mpegvideo.c 101.4 KB
Newer Older
F
Fabrice Bellard 已提交
1 2
/*
 * The simplest mpeg encoder (well, it was the simplest!)
3
 * Copyright (c) 2000,2001 Fabrice Bellard
4
 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
F
Fabrice Bellard 已提交
5
 *
6 7
 * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
 *
8
 * This file is part of Libav.
9
 *
10
 * Libav 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
 * Libav 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 Libav; 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
 * @file
M
Michael Niedermayer 已提交
27
 * The simplest mpeg encoder (well, it was the simplest!).
28 29
 */

30
#include "libavutil/intmath.h"
31
#include "libavutil/imgutils.h"
F
Fabrice Bellard 已提交
32 33
#include "avcodec.h"
#include "dsputil.h"
34
#include "internal.h"
F
Fabrice Bellard 已提交
35
#include "mpegvideo.h"
B
Benoit Fouet 已提交
36
#include "mpegvideo_common.h"
37
#include "mjpegenc.h"
38
#include "msmpeg4.h"
M
Michael Niedermayer 已提交
39
#include "faandct.h"
40
#include "xvmc_internal.h"
41
#include "thread.h"
42
#include <limits.h>
F
Fabrice Bellard 已提交
43

44 45
//#undef NDEBUG
//#include <assert.h>
46

47
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
48
                                   DCTELEM *block, int n, int qscale);
49
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
50
                                   DCTELEM *block, int n, int qscale);
51 52
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
53 54
static void dct_unquantize_mpeg2_intra_bitexact(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
55 56
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
57
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
58
                                  DCTELEM *block, int n, int qscale);
59
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
60
                                  DCTELEM *block, int n, int qscale);
M
Michael Niedermayer 已提交
61

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

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

//#define DEBUG

68

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

74 75 76 77 78 79 80 81
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,
};

82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
static const uint8_t mpeg2_dc_scale_table1[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
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
    4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
};

static const uint8_t mpeg2_dc_scale_table2[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
    2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
    2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
    2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
    2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
};

static const uint8_t mpeg2_dc_scale_table3[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
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
};

106
const uint8_t * const ff_mpeg2_dc_scale_table[4]={
107 108 109 110 111 112
    ff_mpeg1_dc_scale_table,
    mpeg2_dc_scale_table1,
    mpeg2_dc_scale_table2,
    mpeg2_dc_scale_table3,
};

113
const enum PixelFormat ff_pixfmt_list_420[] = {
114 115 116 117 118
    PIX_FMT_YUV420P,
    PIX_FMT_NONE
};

const enum PixelFormat ff_hwaccel_pixfmt_list_420[] = {
L
Laurent Aimar 已提交
119
    PIX_FMT_DXVA2_VLD,
120
    PIX_FMT_VAAPI_VLD,
121 122 123
    PIX_FMT_YUV420P,
    PIX_FMT_NONE
};
124

M
Michael Niedermayer 已提交
125 126 127
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 已提交
128 129 130 131
    assert(p<=end);
    if(p>=end)
        return end;

M
Michael Niedermayer 已提交
132 133 134 135 136 137 138 139
    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 已提交
140 141 142
        if     (p[-1] > 1      ) p+= 3;
        else if(p[-2]          ) p+= 2;
        else if(p[-3]|(p[-1]-1)) p++;
M
Michael Niedermayer 已提交
143 144 145 146 147 148
        else{
            p++;
            break;
        }
    }

M
Michael Niedermayer 已提交
149
    p= FFMIN(p, end)-4;
150
    *state= AV_RB32(p);
M
Michael Niedermayer 已提交
151 152 153 154

    return p+4;
}

155
/* init common dct for both encoder and decoder */
156
av_cold int ff_dct_common_init(MpegEncContext *s)
F
Fabrice Bellard 已提交
157
{
158 159 160 161 162
    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;
163 164
    if(s->flags & CODEC_FLAG_BITEXACT)
        s->dct_unquantize_mpeg2_intra = dct_unquantize_mpeg2_intra_bitexact;
165
    s->dct_unquantize_mpeg2_inter = dct_unquantize_mpeg2_inter_c;
166

167
#if   HAVE_MMX
168
    MPV_common_init_mmx(s);
169
#elif ARCH_ALPHA
170
    MPV_common_init_axp(s);
171
#elif CONFIG_MLIB
172
    MPV_common_init_mlib(s);
173
#elif HAVE_MMI
174
    MPV_common_init_mmi(s);
175
#elif ARCH_ARM
M
Måns Rullgård 已提交
176
    MPV_common_init_arm(s);
177
#elif HAVE_ALTIVEC
178
    MPV_common_init_altivec(s);
179
#elif ARCH_BFIN
180 181
    MPV_common_init_bfin(s);
#endif
182

183
    /* load & permutate scantables
184
       note: only wmv uses different ones
185
    */
M
Michael Niedermayer 已提交
186 187 188 189 190 191 192
    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 已提交
193 194
    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);
195

196 197 198
    return 0;
}

199
void ff_copy_picture(Picture *dst, Picture *src){
200 201 202 203
    *dst = *src;
    dst->type= FF_BUFFER_TYPE_COPY;
}

204
/**
205
 * Release a frame buffer
206 207 208
 */
static void free_frame_buffer(MpegEncContext *s, Picture *pic)
{
209
    ff_thread_release_buffer(s->avctx, (AVFrame*)pic);
210
    av_freep(&pic->hwaccel_picture_private);
211 212 213
}

/**
214
 * Allocate a frame buffer
215 216 217 218 219
 */
static int alloc_frame_buffer(MpegEncContext *s, Picture *pic)
{
    int r;

220
    if (s->avctx->hwaccel) {
221
        assert(!pic->hwaccel_picture_private);
222
        if (s->avctx->hwaccel->priv_data_size) {
223 224
            pic->hwaccel_picture_private = av_mallocz(s->avctx->hwaccel->priv_data_size);
            if (!pic->hwaccel_picture_private) {
225 226 227 228 229 230
                av_log(s->avctx, AV_LOG_ERROR, "alloc_frame_buffer() failed (hwaccel private data allocation)\n");
                return -1;
            }
        }
    }

231
    r = ff_thread_get_buffer(s->avctx, (AVFrame*)pic);
232 233 234

    if (r<0 || !pic->age || !pic->type || !pic->data[0]) {
        av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (%d %d %d %p)\n", r, pic->age, pic->type, pic->data[0]);
235
        av_freep(&pic->hwaccel_picture_private);
236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253
        return -1;
    }

    if (s->linesize && (s->linesize != pic->linesize[0] || s->uvlinesize != pic->linesize[1])) {
        av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (stride changed)\n");
        free_frame_buffer(s, pic);
        return -1;
    }

    if (pic->linesize[1] != pic->linesize[2]) {
        av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed (uv stride mismatch)\n");
        free_frame_buffer(s, pic);
        return -1;
    }

    return 0;
}

M
cleanup  
Michael Niedermayer 已提交
254
/**
M
Michael Niedermayer 已提交
255 256
 * allocates a Picture
 * The pixels are allocated/set by calling get_buffer() if shared=0
M
cleanup  
Michael Niedermayer 已提交
257
 */
258
int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared){
D
Diego Biurrun 已提交
259
    const int big_mb_num= s->mb_stride*(s->mb_height+1) + 1; //the +1 is needed so memset(,,stride*height) does not sig11
260
    const int mb_array_size= s->mb_stride*s->mb_height;
261 262
    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 已提交
263
    int i;
D
Daniel Kristjansson 已提交
264
    int r= -1;
265

M
Michael Niedermayer 已提交
266 267 268 269 270 271
    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]);
272

273
        if (alloc_frame_buffer(s, pic) < 0)
M
Michael Niedermayer 已提交
274 275 276 277
            return -1;

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

M
Michael Niedermayer 已提交
280
    if(pic->qscale_table==NULL){
281
        if (s->encoding) {
282 283 284
            FF_ALLOCZ_OR_GOTO(s->avctx, pic->mb_var   , mb_array_size * sizeof(int16_t)  , fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, pic->mc_mb_var, mb_array_size * sizeof(int16_t)  , fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, pic->mb_mean  , mb_array_size * sizeof(int8_t )  , fail)
M
Michael Niedermayer 已提交
285
        }
M
cleanup  
Michael Niedermayer 已提交
286

287
        FF_ALLOCZ_OR_GOTO(s->avctx, pic->mbskip_table , mb_array_size * sizeof(uint8_t)+2, fail) //the +2 is for the slice end check
288
        FF_ALLOCZ_OR_GOTO(s->avctx, pic->qscale_table_base , (big_mb_num + s->mb_stride) * sizeof(uint8_t)  , fail)
289
        FF_ALLOCZ_OR_GOTO(s->avctx, pic->mb_type_base , (big_mb_num + s->mb_stride) * sizeof(uint32_t), fail)
290
        pic->mb_type= pic->mb_type_base + 2*s->mb_stride+1;
291
        pic->qscale_table = pic->qscale_table_base + 2*s->mb_stride + 1;
M
Michael Niedermayer 已提交
292 293
        if(s->out_format == FMT_H264){
            for(i=0; i<2; i++){
294
                FF_ALLOCZ_OR_GOTO(s->avctx, pic->motion_val_base[i], 2 * (b4_array_size+4)  * sizeof(int16_t), fail)
M
Michael Niedermayer 已提交
295
                pic->motion_val[i]= pic->motion_val_base[i]+4;
296
                FF_ALLOCZ_OR_GOTO(s->avctx, pic->ref_index[i], 4*mb_array_size * sizeof(uint8_t), fail)
297
            }
M
Michael Niedermayer 已提交
298
            pic->motion_subsample_log2= 2;
299
        }else if(s->out_format == FMT_H263 || s->encoding || (s->avctx->debug&FF_DEBUG_MV) || (s->avctx->debug_mv)){
300
            for(i=0; i<2; i++){
301
                FF_ALLOCZ_OR_GOTO(s->avctx, pic->motion_val_base[i], 2 * (b8_array_size+4) * sizeof(int16_t), fail)
M
Michael Niedermayer 已提交
302
                pic->motion_val[i]= pic->motion_val_base[i]+4;
303
                FF_ALLOCZ_OR_GOTO(s->avctx, pic->ref_index[i], 4*mb_array_size * sizeof(uint8_t), fail)
M
Michael Niedermayer 已提交
304
            }
M
Michael Niedermayer 已提交
305
            pic->motion_subsample_log2= 3;
M
Michael Niedermayer 已提交
306
        }
307
        if(s->avctx->debug&FF_DEBUG_DCT_COEFF) {
308
            FF_ALLOCZ_OR_GOTO(s->avctx, pic->dct_coeff, 64 * mb_array_size * sizeof(DCTELEM)*6, fail)
309
        }
310
        pic->qstride= s->mb_stride;
311
        FF_ALLOCZ_OR_GOTO(s->avctx, pic->pan_scan , 1 * sizeof(AVPanScan), fail)
M
Michael Niedermayer 已提交
312
    }
M
Michael Niedermayer 已提交
313

D
Diego Biurrun 已提交
314 315
    /* It might be nicer if the application would keep track of these
     * but it would require an API change. */
316
    memmove(s->prev_pict_types+1, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE-1);
317 318
    s->prev_pict_types[0]= s->dropable ? AV_PICTURE_TYPE_B : s->pict_type;
    if(pic->age < PREV_PICT_TYPES_BUFFER_SIZE && s->prev_pict_types[pic->age] == AV_PICTURE_TYPE_B)
D
Diego Biurrun 已提交
319
        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.
320
    pic->owner2 = NULL;
321

M
cleanup  
Michael Niedermayer 已提交
322
    return 0;
323
fail: //for the FF_ALLOCZ_OR_GOTO macro
D
Daniel Kristjansson 已提交
324
    if(r>=0)
325
        free_frame_buffer(s, pic);
