15#ifndef AOM_AV1_ENCODER_ENCODER_H_
16#define AOM_AV1_ENCODER_ENCODER_H_
21#include "config/aom_config.h"
24#include "aom_util/aom_pthread.h"
26#include "av1/common/alloccommon.h"
27#include "av1/common/av1_common_int.h"
28#include "av1/common/blockd.h"
29#include "av1/common/entropymode.h"
31#include "av1/common/reconintra.h"
32#include "av1/common/resize.h"
33#include "av1/common/thread_common.h"
34#include "av1/common/timing.h"
36#include "av1/encoder/aq_cyclicrefresh.h"
37#include "av1/encoder/av1_quantize.h"
39#include "av1/encoder/context_tree.h"
40#include "av1/encoder/encodemb.h"
41#include "av1/encoder/external_partition.h"
42#include "av1/encoder/firstpass.h"
43#include "av1/encoder/global_motion.h"
44#include "av1/encoder/level.h"
46#include "av1/encoder/mcomp.h"
47#include "av1/encoder/pickcdef.h"
48#include "av1/encoder/ratectrl.h"
49#include "av1/encoder/rd.h"
51#include "av1/encoder/svc_layercontext.h"
52#include "av1/encoder/temporal_filter.h"
53#include "av1/encoder/thirdpass.h"
54#include "av1/encoder/tokenize.h"
55#include "av1/encoder/tpl_model.h"
56#include "av1/encoder/av1_noise_estimate.h"
57#include "av1/encoder/bitstream.h"
59#if CONFIG_INTERNAL_STATS
60#include "aom_dsp/ssim.h"
62#include "aom_dsp/variance.h"
64#include "aom_dsp/noise_model.h"
67#include "av1/encoder/tune_vmaf.h"
69#if CONFIG_AV1_TEMPORAL_DENOISING
70#include "av1/encoder/av1_temporal_denoiser.h"
72#if CONFIG_TUNE_BUTTERAUGLI
73#include "av1/encoder/tune_butteraugli.h"
76#include "aom/internal/aom_codec_internal.h"
77#include "aom_util/aom_thread.h"
87#define SCENE_CUT_KEY_TEST_INTERVAL 16
90#define TF_LOOKAHEAD_IDX_THR 7
92#define HDR_QP_LEVELS 10
93#define CHROMA_CB_QP_SCALE 1.04
94#define CHROMA_CR_QP_SCALE 1.04
95#define CHROMA_QP_SCALE -0.46
96#define CHROMA_QP_OFFSET 9.26
97#define QP_SCALE_FACTOR 2.0
98#define DISABLE_HDR_LUMA_DELTAQ 1
102typedef struct aom_rational64 {
119 FRAMEFLAGS_KEY = 1 << 0,
120 FRAMEFLAGS_GOLDEN = 1 << 1,
121 FRAMEFLAGS_BWDREF = 1 << 2,
123 FRAMEFLAGS_ALTREF = 1 << 3,
124 FRAMEFLAGS_INTRAONLY = 1 << 4,
125 FRAMEFLAGS_SWITCH = 1 << 5,
126 FRAMEFLAGS_ERROR_RESILIENT = 1 << 6,
127} UENUM1BYTE(FRAMETYPE_FLAGS);
132 PARALLEL_SIMULATION_ENCODE,
133 NUM_FPMT_TEST_ENCODES
134} UENUM1BYTE(FPMT_TEST_ENC_CFG);
138#define MIN_PYR_LEVEL 1
139static INLINE
int get_true_pyr_level(
int frame_level,
int frame_order,
140 int max_layer_depth) {
141 if (frame_order == 0) {
143 return MIN_PYR_LEVEL;
144 }
else if (frame_level == MAX_ARF_LAYERS) {
146 return max_layer_depth;
147 }
else if (frame_level == (MAX_ARF_LAYERS + 1)) {
149 return MIN_PYR_LEVEL;
151 return AOMMAX(MIN_PYR_LEVEL, frame_level);
158 CYCLIC_REFRESH_AQ = 3,
160} UENUM1BYTE(AQ_MODE);
163 DELTA_Q_OBJECTIVE = 1,
164 DELTA_Q_PERCEPTUAL = 2,
165 DELTA_Q_PERCEPTUAL_AI = 3,
166 DELTA_Q_USER_RATING_BASED = 4,
169} UENUM1BYTE(DELTAQ_MODE);
177} UENUM1BYTE(RESIZE_MODE);
181 SS_CFG_LOOKAHEAD = 1,
184} UENUM1BYTE(SS_CFG_OFFSET);
188 ENABLE_SCENECUT_MODE_1,
189 ENABLE_SCENECUT_MODE_2
190} UENUM1BYTE(SCENECUT_MODE);
192#define MAX_VBR_CORPUS_COMPLEXITY 10000
208} MULTI_THREADED_MODULES;
242 SKIP_APPLY_RESTORATION = 1 << 0,
243 SKIP_APPLY_SUPERRES = 1 << 1,
244 SKIP_APPLY_CDEF = 1 << 2,
245 SKIP_APPLY_LOOPFILTER = 1 << 3,
634 int gf_min_pyr_height;
636 int gf_max_pyr_height;
638 bool enable_auto_arf;
641 bool enable_auto_brf;
646 unsigned int num_tile_groups;
655 int tile_width_count;
657 int tile_height_count;
659 int tile_widths[MAX_TILE_COLS];
661 int tile_heights[MAX_TILE_ROWS];
663 bool enable_large_scale_tile;
665 bool enable_single_tile_decoding;
667 bool enable_ext_tile_debug;
677 int forced_max_frame_width;
680 int forced_max_frame_height;
691 bool enable_warped_motion;
693 bool allow_warped_motion;
700 aom_timing_info_t timing_info;
702 uint32_t num_units_in_decoding_tick;
705 bool decoder_model_info_present_flag;
708 bool display_model_info_present_flag;
710 bool timing_info_present;
726 unsigned int max_reference_frames;
728 bool enable_reduced_reference_set;
730 bool enable_onesided_comp;
748 unsigned int motion_vector_unit_test;
750 unsigned int sb_multipass_unit_test;
755 const char *vmaf_model_path;
757 const char *film_grain_table_filename;
763 int film_grain_test_vector;
770 double init_framerate;
772 unsigned int input_bit_depth;
776 unsigned int chroma_subsampling_x;
778 unsigned int chroma_subsampling_y;
785 int use_fixed_qp_offsets;
795 DELTAQ_MODE deltaq_mode;
797 DELTAQ_MODE deltaq_strength;
799 bool enable_chroma_deltaq;
801 bool enable_hdr_deltaq;
880 bool enable_deltalf_mode;
882 CDEF_CONTROL cdef_control;
884 bool enable_restoration;
886 bool force_video_mode;
888 bool error_resilient_mode;
890 bool frame_parallel_decoding_mode;
892 bool enable_monochrome;
895 bool full_still_picture_hdr;
897 bool enable_dual_filter;
899 bool enable_order_hint;
901 bool ref_frame_mvs_present;
903 bool enable_ref_frame_mvs;
905 bool enable_interintra_comp;
907 bool enable_global_motion;
922 FrameDimensionCfg frm_dim_cfg;
942 QuantizationCfg q_cfg;
974 DecoderModelCfg dec_model_cfg;
977 RefFrameCfg ref_frm_cfg;
980 UnitTestCfg unit_test_cfg;
983 MotionModeCfg motion_mode_cfg;
998 CostUpdateFreq cost_upd_freq;
1004 int noise_block_size;
1006 int enable_dnl_denoising;
1009#if CONFIG_AV1_TEMPORAL_DENOISING
1011 int noise_sensitivity;
1015 unsigned int tier_mask;
1019 int border_in_pixels;
1028 AV1_LEVEL target_seq_level_idx[MAX_NUM_OPERATING_POINTS];
1031 BITSTREAM_PROFILE profile;
1048 const char *two_pass_output;
1051 const char *second_pass_log;
1064 bool use_highbitdepth;
1069 bool save_as_annexb;
1073 const char *partition_info_path;
1078 unsigned int enable_rate_guide_deltaq;
1082 const char *rate_distribution_info;
1085 int strict_level_conformance;
1088 int kf_max_pyr_height;
1096static INLINE
int is_lossless_requested(
const RateControlCfg *
const rc_cfg) {
1110 int obmc_probs[FRAME_UPDATE_TYPES][BLOCK_SIZES_ALL];
1117 int warped_probs[FRAME_UPDATE_TYPES];
1125 int tx_type_probs[FRAME_UPDATE_TYPES][TX_SIZES_ALL][TX_TYPES];
1133 int switchable_interp_probs[FRAME_UPDATE_TYPES][SWITCHABLE_FILTER_CONTEXTS]
1134 [SWITCHABLE_FILTERS];
1139typedef struct FRAME_COUNTS {
1142#if CONFIG_ENTROPY_STATS
1143 unsigned int kf_y_mode[KF_MODE_CONTEXTS][KF_MODE_CONTEXTS][INTRA_MODES];
1144 unsigned int angle_delta[DIRECTIONAL_MODES][2 * MAX_ANGLE_DELTA + 1];
1145 unsigned int y_mode[BLOCK_SIZE_GROUPS][INTRA_MODES];
1146 unsigned int uv_mode[CFL_ALLOWED_TYPES][INTRA_MODES][UV_INTRA_MODES];
1147 unsigned int cfl_sign[CFL_JOINT_SIGNS];
1148 unsigned int cfl_alpha[CFL_ALPHA_CONTEXTS][CFL_ALPHABET_SIZE];
1149 unsigned int palette_y_mode[PALATTE_BSIZE_CTXS][PALETTE_Y_MODE_CONTEXTS][2];
1150 unsigned int palette_uv_mode[PALETTE_UV_MODE_CONTEXTS][2];
1151 unsigned int palette_y_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES];
1152 unsigned int palette_uv_size[PALATTE_BSIZE_CTXS][PALETTE_SIZES];
1153 unsigned int palette_y_color_index[PALETTE_SIZES]
1154 [PALETTE_COLOR_INDEX_CONTEXTS]
1156 unsigned int palette_uv_color_index[PALETTE_SIZES]
1157 [PALETTE_COLOR_INDEX_CONTEXTS]
1159 unsigned int partition[PARTITION_CONTEXTS][EXT_PARTITION_TYPES];
1160 unsigned int txb_skip[TOKEN_CDF_Q_CTXS][TX_SIZES][TXB_SKIP_CONTEXTS][2];
1161 unsigned int eob_extra[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1162 [EOB_COEF_CONTEXTS][2];
1163 unsigned int dc_sign[PLANE_TYPES][DC_SIGN_CONTEXTS][2];
1164 unsigned int coeff_lps[TX_SIZES][PLANE_TYPES][BR_CDF_SIZE - 1][LEVEL_CONTEXTS]
1166 unsigned int eob_flag[TX_SIZES][PLANE_TYPES][EOB_COEF_CONTEXTS][2];
1167 unsigned int eob_multi16[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][5];
1168 unsigned int eob_multi32[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][6];
1169 unsigned int eob_multi64[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][7];
1170 unsigned int eob_multi128[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][8];
1171 unsigned int eob_multi256[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][9];
1172 unsigned int eob_multi512[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][10];
1173 unsigned int eob_multi1024[TOKEN_CDF_Q_CTXS][PLANE_TYPES][2][11];
1174 unsigned int coeff_lps_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1175 [LEVEL_CONTEXTS][BR_CDF_SIZE];
1176 unsigned int coeff_base_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1177 [SIG_COEF_CONTEXTS][NUM_BASE_LEVELS + 2];
1178 unsigned int coeff_base_eob_multi[TOKEN_CDF_Q_CTXS][TX_SIZES][PLANE_TYPES]
1179 [SIG_COEF_CONTEXTS_EOB][NUM_BASE_LEVELS + 1];
1180 unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2];
1181 unsigned int zeromv_mode[GLOBALMV_MODE_CONTEXTS][2];
1182 unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2];
1183 unsigned int drl_mode[DRL_MODE_CONTEXTS][2];
1184 unsigned int inter_compound_mode[INTER_MODE_CONTEXTS][INTER_COMPOUND_MODES];
1185 unsigned int wedge_idx[BLOCK_SIZES_ALL][16];
1186 unsigned int interintra[BLOCK_SIZE_GROUPS][2];
1187 unsigned int interintra_mode[BLOCK_SIZE_GROUPS][INTERINTRA_MODES];
1188 unsigned int wedge_interintra[BLOCK_SIZES_ALL][2];
1189 unsigned int compound_type[BLOCK_SIZES_ALL][MASKED_COMPOUND_TYPES];
1190 unsigned int motion_mode[BLOCK_SIZES_ALL][MOTION_MODES];
1191 unsigned int obmc[BLOCK_SIZES_ALL][2];
1192 unsigned int intra_inter[INTRA_INTER_CONTEXTS][2];
1193 unsigned int comp_inter[COMP_INTER_CONTEXTS][2];
1194 unsigned int comp_ref_type[COMP_REF_TYPE_CONTEXTS][2];
1195 unsigned int uni_comp_ref[UNI_COMP_REF_CONTEXTS][UNIDIR_COMP_REFS - 1][2];
1196 unsigned int single_ref[REF_CONTEXTS][SINGLE_REFS - 1][2];
1197 unsigned int comp_ref[REF_CONTEXTS][FWD_REFS - 1][2];
1198 unsigned int comp_bwdref[REF_CONTEXTS][BWD_REFS - 1][2];
1199 unsigned int intrabc[2];
1201 unsigned int txfm_partition[TXFM_PARTITION_CONTEXTS][2];
1202 unsigned int intra_tx_size[MAX_TX_CATS][TX_SIZE_CONTEXTS][MAX_TX_DEPTH + 1];
1203 unsigned int skip_mode[SKIP_MODE_CONTEXTS][2];
1204 unsigned int skip_txfm[SKIP_CONTEXTS][2];
1205 unsigned int compound_index[COMP_INDEX_CONTEXTS][2];
1206 unsigned int comp_group_idx[COMP_GROUP_IDX_CONTEXTS][2];
1207 unsigned int delta_q[DELTA_Q_PROBS][2];
1208 unsigned int delta_lf_multi[FRAME_LF_COUNT][DELTA_LF_PROBS][2];
1209 unsigned int delta_lf[DELTA_LF_PROBS][2];
1211 unsigned int inter_ext_tx[EXT_TX_SETS_INTER][EXT_TX_SIZES][TX_TYPES];
1212 unsigned int intra_ext_tx[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES]
1214 unsigned int filter_intra_mode[FILTER_INTRA_MODES];
1215 unsigned int filter_intra[BLOCK_SIZES_ALL][2];
1217 unsigned int wiener_restore[2];
1218 unsigned int sgrproj_restore[2];
1221 unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS]
1222 [SWITCHABLE_FILTERS];
1225#define INTER_MODE_RD_DATA_OVERALL_SIZE 6400
1234 double sse_sse_mean;
1250#define MAX_INTER_MODES 1024
1308 uint32_t sum_square_error;
1321 VPVariance part_variances;
1326 VPVariance part_variances;
1331 VPVariance part_variances;
1336 VPVariance part_variances;
1341 VPVariance part_variances;
1346 VPVariance part_variances;
1364 int64_t thresholds[5];
1377#if CONFIG_MULTITHREAD
1391 int *num_finished_cols;
1423typedef struct TileDataEnc {
1425 DECLARE_ALIGNED(16, FRAME_CONTEXT, tctx);
1426 FRAME_CONTEXT *row_ctx;
1427 uint64_t abs_sum_level;
1428 uint8_t allow_update_cdf;
1429 InterModeRdModel inter_mode_rd_models[BLOCK_SIZES_ALL];
1431 MV firstpass_top_mv;
1434typedef struct RD_COUNTS {
1435 int compound_ref_used_flag;
1436 int skip_mode_used_flag;
1437 int tx_type_used[TX_SIZES_ALL][TX_TYPES];
1438 int obmc_used[BLOCK_SIZES_ALL][2];
1440 int newmv_or_intra_blocks;
1441 uint64_t seg_tmp_pred_cost[2];
1444typedef struct ThreadData {
1448 RD_COUNTS rd_counts;
1449 FRAME_COUNTS *counts;
1450 PC_TREE_SHARED_BUFFERS shared_coeff_buf;
1451 SIMPLE_MOTION_DATA_TREE *sms_tree;
1452 SIMPLE_MOTION_DATA_TREE *sms_root;
1453 uint32_t *hash_value_buffer[2][2];
1457 CONV_BUF_TYPE *tmp_conv_dst;
1458 uint64_t abs_sum_level;
1459 uint8_t *tmp_pred_bufs[2];
1460 uint8_t *wiener_tmp_pred_buf;
1463 int coefficient_size;
1464 int max_mv_magnitude;
1465 int interp_filter_selected[SWITCHABLE];
1466 FRAME_CONTEXT *tctx;
1468 int32_t num_64x64_blocks;
1469 PICK_MODE_CONTEXT *firstpass_ctx;
1470 TemporalFilterData tf_data;
1471 TplBuffers tpl_tmp_buffers;
1472 TplTxfmStats tpl_txfm_stats;
1473 GlobalMotionData gm_data;
1477 PixelLevelGradientInfo *pixel_gradient_info;
1482 Block4x4VarInfo *src_var_info_of_4x4_sub_blocks;
1487struct EncWorkerData;
1521 int thread_id_to_tile_id[MAX_NUM_THREADS];
1555#if CONFIG_MULTITHREAD
1585#if CONFIG_MULTITHREAD
1614#define NUM_RECODES_PER_FRAME 10
1619#define MAX_PARALLEL_FRAMES 4
1834typedef struct ActiveMap {
1850 double cs_rate_array[32];
1863#if CONFIG_INTERNAL_STATS
1871} UENUM1BYTE(StatType);
1873typedef struct IMAGE_STAT {
1874 double stat[NUM_STAT_TYPES];
1909#if CONFIG_COLLECT_PARTITION_STATS
1910typedef struct FramePartitionTimingStats {
1911 int partition_decisions[6][EXT_PARTITION_TYPES];
1912 int partition_attempts[6][EXT_PARTITION_TYPES];
1913 int64_t partition_times[6][EXT_PARTITION_TYPES];
1916} FramePartitionTimingStats;
1919#if CONFIG_COLLECT_COMPONENT_TIMING
1920#include "aom_ports/aom_timer.h"
1923 av1_encode_strategy_time,
1924 av1_get_one_pass_rt_params_time,
1925 av1_get_second_pass_params_time,
1926 denoise_and_encode_time,
1927 apply_filtering_time,
1928 av1_tpl_setup_stats_time,
1929 encode_frame_to_data_rate_time,
1930 encode_with_or_without_recode_time,
1933 loop_restoration_time,
1934 av1_pack_bitstream_final_time,
1935 av1_encode_frame_time,
1936 av1_compute_global_motion_time,
1937 av1_setup_motion_field_time,
1940 rd_pick_partition_time,
1941 rd_use_partition_time,
1942 choose_var_based_partitioning_time,
1943 av1_prune_partitions_time,
1944 none_partition_search_time,
1945 split_partition_search_time,
1946 rectangular_partition_search_time,
1947 ab_partitions_search_time,
1948 rd_pick_4partition_time,
1951 rd_pick_sb_modes_time,
1952 av1_rd_pick_intra_mode_sb_time,
1953 av1_rd_pick_inter_mode_sb_time,
1954 set_params_rd_pick_inter_mode_time,
1955 skip_inter_mode_time,
1956 handle_inter_mode_time,
1957 evaluate_motion_mode_for_winner_candidates_time,
1959 handle_intra_mode_time,
1960 refine_winner_mode_tx_time,
1961 av1_search_palette_mode_time,
1963 compound_type_rd_time,
1964 interpolation_filter_search_time,
1965 motion_mode_rd_time,
1967 nonrd_use_partition_time,
1968 pick_sb_modes_nonrd_time,
1969 hybrid_intra_mode_search_time,
1970 nonrd_pick_inter_mode_sb_time,
1971 encode_b_nonrd_time,
1974} UENUM1BYTE(TIMING_COMPONENT);
1976static INLINE
char const *get_component_name(
int index) {
1978 case av1_encode_strategy_time:
return "av1_encode_strategy_time";
1979 case av1_get_one_pass_rt_params_time:
1980 return "av1_get_one_pass_rt_params_time";
1981 case av1_get_second_pass_params_time:
1982 return "av1_get_second_pass_params_time";
1983 case denoise_and_encode_time:
return "denoise_and_encode_time";
1984 case apply_filtering_time:
return "apply_filtering_time";
1985 case av1_tpl_setup_stats_time:
return "av1_tpl_setup_stats_time";
1986 case encode_frame_to_data_rate_time:
1987 return "encode_frame_to_data_rate_time";
1988 case encode_with_or_without_recode_time:
1989 return "encode_with_or_without_recode_time";
1990 case loop_filter_time:
return "loop_filter_time";
1991 case cdef_time:
return "cdef_time";
1992 case loop_restoration_time:
return "loop_restoration_time";
1993 case av1_pack_bitstream_final_time:
return "av1_pack_bitstream_final_time";
1994 case av1_encode_frame_time:
return "av1_encode_frame_time";
1995 case av1_compute_global_motion_time:
1996 return "av1_compute_global_motion_time";
1997 case av1_setup_motion_field_time:
return "av1_setup_motion_field_time";
1998 case encode_sb_row_time:
return "encode_sb_row_time";
2000 case rd_pick_partition_time:
return "rd_pick_partition_time";
2001 case rd_use_partition_time:
return "rd_use_partition_time";
2002 case choose_var_based_partitioning_time:
2003 return "choose_var_based_partitioning_time";
2004 case av1_prune_partitions_time:
return "av1_prune_partitions_time";
2005 case none_partition_search_time:
return "none_partition_search_time";
2006 case split_partition_search_time:
return "split_partition_search_time";
2007 case rectangular_partition_search_time:
2008 return "rectangular_partition_search_time";
2009 case ab_partitions_search_time:
return "ab_partitions_search_time";
2010 case rd_pick_4partition_time:
return "rd_pick_4partition_time";
2011 case encode_sb_time:
return "encode_sb_time";
2013 case rd_pick_sb_modes_time:
return "rd_pick_sb_modes_time";
2014 case av1_rd_pick_intra_mode_sb_time:
2015 return "av1_rd_pick_intra_mode_sb_time";
2016 case av1_rd_pick_inter_mode_sb_time:
2017 return "av1_rd_pick_inter_mode_sb_time";
2018 case set_params_rd_pick_inter_mode_time:
2019 return "set_params_rd_pick_inter_mode_time";
2020 case skip_inter_mode_time:
return "skip_inter_mode_time";
2021 case handle_inter_mode_time:
return "handle_inter_mode_time";
2022 case evaluate_motion_mode_for_winner_candidates_time:
2023 return "evaluate_motion_mode_for_winner_candidates_time";
2024 case do_tx_search_time:
return "do_tx_search_time";
2025 case handle_intra_mode_time:
return "handle_intra_mode_time";
2026 case refine_winner_mode_tx_time:
return "refine_winner_mode_tx_time";
2027 case av1_search_palette_mode_time:
return "av1_search_palette_mode_time";
2028 case handle_newmv_time:
return "handle_newmv_time";
2029 case compound_type_rd_time:
return "compound_type_rd_time";
2030 case interpolation_filter_search_time:
2031 return "interpolation_filter_search_time";
2032 case motion_mode_rd_time:
return "motion_mode_rd_time";
2034 case nonrd_use_partition_time:
return "nonrd_use_partition_time";
2035 case pick_sb_modes_nonrd_time:
return "pick_sb_modes_nonrd_time";
2036 case hybrid_intra_mode_search_time:
return "hybrid_intra_mode_search_time";
2037 case nonrd_pick_inter_mode_sb_time:
return "nonrd_pick_inter_mode_sb_time";
2038 case encode_b_nonrd_time:
return "encode_b_nonrd_time";
2047#define MAX_INTERNAL_ARFS (REF_FRAMES - BWDREF_FRAME - 1)
2072 int num_ref_frames[MAX_DIRECTIONS];
2080 FrameDistPair reference_frames[MAX_DIRECTIONS][REF_FRAMES - 1];
2136 search_site_config search_site_cfg[SS_CFG_TOTAL][NUM_DISTINCT_SEARCH_METHODS];
2172 int ref_relative_dist[INTER_REFS_PER_FRAME];
2206 unsigned int coeff_opt_thresholds[MODE_EVAL_TYPES][2];
2212 TX_SIZE_SEARCH_METHOD tx_size_search_methods[MODE_EVAL_TYPES];
2220 unsigned int use_transform_domain_distortion[MODE_EVAL_TYPES];
2227 unsigned int tx_domain_dist_threshold[MODE_EVAL_TYPES];
2234 unsigned int skip_txfm_level[MODE_EVAL_TYPES];
2241 unsigned int predict_dc_level[MODE_EVAL_TYPES];
2322 int mv_joint_count[4];
2324 int last_bit_nonzero;
2328 int hp_total_mv_rate;
2329 int lp_total_mv_rate;
2340typedef struct WeberStats {
2341 int64_t mb_wiener_variance;
2342 int64_t src_variance;
2343 int64_t rec_variance;
2344 int16_t src_pix_max;
2345 int16_t rec_pix_max;
2352 struct loopfilter lf;
2376 int show_frame_count;
2436#if !CONFIG_REALTIME_ONLY
2440 DUCKY_ENCODE_FRAME_MODE_NONE,
2441 DUCKY_ENCODE_FRAME_MODE_QINDEX,
2443 DUCKY_ENCODE_FRAME_MODE_QINDEX_RDMULT,
2445} UENUM1BYTE(DUCKY_ENCODE_FRAME_MODE);
2448 DUCKY_ENCODE_GOP_MODE_NONE,
2449 DUCKY_ENCODE_GOP_MODE_RCL,
2450} UENUM1BYTE(DUCKY_ENCODE_GOP_MODE);
2452typedef struct DuckyEncodeFrameInfo {
2453 DUCKY_ENCODE_FRAME_MODE qp_mode;
2454 DUCKY_ENCODE_GOP_MODE gop_mode;
2458 int *superblock_encode_qindex;
2459 int *superblock_encode_rdmult;
2460 int delta_q_enabled;
2461} DuckyEncodeFrameInfo;
2463typedef struct DuckyEncodeFrameResult {
2464 int global_order_idx;
2470} DuckyEncodeFrameResult;
2472typedef struct DuckyEncodeInfo {
2473 DuckyEncodeFrameInfo frame_info;
2474 DuckyEncodeFrameResult frame_result;
2480typedef struct RTC_REF {
2485 int reference[INTER_REFS_PER_FRAME];
2486 int ref_idx[INTER_REFS_PER_FRAME];
2487 int refresh[REF_FRAMES];
2488 int set_ref_frame_config;
2489 int non_reference_frame;
2490 int ref_frame_comp[3];
2495 unsigned int buffer_time_index[REF_FRAMES];
2499 unsigned char buffer_spatial_layer[REF_FRAMES];
2503 bool reference_was_previous_frame;
2508 bool bias_recovery_frame;
2587 FPMT_TEST_ENC_CFG fpmt_unit_test_cfg;
2605 int temp_valid_gm_model_found[FRAME_UPDATE_TYPES];
2612 RefCntBuffer *ref_frame_map_copy[REF_FRAMES];
2763 aom_variance_fn_ptr_t
fn_ptr[BLOCK_SIZES_ALL];
2781#if CONFIG_INTERNAL_STATS
2783 uint64_t total_time_receive_data;
2784 uint64_t total_time_compress_data;
2786 unsigned int total_mode_chosen_counts[MAX_MODES];
2789 uint64_t total_sq_error[2];
2790 uint64_t total_samples[2];
2793 double total_blockiness;
2794 double worst_blockiness;
2797 double summed_quality;
2798 double summed_weights;
2799 double summed_quality_hbd;
2800 double summed_weights_hbd;
2801 unsigned int total_recode_hits;
2803 double worst_ssim_hbd;
2808 int b_calculate_blockiness;
2809 int b_calculate_consistency;
2811 double total_inconsistency;
2812 double worst_consistency;
2818#if CONFIG_ENTROPY_STATS
2822 FRAME_COUNTS aggregate_fc;
3126#if CONFIG_INTERNAL_STATS
3128 uint64_t time_compress_data;
3130 unsigned int mode_chosen_counts[MAX_MODES];
3132 unsigned int frame_recode_hits;
3136#if CONFIG_SPEED_STATS
3140 unsigned int tx_search_count;
3264 double temp_framerate;
3345#if CONFIG_COLLECT_PARTITION_STATS
3349 FramePartitionTimingStats partition_stats;
3352#if CONFIG_COLLECT_COMPONENT_TIMING
3356 uint64_t component_time[kTimingComponents];
3361 struct aom_usec_timer component_timer[kTimingComponents];
3365 uint64_t frame_component_time[kTimingComponents];
3394 TuneVMAFInfo vmaf_info;
3397#if CONFIG_TUNE_BUTTERAUGLI
3401 TuneButteraugliInfo butteraugli_info;
3442#if CONFIG_AV1_TEMPORAL_DENOISING
3446 AV1_DENOISER denoiser;
3521#if CONFIG_RD_COMMAND
3525 RD_COMMAND rd_command;
3570#if CONFIG_BITRATE_ACCURACY
3574 VBR_RATECTRL_INFO vbr_rc_info;
3577#if CONFIG_RATECTRL_LOG
3581 RATECTRL_LOG rc_log;
3617#if !CONFIG_REALTIME_ONLY
3640#if CONFIG_SALIENCY_MAP
3644 uint8_t *saliency_map;
3649 double *sm_scaling_factor;
3672 int64_t ts_duration;
3691 int primary_ref_frame;
3701 int refresh_frame_flags;
3703 int show_existing_frame;
3704 int existing_fb_idx_to_show;
3735} EncodeFrameResults;
3737void av1_initialize_enc(
unsigned int usage,
enum aom_rc_mode end_usage);
3741 BufferPool *
const pool,
3742 COMPRESSOR_STAGE stage,
3743 int lap_lag_in_frames);
3746 struct aom_codec_pkt_list *pkt_list_head,
int num_lap_buffers,
3751void av1_remove_primary_compressor(
AV1_PRIMARY *ppi);
3753#if CONFIG_ENTROPY_STATS
3756#if CONFIG_INTERNAL_STATS
3761 bool *sb_size_changed);
3764 bool sb_size_changed);
3767 int subsampling_x,
int subsampling_y);
3769void av1_init_seq_coding_tools(
AV1_PRIMARY *
const ppi,
3775void av1_scale_references_fpmt(
AV1_COMP *
cpi,
int *ref_buffers_used_map);
3777void av1_increment_scaled_ref_counts_fpmt(BufferPool *buffer_pool,
3778 int ref_buffers_used_map);
3780void av1_release_scaled_references_fpmt(
AV1_COMP *
cpi);
3782void av1_decrement_ref_counts_fpmt(BufferPool *buffer_pool,
3783 int ref_buffers_used_map);
3785void av1_init_sc_decisions(
AV1_PRIMARY *
const ppi);
3790int av1_init_parallel_frame_context(
const AV1_COMP_DATA *
const first_cpi_data,
3792 int *ref_buffers_used_map);
3813 int64_t end_time_stamp);
3847 EncodeFrameResults *
const frame_results);
3858int av1_use_as_reference(
int *ext_ref_frame_flags,
int ref_frame_flags);
3864void av1_set_frame_size(
AV1_COMP *
cpi,
int width,
int height);
3868int av1_set_active_map(
AV1_COMP *
cpi,
unsigned char *map,
int rows,
int cols);
3870int av1_get_active_map(
AV1_COMP *
cpi,
unsigned char *map,
int rows,
int cols);
3879int av1_convert_sect5obus_to_annexb(uint8_t *buffer,
size_t *input_size);
3881void av1_alloc_mb_wiener_var_pred_buf(
AV1_COMMON *cm, ThreadData *td);
3883void av1_dealloc_mb_wiener_var_pred_buf(ThreadData *td);
3896void av1_set_screen_content_options(
struct AV1_COMP *
cpi,
3906static INLINE
void init_ref_map_pair(
3907 AV1_COMP *cpi, RefFrameMapPair ref_frame_map_pairs[REF_FRAMES]) {
3909 memset(ref_frame_map_pairs, -1,
sizeof(*ref_frame_map_pairs) * REF_FRAMES);
3912 memset(ref_frame_map_pairs, 0,
sizeof(*ref_frame_map_pairs) * REF_FRAMES);
3913 for (
int map_idx = 0; map_idx < REF_FRAMES; map_idx++) {
3916 if (ref_frame_map_pairs[map_idx].disp_order == -1)
continue;
3918 ref_frame_map_pairs[map_idx].disp_order = -1;
3919 ref_frame_map_pairs[map_idx].pyr_level = -1;
3921 }
else if (buf->ref_count > 1) {
3927 for (
int idx2 = map_idx + 1; idx2 < REF_FRAMES; ++idx2) {
3930 ref_frame_map_pairs[idx2].disp_order = -1;
3931 ref_frame_map_pairs[idx2].pyr_level = -1;
3935 ref_frame_map_pairs[map_idx].disp_order = (int)buf->display_order_hint;
3936 ref_frame_map_pairs[map_idx].pyr_level = buf->pyramid_level;
3941static AOM_INLINE
void calc_frame_data_update_flag(
3942 GF_GROUP *
const gf_group,
int gf_frame_index,
3943 bool *
const do_frame_data_update) {
3944 *do_frame_data_update =
true;
3947 if (gf_group->frame_parallel_level[gf_frame_index] == 1) {
3948 *do_frame_data_update =
false;
3949 }
else if (gf_group->frame_parallel_level[gf_frame_index] == 2) {
3951 for (
int i = gf_frame_index + 1; i < gf_group->size; i++) {
3952 if ((gf_group->frame_parallel_level[i] == 0 &&
3953 (gf_group->update_type[i] == ARF_UPDATE ||
3954 gf_group->update_type[i] == INTNL_ARF_UPDATE)) ||
3955 gf_group->frame_parallel_level[i] == 1) {
3957 }
else if (gf_group->frame_parallel_level[i] == 2) {
3958 *do_frame_data_update =
false;
3967#define TICKS_PER_SEC 10000000LL
3969static INLINE int64_t
3970timebase_units_to_ticks(
const aom_rational64_t *timestamp_ratio, int64_t n) {
3971 return n * timestamp_ratio->num / timestamp_ratio->den;
3974static INLINE int64_t
3975ticks_to_timebase_units(
const aom_rational64_t *timestamp_ratio, int64_t n) {
3976 int64_t round = timestamp_ratio->num / 2;
3977 if (round > 0) --round;
3978 return (n * timestamp_ratio->den + round) / timestamp_ratio->num;
3981static INLINE
int frame_is_kf_gf_arf(
const AV1_COMP *cpi) {
3983 const FRAME_UPDATE_TYPE update_type =
3986 return frame_is_intra_only(&cpi->
common) || update_type == ARF_UPDATE ||
3987 update_type == GF_UPDATE;
3991static INLINE
int av1_use_hash_me(
const AV1_COMP *
const cpi) {
3994 frame_is_intra_only(&cpi->
common));
3998 const AV1_COMMON *
const cm, MV_REFERENCE_FRAME ref_frame) {
3999 const RefCntBuffer *
const buf = get_ref_frame_buf(cm, ref_frame);
4000 return buf != NULL ? &buf->buf : NULL;
4003static INLINE
void alloc_frame_mvs(
AV1_COMMON *
const cm, RefCntBuffer *buf) {
4004 assert(buf != NULL);
4005 ensure_mv_buffer(buf, cm);
4006 buf->width = cm->
width;
4007 buf->height = cm->
height;
4012static INLINE
unsigned int allocated_tokens(
const TileInfo *tile,
4013 int sb_size_log2,
int num_planes) {
4015 ROUND_POWER_OF_TWO(tile->mi_row_end - tile->mi_row_start, 2);
4017 ROUND_POWER_OF_TWO(tile->mi_col_end - tile->mi_col_start, 2);
4019 return get_token_alloc(tile_mb_rows, tile_mb_cols, sb_size_log2, num_planes);
4022static INLINE
void get_start_tok(
AV1_COMP *cpi,
int tile_row,
int tile_col,
4023 int mi_row, TokenExtra **tok,
int sb_size_log2,
4027 TileDataEnc *this_tile = &cpi->
tile_data[tile_row * tile_cols + tile_col];
4028 const TileInfo *
const tile_info = &this_tile->tile_info;
4030 const int tile_mb_cols =
4031 (tile_info->mi_col_end - tile_info->mi_col_start + 2) >> 2;
4032 const int tile_mb_row = (mi_row - tile_info->mi_row_start + 2) >> 2;
4034 *tok = cpi->
token_info.tile_tok[tile_row][tile_col] +
4035 get_token_alloc(tile_mb_row, tile_mb_cols, sb_size_log2, num_planes);
4040#define ALT_MIN_LAG 3
4041static INLINE
int is_altref_enabled(
int lag_in_frames,
bool enable_auto_arf) {
4042 return lag_in_frames >= ALT_MIN_LAG && enable_auto_arf;
4045static AOM_INLINE
int can_disable_altref(
const GFConfig *gf_cfg) {
4046 return is_altref_enabled(gf_cfg->lag_in_frames, gf_cfg->enable_auto_arf) &&
4047 (gf_cfg->gf_min_pyr_height == 0);
4051static INLINE
int get_num_blocks(
const int frame_length,
const int mb_length) {
4052 return (frame_length + mb_length - 1) / mb_length;
4056static INLINE
int is_stat_generation_stage(
const AV1_COMP *
const cpi) {
4063static INLINE
int is_stat_consumption_stage_twopass(
const AV1_COMP *
const cpi) {
4068static INLINE
int is_stat_consumption_stage(
const AV1_COMP *
const cpi) {
4069 return (is_stat_consumption_stage_twopass(cpi) ||
4075static AOM_INLINE
bool av1_need_dv_costs(
const AV1_COMP *
const cpi) {
4076 return !cpi->
sf.
rt_sf.use_nonrd_pick_mode &&
4077 av1_allow_intrabc(&cpi->
common) && !is_stat_generation_stage(cpi);
4097static INLINE
int is_one_pass_rt_params(
const AV1_COMP *cpi) {
4099 cpi->
oxcf.gf_cfg.lag_in_frames == 0;
4103static INLINE
int use_rtc_reference_structure_one_layer(
const AV1_COMP *cpi) {
4110static INLINE
int get_stats_buf_size(
int num_lap_buffer,
int num_lag_buffer) {
4112 return (num_lap_buffer > 0 ? num_lap_buffer + 1 : num_lag_buffer);
4118 MV_REFERENCE_FRAME ref0,
4119 MV_REFERENCE_FRAME ref1) {
4121 get_ref_scale_factors_const(cm, ref0 >= LAST_FRAME ? ref0 : 1);
4123 get_ref_scale_factors_const(cm, ref1 >= LAST_FRAME ? ref1 : 1);
4126static INLINE
int get_chessboard_index(
int frame_index) {
4127 return frame_index & 0x1;
4130static INLINE
const int *cond_cost_list_const(
const struct AV1_COMP *cpi,
4131 const int *cost_list) {
4132 const int use_cost_list = cpi->
sf.
mv_sf.subpel_search_method != SUBPEL_TREE &&
4133 cpi->
sf.
mv_sf.use_fullpel_costlist;
4134 return use_cost_list ? cost_list : NULL;
4137static INLINE
int *cond_cost_list(
const struct AV1_COMP *cpi,
int *cost_list) {
4138 const int use_cost_list = cpi->
sf.
mv_sf.subpel_search_method != SUBPEL_TREE &&
4139 cpi->
sf.
mv_sf.use_fullpel_costlist;
4140 return use_cost_list ? cost_list : NULL;
4144double av1_get_compression_ratio(
const AV1_COMMON *
const cm,
4145 size_t encoded_frame_size);
4147void av1_new_framerate(
AV1_COMP *cpi,
double framerate);
4149void av1_setup_frame_size(
AV1_COMP *cpi);
4151#define LAYER_IDS_TO_IDX(sl, tl, num_tl) ((sl) * (num_tl) + (tl))
4154static INLINE
int av1_resize_scaled(
const AV1_COMMON *cm) {
4159static INLINE
int av1_frame_scaled(
const AV1_COMMON *cm) {
4160 return av1_superres_scaled(cm) || av1_resize_scaled(cm);
4166static INLINE
int encode_show_existing_frame(
const AV1_COMMON *cm) {
4173static INLINE
int get_mi_ext_idx(
const int mi_row,
const int mi_col,
4174 const BLOCK_SIZE mi_alloc_bsize,
4175 const int mbmi_ext_stride) {
4176 const int mi_ext_size_1d = mi_size_wide[mi_alloc_bsize];
4177 const int mi_ext_row = mi_row / mi_ext_size_1d;
4178 const int mi_ext_col = mi_col / mi_ext_size_1d;
4179 return mi_ext_row * mbmi_ext_stride + mi_ext_col;
4184static INLINE
void set_mode_info_offsets(
4188 set_mi_offsets(mi_params, xd, mi_row, mi_col);
4189 const int ext_idx = get_mi_ext_idx(mi_row, mi_col, mi_params->
mi_alloc_bsize,
4197static INLINE BLOCK_SIZE find_partition_size(BLOCK_SIZE bsize,
int rows_left,
4198 int cols_left,
int *bh,
int *bw) {
4199 int int_size = (int)bsize;
4200 if (rows_left <= 0 || cols_left <= 0) {
4201 return AOMMIN(bsize, BLOCK_8X8);
4203 for (; int_size > 0; int_size -= 3) {
4204 *bh = mi_size_high[int_size];
4205 *bw = mi_size_wide[int_size];
4206 if ((*bh <= rows_left) && (*bw <= cols_left)) {
4211 return (BLOCK_SIZE)int_size;
4214static const uint8_t av1_ref_frame_flag_list[REF_FRAMES] = { 0,
4225static const MV_REFERENCE_FRAME disable_order[] = {
4232static const MV_REFERENCE_FRAME
4233 ref_frame_priority_order[INTER_REFS_PER_FRAME] = {
4234 LAST_FRAME, ALTREF_FRAME, BWDREF_FRAME, GOLDEN_FRAME,
4235 ALTREF2_FRAME, LAST2_FRAME, LAST3_FRAME,
4238static INLINE
int get_ref_frame_flags(
const SPEED_FEATURES *
const sf,
4239 const int use_one_pass_rt_params,
4241 const int ext_ref_frame_flags) {
4246 int flags = ext_ref_frame_flags;
4248 for (
int i = 1; i < INTER_REFS_PER_FRAME; ++i) {
4254 (use_one_pass_rt_params && ref_frame_priority_order[i] == GOLDEN_FRAME)
4255 ? (1 + sf->
rt_sf.use_nonrd_altref_frame)
4257 for (
int j = 0; j < index; ++j) {
4261 if (this_ref == ref_frames[j] &&
4262 (flags & (1 << (ref_frame_priority_order[j] - 1)))) {
4263 flags &= ~(1 << (ref_frame_priority_order[i] - 1));
4282#define MAX_GFUBOOST_FACTOR 10.0
4283#define MIN_GFUBOOST_FACTOR 4.0
4285static INLINE
int is_frame_tpl_eligible(
const GF_GROUP *
const gf_group,
4287 const FRAME_UPDATE_TYPE update_type = gf_group->update_type[index];
4288 return update_type == ARF_UPDATE || update_type == GF_UPDATE ||
4289 update_type == KF_UPDATE;
4292static INLINE
int is_frame_eligible_for_ref_pruning(
const GF_GROUP *gf_group,
4293 int selective_ref_frame,
4294 int prune_ref_frames,
4296 return (selective_ref_frame > 0) && (prune_ref_frames > 0) &&
4297 !is_frame_tpl_eligible(gf_group, gf_index);
4301static INLINE FRAME_UPDATE_TYPE get_frame_update_type(
const GF_GROUP *gf_group,
4302 int gf_frame_index) {
4303 return gf_group->update_type[gf_frame_index];
4306static INLINE
int av1_pixels_to_mi(
int pixels) {
4307 return ALIGN_POWER_OF_TWO(pixels, 3) >> MI_SIZE_LOG2;
4310static AOM_INLINE
int is_psnr_calc_enabled(
const AV1_COMP *cpi) {
4317static INLINE
int is_frame_resize_pending(
const AV1_COMP *
const cpi) {
4320 return (resize_pending_params->
width && resize_pending_params->
height &&
4326static INLINE
int is_loopfilter_used(
const AV1_COMMON *
const cm) {
4331static INLINE
int is_cdef_used(
const AV1_COMMON *
const cm) {
4337static INLINE
int is_restoration_used(
const AV1_COMMON *
const cm) {
4347static INLINE
unsigned int derive_skip_apply_postproc_filters(
4348 const AV1_COMP *cpi,
int use_loopfilter,
int use_cdef,
int use_superres,
4349 int use_restoration) {
4356 return (SKIP_APPLY_LOOPFILTER | SKIP_APPLY_CDEF);
4360 assert(cpi->
oxcf.mode == ALLINTRA);
4369 if (use_restoration)
return SKIP_APPLY_RESTORATION;
4370 if (use_superres)
return SKIP_APPLY_SUPERRES;
4377 ? (SKIP_APPLY_LOOPFILTER | SKIP_APPLY_CDEF)
4380 if (use_loopfilter)
return SKIP_APPLY_LOOPFILTER;
4387static INLINE
void set_postproc_filter_default_params(
AV1_COMMON *cm) {
4388 struct loopfilter *
const lf = &cm->
lf;
4392 lf->filter_level[0] = 0;
4393 lf->filter_level[1] = 0;
4403static INLINE
int is_inter_tx_size_search_level_one(
4404 const TX_SPEED_FEATURES *tx_sf) {
4405 return (tx_sf->inter_tx_size_search_init_depth_rect >= 1 &&
4406 tx_sf->inter_tx_size_search_init_depth_sqr >= 1);
4410 int lpf_opt_level = 0;
4411 if (is_inter_tx_size_search_level_one(&sf->
tx_sf))
4412 lpf_opt_level = (sf->
lpf_sf.lpf_pick == LPF_PICK_FROM_Q) ? 2 : 1;
4413 return lpf_opt_level;
4417static INLINE
bool is_switchable_motion_mode_allowed(
bool allow_warped_motion,
4419 return (allow_warped_motion || enable_obmc);
4422#if CONFIG_AV1_TEMPORAL_DENOISING
4423static INLINE
int denoise_svc(
const struct AV1_COMP *
const cpi) {
4426 cpi->
svc.spatial_layer_id >= cpi->
svc.first_layer_denoise));
4430#if CONFIG_COLLECT_PARTITION_STATS == 2
4431static INLINE
void av1_print_fr_partition_timing_stats(
4432 const FramePartitionTimingStats *part_stats,
const char *filename) {
4433 FILE *f = fopen(filename,
"w");
4438 fprintf(f,
"bsize,redo,");
4439 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4440 fprintf(f,
"decision_%d,", part);
4442 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4443 fprintf(f,
"attempt_%d,", part);
4445 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4446 fprintf(f,
"time_%d,", part);
4450 static const int bsizes[6] = { 128, 64, 32, 16, 8, 4 };
4452 for (
int bsize_idx = 0; bsize_idx < 6; bsize_idx++) {
4453 fprintf(f,
"%d,%d,", bsizes[bsize_idx], part_stats->partition_redo);
4454 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4455 fprintf(f,
"%d,", part_stats->partition_decisions[bsize_idx][part]);
4457 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4458 fprintf(f,
"%d,", part_stats->partition_attempts[bsize_idx][part]);
4460 for (
int part = 0; part < EXT_PARTITION_TYPES; part++) {
4461 fprintf(f,
"%ld,", part_stats->partition_times[bsize_idx][part]);
4469#if CONFIG_COLLECT_PARTITION_STATS
4470static INLINE
int av1_get_bsize_idx_for_part_stats(BLOCK_SIZE bsize) {
4471 assert(bsize == BLOCK_128X128 || bsize == BLOCK_64X64 ||
4472 bsize == BLOCK_32X32 || bsize == BLOCK_16X16 || bsize == BLOCK_8X8 ||
4473 bsize == BLOCK_4X4);
4475 case BLOCK_128X128:
return 0;
4476 case BLOCK_64X64:
return 1;
4477 case BLOCK_32X32:
return 2;
4478 case BLOCK_16X16:
return 3;
4479 case BLOCK_8X8:
return 4;
4480 case BLOCK_4X4:
return 5;
4481 default: assert(0 &&
"Invalid bsize for partition_stats.");
return -1;
4486#if CONFIG_COLLECT_COMPONENT_TIMING
4487static INLINE
void start_timing(
AV1_COMP *cpi,
int component) {
4488 aom_usec_timer_start(&cpi->component_timer[component]);
4490static INLINE
void end_timing(
AV1_COMP *cpi,
int component) {
4491 aom_usec_timer_mark(&cpi->component_timer[component]);
4492 cpi->frame_component_time[component] +=
4493 aom_usec_timer_elapsed(&cpi->component_timer[component]);
4495static INLINE
char const *get_frame_type_enum(
int type) {
4497 case 0:
return "KEY_FRAME";
4498 case 1:
return "INTER_FRAME";
4499 case 2:
return "INTRA_ONLY_FRAME";
4500 case 3:
return "S_FRAME";
enum aom_chroma_sample_position aom_chroma_sample_position_t
List of chroma sample positions.
enum aom_transfer_characteristics aom_transfer_characteristics_t
List of supported transfer functions.
enum aom_color_range aom_color_range_t
List of supported color range.
enum aom_color_primaries aom_color_primaries_t
List of supported color primaries.
enum aom_matrix_coefficients aom_matrix_coefficients_t
List of supported matrix coefficients.
Provides definitions for using AOM or AV1 encoder algorithm within the aom Codec Interface.
struct AV1_COMP AV1_COMP
Top level encoder structure.
COST_UPDATE_TYPE
This enum controls how often the entropy costs should be updated.
Definition encoder.h:218
@ COST_UPD_SBROW
Definition encoder.h:220
@ COST_UPD_TILE
Definition encoder.h:221
@ COST_UPD_OFF
Definition encoder.h:222
@ NUM_COST_UPDATE_TYPES
Definition encoder.h:223
@ COST_UPD_SB
Definition encoder.h:219
struct EncodeFrameParams EncodeFrameParams
contains per-frame encoding parameters decided upon by av1_encode_strategy() and passed down to av1_e...
struct PrimaryMultiThreadInfo PrimaryMultiThreadInfo
Primary Encoder parameters related to multi-threading.
struct EncodeFrameInput EncodeFrameInput
Input frames and last input frame.
struct MultiThreadInfo MultiThreadInfo
Encoder parameters related to multi-threading.
struct AV1_COMP_DATA AV1_COMP_DATA
Structure to hold data corresponding to an encoded frame.
LOOPFILTER_CONTROL
This enum controls to which frames loopfilter is applied.
Definition encoder.h:229
@ LOOPFILTER_ALL
Definition encoder.h:231
@ LOOPFILTER_SELECTIVELY
Definition encoder.h:233
@ LOOPFILTER_REFERENCE
Definition encoder.h:232
@ LOOPFILTER_NONE
Definition encoder.h:230
#define NUM_RECODES_PER_FRAME
Max number of recodes used to track the frame probabilities.
Definition encoder.h:1614
struct AV1_PRIMARY AV1_PRIMARY
Top level primary encoder structure.
struct inter_modes_info InterModesInfo
Struct used to hold inter mode data for fast tx search.
SKIP_APPLY_POSTPROC_FILTER
This enum controls the application of post-processing filters on a reconstructed frame.
Definition encoder.h:241
struct RestoreStateBuffers RestoreStateBuffers
Buffers to be backed up during parallel encode set to be restored later.
struct AV1EncoderConfig AV1EncoderConfig
Main encoder configuration data structure.
#define MAX_PARALLEL_FRAMES
Max number of frames that can be encoded in a parallel encode set.
Definition encoder.h:1619
RestorationType
This enumeration defines various restoration types supported.
Definition enums.h:609
@ RESTORE_NONE
Definition enums.h:610
@ RESTORE_SWITCHABLE_TYPES
Definition enums.h:614
@ RESTORE_TYPES
Definition enums.h:615
aom_dist_metric
Distortion metric to use for RD optimization.
Definition aomcx.h:1653
aom_tune_content
Definition aomcx.h:1614
enum aom_scaling_mode_1d AOM_SCALING_MODE
aom 1-D scaling mode
aom_tune_metric
Model tuning parameters.
Definition aomcx.h:1633
enum aom_bit_depth aom_bit_depth_t
Bit depth for codecThis enumeration determines the bit depth of the codec.
enum aom_superblock_size aom_superblock_size_t
Superblock size selection.
aom_codec_err_t
Algorithm return codes.
Definition aom_codec.h:155
aom_superres_mode
Frame super-resolution mode.
Definition aom_encoder.h:205
aom_rc_mode
Rate control mode.
Definition aom_encoder.h:183
aom_enc_pass
Multi-pass Encoding Pass.
Definition aom_encoder.h:174
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition aom_encoder.h:375
@ AOM_RC_ONE_PASS
Definition aom_encoder.h:175
@ AOM_RC_SECOND_PASS
Definition aom_encoder.h:177
@ AOM_RC_FIRST_PASS
Definition aom_encoder.h:176
int av1_get_compressed_data(AV1_COMP *cpi, AV1_COMP_DATA *const cpi_data)
Encode a frame.
Definition encoder.c:4693
int av1_receive_raw_frame(AV1_COMP *cpi, aom_enc_frame_flags_t frame_flags, const YV12_BUFFER_CONFIG *sd, int64_t time_stamp, int64_t end_time_stamp)
Obtain the raw frame data.
Definition encoder.c:4084
int av1_encode(AV1_COMP *const cpi, uint8_t *const dest, const EncodeFrameInput *const frame_input, const EncodeFrameParams *const frame_params, EncodeFrameResults *const frame_results)
Run 1-pass/2-pass encoding.
Definition encoder.c:3986
static int has_no_stats_stage(const AV1_COMP *const cpi)
Check if the current stage has statistics.
Definition encoder.h:4089
Describes look ahead buffer operations.
@ CDEF_PICK_FROM_Q
Definition speed_features.h:172
Top level common structure used by both encoder and decoder.
Definition av1_common_int.h:752
RestorationInfo rst_info[3]
Definition av1_common_int.h:948
int superres_upscaled_width
Definition av1_common_int.h:801
int superres_upscaled_height
Definition av1_common_int.h:802
SequenceHeader * seq_params
Definition av1_common_int.h:978
int width
Definition av1_common_int.h:777
CdefInfo cdef_info
Definition av1_common_int.h:957
CurrentFrame current_frame
Definition av1_common_int.h:756
int show_existing_frame
Definition av1_common_int.h:902
struct loopfilter lf
Definition av1_common_int.h:941
FeatureFlags features
Definition av1_common_int.h:907
int show_frame
Definition av1_common_int.h:887
RefCntBuffer * ref_frame_map[REF_FRAMES]
Definition av1_common_int.h:880
CommonTileParams tiles
Definition av1_common_int.h:994
int height
Definition av1_common_int.h:778
int render_width
Definition av1_common_int.h:788
int render_height
Definition av1_common_int.h:789
Encoder data related to multi-threading for allintra deltaq-mode=3.
Definition encoder.h:1584
pthread_mutex_t * mutex_
Definition encoder.h:1589
pthread_cond_t * cond_
Definition encoder.h:1593
Encoder data related to row-based multi-threading.
Definition encoder.h:1494
int allocated_sb_rows
Definition encoder.h:1533
pthread_mutex_t * mutex_
Definition encoder.h:1559
int allocated_tile_cols
Definition encoder.h:1502
bool firstpass_mt_exit
Definition encoder.h:1546
int allocated_cols
Definition encoder.h:1516
bool mb_wiener_mt_exit
Definition encoder.h:1553
pthread_cond_t * cond_
Definition encoder.h:1563
bool row_mt_exit
Definition encoder.h:1539
int allocated_tile_rows
Definition encoder.h:1498
int allocated_rows
Definition encoder.h:1509
int * num_tile_cols_done
Definition encoder.h:1527
Encoder parameters for synchronization of row based multi-threading.
Definition encoder.h:1376
int rows
Definition encoder.h:1409
int sync_range
Definition encoder.h:1398
int intrabc_extra_top_right_sb_delay
Definition encoder.h:1405
int next_mi_row
Definition encoder.h:1413
pthread_mutex_t * mutex_
Definition encoder.h:1382
pthread_cond_t * cond_
Definition encoder.h:1383
int num_threads_working
Definition encoder.h:1417
Main encoder configuration data structure.
Definition encoder.h:916
RateControlCfg rc_cfg
Definition encoder.h:938
KeyFrameCfg kf_cfg
Definition encoder.h:933
enum aom_enc_pass pass
Definition encoder.h:1041
AlgoCfg algo_cfg
Definition encoder.h:928
aom_fixed_buf_t twopass_stats_in
Definition encoder.h:955
Structure to hold search parameter per restoration unit and intermediate buffer of Wiener filter used...
Definition encoder.h:1673
int16_t * dgd_avg
Definition encoder.h:1683
Structure to hold data corresponding to an encoded frame.
Definition encoder.h:2515
int64_t ts_frame_end
Definition encoder.h:2544
int pop_lookahead
Definition encoder.h:2559
int64_t ts_frame_start
Definition encoder.h:2539
unsigned char * cx_data
Definition encoder.h:2519
size_t cx_data_sz
Definition encoder.h:2524
int flush
Definition encoder.h:2549
int frame_display_order_hint
Definition encoder.h:2564
unsigned int lib_flags
Definition encoder.h:2534
size_t frame_size
Definition encoder.h:2529
const aom_rational64_t * timestamp_ratio
Definition encoder.h:2554
Top level encoder structure.
Definition encoder.h:2866
double * ext_rate_distribution
Definition encoder.h:3543
int do_update_frame_probs_warp[10]
Definition encoder.h:3252
uint8_t * consec_zero_mv
Definition encoder.h:3453
int do_update_frame_probs_obmc[10]
Definition encoder.h:3247
struct aom_denoise_and_model_t * denoise_and_model
Definition encoder.h:3320
int skip_tpl_setup_stats
Definition encoder.h:2974
int frames_since_last_update
Definition encoder.h:3627
int * mb_delta_q
Definition encoder.h:3563
int vaq_refresh
Definition encoder.h:3222
FRAME_TYPE last_frame_type
Definition encoder.h:3418
YV12_BUFFER_CONFIG * unscaled_source
Definition encoder.h:2942
int palette_pixel_num
Definition encoder.h:3656
CYCLIC_REFRESH * cyclic_refresh
Definition encoder.h:3114
RATE_CONTROL rc
Definition encoder.h:3073
int deltaq_used
Definition encoder.h:3376
ActiveMap active_map
Definition encoder.h:3119
bool frame_size_related_setup_done
Definition encoder.h:3185
WeberStats * mb_weber_stats
Definition encoder.h:3531
bool refresh_idx_available
Definition encoder.h:3496
TokenInfo token_info
Definition encoder.h:3217
int64_t ambient_err
Definition encoder.h:3042
aom_film_grain_table_t * film_grain_table
Definition encoder.h:3313
int ref_refresh_index
Definition encoder.h:3490
size_t available_bs_size
Definition encoder.h:3474
SPEED_FEATURES sf
Definition encoder.h:3093
TRELLIS_OPT_TYPE optimize_seg_arr[8]
Definition encoder.h:2920
ExtPartController ext_part_controller
Definition encoder.h:3480
FILE * second_pass_log_stream
Definition encoder.h:3597
double * ssim_rdmult_scaling_factors
Definition encoder.h:3388
RD_OPT rd
Definition encoder.h:3047
int data_alloc_height
Definition encoder.h:3171
int num_tg
Definition encoder.h:3423
WinnerModeParams winner_mode_params
Definition encoder.h:3063
ExternalFlags ext_flags
Definition encoder.h:3025
bool alloc_pyramid
Definition encoder.h:3638
EncSegmentationInfo enc_seg
Definition encoder.h:3109
MotionVectorSearchParams mv_search_params
Definition encoder.h:3098
int use_screen_content_tools
Definition encoder.h:3335
int do_update_frame_probs_interpfilter[10]
Definition encoder.h:3257
CODING_CONTEXT coding_context
Definition encoder.h:3053
TemporalFilterCtx tf_ctx
Definition encoder.h:2989
ForceIntegerMVInfo force_intpel_info
Definition encoder.h:2999
GlobalMotionInfo gm_info
Definition encoder.h:3058
int consec_zero_mv_alloc_size
Definition encoder.h:3458
CoeffBufferPool coeff_buffer_pool
Definition encoder.h:2904
FRAME_INDEX_SET frame_index_set
Definition encoder.h:3157
int ref_frame_flags
Definition encoder.h:3083
RefCntBuffer * scaled_ref_buf[INTER_REFS_PER_FRAME]
Definition encoder.h:3005
unsigned char gf_frame_index
Definition encoder.h:3124
AV1EncoderConfig oxcf
Definition encoder.h:2914
AV1_COMMON common
Definition encoder.h:2909
AV1LrStruct lr_ctxt
Definition encoder.h:3303
bool do_frame_data_update
Definition encoder.h:3519
CdefSearchCtx * cdef_search_ctx
Definition encoder.h:2994
int data_alloc_width
Definition encoder.h:3164
int do_update_frame_probs_txtype[10]
Definition encoder.h:3242
FRAME_COUNTS counts
Definition encoder.h:2886
COMPRESSOR_STAGE compressor_stage
Definition encoder.h:3412
int intrabc_used
Definition encoder.h:3293
int num_frame_recode
Definition encoder.h:3232
int rt_reduce_num_ref_buffers
Definition encoder.h:3020
RefreshFrameInfo refresh_frame
Definition encoder.h:3015
int prune_ref_frame_mask
Definition encoder.h:3298
YV12_BUFFER_CONFIG * unscaled_last_source
Definition encoder.h:2952
THIRD_PASS_DEC_CTX * third_pass_ctx
Definition encoder.h:3592
int all_one_sided_refs
Definition encoder.h:3104
MultiThreadInfo mt_info
Definition encoder.h:3281
VarBasedPartitionInfo vbp_info
Definition encoder.h:3227
int scaled_last_source_available
Definition encoder.h:3662
YV12_BUFFER_CONFIG * last_source
Definition encoder.h:2936
int existing_fb_idx_to_show
Definition encoder.h:3288
YV12_BUFFER_CONFIG * unfiltered_source
Definition encoder.h:2963
unsigned int zeromv_skip_thresh_exit_part[BLOCK_SIZES_ALL]
Definition encoder.h:3632
FRAME_INFO frame_info
Definition encoder.h:3152
int last_coded_height
Definition encoder.h:3197
int frame_header_count
Definition encoder.h:3371
int droppable
Definition encoder.h:3147
RefCntBuffer * last_show_frame_buf
Definition encoder.h:3010
aom_superres_mode superres_mode
Definition encoder.h:3430
MBMIExtFrameBufferInfo mbmi_ext_info
Definition encoder.h:2891
AV1LrPickStruct pick_lr_ctxt
Definition encoder.h:3308
double new_framerate
Definition encoder.h:3271
AV1_PRIMARY * ppi
Definition encoder.h:2870
uint64_t * src_sad_blk_64x64
Definition encoder.h:3602
int64_t norm_wiener_variance
Definition encoder.h:3558
double * tpl_rdmult_scaling_factors
Definition encoder.h:2984
int sb_counter
Definition encoder.h:3469
int last_coded_width
Definition encoder.h:3191
TileDataEnc * tile_data
Definition encoder.h:3208
int is_screen_content_type
Definition encoder.h:3343
BLOCK_SIZE weber_bsize
Definition encoder.h:3553
InterpSearchFlags interp_search_flags
Definition encoder.h:3326
TimeStamps time_stamps
Definition encoder.h:3068
int ref_idx_to_skip
Definition encoder.h:3503
YV12_BUFFER_CONFIG orig_source
Definition encoder.h:2969
FirstPassData firstpass_data
Definition encoder.h:3435
double framerate
Definition encoder.h:3078
int speed
Definition encoder.h:3088
BLOCK_SIZE fp_block_size
Definition encoder.h:3463
int use_ducky_encode
Definition encoder.h:3615
YV12_BUFFER_CONFIG trial_frame_rst
Definition encoder.h:3037
bool is_dropped_frame
Definition encoder.h:3568
ThreadData td
Definition encoder.h:2881
ResizePendingParams resize_pending_params
Definition encoder.h:3202
YV12_BUFFER_CONFIG scaled_source
Definition encoder.h:2947
int do_update_vbr_bits_off_target_fast
Definition encoder.h:3276
YV12_BUFFER_CONFIG last_frame_uf
Definition encoder.h:3031
EncQuantDequantParams enc_quant_dequant_params
Definition encoder.h:2876
RefFrameDistanceInfo ref_frame_dist_info
Definition encoder.h:3381
int * prep_rate_estimates
Definition encoder.h:3537
DuckyEncodeInfo ducky_encode_info
Definition encoder.h:3621
double ext_rate_scale
Definition encoder.h:3548
int initial_mbs
Definition encoder.h:3179
uint64_t rec_sse
Definition encoder.h:3609
YV12_BUFFER_CONFIG scaled_last_source
Definition encoder.h:2957
MV_STATS mv_stats
Definition encoder.h:3486
FrameProbInfo frame_new_probs[10]
Definition encoder.h:3237
YV12_BUFFER_CONFIG * source
Definition encoder.h:2927
int allocated_tiles
Definition encoder.h:3212
SVC svc
Definition encoder.h:3407
CB_COEFF_BUFFER * coeff_buffer_base
Definition encoder.h:2898
NOISE_ESTIMATE noise_estimate
Definition encoder.h:3440
TWO_PASS_FRAME twopass_frame
Definition encoder.h:3587
Top level primary encoder structure.
Definition encoder.h:2570
int num_fp_contexts
Definition encoder.h:2627
AV1EncRowMultiThreadSync intra_row_mt_sync
Definition encoder.h:2860
bool buffer_removal_time_present
Definition encoder.h:2741
int valid_gm_model_found[FRAME_UPDATE_TYPES]
Definition encoder.h:2849
struct aom_codec_pkt_list * output_pkt_list
Definition encoder.h:2671
int filter_level[2]
Definition encoder.h:2632
SequenceHeader seq_params
Definition encoder.h:2731
MV_STATS mv_stats
Definition encoder.h:2779
struct AV1_COMP * cpi
Definition encoder.h:2648
AV1LevelParams level_params
Definition encoder.h:2701
int internal_altref_allowed
Definition encoder.h:2676
RTC_REF rtc_ref
Definition encoder.h:2854
int b_calculate_psnr
Definition encoder.h:2706
PrimaryMultiThreadInfo p_mt_info
Definition encoder.h:2836
TEMPORAL_FILTER_INFO tf_info
Definition encoder.h:2726
TWO_PASS twopass
Definition encoder.h:2716
int frames_left
Definition encoder.h:2711
int64_t ts_start_last_show_frame
Definition encoder.h:2617
PRIMARY_RATE_CONTROL p_rc
Definition encoder.h:2721
int lap_enabled
Definition encoder.h:2696
FrameProbInfo frame_probs
Definition encoder.h:2841
int show_existing_alt_ref
Definition encoder.h:2681
int fb_of_context_type[REF_FRAMES]
Definition encoder.h:2831
int use_svc
Definition encoder.h:2736
double * tpl_sb_rdmult_scaling_factors
Definition encoder.h:2769
int filter_level_v
Definition encoder.h:2642
int filter_level_u
Definition encoder.h:2637
struct AV1_COMP * cpi_lap
Definition encoder.h:2653
struct AV1_COMP * parallel_cpi[4]
Definition encoder.h:2574
int64_t ts_end_last_show_frame
Definition encoder.h:2622
struct lookahead_ctx * lookahead
Definition encoder.h:2658
GF_STATE gf_state
Definition encoder.h:2691
aom_variance_fn_ptr_t fn_ptr[BLOCK_SIZES_ALL]
Definition encoder.h:2763
GF_GROUP gf_group
Definition encoder.h:2686
struct AV1_COMP_DATA parallel_frames_data[4 - 1]
Definition encoder.h:2580
TplParams tpl_data
Definition encoder.h:2774
unsigned int number_temporal_layers
Definition encoder.h:2746
unsigned int number_spatial_layers
Definition encoder.h:2751
int seq_params_locked
Definition encoder.h:2665
struct aom_internal_error_info error
Definition encoder.h:2756
Algorithm configuration parameters.
Definition encoder.h:810
int disable_trellis_quant
Definition encoder.h:826
int sharpness
Definition encoder.h:817
bool skip_postproc_filtering
Definition encoder.h:870
int arnr_max_frames
Definition encoder.h:831
bool enable_tpl_model
Definition encoder.h:849
LOOPFILTER_CONTROL loopfilter_control
Definition encoder.h:864
uint8_t cdf_update_mode
Definition encoder.h:844
bool enable_overlay
Definition encoder.h:855
int arnr_strength
Definition encoder.h:836
Stores the transforms coefficients for the whole superblock.
Definition block.h:206
The stucture of CYCLIC_REFRESH.
Definition aq_cyclicrefresh.h:36
Parameters related to CDEF.
Definition av1_common_int.h:198
int cdef_bits
Number of CDEF strength values in bits.
Definition av1_common_int.h:220
int cdef_uv_strengths[16]
CDEF strength values for chroma.
Definition av1_common_int.h:218
int cdef_strengths[16]
CDEF strength values for luma.
Definition av1_common_int.h:216
int nb_cdef_strengths
Number of CDEF strength values.
Definition av1_common_int.h:214
Definition encoder.h:2421
uint8_t * entropy_ctx
Definition encoder.h:2433
tran_low_t * tcoeff
Definition encoder.h:2425
uint16_t * eobs
Definition encoder.h:2429
Params related to MB_MODE_INFO arrays and related info.
Definition av1_common_int.h:503
BLOCK_SIZE mi_alloc_bsize
Definition av1_common_int.h:552
int cols
Definition av1_common_int.h:430
unsigned int large_scale
Definition av1_common_int.h:490
Encoder flags for compound prediction modes.
Definition encoder.h:396
bool enable_masked_comp
Definition encoder.h:405
bool enable_diff_wtd_comp
Definition encoder.h:413
bool enable_smooth_interintra
Definition encoder.h:409
bool enable_interintra_wedge
Definition encoder.h:421
bool enable_interinter_wedge
Definition encoder.h:417
bool enable_dist_wtd_comp
Definition encoder.h:400
Contains buffers used by av1_compound_type_rd()
Definition block.h:366
Segmentation related information for the current frame.
Definition encoder.h:2384
uint8_t * map
Definition encoder.h:2390
bool has_lossless_segment
Definition encoder.h:2396
contains per-frame encoding parameters decided upon by av1_encode_strategy() and passed down to av1_e...
Definition encoder.h:3680
int error_resilient_mode
Definition encoder.h:3684
int remapped_ref_idx[REF_FRAMES]
Definition encoder.h:3715
int ref_frame_flags
Definition encoder.h:3710
int speed
Definition encoder.h:3726
FRAME_TYPE frame_type
Definition encoder.h:3688
int show_frame
Definition encoder.h:3698
RefreshFrameInfo refresh_frame
Definition encoder.h:3721
Frame refresh flags set by the external interface.
Definition encoder.h:2251
bool golden_frame
Definition encoder.h:2253
bool bwd_ref_frame
Definition encoder.h:2254
bool update_pending
Definition encoder.h:2260
bool last_frame
Definition encoder.h:2252
bool alt_ref_frame
Definition encoder.h:2256
bool alt2_ref_frame
Definition encoder.h:2255
Flags signalled by the external interface at frame level.
Definition encoder.h:2266
bool use_primary_ref_none
Definition encoder.h:2307
bool use_ref_frame_mvs
Definition encoder.h:2291
ExtRefreshFrameFlagsInfo refresh_frame
Definition encoder.h:2275
int ref_frame_flags
Definition encoder.h:2270
bool use_error_resilient
Definition encoder.h:2296
bool use_s_frame
Definition encoder.h:2301
bool refresh_frame_context
Definition encoder.h:2280
bool refresh_frame_context_pending
Definition encoder.h:2286
Frame level features.
Definition av1_common_int.h:360
bool allow_screen_content_tools
Definition av1_common_int.h:377
bool allow_intrabc
Definition av1_common_int.h:378
bool coded_lossless
Definition av1_common_int.h:387
bool error_resilient_mode
Definition av1_common_int.h:402
bool all_lossless
Definition av1_common_int.h:391
Encoder info used for decision on forcing integer motion vectors.
Definition encoder.h:1845
int rate_size
Definition encoder.h:1858
int rate_index
Definition encoder.h:1854
Encoder-side probabilities for pruning of various AV1 tools.
Definition encoder.h:1104
Data related to the current GF/ARF group and the individual frames within the group.
Definition firstpass.h:354
Parameters related to global motion search.
Definition encoder.h:2054
bool search_done
Definition encoder.h:2058
int segment_map_h
Definition encoder.h:2087
int segment_map_w
Definition encoder.h:2086
Flags related to interpolation filter search.
Definition encoder.h:2094
int default_interp_skip_flags
Definition encoder.h:2099
uint16_t interp_filter_search_mask
Definition encoder.h:2103
Holds mv costs for intrabc.
Definition block.h:789
Encoder flags for intra prediction.
Definition encoder.h:299
bool enable_diagonal_intra
Definition encoder.h:329
bool enable_smooth_intra
Definition encoder.h:312
bool auto_intra_tools_off
Definition encoder.h:348
bool enable_filter_intra
Definition encoder.h:308
bool enable_directional_intra
Definition encoder.h:324
bool enable_paeth_intra
Definition encoder.h:316
bool enable_intra_edge_filter
Definition encoder.h:303
bool enable_cfl_intra
Definition encoder.h:320
bool enable_angle_delta
Definition encoder.h:334
Encoder config related to the coding of key frames.
Definition encoder.h:463
int key_freq_max
Definition encoder.h:472
int sframe_mode
Definition encoder.h:490
bool auto_key
Definition encoder.h:495
bool enable_intrabc
Definition encoder.h:515
int sframe_dist
Definition encoder.h:483
bool enable_sframe
Definition encoder.h:510
int enable_keyframe_filtering
Definition encoder.h:477
int fwd_kf_dist
Definition encoder.h:500
int key_freq_min
Definition encoder.h:467
bool fwd_kf_enabled
Definition encoder.h:505
Buffer to store mode information at mi_alloc_bsize (4x4 or 8x8) level.
Definition encoder.h:1891
int alloc_size
Definition encoder.h:1900
int stride
Definition encoder.h:1904
MB_MODE_INFO_EXT_FRAME * frame_base
Definition encoder.h:1896
Stores best extended mode information at frame level.
Definition block.h:242
Stores the prediction/txfm mode of the current coding block.
Definition blockd.h:222
Parameters for motion vector search process.
Definition encoder.h:2109
int max_mv_magnitude
Definition encoder.h:2115
fractional_mv_step_fp * find_fractional_mv_step
Definition encoder.h:2129
int mv_step_param
Definition encoder.h:2120
Encoder parameters related to multi-threading.
Definition encoder.h:1736
RestoreStateBuffers restore_state_buf
Definition encoder.h:1823
AV1CdefWorkerData * cdef_worker
Definition encoder.h:1818
AV1LrSync lr_row_sync
Definition encoder.h:1793
struct EncWorkerData * tile_thr_data
Definition encoder.h:1756
AV1TplRowMultiThreadInfo tpl_row_mt
Definition encoder.h:1783
AV1EncPackBSSync pack_bs_sync
Definition encoder.h:1798
AV1EncRowMultiThreadInfo enc_row_mt
Definition encoder.h:1772
AV1LfSync lf_row_sync
Definition encoder.h:1788
AV1CdefSync cdef_sync
Definition encoder.h:1813
int num_mod_workers[NUM_MT_MODULES]
Definition encoder.h:1745
AV1EncAllIntraMultiThreadInfo intra_mt
Definition encoder.h:1778
int num_workers
Definition encoder.h:1740
int pipeline_lpf_mt_with_enc
Definition encoder.h:1829
AVxWorker * workers
Definition encoder.h:1750
bool pack_bs_mt_enabled
Definition encoder.h:1767
bool row_mt_enabled
Definition encoder.h:1762
AV1TemporalFilterSync tf_sync
Definition encoder.h:1808
AV1GlobalMotionSync gm_sync
Definition encoder.h:1803
Holds mv costs for encoding and motion search.
Definition block.h:758
Contains buffers used to speed up rdopt for obmc.
Definition block.h:329
Contains color maps used in palette mode.
Definition block.h:354
Primary Rate Control parameters and status.
Definition ratectrl.h:291
Encoder config for coding block partitioning.
Definition encoder.h:271
bool enable_rect_partitions
Definition encoder.h:275
bool enable_1to4_partitions
Definition encoder.h:283
BLOCK_SIZE max_partition_size
Definition encoder.h:293
bool enable_ab_partitions
Definition encoder.h:279
BLOCK_SIZE min_partition_size
Definition encoder.h:288
Primary Encoder parameters related to multi-threading.
Definition encoder.h:1689
struct EncWorkerData * tile_thr_data
Definition encoder.h:1709
AV1CdefWorkerData * cdef_worker
Definition encoder.h:1714
int num_workers
Definition encoder.h:1693
int prev_num_enc_workers
Definition encoder.h:1730
AVxWorker * workers
Definition encoder.h:1703
AVxWorker * p_workers[4]
Definition encoder.h:1720
int p_num_workers
Definition encoder.h:1725
int num_mod_workers[NUM_MT_MODULES]
Definition encoder.h:1698
Rate Control parameters and status.
Definition ratectrl.h:134
Encoder rate control configuration parameters.
Definition encoder.h:521
int worst_allowed_q
Definition encoder.h:591
int over_shoot_pct
Definition encoder.h:586
unsigned int max_intra_bitrate_pct
Definition encoder.h:556
int drop_frames_water_mark
Definition encoder.h:574
int vbrmax_section
Definition encoder.h:622
int64_t maximum_buffer_size_ms
Definition encoder.h:540
unsigned int vbr_corpus_complexity_lap
Definition encoder.h:551
unsigned int min_cr
Definition encoder.h:570
int vbrbias
Definition encoder.h:612
unsigned int gf_cbr_boost_pct
Definition encoder.h:565
int vbrmin_section
Definition encoder.h:617
enum aom_rc_mode mode
Definition encoder.h:605
unsigned int max_inter_bitrate_pct
Definition encoder.h:561
int64_t starting_buffer_level_ms
Definition encoder.h:530
int best_allowed_q
Definition encoder.h:596
int under_shoot_pct
Definition encoder.h:580
int64_t target_bandwidth
Definition encoder.h:545
int64_t optimal_buffer_level_ms
Definition encoder.h:535
int cq_level
Definition encoder.h:600
Refrence frame distance related variables.
Definition encoder.h:2168
int8_t nearest_past_ref
Definition encoder.h:2176
int8_t nearest_future_ref
Definition encoder.h:2180
Refresh frame flags for different type of frames.
Definition encoder.h:2147
bool bwd_ref_frame
Definition encoder.h:2149
bool golden_frame
Definition encoder.h:2148
bool alt_ref_frame
Definition encoder.h:2150
Encoder config related to resize.
Definition encoder.h:251
uint8_t resize_scale_denominator
Definition encoder.h:260
uint8_t resize_kf_scale_denominator
Definition encoder.h:265
RESIZE_MODE resize_mode
Definition encoder.h:255
Desired dimensions for an externally triggered resize.
Definition encoder.h:2160
int width
Definition encoder.h:2161
int height
Definition encoder.h:2162
Parameters related to restoration types.
Definition encoder.h:1650
WienerInfo wiener
Definition encoder.h:1654
SgrprojInfo sgrproj
Definition encoder.h:1659
Parameters related to Restoration Info.
Definition restoration.h:246
RestorationType frame_restoration_type
Definition restoration.h:250
Buffers to be backed up during parallel encode set to be restored later.
Definition encoder.h:1625
int32_t * rst_tmpbuf
Definition encoder.h:1639
RestorationLineBuffers * rlbs
Definition encoder.h:1644
uint16_t * cdef_colbuf[3]
Definition encoder.h:1634
uint16_t * cdef_srcbuf
Definition encoder.h:1629
Top level speed vs quality trade off data struture.
Definition speed_features.h:1912
MV_SPEED_FEATURES mv_sf
Definition speed_features.h:1941
LOOP_FILTER_SPEED_FEATURES lpf_sf
Definition speed_features.h:1976
TX_SPEED_FEATURES tx_sf
Definition speed_features.h:1961
REAL_TIME_SPEED_FEATURES rt_sf
Definition speed_features.h:1981
The stucture of SVC.
Definition svc_layercontext.h:88
Parameters related to Sgrproj Filter.
Definition blockd.h:507
Encoder config related to frame super-resolution.
Definition encoder.h:427
uint8_t superres_kf_scale_denominator
Definition encoder.h:449
aom_superres_mode superres_mode
Definition encoder.h:453
int superres_kf_qthresh
Definition encoder.h:437
bool enable_superres
Definition encoder.h:457
uint8_t superres_scale_denominator
Definition encoder.h:443
int superres_qthresh
Definition encoder.h:432
Temporal filter info for a gop.
Definition temporal_filter.h:161
Frame level Two pass status and control data.
Definition firstpass.h:473
Two pass status and control data.
Definition firstpass.h:431
Parameters related to temporal filtering.
Definition temporal_filter.h:98
Frame time stamps.
Definition encoder.h:2402
int64_t prev_ts_start
Definition encoder.h:2406
int64_t first_ts_start
Definition encoder.h:2414
int64_t prev_ts_end
Definition encoder.h:2410
Params related to temporal dependency model.
Definition tpl_model.h:165
Encoder flags for transform sizes and types.
Definition encoder.h:354
bool enable_tx64
Definition encoder.h:358
bool use_inter_dct_only
Definition encoder.h:381
bool enable_flip_idtx
Definition encoder.h:362
bool use_intra_default_tx_only
Definition encoder.h:386
bool use_intra_dct_only
Definition encoder.h:376
bool enable_rect_tx
Definition encoder.h:366
bool reduced_tx_type_set
Definition encoder.h:371
bool enable_tx_size_search
Definition encoder.h:390
Thresholds for variance based partitioning.
Definition encoder.h:1355
int64_t threshold_minmax
Definition encoder.h:1370
Parameters related to Wiener Filter.
Definition blockd.h:494
Parameters used for winner mode processing.
Definition encoder.h:2198
Generic fixed size buffer structure.
Definition aom_encoder.h:86
Struct used to hold inter mode data for fast tx search.
Definition encoder.h:1260
RD_STATS rd_cost_arr[MAX_INTER_MODES]
Definition encoder.h:1289
int64_t est_rd_arr[MAX_INTER_MODES]
Definition encoder.h:1281
int64_t sse_arr[MAX_INTER_MODES]
Definition encoder.h:1277
RD_STATS rd_cost_y_arr[MAX_INTER_MODES]
Definition encoder.h:1293
RD_STATS rd_cost_uv_arr[MAX_INTER_MODES]
Definition encoder.h:1297
MB_MODE_INFO mbmi_arr[MAX_INTER_MODES]
Definition encoder.h:1269
RdIdxPair rd_idx_pair_arr[MAX_INTER_MODES]
Definition encoder.h:1285
int mode_rate_arr[MAX_INTER_MODES]
Definition encoder.h:1273
int num
Definition encoder.h:1265
Encoder's parameters related to the current coding block.
Definition block.h:878
MB_MODE_INFO_EXT_FRAME * mbmi_ext_frame
Finalized mbmi_ext for the whole frame.
Definition block.h:910
Variables related to current coding block.
Definition blockd.h:570
const struct scale_factors * block_ref_scale_factors[2]
Definition blockd.h:687
YV12 frame buffer data structure.
Definition yv12config.h:46