vp9_decodemv.c 27.8 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 "vp9/common/vp9_common.h"
#include "vp9/common/vp9_entropy.h"
13
#include "vp9/common/vp9_entropymode.h"
14
#include "vp9/common/vp9_entropymv.h"
15
#include "vp9/common/vp9_findnearmv.h"
16
#include "vp9/common/vp9_mvref_common.h"
17
#include "vp9/common/vp9_pred_common.h"
18 19 20
#include "vp9/common/vp9_reconinter.h"
#include "vp9/common/vp9_seg_common.h"

21
#include "vp9/decoder/vp9_decodemv.h"
22
#include "vp9/decoder/vp9_decodframe.h"
23
#include "vp9/decoder/vp9_onyxd_int.h"
24
#include "vp9/decoder/vp9_dsubexp.h"
25 26
#include "vp9/decoder/vp9_treereader.h"

27

John Koleszar's avatar
John Koleszar committed
28 29 30
#if CONFIG_DEBUG
#include <assert.h>
#endif
31

John Koleszar's avatar
John Koleszar committed
32
// #define DEBUG_DEC_MV
33 34 35
#ifdef DEBUG_DEC_MV
int dec_mvcount = 0;
#endif
Dmitry Kovalev's avatar
Dmitry Kovalev committed
36

37 38 39 40
// #define DEC_DEBUG
#ifdef DEC_DEBUG
extern int dec_debug;
#endif
41

42
static MB_PREDICTION_MODE read_intra_mode(vp9_reader *r, const vp9_prob *p) {
43
  return treed_read(r, vp9_intra_mode_tree, p);
44 45
}

46
static int read_segment_id(vp9_reader *r, MACROBLOCKD *xd) {
47
  return treed_read(r, vp9_segment_tree, xd->mb_segment_tree_probs);
Scott LaVarnway's avatar
Scott LaVarnway committed
48
}
49

50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
static TX_SIZE select_txfm_size(VP9_COMMON *cm, MACROBLOCKD *xd,
                                vp9_reader *r, BLOCK_SIZE_TYPE bsize) {
  const int context = vp9_get_pred_context(cm, xd, PRED_TX_SIZE);
  const vp9_prob *tx_probs = vp9_get_pred_probs(cm, xd, PRED_TX_SIZE);
  TX_SIZE txfm_size = vp9_read(r, tx_probs[0]);
  if (txfm_size != TX_4X4 && bsize >= BLOCK_SIZE_MB16X16) {
    txfm_size += vp9_read(r, tx_probs[1]);
    if (txfm_size != TX_8X8 && bsize >= BLOCK_SIZE_SB32X32)
      txfm_size += vp9_read(r, tx_probs[2]);
  }

  if (bsize >= BLOCK_SIZE_SB32X32)
    cm->fc.tx_count_32x32p[context][txfm_size]++;
  else if (bsize >= BLOCK_SIZE_MB16X16)
    cm->fc.tx_count_16x16p[context][txfm_size]++;
  else
    cm->fc.tx_count_8x8p[context][txfm_size]++;

  return txfm_size;
}

static TX_SIZE get_txfm_size(VP9D_COMP *pbi, TXFM_MODE txfm_mode,
                             BLOCK_SIZE_TYPE bsize, int select_cond,
                             vp9_reader *r) {
  VP9_COMMON *const cm = &pbi->common;
  MACROBLOCKD *const xd = &pbi->mb;

  if (txfm_mode == TX_MODE_SELECT && bsize >= BLOCK_SIZE_SB8X8 && select_cond)
    return select_txfm_size(cm, xd, r, bsize);
  else if (txfm_mode >= ALLOW_32X32 && bsize >= BLOCK_SIZE_SB32X32)
    return TX_32X32;
  else if (txfm_mode >= ALLOW_16X16 && bsize >= BLOCK_SIZE_MB16X16)
    return TX_16X16;
  else if (txfm_mode >= ALLOW_8X8 && bsize >= BLOCK_SIZE_SB8X8)
    return TX_8X8;
  else
    return TX_4X4;
}

89
static void set_segment_id(VP9_COMMON *cm, BLOCK_SIZE_TYPE bsize,
90
                           int mi_row, int mi_col, int segment_id) {
91 92 93
  const int mi_offset = mi_row * cm->mi_cols + mi_col;
  const int bw = 1 << mi_width_log2(bsize);
  const int bh = 1 << mi_height_log2(bsize);
94
  const int xmis = MIN(cm->mi_cols - mi_col, bw);
95
  const int ymis = MIN(cm->mi_rows - mi_row, bh);
96 97
  int x, y;

98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
  assert(segment_id >= 0 && segment_id < MAX_MB_SEGMENTS);

  for (y = 0; y < ymis; y++)
    for (x = 0; x < xmis; x++)
      cm->last_frame_seg_map[mi_offset + y * cm->mi_cols + x] = segment_id;
}

static int read_intra_segment_id(VP9D_COMP *pbi, int mi_row, int mi_col,
                                 vp9_reader *r) {
  VP9_COMMON *const cm = &pbi->common;
  MACROBLOCKD *const xd = &pbi->mb;
  const BLOCK_SIZE_TYPE bsize = xd->mode_info_context->mbmi.sb_type;

  if (xd->segmentation_enabled && xd->update_mb_segmentation_map) {
    const int segment_id = read_segment_id(r, xd);
    set_segment_id(cm, bsize, mi_row, mi_col, segment_id);
    return segment_id;
  } else {
    return 0;
117 118 119
  }
}

