vp9_speed_features.c 14.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
enum {
17
  INTRA_ALL       = (1 << DC_PRED) |
18 19 20 21 22
                    (1 << V_PRED) | (1 << H_PRED) |
                    (1 << D45_PRED) | (1 << D135_PRED) |
                    (1 << D117_PRED) | (1 << D153_PRED) |
                    (1 << D207_PRED) | (1 << D63_PRED) |
                    (1 << TM_PRED),
23 24
  INTRA_DC        = (1 << DC_PRED),
  INTRA_DC_TM     = (1 << DC_PRED) | (1 << TM_PRED),
25
  INTRA_DC_H_V    = (1 << DC_PRED) | (1 << V_PRED) | (1 << H_PRED),
26 27
  INTRA_DC_TM_H_V = (1 << DC_PRED) | (1 << TM_PRED) | (1 << V_PRED) |
                    (1 << H_PRED)
28 29
};

30 31 32 33 34 35
enum {
  INTER_ALL = (1 << NEARESTMV) | (1 << NEARMV) | (1 << ZEROMV) | (1 << NEWMV),
  INTER_NEAREST = (1 << NEARESTMV),
  INTER_NEAREST_NEAR_NEW = (1 << NEARESTMV) | (1 << NEARMV) | (1 << NEWMV)
};

36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
enum {
  DISABLE_ALL_INTER_SPLIT   = (1 << THR_COMP_GA) |
                              (1 << THR_COMP_LA) |
                              (1 << THR_ALTR) |
                              (1 << THR_GOLD) |
                              (1 << THR_LAST),

  DISABLE_ALL_SPLIT         = (1 << THR_INTRA) | DISABLE_ALL_INTER_SPLIT,

  DISABLE_COMPOUND_SPLIT    = (1 << THR_COMP_GA) | (1 << THR_COMP_LA),

  LAST_AND_INTRA_SPLIT_ONLY = (1 << THR_COMP_GA) |
                              (1 << THR_COMP_LA) |
                              (1 << THR_ALTR) |
                              (1 << THR_GOLD)
};
52

53 54
static void set_good_speed_feature(VP9_COMP *cpi, VP9_COMMON *cm,
                                   SPEED_FEATURES *sf, int speed) {
55
  sf->adaptive_rd_thresh = 1;
56
  sf->recode_loop = (speed < 1) ? ALLOW_RECODE : ALLOW_RECODE_KFMAXBW;
57 58 59 60 61
  sf->allow_skip_recode = 1;

  if (speed >= 1) {
    sf->use_square_partition_only = !frame_is_intra_only(cm);
    sf->less_rectangular_check  = 1;
62 63
    sf->tx_size_search_method = frame_is_boosted(cpi) ? USE_FULL_RD
                                                      : USE_LARGESTALL;
64 65

    if (MIN(cm->width, cm->height) >= 720)
66 67
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
68 69 70 71
    else
      sf->disable_split_mask = DISABLE_COMPOUND_SPLIT;
    sf->use_rd_breakout = 1;
    sf->adaptive_motion_search = 1;
72
    sf->mv.auto_mv_step_size = 1;
73
    sf->adaptive_rd_thresh = 2;
74
    sf->mv.subpel_iters_per_step = 1;
75 76 77 78 79 80 81 82 83
    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;
  }
84

85
  if (speed >= 2) {
86
    if (MIN(cm->width, cm->height) >= 720) {
Paul Wilkins's avatar
Paul Wilkins committed
87
      sf->lf_motion_threshold = LOW_MOTION_THRESHOLD;
88
      sf->last_partitioning_redo_frequency = 3;
89 90
      sf->disable_split_mask = cm->show_frame ? DISABLE_ALL_SPLIT
                                              : DISABLE_ALL_INTER_SPLIT;
91
      sf->adaptive_pred_interp_filter = 0;
92
    } else {
93
      sf->disable_split_mask = LAST_AND_INTRA_SPLIT_ONLY;
94
      sf->last_partitioning_redo_frequency = 2;
Paul Wilkins's avatar
Paul Wilkins committed
95
      sf->lf_motion_threshold = NO_MOTION_THRESHOLD;
96
    }
Paul Wilkins's avatar
Paul Wilkins committed
97

98 99 100 101 102 103 104 105 106 107 108
    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;
    sf->auto_min_max_partition_size = RELAXED_NEIGHBORING_MIN_MAX;
    sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_LOW_MOTION;
    sf->adjust_partitioning_from_last_frame = 1;
  }
109

110
  if (speed >= 3) {
111 112
    sf->tx_size_search_method = frame_is_intra_only(cm) ? USE_FULL_RD
                                                        : USE_LARGESTALL;
113 114 115 116 117
    if (MIN(cm->width, cm->height) >= 720)
      sf->disable_split_mask = DISABLE_ALL_SPLIT;
    else
      sf->disable_split_mask = DISABLE_ALL_INTER_SPLIT;

118
    sf->adaptive_pred_interp_filter = 0;
119
    sf->cb_pred_filter_search = 1;
120

Paul Wilkins's avatar
Paul Wilkins committed
121
    sf->lf_motion_threshold = LOW_MOTION_THRESHOLD;
122
    sf->last_partitioning_redo_frequency = 3;
123 124 125
    sf->recode_loop = ALLOW_RECODE_KFMAXBW;
    sf->adaptive_rd_thresh = 3;
    sf->mode_skip_start = 6;
126
    sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
127 128
    sf->use_fast_coef_costing = 1;
  }
129

130 131 132 133 134 135 136 137 138 139 140
  if (speed >= 4) {
    sf->use_square_partition_only = 1;
    sf->tx_size_search_method = USE_LARGESTALL;
    sf->disable_split_mask = DISABLE_ALL_SPLIT;
    sf->adaptive_rd_thresh = 4;
    sf->mode_search_skip_flags |= FLAG_SKIP_COMP_REFMISMATCH |
                                  FLAG_EARLY_TERMINATE;
    sf->disable_filter_search_var_thresh = 200;
    sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_ALL;
    sf->use_lp32x32fdct = 1;
  }
141

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

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

160
static void set_rt_speed_feature(VP9_COMP *cpi, SPEED_FEATURES *sf,
161
                                 int speed) {
162 163 164
  VP9_COMMON *const cm = &cpi->common;
  const int frames_since_key =
      cm->frame_type == KEY_FRAME ? 0 : cpi->rc.frames_since_key;
165 166 167 168
  sf->static_segmentation = 0;
  sf->adaptive_rd_thresh = 1;
  sf->use_fast_coef_costing = 1;

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

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

    sf->use_rd_breakout = 1;
    sf->adaptive_motion_search = 1;
    sf->adaptive_pred_interp_filter = 1;
184
    sf->mv.auto_mv_step_size = 1;
185 186 187 188 189
    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;
  }
190

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

198 199 200 201
    sf->mode_search_skip_flags = FLAG_SKIP_INTRA_DIRMISMATCH |
                                 FLAG_SKIP_INTRA_BESTINTER |
                                 FLAG_SKIP_COMP_BESTINTRA |
                                 FLAG_SKIP_INTRA_LOWVAR;
202 203 204 205 206 207
    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
208
    sf->lf_motion_threshold = LOW_MOTION_THRESHOLD;
209 210 211 212 213 214
    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;
  }
