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

Enable per motion component filter type selection

Browse Source 
Change-Id: I73823fc94f296d225dece7156de71b30bae3fcb7
This commit is contained in:
Jingning Han
2016-05-02 18:09:47 -07:00
parent c4e7fde68a
commit d567e14e81
2 changed files with 98 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
vp10/encoder/encodeframe.c
View File

@@ -4560,13 +4560,6 @@ static void encode_frame_internal(VP10_COMP *cpi) {
#endif #endif
} }
#if !CONFIG_DUAL_FILTER
static INTERP_FILTER get_cm_interp_filter(VP10_COMP *cpi) {
(void)cpi;
return SWITCHABLE;
}
#endif
void vp10_encode_frame(VP10_COMP *cpi) { void vp10_encode_frame(VP10_COMP *cpi) {
VP10_COMMON *const cm = &cpi->common; VP10_COMMON *const cm = &cpi->common;
@@ -4633,10 +4626,8 @@ void vp10_encode_frame(VP10_COMP *cpi) {
else else
cm->reference_mode = REFERENCE_MODE_SELECT; cm->reference_mode = REFERENCE_MODE_SELECT;
#if !CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
if (cm->interp_filter == SWITCHABLE) { cm->interp_filter = SWITCHABLE;
cm->interp_filter = get_cm_interp_filter(cpi);
}
#endif #endif
encode_frame_internal(cpi); encode_frame_internal(cpi);

113
vp10/encoder/rdopt.c
View File

@@ -322,6 +322,18 @@ static const REF_DEFINITION vp10_ref_order[MAX_REFS] = {
{{INTRA_FRAME, NONE}}, {{INTRA_FRAME, NONE}},
}; };
#if CONFIG_DUAL_FILTER
// TODO(jingning): The magic number 9 here really means the combination
// of prediction filter types for vertical and horizontal directions.
// It will be replaced after we integrate the dual filter experiment with
// the ext-interp experiment.
static int filter_sets[9][2] = {
{0, 0}, {0, 1}, {0, 2},
{1, 0}, {1, 1}, {1, 2},
{2, 0}, {2, 1}, {2, 2},
};
#endif
static INLINE int write_uniform_cost(int n, int v) { static INLINE int write_uniform_cost(int n, int v) {
int l = get_unsigned_bits(n), m = (1 << l) - n; int l = get_unsigned_bits(n), m = (1 << l) - n;
if (l == 0) if (l == 0)
@@ -6431,6 +6443,7 @@ static int estimate_wedge_sign(const VP10_COMP *cpi,
} }
#endif // CONFIG_EXT_INTER #endif // CONFIG_EXT_INTER
#if !CONFIG_DUAL_FILTER
static INTERP_FILTER predict_interp_filter(const VP10_COMP *cpi, static INTERP_FILTER predict_interp_filter(const VP10_COMP *cpi,
const MACROBLOCK *x, const MACROBLOCK *x,
const BLOCK_SIZE bsize, const BLOCK_SIZE bsize,
@@ -6550,6 +6563,7 @@ static INTERP_FILTER predict_interp_filter(const VP10_COMP *cpi,
} }
return best_filter; return best_filter;
} }
#endif
static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x, static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x,
BLOCK_SIZE bsize, BLOCK_SIZE bsize,
@@ -6637,7 +6651,16 @@ static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x,
uint8_t *orig_dst[MAX_MB_PLANE]; uint8_t *orig_dst[MAX_MB_PLANE];
int orig_dst_stride[MAX_MB_PLANE]; int orig_dst_stride[MAX_MB_PLANE];
int rs = 0; int rs = 0;
#if CONFIG_DUAL_FILTER
// Index use case:
// {0, 1} -> (vertical, horizontal) filter types for the first ref frame
// {2, 3} -> (vertical, horizontal) filter types for the second ref frame
INTERP_FILTER best_filter[4] = {SWITCHABLE, SWITCHABLE,
SWITCHABLE, SWITCHABLE,
};
#else
INTERP_FILTER best_filter = SWITCHABLE; INTERP_FILTER best_filter = SWITCHABLE;
#endif
uint8_t skip_txfm[MAX_MB_PLANE][MAX_TX_BLOCKS_IN_MAX_SB] = {{0}}; uint8_t skip_txfm[MAX_MB_PLANE][MAX_TX_BLOCKS_IN_MAX_SB] = {{0}};
int64_t bsse[MAX_MB_PLANE][MAX_TX_BLOCKS_IN_MAX_SB] = {{0}}; int64_t bsse[MAX_MB_PLANE][MAX_TX_BLOCKS_IN_MAX_SB] = {{0}};
@@ -6934,22 +6957,31 @@ static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x,
if (is_comp_pred) if (is_comp_pred)
intpel_mv &= !mv_has_subpel(&mbmi->mv[1].as_mv); intpel_mv &= !mv_has_subpel(&mbmi->mv[1].as_mv);
#if !CONFIG_DUAL_FILTER
best_filter = predict_interp_filter(cpi, x, bsize, mi_row, mi_col, best_filter = predict_interp_filter(cpi, x, bsize, mi_row, mi_col,
single_filter); single_filter);
if (cm->interp_filter != BILINEAR && best_filter == SWITCHABLE) { #endif
if (cm->interp_filter != BILINEAR) {
int newbest; int newbest;
int tmp_rate_sum = 0; int tmp_rate_sum = 0;
int64_t tmp_dist_sum = 0; int64_t tmp_dist_sum = 0;
#if CONFIG_DUAL_FILTER
for (i = 0; i < 9; ++i) {
#else
for (i = 0; i < SWITCHABLE_FILTERS; ++i) { for (i = 0; i < SWITCHABLE_FILTERS; ++i) {
#endif
int j; int j;
int64_t rs_rd; int64_t rs_rd;
int tmp_skip_sb = 0; int tmp_skip_sb = 0;
int64_t tmp_skip_sse = INT64_MAX; int64_t tmp_skip_sse = INT64_MAX;
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
for (j = 0; j < 4; ++j) mbmi->interp_filter[0] = filter_sets[i][0];
mbmi->interp_filter[j] = i; mbmi->interp_filter[1] = filter_sets[i][1];
mbmi->interp_filter[2] = filter_sets[i][0];
mbmi->interp_filter[3] = filter_sets[i][1];
#else #else
mbmi->interp_filter = i; mbmi->interp_filter = i;
#endif #endif
@@ -7017,7 +7049,10 @@ static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x,
if (newbest) { if (newbest) {
best_rd = rd; best_rd = rd;
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
best_filter = mbmi->interp_filter[0]; best_filter[0] = mbmi->interp_filter[0];
best_filter[1] = mbmi->interp_filter[1];
best_filter[2] = mbmi->interp_filter[2];
best_filter[3] = mbmi->interp_filter[3];
#else #else
best_filter = mbmi->interp_filter; best_filter = mbmi->interp_filter;
#endif #endif
@@ -7049,11 +7084,15 @@ static int64_t handle_inter_mode(VP10_COMP *cpi, MACROBLOCK *x,
// Set the appropriate filter // Set the appropriate filter
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
for (i = 0; i < 4; ++i) { mbmi->interp_filter[0] = cm->interp_filter != SWITCHABLE ?
const int frame_idx = (i >> 1); cm->interp_filter : best_filter[0];
if (mbmi->ref_frame[frame_idx] > INTRA_FRAME) mbmi->interp_filter[1] = cm->interp_filter != SWITCHABLE ?
mbmi->interp_filter[i] = cm->interp_filter != SWITCHABLE ? cm->interp_filter : best_filter[1];
cm->interp_filter : best_filter; if (mbmi->ref_frame[1] > INTRA_FRAME) {
mbmi->interp_filter[2] = cm->interp_filter != SWITCHABLE ?
cm->interp_filter : best_filter[2];
mbmi->interp_filter[3] = cm->interp_filter != SWITCHABLE ?
cm->interp_filter : best_filter[3];
} }
#else #else
mbmi->interp_filter = cm->interp_filter != SWITCHABLE ? mbmi->interp_filter = cm->interp_filter != SWITCHABLE ?
@@ -9743,7 +9782,11 @@ void vp10_rd_pick_inter_mode_sub8x8(struct VP10_COMP *cpi,
int ref_index, best_ref_index = 0; int ref_index, best_ref_index = 0;
unsigned int ref_costs_single[MAX_REF_FRAMES], ref_costs_comp[MAX_REF_FRAMES]; unsigned int ref_costs_single[MAX_REF_FRAMES], ref_costs_comp[MAX_REF_FRAMES];
vpx_prob comp_mode_p; vpx_prob comp_mode_p;
#if CONFIG_DUAL_FILTER
INTERP_FILTER tmp_best_filter[4] = { 0 };
#else
INTERP_FILTER tmp_best_filter = SWITCHABLE; INTERP_FILTER tmp_best_filter = SWITCHABLE;
#endif
int rate_uv_intra, rate_uv_tokenonly; int rate_uv_intra, rate_uv_tokenonly;
int64_t dist_uv; int64_t dist_uv;
int skip_uv; int skip_uv;
@@ -10037,7 +10080,11 @@ void vp10_rd_pick_inter_mode_sub8x8(struct VP10_COMP *cpi,
&x->mbmi_ext->ref_mvs[second_ref_frame][0] : NULL; &x->mbmi_ext->ref_mvs[second_ref_frame][0] : NULL;
b_mode_info tmp_best_bmodes[16]; // Should this be 4 ? b_mode_info tmp_best_bmodes[16]; // Should this be 4 ?
MB_MODE_INFO tmp_best_mbmode; MB_MODE_INFO tmp_best_mbmode;
#if CONFIG_DUAL_FILTER
BEST_SEG_INFO bsi[9];
#else
BEST_SEG_INFO bsi[SWITCHABLE_FILTERS]; BEST_SEG_INFO bsi[SWITCHABLE_FILTERS];
#endif
int pred_exists = 0; int pred_exists = 0;
int uv_skippable; int uv_skippable;
#if CONFIG_EXT_INTER #if CONFIG_EXT_INTER
@@ -10067,26 +10114,50 @@ void vp10_rd_pick_inter_mode_sub8x8(struct VP10_COMP *cpi,
mbmi->tx_type = DCT_DCT; mbmi->tx_type = DCT_DCT;
if (cm->interp_filter != BILINEAR) { if (cm->interp_filter != BILINEAR) {
#if CONFIG_DUAL_FILTER
tmp_best_filter[0] = EIGHTTAP_REGULAR;
#else
tmp_best_filter = EIGHTTAP_REGULAR; tmp_best_filter = EIGHTTAP_REGULAR;
#endif
if (x->source_variance < sf->disable_filter_search_var_thresh) { if (x->source_variance < sf->disable_filter_search_var_thresh) {
#if CONFIG_DUAL_FILTER
tmp_best_filter[0] = EIGHTTAP_REGULAR;
#else
tmp_best_filter = EIGHTTAP_REGULAR; tmp_best_filter = EIGHTTAP_REGULAR;
#endif
} else if (sf->adaptive_pred_interp_filter == 1 && } else if (sf->adaptive_pred_interp_filter == 1 &&
ctx->pred_interp_filter < SWITCHABLE) { ctx->pred_interp_filter < SWITCHABLE) {
#if CONFIG_DUAL_FILTER
tmp_best_filter[0] = ctx->pred_interp_filter;
#else
tmp_best_filter = ctx->pred_interp_filter; tmp_best_filter = ctx->pred_interp_filter;
#endif
} else if (sf->adaptive_pred_interp_filter == 2) { } else if (sf->adaptive_pred_interp_filter == 2) {
tmp_best_filter = ctx->pred_interp_filter < SWITCHABLE ? #if CONFIG_DUAL_FILTER
tmp_best_filter[0] = ctx->pred_interp_filter < SWITCHABLE ?
ctx->pred_interp_filter : 0; ctx->pred_interp_filter : 0;
#else
tmp_best_filter = ctx->pred_interp_filter < SWITCHABLE ?
ctx->pred_interp_filter : 0;
#endif
} else { } else {
#if CONFIG_DUAL_FILTER
for (switchable_filter_index = 0;
switchable_filter_index < 9;
++switchable_filter_index) {
#else
for (switchable_filter_index = 0; for (switchable_filter_index = 0;
switchable_filter_index < SWITCHABLE_FILTERS; switchable_filter_index < SWITCHABLE_FILTERS;
++switchable_filter_index) { ++switchable_filter_index) {
#endif
int newbest, rs; int newbest, rs;
int64_t rs_rd; int64_t rs_rd;
MB_MODE_INFO_EXT *mbmi_ext = x->mbmi_ext; MB_MODE_INFO_EXT *mbmi_ext = x->mbmi_ext;
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
int dir; mbmi->interp_filter[0] = filter_sets[switchable_filter_index][0];
for (dir = 0; dir < 4; ++dir) mbmi->interp_filter[1] = filter_sets[switchable_filter_index][1];
mbmi->interp_filter[dir] = switchable_filter_index; mbmi->interp_filter[2] = filter_sets[switchable_filter_index][0];
mbmi->interp_filter[3] = filter_sets[switchable_filter_index][1];
#else #else
mbmi->interp_filter = switchable_filter_index; mbmi->interp_filter = switchable_filter_index;
#endif #endif
@@ -10122,7 +10193,10 @@ void vp10_rd_pick_inter_mode_sub8x8(struct VP10_COMP *cpi,
newbest = (tmp_rd < tmp_best_rd); newbest = (tmp_rd < tmp_best_rd);
if (newbest) { if (newbest) {
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
tmp_best_filter = mbmi->interp_filter[0]; tmp_best_filter[0] = mbmi->interp_filter[0];
tmp_best_filter[1] = mbmi->interp_filter[1];
tmp_best_filter[2] = mbmi->interp_filter[2];
tmp_best_filter[3] = mbmi->interp_filter[3];
#else #else
tmp_best_filter = mbmi->interp_filter; tmp_best_filter = mbmi->interp_filter;
#endif #endif
@@ -10158,9 +10232,14 @@ void vp10_rd_pick_inter_mode_sub8x8(struct VP10_COMP *cpi,
continue; continue;
#if CONFIG_DUAL_FILTER #if CONFIG_DUAL_FILTER
for (i = 0; i < 4; ++i) mbmi->interp_filter[0] = (cm->interp_filter == SWITCHABLE ?
mbmi->interp_filter[i] = (cm->interp_filter == SWITCHABLE ? tmp_best_filter[0] : cm->interp_filter);
tmp_best_filter : cm->interp_filter); mbmi->interp_filter[1] = (cm->interp_filter == SWITCHABLE ?
tmp_best_filter[1] : cm->interp_filter);
mbmi->interp_filter[2] = (cm->interp_filter == SWITCHABLE ?
tmp_best_filter[2] : cm->interp_filter);
mbmi->interp_filter[3] = (cm->interp_filter == SWITCHABLE ?
tmp_best_filter[3] : cm->interp_filter);
#else #else
mbmi->interp_filter = (cm->interp_filter == SWITCHABLE ? mbmi->interp_filter = (cm->interp_filter == SWITCHABLE ?
tmp_best_filter : cm->interp_filter); tmp_best_filter : cm->interp_filter);