360 строки
13 KiB
C++
360 строки
13 KiB
C++
/*
|
|
* Copyright (c) 2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
|
|
#include "third_party/googletest/src/include/gtest/gtest.h"
|
|
|
|
#include "./aom_config.h"
|
|
#include "./aom_dsp_rtcd.h"
|
|
#include "aom_ports/aom_timer.h"
|
|
#include "test/acm_random.h"
|
|
#include "test/clear_system_state.h"
|
|
#include "test/register_state_check.h"
|
|
#include "test/util.h"
|
|
|
|
using libaom_test::ACMRandom;
|
|
|
|
namespace {
|
|
|
|
typedef void (*clpf_block_t)(const uint8_t *src, uint8_t *dst, int sstride,
|
|
int dstride, int x0, int y0, int sizex, int sizey,
|
|
int width, int height, unsigned int strength);
|
|
|
|
typedef std::tr1::tuple<clpf_block_t, clpf_block_t, int, int>
|
|
clpf_block_param_t;
|
|
|
|
class ClpfBlockTest : public ::testing::TestWithParam<clpf_block_param_t> {
|
|
public:
|
|
virtual ~ClpfBlockTest() {}
|
|
virtual void SetUp() {
|
|
clpf = GET_PARAM(0);
|
|
ref_clpf = GET_PARAM(1);
|
|
sizex = GET_PARAM(2);
|
|
sizey = GET_PARAM(3);
|
|
}
|
|
|
|
virtual void TearDown() { libaom_test::ClearSystemState(); }
|
|
|
|
protected:
|
|
int sizex;
|
|
int sizey;
|
|
clpf_block_t clpf;
|
|
clpf_block_t ref_clpf;
|
|
};
|
|
|
|
typedef ClpfBlockTest ClpfSpeedTest;
|
|
|
|
#if CONFIG_AOM_HIGHBITDEPTH
|
|
typedef void (*clpf_block_hbd_t)(const uint16_t *src, uint16_t *dst,
|
|
int sstride, int dstride, int x0, int y0,
|
|
int sizex, int sizey, int width, int height,
|
|
unsigned int strength);
|
|
|
|
typedef std::tr1::tuple<clpf_block_hbd_t, clpf_block_hbd_t, int, int>
|
|
clpf_block_hbd_param_t;
|
|
|
|
class ClpfBlockHbdTest
|
|
: public ::testing::TestWithParam<clpf_block_hbd_param_t> {
|
|
public:
|
|
virtual ~ClpfBlockHbdTest() {}
|
|
virtual void SetUp() {
|
|
clpf = GET_PARAM(0);
|
|
ref_clpf = GET_PARAM(1);
|
|
sizex = GET_PARAM(2);
|
|
sizey = GET_PARAM(3);
|
|
}
|
|
|
|
virtual void TearDown() { libaom_test::ClearSystemState(); }
|
|
|
|
protected:
|
|
int sizex;
|
|
int sizey;
|
|
clpf_block_hbd_t clpf;
|
|
clpf_block_hbd_t ref_clpf;
|
|
};
|
|
|
|
typedef ClpfBlockHbdTest ClpfHbdSpeedTest;
|
|
#endif
|
|
|
|
template <typename pixel>
|
|
void test_clpf(int w, int h, int depth, int iterations,
|
|
void (*clpf)(const pixel *src, pixel *dst, int sstride,
|
|
int dstride, int x0, int y0, int sizex, int sizey,
|
|
int width, int height, unsigned int strength),
|
|
void (*ref_clpf)(const pixel *src, pixel *dst, int sstride,
|
|
int dstride, int x0, int y0, int sizex,
|
|
int sizey, int width, int height,
|
|
unsigned int strength)) {
|
|
const int size = 24;
|
|
ACMRandom rnd(ACMRandom::DeterministicSeed());
|
|
DECLARE_ALIGNED(16, pixel, s[size * size]);
|
|
DECLARE_ALIGNED(16, pixel, d[size * size]);
|
|
DECLARE_ALIGNED(16, pixel, ref_d[size * size]);
|
|
memset(ref_d, 0, size * size * sizeof(*ref_d));
|
|
memset(d, 0, size * size * sizeof(*d));
|
|
|
|
int error = 0, pos = 0, strength = 0, xpos = 0, ypos = 0;
|
|
int bits, level, count;
|
|
|
|
// Test every combination of:
|
|
// * Input with up to <depth> bits of noise
|
|
// * Noise level around every value from 0 to (1<<depth)-1
|
|
// * Blocks anywhere in the frame (along all egdes and also fully inside)
|
|
// * All strengths
|
|
// If clpf and ref_clpf are the same, we're just testing speed
|
|
for (count = 0; count < iterations; count++) {
|
|
for (level = 0; level < (1 << depth) && !error; level++) {
|
|
for (bits = 1; bits <= depth && !error; bits++) {
|
|
for (int i = 0; i < size * size; i++)
|
|
s[i] = clamp((rnd.Rand16() & ((1 << bits) - 1)) + level, 0,
|
|
(1 << depth) - 1);
|
|
|
|
for (ypos = 0; ypos < size && !error; ypos += h * !error) {
|
|
for (xpos = 0; xpos < size && !error; xpos += w * !error) {
|
|
for (strength = depth - 8; strength < depth - 5 && !error;
|
|
strength += !error) {
|
|
ref_clpf(s, ref_d, size, size, xpos, ypos, w, h, size, size,
|
|
1 << strength);
|
|
if (clpf != ref_clpf)
|
|
ASM_REGISTER_STATE_CHECK(clpf(s, d, size, size, xpos, ypos, w,
|
|
h, size, size, 1 << strength));
|
|
if (ref_clpf != clpf)
|
|
for (pos = 0; pos < size * size && !error; pos++) {
|
|
error = ref_d[pos] != d[pos];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pos--;
|
|
EXPECT_EQ(0, error)
|
|
<< "Error: ClpfBlockTest, SIMD and C mismatch." << std::endl
|
|
<< "First error at " << pos % size << "," << pos / size << " ("
|
|
<< (int16_t)ref_d[pos] << " != " << (int16_t)d[pos] << ") " << std::endl
|
|
<< "strength: " << (1 << strength) << std::endl
|
|
<< "xpos: " << xpos << std::endl
|
|
<< "ypos: " << ypos << std::endl
|
|
<< "w: " << w << std::endl
|
|
<< "h: " << h << std::endl
|
|
<< "A=" << (pos > size ? (int16_t)s[pos - size] : -1) << std::endl
|
|
<< "B=" << (pos % size - 2 >= 0 ? (int16_t)s[pos - 2] : -1) << std::endl
|
|
<< "C=" << (pos % size - 1 >= 0 ? (int16_t)s[pos - 1] : -1) << std::endl
|
|
<< "X=" << (int16_t)s[pos] << std::endl
|
|
<< "D=" << (pos % size + 1 < size ? (int16_t)s[pos + 1] : -1) << std::endl
|
|
<< "E=" << (pos % size + 2 < size ? (int16_t)s[pos + 2] : -1) << std::endl
|
|
<< "F=" << (pos + size < size * size ? (int16_t)s[pos + size] : -1)
|
|
<< std::endl;
|
|
}
|
|
|
|
template <typename pixel>
|
|
void test_clpf_speed(int w, int h, int depth, int iterations,
|
|
void (*clpf)(const pixel *src, pixel *dst, int sstride,
|
|
int dstride, int x0, int y0, int sizex,
|
|
int sizey, int width, int height,
|
|
unsigned int strength),
|
|
void (*ref_clpf)(const pixel *src, pixel *dst, int sstride,
|
|
int dstride, int x0, int y0, int sizex,
|
|
int sizey, int width, int height,
|
|
unsigned int strength)) {
|
|
aom_usec_timer ref_timer;
|
|
aom_usec_timer timer;
|
|
|
|
aom_usec_timer_start(&ref_timer);
|
|
test_clpf(w, h, depth, iterations, ref_clpf, ref_clpf);
|
|
aom_usec_timer_mark(&ref_timer);
|
|
int ref_elapsed_time = (int)aom_usec_timer_elapsed(&ref_timer);
|
|
|
|
aom_usec_timer_start(&timer);
|
|
test_clpf(w, h, depth, iterations, clpf, clpf);
|
|
aom_usec_timer_mark(&timer);
|
|
int elapsed_time = (int)aom_usec_timer_elapsed(&timer);
|
|
|
|
#if 0
|
|
std::cout << "[ ] C time = " << ref_elapsed_time / 1000
|
|
<< " ms, SIMD time = " << elapsed_time / 1000 << " ms" << std::endl;
|
|
#endif
|
|
|
|
EXPECT_GT(ref_elapsed_time, elapsed_time)
|
|
<< "Error: ClpfSpeedTest, SIMD slower than C." << std::endl
|
|
<< "C time: " << ref_elapsed_time << " us" << std::endl
|
|
<< "SIMD time: " << elapsed_time << " us" << std::endl;
|
|
}
|
|
|
|
TEST_P(ClpfBlockTest, TestSIMDNoMismatch) {
|
|
test_clpf(sizex, sizey, 8, 1, clpf, ref_clpf);
|
|
}
|
|
|
|
TEST_P(ClpfSpeedTest, TestSpeed) {
|
|
test_clpf_speed(sizex, sizey, 8, 16, clpf, ref_clpf);
|
|
}
|
|
|
|
#if CONFIG_AOM_HIGHBITDEPTH
|
|
TEST_P(ClpfBlockHbdTest, TestSIMDNoMismatch) {
|
|
test_clpf(sizex, sizey, 12, 1, clpf, ref_clpf);
|
|
}
|
|
|
|
TEST_P(ClpfHbdSpeedTest, TestSpeed) {
|
|
test_clpf_speed(sizex, sizey, 12, 1, clpf, ref_clpf);
|
|
}
|
|
#endif
|
|
|
|
using std::tr1::make_tuple;
|
|
|
|
// VS compiling for 32 bit targets does not support vector types in
|
|
// structs as arguments, which makes the v256 type of the intrinsics
|
|
// hard to support, so optimizations for this target are disabled.
|
|
#if defined(_WIN64) || !defined(_MSC_VER) || defined(__clang__)
|
|
// Test all supported architectures and block sizes
|
|
#if HAVE_SSE2
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSE2, ClpfBlockTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_sse2, &aom_clpf_block_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_sse2, &aom_clpf_block_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_sse2, &aom_clpf_block_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_sse2, &aom_clpf_block_c, 4,
|
|
4)));
|
|
#endif
|
|
|
|
#if HAVE_SSSE3
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSSE3, ClpfBlockTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_ssse3, &aom_clpf_block_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_ssse3, &aom_clpf_block_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_ssse3, &aom_clpf_block_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_ssse3, &aom_clpf_block_c, 4, 4)));
|
|
#endif
|
|
|
|
#if HAVE_SSE4_1
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSSE4_1, ClpfBlockTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_sse4_1, &aom_clpf_block_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_sse4_1, &aom_clpf_block_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_sse4_1, &aom_clpf_block_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_sse4_1, &aom_clpf_block_c, 4, 4)));
|
|
#endif
|
|
|
|
#if HAVE_NEON
|
|
INSTANTIATE_TEST_CASE_P(
|
|
NEON, ClpfBlockTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_neon, &aom_clpf_block_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_neon, &aom_clpf_block_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_neon, &aom_clpf_block_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_neon, &aom_clpf_block_c, 4,
|
|
4)));
|
|
#endif
|
|
|
|
#if CONFIG_AOM_HIGHBITDEPTH
|
|
#if HAVE_SSE2
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSE2, ClpfBlockHbdTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_hbd_sse2, &aom_clpf_block_hbd_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_hbd_sse2, &aom_clpf_block_hbd_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_hbd_sse2, &aom_clpf_block_hbd_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_hbd_sse2, &aom_clpf_block_hbd_c, 4, 4)));
|
|
#endif
|
|
|
|
#if HAVE_SSSE3
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSSE3, ClpfBlockHbdTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_hbd_ssse3, &aom_clpf_block_hbd_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_hbd_ssse3, &aom_clpf_block_hbd_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_hbd_ssse3, &aom_clpf_block_hbd_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_hbd_ssse3, &aom_clpf_block_hbd_c, 4, 4)));
|
|
#endif
|
|
|
|
#if HAVE_SSE4_1
|
|
INSTANTIATE_TEST_CASE_P(
|
|
SSSE4_1, ClpfBlockHbdTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_hbd_sse4_1, &aom_clpf_block_hbd_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_hbd_sse4_1, &aom_clpf_block_hbd_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_hbd_sse4_1, &aom_clpf_block_hbd_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_hbd_sse4_1, &aom_clpf_block_hbd_c, 4, 4)));
|
|
#endif
|
|
|
|
#if HAVE_NEON
|
|
INSTANTIATE_TEST_CASE_P(
|
|
NEON, ClpfBlockHbdTest,
|
|
::testing::Values(
|
|
make_tuple(&aom_clpf_block_hbd_neon, &aom_clpf_block_hbd_c, 8, 8),
|
|
make_tuple(&aom_clpf_block_hbd_neon, &aom_clpf_block_hbd_c, 8, 4),
|
|
make_tuple(&aom_clpf_block_hbd_neon, &aom_clpf_block_hbd_c, 4, 8),
|
|
make_tuple(&aom_clpf_block_hbd_neon, &aom_clpf_block_hbd_c, 4, 4)));
|
|
#endif
|
|
#endif // CONFIG_AOM_HIGHBITDEPTH
|
|
|
|
// Test speed for all supported architectures
|
|
#if HAVE_SSE2
|
|
INSTANTIATE_TEST_CASE_P(SSE2, ClpfSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_sse2,
|
|
&aom_clpf_block_c, 8, 8)));
|
|
#endif
|
|
|
|
#if HAVE_SSSE3
|
|
INSTANTIATE_TEST_CASE_P(SSSE3, ClpfSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_ssse3,
|
|
&aom_clpf_block_c, 8, 8)));
|
|
#endif
|
|
|
|
#if HAVE_SSE4_1
|
|
INSTANTIATE_TEST_CASE_P(SSSE4_1, ClpfSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_ssse3,
|
|
&aom_clpf_block_c, 8, 8)));
|
|
#endif
|
|
|
|
#if HAVE_NEON
|
|
INSTANTIATE_TEST_CASE_P(NEON, ClpfSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_neon,
|
|
&aom_clpf_block_c, 8, 8)));
|
|
#endif
|
|
|
|
#if CONFIG_AOM_HIGHBITDEPTH
|
|
#if HAVE_SSE2
|
|
INSTANTIATE_TEST_CASE_P(SSE2, ClpfHbdSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_hbd_sse2,
|
|
&aom_clpf_block_hbd_c, 8,
|
|
8)));
|
|
#endif
|
|
|
|
#if HAVE_SSSE3
|
|
INSTANTIATE_TEST_CASE_P(SSSE3, ClpfHbdSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_hbd_ssse3,
|
|
&aom_clpf_block_hbd_c, 8,
|
|
8)));
|
|
#endif
|
|
|
|
#if HAVE_SSE4_1
|
|
INSTANTIATE_TEST_CASE_P(SSSE4_1, ClpfHbdSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_hbd_ssse3,
|
|
&aom_clpf_block_hbd_c, 8,
|
|
8)));
|
|
#endif
|
|
|
|
#if HAVE_NEON
|
|
INSTANTIATE_TEST_CASE_P(NEON, ClpfHbdSpeedTest,
|
|
::testing::Values(make_tuple(&aom_clpf_block_hbd_neon,
|
|
&aom_clpf_block_hbd_c, 8,
|
|
8)));
|
|
#endif
|
|
#endif // CONFIG_AOM_HIGHBITDEPTH
|
|
#endif // defined(_WIN64) || !defined(_MSC_VER)
|
|
|
|
} // namespace
|