mpegvideo.c 87.0 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);
M
Michael Niedermayer 已提交
56

57 58 59
int  XVMC_field_start(MpegEncContext*s, AVCodecContext *avctx);
void XVMC_field_end(MpegEncContext *s);
void XVMC_decode_mb(MpegEncContext *s);
I
Ivan Kalvachev 已提交
60

F
Fabrice Bellard 已提交
61 62 63 64 65 66

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

//#define DEBUG

67

M
Michael Niedermayer 已提交
68 69 70 71 72
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
};

73 74 75 76 77 78 79 80
const uint8_t ff_mpeg1_dc_scale_table[128]={
//  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
    8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
    8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
    8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
    8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
};

81

M
Michael Niedermayer 已提交
82 83 84
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 已提交
85 86 87 88
    assert(p<=end);
    if(p>=end)
        return end;

M
Michael Niedermayer 已提交
89 90 91 92 93 94 95 96
    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 已提交
97 98 99
        if     (p[-1] > 1      ) p+= 3;
        else if(p[-2]          ) p+= 2;
        else if(p[-3]|(p[-1]-1)) p++;
M
Michael Niedermayer 已提交
100 101 102 103 104 105
        else{
            p++;
            break;
        }
    }

M
Michael Niedermayer 已提交
106
    p= FFMIN(p, end)-4;
107
    *state= AV_RB32(p);
M
Michael Niedermayer 已提交
108 109 110 111

    return p+4;
}

112
/* init common dct for both encoder and decoder */
113
int ff_dct_common_init(MpegEncContext *s)
F
Fabrice Bellard 已提交
114
{
115 116 117 118 119
    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;
120 121
    if(s->flags & CODEC_FLAG_BITEXACT)
        s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_bitexact;
122
    s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_c;
123

124
#if   HAVE_MMX
125
    MPV_common_init_mmx(s);
126
#elif ARCH_ALPHA
127
    MPV_common_init_axp(s);
128
#elif CONFIG_MLIB
129
    MPV_common_init_mlib(s);
130
#elif HAVE_MMI
131
    MPV_common_init_mmi(s);
132
#elif ARCH_ARM
M
Måns Rullgård 已提交
133
    MPV_common_init_arm(s);
134
#elif HAVE_ALTIVEC
135
    MPV_common_init_altivec(s);
136
#elif ARCH_BFIN
137 138
    MPV_common_init_bfin(s);
#endif
139

140
    /* load & permutate scantables
141
       note: only wmv uses different ones
142
    */
M
Michael Niedermayer 已提交
143 144 145 146 147 148 149
    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 已提交
150 151
    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);
152

153 154 155
    return 0;
}

156
void ff_copy_picture(Picture *dst, Picture *src){
157 158 159 160
    *dst = *src;
    dst->type= FF_BUFFER_TYPE_COPY;
}

M
cleanup  
Michael Niedermayer 已提交
161
/**
M
Michael Niedermayer 已提交
162 163
 * allocates a Picture
 * The pixels are allocated/set by calling get_buffer() if shared=0
M
cleanup  
Michael Niedermayer 已提交
164
 */
B
Benoit Fouet 已提交
165
int alloc_picture(MpegEncContext *s, Picture *pic, int shared){
D
Diego Biurrun 已提交
166
    const int big_mb_num= s->mb_stride*(s->mb_height+1) + 1; //the +1 is needed so memset(,,stride*height) does not sig11
167
    const int mb_array_size= s->mb_stride*s->mb_height;
168 169
    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 已提交
170
    int i;
D
Daniel Kristjansson 已提交
171
    int r= -1;
172

M
Michael Niedermayer 已提交
173 174 175 176 177 178
    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]);
179

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

M
Michael Niedermayer 已提交
182
        if(r<0 || !pic->age || !pic->type || !pic->data[0]){
183
            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 已提交
184 185 186 187
            return -1;
        }

        if(s->linesize && (s->linesize != pic->linesize[0] || s->uvlinesize != pic->linesize[1])){
188
            av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (stride changed)\n");
D
Daniel Kristjansson 已提交
189
            s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
M
Michael Niedermayer 已提交
190 191 192 193
            return -1;
        }

        if(pic->linesize[1] != pic->linesize[2]){
D
Diego Biurrun 已提交
194
            av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (uv stride mismatch)\n");
D
Daniel Kristjansson 已提交
195
            s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
M
Michael Niedermayer 已提交
196 197 198 199 200
            return -1;
        }

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

M
Michael Niedermayer 已提交
203
    if(pic->qscale_table==NULL){
204
        if (s->encoding) {
205 206 207
            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 已提交
208
        }
M
cleanup  
Michael Niedermayer 已提交
209

210
        CHECKED_ALLOCZ(pic->mbskip_table , mb_array_size * sizeof(uint8_t)+2) //the +2 is for the slice end check
211
        CHECKED_ALLOCZ(pic->qscale_table , mb_array_size * sizeof(uint8_t))
212 213
        CHECKED_ALLOCZ(pic->mb_type_base , (big_mb_num + s->mb_stride) * sizeof(uint32_t))
        pic->mb_type= pic->mb_type_base + 2*s->mb_stride+1;
M
Michael Niedermayer 已提交
214 215
        if(s->out_format == FMT_H264){
            for(i=0; i<2; i++){
M
Michael Niedermayer 已提交
216 217
                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;
218
                CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
219
            }
M
Michael Niedermayer 已提交
220
            pic->motion_subsample_log2= 2;
221
        }else if(s->out_format == FMT_H263 || s->encoding || (s->avctx->debug&FF_DEBUG_MV) || (s->avctx->debug_mv)){
222
            for(i=0; i<2; i++){
M
Michael Niedermayer 已提交
223 224
                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;
225
                CHECKED_ALLOCZ(pic->ref_index[i], b8_array_size * sizeof(uint8_t))
M
Michael Niedermayer 已提交
226
            }
M
Michael Niedermayer 已提交
227
            pic->motion_subsample_log2= 3;
M
Michael Niedermayer 已提交
228
        }
229 230 231
        if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
            CHECKED_ALLOCZ(pic->dct_coeff, 64 * mb_array_size * sizeof(DCTELEM)*6)
        }
232
        pic->qstride= s->mb_stride;
233
        CHECKED_ALLOCZ(pic->pan_scan , 1 * sizeof(AVPanScan))
M
Michael Niedermayer 已提交
234
    }
M
Michael Niedermayer 已提交
235

D
Diego Biurrun 已提交
236 237
    /* It might be nicer if the application would keep track of these
     * but it would require an API change. */
238
    memmove(s->prev_pict_types+1, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE-1);
239
    s->prev_pict_types[0]= s->dropable ? FF_B_TYPE : s->pict_type;
240
    if(pic->age < PREV_PICT_TYPES_BUFFER_SIZE && s->prev_pict_types[pic->age] == FF_B_TYPE)
D
Diego Biurrun 已提交
241
        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.
242

M
cleanup  
Michael Niedermayer 已提交
243 244
    return 0;
fail: //for the CHECKED_ALLOCZ macro
D
Daniel Kristjansson 已提交
245 246
    if(r>=0)
        s->avctx->release_buffer(s->avctx, (AVFrame*)pic);
M
cleanup  
Michael Niedermayer 已提交
247 248 249
    return -1;
}

M
Michael Niedermayer 已提交
250 251 252
/**
 * deallocates a picture
 */
M
cleanup  
Michael Niedermayer 已提交
253 254
static void free_picture(MpegEncContext *s, Picture *pic){
    int i;
M
Michael Niedermayer 已提交
255 256

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

M
cleanup  
Michael Niedermayer 已提交
260 261 262 263 264
    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 已提交
265
    av_freep(&pic->mb_type_base);
266
    av_freep(&pic->dct_coeff);
267
    av_freep(&pic->pan_scan);
M
Michael Niedermayer 已提交
268 269
    pic->mb_type= NULL;
    for(i=0; i<2; i++){
270
        av_freep(&pic->motion_val_base[i]);
M
Michael Niedermayer 已提交
271 272
        av_freep(&pic->ref_index[i]);
    }
273

274
    if(pic->type == FF_BUFFER_TYPE_SHARED){
M
Michael Niedermayer 已提交
275 276 277 278
        for(i=0; i<4; i++){
            pic->base[i]=
            pic->data[i]= NULL;
        }
279
        pic->type= 0;
M
cleanup  
Michael Niedermayer 已提交
280 281 282
    }
}

