| /* |
| * 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 <stdint.h> |
| #include <stdio.h> |
| #include <string.h> |
| #include <tuple> |
| |
| #include "third_party/googletest/src/googletest/include/gtest/gtest.h" |
| |
| #include "config/aom_config.h" |
| #include "config/av1_rtcd.h" |
| |
| #include "aom_ports/aom_timer.h" |
| #include "aom_ports/mem.h" |
| #include "av1/common/av1_common_int.h" |
| #include "av1/common/idct.h" |
| #include "av1/common/scan.h" |
| #include "av1/common/txb_common.h" |
| #include "test/acm_random.h" |
| #include "test/clear_system_state.h" |
| #include "test/register_state_check.h" |
| #include "test/util.h" |
| |
| namespace { |
| using libaom_test::ACMRandom; |
| |
| typedef void (*GetNzMapContextsFunc)(const uint8_t *const levels, |
| const int16_t *const scan, |
| const uint16_t eob, const TX_SIZE tx_size, |
| const TX_CLASS tx_class, |
| int8_t *const coeff_contexts); |
| |
| class EncodeTxbTest : public ::testing::TestWithParam<GetNzMapContextsFunc> { |
| public: |
| EncodeTxbTest() : get_nz_map_contexts_func_(GetParam()) {} |
| |
| virtual ~EncodeTxbTest() {} |
| |
| virtual void SetUp() { |
| coeff_contexts_ref_ = reinterpret_cast<int8_t *>( |
| aom_memalign(16, sizeof(*coeff_contexts_ref_) * MAX_TX_SQUARE)); |
| ASSERT_TRUE(coeff_contexts_ref_ != NULL); |
| coeff_contexts_ = reinterpret_cast<int8_t *>( |
| aom_memalign(16, sizeof(*coeff_contexts_) * MAX_TX_SQUARE)); |
| ASSERT_TRUE(coeff_contexts_ != NULL); |
| } |
| |
| virtual void TearDown() { |
| aom_free(coeff_contexts_ref_); |
| aom_free(coeff_contexts_); |
| libaom_test::ClearSystemState(); |
| } |
| |
| void GetNzMapContextsRun() { |
| const int kNumTests = 10; |
| int result = 0; |
| |
| for (int is_inter = 0; is_inter < 2; ++is_inter) { |
| for (int tx_type = DCT_DCT; tx_type < TX_TYPES; ++tx_type) { |
| const TX_CLASS tx_class = tx_type_to_class[tx_type]; |
| for (int tx_size = TX_4X4; tx_size < TX_SIZES_ALL; ++tx_size) { |
| const int bwl = get_txb_bwl((TX_SIZE)tx_size); |
| const int width = get_txb_wide((TX_SIZE)tx_size); |
| const int height = get_txb_high((TX_SIZE)tx_size); |
| const int real_width = tx_size_wide[tx_size]; |
| const int real_height = tx_size_high[tx_size]; |
| const int16_t *const scan = av1_scan_orders[tx_size][tx_type].scan; |
| |
| levels_ = set_levels(levels_buf_, width); |
| for (int i = 0; i < kNumTests && !result; ++i) { |
| for (int eob = 1; eob <= width * height && !result; ++eob) { |
| InitDataWithEob(scan, bwl, eob); |
| |
| av1_get_nz_map_contexts_c(levels_, scan, eob, (TX_SIZE)tx_size, |
| tx_class, coeff_contexts_ref_); |
| get_nz_map_contexts_func_(levels_, scan, eob, (TX_SIZE)tx_size, |
| tx_class, coeff_contexts_); |
| |
| result = Compare(scan, eob); |
| |
| EXPECT_EQ(result, 0) |
| << " tx_class " << tx_class << " width " << real_width |
| << " height " << real_height << " eob " << eob; |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| void SpeedTestGetNzMapContextsRun() { |
| const int kNumTests = 2000000000; |
| aom_usec_timer timer; |
| aom_usec_timer timer_ref; |
| |
| printf("Note: Only test the largest possible eob case!\n"); |
| for (int tx_size = TX_4X4; tx_size < TX_SIZES_ALL; ++tx_size) { |
| const int bwl = get_txb_bwl((TX_SIZE)tx_size); |
| const int width = get_txb_wide((TX_SIZE)tx_size); |
| const int height = get_txb_high((TX_SIZE)tx_size); |
| const int real_width = tx_size_wide[tx_size]; |
| const int real_height = tx_size_high[tx_size]; |
| const TX_TYPE tx_type = DCT_DCT; |
| const TX_CLASS tx_class = tx_type_to_class[tx_type]; |
| const int16_t *const scan = av1_scan_orders[tx_size][tx_type].scan; |
| const int eob = width * height; |
| const int numTests = kNumTests / (width * height); |
| |
| levels_ = set_levels(levels_buf_, width); |
| InitDataWithEob(scan, bwl, eob); |
| |
| aom_usec_timer_start(&timer_ref); |
| for (int i = 0; i < numTests; ++i) { |
| av1_get_nz_map_contexts_c(levels_, scan, eob, (TX_SIZE)tx_size, |
| tx_class, coeff_contexts_ref_); |
| } |
| aom_usec_timer_mark(&timer_ref); |
| |
| levels_ = set_levels(levels_buf_, width); |
| InitDataWithEob(scan, bwl, eob); |
| |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < numTests; ++i) { |
| get_nz_map_contexts_func_(levels_, scan, eob, (TX_SIZE)tx_size, |
| tx_class, coeff_contexts_); |
| } |
| aom_usec_timer_mark(&timer); |
| |
| const int elapsed_time_ref = |
| static_cast<int>(aom_usec_timer_elapsed(&timer_ref)); |
| const int elapsed_time = static_cast<int>(aom_usec_timer_elapsed(&timer)); |
| |
| printf("get_nz_map_contexts_%2dx%2d: %7.1f ms ref %7.1f ms gain %4.2f\n", |
| real_width, real_height, elapsed_time / 1000.0, |
| elapsed_time_ref / 1000.0, |
| (elapsed_time_ref * 1.0) / (elapsed_time * 1.0)); |
| } |
| } |
| |
| private: |
| void InitDataWithEob(const int16_t *const scan, const int bwl, |
| const int eob) { |
| memset(levels_buf_, 0, sizeof(levels_buf_)); |
| memset(coeff_contexts_, 0, sizeof(*coeff_contexts_) * MAX_TX_SQUARE); |
| |
| for (int c = 0; c < eob; ++c) { |
| levels_[get_padded_idx(scan[c], bwl)] = |
| static_cast<uint8_t>(clamp(rnd_.Rand8(), 0, INT8_MAX)); |
| coeff_contexts_[scan[c]] = static_cast<int8_t>(rnd_.Rand16() >> 1); |
| } |
| |
| memcpy(coeff_contexts_ref_, coeff_contexts_, |
| sizeof(*coeff_contexts_) * MAX_TX_SQUARE); |
| } |
| |
| bool Compare(const int16_t *const scan, const int eob) const { |
| bool result = false; |
| if (memcmp(coeff_contexts_, coeff_contexts_ref_, |
| sizeof(*coeff_contexts_ref_) * MAX_TX_SQUARE)) { |
| for (int i = 0; i < eob; i++) { |
| const int pos = scan[i]; |
| if (coeff_contexts_ref_[pos] != coeff_contexts_[pos]) { |
| printf("coeff_contexts_[%d] diff:%6d (ref),%6d (opt)\n", pos, |
| coeff_contexts_ref_[pos], coeff_contexts_[pos]); |
| result = true; |
| break; |
| } |
| } |
| } |
| return result; |
| } |
| |
| GetNzMapContextsFunc get_nz_map_contexts_func_; |
| ACMRandom rnd_; |
| uint8_t levels_buf_[TX_PAD_2D]; |
| uint8_t *levels_; |
| int8_t *coeff_contexts_ref_; |
| int8_t *coeff_contexts_; |
| }; |
| GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(EncodeTxbTest); |
| |
| TEST_P(EncodeTxbTest, GetNzMapContexts) { GetNzMapContextsRun(); } |
| |
| TEST_P(EncodeTxbTest, DISABLED_SpeedTestGetNzMapContexts) { |
| SpeedTestGetNzMapContextsRun(); |
| } |
| |
| #if HAVE_SSE2 |
| INSTANTIATE_TEST_SUITE_P(SSE2, EncodeTxbTest, |
| ::testing::Values(av1_get_nz_map_contexts_sse2)); |
| #endif |
| |
| #if HAVE_NEON |
| INSTANTIATE_TEST_SUITE_P(NEON, EncodeTxbTest, |
| ::testing::Values(av1_get_nz_map_contexts_neon)); |
| #endif |
| |
| typedef void (*av1_txb_init_levels_func)(const tran_low_t *const coeff, |
| const int width, const int height, |
| uint8_t *const levels); |
| |
| typedef std::tuple<av1_txb_init_levels_func, int> TxbInitLevelParam; |
| |
| class EncodeTxbInitLevelTest |
| : public ::testing::TestWithParam<TxbInitLevelParam> { |
| public: |
| virtual ~EncodeTxbInitLevelTest() {} |
| virtual void TearDown() { libaom_test::ClearSystemState(); } |
| void RunTest(av1_txb_init_levels_func test_func, int tx_size, int is_speed); |
| }; |
| GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(EncodeTxbInitLevelTest); |
| |
| void EncodeTxbInitLevelTest::RunTest(av1_txb_init_levels_func test_func, |
| int tx_size, int is_speed) { |
| const int width = get_txb_wide((TX_SIZE)tx_size); |
| const int height = get_txb_high((TX_SIZE)tx_size); |
| tran_low_t coeff[MAX_TX_SQUARE]; |
| |
| uint8_t levels_buf[2][TX_PAD_2D]; |
| uint8_t *const levels0 = set_levels(levels_buf[0], width); |
| uint8_t *const levels1 = set_levels(levels_buf[1], width); |
| |
| ACMRandom rnd(ACMRandom::DeterministicSeed()); |
| for (int i = 0; i < width * height; i++) { |
| coeff[i] = rnd.Rand15Signed() + rnd.Rand15Signed(); |
| } |
| for (int i = 0; i < TX_PAD_2D; i++) { |
| levels_buf[0][i] = rnd.Rand8(); |
| levels_buf[1][i] = rnd.Rand8(); |
| } |
| const int run_times = is_speed ? (width * height) * 10000 : 1; |
| aom_usec_timer timer; |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < run_times; ++i) { |
| av1_txb_init_levels_c(coeff, width, height, levels0); |
| } |
| const double t1 = get_time_mark(&timer); |
| aom_usec_timer_start(&timer); |
| for (int i = 0; i < run_times; ++i) { |
| test_func(coeff, width, height, levels1); |
| } |
| const double t2 = get_time_mark(&timer); |
| if (is_speed) { |
| printf("init %3dx%-3d:%7.2f/%7.2fns", width, height, t1, t2); |
| printf("(%3.2f)\n", t1 / t2); |
| } |
| const int stride = width + TX_PAD_HOR; |
| #if CONFIG_FORWARDSKIP |
| for (int r = TX_PAD_TOP; r < height + TX_PAD_VER; ++r) { |
| #else |
| for (int r = 0; r < height + TX_PAD_VER; ++r) { |
| #endif // CONFIG_FORWARDSKIP |
| for (int c = 0; c < stride; ++c) { |
| ASSERT_EQ(levels_buf[0][c + r * stride], levels_buf[1][c + r * stride]) |
| << "[" << r << "," << c << "] " << run_times << width << "x" |
| << height; |
| } |
| } |
| } |
| |
| TEST_P(EncodeTxbInitLevelTest, match) { |
| RunTest(GET_PARAM(0), GET_PARAM(1), 0); |
| } |
| |
| TEST_P(EncodeTxbInitLevelTest, DISABLED_Speed) { |
| RunTest(GET_PARAM(0), GET_PARAM(1), 1); |
| } |
| |
| #if HAVE_SSE4_1 |
| INSTANTIATE_TEST_SUITE_P( |
| SSE4_1, EncodeTxbInitLevelTest, |
| ::testing::Combine(::testing::Values(&av1_txb_init_levels_sse4_1), |
| ::testing::Range(0, static_cast<int>(TX_SIZES_ALL), 1))); |
| #endif |
| #if HAVE_AVX2 |
| INSTANTIATE_TEST_SUITE_P( |
| AVX2, EncodeTxbInitLevelTest, |
| ::testing::Combine(::testing::Values(&av1_txb_init_levels_avx2), |
| ::testing::Range(0, static_cast<int>(TX_SIZES_ALL), 1))); |
| #endif |
| #if HAVE_NEON |
| INSTANTIATE_TEST_SUITE_P( |
| NEON, EncodeTxbInitLevelTest, |
| ::testing::Combine(::testing::Values(&av1_txb_init_levels_neon), |
| ::testing::Range(0, static_cast<int>(TX_SIZES_ALL), 1))); |
| #endif |
| } // namespace |