M
cleanup  
Michael Niedermayer 已提交
326 327 328
    return -1;
}

M
Michael Niedermayer 已提交
329 330 331
/**
 * deallocates a picture
 */
M
cleanup  
Michael Niedermayer 已提交
332 333
static void free_picture(MpegEncContext *s, Picture *pic){
    int i;
M
Michael Niedermayer 已提交
334 335

    if(pic->data[0] && pic->type!=FF_BUFFER_TYPE_SHARED){
336
        free_frame_buffer(s, pic);
M
Michael Niedermayer 已提交
337 338
    }

M
cleanup  
Michael Niedermayer 已提交
339 340 341 342
    av_freep(&pic->mb_var);
    av_freep(&pic->mc_mb_var);
    av_freep(&pic->mb_mean);
    av_freep(&pic->mbskip_table);
343
    av_freep(&pic->qscale_table_base);
M
Michael Niedermayer 已提交
344
    av_freep(&pic->mb_type_base);
345
    av_freep(&pic->dct_coeff);
346
    av_freep(&pic->pan_scan);
M
Michael Niedermayer 已提交
347 348
    pic->mb_type= NULL;
    for(i=0; i<2; i++){
349
        av_freep(&pic->motion_val_base[i]);
M
Michael Niedermayer 已提交
350 351
        av_freep(&pic->ref_index[i]);
    }
352

353
    if(pic->type == FF_BUFFER_TYPE_SHARED){
M
Michael Niedermayer 已提交
354 355 356 357
        for(i=0; i<4; i++){
            pic->base[i]=
            pic->data[i]= NULL;
        }
358
        pic->type= 0;
M
cleanup  
Michael Niedermayer 已提交
359 360 361
    }
}

362
static int init_duplicate_context(MpegEncContext *s, MpegEncContext *base){
363 364 365
    int y_size = s->b8_stride * (2 * s->mb_height + 1);
    int c_size = s->mb_stride * (s->mb_height + 1);
    int yc_size = y_size + 2 * c_size;
366 367
    int i;

368
    // edge emu needs blocksize + filter length - 1 (=17x17 for halfpel / 21x21 for h264)
369
    FF_ALLOCZ_OR_GOTO(s->avctx, s->allocated_edge_emu_buffer, (s->width+64)*2*21*2, fail); //(width + edge + align)*interlaced*MBsize*tolerance
L
Loren Merritt 已提交
370
    s->edge_emu_buffer= s->allocated_edge_emu_buffer + (s->width+64)*2*21;
371

D
Diego Biurrun 已提交
372
     //FIXME should be linesize instead of s->width*2 but that is not known before get_buffer()
373
    FF_ALLOCZ_OR_GOTO(s->avctx, s->me.scratchpad,  (s->width+64)*4*16*2*sizeof(uint8_t), fail)
374
    s->me.temp=         s->me.scratchpad;
375 376 377 378
    s->rd_scratchpad=   s->me.scratchpad;
    s->b_scratchpad=    s->me.scratchpad;
    s->obmc_scratchpad= s->me.scratchpad + 16;
    if (s->encoding) {
379 380
        FF_ALLOCZ_OR_GOTO(s->avctx, s->me.map      , ME_MAP_SIZE*sizeof(uint32_t), fail)
        FF_ALLOCZ_OR_GOTO(s->avctx, s->me.score_map, ME_MAP_SIZE*sizeof(uint32_t), fail)
381
        if(s->avctx->noise_reduction){
382
            FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_error_sum, 2 * 64 * sizeof(int), fail)
383
        }
384
    }
385
    FF_ALLOCZ_OR_GOTO(s->avctx, s->blocks, 64*12*2 * sizeof(DCTELEM), fail)
386 387 388
    s->block= s->blocks[0];

    for(i=0;i<12;i++){
389
        s->pblocks[i] = &s->block[i];
390
    }
391

392 393
    if (s->out_format == FMT_H263) {
        /* ac values */
394 395 396 397 398 399
        FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_val_base, yc_size * sizeof(int16_t) * 16, fail);
        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;
        s->ac_val[2] = s->ac_val[1] + c_size;
    }

400 401 402 403 404 405 406 407 408 409
    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);
410
    s->me.temp=
411 412
    s->rd_scratchpad=
    s->b_scratchpad=
413
    s->obmc_scratchpad= NULL;
414

415 416 417 418
    av_freep(&s->dct_error_sum);
    av_freep(&s->me.map);
    av_freep(&s->me.score_map);
    av_freep(&s->blocks);
419
    av_freep(&s->ac_val_base);
420 421 422 423 424 425 426 427
    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);
428
    COPY(me.temp);
429 430 431 432 433 434 435 436 437 438 439 440
    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);
441 442
    COPY(dct_count[0]);
    COPY(dct_count[1]);
443 444 445 446
    COPY(ac_val_base);
    COPY(ac_val[0]);
    COPY(ac_val[1]);
    COPY(ac_val[2]);
447 448 449
#undef COPY
}

450
void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src){
451
    MpegEncContext bak;
452
    int i;
453 454 455 456 457
    //FIXME copy only needed parts
//START_TIMER
    backup_duplicate_context(&bak, dst);
    memcpy(dst, src, sizeof(MpegEncContext));
    backup_duplicate_context(dst, &bak);
458
    for(i=0;i<12;i++){
459
        dst->pblocks[i] = &dst->block[i];
460
    }
461 462 463
//STOP_TIMER("update_duplicate_context") //about 10k cycles / 0.01 sec for 1000frames on 1ghz with 2 threads
}

464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538
int ff_mpeg_update_thread_context(AVCodecContext *dst, const AVCodecContext *src)
{
    MpegEncContext *s = dst->priv_data, *s1 = src->priv_data;

    if(dst == src || !s1->context_initialized) return 0;

    //FIXME can parameters change on I-frames? in that case dst may need a reinit
    if(!s->context_initialized){
        memcpy(s, s1, sizeof(MpegEncContext));

        s->avctx                 = dst;
        s->picture_range_start  += MAX_PICTURE_COUNT;
        s->picture_range_end    += MAX_PICTURE_COUNT;
        s->bitstream_buffer      = NULL;
        s->bitstream_buffer_size = s->allocated_bitstream_buffer_size = 0;

        MPV_common_init(s);
    }

    s->avctx->coded_height  = s1->avctx->coded_height;
    s->avctx->coded_width   = s1->avctx->coded_width;
    s->avctx->width         = s1->avctx->width;
    s->avctx->height        = s1->avctx->height;

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

    memcpy(s->picture, s1->picture, s1->picture_count * sizeof(Picture));
    memcpy(&s->last_picture, &s1->last_picture, (char*)&s1->last_picture_ptr - (char*)&s1->last_picture);

    s->last_picture_ptr     = REBASE_PICTURE(s1->last_picture_ptr,    s, s1);
    s->current_picture_ptr  = REBASE_PICTURE(s1->current_picture_ptr, s, s1);
    s->next_picture_ptr     = REBASE_PICTURE(s1->next_picture_ptr,    s, s1);

    memcpy(s->prev_pict_types, s1->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE);

    //Error/bug resilience
    s->next_p_frame_damaged = s1->next_p_frame_damaged;
    s->workaround_bugs      = s1->workaround_bugs;

    //MPEG4 timing info
    memcpy(&s->time_increment_bits, &s1->time_increment_bits, (char*)&s1->shape - (char*)&s1->time_increment_bits);

    //B-frame info
    s->max_b_frames         = s1->max_b_frames;
    s->low_delay            = s1->low_delay;
    s->dropable             = s1->dropable;

    //DivX handling (doesn't work)
    s->divx_packed          = s1->divx_packed;

    if(s1->bitstream_buffer){
        if (s1->bitstream_buffer_size + FF_INPUT_BUFFER_PADDING_SIZE > s->allocated_bitstream_buffer_size)
            av_fast_malloc(&s->bitstream_buffer, &s->allocated_bitstream_buffer_size, s1->allocated_bitstream_buffer_size);
        s->bitstream_buffer_size  = s1->bitstream_buffer_size;
        memcpy(s->bitstream_buffer, s1->bitstream_buffer, s1->bitstream_buffer_size);
        memset(s->bitstream_buffer+s->bitstream_buffer_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
    }

    //MPEG2/interlacing info
    memcpy(&s->progressive_sequence, &s1->progressive_sequence, (char*)&s1->rtp_mode - (char*)&s1->progressive_sequence);

    if(!s1->first_field){
        s->last_pict_type= s1->pict_type;
        if (s1->current_picture_ptr) s->last_lambda_for[s1->pict_type] = s1->current_picture_ptr->quality;

        if(s1->pict_type!=FF_B_TYPE){
            s->last_non_b_pict_type= s1->pict_type;
        }
    }

    return 0;
}

M
Michael Niedermayer 已提交
539 540 541 542
/**
 * 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 已提交
543
void MPV_common_defaults(MpegEncContext *s){
M
Michael Niedermayer 已提交
544 545 546 547 548 549 550 551 552 553 554 555
    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;
556 557 558

    s->f_code = 1;
    s->b_code = 1;
559 560 561

    s->picture_range_start = 0;
    s->picture_range_end = MAX_PICTURE_COUNT;
M
Michael Niedermayer 已提交
562 563 564 565 566 567 568 569 570 571
}

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

572
/**
M
Michael Niedermayer 已提交
573 574 575
 * init common structure for both encoder and decoder.
 * this assumes that some variables like width/height are already set
 */
576
av_cold int MPV_common_init(MpegEncContext *s)
577
{
578 579 580 581 582
    int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y,
        threads = (s->encoding ||
                   (HAVE_THREADS &&
                    s->avctx->active_thread_type & FF_THREAD_SLICE)) ?
                  s->avctx->thread_count : 1;
583

584 585
    if(s->codec_id == CODEC_ID_MPEG2VIDEO && !s->progressive_sequence)
        s->mb_height = (s->height + 31) / 32 * 2;
586
    else if (s->codec_id != CODEC_ID_H264)
B
Baptiste Coudurier 已提交
587
        s->mb_height = (s->height + 15) / 16;
L
Loren Merritt 已提交
588

589 590 591 592 593
    if(s->avctx->pix_fmt == PIX_FMT_NONE){
        av_log(s->avctx, AV_LOG_ERROR, "decoding to PIX_FMT_NONE is not supported.\n");
        return -1;
    }

594
    if((s->encoding || (s->avctx->active_thread_type & FF_THREAD_SLICE)) &&
595
       (s->avctx->thread_count > MAX_THREADS || (s->avctx->thread_count > s->mb_height && s->mb_height))){
596 597 598 599
        int max_threads = FFMIN(MAX_THREADS, s->mb_height);
        av_log(s->avctx, AV_LOG_WARNING, "too many threads (%d), reducing to %d\n",
               s->avctx->thread_count, max_threads);
        threads = max_threads;
M
10l  
Michael Niedermayer 已提交
600 601
    }

602
    if((s->width || s->height) && av_image_check_size(s->width, s->height, 0, s->avctx))
603 604
        return -1;

605
    dsputil_init(&s->dsp, s->avctx);
606
    ff_dct_common_init(s);
607

608
    s->flags= s->avctx->flags;
609
    s->flags2= s->avctx->flags2;
610

611
    if (s->width && s->height) {
R
Ronald S. Bultje 已提交
612 613 614 615 616 617
        s->mb_width  = (s->width  + 15) / 16;
        s->mb_stride = s->mb_width + 1;
        s->b8_stride = s->mb_width*2 + 1;
        s->b4_stride = s->mb_width*4 + 1;
        mb_array_size= s->mb_height * s->mb_stride;
        mv_table_size= (s->mb_height+2) * s->mb_stride + 1;
618

R
Ronald S. Bultje 已提交
619 620 621
        /* set chroma shifts */
        avcodec_get_chroma_sub_sample(s->avctx->pix_fmt,&(s->chroma_x_shift),
                                      &(s->chroma_y_shift) );
622

R
Ronald S. Bultje 已提交
623 624 625
        /* 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;
626

R
Ronald S. Bultje 已提交
627
        s->mb_num = s->mb_width * s->mb_height;
628

R
Ronald S. Bultje 已提交
629 630 631 632 633 634
        s->block_wrap[0]=
        s->block_wrap[1]=
        s->block_wrap[2]=
        s->block_wrap[3]= s->b8_stride;
        s->block_wrap[4]=
        s->block_wrap[5]= s->mb_stride;
635

R
Ronald S. Bultje 已提交
636 637 638
        y_size = s->b8_stride * (2 * s->mb_height + 1);
        c_size = s->mb_stride * (s->mb_height + 1);
        yc_size = y_size + 2 * c_size;
639

R
Ronald S. Bultje 已提交
640 641
        /* convert fourcc to upper case */
        s->codec_tag = ff_toupper4(s->avctx->codec_tag);
642

R
Ronald S. Bultje 已提交
643
        s->stream_codec_tag = ff_toupper4(s->avctx->stream_codec_tag);
644

R
Ronald S. Bultje 已提交
645
        s->avctx->coded_frame= (AVFrame*)&s->current_picture;
646

R
Ronald S. Bultje 已提交
647 648 649 650 651
        FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_index2xy, (s->mb_num+1)*sizeof(int), fail) //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;
            }
652
        }
