mpegvideo.c 89.1 KB
Newer Older
F
Fabrice Bellard 已提交
1 2
/*
 * The simplest mpeg encoder (well, it was the simplest!)
F
Fabrice Bellard 已提交
3
 * Copyright (c) 2000,2001 Fabrice Bellard.
4
 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
F
Fabrice Bellard 已提交
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
F
Fabrice Bellard 已提交
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.
F
Fabrice Bellard 已提交
14
 *
15
 * FFmpeg is distributed in the hope that it will be useful,
F
Fabrice Bellard 已提交
16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
F
Fabrice Bellard 已提交
17 18
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
F
Fabrice Bellard 已提交
19
 *
F
Fabrice Bellard 已提交
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
F
Fabrice Bellard 已提交
23
 */
24

M
Michael Niedermayer 已提交
25 26 27
/**
 * @file mpegvideo.c
 * The simplest mpeg encoder (well, it was the simplest!).
28 29
 */

F
Fabrice Bellard 已提交
30 31 32
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"
B
Benoit Fouet 已提交
33
#include "mpegvideo_common.h"
34
#include "mjpegenc.h"
35
#include "msmpeg4.h"
M
Michael Niedermayer 已提交
36
#include "faandct.h"
37
#include <limits.h>
F
Fabrice Bellard 已提交
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);
M
Michael Niedermayer 已提交
57

I
Ivan Kalvachev 已提交
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);
I
Ivan Kalvachev 已提交
62 63
#endif

64
void (*draw_edges)(uint8_t *buf, int wrap, int width, int height, int w)= draw_edges_c;
M
Michael Niedermayer 已提交
65

F
Fabrice Bellard 已提交
66 67 68 69 70 71

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

//#define DEBUG

72

M
Michael Niedermayer 已提交
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
};

M
Michael Niedermayer 已提交
78
void ff_init_scantable(uint8_t *permutation, ScanTable *st, const uint8_t *src_scantable){
79 80
    int i;
    int end;
81

M
Michael Niedermayer 已提交
82
    st->scantable= src_scantable;
83 84 85 86

    for(i=0; i<64; i++){
        int j;
        j = src_scantable[i];
M
Michael Niedermayer 已提交
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;
    }
}

M
Michael Niedermayer 已提交
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;

M
Michael Niedermayer 已提交
105 106 107 108
    assert(p<=end);
    if(p>=end)
        return end;

M
Michael Niedermayer 已提交
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){
M
Michael Niedermayer 已提交
117 118 119
        if     (p[-1] > 1      ) p+= 3;
        else if(p[-2]          ) p+= 2;
        else if(p[-3]|(p[-1]-1)) p++;
M
Michael Niedermayer 已提交
120 121 122 123 124 125
        else{
            p++;
            break;
        }
    }

M
Michael Niedermayer 已提交
126
    p= FFMIN(p, end)-4;
127
    *state= AV_RB32(p);
M
Michael Niedermayer 已提交
128 129 130 131

    return p+4;
}

132
/* init common dct for both encoder and decoder */
A
Aurelien Jacobs 已提交
133
static int DCT_common_init(MpegEncContext *s)
F
Fabrice Bellard 已提交
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(ARCH_POWERPC)
155
    MPV_common_init_ppc(s);
M
Marc Hoffman 已提交
156
#elif defined(HAVE_BFIN)
157 158
    MPV_common_init_bfin(s);
#endif
159

160
    /* load & permutate scantables
161
       note: only wmv uses different ones
162
    */
M
Michael Niedermayer 已提交
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);
    }
M
Michael Niedermayer 已提交
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;
}

B
Benoit Fouet 已提交
176
void copy_picture(Picture *dst, Picture *src){
177 178 179 180
    *dst = *src;
    dst->type= FF_BUFFER_TYPE_COPY;
}

M
cleanup  
Michael Niedermayer 已提交
181
/**
M
Michael Niedermayer 已提交
182 183
 * allocates a Picture
 * The pixels are allocated/set by calling get_buffer() if shared=0
M
cleanup  
Michael Niedermayer 已提交
184
 */
B
Benoit Fouet 已提交
185
int alloc_picture(MpegEncContext *s, Picture *pic, int shared){
D
Diego Biurrun 已提交
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;
M
Michael Niedermayer 已提交
190
    int i;
D
Daniel Kristjansson 已提交
191
    int r= -1;
192

M
Michael Niedermayer 已提交
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

M
Michael Niedermayer 已提交
200
        r= s->avctx->get_buffer(s->avctx, (AVFrame*)pic);
201

M
Michael Niedermayer 已提交
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]);
M
Michael Niedermayer 已提交
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");
D
Daniel Kristjansson 已提交
209
            s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
M
Michael Niedermayer 已提交
210 211 212 213
            return -1;
        }

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

        s->linesize  = pic->linesize[0];
        s->uvlinesize= pic->linesize[1];
M
cleanup  
Michael Niedermayer 已提交
221
    }
222

M
Michael Niedermayer 已提交
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))
M
Michael Niedermayer 已提交
228
        }
M
cleanup  
Michael Niedermayer 已提交
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
        CHECKED_ALLOCZ(pic->mb_type_base , big_mb_num    * sizeof(uint32_t))
233
        pic->mb_type= pic->mb_type_base + s->mb_stride+1;
M
Michael Niedermayer 已提交
234 235
        if(s->out_format == FMT_H264){
            for(i=0; i<2; i++){
M
Michael Niedermayer 已提交
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
            }
M
Michael Niedermayer 已提交
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++){
M
Michael Niedermayer 已提交
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))
M
Michael Niedermayer 已提交
246
            }
M
Michael Niedermayer 已提交
247
            pic->motion_subsample_log2= 3;
M
Michael Niedermayer 已提交
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))
M
Michael Niedermayer 已提交
254
    }
M
Michael Niedermayer 已提交
255

D
Diego Biurrun 已提交
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)
D
Diego Biurrun 已提交
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

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

M
Michael Niedermayer 已提交
270 271 272
/**
 * deallocates a picture
 */
M
cleanup  
Michael Niedermayer 已提交
273 274
static void free_picture(MpegEncContext *s, Picture *pic){
    int i;
M
Michael Niedermayer 已提交
275 276

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

M
cleanup  
Michael Niedermayer 已提交
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);
M
Michael Niedermayer 已提交
285
    av_freep(&pic->mb_type_base);
286
    av_freep(&pic->dct_coeff);
287
    av_freep(&pic->pan_scan);
M
Michael Niedermayer 已提交
288 289
    pic->mb_type= NULL;
    for(i=0; i<2; i++){
290
        av_freep(&pic->motion_val_base[i]);
M
Michael Niedermayer 已提交
291 292
        av_freep(&pic->ref_index[i]);
    }
