vp9_detokenize.c 16.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
13
#include "vp9/common/vp9_type_aliases.h"
#include "vp9/common/vp9_blockd.h"
14
#include "vp9/decoder/vp9_onyxd_int.h"
John Koleszar's avatar
John Koleszar committed
15
16
#include "vpx_mem/vpx_mem.h"
#include "vpx_ports/mem.h"
17
#include "vp9/decoder/vp9_detokenize.h"
John Koleszar's avatar
John Koleszar committed
18

19
#include "vp9/common/vp9_seg_common.h"
20

John Koleszar's avatar
John Koleszar committed
21
22
23
24
25
26
27
28
29
30
31
32
#define EOB_CONTEXT_NODE            0
#define ZERO_CONTEXT_NODE           1
#define ONE_CONTEXT_NODE            2
#define LOW_VAL_CONTEXT_NODE        3
#define TWO_CONTEXT_NODE            4
#define THREE_CONTEXT_NODE          5
#define HIGH_LOW_CONTEXT_NODE       6
#define CAT_ONE_CONTEXT_NODE        7
#define CAT_THREEFOUR_CONTEXT_NODE  8
#define CAT_THREE_CONTEXT_NODE      9
#define CAT_FIVE_CONTEXT_NODE       10

Scott LaVarnway's avatar
Scott LaVarnway committed
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#define CAT1_MIN_VAL    5
#define CAT2_MIN_VAL    7
#define CAT3_MIN_VAL   11
#define CAT4_MIN_VAL   19
#define CAT5_MIN_VAL   35
#define CAT6_MIN_VAL   67
#define CAT1_PROB0    159
#define CAT2_PROB0    145
#define CAT2_PROB1    165

#define CAT3_PROB0 140
#define CAT3_PROB1 148
#define CAT3_PROB2 173

#define CAT4_PROB0 135
#define CAT4_PROB1 140
#define CAT4_PROB2 155
#define CAT4_PROB3 176

#define CAT5_PROB0 130
#define CAT5_PROB1 134
#define CAT5_PROB2 141
#define CAT5_PROB3 157
#define CAT5_PROB4 180

58
59
60
static const unsigned char cat6_prob[15] = {
  254, 254, 254, 252, 249, 243, 230, 196, 177, 153, 140, 133, 130, 129, 0
};
John Koleszar's avatar
John Koleszar committed
61

62
void vp9_reset_mb_tokens_context(MACROBLOCKD* const xd) {
63
  /* Clear entropy contexts */
Paul Wilkins's avatar
Paul Wilkins committed
64
  if ((xd->mode_info_context->mbmi.mode != B_PRED &&
65
66
67
       xd->mode_info_context->mbmi.mode != I8X8_PRED &&
       xd->mode_info_context->mbmi.mode != SPLITMV)
      || xd->mode_info_context->mbmi.txfm_size == TX_16X16) {
Paul Wilkins's avatar
Paul Wilkins committed
68
69
    vpx_memset(xd->above_context, 0, sizeof(ENTROPY_CONTEXT_PLANES));
    vpx_memset(xd->left_context, 0, sizeof(ENTROPY_CONTEXT_PLANES));
John Koleszar's avatar
John Koleszar committed
70
  } else {
Paul Wilkins's avatar
Paul Wilkins committed
71
72
    vpx_memset(xd->above_context, 0, sizeof(ENTROPY_CONTEXT_PLANES) - 1);
    vpx_memset(xd->left_context, 0, sizeof(ENTROPY_CONTEXT_PLANES) - 1);
Yaowu Xu's avatar
Yaowu Xu committed
73
74
    xd->above_context->y2 = 1;
    xd->left_context->y2 = 1;
John Koleszar's avatar
John Koleszar committed
75
  }
John Koleszar's avatar
John Koleszar committed
76
}
77

78
DECLARE_ALIGNED(16, extern const unsigned char, vp9_norm[256]);
79

