xref: /aosp_15_r20/external/libvpx/vp9/encoder/vp9_speed_features.h (revision fb1b10ab9aebc7c7068eedab379b749d7e3900be)
1*fb1b10abSAndroid Build Coastguard Worker /*
2*fb1b10abSAndroid Build Coastguard Worker  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3*fb1b10abSAndroid Build Coastguard Worker  *
4*fb1b10abSAndroid Build Coastguard Worker  *  Use of this source code is governed by a BSD-style license
5*fb1b10abSAndroid Build Coastguard Worker  *  that can be found in the LICENSE file in the root of the source
6*fb1b10abSAndroid Build Coastguard Worker  *  tree. An additional intellectual property rights grant can be found
7*fb1b10abSAndroid Build Coastguard Worker  *  in the file PATENTS.  All contributing project authors may
8*fb1b10abSAndroid Build Coastguard Worker  *  be found in the AUTHORS file in the root of the source tree.
9*fb1b10abSAndroid Build Coastguard Worker  */
10*fb1b10abSAndroid Build Coastguard Worker 
11*fb1b10abSAndroid Build Coastguard Worker #ifndef VPX_VP9_ENCODER_VP9_SPEED_FEATURES_H_
12*fb1b10abSAndroid Build Coastguard Worker #define VPX_VP9_ENCODER_VP9_SPEED_FEATURES_H_
13*fb1b10abSAndroid Build Coastguard Worker 
14*fb1b10abSAndroid Build Coastguard Worker #include "vp9/common/vp9_enums.h"
15*fb1b10abSAndroid Build Coastguard Worker 
16*fb1b10abSAndroid Build Coastguard Worker #ifdef __cplusplus
17*fb1b10abSAndroid Build Coastguard Worker extern "C" {
18*fb1b10abSAndroid Build Coastguard Worker #endif
19*fb1b10abSAndroid Build Coastguard Worker 
20*fb1b10abSAndroid Build Coastguard Worker enum {
21*fb1b10abSAndroid Build Coastguard Worker   INTRA_ALL = (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED) | (1 << D45_PRED) |
22*fb1b10abSAndroid Build Coastguard Worker               (1 << D135_PRED) | (1 << D117_PRED) | (1 << D153_PRED) |
23*fb1b10abSAndroid Build Coastguard Worker               (1 << D207_PRED) | (1 << D63_PRED) | (1 << TM_PRED),
24*fb1b10abSAndroid Build Coastguard Worker   INTRA_DC = (1 << DC_PRED),
25*fb1b10abSAndroid Build Coastguard Worker   INTRA_DC_TM = (1 << DC_PRED) | (1 << TM_PRED),
26*fb1b10abSAndroid Build Coastguard Worker   INTRA_DC_H_V = (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED),
27*fb1b10abSAndroid Build Coastguard Worker   INTRA_DC_TM_H_V =
28*fb1b10abSAndroid Build Coastguard Worker       (1 << DC_PRED) | (1 << TM_PRED) | (1 << V_PRED) | (1 << H_PRED)
29*fb1b10abSAndroid Build Coastguard Worker };
30*fb1b10abSAndroid Build Coastguard Worker 
31*fb1b10abSAndroid Build Coastguard Worker enum {
32*fb1b10abSAndroid Build Coastguard Worker   INTER_ALL = (1 << NEARESTMV) | (1 << NEARMV) | (1 << ZEROMV) | (1 << NEWMV),
33*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST = (1 << NEARESTMV),
34*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST_NEW = (1 << NEARESTMV) | (1 << NEWMV),
35*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST_ZERO = (1 << NEARESTMV) | (1 << ZEROMV),
36*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST_NEW_ZERO = (1 << NEARESTMV) | (1 << ZEROMV) | (1 << NEWMV),
37*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST_NEAR_NEW = (1 << NEARESTMV) | (1 << NEARMV) | (1 << NEWMV),
38*fb1b10abSAndroid Build Coastguard Worker   INTER_NEAREST_NEAR_ZERO = (1 << NEARESTMV) | (1 << NEARMV) | (1 << ZEROMV),
39*fb1b10abSAndroid Build Coastguard Worker };
40*fb1b10abSAndroid Build Coastguard Worker 
41*fb1b10abSAndroid Build Coastguard Worker enum {
42*fb1b10abSAndroid Build Coastguard Worker   DISABLE_ALL_INTER_SPLIT = (1 << THR_COMP_GA) | (1 << THR_COMP_LA) |
43*fb1b10abSAndroid Build Coastguard Worker                             (1 << THR_ALTR) | (1 << THR_GOLD) | (1 << THR_LAST),
44*fb1b10abSAndroid Build Coastguard Worker 
45*fb1b10abSAndroid Build Coastguard Worker   DISABLE_ALL_SPLIT = (1 << THR_INTRA) | DISABLE_ALL_INTER_SPLIT,
46*fb1b10abSAndroid Build Coastguard Worker 
47*fb1b10abSAndroid Build Coastguard Worker   DISABLE_COMPOUND_SPLIT = (1 << THR_COMP_GA) | (1 << THR_COMP_LA),
48*fb1b10abSAndroid Build Coastguard Worker 
49*fb1b10abSAndroid Build Coastguard Worker   LAST_AND_INTRA_SPLIT_ONLY = (1 << THR_COMP_GA) | (1 << THR_COMP_LA) |
50*fb1b10abSAndroid Build Coastguard Worker                               (1 << THR_ALTR) | (1 << THR_GOLD)
51*fb1b10abSAndroid Build Coastguard Worker };
52*fb1b10abSAndroid Build Coastguard Worker 
53*fb1b10abSAndroid Build Coastguard Worker typedef enum {
54*fb1b10abSAndroid Build Coastguard Worker   DIAMOND = 0,
55*fb1b10abSAndroid Build Coastguard Worker   NSTEP = 1,
56*fb1b10abSAndroid Build Coastguard Worker   HEX = 2,
57*fb1b10abSAndroid Build Coastguard Worker   BIGDIA = 3,
58*fb1b10abSAndroid Build Coastguard Worker   SQUARE = 4,
59*fb1b10abSAndroid Build Coastguard Worker   FAST_HEX = 5,
60*fb1b10abSAndroid Build Coastguard Worker   FAST_DIAMOND = 6,
61*fb1b10abSAndroid Build Coastguard Worker   MESH = 7
62*fb1b10abSAndroid Build Coastguard Worker } SEARCH_METHODS;
63*fb1b10abSAndroid Build Coastguard Worker 
64*fb1b10abSAndroid Build Coastguard Worker typedef enum {
65*fb1b10abSAndroid Build Coastguard Worker   // No recode.
66*fb1b10abSAndroid Build Coastguard Worker   DISALLOW_RECODE = 0,
67*fb1b10abSAndroid Build Coastguard Worker   // Allow recode for KF and exceeding maximum frame bandwidth.
68*fb1b10abSAndroid Build Coastguard Worker   ALLOW_RECODE_KFMAXBW = 1,
69*fb1b10abSAndroid Build Coastguard Worker   // Allow recode only for KF/ARF/GF frames.
70*fb1b10abSAndroid Build Coastguard Worker   ALLOW_RECODE_KFARFGF = 2,
71*fb1b10abSAndroid Build Coastguard Worker   // Allow recode for ARF/GF/KF and first normal frame in each group.
72*fb1b10abSAndroid Build Coastguard Worker   ALLOW_RECODE_FIRST = 3,
73*fb1b10abSAndroid Build Coastguard Worker   // Allow recode for all frames based on bitrate constraints.
74*fb1b10abSAndroid Build Coastguard Worker   ALLOW_RECODE = 4,
75*fb1b10abSAndroid Build Coastguard Worker } RECODE_LOOP_TYPE;
76*fb1b10abSAndroid Build Coastguard Worker 
77*fb1b10abSAndroid Build Coastguard Worker typedef enum {
78*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_TREE = 0,
79*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_TREE_PRUNED = 1,           // Prunes 1/2-pel searches
80*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_TREE_PRUNED_MORE = 2,      // Prunes 1/2-pel searches more aggressively
81*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_TREE_PRUNED_EVENMORE = 3,  // Prunes 1/2- and 1/4-pel searches
82*fb1b10abSAndroid Build Coastguard Worker   // Other methods to come
83*fb1b10abSAndroid Build Coastguard Worker } SUBPEL_SEARCH_METHODS;
84*fb1b10abSAndroid Build Coastguard Worker 
85*fb1b10abSAndroid Build Coastguard Worker typedef enum {
86*fb1b10abSAndroid Build Coastguard Worker   NO_MOTION_THRESHOLD = 0,
87*fb1b10abSAndroid Build Coastguard Worker   LOW_MOTION_THRESHOLD = 7
88*fb1b10abSAndroid Build Coastguard Worker } MOTION_THRESHOLD;
89*fb1b10abSAndroid Build Coastguard Worker 
90*fb1b10abSAndroid Build Coastguard Worker typedef enum {
91*fb1b10abSAndroid Build Coastguard Worker   USE_FULL_RD = 0,
92*fb1b10abSAndroid Build Coastguard Worker   USE_LARGESTALL,
93*fb1b10abSAndroid Build Coastguard Worker   USE_TX_8X8
94*fb1b10abSAndroid Build Coastguard Worker } TX_SIZE_SEARCH_METHOD;
95*fb1b10abSAndroid Build Coastguard Worker 
96*fb1b10abSAndroid Build Coastguard Worker typedef enum {
97*fb1b10abSAndroid Build Coastguard Worker   NOT_IN_USE = 0,
98*fb1b10abSAndroid Build Coastguard Worker   RELAXED_NEIGHBORING_MIN_MAX = 1,
99*fb1b10abSAndroid Build Coastguard Worker   STRICT_NEIGHBORING_MIN_MAX = 2
100*fb1b10abSAndroid Build Coastguard Worker } AUTO_MIN_MAX_MODE;
101*fb1b10abSAndroid Build Coastguard Worker 
102*fb1b10abSAndroid Build Coastguard Worker typedef enum {
103*fb1b10abSAndroid Build Coastguard Worker   // Try the full image with different values.
104*fb1b10abSAndroid Build Coastguard Worker   LPF_PICK_FROM_FULL_IMAGE,
105*fb1b10abSAndroid Build Coastguard Worker   // Try a small portion of the image with different values.
106*fb1b10abSAndroid Build Coastguard Worker   LPF_PICK_FROM_SUBIMAGE,
107*fb1b10abSAndroid Build Coastguard Worker   // Estimate the level based on quantizer and frame type
108*fb1b10abSAndroid Build Coastguard Worker   LPF_PICK_FROM_Q,
109*fb1b10abSAndroid Build Coastguard Worker   // Pick 0 to disable LPF if LPF was enabled last frame
110*fb1b10abSAndroid Build Coastguard Worker   LPF_PICK_MINIMAL_LPF
111*fb1b10abSAndroid Build Coastguard Worker } LPF_PICK_METHOD;
112*fb1b10abSAndroid Build Coastguard Worker 
113*fb1b10abSAndroid Build Coastguard Worker typedef enum {
114*fb1b10abSAndroid Build Coastguard Worker   // Terminate search early based on distortion so far compared to
115*fb1b10abSAndroid Build Coastguard Worker   // qp step, distortion in the neighborhood of the frame, etc.
116*fb1b10abSAndroid Build Coastguard Worker   FLAG_EARLY_TERMINATE = 1 << 0,
117*fb1b10abSAndroid Build Coastguard Worker 
118*fb1b10abSAndroid Build Coastguard Worker   // Skips comp inter modes if the best so far is an intra mode.
119*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_COMP_BESTINTRA = 1 << 1,
120*fb1b10abSAndroid Build Coastguard Worker 
121*fb1b10abSAndroid Build Coastguard Worker   // Skips oblique intra modes if the best so far is an inter mode.
122*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_INTRA_BESTINTER = 1 << 3,
123*fb1b10abSAndroid Build Coastguard Worker 
124*fb1b10abSAndroid Build Coastguard Worker   // Skips oblique intra modes  at angles 27, 63, 117, 153 if the best
125*fb1b10abSAndroid Build Coastguard Worker   // intra so far is not one of the neighboring directions.
126*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_INTRA_DIRMISMATCH = 1 << 4,
127*fb1b10abSAndroid Build Coastguard Worker 
128*fb1b10abSAndroid Build Coastguard Worker   // Skips intra modes other than DC_PRED if the source variance is small
129*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_INTRA_LOWVAR = 1 << 5,
130*fb1b10abSAndroid Build Coastguard Worker } MODE_SEARCH_SKIP_LOGIC;
131*fb1b10abSAndroid Build Coastguard Worker 
132*fb1b10abSAndroid Build Coastguard Worker typedef enum {
133*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_EIGHTTAP = 1 << EIGHTTAP,
134*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_EIGHTTAP_SMOOTH = 1 << EIGHTTAP_SMOOTH,
135*fb1b10abSAndroid Build Coastguard Worker   FLAG_SKIP_EIGHTTAP_SHARP = 1 << EIGHTTAP_SHARP,
136*fb1b10abSAndroid Build Coastguard Worker } INTERP_FILTER_MASK;
137*fb1b10abSAndroid Build Coastguard Worker 
138*fb1b10abSAndroid Build Coastguard Worker typedef enum {
139*fb1b10abSAndroid Build Coastguard Worker   // Search partitions using RD/NONRD criterion.
140*fb1b10abSAndroid Build Coastguard Worker   SEARCH_PARTITION,
141*fb1b10abSAndroid Build Coastguard Worker 
142*fb1b10abSAndroid Build Coastguard Worker   // Always use a fixed size partition.
143*fb1b10abSAndroid Build Coastguard Worker   FIXED_PARTITION,
144*fb1b10abSAndroid Build Coastguard Worker 
145*fb1b10abSAndroid Build Coastguard Worker   REFERENCE_PARTITION,
146*fb1b10abSAndroid Build Coastguard Worker 
147*fb1b10abSAndroid Build Coastguard Worker   // Use an arbitrary partitioning scheme based on source variance within
148*fb1b10abSAndroid Build Coastguard Worker   // a 64X64 SB.
149*fb1b10abSAndroid Build Coastguard Worker   VAR_BASED_PARTITION,
150*fb1b10abSAndroid Build Coastguard Worker 
151*fb1b10abSAndroid Build Coastguard Worker   // Use non-fixed partitions based on source variance.
152*fb1b10abSAndroid Build Coastguard Worker   SOURCE_VAR_BASED_PARTITION,
153*fb1b10abSAndroid Build Coastguard Worker 
154*fb1b10abSAndroid Build Coastguard Worker   // Make partition decisions with machine learning models.
155*fb1b10abSAndroid Build Coastguard Worker   ML_BASED_PARTITION
156*fb1b10abSAndroid Build Coastguard Worker } PARTITION_SEARCH_TYPE;
157*fb1b10abSAndroid Build Coastguard Worker 
158*fb1b10abSAndroid Build Coastguard Worker typedef enum {
159*fb1b10abSAndroid Build Coastguard Worker   // Does a dry run to see if any of the contexts need to be updated or not,
160*fb1b10abSAndroid Build Coastguard Worker   // before the final run.
161*fb1b10abSAndroid Build Coastguard Worker   TWO_LOOP = 0,
162*fb1b10abSAndroid Build Coastguard Worker 
163*fb1b10abSAndroid Build Coastguard Worker   // No dry run, also only half the coef contexts and bands are updated.
164*fb1b10abSAndroid Build Coastguard Worker   // The rest are not updated at all.
165*fb1b10abSAndroid Build Coastguard Worker   ONE_LOOP_REDUCED = 1
166*fb1b10abSAndroid Build Coastguard Worker } FAST_COEFF_UPDATE;
167*fb1b10abSAndroid Build Coastguard Worker 
168*fb1b10abSAndroid Build Coastguard Worker typedef enum { EIGHTH_PEL, QUARTER_PEL, HALF_PEL, FULL_PEL } SUBPEL_FORCE_STOP;
169*fb1b10abSAndroid Build Coastguard Worker 
170*fb1b10abSAndroid Build Coastguard Worker typedef struct ADAPT_SUBPEL_FORCE_STOP {
171*fb1b10abSAndroid Build Coastguard Worker   // Threshold for full pixel motion vector;
172*fb1b10abSAndroid Build Coastguard Worker   int mv_thresh;
173*fb1b10abSAndroid Build Coastguard Worker 
174*fb1b10abSAndroid Build Coastguard Worker   // subpel_force_stop if full pixel MV is below the threshold.
175*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_FORCE_STOP force_stop_below;
176*fb1b10abSAndroid Build Coastguard Worker 
177*fb1b10abSAndroid Build Coastguard Worker   // subpel_force_stop if full pixel MV is equal to or above the threshold.
178*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_FORCE_STOP force_stop_above;
179*fb1b10abSAndroid Build Coastguard Worker } ADAPT_SUBPEL_FORCE_STOP;
180*fb1b10abSAndroid Build Coastguard Worker 
181*fb1b10abSAndroid Build Coastguard Worker typedef struct MV_SPEED_FEATURES {
182*fb1b10abSAndroid Build Coastguard Worker   // Motion search method (Diamond, NSTEP, Hex, Big Diamond, Square, etc).
183*fb1b10abSAndroid Build Coastguard Worker   SEARCH_METHODS search_method;
184*fb1b10abSAndroid Build Coastguard Worker 
185*fb1b10abSAndroid Build Coastguard Worker   // This parameter controls which step in the n-step process we start at.
186*fb1b10abSAndroid Build Coastguard Worker   // It's changed adaptively based on circumstances.
187*fb1b10abSAndroid Build Coastguard Worker   int reduce_first_step_size;
188*fb1b10abSAndroid Build Coastguard Worker 
189*fb1b10abSAndroid Build Coastguard Worker   // If this is set to 1, we limit the motion search range to 2 times the
190*fb1b10abSAndroid Build Coastguard Worker   // largest motion vector found in the last frame.
191*fb1b10abSAndroid Build Coastguard Worker   int auto_mv_step_size;
192*fb1b10abSAndroid Build Coastguard Worker 
193*fb1b10abSAndroid Build Coastguard Worker   // Subpel_search_method can only be subpel_tree which does a subpixel
194*fb1b10abSAndroid Build Coastguard Worker   // logarithmic search that keeps stepping at 1/2 pixel units until
195*fb1b10abSAndroid Build Coastguard Worker   // you stop getting a gain, and then goes on to 1/4 and repeats
196*fb1b10abSAndroid Build Coastguard Worker   // the same process. Along the way it skips many diagonals.
197*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_SEARCH_METHODS subpel_search_method;
198*fb1b10abSAndroid Build Coastguard Worker 
199*fb1b10abSAndroid Build Coastguard Worker   // Subpel MV search level. Can take values 0 - 2. Higher values mean more
200*fb1b10abSAndroid Build Coastguard Worker   // extensive subpel search.
201*fb1b10abSAndroid Build Coastguard Worker   int subpel_search_level;
202*fb1b10abSAndroid Build Coastguard Worker 
203*fb1b10abSAndroid Build Coastguard Worker   // When to stop subpel motion search.
204*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_FORCE_STOP subpel_force_stop;
205*fb1b10abSAndroid Build Coastguard Worker 
206*fb1b10abSAndroid Build Coastguard Worker   // If it's enabled, different subpel_force_stop will be used for different MV.
207*fb1b10abSAndroid Build Coastguard Worker   int enable_adaptive_subpel_force_stop;
208*fb1b10abSAndroid Build Coastguard Worker 
209*fb1b10abSAndroid Build Coastguard Worker   ADAPT_SUBPEL_FORCE_STOP adapt_subpel_force_stop;
210*fb1b10abSAndroid Build Coastguard Worker 
211*fb1b10abSAndroid Build Coastguard Worker   // This variable sets the step_param used in full pel motion search.
212*fb1b10abSAndroid Build Coastguard Worker   int fullpel_search_step_param;
213*fb1b10abSAndroid Build Coastguard Worker 
214*fb1b10abSAndroid Build Coastguard Worker   // Whether to downsample the rows in sad calculation during motion search.
215*fb1b10abSAndroid Build Coastguard Worker   // This is only active when there are at least 8 rows.
216*fb1b10abSAndroid Build Coastguard Worker   int use_downsampled_sad;
217*fb1b10abSAndroid Build Coastguard Worker } MV_SPEED_FEATURES;
218*fb1b10abSAndroid Build Coastguard Worker 
219*fb1b10abSAndroid Build Coastguard Worker typedef struct PARTITION_SEARCH_BREAKOUT_THR {
220*fb1b10abSAndroid Build Coastguard Worker   int64_t dist;
221*fb1b10abSAndroid Build Coastguard Worker   int rate;
222*fb1b10abSAndroid Build Coastguard Worker } PARTITION_SEARCH_BREAKOUT_THR;
223*fb1b10abSAndroid Build Coastguard Worker 
224*fb1b10abSAndroid Build Coastguard Worker #define MAX_MESH_STEP 4
225*fb1b10abSAndroid Build Coastguard Worker 
226*fb1b10abSAndroid Build Coastguard Worker typedef struct MESH_PATTERN {
227*fb1b10abSAndroid Build Coastguard Worker   int range;
228*fb1b10abSAndroid Build Coastguard Worker   int interval;
229*fb1b10abSAndroid Build Coastguard Worker } MESH_PATTERN;
230*fb1b10abSAndroid Build Coastguard Worker 
231*fb1b10abSAndroid Build Coastguard Worker typedef enum {
232*fb1b10abSAndroid Build Coastguard Worker   // No reaction to rate control on a detected slide/scene change.
233*fb1b10abSAndroid Build Coastguard Worker   NO_DETECTION = 0,
234*fb1b10abSAndroid Build Coastguard Worker 
235*fb1b10abSAndroid Build Coastguard Worker   // Set to larger Q (max_q set by user) based only on the
236*fb1b10abSAndroid Build Coastguard Worker   // detected slide/scene change and current/past Q.
237*fb1b10abSAndroid Build Coastguard Worker   FAST_DETECTION_MAXQ = 1,
238*fb1b10abSAndroid Build Coastguard Worker 
239*fb1b10abSAndroid Build Coastguard Worker   // Based on (first pass) encoded frame, if large frame size is detected
240*fb1b10abSAndroid Build Coastguard Worker   // then set to higher Q for the second re-encode. This involves 2 pass
241*fb1b10abSAndroid Build Coastguard Worker   // encoding on slide change, so slower than 1, but more accurate for
242*fb1b10abSAndroid Build Coastguard Worker   // detecting overshoot.
243*fb1b10abSAndroid Build Coastguard Worker   RE_ENCODE_MAXQ = 2
244*fb1b10abSAndroid Build Coastguard Worker } OVERSHOOT_DETECTION_CBR_RT;
245*fb1b10abSAndroid Build Coastguard Worker 
246*fb1b10abSAndroid Build Coastguard Worker typedef enum {
247*fb1b10abSAndroid Build Coastguard Worker   USE_2_TAPS = 0,
248*fb1b10abSAndroid Build Coastguard Worker   USE_4_TAPS,
249*fb1b10abSAndroid Build Coastguard Worker   USE_8_TAPS,
250*fb1b10abSAndroid Build Coastguard Worker   USE_8_TAPS_SHARP,
251*fb1b10abSAndroid Build Coastguard Worker } SUBPEL_SEARCH_TYPE;
252*fb1b10abSAndroid Build Coastguard Worker 
253*fb1b10abSAndroid Build Coastguard Worker typedef enum {
254*fb1b10abSAndroid Build Coastguard Worker   // Disable trellis coefficient optimization
255*fb1b10abSAndroid Build Coastguard Worker   DISABLE_TRELLIS_OPT,
256*fb1b10abSAndroid Build Coastguard Worker   // Enable trellis coefficient optimization
257*fb1b10abSAndroid Build Coastguard Worker   ENABLE_TRELLIS_OPT,
258*fb1b10abSAndroid Build Coastguard Worker   // Enable trellis coefficient optimization based on source variance of the
259*fb1b10abSAndroid Build Coastguard Worker   // prediction block during transform RD
260*fb1b10abSAndroid Build Coastguard Worker   ENABLE_TRELLIS_OPT_TX_RD_SRC_VAR,
261*fb1b10abSAndroid Build Coastguard Worker   // Enable trellis coefficient optimization based on residual mse of the
262*fb1b10abSAndroid Build Coastguard Worker   // transform block during transform RD
263*fb1b10abSAndroid Build Coastguard Worker   ENABLE_TRELLIS_OPT_TX_RD_RESIDUAL_MSE,
264*fb1b10abSAndroid Build Coastguard Worker } ENABLE_TRELLIS_OPT_METHOD;
265*fb1b10abSAndroid Build Coastguard Worker 
266*fb1b10abSAndroid Build Coastguard Worker typedef struct TRELLIS_OPT_CONTROL {
267*fb1b10abSAndroid Build Coastguard Worker   ENABLE_TRELLIS_OPT_METHOD method;
268*fb1b10abSAndroid Build Coastguard Worker   double thresh;
269*fb1b10abSAndroid Build Coastguard Worker } TRELLIS_OPT_CONTROL;
270*fb1b10abSAndroid Build Coastguard Worker 
271*fb1b10abSAndroid Build Coastguard Worker typedef struct SPEED_FEATURES {
272*fb1b10abSAndroid Build Coastguard Worker   MV_SPEED_FEATURES mv;
273*fb1b10abSAndroid Build Coastguard Worker 
274*fb1b10abSAndroid Build Coastguard Worker   // Frame level coding parameter update
275*fb1b10abSAndroid Build Coastguard Worker   int frame_parameter_update;
276*fb1b10abSAndroid Build Coastguard Worker 
277*fb1b10abSAndroid Build Coastguard Worker   RECODE_LOOP_TYPE recode_loop;
278*fb1b10abSAndroid Build Coastguard Worker 
279*fb1b10abSAndroid Build Coastguard Worker   // Trellis (dynamic programming) optimization of quantized values (+1, 0).
280*fb1b10abSAndroid Build Coastguard Worker   int optimize_coefficients;
281*fb1b10abSAndroid Build Coastguard Worker 
282*fb1b10abSAndroid Build Coastguard Worker   // Always set to 0. If on it enables 0 cost background transmission
283*fb1b10abSAndroid Build Coastguard Worker   // (except for the initial transmission of the segmentation). The feature is
284*fb1b10abSAndroid Build Coastguard Worker   // disabled because the addition of very large block sizes make the
285*fb1b10abSAndroid Build Coastguard Worker   // backgrounds very to cheap to encode, and the segmentation we have
286*fb1b10abSAndroid Build Coastguard Worker   // adds overhead.
287*fb1b10abSAndroid Build Coastguard Worker   int static_segmentation;
288*fb1b10abSAndroid Build Coastguard Worker 
289*fb1b10abSAndroid Build Coastguard Worker   // The best compound predictor is found using an iterative log search process
290*fb1b10abSAndroid Build Coastguard Worker   // that searches for best ref0 mv using error of combined predictor and then
291*fb1b10abSAndroid Build Coastguard Worker   // searches for best ref1 mv. This sf determines the number of iterations of
292*fb1b10abSAndroid Build Coastguard Worker   // this process based on block size. The sf becomes more aggressive from level
293*fb1b10abSAndroid Build Coastguard Worker   // 0 to 2. The following table indicates the number of iterations w.r.t bsize:
294*fb1b10abSAndroid Build Coastguard Worker   //  -----------------------------------------------
295*fb1b10abSAndroid Build Coastguard Worker   // |sf (level)|bsize < 8X8| [8X8, 16X16] | > 16X16 |
296*fb1b10abSAndroid Build Coastguard Worker   // |    0     |     4     |      4       |    4    |
297*fb1b10abSAndroid Build Coastguard Worker   // |    1     |     0     |      2       |    4    |
298*fb1b10abSAndroid Build Coastguard Worker   // |    2     |     0     |      0       |    0    |
299*fb1b10abSAndroid Build Coastguard Worker   //  -----------------------------------------------
300*fb1b10abSAndroid Build Coastguard Worker   // Here, 0 iterations indicate using the best single motion vector selected
301*fb1b10abSAndroid Build Coastguard Worker   // for each ref frame without any iterative refinement.
302*fb1b10abSAndroid Build Coastguard Worker   int comp_inter_joint_search_iter_level;
303*fb1b10abSAndroid Build Coastguard Worker 
304*fb1b10abSAndroid Build Coastguard Worker   // This variable is used to cap the maximum number of times we skip testing a
305*fb1b10abSAndroid Build Coastguard Worker   // mode to be evaluated. A high value means we will be faster.
306*fb1b10abSAndroid Build Coastguard Worker   // Turned off when (row_mt_bit_exact == 1 && adaptive_rd_thresh_row_mt == 0).
307*fb1b10abSAndroid Build Coastguard Worker   int adaptive_rd_thresh;
308*fb1b10abSAndroid Build Coastguard Worker 
309*fb1b10abSAndroid Build Coastguard Worker   // Flag to use adaptive_rd_thresh when row-mt is enabled, only for non-rd
310*fb1b10abSAndroid Build Coastguard Worker   // pickmode.
311*fb1b10abSAndroid Build Coastguard Worker   int adaptive_rd_thresh_row_mt;
312*fb1b10abSAndroid Build Coastguard Worker 
313*fb1b10abSAndroid Build Coastguard Worker   // Enables skipping the reconstruction step (idct, recon) in the
314*fb1b10abSAndroid Build Coastguard Worker   // intermediate steps assuming the last frame didn't have too many intra
315*fb1b10abSAndroid Build Coastguard Worker   // blocks and the q is less than a threshold.
316*fb1b10abSAndroid Build Coastguard Worker   int skip_encode_sb;
317*fb1b10abSAndroid Build Coastguard Worker   int skip_encode_frame;
318*fb1b10abSAndroid Build Coastguard Worker   // Speed feature to allow or disallow skipping of recode at block
319*fb1b10abSAndroid Build Coastguard Worker   // level within a frame.
320*fb1b10abSAndroid Build Coastguard Worker   int allow_skip_recode;
321*fb1b10abSAndroid Build Coastguard Worker 
322*fb1b10abSAndroid Build Coastguard Worker   // Coefficient probability model approximation step size
323*fb1b10abSAndroid Build Coastguard Worker   int coeff_prob_appx_step;
324*fb1b10abSAndroid Build Coastguard Worker 
325*fb1b10abSAndroid Build Coastguard Worker   // Enable uniform quantizer followed by trellis coefficient optimization
326*fb1b10abSAndroid Build Coastguard Worker   // during transform RD
327*fb1b10abSAndroid Build Coastguard Worker   TRELLIS_OPT_CONTROL trellis_opt_tx_rd;
328*fb1b10abSAndroid Build Coastguard Worker 
329*fb1b10abSAndroid Build Coastguard Worker   // Enable asymptotic closed-loop encoding decision for key frame and
330*fb1b10abSAndroid Build Coastguard Worker   // alternate reference frames.
331*fb1b10abSAndroid Build Coastguard Worker   int allow_acl;
332*fb1b10abSAndroid Build Coastguard Worker 
333*fb1b10abSAndroid Build Coastguard Worker   // Temporal dependency model based encoding mode optimization
334*fb1b10abSAndroid Build Coastguard Worker   int enable_tpl_model;
335*fb1b10abSAndroid Build Coastguard Worker 
336*fb1b10abSAndroid Build Coastguard Worker   // Use transform domain distortion. Use pixel domain distortion in speed 0
337*fb1b10abSAndroid Build Coastguard Worker   // and certain situations in higher speed to improve the RD model precision.
338*fb1b10abSAndroid Build Coastguard Worker   int allow_txfm_domain_distortion;
339*fb1b10abSAndroid Build Coastguard Worker   double tx_domain_thresh;
340*fb1b10abSAndroid Build Coastguard Worker 
341*fb1b10abSAndroid Build Coastguard Worker   // The threshold is to determine how slow the motino is, it is used when
342*fb1b10abSAndroid Build Coastguard Worker   // use_lastframe_partitioning is set to LAST_FRAME_PARTITION_LOW_MOTION
343*fb1b10abSAndroid Build Coastguard Worker   MOTION_THRESHOLD lf_motion_threshold;
344*fb1b10abSAndroid Build Coastguard Worker 
345*fb1b10abSAndroid Build Coastguard Worker   // Determine which method we use to determine transform size. We can choose
346*fb1b10abSAndroid Build Coastguard Worker   // between options like full rd, largest for prediction size, largest
347*fb1b10abSAndroid Build Coastguard Worker   // for intra and model coefs for the rest.
348*fb1b10abSAndroid Build Coastguard Worker   TX_SIZE_SEARCH_METHOD tx_size_search_method;
349*fb1b10abSAndroid Build Coastguard Worker 
350*fb1b10abSAndroid Build Coastguard Worker   // How many levels of tx size to search, starting from the largest.
351*fb1b10abSAndroid Build Coastguard Worker   int tx_size_search_depth;
352*fb1b10abSAndroid Build Coastguard Worker 
353*fb1b10abSAndroid Build Coastguard Worker   // Low precision 32x32 fdct keeps everything in 16 bits and thus is less
354*fb1b10abSAndroid Build Coastguard Worker   // precise but significantly faster than the non lp version.
355*fb1b10abSAndroid Build Coastguard Worker   int use_lp32x32fdct;
356*fb1b10abSAndroid Build Coastguard Worker 
357*fb1b10abSAndroid Build Coastguard Worker   // After looking at the first set of modes (set by index here), skip
358*fb1b10abSAndroid Build Coastguard Worker   // checking modes for reference frames that don't match the reference frame
359*fb1b10abSAndroid Build Coastguard Worker   // of the best so far.
360*fb1b10abSAndroid Build Coastguard Worker   int mode_skip_start;
361*fb1b10abSAndroid Build Coastguard Worker 
362*fb1b10abSAndroid Build Coastguard Worker   // TODO(JBB): Remove this.
363*fb1b10abSAndroid Build Coastguard Worker   int reference_masking;
364*fb1b10abSAndroid Build Coastguard Worker 
365*fb1b10abSAndroid Build Coastguard Worker   PARTITION_SEARCH_TYPE partition_search_type;
366*fb1b10abSAndroid Build Coastguard Worker 
367*fb1b10abSAndroid Build Coastguard Worker   // Used if partition_search_type = FIXED_PARTITION
368*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE always_this_block_size;
369*fb1b10abSAndroid Build Coastguard Worker 
370*fb1b10abSAndroid Build Coastguard Worker   // Skip rectangular partition test when partition type none gives better
371*fb1b10abSAndroid Build Coastguard Worker   // rd than partition type split.
372*fb1b10abSAndroid Build Coastguard Worker   int less_rectangular_check;
373*fb1b10abSAndroid Build Coastguard Worker 
374*fb1b10abSAndroid Build Coastguard Worker   // Disable testing non square partitions(eg 16x32) for block sizes larger than
375*fb1b10abSAndroid Build Coastguard Worker   // use_square_only_thresh_high or smaller than use_square_only_thresh_low.
376*fb1b10abSAndroid Build Coastguard Worker   int use_square_partition_only;
377*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE use_square_only_thresh_high;
378*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE use_square_only_thresh_low;
379*fb1b10abSAndroid Build Coastguard Worker 
380*fb1b10abSAndroid Build Coastguard Worker   // Prune reference frames for rectangular partitions.
381*fb1b10abSAndroid Build Coastguard Worker   int prune_ref_frame_for_rect_partitions;
382*fb1b10abSAndroid Build Coastguard Worker 
383*fb1b10abSAndroid Build Coastguard Worker   // Sets min and max partition sizes for this 64x64 region based on the
384*fb1b10abSAndroid Build Coastguard Worker   // same 64x64 in last encoded frame, and the left and above neighbor.
385*fb1b10abSAndroid Build Coastguard Worker   AUTO_MIN_MAX_MODE auto_min_max_partition_size;
386*fb1b10abSAndroid Build Coastguard Worker   // Ensures the rd based auto partition search will always
387*fb1b10abSAndroid Build Coastguard Worker   // go down at least to the specified level.
388*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE rd_auto_partition_min_limit;
389*fb1b10abSAndroid Build Coastguard Worker 
390*fb1b10abSAndroid Build Coastguard Worker   // Min and max partition size we enable (block_size) as per auto
391*fb1b10abSAndroid Build Coastguard Worker   // min max, but also used by adjust partitioning, and pick_partitioning.
392*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE default_min_partition_size;
393*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE default_max_partition_size;
394*fb1b10abSAndroid Build Coastguard Worker 
395*fb1b10abSAndroid Build Coastguard Worker   // Whether or not we allow partitions one smaller or one greater than the last
396*fb1b10abSAndroid Build Coastguard Worker   // frame's partitioning. Only used if use_lastframe_partitioning is set.
397*fb1b10abSAndroid Build Coastguard Worker   int adjust_partitioning_from_last_frame;
398*fb1b10abSAndroid Build Coastguard Worker 
399*fb1b10abSAndroid Build Coastguard Worker   // How frequently we re do the partitioning from scratch. Only used if
400*fb1b10abSAndroid Build Coastguard Worker   // use_lastframe_partitioning is set.
401*fb1b10abSAndroid Build Coastguard Worker   int last_partitioning_redo_frequency;
402*fb1b10abSAndroid Build Coastguard Worker 
403*fb1b10abSAndroid Build Coastguard Worker   // Disables sub 8x8 blocksizes in different scenarios: Choices are to disable
404*fb1b10abSAndroid Build Coastguard Worker   // it always, to allow it for only Last frame and Intra, disable it for all
405*fb1b10abSAndroid Build Coastguard Worker   // inter modes or to enable it always.
406*fb1b10abSAndroid Build Coastguard Worker   int disable_split_mask;
407*fb1b10abSAndroid Build Coastguard Worker 
408*fb1b10abSAndroid Build Coastguard Worker   // TODO(jingning): combine the related motion search speed features
409*fb1b10abSAndroid Build Coastguard Worker   // This allows us to use motion search at other sizes as a starting
410*fb1b10abSAndroid Build Coastguard Worker   // point for this motion search and limits the search range around it.
411*fb1b10abSAndroid Build Coastguard Worker   int adaptive_motion_search;
412*fb1b10abSAndroid Build Coastguard Worker 
413*fb1b10abSAndroid Build Coastguard Worker   // Do extra full pixel motion search to obtain better motion vector.
414*fb1b10abSAndroid Build Coastguard Worker   int enhanced_full_pixel_motion_search;
415*fb1b10abSAndroid Build Coastguard Worker 
416*fb1b10abSAndroid Build Coastguard Worker   // Threshold for allowing exhaistive motion search.
417*fb1b10abSAndroid Build Coastguard Worker   int exhaustive_searches_thresh;
418*fb1b10abSAndroid Build Coastguard Worker 
419*fb1b10abSAndroid Build Coastguard Worker   // Pattern to be used for any exhaustive mesh searches.
420*fb1b10abSAndroid Build Coastguard Worker   MESH_PATTERN mesh_patterns[MAX_MESH_STEP];
421*fb1b10abSAndroid Build Coastguard Worker 
422*fb1b10abSAndroid Build Coastguard Worker   int schedule_mode_search;
423*fb1b10abSAndroid Build Coastguard Worker 
424*fb1b10abSAndroid Build Coastguard Worker   // Allows sub 8x8 modes to use the prediction filter that was determined
425*fb1b10abSAndroid Build Coastguard Worker   // best for 8x8 mode. If set to 0 we always re check all the filters for
426*fb1b10abSAndroid Build Coastguard Worker   // sizes less than 8x8, 1 means we check all filter modes if no 8x8 filter
427*fb1b10abSAndroid Build Coastguard Worker   // was selected, and 2 means we use 8 tap if no 8x8 filter mode was selected.
428*fb1b10abSAndroid Build Coastguard Worker   int adaptive_pred_interp_filter;
429*fb1b10abSAndroid Build Coastguard Worker 
430*fb1b10abSAndroid Build Coastguard Worker   // Adaptive prediction mode search
431*fb1b10abSAndroid Build Coastguard Worker   int adaptive_mode_search;
432*fb1b10abSAndroid Build Coastguard Worker 
433*fb1b10abSAndroid Build Coastguard Worker   // Prune NEAREST and ZEROMV single reference modes based on motion vector
434*fb1b10abSAndroid Build Coastguard Worker   // difference and mode rate
435*fb1b10abSAndroid Build Coastguard Worker   int prune_single_mode_based_on_mv_diff_mode_rate;
436*fb1b10abSAndroid Build Coastguard Worker 
437*fb1b10abSAndroid Build Coastguard Worker   // Chessboard pattern prediction for interp filter. Aggressiveness increases
438*fb1b10abSAndroid Build Coastguard Worker   // with levels.
439*fb1b10abSAndroid Build Coastguard Worker   // 0: disable
440*fb1b10abSAndroid Build Coastguard Worker   // 1: cb pattern in eval when filter is not switchable
441*fb1b10abSAndroid Build Coastguard Worker   // 2: cb pattern prediction for filter search
442*fb1b10abSAndroid Build Coastguard Worker   int cb_pred_filter_search;
443*fb1b10abSAndroid Build Coastguard Worker 
444*fb1b10abSAndroid Build Coastguard Worker   // This variable enables an early termination of interpolation filter eval
445*fb1b10abSAndroid Build Coastguard Worker   // based on the current rd cost after processing each plane
446*fb1b10abSAndroid Build Coastguard Worker   int early_term_interp_search_plane_rd;
447*fb1b10abSAndroid Build Coastguard Worker 
448*fb1b10abSAndroid Build Coastguard Worker   int cb_partition_search;
449*fb1b10abSAndroid Build Coastguard Worker 
450*fb1b10abSAndroid Build Coastguard Worker   int motion_field_mode_search;
451*fb1b10abSAndroid Build Coastguard Worker 
452*fb1b10abSAndroid Build Coastguard Worker   int alt_ref_search_fp;
453*fb1b10abSAndroid Build Coastguard Worker 
454*fb1b10abSAndroid Build Coastguard Worker   // Fast quantization process path
455*fb1b10abSAndroid Build Coastguard Worker   int use_quant_fp;
456*fb1b10abSAndroid Build Coastguard Worker 
457*fb1b10abSAndroid Build Coastguard Worker   // Use finer quantizer in every other few frames that run variable block
458*fb1b10abSAndroid Build Coastguard Worker   // partition type search.
459*fb1b10abSAndroid Build Coastguard Worker   int force_frame_boost;
460*fb1b10abSAndroid Build Coastguard Worker 
461*fb1b10abSAndroid Build Coastguard Worker   // Maximally allowed base quantization index fluctuation.
462*fb1b10abSAndroid Build Coastguard Worker   int max_delta_qindex;
463*fb1b10abSAndroid Build Coastguard Worker 
464*fb1b10abSAndroid Build Coastguard Worker   // Implements various heuristics to skip searching modes
465*fb1b10abSAndroid Build Coastguard Worker   // The heuristics selected are based on  flags
466*fb1b10abSAndroid Build Coastguard Worker   // defined in the MODE_SEARCH_SKIP_HEURISTICS enum
467*fb1b10abSAndroid Build Coastguard Worker   unsigned int mode_search_skip_flags;
468*fb1b10abSAndroid Build Coastguard Worker 
469*fb1b10abSAndroid Build Coastguard Worker   // A source variance threshold below which filter search is disabled
470*fb1b10abSAndroid Build Coastguard Worker   // Choose a very large value (UINT_MAX) to use 8-tap always
471*fb1b10abSAndroid Build Coastguard Worker   unsigned int disable_filter_search_var_thresh;
472*fb1b10abSAndroid Build Coastguard Worker 
473*fb1b10abSAndroid Build Coastguard Worker   // These bit masks allow you to enable or disable intra modes for each
474*fb1b10abSAndroid Build Coastguard Worker   // transform size separately.
475*fb1b10abSAndroid Build Coastguard Worker   int intra_y_mode_mask[TX_SIZES];
476*fb1b10abSAndroid Build Coastguard Worker   int intra_uv_mode_mask[TX_SIZES];
477*fb1b10abSAndroid Build Coastguard Worker 
478*fb1b10abSAndroid Build Coastguard Worker   // These bit masks allow you to enable or disable intra modes for each
479*fb1b10abSAndroid Build Coastguard Worker   // prediction block size separately.
480*fb1b10abSAndroid Build Coastguard Worker   int intra_y_mode_bsize_mask[BLOCK_SIZES];
481*fb1b10abSAndroid Build Coastguard Worker 
482*fb1b10abSAndroid Build Coastguard Worker   // This variable enables an early break out of mode testing if the model for
483*fb1b10abSAndroid Build Coastguard Worker   // rd built from the prediction signal indicates a value that's much
484*fb1b10abSAndroid Build Coastguard Worker   // higher than the best rd we've seen so far.
485*fb1b10abSAndroid Build Coastguard Worker   int use_rd_breakout;
486*fb1b10abSAndroid Build Coastguard Worker 
487*fb1b10abSAndroid Build Coastguard Worker   // This enables us to use an estimate for intra rd based on dc mode rather
488*fb1b10abSAndroid Build Coastguard Worker   // than choosing an actual uv mode in the stage of encoding before the actual
489*fb1b10abSAndroid Build Coastguard Worker   // final encode.
490*fb1b10abSAndroid Build Coastguard Worker   int use_uv_intra_rd_estimate;
491*fb1b10abSAndroid Build Coastguard Worker 
492*fb1b10abSAndroid Build Coastguard Worker   // This feature controls how the loop filter level is determined.
493*fb1b10abSAndroid Build Coastguard Worker   LPF_PICK_METHOD lpf_pick;
494*fb1b10abSAndroid Build Coastguard Worker 
495*fb1b10abSAndroid Build Coastguard Worker   // This feature limits the number of coefficients updates we actually do
496*fb1b10abSAndroid Build Coastguard Worker   // by only looking at counts from 1/2 the bands.
497*fb1b10abSAndroid Build Coastguard Worker   FAST_COEFF_UPDATE use_fast_coef_updates;
498*fb1b10abSAndroid Build Coastguard Worker 
499*fb1b10abSAndroid Build Coastguard Worker   // This flag controls the use of non-RD mode decision.
500*fb1b10abSAndroid Build Coastguard Worker   int use_nonrd_pick_mode;
501*fb1b10abSAndroid Build Coastguard Worker 
502*fb1b10abSAndroid Build Coastguard Worker   // A binary mask indicating if NEARESTMV, NEARMV, ZEROMV, NEWMV
503*fb1b10abSAndroid Build Coastguard Worker   // modes are used in order from LSB to MSB for each BLOCK_SIZE.
504*fb1b10abSAndroid Build Coastguard Worker   int inter_mode_mask[BLOCK_SIZES];
505*fb1b10abSAndroid Build Coastguard Worker 
506*fb1b10abSAndroid Build Coastguard Worker   // This feature controls whether we do the expensive context update and
507*fb1b10abSAndroid Build Coastguard Worker   // calculation in the rd coefficient costing loop.
508*fb1b10abSAndroid Build Coastguard Worker   int use_fast_coef_costing;
509*fb1b10abSAndroid Build Coastguard Worker 
510*fb1b10abSAndroid Build Coastguard Worker   // This feature controls the tolerence vs target used in deciding whether to
511*fb1b10abSAndroid Build Coastguard Worker   // recode a frame. It has no meaning if recode is disabled.
512*fb1b10abSAndroid Build Coastguard Worker   int recode_tolerance_low;
513*fb1b10abSAndroid Build Coastguard Worker   int recode_tolerance_high;
514*fb1b10abSAndroid Build Coastguard Worker 
515*fb1b10abSAndroid Build Coastguard Worker   // This variable controls the maximum block size where intra blocks can be
516*fb1b10abSAndroid Build Coastguard Worker   // used in inter frames.
517*fb1b10abSAndroid Build Coastguard Worker   // TODO(aconverse): Fold this into one of the other many mode skips
518*fb1b10abSAndroid Build Coastguard Worker   BLOCK_SIZE max_intra_bsize;
519*fb1b10abSAndroid Build Coastguard Worker 
520*fb1b10abSAndroid Build Coastguard Worker   // The frequency that we check if SOURCE_VAR_BASED_PARTITION or
521*fb1b10abSAndroid Build Coastguard Worker   // FIXED_PARTITION search type should be used.
522*fb1b10abSAndroid Build Coastguard Worker   int search_type_check_frequency;
523*fb1b10abSAndroid Build Coastguard Worker 
524*fb1b10abSAndroid Build Coastguard Worker   // When partition is pre-set, the inter prediction result from pick_inter_mode
525*fb1b10abSAndroid Build Coastguard Worker   // can be reused in final block encoding process. It is enabled only for real-
526*fb1b10abSAndroid Build Coastguard Worker   // time mode speed 6.
527*fb1b10abSAndroid Build Coastguard Worker   int reuse_inter_pred_sby;
528*fb1b10abSAndroid Build Coastguard Worker 
529*fb1b10abSAndroid Build Coastguard Worker   // This variable sets the encode_breakout threshold. Currently, it is only
530*fb1b10abSAndroid Build Coastguard Worker   // enabled in real time mode.
531*fb1b10abSAndroid Build Coastguard Worker   int encode_breakout_thresh;
532*fb1b10abSAndroid Build Coastguard Worker 
533*fb1b10abSAndroid Build Coastguard Worker   // default interp filter choice
534*fb1b10abSAndroid Build Coastguard Worker   INTERP_FILTER default_interp_filter;
535*fb1b10abSAndroid Build Coastguard Worker 
536*fb1b10abSAndroid Build Coastguard Worker   // Early termination in transform size search, which only applies while
537*fb1b10abSAndroid Build Coastguard Worker   // tx_size_search_method is USE_FULL_RD.
538*fb1b10abSAndroid Build Coastguard Worker   int tx_size_search_breakout;
539*fb1b10abSAndroid Build Coastguard Worker 
540*fb1b10abSAndroid Build Coastguard Worker   // adaptive interp_filter search to allow skip of certain filter types.
541*fb1b10abSAndroid Build Coastguard Worker   int adaptive_interp_filter_search;
542*fb1b10abSAndroid Build Coastguard Worker 
543*fb1b10abSAndroid Build Coastguard Worker   // mask for skip evaluation of certain interp_filter type.
544*fb1b10abSAndroid Build Coastguard Worker   INTERP_FILTER_MASK interp_filter_search_mask;
545*fb1b10abSAndroid Build Coastguard Worker 
546*fb1b10abSAndroid Build Coastguard Worker   // Partition search early breakout thresholds.
547*fb1b10abSAndroid Build Coastguard Worker   PARTITION_SEARCH_BREAKOUT_THR partition_search_breakout_thr;
548*fb1b10abSAndroid Build Coastguard Worker 
549*fb1b10abSAndroid Build Coastguard Worker   struct {
550*fb1b10abSAndroid Build Coastguard Worker     // Use ML-based partition search early breakout.
551*fb1b10abSAndroid Build Coastguard Worker     int search_breakout;
552*fb1b10abSAndroid Build Coastguard Worker     // Higher values mean more aggressiveness for partition search breakout that
553*fb1b10abSAndroid Build Coastguard Worker     // results in better encoding  speed but worse compression performance.
554*fb1b10abSAndroid Build Coastguard Worker     float search_breakout_thresh[3];
555*fb1b10abSAndroid Build Coastguard Worker 
556*fb1b10abSAndroid Build Coastguard Worker     // Machine-learning based partition search early termination
557*fb1b10abSAndroid Build Coastguard Worker     int search_early_termination;
558*fb1b10abSAndroid Build Coastguard Worker 
559*fb1b10abSAndroid Build Coastguard Worker     // Machine-learning based partition search pruning using prediction residue
560*fb1b10abSAndroid Build Coastguard Worker     // variance.
561*fb1b10abSAndroid Build Coastguard Worker     int var_pruning;
562*fb1b10abSAndroid Build Coastguard Worker 
563*fb1b10abSAndroid Build Coastguard Worker     // Threshold values used for ML based rectangular partition search pruning.
564*fb1b10abSAndroid Build Coastguard Worker     // If < 0, the feature is turned off.
565*fb1b10abSAndroid Build Coastguard Worker     // Higher values mean more aggressiveness to skip rectangular partition
566*fb1b10abSAndroid Build Coastguard Worker     // search that results in better encoding speed but worse coding
567*fb1b10abSAndroid Build Coastguard Worker     // performance.
568*fb1b10abSAndroid Build Coastguard Worker     int prune_rect_thresh[4];
569*fb1b10abSAndroid Build Coastguard Worker   } rd_ml_partition;
570*fb1b10abSAndroid Build Coastguard Worker 
571*fb1b10abSAndroid Build Coastguard Worker   // Fast approximation of vp9_model_rd_from_var_lapndz
572*fb1b10abSAndroid Build Coastguard Worker   int simple_model_rd_from_var;
573*fb1b10abSAndroid Build Coastguard Worker 
574*fb1b10abSAndroid Build Coastguard Worker   // Skip a number of expensive mode evaluations for blocks with zero source
575*fb1b10abSAndroid Build Coastguard Worker   // variance.
576*fb1b10abSAndroid Build Coastguard Worker   int short_circuit_flat_blocks;
577*fb1b10abSAndroid Build Coastguard Worker 
578*fb1b10abSAndroid Build Coastguard Worker   // Skip a number of expensive mode evaluations for blocks with very low
579*fb1b10abSAndroid Build Coastguard Worker   // temporal variance. If the low temporal variance flag is set for a block,
580*fb1b10abSAndroid Build Coastguard Worker   // do the following:
581*fb1b10abSAndroid Build Coastguard Worker   // 1: Skip all golden modes and ALL INTRA for bsize >= 32x32.
582*fb1b10abSAndroid Build Coastguard Worker   // 2: Skip golden non-zeromv and newmv-last for bsize >= 16x16, skip ALL
583*fb1b10abSAndroid Build Coastguard Worker   // INTRA for bsize >= 32x32 and vert/horz INTRA for bsize 16x16, 16x32 and
584*fb1b10abSAndroid Build Coastguard Worker   // 32x16.
585*fb1b10abSAndroid Build Coastguard Worker   // 3: Same as (2), but also skip golden zeromv.
586*fb1b10abSAndroid Build Coastguard Worker   int short_circuit_low_temp_var;
587*fb1b10abSAndroid Build Coastguard Worker 
588*fb1b10abSAndroid Build Coastguard Worker   // Limits the rd-threshold update for early exit for the newmv-last mode,
589*fb1b10abSAndroid Build Coastguard Worker   // for non-rd mode.
590*fb1b10abSAndroid Build Coastguard Worker   int limit_newmv_early_exit;
591*fb1b10abSAndroid Build Coastguard Worker 
592*fb1b10abSAndroid Build Coastguard Worker   // Adds a bias against golden reference, for non-rd mode.
593*fb1b10abSAndroid Build Coastguard Worker   int bias_golden;
594*fb1b10abSAndroid Build Coastguard Worker 
595*fb1b10abSAndroid Build Coastguard Worker   // Bias to use base mv and skip 1/4 subpel search when use base mv in
596*fb1b10abSAndroid Build Coastguard Worker   // enhancement layer.
597*fb1b10abSAndroid Build Coastguard Worker   int base_mv_aggressive;
598*fb1b10abSAndroid Build Coastguard Worker 
599*fb1b10abSAndroid Build Coastguard Worker   // Global flag to enable partition copy from the previous frame.
600*fb1b10abSAndroid Build Coastguard Worker   int copy_partition_flag;
601*fb1b10abSAndroid Build Coastguard Worker 
602*fb1b10abSAndroid Build Coastguard Worker   // Compute the source sad for every superblock of the frame,
603*fb1b10abSAndroid Build Coastguard Worker   // prior to encoding the frame, to be used to bypass some encoder decisions.
604*fb1b10abSAndroid Build Coastguard Worker   int use_source_sad;
605*fb1b10abSAndroid Build Coastguard Worker 
606*fb1b10abSAndroid Build Coastguard Worker   int use_simple_block_yrd;
607*fb1b10abSAndroid Build Coastguard Worker 
608*fb1b10abSAndroid Build Coastguard Worker   // If source sad of superblock is high (> adapt_partition_thresh), will switch
609*fb1b10abSAndroid Build Coastguard Worker   // from VARIANCE_PARTITION to REFERENCE_PARTITION (which selects partition
610*fb1b10abSAndroid Build Coastguard Worker   // based on the nonrd-pickmode).
611*fb1b10abSAndroid Build Coastguard Worker   int adapt_partition_source_sad;
612*fb1b10abSAndroid Build Coastguard Worker   int adapt_partition_thresh;
613*fb1b10abSAndroid Build Coastguard Worker 
614*fb1b10abSAndroid Build Coastguard Worker   // Enable use of alt-refs in 1 pass VBR.
615*fb1b10abSAndroid Build Coastguard Worker   int use_altref_onepass;
616*fb1b10abSAndroid Build Coastguard Worker 
617*fb1b10abSAndroid Build Coastguard Worker   // Enable use of compound prediction, for nonrd_pickmode with nonzero lag.
618*fb1b10abSAndroid Build Coastguard Worker   int use_compound_nonrd_pickmode;
619*fb1b10abSAndroid Build Coastguard Worker 
620*fb1b10abSAndroid Build Coastguard Worker   // Always use nonrd_pick_intra for all block sizes on keyframes.
621*fb1b10abSAndroid Build Coastguard Worker   int nonrd_keyframe;
622*fb1b10abSAndroid Build Coastguard Worker 
623*fb1b10abSAndroid Build Coastguard Worker   // For SVC: enables use of partition from lower spatial resolution.
624*fb1b10abSAndroid Build Coastguard Worker   int svc_use_lowres_part;
625*fb1b10abSAndroid Build Coastguard Worker 
626*fb1b10abSAndroid Build Coastguard Worker   // Flag to indicate process for handling overshoot on slide/scene change,
627*fb1b10abSAndroid Build Coastguard Worker   // for real-time CBR mode.
628*fb1b10abSAndroid Build Coastguard Worker   OVERSHOOT_DETECTION_CBR_RT overshoot_detection_cbr_rt;
629*fb1b10abSAndroid Build Coastguard Worker 
630*fb1b10abSAndroid Build Coastguard Worker   // Disable partitioning of 16x16 blocks.
631*fb1b10abSAndroid Build Coastguard Worker   int disable_16x16part_nonkey;
632*fb1b10abSAndroid Build Coastguard Worker 
633*fb1b10abSAndroid Build Coastguard Worker   // Allow for disabling golden reference.
634*fb1b10abSAndroid Build Coastguard Worker   int disable_golden_ref;
635*fb1b10abSAndroid Build Coastguard Worker 
636*fb1b10abSAndroid Build Coastguard Worker   // Allow sub-pixel search to use interpolation filters with different taps in
637*fb1b10abSAndroid Build Coastguard Worker   // order to achieve accurate motion search result.
638*fb1b10abSAndroid Build Coastguard Worker   SUBPEL_SEARCH_TYPE use_accurate_subpel_search;
639*fb1b10abSAndroid Build Coastguard Worker 
640*fb1b10abSAndroid Build Coastguard Worker   // Search method used by temporal filtering in full_pixel_motion_search.
641*fb1b10abSAndroid Build Coastguard Worker   SEARCH_METHODS temporal_filter_search_method;
642*fb1b10abSAndroid Build Coastguard Worker 
643*fb1b10abSAndroid Build Coastguard Worker   // Use machine learning based partition search.
644*fb1b10abSAndroid Build Coastguard Worker   int nonrd_use_ml_partition;
645*fb1b10abSAndroid Build Coastguard Worker 
646*fb1b10abSAndroid Build Coastguard Worker   // Multiplier for base threshold for variance partitioning.
647*fb1b10abSAndroid Build Coastguard Worker   int variance_part_thresh_mult;
648*fb1b10abSAndroid Build Coastguard Worker 
649*fb1b10abSAndroid Build Coastguard Worker   // Force subpel motion filter to always use SMOOTH_FILTER.
650*fb1b10abSAndroid Build Coastguard Worker   int force_smooth_interpol;
651*fb1b10abSAndroid Build Coastguard Worker 
652*fb1b10abSAndroid Build Coastguard Worker   // For real-time mode: force DC only under intra search when content
653*fb1b10abSAndroid Build Coastguard Worker   // does not have high souce SAD.
654*fb1b10abSAndroid Build Coastguard Worker   int rt_intra_dc_only_low_content;
655*fb1b10abSAndroid Build Coastguard Worker 
656*fb1b10abSAndroid Build Coastguard Worker   // The encoder has a feature that skips forward transform and quantization
657*fb1b10abSAndroid Build Coastguard Worker   // based on a model rd estimation to reduce encoding time.
658*fb1b10abSAndroid Build Coastguard Worker   // However, this feature is dangerous since it could lead to bad perceptual
659*fb1b10abSAndroid Build Coastguard Worker   // quality. This flag is added to guard the feature.
660*fb1b10abSAndroid Build Coastguard Worker   int allow_skip_txfm_ac_dc;
661*fb1b10abSAndroid Build Coastguard Worker } SPEED_FEATURES;
662*fb1b10abSAndroid Build Coastguard Worker 
663*fb1b10abSAndroid Build Coastguard Worker struct VP9_COMP;
664*fb1b10abSAndroid Build Coastguard Worker 
665*fb1b10abSAndroid Build Coastguard Worker void vp9_set_speed_features_framesize_independent(struct VP9_COMP *cpi,
666*fb1b10abSAndroid Build Coastguard Worker                                                   int speed);
667*fb1b10abSAndroid Build Coastguard Worker void vp9_set_speed_features_framesize_dependent(struct VP9_COMP *cpi,
668*fb1b10abSAndroid Build Coastguard Worker                                                 int speed);
669*fb1b10abSAndroid Build Coastguard Worker 
670*fb1b10abSAndroid Build Coastguard Worker #ifdef __cplusplus
671*fb1b10abSAndroid Build Coastguard Worker }  // extern "C"
672*fb1b10abSAndroid Build Coastguard Worker #endif
673*fb1b10abSAndroid Build Coastguard Worker 
674*fb1b10abSAndroid Build Coastguard Worker #endif  // VPX_VP9_ENCODER_VP9_SPEED_FEATURES_H_
675