pickinter.c 42 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
7
 *  in the file PATENTS.  All contributing project authors may
8
 *  be found in the AUTHORS file in the root of the source tree.
John Koleszar's avatar
John Koleszar committed
9 10 11 12
 */


#include <limits.h>
13
#include "vpx_config.h"
John Koleszar's avatar
John Koleszar committed
14 15 16
#include "onyx_int.h"
#include "modecosts.h"
#include "encodeintra.h"
John Koleszar's avatar
John Koleszar committed
17
#include "vp8/common/entropymode.h"
John Koleszar's avatar
John Koleszar committed
18
#include "pickinter.h"
John Koleszar's avatar
John Koleszar committed
19
#include "vp8/common/findnearmv.h"
John Koleszar's avatar
John Koleszar committed
20
#include "encodemb.h"
John Koleszar's avatar
John Koleszar committed
21 22
#include "vp8/common/reconinter.h"
#include "vp8/common/reconintra4x4.h"
23
#include "vp8/common/variance.h"
John Koleszar's avatar
John Koleszar committed
24
#include "mcomp.h"
25
#include "rdopt.h"
John Koleszar's avatar
John Koleszar committed
26
#include "vpx_mem/vpx_mem.h"
27 28 29
#if CONFIG_TEMPORAL_DENOISING
#include "denoising.h"
#endif
John Koleszar's avatar
John Koleszar committed
30

John Koleszar's avatar
John Koleszar committed
31
extern int VP8_UVSSE(MACROBLOCK *x);
John Koleszar's avatar
John Koleszar committed
32 33 34 35 36

#ifdef SPEEDSTATS
extern unsigned int cnt_pm;
#endif

37
extern const int vp8_ref_frame_order[MAX_MODES];
John Koleszar's avatar
John Koleszar committed
38 39 40 41 42
extern const MB_PREDICTION_MODE vp8_mode_order[MAX_MODES];

extern int vp8_cost_mv_ref(MB_PREDICTION_MODE m, const int near_mv_ref_ct[4]);


Scott LaVarnway's avatar
Scott LaVarnway committed
43 44 45 46 47 48
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)
John Koleszar's avatar
John Koleszar committed
49 50 51 52 53
{
    (void) b;
    (void) d;
    (void) ref_mv;
    (void) error_per_bit;
54
    (void) vfp;
John Koleszar's avatar
John Koleszar committed
55
    (void) mvcost;
56
    (void) distortion;
57
    (void) sse;
Scott LaVarnway's avatar
Scott LaVarnway committed
58 59
    bestmv->as_mv.row <<= 3;
    bestmv->as_mv.col <<= 3;
John Koleszar's avatar
John Koleszar committed
60 61 62 63
    return 0;
}


64
int vp8_get_inter_mbpred_error(MACROBLOCK *mb,
65 66 67
                                  const vp8_variance_fn_ptr_t *vfp,
                                  unsigned int *sse,
                                  int_mv this_mv)
John Koleszar's avatar
John Koleszar committed
68 69 70 71 72 73
{

    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;
Scott LaVarnway's avatar
Scott LaVarnway committed
74 75 76
    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;
77 78
    int xoffset = this_mv.as_mv.col & 7;
    int yoffset = this_mv.as_mv.row & 7;
John Koleszar's avatar
John Koleszar committed
79

Scott LaVarnway's avatar
Scott LaVarnway committed
80
    in_what += (this_mv.as_mv.row >> 3) * pre_stride + (this_mv.as_mv.col >> 3);
John Koleszar's avatar
John Koleszar committed
81 82 83

    if (xoffset | yoffset)
    {
84
        return vfp->svf(in_what, in_what_stride, xoffset, yoffset, what, what_stride, sse);
John Koleszar's avatar
John Koleszar committed
85 86 87
    }
    else
    {
88
        return vfp->vf(what, what_stride, in_what, in_what_stride, sse);
John Koleszar's avatar
John Koleszar committed
89 90 91 92 93 94 95
    }

}


