From e4ab5141d33d0c94a1c82091a02cfc7ecffc65fd Mon Sep 17 00:00:00 2001
From: John Glover <j@johnglover.net>
Date: Wed, 22 Aug 2012 16:44:52 +0100
Subject: [loris] Add C++ implementation of LorisPartialTracking.

---
 tests/test_partial_tracking.cpp | 65 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 tests/test_partial_tracking.cpp

(limited to 'tests')

diff --git a/tests/test_partial_tracking.cpp b/tests/test_partial_tracking.cpp
new file mode 100644
index 0000000..25fcfff
--- /dev/null
+++ b/tests/test_partial_tracking.cpp
@@ -0,0 +1,65 @@
+#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
+{
+
+// ---------------------------------------------------------------------------
+//	TestLorisPartialTracking
+// ---------------------------------------------------------------------------
+class TestLorisPartialTracking : public CPPUNIT_NS::TestCase {
+    CPPUNIT_TEST_SUITE(TestLorisPartialTracking);
+    CPPUNIT_TEST(test_basic);
+    CPPUNIT_TEST_SUITE_END();
+
+protected:
+    static const double PRECISION = 0.001;
+    LorisPeakDetection* pd;
+    LorisPartialTracking* pt;
+    SndfileHandle sf;
+    int num_samples;
+
+    void test_basic() {
+        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);
+        }
+    }
+
+public:
+    void setUp() {
+        pd = new LorisPeakDetection();
+        pt = new LorisPartialTracking();
+        sf = SndfileHandle("../tests/audio/flute.wav");
+        num_samples = 4096;
+    }
+
+    void tearDown() {
+        delete pd;
+    }
+};
+
+} // end of namespace simpl
+
+CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestLorisPartialTracking);
+
+int main(int arg, char **argv) {
+    CppUnit::TextTestRunner runner;
+    runner.addTest(CppUnit::TestFactoryRegistry::getRegistry().makeTest());
+    return runner.run("", false);
+}
-- 
cgit v1.2.3