283 284 285
static int init_duplicate_context(MpegEncContext *s, MpegEncContext *base){
    int i;

286
    // edge emu needs blocksize + filter length - 1 (=17x17 for halfpel / 21x21 for h264)
L
Loren Merritt 已提交
287 288
    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;
289

D
Diego Biurrun 已提交
290
     //FIXME should be linesize instead of s->width*2 but that is not known before get_buffer()
291
    CHECKED_ALLOCZ(s->me.scratchpad,  (s->width+64)*4*16*2*sizeof(uint8_t))
292
    s->me.temp=         s->me.scratchpad;
293 294 295 296 297 298 299 300 301
    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))
        }
302
    }
303
    CHECKED_ALLOCZ(s->blocks, 64*12*2 * sizeof(DCTELEM))
304 305 306 307 308 309 310 311 312 313 314 315 316 317 318
    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);
319
    s->me.temp=
320 321
    s->rd_scratchpad=
    s->b_scratchpad=
322
    s->obmc_scratchpad= NULL;
323

324 325 326 327 328 329 330 331 332 333 334 335
    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);
336
    COPY(me.temp);
337 338 339 340 341 342 343 344 345 346 347 348
    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);
349 350
    COPY(dct_count[0]);
    COPY(dct_count[1]);
351 352 353
#undef COPY
}

354
void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src){
355
    MpegEncContext bak;
356
    int i;
357 358 359 360 361
    //FIXME copy only needed parts
//START_TIMER
    backup_duplicate_context(&bak, dst);
    memcpy(dst, src, sizeof(MpegEncContext));
    backup_duplicate_context(dst, &bak);
362 363 364
    for(i=0;i<12;i++){
        dst->pblocks[i] = (short *)(&dst->block[i]);
    }
365 366 367
//STOP_TIMER("update_duplicate_context") //about 10k cycles / 0.01 sec for 1000frames on 1ghz with 2 threads
}

M
Michael Niedermayer 已提交
368 369 370 371
/**
 * 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 已提交
372
void MPV_common_defaults(MpegEncContext *s){
M
Michael Niedermayer 已提交
373 374 375 376 377 378 379 380 381 382 383 384
    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;
385 386 387

    s->f_code = 1;
    s->b_code = 1;
M
Michael Niedermayer 已提交
388 389 390 391 392 393 394 395 396 397
}

/**
 * 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);
}

398
/**
M
Michael Niedermayer 已提交
399 400 401
 * init common structure for both encoder and decoder.
 * this assumes that some variables like width/height are already set
 */
402 403
int MPV_common_init(MpegEncContext *s)
{
404
    int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y, threads;
405

L
Loren Merritt 已提交
406 407 408
    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 已提交
409 410 411 412
        av_log(s->avctx, AV_LOG_ERROR, "too many threads\n");
        return -1;
    }

413 414 415
    if((s->width || s->height) && avcodec_check_dimensions(s->avctx, s->width, s->height))
        return -1;

416
    dsputil_init(&s->dsp, s->avctx);
417
    ff_dct_common_init(s);
418

419
    s->flags= s->avctx->flags;
420
    s->flags2= s->avctx->flags2;
421

M
cleanup  
Michael Niedermayer 已提交
422
    s->mb_width  = (s->width  + 15) / 16;
423
    s->mb_stride = s->mb_width + 1;
424 425
    s->b8_stride = s->mb_width*2 + 1;
    s->b4_stride = s->mb_width*4 + 1;
426
    mb_array_size= s->mb_height * s->mb_stride;
M
Michael Niedermayer 已提交
427
    mv_table_size= (s->mb_height+2) * s->mb_stride + 1;
428

429 430 431 432
    /* set chroma shifts */
    avcodec_get_chroma_sub_sample(s->avctx->pix_fmt,&(s->chroma_x_shift),
                                                    &(s->chroma_y_shift) );

433 434 435
    /* 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;
436 437

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

439 440 441
    s->block_wrap[0]=
    s->block_wrap[1]=
    s->block_wrap[2]=
442
    s->block_wrap[3]= s->b8_stride;
443
    s->block_wrap[4]=
444
    s->block_wrap[5]= s->mb_stride;
445

446 447
    y_size = s->b8_stride * (2 * s->mb_height + 1);
    c_size = s->mb_stride * (s->mb_height + 1);
448
    yc_size = y_size + 2 * c_size;
449

450
    /* convert fourcc to upper case */
451
    s->codec_tag=          toupper( s->avctx->codec_tag     &0xFF)
452
                        + (toupper((s->avctx->codec_tag>>8 )&0xFF)<<8 )
453
                        + (toupper((s->avctx->codec_tag>>16)&0xFF)<<16)
454
                        + (toupper((s->avctx->codec_tag>>24)&0xFF)<<24);
455

456
    s->stream_codec_tag=          toupper( s->avctx->stream_codec_tag     &0xFF)
457
                               + (toupper((s->avctx->stream_codec_tag>>8 )&0xFF)<<8 )
458
                               + (toupper((s->avctx->stream_codec_tag>>16)&0xFF)<<16)
459 460
                               + (toupper((s->avctx->stream_codec_tag>>24)&0xFF)<<24);

M
Michael Niedermayer 已提交
461
    s->avctx->coded_frame= (AVFrame*)&s->current_picture;
M
cleanup  
Michael Niedermayer 已提交
462

463 464 465 466 467 468 469
    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?
470

471
    if (s->encoding) {
472
        /* Allocate MV tables */
473 474 475 476 477 478 479 480 481 482 483 484
        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;
485

486 487 488
        if(s->msmpeg4_version){
            CHECKED_ALLOCZ(s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int));
        }
M
Michael Niedermayer 已提交
489
        CHECKED_ALLOCZ(s->avctx->stats_out, 256);
490 491

        /* Allocate MB type table */
M
Michael Niedermayer 已提交
492
        CHECKED_ALLOCZ(s->mb_type  , mb_array_size * sizeof(uint16_t)) //needed for encoding
493

494
        CHECKED_ALLOCZ(s->lambda_table, mb_array_size * sizeof(int))
495

496 497
        CHECKED_ALLOCZ(s->q_intra_matrix, 64*32 * sizeof(int))
        CHECKED_ALLOCZ(s->q_inter_matrix, 64*32 * sizeof(int))
498 499
        CHECKED_ALLOCZ(s->q_intra_matrix16, 64*32*2 * sizeof(uint16_t))
        CHECKED_ALLOCZ(s->q_inter_matrix16, 64*32*2 * sizeof(uint16_t))
500 501
        CHECKED_ALLOCZ(s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
        CHECKED_ALLOCZ(s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture*))
502

503 504 505
        if(s->avctx->noise_reduction){
            CHECKED_ALLOCZ(s->dct_offset, 2 * 64 * sizeof(uint16_t))
        }
506
    }
507 508
    CHECKED_ALLOCZ(s->picture, MAX_PICTURE_COUNT * sizeof(Picture))

509
    CHECKED_ALLOCZ(s->error_status_table, mb_array_size*sizeof(uint8_t))
510

M
Michael Niedermayer 已提交
511
    if(s->codec_id==CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){
512
        /* interlaced direct mode decoding tables */
M
Michael Niedermayer 已提交
513 514 515 516 517 518 519 520 521 522 523 524 525
            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 已提交
526
    }