80
static int get_signed(BOOL_DECODER *br, int value_to_sign) {
John Koleszar's avatar
John Koleszar committed
81
  const int split = (br->range + 1) >> 1;
82
  const VP9_BD_VALUE bigsplit = (VP9_BD_VALUE)split << (VP9_BD_VALUE_SIZE - 8);
John Koleszar's avatar
John Koleszar committed
83
  int v;
John Koleszar's avatar
John Koleszar committed
84

John Koleszar's avatar
John Koleszar committed
85
  if (br->count < 0)
86
    vp9_bool_decoder_fill(br);
John Koleszar's avatar
John Koleszar committed
87

John Koleszar's avatar
John Koleszar committed
88
89
90
91
92
93
94
95
96
97
98
  if (br->value < bigsplit) {
    br->range = split;
    v = value_to_sign;
  } else {
    br->range = br->range - split;
    br->value = br->value - bigsplit;
    v = -value_to_sign;
  }
  br->range += br->range;
  br->value += br->value;
  --br->count;
Fritz Koenig's avatar
Fritz Koenig committed
99

John Koleszar's avatar
John Koleszar committed
100
  return v;
Daniel Kang's avatar
Daniel Kang committed
101
}
John Koleszar's avatar
John Koleszar committed
102

103
104
105
106
107
108
109
#define INCREMENT_COUNT(token)               \
  do {                                       \
    coef_counts[coef_bands[c]][pt][token]++; \
    pt = vp9_prev_token_class[token];        \
  } while (0)

#define WRITE_COEF_CONTINUE(val, token)                       \
Daniel Kang's avatar
Daniel Kang committed
110
  {                                                           \
111
    qcoeff_ptr[scan[c]] = (INT16) get_signed(br, val);        \
112
    INCREMENT_COUNT(token);                                   \
John Koleszar's avatar
John Koleszar committed
113
114
115
    c++;                                                      \
    continue;                                                 \
  }
John Koleszar's avatar
John Koleszar committed
116

Daniel Kang's avatar
Daniel Kang committed
117
118
#define ADJUST_COEF(prob, bits_count)  \
  do {                                 \
119
    if (vp9_read(br, prob))            \
John Koleszar's avatar
John Koleszar committed
120
121
      val += (UINT16)(1 << bits_count);\
  } while (0);
John Koleszar's avatar
John Koleszar committed
122

