More goertzel improvements

[mirrors/Programs.git] / c / goertzel / goertzel.c

diff --git a/c/goertzel/goertzel.c b/c/goertzel/goertzel.c
index bcf7a919418e15815c2829dfcd53672970b909f2..a26975023f6858367a813670f89d4f2bfc865388 100644 (file)
--- a/c/goertzel/goertzel.c
+++ b/c/goertzel/goertzel.c
@@ -1,10 +1,15 @@
 #include <stdio.h>
 #include <math.h>
+#include <getopt.h>
+
 
 /*
        Usage examples
        arecord | ./goertzel
        sox input.mp3 -b 8 -c 1 -r 8000 -t wav - | ./goertzel
+
+       Arguments for DTMF decoding:
+       -f 697 -f 770 -f 852 -f 941 -f 1209 -f 1336 -f 1477 -f 1633 -t 10
 */
 
 float goertzel_mag(int numSamples,int TARGET_FREQUENCY,int SAMPLING_RATE, float* data)
@@ -48,7 +53,18 @@ float goertzel_mag(int numSamples,int TARGET_FREQUENCY,int SAMPLING_RATE, float*
     return magnitude;
 }
 
-int main() {
+void print_help(char ** argv) {
+       printf("help me %s\n", argv[0]);
+}
+
+void addfreq(int *freqs, int freq) {
+       int i = 0;
+       while(freqs[i]!=-1) i++;
+       freqs[i]=freq;
+       freqs[i+1]=-1;
+}
+
+int main(int argc, char ** argv) {
        /*
        int samples[] = {0,1,2,3,4,5,6,7,8,9,8,7,6,5,4,3,2,1};
        int samplecount = 18;
@@ -58,19 +74,108 @@ int main() {
 
        int samplerate = 8000;
        int samplecount = 4000;
+       int treshold = -1;
+       char noreturn = 0;
+       char repeat = 1;
+       char integers=0;
+       char verbose=1;
+       int freqs[argc+1]; freqs[0]=-1;
+
+       int opt;
+       while ((opt = getopt(argc, argv, "?r:s:f:t:iqna")) != -1) {
+               switch (opt) {
+                       case 'r':
+                               samplerate = atoi(optarg);
+                               break;
+                       case 's':
+                               samplecount = atoi(optarg);
+                               break;
+                       case 'f':
+                               addfreq(freqs, atoi(optarg));
+                               break;
+                       case 't':
+                               treshold = atoi(optarg);
+                               break;
+                       case 'i':
+                               integers = 1;
+                               break;
+                       case 'a':
+                               repeat = 0;
+                               break;
+                       case 'n':
+                               noreturn = 1;
+                               break;
+                       case 'q':
+                               verbose = 0;
+                               break;
+                       case '?':
+                               print_help(argv);
+                               return 0;
+                               break;
+               }
+       }
+
+       if(freqs[0]==-1) addfreq(freqs, 440);
        float samples[samplecount];
        float position = 0;
-       fprintf(stderr,"Position (Secs)\tMagnitude\n");
+
+       if(verbose) {
+               fprintf(stderr,
+                       "#Detected tone: %d Hz\n"
+                       "#Samplerate: %d Hz\n"
+                       "#Frame lenght: %d samples\n"
+                       "#Treshold: %d\n"
+                       "#\n"
+                       ,freqs[0],samplerate,samplecount,treshold);
+               fflush(stderr);
+
+               printf("#Position");
+               int i; for(i=0;freqs[i]!=-1;i++) {
+                       printf("\t%2dHz",freqs[i]);
+               }
+               puts("");
+       }
+
+       char print=0, printnow=0, printlast = 0;
        while(!feof(stdin)) {
                int i;
+
+               //Sample data
                for(i=0;i<samplecount && !feof(stdin);i++) {
                        unsigned char sample;
                        fread(&sample,1,1,stdin);
                        samples[i]=sample;
                        //printf("%d\n", sample);
                }
+
+               //Apply goertzel
+               float power[argc];
+               print=0;
+               for(i=0;freqs[i]!=-1;i++) {
+                       power[i] = goertzel_mag(samplecount, freqs[i], samplerate, samples);
+
+                       //Set print true if over treshold or if changed to false (print for the last time after going under treshold)
+                       printnow = power[i] > treshold;
+                       print = !(!repeat && printlast && !(!printnow)) && (print || printnow || (printlast && !noreturn));
+               }
+               printlast = printnow;
+               fflush(stdout);
+
+               //Print data
+               if(print) {
+                       printf("%8.2f", position);
+                       for(i=0;freqs[i]!=-1;i++) {
+                               printf("\t");
+                               if(integers)
+                                       printf("%d",(int)power[i]);
+                               else
+                                       printf("%.4f",power[i]);
+                       }
+                       puts("");
+                       fflush(stdout);
+               }
+
+               //Increase time
                position += ((float)samplecount/(float)samplerate);
-               float power = goertzel_mag(samplecount, 440, samplerate, samples);
-               printf("%f\t%f\n", position, power);
        }
 }