vp9_speed_features.c 15.9 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include <limits.h>

Dmitry Kovalev's avatar
Dmitry Kovalev committed
13
#include "vp9/encoder/vp9_encoder.h"
14
15
#include "vp9/encoder/vp9_speed_features.h"

16
17
18
19
20
21
22
23
24
25
// 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 VP9_COMP *cpi) {
  return frame_is_intra_only(&cpi->common) ||
         cpi->refresh_alt_ref_frame ||
         (cpi->refresh_golden_frame && !cpi->rc.is_src_frame_alt_ref) ||
         vp9_is_upper_layer_key_frame(cpi);
}


26
27
static void set_good_speed_feature(VP9_COMP *cpi, VP9_COMMON *cm,
                                   SPEED_FEATURES *sf, int speed) {
28
29
  const int boosted = frame_is_boosted(cpi);

30
31
32
33
34
35
36
37
  sf->adaptive_rd_thresh = 1;
  sf->allow_skip_recode = 1;

  if (speed >= 1) {
    sf->use_square_partition_only = !frame_is_intra_only(cm);
    sf->less_rectangular_check  = 1;

    if (MIN(cm->width, cm->height) >= 720)
38
39
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
40
41
42
43
    else
      sf->disable_split_mask = DISABLE_COMPOUND_SPLIT;
    sf->use_rd_breakout = 1;
    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
48
49
50
51
52
53
54
    sf->mode_skip_start = 10;
    sf->adaptive_pred_interp_filter = 1;

    sf->recode_loop = ALLOW_RECODE_KFARFGF;
    sf->intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V;
    sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC_H_V;
    sf->intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V;
    sf->intra_uv_mode_mask[TX_16X16] = INTRA_DC_H_V;
55
56

    sf->tx_size_search_breakout = 1;
57
58
59
60
61

    if (MIN(cm->width, cm->height) >= 720)
      sf->partition_search_breakout_dist_thr = (1 << 23);
    else
      sf->partition_search_breakout_dist_thr = (1 << 21);
62
    sf->partition_search_breakout_rate_thr = 80;
63
  }
64

65
  if (speed >= 2) {
66
67
68
    sf->tx_size_search_method = frame_is_boosted(cpi) ? USE_FULL_RD
                                                      : USE_LARGESTALL;

69
    if (MIN(cm->width, cm->height) >= 720) {
70
71
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
72
      sf->adaptive_pred_interp_filter = 0;
73
74
      sf->partition_search_breakout_dist_thr = (1 << 24);
      sf->partition_search_breakout_rate_thr = 120;
75
    } else {
76
      sf->disable_split_mask = LAST_AND_INTRA_SPLIT_ONLY;
77
78
      sf->partition_search_breakout_dist_thr = (1 << 22);
      sf->partition_search_breakout_rate_thr = 100;
79
    }
Paul Wilkins's avatar
Paul Wilkins committed
80

81
82
83
84
85
86
87
    sf->reference_masking = 1;
    sf->mode_search_skip_flags = FLAG_SKIP_INTRA_DIRMISMATCH |
                                 FLAG_SKIP_INTRA_BESTINTER |
                                 FLAG_SKIP_COMP_BESTINTRA |
                                 FLAG_SKIP_INTRA_LOWVAR;
    sf->disable_filter_search_var_thresh = 100;
    sf->comp_inter_joint_search_thresh = BLOCK_SIZES;
88
    sf->auto_min_max_partition_size = CONSTRAIN_NEIGHBORING_MIN_MAX;
89

90
    sf->allow_partition_search_skip = 1;
91
  }
92

93
  if (speed >= 3) {
94
95
    sf->tx_size_search_method = frame_is_intra_only(cm) ? USE_FULL_RD
                                                        : USE_LARGESTALL;
Yaowu Xu's avatar
Yaowu Xu committed
96
    if (MIN(cm->width, cm->height) >= 720) {
97
      sf->disable_split_mask = DISABLE_ALL_SPLIT;
Jingning Han's avatar
Jingning Han committed
98
      sf->schedule_mode_search = cm->base_qindex < 220 ? 1 : 0;
99
      sf->partition_search_breakout_dist_thr = (1 << 25);
100
      sf->partition_search_breakout_rate_thr = 200;
Yaowu Xu's avatar
Yaowu Xu committed
101
102
    } else {
      sf->max_intra_bsize = BLOCK_32X32;
103
      sf->disable_split_mask = DISABLE_ALL_INTER_SPLIT;
Jingning Han's avatar
Jingning Han committed
104
      sf->schedule_mode_search = cm->base_qindex < 175 ? 1 : 0;
105
      sf->partition_search_breakout_dist_thr = (1 << 23);
106
      sf->partition_search_breakout_rate_thr = 120;
Yaowu Xu's avatar
Yaowu Xu committed
107
    }
108
    sf->mv.subpel_search_method = SUBPEL_TREE_PRUNED;
109
    sf->adaptive_pred_interp_filter = 0;
110
    sf->adaptive_mode_search = 1;
111
    sf->cb_partition_search = !boosted;
112
    sf->cb_pred_filter_search = 1;
113
    sf->alt_ref_search_fp = 1;
114
115
116
    sf->recode_loop = ALLOW_RECODE_KFMAXBW;
    sf->adaptive_rd_thresh = 3;
    sf->mode_skip_start = 6;
117
118
    sf->intra_y_mode_mask[TX_32X32] = INTRA_DC;
    sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC;
119
    sf->adaptive_interp_filter_search = 1;
120
  }
121

122
123
124
125
  if (speed >= 4) {
    sf->use_square_partition_only = 1;
    sf->tx_size_search_method = USE_LARGESTALL;
    sf->disable_split_mask = DISABLE_ALL_SPLIT;
126
    sf->mv.search_method = BIGDIA;
127
    sf->mv.subpel_search_method = SUBPEL_TREE_PRUNED_MORE;
128
    sf->adaptive_rd_thresh = 4;
Jingning Han's avatar
Jingning Han committed
129
    sf->mode_search_skip_flags |= FLAG_EARLY_TERMINATE;
130
131
    sf->disable_filter_search_var_thresh = 200;
    sf->use_lp32x32fdct = 1;
132
133
    sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
    sf->use_fast_coef_costing = 1;
134
    sf->motion_field_mode_search = !boosted;
135
136
137
138
139

    if (MIN(cm->width, cm->height) >= 720)
      sf->partition_search_breakout_dist_thr = (1 << 26);
    else
      sf->partition_search_breakout_dist_thr = (1 << 24);
140
    sf->partition_search_breakout_rate_thr = 300;
141
  }
142

143
  if (speed >= 5) {
144
145
    int i;

146
147
    sf->partition_search_type = FIXED_PARTITION;
    sf->optimize_coefficients = 0;
148
    sf->mv.search_method = HEX;
149
    sf->disable_filter_search_var_thresh = 500;
150
    for (i = 0; i < TX_SIZES; ++i) {
151
152
      sf->intra_y_mode_mask[i] = INTRA_DC;
      sf->intra_uv_mode_mask[i] = INTRA_DC;
153
154
155
    }
    cpi->allow_encode_breakout = ENCODE_BREAKOUT_ENABLED;
  }
156
157
158
  if (speed >= 6) {
    sf->mv.reduce_first_step_size = 1;
  }
159
160
}

