decodframe.c 45.7 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 "onyxd_int.h"
John Koleszar's avatar
John Koleszar committed
13
14
15
16
#include "vp8/common/header.h"
#include "vp8/common/reconintra.h"
#include "vp8/common/reconintra4x4.h"
#include "vp8/common/reconinter.h"
John Koleszar's avatar
John Koleszar committed
17
18
#include "dequantize.h"
#include "detokenize.h"
John Koleszar's avatar
John Koleszar committed
19
20
21
22
#include "vp8/common/invtrans.h"
#include "vp8/common/alloccommon.h"
#include "vp8/common/entropymode.h"
#include "vp8/common/quant_common.h"
23
24
#include "vpx_scale/vpxscale.h"
#include "vpx_scale/yv12extend.h"
John Koleszar's avatar
John Koleszar committed
25
#include "vp8/common/setupintrarecon.h"
Scott LaVarnway's avatar
Scott LaVarnway committed
26

John Koleszar's avatar
John Koleszar committed
27
#include "decodemv.h"
John Koleszar's avatar
John Koleszar committed
28
#include "vp8/common/extend.h"
Yaowu Xu's avatar
Yaowu Xu committed
29
#include "vp8/common/modecont.h"
John Koleszar's avatar
John Koleszar committed
30
#include "vpx_mem/vpx_mem.h"
John Koleszar's avatar
John Koleszar committed
31
#include "vp8/common/idct.h"
John Koleszar's avatar
John Koleszar committed
32
33
34
#include "dequantize.h"
#include "dboolhuff.h"

35
#include "vp8/common/seg_common.h"
36
#include "vp8/common/entropy.h"
37
#include "vpx_rtcd.h"
38

John Koleszar's avatar
John Koleszar committed
39
40
41
#include <assert.h>
#include <stdio.h>

42

43
44
#define COEFCOUNT_TESTING

John Koleszar's avatar
John Koleszar committed
45
46
47
48
49
50
51
52
53
54
55
56
static int merge_index(int v, int n, int modulus) {
  int max1 = (n - 1 - modulus / 2) / modulus + 1;
  if (v < max1) v = v * modulus + modulus / 2;
  else {
    int w;
    v -= max1;
    w = v;
    v += (v + modulus - modulus / 2) / modulus;
    while (v % modulus == modulus / 2 ||
           w != v - (v + modulus - modulus / 2) / modulus) v++;
  }
  return v;
57
58
}

John Koleszar's avatar
John Koleszar committed
59
60
61
static int inv_remap_prob(int v, int m) {
  const int n = 256;
  const int modulus = MODULUS_PARAM;
62
  int i;
John Koleszar's avatar
John Koleszar committed
63
64
65
66
67
68
69
  v = merge_index(v, n - 1, modulus);
  if ((m << 1) <= n) {
    i = inv_recenter_nonneg(v + 1, m);
  } else {
    i = n - 1 - inv_recenter_nonneg(v + 1, n - 1 - m);
  }
  return i;
70
}
71

John Koleszar's avatar
John Koleszar committed
72
73
74
static vp8_prob read_prob_diff_update(vp8_reader *const bc, int oldp) {
  int delp = vp8_decode_term_subexp(bc, SUBEXP_PARAM, 255);
  return (vp8_prob)inv_remap_prob(delp, oldp);
75
}
76

John Koleszar's avatar
John Koleszar committed
77
78
79
void vp8cx_init_de_quantizer(VP8D_COMP *pbi) {
  int i;
  int Q;
John Koleszar's avatar
John Koleszar committed
80
  VP8_COMMON *const pc = &pbi->common;
John Koleszar's avatar
John Koleszar committed
81
82
83
84
85
86
87
88
89
90
91
92
93

  for (Q = 0; Q < QINDEX_RANGE; Q++) {
    pc->Y1dequant[Q][0] = (short)vp8_dc_quant(Q, pc->y1dc_delta_q);
    pc->Y2dequant[Q][0] = (short)vp8_dc2quant(Q, pc->y2dc_delta_q);
    pc->UVdequant[Q][0] = (short)vp8_dc_uv_quant(Q, pc->uvdc_delta_q);

    /* all the ac values =; */
    for (i = 1; i < 16; i++) {
      int rc = vp8_default_zig_zag1d[i];

      pc->Y1dequant[Q][rc] = (short)vp8_ac_yquant(Q);
      pc->Y2dequant[Q][rc] = (short)vp8_ac2quant(Q, pc->y2ac_delta_q);
      pc->UVdequant[Q][rc] = (short)vp8_ac_uv_quant(Q, pc->uvac_delta_q);
John Koleszar's avatar
John Koleszar committed
94
    }
John Koleszar's avatar
John Koleszar committed
95
  }
John Koleszar's avatar
John Koleszar committed
96
97
}

John Koleszar's avatar
John Koleszar committed
98
99
100
void mb_init_dequantizer(VP8D_COMP *pbi, MACROBLOCKD *xd) {
  int i;
  int QIndex;
John Koleszar's avatar
John Koleszar committed
101
  VP8_COMMON *const pc = &pbi->common;
John Koleszar's avatar
John Koleszar committed
102
103
104
105
106
107
108
109
110
111
112
113
114
  int segment_id = xd->mode_info_context->mbmi.segment_id;

  // Set the Q baseline allowing for any segment level adjustment
  if (segfeature_active(xd, segment_id, SEG_LVL_ALT_Q)) {
    /* Abs Value */
    if (xd->mb_segment_abs_delta == SEGMENT_ABSDATA)
      QIndex = get_segdata(xd, segment_id, SEG_LVL_ALT_Q);

    /* Delta Value */
    else {
      QIndex = pc->base_qindex +
               get_segdata(xd, segment_id, SEG_LVL_ALT_Q);
      QIndex = (QIndex >= 0) ? ((QIndex <= MAXQ) ? QIndex : MAXQ) : 0;    /* Clamp to valid range */
John Koleszar's avatar
John Koleszar committed
115
    }
John Koleszar's avatar
John Koleszar committed
116
117
  } else
    QIndex = pc->base_qindex;
Deb Mukherjee's avatar
Deb Mukherjee committed
118
  xd->q_index = QIndex;
John Koleszar's avatar
John Koleszar committed
119

John Koleszar's avatar
John Koleszar committed
120
121
122
123
  /* Set up the block level dequant pointers */
  for (i = 0; i < 16; i++) {
    xd->block[i].dequant = pc->Y1dequant[QIndex];
  }
John Koleszar's avatar
John Koleszar committed
124

Hui Su's avatar
Hui Su committed
125
#if CONFIG_LOSSLESS
John Koleszar's avatar
John Koleszar committed
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
  if (!QIndex) {
    pbi->common.rtcd.idct.idct1        = vp8_short_inv_walsh4x4_1_x8_c;
    pbi->common.rtcd.idct.idct16       = vp8_short_inv_walsh4x4_x8_c;
    pbi->common.rtcd.idct.idct1_scalar_add  = vp8_dc_only_inv_walsh_add_c;
    pbi->common.rtcd.idct.iwalsh1      = vp8_short_inv_walsh4x4_1_lossless_c;
    pbi->common.rtcd.idct.iwalsh16     = vp8_short_inv_walsh4x4_lossless_c;
    pbi->dequant.idct_add            = vp8_dequant_idct_add_lossless_c;
    pbi->dequant.dc_idct_add         = vp8_dequant_dc_idct_add_lossless_c;
    pbi->dequant.dc_idct_add_y_block = vp8_dequant_dc_idct_add_y_block_lossless_c;
    pbi->dequant.idct_add_y_block    = vp8_dequant_idct_add_y_block_lossless_c;
    pbi->dequant.idct_add_uv_block   = vp8_dequant_idct_add_uv_block_lossless_c;
  } else {
    pbi->common.rtcd.idct.idct1        = vp8_short_idct4x4llm_1_c;
    pbi->common.rtcd.idct.idct16       = vp8_short_idct4x4llm_c;
    pbi->common.rtcd.idct.idct1_scalar_add  = vp8_dc_only_idct_add_c;
    pbi->common.rtcd.idct.iwalsh1      = vp8_short_inv_walsh4x4_1_c;
    pbi->common.rtcd.idct.iwalsh16     = vp8_short_inv_walsh4x4_c;
    pbi->dequant.idct_add            = vp8_dequant_idct_add_c;
    pbi->dequant.dc_idct_add         = vp8_dequant_dc_idct_add_c;
    pbi->dequant.dc_idct_add_y_block = vp8_dequant_dc_idct_add_y_block_c;
    pbi->dequant.idct_add_y_block    = vp8_dequant_idct_add_y_block_c;
    pbi->dequant.idct_add_uv_block   = vp8_dequant_idct_add_uv_block_c;
  }
Hui Su's avatar
Hui Su committed
149
150
#endif

John Koleszar's avatar
John Koleszar committed
151
152
153
  for (i = 16; i < 24; i++) {
    xd->block[i].dequant = pc->UVdequant[QIndex];
  }
John Koleszar's avatar
John Koleszar committed
154

John Koleszar's avatar
John Koleszar committed
155
  xd->block[24].dequant = pc->Y2dequant[QIndex];
John Koleszar's avatar
John Koleszar committed
156
157
158
159
160
161
162
163
164

}

