12#ifndef AOM_AV1_ENCODER_SPEED_FEATURES_H_
13#define AOM_AV1_ENCODER_SPEED_FEATURES_H_
16#include "av1/encoder/enc_enums.h"
17#include "av1/encoder/mcomp.h"
18#include "av1/encoder/encodemb.h"
27#define MAX_MESH_STEP 4
29typedef struct MESH_PATTERN {
36 GM_REDUCED_REF_SEARCH_SKIP_L2_L3,
37 GM_REDUCED_REF_SEARCH_SKIP_L2_L3_ARF2,
39} UENUM1BYTE(GM_SEARCH_TYPE);
42 DIST_WTD_COMP_ENABLED,
43 DIST_WTD_COMP_SKIP_MV_SEARCH,
44 DIST_WTD_COMP_DISABLED,
45} UENUM1BYTE(DIST_WTD_COMP_FLAG);
48 INTRA_ALL = (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED) | (1 << D45_PRED) |
49 (1 << D135_PRED) | (1 << D113_PRED) | (1 << D157_PRED) |
50 (1 << D203_PRED) | (1 << D67_PRED) | (1 << SMOOTH_PRED) |
51 (1 << SMOOTH_V_PRED) | (1 << SMOOTH_H_PRED) | (1 << PAETH_PRED),
53 (1 << UV_DC_PRED) | (1 << UV_V_PRED) | (1 << UV_H_PRED) |
54 (1 << UV_D45_PRED) | (1 << UV_D135_PRED) | (1 << UV_D113_PRED) |
55 (1 << UV_D157_PRED) | (1 << UV_D203_PRED) | (1 << UV_D67_PRED) |
56 (1 << UV_SMOOTH_PRED) | (1 << UV_SMOOTH_V_PRED) |
57 (1 << UV_SMOOTH_H_PRED) | (1 << UV_PAETH_PRED) | (1 << UV_CFL_PRED),
58 UV_INTRA_DC = (1 << UV_DC_PRED),
59 UV_INTRA_DC_CFL = (1 << UV_DC_PRED) | (1 << UV_CFL_PRED),
60 UV_INTRA_DC_TM = (1 << UV_DC_PRED) | (1 << UV_PAETH_PRED),
61 UV_INTRA_DC_PAETH_CFL =
62 (1 << UV_DC_PRED) | (1 << UV_PAETH_PRED) | (1 << UV_CFL_PRED),
63 UV_INTRA_DC_H_V = (1 << UV_DC_PRED) | (1 << UV_V_PRED) | (1 << UV_H_PRED),
64 UV_INTRA_DC_H_V_CFL = (1 << UV_DC_PRED) | (1 << UV_V_PRED) |
65 (1 << UV_H_PRED) | (1 << UV_CFL_PRED),
66 UV_INTRA_DC_PAETH_H_V = (1 << UV_DC_PRED) | (1 << UV_PAETH_PRED) |
67 (1 << UV_V_PRED) | (1 << UV_H_PRED),
68 UV_INTRA_DC_PAETH_H_V_CFL = (1 << UV_DC_PRED) | (1 << UV_PAETH_PRED) |
69 (1 << UV_V_PRED) | (1 << UV_H_PRED) |
71 INTRA_DC = (1 << DC_PRED),
72 INTRA_DC_TM = (1 << DC_PRED) | (1 << PAETH_PRED),
73 INTRA_DC_H_V = (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED),
75 (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED) | (1 << SMOOTH_PRED),
77 (1 << DC_PRED) | (1 << PAETH_PRED) | (1 << V_PRED) | (1 << H_PRED)
81 INTER_ALL = (1 << NEARESTMV) | (1 << NEARMV) | (1 << GLOBALMV) |
82 (1 << NEWMV) | (1 << NEAREST_NEARESTMV) | (1 << NEAR_NEARMV) |
83 (1 << NEW_NEWMV) | (1 << NEAREST_NEWMV) | (1 << NEAR_NEWMV) |
84 (1 << NEW_NEARMV) | (1 << NEW_NEARESTMV) | (1 << GLOBAL_GLOBALMV),
85 INTER_NEAREST_NEAR_ZERO = (1 << NEARESTMV) | (1 << NEARMV) | (1 << GLOBALMV) |
86 (1 << NEAREST_NEARESTMV) | (1 << GLOBAL_GLOBALMV) |
87 (1 << NEAREST_NEWMV) | (1 << NEW_NEARESTMV) |
88 (1 << NEW_NEARMV) | (1 << NEAR_NEWMV) |
93 DISABLE_ALL_INTER_SPLIT = (1 << THR_COMP_GA) | (1 << THR_COMP_LA) |
94 (1 << THR_ALTR) | (1 << THR_GOLD) | (1 << THR_LAST),
96 DISABLE_ALL_SPLIT = (1 << THR_INTRA) | DISABLE_ALL_INTER_SPLIT,
98 DISABLE_COMPOUND_SPLIT = (1 << THR_COMP_GA) | (1 << THR_COMP_LA),
100 LAST_AND_INTRA_SPLIT_ONLY = (1 << THR_COMP_GA) | (1 << THR_COMP_LA) |
101 (1 << THR_ALTR) | (1 << THR_GOLD)
113} UENUM1BYTE(DEV_SPEED_FEATURES);
126 ALLOW_RECODE_KFARFGF = 1,
131} UENUM1BYTE(RECODE_LOOP_TYPE);
135 SUBPEL_TREE_PRUNED = 1,
136 SUBPEL_TREE_PRUNED_MORE = 2,
137} UENUM1BYTE(SUBPEL_SEARCH_METHODS);
141 LPF_PICK_FROM_FULL_IMAGE,
143 LPF_PICK_FROM_FULL_IMAGE_NON_DUAL,
145 LPF_PICK_FROM_SUBIMAGE,
150} UENUM1BYTE(LPF_PICK_METHOD);
171 FLAG_EARLY_TERMINATE = 1 << 0,
174 FLAG_SKIP_COMP_BESTINTRA = 1 << 1,
177 FLAG_SKIP_INTRA_BESTINTER = 1 << 3,
181 FLAG_SKIP_INTRA_DIRMISMATCH = 1 << 4,
184 FLAG_SKIP_INTRA_LOWVAR = 1 << 5,
185} UENUM1BYTE(MODE_SEARCH_SKIP_LOGIC);
199} UENUM1BYTE(TX_TYPE_PRUNE_MODE);
207 FAST_DETECTION_MAXQ = 1,
208} UENUM1BYTE(OVERSHOOT_DETECTION_CBR);
214 MULTI_WINNER_MODE_OFF = 0,
217 MULTI_WINNER_MODE_FAST = 1,
221 MULTI_WINNER_MODE_DEFAULT = 2,
222} UENUM1BYTE(MULTI_WINNER_MODE_TYPE);
225 PRUNE_NEARMV_OFF = 0,
226 PRUNE_NEARMV_LEVEL1 = 1,
227 PRUNE_NEARMV_LEVEL2 = 2,
228 PRUNE_NEARMV_LEVEL3 = 3,
229 PRUNE_NEARMV_MAX = PRUNE_NEARMV_LEVEL3,
230} UENUM1BYTE(PRUNE_NEARMV_LEVEL);
233 TX_TYPE_PRUNE_MODE prune_2d_txfm_mode;
234 int fast_intra_tx_type_search;
241 int fast_inter_tx_type_prob_thresh;
245 int use_reduced_intra_txset;
249 int use_skip_flag_prediction;
252 int ml_tx_split_thresh;
259 int prune_tx_type_using_stats;
261 int prune_tx_type_est_rd;
267 int winner_mode_tx_type_pruning;
280#if CONFIG_RT_ML_PARTITIONING
284} UENUM1BYTE(PARTITION_SEARCH_TYPE);
291} UENUM1BYTE(MAX_PART_PRED_MODE);
297} UENUM1BYTE(MV_PREC_LOGIC);
303} UENUM1BYTE(SUPERRES_AUTO_SEARCH_TYPE);
401typedef struct TPL_SPEED_FEATURES {
409 int gop_length_decision_method;
413 int prune_intra_modes;
415 int reduce_first_step_size;
422 int skip_alike_starting_mv;
425 SUBPEL_FORCE_STOP subpel_force_stop;
428 SEARCH_METHODS search_method;
431 int prune_starting_mv;
434 int disable_filtered_key_tpl;
437 int prune_ref_frames_in_tpl;
440 int allow_compound_pred;
443 int use_y_only_rate_distortion;
446typedef struct GLOBAL_MOTION_SPEED_FEATURES {
447 GM_SEARCH_TYPE gm_search_type;
452 int prune_ref_frame_for_gm_search;
457} GLOBAL_MOTION_SPEED_FEATURES;
459typedef struct PARTITION_SPEED_FEATURES {
460 PARTITION_SEARCH_TYPE partition_search_type;
463 BLOCK_SIZE fixed_partition_size;
468 int prune_ext_partition_types_search_level;
471 int prune_part4_search;
475 int ml_prune_partition;
480 int ml_early_term_after_part_split_level;
485 int less_rectangular_check_level;
488 BLOCK_SIZE use_square_partition_only_threshold;
493 MAX_PART_PRED_MODE auto_max_partition_based_on_simple_motion;
497 BLOCK_SIZE default_min_partition_size;
498 BLOCK_SIZE default_max_partition_size;
505 int adjust_var_based_rd_partitioning;
508 int64_t partition_search_breakout_dist_thr;
509 int partition_search_breakout_rate_thr;
512 int ml_partition_search_breakout_thresh[PARTITION_BLOCK_SIZES];
516 int simple_motion_search_prune_agg;
520 int simple_motion_search_prune_rect;
527 int simple_motion_search_split;
531 int simple_motion_search_early_term_none;
537 int simple_motion_search_reduce_search_steps;
542 BLOCK_SIZE max_intra_bsize;
549 int intra_cnn_based_part_prune_level;
552 int ext_partition_eval_thresh;
555 int rect_partition_eval_thresh;
561 int prune_ext_part_using_split_info;
567 int prune_rectangular_split_based_on_qidx;
571 int early_term_after_none_split;
577 int ml_predict_breakout_level;
583 int prune_sub_8x8_partition_level;
587 int simple_motion_search_rect_split;
620 int reuse_prev_rd_results_for_part_ab;
624 int reuse_best_prediction_for_part_ab;
634 int use_best_rd_for_pruning;
642 int skip_non_sq_part_based_on_none;
643} PARTITION_SPEED_FEATURES;
645typedef struct MV_SPEED_FEATURES {
647 SEARCH_METHODS search_method;
653 int use_bsize_dependent_search_method;
657 int auto_mv_step_size;
663 SUBPEL_SEARCH_METHODS subpel_search_method;
666 int subpel_iters_per_step;
669 SUBPEL_FORCE_STOP subpel_force_stop;
672 SUBPEL_FORCE_STOP simple_motion_subpel_force_stop;
676 SUBPEL_SEARCH_TYPE use_accurate_subpel_search;
679 int exhaustive_searches_thresh;
682 MESH_PATTERN mesh_patterns[MAX_MESH_STEP];
685 MESH_PATTERN intrabc_mesh_patterns[MAX_MESH_STEP];
688 int reduce_search_range;
691 int prune_mesh_search;
694 int use_fullpel_costlist;
699 int obmc_full_pixel_search_level;
702 int full_pixel_search_level;
706 int use_downsampled_sad;
709 int disable_extensive_joint_motion_search;
715 int disable_second_mv;
718typedef struct INTER_MODE_SPEED_FEATURES {
725 int inter_mode_rd_model_estimation;
728 int txfm_rd_gate_level;
731 int reduce_inter_modes;
735 int adaptive_rd_thresh;
738 int prune_inter_modes_if_skippable;
743 int selective_ref_frame;
749 int prune_ref_frame_for_rect_partitions;
751 int alt_ref_search_fp;
765 int skip_newmv_in_drl;
771 int skip_repeated_ref_mv;
774 int perform_best_rd_based_gating_for_chroma;
778 int reuse_inter_intra_mode;
782 int prune_comp_type_by_model_rd;
786 int prune_comp_type_by_comp_avg;
794 int prune_comp_search_by_single_result;
800 BLOCK_SIZE comp_inter_joint_search_thresh;
805 int prune_mode_search_simple_translation;
810 int prune_compound_using_single_ref;
819 int prune_ext_comp_using_neighbors;
822 int skip_ext_comp_nearmv_mode;
829 int prune_comp_using_best_single_mode_ref;
832 int prune_nearest_near_mv_using_refmv_weight;
835 int prune_ref_mv_idx_search;
838 int disable_onesided_comp;
842 int prune_obmc_prob_thresh;
845 int prune_warped_prob_thresh;
848 unsigned int disable_interintra_wedge_var_thresh;
851 unsigned int disable_interinter_wedge_var_thresh;
854 int fast_interintra_wedge_search;
857 int fast_wedge_sign_estimate;
860 int disable_interinter_wedge_newmv_search;
863 DIST_WTD_COMP_FLAG use_dist_wtd_comp_flag;
877 int prune_inter_modes_based_on_tpl;
881 PRUNE_NEARMV_LEVEL prune_nearmv_using_neighbors;
886 int model_based_post_interp_filter_breakout;
891 int reuse_compound_type_decision;
894 int disable_masked_comp;
897 int enable_fast_compound_mode_search;
900 int reuse_mask_search_results;
903 int enable_fast_wedge_mask_search;
906 int inter_mode_txfm_breakout;
912 int limit_inter_mode_cands;
916 int limit_txfm_eval_per_mode;
917} INTER_MODE_SPEED_FEATURES;
919typedef struct INTERP_FILTER_SPEED_FEATURES {
922 int use_fast_interpolation_filter_search;
925 int disable_dual_filter;
930 int use_interp_filter;
934 int skip_sharp_interp_filter_search;
936 int cb_pred_filter_search;
939 int adaptive_interp_filter_search;
940} INTERP_FILTER_SPEED_FEATURES;
942typedef struct INTRA_MODE_SPEED_FEATURES {
945 int intra_y_mode_mask[TX_SIZES];
946 int intra_uv_mode_mask[TX_SIZES];
949 int skip_intra_in_interframe;
955 int intra_pruning_with_hog;
961 int chroma_intra_pruning_with_hog;
964 int disable_smooth_intra;
970 int prune_filter_intra_level;
979 int prune_palette_search_level;
999 int prune_luma_palette_size_search_level;
1005 int prune_chroma_modes_using_luma_winner;
1021 int cfl_search_range;
1026 int top_intra_model_count_allowed;
1038 int early_term_chroma_palette_size_search;
1042 int skip_filter_intra_in_inter_frames;
1043} INTRA_MODE_SPEED_FEATURES;
1045typedef struct TX_SPEED_FEATURES {
1049 int inter_tx_size_search_init_depth_sqr;
1050 int inter_tx_size_search_init_depth_rect;
1051 int intra_tx_size_search_init_depth_sqr;
1052 int intra_tx_size_search_init_depth_rect;
1056 int tx_size_search_lgr_block;
1058 TX_TYPE_SEARCH tx_type_search;
1068 int adaptive_txb_search_level;
1073 int model_based_prune_tx_search_level;
1079 int use_intra_txb_hash;
1083 int use_inter_txb_hash;
1086 int refine_fast_tx_search_results;
1091 int prune_tx_size_level;
1094typedef struct RD_CALC_SPEED_FEATURES {
1096 int simple_model_rd_from_var;
1104 int tx_domain_dist_level;
1107 int tx_domain_dist_thres_level;
1110 TRELLIS_OPT_TYPE optimize_coefficients;
1117 int perform_coeff_opt;
1118} RD_CALC_SPEED_FEATURES;
1120typedef struct WINNER_MODE_SPEED_FEATURES {
1123 int enable_winner_mode_for_coeff_opt;
1127 int enable_winner_mode_for_tx_size_srch;
1135 int tx_size_search_level;
1139 int enable_winner_mode_for_use_tx_domain_dist;
1142 MULTI_WINNER_MODE_TYPE multi_winner_mode_type;
1147 int motion_mode_for_winner_cand;
1152 int dc_blk_pred_level;
1153} WINNER_MODE_SPEED_FEATURES;
1155typedef struct LOOP_FILTER_SPEED_FEATURES {
1157 LPF_PICK_METHOD lpf_pick;
1161 int use_coarse_filter_level_search;
1169 int dual_sgr_penalty_level;
1172 int enable_sgr_ep_pruning;
1175 int disable_loop_restoration_chroma;
1178 int disable_loop_restoration_luma;
1184 int prune_wiener_based_on_src_var;
1191 int prune_sgr_based_on_wiener;
1194 int reduce_wiener_window_size;
1197 int disable_lr_filter;
1198} LOOP_FILTER_SPEED_FEATURES;
1200typedef struct REAL_TIME_SPEED_FEATURES {
1202 int check_intra_pred_nonrd;
1205 int skip_intra_pred_if_tx_skip;
1208 int estimate_motion_for_var_based_partition;
1214 int nonrd_check_partition_merge_mode;
1217 int nonrd_check_partition_split;
1222 unsigned int mode_search_skip_flags;
1228 int nonrd_prune_ref_frame_search;
1231 int use_nonrd_pick_mode;
1234 int use_nonrd_altref_frame;
1237 int use_golden_frame;
1240 int use_comp_ref_nonrd;
1244 int ref_frame_comp_nonrd[3];
1247 int use_real_time_ref_set;
1251 int short_circuit_low_temp_var;
1254 int use_modeled_non_rd_cost;
1257 int reuse_inter_pred_nonrd;
1260 int num_inter_modes_for_tx_search;
1263 int force_tx_search_off;
1266 int use_nonrd_filter_search;
1269 int use_simple_rd_model;
1272 int skip_interp_filter_search;
1277 int hybrid_intra_pickmode;
1280 int source_metrics_sb_nonrd;
1284 OVERSHOOT_DETECTION_CBR overshoot_detection_cbr;
1287 int check_scene_detection;
1290 int force_large_partition_blocks;
1293 int use_temporal_noise_estimate;
1299 int fullpel_search_step_param;
1303 int skip_loopfilter_non_reference;
1307 int intra_y_mode_bsize_mask_nrd[BLOCK_SIZES];
1310 int nonrd_agressive_skip;
1319 int force_large_partition_blocks_intra;
1322 int skip_tx_no_split_var_based_partition;
1323} REAL_TIME_SPEED_FEATURES;
static int prune_zero_mv_with_sse(const aom_variance_fn_ptr_t *fn_ptr, const MACROBLOCK *x, BLOCK_SIZE bsize, const HandleInterModeArgs *args)
Prunes ZeroMV Search Using Best NEWMV's SSE.
Definition rdopt.c:2464
void av1_set_speed_features_framesize_independent(struct AV1_COMP *cpi, int speed)
Frame size independent speed vs quality trade off flags.
void av1_set_speed_features_qindex_dependent(struct AV1_COMP *cpi, int speed)
Q index dependent speed vs quality trade off flags.
void av1_set_speed_features_framesize_dependent(struct AV1_COMP *cpi, int speed)
Frame size dependent speed vs quality trade off flags.
INTERNAL_COST_UPDATE_TYPE
This enum decides internally how often to update the entropy costs.
Definition speed_features.h:314
@ INTERNAL_COST_UPD_OFF
Definition speed_features.h:315
@ INTERNAL_COST_UPD_SBROW_SET
Definition speed_features.h:316
@ INTERNAL_COST_UPD_SBROW
Definition speed_features.h:317
@ INTERNAL_COST_UPD_SB
Definition speed_features.h:318
CDEF_PICK_METHOD
This enumeration defines a variety of CDEF pick methods.
Definition speed_features.h:156
@ CDEF_FAST_SEARCH_LVL2
Definition speed_features.h:159
@ CDEF_FAST_SEARCH_LVL1
Definition speed_features.h:158
@ CDEF_FULL_SEARCH
Definition speed_features.h:157
@ CDEF_PICK_FROM_Q
Definition speed_features.h:163
@ CDEF_FAST_SEARCH_LVL4
Definition speed_features.h:162
@ CDEF_FAST_SEARCH_LVL3
Definition speed_features.h:160
Definition speed_features.h:375
int disable_recon
Skips reconstruction by using source buffers for prediction.
Definition speed_features.h:392
int reduce_mv_step_param
Reduces the mv search window. By default, the initial search window is around MIN(MIN(dims),...
Definition speed_features.h:382
int skip_zeromv_motion_search
Skips the motion search centered on 0,0 mv.
Definition speed_features.h:397
int skip_motion_search_threshold
Skips the motion search when the zero mv has small sse.
Definition speed_features.h:387
Sequence/frame level speed vs quality features.
Definition speed_features.h:324
int second_alt_ref_filtering
Definition speed_features.h:369
int frame_parameter_update
Definition speed_features.h:326
MV_PREC_LOGIC high_precision_mv_usage
Definition speed_features.h:345
int disable_extra_sc_testing
Definition speed_features.h:364
int recode_tolerance
Definition speed_features.h:337
SUPERRES_AUTO_SEARCH_TYPE superres_auto_search_type
Definition speed_features.h:359
int static_segmentation
Definition speed_features.h:354
RECODE_LOOP_TYPE recode_loop
Definition speed_features.h:331
Top level speed vs quality trade off data struture.
Definition speed_features.h:1330
MV_SPEED_FEATURES mv_sf
Definition speed_features.h:1359
TPL_SPEED_FEATURES tpl_sf
Definition speed_features.h:1344
LOOP_FILTER_SPEED_FEATURES lpf_sf
Definition speed_features.h:1394
TX_SPEED_FEATURES tx_sf
Definition speed_features.h:1379
INTER_MODE_SPEED_FEATURES inter_sf
Definition speed_features.h:1364
RD_CALC_SPEED_FEATURES rd_sf
Definition speed_features.h:1384
PARTITION_SPEED_FEATURES part_sf
Definition speed_features.h:1354
GLOBAL_MOTION_SPEED_FEATURES gm_sf
Definition speed_features.h:1349
INTERP_FILTER_SPEED_FEATURES interp_sf
Definition speed_features.h:1369
FIRST_PASS_SPEED_FEATURES fp_sf
Definition speed_features.h:1339
INTRA_MODE_SPEED_FEATURES intra_sf
Definition speed_features.h:1374
WINNER_MODE_SPEED_FEATURES winner_mode_sf
Definition speed_features.h:1389
REAL_TIME_SPEED_FEATURES rt_sf
Definition speed_features.h:1399
HIGH_LEVEL_SPEED_FEATURES hl_sf
Definition speed_features.h:1334