/* * 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. */ #ifndef AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_ #define AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_ #include "av1/common/blockd.h" #include "av1/encoder/block.h" #include "av1/encoder/tokenize.h" #ifdef __cplusplus extern "C" { #endif // The segment ids used in cyclic refresh: from base (no boost) to increasing // boost (higher delta-qp). #define CR_SEGMENT_ID_BASE … #define CR_SEGMENT_ID_BOOST1 … #define CR_SEGMENT_ID_BOOST2 … // Maximum rate target ratio for setting segment delta-qp. #define CR_MAX_RATE_TARGET_RATIO … /*! * \brief The stucture of CYCLIC_REFRESH. * \ingroup cyclic_refresh */ struct CYCLIC_REFRESH { … }; struct AV1_COMP; CYCLIC_REFRESH; CYCLIC_REFRESH *av1_cyclic_refresh_alloc(int mi_rows, int mi_cols); void av1_cyclic_refresh_free(CYCLIC_REFRESH *cr); /*!\brief Estimate the bits, incorporating the delta-q from the segments. * * For the just encoded frame, estimate the bits, incorporating the delta-q * from non-base segment(s). Note this function is called in the postencode * (called from rc_update_rate_correction_factors()). * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * \param[in] correction_factor rate correction factor * * \return Return the estimated bits at given q. */ int av1_cyclic_refresh_estimate_bits_at_q(const struct AV1_COMP *cpi, double correction_factor); /*!\brief Estimate the bits per mb, for given q = i and delta-q. * * Prior to encoding the frame, estimate the bits per mb, for a given q = i and * a corresponding delta-q (for segment 1). This function is called in the * rc_regulate_q() to set the base qp index. Note: the segment map is set to * either 0/CR_SEGMENT_ID_BASE (no refresh) or to 1/CR_SEGMENT_ID_BOOST1 * (refresh) for each superblock, prior to encoding. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * \param[in] i q index * \param[in] correction_factor rate correction factor * * \return Return the estimated bits for q = i and delta-q (segment 1). */ int av1_cyclic_refresh_rc_bits_per_mb(const struct AV1_COMP *cpi, int i, double correction_factor); /*!\brief Update segment_id for blocks are skipped. * * After encoding a given prediction block, of size bsize at (mi_row, mi_col), * check if we should reset the segment_id based on skip_txfm, * and update the cyclic_refresh map and segmentation counters. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * \param[in] x Pointer to MACROBLOCK structure * \param[in] mi_row Row coordinate of the block in a step size of MI_SIZE * \param[in] mi_col Col coordinate of the block in a step size of MI_SIZE * \param[in] bsize Block size * \param[in] dry_run A code indicating whether it is part of the final * pass for reconstructing the superblock * * \remark Update the \c mbmi->segment_id, the \c cpi->cyclic_refresh and * the \c cm->cpi->enc_seg.map. */ void av1_cyclic_reset_segment_skip(const struct AV1_COMP *cpi, MACROBLOCK *const x, int mi_row, int mi_col, BLOCK_SIZE bsize, RUN_TYPE dry_run); /*!\brief Update segment_id for block based on mode selected. * * Prior to coding a given prediction block, of size bsize at (mi_row, mi_col), * check if we should reset the segment_id (based on mode/motion/skip selected * for that block) and update the cyclic_refresh map and segmentation map. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * \param[in] x Pointer to MACROBLOCK structure * \param[in] mi_row Row coordinate of the block in a step size of MI_SIZE * \param[in] mi_col Col coordinate of the block in a step size of MI_SIZE * \param[in] bsize Block size * \param[in] rate Projected block rate from pickmode * \param[in] dist Projected block dist from pickmode * \param[in] skip Skip flag set from picmode * \param[in] dry_run A code indicating whether it is part of the final * pass for reconstructing the superblock * * \remark Update the \c mbmi->segment_id, the \c cpi->cyclic_refresh and * the \c cm->cpi->enc_seg.map. */ void av1_cyclic_refresh_update_segment(const struct AV1_COMP *cpi, MACROBLOCK *const x, int mi_row, int mi_col, BLOCK_SIZE bsize, int64_t rate, int64_t dist, int skip, RUN_TYPE dry_run); /*!\brief Initialize counters used for cyclic refresh. * * Initializes cyclic refresh counters actual_num_seg1_blocks and * actual_num_seg2_blocks. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] x Pointer to MACROBLOCK structure * * \remark Update the \c x->actual_num_seg1_blocks and the * \c x->actual_num_seg2_blocks. */ void av1_init_cyclic_refresh_counters(MACROBLOCK *const x); /*!\brief Accumulate cyclic refresh counters. * * Accumulates cyclic refresh counters actual_num_seg1_blocks and * actual_num_seg2_blocks from MACROBLOCK strcture to CYCLIC_REFRESH strcture. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cyclic_refresh Pointer to CYCLIC_REFRESH structure * \param[in] x Pointer to MACROBLOCK structure * * \remark Update the \c cyclic_refresh->actual_num_seg1_blocks and the * \c cyclic_refresh->actual_num_seg2_blocks. */ void av1_accumulate_cyclic_refresh_counters( CYCLIC_REFRESH *const cyclic_refresh, const MACROBLOCK *const x); /*!\brief Set golden frame update interval nased on cyclic refresh. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * * \remark Returns the interval in \c cpi->rc.baseline_gf_interval. */ void av1_cyclic_refresh_set_golden_update(struct AV1_COMP *const cpi); /*!\brief Set the global/frame level parameters for cyclic refresh. * * First call to the cyclic refresh, before encoding the frame. * Sets the flag on whether cyclic refresh should be applied, sets * the amount/percent of refresh, and the amount of boost applied to * the two segments (set by rate_ratio_qdelta and rate_boost_fac). * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * * \remark Updates the \c cpi->cyclic_refresh with the settings. */ void av1_cyclic_refresh_update_parameters(struct AV1_COMP *const cpi); /*!\brief Setup the cyclic background refresh. * * Set the delta q for the segment(s), and set the segmentation map. * * \ingroup cyclic_refresh * \callgraph * \callergraph * * \param[in] cpi Top level encoder structure * * \remark Updates the \c cpi->cyclic_refresh with the cyclic refresh * parameters and the \c cm->seg with the segmentation data. */ void av1_cyclic_refresh_setup(struct AV1_COMP *const cpi); int av1_cyclic_refresh_get_rdmult(const CYCLIC_REFRESH *cr); int av1_cyclic_refresh_disable_lf_cdef(struct AV1_COMP *const cpi); static inline int cyclic_refresh_segment_id_boosted(int segment_id) { … } static inline int cyclic_refresh_segment_id(int segment_id) { … } #ifdef __cplusplus } // extern "C" #endif #endif // AOM_AV1_ENCODER_AQ_CYCLICREFRESH_H_
Codebase Browser
chromium