#if CONFIG_RUNTIME_CPU_DETECT
#define RTCD_VTABLE(x) (&(pbi)->common.rtcd.x)
#else
#define RTCD_VTABLE(x) NULL
#endif

165
166
167
/* skip_recon_mb() is Modified: Instead of writing the result to predictor buffer and then copying it
 *  to dst buffer, we can write the result directly to dst buffer. This eliminates unnecessary copy.
 */
John Koleszar's avatar
John Koleszar committed
168
169
static void skip_recon_mb(VP8D_COMP *pbi, MACROBLOCKD *xd) {
  if (xd->mode_info_context->mbmi.ref_frame == INTRA_FRAME) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
170
171
#if CONFIG_SUPERBLOCKS
    if (xd->mode_info_context->mbmi.encoded_as_sb) {
172
173
      vp8_build_intra_predictors_sbuv_s(xd);
      vp8_build_intra_predictors_sby_s(xd);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
174
175
    } else {
#endif
176
177
    vp8_build_intra_predictors_mbuv_s(xd);
    vp8_build_intra_predictors_mby_s(xd);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
178
179
180
#if CONFIG_SUPERBLOCKS
    }
#endif
John Koleszar's avatar
John Koleszar committed
181
  } else {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
182
183
184
185
186
187
188
#if CONFIG_SUPERBLOCKS
    if (xd->mode_info_context->mbmi.encoded_as_sb) {
      vp8_build_inter32x32_predictors_sb(xd, xd->dst.y_buffer,
                                         xd->dst.u_buffer, xd->dst.v_buffer,
                                         xd->dst.y_stride, xd->dst.uv_stride);
    } else {
#endif
189
190
191
    vp8_build_1st_inter16x16_predictors_mb(xd, xd->dst.y_buffer,
                                           xd->dst.u_buffer, xd->dst.v_buffer,
                                           xd->dst.y_stride, xd->dst.uv_stride);
John Koleszar's avatar
John Koleszar committed
192
193
194
195
196

    if (xd->mode_info_context->mbmi.second_ref_frame) {
      vp8_build_2nd_inter16x16_predictors_mb(xd, xd->dst.y_buffer,
                                             xd->dst.u_buffer, xd->dst.v_buffer,
                                             xd->dst.y_stride, xd->dst.uv_stride);
197
    }
Ronald S. Bultje's avatar
Ronald S. Bultje committed
198
199
200
#if CONFIG_SUPERBLOCKS
    }
#endif
John Koleszar's avatar
John Koleszar committed
201
  }
John Koleszar's avatar
John Koleszar committed
202
203
}