123
static int decode_coefs(VP9D_COMP *dx, const MACROBLOCKD *xd,
John Koleszar's avatar
John Koleszar committed
124
125
126
127
                        BOOL_DECODER* const br,
                        ENTROPY_CONTEXT *a, ENTROPY_CONTEXT *l,
                        PLANE_TYPE type,
                        TX_TYPE tx_type,
128
                        int seg_eob, INT16 *qcoeff_ptr,
129
                        const int *const scan, TX_SIZE txfm_size,
John Koleszar's avatar
John Koleszar committed
130
                        const int *coef_bands) {
John Koleszar's avatar
John Koleszar committed
131
  FRAME_CONTEXT *const fc = &dx->common.fc;
132
  int pt, c = (type == PLANE_TYPE_Y_NO_DC);
133
  vp9_prob (*coef_probs)[PREV_COEF_CONTEXTS][ENTROPY_NODES], *prob;
134
  unsigned int (*coef_counts)[PREV_COEF_CONTEXTS][MAX_ENTROPY_TOKENS];
Daniel Kang's avatar
Daniel Kang committed
135

136
  switch (txfm_size) {
Daniel Kang's avatar
Daniel Kang committed
137
    default:
Daniel Kang's avatar
Daniel Kang committed
138
    case TX_4X4:
139
      if (tx_type == DCT_DCT) {
140
        coef_probs  = fc->coef_probs[type];
141
142
        coef_counts = fc->coef_counts[type];
      } else {
143
        coef_probs  = fc->hybrid_coef_probs[type];
144
145
        coef_counts = fc->hybrid_coef_counts[type];
      }
Daniel Kang's avatar
Daniel Kang committed
146
147
      break;
    case TX_8X8:
148
      if (tx_type == DCT_DCT) {
149
        coef_probs  = fc->coef_probs_8x8[type];
150
151
        coef_counts = fc->coef_counts_8x8[type];
      } else {
152
        coef_probs  = fc->hybrid_coef_probs_8x8[type];
153
154
        coef_counts = fc->hybrid_coef_counts_8x8[type];
      }
Daniel Kang's avatar
Daniel Kang committed
155
      break;
Daniel Kang's avatar
Daniel Kang committed
156
    case TX_16X16:
157
      if (tx_type == DCT_DCT) {
158
        coef_probs  = fc->coef_probs_16x16[type];
159
160
        coef_counts = fc->coef_counts_16x16[type];
      } else {
161
        coef_probs  = fc->hybrid_coef_probs_16x16[type];
162
163
        coef_counts = fc->hybrid_coef_counts_16x16[type];
      }
Daniel Kang's avatar
Daniel Kang committed
164
      break;
165
166
167
168
169
170
#if CONFIG_TX32X32 && CONFIG_SUPERBLOCKS
    case TX_32X32:
      coef_probs = fc->coef_probs_32x32[type];
      coef_counts = fc->coef_counts_32x32[type];
      break;
#endif
Daniel Kang's avatar
Daniel Kang committed
171
  }
John Koleszar's avatar
John Koleszar committed
172

173
  VP9_COMBINEENTROPYCONTEXTS(pt, *a, *l);
John Koleszar's avatar
John Koleszar committed
174
175
176
  while (1) {
    int val;
    const uint8_t *cat6 = cat6_prob;
177
    if (c >= seg_eob) break;
178
    prob = coef_probs[coef_bands[c]][pt];
179
    if (!vp9_read(br, prob[EOB_CONTEXT_NODE]))
John Koleszar's avatar
John Koleszar committed
180
      break;
Daniel Kang's avatar
Daniel Kang committed
181
SKIP_START:
182
    if (c >= seg_eob) break;
183
    if (!vp9_read(br, prob[ZERO_CONTEXT_NODE])) {
184
      INCREMENT_COUNT(ZERO_TOKEN);
John Koleszar's avatar
John Koleszar committed
185
      ++c;
186
      prob = coef_probs[coef_bands[c]][pt];
John Koleszar's avatar
John Koleszar committed
187
188
189
      goto SKIP_START;
    }
    // ONE_CONTEXT_NODE_0_
190
    if (!vp9_read(br, prob[ONE_CONTEXT_NODE])) {
191
      WRITE_COEF_CONTINUE(1, ONE_TOKEN);
John Koleszar's avatar
John Koleszar committed
192
193
    }
    // LOW_VAL_CONTEXT_NODE_0_
194
195
    if (!vp9_read(br, prob[LOW_VAL_CONTEXT_NODE])) {
      if (!vp9_read(br, prob[TWO_CONTEXT_NODE])) {
196
        WRITE_COEF_CONTINUE(2, TWO_TOKEN);
John Koleszar's avatar
John Koleszar committed
197
      }
198
      if (!vp9_read(br, prob[THREE_CONTEXT_NODE])) {
199
        WRITE_COEF_CONTINUE(3, THREE_TOKEN);
John Koleszar's avatar
John Koleszar committed
200
      }
201
      WRITE_COEF_CONTINUE(4, FOUR_TOKEN);
John Koleszar's avatar
John Koleszar committed
202
203
    }
    // HIGH_LOW_CONTEXT_NODE_0_
204
205
    if (!vp9_read(br, prob[HIGH_LOW_CONTEXT_NODE])) {
      if (!vp9_read(br, prob[CAT_ONE_CONTEXT_NODE])) {
John Koleszar's avatar
John Koleszar committed
206
207
        val = CAT1_MIN_VAL;
        ADJUST_COEF(CAT1_PROB0, 0);
208
        WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY1);
John Koleszar's avatar
John Koleszar committed
209
210
211
212
      }
      val = CAT2_MIN_VAL;
      ADJUST_COEF(CAT2_PROB1, 1);
      ADJUST_COEF(CAT2_PROB0, 0);
213
      WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY2);
Daniel Kang's avatar
Daniel Kang committed
214
    }
John Koleszar's avatar
John Koleszar committed
215
    // CAT_THREEFOUR_CONTEXT_NODE_0_
