mpegvideo.c 89.9 KB
Newer Older
Fabrice Bellard's avatar
Fabrice Bellard committed
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>
Fabrice Bellard's avatar
Fabrice Bellard committed
5
 *
6 7
 * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
 *
8 9 10
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
11 12
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
13
 * version 2.1 of the License, or (at your option) any later version.
Fabrice Bellard's avatar
Fabrice Bellard committed
14
 *
15
 * FFmpeg is distributed in the hope that it will be useful,
Fabrice Bellard's avatar
Fabrice Bellard committed
16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 18
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
Fabrice Bellard's avatar
Fabrice Bellard committed
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
Fabrice Bellard's avatar
Fabrice Bellard committed
23
 */
24

Michael Niedermayer's avatar
Michael Niedermayer committed
25 26 27
/**
 * @file mpegvideo.c
 * The simplest mpeg encoder (well, it was the simplest!).
28 29
 */

Fabrice Bellard's avatar
Fabrice Bellard committed
30 31 32
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"
33
#include "mpegvideo_common.h"
34
#include "mjpegenc.h"
35
#include "msmpeg4.h"
36
#include "faandct.h"
37
#include <limits.h>
Fabrice Bellard's avatar
Fabrice Bellard committed
38

39 40
//#undef NDEBUG
//#include <assert.h>
41

42
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
43
                                   DCTELEM *block, int n, int qscale);
44
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
45
                                   DCTELEM *block, int n, int qscale);
46 47
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
48 49
static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
50 51
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
52
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
53
                                  DCTELEM *block, int n, int qscale);
54
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
55
                                  DCTELEM *block, int n, int qscale);
56
static void draw_edges_c(uint8_t *buf, int wrap, int width, int height, int w);
57

Ivan Kalvachev's avatar
Ivan Kalvachev committed
58 59 60
#ifdef HAVE_XVMC
extern int  XVMC_field_start(MpegEncContext*s, AVCodecContext *avctx);
extern void XVMC_field_end(MpegEncContext *s);
61
extern void XVMC_decode_mb(MpegEncContext *s);
Ivan Kalvachev's avatar
Ivan Kalvachev committed
62 63
#endif

64
void (*draw_edges)(uint8_t *buf, int wrap, int width, int height, int w)= draw_edges_c;
65

Fabrice Bellard's avatar
Fabrice Bellard committed
66 67 68 69 70 71

/* enable all paranoid tests for rounding, overflows, etc... */
//#define PARANOID

//#define DEBUG

72

Michael Niedermayer's avatar
Michael Niedermayer committed
73 74 75 76 77
static const uint8_t ff_default_chroma_qscale_table[32]={
//  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
    0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31
};

Michael Niedermayer's avatar
Michael Niedermayer committed
78
void ff_init_scantable(uint8_t *permutation, ScanTable *st, const uint8_t *src_scantable){
79 80
    int i;
    int end;
81

82
    st->scantable= src_scantable;
83 84 85 86

    for(i=0; i<64; i++){
        int j;
        j = src_scantable[i];
Michael Niedermayer's avatar
Michael Niedermayer committed
87
        st->permutated[i] = permutation[j];
88 89 90
#ifdef ARCH_POWERPC
        st->inverse[j] = i;
#endif
91
    }
92

93 94 95 96 97 98 99 100 101
    end=-1;
    for(i=0; i<64; i++){
        int j;
        j = st->permutated[i];
        if(j>end) end=j;
        st->raster_end[i]= end;
    }
}

102 103 104
const uint8_t *ff_find_start_code(const uint8_t * restrict p, const uint8_t *end, uint32_t * restrict state){
    int i;

Michael Niedermayer's avatar
Michael Niedermayer committed
105 106 107 108
    assert(p<=end);
    if(p>=end)
        return end;

109 110 111 112 113 114 115 116
    for(i=0; i<3; i++){
        uint32_t tmp= *state << 8;
        *state= tmp + *(p++);
        if(tmp == 0x100 || p==end)
            return p;
    }

    while(p<end){
Michael Niedermayer's avatar
Michael Niedermayer committed
117 118 119
        if     (p[-1] > 1      ) p+= 3;
        else if(p[-2]          ) p+= 2;
        else if(p[-3]|(p[-1]-1)) p++;
120 121 122 123 124 125
        else{
            p++;
            break;
        }
    }

Michael Niedermayer's avatar
Michael Niedermayer committed
126
    p= FFMIN(p, end)-4;
127
    *state= AV_RB32(p);
128 129 130 131

    return p+4;
}

132
/* init common dct for both encoder and decoder */
133
int ff_dct_common_init(MpegEncContext *s)
Fabrice Bellard's avatar
Fabrice Bellard committed
134
{
135 136 137 138 139
    s->dct_unquantize_h263_intra = dct_unquantize_h263_intra_c;
    s->dct_unquantize_h263_inter = dct_unquantize_h263_inter_c;
    s->dct_unquantize_mpeg1_intra = dct_unquantize_mpeg1_intra_c;
    s->dct_unquantize_mpeg1_inter = dct_unquantize_mpeg1_inter_c;
    s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_c;
140 141
    if(s->flags & CODEC_FLAG_BITEXACT)
        s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_bitexact;
142
    s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_c;
143

144
#if defined(HAVE_MMX)
145
    MPV_common_init_mmx(s);
146
#elif defined(ARCH_ALPHA)
147
    MPV_common_init_axp(s);
148
#elif defined(HAVE_MLIB)
149
    MPV_common_init_mlib(s);
150
#elif defined(HAVE_MMI)
151
    MPV_common_init_mmi(s);
152
#elif defined(ARCH_ARMV4L)
153
    MPV_common_init_armv4l(s);
154
#elif defined(HAVE_ALTIVEC)
155
    MPV_common_init_altivec(s);
156
#elif defined(ARCH_BFIN)
157 158
    MPV_common_init_bfin(s);
#endif
159

160
    /* load & permutate scantables
161
       note: only wmv uses different ones
162
    */
163 164 165 166 167 168 169
    if(s->alternate_scan){
        ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_alternate_vertical_scan);
        ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_alternate_vertical_scan);
    }else{
        ff_init_scantable(s->dsp.idct_permutation, &s->inter_scantable  , ff_zigzag_direct);
        ff_init_scantable(s->dsp.idct_permutation, &s->intra_scantable  , ff_zigzag_direct);
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
170 171
    ff_init_scantable(s->dsp.idct_permutation, &s->intra_h_scantable, ff_alternate_horizontal_scan);
    ff_init_scantable(s->dsp.idct_permutation, &s->intra_v_scantable, ff_alternate_vertical_scan);
172

173 174 175
    return 0;
}

176
void copy_picture(Picture *dst, Picture *src){
177 178 179 180
    *dst = *src;
    dst->type= FF_BUFFER_TYPE_COPY;
}

Michael Niedermayer's avatar
Michael Niedermayer committed
181
/**
Michael Niedermayer's avatar
Michael Niedermayer committed
182 183
 * allocates a Picture
 * The pixels are allocated/set by calling get_buffer() if shared=0
Michael Niedermayer's avatar
Michael Niedermayer committed
184
 */
185
int alloc_picture(MpegEncContext *s, Picture *pic, int shared){
Diego Biurrun's avatar
Diego Biurrun committed
186
    const int big_mb_num= s->mb_stride*(s->mb_height+1) + 1; //the +1 is needed so memset(,,stride*height) does not sig11
187
    const int mb_array_size= s->mb_stride*s->mb_height;
188 189
    const int b8_array_size= s->b8_stride*s->mb_height*2;
    const int b4_array_size= s->b4_stride*s->mb_height*4;
190
    int i;
Daniel Kristjansson's avatar
Daniel Kristjansson committed
191
    int r= -1;
192

Michael Niedermayer's avatar
Michael Niedermayer committed
193 194 195 196 197 198
    if(shared){
        assert(pic->data[0]);
        assert(pic->type == 0 || pic->type == FF_BUFFER_TYPE_SHARED);
        pic->type= FF_BUFFER_TYPE_SHARED;
    }else{
        assert(!pic->data[0]);
199

200
        r= s->avctx->get_buffer(s->avctx, (AVFrame*)pic);
201

Michael Niedermayer's avatar
Michael Niedermayer committed
202
        if(r<0 || !pic->age || !pic->type || !pic->data[0]){
203
            av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (%d %d %d %p)\n", r, pic->age, pic->type, pic->data[0]);
Michael Niedermayer's avatar
Michael Niedermayer committed
204 205 206 207
            return -1;
        }

        if(s->linesize && (s->linesize != pic->linesize[0] || s->uvlinesize != pic->linesize[1])){
208
            av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (stride changed)\n");
Daniel Kristjansson's avatar
Daniel Kristjansson committed
209
            s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
Michael Niedermayer's avatar
Michael Niedermayer committed
210 211 212 213
            return -1;
        }

        if(pic->linesize[1] != pic->linesize[2]){
Diego Biurrun's avatar
Diego Biurrun committed
214
            av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (uv stride mismatch)\n");
Daniel Kristjansson's avatar
Daniel Kristjansson committed
215
            s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
Michael Niedermayer's avatar
Michael Niedermayer committed
216 217 218 219 220
            return -1;
        }

        s->linesize  = pic->linesize[0];
        s->uvlinesize= pic->linesize[1];
Michael Niedermayer's avatar
Michael Niedermayer committed
221
    }
222

Michael Niedermayer's avatar
Michael Niedermayer committed
223
    if(pic->qscale_table==NULL){
224
        if (s->encoding) {
225 226 227
            CHECKED_ALLOCZ(pic->mb_var   , mb_array_size * sizeof(int16_t))
            CHECKED_ALLOCZ(pic->mc_mb_var, mb_array_size * sizeof(int16_t))
            CHECKED_ALLOCZ(pic->mb_mean  , mb_array_size * sizeof(int8_t))
Michael Niedermayer's avatar
Michael Niedermayer committed
228
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
229

230
        CHECKED_ALLOCZ(pic->mbskip_table , mb_array_size * sizeof(uint8_t)+2) //the +2 is for the slice end check
231
        CHECKED_ALLOCZ(pic->qscale_table , mb_array_size * sizeof(uint8_t))
232 233
        CHECKED_ALLOCZ(pic->mb_type_base , (big_mb_num + s->mb_stride) * sizeof(uint32_t))
        pic->mb_type= pic->mb_type_base + 2*s->mb_stride+1;
234 235
        if(s->out_format == FMT_H264){
            for(i=0; i<2; i++){
Michael Niedermayer's avatar
Michael Niedermayer committed
236 237
                CHECKED_ALLOCZ(pic->motion_val_base[i], 2 * (b4_array_size+4)  * sizeof(int16_t))
                pic->motion_val[i]= pic->motion_val_base[i]+4;
238
                CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
239
            }
240
            pic->motion_subsample_log2= 2;
241
        }else if(s->out_format == FMT_H263 || s->encoding || (s->avctx->debug&FF_DEBUG_MV) || (s->avctx->debug_mv)){
242
            for(i=0; i<2; i++){
Michael Niedermayer's avatar
Michael Niedermayer committed
243 244
                CHECKED_ALLOCZ(pic->motion_val_base[i], 2 * (b8_array_size+4) * sizeof(int16_t))
                pic->motion_val[i]= pic->motion_val_base[i]+4;
245
                CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
246
            }
247
            pic->motion_subsample_log2= 3;
248
        }
249 250 251
        if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
            CHECKED_ALLOCZ(pic->dct_coeff, 64 * mb_array_size * sizeof(DCTELEM)*6)
        }
252
        pic->qstride= s->mb_stride;
253
        CHECKED_ALLOCZ(pic->pan_scan , 1 * sizeof(AVPanScan))
Michael Niedermayer's avatar
Michael Niedermayer committed
254
    }
255

Diego Biurrun's avatar
Diego Biurrun committed
256 257
    /* It might be nicer if the application would keep track of these
     * but it would require an API change. */
258 259 260
    memmove(s->prev_pict_types+1, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE-1);
    s->prev_pict_types[0]= s->pict_type;
    if(pic->age < PREV_PICT_TYPES_BUFFER_SIZE && s->prev_pict_types[pic->age] == B_TYPE)
Diego Biurrun's avatar
Diego Biurrun committed
261
        pic->age= INT_MAX; // Skipped MBs in B-frames are quite rare in MPEG-1/2 and it is a bit tricky to skip them anyway.
262

Michael Niedermayer's avatar
Michael Niedermayer committed
263 264
    return 0;
fail: //for the CHECKED_ALLOCZ macro
Daniel Kristjansson's avatar
Daniel Kristjansson committed
265 266
    if(r>=0)
        s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
Michael Niedermayer's avatar
Michael Niedermayer committed
267 268 269
    return -1;
}

