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
|
#include <gtest/gtest.h>
#include <QtDebug>
#include <QDir>
#include "test/mixxxtest.h"
#include "analyzer/analyzerwaveform.h"
#include "library/dao/analysisdao.h"
#include "track/track.h"
#define BIGBUF_SIZE (1024 * 1024) //Megabyte
#define CANARY_SIZE (1024*4)
#define MAGIC_FLOAT 1234.567890f
#define CANARY_FLOAT 0.0f
namespace {
class AnalyzerWaveformTest: public MixxxTest {
protected:
AnalyzerWaveformTest()
: analysisDao(config()),
aw(&analysisDao),
bigbuf(nullptr),
canaryBigBuf(nullptr) {
}
void SetUp() override {
tio = Track::newTemporary();
tio->setSampleRate(44100);
bigbuf = new CSAMPLE[BIGBUF_SIZE];
for (int i = 0; i < BIGBUF_SIZE; i++)
bigbuf[i] = MAGIC_FLOAT;
//Memory layout for canaryBigBuf looks like
// [ canary | big buf | canary ]
canaryBigBuf = new CSAMPLE[BIGBUF_SIZE + 2*CANARY_SIZE];
for (int i = 0; i < CANARY_SIZE; i++)
canaryBigBuf[i] = CANARY_FLOAT;
for (int i = CANARY_SIZE; i < CANARY_SIZE+BIGBUF_SIZE; i++)
canaryBigBuf[i] = MAGIC_FLOAT;
for (int i = CANARY_SIZE+BIGBUF_SIZE; i < 2*CANARY_SIZE+BIGBUF_SIZE; i++)
canaryBigBuf[i] = CANARY_FLOAT;
}
void TearDown() override {
delete [] bigbuf;
delete [] canaryBigBuf;
}
protected:
AnalysisDao analysisDao;
AnalyzerWaveform aw;
TrackPointer tio;
CSAMPLE* bigbuf;
CSAMPLE* canaryBigBuf;
};
//Test to make sure we don't modify the source buffer.
TEST_F(AnalyzerWaveformTest, simpleAnalyze) {
aw.initialize(tio, tio->getSampleRate(), BIGBUF_SIZE);
aw.process(bigbuf, BIGBUF_SIZE);
aw.finalize(tio);
for (int i = 0; i < BIGBUF_SIZE; i++) {
EXPECT_FLOAT_EQ(bigbuf[i], MAGIC_FLOAT);
}
}
//Basic test to make sure we don't step out of bounds.
TEST_F(AnalyzerWaveformTest, canary) {
aw.initialize(tio, tio->getSampleRate(), BIGBUF_SIZE);
aw.process(&canaryBigBuf[CANARY_SIZE], BIGBUF_SIZE);
aw.finalize(tio);
for (int i = 0; i < CANARY_SIZE; i++) {
EXPECT_FLOAT_EQ(canaryBigBuf[i], CANARY_FLOAT);
}
for (int i = CANARY_SIZE+BIGBUF_SIZE; i < 2*CANARY_SIZE+BIGBUF_SIZE; i++) {
EXPECT_FLOAT_EQ(canaryBigBuf[i], CANARY_FLOAT);
}
}
//Test to make sure that if an incorrect totalSamples is passed to
//initialize(..) and process(..) is told to process more samples than that,
//that we don't step out of bounds.
TEST_F(AnalyzerWaveformTest, wrongTotalSamples) {
aw.initialize(tio, tio->getSampleRate(), BIGBUF_SIZE/2);
// Deliver double the expected samples
int wrongTotalSamples = BIGBUF_SIZE;
int blockSize = 2*32768;
for (int i = CANARY_SIZE; i < CANARY_SIZE+wrongTotalSamples; i += blockSize) {
aw.process(&canaryBigBuf[i], blockSize);
}
aw.finalize(tio);
//Ensure the source buffer is intact
for (int i = CANARY_SIZE; i < BIGBUF_SIZE; i++) {
EXPECT_FLOAT_EQ(canaryBigBuf[i], MAGIC_FLOAT);
}
//Make sure our canaries are still OK
for (int i = 0; i < CANARY_SIZE; i++) {
EXPECT_FLOAT_EQ(canaryBigBuf[i], CANARY_FLOAT);
}
for (int i = CANARY_SIZE+BIGBUF_SIZE; i < 2*CANARY_SIZE+BIGBUF_SIZE; i++) {
EXPECT_FLOAT_EQ(canaryBigBuf[i], CANARY_FLOAT);
}
}
}
|
