Mercurial > hg > match-vamp
changeset 126:0ed1adb2d522 refactors
Chroma feature test case
| author | Chris Cannam |
|---|---|
| date | Thu, 11 Dec 2014 11:41:25 +0000 |
| parents | 4ca5e4219684 |
| children | 2ed42b7616c5 |
| files | Makefile.inc Makefile.linux test/TestFeatureExtractor test/TestFeatureExtractor.cpp |
| diffstat | 4 files changed, 101 insertions(+), 5 deletions(-) [+] |
line wrap: on
line diff
--- a/Makefile.inc Thu Dec 11 11:41:11 2014 +0000 +++ b/Makefile.inc Thu Dec 11 11:41:25 2014 +0000 @@ -6,7 +6,7 @@ CXXFLAGS := -Isrc $(CXXFLAGS) -TEST_LDFLAGS := $(LDFLAGS) -lboost_unit_test_framework +TEST_LDFLAGS += -lboost_unit_test_framework HEADERS := $(wildcard src/*.h) SOURCES := $(wildcard src/*.cpp) @@ -22,8 +22,8 @@ $(PLUGIN): $(OBJECTS) $(CXX) -o $@ $^ $(LDFLAGS) $(PLUGIN_LDFLAGS) -test/%: test/%.o - $(CXX) -o $@ $^ $(TEST_LDFLAGS) +test/%: test/%.o $(OBJECTS) + $(CXX) -o $@ $^ $(LDFLAGS) $(TEST_LDFLAGS) clean: rm -f $(OBJECTS) $(TEST_OBJECTS) $(TESTS)
--- a/Makefile.linux Thu Dec 11 11:41:11 2014 +0000 +++ b/Makefile.linux Thu Dec 11 11:41:25 2014 +0000 @@ -2,8 +2,8 @@ #CXXFLAGS += -fPIC -ffast-math -O3 -Wall -Werror -Wfloat-conversion CXXFLAGS += -fPIC -g -Wall -Werror -DPERFORM_ERROR_CHECKS=1 -LDFLAGS += -PLUGIN_LDFLAGS += -shared -Wl,-Bstatic -lvamp-sdk -Wl,-Bdynamic -Wl,-Bsymbolic -Wl,-z,defs -lpthread -Wl,--version-script=vamp-plugin.map +LDFLAGS += -Wl,-Bstatic -lvamp-sdk -Wl,-Bdynamic +PLUGIN_LDFLAGS += -shared -Wl,-Bsymbolic -Wl,-z,defs -lpthread -Wl,--version-script=vamp-plugin.map include Makefile.inc
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/TestFeatureExtractor.cpp Thu Dec 11 11:41:25 2014 +0000 @@ -0,0 +1,96 @@ + +#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()