Michael Niedermayer's avatar
Michael Niedermayer committed
270 271 272
/**
 * deallocates a picture
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
273 274
static void free_picture(MpegEncContext *s, Picture *pic){
    int i;
Michael Niedermayer's avatar
Michael Niedermayer committed
275 276

    if(pic->data[0] && pic->type!=FF_BUFFER_TYPE_SHARED){
277
        s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
Michael Niedermayer's avatar
Michael Niedermayer committed
278 279
    }

Michael Niedermayer's avatar
Michael Niedermayer committed
280 281 282 283 284
    av_freep(&pic->mb_var);
    av_freep(&pic->mc_mb_var);
    av_freep(&pic->mb_mean);
    av_freep(&pic->mbskip_table);
    av_freep(&pic->qscale_table);
285
    av_freep(&pic->mb_type_base);
286
    av_freep(&pic->dct_coeff);
287
    av_freep(&pic->pan_scan);
288 289
    pic->mb_type= NULL;
    for(i=0; i<2; i++){
290
        av_freep(&pic->motion_val_base[i]);
291 292
        av_freep(&pic->ref_index[i]);
    }
293

294
    if(pic->type == FF_BUFFER_TYPE_SHARED){
Michael Niedermayer's avatar
Michael Niedermayer committed
295 296 297 298
        for(i=0; i<4; i++){
            pic->base[i]=
            pic->data[i]= NULL;
        }
299
        pic->type= 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
300 301 302
    }
}

303 304 305
static int init_duplicate_context(MpegEncContext *s, MpegEncContext *base){
    int i;

306
    // edge emu needs blocksize + filter length - 1 (=17x17 for halfpel / 21x21 for h264)
307 308
    CHECKED_ALLOCZ(s->allocated_edge_emu_buffer, (s->width+64)*2*21*2); //(width + edge + align)*interlaced*MBsize*tolerance
    s->edge_emu_buffer= s->allocated_edge_emu_buffer + (s->width+64)*2*21;
309

Diego Biurrun's avatar
Diego Biurrun committed
310
     //FIXME should be linesize instead of s->width*2 but that is not known before get_buffer()
311
    CHECKED_ALLOCZ(s->me.scratchpad,  (s->width+64)*4*16*2*sizeof(uint8_t))
312 313 314 315 316 317 318 319 320
    s->rd_scratchpad=   s->me.scratchpad;
    s->b_scratchpad=    s->me.scratchpad;
    s->obmc_scratchpad= s->me.scratchpad + 16;
    if (s->encoding) {
        CHECKED_ALLOCZ(s->me.map      , ME_MAP_SIZE*sizeof(uint32_t))
        CHECKED_ALLOCZ(s->me.score_map, ME_MAP_SIZE*sizeof(uint32_t))
        if(s->avctx->noise_reduction){
            CHECKED_ALLOCZ(s->dct_error_sum, 2 * 64 * sizeof(int))
        }
321
    }
322
    CHECKED_ALLOCZ(s->blocks, 64*12*2 * sizeof(DCTELEM))
323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
    s->block= s->blocks[0];

    for(i=0;i<12;i++){
        s->pblocks[i] = (short *)(&s->block[i]);
    }
    return 0;
fail:
    return -1; //free() through MPV_common_end()
}

static void free_duplicate_context(MpegEncContext *s){
    if(s==NULL) return;

    av_freep(&s->allocated_edge_emu_buffer); s->edge_emu_buffer= NULL;
    av_freep(&s->me.scratchpad);
338 339
    s->rd_scratchpad=
    s->b_scratchpad=
340
    s->obmc_scratchpad= NULL;
341

342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365
    av_freep(&s->dct_error_sum);
    av_freep(&s->me.map);
    av_freep(&s->me.score_map);
    av_freep(&s->blocks);
    s->block= NULL;
}

static void backup_duplicate_context(MpegEncContext *bak, MpegEncContext *src){
#define COPY(a) bak->a= src->a
    COPY(allocated_edge_emu_buffer);
    COPY(edge_emu_buffer);
    COPY(me.scratchpad);
    COPY(rd_scratchpad);
    COPY(b_scratchpad);
    COPY(obmc_scratchpad);
    COPY(me.map);
    COPY(me.score_map);
    COPY(blocks);
    COPY(block);
    COPY(start_mb_y);
    COPY(end_mb_y);
    COPY(me.map_generation);
    COPY(pb);
    COPY(dct_error_sum);
366 367
    COPY(dct_count[0]);
    COPY(dct_count[1]);
368 369 370
#undef COPY
}

371
void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src){
372
    MpegEncContext bak;
373
    int i;
374 375 376 377 378
    //FIXME copy only needed parts
//START_TIMER
    backup_duplicate_context(&bak, dst);
    memcpy(dst, src, sizeof(MpegEncContext));
    backup_duplicate_context(dst, &bak);
379 380 381
    for(i=0;i<12;i++){
        dst->pblocks[i] = (short *)(&dst->block[i]);
    }
382 383 384
//STOP_TIMER("update_duplicate_context") //about 10k cycles / 0.01 sec for 1000frames on 1ghz with 2 threads
}

Michael Niedermayer's avatar
Michael Niedermayer committed
385 386 387 388
/**
 * sets the given MpegEncContext to common defaults (same for encoding and decoding).
 * the changed fields will not depend upon the prior state of the MpegEncContext.
 */
389
void MPV_common_defaults(MpegEncContext *s){
Michael Niedermayer's avatar
Michael Niedermayer committed
390 391 392 393 394 395 396 397 398 399 400 401
    s->y_dc_scale_table=
    s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
    s->chroma_qscale_table= ff_default_chroma_qscale_table;
    s->progressive_frame= 1;
    s->progressive_sequence= 1;
    s->picture_structure= PICT_FRAME;

    s->coded_picture_number = 0;
    s->picture_number = 0;
    s->input_picture_number = 0;

    s->picture_in_gop_number = 0;
402 403 404

    s->f_code = 1;
    s->b_code = 1;
Michael Niedermayer's avatar
Michael Niedermayer committed
405 406 407 408 409 410 411 412 413 414
}

/**
 * sets the given MpegEncContext to defaults for decoding.
 * the changed fields will not depend upon the prior state of the MpegEncContext.
 */
void MPV_decode_defaults(MpegEncContext *s){
    MPV_common_defaults(s);
}

415
/**
Michael Niedermayer's avatar
Michael Niedermayer committed
416 417 418
 * init common structure for both encoder and decoder.
 * this assumes that some variables like width/height are already set
 */
