comparison projects/tank_wars/main.cpp @ 22:fbfeb5895efd matrix_gpio

Updated tank wars demo for new API

 #include <iostream>
 #include <cstdlib>
 #include <libgen.h>
 #include <signal.h>
 #include <getopt.h>
+#include <sndfile.h>
 #include "../../include/RTAudio.h"
 
 extern int gScreenFramesPerSecond;
 
+float *gMusicBuffer = 0;
+int gMusicBufferLength = 0;
+float *gSoundBoomBuffer = 0;
+int gSoundBoomBufferLength = 0;
+
 using namespace std;
+
+// Load a sound sample from file
+int loadSoundFile(const string& path, float **buffer, int *bufferLength)
+{
+	SNDFILE *sndfile ;
+	SF_INFO sfinfo ;
+
+	if (!(sndfile = sf_open (path.c_str(), SFM_READ, &sfinfo))) {
+		cout << "Couldn't open file " << path << endl;
+		return 1;
+	}
+
+	int numChan = sfinfo.channels;
+	if(numChan != 1)
+	{
+		cout << "Error: " << path << " is not a mono file" << endl;
+		return 1;
+	}
+
+	*bufferLength = sfinfo.frames * numChan;
+	*buffer = new float[*bufferLength];
+	if(*buffer == 0){
+		cout << "Could not allocate buffer" << endl;
+		return 1;
+	}
+
+	int subformat = sfinfo.format & SF_FORMAT_SUBMASK;
+	int readcount = sf_read_float(sndfile, *buffer, *bufferLength);
+
+	// Pad with zeros in case we couldn't read whole file
+	for(int k = readcount; k < *bufferLength; k++)
+		(*buffer)[k] = 0;
+
+	sf_close(sndfile);
+	return 0;
+}
 
 // Handle Ctrl-C by requesting that the audio rendering stop
 void interrupt_handler(int var)
 {
 	gShouldStop = true;

 }
 
 int main(int argc, char *argv[])
 {
 	RTAudioSettings settings;	// Standard audio settings
+	string musicFileName = "music.wav";
+	string soundBoomFileName = "boom.wav";
 
 	struct option customOptions[] =
 	{
 		{"help", 0, NULL, 'h'},
 		{"fps", 1, NULL, 'f'},

 				usage(basename(argv[0]));
 				exit(1);
 		}
 	}
 
+	// Load the sound files
+	if(loadSoundFile(musicFileName, &gMusicBuffer, &gMusicBufferLength) != 0) {
+		cout << "Warning: unable to load sound file " << musicFileName << endl;
+	}
+	if(loadSoundFile(soundBoomFileName, &gSoundBoomBuffer, &gSoundBoomBufferLength) != 0) {
+		cout << "Warning: unable to load sound file " << soundBoomFileName << endl;
+	}
+
 	// Initialise the PRU audio device
 	if(BeagleRT_initAudio(&settings, 0) != 0) {
 		cout << "Error: unable to initialise audio" << endl;
 		return -1;
 	}

 	BeagleRT_stopAudio();
 
 	// Clean up any resources allocated for audio
 	BeagleRT_cleanupAudio();
 
+	// Release sound files
+	if(gMusicBuffer != 0)
+		free(gMusicBuffer);
+	if(gSoundBoomBuffer != 0)
+		free(gSoundBoomBuffer);
+
 	// All done!
 	return 0;
 }