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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#include "test_peak_detection.h"
using namespace simpl;
// ---------------------------------------------------------------------------
// TestMQPeakDetection
// ---------------------------------------------------------------------------
void TestMQPeakDetection::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 TestMQPeakDetection::test_find_peaks_in_frame_basic() {
int frame_size = 2048;
_pd.clear();
_pd.frame_size(frame_size);
Frame f = Frame(frame_size, true);
_pd.find_peaks_in_frame(&f);
CPPUNIT_ASSERT(f.num_peaks() == 0);
}
void TestMQPeakDetection::test_find_peaks_basic() {
int frame_size = 512;
std::vector<sample> audio(frame_size * 2, 0.0);
_pd.clear();
_pd.frame_size(frame_size);
Frames frames = _pd.find_peaks(audio.size(), &audio[0]);
CPPUNIT_ASSERT(frames.size() == 2);
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
}
}
void TestMQPeakDetection::test_find_peaks_audio() {
int num_frames = 5;
int num_samples = _pd.frame_size() + (_pd.hop_size() * num_frames);
std::vector<sample> audio(_sf.frames(), 0.0);
_sf.read(&audio[0], (int)_sf.frames());
_pd.clear();
Frames frames = _pd.find_peaks(num_samples,
&(audio[(int)_sf.frames() / 2]));
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
}
}
void TestMQPeakDetection::test_find_peaks_change_hop_frame_size() {
int num_samples = 1024;
std::vector<sample> audio(num_samples, 0.0);
_pd.clear();
_pd.frame_size(256);
_pd.hop_size(256);
Frames frames = _pd.find_peaks(num_samples, &audio[0]);
CPPUNIT_ASSERT(frames.size() == 4);
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
}
}
// ---------------------------------------------------------------------------
// TestTWM
// ---------------------------------------------------------------------------
void TestTWM::test_basic() {
int num_peaks = 100;
int base_freq = 110;
Peaks peaks;
for(int i = 0; i < num_peaks; i++) {
Peak* p = new Peak();
p->amplitude = 0.4;
p->frequency = base_freq * (i + 1);
peaks.push_back(p);
}
CPPUNIT_ASSERT_DOUBLES_EQUAL(base_freq, twm(peaks), PRECISION);
for(int i = 0; i < num_peaks; i++) {
delete peaks[i];
}
}
// ---------------------------------------------------------------------------
// TestLorisPeakDetection
// ---------------------------------------------------------------------------
void TestLorisPeakDetection::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 TestLorisPeakDetection::test_find_peaks_in_frame_basic() {
int frame_size = 2048;
_pd.clear();
_pd.frame_size(frame_size);
Frame f = Frame(frame_size, true);
_pd.find_peaks_in_frame(&f);
CPPUNIT_ASSERT(f.num_peaks() == 0);
}
void TestLorisPeakDetection::test_find_peaks_basic() {
int frame_size = 512;
std::vector<sample> audio(frame_size * 2, 0.0);
_pd.clear();
_pd.frame_size(frame_size);
Frames frames = _pd.find_peaks(audio.size(), &audio[0]);
CPPUNIT_ASSERT(frames.size() == 2);
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
}
}
void TestLorisPeakDetection::test_find_peaks_audio() {
int num_frames = 5;
int num_samples = _pd.frame_size() + (_pd.hop_size() * num_frames);
std::vector<sample> audio(_sf.frames(), 0.0);
_sf.read(&audio[0], (int)_sf.frames());
_pd.clear();
Frames frames = _pd.find_peaks(num_samples,
&(audio[(int)_sf.frames() / 2]));
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
}
}
void TestLorisPeakDetection::test_find_peaks_change_hop_frame_size() {
int num_samples = 1024;
std::vector<sample> audio(num_samples, 0.0);
_pd.clear();
_pd.frame_size(256);
_pd.hop_size(256);
Frames frames = _pd.find_peaks(num_samples, &audio[0]);
CPPUNIT_ASSERT(frames.size() == 4);
for(int i = 0; i < frames.size(); i++) {
CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
}
}
|