419 420
int MPV_common_init(MpegEncContext *s)
{
421
    int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y, threads;
422

423 424 425
    s->mb_height = (s->height + 15) / 16;

    if(s->avctx->thread_count > MAX_THREADS || (s->avctx->thread_count > s->mb_height && s->mb_height)){
Michael Niedermayer's avatar
Michael Niedermayer committed
426 427 428 429
        av_log(s->avctx, AV_LOG_ERROR, "too many threads\n");
        return -1;
    }

430 431 432
    if((s->width || s->height) && avcodec_check_dimensions(s->avctx, s->width, s->height))
        return -1;

433
    dsputil_init(&s->dsp, s->avctx);
434
    ff_dct_common_init(s);
435

436
    s->flags= s->avctx->flags;
437
    s->flags2= s->avctx->flags2;
438

Michael Niedermayer's avatar
Michael Niedermayer committed
439
    s->mb_width  = (s->width  + 15) / 16;
440
    s->mb_stride = s->mb_width + 1;
441 442
    s->b8_stride = s->mb_width*2 + 1;
    s->b4_stride = s->mb_width*4 + 1;
443
    mb_array_size= s->mb_height * s->mb_stride;
444
    mv_table_size= (s->mb_height+2) * s->mb_stride + 1;
445

446 447 448 449
    /* set chroma shifts */
    avcodec_get_chroma_sub_sample(s->avctx->pix_fmt,&(s->chroma_x_shift),
                                                    &(s->chroma_y_shift) );

450 451 452
    /* set default edge pos, will be overriden in decode_header if needed */
    s->h_edge_pos= s->mb_width*16;
    s->v_edge_pos= s->mb_height*16;
453 454

    s->mb_num = s->mb_width * s->mb_height;
455

456 457 458
    s->block_wrap[0]=
    s->block_wrap[1]=
    s->block_wrap[2]=
459
    s->block_wrap[3]= s->b8_stride;
460
    s->block_wrap[4]=
461
    s->block_wrap[5]= s->mb_stride;
462

463 464
    y_size = s->b8_stride * (2 * s->mb_height + 1);
    c_size = s->mb_stride * (s->mb_height + 1);
465
    yc_size = y_size + 2 * c_size;
466

467
    /* convert fourcc to upper case */
468
    s->codec_tag=          toupper( s->avctx->codec_tag     &0xFF)
469
                        + (toupper((s->avctx->codec_tag>>8 )&0xFF)<<8 )
470
                        + (toupper((s->avctx->codec_tag>>16)&0xFF)<<16)
471
                        + (toupper((s->avctx->codec_tag>>24)&0xFF)<<24);
472

473
    s->stream_codec_tag=          toupper( s->avctx->stream_codec_tag     &0xFF)
474
                               + (toupper((s->avctx->stream_codec_tag>>8 )&0xFF)<<8 )
475
                               + (toupper((s->avctx->stream_codec_tag>>16)&0xFF)<<16)
476 477
                               + (toupper((s->avctx->stream_codec_tag>>24)&0xFF)<<24);

478
    s->avctx->coded_frame= (AVFrame*)&s->current_picture;
Michael Niedermayer's avatar
Michael Niedermayer committed
479

480 481 482 483 484 485 486
    CHECKED_ALLOCZ(s->mb_index2xy, (s->mb_num+1)*sizeof(int)) //error ressilience code looks cleaner with this
    for(y=0; y<s->mb_height; y++){
        for(x=0; x<s->mb_width; x++){
            s->mb_index2xy[ x + y*s->mb_width ] = x + y*s->mb_stride;
        }
    }
    s->mb_index2xy[ s->mb_height*s->mb_width ] = (s->mb_height-1)*s->mb_stride + s->mb_width; //FIXME really needed?
487

488
    if (s->encoding) {
489
        /* Allocate MV tables */
490 491 492 493 494 495 496 497 498 499 500 501
        CHECKED_ALLOCZ(s->p_mv_table_base            , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_forw_mv_table_base       , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_back_mv_table_base       , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_bidir_forw_mv_table_base , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_bidir_back_mv_table_base , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_direct_mv_table_base     , mv_table_size * 2 * sizeof(int16_t))
        s->p_mv_table           = s->p_mv_table_base            + s->mb_stride + 1;
        s->b_forw_mv_table      = s->b_forw_mv_table_base       + s->mb_stride + 1;
        s->b_back_mv_table      = s->b_back_mv_table_base       + s->mb_stride + 1;
        s->b_bidir_forw_mv_table= s->b_bidir_forw_mv_table_base + s->mb_stride + 1;
        s->b_bidir_back_mv_table= s->b_bidir_back_mv_table_base + s->mb_stride + 1;
        s->b_direct_mv_table    = s->b_direct_mv_table_base     + s->mb_stride + 1;
502

503 504 505
        if(s->msmpeg4_version){
            CHECKED_ALLOCZ(s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int));
        }
506
        CHECKED_ALLOCZ(s->avctx->stats_out, 256);
507 508

        /* Allocate MB type table */
509
        CHECKED_ALLOCZ(s->mb_type  , mb_array_size * sizeof(uint16_t)) //needed for encoding
510

511
        CHECKED_ALLOCZ(s->lambda_table, mb_array_size * sizeof(int))
512

513 514
        CHECKED_ALLOCZ(s->q_intra_matrix, 64*32 * sizeof(int))
        CHECKED_ALLOCZ(s->q_inter_matrix, 64*32 * sizeof(int))
515 516
        CHECKED_ALLOCZ(s->q_intra_matrix16, 64*32*2 * sizeof(uint16_t))
        CHECKED_ALLOCZ(s->q_inter_matrix16, 64*32*2 * sizeof(uint16_t))
517 518
        CHECKED_ALLOCZ(s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
        CHECKED_ALLOCZ(s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
519

520 521 522
        if(s->avctx->noise_reduction){
            CHECKED_ALLOCZ(s->dct_offset, 2 * 64 * sizeof(uint16_t))
        }
523
    }
524 525
    CHECKED_ALLOCZ(s->picture, MAX_PICTURE_COUNT * sizeof(Picture))

526
    CHECKED_ALLOCZ(s->error_status_table, mb_array_size*sizeof(uint8_t))
527

528
    if(s->codec_id==CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){
529
        /* interlaced direct mode decoding tables */
530 531 532 533 534 535 536 537 538 539 540 541 542
            for(i=0; i<2; i++){
                int j, k;
                for(j=0; j<2; j++){
                    for(k=0; k<2; k++){
                        CHECKED_ALLOCZ(s->b_field_mv_table_base[i][j][k]     , mv_table_size * 2 * sizeof(int16_t))
                        s->b_field_mv_table[i][j][k]    = s->b_field_mv_table_base[i][j][k]     + s->mb_stride + 1;
                    }
                    CHECKED_ALLOCZ(s->b_field_select_table[i][j]     , mb_array_size * 2 * sizeof(uint8_t))
                    CHECKED_ALLOCZ(s->p_field_mv_table_base[i][j]     , mv_table_size * 2 * sizeof(int16_t))
                    s->p_field_mv_table[i][j]    = s->p_field_mv_table_base[i][j]     + s->mb_stride + 1;
                }
                CHECKED_ALLOCZ(s->p_field_select_table[i]      , mb_array_size * 2 * sizeof(uint8_t))
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
543
    }
544
    if (s->out_format == FMT_H263) {
Fabrice Bellard's avatar
Fabrice Bellard committed
545
        /* ac values */
546 547 548
        CHECKED_ALLOCZ(s->ac_val_base, yc_size * sizeof(int16_t) * 16);
        s->ac_val[0] = s->ac_val_base + s->b8_stride + 1;
        s->ac_val[1] = s->ac_val_base + y_size + s->mb_stride + 1;
Fabrice Bellard's avatar
Fabrice Bellard committed
549
        s->ac_val[2] = s->ac_val[1] + c_size;
550

Fabrice Bellard's avatar
Fabrice Bellard committed
551
        /* cbp values */
552 553
        CHECKED_ALLOCZ(s->coded_block_base, y_size);
        s->coded_block= s->coded_block_base + s->b8_stride + 1;
554

555
        /* cbp, ac_pred, pred_dir */
556 557
        CHECKED_ALLOCZ(s->cbp_table  , mb_array_size * sizeof(uint8_t))
        CHECKED_ALLOCZ(s->pred_dir_table, mb_array_size * sizeof(uint8_t))
558
    }
559

560 561 562
    if (s->h263_pred || s->h263_plus || !s->encoding) {
        /* dc values */
        //MN: we need these for error resilience of intra-frames
563 564 565
        CHECKED_ALLOCZ(s->dc_val_base, yc_size * sizeof(int16_t));
        s->dc_val[0] = s->dc_val_base + s->b8_stride + 1;
        s->dc_val[1] = s->dc_val_base + y_size + s->mb_stride + 1;
566 567
        s->dc_val[2] = s->dc_val[1] + c_size;
        for(i=0;i<yc_size;i++)
568
            s->dc_val_base[i] = 1024;
569 570
    }

571
    /* which mb is a intra block */
572 573
    CHECKED_ALLOCZ(s->mbintra_table, mb_array_size);
    memset(s->mbintra_table, 1, mb_array_size);
574

575
    /* init macroblock skip table */
576
    CHECKED_ALLOCZ(s->mbskip_table, mb_array_size+2);
577
    //Note the +1 is for a quicker mpeg4 slice_end detection
578
    CHECKED_ALLOCZ(s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE);
579

580
    s->parse_context.state= -1;
581 582 583 584 585
    if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
       s->visualization_buffer[0] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
       s->visualization_buffer[1] = av_malloc((s->mb_width*8 + EDGE_WIDTH) * s->mb_height*8 + EDGE_WIDTH);
       s->visualization_buffer[2] = av_malloc((s->mb_width*8 + EDGE_WIDTH) * s->mb_height*8 + EDGE_WIDTH);
    }
586

Fabrice Bellard's avatar
Fabrice Bellard committed
587
    s->context_initialized = 1;
588 589

    s->thread_context[0]= s;
590
    threads = s->avctx->thread_count;
591 592

    for(i=1; i<threads; i++){
593 594 595 596
        s->thread_context[i]= av_malloc(sizeof(MpegEncContext));
        memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
    }

597
    for(i=0; i<threads; i++){
598 599 600 601 602 603
        if(init_duplicate_context(s->thread_context[i], s) < 0)
           goto fail;
        s->thread_context[i]->start_mb_y= (s->mb_height*(i  ) + s->avctx->thread_count/2) / s->avctx->thread_count;
        s->thread_context[i]->end_mb_y  = (s->mb_height*(i+1) + s->avctx->thread_count/2) / s->avctx->thread_count;
    }

Fabrice Bellard's avatar
Fabrice Bellard committed
604 605
    return 0;
 fail:
606
    MPV_common_end(s);
Fabrice Bellard's avatar
Fabrice Bellard committed
607 608 609 610 611 612
    return -1;
}

/* init common structure for both encoder and decoder */
void MPV_common_end(MpegEncContext *s)
{
613
    int i, j, k;
Fabrice Bellard's avatar
Fabrice Bellard committed
614

615 616 617 618 619 620 621
    for(i=0; i<s->avctx->thread_count; i++){
        free_duplicate_context(s->thread_context[i]);
    }
    for(i=1; i<s->avctx->thread_count; i++){
        av_freep(&s->thread_context[i]);
    }

622 623 624
    av_freep(&s->parse_context.buffer);
    s->parse_context.buffer_size=0;

625
    av_freep(&s->mb_type);
626 627 628 629 630 631 632 633 634 635 636 637
    av_freep(&s->p_mv_table_base);
    av_freep(&s->b_forw_mv_table_base);
    av_freep(&s->b_back_mv_table_base);
    av_freep(&s->b_bidir_forw_mv_table_base);
    av_freep(&s->b_bidir_back_mv_table_base);
    av_freep(&s->b_direct_mv_table_base);
    s->p_mv_table= NULL;
    s->b_forw_mv_table= NULL;
    s->b_back_mv_table= NULL;
    s->b_bidir_forw_mv_table= NULL;
    s->b_bidir_back_mv_table= NULL;
    s->b_direct_mv_table= NULL;
638 639 640 641 642 643 644 645 646 647 648 649
    for(i=0; i<2; i++){
        for(j=0; j<2; j++){
            for(k=0; k<2; k++){
                av_freep(&s->b_field_mv_table_base[i][j][k]);
                s->b_field_mv_table[i][j][k]=NULL;
            }
            av_freep(&s->b_field_select_table[i][j]);
            av_freep(&s->p_field_mv_table_base[i][j]);
            s->p_field_mv_table[i][j]=NULL;
        }
        av_freep(&s->p_field_select_table[i]);
    }
650

651 652 653
    av_freep(&s->dc_val_base);
    av_freep(&s->ac_val_base);
    av_freep(&s->coded_block_base);
654
    av_freep(&s->mbintra_table);
655 656
    av_freep(&s->cbp_table);
    av_freep(&s->pred_dir_table);
657

658
    av_freep(&s->mbskip_table);
659
    av_freep(&s->prev_pict_types);
660
    av_freep(&s->bitstream_buffer);
661 662
    s->allocated_bitstream_buffer_size=0;

663
    av_freep(&s->avctx->stats_out);
664
    av_freep(&s->ac_stats);
665
    av_freep(&s->error_status_table);
666
    av_freep(&s->mb_index2xy);
667
    av_freep(&s->lambda_table);
668 669
    av_freep(&s->q_intra_matrix);
    av_freep(&s->q_inter_matrix);
670 671
    av_freep(&s->q_intra_matrix16);
    av_freep(&s->q_inter_matrix16);
672 673
    av_freep(&s->input_picture);
    av_freep(&s->reordered_input_picture);
674
    av_freep(&s->dct_offset);
Michael Niedermayer's avatar
Michael Niedermayer committed
675

Michael Niedermayer's avatar
Michael Niedermayer committed
676 677 678 679
    if(s->picture){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            free_picture(s, &s->picture[i]);
        }
Fabrice Bellard's avatar
Fabrice Bellard committed
680
    }
681
    av_freep(&s->picture);
Fabrice Bellard's avatar
Fabrice Bellard committed
682
    s->context_initialized = 0;
683 684 685
    s->last_picture_ptr=
    s->next_picture_ptr=
    s->current_picture_ptr= NULL;
686
    s->linesize= s->uvlinesize= 0;
687

688
    for(i=0; i<3; i++)
689
        av_freep(&s->visualization_buffer[i]);
690 691

    avcodec_default_free_buffers(s->avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
692 693
}

694
void init_rl(RLTable *rl, uint8_t static_store[2][2*MAX_RUN + MAX_LEVEL + 3])
695
{
696 697
    int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
    uint8_t index_run[MAX_RUN+1];
698 699
    int last, run, level, start, end, i;

700
    /* If table is static, we can quit if rl->max_level[0] is not NULL */
701
    if(static_store && rl->max_level[0])
702 703
        return;

704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726
    /* compute max_level[], max_run[] and index_run[] */
    for(last=0;last<2;last++) {
        if (last == 0) {
            start = 0;
            end = rl->last;
        } else {
            start = rl->last;
            end = rl->n;
        }

        memset(max_level, 0, MAX_RUN + 1);
        memset(max_run, 0, MAX_LEVEL + 1);
        memset(index_run, rl->n, MAX_RUN + 1);
        for(i=start;i<end;i++) {
            run = rl->table_run[i];
            level = rl->table_level[i];
            if (index_run[run] == rl->n)
                index_run[run] = i;
            if (level > max_level[run])
                max_level[run] = level;
            if (run > max_run[level])
                max_run[level] = run;
        }
727 728
        if(static_store)
            rl->max_level[last] = static_store[last];
729 730
        else
            rl->max_level[last] = av_malloc(MAX_RUN + 1);
731
        memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
732 733
        if(static_store)
            rl->max_run[last] = static_store[last] + MAX_RUN + 1;
734 735
        else
            rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
736
        memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
737 738
        if(static_store)
            rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
739 740
        else
            rl->index_run[last] = av_malloc(MAX_RUN + 1);
741 742 743 744
        memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
    }
}

745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797
void init_vlc_rl(RLTable *rl, int use_static)
{
    int i, q;

    /* Return if static table is already initialized */
    if(use_static && rl->rl_vlc[0])
        return;

    init_vlc(&rl->vlc, 9, rl->n + 1,
             &rl->table_vlc[0][1], 4, 2,
             &rl->table_vlc[0][0], 4, 2, use_static);


    for(q=0; q<32; q++){
        int qmul= q*2;
        int qadd= (q-1)|1;

        if(q==0){
            qmul=1;
            qadd=0;
        }
        if(use_static)
            rl->rl_vlc[q]= av_mallocz_static(rl->vlc.table_size*sizeof(RL_VLC_ELEM));
        else
            rl->rl_vlc[q]= av_malloc(rl->vlc.table_size*sizeof(RL_VLC_ELEM));
        for(i=0; i<rl->vlc.table_size; i++){
            int code= rl->vlc.table[i][0];
            int len = rl->vlc.table[i][1];
            int level, run;

            if(len==0){ // illegal code
                run= 66;
                level= MAX_LEVEL;
            }else if(len<0){ //more bits needed
                run= 0;
                level= code;
            }else{
                if(code==rl->n){ //esc
                    run= 66;
                    level= 0;
                }else{
                    run=   rl->table_run  [code] + 1;
                    level= rl->table_level[code] * qmul + qadd;
                    if(code >= rl->last) run+=192;
                }
            }
            rl->rl_vlc[q][i].len= len;
            rl->rl_vlc[q][i].level= level;
            rl->rl_vlc[q][i].run= run;
        }
    }
}

Fabrice Bellard's avatar
Fabrice Bellard committed
798
/* draw the edges of width 'w' of an image of size width, height */
799
//FIXME check that this is ok for mpeg4 interlaced
800
static void draw_edges_c(uint8_t *buf, int wrap, int width, int height, int w)
Fabrice Bellard's avatar
Fabrice Bellard committed
801
{
802
    uint8_t *ptr, *last_line;
Fabrice Bellard's avatar
Fabrice Bellard committed
803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826
    int i;

    last_line = buf + (height - 1) * wrap;
    for(i=0;i<w;i++) {
        /* top and bottom */
        memcpy(buf - (i + 1) * wrap, buf, width);
        memcpy(last_line + (i + 1) * wrap, last_line, width);
    }
    /* left and right */
    ptr = buf;
    for(i=0;i<height;i++) {
        memset(ptr - w, ptr[0], w);
        memset(ptr + width, ptr[width-1], w);
        ptr += wrap;
    }
    /* corners */
    for(i=0;i<w;i++) {
        memset(buf - (i + 1) * wrap - w, buf[0], w); /* top left */
        memset(buf - (i + 1) * wrap + width, buf[width-1], w); /* top right */
        memset(last_line + (i + 1) * wrap - w, last_line[0], w); /* top left */
        memset(last_line + (i + 1) * wrap + width, last_line[width-1], w); /* top right */
    }
}

827
int ff_find_unused_picture(MpegEncContext *s, int shared){
Michael Niedermayer's avatar
Michael Niedermayer committed
828
    int i;
829

Michael Niedermayer's avatar
Michael Niedermayer committed
830 831
    if(shared){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
832
            if(s->picture[i].data[0]==NULL && s->picture[i].type==0) return i;
Michael Niedermayer's avatar
Michael Niedermayer committed
833 834 835
        }
    }else{
        for(i=0; i<MAX_PICTURE_COUNT; i++){
836
            if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) return i; //FIXME
Michael Niedermayer's avatar
Michael Niedermayer committed
837 838
        }
        for(i=0; i<MAX_PICTURE_COUNT; i++){
839
            if(s->picture[i].data[0]==NULL) return i;
Michael Niedermayer's avatar
Michael Niedermayer committed
840 841 842
        }
    }

843
    av_log(s->avctx, AV_LOG_FATAL, "Internal error, picture buffer overflow\n");
844 845 846 847 848 849 850 851 852 853 854
    /* We could return -1, but the codec would crash trying to draw into a
     * non-existing frame anyway. This is safer than waiting for a random crash.
     * Also the return of this is never useful, an encoder must only allocate
     * as much as allowed in the specification. This has no relationship to how
     * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
     * enough for such valid streams).
     * Plus, a decoder has to check stream validity and remove frames if too
     * many reference frames are around. Waiting for "OOM" is not correct at
     * all. Similarly, missing reference frames have to be replaced by
     * interpolated/MC frames, anything else is a bug in the codec ...
     */
855
    abort();
856
    return -1;
Michael Niedermayer's avatar
Michael Niedermayer committed
857 858
}