unsigned int vp8_get4x4sse_cs_c
(
96
    const unsigned char *src_ptr,
John Koleszar's avatar
John Koleszar committed
97
    int  source_stride,
98
    const unsigned char *ref_ptr,
99
    int  recon_stride
John Koleszar's avatar
John Koleszar committed
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
)
{
    int distortion = 0;
    int r, c;

    for (r = 0; r < 4; r++)
    {
        for (c = 0; c < 4; c++)
        {
            int diff = src_ptr[c] - ref_ptr[c];
            distortion += diff * diff;
        }

        src_ptr += source_stride;
        ref_ptr += recon_stride;
    }

    return distortion;
}

John Koleszar's avatar
John Koleszar committed
120
static int get_prediction_error(BLOCK *be, BLOCKD *b)
John Koleszar's avatar
John Koleszar committed
121 122 123 124 125 126
{
    unsigned char *sptr;
    unsigned char *dptr;
    sptr = (*(be->base_src) + be->src);
    dptr = b->predictor;

John Koleszar's avatar
John Koleszar committed
127
    return vp8_get4x4sse_cs(sptr, be->src_stride, dptr, 16);
John Koleszar's avatar
John Koleszar committed
128 129 130 131 132

}

static int pick_intra4x4block(
    MACROBLOCK *x,
133
    int ib,
John Koleszar's avatar
John Koleszar committed
134
    B_PREDICTION_MODE *best_mode,
135
    const int *mode_costs,
John Koleszar's avatar
John Koleszar committed
136 137 138 139

    int *bestrate,
    int *bestdistortion)
{
140 141 142

    BLOCKD *b = &x->e_mbd.block[ib];
    BLOCK *be = &x->block[ib];
Scott LaVarnway's avatar
Scott LaVarnway committed
143
    int dst_stride = x->e_mbd.dst.y_stride;
144
    unsigned char *dst = x->e_mbd.dst.y_buffer + b->offset;
John Koleszar's avatar
John Koleszar committed
145
    B_PREDICTION_MODE mode;
John Koleszar's avatar
John Koleszar committed
146
    int best_rd = INT_MAX;
John Koleszar's avatar
John Koleszar committed
147 148 149
    int rate;
    int distortion;

150 151 152 153
    unsigned char *Above = dst - dst_stride;
    unsigned char *yleft = dst - 1;
    unsigned char top_left = Above[-1];

154
    for (mode = B_DC_PRED; mode <= B_HE_PRED; mode++)
John Koleszar's avatar
John Koleszar committed
155 156 157 158
    {
        int this_rd;

        rate = mode_costs[mode];
159

Johann's avatar
Johann committed
160 161
        vp8_intra4x4_predict(Above, yleft, dst_stride, mode,
                             b->predictor, 16, top_left);
John Koleszar's avatar
John Koleszar committed
162
        distortion = get_prediction_error(be, b);
Yunqing Wang's avatar
Yunqing Wang committed
163
        this_rd = RDCOST(x->rdmult, x->rddiv, rate, distortion);
John Koleszar's avatar
John Koleszar committed
164 165 166 167 168 169 170 171 172

        if (this_rd < best_rd)
        {
            *bestrate = rate;
            *bestdistortion = distortion;
            best_rd = this_rd;
            *best_mode = mode;
        }
    }
Scott LaVarnway's avatar
Scott LaVarnway committed
173

174
    b->bmi.as_mode = *best_mode;
John Koleszar's avatar
John Koleszar committed
175
    vp8_encode_intra4x4block(x, ib);
John Koleszar's avatar
John Koleszar committed
176 177 178 179
    return best_rd;
}