R
Ronald S. Bultje 已提交
653
        s->mb_index2xy[ s->mb_height*s->mb_width ] = (s->mb_height-1)*s->mb_stride + s->mb_width; //FIXME really needed?
654

R
Ronald S. Bultje 已提交
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
        if (s->encoding) {
            /* Allocate MV tables */
            FF_ALLOCZ_OR_GOTO(s->avctx, s->p_mv_table_base            , mv_table_size * 2 * sizeof(int16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->b_forw_mv_table_base       , mv_table_size * 2 * sizeof(int16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->b_back_mv_table_base       , mv_table_size * 2 * sizeof(int16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_forw_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_back_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->b_direct_mv_table_base     , mv_table_size * 2 * sizeof(int16_t), fail)
            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;

            if(s->msmpeg4_version){
                FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int), fail);
            }
            FF_ALLOCZ_OR_GOTO(s->avctx, s->avctx->stats_out, 256, fail);
674

R
Ronald S. Bultje 已提交
675 676
            /* Allocate MB type table */
            FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_type  , mb_array_size * sizeof(uint16_t), fail) //needed for encoding
677

R
Ronald S. Bultje 已提交
678
            FF_ALLOCZ_OR_GOTO(s->avctx, s->lambda_table, mb_array_size * sizeof(int), fail)
679

R
Ronald S. Bultje 已提交
680 681 682 683 684 685
            FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix  , 64*32   * sizeof(int), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix  , 64*32   * sizeof(int), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix16, 64*32*2 * sizeof(uint16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix16, 64*32*2 * sizeof(uint16_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture*), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture*), fail)
686

R
Ronald S. Bultje 已提交
687 688 689
            if(s->avctx->noise_reduction){
                FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_offset, 2 * 64 * sizeof(uint16_t), fail)
            }
690
        }
691
    }
692

693 694 695
    s->picture_count = MAX_PICTURE_COUNT * FFMAX(1, s->avctx->thread_count);
    FF_ALLOCZ_OR_GOTO(s->avctx, s->picture, s->picture_count * sizeof(Picture), fail)
    for(i = 0; i < s->picture_count; i++) {
696 697
        avcodec_get_frame_defaults((AVFrame *)&s->picture[i]);
    }
698

699
    if (s->width && s->height) {
R
Ronald S. Bultje 已提交
700
        FF_ALLOCZ_OR_GOTO(s->avctx, s->error_status_table, mb_array_size*sizeof(uint8_t), fail)
701

R
Ronald S. Bultje 已提交
702 703
        if(s->codec_id==CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){
            /* interlaced direct mode decoding tables */
M
Michael Niedermayer 已提交
704 705 706 707
            for(i=0; i<2; i++){
                int j, k;
                for(j=0; j<2; j++){
                    for(k=0; k<2; k++){
708 709
                        FF_ALLOCZ_OR_GOTO(s->avctx,    s->b_field_mv_table_base[i][j][k], mv_table_size * 2 * sizeof(int16_t), fail)
                        s->b_field_mv_table[i][j][k] = s->b_field_mv_table_base[i][j][k] + s->mb_stride + 1;
M
Michael Niedermayer 已提交
710
                    }
711 712 713
                    FF_ALLOCZ_OR_GOTO(s->avctx, s->b_field_select_table [i][j], mb_array_size * 2 * sizeof(uint8_t), fail)
                    FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_mv_table_base[i][j], mv_table_size * 2 * sizeof(int16_t), fail)
                    s->p_field_mv_table[i][j] = s->p_field_mv_table_base[i][j]+ s->mb_stride + 1;
M
Michael Niedermayer 已提交
714
                }
715
                FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_select_table[i], mb_array_size * 2 * sizeof(uint8_t), fail)
M
Michael Niedermayer 已提交
716
            }
R
Ronald S. Bultje 已提交
717 718 719 720 721 722 723 724 725 726
        }
        if (s->out_format == FMT_H263) {
            /* cbp values */
            FF_ALLOCZ_OR_GOTO(s->avctx, s->coded_block_base, y_size, fail);
            s->coded_block= s->coded_block_base + s->b8_stride + 1;

            /* cbp, ac_pred, pred_dir */
            FF_ALLOCZ_OR_GOTO(s->avctx, s->cbp_table     , mb_array_size * sizeof(uint8_t), fail)
            FF_ALLOCZ_OR_GOTO(s->avctx, s->pred_dir_table, mb_array_size * sizeof(uint8_t), fail)
        }
727

R
Ronald S. Bultje 已提交
728 729 730 731 732 733 734 735 736 737
        if (s->h263_pred || s->h263_plus || !s->encoding) {
            /* dc values */
            //MN: we need these for error resilience of intra-frames
            FF_ALLOCZ_OR_GOTO(s->avctx, s->dc_val_base, yc_size * sizeof(int16_t), fail);
            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;
            s->dc_val[2] = s->dc_val[1] + c_size;
            for(i=0;i<yc_size;i++)
                s->dc_val_base[i] = 1024;
        }
738

R
Ronald S. Bultje 已提交
739 740 741
        /* which mb is a intra block */
        FF_ALLOCZ_OR_GOTO(s->avctx, s->mbintra_table, mb_array_size, fail);
        memset(s->mbintra_table, 1, mb_array_size);
742

R
Ronald S. Bultje 已提交
743 744 745 746
        /* init macroblock skip table */
        FF_ALLOCZ_OR_GOTO(s->avctx, s->mbskip_table, mb_array_size+2, fail);
        //Note the +1 is for a quicker mpeg4 slice_end detection
        FF_ALLOCZ_OR_GOTO(s->avctx, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE, fail);
747

R
Ronald S. Bultje 已提交
748 749 750 751 752 753
        s->parse_context.state= -1;
        if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
            s->visualization_buffer[0] = av_malloc((s->mb_width*16 + 2*EDGE_WIDTH) * s->mb_height*16 + 2*EDGE_WIDTH);
            s->visualization_buffer[1] = av_malloc((s->mb_width*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);
        }
754
    }
755

F
Fabrice Bellard 已提交
756
    s->context_initialized = 1;
757
    s->thread_context[0]= s;
758

759
    if (s->width && s->height) {
760
    if (s->encoding || (HAVE_THREADS && s->avctx->active_thread_type&FF_THREAD_SLICE)) {
R
Ronald S. Bultje 已提交
761 762 763 764
        for(i=1; i<threads; i++){
            s->thread_context[i]= av_malloc(sizeof(MpegEncContext));
            memcpy(s->thread_context[i], s, sizeof(MpegEncContext));
        }
765

R
Ronald S. Bultje 已提交
766 767 768 769 770 771
        for(i=0; i<threads; i++){
            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;
        }
772 773 774 775 776
    } else {
        if(init_duplicate_context(s, s) < 0) goto fail;
        s->start_mb_y = 0;
        s->end_mb_y   = s->mb_height;
    }
777 778
    }

F
Fabrice Bellard 已提交
779 780
    return 0;
 fail:
781
    MPV_common_end(s);
F
Fabrice Bellard 已提交
782 783 784 785 786 787
    return -1;
}

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

790
    if (s->encoding || (HAVE_THREADS && s->avctx->active_thread_type&FF_THREAD_SLICE)) {
791 792 793 794 795 796 797
        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]);
        }
    } else free_duplicate_context(s);
798

799 800 801
    av_freep(&s->parse_context.buffer);
    s->parse_context.buffer_size=0;

802
    av_freep(&s->mb_type);
803 804 805 806 807 808 809 810 811 812 813 814
    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 已提交
815 816 817 818 819 820 821 822 823 824 825 826
    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]);
    }
827

828 829
    av_freep(&s->dc_val_base);
    av_freep(&s->coded_block_base);
830
    av_freep(&s->mbintra_table);
831 832
    av_freep(&s->cbp_table);
    av_freep(&s->pred_dir_table);
833

834
    av_freep(&s->mbskip_table);
835
    av_freep(&s->prev_pict_types);
836
    av_freep(&s->bitstream_buffer);
837 838
    s->allocated_bitstream_buffer_size=0;

M
Michael Niedermayer 已提交
839
    av_freep(&s->avctx->stats_out);
840
    av_freep(&s->ac_stats);
841
    av_freep(&s->error_status_table);
842
    av_freep(&s->mb_index2xy);
843
    av_freep(&s->lambda_table);
844 845
    av_freep(&s->q_intra_matrix);
    av_freep(&s->q_inter_matrix);
846 847
    av_freep(&s->q_intra_matrix16);
    av_freep(&s->q_inter_matrix16);
848 849
    av_freep(&s->input_picture);
    av_freep(&s->reordered_input_picture);
850
    av_freep(&s->dct_offset);
M
cleanup  
Michael Niedermayer 已提交
851

852 853
    if(s->picture && !s->avctx->is_copy){
        for(i=0; i<s->picture_count; i++){
M
Michael Niedermayer 已提交
854 855
            free_picture(s, &s->picture[i]);
        }
F
Fabrice Bellard 已提交
856
    }
857
    av_freep(&s->picture);
F
Fabrice Bellard 已提交
858
    s->context_initialized = 0;
859 860 861
    s->last_picture_ptr=
    s->next_picture_ptr=
    s->current_picture_ptr= NULL;
M
Michael Niedermayer 已提交
862
    s->linesize= s->uvlinesize= 0;
863

864
    for(i=0; i<3; i++)
865
        av_freep(&s->visualization_buffer[i]);
M
Michael Niedermayer 已提交
866

867 868
    if(!(s->avctx->active_thread_type&FF_THREAD_FRAME))
        avcodec_default_free_buffers(s->avctx);
F
Fabrice Bellard 已提交
869 870
}