859 860 861 862 863 864 865 866 867 868
static void update_noise_reduction(MpegEncContext *s){
    int intra, i;

    for(intra=0; intra<2; intra++){
        if(s->dct_count[intra] > (1<<16)){
            for(i=0; i<64; i++){
                s->dct_error_sum[intra][i] >>=1;
            }
            s->dct_count[intra] >>= 1;
        }
869

870 871 872 873 874 875
        for(i=0; i<64; i++){
            s->dct_offset[intra][i]= (s->avctx->noise_reduction * s->dct_count[intra] + s->dct_error_sum[intra][i]/2) / (s->dct_error_sum[intra][i]+1);
        }
    }
}

876 877 878
/**
 * generic function for encode/decode called after coding/decoding the header and before a frame is coded/decoded
 */
879
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
Fabrice Bellard's avatar
Fabrice Bellard committed
880
{
Michael Niedermayer's avatar
Michael Niedermayer committed
881
    int i;
882
    AVFrame *pic;
883
    s->mb_skipped = 0;
884

885
    assert(s->last_picture_ptr==NULL || s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3);
886

Michael Niedermayer's avatar
Michael Niedermayer committed
887
    /* mark&release old frames */
888
    if (s->pict_type != B_TYPE && s->last_picture_ptr && s->last_picture_ptr != s->next_picture_ptr && s->last_picture_ptr->data[0]) {
889
      if(s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3){
890
        avctx->release_buffer(avctx, (AVFrame*)s->last_picture_ptr);
Michael Niedermayer's avatar
Michael Niedermayer committed
891 892 893 894 895

        /* release forgotten pictures */
        /* if(mpeg124/h263) */
        if(!s->encoding){
            for(i=0; i<MAX_PICTURE_COUNT; i++){
896
                if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
897
                    av_log(avctx, AV_LOG_ERROR, "releasing zombie picture\n");
898
                    avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
Michael Niedermayer's avatar
Michael Niedermayer committed
899 900
                }
            }
901
        }
902
      }
903
    }
904 905
alloc:
    if(!s->encoding){
Diego Biurrun's avatar
Diego Biurrun committed
906
        /* release non reference frames */
907 908 909 910 911 912
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
                s->avctx->release_buffer(s->avctx, (AVFrame*)&s->picture[i]);
            }
        }

913 914 915 916 917 918 919
        if(s->current_picture_ptr && s->current_picture_ptr->data[0]==NULL)
            pic= (AVFrame*)s->current_picture_ptr; //we allready have a unused image (maybe it was set before reading the header)
        else{
            i= ff_find_unused_picture(s, 0);
            pic= (AVFrame*)&s->picture[i];
        }

920 921 922 923 924 925 926
        pic->reference= 0;
        if (!s->dropable){
            if (s->codec_id == CODEC_ID_H264)
                pic->reference = s->picture_structure;
            else if (s->pict_type != B_TYPE)
                pic->reference = 3;
        }
927

928
        pic->coded_picture_number= s->coded_picture_number++;
929

930 931
        if( alloc_picture(s, (Picture*)pic, 0) < 0)
            return -1;
932

933
        s->current_picture_ptr= (Picture*)pic;
934
        s->current_picture_ptr->top_field_first= s->top_field_first; //FIXME use only the vars from current_pic
935
        s->current_picture_ptr->interlaced_frame= !s->progressive_frame && !s->progressive_sequence;
Michael Niedermayer's avatar
Michael Niedermayer committed
936
    }
937

938
    s->current_picture_ptr->pict_type= s->pict_type;
939
//    if(s->flags && CODEC_FLAG_QSCALE)
940
  //      s->current_picture_ptr->quality= s->new_picture_ptr->quality;
941
    s->current_picture_ptr->key_frame= s->pict_type == I_TYPE;
942

943
    copy_picture(&s->current_picture, s->current_picture_ptr);
944

Michael Niedermayer's avatar
Michael Niedermayer committed
945
    if (s->pict_type != B_TYPE) {
946
        s->last_picture_ptr= s->next_picture_ptr;
947 948
        if(!s->dropable)
            s->next_picture_ptr= s->current_picture_ptr;
Fabrice Bellard's avatar
Fabrice Bellard committed
949
    }
950
/*    av_log(s->avctx, AV_LOG_DEBUG, "L%p N%p C%p L%p N%p C%p type:%d drop:%d\n", s->last_picture_ptr, s->next_picture_ptr,s->current_picture_ptr,
951 952
        s->last_picture_ptr    ? s->last_picture_ptr->data[0] : NULL,
        s->next_picture_ptr    ? s->next_picture_ptr->data[0] : NULL,
953 954
        s->current_picture_ptr ? s->current_picture_ptr->data[0] : NULL,
        s->pict_type, s->dropable);*/
955

956 957
    if(s->last_picture_ptr) copy_picture(&s->last_picture, s->last_picture_ptr);
    if(s->next_picture_ptr) copy_picture(&s->next_picture, s->next_picture_ptr);
958

959
    if(s->pict_type != I_TYPE && (s->last_picture_ptr==NULL || s->last_picture_ptr->data[0]==NULL) && !s->dropable){
960
        av_log(avctx, AV_LOG_ERROR, "warning: first frame is no keyframe\n");
Diego Biurrun's avatar
Diego Biurrun committed
961
        assert(s->pict_type != B_TYPE); //these should have been dropped if we don't have a reference
962 963 964 965 966
        goto alloc;
    }

    assert(s->pict_type == I_TYPE || (s->last_picture_ptr && s->last_picture_ptr->data[0]));

967
    if(s->picture_structure!=PICT_FRAME && s->out_format != FMT_H264){
968 969 970 971
        int i;
        for(i=0; i<4; i++){
            if(s->picture_structure == PICT_BOTTOM_FIELD){
                 s->current_picture.data[i] += s->current_picture.linesize[i];
972
            }
973 974 975 976 977
            s->current_picture.linesize[i] *= 2;
            s->last_picture.linesize[i] *=2;
            s->next_picture.linesize[i] *=2;
        }
    }
978

979 980 981
    s->hurry_up= s->avctx->hurry_up;
    s->error_resilience= avctx->error_resilience;

Diego Biurrun's avatar
Diego Biurrun committed
982
    /* set dequantizer, we can't do it during init as it might change for mpeg4
Diego Biurrun's avatar
Diego Biurrun committed
983
       and we can't do it in the header decode as init is not called for mpeg4 there yet */
984 985 986
    if(s->mpeg_quant || s->codec_id == CODEC_ID_MPEG2VIDEO){
        s->dct_unquantize_intra = s->dct_unquantize_mpeg2_intra;
        s->dct_unquantize_inter = s->dct_unquantize_mpeg2_inter;
987
    }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
988 989 990 991 992 993
        s->dct_unquantize_intra = s->dct_unquantize_h263_intra;
        s->dct_unquantize_inter = s->dct_unquantize_h263_inter;
    }else{
        s->dct_unquantize_intra = s->dct_unquantize_mpeg1_intra;
        s->dct_unquantize_inter = s->dct_unquantize_mpeg1_inter;
    }
994

995 996 997 998 999
    if(s->dct_error_sum){
        assert(s->avctx->noise_reduction && s->encoding);

        update_noise_reduction(s);
    }
1000

Ivan Kalvachev's avatar
Ivan Kalvachev committed
1001 1002 1003 1004
#ifdef HAVE_XVMC
    if(s->avctx->xvmc_acceleration)
        return XVMC_field_start(s, avctx);
#endif
1005
    return 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
1006
}
1007

Fabrice Bellard's avatar
Fabrice Bellard committed
1008 1009 1010
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
Michael Niedermayer's avatar
Michael Niedermayer committed
1011
    int i;
Fabrice Bellard's avatar
Fabrice Bellard committed
1012
    /* draw edge for correct motion prediction if outside */
Ivan Kalvachev's avatar
Ivan Kalvachev committed
1013 1014 1015 1016 1017 1018
#ifdef HAVE_XVMC
//just to make sure that all data is rendered.
    if(s->avctx->xvmc_acceleration){
        XVMC_field_end(s);
    }else
#endif
1019
    if(s->unrestricted_mv && s->current_picture.reference && !s->intra_only && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
Michael Niedermayer's avatar
Michael Niedermayer committed
1020 1021 1022
            draw_edges(s->current_picture.data[0], s->linesize  , s->h_edge_pos   , s->v_edge_pos   , EDGE_WIDTH  );
            draw_edges(s->current_picture.data[1], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
            draw_edges(s->current_picture.data[2], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
Fabrice Bellard's avatar
Fabrice Bellard committed
1023
    }
1024
    emms_c();
1025

1026
    s->last_pict_type    = s->pict_type;
1027
    s->last_lambda_for[s->pict_type]= s->current_picture_ptr->quality;
1028 1029 1030
    if(s->pict_type!=B_TYPE){
        s->last_non_b_pict_type= s->pict_type;
    }
1031 1032
#if 0
        /* copy back current_picture variables */
Michael Niedermayer's avatar
Michael Niedermayer committed
1033 1034 1035 1036
    for(i=0; i<MAX_PICTURE_COUNT; i++){
        if(s->picture[i].data[0] == s->current_picture.data[0]){
            s->picture[i]= s->current_picture;
            break;
1037
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
1038 1039
    }
    assert(i<MAX_PICTURE_COUNT);
1040
#endif
Michael Niedermayer's avatar
Michael Niedermayer committed
1041

1042
    if(s->encoding){
Diego Biurrun's avatar
Diego Biurrun committed
1043
        /* release non-reference frames */
1044 1045 1046 1047 1048
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
                s->avctx->release_buffer(s->avctx, (AVFrame*)&s->picture[i]);
            }
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
1049
    }
1050 1051 1052 1053 1054 1055
    // clear copies, to avoid confusion
#if 0
    memset(&s->last_picture, 0, sizeof(Picture));
    memset(&s->next_picture, 0, sizeof(Picture));
    memset(&s->current_picture, 0, sizeof(Picture));
#endif
1056
    s->avctx->coded_frame= (AVFrame*)s->current_picture_ptr;
Fabrice Bellard's avatar
Fabrice Bellard committed
1057 1058
}

1059 1060 1061 1062 1063 1064 1065 1066
/**
 * draws an line from (ex, ey) -> (sx, sy).
 * @param w width of the image
 * @param h height of the image
 * @param stride stride/linesize of the image
 * @param color color of the arrow
 */
static void draw_line(uint8_t *buf, int sx, int sy, int ex, int ey, int w, int h, int stride, int color){
Michael Niedermayer's avatar
Michael Niedermayer committed
1067
    int x, y, fr, f;
1068

1069 1070 1071 1072
    sx= av_clip(sx, 0, w-1);
    sy= av_clip(sy, 0, h-1);
    ex= av_clip(ex, 0, w-1);
    ey= av_clip(ey, 0, h-1);
1073

1074
    buf[sy*stride + sx]+= color;
1075

1076
    if(FFABS(ex - sx) > FFABS(ey - sy)){
1077
        if(sx > ex){
1078 1079
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1080 1081 1082 1083 1084
        }
        buf+= sx + sy*stride;
        ex-= sx;
        f= ((ey-sy)<<16)/ex;
        for(x= 0; x <= ex; x++){
1085 1086 1087 1088
            y = (x*f)>>16;
            fr= (x*f)&0xFFFF;
            buf[ y   *stride + x]+= (color*(0x10000-fr))>>16;
            buf[(y+1)*stride + x]+= (color*         fr )>>16;
1089 1090 1091
        }
    }else{
        if(sy > ey){
1092 1093
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1094 1095 1096 1097 1098 1099
        }
        buf+= sx + sy*stride;
        ey-= sy;
        if(ey) f= ((ex-sx)<<16)/ey;
        else   f= 0;
        for(y= 0; y <= ey; y++){
1100 1101 1102 1103
            x = (y*f)>>16;
            fr= (y*f)&0xFFFF;
            buf[y*stride + x  ]+= (color*(0x10000-fr))>>16;;
            buf[y*stride + x+1]+= (color*         fr )>>16;;
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
        }
    }
}

/**
 * draws an arrow from (ex, ey) -> (sx, sy).
 * @param w width of the image
 * @param h height of the image
 * @param stride stride/linesize of the image
 * @param color color of the arrow
 */
1115
static void draw_arrow(uint8_t *buf, int sx, int sy, int ex, int ey, int w, int h, int stride, int color){
1116 1117
    int dx,dy;

1118 1119 1120 1121
    sx= av_clip(sx, -100, w+100);
    sy= av_clip(sy, -100, h+100);
    ex= av_clip(ex, -100, w+100);
    ey= av_clip(ey, -100, h+100);
1122

1123 1124
    dx= ex - sx;
    dy= ey - sy;
1125

1126 1127 1128 1129
    if(dx*dx + dy*dy > 3*3){
        int rx=  dx + dy;
        int ry= -dx + dy;
        int length= ff_sqrt((rx*rx + ry*ry)<<8);
1130

1131 1132 1133
        //FIXME subpixel accuracy
        rx= ROUNDED_DIV(rx*3<<4, length);
        ry= ROUNDED_DIV(ry*3<<4, length);
1134

1135 1136 1137 1138 1139 1140
        draw_line(buf, sx, sy, sx + rx, sy + ry, w, h, stride, color);
        draw_line(buf, sx, sy, sx - ry, sy + rx, w, h, stride, color);
    }
    draw_line(buf, sx, sy, ex, ey, w, h, stride, color);
}

1141 1142 1143
/**
 * prints debuging info for the given picture.
 */
1144
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
1145 1146 1147 1148 1149

    if(!pict || !pict->mb_type) return;

    if(s->avctx->debug&(FF_DEBUG_SKIP | FF_DEBUG_QP | FF_DEBUG_MB_TYPE)){
        int x,y;
1150

1151 1152 1153 1154 1155 1156 1157
        av_log(s->avctx,AV_LOG_DEBUG,"New frame, type: ");
        switch (pict->pict_type) {
            case FF_I_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"I\n"); break;
            case FF_P_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"P\n"); break;
            case FF_B_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"B\n"); break;
            case FF_S_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"S\n"); break;
            case FF_SI_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SI\n"); break;
