diff options
Diffstat (limited to 'third_party/aom/test/mv_cost_test.cc')
-rw-r--r-- | third_party/aom/test/mv_cost_test.cc | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/third_party/aom/test/mv_cost_test.cc b/third_party/aom/test/mv_cost_test.cc new file mode 100644 index 0000000000..73d56665bf --- /dev/null +++ b/third_party/aom/test/mv_cost_test.cc @@ -0,0 +1,125 @@ +/* + * Copyright (c) 2022, 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 "av1/encoder/cost.h" +#include "av1/encoder/encodemv.h" +#include "third_party/googletest/src/googletest/include/gtest/gtest.h" + +namespace { + +void ReferenceBuildNmvComponentCostTable(int *mvcost, + const nmv_component *const mvcomp, + MvSubpelPrecision precision) { + int i, v; + int sign_cost[2], class_cost[MV_CLASSES], class0_cost[CLASS0_SIZE]; + int bits_cost[MV_OFFSET_BITS][2]; + int class0_fp_cost[CLASS0_SIZE][MV_FP_SIZE], fp_cost[MV_FP_SIZE]; + int class0_hp_cost[2], hp_cost[2]; + av1_cost_tokens_from_cdf(sign_cost, mvcomp->sign_cdf, nullptr); + av1_cost_tokens_from_cdf(class_cost, mvcomp->classes_cdf, nullptr); + av1_cost_tokens_from_cdf(class0_cost, mvcomp->class0_cdf, nullptr); + for (i = 0; i < MV_OFFSET_BITS; ++i) { + av1_cost_tokens_from_cdf(bits_cost[i], mvcomp->bits_cdf[i], nullptr); + } + for (i = 0; i < CLASS0_SIZE; ++i) + av1_cost_tokens_from_cdf(class0_fp_cost[i], mvcomp->class0_fp_cdf[i], + nullptr); + av1_cost_tokens_from_cdf(fp_cost, mvcomp->fp_cdf, nullptr); + if (precision > MV_SUBPEL_LOW_PRECISION) { + av1_cost_tokens_from_cdf(class0_hp_cost, mvcomp->class0_hp_cdf, nullptr); + av1_cost_tokens_from_cdf(hp_cost, mvcomp->hp_cdf, nullptr); + } + mvcost[0] = 0; + for (v = 1; v <= MV_MAX; ++v) { + int z, c, o, d, e, f, cost = 0; + z = v - 1; + c = av1_get_mv_class(z, &o); + cost += class_cost[c]; + d = (o >> 3); /* int mv data */ + f = (o >> 1) & 3; /* fractional pel mv data */ + e = (o & 1); /* high precision mv data */ + if (c == MV_CLASS_0) { + cost += class0_cost[d]; + } else { + const int b = c + CLASS0_BITS - 1; /* number of bits */ + for (i = 0; i < b; ++i) cost += bits_cost[i][((d >> i) & 1)]; + } + if (precision > MV_SUBPEL_NONE) { + if (c == MV_CLASS_0) { + cost += class0_fp_cost[d][f]; + } else { + cost += fp_cost[f]; + } + if (precision > MV_SUBPEL_LOW_PRECISION) { + if (c == MV_CLASS_0) { + cost += class0_hp_cost[e]; + } else { + cost += hp_cost[e]; + } + } + } + mvcost[v] = cost + sign_cost[0]; + mvcost[-v] = cost + sign_cost[1]; + } +} + +// Test using the default context, except for sign +static const nmv_component kTestComponentContext = { + { AOM_CDF11(28672, 30976, 31858, 32320, 32551, 32656, 32740, 32757, 32762, + 32767) }, // class_cdf // fp + { { AOM_CDF4(16384, 24576, 26624) }, + { AOM_CDF4(12288, 21248, 24128) } }, // class0_fp_cdf + { AOM_CDF4(8192, 17408, 21248) }, // fp_cdf + { AOM_CDF2(70 * 128) }, // sign_cdf + { AOM_CDF2(160 * 128) }, // class0_hp_cdf + { AOM_CDF2(128 * 128) }, // hp_cdf + { AOM_CDF2(216 * 128) }, // class0_cdf + { { AOM_CDF2(128 * 136) }, + { AOM_CDF2(128 * 140) }, + { AOM_CDF2(128 * 148) }, + { AOM_CDF2(128 * 160) }, + { AOM_CDF2(128 * 176) }, + { AOM_CDF2(128 * 192) }, + { AOM_CDF2(128 * 224) }, + { AOM_CDF2(128 * 234) }, + { AOM_CDF2(128 * 234) }, + { AOM_CDF2(128 * 240) } }, // bits_cdf +}; + +void TestMvComponentCostTable(MvSubpelPrecision precision) { + std::unique_ptr<int[]> mvcost_ref_buf(new int[MV_VALS]); + std::unique_ptr<int[]> mvcost_buf(new int[MV_VALS]); + int *mvcost_ref = mvcost_ref_buf.get() + MV_MAX; + int *mvcost = mvcost_buf.get() + MV_MAX; + + ReferenceBuildNmvComponentCostTable(mvcost_ref, &kTestComponentContext, + precision); + av1_build_nmv_component_cost_table(mvcost, &kTestComponentContext, precision); + + for (int v = 0; v <= MV_MAX; ++v) { + ASSERT_EQ(mvcost_ref[v], mvcost[v]) << "v = " << v; + ASSERT_EQ(mvcost_ref[-v], mvcost[-v]) << "v = " << v; + } +} + +TEST(MvCostTest, BuildMvComponentCostTableTest1) { + TestMvComponentCostTable(MV_SUBPEL_NONE); +} + +TEST(MvCostTest, BuildMvComponentCostTableTest2) { + TestMvComponentCostTable(MV_SUBPEL_LOW_PRECISION); +} + +TEST(MvCostTest, BuildMvComponentCostTableTest3) { + TestMvComponentCostTable(MV_SUBPEL_HIGH_PRECISION); +} + +} // namespace
\ No newline at end of file |