chromium/third_party/libaom/source/libaom/test/encodemb_test.cc

/*
 * Copyright (c) 2021, 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 <stdint.h>
#include <vector>

#include "gtest/gtest.h"

#include "av1/encoder/block.h"
#include "av1/encoder/encodemb.h"
#include "av1/common/scan.h"

namespace {

// Reorders 'qcoeff_lexico', which is in lexicographic order (row by row), into
// scan order (zigzag) in 'qcoeff_scan'.
void ToScanOrder(TX_SIZE tx_size, TX_TYPE tx_type, tran_low_t *qcoeff_lexico,
                 tran_low_t *qcoeff_scan) { … }

// Reorders 'qcoeff_scan', which is in scan order (zigzag), into lexicographic
// order (row by row) in 'qcoeff_lexico'.
void ToLexicoOrder(TX_SIZE tx_size, TX_TYPE tx_type, tran_low_t *qcoeff_scan,
                   tran_low_t *qcoeff_lexico) { … }

// Runs coefficient dropout on 'qcoeff_scan'.
void Dropout(TX_SIZE tx_size, TX_TYPE tx_type, int dropout_num_before,
             int dropout_num_after, tran_low_t *qcoeff_scan) { … }

void ExpectArrayEq(tran_low_t *actual, std::vector<tran_low_t> expected) { … }

static constexpr TX_TYPE kTxType = …;

TEST(DropoutTest, KeepsLargeCoeffs) { … }

TEST(DropoutTest, RemovesSmallIsolatedCoeffs) { … }

TEST(DropoutTest, KeepsSmallCoeffsAmongLargeOnes) { … }

TEST(DropoutTest, KeepsSmallCoeffsCloseToStartOrEnd) { … }

TEST(DropoutTest, RemovesSmallClusterOfCoeffs) { … }

TEST(DropoutTest, KeepsLargeClusterOfCoeffs) { … }

TEST(DropoutTest, NumBeforeLargerThanNumAfter) { … }

// More complex test combining other test cases.
TEST(DropoutTest, ComplexTest) { … }

}  // namespace