120
static void kfread_modes(VP9D_COMP *pbi, MODE_INFO *m,
121
                         int mi_row, int mi_col,
122
                         vp9_reader *r) {
123
  VP9_COMMON *const cm = &pbi->common;
124
  MACROBLOCKD *const xd = &pbi->mb;
125
  const int mis = cm->mode_info_stride;
126

127
  m->mbmi.segment_id = read_intra_segment_id(pbi, mi_row, mi_col, r);
128
  m->mbmi.mb_skip_coeff = vp9_segfeature_active(xd, m->mbmi.segment_id,
129
                                                SEG_LVL_SKIP);
Deb Mukherjee's avatar
Deb Mukherjee committed
130
  if (!m->mbmi.mb_skip_coeff) {
131
    m->mbmi.mb_skip_coeff = vp9_read(r, vp9_get_pred_prob(cm, xd, PRED_MBSKIP));
Deb Mukherjee's avatar
Deb Mukherjee committed
132 133 134
    cm->fc.mbskip_count[vp9_get_pred_context(cm, xd, PRED_MBSKIP)]
                       [m->mbmi.mb_skip_coeff]++;
  }
John Koleszar's avatar
John Koleszar committed
135

136 137
  m->mbmi.txfm_size = get_txfm_size(pbi, cm->txfm_mode, m->mbmi.sb_type,
                                    1, r);
138

139
  // luma mode
Ronald S. Bultje's avatar
Ronald S. Bultje committed
140
  m->mbmi.ref_frame[0] = INTRA_FRAME;
141 142 143 144
  if (m->mbmi.sb_type >= BLOCK_SIZE_SB8X8) {
    const MB_PREDICTION_MODE A = above_block_mode(m, 0, mis);
    const MB_PREDICTION_MODE L = xd->left_available ?
                                  left_block_mode(m, 0) : DC_PRED;
145
    m->mbmi.mode = read_intra_mode(r, cm->kf_y_mode_prob[A][L]);
146
  } else {
147 148 149
    int idx, idy;
    int bw = 1 << b_width_log2(m->mbmi.sb_type);
    int bh = 1 << b_height_log2(m->mbmi.sb_type);
150 151 152 153 154 155 156 157 158

    for (idy = 0; idy < 2; idy += bh) {
      for (idx = 0; idx < 2; idx += bw) {
        int ib = idy * 2 + idx;
        int k;
        const MB_PREDICTION_MODE A = above_block_mode(m, ib, mis);
        const MB_PREDICTION_MODE L = (xd->left_available || idx) ?
                                      left_block_mode(m, ib) : DC_PRED;
        m->bmi[ib].as_mode.first =
159
            read_intra_mode(r, cm->kf_y_mode_prob[A][L]);
160 161 162 163 164 165
        for (k = 1; k < bh; ++k)
          m->bmi[ib + k * 2].as_mode.first = m->bmi[ib].as_mode.first;
        for (k = 1; k < bw; ++k)
          m->bmi[ib + k].as_mode.first = m->bmi[ib].as_mode.first;
      }
    }
Ronald S. Bultje's avatar
Ronald S. Bultje committed
166
    m->mbmi.mode = m->bmi[3].as_mode.first;
John Koleszar's avatar
John Koleszar committed
167
  }
Dmitry Kovalev's avatar
Dmitry Kovalev committed
168

169
  m->mbmi.uv_mode = read_intra_mode(r, cm->kf_uv_mode_prob[m->mbmi.mode]);
Scott LaVarnway's avatar
Scott LaVarnway committed
170
}
John Koleszar's avatar
John Koleszar committed
171

172 173
static int read_mv_component(vp9_reader *r,
                             const nmv_component *mvcomp, int usehp) {
174

175
  int mag, d, fr, hp;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
176 177 178
  const int sign = vp9_read(r, mvcomp->sign);
  const int mv_class = treed_read(r, vp9_mv_class_tree, mvcomp->classes);

179
  // Integer part
Dmitry Kovalev's avatar
Dmitry Kovalev committed
180
  if (mv_class == MV_CLASS_0) {
181
    d = treed_read(r, vp9_mv_class0_tree, mvcomp->class0);
182
  } else {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
183
    int i;
184
    const int n = mv_class + CLASS0_BITS - 1;  // number of bits
Dmitry Kovalev's avatar
Dmitry Kovalev committed
185

186
    d = 0;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
187 188
    for (i = 0; i < n; ++i)
      d |= vp9_read(r, mvcomp->bits[i]) << i;
189 190
  }

191 192 193
  // Fractional part
  fr = treed_read(r, vp9_mv_fp_tree,
                  mv_class == MV_CLASS_0 ? mvcomp->class0_fp[d] : mvcomp->fp);
194 195


196 197 198 199
  // High precision part (if hp is not used, the default value of the hp is 1)
  hp = usehp ? vp9_read(r,
                        mv_class == MV_CLASS_0 ? mvcomp->class0_hp : mvcomp->hp)
             : 1;
200

201 202 203
  // result
  mag = vp9_get_mv_mag(mv_class, (d << 3) | (fr << 1) | hp) + 1;
  return sign ? -mag : mag;
204 205
}

Dmitry Kovalev's avatar
Dmitry Kovalev committed
206
static void update_mv(vp9_reader *r, vp9_prob *const p,
207
                       const vp9_prob upd_p) {
208
  if (vp9_read(r, upd_p)) {
209
#ifdef LOW_PRECISION_MV_UPDATE
210
    *p = (vp9_read_literal(r, 7) << 1) | 1;
211
#else
212
    *p = (vp9_read_literal(r, 8));
213 214 215 216
#endif
  }
}

