git.sesse.net Git - ffmpeg/blob - 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, !!(s->flags & CODEC_FLAG_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     s->mpeg_enc_ctx.flags2= avctx->flags2;
1710     common_init(&s->mpeg_enc_ctx);
1711     init_vlcs();
1712
1713     s->mpeg_enc_ctx_allocated = 0;
1714     s->mpeg_enc_ctx.picture_number = 0;
1715     s->repeat_field = 0;
1716     s->mpeg_enc_ctx.codec_id= avctx->codec->id;
1717     return 0;
1718 }
1719
1720 /* return the 8 bit start code value and update the search
1721    state. Return -1 if no start code found */
1722 static int find_start_code(uint8_t **pbuf_ptr, uint8_t *buf_end)
1723 {
1724     uint8_t *buf_ptr;
1725     unsigned int state=0xFFFFFFFF, v;
1726     int val;
1727
1728     buf_ptr = *pbuf_ptr;
1729     while (buf_ptr < buf_end) {
1730         v = *buf_ptr++;
1731         if (state == 0x000001) {
1732             state = ((state << 8) | v) & 0xffffff;
1733             val = state;
1734             goto found;
1735         }
1736         state = ((state << 8) | v) & 0xffffff;
1737     }
1738     val = -1;
1739  found:
1740     *pbuf_ptr = buf_ptr;
1741     return val;
1742 }
1743
1744 static int mpeg1_decode_picture(AVCodecContext *avctx,
1745                                 uint8_t *buf, int buf_size)
1746 {
1747     Mpeg1Context *s1 = avctx->priv_data;
1748     MpegEncContext *s = &s1->mpeg_enc_ctx;
1749     int ref, f_code, vbv_delay;
1750
1751     init_get_bits(&s->gb, buf, buf_size*8);
1752
1753     ref = get_bits(&s->gb, 10); /* temporal ref */
1754     s->pict_type = get_bits(&s->gb, 3);
1755
1756     vbv_delay= get_bits(&s->gb, 16);
1757     if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) {
1758         s->full_pel[0] = get_bits1(&s->gb);
1759         f_code = get_bits(&s->gb, 3);
1760         if (f_code == 0)
1761             return -1;
1762         s->mpeg_f_code[0][0] = f_code;
1763         s->mpeg_f_code[0][1] = f_code;
1764     }
1765     if (s->pict_type == B_TYPE) {
1766         s->full_pel[1] = get_bits1(&s->gb);
1767         f_code = get_bits(&s->gb, 3);
1768         if (f_code == 0)
1769             return -1;
1770         s->mpeg_f_code[1][0] = f_code;
1771         s->mpeg_f_code[1][1] = f_code;
1772     }
1773     s->current_picture.pict_type= s->pict_type;
1774     s->current_picture.key_frame= s->pict_type == I_TYPE;
1775
1776 //    if(avctx->debug & FF_DEBUG_PICT_INFO)
1777 //        av_log(avctx, AV_LOG_DEBUG, "vbv_delay %d, ref %d\n", vbv_delay, ref);
1778
1779     s->y_dc_scale = 8;
1780     s->c_dc_scale = 8;
1781     s->first_slice = 1;
1782     return 0;
1783 }
1784
1785 static void mpeg_decode_sequence_extension(MpegEncContext *s)
1786 {
1787     int horiz_size_ext, vert_size_ext;
1788     int bit_rate_ext;
1789     int frame_rate_ext_n, frame_rate_ext_d;
1790     int level, profile;
1791
1792     skip_bits(&s->gb, 1); /* profil and level esc*/
1793     profile= get_bits(&s->gb, 3);
1794     level= get_bits(&s->gb, 4);
1795     s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
1796     skip_bits(&s->gb, 2); /* chroma_format */
1797     horiz_size_ext = get_bits(&s->gb, 2);
1798     vert_size_ext = get_bits(&s->gb, 2);
1799     s->width |= (horiz_size_ext << 12);
1800     s->height |= (vert_size_ext << 12);
1801     bit_rate_ext = get_bits(&s->gb, 12);  /* XXX: handle it */
1802     s->bit_rate = ((s->bit_rate / 400) | (bit_rate_ext << 12)) * 400;
1803     skip_bits1(&s->gb); /* marker */
1804     s->avctx->rc_buffer_size += get_bits(&s->gb, 8)*1024*16<<10;
1805
1806     s->low_delay = get_bits1(&s->gb);
1807     if(s->flags & CODEC_FLAG_LOW_DELAY) s->low_delay=1;
1808
1809     frame_rate_ext_n = get_bits(&s->gb, 2);
1810     frame_rate_ext_d = get_bits(&s->gb, 5);
1811     av_reduce(
1812         &s->avctx->frame_rate,
1813         &s->avctx->frame_rate_base,
1814         frame_rate_tab[s->frame_rate_index] * (frame_rate_ext_n+1),
1815         MPEG1_FRAME_RATE_BASE * (frame_rate_ext_d+1),
1816         1<<30);
1817
1818     dprintf("sequence extension\n");
1819     s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG2VIDEO;
1820     s->avctx->sub_id = 2; /* indicates mpeg2 found */
1821
1822     if(s->aspect_ratio_info <= 1)
1823         s->avctx->sample_aspect_ratio= mpeg2_aspect[s->aspect_ratio_info];
1824     else{
1825         s->avctx->sample_aspect_ratio=
1826             av_div_q(
1827                 mpeg2_aspect[s->aspect_ratio_info],
1828                 (AVRational){s->width, s->height}
1829             );
1830     }
1831
1832     if(s->avctx->debug & FF_DEBUG_PICT_INFO)
1833         av_log(s->avctx, AV_LOG_DEBUG, "profile: %d, level: %d vbv buffer: %d, bitrate:%d\n",
1834                profile, level, s->avctx->rc_buffer_size, s->bit_rate);
1835 }
1836
1837 static void mpeg_decode_sequence_display_extension(Mpeg1Context *s1)
1838 {
1839     MpegEncContext *s= &s1->mpeg_enc_ctx;
1840     int color_description, w, h;
1841
1842     skip_bits(&s->gb, 3); /* video format */
1843     color_description= get_bits1(&s->gb);
1844     if(color_description){
1845         skip_bits(&s->gb, 8); /* color primaries */
1846         skip_bits(&s->gb, 8); /* transfer_characteristics */
1847         skip_bits(&s->gb, 8); /* matrix_coefficients */
1848     }
1849     w= get_bits(&s->gb, 14);
1850     skip_bits(&s->gb, 1); //marker
1851     h= get_bits(&s->gb, 14);
1852     skip_bits(&s->gb, 1); //marker
1853
1854     s1->pan_scan.width= 16*w;
1855     s1->pan_scan.height=16*h;
1856
1857     if(s->aspect_ratio_info > 1)
1858         s->avctx->sample_aspect_ratio=
1859             av_div_q(
1860                 mpeg2_aspect[s->aspect_ratio_info],
1861                 (AVRational){w, h}
1862             );
1863
1864     if(s->avctx->debug & FF_DEBUG_PICT_INFO)
1865         av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
1866 }
1867
1868 static void mpeg_decode_picture_display_extension(Mpeg1Context *s1)
1869 {
1870     MpegEncContext *s= &s1->mpeg_enc_ctx;
1871     int i;
1872
1873     for(i=0; i<1; i++){ //FIXME count
1874         s1->pan_scan.position[i][0]= get_sbits(&s->gb, 16);
1875         skip_bits(&s->gb, 1); //marker
1876         s1->pan_scan.position[i][1]= get_sbits(&s->gb, 16);
1877         skip_bits(&s->gb, 1); //marker
1878     }
1879
1880     if(s->avctx->debug & FF_DEBUG_PICT_INFO)
1881         av_log(s->avctx, AV_LOG_DEBUG, "pde (%d,%d) (%d,%d) (%d,%d)\n",
1882             s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
1883             s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
1884             s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]
1885         );
1886 }
1887
1888 static void mpeg_decode_quant_matrix_extension(MpegEncContext *s)
1889 {
1890     int i, v, j;
1891
1892     dprintf("matrix extension\n");
1893
1894     if (get_bits1(&s->gb)) {
1895         for(i=0;i<64;i++) {
1896             v = get_bits(&s->gb, 8);
1897             j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
1898             s->intra_matrix[j] = v;
1899             s->chroma_intra_matrix[j] = v;
1900         }
1901     }
1902     if (get_bits1(&s->gb)) {
1903         for(i=0;i<64;i++) {
1904             v = get_bits(&s->gb, 8);
1905             j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
1906             s->inter_matrix[j] = v;
1907             s->chroma_inter_matrix[j] = v;
1908         }
1909     }
1910     if (get_bits1(&s->gb)) {
1911         for(i=0;i<64;i++) {
1912             v = get_bits(&s->gb, 8);
1913             j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
1914             s->chroma_intra_matrix[j] = v;
1915         }
1916     }
1917     if (get_bits1(&s->gb)) {
1918         for(i=0;i<64;i++) {
1919             v = get_bits(&s->gb, 8);
1920             j= s->dsp.idct_permutation[ ff_zigzag_direct[i] ];
1921             s->chroma_inter_matrix[j] = v;
1922         }
1923     }
1924 }
1925
1926 static void mpeg_decode_picture_coding_extension(MpegEncContext *s)
1927 {
1928     s->full_pel[0] = s->full_pel[1] = 0;
1929     s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
1930     s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
1931     s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
1932     s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
1933     s->intra_dc_precision = get_bits(&s->gb, 2);
1934     s->picture_structure = get_bits(&s->gb, 2);
1935     s->top_field_first = get_bits1(&s->gb);
1936     s->frame_pred_frame_dct = get_bits1(&s->gb);
1937     s->concealment_motion_vectors = get_bits1(&s->gb);
1938     s->q_scale_type = get_bits1(&s->gb);
1939     s->intra_vlc_format = get_bits1(&s->gb);
1940     s->alternate_scan = get_bits1(&s->gb);
1941     s->repeat_first_field = get_bits1(&s->gb);
1942     s->chroma_420_type = get_bits1(&s->gb);
1943     s->progressive_frame = get_bits1(&s->gb);
1944
1945     if(s->picture_structure == PICT_FRAME)
1946         s->first_field=0;
1947     else{
1948         s->first_field ^= 1;
1949         memset(s->mbskip_table, 0, s->mb_stride*s->mb_height);
1950     }
1951
1952     if(s->alternate_scan){
1953         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_alternate_vertical_scan);
1954         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_alternate_vertical_scan);
1955     }else{
1956         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_zigzag_direct);
1957         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_zigzag_direct);
1958     }
1959
1960     /* composite display not parsed */
1961     dprintf("intra_dc_precision=%d\n", s->intra_dc_precision);
1962     dprintf("picture_structure=%d\n", s->picture_structure);
1963     dprintf("top field first=%d\n", s->top_field_first);
1964     dprintf("repeat first field=%d\n", s->repeat_first_field);
1965     dprintf("conceal=%d\n", s->concealment_motion_vectors);
1966     dprintf("intra_vlc_format=%d\n", s->intra_vlc_format);
1967     dprintf("alternate_scan=%d\n", s->alternate_scan);
1968     dprintf("frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
1969     dprintf("progressive_frame=%d\n", s->progressive_frame);
1970 }
1971
1972 static void mpeg_decode_extension(AVCodecContext *avctx,
1973                                   uint8_t *buf, int buf_size)
1974 {
1975     Mpeg1Context *s1 = avctx->priv_data;
1976     MpegEncContext *s = &s1->mpeg_enc_ctx;
1977     int ext_type;
1978
1979     init_get_bits(&s->gb, buf, buf_size*8);
1980
1981     ext_type = get_bits(&s->gb, 4);
1982     switch(ext_type) {
1983     case 0x1:
1984         mpeg_decode_sequence_extension(s);
1985         break;
1986     case 0x2:
1987         mpeg_decode_sequence_display_extension(s1);
1988         break;
1989     case 0x3:
1990         mpeg_decode_quant_matrix_extension(s);
1991         break;
1992     case 0x7:
1993         mpeg_decode_picture_display_extension(s1);
1994         break;
1995     case 0x8:
1996         mpeg_decode_picture_coding_extension(s);
1997         break;
1998     }
1999 }
2000
2001 static void exchange_uv(MpegEncContext *s){
2002 short * tmp;
2003
2004     tmp = s->pblocks[4];
2005     s->pblocks[4] = s->pblocks[5];
2006     s->pblocks[5] = tmp;
2007 }
2008
2009 #define DECODE_SLICE_FATAL_ERROR -2
2010 #define DECODE_SLICE_ERROR -1
2011 #define DECODE_SLICE_OK 0
2012
2013 /**
2014  * decodes a slice.
2015  * @return DECODE_SLICE_FATAL_ERROR if a non recoverable error occured<br>
2016  *         DECODE_SLICE_ERROR if the slice is damaged<br>
2017  *         DECODE_SLICE_OK if this slice is ok<br>
2018  */
2019 static int mpeg_decode_slice(AVCodecContext *avctx,
2020                               AVFrame *pict,
2021                               int start_code,
2022                               uint8_t **buf, int buf_size)
2023 {
2024     Mpeg1Context *s1 = avctx->priv_data;
2025     MpegEncContext *s = &s1->mpeg_enc_ctx;
2026     int ret;
2027     const int field_pic= s->picture_structure != PICT_FRAME;
2028
2029     s->resync_mb_x= s->mb_x =
2030     s->resync_mb_y= s->mb_y = -1;
2031
2032     start_code = (start_code - 1) & 0xff;
2033     if (start_code >= s->mb_height){
2034         av_log(s->avctx, AV_LOG_ERROR, "slice below image (%d >= %d)\n", start_code, s->mb_height);
2035         return -1;
2036     }
2037
2038     ff_mpeg1_clean_buffers(s);
2039     s->interlaced_dct = 0;
2040
2041     /* start frame decoding */
2042     if (s->first_slice) {
2043       if(s->first_field || s->picture_structure==PICT_FRAME){
2044         if(MPV_frame_start(s, avctx) < 0)
2045             return DECODE_SLICE_FATAL_ERROR;
2046
2047         ff_er_frame_start(s);
2048
2049         /* first check if we must repeat the frame */
2050         s->current_picture_ptr->repeat_pict = 0;
2051         if (s->repeat_first_field) {
2052             if (s->progressive_sequence) {
2053                 if (s->top_field_first)
2054                     s->current_picture_ptr->repeat_pict = 4;
2055                 else
2056                     s->current_picture_ptr->repeat_pict = 2;
2057             } else if (s->progressive_frame) {
2058                 s->current_picture_ptr->repeat_pict = 1;
2059             }
2060         }
2061
2062         *s->current_picture_ptr->pan_scan= s1->pan_scan;
2063       }else{ //second field
2064             int i;
2065
2066             if(!s->current_picture_ptr){
2067                 av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
2068                 return -1;
2069             }
2070
2071             for(i=0; i<4; i++){
2072                 s->current_picture.data[i] = s->current_picture_ptr->data[i];
2073                 if(s->picture_structure == PICT_BOTTOM_FIELD){
2074                     s->current_picture.data[i] += s->current_picture_ptr->linesize[i];
2075                 }
2076             }
2077       }
2078 #ifdef HAVE_XVMC
2079 // MPV_frame_start will call this function too,
2080 // but we need to call it on every field
2081       if(s->avctx->xvmc_acceleration)
2082          XVMC_field_start(s,avctx);
2083 #endif
2084     }//fi(s->first_slice)
2085
2086     init_get_bits(&s->gb, *buf, buf_size*8);
2087
2088     s->qscale = get_qscale(s);
2089     if (s->first_slice && (s->first_field || s->picture_structure==PICT_FRAME)) {
2090         if(s->avctx->debug&FF_DEBUG_PICT_INFO){
2091              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",
2092                  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],
2093                  s->pict_type == I_TYPE ? "I" : (s->pict_type == P_TYPE ? "P" : (s->pict_type == B_TYPE ? "B" : "S")),
2094                  s->progressive_sequence ? "ps" :"", s->progressive_frame ? "pf" : "", s->alternate_scan ? "alt" :"", s->top_field_first ? "top" :"",
2095                  s->intra_dc_precision, s->picture_structure, s->frame_pred_frame_dct, s->concealment_motion_vectors,
2096                  s->q_scale_type, s->intra_vlc_format, s->repeat_first_field, s->chroma_420_type ? "420" :"");
2097         }
2098     }
2099
2100     s->first_slice = 0;
2101     if(s->qscale == 0){
2102         av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
2103         return -1;
2104     }
2105
2106     /* extra slice info */
2107     while (get_bits1(&s->gb) != 0) {
2108         skip_bits(&s->gb, 8);
2109     }
2110
2111     s->mb_x=0;
2112
2113     for(;;) {
2114         int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
2115         if (code < 0){
2116             av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
2117             return -1;
2118         }
2119         if (code >= 33) {
2120             if (code == 33) {
2121                 s->mb_x += 33;
2122             }
2123             /* otherwise, stuffing, nothing to do */
2124         } else {
2125             s->mb_x += code;
2126             break;
2127         }
2128     }
2129
2130     s->resync_mb_x= s->mb_x;
2131     s->resync_mb_y= s->mb_y = start_code;
2132     s->mb_skip_run= 0;
2133     ff_init_block_index(s);
2134
2135     for(;;) {
2136 #ifdef HAVE_XVMC
2137         //one 1 we memcpy blocks in xvmcvideo
2138         if(s->avctx->xvmc_acceleration > 1)
2139             XVMC_init_block(s);//set s->block
2140 #endif
2141
2142         s->dsp.clear_blocks(s->block[0]);
2143
2144         ret = mpeg_decode_mb(s, s->block);
2145         s->chroma_qscale= s->qscale;
2146
2147         dprintf("ret=%d\n", ret);
2148         if (ret < 0)
2149             return -1;
2150
2151         if(s->current_picture.motion_val[0] && !s->encoding){ //note motion_val is normally NULL unless we want to extract the MVs
2152             const int wrap = field_pic ? 2*s->block_wrap[0] : s->block_wrap[0];
2153             int xy = s->mb_x*2 + 1 + (s->mb_y*2 +1)*wrap;
2154             int motion_for_top_x, motion_for_top_y, motion_back_top_x, motion_back_top_y;
2155             int motion_for_bottom_x, motion_for_bottom_y, motion_back_bottom_x, motion_back_bottom_y;
2156             if(field_pic && !s->first_field)
2157                 xy += wrap/2;
2158
2159             if (s->mb_intra) {
2160                 motion_for_top_x = motion_for_top_y = motion_back_top_x = motion_back_top_y =
2161                 motion_for_bottom_x = motion_for_bottom_y = motion_back_bottom_x = motion_back_bottom_y = 0;
2162             }else if (s->mv_type == MV_TYPE_16X16){
2163                 motion_for_top_x = motion_for_bottom_x = s->mv[0][0][0];
2164                 motion_for_top_y = motion_for_bottom_y = s->mv[0][0][1];
2165                 motion_back_top_x = motion_back_bottom_x = s->mv[1][0][0];
2166                 motion_back_top_y = motion_back_bottom_y = s->mv[1][0][1];
2167             } else /*if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8))*/ {
2168                 motion_for_top_x = s->mv[0][0][0];
2169                 motion_for_top_y = s->mv[0][0][1];
2170                 motion_for_bottom_x = s->mv[0][1][0];
2171                 motion_for_bottom_y = s->mv[0][1][1];
2172                 motion_back_top_x = s->mv[1][0][0];
2173                 motion_back_top_y = s->mv[1][0][1];
2174                 motion_back_bottom_x = s->mv[1][1][0];
2175                 motion_back_bottom_y = s->mv[1][1][1];
2176             }
2177
2178             s->current_picture.motion_val[0][xy][0] = motion_for_top_x;
2179             s->current_picture.motion_val[0][xy][1] = motion_for_top_y;
2180             s->current_picture.motion_val[0][xy + 1][0] = motion_for_top_x;
2181             s->current_picture.motion_val[0][xy + 1][1] = motion_for_top_y;
2182             s->current_picture.motion_val[0][xy + wrap][0] = motion_for_bottom_x;
2183             s->current_picture.motion_val[0][xy + wrap][1] = motion_for_bottom_y;
2184             s->current_picture.motion_val[0][xy + 1 + wrap][0] = motion_for_bottom_x;
2185             s->current_picture.motion_val[0][xy + 1 + wrap][1] = motion_for_bottom_y;
2186
2187             if(s->pict_type != B_TYPE){
2188                 motion_back_top_x = motion_back_top_y = motion_back_bottom_x = motion_back_bottom_y = 0;
2189             }
2190
2191             s->current_picture.motion_val[1][xy][0] = motion_back_top_x;
2192             s->current_picture.motion_val[1][xy][1] = motion_back_top_y;
2193             s->current_picture.motion_val[1][xy + 1][0] = motion_back_top_x;
2194             s->current_picture.motion_val[1][xy + 1][1] = motion_back_top_y;
2195             s->current_picture.motion_val[1][xy + wrap][0] = motion_back_bottom_x;
2196             s->current_picture.motion_val[1][xy + wrap][1] = motion_back_bottom_y;
2197             s->current_picture.motion_val[1][xy + 1 + wrap][0] = motion_back_bottom_x;
2198             s->current_picture.motion_val[1][xy + 1 + wrap][1] = motion_back_bottom_y;
2199         }
2200
2201         s->dest[0] += 16;
2202         s->dest[1] += 8;
2203         s->dest[2] += 8;
2204
2205         MPV_decode_mb(s, s->block);
2206
2207         if (++s->mb_x >= s->mb_width) {
2208
2209             ff_draw_horiz_band(s, 16*s->mb_y, 16);
2210
2211             s->mb_x = 0;
2212             s->mb_y++;
2213
2214             if(s->mb_y<<field_pic >= s->mb_height){
2215                 int left= s->gb.size_in_bits - get_bits_count(&s->gb);
2216
2217                 if(left < 0 || (left && show_bits(&s->gb, FFMIN(left, 23)))
2218                    || (avctx->error_resilience >= FF_ER_AGGRESSIVE && left>8)){
2219                     av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d\n", left);
2220                     return -1;
2221                 }else
2222                     goto eos;
2223             }
2224
2225             ff_init_block_index(s);
2226         }
2227
2228         /* skip mb handling */
2229         if (s->mb_skip_run == -1) {
2230             /* read again increment */
2231             s->mb_skip_run = 0;
2232             for(;;) {
2233                 int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
2234                 if (code < 0){
2235                     av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
2236                     return -1;
2237                 }
2238                 if (code >= 33) {
2239                     if (code == 33) {
2240                         s->mb_skip_run += 33;
2241                     }else if(code == 35){
2242                         if(s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0){
2243                             av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
2244                             return -1;
2245                         }
2246                         goto eos; /* end of slice */
2247                     }
2248                     /* otherwise, stuffing, nothing to do */
2249                 } else {
2250                     s->mb_skip_run += code;
2251                     break;
2252                 }
2253             }
2254         }
2255     }
2256 eos: // end of slice
2257     *buf += get_bits_count(&s->gb)/8 - 1;
2258 //printf("y %d %d %d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
2259     return 0;
2260 }
2261
2262 /**
2263  * handles slice ends.
2264  * @return 1 if it seems to be the last slice of
2265  */
2266 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
2267 {
2268     Mpeg1Context *s1 = avctx->priv_data;
2269     MpegEncContext *s = &s1->mpeg_enc_ctx;
2270
2271     if (!s1->mpeg_enc_ctx_allocated || !s->current_picture_ptr)
2272         return 0;
2273
2274 #ifdef HAVE_XVMC
2275     if(s->avctx->xvmc_acceleration)
2276         XVMC_field_end(s);
2277 #endif
2278     /* end of slice reached */
2279     if (/*s->mb_y<<field_pic == s->mb_height &&*/ !s->first_field) {
2280         /* end of image */
2281
2282         s->current_picture_ptr->qscale_type= FF_QSCALE_TYPE_MPEG2;
2283
2284         ff_er_frame_end(s);
2285
2286         MPV_frame_end(s);
2287
2288         if (s->pict_type == B_TYPE || s->low_delay) {
2289             *pict= *(AVFrame*)s->current_picture_ptr;
2290             ff_print_debug_info(s, pict);
2291         } else {
2292             s->picture_number++;
2293             /* latency of 1 frame for I and P frames */
2294             /* XXX: use another variable than picture_number */
2295             if (s->last_picture_ptr != NULL) {
2296                 *pict= *(AVFrame*)s->last_picture_ptr;
2297                  ff_print_debug_info(s, pict);
2298             }
2299         }
2300
2301         return 1;
2302     } else {
2303         return 0;
2304     }
2305 }
2306
2307 static int mpeg1_decode_sequence(AVCodecContext *avctx,
2308                                  uint8_t *buf, int buf_size)
2309 {
2310     Mpeg1Context *s1 = avctx->priv_data;
2311     MpegEncContext *s = &s1->mpeg_enc_ctx;
2312     int width, height, i, v, j;
2313     float aspect;
2314
2315     init_get_bits(&s->gb, buf, buf_size*8);
2316
2317     width = get_bits(&s->gb, 12);
2318     height = get_bits(&s->gb, 12);
2319     s->aspect_ratio_info= get_bits(&s->gb, 4);
2320     if (s->aspect_ratio_info == 0)
2321         return -1;
2322     aspect= 1.0/mpeg1_aspect[s->aspect_ratio_info];
2323     avctx->sample_aspect_ratio= av_d2q(aspect, 255);
2324
2325     s->frame_rate_index = get_bits(&s->gb, 4);
2326     if (s->frame_rate_index == 0)
2327         return -1;
2328     s->bit_rate = get_bits(&s->gb, 18) * 400;
2329     if (get_bits1(&s->gb) == 0) /* marker */
2330         return -1;
2331     if (width <= 0 || height <= 0 ||
2332         (width % 2) != 0 || (height % 2) != 0)
2333         return -1;
2334     if (width != s->width ||
2335         height != s->height) {
2336         /* start new mpeg1 context decoding */
2337         s->out_format = FMT_MPEG1;
2338         if (s1->mpeg_enc_ctx_allocated) {
2339             MPV_common_end(s);
2340         }
2341         s->width = width;
2342         s->height = height;
2343         avctx->has_b_frames= 1;
2344         avctx->width = width;
2345         avctx->height = height;
2346         av_reduce(
2347             &avctx->frame_rate,
2348             &avctx->frame_rate_base,
2349             frame_rate_tab[s->frame_rate_index],
2350             MPEG1_FRAME_RATE_BASE, //FIXME store in allready reduced form
2351             1<<30
2352             );
2353         avctx->bit_rate = s->bit_rate;
2354
2355         //get_format() or set_video(width,height,aspect,pix_fmt);
2356         //until then pix_fmt may be changed right after codec init
2357         if( avctx->pix_fmt == PIX_FMT_XVMC_MPEG2_IDCT )
2358             if( avctx->idct_algo == FF_IDCT_AUTO )
2359                 avctx->idct_algo = FF_IDCT_SIMPLE;
2360
2361         if (MPV_common_init(s) < 0)
2362             return -1;
2363         s1->mpeg_enc_ctx_allocated = 1;
2364         s->swap_uv = 0;//just in case vcr2 and mpeg2 stream have been concatinated
2365     }
2366
2367     s->avctx->rc_buffer_size= get_bits(&s->gb, 10) * 1024*16;
2368     skip_bits(&s->gb, 1);
2369
2370     /* get matrix */
2371     if (get_bits1(&s->gb)) {
2372         for(i=0;i<64;i++) {
2373             v = get_bits(&s->gb, 8);
2374             j = s->intra_scantable.permutated[i];
2375             s->intra_matrix[j] = v;
2376             s->chroma_intra_matrix[j] = v;
2377         }
2378 #ifdef DEBUG
2379         dprintf("intra matrix present\n");
2380         for(i=0;i<64;i++)
2381             dprintf(" %d", s->intra_matrix[s->intra_scantable.permutated[i]]);
2382         printf("\n");
2383 #endif
2384     } else {
2385         for(i=0;i<64;i++) {
2386             int j= s->dsp.idct_permutation[i];
2387             v = ff_mpeg1_default_intra_matrix[i];
2388             s->intra_matrix[j] = v;
2389             s->chroma_intra_matrix[j] = v;
2390         }
2391     }
2392     if (get_bits1(&s->gb)) {
2393         for(i=0;i<64;i++) {
2394             v = get_bits(&s->gb, 8);
2395             j = s->intra_scantable.permutated[i];
2396             s->inter_matrix[j] = v;
2397             s->chroma_inter_matrix[j] = v;
2398         }
2399 #ifdef DEBUG
2400         dprintf("non intra matrix present\n");
2401         for(i=0;i<64;i++)
2402             dprintf(" %d", s->inter_matrix[s->intra_scantable.permutated[i]]);
2403         printf("\n");
2404 #endif
2405     } else {
2406         for(i=0;i<64;i++) {
2407             int j= s->dsp.idct_permutation[i];
2408             v = ff_mpeg1_default_non_intra_matrix[i];
2409             s->inter_matrix[j] = v;
2410             s->chroma_inter_matrix[j] = v;
2411         }
2412     }
2413
2414     /* we set mpeg2 parameters so that it emulates mpeg1 */
2415     s->progressive_sequence = 1;
2416     s->progressive_frame = 1;
2417     s->picture_structure = PICT_FRAME;
2418     s->frame_pred_frame_dct = 1;
2419     s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG1VIDEO;
2420     avctx->sub_id = 1; /* indicates mpeg1 */
2421     if(s->flags & CODEC_FLAG_LOW_DELAY) s->low_delay=1;
2422
2423     if(s->avctx->debug & FF_DEBUG_PICT_INFO)
2424         av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%d\n",
2425                s->avctx->rc_buffer_size, s->bit_rate);
2426
2427     return 0;
2428 }
2429
2430 static int vcr2_init_sequence(AVCodecContext *avctx)
2431 {
2432     Mpeg1Context *s1 = avctx->priv_data;
2433     MpegEncContext *s = &s1->mpeg_enc_ctx;
2434     int i, v;
2435
2436     /* start new mpeg1 context decoding */
2437     s->out_format = FMT_MPEG1;
2438     if (s1->mpeg_enc_ctx_allocated) {
2439         MPV_common_end(s);
2440     }
2441     s->width = avctx->width;
2442     s->height = avctx->height;
2443     avctx->has_b_frames= 0; //true?
2444     s->low_delay= 1;
2445
2446     //get_format() or set_video(width,height,aspect,pix_fmt);
2447     //until then pix_fmt may be changed right after codec init
2448     if( avctx->pix_fmt == PIX_FMT_XVMC_MPEG2_IDCT )
2449         if( avctx->idct_algo == FF_IDCT_AUTO )
2450             avctx->idct_algo = FF_IDCT_SIMPLE;
2451
2452     if (MPV_common_init(s) < 0)
2453         return -1;
2454     exchange_uv(s);//common init reset pblocks, so we swap them here
2455     s->swap_uv = 1;// in case of xvmc we need to swap uv for each MB
2456     s1->mpeg_enc_ctx_allocated = 1;
2457
2458     for(i=0;i<64;i++) {
2459         int j= s->dsp.idct_permutation[i];
2460         v = ff_mpeg1_default_intra_matrix[i];
2461         s->intra_matrix[j] = v;
2462         s->chroma_intra_matrix[j] = v;
2463
2464         v = ff_mpeg1_default_non_intra_matrix[i];
2465         s->inter_matrix[j] = v;
2466         s->chroma_inter_matrix[j] = v;
2467     }
2468
2469     s->progressive_sequence = 1;
2470     s->progressive_frame = 1;
2471     s->picture_structure = PICT_FRAME;
2472     s->frame_pred_frame_dct = 1;
2473     s->codec_id= s->avctx->codec_id= CODEC_ID_MPEG2VIDEO;
2474     avctx->sub_id = 2; /* indicates mpeg2 */
2475     return 0;
2476 }
2477
2478
2479 static void mpeg_decode_user_data(AVCodecContext *avctx,
2480                                   const uint8_t *buf, int buf_size)
2481 {
2482     const uint8_t *p;
2483     int len, flags;
2484     p = buf;
2485     len = buf_size;
2486
2487     /* we parse the DTG active format information */
2488     if (len >= 5 &&
2489         p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2490         flags = p[4];
2491         p += 5;
2492         len -= 5;
2493         if (flags & 0x80) {
2494             /* skip event id */
2495             if (len < 2)
2496                 return;
2497             p += 2;
2498             len -= 2;
2499         }
2500         if (flags & 0x40) {
2501             if (len < 1)
2502                 return;
2503             avctx->dtg_active_format = p[0] & 0x0f;
2504         }
2505     }
2506 }
2507
2508 /**
2509  * finds the end of the current frame in the bitstream.
2510  * @return the position of the first byte of the next frame, or -1
2511  */
2512 static int mpeg1_find_frame_end(MpegEncContext *s, uint8_t *buf, int buf_size){
2513     ParseContext *pc= &s->parse_context;
2514     int i;
2515     uint32_t state;
2516
2517     state= pc->state;
2518
2519     i=0;
2520     if(!pc->frame_start_found){
2521         for(i=0; i<buf_size; i++){
2522             state= (state<<8) | buf[i];
2523             if(state >= SLICE_MIN_START_CODE && state <= SLICE_MAX_START_CODE){
2524                 i++;
2525                 pc->frame_start_found=1;
2526                 break;
2527             }
2528         }
2529     }
2530
2531     if(pc->frame_start_found){
2532         for(; i<buf_size; i++){
2533             state= (state<<8) | buf[i];
2534             if((state&0xFFFFFF00) == 0x100){
2535                 if(state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE){
2536                     pc->frame_start_found=0;
2537                     pc->state=-1;
2538                     return i-3;
2539                 }
2540             }
2541         }
2542     }
2543     pc->state= state;
2544     return END_NOT_FOUND;
2545 }
2546
2547 /* handle buffering and image synchronisation */
2548 static int mpeg_decode_frame(AVCodecContext *avctx,
2549                              void *data, int *data_size,
2550                              uint8_t *buf, int buf_size)
2551 {
2552     Mpeg1Context *s = avctx->priv_data;
2553     uint8_t *buf_end, *buf_ptr;
2554     int ret, start_code, input_size;
2555     AVFrame *picture = data;
2556     MpegEncContext *s2 = &s->mpeg_enc_ctx;
2557     dprintf("fill_buffer\n");
2558
2559     *data_size = 0;
2560
2561     /* special case for last picture */
2562     if (buf_size == 0 && s2->low_delay==0 && s2->next_picture_ptr) {
2563         *picture= *(AVFrame*)s2->next_picture_ptr;
2564         s2->next_picture_ptr= NULL;
2565
2566         *data_size = sizeof(AVFrame);
2567         return 0;
2568     }
2569
2570     if(s2->flags&CODEC_FLAG_TRUNCATED){
2571         int next= mpeg1_find_frame_end(s2, buf, buf_size);
2572
2573         if( ff_combine_frame(s2, next, &buf, &buf_size) < 0 )
2574             return buf_size;
2575     }
2576
2577     buf_ptr = buf;
2578     buf_end = buf + buf_size;
2579
2580 #if 0
2581     if (s->repeat_field % 2 == 1) {
2582         s->repeat_field++;
2583         //fprintf(stderr,"\nRepeating last frame: %d -> %d! pict: %d %d", avctx->frame_number-1, avctx->frame_number,
2584         //        s2->picture_number, s->repeat_field);
2585         if (avctx->flags & CODEC_FLAG_REPEAT_FIELD) {
2586             *data_size = sizeof(AVPicture);
2587             goto the_end;
2588         }
2589     }
2590 #endif
2591
2592     if(s->mpeg_enc_ctx_allocated==0 && avctx->codec_tag == ff_get_fourcc("VCR2"))
2593         vcr2_init_sequence(avctx);
2594
2595     for(;;) {
2596         /* find start next code */
2597         start_code = find_start_code(&buf_ptr, buf_end);
2598         if (start_code < 0){
2599             if(s2->pict_type != B_TYPE || avctx->hurry_up==0){
2600                 if (slice_end(avctx, picture)) {
2601                     if(s2->last_picture_ptr || s2->low_delay) //FIXME merge with the stuff in mpeg_decode_slice
2602                         *data_size = sizeof(AVPicture);
2603                 }
2604             }
2605             return FFMAX(0, buf_ptr - buf - s2->parse_context.last_index);
2606         }
2607
2608         input_size = buf_end - buf_ptr;
2609
2610         if(avctx->debug & FF_DEBUG_STARTCODE){
2611             av_log(avctx, AV_LOG_DEBUG, "%3X at %d left %d\n", start_code, buf_ptr-buf, input_size);
2612         }
2613
2614                 /* prepare data for next start code */
2615                 switch(start_code) {
2616                 case SEQ_START_CODE:
2617                     mpeg1_decode_sequence(avctx, buf_ptr,
2618                                           input_size);
2619                     break;
2620
2621                 case PICTURE_START_CODE:
2622                     /* we have a complete image : we try to decompress it */
2623                     mpeg1_decode_picture(avctx,
2624                                          buf_ptr, input_size);
2625                     break;
2626                 case EXT_START_CODE:
2627                     mpeg_decode_extension(avctx,
2628                                           buf_ptr, input_size);
2629                     break;
2630                 case USER_START_CODE:
2631                     mpeg_decode_user_data(avctx,
2632                                           buf_ptr, input_size);
2633                     break;
2634                 case GOP_START_CODE:
2635                     s2->first_field=0;
2636                     break;
2637                 default:
2638                     if (start_code >= SLICE_MIN_START_CODE &&
2639                         start_code <= SLICE_MAX_START_CODE) {
2640
2641                         /* skip b frames if we dont have reference frames */
2642                         if(s2->last_picture_ptr==NULL && s2->pict_type==B_TYPE) break;
2643                         /* skip b frames if we are in a hurry */
2644                         if(avctx->hurry_up && s2->pict_type==B_TYPE) break;
2645                         /* skip everything if we are in a hurry>=5 */
2646                         if(avctx->hurry_up>=5) break;
2647
2648                         if (!s->mpeg_enc_ctx_allocated) break;
2649
2650                         ret = mpeg_decode_slice(avctx, picture,
2651                                                 start_code, &buf_ptr, input_size);
2652                         emms_c();
2653
2654                         if(ret < 0){
2655                             if(s2->resync_mb_x>=0 && s2->resync_mb_y>=0)
2656                                 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);
2657                             if(ret==DECODE_SLICE_FATAL_ERROR) return -1;
2658                         }else{
2659                             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);
2660                         }
2661                     }
2662                     break;
2663                 }
2664     }
2665 }
2666
2667 static int mpeg_decode_end(AVCodecContext *avctx)
2668 {
2669     Mpeg1Context *s = avctx->priv_data;
2670
2671     if (s->mpeg_enc_ctx_allocated)
2672         MPV_common_end(&s->mpeg_enc_ctx);
2673     return 0;
2674 }
2675
2676 AVCodec mpeg1video_decoder = {
2677     "mpeg1video",
2678     CODEC_TYPE_VIDEO,
2679     CODEC_ID_MPEG1VIDEO,
2680     sizeof(Mpeg1Context),
2681     mpeg_decode_init,
2682     NULL,
2683     mpeg_decode_end,
2684     mpeg_decode_frame,
2685     CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED,
2686     .flush= ff_mpeg_flush,
2687 };
2688
2689 AVCodec mpeg2video_decoder = {
2690     "mpeg2video",
2691     CODEC_TYPE_VIDEO,
2692     CODEC_ID_MPEG2VIDEO,
2693     sizeof(Mpeg1Context),
2694     mpeg_decode_init,
2695     NULL,
2696     mpeg_decode_end,
2697     mpeg_decode_frame,
2698     CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED,
2699     .flush= ff_mpeg_flush,
2700 };
2701
2702 //legacy decoder
2703 AVCodec mpegvideo_decoder = {
2704     "mpegvideo",
2705     CODEC_TYPE_VIDEO,
2706     CODEC_ID_MPEG2VIDEO,
2707     sizeof(Mpeg1Context),
2708     mpeg_decode_init,
2709     NULL,
2710     mpeg_decode_end,
2711     mpeg_decode_frame,
2712     CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED,
2713     .flush= ff_mpeg_flush,
2714 };
2715
2716 #ifdef CONFIG_ENCODERS
2717
2718 AVCodec mpeg1video_encoder = {
2719     "mpeg1video",
2720     CODEC_TYPE_VIDEO,
2721     CODEC_ID_MPEG1VIDEO,
2722     sizeof(MpegEncContext),
2723     encode_init,
2724     MPV_encode_picture,
2725     MPV_encode_end,
2726 };
2727
2728 #ifdef CONFIG_RISKY
2729
2730 AVCodec mpeg2video_encoder = {
2731     "mpeg2video",
2732     CODEC_TYPE_VIDEO,
2733     CODEC_ID_MPEG2VIDEO,
2734     sizeof(MpegEncContext),
2735     encode_init,
2736     MPV_encode_picture,
2737     MPV_encode_end,
2738 };
2739 #endif
2740 #endif
2741
2742 #ifdef HAVE_XVMC
2743 static int mpeg_mc_decode_init(AVCodecContext *avctx){
2744     Mpeg1Context *s;
2745
2746     if( !(avctx->slice_flags & SLICE_FLAG_CODED_ORDER) )
2747         return -1;
2748     if( !(avctx->slice_flags & SLICE_FLAG_ALLOW_FIELD) ){
2749         dprintf("mpeg12.c: XvMC decoder will work better if SLICE_FLAG_ALLOW_FIELD is set\n");
2750     }
2751     mpeg_decode_init(avctx);
2752     s = avctx->priv_data;
2753
2754     avctx->pix_fmt = PIX_FMT_XVMC_MPEG2_IDCT;
2755     avctx->xvmc_acceleration = 2;//2 - the blocks are packed!
2756
2757     return 0;
2758 }
2759
2760 AVCodec mpeg_xvmc_decoder = {
2761     "mpegvideo_xvmc",
2762     CODEC_TYPE_VIDEO,
2763     CODEC_ID_MPEG2VIDEO_XVMC,
2764     sizeof(Mpeg1Context),
2765     mpeg_mc_decode_init,
2766     NULL,
2767     mpeg_decode_end,
2768     mpeg_decode_frame,
2769     CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED,
2770 };
2771
2772 #endif
2773
2774 /* this is ugly i know, but the alternative is too make
2775    hundreds of vars global and prefix them with ff_mpeg1_
2776    which is far uglier. */
2777 #include "mdec.c"