293

294
    if(pic->type == FF_BUFFER_TYPE_SHARED){
M
Michael Niedermayer 已提交
295 296 297 298
        for(i=0; i<4; i++){
            pic->base[i]=
            pic->data[i]= NULL;
        }
299
        pic->type= 0;
M
cleanup  
Michael Niedermayer 已提交
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)
L
Loren Merritt 已提交
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

D
Diego Biurrun 已提交
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
}

M
Michael Niedermayer 已提交
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.
 */
B
Benoit Fouet 已提交
389
void MPV_common_defaults(MpegEncContext *s){
M
Michael Niedermayer 已提交
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;
M
Michael Niedermayer 已提交
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
/**
M
Michael Niedermayer 已提交
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

L
Loren Merritt 已提交
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)){
M
10l  
Michael Niedermayer 已提交
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
    DCT_common_init(s);
435

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

M
cleanup  
Michael Niedermayer 已提交
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;
M
Michael Niedermayer 已提交
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);

M
Michael Niedermayer 已提交
478
    s->avctx->coded_frame= (AVFrame*)&s->current_picture;
M
cleanup  
Michael Niedermayer 已提交
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));
        }
M
Michael Niedermayer 已提交
506
        CHECKED_ALLOCZ(s->avctx->stats_out, 256);
507 508

        /* Allocate MB type table */
M
Michael Niedermayer 已提交
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

M
Michael Niedermayer 已提交
528
    if(s->codec_id==CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){
529
        /* interlaced direct mode decoding tables */
M
Michael Niedermayer 已提交
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))
            }
F
Fabrice Bellard 已提交
543
    }
M
Michael Niedermayer 已提交
544
    if (s->out_format == FMT_H263) {
F
Fabrice Bellard 已提交
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;
F
Fabrice Bellard 已提交
549
        s->ac_val[2] = s->ac_val[1] + c_size;
550

F
Fabrice Bellard 已提交
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

F
Fabrice Bellard 已提交
587
    s->context_initialized = 1;
588 589

    s->thread_context[0]= s;
590 591 592 593 594
    /* h264 does thread context setup itself, but it needs context[0]
     * to be fully initialized for the error resilience code */
    threads = s->codec_id == CODEC_ID_H264 ? 1 : s->avctx->thread_count;

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

599
    for(i=0; i<threads; i++){
600 601 602 603 604 605
        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;
    }

F
Fabrice Bellard 已提交
606 607
    return 0;
 fail:
608
    MPV_common_end(s);
F
Fabrice Bellard 已提交
609 610 611 612 613 614
    return -1;
}

/* init common structure for both encoder and decoder */
void MPV_common_end(MpegEncContext *s)
{
M
Michael Niedermayer 已提交
615
    int i, j, k;
F
Fabrice Bellard 已提交
616

617 618 619 620 621 622 623
    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]);
    }

624 625 626
    av_freep(&s->parse_context.buffer);
    s->parse_context.buffer_size=0;

627
    av_freep(&s->mb_type);
628 629 630 631 632 633 634 635 636 637 638 639
    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;
M
Michael Niedermayer 已提交
640 641 642 643 644 645 646 647 648 649 650 651
    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]);
    }
652

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

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

M
Michael Niedermayer 已提交
665
    av_freep(&s->avctx->stats_out);
666
    av_freep(&s->ac_stats);
667
    av_freep(&s->error_status_table);
668
    av_freep(&s->mb_index2xy);
669
    av_freep(&s->lambda_table);
670 671
    av_freep(&s->q_intra_matrix);
    av_freep(&s->q_inter_matrix);
672 673
    av_freep(&s->q_intra_matrix16);
    av_freep(&s->q_inter_matrix16);
674 675
    av_freep(&s->input_picture);
    av_freep(&s->reordered_input_picture);
676
    av_freep(&s->dct_offset);
M
cleanup  
Michael Niedermayer 已提交
677

M
Michael Niedermayer 已提交
678 679 680 681
    if(s->picture){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            free_picture(s, &s->picture[i]);
        }
F
Fabrice Bellard 已提交
682
    }
683
    av_freep(&s->picture);
F
Fabrice Bellard 已提交
684
    s->context_initialized = 0;
685 686 687
    s->last_picture_ptr=
    s->next_picture_ptr=
    s->current_picture_ptr= NULL;
M
Michael Niedermayer 已提交
688
    s->linesize= s->uvlinesize= 0;
689

690
    for(i=0; i<3; i++)
691
        av_freep(&s->visualization_buffer[i]);
M
Michael Niedermayer 已提交
692 693

    avcodec_default_free_buffers(s->avctx);
F
Fabrice Bellard 已提交
694 695
}

696
void init_rl(RLTable *rl, uint8_t static_store[2][2*MAX_RUN + MAX_LEVEL + 3])
M
Michael Niedermayer 已提交
697
{
698 699
    int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
    uint8_t index_run[MAX_RUN+1];
M
Michael Niedermayer 已提交
700 701
    int last, run, level, start, end, i;

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

M
Michael Niedermayer 已提交
706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
    /* 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;
        }
729 730
        if(static_store)
            rl->max_level[last] = static_store[last];
731 732
        else
            rl->max_level[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
733
        memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
734 735
        if(static_store)
            rl->max_run[last] = static_store[last] + MAX_RUN + 1;
736 737
        else
            rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
M
Michael Niedermayer 已提交
738
        memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
739 740
        if(static_store)
            rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
741 742
        else
            rl->index_run[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
743 744 745 746
        memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
    }
}

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 798 799
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;
        }
    }
}

F
Fabrice Bellard 已提交
800
/* draw the edges of width 'w' of an image of size width, height */
801
//FIXME check that this is ok for mpeg4 interlaced
802
static void draw_edges_c(uint8_t *buf, int wrap, int width, int height, int w)
F
Fabrice Bellard 已提交
803
{
804
    uint8_t *ptr, *last_line;
F
Fabrice Bellard 已提交
805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828
    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 */
    }
}

M
Michael Niedermayer 已提交
829
int ff_find_unused_picture(MpegEncContext *s, int shared){
M
Michael Niedermayer 已提交
830
    int i;
831

M
Michael Niedermayer 已提交
832 833
    if(shared){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
834
            if(s->picture[i].data[0]==NULL && s->picture[i].type==0) return i;
M
Michael Niedermayer 已提交
835 836 837
        }
    }else{
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
838
            if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) return i; //FIXME
M
Michael Niedermayer 已提交
839 840
        }
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
841
            if(s->picture[i].data[0]==NULL) return i;
M
Michael Niedermayer 已提交
842 843 844
        }
    }

M
Michael Niedermayer 已提交
845 846
    assert(0);
    return -1;
M
Michael Niedermayer 已提交
847 848
}

