Commit | Line | Data |
---|---|---|
21aa398f AB |
1 | /* |
2 | * VC-9 and WMV3 decoder | |
3 | * Copyright (c) 2005 Anonymous | |
4 | * Copyright (c) 2005 Alex Beregszaszi | |
5 | * | |
6 | * This library is free software; you can redistribute it and/or | |
7 | * modify it under the terms of the GNU Lesser General Public | |
8 | * License as published by the Free Software Foundation; either | |
9 | * version 2 of the License, or (at your option) any later version. | |
10 | * | |
11 | * This library is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | * Lesser General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU Lesser General Public | |
17 | * License along with this library; if not, write to the Free Software | |
18 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
19 | * | |
20 | */ | |
21 | ||
22 | /** | |
23 | * @file vc9.c | |
24 | * VC-9 and WMV3 decoder | |
25 | * | |
26 | * TODO: Norm-6 bitplane imode, most AP stuff, optimize, all of MB layer :) | |
27 | * TODO: use MPV_ !! | |
28 | */ | |
29 | #include "common.h" | |
30 | #include "dsputil.h" | |
31 | #include "avcodec.h" | |
32 | #include "mpegvideo.h" | |
33 | #include "vc9data.h" | |
34 | ||
35 | /* Some inhibiting stuff */ | |
36 | #define HAS_ADVANCED_PROFILE 1 | |
37 | #define TRACE 1 | |
38 | ||
39 | #if TRACE | |
40 | # define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
41 | codes, codes_wrap, codes_size, use_static) \ | |
42 | if (init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
43 | codes, codes_wrap, codes_size, use_static) < 0) \ | |
44 | { \ | |
45 | av_log(v->avctx, AV_LOG_ERROR, "Error for " # vlc " (%i)\n", i); \ | |
46 | return -1; \ | |
47 | } | |
48 | #else | |
49 | # define INIT_VLC(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
50 | codes, codes_wrap, codes_size, use_static) \ | |
51 | init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, \ | |
52 | codes, codes_wrap, codes_size, use_static) | |
53 | #endif | |
54 | ||
55 | #define PROFILE_SIMPLE 0 | |
56 | #define PROFILE_MAIN 1 | |
57 | #define PROFILE_ADVANCED 3 | |
58 | ||
59 | #define QUANT_FRAME_IMPLICIT 0 | |
60 | #define QUANT_FRAME_EXPLICIT 1 | |
61 | #define QUANT_NON_UNIFORM 2 | |
62 | #define QUANT_UNIFORM 3 | |
63 | ||
64 | /* Where quant can be changed */ | |
65 | #define DQPROFILE_FOUR_EDGES 0 | |
66 | #define DQPROFILE_DOUBLE_EDGES 1 | |
67 | #define DQPROFILE_SINGLE_EDGE 2 | |
68 | #define DQPROFILE_ALL_MBS 3 | |
69 | ||
70 | /* Which edge is quantized with ALTPQUANT */ | |
71 | #define DQSINGLE_BEDGE_LEFT 0 | |
72 | #define DQSINGLE_BEDGE_TOP 1 | |
73 | #define DQSINGLE_BEDGE_RIGHT 2 | |
74 | #define DQSINGLE_BEDGE_BOTTOM 3 | |
75 | ||
76 | /* Which pair of edges is quantized with ALTPQUANT */ | |
77 | #define DQDOUBLE_BEDGE_TOPLEFT 0 | |
78 | #define DQDOUBLE_BEDGE_TOPRIGHT 1 | |
79 | #define DQDOUBLE_BEDGE_BOTTOMRIGHT 2 | |
80 | #define DQDOUBLE_BEDGE_BOTTOMLEFT 3 | |
81 | ||
82 | ||
83 | /* Start Codes */ | |
84 | #define SEQ_SC 0x00000010F /* Sequence Start Code */ | |
85 | #define SEQ_EC 0x00000000A /* Sequence End code */ | |
86 | #define SEQ_HDR /* Sequence Header */ | |
87 | #define ENTRY_SC 0x00000010E /* Entry Point Start Code */ | |
88 | #define ENTRY_HDR /* Entry Point Header */ | |
89 | #define FRM_SC 0x00000010D /* Frame Start Code */ | |
90 | #define FRM_DAT /* Frame Data (includes a Frame Header) */ | |
91 | #define FLD_SC 0x00000010C /* Field Start Code */ | |
92 | #define FLD1_DAT /* Field 1 Data (includes a Frame Header) */ | |
93 | #define FLD2_DAT /* Field 2 Data (includes a Field Header) */ | |
94 | #define SLC_SC 0x00000010B /* Slice Start Code */ | |
95 | #define SLC_HDR /* Slice Header */ | |
96 | #define SLC_DAT /* Slice Data (FrH or FiH possible) */ | |
97 | ||
98 | /* MV P modes */ | |
99 | #define MV_PMODE_1MV_HPEL_BILIN 0 | |
100 | #define MV_PMODE_1MV 1 | |
101 | #define MV_PMODE_1MV_HPEL 2 | |
102 | #define MV_PMODE_MIXED_MV 3 | |
103 | #define MV_PMODE_INTENSITY_COMP 4 | |
104 | ||
105 | /* MV P mode - the 5th element is only used for mode 1 */ | |
106 | static const uint8_t mv_pmode_table[2][5] = { | |
107 | { MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_1MV, MV_PMODE_1MV_HPEL, MV_PMODE_MIXED_MV, MV_PMODE_INTENSITY_COMP }, | |
108 | { MV_PMODE_1MV, MV_PMODE_MIXED_MV, MV_PMODE_1MV_HPEL, MV_PMODE_1MV_HPEL_BILIN, MV_PMODE_INTENSITY_COMP } | |
109 | }; | |
110 | ||
111 | /* One more frame type */ | |
112 | #define BI_TYPE 7 | |
113 | ||
114 | /* FIXME Worse than ugly */ | |
115 | static const int fps_nr[5] = { 24, 25, 30, 50, 60 }, | |
116 | fps_dr[2] = { 1000, 1001 }; | |
117 | static const uint8_t pquant_table[3][32] = { | |
118 | { /* Implicit quantizer */ | |
119 | 0, 1, 2, 3, 4, 5, 6, 7, 8, 6, 7, 8, 9, 10, 11, 12, | |
120 | 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 27, 29, 31 | |
121 | }, | |
122 | { /* Explicit quantizer, pquantizer uniform */ | |
123 | 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, | |
124 | 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31 | |
125 | }, | |
126 | { /* Explicit quantizer, pquantizer non-uniform */ | |
127 | 0, 1, 1, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, | |
128 | 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 29, 31 | |
129 | } | |
130 | }; | |
131 | ||
132 | // FIXME move this into the context | |
133 | #define VC9_BFRACTION_VLC_BITS 7 | |
134 | static VLC vc9_bfraction_vlc; | |
135 | #define VC9_IMODE_VLC_BITS 4 | |
136 | static VLC vc9_imode_vlc; | |
137 | #define VC9_NORM2_VLC_BITS 3 | |
138 | static VLC vc9_norm2_vlc; | |
139 | /* Could be optimized, one table only needs 8 bits */ | |
140 | #define VC9_TTMB_VLC_BITS 12 | |
141 | static VLC vc9_ttmb_vlc[3]; | |
142 | #define VC9_MV_DIFF_VLC_BITS 15 | |
143 | static VLC vc9_mv_diff_vlc[4]; | |
144 | #define VC9_CBPCY_I_VLC_BITS 13 | |
145 | static VLC vc9_cbpcy_i_vlc; | |
146 | #define VC9_CBPCY_P_VLC_BITS 14 | |
147 | static VLC vc9_cbpcy_p_vlc[4]; | |
148 | #define VC9_4MV_BLOCK_PATTERN_VLC_BITS 6 | |
149 | static VLC vc9_4mv_block_pattern_vlc[4]; | |
150 | #define VC9_LUMA_DC_VLC_BITS 26 | |
151 | static VLC vc9_luma_dc_vlc[2]; | |
152 | ||
153 | typedef struct VC9Context{ | |
154 | /* No MpegEnc context, might be good to use it */ | |
155 | GetBitContext gb; | |
156 | AVCodecContext *avctx; | |
157 | ||
158 | /***************************/ | |
159 | /* Sequence Header */ | |
160 | /***************************/ | |
161 | /* Simple/Main Profile */ | |
162 | int res_sm; //reserved, 2b | |
163 | int res_x8; //reserved | |
164 | int multires; //frame-level RESPIC syntax element present | |
165 | int res_fasttx; //always 1 | |
166 | int res_transtab; //always 0 | |
167 | int syncmarker; //Sync markers presents | |
168 | int rangered; //RANGEREDFRM (range reduction) syntax element present | |
169 | int res_rtm_flag; //reserved, set to 1 | |
170 | int reserved; //duh | |
171 | ||
172 | #if HAS_ADVANCED_PROFILE | |
173 | /* Advanced Profile */ | |
174 | int level; //3 | |
175 | int chromaformat; //2 | |
176 | int postprocflag; //frame-based processing use | |
177 | int broadcast; //TFF/RFF present | |
178 | int interlace; //Progressive/interlaced (RPTFTM syntax element) | |
179 | int tfcntrflag; //TFCNTR present | |
180 | int panscanflag; //NUMPANSCANWIN, TOPLEFT{X,Y}, BOTRIGHT{X,Y} presents | |
181 | int extended_dmv; | |
182 | int color_prim; //8 | |
183 | int transfer_char; //8 | |
184 | int matrix_coef; //8 | |
185 | int hrd_param_flag; | |
186 | #endif | |
187 | ||
188 | /* All Profiles */ | |
189 | /* TODO: move all int to flags */ | |
190 | int profile; //2 | |
191 | int frmrtq_postproc; //3 | |
192 | int bitrtq_postproc; //5 | |
193 | int loopfilter; | |
194 | int fastuvmc; //Rounding of qpel vector to hpel ? (not in Simple) | |
195 | int extended_mv; //Ext MV in P/B (not in Simple) | |
196 | int dquant; //Q varies with MBs, 2bits (not in Simple) | |
197 | int vstransform; //variable-size transform46 | |
198 | ||
199 | int overlap; //overlapped transforms in use | |
200 | int quantizer_mode; //2, quantizer mode used for sequence, see QUANT_* | |
201 | int finterpflag; //INTERPFRM present | |
202 | ||
203 | ||
204 | /*****************************/ | |
205 | /* Frame decoding */ | |
206 | /*****************************/ | |
207 | /* All profiles */ | |
208 | uint8_t mv_mode, mv_mode2; /* MV coding mode */ | |
209 | uint8_t pict_type; /* Picture type, mapped on MPEG types */ | |
210 | uint8_t pq, altpq; /* Quantizers */ | |
211 | uint8_t dquantfrm, dqprofile, dqsbedge, dqbilevel; /* pquant parameters */ | |
212 | int width_mb, height_mb; | |
213 | int tile; /* 3x2 if (width_mb%3) else 2x3 */ | |
214 | int transacfrm2, transacfrm, transacdctab; //1bit elements | |
215 | uint8_t ttmbf, ttfrm; /* Transform type */ | |
216 | uint8_t lumscale, lumshift; /* Luma compensation parameters */ | |
217 | int16_t bfraction; /* Relative position % anchors=> how to scale MVs */ | |
218 | uint8_t halfpq; /* Uniform quant over image and qp+.5 */ | |
219 | uint8_t respic; | |
220 | /* Ranges: | |
221 | * 0 -> [-64n 63.f] x [-32, 31.f] | |
222 | * 1 -> [-128, 127.f] x [-64, 63.f] | |
223 | * 2 -> [-512, 511.f] x [-128, 127.f] | |
224 | * 3 -> [-1024, 1023.f] x [-256, 255.f] | |
225 | */ | |
226 | uint8_t mvrange; | |
227 | uint8_t pquantizer; | |
228 | VLC *cbpcy_vlc /* Current CBPCY VLC table */, | |
229 | *mv_diff_vlc /* Current MV Diff VLC table */, | |
230 | *ttmb_vlc /* Current MB Transform Type VLC table */; | |
231 | uint8_t *mv_type_mb_plane; /* bitplane for mv_type == "raw" */ | |
232 | uint8_t *skip_mb_plane, /* bitplane for skipped MBs */ | |
233 | *direct_mb_plane; /* bitplane for "direct" MBs */ | |
234 | ||
235 | /* S/M only ? */ | |
236 | uint8_t rangeredfrm, interpfrm; | |
237 | ||
238 | #if HAS_ADVANCED_PROFILE | |
239 | /* Advanced */ | |
240 | uint8_t fcm; //0->Progressive, 2->Frame-Interlace, 3->Field-Interlace | |
241 | uint8_t numpanscanwin; | |
242 | uint8_t tfcntr; | |
243 | uint8_t rptfrm, tff, rff; | |
244 | uint8_t topleftx; | |
245 | uint8_t toplefty; | |
246 | uint8_t bottomrightx; | |
247 | uint8_t bottomrighty; | |
248 | uint8_t rndctrl; | |
249 | uint8_t uvsamp; | |
250 | uint8_t postproc; | |
251 | int hrd_num_leaky_buckets; | |
252 | uint8_t bit_rate_exponent; | |
253 | uint8_t buffer_size_exponent; | |
254 | uint8_t *ac_pred_plane; | |
255 | uint8_t *over_flags_plane; | |
256 | uint16_t *hrd_rate, *hrd_buffer; | |
257 | #endif | |
258 | } VC9Context; | |
259 | ||
260 | /* FIXME Slow and ugly */ | |
261 | static int get_prefix(GetBitContext *gb, int stop, int len) | |
262 | { | |
263 | int i = 0, tmp = !stop; | |
264 | ||
265 | while (i != len && tmp != stop) | |
266 | { | |
267 | tmp = get_bits(gb, 1); | |
268 | i++; | |
269 | } | |
270 | return i; | |
271 | } | |
272 | ||
273 | static int init_common(VC9Context *v) | |
274 | { | |
275 | static int done = 0; | |
276 | int i; | |
277 | ||
278 | v->mv_type_mb_plane = v->direct_mb_plane = v->skip_mb_plane = NULL; | |
279 | v->pq = -1; | |
280 | #if HAS_ADVANCED_PROFILE | |
281 | v->ac_pred_plane = v->over_flags_plane = NULL; | |
282 | v->hrd_rate = v->hrd_buffer = NULL; | |
283 | #endif | |
284 | if(!done) | |
285 | { | |
286 | done = 1; | |
287 | INIT_VLC(&vc9_bfraction_vlc, VC9_BFRACTION_VLC_BITS, 23, | |
288 | vc9_bfraction_bits, 1, 1, | |
289 | vc9_bfraction_codes, 1, 1, 1); | |
290 | INIT_VLC(&vc9_norm2_vlc, VC9_NORM2_VLC_BITS, 4, | |
291 | vc9_norm2_bits, 1, 1, | |
292 | vc9_norm2_codes, 1, 1, 1); | |
293 | INIT_VLC(&vc9_cbpcy_i_vlc, VC9_CBPCY_I_VLC_BITS, 64, | |
294 | vc9_cbpcy_i_bits, 1, 1, | |
295 | vc9_cbpcy_i_codes, 2, 2, 1); | |
296 | INIT_VLC(&vc9_imode_vlc, VC9_IMODE_VLC_BITS, 7, | |
297 | vc9_imode_bits, 1, 1, | |
298 | vc9_imode_codes, 1, 1, 1); | |
299 | for(i=0; i<3; i++) | |
300 | { | |
301 | INIT_VLC(&vc9_4mv_block_pattern_vlc[i], VC9_4MV_BLOCK_PATTERN_VLC_BITS, 16, | |
302 | vc9_4mv_block_pattern_bits[i], 1, 1, | |
303 | vc9_4mv_block_pattern_codes[i], 1, 1, 1); | |
304 | INIT_VLC(&vc9_cbpcy_p_vlc[i], VC9_CBPCY_P_VLC_BITS, 64, | |
305 | vc9_cbpcy_p_bits[i], 1, 1, | |
306 | vc9_cbpcy_p_codes[i], 2, 2, 1); | |
307 | } | |
308 | for (i=0; i<2; i++) | |
309 | { | |
310 | INIT_VLC(&vc9_mv_diff_vlc[i], VC9_MV_DIFF_VLC_BITS, 73, | |
311 | vc9_mv_diff_bits[i], 1, 1, | |
312 | vc9_mv_diff_codes[i], 2, 2, 1); | |
313 | INIT_VLC(&vc9_luma_dc_vlc[i], VC9_LUMA_DC_VLC_BITS, 26, | |
314 | vc9_luma_dc_bits[i], 1, 1, | |
315 | vc9_luma_dc_codes[i], 4, 4, 1); | |
316 | INIT_VLC(&vc9_ttmb_vlc[i], VC9_TTMB_VLC_BITS, 16, | |
317 | vc9_ttmb_bits[i], 1, 1, | |
318 | vc9_ttmb_codes[i], 2, 2, 1); | |
319 | } | |
320 | } | |
321 | ||
322 | return 0; | |
323 | } | |
324 | ||
325 | #if HAS_ADVANCED_PROFILE | |
326 | static int decode_hrd(VC9Context *v, GetBitContext *gb) | |
327 | { | |
328 | int i, num; | |
329 | ||
330 | num = get_bits(gb, 5); | |
331 | ||
332 | if (v->hrd_rate || num != v->hrd_num_leaky_buckets) | |
333 | { | |
334 | av_free(v->hrd_rate); | |
335 | v->hrd_rate = NULL; | |
336 | } | |
337 | if (!v->hrd_rate) v->hrd_rate = av_malloc(num); | |
338 | if (!v->hrd_rate) return -1; | |
339 | ||
340 | if (v->hrd_buffer || num != v->hrd_num_leaky_buckets) | |
341 | { | |
342 | av_free(v->hrd_buffer); | |
343 | v->hrd_buffer = NULL; | |
344 | } | |
345 | if (!v->hrd_buffer) v->hrd_buffer = av_malloc(num); | |
346 | if (!v->hrd_buffer) return -1; | |
347 | ||
348 | v->hrd_num_leaky_buckets = num; | |
349 | ||
350 | //exponent in base-2 for rate | |
351 | v->bit_rate_exponent = get_bits(gb, 4); | |
352 | //exponent in base-2 for buffer_size | |
353 | v->buffer_size_exponent = get_bits(gb, 4); | |
354 | ||
355 | for (i=0; i<num; i++) | |
356 | { | |
357 | //mantissae, ordered (if not, use a function ? | |
358 | v->hrd_rate[i] = get_bits(gb, 16); | |
359 | if (i && v->hrd_rate[i-1]>=v->hrd_rate[i]) | |
360 | { | |
361 | av_log(v, AV_LOG_ERROR, "HDR Rates aren't strictly increasing:" | |
362 | "%i vs %i\n", v->hrd_rate[i-1], v->hrd_rate[i]); | |
363 | return -1; | |
364 | } | |
365 | v->hrd_buffer[i] = get_bits(gb, 16); | |
366 | if (i && v->hrd_buffer[i-1]<v->hrd_buffer[i]) | |
367 | { | |
368 | av_log(v, AV_LOG_ERROR, "HDR Buffers aren't decreasing:" | |
369 | "%i vs %i\n", v->hrd_buffer[i-1], v->hrd_buffer[i]); | |
370 | return -1; | |
371 | } | |
372 | } | |
373 | return 0; | |
374 | } | |
375 | ||
376 | static int decode_advanced_sequence_header(AVCodecContext *avctx, GetBitContext *gb) | |
377 | { | |
378 | VC9Context *v = avctx->priv_data; | |
379 | int nr, dr, aspect_ratio; | |
380 | ||
381 | v->postprocflag = get_bits(gb, 1); | |
382 | v->broadcast = get_bits(gb, 1); | |
383 | v->interlace = get_bits(gb, 1); | |
384 | ||
385 | v->tfcntrflag = get_bits(gb, 1); | |
386 | v->finterpflag = get_bits(gb, 1); //common | |
387 | v->panscanflag = get_bits(gb, 1); | |
388 | v->reserved = get_bits(gb, 1); | |
389 | if (v->reserved) | |
390 | { | |
391 | av_log(avctx, AV_LOG_ERROR, "RESERVED should be 0 (is %i)\n", | |
392 | v->reserved); | |
393 | return -1; | |
394 | } | |
395 | if (v->extended_mv) | |
396 | v->extended_dmv = get_bits(gb, 1); | |
397 | ||
398 | if (get_bits(gb, 1) /* pic_size_flag */) | |
399 | { | |
400 | avctx->coded_width = get_bits(gb, 12); | |
401 | avctx->coded_height = get_bits(gb, 12); | |
402 | if ( get_bits(gb, 1) /* disp_size_flag */) | |
403 | { | |
404 | avctx->width = get_bits(gb, 14); | |
405 | avctx->height = get_bits(gb, 14); | |
406 | } | |
407 | ||
408 | if ( get_bits(gb, 1) /* aspect_ratio_flag */) | |
409 | { | |
410 | aspect_ratio = get_bits(gb, 4); //SAR | |
411 | if (aspect_ratio == 0x0F) //FF_ASPECT_EXTENDED | |
412 | { | |
413 | avctx->sample_aspect_ratio.num = get_bits(gb, 8); | |
414 | avctx->sample_aspect_ratio.den = get_bits(gb, 8); | |
415 | } | |
416 | else if (aspect_ratio == 0x0E) | |
417 | { | |
418 | av_log(avctx, AV_LOG_DEBUG, "Reserved AR found\n"); | |
419 | } | |
420 | else | |
421 | { | |
422 | avctx->sample_aspect_ratio = vc9_pixel_aspect[aspect_ratio]; | |
423 | } | |
424 | } | |
425 | } | |
426 | else | |
427 | { | |
428 | avctx->coded_width = avctx->width; | |
429 | avctx->coded_height = avctx->height; | |
430 | } | |
431 | ||
432 | if ( get_bits(gb, 1) /* framerateflag */) | |
433 | { | |
434 | if ( get_bits(gb, 1) /* framerateind */) | |
435 | { | |
436 | nr = get_bits(gb, 8); | |
437 | dr = get_bits(gb, 4); | |
438 | if (nr<1) | |
439 | { | |
440 | av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATENR\n"); | |
441 | return -1; | |
442 | } | |
443 | if (nr>5) | |
444 | { | |
445 | av_log(avctx, AV_LOG_ERROR, | |
446 | "Reserved FRAMERATENR %i not handled\n", nr); | |
447 | } | |
448 | if (dr<1) | |
449 | { | |
450 | av_log(avctx, AV_LOG_ERROR, "0 is forbidden for FRAMERATEDR\n"); | |
451 | } | |
452 | if (dr>2) | |
453 | { | |
454 | av_log(avctx, AV_LOG_ERROR, | |
455 | "Reserved FRAMERATEDR %i not handled\n", dr); | |
456 | } | |
457 | avctx->frame_rate_base = fps_nr[dr]; | |
458 | avctx->frame_rate = fps_nr[nr]; | |
459 | } | |
460 | else | |
461 | { | |
462 | nr = get_bits(gb, 16); | |
463 | // 0.03125->2048Hz / 0.03125Hz | |
464 | avctx->frame_rate = 1000000; | |
465 | avctx->frame_rate_base = 31250*(1+nr); | |
466 | } | |
467 | } | |
468 | ||
469 | if ( get_bits(gb, 1) /* color_format_flag */) | |
470 | { | |
471 | //Chromacity coordinates of color primaries | |
472 | //like ITU-R BT.709-2, BT.470-2, ... | |
473 | v->color_prim = get_bits(gb, 8); | |
474 | if (v->color_prim<1) | |
475 | { | |
476 | av_log(avctx, AV_LOG_ERROR, "0 for COLOR_PRIM is reserved\n"); | |
477 | return -1; | |
478 | } | |
479 | if (v->color_prim == 3 || v->color_prim>6) | |
480 | { | |
481 | av_log(avctx, AV_LOG_DEBUG, "Reserved COLOR_PRIM %i found\n", | |
482 | v->color_prim); | |
483 | return -1; | |
484 | } | |
485 | ||
486 | //Opto-electronic transfer characteristics | |
487 | v->transfer_char = get_bits(gb, 8); | |
488 | if (v->transfer_char == 3 || v->transfer_char>8) | |
489 | { | |
490 | av_log(avctx, AV_LOG_DEBUG, "Reserved TRANSFERT_CHAR %i found\n", | |
491 | v->color_prim); | |
492 | return -1; | |
493 | } | |
494 | ||
495 | //Matrix coefficient for primariev->YCbCr | |
496 | v->matrix_coef = get_bits(gb, 8); | |
497 | if (v->matrix_coef < 1) return -1; //forbidden | |
498 | if ((v->matrix_coef>3 && v->matrix_coef<6) || v->matrix_coef>7) | |
499 | { | |
500 | av_log(avctx, AV_LOG_DEBUG, "Reserved MATRIX_COEF %i found\n", | |
501 | v->color_prim); | |
502 | return -1; | |
503 | } | |
504 | } | |
505 | ||
506 | //Hypothetical reference decoder indicator flag | |
507 | v->hrd_param_flag = get_bits(gb, 1); | |
508 | if (v->hrd_param_flag) | |
509 | { | |
510 | if (decode_hrd(v, gb) < 0) return -1; | |
511 | } | |
512 | ||
513 | av_log(avctx, AV_LOG_DEBUG, "Advanced profile not supported yet\n"); | |
514 | return -1; | |
515 | } | |
516 | #endif | |
517 | ||
518 | static int decode_sequence_header(AVCodecContext *avctx, GetBitContext *gb) | |
519 | { | |
520 | VC9Context *v = avctx->priv_data; | |
521 | ||
522 | v->profile = get_bits(gb, 2); | |
523 | av_log(avctx, AV_LOG_DEBUG, "Profile: %i\n", v->profile); | |
524 | ||
525 | #if HAS_ADVANCED_PROFILE | |
526 | if (v->profile > PROFILE_MAIN) | |
527 | { | |
528 | v->level = get_bits(gb, 3); | |
529 | v->chromaformat = get_bits(gb, 2); | |
530 | if (v->chromaformat != 1) | |
531 | { | |
532 | av_log(avctx, AV_LOG_ERROR, | |
533 | "Only 4:2:0 chroma format supported\n"); | |
534 | return -1; | |
535 | } | |
536 | } | |
537 | else | |
538 | #endif | |
539 | { | |
540 | v->res_sm = get_bits(gb, 2); //reserved | |
541 | if (v->res_sm) | |
542 | { | |
543 | av_log(avctx, AV_LOG_ERROR, | |
544 | "Reserved RES_SM=%i is forbidden\n", v->res_sm); | |
545 | //return -1; | |
546 | } | |
547 | } | |
548 | ||
549 | // (fps-2)/4 (->30) | |
550 | v->frmrtq_postproc = get_bits(gb, 3); //common | |
551 | // (bitrate-32kbps)/64kbps | |
552 | v->bitrtq_postproc = get_bits(gb, 5); //common | |
553 | v->loopfilter = get_bits(gb, 1); //common | |
554 | ||
555 | #if HAS_ADVANCED_PROFILE | |
556 | if (v->profile <= PROFILE_MAIN) | |
557 | #endif | |
558 | { | |
559 | v->res_x8 = get_bits(gb, 1); //reserved | |
560 | if (v->res_x8) | |
561 | { | |
562 | av_log(avctx, AV_LOG_ERROR, | |
563 | "1 for reserved RES_X8 is forbidden\n"); | |
564 | return -1; | |
565 | } | |
566 | v->multires = get_bits(gb, 1); | |
567 | v->res_fasttx = get_bits(gb, 1); | |
568 | if (!v->res_fasttx) | |
569 | { | |
570 | av_log(avctx, AV_LOG_ERROR, | |
571 | "0 for reserved RES_FASTTX is forbidden\n"); | |
572 | //return -1; | |
573 | } | |
574 | } | |
575 | ||
576 | v->fastuvmc = get_bits(gb, 1); //common | |
577 | if (!v->profile && !v->fastuvmc) | |
578 | { | |
579 | av_log(avctx, AV_LOG_ERROR, | |
580 | "FASTUVMC unavailable in Simple Profile\n"); | |
581 | return -1; | |
582 | } | |
583 | v->extended_mv = get_bits(gb, 1); //common | |
584 | if (!v->profile && v->extended_mv) | |
585 | { | |
586 | av_log(avctx, AV_LOG_ERROR, | |
587 | "Extended MVs unavailable in Simple Profile\n"); | |
588 | return -1; | |
589 | } | |
590 | v->dquant = get_bits(gb, 2); //common | |
591 | v->vstransform = get_bits(gb, 1); //common | |
592 | ||
593 | #if HAS_ADVANCED_PROFILE | |
594 | if (v->profile <= PROFILE_MAIN) | |
595 | #endif | |
596 | { | |
597 | v->res_transtab = get_bits(gb, 1); | |
598 | if (v->res_transtab) | |
599 | { | |
600 | av_log(avctx, AV_LOG_ERROR, | |
601 | "1 for reserved RES_TRANSTAB is forbidden\n"); | |
602 | return -1; | |
603 | } | |
604 | } | |
605 | ||
606 | v->overlap = get_bits(gb, 1); //common | |
607 | ||
608 | #if HAS_ADVANCED_PROFILE | |
609 | if (v->profile <= PROFILE_MAIN) | |
610 | #endif | |
611 | { | |
612 | v->syncmarker = get_bits(gb, 1); | |
613 | v->rangered = get_bits(gb, 1); | |
614 | } | |
615 | ||
616 | avctx->max_b_frames = get_bits(gb, 3); //common | |
617 | v->quantizer_mode = get_bits(gb, 2); //common | |
618 | ||
619 | #if HAS_ADVANCED_PROFILE | |
620 | if (v->profile <= PROFILE_MAIN) | |
621 | #endif | |
622 | { | |
623 | v->finterpflag = get_bits(gb, 1); //common | |
624 | v->res_rtm_flag = get_bits(gb, 1); //reserved | |
625 | if (!v->res_rtm_flag) | |
626 | { | |
627 | av_log(avctx, AV_LOG_ERROR, | |
628 | "0 for reserved RES_RTM_FLAG is forbidden\n"); | |
629 | //return -1; | |
630 | } | |
631 | #if TRACE | |
632 | av_log(avctx, AV_LOG_INFO, | |
633 | "Profile %i:\nfrmrtq_postproc=%i, bitrtq_postproc=%i\n" | |
634 | "LoopFilter=%i, MultiRes=%i, FastUVMV=%i, Extended MV=%i\n" | |
635 | "Rangered=%i, VSTransform=%i, Overlap=%i, SyncMarker=%i\n" | |
636 | "DQuant=%i, Quantizer mode=%i, Max B frames=%i\n", | |
637 | v->profile, v->frmrtq_postproc, v->bitrtq_postproc, | |
638 | v->loopfilter, v->multires, v->fastuvmc, v->extended_mv, | |
639 | v->rangered, v->vstransform, v->overlap, v->syncmarker, | |
640 | v->dquant, v->quantizer_mode, avctx->max_b_frames | |
641 | ); | |
642 | #endif | |
643 | } | |
644 | #if HAS_ADVANCED_PROFILE | |
645 | else decode_advanced_sequence_header(avctx, gb); | |
646 | #endif | |
647 | } | |
648 | ||
649 | ||
650 | #if HAS_ADVANCED_PROFILE | |
651 | /*****************************************************************************/ | |
652 | /* Entry point decoding (Advanced Profile) */ | |
653 | /*****************************************************************************/ | |
654 | static int advanced_entry_point_process(AVCodecContext *avctx, GetBitContext *gb) | |
655 | { | |
656 | VC9Context *v = avctx->priv_data; | |
657 | int range_mapy_flag, range_mapuv_flag, i; | |
658 | if (v->profile != PROFILE_ADVANCED) | |
659 | { | |
660 | av_log(avctx, AV_LOG_ERROR, | |
661 | "Entry point are only defined in Advanced Profile!\n"); | |
662 | return -1; //Only for advanced profile! | |
663 | } | |
664 | if (v->hrd_param_flag) | |
665 | { | |
666 | //Update buffer fullness | |
667 | av_log(avctx, AV_LOG_DEBUG, "Buffer fullness update\n"); | |
668 | for (i=0; i<v->hrd_num_leaky_buckets; i++) | |
669 | skip_bits(gb, 8); | |
670 | } | |
671 | if ((range_mapy_flag = get_bits(gb, 1))) | |
672 | { | |
673 | //RANGE_MAPY | |
674 | av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPY\n"); | |
675 | skip_bits(gb, 3); | |
676 | } | |
677 | if ((range_mapuv_flag = get_bits(gb, 1))) | |
678 | { | |
679 | //RANGE_MAPUV | |
680 | av_log(avctx, AV_LOG_DEBUG, "RANGE_MAPUV\n"); | |
681 | skip_bits(gb, 3); | |
682 | } | |
683 | if (v->panscanflag) | |
684 | { | |
685 | //NUMPANSCANWIN | |
686 | v->numpanscanwin = get_bits(gb, 3); | |
687 | av_log(avctx, AV_LOG_DEBUG, "NUMPANSCANWIN: %u\n", v->numpanscanwin); | |
688 | } | |
689 | return 0; | |
690 | } | |
691 | #endif | |
692 | ||
693 | /******************************************************************************/ | |
694 | /* Bitplane decoding */ | |
695 | /******************************************************************************/ | |
696 | #define IMODE_RAW 0 | |
697 | #define IMODE_NORM2 1 | |
698 | #define IMODE_DIFF2 2 | |
699 | #define IMODE_NORM6 3 | |
700 | #define IMODE_DIFF6 4 | |
701 | #define IMODE_ROWSKIP 5 | |
702 | #define IMODE_COLSKIP 6 | |
703 | ||
704 | //FIXME optimize | |
705 | static int bitplane_decoding(uint8_t* plane, int width, int height, VC9Context *v) | |
706 | { | |
707 | int imode, x, y, code; | |
708 | uint8_t invert, *planep = plane; | |
709 | ||
710 | invert = get_bits(&v->gb, 1); | |
711 | imode = get_vlc2(&v->gb, vc9_imode_vlc.table, VC9_IMODE_VLC_BITS, 2); | |
712 | av_log(v->avctx, AV_LOG_DEBUG, "Bitplane: imode=%i, invert=%i\n", | |
713 | imode, invert); | |
714 | ||
715 | switch (imode) | |
716 | { | |
717 | case IMODE_RAW: | |
718 | for (y=0; y<height; y++) | |
719 | { | |
720 | for (x=0; x<width; x++) | |
721 | planep[x] = (-get_bits(&v->gb, 1)); //-1=0xFF | |
722 | planep += width; | |
723 | } | |
724 | break; | |
725 | case IMODE_DIFF2: | |
726 | case IMODE_NORM2: | |
727 | if ((height*width) & 1) *(++planep) = get_bits(&v->gb, 1); | |
728 | code = get_vlc2(&v->gb, vc9_norm2_vlc.table, VC9_NORM2_VLC_BITS, 2); | |
729 | *(++planep) = code&1; //lsb => left | |
730 | *(++planep) = code&2; //msb => right - this is a bitplane, so only !0 matters | |
731 | break; | |
732 | case IMODE_DIFF6: | |
733 | case IMODE_NORM6: | |
734 | av_log(v->avctx, AV_LOG_ERROR, "Imode using Norm-6 isn't supported\n"); | |
735 | return -1; | |
736 | break; | |
737 | case IMODE_ROWSKIP: | |
738 | for (y=0; y<height; y++) | |
739 | { | |
740 | if (get_bits(&v->gb, 1)) //rowskip | |
741 | memset(planep, 0, width); | |
742 | else for (x=0; x<width; x++) planep[x] = get_bits(&v->gb, 1); | |
743 | planep += width; | |
744 | } | |
745 | break; | |
746 | case IMODE_COLSKIP: //Teh ugly | |
747 | for (x=0; x<width; x++) | |
748 | { | |
749 | planep = plane; | |
750 | if (get_bits(&v->gb, 1)) //colskip | |
751 | { | |
752 | for (y=0; y<height; y++) | |
753 | { | |
754 | planep[x] = 0; | |
755 | planep += width; | |
756 | } | |
757 | } | |
758 | else | |
759 | { | |
760 | for (y=0; y<height; y++) | |
761 | { | |
762 | planep[x] = get_bits(&v->gb, 1); | |
763 | planep += width; | |
764 | } | |
765 | } | |
766 | } | |
767 | break; | |
768 | default: break; | |
769 | } | |
770 | ||
771 | /* Applying diff operator */ | |
772 | if (imode == IMODE_DIFF2 || imode == IMODE_DIFF2) | |
773 | { | |
774 | /* And what about j=0 !? */ | |
775 | planep = plane + width; | |
776 | memset(plane, invert, width); | |
777 | for (y=0; y<height; y++) | |
778 | { | |
779 | planep[0] = planep[-width]; | |
780 | for (x=1; x<width; x++) | |
781 | { | |
782 | if (planep[x-1] != planep[-width]) planep[x] = invert; | |
783 | else planep[x] = planep[x-1]; | |
784 | } | |
785 | planep += width; | |
786 | } | |
787 | } | |
788 | else if (invert) | |
789 | { | |
790 | planep = plane; | |
791 | for (x=0; x<width*height; x++) planep[x] = !planep[x]; | |
792 | } | |
793 | return 0; | |
794 | } | |
795 | ||
796 | /*****************************************************************************/ | |
797 | /* VOP Dquant decoding */ | |
798 | /*****************************************************************************/ | |
799 | static int vop_dquant_decoding(VC9Context *v) | |
800 | { | |
801 | int pqdiff; | |
802 | ||
803 | //variable size | |
804 | if (v->dquant == 2) | |
805 | { | |
806 | pqdiff = get_bits(&v->gb, 3); | |
807 | if (pqdiff == 7) v->altpq = get_bits(&v->gb, 5); | |
808 | else v->altpq = v->pq + pqdiff + 1; | |
809 | } | |
810 | else | |
811 | { | |
812 | v->dquantfrm = get_bits(&v->gb, 1); | |
813 | if ( v->dquantfrm ) | |
814 | { | |
815 | v->dqprofile = get_bits(&v->gb, 2); | |
816 | switch (v->dqprofile) | |
817 | { | |
818 | case DQPROFILE_SINGLE_EDGE: | |
819 | case DQPROFILE_DOUBLE_EDGES: | |
820 | v->dqsbedge = get_bits(&v->gb, 2); | |
821 | break; | |
822 | case DQPROFILE_ALL_MBS: | |
823 | v->dqbilevel = get_bits(&v->gb, 1); | |
824 | default: break; //Forbidden ? | |
825 | } | |
826 | if (!v->dqbilevel || v->dqprofile != DQPROFILE_ALL_MBS) | |
827 | { | |
828 | pqdiff = get_bits(&v->gb, 3); | |
829 | if (pqdiff == 7) v->altpq = get_bits(&v->gb, 5); | |
830 | else v->altpq = v->pq + pqdiff + 1; | |
831 | } | |
832 | } | |
833 | } | |
834 | return 0; | |
835 | } | |
836 | ||
837 | /*****************************************************************************/ | |
838 | /* All Profiles picture header decoding specific functions */ | |
839 | /* Only pro/epilog differs between Simple/Main and Advanced => check caller */ | |
840 | /*****************************************************************************/ | |
841 | static int decode_bi_picture_header(VC9Context *v) | |
842 | { | |
843 | /* Very particular case: | |
844 | - for S/M Profiles, decode_b_picture_header reads BF, | |
845 | bfraction then determine if this is a BI frame, calling | |
846 | this function afterwards | |
847 | - for A Profile, PTYPE already tells so and we can go | |
848 | directly there | |
849 | */ | |
850 | int pqindex; | |
851 | ||
852 | /* Read the quantization stuff */ | |
853 | pqindex = get_bits(&v->gb, 5); | |
854 | if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
855 | v->pq = pquant_table[0][pqindex]; | |
856 | else | |
857 | { | |
858 | v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
859 | } | |
860 | if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); | |
861 | if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
862 | v->pquantizer = get_bits(&v->gb, 1); | |
863 | ||
864 | /* Read the MV type/mode */ | |
865 | if (v->extended_mv == 1) | |
866 | v->mvrange = get_prefix(&v->gb, 0, 3); | |
867 | ||
868 | /* FIXME: what table are used in that case ? */ | |
869 | v->mv_diff_vlc = &vc9_mv_diff_vlc[0]; | |
870 | v->cbpcy_vlc = &vc9_cbpcy_i_vlc; | |
871 | ||
872 | av_log(v->avctx, AV_LOG_DEBUG, "B frame, QP=%i\n", v->pq); | |
873 | av_log(v->avctx, AV_LOG_ERROR, "BI_TYPE not supported yet\n"); | |
874 | /* Epilog should be done in caller */ | |
875 | return -1; | |
876 | } | |
877 | ||
878 | static int decode_b_picture_header(VC9Context *v) | |
879 | { | |
880 | int pqindex; | |
881 | ||
882 | /* Prolog common to all frametypes should be done in caller */ | |
883 | if (v->profile == PROFILE_SIMPLE) | |
884 | { | |
885 | av_log(v, AV_LOG_ERROR, "Found a B frame while in Simple Profile!\n"); | |
886 | return FRAME_SKIPED; | |
887 | } | |
888 | ||
889 | v->bfraction = vc9_bfraction_lut[get_vlc2(&v->gb, vc9_bfraction_vlc.table, | |
890 | VC9_BFRACTION_VLC_BITS, 2)]; | |
891 | if (v->bfraction < -1) | |
892 | { | |
893 | av_log(v, AV_LOG_ERROR, "Invalid BFRaction\n"); | |
894 | return FRAME_SKIPED; | |
895 | } | |
896 | else if (!v->bfraction) | |
897 | { | |
898 | /* We actually have a BI frame */ | |
899 | return decode_bi_picture_header(v); | |
900 | } | |
901 | ||
902 | /* Read the quantization stuff */ | |
903 | pqindex = get_bits(&v->gb, 5); | |
904 | if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
905 | v->pq = pquant_table[0][pqindex]; | |
906 | else | |
907 | { | |
908 | v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
909 | } | |
910 | if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); | |
911 | if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
912 | v->pquantizer = get_bits(&v->gb, 1); | |
913 | ||
914 | /* Read the MV type/mode */ | |
915 | if (v->extended_mv == 1) | |
916 | v->mvrange = get_prefix(&v->gb, 0, 3); | |
917 | v->mv_mode = get_bits(&v->gb, 1); | |
918 | if (v->pq < 13) | |
919 | { | |
920 | if (!v->mv_mode) | |
921 | { | |
922 | v->mv_mode = get_bits(&v->gb, 2); | |
923 | if (v->mv_mode) | |
924 | av_log(v, AV_LOG_ERROR, | |
925 | "mv_mode for lowquant B frame was %i\n", v->mv_mode); | |
926 | } | |
927 | } | |
928 | else | |
929 | { | |
930 | if (!v->mv_mode) | |
931 | { | |
932 | if (get_bits(&v->gb, 1)) | |
933 | av_log(v, AV_LOG_ERROR, | |
934 | "mv_mode for highquant B frame was %i\n", v->mv_mode); | |
935 | } | |
936 | v->mv_mode = 1-v->mv_mode; //To match (pq < 13) mapping | |
937 | } | |
938 | ||
939 | if (v->mv_mode == MV_PMODE_MIXED_MV) | |
940 | { | |
941 | if (bitplane_decoding( v->mv_type_mb_plane, v->width_mb, | |
942 | v->height_mb, v)<0) | |
943 | return -1; | |
944 | } | |
945 | ||
946 | //bitplane | |
947 | bitplane_decoding(v->direct_mb_plane, v->width_mb, v->height_mb, v); | |
948 | bitplane_decoding(v->skip_mb_plane, v->width_mb, v->height_mb, v); | |
949 | ||
950 | /* FIXME: what is actually chosen for B frames ? */ | |
951 | v->mv_diff_vlc = &vc9_mv_diff_vlc[get_bits(&v->gb, 2)]; | |
952 | v->cbpcy_vlc = &vc9_cbpcy_p_vlc[get_bits(&v->gb, 2)]; | |
953 | if (v->dquant) | |
954 | { | |
955 | vop_dquant_decoding(v); | |
956 | } | |
957 | ||
958 | if (v->vstransform) | |
959 | { | |
960 | v->ttmbf = get_bits(&v->gb, 1); | |
961 | if (v->ttmbf) | |
962 | { | |
963 | v->ttfrm = get_bits(&v->gb, 2); | |
964 | av_log(v, AV_LOG_INFO, "Transform used: %ix%i\n", | |
965 | (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); | |
966 | } | |
967 | } | |
968 | /* Epilog should be done in caller */ | |
969 | return 0; | |
970 | } | |
971 | ||
972 | static int decode_i_picture_header(VC9Context *v) | |
973 | { | |
974 | int pqindex, status = 0, ac_pred, condover; | |
975 | ||
976 | /* Prolog common to all frametypes should be done in caller */ | |
977 | //BF = Buffer Fullness | |
978 | if (v->profile <= PROFILE_MAIN && get_bits(&v->gb, 7)) | |
979 | { | |
980 | av_log(v, AV_LOG_DEBUG, "I BufferFullness not 0\n"); | |
981 | } | |
982 | ||
983 | /* Quantizer stuff */ | |
984 | pqindex = get_bits(&v->gb, 5); | |
985 | if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
986 | v->pq = pquant_table[0][pqindex]; | |
987 | else | |
988 | { | |
989 | v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
990 | } | |
991 | if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); | |
992 | if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
993 | v->pquantizer = get_bits(&v->gb, 1); | |
994 | av_log(v->avctx, AV_LOG_DEBUG, "I frame: QP=%i (+%i/2)\n", | |
995 | v->pq, v->halfpq); | |
996 | #if HAS_ADVANCED_PROFILE | |
997 | if (v->profile <= PROFILE_MAIN) | |
998 | #endif | |
999 | { | |
1000 | if (v->extended_mv) v->mvrange = get_prefix(&v->gb, 0, 3); | |
1001 | if (v->multires) v->respic = get_bits(&v->gb, 2); | |
1002 | } | |
1003 | #if HAS_ADVANCED_PROFILE | |
1004 | else | |
1005 | { | |
1006 | ac_pred = get_bits(&v->gb, 1); | |
1007 | if (v->postprocflag) v->postproc = get_bits(&v->gb, 1); | |
1008 | /* 7.1.1.34 + 8.5.2 */ | |
1009 | if (v->overlap && v->pq<9) | |
1010 | { | |
1011 | condover = get_bits(&v->gb, 1); | |
1012 | if (condover) | |
1013 | { | |
1014 | condover = 2+get_bits(&v->gb, 1); | |
1015 | if (condover == 3) | |
1016 | status = bitplane_decoding(v->over_flags_plane, | |
1017 | v->width_mb, v->height_mb, v); | |
1018 | } | |
1019 | } | |
1020 | } | |
1021 | #endif | |
1022 | ||
1023 | /* Epilog should be done in caller */ | |
1024 | return status; | |
1025 | } | |
1026 | ||
1027 | static int decode_p_picture_header(VC9Context *v) | |
1028 | { | |
1029 | /* INTERFRM, FRMCNT, RANGEREDFRM read in caller */ | |
1030 | int lowquant, pqindex; | |
1031 | ||
1032 | pqindex = get_bits(&v->gb, 5); | |
1033 | if (v->quantizer_mode == QUANT_FRAME_IMPLICIT) | |
1034 | v->pq = pquant_table[0][pqindex]; | |
1035 | else | |
1036 | { | |
1037 | v->pq = pquant_table[v->quantizer_mode-1][pqindex]; | |
1038 | } | |
1039 | if (pqindex < 9) v->halfpq = get_bits(&v->gb, 1); | |
1040 | if (v->quantizer_mode == QUANT_FRAME_EXPLICIT) | |
1041 | v->pquantizer = get_bits(&v->gb, 1); | |
1042 | av_log(v->avctx, AV_LOG_DEBUG, "P Frame: QP=%i (+%i/2)\n", | |
1043 | v->pq, v->halfpq); | |
1044 | if (v->extended_mv == 1) v->mvrange = get_prefix(&v->gb, 0, 3); | |
1045 | #if HAS_ADVANCED_PROFILE | |
1046 | if (v->profile > PROFILE_MAIN) | |
1047 | { | |
1048 | if (v->postprocflag) v->postproc = get_bits(&v->gb, 1); | |
1049 | } | |
1050 | else | |
1051 | #endif | |
1052 | if (v->multires) v->respic = get_bits(&v->gb, 2); | |
1053 | lowquant = (v->pquantizer>12) ? 0 : 1; | |
1054 | v->mv_mode = mv_pmode_table[lowquant][get_prefix(&v->gb, 1, 4)]; | |
1055 | if (v->mv_mode == MV_PMODE_INTENSITY_COMP) | |
1056 | { | |
1057 | v->mv_mode2 = mv_pmode_table[lowquant][get_prefix(&v->gb, 1, 3)]; | |
1058 | v->lumscale = get_bits(&v->gb, 6); | |
1059 | v->lumshift = get_bits(&v->gb, 6); | |
1060 | } | |
1061 | ||
1062 | if ((v->mv_mode == MV_PMODE_INTENSITY_COMP && | |
1063 | v->mv_mode2 == MV_PMODE_MIXED_MV) | |
1064 | || v->mv_mode == MV_PMODE_MIXED_MV) | |
1065 | { | |
1066 | if (bitplane_decoding(v->mv_type_mb_plane, v->width_mb, | |
1067 | v->height_mb, v) < 0) | |
1068 | return -1; | |
1069 | } | |
1070 | ||
1071 | if (bitplane_decoding(v->skip_mb_plane, v->width_mb, | |
1072 | v->height_mb, v) < 0) | |
1073 | return -1; | |
1074 | ||
1075 | /* Hopefully this is correct for P frames */ | |
1076 | v->mv_diff_vlc = &vc9_mv_diff_vlc[get_bits(&v->gb, 2)]; | |
1077 | v->cbpcy_vlc = &vc9_cbpcy_p_vlc[get_bits(&v->gb, 2)]; | |
1078 | ||
1079 | if (v->dquant) | |
1080 | { | |
1081 | av_log(v->avctx, AV_LOG_INFO, "VOP DQuant info\n"); | |
1082 | vop_dquant_decoding(v); | |
1083 | } | |
1084 | ||
1085 | if (v->vstransform) | |
1086 | { | |
1087 | v->ttmbf = get_bits(&v->gb, 1); | |
1088 | if (v->ttmbf) | |
1089 | { | |
1090 | v->ttfrm = get_bits(&v->gb, 2); | |
1091 | av_log(v->avctx, AV_LOG_INFO, "Transform used: %ix%i\n", | |
1092 | (v->ttfrm & 2) ? 4 : 8, (v->ttfrm & 1) ? 4 : 8); | |
1093 | } | |
1094 | } | |
1095 | /* Epilog should be done in caller */ | |
1096 | return 0; | |
1097 | } | |
1098 | ||
1099 | ||
1100 | static int standard_decode_picture_header(VC9Context *v) | |
1101 | { | |
1102 | int status = 0; | |
1103 | ||
1104 | if (v->finterpflag) v->interpfrm = get_bits(&v->gb, 1); | |
1105 | skip_bits(&v->gb, 2); //framecnt unused | |
1106 | if (v->rangered) v->rangeredfrm = get_bits(&v->gb, 1); | |
1107 | v->pict_type = get_bits(&v->gb, 1); | |
1108 | if (v->avctx->max_b_frames && !v->pict_type) | |
1109 | { | |
1110 | if (get_bits(&v->gb, 1)) v->pict_type = I_TYPE; | |
1111 | else v->pict_type = P_TYPE; | |
1112 | } | |
1113 | else v->pict_type++; //P_TYPE | |
1114 | ||
1115 | switch (v->pict_type) | |
1116 | { | |
1117 | case I_TYPE: status = decode_i_picture_header(v); break; | |
1118 | case BI_TYPE: status = decode_b_picture_header(v); break; | |
1119 | case P_TYPE: status = decode_p_picture_header(v); break; | |
1120 | case B_TYPE: status = decode_b_picture_header(v); break; | |
1121 | } | |
1122 | ||
1123 | if (status == FRAME_SKIPED) | |
1124 | { | |
1125 | av_log(v, AV_LOG_INFO, "Skipping frame...\n"); | |
1126 | return status; | |
1127 | } | |
1128 | ||
1129 | /* AC/DC Syntax */ | |
1130 | v->transacfrm = get_bits(&v->gb, 1); | |
1131 | if (v->transacfrm) v->transacfrm += get_bits(&v->gb, 1); | |
1132 | if (v->pict_type == I_TYPE || v->pict_type == BI_TYPE) | |
1133 | { | |
1134 | v->transacfrm2 = get_bits(&v->gb, 1); | |
1135 | if (v->transacfrm2) v->transacfrm2 += get_bits(&v->gb, 1); | |
1136 | } | |
1137 | v->transacdctab = get_bits(&v->gb, 1); | |
1138 | ||
1139 | return 0; | |
1140 | } | |
1141 | ||
1142 | ||
1143 | #if HAS_ADVANCED_PROFILE | |
1144 | /******************************************************************************/ | |
1145 | /* Advanced Profile picture header decoding specific functions */ | |
1146 | /******************************************************************************/ | |
1147 | static int advanced_decode_picture_header(VC9Context *v) | |
1148 | { | |
1149 | static const int type_table[4] = { P_TYPE, B_TYPE, I_TYPE, BI_TYPE }; | |
1150 | int type, i, ret; | |
1151 | ||
1152 | if (v->interlace) | |
1153 | { | |
1154 | v->fcm = get_bits(&v->gb, 1); | |
1155 | if (v->fcm) v->fcm = 2+get_bits(&v->gb, 1); | |
1156 | } | |
1157 | ||
1158 | type = get_prefix(&v->gb, 0, 4); | |
1159 | if (type > 4 || type < 0) return FRAME_SKIPED; | |
1160 | v->pict_type = type_table[type]; | |
1161 | av_log(v->avctx, AV_LOG_INFO, "AP Frame Type: %i\n", v->pict_type); | |
1162 | ||
1163 | if (v->tfcntrflag) v->tfcntr = get_bits(&v->gb, 8); | |
1164 | if (v->broadcast) | |
1165 | { | |
1166 | if (!v->interlace) v->rptfrm = get_bits(&v->gb, 2); | |
1167 | else | |
1168 | { | |
1169 | v->tff = get_bits(&v->gb, 1); | |
1170 | v->rff = get_bits(&v->gb, 1); | |
1171 | } | |
1172 | } | |
1173 | ||
1174 | if (v->panscanflag) | |
1175 | { | |
1176 | #if 0 | |
1177 | for (i=0; i<v->numpanscanwin; i++) | |
1178 | { | |
1179 | v->topleftx[i] = get_bits(&v->gb, 16); | |
1180 | v->toplefty[i] = get_bits(&v->gb, 16); | |
1181 | v->bottomrightx[i] = get_bits(&v->gb, 16); | |
1182 | v->bottomrighty[i] = get_bits(&v->gb, 16); | |
1183 | } | |
1184 | #else | |
1185 | skip_bits(&v->gb, 16*4*v->numpanscanwin); | |
1186 | #endif | |
1187 | } | |
1188 | v->rndctrl = get_bits(&v->gb, 1); | |
1189 | v->uvsamp = get_bits(&v->gb, 1); | |
1190 | if (v->finterpflag == 1) v->interpfrm = get_bits(&v->gb, 1); | |
1191 | ||
1192 | switch(v->pict_type) | |
1193 | { | |
1194 | case I_TYPE: if (decode_i_picture_header(v) < 0) return -1; | |
1195 | case P_TYPE: if (decode_p_picture_header(v) < 0) return -1; | |
1196 | case BI_TYPE: | |
1197 | case B_TYPE: if (decode_b_picture_header(v) < 0) return FRAME_SKIPED; | |
1198 | default: break; | |
1199 | } | |
1200 | ||
1201 | /* AC/DC Syntax */ | |
1202 | v->transacfrm = get_bits(&v->gb, 1); | |
1203 | if (v->transacfrm) v->transacfrm += get_bits(&v->gb, 1); | |
1204 | if (v->pict_type == I_TYPE || v->pict_type == BI_TYPE) | |
1205 | { | |
1206 | v->transacfrm2 = get_bits(&v->gb, 1); | |
1207 | if (v->transacfrm2) v->transacfrm2 += get_bits(&v->gb, 1); | |
1208 | } | |
1209 | v->transacdctab = get_bits(&v->gb, 1); | |
1210 | if (v->pict_type == I_TYPE) vop_dquant_decoding(v); | |
1211 | ||
1212 | return 0; | |
1213 | } | |
1214 | #endif | |
1215 | ||
1216 | /******************************************************************************/ | |
1217 | /* MacroBlock decoding functions */ | |
1218 | /******************************************************************************/ | |
1219 | static int standard_decode_i_mbs(VC9Context *v) | |
1220 | { | |
1221 | int x, y, ac_pred, cbpcy; | |
1222 | ||
1223 | /* Select ttmb table depending on pq */ | |
1224 | if (v->pq < 5) v->ttmb_vlc = &vc9_ttmb_vlc[0]; | |
1225 | else if (v->pq < 13) v->ttmb_vlc = &vc9_ttmb_vlc[1]; | |
1226 | else v->ttmb_vlc = &vc9_ttmb_vlc[2]; | |
1227 | ||
1228 | for (y=0; y<v->height_mb; y++) | |
1229 | { | |
1230 | for (x=0; x<v->width_mb; x++) | |
1231 | { | |
1232 | cbpcy = get_vlc2(&v->gb, vc9_cbpcy_i_vlc.table, | |
1233 | VC9_CBPCY_I_VLC_BITS, 2); | |
1234 | ac_pred = get_bits(&v->gb, 1); | |
1235 | //Decode blocks from that mb wrt cbpcy | |
1236 | } | |
1237 | } | |
1238 | return 0; | |
1239 | } | |
1240 | ||
1241 | #define GET_MQUANT() \ | |
1242 | if (v->dquantfrm) \ | |
1243 | { \ | |
1244 | if (v->dqprofile == DQPROFILE_ALL_MBS) \ | |
1245 | { \ | |
1246 | if (v->dqbilevel) \ | |
1247 | { \ | |
1248 | mquant = (get_bits(&v->gb, 1)) ? v->pq : v->altpq; \ | |
1249 | } \ | |
1250 | else \ | |
1251 | { \ | |
1252 | mqdiff = get_bits(&v->gb, 3); \ | |
1253 | if (mqdiff != 7) mquant = v->pq + mqdiff; \ | |
1254 | else mquant = get_bits(&v->gb, 5); \ | |
1255 | } \ | |
1256 | } \ | |
1257 | } | |
1258 | ||
1259 | /* MVDATA decoding from 8.3.5.2 */ | |
1260 | #define GET_MVDATA() \ | |
1261 | index = 1 + get_vlc2(&v->gb, v->mv_diff_vlc->table, \ | |
1262 | VC9_MV_DIFF_VLC_BITS, 2); \ | |
1263 | if (index > 36) \ | |
1264 | { \ | |
1265 | mb_has_coeffs = 1; \ | |
1266 | index -= 37; \ | |
1267 | } \ | |
1268 | else mb_has_coeffs = 0; \ | |
1269 | mb_is_intra = 0; \ | |
1270 | if (!index) { dmv_x = dmv_y = 0; } \ | |
1271 | else if (index == 35) \ | |
1272 | { \ | |
1273 | dmv_x = get_bits(&v->gb, k_x); \ | |
1274 | dmv_y = get_bits(&v->gb, k_y); \ | |
1275 | mb_is_intra = 1; \ | |
1276 | } \ | |
1277 | else \ | |
1278 | { \ | |
1279 | index1 = index%6; \ | |
1280 | if (hpel_flag && index1 == 5) val = 1; \ | |
1281 | else val = 0; \ | |
1282 | val = get_bits(&v->gb, size_table[index1] - val); \ | |
1283 | sign = 0 - (val&1); \ | |
1284 | dmv_x = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ | |
1285 | \ | |
1286 | index1 = index/6; \ | |
1287 | if (hpel_flag && index1 == 5) val = 1; \ | |
1288 | else val = 0; \ | |
1289 | val = get_bits(&v->gb, size_table[index1] - val); \ | |
1290 | sign = 0 - (val&1); \ | |
1291 | dmv_y = (sign ^ ((val>>1) + offset_table[index1])) - sign; \ | |
1292 | } | |
1293 | ||
1294 | static int decode_p_mbs(VC9Context *v) | |
1295 | { | |
1296 | int x, y, current_mb = 0, i; /* MB/Block Position info */ | |
1297 | int skip_mb_bit = 0, cbpcy; /* MB/B skip */ | |
1298 | int hybrid_pred, ac_pred; /* Prediction types */ | |
1299 | int mb_has_coeffs = 1 /* last_flag */, mb_is_intra; | |
1300 | int dmv_x, dmv_y; /* Differential MV components */ | |
1301 | int mv_mode_bit = 0; /* mv_mode_bit: 1MV=0, 4MV=0 */ | |
1302 | int mqdiff, mquant; /* MB quantization */ | |
1303 | int tt_block; /* MB Transform type */ | |
1304 | static const int size_table[6] = { 0, 2, 3, 4, 5, 8 }, | |
1305 | offset_table[6] = { 0, 1, 3, 7, 15, 31 }; | |
1306 | int k_x, k_y; /* Long MV fixed bitlength */ | |
1307 | int hpel_flag, intra_flag; /* Some MB properties */ | |
1308 | int index, index1; /* LUT indices */ | |
1309 | int val, sign; | |
1310 | ||
1311 | /* Select ttmb table depending on pq */ | |
1312 | if (v->pq < 5) v->ttmb_vlc = &vc9_ttmb_vlc[0]; | |
1313 | else if (v->pq < 13) v->ttmb_vlc = &vc9_ttmb_vlc[1]; | |
1314 | else v->ttmb_vlc = &vc9_ttmb_vlc[2]; | |
1315 | ||
1316 | /* Select proper long MV range */ | |
1317 | switch (v->mvrange) | |
1318 | { | |
1319 | case 1: k_x = 10; k_y = 9; break; | |
1320 | case 2: k_x = 12; k_y = 10; break; | |
1321 | case 3: k_x = 13; k_y = 11; break; | |
1322 | default: /*case 0 too */ k_x = 9; k_y = 8; break; | |
1323 | } | |
1324 | ||
1325 | hpel_flag = v->mv_mode & 1; //MV_PMODE is HPEL | |
1326 | k_x -= hpel_flag; | |
1327 | k_y -= hpel_flag; | |
1328 | ||
1329 | for (y=0; y<v->height_mb; y++) | |
1330 | { | |
1331 | for (x=0; x<v->width_mb; x++) | |
1332 | { | |
1333 | if (v->mv_type_mb_plane[current_mb]) | |
1334 | mv_mode_bit = get_bits(&v->gb, 1); | |
1335 | if (0) //skipmb is rawmode | |
1336 | skip_mb_bit = get_bits(&v->gb, 1); | |
1337 | if (!mv_mode_bit) /* 1MV mode */ | |
1338 | { | |
1339 | if (!v->skip_mb_plane[current_mb]) | |
1340 | { | |
1341 | GET_MVDATA(); | |
1342 | ||
1343 | /* hybrid mv pred, 8.3.5.3.4 */ | |
1344 | if (v->mv_mode == MV_PMODE_1MV || | |
1345 | v->mv_mode == MV_PMODE_MIXED_MV) | |
1346 | hybrid_pred = get_bits(&v->gb, 1); | |
1347 | if (mb_is_intra && !mb_has_coeffs) | |
1348 | { | |
1349 | GET_MQUANT(); | |
1350 | ac_pred = get_bits(&v->gb, 1); | |
1351 | } | |
1352 | else if (mb_has_coeffs) | |
1353 | { | |
1354 | if (mb_is_intra) ac_pred = get_bits(&v->gb, 1); | |
1355 | cbpcy = get_vlc2(&v->gb, v->cbpcy_vlc->table, | |
1356 | VC9_CBPCY_P_VLC_BITS, 2); | |
1357 | GET_MQUANT(); | |
1358 | } | |
1359 | if (!v->ttmbf) | |
1360 | v->ttfrm = get_vlc2(&v->gb, v->ttmb_vlc->table, | |
1361 | VC9_TTMB_VLC_BITS, 12); | |
1362 | //Decode blocks from that mb wrt cbpcy | |
1363 | } | |
1364 | else //Skipped | |
1365 | { | |
1366 | /* hybrid mv pred, 8.3.5.3.4 */ | |
1367 | if (v->mv_mode == MV_PMODE_1MV || | |
1368 | v->mv_mode == MV_PMODE_MIXED_MV) | |
1369 | hybrid_pred = get_bits(&v->gb, 1); | |
1370 | } | |
1371 | } //1MV mode | |
1372 | else //4MV mode | |
1373 | { | |
1374 | if (!v->skip_mb_plane[current_mb] /* unskipped MB */) | |
1375 | { | |
1376 | cbpcy = get_vlc2(&v->gb, v->cbpcy_vlc->table, | |
1377 | VC9_CBPCY_P_VLC_BITS, 2); | |
1378 | for (i=0; i<4; i++) //For all 4 Y blocks | |
1379 | { | |
1380 | if (cbpcy & (1<<6) /* cbpcy set for this block */) | |
1381 | { | |
1382 | GET_MVDATA(); | |
1383 | } | |
1384 | if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) | |
1385 | hybrid_pred = get_bits(&v->gb, 1); | |
1386 | GET_MQUANT(); | |
1387 | if (mb_is_intra /* One of the 4 blocks is intra */ && | |
1388 | index /* non-zero pred for that block */) | |
1389 | ac_pred = get_bits(&v->gb, 1); | |
1390 | if (!v->ttmbf) | |
1391 | tt_block = get_vlc2(&v->gb, v->ttmb_vlc->table, | |
1392 | VC9_TTMB_VLC_BITS, 12); | |
1393 | ||
1394 | /* TODO: Process blocks wrt cbpcy */ | |
1395 | ||
1396 | /* Prepare cbpcy for next block */ | |
1397 | cbpcy <<= 1; | |
1398 | } | |
1399 | } | |
1400 | else //Skipped MB | |
1401 | { | |
1402 | for (i=0; i<4; i++) //All 4 Y blocks | |
1403 | { | |
1404 | if (v->mv_mode == MV_PMODE_MIXED_MV /* Hybrid pred */) | |
1405 | hybrid_pred = get_bits(&v->gb, 1); | |
1406 | ||
1407 | /* FIXME: do something */ | |
1408 | } | |
1409 | } | |
1410 | } | |
1411 | } | |
1412 | current_mb++; | |
1413 | } | |
1414 | return 0; | |
1415 | } | |
1416 | ||
1417 | static int decode_b_mbs(VC9Context *v) | |
1418 | { | |
1419 | int x, y, current_mb = 0 , last_mb = v->height_mb*v->width_mb, | |
1420 | i /* MB / B postion information */; | |
1421 | int direct_b_bit = 0, skip_mb_bit = 0; | |
1422 | int ac_pred; | |
1423 | int b_mv1 = 0, b_mv2 = 0, b_mv_type = 0; | |
1424 | int mquant, mqdiff; /* MB quant stuff */ | |
1425 | int tt_block; /* Block transform type */ | |
1426 | ||
1427 | for (y=0; y<v->height_mb; y++) | |
1428 | { | |
1429 | for (x=0; x<v->width_mb; x++) | |
1430 | { | |
1431 | if (v->direct_mb_plane[current_mb]) | |
1432 | direct_b_bit = get_bits(&v->gb, 1); | |
1433 | if (1 /* Skip mode is raw */) | |
1434 | { | |
1435 | /* FIXME getting tired commenting */ | |
1436 | #if 0 | |
1437 | skip_mb_bit = get_bits(&v->gb, n); //vlc | |
1438 | #endif | |
1439 | } | |
1440 | if (!direct_b_bit) | |
1441 | { | |
1442 | if (skip_mb_bit) | |
1443 | { | |
1444 | /* FIXME getting tired commenting */ | |
1445 | #if 0 | |
1446 | b_mv_type = get_bits(&v->gb, n); //vlc | |
1447 | #endif | |
1448 | } | |
1449 | else | |
1450 | { | |
1451 | /* FIXME getting tired commenting */ | |
1452 | #if 0 | |
1453 | b_mv1 = get_bits(&v->gb, n); //VLC | |
1454 | #endif | |
1455 | if (1 /* b_mv1 isn't intra */) | |
1456 | { | |
1457 | /* FIXME: actually read it */ | |
1458 | b_mv_type = 0; //vlc | |
1459 | } | |
1460 | } | |
1461 | } | |
1462 | if (!skip_mb_bit) | |
1463 | { | |
1464 | if (b_mv1 != last_mb) | |
1465 | { | |
1466 | GET_MQUANT(); | |
1467 | if (1 /* intra mb */) | |
1468 | ac_pred = get_bits(&v->gb, 1); | |
1469 | } | |
1470 | else | |
1471 | { | |
1472 | if (1 /* forward_mb is interpolate */) | |
1473 | { | |
1474 | /* FIXME: actually read it */ | |
1475 | b_mv2 = 0; //vlc | |
1476 | } | |
1477 | if (1 /* b_mv2 isn't the last */) | |
1478 | { | |
1479 | if (1 /* intra_mb */) | |
1480 | ac_pred = get_bits(&v->gb, 1); | |
1481 | GET_MQUANT(); | |
1482 | } | |
1483 | } | |
1484 | } | |
1485 | //End1 | |
1486 | /* FIXME getting tired, commenting */ | |
1487 | #if 0 | |
1488 | if (v->ttmbf) | |
1489 | v->ttmb = get_bits(&v->gb, n); //vlc | |
1490 | #endif | |
1491 | } | |
1492 | //End2 | |
1493 | for (i=0; i<6; i++) | |
1494 | { | |
1495 | /* FIXME: process the block */ | |
1496 | } | |
1497 | ||
1498 | current_mb++; | |
1499 | } | |
1500 | return 0; | |
1501 | } | |
1502 | ||
1503 | #if HAS_ADVANCED_PROFILE | |
1504 | static int advanced_decode_i_mbs(VC9Context *v) | |
1505 | { | |
1506 | int i, x, y, cbpcy, mqdiff, absmq, mquant, ac_pred, condover, | |
1507 | current_mb = 0, over_flags_mb = 0; | |
1508 | ||
1509 | for (y=0; y<v->height_mb; y++) | |
1510 | { | |
1511 | for (x=0; x<v->width_mb; x++) | |
1512 | { | |
1513 | if (v->ac_pred_plane[i]) | |
1514 | ac_pred = get_bits(&v->gb, 1); | |
1515 | if (condover == 3 && v->over_flags_plane) | |
1516 | over_flags_mb = get_bits(&v->gb, 1); | |
1517 | GET_MQUANT(); | |
1518 | } | |
1519 | current_mb++; | |
1520 | } | |
1521 | return 0; | |
1522 | } | |
1523 | #endif | |
1524 | ||
1525 | static int vc9_decode_init(AVCodecContext *avctx) | |
1526 | { | |
1527 | VC9Context *v = avctx->priv_data; | |
1528 | GetBitContext gb; | |
1529 | ||
1530 | if (!avctx->extradata_size || !avctx->extradata) return -1; | |
1531 | avctx->pix_fmt = PIX_FMT_YUV420P; | |
1532 | v->avctx = avctx; | |
1533 | ||
1534 | if (init_common(v) < 0) return -1; | |
1535 | ||
1536 | if (avctx->codec_id == CODEC_ID_WMV3) | |
1537 | { | |
1538 | int count = 0; | |
1539 | ||
1540 | // looks like WMV3 has a sequence header stored in the extradata | |
1541 | // advanced sequence header may be before the first frame | |
1542 | // the last byte of the extradata is a version number, 1 for the | |
1543 | // samples we can decode | |
1544 | ||
1545 | init_get_bits(&gb, avctx->extradata, avctx->extradata_size); | |
1546 | ||
1547 | decode_sequence_header(avctx, &gb); | |
1548 | ||
1549 | count = avctx->extradata_size*8 - get_bits_count(&gb); | |
1550 | if (count>0) | |
1551 | { | |
1552 | av_log(avctx, AV_LOG_INFO, "Extra data: %i bits left, value: %X\n", | |
1553 | count, get_bits(&gb, count)); | |
1554 | } | |
1555 | else | |
1556 | { | |
1557 | av_log(avctx, AV_LOG_INFO, "Read %i bits in overflow\n", -count); | |
1558 | } | |
1559 | } | |
1560 | ||
1561 | /* Done with header parsing */ | |
1562 | //FIXME I feel like this is wrong | |
1563 | v->width_mb = (avctx->coded_width+15)>>4; | |
1564 | v->height_mb = (avctx->coded_height+15)>>4; | |
1565 | ||
1566 | /* Allocate mb bitplanes */ | |
1567 | v->mv_type_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); | |
1568 | if (!v->mv_type_mb_plane) return -1; | |
1569 | v->skip_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); | |
1570 | if (!v->skip_mb_plane) return -1; | |
1571 | v->direct_mb_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); | |
1572 | if (!v->direct_mb_plane) return -1; | |
1573 | ||
1574 | #if HAS_ADVANCED_PROFILE | |
1575 | if (v->profile > PROFILE_MAIN) | |
1576 | { | |
1577 | v->over_flags_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); | |
1578 | if (!v->over_flags_plane) return -1; | |
1579 | v->ac_pred_plane = (uint8_t *)av_malloc(v->width_mb*v->height_mb); | |
1580 | if (!v->ac_pred_plane) return -1; | |
1581 | } | |
1582 | #endif | |
1583 | ||
1584 | return 0; | |
1585 | } | |
1586 | ||
1587 | static int vc9_decode_frame(AVCodecContext *avctx, | |
1588 | void *data, int *data_size, | |
1589 | uint8_t *buf, int buf_size) | |
1590 | { | |
1591 | VC9Context *v = avctx->priv_data; | |
1592 | int ret = FRAME_SKIPED, len, start_code; | |
1593 | AVFrame *pict = data; | |
1594 | uint8_t *tmp_buf; | |
1595 | v->avctx = avctx; | |
1596 | ||
1597 | //buf_size = 0 -> last frame | |
1598 | if (!buf_size) return 0; | |
1599 | ||
1600 | len = avpicture_get_size(avctx->pix_fmt, avctx->width, | |
1601 | avctx->height); | |
1602 | tmp_buf = (uint8_t *)av_mallocz(len); | |
1603 | avpicture_fill((AVPicture *)pict, tmp_buf, avctx->pix_fmt, | |
1604 | avctx->width, avctx->height); | |
1605 | ||
1606 | if (avctx->codec_id == CODEC_ID_WMV3) | |
1607 | { | |
1608 | init_get_bits(&v->gb, buf, buf_size*8); | |
1609 | av_log(avctx, AV_LOG_INFO, "Frame: %i bits to decode\n", buf_size*8); | |
1610 | ||
1611 | #if HAS_ADVANCED_PROFILE | |
1612 | if (v->profile > PROFILE_MAIN) | |
1613 | { | |
1614 | if (advanced_decode_picture_header(v) == FRAME_SKIPED) return buf_size; | |
1615 | switch(v->pict_type) | |
1616 | { | |
1617 | case I_TYPE: ret = advanced_decode_i_mbs(v); break; | |
1618 | case P_TYPE: ret = decode_p_mbs(v); break; | |
1619 | case B_TYPE: | |
1620 | case BI_TYPE: ret = decode_b_mbs(v); break; | |
1621 | default: ret = FRAME_SKIPED; | |
1622 | } | |
1623 | if (ret == FRAME_SKIPED) return buf_size; //We ignore for now failures | |
1624 | } | |
1625 | else | |
1626 | #endif | |
1627 | { | |
1628 | if (standard_decode_picture_header(v) == FRAME_SKIPED) return buf_size; | |
1629 | switch(v->pict_type) | |
1630 | { | |
1631 | case I_TYPE: ret = standard_decode_i_mbs(v); break; | |
1632 | case P_TYPE: ret = decode_p_mbs(v); break; | |
1633 | case B_TYPE: | |
1634 | case BI_TYPE: ret = decode_b_mbs(v); break; | |
1635 | default: ret = FRAME_SKIPED; | |
1636 | } | |
1637 | if (ret == FRAME_SKIPED) return buf_size; | |
1638 | } | |
1639 | ||
1640 | /* Size of the output data = image */ | |
1641 | av_log(avctx, AV_LOG_DEBUG, "Consumed %i/%i bits\n", | |
1642 | get_bits_count(&v->gb), buf_size*8); | |
1643 | } | |
1644 | else | |
1645 | { | |
1646 | #if 0 | |
1647 | // search for IDU's | |
1648 | // FIXME | |
1649 | uint32_t scp = 0; | |
1650 | int scs = 0, i = 0; | |
1651 | ||
1652 | while (i < buf_size) | |
1653 | { | |
1654 | for (; i < buf_size && scp != 0x000001; i++) | |
1655 | scp = ((scp<<8)|buf[i])&0xffffff; | |
1656 | ||
1657 | if (scp != 0x000001) | |
1658 | break; // eof ? | |
1659 | ||
1660 | scs = buf[i++]; | |
1661 | ||
1662 | init_get_bits(&v->gb, buf+i, (buf_size-i)*8); | |
1663 | ||
1664 | switch(scs) | |
1665 | { | |
1666 | case 0xf: | |
1667 | decode_sequence_header(avctx, &v->gb); | |
1668 | break; | |
1669 | // to be finished | |
1670 | } | |
1671 | ||
1672 | i += get_bits_count(&v->gb)*8; | |
1673 | } | |
1674 | #else | |
1675 | av_abort(); | |
1676 | #endif | |
1677 | } | |
1678 | ||
1679 | *data_size = len; | |
1680 | ||
1681 | /* Fake consumption of all data */ | |
1682 | return buf_size; //Number of bytes consumed | |
1683 | } | |
1684 | ||
1685 | static int vc9_decode_end(AVCodecContext *avctx) | |
1686 | { | |
1687 | VC9Context *v = avctx->priv_data; | |
1688 | ||
1689 | #if HAS_ADVANCED_PROFILE | |
1690 | if (v->hrd_rate) av_free(v->hrd_rate); | |
1691 | if (v->hrd_buffer) av_free(v->hrd_buffer); | |
1692 | #endif | |
1693 | if (v->mv_type_mb_plane) av_free(v->mv_type_mb_plane); | |
1694 | if (v->skip_mb_plane) av_free(v->skip_mb_plane); | |
1695 | if (v->direct_mb_plane) av_free(v->direct_mb_plane); | |
1696 | return 0; | |
1697 | } | |
1698 | ||
1699 | AVCodec vc9_decoder = { | |
1700 | "vc9", | |
1701 | CODEC_TYPE_VIDEO, | |
1702 | CODEC_ID_VC9, | |
1703 | sizeof(VC9Context), | |
1704 | vc9_decode_init, | |
1705 | NULL, | |
1706 | vc9_decode_end, | |
1707 | vc9_decode_frame, | |
1708 | 0, | |
1709 | NULL | |
1710 | }; | |
1711 | ||
1712 | AVCodec wmv3_decoder = { | |
1713 | "wmv3", | |
1714 | CODEC_TYPE_VIDEO, | |
1715 | CODEC_ID_WMV3, | |
1716 | sizeof(VC9Context), | |
1717 | vc9_decode_init, | |
1718 | NULL, | |
1719 | vc9_decode_end, | |
1720 | vc9_decode_frame, | |
1721 | 0, | |
1722 | NULL | |
1723 | }; |