summaryrefslogtreecommitdiff
path: root/src/gallium/auxiliary/vl/vp8/vp8_decoder.h
blob: 58f9b6454e1626d84095f816e8e59075e921c141 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */


#ifndef VP8_DECODER_H
#define VP8_DECODER_H

#ifdef __cplusplus
extern "C"
{
#endif

#include <stdint.h>

#include "pipe/p_video_decoder.h"

#include "vp8_debug.h"
#include "entropymv.h"
#include "entropy.h"
#include "yv12utils.h"
#include "treereader.h"
#include "dequantize.h"
#include "recon_dispatch.h"
#include "idct_dispatch.h"

#define MINQ 0
#define MAXQ 127
#define QINDEX_RANGE (MAXQ + 1)

#define NUM_YV12_BUFFERS 4

typedef struct
{
    vp8_prob bmode_prob[VP8_BINTRAMODES - 1];
    vp8_prob ymode_prob[VP8_YMODES - 1];     /**< interframe intra mode probs */
    vp8_prob uv_mode_prob[VP8_UV_MODES - 1];
    vp8_prob sub_mv_ref_prob[VP8_SUBMVREFS - 1];
    vp8_prob coef_probs[BLOCK_TYPES][COEF_BANDS][PREV_COEF_CONTEXTS][ENTROPY_NODES];
    MV_CONTEXT mvc[2];
} FRAME_CONTEXT;

typedef struct
{
    int16_t min_val;
    int16_t length;
    uint8_t probs[12];
} TOKEN_EXTRABITS;

typedef enum
{
    ONE_PARTITION   = 0,
    TWO_PARTITION   = 1,
    FOUR_PARTITION  = 2,
    EIGHT_PARTITION = 3
} TOKEN_PARTITION;

typedef enum
{
    RECON_CLAMP_REQUIRED    = 0,
    RECON_CLAMP_NOTREQUIRED = 1
} CLAMP_TYPE;

typedef enum
{
    SIXTAP   = 0,
    BILINEAR = 1
} INTERPOLATIONFILTER_TYPE;

typedef enum
{
    NORMAL_LOOPFILTER = 0,
    SIMPLE_LOOPFILTER = 1
} LOOPFILTER_TYPE;

typedef struct VP8Common
{
    struct vpx_internal_error_info error;

    /* Header content */
    FRAME_TYPE frame_type;
    int version;
    int show_frame;

    int width;
    int height;
    int horiz_scale;
    int vert_scale;

    YUV_TYPE clr_type;
    CLAMP_TYPE clamp_type;

    INTERPOLATIONFILTER_TYPE mcomp_filter_type;
    LOOPFILTER_TYPE filter_type;

    int filter_level;
    int last_sharpness_level;
    int sharpness_level;

    int refresh_last_frame;    /**< Two state 0 = NO, 1 = YES */
    int refresh_golden_frame;  /**< Two state 0 = NO, 1 = YES */
    int refresh_alt_ref_frame; /**< Two state 0 = NO, 1 = YES */
    int refresh_entropy_probs; /**< Two state 0 = NO, 1 = YES */

    /* Frame buffers */
    YV12_BUFFER_CONFIG *frame_to_show;
    YV12_BUFFER_CONFIG yv12_fb[NUM_YV12_BUFFERS];
    int fb_idx_ref_cnt[NUM_YV12_BUFFERS];
    int new_fb_idx, lst_fb_idx, gld_fb_idx, alt_fb_idx;

    int frame_flags;
    int MBs;
    int mb_rows;
    int mb_cols;
    int mode_info_stride;

    /* Profile settings */
    int mb_no_coeff_skip;
    int no_lpf;
    int use_bilinear_mc_filter;
    int full_pixel;

    /* Quantization */
    DECLARE_ALIGNED(16, short, Y1dequant[QINDEX_RANGE][16]);
    DECLARE_ALIGNED(16, short, Y2dequant[QINDEX_RANGE][16]);
    DECLARE_ALIGNED(16, short, UVdequant[QINDEX_RANGE][16]);

    int base_qindex;
    int last_kf_gf_q;  /**< Q used on the last GF or KF */

    int y1dc_delta_q;
    int y2dc_delta_q;
    int y2ac_delta_q;
    int uvdc_delta_q;
    int uvac_delta_q;

    /* We allocate a MODE_INFO struct for each macroblock, together with
       an extra row on top and column on the left to simplify prediction. */

    MODE_INFO *mip; /**< Base of allocated array */
    MODE_INFO *mi;  /**< Corresponds to upper left visible macroblock */

    int copy_buffer_to_gf;     /**< 0 none, 1 Last to GF, 2 ARF to GF */
    int copy_buffer_to_arf;    /**< 0 none, 1 Last to ARF, 2 GF to ARF */

    int ref_frame_sign_bias[MAX_REF_FRAMES]; /**< Two state 0, 1 */

    ENTROPY_CONTEXT_PLANES *above_context; /**< Row of context for each plane */
    ENTROPY_CONTEXT_PLANES left_context;   /**< (up to) 4 contexts */

    /* keyframe block modes are predicted by their above, left neighbors */

    vp8_prob kf_bmode_prob[VP8_BINTRAMODES][VP8_BINTRAMODES][VP8_BINTRAMODES - 1];
    vp8_prob kf_ymode_prob[VP8_YMODES - 1];  /**< keyframe */
    vp8_prob kf_uv_mode_prob[VP8_UV_MODES - 1];

    FRAME_CONTEXT lfc; /**< Last frame entropy */
    FRAME_CONTEXT fc;  /**< Current frame entropy */

    TOKEN_PARTITION multi_token_partition;

} VP8_COMMON;

typedef struct
{
    DECLARE_ALIGNED(16, MACROBLOCKD, mb);
    DECLARE_ALIGNED(16, VP8_COMMON, common);

    BOOL_DECODER bd, bd2;
    BOOL_DECODER *mbd;

    const unsigned char *data;
    unsigned int         data_size;

    vp8_prob prob_intra;
    vp8_prob prob_last;
    vp8_prob prob_gf;
    vp8_prob prob_skip_false;

} VP8D_COMP;

/* ************************************************************************** */

VP8D_COMP *vp8_decoder_create();

int vp8_decoder_start(VP8D_COMP *pbi, struct pipe_vp8_picture_desc *frame_header,
                      const unsigned char *data, unsigned data_size);

int vp8_decoder_getframe(VP8D_COMP *pbi, YV12_BUFFER_CONFIG *sd);

void vp8_decoder_remove(VP8D_COMP *pbi);

int vp8_frame_decode(VP8D_COMP *pbi, struct pipe_vp8_picture_desc *frame_header);

#ifdef __cplusplus
}
#endif

#endif /* VP8_DECODER_H */