summaryrefslogtreecommitdiff
path: root/libavcodec/msmpeg4dec.c (plain)
blob: cf43027a0089331acac2b0c30ce6ac375b18b779
1/*
2 * MSMPEG4 backend for encoder and decoder
3 * Copyright (c) 2001 Fabrice Bellard
4 * Copyright (c) 2002-2013 Michael Niedermayer <michaelni@gmx.at>
5 *
6 * msmpeg4v1 & v2 stuff by Michael Niedermayer <michaelni@gmx.at>
7 *
8 * This file is part of FFmpeg.
9 *
10 * FFmpeg is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * FFmpeg is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with FFmpeg; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
25#include "avcodec.h"
26#include "internal.h"
27#include "mpegutils.h"
28#include "mpegvideo.h"
29#include "msmpeg4.h"
30#include "libavutil/imgutils.h"
31#include "h263.h"
32#include "mpeg4video.h"
33#include "msmpeg4data.h"
34#include "vc1data.h"
35#include "wmv2.h"
36
37#define DC_VLC_BITS 9
38#define V2_INTRA_CBPC_VLC_BITS 3
39#define V2_MB_TYPE_VLC_BITS 7
40#define MV_VLC_BITS 9
41#define V2_MV_VLC_BITS 9
42#define TEX_VLC_BITS 9
43
44#define DEFAULT_INTER_INDEX 3
45
46static inline int msmpeg4v1_pred_dc(MpegEncContext * s, int n,
47 int32_t **dc_val_ptr)
48{
49 int i;
50
51 if (n < 4) {
52 i= 0;
53 } else {
54 i= n-3;
55 }
56
57 *dc_val_ptr= &s->last_dc[i];
58 return s->last_dc[i];
59}
60
61/****************************************/
62/* decoding stuff */
63
64VLC ff_mb_non_intra_vlc[4];
65static VLC v2_dc_lum_vlc;
66static VLC v2_dc_chroma_vlc;
67static VLC v2_intra_cbpc_vlc;
68static VLC v2_mb_type_vlc;
69static VLC v2_mv_vlc;
70VLC ff_inter_intra_vlc;
71
72/* This is identical to H.263 except that its range is multiplied by 2. */
73static int msmpeg4v2_decode_motion(MpegEncContext * s, int pred, int f_code)
74{
75 int code, val, sign, shift;
76
77 code = get_vlc2(&s->gb, v2_mv_vlc.table, V2_MV_VLC_BITS, 2);
78 ff_dlog(s, "MV code %d at %d %d pred: %d\n", code, s->mb_x,s->mb_y, pred);
79 if (code < 0)
80 return 0xffff;
81
82 if (code == 0)
83 return pred;
84 sign = get_bits1(&s->gb);
85 shift = f_code - 1;
86 val = code;
87 if (shift) {
88 val = (val - 1) << shift;
89 val |= get_bits(&s->gb, shift);
90 val++;
91 }
92 if (sign)
93 val = -val;
94
95 val += pred;
96 if (val <= -64)
97 val += 64;
98 else if (val >= 64)
99 val -= 64;
100
101 return val;
102}
103
104static int msmpeg4v12_decode_mb(MpegEncContext *s, int16_t block[6][64])
105{
106 int cbp, code, i;
107 uint32_t * const mb_type_ptr = &s->current_picture.mb_type[s->mb_x + s->mb_y*s->mb_stride];
108
109 if (s->pict_type == AV_PICTURE_TYPE_P) {
110 if (s->use_skip_mb_code) {
111 if (get_bits1(&s->gb)) {
112 /* skip mb */
113 s->mb_intra = 0;
114 for(i=0;i<6;i++)
115 s->block_last_index[i] = -1;
116 s->mv_dir = MV_DIR_FORWARD;
117 s->mv_type = MV_TYPE_16X16;
118 s->mv[0][0][0] = 0;
119 s->mv[0][0][1] = 0;
120 s->mb_skipped = 1;
121 *mb_type_ptr = MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
122 return 0;
123 }
124 }
125
126 if(s->msmpeg4_version==2)
127 code = get_vlc2(&s->gb, v2_mb_type_vlc.table, V2_MB_TYPE_VLC_BITS, 1);
128 else
129 code = get_vlc2(&s->gb, ff_h263_inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
130 if(code<0 || code>7){
131 av_log(s->avctx, AV_LOG_ERROR, "cbpc %d invalid at %d %d\n", code, s->mb_x, s->mb_y);
132 return -1;
133 }
134
135 s->mb_intra = code >>2;
136
137 cbp = code & 0x3;
138 } else {
139 s->mb_intra = 1;
140 if(s->msmpeg4_version==2)
141 cbp= get_vlc2(&s->gb, v2_intra_cbpc_vlc.table, V2_INTRA_CBPC_VLC_BITS, 1);
142 else
143 cbp= get_vlc2(&s->gb, ff_h263_intra_MCBPC_vlc.table, INTRA_MCBPC_VLC_BITS, 1);
144 if(cbp<0 || cbp>3){
145 av_log(s->avctx, AV_LOG_ERROR, "cbpc %d invalid at %d %d\n", cbp, s->mb_x, s->mb_y);
146 return -1;
147 }
148 }
149
150 if (!s->mb_intra) {
151 int mx, my, cbpy;
152
153 cbpy= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1);
154 if(cbpy<0){
155 av_log(s->avctx, AV_LOG_ERROR, "cbpy %d invalid at %d %d\n", cbp, s->mb_x, s->mb_y);
156 return -1;
157 }
158
159 cbp|= cbpy<<2;
160 if(s->msmpeg4_version==1 || (cbp&3) != 3) cbp^= 0x3C;
161
162 ff_h263_pred_motion(s, 0, 0, &mx, &my);
163 mx= msmpeg4v2_decode_motion(s, mx, 1);
164 my= msmpeg4v2_decode_motion(s, my, 1);
165
166 s->mv_dir = MV_DIR_FORWARD;
167 s->mv_type = MV_TYPE_16X16;
168 s->mv[0][0][0] = mx;
169 s->mv[0][0][1] = my;
170 *mb_type_ptr = MB_TYPE_L0 | MB_TYPE_16x16;
171 } else {
172 if(s->msmpeg4_version==2){
173 s->ac_pred = get_bits1(&s->gb);
174 cbp|= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1)<<2; //FIXME check errors
175 } else{
176 s->ac_pred = 0;
177 cbp|= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1)<<2; //FIXME check errors
178 if(s->pict_type==AV_PICTURE_TYPE_P) cbp^=0x3C;
179 }
180 *mb_type_ptr = MB_TYPE_INTRA;
181 }
182
183 s->bdsp.clear_blocks(s->block[0]);
184 for (i = 0; i < 6; i++) {
185 if (ff_msmpeg4_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1, NULL) < 0)
186 {
187 av_log(s->avctx, AV_LOG_ERROR, "\nerror while decoding block: %d x %d (%d)\n", s->mb_x, s->mb_y, i);
188 return -1;
189 }
190 }
191 return 0;
192}
193
194static int msmpeg4v34_decode_mb(MpegEncContext *s, int16_t block[6][64])
195{
196 int cbp, code, i;
197 uint8_t *coded_val;
198 uint32_t * const mb_type_ptr = &s->current_picture.mb_type[s->mb_x + s->mb_y*s->mb_stride];
199
200 if (s->pict_type == AV_PICTURE_TYPE_P) {
201 if (s->use_skip_mb_code) {
202 if (get_bits1(&s->gb)) {
203 /* skip mb */
204 s->mb_intra = 0;
205 for(i=0;i<6;i++)
206 s->block_last_index[i] = -1;
207 s->mv_dir = MV_DIR_FORWARD;
208 s->mv_type = MV_TYPE_16X16;
209 s->mv[0][0][0] = 0;
210 s->mv[0][0][1] = 0;
211 s->mb_skipped = 1;
212 *mb_type_ptr = MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
213
214 return 0;
215 }
216 }
217
218 code = get_vlc2(&s->gb, ff_mb_non_intra_vlc[DEFAULT_INTER_INDEX].table, MB_NON_INTRA_VLC_BITS, 3);
219 if (code < 0)
220 return -1;
221 //s->mb_intra = (code & 0x40) ? 0 : 1;
222 s->mb_intra = (~code & 0x40) >> 6;
223
224 cbp = code & 0x3f;
225 } else {
226 s->mb_intra = 1;
227 code = get_vlc2(&s->gb, ff_msmp4_mb_i_vlc.table, MB_INTRA_VLC_BITS, 2);
228 if (code < 0)
229 return -1;
230 /* predict coded block pattern */
231 cbp = 0;
232 for(i=0;i<6;i++) {
233 int val = ((code >> (5 - i)) & 1);
234 if (i < 4) {
235 int pred = ff_msmpeg4_coded_block_pred(s, i, &coded_val);
236 val = val ^ pred;
237 *coded_val = val;
238 }
239 cbp |= val << (5 - i);
240 }
241 }
242
243 if (!s->mb_intra) {
244 int mx, my;
245 if(s->per_mb_rl_table && cbp){
246 s->rl_table_index = decode012(&s->gb);
247 s->rl_chroma_table_index = s->rl_table_index;
248 }
249 ff_h263_pred_motion(s, 0, 0, &mx, &my);
250 if (ff_msmpeg4_decode_motion(s, &mx, &my) < 0)
251 return -1;
252 s->mv_dir = MV_DIR_FORWARD;
253 s->mv_type = MV_TYPE_16X16;
254 s->mv[0][0][0] = mx;
255 s->mv[0][0][1] = my;
256 *mb_type_ptr = MB_TYPE_L0 | MB_TYPE_16x16;
257 } else {
258 ff_dlog(s, "I at %d %d %d %06X\n", s->mb_x, s->mb_y,
259 ((cbp & 3) ? 1 : 0) +((cbp & 0x3C)? 2 : 0),
260 show_bits(&s->gb, 24));
261 s->ac_pred = get_bits1(&s->gb);
262 *mb_type_ptr = MB_TYPE_INTRA;
263 if(s->inter_intra_pred){
264 s->h263_aic_dir= get_vlc2(&s->gb, ff_inter_intra_vlc.table, INTER_INTRA_VLC_BITS, 1);
265 ff_dlog(s, "%d%d %d %d/",
266 s->ac_pred, s->h263_aic_dir, s->mb_x, s->mb_y);
267 }
268 if(s->per_mb_rl_table && cbp){
269 s->rl_table_index = decode012(&s->gb);
270 s->rl_chroma_table_index = s->rl_table_index;
271 }
272 }
273
274 s->bdsp.clear_blocks(s->block[0]);
275 for (i = 0; i < 6; i++) {
276 if (ff_msmpeg4_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1, NULL) < 0)
277 {
278 av_log(s->avctx, AV_LOG_ERROR, "\nerror while decoding block: %d x %d (%d)\n", s->mb_x, s->mb_y, i);
279 return -1;
280 }
281 }
282
283 return 0;
284}
285
286/* init all vlc decoding tables */
287av_cold int ff_msmpeg4_decode_init(AVCodecContext *avctx)
288{
289 MpegEncContext *s = avctx->priv_data;
290 static volatile int done = 0;
291 int i, ret;
292 MVTable *mv;
293
294 if ((ret = av_image_check_size(avctx->width, avctx->height, 0, avctx)) < 0)
295 return ret;
296
297 if (ff_h263_decode_init(avctx) < 0)
298 return -1;
299
300 ff_msmpeg4_common_init(s);
301
302 if (!done) {
303 for(i=0;i<NB_RL_TABLES;i++) {
304 ff_rl_init(&ff_rl_table[i], ff_static_rl_table_store[i]);
305 }
306 INIT_VLC_RL(ff_rl_table[0], 642);
307 INIT_VLC_RL(ff_rl_table[1], 1104);
308 INIT_VLC_RL(ff_rl_table[2], 554);
309 INIT_VLC_RL(ff_rl_table[3], 940);
310 INIT_VLC_RL(ff_rl_table[4], 962);
311 INIT_VLC_RL(ff_rl_table[5], 554);
312
313 mv = &ff_mv_tables[0];
314 INIT_VLC_STATIC(&mv->vlc, MV_VLC_BITS, mv->n + 1,
315 mv->table_mv_bits, 1, 1,
316 mv->table_mv_code, 2, 2, 3714);
317 mv = &ff_mv_tables[1];
318 INIT_VLC_STATIC(&mv->vlc, MV_VLC_BITS, mv->n + 1,
319 mv->table_mv_bits, 1, 1,
320 mv->table_mv_code, 2, 2, 2694);
321
322 INIT_VLC_STATIC(&ff_msmp4_dc_luma_vlc[0], DC_VLC_BITS, 120,
323 &ff_table0_dc_lum[0][1], 8, 4,
324 &ff_table0_dc_lum[0][0], 8, 4, 1158);
325 INIT_VLC_STATIC(&ff_msmp4_dc_chroma_vlc[0], DC_VLC_BITS, 120,
326 &ff_table0_dc_chroma[0][1], 8, 4,
327 &ff_table0_dc_chroma[0][0], 8, 4, 1118);
328 INIT_VLC_STATIC(&ff_msmp4_dc_luma_vlc[1], DC_VLC_BITS, 120,
329 &ff_table1_dc_lum[0][1], 8, 4,
330 &ff_table1_dc_lum[0][0], 8, 4, 1476);
331 INIT_VLC_STATIC(&ff_msmp4_dc_chroma_vlc[1], DC_VLC_BITS, 120,
332 &ff_table1_dc_chroma[0][1], 8, 4,
333 &ff_table1_dc_chroma[0][0], 8, 4, 1216);
334
335 INIT_VLC_STATIC(&v2_dc_lum_vlc, DC_VLC_BITS, 512,
336 &ff_v2_dc_lum_table[0][1], 8, 4,
337 &ff_v2_dc_lum_table[0][0], 8, 4, 1472);
338 INIT_VLC_STATIC(&v2_dc_chroma_vlc, DC_VLC_BITS, 512,
339 &ff_v2_dc_chroma_table[0][1], 8, 4,
340 &ff_v2_dc_chroma_table[0][0], 8, 4, 1506);
341
342 INIT_VLC_STATIC(&v2_intra_cbpc_vlc, V2_INTRA_CBPC_VLC_BITS, 4,
343 &ff_v2_intra_cbpc[0][1], 2, 1,
344 &ff_v2_intra_cbpc[0][0], 2, 1, 8);
345 INIT_VLC_STATIC(&v2_mb_type_vlc, V2_MB_TYPE_VLC_BITS, 8,
346 &ff_v2_mb_type[0][1], 2, 1,
347 &ff_v2_mb_type[0][0], 2, 1, 128);
348 INIT_VLC_STATIC(&v2_mv_vlc, V2_MV_VLC_BITS, 33,
349 &ff_mvtab[0][1], 2, 1,
350 &ff_mvtab[0][0], 2, 1, 538);
351
352 INIT_VLC_STATIC(&ff_mb_non_intra_vlc[0], MB_NON_INTRA_VLC_BITS, 128,
353 &ff_wmv2_inter_table[0][0][1], 8, 4,
354 &ff_wmv2_inter_table[0][0][0], 8, 4, 1636);
355 INIT_VLC_STATIC(&ff_mb_non_intra_vlc[1], MB_NON_INTRA_VLC_BITS, 128,
356 &ff_wmv2_inter_table[1][0][1], 8, 4,
357 &ff_wmv2_inter_table[1][0][0], 8, 4, 2648);
358 INIT_VLC_STATIC(&ff_mb_non_intra_vlc[2], MB_NON_INTRA_VLC_BITS, 128,
359 &ff_wmv2_inter_table[2][0][1], 8, 4,
360 &ff_wmv2_inter_table[2][0][0], 8, 4, 1532);
361 INIT_VLC_STATIC(&ff_mb_non_intra_vlc[3], MB_NON_INTRA_VLC_BITS, 128,
362 &ff_wmv2_inter_table[3][0][1], 8, 4,
363 &ff_wmv2_inter_table[3][0][0], 8, 4, 2488);
364
365 INIT_VLC_STATIC(&ff_msmp4_mb_i_vlc, MB_INTRA_VLC_BITS, 64,
366 &ff_msmp4_mb_i_table[0][1], 4, 2,
367 &ff_msmp4_mb_i_table[0][0], 4, 2, 536);
368
369 INIT_VLC_STATIC(&ff_inter_intra_vlc, INTER_INTRA_VLC_BITS, 4,
370 &ff_table_inter_intra[0][1], 2, 1,
371 &ff_table_inter_intra[0][0], 2, 1, 8);
372 done = 1;
373 }
374
375 switch(s->msmpeg4_version){
376 case 1:
377 case 2:
378 s->decode_mb= msmpeg4v12_decode_mb;
379 break;
380 case 3:
381 case 4:
382 s->decode_mb= msmpeg4v34_decode_mb;
383 break;
384 case 5:
385 if (CONFIG_WMV2_DECODER)
386 s->decode_mb= ff_wmv2_decode_mb;
387 case 6:
388 //FIXME + TODO VC1 decode mb
389 break;
390 }
391
392 s->slice_height= s->mb_height; //to avoid 1/0 if the first frame is not a keyframe
393
394 return 0;
395}
396
397int ff_msmpeg4_decode_picture_header(MpegEncContext * s)
398{
399 int code;
400
401 if(s->msmpeg4_version==1){
402 int start_code = get_bits_long(&s->gb, 32);
403 if(start_code!=0x00000100){
404 av_log(s->avctx, AV_LOG_ERROR, "invalid startcode\n");
405 return -1;
406 }
407
408 skip_bits(&s->gb, 5); // frame number */
409 }
410
411 s->pict_type = get_bits(&s->gb, 2) + 1;
412 if (s->pict_type != AV_PICTURE_TYPE_I &&
413 s->pict_type != AV_PICTURE_TYPE_P){
414 av_log(s->avctx, AV_LOG_ERROR, "invalid picture type\n");
415 return -1;
416 }
417 s->chroma_qscale= s->qscale = get_bits(&s->gb, 5);
418 if(s->qscale==0){
419 av_log(s->avctx, AV_LOG_ERROR, "invalid qscale\n");
420 return -1;
421 }
422
423 if (s->pict_type == AV_PICTURE_TYPE_I) {
424 code = get_bits(&s->gb, 5);
425 if(s->msmpeg4_version==1){
426 if(code==0 || code>s->mb_height){
427 av_log(s->avctx, AV_LOG_ERROR, "invalid slice height %d\n", code);
428 return -1;
429 }
430
431 s->slice_height = code;
432 }else{
433 /* 0x17: one slice, 0x18: two slices, ... */
434 if (code < 0x17){
435 av_log(s->avctx, AV_LOG_ERROR, "error, slice code was %X\n", code);
436 return -1;
437 }
438
439 s->slice_height = s->mb_height / (code - 0x16);
440 }
441
442 switch(s->msmpeg4_version){
443 case 1:
444 case 2:
445 s->rl_chroma_table_index = 2;
446 s->rl_table_index = 2;
447
448 s->dc_table_index = 0; //not used
449 break;
450 case 3:
451 s->rl_chroma_table_index = decode012(&s->gb);
452 s->rl_table_index = decode012(&s->gb);
453
454 s->dc_table_index = get_bits1(&s->gb);
455 break;
456 case 4:
457 ff_msmpeg4_decode_ext_header(s, (2+5+5+17+7)/8);
458
459 if(s->bit_rate > MBAC_BITRATE) s->per_mb_rl_table= get_bits1(&s->gb);
460 else s->per_mb_rl_table= 0;
461
462 if(!s->per_mb_rl_table){
463 s->rl_chroma_table_index = decode012(&s->gb);
464 s->rl_table_index = decode012(&s->gb);
465 }
466
467 s->dc_table_index = get_bits1(&s->gb);
468 s->inter_intra_pred= 0;
469 break;
470 }
471 s->no_rounding = 1;
472 if(s->avctx->debug&FF_DEBUG_PICT_INFO)
473 av_log(s->avctx, AV_LOG_DEBUG, "qscale:%d rlc:%d rl:%d dc:%d mbrl:%d slice:%d \n",
474 s->qscale,
475 s->rl_chroma_table_index,
476 s->rl_table_index,
477 s->dc_table_index,
478 s->per_mb_rl_table,
479 s->slice_height);
480 } else {
481 switch(s->msmpeg4_version){
482 case 1:
483 case 2:
484 if(s->msmpeg4_version==1)
485 s->use_skip_mb_code = 1;
486 else
487 s->use_skip_mb_code = get_bits1(&s->gb);
488 s->rl_table_index = 2;
489 s->rl_chroma_table_index = s->rl_table_index;
490 s->dc_table_index = 0; //not used
491 s->mv_table_index = 0;
492 break;
493 case 3:
494 s->use_skip_mb_code = get_bits1(&s->gb);
495 s->rl_table_index = decode012(&s->gb);
496 s->rl_chroma_table_index = s->rl_table_index;
497
498 s->dc_table_index = get_bits1(&s->gb);
499
500 s->mv_table_index = get_bits1(&s->gb);
501 break;
502 case 4:
503 s->use_skip_mb_code = get_bits1(&s->gb);
504
505 if(s->bit_rate > MBAC_BITRATE) s->per_mb_rl_table= get_bits1(&s->gb);
506 else s->per_mb_rl_table= 0;
507
508 if(!s->per_mb_rl_table){
509 s->rl_table_index = decode012(&s->gb);
510 s->rl_chroma_table_index = s->rl_table_index;
511 }
512
513 s->dc_table_index = get_bits1(&s->gb);
514
515 s->mv_table_index = get_bits1(&s->gb);
516 s->inter_intra_pred= (s->width*s->height < 320*240 && s->bit_rate<=II_BITRATE);
517 break;
518 }
519
520 if(s->avctx->debug&FF_DEBUG_PICT_INFO)
521 av_log(s->avctx, AV_LOG_DEBUG, "skip:%d rl:%d rlc:%d dc:%d mv:%d mbrl:%d qp:%d \n",
522 s->use_skip_mb_code,
523 s->rl_table_index,
524 s->rl_chroma_table_index,
525 s->dc_table_index,
526 s->mv_table_index,
527 s->per_mb_rl_table,
528 s->qscale);
529
530 if(s->flipflop_rounding){
531 s->no_rounding ^= 1;
532 }else{
533 s->no_rounding = 0;
534 }
535 }
536 ff_dlog(s->avctx, "%d %"PRId64" %d %d %d\n", s->pict_type, s->bit_rate,
537 s->inter_intra_pred, s->width, s->height);
538
539 s->esc3_level_length= 0;
540 s->esc3_run_length= 0;
541
542 return 0;
543}
544
545int ff_msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size)
546{
547 int left= buf_size*8 - get_bits_count(&s->gb);
548 int length= s->msmpeg4_version>=3 ? 17 : 16;
549 /* the alt_bitstream reader could read over the end so we need to check it */
550 if(left>=length && left<length+8)
551 {
552 skip_bits(&s->gb, 5); /* fps */
553 s->bit_rate= get_bits(&s->gb, 11)*1024;
554 if(s->msmpeg4_version>=3)
555 s->flipflop_rounding= get_bits1(&s->gb);
556 else
557 s->flipflop_rounding= 0;
558 }
559 else if(left<length+8)
560 {
561 s->flipflop_rounding= 0;
562 if(s->msmpeg4_version != 2)
563 av_log(s->avctx, AV_LOG_ERROR, "ext header missing, %d left\n", left);
564 }
565 else
566 {
567 av_log(s->avctx, AV_LOG_ERROR, "I-frame too long, ignoring ext header\n");
568 }
569
570 return 0;
571}
572
573static int msmpeg4_decode_dc(MpegEncContext * s, int n, int *dir_ptr)
574{
575 int level, pred;
576
577 if(s->msmpeg4_version<=2){
578 if (n < 4) {
579 level = get_vlc2(&s->gb, v2_dc_lum_vlc.table, DC_VLC_BITS, 3);
580 } else {
581 level = get_vlc2(&s->gb, v2_dc_chroma_vlc.table, DC_VLC_BITS, 3);
582 }
583 if (level < 0) {
584 av_log(s->avctx, AV_LOG_ERROR, "illegal dc vlc\n");
585 *dir_ptr = 0;
586 return -1;
587 }
588 level-=256;
589 }else{ //FIXME optimize use unified tables & index
590 if (n < 4) {
591 level = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
592 } else {
593 level = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
594 }
595 if (level < 0){
596 av_log(s->avctx, AV_LOG_ERROR, "illegal dc vlc\n");
597 *dir_ptr = 0;
598 return -1;
599 }
600
601 if (level == DC_MAX) {
602 level = get_bits(&s->gb, 8);
603 if (get_bits1(&s->gb))
604 level = -level;
605 } else if (level != 0) {
606 if (get_bits1(&s->gb))
607 level = -level;
608 }
609 }
610
611 if(s->msmpeg4_version==1){
612 int32_t *dc_val;
613 pred = msmpeg4v1_pred_dc(s, n, &dc_val);
614 level += pred;
615
616 /* update predictor */
617 *dc_val= level;
618 }else{
619 int16_t *dc_val;
620 pred = ff_msmpeg4_pred_dc(s, n, &dc_val, dir_ptr);
621 level += pred;
622
623 /* update predictor */
624 if (n < 4) {
625 *dc_val = level * s->y_dc_scale;
626 } else {
627 *dc_val = level * s->c_dc_scale;
628 }
629 }
630
631 return level;
632}
633
634int ff_msmpeg4_decode_block(MpegEncContext * s, int16_t * block,
635 int n, int coded, const uint8_t *scan_table)
636{
637 int level, i, last, run, run_diff;
638 int av_uninit(dc_pred_dir);
639 RLTable *rl;
640 RL_VLC_ELEM *rl_vlc;
641 int qmul, qadd;
642
643 if (s->mb_intra) {
644 qmul=1;
645 qadd=0;
646
647 /* DC coef */
648 level = msmpeg4_decode_dc(s, n, &dc_pred_dir);
649
650 if (level < 0){
651 av_log(s->avctx, AV_LOG_ERROR, "dc overflow- block: %d qscale: %d//\n", n, s->qscale);
652 if(s->inter_intra_pred) level=0;
653 }
654 if (n < 4) {
655 rl = &ff_rl_table[s->rl_table_index];
656 if(level > 256*s->y_dc_scale){
657 av_log(s->avctx, AV_LOG_ERROR, "dc overflow+ L qscale: %d//\n", s->qscale);
658 if(!s->inter_intra_pred) return -1;
659 }
660 } else {
661 rl = &ff_rl_table[3 + s->rl_chroma_table_index];
662 if(level > 256*s->c_dc_scale){
663 av_log(s->avctx, AV_LOG_ERROR, "dc overflow+ C qscale: %d//\n", s->qscale);
664 if(!s->inter_intra_pred) return -1;
665 }
666 }
667 block[0] = level;
668
669 run_diff = s->msmpeg4_version >= 4;
670 i = 0;
671 if (!coded) {
672 goto not_coded;
673 }
674 if (s->ac_pred) {
675 if (dc_pred_dir == 0)
676 scan_table = s->intra_v_scantable.permutated; /* left */
677 else
678 scan_table = s->intra_h_scantable.permutated; /* top */
679 } else {
680 scan_table = s->intra_scantable.permutated;
681 }
682 rl_vlc= rl->rl_vlc[0];
683 } else {
684 qmul = s->qscale << 1;
685 qadd = (s->qscale - 1) | 1;
686 i = -1;
687 rl = &ff_rl_table[3 + s->rl_table_index];
688
689 if(s->msmpeg4_version==2)
690 run_diff = 0;
691 else
692 run_diff = 1;
693
694 if (!coded) {
695 s->block_last_index[n] = i;
696 return 0;
697 }
698 if(!scan_table)
699 scan_table = s->inter_scantable.permutated;
700 rl_vlc= rl->rl_vlc[s->qscale];
701 }
702 {
703 OPEN_READER(re, &s->gb);
704 for(;;) {
705 UPDATE_CACHE(re, &s->gb);
706 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 0);
707 if (level==0) {
708 int cache;
709 cache= GET_CACHE(re, &s->gb);
710 /* escape */
711 if (s->msmpeg4_version==1 || (cache&0x80000000)==0) {
712 if (s->msmpeg4_version==1 || (cache&0x40000000)==0) {
713 /* third escape */
714 if(s->msmpeg4_version!=1) LAST_SKIP_BITS(re, &s->gb, 2);
715 UPDATE_CACHE(re, &s->gb);
716 if(s->msmpeg4_version<=3){
717 last= SHOW_UBITS(re, &s->gb, 1); SKIP_CACHE(re, &s->gb, 1);
718 run= SHOW_UBITS(re, &s->gb, 6); SKIP_CACHE(re, &s->gb, 6);
719 level= SHOW_SBITS(re, &s->gb, 8);
720 SKIP_COUNTER(re, &s->gb, 1+6+8);
721 }else{
722 int sign;
723 last= SHOW_UBITS(re, &s->gb, 1); SKIP_BITS(re, &s->gb, 1);
724 if(!s->esc3_level_length){
725 int ll;
726 ff_dlog(s->avctx, "ESC-3 %X at %d %d\n",
727 show_bits(&s->gb, 24), s->mb_x, s->mb_y);
728 if(s->qscale<8){
729 ll= SHOW_UBITS(re, &s->gb, 3); SKIP_BITS(re, &s->gb, 3);
730 if(ll==0){
731 ll= 8+SHOW_UBITS(re, &s->gb, 1); SKIP_BITS(re, &s->gb, 1);
732 }
733 }else{
734 ll=2;
735 while(ll<8 && SHOW_UBITS(re, &s->gb, 1)==0){
736 ll++;
737 SKIP_BITS(re, &s->gb, 1);
738 }
739 if(ll<8) SKIP_BITS(re, &s->gb, 1);
740 }
741
742 s->esc3_level_length= ll;
743 s->esc3_run_length= SHOW_UBITS(re, &s->gb, 2) + 3; SKIP_BITS(re, &s->gb, 2);
744 UPDATE_CACHE(re, &s->gb);
745 }
746 run= SHOW_UBITS(re, &s->gb, s->esc3_run_length);
747 SKIP_BITS(re, &s->gb, s->esc3_run_length);
748
749 sign= SHOW_UBITS(re, &s->gb, 1);
750 SKIP_BITS(re, &s->gb, 1);
751
752 level= SHOW_UBITS(re, &s->gb, s->esc3_level_length);
753 SKIP_BITS(re, &s->gb, s->esc3_level_length);
754 if(sign) level= -level;
755 }
756
757 //level = level * qmul + (level>0) * qadd - (level<=0) * qadd ;
758 if (level>0) level= level * qmul + qadd;
759 else level= level * qmul - qadd;
760 i+= run + 1;
761 if(last) i+=192;
762 } else {
763 /* second escape */
764 SKIP_BITS(re, &s->gb, 2);
765 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
766 i+= run + rl->max_run[run>>7][level/qmul] + run_diff; //FIXME opt indexing
767 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
768 LAST_SKIP_BITS(re, &s->gb, 1);
769 }
770 } else {
771 /* first escape */
772 SKIP_BITS(re, &s->gb, 1);
773 GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
774 i+= run;
775 level = level + rl->max_level[run>>7][(run-1)&63] * qmul;//FIXME opt indexing
776 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
777 LAST_SKIP_BITS(re, &s->gb, 1);
778 }
779 } else {
780 i+= run;
781 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
782 LAST_SKIP_BITS(re, &s->gb, 1);
783 }
784 if (i > 62){
785 i-= 192;
786 if(i&(~63)){
787 const int left= get_bits_left(&s->gb);
788 if (((i + 192 == 64 && level / qmul == -1) ||
789 !(s->avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT))) &&
790 left >= 0) {
791 av_log(s->avctx, AV_LOG_ERROR, "ignoring overflow at %d %d\n", s->mb_x, s->mb_y);
792 i = 63;
793 break;
794 }else{
795 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
796 return -1;
797 }
798 }
799
800 block[scan_table[i]] = level;
801 break;
802 }
803
804 block[scan_table[i]] = level;
805 }
806 CLOSE_READER(re, &s->gb);
807 }
808 not_coded:
809 if (s->mb_intra) {
810 ff_mpeg4_pred_ac(s, block, n, dc_pred_dir);
811 if (s->ac_pred) {
812 i = 63; /* XXX: not optimal */
813 }
814 }
815 if(s->msmpeg4_version>=4 && i>0) i=63; //FIXME/XXX optimize
816 s->block_last_index[n] = i;
817
818 return 0;
819}
820
821int ff_msmpeg4_decode_motion(MpegEncContext * s,
822 int *mx_ptr, int *my_ptr)
823{
824 MVTable *mv;
825 int code, mx, my;
826
827 mv = &ff_mv_tables[s->mv_table_index];
828
829 code = get_vlc2(&s->gb, mv->vlc.table, MV_VLC_BITS, 2);
830 if (code < 0){
831 av_log(s->avctx, AV_LOG_ERROR, "illegal MV code at %d %d\n", s->mb_x, s->mb_y);
832 return -1;
833 }
834 if (code == mv->n) {
835 mx = get_bits(&s->gb, 6);
836 my = get_bits(&s->gb, 6);
837 } else {
838 mx = mv->table_mvx[code];
839 my = mv->table_mvy[code];
840 }
841
842 mx += *mx_ptr - 32;
843 my += *my_ptr - 32;
844 /* WARNING : they do not do exactly modulo encoding */
845 if (mx <= -64)
846 mx += 64;
847 else if (mx >= 64)
848 mx -= 64;
849
850 if (my <= -64)
851 my += 64;
852 else if (my >= 64)
853 my -= 64;
854 *mx_ptr = mx;
855 *my_ptr = my;
856 return 0;
857}
858
859AVCodec ff_msmpeg4v1_decoder = {
860 .name = "msmpeg4v1",
861 .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 1"),
862 .type = AVMEDIA_TYPE_VIDEO,
863 .id = AV_CODEC_ID_MSMPEG4V1,
864 .priv_data_size = sizeof(MpegEncContext),
865 .init = ff_msmpeg4_decode_init,
866 .close = ff_h263_decode_end,
867 .decode = ff_h263_decode_frame,
868 .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
869 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
870 .max_lowres = 3,
871 .pix_fmts = (const enum AVPixelFormat[]) {
872 AV_PIX_FMT_YUV420P,
873 AV_PIX_FMT_NONE
874 },
875};
876
877AVCodec ff_msmpeg4v2_decoder = {
878 .name = "msmpeg4v2",
879 .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 2"),
880 .type = AVMEDIA_TYPE_VIDEO,
881 .id = AV_CODEC_ID_MSMPEG4V2,
882 .priv_data_size = sizeof(MpegEncContext),
883 .init = ff_msmpeg4_decode_init,
884 .close = ff_h263_decode_end,
885 .decode = ff_h263_decode_frame,
886 .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
887 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
888 .max_lowres = 3,
889 .pix_fmts = (const enum AVPixelFormat[]) {
890 AV_PIX_FMT_YUV420P,
891 AV_PIX_FMT_NONE
892 },
893};
894
895AVCodec ff_msmpeg4v3_decoder = {
896 .name = "msmpeg4",
897 .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 3"),
898 .type = AVMEDIA_TYPE_VIDEO,
899 .id = AV_CODEC_ID_MSMPEG4V3,
900 .priv_data_size = sizeof(MpegEncContext),
901 .init = ff_msmpeg4_decode_init,
902 .close = ff_h263_decode_end,
903 .decode = ff_h263_decode_frame,
904 .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
905 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
906 .max_lowres = 3,
907 .pix_fmts = (const enum AVPixelFormat[]) {
908 AV_PIX_FMT_YUV420P,
909 AV_PIX_FMT_NONE
910 },
911};
912
913AVCodec ff_wmv1_decoder = {
914 .name = "wmv1",
915 .long_name = NULL_IF_CONFIG_SMALL("Windows Media Video 7"),
916 .type = AVMEDIA_TYPE_VIDEO,
917 .id = AV_CODEC_ID_WMV1,
918 .priv_data_size = sizeof(MpegEncContext),
919 .init = ff_msmpeg4_decode_init,
920 .close = ff_h263_decode_end,
921 .decode = ff_h263_decode_frame,
922 .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
923 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
924 .max_lowres = 3,
925 .pix_fmts = (const enum AVPixelFormat[]) {
926 AV_PIX_FMT_YUV420P,
927 AV_PIX_FMT_NONE
928 },
929};
930
generated by cgit at 2024-05-17 06:57:16 (GMT)