Fix another missing #include of string.h.
[pitch] / pitchdetector.cpp
index cf0abe0..a73fb86 100644 (file)
@@ -1,4 +1,5 @@
 #include <stdio.h>
+#include <string.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <fcntl.h>
@@ -80,6 +81,13 @@ std::pair<double, double> PitchDetector::find_peak(double *in, unsigned num_samp
                        best_peak = in[i];
                        best_bin = i;
                }
+#if 0
+               if (20.0 * log10(in[i] / fft_length) > 0.0) {
+                       printf("PEAK: %+4.2f dB   %5.2f Hz\n",
+                               20.0 * log10(in[i] / fft_length),
+                               bin_to_freq(i, num_samples));
+               }
+#endif
        }
        
        if (best_bin == 0 || best_bin == num_samples / 2) {
@@ -97,10 +105,11 @@ std::pair<double, double> PitchDetector::find_peak(double *in, unsigned num_samp
                20.0 * log10(in[best_bin/4] / fft_length));
 #endif
 
-       // see if we might have hit an overtone (set a limit of 5dB)
-       for (unsigned i = 4; i >= 1; --i) {
+       // see if we might have hit an overtone (set a limit of 10dB)
+       for (unsigned i = 6; i >= 1; --i) {
                if (best_bin != best_bin / i &&
-                   20.0 * log10(in[best_bin] / in[best_bin / i]) < 5.0f) {
+                   20.0 * log10(in[best_bin] / in[best_bin / i]) < 10.0f &&
+                   best_bin / i >= 5) {
 #if 0
                        printf("Overtone of degree %u!\n", i);
 #endif
@@ -244,7 +253,7 @@ std::pair<double, double> PitchDetector::interpolate_peak(double ym1, double y0,
        double c = y0;
        
        double xmax = (ym1 - y1) / (2.0 * (y1 + ym1 - 2.0 * y0));
-       double ymax = 20.0 * (a * xmax * xmax + b * xmax + c) - 90.0;
+       double ymax = 20.0 * (a * xmax * xmax + b * xmax + c) - 70.0;
 
        return std::make_pair(xmax, ymax);
 }