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