125 lines
No EOL
4.5 KiB
C++
125 lines
No EOL
4.5 KiB
C++
/*
|
|
* 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
|