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
|
// Copyright (C) 2010-2016 Lukas Lalinsky
// Distributed under the MIT license, see the LICENSE file for details.
#include <gtest/gtest.h>
#include "filter_utils.h"
#include "utils/rolling_integral_image.h"
namespace chromaprint {
TEST(FilterUtils, CompareSubtract) {
double res = Subtract(2.0, 1.0);
EXPECT_FLOAT_EQ(1.0, res);
}
TEST(FilterUtils, CompareSubtractLog) {
double res = SubtractLog(2.0, 1.0);
EXPECT_FLOAT_EQ(0.4054651, res);
}
TEST(FilterUtils, Filter0) {
double data[] = {
1.0, 2.0, 3.0,
4.0, 5.0, 6.0,
7.0, 8.0, 9.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter0(integral_image, 0, 0, 1, 1, Subtract);
EXPECT_FLOAT_EQ(1.0, res);
res = Filter0(integral_image, 0, 0, 2, 2, Subtract);
EXPECT_FLOAT_EQ(12.0, res);
res = Filter0(integral_image, 0, 0, 3, 3, Subtract);
EXPECT_FLOAT_EQ(45.0, res);
res = Filter0(integral_image, 1, 1, 2, 2, Subtract);
EXPECT_FLOAT_EQ(28.0, res);
res = Filter0(integral_image, 2, 2, 1, 1, Subtract);
EXPECT_FLOAT_EQ(9.0, res);
res = Filter0(integral_image, 0, 0, 3, 1, Subtract);
EXPECT_FLOAT_EQ(12.0, res);
res = Filter0(integral_image, 0, 0, 1, 3, Subtract);
EXPECT_FLOAT_EQ(6.0, res);
}
TEST(FilterUtils, Filter1) {
double data[] = {
1.0, 2.1, 3.4,
3.1, 4.1, 5.1,
6.0, 7.1, 8.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter1(integral_image, 0, 0, 1, 1, Subtract);
EXPECT_FLOAT_EQ(1.0 - 0.0, res);
res = Filter1(integral_image, 1, 1, 1, 1, Subtract);
EXPECT_FLOAT_EQ(4.1 - 0.0, res);
res = Filter1(integral_image, 0, 0, 1, 2, Subtract);
EXPECT_FLOAT_EQ(2.1 - 1.0, res);
res = Filter1(integral_image, 0, 0, 2, 2, Subtract);
EXPECT_FLOAT_EQ((2.1 + 4.1) - (1.0 + 3.1), res);
res = Filter1(integral_image, 0, 0, 3, 2, Subtract);
EXPECT_FLOAT_EQ((2.1 + 4.1 + 7.1) - (1.0 + 3.1 + 6.0), res);
}
TEST(FilterUtils, Filter2) {
double data[] = {
1.0, 2.0, 3.0,
3.0, 4.0, 5.0,
6.0, 7.0, 8.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter2(integral_image, 0, 0, 2, 1, Subtract);
EXPECT_FLOAT_EQ(2.0, res); // 3 - 1
res = Filter2(integral_image, 0, 0, 2, 2, Subtract);
EXPECT_FLOAT_EQ(4.0, res); // 3+4 - 1+2
res = Filter2(integral_image, 0, 0, 2, 3, Subtract);
EXPECT_FLOAT_EQ(6.0, res); // 3+4+5 - 1+2+3
}
TEST(FilterUtils, Filter3) {
double data[] = {
1.0, 2.1, 3.4,
3.1, 4.1, 5.1,
6.0, 7.1, 8.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter3(integral_image, 0, 0, 2, 2, Subtract);
EXPECT_FLOAT_EQ(0.1, res); // 2.1+3.1 - 1+4.1
res = Filter3(integral_image, 1, 1, 2, 2, Subtract);
EXPECT_FLOAT_EQ(0.1, res); // 4+8 - 5+7
res = Filter3(integral_image, 0, 1, 2, 2, Subtract);
EXPECT_FLOAT_EQ(0.3, res); // 2.1+5.1 - 3.4+4.1
}
TEST(FilterUtils, Filter4) {
double data[] = {
1.0, 2.0, 3.0,
3.0, 4.0, 5.0,
6.0, 7.0, 8.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter4(integral_image, 0, 0, 3, 3, Subtract);
EXPECT_FLOAT_EQ(-13.0, res); // 2+4+7 - (1+3+6) - (3+5+8)
}
TEST(FilterUtils, Filter5) {
double data[] = {
1.0, 2.0, 3.0,
3.0, 4.0, 5.0,
6.0, 7.0, 8.0,
};
RollingIntegralImage integral_image(3, data, data + 9);
double res;
res = Filter5(integral_image, 0, 0, 3, 3, Subtract);
EXPECT_FLOAT_EQ(-15.0, res); // 3+4+5 - (1+2+3) - (6+7+8)
}
}; // namespace chromaprint
|