180
static int pick_intra4x4mby_modes
181 182 183 184 185
(
    MACROBLOCK *mb,
    int *Rate,
    int *best_dist
)
John Koleszar's avatar
John Koleszar committed
186 187 188 189
{
    MACROBLOCKD *const xd = &mb->e_mbd;
    int i;
    int cost = mb->mbmode_cost [xd->frame_type] [B_PRED];
190
    int error;
John Koleszar's avatar
John Koleszar committed
191
    int distortion = 0;
192
    const int *bmode_costs;
John Koleszar's avatar
John Koleszar committed
193

Scott LaVarnway's avatar
Scott LaVarnway committed
194
    intra_prediction_down_copy(xd, xd->dst.y_buffer - xd->dst.y_stride + 16);
John Koleszar's avatar
John Koleszar committed
195

Scott LaVarnway's avatar
Scott LaVarnway committed
196 197
    bmode_costs = mb->inter_bmode_costs;

John Koleszar's avatar
John Koleszar committed
198 199 200 201
    for (i = 0; i < 16; i++)
    {
        MODE_INFO *const mic = xd->mode_info_context;
        const int mis = xd->mode_info_stride;
Scott LaVarnway's avatar
Scott LaVarnway committed
202

John Koleszar's avatar
John Koleszar committed
203 204 205
        B_PREDICTION_MODE UNINITIALIZED_IS_SAFE(best_mode);
        int UNINITIALIZED_IS_SAFE(r), UNINITIALIZED_IS_SAFE(d);

Scott LaVarnway's avatar
Scott LaVarnway committed
206 207 208 209 210 211 212 213 214
        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];
        }


John Koleszar's avatar
John Koleszar committed
215
        pick_intra4x4block(mb, i, &best_mode, bmode_costs, &r, &d);
John Koleszar's avatar
John Koleszar committed
216 217 218

        cost += r;
        distortion += d;
Scott LaVarnway's avatar
Scott LaVarnway committed
219
        mic->bmi[i].as_mode = best_mode;
John Koleszar's avatar
John Koleszar committed
220

John Koleszar's avatar
John Koleszar committed
221 222 223
        /* Break out case where we have already exceeded best so far value
         * that was passed in
         */
John Koleszar's avatar
John Koleszar committed
224 225 226 227 228 229 230
        if (distortion > *best_dist)
            break;
    }

    *Rate = cost;

    if (i == 16)
231
    {
John Koleszar's avatar
John Koleszar committed
232
        *best_dist = distortion;
Yunqing Wang's avatar
Yunqing Wang committed
233
        error = RDCOST(mb->rdmult, mb->rddiv, cost, distortion);
234
    }
John Koleszar's avatar
John Koleszar committed
235
    else
236
    {
John Koleszar's avatar
John Koleszar committed
237
        *best_dist = INT_MAX;
238 239
        error = INT_MAX;
    }
John Koleszar's avatar
John Koleszar committed
240 241 242 243

    return error;
}

244
static void pick_intra_mbuv_mode(MACROBLOCK *mb)
John Koleszar's avatar
John Koleszar committed
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
{

    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;
        }
    }


388
    mb->e_mbd.mode_info_context->mbmi.uv_mode = best_mode;
John Koleszar's avatar
John Koleszar committed
389 390 391

}

392
static void update_mvcount(VP8_COMP *cpi, MACROBLOCKD *xd, int_mv *best_ref_mv)
393
{
394 395
    /* Split MV modes currently not supported when RD is nopt enabled,
     * therefore, only need to modify MVcount in NEWMV mode. */
396 397
    if (xd->mode_info_context->mbmi.mode == NEWMV)
    {
Scott LaVarnway's avatar
Scott LaVarnway committed
398
        cpi->MVcount[0][mv_max+((xd->mode_info_context->mbmi.mv.as_mv.row -
399
                                      best_ref_mv->as_mv.row) >> 1)]++;
Scott LaVarnway's avatar
Scott LaVarnway committed
400
        cpi->MVcount[1][mv_max+((xd->mode_info_context->mbmi.mv.as_mv.col -
401
                                      best_ref_mv->as_mv.col) >> 1)]++;
402 403 404
    }
}

