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