871
void init_rl(RLTable *rl, uint8_t static_store[2][2*MAX_RUN + MAX_LEVEL + 3])
M
Michael Niedermayer 已提交
872
{
873 874
    int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
    uint8_t index_run[MAX_RUN+1];
M
Michael Niedermayer 已提交
875 876
    int last, run, level, start, end, i;

877
    /* If table is static, we can quit if rl->max_level[0] is not NULL */
878
    if(static_store && rl->max_level[0])
879 880
        return;

M
Michael Niedermayer 已提交
881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903
    /* 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;
        }
904 905
        if(static_store)
            rl->max_level[last] = static_store[last];
906 907
        else
            rl->max_level[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
908
        memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
909 910
        if(static_store)
            rl->max_run[last] = static_store[last] + MAX_RUN + 1;
911 912
        else
            rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
M
Michael Niedermayer 已提交
913
        memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
914 915
        if(static_store)
            rl->index_run[last] = static_store[last] + MAX_RUN + MAX_LEVEL + 2;
916 917
        else
            rl->index_run[last] = av_malloc(MAX_RUN + 1);
M
Michael Niedermayer 已提交
918 919 920 921
        memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
    }
}

922
void init_vlc_rl(RLTable *rl)
923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961
{
    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;
        }
    }
}

962 963 964 965 966 967 968 969 970 971 972 973 974 975 976
void ff_release_unused_pictures(MpegEncContext *s, int remove_current)
{
    int i;

    /* release non reference frames */
    for(i=0; i<s->picture_count; i++){
        if(s->picture[i].data[0] && !s->picture[i].reference
           && (!s->picture[i].owner2 || s->picture[i].owner2 == s)
           && (remove_current || &s->picture[i] != s->current_picture_ptr)
           /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
            free_frame_buffer(s, &s->picture[i]);
        }
    }
}

M
Michael Niedermayer 已提交
977
int ff_find_unused_picture(MpegEncContext *s, int shared){
M
Michael Niedermayer 已提交
978
    int i;
979

M
Michael Niedermayer 已提交
980
    if(shared){
981
        for(i=s->picture_range_start; i<s->picture_range_end; i++){
M
Michael Niedermayer 已提交
982
            if(s->picture[i].data[0]==NULL && s->picture[i].type==0) return i;
M
Michael Niedermayer 已提交
983 984
        }
    }else{
985
        for(i=s->picture_range_start; i<s->picture_range_end; i++){
M
Michael Niedermayer 已提交
986
            if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) return i; //FIXME
M
Michael Niedermayer 已提交
987
        }
988
        for(i=s->picture_range_start; i<s->picture_range_end; i++){
M
Michael Niedermayer 已提交
989
            if(s->picture[i].data[0]==NULL) return i;
M
Michael Niedermayer 已提交
990 991 992
        }
    }

993
    av_log(s->avctx, AV_LOG_FATAL, "Internal error, picture buffer overflow\n");
994 995 996 997 998 999 1000 1001 1002 1003 1004
    /* 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 ...
     */
1005
    abort();
M
Michael Niedermayer 已提交
1006
    return -1;
M
Michael Niedermayer 已提交
1007 1008
}

1009 1010 1011 1012 1013 1014 1015 1016 1017 1018
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;
        }
1019

1020 1021 1022 1023 1024 1025
        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 已提交
1026 1027 1028
/**
 * generic function for encode/decode called after coding/decoding the header and before a frame is coded/decoded
 */
1029
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
F
Fabrice Bellard 已提交
1030
{
M
Michael Niedermayer 已提交
1031
    int i;
1032
    Picture *pic;
1033
    s->mb_skipped = 0;
M
Michael Niedermayer 已提交
1034

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

M
cleanup  
Michael Niedermayer 已提交
1037
    /* mark&release old frames */
1038
    if (s->pict_type != AV_PICTURE_TYPE_B && s->last_picture_ptr && s->last_picture_ptr != s->next_picture_ptr && s->last_picture_ptr->data[0]) {
1039
      if(s->out_format != FMT_H264 || s->codec_id == CODEC_ID_SVQ3){
1040
          free_frame_buffer(s, s->last_picture_ptr);
M
cleanup  
Michael Niedermayer 已提交
1041 1042 1043 1044

        /* release forgotten pictures */
        /* if(mpeg124/h263) */
        if(!s->encoding){
1045
            for(i=0; i<s->picture_count; i++){
M
Michael Niedermayer 已提交
1046
                if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
1047
                    av_log(avctx, AV_LOG_ERROR, "releasing zombie picture\n");
1048
                    free_frame_buffer(s, &s->picture[i]);
M
cleanup  
Michael Niedermayer 已提交
1049 1050
                }
            }
1051
        }
1052
      }
1053
    }
1054

1055
    if(!s->encoding){
1056
        ff_release_unused_pictures(s, 1);
M
Michael Niedermayer 已提交
1057

M
Michael Niedermayer 已提交
1058
        if(s->current_picture_ptr && s->current_picture_ptr->data[0]==NULL)
1059
            pic= s->current_picture_ptr; //we already have a unused image (maybe it was set before reading the header)
M
Michael Niedermayer 已提交
1060 1061
        else{
            i= ff_find_unused_picture(s, 0);
1062
            pic= &s->picture[i];
M
Michael Niedermayer 已提交
1063 1064
        }

1065 1066 1067 1068
        pic->reference= 0;
        if (!s->dropable){
            if (s->codec_id == CODEC_ID_H264)
                pic->reference = s->picture_structure;
1069
            else if (s->pict_type != AV_PICTURE_TYPE_B)
1070 1071
                pic->reference = 3;
        }
M
Michael Niedermayer 已提交
1072

M
Michael Niedermayer 已提交
1073
        pic->coded_picture_number= s->coded_picture_number++;
1074

1075
        if(ff_alloc_picture(s, pic, 0) < 0)
1076
            return -1;
1077

1078
        s->current_picture_ptr= pic;
1079
        //FIXME use only the vars from current_pic
1080
        s->current_picture_ptr->top_field_first= s->top_field_first;
1081
        if(s->codec_id == CODEC_ID_MPEG1VIDEO || s->codec_id == CODEC_ID_MPEG2VIDEO) {
1082
            if(s->picture_structure != PICT_FRAME)
1083
                s->current_picture_ptr->top_field_first= (s->picture_structure == PICT_TOP_FIELD) == s->first_field;
1084
        }
1085
        s->current_picture_ptr->interlaced_frame= !s->progressive_frame && !s->progressive_sequence;
1086
        s->current_picture_ptr->field_picture= s->picture_structure != PICT_FRAME;
M
cleanup  
Michael Niedermayer 已提交
1087
    }
1088

M
Michael Niedermayer 已提交
1089
    s->current_picture_ptr->pict_type= s->pict_type;
1090
//    if(s->flags && CODEC_FLAG_QSCALE)
1091
  //      s->current_picture_ptr->quality= s->new_picture_ptr->quality;
1092
    s->current_picture_ptr->key_frame= s->pict_type == AV_PICTURE_TYPE_I;
M
Michael Niedermayer 已提交
1093

1094
    ff_copy_picture(&s->current_picture, s->current_picture_ptr);
1095

1096
    if (s->pict_type != AV_PICTURE_TYPE_B) {
M
Michael Niedermayer 已提交
1097
        s->last_picture_ptr= s->next_picture_ptr;
1098 1099
        if(!s->dropable)
            s->next_picture_ptr= s->current_picture_ptr;
F
Fabrice Bellard 已提交
1100
    }
1101
/*    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,
1102 1103
        s->last_picture_ptr    ? s->last_picture_ptr->data[0] : NULL,
        s->next_picture_ptr    ? s->next_picture_ptr->data[0] : NULL,
1104 1105
        s->current_picture_ptr ? s->current_picture_ptr->data[0] : NULL,
        s->pict_type, s->dropable);*/
1106

1107
    if(s->codec_id != CODEC_ID_H264){
1108 1109 1110 1111 1112 1113 1114
        if((s->last_picture_ptr==NULL || s->last_picture_ptr->data[0]==NULL) &&
           (s->pict_type!=AV_PICTURE_TYPE_I || s->picture_structure != PICT_FRAME)){
            if (s->pict_type != AV_PICTURE_TYPE_I)
                av_log(avctx, AV_LOG_ERROR, "warning: first frame is no keyframe\n");
            else if (s->picture_structure != PICT_FRAME)
                av_log(avctx, AV_LOG_INFO, "allocate dummy last picture for field based first keyframe\n");

1115 1116 1117 1118 1119
            /* Allocate a dummy frame */
            i= ff_find_unused_picture(s, 0);
            s->last_picture_ptr= &s->picture[i];
            if(ff_alloc_picture(s, s->last_picture_ptr, 0) < 0)
                return -1;
1120 1121
            ff_thread_report_progress((AVFrame*)s->last_picture_ptr, INT_MAX, 0);
            ff_thread_report_progress((AVFrame*)s->last_picture_ptr, INT_MAX, 1);
1122
        }
1123
        if((s->next_picture_ptr==NULL || s->next_picture_ptr->data[0]==NULL) && s->pict_type==AV_PICTURE_TYPE_B){
1124 1125 1126 1127 1128
            /* Allocate a dummy frame */
            i= ff_find_unused_picture(s, 0);
            s->next_picture_ptr= &s->picture[i];
            if(ff_alloc_picture(s, s->next_picture_ptr, 0) < 0)
                return -1;
1129 1130
            ff_thread_report_progress((AVFrame*)s->next_picture_ptr, INT_MAX, 0);
            ff_thread_report_progress((AVFrame*)s->next_picture_ptr, INT_MAX, 1);
1131 1132 1133
        }
    }

1134 1135
    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);
1136

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

1139
    if(s->picture_structure!=PICT_FRAME && s->out_format != FMT_H264){
M
Michael Niedermayer 已提交
1140 1141 1142 1143
        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];
1144
            }
M
Michael Niedermayer 已提交
1145 1146 1147 1148 1149
            s->current_picture.linesize[i] *= 2;
            s->last_picture.linesize[i] *=2;
            s->next_picture.linesize[i] *=2;
        }
    }
1150

1151
    s->error_recognition= avctx->error_recognition;
1152

D
Diego Biurrun 已提交
1153
    /* set dequantizer, we can't do it during init as it might change for mpeg4
D
Diego Biurrun 已提交
1154
       and we can't do it in the header decode as init is not called for mpeg4 there yet */
1155 1156 1157
    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 已提交
1158
    }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
1159 1160 1161 1162 1163 1164
        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;
    }
1165

1166 1167 1168 1169 1170
    if(s->dct_error_sum){
        assert(s->avctx->noise_reduction && s->encoding);

        update_noise_reduction(s);
    }
1171

1172
    if(CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration)
1173
        return ff_xvmc_field_start(s, avctx);
1174

1175
    return 0;
F
Fabrice Bellard 已提交
1176
}
1177

F
Fabrice Bellard 已提交
1178 1179 1180
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
M
cleanup  
Michael Niedermayer 已提交
1181
    int i;
1182
    /* redraw edges for the frame if decoding didn't complete */
1183
    //just to make sure that all data is rendered.
1184
    if(CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration){
1185
        ff_xvmc_field_end(s);
1186 1187
   }else if((s->error_count || s->encoding)
       && !s->avctx->hwaccel
1188
       && !(s->avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU)
1189 1190 1191 1192
       && s->unrestricted_mv
       && s->current_picture.reference
       && !s->intra_only
       && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
1193 1194
            int hshift = av_pix_fmt_descriptors[s->avctx->pix_fmt].log2_chroma_w;
            int vshift = av_pix_fmt_descriptors[s->avctx->pix_fmt].log2_chroma_h;
1195
            s->dsp.draw_edges(s->current_picture.data[0], s->linesize  ,
1196 1197
                              s->h_edge_pos             , s->v_edge_pos,
                              EDGE_WIDTH        , EDGE_WIDTH        , EDGE_TOP | EDGE_BOTTOM);
1198
            s->dsp.draw_edges(s->current_picture.data[1], s->uvlinesize,
1199 1200
                              s->h_edge_pos>>hshift, s->v_edge_pos>>vshift,
                              EDGE_WIDTH>>hshift, EDGE_WIDTH>>vshift, EDGE_TOP | EDGE_BOTTOM);
1201
            s->dsp.draw_edges(s->current_picture.data[2], s->uvlinesize,
1202 1203
                              s->h_edge_pos>>hshift, s->v_edge_pos>>vshift,
                              EDGE_WIDTH>>hshift, EDGE_WIDTH>>vshift, EDGE_TOP | EDGE_BOTTOM);
F
Fabrice Bellard 已提交
1204
    }