M
Michael Niedermayer 已提交
527
    if (s->out_format == FMT_H263) {
F
Fabrice Bellard 已提交
528
        /* ac values */
529 530 531
        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 已提交
532
        s->ac_val[2] = s->ac_val[1] + c_size;
533

F
Fabrice Bellard 已提交
534
        /* cbp values */
535 536
        CHECKED_ALLOCZ(s->coded_block_base, y_size);
        s->coded_block= s->coded_block_base + s->b8_stride + 1;
537

538
        /* cbp, ac_pred, pred_dir */
539 540
        CHECKED_ALLOCZ(s->cbp_table  , mb_array_size * sizeof(uint8_t))
        CHECKED_ALLOCZ(s->pred_dir_table, mb_array_size * sizeof(uint8_t))
541
    }
542

543 544 545
    if (s->h263_pred || s->h263_plus || !s->encoding) {
        /* dc values */
        //MN: we need these for error resilience of intra-frames
546 547 548
        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;
549 550
        s->dc_val[2] = s->dc_val[1] + c_size;
        for(i=0;i<yc_size;i++)
551
            s->dc_val_base[i] = 1024;
552 553
    }

554
    /* which mb is a intra block */
555 556
    CHECKED_ALLOCZ(s->mbintra_table, mb_array_size);
    memset(s->mbintra_table, 1, mb_array_size);
557

558
    /* init macroblock skip table */
559
    CHECKED_ALLOCZ(s->mbskip_table, mb_array_size+2);
560
    //Note the +1 is for a quicker mpeg4 slice_end detection
561
    CHECKED_ALLOCZ(s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE);
562

563
    s->parse_context.state= -1;
564 565
    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);
566 567
       s->visualization_buffer[1] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
       s->visualization_buffer[2] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
568
    }
569

F
Fabrice Bellard 已提交
570
    s->context_initialized = 1;
571 572

    s->thread_context[0]= s;
573
    threads = s->avctx->thread_count;
574 575

    for(i=1; i<threads; i++){
576 577 578 579
        s->thread_context[i]= av_malloc(sizeof(MpegEncContext));
        memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
    }

580
    for(i=0; i<threads; i++){
581 582 583 584 585 586
        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 已提交
587 588
    return 0;
 fail:
589
    MPV_common_end(s);
F
Fabrice Bellard 已提交
590 591 592 593 594 595
    return -1;
}

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

598 599 600 601 602 603 604
    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]);
    }

605 606 607
    av_freep(&s->parse_context.buffer);
    s->parse_context.buffer_size=0;

608
    av_freep(&s->mb_type);
609 610 611 612 613 614 615 616 617 618 619 620
    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 已提交
621 622 623 624 625 626 627 628 629 630 631 632
    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]);
    }
633

634 635 636
    av_freep(&s->dc_val_base);
    av_freep(&s->ac_val_base);
    av_freep(&s->coded_block_base);
637
    av_freep(&s->mbintra_table);
638 639
    av_freep(&s->cbp_table);
    av_freep(&s->pred_dir_table);
640

641
    av_freep(&s->mbskip_table);
642
    av_freep(&s->prev_pict_types);
643
    av_freep(&s->bitstream_buffer);
644 645
    s->allocated_bitstream_buffer_size=0;

M
Michael Niedermayer 已提交
646
    av_freep(&s->avctx->stats_out);
647
    av_freep(&s->ac_stats);
648
    av_freep(&s->error_status_table);
649
    av_freep(&s->mb_index2xy);
650
    av_freep(&s->lambda_table);
651 652
    av_freep(&s->q_intra_matrix);
    av_freep(&s->q_inter_matrix);
653 654
    av_freep(&s->q_intra_matrix16);
    av_freep(&s->q_inter_matrix16);
655 656
    av_freep(&s->input_picture);
    av_freep(&s->reordered_input_picture);
657
    av_freep(&s->dct_offset);
M
cleanup  
Michael Niedermayer 已提交
658

M
Michael Niedermayer 已提交
659 660 661 662
    if(s->picture){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            free_picture(s, &s->picture[i]);
        }
F
Fabrice Bellard 已提交
663
    }
664
    av_freep(&s->picture);
F
Fabrice Bellard 已提交
665
    s->context_initialized = 0;
666 667 668
    s->last_picture_ptr=
    s->next_picture_ptr=
    s->current_picture_ptr= NULL;
M
Michael Niedermayer 已提交
669
    s->linesize= s->uvlinesize= 0;
670

671
    for(i=0; i<3; i++)
672
        av_freep(&s->visualization_buffer[i]);
M
Michael Niedermayer 已提交
673 674

    avcodec_default_free_buffers(s->avctx);
F
Fabrice Bellard 已提交
675 676
}

677
void init_rl(RLTable *rl, uint8_t static_store[2][2*MAX_RUN + MAX_LEVEL + 3])
M
Michael Niedermayer 已提交
678
{
679 680
    int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
    uint8_t index_run[MAX_RUN+1];
M
Michael Niedermayer 已提交
681 682
    int last, run, level, start, end, i;

683
    /* If table is static, we can quit if rl->max_level[0] is not NULL */
684
    if(static_store && rl->max_level[0])
685 686
        return;

M
Michael Niedermayer 已提交
687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
    /* 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;
        }
710 711
        if(static_store)
            rl->max_level[last] = static_store[last];
712 713
        else
            rl->max_level[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
714
        memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
715 716
        if(static_store)
            rl->max_run[last] = static_store[last] + MAX_RUN + 1;
717 718
        else
            rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
M
Michael Niedermayer 已提交
719
        memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
720 721
        if(static_store)
            rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
722 723
        else
            rl->index_run[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
724 725 726 727
        memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
    }
}

728
void init_vlc_rl(RLTable *rl)
729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767
{
    int i, q;

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

        if(q==0){
            qmul=1;
            qadd=0;
        }
        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;
        }
    }
}

M
Michael Niedermayer 已提交
768
int ff_find_unused_picture(MpegEncContext *s, int shared){
M
Michael Niedermayer 已提交
769
    int i;
770

M
Michael Niedermayer 已提交
771 772
    if(shared){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
773
            if(s->picture[i].data[0]==NULL && s->picture[i].type==0) return i;
M
Michael Niedermayer 已提交
774 775 776
        }
    }else{
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
777
            if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) return i; //FIXME
M
Michael Niedermayer 已提交
778 779
        }
        for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
780
            if(s->picture[i].data[0]==NULL) return i;
M
Michael Niedermayer 已提交
781 782 783
        }
    }

784
    av_log(s->avctx, AV_LOG_FATAL, "Internal error, picture buffer overflow\n");
785 786 787 788 789 790 791 792 793 794 795
    /* We could return -1, but the codec would crash trying to draw into a
     * non-existing frame anyway. This is safer than waiting for a random crash.
     * Also the return of this is never useful, an encoder must only allocate
     * as much as allowed in the specification. This has no relationship to how
     * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
     * enough for such valid streams).
     * Plus, a decoder has to check stream validity and remove frames if too
     * many reference frames are around. Waiting for "OOM" is not correct at
     * all. Similarly, missing reference frames have to be replaced by
     * interpolated/MC frames, anything else is a bug in the codec ...
     */
796
    abort();
M
Michael Niedermayer 已提交
797
    return -1;
M
Michael Niedermayer 已提交
798 799
}

800 801 802 803 804 805 806 807 808 809
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;
        }
810

811 812 813 814 815 816
        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 已提交
817 818 819
/**
 * generic function for encode/decode called after coding/decoding the header and before a frame is coded/decoded
 */