849 850 851 852 853 854 855 856 857 858
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;
        }
859

860 861 862 863 864 865
        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);
        }
    }
}

M
Michael Niedermayer 已提交
866 867 868
/**
 * generic function for encode/decode called after coding/decoding the header and before a frame is coded/decoded
 */
869
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
F
Fabrice Bellard 已提交
870
{
M
Michael Niedermayer 已提交
871
    int i;
M
Michael Niedermayer 已提交
872
    AVFrame *pic;
873
    s->mb_skipped = 0;
M
Michael Niedermayer 已提交
874

M
Michael Niedermayer 已提交
875
    assert(s->last_picture_ptr==NULL || s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3);
M
Michael Niedermayer 已提交
876

M
cleanup  
Michael Niedermayer 已提交
877
    /* mark&release old frames */
878
    if (s->pict_type != B_TYPE && s->last_picture_ptr && s->last_picture_ptr != s->next_picture_ptr && s->last_picture_ptr->data[0]) {
879
      if(s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3){
M
Michael Niedermayer 已提交
880
        avctx->release_buffer(avctx, (AVFrame*)s->last_picture_ptr);
M
cleanup  
Michael Niedermayer 已提交
881 882 883 884 885

        /* release forgotten pictures */
        /* if(mpeg124/h263) */
        if(!s->encoding){
            for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
886
                if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
887
                    av_log(avctx, AV_LOG_ERROR, "releasing zombie picture\n");
888
                    avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
M
cleanup  
Michael Niedermayer 已提交
889 890
                }
            }
891
        }
892
      }
893
    }
894 895
alloc:
    if(!s->encoding){
D
Diego Biurrun 已提交
896
        /* release non reference frames */
M
Michael Niedermayer 已提交
897 898 899 900 901 902
        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]);
            }
        }

M
Michael Niedermayer 已提交
903 904 905 906 907 908 909
        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];
        }

910 911
        pic->reference= (s->pict_type != B_TYPE || s->codec_id == CODEC_ID_H264)
                        && !s->dropable ? 3 : 0;
M
Michael Niedermayer 已提交
912

M
Michael Niedermayer 已提交
913
        pic->coded_picture_number= s->coded_picture_number++;
914

915 916
        if( alloc_picture(s, (Picture*)pic, 0) < 0)
            return -1;
917

M
Michael Niedermayer 已提交
918
        s->current_picture_ptr= (Picture*)pic;
919
        s->current_picture_ptr->top_field_first= s->top_field_first; //FIXME use only the vars from current_pic
920
        s->current_picture_ptr->interlaced_frame= !s->progressive_frame && !s->progressive_sequence;
M
cleanup  
Michael Niedermayer 已提交
921
    }
922

M
Michael Niedermayer 已提交
923
    s->current_picture_ptr->pict_type= s->pict_type;
924
//    if(s->flags && CODEC_FLAG_QSCALE)
925
  //      s->current_picture_ptr->quality= s->new_picture_ptr->quality;
926
    s->current_picture_ptr->key_frame= s->pict_type == I_TYPE;
M
Michael Niedermayer 已提交
927

928
    copy_picture(&s->current_picture, s->current_picture_ptr);
929

M
cleanup  
Michael Niedermayer 已提交
930
    if (s->pict_type != B_TYPE) {
M
Michael Niedermayer 已提交
931
        s->last_picture_ptr= s->next_picture_ptr;
932 933
        if(!s->dropable)
            s->next_picture_ptr= s->current_picture_ptr;
F
Fabrice Bellard 已提交
934
    }
935
/*    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,
936 937
        s->last_picture_ptr    ? s->last_picture_ptr->data[0] : NULL,
        s->next_picture_ptr    ? s->next_picture_ptr->data[0] : NULL,
938 939
        s->current_picture_ptr ? s->current_picture_ptr->data[0] : NULL,
        s->pict_type, s->dropable);*/
940

941 942
    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);
943

944
    if(s->pict_type != I_TYPE && (s->last_picture_ptr==NULL || s->last_picture_ptr->data[0]==NULL) && !s->dropable){
945
        av_log(avctx, AV_LOG_ERROR, "warning: first frame is no keyframe\n");
D
Diego Biurrun 已提交
946
        assert(s->pict_type != B_TYPE); //these should have been dropped if we don't have a reference
947 948 949 950 951
        goto alloc;
    }

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

M
Michael Niedermayer 已提交
952 953 954 955 956
    if(s->picture_structure!=PICT_FRAME){
        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];
957
            }
M
Michael Niedermayer 已提交
958 959 960 961 962
            s->current_picture.linesize[i] *= 2;
            s->last_picture.linesize[i] *=2;
            s->next_picture.linesize[i] *=2;
        }
    }
963

964 965 966
    s->hurry_up= s->avctx->hurry_up;
    s->error_resilience= avctx->error_resilience;

D
Diego Biurrun 已提交
967
    /* set dequantizer, we can't do it during init as it might change for mpeg4
D
Diego Biurrun 已提交
968
       and we can't do it in the header decode as init is not called for mpeg4 there yet */
969 970 971
    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;
M
Maarten Daniels 已提交
972
    }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
973 974 975 976 977 978
        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;
    }
979

980 981 982 983 984
    if(s->dct_error_sum){
        assert(s->avctx->noise_reduction && s->encoding);

        update_noise_reduction(s);
    }
985

I
Ivan Kalvachev 已提交
986 987 988 989
#ifdef HAVE_XVMC
    if(s->avctx->xvmc_acceleration)
        return XVMC_field_start(s, avctx);
#endif
990
    return 0;
F
Fabrice Bellard 已提交
991
}
992

F
Fabrice Bellard 已提交
993 994 995
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
M
cleanup  
Michael Niedermayer 已提交
996
    int i;
F
Fabrice Bellard 已提交
997
    /* draw edge for correct motion prediction if outside */
I
Ivan Kalvachev 已提交
998 999 1000 1001 1002 1003
#ifdef HAVE_XVMC
//just to make sure that all data is rendered.
    if(s->avctx->xvmc_acceleration){
        XVMC_field_end(s);
    }else
#endif
1004
    if(s->unrestricted_mv && s->current_picture.reference && !s->intra_only && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
M
cleanup  
Michael Niedermayer 已提交
1005 1006 1007
            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);
F
Fabrice Bellard 已提交
1008
    }
M
Michael Niedermayer 已提交
1009
    emms_c();
1010

M
Michael Niedermayer 已提交
1011
    s->last_pict_type    = s->pict_type;
1012
    s->last_lambda_for[s->pict_type]= s->current_picture_ptr->quality;
1013 1014 1015
    if(s->pict_type!=B_TYPE){
        s->last_non_b_pict_type= s->pict_type;
    }
