0ce98b423b
this code is similar to other existing uses and/or vp9 Change-Id: I56e646931379759d9f7332ea6d746060007c75ee
1430 lines
49 KiB
C
1430 lines
49 KiB
C
/*
|
|
* 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>
|
|
#include "vpx_config.h"
|
|
#include "./vpx_dsp_rtcd.h"
|
|
#include "onyx_int.h"
|
|
#include "modecosts.h"
|
|
#include "encodeintra.h"
|
|
#include "vp8/common/common.h"
|
|
#include "vp8/common/entropymode.h"
|
|
#include "pickinter.h"
|
|
#include "vp8/common/findnearmv.h"
|
|
#include "encodemb.h"
|
|
#include "vp8/common/reconinter.h"
|
|
#include "vp8/common/reconintra.h"
|
|
#include "vp8/common/reconintra4x4.h"
|
|
#include "vpx_dsp/variance.h"
|
|
#include "mcomp.h"
|
|
#include "rdopt.h"
|
|
#include "vpx_dsp/vpx_dsp_common.h"
|
|
#include "vpx_mem/vpx_mem.h"
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
#include "denoising.h"
|
|
#endif
|
|
|
|
#ifdef SPEEDSTATS
|
|
extern unsigned int cnt_pm;
|
|
#endif
|
|
|
|
#define MODEL_MODE 1
|
|
|
|
extern const int vp8_ref_frame_order[MAX_MODES];
|
|
extern const MB_PREDICTION_MODE vp8_mode_order[MAX_MODES];
|
|
|
|
// Fixed point implementation of a skin color classifier. Skin color
|
|
// is model by a Gaussian distribution in the CbCr color space.
|
|
// See ../../test/skin_color_detector_test.cc where the reference
|
|
// skin color classifier is defined.
|
|
|
|
// Fixed-point skin color model parameters.
|
|
static const int skin_mean[5][2] = { { 7463, 9614 },
|
|
{ 6400, 10240 },
|
|
{ 7040, 10240 },
|
|
{ 8320, 9280 },
|
|
{ 6800, 9614 } };
|
|
static const int skin_inv_cov[4] = { 4107, 1663, 1663, 2157 }; // q16
|
|
static const int skin_threshold[6] = { 1570636, 1400000, 800000,
|
|
800000, 800000, 800000 }; // q18
|
|
|
|
// Evaluates the Mahalanobis distance measure for the input CbCr values.
|
|
static int evaluate_skin_color_difference(int cb, int cr, int idx) {
|
|
const int cb_q6 = cb << 6;
|
|
const int cr_q6 = cr << 6;
|
|
const int cb_diff_q12 =
|
|
(cb_q6 - skin_mean[idx][0]) * (cb_q6 - skin_mean[idx][0]);
|
|
const int cbcr_diff_q12 =
|
|
(cb_q6 - skin_mean[idx][0]) * (cr_q6 - skin_mean[idx][1]);
|
|
const int cr_diff_q12 =
|
|
(cr_q6 - skin_mean[idx][1]) * (cr_q6 - skin_mean[idx][1]);
|
|
const int cb_diff_q2 = (cb_diff_q12 + (1 << 9)) >> 10;
|
|
const int cbcr_diff_q2 = (cbcr_diff_q12 + (1 << 9)) >> 10;
|
|
const int cr_diff_q2 = (cr_diff_q12 + (1 << 9)) >> 10;
|
|
const int skin_diff =
|
|
skin_inv_cov[0] * cb_diff_q2 + skin_inv_cov[1] * cbcr_diff_q2 +
|
|
skin_inv_cov[2] * cbcr_diff_q2 + skin_inv_cov[3] * cr_diff_q2;
|
|
return skin_diff;
|
|
}
|
|
|
|
// Checks if the input yCbCr values corresponds to skin color.
|
|
static int is_skin_color(int y, int cb, int cr, int consec_zeromv) {
|
|
if (y < 40 || y > 220) {
|
|
return 0;
|
|
} else {
|
|
if (MODEL_MODE == 0) {
|
|
return (evaluate_skin_color_difference(cb, cr, 0) < skin_threshold[0]);
|
|
} else {
|
|
int i = 0;
|
|
// No skin if block has been zero motion for long consecutive time.
|
|
if (consec_zeromv > 60) return 0;
|
|
// Exit on grey.
|
|
if (cb == 128 && cr == 128) return 0;
|
|
// Exit on very strong cb.
|
|
if (cb > 150 && cr < 110) return 0;
|
|
for (; i < 5; ++i) {
|
|
int skin_color_diff = evaluate_skin_color_difference(cb, cr, i);
|
|
if (skin_color_diff < skin_threshold[i + 1]) {
|
|
if (y < 60 && skin_color_diff > 3 * (skin_threshold[i + 1] >> 2)) {
|
|
return 0;
|
|
} else if (consec_zeromv > 25 &&
|
|
skin_color_diff > (skin_threshold[i + 1] >> 1)) {
|
|
return 0;
|
|
} else {
|
|
return 1;
|
|
}
|
|
}
|
|
// Exit if difference is much large than the threshold.
|
|
if (skin_color_diff > (skin_threshold[i + 1] << 3)) {
|
|
return 0;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int macroblock_corner_grad(unsigned char *signal, int stride,
|
|
int offsetx, int offsety, int sgnx,
|
|
int sgny) {
|
|
int y1 = signal[offsetx * stride + offsety];
|
|
int y2 = signal[offsetx * stride + offsety + sgny];
|
|
int y3 = signal[(offsetx + sgnx) * stride + offsety];
|
|
int y4 = signal[(offsetx + sgnx) * stride + offsety + sgny];
|
|
return VPXMAX(VPXMAX(abs(y1 - y2), abs(y1 - y3)), abs(y1 - y4));
|
|
}
|
|
|
|
static int check_dot_artifact_candidate(VP8_COMP *cpi, MACROBLOCK *x,
|
|
unsigned char *target_last, int stride,
|
|
unsigned char *last_ref, int mb_row,
|
|
int mb_col, int channel) {
|
|
int threshold1 = 6;
|
|
int threshold2 = 3;
|
|
unsigned int max_num = (cpi->common.MBs) / 10;
|
|
int grad_last = 0;
|
|
int grad_source = 0;
|
|
int index = mb_row * cpi->common.mb_cols + mb_col;
|
|
// Threshold for #consecutive (base layer) frames using zero_last mode.
|
|
int num_frames = 30;
|
|
int shift = 15;
|
|
if (channel > 0) {
|
|
shift = 7;
|
|
}
|
|
if (cpi->oxcf.number_of_layers > 1) {
|
|
num_frames = 20;
|
|
}
|
|
x->zero_last_dot_suppress = 0;
|
|
// Blocks on base layer frames that have been using ZEROMV_LAST repeatedly
|
|
// (i.e, at least |x| consecutive frames are candidates for increasing the
|
|
// rd adjustment for zero_last mode.
|
|
// Only allow this for at most |max_num| blocks per frame.
|
|
// Don't allow this for screen content input.
|
|
if (cpi->current_layer == 0 &&
|
|
cpi->consec_zero_last_mvbias[index] > num_frames &&
|
|
x->mbs_zero_last_dot_suppress < max_num &&
|
|
!cpi->oxcf.screen_content_mode) {
|
|
// If this block is checked here, label it so we don't check it again until
|
|
// ~|x| framaes later.
|
|
x->zero_last_dot_suppress = 1;
|
|
// Dot artifact is noticeable as strong gradient at corners of macroblock,
|
|
// for flat areas. As a simple detector for now, we look for a high
|
|
// corner gradient on last ref, and a smaller gradient on source.
|
|
// Check 4 corners, return if any satisfy condition.
|
|
// Top-left:
|
|
grad_last = macroblock_corner_grad(last_ref, stride, 0, 0, 1, 1);
|
|
grad_source = macroblock_corner_grad(target_last, stride, 0, 0, 1, 1);
|
|
if (grad_last >= threshold1 && grad_source <= threshold2) {
|
|
x->mbs_zero_last_dot_suppress++;
|
|
return 1;
|
|
}
|
|
// Top-right:
|
|
grad_last = macroblock_corner_grad(last_ref, stride, 0, shift, 1, -1);
|
|
grad_source = macroblock_corner_grad(target_last, stride, 0, shift, 1, -1);
|
|
if (grad_last >= threshold1 && grad_source <= threshold2) {
|
|
x->mbs_zero_last_dot_suppress++;
|
|
return 1;
|
|
}
|
|
// Bottom-left:
|
|
grad_last = macroblock_corner_grad(last_ref, stride, shift, 0, -1, 1);
|
|
grad_source = macroblock_corner_grad(target_last, stride, shift, 0, -1, 1);
|
|
if (grad_last >= threshold1 && grad_source <= threshold2) {
|
|
x->mbs_zero_last_dot_suppress++;
|
|
return 1;
|
|
}
|
|
// Bottom-right:
|
|
grad_last = macroblock_corner_grad(last_ref, stride, shift, shift, -1, -1);
|
|
grad_source =
|
|
macroblock_corner_grad(target_last, stride, shift, shift, -1, -1);
|
|
if (grad_last >= threshold1 && grad_source <= threshold2) {
|
|
x->mbs_zero_last_dot_suppress++;
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int vp8_skip_fractional_mv_step(MACROBLOCK *mb, BLOCK *b, BLOCKD *d,
|
|
int_mv *bestmv, int_mv *ref_mv,
|
|
int error_per_bit,
|
|
const vp8_variance_fn_ptr_t *vfp,
|
|
int *mvcost[2], int *distortion,
|
|
unsigned int *sse) {
|
|
(void)b;
|
|
(void)d;
|
|
(void)ref_mv;
|
|
(void)error_per_bit;
|
|
(void)vfp;
|
|
(void)mb;
|
|
(void)mvcost;
|
|
(void)distortion;
|
|
(void)sse;
|
|
bestmv->as_mv.row <<= 3;
|
|
bestmv->as_mv.col <<= 3;
|
|
return 0;
|
|
}
|
|
|
|
int vp8_get_inter_mbpred_error(MACROBLOCK *mb, const vp8_variance_fn_ptr_t *vfp,
|
|
unsigned int *sse, int_mv this_mv) {
|
|
BLOCK *b = &mb->block[0];
|
|
BLOCKD *d = &mb->e_mbd.block[0];
|
|
unsigned char *what = (*(b->base_src) + b->src);
|
|
int what_stride = b->src_stride;
|
|
int pre_stride = mb->e_mbd.pre.y_stride;
|
|
unsigned char *in_what = mb->e_mbd.pre.y_buffer + d->offset;
|
|
int in_what_stride = pre_stride;
|
|
int xoffset = this_mv.as_mv.col & 7;
|
|
int yoffset = this_mv.as_mv.row & 7;
|
|
|
|
in_what += (this_mv.as_mv.row >> 3) * pre_stride + (this_mv.as_mv.col >> 3);
|
|
|
|
if (xoffset | yoffset) {
|
|
return vfp->svf(in_what, in_what_stride, xoffset, yoffset, what,
|
|
what_stride, sse);
|
|
} else {
|
|
return vfp->vf(what, what_stride, in_what, in_what_stride, sse);
|
|
}
|
|
}
|
|
|
|
static int get_prediction_error(BLOCK *be, BLOCKD *b) {
|
|
unsigned char *sptr;
|
|
unsigned char *dptr;
|
|
sptr = (*(be->base_src) + be->src);
|
|
dptr = b->predictor;
|
|
|
|
return vpx_get4x4sse_cs(sptr, be->src_stride, dptr, 16);
|
|
}
|
|
|
|
static int pick_intra4x4block(MACROBLOCK *x, int ib,
|
|
B_PREDICTION_MODE *best_mode,
|
|
const int *mode_costs,
|
|
|
|
int *bestrate, int *bestdistortion) {
|
|
BLOCKD *b = &x->e_mbd.block[ib];
|
|
BLOCK *be = &x->block[ib];
|
|
int dst_stride = x->e_mbd.dst.y_stride;
|
|
unsigned char *dst = x->e_mbd.dst.y_buffer + b->offset;
|
|
B_PREDICTION_MODE mode;
|
|
int best_rd = INT_MAX;
|
|
int rate;
|
|
int distortion;
|
|
|
|
unsigned char *Above = dst - dst_stride;
|
|
unsigned char *yleft = dst - 1;
|
|
unsigned char top_left = Above[-1];
|
|
|
|
for (mode = B_DC_PRED; mode <= B_HE_PRED; ++mode) {
|
|
int this_rd;
|
|
|
|
rate = mode_costs[mode];
|
|
|
|
vp8_intra4x4_predict(Above, yleft, dst_stride, mode, b->predictor, 16,
|
|
top_left);
|
|
distortion = get_prediction_error(be, b);
|
|
this_rd = RDCOST(x->rdmult, x->rddiv, rate, distortion);
|
|
|
|
if (this_rd < best_rd) {
|
|
*bestrate = rate;
|
|
*bestdistortion = distortion;
|
|
best_rd = this_rd;
|
|
*best_mode = mode;
|
|
}
|
|
}
|
|
|
|
b->bmi.as_mode = *best_mode;
|
|
vp8_encode_intra4x4block(x, ib);
|
|
return best_rd;
|
|
}
|
|
|
|
static int pick_intra4x4mby_modes(MACROBLOCK *mb, int *Rate, int *best_dist) {
|
|
MACROBLOCKD *const xd = &mb->e_mbd;
|
|
int i;
|
|
int cost = mb->mbmode_cost[xd->frame_type][B_PRED];
|
|
int error;
|
|
int distortion = 0;
|
|
const int *bmode_costs;
|
|
|
|
intra_prediction_down_copy(xd, xd->dst.y_buffer - xd->dst.y_stride + 16);
|
|
|
|
bmode_costs = mb->inter_bmode_costs;
|
|
|
|
for (i = 0; i < 16; ++i) {
|
|
MODE_INFO *const mic = xd->mode_info_context;
|
|
const int mis = xd->mode_info_stride;
|
|
|
|
B_PREDICTION_MODE UNINITIALIZED_IS_SAFE(best_mode);
|
|
int UNINITIALIZED_IS_SAFE(r), UNINITIALIZED_IS_SAFE(d);
|
|
|
|
if (mb->e_mbd.frame_type == KEY_FRAME) {
|
|
const B_PREDICTION_MODE A = above_block_mode(mic, i, mis);
|
|
const B_PREDICTION_MODE L = left_block_mode(mic, i);
|
|
|
|
bmode_costs = mb->bmode_costs[A][L];
|
|
}
|
|
|
|
pick_intra4x4block(mb, i, &best_mode, bmode_costs, &r, &d);
|
|
|
|
cost += r;
|
|
distortion += d;
|
|
mic->bmi[i].as_mode = best_mode;
|
|
|
|
/* Break out case where we have already exceeded best so far value
|
|
* that was passed in
|
|
*/
|
|
if (distortion > *best_dist) break;
|
|
}
|
|
|
|
*Rate = cost;
|
|
|
|
if (i == 16) {
|
|
*best_dist = distortion;
|
|
error = RDCOST(mb->rdmult, mb->rddiv, cost, distortion);
|
|
} else {
|
|
*best_dist = INT_MAX;
|
|
error = INT_MAX;
|
|
}
|
|
|
|
return error;
|
|
}
|
|
|
|
static void pick_intra_mbuv_mode(MACROBLOCK *mb) {
|
|
MACROBLOCKD *x = &mb->e_mbd;
|
|
unsigned char *uabove_row = x->dst.u_buffer - x->dst.uv_stride;
|
|
unsigned char *vabove_row = x->dst.v_buffer - x->dst.uv_stride;
|
|
unsigned char *usrc_ptr = (mb->block[16].src + *mb->block[16].base_src);
|
|
unsigned char *vsrc_ptr = (mb->block[20].src + *mb->block[20].base_src);
|
|
int uvsrc_stride = mb->block[16].src_stride;
|
|
unsigned char uleft_col[8];
|
|
unsigned char vleft_col[8];
|
|
unsigned char utop_left = uabove_row[-1];
|
|
unsigned char vtop_left = vabove_row[-1];
|
|
int i, j;
|
|
int expected_udc;
|
|
int expected_vdc;
|
|
int shift;
|
|
int Uaverage = 0;
|
|
int Vaverage = 0;
|
|
int diff;
|
|
int pred_error[4] = { 0, 0, 0, 0 }, best_error = INT_MAX;
|
|
MB_PREDICTION_MODE UNINITIALIZED_IS_SAFE(best_mode);
|
|
|
|
for (i = 0; i < 8; ++i) {
|
|
uleft_col[i] = x->dst.u_buffer[i * x->dst.uv_stride - 1];
|
|
vleft_col[i] = x->dst.v_buffer[i * x->dst.uv_stride - 1];
|
|
}
|
|
|
|
if (!x->up_available && !x->left_available) {
|
|
expected_udc = 128;
|
|
expected_vdc = 128;
|
|
} else {
|
|
shift = 2;
|
|
|
|
if (x->up_available) {
|
|
for (i = 0; i < 8; ++i) {
|
|
Uaverage += uabove_row[i];
|
|
Vaverage += vabove_row[i];
|
|
}
|
|
|
|
shift++;
|
|
}
|
|
|
|
if (x->left_available) {
|
|
for (i = 0; i < 8; ++i) {
|
|
Uaverage += uleft_col[i];
|
|
Vaverage += vleft_col[i];
|
|
}
|
|
|
|
shift++;
|
|
}
|
|
|
|
expected_udc = (Uaverage + (1 << (shift - 1))) >> shift;
|
|
expected_vdc = (Vaverage + (1 << (shift - 1))) >> shift;
|
|
}
|
|
|
|
for (i = 0; i < 8; ++i) {
|
|
for (j = 0; j < 8; ++j) {
|
|
int predu = uleft_col[i] + uabove_row[j] - utop_left;
|
|
int predv = vleft_col[i] + vabove_row[j] - vtop_left;
|
|
int u_p, v_p;
|
|
|
|
u_p = usrc_ptr[j];
|
|
v_p = vsrc_ptr[j];
|
|
|
|
if (predu < 0) predu = 0;
|
|
|
|
if (predu > 255) predu = 255;
|
|
|
|
if (predv < 0) predv = 0;
|
|
|
|
if (predv > 255) predv = 255;
|
|
|
|
diff = u_p - expected_udc;
|
|
pred_error[DC_PRED] += diff * diff;
|
|
diff = v_p - expected_vdc;
|
|
pred_error[DC_PRED] += diff * diff;
|
|
|
|
diff = u_p - uabove_row[j];
|
|
pred_error[V_PRED] += diff * diff;
|
|
diff = v_p - vabove_row[j];
|
|
pred_error[V_PRED] += diff * diff;
|
|
|
|
diff = u_p - uleft_col[i];
|
|
pred_error[H_PRED] += diff * diff;
|
|
diff = v_p - vleft_col[i];
|
|
pred_error[H_PRED] += diff * diff;
|
|
|
|
diff = u_p - predu;
|
|
pred_error[TM_PRED] += diff * diff;
|
|
diff = v_p - predv;
|
|
pred_error[TM_PRED] += diff * diff;
|
|
}
|
|
|
|
usrc_ptr += uvsrc_stride;
|
|
vsrc_ptr += uvsrc_stride;
|
|
|
|
if (i == 3) {
|
|
usrc_ptr = (mb->block[18].src + *mb->block[18].base_src);
|
|
vsrc_ptr = (mb->block[22].src + *mb->block[22].base_src);
|
|
}
|
|
}
|
|
|
|
for (i = DC_PRED; i <= TM_PRED; ++i) {
|
|
if (best_error > pred_error[i]) {
|
|
best_error = pred_error[i];
|
|
best_mode = (MB_PREDICTION_MODE)i;
|
|
}
|
|
}
|
|
|
|
mb->e_mbd.mode_info_context->mbmi.uv_mode = best_mode;
|
|
}
|
|
|
|
static void update_mvcount(MACROBLOCK *x, int_mv *best_ref_mv) {
|
|
MACROBLOCKD *xd = &x->e_mbd;
|
|
/* Split MV modes currently not supported when RD is nopt enabled,
|
|
* therefore, only need to modify MVcount in NEWMV mode. */
|
|
if (xd->mode_info_context->mbmi.mode == NEWMV) {
|
|
x->MVcount[0][mv_max + ((xd->mode_info_context->mbmi.mv.as_mv.row -
|
|
best_ref_mv->as_mv.row) >>
|
|
1)]++;
|
|
x->MVcount[1][mv_max + ((xd->mode_info_context->mbmi.mv.as_mv.col -
|
|
best_ref_mv->as_mv.col) >>
|
|
1)]++;
|
|
}
|
|
}
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
static void get_lower_res_motion_info(VP8_COMP *cpi, MACROBLOCKD *xd,
|
|
int *dissim, int *parent_ref_frame,
|
|
MB_PREDICTION_MODE *parent_mode,
|
|
int_mv *parent_ref_mv, int mb_row,
|
|
int mb_col) {
|
|
LOWER_RES_MB_INFO *store_mode_info =
|
|
((LOWER_RES_FRAME_INFO *)cpi->oxcf.mr_low_res_mode_info)->mb_info;
|
|
unsigned int parent_mb_index;
|
|
|
|
/* Consider different down_sampling_factor. */
|
|
{
|
|
/* TODO: Removed the loop that supports special down_sampling_factor
|
|
* such as 2, 4, 8. Will revisit it if needed.
|
|
* Should also try using a look-up table to see if it helps
|
|
* performance. */
|
|
int parent_mb_row, parent_mb_col;
|
|
|
|
parent_mb_row = mb_row * cpi->oxcf.mr_down_sampling_factor.den /
|
|
cpi->oxcf.mr_down_sampling_factor.num;
|
|
parent_mb_col = mb_col * cpi->oxcf.mr_down_sampling_factor.den /
|
|
cpi->oxcf.mr_down_sampling_factor.num;
|
|
parent_mb_index = parent_mb_row * cpi->mr_low_res_mb_cols + parent_mb_col;
|
|
}
|
|
|
|
/* Read lower-resolution mode & motion result from memory.*/
|
|
*parent_ref_frame = store_mode_info[parent_mb_index].ref_frame;
|
|
*parent_mode = store_mode_info[parent_mb_index].mode;
|
|
*dissim = store_mode_info[parent_mb_index].dissim;
|
|
|
|
/* For highest-resolution encoder, adjust dissim value. Lower its quality
|
|
* for good performance. */
|
|
if (cpi->oxcf.mr_encoder_id == (cpi->oxcf.mr_total_resolutions - 1))
|
|
*dissim >>= 1;
|
|
|
|
if (*parent_ref_frame != INTRA_FRAME) {
|
|
/* Consider different down_sampling_factor.
|
|
* The result can be rounded to be more precise, but it takes more time.
|
|
*/
|
|
(*parent_ref_mv).as_mv.row = store_mode_info[parent_mb_index].mv.as_mv.row *
|
|
cpi->oxcf.mr_down_sampling_factor.num /
|
|
cpi->oxcf.mr_down_sampling_factor.den;
|
|
(*parent_ref_mv).as_mv.col = store_mode_info[parent_mb_index].mv.as_mv.col *
|
|
cpi->oxcf.mr_down_sampling_factor.num /
|
|
cpi->oxcf.mr_down_sampling_factor.den;
|
|
|
|
vp8_clamp_mv2(parent_ref_mv, xd);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static void check_for_encode_breakout(unsigned int sse, MACROBLOCK *x) {
|
|
MACROBLOCKD *xd = &x->e_mbd;
|
|
|
|
unsigned int threshold =
|
|
(xd->block[0].dequant[1] * xd->block[0].dequant[1] >> 4);
|
|
|
|
if (threshold < x->encode_breakout) threshold = x->encode_breakout;
|
|
|
|
if (sse < threshold) {
|
|
/* Check u and v to make sure skip is ok */
|
|
unsigned int sse2 = 0;
|
|
|
|
sse2 = VP8_UVSSE(x);
|
|
|
|
if (sse2 * 2 < x->encode_breakout) {
|
|
x->skip = 1;
|
|
} else {
|
|
x->skip = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int evaluate_inter_mode(unsigned int *sse, int rate2, int *distortion2,
|
|
VP8_COMP *cpi, MACROBLOCK *x, int rd_adj) {
|
|
MB_PREDICTION_MODE this_mode = x->e_mbd.mode_info_context->mbmi.mode;
|
|
int_mv mv = x->e_mbd.mode_info_context->mbmi.mv;
|
|
int this_rd;
|
|
int denoise_aggressive = 0;
|
|
/* Exit early and don't compute the distortion if this macroblock
|
|
* is marked inactive. */
|
|
if (cpi->active_map_enabled && x->active_ptr[0] == 0) {
|
|
*sse = 0;
|
|
*distortion2 = 0;
|
|
x->skip = 1;
|
|
return INT_MAX;
|
|
}
|
|
|
|
if ((this_mode != NEWMV) || !(cpi->sf.half_pixel_search) ||
|
|
cpi->common.full_pixel == 1) {
|
|
*distortion2 =
|
|
vp8_get_inter_mbpred_error(x, &cpi->fn_ptr[BLOCK_16X16], sse, mv);
|
|
}
|
|
|
|
this_rd = RDCOST(x->rdmult, x->rddiv, rate2, *distortion2);
|
|
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity > 0) {
|
|
denoise_aggressive =
|
|
(cpi->denoiser.denoiser_mode == kDenoiserOnYUVAggressive) ? 1 : 0;
|
|
}
|
|
#endif
|
|
|
|
// Adjust rd for ZEROMV and LAST, if LAST is the closest reference frame.
|
|
// TODO: We should also add condition on distance of closest to current.
|
|
if (!cpi->oxcf.screen_content_mode && this_mode == ZEROMV &&
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame == LAST_FRAME &&
|
|
(denoise_aggressive || (cpi->closest_reference_frame == LAST_FRAME))) {
|
|
// No adjustment if block is considered to be skin area.
|
|
if (x->is_skin) rd_adj = 100;
|
|
|
|
this_rd = (int)(((int64_t)this_rd) * rd_adj / 100);
|
|
}
|
|
|
|
check_for_encode_breakout(*sse, x);
|
|
return this_rd;
|
|
}
|
|
|
|
static void calculate_zeromv_rd_adjustment(VP8_COMP *cpi, MACROBLOCK *x,
|
|
int *rd_adjustment) {
|
|
MODE_INFO *mic = x->e_mbd.mode_info_context;
|
|
int_mv mv_l, mv_a, mv_al;
|
|
int local_motion_check = 0;
|
|
|
|
if (cpi->lf_zeromv_pct > 40) {
|
|
/* left mb */
|
|
mic -= 1;
|
|
mv_l = mic->mbmi.mv;
|
|
|
|
if (mic->mbmi.ref_frame != INTRA_FRAME) {
|
|
if (abs(mv_l.as_mv.row) < 8 && abs(mv_l.as_mv.col) < 8) {
|
|
local_motion_check++;
|
|
}
|
|
}
|
|
|
|
/* above-left mb */
|
|
mic -= x->e_mbd.mode_info_stride;
|
|
mv_al = mic->mbmi.mv;
|
|
|
|
if (mic->mbmi.ref_frame != INTRA_FRAME) {
|
|
if (abs(mv_al.as_mv.row) < 8 && abs(mv_al.as_mv.col) < 8) {
|
|
local_motion_check++;
|
|
}
|
|
}
|
|
|
|
/* above mb */
|
|
mic += 1;
|
|
mv_a = mic->mbmi.mv;
|
|
|
|
if (mic->mbmi.ref_frame != INTRA_FRAME) {
|
|
if (abs(mv_a.as_mv.row) < 8 && abs(mv_a.as_mv.col) < 8) {
|
|
local_motion_check++;
|
|
}
|
|
}
|
|
|
|
if (((!x->e_mbd.mb_to_top_edge || !x->e_mbd.mb_to_left_edge) &&
|
|
local_motion_check > 0) ||
|
|
local_motion_check > 2) {
|
|
*rd_adjustment = 80;
|
|
} else if (local_motion_check > 0) {
|
|
*rd_adjustment = 90;
|
|
}
|
|
}
|
|
}
|
|
|
|
void vp8_pick_inter_mode(VP8_COMP *cpi, MACROBLOCK *x, int recon_yoffset,
|
|
int recon_uvoffset, int *returnrate,
|
|
int *returndistortion, int *returnintra, int mb_row,
|
|
int mb_col) {
|
|
BLOCK *b = &x->block[0];
|
|
BLOCKD *d = &x->e_mbd.block[0];
|
|
MACROBLOCKD *xd = &x->e_mbd;
|
|
MB_MODE_INFO best_mbmode;
|
|
|
|
int_mv best_ref_mv_sb[2];
|
|
int_mv mode_mv_sb[2][MB_MODE_COUNT];
|
|
int_mv best_ref_mv;
|
|
int_mv *mode_mv;
|
|
MB_PREDICTION_MODE this_mode;
|
|
int num00;
|
|
int mdcounts[4];
|
|
int best_rd = INT_MAX;
|
|
int rd_adjustment = 100;
|
|
int best_intra_rd = INT_MAX;
|
|
int mode_index;
|
|
int rate;
|
|
int rate2;
|
|
int distortion2;
|
|
int bestsme = INT_MAX;
|
|
int best_mode_index = 0;
|
|
unsigned int sse = UINT_MAX, best_rd_sse = UINT_MAX;
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
unsigned int zero_mv_sse = UINT_MAX, best_sse = UINT_MAX;
|
|
#endif
|
|
|
|
int sf_improved_mv_pred = cpi->sf.improved_mv_pred;
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
int dissim = INT_MAX;
|
|
int parent_ref_frame = 0;
|
|
int_mv parent_ref_mv;
|
|
MB_PREDICTION_MODE parent_mode = 0;
|
|
int parent_ref_valid = 0;
|
|
#endif
|
|
|
|
int_mv mvp;
|
|
|
|
int near_sadidx[8] = { 0, 1, 2, 3, 4, 5, 6, 7 };
|
|
int saddone = 0;
|
|
/* search range got from mv_pred(). It uses step_param levels. (0-7) */
|
|
int sr = 0;
|
|
|
|
unsigned char *plane[4][3];
|
|
int ref_frame_map[4];
|
|
int sign_bias = 0;
|
|
int dot_artifact_candidate = 0;
|
|
get_predictor_pointers(cpi, plane, recon_yoffset, recon_uvoffset);
|
|
|
|
// If the current frame is using LAST as a reference, check for
|
|
// biasing the mode selection for dot artifacts.
|
|
if (cpi->ref_frame_flags & VP8_LAST_FRAME) {
|
|
unsigned char *target_y = x->src.y_buffer;
|
|
unsigned char *target_u = x->block[16].src + *x->block[16].base_src;
|
|
unsigned char *target_v = x->block[20].src + *x->block[20].base_src;
|
|
int stride = x->src.y_stride;
|
|
int stride_uv = x->block[16].src_stride;
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity) {
|
|
const int uv_denoise = (cpi->oxcf.noise_sensitivity >= 2) ? 1 : 0;
|
|
target_y =
|
|
cpi->denoiser.yv12_running_avg[LAST_FRAME].y_buffer + recon_yoffset;
|
|
stride = cpi->denoiser.yv12_running_avg[LAST_FRAME].y_stride;
|
|
if (uv_denoise) {
|
|
target_u = cpi->denoiser.yv12_running_avg[LAST_FRAME].u_buffer +
|
|
recon_uvoffset;
|
|
target_v = cpi->denoiser.yv12_running_avg[LAST_FRAME].v_buffer +
|
|
recon_uvoffset;
|
|
stride_uv = cpi->denoiser.yv12_running_avg[LAST_FRAME].uv_stride;
|
|
}
|
|
}
|
|
#endif
|
|
dot_artifact_candidate = check_dot_artifact_candidate(
|
|
cpi, x, target_y, stride, plane[LAST_FRAME][0], mb_row, mb_col, 0);
|
|
// If not found in Y channel, check UV channel.
|
|
if (!dot_artifact_candidate) {
|
|
dot_artifact_candidate = check_dot_artifact_candidate(
|
|
cpi, x, target_u, stride_uv, plane[LAST_FRAME][1], mb_row, mb_col, 1);
|
|
if (!dot_artifact_candidate) {
|
|
dot_artifact_candidate = check_dot_artifact_candidate(
|
|
cpi, x, target_v, stride_uv, plane[LAST_FRAME][2], mb_row, mb_col,
|
|
2);
|
|
}
|
|
}
|
|
}
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
// |parent_ref_valid| will be set here if potentially we can do mv resue for
|
|
// this higher resol (|cpi->oxcf.mr_encoder_id| > 0) frame.
|
|
// |parent_ref_valid| may be reset depending on |parent_ref_frame| for
|
|
// the current macroblock below.
|
|
parent_ref_valid = cpi->oxcf.mr_encoder_id && cpi->mr_low_res_mv_avail;
|
|
if (parent_ref_valid) {
|
|
int parent_ref_flag;
|
|
|
|
get_lower_res_motion_info(cpi, xd, &dissim, &parent_ref_frame, &parent_mode,
|
|
&parent_ref_mv, mb_row, mb_col);
|
|
|
|
/* TODO(jkoleszar): The references available (ref_frame_flags) to the
|
|
* lower res encoder should match those available to this encoder, but
|
|
* there seems to be a situation where this mismatch can happen in the
|
|
* case of frame dropping and temporal layers. For example,
|
|
* GOLD being disallowed in ref_frame_flags, but being returned as
|
|
* parent_ref_frame.
|
|
*
|
|
* In this event, take the conservative approach of disabling the
|
|
* lower res info for this MB.
|
|
*/
|
|
|
|
parent_ref_flag = 0;
|
|
// Note availability for mv reuse is only based on last and golden.
|
|
if (parent_ref_frame == LAST_FRAME)
|
|
parent_ref_flag = (cpi->ref_frame_flags & VP8_LAST_FRAME);
|
|
else if (parent_ref_frame == GOLDEN_FRAME)
|
|
parent_ref_flag = (cpi->ref_frame_flags & VP8_GOLD_FRAME);
|
|
|
|
// assert(!parent_ref_frame || parent_ref_flag);
|
|
|
|
// If |parent_ref_frame| did not match either last or golden then
|
|
// shut off mv reuse.
|
|
if (parent_ref_frame && !parent_ref_flag) parent_ref_valid = 0;
|
|
|
|
// Don't do mv reuse since we want to allow for another mode besides
|
|
// ZEROMV_LAST to remove dot artifact.
|
|
if (dot_artifact_candidate) parent_ref_valid = 0;
|
|
}
|
|
#endif
|
|
|
|
// Check if current macroblock is in skin area.
|
|
{
|
|
const int y = (x->src.y_buffer[7 * x->src.y_stride + 7] +
|
|
x->src.y_buffer[7 * x->src.y_stride + 8] +
|
|
x->src.y_buffer[8 * x->src.y_stride + 7] +
|
|
x->src.y_buffer[8 * x->src.y_stride + 8]) >>
|
|
2;
|
|
const int cb = (x->src.u_buffer[3 * x->src.uv_stride + 3] +
|
|
x->src.u_buffer[3 * x->src.uv_stride + 4] +
|
|
x->src.u_buffer[4 * x->src.uv_stride + 3] +
|
|
x->src.u_buffer[4 * x->src.uv_stride + 4]) >>
|
|
2;
|
|
const int cr = (x->src.v_buffer[3 * x->src.uv_stride + 3] +
|
|
x->src.v_buffer[3 * x->src.uv_stride + 4] +
|
|
x->src.v_buffer[4 * x->src.uv_stride + 3] +
|
|
x->src.v_buffer[4 * x->src.uv_stride + 4]) >>
|
|
2;
|
|
x->is_skin = 0;
|
|
if (!cpi->oxcf.screen_content_mode) {
|
|
int block_index = mb_row * cpi->common.mb_cols + mb_col;
|
|
x->is_skin = is_skin_color(y, cb, cr, cpi->consec_zero_last[block_index]);
|
|
}
|
|
}
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity) {
|
|
// Under aggressive denoising mode, should we use skin map to reduce
|
|
// denoiser
|
|
// and ZEROMV bias? Will need to revisit the accuracy of this detection for
|
|
// very noisy input. For now keep this as is (i.e., don't turn it off).
|
|
// if (cpi->denoiser.denoiser_mode == kDenoiserOnYUVAggressive)
|
|
// x->is_skin = 0;
|
|
}
|
|
#endif
|
|
|
|
mode_mv = mode_mv_sb[sign_bias];
|
|
best_ref_mv.as_int = 0;
|
|
memset(mode_mv_sb, 0, sizeof(mode_mv_sb));
|
|
memset(&best_mbmode, 0, sizeof(best_mbmode));
|
|
|
|
/* Setup search priorities */
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
if (parent_ref_valid && parent_ref_frame && dissim < 8) {
|
|
ref_frame_map[0] = -1;
|
|
ref_frame_map[1] = parent_ref_frame;
|
|
ref_frame_map[2] = -1;
|
|
ref_frame_map[3] = -1;
|
|
} else
|
|
#endif
|
|
get_reference_search_order(cpi, ref_frame_map);
|
|
|
|
/* Check to see if there is at least 1 valid reference frame that we need
|
|
* to calculate near_mvs.
|
|
*/
|
|
if (ref_frame_map[1] > 0) {
|
|
sign_bias = vp8_find_near_mvs_bias(
|
|
&x->e_mbd, x->e_mbd.mode_info_context, mode_mv_sb, best_ref_mv_sb,
|
|
mdcounts, ref_frame_map[1], cpi->common.ref_frame_sign_bias);
|
|
|
|
mode_mv = mode_mv_sb[sign_bias];
|
|
best_ref_mv.as_int = best_ref_mv_sb[sign_bias].as_int;
|
|
}
|
|
|
|
/* Count of the number of MBs tested so far this frame */
|
|
x->mbs_tested_so_far++;
|
|
|
|
*returnintra = INT_MAX;
|
|
x->skip = 0;
|
|
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame = INTRA_FRAME;
|
|
|
|
/* If the frame has big static background and current MB is in low
|
|
* motion area, its mode decision is biased to ZEROMV mode.
|
|
* No adjustment if cpu_used is <= -12 (i.e., cpi->Speed >= 12).
|
|
* At such speed settings, ZEROMV is already heavily favored.
|
|
*/
|
|
if (cpi->Speed < 12) {
|
|
calculate_zeromv_rd_adjustment(cpi, x, &rd_adjustment);
|
|
}
|
|
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity) {
|
|
rd_adjustment = (int)(rd_adjustment *
|
|
cpi->denoiser.denoise_pars.pickmode_mv_bias / 100);
|
|
}
|
|
#endif
|
|
|
|
if (dot_artifact_candidate) {
|
|
// Bias against ZEROMV_LAST mode.
|
|
rd_adjustment = 150;
|
|
}
|
|
|
|
/* if we encode a new mv this is important
|
|
* find the best new motion vector
|
|
*/
|
|
for (mode_index = 0; mode_index < MAX_MODES; ++mode_index) {
|
|
int frame_cost;
|
|
int this_rd = INT_MAX;
|
|
int this_ref_frame = ref_frame_map[vp8_ref_frame_order[mode_index]];
|
|
|
|
if (best_rd <= x->rd_threshes[mode_index]) continue;
|
|
|
|
if (this_ref_frame < 0) continue;
|
|
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame = this_ref_frame;
|
|
|
|
/* everything but intra */
|
|
if (x->e_mbd.mode_info_context->mbmi.ref_frame) {
|
|
x->e_mbd.pre.y_buffer = plane[this_ref_frame][0];
|
|
x->e_mbd.pre.u_buffer = plane[this_ref_frame][1];
|
|
x->e_mbd.pre.v_buffer = plane[this_ref_frame][2];
|
|
|
|
if (sign_bias != cpi->common.ref_frame_sign_bias[this_ref_frame]) {
|
|
sign_bias = cpi->common.ref_frame_sign_bias[this_ref_frame];
|
|
mode_mv = mode_mv_sb[sign_bias];
|
|
best_ref_mv.as_int = best_ref_mv_sb[sign_bias].as_int;
|
|
}
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
if (parent_ref_valid) {
|
|
if (vp8_mode_order[mode_index] == NEARESTMV &&
|
|
mode_mv[NEARESTMV].as_int == 0)
|
|
continue;
|
|
if (vp8_mode_order[mode_index] == NEARMV && mode_mv[NEARMV].as_int == 0)
|
|
continue;
|
|
|
|
if (vp8_mode_order[mode_index] == NEWMV && parent_mode == ZEROMV &&
|
|
best_ref_mv.as_int == 0)
|
|
continue;
|
|
else if (vp8_mode_order[mode_index] == NEWMV && dissim == 0 &&
|
|
best_ref_mv.as_int == parent_ref_mv.as_int)
|
|
continue;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/* Check to see if the testing frequency for this mode is at its max
|
|
* If so then prevent it from being tested and increase the threshold
|
|
* for its testing */
|
|
if (x->mode_test_hit_counts[mode_index] &&
|
|
(cpi->mode_check_freq[mode_index] > 1)) {
|
|
if (x->mbs_tested_so_far <= (cpi->mode_check_freq[mode_index] *
|
|
x->mode_test_hit_counts[mode_index])) {
|
|
/* Increase the threshold for coding this mode to make it less
|
|
* likely to be chosen */
|
|
x->rd_thresh_mult[mode_index] += 4;
|
|
|
|
if (x->rd_thresh_mult[mode_index] > MAX_THRESHMULT) {
|
|
x->rd_thresh_mult[mode_index] = MAX_THRESHMULT;
|
|
}
|
|
|
|
x->rd_threshes[mode_index] =
|
|
(cpi->rd_baseline_thresh[mode_index] >> 7) *
|
|
x->rd_thresh_mult[mode_index];
|
|
continue;
|
|
}
|
|
}
|
|
|
|
/* We have now reached the point where we are going to test the current
|
|
* mode so increment the counter for the number of times it has been
|
|
* tested */
|
|
x->mode_test_hit_counts[mode_index]++;
|
|
|
|
rate2 = 0;
|
|
distortion2 = 0;
|
|
|
|
this_mode = vp8_mode_order[mode_index];
|
|
|
|
x->e_mbd.mode_info_context->mbmi.mode = this_mode;
|
|
x->e_mbd.mode_info_context->mbmi.uv_mode = DC_PRED;
|
|
|
|
/* Work out the cost assosciated with selecting the reference frame */
|
|
frame_cost = x->ref_frame_cost[x->e_mbd.mode_info_context->mbmi.ref_frame];
|
|
rate2 += frame_cost;
|
|
|
|
/* Only consider ZEROMV/ALTREF_FRAME for alt ref frame,
|
|
* unless ARNR filtering is enabled in which case we want
|
|
* an unfiltered alternative */
|
|
if (cpi->is_src_frame_alt_ref && (cpi->oxcf.arnr_max_frames == 0)) {
|
|
if (this_mode != ZEROMV ||
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame != ALTREF_FRAME) {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
switch (this_mode) {
|
|
case B_PRED:
|
|
/* Pass best so far to pick_intra4x4mby_modes to use as breakout */
|
|
distortion2 = best_rd_sse;
|
|
pick_intra4x4mby_modes(x, &rate, &distortion2);
|
|
|
|
if (distortion2 == INT_MAX) {
|
|
this_rd = INT_MAX;
|
|
} else {
|
|
rate2 += rate;
|
|
distortion2 = vpx_variance16x16(*(b->base_src), b->src_stride,
|
|
x->e_mbd.predictor, 16, &sse);
|
|
this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2);
|
|
|
|
if (this_rd < best_intra_rd) {
|
|
best_intra_rd = this_rd;
|
|
*returnintra = distortion2;
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
case SPLITMV:
|
|
|
|
/* Split MV modes currently not supported when RD is not enabled. */
|
|
break;
|
|
|
|
case DC_PRED:
|
|
case V_PRED:
|
|
case H_PRED:
|
|
case TM_PRED:
|
|
vp8_build_intra_predictors_mby_s(
|
|
xd, xd->dst.y_buffer - xd->dst.y_stride, xd->dst.y_buffer - 1,
|
|
xd->dst.y_stride, xd->predictor, 16);
|
|
distortion2 = vpx_variance16x16(*(b->base_src), b->src_stride,
|
|
x->e_mbd.predictor, 16, &sse);
|
|
rate2 += x->mbmode_cost[x->e_mbd.frame_type]
|
|
[x->e_mbd.mode_info_context->mbmi.mode];
|
|
this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2);
|
|
|
|
if (this_rd < best_intra_rd) {
|
|
best_intra_rd = this_rd;
|
|
*returnintra = distortion2;
|
|
}
|
|
break;
|
|
|
|
case NEWMV: {
|
|
int thissme;
|
|
int step_param;
|
|
int further_steps;
|
|
int n = 0;
|
|
int sadpb = x->sadperbit16;
|
|
int_mv mvp_full;
|
|
|
|
int col_min = ((best_ref_mv.as_mv.col + 7) >> 3) - MAX_FULL_PEL_VAL;
|
|
int row_min = ((best_ref_mv.as_mv.row + 7) >> 3) - MAX_FULL_PEL_VAL;
|
|
int col_max = (best_ref_mv.as_mv.col >> 3) + MAX_FULL_PEL_VAL;
|
|
int row_max = (best_ref_mv.as_mv.row >> 3) + MAX_FULL_PEL_VAL;
|
|
|
|
int tmp_col_min = x->mv_col_min;
|
|
int tmp_col_max = x->mv_col_max;
|
|
int tmp_row_min = x->mv_row_min;
|
|
int tmp_row_max = x->mv_row_max;
|
|
|
|
int speed_adjust = (cpi->Speed > 5) ? ((cpi->Speed >= 8) ? 3 : 2) : 1;
|
|
|
|
/* Further step/diamond searches as necessary */
|
|
step_param = cpi->sf.first_step + speed_adjust;
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
/* If lower-res frame is not available for mv reuse (because of
|
|
frame dropping or different temporal layer pattern), then higher
|
|
resol encoder does motion search without any previous knowledge.
|
|
Also, since last frame motion info is not stored, then we can not
|
|
use improved_mv_pred. */
|
|
if (cpi->oxcf.mr_encoder_id) sf_improved_mv_pred = 0;
|
|
|
|
// Only use parent MV as predictor if this candidate reference frame
|
|
// (|this_ref_frame|) is equal to |parent_ref_frame|.
|
|
if (parent_ref_valid && (parent_ref_frame == this_ref_frame)) {
|
|
/* Use parent MV as predictor. Adjust search range
|
|
* accordingly.
|
|
*/
|
|
mvp.as_int = parent_ref_mv.as_int;
|
|
mvp_full.as_mv.col = parent_ref_mv.as_mv.col >> 3;
|
|
mvp_full.as_mv.row = parent_ref_mv.as_mv.row >> 3;
|
|
|
|
if (dissim <= 32)
|
|
step_param += 3;
|
|
else if (dissim <= 128)
|
|
step_param += 2;
|
|
else
|
|
step_param += 1;
|
|
} else
|
|
#endif
|
|
{
|
|
if (sf_improved_mv_pred) {
|
|
if (!saddone) {
|
|
vp8_cal_sad(cpi, xd, x, recon_yoffset, &near_sadidx[0]);
|
|
saddone = 1;
|
|
}
|
|
|
|
vp8_mv_pred(cpi, &x->e_mbd, x->e_mbd.mode_info_context, &mvp,
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame,
|
|
cpi->common.ref_frame_sign_bias, &sr, &near_sadidx[0]);
|
|
|
|
sr += speed_adjust;
|
|
/* adjust search range according to sr from mv prediction */
|
|
if (sr > step_param) step_param = sr;
|
|
|
|
mvp_full.as_mv.col = mvp.as_mv.col >> 3;
|
|
mvp_full.as_mv.row = mvp.as_mv.row >> 3;
|
|
} else {
|
|
mvp.as_int = best_ref_mv.as_int;
|
|
mvp_full.as_mv.col = best_ref_mv.as_mv.col >> 3;
|
|
mvp_full.as_mv.row = best_ref_mv.as_mv.row >> 3;
|
|
}
|
|
}
|
|
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
if (parent_ref_valid && (parent_ref_frame == this_ref_frame) &&
|
|
dissim <= 2 &&
|
|
VPXMAX(abs(best_ref_mv.as_mv.row - parent_ref_mv.as_mv.row),
|
|
abs(best_ref_mv.as_mv.col - parent_ref_mv.as_mv.col)) <= 4) {
|
|
d->bmi.mv.as_int = mvp_full.as_int;
|
|
mode_mv[NEWMV].as_int = mvp_full.as_int;
|
|
|
|
cpi->find_fractional_mv_step(
|
|
x, b, d, &d->bmi.mv, &best_ref_mv, x->errorperbit,
|
|
&cpi->fn_ptr[BLOCK_16X16], cpi->mb.mvcost, &distortion2, &sse);
|
|
} else
|
|
#endif
|
|
{
|
|
/* Get intersection of UMV window and valid MV window to
|
|
* reduce # of checks in diamond search. */
|
|
if (x->mv_col_min < col_min) x->mv_col_min = col_min;
|
|
if (x->mv_col_max > col_max) x->mv_col_max = col_max;
|
|
if (x->mv_row_min < row_min) x->mv_row_min = row_min;
|
|
if (x->mv_row_max > row_max) x->mv_row_max = row_max;
|
|
|
|
further_steps =
|
|
(cpi->Speed >= 8)
|
|
? 0
|
|
: (cpi->sf.max_step_search_steps - 1 - step_param);
|
|
|
|
if (cpi->sf.search_method == HEX) {
|
|
#if CONFIG_MULTI_RES_ENCODING
|
|
/* TODO: In higher-res pick_inter_mode, step_param is used to
|
|
* modify hex search range. Here, set step_param to 0 not to
|
|
* change the behavior in lowest-resolution encoder.
|
|
* Will improve it later.
|
|
*/
|
|
/* Set step_param to 0 to ensure large-range motion search
|
|
* when mv reuse if not valid (i.e. |parent_ref_valid| = 0),
|
|
* or if this candidate reference frame (|this_ref_frame|) is
|
|
* not equal to |parent_ref_frame|.
|
|
*/
|
|
if (!parent_ref_valid || (parent_ref_frame != this_ref_frame))
|
|
step_param = 0;
|
|
#endif
|
|
bestsme = vp8_hex_search(x, b, d, &mvp_full, &d->bmi.mv, step_param,
|
|
sadpb, &cpi->fn_ptr[BLOCK_16X16],
|
|
x->mvsadcost, x->mvcost, &best_ref_mv);
|
|
mode_mv[NEWMV].as_int = d->bmi.mv.as_int;
|
|
} else {
|
|
bestsme = cpi->diamond_search_sad(
|
|
x, b, d, &mvp_full, &d->bmi.mv, step_param, sadpb, &num00,
|
|
&cpi->fn_ptr[BLOCK_16X16], x->mvcost, &best_ref_mv);
|
|
mode_mv[NEWMV].as_int = d->bmi.mv.as_int;
|
|
|
|
/* Further step/diamond searches as necessary */
|
|
n = num00;
|
|
num00 = 0;
|
|
|
|
while (n < further_steps) {
|
|
n++;
|
|
|
|
if (num00) {
|
|
num00--;
|
|
} else {
|
|
thissme = cpi->diamond_search_sad(
|
|
x, b, d, &mvp_full, &d->bmi.mv, step_param + n, sadpb,
|
|
&num00, &cpi->fn_ptr[BLOCK_16X16], x->mvcost, &best_ref_mv);
|
|
if (thissme < bestsme) {
|
|
bestsme = thissme;
|
|
mode_mv[NEWMV].as_int = d->bmi.mv.as_int;
|
|
} else {
|
|
d->bmi.mv.as_int = mode_mv[NEWMV].as_int;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
x->mv_col_min = tmp_col_min;
|
|
x->mv_col_max = tmp_col_max;
|
|
x->mv_row_min = tmp_row_min;
|
|
x->mv_row_max = tmp_row_max;
|
|
|
|
if (bestsme < INT_MAX) {
|
|
cpi->find_fractional_mv_step(
|
|
x, b, d, &d->bmi.mv, &best_ref_mv, x->errorperbit,
|
|
&cpi->fn_ptr[BLOCK_16X16], cpi->mb.mvcost, &distortion2, &sse);
|
|
}
|
|
}
|
|
|
|
mode_mv[NEWMV].as_int = d->bmi.mv.as_int;
|
|
// The clamp below is not necessary from the perspective
|
|
// of VP8 bitstream, but is added to improve ChromeCast
|
|
// mirroring's robustness. Please do not remove.
|
|
vp8_clamp_mv2(&mode_mv[this_mode], xd);
|
|
/* mv cost; */
|
|
rate2 +=
|
|
vp8_mv_bit_cost(&mode_mv[NEWMV], &best_ref_mv, cpi->mb.mvcost, 128);
|
|
}
|
|
|
|
case NEARESTMV:
|
|
case NEARMV:
|
|
if (mode_mv[this_mode].as_int == 0) continue;
|
|
|
|
case ZEROMV:
|
|
|
|
/* Trap vectors that reach beyond the UMV borders
|
|
* Note that ALL New MV, Nearest MV Near MV and Zero MV code drops
|
|
* through to this point because of the lack of break statements
|
|
* in the previous two cases.
|
|
*/
|
|
if (((mode_mv[this_mode].as_mv.row >> 3) < x->mv_row_min) ||
|
|
((mode_mv[this_mode].as_mv.row >> 3) > x->mv_row_max) ||
|
|
((mode_mv[this_mode].as_mv.col >> 3) < x->mv_col_min) ||
|
|
((mode_mv[this_mode].as_mv.col >> 3) > x->mv_col_max)) {
|
|
continue;
|
|
}
|
|
|
|
rate2 += vp8_cost_mv_ref(this_mode, mdcounts);
|
|
x->e_mbd.mode_info_context->mbmi.mv.as_int = mode_mv[this_mode].as_int;
|
|
this_rd = evaluate_inter_mode(&sse, rate2, &distortion2, cpi, x,
|
|
rd_adjustment);
|
|
|
|
break;
|
|
default: break;
|
|
}
|
|
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity) {
|
|
/* Store for later use by denoiser. */
|
|
// Dont' denoise with GOLDEN OR ALTREF is they are old reference
|
|
// frames (greater than MAX_GF_ARF_DENOISE_RANGE frames in past).
|
|
int skip_old_reference = ((this_ref_frame != LAST_FRAME) &&
|
|
(cpi->common.current_video_frame -
|
|
cpi->current_ref_frames[this_ref_frame] >
|
|
MAX_GF_ARF_DENOISE_RANGE))
|
|
? 1
|
|
: 0;
|
|
if (this_mode == ZEROMV && sse < zero_mv_sse && !skip_old_reference) {
|
|
zero_mv_sse = sse;
|
|
x->best_zeromv_reference_frame =
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame;
|
|
}
|
|
|
|
// Store the best NEWMV in x for later use in the denoiser.
|
|
if (x->e_mbd.mode_info_context->mbmi.mode == NEWMV && sse < best_sse &&
|
|
!skip_old_reference) {
|
|
best_sse = sse;
|
|
x->best_sse_inter_mode = NEWMV;
|
|
x->best_sse_mv = x->e_mbd.mode_info_context->mbmi.mv;
|
|
x->need_to_clamp_best_mvs =
|
|
x->e_mbd.mode_info_context->mbmi.need_to_clamp_mvs;
|
|
x->best_reference_frame = x->e_mbd.mode_info_context->mbmi.ref_frame;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
if (this_rd < best_rd || x->skip) {
|
|
/* Note index of best mode */
|
|
best_mode_index = mode_index;
|
|
|
|
*returnrate = rate2;
|
|
*returndistortion = distortion2;
|
|
best_rd_sse = sse;
|
|
best_rd = this_rd;
|
|
memcpy(&best_mbmode, &x->e_mbd.mode_info_context->mbmi,
|
|
sizeof(MB_MODE_INFO));
|
|
|
|
/* Testing this mode gave rise to an improvement in best error
|
|
* score. Lower threshold a bit for next time
|
|
*/
|
|
x->rd_thresh_mult[mode_index] =
|
|
(x->rd_thresh_mult[mode_index] >= (MIN_THRESHMULT + 2))
|
|
? x->rd_thresh_mult[mode_index] - 2
|
|
: MIN_THRESHMULT;
|
|
x->rd_threshes[mode_index] = (cpi->rd_baseline_thresh[mode_index] >> 7) *
|
|
x->rd_thresh_mult[mode_index];
|
|
}
|
|
|
|
/* If the mode did not help improve the best error case then raise the
|
|
* threshold for testing that mode next time around.
|
|
*/
|
|
else {
|
|
x->rd_thresh_mult[mode_index] += 4;
|
|
|
|
if (x->rd_thresh_mult[mode_index] > MAX_THRESHMULT) {
|
|
x->rd_thresh_mult[mode_index] = MAX_THRESHMULT;
|
|
}
|
|
|
|
x->rd_threshes[mode_index] = (cpi->rd_baseline_thresh[mode_index] >> 7) *
|
|
x->rd_thresh_mult[mode_index];
|
|
}
|
|
|
|
if (x->skip) break;
|
|
}
|
|
|
|
/* Reduce the activation RD thresholds for the best choice mode */
|
|
if ((cpi->rd_baseline_thresh[best_mode_index] > 0) &&
|
|
(cpi->rd_baseline_thresh[best_mode_index] < (INT_MAX >> 2))) {
|
|
int best_adjustment = (x->rd_thresh_mult[best_mode_index] >> 3);
|
|
|
|
x->rd_thresh_mult[best_mode_index] =
|
|
(x->rd_thresh_mult[best_mode_index] >=
|
|
(MIN_THRESHMULT + best_adjustment))
|
|
? x->rd_thresh_mult[best_mode_index] - best_adjustment
|
|
: MIN_THRESHMULT;
|
|
x->rd_threshes[best_mode_index] =
|
|
(cpi->rd_baseline_thresh[best_mode_index] >> 7) *
|
|
x->rd_thresh_mult[best_mode_index];
|
|
}
|
|
|
|
{
|
|
int this_rdbin = (*returndistortion >> 7);
|
|
|
|
if (this_rdbin >= 1024) {
|
|
this_rdbin = 1023;
|
|
}
|
|
|
|
x->error_bins[this_rdbin]++;
|
|
}
|
|
|
|
#if CONFIG_TEMPORAL_DENOISING
|
|
if (cpi->oxcf.noise_sensitivity) {
|
|
int block_index = mb_row * cpi->common.mb_cols + mb_col;
|
|
int reevaluate = 0;
|
|
int is_noisy = 0;
|
|
if (x->best_sse_inter_mode == DC_PRED) {
|
|
/* No best MV found. */
|
|
x->best_sse_inter_mode = best_mbmode.mode;
|
|
x->best_sse_mv = best_mbmode.mv;
|
|
x->need_to_clamp_best_mvs = best_mbmode.need_to_clamp_mvs;
|
|
x->best_reference_frame = best_mbmode.ref_frame;
|
|
best_sse = best_rd_sse;
|
|
}
|
|
// For non-skin blocks that have selected ZEROMV for this current frame,
|
|
// and have been selecting ZEROMV_LAST (on the base layer frame) at
|
|
// least |x~20| consecutive past frames in a row, label the block for
|
|
// possible increase in denoising strength. We also condition this
|
|
// labeling on there being significant denoising in the scene
|
|
if (cpi->oxcf.noise_sensitivity == 4) {
|
|
if (cpi->denoiser.nmse_source_diff >
|
|
70 * cpi->denoiser.threshold_aggressive_mode / 100) {
|
|
is_noisy = 1;
|
|
}
|
|
} else {
|
|
if (cpi->mse_source_denoised > 1000) is_noisy = 1;
|
|
}
|
|
x->increase_denoising = 0;
|
|
if (!x->is_skin && x->best_sse_inter_mode == ZEROMV &&
|
|
(x->best_reference_frame == LAST_FRAME ||
|
|
x->best_reference_frame == cpi->closest_reference_frame) &&
|
|
cpi->consec_zero_last[block_index] >= 20 && is_noisy) {
|
|
x->increase_denoising = 1;
|
|
}
|
|
x->denoise_zeromv = 0;
|
|
vp8_denoiser_denoise_mb(&cpi->denoiser, x, best_sse, zero_mv_sse,
|
|
recon_yoffset, recon_uvoffset, &cpi->common.lf_info,
|
|
mb_row, mb_col, block_index,
|
|
cpi->consec_zero_last_mvbias[block_index]);
|
|
|
|
// Reevaluate ZEROMV after denoising: for large noise content
|
|
// (i.e., cpi->mse_source_denoised is above threshold), do this for all
|
|
// blocks that did not pick ZEROMV as best mode but are using ZEROMV
|
|
// for denoising. Otherwise, always re-evaluate for blocks that picked
|
|
// INTRA mode as best mode.
|
|
// Avoid blocks that have been biased against ZERO_LAST
|
|
// (i.e., dot artifact candidate blocks).
|
|
reevaluate = (best_mbmode.ref_frame == INTRA_FRAME) ||
|
|
(best_mbmode.mode != ZEROMV && x->denoise_zeromv &&
|
|
cpi->mse_source_denoised > 2000);
|
|
if (!dot_artifact_candidate && reevaluate &&
|
|
x->best_zeromv_reference_frame != INTRA_FRAME) {
|
|
int this_rd = 0;
|
|
int this_ref_frame = x->best_zeromv_reference_frame;
|
|
rd_adjustment = 100;
|
|
rate2 =
|
|
x->ref_frame_cost[this_ref_frame] + vp8_cost_mv_ref(ZEROMV, mdcounts);
|
|
distortion2 = 0;
|
|
|
|
/* set up the proper prediction buffers for the frame */
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame = this_ref_frame;
|
|
x->e_mbd.pre.y_buffer = plane[this_ref_frame][0];
|
|
x->e_mbd.pre.u_buffer = plane[this_ref_frame][1];
|
|
x->e_mbd.pre.v_buffer = plane[this_ref_frame][2];
|
|
|
|
x->e_mbd.mode_info_context->mbmi.mode = ZEROMV;
|
|
x->e_mbd.mode_info_context->mbmi.uv_mode = DC_PRED;
|
|
x->e_mbd.mode_info_context->mbmi.mv.as_int = 0;
|
|
this_rd =
|
|
evaluate_inter_mode(&sse, rate2, &distortion2, cpi, x, rd_adjustment);
|
|
|
|
if (this_rd < best_rd) {
|
|
memcpy(&best_mbmode, &x->e_mbd.mode_info_context->mbmi,
|
|
sizeof(MB_MODE_INFO));
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
if (cpi->is_src_frame_alt_ref &&
|
|
(best_mbmode.mode != ZEROMV || best_mbmode.ref_frame != ALTREF_FRAME)) {
|
|
x->e_mbd.mode_info_context->mbmi.mode = ZEROMV;
|
|
x->e_mbd.mode_info_context->mbmi.ref_frame = ALTREF_FRAME;
|
|
x->e_mbd.mode_info_context->mbmi.mv.as_int = 0;
|
|
x->e_mbd.mode_info_context->mbmi.uv_mode = DC_PRED;
|
|
x->e_mbd.mode_info_context->mbmi.mb_skip_coeff =
|
|
(cpi->common.mb_no_coeff_skip);
|
|
x->e_mbd.mode_info_context->mbmi.partitioning = 0;
|
|
|
|
return;
|
|
}
|
|
|
|
/* set to the best mb mode, this copy can be skip if x->skip since it
|
|
* already has the right content */
|
|
if (!x->skip) {
|
|
memcpy(&x->e_mbd.mode_info_context->mbmi, &best_mbmode,
|
|
sizeof(MB_MODE_INFO));
|
|
}
|
|
|
|
if (best_mbmode.mode <= B_PRED) {
|
|
/* set mode_info_context->mbmi.uv_mode */
|
|
pick_intra_mbuv_mode(x);
|
|
}
|
|
|
|
if (sign_bias !=
|
|
cpi->common.ref_frame_sign_bias[xd->mode_info_context->mbmi.ref_frame]) {
|
|
best_ref_mv.as_int = best_ref_mv_sb[!sign_bias].as_int;
|
|
}
|
|
|
|
update_mvcount(x, &best_ref_mv);
|
|
}
|
|
|
|
void vp8_pick_intra_mode(MACROBLOCK *x, int *rate_) {
|
|
int error4x4, error16x16 = INT_MAX;
|
|
int rate, best_rate = 0, distortion, best_sse;
|
|
MB_PREDICTION_MODE mode, best_mode = DC_PRED;
|
|
int this_rd;
|
|
unsigned int sse;
|
|
BLOCK *b = &x->block[0];
|
|
MACROBLOCKD *xd = &x->e_mbd;
|
|
|
|
xd->mode_info_context->mbmi.ref_frame = INTRA_FRAME;
|
|
|
|
pick_intra_mbuv_mode(x);
|
|
|
|
for (mode = DC_PRED; mode <= TM_PRED; ++mode) {
|
|
xd->mode_info_context->mbmi.mode = mode;
|
|
vp8_build_intra_predictors_mby_s(xd, xd->dst.y_buffer - xd->dst.y_stride,
|
|
xd->dst.y_buffer - 1, xd->dst.y_stride,
|
|
xd->predictor, 16);
|
|
distortion = vpx_variance16x16(*(b->base_src), b->src_stride, xd->predictor,
|
|
16, &sse);
|
|
rate = x->mbmode_cost[xd->frame_type][mode];
|
|
this_rd = RDCOST(x->rdmult, x->rddiv, rate, distortion);
|
|
|
|
if (error16x16 > this_rd) {
|
|
error16x16 = this_rd;
|
|
best_mode = mode;
|
|
best_sse = sse;
|
|
best_rate = rate;
|
|
}
|
|
}
|
|
xd->mode_info_context->mbmi.mode = best_mode;
|
|
|
|
error4x4 = pick_intra4x4mby_modes(x, &rate, &best_sse);
|
|
if (error4x4 < error16x16) {
|
|
xd->mode_info_context->mbmi.mode = B_PRED;
|
|
best_rate = rate;
|
|
}
|
|
|
|
*rate_ = best_rate;
|
|
}
|