820
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
F
Fabrice Bellard 已提交
821
{
M
Michael Niedermayer 已提交
822
    int i;
M
Michael Niedermayer 已提交
823
    AVFrame *pic;
824
    s->mb_skipped = 0;
M
Michael Niedermayer 已提交
825

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

M
cleanup  
Michael Niedermayer 已提交
828
    /* mark&release old frames */
829
    if (s->pict_type != FF_B_TYPE && s->last_picture_ptr && s->last_picture_ptr != s->next_picture_ptr && s->last_picture_ptr->data[0]) {
830
      if(s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3){
M
Michael Niedermayer 已提交
831
        avctx->release_buffer(avctx, (AVFrame*)s->last_picture_ptr);
M
cleanup  
Michael Niedermayer 已提交
832 833 834 835 836

        /* release forgotten pictures */
        /* if(mpeg124/h263) */
        if(!s->encoding){
            for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
837
                if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
838
                    av_log(avctx, AV_LOG_ERROR, "releasing zombie picture\n");
839
                    avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);
M
cleanup  
Michael Niedermayer 已提交
840 841
                }
            }
842
        }
843
      }
844
    }
845 846
alloc:
    if(!s->encoding){
D
Diego Biurrun 已提交
847
        /* release non reference frames */
M
Michael Niedermayer 已提交
848 849 850 851 852 853
        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 已提交
854
        if(s->current_picture_ptr && s->current_picture_ptr->data[0]==NULL)
D
Diego Biurrun 已提交
855
            pic= (AVFrame*)s->current_picture_ptr; //we already have a unused image (maybe it was set before reading the header)
M
Michael Niedermayer 已提交
856 857 858 859 860
        else{
            i= ff_find_unused_picture(s, 0);
            pic= (AVFrame*)&s->picture[i];
        }

861 862 863 864
        pic->reference= 0;
        if (!s->dropable){
            if (s->codec_id == CODEC_ID_H264)
                pic->reference = s->picture_structure;
865
            else if (s->pict_type != FF_B_TYPE)
866 867
                pic->reference = 3;
        }
M
Michael Niedermayer 已提交
868

M
Michael Niedermayer 已提交
869
        pic->coded_picture_number= s->coded_picture_number++;
870

871 872
        if( alloc_picture(s, (Picture*)pic, 0) < 0)
            return -1;
873

M
Michael Niedermayer 已提交
874
        s->current_picture_ptr= (Picture*)pic;
875
        s->current_picture_ptr->top_field_first= s->top_field_first; //FIXME use only the vars from current_pic
876
        s->current_picture_ptr->interlaced_frame= !s->progressive_frame && !s->progressive_sequence;
M
cleanup  
Michael Niedermayer 已提交
877
    }
878

M
Michael Niedermayer 已提交
879
    s->current_picture_ptr->pict_type= s->pict_type;
880
//    if(s->flags && CODEC_FLAG_QSCALE)
881
  //      s->current_picture_ptr->quality= s->new_picture_ptr->quality;
882
    s->current_picture_ptr->key_frame= s->pict_type == FF_I_TYPE;
M
Michael Niedermayer 已提交
883

884
    ff_copy_picture(&s->current_picture, s->current_picture_ptr);
885

886
    if (s->pict_type != FF_B_TYPE) {
M
Michael Niedermayer 已提交
887
        s->last_picture_ptr= s->next_picture_ptr;
888 889
        if(!s->dropable)
            s->next_picture_ptr= s->current_picture_ptr;
F
Fabrice Bellard 已提交
890
    }
891
/*    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,
892 893
        s->last_picture_ptr    ? s->last_picture_ptr->data[0] : NULL,
        s->next_picture_ptr    ? s->next_picture_ptr->data[0] : NULL,
894 895
        s->current_picture_ptr ? s->current_picture_ptr->data[0] : NULL,
        s->pict_type, s->dropable);*/
896

897 898
    if(s->last_picture_ptr) ff_copy_picture(&s->last_picture, s->last_picture_ptr);
    if(s->next_picture_ptr) ff_copy_picture(&s->next_picture, s->next_picture_ptr);
899

900
    if(s->pict_type != FF_I_TYPE && (s->last_picture_ptr==NULL || s->last_picture_ptr->data[0]==NULL) && !s->dropable && s->codec_id != CODEC_ID_H264){
901
        av_log(avctx, AV_LOG_ERROR, "warning: first frame is no keyframe\n");
902
        assert(s->pict_type != FF_B_TYPE); //these should have been dropped if we don't have a reference
903 904 905
        goto alloc;
    }

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

908
    if(s->picture_structure!=PICT_FRAME && s->out_format != FMT_H264){
M
Michael Niedermayer 已提交
909 910 911 912
        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];
913
            }
M
Michael Niedermayer 已提交
914 915 916 917 918
            s->current_picture.linesize[i] *= 2;
            s->last_picture.linesize[i] *=2;
            s->next_picture.linesize[i] *=2;
        }
    }
919

920
    s->hurry_up= s->avctx->hurry_up;
921
    s->error_recognition= avctx->error_recognition;
922

D
Diego Biurrun 已提交
923
    /* set dequantizer, we can't do it during init as it might change for mpeg4
D
Diego Biurrun 已提交
924
       and we can't do it in the header decode as init is not called for mpeg4 there yet */
925 926 927
    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 已提交
928
    }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
929 930 931 932 933 934
        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;
    }
935

936 937 938 939 940
    if(s->dct_error_sum){
        assert(s->avctx->noise_reduction && s->encoding);

        update_noise_reduction(s);
    }
941

942
#if CONFIG_XVMC
I
Ivan Kalvachev 已提交
943 944 945
    if(s->avctx->xvmc_acceleration)
        return XVMC_field_start(s, avctx);
#endif
946
    return 0;
F
Fabrice Bellard 已提交
947
}
948

F
Fabrice Bellard 已提交
949 950 951
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
M
cleanup  
Michael Niedermayer 已提交
952
    int i;
F
Fabrice Bellard 已提交
953
    /* draw edge for correct motion prediction if outside */
954
#if CONFIG_XVMC
I
Ivan Kalvachev 已提交
955 956 957 958 959
//just to make sure that all data is rendered.
    if(s->avctx->xvmc_acceleration){
        XVMC_field_end(s);
    }else
#endif
960 961 962 963 964
    if(!(s->avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU)
       && s->unrestricted_mv
       && s->current_picture.reference
       && !s->intra_only
       && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
A
Aurelien Jacobs 已提交
965 966 967
            s->dsp.draw_edges(s->current_picture.data[0], s->linesize  , s->h_edge_pos   , s->v_edge_pos   , EDGE_WIDTH  );
            s->dsp.draw_edges(s->current_picture.data[1], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
            s->dsp.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 已提交
968
    }
M
Michael Niedermayer 已提交
969
    emms_c();
970

M
Michael Niedermayer 已提交
971
    s->last_pict_type    = s->pict_type;
972
    s->last_lambda_for[s->pict_type]= s->current_picture_ptr->quality;
973
    if(s->pict_type!=FF_B_TYPE){
974 975
        s->last_non_b_pict_type= s->pict_type;
    }
M
Michael Niedermayer 已提交
976 977
#if 0
        /* copy back current_picture variables */
M
cleanup  
Michael Niedermayer 已提交
978 979 980 981
    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;
982
        }
M
cleanup  
Michael Niedermayer 已提交
983 984
    }
    assert(i<MAX_PICTURE_COUNT);
985
#endif
M
cleanup  
Michael Niedermayer 已提交
986

M
Michael Niedermayer 已提交
987
    if(s->encoding){
D
Diego Biurrun 已提交
988
        /* release non-reference frames */
M
Michael Niedermayer 已提交
989 990 991 992 993
        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 已提交
994
    }
M
Michael Niedermayer 已提交
995 996 997 998 999 1000
    // 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 已提交
1001
    s->avctx->coded_frame= (AVFrame*)s->current_picture_ptr;
F
Fabrice Bellard 已提交
1002 1003
}