M
Michael Niedermayer 已提交
1016 1017
#if 0
        /* copy back current_picture variables */
M
cleanup  
Michael Niedermayer 已提交
1018 1019 1020 1021
    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;
1022
        }
M
cleanup  
Michael Niedermayer 已提交
1023 1024
    }
    assert(i<MAX_PICTURE_COUNT);
1025
#endif
M
cleanup  
Michael Niedermayer 已提交
1026

M
Michael Niedermayer 已提交
1027
    if(s->encoding){
D
Diego Biurrun 已提交
1028
        /* release non-reference frames */
M
Michael Niedermayer 已提交
1029 1030 1031 1032 1033
        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]);
            }
        }
M
cleanup  
Michael Niedermayer 已提交
1034
    }
M
Michael Niedermayer 已提交
1035 1036 1037 1038 1039 1040
    // 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
M
Michael Niedermayer 已提交
1041
    s->avctx->coded_frame= (AVFrame*)s->current_picture_ptr;
F
Fabrice Bellard 已提交
1042 1043
}

1044 1045 1046 1047 1048 1049 1050 1051
/**
 * 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){
M
SWAP  
Michael Niedermayer 已提交
1052
    int x, y, fr, f;
1053

1054 1055 1056 1057
    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);
1058

1059
    buf[sy*stride + sx]+= color;
1060

D
Diego Biurrun 已提交
1061
    if(FFABS(ex - sx) > FFABS(ey - sy)){
1062
        if(sx > ex){
D
Diego Biurrun 已提交
1063 1064
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1065 1066 1067 1068 1069
        }
        buf+= sx + sy*stride;
        ex-= sx;
        f= ((ey-sy)<<16)/ex;
        for(x= 0; x <= ex; x++){
1070 1071 1072 1073
            y = (x*f)>>16;
            fr= (x*f)&0xFFFF;
            buf[ y   *stride + x]+= (color*(0x10000-fr))>>16;
            buf[(y+1)*stride + x]+= (color*         fr )>>16;
1074 1075 1076
        }
    }else{
        if(sy > ey){
D
Diego Biurrun 已提交
1077 1078
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1079 1080 1081 1082 1083 1084
        }
        buf+= sx + sy*stride;
        ey-= sy;
        if(ey) f= ((ex-sx)<<16)/ey;
        else   f= 0;
        for(y= 0; y <= ey; y++){
1085 1086 1087 1088
            x = (y*f)>>16;
            fr= (y*f)&0xFFFF;
            buf[y*stride + x  ]+= (color*(0x10000-fr))>>16;;
            buf[y*stride + x+1]+= (color*         fr )>>16;;
1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
        }
    }
}

/**
 * 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
 */
1100
static void draw_arrow(uint8_t *buf, int sx, int sy, int ex, int ey, int w, int h, int stride, int color){
1101 1102
    int dx,dy;

1103 1104 1105 1106
    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);
1107

1108 1109
    dx= ex - sx;
    dy= ey - sy;
1110

1111 1112 1113 1114
    if(dx*dx + dy*dy > 3*3){
        int rx=  dx + dy;
        int ry= -dx + dy;
        int length= ff_sqrt((rx*rx + ry*ry)<<8);
1115

1116 1117 1118
        //FIXME subpixel accuracy
        rx= ROUNDED_DIV(rx*3<<4, length);
        ry= ROUNDED_DIV(ry*3<<4, length);
1119

1120 1121 1122 1123 1124 1125
        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);
}

1126 1127 1128
/**
 * prints debuging info for the given picture.
 */
1129
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
1130 1131 1132 1133 1134

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

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

1136 1137 1138 1139 1140 1141 1142
        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;
1143
            case FF_SP_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SP\n"); break;
1144
        }
1145 1146 1147 1148 1149
        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;
1150
                    av_log(s->avctx, AV_LOG_DEBUG, "%1d", count);
1151 1152
                }
                if(s->avctx->debug&FF_DEBUG_QP){
1153
                    av_log(s->avctx, AV_LOG_DEBUG, "%2d", pict->qscale_table[x + y*s->mb_stride]);
1154 1155 1156 1157 1158
                }
                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))
1159
                        av_log(s->avctx, AV_LOG_DEBUG, "P");
1160
                    else if(IS_INTRA(mb_type) && IS_ACPRED(mb_type))
1161
                        av_log(s->avctx, AV_LOG_DEBUG, "A");
1162
                    else if(IS_INTRA4x4(mb_type))
1163
                        av_log(s->avctx, AV_LOG_DEBUG, "i");
1164
                    else if(IS_INTRA16x16(mb_type))
1165
                        av_log(s->avctx, AV_LOG_DEBUG, "I");
1166
                    else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type))
1167
                        av_log(s->avctx, AV_LOG_DEBUG, "d");
1168
                    else if(IS_DIRECT(mb_type))
1169
                        av_log(s->avctx, AV_LOG_DEBUG, "D");
1170
                    else if(IS_GMC(mb_type) && IS_SKIP(mb_type))
1171
                        av_log(s->avctx, AV_LOG_DEBUG, "g");
1172
                    else if(IS_GMC(mb_type))
1173
                        av_log(s->avctx, AV_LOG_DEBUG, "G");
1174
                    else if(IS_SKIP(mb_type))
1175
                        av_log(s->avctx, AV_LOG_DEBUG, "S");
1176
                    else if(!USES_LIST(mb_type, 1))
1177
                        av_log(s->avctx, AV_LOG_DEBUG, ">");
1178
                    else if(!USES_LIST(mb_type, 0))
1179
                        av_log(s->avctx, AV_LOG_DEBUG, "<");
1180 1181
                    else{
                        assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
1182
                        av_log(s->avctx, AV_LOG_DEBUG, "X");
1183
                    }
1184

1185 1186
                    //segmentation
                    if(IS_8X8(mb_type))
1187
                        av_log(s->avctx, AV_LOG_DEBUG, "+");
1188
                    else if(IS_16X8(mb_type))
1189
                        av_log(s->avctx, AV_LOG_DEBUG, "-");
1190
                    else if(IS_8X16(mb_type))
1191
                        av_log(s->avctx, AV_LOG_DEBUG, "|");
1192
                    else if(IS_INTRA(mb_type) || IS_16X16(mb_type))
1193
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1194
                    else
1195
                        av_log(s->avctx, AV_LOG_DEBUG, "?");
1196 1197


1198
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264)
1199
                        av_log(s->avctx, AV_LOG_DEBUG, "=");
1200
                    else
1201
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1202
                }
1203
//                av_log(s->avctx, AV_LOG_DEBUG, " ");
1204
            }
1205
            av_log(s->avctx, AV_LOG_DEBUG, "\n");
1206 1207
        }
    }
