comparison test/TestFeatureExtractor.cpp @ 126:0ed1adb2d522 refactors

Chroma feature test case

+
+#include "FeatureExtractor.h"
+
+#include <vector>
+#include <iostream>
+#include <cmath>
+
+using namespace std;
+
+#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MAIN
+
+#include <boost/test/unit_test.hpp>
+
+static int freq2mid(double freq)
+{
+    return round(57.0 + 12.0 * log(freq / 220.) / log(2.));
+}
+
+static int freq2chroma(double freq)
+{
+    return freq2mid(freq) % 12;
+}
+
+BOOST_AUTO_TEST_SUITE(TestFeatureExtractor)
+
+BOOST_AUTO_TEST_CASE(chroma)
+{
+    int rate = 44100;
+    int sz = 2048;
+    int hs = sz / 2 + 1;
+    int fsz = 13;
+    
+    FeatureExtractor::Parameters params(rate, sz);
+    params.useChromaFrequencyMap = true;
+    FeatureExtractor fe(params);
+    BOOST_CHECK_EQUAL(fe.getFeatureSize(), fsz);
+
+    for (int bin = 0; bin < hs; ++bin) {
+
+	vector<double> real, imag;
+	real.resize(hs, 0.0);
+	imag.resize(hs, 0.0);
+	
+	real[bin] += 10.0;
+	imag[bin] += 10.0;
+
+	// use two input sweeps, so we can test that they are properly
+	// summed into the output bin
+	real[hs-bin-1] += 5.0;
+	imag[hs-bin-1] += 5.0;
+
+	vector<double> out = fe.process(real, imag);
+
+	// We expect to find all bins are 0 except for:
+	// 
+	// * two bins of 200 and 50 respectively, if the two input
+	// bins are distinct and their output chroma are also distinct
+	//
+	// * one bin of value 250 (= 10^2 + 5^2), if the two input
+	// bins are distinct but their output chroma are not
+	//
+	// * one bin of value 450 (= 15^2 + 15^2), if the input bins
+	// are not distinct
+	// 
+	// The bin corresponding to each input frequency is that of
+	// its semitone value (with C=0), except that input bin
+	// frequencies less than 362Hz are shepherded into the
+	// separate bin 0 (see docs in FeatureExtractor.h)
+
+	vector<double> expected(fsz);
+
+	double infreq1 = (double(bin) * rate) / sz;
+
+	if (bin == hs-bin-1) {
+	    expected[freq2chroma(infreq1) + 1] += 450;
+	} else {
+	    if (infreq1 < 362) {
+		expected[0] += 200;
+	    } else {
+		expected[freq2chroma(infreq1) + 1] += 200;
+	    }
+	    double infreq2 = (double(hs-bin-1) * rate) / sz;
+	    if (infreq2 < 362) {
+		expected[0] += 50;
+	    } else {
+		expected[freq2chroma(infreq2) + 1] += 50;
+	    }
+	}
+
+	BOOST_CHECK_EQUAL_COLLECTIONS(out.begin(), out.end(),
+				      expected.begin(), expected.end());
+    }
+}
+
+BOOST_AUTO_TEST_SUITE_END()