| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| * Media Patent License 1.0 was not distributed with this source code in the |
| * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
| */ |
| |
| #include "test/av1_txfm_test.h" |
| #include "av1/common/av1_fwd_txfm1d.h" |
| #include "av1/common/av1_inv_txfm1d.h" |
| |
| using libaom_test::ACMRandom; |
| using libaom_test::input_base; |
| |
| namespace { |
| const int txfm_type_num = 2; |
| const int txfm_size_num = 5; |
| const int txfm_size_ls[5] = { 4, 8, 16, 32, 64 }; |
| |
| const TxfmFunc fwd_txfm_func_ls[2][5] = { |
| #if CONFIG_TX64X64 |
| { av1_fdct4_new, av1_fdct8_new, av1_fdct16_new, av1_fdct32_new, |
| av1_fdct64_new }, |
| #else |
| { av1_fdct4_new, av1_fdct8_new, av1_fdct16_new, av1_fdct32_new, NULL }, |
| #endif |
| { av1_fadst4_new, av1_fadst8_new, av1_fadst16_new, av1_fadst32_new, NULL } |
| }; |
| |
| const TxfmFunc inv_txfm_func_ls[2][5] = { |
| #if CONFIG_TX64X64 |
| { av1_idct4_new, av1_idct8_new, av1_idct16_new, av1_idct32_new, |
| av1_idct64_new }, |
| #else |
| { av1_idct4_new, av1_idct8_new, av1_idct16_new, av1_idct32_new, NULL }, |
| #endif |
| { av1_iadst4_new, av1_iadst8_new, av1_iadst16_new, av1_iadst32_new, NULL } |
| }; |
| |
| // the maximum stage number of fwd/inv 1d dct/adst txfm is 12 |
| const int8_t cos_bit[12] = { 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14 }; |
| const int8_t range_bit[12] = { 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32 }; |
| |
| TEST(av1_inv_txfm1d, round_trip) { |
| ACMRandom rnd(ACMRandom::DeterministicSeed()); |
| for (int si = 0; si < txfm_size_num; ++si) { |
| int txfm_size = txfm_size_ls[si]; |
| int32_t *input = new int32_t[txfm_size]; |
| int32_t *output = new int32_t[txfm_size]; |
| int32_t *round_trip_output = new int32_t[txfm_size]; |
| |
| for (int ti = 0; ti < txfm_type_num; ++ti) { |
| TxfmFunc fwd_txfm_func = fwd_txfm_func_ls[ti][si]; |
| TxfmFunc inv_txfm_func = inv_txfm_func_ls[ti][si]; |
| int max_error = 2; |
| |
| if (fwd_txfm_func != NULL) { |
| const int count_test_block = 5000; |
| for (int ci = 0; ci < count_test_block; ++ci) { |
| for (int ni = 0; ni < txfm_size; ++ni) { |
| input[ni] = rnd.Rand16() % input_base - rnd.Rand16() % input_base; |
| } |
| |
| fwd_txfm_func(input, output, cos_bit, range_bit); |
| inv_txfm_func(output, round_trip_output, cos_bit, range_bit); |
| |
| for (int ni = 0; ni < txfm_size; ++ni) { |
| int node_err = |
| abs(input[ni] - round_shift(round_trip_output[ni], |
| get_max_bit(txfm_size) - 1)); |
| EXPECT_LE(node_err, max_error); |
| } |
| } |
| } |
| } |
| delete[] input; |
| delete[] output; |
| delete[] round_trip_output; |
| } |
| } |
| |
| } // namespace |