1158
            case FF_SP_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SP\n"); break;
1159
        }
1160 1161 1162 1163 1164
        for(y=0; y<s->mb_height; y++){
            for(x=0; x<s->mb_width; x++){
                if(s->avctx->debug&FF_DEBUG_SKIP){
                    int count= s->mbskip_table[x + y*s->mb_stride];
                    if(count>9) count=9;
1165
                    av_log(s->avctx, AV_LOG_DEBUG, "%1d", count);
1166 1167
                }
                if(s->avctx->debug&FF_DEBUG_QP){
1168
                    av_log(s->avctx, AV_LOG_DEBUG, "%2d", pict->qscale_table[x + y*s->mb_stride]);
1169 1170 1171 1172 1173
                }
                if(s->avctx->debug&FF_DEBUG_MB_TYPE){
                    int mb_type= pict->mb_type[x + y*s->mb_stride];
                    //Type & MV direction
                    if(IS_PCM(mb_type))
1174
                        av_log(s->avctx, AV_LOG_DEBUG, "P");
1175
                    else if(IS_INTRA(mb_type) && IS_ACPRED(mb_type))
1176
                        av_log(s->avctx, AV_LOG_DEBUG, "A");
1177
                    else if(IS_INTRA4x4(mb_type))
1178
                        av_log(s->avctx, AV_LOG_DEBUG, "i");
1179
                    else if(IS_INTRA16x16(mb_type))
1180
                        av_log(s->avctx, AV_LOG_DEBUG, "I");
1181
                    else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type))
1182
                        av_log(s->avctx, AV_LOG_DEBUG, "d");
1183
                    else if(IS_DIRECT(mb_type))
1184
                        av_log(s->avctx, AV_LOG_DEBUG, "D");
1185
                    else if(IS_GMC(mb_type) && IS_SKIP(mb_type))
1186
                        av_log(s->avctx, AV_LOG_DEBUG, "g");
1187
                    else if(IS_GMC(mb_type))
1188
                        av_log(s->avctx, AV_LOG_DEBUG, "G");
1189
                    else if(IS_SKIP(mb_type))
1190
                        av_log(s->avctx, AV_LOG_DEBUG, "S");
1191
                    else if(!USES_LIST(mb_type, 1))
1192
                        av_log(s->avctx, AV_LOG_DEBUG, ">");
1193
                    else if(!USES_LIST(mb_type, 0))
1194
                        av_log(s->avctx, AV_LOG_DEBUG, "<");
1195 1196
                    else{
                        assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
1197
                        av_log(s->avctx, AV_LOG_DEBUG, "X");
1198
                    }
1199

1200 1201
                    //segmentation
                    if(IS_8X8(mb_type))
1202
                        av_log(s->avctx, AV_LOG_DEBUG, "+");
1203
                    else if(IS_16X8(mb_type))
1204
                        av_log(s->avctx, AV_LOG_DEBUG, "-");
1205
                    else if(IS_8X16(mb_type))
1206
                        av_log(s->avctx, AV_LOG_DEBUG, "|");
1207
                    else if(IS_INTRA(mb_type) || IS_16X16(mb_type))
1208
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1209
                    else
1210
                        av_log(s->avctx, AV_LOG_DEBUG, "?");
1211 1212


1213
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264)
1214
                        av_log(s->avctx, AV_LOG_DEBUG, "=");
1215
                    else
1216
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1217
                }
1218
//                av_log(s->avctx, AV_LOG_DEBUG, " ");
1219
            }
1220
            av_log(s->avctx, AV_LOG_DEBUG, "\n");
1221 1222
        }
    }
1223

1224
    if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
1225 1226
        const int shift= 1 + s->quarter_sample;
        int mb_y;
1227 1228
        uint8_t *ptr;
        int i;
1229
        int h_chroma_shift, v_chroma_shift;
1230 1231
        const int width = s->avctx->width;
        const int height= s->avctx->height;
1232
        const int mv_sample_log2= 4 - pict->motion_subsample_log2;
1233
        const int mv_stride= (s->mb_width << mv_sample_log2) + (s->codec_id == CODEC_ID_H264 ? 0 : 1);
Michael Niedermayer's avatar
Michael Niedermayer committed
1234
        s->low_delay=0; //needed to see the vectors without trashing the buffers
1235

1236
        avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1237
        for(i=0; i<3; i++){
1238
            memcpy(s->visualization_buffer[i], pict->data[i], (i==0) ? pict->linesize[i]*height:pict->linesize[i]*height >> v_chroma_shift);
1239 1240 1241 1242
            pict->data[i]= s->visualization_buffer[i];
        }
        pict->type= FF_BUFFER_TYPE_COPY;
        ptr= pict->data[0];
1243 1244 1245 1246 1247

        for(mb_y=0; mb_y<s->mb_height; mb_y++){
            int mb_x;
            for(mb_x=0; mb_x<s->mb_width; mb_x++){
                const int mb_index= mb_x + mb_y*s->mb_stride;
1248 1249 1250
                if((s->avctx->debug_mv) && pict->motion_val){
                  int type;
                  for(type=0; type<3; type++){
1251
                    int direction = 0;
1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
                    switch (type) {
                      case 0: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_P_FOR)) || (pict->pict_type!=FF_P_TYPE))
                                continue;
                              direction = 0;
                              break;
                      case 1: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_FOR)) || (pict->pict_type!=FF_B_TYPE))
                                continue;
                              direction = 0;
                              break;
                      case 2: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_BACK)) || (pict->pict_type!=FF_B_TYPE))
                                continue;
                              direction = 1;
                              break;
                    }
1266 1267 1268
                    if(!USES_LIST(pict->mb_type[mb_index], direction))
                        continue;

1269 1270 1271
                    if(IS_8X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<4; i++){
1272 1273
                        int sx= mb_x*16 + 4 + 8*(i&1);
                        int sy= mb_y*16 + 4 + 8*(i>>1);
1274
                        int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
1275 1276
                        int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                        int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1277
                        draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1278 1279 1280 1281
                      }
                    }else if(IS_16X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<2; i++){
1282 1283
                        int sx=mb_x*16 + 8;
                        int sy=mb_y*16 + 4 + 8*i;
1284
                        int xy= (mb_x*2 + (mb_y*2 + i)*mv_stride) << (mv_sample_log2-1);
1285 1286
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1287

1288 1289
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1290

1291 1292 1293 1294 1295 1296 1297
                        draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
                      }
                    }else if(IS_8X16(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<2; i++){
                        int sx=mb_x*16 + 4 + 8*i;
                        int sy=mb_y*16 + 8;
1298
                        int xy= (mb_x*2 + i + mb_y*2*mv_stride) << (mv_sample_log2-1);
1299 1300
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1301

1302 1303
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1304

1305
                        draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
1306 1307 1308 1309
                      }
                    }else{
                      int sx= mb_x*16 + 8;
                      int sy= mb_y*16 + 8;
1310
                      int xy= (mb_x + mb_y*mv_stride) << mv_sample_log2;
1311 1312
                      int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                      int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1313
                      draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1314
                    }
1315
                  }
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
                }
                if((s->avctx->debug&FF_DEBUG_VIS_QP) && pict->motion_val){
                    uint64_t c= (pict->qscale_table[mb_index]*128/31) * 0x0101010101010101ULL;
                    int y;
                    for(y=0; y<8; y++){
                        *(uint64_t*)(pict->data[1] + 8*mb_x + (8*mb_y + y)*pict->linesize[1])= c;
                        *(uint64_t*)(pict->data[2] + 8*mb_x + (8*mb_y + y)*pict->linesize[2])= c;
                    }
                }
                if((s->avctx->debug&FF_DEBUG_VIS_MB_TYPE) && pict->motion_val){
                    int mb_type= pict->mb_type[mb_index];
                    uint64_t u,v;
                    int y;
#define COLOR(theta, r)\
u= (int)(128 + r*cos(theta*3.141592/180));\
v= (int)(128 + r*sin(theta*3.141592/180));

1333

1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
                    u=v=128;
                    if(IS_PCM(mb_type)){
                        COLOR(120,48)
                    }else if((IS_INTRA(mb_type) && IS_ACPRED(mb_type)) || IS_INTRA16x16(mb_type)){
                        COLOR(30,48)
                    }else if(IS_INTRA4x4(mb_type)){
                        COLOR(90,48)
                    }else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type)){
//                        COLOR(120,48)
                    }else if(IS_DIRECT(mb_type)){
                        COLOR(150,48)
                    }else if(IS_GMC(mb_type) && IS_SKIP(mb_type)){
                        COLOR(170,48)
                    }else if(IS_GMC(mb_type)){
                        COLOR(190,48)
                    }else if(IS_SKIP(mb_type)){
//                        COLOR(180,48)
                    }else if(!USES_LIST(mb_type, 1)){
                        COLOR(240,48)
                    }else if(!USES_LIST(mb_type, 0)){
                        COLOR(0,48)
                    }else{
                        assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
                        COLOR(300,48)
                    }

                    u*= 0x0101010101010101ULL;
                    v*= 0x0101010101010101ULL;
                    for(y=0; y<8; y++){
                        *(uint64_t*)(pict->data[1] + 8*mb_x + (8*mb_y + y)*pict->linesize[1])= u;
                        *(uint64_t*)(pict->data[2] + 8*mb_x + (8*mb_y + y)*pict->linesize[2])= v;
                    }

                    //segmentation
                    if(IS_8X8(mb_type) || IS_16X8(mb_type)){
                        *(uint64_t*)(pict->data[0] + 16*mb_x + 0 + (16*mb_y + 8)*pict->linesize[0])^= 0x8080808080808080ULL;
                        *(uint64_t*)(pict->data[0] + 16*mb_x + 8 + (16*mb_y + 8)*pict->linesize[0])^= 0x8080808080808080ULL;
                    }
                    if(IS_8X8(mb_type) || IS_8X16(mb_type)){
                        for(y=0; y<16; y++)
                            pict->data[0][16*mb_x + 8 + (16*mb_y + y)*pict->linesize[0]]^= 0x80;
                    }
1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390
                    if(IS_8X8(mb_type) && mv_sample_log2 >= 2){
                        int dm= 1 << (mv_sample_log2-2);
                        for(i=0; i<4; i++){
                            int sx= mb_x*16 + 8*(i&1);
                            int sy= mb_y*16 + 8*(i>>1);
                            int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
                            //FIXME bidir
                            int32_t *mv = (int32_t*)&pict->motion_val[0][xy];
                            if(mv[0] != mv[dm] || mv[dm*mv_stride] != mv[dm*(mv_stride+1)])
                                for(y=0; y<8; y++)
                                    pict->data[0][sx + 4 + (sy + y)*pict->linesize[0]]^= 0x80;
                            if(mv[0] != mv[dm*mv_stride] || mv[dm] != mv[dm*(mv_stride+1)])
                                *(uint64_t*)(pict->data[0] + sx + (sy + 4)*pict->linesize[0])^= 0x8080808080808080ULL;
                        }
                    }
1391

1392 1393 1394
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264){
                        // hmm
                    }
1395 1396 1397 1398 1399
                }
                s->mbskip_table[mb_index]=0;
            }
        }
    }
1400 1401
}

1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
/**
 * Copies a rectangular area of samples to a temporary buffer and replicates the boarder samples.
 * @param buf destination buffer
 * @param src source buffer
 * @param linesize number of bytes between 2 vertically adjacent samples in both the source and destination buffers
 * @param block_w width of block
 * @param block_h height of block
 * @param src_x x coordinate of the top left sample of the block in the source buffer
 * @param src_y y coordinate of the top left sample of the block in the source buffer
 * @param w width of the source buffer
 * @param h height of the source buffer
 */
1414
void ff_emulated_edge_mc(uint8_t *buf, uint8_t *src, int linesize, int block_w, int block_h,
1415 1416 1417
                                    int src_x, int src_y, int w, int h){
    int x, y;
    int start_y, start_x, end_y, end_x;
1418

1419 1420 1421
    if(src_y>= h){
        src+= (h-1-src_y)*linesize;
        src_y=h-1;
1422 1423 1424
    }else if(src_y<=-block_h){
        src+= (1-block_h-src_y)*linesize;
        src_y=1-block_h;
1425 1426 1427 1428
    }
    if(src_x>= w){
        src+= (w-1-src_x);
        src_x=w-1;
1429 1430 1431
    }else if(src_x<=-block_w){
        src+= (1-block_w-src_x);
        src_x=1-block_w;
1432 1433
    }

1434 1435 1436 1437
    start_y= FFMAX(0, -src_y);
    start_x= FFMAX(0, -src_x);
    end_y= FFMIN(block_h, h-src_y);
    end_x= FFMIN(block_w, w-src_x);
1438

1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458
    // copy existing part
    for(y=start_y; y<end_y; y++){
        for(x=start_x; x<end_x; x++){
            buf[x + y*linesize]= src[x + y*linesize];
        }
    }

    //top
    for(y=0; y<start_y; y++){
        for(x=start_x; x<end_x; x++){
            buf[x + y*linesize]= buf[x + start_y*linesize];
        }
    }

    //bottom
    for(y=end_y; y<block_h; y++){
        for(x=start_x; x<end_x; x++){
            buf[x + y*linesize]= buf[x + (end_y-1)*linesize];
        }
    }
1459

1460 1461 1462 1463 1464
    for(y=0; y<block_h; y++){
       //left
        for(x=0; x<start_x; x++){
            buf[x + y*linesize]= buf[start_x + y*linesize];
        }
1465

1466 1467 1468 1469 1470 1471 1472
       //right
        for(x=end_x; x<block_w; x++){
            buf[x + y*linesize]= buf[end_x - 1 + y*linesize];
        }
    }
}

