Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Copyright (C) 2006 Aurelien Jacobs <aurel@gnuage.org> | ||
3 | * | ||
4 | * This file is part of FFmpeg. | ||
5 | * | ||
6 | * FFmpeg is free software; you can redistribute it and/or | ||
7 | * modify it under the terms of the GNU Lesser General Public | ||
8 | * License as published by the Free Software Foundation; either | ||
9 | * version 2.1 of the License, or (at your option) any later version. | ||
10 | * | ||
11 | * FFmpeg is distributed in the hope that it will be useful, | ||
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
14 | * Lesser General Public License for more details. | ||
15 | * | ||
16 | * You should have received a copy of the GNU Lesser General Public | ||
17 | * License along with FFmpeg; if not, write to the Free Software | ||
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
19 | */ | ||
20 | |||
21 | /** | ||
22 | * @file | ||
23 | * VP5 and VP6 compatible video decoder (common features) | ||
24 | */ | ||
25 | |||
26 | #ifndef AVCODEC_VP56_H | ||
27 | #define AVCODEC_VP56_H | ||
28 | |||
29 | #include "libavutil/mem_internal.h" | ||
30 | |||
31 | #include "avcodec.h" | ||
32 | #include "get_bits.h" | ||
33 | #include "hpeldsp.h" | ||
34 | #include "h264chroma.h" | ||
35 | #include "videodsp.h" | ||
36 | #include "vp3dsp.h" | ||
37 | #include "vp56dsp.h" | ||
38 | #include "vpx_rac.h" | ||
39 | |||
40 | typedef struct vp56_context VP56Context; | ||
41 | |||
42 | typedef enum { | ||
43 | VP56_FRAME_NONE =-1, | ||
44 | VP56_FRAME_CURRENT = 0, | ||
45 | VP56_FRAME_PREVIOUS = 1, | ||
46 | VP56_FRAME_GOLDEN = 2, | ||
47 | } VP56Frame; | ||
48 | |||
49 | typedef enum { | ||
50 | VP56_MB_INTER_NOVEC_PF = 0, /**< Inter MB, no vector, from previous frame */ | ||
51 | VP56_MB_INTRA = 1, /**< Intra MB */ | ||
52 | VP56_MB_INTER_DELTA_PF = 2, /**< Inter MB, above/left vector + delta, from previous frame */ | ||
53 | VP56_MB_INTER_V1_PF = 3, /**< Inter MB, first vector, from previous frame */ | ||
54 | VP56_MB_INTER_V2_PF = 4, /**< Inter MB, second vector, from previous frame */ | ||
55 | VP56_MB_INTER_NOVEC_GF = 5, /**< Inter MB, no vector, from golden frame */ | ||
56 | VP56_MB_INTER_DELTA_GF = 6, /**< Inter MB, above/left vector + delta, from golden frame */ | ||
57 | VP56_MB_INTER_4V = 7, /**< Inter MB, 4 vectors, from previous frame */ | ||
58 | VP56_MB_INTER_V1_GF = 8, /**< Inter MB, first vector, from golden frame */ | ||
59 | VP56_MB_INTER_V2_GF = 9, /**< Inter MB, second vector, from golden frame */ | ||
60 | } VP56mb; | ||
61 | |||
62 | typedef struct VP56Tree { | ||
63 | int8_t val; | ||
64 | int8_t prob_idx; | ||
65 | } VP56Tree; | ||
66 | |||
67 | typedef struct VP56mv { | ||
68 | DECLARE_ALIGNED(4, int16_t, x); | ||
69 | int16_t y; | ||
70 | } VP56mv; | ||
71 | |||
72 | #define VP56_SIZE_CHANGE 1 | ||
73 | |||
74 | typedef void (*VP56ParseVectorAdjustment)(VP56Context *s, | ||
75 | VP56mv *vect); | ||
76 | typedef void (*VP56Filter)(VP56Context *s, uint8_t *dst, uint8_t *src, | ||
77 | int offset1, int offset2, ptrdiff_t stride, | ||
78 | VP56mv mv, int mask, int select, int luma); | ||
79 | typedef int (*VP56ParseCoeff)(VP56Context *s); | ||
80 | typedef void (*VP56DefaultModelsInit)(VP56Context *s); | ||
81 | typedef void (*VP56ParseVectorModels)(VP56Context *s); | ||
82 | typedef int (*VP56ParseCoeffModels)(VP56Context *s); | ||
83 | typedef int (*VP56ParseHeader)(VP56Context *s, const uint8_t *buf, | ||
84 | int buf_size); | ||
85 | |||
86 | typedef struct VP56RefDc { | ||
87 | uint8_t not_null_dc; | ||
88 | VP56Frame ref_frame; | ||
89 | int16_t dc_coeff; | ||
90 | } VP56RefDc; | ||
91 | |||
92 | typedef struct VP56Macroblock { | ||
93 | uint8_t type; | ||
94 | VP56mv mv; | ||
95 | } VP56Macroblock; | ||
96 | |||
97 | typedef struct VP56Model { | ||
98 | uint8_t coeff_reorder[64]; /* used in vp6 only */ | ||
99 | uint8_t coeff_index_to_pos[64]; /* used in vp6 only */ | ||
100 | uint8_t coeff_index_to_idct_selector[64]; /* used in vp6 only */ | ||
101 | uint8_t vector_sig[2]; /* delta sign */ | ||
102 | uint8_t vector_dct[2]; /* delta coding types */ | ||
103 | uint8_t vector_pdi[2][2]; /* predefined delta init */ | ||
104 | uint8_t vector_pdv[2][7]; /* predefined delta values */ | ||
105 | uint8_t vector_fdv[2][8]; /* 8 bit delta value definition */ | ||
106 | uint8_t coeff_dccv[2][11]; /* DC coeff value */ | ||
107 | uint8_t coeff_ract[2][3][6][11]; /* Run/AC coding type and AC coeff value */ | ||
108 | uint8_t coeff_acct[2][3][3][6][5];/* vp5 only AC coding type for coding group < 3 */ | ||
109 | uint8_t coeff_dcct[2][36][5]; /* DC coeff coding type */ | ||
110 | uint8_t coeff_runv[2][14]; /* run value (vp6 only) */ | ||
111 | uint8_t mb_type[3][10][10]; /* model for decoding MB type */ | ||
112 | uint8_t mb_types_stats[3][10][2];/* contextual, next MB type stats */ | ||
113 | } VP56Model; | ||
114 | |||
115 | struct vp56_context { | ||
116 | AVCodecContext *avctx; | ||
117 | H264ChromaContext h264chroma; | ||
118 | HpelDSPContext hdsp; | ||
119 | VideoDSPContext vdsp; | ||
120 | VP3DSPContext vp3dsp; | ||
121 | VP56DSPContext vp56dsp; | ||
122 | uint8_t idct_scantable[64]; | ||
123 | AVFrame *frames[4]; | ||
124 | uint8_t *edge_emu_buffer_alloc; | ||
125 | uint8_t *edge_emu_buffer; | ||
126 | VPXRangeCoder c; | ||
127 | VPXRangeCoder cc; | ||
128 | VPXRangeCoder *ccp; | ||
129 | int sub_version; | ||
130 | |||
131 | /* frame info */ | ||
132 | int golden_frame; | ||
133 | int plane_width[4]; | ||
134 | int plane_height[4]; | ||
135 | int mb_width; /* number of horizontal MB */ | ||
136 | int mb_height; /* number of vertical MB */ | ||
137 | int block_offset[6]; | ||
138 | |||
139 | int quantizer; | ||
140 | uint16_t dequant_dc; | ||
141 | uint16_t dequant_ac; | ||
142 | |||
143 | /* DC predictors management */ | ||
144 | VP56RefDc *above_blocks; | ||
145 | VP56RefDc left_block[4]; | ||
146 | int above_block_idx[6]; | ||
147 | int16_t prev_dc[3][3]; /* [plan][ref_frame] */ | ||
148 | |||
149 | /* blocks / macroblock */ | ||
150 | VP56mb mb_type; | ||
151 | VP56Macroblock *macroblocks; | ||
152 | DECLARE_ALIGNED(16, int16_t, block_coeff)[6][64]; | ||
153 | int idct_selector[6]; | ||
154 | |||
155 | /* motion vectors */ | ||
156 | VP56mv mv[6]; /* vectors for each block in MB */ | ||
157 | VP56mv vector_candidate[2]; | ||
158 | int vector_candidate_pos; | ||
159 | |||
160 | /* filtering hints */ | ||
161 | int filter_header; /* used in vp6 only */ | ||
162 | int deblock_filtering; | ||
163 | int filter_selection; | ||
164 | int filter_mode; | ||
165 | int max_vector_length; | ||
166 | int sample_variance_threshold; | ||
167 | DECLARE_ALIGNED(8, int, bounding_values_array)[256]; | ||
168 | |||
169 | uint8_t coeff_ctx[4][64]; /* used in vp5 only */ | ||
170 | uint8_t coeff_ctx_last[4]; /* used in vp5 only */ | ||
171 | |||
172 | int has_alpha; | ||
173 | |||
174 | /* upside-down flipping hints */ | ||
175 | int flip; /* are we flipping ? */ | ||
176 | int frbi; /* first row block index in MB */ | ||
177 | int srbi; /* second row block index in MB */ | ||
178 | ptrdiff_t stride[4]; /* stride for each plan */ | ||
179 | |||
180 | const uint8_t *vp56_coord_div; | ||
181 | VP56ParseVectorAdjustment parse_vector_adjustment; | ||
182 | VP56Filter filter; | ||
183 | VP56ParseCoeff parse_coeff; | ||
184 | VP56DefaultModelsInit default_models_init; | ||
185 | VP56ParseVectorModels parse_vector_models; | ||
186 | VP56ParseCoeffModels parse_coeff_models; | ||
187 | VP56ParseHeader parse_header; | ||
188 | |||
189 | /* for "slice" parallelism between YUV and A */ | ||
190 | VP56Context *alpha_context; | ||
191 | |||
192 | VP56Model *modelp; | ||
193 | VP56Model model; | ||
194 | |||
195 | /* huffman decoding */ | ||
196 | int use_huffman; | ||
197 | GetBitContext gb; | ||
198 | VLC dccv_vlc[2]; | ||
199 | VLC runv_vlc[2]; | ||
200 | VLC ract_vlc[2][3][6]; | ||
201 | unsigned int nb_null[2][2]; /* number of consecutive NULL DC/AC */ | ||
202 | |||
203 | int have_undamaged_frame; | ||
204 | int discard_frame; | ||
205 | }; | ||
206 | |||
207 | |||
208 | /** | ||
209 | * Initializes an VP56Context. Expects its caller to clean up | ||
210 | * in case of error. | ||
211 | */ | ||
212 | int ff_vp56_init_context(AVCodecContext *avctx, VP56Context *s, | ||
213 | int flip, int has_alpha); | ||
214 | int ff_vp56_free_context(VP56Context *s); | ||
215 | void ff_vp56_init_dequant(VP56Context *s, int quantizer); | ||
216 | int ff_vp56_decode_frame(AVCodecContext *avctx, AVFrame *frame, | ||
217 | int *got_frame, AVPacket *avpkt); | ||
218 | |||
219 | |||
220 | /** | ||
221 | * vp56 specific range coder implementation | ||
222 | */ | ||
223 | |||
224 | 52284 | static int vp56_rac_gets(VPXRangeCoder *c, int bits) | |
225 | { | ||
226 | 52284 | int value = 0; | |
227 | |||
228 |
2/2✓ Branch 0 taken 151547 times.
✓ Branch 1 taken 52284 times.
|
203831 | while (bits--) { |
229 | 151547 | value = (value << 1) | vpx_rac_get(c); | |
230 | } | ||
231 | |||
232 | 52284 | return value; | |
233 | } | ||
234 | |||
235 | // P(7) | ||
236 | 8535 | static av_unused int vp56_rac_gets_nn(VPXRangeCoder *c, int bits) | |
237 | { | ||
238 | 8535 | int v = vp56_rac_gets(c, 7) << 1; | |
239 | 8535 | return v + !v; | |
240 | } | ||
241 | |||
242 | static av_always_inline | ||
243 | 445956 | int vp56_rac_get_tree(VPXRangeCoder *c, | |
244 | const VP56Tree *tree, | ||
245 | const uint8_t *probs) | ||
246 | { | ||
247 |
2/2✓ Branch 0 taken 1320748 times.
✓ Branch 1 taken 445956 times.
|
1766704 | while (tree->val > 0) { |
248 |
2/2✓ Branch 1 taken 451188 times.
✓ Branch 2 taken 869560 times.
|
1320748 | if (vpx_rac_get_prob_branchy(c, probs[tree->prob_idx])) |
249 | 451188 | tree += tree->val; | |
250 | else | ||
251 | 869560 | tree++; | |
252 | } | ||
253 | 445956 | return -tree->val; | |
254 | } | ||
255 | |||
256 | #endif /* AVCODEC_VP56_H */ | ||
257 |