215

216 217 218 219
  if (speed >= 3) {
    sf->use_square_partition_only = 1;
    sf->disable_filter_search_var_thresh = 100;
    sf->use_lastframe_partitioning = LAST_FRAME_PARTITION_ALL;
220
    sf->constrain_copy_partition = 1;
221 222
    sf->use_uv_intra_rd_estimate = 1;
    sf->skip_encode_sb = 1;
223
    sf->mv.subpel_iters_per_step = 1;
224
    sf->use_fast_coef_updates = ONE_LOOP_REDUCED;
225 226 227 228 229 230 231
    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;
  }
232

Yaowu Xu's avatar
Yaowu Xu committed
233
  if (speed >= 4) {
234 235 236 237 238 239 240
    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 ==
241
        (frames_since_key + 1) % sf->last_partitioning_redo_frequency);
242
    sf->mv.subpel_force_stop = 1;
243 244
    for (i = 0; i < TX_SIZES; i++) {
      sf->intra_y_mode_mask[i] = INTRA_DC_H_V;
245
      sf->intra_uv_mode_mask[i] = INTRA_DC;
246
    }
247
    sf->intra_y_mode_mask[TX_32X32] = INTRA_DC;
248
    sf->frame_parameter_update = 0;
249
    sf->mv.search_method = FAST_HEX;
250 251 252 253
    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;
254 255 256
    sf->max_intra_bsize = BLOCK_32X32;
    sf->allow_skip_recode = 1;
  }
257

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

Yaowu Xu's avatar
Yaowu Xu committed
275
  if (speed >= 6) {
276 277 278
    // Adaptively switch between SOURCE_VAR_BASED_PARTITION and FIXED_PARTITION.
    sf->partition_search_type = SOURCE_VAR_BASED_PARTITION;
    sf->search_type_check_frequency = 50;
279

280 281
    sf->tx_size_search_method = (cm->frame_type == KEY_FRAME) ?
        USE_LARGESTALL : USE_TX_8X8;
282
    sf->max_intra_bsize = BLOCK_8X8;
283 284 285

    // This feature is only enabled when partition search is disabled.
    sf->reuse_inter_pred_sby = 1;
286 287 288

    // Increase mode checking threshold for NEWMV.
    sf->elevate_newmv_thresh = 2000;
289 290

    sf->mv.reduce_first_step_size = 1;
291
  }
Yaowu Xu's avatar
Yaowu Xu committed
292
  if (speed >= 7) {
293
    sf->mv.search_method = FAST_DIAMOND;
294
    sf->mv.fullpel_search_step_param = 10;
Yaowu Xu's avatar
Yaowu Xu committed
295
    sf->lpf_pick = LPF_PICK_MINIMAL_LPF;
296 297
    sf->encode_breakout_thresh = (MIN(cm->width, cm->height) >= 720) ?
        800 : 300;
298
    sf->elevate_newmv_thresh = 2500;
Yaowu Xu's avatar
Yaowu Xu committed
299
  }