1473
static inline int hpel_motion_lowres(MpegEncContext *s,
Michael Niedermayer's avatar
Michael Niedermayer committed
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495
                                  uint8_t *dest, uint8_t *src,
                                  int field_based, int field_select,
                                  int src_x, int src_y,
                                  int width, int height, int stride,
                                  int h_edge_pos, int v_edge_pos,
                                  int w, int h, h264_chroma_mc_func *pix_op,
                                  int motion_x, int motion_y)
{
    const int lowres= s->avctx->lowres;
    const int s_mask= (2<<lowres)-1;
    int emu=0;
    int sx, sy;

    if(s->quarter_sample){
        motion_x/=2;
        motion_y/=2;
    }

    sx= motion_x & s_mask;
    sy= motion_y & s_mask;
    src_x += motion_x >> (lowres+1);
    src_y += motion_y >> (lowres+1);
1496

Michael Niedermayer's avatar
Michael Niedermayer committed
1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
    src += src_y * stride + src_x;

    if(   (unsigned)src_x > h_edge_pos                 - (!!sx) - w
       || (unsigned)src_y >(v_edge_pos >> field_based) - (!!sy) - h){
        ff_emulated_edge_mc(s->edge_emu_buffer, src, s->linesize, w+1, (h+1)<<field_based,
                            src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
        src= s->edge_emu_buffer;
        emu=1;
    }

    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    if(field_select)
        src += s->linesize;
    pix_op[lowres](dest, src, stride, h, sx, sy);
    return emu;
}

Fabrice Bellard's avatar
Fabrice Bellard committed
1515
/* apply one mpeg motion vector to the three components */
1516
static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
1517
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1518
                               int field_based, int bottom_field, int field_select,
1519
                               uint8_t **ref_picture, h264_chroma_mc_func *pix_op,
Fabrice Bellard's avatar
Fabrice Bellard committed
1520 1521
                               int motion_x, int motion_y, int h)
{
1522
    uint8_t *ptr_y, *ptr_cb, *ptr_cr;
1523 1524 1525 1526 1527 1528
    int mx, my, src_x, src_y, uvsrc_x, uvsrc_y, uvlinesize, linesize, sx, sy, uvsx, uvsy;
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
    const int s_mask= (2<<lowres)-1;
    const int h_edge_pos = s->h_edge_pos >> lowres;
    const int v_edge_pos = s->v_edge_pos >> lowres;
1529
    linesize   = s->current_picture.linesize[0] << field_based;
1530
    uvlinesize = s->current_picture.linesize[1] << field_based;
1531

1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
    if(s->quarter_sample){ //FIXME obviously not perfect but qpel wont work in lowres anyway
        motion_x/=2;
        motion_y/=2;
    }

    if(field_based){
        motion_y += (bottom_field - field_select)*((1<<lowres)-1);
    }

    sx= motion_x & s_mask;
    sy= motion_y & s_mask;
    src_x = s->mb_x*2*block_s               + (motion_x >> (lowres+1));
    src_y =(s->mb_y*2*block_s>>field_based) + (motion_y >> (lowres+1));
1545

1546
    if (s->out_format == FMT_H263) {
Michael Niedermayer's avatar
Michael Niedermayer committed
1547 1548
        uvsx = ((motion_x>>1) & s_mask) | (sx&1);
        uvsy = ((motion_y>>1) & s_mask) | (sy&1);
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563
        uvsrc_x = src_x>>1;
        uvsrc_y = src_y>>1;
    }else if(s->out_format == FMT_H261){//even chroma mv's are full pel in H261
        mx = motion_x / 4;
        my = motion_y / 4;
        uvsx = (2*mx) & s_mask;
        uvsy = (2*my) & s_mask;
        uvsrc_x = s->mb_x*block_s               + (mx >> lowres);
        uvsrc_y = s->mb_y*block_s               + (my >> lowres);
    } else {
        mx = motion_x / 2;
        my = motion_y / 2;
        uvsx = mx & s_mask;
        uvsy = my & s_mask;
        uvsrc_x = s->mb_x*block_s               + (mx >> (lowres+1));
1564
        uvsrc_y =(s->mb_y*block_s>>field_based) + (my >> (lowres+1));
1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575
    }

    ptr_y  = ref_picture[0] + src_y * linesize + src_x;
    ptr_cb = ref_picture[1] + uvsrc_y * uvlinesize + uvsrc_x;
    ptr_cr = ref_picture[2] + uvsrc_y * uvlinesize + uvsrc_x;

    if(   (unsigned)src_x > h_edge_pos                 - (!!sx) - 2*block_s
       || (unsigned)src_y >(v_edge_pos >> field_based) - (!!sy) - h){
            ff_emulated_edge_mc(s->edge_emu_buffer, ptr_y, s->linesize, 17, 17+field_based,
                             src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
            ptr_y = s->edge_emu_buffer;
1576
            if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1577
                uint8_t *uvbuf= s->edge_emu_buffer+18*s->linesize;
1578
                ff_emulated_edge_mc(uvbuf  , ptr_cb, s->uvlinesize, 9, 9+field_based,
1579
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
1580
                ff_emulated_edge_mc(uvbuf+16, ptr_cr, s->uvlinesize, 9, 9+field_based,
1581 1582 1583 1584 1585 1586
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
                ptr_cb= uvbuf;
                ptr_cr= uvbuf+16;
            }
    }

1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
    if(bottom_field){ //FIXME use this for field pix too instead of the obnoxious hack which changes picture.data
        dest_y += s->linesize;
        dest_cb+= s->uvlinesize;
        dest_cr+= s->uvlinesize;
    }

    if(field_select){
        ptr_y += s->linesize;
        ptr_cb+= s->uvlinesize;
        ptr_cr+= s->uvlinesize;
    }

1599 1600 1601
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres-1](dest_y, ptr_y, linesize, h, sx, sy);
1602

1603
    if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1604 1605 1606 1607 1608
        uvsx <<= 2 - lowres;
        uvsy <<= 2 - lowres;
        pix_op[lowres](dest_cb, ptr_cb, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
        pix_op[lowres](dest_cr, ptr_cr, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
    }
1609
    //FIXME h261 lowres loop filter
1610 1611
}

Michael Niedermayer's avatar
Michael Niedermayer committed
1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623
static inline void chroma_4mv_motion_lowres(MpegEncContext *s,
                                     uint8_t *dest_cb, uint8_t *dest_cr,
                                     uint8_t **ref_picture,
                                     h264_chroma_mc_func *pix_op,
                                     int mx, int my){
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
    const int s_mask= (2<<lowres)-1;
    const int h_edge_pos = s->h_edge_pos >> (lowres+1);
    const int v_edge_pos = s->v_edge_pos >> (lowres+1);
    int emu=0, src_x, src_y, offset, sx, sy;
    uint8_t *ptr;
1624

Michael Niedermayer's avatar
Michael Niedermayer committed
1625 1626 1627 1628 1629 1630 1631 1632 1633
    if(s->quarter_sample){
        mx/=2;
        my/=2;
    }

    /* In case of 8X8, we construct a single chroma motion vector
       with a special rounding */
    mx= ff_h263_round_chroma(mx);
    my= ff_h263_round_chroma(my);
1634

Michael Niedermayer's avatar
Michael Niedermayer committed
1635 1636 1637 1638
    sx= mx & s_mask;
    sy= my & s_mask;
    src_x = s->mb_x*block_s + (mx >> (lowres+1));
    src_y = s->mb_y*block_s + (my >> (lowres+1));
1639

Michael Niedermayer's avatar
Michael Niedermayer committed
1640 1641 1642 1643 1644 1645 1646 1647 1648
    offset = src_y * s->uvlinesize + src_x;
    ptr = ref_picture[1] + offset;
    if(s->flags&CODEC_FLAG_EMU_EDGE){
        if(   (unsigned)src_x > h_edge_pos - (!!sx) - block_s
           || (unsigned)src_y > v_edge_pos - (!!sy) - block_s){
            ff_emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
            ptr= s->edge_emu_buffer;
            emu=1;
        }
1649
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
1650 1651 1652
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
1653

Michael Niedermayer's avatar
Michael Niedermayer committed
1654 1655 1656 1657 1658 1659 1660 1661
    ptr = ref_picture[2] + offset;
    if(emu){
        ff_emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
        ptr= s->edge_emu_buffer;
    }
    pix_op[lowres](dest_cr, ptr, s->uvlinesize, block_s, sx, sy);
}

Michael Niedermayer's avatar
Michael Niedermayer committed
1662
/**
Diego Biurrun's avatar
Diego Biurrun committed
1663
 * motion compensation of a single macroblock
Michael Niedermayer's avatar
Michael Niedermayer committed
1664 1665 1666 1667 1668 1669 1670 1671 1672
 * @param s context
 * @param dest_y luma destination pointer
 * @param dest_cb chroma cb/u destination pointer
 * @param dest_cr chroma cr/v destination pointer
 * @param dir direction (0->forward, 1->backward)
 * @param ref_picture array[3] of pointers to the 3 planes of the reference picture
 * @param pic_op halfpel motion compensation function (average or put normally)
 * the motion vectors are taken from s->mv and the MV type from s->mv_type
 */
1673
static inline void MPV_motion_lowres(MpegEncContext *s,
1674
                              uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1675
                              int dir, uint8_t **ref_picture,
1676
                              h264_chroma_mc_func *pix_op)
Fabrice Bellard's avatar
Fabrice Bellard committed
1677
{
1678
    int mx, my;
Fabrice Bellard's avatar
Fabrice Bellard committed
1679
    int mb_x, mb_y, i;
1680 1681
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
Fabrice Bellard's avatar
Fabrice Bellard committed
1682 1683 1684 1685 1686 1687

    mb_x = s->mb_x;
    mb_y = s->mb_y;

    switch(s->mv_type) {
    case MV_TYPE_16X16:
1688 1689 1690 1691
        mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                    0, 0, 0,
                    ref_picture, pix_op,
                    s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
Fabrice Bellard's avatar
Fabrice Bellard committed
1692 1693
        break;
    case MV_TYPE_8X8:
Michael Niedermayer's avatar
Michael Niedermayer committed
1694 1695 1696
        mx = 0;
        my = 0;
            for(i=0;i<4;i++) {
1697
                hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) * s->linesize)*block_s,
1698
                            ref_picture[0], 0, 0,
1699
                            (2*mb_x + (i & 1))*block_s, (2*mb_y + (i >>1))*block_s,
1700
                            s->width, s->height, s->linesize,
1701 1702
                            s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
                            block_s, block_s, pix_op,
1703
                            s->mv[dir][i][0], s->mv[dir][i][1]);
Michael Niedermayer's avatar
Michael Niedermayer committed
1704 1705 1706

                mx += s->mv[dir][i][0];
                my += s->mv[dir][i][1];
1707
            }
Michael Niedermayer's avatar
Michael Niedermayer committed
1708

1709
        if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY))
1710
            chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture, pix_op, mx, my);
Fabrice Bellard's avatar
Fabrice Bellard committed
1711 1712 1713
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
1714 1715 1716 1717 1718 1719 1720 1721 1722 1723
            /* top field */
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                        1, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
                        s->mv[dir][0][0], s->mv[dir][0][1], block_s);
            /* bottom field */
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                        1, 1, s->field_select[dir][1],
                        ref_picture, pix_op,
                        s->mv[dir][1][0], s->mv[dir][1][1], block_s);
Fabrice Bellard's avatar
Fabrice Bellard committed
1724
        } else {
1725 1726
            if(s->picture_structure != s->field_select[dir][0] + 1 && s->pict_type != B_TYPE && !s->first_field){
                ref_picture= s->current_picture_ptr->data;
1727
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
1728

1729
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1730 1731
                        0, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
1732
                        s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
Fabrice Bellard's avatar
Fabrice Bellard committed
1733
        }
1734
        break;
1735 1736 1737
    case MV_TYPE_16X8:
        for(i=0; i<2; i++){
            uint8_t ** ref2picture;
1738

1739
            if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == B_TYPE || s->first_field){
1740 1741
                ref2picture= ref_picture;
            }else{
1742
                ref2picture= s->current_picture_ptr->data;
1743
            }
1744

1745
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1746
                        0, 0, s->field_select[dir][i],
1747
                        ref2picture, pix_op,
1748
                        s->mv[dir][i][0], s->mv[dir][i][1] + 2*block_s*i, block_s);
1749

1750 1751 1752
            dest_y += 2*block_s*s->linesize;
            dest_cb+= (2*block_s>>s->chroma_y_shift)*s->uvlinesize;
            dest_cr+= (2*block_s>>s->chroma_y_shift)*s->uvlinesize;
