comparison spectrum-compare-plugin/TuningDifference.cpp @ 14:812e4d021443

Merge

     int targetBinMin = int(floor(targetFmin * m_blockSize / m_inputSampleRate));
     int targetBinMax = int(ceil(targetFmax * m_blockSize / m_inputSampleRate));
     cerr << "target bin range: " << targetBinMin << " -> " << targetBinMax << endl;
 
     d.identifier = "averages";
-    d.name = "Spectrum averages";
-    d.description = "Average magnitude spectrum for each channel.";
+    d.name = "Harmonic spectrum averages";
+    d.description = "Average of harmonic spectrum for each channel.";
     d.unit = "";
     d.hasFixedBinCount = true;
     d.binCount = (targetBinMax > targetBinMin ? targetBinMax - targetBinMin + 1 : 100);
     d.hasKnownExtents = false;
     d.isQuantized = false;

 {
     for (int c = 0; c < 2; ++c) {
 	if (m_sum[c].size() == 0) {
 	    m_sum[c].resize(m_blockSize/2 + 1, 0.0);
 	}
+	vector<double> mags(m_blockSize/2 + 1, 0.0);
 	for (int i = 0; i <= m_blockSize/2; ++i) {
 	    double energy =
 		inputBuffers[c][i*2  ] * inputBuffers[c][i*2  ] +
 		inputBuffers[c][i*2+1] * inputBuffers[c][i*2+1];
 	    double mag = sqrt(energy);
-	    m_sum[c][i] += mag;
-	    m_sum[c][i/2] += mag;
+	    mags[i] = mag;
+	}
+	for (int i = 0; i <= m_blockSize/2; ++i) {
+	    for (int h = 2; h <= 4; ++h) {
+		double max = 0.0;
+		for (int j = 0; j <= h; ++j) {
+		    int ix = i * h + j - h/2;
+		    if (ix <= m_blockSize/2) {
+			double harmMag = mags[ix];
+			if (j == 0 || harmMag > max) {
+			    max = harmMag;
+			}
+		    }
+		}
+		mags[i] += max;
+	    }
+	}
+	for (int i = 0; i <= m_blockSize/2; ++i) {
+	    m_sum[c][i] += mags[i];
 	}
     }
     
     ++m_frameCount;
     return FeatureSet();

 	m_sum[0][i] /= m_frameCount;
 	m_sum[1][i] /= m_frameCount;
     }
 
     for (int c = 0; c < 2; ++c) {
+	for (int i = 0; i < n; ++i) {
+	    if (i == 0 || i == n-1 ||
+		m_sum[c][i] < m_sum[c][i-1] ||
+		m_sum[c][i] < m_sum[c][i+1]) {
+		m_sum[c][i] = 0.0;
+	    }
+	}
+    }
+
+    int targetBinMin = int(floor(targetFmin * m_blockSize / m_inputSampleRate));
+    int targetBinMax = int(ceil(targetFmax * m_blockSize / m_inputSampleRate));
+    cerr << "target bin range: " << targetBinMin << " -> " << targetBinMax << endl;
+
+    for (int c = 0; c < 2; ++c) {
 	double tot = 0.0;
-	for (int i = 0; i < n; ++i) {
+	for (int i = targetBinMin; i <= targetBinMax; ++i) {
 	    tot += m_sum[c][i];
 	}
 	if (tot != 0.0) {
 	    for (int i = 0; i < n; ++i) {
 		m_sum[c][i] /= tot;
 	    }
 	}
     }
-
-    int targetBinMin = int(floor(targetFmin * m_blockSize / m_inputSampleRate));
-    int targetBinMax = int(ceil(targetFmax * m_blockSize / m_inputSampleRate));
-    cerr << "target bin range: " << targetBinMin << " -> " << targetBinMax << endl;
 	
     f.values.clear();
     for (int i = targetBinMin; i < targetBinMax; ++i) {
 	f.values.push_back(m_sum[0][i]);
     }

     f.timestamp = Vamp::RealTime::zeroTime;
     f.hasTimestamp = true;
     f.label = "";
 
     f.values.clear();
-//    cerr << "best dist = " << bestdist << " at shift " << bestshift << endl;
+    cerr << "best dist = " << bestdist << " at shift " << bestshift << endl;
     f.values.push_back(-bestshift);
     fs[0].push_back(f);
 
     f.values.clear();
     f.values.push_back(440.0 / pow(2.0, double(bestshift) / 1200.0));