/* * 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. */ #ifndef AOM_AV1_ENCODER_TXB_RDOPT_H_ #define AOM_AV1_ENCODER_TXB_RDOPT_H_ #include "av1/common/blockd.h" #include "av1/common/txb_common.h" #include "av1/encoder/encoder.h" #ifdef __cplusplus extern "C" { #endif /*!\brief Adjust the magnitude of quantized coefficients to achieve better * rate-distortion (RD) trade-off. * * \ingroup coefficient_coding * * This function goes through each coefficient and greedily choose to lower * the coefficient magnitude by 1 or not based on the RD score. * * The coefficients are processing in reversed scan order. * * Note that, the end of block position (eob) may change if the original last * coefficient is lowered to zero. * * \param[in] cpi Top-level encoder structure * \param[in] x Pointer to structure holding the data for the current encoding macroblock * \param[in] plane The index of the current plane * \param[in] block The index of the current transform block in the * \param[in] tx_size The transform size * \param[in] tx_type The transform type * \param[in] txb_ctx Context info for entropy coding transform block * skip flag (tx_skip) and the sign of DC coefficient (dc_sign). * \param[out] rate_cost The entropy cost of coding the transform block * after adjustment of coefficients. * \param[in] sharpness When sharpness > 0, the function will be less * aggressive towards lowering the magnitude of coefficients. * In this way, the transform block will contain more high-frequency * coefficients and therefore will preserve the sharpness of the reconstructed * block. */ int av1_optimize_txb(const struct AV1_COMP *cpi, MACROBLOCK *x, int plane, int block, TX_SIZE tx_size, TX_TYPE tx_type, const TXB_CTX *const txb_ctx, int *rate_cost, int sharpness); /*!\brief Compute the entropy cost of coding coefficients in a transform block. * * \ingroup coefficient_coding * * \param[in] x Pointer to structure holding the data for the current encoding macroblock. * \param[in] plane The index of the current plane. * \param[in] block The index of the current transform block in the * macroblock. It's defined by number of 4x4 units that have been coded before * the currernt transform block. * \param[in] tx_size The transform size. * \param[in] tx_type The transform type. * \param[in] txb_ctx Context info for entropy coding transform block * skip flag (tx_skip) and the sign of DC coefficient (dc_sign). * \param[in] reduced_tx_set_used Whether the transform type is chosen from * a reduced set. */ int av1_cost_coeffs_txb(const MACROBLOCK *x, const int plane, const int block, const TX_SIZE tx_size, const TX_TYPE tx_type, const TXB_CTX *const txb_ctx, int reduced_tx_set_used); /*!\brief Estimate the entropy cost of coding a transform block using Laplacian * distribution. * * \ingroup coefficient_coding * * This function compute the entropy costs of the end of block position (eob) * and the transform type (tx_type) precisely. * * Then using \ref av1_cost_coeffs_txb_estimate to estimate the entropy costs * of coefficients in the transform block. * * In the end, the function returns the sum of entropy costs of end of block * position (eob), transform type (tx_type) and coefficients. * * Compared to \ref av1_cost_coeffs_txb, this function is much faster but less * accurate. * * \param[in] x Pointer to structure holding the data for the current encoding macroblock * \param[in] plane The index of the current plane * \param[in] block The index of the current transform block in the * macroblock. It's defined by number of 4x4 units that have been coded before * the currernt transform block * \param[in] tx_size The transform size * \param[in] tx_type The transform type * \param[in] txb_ctx Context info for entropy coding transform block * skip flag (tx_skip) and the sign of DC coefficient (dc_sign). * \param[in] reduced_tx_set_used Whether the transform type is chosen from * a reduced set. * \param[in] adjust_eob Whether to adjust the end of block position (eob) * or not. * \return int Estimated entropy cost of coding the transform block. */ int av1_cost_coeffs_txb_laplacian(const MACROBLOCK *x, const int plane, const int block, const TX_SIZE tx_size, const TX_TYPE tx_type, const TXB_CTX *const txb_ctx, const int reduced_tx_set_used, const int adjust_eob); /*!\brief Estimate the entropy cost of transform coefficients using Laplacian * distribution. * * \ingroup coefficient_coding * * This function assumes each transform coefficient is of its own Laplacian * distribution and the coefficient is the only observation of the Laplacian * distribution. * * Based on that, each coefficient's coding cost can be estimated by computing * the entropy of the corresponding Laplacian distribution. * * This function then return the sum of the estimated entropy cost for all * coefficients in the transform block. * * Note that the entropy cost of end of block (eob) and transform type (tx_type) * are not included. * * \param[in] x Pointer to structure holding the data for the current encoding macroblock * \param[in] plane The index of the current plane * \param[in] block The index of the current transform block in the * macroblock. It's defined by number of 4x4 units that have been coded before * the currernt transform block * \param[in] tx_size The transform size * \param[in] tx_type The transform type * \return int Estimated entropy cost of coefficients in the * transform block. */ int av1_cost_coeffs_txb_estimate(const MACROBLOCK *x, const int plane, const int block, const TX_SIZE tx_size, const TX_TYPE tx_type); #ifdef __cplusplus } #endif #endif // AOM_AV1_ENCODER_TXB_RDOPT_H_