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

I
Ivan Kalvachev 已提交
57 58
extern int  XVMC_field_start(MpegEncContext*s, AVCodecContext *avctx);
extern void XVMC_field_end(MpegEncContext *s);
59
extern 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 defined(HAVE_MMX)
125
    MPV_common_init_mmx(s);
126
#elif defined(ARCH_ALPHA)
127
    MPV_common_init_axp(s);
128
#elif defined(CONFIG_MLIB)
129
    MPV_common_init_mlib(s);
130
#elif defined(HAVE_MMI)
131
    MPV_common_init_mmi(s);
132
#elif defined(ARCH_ARMV4L)
133
    MPV_common_init_armv4l(s);
134
#elif defined(HAVE_ALTIVEC)
135
    MPV_common_init_altivec(s);
136
#elif defined(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

I
Ivan Kalvachev 已提交
942 943 944 945
#ifdef HAVE_XVMC
    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 */
I
Ivan Kalvachev 已提交
954 955 956 957 958 959
#ifdef HAVE_XVMC
//just to make sure that all data is rendered.
    if(s->avctx->xvmc_acceleration){
        XVMC_field_end(s);
    }else
#endif
960
    if(s->unrestricted_mv && s->current_picture.reference && !s->intra_only && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
A
Aurelien Jacobs 已提交
961 962 963
            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 已提交
964
    }
M
Michael Niedermayer 已提交
965
    emms_c();
966

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

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

1000 1001 1002 1003 1004 1005 1006 1007
/**
 * 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 已提交
1008
    int x, y, fr, f;
1009

1010 1011 1012 1013
    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);
1014

1015
    buf[sy*stride + sx]+= color;
1016

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

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

1059 1060 1061 1062
    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);
1063

1064 1065
    dx= ex - sx;
    dy= ey - sy;
1066

1067 1068 1069 1070
    if(dx*dx + dy*dy > 3*3){
        int rx=  dx + dy;
        int ry= -dx + dy;
        int length= ff_sqrt((rx*rx + ry*ry)<<8);
1071

1072 1073 1074
        //FIXME subpixel accuracy
        rx= ROUNDED_DIV(rx*3<<4, length);
        ry= ROUNDED_DIV(ry*3<<4, length);
1075

1076 1077 1078 1079 1080 1081
        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);
}

1082 1083 1084
/**
 * prints debuging info for the given picture.
 */
1085
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
1086 1087 1088 1089 1090

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

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

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

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


1154
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264)
1155
                        av_log(s->avctx, AV_LOG_DEBUG, "=");
1156
                    else
1157
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1158
                }
1159
//                av_log(s->avctx, AV_LOG_DEBUG, " ");
1160
            }
1161
            av_log(s->avctx, AV_LOG_DEBUG, "\n");
1162 1163
        }
    }
1164

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

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

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

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

1230 1231
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1232

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

1244 1245
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1246

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

1275

M
Michael Niedermayer 已提交
1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303
                    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;
1304 1305 1306
                    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 已提交
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
                    }

                    //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;
                    }
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
                    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;
                        }
                    }
1333

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

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

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

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

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

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

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

1474
    if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1475 1476 1477 1478 1479
        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 已提交
1480
    //FIXME h261 lowres loop filter
M
Michael Niedermayer 已提交
1481 1482
}

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

M
Michael Niedermayer 已提交
1496 1497 1498 1499 1500 1501 1502 1503 1504
    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);
1505

M
Michael Niedermayer 已提交
1506 1507 1508 1509
    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));
1510

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

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

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

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

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

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

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

1610
            if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == FF_B_TYPE || s->first_field){
1611 1612
                ref2picture= ref_picture;
            }else{
1613
                ref2picture= s->current_picture_ptr->data;
1614
            }
1615

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

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

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

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

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