Dmitry Kovalev's avatar
Dmitry Kovalev committed
217
static void read_mv_probs(vp9_reader *r, nmv_context *mvc, int usehp) {
218
  int i, j, k;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
219

220
#ifdef MV_GROUP_UPDATE
221
  if (!vp9_read_bit(r))
Dmitry Kovalev's avatar
Dmitry Kovalev committed
222
    return;
223
#endif
Dmitry Kovalev's avatar
Dmitry Kovalev committed
224
  for (j = 0; j < MV_JOINTS - 1; ++j)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
225
    update_mv(r, &mvc->joints[j], VP9_NMV_UPDATE_PROB);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
226

227
  for (i = 0; i < 2; ++i) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
228
    update_mv(r, &mvc->comps[i].sign, VP9_NMV_UPDATE_PROB);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
229
    for (j = 0; j < MV_CLASSES - 1; ++j)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
230
      update_mv(r, &mvc->comps[i].classes[j], VP9_NMV_UPDATE_PROB);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
231 232

    for (j = 0; j < CLASS0_SIZE - 1; ++j)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
233
      update_mv(r, &mvc->comps[i].class0[j], VP9_NMV_UPDATE_PROB);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
234 235

    for (j = 0; j < MV_OFFSET_BITS; ++j)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
236
      update_mv(r, &mvc->comps[i].bits[j], VP9_NMV_UPDATE_PROB);
237 238 239
  }

  for (i = 0; i < 2; ++i) {
240
    for (j = 0; j < CLASS0_SIZE; ++j)
241
      for (k = 0; k < 3; ++k)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
242
        update_mv(r, &mvc->comps[i].class0_fp[j][k], VP9_NMV_UPDATE_PROB);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
243 244

    for (j = 0; j < 3; ++j)
Dmitry Kovalev's avatar
Dmitry Kovalev committed
245
      update_mv(r, &mvc->comps[i].fp[j], VP9_NMV_UPDATE_PROB);
246 247 248 249
  }

  if (usehp) {
    for (i = 0; i < 2; ++i) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
250 251
      update_mv(r, &mvc->comps[i].class0_hp, VP9_NMV_UPDATE_PROB);
      update_mv(r, &mvc->comps[i].hp, VP9_NMV_UPDATE_PROB);
252 253 254 255
    }
  }
}

256
// Read the referncence frame
Ronald S. Bultje's avatar
Ronald S. Bultje committed
257 258
static void read_ref_frame(VP9D_COMP *pbi, vp9_reader *r,
                           int segment_id, MV_REFERENCE_FRAME ref_frame[2]) {
259
  VP9_COMMON *const cm = &pbi->common;
John Koleszar's avatar
John Koleszar committed
260
  MACROBLOCKD *const xd = &pbi->mb;
261 262
  const int seg_ref_active = vp9_segfeature_active(xd, segment_id,
                                                   SEG_LVL_REF_FRAME);
263 264 265

  // Segment reference frame features not available.
  if (!seg_ref_active) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
266 267
    int is_comp;
    int comp_ctx = vp9_get_pred_context(cm, xd, PRED_COMP_INTER_INTER);
John Koleszar's avatar
John Koleszar committed
268

Ronald S. Bultje's avatar
Ronald S. Bultje committed
269 270 271 272 273 274
    if (cm->comp_pred_mode == HYBRID_PREDICTION) {
      is_comp = vp9_read(r, cm->fc.comp_inter_prob[comp_ctx]);
      cm->fc.comp_inter_count[comp_ctx][is_comp]++;
    } else {
      is_comp = cm->comp_pred_mode == COMP_PREDICTION_ONLY;
    }
John Koleszar's avatar
John Koleszar committed
275

Ronald S. Bultje's avatar
Ronald S. Bultje committed
276 277 278 279
    // FIXME(rbultje) I'm pretty sure this breaks segmentation ref frame coding
    if (is_comp) {
      int b, fix_ref_idx = cm->ref_frame_sign_bias[cm->comp_fixed_ref];
      int ref_ctx = vp9_get_pred_context(cm, xd, PRED_COMP_REF_P);
John Koleszar's avatar
John Koleszar committed
280

Ronald S. Bultje's avatar
Ronald S. Bultje committed
281 282 283 284
      ref_frame[fix_ref_idx]  = cm->comp_fixed_ref;
      b = vp9_read(r, cm->fc.comp_ref_prob[ref_ctx]);
      cm->fc.comp_ref_count[ref_ctx][b]++;
      ref_frame[!fix_ref_idx] = cm->comp_var_ref[b];
Dmitry Kovalev's avatar
Dmitry Kovalev committed
285
    } else {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
286 287 288 289 290 291 292 293 294 295 296
      int ref1_ctx = vp9_get_pred_context(cm, xd, PRED_SINGLE_REF_P1);
      ref_frame[1] = NONE;
      if (vp9_read(r, cm->fc.single_ref_prob[ref1_ctx][0])) {
        int ref2_ctx = vp9_get_pred_context(cm, xd, PRED_SINGLE_REF_P2);
        int b2 = vp9_read(r, cm->fc.single_ref_prob[ref2_ctx][1]);
        ref_frame[0] = b2 ? ALTREF_FRAME : GOLDEN_FRAME;
        cm->fc.single_ref_count[ref1_ctx][0][1]++;
        cm->fc.single_ref_count[ref2_ctx][1][b2]++;
      } else {
        ref_frame[0] = LAST_FRAME;
        cm->fc.single_ref_count[ref1_ctx][0][0]++;
John Koleszar's avatar
John Koleszar committed
297
      }
298
    }
Dmitry Kovalev's avatar
Dmitry Kovalev committed
299
  } else {
300
    ref_frame[0] = vp9_get_segdata(xd, segment_id, SEG_LVL_REF_FRAME);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
301
    ref_frame[1] = NONE;
John Koleszar's avatar
John Koleszar committed
302
  }
303
}
John Koleszar's avatar
John Koleszar committed
304

305 306
static MB_PREDICTION_MODE read_sb_mv_ref(vp9_reader *r, const vp9_prob *p) {
  return (MB_PREDICTION_MODE) treed_read(r, vp9_sb_mv_ref_tree, p);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
307
}
John Koleszar's avatar
John Koleszar committed
308