1205

M
Michael Niedermayer 已提交
1206
    emms_c();
1207

M
Michael Niedermayer 已提交
1208
    s->last_pict_type    = s->pict_type;
1209
    s->last_lambda_for[s->pict_type]= s->current_picture_ptr->quality;
1210
    if(s->pict_type!=AV_PICTURE_TYPE_B){
1211 1212
        s->last_non_b_pict_type= s->pict_type;
    }
M
Michael Niedermayer 已提交
1213 1214
#if 0
        /* copy back current_picture variables */
M
cleanup  
Michael Niedermayer 已提交
1215 1216 1217 1218
    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;
1219
        }
M
cleanup  
Michael Niedermayer 已提交
1220 1221
    }
    assert(i<MAX_PICTURE_COUNT);
1222
#endif
M
cleanup  
Michael Niedermayer 已提交
1223

M
Michael Niedermayer 已提交
1224
    if(s->encoding){
D
Diego Biurrun 已提交
1225
        /* release non-reference frames */
1226
        for(i=0; i<s->picture_count; i++){
M
Michael Niedermayer 已提交
1227
            if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/){
1228
                free_frame_buffer(s, &s->picture[i]);
M
Michael Niedermayer 已提交
1229 1230
            }
        }
M
cleanup  
Michael Niedermayer 已提交
1231
    }
M
Michael Niedermayer 已提交
1232 1233 1234 1235 1236 1237
    // 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 已提交
1238
    s->avctx->coded_frame= (AVFrame*)s->current_picture_ptr;
1239 1240 1241 1242

    if (s->codec_id != CODEC_ID_H264 && s->current_picture.reference) {
        ff_thread_report_progress((AVFrame*)s->current_picture_ptr, s->mb_height-1, 0);
    }
F
Fabrice Bellard 已提交
1243 1244
}

1245 1246 1247 1248 1249 1250 1251 1252
/**
 * 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 已提交
1253
    int x, y, fr, f;
1254

1255 1256 1257 1258
    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);
1259

1260
    buf[sy*stride + sx]+= color;
1261

D
Diego Biurrun 已提交
1262
    if(FFABS(ex - sx) > FFABS(ey - sy)){
1263
        if(sx > ex){
D
Diego Biurrun 已提交
1264 1265
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1266 1267 1268 1269 1270
        }
        buf+= sx + sy*stride;
        ex-= sx;
        f= ((ey-sy)<<16)/ex;
        for(x= 0; x <= ex; x++){
1271 1272 1273 1274
            y = (x*f)>>16;
            fr= (x*f)&0xFFFF;
            buf[ y   *stride + x]+= (color*(0x10000-fr))>>16;
            buf[(y+1)*stride + x]+= (color*         fr )>>16;
1275 1276 1277
        }
    }else{
        if(sy > ey){
D
Diego Biurrun 已提交
1278 1279
            FFSWAP(int, sx, ex);
            FFSWAP(int, sy, ey);
1280 1281 1282 1283 1284 1285
        }
        buf+= sx + sy*stride;
        ey-= sy;
        if(ey) f= ((ex-sx)<<16)/ey;
        else   f= 0;
        for(y= 0; y <= ey; y++){
1286 1287
            x = (y*f)>>16;
            fr= (y*f)&0xFFFF;
M
Fix ;;  
Michael Niedermayer 已提交
1288 1289
            buf[y*stride + x  ]+= (color*(0x10000-fr))>>16;
            buf[y*stride + x+1]+= (color*         fr )>>16;
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300
        }
    }
}

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

1304 1305 1306 1307
    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);
1308

1309 1310
    dx= ex - sx;
    dy= ey - sy;
1311

1312 1313 1314 1315
    if(dx*dx + dy*dy > 3*3){
        int rx=  dx + dy;
        int ry= -dx + dy;
        int length= ff_sqrt((rx*rx + ry*ry)<<8);
1316

1317 1318 1319
        //FIXME subpixel accuracy
        rx= ROUNDED_DIV(rx*3<<4, length);
        ry= ROUNDED_DIV(ry*3<<4, length);
1320

1321 1322 1323 1324 1325 1326
        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);
}

1327 1328 1329
/**
 * prints debuging info for the given picture.
 */
1330
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict){
1331

1332
    if(s->avctx->hwaccel || !pict || !pict->mb_type) return;
1333 1334 1335

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

1337 1338
        av_log(s->avctx,AV_LOG_DEBUG,"New frame, type: ");
        switch (pict->pict_type) {
1339 1340 1341 1342 1343 1344
            case AV_PICTURE_TYPE_I: av_log(s->avctx,AV_LOG_DEBUG,"I\n"); break;
            case AV_PICTURE_TYPE_P: av_log(s->avctx,AV_LOG_DEBUG,"P\n"); break;
            case AV_PICTURE_TYPE_B: av_log(s->avctx,AV_LOG_DEBUG,"B\n"); break;
            case AV_PICTURE_TYPE_S: av_log(s->avctx,AV_LOG_DEBUG,"S\n"); break;
            case AV_PICTURE_TYPE_SI: av_log(s->avctx,AV_LOG_DEBUG,"SI\n"); break;
            case AV_PICTURE_TYPE_SP: av_log(s->avctx,AV_LOG_DEBUG,"SP\n"); break;
1345
        }
1346 1347 1348 1349 1350
        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;
1351
                    av_log(s->avctx, AV_LOG_DEBUG, "%1d", count);
1352 1353
                }
                if(s->avctx->debug&FF_DEBUG_QP){
1354
                    av_log(s->avctx, AV_LOG_DEBUG, "%2d", pict->qscale_table[x + y*s->mb_stride]);
1355 1356 1357 1358 1359
                }
                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))
1360
                        av_log(s->avctx, AV_LOG_DEBUG, "P");
1361
                    else if(IS_INTRA(mb_type) && IS_ACPRED(mb_type))
1362
                        av_log(s->avctx, AV_LOG_DEBUG, "A");
1363
                    else if(IS_INTRA4x4(mb_type))
1364
                        av_log(s->avctx, AV_LOG_DEBUG, "i");
1365
                    else if(IS_INTRA16x16(mb_type))
1366
                        av_log(s->avctx, AV_LOG_DEBUG, "I");
1367
                    else if(IS_DIRECT(mb_type) && IS_SKIP(mb_type))
1368
                        av_log(s->avctx, AV_LOG_DEBUG, "d");
1369
                    else if(IS_DIRECT(mb_type))
1370
                        av_log(s->avctx, AV_LOG_DEBUG, "D");
1371
                    else if(IS_GMC(mb_type) && IS_SKIP(mb_type))
1372
                        av_log(s->avctx, AV_LOG_DEBUG, "g");
1373
                    else if(IS_GMC(mb_type))
1374
                        av_log(s->avctx, AV_LOG_DEBUG, "G");
1375
                    else if(IS_SKIP(mb_type))
1376
                        av_log(s->avctx, AV_LOG_DEBUG, "S");
1377
                    else if(!USES_LIST(mb_type, 1))
1378
                        av_log(s->avctx, AV_LOG_DEBUG, ">");
1379
                    else if(!USES_LIST(mb_type, 0))
1380
                        av_log(s->avctx, AV_LOG_DEBUG, "<");
1381 1382
                    else{
                        assert(USES_LIST(mb_type, 0) && USES_LIST(mb_type, 1));
1383
                        av_log(s->avctx, AV_LOG_DEBUG, "X");
1384
                    }
1385

1386 1387
                    //segmentation
                    if(IS_8X8(mb_type))
1388
                        av_log(s->avctx, AV_LOG_DEBUG, "+");
1389
                    else if(IS_16X8(mb_type))
1390
                        av_log(s->avctx, AV_LOG_DEBUG, "-");
1391
                    else if(IS_8X16(mb_type))
1392
                        av_log(s->avctx, AV_LOG_DEBUG, "|");
1393
                    else if(IS_INTRA(mb_type) || IS_16X16(mb_type))
1394
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1395
                    else
1396
                        av_log(s->avctx, AV_LOG_DEBUG, "?");
1397 1398


1399
                    if(IS_INTERLACED(mb_type))
1400
                        av_log(s->avctx, AV_LOG_DEBUG, "=");
1401
                    else
1402
                        av_log(s->avctx, AV_LOG_DEBUG, " ");
1403
                }
1404
//                av_log(s->avctx, AV_LOG_DEBUG, " ");
1405
            }
1406
            av_log(s->avctx, AV_LOG_DEBUG, "\n");
1407 1408
        }
    }
1409

1410
    if((s->avctx->debug&(FF_DEBUG_VIS_QP|FF_DEBUG_VIS_MB_TYPE)) || (s->avctx->debug_mv)){
1411 1412
        const int shift= 1 + s->quarter_sample;
        int mb_y;
1413 1414
        uint8_t *ptr;
        int i;
1415
        int h_chroma_shift, v_chroma_shift, block_height;
1416 1417
        const int width = s->avctx->width;
        const int height= s->avctx->height;
1418
        const int mv_sample_log2= 4 - pict->motion_subsample_log2;
1419
        const int mv_stride= (s->mb_width << mv_sample_log2) + (s->codec_id == CODEC_ID_H264 ? 0 : 1);
M
100000l  
Michael Niedermayer 已提交
1420
        s->low_delay=0; //needed to see the vectors without trashing the buffers
1421

1422
        avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1423
        for(i=0; i<3; i++){
1424
            memcpy(s->visualization_buffer[i], pict->data[i], (i==0) ? pict->linesize[i]*height:pict->linesize[i]*height >> v_chroma_shift);
1425 1426 1427 1428
            pict->data[i]= s->visualization_buffer[i];
        }
        pict->type= FF_BUFFER_TYPE_COPY;
        ptr= pict->data[0];
1429
        block_height = 16>>v_chroma_shift;
1430 1431 1432 1433 1434

        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;
1435 1436 1437
                if((s->avctx->debug_mv) && pict->motion_val){
                  int type;
                  for(type=0; type<3; type++){
1438
                    int direction = 0;
1439
                    switch (type) {
1440
                      case 0: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_P_FOR)) || (pict->pict_type!=AV_PICTURE_TYPE_P))
1441 1442 1443
                                continue;
                              direction = 0;
                              break;
1444
                      case 1: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_FOR)) || (pict->pict_type!=AV_PICTURE_TYPE_B))
1445 1446 1447
                                continue;
                              direction = 0;
                              break;
1448
                      case 2: if ((!(s->avctx->debug_mv&FF_DEBUG_VIS_MV_B_BACK)) || (pict->pict_type!=AV_PICTURE_TYPE_B))
1449 1450 1451 1452
                                continue;
                              direction = 1;
                              break;
                    }
1453 1454 1455
                    if(!USES_LIST(pict->mb_type[mb_index], direction))
                        continue;

1456 1457 1458
                    if(IS_8X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<4; i++){
1459 1460
                        int sx= mb_x*16 + 4 + 8*(i&1);
                        int sy= mb_y*16 + 4 + 8*(i>>1);
1461
                        int xy= (mb_x*2 + (i&1) + (mb_y*2 + (i>>1))*mv_stride) << (mv_sample_log2-1);
1462 1463
                        int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                        int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1464
                        draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1465 1466 1467 1468
                      }
                    }else if(IS_16X8(pict->mb_type[mb_index])){
                      int i;
                      for(i=0; i<2; i++){
1469 1470
                        int sx=mb_x*16 + 8;
                        int sy=mb_y*16 + 4 + 8*i;
1471
                        int xy= (mb_x*2 + (mb_y*2 + i)*mv_stride) << (mv_sample_log2-1);
1472 1473
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1474

1475 1476
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1477

1478 1479 1480 1481 1482 1483 1484
                        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;
1485
                        int xy= (mb_x*2 + i + mb_y*2*mv_stride) << (mv_sample_log2-1);
1486 1487
                        int mx=(pict->motion_val[direction][xy][0]>>shift);
                        int my=(pict->motion_val[direction][xy][1]>>shift);
1488

1489 1490
                        if(IS_INTERLACED(pict->mb_type[mb_index]))
                            my*=2;
1491

1492
                        draw_arrow(ptr, sx, sy, mx+sx, my+sy, width, height, s->linesize, 100);
1493 1494 1495 1496
                      }
                    }else{
                      int sx= mb_x*16 + 8;
                      int sy= mb_y*16 + 8;
1497
                      int xy= (mb_x + mb_y*mv_stride) << mv_sample_log2;
1498 1499
                      int mx= (pict->motion_val[direction][xy][0]>>shift) + sx;
                      int my= (pict->motion_val[direction][xy][1]>>shift) + sy;
1500
                      draw_arrow(ptr, sx, sy, mx, my, width, height, s->linesize, 100);
1501
                    }