204
static void decode_macroblock(VP8D_COMP *pbi, MACROBLOCKD *xd,
205
                              unsigned int mb_col,
John Koleszar's avatar
John Koleszar committed
206
                              BOOL_DECODER* const bc) {
John Koleszar's avatar
John Koleszar committed
207
208
209
  int eobtotal = 0;
  MB_PREDICTION_MODE mode;
  int i;
Deb Mukherjee's avatar
Deb Mukherjee committed
210
211
  int tx_size;
  TX_TYPE tx_type;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
212
  VP8_COMMON *pc = &pbi->common;
213
#if CONFIG_SUPERBLOCKS
Ronald S. Bultje's avatar
Ronald S. Bultje committed
214
215
  int orig_skip_flag = xd->mode_info_context->mbmi.mb_skip_coeff;
#endif
John Koleszar's avatar
John Koleszar committed
216

Jingning Han's avatar
Jingning Han committed
217
218
219
220
  // re-initialize macroblock dequantizer before detokenization
  if (xd->segmentation_enabled)
    mb_init_dequantizer(pbi, xd);

221
222
223
#if CONFIG_SUPERBLOCKS
  if (xd->mode_info_context->mbmi.encoded_as_sb) {
    xd->mode_info_context->mbmi.txfm_size = TX_8X8;
Deb Mukherjee's avatar
Deb Mukherjee committed
224
225
226
  }
#endif

Deb Mukherjee's avatar
Deb Mukherjee committed
227
  tx_size = xd->mode_info_context->mbmi.txfm_size;
Deb Mukherjee's avatar
Deb Mukherjee committed
228
229
  mode = xd->mode_info_context->mbmi.mode;

John Koleszar's avatar
John Koleszar committed
230
231
  if (xd->mode_info_context->mbmi.mb_skip_coeff) {
    vp8_reset_mb_tokens_context(xd);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
232
233
234
235
236
237
238
239
240
#if CONFIG_SUPERBLOCKS
    if (xd->mode_info_context->mbmi.encoded_as_sb) {
      xd->above_context++;
      xd->left_context++;
      vp8_reset_mb_tokens_context(xd);
      xd->above_context--;
      xd->left_context--;
    }
#endif
John Koleszar's avatar
John Koleszar committed
241
  } else if (!vp8dx_bool_error(bc)) {
John Koleszar's avatar
John Koleszar committed
242
243
244
    for (i = 0; i < 25; i++) {
      xd->block[i].eob = 0;
      xd->eobs[i] = 0;
245
    }
Deb Mukherjee's avatar
Deb Mukherjee committed
246
    if (tx_size == TX_16X16) {
John Koleszar's avatar
John Koleszar committed
247
      eobtotal = vp8_decode_mb_tokens_16x16(pbi, xd, bc);
Deb Mukherjee's avatar
Deb Mukherjee committed
248
    } else if (tx_size == TX_8X8) {
John Koleszar's avatar
John Koleszar committed
249
      eobtotal = vp8_decode_mb_tokens_8x8(pbi, xd, bc);
Deb Mukherjee's avatar
Deb Mukherjee committed
250
    } else {
John Koleszar's avatar
John Koleszar committed
251
      eobtotal = vp8_decode_mb_tokens(pbi, xd, bc);
Deb Mukherjee's avatar
Deb Mukherjee committed
252
    }
John Koleszar's avatar
John Koleszar committed
253
  }
254

Deb Mukherjee's avatar
Deb Mukherjee committed
255
  //mode = xd->mode_info_context->mbmi.mode;
256
257
258
  if (pbi->common.frame_type != KEY_FRAME)
    vp8_setup_interp_filters(xd, xd->mode_info_context->mbmi.interp_filter,
                             &pbi->common);
Gaute Strokkenes's avatar
Gaute Strokkenes committed
259

John Koleszar's avatar
John Koleszar committed
260
261
  if (eobtotal == 0 && mode != B_PRED && mode != SPLITMV
      && mode != I8X8_PRED
John Koleszar's avatar
John Koleszar committed
262
      && !vp8dx_bool_error(bc)) {
John Koleszar's avatar
John Koleszar committed
263
264
265
266
    /* Special case:  Force the loopfilter to skip when eobtotal and
     * mb_skip_coeff are zero.
     * */
    xd->mode_info_context->mbmi.mb_skip_coeff = 1;
Scott LaVarnway's avatar
Scott LaVarnway committed
267

Ronald S. Bultje's avatar
Ronald S. Bultje committed
268
269
270
271
272
273
274
#if CONFIG_SUPERBLOCKS
    if (!xd->mode_info_context->mbmi.encoded_as_sb || orig_skip_flag)
#endif
    {
      skip_recon_mb(pbi, xd);
      return;
    }
John Koleszar's avatar
John Koleszar committed
275
  }
Yaowu Xu's avatar
Yaowu Xu committed
276

Jingning Han's avatar
Jingning Han committed
277
278
279
  // moved to be performed before detokenization
//  if (xd->segmentation_enabled)
//    mb_init_dequantizer(pbi, xd);
280

John Koleszar's avatar
John Koleszar committed
281
282
  /* do prediction */
  if (xd->mode_info_context->mbmi.ref_frame == INTRA_FRAME) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
283
284
#if CONFIG_SUPERBLOCKS
    if (xd->mode_info_context->mbmi.encoded_as_sb) {
285
286
      vp8_build_intra_predictors_sby_s(xd);
      vp8_build_intra_predictors_sbuv_s(xd);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
287
288
    } else
#endif
John Koleszar's avatar
John Koleszar committed
289
    if (mode != I8X8_PRED) {
290
      vp8_build_intra_predictors_mbuv(xd);
John Koleszar's avatar
John Koleszar committed
291
      if (mode != B_PRED) {
292
        vp8_build_intra_predictors_mby(xd);
John Koleszar's avatar
John Koleszar committed
293
      }
John Koleszar's avatar
John Koleszar committed
294
    }
John Koleszar's avatar
John Koleszar committed
295
  } else {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
296
297
298
299
300
301
302
#if CONFIG_SUPERBLOCKS
    if (xd->mode_info_context->mbmi.encoded_as_sb) {
      vp8_build_inter32x32_predictors_sb(xd, xd->dst.y_buffer,
                                         xd->dst.u_buffer, xd->dst.v_buffer,
                                         xd->dst.y_stride, xd->dst.uv_stride);
    } else
#endif
John Koleszar's avatar
John Koleszar committed
303
304
305
306
307
308
309
310
311
312
313
314
    vp8_build_inter_predictors_mb(xd);
  }

  /* dequantization and idct */
  if (mode == I8X8_PRED) {
    for (i = 0; i < 4; i++) {
      int ib = vp8_i8x8_block[i];
      const int iblock[4] = {0, 1, 4, 5};
      int j;
      int i8x8mode;
      BLOCKD *b;

315
316
317
318
319
320
321
322
      int idx = (ib & 0x02) ? (ib + 2) : ib;

      short *q  = xd->block[idx].qcoeff;
      short *dq = xd->block[0].dequant;
      unsigned char *pre = xd->block[ib].predictor;
      unsigned char *dst = *(xd->block[ib].base_dst) + xd->block[ib].dst;
      int stride = xd->dst.y_stride;

John Koleszar's avatar
John Koleszar committed
323
324
      b = &xd->block[ib];
      i8x8mode = b->bmi.as_mode.first;
325
      vp8_intra8x8_predict(b, i8x8mode, b->predictor);
John Koleszar's avatar
John Koleszar committed
326

327
      if (xd->mode_info_context->mbmi.txfm_size == TX_8X8) {
Deb Mukherjee's avatar
Deb Mukherjee committed
328
329
330
331
332
333
334
335
        tx_type = get_tx_type(xd, &xd->block[idx]);
        if (tx_type != DCT_DCT) {
          vp8_ht_dequant_idct_add_8x8_c(tx_type,
                                        q, dq, pre, dst, 16, stride);
        } else {
          vp8_dequant_idct_add_8x8_c(q, dq, pre, dst, 16, stride);
        }
        q += 64;
336
337
338
339
340
341
342
      } else {
        for (j = 0; j < 4; j++) {
          b = &xd->block[ib + iblock[j]];
          vp8_dequant_idct_add_c(b->qcoeff, b->dequant, b->predictor,
                                 *(b->base_dst) + b->dst, 16, b->dst_stride);
        }
      }
John Koleszar's avatar
John Koleszar committed
343
344

      b = &xd->block[16 + i];
Deb Mukherjee's avatar
Deb Mukherjee committed
345
      vp8_intra_uv4x4_predict(b, i8x8mode, b->predictor);
346
347
348
349
      DEQUANT_INVOKE(&pbi->dequant, idct_add)(b->qcoeff, b->dequant,
                                              b->predictor,
                                              *(b->base_dst) + b->dst, 8,
                                              b->dst_stride);
John Koleszar's avatar
John Koleszar committed
350
      b = &xd->block[20 + i];
351
      vp8_intra_uv4x4_predict(b, i8x8mode, b->predictor);
352
353
354
355
      DEQUANT_INVOKE(&pbi->dequant, idct_add)(b->qcoeff, b->dequant,
                                              b->predictor,
                                              *(b->base_dst) + b->dst, 8,
                                              b->dst_stride);
Yaowu Xu's avatar
Yaowu Xu committed
356
    }
John Koleszar's avatar
John Koleszar committed
357
358
359
360
  } else if (mode == B_PRED) {
    for (i = 0; i < 16; i++) {
      BLOCKD *b = &xd->block[i];
      int b_mode = xd->mode_info_context->bmi[i].as_mode.first;
361
#if CONFIG_COMP_INTRA_PRED
John Koleszar's avatar
John Koleszar committed
362
      int b_mode2 = xd->mode_info_context->bmi[i].as_mode.second;
363

John Koleszar's avatar
John Koleszar committed
364
      if (b_mode2 == (B_PREDICTION_MODE)(B_DC_PRED - 1)) {
365
#endif
366
        vp8_intra4x4_predict(b, b_mode, b->predictor);
367
#if CONFIG_COMP_INTRA_PRED
John Koleszar's avatar
John Koleszar committed
368
      } else {
369
        vp8_comp_intra4x4_predict(b, b_mode, b_mode2, b->predictor);
John Koleszar's avatar
John Koleszar committed
370
      }
371
#endif
John Koleszar's avatar
John Koleszar committed
372

Deb Mukherjee's avatar
Deb Mukherjee committed
373
374
375
376
377
378
      tx_type = get_tx_type(xd, b);
      if (tx_type != DCT_DCT) {
        vp8_ht_dequant_idct_add_c(tx_type, b->qcoeff,
                                  b->dequant, b->predictor,
                                  *(b->base_dst) + b->dst, 16, b->dst_stride);
      } else {
Jingning Han's avatar
Jingning Han committed
379
380
        vp8_dequant_idct_add_c(b->qcoeff, b->dequant, b->predictor,
                               *(b->base_dst) + b->dst, 16, b->dst_stride);
Deb Mukherjee's avatar
Deb Mukherjee committed
381
      }
John Koleszar's avatar
John Koleszar committed
382
    }
John Koleszar's avatar
John Koleszar committed
383
  } else if (mode == SPLITMV) {
384
385
386
387
388
389
390
391
392
393
    if (tx_size == TX_8X8) {
      vp8_dequant_idct_add_y_block_8x8_c(xd->qcoeff, xd->block[0].dequant,
                                         xd->predictor, xd->dst.y_buffer,
                                         xd->dst.y_stride, xd->eobs, xd);
    } else {
      DEQUANT_INVOKE(&pbi->dequant,
                     idct_add_y_block)(xd->qcoeff, xd->block[0].dequant,
                                       xd->predictor, xd->dst.y_buffer,
                                       xd->dst.y_stride, xd->eobs);
    }
John Koleszar's avatar
John Koleszar committed
394
395
  } else {
    BLOCKD *b = &xd->block[24];
Daniel Kang's avatar
Daniel Kang committed
396

Deb Mukherjee's avatar
Deb Mukherjee committed
397
398
399
400
401
    if (tx_size == TX_16X16) {
      BLOCKD *bd = &xd->block[0];
      tx_type = get_tx_type(xd, bd);
      if (tx_type != DCT_DCT) {
        vp8_ht_dequant_idct_add_16x16_c(tx_type, xd->qcoeff,
402
403
404
405
406
407
408
                                        xd->block[0].dequant, xd->predictor,
                                        xd->dst.y_buffer, 16, xd->dst.y_stride);
      } else {
        vp8_dequant_idct_add_16x16_c(xd->qcoeff, xd->block[0].dequant,
                                     xd->predictor, xd->dst.y_buffer,
                                     16, xd->dst.y_stride);
      }
Deb Mukherjee's avatar
Deb Mukherjee committed
409
    } else if (tx_size == TX_8X8) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
#if CONFIG_SUPERBLOCKS
      void *orig = xd->mode_info_context;
      int n, num = xd->mode_info_context->mbmi.encoded_as_sb ? 4 : 1;
      for (n = 0; n < num; n++) {
        if (n != 0) {
          for (i = 0; i < 25; i++) {
            xd->block[i].eob = 0;
            xd->eobs[i] = 0;
          }
          xd->above_context = pc->above_context + mb_col + (n & 1);
          xd->left_context = pc->left_context + (n >> 1);
          xd->mode_info_context = orig;
          xd->mode_info_context += (n & 1);
          xd->mode_info_context += (n >> 1) * pc->mode_info_stride;
          if (!orig_skip_flag) {
John Koleszar's avatar
John Koleszar committed
425
            eobtotal = vp8_decode_mb_tokens_8x8(pbi, xd, bc);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
426
427
428
429
430
431
432
433
434
435
            if (eobtotal == 0) // skip loopfilter
              xd->mode_info_context->mbmi.mb_skip_coeff = 1;
          } else {
            vp8_reset_mb_tokens_context(xd);
          }
        }

        if (xd->mode_info_context->mbmi.mb_skip_coeff)
          continue; // only happens for SBs, which are already in dest buffer
#endif
John Koleszar's avatar
John Koleszar committed
436
437
438
439
440
441
442
443
444
445
      DEQUANT_INVOKE(&pbi->dequant, block_2x2)(b);
      IDCT_INVOKE(RTCD_VTABLE(idct), ihaar2)(&b->dqcoeff[0], b->diff, 8);
      ((int *)b->qcoeff)[0] = 0;// 2nd order block are set to 0 after inverse transform
      ((int *)b->qcoeff)[1] = 0;
      ((int *)b->qcoeff)[2] = 0;
      ((int *)b->qcoeff)[3] = 0;
      ((int *)b->qcoeff)[4] = 0;
      ((int *)b->qcoeff)[5] = 0;
      ((int *)b->qcoeff)[6] = 0;
      ((int *)b->qcoeff)[7] = 0;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
#if CONFIG_SUPERBLOCKS
      if (xd->mode_info_context->mbmi.encoded_as_sb) {
        vp8_dequant_dc_idct_add_y_block_8x8_inplace_c(xd->qcoeff,
          xd->block[0].dequant,
          xd->dst.y_buffer + (n >> 1) * 16 * xd->dst.y_stride + (n & 1) * 16,
          xd->dst.y_stride, xd->eobs, xd->block[24].diff, xd);
        // do UV inline also
        vp8_dequant_idct_add_uv_block_8x8_inplace_c(xd->qcoeff + 16 * 16,
          xd->block[16].dequant,
          xd->dst.u_buffer + (n >> 1) * 8 * xd->dst.uv_stride + (n & 1) * 8,
          xd->dst.v_buffer + (n >> 1) * 8 * xd->dst.uv_stride + (n & 1) * 8,
          xd->dst.uv_stride, xd->eobs + 16, xd);
      } else
#endif
        DEQUANT_INVOKE(&pbi->dequant, dc_idct_add_y_block_8x8)(xd->qcoeff,
          xd->block[0].dequant, xd->predictor, xd->dst.y_buffer,
          xd->dst.y_stride, xd->eobs, xd->block[24].diff, xd);
#if CONFIG_SUPERBLOCKS
      }
      xd->mode_info_context = orig;
#endif
John Koleszar's avatar
John Koleszar committed
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
    } else {
      DEQUANT_INVOKE(&pbi->dequant, block)(b);
      if (xd->eobs[24] > 1) {
        IDCT_INVOKE(RTCD_VTABLE(idct), iwalsh16)(&b->dqcoeff[0], b->diff);
        ((int *)b->qcoeff)[0] = 0;
        ((int *)b->qcoeff)[1] = 0;
        ((int *)b->qcoeff)[2] = 0;
        ((int *)b->qcoeff)[3] = 0;
        ((int *)b->qcoeff)[4] = 0;
        ((int *)b->qcoeff)[5] = 0;
        ((int *)b->qcoeff)[6] = 0;
        ((int *)b->qcoeff)[7] = 0;
      } else {
        IDCT_INVOKE(RTCD_VTABLE(idct), iwalsh1)(&b->dqcoeff[0], b->diff);
        ((int *)b->qcoeff)[0] = 0;
      }

      DEQUANT_INVOKE(&pbi->dequant, dc_idct_add_y_block)
      (xd->qcoeff, xd->block[0].dequant,
       xd->predictor, xd->dst.y_buffer,
       xd->dst.y_stride, xd->eobs, xd->block[24].diff);
488
    }