1208

1209
    if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
1210 1211
        const int shift= 1 + s->quarter_sample;
        int mb_y;
1212 1213
        uint8_t *ptr;
        int i;
1214
        int h_chroma_shift, v_chroma_shift;
1215 1216
        const int width = s->avctx->width;
        const int height= s->avctx->height;
1217
        const int mv_sample_log2= 4 - pict->motion_subsample_log2;
1218
        const int mv_stride= (s->mb_width << mv_sample_log2) + (s->codec_id == CODEC_ID_H264 ? 0 : 1);
M
100000l  
Michael Niedermayer 已提交
1219
        s->low_delay=0; //needed to see the vectors without trashing the buffers
1220

1221
        avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1222
        for(i=0; i<3; i++){
1223
            memcpy(s->visualization_buffer[i], pict->data[i], (i==0) ? pict->linesize[i]*height:pict->linesize[i]*height >> v_chroma_shift);
1224 1225 1226 1227
            pict->data[i]= s->visualization_buffer[i];
        }
        pict->type= FF_BUFFER_TYPE_COPY;
        ptr= pict->data[0];
1228 1229 1230 1231 1232

        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;
1233 1234 1235
                if((s->avctx->debug_mv) && pict->motion_val){
                  int type;
                  for(type=0; type<3; type++){
1236
                    int direction = 0;
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250
                    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;
                    }
1251 1252 1253
                    if(!USES_LIST(pict->mb_type[mb_index], direction))
                        continue;

1254 1255 1256
                    if(IS_8X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<4; i++){
1257 1258
                        int sx= mb_x*16 + 4 + 8*(i&1);
                        int sy= mb_y*16 + 4 + 8*(i>>1);
1259
                        int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
1260 1261
                        int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                        int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1262
                        draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1263 1264 1265 1266
                      }
                    }else if(IS_16X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<2; i++){
1267 1268
                        int sx=mb_x*16 + 8;
                        int sy=mb_y*16 + 4 + 8*i;
1269
                        int xy= (mb_x*2 + (mb_y*2 + i)*mv_stride) << (mv_sample_log2-1);
1270 1271
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1272

1273 1274
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1275

1276 1277 1278 1279 1280 1281 1282
                        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;
1283
                        int xy= (mb_x*2 + i + mb_y*2*mv_stride) << (mv_sample_log2-1);
1284 1285
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1286

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

1290
                        draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
1291 1292 1293 1294
                      }
                    }else{
                      int sx= mb_x*16 + 8;
                      int sy= mb_y*16 + 8;
1295
                      int xy= (mb_x + mb_y*mv_stride) << mv_sample_log2;
1296 1297
                      int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                      int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1298
                      draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1299
                    }
1300
                  }
M
Michael Niedermayer 已提交
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
                }
                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));

1318

M
Michael Niedermayer 已提交
1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 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
                    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;
                    }
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
                    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;
                        }
                    }
1376

M
Michael Niedermayer 已提交
1377 1378 1379
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264){
                        // hmm
                    }
1380 1381 1382 1383 1384
                }
                s->mbskip_table[mb_index]=0;
            }
        }
    }
1385 1386
}

1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398
/**
 * 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
 */
1399
void ff_emulated_edge_mc(uint8_t *buf, uint8_t *src, int linesize, int block_w, int block_h,
1400 1401 1402
                                    int src_x, int src_y, int w, int h){
    int x, y;
    int start_y, start_x, end_y, end_x;
1403

1404 1405 1406
    if(src_y>= h){
        src+= (h-1-src_y)*linesize;
        src_y=h-1;
M
Michael Niedermayer 已提交
1407 1408 1409
    }else if(src_y<=-block_h){
        src+= (1-block_h-src_y)*linesize;
        src_y=1-block_h;
1410 1411 1412 1413
    }
    if(src_x>= w){
        src+= (w-1-src_x);
        src_x=w-1;
M
Michael Niedermayer 已提交
1414 1415 1416
    }else if(src_x<=-block_w){
        src+= (1-block_w-src_x);
        src_x=1-block_w;
1417 1418
    }

M
Michael Niedermayer 已提交
1419 1420 1421 1422
    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);
M
Michael Niedermayer 已提交
1423

1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443
    // 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];
        }
    }
1444

1445 1446 1447 1448 1449
    for(y=0; y<block_h; y++){
       //left
        for(x=0; x<start_x; x++){
            buf[x + y*linesize]= buf[start_x + y*linesize];
        }
1450

1451 1452 1453 1454 1455 1456 1457
       //right
        for(x=end_x; x<block_w; x++){
            buf[x + y*linesize]= buf[end_x - 1 + y*linesize];
        }
    }
}

1458
static inline int hpel_motion_lowres(MpegEncContext *s,
M
Michael Niedermayer 已提交
1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480
                                  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);
1481

M
Michael Niedermayer 已提交
1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499
    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;
}

F
Fabrice Bellard 已提交
1500
/* apply one mpeg motion vector to the three components */
B
Benoit Fouet 已提交
1501
static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
1502
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1503
                               int field_based, int bottom_field, int field_select,
B
Benoit Fouet 已提交
1504
                               uint8_t **ref_picture, h264_chroma_mc_func *pix_op,
F
Fabrice Bellard 已提交
1505 1506
                               int motion_x, int motion_y, int h)
{
1507
    uint8_t *ptr_y, *ptr_cb, *ptr_cr;
B
Benoit Fouet 已提交
1508 1509 1510 1511 1512 1513
    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;
1514
    linesize   = s->current_picture.linesize[0] << field_based;
M
Michael Niedermayer 已提交
1515
    uvlinesize = s->current_picture.linesize[1] << field_based;
1516

B
Benoit Fouet 已提交
1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529
    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));
1530

M
Michael Niedermayer 已提交
1531
    if (s->out_format == FMT_H263) {
M
Michael Niedermayer 已提交
1532 1533
        uvsx = ((motion_x>>1) & s_mask) | (sx&1);
        uvsy = ((motion_y>>1) & s_mask) | (sy&1);
M
Michael Niedermayer 已提交
1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
        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));
1549
        uvsrc_y =(s->mb_y*block_s>>field_based) + (my >> (lowres+1));
M
Michael Niedermayer 已提交
1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560
    }

    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;
1561
            if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1562
                uint8_t *uvbuf= s->edge_emu_buffer+18*s->linesize;
1563
                ff_emulated_edge_mc(uvbuf  , ptr_cb, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1564
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
1565
                ff_emulated_edge_mc(uvbuf+16, ptr_cr, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1566 1567 1568 1569 1570 1571
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
                ptr_cb= uvbuf;
                ptr_cr= uvbuf+16;
            }
    }

1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583
    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;
    }

