Mercurial > libavcodec.hg
comparison vc1.c @ 3359:87187ebe2c28 libavcodec
Rename VC9 to VC1 as there is no such thing as VC9
author | kostya |
---|---|
date | Tue, 27 Jun 2006 02:55:54 +0000 |
parents | |
children | 2c4ddf5b9217 |
comparison
equal
deleted
inserted
replaced
3358:4ae69b5b596b | 3359:87187ebe2c28 |
---|---|
1 /* | |
2 * VC-1 and WMV3 decoder | |
3 * Copyright (c) 2005 Anonymous | |
4 * Copyright (c) 2005 Alex Beregszaszi | |
5 * Copyright (c) 2005 Michael Niedermayer | |
6 * | |
7 * This library is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public | |
9 * License as published by the Free Software Foundation; either | |
10 * version 2 of the License, or (at your option) any later version. | |
11 * | |
12 * This library is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
18 * License along with this library; if not, write to the Free Software | |
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
20 * | |
21 */ | |
22 | |
23 /** | |
24 * @file vc1.c | |
25 * VC-1 and WMV3 decoder | |
26 * | |
27 * TODO: most AP stuff, optimize, most of MB layer, transform, filtering and motion compensation, etc | |
28 * TODO: use MPV_ !! | |
29 */ | |
30 #include "common.h" | |
31 #include "dsputil.h" | |
32 #include "avcodec.h" | |
33 #include "mpegvideo.h" | |
34 #include "vc1data.h" | |
35 | |
36 #undef NDEBUG | |
37 #include <assert.h> | |
38 | |
39 extern const uint32_t ff_table0_dc_lum[120][2], ff_table1_dc_lum[120][2]; | |
40 extern const uint32_t ff_table0_dc_chroma[120][2], ff_table1_dc_chroma[120][2]; | |
41 extern VLC ff_msmp4_dc_luma_vlc[2], ff_msmp4_dc_chroma_vlc[2]; | |
42 #define MB_INTRA_VLC_BITS 9 | |
43 extern VLC ff_msmp4_mb_i_vlc; | |
44 #define DC_VLC_BITS 9 | |
45 static const uint16_t table_mb_intra[64][2]; | |
46 | |
47 /* Some inhibiting stuff */ | |
48 #define HAS_ADVANCED_PROFILE 0 | |
49 #define TRACE 1 | |
50 | |
51 #if TRACE | |
52 # define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
53 codes, codes_wrap, codes_size, use_static) \ | |
54 if (init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
55 codes, codes_wrap, codes_size, use_static) < 0) \ | |
56 { \ | |
57 av_log(v->s.avctx, AV_LOG_ERROR, "Error for " # vlc " (%i)\n", i); \ | |
58 return -1; \ | |
59 } | |
60 #else | |
61 # define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
62 codes, codes_wrap, codes_size, use_static) \ | |
63 init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
64 codes, codes_wrap, codes_size, use_static) | |
65 #endif | |
66 | |
67 /** Available Profiles */ | |
68 //@{ | |
69 #define PROFILE_SIMPLE 0 | |
70 #define PROFILE_MAIN 1 | |
71 #define PROFILE_COMPLEX 2 ///< TODO: WMV9 specific | |
72 #define PROFILE_ADVANCED 3 | |
73 //@} | |
74 | |
75 /** Sequence quantizer mode */ | |
76 //@{ | |
77 #define QUANT_FRAME_IMPLICIT 0 ///< Implicitly specified at frame level | |
78 #define QUANT_FRAME_EXPLICIT 1 ///< Explicitly specified at frame level | |
79 #define QUANT_NON_UNIFORM 2 ///< Non-uniform quant used for all frames | |
80 #define QUANT_UNIFORM 3 ///< Uniform quant used for all frames | |
81 //@} | |
82 | |
83 /** Where quant can be changed */ | |
84 //@{ | |
85 #define DQPROFILE_FOUR_EDGES 0 | |
86 #define DQPROFILE_DOUBLE_EDGES 1 | |
87 #define DQPROFILE_SINGLE_EDGE 2 | |
88 #define DQPROFILE_ALL_MBS 3 | |
89 //@} | |
90 | |
91 /** @name Where quant can be changed | |
92 */ | |
93 //@{ | |
94 #define DQPROFILE_FOUR_EDGES 0 | |
95 #define DQSINGLE_BEDGE_LEFT 0 | |
96 #define DQSINGLE_BEDGE_TOP 1 | |
97 #define DQSINGLE_BEDGE_RIGHT 2 | |
98 #define DQSINGLE_BEDGE_BOTTOM 3 | |
99 //@} | |
100 | |
101 /** Which pair of edges is quantized with ALTPQUANT */ | |
102 //@{ | |
103 #define DQDOUBLE_BEDGE_TOPLEFT 0 | |
104 #define DQDOUBLE_BEDGE_TOPRIGHT 1 | |
105 #define DQDOUBLE_BEDGE_BOTTOMRIGHT 2 | |
106 #define DQDOUBLE_BEDGE_BOTTOMLEFT 3 | |
107 //@} | |
108 | |
109 /** MV modes for P frames */ | |
110 //@{ | |
111 #define MV_PMODE_1MV_HPEL_BILIN 0 | |
112 #define MV_PMODE_1MV 1 | |
113 #define MV_PMODE_1MV_HPEL 2 | |
114 #define MV_PMODE_MIXED_MV 3 | |
115 #define MV_PMODE_INTENSITY_COMP 4 | |
116 //@} | |
117 | |
118 /** @name MV types for B frames */ | |
119 //@{ | |
120 #define BMV_TYPE_BACKWARD 0 | |
121 #define BMV_TYPE_BACKWARD 0 | |
122 #define BMV_TYPE_FORWARD 1 | |
123 #define BMV_TYPE_INTERPOLATED 3 | |
124 //@} | |
125 | |
126 /** MV P mode - the 5th element is only used for mode 1 */ | |
127 static const uint8_t mv_pmode_table[2][5] = { | |
128 { MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_1MV, MV_PMODE_1MV_HPEL, MV_PMODE_MIXED_MV, MV_PMODE_INTENSITY_COMP }, | |
129 { MV_PMODE_1MV, MV_PMODE_MIXED_MV, MV_PMODE_1MV_HPEL, MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_INTENSITY_COMP } | |
130 }; | |
131 | |
132 /** One more frame type */ | |
133 #define BI_TYPE 7 | |
134 | |
135 static const int fps_nr[5] = { 24, 25, 30, 50, 60 }, | |
136 fps_dr[2] = { 1000, 1001 }; | |
137 static const uint8_t pquant_table[3][32] = { | |
138 { /* Implicit quantizer */ | |
139 0, 1, 2, 3, 4, 5, 6, 7, 8, 6, 7, 8, 9, 10, 11, 12, | |
140 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 27, 29, 31 | |
141 }, | |
142 { /* Explicit quantizer, pquantizer uniform */ | |
143 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, | |
144 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31 | |
145 }, | |
146 { /* Explicit quantizer, pquantizer non-uniform */ | |
147 0, 1, 1, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, | |
148 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 31 | |
149 } | |
150 }; | |
151 | |
152 /** @name VC-1 VLC tables and defines | |
153 * @todo TODO move this into the context | |
154 */ | |
155 //@{ | |
156 #define VC1_BFRACTION_VLC_BITS 7 | |
157 static VLC vc1_bfraction_vlc; | |
158 #define VC1_IMODE_VLC_BITS 4 | |
159 static VLC vc1_imode_vlc; | |
160 #define VC1_NORM2_VLC_BITS 3 | |
161 static VLC vc1_norm2_vlc; | |
162 #define VC1_NORM6_VLC_BITS 9 | |
163 static VLC vc1_norm6_vlc; | |
164 /* Could be optimized, one table only needs 8 bits */ | |
165 #define VC1_TTMB_VLC_BITS 9 //12 | |
166 static VLC vc1_ttmb_vlc[3]; | |
167 #define VC1_MV_DIFF_VLC_BITS 9 //15 | |
168 static VLC vc1_mv_diff_vlc[4]; | |
169 #define VC1_CBPCY_P_VLC_BITS 9 //14 | |
170 static VLC vc1_cbpcy_p_vlc[4]; | |
171 #define VC1_4MV_BLOCK_PATTERN_VLC_BITS 6 | |
172 static VLC vc1_4mv_block_pattern_vlc[4]; | |
173 #define VC1_TTBLK_VLC_BITS 5 | |
174 static VLC vc1_ttblk_vlc[3]; | |
175 #define VC1_SUBBLKPAT_VLC_BITS 6 | |
176 static VLC vc1_subblkpat_vlc[3]; | |
177 //@} | |
178 | |
179 /** Bitplane struct | |
180 * We mainly need data and is_raw, so this struct could be avoided | |
181 * to save a level of indirection; feel free to modify | |
182 * @fixme For now, stride=width | |
183 * @warning Data are bits, either 1 or 0 | |
184 */ | |
185 typedef struct BitPlane { | |
186 uint8_t *data; ///< Data buffer | |
187 int width; ///< Width of the buffer | |
188 int stride; ///< Stride of the buffer | |
189 int height; ///< Plane height | |
190 uint8_t is_raw; ///< Bit values must be read at MB level | |
191 } BitPlane; | |
192 | |
193 /** The VC1 Context | |
194 * @fixme Change size wherever another size is more efficient | |
195 * Many members are only used for Advanced Profile | |
196 */ | |
197 typedef struct VC1Context{ | |
198 MpegEncContext s; | |
199 | |
200 /** Simple/Main Profile sequence header */ | |
201 //@{ | |
202 int res_sm; ///< reserved, 2b | |
203 int res_x8; ///< reserved | |
204 int multires; ///< frame-level RESPIC syntax element present | |
205 int res_fasttx; ///< reserved, always 1 | |
206 int res_transtab; ///< reserved, always 0 | |
207 int rangered; ///< RANGEREDFRM (range reduction) syntax element present | |
208 ///< at frame level | |
209 int res_rtm_flag; ///< reserved, set to 1 | |
210 int reserved; ///< reserved | |
211 //@} | |
212 | |
213 #if HAS_ADVANCED_PROFILE | |
214 /** Advanced Profile */ | |
215 //@{ | |
216 int level; ///< 3bits, for Advanced/Simple Profile, provided by TS layer | |
217 int chromaformat; ///< 2bits, 2=4:2:0, only defined | |
218 int postprocflag; ///< Per-frame processing suggestion flag present | |
219 int broadcast; ///< TFF/RFF present | |
220 int interlace; ///< Progressive/interlaced (RPTFTM syntax element) | |
221 int tfcntrflag; ///< TFCNTR present | |
222 int panscanflag; ///< NUMPANSCANWIN, TOPLEFT{X,Y}, BOTRIGHT{X,Y} present | |
223 int extended_dmv; ///< Additional extended dmv range at P/B frame-level | |
224 int color_prim; ///< 8bits, chroma coordinates of the color primaries | |
225 int transfer_char; ///< 8bits, Opto-electronic transfer characteristics | |
226 int matrix_coef; ///< 8bits, Color primaries->YCbCr transform matrix | |
227 int hrd_param_flag; ///< Presence of Hypothetical Reference | |
228 ///< Decoder parameters | |
229 //@} | |
230 #endif | |
231 | |
232 | |
233 /** Sequence header data for all Profiles | |
234 * TODO: choose between ints, uint8_ts and monobit flags | |
235 */ | |
236 //@{ | |
237 int profile; ///< 2bits, Profile | |
238 int frmrtq_postproc; ///< 3bits, | |
239 int bitrtq_postproc; ///< 5bits, quantized framerate-based postprocessing strength | |
240 int fastuvmc; ///< Rounding of qpel vector to hpel ? (not in Simple) | |
241 int extended_mv; ///< Ext MV in P/B (not in Simple) | |
242 int dquant; ///< How qscale varies with MBs, 2bits (not in Simple) | |
243 int vstransform; ///< variable-size [48]x[48] transform type + info | |
244 int overlap; ///< overlapped transforms in use | |
245 int quantizer_mode; ///< 2bits, quantizer mode used for sequence, see QUANT_* | |
246 int finterpflag; ///< INTERPFRM present | |
247 //@} | |
248 | |
249 /** Frame decoding info for all profiles */ | |
250 //@{ | |
251 uint8_t mv_mode; ///< MV coding monde | |
252 uint8_t mv_mode2; ///< Secondary MV coding mode (B frames) | |
253 int k_x; ///< Number of bits for MVs (depends on MV range) | |
254 int k_y; ///< Number of bits for MVs (depends on MV range) | |
255 uint8_t pq, altpq; ///< Current/alternate frame quantizer scale | |
256 /** pquant parameters */ | |
257 //@{ | |
258 uint8_t dquantfrm; | |
259 uint8_t dqprofile; | |
260 uint8_t dqsbedge; | |
261 uint8_t dqbilevel; | |
262 //@} | |
263 /** AC coding set indexes | |
264 * @see 8.1.1.10, p(1)10 | |
265 */ | |
266 //@{ | |
267 int c_ac_table_index; ///< Chroma index from ACFRM element | |
268 int y_ac_table_index; ///< Luma index from AC2FRM element | |
269 //@} | |
270 int ttfrm; ///< Transform type info present at frame level | |
271 uint8_t ttmbf; ///< Transform type flag | |
272 int ttmb; ///< Transform type | |
273 uint8_t ttblk4x4; ///< Value of ttblk which indicates a 4x4 transform | |
274 /** Luma compensation parameters */ | |
275 //@{ | |
276 uint8_t lumscale; | |
277 uint8_t lumshift; | |
278 //@} | |
279 int16_t bfraction; ///< Relative position % anchors=> how to scale MVs | |
280 uint8_t halfpq; ///< Uniform quant over image and qp+.5 | |
281 uint8_t respic; ///< Frame-level flag for resized images | |
282 int buffer_fullness; ///< HRD info | |
283 /** Ranges: | |
284 * -# 0 -> [-64n 63.f] x [-32, 31.f] | |
285 * -# 1 -> [-128, 127.f] x [-64, 63.f] | |
286 * -# 2 -> [-512, 511.f] x [-128, 127.f] | |
287 * -# 3 -> [-1024, 1023.f] x [-256, 255.f] | |
288 */ | |
289 uint8_t mvrange; | |
290 uint8_t pquantizer; ///< Uniform (over sequence) quantizer in use | |
291 uint8_t *previous_line_cbpcy; ///< To use for predicted CBPCY | |
292 VLC *cbpcy_vlc; ///< CBPCY VLC table | |
293 int tt_index; ///< Index for Transform Type tables | |
294 BitPlane mv_type_mb_plane; ///< bitplane for mv_type == (4MV) | |
295 BitPlane skip_mb_plane; ///< bitplane for skipped MBs | |
296 BitPlane direct_mb_plane; ///< bitplane for "direct" MBs | |
297 | |
298 /** Frame decoding info for S/M profiles only */ | |
299 //@{ | |
300 uint8_t rangeredfrm; ///< out_sample = CLIP((in_sample-128)*2+128) | |
301 uint8_t interpfrm; | |
302 //@} | |
303 | |
304 #if HAS_ADVANCED_PROFILE | |
305 /** Frame decoding info for Advanced profile */ | |
306 //@{ | |
307 uint8_t fcm; ///< 0->Progressive, 2->Frame-Interlace, 3->Field-Interlace | |
308 uint8_t numpanscanwin; | |
309 uint8_t tfcntr; | |
310 uint8_t rptfrm, tff, rff; | |
311 uint16_t topleftx; | |
312 uint16_t toplefty; | |
313 uint16_t bottomrightx; | |
314 uint16_t bottomrighty; | |
315 uint8_t uvsamp; | |
316 uint8_t postproc; | |
317 int hrd_num_leaky_buckets; | |
318 uint8_t bit_rate_exponent; | |
319 uint8_t buffer_size_exponent; | |
320 BitPlane ac_pred_plane; ///< AC prediction flags bitplane | |
321 BitPlane over_flags_plane; ///< Overflags bitplane | |
322 uint8_t condover; | |
323 uint16_t *hrd_rate, *hrd_buffer; | |
324 uint8_t *hrd_fullness; | |
325 uint8_t range_mapy_flag; | |
326 uint8_t range_mapuv_flag; | |
327 uint8_t range_mapy; | |
328 uint8_t range_mapuv; | |
329 //@} | |
330 #endif | |
331 } VC1Context; | |
332 | |
333 /** | |
334 * Get unary code of limited length | |
335 * @fixme FIXME Slow and ugly | |
336 * @param gb GetBitContext | |
337 * @param[in] stop The bitstop value (unary code of 1's or 0's) | |
338 * @param[in] len Maximum length | |
339 * @return Unary length/index | |
340 */ | |
341 static int get_prefix(GetBitContext *gb, int stop, int len) | |
342 { | |
343 #if 1 | |
344 int i = 0, tmp = !stop; | |
345 | |
346 while (i != len && tmp != stop) | |
347 { | |
348 tmp = get_bits(gb, 1); | |
349 i++; | |
350 } | |
351 if (i == len && tmp != stop) return len+1; | |
352 return i; | |
353 #else | |
354 unsigned int buf; | |
355 int log; | |
356 | |
357 OPEN_READER(re, gb); | |
358 UPDATE_CACHE(re, gb); | |
359 buf=GET_CACHE(re, gb); //Still not sure | |
360 if (stop) buf = ~buf; | |
361 | |
362 log= av_log2(-buf); //FIXME: -? | |
363 if (log < limit){ | |
364 LAST_SKIP_BITS(re, gb, log+1); | |
365 CLOSE_READER(re, gb); | |
366 return log; | |
367 } | |
368 | |
369 LAST_SKIP_BITS(re, gb, limit); | |
370 CLOSE_READER(re, gb); | |
371 return limit; | |
372 #endif | |
373 } | |
374 | |
375 /** | |
376 * Init VC-1 specific tables and VC1Context members | |
377 * @param v The VC1Context to initialize | |
378 * @return Status | |
379 */ | |
380 static int vc1_init_common(VC1Context *v) | |
381 { | |
382 static int done = 0; | |
383 int i = 0; | |
384 | |
385 /* Set the bit planes */ | |
386 v->mv_type_mb_plane = (struct BitPlane) { NULL, 0, 0, 0 }; | |
387 v->direct_mb_plane = (struct BitPlane) { NULL, 0, 0, 0 }; | |
388 v->skip_mb_plane = (struct BitPlane) { NULL, 0, 0, 0 }; | |
389 #if HAS_ADVANCED_PROFILE | |
390 v->ac_pred_plane = v->over_flags_plane = (struct BitPlane) { NULL, 0, 0, 0 }; | |
391 v->hrd_rate = v->hrd_buffer = NULL; | |
392 #endif | |
393 | |
394 /* VLC tables */ | |
395 #if 0 // spec -> actual tables converter | |
396 for(i=0; i<64; i++){ | |
397 int code= (vc1_norm6_spec[i][1] << vc1_norm6_spec[i][4]) + vc1_norm6_spec[i][3]; | |
398 av_log(NULL, AV_LOG_DEBUG, "0x%03X, ", code); | |
399 if(i%16==15) av_log(NULL, AV_LOG_DEBUG, "\n"); | |
400 } | |
401 for(i=0; i<64; i++){ | |
402 int code= vc1_norm6_spec[i][2] + vc1_norm6_spec[i][4]; | |
403 av_log(NULL, AV_LOG_DEBUG, "%2d, ", code); | |
404 if(i%16==15) av_log(NULL, AV_LOG_DEBUG, "\n"); | |
405 } | |
406 #endif | |
407 if(!done) | |
408 { | |
409 done = 1; | |
410 INIT_VLC(&vc1_bfraction_vlc, VC1_BFRACTION_VLC_BITS, 23, | |
411 vc1_bfraction_bits, 1, 1, | |
412 vc1_bfraction_codes, 1, 1, 1); | |
413 INIT_VLC(&vc1_norm2_vlc, VC1_NORM2_VLC_BITS, 4, | |
414 vc1_norm2_bits, 1, 1, | |
415 vc1_norm2_codes, 1, 1, 1); | |
416 INIT_VLC(&vc1_norm6_vlc, VC1_NORM6_VLC_BITS, 64, | |
417 vc1_norm6_bits, 1, 1, | |
418 vc1_norm6_codes, 2, 2, 1); | |
419 INIT_VLC(&vc1_imode_vlc, VC1_IMODE_VLC_BITS, 7, | |
420 vc1_imode_bits, 1, 1, | |
421 vc1_imode_codes, 1, 1, 1); | |
422 for (i=0; i<3; i++) | |
423 { | |
424 INIT_VLC(&vc1_ttmb_vlc[i], VC1_TTMB_VLC_BITS, 16, | |
425 vc1_ttmb_bits[i], 1, 1, | |
426 vc1_ttmb_codes[i], 2, 2, 1); | |
427 INIT_VLC(&vc1_ttblk_vlc[i], VC1_TTBLK_VLC_BITS, 8, | |
428 vc1_ttblk_bits[i], 1, 1, | |
429 vc1_ttblk_codes[i], 1, 1, 1); | |
430 INIT_VLC(&vc1_subblkpat_vlc[i], VC1_SUBBLKPAT_VLC_BITS, 15, | |
431 vc1_subblkpat_bits[i], 1, 1, | |
432 vc1_subblkpat_codes[i], 1, 1, 1); | |
433 } | |
434 for(i=0; i<4; i++) | |
435 { | |
436 INIT_VLC(&vc1_4mv_block_pattern_vlc[i], VC1_4MV_BLOCK_PATTERN_VLC_BITS, 16, | |
437 vc1_4mv_block_pattern_bits[i], 1, 1, | |
438 vc1_4mv_block_pattern_codes[i], 1, 1, 1); | |
439 INIT_VLC(&vc1_cbpcy_p_vlc[i], VC1_CBPCY_P_VLC_BITS, 64, | |
440 vc1_cbpcy_p_bits[i], 1, 1, | |
441 vc1_cbpcy_p_codes[i], 2, 2, 1); | |
442 INIT_VLC(&vc1_mv_diff_vlc[i], VC1_MV_DIFF_VLC_BITS, 73, | |
443 vc1_mv_diff_bits[i], 1, 1, | |
444 vc1_mv_diff_codes[i], 2, 2, 1); | |
445 } | |
446 } | |
447 | |
448 /* Other defaults */ | |
449 v->pq = -1; | |
450 v->mvrange = 0; /* 7.1.1.18, p80 */ | |
451 | |
452 return 0; | |
453 } | |
454 | |
455 #if HAS_ADVANCED_PROFILE | |
456 /** | |
457 * Decode sequence header's Hypothetic Reference Decoder data | |
458 * @see 6.2.1, p32 | |
459 * @param v The VC1Context to initialize | |
460 * @param gb A GetBitContext initialized from AVCodecContext extra_data | |
461 * @return Status | |
462 */ | |
463 static int decode_hrd(VC1Context *v, GetBitContext *gb) | |
464 { | |
465 int i, num; | |
466 | |
467 num = 1 + get_bits(gb, 5); | |
468 | |
469 /*hrd rate*/ | |
470 if (v->hrd_rate || num != v->hrd_num_leaky_buckets) | |
471 { | |
472 av_freep(&v->hrd_rate); | |
473 } | |
474 if (!v->hrd_rate) v->hrd_rate = av_malloc(num*sizeof(uint16_t)); | |
475 if (!v->hrd_rate) return -1; | |
476 | |
477 /*hrd buffer*/ | |
478 if (v->hrd_buffer || num != v->hrd_num_leaky_buckets) | |
479 { | |
480 av_freep(&v->hrd_buffer); | |
481 } | |
482 if (!v->hrd_buffer) v->hrd_buffer = av_malloc(num*sizeof(uint16_t)); | |
483 if (!v->hrd_buffer) | |
484 { | |
485 av_freep(&v->hrd_rate); | |
486 return -1; | |
487 } | |
488 | |
489 /*hrd fullness*/ | |
490 if (v->hrd_fullness || num != v->hrd_num_leaky_buckets) | |
491 { | |
492 av_freep(&v->hrd_buffer); | |
493 } | |
494 if (!v->hrd_fullness) v->hrd_fullness = av_malloc(num*sizeof(uint8_t)); | |
495 if (!v->hrd_fullness) | |
496 { | |
497 av_freep(&v->hrd_rate); | |
498 av_freep(&v->hrd_buffer); | |
499 return -1; | |
500 } | |
501 v->hrd_num_leaky_buckets = num; | |
502 | |
503 //exponent in base-2 for rate | |
504 v->bit_rate_exponent = 6 + get_bits(gb, 4); | |
505 //exponent in base-2 for buffer_size | |
506 v->buffer_size_exponent = 4 + get_bits(gb, 4); | |
507 | |
508 for (i=0; i<num; i++) | |
509 { | |
510 //mantissae, ordered (if not, use a function ? | |
511 v->hrd_rate[i] = 1 + get_bits(gb, 16); | |
512 if (i && v->hrd_rate[i-1]>=v->hrd_rate[i]) | |
513 { | |
514 av_log(v->s.avctx, AV_LOG_ERROR, "HDR Rates aren't strictly increasing:" | |
515 "%i vs %i\n", v->hrd_rate[i-1], v->hrd_rate[i]); | |
516 return -1; | |
517 } | |
518 v->hrd_buffer[i] = 1 + get_bits(gb, 16); | |
519 if (i && v->hrd_buffer[i-1]<v->hrd_buffer[i]) | |
520 { | |
521 av_log(v->s.avctx, AV_LOG_ERROR, "HDR Buffers aren't decreasing:" | |
522 "%i vs %i\n", v->hrd_buffer[i-1], v->hrd_buffer[i]); | |
523 return -1; | |
524 } | |
525 } | |
526 return 0; | |
527 } | |
528 | |
529 /** | |
530 * Decode sequence header for Advanced Profile | |
531 * @see Table 2, p18 | |
532 * @see 6.1.7, pp21-27 | |
533 * @param v The VC1Context to initialize | |
534 * @param gb A GetBitContext initialized from AVCodecContext extra_data | |
535 * @return Status | |
536 */ | |
537 static int decode_advanced_sequence_header(AVCodecContext *avctx, GetBitContext *gb) | |
538 { | |
539 VC1Context *v = avctx->priv_data; | |
540 int nr, dr, aspect_ratio; | |
541 | |
542 v->postprocflag = get_bits(gb, 1); | |
543 v->broadcast = get_bits(gb, 1); | |
544 v->interlace = get_bits(gb, 1); | |
545 | |
546 v->tfcntrflag = get_bits(gb, 1); | |
547 v->finterpflag = get_bits(gb, 1); //common | |
548 v->panscanflag = get_bits(gb, 1); | |
549 v->reserved = get_bits(gb, 1); | |
550 if (v->reserved) | |
551 { | |
552 av_log(avctx, AV_LOG_ERROR, "RESERVED should be 0 (is %i)\n", | |
553 v->reserved); | |
554 return -1; | |
555 } | |
556 if (v->extended_mv) | |
557 v->extended_dmv = get_bits(gb, 1); | |
558 | |
559 /* 6.1.7, p21 */ | |
560 if (get_bits(gb, 1) /* pic_size_flag */) | |
561 { | |
562 avctx->coded_width = get_bits(gb, 12) << 1; | |
563 avctx->coded_height = get_bits(gb, 12) << 1; | |
564 if ( get_bits(gb, 1) /* disp_size_flag */) | |
565 { | |
566 avctx->width = get_bits(gb, 14); | |
567 avctx->height = get_bits(gb, 14); | |
568 } | |
569 | |
570 /* 6.1.7.4, p23 */ | |
571 if ( get_bits(gb, 1) /* aspect_ratio_flag */) | |
572 { | |
573 aspect_ratio = get_bits(gb, 4); //SAR | |
574 if (aspect_ratio == 0x0F) //FF_ASPECT_EXTENDED | |
575 { | |
576 avctx->sample_aspect_ratio.num = 1 + get_bits(gb, 8); | |
577 avctx->sample_aspect_ratio.den = 1 + get_bits(gb, 8); | |
578 } | |
579 else if (aspect_ratio == 0x0E) | |
580 { | |
581 av_log(avctx, AV_LOG_DEBUG, "Reserved AR found\n"); | |
582 } | |
583 else | |
584 { | |
585 avctx->sample_aspect_ratio = vc1_pixel_aspect[aspect_ratio]; | |
586 } | |
587 } | |
588 } | |
589 else | |
590 { | |
591 avctx->coded_width = avctx->width; | |
592 avctx->coded_height = avctx->height; | |
593 } | |
594 | |
595 /* 6.1.8, p23 */ | |
596 if ( get_bits(gb, 1) /* framerateflag */) | |
597 { | |
598 if ( !get_bits(gb, 1) /* framerateind */) | |
599 { | |
600 nr = get_bits(gb, 8); | |
601 dr = get_bits(gb, 4); | |
602 if (nr<1) | |
603 { | |
604 av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATENR\n"); | |
605 return -1; | |
606 } | |
607 if (nr>5) | |
608 { | |
609 av_log(avctx, AV_LOG_ERROR, | |
610 "Reserved FRAMERATENR %i not handled\n", nr); | |
611 nr = 5; /* overflow protection */ | |
612 } | |
613 if (dr<1) | |
614 { | |
615 av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATEDR\n"); | |
616 return -1; | |
617 } | |
618 if (dr>2) | |
619 { | |
620 av_log(avctx, AV_LOG_ERROR, | |
621 "Reserved FRAMERATEDR %i not handled\n", dr); | |
622 dr = 2; /* overflow protection */ | |
623 } | |
624 avctx->time_base.num = fps_nr[dr - 1]; | |
625 avctx->time_base.den = fps_nr[nr - 1]; | |
626 } | |
627 else | |
628 { | |
629 nr = get_bits(gb, 16); | |
630 // 0.03125->2048Hz / 0.03125Hz | |
631 avctx->time_base.den = 1000000; | |
632 avctx->time_base.num = 31250*(1+nr); | |
633 } | |
634 } | |
635 | |
636 /* 6.1.9, p25 */ | |
637 if ( get_bits(gb, 1) /* color_format_flag */) | |
638 { | |
639 //Chromacity coordinates of color primaries | |
640 //like ITU-R BT.709-2, BT.470-2, ... | |
641 v->color_prim = get_bits(gb, 8); | |
642 if (v->color_prim<1) | |
643 { | |
644 av_log(avctx, AV_LOG_ERROR, "0 for COLOR_PRIM is forbidden\n"); | |
645 return -1; | |
646 } | |
647 if (v->color_prim == 3 || v->color_prim>6) | |
648 { | |
649 av_log(avctx, AV_LOG_DEBUG, "Reserved COLOR_PRIM %i found\n", | |
650 v->color_prim); | |
651 return -1; | |
652 } | |
653 | |
654 //Opto-electronic transfer characteristics | |
655 v->transfer_char = get_bits(gb, 8); | |
656 if (v->transfer_char < 1) | |
657 { | |
658 av_log(avctx, AV_LOG_ERROR, "0 for TRAMSFER_CHAR is forbidden\n"); | |
659 return -1; | |
660 } | |
661 if (v->transfer_char == 3 || v->transfer_char>8) | |
662 { | |
663 av_log(avctx, AV_LOG_DEBUG, "Reserved TRANSFERT_CHAR %i found\n", | |
664 v->color_prim); | |
665 return -1; | |
666 } | |
667 | |
668 //Matrix coefficient for primariev->YCbCr | |
669 v->matrix_coef = get_bits(gb, 8); | |
670 if (v->matrix_coef < 1) | |
671 { | |
672 av_log(avctx, AV_LOG_ERROR, "0 for MATRIX_COEF is forbidden\n"); | |
673 return -1; | |
674 } | |
675 if ((v->matrix_coef > 2 && v->matrix_coef < 6) || v->matrix_coef > 7) | |
676 { | |
677 av_log(avctx, AV_LOG_DEBUG, "Reserved MATRIX_COEF %i found\n", | |
678 v->color_prim); | |
679 return -1; | |
680 } | |
681 } | |
682 | |
683 //Hypothetical reference decoder indicator flag | |
684 v->hrd_param_flag = get_bits(gb, 1); | |
685 if (v->hrd_param_flag) | |
686 { | |
687 if (decode_hrd(v, gb) < 0) return -1; | |
688 } | |
689 | |
690 /*reset scaling ranges, 6.2.2 & 6.2.3, p33*/ | |
691 v->range_mapy_flag = 0; | |
692 v->range_mapuv_flag = 0; | |
693 | |
694 av_log(avctx, AV_LOG_DEBUG, "Advanced profile not supported yet\n"); | |
695 return -1; | |
696 } | |
697 #endif | |
698 | |
699 /** | |
700 * Decode Simple/Main Profiles sequence header | |
701 * @see Figure 7-8, p16-17 | |
702 * @param avctx Codec context | |
703 * @param gb GetBit context initialized from Codec context extra_data | |
704 * @return Status | |
705 */ | |
706 static int decode_sequence_header(AVCodecContext *avctx, GetBitContext *gb) | |
707 { | |
708 VC1Context *v = avctx->priv_data; | |
709 | |
710 av_log(avctx, AV_LOG_DEBUG, "Header: %0X\n", show_bits(gb, 32)); | |
711 v->profile = get_bits(gb, 2); | |
712 if (v->profile == 2) | |
713 { | |
714 av_log(avctx, AV_LOG_ERROR, "Profile value 2 is forbidden\n"); | |
715 return -1; | |
716 } | |
717 | |
718 #if HAS_ADVANCED_PROFILE | |
719 if (v->profile == PROFILE_ADVANCED) | |
720 { | |
721 v->level = get_bits(gb, 3); | |
722 if(v->level >= 5) | |
723 { | |
724 av_log(avctx, AV_LOG_ERROR, "Reserved LEVEL %i\n",v->level); | |
725 } | |
726 v->chromaformat = get_bits(gb, 2); | |
727 if (v->chromaformat != 1) | |
728 { | |
729 av_log(avctx, AV_LOG_ERROR, | |
730 "Only 4:2:0 chroma format supported\n"); | |
731 return -1; | |
732 } | |
733 } | |
734 else | |
735 #endif | |
736 { | |
737 v->res_sm = get_bits(gb, 2); //reserved | |
738 if (v->res_sm) | |
739 { | |
740 av_log(avctx, AV_LOG_ERROR, | |
741 "Reserved RES_SM=%i is forbidden\n", v->res_sm); | |
742 return -1; | |
743 } | |
744 } | |
745 | |
746 // (fps-2)/4 (->30) | |
747 v->frmrtq_postproc = get_bits(gb, 3); //common | |
748 // (bitrate-32kbps)/64kbps | |
749 v->bitrtq_postproc = get_bits(gb, 5); //common | |
750 v->s.loop_filter = get_bits(gb, 1); //common | |
751 if(v->s.loop_filter == 1 && v->profile == PROFILE_SIMPLE) | |
752 { | |
753 av_log(avctx, AV_LOG_ERROR, | |
754 "LOOPFILTER shell not be enabled in simple profile\n"); | |
755 } | |
756 | |
757 #if HAS_ADVANCED_PROFILE | |
758 if (v->profile < PROFILE_ADVANCED) | |
759 #endif | |
760 { | |
761 v->res_x8 = get_bits(gb, 1); //reserved | |
762 if (v->res_x8) | |
763 { | |
764 av_log(avctx, AV_LOG_ERROR, | |
765 "1 for reserved RES_X8 is forbidden\n"); | |
766 //return -1; | |
767 } | |
768 v->multires = get_bits(gb, 1); | |
769 v->res_fasttx = get_bits(gb, 1); | |
770 if (!v->res_fasttx) | |
771 { | |
772 av_log(avctx, AV_LOG_ERROR, | |
773 "0 for reserved RES_FASTTX is forbidden\n"); | |
774 //return -1; | |
775 } | |
776 } | |
777 | |
778 v->fastuvmc = get_bits(gb, 1); //common | |
779 if (!v->profile && !v->fastuvmc) | |
780 { | |
781 av_log(avctx, AV_LOG_ERROR, | |
782 "FASTUVMC unavailable in Simple Profile\n"); | |
783 return -1; | |
784 } | |
785 v->extended_mv = get_bits(gb, 1); //common | |
786 if (!v->profile && v->extended_mv) | |
787 { | |
788 av_log(avctx, AV_LOG_ERROR, | |
789 "Extended MVs unavailable in Simple Profile\n"); | |
790 return -1; | |
791 } | |
792 v->dquant = get_bits(gb, 2); //common | |
793 v->vstransform = get_bits(gb, 1); //common | |
794 | |
795 #if HAS_ADVANCED_PROFILE | |
796 if (v->profile < PROFILE_ADVANCED) | |
797 #endif | |
798 { | |
799 v->res_transtab = get_bits(gb, 1); | |
800 if (v->res_transtab) | |
801 { | |
802 av_log(avctx, AV_LOG_ERROR, | |
803 "1 for reserved RES_TRANSTAB is forbidden\n"); | |
804 return -1; | |
805 } | |
806 } | |
807 | |
808 v->overlap = get_bits(gb, 1); //common | |
809 | |
810 #if HAS_ADVANCED_PROFILE | |
811 if (v->profile < PROFILE_ADVANCED) | |
812 #endif | |
813 { | |
814 v->s.resync_marker = get_bits(gb, 1); | |
815 v->rangered = get_bits(gb, 1); | |
816 if (v->rangered && v->profile == PROFILE_SIMPLE) | |
817 { | |
818 av_log(avctx, AV_LOG_DEBUG, | |
819 "RANGERED should be set to 0 in simple profile\n"); | |
820 } | |
821 } | |
822 | |
823 v->s.max_b_frames = avctx->max_b_frames = get_bits(gb, 3); //common | |
824 v->quantizer_mode = get_bits(gb, 2); //common | |
825 | |
826 #if HAS_ADVANCED_PROFILE | |
827 if (v->profile < PROFILE_ADVANCED) | |
828 #endif | |
829 { | |
830 v->finterpflag = get_bits(gb, 1); //common | |
831 v->res_rtm_flag = get_bits(gb, 1); //reserved | |
832 if (!v->res_rtm_flag) | |
833 { | |
834 av_log(avctx, AV_LOG_ERROR, | |
835 "0 for reserved RES_RTM_FLAG is forbidden\n"); | |
836 //return -1; | |
837 } | |
838 #if TRACE | |
839 av_log(avctx, AV_LOG_INFO, | |
840 "Profile %i:\nfrmrtq_postproc=%i, bitrtq_postproc=%i\n" | |
841 "LoopFilter=%i, MultiRes=%i, FastUVMV=%i, Extended MV=%i\n" | |
842 "Rangered=%i, VSTransform=%i, Overlap=%i, SyncMarker=%i\n" | |
843 "DQuant=%i, Quantizer mode=%i, Max B frames=%i\n", | |
844 v->profile, v->frmrtq_postproc, v->bitrtq_postproc, | |
845 v->s.loop_filter, v->multires, v->fastuvmc, v->extended_mv, | |
846 v->rangered, v->vstransform, v->overlap, v->s.resync_marker, | |
847 v->dquant, v->quantizer_mode, avctx->max_b_frames | |
848 ); | |
849 return 0; | |
850 #endif | |
851 } | |
852 #if HAS_ADVANCED_PROFILE | |
853 else return decode_advanced_sequence_header(avctx, gb); | |
854 #endif | |
855 } | |
856 | |
857 | |
858 #if HAS_ADVANCED_PROFILE | |
859 /** Entry point decoding (Advanced Profile) | |
860 * @param avctx Codec context | |
861 * @param gb GetBit context initialized from avctx->extra_data | |
862 * @return Status | |
863 */ | |
864 static int advanced_entry_point_process(AVCodecContext *avctx, GetBitContext *gb) | |
865 { | |
866 VC1Context *v = avctx->priv_data; | |
867 int i; | |
868 if (v->profile != PROFILE_ADVANCED) | |
869 { | |
870 av_log(avctx, AV_LOG_ERROR, | |
871 "Entry point are only defined in Advanced Profile!\n"); | |
872 return -1; //Only for advanced profile! | |
873 } | |
874 if (v->hrd_param_flag) | |
875 { | |
876 //Update buffer fullness | |
877 av_log(avctx, AV_LOG_DEBUG, "Buffer fullness update\n"); | |
878 assert(v->hrd_num_leaky_buckets > 0); | |
879 for (i=0; i<v->hrd_num_leaky_buckets; i++) | |
880 v->hrd_fullness[i] = get_bits(gb, 8); | |
881 } | |
882 if ((v->range_mapy_flag = get_bits(gb, 1))) | |
883 { | |
884 //RANGE_MAPY | |
885 av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPY\n"); | |
886 v->range_mapy = get_bits(gb, 3); | |
887 } | |
888 if ((v->range_mapuv_flag = get_bits(gb, 1))) | |
889 { | |
890 //RANGE_MAPUV | |
891 av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPUV\n"); | |
892 v->range_mapuv = get_bits(gb, 3); | |
893 } | |
894 if (v->panscanflag) | |
895 { | |
896 //NUMPANSCANWIN | |
897 v->numpanscanwin = get_bits(gb, 3); | |
898 av_log(avctx, AV_LOG_DEBUG, "NUMPANSCANWIN: %u\n", v->numpanscanwin); | |
899 } | |
900 return 0; | |
901 } | |
902 #endif | |
903 | |
904 /***********************************************************************/ | |
905 /** | |
906 * @defgroup bitplane VC1 Bitplane decoding | |
907 * @see 8.7, p56 | |
908 * @{ | |
909 */ | |
910 | |
911 /** @addtogroup bitplane | |
912 * Imode types | |
913 * @{ | |
914 */ | |
915 #define IMODE_RAW 0 | |
916 #define IMODE_NORM2 1 | |
917 #define IMODE_DIFF2 2 | |
918 #define IMODE_NORM6 3 | |
919 #define IMODE_DIFF6 4 | |
920 #define IMODE_ROWSKIP 5 | |
921 #define IMODE_COLSKIP 6 | |
922 /** @} */ //imode defines | |
923 | |
924 /** Allocate the buffer from a bitplane, given its dimensions | |
925 * @param bp Bitplane which buffer is to allocate | |
926 * @param[in] width Width of the buffer | |
927 * @param[in] height Height of the buffer | |
928 * @return Status | |
929 * @todo TODO: Take into account stride | |
930 * @todo TODO: Allow use of external buffers ? | |
931 */ | |
932 static int alloc_bitplane(BitPlane *bp, int width, int height) | |
933 { | |
934 if (!bp || bp->width<0 || bp->height<0) return -1; | |
935 bp->data = (uint8_t*)av_malloc(width*height); | |
936 if (!bp->data) return -1; | |
937 bp->width = bp->stride = width; | |
938 bp->height = height; | |
939 return 0; | |
940 } | |
941 | |
942 /** Free the bitplane's buffer | |
943 * @param bp Bitplane which buffer is to free | |
944 */ | |
945 static void free_bitplane(BitPlane *bp) | |
946 { | |
947 bp->width = bp->stride = bp->height = 0; | |
948 if (bp->data) av_freep(&bp->data); | |
949 } | |
950 | |
951 /** Decode rows by checking if they are skipped | |
952 * @param plane Buffer to store decoded bits | |
953 * @param[in] width Width of this buffer | |
954 * @param[in] height Height of this buffer | |
955 * @param[in] stride of this buffer | |
956 */ | |
957 static void decode_rowskip(uint8_t* plane, int width, int height, int stride, GetBitContext *gb){ | |
958 int x, y; | |
959 | |
960 for (y=0; y<height; y++){ | |
961 if (!get_bits(gb, 1)) //rowskip | |
962 memset(plane, 0, width); | |
963 else | |
964 for (x=0; x<width; x++) | |
965 plane[x] = get_bits(gb, 1); | |
966 plane += stride; | |
967 } | |
968 } | |
969 | |
970 /** Decode columns by checking if they are skipped | |
971 * @param plane Buffer to store decoded bits | |
972 * @param[in] width Width of this buffer | |
973 * @param[in] height Height of this buffer | |
974 * @param[in] stride of this buffer | |
975 * @fixme FIXME: Optimize | |
976 */ | |
977 static void decode_colskip(uint8_t* plane, int width, int height, int stride, GetBitContext *gb){ | |
978 int x, y; | |
979 | |
980 for (x=0; x<width; x++){ | |
981 if (!get_bits(gb, 1)) //colskip | |
982 for (y=0; y<height; y++) | |
983 plane[y*stride] = 0; | |
984 else | |
985 for (y=0; y<height; y++) | |
986 plane[y*stride] = get_bits(gb, 1); | |
987 plane ++; | |
988 } | |
989 } | |
990 | |
991 /** Decode a bitplane's bits | |
992 * @param bp Bitplane where to store the decode bits | |
993 * @param v VC1 context for bit reading and logging | |
994 * @return Status | |
995 * @fixme FIXME: Optimize | |
996 * @todo TODO: Decide if a struct is needed | |
997 */ | |
998 static int bitplane_decoding(BitPlane *bp, VC1Context *v) | |
999 { | |
1000 GetBitContext *gb = &v->s.gb; | |
1001 | |
1002 int imode, x, y, code, use_vertical_tile, tile_w, tile_h, offset; | |
1003 uint8_t invert, *planep = bp->data; | |
1004 | |
1005 invert = get_bits(gb, 1); | |
1006 imode = get_vlc2(gb, vc1_imode_vlc.table, VC1_IMODE_VLC_BITS, 2); | |
1007 | |
1008 bp->is_raw = 0; | |
1009 switch (imode) | |
1010 { | |
1011 case IMODE_RAW: | |
1012 //Data is actually read in the MB layer (same for all tests == "raw") | |
1013 bp->is_raw = 1; //invert ignored | |
1014 return invert; | |
1015 case IMODE_DIFF2: | |
1016 case IMODE_NORM2: | |
1017 if ((bp->height*bp->width) & 1) | |
1018 { | |
1019 *(++planep) = get_bits(gb, 1); | |
1020 offset = x = 1; | |
1021 } | |
1022 else offset = x = 0; | |
1023 | |
1024 for (y=0; y<bp->height; y++) | |
1025 { | |
1026 for(; x<bp->width; x+=2) | |
1027 { | |
1028 code = get_vlc2(gb, vc1_norm2_vlc.table, VC1_NORM2_VLC_BITS, 2); | |
1029 *(++planep) = code&1; //lsb => left | |
1030 *(++planep) = (code>>1)&1; //msb => right | |
1031 } | |
1032 planep += bp->stride-bp->width; | |
1033 if ((bp->width-offset)&1) //Odd number previously processed | |
1034 { | |
1035 code = get_vlc2(gb, vc1_norm2_vlc.table, VC1_NORM2_VLC_BITS, 2); | |
1036 *planep = code&1; | |
1037 planep += bp->stride-bp->width; | |
1038 *planep = (code>>1)&1; //msb => right | |
1039 offset = x = 1; | |
1040 } | |
1041 else | |
1042 { | |
1043 offset = x = 0; | |
1044 planep += bp->stride-bp->width; | |
1045 } | |
1046 } | |
1047 break; | |
1048 case IMODE_DIFF6: | |
1049 case IMODE_NORM6: | |
1050 use_vertical_tile= bp->height%3==0 && bp->width%3!=0; | |
1051 tile_w= use_vertical_tile ? 2 : 3; | |
1052 tile_h= use_vertical_tile ? 3 : 2; | |
1053 | |
1054 for(y= bp->height%tile_h; y< bp->height; y+=tile_h){ | |
1055 for(x= bp->width%tile_w; x< bp->width; x+=tile_w){ | |
1056 code = get_vlc2(gb, vc1_norm6_vlc.table, VC1_NORM6_VLC_BITS, 2); | |
1057 if(code<0){ | |
1058 av_log(v->s.avctx, AV_LOG_DEBUG, "invalid NORM-6 VLC\n"); | |
1059 return -1; | |
1060 } | |
1061 //FIXME following is a pure guess and probably wrong | |
1062 //FIXME A bitplane (0 | !0), so could the shifts be avoided ? | |
1063 planep[x + 0*bp->stride]= (code>>0)&1; | |
1064 planep[x + 1 + 0*bp->stride]= (code>>1)&1; | |
1065 //FIXME Does branch prediction help here? | |
1066 if(use_vertical_tile){ | |
1067 planep[x + 0 + 1*bp->stride]= (code>>2)&1; | |
1068 planep[x + 1 + 1*bp->stride]= (code>>3)&1; | |
1069 planep[x + 0 + 2*bp->stride]= (code>>4)&1; | |
1070 planep[x + 1 + 2*bp->stride]= (code>>5)&1; | |
1071 }else{ | |
1072 planep[x + 2 + 0*bp->stride]= (code>>2)&1; | |
1073 planep[x + 0 + 1*bp->stride]= (code>>3)&1; | |
1074 planep[x + 1 + 1*bp->stride]= (code>>4)&1; | |
1075 planep[x + 2 + 1*bp->stride]= (code>>5)&1; | |
1076 } | |
1077 } | |
1078 } | |
1079 | |
1080 x= bp->width % tile_w; | |
1081 decode_colskip(bp->data , x, bp->height , bp->stride, &v->s.gb); | |
1082 decode_rowskip(bp->data+x, bp->width - x, bp->height % tile_h, bp->stride, &v->s.gb); | |
1083 | |
1084 break; | |
1085 case IMODE_ROWSKIP: | |
1086 decode_rowskip(bp->data, bp->width, bp->height, bp->stride, &v->s.gb); | |
1087 break; | |
1088 case IMODE_COLSKIP: | |
1089 decode_colskip(bp->data, bp->width, bp->height, bp->stride, &v->s.gb); | |
1090 break; | |
1091 default: break; | |
1092 } | |
1093 | |
1094 /* Applying diff operator */ | |
1095 if (imode == IMODE_DIFF2 || imode == IMODE_DIFF6) | |
1096 { | |
1097 planep = bp->data; | |
1098 planep[0] ^= invert; | |
1099 for (x=1; x<bp->width; x++) | |
1100 planep[x] ^= planep[x-1]; | |
1101 for (y=1; y<bp->height; y++) | |
1102 { | |
1103 planep += bp->stride; | |
1104 planep[0] ^= planep[-bp->stride]; | |
1105 for (x=1; x<bp->width; x++) | |
1106 { | |
1107 if (planep[x-1] != planep[x-bp->stride]) planep[x] ^= invert; | |
1108 else planep[x] ^= planep[x-1]; | |
1109 } | |
1110 } | |
1111 } | |
1112 else if (invert) | |
1113 { | |
1114 planep = bp->data; | |
1115 for (x=0; x<bp->width*bp->height; x++) planep[x] = !planep[x]; //FIXME stride | |
1116 } | |
1117 return (imode<<1) + invert; | |
1118 } | |
1119 /** @} */ //Bitplane group | |
1120 | |
1121 /***********************************************************************/ | |
1122 /** VOP Dquant decoding | |
1123 * @param v VC1 Context | |
1124 */ | |
1125 static int vop_dquant_decoding(VC1Context *v) | |
1126 { | |
1127 GetBitContext *gb = &v->s.gb; | |
1128 int pqdiff; | |
1129 | |
1130 //variable size | |
1131 if (v->dquant == 2) | |
1132 { | |
1133 pqdiff = get_bits(gb, 3); | |
1134 if (pqdiff == 7) v->altpq = get_bits(gb, 5); | |
1135 else v->altpq = v->pq + pqdiff + 1; | |
1136 } | |
1137 else | |
1138 { | |
1139 v->dquantfrm = get_bits(gb, 1); | |
1140 if ( v->dquantfrm ) | |
1141 { | |
1142 v->dqprofile = get_bits(gb, 2); | |
1143 switch (v->dqprofile) | |
1144 { | |
1145 case DQPROFILE_SINGLE_EDGE: | |
1146 case DQPROFILE_DOUBLE_EDGES: | |
1147 v->dqsbedge = get_bits(gb, 2); | |
1148 break; | |
1149 case DQPROFILE_ALL_MBS: | |
1150 v->dqbilevel = get_bits(gb, 1); | |
1151 default: break; //Forbidden ? | |
1152 } | |
1153 if (!v->dqbilevel || v->dqprofile != DQPROFILE_ALL_MBS) | |
1154 { | |
1155 pqdiff = get_bits(gb, 3); | |
1156 if (pqdiff == 7) v->altpq = get_bits(gb, 5); | |
1157 else v->altpq = v->pq + pqdiff + 1; | |
1158 } | |
1159 } | |
1160 } | |
1161 return 0; | |
1162 } | |
1163 | |
1164 /***********************************************************************/ | |
1165 /** | |
1166 * @defgroup all_frame_hdr All VC1 profiles frame header | |
1167 * @brief Part of the frame header decoding from all profiles | |
1168 * @warning Only pro/epilog differs between Simple/Main and Advanced => check caller | |
1169 * @{ | |
1170 */ | |
1171 /** B and BI frame header decoding, primary part | |
1172 * @see Tables 11+12, p62-65 | |
1173 * @param v VC1 context | |
1174 * @return Status | |
1175 * @warning Also handles BI frames | |
1176 */ | |
1177 static int decode_b_picture_primary_header(VC1Context *v) | |
1178 { | |
1179 GetBitContext *gb = &v->s.gb; | |
1180 int pqindex; | |
1181 | |
1182 /* Prolog common to all frametypes should be done in caller */ | |
1183 if (v->profile == PROFILE_SIMPLE) | |
1184 { | |
1185 av_log(v->s.avctx, AV_LOG_ERROR, "Found a B frame while in Simple Profile!\n"); | |
1186 return FRAME_SKIPPED; | |
1187 } | |
1188 v->bfraction = vc1_bfraction_lut[get_vlc2(gb, vc1_bfraction_vlc.table, | |
1189 VC1_BFRACTION_VLC_BITS, 2)]; | |
1190 if (v->bfraction < -1) | |
1191 { | |
1192 av_log(v->s.avctx, AV_LOG_ERROR, "Invalid BFRaction\n"); | |
1193 return FRAME_SKIPPED; | |
1194 } | |
1195 else if (!v->bfraction) | |
1196 { | |
1197 /* We actually have a BI frame */ | |
1198 v->s.pict_type = BI_TYPE; | |
1199 v->buffer_fullness = get_bits(gb, 7); | |
1200 } | |
1201 | |
1202 /* Read the quantization stuff */ | |
1203 pqindex = get_bits(gb, 5); | |
1204 if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
1205 v->pq = pquant_table[0][pqindex]; | |
1206 else | |
1207 { | |
1208 v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
1209 } | |
1210 if (pqindex < 9) v->halfpq = get_bits(gb, 1); | |
1211 if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
1212 v->pquantizer = get_bits(gb, 1); | |
1213 #if HAS_ADVANCED_PROFILE | |
1214 if (v->profile == PROFILE_ADVANCED) | |
1215 { | |
1216 if (v->postprocflag) v->postproc = get_bits(gb, 2); | |
1217 if (v->extended_mv == 1 && v->s.pict_type != BI_TYPE) | |
1218 v->mvrange = get_prefix(gb, 0, 3); | |
1219 } | |
1220 #endif | |
1221 else | |
1222 { | |
1223 if (v->extended_mv == 1) | |
1224 v->mvrange = get_prefix(gb, 0, 3); | |
1225 } | |
1226 /* Read the MV mode */ | |
1227 if (v->s.pict_type != BI_TYPE) | |
1228 { | |
1229 v->mv_mode = get_bits(gb, 1); | |
1230 if (v->pq < 13) | |
1231 { | |
1232 if (!v->mv_mode) | |
1233 { | |
1234 v->mv_mode = get_bits(gb, 2); | |
1235 if (v->mv_mode) | |
1236 av_log(v->s.avctx, AV_LOG_ERROR, | |
1237 "mv_mode for lowquant B frame was %i\n", v->mv_mode); | |
1238 } | |
1239 } | |
1240 else | |
1241 { | |
1242 if (!v->mv_mode) | |
1243 { | |
1244 if (get_bits(gb, 1)) | |
1245 av_log(v->s.avctx, AV_LOG_ERROR, | |
1246 "mv_mode for highquant B frame was %i\n", v->mv_mode); | |
1247 } | |
1248 v->mv_mode = 1-v->mv_mode; //To match (pq < 13) mapping | |
1249 } | |
1250 } | |
1251 | |
1252 return 0; | |
1253 } | |
1254 | |
1255 /** B and BI frame header decoding, secondary part | |
1256 * @see Tables 11+12, p62-65 | |
1257 * @param v VC1 context | |
1258 * @return Status | |
1259 * @warning Also handles BI frames | |
1260 * @warning To call once all MB arrays are allocated | |
1261 * @todo Support Advanced Profile headers | |
1262 */ | |
1263 static int decode_b_picture_secondary_header(VC1Context *v) | |
1264 { | |
1265 GetBitContext *gb = &v->s.gb; | |
1266 int status; | |
1267 | |
1268 status = bitplane_decoding(&v->skip_mb_plane, v); | |
1269 if (status < 0) return -1; | |
1270 #if TRACE | |
1271 if (v->mv_mode == MV_PMODE_MIXED_MV) | |
1272 { | |
1273 status = bitplane_decoding(&v->mv_type_mb_plane, v); | |
1274 if (status < 0) | |
1275 return -1; | |
1276 #if TRACE | |
1277 av_log(v->s.avctx, AV_LOG_DEBUG, "MB MV Type plane encoding: " | |
1278 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1279 #endif | |
1280 } | |
1281 | |
1282 //bitplane | |
1283 status = bitplane_decoding(&v->direct_mb_plane, v); | |
1284 if (status < 0) return -1; | |
1285 #if TRACE | |
1286 av_log(v->s.avctx, AV_LOG_DEBUG, "MB Direct plane encoding: " | |
1287 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1288 #endif | |
1289 | |
1290 av_log(v->s.avctx, AV_LOG_DEBUG, "Skip MB plane encoding: " | |
1291 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1292 #endif | |
1293 | |
1294 /* FIXME: what is actually chosen for B frames ? */ | |
1295 v->s.mv_table_index = get_bits(gb, 2); //but using vc1_ tables | |
1296 v->cbpcy_vlc = &vc1_cbpcy_p_vlc[get_bits(gb, 2)]; | |
1297 | |
1298 if (v->dquant) | |
1299 { | |
1300 vop_dquant_decoding(v); | |
1301 } | |
1302 | |
1303 if (v->vstransform) | |
1304 { | |
1305 v->ttmbf = get_bits(gb, 1); | |
1306 if (v->ttmbf) | |
1307 { | |
1308 v->ttfrm = get_bits(gb, 2); | |
1309 av_log(v->s.avctx, AV_LOG_INFO, "Transform used: %ix%i\n", | |
1310 (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); | |
1311 } | |
1312 } | |
1313 /* Epilog (AC/DC syntax) should be done in caller */ | |
1314 return 0; | |
1315 } | |
1316 | |
1317 /** I frame header decoding, primary part | |
1318 * @see Tables 5+7, p53-54 and 55-57 | |
1319 * @param v VC1 context | |
1320 * @return Status | |
1321 * @todo Support Advanced Profile headers | |
1322 */ | |
1323 static int decode_i_picture_primary_header(VC1Context *v) | |
1324 { | |
1325 GetBitContext *gb = &v->s.gb; | |
1326 int pqindex; | |
1327 | |
1328 /* Prolog common to all frametypes should be done in caller */ | |
1329 //BF = Buffer Fullness | |
1330 if (v->profile < PROFILE_ADVANCED && get_bits(gb, 7)) | |
1331 { | |
1332 av_log(v->s.avctx, AV_LOG_DEBUG, "I BufferFullness not 0\n"); | |
1333 } | |
1334 | |
1335 /* Quantizer stuff */ | |
1336 pqindex = get_bits(gb, 5); | |
1337 if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
1338 v->pq = pquant_table[0][pqindex]; | |
1339 else | |
1340 { | |
1341 v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
1342 } | |
1343 if (pqindex < 9) v->halfpq = get_bits(gb, 1); | |
1344 if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
1345 v->pquantizer = get_bits(gb, 1); | |
1346 av_log(v->s.avctx, AV_LOG_DEBUG, "I frame: QP=%i (+%i/2)\n", | |
1347 v->pq, v->halfpq); | |
1348 return 0; | |
1349 } | |
1350 | |
1351 /** I frame header decoding, secondary part | |
1352 * @param v VC1 context | |
1353 * @return Status | |
1354 * @warning Not called in A/S/C profiles, it seems | |
1355 * @todo Support Advanced Profile headers | |
1356 */ | |
1357 static int decode_i_picture_secondary_header(VC1Context *v) | |
1358 { | |
1359 #if HAS_ADVANCED_PROFILE | |
1360 int status; | |
1361 if (v->profile == PROFILE_ADVANCED) | |
1362 { | |
1363 v->s.ac_pred = get_bits(&v->s.gb, 1); | |
1364 if (v->postprocflag) v->postproc = get_bits(&v->s.gb, 1); | |
1365 /* 7.1.1.34 + 8.5.2 */ | |
1366 if (v->overlap && v->pq<9) | |
1367 { | |
1368 v->condover = get_bits(&v->s.gb, 1); | |
1369 if (v->condover) | |
1370 { | |
1371 v->condover = 2+get_bits(&v->s.gb, 1); | |
1372 if (v->condover == 3) | |
1373 { | |
1374 status = bitplane_decoding(&v->over_flags_plane, v); | |
1375 if (status < 0) return -1; | |
1376 # if TRACE | |
1377 av_log(v->s.avctx, AV_LOG_DEBUG, "Overflags plane encoding: " | |
1378 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1379 # endif | |
1380 } | |
1381 } | |
1382 } | |
1383 } | |
1384 #endif | |
1385 | |
1386 /* Epilog (AC/DC syntax) should be done in caller */ | |
1387 return 0; | |
1388 } | |
1389 | |
1390 /** P frame header decoding, primary part | |
1391 * @see Tables 5+7, p53-54 and 55-57 | |
1392 * @param v VC1 context | |
1393 * @todo Support Advanced Profile headers | |
1394 * @return Status | |
1395 */ | |
1396 static int decode_p_picture_primary_header(VC1Context *v) | |
1397 { | |
1398 /* INTERFRM, FRMCNT, RANGEREDFRM read in caller */ | |
1399 GetBitContext *gb = &v->s.gb; | |
1400 int lowquant, pqindex; | |
1401 | |
1402 pqindex = get_bits(gb, 5); | |
1403 if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
1404 v->pq = pquant_table[0][pqindex]; | |
1405 else | |
1406 { | |
1407 v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
1408 } | |
1409 if (pqindex < 9) v->halfpq = get_bits(gb, 1); | |
1410 if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
1411 v->pquantizer = get_bits(gb, 1); | |
1412 av_log(v->s.avctx, AV_LOG_DEBUG, "P Frame: QP=%i (+%i/2)\n", | |
1413 v->pq, v->halfpq); | |
1414 if (v->extended_mv == 1) v->mvrange = get_prefix(gb, 0, 3); | |
1415 #if HAS_ADVANCED_PROFILE | |
1416 if (v->profile == PROFILE_ADVANCED) | |
1417 { | |
1418 if (v->postprocflag) v->postproc = get_bits(gb, 1); | |
1419 } | |
1420 else | |
1421 #endif | |
1422 if (v->multires) v->respic = get_bits(gb, 2); | |
1423 lowquant = (v->pquantizer>12) ? 0 : 1; | |
1424 v->mv_mode = mv_pmode_table[lowquant][get_prefix(gb, 1, 4)]; | |
1425 if (v->mv_mode == MV_PMODE_INTENSITY_COMP) | |
1426 { | |
1427 v->mv_mode2 = mv_pmode_table[lowquant][get_prefix(gb, 1, 3)]; | |
1428 v->lumscale = get_bits(gb, 6); | |
1429 v->lumshift = get_bits(gb, 6); | |
1430 } | |
1431 return 0; | |
1432 } | |
1433 | |
1434 /** P frame header decoding, secondary part | |
1435 * @see Tables 5+7, p53-54 and 55-57 | |
1436 * @param v VC1 context | |
1437 * @warning To call once all MB arrays are allocated | |
1438 * @return Status | |
1439 */ | |
1440 static int decode_p_picture_secondary_header(VC1Context *v) | |
1441 { | |
1442 GetBitContext *gb = &v->s.gb; | |
1443 int status = 0; | |
1444 if ((v->mv_mode == MV_PMODE_INTENSITY_COMP && | |
1445 v->mv_mode2 == MV_PMODE_MIXED_MV) | |
1446 || v->mv_mode == MV_PMODE_MIXED_MV) | |
1447 { | |
1448 status = bitplane_decoding(&v->mv_type_mb_plane, v); | |
1449 if (status < 0) return -1; | |
1450 #if TRACE | |
1451 av_log(v->s.avctx, AV_LOG_DEBUG, "MB MV Type plane encoding: " | |
1452 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1453 #endif | |
1454 } | |
1455 | |
1456 status = bitplane_decoding(&v->skip_mb_plane, v); | |
1457 if (status < 0) return -1; | |
1458 #if TRACE | |
1459 av_log(v->s.avctx, AV_LOG_DEBUG, "MB Skip plane encoding: " | |
1460 "Imode: %i, Invert: %i\n", status>>1, status&1); | |
1461 #endif | |
1462 | |
1463 /* Hopefully this is correct for P frames */ | |
1464 v->s.mv_table_index =get_bits(gb, 2); //but using vc1_ tables | |
1465 v->cbpcy_vlc = &vc1_cbpcy_p_vlc[get_bits(gb, 2)]; | |
1466 | |
1467 if (v->dquant) | |
1468 { | |
1469 av_log(v->s.avctx, AV_LOG_INFO, "VOP DQuant info\n"); | |
1470 vop_dquant_decoding(v); | |
1471 } | |
1472 | |
1473 v->ttfrm = 0; //FIXME Is that so ? | |
1474 if (v->vstransform) | |
1475 { | |
1476 v->ttmbf = get_bits(gb, 1); | |
1477 if (v->ttmbf) | |
1478 { | |
1479 v->ttfrm = get_bits(gb, 2); | |
1480 av_log(v->s.avctx, AV_LOG_INFO, "Transform used: %ix%i\n", | |
1481 (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); | |
1482 } | |
1483 } | |
1484 /* Epilog (AC/DC syntax) should be done in caller */ | |
1485 return 0; | |
1486 } | |
1487 /** @} */ //End of group all_frm_hdr | |
1488 | |
1489 | |
1490 /***********************************************************************/ | |
1491 /** | |
1492 * @defgroup std_frame_hdr VC1 Simple/Main Profiles header decoding | |
1493 * @brief Part of the frame header decoding belonging to Simple/Main Profiles | |
1494 * @warning Only pro/epilog differs between Simple/Main and Advanced => | |
1495 * check caller | |
1496 * @{ | |
1497 */ | |
1498 | |
1499 /** Frame header decoding, first part, in Simple and Main profiles | |
1500 * @see Tables 5+7, p53-54 and 55-57 | |
1501 * @param v VC1 context | |
1502 * @todo FIXME: RANGEREDFRM element not read if BI frame from Table6, P54 | |
1503 * However, 7.1.1.8 says "all frame types, for main profiles" | |
1504 * @return Status | |
1505 */ | |
1506 static int standard_decode_picture_primary_header(VC1Context *v) | |
1507 { | |
1508 GetBitContext *gb = &v->s.gb; | |
1509 int status = 0; | |
1510 | |
1511 if (v->finterpflag) v->interpfrm = get_bits(gb, 1); | |
1512 skip_bits(gb, 2); //framecnt unused | |
1513 if (v->rangered) v->rangeredfrm = get_bits(gb, 1); | |
1514 v->s.pict_type = get_bits(gb, 1); | |
1515 if (v->s.avctx->max_b_frames) | |
1516 { | |
1517 if (!v->s.pict_type) | |
1518 { | |
1519 if (get_bits(gb, 1)) v->s.pict_type = I_TYPE; | |
1520 else v->s.pict_type = B_TYPE; | |
1521 } | |
1522 else v->s.pict_type = P_TYPE; | |
1523 } | |
1524 else v->s.pict_type++; | |
1525 | |
1526 switch (v->s.pict_type) | |
1527 { | |
1528 case I_TYPE: status = decode_i_picture_primary_header(v); break; | |
1529 case P_TYPE: status = decode_p_picture_primary_header(v); break; | |
1530 case BI_TYPE: //Same as B | |
1531 case B_TYPE: status = decode_b_picture_primary_header(v); break; | |
1532 } | |
1533 | |
1534 if (status == FRAME_SKIPPED) | |
1535 { | |
1536 av_log(v->s.avctx, AV_LOG_INFO, "Skipping frame...\n"); | |
1537 return status; | |
1538 } | |
1539 return 0; | |
1540 } | |
1541 | |
1542 /** Frame header decoding, secondary part | |
1543 * @param v VC1 context | |
1544 * @warning To call once all MB arrays are allocated | |
1545 * @return Status | |
1546 */ | |
1547 static int standard_decode_picture_secondary_header(VC1Context *v) | |
1548 { | |
1549 GetBitContext *gb = &v->s.gb; | |
1550 int status = 0; | |
1551 | |
1552 switch (v->s.pict_type) | |
1553 { | |
1554 case P_TYPE: status = decode_p_picture_secondary_header(v); break; | |
1555 case B_TYPE: status = decode_b_picture_secondary_header(v); break; | |
1556 case BI_TYPE: | |
1557 case I_TYPE: break; //Nothing needed as it's done in the epilog | |
1558 } | |
1559 if (status < 0) return FRAME_SKIPPED; | |
1560 | |
1561 /* AC Syntax */ | |
1562 v->c_ac_table_index = decode012(gb); | |
1563 if (v->s.pict_type == I_TYPE || v->s.pict_type == BI_TYPE) | |
1564 { | |
1565 v->y_ac_table_index = decode012(gb); | |
1566 } | |
1567 /* DC Syntax */ | |
1568 v->s.dc_table_index = decode012(gb); | |
1569 | |
1570 return 0; | |
1571 } | |
1572 /** @} */ //End for group std_frame_hdr | |
1573 | |
1574 #if HAS_ADVANCED_PROFILE | |
1575 /***********************************************************************/ | |
1576 /** | |
1577 * @defgroup adv_frame_hdr VC1 Advanced Profile header decoding | |
1578 * @brief Part of the frame header decoding belonging to Advanced Profiles | |
1579 * @warning Only pro/epilog differs between Simple/Main and Advanced => | |
1580 * check caller | |
1581 * @{ | |
1582 */ | |
1583 /** Frame header decoding, primary part | |
1584 * @param v VC1 context | |
1585 * @return Status | |
1586 */ | |
1587 static int advanced_decode_picture_primary_header(VC1Context *v) | |
1588 { | |
1589 GetBitContext *gb = &v->s.gb; | |
1590 static const int type_table[4] = { P_TYPE, B_TYPE, I_TYPE, BI_TYPE }; | |
1591 int type; | |
1592 | |
1593 if (v->interlace) | |
1594 { | |
1595 v->fcm = get_bits(gb, 1); | |
1596 if (v->fcm) v->fcm = 2+get_bits(gb, 1); | |
1597 } | |
1598 | |
1599 type = get_prefix(gb, 0, 4); | |
1600 if (type > 4 || type < 0) return FRAME_SKIPPED; | |
1601 v->s.pict_type = type_table[type]; | |
1602 av_log(v->s.avctx, AV_LOG_INFO, "AP Frame Type: %i\n", v->s.pict_type); | |
1603 | |
1604 if (v->tfcntrflag) v->tfcntr = get_bits(gb, 8); | |
1605 if (v->broadcast) | |
1606 { | |
1607 if (!v->interlace) v->rptfrm = get_bits(gb, 2); | |
1608 else | |
1609 { | |
1610 v->tff = get_bits(gb, 1); | |
1611 v->rff = get_bits(gb, 1); | |
1612 } | |
1613 } | |
1614 | |
1615 if (v->panscanflag) | |
1616 { | |
1617 #if 0 | |
1618 for (i=0; i<v->numpanscanwin; i++) | |
1619 { | |
1620 v->topleftx[i] = get_bits(gb, 16); | |
1621 v->toplefty[i] = get_bits(gb, 16); | |
1622 v->bottomrightx[i] = get_bits(gb, 16); | |
1623 v->bottomrighty[i] = get_bits(gb, 16); | |
1624 } | |
1625 #else | |
1626 skip_bits(gb, 16*4*v->numpanscanwin); | |
1627 #endif | |
1628 } | |
1629 v->s.no_rounding = !get_bits(gb, 1); | |
1630 v->uvsamp = get_bits(gb, 1); | |
1631 if (v->finterpflag == 1) v->interpfrm = get_bits(gb, 1); | |
1632 | |
1633 switch(v->s.pict_type) | |
1634 { | |
1635 case I_TYPE: if (decode_i_picture_primary_header(v) < 0) return -1; | |
1636 case P_TYPE: if (decode_p_picture_primary_header(v) < 0) return -1; | |
1637 case BI_TYPE: | |
1638 case B_TYPE: if (decode_b_picture_primary_header(v) < 0) return FRAME_SKIPPED; | |
1639 default: return -1; | |
1640 } | |
1641 } | |
1642 | |
1643 /** Frame header decoding, secondary part | |
1644 * @param v VC1 context | |
1645 * @return Status | |
1646 */ | |
1647 static int advanced_decode_picture_secondary_header(VC1Context *v) | |
1648 { | |
1649 GetBitContext *gb = &v->s.gb; | |
1650 int status = 0; | |
1651 | |
1652 switch(v->s.pict_type) | |
1653 { | |
1654 case P_TYPE: status = decode_p_picture_secondary_header(v); break; | |
1655 case B_TYPE: status = decode_b_picture_secondary_header(v); break; | |
1656 case BI_TYPE: | |
1657 case I_TYPE: status = decode_i_picture_secondary_header(v); break; | |
1658 } | |
1659 if (status<0) return FRAME_SKIPPED; | |
1660 | |
1661 /* AC Syntax */ | |
1662 v->c_ac_table_index = decode012(gb); | |
1663 if (v->s.pict_type == I_TYPE || v->s.pict_type == BI_TYPE) | |
1664 { | |
1665 v->y_ac_table_index = decode012(gb); | |
1666 } | |
1667 /* DC Syntax */ | |
1668 v->s.dc_table_index = decode012(gb); | |
1669 | |
1670 return 0; | |
1671 } | |
1672 #endif | |
1673 /** @} */ //End for adv_frame_hdr | |
1674 | |
1675 /***********************************************************************/ | |
1676 /** | |
1677 * @defgroup block VC1 Block-level functions | |
1678 * @see 7.1.4, p91 and 8.1.1.7, p(1)04 | |
1679 * @todo TODO: Integrate to MpegEncContext facilities | |
1680 * @{ | |
1681 */ | |
1682 | |
1683 /** | |
1684 * @def GET_MQUANT | |
1685 * @brief Get macroblock-level quantizer scale | |
1686 * @warning XXX: qdiff to the frame quant, not previous quant ? | |
1687 * @fixme XXX: Don't know how to initialize mquant otherwise in last case | |
1688 */ | |
1689 #define GET_MQUANT() \ | |
1690 if (v->dquantfrm) \ | |
1691 { \ | |
1692 if (v->dqprofile == DQPROFILE_ALL_MBS) \ | |
1693 { \ | |
1694 if (v->dqbilevel) \ | |
1695 { \ | |
1696 mquant = (get_bits(gb, 1)) ? v->pq : v->altpq; \ | |
1697 } \ | |
1698 else \ | |
1699 { \ | |
1700 mqdiff = get_bits(gb, 3); \ | |
1701 if (mqdiff != 7) mquant = v->pq + mqdiff; \ | |
1702 else mquant = get_bits(gb, 5); \ | |
1703 } \ | |
1704 } \ | |
1705 else mquant = v->pq; \ | |
1706 } | |
1707 | |
1708 /** | |
1709 * @def GET_MVDATA(_dmv_x, _dmv_y) | |
1710 * @brief Get MV differentials | |
1711 * @see MVDATA decoding from 8.3.5.2, p(1)20 | |
1712 * @param _dmv_x Horizontal differential for decoded MV | |
1713 * @param _dmv_y Vertical differential for decoded MV | |
1714 * @todo TODO: Use MpegEncContext arrays to store them | |
1715 */ | |
1716 #define GET_MVDATA(_dmv_x, _dmv_y) \ | |
1717 index = 1 + get_vlc2(gb, vc1_mv_diff_vlc[s->mv_table_index].table,\ | |
1718 VC1_MV_DIFF_VLC_BITS, 2); \ | |
1719 if (index > 36) \ | |
1720 { \ | |
1721 mb_has_coeffs = 1; \ | |
1722 index -= 37; \ | |
1723 } \ | |
1724 else mb_has_coeffs = 0; \ | |
1725 s->mb_intra = 0; \ | |
1726 if (!index) { _dmv_x = _dmv_y = 0; } \ | |
1727 else if (index == 35) \ | |
1728 { \ | |
1729 _dmv_x = get_bits(gb, v->k_x); \ | |
1730 _dmv_y = get_bits(gb, v->k_y); \ | |
1731 s->mb_intra = 1; \ | |
1732 } \ | |
1733 else \ | |
1734 { \ | |
1735 index1 = index%6; \ | |
1736 if (s->mspel && index1 == 5) val = 1; \ | |
1737 else val = 0; \ | |
1738 val = get_bits(gb, size_table[index1] - val); \ | |
1739 sign = 0 - (val&1); \ | |
1740 _dmv_x = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ | |
1741 \ | |
1742 index1 = index/6; \ | |
1743 if (s->mspel && index1 == 5) val = 1; \ | |
1744 else val = 0; \ | |
1745 val = get_bits(gb, size_table[index1] - val); \ | |
1746 sign = 0 - (val&1); \ | |
1747 _dmv_y = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ | |
1748 } | |
1749 | |
1750 /** Get predicted DC value | |
1751 * prediction dir: left=0, top=1 | |
1752 * @param s MpegEncContext | |
1753 * @param[in] n block index in the current MB | |
1754 * @param dc_val_ptr Pointer to DC predictor | |
1755 * @param dir_ptr Prediction direction for use in AC prediction | |
1756 * @todo TODO: Actually do it the VC1 way | |
1757 * @todo TODO: Handle properly edges | |
1758 */ | |
1759 static inline int vc1_pred_dc(MpegEncContext *s, int n, | |
1760 uint16_t **dc_val_ptr, int *dir_ptr) | |
1761 { | |
1762 int a, b, c, wrap, pred, scale; | |
1763 int16_t *dc_val; | |
1764 static const uint16_t dcpred[31] = { | |
1765 1024, 512, 341, 256, 205, 171, 146, 128, | |
1766 114, 102, 93, 85, 79, 73, 68, 64, | |
1767 60, 57, 54, 51, 49, 47, 45, 43, | |
1768 41, 39, 38, 37, 35, 34, 33 | |
1769 }; | |
1770 | |
1771 /* find prediction - wmv3_dc_scale always used here in fact */ | |
1772 if (n < 4) scale = s->y_dc_scale; | |
1773 else scale = s->c_dc_scale; | |
1774 | |
1775 wrap = s->block_wrap[n]; | |
1776 dc_val= s->dc_val[0] + s->block_index[n]; | |
1777 | |
1778 /* B C | |
1779 * A X | |
1780 */ | |
1781 a = dc_val[ - 1]; | |
1782 b = dc_val[ - 1 - wrap]; | |
1783 c = dc_val[ - wrap]; | |
1784 | |
1785 /* XXX: Rule B is used only for I and BI frames in S/M/C profile | |
1786 * with overlap filtering off | |
1787 */ | |
1788 if ((s->pict_type == I_TYPE || s->pict_type == BI_TYPE) && | |
1789 1 /* XXX: overlap filtering off */) | |
1790 { | |
1791 /* Set outer values */ | |
1792 if (s->first_slice_line && n!=2) b=c=dcpred[scale]; | |
1793 if (s->mb_x == 0) b=a=dcpred[scale]; | |
1794 } | |
1795 else | |
1796 { | |
1797 /* Set outer values */ | |
1798 if (s->first_slice_line && n!=2) b=c=0; | |
1799 if (s->mb_x == 0) b=a=0; | |
1800 | |
1801 /* XXX: Rule A needs to know if blocks are inter or intra :/ */ | |
1802 if (0) | |
1803 { | |
1804 /* update predictor */ | |
1805 *dc_val_ptr = &dc_val[0]; | |
1806 dir_ptr = 0; | |
1807 return a; | |
1808 } | |
1809 } | |
1810 | |
1811 if (abs(a - b) <= abs(b - c)) { | |
1812 pred = c; | |
1813 *dir_ptr = 1; | |
1814 } else { | |
1815 pred = a; | |
1816 *dir_ptr = 0; | |
1817 } | |
1818 | |
1819 /* update predictor */ | |
1820 *dc_val_ptr = &dc_val[0]; | |
1821 return pred; | |
1822 } | |
1823 | |
1824 /** Decode one block, inter or intra | |
1825 * @param v The VC1 context | |
1826 * @param block 8x8 DCT block | |
1827 * @param n Block index in the current MB (<4=>luma) | |
1828 * @param coded If the block is coded | |
1829 * @param mquant Quantizer step for the current block | |
1830 * @see Inter TT: Table 21, p73 + p91-85 | |
1831 * @see Intra TT: Table 20, p72 + p(1)05-(1)07 | |
1832 * @todo TODO: Process the blocks | |
1833 * @todo TODO: Use M$ MPEG-4 cbp prediction | |
1834 */ | |
1835 static int vc1_decode_block(VC1Context *v, DCTELEM block[64], int n, int coded, int mquant) | |
1836 { | |
1837 GetBitContext *gb = &v->s.gb; | |
1838 MpegEncContext *s = &v->s; | |
1839 int ttblk; /* Transform Type per Block */ | |
1840 int subblkpat; /* Sub-block Transform Type Pattern */ | |
1841 int dc_pred_dir; /* Direction of the DC prediction used */ | |
1842 int run_diff, i; | |
1843 | |
1844 /* XXX: Guard against dumb values of mquant */ | |
1845 mquant = (mquant < 1) ? 0 : ( (mquant>31) ? 31 : mquant ); | |
1846 | |
1847 /* Set DC scale - y and c use the same */ | |
1848 s->y_dc_scale = s->y_dc_scale_table[mquant]; | |
1849 s->c_dc_scale = s->c_dc_scale_table[mquant]; | |
1850 | |
1851 if (s->mb_intra) | |
1852 { | |
1853 int dcdiff; | |
1854 uint16_t *dc_val; | |
1855 | |
1856 /* Get DC differential */ | |
1857 if (n < 4) { | |
1858 dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3); | |
1859 } else { | |
1860 dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3); | |
1861 } | |
1862 if (dcdiff < 0){ | |
1863 av_log(s->avctx, AV_LOG_ERROR, "Illegal DC VLC\n"); | |
1864 return -1; | |
1865 } | |
1866 if (dcdiff) | |
1867 { | |
1868 if (dcdiff == 119 /* ESC index value */) | |
1869 { | |
1870 /* TODO: Optimize */ | |
1871 if (mquant == 1) dcdiff = get_bits(gb, 10); | |
1872 else if (mquant == 2) dcdiff = get_bits(gb, 9); | |
1873 else dcdiff = get_bits(gb, 8); | |
1874 } | |
1875 else | |
1876 { | |
1877 if (mquant == 1) | |
1878 dcdiff = (dcdiff<<2) + get_bits(gb, 2) - 3; | |
1879 else if (mquant == 2) | |
1880 dcdiff = (dcdiff<<1) + get_bits(gb, 1) - 1; | |
1881 } | |
1882 if (get_bits(gb, 1)) | |
1883 dcdiff = -dcdiff; | |
1884 } | |
1885 | |
1886 /* Prediction */ | |
1887 dcdiff += vc1_pred_dc(s, n, &dc_val, &dc_pred_dir); | |
1888 *dc_val = dcdiff; | |
1889 /* Store the quantized DC coeff, used for prediction */ | |
1890 | |
1891 if (n < 4) { | |
1892 block[0] = dcdiff * s->y_dc_scale; | |
1893 } else { | |
1894 block[0] = dcdiff * s->c_dc_scale; | |
1895 } | |
1896 if (block[0] < 0) { | |
1897 #if TRACE | |
1898 //av_log(s->avctx, AV_LOG_ERROR, "DC=%i<0\n", dcdiff); | |
1899 #endif | |
1900 //return -1; | |
1901 } | |
1902 /* Skip ? */ | |
1903 run_diff = 0; | |
1904 i = 0; | |
1905 if (!coded) { | |
1906 goto not_coded; | |
1907 } | |
1908 } | |
1909 else | |
1910 { | |
1911 mquant = v->pq; | |
1912 | |
1913 /* Get TTBLK */ | |
1914 if (v->ttmb < 8) /* per block */ | |
1915 ttblk = get_vlc2(gb, vc1_ttblk_vlc[v->tt_index].table, VC1_TTBLK_VLC_BITS, 2); | |
1916 else /* Per frame */ | |
1917 ttblk = 0; //FIXME, depends on ttfrm | |
1918 | |
1919 /* Get SUBBLKPAT */ | |
1920 if (ttblk == v->ttblk4x4) /* 4x4 transform for that qp value */ | |
1921 subblkpat = 1+get_vlc2(gb, vc1_subblkpat_vlc[v->tt_index].table, | |
1922 VC1_SUBBLKPAT_VLC_BITS, 2); | |
1923 else /* All others: 8x8, 4x8, 8x4 */ | |
1924 subblkpat = decode012(gb); | |
1925 } | |
1926 | |
1927 //TODO AC Decoding | |
1928 i = 63; //XXX: nothing done yet | |
1929 | |
1930 | |
1931 not_coded: | |
1932 if (s->mb_intra) { | |
1933 mpeg4_pred_ac(s, block, n, dc_pred_dir); | |
1934 if (s->ac_pred) { | |
1935 i = 63; /* XXX: not optimal */ | |
1936 } | |
1937 } | |
1938 if(i>0) i=63; //FIXME/XXX optimize | |
1939 s->block_last_index[n] = i; | |
1940 return 0; | |
1941 } | |
1942 | |
1943 /** @} */ //End for group block | |
1944 | |
1945 /***********************************************************************/ | |
1946 /** | |
1947 * @defgroup std_mb VC1 Macroblock-level functions in Simple/Main Profiles | |
1948 * @see 7.1.4, p91 and 8.1.1.7, p(1)04 | |
1949 * @todo TODO: Integrate to MpegEncContext facilities | |
1950 * @{ | |
1951 */ | |
1952 | |
1953 static inline int vc1_coded_block_pred(MpegEncContext * s, int n, uint8_t **coded_block_ptr) | |
1954 { | |
1955 int xy, wrap, pred, a, b, c; | |
1956 | |
1957 xy = s->block_index[n]; | |
1958 wrap = s->b8_stride; | |
1959 | |
1960 /* B C | |
1961 * A X | |
1962 */ | |
1963 a = s->coded_block[xy - 1 ]; | |
1964 b = s->coded_block[xy - 1 - wrap]; | |
1965 c = s->coded_block[xy - wrap]; | |
1966 | |
1967 if (b == c) { | |
1968 pred = a; | |
1969 } else { | |
1970 pred = c; | |
1971 } | |
1972 | |
1973 /* store value */ | |
1974 *coded_block_ptr = &s->coded_block[xy]; | |
1975 | |
1976 return pred; | |
1977 } | |
1978 | |
1979 /** Decode one I-frame MB (in Simple/Main profile) | |
1980 * @todo TODO: Extend to AP | |
1981 */ | |
1982 static int vc1_decode_i_mb(VC1Context *v, DCTELEM block[6][64]) | |
1983 { | |
1984 int i, cbp, val; | |
1985 uint8_t *coded_val; | |
1986 // uint32_t * const mb_type_ptr= &v->s.current_picture.mb_type[ v->s.mb_x + v->s.mb_y*v->s.mb_stride ]; | |
1987 | |
1988 v->s.mb_intra = 1; | |
1989 cbp = get_vlc2(&v->s.gb, ff_msmp4_mb_i_vlc.table, MB_INTRA_VLC_BITS, 2); | |
1990 if (cbp < 0) return -1; | |
1991 v->s.ac_pred = get_bits(&v->s.gb, 1); | |
1992 | |
1993 for (i=0; i<6; i++) | |
1994 { | |
1995 val = ((cbp >> (5 - i)) & 1); | |
1996 if (i < 4) { | |
1997 int pred = vc1_coded_block_pred(&v->s, i, &coded_val); | |
1998 val = val ^ pred; | |
1999 *coded_val = val; | |
2000 } | |
2001 cbp |= val << (5 - i); | |
2002 if (vc1_decode_block(v, block[i], i, val, v->pq) < 0) //FIXME Should be mquant | |
2003 { | |
2004 av_log(v->s.avctx, AV_LOG_ERROR, | |
2005 "\nerror while decoding block: %d x %d (%d)\n", v->s.mb_x, v->s.mb_y, i); | |
2006 return -1; | |
2007 } | |
2008 } | |
2009 return 0; | |
2010 } | |
2011 | |
2012 /** Decode one P-frame MB (in Simple/Main profile) | |
2013 * @todo TODO: Extend to AP | |
2014 * @fixme FIXME: DC value for inter blocks not set | |
2015 */ | |
2016 static int vc1_decode_p_mb(VC1Context *v, DCTELEM block[6][64]) | |
2017 { | |
2018 MpegEncContext *s = &v->s; | |
2019 GetBitContext *gb = &s->gb; | |
2020 int i, mb_offset = s->mb_x + s->mb_y*s->mb_width; /* XXX: mb_stride */ | |
2021 int cbp; /* cbp decoding stuff */ | |
2022 int hybrid_pred; /* Prediction types */ | |
2023 int mv_mode_bit = 0; | |
2024 int mqdiff, mquant; /* MB quantization */ | |
2025 int ttmb; /* MB Transform type */ | |
2026 int status; | |
2027 uint8_t *coded_val; | |
2028 | |
2029 static const int size_table[6] = { 0, 2, 3, 4, 5, 8 }, | |
2030 offset_table[6] = { 0, 1, 3, 7, 15, 31 }; | |
2031 int mb_has_coeffs = 1; /* last_flag */ | |
2032 int dmv_x, dmv_y; /* Differential MV components */ | |
2033 int index, index1; /* LUT indices */ | |
2034 int val, sign; /* temp values */ | |
2035 | |
2036 mquant = v->pq; /* Loosy initialization */ | |
2037 | |
2038 if (v->mv_type_mb_plane.is_raw) | |
2039 v->mv_type_mb_plane.data[mb_offset] = get_bits(gb, 1); | |
2040 if (v->skip_mb_plane.is_raw) | |
2041 v->skip_mb_plane.data[mb_offset] = get_bits(gb, 1); | |
2042 if (!mv_mode_bit) /* 1MV mode */ | |
2043 { | |
2044 if (!v->skip_mb_plane.data[mb_offset]) | |
2045 { | |
2046 GET_MVDATA(dmv_x, dmv_y); | |
2047 | |
2048 /* hybrid mv pred, 8.3.5.3.4 */ | |
2049 if (v->mv_mode == MV_PMODE_1MV || | |
2050 v->mv_mode == MV_PMODE_MIXED_MV) | |
2051 hybrid_pred = get_bits(gb, 1); | |
2052 /* FIXME Set DC val for inter block ? */ | |
2053 if (s->mb_intra && !mb_has_coeffs) | |
2054 { | |
2055 GET_MQUANT(); | |
2056 s->ac_pred = get_bits(gb, 1); | |
2057 /* XXX: how to handle cbp ? */ | |
2058 cbp = 0; | |
2059 for (i=0; i<6; i++) | |
2060 { | |
2061 s->coded_block[s->block_index[i]] = 0; | |
2062 vc1_decode_block(v, block[i], i, 0, mquant); | |
2063 } | |
2064 return 0; | |
2065 } | |
2066 else if (mb_has_coeffs) | |
2067 { | |
2068 if (s->mb_intra) s->ac_pred = get_bits(gb, 1); | |
2069 cbp = get_vlc2(&v->s.gb, v->cbpcy_vlc->table, VC1_CBPCY_P_VLC_BITS, 2); | |
2070 GET_MQUANT(); | |
2071 } | |
2072 else | |
2073 { | |
2074 mquant = v->pq; | |
2075 /* XXX: how to handle cbp ? */ | |
2076 /* XXX: how to set values for following predictions ? */ | |
2077 cbp = 0; | |
2078 } | |
2079 | |
2080 if (!v->ttmbf) | |
2081 ttmb = get_vlc2(gb, vc1_ttmb_vlc[v->tt_index].table, | |
2082 VC1_TTMB_VLC_BITS, 12); | |
2083 | |
2084 for (i=0; i<6; i++) | |
2085 { | |
2086 val = ((cbp >> (5 - i)) & 1); | |
2087 if (i < 4) { | |
2088 int pred = vc1_coded_block_pred(&v->s, i, &coded_val); | |
2089 val = val ^ pred; | |
2090 *coded_val = val; | |
2091 } | |
2092 vc1_decode_block(v, block[i], i, val, mquant); //FIXME | |
2093 } | |
2094 } | |
2095 else //Skipped | |
2096 { | |
2097 /* hybrid mv pred, 8.3.5.3.4 */ | |
2098 if (v->mv_mode == MV_PMODE_1MV || | |
2099 v->mv_mode == MV_PMODE_MIXED_MV) | |
2100 hybrid_pred = get_bits(gb, 1); | |
2101 | |
2102 /* TODO: blah */ | |
2103 return 0; | |
2104 } | |
2105 } //1MV mode | |
2106 else //4MV mode | |
2107 { | |
2108 if (!v->skip_mb_plane.data[mb_offset] /* unskipped MB */) | |
2109 { | |
2110 /* Get CBPCY */ | |
2111 cbp = get_vlc2(&v->s.gb, v->cbpcy_vlc->table, VC1_CBPCY_P_VLC_BITS, 2); | |
2112 for (i=0; i<6; i++) | |
2113 { | |
2114 val = ((cbp >> (5 - i)) & 1); | |
2115 if (i < 4) { | |
2116 int pred = vc1_coded_block_pred(&v->s, i, &coded_val); | |
2117 val = val ^ pred; | |
2118 *coded_val = val; | |
2119 } | |
2120 if (i<4 && val) | |
2121 { | |
2122 GET_MVDATA(dmv_x, dmv_y); | |
2123 } | |
2124 if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) | |
2125 hybrid_pred = get_bits(gb, 1); | |
2126 GET_MQUANT(); | |
2127 | |
2128 if (s->mb_intra /* One of the 4 blocks is intra */ && | |
2129 index /* non-zero pred for that block */) | |
2130 s->ac_pred = get_bits(gb, 1); | |
2131 if (!v->ttmbf) | |
2132 ttmb = get_vlc2(gb, vc1_ttmb_vlc[v->tt_index].table, | |
2133 VC1_TTMB_VLC_BITS, 12); | |
2134 status = vc1_decode_block(v, block[i], i, val, mquant); | |
2135 } | |
2136 return status; | |
2137 } | |
2138 else //Skipped MB | |
2139 { | |
2140 /* XXX: Skipped => cbp=0 and mquant doesn't matter ? */ | |
2141 for (i=0; i<4; i++) | |
2142 { | |
2143 if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) | |
2144 hybrid_pred = get_bits(gb, 1); | |
2145 vc1_decode_block(v, block[i], i, 0, v->pq); //FIXME | |
2146 } | |
2147 vc1_decode_block(v, block[4], 4, 0, v->pq); //FIXME | |
2148 vc1_decode_block(v, block[5], 5, 0, v->pq); //FIXME | |
2149 /* TODO: blah */ | |
2150 return 0; | |
2151 } | |
2152 } | |
2153 | |
2154 /* Should never happen */ | |
2155 return -1; | |
2156 } | |
2157 | |
2158 /** Decode one B-frame MB (in Simple/Main profile) | |
2159 * @todo TODO: Extend to AP | |
2160 * @warning XXX: Used for decoding BI MBs | |
2161 * @fixme FIXME: DC value for inter blocks not set | |
2162 */ | |
2163 static int vc1_decode_b_mb(VC1Context *v, DCTELEM block[6][64]) | |
2164 { | |
2165 MpegEncContext *s = &v->s; | |
2166 GetBitContext *gb = &v->s.gb; | |
2167 int mb_offset, i /* MB / B postion information */; | |
2168 int b_mv_type = BMV_TYPE_BACKWARD; | |
2169 int mquant, mqdiff; /* MB quant stuff */ | |
2170 int ttmb; /* MacroBlock transform type */ | |
2171 | |
2172 static const int size_table[6] = { 0, 2, 3, 4, 5, 8 }, | |
2173 offset_table[6] = { 0, 1, 3, 7, 15, 31 }; | |
2174 int mb_has_coeffs = 1; /* last_flag */ | |
2175 int dmv1_x, dmv1_y, dmv2_x, dmv2_y; /* Differential MV components */ | |
2176 int index, index1; /* LUT indices */ | |
2177 int val, sign; /* MVDATA temp values */ | |
2178 | |
2179 mb_offset = s->mb_width*s->mb_y + s->mb_x; //FIXME: arrays aren't using stride | |
2180 | |
2181 if (v->direct_mb_plane.is_raw) | |
2182 v->direct_mb_plane.data[mb_offset] = get_bits(gb, 1); | |
2183 if (v->skip_mb_plane.is_raw) | |
2184 v->skip_mb_plane.data[mb_offset] = get_bits(gb, 1); | |
2185 | |
2186 if (!v->direct_mb_plane.data[mb_offset]) | |
2187 { | |
2188 if (v->skip_mb_plane.data[mb_offset]) | |
2189 { | |
2190 b_mv_type = decode012(gb); | |
2191 if (v->bfraction > 420 /*1/2*/ && | |
2192 b_mv_type < 3) b_mv_type = 1-b_mv_type; | |
2193 } | |
2194 else | |
2195 { | |
2196 GET_MVDATA(dmv1_x, dmv1_y); | |
2197 if (!s->mb_intra /* b_mv1 tells not intra */) | |
2198 { | |
2199 b_mv_type = decode012(gb); | |
2200 if (v->bfraction > 420 /*1/2*/ && | |
2201 b_mv_type < 3) b_mv_type = 1-b_mv_type; | |
2202 } | |
2203 } | |
2204 } | |
2205 if (!v->skip_mb_plane.data[mb_offset]) | |
2206 { | |
2207 if (mb_has_coeffs /* BMV1 == "last" */) | |
2208 { | |
2209 GET_MQUANT(); | |
2210 if (s->mb_intra /* intra mb */) | |
2211 s->ac_pred = get_bits(gb, 1); | |
2212 } | |
2213 else | |
2214 { | |
2215 /* if bmv1 tells MVs are interpolated */ | |
2216 if (b_mv_type == BMV_TYPE_INTERPOLATED) | |
2217 { | |
2218 GET_MVDATA(dmv2_x, dmv2_y); | |
2219 mquant = v->pq; //FIXME: initialization not necessary ? | |
2220 } | |
2221 /* GET_MVDATA has reset some stuff */ | |
2222 if (mb_has_coeffs /* b_mv2 == "last" */) | |
2223 { | |
2224 if (s->mb_intra /* intra_mb */) | |
2225 s->ac_pred = get_bits(gb, 1); | |
2226 GET_MQUANT(); | |
2227 } | |
2228 } | |
2229 } | |
2230 | |
2231 //End1 | |
2232 if (v->ttmbf) | |
2233 ttmb = get_vlc2(gb, vc1_ttmb_vlc[v->tt_index].table, | |
2234 VC1_TTMB_VLC_BITS, 12); | |
2235 | |
2236 //End2 | |
2237 for (i=0; i<6; i++) | |
2238 { | |
2239 vc1_decode_block(v, block[i], i, 0 /*cbp[i]*/, mquant); //FIXME | |
2240 } | |
2241 return 0; | |
2242 } | |
2243 | |
2244 /** Decode all MBs for an I frame in Simple/Main profile | |
2245 * @todo TODO: Move out of the loop the picture type case? | |
2246 (branch prediction should help there though) | |
2247 */ | |
2248 static int standard_decode_mbs(VC1Context *v) | |
2249 { | |
2250 MpegEncContext *s = &v->s; | |
2251 | |
2252 /* Set transform type info depending on pq */ | |
2253 if (v->pq < 5) | |
2254 { | |
2255 v->tt_index = 0; | |
2256 v->ttblk4x4 = 3; | |
2257 } | |
2258 else if (v->pq < 13) | |
2259 { | |
2260 v->tt_index = 1; | |
2261 v->ttblk4x4 = 3; | |
2262 } | |
2263 else | |
2264 { | |
2265 v->tt_index = 2; | |
2266 v->ttblk4x4 = 2; | |
2267 } | |
2268 | |
2269 if (s->pict_type != I_TYPE) | |
2270 { | |
2271 /* Select proper long MV range */ | |
2272 switch (v->mvrange) | |
2273 { | |
2274 case 1: v->k_x = 10; v->k_y = 9; break; | |
2275 case 2: v->k_x = 12; v->k_y = 10; break; | |
2276 case 3: v->k_x = 13; v->k_y = 11; break; | |
2277 default: /*case 0 too */ v->k_x = 9; v->k_y = 8; break; | |
2278 } | |
2279 | |
2280 s->mspel = v->mv_mode & 1; //MV_PMODE is HPEL | |
2281 v->k_x -= s->mspel; | |
2282 v->k_y -= s->mspel; | |
2283 } | |
2284 | |
2285 for (s->mb_y=0; s->mb_y<s->mb_height; s->mb_y++) | |
2286 { | |
2287 for (s->mb_x=0; s->mb_x<s->mb_width; s->mb_x++) | |
2288 { | |
2289 //FIXME Get proper MB DCTELEM | |
2290 //TODO Move out of the loop | |
2291 switch (s->pict_type) | |
2292 { | |
2293 case I_TYPE: vc1_decode_i_mb(v, s->block); break; | |
2294 case P_TYPE: vc1_decode_p_mb(v, s->block); break; | |
2295 case BI_TYPE: | |
2296 case B_TYPE: vc1_decode_b_mb(v, s->block); break; | |
2297 } | |
2298 } | |
2299 //Add a check for overconsumption ? | |
2300 } | |
2301 return 0; | |
2302 } | |
2303 /** @} */ //End for group std_mb | |
2304 | |
2305 #if HAS_ADVANCED_PROFILE | |
2306 /***********************************************************************/ | |
2307 /** | |
2308 * @defgroup adv_mb VC1 Macroblock-level functions in Advanced Profile | |
2309 * @todo TODO: Integrate to MpegEncContext facilities | |
2310 * @todo TODO: Code P, B and BI | |
2311 * @{ | |
2312 */ | |
2313 static int advanced_decode_i_mbs(VC1Context *v) | |
2314 { | |
2315 MpegEncContext *s = &v->s; | |
2316 GetBitContext *gb = &v->s.gb; | |
2317 int mqdiff, mquant, mb_offset = 0, over_flags_mb = 0; | |
2318 | |
2319 for (s->mb_y=0; s->mb_y<s->mb_height; s->mb_y++) | |
2320 { | |
2321 for (s->mb_x=0; s->mb_x<s->mb_width; s->mb_x++) | |
2322 { | |
2323 if (v->ac_pred_plane.is_raw) | |
2324 s->ac_pred = get_bits(gb, 1); | |
2325 else | |
2326 s->ac_pred = v->ac_pred_plane.data[mb_offset]; | |
2327 if (v->condover == 3 && v->over_flags_plane.is_raw) | |
2328 over_flags_mb = get_bits(gb, 1); | |
2329 GET_MQUANT(); | |
2330 | |
2331 /* TODO: lots */ | |
2332 } | |
2333 mb_offset++; | |
2334 } | |
2335 return 0; | |
2336 } | |
2337 /** @} */ //End for group adv_mb | |
2338 #endif | |
2339 | |
2340 /** Initialize a VC1/WMV3 decoder | |
2341 * @todo TODO: Handle VC-1 IDUs (Transport level?) | |
2342 * @todo TODO: Decypher remaining bits in extra_data | |
2343 */ | |
2344 static int vc1_decode_init(AVCodecContext *avctx) | |
2345 { | |
2346 VC1Context *v = avctx->priv_data; | |
2347 MpegEncContext *s = &v->s; | |
2348 GetBitContext gb; | |
2349 | |
2350 if (!avctx->extradata_size || !avctx->extradata) return -1; | |
2351 avctx->pix_fmt = PIX_FMT_YUV420P; | |
2352 v->s.avctx = avctx; | |
2353 | |
2354 if(ff_h263_decode_init(avctx) < 0) | |
2355 return -1; | |
2356 if (vc1_init_common(v) < 0) return -1; | |
2357 | |
2358 av_log(avctx, AV_LOG_INFO, "This decoder is not supposed to produce picture. Dont report this as a bug!\n"); | |
2359 | |
2360 avctx->coded_width = avctx->width; | |
2361 avctx->coded_height = avctx->height; | |
2362 if (avctx->codec_id == CODEC_ID_WMV3) | |
2363 { | |
2364 int count = 0; | |
2365 | |
2366 // looks like WMV3 has a sequence header stored in the extradata | |
2367 // advanced sequence header may be before the first frame | |
2368 // the last byte of the extradata is a version number, 1 for the | |
2369 // samples we can decode | |
2370 | |
2371 init_get_bits(&gb, avctx->extradata, avctx->extradata_size*8); | |
2372 | |
2373 if (decode_sequence_header(avctx, &gb) < 0) | |
2374 return -1; | |
2375 | |
2376 count = avctx->extradata_size*8 - get_bits_count(&gb); | |
2377 if (count>0) | |
2378 { | |
2379 av_log(avctx, AV_LOG_INFO, "Extra data: %i bits left, value: %X\n", | |
2380 count, get_bits(&gb, count)); | |
2381 } | |
2382 else if (count < 0) | |
2383 { | |
2384 av_log(avctx, AV_LOG_INFO, "Read %i bits in overflow\n", -count); | |
2385 } | |
2386 } | |
2387 avctx->has_b_frames= !!(avctx->max_b_frames); | |
2388 | |
2389 s->mb_width = (avctx->coded_width+15)>>4; | |
2390 s->mb_height = (avctx->coded_height+15)>>4; | |
2391 | |
2392 /* Allocate mb bitplanes */ | |
2393 if (alloc_bitplane(&v->mv_type_mb_plane, s->mb_width, s->mb_height) < 0) | |
2394 return -1; | |
2395 if (alloc_bitplane(&v->mv_type_mb_plane, s->mb_width, s->mb_height) < 0) | |
2396 return -1; | |
2397 if (alloc_bitplane(&v->skip_mb_plane, s->mb_width, s->mb_height) < 0) | |
2398 return -1; | |
2399 if (alloc_bitplane(&v->direct_mb_plane, s->mb_width, s->mb_height) < 0) | |
2400 return -1; | |
2401 | |
2402 /* For predictors */ | |
2403 v->previous_line_cbpcy = (uint8_t *)av_malloc(s->mb_stride*4); | |
2404 if (!v->previous_line_cbpcy) return -1; | |
2405 | |
2406 #if HAS_ADVANCED_PROFILE | |
2407 if (v->profile == PROFILE_ADVANCED) | |
2408 { | |
2409 if (alloc_bitplane(&v->over_flags_plane, s->mb_width, s->mb_height) < 0) | |
2410 return -1; | |
2411 if (alloc_bitplane(&v->ac_pred_plane, s->mb_width, s->mb_height) < 0) | |
2412 return -1; | |
2413 } | |
2414 #endif | |
2415 | |
2416 return 0; | |
2417 } | |
2418 | |
2419 /** Decode a VC1/WMV3 frame | |
2420 * @todo TODO: Handle VC-1 IDUs (Transport level?) | |
2421 * @warning Initial try at using MpegEncContext stuff | |
2422 */ | |
2423 static int vc1_decode_frame(AVCodecContext *avctx, | |
2424 void *data, int *data_size, | |
2425 uint8_t *buf, int buf_size) | |
2426 { | |
2427 VC1Context *v = avctx->priv_data; | |
2428 MpegEncContext *s = &v->s; | |
2429 int ret = FRAME_SKIPPED, len; | |
2430 AVFrame *pict = data; | |
2431 uint8_t *tmp_buf; | |
2432 v->s.avctx = avctx; | |
2433 | |
2434 //buf_size = 0 -> last frame | |
2435 if (!buf_size) return 0; | |
2436 | |
2437 len = avpicture_get_size(avctx->pix_fmt, avctx->width, | |
2438 avctx->height); | |
2439 tmp_buf = (uint8_t *)av_mallocz(len); | |
2440 avpicture_fill((AVPicture *)pict, tmp_buf, avctx->pix_fmt, | |
2441 avctx->width, avctx->height); | |
2442 | |
2443 if (avctx->codec_id == CODEC_ID_VC1) | |
2444 { | |
2445 #if 0 | |
2446 // search for IDU's | |
2447 // FIXME | |
2448 uint32_t scp = 0; | |
2449 int scs = 0, i = 0; | |
2450 | |
2451 while (i < buf_size) | |
2452 { | |
2453 for (; i < buf_size && scp != 0x000001; i++) | |
2454 scp = ((scp<<8)|buf[i])&0xffffff; | |
2455 | |
2456 if (scp != 0x000001) | |
2457 break; // eof ? | |
2458 | |
2459 scs = buf[i++]; | |
2460 | |
2461 init_get_bits(gb, buf+i, (buf_size-i)*8); | |
2462 | |
2463 switch(scs) | |
2464 { | |
2465 case 0x0A: //Sequence End Code | |
2466 return 0; | |
2467 case 0x0B: //Slice Start Code | |
2468 av_log(avctx, AV_LOG_ERROR, "Slice coding not supported\n"); | |
2469 return -1; | |
2470 case 0x0C: //Field start code | |
2471 av_log(avctx, AV_LOG_ERROR, "Interlaced coding not supported\n"); | |
2472 return -1; | |
2473 case 0x0D: //Frame start code | |
2474 break; | |
2475 case 0x0E: //Entry point Start Code | |
2476 if (v->profile < PROFILE_ADVANCED) | |
2477 av_log(avctx, AV_LOG_ERROR, | |
2478 "Found an entry point in profile %i\n", v->profile); | |
2479 advanced_entry_point_process(avctx, gb); | |
2480 break; | |
2481 case 0x0F: //Sequence header Start Code | |
2482 decode_sequence_header(avctx, gb); | |
2483 break; | |
2484 default: | |
2485 av_log(avctx, AV_LOG_ERROR, | |
2486 "Unsupported IDU suffix %lX\n", scs); | |
2487 } | |
2488 | |
2489 i += get_bits_count(gb)*8; | |
2490 } | |
2491 #else | |
2492 av_abort(); | |
2493 #endif | |
2494 } | |
2495 else | |
2496 init_get_bits(&v->s.gb, buf, buf_size*8); | |
2497 | |
2498 s->flags= avctx->flags; | |
2499 s->flags2= avctx->flags2; | |
2500 | |
2501 /* no supplementary picture */ | |
2502 if (buf_size == 0) { | |
2503 /* special case for last picture */ | |
2504 if (s->low_delay==0 && s->next_picture_ptr) { | |
2505 *pict= *(AVFrame*)s->next_picture_ptr; | |
2506 s->next_picture_ptr= NULL; | |
2507 | |
2508 *data_size = sizeof(AVFrame); | |
2509 } | |
2510 | |
2511 return 0; | |
2512 } | |
2513 | |
2514 //No IDU - we mimic ff_h263_decode_frame | |
2515 s->bitstream_buffer_size=0; | |
2516 | |
2517 if (!s->context_initialized) { | |
2518 if (MPV_common_init(s) < 0) //we need the idct permutaton for reading a custom matrix | |
2519 return -1; | |
2520 } | |
2521 | |
2522 //we need to set current_picture_ptr before reading the header, otherwise we cant store anyting im there | |
2523 if(s->current_picture_ptr==NULL || s->current_picture_ptr->data[0]){ | |
2524 s->current_picture_ptr= &s->picture[ff_find_unused_picture(s, 0)]; | |
2525 } | |
2526 #if HAS_ADVANCED_PROFILE | |
2527 if (v->profile == PROFILE_ADVANCED) | |
2528 ret= advanced_decode_picture_primary_header(v); | |
2529 else | |
2530 #endif | |
2531 ret= standard_decode_picture_primary_header(v); | |
2532 if (ret == FRAME_SKIPPED) return buf_size; | |
2533 /* skip if the header was thrashed */ | |
2534 if (ret < 0){ | |
2535 av_log(s->avctx, AV_LOG_ERROR, "header damaged\n"); | |
2536 return -1; | |
2537 } | |
2538 | |
2539 //No bug workaround yet, no DCT conformance | |
2540 | |
2541 //WMV9 does have resized images | |
2542 if (v->profile < PROFILE_ADVANCED && v->multires){ | |
2543 //Parse context stuff in here, don't know how appliable it is | |
2544 } | |
2545 //Not sure about context initialization | |
2546 | |
2547 // for hurry_up==5 | |
2548 s->current_picture.pict_type= s->pict_type; | |
2549 s->current_picture.key_frame= s->pict_type == I_TYPE; | |
2550 | |
2551 /* skip b frames if we dont have reference frames */ | |
2552 if(s->last_picture_ptr==NULL && (s->pict_type==B_TYPE || s->dropable)) | |
2553 return buf_size; //FIXME simulating all buffer consumed | |
2554 /* skip b frames if we are in a hurry */ | |
2555 if(avctx->hurry_up && s->pict_type==B_TYPE) | |
2556 return buf_size; //FIXME simulating all buffer consumed | |
2557 /* skip everything if we are in a hurry>=5 */ | |
2558 if(avctx->hurry_up>=5) | |
2559 return buf_size; //FIXME simulating all buffer consumed | |
2560 | |
2561 if(s->next_p_frame_damaged){ | |
2562 if(s->pict_type==B_TYPE) | |
2563 return buf_size; //FIXME simulating all buffer consumed | |
2564 else | |
2565 s->next_p_frame_damaged=0; | |
2566 } | |
2567 | |
2568 if(MPV_frame_start(s, avctx) < 0) | |
2569 return -1; | |
2570 | |
2571 ff_er_frame_start(s); | |
2572 | |
2573 //wmv9 may or may not have skip bits | |
2574 #if HAS_ADVANCED_PROFILE | |
2575 if (v->profile == PROFILE_ADVANCED) | |
2576 ret= advanced_decode_picture_secondary_header(v); | |
2577 else | |
2578 #endif | |
2579 ret = standard_decode_picture_secondary_header(v); | |
2580 if (ret<0) return FRAME_SKIPPED; //FIXME Non fatal for now | |
2581 | |
2582 //We consider the image coded in only one slice | |
2583 #if HAS_ADVANCED_PROFILE | |
2584 if (v->profile == PROFILE_ADVANCED) | |
2585 { | |
2586 switch(s->pict_type) | |
2587 { | |
2588 case I_TYPE: ret = advanced_decode_i_mbs(v); break; | |
2589 case P_TYPE: ret = decode_p_mbs(v); break; | |
2590 case B_TYPE: | |
2591 case BI_TYPE: ret = decode_b_mbs(v); break; | |
2592 default: ret = FRAME_SKIPPED; | |
2593 } | |
2594 if (ret == FRAME_SKIPPED) return buf_size; //We ignore for now failures | |
2595 } | |
2596 else | |
2597 #endif | |
2598 { | |
2599 ret = standard_decode_mbs(v); | |
2600 if (ret == FRAME_SKIPPED) return buf_size; | |
2601 } | |
2602 | |
2603 ff_er_frame_end(s); | |
2604 | |
2605 MPV_frame_end(s); | |
2606 | |
2607 assert(s->current_picture.pict_type == s->current_picture_ptr->pict_type); | |
2608 assert(s->current_picture.pict_type == s->pict_type); | |
2609 | |
2610 if (s->pict_type == B_TYPE || s->low_delay) { | |
2611 *pict= *(AVFrame*)s->current_picture_ptr; | |
2612 } else if (s->last_picture_ptr != NULL) { | |
2613 *pict= *(AVFrame*)s->last_picture_ptr; | |
2614 } | |
2615 | |
2616 if(s->last_picture_ptr || s->low_delay){ | |
2617 *data_size = sizeof(AVFrame); | |
2618 ff_print_debug_info(s, pict); | |
2619 } | |
2620 | |
2621 /* Return the Picture timestamp as the frame number */ | |
2622 /* we substract 1 because it is added on utils.c */ | |
2623 avctx->frame_number = s->picture_number - 1; | |
2624 | |
2625 av_log(avctx, AV_LOG_DEBUG, "Consumed %i/%i bits\n", | |
2626 get_bits_count(&s->gb), buf_size*8); | |
2627 | |
2628 /* Fake consumption of all data */ | |
2629 *data_size = len; | |
2630 return buf_size; //Number of bytes consumed | |
2631 } | |
2632 | |
2633 /** Close a VC1/WMV3 decoder | |
2634 * @warning Initial try at using MpegEncContext stuff | |
2635 */ | |
2636 static int vc1_decode_end(AVCodecContext *avctx) | |
2637 { | |
2638 VC1Context *v = avctx->priv_data; | |
2639 | |
2640 #if HAS_ADVANCED_PROFILE | |
2641 av_freep(&v->hrd_rate); | |
2642 av_freep(&v->hrd_buffer); | |
2643 #endif | |
2644 MPV_common_end(&v->s); | |
2645 free_bitplane(&v->mv_type_mb_plane); | |
2646 free_bitplane(&v->skip_mb_plane); | |
2647 free_bitplane(&v->direct_mb_plane); | |
2648 return 0; | |
2649 } | |
2650 | |
2651 AVCodec vc1_decoder = { | |
2652 "vc1", | |
2653 CODEC_TYPE_VIDEO, | |
2654 CODEC_ID_VC1, | |
2655 sizeof(VC1Context), | |
2656 vc1_decode_init, | |
2657 NULL, | |
2658 vc1_decode_end, | |
2659 vc1_decode_frame, | |
2660 CODEC_CAP_DELAY, | |
2661 NULL | |
2662 }; | |
2663 | |
2664 AVCodec wmv3_decoder = { | |
2665 "wmv3", | |
2666 CODEC_TYPE_VIDEO, | |
2667 CODEC_ID_WMV3, | |
2668 sizeof(VC1Context), | |
2669 vc1_decode_init, | |
2670 NULL, | |
2671 vc1_decode_end, | |
2672 vc1_decode_frame, | |
2673 CODEC_CAP_DELAY, | |
2674 NULL | |
2675 }; |