1502
                  }
M
Michael Niedermayer 已提交
1503 1504 1505 1506
                }
                if((s->avctx->debug&FF_DEBUG_VIS_QP) && pict->motion_val){
                    uint64_t c= (pict->qscale_table[mb_index]*128/31) * 0x0101010101010101ULL;
                    int y;
1507 1508 1509
                    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 已提交
1510 1511 1512 1513 1514 1515 1516 1517 1518 1519
                    }
                }
                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));

1520

M
Michael Niedermayer 已提交
1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
                    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;
1549 1550 1551
                    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 已提交
1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
                    }

                    //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;
                    }
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
                    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;
                        }
                    }
1578

M
Michael Niedermayer 已提交
1579 1580 1581
                    if(IS_INTERLACED(mb_type) && s->codec_id == CODEC_ID_H264){
                        // hmm
                    }
1582 1583 1584 1585 1586
                }
                s->mbskip_table[mb_index]=0;
            }
        }
    }
1587 1588
}

1589
static inline int hpel_motion_lowres(MpegEncContext *s,
M
Michael Niedermayer 已提交
1590 1591 1592 1593 1594 1595 1596 1597 1598
                                  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;
1599
    const int op_index= FFMIN(lowres, 2);
M
Michael Niedermayer 已提交
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
    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);
1613

M
Michael Niedermayer 已提交
1614 1615 1616 1617
    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){
1618
        s->dsp.emulated_edge_mc(s->edge_emu_buffer, src, s->linesize, w+1, (h+1)<<field_based,
M
Michael Niedermayer 已提交
1619 1620 1621 1622 1623
                            src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
        src= s->edge_emu_buffer;
        emu=1;
    }

1624 1625
    sx= (sx << 2) >> lowres;
    sy= (sy << 2) >> lowres;
M
Michael Niedermayer 已提交
1626 1627
    if(field_select)
        src += s->linesize;
1628
    pix_op[op_index](dest, src, stride, h, sx, sy);
M
Michael Niedermayer 已提交
1629 1630 1631
    return emu;
}

F
Fabrice Bellard 已提交
1632
/* apply one mpeg motion vector to the three components */
B
Benoit Fouet 已提交
1633
static av_always_inline void mpeg_motion_lowres(MpegEncContext *s,
1634
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1635
                               int field_based, int bottom_field, int field_select,
B
Benoit Fouet 已提交
1636
                               uint8_t **ref_picture, h264_chroma_mc_func *pix_op,
1637
                               int motion_x, int motion_y, int h, int mb_y)
F
Fabrice Bellard 已提交
1638
{
1639
    uint8_t *ptr_y, *ptr_cb, *ptr_cr;
B
Benoit Fouet 已提交
1640 1641
    int mx, my, src_x, src_y, uvsrc_x, uvsrc_y, uvlinesize, linesize, sx, sy, uvsx, uvsy;
    const int lowres= s->avctx->lowres;
1642
    const int op_index= FFMIN(lowres, 2);
B
Benoit Fouet 已提交
1643 1644 1645 1646
    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;
1647
    linesize   = s->current_picture.linesize[0] << field_based;
M
Michael Niedermayer 已提交
1648
    uvlinesize = s->current_picture.linesize[1] << field_based;
1649

1650
    if(s->quarter_sample){ //FIXME obviously not perfect but qpel will not work in lowres anyway
B
Benoit Fouet 已提交
1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661
        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));
1662
    src_y =(   mb_y*2*block_s>>field_based) + (motion_y >> (lowres+1));
1663

M
Michael Niedermayer 已提交
1664
    if (s->out_format == FMT_H263) {
M
Michael Niedermayer 已提交
1665 1666
        uvsx = ((motion_x>>1) & s_mask) | (sx&1);
        uvsy = ((motion_y>>1) & s_mask) | (sy&1);
M
Michael Niedermayer 已提交
1667 1668 1669 1670 1671 1672 1673 1674
        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);
1675
        uvsrc_y =    mb_y*block_s               + (my >> lowres);
M
Michael Niedermayer 已提交
1676 1677 1678 1679 1680 1681
    } 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));
1682
        uvsrc_y =(   mb_y*block_s>>field_based) + (my >> (lowres+1));
M
Michael Niedermayer 已提交
1683 1684 1685 1686 1687 1688 1689 1690
    }

    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){
1691
            s->dsp.emulated_edge_mc(s->edge_emu_buffer, ptr_y, s->linesize, 17, 17+field_based,
M
Michael Niedermayer 已提交
1692 1693
                             src_x, src_y<<field_based, h_edge_pos, v_edge_pos);
            ptr_y = s->edge_emu_buffer;
1694
            if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1695
                uint8_t *uvbuf= s->edge_emu_buffer+18*s->linesize;
1696
                s->dsp.emulated_edge_mc(uvbuf  , ptr_cb, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1697
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
1698
                s->dsp.emulated_edge_mc(uvbuf+16, ptr_cr, s->uvlinesize, 9, 9+field_based,
M
Michael Niedermayer 已提交
1699 1700 1701 1702 1703 1704
                                 uvsrc_x, uvsrc_y<<field_based, h_edge_pos>>1, v_edge_pos>>1);
                ptr_cb= uvbuf;
                ptr_cr= uvbuf+16;
            }
    }

1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
    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;
    }

1717 1718
    sx= (sx << 2) >> lowres;
    sy= (sy << 2) >> lowres;
M
Michael Niedermayer 已提交
1719
    pix_op[lowres-1](dest_y, ptr_y, linesize, h, sx, sy);
1720

1721
    if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
1722 1723 1724 1725
        uvsx= (uvsx << 2) >> lowres;
        uvsy= (uvsy << 2) >> lowres;
        pix_op[op_index](dest_cb, ptr_cb, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
        pix_op[op_index](dest_cr, ptr_cr, uvlinesize, h >> s->chroma_y_shift, uvsx, uvsy);
M
Michael Niedermayer 已提交
1726
    }
M
Michael Niedermayer 已提交
1727
    //FIXME h261 lowres loop filter
M
Michael Niedermayer 已提交
1728 1729
}

M
Michael Niedermayer 已提交
1730 1731 1732 1733 1734 1735
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;
1736
    const int op_index= FFMIN(lowres, 2);
M
Michael Niedermayer 已提交
1737 1738 1739 1740 1741 1742
    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;
1743

M
Michael Niedermayer 已提交
1744 1745 1746 1747 1748 1749 1750 1751 1752
    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);
1753

M
Michael Niedermayer 已提交
1754 1755 1756 1757
    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));
1758

M
Michael Niedermayer 已提交
1759 1760 1761 1762 1763
    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){
1764
            s->dsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
M
Michael Niedermayer 已提交
1765 1766 1767
            ptr= s->edge_emu_buffer;
            emu=1;
        }
1768
    }
1769 1770 1771
    sx= (sx << 2) >> lowres;
    sy= (sy << 2) >> lowres;
    pix_op[op_index](dest_cb, ptr, s->uvlinesize, block_s, sx, sy);
1772

M
Michael Niedermayer 已提交
1773 1774
    ptr = ref_picture[2] + offset;
    if(emu){
1775
        s->dsp.emulated_edge_mc(s->edge_emu_buffer, ptr, s->uvlinesize, 9, 9, src_x, src_y, h_edge_pos, v_edge_pos);
M
Michael Niedermayer 已提交
1776 1777
        ptr= s->edge_emu_buffer;
    }
1778
    pix_op[op_index](dest_cr, ptr, s->uvlinesize, block_s, sx, sy);
M
Michael Niedermayer 已提交
1779 1780
}

M
doxy  
Michael Niedermayer 已提交
1781
/**
D
Diego Biurrun 已提交
1782
 * motion compensation of a single macroblock
M
doxy  
Michael Niedermayer 已提交
1783 1784 1785 1786 1787 1788
 * @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
1789
 * @param pix_op halfpel motion compensation function (average or put normally)
M
doxy  
Michael Niedermayer 已提交
1790 1791
 * the motion vectors are taken from s->mv and the MV type from s->mv_type
 */
B
Benoit Fouet 已提交
1792
static inline void MPV_motion_lowres(MpegEncContext *s,
1793
                              uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1794
                              int dir, uint8_t **ref_picture,
B
Benoit Fouet 已提交
1795
                              h264_chroma_mc_func *pix_op)
F
Fabrice Bellard 已提交
1796
{
B
Benoit Fouet 已提交
1797
    int mx, my;
F
Fabrice Bellard 已提交
1798
    int mb_x, mb_y, i;
B
Benoit Fouet 已提交
1799 1800
    const int lowres= s->avctx->lowres;
    const int block_s= 8>>lowres;
F
Fabrice Bellard 已提交
1801 1802 1803 1804 1805 1806

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

    switch(s->mv_type) {
    case MV_TYPE_16X16:
B
Benoit Fouet 已提交
1807 1808 1809
        mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                    0, 0, 0,
                    ref_picture, pix_op,
1810
                    s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s, mb_y);
F
Fabrice Bellard 已提交
1811 1812
        break;
    case MV_TYPE_8X8:
M
Michael Niedermayer 已提交
1813 1814 1815
        mx = 0;
        my = 0;
            for(i=0;i<4;i++) {
B
Benoit Fouet 已提交
1816
                hpel_motion_lowres(s, dest_y + ((i & 1) + (i >> 1) * s->linesize)*block_s,
1817
                            ref_picture[0], 0, 0,
B
Benoit Fouet 已提交
1818
                            (2*mb_x + (i & 1))*block_s, (2*mb_y + (i >>1))*block_s,
M
Michael Niedermayer 已提交
1819
                            s->width, s->height, s->linesize,
B
Benoit Fouet 已提交
1820 1821
                            s->h_edge_pos >> lowres, s->v_edge_pos >> lowres,
                            block_s, block_s, pix_op,
M
Michael Niedermayer 已提交
1822
                            s->mv[dir][i][0], s->mv[dir][i][1]);
M
Michael Niedermayer 已提交
1823 1824 1825

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

1828
        if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY))
B
Benoit Fouet 已提交
1829
            chroma_4mv_motion_lowres(s, dest_cb, dest_cr, ref_picture, pix_op, mx, my);
F
Fabrice Bellard 已提交
1830 1831 1832
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
B
Benoit Fouet 已提交
1833 1834 1835 1836
            /* top field */
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                        1, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
1837
                        s->mv[dir][0][0], s->mv[dir][0][1], block_s, mb_y);
B
Benoit Fouet 已提交
1838 1839 1840 1841
            /* bottom field */
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
                        1, 1, s->field_select[dir][1],
                        ref_picture, pix_op,
1842
                        s->mv[dir][1][0], s->mv[dir][1][1], block_s, mb_y);
F
Fabrice Bellard 已提交
1843
        } else {
1844
            if(s->picture_structure != s->field_select[dir][0] + 1 && s->pict_type != AV_PICTURE_TYPE_B && !s->first_field){
1845
                ref_picture= s->current_picture_ptr->data;
1846
            }
F
Fabrice Bellard 已提交
1847

B
Benoit Fouet 已提交
1848
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
1849 1850
                        0, 0, s->field_select[dir][0],
                        ref_picture, pix_op,
1851
                        s->mv[dir][0][0], s->mv[dir][0][1], 2*block_s, mb_y>>1);