216
217
    if (!vp9_read(br, prob[CAT_THREEFOUR_CONTEXT_NODE])) {
      if (!vp9_read(br, prob[CAT_THREE_CONTEXT_NODE])) {
John Koleszar's avatar
John Koleszar committed
218
219
220
221
        val = CAT3_MIN_VAL;
        ADJUST_COEF(CAT3_PROB2, 2);
        ADJUST_COEF(CAT3_PROB1, 1);
        ADJUST_COEF(CAT3_PROB0, 0);
222
        WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY3);
John Koleszar's avatar
John Koleszar committed
223
224
225
226
227
228
      }
      val = CAT4_MIN_VAL;
      ADJUST_COEF(CAT4_PROB3, 3);
      ADJUST_COEF(CAT4_PROB2, 2);
      ADJUST_COEF(CAT4_PROB1, 1);
      ADJUST_COEF(CAT4_PROB0, 0);
229
      WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY4);
John Koleszar's avatar
John Koleszar committed
230
231
    }
    // CAT_FIVE_CONTEXT_NODE_0_:
232
    if (!vp9_read(br, prob[CAT_FIVE_CONTEXT_NODE])) {
John Koleszar's avatar
John Koleszar committed
233
234
235
236
237
238
      val = CAT5_MIN_VAL;
      ADJUST_COEF(CAT5_PROB4, 4);
      ADJUST_COEF(CAT5_PROB3, 3);
      ADJUST_COEF(CAT5_PROB2, 2);
      ADJUST_COEF(CAT5_PROB1, 1);
      ADJUST_COEF(CAT5_PROB0, 0);
239
      WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY5);
John Koleszar's avatar
John Koleszar committed
240
241
242
    }
    val = 0;
    while (*cat6) {
243
      val = (val << 1) | vp9_read(br, *cat6++);
John Koleszar's avatar
John Koleszar committed
244
245
    }
    val += CAT6_MIN_VAL;
246
    WRITE_COEF_CONTINUE(val, DCT_VAL_CATEGORY6);
John Koleszar's avatar
John Koleszar committed
247
  }
Jingning Han's avatar
Jingning Han committed
248

249
250
  if (c < seg_eob)
    coef_counts[coef_bands[c]][pt][DCT_EOB_TOKEN]++;
251

252
  a[0] = l[0] = (c > !type);
253

John Koleszar's avatar
John Koleszar committed
254
  return c;
Daniel Kang's avatar
Daniel Kang committed
255
}
John Koleszar's avatar
John Koleszar committed
256

257
static int get_eob(MACROBLOCKD* const xd, int segment_id, int eob_max) {
258
259
260
261
262
263
264
265
  int active = vp9_segfeature_active(xd, segment_id, SEG_LVL_EOB);
  int eob = vp9_get_segdata(xd, segment_id, SEG_LVL_EOB);

  if (!active || eob > eob_max)
    eob = eob_max;
  return eob;
}

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
#if CONFIG_TX32X32 && CONFIG_SUPERBLOCKS
int vp9_decode_sb_tokens(VP9D_COMP* const pbi,
                         MACROBLOCKD* const xd,
                         BOOL_DECODER* const bc) {
  ENTROPY_CONTEXT* const A = (ENTROPY_CONTEXT *)xd->above_context;
  ENTROPY_CONTEXT* const L = (ENTROPY_CONTEXT *)xd->left_context;
  unsigned short* const eobs = xd->eobs;
  const int segment_id = xd->mode_info_context->mbmi.segment_id;
  int c, i, eobtotal = 0, seg_eob;

  // Luma block
  eobs[0] = c = decode_coefs(pbi, xd, bc, A, L, PLANE_TYPE_Y_WITH_DC,
                             DCT_DCT, get_eob(xd, segment_id, 1024),
                             xd->sb_coeff_data.qcoeff,
                             vp9_default_zig_zag1d_32x32,
                             TX_32X32, vp9_coef_bands_32x32);
  A[1] = A[2] = A[3] = A[0];
  L[1] = L[2] = L[3] = L[0];
  eobtotal += c;

  // 16x16 chroma blocks
  seg_eob = get_eob(xd, segment_id, 256);
  for (i = 16; i < 24; i += 4) {
    ENTROPY_CONTEXT* const a = A + vp9_block2above_8x8[i];
    ENTROPY_CONTEXT* const l = L + vp9_block2left_8x8[i];

    eobs[i] = c = decode_coefs(pbi, xd, bc, a, l, PLANE_TYPE_UV,
                               DCT_DCT, seg_eob,
                               xd->sb_coeff_data.qcoeff + 1024 + (i - 16) * 64,
                               vp9_default_zig_zag1d_16x16,
                               TX_16X16, vp9_coef_bands_16x16);
    a[1] = a[0];
    l[1] = l[0];
    eobtotal += c;
  }

  // no Y2 block
  vpx_memset(&A[8], 0, sizeof(A[8]));
  vpx_memset(&L[8], 0, sizeof(L[8]));

  vpx_memcpy(xd->above_context + 1, xd->above_context,
             sizeof(ENTROPY_CONTEXT_PLANES));
  vpx_memcpy(xd->left_context + 1, xd->left_context,
             sizeof(ENTROPY_CONTEXT_PLANES));

  return eobtotal;
}
#endif
314

