summaryrefslogtreecommitdiff
path: root/tests/test_partial_tracking.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_partial_tracking.cpp')
-rw-r--r--tests/test_partial_tracking.cpp496
1 files changed, 225 insertions, 271 deletions
diff --git a/tests/test_partial_tracking.cpp b/tests/test_partial_tracking.cpp
index 86cc878..4d7d7f6 100644
--- a/tests/test_partial_tracking.cpp
+++ b/tests/test_partial_tracking.cpp
@@ -1,308 +1,262 @@
-#include <iostream>
-#include <cppunit/ui/text/TextTestRunner.h>
-#include <cppunit/TestResult.h>
-#include <cppunit/TestResultCollector.h>
-#include <cppunit/extensions/HelperMacros.h>
-#include <cppunit/BriefTestProgressListener.h>
-#include <cppunit/extensions/TestFactoryRegistry.h>
-#include <sndfile.hh>
-
-#include "../src/simpl/base.h"
-#include "../src/simpl/peak_detection.h"
-#include "../src/simpl/partial_tracking.h"
-
-namespace simpl
-{
+#include "test_partial_tracking.h"
+
+using namespace simpl;
// ---------------------------------------------------------------------------
// TestMQPartialTracking
// ---------------------------------------------------------------------------
-class TestMQPartialTracking : public CPPUNIT_NS::TestCase {
- CPPUNIT_TEST_SUITE(TestMQPartialTracking);
- CPPUNIT_TEST(test_basic);
- CPPUNIT_TEST(test_peaks);
- CPPUNIT_TEST_SUITE_END();
-
-protected:
- static const double PRECISION = 0.001;
- MQPeakDetection* pd;
- MQPartialTracking* pt;
- SndfileHandle sf;
- int num_samples;
-
- void test_basic() {
- pt->reset();
- pd->hop_size(256);
- pd->frame_size(2048);
-
- sample* audio = new sample[(int)sf.frames()];
- sf.read(audio, (int)sf.frames());
-
- Frames frames = pd->find_peaks(
- num_samples, &(audio[(int)sf.frames() / 2])
- );
- frames = pt->find_partials(frames);
-
- for(int i = 0; i < frames.size(); i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- }
+void TestMQPartialTracking::setUp() {
+ _sf = SndfileHandle(TEST_AUDIO_FILE);
+
+ if(_sf.error() > 0) {
+ throw Exception(std::string("Could not open audio file: ") +
+ std::string(TEST_AUDIO_FILE));
+ }
+}
+
+void TestMQPartialTracking::test_basic() {
+ int hop_size = 256;
+ int frame_size = 2048;
+ int num_samples = 4096;
+
+ _pd.clear();
+ _pt.reset();
+
+ _pd.hop_size(hop_size);
+ _pd.frame_size(frame_size);
+
+ std::vector<sample> audio(_sf.frames(), 0.0);
+ _sf.read(&audio[0], (int)_sf.frames());
+
+ Frames frames = _pd.find_peaks(num_samples,
+ &(audio[(int)_sf.frames() / 2]));
+ frames = _pt.find_partials(frames);
+
+ for(int i = 0; i < frames.size(); i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
+ }
+}
+
+void TestMQPartialTracking::test_peaks() {
+ int num_frames = 8;
+ Frames frames;
+ Peaks peaks;
+
+ _pd.clear();
+ _pt.reset();
+
+ for(int i = 0; i < num_frames; i++) {
+ Peak* p = new Peak();
+ p->amplitude = 0.2;
+ p->frequency = 220;
+
+ Peak* p2 = new Peak();
+ p2->amplitude = 0.2;
+ p2->frequency = 440;
+
+ Frame* f = new Frame();
+ f->add_peak(p);
+ f->add_peak(p2);
+
+ frames.push_back(f);
+ peaks.push_back(p);
+ peaks.push_back(p2);
}
- void test_peaks() {
- pt->reset();
-
- Frames frames;
- Peaks peaks;
- int num_frames = 8;
-
- for(int i = 0; i < num_frames; i++) {
- Peak* p = new Peak();
- p->amplitude = 0.2;
- p->frequency = 220;
-
- Peak* p2 = new Peak();
- p2->amplitude = 0.2;
- p2->frequency = 440;
-
- Frame* f = new Frame();
- f->add_peak(p);
- f->add_peak(p2);
-
- frames.push_back(f);
- peaks.push_back(p);
- peaks.push_back(p2);
- }
-
- pt->find_partials(frames);
- for(int i = 0; i < num_frames; i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- CPPUNIT_ASSERT(frames[i]->partial(0)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(0)->frequency == 220);
- CPPUNIT_ASSERT(frames[i]->partial(1)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(1)->frequency == 440);
- }
-
- for(int i = 0; i < num_frames * 2; i++) {
- delete peaks[i];
- }
-
- for(int i = 0; i < num_frames; i++) {
- delete frames[i];
- }
+ _pt.find_partials(frames);
+ for(int i = 0; i < num_frames; i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(0)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(220, frames[i]->partial(0)->frequency,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(1)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(440, frames[i]->partial(1)->frequency,
+ PRECISION);
}
-public:
- void setUp() {
- pd = new MQPeakDetection();
- pt = new MQPartialTracking();
- sf = SndfileHandle("../tests/audio/flute.wav");
- num_samples = 4096;
+ for(int i = 0; i < num_frames * 2; i++) {
+ delete peaks[i];
}
- void tearDown() {
- delete pd;
- delete pt;
+ for(int i = 0; i < num_frames; i++) {
+ delete frames[i];
}
-};
+}
// ---------------------------------------------------------------------------
// TestSMSPartialTracking
// ---------------------------------------------------------------------------
-class TestSMSPartialTracking : public CPPUNIT_NS::TestCase {
- CPPUNIT_TEST_SUITE(TestSMSPartialTracking);
- CPPUNIT_TEST(test_basic);
- CPPUNIT_TEST(test_peaks);
- CPPUNIT_TEST_SUITE_END();
-
-protected:
- static const double PRECISION = 0.001;
- SMSPeakDetection* pd;
- SMSPartialTracking* pt;
- SndfileHandle sf;
- int num_samples;
-
- void test_basic() {
- pt->reset();
- pd->hop_size(256);
- pd->frame_size(2048);
-
- sample* audio = new sample[(int)sf.frames()];
- sf.read(audio, (int)sf.frames());
-
- Frames frames = pd->find_peaks(
- num_samples, &(audio[(int)sf.frames() / 2])
- );
- frames = pt->find_partials(frames);
-
- for(int i = 0; i < frames.size(); i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- }
+void TestSMSPartialTracking::setUp() {
+ _sf = SndfileHandle(TEST_AUDIO_FILE);
+
+ if(_sf.error() > 0) {
+ throw Exception(std::string("Could not open audio file: ") +
+ std::string(TEST_AUDIO_FILE));
}
- void test_peaks() {
- pt->reset();
-
- Frames frames;
- Peaks peaks;
- int num_frames = 8;
-
- for(int i = 0; i < num_frames; i++) {
- Peak* p = new Peak();
- p->amplitude = 0.2;
- p->frequency = 220;
-
- Peak* p2 = new Peak();
- p2->amplitude = 0.2;
- p2->frequency = 440;
-
- Frame* f = new Frame();
- f->add_peak(p);
- f->add_peak(p2);
-
- frames.push_back(f);
- peaks.push_back(p);
- peaks.push_back(p2);
- }
-
- pt->find_partials(frames);
- for(int i = 0; i < num_frames; i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- CPPUNIT_ASSERT(frames[i]->partial(0)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(0)->frequency == 220);
- CPPUNIT_ASSERT(frames[i]->partial(1)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(1)->frequency == 440);
- }
-
- for(int i = 0; i < num_frames * 2; i++) {
- delete peaks[i];
- }
-
- for(int i = 0; i < num_frames; i++) {
- delete frames[i];
- }
+ _pt.realtime(1);
+}
+
+void TestSMSPartialTracking::test_basic() {
+ int hop_size = 256;
+ int frame_size = 2048;
+ int num_samples = 4096;
+
+ _pd.clear();
+ _pt.reset();
+
+ _pd.hop_size(hop_size);
+ _pd.frame_size(frame_size);
+
+ std::vector<sample> audio(_sf.frames(), 0.0);
+ _sf.read(&audio[0], (int)_sf.frames());
+
+ Frames frames = _pd.find_peaks(num_samples,
+ &(audio[(int)_sf.frames() / 2]));
+ frames = _pt.find_partials(frames);
+
+ for(int i = 0; i < frames.size(); i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
}
+}
-public:
- void setUp() {
- pd = new SMSPeakDetection();
- pt = new SMSPartialTracking();
- pt->realtime(1);
- sf = SndfileHandle("../tests/audio/flute.wav");
- num_samples = 4096;
+void TestSMSPartialTracking::test_peaks() {
+ int num_frames = 8;
+ Frames frames;
+ Peaks peaks;
+
+ _pd.clear();
+ _pt.reset();
+
+ for(int i = 0; i < num_frames; i++) {
+ Peak* p = new Peak();
+ p->amplitude = 0.2;
+ p->frequency = 220;
+
+ Peak* p2 = new Peak();
+ p2->amplitude = 0.2;
+ p2->frequency = 440;
+
+ Frame* f = new Frame();
+ f->add_peak(p);
+ f->add_peak(p2);
+
+ frames.push_back(f);
+ peaks.push_back(p);
+ peaks.push_back(p2);
+ }
+
+ _pt.find_partials(frames);
+ for(int i = 0; i < num_frames; i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(0)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(220, frames[i]->partial(0)->frequency,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(1)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(440, frames[i]->partial(1)->frequency,
+ PRECISION);
}
- void tearDown() {
- delete pd;
- delete pt;
+ for(int i = 0; i < num_frames * 2; i++) {
+ delete peaks[i];
}
-};
+
+ for(int i = 0; i < num_frames; i++) {
+ delete frames[i];
+ }
+}
+
// ---------------------------------------------------------------------------
// TestLorisPartialTracking
// ---------------------------------------------------------------------------
-class TestLorisPartialTracking : public CPPUNIT_NS::TestCase {
- CPPUNIT_TEST_SUITE(TestLorisPartialTracking);
- CPPUNIT_TEST(test_basic);
- CPPUNIT_TEST(test_peaks);
- CPPUNIT_TEST_SUITE_END();
-
-protected:
- static const double PRECISION = 0.001;
- LorisPeakDetection* pd;
- LorisPartialTracking* pt;
- SndfileHandle sf;
- int num_samples;
-
- void test_basic() {
- pt->reset();
- pd->hop_size(256);
- pd->frame_size(2048);
-
- sample* audio = new sample[(int)sf.frames()];
- sf.read(audio, (int)sf.frames());
-
- Frames frames = pd->find_peaks(
- num_samples, &(audio[(int)sf.frames() / 2])
- );
- frames = pt->find_partials(frames);
-
- for(int i = 0; i < frames.size(); i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- }
- }
+void TestLorisPartialTracking::setUp() {
+ _sf = SndfileHandle(TEST_AUDIO_FILE);
- void test_peaks() {
- pt->reset();
-
- Frames frames;
- Peaks peaks;
- int num_frames = 8;
-
- for(int i = 0; i < num_frames; i++) {
- Peak* p = new Peak();
- p->amplitude = 0.2;
- p->frequency = 220;
-
- Peak* p2 = new Peak();
- p2->amplitude = 0.2;
- p2->frequency = 440;
-
- Frame* f = new Frame();
- f->add_peak(p);
- f->add_peak(p2);
-
- frames.push_back(f);
- peaks.push_back(p);
- peaks.push_back(p2);
- }
-
- pt->find_partials(frames);
- for(int i = 0; i < num_frames; i++) {
- CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
- CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
- CPPUNIT_ASSERT(frames[i]->partial(0)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(0)->frequency == 220);
- CPPUNIT_ASSERT(frames[i]->partial(1)->amplitude == 0.2);
- CPPUNIT_ASSERT(frames[i]->partial(1)->frequency == 440);
- }
-
- for(int i = 0; i < num_frames * 2; i++) {
- delete peaks[i];
- }
-
- for(int i = 0; i < num_frames; i++) {
- delete frames[i];
- }
+ if(_sf.error() > 0) {
+ throw Exception(std::string("Could not open audio file: ") +
+ std::string(TEST_AUDIO_FILE));
}
+}
+
+void TestLorisPartialTracking::test_basic() {
+ int hop_size = 256;
+ int frame_size = 2048;
+ int num_samples = 4096;
-public:
- void setUp() {
- pd = new LorisPeakDetection();
- pt = new LorisPartialTracking();
- sf = SndfileHandle("../tests/audio/flute.wav");
- num_samples = 4096;
+ _pd.clear();
+ _pt.reset();
+
+ _pd.hop_size(hop_size);
+ _pd.frame_size(frame_size);
+
+ std::vector<sample> audio(_sf.frames(), 0.0);
+ _sf.read(&audio[0], (int)_sf.frames());
+
+ Frames frames = _pd.find_peaks(num_samples,
+ &(audio[(int)_sf.frames() / 2]));
+ frames = _pt.find_partials(frames);
+
+ for(int i = 0; i < frames.size(); i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
}
+}
+
+void TestLorisPartialTracking::test_peaks() {
+ int num_frames = 8;
+ Frames frames;
+ Peaks peaks;
+
+ _pd.clear();
+ _pt.reset();
- void tearDown() {
- delete pd;
- delete pt;
+ for(int i = 0; i < num_frames; i++) {
+ Peak* p = new Peak();
+ p->amplitude = 0.2;
+ p->frequency = 220;
+
+ Peak* p2 = new Peak();
+ p2->amplitude = 0.2;
+ p2->frequency = 440;
+
+ Frame* f = new Frame();
+ f->add_peak(p);
+ f->add_peak(p2);
+
+ frames.push_back(f);
+ peaks.push_back(p);
+ peaks.push_back(p2);
}
-};
-} // end of namespace simpl
+ _pt.find_partials(frames);
+ for(int i = 0; i < num_frames; i++) {
+ CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
+ CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(0)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(220, frames[i]->partial(0)->frequency,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(0.2, frames[i]->partial(1)->amplitude,
+ PRECISION);
+ CPPUNIT_ASSERT_DOUBLES_EQUAL(440, frames[i]->partial(1)->frequency,
+ PRECISION);
+ }
-CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestMQPartialTracking);
-CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestSMSPartialTracking);
-CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestLorisPartialTracking);
+ for(int i = 0; i < num_frames * 2; i++) {
+ delete peaks[i];
+ }
-int main(int arg, char **argv) {
- CppUnit::TextTestRunner runner;
- runner.addTest(CppUnit::TestFactoryRegistry::getRegistry().makeTest());
- return runner.run("", false);
+ for(int i = 0; i < num_frames; i++) {
+ delete frames[i];
+ }
}
generated by cgit v1.2.3 (git 2.25.1) at 2025-04-05 09:35:23 +0000