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