1004 1005 1006 1007 1008 1009 1010 1011
/**
 * 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 已提交
1012
    int x, y, fr, f;
1013

1014 1015 1016 1017
    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);
1018

1019
    buf[sy*stride + sx]+= color;
1020

D
Diego Biurrun 已提交
1021
    if(FFABS(ex - sx) > FFABS(ey - sy)){
1022
        if(sx > ex){
D
Diego Biurrun 已提交
1023 1024
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1025 1026 1027 1028 1029
        }
        buf+= sx + sy*stride;
        ex-= sx;
        f= ((ey-sy)<<16)/ex;
        for(x= 0; x <= ex; x++){
1030 1031 1032 1033
            y = (x*f)>>16;
            fr= (x*f)&0xFFFF;
            buf[ y   *stride + x]+= (color*(0x10000-fr))>>16;
            buf[(y+1)*stride + x]+= (color*         fr )>>16;
1034 1035 1036
        }
    }else{
        if(sy > ey){
D
Diego Biurrun 已提交
1037 1038
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1039 1040 1041 1042 1043 1044
        }
        buf+= sx + sy*stride;
        ey-= sy;
        if(ey) f= ((ex-sx)<<16)/ey;
        else   f= 0;
        for(y= 0; y <= ey; y++){
1045 1046
            x = (y*f)>>16;
            fr= (y*f)&0xFFFF;
M
Fix ;;  
Michael Niedermayer 已提交
1047 1048
            buf[y*stride + x  ]+= (color*(0x10000-fr))>>16;
            buf[y*stride + x+1]+= (color*         fr )>>16;
1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059
        }
    }
}

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

1063 1064 1065 1066
    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);
1067

1068 1069
    dx= ex - sx;
    dy= ey - sy;
1070

1071 1072 1073 1074
    if(dx*dx + dy*dy > 3*3){
        int rx=  dx + dy;
        int ry= -dx + dy;
        int length= ff_sqrt((rx*rx + ry*ry)<<8);
1075

1076 1077 1078
        //FIXME subpixel accuracy
        rx= ROUNDED_DIV(rx*3<<4, length);
        ry= ROUNDED_DIV(ry*3<<4, length);
1079

1080 1081 1082 1083 1084 1085
        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);
}

1086 1087 1088
/**
 * prints debuging info for the given picture.
 */
1089
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
1090 1091 1092 1093 1094

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

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

1096 1097 1098 1099 1100 1101 1102
        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;
1103
            case FF_SP_TYPE: av_log(s->avctx,AV_LOG_DEBUG,"SP\n"); break;
1104
        }
1105 1106 1107 1108 1109
        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;
1110
                    av_log(s->avctx, AV_LOG_DEBUG, "%1d", count);
1111 1112
                }
                if(s->avctx->debug&FF_DEBUG_QP){
1113
                    av_log(s->avctx, AV_LOG_DEBUG, "%2d", pict->qscale_table[x + y*s->mb_stride]);
1114 1115 1116 1117 1118
                }
                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))
1119
                        av_log(s->avctx, AV_LOG_DEBUG, "P");
1120
                    else if(IS_INTRA(mb_type) && IS_ACPRED(mb_type))
1121
                        av_log(s->avctx, AV_LOG_DEBUG, "A");
1122
                    else if(IS_INTRA4x4(mb_type))
1123
                        av_log(s->avctx, AV_LOG_DEBUG, "i");
1124
                    else if(IS_INTRA16x16(mb_type))
1125
                        av_log(s->avctx, AV_LOG_DEBUG, "I");
1126
                    else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type))
1127
                        av_log(s->avctx, AV_LOG_DEBUG, "d");
1128
                    else if(IS_DIRECT(mb_type))
1129
                        av_log(s->avctx, AV_LOG_DEBUG, "D");
1130
                    else if(IS_GMC(mb_type) && IS_SKIP(mb_type))
1131
                        av_log(s->avctx, AV_LOG_DEBUG, "g");
1132
                    else if(IS_GMC(mb_type))
1133
                        av_log(s->avctx, AV_LOG_DEBUG, "G");
1134
                    else if(IS_SKIP(mb_type))
1135
                        av_log(s->avctx, AV_LOG_DEBUG, "S");
1136
                    else if(!USES_LIST(mb_type, 1))
1137
                        av_log(s->avctx, AV_LOG_DEBUG, ">");
1138
                    else if(!USES_LIST(mb_type, 0))
1139
                        av_log(s->avctx, AV_LOG_DEBUG, "<");
1140 1141
                    else{
                        assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
1142
                        av_log(s->avctx, AV_LOG_DEBUG, "X");
1143
                    }
1144

1145 1146
                    //segmentation
                    if(IS_8X8(mb_type))
1147
                        av_log(s->avctx, AV_LOG_DEBUG, "+");
1148
                    else if(IS_16X8(mb_type))
1149
                        av_log(s->avctx, AV_LOG_DEBUG, "-");
1150
                    else if(IS_8X16(mb_type))
1151
                        av_log(s->avctx, AV_LOG_DEBUG, "|");
1152
                    else if(IS_INTRA(mb_type) || IS_16X16(mb_type))
1153
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1154
                    else
1155
                        av_log(s->avctx, AV_LOG_DEBUG, "?");
1156 1157


1158
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264)
1159
                        av_log(s->avctx, AV_LOG_DEBUG, "=");
1160
                    else
1161
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1162
                }
1163
//                av_log(s->avctx, AV_LOG_DEBUG, " ");
1164
            }
1165
            av_log(s->avctx, AV_LOG_DEBUG, "\n");
1166 1167
        }
    }
1168

1169
    if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
1170 1171
        const int shift= 1 + s->quarter_sample;
        int mb_y;
1172 1173
        uint8_t *ptr;
        int i;
1174
        int h_chroma_shift, v_chroma_shift, block_height;
1175 1176
        const int width = s->avctx->width;
        const int height= s->avctx->height;
1177
        const int mv_sample_log2= 4 - pict->motion_subsample_log2;
1178
        const int mv_stride= (s->mb_width << mv_sample_log2) + (s->codec_id == CODEC_ID_H264 ? 0 : 1);
M
100000l  
Michael Niedermayer 已提交
1179
        s->low_delay=0; //needed to see the vectors without trashing the buffers
1180

1181
        avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1182
        for(i=0; i<3; i++){
1183
            memcpy(s->visualization_buffer[i], pict->data[i], (i==0) ? pict->linesize[i]*height:pict->linesize[i]*height >> v_chroma_shift);
1184 1185 1186 1187
            pict->data[i]= s->visualization_buffer[i];
        }
        pict->type= FF_BUFFER_TYPE_COPY;
        ptr= pict->data[0];
1188
        block_height = 16>>v_chroma_shift;
1189 1190 1191 1192 1193

        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;
1194 1195 1196
                if((s->avctx->debug_mv) && pict->motion_val){
                  int type;
                  for(type=0; type<3; type++){
1197
                    int direction = 0;
1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211
                    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;
                    }
1212 1213 1214
                    if(!USES_LIST(pict->mb_type[mb_index], direction))
                        continue;

1215 1216 1217
                    if(IS_8X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<4; i++){
1218 1219
                        int sx= mb_x*16 + 4 + 8*(i&1);
                        int sy= mb_y*16 + 4 + 8*(i>>1);
1220
                        int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
1221 1222
                        int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                        int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1223
                        draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1224 1225 1226 1227
                      }
                    }else if(IS_16X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<2; i++){
1228 1229
                        int sx=mb_x*16 + 8;
                        int sy=mb_y*16 + 4 + 8*i;
1230
                        int xy= (mb_x*2 + (mb_y*2 + i)*mv_stride) << (mv_sample_log2-1);
1231 1232
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1233

1234 1235
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1236

1237 1238 1239 1240 1241 1242 1243
                        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;
1244
                        int xy= (mb_x*2 + i + mb_y*2*mv_stride) << (mv_sample_log2-1);
1245 1246
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1247

1248 1249
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1250

1251
                        draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
1252 1253 1254 1255
                      }
                    }else{
                      int sx= mb_x*16 + 8;
                      int sy= mb_y*16 + 8;
1256
                      int xy= (mb_x + mb_y*mv_stride) << mv_sample_log2;
1257 1258
                      int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                      int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1259
                      draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1260
                    }
1261
                  }