F
Fabrice Bellard 已提交
1852
        }
1853
        break;
M
Michael Niedermayer 已提交
1854 1855 1856
    case MV_TYPE_16X8:
        for(i=0; i<2; i++){
            uint8_t ** ref2picture;
1857

1858
            if(s->picture_structure == s->field_select[dir][i] + 1 || s->pict_type == AV_PICTURE_TYPE_B || s->first_field){
1859 1860
                ref2picture= ref_picture;
            }else{
1861
                ref2picture= s->current_picture_ptr->data;
1862
            }
1863

B
Benoit Fouet 已提交
1864
            mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1865
                        0, 0, s->field_select[dir][i],
1866
                        ref2picture, pix_op,
1867
                        s->mv[dir][i][0], s->mv[dir][i][1] + 2*block_s*i, block_s, mb_y>>1);
1868

B
Benoit Fouet 已提交
1869 1870 1871
            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;
1872
        }
F
Fabrice Bellard 已提交
1873
        break;
1874 1875
    case MV_TYPE_DMV:
        if(s->picture_structure == PICT_FRAME){
M
Michael Niedermayer 已提交
1876 1877 1878
            for(i=0; i<2; i++){
                int j;
                for(j=0; j<2; j++){
B
Benoit Fouet 已提交
1879
                    mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1880 1881
                                1, j, j^i,
                                ref_picture, pix_op,
1882
                                s->mv[dir][2*i + j][0], s->mv[dir][2*i + j][1], block_s, mb_y);
M
Michael Niedermayer 已提交
1883
                }
B
Benoit Fouet 已提交
1884
                pix_op = s->dsp.avg_h264_chroma_pixels_tab;
M
Michael Niedermayer 已提交
1885
            }
1886
        }else{
M
Michael Niedermayer 已提交
1887
            for(i=0; i<2; i++){
B
Benoit Fouet 已提交
1888
                mpeg_motion_lowres(s, dest_y, dest_cb, dest_cr,
M
Michael Niedermayer 已提交
1889 1890
                            0, 0, s->picture_structure != i+1,
                            ref_picture, pix_op,
1891
                            s->mv[dir][2*i][0],s->mv[dir][2*i][1],2*block_s, mb_y>>1);
1892

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

M
Michael Niedermayer 已提交
1896 1897
                //opposite parity is always in the same frame if this is second field
                if(!s->first_field){
1898
                    ref_picture = s->current_picture_ptr->data;
M
Michael Niedermayer 已提交
1899
                }
1900
            }
1901 1902
        }
    break;
M
Michael Niedermayer 已提交
1903
    default: assert(0);
F
Fabrice Bellard 已提交
1904 1905 1906
    }
}

1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943
/**
 * find the lowest MB row referenced in the MVs
 */
int MPV_lowest_referenced_row(MpegEncContext *s, int dir)
{
    int my_max = INT_MIN, my_min = INT_MAX, qpel_shift = !s->quarter_sample;
    int my, off, i, mvs;

    if (s->picture_structure != PICT_FRAME) goto unhandled;

    switch (s->mv_type) {
        case MV_TYPE_16X16:
            mvs = 1;
            break;
        case MV_TYPE_16X8:
            mvs = 2;
            break;
        case MV_TYPE_8X8:
            mvs = 4;
            break;
        default:
            goto unhandled;
    }

    for (i = 0; i < mvs; i++) {
        my = s->mv[dir][i][1]<<qpel_shift;
        my_max = FFMAX(my_max, my);
        my_min = FFMIN(my_min, my);
    }

    off = (FFMAX(-my_min, my_max) + 63) >> 6;

    return FFMIN(FFMAX(s->mb_y + off, 0), s->mb_height-1);
unhandled:
    return s->mb_height-1;
}

B
Benoit Fouet 已提交
1944 1945 1946
/* 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 已提交
1947
{
B
Benoit Fouet 已提交
1948 1949 1950
    s->dct_unquantize_intra(s, block, i, qscale);
    s->dsp.idct_put (dest, line_size, block);
}
1951

B
Benoit Fouet 已提交
1952 1953 1954 1955 1956 1957 1958 1959
/* 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);
    }
}
1960

1961
static inline void add_dequant_dct(MpegEncContext *s,
M
Michael Niedermayer 已提交
1962
                           DCTELEM *block, int i, uint8_t *dest, int line_size, int qscale)
M
Michael Niedermayer 已提交
1963
{
F
Fabrice Bellard 已提交
1964
    if (s->block_last_index[i] >= 0) {
1965
        s->dct_unquantize_inter(s, block, i, qscale);
1966

1967
        s->dsp.idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1968 1969 1970
    }
}

1971 1972 1973 1974 1975
/**
 * cleans dc, ac, coded_block for the current non intra MB
 */
void ff_clean_intra_table_entries(MpegEncContext *s)
{
1976
    int wrap = s->b8_stride;
1977
    int xy = s->block_index[0];
1978 1979 1980

    s->dc_val[0][xy           ] =
    s->dc_val[0][xy + 1       ] =
1981 1982 1983
    s->dc_val[0][xy     + wrap] =
    s->dc_val[0][xy + 1 + wrap] = 1024;
    /* ac pred */
1984 1985
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
1986 1987 1988 1989 1990 1991 1992
    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 */
1993 1994
    wrap = s->mb_stride;
    xy = s->mb_x + s->mb_y * wrap;
1995 1996 1997
    s->dc_val[1][xy] =
    s->dc_val[2][xy] = 1024;
    /* ac pred */
1998 1999
    memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
2000

2001
    s->mbintra_table[xy]= 0;
2002 2003
}

F
Fabrice Bellard 已提交
2004 2005 2006 2007 2008 2009 2010 2011 2012 2013
/* 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)
 */
2014 2015
static av_always_inline
void MPV_decode_mb_internal(MpegEncContext *s, DCTELEM block[12][64],
K
Keiji Costantini 已提交
2016
                            int lowres_flag, int is_mpeg12)
F
Fabrice Bellard 已提交
2017
{
2018
    const int mb_xy = s->mb_y * s->mb_stride + s->mb_x;
2019
    if(CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration){
2020
        ff_xvmc_decode_mb(s);//xvmc uses pblocks
I
Ivan Kalvachev 已提交
2021 2022
        return;
    }
F
Fabrice Bellard 已提交
2023

2024 2025 2026 2027
    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];
2028 2029 2030
       av_log(s->avctx, AV_LOG_DEBUG, "DCT coeffs of MB at %dx%d:\n", s->mb_x, s->mb_y);
       for(i=0; i<6; i++){
           for(j=0; j<64; j++){
2031
               *dct++ = block[i][s->dsp.idct_permutation[j]];
2032 2033 2034 2035
               av_log(s->avctx, AV_LOG_DEBUG, "%5d", dct[-1]);
           }
           av_log(s->avctx, AV_LOG_DEBUG, "\n");
       }
2036 2037
    }

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

F
Fabrice Bellard 已提交
2040 2041
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
K
Keiji Costantini 已提交
2042
        if (!is_mpeg12 && (s->h263_pred || s->h263_aic)) {
M
Michael Niedermayer 已提交
2043
            if(s->mbintra_table[mb_xy])
2044
                ff_clean_intra_table_entries(s);
F
Fabrice Bellard 已提交
2045
        } else {
2046 2047
            s->last_dc[0] =
            s->last_dc[1] =
F
Fabrice Bellard 已提交
2048 2049 2050
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
K
Keiji Costantini 已提交
2051
    else if (!is_mpeg12 && (s->h263_pred || s->h263_aic))
M
Michael Niedermayer 已提交
2052
        s->mbintra_table[mb_xy]=1;
2053

2054
    if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==AV_PICTURE_TYPE_B) && s->avctx->mb_decision != FF_MB_DECISION_RD)) { //FIXME precalc
2055
        uint8_t *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
2056
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
2057 2058
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
2059
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this would be wrong for field pics
M
Michael Niedermayer 已提交
2060
        const int uvlinesize= s->current_picture.linesize[1];
2061
        const int readable= s->pict_type != AV_PICTURE_TYPE_B || s->encoding || s->avctx->draw_horiz_band || lowres_flag;
M
Michael Niedermayer 已提交
2062
        const int block_size= lowres_flag ? 8>>s->avctx->lowres : 8;
2063

M
cleanup  
Michael Niedermayer 已提交
2064 2065 2066
        /* 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){
2067
            uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
M
cleanup  
Michael Niedermayer 已提交
2068
            const int age= s->current_picture.age;
M
Michael Niedermayer 已提交
2069

M
cleanup  
Michael Niedermayer 已提交
2070 2071
            assert(age);

2072 2073
            if (s->mb_skipped) {
                s->mb_skipped= 0;
2074
                assert(s->pict_type!=AV_PICTURE_TYPE_I);
2075

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

M
cleanup  
Michael Niedermayer 已提交
2079
                /* if previous was skipped too, then nothing to do !  */
2080 2081
                if (*mbskip_ptr >= age && s->current_picture.reference){
                    return;
M
cleanup  
Michael Niedermayer 已提交
2082
                }
2083 2084 2085 2086
            } 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{
2087 2088
                *mbskip_ptr = 0; /* not skipped */
            }
M
Michael Niedermayer 已提交
2089
        }
2090

2091
        dct_linesize = linesize << s->interlaced_dct;
M
Michael Niedermayer 已提交
2092
        dct_offset =(s->interlaced_dct)? linesize : linesize*block_size;
2093

2094 2095 2096 2097 2098
        if(readable){
            dest_y=  s->dest[0];
            dest_cb= s->dest[1];
            dest_cr= s->dest[2];
        }else{
2099
            dest_y = s->b_scratchpad;
2100
            dest_cb= s->b_scratchpad+16*linesize;
2101
            dest_cr= s->b_scratchpad+32*linesize;
2102
        }
M
Michael Niedermayer 已提交
2103

F
Fabrice Bellard 已提交
2104 2105
        if (!s->mb_intra) {
            /* motion handling */
2106
            /* decoding or more than one mb_type (MC was already done otherwise) */
2107
            if(!s->encoding){
2108 2109 2110 2111 2112 2113 2114 2115 2116 2117

                if(HAVE_PTHREADS && s->avctx->active_thread_type&FF_THREAD_FRAME) {
                    if (s->mv_dir & MV_DIR_FORWARD) {
                        ff_thread_await_progress((AVFrame*)s->last_picture_ptr, MPV_lowest_referenced_row(s, 0), 0);
                    }
                    if (s->mv_dir & MV_DIR_BACKWARD) {
                        ff_thread_await_progress((AVFrame*)s->next_picture_ptr, MPV_lowest_referenced_row(s, 1), 0);
                    }
                }

M
Michael Niedermayer 已提交
2118 2119
                if(lowres_flag){
                    h264_chroma_mc_func *op_pix = s->dsp.put_h264_chroma_pixels_tab;
F
Fabrice Bellard 已提交
2120

M
Michael Niedermayer 已提交
2121 2122 2123 2124 2125 2126 2127 2128
                    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{
2129
                    op_qpix= s->me.qpel_put;
2130
                    if ((!s->no_rounding) || s->pict_type==AV_PICTURE_TYPE_B){
M
Michael Niedermayer 已提交
2131 2132 2133 2134 2135 2136 2137
                        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;
2138
                        op_qpix= s->me.qpel_avg;
M
Michael Niedermayer 已提交
2139 2140 2141 2142
                    }
                    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);
                    }
2143
                }
F
Fabrice Bellard 已提交
2144 2145
            }

M
Michael Niedermayer 已提交
2146
            /* skip dequant / idct if we are really late ;) */
