/* * Copyright (c) 2019, 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_PASS2_STRATEGY_H_ #define AOM_AV1_ENCODER_PASS2_STRATEGY_H_ #ifdef __cplusplus extern "C" { #endif struct AV1_COMP; struct EncodeFrameParams; #include "av1/encoder/encoder.h" /*! * \brief accumulated stats and features in a gf group */ GF_GROUP_STATS; /*! * \brief accumulated stats and features for a frame */ GF_FRAME_STATS; /*!\cond */ void av1_init_second_pass(struct AV1_COMP *cpi); void av1_init_single_pass_lap(AV1_COMP *cpi); /*!\endcond */ /*!\brief Main per frame entry point for second pass of two pass encode * *\ingroup rate_control * * This function is called for each frame in the second pass of a two pass * encode. It checks the frame type and if a new KF or GF/ARF is due. * When a KF is due it calls find_next_key_frame() to work out how long * this key frame group will be and assign bits to the key frame. * At the start of a new GF/ARF group it calls calculate_gf_length() * and define_gf_group() which are the main functions responsible for * defining the size and structure of the new GF/ARF group. * * \param[in] cpi Top - level encoder instance structure * \param[in] frame_params Per frame encoding parameters * \param[in] frame_flags Frame type and coding flags * * \remark No return but analyses first pass stats and assigns a target * number of bits to the current frame and a target Q range. */ void av1_get_second_pass_params(struct AV1_COMP *cpi, struct EncodeFrameParams *const frame_params, unsigned int frame_flags); /*!\brief Adjustments to two pass and rate control after each frame. * *\ingroup rate_control * * This function is called after each frame to make adjustments to * heuristics and data structures that relate to rate control. * * \param[in] cpi Top - level encoder instance structure * * \remark No return value but this function updates various rate control * related data structures that for example track overshoot and * undershoot. */ void av1_twopass_postencode_update(struct AV1_COMP *cpi); /*!\brief Distributes bits to frames in a group * *\ingroup rate_control * * This function decides on the allocation of bits between the different * frames and types of frame in a GF/ARF group. * * \param[in] cpi Top - level encoder instance structure * \param[in] rc Rate control data * \param[in] gf_group GF/ARF group data structure * \param[in] is_key_frame Indicates if the first frame in the group is * also a key frame. * \param[in] use_arf Are ARF frames enabled or is this a GF only * uni-directional group. * \param[in] gf_group_bits Bits available to be allocated. * * \remark No return but updates the rate control and group data structures * to reflect the allocation of bits. */ void av1_gop_bit_allocation(const AV1_COMP *cpi, RATE_CONTROL *const rc, GF_GROUP *gf_group, int is_key_frame, int use_arf, int64_t gf_group_bits); int av1_calc_arf_boost(const TWO_PASS *twopass, const TWO_PASS_FRAME *twopass_frame, const PRIMARY_RATE_CONTROL *p_rc, FRAME_INFO *frame_info, int offset, int f_frames, int b_frames, int *num_fpstats_used, int *num_fpstats_required, int project_gfu_boost); void av1_mark_flashes(FIRSTPASS_STATS *first_stats, FIRSTPASS_STATS *last_stats); void av1_estimate_noise(FIRSTPASS_STATS *first_stats, FIRSTPASS_STATS *last_stats, struct aom_internal_error_info *error_info); void av1_estimate_coeff(FIRSTPASS_STATS *first_stats, FIRSTPASS_STATS *last_stats); #ifdef __cplusplus } // extern "C" #endif #endif // AOM_AV1_ENCODER_PASS2_STRATEGY_H_
Codebase Browser
chromium