405 406 407 408 409 410 411 412

#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)
{
413 414
    LOWER_RES_MB_INFO* store_mode_info
                          = ((LOWER_RES_FRAME_INFO*)cpi->oxcf.mr_low_res_mode_info)->mb_info;
415 416 417 418 419 420 421 422 423 424
    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;

425
        parent_mb_row = mb_row*cpi->oxcf.mr_down_sampling_factor.den
426
                    /cpi->oxcf.mr_down_sampling_factor.num;
427
        parent_mb_col = mb_col*cpi->oxcf.mr_down_sampling_factor.den
428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458
                    /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

459 460
static void check_for_encode_breakout(unsigned int sse, MACROBLOCK* x)
{
461 462 463 464 465 466 467 468 469
    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 )
470
    {
John Koleszar's avatar
John Koleszar committed
471
        /* Check u and v to make sure skip is ok */
472
        unsigned int sse2 = 0;
473 474 475 476 477 478 479 480 481 482

        sse2 = VP8_UVSSE(x);

        if (sse2 * 2 < x->encode_breakout)
            x->skip = 1;
        else
            x->skip = 0;
    }
}

483 484
static int evaluate_inter_mode(unsigned int* sse, int rate2, int* distortion2,
                               VP8_COMP *cpi, MACROBLOCK *x, int rd_adj)
485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500
{
    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;
    /* 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)
501
        *distortion2 = vp8_get_inter_mbpred_error(x,
502 503 504 505 506
                                              &cpi->fn_ptr[BLOCK_16X16],
                                              sse, mv);

    this_rd = RDCOST(x->rdmult, x->rddiv, rate2, *distortion2);

507 508 509 510 511 512 513 514 515
    /* Adjust rd to bias to ZEROMV */
    if(this_mode == ZEROMV)
    {
        /* Bias to ZEROMV on LAST_FRAME reference when it is available. */
        if ((cpi->ref_frame_flags & VP8_LAST_FRAME &
            cpi->common.refresh_last_frame)
            && x->e_mbd.mode_info_context->mbmi.ref_frame != LAST_FRAME)
            rd_adj = 100;

516 517
        // rd_adj <= 100
        this_rd = ((int64_t)this_rd) * rd_adj / 100;
518 519
    }

520 521 522
    check_for_encode_breakout(*sse, x);
    return this_rd;
}
523

524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
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;
    }
}

565 566
void vp8_pick_inter_mode(VP8_COMP *cpi, MACROBLOCK *x, int recon_yoffset,
                         int recon_uvoffset, int *returnrate,
567 568
                         int *returndistortion, int *returnintra, int mb_row,
                         int mb_col)