Scott LaVarnway's avatar
Scott LaVarnway committed
309
#ifdef VPX_MODE_COUNT
310
unsigned int vp9_mv_cont_count[5][4] = {
John Koleszar's avatar
John Koleszar committed
311 312 313 314 315
  { 0, 0, 0, 0 },
  { 0, 0, 0, 0 },
  { 0, 0, 0, 0 },
  { 0, 0, 0, 0 },
  { 0, 0, 0, 0 }
John Koleszar's avatar
John Koleszar committed
316
};
Scott LaVarnway's avatar
Scott LaVarnway committed
317
#endif
John Koleszar's avatar
John Koleszar committed
318

319
static void read_switchable_interp_probs(FRAME_CONTEXT *fc, vp9_reader *r) {
320
  int i, j;
321 322 323
  for (j = 0; j < VP9_SWITCHABLE_FILTERS + 1; ++j)
    for (i = 0; i < VP9_SWITCHABLE_FILTERS - 1; ++i)
      if (vp9_read(r, VP9_MODE_UPDATE_PROB))
324
        vp9_diff_update_prob(r, &fc->switchable_interp_prob[j][i]);
325 326
}

327
static void read_inter_mode_probs(FRAME_CONTEXT *fc, vp9_reader *r) {
328 329
  int i, j;
  for (i = 0; i < INTER_MODE_CONTEXTS; ++i)
330 331
    for (j = 0; j < VP9_INTER_MODES - 1; ++j)
      if (vp9_read(r, VP9_MODE_UPDATE_PROB))
332
        vp9_diff_update_prob(r, &fc->inter_mode_probs[i][j]);
333
}
John Koleszar's avatar
John Koleszar committed
334

335 336 337 338 339 340 341 342
static INLINE COMPPREDMODE_TYPE read_comp_pred_mode(vp9_reader *r) {
  COMPPREDMODE_TYPE mode = vp9_read_bit(r);
  if (mode)
     mode += vp9_read_bit(r);
  return mode;
}

static void mb_mode_mv_init(VP9D_COMP *pbi, vp9_reader *r) {
343
  VP9_COMMON *const cm = &pbi->common;
John Koleszar's avatar
John Koleszar committed
344

345
  if (cm->frame_type != KEY_FRAME && !cm->intra_only) {
346 347
    nmv_context *const nmvc = &pbi->common.fc.nmvc;
    MACROBLOCKD *const xd = &pbi->mb;
348
    int i, j;
349

350
    read_inter_mode_probs(&cm->fc, r);
351

352
    if (cm->mcomp_filter_type == SWITCHABLE)
353
      read_switchable_interp_probs(&cm->fc, r);
354

355
    for (i = 0; i < INTRA_INTER_CONTEXTS; i++)
356
      if (vp9_read(r, VP9_MODE_UPDATE_PROB))
357
        vp9_diff_update_prob(r, &cm->fc.intra_inter_prob[i]);
John Koleszar's avatar
John Koleszar committed
358

Ronald S. Bultje's avatar
Ronald S. Bultje committed
359 360 361 362
    if (cm->allow_comp_inter_inter) {
      cm->comp_pred_mode = read_comp_pred_mode(r);
      if (cm->comp_pred_mode == HYBRID_PREDICTION)
        for (i = 0; i < COMP_INTER_CONTEXTS; i++)
363
          if (vp9_read(r, VP9_MODE_UPDATE_PROB))
364
            vp9_diff_update_prob(r, &cm->fc.comp_inter_prob[i]);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
365 366 367 368 369 370
    } else {
      cm->comp_pred_mode = SINGLE_PREDICTION_ONLY;
    }

    if (cm->comp_pred_mode != COMP_PREDICTION_ONLY)
      for (i = 0; i < REF_CONTEXTS; i++) {
371
        if (vp9_read(r, VP9_MODE_UPDATE_PROB))
372 373
          vp9_diff_update_prob(r, &cm->fc.single_ref_prob[i][0]);

374
        if (vp9_read(r, VP9_MODE_UPDATE_PROB))
375
          vp9_diff_update_prob(r, &cm->fc.single_ref_prob[i][1]);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
376
      }
John Koleszar's avatar
John Koleszar committed
377

Ronald S. Bultje's avatar
Ronald S. Bultje committed
378 379
    if (cm->comp_pred_mode != SINGLE_PREDICTION_ONLY)
      for (i = 0; i < REF_CONTEXTS; i++)
380
        if (vp9_read(r, VP9_MODE_UPDATE_PROB))
381
          vp9_diff_update_prob(r, &cm->fc.comp_ref_prob[i]);
John Koleszar's avatar
John Koleszar committed
382

383
    // VP9_INTRA_MODES
384 385 386 387 388 389 390 391 392
    for (j = 0; j < BLOCK_SIZE_GROUPS; j++)
      for (i = 0; i < VP9_INTRA_MODES - 1; ++i)
        if (vp9_read(r, VP9_MODE_UPDATE_PROB))
          vp9_diff_update_prob(r, &cm->fc.y_mode_prob[j][i]);

    for (j = 0; j < NUM_PARTITION_CONTEXTS; ++j)
      for (i = 0; i < PARTITION_TYPES - 1; ++i)
        if (vp9_read(r, VP9_MODE_UPDATE_PROB))
          vp9_diff_update_prob(r, &cm->fc.partition_prob[INTER_FRAME][j][i]);
393

Dmitry Kovalev's avatar
Dmitry Kovalev committed
394
    read_mv_probs(r, nmvc, xd->allow_high_precision_mv);
John Koleszar's avatar
John Koleszar committed
395
  }
Scott LaVarnway's avatar
Scott LaVarnway committed
396
}
John Koleszar's avatar
John Koleszar committed
397

