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
|
#include <vector>
#include <algorithm>
#include <gtest/gtest.h>
#include "fft_frame.h"
#include "chroma.h"
using namespace chromaprint;
class FeatureVectorBuffer : public FeatureVectorConsumer
{
public:
void Consume(std::vector<double> &features)
{
m_features = features;
}
std::vector<double> m_features;
};
TEST(Chroma, NormalA) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[113] = 1.0;
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
1.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
TEST(Chroma, NormalGSharp) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[112] = 1.0;
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
TEST(Chroma, NormalB) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[64] = 1.0; // 250 Hz
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
0.0, 0.0, 1.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
TEST(Chroma, InterpolatedB) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
chroma.set_interpolate(true);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[64] = 1.0;
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
0.0, 0.286905, 0.713095, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
TEST(Chroma, InterpolatedA) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
chroma.set_interpolate(true);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[113] = 1.0;
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
0.555242, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.444758,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
TEST(Chroma, InterpolatedGSharp) {
FeatureVectorBuffer buffer;
Chroma chroma(10, 510, 256, 1000, &buffer);
chroma.set_interpolate(true);
FFTFrame frame(128);
std::fill(frame.data(), frame.data() + frame.size(), 0.0);
frame.data()[112] = 1.0;
chroma.Consume(frame);
ASSERT_EQ(12, buffer.m_features.size());
double expected_features[12] = {
0.401354, 0.0, 0.0, 0.0, 0.0, 0.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.598646,
};
for (int i = 0; i < 12; i++) {
EXPECT_NEAR(expected_features[i], buffer.m_features[i], 0.0001) << "Different value at index " << i;
}
}
|
