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,