315
316
317
static int vp9_decode_mb_tokens_16x16(VP9D_COMP* const pbi,
                                      MACROBLOCKD* const xd,
                                      BOOL_DECODER* const bc) {
Daniel Kang's avatar
Daniel Kang committed
318
319
  ENTROPY_CONTEXT* const A = (ENTROPY_CONTEXT *)xd->above_context;
  ENTROPY_CONTEXT* const L = (ENTROPY_CONTEXT *)xd->left_context;
Yunqing Wang's avatar
Yunqing Wang committed
320
  unsigned short* const eobs = xd->eobs;
Daniel Kang's avatar
Daniel Kang committed
321
  const int segment_id = xd->mode_info_context->mbmi.segment_id;
322
  int c, i, eobtotal = 0, seg_eob;
Daniel Kang's avatar
Daniel Kang committed
323
324

  // Luma block
325
326
327
328
329
330
331
332
  eobs[0] = c = decode_coefs(pbi, xd, bc, A, L, PLANE_TYPE_Y_WITH_DC,
                             get_tx_type(xd, &xd->block[0]),
                             get_eob(xd, segment_id, 256),
                             xd->qcoeff, vp9_default_zig_zag1d_16x16,
                             TX_16X16, vp9_coef_bands_16x16);
  A[1] = A[2] = A[3] = A[0];
  L[1] = L[2] = L[3] = L[0];
  eobtotal += c;
Daniel Kang's avatar
Daniel Kang committed
333
334

  // 8x8 chroma blocks
335
  seg_eob = get_eob(xd, segment_id, 64);
Daniel Kang's avatar
Daniel Kang committed
336
  for (i = 16; i < 24; i += 4) {
337
338
    ENTROPY_CONTEXT* const a = A + vp9_block2above_8x8[i];
    ENTROPY_CONTEXT* const l = L + vp9_block2left_8x8[i];
Daniel Kang's avatar
Daniel Kang committed
339

340
341
342
343
    eobs[i] = c = decode_coefs(pbi, xd, bc, a, l, PLANE_TYPE_UV,
                               DCT_DCT, seg_eob, xd->block[i].qcoeff,
                               vp9_default_zig_zag1d_8x8,
                               TX_8X8, vp9_coef_bands_8x8);
Daniel Kang's avatar
Daniel Kang committed
344
345
346
347
    a[1] = a[0];
    l[1] = l[0];
    eobtotal += c;
  }
348
349
  A[8] = 0;
  L[8] = 0;
Daniel Kang's avatar
Daniel Kang committed
350
351
352
  return eobtotal;
}

