| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| * Media Patent License 1.0 was not distributed with this source code in the |
| * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
| */ |
| |
| #include <limits.h> |
| |
| #include "av1/common/reconintra.h" |
| |
| #include "av1/encoder/encoder.h" |
| #include "av1/encoder/speed_features.h" |
| #include "av1/encoder/rdopt.h" |
| |
| #include "aom_dsp/aom_dsp_common.h" |
| |
| #define MAX_MESH_SPEED 5 // Max speed setting for mesh motion method |
| // Max speed setting for tx domain evaluation |
| #define MAX_TX_DOMAIN_EVAL_SPEED 5 |
| static MESH_PATTERN |
| good_quality_mesh_patterns[MAX_MESH_SPEED + 1][MAX_MESH_STEP] = { |
| { { 64, 8 }, { 28, 4 }, { 15, 1 }, { 7, 1 } }, |
| { { 64, 8 }, { 28, 4 }, { 15, 1 }, { 7, 1 } }, |
| { { 64, 8 }, { 14, 2 }, { 7, 1 }, { 7, 1 } }, |
| { { 64, 16 }, { 24, 8 }, { 12, 4 }, { 7, 1 } }, |
| { { 64, 16 }, { 24, 8 }, { 12, 4 }, { 7, 1 } }, |
| { { 64, 16 }, { 24, 8 }, { 12, 4 }, { 7, 1 } }, |
| }; |
| static unsigned char good_quality_max_mesh_pct[MAX_MESH_SPEED + 1] = { 50, 50, |
| 25, 15, |
| 5, 1 }; |
| |
| // TODO(huisu@google.com): These settings are pretty relaxed, tune them for |
| // each speed setting |
| static MESH_PATTERN intrabc_mesh_patterns[MAX_MESH_SPEED + 1][MAX_MESH_STEP] = { |
| { { 256, 1 }, { 256, 1 }, { 0, 0 }, { 0, 0 } }, |
| { { 256, 1 }, { 256, 1 }, { 0, 0 }, { 0, 0 } }, |
| { { 64, 1 }, { 64, 1 }, { 0, 0 }, { 0, 0 } }, |
| { { 64, 1 }, { 64, 1 }, { 0, 0 }, { 0, 0 } }, |
| { { 64, 4 }, { 16, 1 }, { 0, 0 }, { 0, 0 } }, |
| { { 64, 4 }, { 16, 1 }, { 0, 0 }, { 0, 0 } }, |
| }; |
| static uint8_t intrabc_max_mesh_pct[MAX_MESH_SPEED + 1] = { 100, 100, 100, |
| 25, 25, 10 }; |
| |
| // Threshold values to be used for pruning the txfm_domain_distortion |
| // based on block MSE |
| // Index 0: Default mode evaluation, Winner mode processing is not |
| // applicable (Eg : IntraBc). Index 1: Mode evaluation. |
| // Index 2: Winner mode evaluation. Index 1 and 2 are applicable when |
| // enable_winner_mode_for_use_tx_domain_dist speed feature is ON |
| // TODO(any): Experiment the threshold logic based on variance metric |
| static unsigned int tx_domain_dist_thresholds[3][MODE_EVAL_TYPES] = { |
| { UINT_MAX, UINT_MAX, UINT_MAX }, { 22026, 22026, 22026 }, { 0, 0, 0 } |
| }; |
| |
| // Transform domain distortion type to be used for default, mode and winner mode |
| // evaluation Index 0: Default mode evaluation, Winner mode processing is not |
| // applicable (Eg : IntraBc). Index 1: Mode evaluation. Index 2: Winner mode |
| // evaluation. Index 1 and 2 are applicable when |
| // enable_winner_mode_for_use_tx_domain_dist speed feature is ON |
| static unsigned int tx_domain_dist_types[3][MODE_EVAL_TYPES] = { { 0, 2, 0 }, |
| { 1, 2, 0 }, |
| { 2, 2, 0 } }; |
| |
| // Indicates number of winner simple translation modes to be used |
| static unsigned int num_winner_motion_modes[3] = { 0, 10, 3 }; |
| |
| // Threshold values to be used for disabling coeff RD-optimization |
| // based on block MSE |
| // TODO(any): Experiment the threshold logic based on variance metric |
| // Index 0: Default mode evaluation, Winner mode processing is not applicable |
| // (Eg : IntraBc) Index 1: Mode evaluation. Index 2: Winner mode evaluation. |
| // Index 1 and 2 are applicable when enable_winner_mode_for_coeff_opt speed |
| // feature is ON |
| static unsigned int coeff_opt_dist_thresholds[5][MODE_EVAL_TYPES] = { |
| { UINT_MAX, UINT_MAX, UINT_MAX }, |
| { 442413, 36314, UINT_MAX }, |
| { 162754, 36314, UINT_MAX }, |
| { 22026, 22026, UINT_MAX }, |
| { 22026, 22026, UINT_MAX } |
| }; |
| |
| // Transform size to be used for default, mode and winner mode evaluation |
| // Index 0: Default mode evaluation, Winner mode processing is not applicable |
| // (Eg : IntraBc) Index 1: Mode evaluation. Index 2: Winner mode evaluation. |
| // Index 1 and 2 are applicable when enable_winner_mode_for_tx_size_srch speed |
| // feature is ON |
| static TX_SIZE_SEARCH_METHOD tx_size_search_methods[3][MODE_EVAL_TYPES] = { |
| { USE_FULL_RD, USE_LARGESTALL, USE_FULL_RD }, |
| { USE_FAST_RD, USE_LARGESTALL, USE_FULL_RD }, |
| { USE_LARGESTALL, USE_LARGESTALL, USE_FULL_RD } |
| }; |
| |
| // Predict transform skip levels to be used for default, mode and winner mode |
| // evaluation. Index 0: Default mode evaluation, Winner mode processing is not |
| // applicable. Index 1: Mode evaluation, Index 2: Winner mode evaluation |
| // Values indicate the aggressiveness of skip flag prediction. |
| // 0 : no early skip prediction |
| // 1 : conservative early skip prediction using DCT_DCT |
| // 2 : early skip prediction based on SSE |
| static unsigned int predict_skip_levels[3][MODE_EVAL_TYPES] = { { 0, 0, 0 }, |
| { 1, 1, 1 }, |
| { 1, 2, 1 } }; |
| |
| // scaling values to be used for gating wedge/compound segment based on best |
| // approximate rd |
| static int comp_type_rd_threshold_mul[3] = { 1, 11, 12 }; |
| static int comp_type_rd_threshold_div[3] = { 3, 16, 16 }; |
| |
| // Intra only frames, golden frames (except alt ref overlays) and |
| // alt ref frames tend to be coded at a higher than ambient quality |
| static int frame_is_boosted(const AV1_COMP *cpi) { |
| return frame_is_kf_gf_arf(cpi); |
| } |
| |
| static BLOCK_SIZE dim_to_size(int dim) { |
| switch (dim) { |
| case 4: return BLOCK_4X4; |
| case 8: return BLOCK_8X8; |
| case 16: return BLOCK_16X16; |
| case 32: return BLOCK_32X32; |
| case 64: return BLOCK_64X64; |
| case 128: return BLOCK_128X128; |
| default: assert(0); return 0; |
| } |
| } |
| |
| static void set_good_speed_feature_framesize_dependent( |
| const AV1_COMP *const cpi, SPEED_FEATURES *const sf, int speed) { |
| const AV1_COMMON *const cm = &cpi->common; |
| const int is_720p_or_larger = AOMMIN(cm->width, cm->height) >= 720; |
| const int is_480p_or_larger = AOMMIN(cm->width, cm->height) >= 480; |
| const int is_4k_or_larger = AOMMIN(cm->width, cm->height) >= 2160; |
| |
| if (is_480p_or_larger) { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_128X128; |
| if (is_720p_or_larger) |
| sf->part_sf.auto_max_partition_based_on_simple_motion = ADAPT_PRED; |
| else |
| sf->part_sf.auto_max_partition_based_on_simple_motion = RELAXED_PRED; |
| } else { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_64X64; |
| sf->part_sf.auto_max_partition_based_on_simple_motion = DIRECT_PRED; |
| } |
| |
| if (is_4k_or_larger) { |
| sf->part_sf.default_min_partition_size = BLOCK_8X8; |
| } |
| |
| // TODO(huisu@google.com): train models for 720P and above. |
| if (!is_720p_or_larger) { |
| sf->part_sf.ml_partition_search_breakout_thresh[0] = 200; // BLOCK_8X8 |
| sf->part_sf.ml_partition_search_breakout_thresh[1] = 250; // BLOCK_16X16 |
| sf->part_sf.ml_partition_search_breakout_thresh[2] = 300; // BLOCK_32X32 |
| sf->part_sf.ml_partition_search_breakout_thresh[3] = 500; // BLOCK_64X64 |
| sf->part_sf.ml_partition_search_breakout_thresh[4] = -1; // BLOCK_128X128 |
| sf->part_sf.ml_early_term_after_part_split_level = 1; |
| } |
| |
| if (speed >= 1) { |
| if (is_720p_or_larger) { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_128X128; |
| } else if (is_480p_or_larger) { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_64X64; |
| } else { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_32X32; |
| } |
| |
| if (!is_720p_or_larger) { |
| sf->part_sf.ml_partition_search_breakout_thresh[0] = 200; // BLOCK_8X8 |
| sf->part_sf.ml_partition_search_breakout_thresh[1] = 250; // BLOCK_16X16 |
| sf->part_sf.ml_partition_search_breakout_thresh[2] = 300; // BLOCK_32X32 |
| sf->part_sf.ml_partition_search_breakout_thresh[3] = 300; // BLOCK_64X64 |
| sf->part_sf.ml_partition_search_breakout_thresh[4] = -1; // BLOCK_128X128 |
| } |
| sf->part_sf.ml_early_term_after_part_split_level = 2; |
| } |
| |
| if (speed >= 2) { |
| if (is_720p_or_larger) { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_64X64; |
| } else if (is_480p_or_larger) { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_32X32; |
| } else { |
| sf->part_sf.use_square_partition_only_threshold = BLOCK_32X32; |
| } |
| |
| if (is_720p_or_larger) { |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 24); |
| sf->part_sf.partition_search_breakout_rate_thr = 120; |
| } else { |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 22); |
| sf->part_sf.partition_search_breakout_rate_thr = 100; |
| } |
| |
| if (is_720p_or_larger) { |
| sf->inter_sf.prune_obmc_prob_thresh = 16; |
| } else { |
| sf->inter_sf.prune_obmc_prob_thresh = 8; |
| } |
| |
| if (is_480p_or_larger) { |
| sf->tx_sf.tx_type_search.prune_tx_type_using_stats = 1; |
| } |
| } |
| |
| if (speed >= 3) { |
| sf->part_sf.ml_early_term_after_part_split_level = 0; |
| |
| if (is_720p_or_larger) { |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 25); |
| sf->part_sf.partition_search_breakout_rate_thr = 200; |
| } else { |
| sf->part_sf.max_intra_bsize = BLOCK_32X32; |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 23); |
| sf->part_sf.partition_search_breakout_rate_thr = 120; |
| } |
| } |
| |
| if (speed >= 4) { |
| if (is_720p_or_larger) { |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 26); |
| } else { |
| sf->part_sf.partition_search_breakout_dist_thr = (1 << 24); |
| } |
| |
| if (is_480p_or_larger) { |
| sf->tx_sf.tx_type_search.prune_tx_type_using_stats = 2; |
| } |
| |
| sf->inter_sf.prune_obmc_prob_thresh = 16; |
| } |
| |
| if (speed >= 5) { |
| if (is_720p_or_larger) { |
| sf->inter_sf.prune_warped_prob_thresh = 16; |
| } else if (is_480p_or_larger) { |
| sf->inter_sf.prune_warped_prob_thresh = 8; |
| } |
| } |
| } |
| |
| static void set_rt_speed_feature_framesize_dependent(const AV1_COMP *const cpi, |
| SPEED_FEATURES *const sf, |
| int speed) { |
| const AV1_COMMON *const cm = &cpi->common; |
| const int is_720p_or_larger = AOMMIN(cm->width, cm->height) >= 720; |
| const int is_480p_or_larger = AOMMIN(cm->width, cm->height) >= 480; |
| |
| (void)is_720p_or_larger; // Not used so far |
| |
| if (!is_480p_or_larger) { |
| if (speed >= 8) { |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE; |
| |
| sf->rt_sf.estimate_motion_for_var_based_partition = 1; |
| } |
| } |
| } |
| |
| static void set_good_speed_features_framesize_independent( |
| const AV1_COMP *const cpi, SPEED_FEATURES *const sf, int speed) { |
| const AV1_COMMON *const cm = &cpi->common; |
| const GF_GROUP *const gf_group = &cpi->gf_group; |
| const int boosted = frame_is_boosted(cpi); |
| const int is_boosted_arf2_bwd_type = |
| boosted || gf_group->update_type[gf_group->index] == INTNL_ARF_UPDATE; |
| |
| // Speed 0 for all speed features that give neutral coding performance change. |
| sf->gm_sf.gm_disable_recode = 1; |
| sf->gm_sf.gm_search_type = GM_REDUCED_REF_SEARCH_SKIP_L2_L3; |
| |
| sf->part_sf.less_rectangular_check_level = 1; |
| sf->part_sf.ml_prune_4_partition = 1; |
| sf->part_sf.ml_prune_ab_partition = 1; |
| sf->part_sf.ml_prune_rect_partition = 1; |
| sf->part_sf.prune_ext_partition_types_search_level = 1; |
| sf->part_sf.simple_motion_search_prune_rect = 1; |
| |
| // TODO(any): Clean-up code related to hash_me in inter frames |
| sf->mv_sf.disable_hash_me = frame_is_intra_only(cm) ? 0 : 1; |
| |
| sf->inter_sf.disable_wedge_search_edge_thresh = 0; |
| sf->inter_sf.disable_wedge_search_var_thresh = 0; |
| // TODO(debargha): Test, tweak and turn on either 1 or 2 |
| sf->inter_sf.inter_mode_rd_model_estimation = 1; |
| sf->inter_sf.model_based_post_interp_filter_breakout = 1; |
| sf->inter_sf.prune_compound_using_single_ref = 1; |
| sf->inter_sf.prune_mode_search_simple_translation = 1; |
| sf->inter_sf.prune_motion_mode_level = 1; |
| sf->inter_sf.prune_ref_frame_for_rect_partitions = |
| (boosted || (cm->allow_screen_content_tools)) |
| ? 0 |
| : (is_boosted_arf2_bwd_type ? 1 : 2); |
| sf->inter_sf.prune_wedge_pred_diff_based = 1; |
| sf->inter_sf.reduce_inter_modes = 1; |
| sf->inter_sf.selective_ref_frame = 1; |
| sf->inter_sf.use_dist_wtd_comp_flag = DIST_WTD_COMP_SKIP_MV_SEARCH; |
| |
| sf->interp_sf.cb_pred_filter_search = 0; |
| sf->interp_sf.use_fast_interpolation_filter_search = 1; |
| |
| sf->intra_sf.intra_pruning_with_hog = 1; |
| sf->intra_sf.intra_pruning_with_hog_thresh = -1.2f; |
| |
| sf->tx_sf.adaptive_txb_search_level = 1; |
| sf->tx_sf.intra_tx_size_search_init_depth_sqr = 1; |
| sf->tx_sf.model_based_prune_tx_search_level = 1; |
| sf->tx_sf.tx_type_search.use_reduced_intra_txset = 1; |
| |
| sf->rt_sf.use_nonrd_pick_mode = 0; |
| sf->rt_sf.use_real_time_ref_set = 0; |
| |
| if (speed >= 1) { |
| sf->gm_sf.disable_adaptive_warp_error_thresh = 0; |
| sf->gm_sf.gm_search_type = GM_REDUCED_REF_SEARCH_SKIP_L2_L3_ARF2; |
| sf->gm_sf.prune_ref_frame_for_gm_search = boosted ? 0 : 1; |
| |
| sf->part_sf.intra_cnn_split = 1; |
| sf->part_sf.simple_motion_search_early_term_none = 1; |
| // TODO(Venkat): Clean-up frame type dependency for |
| // simple_motion_search_split in partition search function and set the |
| // speed feature accordingly |
| sf->part_sf.simple_motion_search_split = |
| cm->allow_screen_content_tools ? 1 : 2; |
| |
| sf->mv_sf.use_accurate_subpel_search = USE_4_TAPS; |
| |
| sf->inter_sf.disable_interinter_wedge_newmv_search = boosted ? 0 : 1; |
| sf->inter_sf.obmc_full_pixel_search_level = 1; |
| sf->inter_sf.prune_comp_search_by_single_result = boosted ? 2 : 1; |
| sf->inter_sf.prune_comp_type_by_comp_avg = 1; |
| sf->inter_sf.prune_comp_type_by_model_rd = boosted ? 0 : 1; |
| sf->inter_sf.prune_motion_mode_level = 2; |
| sf->inter_sf.prune_ref_frame_for_rect_partitions = |
| (frame_is_intra_only(&cpi->common) || (cm->allow_screen_content_tools)) |
| ? 0 |
| : (boosted ? 1 : 2); |
| sf->inter_sf.reduce_inter_modes = boosted ? 1 : 2; |
| sf->inter_sf.reuse_inter_intra_mode = 1; |
| sf->inter_sf.selective_ref_frame = 2; |
| sf->inter_sf.skip_repeated_newmv = 1; |
| |
| sf->interp_sf.cb_pred_filter_search = 0; |
| sf->interp_sf.use_interp_filter = 1; |
| sf->intra_sf.prune_palette_search_level = 1; |
| |
| sf->tx_sf.adaptive_txb_search_level = 2; |
| sf->tx_sf.inter_tx_size_search_init_depth_rect = 1; |
| sf->tx_sf.inter_tx_size_search_init_depth_sqr = 1; |
| sf->tx_sf.intra_tx_size_search_init_depth_rect = 1; |
| sf->tx_sf.model_based_prune_tx_search_level = 0; |
| sf->tx_sf.tx_type_search.ml_tx_split_thresh = 4000; |
| sf->tx_sf.tx_type_search.prune_mode = PRUNE_2D_FAST; |
| sf->tx_sf.tx_type_search.skip_tx_search = 1; |
| sf->tx_sf.use_intra_txb_hash = 1; |
| |
| sf->rd_sf.perform_coeff_opt = boosted ? 1 : 2; |
| sf->rd_sf.tx_domain_dist_level = boosted ? 1 : 2; |
| sf->rd_sf.tx_domain_dist_thres_level = 1; |
| |
| sf->lpf_sf.cdef_pick_method = CDEF_FAST_SEARCH; |
| sf->lpf_sf.dual_sgr_penalty_level = 1; |
| sf->lpf_sf.enable_sgr_ep_pruning = 1; |
| } |
| |
| if (speed >= 2) { |
| sf->gm_sf.gm_erroradv_type = GM_ERRORADV_TR_2; |
| |
| sf->part_sf.allow_partition_search_skip = 1; |
| |
| sf->mv_sf.auto_mv_step_size = 1; |
| sf->mv_sf.subpel_iters_per_step = 1; |
| |
| // TODO(chiyotsai@google.com): We can get 10% speed up if we move |
| // adaptive_rd_thresh to speed 1. But currently it performs poorly on some |
| // clips (e.g. 5% loss on dinner_1080p). We need to examine the sequence a |
| // bit more closely to figure out why. |
| sf->inter_sf.adaptive_rd_thresh = 1; |
| sf->inter_sf.comp_inter_joint_search_thresh = BLOCK_SIZES_ALL; |
| sf->inter_sf.disable_interinter_wedge_newmv_search = 1; |
| sf->inter_sf.disable_wedge_search_edge_thresh = 0; |
| sf->inter_sf.disable_wedge_search_var_thresh = 100; |
| sf->inter_sf.fast_interintra_wedge_search = 1; |
| sf->inter_sf.fast_wedge_sign_estimate = 1; |
| sf->inter_sf.prune_comp_search_by_single_result = boosted ? 4 : 1; |
| sf->inter_sf.prune_comp_type_by_comp_avg = 2; |
| sf->inter_sf.prune_warp_using_wmtype = 1; |
| sf->inter_sf.selective_ref_frame = 3; |
| sf->inter_sf.use_dist_wtd_comp_flag = DIST_WTD_COMP_DISABLED; |
| |
| // TODO(Sachin): Enable/Enhance this speed feature for speed 2 & 3 |
| sf->interp_sf.adaptive_interp_filter_search = 1; |
| sf->interp_sf.disable_dual_filter = 1; |
| sf->interp_sf.disable_filter_search_var_thresh = 100; |
| |
| sf->intra_sf.disable_smooth_intra = |
| !frame_is_intra_only(&cpi->common) || (cpi->rc.frames_to_key != 1); |
| |
| sf->rd_sf.perform_coeff_opt = is_boosted_arf2_bwd_type ? 2 : 3; |
| |
| sf->lpf_sf.prune_sgr_based_on_wiener = |
| cm->allow_screen_content_tools ? 0 : 1; |
| } |
| |
| if (speed >= 3) { |
| sf->hl_sf.recode_loop = ALLOW_RECODE_KFARFGF; |
| |
| sf->gm_sf.gm_search_type = GM_DISABLE_SEARCH; |
| |
| sf->part_sf.less_rectangular_check_level = 2; |
| sf->part_sf.simple_motion_search_prune_agg = 1; |
| |
| // adaptive_motion_search breaks encoder multi-thread tests. |
| // The values in x->pred_mv[] differ for single and multi-thread cases. |
| // See aomedia:1778. |
| // sf->mv_sf.adaptive_motion_search = 1; |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE_PRUNED; |
| sf->mv_sf.use_accurate_subpel_search = USE_2_TAPS; |
| sf->mv_sf.search_method = DIAMOND; |
| sf->inter_sf.disable_sb_level_mv_cost_upd = 1; |
| // TODO(yunqing): evaluate this speed feature for speed 1 & 2, and combine |
| // it with cpi->sf.disable_wedge_search_var_thresh. |
| sf->inter_sf.disable_wedge_interintra_search = 1; |
| // TODO(any): Experiment with the early exit mechanism for speeds 0, 1 and 2 |
| // and clean-up the speed feature |
| sf->inter_sf.perform_best_rd_based_gating_for_chroma = 1; |
| sf->inter_sf.prune_comp_search_by_single_result = boosted ? 4 : 2; |
| sf->inter_sf.prune_motion_mode_level = boosted ? 2 : 3; |
| if (cpi->oxcf.enable_smooth_interintra) |
| sf->inter_sf.disable_smooth_interintra = boosted ? 0 : 1; |
| sf->inter_sf.reuse_compound_type_decision = 1; |
| |
| sf->intra_sf.prune_palette_search_level = 2; |
| |
| sf->tx_sf.tx_type_search.use_skip_flag_prediction = |
| cm->allow_screen_content_tools ? 1 : 2; |
| |
| // TODO(any): Refactor the code related to following winner mode speed |
| // features |
| sf->winner_mode_sf.enable_winner_mode_for_coeff_opt = 1; |
| // TODO(any): Experiment with this speed feature by enabling for key frames |
| sf->winner_mode_sf.enable_winner_mode_for_tx_size_srch = |
| frame_is_intra_only(&cpi->common) ? 0 : 1; |
| sf->winner_mode_sf.enable_winner_mode_for_use_tx_domain_dist = |
| cm->allow_screen_content_tools ? 0 : 1; |
| sf->winner_mode_sf.motion_mode_for_winner_cand = |
| boosted |
| ? 0 |
| : gf_group->update_type[gf_group->index] == INTNL_ARF_UPDATE ? 1 |
| : 2; |
| |
| sf->lpf_sf.prune_wiener_based_on_src_var = 1; |
| sf->lpf_sf.prune_sgr_based_on_wiener = |
| cm->allow_screen_content_tools ? 0 : 2; |
| sf->lpf_sf.reduce_wiener_window_size = is_boosted_arf2_bwd_type ? 0 : 1; |
| sf->hl_sf.second_alt_ref_filtering = 0; |
| |
| sf->tpl_sf.skip_repeated_mv_level = 1; |
| } |
| |
| if (speed >= 4) { |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE_PRUNED_MORE; |
| |
| sf->part_sf.simple_motion_search_prune_agg = 2; |
| |
| sf->inter_sf.adaptive_mode_search = 1; |
| sf->inter_sf.alt_ref_search_fp = 1; |
| sf->inter_sf.prune_ref_mv_idx_search = 1; |
| sf->inter_sf.selective_ref_frame = 4; |
| |
| sf->interp_sf.cb_pred_filter_search = 1; |
| sf->interp_sf.skip_sharp_interp_filter_search = 1; |
| sf->interp_sf.use_interp_filter = 2; |
| |
| sf->intra_sf.intra_uv_mode_mask[TX_16X16] = UV_INTRA_DC_H_V_CFL; |
| sf->intra_sf.intra_uv_mode_mask[TX_32X32] = UV_INTRA_DC_H_V_CFL; |
| sf->intra_sf.intra_uv_mode_mask[TX_64X64] = UV_INTRA_DC_H_V_CFL; |
| sf->intra_sf.intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V; |
| sf->intra_sf.intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V; |
| sf->intra_sf.intra_y_mode_mask[TX_64X64] = INTRA_DC_H_V; |
| // TODO(any): Experiment with this speed feature set to 2 for higher quality |
| // presets as well |
| sf->intra_sf.skip_intra_in_interframe = 2; |
| |
| sf->tx_sf.adaptive_txb_search_level = boosted ? 2 : 3; |
| sf->tx_sf.tx_type_search.enable_winner_mode_tx_type_pruning = 1; |
| sf->tx_sf.tx_type_search.fast_intra_tx_type_search = 1; |
| sf->tx_sf.tx_type_search.prune_mode = PRUNE_2D_MORE; |
| // TODO(any): Experiment with enabling of this speed feature as hash state |
| // is reset during winner mode processing |
| sf->tx_sf.use_intra_txb_hash = 0; |
| |
| sf->rd_sf.perform_coeff_opt = is_boosted_arf2_bwd_type ? 2 : 4; |
| sf->rd_sf.tx_domain_dist_thres_level = 2; |
| |
| // TODO(any): Extend multi-winner mode processing support for inter frames |
| sf->winner_mode_sf.enable_multiwinner_mode_process = |
| frame_is_intra_only(&cpi->common) ? 1 : 0; |
| sf->winner_mode_sf.enable_winner_mode_for_tx_size_srch = 1; |
| |
| sf->lpf_sf.disable_loop_restoration_chroma = |
| (boosted || cm->allow_screen_content_tools) ? 0 : 1; |
| sf->lpf_sf.reduce_wiener_window_size = !boosted; |
| sf->lpf_sf.prune_wiener_based_on_src_var = 2; |
| |
| // TODO(any): The following features have no impact on quality and speed, |
| // and are disabled. |
| // sf->part_sf.partition_search_breakout_rate_thr = 300; |
| // sf->interp_sf.disable_filter_search_var_thresh = 200; |
| // sf->rd_sf.use_fast_coef_costing = 1; |
| |
| // TODO(any): The following features give really bad quality/speed trade |
| // off. Needs to be re-worked. |
| // sf->mv_sf.search_method = BIGDIA; |
| // sf->inter_sf.adaptive_rd_thresh = 4; |
| // sf->rd_sf.tx_domain_dist_level = 2; |
| // sf->rt_sf.mode_search_skip_flags = |
| // (cm->current_frame.frame_type == KEY_FRAME) |
| // ? 0 |
| // : FLAG_SKIP_INTRA_DIRMISMATCH | FLAG_SKIP_INTRA_BESTINTER | |
| // FLAG_SKIP_COMP_BESTINTRA | FLAG_SKIP_INTRA_LOWVAR | |
| // FLAG_EARLY_TERMINATE; |
| } |
| |
| if (speed >= 5) { |
| sf->tpl_sf.prune_intra_modes = 1; |
| sf->tpl_sf.reduce_first_step_size = 6; |
| |
| sf->inter_sf.disable_interinter_wedge = 1; |
| sf->inter_sf.disable_obmc = 1; |
| sf->inter_sf.disable_onesided_comp = 1; |
| sf->inter_sf.disable_smooth_interintra = 1; |
| |
| sf->lpf_sf.disable_lr_filter = 1; |
| } |
| } |
| |
| // TODO(kyslov): now this is very similar to |
| // set_good_speed_features_framesize_independent |
| // except it sets non-rd flag on speed8. This function will likely |
| // be modified in the future with RT-specific speed features |
| static void set_rt_speed_features_framesize_independent(AV1_COMP *cpi, |
| SPEED_FEATURES *sf, |
| int speed) { |
| AV1_COMMON *const cm = &cpi->common; |
| const int boosted = frame_is_boosted(cpi); |
| |
| // Speed 0 for all speed features that give neutral coding performance change. |
| sf->gm_sf.gm_disable_recode = 1; |
| sf->gm_sf.gm_search_type = GM_REDUCED_REF_SEARCH_SKIP_L2_L3; |
| |
| sf->part_sf.less_rectangular_check_level = 1; |
| sf->part_sf.ml_prune_4_partition = 1; |
| sf->part_sf.ml_prune_ab_partition = 1; |
| sf->part_sf.ml_prune_rect_partition = 1; |
| sf->part_sf.prune_ext_partition_types_search_level = 1; |
| |
| // TODO(debargha): Test, tweak and turn on either 1 or 2 |
| sf->inter_sf.inter_mode_rd_model_estimation = 0; |
| sf->inter_sf.disable_wedge_search_edge_thresh = 0; |
| sf->inter_sf.disable_wedge_search_var_thresh = 0; |
| sf->inter_sf.model_based_post_interp_filter_breakout = 1; |
| sf->inter_sf.prune_compound_using_single_ref = 0; |
| sf->inter_sf.prune_mode_search_simple_translation = 1; |
| sf->inter_sf.prune_motion_mode_level = 1; |
| sf->inter_sf.prune_ref_frame_for_rect_partitions = !boosted; |
| sf->inter_sf.prune_wedge_pred_diff_based = 1; |
| sf->inter_sf.reduce_inter_modes = 1; |
| sf->inter_sf.selective_ref_frame = 1; |
| sf->inter_sf.use_dist_wtd_comp_flag = DIST_WTD_COMP_SKIP_MV_SEARCH; |
| |
| sf->interp_sf.cb_pred_filter_search = 0; |
| sf->interp_sf.use_fast_interpolation_filter_search = 1; |
| |
| sf->intra_sf.intra_pruning_with_hog = 1; |
| sf->intra_sf.intra_pruning_with_hog_thresh = -1.2f; |
| |
| sf->rt_sf.check_intra_pred_nonrd = 1; |
| sf->rt_sf.estimate_motion_for_var_based_partition = 1; |
| sf->rt_sf.hybrid_intra_pickmode = 0; |
| sf->rt_sf.nonrd_reduce_golden_mode_search = 0; |
| sf->rt_sf.nonrd_use_blockyrd_interp_filter = 0; |
| sf->rt_sf.reuse_inter_pred_nonrd = 0; |
| sf->rt_sf.use_comp_ref_nonrd = 1; |
| sf->rt_sf.use_nonrd_filter_search = 1; |
| sf->rt_sf.use_nonrd_pick_mode = 0; |
| sf->rt_sf.use_real_time_ref_set = 0; |
| sf->tx_sf.adaptive_txb_search_level = 1; |
| sf->tx_sf.intra_tx_size_search_init_depth_sqr = 1; |
| sf->tx_sf.model_based_prune_tx_search_level = 1; |
| sf->tx_sf.tx_type_search.use_reduced_intra_txset = 1; |
| |
| if (speed >= 1) { |
| sf->gm_sf.gm_erroradv_type = GM_ERRORADV_TR_1; |
| sf->gm_sf.gm_search_type = GM_REDUCED_REF_SEARCH_SKIP_L2_L3_ARF2; |
| |
| sf->part_sf.prune_ext_partition_types_search_level = 2; |
| sf->part_sf.simple_motion_search_prune_rect = 1; |
| |
| sf->mv_sf.use_accurate_subpel_search = USE_4_TAPS; |
| |
| sf->inter_sf.obmc_full_pixel_search_level = 1; |
| sf->inter_sf.prune_comp_search_by_single_result = 1; |
| sf->inter_sf.reuse_inter_intra_mode = 1; |
| sf->inter_sf.selective_ref_frame = 2; |
| sf->inter_sf.skip_repeated_newmv = 1; |
| sf->inter_sf.disable_wedge_search_var_thresh = 0; |
| sf->inter_sf.disable_wedge_search_edge_thresh = 0; |
| sf->inter_sf.prune_comp_type_by_comp_avg = 1; |
| sf->inter_sf.prune_motion_mode_level = 2; |
| sf->inter_sf.prune_single_motion_modes_by_simple_trans = 1; |
| |
| sf->interp_sf.cb_pred_filter_search = 1; |
| sf->interp_sf.use_interp_filter = 1; |
| |
| sf->tx_sf.adaptive_txb_search_level = 2; |
| sf->tx_sf.intra_tx_size_search_init_depth_rect = 1; |
| sf->tx_sf.tx_size_search_lgr_block = 1; |
| sf->tx_sf.tx_type_search.ml_tx_split_thresh = 4000; |
| sf->tx_sf.tx_type_search.skip_tx_search = 1; |
| sf->tx_sf.use_intra_txb_hash = 1; |
| |
| sf->rd_sf.optimize_b_precheck = 1; |
| sf->rd_sf.tx_domain_dist_level = boosted ? 0 : 1; |
| sf->rd_sf.tx_domain_dist_thres_level = 1; |
| |
| sf->lpf_sf.dual_sgr_penalty_level = 1; |
| } |
| |
| if (speed >= 2) { |
| sf->gm_sf.gm_erroradv_type = GM_ERRORADV_TR_2; |
| |
| sf->part_sf.allow_partition_search_skip = 1; |
| sf->part_sf.partition_search_breakout_rate_thr = 80; |
| |
| sf->mv_sf.auto_mv_step_size = 1; |
| sf->mv_sf.subpel_iters_per_step = 1; |
| |
| sf->inter_sf.adaptive_rd_thresh = 1; |
| sf->inter_sf.comp_inter_joint_search_thresh = BLOCK_SIZES_ALL; |
| sf->inter_sf.disable_wedge_search_edge_thresh = 0; |
| sf->inter_sf.disable_wedge_search_var_thresh = 100; |
| sf->inter_sf.fast_wedge_sign_estimate = 1; |
| sf->inter_sf.prune_comp_type_by_comp_avg = 2; |
| sf->inter_sf.selective_ref_frame = 3; |
| sf->inter_sf.use_dist_wtd_comp_flag = DIST_WTD_COMP_DISABLED; |
| |
| sf->interp_sf.adaptive_interp_filter_search = 1; |
| sf->interp_sf.cb_pred_filter_search = 0; |
| sf->interp_sf.disable_dual_filter = 1; |
| sf->interp_sf.disable_filter_search_var_thresh = 100; |
| |
| sf->tx_sf.inter_tx_size_search_init_depth_rect = 1; |
| sf->tx_sf.inter_tx_size_search_init_depth_sqr = 1; |
| sf->tx_sf.model_based_prune_tx_search_level = 0; |
| |
| sf->lpf_sf.cdef_pick_method = CDEF_FAST_SEARCH; |
| } |
| |
| if (speed >= 3) { |
| sf->hl_sf.recode_loop = ALLOW_RECODE_KFARFGF; |
| |
| sf->gm_sf.gm_search_type = GM_DISABLE_SEARCH; |
| |
| sf->part_sf.less_rectangular_check_level = 2; |
| |
| sf->mv_sf.use_accurate_subpel_search = USE_2_TAPS; |
| // adaptive_motion_search breaks encoder multi-thread tests. |
| // The values in x->pred_mv[] differ for single and multi-thread cases. |
| // See aomedia:1778. |
| // sf->mv_sf.adaptive_motion_search = 1; |
| |
| sf->inter_sf.adaptive_rd_thresh = 2; |
| sf->inter_sf.disable_sb_level_mv_cost_upd = 1; |
| // TODO(yunqing): evaluate this speed feature for speed 1 & 2, and combine |
| // it with cpi->sf.disable_wedge_search_var_thresh. |
| sf->inter_sf.disable_wedge_interintra_search = 1; |
| sf->inter_sf.prune_comp_search_by_single_result = 2; |
| sf->inter_sf.prune_motion_mode_level = boosted ? 2 : 3; |
| sf->inter_sf.prune_warp_using_wmtype = 1; |
| sf->inter_sf.selective_ref_frame = 4; |
| |
| sf->tx_sf.tx_type_search.prune_mode = PRUNE_2D_FAST; |
| |
| sf->rd_sf.tx_domain_dist_level = 1; |
| |
| sf->winner_mode_sf.tx_size_search_level = boosted ? 0 : 2; |
| } |
| |
| if (speed >= 4) { |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE_PRUNED; |
| |
| sf->inter_sf.adaptive_mode_search = 1; |
| sf->inter_sf.alt_ref_search_fp = 1; |
| |
| sf->interp_sf.skip_sharp_interp_filter_search = 1; |
| |
| sf->tx_sf.tx_type_search.fast_inter_tx_type_search = 1; |
| sf->tx_sf.tx_type_search.fast_intra_tx_type_search = 1; |
| sf->tx_sf.use_intra_txb_hash = 0; |
| |
| sf->rd_sf.use_mb_rd_hash = 0; |
| |
| sf->winner_mode_sf.tx_size_search_level = frame_is_intra_only(cm) ? 0 : 2; |
| } |
| |
| if (speed >= 5) { |
| sf->hl_sf.recode_loop = ALLOW_RECODE_KFMAXBW; |
| |
| sf->part_sf.partition_search_breakout_rate_thr = 300; |
| |
| sf->mv_sf.search_method = BIGDIA; |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE_PRUNED_MORE; |
| |
| sf->inter_sf.adaptive_rd_thresh = 4; |
| sf->interp_sf.disable_filter_search_var_thresh = 200; |
| |
| sf->intra_sf.intra_y_mode_mask[TX_64X64] = INTRA_DC_H_V; |
| sf->intra_sf.intra_uv_mode_mask[TX_64X64] = UV_INTRA_DC_H_V_CFL; |
| sf->intra_sf.intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V; |
| sf->intra_sf.intra_uv_mode_mask[TX_32X32] = UV_INTRA_DC_H_V_CFL; |
| sf->intra_sf.intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V; |
| sf->intra_sf.intra_uv_mode_mask[TX_16X16] = UV_INTRA_DC_H_V_CFL; |
| |
| sf->rd_sf.use_fast_coef_costing = 1; |
| sf->rd_sf.tx_domain_dist_level = 2; |
| sf->rd_sf.tx_domain_dist_thres_level = 2; |
| |
| sf->winner_mode_sf.tx_size_search_level = 2; |
| |
| sf->rt_sf.mode_search_skip_flags = |
| (cm->current_frame.frame_type == KEY_FRAME) |
| ? 0 |
| : FLAG_SKIP_INTRA_DIRMISMATCH | FLAG_SKIP_INTRA_BESTINTER | |
| FLAG_SKIP_COMP_BESTINTRA | FLAG_SKIP_INTRA_LOWVAR | |
| FLAG_EARLY_TERMINATE; |
| } |
| |
| if (speed >= 6) { |
| sf->hl_sf.frame_parameter_update = 0; |
| |
| sf->part_sf.default_max_partition_size = BLOCK_128X128; |
| sf->part_sf.default_min_partition_size = BLOCK_8X8; |
| sf->part_sf.max_intra_bsize = BLOCK_32X32; |
| sf->part_sf.partition_search_breakout_rate_thr = 500; |
| sf->part_sf.partition_search_type = VAR_BASED_PARTITION; |
| |
| sf->mv_sf.search_method = FAST_DIAMOND; |
| sf->mv_sf.subpel_force_stop = QUARTER_PEL; |
| |
| sf->inter_sf.adaptive_mode_search = 2; |
| sf->inter_sf.inter_mode_rd_model_estimation = 2; |
| |
| for (int i = 0; i < TX_SIZES; ++i) { |
| sf->intra_sf.intra_y_mode_mask[i] = INTRA_DC; |
| sf->intra_sf.intra_uv_mode_mask[i] = UV_INTRA_DC_CFL; |
| } |
| |
| sf->tx_sf.tx_type_search.prune_mode = PRUNE_2D_MORE; |
| sf->tx_sf.use_inter_txb_hash = 0; |
| |
| sf->rd_sf.optimize_coefficients = NO_TRELLIS_OPT; |
| sf->rd_sf.simple_model_rd_from_var = 1; |
| |
| sf->winner_mode_sf.tx_size_search_level = 1; |
| |
| sf->lpf_sf.cdef_pick_method = CDEF_PICK_FROM_Q; |
| sf->lpf_sf.lpf_pick = LPF_PICK_FROM_Q; |
| |
| sf->rt_sf.force_tx_search_off = 1; |
| sf->rt_sf.mode_search_skip_flags |= FLAG_SKIP_INTRA_DIRMISMATCH; |
| sf->rt_sf.num_inter_modes_for_tx_search = 5; |
| sf->rt_sf.skip_interp_filter_search = 1; |
| sf->rt_sf.use_comp_ref_nonrd = 0; |
| sf->rt_sf.use_real_time_ref_set = 1; |
| sf->rt_sf.use_simple_rd_model = 1; |
| } |
| |
| if (speed >= 7) { |
| sf->hl_sf.frame_parameter_update = 0; |
| |
| sf->part_sf.default_max_partition_size = BLOCK_128X128; |
| sf->part_sf.default_min_partition_size = BLOCK_8X8; |
| sf->part_sf.partition_search_type = VAR_BASED_PARTITION; |
| |
| sf->mv_sf.search_method = FAST_DIAMOND; |
| sf->mv_sf.subpel_force_stop = QUARTER_PEL; |
| sf->mv_sf.subpel_search_method = SUBPEL_TREE; |
| |
| sf->inter_sf.inter_mode_rd_model_estimation = 2; |
| |
| sf->lpf_sf.cdef_pick_method = CDEF_PICK_FROM_Q; |
| sf->lpf_sf.lpf_pick = LPF_PICK_FROM_Q; |
| |
| sf->rt_sf.mode_search_skip_flags |= FLAG_SKIP_INTRA_DIRMISMATCH; |
| sf->rt_sf.nonrd_reduce_golden_mode_search = 0; |
| sf->rt_sf.nonrd_use_blockyrd_interp_filter = 1; |
| sf->rt_sf.reuse_inter_pred_nonrd = 0; |
| sf->rt_sf.short_circuit_low_temp_var = 0; |
| sf->rt_sf.skip_interp_filter_search = 0; |
| sf->rt_sf.use_comp_ref_nonrd = 0; |
| sf->rt_sf.use_nonrd_altref_frame = 1; |
| sf->rt_sf.use_nonrd_pick_mode = 1; |
| sf->rt_sf.nonrd_check_partition_merge = 1; |
| sf->rt_sf.nonrd_check_partition_split = 0; |
| sf->rt_sf.hybrid_intra_pickmode = 1; |
| } |
| |
| if (speed >= 8) { |
| sf->rt_sf.estimate_motion_for_var_based_partition = 0; |
| sf->rt_sf.short_circuit_low_temp_var = 1; |
| sf->rt_sf.reuse_inter_pred_nonrd = 1; |
| sf->rt_sf.nonrd_use_blockyrd_interp_filter = 0; |
| sf->rt_sf.use_nonrd_altref_frame = 0; |
| sf->rt_sf.nonrd_reduce_golden_mode_search = 1; |
| sf->rt_sf.nonrd_check_partition_merge = 0; |
| sf->rt_sf.nonrd_check_partition_split = 0; |
| |
| // TODO(kyslov) Enable when better model is available |
| // It gives +5% speedup and 11% overall BDRate degradation |
| // So, can not enable now until better CurvFit is there |
| #if 0 |
| sf->rt_sf.use_modeled_non_rd_cost = 1; |
| #endif |
| } |
| } |
| |
| static AOM_INLINE void init_hl_sf(HIGH_LEVEL_SPEED_FEATURES *hl_sf) { |
| // best quality defaults |
| hl_sf->frame_parameter_update = 1; |
| hl_sf->recode_loop = ALLOW_RECODE; |
| hl_sf->disable_overlay_frames = 0; |
| hl_sf->adaptive_overlay_encoding = 1; |
| // Recode loop tolerance %. |
| hl_sf->recode_tolerance = 25; |
| hl_sf->high_precision_mv_usage = CURRENT_Q; |
| hl_sf->second_alt_ref_filtering = 1; |
| } |
| |
| static AOM_INLINE void init_tpl_sf(TPL_SPEED_FEATURES *tpl_sf) { |
| tpl_sf->prune_intra_modes = 0; |
| tpl_sf->reduce_first_step_size = 0; |
| tpl_sf->skip_repeated_mv_level = 0; |
| } |
| |
| static AOM_INLINE void init_gm_sf(GLOBAL_MOTION_SPEED_FEATURES *gm_sf) { |
| gm_sf->gm_erroradv_type = GM_ERRORADV_TR_0; |
| gm_sf->disable_adaptive_warp_error_thresh = 1; |
| gm_sf->selective_ref_gm = 1; |
| gm_sf->gm_search_type = GM_FULL_SEARCH; |
| gm_sf->gm_disable_recode = 0; |
| gm_sf->prune_ref_frame_for_gm_search = 0; |
| } |
| |
| static AOM_INLINE void init_part_sf(PARTITION_SPEED_FEATURES *part_sf) { |
| part_sf->partition_search_type = SEARCH_PARTITION; |
| part_sf->less_rectangular_check_level = 0; |
| part_sf->use_square_partition_only_threshold = BLOCK_128X128; |
| part_sf->auto_max_partition_based_on_simple_motion = NOT_IN_USE; |
| part_sf->auto_min_partition_based_on_simple_motion = 0; |
| part_sf->default_max_partition_size = BLOCK_LARGEST; |
| part_sf->default_min_partition_size = BLOCK_4X4; |
| part_sf->adjust_partitioning_from_last_frame = 0; |
| part_sf->allow_partition_search_skip = 0; |
| part_sf->max_intra_bsize = BLOCK_LARGEST; |
| // This setting only takes effect when partition_search_type is set |
| // to FIXED_PARTITION. |
| part_sf->always_this_block_size = BLOCK_16X16; |
| // Recode loop tolerance %. |
| part_sf->partition_search_breakout_dist_thr = 0; |
| part_sf->partition_search_breakout_rate_thr = 0; |
| part_sf->prune_ext_partition_types_search_level = 0; |
| part_sf->ml_prune_rect_partition = 0; |
| part_sf->ml_prune_ab_partition = 0; |
| part_sf->ml_prune_4_partition = 0; |
| part_sf->ml_early_term_after_part_split_level = 0; |
| for (int i = 0; i < PARTITION_BLOCK_SIZES; ++i) { |
| part_sf->ml_partition_search_breakout_thresh[i] = |
| -1; // -1 means not enabled. |
| } |
| part_sf->simple_motion_search_prune_agg = 0; |
| part_sf->simple_motion_search_split = 0; |
| part_sf->simple_motion_search_prune_rect = 0; |
| part_sf->simple_motion_search_early_term_none = 0; |
| part_sf->intra_cnn_split = 0; |
| } |
| |
| static AOM_INLINE void init_mv_sf(MV_SPEED_FEATURES *mv_sf) { |
| mv_sf->search_method = NSTEP; |
| mv_sf->subpel_search_method = SUBPEL_TREE; |
| mv_sf->subpel_iters_per_step = 2; |
| mv_sf->subpel_force_stop = EIGHTH_PEL; |
| mv_sf->auto_mv_step_size = 0; |
| mv_sf->adaptive_motion_search = 0; |
| mv_sf->use_accurate_subpel_search = USE_8_TAPS; |
| mv_sf->disable_hash_me = 0; |
| } |
| |
| static AOM_INLINE void init_inter_sf(INTER_MODE_SPEED_FEATURES *inter_sf) { |
| inter_sf->comp_inter_joint_search_thresh = BLOCK_4X4; |
| inter_sf->adaptive_rd_thresh = 0; |
| inter_sf->model_based_post_interp_filter_breakout = 0; |
| inter_sf->reduce_inter_modes = 0; |
| inter_sf->adaptive_mode_search = 0; |
| inter_sf->alt_ref_search_fp = 0; |
| inter_sf->selective_ref_frame = 0; |
| inter_sf->prune_ref_frame_for_rect_partitions = 0; |
| inter_sf->disable_wedge_search_edge_thresh = 0; |
| inter_sf->disable_wedge_search_var_thresh = 0; |
| inter_sf->fast_wedge_sign_estimate = 0; |
| inter_sf->prune_wedge_pred_diff_based = 0; |
| inter_sf->use_dist_wtd_comp_flag = DIST_WTD_COMP_ENABLED; |
| inter_sf->reuse_inter_intra_mode = 0; |
| inter_sf->disable_sb_level_coeff_cost_upd = 0; |
| inter_sf->disable_sb_level_mv_cost_upd = 0; |
| inter_sf->prune_comp_search_by_single_result = 0; |
| inter_sf->skip_repeated_newmv = 0; |
| inter_sf->prune_single_motion_modes_by_simple_trans = 0; |
| inter_sf->inter_mode_rd_model_estimation = 0; |
| inter_sf->prune_compound_using_single_ref = 0; |
| inter_sf->disable_onesided_comp = 0; |
| inter_sf->prune_mode_search_simple_translation = 0; |
| inter_sf->obmc_full_pixel_search_level = 0; |
| inter_sf->prune_comp_type_by_comp_avg = 0; |
| inter_sf->disable_interinter_wedge_newmv_search = 0; |
| inter_sf->enable_interinter_diffwtd_newmv_search = 0; |
| inter_sf->disable_smooth_interintra = 0; |
| inter_sf->prune_motion_mode_level = 0; |
| inter_sf->prune_warp_using_wmtype = 0; |
| inter_sf->disable_wedge_interintra_search = 0; |
| inter_sf->fast_interintra_wedge_search = 0; |
| inter_sf->prune_comp_type_by_model_rd = 0; |
| inter_sf->perform_best_rd_based_gating_for_chroma = 0; |
| inter_sf->prune_obmc_prob_thresh = 0; |
| inter_sf->disable_obmc = 0; |
| inter_sf->disable_interinter_wedge = 0; |
| inter_sf->prune_ref_mv_idx_search = 0; |
| inter_sf->prune_warped_prob_thresh = 0; |
| inter_sf->reuse_compound_type_decision = 0; |
| } |
| |
| static AOM_INLINE void init_interp_sf(INTERP_FILTER_SPEED_FEATURES *interp_sf) { |
| interp_sf->disable_filter_search_var_thresh = 0; |
| interp_sf->adaptive_interp_filter_search = 0; |
| interp_sf->use_fast_interpolation_filter_search = 0; |
| interp_sf->disable_dual_filter = 0; |
| interp_sf->use_interp_filter = 0; |
| interp_sf->skip_sharp_interp_filter_search = 0; |
| } |
| |
| static AOM_INLINE void init_intra_sf(INTRA_MODE_SPEED_FEATURES *intra_sf) { |
| intra_sf->skip_intra_in_interframe = 1; |
| intra_sf->intra_pruning_with_hog = 0; |
| intra_sf->src_var_thresh_intra_skip = 1; |
| intra_sf->prune_palette_search_level = 0; |
| |
| for (int i = 0; i < TX_SIZES; i++) { |
| intra_sf->intra_y_mode_mask[i] = INTRA_ALL; |
| intra_sf->intra_uv_mode_mask[i] = UV_INTRA_ALL; |
| } |
| intra_sf->disable_smooth_intra = 0; |
| } |
| |
| static AOM_INLINE void init_tx_sf(TX_SPEED_FEATURES *tx_sf) { |
| tx_sf->inter_tx_size_search_init_depth_sqr = 0; |
| tx_sf->inter_tx_size_search_init_depth_rect = 0; |
| tx_sf->intra_tx_size_search_init_depth_rect = 0; |
| tx_sf->intra_tx_size_search_init_depth_sqr = 0; |
| tx_sf->tx_size_search_lgr_block = 0; |
| tx_sf->model_based_prune_tx_search_level = 0; |
| tx_sf->tx_type_search.prune_mode = PRUNE_2D_ACCURATE; |
| tx_sf->tx_type_search.ml_tx_split_thresh = 8500; |
| tx_sf->tx_type_search.use_skip_flag_prediction = 1; |
| tx_sf->tx_type_search.use_reduced_intra_txset = 0; |
| tx_sf->tx_type_search.fast_intra_tx_type_search = 0; |
| tx_sf->tx_type_search.fast_inter_tx_type_search = 0; |
| tx_sf->tx_type_search.skip_tx_search = 0; |
| tx_sf->tx_type_search.prune_tx_type_using_stats = 0; |
| tx_sf->tx_type_search.enable_winner_mode_tx_type_pruning = 0; |
| tx_sf->txb_split_cap = 1; |
| tx_sf->adaptive_txb_search_level = 0; |
| tx_sf->use_intra_txb_hash = 0; |
| tx_sf->use_inter_txb_hash = 1; |
| } |
| |
| static AOM_INLINE void init_rd_sf(RD_CALC_SPEED_FEATURES *rd_sf, |
| const AV1_COMP *cpi) { |
| if (cpi->oxcf.disable_trellis_quant == 3) { |
| rd_sf->optimize_coefficients = !is_lossless_requested(&cpi->oxcf) |
| ? NO_ESTIMATE_YRD_TRELLIS_OPT |
| : NO_TRELLIS_OPT; |
| } else if (cpi->oxcf.disable_trellis_quant == 2) { |
| rd_sf->optimize_coefficients = !is_lossless_requested(&cpi->oxcf) |
| ? FINAL_PASS_TRELLIS_OPT |
| : NO_TRELLIS_OPT; |
| } else if (cpi->oxcf.disable_trellis_quant == 0) { |
| if (is_lossless_requested(&cpi->oxcf)) { |
| rd_sf->optimize_coefficients = NO_TRELLIS_OPT; |
| } else { |
| rd_sf->optimize_coefficients = FULL_TRELLIS_OPT; |
| } |
| } else if (cpi->oxcf.disable_trellis_quant == 1) { |
| rd_sf->optimize_coefficients = NO_TRELLIS_OPT; |
| } else { |
| assert(0 && "Invalid disable_trellis_quant value"); |
| } |
| // TODO(sarahparker) Pair this with a speed setting once experiments are done |
| rd_sf->trellis_eob_fast = 0; |
| rd_sf->use_mb_rd_hash = 1; |
| rd_sf->optimize_b_precheck = 0; |
| rd_sf->use_fast_coef_costing = 0; |
| rd_sf->simple_model_rd_from_var = 0; |
| rd_sf->tx_domain_dist_level = 0; |
| rd_sf->tx_domain_dist_thres_level = 0; |
| rd_sf->use_hash_based_trellis = 0; |
| rd_sf->perform_coeff_opt = 0; |
| } |
| |
| static AOM_INLINE void init_winner_mode_sf( |
| WINNER_MODE_SPEED_FEATURES *winner_mode_sf) { |
| winner_mode_sf->motion_mode_for_winner_cand = 0; |
| // Set this at the appropriate speed levels |
| winner_mode_sf->tx_size_search_level = USE_FULL_RD; |
| winner_mode_sf->enable_winner_mode_for_coeff_opt = 0; |
| winner_mode_sf->enable_winner_mode_for_tx_size_srch = 0; |
| winner_mode_sf->enable_winner_mode_for_use_tx_domain_dist = 0; |
| winner_mode_sf->enable_multiwinner_mode_process = 0; |
| } |
| |
| static AOM_INLINE void init_lpf_sf(LOOP_FILTER_SPEED_FEATURES *lpf_sf) { |
| lpf_sf->disable_loop_restoration_chroma = 0; |
| lpf_sf->prune_wiener_based_on_src_var = 0; |
| lpf_sf->prune_sgr_based_on_wiener = 0; |
| lpf_sf->enable_sgr_ep_pruning = 0; |
| lpf_sf->reduce_wiener_window_size = 0; |
| lpf_sf->lpf_pick = LPF_PICK_FROM_FULL_IMAGE; |
| lpf_sf->cdef_pick_method = CDEF_FULL_SEARCH; |
| // Set decoder side speed feature to use less dual sgr modes |
| lpf_sf->dual_sgr_penalty_level = 0; |
| lpf_sf->disable_lr_filter = 0; |
| } |
| |
| static AOM_INLINE void init_rt_sf(REAL_TIME_SPEED_FEATURES *rt_sf) { |
| rt_sf->mode_search_skip_flags = 0; |
| rt_sf->skip_interp_filter_search = 0; |
| rt_sf->force_tx_search_off = 0; |
| rt_sf->num_inter_modes_for_tx_search = INT_MAX; |
| rt_sf->use_simple_rd_model = 0; |
| rt_sf->nonrd_check_partition_merge = 0; |
| rt_sf->nonrd_check_partition_split = 0; |
| } |
| |
| void av1_set_speed_features_framesize_dependent(AV1_COMP *cpi, int speed) { |
| SPEED_FEATURES *const sf = &cpi->sf; |
| const AV1EncoderConfig *const oxcf = &cpi->oxcf; |
| |
| if (oxcf->mode == GOOD) { |
| set_good_speed_feature_framesize_dependent(cpi, sf, speed); |
| } else if (oxcf->mode == REALTIME) { |
| set_rt_speed_feature_framesize_dependent(cpi, sf, speed); |
| } |
| |
| // This is only used in motion vector unit test. |
| if (cpi->oxcf.motion_vector_unit_test == 1) |
| cpi->find_fractional_mv_step = av1_return_max_sub_pixel_mv; |
| else if (cpi->oxcf.motion_vector_unit_test == 2) |
| cpi->find_fractional_mv_step = av1_return_min_sub_pixel_mv; |
| |
| MACROBLOCK *const x = &cpi->td.mb; |
| AV1_COMMON *const cm = &cpi->common; |
| x->min_partition_size = AOMMAX(sf->part_sf.default_min_partition_size, |
| dim_to_size(cpi->oxcf.min_partition_size)); |
| x->max_partition_size = AOMMIN(sf->part_sf.default_max_partition_size, |
| dim_to_size(cpi->oxcf.max_partition_size)); |
| x->min_partition_size = AOMMIN(x->min_partition_size, cm->seq_params.sb_size); |
| x->max_partition_size = AOMMIN(x->max_partition_size, cm->seq_params.sb_size); |
| } |
| |
| void av1_set_speed_features_framesize_independent(AV1_COMP *cpi, int speed) { |
| AV1_COMMON *const cm = &cpi->common; |
| SPEED_FEATURES *const sf = &cpi->sf; |
| MACROBLOCK *const x = &cpi->td.mb; |
| const AV1EncoderConfig *const oxcf = &cpi->oxcf; |
| int i; |
| |
| init_hl_sf(&sf->hl_sf); |
| init_tpl_sf(&sf->tpl_sf); |
| init_gm_sf(&sf->gm_sf); |
| init_part_sf(&sf->part_sf); |
| init_mv_sf(&sf->mv_sf); |
| init_inter_sf(&sf->inter_sf); |
| init_interp_sf(&sf->interp_sf); |
| init_intra_sf(&sf->intra_sf); |
| init_tx_sf(&sf->tx_sf); |
| init_rd_sf(&sf->rd_sf, cpi); |
| init_winner_mode_sf(&sf->winner_mode_sf); |
| init_lpf_sf(&sf->lpf_sf); |
| init_rt_sf(&sf->rt_sf); |
| |
| if (oxcf->mode == GOOD) |
| set_good_speed_features_framesize_independent(cpi, sf, speed); |
| else if (oxcf->mode == REALTIME) |
| set_rt_speed_features_framesize_independent(cpi, sf, speed); |
| |
| if (!cpi->seq_params_locked) { |
| cpi->common.seq_params.enable_dual_filter &= |
| !sf->interp_sf.disable_dual_filter; |
| cpi->common.seq_params.enable_restoration &= !sf->lpf_sf.disable_lr_filter; |
| } |
| |
| // sf->part_sf.partition_search_breakout_dist_thr is set assuming max 64x64 |
| // blocks. Normalise this if the blocks are bigger. |
| if (MAX_SB_SIZE_LOG2 > 6) { |
| sf->part_sf.partition_search_breakout_dist_thr <<= |
| 2 * (MAX_SB_SIZE_LOG2 - 6); |
| } |
| |
| sf->mv_sf.allow_exhaustive_searches = 1; |
| |
| const int mesh_speed = AOMMIN(speed, MAX_MESH_SPEED); |
| if (cpi->twopass.fr_content_type == FC_GRAPHICS_ANIMATION) |
| sf->mv_sf.exhaustive_searches_thresh = (1 << 24); |
| else |
| sf->mv_sf.exhaustive_searches_thresh = (1 << 25); |
| sf->mv_sf.max_exhaustive_pct = good_quality_max_mesh_pct[mesh_speed]; |
| if (mesh_speed > 0) |
| sf->mv_sf.exhaustive_searches_thresh = sf->mv_sf.exhaustive_searches_thresh |
| << 1; |
| |
| for (i = 0; i < MAX_MESH_STEP; ++i) { |
| sf->mv_sf.mesh_patterns[i].range = |
| good_quality_mesh_patterns[mesh_speed][i].range; |
| sf->mv_sf.mesh_patterns[i].interval = |
| good_quality_mesh_patterns[mesh_speed][i].interval; |
| } |
| |
| // Update the mesh pattern of exhaustive motion search for intraBC |
| // Though intraBC mesh pattern is populated for all frame types, it is used |
| // only for intra frames of screen contents |
| for (i = 0; i < MAX_MESH_STEP; ++i) { |
| sf->mv_sf.intrabc_mesh_patterns[i].range = |
| intrabc_mesh_patterns[mesh_speed][i].range; |
| sf->mv_sf.intrabc_mesh_patterns[i].interval = |
| intrabc_mesh_patterns[mesh_speed][i].interval; |
| } |
| sf->mv_sf.intrabc_max_exhaustive_pct = intrabc_max_mesh_pct[mesh_speed]; |
| |
| // Slow quant, dct and trellis not worthwhile for first pass |
| // so make sure they are always turned off. |
| if (is_stat_generation_stage(cpi)) |
| sf->rd_sf.optimize_coefficients = NO_TRELLIS_OPT; |
| |
| // No recode or trellis for 1 pass. |
| if (oxcf->pass == 0) sf->hl_sf.recode_loop = DISALLOW_RECODE; |
| |
| if (sf->mv_sf.subpel_search_method == SUBPEL_TREE) { |
| cpi->find_fractional_mv_step = av1_find_best_sub_pixel_tree; |
| } else if (sf->mv_sf.subpel_search_method == SUBPEL_TREE_PRUNED) { |
| cpi->find_fractional_mv_step = av1_find_best_sub_pixel_tree_pruned; |
| } else if (sf->mv_sf.subpel_search_method == SUBPEL_TREE_PRUNED_MORE) { |
| cpi->find_fractional_mv_step = av1_find_best_sub_pixel_tree_pruned_more; |
| } else if (sf->mv_sf.subpel_search_method == SUBPEL_TREE_PRUNED_EVENMORE) { |
| cpi->find_fractional_mv_step = av1_find_best_sub_pixel_tree_pruned_evenmore; |
| } |
| |
| x->min_partition_size = AOMMAX(sf->part_sf.default_min_partition_size, |
| dim_to_size(cpi->oxcf.min_partition_size)); |
| x->max_partition_size = AOMMIN(sf->part_sf.default_max_partition_size, |
| dim_to_size(cpi->oxcf.max_partition_size)); |
| x->min_partition_size = AOMMIN(x->min_partition_size, cm->seq_params.sb_size); |
| x->max_partition_size = AOMMIN(x->max_partition_size, cm->seq_params.sb_size); |
| |
| // This is only used in motion vector unit test. |
| if (cpi->oxcf.motion_vector_unit_test == 1) |
| cpi->find_fractional_mv_step = av1_return_max_sub_pixel_mv; |
| else if (cpi->oxcf.motion_vector_unit_test == 2) |
| cpi->find_fractional_mv_step = av1_return_min_sub_pixel_mv; |
| cpi->max_comp_type_rd_threshold_mul = |
| comp_type_rd_threshold_mul[sf->inter_sf.prune_comp_type_by_comp_avg]; |
| cpi->max_comp_type_rd_threshold_div = |
| comp_type_rd_threshold_div[sf->inter_sf.prune_comp_type_by_comp_avg]; |
| |
| // assert ensures that tx_domain_dist_level is accessed correctly |
| assert(cpi->sf.rd_sf.tx_domain_dist_thres_level >= 0 && |
| cpi->sf.rd_sf.tx_domain_dist_thres_level < 3); |
| memcpy(cpi->tx_domain_dist_threshold, |
| tx_domain_dist_thresholds[cpi->sf.rd_sf.tx_domain_dist_thres_level], |
| sizeof(cpi->tx_domain_dist_threshold)); |
| |
| assert(cpi->sf.rd_sf.tx_domain_dist_level >= 0 && |
| cpi->sf.rd_sf.tx_domain_dist_level < 3); |
| memcpy(cpi->use_transform_domain_distortion, |
| tx_domain_dist_types[cpi->sf.rd_sf.tx_domain_dist_level], |
| sizeof(cpi->use_transform_domain_distortion)); |
| |
| // Update the number of winner motion modes to be used appropriately |
| cpi->num_winner_motion_modes = |
| num_winner_motion_modes[cpi->sf.winner_mode_sf |
| .motion_mode_for_winner_cand]; |
| assert(cpi->num_winner_motion_modes <= MAX_WINNER_MOTION_MODES); |
| |
| // assert ensures that coeff_opt_dist_thresholds is accessed correctly |
| assert(cpi->sf.rd_sf.perform_coeff_opt >= 0 && |
| cpi->sf.rd_sf.perform_coeff_opt < 5); |
| memcpy(cpi->coeff_opt_dist_threshold, |
| coeff_opt_dist_thresholds[cpi->sf.rd_sf.perform_coeff_opt], |
| sizeof(cpi->coeff_opt_dist_threshold)); |
| |
| // assert ensures that predict_skip_levels is accessed correctly |
| assert(cpi->sf.tx_sf.tx_type_search.use_skip_flag_prediction >= 0 && |
| cpi->sf.tx_sf.tx_type_search.use_skip_flag_prediction < 3); |
| memcpy(cpi->predict_skip_level, |
| predict_skip_levels[cpi->sf.tx_sf.tx_type_search |
| .use_skip_flag_prediction], |
| sizeof(cpi->predict_skip_level)); |
| |
| // assert ensures that tx_size_search_level is accessed correctly |
| assert(cpi->sf.winner_mode_sf.tx_size_search_level >= 0 && |
| cpi->sf.winner_mode_sf.tx_size_search_level < 3); |
| memcpy(cpi->tx_size_search_methods, |
| tx_size_search_methods[cpi->sf.winner_mode_sf.tx_size_search_level], |
| sizeof(cpi->tx_size_search_methods)); |
| |
| #if CONFIG_DIST_8X8 |
| if (sf->tx_domain_dist_level > 0) cpi->oxcf.using_dist_8x8 = 0; |
| |
| if (cpi->oxcf.using_dist_8x8) x->min_partition_size = BLOCK_8X8; |
| #endif // CONFIG_DIST_8X8 |
| if (cpi->oxcf.row_mt == 1 && (cpi->oxcf.max_threads > 1)) { |
| if (sf->inter_sf.inter_mode_rd_model_estimation == 1) { |
| // Revert to type 2 |
| sf->inter_sf.inter_mode_rd_model_estimation = 2; |
| } |
| } |
| } |