M
Michael Niedermayer 已提交
1584 1585 1586
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres-1](dest_y, ptr_y, linesize, h, sx, sy);
1587

1588
    if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1589 1590 1591 1592 1593
        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);
    }
M
Michael Niedermayer 已提交
1594
    //FIXME h261 lowres loop filter
M
Michael Niedermayer 已提交
1595 1596
}

M
Michael Niedermayer 已提交
1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608
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;
1609

M
Michael Niedermayer 已提交
1610 1611 1612 1613 1614 1615 1616 1617 1618
    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);
1619

M
Michael Niedermayer 已提交
1620 1621 1622 1623
    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));
1624

M
Michael Niedermayer 已提交
1625 1626 1627 1628 1629 1630 1631 1632 1633
    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;
        }
1634
    }
M
Michael Niedermayer 已提交
1635 1636 1637
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
1638

M
Michael Niedermayer 已提交
1639 1640 1641 1642 1643 1644 1645 1646
    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);
}

M
doxy  
Michael Niedermayer 已提交
1647
/**
D
Diego Biurrun 已提交
1648
 * motion compensation of a single macroblock
M
doxy  
Michael Niedermayer 已提交
1649 1650 1651 1652 1653 1654 1655 1656 1657
 * @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
 */
B
Benoit Fouet 已提交
1658
static inline void MPV_motion_lowres(MpegEncContext *s,
1659
                              uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1660
                              int dir, uint8_t **ref_picture,
B
Benoit Fouet 已提交
1661
                              h264_chroma_mc_func *pix_op)
F
Fabrice Bellard 已提交
1662
{
B
Benoit Fouet 已提交
1663
    int mx, my;
F
Fabrice Bellard 已提交
1664
    int mb_x, mb_y, i;
B
Benoit Fouet 已提交
1665 1666
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
F
Fabrice Bellard 已提交
1667 1668 1669 1670 1671 1672

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

    switch(s->mv_type) {
    case MV_TYPE_16X16:
B
Benoit Fouet 已提交
1673 1674 1675 1676
        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);
F
Fabrice Bellard 已提交
1677 1678
        break;
    case MV_TYPE_8X8:
M
Michael Niedermayer 已提交
1679 1680 1681
        mx = 0;
        my = 0;
            for(i=0;i<4;i++) {
B
Benoit Fouet 已提交
1682
                hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) * s->linesize)*block_s,
1683
                            ref_picture[0], 0, 0,
B
Benoit Fouet 已提交
1684
                            (2*mb_x + (i & 1))*block_s, (2*mb_y + (i >>1))*block_s,
M
Michael Niedermayer 已提交
1685
                            s->width, s->height, s->linesize,
B
Benoit Fouet 已提交
1686 1687
                            s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
                            block_s, block_s, pix_op,
M
Michael Niedermayer 已提交
1688
                            s->mv[dir][i][0], s->mv[dir][i][1]);
M
Michael Niedermayer 已提交
1689 1690 1691

                mx += s->mv[dir][i][0];
                my += s->mv[dir][i][1];
M
Michael Niedermayer 已提交
1692
            }
M
Michael Niedermayer 已提交
1693

1694
        if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY))
B
Benoit Fouet 已提交
1695
            chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture, pix_op, mx, my);
F
Fabrice Bellard 已提交
1696 1697 1698
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
B
Benoit Fouet 已提交
1699 1700 1701 1702 1703 1704 1705 1706 1707 1708
            /* 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);
F
Fabrice Bellard 已提交
1709
        } else {
1710 1711
            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;
1712
            }
F
Fabrice Bellard 已提交
1713

B
Benoit Fouet 已提交
1714
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1715 1716
                        0, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
B
Benoit Fouet 已提交
1717
                        s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
F
Fabrice Bellard 已提交
1718
        }
1719
        break;
M
Michael Niedermayer 已提交
1720 1721 1722
    case MV_TYPE_16X8:
        for(i=0; i<2; i++){
            uint8_t ** ref2picture;
1723

M
Michael Niedermayer 已提交
1724
            if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == B_TYPE || s->first_field){
1725 1726
                ref2picture= ref_picture;
            }else{
1727
                ref2picture= s->current_picture_ptr->data;
1728
            }
1729

B
Benoit Fouet 已提交
1730
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1731
                        0, 0, s->field_select[dir][i],
1732
                        ref2picture, pix_op,
B
Benoit Fouet 已提交
1733
                        s->mv[dir][i][0], s->mv[dir][i][1] + 2*block_s*i, block_s);
1734

B
Benoit Fouet 已提交
1735 1736 1737
            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;
1738
        }
F
Fabrice Bellard 已提交
1739
        break;
1740 1741
    case MV_TYPE_DMV:
        if(s->picture_structure == PICT_FRAME){
M
Michael Niedermayer 已提交
1742 1743 1744
            for(i=0; i<2; i++){
                int j;
                for(j=0; j<2; j++){
B
Benoit Fouet 已提交
1745
                    mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1746 1747
                                1, j, j^i,
                                ref_picture, pix_op,
B
Benoit Fouet 已提交
1748
                                s->mv[dir][2*i + j][0], s->mv[dir][2*i + j][1], block_s);
M
Michael Niedermayer 已提交
1749
                }
B
Benoit Fouet 已提交
1750
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
M
Michael Niedermayer 已提交
1751
            }
1752
        }else{
M
Michael Niedermayer 已提交
1753
            for(i=0; i<2; i++){
B
Benoit Fouet 已提交
1754
                mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1755 1756
                            0, 0, s->picture_structure != i+1,
                            ref_picture, pix_op,
B
Benoit Fouet 已提交
1757
                            s->mv[dir][2*i][0],s->mv[dir][2*i][1],2*block_s);
1758

M
Michael Niedermayer 已提交
1759
                // after put we make avg of the same block
B
Benoit Fouet 已提交
1760
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
1761

M
Michael Niedermayer 已提交
1762 1763
                //opposite parity is always in the same frame if this is second field
                if(!s->first_field){
1764
                    ref_picture = s->current_picture_ptr->data;
M
Michael Niedermayer 已提交
1765
                }
1766
            }
1767 1768
        }
    break;
M
Michael Niedermayer 已提交
1769
    default: assert(0);
F
Fabrice Bellard 已提交
1770 1771 1772
    }
}

B
Benoit Fouet 已提交
1773 1774 1775
/* put block[] to dest[] */
static inline void put_dct(MpegEncContext *s,
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
M
Michael Niedermayer 已提交
1776
{
B
Benoit Fouet 已提交
1777 1778 1779
    s->dct_unquantize_intra(s, block, i, qscale);
    s->dsp.idct_put (dest, line_size, block);
}
1780

B
Benoit Fouet 已提交
1781 1782 1783 1784 1785 1786 1787 1788
/* 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);
    }
}
1789

1790
static inline void add_dequant_dct(MpegEncContext *s,
M
Michael Niedermayer 已提交
1791
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
M
Michael Niedermayer 已提交
1792
{
F
Fabrice Bellard 已提交
1793
    if (s->block_last_index[i] >= 0) {
1794
        s->dct_unquantize_inter(s, block, i, qscale);
1795

1796
        s->dsp.idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1797 1798 1799
    }
}

1800 1801 1802 1803 1804
/**
 * cleans dc, ac, coded_block for the current non intra MB
 */
void ff_clean_intra_table_entries(MpegEncContext *s)
{
1805
    int wrap = s->b8_stride;
1806
    int xy = s->block_index[0];
1807 1808 1809

    s->dc_val[0][xy           ] =
    s->dc_val[0][xy + 1       ] =
1810 1811 1812
    s->dc_val[0][xy     + wrap] =
    s->dc_val[0][xy + 1 + wrap] = 1024;
    /* ac pred */
1813 1814
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
1815 1816 1817 1818 1819 1820 1821
    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 */
1822 1823
    wrap = s->mb_stride;
    xy = s->mb_x + s->mb_y * wrap;
1824 1825 1826
    s->dc_val[1][xy] =
    s->dc_val[2][xy] = 1024;
    /* ac pred */
1827 1828
    memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
1829

1830
    s->mbintra_table[xy]= 0;
1831 1832
}

F
Fabrice Bellard 已提交
1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
/* 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)
 */
1843
static av_always_inline void MPV_decode_mb_internal(MpegEncContext *s, DCTELEM block[12][64], int lowres_flag)
F
Fabrice Bellard 已提交
1844
{
1845
    int mb_x, mb_y;
1846
    const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
I
Ivan Kalvachev 已提交
1847 1848
#ifdef HAVE_XVMC
    if(s->avctx->xvmc_acceleration){
1849
        XVMC_decode_mb(s);//xvmc uses pblocks
I
Ivan Kalvachev 已提交
1850 1851 1852
        return;
    }
#endif
F
Fabrice Bellard 已提交
1853 1854 1855 1856

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

1857 1858 1859 1860 1861 1862 1863 1864 1865
    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]];
    }