John Koleszar's avatar
John Koleszar committed
489
490
  }

Ronald S. Bultje's avatar
Ronald S. Bultje committed
491
#if CONFIG_SUPERBLOCKS
492
  if (!xd->mode_info_context->mbmi.encoded_as_sb) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
493
#endif
Deb Mukherjee's avatar
Deb Mukherjee committed
494
    if ((tx_size == TX_8X8 &&
495
496
497
498
         xd->mode_info_context->mbmi.mode != I8X8_PRED &&
         xd->mode_info_context->mbmi.mode != SPLITMV)
        || tx_size == TX_16X16
       )
499
500
501
502
503
504
505
506
507
      DEQUANT_INVOKE(&pbi->dequant, idct_add_uv_block_8x8) //
          (xd->qcoeff + 16 * 16, xd->block[16].dequant,
           xd->predictor + 16 * 16, xd->dst.u_buffer, xd->dst.v_buffer,
           xd->dst.uv_stride, xd->eobs + 16, xd); //
    else if (xd->mode_info_context->mbmi.mode != I8X8_PRED)
      DEQUANT_INVOKE(&pbi->dequant, idct_add_uv_block)
          (xd->qcoeff + 16 * 16, xd->block[16].dequant,
           xd->predictor + 16 * 16, xd->dst.u_buffer, xd->dst.v_buffer,
           xd->dst.uv_stride, xd->eobs + 16);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