398 399
static int read_inter_segment_id(VP9D_COMP *pbi, int mi_row, int mi_col,
                                 vp9_reader *r) {
400
  VP9_COMMON *const cm = &pbi->common;
401
  MACROBLOCKD *const xd = &pbi->mb;
402 403 404 405
  const BLOCK_SIZE_TYPE bsize = xd->mode_info_context->mbmi.sb_type;
  const int pred_segment_id = vp9_get_segment_id(cm, cm->last_frame_seg_map,
                                                 bsize, mi_row, mi_col);
  int segment_id;
John Koleszar's avatar
John Koleszar committed
406

407 408 409
  if (!xd->segmentation_enabled)
    return 0;  // Default for disabled segmentation

410 411
  if (!xd->update_mb_segmentation_map)
    return pred_segment_id;
412

413 414 415 416 417 418
  if (cm->temporal_update) {
    const vp9_prob pred_prob = vp9_get_pred_prob(cm, xd, PRED_SEG_ID);
    const int pred_flag = vp9_read(r, pred_prob);
    vp9_set_pred_flag(xd, PRED_SEG_ID, pred_flag);
    segment_id = pred_flag ? pred_segment_id
                           : read_segment_id(r, xd);
John Koleszar's avatar
John Koleszar committed
419
  } else {
420
    segment_id = read_segment_id(r, xd);
John Koleszar's avatar
John Koleszar committed
421
  }
422 423
  set_segment_id(cm, bsize, mi_row, mi_col, segment_id);
  return segment_id;
424
}
425

Dmitry Kovalev's avatar
Dmitry Kovalev committed
426 427 428 429 430 431 432 433 434 435 436

static INLINE void assign_and_clamp_mv(int_mv *dst, const int_mv *src,
                                       int mb_to_left_edge,
                                       int mb_to_right_edge,
                                       int mb_to_top_edge,
                                       int mb_to_bottom_edge) {
  dst->as_int = src->as_int;
  clamp_mv(dst, mb_to_left_edge, mb_to_right_edge, mb_to_top_edge,
           mb_to_bottom_edge);
}

437 438 439 440 441 442 443
static INLINE void decode_mv(vp9_reader *r, MV *mv, const MV *ref,
                             const nmv_context *ctx,
                             nmv_context_counts *counts,
                             int usehp) {
  const MV_JOINT_TYPE j = treed_read(r, vp9_mv_joint_tree, ctx->joints);
  MV diff = {0, 0};

444
  usehp = usehp && vp9_use_mv_hp(ref);
445 446 447 448 449 450
  if (mv_joint_vertical(j))
    diff.row = read_mv_component(r, &ctx->comps[0], usehp);

  if (mv_joint_horizontal(j))
    diff.col = read_mv_component(r, &ctx->comps[1], usehp);

Dmitry Kovalev's avatar
Dmitry Kovalev committed
451
  vp9_inc_mv(&diff, ref, counts, usehp);
452 453 454

  mv->row = diff.row + ref->row;
  mv->col = diff.col + ref->col;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
455 456
}

457
static INLINE INTERPOLATIONFILTERTYPE read_switchable_filter_type(
458
    VP9D_COMP *pbi, vp9_reader *r) {
459 460 461 462 463 464
  VP9_COMMON *const cm = &pbi->common;
  MACROBLOCKD *const xd = &pbi->mb;
  const vp9_prob *probs = vp9_get_pred_probs(cm, xd, PRED_SWITCHABLE_INTERP);
  const int index = treed_read(r, vp9_switchable_interp_tree, probs);
  const int ctx = vp9_get_pred_context(cm, xd, PRED_SWITCHABLE_INTERP);
  ++cm->fc.switchable_interp_count[ctx][index];
465 466 467
  return vp9_switchable_interp[index];
}

468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
static void read_intra_block_modes(VP9D_COMP *pbi, MODE_INFO *mi,
                                   MB_MODE_INFO *mbmi, vp9_reader *r) {
  VP9_COMMON *const cm = &pbi->common;
  MACROBLOCKD *const xd = &pbi->mb;
  const BLOCK_SIZE_TYPE bsize = mi->mbmi.sb_type;
  const int bw = 1 << b_width_log2(bsize);
  const int bh = 1 << b_height_log2(bsize);

  if (bsize >= BLOCK_SIZE_SB8X8) {
    const BLOCK_SIZE_TYPE bsize = xd->mode_info_context->mbmi.sb_type;
    const int bwl = b_width_log2(bsize), bhl = b_height_log2(bsize);
    const int bsl = MIN(bwl, bhl);
    mbmi->mode = read_intra_mode(r, cm->fc.y_mode_prob[MIN(3, bsl)]);
    cm->fc.y_mode_counts[MIN(3, bsl)][mbmi->mode]++;
  } else {
     int idx, idy;
     for (idy = 0; idy < 2; idy += bh) {
       for (idx = 0; idx < 2; idx += bw) {
         int ib = idy * 2 + idx, k;
         int m = read_intra_mode(r, cm->fc.y_mode_prob[0]);
         mi->bmi[ib].as_mode.first = m;
         cm->fc.y_mode_counts[0][m]++;
         for (k = 1; k < bh; ++k)
           mi->bmi[ib + k * 2].as_mode.first = m;
         for (k = 1; k < bw; ++k)
           mi->bmi[ib + k].as_mode.first = m;
      }
    }
    mbmi->mode = mi->bmi[3].as_mode.first;
  }

  mbmi->uv_mode = read_intra_mode(r, cm->fc.uv_mode_prob[mbmi->mode]);
  cm->fc.uv_mode_counts[mbmi->mode][mbmi->uv_mode]++;
}