M
cleanup  
Michael Niedermayer 已提交
1866
    s->current_picture.qscale_table[mb_xy]= s->qscale;
1867

F
Fabrice Bellard 已提交
1868 1869
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
1870
        if (s->h263_pred || s->h263_aic) {
M
Michael Niedermayer 已提交
1871
            if(s->mbintra_table[mb_xy])
1872
                ff_clean_intra_table_entries(s);
F
Fabrice Bellard 已提交
1873
        } else {
1874 1875
            s->last_dc[0] =
            s->last_dc[1] =
F
Fabrice Bellard 已提交
1876 1877 1878
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
1879
    else if (s->h263_pred || s->h263_aic)
M
Michael Niedermayer 已提交
1880
        s->mbintra_table[mb_xy]=1;
1881

1882
    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
1883
        uint8_t *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
1884
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
1885 1886
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
1887
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
M
Michael Niedermayer 已提交
1888
        const int uvlinesize= s->current_picture.linesize[1];
M
Michael Niedermayer 已提交
1889 1890
        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;
1891

M
cleanup  
Michael Niedermayer 已提交
1892 1893 1894
        /* 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){
1895
            uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
M
cleanup  
Michael Niedermayer 已提交
1896
            const int age= s->current_picture.age;
M
Michael Niedermayer 已提交
1897

M
cleanup  
Michael Niedermayer 已提交
1898 1899
            assert(age);

1900 1901
            if (s->mb_skipped) {
                s->mb_skipped= 0;
M
cleanup  
Michael Niedermayer 已提交
1902
                assert(s->pict_type!=I_TYPE);
1903

1904
                (*mbskip_ptr) ++; /* indicate that this time we skipped it */
M
Michael Niedermayer 已提交
1905 1906
                if(*mbskip_ptr >99) *mbskip_ptr= 99;

M
cleanup  
Michael Niedermayer 已提交
1907
                /* if previous was skipped too, then nothing to do !  */
1908 1909
                if (*mbskip_ptr >= age && s->current_picture.reference){
                    return;
M
cleanup  
Michael Niedermayer 已提交
1910
                }
1911 1912 1913 1914
            } 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{
1915 1916
                *mbskip_ptr = 0; /* not skipped */
            }
M
Michael Niedermayer 已提交
1917
        }
1918

1919
        dct_linesize = linesize << s->interlaced_dct;
M
Michael Niedermayer 已提交
1920
        dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
1921

1922 1923 1924 1925 1926
        if(readable){
            dest_y=  s->dest[0];
            dest_cb= s->dest[1];
            dest_cr= s->dest[2];
        }else{
1927
            dest_y = s->b_scratchpad;
1928
            dest_cb= s->b_scratchpad+16*linesize;
1929
            dest_cr= s->b_scratchpad+32*linesize;
1930
        }
M
Michael Niedermayer 已提交
1931

F
Fabrice Bellard 已提交
1932 1933
        if (!s->mb_intra) {
            /* motion handling */
1934
            /* decoding or more than one mb_type (MC was already done otherwise) */
1935
            if(!s->encoding){
M
Michael Niedermayer 已提交
1936 1937
                if(lowres_flag){
                    h264_chroma_mc_func *op_pix = s->dsp.put_h264_chroma_pixels_tab;
F
Fabrice Bellard 已提交
1938

M
Michael Niedermayer 已提交
1939 1940 1941 1942 1943 1944 1945 1946
                    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{
1947
                    op_qpix= s->me.qpel_put;
1948
                    if ((!s->no_rounding) || s->pict_type==B_TYPE){
M
Michael Niedermayer 已提交
1949 1950 1951 1952 1953 1954 1955
                        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;
1956
                        op_qpix= s->me.qpel_avg;
M
Michael Niedermayer 已提交
1957 1958 1959 1960
                    }
                    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);
                    }
1961
                }
F
Fabrice Bellard 已提交
1962 1963
            }

M
Michael Niedermayer 已提交
1964
            /* skip dequant / idct if we are really late ;) */
M
Michael Niedermayer 已提交
1965 1966 1967 1968 1969 1970 1971
            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;
            }
M
Michael Niedermayer 已提交
1972

F
Fabrice Bellard 已提交
1973
            /* add dct residue */
M
Michael Niedermayer 已提交
1974
            if(s->encoding || !(   s->h263_msmpeg4 || s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO
1975
                                || (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant))){
M
Michael Niedermayer 已提交
1976 1977 1978 1979
                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);
M
Michael Niedermayer 已提交
1980

1981
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
                    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);
                    }