161
static void set_rt_speed_feature(VP9_COMP *cpi, SPEED_FEATURES *sf,
162
                                 int speed, vp9e_tune_content content) {
163
  VP9_COMMON *const cm = &cpi->common;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
164
165
  const int is_keyframe = cm->frame_type == KEY_FRAME;
  const int frames_since_key = is_keyframe ? 0 : cpi->rc.frames_since_key;
166
167
168
169
  sf->static_segmentation = 0;
  sf->adaptive_rd_thresh = 1;
  sf->use_fast_coef_costing = 1;

170
  if (speed >= 1) {
171
172
    sf->use_square_partition_only = !frame_is_intra_only(cm);
    sf->less_rectangular_check = 1;
173
174
    sf->tx_size_search_method = frame_is_intra_only(cm) ? USE_FULL_RD
                                                        : USE_LARGESTALL;
175
176

    if (MIN(cm->width, cm->height) >= 720)
177
178
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
179
180
181
182
    else
      sf->disable_split_mask = DISABLE_COMPOUND_SPLIT;

    sf->use_rd_breakout = 1;
Yaowu Xu's avatar
Yaowu Xu committed
183

184
185
    sf->adaptive_motion_search = 1;
    sf->adaptive_pred_interp_filter = 1;
186
    sf->mv.auto_mv_step_size = 1;
187
188
189
190
191
    sf->adaptive_rd_thresh = 2;
    sf->intra_y_mode_mask[TX_32X32] = INTRA_DC_H_V;
    sf->intra_uv_mode_mask[TX_32X32] = INTRA_DC_H_V;
    sf->intra_uv_mode_mask[TX_16X16] = INTRA_DC_H_V;
  }
192

193
194
  if (speed >= 2) {
    if (MIN(cm->width, cm->height) >= 720)
195
196
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
197
198
199
    else
      sf->disable_split_mask = LAST_AND_INTRA_SPLIT_ONLY;

200
201
202
203
    sf->mode_search_skip_flags = FLAG_SKIP_INTRA_DIRMISMATCH |
                                 FLAG_SKIP_INTRA_BESTINTER |
                                 FLAG_SKIP_COMP_BESTINTRA |
                                 FLAG_SKIP_INTRA_LOWVAR;
204
205
206
207
208
209
    sf->adaptive_pred_interp_filter = 2;
    sf->reference_masking = 1;
    sf->disable_filter_search_var_thresh = 50;
    sf->comp_inter_joint_search_thresh = BLOCK_SIZES;
    sf->auto_min_max_partition_size = RELAXED_NEIGHBORING_MIN_MAX;
    sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_LOW_MOTION;
Paul Wilkins's avatar
Paul Wilkins committed
210
    sf->lf_motion_threshold = LOW_MOTION_THRESHOLD;
211
212
213
214
215
216
    sf->adjust_partitioning_from_last_frame = 1;
    sf->last_partitioning_redo_frequency = 3;
    sf->use_lp32x32fdct = 1;
    sf->mode_skip_start = 11;
    sf->intra_y_mode_mask[TX_16X16] = INTRA_DC_H_V;
  }
217

218
219
220
221
  if (speed >= 3) {
    sf->use_square_partition_only = 1;
    sf->disable_filter_search_var_thresh = 100;
    sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_ALL;
222
    sf->constrain_copy_partition = 1;
223
224
    sf->use_uv_intra_rd_estimate = 1;
    sf->skip_encode_sb = 1;
225
    sf->mv.subpel_iters_per_step = 1;
226
    sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
227
228
229
230
231
232
233
    sf->adaptive_rd_thresh = 4;
    sf->mode_skip_start = 6;
    sf->allow_skip_recode = 0;
    sf->optimize_coefficients = 0;
    sf->disable_split_mask = DISABLE_ALL_SPLIT;
    sf->lpf_pick = LPF_PICK_FROM_Q;
  }
234

Yaowu Xu's avatar
Yaowu Xu committed
235
  if (speed >= 4) {
236
237
238
239
240
241
242
    int i;
    sf->last_partitioning_redo_frequency = 4;
    sf->adaptive_rd_thresh = 5;
    sf->use_fast_coef_costing = 0;
    sf->auto_min_max_partition_size = STRICT_NEIGHBORING_MIN_MAX;
    sf->adjust_partitioning_from_last_frame =
        cm->last_frame_type != cm->frame_type || (0 ==
243
        (frames_since_key + 1) % sf->last_partitioning_redo_frequency);
244
    sf->mv.subpel_force_stop = 1;
245
246
    for (i = 0; i < TX_SIZES; i++) {
      sf->intra_y_mode_mask[i] = INTRA_DC_H_V;
247
      sf->intra_uv_mode_mask[i] = INTRA_DC;
248
    }
249
    sf->intra_y_mode_mask[TX_32X32] = INTRA_DC;
250
    sf->frame_parameter_update = 0;
251
    sf->mv.search_method = FAST_HEX;
252
253
254
255
    sf->inter_mode_mask[BLOCK_32X32] = INTER_NEAREST_NEAR_NEW;
    sf->inter_mode_mask[BLOCK_32X64] = INTER_NEAREST;
    sf->inter_mode_mask[BLOCK_64X32] = INTER_NEAREST;
    sf->inter_mode_mask[BLOCK_64X64] = INTER_NEAREST;
256
257
258
    sf->max_intra_bsize = BLOCK_32X32;
    sf->allow_skip_recode = 1;
  }
259

Yaowu Xu's avatar
Yaowu Xu committed
260
  if (speed >= 5) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
261
262
263
    sf->use_quant_fp = !is_keyframe;
    sf->auto_min_max_partition_size = is_keyframe ? RELAXED_NEIGHBORING_MIN_MAX
                                                  : STRICT_NEIGHBORING_MIN_MAX;
264
265
266
    sf->max_partition_size = BLOCK_32X32;
    sf->min_partition_size = BLOCK_8X8;
    sf->partition_check =
267
        (frames_since_key % sf->last_partitioning_redo_frequency == 1);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
268
269
270
    sf->force_frame_boost = is_keyframe ||
        (frames_since_key % (sf->last_partitioning_redo_frequency << 1) == 1);
    sf->max_delta_qindex = is_keyframe ? 20 : 15;
271
272
273
274
    sf->partition_search_type = REFERENCE_PARTITION;
    sf->use_nonrd_pick_mode = 1;
    sf->allow_skip_recode = 0;
  }