John Koleszar's avatar
John Koleszar committed
569 570 571 572 573
{
    BLOCK *b = &x->block[0];
    BLOCKD *d = &x->e_mbd.block[0];
    MACROBLOCKD *xd = &x->e_mbd;
    MB_MODE_INFO best_mbmode;
574

575 576
    int_mv best_ref_mv_sb[2];
    int_mv mode_mv_sb[2][MB_MODE_COUNT];
Scott LaVarnway's avatar
Scott LaVarnway committed
577
    int_mv best_ref_mv;
578
    int_mv *mode_mv;
John Koleszar's avatar
John Koleszar committed
579 580 581
    MB_PREDICTION_MODE this_mode;
    int num00;
    int mdcounts[4];
John Koleszar's avatar
John Koleszar committed
582
    int best_rd = INT_MAX;
583
    int rd_adjustment = 100;
John Koleszar's avatar
John Koleszar committed
584 585 586 587 588
    int best_intra_rd = INT_MAX;
    int mode_index;
    int rate;
    int rate2;
    int distortion2;
589
    int bestsme = INT_MAX;
John Koleszar's avatar
John Koleszar committed
590
    int best_mode_index = 0;
591 592
    unsigned int sse = INT_MAX, best_rd_sse = INT_MAX;
#if CONFIG_TEMPORAL_DENOISING
593
    unsigned int zero_mv_sse = INT_MAX, best_sse = INT_MAX;
594
#endif
John Koleszar's avatar
John Koleszar committed
595

Scott LaVarnway's avatar
Scott LaVarnway committed
596
    int_mv mvp;
597

598 599
    int near_sadidx[8] = {0, 1, 2, 3, 4, 5, 6, 7};
    int saddone=0;
John Koleszar's avatar
John Koleszar committed
600 601
    /* search range got from mv_pred(). It uses step_param levels. (0-7) */
    int sr=0;
602

603
    unsigned char *plane[4][3];
604
    int ref_frame_map[4];
605
    int sign_bias = 0;
John Koleszar's avatar
John Koleszar committed
606

607 608 609
#if CONFIG_MULTI_RES_ENCODING
    int dissim = INT_MAX;
    int parent_ref_frame = 0;
610
    int parent_ref_valid = cpi->oxcf.mr_encoder_id && cpi->mr_low_res_mv_avail;
611 612 613
    int_mv parent_ref_mv;
    MB_PREDICTION_MODE parent_mode = 0;

614
    if (parent_ref_valid)
615 616 617
    {
        int parent_ref_flag;

618 619
        get_lower_res_motion_info(cpi, xd, &dissim, &parent_ref_frame,
                                  &parent_mode, &parent_ref_mv, mb_row, mb_col);
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639

        /* 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;
        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);
        else if (parent_ref_frame == ALTREF_FRAME)
            parent_ref_flag = (cpi->ref_frame_flags & VP8_ALTR_FRAME);

        //assert(!parent_ref_frame || parent_ref_flag);
640 641
        if (parent_ref_frame && !parent_ref_flag)
            parent_ref_valid = 0;
642
    }
643
#endif
644

645 646 647
    mode_mv = mode_mv_sb[sign_bias];
    best_ref_mv.as_int = 0;
    vpx_memset(mode_mv_sb, 0, sizeof(mode_mv_sb));
648
    vpx_memset(&best_mbmode, 0, sizeof(best_mbmode));
John Koleszar's avatar
John Koleszar committed
649

650
    /* Setup search priorities */
651
#if CONFIG_MULTI_RES_ENCODING
652
    if (parent_ref_valid && parent_ref_frame && dissim < 8)
653
    {
654
        ref_frame_map[0] = -1;
655 656 657 658 659
        ref_frame_map[1] = parent_ref_frame;
        ref_frame_map[2] = -1;
        ref_frame_map[3] = -1;
    } else
#endif
660
    get_reference_search_order(cpi, ref_frame_map);
John Koleszar's avatar
John Koleszar committed
661

662 663 664 665 666
    /* 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)
    {
667 668 669 670 671 672 673 674 675 676
        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;
677 678
    }

679
    get_predictor_pointers(cpi, plane, recon_yoffset, recon_uvoffset);
John Koleszar's avatar
John Koleszar committed
680

John Koleszar's avatar
John Koleszar committed
681 682
    /* Count of the number of MBs tested so far this frame */
    cpi->mbs_tested_so_far++;
John Koleszar's avatar
John Koleszar committed
683

Paul Wilkins's avatar
Paul Wilkins committed
684
    *returnintra = INT_MAX;
John Koleszar's avatar
John Koleszar committed
685 686
    x->skip = 0;

687
    x->e_mbd.mode_info_context->mbmi.ref_frame = INTRA_FRAME;
John Koleszar's avatar
John Koleszar committed
688

689 690 691 692 693
    /* If the frame has big static background and current MB is in low
     * motion area, its mode decision is biased to ZEROMV mode.
     */
    calculate_zeromv_rd_adjustment(cpi, x, &rd_adjustment);

John Koleszar's avatar
John Koleszar committed
694 695 696
    /* if we encode a new mv this is important
     * find the best new motion vector
     */
John Koleszar's avatar
John Koleszar committed
697 698 699 700
    for (mode_index = 0; mode_index < MAX_MODES; mode_index++)
    {
        int frame_cost;
        int this_rd = INT_MAX;
701
        int this_ref_frame = ref_frame_map[vp8_ref_frame_order[mode_index]];
John Koleszar's avatar
John Koleszar committed
702 703 704 705

        if (best_rd <= cpi->rd_threshes[mode_index])
            continue;

706
        if (this_ref_frame < 0)
John Koleszar's avatar
John Koleszar committed
707 708
            continue;

709 710
        x->e_mbd.mode_info_context->mbmi.ref_frame = this_ref_frame;

John Koleszar's avatar
John Koleszar committed
711
        /* everything but intra */
712
        if (x->e_mbd.mode_info_context->mbmi.ref_frame)
John Koleszar's avatar
John Koleszar committed
713
        {
714 715 716
            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];
717

718
            if (sign_bias != cpi->common.ref_frame_sign_bias[this_ref_frame])
719
            {
720 721 722
                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;
723
            }
724 725

#if CONFIG_MULTI_RES_ENCODING
726
            if (parent_ref_valid)
727 728 729 730 731 732 733 734 735
            {
                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
John Koleszar's avatar
John Koleszar committed
736
                    && best_ref_mv.as_int==0)
737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765
                    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 (cpi->mode_test_hit_counts[mode_index] &&
                                         (cpi->mode_check_freq[mode_index] > 1))
        {
            if (cpi->mbs_tested_so_far <= (cpi->mode_check_freq[mode_index] *
                                         cpi->mode_test_hit_counts[mode_index]))
            {
                /* Increase the threshold for coding this mode to make it less
                 * likely to be chosen */
                cpi->rd_thresh_mult[mode_index] += 4;

                if (cpi->rd_thresh_mult[mode_index] > MAX_THRESHMULT)
                    cpi->rd_thresh_mult[mode_index] = MAX_THRESHMULT;

                cpi->rd_threshes[mode_index] =
                                 (cpi->rd_baseline_thresh[mode_index] >> 7) *
                                 cpi->rd_thresh_mult[mode_index];
                continue;
            }
John Koleszar's avatar
John Koleszar committed
766 767
        }

768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
        /* 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 */
        cpi->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 =
783
            x->ref_frame_cost[x->e_mbd.mode_info_context->mbmi.ref_frame];
784 785 786 787 788
        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 */
789
        if (cpi->is_src_frame_alt_ref && (cpi->oxcf.arnr_max_frames == 0))
John Koleszar's avatar
John Koleszar committed
790
        {
791 792
            if (this_mode != ZEROMV ||
                x->e_mbd.mode_info_context->mbmi.ref_frame != ALTREF_FRAME)
John Koleszar's avatar
John Koleszar committed
793 794 795 796 797 798
                continue;
        }

        switch (this_mode)
        {
        case B_PRED:
799
            /* Pass best so far to pick_intra4x4mby_modes to use as breakout */
800
            distortion2 = best_rd_sse;
John Koleszar's avatar
John Koleszar committed
801
            pick_intra4x4mby_modes(x, &rate, &distortion2);
John Koleszar's avatar
John Koleszar committed
802 803 804 805 806 807 808

            if (distortion2 == INT_MAX)
            {
                this_rd = INT_MAX;
            }
            else
            {
809
                rate2 += rate;
John Koleszar's avatar
John Koleszar committed
810
                distortion2 = vp8_variance16x16(
811
                                    *(b->base_src), b->src_stride,
812
                                    x->e_mbd.predictor, 16, &sse);
Yunqing Wang's avatar
Yunqing Wang committed
813
                this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2);
John Koleszar's avatar
John Koleszar committed
814 815 816 817

                if (this_rd < best_intra_rd)
                {
                    best_intra_rd = this_rd;
Paul Wilkins's avatar
Paul Wilkins committed
818
                    *returnintra = distortion2;
John Koleszar's avatar
John Koleszar committed
819 820 821 822 823 824 825
                }
            }

            break;

        case SPLITMV:

John Koleszar's avatar
John Koleszar committed
826
            /* Split MV modes currently not supported when RD is not enabled. */
John Koleszar's avatar
John Koleszar committed
827 828 829 830 831 832
            break;

        case DC_PRED:
        case V_PRED:
        case H_PRED:
        case TM_PRED:
833 834 835 836 837 838
            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);
John Koleszar's avatar
John Koleszar committed
839
            distortion2 = vp8_variance16x16
840
                                          (*(b->base_src), b->src_stride,
841
                                          x->e_mbd.predictor, 16, &sse);
842
            rate2 += x->mbmode_cost[x->e_mbd.frame_type][x->e_mbd.mode_info_context->mbmi.mode];
Yunqing Wang's avatar
Yunqing Wang committed
843
            this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2);
John Koleszar's avatar
John Koleszar committed
844 845 846 847

            if (this_rd < best_intra_rd)
            {
                best_intra_rd = this_rd;
Paul Wilkins's avatar
Paul Wilkins committed
848
                *returnintra = distortion2;
John Koleszar's avatar
John Koleszar committed
849 850 851 852 853 854 855 856 857
            }
            break;

        case NEWMV:
        {
            int thissme;
            int step_param;
            int further_steps;
            int n = 0;
858
            int sadpb = x->sadperbit16;
859
            int_mv mvp_full;
John Koleszar's avatar
John Koleszar committed
860

861 862
            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;
863 864 865 866
            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;
867 868 869 870 871 872

            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;

873
            int speed_adjust = (cpi->Speed > 5) ? ((cpi->Speed >= 8)? 3 : 2) : 1;
John Koleszar's avatar
John Koleszar committed
874

John Koleszar's avatar
John Koleszar committed
875
            /* Further step/diamond searches as necessary */
876 877
            step_param = cpi->sf.first_step + speed_adjust;

878
#if CONFIG_MULTI_RES_ENCODING
879 880 881 882
            /* If lower-res drops this frame, then higher-res 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. */
883
            if (cpi->oxcf.mr_encoder_id && !parent_ref_valid)
884 885
                cpi->sf.improved_mv_pred = 0;

886
            if (parent_ref_valid && parent_ref_frame)
887
            {
John Koleszar's avatar
John Koleszar committed
888 889 890
                /* Use parent MV as predictor. Adjust search range
                 * accordingly.
                 */
891 892 893 894 895 896 897 898 899
                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
John Koleszar's avatar
John Koleszar committed
900
            {
901
                if(cpi->sf.improved_mv_pred)
902
                {
903 904 905 906 907
                    if(!saddone)
                    {
                        vp8_cal_sad(cpi,xd,x, recon_yoffset ,&near_sadidx[0] );
                        saddone = 1;
                    }
908

909 910 911 912
                    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]);
913

914
                    sr += speed_adjust;
John Koleszar's avatar
John Koleszar committed
915
                    /* adjust search range according to sr from mv prediction */
916 917
                    if(sr > step_param)
                        step_param = sr;
918

919 920 921 922 923 924 925 926 927
                    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;
                }
            }
928

929
#if CONFIG_MULTI_RES_ENCODING
930
            if (parent_ref_valid && parent_ref_frame && dissim <= 2 &&
931 932
                MAX(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)
933
            {
934 935
                d->bmi.mv.as_int = mvp_full.as_int;
                mode_mv[NEWMV].as_int = mvp_full.as_int;
John Koleszar's avatar
John Koleszar committed
936

937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
                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;
955

956 957
                further_steps = (cpi->Speed >= 8)?
                           0: (cpi->sf.max_step_search_steps - 1 - step_param);
958

959 960
                if (cpi->sf.search_method == HEX)
                {
Yunqing Wang's avatar
Yunqing Wang committed
961 962 963 964 965 966
#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.
                 */
967 968 969
                 /* Set step_param to 0 to ensure large-range motion search
                    when encoder drops this frame at lower-resolution.
                  */
970
                if (!parent_ref_valid)
971
                    step_param = 0;
Yunqing Wang's avatar
Yunqing Wang committed
972
#endif
973 974 975 976 977 978 979 980 981 982 983 984 985
                    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;
John Koleszar's avatar
John Koleszar committed
986

John Koleszar's avatar
John Koleszar committed
987
                    /* Further step/diamond searches as necessary */
988 989
                    n = num00;
                    num00 = 0;
John Koleszar's avatar
John Koleszar committed
990

991
                    while (n < further_steps)
John Koleszar's avatar
John Koleszar committed
992
                    {
993 994 995 996
                        n++;

                        if (num00)
                            num00--;
John Koleszar's avatar
John Koleszar committed
997 998
                        else
                        {
999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
                            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;
                            }
John Koleszar's avatar
John Koleszar committed
1015 1016 1017 1018
                        }
                    }
                }

1019 1020 1021 1022
                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;
John Koleszar's avatar
John Koleszar committed
1023

1024 1025 1026
                if (bestsme < INT_MAX)
                    cpi->find_fractional_mv_step(x, b, d, &d->bmi.mv,
                                             &best_ref_mv, x->errorperbit,
1027 1028 1029
                                             &cpi->fn_ptr[BLOCK_16X16],
                                             cpi->mb.mvcost,
                                             &distortion2,&sse);
1030
            }
John Koleszar's avatar
John Koleszar committed
1031

Scott LaVarnway's avatar
Scott LaVarnway committed
1032
            mode_mv[NEWMV].as_int = d->bmi.mv.as_int;
John Koleszar's avatar
John Koleszar committed
1033

John Koleszar's avatar
John Koleszar committed
1034
            /* mv cost; */
1035 1036
            rate2 += vp8_mv_bit_cost(&mode_mv[NEWMV], &best_ref_mv,
                                     cpi->mb.mvcost, 128);
1037
        }