M
Michael Niedermayer 已提交
1262 1263 1264 1265
                }
                if((s->avctx->debug&FF_DEBUG_VIS_QP) && pict->motion_val){
                    uint64_t c= (pict->qscale_table[mb_index]*128/31) * 0x0101010101010101ULL;
                    int y;
1266 1267 1268
                    for(y=0; y<block_height; y++){
                        *(uint64_t*)(pict->data[1] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[1])= c;
                        *(uint64_t*)(pict->data[2] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[2])= c;
M
Michael Niedermayer 已提交
1269 1270 1271 1272 1273 1274 1275 1276 1277 1278
                    }
                }
                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));

1279

M
Michael Niedermayer 已提交
1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307
                    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;
1308 1309 1310
                    for(y=0; y<block_height; y++){
                        *(uint64_t*)(pict->data[1] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[1])= u;
                        *(uint64_t*)(pict->data[2] + 8*mb_x + (block_height*mb_y + y)*pict->linesize[2])= v;
M
Michael Niedermayer 已提交
1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321
                    }

                    //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;
                    }
1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336
                    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;
                        }
                    }
1337

M
Michael Niedermayer 已提交
1338 1339 1340
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264){
                        // hmm
                    }
1341 1342 1343 1344 1345
                }
                s->mbskip_table[mb_index]=0;
            }
        }
    }
1346 1347
}

1348
static inline int hpel_motion_lowres(MpegEncContext *s,
M
Michael Niedermayer 已提交
1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
                                  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);
1371

M
Michael Niedermayer 已提交
1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389
    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 已提交
1390
/* apply one mpeg motion vector to the three components */
B
Benoit Fouet 已提交
1391
static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
1392
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1393
                               int field_based, int bottom_field, int field_select,
B
Benoit Fouet 已提交
1394
                               uint8_t **ref_picture, h264_chroma_mc_func *pix_op,
F
Fabrice Bellard 已提交
1395 1396
                               int motion_x, int motion_y, int h)
{
1397
    uint8_t *ptr_y, *ptr_cb, *ptr_cr;
B
Benoit Fouet 已提交
1398 1399 1400 1401 1402 1403
    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;
1404
    linesize   = s->current_picture.linesize[0] << field_based;
M
Michael Niedermayer 已提交
1405
    uvlinesize = s->current_picture.linesize[1] << field_based;
1406

1407
    if(s->quarter_sample){ //FIXME obviously not perfect but qpel will not work in lowres anyway
B
Benoit Fouet 已提交
1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
        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));
1420

M
Michael Niedermayer 已提交
1421
    if (s->out_format == FMT_H263) {
M
Michael Niedermayer 已提交
1422 1423
        uvsx = ((motion_x>>1) & s_mask) | (sx&1);
        uvsy = ((motion_y>>1) & s_mask) | (sy&1);
M
Michael Niedermayer 已提交
1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438
        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));
1439
        uvsrc_y =(s->mb_y*block_s>>field_based) + (my >> (lowres+1));
M
Michael Niedermayer 已提交
1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450
    }

    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;
1451
            if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1452
                uint8_t *uvbuf= s->edge_emu_buffer+18*s->linesize;
1453
                ff_emulated_edge_mc(uvbuf  , ptr_cb, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1454
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
1455
                ff_emulated_edge_mc(uvbuf+16, ptr_cr, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1456 1457 1458 1459 1460 1461
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
                ptr_cb= uvbuf;
                ptr_cr= uvbuf+16;
            }
    }

1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473
    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 已提交
1474 1475 1476
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres-1](dest_y, ptr_y, linesize, h, sx, sy);
1477

1478
    if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1479 1480 1481 1482 1483
        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 已提交
1484
    //FIXME h261 lowres loop filter
M
Michael Niedermayer 已提交
1485 1486
}

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

M
Michael Niedermayer 已提交
1500 1501 1502 1503 1504 1505 1506 1507 1508
    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);
1509

M
Michael Niedermayer 已提交
1510 1511 1512 1513
    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));
1514

M
Michael Niedermayer 已提交
1515 1516 1517 1518 1519 1520 1521 1522 1523
    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;
        }
1524
    }