275

Yaowu Xu's avatar
Yaowu Xu committed
276
  if (speed >= 6) {
277
278
279
280
281
282
283
    if (content == VP9E_CONTENT_SCREEN) {
      int i;
      // Allow fancy modes at all sizes since SOURCE_VAR_BASED_PARTITION is used
      for (i = 0; i < BLOCK_SIZES; ++i)
        sf->inter_mode_mask[i] = INTER_ALL;
    }

284
285
286
    // Adaptively switch between SOURCE_VAR_BASED_PARTITION and FIXED_PARTITION.
    sf->partition_search_type = SOURCE_VAR_BASED_PARTITION;
    sf->search_type_check_frequency = 50;
287

Dmitry Kovalev's avatar
Dmitry Kovalev committed
288
    sf->tx_size_search_method = is_keyframe ? USE_LARGESTALL : USE_TX_8X8;
289
290
291

    // This feature is only enabled when partition search is disabled.
    sf->reuse_inter_pred_sby = 1;
292
293
294

    // Increase mode checking threshold for NEWMV.
    sf->elevate_newmv_thresh = 2000;
295
296

    sf->mv.reduce_first_step_size = 1;
297
  }
Dmitry Kovalev's avatar
Dmitry Kovalev committed
298

Yaowu Xu's avatar
Yaowu Xu committed
299
  if (speed >= 7) {
300
    sf->mv.search_method = FAST_DIAMOND;
301
    sf->mv.fullpel_search_step_param = 10;
Yaowu Xu's avatar
Yaowu Xu committed
302
    sf->lpf_pick = LPF_PICK_MINIMAL_LPF;
303
304
    sf->encode_breakout_thresh = (MIN(cm->width, cm->height) >= 720) ?
        800 : 300;
305
    sf->elevate_newmv_thresh = 2500;
Yaowu Xu's avatar
Yaowu Xu committed
306
  }