508
509
510
#if CONFIG_SUPERBLOCKS
  }
#endif
Yaowu Xu's avatar
Yaowu Xu committed
511
}
John Koleszar's avatar
John Koleszar committed
512

513

John Koleszar's avatar
John Koleszar committed
514
515
static int get_delta_q(vp8_reader *bc, int prev, int *q_update) {
  int ret_val = 0;
John Koleszar's avatar
John Koleszar committed
516

John Koleszar's avatar
John Koleszar committed
517
518
  if (vp8_read_bit(bc)) {
    ret_val = vp8_read_literal(bc, 4);
John Koleszar's avatar
John Koleszar committed
519

John Koleszar's avatar
John Koleszar committed
520
521
522
    if (vp8_read_bit(bc))
      ret_val = -ret_val;
  }
John Koleszar's avatar
John Koleszar committed
523

John Koleszar's avatar
John Koleszar committed
524
525
526
  /* Trigger a quantizer update if the delta-q value has changed */
  if (ret_val != prev)
    *q_update = 1;
John Koleszar's avatar
John Koleszar committed
527

John Koleszar's avatar
John Koleszar committed
528
  return ret_val;
John Koleszar's avatar
John Koleszar committed
529
530
531
532
533
534
535
}

#ifdef PACKET_TESTING
#include <stdio.h>
FILE *vpxlog = 0;
#endif