John Koleszar's avatar
John Koleszar committed
1038 1039 1040 1041

        case NEARESTMV:
        case NEARMV:

Scott LaVarnway's avatar
Scott LaVarnway committed
1042
            if (mode_mv[this_mode].as_int == 0)
John Koleszar's avatar
John Koleszar committed
1043 1044 1045 1046
                continue;

        case ZEROMV:

1047 1048 1049 1050 1051 1052 1053 1054 1055
            /* 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))
John Koleszar's avatar
John Koleszar committed
1056 1057 1058
                continue;

            rate2 += vp8_cost_mv_ref(this_mode, mdcounts);
1059 1060
            x->e_mbd.mode_info_context->mbmi.mv.as_int =
                                                    mode_mv[this_mode].as_int;
1061 1062
            this_rd = evaluate_inter_mode(&sse, rate2, &distortion2, cpi, x,
                                          rd_adjustment);
John Koleszar's avatar
John Koleszar committed
1063 1064 1065 1066 1067 1068

            break;
        default:
            break;
        }

1069 1070 1071
#if CONFIG_TEMPORAL_DENOISING
        if (cpi->oxcf.noise_sensitivity)
        {
1072

John Koleszar's avatar
John Koleszar committed
1073
            /* Store for later use by denoiser. */
1074 1075 1076 1077 1078 1079 1080
            if (this_mode == ZEROMV && sse < zero_mv_sse )
            {
                zero_mv_sse = sse;
                x->best_zeromv_reference_frame =
                        x->e_mbd.mode_info_context->mbmi.ref_frame;
            }

John Koleszar's avatar
John Koleszar committed
1081
            /* Store the best NEWMV in x for later use in the denoiser. */
1082 1083 1084