M
Michael Niedermayer 已提交
1525 1526 1527
    sx <<= 2 - lowres;
    sy <<= 2 - lowres;
    pix_op[lowres](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
1528

M
Michael Niedermayer 已提交
1529 1530 1531 1532 1533 1534 1535 1536
    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 已提交
1537
/**
D
Diego Biurrun 已提交
1538
 * motion compensation of a single macroblock
M
doxy  
Michael Niedermayer 已提交
1539 1540 1541 1542 1543 1544 1545 1546 1547
 * @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 已提交
1548
static inline void MPV_motion_lowres(MpegEncContext *s,
1549
                              uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1550
                              int dir, uint8_t **ref_picture,
B
Benoit Fouet 已提交
1551
                              h264_chroma_mc_func *pix_op)
F
Fabrice Bellard 已提交
1552
{
B
Benoit Fouet 已提交
1553
    int mx, my;
F
Fabrice Bellard 已提交
1554
    int mb_x, mb_y, i;
B
Benoit Fouet 已提交
1555 1556
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
F
Fabrice Bellard 已提交
1557 1558 1559 1560 1561 1562

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

    switch(s->mv_type) {
    case MV_TYPE_16X16:
B
Benoit Fouet 已提交
1563 1564 1565 1566
        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 已提交
1567 1568
        break;
    case MV_TYPE_8X8:
M
Michael Niedermayer 已提交
1569 1570 1571
        mx = 0;
        my = 0;
            for(i=0;i<4;i++) {
B
Benoit Fouet 已提交
1572
                hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) * s->linesize)*block_s,
1573
                            ref_picture[0], 0, 0,
B
Benoit Fouet 已提交
1574
                            (2*mb_x + (i & 1))*block_s, (2*mb_y + (i >>1))*block_s,
M
Michael Niedermayer 已提交
1575
                            s->width, s->height, s->linesize,
B
Benoit Fouet 已提交
1576 1577
                            s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
                            block_s, block_s, pix_op,
M
Michael Niedermayer 已提交
1578
                            s->mv[dir][i][0], s->mv[dir][i][1]);
M
Michael Niedermayer 已提交
1579 1580 1581

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

1584
        if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY))
B
Benoit Fouet 已提交
1585
            chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture, pix_op, mx, my);
F
Fabrice Bellard 已提交
1586 1587 1588
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
B
Benoit Fouet 已提交
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
            /* 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 已提交
1599
        } else {
1600
            if(s->picture_structure != s->field_select[dir][0] + 1 && s->pict_type != FF_B_TYPE && !s->first_field){
1601
                ref_picture= s->current_picture_ptr->data;
1602
            }
F
Fabrice Bellard 已提交
1603

B
Benoit Fouet 已提交
1604
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1605 1606
                        0, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
B
Benoit Fouet 已提交
1607
                        s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s);
F
Fabrice Bellard 已提交
1608
        }
1609
        break;
M
Michael Niedermayer 已提交
1610 1611 1612
    case MV_TYPE_16X8:
        for(i=0; i<2; i++){
            uint8_t ** ref2picture;
1613

1614
            if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == FF_B_TYPE || s->first_field){
1615 1616
                ref2picture= ref_picture;
            }else{
1617
                ref2picture= s->current_picture_ptr->data;
1618
            }
1619

B
Benoit Fouet 已提交
1620
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1621
                        0, 0, s->field_select[dir][i],
1622
                        ref2picture, pix_op,
B
Benoit Fouet 已提交
1623
                        s->mv[dir][i][0], s->mv[dir][i][1] + 2*block_s*i, block_s);
1624

B
Benoit Fouet 已提交
1625 1626 1627
            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;
1628
        }
F
Fabrice Bellard 已提交
1629
        break;
1630 1631
    case MV_TYPE_DMV:
        if(s->picture_structure == PICT_FRAME){
M
Michael Niedermayer 已提交
1632 1633 1634
            for(i=0; i<2; i++){
                int j;
                for(j=0; j<2; j++){
B
Benoit Fouet 已提交
1635
                    mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1636 1637
                                1, j, j^i,
                                ref_picture, pix_op,
B
Benoit Fouet 已提交
1638
                                s->mv[dir][2*i + j][0], s->mv[dir][2*i + j][1], block_s);
M
Michael Niedermayer 已提交
1639
                }
B
Benoit Fouet 已提交
1640
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
M
Michael Niedermayer 已提交
1641
            }
1642
        }else{
M
Michael Niedermayer 已提交
1643
            for(i=0; i<2; i++){
B
Benoit Fouet 已提交
1644
                mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1645 1646
                            0, 0, s->picture_structure != i+1,
                            ref_picture, pix_op,
B
Benoit Fouet 已提交
1647
                            s->mv[dir][2*i][0],s->mv[dir][2*i][1],2*block_s);
1648

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

M
Michael Niedermayer 已提交
1652 1653
                //opposite parity is always in the same frame if this is second field
                if(!s->first_field){
1654
                    ref_picture = s->current_picture_ptr->data;
M
Michael Niedermayer 已提交
1655
                }
1656
            }
1657 1658
        }
    break;
M
Michael Niedermayer 已提交
1659
    default: assert(0);
F
Fabrice Bellard 已提交
1660 1661 1662
    }
}

B
Benoit Fouet 已提交
1663 1664 1665
/* 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 已提交
1666
{
B
Benoit Fouet 已提交
1667 1668 1669
    s->dct_unquantize_intra(s, block, i, qscale);
    s->dsp.idct_put (dest, line_size, block);
}
1670

B
Benoit Fouet 已提交
1671 1672 1673 1674 1675 1676 1677 1678
/* 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);
    }
}
1679

1680
static inline void add_dequant_dct(MpegEncContext *s,
M
Michael Niedermayer 已提交
1681
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
M
Michael Niedermayer 已提交
1682
{
F
Fabrice Bellard 已提交
1683
    if (s->block_last_index[i] >= 0) {
1684
        s->dct_unquantize_inter(s, block, i, qscale);
1685

1686
        s->dsp.idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1687 1688 1689
    }
}

1690 1691 1692 1693 1694
/**
 * cleans dc, ac, coded_block for the current non intra MB
 */
void ff_clean_intra_table_entries(MpegEncContext *s)
{
1695
    int wrap = s->b8_stride;
1696
    int xy = s->block_index[0];
1697 1698 1699

    s->dc_val[0][xy           ] =
    s->dc_val[0][xy + 1       ] =
1700 1701 1702
    s->dc_val[0][xy     + wrap] =
    s->dc_val[0][xy + 1 + wrap] = 1024;
    /* ac pred */
1703 1704
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
1705 1706 1707 1708 1709 1710 1711
    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 */
1712 1713
    wrap = s->mb_stride;
    xy = s->mb_x + s->mb_y * wrap;
1714 1715 1716
    s->dc_val[1][xy] =
    s->dc_val[2][xy] = 1024;
    /* ac pred */
1717 1718
    memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
1719

1720
    s->mbintra_table[xy]= 0;
1721 1722
}

F
Fabrice Bellard 已提交
1723 1724 1725 1726 1727 1728 1729 1730 1731 1732
/* 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)
 */
1733 1734
static av_always_inline
void MPV_decode_mb_internal(MpegEncContext *s, DCTELEM block[12][64],
K
Keiji Costantini 已提交
1735
                            int lowres_flag, int is_mpeg12)
F
Fabrice Bellard 已提交
1736
{
1737
    int mb_x, mb_y;
1738
    const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
1739
#if CONFIG_XVMC
I
Ivan Kalvachev 已提交
1740
    if(s->avctx->xvmc_acceleration){
1741
        XVMC_decode_mb(s);//xvmc uses pblocks
I
Ivan Kalvachev 已提交
1742 1743 1744
        return;
    }
#endif
F
Fabrice Bellard 已提交
1745 1746 1747 1748

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

1749 1750 1751 1752 1753 1754 1755 1756 1757
    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 已提交
1758
    s->current_picture.qscale_table[mb_xy]= s->qscale;
1759

F
Fabrice Bellard 已提交
1760 1761
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
K
Keiji Costantini 已提交
1762
        if (!is_mpeg12 && (s->h263_pred || s->h263_aic)) {
M
Michael Niedermayer 已提交
1763
            if(s->mbintra_table[mb_xy])
1764
                ff_clean_intra_table_entries(s);
F
Fabrice Bellard 已提交
1765
        } else {
1766 1767
            s->last_dc[0] =
            s->last_dc[1] =
F
Fabrice Bellard 已提交
1768 1769 1770
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
K
Keiji Costantini 已提交
1771
    else if (!is_mpeg12 && (s->h263_pred || s->h263_aic))
M
Michael Niedermayer 已提交
1772
        s->mbintra_table[mb_xy]=1;
1773

1774
    if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==FF_B_TYPE) && s->avctx->mb_decision != FF_MB_DECISION_RD)) { //FIXME precalc
1775
        uint8_t *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
1776
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
1777 1778
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
1779
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
M
Michael Niedermayer 已提交
1780
        const int uvlinesize= s->current_picture.linesize[1];
1781
        const int readable= s->pict_type != FF_B_TYPE || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
M
Michael Niedermayer 已提交
1782
        const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
1783

M
cleanup  
Michael Niedermayer 已提交
1784 1785 1786
        /* 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){
1787
            uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
M
cleanup  
Michael Niedermayer 已提交
1788
            const int age= s->current_picture.age;
M
Michael Niedermayer 已提交
1789

M
cleanup  
Michael Niedermayer 已提交
1790 1791
            assert(age);

1792 1793
            if (s->mb_skipped) {
                s->mb_skipped= 0;
1794
                assert(s->pict_type!=FF_I_TYPE);
1795

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

M
cleanup  
Michael Niedermayer 已提交
1799
                /* if previous was skipped too, then nothing to do !  */
1800 1801
                if (*mbskip_ptr >= age && s->current_picture.reference){
                    return;
M
cleanup  
Michael Niedermayer 已提交
1802
                }
1803 1804 1805 1806
            } 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{
1807 1808
                *mbskip_ptr = 0; /* not skipped */
            }
M
Michael Niedermayer 已提交
1809
        }
1810

1811
        dct_linesize = linesize << s->interlaced_dct;
M
Michael Niedermayer 已提交
1812
        dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
1813

1814 1815 1816 1817 1818
        if(readable){
            dest_y=  s->dest[0];
            dest_cb= s->dest[1];
            dest_cr= s->dest[2];
        }else{
1819
            dest_y = s->b_scratchpad;
1820
            dest_cb= s->b_scratchpad+16*linesize;
1821
            dest_cr= s->b_scratchpad+32*linesize;
1822
        }
M
Michael Niedermayer 已提交
1823

F
Fabrice Bellard 已提交
1824 1825
        if (!s->mb_intra) {
            /* motion handling */
1826
            /* decoding or more than one mb_type (MC was already done otherwise) */
1827
            if(!s->encoding){
M
Michael Niedermayer 已提交
1828 1829
                if(lowres_flag){
                    h264_chroma_mc_func *op_pix = s->dsp.put_h264_chroma_pixels_tab;
F
Fabrice Bellard 已提交
1830

M
Michael Niedermayer 已提交
1831 1832 1833 1834 1835 1836 1837 1838
                    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{
1839
                    op_qpix= s->me.qpel_put;
1840
                    if ((!s->no_rounding) || s->pict_type==FF_B_TYPE){
M
Michael Niedermayer 已提交
1841 1842 1843 1844 1845 1846 1847
                        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;
1848
                        op_qpix= s->me.qpel_avg;
M
Michael Niedermayer 已提交
1849 1850 1851 1852
                    }
                    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);
                    }
1853
                }
F
Fabrice Bellard 已提交
1854 1855
            }

M
Michael Niedermayer 已提交
1856
            /* skip dequant / idct if we are really late ;) */