Adrian Grange's avatar
Adrian Grange committed
536
/* Decode a row of Superblocks (2x2 region of MBs) */
537
static void
John Koleszar's avatar
John Koleszar committed
538
539
decode_sb_row(VP8D_COMP *pbi, VP8_COMMON *pc, int mbrow, MACROBLOCKD *xd,
              BOOL_DECODER* const bc) {
John Koleszar's avatar
John Koleszar committed
540
541
542
543
544
545
546
547
548
549
550
551
552
  int i;
  int sb_col;
  int mb_row, mb_col;
  int recon_yoffset, recon_uvoffset;
  int ref_fb_idx = pc->lst_fb_idx;
  int dst_fb_idx = pc->new_fb_idx;
  int recon_y_stride = pc->yv12_fb[ref_fb_idx].y_stride;
  int recon_uv_stride = pc->yv12_fb[ref_fb_idx].uv_stride;
  int row_delta[4] = { 0, +1,  0, -1};
  int col_delta[4] = { +1, -1, +1, +1};
  int sb_cols = (pc->mb_cols + 1) >> 1;

  // For a SB there are 2 left contexts, each pertaining to a MB row within
Ronald S. Bultje's avatar
Ronald S. Bultje committed
553
  vpx_memset(pc->left_context, 0, sizeof(pc->left_context));
John Koleszar's avatar
John Koleszar committed
554
555
556
557
558

  mb_row = mbrow;
  mb_col = 0;

  for (sb_col = 0; sb_col < sb_cols; sb_col++) {
Ronald S. Bultje's avatar
Ronald S. Bultje committed
559
560
561
    MODE_INFO *mi = xd->mode_info_context;

#if CONFIG_SUPERBLOCKS
John Koleszar's avatar
John Koleszar committed
562
    mi->mbmi.encoded_as_sb = vp8_read(bc, pc->sb_coded);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
563
564
#endif

John Koleszar's avatar
John Koleszar committed
565
566
567
568
569
570
571
    // Process the 4 MBs within the SB in the order:
    // top-left, top-right, bottom-left, bottom-right
    for (i = 0; i < 4; i++) {
      int dy = row_delta[i];
      int dx = col_delta[i];
      int offset_extended = dy * xd->mode_info_stride + dx;

572
573
      xd->mb_index = i;

Ronald S. Bultje's avatar
Ronald S. Bultje committed
574
      mi = xd->mode_info_context;
John Koleszar's avatar
John Koleszar committed
575
576
577
578
579
      if ((mb_row >= pc->mb_rows) || (mb_col >= pc->mb_cols)) {
        // MB lies outside frame, skip on to next
        mb_row += dy;
        mb_col += dx;
        xd->mode_info_context += offset_extended;
580
        xd->prev_mode_info_context += offset_extended;
John Koleszar's avatar
John Koleszar committed
581
582
        continue;
      }
Yaowu Xu's avatar
Yaowu Xu committed
583

John Koleszar's avatar
John Koleszar committed
584
585
      // Set above context pointer
      xd->above_context = pc->above_context + mb_col;
Ronald S. Bultje's avatar
Ronald S. Bultje committed
586
      xd->left_context = pc->left_context + (i >> 1);
John Koleszar's avatar
John Koleszar committed
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608

      /* Distance of Mb to the various image edges.
       * These are specified to 8th pel as they are always compared to
       * values that are in 1/8th pel units
       */
      xd->mb_to_top_edge = -((mb_row * 16)) << 3;
      xd->mb_to_bottom_edge = ((pc->mb_rows - 1 - mb_row) * 16) << 3;

      xd->mb_to_left_edge = -((mb_col * 16) << 3);
      xd->mb_to_right_edge = ((pc->mb_cols - 1 - mb_col) * 16) << 3;

      xd->up_available = (mb_row != 0);
      xd->left_available = (mb_col != 0);


      recon_yoffset = (mb_row * recon_y_stride * 16) + (mb_col * 16);
      recon_uvoffset = (mb_row * recon_uv_stride * 8) + (mb_col * 8);

      xd->dst.y_buffer = pc->yv12_fb[dst_fb_idx].y_buffer + recon_yoffset;
      xd->dst.u_buffer = pc->yv12_fb[dst_fb_idx].u_buffer + recon_uvoffset;
      xd->dst.v_buffer = pc->yv12_fb[dst_fb_idx].v_buffer + recon_uvoffset;

Ronald S. Bultje's avatar
Ronald S. Bultje committed
609
610
611
612
#if CONFIG_SUPERBLOCKS
      if (i)
        mi->mbmi.encoded_as_sb = 0;
#endif
John Koleszar's avatar
John Koleszar committed
613
      vpx_decode_mb_mode_mv(pbi, xd, mb_row, mb_col, bc);
614
615
616

      update_blockd_bmi(xd);

John Koleszar's avatar
John Koleszar committed
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
      /* Select the appropriate reference frame for this MB */
      if (xd->mode_info_context->mbmi.ref_frame == LAST_FRAME)
        ref_fb_idx = pc->lst_fb_idx;
      else if (xd->mode_info_context->mbmi.ref_frame == GOLDEN_FRAME)
        ref_fb_idx = pc->gld_fb_idx;
      else
        ref_fb_idx = pc->alt_fb_idx;

      xd->pre.y_buffer = pc->yv12_fb[ref_fb_idx].y_buffer + recon_yoffset;
      xd->pre.u_buffer = pc->yv12_fb[ref_fb_idx].u_buffer + recon_uvoffset;
      xd->pre.v_buffer = pc->yv12_fb[ref_fb_idx].v_buffer + recon_uvoffset;

      if (xd->mode_info_context->mbmi.second_ref_frame) {
        int second_ref_fb_idx;

        /* Select the appropriate reference frame for this MB */
        if (xd->mode_info_context->mbmi.second_ref_frame == LAST_FRAME)
          second_ref_fb_idx = pc->lst_fb_idx;
        else if (xd->mode_info_context->mbmi.second_ref_frame ==
                 GOLDEN_FRAME)
          second_ref_fb_idx = pc->gld_fb_idx;
        else
          second_ref_fb_idx = pc->alt_fb_idx;

        xd->second_pre.y_buffer =
          pc->yv12_fb[second_ref_fb_idx].y_buffer + recon_yoffset;
        xd->second_pre.u_buffer =
          pc->yv12_fb[second_ref_fb_idx].u_buffer + recon_uvoffset;
        xd->second_pre.v_buffer =
          pc->yv12_fb[second_ref_fb_idx].v_buffer + recon_uvoffset;
      }

      if (xd->mode_info_context->mbmi.ref_frame != INTRA_FRAME) {
        /* propagate errors from reference frames */
        xd->corrupted |= pc->yv12_fb[ref_fb_idx].corrupted;
      }

Ronald S. Bultje's avatar
Ronald S. Bultje committed
654
655
656
657
658
659
660
#if CONFIG_SUPERBLOCKS
      if (xd->mode_info_context->mbmi.encoded_as_sb) {
        mi[1] = mi[0];
        mi[pc->mode_info_stride] = mi[0];
        mi[pc->mode_info_stride + 1] = mi[0];
      }
#endif
661
      vp8_intra_prediction_down_copy(xd);
662
      decode_macroblock(pbi, xd, mb_col, bc);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
663
664
665
666
667
668
669
#if CONFIG_SUPERBLOCKS
      if (xd->mode_info_context->mbmi.encoded_as_sb) {
        mi[1].mbmi.txfm_size = mi[0].mbmi.txfm_size;
        mi[pc->mode_info_stride].mbmi.txfm_size = mi[0].mbmi.txfm_size;
        mi[pc->mode_info_stride + 1].mbmi.txfm_size = mi[0].mbmi.txfm_size;
      }
#endif
John Koleszar's avatar
John Koleszar committed
670
671

      /* check if the boolean decoder has suffered an error */
John Koleszar's avatar
John Koleszar committed
672
      xd->corrupted |= vp8dx_bool_error(bc);
John Koleszar's avatar
John Koleszar committed
673

Ronald S. Bultje's avatar
Ronald S. Bultje committed
674
675
676
677
678
679
680
681
682
#if CONFIG_SUPERBLOCKS
      if (mi->mbmi.encoded_as_sb) {
        assert(!i);
        mb_col += 2;
        xd->mode_info_context += 2;
        xd->prev_mode_info_context += 2;
        break;
      }
#endif
John Koleszar's avatar
John Koleszar committed
683
684
685

      // skip to next MB
      xd->mode_info_context += offset_extended;
686
      xd->prev_mode_info_context += offset_extended;
John Koleszar's avatar
John Koleszar committed
687
688
      mb_row += dy;
      mb_col += dx;
John Koleszar's avatar
John Koleszar committed
689
    }
John Koleszar's avatar
John Koleszar committed
690
  }
John Koleszar's avatar
John Koleszar committed
691

John Koleszar's avatar
John Koleszar committed
692
693
  /* skip prediction column */
  xd->mode_info_context += 1 - (pc->mb_cols & 0x1) + xd->mode_info_stride;
694
  xd->prev_mode_info_context += 1 - (pc->mb_cols & 0x1) + xd->mode_info_stride;
John Koleszar's avatar
John Koleszar committed
695
696
}

John Koleszar's avatar
John Koleszar committed
697
698
699
700
static unsigned int read_partition_size(const unsigned char *cx_size) {
  const unsigned int size =
    cx_size[0] + (cx_size[1] << 8) + (cx_size[2] << 16);
  return size;
John Koleszar's avatar
John Koleszar committed
701
702
}

703
704
static int read_is_valid(const unsigned char *start,
                         size_t               len,
John Koleszar's avatar
John Koleszar committed
705
706
                         const unsigned char *end) {
  return (start + len > start && start + len <= end);
707
708
709
}


John Koleszar's avatar
John Koleszar committed
710
static void setup_token_decoder(VP8D_COMP *pbi,
John Koleszar's avatar
John Koleszar committed
711
712
                                const unsigned char *cx_data,
                                BOOL_DECODER* const bool_decoder) {
John Koleszar's avatar
John Koleszar committed
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
  VP8_COMMON          *pc = &pbi->common;
  const unsigned char *user_data_end = pbi->Source + pbi->source_sz;
  const unsigned char *partition;

  ptrdiff_t            partition_size;
  ptrdiff_t            bytes_left;

  // Set up pointers to token partition
  partition = cx_data;
  bytes_left = user_data_end - partition;
  partition_size = bytes_left;

  /* Validate the calculated partition length. If the buffer
   * described by the partition can't be fully read, then restrict
   * it to the portion that can be (for EC mode) or throw an error.
   */
  if (!read_is_valid(partition, partition_size, user_data_end)) {
    vpx_internal_error(&pc->error, VPX_CODEC_CORRUPT_FRAME,
                       "Truncated packet or corrupt partition "
                       "%d length", 1);
  }

  if (vp8dx_start_decode(bool_decoder, partition, partition_size))
    vpx_internal_error(&pc->error, VPX_CODEC_MEM_ERROR,
                       "Failed to allocate bool decoder %d", 1);
John Koleszar's avatar
John Koleszar committed
738
739
}

