blob: 243af6208d8cd0f5af481f0fafe3fa9cf98a7a65 [file] [log] [blame]
/*
* 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/.
*/
#ifndef AOM_TEST_TRANSFORM_TEST_BASE_H_
#define AOM_TEST_TRANSFORM_TEST_BASE_H_
#include "config/aom_config.h"
#include "aom_mem/aom_mem.h"
#include "aom/aom_codec.h"
#include "aom_dsp/txfm_common.h"
namespace libaom_test {
// Note:
// Same constant are defined in av1/common/av1_entropy.h and
// av1/common/entropy.h. Goal is to make this base class
// to use for future codec transform testing. But including
// either of them would lead to compiling error when we do
// unit test for another codec. Suggest to move the definition
// to a aom header file.
const int kDctMaxValue = 16384;
template <typename OutputType>
using FhtFunc = void (*)(const int16_t *in, OutputType *out, int stride,
TxfmParam *txfm_param);
template <typename OutputType>
using IhtFunc = void (*)(const tran_low_t *in, uint16_t *out, int stride,
const TxfmParam *txfm_param);
template <typename OutType>
class TransformTestBase {
public:
virtual ~TransformTestBase() {}
protected:
virtual void RunFwdTxfm(const int16_t *in, OutType *out, int stride) = 0;
virtual void RunInvTxfm(const OutType *out, uint16_t *dst, int stride) = 0;
void RunAccuracyCheck(uint32_t ref_max_error, double ref_avg_error) {
ACMRandom rnd(ACMRandom::DeterministicSeed());
uint32_t max_error = 0;
int64_t total_error = 0;
const int count_test_block = 10000;
int16_t *test_input_block = reinterpret_cast<int16_t *>(
aom_memalign(16, sizeof(int16_t) * num_coeffs_));
OutType *test_temp_block = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(test_temp_block[0]) * num_coeffs_));
uint16_t *dst16 = reinterpret_cast<uint16_t *>(
aom_memalign(16, sizeof(uint16_t) * num_coeffs_));
uint16_t *src16 = reinterpret_cast<uint16_t *>(
aom_memalign(16, sizeof(uint16_t) * num_coeffs_));
for (int i = 0; i < count_test_block; ++i) {
// Initialize a test block with input range [-255, 255].
for (int j = 0; j < num_coeffs_; ++j) {
src16[j] = rnd.Rand16() & mask_;
dst16[j] = rnd.Rand16() & mask_;
test_input_block[j] = src16[j] - dst16[j];
}
ASM_REGISTER_STATE_CHECK(
RunFwdTxfm(test_input_block, test_temp_block, pitch_));
ASM_REGISTER_STATE_CHECK(RunInvTxfm(test_temp_block, dst16, pitch_));
for (int j = 0; j < num_coeffs_; ++j) {
const int diff = dst16[j] - src16[j];
const uint32_t error = diff * diff;
if (max_error < error) max_error = error;
total_error += error;
}
}
double avg_error = total_error * 1. / count_test_block / num_coeffs_;
EXPECT_GE(ref_max_error, max_error)
<< "Error: FHT/IHT has an individual round trip error > "
<< ref_max_error;
EXPECT_GE(ref_avg_error, avg_error)
<< "Error: FHT/IHT has average round trip error > " << ref_avg_error
<< " per block";
aom_free(test_input_block);
aom_free(test_temp_block);
aom_free(dst16);
aom_free(src16);
}
void RunCoeffCheck() {
ACMRandom rnd(ACMRandom::DeterministicSeed());
const int count_test_block = 5000;
// Use a stride value which is not the width of any transform, to catch
// cases where the transforms use the stride incorrectly.
int stride = 96;
int16_t *input_block = reinterpret_cast<int16_t *>(
aom_memalign(16, sizeof(int16_t) * stride * height_));
OutType *output_ref_block = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(output_ref_block[0]) * num_coeffs_));
OutType *output_block = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(output_block[0]) * num_coeffs_));
for (int i = 0; i < count_test_block; ++i) {
int j, k;
for (j = 0; j < height_; ++j) {
for (k = 0; k < pitch_; ++k) {
int in_idx = j * stride + k;
int out_idx = j * pitch_ + k;
input_block[in_idx] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_);
if (bit_depth_ == AOM_BITS_8) {
output_block[out_idx] = output_ref_block[out_idx] = rnd.Rand8();
} else {
output_block[out_idx] = output_ref_block[out_idx] =
rnd.Rand16() & mask_;
}
}
}
fwd_txfm_ref(input_block, output_ref_block, stride, &txfm_param_);
ASM_REGISTER_STATE_CHECK(RunFwdTxfm(input_block, output_block, stride));
// The minimum quant value is 4.
for (j = 0; j < height_; ++j) {
for (k = 0; k < pitch_; ++k) {
int out_idx = j * pitch_ + k;
ASSERT_EQ(output_block[out_idx], output_ref_block[out_idx])
<< "Error: not bit-exact result at index: " << out_idx
<< " at test block: " << i;
}
}
}
aom_free(input_block);
aom_free(output_ref_block);
aom_free(output_block);
}
void RunMemCheck() {
ACMRandom rnd(ACMRandom::DeterministicSeed());
const int count_test_block = 5000;
int16_t *input_extreme_block = reinterpret_cast<int16_t *>(
aom_memalign(16, sizeof(int16_t) * num_coeffs_));
OutType *output_ref_block = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(output_ref_block[0]) * num_coeffs_));
OutType *output_block = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(output_block[0]) * num_coeffs_));
for (int i = 0; i < count_test_block; ++i) {
// Initialize a test block with input range [-mask_, mask_].
for (int j = 0; j < num_coeffs_; ++j) {
input_extreme_block[j] = rnd.Rand8() % 2 ? mask_ : -mask_;
}
if (i == 0) {
for (int j = 0; j < num_coeffs_; ++j) input_extreme_block[j] = mask_;
} else if (i == 1) {
for (int j = 0; j < num_coeffs_; ++j) input_extreme_block[j] = -mask_;
}
fwd_txfm_ref(input_extreme_block, output_ref_block, pitch_, &txfm_param_);
ASM_REGISTER_STATE_CHECK(
RunFwdTxfm(input_extreme_block, output_block, pitch_));
int row_length = FindRowLength();
// The minimum quant value is 4.
for (int j = 0; j < num_coeffs_; ++j) {
ASSERT_EQ(output_block[j], output_ref_block[j])
<< "Not bit-exact at test index: " << i << ", "
<< "j = " << j << std::endl;
EXPECT_GE(row_length * kDctMaxValue << (bit_depth_ - 8),
abs(output_block[j]))
<< "Error: NxN FDCT has coefficient larger than N*DCT_MAX_VALUE";
}
}
aom_free(input_extreme_block);
aom_free(output_ref_block);
aom_free(output_block);
}
void RunInvAccuracyCheck(int limit) {
ACMRandom rnd(ACMRandom::DeterministicSeed());
const int count_test_block = 1000;
int16_t *in = reinterpret_cast<int16_t *>(
aom_memalign(16, sizeof(int16_t) * num_coeffs_));
OutType *coeff = reinterpret_cast<OutType *>(
aom_memalign(16, sizeof(coeff[0]) * num_coeffs_));
uint16_t *dst16 = reinterpret_cast<uint16_t *>(
aom_memalign(16, sizeof(uint16_t) * num_coeffs_));
uint16_t *src16 = reinterpret_cast<uint16_t *>(
aom_memalign(16, sizeof(uint16_t) * num_coeffs_));
for (int i = 0; i < count_test_block; ++i) {
// Initialize a test block with input range [-mask_, mask_].
for (int j = 0; j < num_coeffs_; ++j) {
src16[j] = rnd.Rand16() & mask_;
dst16[j] = rnd.Rand16() & mask_;
in[j] = src16[j] - dst16[j];
}
fwd_txfm_ref(in, coeff, pitch_, &txfm_param_);
ASM_REGISTER_STATE_CHECK(RunInvTxfm(coeff, dst16, pitch_));
for (int j = 0; j < num_coeffs_; ++j) {
const int diff = dst16[j] - src16[j];
const uint32_t error = diff * diff;
ASSERT_GE(static_cast<uint32_t>(limit), error)
<< "Error: 4x4 IDCT has error " << error << " at index " << j;
}
}
aom_free(in);
aom_free(coeff);
aom_free(src16);
aom_free(dst16);
}
int pitch_;
int height_;
FhtFunc<OutType> fwd_txfm_ref;
IhtFunc<OutType> inv_txfm_ref;
aom_bit_depth_t bit_depth_;
int mask_;
int num_coeffs_;
TxfmParam txfm_param_;
private:
// Assume transform size is 4x4, 8x8, 16x16,...
int FindRowLength() const {
int row = 4;
if (16 == num_coeffs_) {
row = 4;
} else if (64 == num_coeffs_) {
row = 8;
} else if (256 == num_coeffs_) {
row = 16;
} else if (1024 == num_coeffs_) {
row = 32;
}
return row;
}
};
} // namespace libaom_test
#endif // AOM_TEST_TRANSFORM_TEST_BASE_H_