M
Michael Niedermayer 已提交
1993
                }
M
Michael Niedermayer 已提交
1994
            } else if(s->codec_id != CODEC_ID_WMV2){
M
Michael Niedermayer 已提交
1995 1996 1997 1998
                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);
F
Fabrice Bellard 已提交
1999

2000
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021
                    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
            }
2022
            else if (ENABLE_WMV2) {
M
Michael Niedermayer 已提交
2023
                ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
M
Michael Niedermayer 已提交
2024
            }
F
Fabrice Bellard 已提交
2025 2026
        } else {
            /* dct only in intra block */
M
Michael Niedermayer 已提交
2027
            if(s->encoding || !(s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO)){
M
Michael Niedermayer 已提交
2028 2029 2030 2031
                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);
2032

2033
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2034 2035 2036 2037 2038 2039 2040 2041 2042
                    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);
B
Benoit Fouet 已提交
2043
                        put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
M
Michael Niedermayer 已提交
2044 2045 2046
                    }
                }
            }else{
B
Benoit Fouet 已提交
2047 2048 2049 2050
                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]);
2051

2052
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
B
Benoit Fouet 已提交
2053 2054 2055 2056
                    if(s->chroma_y_shift){
                        s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                        s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
                    }else{
2057

B
Benoit Fouet 已提交
2058 2059
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
2060

B
Benoit Fouet 已提交
2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072
                        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
2073
            }
M
Michael Niedermayer 已提交
2074
        }
B
Benoit Fouet 已提交
2075 2076 2077 2078 2079
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);
2080
        }
M
Michael Niedermayer 已提交
2081 2082 2083
    }
}

B
Benoit Fouet 已提交
2084 2085 2086
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);
M
Michael Niedermayer 已提交
2087 2088
}

B
Benoit Fouet 已提交
2089 2090 2091 2092 2093 2094 2095 2096
/**
 *
 * @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];
M
Michael Niedermayer 已提交
2097

B
Benoit Fouet 已提交
2098 2099 2100 2101
        if(s->picture_structure != PICT_FRAME){
            h <<= 1;
            y <<= 1;
            if(s->first_field  && !(s->avctx->slice_flags&SLICE_FLAG_ALLOW_FIELD)) return;
M
Michael Niedermayer 已提交
2102 2103
        }

B
Benoit Fouet 已提交
2104
        h= FFMIN(h, s->avctx->height - y);
M
Michael Niedermayer 已提交
2105

B
Benoit Fouet 已提交
2106 2107 2108 2109 2110 2111
        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;
2112

B
Benoit Fouet 已提交
2113 2114 2115 2116 2117
        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;
M
Michael Niedermayer 已提交
2118
        }else{
B
Benoit Fouet 已提交
2119 2120 2121 2122
            offset[0]= y * s->linesize;;
            offset[1]=
            offset[2]= (y >> s->chroma_y_shift) * s->uvlinesize;
            offset[3]= 0;
M
Michael Niedermayer 已提交
2123
        }
2124

B
Benoit Fouet 已提交
2125
        emms_c();
M
Michael Niedermayer 已提交
2126

B
Benoit Fouet 已提交
2127 2128 2129 2130
        s->avctx->draw_horiz_band(s->avctx, src, offset,
                                  y, s->picture_structure, h);
    }
}
2131

B
Benoit Fouet 已提交
2132 2133 2134 2135
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;
M
Michael Niedermayer 已提交
2136

B
Benoit Fouet 已提交
2137 2138 2139 2140 2141 2142 2143
    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
2144

B
Benoit Fouet 已提交
2145 2146 2147
    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));
2148

B
Benoit Fouet 已提交
2149 2150 2151 2152 2153
    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);
M
Michael Niedermayer 已提交
2154 2155
    }
}
2156

B
Benoit Fouet 已提交
2157 2158 2159
void ff_mpeg_flush(AVCodecContext *avctx){
    int i;
    MpegEncContext *s = avctx->priv_data;
M
Michael Niedermayer 已提交
2160

B
Benoit Fouet 已提交
2161 2162
    if(s==NULL || s->picture==NULL)
        return;
M
Michael Niedermayer 已提交
2163

B
Benoit Fouet 已提交
2164 2165 2166 2167
    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]);
F
Fabrice Bellard 已提交
2168
    }
B
Benoit Fouet 已提交
2169
    s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
2170

B
Benoit Fouet 已提交
2171
    s->mb_x= s->mb_y= 0;
M
Michael Niedermayer 已提交
2172

B
Benoit Fouet 已提交
2173 2174 2175 2176 2177 2178 2179 2180
    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;
F
Fabrice Bellard 已提交
2181 2182
}

2183
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
2184
                                   DCTELEM *block, int n, int qscale)
F
Fabrice Bellard 已提交
2185
{
M
Michael Niedermayer 已提交
2186
    int i, level, nCoeffs;
2187
    const uint16_t *quant_matrix;
F
Fabrice Bellard 已提交
2188

2189
    nCoeffs= s->block_last_index[n];
2190 2191

    if (n < 4)
2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208
        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;
F
Fabrice Bellard 已提交
2209
            }
2210
            block[j] = level;
F
Fabrice Bellard 已提交
2211
        }
2212 2213 2214
    }
}

2215
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
2216 2217 2218 2219 2220 2221
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;

    nCoeffs= s->block_last_index[n];
2222

2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237
    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;
F
Fabrice Bellard 已提交
2238
            }
2239
            block[j] = level;
F
Fabrice Bellard 已提交
2240 2241 2242
        }
    }
}
2243

2244
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
2245 2246 2247
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
2248
    const uint16_t *quant_matrix;
2249

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

    if (n < 4)
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273
        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;
        }
    }
}

2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306
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;
}

2307
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
2308 2309 2310 2311 2312 2313 2314 2315
                                   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];
2316

2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337
    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;
}

2338
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
2339 2340 2341 2342
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
    int nCoeffs;
2343

2344
    assert(s->block_last_index[n]>=0);
2345

2346
    qmul = qscale << 1;
2347

2348
    if (!s->h263_aic) {
2349
        if (n < 4)
2350 2351 2352
            block[0] = block[0] * s->y_dc_scale;
        else
            block[0] = block[0] * s->c_dc_scale;
2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368
        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;
2369
            }
2370
            block[i] = level;
2371 2372 2373 2374
        }
    }
}

2375
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
2376 2377 2378
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
2379
    int nCoeffs;
2380

2381
    assert(s->block_last_index[n]>=0);
2382

2383 2384
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
2385

2386
    nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
2387

2388
    for(i=0; i<=nCoeffs; i++) {
2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
            }
            block[i] = level;
        }
    }
}
F
Fabrice Bellard 已提交
2400

2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416
/**
 * 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 ];
}