John Koleszar's avatar
John Koleszar committed
740
static void init_frame(VP8D_COMP *pbi) {
John Koleszar's avatar
John Koleszar committed
741
742
  VP8_COMMON *const pc = &pbi->common;
  MACROBLOCKD *const xd  = &pbi->mb;
John Koleszar's avatar
John Koleszar committed
743

John Koleszar's avatar
John Koleszar committed
744
745
  if (pc->frame_type == KEY_FRAME) {
    /* Various keyframe initializations */
746
    vp8_init_mv_probs(pc);
John Koleszar's avatar
John Koleszar committed
747

John Koleszar's avatar
John Koleszar committed
748
    vp8_init_mbmode_probs(pc);
749
    vp8_default_bmode_probs(pc->fc.bmode_prob);
John Koleszar's avatar
John Koleszar committed
750

John Koleszar's avatar
John Koleszar committed
751
752
    vp8_default_coef_probs(pc);
    vp8_kf_default_bmode_probs(pc->kf_bmode_prob);
John Koleszar's avatar
John Koleszar committed
753

John Koleszar's avatar
John Koleszar committed
754
755
756
    // Reset the segment feature data to the default stats:
    // Features disabled, 0, with delta coding (Default state).
    clearall_segfeatures(xd);
Paul Wilkins's avatar
Paul Wilkins committed
757

John Koleszar's avatar
John Koleszar committed
758
    xd->mb_segment_abs_delta = SEGMENT_DELTADATA;
John Koleszar's avatar
John Koleszar committed
759

John Koleszar's avatar
John Koleszar committed
760
761
762
    /* reset the mode ref deltasa for loop filter */
    vpx_memset(xd->ref_lf_deltas, 0, sizeof(xd->ref_lf_deltas));
    vpx_memset(xd->mode_lf_deltas, 0, sizeof(xd->mode_lf_deltas));
John Koleszar's avatar
John Koleszar committed
763

John Koleszar's avatar
John Koleszar committed
764
765
766
767
768
    /* All buffers are implicitly updated on key frames. */
    pc->refresh_golden_frame = 1;
    pc->refresh_alt_ref_frame = 1;
    pc->copy_buffer_to_gf = 0;
    pc->copy_buffer_to_arf = 0;
John Koleszar's avatar
John Koleszar committed
769

John Koleszar's avatar
John Koleszar committed
770
771
772
773
774
    /* Note that Golden and Altref modes cannot be used on a key frame so
     * ref_frame_sign_bias[] is undefined and meaningless
     */
    pc->ref_frame_sign_bias[GOLDEN_FRAME] = 0;
    pc->ref_frame_sign_bias[ALTREF_FRAME] = 0;
775

John Koleszar's avatar
John Koleszar committed
776
777
778
    vp8_init_mode_contexts(&pbi->common);
    vpx_memcpy(&pc->lfc, &pc->fc, sizeof(pc->fc));
    vpx_memcpy(&pc->lfc_a, &pc->fc, sizeof(pc->fc));
Yaowu Xu's avatar
Yaowu Xu committed
779

John Koleszar's avatar
John Koleszar committed
780
781
782
    vpx_memcpy(pbi->common.fc.vp8_mode_contexts,
               pbi->common.fc.mode_context,
               sizeof(pbi->common.fc.mode_context));
783
784
785
786
787
788
789
790
    vpx_memset(pc->prev_mip, 0,
               (pc->mb_cols + 1) * (pc->mb_rows + 1)* sizeof(MODE_INFO));
    vpx_memset(pc->mip, 0,
               (pc->mb_cols + 1) * (pc->mb_rows + 1)* sizeof(MODE_INFO));

    update_mode_info_border(pc, pc->mip);
    update_mode_info_in_image(pc, pc->mi);

John Koleszar's avatar
John Koleszar committed
791
  } else {
792

John Koleszar's avatar
John Koleszar committed
793
794
795
796
797
    if (!pc->use_bilinear_mc_filter)
      pc->mcomp_filter_type = EIGHTTAP;
    else
      pc->mcomp_filter_type = BILINEAR;

798
    /* To enable choice of different interpolation filters */
799
    vp8_setup_interp_filters(xd, pc->mcomp_filter_type, pc);
John Koleszar's avatar
John Koleszar committed
800
  }
John Koleszar's avatar
John Koleszar committed
801

John Koleszar's avatar
John Koleszar committed
802
  xd->mode_info_context = pc->mi;
803
  xd->prev_mode_info_context = pc->prev_mi;
John Koleszar's avatar
John Koleszar committed
804
805
806
807
  xd->frame_type = pc->frame_type;
  xd->mode_info_context->mbmi.mode = DC_PRED;
  xd->mode_info_stride = pc->mode_info_stride;
  xd->corrupted = 0; /* init without corruption */
808

John Koleszar's avatar
John Koleszar committed
809
810
811
  xd->fullpixel_mask = 0xffffffff;
  if (pc->full_pixel)
    xd->fullpixel_mask = 0xfffffff8;
812

John Koleszar's avatar
John Koleszar committed
813
814
}

815
#if 0
John Koleszar's avatar
John Koleszar committed
816
817
818
static void read_coef_probs2(VP8D_COMP *pbi) {
  const vp8_prob grpupd = 192;
  int i, j, k, l;
John Koleszar's avatar
John Koleszar committed
819
820
  vp8_reader *const bc = &pbi->bc;
  VP8_COMMON *const pc = &pbi->common;
John Koleszar's avatar
John Koleszar committed
821
822
823
824
825
826
827
828
829
  for (l = 0; l < ENTROPY_NODES; l++) {
    if (vp8_read(bc, grpupd)) {
      // printf("Decoding %d\n", l);
      for (i = 0; i < BLOCK_TYPES; i++)
        for (j = !i; j < COEF_BANDS; j++)
          for (k = 0; k < PREV_COEF_CONTEXTS; k++) {
            if (k >= 3 && ((i == 0 && j == 1) ||
                           (i > 0 && j == 0)))
              continue;
830
            {
John Koleszar's avatar
John Koleszar committed
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
              vp8_prob *const p = pc->fc.coef_probs [i][j][k] + l;
              int u = vp8_read(bc, COEF_UPDATE_PROB);
              if (u) *p = read_prob_diff_update(bc, *p);
            }
          }
    }
  }
  if (pbi->common.txfm_mode == ALLOW_8X8) {
    for (l = 0; l < ENTROPY_NODES; l++) {
      if (vp8_read(bc, grpupd)) {
        for (i = 0; i < BLOCK_TYPES_8X8; i++)
          for (j = !i; j < COEF_BANDS; j++)
            for (k = 0; k < PREV_COEF_CONTEXTS; k++) {
              if (k >= 3 && ((i == 0 && j == 1) ||
                             (i > 0 && j == 0)))
                continue;
              {
                vp8_prob *const p = pc->fc.coef_probs_8x8 [i][j][k] + l;
849

John Koleszar's avatar
John Koleszar committed
850
851
852
                int u = vp8_read(bc, COEF_UPDATE_PROB_8X8);
                if (u) *p = read_prob_diff_update(bc, *p);
              }
853
            }
John Koleszar's avatar
John Koleszar committed
854
      }
855
    }
John Koleszar's avatar
John Koleszar committed
856
  }
857
}
858
#endif
859