1753
        }
Fabrice Bellard's avatar
Fabrice Bellard committed
1754
        break;
1755 1756
    case MV_TYPE_DMV:
        if(s->picture_structure == PICT_FRAME){
1757 1758 1759
            for(i=0; i<2; i++){
                int j;
                for(j=0; j<2; j++){
1760
                    mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1761 1762
                                1, j, j^i,
                                ref_picture, pix_op,
1763
                                s->mv[dir][2*i + j][0], s->mv[dir][2*i + j][1], block_s);
1764
                }
1765
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
1766
            }
1767
        }else{
1768
            for(i=0; i<2; i++){
1769
                mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1770 1771
                            0, 0, s->picture_structure != i+1,
                            ref_picture, pix_op,
1772
                            s->mv[dir][2*i][0],s->mv[dir][2*i][1],2*block_s);
1773

1774
                // after put we make avg of the same block
1775
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
1776

1777 1778
                //opposite parity is always in the same frame if this is second field
                if(!s->first_field){
1779
                    ref_picture = s->current_picture_ptr->data;
1780
                }
1781
            }
1782 1783
        }
    break;
1784
    default: assert(0);
Fabrice Bellard's avatar
Fabrice Bellard committed
1785 1786 1787
    }
}

1788 1789 1790
/* put block[] to dest[] */
static inline void put_dct(MpegEncContext *s,
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
1791
{
1792 1793 1794
    s->dct_unquantize_intra(s, block, i, qscale);
    s->dsp.idct_put (dest, line_size, block);
}
1795

1796 1797 1798 1799 1800 1801 1802 1803
/* add block[] to dest[] */
static inline void add_dct(MpegEncContext *s,
                           DCTELEM *block, int i, uint8_t *dest, int line_size)
{
    if (s->block_last_index[i] >= 0) {
        s->dsp.idct_add (dest, line_size, block);
    }
}
1804

1805
static inline void add_dequant_dct(MpegEncContext *s,
Michael Niedermayer's avatar
Michael Niedermayer committed
1806
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
1807
{
Fabrice Bellard's avatar
Fabrice Bellard committed
1808
    if (s->block_last_index[i] >= 0) {
1809
        s->dct_unquantize_inter(s, block, i, qscale);
1810

1811
        s->dsp.idct_add (dest, line_size, block);
Fabrice Bellard's avatar
Fabrice Bellard committed
1812 1813 1814
    }
}

1815 1816 1817 1818 1819
/**
 * cleans dc, ac, coded_block for the current non intra MB
 */
void ff_clean_intra_table_entries(MpegEncContext *s)
{
1820
    int wrap = s->b8_stride;
1821
    int xy = s->block_index[0];
1822 1823 1824

    s->dc_val[0][xy           ] =
    s->dc_val[0][xy + 1       ] =
1825 1826 1827
    s->dc_val[0][xy     + wrap] =
    s->dc_val[0][xy + 1 + wrap] = 1024;
    /* ac pred */
1828 1829
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
1830 1831 1832 1833 1834 1835 1836
    if (s->msmpeg4_version>=3) {
        s->coded_block[xy           ] =
        s->coded_block[xy + 1       ] =
        s->coded_block[xy     + wrap] =
        s->coded_block[xy + 1 + wrap] = 0;
    }
    /* chroma */
1837 1838
    wrap = s->mb_stride;
    xy = s->mb_x + s->mb_y * wrap;
1839 1840 1841
    s->dc_val[1][xy] =
    s->dc_val[2][xy] = 1024;
    /* ac pred */
1842 1843
    memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
1844

1845
    s->mbintra_table[xy]= 0;
1846 1847
}

Fabrice Bellard's avatar
Fabrice Bellard committed
1848 1849 1850 1851 1852 1853 1854 1855 1856 1857
/* generic function called after a macroblock has been parsed by the
   decoder or after it has been encoded by the encoder.

   Important variables used:
   s->mb_intra : true if intra macroblock
   s->mv_dir   : motion vector direction
   s->mv_type  : motion vector type
   s->mv       : motion vector
   s->interlaced_dct : true if interlaced dct used (mpeg2)
 */
1858
static av_always_inline void MPV_decode_mb_internal(MpegEncContext *s, DCTELEM block[12][64], int lowres_flag)
Fabrice Bellard's avatar
Fabrice Bellard committed
1859
{
1860
    int mb_x, mb_y;
1861
    const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
Ivan Kalvachev's avatar
Ivan Kalvachev committed
1862 1863
#ifdef HAVE_XVMC
    if(s->avctx->xvmc_acceleration){
1864
        XVMC_decode_mb(s);//xvmc uses pblocks
Ivan Kalvachev's avatar
Ivan Kalvachev committed
1865 1866 1867
        return;
    }
#endif
Fabrice Bellard's avatar
Fabrice Bellard committed
1868 1869 1870 1871

    mb_x = s->mb_x;
    mb_y = s->mb_y;

1872 1873 1874 1875 1876 1877 1878 1879 1880
    if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
       /* save DCT coefficients */
       int i,j;
       DCTELEM *dct = &s->current_picture.dct_coeff[mb_xy*64*6];
       for(i=0; i<6; i++)
           for(j=0; j<64; j++)
               *dct++ = block[i][s->dsp.idct_permutation[j]];
    }

Michael Niedermayer's avatar
Michael Niedermayer committed
1881
    s->current_picture.qscale_table[mb_xy]= s->qscale;
1882

Fabrice Bellard's avatar
Fabrice Bellard committed
1883 1884
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
1885
        if (s->h263_pred || s->h263_aic) {
1886
            if(s->mbintra_table[mb_xy])
1887
                ff_clean_intra_table_entries(s);
Fabrice Bellard's avatar
Fabrice Bellard committed
1888
        } else {
1889 1890
            s->last_dc[0] =
            s->last_dc[1] =
Fabrice Bellard's avatar
Fabrice Bellard committed
1891 1892 1893
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
1894
    else if (s->h263_pred || s->h263_aic)
1895
        s->mbintra_table[mb_xy]=1;
1896

1897
    if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==B_TYPE) && s->avctx->mb_decision != FF_MB_DECISION_RD)) { //FIXME precalc
1898
        uint8_t *dest_y, *dest_cb, *dest_cr;
1899
        int dct_linesize, dct_offset;
Michael Niedermayer's avatar
Michael Niedermayer committed
1900 1901
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
1902
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
1903
        const int uvlinesize= s->current_picture.linesize[1];
1904 1905
        const int readable= s->pict_type != B_TYPE || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
        const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
1906

Michael Niedermayer's avatar
Michael Niedermayer committed
1907 1908 1909
        /* avoid copy if macroblock skipped in last frame too */
        /* skip only during decoding as we might trash the buffers during encoding a bit */
        if(!s->encoding){
1910
            uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
Michael Niedermayer's avatar
Michael Niedermayer committed
1911
            const int age= s->current_picture.age;
Michael Niedermayer's avatar
Michael Niedermayer committed
1912

Michael Niedermayer's avatar
Michael Niedermayer committed
1913 1914
            assert(age);

1915 1916
            if (s->mb_skipped) {
                s->mb_skipped= 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
1917
                assert(s->pict_type!=I_TYPE);
1918

1919
                (*mbskip_ptr) ++; /* indicate that this time we skipped it */
Michael Niedermayer's avatar
Michael Niedermayer committed
1920 1921
                if(*mbskip_ptr >99) *mbskip_ptr= 99;

Michael Niedermayer's avatar
Michael Niedermayer committed
1922
                /* if previous was skipped too, then nothing to do !  */
1923 1924
                if (*mbskip_ptr >= age && s->current_picture.reference){
                    return;
Michael Niedermayer's avatar
Michael Niedermayer committed
1925
                }
1926 1927 1928 1929
            } else if(!s->current_picture.reference){
                (*mbskip_ptr) ++; /* increase counter so the age can be compared cleanly */
                if(*mbskip_ptr >99) *mbskip_ptr= 99;
            } else{
1930 1931
                *mbskip_ptr = 0; /* not skipped */
            }
Michael Niedermayer's avatar
Michael Niedermayer committed
1932
        }
1933

1934
        dct_linesize = linesize << s->interlaced_dct;
1935
        dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
1936

1937 1938 1939 1940 1941
        if(readable){
            dest_y=  s->dest[0];
            dest_cb= s->dest[1];
            dest_cr= s->dest[2];
        }else{
1942
            dest_y = s->b_scratchpad;
1943
            dest_cb= s->b_scratchpad+16*linesize;
1944
            dest_cr= s->b_scratchpad+32*linesize;
1945
        }
1946

Fabrice Bellard's avatar
Fabrice Bellard committed
1947 1948
        if (!s->mb_intra) {
            /* motion handling */
1949
            /* decoding or more than one mb_type (MC was already done otherwise) */
1950
            if(!s->encoding){
1951 1952
                if(lowres_flag){
                    h264_chroma_mc_func *op_pix = s->dsp.put_h264_chroma_pixels_tab;
Fabrice Bellard's avatar
Fabrice Bellard committed
1953

1954 1955 1956 1957 1958 1959 1960 1961
                    if (s->mv_dir & MV_DIR_FORWARD) {
                        MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix);
                        op_pix = s->dsp.avg_h264_chroma_pixels_tab;
                    }
                    if (s->mv_dir & MV_DIR_BACKWARD) {
                        MPV_motion_lowres(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix);
                    }
                }else{
1962
                    op_qpix= s->me.qpel_put;
1963
                    if ((!s->no_rounding) || s->pict_type==B_TYPE){
1964 1965 1966 1967 1968 1969 1970
                        op_pix = s->dsp.put_pixels_tab;
                    }else{
                        op_pix = s->dsp.put_no_rnd_pixels_tab;
                    }
                    if (s->mv_dir & MV_DIR_FORWARD) {
                        MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix, op_qpix);
                        op_pix = s->dsp.avg_pixels_tab;
1971
                        op_qpix= s->me.qpel_avg;
1972 1973 1974 1975
                    }
                    if (s->mv_dir & MV_DIR_BACKWARD) {
                        MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix, op_qpix);
                    }
1976
                }
Fabrice Bellard's avatar
Fabrice Bellard committed
1977 1978
            }

1979
            /* skip dequant / idct if we are really late ;) */
Michael Niedermayer's avatar
Michael Niedermayer committed
1980 1981 1982 1983 1984 1985 1986
            if(s->hurry_up>1) goto skip_idct;
            if(s->avctx->skip_idct){
                if(  (s->avctx->skip_idct >= AVDISCARD_NONREF && s->pict_type == B_TYPE)
                   ||(s->avctx->skip_idct >= AVDISCARD_NONKEY && s->pict_type != I_TYPE)
                   || s->avctx->skip_idct >= AVDISCARD_ALL)
                    goto skip_idct;
            }
1987

Fabrice Bellard's avatar
Fabrice Bellard committed
1988
            /* add dct residue */
Michael Niedermayer's avatar
Michael Niedermayer committed
1989
            if(s->encoding || !(   s->h263_msmpeg4 || s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO
1990
                                || (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant))){
1991 1992 1993 1994
                add_dequant_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
                add_dequant_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
                add_dequant_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
                add_dequant_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
1995

1996
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007
                    if (s->chroma_y_shift){
                        add_dequant_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
                        add_dequant_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
                    }else{
                        dct_linesize >>= 1;
                        dct_offset >>=1;
                        add_dequant_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
                        add_dequant_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
                        add_dequant_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
                        add_dequant_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
                    }
2008
                }
Michael Niedermayer's avatar
Michael Niedermayer committed
2009
            } else if(s->codec_id != CODEC_ID_WMV2){
2010 2011 2012 2013
                add_dct(s, block[0], 0, dest_y                          , dct_linesize);
                add_dct(s, block[1], 1, dest_y              + block_size, dct_linesize);
                add_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize);
                add_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize);
Fabrice Bellard's avatar
Fabrice Bellard committed
2014

2015
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
                    if(s->chroma_y_shift){//Chroma420
                        add_dct(s, block[4], 4, dest_cb, uvlinesize);
                        add_dct(s, block[5], 5, dest_cr, uvlinesize);
                    }else{
                        //chroma422
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;

                        add_dct(s, block[4], 4, dest_cb, dct_linesize);
                        add_dct(s, block[5], 5, dest_cr, dct_linesize);
                        add_dct(s, block[6], 6, dest_cb+dct_offset, dct_linesize);
                        add_dct(s, block[7], 7, dest_cr+dct_offset, dct_linesize);
                        if(!s->chroma_x_shift){//Chroma444
                            add_dct(s, block[8], 8, dest_cb+8, dct_linesize);
                            add_dct(s, block[9], 9, dest_cr+8, dct_linesize);
                            add_dct(s, block[10], 10, dest_cb+8+dct_offset, dct_linesize);
                            add_dct(s, block[11], 11, dest_cr+8+dct_offset, dct_linesize);
                        }
                    }
                }//fi gray
            }