Dmitry Kovalev's avatar
Dmitry Kovalev committed
307

Yaowu Xu's avatar
Yaowu Xu committed
308
309
310
311
  if (speed >= 12) {
    sf->elevate_newmv_thresh = 4000;
    sf->mv.subpel_force_stop = 2;
  }
Dmitry Kovalev's avatar
Dmitry Kovalev committed
312

Yaowu Xu's avatar
Yaowu Xu committed
313
  if (speed >= 13) {
314
    int i;
315
    sf->max_intra_bsize = BLOCK_32X32;
316
    for (i = 0; i < BLOCK_SIZES; ++i)
317
      sf->inter_mode_mask[i] = INTER_NEAREST;
318
319
320
321
322
323
  }
}

void vp9_set_speed_features(VP9_COMP *cpi) {
  SPEED_FEATURES *const sf = &cpi->sf;
  VP9_COMMON *const cm = &cpi->common;
324
  const VP9EncoderConfig *const oxcf = &cpi->oxcf;
325
326
327
328
  int i;

  // best quality defaults
  sf->frame_parameter_update = 1;
329
  sf->mv.search_method = NSTEP;
330
  sf->recode_loop = ALLOW_RECODE;
331
332
333
  sf->mv.subpel_search_method = SUBPEL_TREE;
  sf->mv.subpel_iters_per_step = 2;
  sf->mv.subpel_force_stop = 0;
334
  sf->optimize_coefficients = !is_lossless_requested(&cpi->oxcf);
335
336
  sf->mv.reduce_first_step_size = 0;
  sf->mv.auto_mv_step_size = 0;
337
  sf->mv.fullpel_search_step_param = 6;
338
339
340
341
342
343
344
  sf->comp_inter_joint_search_thresh = BLOCK_4X4;
  sf->adaptive_rd_thresh = 0;
  sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_OFF;
  sf->tx_size_search_method = USE_FULL_RD;
  sf->use_lp32x32fdct = 0;
  sf->adaptive_motion_search = 0;
  sf->adaptive_pred_interp_filter = 0;
345
  sf->adaptive_mode_search = 0;
346
  sf->cb_pred_filter_search = 0;
347
  sf->cb_partition_search = 0;
348
  sf->motion_field_mode_search = 0;
349
  sf->alt_ref_search_fp = 0;
Jingning Han's avatar
Jingning Han committed
350
  sf->use_quant_fp = 0;
351
352
353
354
355
356
357
358
359
  sf->reference_masking = 0;
  sf->partition_search_type = SEARCH_PARTITION;
  sf->less_rectangular_check = 0;
  sf->use_square_partition_only = 0;
  sf->auto_min_max_partition_size = NOT_IN_USE;
  sf->max_partition_size = BLOCK_64X64;
  sf->min_partition_size = BLOCK_4X4;
  sf->adjust_partitioning_from_last_frame = 0;
  sf->last_partitioning_redo_frequency = 4;
360
  sf->constrain_copy_partition = 0;
361
362
  sf->disable_split_mask = 0;
  sf->mode_search_skip_flags = 0;
363
  sf->force_frame_boost = 0;
364
  sf->max_delta_qindex = 0;
365
  sf->disable_filter_search_var_thresh = 0;
366
  sf->adaptive_interp_filter_search = 0;
367
  sf->allow_partition_search_skip = 0;
368

369
  for (i = 0; i < TX_SIZES; i++) {
370
371
    sf->intra_y_mode_mask[i] = INTRA_ALL;
    sf->intra_uv_mode_mask[i] = INTRA_ALL;
372
373
374
375
376
377
  }
  sf->use_rd_breakout = 0;
  sf->skip_encode_sb = 0;
  sf->use_uv_intra_rd_estimate = 0;
  sf->allow_skip_recode = 0;
  sf->lpf_pick = LPF_PICK_FROM_FULL_IMAGE;
378
  sf->use_fast_coef_updates = TWO_LOOP;
379
380
  sf->use_fast_coef_costing = 0;
  sf->mode_skip_start = MAX_MODES;  // Mode index at which mode skip mask set
Jingning Han's avatar
Jingning Han committed
381
  sf->schedule_mode_search = 0;
382
383
  sf->use_nonrd_pick_mode = 0;
  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
388
389
  // This setting only takes effect when partition_search_type is set
  // to FIXED_PARTITION.
  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
394
  // Recode loop tolerence %.
  sf->recode_tolerance = 25;
395
  sf->default_interp_filter = SWITCHABLE;
396
  sf->tx_size_search_breakout = 0;
397
398
  sf->partition_search_breakout_dist_thr = 0;
  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
403
  else if (oxcf->mode == GOOD)
    set_good_speed_feature(cpi, cm, sf, oxcf->speed);
404

405
  cpi->full_search_sad = vp9_full_search_sad;
406
407
  cpi->diamond_search_sad = oxcf->mode == BEST ? vp9_full_range_search
                                               : vp9_diamond_search_sad;
408
409
410
  cpi->refining_search_sad = vp9_refining_search_sad;


411
412
  // Slow quant, dct and trellis not worthwhile for first pass
  // so make sure they are always turned off.
413
  if (oxcf->pass == 1)
414
415
416
    sf->optimize_coefficients = 0;

  // No recode for 1 pass.
417
  if (oxcf->pass == 0) {
418
419
420
421
    sf->recode_loop = DISALLOW_RECODE;
    sf->optimize_coefficients = 0;
  }

422
  if (sf->mv.subpel_search_method == SUBPEL_TREE) {
423
    cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree;
424
425
  } else if (sf->mv.subpel_search_method == SUBPEL_TREE_PRUNED) {
    cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree_pruned;
426
427
  } else if (sf->mv.subpel_search_method == SUBPEL_TREE_PRUNED_MORE) {
    cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree_pruned_more;
428
429
  } else if (sf->mv.subpel_search_method == SUBPEL_TREE_PRUNED_EVENMORE) {
    cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree_pruned_evenmore;
430
431
  }

432
  cpi->mb.optimize = sf->optimize_coefficients == 1 && oxcf->pass != 1;
433
434
435

  if (sf->disable_split_mask == DISABLE_ALL_SPLIT)
    sf->adaptive_pred_interp_filter = 0;
436
437
438
439

  if (!cpi->oxcf.frame_periodic_boost) {
    sf->max_delta_qindex = 0;
  }
440
441
442
443

  if (cpi->encode_breakout && oxcf->mode == REALTIME &&
      sf->encode_breakout_thresh > cpi->encode_breakout)
    cpi->encode_breakout = sf->encode_breakout_thresh;
444
}