1/*
2 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  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.
9 */
10
11
12#include "vp8/common/type_aliases.h"
13#include "vp8/common/blockd.h"
14#include "onyxd_int.h"
15#include "vpx_mem/vpx_mem.h"
16#include "vpx_ports/mem.h"
17#include "detokenize.h"
18
19#define BOOL_DATA UINT8
20
21#define OCB_X PREV_COEF_CONTEXTS * ENTROPY_NODES
22DECLARE_ALIGNED(16, static const unsigned char, coef_bands_x[16]) =
23{
24    0 * OCB_X, 1 * OCB_X, 2 * OCB_X, 3 * OCB_X,
25    6 * OCB_X, 4 * OCB_X, 5 * OCB_X, 6 * OCB_X,
26    6 * OCB_X, 6 * OCB_X, 6 * OCB_X, 6 * OCB_X,
27    6 * OCB_X, 6 * OCB_X, 6 * OCB_X, 7 * OCB_X
28};
29#define EOB_CONTEXT_NODE            0
30#define ZERO_CONTEXT_NODE           1
31#define ONE_CONTEXT_NODE            2
32#define LOW_VAL_CONTEXT_NODE        3
33#define TWO_CONTEXT_NODE            4
34#define THREE_CONTEXT_NODE          5
35#define HIGH_LOW_CONTEXT_NODE       6
36#define CAT_ONE_CONTEXT_NODE        7
37#define CAT_THREEFOUR_CONTEXT_NODE  8
38#define CAT_THREE_CONTEXT_NODE      9
39#define CAT_FIVE_CONTEXT_NODE       10
40
41/*
42//the definition is put in "onyxd_int.h"
43typedef struct
44{
45    INT16         min_val;
46    INT16         Length;
47    UINT8 Probs[12];
48} TOKENEXTRABITS;
49*/
50
51DECLARE_ALIGNED(16, static const TOKENEXTRABITS, vp8d_token_extra_bits2[MAX_ENTROPY_TOKENS]) =
52{
53    {  0, -1, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },  /* ZERO_TOKEN */
54    {  1, 0, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },   /* ONE_TOKEN */
55    {  2, 0, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },   /* TWO_TOKEN */
56    {  3, 0, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },   /* THREE_TOKEN */
57    {  4, 0, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },   /* FOUR_TOKEN */
58    {  5, 0, { 159, 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },  /* DCT_VAL_CATEGORY1 */
59    {  7, 1, { 145, 165, 0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } }, /* DCT_VAL_CATEGORY2 */
60    { 11, 2, { 140, 148, 173, 0,  0,  0,  0,  0,  0,  0,  0,  0   } }, /* DCT_VAL_CATEGORY3 */
61    { 19, 3, { 135, 140, 155, 176, 0,  0,  0,  0,  0,  0,  0,  0   } }, /* DCT_VAL_CATEGORY4 */
62    { 35, 4, { 130, 134, 141, 157, 180, 0,  0,  0,  0,  0,  0,  0   } }, /* DCT_VAL_CATEGORY5 */
63    { 67, 10, { 129, 130, 133, 140, 153, 177, 196, 230, 243, 254, 254, 0   } }, /* DCT_VAL_CATEGORY6 */
64    {  0, -1, { 0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0   } },  /*  EOB TOKEN */
65};
66
67
68void vp8_reset_mb_tokens_context(MACROBLOCKD *x)
69{
70    /* Clear entropy contexts for Y2 blocks */
71    if (x->mode_info_context->mbmi.mode != B_PRED && x->mode_info_context->mbmi.mode != SPLITMV)
72    {
73        vpx_memset(x->above_context, 0, sizeof(ENTROPY_CONTEXT_PLANES));
74        vpx_memset(x->left_context, 0, sizeof(ENTROPY_CONTEXT_PLANES));
75    }
76    else
77    {
78        vpx_memset(x->above_context, 0, sizeof(ENTROPY_CONTEXT_PLANES)-1);
79        vpx_memset(x->left_context, 0, sizeof(ENTROPY_CONTEXT_PLANES)-1);
80    }
81}
82
83DECLARE_ALIGNED(16, extern const unsigned char, vp8dx_bitreader_norm[256]);
84#define FILL \
85    if(count < 0) \
86        VP8DX_BOOL_DECODER_FILL(count, value, bufptr, bufend);
87
88#define NORMALIZE \
89    /*if(range < 0x80)*/                            \
90    { \
91        shift = vp8dx_bitreader_norm[range]; \
92        range <<= shift; \
93        value <<= shift; \
94        count -= shift; \
95    }
96
97#define DECODE_AND_APPLYSIGN(value_to_sign) \
98    split = (range + 1) >> 1; \
99    bigsplit = (VP8_BD_VALUE)split << (VP8_BD_VALUE_SIZE - 8); \
100    FILL \
101    if ( value < bigsplit ) \
102    { \
103        range = split; \
104        v= value_to_sign; \
105    } \
106    else \
107    { \
108        range = range-split; \
109        value = value-bigsplit; \
110        v = -value_to_sign; \
111    } \
112    range +=range;                   \
113    value +=value;                   \
114    count--;
115
116#define DECODE_AND_BRANCH_IF_ZERO(probability,branch) \
117    { \
118        split = 1 +  ((( probability*(range-1) ) )>> 8); \
119        bigsplit = (VP8_BD_VALUE)split << (VP8_BD_VALUE_SIZE - 8); \
120        FILL \
121        if ( value < bigsplit ) \
122        { \
123            range = split; \
124            NORMALIZE \
125            goto branch; \
126        } \
127        value -= bigsplit; \
128        range = range - split; \
129        NORMALIZE \
130    }
131
132#define DECODE_AND_LOOP_IF_ZERO(probability,branch) \
133    { \
134        split = 1 + ((( probability*(range-1) ) ) >> 8); \
135        bigsplit = (VP8_BD_VALUE)split << (VP8_BD_VALUE_SIZE - 8); \
136        FILL \
137        if ( value < bigsplit ) \
138        { \
139            range = split; \
140            NORMALIZE \
141            Prob = coef_probs; \
142            if(c<15) {\
143            ++c; \
144            Prob += coef_bands_x[c]; \
145            goto branch; \
146            } goto BLOCK_FINISHED; /*for malformed input */\
147        } \
148        value -= bigsplit; \
149        range = range - split; \
150        NORMALIZE \
151    }
152
153#define DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val) \
154    DECODE_AND_APPLYSIGN(val) \
155    Prob = coef_probs + (ENTROPY_NODES*2); \
156    if(c < 15){\
157        qcoeff_ptr [ scan[c] ] = (INT16) v; \
158        ++c; \
159        goto DO_WHILE; }\
160    qcoeff_ptr [ scan[15] ] = (INT16) v; \
161    goto BLOCK_FINISHED;
162
163
164#define DECODE_EXTRABIT_AND_ADJUST_VAL(t,bits_count)\
165    split = 1 +  (((range-1) * vp8d_token_extra_bits2[t].Probs[bits_count]) >> 8); \
166    bigsplit = (VP8_BD_VALUE)split << (VP8_BD_VALUE_SIZE - 8); \
167    FILL \
168    if(value >= bigsplit)\
169    {\
170        range = range-split;\
171        value = value-bigsplit;\
172        val += ((UINT16)1<<bits_count);\
173    }\
174    else\
175    {\
176        range = split;\
177    }\
178    NORMALIZE
179
180int vp8_decode_mb_tokens(VP8D_COMP *dx, MACROBLOCKD *x)
181{
182    ENTROPY_CONTEXT *A = (ENTROPY_CONTEXT *)x->above_context;
183    ENTROPY_CONTEXT *L = (ENTROPY_CONTEXT *)x->left_context;
184    const VP8_COMMON *const oc = & dx->common;
185
186    BOOL_DECODER *bc = x->current_bc;
187
188    char *eobs = x->eobs;
189
190    ENTROPY_CONTEXT *a;
191    ENTROPY_CONTEXT *l;
192    int i;
193
194    int eobtotal = 0;
195
196    register int count;
197
198    const BOOL_DATA *bufptr;
199    const BOOL_DATA *bufend;
200    register unsigned int range;
201    VP8_BD_VALUE value;
202    const int *scan;
203    register unsigned int shift;
204    UINT32 split;
205    VP8_BD_VALUE bigsplit;
206    INT16 *qcoeff_ptr;
207
208    const vp8_prob *coef_probs;
209    int type;
210    int stop;
211    INT16 val, bits_count;
212    INT16 c;
213    INT16 v;
214    const vp8_prob *Prob;
215
216    type = 3;
217    i = 0;
218    stop = 16;
219
220    scan = vp8_default_zig_zag1d;
221    qcoeff_ptr = &x->qcoeff[0];
222
223    if (x->mode_info_context->mbmi.mode != B_PRED && x->mode_info_context->mbmi.mode != SPLITMV)
224    {
225        i = 24;
226        stop = 24;
227        type = 1;
228        qcoeff_ptr += 24*16;
229        eobtotal -= 16;
230    }
231
232    bufend  = bc->user_buffer_end;
233    bufptr  = bc->user_buffer;
234    value   = bc->value;
235    count   = bc->count;
236    range   = bc->range;
237
238
239    coef_probs = oc->fc.coef_probs [type] [ 0 ] [0];
240
241BLOCK_LOOP:
242    a = A + vp8_block2above[i];
243    l = L + vp8_block2left[i];
244
245    c = (INT16)(!type);
246
247    /*Dest = ((A)!=0) + ((B)!=0);*/
248    VP8_COMBINEENTROPYCONTEXTS(v, *a, *l);
249    Prob = coef_probs;
250    Prob += v * ENTROPY_NODES;
251
252DO_WHILE:
253    Prob += coef_bands_x[c];
254    DECODE_AND_BRANCH_IF_ZERO(Prob[EOB_CONTEXT_NODE], BLOCK_FINISHED);
255
256CHECK_0_:
257    DECODE_AND_LOOP_IF_ZERO(Prob[ZERO_CONTEXT_NODE], CHECK_0_);
258    DECODE_AND_BRANCH_IF_ZERO(Prob[ONE_CONTEXT_NODE], ONE_CONTEXT_NODE_0_);
259    DECODE_AND_BRANCH_IF_ZERO(Prob[LOW_VAL_CONTEXT_NODE], LOW_VAL_CONTEXT_NODE_0_);
260    DECODE_AND_BRANCH_IF_ZERO(Prob[HIGH_LOW_CONTEXT_NODE], HIGH_LOW_CONTEXT_NODE_0_);
261    DECODE_AND_BRANCH_IF_ZERO(Prob[CAT_THREEFOUR_CONTEXT_NODE], CAT_THREEFOUR_CONTEXT_NODE_0_);
262    DECODE_AND_BRANCH_IF_ZERO(Prob[CAT_FIVE_CONTEXT_NODE], CAT_FIVE_CONTEXT_NODE_0_);
263    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY6].min_val;
264    bits_count = vp8d_token_extra_bits2[DCT_VAL_CATEGORY6].Length;
265
266    do
267    {
268        DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY6, bits_count);
269        bits_count -- ;
270    }
271    while (bits_count >= 0);
272
273    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
274
275CAT_FIVE_CONTEXT_NODE_0_:
276    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY5].min_val;
277    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY5, 4);
278    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY5, 3);
279    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY5, 2);
280    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY5, 1);
281    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY5, 0);
282    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
283
284CAT_THREEFOUR_CONTEXT_NODE_0_:
285    DECODE_AND_BRANCH_IF_ZERO(Prob[CAT_THREE_CONTEXT_NODE], CAT_THREE_CONTEXT_NODE_0_);
286    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY4].min_val;
287    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY4, 3);
288    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY4, 2);
289    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY4, 1);
290    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY4, 0);
291    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
292
293CAT_THREE_CONTEXT_NODE_0_:
294    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY3].min_val;
295    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY3, 2);
296    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY3, 1);
297    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY3, 0);
298    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
299
300HIGH_LOW_CONTEXT_NODE_0_:
301    DECODE_AND_BRANCH_IF_ZERO(Prob[CAT_ONE_CONTEXT_NODE], CAT_ONE_CONTEXT_NODE_0_);
302
303    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY2].min_val;
304    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY2, 1);
305    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY2, 0);
306    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
307
308CAT_ONE_CONTEXT_NODE_0_:
309    val = vp8d_token_extra_bits2[DCT_VAL_CATEGORY1].min_val;
310    DECODE_EXTRABIT_AND_ADJUST_VAL(DCT_VAL_CATEGORY1, 0);
311    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(val);
312
313LOW_VAL_CONTEXT_NODE_0_:
314    DECODE_AND_BRANCH_IF_ZERO(Prob[TWO_CONTEXT_NODE], TWO_CONTEXT_NODE_0_);
315    DECODE_AND_BRANCH_IF_ZERO(Prob[THREE_CONTEXT_NODE], THREE_CONTEXT_NODE_0_);
316    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(4);
317
318THREE_CONTEXT_NODE_0_:
319    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(3);
320
321TWO_CONTEXT_NODE_0_:
322    DECODE_SIGN_WRITE_COEFF_AND_CHECK_EXIT(2);
323
324ONE_CONTEXT_NODE_0_:
325    DECODE_AND_APPLYSIGN(1);
326    Prob = coef_probs + ENTROPY_NODES;
327
328    if (c < 15)
329    {
330        qcoeff_ptr [ scan[c] ] = (INT16) v;
331        ++c;
332        goto DO_WHILE;
333    }
334
335    qcoeff_ptr [ scan[15] ] = (INT16) v;
336BLOCK_FINISHED:
337    *a = *l = ((eobs[i] = c) != !type);   /* any nonzero data? */
338    eobtotal += c;
339    qcoeff_ptr += 16;
340
341    i++;
342
343    if (i < stop)
344        goto BLOCK_LOOP;
345
346    if (i == 25)
347    {
348        type = 0;
349        i = 0;
350        stop = 16;
351        coef_probs = oc->fc.coef_probs [type] [ 0 ] [0];
352        qcoeff_ptr -= (24*16 + 16);
353        goto BLOCK_LOOP;
354    }
355
356    if (i == 16)
357    {
358        type = 2;
359        coef_probs = oc->fc.coef_probs [type] [ 0 ] [0];
360        stop = 24;
361        goto BLOCK_LOOP;
362    }
363
364    FILL
365    bc->user_buffer = bufptr;
366    bc->value = value;
367    bc->count = count;
368    bc->range = range;
369    return eobtotal;
370
371}
372