860
861
862
863
static void read_coef_probs_common(
    BOOL_DECODER* const bc,
    vp8_prob coef_probs[BLOCK_TYPES][COEF_BANDS]
                       [PREV_COEF_CONTEXTS][ENTROPY_NODES]) {
John Koleszar's avatar
John Koleszar committed
864
865
  int i, j, k, l;

866
867
868
869
  if (vp8_read_bit(bc)) {
    for (i = 0; i < BLOCK_TYPES; i++) {
      for (j = !i; j < COEF_BANDS; j++) {
        /* NB: This j loop starts from 1 on block type i == 0 */
870
871
872
873
874
        for (k = 0; k < PREV_COEF_CONTEXTS; k++) {
          if (k >= 3 && ((i == 0 && j == 1) ||
                         (i > 0 && j == 0)))
            continue;
          for (l = 0; l < ENTROPY_NODES; l++) {
875
            vp8_prob *const p = coef_probs[i][j][k] + l;
Daniel Kang's avatar
Daniel Kang committed
876

877
            if (vp8_read(bc, COEF_UPDATE_PROB)) {
878
879
880
881
              *p = read_prob_diff_update(bc, *p);
            }
          }
        }
882
883
      }
    }
884
  }
885
}
886

887
888
static void read_coef_probs(VP8D_COMP *pbi, BOOL_DECODER* const bc) {
  VP8_COMMON *const pc = &pbi->common;
Daniel Kang's avatar
Daniel Kang committed
889

890
891
  read_coef_probs_common(bc, pc->fc.coef_probs);
  read_coef_probs_common(bc, pc->fc.hybrid_coef_probs);
892

893
894
895
  if (pbi->common.txfm_mode != ONLY_4X4) {
    read_coef_probs_common(bc, pc->fc.coef_probs_8x8);
    read_coef_probs_common(bc, pc->fc.hybrid_coef_probs_8x8);
Daniel Kang's avatar
Daniel Kang committed
896
  }
897
898
899
  if (pbi->common.txfm_mode > ALLOW_8X8) {
    read_coef_probs_common(bc, pc->fc.coef_probs_16x16);
    read_coef_probs_common(bc, pc->fc.hybrid_coef_probs_16x16);
900
  }
901
902
}

John Koleszar's avatar
John Koleszar committed
903
int vp8_decode_frame(VP8D_COMP *pbi) {
John Koleszar's avatar
John Koleszar committed
904
  BOOL_DECODER header_bc, residual_bc;
John Koleszar's avatar
John Koleszar committed
905
906
  VP8_COMMON *const pc = &pbi->common;
  MACROBLOCKD *const xd  = &pbi->mb;
John Koleszar's avatar
John Koleszar committed
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
  const unsigned char *data = (const unsigned char *)pbi->Source;
  const unsigned char *data_end = data + pbi->source_sz;
  ptrdiff_t first_partition_length_in_bytes = 0;

  int mb_row;
  int i, j;
  int corrupt_tokens = 0;

  /* start with no corruption of current frame */
  xd->corrupted = 0;
  pc->yv12_fb[pc->new_fb_idx].corrupted = 0;

  if (data_end - data < 3) {
    vpx_internal_error(&pc->error, VPX_CODEC_CORRUPT_FRAME,
                       "Truncated packet");
  } else {
    pc->last_frame_type = pc->frame_type;
    pc->frame_type = (FRAME_TYPE)(data[0] & 1);
    pc->version = (data[0] >> 1) & 7;
    pc->show_frame = (data[0] >> 4) & 1;
    first_partition_length_in_bytes =
      (data[0] | (data[1] << 8) | (data[2] << 16)) >> 5;

    if ((data + first_partition_length_in_bytes > data_end
         || data + first_partition_length_in_bytes < data))
      vpx_internal_error(&pc->error, VPX_CODEC_CORRUPT_FRAME,
                         "Truncated packet or corrupt partition 0 length");

    data += 3;

    vp8_setup_version(pc);
Yaowu Xu's avatar
Yaowu Xu committed
938

John Koleszar's avatar
John Koleszar committed
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
    if (pc->frame_type == KEY_FRAME) {
      const int Width = pc->Width;
      const int Height = pc->Height;

      /* vet via sync code */
      /* When error concealment is enabled we should only check the sync
       * code if we have enough bits available
       */
      if (data + 3 < data_end) {
        if (data[0] != 0x9d || data[1] != 0x01 || data[2] != 0x2a)
          vpx_internal_error(&pc->error, VPX_CODEC_UNSUP_BITSTREAM,
                             "Invalid frame sync code");
      }

      /* If error concealment is enabled we should only parse the new size
       * if we have enough data. Otherwise we will end up with the wrong
       * size.
       */
      if (data + 6 < data_end) {
        pc->Width = (data[3] | (data[4] << 8)) & 0x3fff;
        pc->horiz_scale = data[4] >> 6;
        pc->Height = (data[5] | (data[6] << 8)) & 0x3fff;
        pc->vert_scale = data[6] >> 6;
      }
      data += 7;

      if (Width != pc->Width  ||  Height != pc->Height) {
        if (pc->Width <= 0) {
          pc->Width = Width;
          vpx_internal_error(&pc->error, VPX_CODEC_CORRUPT_FRAME,
                             "Invalid frame width");
        }
John Koleszar's avatar
John Koleszar committed
971

John Koleszar's avatar
John Koleszar committed
972
973
974
975
976
        if (pc->Height <= 0) {
          pc->Height = Height;
          vpx_internal_error(&pc->error, VPX_CODEC_CORRUPT_FRAME,
                             "Invalid frame height");
        }
977

John Koleszar's avatar
John Koleszar committed
978
979
980
981
        if (vp8_alloc_frame_buffers(pc, pc->Width, pc->Height))
          vpx_internal_error(&pc->error, VPX_CODEC_MEM_ERROR,
                             "Failed to allocate frame buffers");
      }
982
    }
John Koleszar's avatar
John Koleszar committed
983
  }
John Koleszar's avatar
John Koleszar committed
984

John Koleszar's avatar
John Koleszar committed
985
986
987
988
  if ((!pbi->decoded_key_frame && pc->frame_type != KEY_FRAME) ||
      pc->Width == 0 || pc->Height == 0) {
    return -1;
  }
989

John Koleszar's avatar
John Koleszar committed
990
  init_frame(pbi);
John Koleszar's avatar
John Koleszar committed
991

John Koleszar's avatar
John Koleszar committed
992
  if (vp8dx_start_decode(&header_bc, data, first_partition_length_in_bytes))
John Koleszar's avatar
John Koleszar committed
993
994
995
    vpx_internal_error(&pc->error, VPX_CODEC_MEM_ERROR,
                       "Failed to allocate bool decoder 0");
  if (pc->frame_type == KEY_FRAME) {
John Koleszar's avatar
John Koleszar committed
996
997
    pc->clr_type    = (YUV_TYPE)vp8_read_bit(&header_bc);
    pc->clamp_type  = (CLAMP_TYPE)vp8_read_bit(&header_bc);
John Koleszar's avatar
John Koleszar committed
998
  }
John Koleszar's avatar
John Koleszar committed
999

John Koleszar's avatar
John Koleszar committed
1000
  /* Is segmentation enabled */
For faster browsing, not all history is shown. View entire blame