• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #include <limits.h>
12 
13 #include "vp9/encoder/vp9_encoder.h"
14 #include "vp9/encoder/vp9_speed_features.h"
15 
16 // Intra only frames, golden frames (except alt ref overlays) and
17 // alt ref frames tend to be coded at a higher than ambient quality
frame_is_boosted(const VP9_COMP * cpi)18 static int frame_is_boosted(const VP9_COMP *cpi) {
19   return frame_is_intra_only(&cpi->common) ||
20          cpi->refresh_alt_ref_frame ||
21          (cpi->refresh_golden_frame && !cpi->rc.is_src_frame_alt_ref) ||
22          vp9_is_upper_layer_key_frame(cpi);
23 }
24 
25 
set_good_speed_feature(VP9_COMP * cpi,VP9_COMMON * cm,SPEED_FEATURES * sf,int speed)26 static void set_good_speed_feature(VP9_COMP *cpi, VP9_COMMON *cm,
27                                    SPEED_FEATURES *sf, int speed) {
28   const int boosted = frame_is_boosted(cpi);
29 
30   sf->adaptive_rd_thresh = 1;
31   sf->allow_skip_recode = 1;
32 
33   if (speed >= 1) {
34     sf->use_square_partition_only = !frame_is_intra_only(cm);
35     sf->less_rectangular_check  = 1;
36 
37     if (MIN(cm->width, cm->height) >= 720)
38       sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
39                                               : DISABLE_ALL_INTER_SPLIT;
40     else
41       sf->disable_split_mask = DISABLE_COMPOUND_SPLIT;
42     sf->use_rd_breakout = 1;
43     sf->adaptive_motion_search = 1;
44     sf->mv.auto_mv_step_size = 1;
45     sf->adaptive_rd_thresh = 2;
46     sf->mv.subpel_iters_per_step = 1;
47     sf->mode_skip_start = 10;
48     sf->adaptive_pred_interp_filter = 1;
49 
50     sf->recode_loop = ALLOW_RECODE_KFARFGF;
51     sf->intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V;
52     sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC_H_V;
53     sf->intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V;
54     sf->intra_uv_mode_mask[TX_16X16] = INTRA_DC_H_V;
55 
56     sf->tx_size_search_breakout = 1;
57 
58     if (MIN(cm->width, cm->height) >= 720)
59       sf->partition_search_breakout_dist_thr = (1 << 23);
60     else
61       sf->partition_search_breakout_dist_thr = (1 << 21);
62     sf->partition_search_breakout_rate_thr = 500;
63   }
64 
65   if (speed >= 2) {
66     sf->tx_size_search_method = frame_is_boosted(cpi) ? USE_FULL_RD
67                                                       : USE_LARGESTALL;
68 
69     if (MIN(cm->width, cm->height) >= 720) {
70       sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
71                                               : DISABLE_ALL_INTER_SPLIT;
72       sf->adaptive_pred_interp_filter = 0;
73     } else {
74       sf->disable_split_mask = LAST_AND_INTRA_SPLIT_ONLY;
75     }
76 
77     sf->reference_masking = 1;
78     sf->mode_search_skip_flags = FLAG_SKIP_INTRA_DIRMISMATCH |
79                                  FLAG_SKIP_INTRA_BESTINTER |
80                                  FLAG_SKIP_COMP_BESTINTRA |
81                                  FLAG_SKIP_INTRA_LOWVAR;
82     sf->disable_filter_search_var_thresh = 100;
83     sf->comp_inter_joint_search_thresh = BLOCK_SIZES;
84     sf->auto_min_max_partition_size = CONSTRAIN_NEIGHBORING_MIN_MAX;
85 
86     if (MIN(cm->width, cm->height) >= 720)
87       sf->partition_search_breakout_dist_thr = (1 << 24);
88     else
89       sf->partition_search_breakout_dist_thr = (1 << 22);
90     sf->partition_search_breakout_rate_thr = 700;
91   }
92 
93   if (speed >= 3) {
94     sf->tx_size_search_method = frame_is_intra_only(cm) ? USE_FULL_RD
95                                                         : USE_LARGESTALL;
96     if (MIN(cm->width, cm->height) >= 720) {
97       sf->disable_split_mask = DISABLE_ALL_SPLIT;
98       sf->schedule_mode_search = cm->base_qindex < 220 ? 1 : 0;
99     } else {
100       sf->max_intra_bsize = BLOCK_32X32;
101       sf->disable_split_mask = DISABLE_ALL_INTER_SPLIT;
102       sf->schedule_mode_search = cm->base_qindex < 175 ? 1 : 0;
103     }
104     sf->adaptive_pred_interp_filter = 0;
105     sf->adaptive_mode_search = 1;
106     sf->cb_partition_search = !boosted;
107     sf->cb_pred_filter_search = 1;
108     sf->alt_ref_search_fp = 1;
109     sf->motion_field_mode_search = !boosted;
110     sf->recode_loop = ALLOW_RECODE_KFMAXBW;
111     sf->adaptive_rd_thresh = 3;
112     sf->mode_skip_start = 6;
113     sf->intra_y_mode_mask[TX_32X32] = INTRA_DC;
114     sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC;
115     sf->adaptive_interp_filter_search = 1;
116 
117     if (MIN(cm->width, cm->height) >= 720)
118       sf->partition_search_breakout_dist_thr = (1 << 25);
119     else
120       sf->partition_search_breakout_dist_thr = (1 << 23);
121     sf->partition_search_breakout_rate_thr = 1000;
122   }
123 
124   if (speed >= 4) {
125     sf->use_square_partition_only = 1;
126     sf->tx_size_search_method = USE_LARGESTALL;
127     sf->disable_split_mask = DISABLE_ALL_SPLIT;
128     sf->mv.search_method = BIGDIA;
129     sf->mv.subpel_search_method = SUBPEL_TREE_PRUNED;
130     sf->adaptive_rd_thresh = 4;
131     sf->mode_search_skip_flags |= FLAG_EARLY_TERMINATE;
132     sf->disable_filter_search_var_thresh = 200;
133     sf->use_lp32x32fdct = 1;
134     sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
135     sf->use_fast_coef_costing = 1;
136 
137     if (MIN(cm->width, cm->height) >= 720)
138       sf->partition_search_breakout_dist_thr = (1 << 26);
139     else
140       sf->partition_search_breakout_dist_thr = (1 << 24);
141     sf->partition_search_breakout_rate_thr = 1500;
142   }
143 
144   if (speed >= 5) {
145     int i;
146 
147     sf->partition_search_type = FIXED_PARTITION;
148     sf->optimize_coefficients = 0;
149     sf->mv.search_method = HEX;
150     sf->disable_filter_search_var_thresh = 500;
151     for (i = 0; i < TX_SIZES; ++i) {
152       sf->intra_y_mode_mask[i] = INTRA_DC;
153       sf->intra_uv_mode_mask[i] = INTRA_DC;
154     }
155     cpi->allow_encode_breakout = ENCODE_BREAKOUT_ENABLED;
156   }
157   if (speed >= 6) {
158     sf->mv.reduce_first_step_size = 1;
159   }
160 }
161 
set_rt_speed_feature(VP9_COMP * cpi,SPEED_FEATURES * sf,int speed,vp9e_tune_content content)162 static void set_rt_speed_feature(VP9_COMP *cpi, SPEED_FEATURES *sf,
163                                  int speed, vp9e_tune_content content) {
164   VP9_COMMON *const cm = &cpi->common;
165   const int is_keyframe = cm->frame_type == KEY_FRAME;
166   const int frames_since_key = is_keyframe ? 0 : cpi->rc.frames_since_key;
167   sf->static_segmentation = 0;
168   sf->adaptive_rd_thresh = 1;
169   sf->use_fast_coef_costing = 1;
170 
171   if (speed >= 1) {
172     sf->use_square_partition_only = !frame_is_intra_only(cm);
173     sf->less_rectangular_check = 1;
174     sf->tx_size_search_method = frame_is_intra_only(cm) ? USE_FULL_RD
175                                                         : USE_LARGESTALL;
176 
177     if (MIN(cm->width, cm->height) >= 720)
178       sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
179                                               : DISABLE_ALL_INTER_SPLIT;
180     else
181       sf->disable_split_mask = DISABLE_COMPOUND_SPLIT;
182 
183     sf->use_rd_breakout = 1;
184 
185     sf->adaptive_motion_search = 1;
186     sf->adaptive_pred_interp_filter = 1;
187     sf->mv.auto_mv_step_size = 1;
188     sf->adaptive_rd_thresh = 2;
189     sf->intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V;
190     sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC_H_V;
191     sf->intra_uv_mode_mask[TX_16X16] = INTRA_DC_H_V;
192   }
193 
194   if (speed >= 2) {
195     if (MIN(cm->width, cm->height) >= 720)
196       sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
197                                               : DISABLE_ALL_INTER_SPLIT;
198     else
199       sf->disable_split_mask = LAST_AND_INTRA_SPLIT_ONLY;
200 
201     sf->mode_search_skip_flags = FLAG_SKIP_INTRA_DIRMISMATCH |
202                                  FLAG_SKIP_INTRA_BESTINTER |
203                                  FLAG_SKIP_COMP_BESTINTRA |
204                                  FLAG_SKIP_INTRA_LOWVAR;
205     sf->adaptive_pred_interp_filter = 2;
206     sf->reference_masking = 1;
207     sf->disable_filter_search_var_thresh = 50;
208     sf->comp_inter_joint_search_thresh = BLOCK_SIZES;
209     sf->auto_min_max_partition_size = RELAXED_NEIGHBORING_MIN_MAX;
210     sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_LOW_MOTION;
211     sf->lf_motion_threshold = LOW_MOTION_THRESHOLD;
212     sf->adjust_partitioning_from_last_frame = 1;
213     sf->last_partitioning_redo_frequency = 3;
214     sf->use_lp32x32fdct = 1;
215     sf->mode_skip_start = 11;
216     sf->intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V;
217   }
218 
219   if (speed >= 3) {
220     sf->use_square_partition_only = 1;
221     sf->disable_filter_search_var_thresh = 100;
222     sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_ALL;
223     sf->constrain_copy_partition = 1;
224     sf->use_uv_intra_rd_estimate = 1;
225     sf->skip_encode_sb = 1;
226     sf->mv.subpel_iters_per_step = 1;
227     sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
228     sf->adaptive_rd_thresh = 4;
229     sf->mode_skip_start = 6;
230     sf->allow_skip_recode = 0;
231     sf->optimize_coefficients = 0;
232     sf->disable_split_mask = DISABLE_ALL_SPLIT;
233     sf->lpf_pick = LPF_PICK_FROM_Q;
234   }
235 
236   if (speed >= 4) {
237     int i;
238     sf->last_partitioning_redo_frequency = 4;
239     sf->adaptive_rd_thresh = 5;
240     sf->use_fast_coef_costing = 0;
241     sf->auto_min_max_partition_size = STRICT_NEIGHBORING_MIN_MAX;
242     sf->adjust_partitioning_from_last_frame =
243         cm->last_frame_type != cm->frame_type || (0 ==
244         (frames_since_key + 1) % sf->last_partitioning_redo_frequency);
245     sf->mv.subpel_force_stop = 1;
246     for (i = 0; i < TX_SIZES; i++) {
247       sf->intra_y_mode_mask[i] = INTRA_DC_H_V;
248       sf->intra_uv_mode_mask[i] = INTRA_DC;
249     }
250     sf->intra_y_mode_mask[TX_32X32] = INTRA_DC;
251     sf->frame_parameter_update = 0;
252     sf->mv.search_method = FAST_HEX;
253     sf->inter_mode_mask[BLOCK_32X32] = INTER_NEAREST_NEAR_NEW;
254     sf->inter_mode_mask[BLOCK_32X64] = INTER_NEAREST;
255     sf->inter_mode_mask[BLOCK_64X32] = INTER_NEAREST;
256     sf->inter_mode_mask[BLOCK_64X64] = INTER_NEAREST;
257     sf->max_intra_bsize = BLOCK_32X32;
258     sf->allow_skip_recode = 1;
259   }
260 
261   if (speed >= 5) {
262     sf->use_quant_fp = !is_keyframe;
263     sf->auto_min_max_partition_size = is_keyframe ? RELAXED_NEIGHBORING_MIN_MAX
264                                                   : STRICT_NEIGHBORING_MIN_MAX;
265     sf->max_partition_size = BLOCK_32X32;
266     sf->min_partition_size = BLOCK_8X8;
267     sf->partition_check =
268         (frames_since_key % sf->last_partitioning_redo_frequency == 1);
269     sf->force_frame_boost = is_keyframe ||
270         (frames_since_key % (sf->last_partitioning_redo_frequency << 1) == 1);
271     sf->max_delta_qindex = is_keyframe ? 20 : 15;
272     sf->partition_search_type = REFERENCE_PARTITION;
273     sf->use_nonrd_pick_mode = 1;
274     sf->allow_skip_recode = 0;
275   }
276 
277   if (speed >= 6) {
278     if (content == VP9E_CONTENT_SCREEN) {
279       int i;
280       // Allow fancy modes at all sizes since SOURCE_VAR_BASED_PARTITION is used
281       for (i = 0; i < BLOCK_SIZES; ++i)
282         sf->inter_mode_mask[i] = INTER_ALL;
283     }
284 
285     // Adaptively switch between SOURCE_VAR_BASED_PARTITION and FIXED_PARTITION.
286     sf->partition_search_type = SOURCE_VAR_BASED_PARTITION;
287     sf->search_type_check_frequency = 50;
288 
289     sf->tx_size_search_method = is_keyframe ? USE_LARGESTALL : USE_TX_8X8;
290 
291     // This feature is only enabled when partition search is disabled.
292     sf->reuse_inter_pred_sby = 1;
293 
294     // Increase mode checking threshold for NEWMV.
295     sf->elevate_newmv_thresh = 2000;
296 
297     sf->mv.reduce_first_step_size = 1;
298   }
299 
300   if (speed >= 7) {
301     sf->mv.search_method = FAST_DIAMOND;
302     sf->mv.fullpel_search_step_param = 10;
303     sf->lpf_pick = LPF_PICK_MINIMAL_LPF;
304     sf->encode_breakout_thresh = (MIN(cm->width, cm->height) >= 720) ?
305         800 : 300;
306     sf->elevate_newmv_thresh = 2500;
307   }
308 
309   if (speed >= 12) {
310     sf->elevate_newmv_thresh = 4000;
311     sf->mv.subpel_force_stop = 2;
312   }
313 
314   if (speed >= 13) {
315     int i;
316     sf->max_intra_bsize = BLOCK_32X32;
317     for (i = 0; i < BLOCK_SIZES; ++i)
318       sf->inter_mode_mask[i] = INTER_NEAREST;
319   }
320 }
321 
vp9_set_speed_features(VP9_COMP * cpi)322 void vp9_set_speed_features(VP9_COMP *cpi) {
323   SPEED_FEATURES *const sf = &cpi->sf;
324   VP9_COMMON *const cm = &cpi->common;
325   const VP9EncoderConfig *const oxcf = &cpi->oxcf;
326   int i;
327 
328   // best quality defaults
329   sf->frame_parameter_update = 1;
330   sf->mv.search_method = NSTEP;
331   sf->recode_loop = ALLOW_RECODE;
332   sf->mv.subpel_search_method = SUBPEL_TREE;
333   sf->mv.subpel_iters_per_step = 2;
334   sf->mv.subpel_force_stop = 0;
335   sf->optimize_coefficients = !is_lossless_requested(&cpi->oxcf);
336   sf->mv.reduce_first_step_size = 0;
337   sf->mv.auto_mv_step_size = 0;
338   sf->mv.fullpel_search_step_param = 6;
339   sf->comp_inter_joint_search_thresh = BLOCK_4X4;
340   sf->adaptive_rd_thresh = 0;
341   sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_OFF;
342   sf->tx_size_search_method = USE_FULL_RD;
343   sf->use_lp32x32fdct = 0;
344   sf->adaptive_motion_search = 0;
345   sf->adaptive_pred_interp_filter = 0;
346   sf->adaptive_mode_search = 0;
347   sf->cb_pred_filter_search = 0;
348   sf->cb_partition_search = 0;
349   sf->motion_field_mode_search = 0;
350   sf->alt_ref_search_fp = 0;
351   sf->use_quant_fp = 0;
352   sf->reference_masking = 0;
353   sf->partition_search_type = SEARCH_PARTITION;
354   sf->less_rectangular_check = 0;
355   sf->use_square_partition_only = 0;
356   sf->auto_min_max_partition_size = NOT_IN_USE;
357   sf->max_partition_size = BLOCK_64X64;
358   sf->min_partition_size = BLOCK_4X4;
359   sf->adjust_partitioning_from_last_frame = 0;
360   sf->last_partitioning_redo_frequency = 4;
361   sf->constrain_copy_partition = 0;
362   sf->disable_split_mask = 0;
363   sf->mode_search_skip_flags = 0;
364   sf->force_frame_boost = 0;
365   sf->max_delta_qindex = 0;
366   sf->disable_filter_search_var_thresh = 0;
367   sf->adaptive_interp_filter_search = 0;
368 
369   for (i = 0; i < TX_SIZES; i++) {
370     sf->intra_y_mode_mask[i] = INTRA_ALL;
371     sf->intra_uv_mode_mask[i] = INTRA_ALL;
372   }
373   sf->use_rd_breakout = 0;
374   sf->skip_encode_sb = 0;
375   sf->use_uv_intra_rd_estimate = 0;
376   sf->allow_skip_recode = 0;
377   sf->lpf_pick = LPF_PICK_FROM_FULL_IMAGE;
378   sf->use_fast_coef_updates = TWO_LOOP;
379   sf->use_fast_coef_costing = 0;
380   sf->mode_skip_start = MAX_MODES;  // Mode index at which mode skip mask set
381   sf->schedule_mode_search = 0;
382   sf->use_nonrd_pick_mode = 0;
383   for (i = 0; i < BLOCK_SIZES; ++i)
384     sf->inter_mode_mask[i] = INTER_ALL;
385   sf->max_intra_bsize = BLOCK_64X64;
386   sf->reuse_inter_pred_sby = 0;
387   // This setting only takes effect when partition_search_type is set
388   // to FIXED_PARTITION.
389   sf->always_this_block_size = BLOCK_16X16;
390   sf->search_type_check_frequency = 50;
391   sf->encode_breakout_thresh = 0;
392   sf->elevate_newmv_thresh = 0;
393   // Recode loop tolerence %.
394   sf->recode_tolerance = 25;
395   sf->default_interp_filter = SWITCHABLE;
396   sf->tx_size_search_breakout = 0;
397   sf->partition_search_breakout_dist_thr = 0;
398   sf->partition_search_breakout_rate_thr = 0;
399 
400   if (oxcf->mode == REALTIME)
401     set_rt_speed_feature(cpi, sf, oxcf->speed, oxcf->content);
402   else if (oxcf->mode == GOOD)
403     set_good_speed_feature(cpi, cm, sf, oxcf->speed);
404 
405   cpi->full_search_sad = vp9_full_search_sad;
406   cpi->diamond_search_sad = oxcf->mode == BEST ? vp9_full_range_search
407                                                : vp9_diamond_search_sad;
408   cpi->refining_search_sad = vp9_refining_search_sad;
409 
410 
411   // Slow quant, dct and trellis not worthwhile for first pass
412   // so make sure they are always turned off.
413   if (oxcf->pass == 1)
414     sf->optimize_coefficients = 0;
415 
416   // No recode for 1 pass.
417   if (oxcf->pass == 0) {
418     sf->recode_loop = DISALLOW_RECODE;
419     sf->optimize_coefficients = 0;
420   }
421 
422   if (sf->mv.subpel_search_method == SUBPEL_TREE) {
423     cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree;
424   } else if (sf->mv.subpel_search_method == SUBPEL_TREE_PRUNED) {
425     cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree_pruned;
426   }
427 
428   cpi->mb.optimize = sf->optimize_coefficients == 1 && oxcf->pass != 1;
429 
430   if (sf->disable_split_mask == DISABLE_ALL_SPLIT)
431     sf->adaptive_pred_interp_filter = 0;
432 
433   if (!cpi->oxcf.frame_periodic_boost) {
434     sf->max_delta_qindex = 0;
435   }
436 
437   if (cpi->encode_breakout && oxcf->mode == REALTIME &&
438       sf->encode_breakout_thresh > cpi->encode_breakout)
439     cpi->encode_breakout = sf->encode_breakout_thresh;
440 }
441