M
Michael Niedermayer 已提交
2147
            if(s->avctx->skip_idct){
2148 2149
                if(  (s->avctx->skip_idct >= AVDISCARD_NONREF && s->pict_type == AV_PICTURE_TYPE_B)
                   ||(s->avctx->skip_idct >= AVDISCARD_NONKEY && s->pict_type != AV_PICTURE_TYPE_I)
M
Michael Niedermayer 已提交
2150 2151 2152
                   || s->avctx->skip_idct >= AVDISCARD_ALL)
                    goto skip_idct;
            }
M
Michael Niedermayer 已提交
2153

F
Fabrice Bellard 已提交
2154
            /* add dct residue */
2155
            if(s->encoding || !(   s->msmpeg4_version || s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO
2156
                                || (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant))){
M
Michael Niedermayer 已提交
2157 2158 2159 2160
                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 已提交
2161

2162
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173
                    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 已提交
2174
                }
K
Keiji Costantini 已提交
2175
            } else if(is_mpeg12 || (s->codec_id != CODEC_ID_WMV2)){
M
Michael Niedermayer 已提交
2176 2177 2178 2179
                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 已提交
2180

2181
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202
                    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
            }
2203
            else if (CONFIG_WMV2_DECODER || CONFIG_WMV2_ENCODER) {
M
Michael Niedermayer 已提交
2204
                ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
M
Michael Niedermayer 已提交
2205
            }
F
Fabrice Bellard 已提交
2206 2207
        } else {
            /* dct only in intra block */
M
Michael Niedermayer 已提交
2208
            if(s->encoding || !(s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO)){
M
Michael Niedermayer 已提交
2209 2210 2211 2212
                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);
2213

2214
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
2215 2216 2217 2218 2219 2220 2221 2222 2223
                    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 已提交
2224
                        put_dct(s, block[7], 7, dest_cr + dct_offset, dct_linesize, s->chroma_qscale);
M
Michael Niedermayer 已提交
2225 2226 2227
                    }
                }
            }else{
B
Benoit Fouet 已提交
2228 2229 2230 2231
                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]);
2232

2233
                if(!CONFIG_GRAY || !(s->flags&CODEC_FLAG_GRAY)){
B
Benoit Fouet 已提交
2234 2235 2236 2237
                    if(s->chroma_y_shift){
                        s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                        s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
                    }else{
2238

B
Benoit Fouet 已提交
2239 2240
                        dct_linesize = uvlinesize << s->interlaced_dct;
                        dct_offset =(s->interlaced_dct)? uvlinesize : uvlinesize*8;
2241

B
Benoit Fouet 已提交
2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253
                        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
2254
            }
M
Michael Niedermayer 已提交
2255
        }
B
Benoit Fouet 已提交
2256 2257 2258 2259 2260
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);
2261
        }
M
Michael Niedermayer 已提交
2262 2263 2264
    }
}

B
Benoit Fouet 已提交
2265
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]){
2266
#if !CONFIG_SMALL
K
Keiji Costantini 已提交
2267 2268 2269 2270 2271 2272 2273
    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 已提交
2274 2275
}

B
Benoit Fouet 已提交
2276 2277 2278 2279 2280
/**
 *
 * @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){
2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293
    const int field_pic= s->picture_structure != PICT_FRAME;
    if(field_pic){
        h <<= 1;
        y <<= 1;
    }

    if (!s->avctx->hwaccel
       && !(s->avctx->codec->capabilities&CODEC_CAP_HWACCEL_VDPAU)
       && s->unrestricted_mv
       && s->current_picture.reference
       && !s->intra_only
       && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
        int sides = 0, edge_h;
2294 2295
        int hshift = av_pix_fmt_descriptors[s->avctx->pix_fmt].log2_chroma_w;
        int vshift = av_pix_fmt_descriptors[s->avctx->pix_fmt].log2_chroma_h;
2296 2297 2298 2299 2300
        if (y==0) sides |= EDGE_TOP;
        if (y + h >= s->v_edge_pos) sides |= EDGE_BOTTOM;

        edge_h= FFMIN(h, s->v_edge_pos - y);

2301 2302 2303 2304 2305 2306
        s->dsp.draw_edges(s->current_picture_ptr->data[0] +  y         *s->linesize  , s->linesize,
                          s->h_edge_pos        , edge_h        , EDGE_WIDTH        , EDGE_WIDTH        , sides);
        s->dsp.draw_edges(s->current_picture_ptr->data[1] + (y>>vshift)*s->uvlinesize, s->uvlinesize,
                          s->h_edge_pos>>hshift, edge_h>>hshift, EDGE_WIDTH>>hshift, EDGE_WIDTH>>vshift, sides);
        s->dsp.draw_edges(s->current_picture_ptr->data[2] + (y>>vshift)*s->uvlinesize, s->uvlinesize,
                          s->h_edge_pos>>hshift, edge_h>>hshift, EDGE_WIDTH>>hshift, EDGE_WIDTH>>vshift, sides);
2307 2308 2309 2310 2311 2312
    }

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

    if(field_pic && s->first_field && !(s->avctx->slice_flags&SLICE_FLAG_ALLOW_FIELD)) return;

B
Benoit Fouet 已提交
2313 2314 2315
    if (s->avctx->draw_horiz_band) {
        AVFrame *src;
        int offset[4];
M
Michael Niedermayer 已提交
2316

2317
        if(s->pict_type==AV_PICTURE_TYPE_B || s->low_delay || (s->avctx->slice_flags&SLICE_FLAG_CODED_ORDER))
B
Benoit Fouet 已提交
2318 2319 2320 2321 2322
            src= (AVFrame*)s->current_picture_ptr;
        else if(s->last_picture_ptr)
            src= (AVFrame*)s->last_picture_ptr;
        else
            return;
2323

2324
        if(s->pict_type==AV_PICTURE_TYPE_B && s->picture_structure == PICT_FRAME && s->out_format != FMT_H264){
B
Benoit Fouet 已提交
2325 2326 2327 2328
            offset[0]=
            offset[1]=
            offset[2]=
            offset[3]= 0;
M
Michael Niedermayer 已提交
2329
        }else{
M
Fix ;;  
Michael Niedermayer 已提交
2330
            offset[0]= y * s->linesize;
B
Benoit Fouet 已提交
2331 2332 2333
            offset[1]=
            offset[2]= (y >> s->chroma_y_shift) * s->uvlinesize;
            offset[3]= 0;
M
Michael Niedermayer 已提交
2334
        }
2335

B
Benoit Fouet 已提交
2336
        emms_c();
M
Michael Niedermayer 已提交
2337

B
Benoit Fouet 已提交
2338 2339 2340 2341
        s->avctx->draw_horiz_band(s->avctx, src, offset,
                                  y, s->picture_structure, h);
    }
}
2342

B
Benoit Fouet 已提交
2343 2344 2345 2346
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 已提交
2347

B
Benoit Fouet 已提交
2348 2349 2350 2351 2352 2353 2354
    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
2355

B
Benoit Fouet 已提交
2356 2357 2358
    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));
2359

2360
    if(!(s->pict_type==AV_PICTURE_TYPE_B && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME))
B
Benoit Fouet 已提交
2361
    {
2362
        if(s->picture_structure==PICT_FRAME){
B
Benoit Fouet 已提交
2363 2364 2365
        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);
2366 2367 2368 2369 2370 2371
        }else{
            s->dest[0] += (s->mb_y>>1) *   linesize << mb_size;
            s->dest[1] += (s->mb_y>>1) * uvlinesize << (mb_size - s->chroma_y_shift);
            s->dest[2] += (s->mb_y>>1) * uvlinesize << (mb_size - s->chroma_y_shift);
            assert((s->mb_y&1) == (s->picture_structure == PICT_BOTTOM_FIELD));
        }
M
Michael Niedermayer 已提交
2372 2373
    }
}
2374

B
Benoit Fouet 已提交
2375 2376 2377
void ff_mpeg_flush(AVCodecContext *avctx){
    int i;
    MpegEncContext *s = avctx->priv_data;
M
Michael Niedermayer 已提交
2378

B
Benoit Fouet 已提交
2379 2380
    if(s==NULL || s->picture==NULL)
        return;
M
Michael Niedermayer 已提交
2381

2382
    for(i=0; i<s->picture_count; i++){
B
Benoit Fouet 已提交
2383 2384
       if(s->picture[i].data[0] && (   s->picture[i].type == FF_BUFFER_TYPE_INTERNAL
                                    || s->picture[i].type == FF_BUFFER_TYPE_USER))
2385
        free_frame_buffer(s, &s->picture[i]);
F
Fabrice Bellard 已提交
2386
    }
B
Benoit Fouet 已提交
2387
    s->current_picture_ptr = s->last_picture_ptr = s->next_picture_ptr = NULL;
2388

B
Benoit Fouet 已提交
2389
    s->mb_x= s->mb_y= 0;
2390
    s->closed_gop= 0;
M
Michael Niedermayer 已提交
2391

B
Benoit Fouet 已提交
2392 2393 2394 2395 2396 2397 2398 2399
    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 已提交
2400 2401
}

2402
static void dct_unquantize_mpeg1_intra_c(MpegEncContext *s,
2403
                                   DCTELEM *block, int n, int qscale)
F
Fabrice Bellard 已提交
2404
{
M
Michael Niedermayer 已提交
2405
    int i, level, nCoeffs;
2406
    const uint16_t *quant_matrix;
F
Fabrice Bellard 已提交
2407

2408
    nCoeffs= s->block_last_index[n];
2409 2410

    if (n < 4)
2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427
        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 已提交
2428
            }
2429
            block[j] = level;
F
Fabrice Bellard 已提交
2430
        }
2431 2432 2433
    }
}

2434
static void dct_unquantize_mpeg1_inter_c(MpegEncContext *s,
2435 2436 2437 2438 2439 2440
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const uint16_t *quant_matrix;

    nCoeffs= s->block_last_index[n];
2441

2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456
    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 已提交
2457
            }
2458
            block[j] = level;
F
Fabrice Bellard 已提交
2459 2460 2461
        }
    }
}
2462

2463
static void dct_unquantize_mpeg2_intra_c(MpegEncContext *s,
2464 2465 2466
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
2467
    const uint16_t *quant_matrix;
2468

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

    if (n < 4)
2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492
        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;
        }
    }
}

2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525
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;
}

2526
static void dct_unquantize_mpeg2_inter_c(MpegEncContext *s,
2527 2528 2529 2530 2531 2532 2533 2534
                                   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];
2535

2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556
    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;
}

2557
static void dct_unquantize_h263_intra_c(MpegEncContext *s,
2558 2559 2560 2561
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
    int nCoeffs;
2562

2563
    assert(s->block_last_index[n]>=0);
2564

2565
    qmul = qscale << 1;
2566

2567
    if (!s->h263_aic) {
2568
        if (n < 4)
2569 2570 2571
            block[0] = block[0] * s->y_dc_scale;
        else
            block[0] = block[0] * s->c_dc_scale;
2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587
        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;
2588
            }
2589
            block[i] = level;
2590 2591 2592 2593
        }
    }
}

2594
static void dct_unquantize_h263_inter_c(MpegEncContext *s,
2595 2596 2597
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
2598
    int nCoeffs;
2599

2600
    assert(s->block_last_index[n]>=0);
2601

2602 2603
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
2604

2605
    nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
2606

2607
    for(i=0; i<=nCoeffs; i++) {
2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
            }
            block[i] = level;
        }
    }
}
F
Fabrice Bellard 已提交
2619

2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635
/**
 * 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 ];
}
2636 2637 2638 2639 2640 2641

void MPV_report_decode_progress(MpegEncContext *s)
{
    if (s->pict_type != FF_B_TYPE && !s->partitioned_frame && !s->error_occurred)
        ff_thread_report_progress((AVFrame*)s->current_picture_ptr, s->mb_y, 0);
}