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
|
// Copyright (C) 2016 Lukas Lalinsky
// Distributed under the MIT license, see the LICENSE file for details.
#include <gtest/gtest.h>
#include "utils/gaussian_filter.h"
using namespace chromaprint;
TEST(ReflectIterator, Iterate) {
std::vector<int> data { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
ReflectIterator it(data.size());
for (int i = 0; i < 3; i++) {
it.MoveBack();
}
ASSERT_EQ(3, data[it.pos]);
ASSERT_EQ(0, it.SafeForwardDistance());
it.MoveForward();
ASSERT_EQ(2, data[it.pos]);
ASSERT_EQ(0, it.SafeForwardDistance());
it.MoveForward();
ASSERT_EQ(1, data[it.pos]);
ASSERT_EQ(0, it.SafeForwardDistance());
it.MoveForward();
ASSERT_EQ(1, data[it.pos]);
ASSERT_EQ(8, it.SafeForwardDistance());
it.MoveForward();
ASSERT_EQ(2, data[it.pos]);
// ASSERT_EQ(3, it.SafeForwardDistance());
}
TEST(BoxFilter, Width1)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
BoxFilter(input, output, 1);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(1.0, output[0]);
ASSERT_FLOAT_EQ(2.0, output[1]);
ASSERT_FLOAT_EQ(4.0, output[2]);
}
TEST(BoxFilter, Width2)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
BoxFilter(input, output, 2);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(1.0, output[0]);
ASSERT_FLOAT_EQ(1.5, output[1]);
ASSERT_FLOAT_EQ(3.0, output[2]);
}
TEST(BoxFilter, Width3)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
BoxFilter(input, output, 3);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(1.333333333, output[0]);
ASSERT_FLOAT_EQ(2.333333333, output[1]);
ASSERT_FLOAT_EQ(3.333333333, output[2]);
}
TEST(BoxFilter, Width4)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
BoxFilter(input, output, 4);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(1.5, output[0]);
ASSERT_FLOAT_EQ(2.0, output[1]);
ASSERT_FLOAT_EQ(2.75, output[2]);
}
TEST(BoxFilter, Width5)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
BoxFilter(input, output, 5);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(2.0, output[0]);
ASSERT_FLOAT_EQ(2.4, output[1]);
ASSERT_FLOAT_EQ(2.6, output[2]);
}
TEST(GaussianFilter, Test1)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
GaussianFilter(input, output, 1.6, 3);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(1.88888889, output[0]);
ASSERT_FLOAT_EQ(2.33333333, output[1]);
ASSERT_FLOAT_EQ(2.77777778, output[2]);
}
TEST(GaussianFilter, Test2)
{
std::vector<float> input { 1.0, 2.0, 4.0 };
std::vector<float> output;
GaussianFilter(input, output, 3.6, 4);
ASSERT_EQ(input.size(), output.size());
ASSERT_FLOAT_EQ(2.3322449, output[0]);
ASSERT_FLOAT_EQ(2.33306122, output[1]);
ASSERT_FLOAT_EQ(2.33469388, output[2]);
}
|
