git.sesse.net Git - ffmpeg/blob - libavcodec/mpegvideo.c

   1 /*
   2  * The simplest mpeg encoder (well, it was the simplest!)
   3  * Copyright (c) 2000,2001 Fabrice Bellard
   4  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
   5  *
   6  * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
   7  *
   8  * This file is part of FFmpeg.
   9  *
  10  * FFmpeg is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * FFmpeg is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with FFmpeg; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23  */
  24
  25 /**
  26  * @file
  27  * The simplest mpeg encoder (well, it was the simplest!).
  28  */
  29
  30 #include "libavutil/attributes.h"
  31 #include "libavutil/avassert.h"
  32 #include "libavutil/imgutils.h"
  33 #include "avcodec.h"
  34 #include "dsputil.h"
  35 #include "h264chroma.h"
  36 #include "internal.h"
  37 #include "mathops.h"
  38 #include "mpegvideo.h"
  39 #include "mjpegenc.h"
  40 #include "msmpeg4.h"
  41 #include "xvmc_internal.h"
  42 #include "thread.h"
  43 #include <limits.h>
  44
  45 static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
  46                                    int16_t *block, int n, int qscale);
  47 static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
  48                                    int16_t *block, int n, int qscale);
  49 static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
  50                                    int16_t *block, int n, int qscale);
  51 static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
  52                                    int16_t *block, int n, int qscale);
  53 static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
  54                                    int16_t *block, int n, int qscale);
  55 static void dct_unquantize_h263_intra_c(MpegEncContext *s,
  56                                   int16_t *block, int n, int qscale);
  57 static void dct_unquantize_h263_inter_c(MpegEncContext *s,
  58                                   int16_t *block, int n, int qscale);
  59
  60 static const uint8_t ff_default_chroma_qscale_table[32] = {
  61 //   0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15
  62      0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15,
  63     16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31
  64 };
  65
  66 const uint8_t ff_mpeg1_dc_scale_table[128] = {
  67 //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15
  68     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  69     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  70     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  71     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  72     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  73     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  74     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  75     8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
  76 };
  77
  78 static const uint8_t mpeg2_dc_scale_table1[128] = {
  79 //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15
  80     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  81     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  82     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  83     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  84     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  85     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  86     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  87     4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  88 };
  89
  90 static const uint8_t mpeg2_dc_scale_table2[128] = {
  91 //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15
  92     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  93     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  94     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  95     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  96     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  97     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  98     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  99     2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
 100 };
 101
 102 static const uint8_t mpeg2_dc_scale_table3[128] = {
 103 //  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15
 104     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 105     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 106     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 107     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 108     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 109     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 110     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 111     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 112 };
 113
 114 const uint8_t *const ff_mpeg2_dc_scale_table[4] = {
 115     ff_mpeg1_dc_scale_table,
 116     mpeg2_dc_scale_table1,
 117     mpeg2_dc_scale_table2,
 118     mpeg2_dc_scale_table3,
 119 };
 120
 121 const enum AVPixelFormat ff_pixfmt_list_420[] = {
 122     AV_PIX_FMT_YUV420P,
 123     AV_PIX_FMT_NONE
 124 };
 125
 126 static void mpeg_er_decode_mb(void *opaque, int ref, int mv_dir, int mv_type,
 127                               int (*mv)[2][4][2],
 128                               int mb_x, int mb_y, int mb_intra, int mb_skipped)
 129 {
 130     MpegEncContext *s = opaque;
 131
 132     s->mv_dir     = mv_dir;
 133     s->mv_type    = mv_type;
 134     s->mb_intra   = mb_intra;
 135     s->mb_skipped = mb_skipped;
 136     s->mb_x       = mb_x;
 137     s->mb_y       = mb_y;
 138     memcpy(s->mv, mv, sizeof(*mv));
 139
 140     ff_init_block_index(s);
 141     ff_update_block_index(s);
 142
 143     s->dsp.clear_blocks(s->block[0]);
 144
 145     s->dest[0] = s->current_picture.f.data[0] + (s->mb_y *  16                       * s->linesize)   + s->mb_x *  16;
 146     s->dest[1] = s->current_picture.f.data[1] + (s->mb_y * (16 >> s->chroma_y_shift) * s->uvlinesize) + s->mb_x * (16 >> s->chroma_x_shift);
 147     s->dest[2] = s->current_picture.f.data[2] + (s->mb_y * (16 >> s->chroma_y_shift) * s->uvlinesize) + s->mb_x * (16 >> s->chroma_x_shift);
 148
 149     if (ref)
 150         av_log(s->avctx, AV_LOG_DEBUG, "Interlaced error concealment is not fully implemented\n");
 151     ff_MPV_decode_mb(s, s->block);
 152 }
 153
 154 /* init common dct for both encoder and decoder */
 155 av_cold int ff_dct_common_init(MpegEncContext *s)
 156 {
 157     ff_dsputil_init(&s->dsp, s->avctx);
 158     ff_h264chroma_init(&s->h264chroma, 8); //for lowres
 159     ff_hpeldsp_init(&s->hdsp, s->avctx->flags);
 160     ff_videodsp_init(&s->vdsp, s->avctx->bits_per_raw_sample);
 161
 162     s->dct_unquantize_h263_intra = dct_unquantize_h263_intra_c;
 163     s->dct_unquantize_h263_inter = dct_unquantize_h263_inter_c;
 164     s->dct_unquantize_mpeg1_intra = dct_unquantize_mpeg1_intra_c;
 165     s->dct_unquantize_mpeg1_inter = dct_unquantize_mpeg1_inter_c;
 166     s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_c;
 167     if (s->flags & CODEC_FLAG_BITEXACT)
 168         s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_bitexact;
 169     s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_c;
 170
 171 #if ARCH_X86
 172     ff_MPV_common_init_x86(s);
 173 #elif ARCH_ALPHA
 174     ff_MPV_common_init_axp(s);
 175 #elif ARCH_ARM
 176     ff_MPV_common_init_arm(s);
 177 #elif ARCH_BFIN
 178     ff_MPV_common_init_bfin(s);
 179 #elif ARCH_PPC
 180     ff_MPV_common_init_ppc(s);
 181 #endif
 182
 183     /* load & permutate scantables
 184      * note: only wmv uses different ones
 185      */
 186     if (s->alternate_scan) {
 187         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_alternate_vertical_scan);
 188         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_alternate_vertical_scan);
 189     } else {
 190         ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_zigzag_direct);
 191         ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_zigzag_direct);
 192     }
 193     ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable, ff_alternate_horizontal_scan);
 194     ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable, ff_alternate_vertical_scan);
 195
 196     return 0;
 197 }
 198
 199 int ff_mpv_frame_size_alloc(MpegEncContext *s, int linesize)
 200 {
 201     int alloc_size = FFALIGN(FFABS(linesize) + 64, 32);
 202
 203     // edge emu needs blocksize + filter length - 1
 204     // (= 17x17 for  halfpel / 21x21 for  h264)
 205     // VC1 computes luma and chroma simultaneously and needs 19X19 + 9x9
 206     // at uvlinesize. It supports only YUV420 so 24x24 is enough
 207     // linesize * interlaced * MBsize
 208     FF_ALLOCZ_OR_GOTO(s->avctx, s->edge_emu_buffer, alloc_size * 4 * 24,
 209                       fail);
 210
 211     FF_ALLOCZ_OR_GOTO(s->avctx, s->me.scratchpad, alloc_size * 4 * 16 * 2,
 212                       fail)
 213     s->me.temp         = s->me.scratchpad;
 214     s->rd_scratchpad   = s->me.scratchpad;
 215     s->b_scratchpad    = s->me.scratchpad;
 216     s->obmc_scratchpad = s->me.scratchpad + 16;
 217
 218     return 0;
 219 fail:
 220     av_freep(&s->edge_emu_buffer);
 221     return AVERROR(ENOMEM);
 222 }
 223
 224 /**
 225  * Allocate a frame buffer
 226  */
 227 static int alloc_frame_buffer(MpegEncContext *s, Picture *pic)
 228 {
 229     int r, ret;
 230
 231     pic->tf.f = &pic->f;
 232     if (s->codec_id != AV_CODEC_ID_WMV3IMAGE &&
 233         s->codec_id != AV_CODEC_ID_VC1IMAGE  &&
 234         s->codec_id != AV_CODEC_ID_MSS2)
 235         r = ff_thread_get_buffer(s->avctx, &pic->tf,
 236                                  pic->reference ? AV_GET_BUFFER_FLAG_REF : 0);
 237     else {
 238         pic->f.width  = s->avctx->width;
 239         pic->f.height = s->avctx->height;
 240         pic->f.format = s->avctx->pix_fmt;
 241         r = avcodec_default_get_buffer2(s->avctx, &pic->f, 0);
 242     }
 243
 244     if (r < 0 || !pic->f.data[0]) {
 245         av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (%d %p)\n",
 246                r, pic->f.data[0]);
 247         return -1;
 248     }
 249
 250     if (s->avctx->hwaccel) {
 251         assert(!pic->hwaccel_picture_private);
 252         if (s->avctx->hwaccel->priv_data_size) {
 253             pic->hwaccel_priv_buf = av_buffer_allocz(s->avctx->hwaccel->priv_data_size);
 254             if (!pic->hwaccel_priv_buf) {
 255                 av_log(s->avctx, AV_LOG_ERROR, "alloc_frame_buffer() failed (hwaccel private data allocation)\n");
 256                 return -1;
 257             }
 258             pic->hwaccel_picture_private = pic->hwaccel_priv_buf->data;
 259         }
 260     }
 261
 262     if (s->linesize && (s->linesize   != pic->f.linesize[0] ||
 263                         s->uvlinesize != pic->f.linesize[1])) {
 264         av_log(s->avctx, AV_LOG_ERROR,
 265                "get_buffer() failed (stride changed)\n");
 266         ff_mpeg_unref_picture(s, pic);
 267         return -1;
 268     }
 269
 270     if (pic->f.linesize[1] != pic->f.linesize[2]) {
 271         av_log(s->avctx, AV_LOG_ERROR,
 272                "get_buffer() failed (uv stride mismatch)\n");
 273         ff_mpeg_unref_picture(s, pic);
 274         return -1;
 275     }
 276
 277     if (!s->edge_emu_buffer &&
 278         (ret = ff_mpv_frame_size_alloc(s, pic->f.linesize[0])) < 0) {
 279         av_log(s->avctx, AV_LOG_ERROR,
 280                "get_buffer() failed to allocate context scratch buffers.\n");
 281         ff_mpeg_unref_picture(s, pic);
 282         return ret;
 283     }
 284
 285     return 0;
 286 }
 287
 288 static void free_picture_tables(Picture *pic)
 289 {
 290     int i;
 291
 292     pic->alloc_mb_width  =
 293     pic->alloc_mb_height = 0;
 294
 295     av_buffer_unref(&pic->mb_var_buf);
 296     av_buffer_unref(&pic->mc_mb_var_buf);
 297     av_buffer_unref(&pic->mb_mean_buf);
 298     av_buffer_unref(&pic->mbskip_table_buf);
 299     av_buffer_unref(&pic->qscale_table_buf);
 300     av_buffer_unref(&pic->mb_type_buf);
 301
 302     for (i = 0; i < 2; i++) {
 303         av_buffer_unref(&pic->motion_val_buf[i]);
 304         av_buffer_unref(&pic->ref_index_buf[i]);
 305     }
 306 }
 307
 308 static int alloc_picture_tables(MpegEncContext *s, Picture *pic)
 309 {
 310     const int big_mb_num    = s->mb_stride * (s->mb_height + 1) + 1;
 311     const int mb_array_size = s->mb_stride * s->mb_height;
 312     const int b8_array_size = s->b8_stride * s->mb_height * 2;
 313     int i;
 314
 315
 316     pic->mbskip_table_buf = av_buffer_allocz(mb_array_size + 2);
 317     pic->qscale_table_buf = av_buffer_allocz(big_mb_num + s->mb_stride);
 318     pic->mb_type_buf      = av_buffer_allocz((big_mb_num + s->mb_stride) *
 319                                              sizeof(uint32_t));
 320     if (!pic->mbskip_table_buf || !pic->qscale_table_buf || !pic->mb_type_buf)
 321         return AVERROR(ENOMEM);
 322
 323     if (s->encoding) {
 324         pic->mb_var_buf    = av_buffer_allocz(mb_array_size * sizeof(int16_t));
 325         pic->mc_mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t));
 326         pic->mb_mean_buf   = av_buffer_allocz(mb_array_size);
 327         if (!pic->mb_var_buf || !pic->mc_mb_var_buf || !pic->mb_mean_buf)
 328             return AVERROR(ENOMEM);
 329     }
 330
 331     if (s->out_format == FMT_H263 || s->encoding ||
 332                (s->avctx->debug & FF_DEBUG_MV) || s->avctx->debug_mv) {
 333         int mv_size        = 2 * (b8_array_size + 4) * sizeof(int16_t);
 334         int ref_index_size = 4 * mb_array_size;
 335
 336         for (i = 0; mv_size && i < 2; i++) {
 337             pic->motion_val_buf[i] = av_buffer_allocz(mv_size);
 338             pic->ref_index_buf[i]  = av_buffer_allocz(ref_index_size);
 339             if (!pic->motion_val_buf[i] || !pic->ref_index_buf[i])
 340                 return AVERROR(ENOMEM);
 341         }
 342     }
 343
 344     pic->alloc_mb_width  = s->mb_width;
 345     pic->alloc_mb_height = s->mb_height;
 346
 347     return 0;
 348 }
 349
 350 static int make_tables_writable(Picture *pic)
 351 {
 352     int ret, i;
 353 #define MAKE_WRITABLE(table) \
 354 do {\
 355     if (pic->table &&\
 356        (ret = av_buffer_make_writable(&pic->table)) < 0)\
 357     return ret;\
 358 } while (0)
 359
 360     MAKE_WRITABLE(mb_var_buf);
 361     MAKE_WRITABLE(mc_mb_var_buf);
 362     MAKE_WRITABLE(mb_mean_buf);
 363     MAKE_WRITABLE(mbskip_table_buf);
 364     MAKE_WRITABLE(qscale_table_buf);
 365     MAKE_WRITABLE(mb_type_buf);
 366
 367     for (i = 0; i < 2; i++) {
 368         MAKE_WRITABLE(motion_val_buf[i]);
 369         MAKE_WRITABLE(ref_index_buf[i]);
 370     }
 371
 372     return 0;
 373 }
 374
 375 /**
 376  * Allocate a Picture.
 377  * The pixels are allocated/set by calling get_buffer() if shared = 0
 378  */
 379 int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared)
 380 {
 381     int i, ret;
 382
 383     if (pic->qscale_table_buf)
 384         if (   pic->alloc_mb_width  != s->mb_width
 385             || pic->alloc_mb_height != s->mb_height)
 386             free_picture_tables(pic);
 387
 388     if (shared) {
 389         av_assert0(pic->f.data[0]);
 390         pic->shared = 1;
 391     } else {
 392         av_assert0(!pic->f.data[0]);
 393
 394         if (alloc_frame_buffer(s, pic) < 0)
 395             return -1;
 396
 397         s->linesize   = pic->f.linesize[0];
 398         s->uvlinesize = pic->f.linesize[1];
 399     }
 400
 401     if (!pic->qscale_table_buf)
 402         ret = alloc_picture_tables(s, pic);
 403     else
 404         ret = make_tables_writable(pic);
 405     if (ret < 0)
 406         goto fail;
 407
 408     if (s->encoding) {
 409         pic->mb_var    = (uint16_t*)pic->mb_var_buf->data;
 410         pic->mc_mb_var = (uint16_t*)pic->mc_mb_var_buf->data;
 411         pic->mb_mean   = pic->mb_mean_buf->data;
 412     }
 413
 414     pic->mbskip_table = pic->mbskip_table_buf->data;
 415     pic->qscale_table = pic->qscale_table_buf->data + 2 * s->mb_stride + 1;
 416     pic->mb_type      = (uint32_t*)pic->mb_type_buf->data + 2 * s->mb_stride + 1;
 417
 418     if (pic->motion_val_buf[0]) {
 419         for (i = 0; i < 2; i++) {
 420             pic->motion_val[i] = (int16_t (*)[2])pic->motion_val_buf[i]->data + 4;
 421             pic->ref_index[i]  = pic->ref_index_buf[i]->data;
 422         }
 423     }
 424
 425     return 0;
 426 fail:
 427     av_log(s->avctx, AV_LOG_ERROR, "Error allocating a picture.\n");
 428     ff_mpeg_unref_picture(s, pic);
 429     free_picture_tables(pic);
 430     return AVERROR(ENOMEM);
 431 }
 432
 433 /**
 434  * Deallocate a picture.
 435  */
 436 void ff_mpeg_unref_picture(MpegEncContext *s, Picture *pic)
 437 {
 438     int off = offsetof(Picture, mb_mean) + sizeof(pic->mb_mean);
 439
 440     pic->tf.f = &pic->f;
 441     /* WM Image / Screen codecs allocate internal buffers with different
 442      * dimensions / colorspaces; ignore user-defined callbacks for these. */
 443     if (s->codec_id != AV_CODEC_ID_WMV3IMAGE &&
 444         s->codec_id != AV_CODEC_ID_VC1IMAGE  &&
 445         s->codec_id != AV_CODEC_ID_MSS2)
 446         ff_thread_release_buffer(s->avctx, &pic->tf);
 447     else
 448         av_frame_unref(&pic->f);
 449
 450     av_buffer_unref(&pic->hwaccel_priv_buf);
 451
 452     if (pic->needs_realloc)
 453         free_picture_tables(pic);
 454
 455     memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
 456 }
 457
 458 static int update_picture_tables(Picture *dst, Picture *src)
 459 {
 460      int i;
 461
 462 #define UPDATE_TABLE(table)\
 463 do {\
 464     if (src->table &&\
 465         (!dst->table || dst->table->buffer != src->table->buffer)) {\
 466         av_buffer_unref(&dst->table);\
 467         dst->table = av_buffer_ref(src->table);\
 468         if (!dst->table) {\
 469             free_picture_tables(dst);\
 470             return AVERROR(ENOMEM);\
 471         }\
 472     }\
 473 } while (0)
 474
 475     UPDATE_TABLE(mb_var_buf);
 476     UPDATE_TABLE(mc_mb_var_buf);
 477     UPDATE_TABLE(mb_mean_buf);
 478     UPDATE_TABLE(mbskip_table_buf);
 479     UPDATE_TABLE(qscale_table_buf);
 480     UPDATE_TABLE(mb_type_buf);
 481     for (i = 0; i < 2; i++) {
 482         UPDATE_TABLE(motion_val_buf[i]);
 483         UPDATE_TABLE(ref_index_buf[i]);
 484     }
 485
 486     dst->mb_var        = src->mb_var;
 487     dst->mc_mb_var     = src->mc_mb_var;
 488     dst->mb_mean       = src->mb_mean;
 489     dst->mbskip_table  = src->mbskip_table;
 490     dst->qscale_table  = src->qscale_table;
 491     dst->mb_type       = src->mb_type;
 492     for (i = 0; i < 2; i++) {
 493         dst->motion_val[i] = src->motion_val[i];
 494         dst->ref_index[i]  = src->ref_index[i];
 495     }
 496
 497     dst->alloc_mb_width  = src->alloc_mb_width;
 498     dst->alloc_mb_height = src->alloc_mb_height;
 499
 500     return 0;
 501 }
 502
 503 int ff_mpeg_ref_picture(MpegEncContext *s, Picture *dst, Picture *src)
 504 {
 505     int ret;
 506
 507     av_assert0(!dst->f.buf[0]);
 508     av_assert0(src->f.buf[0]);
 509
 510     src->tf.f = &src->f;
 511     dst->tf.f = &dst->f;
 512     ret = ff_thread_ref_frame(&dst->tf, &src->tf);
 513     if (ret < 0)
 514         goto fail;
 515
 516     ret = update_picture_tables(dst, src);
 517     if (ret < 0)
 518         goto fail;
 519
 520     if (src->hwaccel_picture_private) {
 521         dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf);
 522         if (!dst->hwaccel_priv_buf)
 523             goto fail;
 524         dst->hwaccel_picture_private = dst->hwaccel_priv_buf->data;
 525     }
 526
 527     dst->field_picture           = src->field_picture;
 528     dst->mb_var_sum              = src->mb_var_sum;
 529     dst->mc_mb_var_sum           = src->mc_mb_var_sum;
 530     dst->b_frame_score           = src->b_frame_score;
 531     dst->needs_realloc           = src->needs_realloc;
 532     dst->reference               = src->reference;
 533     dst->shared                  = src->shared;
 534
 535     return 0;
 536 fail:
 537     ff_mpeg_unref_picture(s, dst);
 538     return ret;
 539 }
 540
 541 static int init_duplicate_context(MpegEncContext *s)
 542 {
 543     int y_size = s->b8_stride * (2 * s->mb_height + 1);
 544     int c_size = s->mb_stride * (s->mb_height + 1);
 545     int yc_size = y_size + 2 * c_size;
 546     int i;
 547
 548     s->edge_emu_buffer =
 549     s->me.scratchpad   =
 550     s->me.temp         =
 551     s->rd_scratchpad   =
 552     s->b_scratchpad    =
 553     s->obmc_scratchpad = NULL;
 554
 555     if (s->encoding) {
 556         FF_ALLOCZ_OR_GOTO(s->avctx, s->me.map,
 557                           ME_MAP_SIZE * sizeof(uint32_t), fail)
 558         FF_ALLOCZ_OR_GOTO(s->avctx, s->me.score_map,
 559                           ME_MAP_SIZE * sizeof(uint32_t), fail)
 560         if (s->avctx->noise_reduction) {
 561             FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_error_sum,
 562                               2 * 64 * sizeof(int), fail)
 563         }
 564     }
 565     FF_ALLOCZ_OR_GOTO(s->avctx, s->blocks, 64 * 12 * 2 * sizeof(int16_t), fail)
 566     s->block = s->blocks[0];
 567
 568     for (i = 0; i < 12; i++) {
 569         s->pblocks[i] = &s->block[i];
 570     }
 571
 572     if (s->out_format == FMT_H263) {
 573         /* ac values */
 574         FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_val_base,
 575                           yc_size * sizeof(int16_t) * 16, fail);
 576         s->ac_val[0] = s->ac_val_base + s->b8_stride + 1;
 577         s->ac_val[1] = s->ac_val_base + y_size + s->mb_stride + 1;
 578         s->ac_val[2] = s->ac_val[1] + c_size;
 579     }
 580
 581     return 0;
 582 fail:
 583     return -1; // free() through ff_MPV_common_end()
 584 }
 585
 586 static void free_duplicate_context(MpegEncContext *s)
 587 {
 588     if (s == NULL)
 589         return;
 590
 591     av_freep(&s->edge_emu_buffer);
 592     av_freep(&s->me.scratchpad);
 593     s->me.temp =
 594     s->rd_scratchpad =
 595     s->b_scratchpad =
 596     s->obmc_scratchpad = NULL;
 597
 598     av_freep(&s->dct_error_sum);
 599     av_freep(&s->me.map);
 600     av_freep(&s->me.score_map);
 601     av_freep(&s->blocks);
 602     av_freep(&s->ac_val_base);
 603     s->block = NULL;
 604 }
 605
 606 static void backup_duplicate_context(MpegEncContext *bak, MpegEncContext *src)
 607 {
 608 #define COPY(a) bak->a = src->a
 609     COPY(edge_emu_buffer);
 610     COPY(me.scratchpad);
 611     COPY(me.temp);
 612     COPY(rd_scratchpad);
 613     COPY(b_scratchpad);
 614     COPY(obmc_scratchpad);
 615     COPY(me.map);
 616     COPY(me.score_map);
 617     COPY(blocks);
 618     COPY(block);
 619     COPY(start_mb_y);
 620     COPY(end_mb_y);
 621     COPY(me.map_generation);
 622     COPY(pb);
 623     COPY(dct_error_sum);
 624     COPY(dct_count[0]);
 625     COPY(dct_count[1]);
 626     COPY(ac_val_base);
 627     COPY(ac_val[0]);
 628     COPY(ac_val[1]);
 629     COPY(ac_val[2]);
 630 #undef COPY
 631 }
 632
 633 int ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src)
 634 {
 635     MpegEncContext bak;
 636     int i, ret;
 637     // FIXME copy only needed parts
 638     // START_TIMER
 639     backup_duplicate_context(&bak, dst);
 640     memcpy(dst, src, sizeof(MpegEncContext));
 641     backup_duplicate_context(dst, &bak);
 642     for (i = 0; i < 12; i++) {
 643         dst->pblocks[i] = &dst->block[i];
 644     }
 645     if (!dst->edge_emu_buffer &&
 646         (ret = ff_mpv_frame_size_alloc(dst, dst->linesize)) < 0) {
 647         av_log(dst->avctx, AV_LOG_ERROR, "failed to allocate context "
 648                "scratch buffers.\n");
 649         return ret;
 650     }
 651     // STOP_TIMER("update_duplicate_context")
 652     // about 10k cycles / 0.01 sec for  1000frames on 1ghz with 2 threads
 653     return 0;
 654 }
 655
 656 int ff_mpeg_update_thread_context(AVCodecContext *dst,
 657                                   const AVCodecContext *src)
 658 {
 659     int i, ret;
 660     MpegEncContext *s = dst->priv_data, *s1 = src->priv_data;
 661
 662     if (dst == src)
 663         return 0;
 664
 665     av_assert0(s != s1);
 666
 667     // FIXME can parameters change on I-frames?
 668     // in that case dst may need a reinit
 669     if (!s->context_initialized) {
 670         memcpy(s, s1, sizeof(MpegEncContext));
 671
 672         s->avctx                 = dst;
 673         s->bitstream_buffer      = NULL;
 674         s->bitstream_buffer_size = s->allocated_bitstream_buffer_size = 0;
 675
 676         if (s1->context_initialized){
 677 //             s->picture_range_start  += MAX_PICTURE_COUNT;
 678 //             s->picture_range_end    += MAX_PICTURE_COUNT;
 679             if((ret = ff_MPV_common_init(s)) < 0){
 680                 memset(s, 0, sizeof(MpegEncContext));
 681                 s->avctx = dst;
 682                 return ret;
 683             }
 684         }
 685     }
 686
 687     if (s->height != s1->height || s->width != s1->width || s->context_reinit) {
 688         s->context_reinit = 0;
 689         s->height = s1->height;
 690         s->width  = s1->width;
 691         if ((ret = ff_MPV_common_frame_size_change(s)) < 0)
 692             return ret;
 693     }
 694
 695     s->avctx->coded_height  = s1->avctx->coded_height;
 696     s->avctx->coded_width   = s1->avctx->coded_width;
 697     s->avctx->width         = s1->avctx->width;
 698     s->avctx->height        = s1->avctx->height;
 699
 700     s->coded_picture_number = s1->coded_picture_number;
 701     s->picture_number       = s1->picture_number;
 702     s->input_picture_number = s1->input_picture_number;
 703
 704     av_assert0(!s->picture || s->picture != s1->picture);
 705     if(s->picture)
 706     for (i = 0; i < MAX_PICTURE_COUNT; i++) {
 707         ff_mpeg_unref_picture(s, &s->picture[i]);
 708         if (s1->picture[i].f.data[0] &&
 709             (ret = ff_mpeg_ref_picture(s, &s->picture[i], &s1->picture[i])) < 0)
 710             return ret;
 711     }
 712
 713 #define UPDATE_PICTURE(pic)\
 714 do {\
 715     ff_mpeg_unref_picture(s, &s->pic);\
 716     if (s1->pic.f.data[0])\
 717         ret = ff_mpeg_ref_picture(s, &s->pic, &s1->pic);\
 718     else\
 719         ret = update_picture_tables(&s->pic, &s1->pic);\
 720     if (ret < 0)\
 721         return ret;\
 722 } while (0)
 723
 724     UPDATE_PICTURE(current_picture);
 725     UPDATE_PICTURE(last_picture);
 726     UPDATE_PICTURE(next_picture);
 727
 728     s->last_picture_ptr    = REBASE_PICTURE(s1->last_picture_ptr,    s, s1);
 729     s->current_picture_ptr = REBASE_PICTURE(s1->current_picture_ptr, s, s1);
 730     s->next_picture_ptr    = REBASE_PICTURE(s1->next_picture_ptr,    s, s1);
 731
 732     // Error/bug resilience
 733     s->next_p_frame_damaged = s1->next_p_frame_damaged;
 734     s->workaround_bugs      = s1->workaround_bugs;
 735     s->padding_bug_score    = s1->padding_bug_score;
 736
 737     // MPEG4 timing info
 738     memcpy(&s->time_increment_bits, &s1->time_increment_bits,
 739            (char *) &s1->shape - (char *) &s1->time_increment_bits);
 740
 741     // B-frame info
 742     s->max_b_frames = s1->max_b_frames;
 743     s->low_delay    = s1->low_delay;
 744     s->droppable    = s1->droppable;
 745
 746     // DivX handling (doesn't work)
 747     s->divx_packed  = s1->divx_packed;
 748
 749     if (s1->bitstream_buffer) {
 750         if (s1->bitstream_buffer_size +
 751             FF_INPUT_BUFFER_PADDING_SIZE > s->allocated_bitstream_buffer_size)
 752             av_fast_malloc(&s->bitstream_buffer,
 753                            &s->allocated_bitstream_buffer_size,
 754                            s1->allocated_bitstream_buffer_size);
 755             s->bitstream_buffer_size = s1->bitstream_buffer_size;
 756         memcpy(s->bitstream_buffer, s1->bitstream_buffer,
 757                s1->bitstream_buffer_size);
 758         memset(s->bitstream_buffer + s->bitstream_buffer_size, 0,
 759                FF_INPUT_BUFFER_PADDING_SIZE);
 760     }
 761
 762     // linesize dependend scratch buffer allocation
 763     if (!s->edge_emu_buffer)
 764         if (s1->linesize) {
 765             if (ff_mpv_frame_size_alloc(s, s1->linesize) < 0) {
 766                 av_log(s->avctx, AV_LOG_ERROR, "Failed to allocate context "
 767                        "scratch buffers.\n");
 768                 return AVERROR(ENOMEM);
 769             }
 770         } else {
 771             av_log(s->avctx, AV_LOG_ERROR, "Context scratch buffers could not "
 772                    "be allocated due to unknown size.\n");
 773         }
 774
 775     // MPEG2/interlacing info
 776     memcpy(&s->progressive_sequence, &s1->progressive_sequence,
 777            (char *) &s1->rtp_mode - (char *) &s1->progressive_sequence);
 778
 779     if (!s1->first_field) {
 780         s->last_pict_type = s1->pict_type;
 781         if (s1->current_picture_ptr)
 782             s->last_lambda_for[s1->pict_type] = s1->current_picture_ptr->f.quality;
 783
 784         if (s1->pict_type != AV_PICTURE_TYPE_B) {
 785             s->last_non_b_pict_type = s1->pict_type;
 786         }
 787     }
 788
 789     return 0;
 790 }
 791
 792 /**
 793  * Set the given MpegEncContext to common defaults
 794  * (same for encoding and decoding).
 795  * The changed fields will not depend upon the
 796  * prior state of the MpegEncContext.
 797  */
 798 void ff_MPV_common_defaults(MpegEncContext *s)
 799 {
 800     s->y_dc_scale_table      =
 801     s->c_dc_scale_table      = ff_mpeg1_dc_scale_table;
 802     s->chroma_qscale_table   = ff_default_chroma_qscale_table;
 803     s->progressive_frame     = 1;
 804     s->progressive_sequence  = 1;
 805     s->picture_structure     = PICT_FRAME;
 806
 807     s->coded_picture_number  = 0;
 808     s->picture_number        = 0;
 809     s->input_picture_number  = 0;
 810
 811     s->picture_in_gop_number = 0;
 812
 813     s->f_code                = 1;
 814     s->b_code                = 1;
 815
 816     s->slice_context_count   = 1;
 817 }
 818
 819 /**
 820  * Set the given MpegEncContext to defaults for decoding.
 821  * the changed fields will not depend upon
 822  * the prior state of the MpegEncContext.
 823  */
 824 void ff_MPV_decode_defaults(MpegEncContext *s)
 825 {
 826     ff_MPV_common_defaults(s);
 827 }
 828
 829 static int init_er(MpegEncContext *s)
 830 {
 831     ERContext *er = &s->er;
 832     int mb_array_size = s->mb_height * s->mb_stride;
 833     int i;
 834
 835     er->avctx       = s->avctx;
 836     er->dsp         = &s->dsp;
 837
 838     er->mb_index2xy = s->mb_index2xy;
 839     er->mb_num      = s->mb_num;
 840     er->mb_width    = s->mb_width;
 841     er->mb_height   = s->mb_height;
 842     er->mb_stride   = s->mb_stride;
 843     er->b8_stride   = s->b8_stride;
 844
 845     er->er_temp_buffer     = av_malloc(s->mb_height * s->mb_stride);
 846     er->error_status_table = av_mallocz(mb_array_size);
 847     if (!er->er_temp_buffer || !er->error_status_table)
 848         goto fail;
 849
 850     er->mbskip_table  = s->mbskip_table;
 851     er->mbintra_table = s->mbintra_table;
 852
 853     for (i = 0; i < FF_ARRAY_ELEMS(s->dc_val); i++)
 854         er->dc_val[i] = s->dc_val[i];
 855
 856     er->decode_mb = mpeg_er_decode_mb;
 857     er->opaque    = s;
 858
 859     return 0;
 860 fail:
 861     av_freep(&er->er_temp_buffer);
 862     av_freep(&er->error_status_table);
 863     return AVERROR(ENOMEM);
 864 }
 865
 866 /**
 867  * Initialize and allocates MpegEncContext fields dependent on the resolution.
 868  */
 869 static int init_context_frame(MpegEncContext *s)
 870 {
 871     int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y;
 872
 873     s->mb_width   = (s->width + 15) / 16;
 874     s->mb_stride  = s->mb_width + 1;
 875     s->b8_stride  = s->mb_width * 2 + 1;
 876     s->b4_stride  = s->mb_width * 4 + 1;
 877     mb_array_size = s->mb_height * s->mb_stride;
 878     mv_table_size = (s->mb_height + 2) * s->mb_stride + 1;
 879
 880     /* set default edge pos, will be overriden
 881      * in decode_header if needed */
 882     s->h_edge_pos = s->mb_width * 16;
 883     s->v_edge_pos = s->mb_height * 16;
 884
 885     s->mb_num     = s->mb_width * s->mb_height;
 886
 887     s->block_wrap[0] =
 888     s->block_wrap[1] =
 889     s->block_wrap[2] =
 890     s->block_wrap[3] = s->b8_stride;
 891     s->block_wrap[4] =
 892     s->block_wrap[5] = s->mb_stride;
 893
 894     y_size  = s->b8_stride * (2 * s->mb_height + 1);
 895     c_size  = s->mb_stride * (s->mb_height + 1);
 896     yc_size = y_size + 2   * c_size;
 897
 898     FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_index2xy, (s->mb_num + 1) * sizeof(int), fail); // error ressilience code looks cleaner with this
 899     for (y = 0; y < s->mb_height; y++)
 900         for (x = 0; x < s->mb_width; x++)
 901             s->mb_index2xy[x + y * s->mb_width] = x + y * s->mb_stride;
 902
 903     s->mb_index2xy[s->mb_height * s->mb_width] = (s->mb_height - 1) * s->mb_stride + s->mb_width; // FIXME really needed?
 904
 905     if (s->encoding) {
 906         /* Allocate MV tables */
 907         FF_ALLOCZ_OR_GOTO(s->avctx, s->p_mv_table_base,                 mv_table_size * 2 * sizeof(int16_t), fail)
 908         FF_ALLOCZ_OR_GOTO(s->avctx, s->b_forw_mv_table_base,            mv_table_size * 2 * sizeof(int16_t), fail)
 909         FF_ALLOCZ_OR_GOTO(s->avctx, s->b_back_mv_table_base,            mv_table_size * 2 * sizeof(int16_t), fail)
 910         FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_forw_mv_table_base,      mv_table_size * 2 * sizeof(int16_t), fail)
 911         FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_back_mv_table_base,      mv_table_size * 2 * sizeof(int16_t), fail)
 912         FF_ALLOCZ_OR_GOTO(s->avctx, s->b_direct_mv_table_base,          mv_table_size * 2 * sizeof(int16_t), fail)
 913         s->p_mv_table            = s->p_mv_table_base + s->mb_stride + 1;
 914         s->b_forw_mv_table       = s->b_forw_mv_table_base + s->mb_stride + 1;
 915         s->b_back_mv_table       = s->b_back_mv_table_base + s->mb_stride + 1;
 916         s->b_bidir_forw_mv_table = s->b_bidir_forw_mv_table_base + s->mb_stride + 1;
 917         s->b_bidir_back_mv_table = s->b_bidir_back_mv_table_base + s->mb_stride + 1;
 918         s->b_direct_mv_table     = s->b_direct_mv_table_base + s->mb_stride + 1;
 919
 920         /* Allocate MB type table */
 921         FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_type, mb_array_size * sizeof(uint16_t), fail) // needed for encoding
 922
 923         FF_ALLOCZ_OR_GOTO(s->avctx, s->lambda_table, mb_array_size * sizeof(int), fail)
 924
 925         FF_ALLOC_OR_GOTO(s->avctx, s->cplx_tab,
 926                          mb_array_size * sizeof(float), fail);
 927         FF_ALLOC_OR_GOTO(s->avctx, s->bits_tab,
 928                          mb_array_size * sizeof(float), fail);
 929
 930     }
 931
 932     if (s->codec_id == AV_CODEC_ID_MPEG4 ||
 933         (s->flags & CODEC_FLAG_INTERLACED_ME)) {
 934         /* interlaced direct mode decoding tables */
 935         for (i = 0; i < 2; i++) {
 936             int j, k;
 937             for (j = 0; j < 2; j++) {
 938                 for (k = 0; k < 2; k++) {
 939                     FF_ALLOCZ_OR_GOTO(s->avctx,
 940                                       s->b_field_mv_table_base[i][j][k],
 941                                       mv_table_size * 2 * sizeof(int16_t),
 942                                       fail);
 943                     s->b_field_mv_table[i][j][k] = s->b_field_mv_table_base[i][j][k] +
 944                                                    s->mb_stride + 1;
 945                 }
 946                 FF_ALLOCZ_OR_GOTO(s->avctx, s->b_field_select_table [i][j], mb_array_size * 2 * sizeof(uint8_t), fail)
 947                 FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_mv_table_base[i][j], mv_table_size * 2 * sizeof(int16_t), fail)
 948                 s->p_field_mv_table[i][j] = s->p_field_mv_table_base[i][j] + s->mb_stride + 1;
 949             }
 950             FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_select_table[i], mb_array_size * 2 * sizeof(uint8_t), fail)
 951         }
 952     }
 953     if (s->out_format == FMT_H263) {
 954         /* cbp values */
 955         FF_ALLOCZ_OR_GOTO(s->avctx, s->coded_block_base, y_size, fail);
 956         s->coded_block = s->coded_block_base + s->b8_stride + 1;
 957
 958         /* cbp, ac_pred, pred_dir */
 959         FF_ALLOCZ_OR_GOTO(s->avctx, s->cbp_table     , mb_array_size * sizeof(uint8_t), fail);
 960         FF_ALLOCZ_OR_GOTO(s->avctx, s->pred_dir_table, mb_array_size * sizeof(uint8_t), fail);
 961     }
 962
 963     if (s->h263_pred || s->h263_plus || !s->encoding) {
 964         /* dc values */
 965         // MN: we need these for  error resilience of intra-frames
 966         FF_ALLOCZ_OR_GOTO(s->avctx, s->dc_val_base, yc_size * sizeof(int16_t), fail);
 967         s->dc_val[0] = s->dc_val_base + s->b8_stride + 1;
 968         s->dc_val[1] = s->dc_val_base + y_size + s->mb_stride + 1;
 969         s->dc_val[2] = s->dc_val[1] + c_size;
 970         for (i = 0; i < yc_size; i++)
 971             s->dc_val_base[i] = 1024;
 972     }
 973
 974     /* which mb is a intra block */
 975     FF_ALLOCZ_OR_GOTO(s->avctx, s->mbintra_table, mb_array_size, fail);
 976     memset(s->mbintra_table, 1, mb_array_size);
 977
 978     /* init macroblock skip table */
 979     FF_ALLOCZ_OR_GOTO(s->avctx, s->mbskip_table, mb_array_size + 2, fail);
 980     // Note the + 1 is for  a quicker mpeg4 slice_end detection
 981
 982     return init_er(s);
 983 fail:
 984     return AVERROR(ENOMEM);
 985 }
 986
 987 /**
 988  * init common structure for both encoder and decoder.
 989  * this assumes that some variables like width/height are already set
 990  */
 991 av_cold int ff_MPV_common_init(MpegEncContext *s)
 992 {
 993     int i;
 994     int nb_slices = (HAVE_THREADS &&
 995                      s->avctx->active_thread_type & FF_THREAD_SLICE) ?
 996                     s->avctx->thread_count : 1;
 997
 998     if (s->encoding && s->avctx->slices)
 999         nb_slices = s->avctx->slices;
1000
1001     if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO && !s->progressive_sequence)
1002         s->mb_height = (s->height + 31) / 32 * 2;
1003     else
1004         s->mb_height = (s->height + 15) / 16;
1005
1006     if (s->avctx->pix_fmt == AV_PIX_FMT_NONE) {
1007         av_log(s->avctx, AV_LOG_ERROR,
1008                "decoding to AV_PIX_FMT_NONE is not supported.\n");
1009         return -1;
1010     }
1011
1012     if (nb_slices > MAX_THREADS || (nb_slices > s->mb_height && s->mb_height)) {
1013         int max_slices;
1014         if (s->mb_height)
1015             max_slices = FFMIN(MAX_THREADS, s->mb_height);
1016         else
1017             max_slices = MAX_THREADS;
1018         av_log(s->avctx, AV_LOG_WARNING, "too many threads/slices (%d),"
1019                " reducing to %d\n", nb_slices, max_slices);
1020         nb_slices = max_slices;
1021     }
1022
1023     if ((s->width || s->height) &&
1024         av_image_check_size(s->width, s->height, 0, s->avctx))
1025         return -1;
1026
1027     ff_dct_common_init(s);
1028
1029     s->flags  = s->avctx->flags;
1030     s->flags2 = s->avctx->flags2;
1031
1032     /* set chroma shifts */
1033     avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &s->chroma_x_shift, &s->chroma_y_shift);
1034
1035     /* convert fourcc to upper case */
1036     s->codec_tag        = avpriv_toupper4(s->avctx->codec_tag);
1037     s->stream_codec_tag = avpriv_toupper4(s->avctx->stream_codec_tag);
1038
1039     s->avctx->coded_frame = &s->current_picture.f;
1040
1041     if (s->encoding) {
1042         if (s->msmpeg4_version) {
1043             FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_stats,
1044                                 2 * 2 * (MAX_LEVEL + 1) *
1045                                 (MAX_RUN + 1) * 2 * sizeof(int), fail);
1046         }
1047         FF_ALLOCZ_OR_GOTO(s->avctx, s->avctx->stats_out, 256, fail);
1048
1049         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix,          64 * 32   * sizeof(int), fail)
1050         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix,   64 * 32   * sizeof(int), fail)
1051         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix,          64 * 32   * sizeof(int), fail)
1052         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix16,        64 * 32 * 2 * sizeof(uint16_t), fail)
1053         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix16, 64 * 32 * 2 * sizeof(uint16_t), fail)
1054         FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix16,        64 * 32 * 2 * sizeof(uint16_t), fail)
1055         FF_ALLOCZ_OR_GOTO(s->avctx, s->input_picture,           MAX_PICTURE_COUNT * sizeof(Picture *), fail)
1056         FF_ALLOCZ_OR_GOTO(s->avctx, s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture *), fail)
1057
1058         if (s->avctx->noise_reduction) {
1059             FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_offset, 2 * 64 * sizeof(uint16_t), fail);
1060         }
1061     }
1062
1063     FF_ALLOCZ_OR_GOTO(s->avctx, s->picture,
1064                       MAX_PICTURE_COUNT * sizeof(Picture), fail);
1065     for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1066         avcodec_get_frame_defaults(&s->picture[i].f);
1067     }
1068     memset(&s->next_picture, 0, sizeof(s->next_picture));
1069     memset(&s->last_picture, 0, sizeof(s->last_picture));
1070     memset(&s->current_picture, 0, sizeof(s->current_picture));
1071     avcodec_get_frame_defaults(&s->next_picture.f);
1072     avcodec_get_frame_defaults(&s->last_picture.f);
1073     avcodec_get_frame_defaults(&s->current_picture.f);
1074
1075         if (init_context_frame(s))
1076             goto fail;
1077
1078         s->parse_context.state = -1;
1079
1080         s->context_initialized = 1;
1081         s->thread_context[0]   = s;
1082
1083 //     if (s->width && s->height) {
1084         if (nb_slices > 1) {
1085             for (i = 1; i < nb_slices; i++) {
1086                 s->thread_context[i] = av_malloc(sizeof(MpegEncContext));
1087                 memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
1088             }
1089
1090             for (i = 0; i < nb_slices; i++) {
1091                 if (init_duplicate_context(s->thread_context[i]) < 0)
1092                     goto fail;
1093                     s->thread_context[i]->start_mb_y =
1094                         (s->mb_height * (i) + nb_slices / 2) / nb_slices;
1095                     s->thread_context[i]->end_mb_y   =
1096                         (s->mb_height * (i + 1) + nb_slices / 2) / nb_slices;
1097             }
1098         } else {
1099             if (init_duplicate_context(s) < 0)
1100                 goto fail;
1101             s->start_mb_y = 0;
1102             s->end_mb_y   = s->mb_height;
1103         }
1104         s->slice_context_count = nb_slices;
1105 //     }
1106
1107     return 0;
1108  fail:
1109     ff_MPV_common_end(s);
1110     return -1;
1111 }
1112
1113 /**
1114  * Frees and resets MpegEncContext fields depending on the resolution.
1115  * Is used during resolution changes to avoid a full reinitialization of the
1116  * codec.
1117  */
1118 static int free_context_frame(MpegEncContext *s)
1119 {
1120     int i, j, k;
1121
1122     av_freep(&s->mb_type);
1123     av_freep(&s->p_mv_table_base);
1124     av_freep(&s->b_forw_mv_table_base);
1125     av_freep(&s->b_back_mv_table_base);
1126     av_freep(&s->b_bidir_forw_mv_table_base);
1127     av_freep(&s->b_bidir_back_mv_table_base);
1128     av_freep(&s->b_direct_mv_table_base);
1129     s->p_mv_table            = NULL;
1130     s->b_forw_mv_table       = NULL;
1131     s->b_back_mv_table       = NULL;
1132     s->b_bidir_forw_mv_table = NULL;
1133     s->b_bidir_back_mv_table = NULL;
1134     s->b_direct_mv_table     = NULL;
1135     for (i = 0; i < 2; i++) {
1136         for (j = 0; j < 2; j++) {
1137             for (k = 0; k < 2; k++) {
1138                 av_freep(&s->b_field_mv_table_base[i][j][k]);
1139                 s->b_field_mv_table[i][j][k] = NULL;
1140             }
1141             av_freep(&s->b_field_select_table[i][j]);
1142             av_freep(&s->p_field_mv_table_base[i][j]);
1143             s->p_field_mv_table[i][j] = NULL;
1144         }
1145         av_freep(&s->p_field_select_table[i]);
1146     }
1147
1148     av_freep(&s->dc_val_base);
1149     av_freep(&s->coded_block_base);
1150     av_freep(&s->mbintra_table);
1151     av_freep(&s->cbp_table);
1152     av_freep(&s->pred_dir_table);
1153
1154     av_freep(&s->mbskip_table);
1155
1156     av_freep(&s->er.error_status_table);
1157     av_freep(&s->er.er_temp_buffer);
1158     av_freep(&s->mb_index2xy);
1159     av_freep(&s->lambda_table);
1160
1161     av_freep(&s->cplx_tab);
1162     av_freep(&s->bits_tab);
1163
1164     s->linesize = s->uvlinesize = 0;
1165
1166     return 0;
1167 }
1168
1169 int ff_MPV_common_frame_size_change(MpegEncContext *s)
1170 {
1171     int i, err = 0;
1172
1173     if (s->slice_context_count > 1) {
1174         for (i = 0; i < s->slice_context_count; i++) {
1175             free_duplicate_context(s->thread_context[i]);
1176         }
1177         for (i = 1; i < s->slice_context_count; i++) {
1178             av_freep(&s->thread_context[i]);
1179         }
1180     } else
1181         free_duplicate_context(s);
1182
1183     if ((err = free_context_frame(s)) < 0)
1184         return err;
1185
1186     if (s->picture)
1187         for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1188                 s->picture[i].needs_realloc = 1;
1189         }
1190
1191     s->last_picture_ptr         =
1192     s->next_picture_ptr         =
1193     s->current_picture_ptr      = NULL;
1194
1195     // init
1196     if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO && !s->progressive_sequence)
1197         s->mb_height = (s->height + 31) / 32 * 2;
1198     else
1199         s->mb_height = (s->height + 15) / 16;
1200
1201     if ((s->width || s->height) &&
1202         av_image_check_size(s->width, s->height, 0, s->avctx))
1203         return AVERROR_INVALIDDATA;
1204
1205     if ((err = init_context_frame(s)))
1206         goto fail;
1207
1208     s->thread_context[0]   = s;
1209
1210     if (s->width && s->height) {
1211         int nb_slices = s->slice_context_count;
1212         if (nb_slices > 1) {
1213             for (i = 1; i < nb_slices; i++) {
1214                 s->thread_context[i] = av_malloc(sizeof(MpegEncContext));
1215                 memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
1216             }
1217
1218             for (i = 0; i < nb_slices; i++) {
1219                 if (init_duplicate_context(s->thread_context[i]) < 0)
1220                     goto fail;
1221                     s->thread_context[i]->start_mb_y =
1222                         (s->mb_height * (i) + nb_slices / 2) / nb_slices;
1223                     s->thread_context[i]->end_mb_y   =
1224                         (s->mb_height * (i + 1) + nb_slices / 2) / nb_slices;
1225             }
1226         } else {
1227             err = init_duplicate_context(s);
1228             if (err < 0)
1229                 goto fail;
1230             s->start_mb_y = 0;
1231             s->end_mb_y   = s->mb_height;
1232         }
1233         s->slice_context_count = nb_slices;
1234     }
1235
1236     return 0;
1237  fail:
1238     ff_MPV_common_end(s);
1239     return err;
1240 }
1241
1242 /* init common structure for both encoder and decoder */
1243 void ff_MPV_common_end(MpegEncContext *s)
1244 {
1245     int i;
1246
1247     if (s->slice_context_count > 1) {
1248         for (i = 0; i < s->slice_context_count; i++) {
1249             free_duplicate_context(s->thread_context[i]);
1250         }
1251         for (i = 1; i < s->slice_context_count; i++) {
1252             av_freep(&s->thread_context[i]);
1253         }
1254         s->slice_context_count = 1;
1255     } else free_duplicate_context(s);
1256
1257     av_freep(&s->parse_context.buffer);
1258     s->parse_context.buffer_size = 0;
1259
1260     av_freep(&s->bitstream_buffer);
1261     s->allocated_bitstream_buffer_size = 0;
1262
1263     av_freep(&s->avctx->stats_out);
1264     av_freep(&s->ac_stats);
1265
1266     if(s->q_chroma_intra_matrix   != s->q_intra_matrix  ) av_freep(&s->q_chroma_intra_matrix);
1267     if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16);
1268     s->q_chroma_intra_matrix=   NULL;
1269     s->q_chroma_intra_matrix16= NULL;
1270     av_freep(&s->q_intra_matrix);
1271     av_freep(&s->q_inter_matrix);
1272     av_freep(&s->q_intra_matrix16);
1273     av_freep(&s->q_inter_matrix16);
1274     av_freep(&s->input_picture);
1275     av_freep(&s->reordered_input_picture);
1276     av_freep(&s->dct_offset);
1277
1278     if (s->picture) {
1279         for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1280             free_picture_tables(&s->picture[i]);
1281             ff_mpeg_unref_picture(s, &s->picture[i]);
1282         }
1283     }
1284     av_freep(&s->picture);
1285     free_picture_tables(&s->last_picture);
1286     ff_mpeg_unref_picture(s, &s->last_picture);
1287     free_picture_tables(&s->current_picture);
1288     ff_mpeg_unref_picture(s, &s->current_picture);
1289     free_picture_tables(&s->next_picture);
1290     ff_mpeg_unref_picture(s, &s->next_picture);
1291     free_picture_tables(&s->new_picture);
1292     ff_mpeg_unref_picture(s, &s->new_picture);
1293
1294     free_context_frame(s);
1295
1296     s->context_initialized      = 0;
1297     s->last_picture_ptr         =
1298     s->next_picture_ptr         =
1299     s->current_picture_ptr      = NULL;
1300     s->linesize = s->uvlinesize = 0;
1301 }
1302
1303 av_cold void ff_init_rl(RLTable *rl,
1304                         uint8_t static_store[2][2 * MAX_RUN + MAX_LEVEL + 3])
1305 {
1306     int8_t  max_level[MAX_RUN + 1], max_run[MAX_LEVEL + 1];
1307     uint8_t index_run[MAX_RUN + 1];
1308     int last, run, level, start, end, i;
1309
1310     /* If table is static, we can quit if rl->max_level[0] is not NULL */
1311     if (static_store && rl->max_level[0])
1312         return;
1313
1314     /* compute max_level[], max_run[] and index_run[] */
1315     for (last = 0; last < 2; last++) {
1316         if (last == 0) {
1317             start = 0;
1318             end = rl->last;
1319         } else {
1320             start = rl->last;
1321             end = rl->n;
1322         }
1323
1324         memset(max_level, 0, MAX_RUN + 1);
1325         memset(max_run, 0, MAX_LEVEL + 1);
1326         memset(index_run, rl->n, MAX_RUN + 1);
1327         for (i = start; i < end; i++) {
1328             run   = rl->table_run[i];
1329             level = rl->table_level[i];
1330             if (index_run[run] == rl->n)
1331                 index_run[run] = i;
1332             if (level > max_level[run])
1333                 max_level[run] = level;
1334             if (run > max_run[level])
1335                 max_run[level] = run;
1336         }
1337         if (static_store)
1338             rl->max_level[last] = static_store[last];
1339         else
1340             rl->max_level[last] = av_malloc(MAX_RUN + 1);
1341         memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
1342         if (static_store)
1343             rl->max_run[last]   = static_store[last] + MAX_RUN + 1;
1344         else
1345             rl->max_run[last]   = av_malloc(MAX_LEVEL + 1);
1346         memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
1347         if (static_store)
1348             rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
1349         else
1350             rl->index_run[last] = av_malloc(MAX_RUN + 1);
1351         memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
1352     }
1353 }
1354
1355 av_cold void ff_init_vlc_rl(RLTable *rl)
1356 {
1357     int i, q;
1358
1359     for (q = 0; q < 32; q++) {
1360         int qmul = q * 2;
1361         int qadd = (q - 1) | 1;
1362
1363         if (q == 0) {
1364             qmul = 1;
1365             qadd = 0;
1366         }
1367         for (i = 0; i < rl->vlc.table_size; i++) {
1368             int code = rl->vlc.table[i][0];
1369             int len  = rl->vlc.table[i][1];
1370             int level, run;
1371
1372             if (len == 0) { // illegal code
1373                 run   = 66;
1374                 level = MAX_LEVEL;
1375             } else if (len < 0) { // more bits needed
1376                 run   = 0;
1377                 level = code;
1378             } else {
1379                 if (code == rl->n) { // esc
1380                     run   = 66;
1381                     level =  0;
1382                 } else {
1383                     run   = rl->table_run[code] + 1;
1384                     level = rl->table_level[code] * qmul + qadd;
1385                     if (code >= rl->last) run += 192;
1386                 }
1387             }
1388             rl->rl_vlc[q][i].len   = len;
1389             rl->rl_vlc[q][i].level = level;
1390             rl->rl_vlc[q][i].run   = run;
1391         }
1392     }
1393 }
1394
1395 void ff_release_unused_pictures(MpegEncContext*s, int remove_current)
1396 {
1397     int i;
1398
1399     /* release non reference frames */
1400     for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1401         if (!s->picture[i].reference &&
1402             (remove_current || &s->picture[i] !=  s->current_picture_ptr)) {
1403             ff_mpeg_unref_picture(s, &s->picture[i]);
1404         }
1405     }
1406 }
1407
1408 static inline int pic_is_unused(MpegEncContext *s, Picture *pic)
1409 {
1410     if (pic == s->last_picture_ptr)
1411         return 0;
1412     if (pic->f.data[0] == NULL)
1413         return 1;
1414     if (pic->needs_realloc && !(pic->reference & DELAYED_PIC_REF))
1415         return 1;
1416     return 0;
1417 }
1418
1419 static int find_unused_picture(MpegEncContext *s, int shared)
1420 {
1421     int i;
1422
1423     if (shared) {
1424         for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1425             if (s->picture[i].f.data[0] == NULL && &s->picture[i] != s->last_picture_ptr)
1426                 return i;
1427         }
1428     } else {
1429         for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1430             if (pic_is_unused(s, &s->picture[i]))
1431                 return i;
1432         }
1433     }
1434
1435     av_log(s->avctx, AV_LOG_FATAL,
1436            "Internal error, picture buffer overflow\n");
1437     /* We could return -1, but the codec would crash trying to draw into a
1438      * non-existing frame anyway. This is safer than waiting for a random crash.
1439      * Also the return of this is never useful, an encoder must only allocate
1440      * as much as allowed in the specification. This has no relationship to how
1441      * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
1442      * enough for such valid streams).
1443      * Plus, a decoder has to check stream validity and remove frames if too
1444      * many reference frames are around. Waiting for "OOM" is not correct at
1445      * all. Similarly, missing reference frames have to be replaced by
1446      * interpolated/MC frames, anything else is a bug in the codec ...
1447      */
1448     abort();
1449     return -1;
1450 }
1451
1452 int ff_find_unused_picture(MpegEncContext *s, int shared)
1453 {
1454     int ret = find_unused_picture(s, shared);
1455
1456     if (ret >= 0 && ret < MAX_PICTURE_COUNT) {
1457         if (s->picture[ret].needs_realloc) {
1458             s->picture[ret].needs_realloc = 0;
1459             free_picture_tables(&s->picture[ret]);
1460             ff_mpeg_unref_picture(s, &s->picture[ret]);
1461             avcodec_get_frame_defaults(&s->picture[ret].f);
1462         }
1463     }
1464     return ret;
1465 }
1466
1467 static void update_noise_reduction(MpegEncContext *s)
1468 {
1469     int intra, i;
1470
1471     for (intra = 0; intra < 2; intra++) {
1472         if (s->dct_count[intra] > (1 << 16)) {
1473             for (i = 0; i < 64; i++) {
1474                 s->dct_error_sum[intra][i] >>= 1;
1475             }
1476             s->dct_count[intra] >>= 1;
1477         }
1478
1479         for (i = 0; i < 64; i++) {
1480             s->dct_offset[intra][i] = (s->avctx->noise_reduction *
1481                                        s->dct_count[intra] +
1482                                        s->dct_error_sum[intra][i] / 2) /
1483                                       (s->dct_error_sum[intra][i] + 1);
1484         }
1485     }
1486 }
1487
1488 /**
1489  * generic function for encode/decode called after coding/decoding
1490  * the header and before a frame is coded/decoded.
1491  */
1492 int ff_MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
1493 {
1494     int i, ret;
1495     Picture *pic;
1496     s->mb_skipped = 0;
1497
1498     if (!ff_thread_can_start_frame(avctx)) {
1499         av_log(avctx, AV_LOG_ERROR, "Attempt to start a frame outside SETUP state\n");
1500         return -1;
1501     }
1502
1503     /* mark & release old frames */
1504     if (s->pict_type != AV_PICTURE_TYPE_B && s->last_picture_ptr &&
1505         s->last_picture_ptr != s->next_picture_ptr &&
1506         s->last_picture_ptr->f.data[0]) {
1507         ff_mpeg_unref_picture(s, s->last_picture_ptr);
1508     }
1509
1510     /* release forgotten pictures */
1511     /* if (mpeg124/h263) */
1512     if (!s->encoding) {
1513         for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1514             if (&s->picture[i] != s->last_picture_ptr &&
1515                 &s->picture[i] != s->next_picture_ptr &&
1516                 s->picture[i].reference && !s->picture[i].needs_realloc) {
1517                 if (!(avctx->active_thread_type & FF_THREAD_FRAME))
1518                     av_log(avctx, AV_LOG_ERROR,
1519                            "releasing zombie picture\n");
1520                 ff_mpeg_unref_picture(s, &s->picture[i]);
1521             }
1522         }
1523     }
1524
1525     ff_mpeg_unref_picture(s, &s->current_picture);
1526
1527     if (!s->encoding) {
1528         ff_release_unused_pictures(s, 1);
1529
1530         if (s->current_picture_ptr &&
1531             s->current_picture_ptr->f.data[0] == NULL) {
1532             // we already have a unused image
1533             // (maybe it was set before reading the header)
1534             pic = s->current_picture_ptr;
1535         } else {
1536             i   = ff_find_unused_picture(s, 0);
1537             if (i < 0) {
1538                 av_log(s->avctx, AV_LOG_ERROR, "no frame buffer available\n");
1539                 return i;
1540             }
1541             pic = &s->picture[i];
1542         }
1543
1544         pic->reference = 0;
1545         if (!s->droppable) {
1546             if (s->pict_type != AV_PICTURE_TYPE_B)
1547                 pic->reference = 3;
1548         }
1549
1550         pic->f.coded_picture_number = s->coded_picture_number++;
1551
1552         if (ff_alloc_picture(s, pic, 0) < 0)
1553             return -1;
1554
1555         s->current_picture_ptr = pic;
1556         // FIXME use only the vars from current_pic
1557         s->current_picture_ptr->f.top_field_first = s->top_field_first;
1558         if (s->codec_id == AV_CODEC_ID_MPEG1VIDEO ||
1559             s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
1560             if (s->picture_structure != PICT_FRAME)
1561                 s->current_picture_ptr->f.top_field_first =
1562                     (s->picture_structure == PICT_TOP_FIELD) == s->first_field;
1563         }
1564         s->current_picture_ptr->f.interlaced_frame = !s->progressive_frame &&
1565                                                      !s->progressive_sequence;
1566         s->current_picture_ptr->field_picture      =  s->picture_structure != PICT_FRAME;
1567     }
1568
1569     s->current_picture_ptr->f.pict_type = s->pict_type;
1570     // if (s->flags && CODEC_FLAG_QSCALE)
1571     //     s->current_picture_ptr->quality = s->new_picture_ptr->quality;
1572     s->current_picture_ptr->f.key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1573
1574     if ((ret = ff_mpeg_ref_picture(s, &s->current_picture,
1575                                    s->current_picture_ptr)) < 0)
1576         return ret;
1577
1578     if (s->pict_type != AV_PICTURE_TYPE_B) {
1579         s->last_picture_ptr = s->next_picture_ptr;
1580         if (!s->droppable)
1581             s->next_picture_ptr = s->current_picture_ptr;
1582     }
1583     av_dlog(s->avctx, "L%p N%p C%p L%p N%p C%p type:%d drop:%d\n",
1584             s->last_picture_ptr, s->next_picture_ptr,s->current_picture_ptr,
1585             s->last_picture_ptr    ? s->last_picture_ptr->f.data[0]    : NULL,
1586             s->next_picture_ptr    ? s->next_picture_ptr->f.data[0]    : NULL,
1587             s->current_picture_ptr ? s->current_picture_ptr->f.data[0] : NULL,
1588             s->pict_type, s->droppable);
1589
1590     if ((s->last_picture_ptr == NULL ||
1591          s->last_picture_ptr->f.data[0] == NULL) &&
1592         (s->pict_type != AV_PICTURE_TYPE_I ||
1593          s->picture_structure != PICT_FRAME)) {
1594         int h_chroma_shift, v_chroma_shift;
1595         av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt,
1596                                          &h_chroma_shift, &v_chroma_shift);
1597         if (s->pict_type != AV_PICTURE_TYPE_I)
1598             av_log(avctx, AV_LOG_ERROR,
1599                    "warning: first frame is no keyframe\n");
1600         else if (s->picture_structure != PICT_FRAME)
1601             av_log(avctx, AV_LOG_INFO,
1602                    "allocate dummy last picture for field based first keyframe\n");
1603
1604         /* Allocate a dummy frame */
1605         i = ff_find_unused_picture(s, 0);
1606         if (i < 0) {
1607             av_log(s->avctx, AV_LOG_ERROR, "no frame buffer available\n");
1608             return i;
1609         }
1610         s->last_picture_ptr = &s->picture[i];
1611         s->last_picture_ptr->f.key_frame = 0;
1612         if (ff_alloc_picture(s, s->last_picture_ptr, 0) < 0) {
1613             s->last_picture_ptr = NULL;
1614             return -1;
1615         }
1616
1617         memset(s->last_picture_ptr->f.data[0], 0x80,
1618                avctx->height * s->last_picture_ptr->f.linesize[0]);
1619         memset(s->last_picture_ptr->f.data[1], 0x80,
1620                (avctx->height >> v_chroma_shift) *
1621                s->last_picture_ptr->f.linesize[1]);
1622         memset(s->last_picture_ptr->f.data[2], 0x80,
1623                (avctx->height >> v_chroma_shift) *
1624                s->last_picture_ptr->f.linesize[2]);
1625
1626         if(s->codec_id == AV_CODEC_ID_FLV1 || s->codec_id == AV_CODEC_ID_H263){
1627             for(i=0; i<avctx->height; i++)
1628             memset(s->last_picture_ptr->f.data[0] + s->last_picture_ptr->f.linesize[0]*i, 16, avctx->width);
1629         }
1630
1631         ff_thread_report_progress(&s->last_picture_ptr->tf, INT_MAX, 0);
1632         ff_thread_report_progress(&s->last_picture_ptr->tf, INT_MAX, 1);
1633     }
1634     if ((s->next_picture_ptr == NULL ||
1635          s->next_picture_ptr->f.data[0] == NULL) &&
1636         s->pict_type == AV_PICTURE_TYPE_B) {
1637         /* Allocate a dummy frame */
1638         i = ff_find_unused_picture(s, 0);
1639         if (i < 0) {
1640             av_log(s->avctx, AV_LOG_ERROR, "no frame buffer available\n");
1641             return i;
1642         }
1643         s->next_picture_ptr = &s->picture[i];
1644         s->next_picture_ptr->f.key_frame = 0;
1645         if (ff_alloc_picture(s, s->next_picture_ptr, 0) < 0) {
1646             s->next_picture_ptr = NULL;
1647             return -1;
1648         }
1649         ff_thread_report_progress(&s->next_picture_ptr->tf, INT_MAX, 0);
1650         ff_thread_report_progress(&s->next_picture_ptr->tf, INT_MAX, 1);
1651     }
1652
1653 #if 0 // BUFREF-FIXME
1654     memset(s->last_picture.f.data, 0, sizeof(s->last_picture.f.data));
1655     memset(s->next_picture.f.data, 0, sizeof(s->next_picture.f.data));
1656 #endif
1657     if (s->last_picture_ptr) {
1658         ff_mpeg_unref_picture(s, &s->last_picture);
1659         if (s->last_picture_ptr->f.data[0] &&
1660             (ret = ff_mpeg_ref_picture(s, &s->last_picture,
1661                                        s->last_picture_ptr)) < 0)
1662             return ret;
1663     }
1664     if (s->next_picture_ptr) {
1665         ff_mpeg_unref_picture(s, &s->next_picture);
1666         if (s->next_picture_ptr->f.data[0] &&
1667             (ret = ff_mpeg_ref_picture(s, &s->next_picture,
1668                                        s->next_picture_ptr)) < 0)
1669             return ret;
1670     }
1671
1672     av_assert0(s->pict_type == AV_PICTURE_TYPE_I || (s->last_picture_ptr &&
1673                                                  s->last_picture_ptr->f.data[0]));
1674
1675     if (s->picture_structure!= PICT_FRAME) {
1676         int i;
1677         for (i = 0; i < 4; i++) {
1678             if (s->picture_structure == PICT_BOTTOM_FIELD) {
1679                 s->current_picture.f.data[i] +=
1680                     s->current_picture.f.linesize[i];
1681             }
1682             s->current_picture.f.linesize[i] *= 2;
1683             s->last_picture.f.linesize[i]    *= 2;
1684             s->next_picture.f.linesize[i]    *= 2;
1685         }
1686     }
1687
1688     s->err_recognition = avctx->err_recognition;
1689
1690     /* set dequantizer, we can't do it during init as
1691      * it might change for mpeg4 and we can't do it in the header
1692      * decode as init is not called for mpeg4 there yet */
1693     if (s->mpeg_quant || s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
1694         s->dct_unquantize_intra = s->dct_unquantize_mpeg2_intra;
1695         s->dct_unquantize_inter = s->dct_unquantize_mpeg2_inter;
1696     } else if (s->out_format == FMT_H263 || s->out_format == FMT_H261) {
1697         s->dct_unquantize_intra = s->dct_unquantize_h263_intra;
1698         s->dct_unquantize_inter = s->dct_unquantize_h263_inter;
1699     } else {
1700         s->dct_unquantize_intra = s->dct_unquantize_mpeg1_intra;
1701         s->dct_unquantize_inter = s->dct_unquantize_mpeg1_inter;
1702     }
1703
1704     if (s->dct_error_sum) {
1705         av_assert2(s->avctx->noise_reduction && s->encoding);
1706         update_noise_reduction(s);
1707     }
1708
1709     if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration)
1710         return ff_xvmc_field_start(s, avctx);
1711
1712     return 0;
1713 }
1714
1715 /* generic function for encode/decode called after a
1716  * frame has been coded/decoded. */
1717 void ff_MPV_frame_end(MpegEncContext *s)
1718 {
1719     /* redraw edges for the frame if decoding didn't complete */
1720     // just to make sure that all data is rendered.
1721     if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration) {
1722         ff_xvmc_field_end(s);
1723    } else if ((s->er.error_count || s->encoding || !(s->avctx->codec->capabilities&CODEC_CAP_DRAW_HORIZ_BAND)) &&
1724               !s->avctx->hwaccel &&
1725               s->unrestricted_mv &&
1726               s->current_picture.reference &&
1727               !s->intra_only &&
1728               !(s->flags & CODEC_FLAG_EMU_EDGE) &&
1729               !s->avctx->lowres
1730             ) {
1731         const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
1732         int hshift = desc->log2_chroma_w;
1733         int vshift = desc->log2_chroma_h;
1734         s->dsp.draw_edges(s->current_picture.f.data[0], s->current_picture.f.linesize[0],
1735                           s->h_edge_pos, s->v_edge_pos,
1736                           EDGE_WIDTH, EDGE_WIDTH,
1737                           EDGE_TOP | EDGE_BOTTOM);
1738         s->dsp.draw_edges(s->current_picture.f.data[1], s->current_picture.f.linesize[1],
1739                           s->h_edge_pos >> hshift, s->v_edge_pos >> vshift,
1740                           EDGE_WIDTH >> hshift, EDGE_WIDTH >> vshift,
1741                           EDGE_TOP | EDGE_BOTTOM);
1742         s->dsp.draw_edges(s->current_picture.f.data[2], s->current_picture.f.linesize[2],
1743                           s->h_edge_pos >> hshift, s->v_edge_pos >> vshift,
1744                           EDGE_WIDTH >> hshift, EDGE_WIDTH >> vshift,
1745                           EDGE_TOP | EDGE_BOTTOM);
1746     }
1747
1748     emms_c();
1749
1750     s->last_pict_type                 = s->pict_type;
1751     s->last_lambda_for [s->pict_type] = s->current_picture_ptr->f.quality;
1752     if (s->pict_type!= AV_PICTURE_TYPE_B) {
1753         s->last_non_b_pict_type = s->pict_type;
1754     }
1755 #if 0
1756     /* copy back current_picture variables */
1757     for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1758         if (s->picture[i].f.data[0] == s->current_picture.f.data[0]) {
1759             s->picture[i] = s->current_picture;
1760             break;
1761         }
1762     }
1763     av_assert0(i < MAX_PICTURE_COUNT);
1764 #endif
1765
1766     // clear copies, to avoid confusion
1767 #if 0
1768     memset(&s->last_picture,    0, sizeof(Picture));
1769     memset(&s->next_picture,    0, sizeof(Picture));
1770     memset(&s->current_picture, 0, sizeof(Picture));
1771 #endif
1772     s->avctx->coded_frame = &s->current_picture_ptr->f;
1773
1774     if (s->current_picture.reference)
1775         ff_thread_report_progress(&s->current_picture_ptr->tf, INT_MAX, 0);
1776 }
1777
1778 /**
1779  * Draw a line from (ex, ey) -> (sx, sy).
1780  * @param w width of the image
1781  * @param h height of the image
1782  * @param stride stride/linesize of the image
1783  * @param color color of the arrow
1784  */
1785 static void draw_line(uint8_t *buf, int sx, int sy, int ex, int ey,
1786                       int w, int h, int stride, int color)
1787 {
1788     int x, y, fr, f;
1789
1790     sx = av_clip(sx, 0, w - 1);
1791     sy = av_clip(sy, 0, h - 1);
1792     ex = av_clip(ex, 0, w - 1);
1793     ey = av_clip(ey, 0, h - 1);
1794
1795     buf[sy * stride + sx] += color;
1796
1797     if (FFABS(ex - sx) > FFABS(ey - sy)) {
1798         if (sx > ex) {
1799             FFSWAP(int, sx, ex);
1800             FFSWAP(int, sy, ey);
1801         }
1802         buf += sx + sy * stride;
1803         ex  -= sx;
1804         f    = ((ey - sy) << 16) / ex;
1805         for (x = 0; x <= ex; x++) {
1806             y  = (x * f) >> 16;
1807             fr = (x * f) & 0xFFFF;
1808             buf[y * stride + x]       += (color * (0x10000 - fr)) >> 16;
1809             if(fr) buf[(y + 1) * stride + x] += (color *            fr ) >> 16;
1810         }
1811     } else {
1812         if (sy > ey) {
1813             FFSWAP(int, sx, ex);
1814             FFSWAP(int, sy, ey);
1815         }
1816         buf += sx + sy * stride;
1817         ey  -= sy;
1818         if (ey)
1819             f = ((ex - sx) << 16) / ey;
1820         else
1821             f = 0;
1822         for(y= 0; y <= ey; y++){
1823             x  = (y*f) >> 16;
1824             fr = (y*f) & 0xFFFF;
1825             buf[y * stride + x]     += (color * (0x10000 - fr)) >> 16;
1826             if(fr) buf[y * stride + x + 1] += (color *            fr ) >> 16;
1827         }
1828     }
1829 }
1830
1831 /**
1832  * Draw an arrow from (ex, ey) -> (sx, sy).
1833  * @param w width of the image
1834  * @param h height of the image
1835  * @param stride stride/linesize of the image
1836  * @param color color of the arrow
1837  */
1838 static void draw_arrow(uint8_t *buf, int sx, int sy, int ex,
1839                        int ey, int w, int h, int stride, int color)
1840 {
1841     int dx,dy;
1842
1843     sx = av_clip(sx, -100, w + 100);
1844     sy = av_clip(sy, -100, h + 100);
1845     ex = av_clip(ex, -100, w + 100);
1846     ey = av_clip(ey, -100, h + 100);
1847
1848     dx = ex - sx;
1849     dy = ey - sy;
1850
1851     if (dx * dx + dy * dy > 3 * 3) {
1852         int rx =  dx + dy;
1853         int ry = -dx + dy;
1854         int length = ff_sqrt((rx * rx + ry * ry) << 8);
1855
1856         // FIXME subpixel accuracy
1857         rx = ROUNDED_DIV(rx * 3 << 4, length);
1858         ry = ROUNDED_DIV(ry * 3 << 4, length);
1859
1860         draw_line(buf, sx, sy, sx + rx, sy + ry, w, h, stride, color);
1861         draw_line(buf, sx, sy, sx - ry, sy + rx, w, h, stride, color);
1862     }
1863     draw_line(buf, sx, sy, ex, ey, w, h, stride, color);
1864 }
1865
1866 /**
1867  * Print debugging info for the given picture.
1868  */
1869 void ff_print_debug_info2(AVCodecContext *avctx, Picture *p, AVFrame *pict, uint8_t *mbskip_table,
1870                          int *low_delay,
1871                          int mb_width, int mb_height, int mb_stride, int quarter_sample)
1872 {
1873     if (avctx->hwaccel || !p || !p->mb_type
1874         || (avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU))
1875         return;
1876
1877
1878     if (avctx->debug & (FF_DEBUG_SKIP | FF_DEBUG_QP | FF_DEBUG_MB_TYPE)) {
1879         int x,y;
1880
1881         av_log(avctx, AV_LOG_DEBUG, "New frame, type: %c\n",
1882                av_get_picture_type_char(pict->pict_type));
1883         for (y = 0; y < mb_height; y++) {
1884             for (x = 0; x < mb_width; x++) {
1885                 if (avctx->debug & FF_DEBUG_SKIP) {
1886                     int count = mbskip_table[x + y * mb_stride];
1887                     if (count > 9)
1888                         count = 9;
1889                     av_log(avctx, AV_LOG_DEBUG, "%1d", count);
1890                 }
1891                 if (avctx->debug & FF_DEBUG_QP) {
1892                     av_log(avctx, AV_LOG_DEBUG, "%2d",
1893                            p->qscale_table[x + y * mb_stride]);
1894                 }
1895                 if (avctx->debug & FF_DEBUG_MB_TYPE) {
1896                     int mb_type = p->mb_type[x + y * mb_stride];
1897                     // Type & MV direction
1898                     if (IS_PCM(mb_type))
1899                         av_log(avctx, AV_LOG_DEBUG, "P");
1900                     else if (IS_INTRA(mb_type) && IS_ACPRED(mb_type))
1901                         av_log(avctx, AV_LOG_DEBUG, "A");
1902                     else if (IS_INTRA4x4(mb_type))
1903                         av_log(avctx, AV_LOG_DEBUG, "i");
1904                     else if (IS_INTRA16x16(mb_type))
1905                         av_log(avctx, AV_LOG_DEBUG, "I");
1906                     else if (IS_DIRECT(mb_type) && IS_SKIP(mb_type))
1907                         av_log(avctx, AV_LOG_DEBUG, "d");
1908                     else if (IS_DIRECT(mb_type))
1909                         av_log(avctx, AV_LOG_DEBUG, "D");
1910                     else if (IS_GMC(mb_type) && IS_SKIP(mb_type))
1911                         av_log(avctx, AV_LOG_DEBUG, "g");
1912                     else if (IS_GMC(mb_type))
1913                         av_log(avctx, AV_LOG_DEBUG, "G");
1914                     else if (IS_SKIP(mb_type))
1915                         av_log(avctx, AV_LOG_DEBUG, "S");
1916                     else if (!USES_LIST(mb_type, 1))
1917                         av_log(avctx, AV_LOG_DEBUG, ">");
1918                     else if (!USES_LIST(mb_type, 0))
1919                         av_log(avctx, AV_LOG_DEBUG, "<");
1920                     else {
1921                         av_assert2(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
1922                         av_log(avctx, AV_LOG_DEBUG, "X");
1923                     }
1924
1925                     // segmentation
1926                     if (IS_8X8(mb_type))
1927                         av_log(avctx, AV_LOG_DEBUG, "+");
1928                     else if (IS_16X8(mb_type))
1929                         av_log(avctx, AV_LOG_DEBUG, "-");
1930                     else if (IS_8X16(mb_type))
1931                         av_log(avctx, AV_LOG_DEBUG, "|");
1932                     else if (IS_INTRA(mb_type) || IS_16X16(mb_type))
1933                         av_log(avctx, AV_LOG_DEBUG, " ");
1934                     else
1935                         av_log(avctx, AV_LOG_DEBUG, "?");
1936
1937
1938                     if (IS_INTERLACED(mb_type))
1939                         av_log(avctx, AV_LOG_DEBUG, "=");
1940                     else
1941                         av_log(avctx, AV_LOG_DEBUG, " ");
1942                 }
1943             }
1944             av_log(avctx, AV_LOG_DEBUG, "\n");
1945         }
1946     }
1947
1948     if ((avctx->debug & (FF_DEBUG_VIS_QP | FF_DEBUG_VIS_MB_TYPE)) ||
1949         (avctx->debug_mv)) {
1950         const int shift = 1 + quarter_sample;
1951         int mb_y;
1952         uint8_t *ptr;
1953         int i;
1954         int h_chroma_shift, v_chroma_shift, block_height;
1955         const int width          = avctx->width;
1956         const int height         = avctx->height;
1957         const int mv_sample_log2 = avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_SVQ3 ? 2 : 1;
1958         const int mv_stride      = (mb_width << mv_sample_log2) +
1959                                    (avctx->codec->id == AV_CODEC_ID_H264 ? 0 : 1);
1960
1961         *low_delay = 0; // needed to see the vectors without trashing the buffers
1962
1963         avcodec_get_chroma_sub_sample(avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1964
1965         av_frame_make_writable(pict);
1966
1967         pict->opaque = NULL;
1968         ptr          = pict->data[0];
1969         block_height = 16 >> v_chroma_shift;
1970
1971         for (mb_y = 0; mb_y < mb_height; mb_y++) {
1972             int mb_x;
1973             for (mb_x = 0; mb_x < mb_width; mb_x++) {
1974                 const int mb_index = mb_x + mb_y * mb_stride;
1975                 if ((avctx->debug_mv) && p->motion_val[0]) {
1976                     int type;
1977                     for (type = 0; type < 3; type++) {
1978                         int direction = 0;
1979                         switch (type) {
1980                         case 0:
1981                             if ((!(avctx->debug_mv & FF_DEBUG_VIS_MV_P_FOR)) ||
1982                                 (pict->pict_type!= AV_PICTURE_TYPE_P))
1983                                 continue;
1984                             direction = 0;
1985                             break;
1986                         case 1:
1987                             if ((!(avctx->debug_mv & FF_DEBUG_VIS_MV_B_FOR)) ||
1988                                 (pict->pict_type!= AV_PICTURE_TYPE_B))
1989                                 continue;
1990                             direction = 0;
1991                             break;
1992                         case 2:
1993                             if ((!(avctx->debug_mv & FF_DEBUG_VIS_MV_B_BACK)) ||
1994                                 (pict->pict_type!= AV_PICTURE_TYPE_B))
1995                                 continue;
1996                             direction = 1;
1997                             break;
1998                         }
1999                         if (!USES_LIST(p->mb_type[mb_index], direction))
2000                             continue;
2001
2002                         if (IS_8X8(p->mb_type[mb_index])) {
2003                             int i;
2004                             for (i = 0; i < 4; i++) {
2005                                 int sx = mb_x * 16 + 4 + 8 * (i & 1);
2006                                 int sy = mb_y * 16 + 4 + 8 * (i >> 1);
2007                                 int xy = (mb_x * 2 + (i & 1) +
2008                                           (mb_y * 2 + (i >> 1)) * mv_stride) << (mv_sample_log2 - 1);
2009                                 int mx = (p->motion_val[direction][xy][0] >> shift) + sx;
2010                                 int my = (p->motion_val[direction][xy][1] >> shift) + sy;
2011                                 draw_arrow(ptr, sx, sy, mx, my, width,
2012                                            height, pict->linesize[0], 100);
2013                             }
2014                         } else if (IS_16X8(p->mb_type[mb_index])) {
2015                             int i;
2016                             for (i = 0; i < 2; i++) {
2017                                 int sx = mb_x * 16 + 8;
2018                                 int sy = mb_y * 16 + 4 + 8 * i;
2019                                 int xy = (mb_x * 2 + (mb_y * 2 + i) * mv_stride) << (mv_sample_log2 - 1);
2020                                 int mx = (p->motion_val[direction][xy][0] >> shift);
2021                                 int my = (p->motion_val[direction][xy][1] >> shift);
2022
2023                                 if (IS_INTERLACED(p->mb_type[mb_index]))
2024                                     my *= 2;
2025
2026                             draw_arrow(ptr, sx, sy, mx + sx, my + sy, width,
2027                                        height, pict->linesize[0], 100);
2028                             }
2029                         } else if (IS_8X16(p->mb_type[mb_index])) {
2030                             int i;
2031                             for (i = 0; i < 2; i++) {
2032                                 int sx = mb_x * 16 + 4 + 8 * i;
2033                                 int sy = mb_y * 16 + 8;
2034                                 int xy = (mb_x * 2 + i + mb_y * 2 * mv_stride) << (mv_sample_log2 - 1);
2035                                 int mx = p->motion_val[direction][xy][0] >> shift;
2036                                 int my = p->motion_val[direction][xy][1] >> shift;
2037
2038                                 if (IS_INTERLACED(p->mb_type[mb_index]))
2039                                     my *= 2;
2040
2041                                 draw_arrow(ptr, sx, sy, mx + sx, my + sy, width,
2042                                            height, pict->linesize[0], 100);
2043                             }
2044                         } else {
2045                               int sx= mb_x * 16 + 8;
2046                               int sy= mb_y * 16 + 8;
2047                               int xy= (mb_x + mb_y * mv_stride) << mv_sample_log2;
2048                               int mx= (p->motion_val[direction][xy][0]>>shift) + sx;
2049                               int my= (p->motion_val[direction][xy][1]>>shift) + sy;
2050                               draw_arrow(ptr, sx, sy, mx, my, width, height, pict->linesize[0], 100);
2051                         }
2052                     }
2053                 }
2054                 if ((avctx->debug & FF_DEBUG_VIS_QP)) {
2055                     uint64_t c = (p->qscale_table[mb_index] * 128 / 31) *
2056                                  0x0101010101010101ULL;
2057                     int y;
2058                     for (y = 0; y < block_height; y++) {
2059                         *(uint64_t *)(pict->data[1] + 8 * mb_x +
2060                                       (block_height * mb_y + y) *
2061                                       pict->linesize[1]) = c;
2062                         *(uint64_t *)(pict->data[2] + 8 * mb_x +
2063                                       (block_height * mb_y + y) *
2064                                       pict->linesize[2]) = c;
2065                     }
2066                 }
2067                 if ((avctx->debug & FF_DEBUG_VIS_MB_TYPE) &&
2068                     p->motion_val[0]) {
2069                     int mb_type = p->mb_type[mb_index];
2070                     uint64_t u,v;
2071                     int y;
2072 #define COLOR(theta, r) \
2073     u = (int)(128 + r * cos(theta * 3.141592 / 180)); \
2074     v = (int)(128 + r * sin(theta * 3.141592 / 180));
2075
2076
2077                     u = v = 128;
2078                     if (IS_PCM(mb_type)) {
2079                         COLOR(120, 48)
2080                     } else if ((IS_INTRA(mb_type) && IS_ACPRED(mb_type)) ||
2081                                IS_INTRA16x16(mb_type)) {
2082                         COLOR(30, 48)
2083                     } else if (IS_INTRA4x4(mb_type)) {
2084                         COLOR(90, 48)
2085                     } else if (IS_DIRECT(mb_type) && IS_SKIP(mb_type)) {
2086                         // COLOR(120, 48)
2087                     } else if (IS_DIRECT(mb_type)) {
2088                         COLOR(150, 48)
2089                     } else if (IS_GMC(mb_type) && IS_SKIP(mb_type)) {
2090                         COLOR(170, 48)
2091                     } else if (IS_GMC(mb_type)) {
2092                         COLOR(190, 48)
2093                     } else if (IS_SKIP(mb_type)) {
2094                         // COLOR(180, 48)
2095                     } else if (!USES_LIST(mb_type, 1)) {
2096                         COLOR(240, 48)
2097                     } else if (!USES_LIST(mb_type, 0)) {
2098                         COLOR(0, 48)
2099                     } else {
2100                         av_assert2(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
2101                         COLOR(300,48)
2102                     }
2103
2104                     u *= 0x0101010101010101ULL;
2105                     v *= 0x0101010101010101ULL;
2106                     for (y = 0; y < block_height; y++) {
2107                         *(uint64_t *)(pict->data[1] + 8 * mb_x +
2108                                       (block_height * mb_y + y) * pict->linesize[1]) = u;
2109                         *(uint64_t *)(pict->data[2] + 8 * mb_x +
2110                                       (block_height * mb_y + y) * pict->linesize[2]) = v;
2111                     }
2112
2113                     // segmentation
2114                     if (IS_8X8(mb_type) || IS_16X8(mb_type)) {
2115                         *(uint64_t *)(pict->data[0] + 16 * mb_x + 0 +
2116                                       (16 * mb_y + 8) * pict->linesize[0]) ^= 0x8080808080808080ULL;
2117                         *(uint64_t *)(pict->data[0] + 16 * mb_x + 8 +
2118                                       (16 * mb_y + 8) * pict->linesize[0]) ^= 0x8080808080808080ULL;
2119                     }
2120                     if (IS_8X8(mb_type) || IS_8X16(mb_type)) {
2121                         for (y = 0; y < 16; y++)
2122                             pict->data[0][16 * mb_x + 8 + (16 * mb_y + y) *
2123                                           pict->linesize[0]] ^= 0x80;
2124                     }
2125                     if (IS_8X8(mb_type) && mv_sample_log2 >= 2) {
2126                         int dm = 1 << (mv_sample_log2 - 2);
2127                         for (i = 0; i < 4; i++) {
2128                             int sx = mb_x * 16 + 8 * (i & 1);
2129                             int sy = mb_y * 16 + 8 * (i >> 1);
2130                             int xy = (mb_x * 2 + (i & 1) +
2131                                      (mb_y * 2 + (i >> 1)) * mv_stride) << (mv_sample_log2 - 1);
2132                             // FIXME bidir
2133                             int32_t *mv = (int32_t *) &p->motion_val[0][xy];
2134                             if (mv[0] != mv[dm] ||
2135                                 mv[dm * mv_stride] != mv[dm * (mv_stride + 1)])
2136                                 for (y = 0; y < 8; y++)
2137                                     pict->data[0][sx + 4 + (sy + y) * pict->linesize[0]] ^= 0x80;
2138                             if (mv[0] != mv[dm * mv_stride] || mv[dm] != mv[dm * (mv_stride + 1)])
2139                                 *(uint64_t *)(pict->data[0] + sx + (sy + 4) *
2140                                               pict->linesize[0]) ^= 0x8080808080808080ULL;
2141                         }
2142                     }
2143
2144                     if (IS_INTERLACED(mb_type) &&
2145                         avctx->codec->id == AV_CODEC_ID_H264) {
2146                         // hmm
2147                     }
2148                 }
2149                 mbskip_table[mb_index] = 0;
2150             }
2151         }
2152     }
2153 }
2154
2155 void ff_print_debug_info(MpegEncContext *s, Picture *p, AVFrame *pict)
2156 {
2157     ff_print_debug_info2(s->avctx, p, pict, s->mbskip_table, &s->low_delay,
2158                          s->mb_width, s->mb_height, s->mb_stride, s->quarter_sample);
2159 }
2160
2161 int ff_mpv_export_qp_table(MpegEncContext *s, AVFrame *f, Picture *p, int qp_type)
2162 {
2163     AVBufferRef *ref = av_buffer_ref(p->qscale_table_buf);
2164     int offset = 2*s->mb_stride + 1;
2165     if(!ref)
2166         return AVERROR(ENOMEM);
2167     av_assert0(ref->size >= offset + s->mb_stride * ((f->height+15)/16));
2168     ref->size -= offset;
2169     ref->data += offset;
2170     return av_frame_set_qp_table(f, ref, s->mb_stride, qp_type);
2171 }
2172
2173 static inline int hpel_motion_lowres(MpegEncContext *s,
2174                                      uint8_t *dest, uint8_t *src,
2175                                      int field_based, int field_select,
2176                                      int src_x, int src_y,
2177                                      int width, int height, int stride,
2178                                      int h_edge_pos, int v_edge_pos,
2179                                      int w, int h, h264_chroma_mc_func *pix_op,
2180                                      int motion_x, int motion_y)
2181 {
2182     const int lowres   = s->avctx->lowres;
2183     const int op_index = FFMIN(lowres, 3);
2184     const int s_mask   = (2 << lowres) - 1;
2185     int emu = 0;
2186     int sx, sy;
2187
2188     if (s->quarter_sample) {
2189         motion_x /= 2;
2190         motion_y /= 2;
2191     }
2192
2193     sx = motion_x & s_mask;
2194     sy = motion_y & s_mask;
2195     src_x += motion_x >> lowres + 1;
2196     src_y += motion_y >> lowres + 1;
2197
2198     src   += src_y * stride + src_x;
2199
2200     if ((unsigned)src_x > FFMAX( h_edge_pos - (!!sx) - w,                 0) ||
2201         (unsigned)src_y > FFMAX((v_edge_pos >> field_based) - (!!sy) - h, 0)) {
2202         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, src, s->linesize, w + 1,
2203                                 (h + 1) << field_based, src_x,
2204                                 src_y   << field_based,
2205                                 h_edge_pos,
2206                                 v_edge_pos);
2207         src = s->edge_emu_buffer;
2208         emu = 1;
2209     }
2210
2211     sx = (sx << 2) >> lowres;
2212     sy = (sy << 2) >> lowres;
2213     if (field_select)
2214         src += s->linesize;
2215     pix_op[op_index](dest, src, stride, h, sx, sy);
2216     return emu;
2217 }
2218
2219 /* apply one mpeg motion vector to the three components */
2220 static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
2221                                                 uint8_t *dest_y,
2222                                                 uint8_t *dest_cb,
2223                                                 uint8_t *dest_cr,
2224                                                 int field_based,
2225                                                 int bottom_field,
2226                                                 int field_select,
2227                                                 uint8_t **ref_picture,
2228                                                 h264_chroma_mc_func *pix_op,
2229                                                 int motion_x, int motion_y,
2230                                                 int h, int mb_y)
2231 {
2232     uint8_t *ptr_y, *ptr_cb, *ptr_cr;
2233     int mx, my, src_x, src_y, uvsrc_x, uvsrc_y, uvlinesize, linesize, sx, sy,
2234         uvsx, uvsy;
2235     const int lowres     = s->avctx->lowres;
2236     const int op_index   = FFMIN(lowres-1+s->chroma_x_shift, 3);
2237     const int block_s    = 8>>lowres;
2238     const int s_mask     = (2 << lowres) - 1;
2239     const int h_edge_pos = s->h_edge_pos >> lowres;
2240     const int v_edge_pos = s->v_edge_pos >> lowres;
2241     linesize   = s->current_picture.f.linesize[0] << field_based;
2242     uvlinesize = s->current_picture.f.linesize[1] << field_based;
2243
2244     // FIXME obviously not perfect but qpel will not work in lowres anyway
2245     if (s->quarter_sample) {
2246         motion_x /= 2;
2247         motion_y /= 2;
2248     }
2249
2250     if(field_based){
2251         motion_y += (bottom_field - field_select)*((1 << lowres)-1);
2252     }
2253
2254     sx = motion_x & s_mask;
2255     sy = motion_y & s_mask;
2256     src_x = s->mb_x * 2 * block_s + (motion_x >> lowres + 1);
2257     src_y = (mb_y * 2 * block_s >> field_based) + (motion_y >> lowres + 1);
2258
2259     if (s->out_format == FMT_H263) {
2260         uvsx    = ((motion_x >> 1) & s_mask) | (sx & 1);
2261         uvsy    = ((motion_y >> 1) & s_mask) | (sy & 1);
2262         uvsrc_x = src_x >> 1;
2263         uvsrc_y = src_y >> 1;
2264     } else if (s->out_format == FMT_H261) {
2265         // even chroma mv's are full pel in H261
2266         mx      = motion_x / 4;
2267         my      = motion_y / 4;
2268         uvsx    = (2 * mx) & s_mask;
2269         uvsy    = (2 * my) & s_mask;
2270         uvsrc_x = s->mb_x * block_s + (mx >> lowres);
2271         uvsrc_y =    mb_y * block_s + (my >> lowres);
2272     } else {
2273         if(s->chroma_y_shift){
2274             mx      = motion_x / 2;
2275             my      = motion_y / 2;
2276             uvsx    = mx & s_mask;
2277             uvsy    = my & s_mask;
2278             uvsrc_x = s->mb_x * block_s                 + (mx >> lowres + 1);
2279             uvsrc_y =   (mb_y * block_s >> field_based) + (my >> lowres + 1);
2280         } else {
2281             if(s->chroma_x_shift){
2282             //Chroma422
2283                 mx = motion_x / 2;
2284                 uvsx = mx & s_mask;
2285                 uvsy = motion_y & s_mask;
2286                 uvsrc_y = src_y;
2287                 uvsrc_x = s->mb_x*block_s               + (mx >> (lowres+1));
2288             } else {
2289             //Chroma444
2290                 uvsx = motion_x & s_mask;
2291                 uvsy = motion_y & s_mask;
2292                 uvsrc_x = src_x;
2293                 uvsrc_y = src_y;
2294             }
2295         }
2296     }
2297
2298     ptr_y  = ref_picture[0] + src_y   * linesize   + src_x;
2299     ptr_cb = ref_picture[1] + uvsrc_y * uvlinesize + uvsrc_x;
2300     ptr_cr = ref_picture[2] + uvsrc_y * uvlinesize + uvsrc_x;
2301
2302     if ((unsigned) src_x > FFMAX( h_edge_pos - (!!sx) - 2 * block_s,       0) || uvsrc_y<0 ||
2303         (unsigned) src_y > FFMAX((v_edge_pos >> field_based) - (!!sy) - h, 0)) {
2304         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr_y,
2305                                 linesize >> field_based, 17, 17 + field_based,
2306                                 src_x, src_y << field_based, h_edge_pos,
2307                                 v_edge_pos);
2308         ptr_y = s->edge_emu_buffer;
2309         if (!CONFIG_GRAY || !(s->flags & CODEC_FLAG_GRAY)) {
2310             uint8_t *uvbuf = s->edge_emu_buffer + 18 * s->linesize;
2311             s->vdsp.emulated_edge_mc(uvbuf , ptr_cb, uvlinesize >> field_based, 9,
2312                                     9 + field_based,
2313                                     uvsrc_x, uvsrc_y << field_based,
2314                                     h_edge_pos >> 1, v_edge_pos >> 1);
2315             s->vdsp.emulated_edge_mc(uvbuf + 16, ptr_cr, uvlinesize >> field_based, 9,
2316                                     9 + field_based,
2317                                     uvsrc_x, uvsrc_y << field_based,
2318                                     h_edge_pos >> 1, v_edge_pos >> 1);
2319             ptr_cb = uvbuf;
2320             ptr_cr = uvbuf + 16;
2321         }
2322     }
2323
2324     // FIXME use this for field pix too instead of the obnoxious hack which changes picture.f.data
2325     if (bottom_field) {
2326         dest_y  += s->linesize;
2327         dest_cb += s->uvlinesize;
2328         dest_cr += s->uvlinesize;
2329     }
2330
2331     if (field_select) {
2332         ptr_y   += s->linesize;
2333         ptr_cb  += s->uvlinesize;
2334         ptr_cr  += s->uvlinesize;
2335     }
2336
2337     sx = (sx << 2) >> lowres;
2338     sy = (sy << 2) >> lowres;
2339     pix_op[lowres - 1](dest_y, ptr_y, linesize, h, sx, sy);
2340
2341     if (!CONFIG_GRAY || !(s->flags & CODEC_FLAG_GRAY)) {
2342         int hc = s->chroma_y_shift ? (h+1-bottom_field)>>1 : h;
2343         uvsx = (uvsx << 2) >> lowres;
2344         uvsy = (uvsy << 2) >> lowres;
2345         if (hc) {
2346             pix_op[op_index](dest_cb, ptr_cb, uvlinesize, hc, uvsx, uvsy);
2347             pix_op[op_index](dest_cr, ptr_cr, uvlinesize, hc, uvsx, uvsy);
2348         }
2349     }
2350     // FIXME h261 lowres loop filter
2351 }
2352
2353 static inline void chroma_4mv_motion_lowres(MpegEncContext *s,
2354                                             uint8_t *dest_cb, uint8_t *dest_cr,
2355                                             uint8_t **ref_picture,
2356                                             h264_chroma_mc_func * pix_op,
2357                                             int mx, int my)
2358 {
2359     const int lowres     = s->avctx->lowres;
2360     const int op_index   = FFMIN(lowres, 3);
2361     const int block_s    = 8 >> lowres;
2362     const int s_mask     = (2 << lowres) - 1;
2363     const int h_edge_pos = s->h_edge_pos >> lowres + 1;
2364     const int v_edge_pos = s->v_edge_pos >> lowres + 1;
2365     int emu = 0, src_x, src_y, offset, sx, sy;
2366     uint8_t *ptr;
2367
2368     if (s->quarter_sample) {
2369         mx /= 2;
2370         my /= 2;
2371     }
2372
2373     /* In case of 8X8, we construct a single chroma motion vector
2374        with a special rounding */
2375     mx = ff_h263_round_chroma(mx);
2376     my = ff_h263_round_chroma(my);
2377
2378     sx = mx & s_mask;
2379     sy = my & s_mask;
2380     src_x = s->mb_x * block_s + (mx >> lowres + 1);
2381     src_y = s->mb_y * block_s + (my >> lowres + 1);
2382
2383     offset = src_y * s->uvlinesize + src_x;
2384     ptr = ref_picture[1] + offset;
2385     if (s->flags & CODEC_FLAG_EMU_EDGE) {
2386         if ((unsigned) src_x > FFMAX(h_edge_pos - (!!sx) - block_s, 0) ||
2387             (unsigned) src_y > FFMAX(v_edge_pos - (!!sy) - block_s, 0)) {
2388             s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize,
2389                                     9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
2390             ptr = s->edge_emu_buffer;
2391             emu = 1;
2392         }
2393     }
2394     sx = (sx << 2) >> lowres;
2395     sy = (sy << 2) >> lowres;
2396     pix_op[op_index](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
2397
2398     ptr = ref_picture[2] + offset;
2399     if (emu) {
2400         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9,
2401                                 src_x, src_y, h_edge_pos, v_edge_pos);
2402         ptr = s->edge_emu_buffer;
2403     }
2404     pix_op[op_index](dest_cr, ptr, s->uvlinesize, block_s, sx, sy);
2405 }
2406
2407 /**
2408  * motion compensation of a single macroblock
2409  * @param s context
2410  * @param dest_y luma destination pointer
2411  * @param dest_cb chroma cb/u destination pointer
2412  * @param dest_cr chroma cr/v destination pointer
2413  * @param dir direction (0->forward, 1->backward)
2414  * @param ref_picture array[3] of pointers to the 3 planes of the reference picture
2415  * @param pix_op halfpel motion compensation function (average or put normally)
2416  * the motion vectors are taken from s->mv and the MV type from s->mv_type
2417  */
2418 static inline void MPV_motion_lowres(MpegEncContext *s,
2419                                      uint8_t *dest_y, uint8_t *dest_cb,
2420                                      uint8_t *dest_cr,
2421                                      int dir, uint8_t **ref_picture,
2422                                      h264_chroma_mc_func *pix_op)
2423 {
2424     int mx, my;
2425     int mb_x, mb_y, i;
2426     const int lowres  = s->avctx->lowres;
2427     const int block_s = 8 >>lowres;
2428
2429     mb_x = s->mb_x;
2430     mb_y = s->mb_y;
2431
2432     switch (s->mv_type) {
2433     case MV_TYPE_16X16:
2434         mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2435                            0, 0, 0,
2436                            ref_picture, pix_op,
2437                            s->mv[dir][0][0], s->mv[dir][0][1],
2438                            2 * block_s, mb_y);
2439         break;
2440     case MV_TYPE_8X8:
2441         mx = 0;
2442         my = 0;
2443         for (i = 0; i < 4; i++) {
2444             hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) *
2445                                s->linesize) * block_s,
2446                                ref_picture[0], 0, 0,
2447                                (2 * mb_x + (i & 1)) * block_s,
2448                                (2 * mb_y + (i >> 1)) * block_s,
2449                                s->width, s->height, s->linesize,
2450                                s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
2451                                block_s, block_s, pix_op,
2452                                s->mv[dir][i][0], s->mv[dir][i][1]);
2453
2454             mx += s->mv[dir][i][0];
2455             my += s->mv[dir][i][1];
2456         }
2457
2458         if (!CONFIG_GRAY || !(s->flags & CODEC_FLAG_GRAY))
2459             chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture,
2460                                      pix_op, mx, my);
2461         break;
2462     case MV_TYPE_FIELD:
2463         if (s->picture_structure == PICT_FRAME) {
2464             /* top field */
2465             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2466                                1, 0, s->field_select[dir][0],
2467                                ref_picture, pix_op,
2468                                s->mv[dir][0][0], s->mv[dir][0][1],
2469                                block_s, mb_y);
2470             /* bottom field */
2471             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2472                                1, 1, s->field_select[dir][1],
2473                                ref_picture, pix_op,
2474                                s->mv[dir][1][0], s->mv[dir][1][1],
2475                                block_s, mb_y);
2476         } else {
2477             if (s->picture_structure != s->field_select[dir][0] + 1 &&
2478                 s->pict_type != AV_PICTURE_TYPE_B && !s->first_field) {
2479                 ref_picture = s->current_picture_ptr->f.data;
2480
2481             }
2482             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2483                                0, 0, s->field_select[dir][0],
2484                                ref_picture, pix_op,
2485                                s->mv[dir][0][0],
2486                                s->mv[dir][0][1], 2 * block_s, mb_y >> 1);
2487             }
2488         break;
2489     case MV_TYPE_16X8:
2490         for (i = 0; i < 2; i++) {
2491             uint8_t **ref2picture;
2492
2493             if (s->picture_structure == s->field_select[dir][i] + 1 ||
2494                 s->pict_type == AV_PICTURE_TYPE_B || s->first_field) {
2495                 ref2picture = ref_picture;
2496             } else {
2497                 ref2picture = s->current_picture_ptr->f.data;
2498             }
2499
2500             mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2501                                0, 0, s->field_select[dir][i],
2502                                ref2picture, pix_op,
2503                                s->mv[dir][i][0], s->mv[dir][i][1] +
2504                                2 * block_s * i, block_s, mb_y >> 1);
2505
2506             dest_y  +=  2 * block_s *  s->linesize;
2507             dest_cb += (2 * block_s >> s->chroma_y_shift) * s->uvlinesize;
2508             dest_cr += (2 * block_s >> s->chroma_y_shift) * s->uvlinesize;
2509         }
2510         break;
2511     case MV_TYPE_DMV:
2512         if (s->picture_structure == PICT_FRAME) {
2513             for (i = 0; i < 2; i++) {
2514                 int j;
2515                 for (j = 0; j < 2; j++) {
2516                     mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2517                                        1, j, j ^ i,
2518                                        ref_picture, pix_op,
2519                                        s->mv[dir][2 * i + j][0],
2520                                        s->mv[dir][2 * i + j][1],
2521                                        block_s, mb_y);
2522                 }
2523                 pix_op = s->h264chroma.avg_h264_chroma_pixels_tab;
2524             }
2525         } else {
2526             for (i = 0; i < 2; i++) {
2527                 mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
2528                                    0, 0, s->picture_structure != i + 1,
2529                                    ref_picture, pix_op,
2530                                    s->mv[dir][2 * i][0],s->mv[dir][2 * i][1],
2531                                    2 * block_s, mb_y >> 1);
2532
2533                 // after put we make avg of the same block
2534                 pix_op = s->h264chroma.avg_h264_chroma_pixels_tab;
2535
2536                 // opposite parity is always in the same
2537                 // frame if this is second field
2538                 if (!s->first_field) {
2539                     ref_picture = s->current_picture_ptr->f.data;
2540                 }
2541             }
2542         }
2543         break;
2544     default:
2545         av_assert2(0);
2546     }
2547 }
2548
2549 /**
2550  * find the lowest MB row referenced in the MVs
2551  */
2552 int ff_MPV_lowest_referenced_row(MpegEncContext *s, int dir)
2553 {
2554     int my_max = INT_MIN, my_min = INT_MAX, qpel_shift = !s->quarter_sample;
2555     int my, off, i, mvs;
2556
2557     if (s->picture_structure != PICT_FRAME || s->mcsel)
2558         goto unhandled;
2559
2560     switch (s->mv_type) {
2561         case MV_TYPE_16X16:
2562             mvs = 1;
2563             break;
2564         case MV_TYPE_16X8:
2565             mvs = 2;
2566             break;
2567         case MV_TYPE_8X8:
2568             mvs = 4;
2569             break;
2570         default:
2571             goto unhandled;
2572     }
2573
2574     for (i = 0; i < mvs; i++) {
2575         my = s->mv[dir][i][1]<<qpel_shift;
2576         my_max = FFMAX(my_max, my);
2577         my_min = FFMIN(my_min, my);
2578     }
2579
2580     off = (FFMAX(-my_min, my_max) + 63) >> 6;
2581
2582     return FFMIN(FFMAX(s->mb_y + off, 0), s->mb_height-1);
2583 unhandled:
2584     return s->mb_height-1;
2585 }
2586
2587 /* put block[] to dest[] */
2588 static inline void put_dct(MpegEncContext *s,
2589                            int16_t *block, int i, uint8_t *dest, int line_size, int qscale)
2590 {
2591     s->dct_unquantize_intra(s, block, i, qscale);
2592     s->dsp.idct_put (dest, line_size, block);
2593 }
2594
2595 /* add block[] to dest[] */
2596 static inline void add_dct(MpegEncContext *s,
2597                            int16_t *block, int i, uint8_t *dest, int line_size)
2598 {
2599     if (s->block_last_index[i] >= 0) {
2600         s->dsp.idct_add (dest, line_size, block);
2601     }
2602 }
2603
2604 static inline void add_dequant_dct(MpegEncContext *s,
2605                            int16_t *block, int i, uint8_t *dest, int line_size, int qscale)
2606 {
2607     if (s->block_last_index[i] >= 0) {
2608         s->dct_unquantize_inter(s, block, i, qscale);
2609
2610         s->dsp.idct_add (dest, line_size, block);
2611     }
2612 }
2613
2614 /**
2615  * Clean dc, ac, coded_block for the current non-intra MB.
2616  */
2617 void ff_clean_intra_table_entries(MpegEncContext *s)
2618 {
2619     int wrap = s->b8_stride;
2620     int xy = s->block_index[0];
2621
2622     s->dc_val[0][xy           ] =
2623     s->dc_val[0][xy + 1       ] =
2624     s->dc_val[0][xy     + wrap] =
2625     s->dc_val[0][xy + 1 + wrap] = 1024;
2626     /* ac pred */
2627     memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
2628     memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
2629     if (s->msmpeg4_version>=3) {
2630         s->coded_block[xy           ] =
2631         s->coded_block[xy + 1       ] =
2632         s->coded_block[xy     + wrap] =
2633         s->coded_block[xy + 1 + wrap] = 0;
2634     }
2635     /* chroma */
2636     wrap = s->mb_stride;
2637     xy = s->mb_x + s->mb_y * wrap;
2638     s->dc_val[1][xy] =
2639     s->dc_val[2][xy] = 1024;
2640     /* ac pred */
2641     memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
2642     memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
2643
2644     s->mbintra_table[xy]= 0;
2645 }
2646
2647 /* generic function called after a macroblock has been parsed by the
2648    decoder or after it has been encoded by the encoder.
2649
2650    Important variables used:
2651    s->mb_intra : true if intra macroblock
2652    s->mv_dir   : motion vector direction
2653    s->mv_type  : motion vector type
2654    s->mv       : motion vector
2655    s->interlaced_dct : true if interlaced dct used (mpeg2)
2656  */
2657 static av_always_inline
2658 void MPV_decode_mb_internal(MpegEncContext *s, int16_t block[12][64],
2659                             int lowres_flag, int is_mpeg12)
2660 {
2661     const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
2662     if(CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration){
2663         ff_xvmc_decode_mb(s);//xvmc uses pblocks
2664         return;
2665     }
2666
2667     if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
2668        /* print DCT coefficients */
2669        int i,j;
2670        av_log(s->avctx, AV_LOG_DEBUG, "DCT coeffs of MB at %dx%d:\n", s->mb_x, s->mb_y);
2671        for(i=0; i<6; i++){
2672            for(j=0; j<64; j++){
2673                av_log(s->avctx, AV_LOG_DEBUG, "%5d", block[i][s->dsp.idct_permutation[j]]);
2674            }
2675            av_log(s->avctx, AV_LOG_DEBUG, "\n");
2676        }
2677     }
2678
2679     s->current_picture.qscale_table[mb_xy] = s->qscale;
2680
2681     /* update DC predictors for P macroblocks */
2682     if (!s->mb_intra) {
2683         if (!is_mpeg12 && (s->h263_pred || s->h263_aic)) {
2684             if(s->mbintra_table[mb_xy])
2685                 ff_clean_intra_table_entries(s);
2686         } else {
2687             s->last_dc[0] =
2688             s->last_dc[1] =
2689             s->last_dc[2] = 128 << s->intra_dc_precision;
2690         }
2691     }
2692     else if (!is_mpeg12 && (s->h263_pred || s->h263_aic))
2693         s->mbintra_table[mb_xy]=1;
2694
2695     if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==AV_PICTURE_TYPE_B) && s->avctx->mb_decision != FF_MB_DECISION_RD)) { //FIXME precalc
2696         uint8_t *dest_y, *dest_cb, *dest_cr;
2697         int dct_linesize, dct_offset;
2698         op_pixels_func (*op_pix)[4];
2699         qpel_mc_func (*op_qpix)[16];
2700         const int linesize   = s->current_picture.f.linesize[0]; //not s->linesize as this would be wrong for field pics
2701         const int uvlinesize = s->current_picture.f.linesize[1];
2702         const int readable= s->pict_type != AV_PICTURE_TYPE_B || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
2703         const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
2704
2705         /* avoid copy if macroblock skipped in last frame too */
2706         /* skip only during decoding as we might trash the buffers during encoding a bit */
2707         if(!s->encoding){
2708             uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
2709
2710             if (s->mb_skipped) {
2711                 s->mb_skipped= 0;
2712                 av_assert2(s->pict_type!=AV_PICTURE_TYPE_I);
2713                 *mbskip_ptr = 1;
2714             } else if(!s->current_picture.reference) {
2715                 *mbskip_ptr = 1;
2716             } else{
2717                 *mbskip_ptr = 0; /* not skipped */
2718             }
2719         }
2720
2721         dct_linesize = linesize << s->interlaced_dct;
2722         dct_offset   = s->interlaced_dct ? linesize : linesize * block_size;
2723
2724         if(readable){
2725             dest_y=  s->dest[0];
2726             dest_cb= s->dest[1];
2727             dest_cr= s->dest[2];
2728         }else{
2729             dest_y = s->b_scratchpad;
2730             dest_cb= s->b_scratchpad+16*linesize;
2731             dest_cr= s->b_scratchpad+32*linesize;
2732         }
2733
2734         if (!s->mb_intra) {
2735             /* motion handling */
2736             /* decoding or more than one mb_type (MC was already done otherwise) */
2737             if(!s->encoding){
2738
2739                 if(HAVE_THREADS && s->avctx->active_thread_type&FF_THREAD_FRAME) {
2740                     if (s->mv_dir & MV_DIR_FORWARD) {
2741                         ff_thread_await_progress(&s->last_picture_ptr->tf,
2742                                                  ff_MPV_lowest_referenced_row(s, 0),
2743                                                  0);
2744                     }
2745                     if (s->mv_dir & MV_DIR_BACKWARD) {
2746                         ff_thread_await_progress(&s->next_picture_ptr->tf,
2747                                                  ff_MPV_lowest_referenced_row(s, 1),
2748                                                  0);
2749                     }
2750                 }
2751
2752                 if(lowres_flag){
2753                     h264_chroma_mc_func *op_pix = s->h264chroma.put_h264_chroma_pixels_tab;
2754
2755                     if (s->mv_dir & MV_DIR_FORWARD) {
2756                         MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.f.data, op_pix);
2757                         op_pix = s->h264chroma.avg_h264_chroma_pixels_tab;
2758                     }
2759                     if (s->mv_dir & MV_DIR_BACKWARD) {
2760                         MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.f.data, op_pix);
2761                     }
2762                 }else{
2763                     op_qpix= s->me.qpel_put;
2764                     if ((!s->no_rounding) || s->pict_type==AV_PICTURE_TYPE_B){
2765                         op_pix = s->hdsp.put_pixels_tab;
2766                     }else{
2767                         op_pix = s->hdsp.put_no_rnd_pixels_tab;
2768                     }
2769                     if (s->mv_dir & MV_DIR_FORWARD) {
2770                         ff_MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.f.data, op_pix, op_qpix);
2771                         op_pix = s->hdsp.avg_pixels_tab;
2772                         op_qpix= s->me.qpel_avg;
2773                     }
2774                     if (s->mv_dir & MV_DIR_BACKWARD) {
2775                         ff_MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.f.data, op_pix, op_qpix);
2776                     }
2777                 }
2778             }
2779
2780             /* skip dequant / idct if we are really late ;) */
2781             if(s->avctx->skip_idct){
2782                 if(  (s->avctx->skip_idct >= AVDISCARD_NONREF && s->pict_type == AV_PICTURE_TYPE_B)
2783                    ||(s->avctx->skip_idct >= AVDISCARD_NONKEY && s->pict_type != AV_PICTURE_TYPE_I)
2784                    || s->avctx->skip_idct >= AVDISCARD_ALL)
2785                     goto skip_idct;
2786             }
2787
2788             /* add dct residue */
2789             if(s->encoding || !(   s->msmpeg4_version || s->codec_id==AV_CODEC_ID_MPEG1VIDEO || s->codec_id==AV_CODEC_ID_MPEG2VIDEO
2790                                 || (s->codec_id==AV_CODEC_ID_MPEG4 && !s->mpeg_quant))){
2791                 add_dequant_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
2792                 add_dequant_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
2793                 add_dequant_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
2794                 add_dequant_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
2795
2796                 if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2797                     if (s->chroma_y_shift){
2798                         add_dequant_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
2799                         add_dequant_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
2800                     }else{
2801                         dct_linesize >>= 1;
2802                         dct_offset >>=1;
2803                         add_dequant_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
2804                         add_dequant_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
2805                         add_dequant_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
2806                         add_dequant_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
2807                     }
2808                 }
2809             } else if(is_mpeg12 || (s->codec_id != AV_CODEC_ID_WMV2)){
2810                 add_dct(s, block[0], 0, dest_y                          , dct_linesize);
2811                 add_dct(s, block[1], 1, dest_y              + block_size, dct_linesize);
2812                 add_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize);
2813                 add_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize);
2814
2815                 if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2816                     if(s->chroma_y_shift){//Chroma420
2817                         add_dct(s, block[4], 4, dest_cb, uvlinesize);
2818                         add_dct(s, block[5], 5, dest_cr, uvlinesize);
2819                     }else{
2820                         //chroma422
2821                         dct_linesize = uvlinesize << s->interlaced_dct;
2822                         dct_offset   = s->interlaced_dct ? uvlinesize : uvlinesize*block_size;
2823
2824                         add_dct(s, block[4], 4, dest_cb, dct_linesize);
2825                         add_dct(s, block[5], 5, dest_cr, dct_linesize);
2826                         add_dct(s, block[6], 6, dest_cb+dct_offset, dct_linesize);
2827                         add_dct(s, block[7], 7, dest_cr+dct_offset, dct_linesize);
2828                         if(!s->chroma_x_shift){//Chroma444
2829                             add_dct(s, block[8], 8, dest_cb+block_size, dct_linesize);
2830                             add_dct(s, block[9], 9, dest_cr+block_size, dct_linesize);
2831                             add_dct(s, block[10], 10, dest_cb+block_size+dct_offset, dct_linesize);
2832                             add_dct(s, block[11], 11, dest_cr+block_size+dct_offset, dct_linesize);
2833                         }
2834                     }
2835                 }//fi gray
2836             }
2837             else if (CONFIG_WMV2_DECODER || CONFIG_WMV2_ENCODER) {
2838                 ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
2839             }
2840         } else {
2841             /* dct only in intra block */
2842             if(s->encoding || !(s->codec_id==AV_CODEC_ID_MPEG1VIDEO || s->codec_id==AV_CODEC_ID_MPEG2VIDEO)){
2843                 put_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
2844                 put_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
2845                 put_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
2846                 put_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
2847
2848                 if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2849                     if(s->chroma_y_shift){
2850                         put_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
2851                         put_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
2852                     }else{
2853                         dct_offset >>=1;
2854                         dct_linesize >>=1;
2855                         put_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
2856                         put_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
2857                         put_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
2858                         put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
2859                     }
2860                 }
2861             }else{
2862                 s->dsp.idct_put(dest_y                          , dct_linesize, block[0]);
2863                 s->dsp.idct_put(dest_y              + block_size, dct_linesize, block[1]);
2864                 s->dsp.idct_put(dest_y + dct_offset             , dct_linesize, block[2]);
2865                 s->dsp.idct_put(dest_y + dct_offset + block_size, dct_linesize, block[3]);
2866
2867                 if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2868                     if(s->chroma_y_shift){
2869                         s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
2870                         s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
2871                     }else{
2872
2873                         dct_linesize = uvlinesize << s->interlaced_dct;
2874                         dct_offset   = s->interlaced_dct? uvlinesize : uvlinesize*block_size;
2875
2876                         s->dsp.idct_put(dest_cb,              dct_linesize, block[4]);
2877                         s->dsp.idct_put(dest_cr,              dct_linesize, block[5]);
2878                         s->dsp.idct_put(dest_cb + dct_offset, dct_linesize, block[6]);
2879                         s->dsp.idct_put(dest_cr + dct_offset, dct_linesize, block[7]);
2880                         if(!s->chroma_x_shift){//Chroma444
2881                             s->dsp.idct_put(dest_cb + block_size,              dct_linesize, block[8]);
2882                             s->dsp.idct_put(dest_cr + block_size,              dct_linesize, block[9]);
2883                             s->dsp.idct_put(dest_cb + block_size + dct_offset, dct_linesize, block[10]);
2884                             s->dsp.idct_put(dest_cr + block_size + dct_offset, dct_linesize, block[11]);
2885                         }
2886                     }
2887                 }//gray
2888             }
2889         }
2890 skip_idct:
2891         if(!readable){
2892             s->hdsp.put_pixels_tab[0][0](s->dest[0], dest_y ,   linesize,16);
2893             s->hdsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[1], dest_cb, uvlinesize,16 >> s->chroma_y_shift);
2894             s->hdsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[2], dest_cr, uvlinesize,16 >> s->chroma_y_shift);
2895         }
2896     }
2897 }
2898
2899 void ff_MPV_decode_mb(MpegEncContext *s, int16_t block[12][64]){
2900 #if !CONFIG_SMALL
2901     if(s->out_format == FMT_MPEG1) {
2902         if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 1);
2903         else                 MPV_decode_mb_internal(s, block, 0, 1);
2904     } else
2905 #endif
2906     if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 0);
2907     else                  MPV_decode_mb_internal(s, block, 0, 0);
2908 }
2909
2910 /**
2911  * @param h is the normal height, this will be reduced automatically if needed for the last row
2912  */
2913 void ff_draw_horiz_band(AVCodecContext *avctx, DSPContext *dsp, Picture *cur,
2914                         Picture *last, int y, int h, int picture_structure,
2915                         int first_field, int draw_edges, int low_delay,
2916                         int v_edge_pos, int h_edge_pos)
2917 {
2918     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(avctx->pix_fmt);
2919     int hshift = desc->log2_chroma_w;
2920     int vshift = desc->log2_chroma_h;
2921     const int field_pic = picture_structure != PICT_FRAME;
2922     if(field_pic){
2923         h <<= 1;
2924         y <<= 1;
2925     }
2926
2927     if (!avctx->hwaccel &&
2928         draw_edges &&
2929         cur->reference &&
2930         !(avctx->flags & CODEC_FLAG_EMU_EDGE)) {
2931         int *linesize = cur->f.linesize;
2932         int sides = 0, edge_h;
2933         if (y==0) sides |= EDGE_TOP;
2934         if (y + h >= v_edge_pos)
2935             sides |= EDGE_BOTTOM;
2936
2937         edge_h= FFMIN(h, v_edge_pos - y);
2938
2939         dsp->draw_edges(cur->f.data[0] + y * linesize[0],
2940                         linesize[0], h_edge_pos, edge_h,
2941                         EDGE_WIDTH, EDGE_WIDTH, sides);
2942         dsp->draw_edges(cur->f.data[1] + (y >> vshift) * linesize[1],
2943                         linesize[1], h_edge_pos >> hshift, edge_h >> vshift,
2944                         EDGE_WIDTH >> hshift, EDGE_WIDTH >> vshift, sides);
2945         dsp->draw_edges(cur->f.data[2] + (y >> vshift) * linesize[2],
2946                         linesize[2], h_edge_pos >> hshift, edge_h >> vshift,
2947                         EDGE_WIDTH >> hshift, EDGE_WIDTH >> vshift, sides);
2948     }
2949
2950     h = FFMIN(h, avctx->height - y);
2951
2952     if(field_pic && first_field && !(avctx->slice_flags&SLICE_FLAG_ALLOW_FIELD)) return;
2953
2954     if (avctx->draw_horiz_band) {
2955         AVFrame *src;
2956         int offset[AV_NUM_DATA_POINTERS];
2957         int i;
2958
2959         if(cur->f.pict_type == AV_PICTURE_TYPE_B || low_delay ||
2960            (avctx->slice_flags & SLICE_FLAG_CODED_ORDER))
2961             src = &cur->f;
2962         else if (last)
2963             src = &last->f;
2964         else
2965             return;
2966
2967         if (cur->f.pict_type == AV_PICTURE_TYPE_B &&
2968             picture_structure == PICT_FRAME &&
2969             avctx->codec_id != AV_CODEC_ID_SVQ3) {
2970             for (i = 0; i < AV_NUM_DATA_POINTERS; i++)
2971                 offset[i] = 0;
2972         }else{
2973             offset[0]= y * src->linesize[0];
2974             offset[1]=
2975             offset[2]= (y >> vshift) * src->linesize[1];
2976             for (i = 3; i < AV_NUM_DATA_POINTERS; i++)
2977                 offset[i] = 0;
2978         }
2979
2980         emms_c();
2981
2982         avctx->draw_horiz_band(avctx, src, offset,
2983                                y, picture_structure, h);
2984     }
2985 }
2986
2987 void ff_mpeg_draw_horiz_band(MpegEncContext *s, int y, int h)
2988 {
2989     int draw_edges = s->unrestricted_mv && !s->intra_only;
2990     ff_draw_horiz_band(s->avctx, &s->dsp, &s->current_picture,
2991                        &s->last_picture, y, h, s->picture_structure,
2992                        s->first_field, draw_edges, s->low_delay,
2993                        s->v_edge_pos, s->h_edge_pos);
2994 }
2995
2996 void ff_init_block_index(MpegEncContext *s){ //FIXME maybe rename
2997     const int linesize   = s->current_picture.f.linesize[0]; //not s->linesize as this would be wrong for field pics
2998     const int uvlinesize = s->current_picture.f.linesize[1];
2999     const int mb_size= 4 - s->avctx->lowres;
3000
3001     s->block_index[0]= s->b8_stride*(s->mb_y*2    ) - 2 + s->mb_x*2;
3002     s->block_index[1]= s->b8_stride*(s->mb_y*2    ) - 1 + s->mb_x*2;
3003     s->block_index[2]= s->b8_stride*(s->mb_y*2 + 1) - 2 + s->mb_x*2;
3004     s->block_index[3]= s->b8_stride*(s->mb_y*2 + 1) - 1 + s->mb_x*2;
3005     s->block_index[4]= s->mb_stride*(s->mb_y + 1)                + s->b8_stride*s->mb_height*2 + s->mb_x - 1;
3006     s->block_index[5]= s->mb_stride*(s->mb_y + s->mb_height + 2) + s->b8_stride*s->mb_height*2 + s->mb_x - 1;
3007     //block_index is not used by mpeg2, so it is not affected by chroma_format
3008
3009     s->dest[0] = s->current_picture.f.data[0] + ((s->mb_x - 1) <<  mb_size);
3010     s->dest[1] = s->current_picture.f.data[1] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
3011     s->dest[2] = s->current_picture.f.data[2] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
3012
3013     if(!(s->pict_type==AV_PICTURE_TYPE_B && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME))
3014     {
3015         if(s->picture_structure==PICT_FRAME){
3016         s->dest[0] += s->mb_y *   linesize << mb_size;
3017         s->dest[1] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
3018         s->dest[2] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
3019         }else{
3020             s->dest[0] += (s->mb_y>>1) *   linesize << mb_size;
3021             s->dest[1] += (s->mb_y>>1) * uvlinesize << (mb_size - s->chroma_y_shift);
3022             s->dest[2] += (s->mb_y>>1) * uvlinesize << (mb_size - s->chroma_y_shift);
3023             av_assert1((s->mb_y&1) == (s->picture_structure == PICT_BOTTOM_FIELD));
3024         }
3025     }
3026 }
3027
3028 /**
3029  * Permute an 8x8 block.
3030  * @param block the block which will be permuted according to the given permutation vector
3031  * @param permutation the permutation vector
3032  * @param last the last non zero coefficient in scantable order, used to speed the permutation up
3033  * @param scantable the used scantable, this is only used to speed the permutation up, the block is not
3034  *                  (inverse) permutated to scantable order!
3035  */
3036 void ff_block_permute(int16_t *block, uint8_t *permutation, const uint8_t *scantable, int last)
3037 {
3038     int i;
3039     int16_t temp[64];
3040
3041     if(last<=0) return;
3042     //if(permutation[1]==1) return; //FIXME it is ok but not clean and might fail for some permutations
3043
3044     for(i=0; i<=last; i++){
3045         const int j= scantable[i];
3046         temp[j]= block[j];
3047         block[j]=0;
3048     }
3049
3050     for(i=0; i<=last; i++){
3051         const int j= scantable[i];
3052         const int perm_j= permutation[j];
3053         block[perm_j]= temp[j];
3054     }
3055 }
3056
3057 void ff_mpeg_flush(AVCodecContext *avctx){
3058     int i;
3059     MpegEncContext *s = avctx->priv_data;
3060
3061     if(s==NULL || s->picture==NULL)
3062         return;
3063
3064     for (i = 0; i < MAX_PICTURE_COUNT; i++)
3065         ff_mpeg_unref_picture(s, &s->picture[i]);
3066     s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
3067
3068     ff_mpeg_unref_picture(s, &s->current_picture);
3069     ff_mpeg_unref_picture(s, &s->last_picture);
3070     ff_mpeg_unref_picture(s, &s->next_picture);
3071
3072     s->mb_x= s->mb_y= 0;
3073     s->closed_gop= 0;
3074
3075     s->parse_context.state= -1;
3076     s->parse_context.frame_start_found= 0;
3077     s->parse_context.overread= 0;
3078     s->parse_context.overread_index= 0;
3079     s->parse_context.index= 0;
3080     s->parse_context.last_index= 0;
3081     s->bitstream_buffer_size=0;
3082     s->pp_time=0;
3083 }
3084
3085 static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
3086                                    int16_t *block, int n, int qscale)
3087 {
3088     int i, level, nCoeffs;
3089     const uint16_t *quant_matrix;
3090
3091     nCoeffs= s->block_last_index[n];
3092
3093     block[0] *= n < 4 ? s->y_dc_scale : s->c_dc_scale;
3094     /* XXX: only mpeg1 */
3095     quant_matrix = s->intra_matrix;
3096     for(i=1;i<=nCoeffs;i++) {
3097         int j= s->intra_scantable.permutated[i];
3098         level = block[j];
3099         if (level) {
3100             if (level < 0) {
3101                 level = -level;
3102                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3103                 level = (level - 1) | 1;
3104                 level = -level;
3105             } else {
3106                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3107                 level = (level - 1) | 1;
3108             }
3109             block[j] = level;
3110         }
3111     }
3112 }
3113
3114 static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
3115                                    int16_t *block, int n, int qscale)
3116 {
3117     int i, level, nCoeffs;
3118     const uint16_t *quant_matrix;
3119
3120     nCoeffs= s->block_last_index[n];
3121
3122     quant_matrix = s->inter_matrix;
3123     for(i=0; i<=nCoeffs; i++) {
3124         int j= s->intra_scantable.permutated[i];
3125         level = block[j];
3126         if (level) {
3127             if (level < 0) {
3128                 level = -level;
3129                 level = (((level << 1) + 1) * qscale *
3130                          ((int) (quant_matrix[j]))) >> 4;
3131                 level = (level - 1) | 1;
3132                 level = -level;
3133             } else {
3134                 level = (((level << 1) + 1) * qscale *
3135                          ((int) (quant_matrix[j]))) >> 4;
3136                 level = (level - 1) | 1;
3137             }
3138             block[j] = level;
3139         }
3140     }
3141 }
3142
3143 static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
3144                                    int16_t *block, int n, int qscale)
3145 {
3146     int i, level, nCoeffs;
3147     const uint16_t *quant_matrix;
3148
3149     if(s->alternate_scan) nCoeffs= 63;
3150     else nCoeffs= s->block_last_index[n];
3151
3152     block[0] *= n < 4 ? s->y_dc_scale : s->c_dc_scale;
3153     quant_matrix = s->intra_matrix;
3154     for(i=1;i<=nCoeffs;i++) {
3155         int j= s->intra_scantable.permutated[i];
3156         level = block[j];
3157         if (level) {
3158             if (level < 0) {
3159                 level = -level;
3160                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3161                 level = -level;
3162             } else {
3163                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3164             }
3165             block[j] = level;
3166         }
3167     }
3168 }
3169
3170 static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
3171                                    int16_t *block, int n, int qscale)
3172 {
3173     int i, level, nCoeffs;
3174     const uint16_t *quant_matrix;
3175     int sum=-1;
3176
3177     if(s->alternate_scan) nCoeffs= 63;
3178     else nCoeffs= s->block_last_index[n];
3179
3180     block[0] *= n < 4 ? s->y_dc_scale : s->c_dc_scale;
3181     sum += block[0];
3182     quant_matrix = s->intra_matrix;
3183     for(i=1;i<=nCoeffs;i++) {
3184         int j= s->intra_scantable.permutated[i];
3185         level = block[j];
3186         if (level) {
3187             if (level < 0) {
3188                 level = -level;
3189                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3190                 level = -level;
3191             } else {
3192                 level = (int)(level * qscale * quant_matrix[j]) >> 3;
3193             }
3194             block[j] = level;
3195             sum+=level;
3196         }
3197     }
3198     block[63]^=sum&1;
3199 }
3200
3201 static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
3202                                    int16_t *block, int n, int qscale)
3203 {
3204     int i, level, nCoeffs;
3205     const uint16_t *quant_matrix;
3206     int sum=-1;
3207
3208     if(s->alternate_scan) nCoeffs= 63;
3209     else nCoeffs= s->block_last_index[n];
3210
3211     quant_matrix = s->inter_matrix;
3212     for(i=0; i<=nCoeffs; i++) {
3213         int j= s->intra_scantable.permutated[i];
3214         level = block[j];
3215         if (level) {
3216             if (level < 0) {
3217                 level = -level;
3218                 level = (((level << 1) + 1) * qscale *
3219                          ((int) (quant_matrix[j]))) >> 4;
3220                 level = -level;
3221             } else {
3222                 level = (((level << 1) + 1) * qscale *
3223                          ((int) (quant_matrix[j]))) >> 4;
3224             }
3225             block[j] = level;
3226             sum+=level;
3227         }
3228     }
3229     block[63]^=sum&1;
3230 }
3231
3232 static void dct_unquantize_h263_intra_c(MpegEncContext *s,
3233                                   int16_t *block, int n, int qscale)
3234 {
3235     int i, level, qmul, qadd;
3236     int nCoeffs;
3237
3238     av_assert2(s->block_last_index[n]>=0 || s->h263_aic);
3239
3240     qmul = qscale << 1;
3241
3242     if (!s->h263_aic) {
3243         block[0] *= n < 4 ? s->y_dc_scale : s->c_dc_scale;
3244         qadd = (qscale - 1) | 1;
3245     }else{
3246         qadd = 0;
3247     }
3248     if(s->ac_pred)
3249         nCoeffs=63;
3250     else
3251         nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
3252
3253     for(i=1; i<=nCoeffs; i++) {
3254         level = block[i];
3255         if (level) {
3256             if (level < 0) {
3257                 level = level * qmul - qadd;
3258             } else {
3259                 level = level * qmul + qadd;
3260             }
3261             block[i] = level;
3262         }
3263     }
3264 }
3265
3266 static void dct_unquantize_h263_inter_c(MpegEncContext *s,
3267                                   int16_t *block, int n, int qscale)
3268 {
3269     int i, level, qmul, qadd;
3270     int nCoeffs;
3271
3272     av_assert2(s->block_last_index[n]>=0);
3273
3274     qadd = (qscale - 1) | 1;
3275     qmul = qscale << 1;
3276
3277     nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
3278
3279     for(i=0; i<=nCoeffs; i++) {
3280         level = block[i];
3281         if (level) {
3282             if (level < 0) {
3283                 level = level * qmul - qadd;
3284             } else {
3285                 level = level * qmul + qadd;
3286             }
3287             block[i] = level;
3288         }
3289     }
3290 }
3291
3292 /**
3293  * set qscale and update qscale dependent variables.
3294  */
3295 void ff_set_qscale(MpegEncContext * s, int qscale)
3296 {
3297     if (qscale < 1)
3298         qscale = 1;
3299     else if (qscale > 31)
3300         qscale = 31;
3301
3302     s->qscale = qscale;
3303     s->chroma_qscale= s->chroma_qscale_table[qscale];
3304
3305     s->y_dc_scale= s->y_dc_scale_table[ qscale ];
3306     s->c_dc_scale= s->c_dc_scale_table[ s->chroma_qscale ];
3307 }
3308
3309 void ff_MPV_report_decode_progress(MpegEncContext *s)
3310 {
3311     if (s->pict_type != AV_PICTURE_TYPE_B && !s->partitioned_frame && !s->er.error_occurred)
3312         ff_thread_report_progress(&s->current_picture_ptr->tf, s->mb_y, 0);
3313 }
3314
3315 #if CONFIG_ERROR_RESILIENCE
3316 void ff_mpeg_er_frame_start(MpegEncContext *s)
3317 {
3318     ERContext *er = &s->er;
3319
3320     er->cur_pic  = s->current_picture_ptr;
3321     er->last_pic = s->last_picture_ptr;
3322     er->next_pic = s->next_picture_ptr;
3323
3324     er->pp_time           = s->pp_time;
3325     er->pb_time           = s->pb_time;
3326     er->quarter_sample    = s->quarter_sample;
3327     er->partitioned_frame = s->partitioned_frame;
3328
3329     ff_er_frame_start(er);
3330 }
3331 #endif /* CONFIG_ERROR_RESILIENCE */