blob: ff1591ceb8d4c2a3b9ad323824d9ef222ec8f80b [file] [log] [blame]
/*
* 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
*/
typedef struct {
/*!\cond */
double gf_group_err;
double gf_group_raw_error;
double gf_group_skip_pct;
double gf_group_inactive_zone_rows;
double mv_ratio_accumulator;
double decay_accumulator;
double zero_motion_accumulator;
double loop_decay_rate;
double last_loop_decay_rate;
double this_frame_mv_in_out;
double mv_in_out_accumulator;
double abs_mv_in_out_accumulator;
double avg_sr_coded_error;
double avg_pcnt_second_ref;
double avg_new_mv_count;
double avg_wavelet_energy;
double avg_raw_err_stdev;
int non_zero_stdev_count;
/*!\endcond */
} GF_GROUP_STATS;
/*!
* \brief accumulated stats and features for a frame
*/
typedef struct {
/*!\cond */
double frame_err;
double frame_coded_error;
double frame_sr_coded_error;
/*!\endcond */
} 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);
void av1_estimate_coeff(FIRSTPASS_STATS *first_stats,
FIRSTPASS_STATS *last_stats);
#ifdef __cplusplus
} // extern "C"
#endif
#endif // AOM_AV1_ENCODER_PASS2_STRATEGY_H_