353
354
355
static int vp9_decode_mb_tokens_8x8(VP9D_COMP* const pbi,
                                    MACROBLOCKD* const xd,
                                    BOOL_DECODER* const bc) {
John Koleszar's avatar
John Koleszar committed
356
357
  ENTROPY_CONTEXT *const A = (ENTROPY_CONTEXT *)xd->above_context;
  ENTROPY_CONTEXT *const L = (ENTROPY_CONTEXT *)xd->left_context;
Yunqing Wang's avatar
Yunqing Wang committed
358
  unsigned short *const eobs = xd->eobs;
359
360
  PLANE_TYPE type;
  int c, i, eobtotal = 0, seg_eob;
John Koleszar's avatar
John Koleszar committed
361
  const int segment_id = xd->mode_info_context->mbmi.segment_id;
362

363
  int has_2nd_order = get_2nd_order_usage(xd);
364
  // 2nd order DC block
365
  if (has_2nd_order) {
366
367
    ENTROPY_CONTEXT *const a = A + vp9_block2above_8x8[24];
    ENTROPY_CONTEXT *const l = L + vp9_block2left_8x8[24];
John Koleszar's avatar
John Koleszar committed
368

369
370
371
372
    eobs[24] = c = decode_coefs(pbi, xd, bc, a, l, PLANE_TYPE_Y2,
                                DCT_DCT, get_eob(xd, segment_id, 4),
                                xd->block[24].qcoeff,
                                vp9_default_zig_zag1d, TX_8X8, vp9_coef_bands);
John Koleszar's avatar
John Koleszar committed
373
374
    eobtotal += c - 4;
    type = PLANE_TYPE_Y_NO_DC;
375
  } else {
Yaowu Xu's avatar
Yaowu Xu committed
376
377
    xd->above_context->y2 = 1;
    xd->left_context->y2 = 1;
378
    eobs[24] = 0;
John Koleszar's avatar
John Koleszar committed
379
    type = PLANE_TYPE_Y_WITH_DC;
380
  }
John Koleszar's avatar
John Koleszar committed
381

382
  // luma blocks
383
  seg_eob = get_eob(xd, segment_id, 64);
384
  for (i = 0; i < 16; i += 4) {
385
386
    ENTROPY_CONTEXT *const a = A + vp9_block2above_8x8[i];
    ENTROPY_CONTEXT *const l = L + vp9_block2left_8x8[i];
John Koleszar's avatar
John Koleszar committed
387

388
    eobs[i] = c = decode_coefs(pbi, xd, bc, a, l, type,
389
                               type == PLANE_TYPE_Y_WITH_DC ?
390
                               get_tx_type(xd, xd->block + i) : DCT_DCT,
391
392
393
                               seg_eob, xd->block[i].qcoeff,
                               vp9_default_zig_zag1d_8x8,
                               TX_8X8, vp9_coef_bands_8x8);
John Koleszar's avatar
John Koleszar committed
394
395
396
397
398
    a[1] = a[0];
    l[1] = l[0];
    eobtotal += c;
  }

399
400
401
402
  // chroma blocks
  if (xd->mode_info_context->mbmi.mode == I8X8_PRED ||
      xd->mode_info_context->mbmi.mode == SPLITMV) {
    // use 4x4 transform for U, V components in I8X8/splitmv prediction mode
403
    seg_eob = get_eob(xd, segment_id, 16);
404
    for (i = 16; i < 24; i++) {
405
406
      ENTROPY_CONTEXT *const a = A + vp9_block2above[i];
      ENTROPY_CONTEXT *const l = L + vp9_block2left[i];
407

408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
      eobs[i] = c = decode_coefs(pbi, xd, bc, a, l, PLANE_TYPE_UV,
                                 DCT_DCT, seg_eob, xd->block[i].qcoeff,
                                 vp9_default_zig_zag1d, TX_4X4, vp9_coef_bands);
      eobtotal += c;
    }
  } else {
    for (i = 16; i < 24; i += 4) {
      ENTROPY_CONTEXT *const a = A + vp9_block2above_8x8[i];
      ENTROPY_CONTEXT *const l = L + vp9_block2left_8x8[i];

      eobs[i] = c = decode_coefs(pbi, xd, bc, a, l, PLANE_TYPE_UV,
                                 DCT_DCT, seg_eob, xd->block[i].qcoeff,
                                 vp9_default_zig_zag1d_8x8,
                                 TX_8X8, vp9_coef_bands_8x8);
      a[1] = a[0];
      l[1] = l[0];
424
425
426
427
      eobtotal += c;
    }
  }

John Koleszar's avatar
John Koleszar committed
428
  return eobtotal;
429
430
}