Yaowu Xu's avatar
Yaowu Xu committed
300 301 302 303 304
  if (speed >= 12) {
    sf->elevate_newmv_thresh = 4000;
    sf->mv.subpel_force_stop = 2;
  }
  if (speed >= 13) {
305
    int i;
306
    sf->max_intra_bsize = BLOCK_32X32;
307
    for (i = 0; i < BLOCK_SIZES; ++i)
308
      sf->inter_mode_mask[i] = INTER_NEAREST;
309 310 311 312 313 314
  }
}

void vp9_set_speed_features(VP9_COMP *cpi) {
  SPEED_FEATURES *const sf = &cpi->sf;
  VP9_COMMON *const cm = &cpi->common;
315
  const VP9EncoderConfig *const oxcf = &cpi->oxcf;
316 317 318 319
  int i;

  // best quality defaults
  sf->frame_parameter_update = 1;
320
  sf->mv.search_method = NSTEP;
321
  sf->recode_loop = ALLOW_RECODE;
322 323 324
  sf->mv.subpel_search_method = SUBPEL_TREE;
  sf->mv.subpel_iters_per_step = 2;
  sf->mv.subpel_force_stop = 0;
325
  sf->optimize_coefficients = !is_lossless_requested(&cpi->oxcf);
326 327
  sf->mv.reduce_first_step_size = 0;
  sf->mv.auto_mv_step_size = 0;
328
  sf->mv.fullpel_search_step_param = 6;
329 330 331 332 333 334 335
  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;
336
  sf->cb_pred_filter_search = 0;
Jingning Han's avatar
Jingning Han committed
337
  sf->use_quant_fp = 0;
338 339 340 341 342 343 344 345 346
  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;
347
  sf->constrain_copy_partition = 0;
348 349
  sf->disable_split_mask = 0;
  sf->mode_search_skip_flags = 0;
350
  sf->force_frame_boost = 0;
351
  sf->max_delta_qindex = 0;
352 353 354
  sf->disable_split_var_thresh = 0;
  sf->disable_filter_search_var_thresh = 0;
  for (i = 0; i < TX_SIZES; i++) {
355 356
    sf->intra_y_mode_mask[i] = INTRA_ALL;
    sf->intra_uv_mode_mask[i] = INTRA_ALL;
357 358 359 360 361 362
  }
  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;
363
  sf->use_fast_coef_updates = TWO_LOOP;
364 365 366 367
  sf->use_fast_coef_costing = 0;
  sf->mode_skip_start = MAX_MODES;  // Mode index at which mode skip mask set
  sf->use_nonrd_pick_mode = 0;
  for (i = 0; i < BLOCK_SIZES; ++i)
368
    sf->inter_mode_mask[i] = INTER_ALL;
369
  sf->max_intra_bsize = BLOCK_64X64;
370
  sf->reuse_inter_pred_sby = 0;
371 372 373
  // This setting only takes effect when partition_search_type is set
  // to FIXED_PARTITION.
  sf->always_this_block_size = BLOCK_16X16;
374
  sf->search_type_check_frequency = 50;
375
  sf->encode_breakout_thresh = 0;
376
  sf->elevate_newmv_thresh = 0;
377 378
  // Recode loop tolerence %.
  sf->recode_tolerance = 25;
379
  sf->default_interp_filter = SWITCHABLE;
380

381
  switch (oxcf->mode) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
382 383
    case ONE_PASS_BEST:
    case TWO_PASS_SECOND_BEST:  // This is the best quality mode.
384 385
      cpi->diamond_search_sad = vp9_full_range_search;
      break;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
386 387 388
    case TWO_PASS_FIRST:
    case ONE_PASS_GOOD:
    case TWO_PASS_SECOND_GOOD:
389
      set_good_speed_feature(cpi, cm, sf, oxcf->speed);
390
      break;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
391
    case REALTIME:
392
      set_rt_speed_feature(cpi, sf, oxcf->speed);
393 394 395 396 397
      break;
  }

  // Slow quant, dct and trellis not worthwhile for first pass
  // so make sure they are always turned off.
398
  if (cpi->pass == 1)
399 400 401 402 403 404 405 406
    sf->optimize_coefficients = 0;

  // No recode for 1 pass.
  if (cpi->pass == 0) {
    sf->recode_loop = DISALLOW_RECODE;
    sf->optimize_coefficients = 0;
  }

407
  if (sf->mv.subpel_search_method == SUBPEL_TREE) {
408 409 410
    cpi->find_fractional_mv_step = vp9_find_best_sub_pixel_tree;
  }

411
  cpi->mb.optimize = sf->optimize_coefficients == 1 && cpi->pass != 1;
412 413 414

  if (sf->disable_split_mask == DISABLE_ALL_SPLIT)
    sf->adaptive_pred_interp_filter = 0;
415 416 417 418

  if (!cpi->oxcf.frame_periodic_boost) {
    sf->max_delta_qindex = 0;
  }
419 420 421 422

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