M
Michael Niedermayer 已提交
1857 1858
            if(s->hurry_up>1) goto skip_idct;
            if(s->avctx->skip_idct){
1859 1860
                if(  (s->avctx->skip_idct >= AVDISCARD_NONREF && s->pict_type == FF_B_TYPE)
                   ||(s->avctx->skip_idct >= AVDISCARD_NONKEY && s->pict_type != FF_I_TYPE)
M
Michael Niedermayer 已提交
1861 1862 1863
                   || s->avctx->skip_idct >= AVDISCARD_ALL)
                    goto skip_idct;
            }
M
Michael Niedermayer 已提交
1864

F
Fabrice Bellard 已提交
1865
            /* add dct residue */
M
Michael Niedermayer 已提交
1866
            if(s->encoding || !(   s->h263_msmpeg4 || s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO
1867
                                || (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant))){
M
Michael Niedermayer 已提交
1868 1869 1870 1871
                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 已提交
1872

1873
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884
                    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 已提交
1885
                }
K
Keiji Costantini 已提交
1886
            } else if(is_mpeg12 || (s->codec_id != CODEC_ID_WMV2)){
M
Michael Niedermayer 已提交
1887 1888 1889 1890
                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 已提交
1891

1892
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
                    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
            }
1914
            else if (CONFIG_WMV2) {
M
Michael Niedermayer 已提交
1915
                ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
M
Michael Niedermayer 已提交
1916
            }
F
Fabrice Bellard 已提交
1917 1918
        } else {
            /* dct only in intra block */
M
Michael Niedermayer 已提交
1919
            if(s->encoding || !(s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO)){
M
Michael Niedermayer 已提交
1920 1921 1922 1923
                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);
1924

1925
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1926 1927 1928 1929 1930 1931 1932 1933 1934
                    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 已提交
1935
                        put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
M
Michael Niedermayer 已提交
1936 1937 1938
                    }
                }
            }else{
B
Benoit Fouet 已提交
1939 1940 1941 1942
                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]);
1943

1944
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
B
Benoit Fouet 已提交
1945 1946 1947 1948
                    if(s->chroma_y_shift){
                        s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                        s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
                    }else{
1949

B
Benoit Fouet 已提交
1950 1951
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
1952

B
Benoit Fouet 已提交
1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964
                        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
1965
            }
M
Michael Niedermayer 已提交
1966
        }
B
Benoit Fouet 已提交
1967 1968 1969 1970 1971
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);
1972
        }
M
Michael Niedermayer 已提交
1973 1974 1975
    }
}

B
Benoit Fouet 已提交
1976
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]){
1977
#if !CONFIG_SMALL
K
Keiji Costantini 已提交
1978 1979 1980 1981 1982 1983 1984
    if(s->out_format == FMT_MPEG1) {
        if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 1);
        else                 MPV_decode_mb_internal(s, block, 0, 1);
    } else
#endif
    if(s->avctx->lowres) MPV_decode_mb_internal(s, block, 1, 0);
    else                  MPV_decode_mb_internal(s, block, 0, 0);
M
Michael Niedermayer 已提交
1985 1986
}

B
Benoit Fouet 已提交
1987 1988 1989 1990 1991 1992 1993 1994
/**
 *
 * @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 已提交
1995

B
Benoit Fouet 已提交
1996 1997 1998 1999
        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 已提交
2000 2001
        }

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

2004
        if(s->pict_type==FF_B_TYPE || s->low_delay || (s->avctx->slice_flags&SLICE_FLAG_CODED_ORDER))
B
Benoit Fouet 已提交
2005 2006 2007 2008 2009
            src= (AVFrame*)s->current_picture_ptr;
        else if(s->last_picture_ptr)
            src= (AVFrame*)s->last_picture_ptr;
        else
            return;
2010

2011
        if(s->pict_type==FF_B_TYPE && s->picture_structure == PICT_FRAME && s->out_format != FMT_H264){
B
Benoit Fouet 已提交
2012 2013 2014 2015
            offset[0]=
            offset[1]=
            offset[2]=
            offset[3]= 0;
M
Michael Niedermayer 已提交
2016
        }else{
M
Fix ;;  
Michael Niedermayer 已提交
2017
            offset[0]= y * s->linesize;
B
Benoit Fouet 已提交
2018 2019 2020
            offset[1]=
            offset[2]= (y >> s->chroma_y_shift) * s->uvlinesize;
            offset[3]= 0;
M
Michael Niedermayer 已提交
2021
        }
2022

B
Benoit Fouet 已提交
2023
        emms_c();
M
Michael Niedermayer 已提交
2024

B
Benoit Fouet 已提交
2025 2026 2027 2028
        s->avctx->draw_horiz_band(s->avctx, src, offset,
                                  y, s->picture_structure, h);
    }
}
2029

B
Benoit Fouet 已提交
2030 2031 2032 2033
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 已提交
2034

B
Benoit Fouet 已提交
2035 2036 2037 2038 2039 2040 2041
    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
2042

B
Benoit Fouet 已提交
2043 2044 2045
    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));
2046

2047
    if(!(s->pict_type==FF_B_TYPE && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME))
B
Benoit Fouet 已提交
2048 2049 2050 2051
    {
        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 已提交
2052 2053
    }
}
2054

B
Benoit Fouet 已提交
2055 2056 2057
void ff_mpeg_flush(AVCodecContext *avctx){
    int i;
    MpegEncContext *s = avctx->priv_data;
M
Michael Niedermayer 已提交
2058

B
Benoit Fouet 已提交
2059 2060
    if(s==NULL || s->picture==NULL)
        return;
M
Michael Niedermayer 已提交
2061

B
Benoit Fouet 已提交
2062 2063 2064 2065
    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 已提交
2066
    }
B
Benoit Fouet 已提交
2067
    s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
2068

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

B
Benoit Fouet 已提交
2071 2072 2073 2074 2075 2076 2077 2078
    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 已提交
2079 2080
}

2081
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
2082
                                   DCTELEM *block, int n, int qscale)
F
Fabrice Bellard 已提交
2083
{
M
Michael Niedermayer 已提交
2084
    int i, level, nCoeffs;
2085
    const uint16_t *quant_matrix;
F
Fabrice Bellard 已提交
2086

2087
    nCoeffs= s->block_last_index[n];
2088 2089

    if (n < 4)
2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106
        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 已提交
2107
            }
2108
            block[j] = level;
F
Fabrice Bellard 已提交
2109
        }
2110 2111 2112
    }
}

2113
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
2114 2115 2116 2117 2118 2119
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;

    nCoeffs= s->block_last_index[n];
2120

2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135
    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 已提交
2136
            }
2137
            block[j] = level;
F
Fabrice Bellard 已提交
2138 2139 2140
        }
    }
}
2141

2142
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
2143 2144 2145
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
2146
    const uint16_t *quant_matrix;
2147

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

    if (n < 4)
2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171
        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;
        }
    }
}

2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204
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;
}

2205
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
2206 2207 2208 2209 2210 2211 2212 2213
                                   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];
2214

2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235
    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;
}

2236
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
2237 2238 2239 2240
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
    int nCoeffs;
2241

2242
    assert(s->block_last_index[n]>=0);
2243

2244
    qmul = qscale << 1;
2245

2246
    if (!s->h263_aic) {
2247
        if (n < 4)
2248 2249 2250
            block[0] = block[0] * s->y_dc_scale;
        else
            block[0] = block[0] * s->c_dc_scale;
2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
        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;
2267
            }
2268
            block[i] = level;
2269 2270 2271 2272
        }
    }
}

2273
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
2274 2275 2276
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
2277
    int nCoeffs;
2278

2279
    assert(s->block_last_index[n]>=0);
2280

2281 2282
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
2283

2284
    nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
2285

2286
    for(i=0; i<=nCoeffs; i++) {
2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
            }
            block[i] = level;
        }
    }
}
F
Fabrice Bellard 已提交
2298

2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314
/**
 * 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 ];
}