431
432
433
int vp9_decode_coefs_4x4(VP9D_COMP *dx, MACROBLOCKD *xd,
                         BOOL_DECODER* const bc,
                         PLANE_TYPE type, int i) {
John Koleszar's avatar
John Koleszar committed
434
435
  ENTROPY_CONTEXT *const A = (ENTROPY_CONTEXT *)xd->above_context;
  ENTROPY_CONTEXT *const L = (ENTROPY_CONTEXT *)xd->left_context;
436
437
  ENTROPY_CONTEXT *const a = A + vp9_block2above[i];
  ENTROPY_CONTEXT *const l = L + vp9_block2left[i];
John Koleszar's avatar
John Koleszar committed
438
  INT16 *qcoeff_ptr = &xd->qcoeff[0];
439
440
  const int *scan = vp9_default_zig_zag1d;
  unsigned short *const eobs = xd->eobs;
John Koleszar's avatar
John Koleszar committed
441
  int segment_id = xd->mode_info_context->mbmi.segment_id;
442
443
  int c, seg_eob = get_eob(xd, segment_id, 16);
  TX_TYPE tx_type = DCT_DCT;
444
445

  if (type == PLANE_TYPE_Y_WITH_DC)
446
    tx_type = get_tx_type_4x4(xd, &xd->block[i]);
447
448
449
450
451
452
453
454
455
456
457
458
459
  switch (tx_type) {
    case ADST_DCT :
      scan = vp9_row_scan;
      break;

    case DCT_ADST :
      scan = vp9_col_scan;
      break;

    default :
      scan = vp9_default_zig_zag1d;
      break;
  }
460
461
462
  eobs[i] = c = decode_coefs(dx, xd, bc, a, l, type,
                             tx_type, seg_eob, qcoeff_ptr + i * 16,
                             scan, TX_4X4, vp9_coef_bands);
463
464
465
  return c;
}

466
467
468
469
470
471
472
473
474
475
476
int vp9_decode_mb_tokens_4x4_uv(VP9D_COMP* const dx,
                                MACROBLOCKD* const xd,
                                BOOL_DECODER* const bc) {
  int eobtotal = 0, i;

  for (i = 16; i < 24; i++)
    eobtotal += vp9_decode_coefs_4x4(dx, xd, bc, PLANE_TYPE_UV, i);

  return eobtotal;
}

477
478
479
static int vp9_decode_mb_tokens_4x4(VP9D_COMP* const dx,
                                    MACROBLOCKD* const xd,
                                    BOOL_DECODER* const bc) {
480
481
  int i, eobtotal = 0;
  PLANE_TYPE type;
482

483
484
485
  int has_2nd_order = get_2nd_order_usage(xd);

  if (has_2nd_order) {
486
    eobtotal += vp9_decode_coefs_4x4(dx, xd, bc, PLANE_TYPE_Y2, 24) - 16;
John Koleszar's avatar
John Koleszar committed
487
488
    type = PLANE_TYPE_Y_NO_DC;
  } else {
Yaowu Xu's avatar
Yaowu Xu committed
489
490
    xd->above_context->y2 = 1;
    xd->left_context->y2 = 1;
491
    xd->eobs[24] = 0;
John Koleszar's avatar
John Koleszar committed
492
493
    type = PLANE_TYPE_Y_WITH_DC;
  }
John Koleszar's avatar
John Koleszar committed
494

495
  for (i = 0; i < 16; ++i) {
496
    eobtotal += vp9_decode_coefs_4x4(dx, xd, bc, type, i);
John Koleszar's avatar
John Koleszar committed
497
  }
498
499

  return eobtotal + vp9_decode_mb_tokens_4x4_uv(dx, xd, bc);
John Koleszar's avatar
John Koleszar committed
500
}
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518

int vp9_decode_mb_tokens(VP9D_COMP* const dx,
                         MACROBLOCKD* const xd,
                         BOOL_DECODER* const bc) {
  const TX_SIZE tx_size = xd->mode_info_context->mbmi.txfm_size;
  int eobtotal;

  if (tx_size == TX_16X16) {
    eobtotal = vp9_decode_mb_tokens_16x16(dx, xd, bc);
  } else if (tx_size == TX_8X8) {
    eobtotal = vp9_decode_mb_tokens_8x8(dx, xd, bc);
  } else {
    assert(tx_size == TX_4X4);
    eobtotal = vp9_decode_mb_tokens_4x4(dx, xd, bc);
  }

  return eobtotal;
}