generic-library/vpx
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge "vp9: Speed feature to adapt partition based on source_sad."

Browse Source
This commit is contained in:
Marco Paniconi 2017-07-29 01:45:19 +00:00 committed by Gerrit Code Review
commit e48dfcead1
3 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vp9/encoder/vp9_encodeframe.c
View File

@ -981,8 +981,8 @@ static void chroma_check(VP9_COMP *cpi, MACROBLOCK *x, int bsize,
} }
} }
static void avg_source_sad(VP9_COMP *cpi, MACROBLOCK *x, int shift, static uint64_t avg_source_sad(VP9_COMP *cpi, MACROBLOCK *x, int shift,
int sb_offset) { int sb_offset) {
unsigned int tmp_sse; unsigned int tmp_sse;
uint64_t tmp_sad; uint64_t tmp_sad;
unsigned int tmp_variance; unsigned int tmp_variance;
@ -994,7 +994,7 @@ static void avg_source_sad(VP9_COMP *cpi, MACROBLOCK *x, int shift,
uint64_t avg_source_sad_threshold = 10000; uint64_t avg_source_sad_threshold = 10000;
uint64_t avg_source_sad_threshold2 = 12000; uint64_t avg_source_sad_threshold2 = 12000;
#if CONFIG_VP9_HIGHBITDEPTH #if CONFIG_VP9_HIGHBITDEPTH
if (cpi->common.use_highbitdepth) return; if (cpi->common.use_highbitdepth) return 0;
#endif #endif
src_y += shift; src_y += shift;
last_src_y += shift; last_src_y += shift;
@ -1025,7 +1025,7 @@ static void avg_source_sad(VP9_COMP *cpi, MACROBLOCK *x, int shift,
cpi->content_state_sb_fd[sb_offset] = 0; cpi->content_state_sb_fd[sb_offset] = 0;
} }
} }
return; return tmp_sad;
} }
// This function chooses partitioning based on the variance between source and // This function chooses partitioning based on the variance between source and
@ -4167,7 +4167,9 @@ static void encode_nonrd_sb_row(VP9_COMP *cpi, ThreadData *td,
if (cpi->compute_source_sad_onepass && cpi->sf.use_source_sad) { if (cpi->compute_source_sad_onepass && cpi->sf.use_source_sad) {
int shift = cpi->Source->y_stride * (mi_row << 3) + (mi_col << 3); int shift = cpi->Source->y_stride * (mi_row << 3) + (mi_col << 3);
int sb_offset2 = ((cm->mi_cols + 7) >> 3) * (mi_row >> 3) + (mi_col >> 3); int sb_offset2 = ((cm->mi_cols + 7) >> 3) * (mi_row >> 3) + (mi_col >> 3);
avg_source_sad(cpi, x, shift, sb_offset2); int64_t source_sad = avg_source_sad(cpi, x, shift, sb_offset2);
if (sf->adapt_partition_source_sad && source_sad > 40000)
partition_search_type = REFERENCE_PARTITION;
} }
// Set the partition type of the 64X64 block // Set the partition type of the 64X64 block

22
vp9/encoder/vp9_speed_features.c
View File

@ -364,6 +364,7 @@ static void set_rt_speed_feature_framesize_independent(
sf->copy_partition_flag = 0; sf->copy_partition_flag = 0;
sf->use_source_sad = 0; sf->use_source_sad = 0;
sf->use_simple_block_yrd = 0; sf->use_simple_block_yrd = 0;
sf->adapt_partition_source_sad = 0;
if (speed >= 1) { if (speed >= 1) {
sf->allow_txfm_domain_distortion = 1; sf->allow_txfm_domain_distortion = 1;
@ -527,6 +528,17 @@ static void set_rt_speed_feature_framesize_independent(
sf->mv.search_method = NSTEP; sf->mv.search_method = NSTEP;
sf->mv.reduce_first_step_size = 1; sf->mv.reduce_first_step_size = 1;
sf->skip_encode_sb = 0; sf->skip_encode_sb = 0;
if (!cpi->external_resize) sf->use_source_sad = 1;
if (sf->use_source_sad) {
if (cm->width * cm->height <= 640 * 360)
sf->adapt_partition_source_sad = 1;
if (cpi->content_state_sb_fd == NULL &&
(!cpi->use_svc ||
cpi->svc.spatial_layer_id == cpi->svc.number_spatial_layers - 1)) {
cpi->content_state_sb_fd = (uint8_t *)vpx_calloc(
(cm->mi_stride >> 3) * ((cm->mi_rows >> 3) + 1), sizeof(uint8_t));
}
}
if (cpi->oxcf.rc_mode == VPX_CBR && content != VP9E_CONTENT_SCREEN) { if (cpi->oxcf.rc_mode == VPX_CBR && content != VP9E_CONTENT_SCREEN) {
// Enable short circuit for low temporal variance. // Enable short circuit for low temporal variance.
sf->short_circuit_low_temp_var = 1; sf->short_circuit_low_temp_var = 1;
@ -539,6 +551,7 @@ static void set_rt_speed_feature_framesize_independent(
} }
if (speed >= 7) { if (speed >= 7) {
sf->adapt_partition_source_sad = 0;
sf->adaptive_rd_thresh = 3; sf->adaptive_rd_thresh = 3;
sf->mv.search_method = FAST_DIAMOND; sf->mv.search_method = FAST_DIAMOND;
sf->mv.fullpel_search_step_param = 10; sf->mv.fullpel_search_step_param = 10;
@ -552,15 +565,6 @@ static void set_rt_speed_feature_framesize_independent(
if (cpi->svc.non_reference_frame) if (cpi->svc.non_reference_frame)
sf->mv.subpel_search_method = SUBPEL_TREE_PRUNED_EVENMORE; sf->mv.subpel_search_method = SUBPEL_TREE_PRUNED_EVENMORE;
} }
if (!cpi->external_resize) sf->use_source_sad = 1;
if (sf->use_source_sad) {
if (cpi->content_state_sb_fd == NULL &&
(!cpi->use_svc ||
cpi->svc.spatial_layer_id == cpi->svc.number_spatial_layers - 1)) {
cpi->content_state_sb_fd = (uint8_t *)vpx_calloc(
(cm->mi_stride >> 3) * ((cm->mi_rows >> 3) + 1), sizeof(uint8_t));
}
}
// Enable partition copy. For SVC only enabled for top spatial resolution // Enable partition copy. For SVC only enabled for top spatial resolution
// layer. // layer.
cpi->max_copied_frame = 0; cpi->max_copied_frame = 0;

4
vp9/encoder/vp9_speed_features.h
View File

@ -490,6 +490,10 @@ typedef struct SPEED_FEATURES {
int use_source_sad; int use_source_sad;
int use_simple_block_yrd; int use_simple_block_yrd;
// If source sad of superblock is high, will switch from VARIANCE_PARTITION to
// REFERENCE_PARTITION (which selects partition based on the nonrd-pickmode).
int adapt_partition_source_sad;
} SPEED_FEATURES; } SPEED_FEATURES;
struct VP9_COMP; struct VP9_COMP;