comparison projects/filter_IIR/main.cpp @ 5:09f03ac40fcc

API improvements and cleanups. Now all common audio command-line options can be parsed automatically.

 #include <iostream>
 #include <cstdlib>
 #include <libgen.h>
 #include <signal.h>
 #include <string>
+#include <getopt.h>
 #include <sndfile.h>				// to load audio files
 #include "../../include/RTAudio.h"
 #include "SampleData.h"
 
 using namespace std;

 	sf_close(sndfile);
 
 	return 0;
 }
 
-
 // Handle Ctrl-C by requesting that the audio rendering stop
 void interrupt_handler(int var)
 {
 	//rt_task_delete ((RT_TASK *) &gTriggerSamplesTask);
 	gShouldStop = true;
 }
 
 // Print usage information
 void usage(const char * processName)
 {
-	cerr << "Usage: " << processName << " [-h] [-v] [-p period] [-f frequency]" << endl;
-	cerr << "   -h:           Print this menu\n";
-	cerr << "   -v:           Enable verbose messages\n";
-	cerr << "   -p period:    Set the period (hardware buffer) size in sensor frames\n";
-	cerr << "   -m:           Enable the matrix (ADC and DAC) as well as audio\n";
-	cerr << "   -f filename:  Name of the file to load (default is \"sample.wav\")\n";
-	cerr << "   -c freq:      Set the cut off frequency of the filter in Hz\n";
+	cerr << "Usage: " << processName << " [options]" << endl;
+
+	BeagleRT_usage();
+
+	cerr << "   --file [-f] filename:    Name of the file to load (default is \"longsample.wav\")\n";
+	cerr << "   --cutfreq [-c] freq:     Set the cut off frequency of the filter in Hz\n";
+	cerr << "   --help [-h]:             Print this menu\n";
 }
 
 int main(int argc, char *argv[])
 {
-	int verbose = 0;			// Verbose printing level
-	int periodSize = 8;			// Period size in sensor frames
-	int useMatrix = 0;			// Whether to use the matrix or just audio
+	RTAudioSettings settings;	// Standard audio settings
 	string fileName;			// Name of the sample to load
 
 	SampleData sampleData;		// User define structure to pass data retrieved from file to render function
 	sampleData.samples = 0;
 	sampleData.sampleLen = -1;
 
+
+	struct option customOptions[] =
+	{
+		{"help", 0, NULL, 'h'},
+		{"cutfreq", 1, NULL, 'c'},
+		{"file", 1, NULL, 'f'},
+		{NULL, 0, NULL, 0}
+	};
+
+	// Set default settings
+	BeagleRT_defaultSettings(&settings);
+
 	// Parse command-line arguments
 	while (1) {
 		int c;
-		if ((c = getopt(argc, argv, "hp:vms:")) < 0)
+		if ((c = BeagleRT_getopt_long(argc, argv, "hf:c:", customOptions, &settings)) < 0)
 				break;
 		switch (c) {
 		case 'h':
 				usage(basename(argv[0]));
 				exit(0);
-		case 'p':
-				periodSize = atoi(optarg);
-				if(periodSize < 1)
-					periodSize = 1;
-				break;
-		case 'v':
-				verbose = 1;
-				break;
-		case 'm':
-				useMatrix = 1;
-				break;
 		case 'f':
 				fileName = string((char *)optarg);
 				break;
 		case 'c':
 				gCutFreq = atof(optarg);

 				exit(1);
 		}
 	}
 
 	if(fileName.empty()){
-		fileName = "filter/longsample.wav";
+		fileName = "longsample.wav";
 	}
 
-	// Set verbose logging information (optional by using value > 0; default is 0)
-	setVerboseLevel(verbose);
-
-	if(verbose) {
-		cout << "Starting with period size " << periodSize << " and cut-off frequency" << gCutFreq << endl;
-		if(useMatrix)
-			cout << "Matrix enabled\n";
-		else
-			cout << "Matrix disabled\n";
+	if(settings.verbose) {
 		cout << "Loading file " << fileName << endl;
 	}
 
 	// Load file
 	if(initFile(fileName, &sampleData) != 0)
 	{
 		cout << "Error: unable to load samples " << endl;
 		return -1;
 	}
 
-	if(verbose)
+	if(settings.verbose)
 		cout << "File contains " << sampleData.sampleLen << " samples" << endl;
 
+
 	// Initialise the PRU audio device
-	if(initAudio(periodSize, useMatrix, &sampleData) != 0) {
+	if(BeagleRT_initAudio(&settings, &sampleData) != 0) {
 		cout << "Error: unable to initialise audio" << endl;
 		return -1;
 	}
 
 	// Start the audio device running
-	if(startAudio()) {
+	if(BeagleRT_startAudio()) {
 		cout << "Error: unable to start real-time audio" << endl;
 		return -1;
 	}
 
 	// Set up interrupt handler to catch Control-C

 	while(!gShouldStop) {
 		usleep(100000);
 	}
 
 	// Stop the audio device
-	stopAudio();
-
-	if(verbose) {
-		cout << "Cleaning up..." << endl;
-	}
+	BeagleRT_stopAudio();
 
 	// Clean up any resources allocated for audio
-	cleanupAudio();
+	BeagleRT_cleanupAudio();
 
 	// All done!
 	return 0;
 }
+