| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. 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 |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #include <ostream> |
| #include <set> |
| #include <vector> |
| #include "aom_ports/aom_timer.h" |
| #include "config/av1_rtcd.h" |
| #include "config/aom_dsp_rtcd.h" |
| #pragma GCC diagnostic push |
| #pragma GCC diagnostic ignored "-Wuninitialized" |
| #include "test/acm_random.h" |
| #pragma GCC diagnostic pop |
| #include "test/clear_system_state.h" |
| #include "test/util.h" |
| #include "third_party/googletest/src/googletest/include/gtest/gtest.h" |
| |
| #if CONFIG_LR_IMPROVEMENTS |
| #include "av1/common/restoration.h" |
| #endif // CONFIG_LR_IMPROVEMENTS |
| |
| namespace { |
| |
| // TODO(any): Remove following INTERP_FILTERS_ALL define, so that 12-tap filter |
| // is tested once 12-tap filter SIMD is done. |
| #undef INTERP_FILTERS_ALL |
| #define INTERP_FILTERS_ALL 4 |
| |
| // All single reference convolve tests are parameterized on block size, |
| // bit-depth, and function to test. |
| // |
| // Note that parameterizing on these variables (and not other parameters) is |
| // a conscious decision - Jenkins needs some degree of parallelization to run |
| // the tests within the time limit, but if the number of parameters increases |
| // too much, the gtest framework does not handle it well (increased overhead per |
| // test, huge amount of output to stdout, etc.). |
| // |
| // Also note that the test suites must be named with the architecture, e.g., |
| // C, C_X, AVX2_X, ... The test suite that runs on Jenkins sometimes runs tests |
| // that cannot deal with intrinsics (e.g., the Valgrind tests on 32-bit x86 |
| // binaries) and will disable tests using a filter like |
| // --gtest_filter=-:SSE4_1.*. If the test suites are not named this way, the |
| // testing infrastructure will not selectively filter them properly. |
| class BlockSize { |
| public: |
| BlockSize(int w, int h) : width_(w), height_(h) {} |
| |
| int Width() const { return width_; } |
| int Height() const { return height_; } |
| |
| bool operator<(const BlockSize &other) const { |
| if (Width() == other.Width()) { |
| return Height() < other.Height(); |
| } |
| return Width() < other.Width(); |
| } |
| |
| bool operator==(const BlockSize &other) const { |
| return Width() == other.Width() && Height() == other.Height(); |
| } |
| |
| private: |
| int width_; |
| int height_; |
| }; |
| |
| // Block size / bit depth / test function used to parameterize the tests. |
| template <typename T> |
| class TestParam { |
| public: |
| TestParam(const BlockSize &block, int bd, T test_func) |
| : block_(block), bd_(bd), test_func_(test_func) {} |
| |
| const BlockSize &Block() const { return block_; } |
| int BitDepth() const { return bd_; } |
| T TestFunction() const { return test_func_; } |
| |
| bool operator==(const TestParam &other) const { |
| return Block() == other.Block() && BitDepth() == other.BitDepth() && |
| TestFunction() == other.TestFunction(); |
| } |
| |
| private: |
| BlockSize block_; |
| int bd_; |
| T test_func_; |
| }; |
| |
| template <typename T> |
| std::ostream &operator<<(std::ostream &os, const TestParam<T> &test_arg) { |
| return os << "TestParam { width:" << test_arg.Block().Width() |
| << " height:" << test_arg.Block().Height() |
| << " bd:" << test_arg.BitDepth() << " }"; |
| } |
| |
| // Generate the list of all block widths / heights that need to be tested, |
| // includes chroma and luma sizes, for the given bit-depths. The test |
| // function is the same for all generated parameters. |
| template <typename T> |
| std::vector<TestParam<T>> GetTestParams(std::initializer_list<int> bit_depths, |
| T test_func) { |
| std::set<BlockSize> sizes; |
| for (int b = BLOCK_4X4; b < BLOCK_SIZES_ALL; ++b) { |
| const int w = block_size_wide[b]; |
| const int h = block_size_high[b]; |
| sizes.insert(BlockSize(w, h)); |
| // Add in smaller chroma sizes as well. |
| if (w == 4 || h == 4) { |
| sizes.insert(BlockSize(w / 2, h / 2)); |
| } |
| } |
| std::vector<TestParam<T>> result; |
| for (const BlockSize &block : sizes) { |
| for (int bd : bit_depths) { |
| result.push_back(TestParam<T>(block, bd, test_func)); |
| } |
| } |
| return result; |
| } |
| |
| // Test the test-parameters generators work as expected. |
| class AV1ConvolveParametersTest : public ::testing::Test {}; |
| |
| template <typename T> |
| std::vector<TestParam<T>> GetHighbdTestParams(T test_func) { |
| return GetTestParams({ 10, 12 }, test_func); |
| } |
| |
| template <typename T> |
| ::testing::internal::ParamGenerator<TestParam<T>> BuildHighbdParams( |
| T test_func) { |
| return ::testing::ValuesIn(GetHighbdTestParams(test_func)); |
| } |
| |
| TEST_F(AV1ConvolveParametersTest, GetHighbdTestParams) { |
| auto v = GetHighbdTestParams(av1_highbd_convolve_x_sr_c); |
| #if CONFIG_FLEX_PARTITION |
| ASSERT_EQ(80U, v.size()); |
| #else |
| ASSERT_EQ(60U, v.size()); |
| #endif // CONFIG_FLEX_PARTITION |
| int num_10 = 0; |
| int num_12 = 0; |
| for (const auto &p : v) { |
| ASSERT_TRUE(p.BitDepth() == 10 || p.BitDepth() == 12); |
| bool same_fn = av1_highbd_convolve_x_sr_c == p.TestFunction(); |
| ASSERT_TRUE(same_fn); |
| if (p.BitDepth() == 10) { |
| ++num_10; |
| } else { |
| ++num_12; |
| } |
| } |
| ASSERT_EQ(num_10, num_12); |
| } |
| |
| // AV1ConvolveTest is the base class that all convolve tests should derive from. |
| // It provides storage/methods for generating randomized buffers for both |
| // low bit-depth and high bit-depth, and setup/teardown methods for clearing |
| // system state. Implementors can get the bit-depth / block-size / |
| // test function by calling GetParam(). |
| template <typename T> |
| class AV1ConvolveTest : public ::testing::TestWithParam<TestParam<T>> { |
| public: |
| virtual ~AV1ConvolveTest() { TearDown(); } |
| |
| virtual void SetUp() override { |
| rnd_.Reset(libaom_test::ACMRandom::DeterministicSeed()); |
| } |
| |
| virtual void TearDown() override { libaom_test::ClearSystemState(); } |
| |
| // Randomizes the 8-bit input buffer and returns a pointer to it. Note that |
| // the pointer is safe to use with an 8-tap filter. The stride can range |
| // from width to (width + kPadding). Also note that the pointer is to the |
| // same memory location. |
| static constexpr int kInputPadding = 8; |
| |
| // Get a pointer to a buffer with stride == width. Note that we must have |
| // the test param passed in explicitly -- the gtest framework does not |
| // support calling GetParam() within a templatized class. |
| // Note that FirstRandomInput8 always returns the same pointer -- if two |
| // inputs are needed, also use SecondRandomInput8. |
| const uint8_t *FirstRandomInput8(const TestParam<T> ¶m) { |
| // Note we can't call GetParam() directly -- gtest does not support |
| // this for parameterized types. |
| return RandomInput8(input8_1_, param); |
| } |
| |
| const uint8_t *SecondRandomInput8(const TestParam<T> ¶m) { |
| return RandomInput8(input8_2_, param); |
| } |
| |
| // Some of the intrinsics perform writes in 32 byte chunks. Moreover, some |
| // of the instrinsics assume that the stride is also a multiple of 32. |
| // To satisfy these constraints and also remain simple, output buffer strides |
| // are assumed MAX_SB_SIZE. |
| static constexpr int kOutputStride = MAX_SB_SIZE; |
| |
| // Check that two 8-bit output buffers are identical. |
| void AssertOutputBufferEq(const uint8_t *p1, const uint8_t *p2, int width, |
| int height) { |
| ASSERT_TRUE(p1 != p2) << "Buffers must be at different memory locations"; |
| for (int j = 0; j < height; ++j) { |
| if (memcmp(p1, p2, sizeof(*p1) * width) == 0) { |
| p1 += kOutputStride; |
| p2 += kOutputStride; |
| continue; |
| } |
| for (int i = 0; i < width; ++i) { |
| ASSERT_EQ(p1[i], p2[i]) |
| << width << "x" << height << " Pixel mismatch at (" << i << ", " |
| << j << ")"; |
| } |
| } |
| } |
| |
| // Check that two 16-bit output buffers are identical. |
| void AssertOutputBufferEq(const uint16_t *p1, const uint16_t *p2, int width, |
| int height) { |
| ASSERT_TRUE(p1 != p2) << "Buffers must be in different memory locations"; |
| for (int j = 0; j < height; ++j) { |
| if (memcmp(p1, p2, sizeof(*p1) * width) == 0) { |
| p1 += kOutputStride; |
| p2 += kOutputStride; |
| continue; |
| } |
| for (int i = 0; i < width; ++i) { |
| ASSERT_EQ(p1[i], p2[i]) |
| << width << "x" << height << " Pixel mismatch at (" << i << ", " |
| << j << ")"; |
| } |
| } |
| } |
| |
| // Note that the randomized values are capped by bit-depth. |
| const uint16_t *FirstRandomInput16(const TestParam<T> ¶m) { |
| return RandomInput16(input16_1_, param); |
| } |
| |
| const uint16_t *SecondRandomInput16(const TestParam<T> ¶m) { |
| return RandomInput16(input16_2_, param); |
| } |
| |
| #if CONFIG_LR_IMPROVEMENTS |
| const uint16_t *FirstRandomInput16Extreme(const TestParam<T> ¶m) { |
| return RandomInput16Extreme(input16_1_, param); |
| } |
| #endif // CONFIG_LR_IMPROVEMENTS |
| |
| private: |
| const uint8_t *RandomInput8(uint8_t *p, const TestParam<T> ¶m) { |
| EXPECT_EQ(8, param.BitDepth()); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Width()); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Height()); |
| const int padded_width = param.Block().Width() + kInputPadding; |
| const int padded_height = param.Block().Height() + kInputPadding; |
| Randomize(p, padded_width * padded_height); |
| return p + (kInputPadding / 2) * padded_width + kInputPadding / 2; |
| } |
| |
| void Randomize(uint8_t *p, int size) { |
| for (int i = 0; i < size; ++i) { |
| p[i] = rnd_.Rand8(); |
| } |
| } |
| |
| const uint16_t *RandomInput16(uint16_t *p, const TestParam<T> ¶m) { |
| // Check that this is only called with high bit-depths. |
| EXPECT_TRUE(param.BitDepth() == 10 || param.BitDepth() == 12); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Width()); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Height()); |
| const int padded_width = param.Block().Width() + kInputPadding; |
| const int padded_height = param.Block().Height() + kInputPadding; |
| Randomize(p, padded_width * padded_height, param.BitDepth()); |
| return p + (kInputPadding / 2) * padded_width + kInputPadding / 2; |
| } |
| |
| void Randomize(uint16_t *p, int size, int bit_depth) { |
| for (int i = 0; i < size; ++i) { |
| p[i] = rnd_.Rand16() & ((1 << bit_depth) - 1); |
| } |
| } |
| |
| #if CONFIG_LR_IMPROVEMENTS |
| const uint16_t *RandomInput16Extreme(uint16_t *p, const TestParam<T> ¶m) { |
| // Check that this is only called with high bit-depths. |
| EXPECT_TRUE(param.BitDepth() == 10 || param.BitDepth() == 12); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Width()); |
| EXPECT_GE(MAX_SB_SIZE, param.Block().Height()); |
| const int padded_width = param.Block().Width() + kInputPadding; |
| const int padded_height = param.Block().Height() + kInputPadding; |
| RandomizeExtreme(p, padded_width * padded_height, param.BitDepth()); |
| return p + (kInputPadding / 2) * padded_width + kInputPadding / 2; |
| } |
| |
| void RandomizeExtreme(uint16_t *p, int size, int max_bit_range) { |
| EXPECT_GE(12, max_bit_range); |
| const int max_val = (1 << max_bit_range) - 1; |
| for (int i = 0; i < size; ++i) { |
| p[i] = static_cast<uint16_t>(RandBool() ? max_val : 0); |
| } |
| } |
| |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| #endif // CONFIG_LR_IMPROVEMENTS |
| |
| static constexpr int kInputStride = MAX_SB_SIZE + kInputPadding; |
| |
| libaom_test::ACMRandom rnd_; |
| // Statically allocate all the memory that is needed for the tests. Note |
| // that we cannot allocate output memory here. It must use DECLARE_ALIGNED, |
| // which is a C99 feature and interacts badly with C++ member variables. |
| uint8_t input8_1_[kInputStride * kInputStride]; |
| uint8_t input8_2_[kInputStride * kInputStride]; |
| uint16_t input16_1_[kInputStride * kInputStride]; |
| uint16_t input16_2_[kInputStride * kInputStride]; |
| }; |
| |
| ///////////////////////////////////////////////////////// |
| // Single reference convolve-x functions (high bit-depth) |
| ///////////////////////////////////////////////////////// |
| typedef void (*highbd_convolve_x_func)( |
| const uint16_t *src, int src_stride, uint16_t *dst, int dst_stride, int w, |
| int h, const InterpFilterParams *filter_params_x, const int subpel_x_qn, |
| ConvolveParams *conv_params, int bd); |
| |
| class AV1ConvolveXHighbdTest : public AV1ConvolveTest<highbd_convolve_x_func> { |
| public: |
| void RunTest() { |
| for (int sub_x = 0; sub_x < 16; ++sub_x) { |
| for (int filter = EIGHTTAP_REGULAR; filter < INTERP_FILTERS_ALL; |
| ++filter) { |
| InterpFilter f = static_cast<InterpFilter>(filter); |
| TestConvolve(sub_x, f); |
| } |
| } |
| } |
| |
| private: |
| void TestConvolve(const int sub_x, const InterpFilter filter) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| const InterpFilterParams *filter_params_x = |
| av1_get_interp_filter_params_with_block_size(filter, width); |
| ConvolveParams conv_params1 = |
| get_conv_params_no_round(0, 0, NULL, 0, 0, bit_depth); |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| av1_highbd_convolve_x_sr_c(input, width, reference, kOutputStride, width, |
| height, filter_params_x, sub_x, &conv_params1, |
| bit_depth); |
| |
| ConvolveParams conv_params2 = |
| get_conv_params_no_round(0, 0, NULL, 0, 0, bit_depth); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| GetParam().TestFunction()(input, width, test, kOutputStride, width, height, |
| filter_params_x, sub_x, &conv_params2, bit_depth); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| }; |
| |
| TEST_P(AV1ConvolveXHighbdTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P(C, AV1ConvolveXHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_x_sr_c)); |
| |
| #if HAVE_SSSE3 |
| INSTANTIATE_TEST_SUITE_P(SSSE3, AV1ConvolveXHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_x_sr_ssse3)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1ConvolveXHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_x_sr_avx2)); |
| #endif |
| |
| ///////////////////////////////////////////////////////// |
| // Single reference convolve-y functions (high bit-depth) |
| ///////////////////////////////////////////////////////// |
| typedef void (*highbd_convolve_y_func)( |
| const uint16_t *src, int src_stride, uint16_t *dst, int dst_stride, int w, |
| int h, const InterpFilterParams *filter_params_y, const int subpel_y_qn, |
| int bd); |
| |
| class AV1ConvolveYHighbdTest : public AV1ConvolveTest<highbd_convolve_y_func> { |
| public: |
| void RunTest() { |
| for (int sub_y = 0; sub_y < 16; ++sub_y) { |
| for (int filter = EIGHTTAP_REGULAR; filter < INTERP_FILTERS_ALL; |
| ++filter) { |
| InterpFilter f = static_cast<InterpFilter>(filter); |
| TestConvolve(sub_y, f); |
| } |
| } |
| } |
| |
| private: |
| void TestConvolve(const int sub_y, const InterpFilter filter) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| const InterpFilterParams *filter_params_y = |
| av1_get_interp_filter_params_with_block_size(filter, height); |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| av1_highbd_convolve_y_sr_c(input, width, reference, kOutputStride, width, |
| height, filter_params_y, sub_y, bit_depth); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| GetParam().TestFunction()(input, width, test, kOutputStride, width, height, |
| filter_params_y, sub_y, bit_depth); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| }; |
| |
| TEST_P(AV1ConvolveYHighbdTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P(C, AV1ConvolveYHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_y_sr_c)); |
| |
| #if HAVE_SSSE3 |
| INSTANTIATE_TEST_SUITE_P(SSSE3, AV1ConvolveYHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_y_sr_ssse3)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1ConvolveYHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_y_sr_avx2)); |
| #endif |
| |
| /////////////////////////////////////////////////////////////// |
| // Single reference convolve-copy functions (high bit-depth) |
| /////////////////////////////////////////////////////////////// |
| typedef void (*highbd_convolve_copy_func)(const uint16_t *src, |
| ptrdiff_t src_stride, uint16_t *dst, |
| ptrdiff_t dst_stride, int w, int h); |
| |
| class AV1ConvolveCopyHighbdTest |
| : public AV1ConvolveTest<highbd_convolve_copy_func> { |
| public: |
| void RunTest() { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| aom_highbd_convolve_copy_c(input, width, reference, kOutputStride, width, |
| height); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| GetParam().TestFunction()(input, width, test, kOutputStride, width, height); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| }; |
| |
| TEST_P(AV1ConvolveCopyHighbdTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P(C, AV1ConvolveCopyHighbdTest, |
| BuildHighbdParams(aom_highbd_convolve_copy_c)); |
| |
| #if HAVE_SSE2 |
| INSTANTIATE_TEST_SUITE_P(SSE2, AV1ConvolveCopyHighbdTest, |
| BuildHighbdParams(aom_highbd_convolve_copy_sse2)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1ConvolveCopyHighbdTest, |
| BuildHighbdParams(aom_highbd_convolve_copy_avx2)); |
| #endif |
| |
| ////////////////////////////////////////////////////////// |
| // Single reference convolve-2d functions (high bit-depth) |
| ////////////////////////////////////////////////////////// |
| |
| typedef void (*highbd_convolve_2d_func)( |
| const uint16_t *src, int src_stride, uint16_t *dst, int dst_stride, int w, |
| int h, const InterpFilterParams *filter_params_x, |
| const InterpFilterParams *filter_params_y, const int subpel_x_qn, |
| const int subpel_y_qn, ConvolveParams *conv_params, int bd); |
| |
| class AV1Convolve2DHighbdTest |
| : public AV1ConvolveTest<highbd_convolve_2d_func> { |
| public: |
| void RunTest() { |
| for (int sub_x = 0; sub_x < 16; ++sub_x) { |
| for (int sub_y = 0; sub_y < 16; ++sub_y) { |
| for (int h_f = EIGHTTAP_REGULAR; h_f < INTERP_FILTERS_ALL; ++h_f) { |
| for (int v_f = EIGHTTAP_REGULAR; v_f < INTERP_FILTERS_ALL; ++v_f) { |
| TestConvolve(static_cast<InterpFilter>(h_f), |
| static_cast<InterpFilter>(v_f), sub_x, sub_y); |
| } |
| } |
| } |
| } |
| } |
| |
| private: |
| void TestConvolve(const InterpFilter h_f, const InterpFilter v_f, |
| const int sub_x, const int sub_y) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| const InterpFilterParams *filter_params_x = |
| av1_get_interp_filter_params_with_block_size(h_f, width); |
| const InterpFilterParams *filter_params_y = |
| av1_get_interp_filter_params_with_block_size(v_f, height); |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| ConvolveParams conv_params1 = |
| get_conv_params_no_round(0, 0, NULL, 0, 0, bit_depth); |
| av1_highbd_convolve_2d_sr_c(input, width, reference, kOutputStride, width, |
| height, filter_params_x, filter_params_y, sub_x, |
| sub_y, &conv_params1, bit_depth); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| ConvolveParams conv_params2 = |
| get_conv_params_no_round(0, 0, NULL, 0, 0, bit_depth); |
| GetParam().TestFunction()(input, width, test, kOutputStride, width, height, |
| filter_params_x, filter_params_y, sub_x, sub_y, |
| &conv_params2, bit_depth); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| }; |
| |
| TEST_P(AV1Convolve2DHighbdTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P(C, AV1Convolve2DHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_2d_sr_c)); |
| |
| #if HAVE_SSSE3 |
| INSTANTIATE_TEST_SUITE_P(SSSE3, AV1Convolve2DHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_2d_sr_ssse3)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1Convolve2DHighbdTest, |
| BuildHighbdParams(av1_highbd_convolve_2d_sr_avx2)); |
| #endif |
| |
| ////////////////////////// |
| // Compound Convolve Tests |
| ////////////////////////// |
| |
| // The compound functions do not work for chroma block sizes. Provide |
| // a function to generate test parameters for just luma block sizes. |
| template <typename T> |
| std::vector<TestParam<T>> GetLumaTestParams( |
| std::initializer_list<int> bit_depths, T test_func) { |
| std::set<BlockSize> sizes; |
| for (int b = BLOCK_4X4; b < BLOCK_SIZES_ALL; ++b) { |
| const int w = block_size_wide[b]; |
| const int h = block_size_high[b]; |
| sizes.insert(BlockSize(w, h)); |
| } |
| std::vector<TestParam<T>> result; |
| for (int bit_depth : bit_depths) { |
| for (const auto &block : sizes) { |
| result.push_back(TestParam<T>(block, bit_depth, test_func)); |
| } |
| } |
| return result; |
| } |
| |
| template <typename T> |
| std::vector<TestParam<T>> GetHighbdLumaTestParams(T test_func) { |
| return GetLumaTestParams({ 10, 12 }, test_func); |
| } |
| |
| TEST_F(AV1ConvolveParametersTest, GetHighbdLumaTestParams) { |
| auto v = GetHighbdLumaTestParams(av1_highbd_dist_wtd_convolve_x_c); |
| ASSERT_EQ(static_cast<size_t>(BLOCK_SIZES_ALL * 2), v.size()); |
| int num_10 = 0; |
| int num_12 = 0; |
| for (const auto &e : v) { |
| ASSERT_TRUE(10 == e.BitDepth() || 12 == e.BitDepth()); |
| bool same_fn = av1_highbd_dist_wtd_convolve_x_c == e.TestFunction(); |
| ASSERT_TRUE(same_fn); |
| if (e.BitDepth() == 10) { |
| ++num_10; |
| } else { |
| ++num_12; |
| } |
| } |
| ASSERT_EQ(num_10, num_12); |
| } |
| |
| template <typename T> |
| ::testing::internal::ParamGenerator<TestParam<T>> BuildHighbdLumaParams( |
| T test_func) { |
| return ::testing::ValuesIn(GetHighbdLumaTestParams(test_func)); |
| } |
| |
| // Compound cases also need to test different frame offsets and weightings. |
| class CompoundParam { |
| public: |
| CompoundParam(int fwd_offset, int bck_offset) |
| : fwd_offset_(fwd_offset), bck_offset_(bck_offset) {} |
| |
| bool UseWtdCompAvg() const { |
| return bck_offset_ != (1 << (DIST_PRECISION_BITS - 1)) || |
| fwd_offset_ != (1 << (DIST_PRECISION_BITS - 1)); |
| } |
| int FwdOffset() const { return fwd_offset_; } |
| int BckOffset() const { return bck_offset_; } |
| |
| private: |
| int fwd_offset_; |
| int bck_offset_; |
| }; |
| |
| std::vector<CompoundParam> GetCompoundParams() { |
| std::vector<CompoundParam> result; |
| result.push_back(CompoundParam(1 << (DIST_PRECISION_BITS - 1), |
| 1 << (DIST_PRECISION_BITS - 1))); |
| for (int k = 0; k < 2; ++k) { |
| for (int l = 0; l < 4; ++l) { |
| result.push_back(CompoundParam(quant_dist_lookup_table[l][k], |
| quant_dist_lookup_table[l][1 - k])); |
| } |
| } |
| return result; |
| } |
| |
| TEST_F(AV1ConvolveParametersTest, GetCompoundParams) { |
| auto v = GetCompoundParams(); |
| ASSERT_EQ(9U, v.size()); |
| ASSERT_FALSE(v[0].UseWtdCompAvg()); |
| for (size_t i = 1; i < v.size(); ++i) { |
| ASSERT_TRUE(v[i].UseWtdCompAvg()); |
| } |
| } |
| |
| ///////////////////////////////////////////////// |
| // Compound convolve-x functions (high bit-depth) |
| ///////////////////////////////////////////////// |
| ConvolveParams GetConvolveParams(int do_average, CONV_BUF_TYPE *conv_buf, |
| int width, int bit_depth, |
| const CompoundParam &compound) { |
| ConvolveParams conv_params = |
| get_conv_params_no_round(do_average, 0, conv_buf, width, 1, bit_depth); |
| (void)compound; |
| conv_params.fwd_offset = compound.FwdOffset(); |
| conv_params.bck_offset = compound.BckOffset(); |
| return conv_params; |
| } |
| |
| class AV1ConvolveXHighbdCompoundTest |
| : public AV1ConvolveTest<highbd_convolve_x_func> { |
| public: |
| void RunTest() { |
| auto compound_params = GetCompoundParams(); |
| for (int sub_pix = 0; sub_pix < 16; ++sub_pix) { |
| for (int f = EIGHTTAP_REGULAR; f < INTERP_FILTERS_ALL; ++f) { |
| for (const auto &c : compound_params) { |
| TestConvolve(sub_pix, static_cast<InterpFilter>(f), c); |
| } |
| } |
| } |
| } |
| |
| protected: |
| virtual const InterpFilterParams *FilterParams(InterpFilter f, |
| const BlockSize &block) const { |
| return av1_get_interp_filter_params_with_block_size(f, block.Width()); |
| } |
| |
| virtual highbd_convolve_x_func ReferenceFunc() const { |
| return av1_highbd_dist_wtd_convolve_x_c; |
| } |
| |
| private: |
| void TestConvolve(const int sub_pix, const InterpFilter filter, |
| const CompoundParam &compound) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| |
| const uint16_t *input1 = FirstRandomInput16(GetParam()); |
| const uint16_t *input2 = SecondRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, reference_conv_buf[MAX_SB_SQUARE]); |
| Convolve(ReferenceFunc(), input1, input2, reference, reference_conv_buf, |
| compound, sub_pix, filter); |
| |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, test_conv_buf[MAX_SB_SQUARE]); |
| Convolve(GetParam().TestFunction(), input1, input2, test, test_conv_buf, |
| compound, sub_pix, filter); |
| |
| AssertOutputBufferEq(reference_conv_buf, test_conv_buf, width, height); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| |
| void Convolve(highbd_convolve_x_func test_func, const uint16_t *src1, |
| const uint16_t *src2, uint16_t *dst, CONV_BUF_TYPE *conv_buf, |
| const CompoundParam &compound, const int sub_pix, |
| const InterpFilter filter) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| const InterpFilterParams *filter_params = |
| FilterParams(filter, GetParam().Block()); |
| ConvolveParams conv_params = |
| GetConvolveParams(0, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src1, width, dst, kOutputStride, width, height, filter_params, |
| sub_pix, &conv_params, bit_depth); |
| conv_params = |
| GetConvolveParams(1, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src2, width, dst, kOutputStride, width, height, filter_params, |
| sub_pix, &conv_params, bit_depth); |
| } |
| }; |
| |
| TEST_P(AV1ConvolveXHighbdCompoundTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P( |
| C, AV1ConvolveXHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_x_c)); |
| |
| #if HAVE_SSE4_1 |
| INSTANTIATE_TEST_SUITE_P( |
| SSE4_1, AV1ConvolveXHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_x_sse4_1)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1ConvolveXHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_x_avx2)); |
| #endif |
| |
| ///////////////////////////////////////////////// |
| // Compound convolve-y functions (high bit-depth) |
| ///////////////////////////////////////////////// |
| |
| // Again, the X and Y convolve functions have the same type signature and logic. |
| class AV1ConvolveYHighbdCompoundTest : public AV1ConvolveXHighbdCompoundTest { |
| virtual highbd_convolve_x_func ReferenceFunc() const override { |
| return av1_highbd_dist_wtd_convolve_y_c; |
| } |
| virtual const InterpFilterParams *FilterParams( |
| InterpFilter f, const BlockSize &block) const override { |
| return av1_get_interp_filter_params_with_block_size(f, block.Height()); |
| } |
| }; |
| |
| TEST_P(AV1ConvolveYHighbdCompoundTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P( |
| C, AV1ConvolveYHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_y_c)); |
| |
| #if HAVE_SSE4_1 |
| INSTANTIATE_TEST_SUITE_P( |
| SSE4_1, AV1ConvolveYHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_y_sse4_1)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1ConvolveYHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_y_avx2)); |
| #endif |
| |
| /////////////////////////////////////////////////////// |
| // Compound convolve-2d-copy functions (high bit-depth) |
| /////////////////////////////////////////////////////// |
| typedef void (*highbd_compound_conv_2d_copy_func)(const uint16_t *src, |
| int src_stride, uint16_t *dst, |
| int dst_stride, int w, int h, |
| ConvolveParams *conv_params, |
| int bd); |
| |
| class AV1Convolve2DCopyHighbdCompoundTest |
| : public AV1ConvolveTest<highbd_compound_conv_2d_copy_func> { |
| public: |
| void RunTest() { |
| auto compound_params = GetCompoundParams(); |
| for (const auto &compound : compound_params) { |
| TestConvolve(compound); |
| } |
| } |
| |
| public: |
| void SpeedTest() { |
| auto compound_params = GetCompoundParams(); |
| for (const auto &compound : compound_params) { |
| SpeedTestConvolve(compound); |
| } |
| } |
| |
| private: |
| void SpeedTestConvolve(const CompoundParam &compound) { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| int nob = 100000; |
| |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, conv_buf[MAX_SB_SQUARE]); |
| highbd_compound_conv_2d_copy_func test_func = GetParam().TestFunction(); |
| |
| ConvolveParams conv_params = |
| GetConvolveParams(0, conv_buf, kOutputStride, bit_depth, compound); |
| ConvolveParams conv_params_do_avg = |
| GetConvolveParams(1, conv_buf, kOutputStride, bit_depth, compound); |
| |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < nob; i++) { |
| av1_highbd_dist_wtd_convolve_2d_copy_c(input, width, conv_buf, |
| kOutputStride, width, height, |
| &conv_params, bit_depth); |
| av1_highbd_dist_wtd_convolve_2d_copy_c(input, width, conv_buf, |
| kOutputStride, width, height, |
| &conv_params_do_avg, bit_depth); |
| } |
| aom_usec_timer_mark(&timer); |
| const int elapsed_time = static_cast<int>(aom_usec_timer_elapsed(&timer)); |
| |
| aom_usec_timer timer1; |
| aom_usec_timer_start(&timer1); |
| for (int i = 0; i < nob; i++) { |
| test_func(input, width, conv_buf, kOutputStride, width, height, |
| &conv_params, bit_depth); |
| test_func(input, width, conv_buf, kOutputStride, width, height, |
| &conv_params_do_avg, bit_depth); |
| } |
| aom_usec_timer_mark(&timer1); |
| const int elapsed_time1 = static_cast<int>(aom_usec_timer_elapsed(&timer1)); |
| printf("%d x %d block: bd: %d, Scaling = %.2f\n", width, height, bit_depth, |
| (double)elapsed_time / elapsed_time1); |
| } |
| |
| private: |
| void TestConvolve(const CompoundParam &compound) { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| |
| const uint16_t *input1 = FirstRandomInput16(GetParam()); |
| const uint16_t *input2 = SecondRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, reference_conv_buf[MAX_SB_SQUARE]); |
| Convolve(av1_highbd_dist_wtd_convolve_2d_copy_c, input1, input2, reference, |
| reference_conv_buf, compound); |
| |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, test_conv_buf[MAX_SB_SQUARE]); |
| Convolve(GetParam().TestFunction(), input1, input2, test, test_conv_buf, |
| compound); |
| |
| AssertOutputBufferEq(reference_conv_buf, test_conv_buf, width, height); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| |
| void Convolve(highbd_compound_conv_2d_copy_func test_func, |
| const uint16_t *src1, const uint16_t *src2, uint16_t *dst, |
| uint16_t *conv_buf, const CompoundParam &compound) { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| |
| ConvolveParams conv_params = |
| GetConvolveParams(0, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src1, width, dst, kOutputStride, width, height, &conv_params, |
| bit_depth); |
| |
| conv_params = |
| GetConvolveParams(1, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src2, width, dst, kOutputStride, width, height, &conv_params, |
| bit_depth); |
| } |
| }; |
| |
| TEST_P(AV1Convolve2DCopyHighbdCompoundTest, RunTest) { RunTest(); } |
| TEST_P(AV1Convolve2DCopyHighbdCompoundTest, DISABLED_SpeedTest) { SpeedTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P( |
| C, AV1Convolve2DCopyHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_copy_c)); |
| |
| #if HAVE_SSE4_1 |
| INSTANTIATE_TEST_SUITE_P( |
| SSE4_1, AV1Convolve2DCopyHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_copy_sse4_1)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1Convolve2DCopyHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_copy_avx2)); |
| #endif |
| |
| ////////////////////////////////////////////////// |
| // Compound convolve-2d functions (high bit-depth) |
| ////////////////////////////////////////////////// |
| |
| class AV1Convolve2DHighbdCompoundTest |
| : public AV1ConvolveTest<highbd_convolve_2d_func> { |
| public: |
| void RunTest() { |
| auto compound_params = GetCompoundParams(); |
| for (int h_f = EIGHTTAP_REGULAR; h_f < INTERP_FILTERS_ALL; ++h_f) { |
| for (int v_f = EIGHTTAP_REGULAR; v_f < INTERP_FILTERS_ALL; ++v_f) { |
| for (int sub_x = 0; sub_x < 16; ++sub_x) { |
| for (int sub_y = 0; sub_y < 16; ++sub_y) { |
| for (const auto &compound : compound_params) { |
| TestConvolve(static_cast<InterpFilter>(h_f), |
| static_cast<InterpFilter>(v_f), sub_x, sub_y, |
| compound); |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| private: |
| void TestConvolve(const InterpFilter h_f, const InterpFilter v_f, |
| const int sub_x, const int sub_y, |
| const CompoundParam &compound) { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| const uint16_t *input1 = FirstRandomInput16(GetParam()); |
| const uint16_t *input2 = SecondRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, reference_conv_buf[MAX_SB_SQUARE]); |
| Convolve(av1_highbd_dist_wtd_convolve_2d_c, input1, input2, reference, |
| reference_conv_buf, compound, h_f, v_f, sub_x, sub_y); |
| |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, CONV_BUF_TYPE, test_conv_buf[MAX_SB_SQUARE]); |
| Convolve(GetParam().TestFunction(), input1, input2, test, test_conv_buf, |
| compound, h_f, v_f, sub_x, sub_y); |
| |
| AssertOutputBufferEq(reference_conv_buf, test_conv_buf, width, height); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| |
| private: |
| void Convolve(highbd_convolve_2d_func test_func, const uint16_t *src1, |
| const uint16_t *src2, uint16_t *dst, uint16_t *conv_buf, |
| const CompoundParam &compound, const InterpFilter h_f, |
| const InterpFilter v_f, const int sub_x, const int sub_y) { |
| const BlockSize &block = GetParam().Block(); |
| const int width = block.Width(); |
| const int height = block.Height(); |
| |
| const InterpFilterParams *filter_params_x = |
| av1_get_interp_filter_params_with_block_size(h_f, width); |
| const InterpFilterParams *filter_params_y = |
| av1_get_interp_filter_params_with_block_size(v_f, height); |
| const int bit_depth = GetParam().BitDepth(); |
| ConvolveParams conv_params = |
| GetConvolveParams(0, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src1, width, dst, kOutputStride, width, height, filter_params_x, |
| filter_params_y, sub_x, sub_y, &conv_params, bit_depth); |
| |
| conv_params = |
| GetConvolveParams(1, conv_buf, kOutputStride, bit_depth, compound); |
| test_func(src2, width, dst, kOutputStride, width, height, filter_params_x, |
| filter_params_y, sub_x, sub_y, &conv_params, bit_depth); |
| } |
| }; |
| |
| TEST_P(AV1Convolve2DHighbdCompoundTest, RunTest) { RunTest(); } |
| |
| INSTANTIATE_TEST_SUITE_P( |
| C, AV1Convolve2DHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_c)); |
| |
| #if HAVE_SSE4_1 |
| INSTANTIATE_TEST_SUITE_P( |
| SSE4_1, AV1Convolve2DHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_sse4_1)); |
| #endif |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1Convolve2DHighbdCompoundTest, |
| BuildHighbdLumaParams(av1_highbd_dist_wtd_convolve_2d_avx2)); |
| #endif |
| |
| ////////////////////////////////////////////////////////// |
| // Nonseparable convolve-2d functions (high bit-depth) |
| ////////////////////////////////////////////////////////// |
| |
| #if CONFIG_LR_IMPROVEMENTS |
| typedef void (*highbd_convolve_nonsep_2d_func)( |
| const uint16_t *src, int src_stride, |
| const NonsepFilterConfig *filter_config, const int16_t *filter, |
| uint16_t *dst, int dst_stride, int bit_depth, int block_row_begin, |
| int block_row_end, int block_col_begin, int block_col_end); |
| |
| class AV1ConvolveNonSep2DHighbdTest |
| : public AV1ConvolveTest<highbd_convolve_nonsep_2d_func> { |
| public: |
| void RunTest(RestorationType rtype) { |
| for (int i = 0; i < kTestIterations; i++) { |
| SetFilterTaps(); |
| TestConvolve(FilterTaps_, rtype); |
| } |
| } |
| void RunSpeedTest(RestorationType rtype) { |
| SpeedTestConvolve(FilterTaps_, rtype); |
| }; |
| |
| private: |
| void BitMatchTest(const uint16_t *input, int input_stride, int width, |
| int height, const int16_t *filter, uint16_t *reference, |
| uint16_t *test, int dst_stride, int bit_depth, |
| int block_row_begin, int block_row_end, int block_col_begin, |
| int block_col_end, RestorationType rtype) { |
| const NonsepFilterConfig *filter_config[2] = { NULL, NULL }; |
| highbd_convolve_nonsep_2d_func ref_func = av1_convolve_symmetric_highbd_c; |
| const int num_planes = 2; |
| |
| if (rtype == RESTORE_PC_WIENER) { |
| ref_func = av1_convolve_symmetric_highbd_c; |
| filter_config[0] = &UnconstrainedSumFilterConfig_; |
| filter_config[1] = &PcWienerNonsepFilterConfigChroma_; |
| } |
| |
| // When CONFIG_WIENER_NONSEP=1, luma and chroma plane uses different number |
| // of filter taps and both needs to be tested. Here, luma is tested for |
| // 12/13-tap filtering whereas chroma is tested for 6-tap filtering. |
| if (rtype == RESTORE_WIENER_NONSEP) { |
| ref_func = av1_convolve_symmetric_subtract_center_highbd_c; |
| filter_config[0] = &UnitSumFilterConfig_; |
| filter_config[1] = &UnitSumFilterConfigChroma_; |
| } |
| |
| assert(filter_config[0] != NULL && filter_config[1] != NULL); |
| |
| for (int plane = 0; plane < num_planes; plane++) { |
| ref_func(input, input_stride, filter_config[plane], filter, reference, |
| dst_stride, bit_depth, block_row_begin, block_row_end, |
| block_col_begin, block_col_end); |
| GetParam().TestFunction()(input, input_stride, filter_config[plane], |
| filter, test, dst_stride, bit_depth, |
| block_row_begin, block_row_end, block_col_begin, |
| block_col_end); |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| } |
| void TestConvolve(const int16_t *filter, RestorationType rtype) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| |
| ASSERT_TRUE(kInputPadding >= kMaxTapOffset) |
| << "Not enough padding for 7x7 filters"; |
| const uint16_t *centered_input = |
| input + kMaxTapOffset * width + kMaxTapOffset; |
| const int input_stride = width; |
| BitMatchTest(centered_input, input_stride, width, height, filter, reference, |
| test, kOutputStride, bit_depth, 0, height, 0, width, rtype); |
| // Extreme value test |
| const uint16_t *extreme_input = FirstRandomInput16Extreme(GetParam()); |
| const uint16_t *centered_extreme_input = |
| extreme_input + kMaxTapOffset * width + kMaxTapOffset; |
| int16_t Extream_Tap_[kNumSymmetricTaps + 1]; |
| RandomizeExtreamFilterTap(Extream_Tap_, kNumSymmetricTaps + 1, |
| kMaxPrecisionBeforeOverflow); |
| BitMatchTest(centered_extreme_input, input_stride, width, height, |
| Extream_Tap_, reference, test, kOutputStride, bit_depth, 0, |
| height, 0, width, rtype); |
| } |
| |
| void SpeedTestConvolve(const int16_t *filter, RestorationType rtype) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| const int num_planes = 2; |
| |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| |
| ASSERT_TRUE(kInputPadding >= kMaxTapOffset) |
| << "Not enough padding for 7x7 filters"; |
| const uint16_t *centered_input = |
| input + kMaxTapOffset * width + kMaxTapOffset; |
| |
| // Calculate time taken for C function |
| const NonsepFilterConfig *filter_config[2] = { NULL, NULL }; |
| highbd_convolve_nonsep_2d_func ref_func = av1_convolve_symmetric_highbd_c; |
| |
| if (rtype == RESTORE_PC_WIENER) { |
| ref_func = av1_convolve_symmetric_highbd_c; |
| filter_config[0] = &UnconstrainedSumFilterConfig_; |
| filter_config[1] = &PcWienerNonsepFilterConfigChroma_; |
| } |
| |
| // When CONFIG_WIENER_NONSEP=1, luma and chroma uses different number of |
| // filter taps and both needs to be tested. Here, luma is tested for |
| // 12/13-tap filtering whereas chroma is tested for 6-tap filtering. |
| if (rtype == RESTORE_WIENER_NONSEP) { |
| ref_func = av1_convolve_symmetric_subtract_center_highbd_c; |
| filter_config[0] = &UnitSumFilterConfig_; |
| filter_config[1] = &UnitSumFilterConfigChroma_; |
| } |
| |
| for (int plane = 0; plane < num_planes; plane++) { |
| // Calculate time taken by reference/c function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| ref_func(centered_input, width, filter_config[plane], filter, reference, |
| kOutputStride, bit_depth, 0, height, 0, width); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| GetParam().TestFunction()(centered_input, width, filter_config[plane], |
| filter, test, kOutputStride, bit_depth, 0, |
| height, 0, width); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / (kSpeedIterations * width * height); |
| float opt_time_per_pixel = (float)1000.0 * elapsed_time_opt / |
| (kSpeedIterations * width * height); |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| "plane=%3d, %3dx%-3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| plane, width, height, c_time_per_pixel, opt_time_per_pixel, scaling); |
| } |
| } |
| |
| // Generates NonsepFilterConfig compliant origin symmetric filter tap values. |
| // The first (2 * kNumSymmetricTaps) are for the CONFIG_WIENER_NONSEP use case |
| // where the center tap is constrained so that filter sums to one. The last |
| // added tap at (2 * kNumSymmetricTaps) is unconstrained and intended for |
| // CONFIG_PC_WIENER use case. |
| void SetFilterTaps() { |
| Randomize(FilterTaps_, kNumSymmetricTaps + 1, kMaxPrecisionBeforeOverflow); |
| } |
| |
| // Fills the array p with signed integers. |
| void Randomize(int16_t *p, int size, int max_bit_range) { |
| ASSERT_TRUE(max_bit_range < 16) << "max_bit_range has to be less than 16"; |
| for (int i = 0; i < size; ++i) { |
| p[i] = rnd_.Rand15Signed() & ((1 << max_bit_range) - 1); |
| } |
| } |
| |
| // Fills the array p with maximum and minimum possible integers. |
| void RandomizeExtreamFilterTap(int16_t *p, int size, int max_bit_range) { |
| ASSERT_TRUE(max_bit_range < 16) << "max_bit_range has to be less than 16"; |
| const int sign_max_val = (1 << (max_bit_range - 1)) - 1; |
| for (int i = 0; i < size; ++i) { |
| p[i] = static_cast<uint16_t>(RandBool() ? sign_max_val |
| : -(sign_max_val + 1)); |
| } |
| } |
| |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| |
| libaom_test::ACMRandom rnd_; |
| static constexpr int kMaxPrecisionBeforeOverflow = 12; |
| static constexpr int kNumSymmetricTaps = 12; |
| static constexpr int kNumSymmetricTapsChroma = 6; |
| static constexpr int kMaxTapOffset = 3; // Filters are 7x7. |
| static constexpr int kSpeedIterations = 10000; |
| static constexpr int kTestIterations = 100; |
| |
| // Configuration for nonseparable 7x7 filters for DIAMOND shape. |
| // Format is offset (i) row and (ii) column from center pixel |
| // and the (iii) filter-tap index that multiplies the pixel at |
| // the respective offset. |
| const int NonsepConfig_[25][3] = { |
| { -3, 0, 0 }, { 3, 0, 0 }, { -2, -1, 1 }, { 2, 1, 1 }, { -2, 0, 2 }, |
| { 2, 0, 2 }, { -2, 1, 3 }, { 2, -1, 3 }, { -1, -2, 4 }, { 1, 2, 4 }, |
| { -1, -1, 5 }, { 1, 1, 5 }, { -1, 0, 6 }, { 1, 0, 6 }, { -1, 1, 7 }, |
| { 1, -1, 7 }, { -1, 2, 8 }, { 1, -2, 8 }, { 0, -3, 9 }, { 0, 3, 9 }, |
| { 0, -2, 10 }, { 0, 2, 10 }, { 0, -1, 11 }, { 0, 1, 11 }, { 0, 0, 12 }, |
| }; |
| |
| const int wienerns_wout_subtract_center_config_uv_from_uv_[13][3] = { |
| { 1, 0, 0 }, { -1, 0, 0 }, { 0, 1, 1 }, { 0, -1, 1 }, { 1, 1, 2 }, |
| { -1, -1, 2 }, { -1, 1, 3 }, { 1, -1, 3 }, { 2, 0, 4 }, { -2, 0, 4 }, |
| { 0, 2, 5 }, { 0, -2, 5 }, { 0, 0, 6 }, |
| }; |
| |
| // Filters use all unique taps. |
| const NonsepFilterConfig UnconstrainedSumFilterConfig_ = { |
| kMaxPrecisionBeforeOverflow, |
| 2 * kNumSymmetricTaps + 1, |
| 0, |
| NonsepConfig_, |
| NULL, |
| 0, |
| 0 |
| }; |
| |
| const NonsepFilterConfig PcWienerNonsepFilterConfigChroma_ = { |
| kMaxPrecisionBeforeOverflow, |
| 2 * kNumSymmetricTapsChroma + 1, |
| 0, |
| wienerns_wout_subtract_center_config_uv_from_uv_, |
| NULL, |
| 0, |
| 0 |
| }; |
| |
| // Configuration for UnitSumFilterConfig_ wiener nonseparable 7x7 filters for |
| // DIAMOND shape. Format is offset (i) row and (ii) column from center pixel |
| // and the (iii) filter-tap index that multiplies the pixel at the respective |
| // offset. |
| const int WienerNonsepConfig_[25][3] = { |
| { 1, 0, 0 }, |
| { -1, 0, 0 }, |
| { 0, 1, 1 }, |
| { 0, -1, 1 }, |
| { 2, 0, 2 }, |
| { -2, 0, 2 }, |
| { 0, 2, 3 }, |
| { 0, -2, 3 }, |
| { 1, 1, 4 }, |
| { -1, -1, 4 }, |
| { -1, 1, 5 }, |
| { 1, -1, 5 }, |
| { 2, 1, 6 }, |
| { -2, -1, 6 }, |
| { 2, -1, 7 }, |
| { -2, 1, 7 }, |
| { 1, 2, 8 }, |
| { -1, -2, 8 }, |
| { 1, -2, 9 }, |
| { -1, 2, 9 }, |
| { 3, 0, 10 }, |
| { -3, 0, 10 }, |
| { 0, 3, 11 }, |
| { 0, -3, 11 }, |
| #if USE_CENTER_WIENER_NONSEP |
| { 0, 0, 12 }, |
| #endif // USE_CENTER_WIENER_NONSEP |
| }; |
| |
| const int WienerNonsepConfigChroma_[12][3] = { |
| { 1, 0, 0 }, { -1, 0, 0 }, { 0, 1, 1 }, { 0, -1, 1 }, |
| { 1, 1, 2 }, { -1, -1, 2 }, { -1, 1, 3 }, { 1, -1, 3 }, |
| { 2, 0, 4 }, { -2, 0, 4 }, { 0, 2, 5 }, { 0, -2, 5 }, |
| }; |
| |
| // Filters use only the first (2 * kNumSymmetricTaps) taps. Center tap is |
| // constrained. |
| const NonsepFilterConfig UnitSumFilterConfig_ = { |
| kMaxPrecisionBeforeOverflow, |
| #if USE_CENTER_WIENER_NONSEP |
| 2 * kNumSymmetricTaps + 1, |
| #else |
| 2 * kNumSymmetricTaps, |
| #endif // USE_CENTER_WIENER_NONSEP |
| 0, |
| WienerNonsepConfig_, |
| NULL, |
| 0, |
| 1 |
| }; |
| |
| // Config used for filtering of chroma when CONFIG_WIENER_NONSEP=1. |
| const NonsepFilterConfig UnitSumFilterConfigChroma_ = { |
| kMaxPrecisionBeforeOverflow, |
| 2 * kNumSymmetricTapsChroma, |
| 0, |
| WienerNonsepConfigChroma_, |
| NULL, |
| 0, |
| 1 |
| }; |
| |
| int16_t FilterTaps_[kNumSymmetricTaps + 1]; |
| }; |
| |
| TEST_P(AV1ConvolveNonSep2DHighbdTest, RunTest) { RunTest(RESTORE_PC_WIENER); } |
| |
| TEST_P(AV1ConvolveNonSep2DHighbdTest, DISABLED_Speed) { |
| RunSpeedTest(RESTORE_PC_WIENER); |
| } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1ConvolveNonSep2DHighbdTest, |
| BuildHighbdParams(av1_convolve_symmetric_highbd_avx2)); |
| #endif |
| |
| class AV1ConvolveWienerNonSep2DHighbdTest |
| : public AV1ConvolveNonSep2DHighbdTest {}; |
| |
| TEST_P(AV1ConvolveWienerNonSep2DHighbdTest, RunTest) { |
| RunTest(RESTORE_WIENER_NONSEP); |
| } |
| TEST_P(AV1ConvolveWienerNonSep2DHighbdTest, DISABLED_Speed) { |
| RunSpeedTest(RESTORE_WIENER_NONSEP); |
| } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1ConvolveWienerNonSep2DHighbdTest, |
| BuildHighbdParams(av1_convolve_symmetric_subtract_center_highbd_avx2)); |
| #endif |
| |
| #endif // CONFIG_LR_IMPROVEMENTS |
| |
| ////////////////////////////////////////////////////////// |
| // Nonseparable convolve-2d Dual functions (high bit-depth) |
| ////////////////////////////////////////////////////////// |
| |
| #if CONFIG_WIENER_NONSEP_CROSS_FILT |
| typedef void (*highbd_convolve_nonsep_dual_2d_func)( |
| const uint16_t *dgd, int dgd_stride, const uint16_t *dgd_dual, |
| int dgd_dual_stride, const NonsepFilterConfig *filter_config, |
| const int16_t *filter, uint16_t *dst, int dst_stride, int bit_depth, |
| int block_row_begin, int block_row_end, int block_col_begin, |
| int block_col_end); |
| |
| class AV1ConvolveNon_Sep_dual2DHighbdTest |
| : public AV1ConvolveTest<highbd_convolve_nonsep_dual_2d_func> { |
| public: |
| void RunTest(int is_subtract_center) { |
| for (int i = 0; i < kTestIterations; i++) { |
| SetFilterTaps(); |
| TestConvolve(FilterTaps_, is_subtract_center); |
| } |
| } |
| void RunSpeedTest(int is_subtract_center) { |
| SpeedTestConvolve(FilterTaps_, is_subtract_center); |
| }; |
| |
| private: |
| libaom_test::ACMRandom rnd_; |
| static constexpr int kMaxPrecisionBeforeOverflow = 12; |
| static constexpr int kNumSymmetricTaps = 6; |
| // In dual filtering, 7 taps (6 symmetric + 1 center) are required for each of |
| // the buffer. |
| static constexpr int kNumSubtractCenterOffTaps = (2 * kNumSymmetricTaps) + 2; |
| static constexpr int kMaxTapOffset = 2; // Filters are 5x5. |
| static constexpr int kSpeedIterations = 10000; |
| static constexpr int kTestIterations = 100; |
| |
| // Declare the filter taps for worst case (i.e., for subtract center off |
| // case). |
| int16_t FilterTaps_[kNumSubtractCenterOffTaps]; |
| |
| // Fills the array p with signed integers. |
| void Randomize(int16_t *p, int size, int max_bit_range) { |
| ASSERT_TRUE(max_bit_range < 16) << "max_bit_range has to be less than 16"; |
| for (int i = 0; i < size; ++i) { |
| p[i] = rnd_.Rand15Signed() & ((1 << max_bit_range) - 1); |
| } |
| } |
| |
| void SetFilterTaps() { |
| Randomize(FilterTaps_, kNumSubtractCenterOffTaps, |
| kMaxPrecisionBeforeOverflow); |
| } |
| |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| |
| // Fills the array p with maximum and minimum possible integers. |
| void RandomizeExtreamFilterTap(int16_t *p, int size, int max_bit_range) { |
| ASSERT_TRUE(max_bit_range < 16) << "max_bit_range has to be less than 16"; |
| const int sign_max_val = (1 << (max_bit_range - 1)) - 1; |
| for (int i = 0; i < size; ++i) { |
| p[i] = static_cast<uint16_t>(RandBool() ? sign_max_val |
| : -(sign_max_val + 1)); |
| } |
| } |
| |
| void BitMatchTest(const uint16_t *dgd, const uint16_t *dgd_dual, |
| int dgd_stride, int width, int height, |
| const int16_t *filter, uint16_t *reference, uint16_t *test, |
| int dst_stride, int bit_depth, int block_row_begin, |
| int block_row_end, int block_col_begin, int block_col_end, |
| int is_subtract_center) { |
| // Set filter_config and reference function appropriately. |
| highbd_convolve_nonsep_dual_2d_func ref_func; |
| const NonsepFilterConfig *filter_cfg; |
| |
| filter_cfg = &DualFilterWithCenterConfig_; |
| ref_func = av1_convolve_symmetric_dual_subtract_center_highbd_c; |
| |
| if (!is_subtract_center) { |
| ref_func = av1_convolve_symmetric_dual_highbd_c; |
| filter_cfg = &DualFilterWithoutCenterConfig_; |
| } |
| // Reference function |
| ref_func(dgd, dgd_stride, dgd_dual, dgd_stride, filter_cfg, filter, |
| reference, dst_stride, bit_depth, block_row_begin, block_row_end, |
| block_col_begin, block_col_end); |
| |
| // Test function |
| GetParam().TestFunction()(dgd, dgd_stride, dgd_dual, dgd_stride, filter_cfg, |
| filter, test, dst_stride, bit_depth, |
| block_row_begin, block_row_end, block_col_begin, |
| block_col_end); |
| |
| // Compare the output of reference and test for bit match |
| AssertOutputBufferEq(reference, test, width, height); |
| } |
| |
| void TestConvolve(const int16_t *filter, int is_subtract_center) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| |
| const uint16_t *dgd = FirstRandomInput16(GetParam()); |
| const uint16_t *dgd_dual = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| |
| ASSERT_TRUE(kInputPadding >= kMaxTapOffset) |
| << "Not enough padding for 5x5 filters"; |
| const uint16_t *centered_input1 = |
| dgd + kMaxTapOffset * width + kMaxTapOffset; |
| const uint16_t *centered_input2 = |
| dgd_dual + kMaxTapOffset * width + kMaxTapOffset; |
| const int input_stride = width; |
| BitMatchTest(centered_input1, centered_input2, input_stride, width, height, |
| filter, reference, test, kOutputStride, bit_depth, 0, height, |
| 0, width, is_subtract_center); |
| // Extreme value test |
| const uint16_t *extreme_input1 = FirstRandomInput16Extreme(GetParam()); |
| const uint16_t *extreme_input2 = FirstRandomInput16Extreme(GetParam()); |
| const uint16_t *centered_extreme_input1 = |
| extreme_input1 + kMaxTapOffset * width + kMaxTapOffset; |
| const uint16_t *centered_extreme_input2 = |
| extreme_input2 + kMaxTapOffset * width + kMaxTapOffset; |
| int16_t Extream_Tap_[kNumSubtractCenterOffTaps]; |
| RandomizeExtreamFilterTap(Extream_Tap_, kNumSubtractCenterOffTaps, |
| kMaxPrecisionBeforeOverflow); |
| BitMatchTest(centered_extreme_input1, centered_extreme_input2, input_stride, |
| width, height, Extream_Tap_, reference, test, kOutputStride, |
| bit_depth, 0, height, 0, width, is_subtract_center); |
| } |
| |
| void SpeedTestConvolve(const int16_t *filter, int is_subtract_center) { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| const int bit_depth = GetParam().BitDepth(); |
| |
| const uint16_t *dgd = FirstRandomInput16(GetParam()); |
| const uint16_t *dgd_dual = FirstRandomInput16(GetParam()); |
| DECLARE_ALIGNED(32, uint16_t, test[MAX_SB_SQUARE]); |
| DECLARE_ALIGNED(32, uint16_t, reference[MAX_SB_SQUARE]); |
| |
| ASSERT_TRUE(kInputPadding >= kMaxTapOffset) |
| << "Not enough padding for 5x5 filters"; |
| const uint16_t *centered_input1 = |
| dgd + kMaxTapOffset * width + kMaxTapOffset; |
| const uint16_t *centered_input2 = |
| dgd_dual + kMaxTapOffset * width + kMaxTapOffset; |
| |
| // Set filter_config and reference function appropriately. |
| highbd_convolve_nonsep_dual_2d_func ref_func; |
| const NonsepFilterConfig *filter_cfg; |
| |
| filter_cfg = &DualFilterWithCenterConfig_; |
| ref_func = av1_convolve_symmetric_dual_subtract_center_highbd_c; |
| |
| if (!is_subtract_center) { |
| ref_func = av1_convolve_symmetric_dual_highbd_c; |
| filter_cfg = &DualFilterWithoutCenterConfig_; |
| } |
| |
| // Calculate time taken by reference/c function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| ref_func(centered_input1, width, centered_input2, width, filter_cfg, |
| filter, reference, kOutputStride, bit_depth, 0, height, 0, |
| width); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| GetParam().TestFunction()(centered_input1, width, centered_input2, width, |
| filter_cfg, filter, test, kOutputStride, |
| bit_depth, 0, height, 0, width); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / (kSpeedIterations * width * height); |
| float opt_time_per_pixel = |
| (float)1000.0 * elapsed_time_opt / (kSpeedIterations * width * height); |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| " %3dx%-3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| width, height, c_time_per_pixel, opt_time_per_pixel, scaling); |
| } |
| |
| const int wienerns_config_uv_from_uv[12][3] = { |
| { 1, 0, 0 }, { -1, 0, 0 }, { 0, 1, 1 }, { 0, -1, 1 }, |
| { 1, 1, 2 }, { -1, -1, 2 }, { -1, 1, 3 }, { 1, -1, 3 }, |
| { 2, 0, 4 }, { -2, 0, 4 }, { 0, 2, 5 }, { 0, -2, 5 }, |
| }; |
| |
| const int wienerns_config_uv_from_y[12][3] = { |
| { 1, 0, 6 }, { -1, 0, 6 }, { 0, 1, 7 }, { 0, -1, 7 }, |
| { 1, 1, 8 }, { -1, -1, 8 }, { -1, 1, 9 }, { 1, -1, 9 }, |
| { 2, 0, 10 }, { -2, 0, 10 }, { 0, 2, 11 }, { 0, -2, 11 }, |
| }; |
| |
| const int wienerns_wout_subtract_center_config_uv_from_uv[13][3] = { |
| { 1, 0, 0 }, { -1, 0, 0 }, { 0, 1, 1 }, { 0, -1, 1 }, { 1, 1, 2 }, |
| { -1, -1, 2 }, { -1, 1, 3 }, { 1, -1, 3 }, { 2, 0, 4 }, { -2, 0, 4 }, |
| { 0, 2, 5 }, { 0, -2, 5 }, { 0, 0, 6 }, |
| }; |
| |
| // Adjust the beginning tap to account for the above change and add a tap at |
| // (0, 0). |
| const int wienerns_wout_subtract_center_config_uv_from_y[13][3] = { |
| { 1, 0, 7 }, { -1, 0, 7 }, { 0, 1, 8 }, { 0, -1, 8 }, { 1, 1, 9 }, |
| { -1, -1, 9 }, { -1, 1, 10 }, { 1, -1, 10 }, { 2, 0, 11 }, { -2, 0, 11 }, |
| { 0, 2, 12 }, { 0, -2, 12 }, { 0, 0, 13 }, |
| }; |
| |
| const NonsepFilterConfig DualFilterWithCenterConfig_ = { |
| kMaxPrecisionBeforeOverflow, // prec_bits; |
| sizeof(wienerns_config_uv_from_uv) / |
| sizeof(wienerns_config_uv_from_uv[0]), // num_pixels; |
| sizeof(wienerns_config_uv_from_y) / |
| sizeof(wienerns_config_uv_from_y[0]), // num_pixels2 |
| wienerns_config_uv_from_uv, // config |
| wienerns_config_uv_from_y, // config2 |
| 0, // strict_bounds |
| 1 // subtract_center |
| }; |
| |
| const NonsepFilterConfig DualFilterWithoutCenterConfig_ = { |
| kMaxPrecisionBeforeOverflow, // prec_bits; |
| sizeof(wienerns_wout_subtract_center_config_uv_from_uv) / |
| sizeof( |
| wienerns_wout_subtract_center_config_uv_from_uv[0]), // num_pixels; |
| sizeof(wienerns_wout_subtract_center_config_uv_from_y) / |
| sizeof( |
| wienerns_wout_subtract_center_config_uv_from_y[0]), // num_pixels2 |
| wienerns_wout_subtract_center_config_uv_from_uv, // config |
| wienerns_wout_subtract_center_config_uv_from_y, // config2 |
| 0, // strict_bounds |
| 0 // subtract_center |
| }; |
| }; |
| |
| TEST_P(AV1ConvolveNon_Sep_dual2DHighbdTest, RunTest) { RunTest(1); } |
| TEST_P(AV1ConvolveNon_Sep_dual2DHighbdTest, DISABLED_Speed) { RunSpeedTest(1); } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1ConvolveNon_Sep_dual2DHighbdTest, |
| BuildHighbdParams(av1_convolve_symmetric_dual_subtract_center_highbd_avx2)); |
| #endif // HAVE_AVX2 |
| |
| /* Dual with subtract center off unit-test*/ |
| class AV1ConvolveDualWithoutsubtract2DHighbdTest |
| : public AV1ConvolveNon_Sep_dual2DHighbdTest {}; |
| |
| TEST_P(AV1ConvolveDualWithoutsubtract2DHighbdTest, RunTest) { RunTest(0); } |
| TEST_P(AV1ConvolveDualWithoutsubtract2DHighbdTest, DISABLED_Speed) { |
| RunSpeedTest(0); |
| } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1ConvolveDualWithoutsubtract2DHighbdTest, |
| BuildHighbdParams(av1_convolve_symmetric_dual_highbd_avx2)); |
| #endif |
| |
| #endif // CONFIG_WIENER_NONSEP_CROSS_FILT |
| |
| ////////////////////////////////////////////////////////// |
| // Unit-test corresponds to buffer accumulations to derive filter |
| // index for each block size (pc_wiener_block_size: 4x4) |
| ////////////////////////////////////////////////////////// |
| |
| #if CONFIG_LR_IMPROVEMENTS |
| |
| // Generate the list of all block widths / heights that need to be tested for |
| // pc_wiener. |
| template <typename T> |
| std::vector<TestParam<T>> GetPCWienerTestParams( |
| std::initializer_list<int> bit_depths, T test_func) { |
| std::set<BlockSize> sizes; |
| for (int b = BLOCK_4X4; b < BLOCK_SIZES_ALL; ++b) { |
| const int w = block_size_wide[b]; |
| const int h = block_size_high[b]; |
| if (w > RESTORATION_PROC_UNIT_SIZE || h > RESTORATION_PROC_UNIT_SIZE) { |
| continue; |
| } |
| sizes.insert(BlockSize(w, h)); |
| // Add in smaller chroma sizes as well. |
| if (w == 4 || h == 4) { |
| sizes.insert(BlockSize(w / 2, h / 2)); |
| } |
| } |
| std::vector<TestParam<T>> result; |
| for (const BlockSize &block : sizes) { |
| for (int bd : bit_depths) { |
| result.push_back(TestParam<T>(block, bd, test_func)); |
| } |
| } |
| return result; |
| } |
| template <typename T> |
| ::testing::internal::ParamGenerator<TestParam<T>> BuildHighbdPCWienerParams( |
| T test_func) { |
| return ::testing::ValuesIn(GetPCWienerTestParams({ 10, 12 }, test_func)); |
| } |
| |
| typedef void (*fill_directional_feature_buffers_highbd_func)( |
| int *feature_sum_buffers[], int16_t *feature_line_buffers[], int row, |
| int buffer_row, const uint16_t *dgd, int dgd_stride, int width, |
| int feature_lead, int feature_lag); |
| |
| class AV1FillDirFeatureBufHighbdTest |
| : public AV1ConvolveTest<fill_directional_feature_buffers_highbd_func> { |
| public: |
| void RunTest() { |
| for (int i = 0; i < kTestIterations; i++) { |
| // Set buffer values here. |
| SetBufferValues(); |
| TestConvolve(); |
| } |
| } |
| |
| void RunSpeedTest() { SpeedTestConvolve(); }; |
| |
| protected: |
| virtual void SetUp() { |
| for (int j = 0; j < NUM_FEATURE_LINE_BUFFERS; ++j) { |
| feature_line_buffers_c_[j] = static_cast<int16_t *>( |
| (aom_malloc(buffer_width_ * sizeof(*feature_line_buffers_c_[j])))); |
| ASSERT_NE(feature_line_buffers_c_[j], nullptr); |
| |
| feature_line_buffers_simd_[j] = static_cast<int16_t *>( |
| (aom_malloc(buffer_width_ * sizeof(*feature_line_buffers_simd_[j])))); |
| ASSERT_NE(feature_line_buffers_simd_[j], nullptr); |
| } |
| |
| for (int j = 0; j < NUM_PC_WIENER_FEATURES; ++j) { |
| feature_sum_buffers_c_[j] = static_cast<int *>( |
| (aom_malloc(buffer_width_ * sizeof(*feature_sum_buffers_c_[j])))); |
| ASSERT_NE(feature_sum_buffers_c_[j], nullptr); |
| |
| feature_sum_buffers_simd_[j] = static_cast<int *>( |
| (aom_malloc(buffer_width_ * sizeof(*feature_sum_buffers_simd_[j])))); |
| ASSERT_NE(feature_sum_buffers_simd_[j], nullptr); |
| } |
| } |
| |
| virtual void TearDown() { |
| for (int j = 0; j < NUM_FEATURE_LINE_BUFFERS; ++j) { |
| aom_free(feature_line_buffers_c_[j]); |
| feature_line_buffers_c_[j] = NULL; |
| aom_free(feature_line_buffers_simd_[j]); |
| feature_line_buffers_simd_[j] = NULL; |
| } |
| |
| for (int j = 0; j < NUM_PC_WIENER_FEATURES; ++j) { |
| aom_free(feature_sum_buffers_c_[j]); |
| feature_sum_buffers_c_[j] = NULL; |
| aom_free(feature_sum_buffers_simd_[j]); |
| feature_sum_buffers_simd_[j] = NULL; |
| } |
| } |
| |
| void SetBufferValues() { |
| const int bitdepth = GetParam().BitDepth(); |
| for (int j = 0; j < NUM_FEATURE_LINE_BUFFERS; ++j) { |
| Randomize(feature_line_buffers_c_[j], buffer_width_, bitdepth); |
| memcpy(feature_line_buffers_simd_[j], feature_line_buffers_c_[j], |
| buffer_width_ * sizeof(*feature_line_buffers_simd_[j])); |
| } |
| |
| for (int j = 0; j < NUM_PC_WIENER_FEATURES; ++j) { |
| RandomizeSigned31(feature_sum_buffers_c_[j], buffer_width_, 31); |
| memcpy(feature_sum_buffers_simd_[j], feature_sum_buffers_c_[j], |
| buffer_width_ * sizeof(*feature_sum_buffers_simd_[j])); |
| } |
| } |
| |
| private: |
| libaom_test::ACMRandom rnd_; |
| static constexpr int kSpeedIterations = 10000; |
| static constexpr int kTestIterations = 100; |
| |
| void TestConvolve() { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| // Input buffer allocation. |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| const int input_stride = width; |
| |
| // C function call |
| for (int i = 0; i < height; ++i) { |
| const int row_to_process = AOMMIN(i + feature_lag, height + 3 - 2); |
| fill_directional_feature_buffers_highbd_c( |
| feature_sum_buffers_c_, feature_line_buffers_c_, row_to_process, |
| feature_length - 1, input, input_stride, width, feature_lead, |
| feature_lag); |
| } |
| |
| // SIMD function call |
| for (int i = 0; i < height; ++i) { |
| const int row_to_process = AOMMIN(i + feature_lag, height + 3 - 2); |
| GetParam().TestFunction()(feature_sum_buffers_simd_, |
| feature_line_buffers_simd_, row_to_process, |
| feature_length - 1, input, input_stride, width, |
| feature_lead, feature_lag); |
| } |
| |
| // Compare the outputs of C and SIMD |
| for (int i = 0; i < NUM_PC_WIENER_FEATURES; i++) { |
| int *c_buf = feature_sum_buffers_c_[i]; |
| int *simd_buf = feature_sum_buffers_simd_[i]; |
| for (int j = 0; j < buffer_width_; ++j) { |
| ASSERT_EQ(c_buf[j], simd_buf[j]) |
| << "feature_buf=" << i << " Pixel mismatch at width (" << i << ")"; |
| } |
| } |
| } |
| |
| void SpeedTestConvolve() { |
| const int width = GetParam().Block().Width(); |
| const int height = GetParam().Block().Height(); |
| |
| // Input buffer allocation. |
| const uint16_t *input = FirstRandomInput16(GetParam()); |
| const int input_stride = width; |
| |
| // Calculate time taken for C function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| for (int i = 0; i < height; ++i) { |
| const int row_to_process = AOMMIN(i + feature_lag, height + 3 - 2); |
| fill_directional_feature_buffers_highbd_c( |
| feature_sum_buffers_c_, feature_line_buffers_c_, row_to_process, |
| feature_length - 1, input, input_stride, width, feature_lead, |
| feature_lag); |
| } |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| for (int i = 0; i < height; ++i) { |
| const int row_to_process = AOMMIN(i + feature_lag, height + 3 - 2); |
| GetParam().TestFunction()(feature_sum_buffers_simd_, |
| feature_line_buffers_simd_, row_to_process, |
| feature_length - 1, input, input_stride, |
| width, feature_lead, feature_lag); |
| } |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / (kSpeedIterations * width * height); |
| float opt_time_per_pixel = |
| (float)1000.0 * elapsed_time_opt / (kSpeedIterations * width * height); |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| "%3dx%-3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| width, height, c_time_per_pixel, opt_time_per_pixel, scaling); |
| } |
| |
| // Fills the array p with signed integers. |
| void Randomize(int16_t *p, int size, int max_bit_range) { |
| ASSERT_TRUE(max_bit_range < 16) << "max_bit_range has to be less than 16"; |
| for (int i = 0; i < size; ++i) { |
| p[i] = rnd_.Rand15Signed() & ((1 << max_bit_range) - 1); |
| } |
| } |
| |
| // Fills the array p with signed integers of 31 bit range. |
| void RandomizeSigned31(int *p, int size, uint32_t max_bit_range) { |
| assert(max_bit_range <= 31); |
| uint32_t mask = (uint32_t)(1 << max_bit_range) - 1; |
| for (int i = 0; i < size; ++i) { |
| p[i] = (int)(rnd_.Rand31() & mask); |
| } |
| } |
| |
| int *feature_sum_buffers_c_[NUM_PC_WIENER_FEATURES]; |
| int *feature_sum_buffers_simd_[NUM_PC_WIENER_FEATURES]; |
| int16_t *feature_line_buffers_c_[NUM_FEATURE_LINE_BUFFERS]; |
| int16_t *feature_line_buffers_simd_[NUM_FEATURE_LINE_BUFFERS]; |
| const int feature_lead = PC_WIENER_FEATURE_LEAD_LUMA; |
| const int feature_lag = PC_WIENER_FEATURE_LAG_LUMA; |
| const int feature_length = PC_WIENER_FEATURE_LENGTH_LUMA; |
| const int buffer_width_ = MAX_SB_SIZE + kInputPadding; |
| }; |
| |
| TEST_P(AV1FillDirFeatureBufHighbdTest, RunTest) { RunTest(); } |
| |
| TEST_P(AV1FillDirFeatureBufHighbdTest, DISABLED_Speed) { RunSpeedTest(); } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1FillDirFeatureBufHighbdTest, |
| BuildHighbdPCWienerParams(fill_directional_feature_buffers_highbd_avx2)); |
| #endif // HAVE_AVX2 |
| |
| typedef void (*FillTSkipSumBufferFunc)(int row, const uint8_t *tskip, |
| int tskip_stride, |
| int8_t *tskip_sum_buffer, int width, |
| int height, int tskip_lead, |
| int tskip_lag, bool use_strict_bounds); |
| |
| typedef std::tuple<const FillTSkipSumBufferFunc> AV1FillTSkipSumBufferFuncParam; |
| |
| class AV1Fill_TSkip_Sum_BufferTest |
| : public ::testing::TestWithParam<AV1FillTSkipSumBufferFuncParam> { |
| public: |
| virtual void SetUp() { target_func_ = GET_PARAM(0); } |
| |
| void RunTest() { |
| for (int i = 0; i < kTestIterations; i++) { |
| TestTSkipSum(); |
| } |
| } |
| void RunSpeedTest() { SpeedTestTSkipSum(); }; |
| |
| private: |
| libaom_test::ACMRandom rnd_; |
| FillTSkipSumBufferFunc target_func_; |
| |
| static constexpr int kSpeedIterations = 10000; |
| static constexpr int kTestIterations = 100; |
| static constexpr int kNumPlanes = 1; |
| static constexpr int kWidth = RESTORATION_PROC_UNIT_SIZE; |
| static constexpr int kHeight = RESTORATION_PROC_UNIT_SIZE; |
| static constexpr int kInputWidth = MI_SIZE_64X64; |
| static constexpr int kInputStride = MI_SIZE_64X64; |
| static constexpr int kOutputWidth = |
| (RESTORATION_PROC_UNIT_SIZE + PC_WIENER_FEATURE_LENGTH_LUMA - 1); |
| |
| uint8_t input_buffer_[MI_SIZE_64X64 * MI_SIZE_64X64]; |
| int8_t ref_buffer_[kOutputWidth]; |
| int8_t test_buffer_[kOutputWidth]; |
| const bool tskip_strict_ = true; |
| |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| |
| void TestTSkipSum() { |
| for (int i = 0; i < kInputWidth * kInputStride; ++i) { |
| input_buffer_[i] = static_cast<uint8_t>(RandBool() ? 1 : 0); |
| } |
| |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int ss_y = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int plane_height = kHeight >> ss_y; |
| const int tskip_lead = PC_WIENER_TSKIP_LEAD_LUMA; |
| const int tskip_lag = PC_WIENER_TSKIP_LAG_LUMA; |
| |
| memset(ref_buffer_, 0, sizeof(*ref_buffer_) * kOutputWidth); |
| memset(test_buffer_, 0, sizeof(*test_buffer_) * kOutputWidth); |
| |
| // Reference function |
| for (int row = -tskip_lead; row < (tskip_lag + plane_height); ++row) { |
| av1_fill_tskip_sum_buffer_c(row, input_buffer_, kInputStride, |
| ref_buffer_, plane_width, plane_height, |
| tskip_lead, tskip_lag, tskip_strict_); |
| } |
| |
| // Test function |
| for (int row = -tskip_lead; row < (tskip_lag + plane_height); ++row) { |
| target_func_(row, input_buffer_, kInputStride, test_buffer_, |
| plane_width, plane_height, tskip_lead, tskip_lag, |
| tskip_strict_); |
| } |
| |
| // Compare the output of reference and test for bit match |
| for (int i = 0; i < kOutputWidth; ++i) { |
| ASSERT_EQ(ref_buffer_[i], test_buffer_[i]) |
| << " Mismatch at (" << i << ")"; |
| } |
| } |
| } |
| |
| void SpeedTestTSkipSum() { |
| for (int i = 0; i < kInputWidth * kInputStride; ++i) { |
| input_buffer_[i] = static_cast<uint8_t>(RandBool() ? 1 : 0); |
| } |
| |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int ss_y = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int plane_height = kHeight >> ss_y; |
| const int tskip_lead = PC_WIENER_TSKIP_LEAD_LUMA; |
| const int tskip_lag = PC_WIENER_TSKIP_LAG_LUMA; |
| |
| memset(ref_buffer_, 0, sizeof(*ref_buffer_) * kOutputWidth); |
| memset(test_buffer_, 0, sizeof(*test_buffer_) * kOutputWidth); |
| |
| // Calculate time taken by reference/c function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| // Reference function |
| for (int row = -tskip_lead; row < (tskip_lag + plane_height - 1); |
| ++row) { |
| av1_fill_tskip_sum_buffer_c(row, input_buffer_, kInputStride, |
| ref_buffer_, plane_width, plane_height, |
| tskip_lead, tskip_lag, tskip_strict_); |
| } |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| for (int row = -tskip_lead; row < (tskip_lag + plane_height - 1); |
| ++row) { |
| target_func_(row, input_buffer_, kInputStride, test_buffer_, |
| plane_width, plane_height, tskip_lead, tskip_lag, |
| tskip_strict_); |
| } |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / kSpeedIterations; |
| float opt_time_per_pixel = |
| (float)1000.0 * elapsed_time_opt / kSpeedIterations; |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| " %3dx%-3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| plane_width, plane_height, c_time_per_pixel, opt_time_per_pixel, |
| scaling); |
| } |
| } |
| }; |
| |
| TEST_P(AV1Fill_TSkip_Sum_BufferTest, RunTest) { RunTest(); } |
| TEST_P(AV1Fill_TSkip_Sum_BufferTest, DISABLED_Speed) { RunSpeedTest(); } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P(AVX2, AV1Fill_TSkip_Sum_BufferTest, |
| ::testing::Values(av1_fill_tskip_sum_buffer_avx2)); |
| #endif // HAVE_AVX2 |
| |
| ////////////////////////////////////////////////////////// |
| // unit-test for 'directional_feature_accum' // |
| ////////////////////////////////////////////////////////// |
| typedef void (*FillDirFeatureAccumFunc)( |
| int dir_feature_accum[NUM_PC_WIENER_FEATURES][PC_WIENER_FEATURE_ACC_SIZE], |
| int *feature_sum_buf[NUM_PC_WIENER_FEATURES], int width, int col_offset, |
| int feature_lead, int feature_lag); |
| |
| typedef std::tuple<const FillDirFeatureAccumFunc> |
| AV1FillDirFeatureAccumFuncParam; |
| |
| class AV1FeatureDirAccumHighbdTest |
| : public ::testing::TestWithParam<AV1FillDirFeatureAccumFuncParam> { |
| public: |
| void RunTest() { |
| for (int i = 0; i < kTestIterations; i++) { |
| FillInputBufs(); |
| TestFillDirFeatureAccum(); |
| } |
| } |
| |
| void RunSpeedTest() { SpeedTestConvolve(); }; |
| |
| virtual void SetUp() { |
| target_func_ = GET_PARAM(0); |
| |
| for (int j = 0; j < NUM_PC_WIENER_FEATURES; ++j) { |
| feature_sum_buf[j] = |
| (int *)(aom_malloc(kInputWidth * sizeof(*feature_sum_buf[j]))); |
| } |
| } |
| |
| virtual void TearDown() { |
| for (int j = 0; j < NUM_PC_WIENER_FEATURES; ++j) { |
| aom_free(feature_sum_buf[j]); |
| feature_sum_buf[j] = NULL; |
| } |
| } |
| |
| private: |
| libaom_test::ACMRandom rnd_; |
| FillDirFeatureAccumFunc target_func_; |
| |
| static constexpr int kSpeedIterations = 1000000; |
| static constexpr int kTestIterations = 100; |
| static constexpr int kNumPlanes = 2; |
| static constexpr int kWidth = RESTORATION_PROC_UNIT_SIZE; |
| static constexpr int kInputWidth = |
| (RESTORATION_PROC_UNIT_SIZE + PC_WIENER_FEATURE_LENGTH_LUMA - 1); |
| |
| int *feature_sum_buf[NUM_PC_WIENER_FEATURES]; |
| int dir_feature_accum_buf_c[NUM_PC_WIENER_FEATURES] |
| [PC_WIENER_FEATURE_ACC_SIZE] = { { 0 } }; |
| int dir_feature_accum_buf_simd[NUM_PC_WIENER_FEATURES] |
| [PC_WIENER_FEATURE_ACC_SIZE] = { { 0 } }; |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| |
| void FillInputBufs() { |
| for (int i = 0; i < NUM_PC_WIENER_FEATURES; ++i) { |
| for (int j = 0; j < kInputWidth; ++j) { |
| // For the extreme values case, the maimum input that feature_sum_buf |
| // can take is (kInputWidth * 2 * input_max_value). Hence, clipping the |
| // value generated to 23 bit. |
| const int max_range = (1 << 23); |
| const int value = rnd_.Rand31() % max_range; |
| feature_sum_buf[i][j] = |
| static_cast<uint8_t>(RandBool() ? value : -value); |
| } |
| } |
| // Reset output buffers |
| av1_zero(dir_feature_accum_buf_c); |
| av1_zero(dir_feature_accum_buf_simd); |
| } |
| |
| void TestFillDirFeatureAccum() { |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int feature_lead = PC_WIENER_FEATURE_LEAD_LUMA; |
| const int feature_lag = PC_WIENER_FEATURE_LAG_LUMA; |
| |
| // Reset output buffers |
| av1_zero(dir_feature_accum_buf_c); |
| av1_zero(dir_feature_accum_buf_simd); |
| |
| // C function call |
| av1_fill_directional_feature_accumulators_c( |
| dir_feature_accum_buf_c, feature_sum_buf, plane_width, feature_lag, |
| feature_lead, feature_lag); |
| |
| // SIMD function call |
| target_func_(dir_feature_accum_buf_simd, feature_sum_buf, plane_width, |
| feature_lag, feature_lead, feature_lag); |
| |
| // Compare the output of reference and test for bit match |
| for (int i = 0; i < NUM_PC_WIENER_FEATURES; i++) { |
| for (int j = 0; j < PC_WIENER_FEATURE_ACC_SIZE; j++) { |
| ASSERT_EQ(dir_feature_accum_buf_c[i][j], |
| dir_feature_accum_buf_simd[i][j]) |
| << " Feature_Buf: Pixel mismatch at (" << i << ", " << j << ", " |
| << plane_width << ")"; |
| } |
| } |
| } |
| } |
| |
| void SpeedTestConvolve() { |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int feature_lead = PC_WIENER_FEATURE_LEAD_LUMA; |
| const int feature_lag = PC_WIENER_FEATURE_LAG_LUMA; |
| FillInputBufs(); |
| |
| // Calculate time taken by reference/c function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| av1_fill_directional_feature_accumulators_c( |
| dir_feature_accum_buf_c, feature_sum_buf, plane_width, feature_lag, |
| feature_lead, feature_lag); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| target_func_(dir_feature_accum_buf_simd, feature_sum_buf, plane_width, |
| feature_lag, feature_lead, feature_lag); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / (kSpeedIterations * plane_width); |
| float opt_time_per_pixel = |
| (float)1000.0 * elapsed_time_opt / (kSpeedIterations * plane_width); |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| " %3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| plane_width, c_time_per_pixel, opt_time_per_pixel, scaling); |
| } |
| } |
| }; |
| |
| TEST_P(AV1FeatureDirAccumHighbdTest, RunTest) { RunTest(); } |
| TEST_P(AV1FeatureDirAccumHighbdTest, DISABLED_Speed) { RunSpeedTest(); } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1FeatureDirAccumHighbdTest, |
| ::testing::Values(av1_fill_directional_feature_accumulators_avx2)); |
| #endif // HAVE_AVX2 |
| |
| ////////////////////////////////////////////////////////// |
| // unit-test for 'fill_tskip_feature_accumulator' // |
| ////////////////////////////////////////////////////////// |
| typedef void (*FillTskip_Accumulator_func)( |
| int16_t tskip_feature_accum[PC_WIENER_FEATURE_ACC_SIZE], |
| int8_t *tskip_sum_buff, int width, int col_offset, int tskip_lead, |
| int tskip_lag); |
| typedef std::tuple<const FillTskip_Accumulator_func> |
| AV1FillTSkipAccumBufferFuncParam; |
| |
| class AV1TskipAccumHighbdTest |
| : public ::testing::TestWithParam<AV1FillTSkipAccumBufferFuncParam> { |
| public: |
| virtual void SetUp() { target_func_ = GET_PARAM(0); } |
| |
| void RunTest() { |
| for (int i = 0; i < kTestIterations; i++) TestTskipAccum(); |
| } |
| |
| void RunSpeedTest() { SpeedTestTskipAccum(); }; |
| |
| private: |
| libaom_test::ACMRandom rnd_; |
| FillTskip_Accumulator_func target_func_; |
| |
| static constexpr int kSpeedIterations = 1000000; |
| static constexpr int kTestIterations = 100; |
| static constexpr int kNumPlanes = 2; |
| static constexpr int kWidth = RESTORATION_PROC_UNIT_SIZE; |
| static constexpr int kInputWidth = |
| (RESTORATION_PROC_UNIT_SIZE + PC_WIENER_FEATURE_LENGTH_LUMA - 1); |
| |
| int8_t *tskip_sum_buf; |
| int16_t tskip_feature_accum_c[PC_WIENER_FEATURE_ACC_SIZE] = { 0 }; |
| int16_t tskip_feature_accum_simd[PC_WIENER_FEATURE_ACC_SIZE] = { 0 }; |
| |
| void buffer_alloc_and_set_data() { |
| tskip_sum_buf = |
| (int8_t *)(aom_malloc(kInputWidth * sizeof(*tskip_sum_buf))); |
| // Input buffer filling. Tskip buffer max value will not cross width of |
| // restoration unit size. Hence, the generated values are clipped to the |
| // same. |
| for (int i = 0; i < kInputWidth; ++i) { |
| const int8_t value = |
| static_cast<int8_t>(rnd_.Rand8() % RESTORATION_PROC_UNIT_SIZE); |
| tskip_sum_buf[i] = static_cast<uint8_t>(RandBool() ? value : -value); |
| } |
| } |
| |
| int RandBool() { |
| const uint32_t value = rnd_.Rand8(); |
| // There's a bit more entropy in the upper bits of this implementation. |
| return (value >> 7) & 0x1; |
| } |
| |
| void TestTskipAccum() { |
| // Allocate memory and fill input buffer |
| buffer_alloc_and_set_data(); |
| |
| // Loop over luma and chroma plane |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int tskip_lead = PC_WIENER_TSKIP_LEAD_LUMA; |
| const int tskip_lag = PC_WIENER_TSKIP_LAG_LUMA; |
| av1_zero(tskip_feature_accum_c); |
| av1_zero(tskip_feature_accum_simd); |
| |
| // C function call |
| av1_fill_tskip_feature_accumulator_c(tskip_feature_accum_c, tskip_sum_buf, |
| plane_width, tskip_lag, tskip_lead, |
| tskip_lag); |
| |
| // SIMD function call |
| target_func_(tskip_feature_accum_simd, tskip_sum_buf, plane_width, |
| tskip_lag, tskip_lead, tskip_lag); |
| |
| // Compare the output of reference and test for bit match |
| for (int i = 0; i < PC_WIENER_FEATURE_ACC_SIZE; i++) { |
| ASSERT_EQ(tskip_feature_accum_c[i], tskip_feature_accum_simd[i]) |
| << " Feature_Buf: Pixel mismatch at (" << i << "," << plane_width |
| << ")"; |
| } |
| } |
| aom_free(tskip_sum_buf); |
| tskip_sum_buf = NULL; |
| } |
| |
| void SpeedTestTskipAccum() { |
| // Allocate memory and fill input buffer |
| buffer_alloc_and_set_data(); |
| |
| for (int plane = 0; plane < kNumPlanes; ++plane) { |
| const int is_uv = (plane > 0); |
| const int ss_x = is_uv ? 1 : 0; |
| const int plane_width = kWidth >> ss_x; |
| const int tskip_lead = PC_WIENER_TSKIP_LEAD_LUMA; |
| const int tskip_lag = PC_WIENER_TSKIP_LAG_LUMA; |
| |
| // Calculate time taken by reference/c function |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| av1_fill_tskip_feature_accumulator_c(tskip_feature_accum_c, |
| tskip_sum_buf, plane_width, |
| tskip_lag, tskip_lead, tskip_lag); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_c = aom_usec_timer_elapsed(&timer); |
| |
| // Calculate time taken by optimized/intrinsic function |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < kSpeedIterations; ++i) { |
| target_func_(tskip_feature_accum_simd, tskip_sum_buf, plane_width, |
| tskip_lag, tskip_lead, tskip_lag); |
| } |
| aom_usec_timer_mark(&timer); |
| auto elapsed_time_opt = aom_usec_timer_elapsed(&timer); |
| |
| float c_time_per_pixel = |
| (float)1000.0 * elapsed_time_c / (kSpeedIterations * plane_width); |
| float opt_time_per_pixel = |
| (float)1000.0 * elapsed_time_opt / (kSpeedIterations * plane_width); |
| float scaling = c_time_per_pixel / opt_time_per_pixel; |
| printf( |
| " %3d: c_time_per_pixel=%10.5f, " |
| "opt_time_per_pixel=%10.5f, scaling=%f \n", |
| plane_width, c_time_per_pixel, opt_time_per_pixel, scaling); |
| } |
| aom_free(tskip_sum_buf); |
| tskip_sum_buf = NULL; |
| } |
| }; |
| |
| TEST_P(AV1TskipAccumHighbdTest, RunTest) { RunTest(); } |
| TEST_P(AV1TskipAccumHighbdTest, DISABLED_Speed) { RunSpeedTest(); } |
| |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, AV1TskipAccumHighbdTest, |
| ::testing::Values(av1_fill_tskip_feature_accumulator_avx2)); |
| #endif // HAVE_AVX2 |
| #endif // CONFIG_LR_IMPROVEMENTS |
| } // namespace |