static MV_REFERENCE_FRAME read_reference_frame(VP9D_COMP *pbi, int segment_id,
                                               vp9_reader *r) {
  VP9_COMMON *const cm = &pbi->common;
  MACROBLOCKD *const xd = &pbi->mb;

  MV_REFERENCE_FRAME ref;
  if (!vp9_segfeature_active(xd, segment_id, SEG_LVL_REF_FRAME)) {
    const int ctx = vp9_get_pred_context(cm, xd, PRED_INTRA_INTER);
    ref = (MV_REFERENCE_FRAME)
              vp9_read(r, vp9_get_pred_prob(cm, xd, PRED_INTRA_INTER));
    cm->fc.intra_inter_count[ctx][ref != INTRA_FRAME]++;
  } else {
    ref = (MV_REFERENCE_FRAME)
              vp9_get_segdata(xd, segment_id, SEG_LVL_REF_FRAME) != INTRA_FRAME;
  }
  return ref;
}

521
static void read_mb_modes_mv(VP9D_COMP *pbi, MODE_INFO *mi, MB_MODE_INFO *mbmi,
522
                             int mi_row, int mi_col,
523
                             vp9_reader *r) {
524
  VP9_COMMON *const cm = &pbi->common;
525
  MACROBLOCKD *const xd = &pbi->mb;
526
  nmv_context *const nmvc = &cm->fc.nmvc;
John Koleszar's avatar
John Koleszar committed
527

528 529
  int_mv *const mv0 = &mbmi->mv[0];
  int_mv *const mv1 = &mbmi->mv[1];
530 531 532
  const BLOCK_SIZE_TYPE bsize = mi->mbmi.sb_type;
  const int bw = 1 << b_width_log2(bsize);
  const int bh = 1 << b_height_log2(bsize);
Dmitry Kovalev's avatar
Dmitry Kovalev committed
533

534
  int mb_to_left_edge, mb_to_right_edge, mb_to_top_edge, mb_to_bottom_edge;
535
  int j, idx, idy;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
536

Ronald S. Bultje's avatar
Ronald S. Bultje committed
537
  mbmi->ref_frame[1] = NONE;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
538

539 540 541 542
  // Make sure the MACROBLOCKD mode info pointer is pointed at the
  // correct entry for the current macroblock.
  xd->mode_info_context = mi;

Dmitry Kovalev's avatar
Dmitry Kovalev committed
543 544 545
  // Distance of Mb to the various image edges.
  // These specified to 8th pel as they are always compared to MV values
  // that are in 1/8th pel units
546 547
  set_mi_row_col(cm, xd, mi_row, 1 << mi_height_log2(bsize),
                         mi_col, 1 << mi_width_log2(bsize));
548

549 550
  mb_to_top_edge = xd->mb_to_top_edge - LEFT_TOP_MARGIN;
  mb_to_bottom_edge = xd->mb_to_bottom_edge + RIGHT_BOTTOM_MARGIN;
Dmitry Kovalev's avatar
Dmitry Kovalev committed
551 552
  mb_to_left_edge = xd->mb_to_left_edge - LEFT_TOP_MARGIN;
  mb_to_right_edge = xd->mb_to_right_edge + RIGHT_BOTTOM_MARGIN;
John Koleszar's avatar
John Koleszar committed
553 554

  // Read the macroblock segment id.
555
  mbmi->segment_id = read_inter_segment_id(pbi, mi_row, mi_col, r);
John Koleszar's avatar
John Koleszar committed
556

557 558
  mbmi->mb_skip_coeff = vp9_segfeature_active(xd, mbmi->segment_id,
                                              SEG_LVL_SKIP);
Deb Mukherjee's avatar
Deb Mukherjee committed
559
  if (!mbmi->mb_skip_coeff) {
560
    mbmi->mb_skip_coeff = vp9_read(r, vp9_get_pred_prob(cm, xd, PRED_MBSKIP));
Deb Mukherjee's avatar
Deb Mukherjee committed
561 562 563
    cm->fc.mbskip_count[vp9_get_pred_context(cm, xd, PRED_MBSKIP)]
                       [mbmi->mb_skip_coeff]++;
  }
John Koleszar's avatar
John Koleszar committed
564

565 566 567
  mbmi->ref_frame[0] = read_reference_frame(pbi, mbmi->segment_id, r);
  mbmi->txfm_size = get_txfm_size(pbi, cm->txfm_mode, bsize,
     (mbmi->mb_skip_coeff == 0 || mbmi->ref_frame[0] == INTRA_FRAME), r);
568

John Koleszar's avatar
John Koleszar committed
569
  // If reference frame is an Inter frame
Ronald S. Bultje's avatar
Ronald S. Bultje committed
570
  if (mbmi->ref_frame[0] != INTRA_FRAME) {
John Koleszar's avatar
John Koleszar committed
571 572
    int_mv nearest, nearby, best_mv;
    int_mv nearest_second, nearby_second, best_mv_second;
573
    vp9_prob *mv_ref_p;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
574 575

    read_ref_frame(pbi, r, mbmi->segment_id, mbmi->ref_frame);
576 577

    {
578 579 580 581 582
#ifdef DEC_DEBUG
      if (dec_debug)
        printf("%d %d\n", xd->mode_info_context->mbmi.mv[0].as_mv.row,
               xd->mode_info_context->mbmi.mv[0].as_mv.col);
#endif
Ronald S. Bultje's avatar
Ronald S. Bultje committed
583 584 585
      vp9_find_mv_refs(cm, xd, mi, xd->prev_mode_info_context,
                       mbmi->ref_frame[0], mbmi->ref_mvs[mbmi->ref_frame[0]],
                       cm->ref_frame_sign_bias);
Paul Wilkins's avatar
Paul Wilkins committed
586

587 588
      mv_ref_p = cm->fc.inter_mode_probs[
        mbmi->mb_mode_context[mbmi->ref_frame[0]]];
589

590
      // If the segment level skip mode enabled
Paul Wilkins's avatar
Paul Wilkins committed
591 592
      if (vp9_segfeature_active(xd, mbmi->segment_id, SEG_LVL_SKIP)) {
        mbmi->mode = ZEROMV;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
593 594
      } else if (bsize >= BLOCK_SIZE_SB8X8) {
        mbmi->mode = read_sb_mv_ref(r, mv_ref_p);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
595 596
        vp9_accum_mv_refs(cm, mbmi->mode,
                          mbmi->mb_mode_context[mbmi->ref_frame[0]]);
597 598
      }

Ronald S. Bultje's avatar
Ronald S. Bultje committed
599
      if (bsize < BLOCK_SIZE_SB8X8 || mbmi->mode != ZEROMV) {
600
        vp9_find_best_ref_mvs(xd,
Ronald S. Bultje's avatar
Ronald S. Bultje committed
601
                              mbmi->ref_mvs[mbmi->ref_frame[0]],
602 603
                              &nearest, &nearby);

Ronald S. Bultje's avatar
Ronald S. Bultje committed
604
        best_mv.as_int = mbmi->ref_mvs[mbmi->ref_frame[0]][0].as_int;
605
      }
606

607 608 609 610 611 612
#ifdef DEC_DEBUG
      if (dec_debug)
        printf("[D %d %d] %d %d %d %d\n", ref_frame,
               mbmi->mb_mode_context[ref_frame],
               mv_ref_p[0], mv_ref_p[1], mv_ref_p[2], mv_ref_p[3]);
#endif
Paul Wilkins's avatar
Paul Wilkins committed
613
    }
John Koleszar's avatar
John Koleszar committed
614

615 616 617
    mbmi->interp_filter = cm->mcomp_filter_type == SWITCHABLE
                              ? read_switchable_filter_type(pbi, r)
                              : cm->mcomp_filter_type;
618

Ronald S. Bultje's avatar
Ronald S. Bultje committed
619 620 621 622 623 624 625 626 627 628 629 630
    if (mbmi->ref_frame[1] > INTRA_FRAME) {
      vp9_find_mv_refs(cm, xd, mi, xd->prev_mode_info_context,
                       mbmi->ref_frame[1],
                       mbmi->ref_mvs[mbmi->ref_frame[1]],
                       cm->ref_frame_sign_bias);

      if (bsize < BLOCK_SIZE_SB8X8 || mbmi->mode != ZEROMV) {
        vp9_find_best_ref_mvs(xd,
                              mbmi->ref_mvs[mbmi->ref_frame[1]],
                              &nearest_second,
                              &nearby_second);
        best_mv_second.as_int = mbmi->ref_mvs[mbmi->ref_frame[1]][0].as_int;
631
      }
John Koleszar's avatar
John Koleszar committed
632
    }
633

John Koleszar's avatar
John Koleszar committed
634
    mbmi->uv_mode = DC_PRED;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
635 636 637 638 639 640 641 642 643
    if (mbmi->sb_type < BLOCK_SIZE_SB8X8) {
      for (idy = 0; idy < 2; idy += bh) {
        for (idx = 0; idx < 2; idx += bw) {
          int_mv blockmv, secondmv;
          int blockmode;
          int i;
          j = idy * 2 + idx;

          blockmode = read_sb_mv_ref(r, mv_ref_p);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
644 645
          vp9_accum_mv_refs(cm, blockmode,
                            mbmi->mb_mode_context[mbmi->ref_frame[0]]);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
646
          if (blockmode == NEARESTMV || blockmode == NEARMV) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
647
            MV_REFERENCE_FRAME rf2 = mbmi->ref_frame[1];
Ronald S. Bultje's avatar
Ronald S. Bultje committed
648 649 650 651
            vp9_append_sub8x8_mvs_for_idx(cm, xd, &nearest, &nearby, j, 0);
            if (rf2 > 0) {
              vp9_append_sub8x8_mvs_for_idx(cm, xd,  &nearest_second,
                                            &nearby_second, j, 1);
652
            }
Ronald S. Bultje's avatar
Ronald S. Bultje committed
653
          }
654

Ronald S. Bultje's avatar
Ronald S. Bultje committed
655 656 657 658 659
          switch (blockmode) {
            case NEWMV:
              decode_mv(r, &blockmv.as_mv, &best_mv.as_mv, nmvc,
                         &cm->fc.NMVcount, xd->allow_high_precision_mv);

Ronald S. Bultje's avatar
Ronald S. Bultje committed
660
              if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
661 662 663 664 665 666 667 668 669
                decode_mv(r, &secondmv.as_mv, &best_mv_second.as_mv, nmvc,
                          &cm->fc.NMVcount, xd->allow_high_precision_mv);

#ifdef VPX_MODE_COUNT
              vp9_mv_cont_count[mv_contz][3]++;
#endif
              break;
            case NEARESTMV:
              blockmv.as_int = nearest.as_int;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
670
              if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
671 672 673 674 675 676 677
                secondmv.as_int = nearest_second.as_int;
#ifdef VPX_MODE_COUNT
              vp9_mv_cont_count[mv_contz][0]++;
#endif
              break;
            case NEARMV:
              blockmv.as_int = nearby.as_int;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
678
              if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
679 680 681 682 683 684 685
                secondmv.as_int = nearby_second.as_int;
#ifdef VPX_MODE_COUNT
              vp9_mv_cont_count[mv_contz][1]++;
#endif
              break;
            case ZEROMV:
              blockmv.as_int = 0;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
686
              if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
687 688 689 690 691 692 693
                secondmv.as_int = 0;
#ifdef VPX_MODE_COUNT
              vp9_mv_cont_count[mv_contz][2]++;
#endif
              break;
            default:
              break;
John Koleszar's avatar
John Koleszar committed
694
          }
Ronald S. Bultje's avatar
Ronald S. Bultje committed
695
          mi->bmi[j].as_mv[0].as_int = blockmv.as_int;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
696
          if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
697 698 699 700 701 702 703
            mi->bmi[j].as_mv[1].as_int = secondmv.as_int;

          for (i = 1; i < bh; ++i)
            vpx_memcpy(&mi->bmi[j + i * 2], &mi->bmi[j], sizeof(mi->bmi[j]));
          for (i = 1; i < bw; ++i)
            vpx_memcpy(&mi->bmi[j + i], &mi->bmi[j], sizeof(mi->bmi[j]));
          mi->mbmi.mode = blockmode;
704
        }
Ronald S. Bultje's avatar
Ronald S. Bultje committed
705
      }
John Koleszar's avatar
John Koleszar committed
706

Ronald S. Bultje's avatar
Ronald S. Bultje committed
707 708 709 710 711 712 713 714 715 716
      mv0->as_int = mi->bmi[3].as_mv[0].as_int;
      mv1->as_int = mi->bmi[3].as_mv[1].as_int;
    } else {
      switch (mbmi->mode) {
        case NEARMV:
          // Clip "next_nearest" so that it does not extend to far out of image
          assign_and_clamp_mv(mv0, &nearby, mb_to_left_edge,
                                            mb_to_right_edge,
                                            mb_to_top_edge,
                                            mb_to_bottom_edge);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
717
          if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
718 719 720 721 722 723 724 725 726 727 728 729
            assign_and_clamp_mv(mv1, &nearby_second, mb_to_left_edge,
                                                     mb_to_right_edge,
                                                     mb_to_top_edge,
                                                     mb_to_bottom_edge);
          break;

        case NEARESTMV:
          // Clip "next_nearest" so that it does not extend to far out of image
          assign_and_clamp_mv(mv0, &nearest, mb_to_left_edge,
                                             mb_to_right_edge,
                                             mb_to_top_edge,
                                             mb_to_bottom_edge);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
730
          if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
731 732 733 734 735 736 737 738
            assign_and_clamp_mv(mv1, &nearest_second, mb_to_left_edge,
                                                      mb_to_right_edge,
                                                      mb_to_top_edge,
                                                      mb_to_bottom_edge);
          break;

        case ZEROMV:
          mv0->as_int = 0;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
739
          if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
740 741 742 743 744 745
            mv1->as_int = 0;
          break;

        case NEWMV:
          decode_mv(r, &mv0->as_mv, &best_mv.as_mv, nmvc, &cm->fc.NMVcount,
                    xd->allow_high_precision_mv);
746
          if (mbmi->ref_frame[1] > 0)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
747 748 749 750
            decode_mv(r, &mv1->as_mv, &best_mv_second.as_mv, nmvc,
                      &cm->fc.NMVcount, xd->allow_high_precision_mv);
          break;
        default:
John Koleszar's avatar
John Koleszar committed
751
#if CONFIG_DEBUG
Ronald S. Bultje's avatar
Ronald S. Bultje committed
752
          assert(0);
John Koleszar's avatar
John Koleszar committed
753
#endif
Ronald S. Bultje's avatar
Ronald S. Bultje committed
754 755
          break;
      }
Scott LaVarnway's avatar
Scott LaVarnway committed
756
    }
John Koleszar's avatar
John Koleszar committed
757
  } else {
758 759
    mv0->as_int = 0;  // required for left and above block mv
    read_intra_block_modes(pbi, mi, mbmi, r);
John Koleszar's avatar
John Koleszar committed
760
  }
