mess cleanup
[libav.git] / libavcodec / mpeg12.c
1 /*
2 * MPEG1 codec / MPEG2 decoder
3 * Copyright (c) 2000,2001 Fabrice Bellard.
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
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 * @file mpeg12.c
23 * MPEG1/2 codec
24 */
25
26 //#define DEBUG
27 #include "avcodec.h"
28 #include "dsputil.h"
29 #include "mpegvideo.h"
30
31 #include "mpeg12data.h"
32
33 //#undef NDEBUG
34 //#include <assert.h>
35
36
37 /* Start codes. */
38 #define SEQ_END_CODE 0x000001b7
39 #define SEQ_START_CODE 0x000001b3
40 #define GOP_START_CODE 0x000001b8
41 #define PICTURE_START_CODE 0x00000100
42 #define SLICE_MIN_START_CODE 0x00000101
43 #define SLICE_MAX_START_CODE 0x000001af
44 #define EXT_START_CODE 0x000001b5
45 #define USER_START_CODE 0x000001b2
46
47 #define DC_VLC_BITS 9
48 #define MV_VLC_BITS 9
49 #define MBINCR_VLC_BITS 9
50 #define MB_PAT_VLC_BITS 9
51 #define MB_PTYPE_VLC_BITS 6
52 #define MB_BTYPE_VLC_BITS 6
53 #define TEX_VLC_BITS 9
54
55 #ifdef CONFIG_ENCODERS
56 static void mpeg1_encode_block(MpegEncContext *s,
57 DCTELEM *block,
58 int component);
59 static void mpeg1_encode_motion(MpegEncContext *s, int val, int f_or_b_code); // RAL: f_code parameter added
60 #endif //CONFIG_ENCODERS
61 static inline int mpeg1_decode_block_inter(MpegEncContext *s,
62 DCTELEM *block,
63 int n);
64 static inline int mpeg1_decode_block_intra(MpegEncContext *s,
65 DCTELEM *block,
66 int n);
67 static inline int mpeg1_fast_decode_block_inter(MpegEncContext *s, DCTELEM *block, int n);
68 static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
69 DCTELEM *block,
70 int n);
71 static inline int mpeg2_decode_block_intra(MpegEncContext *s,
72 DCTELEM *block,
73 int n);
74 static inline int mpeg2_fast_decode_block_non_intra(MpegEncContext *s, DCTELEM *block, int n);
75 static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred);
76 static void exchange_uv(MpegEncContext *s);
77
78 #ifdef HAVE_XVMC
79 extern int XVMC_field_start(MpegEncContext *s, AVCodecContext *avctx);
80 extern int XVMC_field_end(MpegEncContext *s);
81 extern void XVMC_pack_pblocks(MpegEncContext *s,int cbp);
82 extern void XVMC_init_block(MpegEncContext *s);//set s->block
83 #endif
84
85 const enum PixelFormat pixfmt_yuv_420[]= {PIX_FMT_YUV420P,-1};
86 const enum PixelFormat pixfmt_yuv_422[]= {PIX_FMT_YUV422P,-1};
87 const enum PixelFormat pixfmt_yuv_444[]= {PIX_FMT_YUV444P,-1};
88 const enum PixelFormat pixfmt_xvmc_mpg2_420[] = {
89 PIX_FMT_XVMC_MPEG2_IDCT,
90 PIX_FMT_XVMC_MPEG2_MC,
91 -1};
92 #ifdef CONFIG_ENCODERS
93 static uint8_t (*mv_penalty)[MAX_MV*2+1]= NULL;
94 static uint8_t fcode_tab[MAX_MV*2+1];
95
96 static uint32_t uni_mpeg1_ac_vlc_bits[64*64*2];
97 static uint8_t uni_mpeg1_ac_vlc_len [64*64*2];
98
99 /* simple include everything table for dc, first byte is bits number next 3 are code*/
100 static uint32_t mpeg1_lum_dc_uni[512];
101 static uint32_t mpeg1_chr_dc_uni[512];
102
103 static uint8_t mpeg1_index_run[2][64];
104 static int8_t mpeg1_max_level[2][64];
105 #endif //CONFIG_ENCODERS
106
107 static void init_2d_vlc_rl(RLTable *rl)
108 {
109 int i;
110
111 init_vlc(&rl->vlc, TEX_VLC_BITS, rl->n + 2,
112 &rl->table_vlc[0][1], 4, 2,
113 &rl->table_vlc[0][0], 4, 2);
114
115
116 rl->rl_vlc[0]= av_malloc(rl->vlc.table_size*sizeof(RL_VLC_ELEM));
117 for(i=0; i<rl->vlc.table_size; i++){
118 int code= rl->vlc.table[i][0];
119 int len = rl->vlc.table[i][1];
120 int level, run;
121
122 if(len==0){ // illegal code
123 run= 65;
124 level= MAX_LEVEL;
125 }else if(len<0){ //more bits needed
126 run= 0;
127 level= code;
128 }else{
129 if(code==rl->n){ //esc
130 run= 65;
131 level= 0;
132 }else if(code==rl->n+1){ //eob
133 run= 0;
134 level= 127;
135 }else{
136 run= rl->table_run [code] + 1;
137 level= rl->table_level[code];
138 }
139 }
140 rl->rl_vlc[0][i].len= len;
141 rl->rl_vlc[0][i].level= level;
142 rl->rl_vlc[0][i].run= run;
143 }
144 }
145
146 #ifdef CONFIG_ENCODERS
147 static void init_uni_ac_vlc(RLTable *rl, uint32_t *uni_ac_vlc_bits, uint8_t *uni_ac_vlc_len){
148 int i;
149
150 for(i=0; i<128; i++){
151 int level= i-64;
152 int run;
153 for(run=0; run<64; run++){
154 int len, bits, code;
155
156 int alevel= ABS(level);
157 int sign= (level>>31)&1;
158
159 if (alevel > rl->max_level[0][run])
160 code= 111; /*rl->n*/
161 else
162 code= rl->index_run[0][run] + alevel - 1;
163
164 if (code < 111 /* rl->n */) {
165 /* store the vlc & sign at once */
166 len= mpeg1_vlc[code][1]+1;
167 bits= (mpeg1_vlc[code][0]<<1) + sign;
168 } else {
169 len= mpeg1_vlc[111/*rl->n*/][1]+6;
170 bits= mpeg1_vlc[111/*rl->n*/][0]<<6;
171
172 bits|= run;
173 if (alevel < 128) {
174 bits<<=8; len+=8;
175 bits|= level & 0xff;
176 } else {
177 bits<<=16; len+=16;
178 bits|= level & 0xff;
179 if (level < 0) {
180 bits|= 0x8001 + level + 255;
181 } else {
182 bits|= level & 0xffff;
183 }
184 }
185 }
186
187 uni_ac_vlc_bits[UNI_AC_ENC_INDEX(run, i)]= bits;
188 uni_ac_vlc_len [UNI_AC_ENC_INDEX(run, i)]= len;
189 }
190 }
191 }
192
193
194 static int find_frame_rate_index(MpegEncContext *s){
195 int i;
196 int64_t dmin= INT64_MAX;
197 int64_t d;
198
199 for(i=1;i<14;i++) {
200 int64_t n0= 1001LL/frame_rate_tab[i].den*frame_rate_tab[i].num*s->avctx->frame_rate_base;
201 int64_t n1= 1001LL*s->avctx->frame_rate;
202 if(s->avctx->strict_std_compliance >= 0 && i>=9) break;
203
204 d = ABS(n0 - n1);
205 if(d < dmin){
206 dmin=d;
207 s->frame_rate_index= i;
208 }
209 }
210 if(dmin)
211 return -1;
212 else
213 return 0;
214 }
215
216 static int encode_init(AVCodecContext *avctx)
217 {
218 MpegEncContext *s = avctx->priv_data;
219
220 if(MPV_encode_init(avctx) < 0)
221 return -1;
222
223 if(find_frame_rate_index(s) < 0){
224 if(s->strict_std_compliance >=0){
225 av_log(avctx, AV_LOG_ERROR, "MPEG1/2 doesnt support %d/%d fps\n", avctx->frame_rate, avctx->frame_rate_base);
226 return -1;
227 }else{
228 av_log(avctx, AV_LOG_INFO, "MPEG1/2 doesnt support %d/%d fps, there may be AV sync issues\n", avctx->frame_rate, avctx->frame_rate_base);
229 }
230 }
231
232 return 0;
233 }
234
235 static void put_header(MpegEncContext *s, int header)
236 {
237 align_put_bits(&s->pb);
238 put_bits(&s->pb, 16, header>>16);
239 put_bits(&s->pb, 16, header&0xFFFF);
240 }
241
242 /* put sequence header if needed */
243 static void mpeg1_encode_sequence_header(MpegEncContext *s)
244 {
245 unsigned int vbv_buffer_size;
246 unsigned int fps, v;
247 int i;
248 uint64_t time_code;
249 float best_aspect_error= 1E10;
250 float aspect_ratio= av_q2d(s->avctx->sample_aspect_ratio);
251 int constraint_parameter_flag;
252
253 if(aspect_ratio==0.0) aspect_ratio= 1.0; //pixel aspect 1:1 (VGA)
254
255 if (s->current_picture.key_frame) {
256 AVRational framerate= frame_rate_tab[s->frame_rate_index];
257
258 /* mpeg1 header repeated every gop */
259 put_header(s, SEQ_START_CODE);
260
261 put_bits(&s->pb, 12, s->width);
262 put_bits(&s->pb, 12, s->height);
263
264 for(i=1; i<15; i++){
265 float error= aspect_ratio;
266 if(s->codec_id == CODEC_ID_MPEG1VIDEO || i <=1)
267 error-= 1.0/mpeg1_aspect[i];
268 else
269 error-= av_q2d(mpeg2_aspect[i])*s->height/s->width;
270
271 error= ABS(error);
272
273 if(error < best_aspect_error){
274 best_aspect_error= error;
275 s->aspect_ratio_info= i;
276 }
277 }
278
279 put_bits(&s->pb, 4, s->aspect_ratio_info);
280 put_bits(&s->pb, 4, s->frame_rate_index);
281
282 if(s->avctx->rc_max_rate){
283 v = (s->avctx->rc_max_rate + 399) / 400;
284 if (v > 0x3ffff && s->codec_id == CODEC_ID_MPEG1VIDEO)
285 v = 0x3ffff;
286 }else{
287 v= 0x3FFFF;
288 }
289
290 if(s->avctx->rc_buffer_size)
291 vbv_buffer_size = s->avctx->rc_buffer_size;
292 else
293 /* VBV calculation: Scaled so that a VCD has the proper VBV size of 40 kilobytes */
294 vbv_buffer_size = (( 20 * s->bit_rate) / (1151929 / 2)) * 8 * 1024;
295 vbv_buffer_size= (vbv_buffer_size + 16383) / 16384;
296
297 put_bits(&s->pb, 18, v & 0x3FFFF);
298 put_bits(&s->pb, 1, 1); /* marker */
299 put_bits(&s->pb, 10, vbv_buffer_size & 0x3FF);
300
301 constraint_parameter_flag=
302 s->width <= 768 && s->height <= 576 &&
303 s->mb_width * s->mb_height <= 396 &&
304 s->mb_width * s->mb_height * framerate.num <= framerate.den*396*25 &&
305 framerate.num <= framerate.den*30 &&
306 vbv_buffer_size <= 20 &&
307 v <= 1856000/400 &&
308 s->codec_id == CODEC_ID_MPEG1VIDEO;
309
310 put_bits(&s->pb, 1, constraint_parameter_flag);
311
312 ff_write_quant_matrix(&s->pb, s->avctx->intra_matrix);
313 ff_write_quant_matrix(&s->pb, s->avctx->inter_matrix);
314
315 if(s->codec_id == CODEC_ID_MPEG2VIDEO){
316 put_header(s, EXT_START_CODE);
317 put_bits(&s->pb, 4, 1); //seq ext
318 put_bits(&s->pb, 1, 0); //esc
319
320 if(s->avctx->profile == FF_PROFILE_UNKNOWN){
321 put_bits(&s->pb, 3, 4); //profile
322 }else{
323 put_bits(&s->pb, 3, s->avctx->profile); //profile
324 }
325
326 if(s->avctx->level == FF_LEVEL_UNKNOWN){
327 put_bits(&s->pb, 4, 8); //level
328 }else{
329 put_bits(&s->pb, 4, s->avctx->level); //level
330 }
331
332 put_bits(&s->pb, 1, s->progressive_sequence);
333 put_bits(&s->pb, 2, 1); //chroma format 4:2:0
334 put_bits(&s->pb, 2, 0); //horizontal size ext
335 put_bits(&s->pb, 2, 0); //vertical size ext
336 put_bits(&s->pb, 12, v>>18); //bitrate ext
337 put_bits(&s->pb, 1, 1); //marker
338 put_bits(&s->pb, 8, vbv_buffer_size >>10); //vbv buffer ext
339 put_bits(&s->pb, 1, s->low_delay);
340 put_bits(&s->pb, 2, 0); // frame_rate_ext_n
341 put_bits(&s->pb, 5, 0); // frame_rate_ext_d
342 }
343
344 put_header(s, GOP_START_CODE);
345 put_bits(&s->pb, 1, 0); /* do drop frame */
346 /* time code : we must convert from the real frame rate to a
347 fake mpeg frame rate in case of low frame rate */
348 fps = (framerate.num + framerate.den/2)/ framerate.den;
349 time_code = s->current_picture_ptr->coded_picture_number;
350
351 s->gop_picture_number = time_code;
352 put_bits(&s->pb, 5, (uint32_t)((time_code / (fps * 3600)) % 24));
353 put_bits(&s->pb, 6, (uint32_t)((time_code / (fps * 60)) % 60));
354 put_bits(&s->pb, 1, 1);
355 put_bits(&s->pb, 6, (uint32_t)((time_code / fps) % 60));
356 put_bits(&s->pb, 6, (uint32_t)((time_code % fps)));
357 put_bits(&s->pb, 1, !!(s->flags & CODEC_FLAG_CLOSED_GOP));
358 put_bits(&s->pb, 1, 0); /* broken link */
359 }
360 }
361
362 static inline void encode_mb_skip_run(MpegEncContext *s, int run){
363 while (run >= 33) {
364 put_bits(&s->pb, 11, 0x008);
365 run -= 33;
366 }
367 put_bits(&s->pb, mbAddrIncrTable[run][1],
368 mbAddrIncrTable[run][0]);
369 }
370 #endif //CONFIG_ENCODERS
371
372 static void common_init(MpegEncContext *s)
373 {
374
375 s->y_dc_scale_table=
376 s->c_dc_scale_table= mpeg2_dc_scale_table[s->intra_dc_precision];
377
378 }
379
380 void ff_mpeg1_clean_buffers(MpegEncContext *s){
381 s->last_dc[0] = 1 << (7 + s->intra_dc_precision);
382 s->last_dc[1] = s->last_dc[0];
383 s->last_dc[2] = s->last_dc[0];
384 memset(s->last_mv, 0, sizeof(s->last_mv));
385 }
386
387 #ifdef CONFIG_ENCODERS
388
389 void ff_mpeg1_encode_slice_header(MpegEncContext *s){
390 put_header(s, SLICE_MIN_START_CODE + s->mb_y);
391 put_bits(&s->pb, 5, s->qscale); /* quantizer scale */
392 put_bits(&s->pb, 1, 0); /* slice extra information */
393 }
394
395 void mpeg1_encode_picture_header(MpegEncContext *s, int picture_number)
396 {
397 mpeg1_encode_sequence_header(s);
398
399 /* mpeg1 picture header */
400 put_header(s, PICTURE_START_CODE);
401 /* temporal reference */
402
403 // RAL: s->picture_number instead of s->fake_picture_number
404 put_bits(&s->pb, 10, (s->picture_number -
405 s->gop_picture_number) & 0x3ff);
406 put_bits(&s->pb, 3, s->pict_type);
407
408 s->vbv_delay_ptr= s->pb.buf + put_bits_count(&s->pb)/8;
409 put_bits(&s->pb, 16, 0xFFFF); /* vbv_delay */
410
411 // RAL: Forward f_code also needed for B frames
412 if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) {
413 put_bits(&s->pb, 1, 0); /* half pel coordinates */
414 if(s->codec_id == CODEC_ID_MPEG1VIDEO)
415 put_bits(&s->pb, 3, s->f_code); /* forward_f_code */
416 else
417 put_bits(&s->pb, 3, 7); /* forward_f_code */
418 }
419
420 // RAL: Backward f_code necessary for B frames
421 if (s->pict_type == B_TYPE) {
422 put_bits(&s->pb, 1, 0); /* half pel coordinates */
423 if(s->codec_id == CODEC_ID_MPEG1VIDEO)
424 put_bits(&s->pb, 3, s->b_code); /* backward_f_code */
425 else
426 put_bits(&s->pb, 3, 7); /* backward_f_code */
427 }
428
429 put_bits(&s->pb, 1, 0); /* extra bit picture */
430
431 s->frame_pred_frame_dct = 1;
432 if(s->codec_id == CODEC_ID_MPEG2VIDEO){
433 put_header(s, EXT_START_CODE);
434 put_bits(&s->pb, 4, 8); //pic ext
435 if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) {
436 put_bits(&s->pb, 4, s->f_code);
437 put_bits(&s->pb, 4, s->f_code);
438 }else{
439 put_bits(&s->pb, 8, 255);
440 }
441 if (s->pict_type == B_TYPE) {
442 put_bits(&s->pb, 4, s->b_code);
443 put_bits(&s->pb, 4, s->b_code);
444 }else{
445 put_bits(&s->pb, 8, 255);
446 }
447 put_bits(&s->pb, 2, s->intra_dc_precision);
448
449 assert(s->picture_structure == PICT_FRAME);
450 put_bits(&s->pb, 2, s->picture_structure);
451 if (s->progressive_sequence) {
452 put_bits(&s->pb, 1, 0); /* no repeat */
453 } else {
454 put_bits(&s->pb, 1, s->current_picture_ptr->top_field_first);
455 }
456 /* XXX: optimize the generation of this flag with entropy
457 measures */
458 s->frame_pred_frame_dct = s->progressive_sequence;
459
460 put_bits(&s->pb, 1, s->frame_pred_frame_dct);
461 put_bits(&s->pb, 1, s->concealment_motion_vectors);
462 put_bits(&s->pb, 1, s->q_scale_type);
463 put_bits(&s->pb, 1, s->intra_vlc_format);
464 put_bits(&s->pb, 1, s->alternate_scan);
465 put_bits(&s->pb, 1, s->repeat_first_field);
466 put_bits(&s->pb, 1, s->chroma_420_type=1);
467 s->progressive_frame = s->progressive_sequence;
468 put_bits(&s->pb, 1, s->progressive_frame);
469 put_bits(&s->pb, 1, 0); //composite_display_flag
470 }
471 if(s->flags & CODEC_FLAG_SVCD_SCAN_OFFSET){
472 int i;
473
474 put_header(s, USER_START_CODE);
475 for(i=0; i<sizeof(svcd_scan_offset_placeholder); i++){
476 put_bits(&s->pb, 8, svcd_scan_offset_placeholder[i]);
477 }
478 }
479
480 s->mb_y=0;
481 ff_mpeg1_encode_slice_header(s);
482 }
483
484 static inline void put_mb_modes(MpegEncContext *s, int n, int bits,
485 int has_mv, int field_motion)
486 {
487 put_bits(&s->pb, n, bits);
488 if (!s->frame_pred_frame_dct) {
489 if (has_mv)
490 put_bits(&s->pb, 2, 2 - field_motion); /* motion_type: frame/field */
491 put_bits(&s->pb, 1, s->interlaced_dct);
492 }
493 }
494
495 void mpeg1_encode_mb(MpegEncContext *s,
496 DCTELEM block[6][64],
497 int motion_x, int motion_y)
498 {
499 int i, cbp;
500 const int mb_x = s->mb_x;
501 const int mb_y = s->mb_y;
502 const int first_mb= mb_x == s->resync_mb_x && mb_y == s->resync_mb_y;
503
504 /* compute cbp */
505 cbp = 0;
506 for(i=0;i<6;i++) {
507 if (s->block_last_index[i] >= 0)
508 cbp |= 1 << (5 - i);
509 }
510
511 if (cbp == 0 && !first_mb && s->mv_type == MV_TYPE_16X16 &&
512 (mb_x != s->mb_width - 1 || (mb_y != s->mb_height - 1 && s->codec_id == CODEC_ID_MPEG1VIDEO)) &&
513 ((s->pict_type == P_TYPE && (motion_x | motion_y) == 0) ||
514 (s->pict_type == B_TYPE && s->mv_dir == s->last_mv_dir && (((s->mv_dir & MV_DIR_FORWARD) ? ((s->mv[0][0][0] - s->last_mv[0][0][0])|(s->mv[0][0][1] - s->last_mv[0][0][1])) : 0) |
515 ((s->mv_dir & MV_DIR_BACKWARD) ? ((s->mv[1][0][0] - s->last_mv[1][0][0])|(s->mv[1][0][1] - s->last_mv[1][0][1])) : 0)) == 0))) {
516 s->mb_skip_run++;
517 s->qscale -= s->dquant;
518 s->skip_count++;
519 s->misc_bits++;
520 s->last_bits++;
521 if(s->pict_type == P_TYPE){
522 s->last_mv[0][1][0]= s->last_mv[0][0][0]=
523 s->last_mv[0][1][1]= s->last_mv[0][0][1]= 0;
524 }
525 } else {
526 if(first_mb){
527 assert(s->mb_skip_run == 0);
528 encode_mb_skip_run(s, s->mb_x);
529 }else{
530 encode_mb_skip_run(s, s->mb_skip_run);
531 }
532
533 if (s->pict_type == I_TYPE) {
534 if(s->dquant && cbp){
535 put_mb_modes(s, 2, 1, 0, 0); /* macroblock_type : macroblock_quant = 1 */
536 put_bits(&s->pb, 5, s->qscale);
537 }else{
538 put_mb_modes(s, 1, 1, 0, 0); /* macroblock_type : macroblock_quant = 0 */
539 s->qscale -= s->dquant;
540 }
541 s->misc_bits+= get_bits_diff(s);
542 s->i_count++;
543 } else if (s->mb_intra) {
544 if(s->dquant && cbp){
545 put_mb_modes(s, 6, 0x01, 0, 0);
546 put_bits(&s->pb, 5, s->qscale);
547 }else{
548 put_mb_modes(s, 5, 0x03, 0, 0);
549 s->qscale -= s->dquant;
550 }
551 s->misc_bits+= get_bits_diff(s);
552 s->i_count++;
553 memset(s->last_mv, 0, sizeof(s->last_mv));
554 } else if (s->pict_type == P_TYPE) {
555 if(s->mv_type == MV_TYPE_16X16){
556 if (cbp != 0) {
557 if ((motion_x|motion_y) == 0) {
558 if(s->dquant){
559 put_mb_modes(s, 5, 1, 0, 0); /* macroblock_pattern & quant */
560 put_bits(&s->pb, 5, s->qscale);
561 }else{
562 put_mb_modes(s, 2, 1, 0, 0); /* macroblock_pattern only */
563 }
564 s->misc_bits+= get_bits_diff(s);
565 } else {
566 if(s->dquant){
567 put_mb_modes(s, 5, 2, 1, 0); /* motion + cbp */
568 put_bits(&s->pb, 5, s->qscale);
569 }else{
570 put_mb_modes(s, 1, 1, 1, 0); /* motion + cbp */
571 }
572 s->misc_bits+= get_bits_diff(s);
573 mpeg1_encode_motion(s, motion_x - s->last_mv[0][0][0], s->f_code); // RAL: f_code parameter added
574 mpeg1_encode_motion(s, motion_y - s->last_mv[0][0][1], s->f_code); // RAL: f_code parameter added
575 s->mv_bits+= get_bits_diff(s);
576 }
577 } else {
578 put_bits(&s->pb, 3, 1); /* motion only */
579 if (!s->frame_pred_frame_dct)
580 put_bits(&s->pb, 2, 2); /* motion_type: frame */
581 s->misc_bits+= get_bits_diff(s);
582 mpeg1_encode_motion(s, motion_x - s->last_mv[0][0][0], s->f_code); // RAL: f_code parameter added
583 mpeg1_encode_motion(s, motion_y - s->last_mv[0][0][1], s->f_code); // RAL: f_code parameter added
584 s->qscale -= s->dquant;
585 s->mv_bits+= get_bits_diff(s);
586 }
587 s->last_mv[0][1][0]= s->last_mv[0][0][0]= motion_x;
588 s->last_mv[0][1][1]= s->last_mv[0][0][1]= motion_y;
589 }else{
590 assert(!s->frame_pred_frame_dct && s->mv_type == MV_TYPE_FIELD);
591
592 if (cbp) {
593 if(s->dquant){
594 put_mb_modes(s, 5, 2, 1, 1); /* motion + cbp */
595 put_bits(&s->pb, 5, s->qscale);
596 }else{
597 put_mb_modes(s, 1, 1, 1, 1); /* motion + cbp */
598 }
599 } else {
600 put_bits(&s->pb, 3, 1); /* motion only */
601 put_bits(&s->pb, 2, 1); /* motion_type: field */
602 s->qscale -= s->dquant;
603 }
604 s->misc_bits+= get_bits_diff(s);
605 for(i=0; i<2; i++){
606 put_bits(&s->pb, 1, s->field_select[0][i]);
607 mpeg1_encode_motion(s, s->mv[0][i][0] - s->last_mv[0][i][0] , s->f_code);
608 mpeg1_encode_motion(s, s->mv[0][i][1] - (s->last_mv[0][i][1]>>1), s->f_code);
609 s->last_mv[0][i][0]= s->mv[0][i][0];
610 s->last_mv[0][i][1]= 2*s->mv[0][i][1];
611 }
612 s->mv_bits+= get_bits_diff(s);
613 }
614 if(cbp)
615 put_bits(&s->pb, mbPatTable[cbp][1], mbPatTable[cbp][0]);
616 s->f_count++;
617 } else{
618 static const int mb_type_len[4]={0,3,4,2}; //bak,for,bi
619
620 if(s->mv_type == MV_TYPE_16X16){
621 if (cbp){ // With coded bloc pattern
622 if (s->dquant) {
623 if(s->mv_dir == MV_DIR_FORWARD)
624 put_mb_modes(s, 6, 3, 1, 0);
625 else
626 put_mb_modes(s, mb_type_len[s->mv_dir]+3, 2, 1, 0);
627 put_bits(&s->pb, 5, s->qscale);
628 } else {
629 put_mb_modes(s, mb_type_len[s->mv_dir], 3, 1, 0);
630 }
631 }else{ // No coded bloc pattern
632 put_bits(&s->pb, mb_type_len[s->mv_dir], 2);
633 if (!s->frame_pred_frame_dct)
634 put_bits(&s->pb, 2, 2); /* motion_type: frame */
635 s->qscale -= s->dquant;
636 }
637 s->misc_bits += get_bits_diff(s);
638 if (s->mv_dir&MV_DIR_FORWARD){
639 mpeg1_encode_motion(s, s->mv[0][0][0] - s->last_mv[0][0][0], s->f_code);
640 mpeg1_encode_motion(s, s->mv[0][0][1] - s->last_mv[0][0][1], s->f_code);
641 s->last_mv[0][0][0]=s->last_mv[0][1][0]= s->mv[0][0][0];
642 s->last_mv[0][0][1]=s->last_mv[0][1][1]= s->mv[0][0][1];
643 s->f_count++;
644 }
645 if (s->mv_dir&MV_DIR_BACKWARD){
646 mpeg1_encode_motion(s, s->mv[1][0][0] - s->last_mv[1][0][0], s->b_code);
647 mpeg1_encode_motion(s, s->mv[1][0][1] - s->last_mv[1][0][1], s->b_code);
648 s->last_mv[1][0][0]=s->last_mv[1][1][0]= s->mv[1][0][0];
649 s->last_mv[1][0][1]=s->last_mv[1][1][1]= s->mv[1][0][1];
650 s->b_count++;
651 }
652 }else{
653 assert(s->mv_type == MV_TYPE_FIELD);
654 assert(!s->frame_pred_frame_dct);
655 if (cbp){ // With coded bloc pattern
656 if (s->dquant) {
657 if(s->mv_dir == MV_DIR_FORWARD)
658 put_mb_modes(s, 6, 3, 1, 1);
659 else
660 put_mb_modes(s, mb_type_len[s->mv_dir]+3, 2, 1, 1);
661 put_bits(&s->pb, 5, s->qscale);
662 } else {
663 put_mb_modes(s, mb_type_len[s->mv_dir], 3, 1, 1);
664 }
665 }else{ // No coded bloc pattern
666 put_bits(&s->pb, mb_type_len[s->mv_dir], 2);
667 put_bits(&s->pb, 2, 1); /* motion_type: field */
668 s->qscale -= s->dquant;
669 }
670 s->misc_bits += get_bits_diff(s);
671 if (s->mv_dir&MV_DIR_FORWARD){
672 for(i=0; i<2; i++){
673 put_bits(&s->pb, 1, s->field_select[0][i]);
674 mpeg1_encode_motion(s, s->mv[0][i][0] - s->last_mv[0][i][0] , s->f_code);
675 mpeg1_encode_motion(s, s->mv[0][i][1] - (s->last_mv[0][i][1]>>1), s->f_code);
676 s->last_mv[0][i][0]= s->mv[0][i][0];
677 s->last_mv[0][i][1]= 2*s->mv[0][i][1];
678 }
679 s->f_count++;
680 }
681 if (s->mv_dir&MV_DIR_BACKWARD){
682 for(i=0; i<2; i++){
683 put_bits(&s->pb, 1, s->field_select[1][i]);
684 mpeg1_encode_motion(s, s->mv[1][i][0] - s->last_mv[1][i][0] , s->b_code);
685 mpeg1_encode_motion(s, s->mv[1][i][1] - (s->last_mv[1][i][1]>>1), s->b_code);
686 s->last_mv[1][i][0]= s->mv[1][i][0];
687 s->last_mv[1][i][1]= 2*s->mv[1][i][1];
688 }
689 s->b_count++;
690 }
691 }
692 s->mv_bits += get_bits_diff(s);
693 if(cbp)
694 put_bits(&s->pb, mbPatTable[cbp][1], mbPatTable[cbp][0]);
695 }
696 for(i=0;i<6;i++) {
697 if (cbp & (1 << (5 - i))) {
698 mpeg1_encode_block(s, block[i], i);
699 }
700 }
701 s->mb_skip_run = 0;
702 if(s->mb_intra)
703 s->i_tex_bits+= get_bits_diff(s);
704 else
705 s->p_tex_bits+= get_bits_diff(s);
706 }
707 }
708
709 // RAL: Parameter added: f_or_b_code
710 static void mpeg1_encode_motion(MpegEncContext *s, int val, int f_or_b_code)
711 {
712 int code, bit_size, l, bits, range, sign;
713
714 if (val == 0) {
715 /* zero vector */
716 code = 0;
717 put_bits(&s->pb,
718 mbMotionVectorTable[0][1],
719 mbMotionVectorTable[0][0]);
720 } else {
721 bit_size = f_or_b_code - 1;
722 range = 1 << bit_size;
723 /* modulo encoding */
724 l= INT_BIT - 5 - bit_size;
725 val= (val<<l)>>l;
726
727 if (val >= 0) {
728 val--;
729 code = (val >> bit_size) + 1;
730 bits = val & (range - 1);
731 sign = 0;
732 } else {
733 val = -val;
734 val--;
735 code = (val >> bit_size) + 1;
736 bits = val & (range - 1);
737 sign = 1;
738 }
739
740 assert(code > 0 && code <= 16);
741
742 put_bits(&s->pb,
743 mbMotionVectorTable[code][1],
744 mbMotionVectorTable[code][0]);
745
746 put_bits(&s->pb, 1, sign);
747 if (bit_size > 0) {
748 put_bits(&s->pb, bit_size, bits);
749 }
750 }
751 }
752
753 void ff_mpeg1_encode_init(MpegEncContext *s)
754 {
755 static int done=0;
756
757 common_init(s);
758
759 if(!done){
760 int f_code;
761 int mv;
762 int i;
763
764 done=1;
765 init_rl(&rl_mpeg1);
766
767 for(i=0; i<64; i++)
768 {
769 mpeg1_max_level[0][i]= rl_mpeg1.max_level[0][i];
770 mpeg1_index_run[0][i]= rl_mpeg1.index_run[0][i];
771 }
772
773 init_uni_ac_vlc(&rl_mpeg1, uni_mpeg1_ac_vlc_bits, uni_mpeg1_ac_vlc_len);
774
775 /* build unified dc encoding tables */
776 for(i=-255; i<256; i++)
777 {
778 int adiff, index;
779 int bits, code;
780 int diff=i;
781
782 adiff = ABS(diff);
783 if(diff<0) diff--;
784 index = av_log2(2*adiff);
785
786 bits= vlc_dc_lum_bits[index] + index;
787 code= (vlc_dc_lum_code[index]<<index) + (diff & ((1 << index) - 1));
788 mpeg1_lum_dc_uni[i+255]= bits + (code<<8);
789
790 bits= vlc_dc_chroma_bits[index] + index;
791 code= (vlc_dc_chroma_code[index]<<index) + (diff & ((1 << index) - 1));
792 mpeg1_chr_dc_uni[i+255]= bits + (code<<8);
793 }
794
795 mv_penalty= av_mallocz( sizeof(uint8_t)*(MAX_FCODE+1)*(2*MAX_MV+1) );
796
797 for(f_code=1; f_code<=MAX_FCODE; f_code++){
798 for(mv=-MAX_MV; mv<=MAX_MV; mv++){
799 int len;
800
801 if(mv==0) len= mbMotionVectorTable[0][1];
802 else{
803 int val, bit_size, range, code;
804
805 bit_size = f_code - 1;
806 range = 1 << bit_size;
807
808 val=mv;
809 if (val < 0)
810 val = -val;
811 val--;
812 code = (val >> bit_size) + 1;
813 if(code<17){
814 len= mbMotionVectorTable[code][1] + 1 + bit_size;
815 }else{
816 len= mbMotionVectorTable[16][1] + 2 + bit_size;
817 }
818 }
819
820 mv_penalty[f_code][mv+MAX_MV]= len;
821 }
822 }
823
824
825 for(f_code=MAX_FCODE; f_code>0; f_code--){
826 for(mv=-(8<<f_code); mv<(8<<f_code); mv++){
827 fcode_tab[mv+MAX_MV]= f_code;
828 }
829 }
830 }
831 s->me.mv_penalty= mv_penalty;
832 s->fcode_tab= fcode_tab;
833 if(s->codec_id == CODEC_ID_MPEG1VIDEO){
834 s->min_qcoeff=-255;
835 s->max_qcoeff= 255;
836 }else{
837 s->min_qcoeff=-2047;
838 s->max_qcoeff= 2047;
839 }
840 s->intra_ac_vlc_length=
841 s->inter_ac_vlc_length=
842 s->intra_ac_vlc_last_length=
843 s->inter_ac_vlc_last_length= uni_mpeg1_ac_vlc_len;
844 }
845
846 static inline void encode_dc(MpegEncContext *s, int diff, int component)
847 {
848 if(((unsigned) (diff+255)) >= 511){
849 int index;
850
851 if(diff<0){
852 index= av_log2_16bit(-2*diff);
853 diff--;
854 }else{
855 index= av_log2_16bit(2*diff);
856 }
857 if (component == 0) {
858 put_bits(
859 &s->pb,
860 vlc_dc_lum_bits[index] + index,
861 (vlc_dc_lum_code[index]<<index) + (diff & ((1 << index) - 1)));
862 }else{
863 put_bits(
864 &s->pb,
865 vlc_dc_chroma_bits[index] + index,
866 (vlc_dc_chroma_code[index]<<index) + (diff & ((1 << index) - 1)));
867 }
868 }else{
869 if (component == 0) {
870 put_bits(
871 &s->pb,
872 mpeg1_lum_dc_uni[diff+255]&0xFF,
873 mpeg1_lum_dc_uni[diff+255]>>8);
874 } else {
875 put_bits(
876 &s->pb,
877 mpeg1_chr_dc_uni[diff+255]&0xFF,
878 mpeg1_chr_dc_uni[diff+255]>>8);
879 }
880 }
881 }
882
883 static void mpeg1_encode_block(MpegEncContext *s,
884 DCTELEM *block,
885 int n)
886 {
887 int alevel, level, last_non_zero, dc, diff, i, j, run, last_index, sign;
888 int code, component;
889 // RLTable *rl = &rl_mpeg1;
890
891 last_index = s->block_last_index[n];
892
893 /* DC coef */
894 if (s->mb_intra) {
895 component = (n <= 3 ? 0 : n - 4 + 1);
896 dc = block[0]; /* overflow is impossible */
897 diff = dc - s->last_dc[component];
898 encode_dc(s, diff, component);
899 s->last_dc[component] = dc;
900 i = 1;
901 /*
902 if (s->intra_vlc_format)
903 rl = &rl_mpeg2;
904 else
905 rl = &rl_mpeg1;
906 */
907 } else {
908 /* encode the first coefficient : needs to be done here because
909 it is handled slightly differently */
910 level = block[0];
911 if (abs(level) == 1) {
912 code = ((uint32_t)level >> 31); /* the sign bit */
913 put_bits(&s->pb, 2, code | 0x02);
914 i = 1;
915 } else {
916 i = 0;
917 last_non_zero = -1;
918 goto next_coef;
919 }
920 }
921
922 /* now quantify & encode AC coefs */
923 last_non_zero = i - 1;
924
925 for(;i<=last_index;i++) {
926 j = s->intra_scantable.permutated[i];
927 level = block[j];
928 next_coef:
929 #if 0
930 if (level != 0)
931 dprintf("level[%d]=%d\n", i, level);
932 #endif
933 /* encode using VLC */
934 if (level != 0) {
935 run = i - last_non_zero - 1;
936
937 alevel= level;
938 MASK_ABS(sign, alevel)
939 sign&=1;
940
941 // code = get_rl_index(rl, 0, run, alevel);
942 if (alevel <= mpeg1_max_level[0][run]){
943 code= mpeg1_index_run[0][run] + alevel - 1;
944 /* store the vlc & sign at once */
945 put_bits(&s->pb, mpeg1_vlc[code][1]+1, (mpeg1_vlc[code][0]<<1) + sign);
946 } else {
947 /* escape seems to be pretty rare <5% so i dont optimize it */
948 put_bits(&s->pb, mpeg1_vlc[111/*rl->n*/][1], mpeg1_vlc[111/*rl->n*/][0]);
949 /* escape: only clip in this case */
950 put_bits(&s->pb, 6, run);
951 if(s->codec_id == CODEC_ID_MPEG1VIDEO){
952 if (alevel < 128) {
953 put_bits(&s->pb, 8, level & 0xff);
954 } else {
955 if (level < 0) {
956 put_bits(&s->pb, 16, 0x8001 + level + 255);
957 } else {
958 put_bits(&s->pb, 16, level & 0xffff);
959 }
960 }
961 }else{
962 put_bits(&s->pb, 12, level & 0xfff);
963 }
964 }
965 last_non_zero = i;
966 }
967 }
968 /* end of block */
969 put_bits(&s->pb, 2, 0x2);
970 }
971 #endif //CONFIG_ENCODERS
972
973 /******************************************/
974 /* decoding */
975
976 static VLC dc_lum_vlc;
977 static VLC dc_chroma_vlc;
978 static VLC mv_vlc;
979 static VLC mbincr_vlc;
980 static VLC mb_ptype_vlc;
981 static VLC mb_btype_vlc;
982 static VLC mb_pat_vlc;
983
984 static void init_vlcs(void)
985 {
986 static int done = 0;
987
988 if (!done) {
989 done = 1;
990
991 init_vlc(&dc_lum_vlc, DC_VLC_BITS, 12,
992 vlc_dc_lum_bits, 1, 1,
993 vlc_dc_lum_code, 2, 2);
994 init_vlc(&dc_chroma_vlc, DC_VLC_BITS, 12,
995 vlc_dc_chroma_bits, 1, 1,
996 vlc_dc_chroma_code, 2, 2);
997 init_vlc(&mv_vlc, MV_VLC_BITS, 17,
998 &mbMotionVectorTable[0][1], 2, 1,
999 &mbMotionVectorTable[0][0], 2, 1);
1000 init_vlc(&mbincr_vlc, MBINCR_VLC_BITS, 36,
1001 &mbAddrIncrTable[0][1], 2, 1,
1002 &mbAddrIncrTable[0][0], 2, 1);
1003 init_vlc(&mb_pat_vlc, MB_PAT_VLC_BITS, 64,
1004 &mbPatTable[0][1], 2, 1,
1005 &mbPatTable[0][0], 2, 1);
1006
1007 init_vlc(&mb_ptype_vlc, MB_PTYPE_VLC_BITS, 7,
1008 &table_mb_ptype[0][1], 2, 1,
1009 &table_mb_ptype[0][0], 2, 1);
1010 init_vlc(&mb_btype_vlc, MB_BTYPE_VLC_BITS, 11,
1011 &table_mb_btype[0][1], 2, 1,
1012 &table_mb_btype[0][0], 2, 1);
1013 init_rl(&rl_mpeg1);
1014 init_rl(&rl_mpeg2);
1015
1016 init_2d_vlc_rl(&rl_mpeg1);
1017 init_2d_vlc_rl(&rl_mpeg2);
1018 }
1019 }
1020
1021 static inline int get_dmv(MpegEncContext *s)
1022 {
1023 if(get_bits1(&s->gb))
1024 return 1 - (get_bits1(&s->gb) << 1);
1025 else
1026 return 0;
1027 }
1028
1029 static inline int get_qscale(MpegEncContext *s)
1030 {
1031 int qscale = get_bits(&s->gb, 5);
1032 if (s->q_scale_type) {
1033 return non_linear_qscale[qscale];
1034 } else {
1035 return qscale << 1;
1036 }
1037 }
1038
1039 /* motion type (for mpeg2) */
1040 #define MT_FIELD 1
1041 #define MT_FRAME 2
1042 #define MT_16X8 2
1043 #define MT_DMV 3
1044
1045 static int mpeg_decode_mb(MpegEncContext *s,
1046 DCTELEM block[12][64])
1047 {
1048 int i, j, k, cbp, val, mb_type, motion_type;
1049 const int mb_block_count = 4 + (1<< s->chroma_format)
1050
1051 dprintf("decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
1052
1053 assert(s->mb_skiped==0);
1054
1055 if (s->mb_skip_run-- != 0) {
1056 if(s->pict_type == I_TYPE){
1057 av_log(s->avctx, AV_LOG_ERROR, "skiped MB in I frame at %d %d\n", s->mb_x, s->mb_y);
1058 return -1;
1059 }
1060
1061 /* skip mb */
1062 s->mb_intra = 0;
1063 for(i=0;i<12;i++)
1064 s->block_last_index[i] = -1;
1065 if(s->picture_structure == PICT_FRAME)
1066 s->mv_type = MV_TYPE_16X16;
1067 else
1068 s->mv_type = MV_TYPE_FIELD;
1069 if (s->pict_type == P_TYPE) {
1070 /* if P type, zero motion vector is implied */
1071 s->mv_dir = MV_DIR_FORWARD;
1072 s->mv[0][0][0] = s->mv[0][0][1] = 0;
1073 s->last_mv[0][0][0] = s->last_mv[0][0][1] = 0;
1074 s->last_mv[0][1][0] = s->last_mv[0][1][1] = 0;
1075 s->field_select[0][0]= s->picture_structure - 1;
1076 s->mb_skiped = 1;
1077 s->current_picture.mb_type[ s->mb_x + s->mb_y*s->mb_stride ]= MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
1078 } else {
1079 /* if B type, reuse previous vectors and directions */
1080 s->mv[0][0][0] = s->last_mv[0][0][0];
1081 s->mv[0][0][1] = s->last_mv[0][0][1];
1082 s->mv[1][0][0] = s->last_mv[1][0][0];
1083 s->mv[1][0][1] = s->last_mv[1][0][1];
1084
1085 s->current_picture.mb_type[ s->mb_x + s->mb_y*s->mb_stride ]=
1086 s->current_picture.mb_type[ s->mb_x + s->mb_y*s->mb_stride - 1] | MB_TYPE_SKIP;
1087 // assert(s->current_picture.mb_type[ s->mb_x + s->mb_y*s->mb_stride - 1]&(MB_TYPE_16x16|MB_TYPE_16x8));
1088
1089 if((s->mv[0][0][0]|s->mv[0][0][1]|s->mv[1][0][0]|s->mv[1][0][1])==0)
1090 s->mb_skiped = 1;
1091 }
1092
1093 return 0;
1094 }
1095
1096 switch(s->pict_type) {
1097 default:
1098 case I_TYPE:
1099 if (get_bits1(&s->gb) == 0) {
1100 if (get_bits1(&s->gb) == 0){
1101 av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in I Frame at %d %d\n", s->mb_x, s->mb_y);
1102 return -1;
1103 }
1104 mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
1105 } else {
1106 mb_type = MB_TYPE_INTRA;
1107 }
1108 break;
1109 case P_TYPE:
1110 mb_type = get_vlc2(&s->gb, mb_ptype_vlc.table, MB_PTYPE_VLC_BITS, 1);
1111 if (mb_type < 0){
1112 av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in P Frame at %d %d\n", s->mb_x, s->mb_y);
1113 return -1;
1114 }
1115 mb_type = ptype2mb_type[ mb_type ];
1116 break;
1117 case B_TYPE:
1118 mb_type = get_vlc2(&s->gb, mb_btype_vlc.table, MB_BTYPE_VLC_BITS, 1);
1119 if (mb_type < 0){
1120 av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in B Frame at %d %d\n", s->mb_x, s->mb_y);
1121 return -1;
1122 }
1123 mb_type = btype2mb_type[ mb_type ];
1124 break;
1125 }
1126 dprintf("mb_type=%x\n", mb_type);
1127 // motion_type = 0; /* avoid warning */
1128 if (IS_INTRA(mb_type)) {
1129 /* compute dct type */
1130 if (s->picture_structure == PICT_FRAME && //FIXME add a interlaced_dct coded var?
1131 !s->frame_pred_frame_dct) {
1132 s->interlaced_dct = get_bits1(&s->gb);
1133 }
1134
1135 if (IS_QUANT(mb_type))
1136 s->qscale = get_qscale(s);
1137
1138 if (s->concealment_motion_vectors) {
1139 /* just parse them */
1140 if (s->picture_structure != PICT_FRAME)
1141 skip_bits1(&s->gb); /* field select */
1142
1143 s->mv[0][0][0]= s->last_mv[0][0][0]= s->last_mv[0][1][0] =
1144 mpeg_decode_motion(s, s->mpeg_f_code[0][0], s->last_mv[0][0][0]);
1145 s->mv[0][0][1]= s->last_mv[0][0][1]= s->last_mv[0][1][1] =
1146 mpeg_decode_motion(s, s->mpeg_f_code[0][1], s->last_mv[0][0][1]);
1147
1148 skip_bits1(&s->gb); /* marker */
1149 }else
1150 memset(s->last_mv, 0, sizeof(s->last_mv)); /* reset mv prediction */
1151 s->mb_intra = 1;
1152 #ifdef HAVE_XVMC
1153 //one 1 we memcpy blocks in xvmcvideo
1154 if(s->avctx->xvmc_acceleration > 1){
1155 XVMC_pack_pblocks(s,-1);//inter are always full blocks
1156 if(s->swap_uv){
1157 exchange_uv(s);
1158 }
1159 }
1160 #endif
1161
1162 if (s->codec_id == CODEC_ID_MPEG2VIDEO) {
1163 for(i=0;i<mb_block_count;i++) {
1164 if (mpeg2_decode_block_intra(s, s->pblocks[i], i) < 0)
1165 return -1;
1166 }
1167 } else {
1168 for(i=0;i<6;i++) {
1169 if (mpeg1_decode_block_intra(s, s->pblocks[i], i) < 0)
1170 return -1;
1171 }
1172 }
1173 } else {
1174 if (mb_type & MB_TYPE_ZERO_MV){
1175 assert(mb_type & MB_TYPE_CBP);
1176
1177 /* compute dct type */
1178 if (s->picture_structure == PICT_FRAME && //FIXME add a interlaced_dct coded var?
1179 !s->frame_pred_frame_dct) {
1180 s->interlaced_dct = get_bits1(&s->gb);
1181 }
1182
1183 if (IS_QUANT(mb_type))
1184 s->qscale = get_qscale(s);
1185
1186 s->mv_dir = MV_DIR_FORWARD;
1187 if(s->picture_structure == PICT_FRAME)
1188 s->mv_type = MV_TYPE_16X16;
1189 else{
1190 s->mv_type = MV_TYPE_FIELD;
1191 mb_type |= MB_TYPE_INTERLACED;
1192 s->field_select[0][0]= s->picture_structure - 1;
1193 }
1194 s->last_mv[0][0][0] = 0;
1195 s->last_mv[0][0][1] = 0;
1196 s->last_mv[0][1][0] = 0;
1197 s->last_mv[0][1][1] = 0;
1198 s->mv[0][0][0] = 0;
1199 s->mv[0][0][1] = 0;
1200 }else{
1201 assert(mb_type & MB_TYPE_L0L1);
1202 //FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
1203 /* get additionnal motion vector type */
1204 if (s->frame_pred_frame_dct)
1205 motion_type = MT_FRAME;
1206 else{
1207 motion_type = get_bits(&s->gb, 2);
1208 }
1209
1210 /* compute dct type */
1211 if (s->picture_structure == PICT_FRAME && //FIXME add a interlaced_dct coded var?
1212 !s->frame_pred_frame_dct && HAS_CBP(mb_type)) {
1213 s->interlaced_dct = get_bits1(&s->gb);
1214 }
1215
1216 if (IS_QUANT(mb_type))
1217 s->qscale = get_qscale(s);
1218
1219 /* motion vectors */
1220 s->mv_dir = 0;
1221 for(i=0;i<2;i++) {
1222 if (USES_LIST(mb_type, i)) {
1223 s->mv_dir |= (MV_DIR_FORWARD >> i);
1224 dprintf("motion_type=%d\n", motion_type);
1225 switch(motion_type) {
1226 case MT_FRAME: /* or MT_16X8 */
1227 if (s->picture_structure == PICT_FRAME) {
1228 /* MT_FRAME */
1229 mb_type |= MB_TYPE_16x16;
1230 s->mv_type = MV_TYPE_16X16;
1231 s->mv[i][0][0]= s->last_mv[i][0][0]= s->last_mv[i][1][0] =
1232 mpeg_decode_motion(s, s->mpeg_f_code[i][0], s->last_mv[i][0][0]);
1233 s->mv[i][0][1]= s->last_mv[i][0][1]= s->last_mv[i][1][1] =
1234 mpeg_decode_motion(s, s->mpeg_f_code[i][1], s->last_mv[i][0][1]);
1235 /* full_pel: only for mpeg1 */
1236 if (s->full_pel[i]){
1237 s->mv[i][0][0] <<= 1;
1238 s->mv[i][0][1] <<= 1;
1239 }
1240 } else {
1241 /* MT_16X8 */
1242 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
1243 s->mv_type = MV_TYPE_16X8;
1244 for(j=0;j<2;j++) {
1245 s->field_select[i][j] = get_bits1(&s->gb);
1246 for(k=0;k<2;k++) {
1247 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
1248 s->last_mv[i][j][k]);
1249 s->last_mv[i][j][k] = val;
1250 s->mv[i][j][k] = val;
1251 }
1252 }
1253 }
1254 break;
1255 case MT_FIELD:
1256 s->mv_type = MV_TYPE_FIELD;
1257 if (s->picture_structure == PICT_FRAME) {
1258 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
1259 for(j=0;j<2;j++) {
1260 s->field_select[i][j] = get_bits1(&s->gb);
1261 val = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
1262 s->last_mv[i][j][0]);
1263 s->last_mv[i][j][0] = val;
1264 s->mv[i][j][0] = val;
1265 dprintf("fmx=%d\n", val);
1266 val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
1267 s->last_mv[i][j][1] >> 1);
1268 s->last_mv[i][j][1] = val << 1;
1269 s->mv[i][j][1] = val;
1270 dprintf("fmy=%d\n", val);
1271 }
1272 } else {
1273 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
1274 s->field_select[i][0] = get_bits1(&s->gb);
1275 for(k=0;k<2;k++) {
1276 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
1277 s->last_mv[i][0][k]);
1278 s->last_mv[i][0][k] = val;
1279 s->last_mv[i][1][k] = val;
1280 s->mv[i][0][k] = val;
1281 }
1282 }
1283 break;
1284 case MT_DMV:
1285 {
1286 int dmx, dmy, mx, my, m;
1287
1288 mx = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
1289 s->last_mv[i][0][0]);
1290 s->last_mv[i][0][0] = mx;
1291 s->last_mv[i][1][0] = mx;
1292 dmx = get_dmv(s);
1293 my = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
1294 s->last_mv[i][0][1] >> 1);
1295 dmy = get_dmv(s);
1296 s->mv_type = MV_TYPE_DMV;
1297
1298
1299 s->last_mv[i][0][1] = my<<1;
1300 s->last_mv[i][1][1] = my<<1;
1301
1302 s->mv[i][0][0] = mx;
1303 s->mv[i][0][1] = my;
1304 s->mv[i][1][0] = mx;//not used
1305 s->mv[i][1][1] = my;//not used
1306
1307 if (s->picture_structure == PICT_FRAME) {
1308 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
1309
1310 //m = 1 + 2 * s->top_field_first;
1311 m = s->top_field_first ? 1 : 3;
1312
1313 /* top -> top pred */
1314 s->mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
1315 s->mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
1316 m = 4 - m;
1317 s->mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
1318 s->mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
1319 } else {
1320 mb_type |= MB_TYPE_16x16;
1321
1322 s->mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
1323 s->mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
1324 if(s->picture_structure == PICT_TOP_FIELD)
1325 s->mv[i][2][1]--;
1326 else
1327 s->mv[i][2][1]++;
1328 }
1329 }
1330 break;
1331 default:
1332 av_log(s->avctx, AV_LOG_ERROR, "00 motion_type at %d %d\n", s->mb_x, s->mb_y);
1333 return -1;
1334 }
1335 }
1336 }
1337 }
1338
1339 s->mb_intra = 0;
1340
1341 if (HAS_CBP(mb_type)) {
1342 cbp = get_vlc2(&s->gb, mb_pat_vlc.table, MB_PAT_VLC_BITS, 1);
1343 if (cbp < 0 || ((cbp == 0) && (s->chroma_format < 2)) ){
1344 av_log(s->avctx, AV_LOG_ERROR, "invalid cbp at %d %d\n", s->mb_x, s->mb_y);
1345 return -1;
1346 }
1347 if(mb_block_count > 6){
1348 cbp<<= mb_block_count-6;
1349 cbp |= get_bits(&s->gb, mb_block_count-6);
1350 }
1351
1352 #ifdef HAVE_XVMC
1353 //on 1 we memcpy blocks in xvmcvideo
1354 if(s->avctx->xvmc_acceleration > 1){
1355 XVMC_pack_pblocks(s,cbp);
1356 if(s->swap_uv){
1357 exchange_uv(s);
1358 }
1359 }
1360 #endif
1361
1362 if (s->codec_id == CODEC_ID_MPEG2VIDEO) {
1363 if(s->flags2 & CODEC_FLAG2_FAST){
1364 for(i=0;i<6;i++) {
1365 if(cbp & 32) {
1366 mpeg2_fast_decode_block_non_intra(s, s->pblocks[i], i);
1367 } else {
1368 s->block_last_index[i] = -1;
1369 }
1370 cbp+=cbp;
1371 }
1372 }else{
1373 cbp<<= 12-mb_block_count;
1374
1375 for(i=0;i<mb_block_count;i++) {
1376 if ( cbp & (1<<11) ) {
1377 if (mpeg2_decode_block_non_intra(s, s->pblocks[i], i) < 0)
1378 return -1;
1379 } else {
1380 s->block_last_index[i] = -1;
1381 }
1382 cbp+=cbp;
1383 }
1384 }
1385 } else {
1386 if(s->flags2 & CODEC_FLAG2_FAST){
1387 for(i=0;i<6;i++) {
1388 if (cbp & 32) {
1389 mpeg1_fast_decode_block_inter(s, s->pblocks[i], i);
1390 } else {
1391 s->block_last_index[i] = -1;
1392 }
1393 cbp+=cbp;
1394 }
1395 }else{
1396 for(i=0;i<6;i++) {
1397 if (cbp & 32) {
1398 if (mpeg1_decode_block_inter(s, s->pblocks[i], i) < 0)
1399 return -1;
1400 } else {
1401 s->block_last_index[i] = -1;
1402 }
1403 cbp+=cbp;
1404 }
1405 }
1406 }
1407 }else{
1408 for(i=0;i<6;i++)
1409 s->block_last_index[i] = -1;
1410 }
1411 }
1412
1413 s->current_picture.mb_type[ s->mb_x + s->mb_y*s->mb_stride ]= mb_type;
1414
1415 return 0;
1416 }
1417
1418 /* as h263, but only 17 codes */
1419 static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
1420 {
1421 int code, sign, val, l, shift;
1422
1423 code = get_vlc2(&s->gb, mv_vlc.table, MV_VLC_BITS, 2);
1424 if (code == 0) {
1425 return pred;
1426 }
1427 if (code < 0) {
1428 return 0xffff;
1429 }
1430
1431 sign = get_bits1(&s->gb);
1432 shift = fcode - 1;
1433 val = code;
1434 if (shift) {
1435 val = (val - 1) << shift;
1436 val |= get_bits(&s->gb, shift);
1437 val++;
1438 }
1439 if (sign)
1440 val = -val;
1441 val += pred;
1442
1443 /* modulo decoding */
1444 l= INT_BIT - 5 - shift;
1445 val = (val<<l)>>l;
1446 return val;
1447 }
1448
1449 static inline int decode_dc(GetBitContext *gb, int component)
1450 {
1451 int code, diff;
1452
1453 if (component == 0) {
1454 code = get_vlc2(gb, dc_lum_vlc.table, DC_VLC_BITS, 2);
1455 } else {
1456 code = get_vlc2(gb, dc_chroma_vlc.table, DC_VLC_BITS, 2);
1457 }
1458 if (code < 0){
1459 av_log(NULL, AV_LOG_ERROR, "invalid dc code at\n");
1460 return 0xffff;
1461 }
1462 if (code == 0) {
1463 diff = 0;
1464 } else {
1465 diff = get_xbits(gb, code);
1466 }
1467 return diff;
1468 }
1469
1470 static inline int mpeg1_decode_block_intra(MpegEncContext *s,
1471 DCTELEM *block,
1472 int n)
1473 {
1474 int level, dc, diff, i, j, run;
1475 int component;
1476 RLTable *rl = &rl_mpeg1;
1477 uint8_t * const scantable= s->intra_scantable.permutated;
1478 const uint16_t *quant_matrix= s->intra_matrix;
1479 const int qscale= s->qscale;
1480
1481 /* DC coef */
1482 component = (n <= 3 ? 0 : n - 4 + 1);
1483 diff = decode_dc(&s->gb, component);
1484 if (diff >= 0xffff)
1485 return -1;
1486 dc = s->last_dc[component];
1487 dc += diff;
1488 s->last_dc[component] = dc;
1489 block[0] = dc<<3;
1490 dprintf("dc=%d diff=%d\n", dc, diff);
1491 i = 0;
1492 {
1493 OPEN_READER(re, &s->gb);
1494 /* now quantify & encode AC coefs */
1495 for(;;) {
1496 UPDATE_CACHE(re, &s->gb);
1497 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1498
1499 if(level == 127){
1500 break;
1501 } else if(level != 0) {
1502 i += run;
1503 j = scantable[i];
1504 level= (level*qscale*quant_matrix[j])>>4;
1505 level= (level-1)|1;
1506 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1507 LAST_SKIP_BITS(re, &s->gb, 1);
1508 } else {
1509 /* escape */
1510 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1511 UPDATE_CACHE(re, &s->gb);
1512 level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
1513 if (level == -128) {
1514 level = SHOW_UBITS(re, &s->gb, 8) - 256; LAST_SKIP_BITS(re, &s->gb, 8);
1515 } else if (level == 0) {
1516 level = SHOW_UBITS(re, &s->gb, 8) ; LAST_SKIP_BITS(re, &s->gb, 8);
1517 }
1518 i += run;
1519 j = scantable[i];
1520 if(level<0){
1521 level= -level;
1522 level= (level*qscale*quant_matrix[j])>>4;
1523 level= (level-1)|1;
1524 level= -level;
1525 }else{
1526 level= (level*qscale*quant_matrix[j])>>4;
1527 level= (level-1)|1;
1528 }
1529 }
1530 if (i > 63){
1531 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
1532 return -1;
1533 }
1534
1535 block[j] = level;
1536 }
1537 CLOSE_READER(re, &s->gb);
1538 }
1539 s->block_last_index[n] = i;
1540 return 0;
1541 }
1542
1543 static inline int mpeg1_decode_block_inter(MpegEncContext *s,
1544 DCTELEM *block,
1545 int n)
1546 {
1547 int level, i, j, run;
1548 RLTable *rl = &rl_mpeg1;
1549 uint8_t * const scantable= s->intra_scantable.permutated;
1550 const uint16_t *quant_matrix= s->inter_matrix;
1551 const int qscale= s->qscale;
1552
1553 {
1554 int v;
1555 OPEN_READER(re, &s->gb);
1556 i = -1;
1557 /* special case for the first coef. no need to add a second vlc table */
1558 UPDATE_CACHE(re, &s->gb);
1559 v= SHOW_UBITS(re, &s->gb, 2);
1560 if (v & 2) {
1561 LAST_SKIP_BITS(re, &s->gb, 2);
1562 level= (3*qscale*quant_matrix[0])>>5;
1563 level= (level-1)|1;
1564 if(v&1)
1565 level= -level;
1566 block[0] = level;
1567 i++;
1568 }
1569
1570 /* now quantify & encode AC coefs */
1571 for(;;) {
1572 UPDATE_CACHE(re, &s->gb);
1573 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1574
1575 if(level == 127){
1576 break;
1577 } else if(level != 0) {
1578 i += run;
1579 j = scantable[i];
1580 level= ((level*2+1)*qscale*quant_matrix[j])>>5;
1581 level= (level-1)|1;
1582 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1583 LAST_SKIP_BITS(re, &s->gb, 1);
1584 } else {
1585 /* escape */
1586 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1587 UPDATE_CACHE(re, &s->gb);
1588 level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
1589 if (level == -128) {
1590 level = SHOW_UBITS(re, &s->gb, 8) - 256; LAST_SKIP_BITS(re, &s->gb, 8);
1591 } else if (level == 0) {
1592 level = SHOW_UBITS(re, &s->gb, 8) ; LAST_SKIP_BITS(re, &s->gb, 8);
1593 }
1594 i += run;
1595 j = scantable[i];
1596 if(level<0){
1597 level= -level;
1598 level= ((level*2+1)*qscale*quant_matrix[j])>>5;
1599 level= (level-1)|1;
1600 level= -level;
1601 }else{
1602 level= ((level*2+1)*qscale*quant_matrix[j])>>5;
1603 level= (level-1)|1;
1604 }
1605 }
1606 if (i > 63){
1607 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
1608 return -1;
1609 }
1610
1611 block[j] = level;
1612 }
1613 CLOSE_READER(re, &s->gb);
1614 }
1615 s->block_last_index[n] = i;
1616 return 0;
1617 }
1618
1619 static inline int mpeg1_fast_decode_block_inter(MpegEncContext *s, DCTELEM *block, int n)
1620 {
1621 int level, i, j, run;
1622 RLTable *rl = &rl_mpeg1;
1623 uint8_t * const scantable= s->intra_scantable.permutated;
1624 const int qscale= s->qscale;
1625
1626 {
1627 int v;
1628 OPEN_READER(re, &s->gb);
1629 i = -1;
1630 /* special case for the first coef. no need to add a second vlc table */
1631 UPDATE_CACHE(re, &s->gb);
1632 v= SHOW_UBITS(re, &s->gb, 2);
1633 if (v & 2) {
1634 LAST_SKIP_BITS(re, &s->gb, 2);
1635 level= (3*qscale)>>1;
1636 level= (level-1)|1;
1637 if(v&1)
1638 level= -level;
1639 block[0] = level;
1640 i++;
1641 }
1642
1643 /* now quantify & encode AC coefs */
1644 for(;;) {
1645 UPDATE_CACHE(re, &s->gb);
1646 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1647
1648 if(level == 127){
1649 break;
1650 } else if(level != 0) {
1651 i += run;
1652 j = scantable[i];
1653 level= ((level*2+1)*qscale)>>1;
1654 level= (level-1)|1;
1655 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1656 LAST_SKIP_BITS(re, &s->gb, 1);
1657 } else {
1658 /* escape */
1659 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1660 UPDATE_CACHE(re, &s->gb);
1661 level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
1662 if (level == -128) {
1663 level = SHOW_UBITS(re, &s->gb, 8) - 256; LAST_SKIP_BITS(re, &s->gb, 8);
1664 } else if (level == 0) {
1665 level = SHOW_UBITS(re, &s->gb, 8) ; LAST_SKIP_BITS(re, &s->gb, 8);
1666 }
1667 i += run;
1668 j = scantable[i];
1669 if(level<0){
1670 level= -level;
1671 level= ((level*2+1)*qscale)>>1;
1672 level= (level-1)|1;
1673 level= -level;
1674 }else{
1675 level= ((level*2+1)*qscale)>>1;
1676 level= (level-1)|1;
1677 }
1678 }
1679
1680 block[j] = level;
1681 }
1682 CLOSE_READER(re, &s->gb);
1683 }
1684 s->block_last_index[n] = i;
1685 return 0;
1686 }
1687
1688
1689 static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
1690 DCTELEM *block,
1691 int n)
1692 {
1693 int level, i, j, run;
1694 RLTable *rl = &rl_mpeg1;
1695 uint8_t * const scantable= s->intra_scantable.permutated;
1696 const uint16_t *quant_matrix;
1697 const int qscale= s->qscale;
1698 int mismatch;
1699
1700 mismatch = 1;
1701
1702 {
1703 int v;
1704 OPEN_READER(re, &s->gb);
1705 i = -1;
1706 if (n < 4)
1707 quant_matrix = s->inter_matrix;
1708 else
1709 quant_matrix = s->chroma_inter_matrix;
1710
1711 /* special case for the first coef. no need to add a second vlc table */
1712 UPDATE_CACHE(re, &s->gb);
1713 v= SHOW_UBITS(re, &s->gb, 2);
1714 if (v & 2) {
1715 LAST_SKIP_BITS(re, &s->gb, 2);
1716 level= (3*qscale*quant_matrix[0])>>5;
1717 if(v&1)
1718 level= -level;
1719 block[0] = level;
1720 mismatch ^= level;
1721 i++;
1722 }
1723
1724 /* now quantify & encode AC coefs */
1725 for(;;) {
1726 UPDATE_CACHE(re, &s->gb);
1727 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1728
1729 if(level == 127){
1730 break;
1731 } else if(level != 0) {
1732 i += run;
1733 j = scantable[i];
1734 level= ((level*2+1)*qscale*quant_matrix[j])>>5;
1735 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1736 LAST_SKIP_BITS(re, &s->gb, 1);
1737 } else {
1738 /* escape */
1739 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1740 UPDATE_CACHE(re, &s->gb);
1741 level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
1742
1743 i += run;
1744 j = scantable[i];
1745 if(level<0){
1746 level= ((-level*2+1)*qscale*quant_matrix[j])>>5;
1747 level= -level;
1748 }else{
1749 level= ((level*2+1)*qscale*quant_matrix[j])>>5;
1750 }
1751 }
1752 if (i > 63){
1753 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
1754 return -1;
1755 }
1756
1757 mismatch ^= level;
1758 block[j] = level;
1759 }
1760 CLOSE_READER(re, &s->gb);
1761 }
1762 block[63] ^= (mismatch & 1);
1763
1764 s->block_last_index[n] = i;
1765 return 0;
1766 }
1767
1768 static inline int mpeg2_fast_decode_block_non_intra(MpegEncContext *s,
1769 DCTELEM *block,
1770 int n)
1771 {
1772 int level, i, j, run;
1773 RLTable *rl = &rl_mpeg1;
1774 uint8_t * const scantable= s->intra_scantable.permutated;
1775 const int qscale= s->qscale;
1776 int v;
1777 OPEN_READER(re, &s->gb);
1778 i = -1;
1779
1780 /* special case for the first coef. no need to add a second vlc table */
1781 UPDATE_CACHE(re, &s->gb);
1782 v= SHOW_UBITS(re, &s->gb, 2);
1783 if (v & 2) {
1784 LAST_SKIP_BITS(re, &s->gb, 2);
1785 level= (3*qscale)>>1;
1786 if(v&1)
1787 level= -level;
1788 block[0] = level;
1789 i++;
1790 }
1791
1792 /* now quantify & encode AC coefs */
1793 for(;;) {
1794 UPDATE_CACHE(re, &s->gb);
1795 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1796
1797 if(level == 127){
1798 break;
1799 } else if(level != 0) {
1800 i += run;
1801 j = scantable[i];
1802 level= ((level*2+1)*qscale)>>1;
1803 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1804 LAST_SKIP_BITS(re, &s->gb, 1);
1805 } else {
1806 /* escape */
1807 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1808 UPDATE_CACHE(re, &s->gb);
1809 level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
1810
1811 i += run;
1812 j = scantable[i];
1813 if(level<0){
1814 level= ((-level*2+1)*qscale)>>1;
1815 level= -level;
1816 }else{
1817 level= ((level*2+1)*qscale)>>1;
1818 }
1819 }
1820
1821 block[j] = level;
1822 }
1823 CLOSE_READER(re, &s->gb);
1824 s->block_last_index[n] = i;
1825 return 0;
1826 }
1827
1828
1829 static inline int mpeg2_decode_block_intra(MpegEncContext *s,
1830 DCTELEM *block,
1831 int n)
1832 {
1833 int level, dc, diff, i, j, run;
1834 int component;
1835 RLTable *rl;
1836 uint8_t * const scantable= s->intra_scantable.permutated;
1837 const uint16_t *quant_matrix;
1838 const int qscale= s->qscale;
1839 int mismatch;
1840
1841 /* DC coef */
1842 if (n < 4){
1843 quant_matrix = s->intra_matrix;
1844 component = 0;
1845 }else{
1846 quant_matrix = s->chroma_intra_matrix;
1847 component = (n&1) + 1;
1848 }
1849 diff = decode_dc(&s->gb, component);
1850 if (diff >= 0xffff)
1851 return -1;
1852 dc = s->last_dc[component];
1853 dc += diff;
1854 s->last_dc[component] = dc;
1855 block[0] = dc << (3 - s->intra_dc_precision);
1856 dprintf("dc=%d\n", block[0]);
1857 mismatch = block[0] ^ 1;
1858 i = 0;
1859 if (s->intra_vlc_format)
1860 rl = &rl_mpeg2;
1861 else
1862 rl = &rl_mpeg1;
1863
1864 {
1865 OPEN_READER(re, &s->gb);
1866 /* now quantify & encode AC coefs */
1867 for(;;) {
1868 UPDATE_CACHE(re, &s->gb);
1869 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2);
1870
1871 if(level == 127){
1872 break;
1873 } else if(level != 0) {
1874 i += run;
1875 j = scantable[i];
1876 level= (level*qscale*quant_matrix[j])>>4;
1877 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
1878 LAST_SKIP_BITS(re, &s->gb, 1);
1879 } else {
1880 /* escape */
1881 run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
1882 UPDATE_CACHE(re, &s->gb);
1883 level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
1884 i += run;
1885 j = scantable[i];
1886 if(level<0){
1887 level= (-level*qscale*quant_matrix[j])>>4;
1888 level= -level;
1889 }else{
1890 level= (level*qscale*quant_matrix[j])>>4;
1891 }
1892 }
1893 if (i > 63){
1894 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
1895 return -1;
1896 }
1897
1898 mismatch^= level;
1899 block[j] = level;
1900 }
1901 CLOSE_READER(re, &s->gb);
1902 }
1903 block[63]^= mismatch&1;
1904
1905 s->block_last_index[n] = i;
1906 return 0;
1907 }
1908
1909 typedef struct Mpeg1Context {
1910 MpegEncContext mpeg_enc_ctx;
1911 int mpeg_enc_ctx_allocated; /* true if decoding context allocated */
1912 int repeat_field; /* true if we must repeat the field */
1913 AVPanScan pan_scan; /** some temporary storage for the panscan */
1914 int slice_count;
1915 int swap_uv;//indicate VCR2
1916 int save_aspect_info;
1917 AVRational frame_rate_ext; ///< MPEG-2 specific framerate modificator
1918
1919 } Mpeg1Context;
1920
1921 static int mpeg_decode_init(AVCodecContext *avctx)
1922 {
1923 Mpeg1Context *s = avctx->priv_data;
1924 MpegEncContext *s2 = &s->mpeg_enc_ctx;
1925 int i;
1926
1927 //we need some parmutation to store
1928 //matrixes, until MPV_common_init()
1929 //set the real permutatuon
1930 for(i=0;i<64;i++)
1931 s2->dsp.idct_permutation[i]=i;
1932
1933 MPV_decode_defaults(s2);
1934
1935 s->mpeg_enc_ctx.avctx= avctx;
1936 s->mpeg_enc_ctx.flags= avctx->flags;
1937 s->mpeg_enc_ctx.flags2= avctx->flags2;
1938 common_init(&s->mpeg_enc_ctx);
1939 init_vlcs();
1940
1941 s->mpeg_enc_ctx_allocated = 0;
1942 s->mpeg_enc_ctx.picture_number = 0;
1943 s->repeat_field = 0;
1944 s->mpeg_enc_ctx.codec_id= avctx->codec->id;
1945 return 0;
1946 }
1947
1948 static void quant_matrix_rebuild(uint16_t *matrix, const uint8_t *old_perm,
1949 const uint8_t *new_perm){
1950 uint16_t temp_matrix[64];
1951 int i;
1952
1953 memcpy(temp_matrix,matrix,64*sizeof(uint16_t));
1954
1955 for(i=0;i<64;i++){
1956 matrix[new_perm[i]] = temp_matrix[old_perm[i]];
1957 }
1958 }
1959
1960 //Call this function when we know all parameters
1961 //it may be called in different places for mpeg1 and mpeg2
1962 static int mpeg_decode_postinit(AVCodecContext *avctx){
1963 Mpeg1Context *s1 = avctx->priv_data;
1964 MpegEncContext *s = &s1->mpeg_enc_ctx;
1965 uint8_t old_permutation[64];
1966
1967 if (
1968 (s1->mpeg_enc_ctx_allocated == 0)||
1969 avctx->width != s->width ||
1970 avctx->height != s->height||
1971 // s1->save_aspect_info != avctx->aspect_ratio_info||
1972 0)
1973 {
1974
1975 if (s1->mpeg_enc_ctx_allocated) {
1976 MPV_common_end(s);
1977 }
1978
1979 if( (s->width == 0 )||(s->height == 0))
1980 return -2;
1981
1982 avctx->width = s->width;
1983 avctx->height = s->height;
1984 avctx->bit_rate = s->bit_rate;
1985 s1->save_aspect_info = s->aspect_ratio_info;
1986
1987 //low_delay may be forced, in this case we will have B frames
1988 //that behave like P frames
1989 avctx->has_b_frames = !(s->low_delay);
1990
1991 if(avctx->sub_id==1){//s->codec_id==avctx->codec_id==CODEC_ID
1992 //mpeg1 fps
1993 avctx->frame_rate = frame_rate_tab[s->frame_rate_index].num;
1994 avctx->frame_rate_base= frame_rate_tab[s->frame_rate_index].den;
1995 //mpeg1 aspect
1996 avctx->sample_aspect_ratio= av_d2q(
1997 1.0/mpeg1_aspect[s->aspect_ratio_info], 255);
1998
1999 }else{//mpeg2
2000 //mpeg2 fps
2001 av_reduce(
2002 &s->avctx->frame_rate,
2003 &s->avctx->frame_rate_base,
2004 frame_rate_tab[s->frame_rate_index].num * s1->frame_rate_ext.num,
2005 frame_rate_tab[s->frame_rate_index].den * s1->frame_rate_ext.den,
2006 1<<30);
2007 //mpeg2 aspect
2008 if(s->aspect_ratio_info > 1){
2009 if( (s1->pan_scan.width == 0 )||(s1->pan_scan.height == 0) ){
2010 s->avctx->sample_aspect_ratio=
2011 av_div_q(
2012 mpeg2_aspect[s->aspect_ratio_info],
2013 (AVRational){s->width, s->height}
2014 );
2015 }else{
2016 s->avctx->sample_aspect_ratio=
2017 av_div_q(
2018 mpeg2_aspect[s->aspect_ratio_info],
2019 (AVRational){s1->pan_scan.width, s1->pan_scan.height}
2020 );
2021 }
2022 }else{
2023 s->avctx->sample_aspect_ratio=
2024 mpeg2_aspect[s->aspect_ratio_info];
2025 }
2026 }//mpeg2
2027
2028 if(avctx->xvmc_acceleration){
2029 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_xvmc_mpg2_420);
2030 }else{
2031 if(s->chroma_format < 2){
2032 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_yuv_420);
2033 }else
2034 if(s->chroma_format == 2){
2035 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_yuv_422);
2036 }else
2037 if(s->chroma_format > 2){
2038 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_yuv_444);
2039 }
2040 }
2041 //until then pix_fmt may be changed right after codec init
2042 if( avctx->pix_fmt == PIX_FMT_XVMC_MPEG2_IDCT )
2043 if( avctx->idct_algo == FF_IDCT_AUTO )
2044 avctx->idct_algo = FF_IDCT_SIMPLE;
2045
2046 //quantization matrixes may need reordering
2047 //if dct permutation is changed
2048 memcpy(old_permutation,s->dsp.idct_permutation,64*sizeof(uint8_t));
2049
2050 if (MPV_common_init(s) < 0)
2051 return -2;
2052
2053 quant_matrix_rebuild(s->intra_matrix, old_permutation,s->dsp.idct_permutation);
2054 quant_matrix_rebuild(s->inter_matrix, old_permutation,s->dsp.idct_permutation);
2055 quant_matrix_rebuild(s->chroma_intra_matrix,old_permutation,s->dsp.idct_permutation);
2056 quant_matrix_rebuild(s->chroma_inter_matrix,old_permutation,s->dsp.idct_permutation);
2057
2058 s1->mpeg_enc_ctx_allocated = 1;
2059 }
2060 return 0;
2061 }
2062
2063 /* return the 8 bit start code value and update the search
2064 state. Return -1 if no start code found */
2065 static int find_start_code(const uint8_t **pbuf_ptr, const uint8_t *buf_end)
2066 {
2067 const uint8_t *buf_ptr= *pbuf_ptr;
2068
2069 buf_ptr++; //gurantees that -1 is within the array
2070 buf_end -= 2; // gurantees that +2 is within the array
2071
2072 while (buf_ptr < buf_end) {
2073 if(*buf_ptr==0){
2074 while(buf_ptr < buf_end && buf_ptr[1]==0)
2075 buf_ptr++;
2076
2077 if(buf_ptr[-1] == 0 && buf_ptr[1] == 1){
2078 *pbuf_ptr = buf_ptr+3;
2079 return buf_ptr[2] + 0x100;
2080 }
2081 }
2082 buf_ptr += 2;
2083 }
2084 buf_end += 2; //undo the hack above
2085
2086 *pbuf_ptr = buf_end;
2087 return -1;
2088 }
2089
2090 static int mpeg1_decode_picture(AVCodecContext *avctx,
2091 const uint8_t *buf, int buf_size)
2092 {
2093 Mpeg1Context *s1 = avctx->priv_data;
2094 MpegEncContext *s = &s1->mpeg_enc_ctx;
2095 int ref, f_code, vbv_delay;
2096
2097 if(mpeg_decode_postinit(s->avctx) < 0)
2098 return -2;
2099
2100 init_get_bits(&s->gb, buf, buf_size*8);
2101
2102 ref = get_bits(&s->gb, 10); /* temporal ref */
2103 s->pict_type = get_bits(&s->gb, 3);
2104
2105 vbv_delay= get_bits(&s->gb, 16);
2106 if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) {
2107 s->full_pel[0] = get_bits1(&s->gb);
2108 f_code = get_bits(&s->gb, 3);
2109 if (f_code == 0)
2110 return -1;
2111 s->mpeg_f_code[0][0] = f_code;
2112 s->mpeg_f_code[0][1] = f_code;
2113 }
2114 if (s->pict_type == B_TYPE) {
2115 s->full_pel[1] = get_bits1(&s->gb);
2116 f_code = get_bits(&s->gb, 3);
2117 if (f_code == 0)
2118 return -1;
2119 s->mpeg_f_code[1][0] = f_code;
2120 s->mpeg_f_code[1][1] = f_code;
2121 }
2122 s->current_picture.pict_type= s->pict_type;
2123 s->current_picture.key_frame= s->pict_type == I_TYPE;
2124
2125 // if(avctx->debug & FF_DEBUG_PICT_INFO)
2126 // av_log(avctx, AV_LOG_DEBUG, "vbv_delay %d, ref %d\n", vbv_delay, ref);
2127
2128 s->y_dc_scale = 8;
2129 s->c_dc_scale = 8;
2130 s->first_slice = 1;
2131 return 0;
2132 }
2133
2134 static void mpeg_decode_sequence_extension(Mpeg1Context *s1)
2135 {
2136 MpegEncContext *s= &s1->mpeg_enc_ctx;
2137 int horiz_size_ext, vert_size_ext;
2138 int bit_rate_ext;
2139
2140 skip_bits(&s->gb, 1); /* profil and level esc*/
2141 s->avctx->profile= get_bits(&s->gb, 3);
2142 s->avctx->level= get_bits(&s->gb, 4);
2143 s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
2144 s->chroma_format = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
2145 horiz_size_ext = get_bits(&s->gb, 2);
2146 vert_size_ext = get_bits(&s->gb, 2);
2147 s->width |= (horiz_size_ext << 12);
2148 s->height |= (vert_size_ext << 12);
2149 bit_rate_ext = get_bits(&s->gb, 12); /* XXX: handle it */
2150 s->bit_rate += (bit_rate_ext << 12) * 400;
2151 skip_bits1(&s->gb); /* marker */
2152 s->avctx->rc_buffer_size += get_bits(&s->gb, 8)*1024*16<<10;
2153
2154 s->low_delay = get_bits1(&s->gb);
2155 if(s->flags & CODEC_FLAG_LOW_DELAY) s->low_delay=1;
2156
2157 s1->frame_rate_ext.num = get_bits(&s->gb, 2)+1;
2158 s1->frame_rate_ext.den = get_bits(&s->gb, 5)+1;
2159
2160 dprintf("sequence extension\n");
2161 s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG2VIDEO;
2162 s->avctx->sub_id = 2; /* indicates mpeg2 found */
2163
2164 if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2165 av_log(s->avctx, AV_LOG_DEBUG, "profile: %d, level: %d vbv buffer: %d, bitrate:%d\n",
2166 s->avctx->profile, s->avctx->level, s->avctx->rc_buffer_size, s->bit_rate);
2167
2168 }
2169
2170 static void mpeg_decode_sequence_display_extension(Mpeg1Context *s1)
2171 {
2172 MpegEncContext *s= &s1->mpeg_enc_ctx;
2173 int color_description, w, h;
2174
2175 skip_bits(&s->gb, 3); /* video format */
2176 color_description= get_bits1(&s->gb);
2177 if(color_description){
2178 skip_bits(&s->gb, 8); /* color primaries */
2179 skip_bits(&s->gb, 8); /* transfer_characteristics */
2180 skip_bits(&s->gb, 8); /* matrix_coefficients */
2181 }
2182 w= get_bits(&s->gb, 14);
2183 skip_bits(&s->gb, 1); //marker
2184 h= get_bits(&s->gb, 14);
2185 skip_bits(&s->gb, 1); //marker
2186
2187 s1->pan_scan.width= 16*w;
2188 s1->pan_scan.height=16*h;
2189
2190 if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2191 av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
2192 }
2193
2194 static void mpeg_decode_picture_display_extension(Mpeg1Context *s1)
2195 {
2196 MpegEncContext *s= &s1->mpeg_enc_ctx;
2197 int i,nofco;
2198
2199 nofco = 1;
2200 if(s->progressive_sequence){
2201 if(s->repeat_first_field){
2202 nofco++;
2203 if(s->top_field_first)
2204 nofco++;
2205 }
2206 }else{
2207 if(s->picture_structure == PICT_FRAME){
2208 nofco++;
2209 if(s->repeat_first_field)
2210 nofco++;
2211 }
2212 }
2213 for(i=0; i<nofco; i++){
2214 s1->pan_scan.position[i][0]= get_sbits(&s->gb, 16);
2215 skip_bits(&s->gb, 1); //marker
2216 s1->pan_scan.position[i][1]= get_sbits(&s->gb, 16);
2217 skip_bits(&s->gb, 1); //marker
2218 }
2219
2220 if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2221 av_log(s->avctx, AV_LOG_DEBUG, "pde (%d,%d) (%d,%d) (%d,%d)\n",
2222 s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
2223 s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
2224 s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]
2225 );
2226 }
2227
2228 static void mpeg_decode_quant_matrix_extension(MpegEncContext *s)
2229 {
2230 int i, v, j;
2231
2232 dprintf("matrix extension\n");
2233
2234 if (get_bits1(&s->gb)) {
2235 for(i=0;i<64;i++) {
2236 v = get_bits(&s->gb, 8);
2237 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2238 s->intra_matrix[j] = v;
2239 s->chroma_intra_matrix[j] = v;
2240 }
2241 }
2242 if (get_bits1(&s->gb)) {
2243 for(i=0;i<64;i++) {
2244 v = get_bits(&s->gb, 8);
2245 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2246 s->inter_matrix[j] = v;
2247 s->chroma_inter_matrix[j] = v;
2248 }
2249 }
2250 if (get_bits1(&s->gb)) {
2251 for(i=0;i<64;i++) {
2252 v = get_bits(&s->gb, 8);
2253 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2254 s->chroma_intra_matrix[j] = v;
2255 }
2256 }
2257 if (get_bits1(&s->gb)) {
2258 for(i=0;i<64;i++) {
2259 v = get_bits(&s->gb, 8);
2260 j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2261 s->chroma_inter_matrix[j] = v;
2262 }
2263 }
2264 }
2265
2266 static void mpeg_decode_picture_coding_extension(MpegEncContext *s)
2267 {
2268 s->full_pel[0] = s->full_pel[1] = 0;
2269 s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
2270 s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
2271 s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
2272 s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
2273 s->intra_dc_precision = get_bits(&s->gb, 2);
2274 s->picture_structure = get_bits(&s->gb, 2);
2275 s->top_field_first = get_bits1(&s->gb);
2276 s->frame_pred_frame_dct = get_bits1(&s->gb);
2277 s->concealment_motion_vectors = get_bits1(&s->gb);
2278 s->q_scale_type = get_bits1(&s->gb);
2279 s->intra_vlc_format = get_bits1(&s->gb);
2280 s->alternate_scan = get_bits1(&s->gb);
2281 s->repeat_first_field = get_bits1(&s->gb);
2282 s->chroma_420_type = get_bits1(&s->gb);
2283 s->progressive_frame = get_bits1(&s->gb);
2284
2285 if(s->picture_structure == PICT_FRAME)
2286 s->first_field=0;
2287 else{
2288 s->first_field ^= 1;
2289 memset(s->mbskip_table, 0, s->mb_stride*s->mb_height);
2290 }
2291
2292 if(s->alternate_scan){
2293 ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable , ff_alternate_vertical_scan);
2294 ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable , ff_alternate_vertical_scan);
2295 }else{
2296 ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable , ff_zigzag_direct);
2297 ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable , ff_zigzag_direct);
2298 }
2299
2300 /* composite display not parsed */
2301 dprintf("intra_dc_precision=%d\n", s->intra_dc_precision);
2302 dprintf("picture_structure=%d\n", s->picture_structure);
2303 dprintf("top field first=%d\n", s->top_field_first);
2304 dprintf("repeat first field=%d\n", s->repeat_first_field);
2305 dprintf("conceal=%d\n", s->concealment_motion_vectors);
2306 dprintf("intra_vlc_format=%d\n", s->intra_vlc_format);
2307 dprintf("alternate_scan=%d\n", s->alternate_scan);
2308 dprintf("frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
2309 dprintf("progressive_frame=%d\n", s->progressive_frame);
2310 }
2311
2312 static void mpeg_decode_extension(AVCodecContext *avctx,
2313 const uint8_t *buf, int buf_size)
2314 {
2315 Mpeg1Context *s1 = avctx->priv_data;
2316 MpegEncContext *s = &s1->mpeg_enc_ctx;
2317 int ext_type;
2318
2319 init_get_bits(&s->gb, buf, buf_size*8);
2320
2321 ext_type = get_bits(&s->gb, 4);
2322 switch(ext_type) {
2323 case 0x1:
2324 mpeg_decode_sequence_extension(s1);
2325 break;
2326 case 0x2:
2327 mpeg_decode_sequence_display_extension(s1);
2328 break;
2329 case 0x3:
2330 mpeg_decode_quant_matrix_extension(s);
2331 break;
2332 case 0x7:
2333 mpeg_decode_picture_display_extension(s1);
2334 break;
2335 case 0x8:
2336 mpeg_decode_picture_coding_extension(s);
2337 break;
2338 }
2339 }
2340
2341 static void exchange_uv(MpegEncContext *s){
2342 short * tmp = s->pblocks[4];
2343 s->pblocks[4] = s->pblocks[5];
2344 s->pblocks[5] = tmp;
2345 }
2346
2347 static int mpeg_field_start(MpegEncContext *s){
2348 AVCodecContext *avctx= s->avctx;
2349 Mpeg1Context *s1 = (Mpeg1Context*)s;
2350
2351 /* start frame decoding */
2352 if(s->first_field || s->picture_structure==PICT_FRAME){
2353 if(MPV_frame_start(s, avctx) < 0)
2354 return -1;
2355
2356 ff_er_frame_start(s);
2357
2358 /* first check if we must repeat the frame */
2359 s->current_picture_ptr->repeat_pict = 0;
2360 if (s->repeat_first_field) {
2361 if (s->progressive_sequence) {
2362 if (s->top_field_first)
2363 s->current_picture_ptr->repeat_pict = 4;
2364 else
2365 s->current_picture_ptr->repeat_pict = 2;
2366 } else if (s->progressive_frame) {
2367 s->current_picture_ptr->repeat_pict = 1;
2368 }
2369 }
2370
2371 *s->current_picture_ptr->pan_scan= s1->pan_scan;
2372 }else{ //second field
2373 int i;
2374
2375 if(!s->current_picture_ptr){
2376 av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
2377 return -1;
2378 }
2379
2380 for(i=0; i<4; i++){
2381 s->current_picture.data[i] = s->current_picture_ptr->data[i];
2382 if(s->picture_structure == PICT_BOTTOM_FIELD){
2383 s->current_picture.data[i] += s->current_picture_ptr->linesize[i];
2384 }
2385 }
2386 }
2387 #ifdef HAVE_XVMC
2388 // MPV_frame_start will call this function too,
2389 // but we need to call it on every field
2390 if(s->avctx->xvmc_acceleration)
2391 XVMC_field_start(s,avctx);
2392 #endif
2393
2394 return 0;
2395 }
2396
2397 #define DECODE_SLICE_ERROR -1
2398 #define DECODE_SLICE_OK 0
2399
2400 /**
2401 * decodes a slice. MpegEncContext.mb_y must be set to the MB row from the startcode
2402 * @return DECODE_SLICE_ERROR if the slice is damaged<br>
2403 * DECODE_SLICE_OK if this slice is ok<br>
2404 */
2405 static int mpeg_decode_slice(Mpeg1Context *s1, int mb_y,
2406 const uint8_t **buf, int buf_size)
2407 {
2408 MpegEncContext *s = &s1->mpeg_enc_ctx;
2409 AVCodecContext *avctx= s->avctx;
2410 int ret;
2411 const int field_pic= s->picture_structure != PICT_FRAME;
2412
2413 s->resync_mb_x=
2414 s->resync_mb_y= -1;
2415
2416 if (mb_y<<field_pic >= s->mb_height){
2417 av_log(s->avctx, AV_LOG_ERROR, "slice below image (%d >= %d)\n", mb_y, s->mb_height);
2418 return -1;
2419 }
2420
2421 init_get_bits(&s->gb, *buf, buf_size*8);
2422
2423 ff_mpeg1_clean_buffers(s);
2424 s->interlaced_dct = 0;
2425
2426 s->qscale = get_qscale(s);
2427
2428 if(s->qscale == 0){
2429 av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
2430 return -1;
2431 }
2432
2433 /* extra slice info */
2434 while (get_bits1(&s->gb) != 0) {
2435 skip_bits(&s->gb, 8);
2436 }
2437
2438 s->mb_x=0;
2439
2440 for(;;) {
2441 int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
2442 if (code < 0){
2443 av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
2444 return -1;
2445 }
2446 if (code >= 33) {
2447 if (code == 33) {
2448 s->mb_x += 33;
2449 }
2450 /* otherwise, stuffing, nothing to do */
2451 } else {
2452 s->mb_x += code;
2453 break;
2454 }
2455 }
2456
2457 s->resync_mb_x= s->mb_x;
2458 s->resync_mb_y= s->mb_y= mb_y;
2459 s->mb_skip_run= 0;
2460 ff_init_block_index(s);
2461
2462 if (s->mb_y==0 && s->mb_x==0 && (s->first_field || s->picture_structure==PICT_FRAME)) {
2463 if(s->avctx->debug&FF_DEBUG_PICT_INFO){
2464 av_log(s->avctx, AV_LOG_DEBUG, "qp:%d fc:%2d%2d%2d%2d %s %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
2465 s->qscale, s->mpeg_f_code[0][0],s->mpeg_f_code[0][1],s->mpeg_f_code[1][0],s->mpeg_f_code[1][1],
2466 s->pict_type == I_TYPE ? "I" : (s->pict_type == P_TYPE ? "P" : (s->pict_type == B_TYPE ? "B" : "S")),
2467 s->progressive_sequence ? "ps" :"", s->progressive_frame ? "pf" : "", s->alternate_scan ? "alt" :"", s->top_field_first ? "top" :"",
2468 s->intra_dc_precision, s->picture_structure, s->frame_pred_frame_dct, s->concealment_motion_vectors,
2469 s->q_scale_type, s->intra_vlc_format, s->repeat_first_field, s->chroma_420_type ? "420" :"");
2470 }
2471 }
2472
2473 for(;;) {
2474 #ifdef HAVE_XVMC
2475 //one 1 we memcpy blocks in xvmcvideo
2476 if(s->avctx->xvmc_acceleration > 1)
2477 XVMC_init_block(s);//set s->block
2478 #endif
2479
2480 s->dsp.clear_blocks(s->block[0]);
2481 if(!s->chroma_y_shift){
2482 s->dsp.clear_blocks(s->block[6]);
2483 }
2484 ret = mpeg_decode_mb(s, s->block);
2485 s->chroma_qscale= s->qscale;
2486
2487 dprintf("ret=%d\n", ret);
2488 if (ret < 0)
2489 return -1;
2490
2491 if(s->current_picture.motion_val[0] && !s->encoding){ //note motion_val is normally NULL unless we want to extract the MVs
2492 const int wrap = field_pic ? 2*s->b8_stride : s->b8_stride;
2493 int xy = s->mb_x*2 + s->mb_y*2*wrap;
2494 int motion_x, motion_y, dir, i;
2495 if(field_pic && !s->first_field)
2496 xy += wrap/2;
2497
2498 for(i=0; i<2; i++){
2499 for(dir=0; dir<2; dir++){
2500 if (s->mb_intra || (dir==1 && s->pict_type != B_TYPE)) {
2501 motion_x = motion_y = 0;
2502 }else if (s->mv_type == MV_TYPE_16X16 || (s->mv_type == MV_TYPE_FIELD && field_pic)){
2503 motion_x = s->mv[dir][0][0];
2504 motion_y = s->mv[dir][0][1];
2505 } else /*if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8))*/ {
2506 motion_x = s->mv[dir][i][0];
2507 motion_y = s->mv[dir][i][1];
2508 }
2509
2510 s->current_picture.motion_val[dir][xy ][0] = motion_x;
2511 s->current_picture.motion_val[dir][xy ][1] = motion_y;
2512 s->current_picture.motion_val[dir][xy + 1][0] = motion_x;
2513 s->current_picture.motion_val[dir][xy + 1][1] = motion_y;
2514 s->current_picture.ref_index [dir][xy ]=
2515 s->current_picture.ref_index [dir][xy + 1]= s->field_select[dir][i];
2516 }
2517 xy += wrap;
2518 }
2519 }
2520
2521 s->dest[0] += 16;
2522 s->dest[1] += 16 >> s->chroma_x_shift;
2523 s->dest[2] += 16 >> s->chroma_x_shift;
2524
2525 MPV_decode_mb(s, s->block);
2526
2527 if (++s->mb_x >= s->mb_width) {
2528
2529 ff_draw_horiz_band(s, 16*s->mb_y, 16);
2530
2531 s->mb_x = 0;
2532 s->mb_y++;
2533
2534 if(s->mb_y<<field_pic >= s->mb_height){
2535 int left= s->gb.size_in_bits - get_bits_count(&s->gb);
2536
2537 if(left < 0 || (left && show_bits(&s->gb, FFMIN(left, 23)))
2538 || (avctx->error_resilience >= FF_ER_AGGRESSIVE && left>8)){
2539 av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d\n", left);
2540 return -1;
2541 }else
2542 goto eos;
2543 }
2544
2545 ff_init_block_index(s);
2546 }
2547
2548 /* skip mb handling */
2549 if (s->mb_skip_run == -1) {
2550 /* read again increment */
2551 s->mb_skip_run = 0;
2552 for(;;) {
2553 int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
2554 if (code < 0){
2555 av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
2556 return -1;
2557 }
2558 if (code >= 33) {
2559 if (code == 33) {
2560 s->mb_skip_run += 33;
2561 }else if(code == 35){
2562 if(s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0){
2563 av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
2564 return -1;
2565 }
2566 goto eos; /* end of slice */
2567 }
2568 /* otherwise, stuffing, nothing to do */
2569 } else {
2570 s->mb_skip_run += code;
2571 break;
2572 }
2573 }
2574 }
2575 }
2576 eos: // end of slice
2577 *buf += get_bits_count(&s->gb)/8 - 1;
2578 //printf("y %d %d %d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
2579 return 0;
2580 }
2581
2582 static int slice_decode_thread(AVCodecContext *c, void *arg){
2583 MpegEncContext *s= arg;
2584 const uint8_t *buf= s->gb.buffer;
2585 int mb_y= s->start_mb_y;
2586
2587 s->error_count= 3*(s->end_mb_y - s->start_mb_y)*s->mb_width;
2588
2589 for(;;){
2590 int start_code, ret;
2591
2592 ret= mpeg_decode_slice((Mpeg1Context*)s, mb_y, &buf, s->gb.buffer_end - buf);
2593 emms_c();
2594 //av_log(c, AV_LOG_DEBUG, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
2595 //ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, s->start_mb_y, s->end_mb_y, s->error_count);
2596 if(ret < 0){
2597 if(s->resync_mb_x>=0 && s->resync_mb_y>=0)
2598 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y, AC_ERROR|DC_ERROR|MV_ERROR);
2599 }else{
2600 ff_er_add_slice(s, s->resync_mb_x, s->resync_mb_y, s->mb_x-1, s->mb_y, AC_END|DC_END|MV_END);
2601 }
2602
2603 if(s->mb_y == s->end_mb_y)
2604 return 0;
2605
2606 start_code = find_start_code(&buf, s->gb.buffer_end);
2607 mb_y= start_code - SLICE_MIN_START_CODE;
2608 if(mb_y < 0 || mb_y >= s->end_mb_y)
2609 return -1;
2610 }
2611
2612 return 0; //not reached
2613 }
2614
2615 /**
2616 * handles slice ends.
2617 * @return 1 if it seems to be the last slice of
2618 */
2619 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
2620 {
2621 Mpeg1Context *s1 = avctx->priv_data;
2622 MpegEncContext *s = &s1->mpeg_enc_ctx;
2623
2624 if (!s1->mpeg_enc_ctx_allocated || !s->current_picture_ptr)
2625 return 0;
2626
2627 #ifdef HAVE_XVMC
2628 if(s->avctx->xvmc_acceleration)
2629 XVMC_field_end(s);
2630 #endif
2631 /* end of slice reached */
2632 if (/*s->mb_y<<field_pic == s->mb_height &&*/ !s->first_field) {
2633 /* end of image */
2634
2635 s->current_picture_ptr->qscale_type= FF_QSCALE_TYPE_MPEG2;
2636
2637 ff_er_frame_end(s);
2638
2639 MPV_frame_end(s);
2640
2641 if (s->pict_type == B_TYPE || s->low_delay) {
2642 *pict= *(AVFrame*)s->current_picture_ptr;
2643 ff_print_debug_info(s, pict);
2644 } else {
2645 s->picture_number++;
2646 /* latency of 1 frame for I and P frames */
2647 /* XXX: use another variable than picture_number */
2648 if (s->last_picture_ptr != NULL) {
2649 *pict= *(AVFrame*)s->last_picture_ptr;
2650 ff_print_debug_info(s, pict);
2651 }
2652 }
2653
2654 return 1;
2655 } else {
2656 return 0;
2657 }
2658 }
2659
2660 static int mpeg1_decode_sequence(AVCodecContext *avctx,
2661 const uint8_t *buf, int buf_size)
2662 {
2663 Mpeg1Context *s1 = avctx->priv_data;
2664 MpegEncContext *s = &s1->mpeg_enc_ctx;
2665 int width,height;
2666 int i, v, j;
2667
2668 init_get_bits(&s->gb, buf, buf_size*8);
2669
2670 width = get_bits(&s->gb, 12);
2671 height = get_bits(&s->gb, 12);
2672 if (width <= 0 || height <= 0 ||
2673 (width % 2) != 0 || (height % 2) != 0)
2674 return -1;
2675 s->aspect_ratio_info= get_bits(&s->gb, 4);
2676 if (s->aspect_ratio_info == 0)
2677 return -1;
2678 s->frame_rate_index = get_bits(&s->gb, 4);
2679 if (s->frame_rate_index == 0 || s->frame_rate_index > 13)
2680 return -1;
2681 s->bit_rate = get_bits(&s->gb, 18) * 400;
2682 if (get_bits1(&s->gb) == 0) /* marker */
2683 return -1;
2684 s->width = width;
2685 s->height = height;
2686
2687 s->avctx->rc_buffer_size= get_bits(&s->gb, 10) * 1024*16;
2688 skip_bits(&s->gb, 1);
2689
2690 /* get matrix */
2691 if (get_bits1(&s->gb)) {
2692 for(i=0;i<64;i++) {
2693 v = get_bits(&s->gb, 8);
2694 if(v==0){
2695 av_log(s->avctx, AV_LOG_ERROR, "intra matrix damaged\n");
2696 return -1;
2697 }
2698 j = s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2699 s->intra_matrix[j] = v;
2700 s->chroma_intra_matrix[j] = v;
2701 }
2702 #ifdef DEBUG
2703 dprintf("intra matrix present\n");
2704 for(i=0;i<64;i++)
2705 dprintf(" %d", s->intra_matrix[s->dsp.idct_permutation[i]);
2706 printf("\n");
2707 #endif
2708 } else {
2709 for(i=0;i<64;i++) {
2710 j = s->dsp.idct_permutation[i];
2711 v = ff_mpeg1_default_intra_matrix[i];
2712 s->intra_matrix[j] = v;
2713 s->chroma_intra_matrix[j] = v;
2714 }
2715 }
2716 if (get_bits1(&s->gb)) {
2717 for(i=0;i<64;i++) {
2718 v = get_bits(&s->gb, 8);
2719 if(v==0){
2720 av_log(s->avctx, AV_LOG_ERROR, "inter matrix damaged\n");
2721 return -1;
2722 }
2723 j = s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
2724 s->inter_matrix[j] = v;
2725 s->chroma_inter_matrix[j] = v;
2726 }
2727 #ifdef DEBUG
2728 dprintf("non intra matrix present\n");
2729 for(i=0;i<64;i++)
2730 dprintf(" %d", s->inter_matrix[s->dsp.idct_permutation[i]);
2731 printf("\n");
2732 #endif
2733 } else {
2734 for(i=0;i<64;i++) {
2735 int j= s->dsp.idct_permutation[i];
2736 v = ff_mpeg1_default_non_intra_matrix[i];
2737 s->inter_matrix[j] = v;
2738 s->chroma_inter_matrix[j] = v;
2739 }
2740 }
2741
2742 if(show_bits(&s->gb, 23) != 0){
2743 av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
2744 return -1;
2745 }
2746
2747 /* we set mpeg2 parameters so that it emulates mpeg1 */
2748 s->progressive_sequence = 1;
2749 s->progressive_frame = 1;
2750 s->picture_structure = PICT_FRAME;
2751 s->frame_pred_frame_dct = 1;
2752 s->chroma_format = 1;
2753 s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG1VIDEO;
2754 avctx->sub_id = 1; /* indicates mpeg1 */
2755 s->out_format = FMT_MPEG1;
2756 s->swap_uv = 0;//AFAIK VCR2 don't have SEQ_HEADER
2757 if(s->flags & CODEC_FLAG_LOW_DELAY) s->low_delay=1;
2758
2759 if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2760 av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%d\n",
2761 s->avctx->rc_buffer_size, s->bit_rate);
2762
2763 return 0;
2764 }
2765
2766 static int vcr2_init_sequence(AVCodecContext *avctx)
2767 {
2768 Mpeg1Context *s1 = avctx->priv_data;
2769 MpegEncContext *s = &s1->mpeg_enc_ctx;
2770 int i, v;
2771
2772 /* start new mpeg1 context decoding */
2773 s->out_format = FMT_MPEG1;
2774 if (s1->mpeg_enc_ctx_allocated) {
2775 MPV_common_end(s);
2776 }
2777 s->width = avctx->width;
2778 s->height = avctx->height;
2779 avctx->has_b_frames= 0; //true?
2780 s->low_delay= 1;
2781
2782 if(avctx->xvmc_acceleration){
2783 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_xvmc_mpg2_420);
2784 }else{
2785 avctx->pix_fmt = avctx->get_format(avctx,pixfmt_yuv_420);
2786 }
2787
2788 if( avctx->pix_fmt == PIX_FMT_XVMC_MPEG2_IDCT )
2789 if( avctx->idct_algo == FF_IDCT_AUTO )
2790 avctx->idct_algo = FF_IDCT_SIMPLE;
2791
2792 if (MPV_common_init(s) < 0)
2793 return -1;
2794 exchange_uv(s);//common init reset pblocks, so we swap them here
2795 s->swap_uv = 1;// in case of xvmc we need to swap uv for each MB
2796 s1->mpeg_enc_ctx_allocated = 1;
2797
2798 for(i=0;i<64;i++) {
2799 int j= s->dsp.idct_permutation[i];
2800 v = ff_mpeg1_default_intra_matrix[i];
2801 s->intra_matrix[j] = v;
2802 s->chroma_intra_matrix[j] = v;
2803
2804 v = ff_mpeg1_default_non_intra_matrix[i];
2805 s->inter_matrix[j] = v;
2806 s->chroma_inter_matrix[j] = v;
2807 }
2808
2809 s->progressive_sequence = 1;
2810 s->progressive_frame = 1;
2811 s->picture_structure = PICT_FRAME;
2812 s->frame_pred_frame_dct = 1;
2813 s->chroma_format = 1;
2814 s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG2VIDEO;
2815 avctx->sub_id = 2; /* indicates mpeg2 */
2816 return 0;
2817 }
2818
2819
2820 static void mpeg_decode_user_data(AVCodecContext *avctx,
2821 const uint8_t *buf, int buf_size)
2822 {
2823 const uint8_t *p;
2824 int len, flags;
2825 p = buf;
2826 len = buf_size;
2827
2828 /* we parse the DTG active format information */
2829 if (len >= 5 &&
2830 p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2831 flags = p[4];
2832 p += 5;
2833 len -= 5;
2834 if (flags & 0x80) {
2835 /* skip event id */
2836 if (len < 2)
2837 return;
2838 p += 2;
2839 len -= 2;
2840 }
2841 if (flags & 0x40) {
2842 if (len < 1)
2843 return;
2844 avctx->dtg_active_format = p[0] & 0x0f;
2845 }
2846 }
2847 }
2848
2849 static void mpeg_decode_gop(AVCodecContext *avctx,
2850 const uint8_t *buf, int buf_size){
2851 Mpeg1Context *s1 = avctx->priv_data;
2852 MpegEncContext *s = &s1->mpeg_enc_ctx;
2853
2854 int drop_frame_flag;
2855 int time_code_hours, time_code_minutes;
2856 int time_code_seconds, time_code_pictures;
2857 int broken_link;
2858
2859 init_get_bits(&s->gb, buf, buf_size*8);
2860
2861 drop_frame_flag = get_bits1(&s->gb);
2862
2863 time_code_hours=get_bits(&s->gb,5);
2864 time_code_minutes = get_bits(&s->gb,6);
2865 skip_bits1(&s->gb);//marker bit
2866 time_code_seconds = get_bits(&s->gb,6);
2867 time_code_pictures = get_bits(&s->gb,6);
2868
2869 /*broken_link indicate that after editing the
2870 reference frames of the first B-Frames after GOP I-Frame
2871 are missing (open gop)*/
2872 broken_link = get_bits1(&s->gb);
2873
2874 if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2875 av_log(s->avctx, AV_LOG_DEBUG, "GOP (%2d:%02d:%02d.[%02d]) broken_link=%d\n",
2876 time_code_hours, time_code_minutes, time_code_seconds,
2877 time_code_pictures, broken_link);
2878 }
2879 /**
2880 * finds the end of the current frame in the bitstream.
2881 * @return the position of the first byte of the next frame, or -1
2882 */
2883 int ff_mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size)
2884 {
2885 int i;
2886 uint32_t state;
2887
2888 state= pc->state;
2889