X-Git-Url: https://git.sesse.net/?p=pitch;a=blobdiff_plain;f=pitch.cpp;h=c8246dcf5f28436277ad51c6cff8bf94f145bc64;hp=13efaf8a5660f9b955c7d9d29cf15e9e9ba03c0c;hb=2bb871a3a6ca4b73e8583de4de97f7b55dd1614c;hpb=2921b00f987d93f2ccc8c918da196158c2f90906

diff --git a/pitch.cpp b/pitch.cpp
index 13efaf8..c8246dc 100644
--- a/pitch.cpp
+++ b/pitch.cpp
@@ -46,7 +46,7 @@ int main()
 	in = reinterpret_cast<double *> (fftw_malloc(sizeof(double) * FFT_LENGTH / PAD_FACTOR));
 	in_windowed = reinterpret_cast<double *> (fftw_malloc(sizeof(double) * FFT_LENGTH));
 	out = reinterpret_cast<std::complex<double> *> (fftw_malloc(sizeof(std::complex<double>) * (FFT_LENGTH / 2 + 1)));
-	bins = reinterpret_cast<double *> (fftw_malloc(sizeof(double) * FFT_LENGTH / 2 + 1));
+	bins = reinterpret_cast<double *> (fftw_malloc(sizeof(double) * (FFT_LENGTH / 2 + 1)));
 
 	memset(in, 0, sizeof(double) * FFT_LENGTH / PAD_FACTOR);
 
@@ -234,6 +234,9 @@ std::pair<double, double> find_peak(double *in, unsigned num_samples)
 		}
 	}
 
+	if (best_bin == 0 || best_bin == num_samples / 2) {
+		return std::make_pair(-1.0, 0.0);
+	}
 	std::pair<double, double> peak = 
 		interpolate_peak(in[best_bin - 1],
 				 in[best_bin],
@@ -248,6 +251,8 @@ std::pair<double, double> find_peak(double *in, unsigned num_samples)
 std::pair<double, double> adjust_for_overtones(std::pair<double, double> base, double *in, unsigned num_samples)
 {
 	double mu = base.first, var = 1.0 / (base.second * base.second);
+		
+	//printf("Base at %.2f (amp=%5.2fdB)\n", base.first, base.second);
 
 	for (unsigned i = 2; i < 10; ++i) {
 		unsigned middle = unsigned(floor(freq_to_bin(base.first, num_samples) * i + 0.5));