/* * Copyright (c) 2016, 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 "config/aom_config.h" #include "config/av1_rtcd.h" #include "config/aom_dsp_rtcd.h" #include "aom_dsp/bitwriter.h" #include "aom_dsp/quantize.h" #include "aom_mem/aom_mem.h" #include "aom_ports/mem.h" #if CONFIG_BITSTREAM_DEBUG || CONFIG_MISMATCH_DEBUG #include "aom_util/debug_util.h" #endif // CONFIG_BITSTREAM_DEBUG || CONFIG_MISMATCH_DEBUG #include "av1/common/cfl.h" #include "av1/common/idct.h" #include "av1/common/reconinter.h" #include "av1/common/reconintra.h" #include "av1/common/scan.h" #include "av1/encoder/av1_quantize.h" #include "av1/encoder/encodemb.h" #include "av1/encoder/hybrid_fwd_txfm.h" #include "av1/encoder/txb_rdopt.h" #include "av1/encoder/rd.h" #include "av1/encoder/rdopt.h" void av1_subtract_block(BitDepthInfo bd_info, int rows, int cols, int16_t *diff, ptrdiff_t diff_stride, const uint8_t *src8, ptrdiff_t src_stride, const uint8_t *pred8, ptrdiff_t pred_stride) { … } void av1_subtract_txb(MACROBLOCK *x, int plane, BLOCK_SIZE plane_bsize, int blk_col, int blk_row, TX_SIZE tx_size) { … } void av1_subtract_plane(MACROBLOCK *x, BLOCK_SIZE plane_bsize, int plane) { … } int av1_optimize_b(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) { … } // Hyper-parameters for dropout optimization, based on following logics. // TODO(yjshen): These settings are tuned by experiments. They may still be // optimized for better performance. // (1) Coefficients which are large enough will ALWAYS be kept. static const tran_low_t DROPOUT_COEFF_MAX = …; // Max dropout-able coefficient. // (2) Continuous coefficients will ALWAYS be kept. Here rigorous continuity is // NOT required. For example, `5 0 0 0 7` is treated as two continuous // coefficients if three zeros do not fulfill the dropout condition. static const int DROPOUT_CONTINUITY_MAX = …; // Max dropout-able continuous coeff. // (3) Dropout operation is NOT applicable to blocks with large or small // quantization index. static const int DROPOUT_Q_MAX = …; static const int DROPOUT_Q_MIN = …; // (4) Recall that dropout optimization will forcibly set some quantized // coefficients to zero. The key logic on determining whether a coefficient // should be dropped is to check the number of continuous zeros before AND // after this coefficient. The exact number of zeros for judgement depends // on block size and quantization index. More concretely, block size // determines the base number of zeros, while quantization index determines // the multiplier. Intuitively, larger block requires more zeros and larger // quantization index also requires more zeros (more information is lost // when using larger quantization index). static const int DROPOUT_BEFORE_BASE_MAX = …; // Max base number for leading zeros. static const int DROPOUT_BEFORE_BASE_MIN = …; // Min base number for leading zeros. static const int DROPOUT_AFTER_BASE_MAX = …; // Max base number for trailing zeros. static const int DROPOUT_AFTER_BASE_MIN = …; // Min base number for trailing zeros. static const int DROPOUT_MULTIPLIER_MAX = …; // Max multiplier on number of zeros. static const int DROPOUT_MULTIPLIER_MIN = …; // Min multiplier on number of zeros. static const int DROPOUT_MULTIPLIER_Q_BASE = …; // Base Q to compute multiplier. void av1_dropout_qcoeff(MACROBLOCK *mb, int plane, int block, TX_SIZE tx_size, TX_TYPE tx_type, int qindex) { … } void av1_dropout_qcoeff_num(MACROBLOCK *mb, int plane, int block, TX_SIZE tx_size, TX_TYPE tx_type, int dropout_num_before, int dropout_num_after) { … } // Settings for optimization type. NOTE: To set optimization type for all intra // frames, both `KEY_BLOCK_OPT_TYPE` and `INTRA_BLOCK_OPT_TYPE` should be set. // TODO(yjshen): These settings are hard-coded and look okay for now. They // should be made configurable later. // Blocks of key frames ONLY. static const OPT_TYPE KEY_BLOCK_OPT_TYPE = …; // Blocks of intra frames (key frames EXCLUSIVE). static const OPT_TYPE INTRA_BLOCK_OPT_TYPE = …; // Blocks of inter frames. (NOTE: Dropout optimization is DISABLED by default // if trellis optimization is on for inter frames.) static const OPT_TYPE INTER_BLOCK_OPT_TYPE = …; enum { … } UENUM1BYTE(…) …; #if CONFIG_AV1_HIGHBITDEPTH static AV1_QUANT_FACADE quant_func_list[AV1_XFORM_QUANT_TYPES][QUANT_FUNC_TYPES] = { { av1_quantize_fp_facade, av1_highbd_quantize_fp_facade }, { av1_quantize_b_facade, av1_highbd_quantize_b_facade }, { av1_quantize_dc_facade, av1_highbd_quantize_dc_facade }, { NULL, NULL } }; #else static AV1_QUANT_FACADE quant_func_list[AV1_XFORM_QUANT_TYPES] = …; #endif // Computes the transform for DC only blocks void av1_xform_dc_only(MACROBLOCK *x, int plane, int block, TxfmParam *txfm_param, int64_t per_px_mean) { … } void av1_xform_quant(MACROBLOCK *x, int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TxfmParam *txfm_param, const QUANT_PARAM *qparam) { … } void av1_xform(MACROBLOCK *x, int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TxfmParam *txfm_param) { … } void av1_quant(MACROBLOCK *x, int plane, int block, TxfmParam *txfm_param, const QUANT_PARAM *qparam) { … } void av1_setup_xform(const AV1_COMMON *cm, MACROBLOCK *x, TX_SIZE tx_size, TX_TYPE tx_type, TxfmParam *txfm_param) { … } void av1_setup_quant(TX_SIZE tx_size, int use_optimize_b, int xform_quant_idx, int use_quant_b_adapt, QUANT_PARAM *qparam) { … } void av1_setup_qmatrix(const CommonQuantParams *quant_params, const MACROBLOCKD *xd, int plane, TX_SIZE tx_size, TX_TYPE tx_type, QUANT_PARAM *qparam) { … } static void encode_block(int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg, RUN_TYPE dry_run) { … } static void encode_block_inter(int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg, RUN_TYPE dry_run) { … } void av1_foreach_transformed_block_in_plane( const MACROBLOCKD *const xd, BLOCK_SIZE plane_bsize, int plane, foreach_transformed_block_visitor visit, void *arg) { … } encode_block_pass1_args; static void encode_block_pass1(int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg) { … } void av1_encode_sby_pass1(AV1_COMP *cpi, MACROBLOCK *x, BLOCK_SIZE bsize) { … } void av1_encode_sb(const struct AV1_COMP *cpi, MACROBLOCK *x, BLOCK_SIZE bsize, RUN_TYPE dry_run) { … } static void encode_block_intra(int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg) { … } static void encode_block_intra_and_set_context(int plane, int block, int blk_row, int blk_col, BLOCK_SIZE plane_bsize, TX_SIZE tx_size, void *arg) { … } void av1_encode_intra_block_plane(const struct AV1_COMP *cpi, MACROBLOCK *x, BLOCK_SIZE bsize, int plane, RUN_TYPE dry_run, TRELLIS_OPT_TYPE enable_optimize_b) { … }
Codebase Browser
chromium