blob: fa49016e6f643b9cb755fc90da2b12cf8888719f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#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"
#include "../src/simpl/synthesis.h"
namespace simpl
{
// ---------------------------------------------------------------------------
// TestMQSynthesis
// ---------------------------------------------------------------------------
class TestMQSynthesis : public CPPUNIT_NS::TestCase {
CPPUNIT_TEST_SUITE(TestMQSynthesis);
CPPUNIT_TEST(test_basic);
CPPUNIT_TEST_SUITE_END();
protected:
static const double PRECISION = 0.001;
MQPeakDetection* pd;
MQPartialTracking* pt;
MQSynthesis* synth;
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);
frames = synth->synth(frames);
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
double energy = 0.f;
for(int j = 0; j < synth->hop_size(); j++) {
energy += frames[i]->synth()[j] * frames[i]->synth()[j];
}
CPPUNIT_ASSERT(energy > 0.f);
}
}
public:
void setUp() {
pd = new MQPeakDetection();
pt = new MQPartialTracking();
synth = new MQSynthesis();
sf = SndfileHandle("../tests/audio/flute.wav");
num_samples = 4096;
}
void tearDown() {
delete pd;
delete pt;
delete synth;
}
};
// ---------------------------------------------------------------------------
// TestLorisSynthesis
// ---------------------------------------------------------------------------
class TestLorisSynthesis : public CPPUNIT_NS::TestCase {
CPPUNIT_TEST_SUITE(TestLorisSynthesis);
CPPUNIT_TEST(test_basic);
CPPUNIT_TEST_SUITE_END();
protected:
static const double PRECISION = 0.001;
LorisPeakDetection* pd;
LorisPartialTracking* pt;
LorisSynthesis* synth;
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);
frames = synth->synth(frames);
for(int i = 0; i < frames.size(); i++) {
// if Loris thinPeaks is used, final frame will have no peaks
// so don't check it
if(i < frames.size() - 1) {
CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
CPPUNIT_ASSERT(frames[i]->num_partials() > 0);
}
double energy = 0.f;
for(int j = 0; j < synth->hop_size(); j++) {
energy += frames[i]->synth()[j] * frames[i]->synth()[j];
}
CPPUNIT_ASSERT(energy > 0.f);
}
}
public:
void setUp() {
pd = new LorisPeakDetection();
pt = new LorisPartialTracking();
synth = new LorisSynthesis();
sf = SndfileHandle("../tests/audio/flute.wav");
num_samples = 4096;
}
void tearDown() {
delete pd;
delete pt;
delete synth;
}
};
} // end of namespace simpl
CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestMQSynthesis);
CPPUNIT_TEST_SUITE_REGISTRATION(simpl::TestLorisSynthesis);
int main(int arg, char **argv) {
CppUnit::TextTestRunner runner;
runner.addTest(CppUnit::TestFactoryRegistry::getRegistry().makeTest());
return runner.run("", false);
}
|