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