B
Benoit Fouet 已提交
1667 1668 1669 1670 1671 1672 1673 1674
/* 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);
    }
}
1675

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

1682
        s->dsp.idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1683 1684 1685
    }
}

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

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

1716
    s->mbintra_table[xy]= 0;
1717 1718
}

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

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

1745 1746 1747 1748 1749 1750 1751 1752 1753
    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 已提交
1754
    s->current_picture.qscale_table[mb_xy]= s->qscale;
1755

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

1770
    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
1771
        uint8_t *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
1772
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
1773 1774
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
1775
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
M
Michael Niedermayer 已提交
1776
        const int uvlinesize= s->current_picture.linesize[1];
1777
        const int readable= s->pict_type != FF_B_TYPE || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
M
Michael Niedermayer 已提交
1778
        const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
1779

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

M
cleanup  
Michael Niedermayer 已提交
1786 1787
            assert(age);

1788 1789
            if (s->mb_skipped) {
                s->mb_skipped= 0;
1790
                assert(s->pict_type!=FF_I_TYPE);
1791

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

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

1807
        dct_linesize = linesize << s->interlaced_dct;
M
Michael Niedermayer 已提交
1808
        dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
1809

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

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

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

M
Michael Niedermayer 已提交
1852
            /* skip dequant / idct if we are really late ;) */
M
Michael Niedermayer 已提交
1853 1854
            if(s->hurry_up>1) goto skip_idct;
            if(s->avctx->skip_idct){
1855 1856
                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 已提交
1857 1858 1859
                   || s->avctx->skip_idct >= AVDISCARD_ALL)
                    goto skip_idct;
            }
M
Michael Niedermayer 已提交
1860

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

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

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

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

1940
                if(!ENABLE_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
B
Benoit Fouet 已提交
1941 1942 1943 1944
                    if(s->chroma_y_shift){
                        s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                        s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
                    }else{
1945

B
Benoit Fouet 已提交
1946 1947
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
1948

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

B
Benoit Fouet 已提交
1972
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]){
K
Keiji Costantini 已提交
1973 1974 1975 1976 1977 1978 1979 1980
#ifndef CONFIG_SMALL
    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 已提交
1981 1982
}

B
Benoit Fouet 已提交
1983 1984 1985 1986 1987 1988 1989 1990
/**
 *
 * @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 已提交
1991

B
Benoit Fouet 已提交
1992 1993 1994 1995
        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 已提交
1996 1997
        }

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

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

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

B
Benoit Fouet 已提交
2019
        emms_c();
M
Michael Niedermayer 已提交
2020

B
Benoit Fouet 已提交
2021 2022 2023 2024
        s->avctx->draw_horiz_band(s->avctx, src, offset,
                                  y, s->picture_structure, h);
    }
}
2025

B
Benoit Fouet 已提交
2026 2027 2028 2029
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 已提交
2030

B
Benoit Fouet 已提交
2031 2032 2033 2034 2035 2036 2037
    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
2038

B
Benoit Fouet 已提交
2039 2040 2041
    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));
2042

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

B
Benoit Fouet 已提交
2051 2052 2053
void ff_mpeg_flush(AVCodecContext *avctx){
    int i;
    MpegEncContext *s = avctx->priv_data;
M
Michael Niedermayer 已提交
2054

B
Benoit Fouet 已提交
2055 2056
    if(s==NULL || s->picture==NULL)
        return;
M
Michael Niedermayer 已提交
2057

B
Benoit Fouet 已提交
2058 2059 2060 2061
    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 已提交
2062
    }
B
Benoit Fouet 已提交
2063
    s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
2064

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

B
Benoit Fouet 已提交
2067 2068 2069 2070 2071 2072 2073 2074
    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 已提交
2075 2076
}

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

2083
    nCoeffs= s->block_last_index[n];
2084 2085

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

2109
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
2110 2111 2112 2113 2114 2115
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;

    nCoeffs= s->block_last_index[n];
2116

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

2138
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
2139 2140 2141
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
2142
    const uint16_t *quant_matrix;
2143

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

    if (n < 4)
2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167
        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;
        }
    }
}

2168 2169 2170 2171 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
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;
}

2201
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
2202 2203 2204 2205 2206 2207 2208 2209
                                   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];
2210

2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231
    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;
}

2232
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
2233 2234 2235 2236
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
    int nCoeffs;
2237

2238
    assert(s->block_last_index[n]>=0);
2239

2240
    qmul = qscale << 1;
2241

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

2269
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
2270 2271 2272
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
2273
    int nCoeffs;
2274

2275
    assert(s->block_last_index[n]>=0);
2276

2277 2278
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
2279

2280
    nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
2281

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

2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310
/**
 * 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 ];
}