123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347 |
- /*
- * 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 <assert.h>
- #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 "vp8/common/vp8_skin_detection.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
- extern const int vp8_ref_frame_order[MAX_MODES];
- extern const MB_PREDICTION_MODE vp8_mode_order[MAX_MODES];
- 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 *= 8;
- bestmv->as_mv.col *= 8;
- 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 best_mode = B_MODE_COUNT;
- int r = 0, d = 0;
- 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;
- assert(best_mode != B_MODE_COUNT);
- 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 best_mode = MB_MODE_COUNT;
- 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;
- }
- }
- assert(best_mode != MB_MODE_COUNT);
- 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] = { { 0 }, { 0 } };
- 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] = { { 0, 0 } };
- 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
- assert(plane[LAST_FRAME][0] != NULL);
- 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) {
- assert(plane[LAST_FRAME][1] != NULL);
- 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) {
- assert(plane[LAST_FRAME][2] != NULL);
- 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.
- x->is_skin = 0;
- if (!cpi->oxcf.screen_content_mode) {
- int block_index = mb_row * cpi->common.mb_cols + mb_col;
- x->is_skin = cpi->skin_map[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);
- }
- // fall through
- case NEARESTMV:
- case NEARMV:
- if (mode_mv[this_mode].as_int == 0) continue;
- // fall through
- 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;
- }
|