2037
            else if (ENABLE_WMV2) {
Michael Niedermayer's avatar
Michael Niedermayer committed
2038
                ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
2039
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
2040 2041
        } else {
            /* dct only in intra block */
Michael Niedermayer's avatar
Michael Niedermayer committed
2042
            if(s->encoding || !(s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO)){
2043 2044 2045 2046
                put_dct(s, block[0], 0, dest_y                          , dct_linesize, s->qscale);
                put_dct(s, block[1], 1, dest_y              + block_size, dct_linesize, s->qscale);
                put_dct(s, block[2], 2, dest_y + dct_offset             , dct_linesize, s->qscale);
                put_dct(s, block[3], 3, dest_y + dct_offset + block_size, dct_linesize, s->qscale);
2047

2048
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2049 2050 2051 2052 2053 2054 2055 2056 2057
                    if(s->chroma_y_shift){
                        put_dct(s, block[4], 4, dest_cb, uvlinesize, s->chroma_qscale);
                        put_dct(s, block[5], 5, dest_cr, uvlinesize, s->chroma_qscale);
                    }else{
                        dct_offset >>=1;
                        dct_linesize >>=1;
                        put_dct(s, block[4], 4, dest_cb,              dct_linesize, s->chroma_qscale);
                        put_dct(s, block[5], 5, dest_cr,              dct_linesize, s->chroma_qscale);
                        put_dct(s, block[6], 6, dest_cb + dct_offset, dct_linesize, s->chroma_qscale);
2058
                        put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
2059 2060 2061
                    }
                }
            }else{
2062 2063 2064 2065
                s->dsp.idct_put(dest_y                          , dct_linesize, block[0]);
                s->dsp.idct_put(dest_y              + block_size, dct_linesize, block[1]);
                s->dsp.idct_put(dest_y + dct_offset             , dct_linesize, block[2]);
                s->dsp.idct_put(dest_y + dct_offset + block_size, dct_linesize, block[3]);
2066

2067
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2068 2069 2070 2071
                    if(s->chroma_y_shift){
                        s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                        s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
                    }else{
2072

2073 2074
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
2075

2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087
                        s->dsp.idct_put(dest_cb,              dct_linesize, block[4]);
                        s->dsp.idct_put(dest_cr,              dct_linesize, block[5]);
                        s->dsp.idct_put(dest_cb + dct_offset, dct_linesize, block[6]);
                        s->dsp.idct_put(dest_cr + dct_offset, dct_linesize, block[7]);
                        if(!s->chroma_x_shift){//Chroma444
                            s->dsp.idct_put(dest_cb + 8,              dct_linesize, block[8]);
                            s->dsp.idct_put(dest_cr + 8,              dct_linesize, block[9]);
                            s->dsp.idct_put(dest_cb + 8 + dct_offset, dct_linesize, block[10]);
                            s->dsp.idct_put(dest_cr + 8 + dct_offset, dct_linesize, block[11]);
                        }
                    }
                }//gray
2088
            }
2089
        }
2090 2091 2092 2093 2094
skip_idct:
        if(!readable){
            s->dsp.put_pixels_tab[0][0](s->dest[0], dest_y ,   linesize,16);
            s->dsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[1], dest_cb, uvlinesize,16 >> s->chroma_y_shift);
            s->dsp.put_pixels_tab[s->chroma_x_shift][0](s->dest[2], dest_cr, uvlinesize,16 >> s->chroma_y_shift);
2095
        }
2096 2097 2098
    }
}

2099 2100 2101
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]){
    if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1);
    else                  MPV_decode_mb_internal(s, block, 0);
2102 2103
}

2104 2105 2106 2107 2108 2109 2110 2111
/**
 *
 * @param h is the normal height, this will be reduced automatically if needed for the last row
 */
void ff_draw_horiz_band(MpegEncContext *s, int y, int h){
    if (s->avctx->draw_horiz_band) {
        AVFrame *src;
        int offset[4];
2112

2113 2114 2115 2116
        if(s->picture_structure != PICT_FRAME){
            h <<= 1;
            y <<= 1;
            if(s->first_field  && !(s->avctx->slice_flags&SLICE_FLAG_ALLOW_FIELD)) return;
2117 2118
        }

2119
        h= FFMIN(h, s->avctx->height - y);
2120

2121 2122 2123 2124 2125 2126
        if(s->pict_type==B_TYPE || s->low_delay || (s->avctx->slice_flags&SLICE_FLAG_CODED_ORDER))
            src= (AVFrame*)s->current_picture_ptr;
        else if(s->last_picture_ptr)
            src= (AVFrame*)s->last_picture_ptr;
        else
            return;
2127

2128 2129 2130 2131 2132
        if(s->pict_type==B_TYPE && s->picture_structure == PICT_FRAME && s->out_format != FMT_H264){
            offset[0]=
            offset[1]=
            offset[2]=
            offset[3]= 0;
2133
        }else{
2134 2135 2136 2137
            offset[0]= y * s->linesize;;
            offset[1]=
            offset[2]= (y >> s->chroma_y_shift) * s->uvlinesize;
            offset[3]= 0;
2138
        }
2139

2140
        emms_c();
2141

2142 2143 2144 2145
        s->avctx->draw_horiz_band(s->avctx, src, offset,
                                  y, s->picture_structure, h);
    }
}
2146

2147 2148 2149 2150
void ff_init_block_index(MpegEncContext *s){ //FIXME maybe rename
    const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
    const int uvlinesize= s->current_picture.linesize[1];
    const int mb_size= 4 - s->avctx->lowres;
2151

2152 2153 2154 2155 2156 2157 2158
    s->block_index[0]= s->b8_stride*(s->mb_y*2    ) - 2 + s->mb_x*2;
    s->block_index[1]= s->b8_stride*(s->mb_y*2    ) - 1 + s->mb_x*2;
    s->block_index[2]= s->b8_stride*(s->mb_y*2 + 1) - 2 + s->mb_x*2;
    s->block_index[3]= s->b8_stride*(s->mb_y*2 + 1) - 1 + s->mb_x*2;
    s->block_index[4]= s->mb_stride*(s->mb_y + 1)                + s->b8_stride*s->mb_height*2 + s->mb_x - 1;
    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;
    //block_index is not used by mpeg2, so it is not affected by chroma_format
2159

2160 2161 2162
    s->dest[0] = s->current_picture.data[0] + ((s->mb_x - 1) << mb_size);
    s->dest[1] = s->current_picture.data[1] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
    s->dest[2] = s->current_picture.data[2] + ((s->mb_x - 1) << (mb_size - s->chroma_x_shift));
2163

2164 2165 2166 2167 2168
    if(!(s->pict_type==B_TYPE && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME))
    {
        s->dest[0] += s->mb_y *   linesize << mb_size;
        s->dest[1] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
        s->dest[2] += s->mb_y * uvlinesize << (mb_size - s->chroma_y_shift);
2169 2170
    }
}
2171

2172 2173 2174
void ff_mpeg_flush(AVCodecContext *avctx){
    int i;
    MpegEncContext *s = avctx->priv_data;
2175

2176 2177
    if(s==NULL || s->picture==NULL)
        return;
2178

2179 2180 2181 2182
    for(i=0; i<MAX_PICTURE_COUNT; i++){
       if(s->picture[i].data[0] && (   s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
                                    || s->picture[i].type == FF_BUFFER_TYPE_USER))
        avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
Fabrice Bellard's avatar
Fabrice Bellard committed
2183
    }
2184
    s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
2185

2186
    s->mb_x= s->mb_y= 0;
2187

2188 2189 2190 2191 2192 2193 2194 2195
    s->parse_context.state= -1;
    s->parse_context.frame_start_found= 0;
    s->parse_context.overread= 0;
    s->parse_context.overread_index= 0;
    s->parse_context.index= 0;
    s->parse_context.last_index= 0;
    s->bitstream_buffer_size=0;
    s->pp_time=0;
Fabrice Bellard's avatar
Fabrice Bellard committed
2196 2197
}

2198
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
2199
                                   DCTELEM *block, int n, int qscale)
Fabrice Bellard's avatar
Fabrice Bellard committed
2200
{
2201
    int i, level, nCoeffs;
2202
    const uint16_t *quant_matrix;
Fabrice Bellard's avatar
Fabrice Bellard committed
2203

2204
    nCoeffs= s->block_last_index[n];
2205 2206

    if (n < 4)
2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223
        block[0] = block[0] * s->y_dc_scale;
    else
        block[0] = block[0] * s->c_dc_scale;
    /* XXX: only mpeg1 */
    quant_matrix = s->intra_matrix;
    for(i=1;i<=nCoeffs;i++) {
        int j= s->intra_scantable.permutated[i];
        level = block[j];
        if (level) {
            if (level < 0) {
                level = -level;
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
                level = (level - 1) | 1;
                level = -level;
            } else {
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
                level = (level - 1) | 1;
Fabrice Bellard's avatar
Fabrice Bellard committed
2224
            }
2225
            block[j] = level;
Fabrice Bellard's avatar
Fabrice Bellard committed
2226
        }
2227 2228 2229
    }
}

2230
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
2231 2232 2233 2234 2235 2236
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;

    nCoeffs= s->block_last_index[n];
2237

2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252
    quant_matrix = s->inter_matrix;
    for(i=0; i<=nCoeffs; i++) {
        int j= s->intra_scantable.permutated[i];
        level = block[j];
        if (level) {
            if (level < 0) {
                level = -level;
                level = (((level << 1) + 1) * qscale *
                         ((int) (quant_matrix[j]))) >> 4;
                level = (level - 1) | 1;
                level = -level;
            } else {
                level = (((level << 1) + 1) * qscale *
                         ((int) (quant_matrix[j]))) >> 4;
                level = (level - 1) | 1;
Fabrice Bellard's avatar
Fabrice Bellard committed
2253
            }
2254
            block[j] = level;
Fabrice Bellard's avatar
Fabrice Bellard committed
2255 2256 2257
        }
    }
}
2258

2259
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
2260 2261 2262
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
2263
    const uint16_t *quant_matrix;
2264

2265 2266
    if(s->alternate_scan) nCoeffs= 63;
    else nCoeffs= s->block_last_index[n];
2267 2268

    if (n < 4)
2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288
        block[0] = block[0] * s->y_dc_scale;
    else
        block[0] = block[0] * s->c_dc_scale;
    quant_matrix = s->intra_matrix;
    for(i=1;i<=nCoeffs;i++) {
        int j= s->intra_scantable.permutated[i];
        level = block[j];
        if (level) {
            if (level < 0) {
                level = -level;
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
                level = -level;
            } else {
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
            }
            block[j] = level;
        }
    }
}

2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321
static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;
    int sum=-1;

    if(s->alternate_scan) nCoeffs= 63;
    else nCoeffs= s->block_last_index[n];

    if (n < 4)
        block[0] = block[0] * s->y_dc_scale;
    else
        block[0] = block[0] * s->c_dc_scale;
    quant_matrix = s->intra_matrix;
    for(i=1;i<=nCoeffs;i++) {
        int j= s->intra_scantable.permutated[i];
        level = block[j];
        if (level) {
            if (level < 0) {
                level = -level;
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
                level = -level;
            } else {
                level = (int)(level * qscale * quant_matrix[j]) >> 3;
            }
            block[j] = level;
            sum+=level;
        }
    }
    block[63]^=sum&1;
}

2322
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
2323 2324 2325 2326 2327 2328 2329 2330
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;
    int sum=-1;

    if(s->alternate_scan) nCoeffs= 63;
    else nCoeffs= s->block_last_index[n];
2331

2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352
    quant_matrix = s->inter_matrix;
    for(i=0; i<=nCoeffs; i++) {
        int j= s->intra_scantable.permutated[i];
        level = block[j];
        if (level) {
            if (level < 0) {
                level = -level;
                level = (((level << 1) + 1) * qscale *
                         ((int) (quant_matrix[j]))) >> 4;
                level = -level;
            } else {
                level = (((level << 1) + 1) * qscale *
                         ((int) (quant_matrix[j]))) >> 4;
            }
            block[j] = level;
            sum+=level;
        }
    }
    block[63]^=sum&1;
}

2353
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
2354 2355 2356 2357
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
    int nCoeffs;
2358

2359
    assert(s->block_last_index[n]>=0);
2360

2361
    qmul = qscale << 1;
2362

2363
    if (!s->h263_aic) {
2364
        if (n < 4)
2365 2366 2367
            block[0] = block[0] * s->y_dc_scale;
        else
            block[0] = block[0] * s->c_dc_scale;
2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383
        qadd = (qscale - 1) | 1;
    }else{
        qadd = 0;
    }
    if(s->ac_pred)
        nCoeffs=63;
    else
        nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];

    for(i=1; i<=nCoeffs; i++) {
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
2384
            }
2385
            block[i] = level;
2386 2387 2388 2389
        }
    }
}

2390
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
2391 2392 2393
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
2394
    int nCoeffs;
2395

2396
    assert(s->block_last_index[n]>=0);
2397

2398 2399
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
2400

2401
    nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
2402

2403
    for(i=0; i<=nCoeffs; i++) {
2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
            }
            block[i] = level;
        }
    }
}
Fabrice Bellard's avatar
Fabrice Bellard committed
2415

2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431
/**
 * set qscale and update qscale dependent variables.
 */
void ff_set_qscale(MpegEncContext * s, int qscale)
{
    if (qscale < 1)
        qscale = 1;
    else if (qscale > 31)
        qscale = 31;

    s->qscale = qscale;
    s->chroma_qscale= s->chroma_qscale_table[qscale];

    s->y_dc_scale= s->y_dc_scale_table[ qscale ];
    s->c_dc_scale= s->c_dc_scale_table[ s->chroma_qscale ];
}