Scott LaVarnway's avatar
Scott LaVarnway committed
761
}
John Koleszar's avatar
John Koleszar committed
762

763
void vp9_decode_mode_mvs_init(VP9D_COMP* const pbi, vp9_reader *r) {
764
  VP9_COMMON *cm = &pbi->common;
765
  int k;
766

767
  // TODO(jkoleszar): does this clear more than MBSKIP_CONTEXTS? Maybe remove.
Deb Mukherjee's avatar
Deb Mukherjee committed
768
  // vpx_memset(cm->fc.mbskip_probs, 0, sizeof(cm->fc.mbskip_probs));
769 770
  for (k = 0; k < MBSKIP_CONTEXTS; ++k)
    if (vp9_read(r, VP9_MODE_UPDATE_PROB))
771
      vp9_diff_update_prob(r, &cm->fc.mbskip_probs[k]);
772

773
  mb_mode_mv_init(pbi, r);
774
}
775

776 777
void vp9_decode_mb_mode_mv(VP9D_COMP* const pbi,
                           MACROBLOCKD* const xd,
778 779
                           int mi_row,
                           int mi_col,
780
                           vp9_reader *r) {
781
  VP9_COMMON *const cm = &pbi->common;
782
  MODE_INFO *mi = xd->mode_info_context;
783
  MB_MODE_INFO *const mbmi = &mi->mbmi;
784

785
  if (cm->frame_type == KEY_FRAME || cm->intra_only) {
786
    kfread_modes(pbi, mi, mi_row, mi_col, r);
787
  } else {
Yunqing Wang's avatar
Yunqing Wang committed
788
    read_mb_modes_mv(pbi, mi, &mi->mbmi, mi_row, mi_col, r);
789
  }
790

791 792 793 794 795
  if (1) {
    const int bw = 1 << mi_width_log2(mbmi->sb_type);
    const int bh = 1 << mi_height_log2(mbmi->sb_type);
    const int y_mis = MIN(bh, cm->mi_rows - mi_row);
    const int x_mis = MIN(bw, cm->mi_cols - mi_col);
796 797 798
    const int mis = cm->mode_info_stride;
    int x, y;

799 800
    for (y = 0; y < y_mis; y++)
      for (x = !y; x < x_mis; x++)
801 802
        mi[y * mis + x] = *mi;
  }
803
}