mpegvideo.c 142.6 KB
Newer Older
F
Fabrice Bellard 已提交
1 2
/*
 * The simplest mpeg encoder (well, it was the simplest!)
F
Fabrice Bellard 已提交
3
 * Copyright (c) 2000,2001 Fabrice Bellard.
F
Fabrice Bellard 已提交
4
 *
F
Fabrice Bellard 已提交
5 6 7 8
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
F
Fabrice Bellard 已提交
9
 *
F
Fabrice Bellard 已提交
10
 * This library is distributed in the hope that it will be useful,
F
Fabrice Bellard 已提交
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
F
Fabrice Bellard 已提交
12 13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
F
Fabrice Bellard 已提交
14
 *
F
Fabrice Bellard 已提交
15 16 17
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
M
Michael Niedermayer 已提交
18
 *
19
 * 4MV & hq & b-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
F
Fabrice Bellard 已提交
20
 */
21
 
M
Michael Niedermayer 已提交
22 23 24 25 26
/**
 * @file mpegvideo.c
 * The simplest mpeg encoder (well, it was the simplest!).
 */ 
 
27
#include <ctype.h>
28
#include <limits.h>
F
Fabrice Bellard 已提交
29 30 31 32
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"

33 34 35 36
#ifdef USE_FASTMEMCPY
#include "fastmemcpy.h"
#endif

37 38 39
//#undef NDEBUG
//#include <assert.h>

40
#ifdef CONFIG_ENCODERS
41
static void encode_picture(MpegEncContext *s, int picture_number);
42
#endif //CONFIG_ENCODERS
43 44
static void dct_unquantize_mpeg1_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale);
45 46
static void dct_unquantize_mpeg2_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
47 48
static void dct_unquantize_h263_c(MpegEncContext *s, 
                                  DCTELEM *block, int n, int qscale);
49
static void draw_edges_c(uint8_t *buf, int wrap, int width, int height, int w);
50
#ifdef CONFIG_ENCODERS
51
static int dct_quantize_c(MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
M
Michael Niedermayer 已提交
52
static int dct_quantize_trellis_c(MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
53
#endif //CONFIG_ENCODERS
M
Michael Niedermayer 已提交
54

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

F
Fabrice Bellard 已提交
57 58 59 60 61 62

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

//#define DEBUG

63

F
Fabrice Bellard 已提交
64 65 66
/* for jpeg fast DCT */
#define CONST_BITS 14

67
static const uint16_t aanscales[64] = {
F
Fabrice Bellard 已提交
68 69 70 71 72 73 74
    /* precomputed values scaled up by 14 bits */
    16384, 22725, 21407, 19266, 16384, 12873,  8867,  4520,
    22725, 31521, 29692, 26722, 22725, 17855, 12299,  6270,
    21407, 29692, 27969, 25172, 21407, 16819, 11585,  5906,
    19266, 26722, 25172, 22654, 19266, 15137, 10426,  5315,
    16384, 22725, 21407, 19266, 16384, 12873,  8867,  4520,
    12873, 17855, 16819, 15137, 12873, 10114,  6967,  3552,
M
Michael Niedermayer 已提交
75 76
    8867 , 12299, 11585, 10426,  8867,  6967,  4799,  2446,
    4520 ,  6270,  5906,  5315,  4520,  3552,  2446,  1247
F
Fabrice Bellard 已提交
77 78
};

79
static const uint8_t h263_chroma_roundtab[16] = {
80
//  0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15
F
Fabrice Bellard 已提交
81 82 83
    0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2,
};

84
#ifdef CONFIG_ENCODERS
85
static uint8_t (*default_mv_penalty)[MAX_MV*2+1]=NULL;
86
static uint8_t default_fcode_tab[MAX_MV*2+1];
87

M
Michael Niedermayer 已提交
88 89
enum PixelFormat ff_yuv420p_list[2]= {PIX_FMT_YUV420P, -1};

90
static void convert_matrix(MpegEncContext *s, int (*qmat)[64], uint16_t (*qmat16)[64], uint16_t (*qmat16_bias)[64],
91
                           const uint16_t *quant_matrix, int bias, int qmin, int qmax)
F
Fabrice Bellard 已提交
92
{
93 94
    int qscale;

95
    for(qscale=qmin; qscale<=qmax; qscale++){
96
        int i;
97
        if (s->dsp.fdct == ff_jpeg_fdct_islow) {
98
            for(i=0;i<64;i++) {
99
                const int j= s->dsp.idct_permutation[i];
100 101 102 103 104
                /* 16 <= qscale * quant_matrix[i] <= 7905 */
                /* 19952         <= aanscales[i] * qscale * quant_matrix[i]           <= 249205026 */
                /* (1<<36)/19952 >= (1<<36)/(aanscales[i] * qscale * quant_matrix[i]) >= (1<<36)/249205026 */
                /* 3444240       >= (1<<36)/(aanscales[i] * qscale * quant_matrix[i]) >= 275 */
                
105
                qmat[qscale][i] = (int)((uint64_t_C(1) << QMAT_SHIFT) / 
106 107
                                (qscale * quant_matrix[j]));
            }
108
        } else if (s->dsp.fdct == fdct_ifast) {
109
            for(i=0;i<64;i++) {
110
                const int j= s->dsp.idct_permutation[i];
111 112 113 114 115
                /* 16 <= qscale * quant_matrix[i] <= 7905 */
                /* 19952         <= aanscales[i] * qscale * quant_matrix[i]           <= 249205026 */
                /* (1<<36)/19952 >= (1<<36)/(aanscales[i] * qscale * quant_matrix[i]) >= (1<<36)/249205026 */
                /* 3444240       >= (1<<36)/(aanscales[i] * qscale * quant_matrix[i]) >= 275 */
                
116
                qmat[qscale][i] = (int)((uint64_t_C(1) << (QMAT_SHIFT + 14)) / 
117 118 119 120
                                (aanscales[i] * qscale * quant_matrix[j]));
            }
        } else {
            for(i=0;i<64;i++) {
121
                const int j= s->dsp.idct_permutation[i];
122 123 124 125 126
                /* We can safely suppose that 16 <= quant_matrix[i] <= 255
                   So 16           <= qscale * quant_matrix[i]             <= 7905
                   so (1<<19) / 16 >= (1<<19) / (qscale * quant_matrix[i]) >= (1<<19) / 7905
                   so 32768        >= (1<<19) / (qscale * quant_matrix[i]) >= 67
                */
127
                qmat[qscale][i] = (int)((uint64_t_C(1) << QMAT_SHIFT) / (qscale * quant_matrix[j]));
M
Michael Niedermayer 已提交
128
//                qmat  [qscale][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[i]);
129
                qmat16[qscale][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[j]);
130 131 132 133

                if(qmat16[qscale][i]==0 || qmat16[qscale][i]==128*256) qmat16[qscale][i]=128*256-1;
                qmat16_bias[qscale][i]= ROUNDED_DIV(bias<<(16-QUANT_BIAS_SHIFT), qmat16[qscale][i]);
            }
F
Fabrice Bellard 已提交
134 135 136
        }
    }
}
137 138
#endif //CONFIG_ENCODERS

139 140 141 142 143 144 145 146 147
// move into common.c perhaps 
#define CHECKED_ALLOCZ(p, size)\
{\
    p= av_mallocz(size);\
    if(p==NULL){\
        perror("malloc");\
        goto fail;\
    }\
}
148

149
void ff_init_scantable(MpegEncContext *s, ScanTable *st, const uint8_t *src_scantable){
150 151
    int i;
    int end;
M
Michael Niedermayer 已提交
152 153
    
    st->scantable= src_scantable;
154 155 156 157

    for(i=0; i<64; i++){
        int j;
        j = src_scantable[i];
158
        st->permutated[i] = s->dsp.idct_permutation[j];
159 160 161
#ifdef ARCH_POWERPC
        st->inverse[j] = i;
#endif
162 163 164 165 166 167 168 169 170 171 172
    }
    
    end=-1;
    for(i=0; i<64; i++){
        int j;
        j = st->permutated[i];
        if(j>end) end=j;
        st->raster_end[i]= end;
    }
}

173 174
/* init common dct for both encoder and decoder */
int DCT_common_init(MpegEncContext *s)
F
Fabrice Bellard 已提交
175
{
176
    s->dct_unquantize_h263 = dct_unquantize_h263_c;
177 178
    s->dct_unquantize_mpeg1 = dct_unquantize_mpeg1_c;
    s->dct_unquantize_mpeg2 = dct_unquantize_mpeg2_c;
179

180
#ifdef CONFIG_ENCODERS
181
    s->dct_quantize= dct_quantize_c;
182
#endif
183 184 185
        
#ifdef HAVE_MMX
    MPV_common_init_mmx(s);
186 187 188
#endif
#ifdef ARCH_ALPHA
    MPV_common_init_axp(s);
N
Nick Kurshev 已提交
189
#endif
190 191 192
#ifdef HAVE_MLIB
    MPV_common_init_mlib(s);
#endif
193 194 195
#ifdef HAVE_MMI
    MPV_common_init_mmi(s);
#endif
196
#ifdef ARCH_ARMV4L
197
    MPV_common_init_armv4l(s);
198
#endif
199 200 201
#ifdef ARCH_POWERPC
    MPV_common_init_ppc(s);
#endif
202

203
#ifdef CONFIG_ENCODERS
204 205
    s->fast_dct_quantize= s->dct_quantize;

M
Michael Niedermayer 已提交
206 207 208 209
    if(s->flags&CODEC_FLAG_TRELLIS_QUANT){
        s->dct_quantize= dct_quantize_trellis_c; //move before MPV_common_init_*
    }

210 211
#endif //CONFIG_ENCODERS

212 213 214 215 216 217 218
    /* load & permutate scantables
       note: only wmv uses differnt ones 
    */
    ff_init_scantable(s, &s->inter_scantable  , ff_zigzag_direct);
    ff_init_scantable(s, &s->intra_scantable  , ff_zigzag_direct);
    ff_init_scantable(s, &s->intra_h_scantable, ff_alternate_horizontal_scan);
    ff_init_scantable(s, &s->intra_v_scantable, ff_alternate_vertical_scan);
219

220 221
    s->picture_structure= PICT_FRAME;
    
222 223 224
    return 0;
}

M
cleanup  
Michael Niedermayer 已提交
225
/**
M
Michael Niedermayer 已提交
226 227
 * allocates a Picture
 * The pixels are allocated/set by calling get_buffer() if shared=0
M
cleanup  
Michael Niedermayer 已提交
228
 */
M
Michael Niedermayer 已提交
229 230 231 232 233 234 235 236 237 238 239
static int alloc_picture(MpegEncContext *s, Picture *pic, int shared){
    
    if(shared){
        assert(pic->data[0]);
        assert(pic->type == 0 || pic->type == FF_BUFFER_TYPE_SHARED);
        pic->type= FF_BUFFER_TYPE_SHARED;
    }else{
        int r;
        
        assert(!pic->data[0]);
        
M
Michael Niedermayer 已提交
240
        r= s->avctx->get_buffer(s->avctx, (AVFrame*)pic);
M
Michael Niedermayer 已提交
241 242
        
        if(r<0 || !pic->age || !pic->type || !pic->data[0]){
243
            fprintf(stderr, "get_buffer() failed (%d %d %d %p)\n", r, pic->age, pic->type, pic->data[0]);
M
Michael Niedermayer 已提交
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
            return -1;
        }

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

        if(pic->linesize[1] != pic->linesize[2]){
            fprintf(stderr, "get_buffer() failed (uv stride missmatch)\n");
            return -1;
        }

        s->linesize  = pic->linesize[0];
        s->uvlinesize= pic->linesize[1];
M
cleanup  
Michael Niedermayer 已提交
259
    }
M
Michael Niedermayer 已提交
260 261 262
    
    if(pic->qscale_table==NULL){
        if (s->encoding) {        
263 264 265
            CHECKED_ALLOCZ(pic->mb_var   , s->mb_num * sizeof(int16_t))
            CHECKED_ALLOCZ(pic->mc_mb_var, s->mb_num * sizeof(int16_t))
            CHECKED_ALLOCZ(pic->mb_mean  , s->mb_num * sizeof(int8_t))
266
            CHECKED_ALLOCZ(pic->mb_cmp_score, s->mb_num * sizeof(int32_t))
M
Michael Niedermayer 已提交
267
        }
M
cleanup  
Michael Niedermayer 已提交
268

269 270
        CHECKED_ALLOCZ(pic->mbskip_table , s->mb_num * sizeof(uint8_t)+1) //the +1 is for the slice end check
        CHECKED_ALLOCZ(pic->qscale_table , s->mb_num * sizeof(uint8_t))
M
Michael Niedermayer 已提交
271 272
        pic->qstride= s->mb_width;
    }
M
cleanup  
Michael Niedermayer 已提交
273
    
274 275 276 277 278 279
    //it might be nicer if the application would keep track of these but it would require a API change
    memmove(s->prev_pict_types+1, s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE-1);
    s->prev_pict_types[0]= s->pict_type;
    if(pic->age < PREV_PICT_TYPES_BUFFER_SIZE && s->prev_pict_types[pic->age] == B_TYPE)
        pic->age= INT_MAX; // skiped MBs in b frames are quite rare in mpeg1/2 and its a bit tricky to skip them anyway
    
M
cleanup  
Michael Niedermayer 已提交
280 281 282 283 284
    return 0;
fail: //for the CHECKED_ALLOCZ macro
    return -1;
}

M
Michael Niedermayer 已提交
285 286 287
/**
 * deallocates a picture
 */
M
cleanup  
Michael Niedermayer 已提交
288 289
static void free_picture(MpegEncContext *s, Picture *pic){
    int i;
M
Michael Niedermayer 已提交
290 291

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

M
cleanup  
Michael Niedermayer 已提交
295 296 297
    av_freep(&pic->mb_var);
    av_freep(&pic->mc_mb_var);
    av_freep(&pic->mb_mean);
298
    av_freep(&pic->mb_cmp_score);
M
cleanup  
Michael Niedermayer 已提交
299 300 301
    av_freep(&pic->mbskip_table);
    av_freep(&pic->qscale_table);
    
M
Michael Niedermayer 已提交
302
    if(pic->type == FF_BUFFER_TYPE_INTERNAL){
M
cleanup  
Michael Niedermayer 已提交
303 304 305 306 307
        for(i=0; i<4; i++){
            av_freep(&pic->base[i]);
            pic->data[i]= NULL;
        }
        av_freep(&pic->opaque);
M
Michael Niedermayer 已提交
308 309 310 311 312 313 314
        pic->type= 0;
    }else if(pic->type == FF_BUFFER_TYPE_SHARED){
        for(i=0; i<4; i++){
            pic->base[i]=
            pic->data[i]= NULL;
        }
        pic->type= 0;        
M
cleanup  
Michael Niedermayer 已提交
315 316 317
    }
}

318 319 320
/* init common structure for both encoder and decoder */
int MPV_common_init(MpegEncContext *s)
{
321
    int y_size, c_size, yc_size, i;
322

323
    dsputil_init(&s->dsp, s->avctx);
324
    DCT_common_init(s);
325

326
    s->flags= s->avctx->flags;
327

M
cleanup  
Michael Niedermayer 已提交
328
    s->mb_width  = (s->width  + 15) / 16;
F
Fabrice Bellard 已提交
329
    s->mb_height = (s->height + 15) / 16;
330

331 332 333
    /* 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;
334 335 336 337 338 339 340

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

    y_size = (2 * s->mb_width + 2) * (2 * s->mb_height + 2);
    c_size = (s->mb_width + 2) * (s->mb_height + 2);
    yc_size = y_size + 2 * c_size;

341
    /* convert fourcc to upper case */
342 343 344 345
    s->avctx->codec_tag=   toupper( s->avctx->codec_tag     &0xFF)          
                        + (toupper((s->avctx->codec_tag>>8 )&0xFF)<<8 )
                        + (toupper((s->avctx->codec_tag>>16)&0xFF)<<16) 
                        + (toupper((s->avctx->codec_tag>>24)&0xFF)<<24);
346

347 348
    CHECKED_ALLOCZ(s->allocated_edge_emu_buffer, (s->width+64)*2*17*2); //(width + edge + align)*interlaced*MBsize*tolerance
    s->edge_emu_buffer= s->allocated_edge_emu_buffer + (s->width+64)*2*17;
M
cleanup  
Michael Niedermayer 已提交
349

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

352
    if (s->encoding) {
353 354 355
        int mv_table_size= (s->mb_width+2)*(s->mb_height+2);

        /* Allocate MV tables */
356 357 358 359 360 361
        CHECKED_ALLOCZ(s->p_mv_table            , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_forw_mv_table       , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_back_mv_table       , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_bidir_forw_mv_table , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_bidir_back_mv_table , mv_table_size * 2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->b_direct_mv_table     , mv_table_size * 2 * sizeof(int16_t))
362

M
cleanup  
Michael Niedermayer 已提交
363
        //FIXME should be linesize instead of s->width*2 but that isnt known before get_buffer()
M
Michael Niedermayer 已提交
364
        CHECKED_ALLOCZ(s->me.scratchpad,  s->width*2*16*3*sizeof(uint8_t)) 
365
        
M
Michael Niedermayer 已提交
366 367
        CHECKED_ALLOCZ(s->me.map      , ME_MAP_SIZE*sizeof(uint32_t))
        CHECKED_ALLOCZ(s->me.score_map, ME_MAP_SIZE*sizeof(uint32_t))
368

369 370 371 372
        if(s->codec_id==CODEC_ID_MPEG4){
            CHECKED_ALLOCZ(s->tex_pb_buffer, PB_BUFFER_SIZE);
            CHECKED_ALLOCZ(   s->pb2_buffer, PB_BUFFER_SIZE);
        }
M
Michael Niedermayer 已提交
373
        
374 375 376
        if(s->msmpeg4_version){
            CHECKED_ALLOCZ(s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int));
        }
M
Michael Niedermayer 已提交
377
        CHECKED_ALLOCZ(s->avctx->stats_out, 256);
378
    }
379
        
380
    CHECKED_ALLOCZ(s->error_status_table, s->mb_num*sizeof(uint8_t))
381
    
382
    if (s->out_format == FMT_H263 || s->encoding) {
F
Fabrice Bellard 已提交
383
        int size;
384
        /* Allocate MB type table */
385
        CHECKED_ALLOCZ(s->mb_type  , s->mb_num * sizeof(uint8_t))
386

F
Fabrice Bellard 已提交
387 388
        /* MV prediction */
        size = (2 * s->mb_width + 2) * (2 * s->mb_height + 2);
389
        CHECKED_ALLOCZ(s->motion_val, size * 2 * sizeof(int16_t));
390 391 392
    }

    if(s->codec_id==CODEC_ID_MPEG4){
393
        /* interlaced direct mode decoding tables */
394 395
        CHECKED_ALLOCZ(s->field_mv_table, s->mb_num*2*2 * sizeof(int16_t))
        CHECKED_ALLOCZ(s->field_select_table, s->mb_num*2* sizeof(int8_t))
F
Fabrice Bellard 已提交
396
    }
397 398
    /* 4mv b frame decoding table */
    //note this is needed for h263 without b frames too (segfault on damaged streams otherwise)
399
    CHECKED_ALLOCZ(s->co_located_type_table, s->mb_num * sizeof(uint8_t))
M
Michael Niedermayer 已提交
400
    if (s->out_format == FMT_H263) {
F
Fabrice Bellard 已提交
401
        /* ac values */
402
        CHECKED_ALLOCZ(s->ac_val[0], yc_size * sizeof(int16_t) * 16);
F
Fabrice Bellard 已提交
403 404 405 406
        s->ac_val[1] = s->ac_val[0] + y_size;
        s->ac_val[2] = s->ac_val[1] + c_size;
        
        /* cbp values */
407
        CHECKED_ALLOCZ(s->coded_block, y_size);
M
Michael Niedermayer 已提交
408 409
        
        /* divx501 bitstream reorder buffer */
410
        CHECKED_ALLOCZ(s->bitstream_buffer, BITSTREAM_BUFFER_SIZE);
M
cleanup  
Michael Niedermayer 已提交
411

412
        /* cbp, ac_pred, pred_dir */
413 414
        CHECKED_ALLOCZ(s->cbp_table  , s->mb_num * sizeof(uint8_t))
        CHECKED_ALLOCZ(s->pred_dir_table, s->mb_num * sizeof(uint8_t))
415
    }
416 417 418 419
    
    if (s->h263_pred || s->h263_plus || !s->encoding) {
        /* dc values */
        //MN: we need these for error resilience of intra-frames
420
        CHECKED_ALLOCZ(s->dc_val[0], yc_size * sizeof(int16_t));
421 422 423 424 425 426
        s->dc_val[1] = s->dc_val[0] + y_size;
        s->dc_val[2] = s->dc_val[1] + c_size;
        for(i=0;i<yc_size;i++)
            s->dc_val[0][i] = 1024;
    }

427 428 429 430
    /* which mb is a intra block */
    CHECKED_ALLOCZ(s->mbintra_table, s->mb_num);
    memset(s->mbintra_table, 1, s->mb_num);
    
F
Fabrice Bellard 已提交
431 432
    /* default structure is frame */
    s->picture_structure = PICT_FRAME;
433
    
434
    /* init macroblock skip table */
435 436
    CHECKED_ALLOCZ(s->mbskip_table, s->mb_num+1);
    //Note the +1 is for a quicker mpeg4 slice_end detection
437
    CHECKED_ALLOCZ(s->prev_pict_types, PREV_PICT_TYPES_BUFFER_SIZE);
438
    
439
    s->block= s->blocks[0];
440

441 442
    s->parse_context.state= -1;

F
Fabrice Bellard 已提交
443 444 445
    s->context_initialized = 1;
    return 0;
 fail:
446
    MPV_common_end(s);
F
Fabrice Bellard 已提交
447 448 449
    return -1;
}

450 451 452

//extern int sads;

F
Fabrice Bellard 已提交
453 454 455 456 457
/* init common structure for both encoder and decoder */
void MPV_common_end(MpegEncContext *s)
{
    int i;

458 459 460 461 462 463 464 465 466 467 468 469
    av_freep(&s->mb_type);
    av_freep(&s->p_mv_table);
    av_freep(&s->b_forw_mv_table);
    av_freep(&s->b_back_mv_table);
    av_freep(&s->b_bidir_forw_mv_table);
    av_freep(&s->b_bidir_back_mv_table);
    av_freep(&s->b_direct_mv_table);
    av_freep(&s->motion_val);
    av_freep(&s->dc_val[0]);
    av_freep(&s->ac_val[0]);
    av_freep(&s->coded_block);
    av_freep(&s->mbintra_table);
470 471
    av_freep(&s->cbp_table);
    av_freep(&s->pred_dir_table);
M
Michael Niedermayer 已提交
472 473 474
    av_freep(&s->me.scratchpad);
    av_freep(&s->me.map);
    av_freep(&s->me.score_map);
475
    
476
    av_freep(&s->mbskip_table);
477
    av_freep(&s->prev_pict_types);
478
    av_freep(&s->bitstream_buffer);
479 480
    av_freep(&s->tex_pb_buffer);
    av_freep(&s->pb2_buffer);
481
    av_freep(&s->allocated_edge_emu_buffer); s->edge_emu_buffer= NULL;
482 483 484
    av_freep(&s->co_located_type_table);
    av_freep(&s->field_mv_table);
    av_freep(&s->field_select_table);
M
Michael Niedermayer 已提交
485
    av_freep(&s->avctx->stats_out);
486
    av_freep(&s->ac_stats);
487
    av_freep(&s->error_status_table);
M
cleanup  
Michael Niedermayer 已提交
488 489 490

    for(i=0; i<MAX_PICTURE_COUNT; i++){
        free_picture(s, &s->picture[i]);
F
Fabrice Bellard 已提交
491 492 493 494
    }
    s->context_initialized = 0;
}

495 496
#ifdef CONFIG_ENCODERS

F
Fabrice Bellard 已提交
497 498 499 500
/* init video encoder */
int MPV_encode_init(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;
501
    int i;
F
Fabrice Bellard 已提交
502

503 504
    avctx->pix_fmt = PIX_FMT_YUV420P;

F
Fabrice Bellard 已提交
505
    s->bit_rate = avctx->bit_rate;
506
    s->bit_rate_tolerance = avctx->bit_rate_tolerance;
F
Fabrice Bellard 已提交
507 508
    s->width = avctx->width;
    s->height = avctx->height;
509
    if(avctx->gop_size > 600){
510
        fprintf(stderr, "Warning keyframe interval too large! reducing it ...\n");
511 512
        avctx->gop_size=600;
    }
F
Fabrice Bellard 已提交
513
    s->gop_size = avctx->gop_size;
514 515
    s->rtp_mode = avctx->rtp_mode;
    s->rtp_payload_size = avctx->rtp_payload_size;
516 517
    if (avctx->rtp_callback)
        s->rtp_callback = avctx->rtp_callback;
518 519 520
    s->max_qdiff= avctx->max_qdiff;
    s->qcompress= avctx->qcompress;
    s->qblur= avctx->qblur;
521
    s->avctx = avctx;
522
    s->flags= avctx->flags;
523
    s->max_b_frames= avctx->max_b_frames;
524
    s->b_frame_strategy= avctx->b_frame_strategy;
525
    s->codec_id= avctx->codec->id;
526 527 528 529
    s->luma_elim_threshold  = avctx->luma_elim_threshold;
    s->chroma_elim_threshold= avctx->chroma_elim_threshold;
    s->strict_std_compliance= avctx->strict_std_compliance;
    s->data_partitioning= avctx->flags & CODEC_FLAG_PART;
M
Michael Niedermayer 已提交
530
    s->quarter_sample= (avctx->flags & CODEC_FLAG_QPEL)!=0;
531
    s->mpeg_quant= avctx->mpeg_quant;
532

F
Fabrice Bellard 已提交
533 534 535 536 537 538
    if (s->gop_size <= 1) {
        s->intra_only = 1;
        s->gop_size = 12;
    } else {
        s->intra_only = 0;
    }
539

M
Michael Niedermayer 已提交
540
    s->me_method = avctx->me_method;
541

542
    /* Fixed QSCALE */
F
Fabrice Bellard 已提交
543
    s->fixed_qscale = (avctx->flags & CODEC_FLAG_QSCALE);
544
    
545
    s->adaptive_quant= (   s->avctx->lumi_masking
546
                        || s->avctx->dark_masking
547 548 549 550
                        || s->avctx->temporal_cplx_masking 
                        || s->avctx->spatial_cplx_masking
                        || s->avctx->p_masking)
                       && !s->fixed_qscale;
551 552
    
    s->progressive_sequence= !(avctx->flags & CODEC_FLAG_INTERLACED_DCT);
553

554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569
    if(s->codec_id==CODEC_ID_MJPEG){
        s->intra_quant_bias= 1<<(QUANT_BIAS_SHIFT-1); //(a + x/2)/x
        s->inter_quant_bias= 0;
    }else if(s->mpeg_quant || s->codec_id==CODEC_ID_MPEG1VIDEO){
        s->intra_quant_bias= 3<<(QUANT_BIAS_SHIFT-3); //(a + x*3/8)/x
        s->inter_quant_bias= 0;
    }else{
        s->intra_quant_bias=0;
        s->inter_quant_bias=-(1<<(QUANT_BIAS_SHIFT-2)); //(a - x/4)/x
    }
    
    if(avctx->intra_quant_bias != FF_DEFAULT_QUANT_BIAS)
        s->intra_quant_bias= avctx->intra_quant_bias;
    if(avctx->inter_quant_bias != FF_DEFAULT_QUANT_BIAS)
        s->inter_quant_bias= avctx->inter_quant_bias;
    
F
Fabrice Bellard 已提交
570 571 572
    switch(avctx->codec->id) {
    case CODEC_ID_MPEG1VIDEO:
        s->out_format = FMT_MPEG1;
573 574
        s->low_delay= 0; //s->max_b_frames ? 0 : 1;
        avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1);
F
Fabrice Bellard 已提交
575 576 577 578
        break;
    case CODEC_ID_MJPEG:
        s->out_format = FMT_MJPEG;
        s->intra_only = 1; /* force intra only for jpeg */
579
        s->mjpeg_write_tables = 1; /* write all tables */
580
	s->mjpeg_data_only_frames = 0; /* write all the needed headers */
581 582 583
        s->mjpeg_vsample[0] = 2; /* set up default sampling factors */
        s->mjpeg_vsample[1] = 1; /* the only currently supported values */
        s->mjpeg_vsample[2] = 1; 
584
        s->mjpeg_hsample[0] = 2;
585 586
        s->mjpeg_hsample[1] = 1; 
        s->mjpeg_hsample[2] = 1; 
F
Fabrice Bellard 已提交
587 588
        if (mjpeg_init(s) < 0)
            return -1;
M
Michael Niedermayer 已提交
589
        avctx->delay=0;
M
Michael Niedermayer 已提交
590
        s->low_delay=1;
F
Fabrice Bellard 已提交
591
        break;
M
Michael Niedermayer 已提交
592
#ifdef CONFIG_RISKY
F
Fabrice Bellard 已提交
593
    case CODEC_ID_H263:
594 595
        if (h263_get_picture_format(s->width, s->height) == 7) {
            printf("Input picture size isn't suitable for h263 codec! try h263+\n");
F
Fabrice Bellard 已提交
596
            return -1;
597
        }
F
Fabrice Bellard 已提交
598
        s->out_format = FMT_H263;
M
Michael Niedermayer 已提交
599
        avctx->delay=0;
M
Michael Niedermayer 已提交
600
        s->low_delay=1;
F
Fabrice Bellard 已提交
601 602 603 604
        break;
    case CODEC_ID_H263P:
        s->out_format = FMT_H263;
        s->h263_plus = 1;
605 606 607 608
	/* Fx */
	s->unrestricted_mv=(avctx->flags & CODEC_FLAG_H263P_UMV) ? 1:0;
	s->h263_aic= (avctx->flags & CODEC_FLAG_H263P_AIC) ? 1:0;
	/* /Fx */
609
        /* These are just to be sure */
M
Michael Niedermayer 已提交
610
        s->umvplus = 1;
M
Michael Niedermayer 已提交
611
        avctx->delay=0;
M
Michael Niedermayer 已提交
612
        s->low_delay=1;
F
Fabrice Bellard 已提交
613 614 615 616
        break;
    case CODEC_ID_RV10:
        s->out_format = FMT_H263;
        s->h263_rv10 = 1;
M
Michael Niedermayer 已提交
617
        avctx->delay=0;
M
Michael Niedermayer 已提交
618
        s->low_delay=1;
F
Fabrice Bellard 已提交
619
        break;
F
Fabrice Bellard 已提交
620
    case CODEC_ID_MPEG4:
F
Fabrice Bellard 已提交
621 622 623
        s->out_format = FMT_H263;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
M
Michael Niedermayer 已提交
624
        s->low_delay= s->max_b_frames ? 0 : 1;
625
        avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1);
F
Fabrice Bellard 已提交
626
        break;
627
    case CODEC_ID_MSMPEG4V1:
F
Fabrice Bellard 已提交
628 629 630 631
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
632
        s->msmpeg4_version= 1;
M
Michael Niedermayer 已提交
633
        avctx->delay=0;
M
Michael Niedermayer 已提交
634
        s->low_delay=1;
635 636 637 638 639 640 641
        break;
    case CODEC_ID_MSMPEG4V2:
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
        s->msmpeg4_version= 2;
M
Michael Niedermayer 已提交
642
        avctx->delay=0;
M
Michael Niedermayer 已提交
643
        s->low_delay=1;
644 645 646 647 648 649 650
        break;
    case CODEC_ID_MSMPEG4V3:
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
        s->msmpeg4_version= 3;
M
Michael Niedermayer 已提交
651
        avctx->delay=0;
M
Michael Niedermayer 已提交
652
        s->low_delay=1;
F
Fabrice Bellard 已提交
653
        break;
M
Michael Niedermayer 已提交
654 655 656 657 658 659 660
    case CODEC_ID_WMV1:
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
        s->msmpeg4_version= 4;
        avctx->delay=0;
M
Michael Niedermayer 已提交
661
        s->low_delay=1;
M
Michael Niedermayer 已提交
662 663 664 665 666 667 668 669
        break;
    case CODEC_ID_WMV2:
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
        s->msmpeg4_version= 5;
        avctx->delay=0;
M
Michael Niedermayer 已提交
670
        s->low_delay=1;
M
Michael Niedermayer 已提交
671
        break;
M
Michael Niedermayer 已提交
672
#endif
F
Fabrice Bellard 已提交
673 674 675
    default:
        return -1;
    }
M
Michael Niedermayer 已提交
676
    
677 678 679 680 681
    { /* set up some save defaults, some codecs might override them later */
        static int done=0;
        if(!done){
            int i;
            done=1;
682

683 684
            default_mv_penalty= av_mallocz( sizeof(uint8_t)*(MAX_FCODE+1)*(2*MAX_MV+1) );
            memset(default_mv_penalty, 0, sizeof(uint8_t)*(MAX_FCODE+1)*(2*MAX_MV+1));
685
            memset(default_fcode_tab , 0, sizeof(uint8_t)*(2*MAX_MV+1));
686 687 688 689 690 691

            for(i=-16; i<16; i++){
                default_fcode_tab[i + MAX_MV]= 1;
            }
        }
    }
M
Michael Niedermayer 已提交
692
    s->me.mv_penalty= default_mv_penalty;
693
    s->fcode_tab= default_fcode_tab;
694 695 696
    s->y_dc_scale_table=
    s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
 
697
    /* dont use mv_penalty table for crap MV as it would be confused */
M
Michael Niedermayer 已提交
698 699
    //FIXME remove after fixing / removing old ME
    if (s->me_method < ME_EPZS) s->me.mv_penalty = default_mv_penalty;
700

701 702
    s->encoding = 1;

F
Fabrice Bellard 已提交
703 704 705 706
    /* init */
    if (MPV_common_init(s) < 0)
        return -1;
    
M
Michael Niedermayer 已提交
707 708
    ff_init_me(s);

709
#ifdef CONFIG_ENCODERS
M
Michael Niedermayer 已提交
710
#ifdef CONFIG_RISKY
711 712 713 714
    if (s->out_format == FMT_H263)
        h263_encode_init(s);
    if(s->msmpeg4_version)
        ff_msmpeg4_encode_init(s);
715
#endif
M
Michael Niedermayer 已提交
716 717 718
    if (s->out_format == FMT_MPEG1)
        ff_mpeg1_encode_init(s);
#endif
719

720 721
    /* init default q matrix */
    for(i=0;i<64;i++) {
722
        int j= s->dsp.idct_permutation[i];
M
Michael Niedermayer 已提交
723
#ifdef CONFIG_RISKY
724
        if(s->codec_id==CODEC_ID_MPEG4 && s->mpeg_quant){
725 726
            s->intra_matrix[j] = ff_mpeg4_default_intra_matrix[i];
            s->inter_matrix[j] = ff_mpeg4_default_non_intra_matrix[i];
727
        }else if(s->out_format == FMT_H263){
728 729
            s->intra_matrix[j] =
            s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
M
Michael Niedermayer 已提交
730 731 732
        }else
#endif
        { /* mpeg1 */
733 734
            s->intra_matrix[j] = ff_mpeg1_default_intra_matrix[i];
            s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
735
        }
736 737 738
    }

    /* precompute matrix */
739
    /* for mjpeg, we do include qscale in the matrix */
740
    if (s->out_format != FMT_MJPEG) {
741
        convert_matrix(s, s->q_intra_matrix, s->q_intra_matrix16, s->q_intra_matrix16_bias, 
742
                       s->intra_matrix, s->intra_quant_bias, 1, 31);
743
        convert_matrix(s, s->q_inter_matrix, s->q_inter_matrix16, s->q_inter_matrix16_bias, 
744
                       s->inter_matrix, s->inter_quant_bias, 1, 31);
745 746
    }

747 748
    if(ff_rate_control_init(s) < 0)
        return -1;
F
Fabrice Bellard 已提交
749 750

    s->picture_number = 0;
751
    s->picture_in_gop_number = 0;
F
Fabrice Bellard 已提交
752 753 754
    s->fake_picture_number = 0;
    /* motion detector init */
    s->f_code = 1;
755
    s->b_code = 1;
F
Fabrice Bellard 已提交
756 757 758 759 760 761 762 763 764 765 766

    return 0;
}

int MPV_encode_end(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;

#ifdef STATS
    print_stats();
#endif
767 768 769

    ff_rate_control_uninit(s);

F
Fabrice Bellard 已提交
770 771 772
    MPV_common_end(s);
    if (s->out_format == FMT_MJPEG)
        mjpeg_close(s);
773
      
F
Fabrice Bellard 已提交
774 775 776
    return 0;
}

777 778
#endif //CONFIG_ENCODERS

M
Michael Niedermayer 已提交
779 780
void init_rl(RLTable *rl)
{
781 782
    int8_t max_level[MAX_RUN+1], max_run[MAX_LEVEL+1];
    uint8_t index_run[MAX_RUN+1];
M
Michael Niedermayer 已提交
783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816
    int last, run, level, start, end, i;

    /* 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;
        }
        rl->max_level[last] = av_malloc(MAX_RUN + 1);
        memcpy(rl->max_level[last], max_level, MAX_RUN + 1);
        rl->max_run[last] = av_malloc(MAX_LEVEL + 1);
        memcpy(rl->max_run[last], max_run, MAX_LEVEL + 1);
        rl->index_run[last] = av_malloc(MAX_RUN + 1);
        memcpy(rl->index_run[last], index_run, MAX_RUN + 1);
    }
}

F
Fabrice Bellard 已提交
817
/* draw the edges of width 'w' of an image of size width, height */
818
//FIXME check that this is ok for mpeg4 interlaced
819
static void draw_edges_c(uint8_t *buf, int wrap, int width, int height, int w)
F
Fabrice Bellard 已提交
820
{
821
    uint8_t *ptr, *last_line;
F
Fabrice Bellard 已提交
822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
    int i;

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

M
Michael Niedermayer 已提交
846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
static int find_unused_picture(MpegEncContext *s, int shared){
    int i;
    
    if(shared){
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            if(s->picture[i].data[0]==NULL && s->picture[i].type==0) break;
        }
    }else{
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            if(s->picture[i].data[0]==NULL && s->picture[i].type!=0) break;
        }
        for(i=0; i<MAX_PICTURE_COUNT; i++){
            if(s->picture[i].data[0]==NULL) break;
        }
    }

    assert(i<MAX_PICTURE_COUNT);
    return i;
}

F
Fabrice Bellard 已提交
866
/* generic function for encode/decode called before a frame is coded/decoded */
867
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
F
Fabrice Bellard 已提交
868
{
M
Michael Niedermayer 已提交
869
    int i;
M
Michael Niedermayer 已提交
870
    AVFrame *pic;
F
Fabrice Bellard 已提交
871

872
    s->mb_skiped = 0;
M
cleanup  
Michael Niedermayer 已提交
873 874
    
    /* mark&release old frames */
M
Michael Niedermayer 已提交
875 876
    if (s->pict_type != B_TYPE && s->last_picture_ptr) {
        avctx->release_buffer(avctx, (AVFrame*)s->last_picture_ptr);
M
cleanup  
Michael Niedermayer 已提交
877 878 879 880 881

        /* release forgotten pictures */
        /* if(mpeg124/h263) */
        if(!s->encoding){
            for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
882
                if(s->picture[i].data[0] && &s->picture[i] != s->next_picture_ptr && s->picture[i].reference){
M
cleanup  
Michael Niedermayer 已提交
883
                    fprintf(stderr, "releasing zombie picture\n");
M
Michael Niedermayer 已提交
884
                    avctx->release_buffer(avctx, (AVFrame*)&s->picture[i]);                
M
cleanup  
Michael Niedermayer 已提交
885 886
                }
            }
887
        }
888
    }
M
Michael Niedermayer 已提交
889
    
890 891
alloc:
    if(!s->encoding){
M
Michael Niedermayer 已提交
892
        i= find_unused_picture(s, 0);
M
cleanup  
Michael Niedermayer 已提交
893
    
M
Michael Niedermayer 已提交
894
        pic= (AVFrame*)&s->picture[i];
M
cleanup  
Michael Niedermayer 已提交
895
        pic->reference= s->pict_type != B_TYPE;
M
Michael Niedermayer 已提交
896 897 898

        if(s->current_picture_ptr)
            pic->coded_picture_number= s->current_picture_ptr->coded_picture_number+1;
M
cleanup  
Michael Niedermayer 已提交
899
        
M
Michael Niedermayer 已提交
900
        alloc_picture(s, (Picture*)pic, 0);
901

M
Michael Niedermayer 已提交
902
        s->current_picture_ptr= &s->picture[i];
M
cleanup  
Michael Niedermayer 已提交
903
    }
904

M
cleanup  
Michael Niedermayer 已提交
905
    if (s->pict_type != B_TYPE) {
M
Michael Niedermayer 已提交
906 907
        s->last_picture_ptr= s->next_picture_ptr;
        s->next_picture_ptr= s->current_picture_ptr;
F
Fabrice Bellard 已提交
908
    }
M
Michael Niedermayer 已提交
909 910 911 912
    s->current_picture= *s->current_picture_ptr;
    if(s->last_picture_ptr) s->last_picture= *s->last_picture_ptr;
    if(s->next_picture_ptr) s->next_picture= *s->next_picture_ptr;
    if(s->new_picture_ptr ) s->new_picture = *s->new_picture_ptr;
913
    
M
Michael Niedermayer 已提交
914 915 916 917 918 919 920 921 922 923 924 925 926
    if(s->picture_structure!=PICT_FRAME){
        int i;
        for(i=0; i<4; i++){
            if(s->picture_structure == PICT_BOTTOM_FIELD){
                 s->current_picture.data[i] += s->current_picture.linesize[i];
            } 
            s->current_picture.linesize[i] *= 2;
            s->last_picture.linesize[i] *=2;
            s->next_picture.linesize[i] *=2;
        }
    }
    
    if(s->pict_type != I_TYPE && s->last_picture_ptr==NULL){
927 928 929 930
        fprintf(stderr, "warning: first frame is no keyframe\n");
        assert(s->pict_type != B_TYPE); //these should have been dropped if we dont have a reference
        goto alloc;
    }
M
cleanup  
Michael Niedermayer 已提交
931
   
932 933 934
    s->hurry_up= s->avctx->hurry_up;
    s->error_resilience= avctx->error_resilience;

935 936 937 938 939 940 941 942 943
    /* set dequantizer, we cant do it during init as it might change for mpeg4
       and we cant do it in the header decode as init isnt called for mpeg4 there yet */
    if(s->out_format == FMT_H263){
        if(s->mpeg_quant)
            s->dct_unquantize = s->dct_unquantize_mpeg2;
        else
            s->dct_unquantize = s->dct_unquantize_h263;
    }else 
        s->dct_unquantize = s->dct_unquantize_mpeg1;
944 945

    return 0;
F
Fabrice Bellard 已提交
946
}
947

F
Fabrice Bellard 已提交
948 949 950
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
M
cleanup  
Michael Niedermayer 已提交
951
    int i;
F
Fabrice Bellard 已提交
952
    /* draw edge for correct motion prediction if outside */
M
cleanup  
Michael Niedermayer 已提交
953 954 955 956 957 958
    if(s->codec_id!=CODEC_ID_SVQ1){
        if (s->pict_type != B_TYPE && !s->intra_only && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
            draw_edges(s->current_picture.data[0], s->linesize  , s->h_edge_pos   , s->v_edge_pos   , EDGE_WIDTH  );
            draw_edges(s->current_picture.data[1], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
            draw_edges(s->current_picture.data[2], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
        }
F
Fabrice Bellard 已提交
959
    }
M
Michael Niedermayer 已提交
960
    emms_c();
961
    
M
Michael Niedermayer 已提交
962
    s->last_pict_type    = s->pict_type;
963 964 965
    if(s->pict_type!=B_TYPE){
        s->last_non_b_pict_type= s->pict_type;
    }
M
Michael Niedermayer 已提交
966 967
#if 0
        /* copy back current_picture variables */
M
cleanup  
Michael Niedermayer 已提交
968 969 970 971 972 973 974
    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;
        }    
    }
    assert(i<MAX_PICTURE_COUNT);
M
Michael Niedermayer 已提交
975 976 977 978
#endif    
    s->current_picture_ptr->quality= s->qscale; //FIXME get average of qscale_table
    s->current_picture_ptr->pict_type= s->pict_type;
    s->current_picture_ptr->key_frame= s->pict_type == I_TYPE;
M
cleanup  
Michael Niedermayer 已提交
979 980 981

    /* release non refernce frames */
    for(i=0; i<MAX_PICTURE_COUNT; i++){
M
Michael Niedermayer 已提交
982
        if(s->picture[i].data[0] && !s->picture[i].reference /*&& s->picture[i].type!=FF_BUFFER_TYPE_SHARED*/)
M
Michael Niedermayer 已提交
983
            s->avctx->release_buffer(s->avctx, (AVFrame*)&s->picture[i]);
M
cleanup  
Michael Niedermayer 已提交
984
    }
M
Michael Niedermayer 已提交
985 986 987 988 989 990 991 992 993 994 995 996
    if(s->avctx->debug&FF_DEBUG_SKIP){
        int x,y;        
        for(y=0; y<s->mb_height; y++){
            for(x=0; x<s->mb_width; x++){
                int count= s->mbskip_table[x + y*s->mb_width];
                if(count>9) count=9;
                printf(" %1d", count);
            }
            printf("\n");
        }
        printf("pict type: %d\n", s->pict_type);
    }
M
Michael Niedermayer 已提交
997 998 999 1000 1001 1002 1003

    // 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
F
Fabrice Bellard 已提交
1004 1005
}

1006 1007
#ifdef CONFIG_ENCODERS

M
Michael Niedermayer 已提交
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
static int get_sae(uint8_t *src, int ref, int stride){
    int x,y;
    int acc=0;
    
    for(y=0; y<16; y++){
        for(x=0; x<16; x++){
            acc+= ABS(src[x+y*stride] - ref);
        }
    }
    
    return acc;
}

static int get_intra_count(MpegEncContext *s, uint8_t *src, uint8_t *ref, int stride){
    int x, y, w, h;
    int acc=0;
    
    w= s->width &~15;
    h= s->height&~15;
    
    for(y=0; y<h; y+=16){
        for(x=0; x<w; x+=16){
            int offset= x + y*stride;
            int sad = s->dsp.pix_abs16x16(src + offset, ref + offset, stride);
            int mean= (s->dsp.pix_sum(src + offset, stride) + 128)>>8;
            int sae = get_sae(src + offset, mean, stride);
            
            acc+= sae + 500 < sad;
        }
    }
    return acc;
}

M
Michael Niedermayer 已提交
1041

M
Michael Niedermayer 已提交
1042 1043
static int load_input_picture(MpegEncContext *s, AVFrame *pic_arg){
    AVFrame *pic;
M
Michael Niedermayer 已提交
1044
    int i;
M
cleanup  
Michael Niedermayer 已提交
1045
    const int encoding_delay= s->max_b_frames;
M
Michael Niedermayer 已提交
1046
    int direct=1;
M
cleanup  
Michael Niedermayer 已提交
1047

M
Michael Niedermayer 已提交
1048 1049 1050 1051 1052 1053
    if(encoding_delay && !(s->flags&CODEC_FLAG_INPUT_PRESERVED)) direct=0;
    if(pic_arg->linesize[0] != s->linesize) direct=0;
    if(pic_arg->linesize[1] != s->uvlinesize) direct=0;
    if(pic_arg->linesize[2] != s->uvlinesize) direct=0;
  
//    printf("%d %d %d %d\n",pic_arg->linesize[0], pic_arg->linesize[1], s->linesize, s->uvlinesize);
M
cleanup  
Michael Niedermayer 已提交
1054
    
M
Michael Niedermayer 已提交
1055 1056
    if(direct){
        i= find_unused_picture(s, 1);
M
cleanup  
Michael Niedermayer 已提交
1057

M
Michael Niedermayer 已提交
1058
        pic= (AVFrame*)&s->picture[i];
M
Michael Niedermayer 已提交
1059
        pic->reference= 1;
M
cleanup  
Michael Niedermayer 已提交
1060
    
M
Michael Niedermayer 已提交
1061 1062 1063 1064 1065 1066 1067
        for(i=0; i<4; i++){
            pic->data[i]= pic_arg->data[i];
            pic->linesize[i]= pic_arg->linesize[i];
        }
        alloc_picture(s, (Picture*)pic, 1);
    }else{
        i= find_unused_picture(s, 0);
M
cleanup  
Michael Niedermayer 已提交
1068

M
Michael Niedermayer 已提交
1069
        pic= (AVFrame*)&s->picture[i];
M
Michael Niedermayer 已提交
1070
        pic->reference= 1;
M
cleanup  
Michael Niedermayer 已提交
1071

M
Michael Niedermayer 已提交
1072
        alloc_picture(s, (Picture*)pic, 0);
M
Michael Niedermayer 已提交
1073 1074 1075 1076 1077 1078
        for(i=0; i<4; i++){
            /* the input will be 16 pixels to the right relative to the actual buffer start
             * and the current_pic, so the buffer can be reused, yes its not beatifull 
             */
            pic->data[i]+= 16; 
        }
M
Michael Niedermayer 已提交
1079 1080 1081 1082

        if(   pic->data[0] == pic_arg->data[0] 
           && pic->data[1] == pic_arg->data[1]
           && pic->data[2] == pic_arg->data[2]){
M
cleanup  
Michael Niedermayer 已提交
1083
       // empty
M
Michael Niedermayer 已提交
1084 1085
        }else{
            int h_chroma_shift, v_chroma_shift;
M
cleanup  
Michael Niedermayer 已提交
1086
        
M
Michael Niedermayer 已提交
1087
            avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
M
cleanup  
Michael Niedermayer 已提交
1088
        
M
Michael Niedermayer 已提交
1089 1090 1091 1092 1093 1094 1095 1096 1097
            for(i=0; i<3; i++){
                int src_stride= pic_arg->linesize[i];
                int dst_stride= i ? s->uvlinesize : s->linesize;
                int h_shift= i ? h_chroma_shift : 0;
                int v_shift= i ? v_chroma_shift : 0;
                int w= s->width >>h_shift;
                int h= s->height>>v_shift;
                uint8_t *src= pic_arg->data[i];
                uint8_t *dst= pic->data[i];
M
cleanup  
Michael Niedermayer 已提交
1098
            
M
Michael Niedermayer 已提交
1099 1100 1101 1102 1103 1104 1105 1106
                if(src_stride==dst_stride)
                    memcpy(dst, src, src_stride*h);
                else{
                    while(h--){
                        memcpy(dst, src, w);
                        dst += dst_stride;
                        src += src_stride;
                    }
M
cleanup  
Michael Niedermayer 已提交
1107
                }
1108
            }
M
cleanup  
Michael Niedermayer 已提交
1109 1110
        }
    }
M
Michael Niedermayer 已提交
1111 1112
    pic->quality= pic_arg->quality;
    pic->pict_type= pic_arg->pict_type;
1113
    pic->pts = pic_arg->pts;
M
Michael Niedermayer 已提交
1114 1115 1116 1117 1118 1119 1120 1121 1122
    
    if(s->input_picture[encoding_delay])
        pic->display_picture_number= s->input_picture[encoding_delay]->display_picture_number + 1;

    /* shift buffer entries */
    for(i=1; i<MAX_PICTURE_COUNT /*s->encoding_delay+1*/; i++)
        s->input_picture[i-1]= s->input_picture[i];
        
    s->input_picture[encoding_delay]= (Picture*)pic;
M
cleanup  
Michael Niedermayer 已提交
1123 1124 1125

    return 0;
}
1126

M
cleanup  
Michael Niedermayer 已提交
1127 1128 1129 1130 1131 1132 1133
static void select_input_picture(MpegEncContext *s){
    int i;
    const int encoding_delay= s->max_b_frames;
    int coded_pic_num=0;    

    if(s->reordered_input_picture[0])
        coded_pic_num= s->reordered_input_picture[0]->coded_picture_number + 1;
M
Michael Niedermayer 已提交
1134

M
cleanup  
Michael Niedermayer 已提交
1135 1136 1137 1138 1139 1140
    for(i=1; i<MAX_PICTURE_COUNT; i++)
        s->reordered_input_picture[i-1]= s->reordered_input_picture[i];
    s->reordered_input_picture[MAX_PICTURE_COUNT-1]= NULL;

    /* set next picture types & ordering */
    if(s->reordered_input_picture[0]==NULL && s->input_picture[0]){
M
Michael Niedermayer 已提交
1141
        if(/*s->picture_in_gop_number >= s->gop_size ||*/ s->next_picture_ptr==NULL || s->intra_only){
M
Michael Niedermayer 已提交
1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156
            s->reordered_input_picture[0]= s->input_picture[0];
            s->reordered_input_picture[0]->pict_type= I_TYPE;
            s->reordered_input_picture[0]->coded_picture_number= coded_pic_num;
        }else{
            int b_frames;
            
            if(s->flags&CODEC_FLAG_PASS2){
                for(i=0; i<s->max_b_frames+1; i++){
                    int pict_num= s->input_picture[0]->display_picture_number + i;
                    int pict_type= s->rc_context.entry[pict_num].new_pict_type;
                    s->input_picture[i]->pict_type= pict_type;
                    
                    if(i + 1 >= s->rc_context.num_entries) break;
                }
            }
M
Michael Niedermayer 已提交
1157

M
Michael Niedermayer 已提交
1158 1159
            if(s->input_picture[0]->pict_type){
                /* user selected pict_type */
1160 1161 1162
                for(b_frames=0; b_frames<s->max_b_frames+1; b_frames++){
                    if(s->input_picture[b_frames]->pict_type!=B_TYPE) break;
                }
M
Michael Niedermayer 已提交
1163
            
1164 1165 1166 1167
                if(b_frames > s->max_b_frames){
                    fprintf(stderr, "warning, too many bframes in a row\n");
                    b_frames = s->max_b_frames;
                }
M
Michael Niedermayer 已提交
1168 1169 1170 1171 1172 1173
            }else if(s->b_frame_strategy==0){
                b_frames= s->max_b_frames;
            }else if(s->b_frame_strategy==1){
                for(i=1; i<s->max_b_frames+1; i++){
                    if(s->input_picture[i]->b_frame_score==0){
                        s->input_picture[i]->b_frame_score= 
M
Michael Niedermayer 已提交
1174 1175
                            get_intra_count(s, s->input_picture[i  ]->data[0], 
                                               s->input_picture[i-1]->data[0], s->linesize) + 1;
M
Michael Niedermayer 已提交
1176 1177 1178 1179 1180 1181 1182
                    }
                }
                for(i=0; i<s->max_b_frames; i++){
                    if(s->input_picture[i]->b_frame_score - 1 > s->mb_num/40) break;
                }
                                
                b_frames= FFMAX(0, i-1);
1183
                
M
Michael Niedermayer 已提交
1184 1185 1186 1187 1188 1189 1190
                /* reset scores */
                for(i=0; i<b_frames+1; i++){
                    s->input_picture[i]->b_frame_score=0;
                }
            }else{
                fprintf(stderr, "illegal b frame strategy\n");
                b_frames=0;
1191
            }
M
Michael Niedermayer 已提交
1192 1193 1194 1195 1196 1197 1198 1199 1200

            emms_c();
//static int b_count=0;
//b_count+= b_frames;
//printf("b_frames: %d\n", b_count);
                        
            s->reordered_input_picture[0]= s->input_picture[b_frames];
            if(   s->picture_in_gop_number + b_frames >= s->gop_size 
               || s->reordered_input_picture[0]->pict_type== I_TYPE)
M
cleanup  
Michael Niedermayer 已提交
1201
                s->reordered_input_picture[0]->pict_type= I_TYPE;
M
Michael Niedermayer 已提交
1202 1203 1204 1205 1206 1207 1208 1209
            else
                s->reordered_input_picture[0]->pict_type= P_TYPE;
            s->reordered_input_picture[0]->coded_picture_number= coded_pic_num;
            for(i=0; i<b_frames; i++){
                coded_pic_num++;
                s->reordered_input_picture[i+1]= s->input_picture[i];
                s->reordered_input_picture[i+1]->pict_type= B_TYPE;
                s->reordered_input_picture[i+1]->coded_picture_number= coded_pic_num;
1210 1211 1212
            }
        }
    }
M
cleanup  
Michael Niedermayer 已提交
1213 1214
    
    if(s->reordered_input_picture[0]){
M
Michael Niedermayer 已提交
1215 1216 1217
        s->reordered_input_picture[0]->reference= s->reordered_input_picture[0]->pict_type!=B_TYPE;

        s->new_picture= *s->reordered_input_picture[0];
M
Michael Niedermayer 已提交
1218 1219

        if(s->reordered_input_picture[0]->type == FF_BUFFER_TYPE_SHARED){
M
Michael Niedermayer 已提交
1220 1221
            // input is a shared pix, so we cant modifiy it -> alloc a new one & ensure that the shared one is reuseable
        
M
Michael Niedermayer 已提交
1222 1223 1224 1225 1226 1227 1228 1229
            int i= find_unused_picture(s, 0);
            Picture *pic= &s->picture[i];

            /* mark us unused / free shared pic */
            for(i=0; i<4; i++)
                s->reordered_input_picture[0]->data[i]= NULL;
            s->reordered_input_picture[0]->type= 0;
            
M
Michael Niedermayer 已提交
1230
            //FIXME bad, copy * except
M
Michael Niedermayer 已提交
1231 1232 1233 1234 1235 1236 1237
            pic->pict_type = s->reordered_input_picture[0]->pict_type;
            pic->quality   = s->reordered_input_picture[0]->quality;
            pic->coded_picture_number = s->reordered_input_picture[0]->coded_picture_number;
            pic->reference = s->reordered_input_picture[0]->reference;
            
            alloc_picture(s, pic, 0);

M
Michael Niedermayer 已提交
1238
            s->current_picture_ptr= pic;
M
Michael Niedermayer 已提交
1239
        }else{
M
Michael Niedermayer 已提交
1240 1241
            // input is not a shared pix -> reuse buffer for current_pix

M
Michael Niedermayer 已提交
1242 1243 1244
            assert(   s->reordered_input_picture[0]->type==FF_BUFFER_TYPE_USER 
                   || s->reordered_input_picture[0]->type==FF_BUFFER_TYPE_INTERNAL);
            
M
Michael Niedermayer 已提交
1245
            s->current_picture_ptr= s->reordered_input_picture[0];
M
Michael Niedermayer 已提交
1246
            for(i=0; i<4; i++){
M
Michael Niedermayer 已提交
1247 1248
                //reverse the +16 we did before storing the input
                s->current_picture_ptr->data[i]-=16;
M
Michael Niedermayer 已提交
1249
            }
M
cleanup  
Michael Niedermayer 已提交
1250
        }
M
Michael Niedermayer 已提交
1251
        s->current_picture= *s->current_picture_ptr;
M
cleanup  
Michael Niedermayer 已提交
1252 1253 1254 1255 1256
    
        s->picture_number= s->new_picture.display_picture_number;
//printf("dpn:%d\n", s->picture_number);
    }else{
       memset(&s->new_picture, 0, sizeof(Picture));
1257 1258 1259
    }
}

F
Fabrice Bellard 已提交
1260 1261 1262 1263
int MPV_encode_picture(AVCodecContext *avctx,
                       unsigned char *buf, int buf_size, void *data)
{
    MpegEncContext *s = avctx->priv_data;
M
Michael Niedermayer 已提交
1264
    AVFrame *pic_arg = data;
1265
    int i;
F
Fabrice Bellard 已提交
1266 1267 1268

    init_put_bits(&s->pb, buf, buf_size, NULL, NULL);

M
cleanup  
Michael Niedermayer 已提交
1269
    s->picture_in_gop_number++;
F
Fabrice Bellard 已提交
1270

M
cleanup  
Michael Niedermayer 已提交
1271
    load_input_picture(s, pic_arg);
1272
    
M
cleanup  
Michael Niedermayer 已提交
1273
    select_input_picture(s);
1274 1275
    
    /* output? */
M
cleanup  
Michael Niedermayer 已提交
1276
    if(s->new_picture.data[0]){
1277

M
cleanup  
Michael Niedermayer 已提交
1278 1279 1280 1281 1282 1283 1284
        s->pict_type= s->new_picture.pict_type;
        if (s->fixed_qscale){ /* the ratecontrol needs the last qscale so we dont touch it for CBR */
            s->qscale= (int)(s->new_picture.quality+0.5);
            assert(s->qscale);
        }
//emms_c();
//printf("qs:%f %f %d\n", s->new_picture.quality, s->current_picture.quality, s->qscale);
1285
        MPV_frame_start(s, avctx);
1286 1287

        encode_picture(s, s->picture_number);
1288
        
J
Juanjo 已提交
1289
        avctx->real_pict_num  = s->picture_number;
1290 1291 1292 1293 1294 1295
        avctx->header_bits = s->header_bits;
        avctx->mv_bits     = s->mv_bits;
        avctx->misc_bits   = s->misc_bits;
        avctx->i_tex_bits  = s->i_tex_bits;
        avctx->p_tex_bits  = s->p_tex_bits;
        avctx->i_count     = s->i_count;
1296
        avctx->p_count     = s->mb_num - s->i_count - s->skip_count; //FIXME f/b_count in avctx
1297 1298 1299 1300 1301 1302
        avctx->skip_count  = s->skip_count;

        MPV_frame_end(s);

        if (s->out_format == FMT_MJPEG)
            mjpeg_picture_trailer(s);
1303 1304 1305
        
        if(s->flags&CODEC_FLAG_PASS1)
            ff_write_pass1_stats(s);
M
Michael Niedermayer 已提交
1306 1307 1308 1309

        for(i=0; i<4; i++){
            avctx->error[i] += s->current_picture_ptr->error[i];
        }
F
Fabrice Bellard 已提交
1310 1311
    }

1312
    s->input_picture_number++;
F
Fabrice Bellard 已提交
1313 1314

    flush_put_bits(&s->pb);
1315
    s->frame_bits  = (pbBufPtr(&s->pb) - s->pb.buf) * 8;
M
Michael Niedermayer 已提交
1316
    
1317
    s->total_bits += s->frame_bits;
1318
    avctx->frame_bits  = s->frame_bits;
1319
    
1320
    return pbBufPtr(&s->pb) - s->pb.buf;
F
Fabrice Bellard 已提交
1321 1322
}

1323 1324
#endif //CONFIG_ENCODERS

M
Michael Niedermayer 已提交
1325
static inline void gmc1_motion(MpegEncContext *s,
1326
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
M
Michael Niedermayer 已提交
1327
                               int dest_offset,
1328
                               uint8_t **ref_picture, int src_offset)
M
Michael Niedermayer 已提交
1329
{
1330
    uint8_t *ptr;
M
Michael Niedermayer 已提交
1331
    int offset, src_x, src_y, linesize, uvlinesize;
M
Michael Niedermayer 已提交
1332
    int motion_x, motion_y;
M
Michael Niedermayer 已提交
1333
    int emu=0;
M
Michael Niedermayer 已提交
1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346

    motion_x= s->sprite_offset[0][0];
    motion_y= s->sprite_offset[0][1];
    src_x = s->mb_x * 16 + (motion_x >> (s->sprite_warping_accuracy+1));
    src_y = s->mb_y * 16 + (motion_y >> (s->sprite_warping_accuracy+1));
    motion_x<<=(3-s->sprite_warping_accuracy);
    motion_y<<=(3-s->sprite_warping_accuracy);
    src_x = clip(src_x, -16, s->width);
    if (src_x == s->width)
        motion_x =0;
    src_y = clip(src_y, -16, s->height);
    if (src_y == s->height)
        motion_y =0;
1347

M
Michael Niedermayer 已提交
1348
    linesize = s->linesize;
M
Michael Niedermayer 已提交
1349
    uvlinesize = s->uvlinesize;
1350
    
M
Michael Niedermayer 已提交
1351 1352 1353
    ptr = ref_picture[0] + (src_y * linesize) + src_x + src_offset;

    dest_y+=dest_offset;
M
Michael Niedermayer 已提交
1354
    if(s->flags&CODEC_FLAG_EMU_EDGE){
1355 1356
        if(src_x<0 || src_y<0 || src_x + 17 >= s->h_edge_pos
                              || src_y + 17 >= s->v_edge_pos){
M
Michael Niedermayer 已提交
1357
            ff_emulated_edge_mc(s, ptr, linesize, 17, 17, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
M
Michael Niedermayer 已提交
1358 1359 1360
            ptr= s->edge_emu_buffer;
        }
    }
1361 1362
    
    if((motion_x|motion_y)&7){
1363 1364
        s->dsp.gmc1(dest_y  , ptr  , linesize, 16, motion_x&15, motion_y&15, 128 - s->no_rounding);
        s->dsp.gmc1(dest_y+8, ptr+8, linesize, 16, motion_x&15, motion_y&15, 128 - s->no_rounding);
1365 1366 1367 1368 1369
    }else{
        int dxy;
        
        dxy= ((motion_x>>3)&1) | ((motion_y>>2)&2);
        if (s->no_rounding){
1370
	    s->dsp.put_no_rnd_pixels_tab[0][dxy](dest_y, ptr, linesize, 16);
1371
        }else{
1372 1373
            s->dsp.put_pixels_tab       [0][dxy](dest_y, ptr, linesize, 16);
        }
1374 1375 1376
    }
    
    if(s->flags&CODEC_FLAG_GRAY) return;
M
Michael Niedermayer 已提交
1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390

    motion_x= s->sprite_offset[1][0];
    motion_y= s->sprite_offset[1][1];
    src_x = s->mb_x * 8 + (motion_x >> (s->sprite_warping_accuracy+1));
    src_y = s->mb_y * 8 + (motion_y >> (s->sprite_warping_accuracy+1));
    motion_x<<=(3-s->sprite_warping_accuracy);
    motion_y<<=(3-s->sprite_warping_accuracy);
    src_x = clip(src_x, -8, s->width>>1);
    if (src_x == s->width>>1)
        motion_x =0;
    src_y = clip(src_y, -8, s->height>>1);
    if (src_y == s->height>>1)
        motion_y =0;

M
Michael Niedermayer 已提交
1391
    offset = (src_y * uvlinesize) + src_x + (src_offset>>1);
M
Michael Niedermayer 已提交
1392
    ptr = ref_picture[1] + offset;
1393 1394 1395 1396 1397 1398 1399
    if(s->flags&CODEC_FLAG_EMU_EDGE){
        if(src_x<0 || src_y<0 || src_x + 9 >= s->h_edge_pos>>1
                              || src_y + 9 >= s->v_edge_pos>>1){
            ff_emulated_edge_mc(s, ptr, uvlinesize, 9, 9, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
            ptr= s->edge_emu_buffer;
            emu=1;
        }
M
Michael Niedermayer 已提交
1400
    }
1401
    s->dsp.gmc1(dest_cb + (dest_offset>>1), ptr, uvlinesize, 8, motion_x&15, motion_y&15, 128 - s->no_rounding);
M
Michael Niedermayer 已提交
1402
    
M
Michael Niedermayer 已提交
1403
    ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1404
    if(emu){
M
Michael Niedermayer 已提交
1405
        ff_emulated_edge_mc(s, ptr, uvlinesize, 9, 9, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
M
Michael Niedermayer 已提交
1406 1407
        ptr= s->edge_emu_buffer;
    }
1408
    s->dsp.gmc1(dest_cr + (dest_offset>>1), ptr, uvlinesize, 8, motion_x&15, motion_y&15, 128 - s->no_rounding);
M
Michael Niedermayer 已提交
1409 1410 1411 1412
    
    return;
}

1413
static inline void gmc_motion(MpegEncContext *s,
1414
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
1415
                               int dest_offset,
1416
                               uint8_t **ref_picture, int src_offset)
1417
{
1418
    uint8_t *ptr;
1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432
    int linesize, uvlinesize;
    const int a= s->sprite_warping_accuracy;
    int ox, oy;

    linesize = s->linesize;
    uvlinesize = s->uvlinesize;

    ptr = ref_picture[0] + src_offset;

    dest_y+=dest_offset;
    
    ox= s->sprite_offset[0][0] + s->sprite_delta[0][0]*s->mb_x*16 + s->sprite_delta[0][1]*s->mb_y*16;
    oy= s->sprite_offset[0][1] + s->sprite_delta[1][0]*s->mb_x*16 + s->sprite_delta[1][1]*s->mb_y*16;

1433
    s->dsp.gmc(dest_y, ptr, linesize, 16,
1434 1435 1436 1437 1438 1439
           ox, 
           oy, 
           s->sprite_delta[0][0], s->sprite_delta[0][1],
           s->sprite_delta[1][0], s->sprite_delta[1][1], 
           a+1, (1<<(2*a+1)) - s->no_rounding,
           s->h_edge_pos, s->v_edge_pos);
1440
    s->dsp.gmc(dest_y+8, ptr, linesize, 16,
1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457
           ox + s->sprite_delta[0][0]*8, 
           oy + s->sprite_delta[1][0]*8, 
           s->sprite_delta[0][0], s->sprite_delta[0][1],
           s->sprite_delta[1][0], s->sprite_delta[1][1], 
           a+1, (1<<(2*a+1)) - s->no_rounding,
           s->h_edge_pos, s->v_edge_pos);

    if(s->flags&CODEC_FLAG_GRAY) return;


    dest_cb+=dest_offset>>1;
    dest_cr+=dest_offset>>1;
    
    ox= s->sprite_offset[1][0] + s->sprite_delta[0][0]*s->mb_x*8 + s->sprite_delta[0][1]*s->mb_y*8;
    oy= s->sprite_offset[1][1] + s->sprite_delta[1][0]*s->mb_x*8 + s->sprite_delta[1][1]*s->mb_y*8;

    ptr = ref_picture[1] + (src_offset>>1);
1458
    s->dsp.gmc(dest_cb, ptr, uvlinesize, 8,
1459 1460 1461 1462 1463 1464 1465 1466
           ox, 
           oy, 
           s->sprite_delta[0][0], s->sprite_delta[0][1],
           s->sprite_delta[1][0], s->sprite_delta[1][1], 
           a+1, (1<<(2*a+1)) - s->no_rounding,
           s->h_edge_pos>>1, s->v_edge_pos>>1);
    
    ptr = ref_picture[2] + (src_offset>>1);
1467
    s->dsp.gmc(dest_cr, ptr, uvlinesize, 8,
1468 1469 1470 1471 1472 1473 1474 1475 1476
           ox, 
           oy, 
           s->sprite_delta[0][0], s->sprite_delta[0][1],
           s->sprite_delta[1][0], s->sprite_delta[1][1], 
           a+1, (1<<(2*a+1)) - s->no_rounding,
           s->h_edge_pos>>1, s->v_edge_pos>>1);
}


1477
void ff_emulated_edge_mc(MpegEncContext *s, uint8_t *src, int linesize, int block_w, int block_h, 
1478 1479 1480
                                    int src_x, int src_y, int w, int h){
    int x, y;
    int start_y, start_x, end_y, end_x;
1481
    uint8_t *buf= s->edge_emu_buffer;
1482

1483 1484 1485
    if(src_y>= h){
        src+= (h-1-src_y)*linesize;
        src_y=h-1;
M
Michael Niedermayer 已提交
1486 1487 1488
    }else if(src_y<=-block_h){
        src+= (1-block_h-src_y)*linesize;
        src_y=1-block_h;
1489 1490 1491 1492
    }
    if(src_x>= w){
        src+= (w-1-src_x);
        src_x=w-1;
M
Michael Niedermayer 已提交
1493 1494 1495
    }else if(src_x<=-block_w){
        src+= (1-block_w-src_x);
        src_x=1-block_w;
1496 1497
    }

M
Michael Niedermayer 已提交
1498 1499 1500 1501
    start_y= FFMAX(0, -src_y);
    start_x= FFMAX(0, -src_x);
    end_y= FFMIN(block_h, h-src_y);
    end_x= FFMIN(block_w, w-src_x);
M
Michael Niedermayer 已提交
1502

1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537
    // copy existing part
    for(y=start_y; y<end_y; y++){
        for(x=start_x; x<end_x; x++){
            buf[x + y*linesize]= src[x + y*linesize];
        }
    }

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

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


F
Fabrice Bellard 已提交
1538 1539
/* apply one mpeg motion vector to the three components */
static inline void mpeg_motion(MpegEncContext *s,
1540
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
F
Fabrice Bellard 已提交
1541
                               int dest_offset,
1542
                               uint8_t **ref_picture, int src_offset,
M
Michael Niedermayer 已提交
1543
                               int field_based, op_pixels_func (*pix_op)[4],
F
Fabrice Bellard 已提交
1544 1545
                               int motion_x, int motion_y, int h)
{
1546
    uint8_t *ptr;
1547
    int dxy, offset, mx, my, src_x, src_y, height, v_edge_pos, linesize, uvlinesize;
1548
    int emu=0;
M
Michael Niedermayer 已提交
1549
#if 0    
M
Michael Niedermayer 已提交
1550 1551 1552 1553 1554
if(s->quarter_sample)
{
    motion_x>>=1;
    motion_y>>=1;
}
M
Michael Niedermayer 已提交
1555
#endif
F
Fabrice Bellard 已提交
1556 1557 1558 1559 1560 1561
    dxy = ((motion_y & 1) << 1) | (motion_x & 1);
    src_x = s->mb_x * 16 + (motion_x >> 1);
    src_y = s->mb_y * (16 >> field_based) + (motion_y >> 1);
                
    /* WARNING: do no forget half pels */
    height = s->height >> field_based;
1562
    v_edge_pos = s->v_edge_pos >> field_based;
F
Fabrice Bellard 已提交
1563 1564 1565 1566 1567 1568
    src_x = clip(src_x, -16, s->width);
    if (src_x == s->width)
        dxy &= ~1;
    src_y = clip(src_y, -16, height);
    if (src_y == height)
        dxy &= ~2;
M
Michael Niedermayer 已提交
1569 1570
    linesize   = s->current_picture.linesize[0] << field_based;
    uvlinesize = s->current_picture.linesize[1] << field_based;
F
Fabrice Bellard 已提交
1571 1572
    ptr = ref_picture[0] + (src_y * linesize) + (src_x) + src_offset;
    dest_y += dest_offset;
1573 1574

    if(s->flags&CODEC_FLAG_EMU_EDGE){
1575 1576
        if(src_x<0 || src_y<0 || src_x + (motion_x&1) + 16 > s->h_edge_pos
                              || src_y + (motion_y&1) + h  > v_edge_pos){
M
Michael Niedermayer 已提交
1577
            ff_emulated_edge_mc(s, ptr - src_offset, s->linesize, 17, 17+field_based,  //FIXME linesize? and uv below
1578 1579
                             src_x, src_y<<field_based, s->h_edge_pos, s->v_edge_pos);
            ptr= s->edge_emu_buffer + src_offset;
1580 1581 1582
            emu=1;
        }
    }
M
Michael Niedermayer 已提交
1583
    pix_op[0][dxy](dest_y, ptr, linesize, h);
F
Fabrice Bellard 已提交
1584

M
Michael Niedermayer 已提交
1585 1586
    if(s->flags&CODEC_FLAG_GRAY) return;

F
Fabrice Bellard 已提交
1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610
    if (s->out_format == FMT_H263) {
        dxy = 0;
        if ((motion_x & 3) != 0)
            dxy |= 1;
        if ((motion_y & 3) != 0)
            dxy |= 2;
        mx = motion_x >> 2;
        my = motion_y >> 2;
    } else {
        mx = motion_x / 2;
        my = motion_y / 2;
        dxy = ((my & 1) << 1) | (mx & 1);
        mx >>= 1;
        my >>= 1;
    }
    
    src_x = s->mb_x * 8 + mx;
    src_y = s->mb_y * (8 >> field_based) + my;
    src_x = clip(src_x, -8, s->width >> 1);
    if (src_x == (s->width >> 1))
        dxy &= ~1;
    src_y = clip(src_y, -8, height >> 1);
    if (src_y == (height >> 1))
        dxy &= ~2;
1611
    offset = (src_y * uvlinesize) + src_x + (src_offset >> 1);
F
Fabrice Bellard 已提交
1612
    ptr = ref_picture[1] + offset;
1613
    if(emu){
M
Michael Niedermayer 已提交
1614
        ff_emulated_edge_mc(s, ptr - (src_offset >> 1), s->uvlinesize, 9, 9+field_based, 
1615 1616
                         src_x, src_y<<field_based, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer + (src_offset >> 1);
1617
    }
M
Michael Niedermayer 已提交
1618
    pix_op[1][dxy](dest_cb + (dest_offset >> 1), ptr, uvlinesize, h >> 1);
1619

F
Fabrice Bellard 已提交
1620
    ptr = ref_picture[2] + offset;
1621
    if(emu){
M
Michael Niedermayer 已提交
1622
        ff_emulated_edge_mc(s, ptr - (src_offset >> 1), s->uvlinesize, 9, 9+field_based, 
1623 1624
                         src_x, src_y<<field_based, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer + (src_offset >> 1);
1625
    }
M
Michael Niedermayer 已提交
1626
    pix_op[1][dxy](dest_cr + (dest_offset >> 1), ptr, uvlinesize, h >> 1);
F
Fabrice Bellard 已提交
1627 1628
}

M
Michael Niedermayer 已提交
1629
static inline void qpel_motion(MpegEncContext *s,
1630
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
M
Michael Niedermayer 已提交
1631
                               int dest_offset,
1632
                               uint8_t **ref_picture, int src_offset,
M
Michael Niedermayer 已提交
1633 1634
                               int field_based, op_pixels_func (*pix_op)[4],
                               qpel_mc_func (*qpix_op)[16],
M
Michael Niedermayer 已提交
1635 1636
                               int motion_x, int motion_y, int h)
{
1637
    uint8_t *ptr;
1638
    int dxy, offset, mx, my, src_x, src_y, height, v_edge_pos, linesize, uvlinesize;
M
Michael Niedermayer 已提交
1639
    int emu=0;
M
Michael Niedermayer 已提交
1640 1641 1642 1643 1644 1645

    dxy = ((motion_y & 3) << 2) | (motion_x & 3);
    src_x = s->mb_x * 16 + (motion_x >> 2);
    src_y = s->mb_y * (16 >> field_based) + (motion_y >> 2);

    height = s->height >> field_based;
1646
    v_edge_pos = s->v_edge_pos >> field_based;
M
Michael Niedermayer 已提交
1647 1648 1649 1650 1651 1652 1653
    src_x = clip(src_x, -16, s->width);
    if (src_x == s->width)
        dxy &= ~3;
    src_y = clip(src_y, -16, height);
    if (src_y == height)
        dxy &= ~12;
    linesize = s->linesize << field_based;
1654
    uvlinesize = s->uvlinesize << field_based;
M
Michael Niedermayer 已提交
1655 1656 1657
    ptr = ref_picture[0] + (src_y * linesize) + src_x + src_offset;
    dest_y += dest_offset;
//printf("%d %d %d\n", src_x, src_y, dxy);
M
Michael Niedermayer 已提交
1658 1659
    
    if(s->flags&CODEC_FLAG_EMU_EDGE){
1660 1661
        if(src_x<0 || src_y<0 || src_x + (motion_x&3) + 16 > s->h_edge_pos
                              || src_y + (motion_y&3) + h  > v_edge_pos){
M
Michael Niedermayer 已提交
1662
            ff_emulated_edge_mc(s, ptr - src_offset, s->linesize, 17, 17+field_based, 
1663 1664
                             src_x, src_y<<field_based, s->h_edge_pos, s->v_edge_pos);
            ptr= s->edge_emu_buffer + src_offset;
M
Michael Niedermayer 已提交
1665 1666 1667
            emu=1;
        }
    }
1668 1669 1670 1671 1672 1673 1674 1675
    if(!field_based)
        qpix_op[0][dxy](dest_y, ptr, linesize);
    else{
        //damn interlaced mode
        //FIXME boundary mirroring is not exactly correct here
        qpix_op[1][dxy](dest_y  , ptr  , linesize);
        qpix_op[1][dxy](dest_y+8, ptr+8, linesize);
    }
M
Michael Niedermayer 已提交
1676

M
Michael Niedermayer 已提交
1677 1678
    if(s->flags&CODEC_FLAG_GRAY) return;

1679 1680 1681
    if(field_based){
        mx= motion_x/2;
        my= motion_y>>1;
M
Michael Niedermayer 已提交
1682 1683 1684 1685
    }else if(s->workaround_bugs&FF_BUG_QPEL_CHROMA2){
        static const int rtab[8]= {0,0,1,1,0,0,0,1};
        mx= (motion_x>>1) + rtab[motion_x&7];
        my= (motion_y>>1) + rtab[motion_y&7];
M
Michael Niedermayer 已提交
1686
    }else if(s->workaround_bugs&FF_BUG_QPEL_CHROMA){
1687 1688 1689 1690 1691 1692 1693 1694
        mx= (motion_x>>1)|(motion_x&1);
        my= (motion_y>>1)|(motion_y&1);
    }else{
        mx= motion_x/2;
        my= motion_y/2;
    }
    mx= (mx>>1)|(mx&1);
    my= (my>>1)|(my&1);
M
Michael Niedermayer 已提交
1695

1696 1697 1698
    dxy= (mx&1) | ((my&1)<<1);
    mx>>=1;
    my>>=1;
M
Michael Niedermayer 已提交
1699 1700 1701 1702 1703 1704 1705 1706 1707 1708

    src_x = s->mb_x * 8 + mx;
    src_y = s->mb_y * (8 >> field_based) + my;
    src_x = clip(src_x, -8, s->width >> 1);
    if (src_x == (s->width >> 1))
        dxy &= ~1;
    src_y = clip(src_y, -8, height >> 1);
    if (src_y == (height >> 1))
        dxy &= ~2;

1709
    offset = (src_y * uvlinesize) + src_x + (src_offset >> 1);
M
Michael Niedermayer 已提交
1710
    ptr = ref_picture[1] + offset;
M
Michael Niedermayer 已提交
1711
    if(emu){
M
Michael Niedermayer 已提交
1712
        ff_emulated_edge_mc(s, ptr - (src_offset >> 1), s->uvlinesize, 9, 9 + field_based, 
1713 1714
                         src_x, src_y<<field_based, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer + (src_offset >> 1);
M
Michael Niedermayer 已提交
1715
    }
1716
    pix_op[1][dxy](dest_cb + (dest_offset >> 1), ptr,  uvlinesize, h >> 1);
M
Michael Niedermayer 已提交
1717
    
M
Michael Niedermayer 已提交
1718
    ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1719
    if(emu){
M
Michael Niedermayer 已提交
1720
        ff_emulated_edge_mc(s, ptr - (src_offset >> 1), s->uvlinesize, 9, 9 + field_based, 
1721 1722
                         src_x, src_y<<field_based, s->h_edge_pos>>1, s->v_edge_pos>>1);
        ptr= s->edge_emu_buffer + (src_offset >> 1);
M
Michael Niedermayer 已提交
1723
    }
1724
    pix_op[1][dxy](dest_cr + (dest_offset >> 1), ptr,  uvlinesize, h >> 1);
M
Michael Niedermayer 已提交
1725 1726
}

1727 1728 1729 1730 1731 1732 1733 1734
inline int ff_h263_round_chroma(int x){
    if (x >= 0)
        return  (h263_chroma_roundtab[x & 0xf] + ((x >> 3) & ~1));
    else {
        x = -x;
        return -(h263_chroma_roundtab[x & 0xf] + ((x >> 3) & ~1));
    }
}
M
Michael Niedermayer 已提交
1735

F
Fabrice Bellard 已提交
1736
static inline void MPV_motion(MpegEncContext *s, 
1737 1738
                              uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
                              int dir, uint8_t **ref_picture, 
M
Michael Niedermayer 已提交
1739
                              op_pixels_func (*pix_op)[4], qpel_mc_func (*qpix_op)[16])
F
Fabrice Bellard 已提交
1740 1741 1742
{
    int dxy, offset, mx, my, src_x, src_y, motion_x, motion_y;
    int mb_x, mb_y, i;
1743
    uint8_t *ptr, *dest;
M
Michael Niedermayer 已提交
1744
    int emu=0;
F
Fabrice Bellard 已提交
1745 1746 1747 1748 1749 1750

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

    switch(s->mv_type) {
    case MV_TYPE_16X16:
M
Michael Niedermayer 已提交
1751
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
1752
        if(s->mcsel){
1753 1754 1755 1756 1757 1758 1759
            if(s->real_sprite_warping_points==1){
                gmc1_motion(s, dest_y, dest_cb, dest_cr, 0,
                            ref_picture, 0);
            }else{
                gmc_motion(s, dest_y, dest_cb, dest_cr, 0,
                            ref_picture, 0);
            }
M
Michael Niedermayer 已提交
1760
        }else if(s->quarter_sample){
M
Michael Niedermayer 已提交
1761 1762 1763 1764
            qpel_motion(s, dest_y, dest_cb, dest_cr, 0,
                        ref_picture, 0,
                        0, pix_op, qpix_op,
                        s->mv[dir][0][0], s->mv[dir][0][1], 16);
M
Michael Niedermayer 已提交
1765 1766 1767 1768
        }else if(s->mspel){
            ff_mspel_motion(s, dest_y, dest_cb, dest_cr,
                        ref_picture, pix_op,
                        s->mv[dir][0][0], s->mv[dir][0][1], 16);
M
Michael Niedermayer 已提交
1769 1770 1771
        }else
#endif
        {
M
Michael Niedermayer 已提交
1772 1773 1774 1775 1776
            mpeg_motion(s, dest_y, dest_cb, dest_cr, 0,
                        ref_picture, 0,
                        0, pix_op,
                        s->mv[dir][0][0], s->mv[dir][0][1], 16);
        }           
F
Fabrice Bellard 已提交
1777 1778
        break;
    case MV_TYPE_8X8:
M
Michael Niedermayer 已提交
1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801
        mx = 0;
        my = 0;
        if(s->quarter_sample){
            for(i=0;i<4;i++) {
                motion_x = s->mv[dir][i][0];
                motion_y = s->mv[dir][i][1];

                dxy = ((motion_y & 3) << 2) | (motion_x & 3);
                src_x = mb_x * 16 + (motion_x >> 2) + (i & 1) * 8;
                src_y = mb_y * 16 + (motion_y >> 2) + (i >>1) * 8;
                    
                /* WARNING: do no forget half pels */
                src_x = clip(src_x, -16, s->width);
                if (src_x == s->width)
                    dxy &= ~3;
                src_y = clip(src_y, -16, s->height);
                if (src_y == s->height)
                    dxy &= ~12;
                    
                ptr = ref_picture[0] + (src_y * s->linesize) + (src_x);
                if(s->flags&CODEC_FLAG_EMU_EDGE){
                    if(src_x<0 || src_y<0 || src_x + (motion_x&3) + 8 > s->h_edge_pos
                                          || src_y + (motion_y&3) + 8 > s->v_edge_pos){
M
Michael Niedermayer 已提交
1802
                        ff_emulated_edge_mc(s, ptr, s->linesize, 9, 9, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
M
Michael Niedermayer 已提交
1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
                        ptr= s->edge_emu_buffer;
                    }
                }
                dest = dest_y + ((i & 1) * 8) + (i >> 1) * 8 * s->linesize;
                qpix_op[1][dxy](dest, ptr, s->linesize);

                mx += s->mv[dir][i][0]/2;
                my += s->mv[dir][i][1]/2;
            }
        }else{
            for(i=0;i<4;i++) {
                motion_x = s->mv[dir][i][0];
                motion_y = s->mv[dir][i][1];
F
Fabrice Bellard 已提交
1816

M
Michael Niedermayer 已提交
1817 1818 1819
                dxy = ((motion_y & 1) << 1) | (motion_x & 1);
                src_x = mb_x * 16 + (motion_x >> 1) + (i & 1) * 8;
                src_y = mb_y * 16 + (motion_y >> 1) + (i >>1) * 8;
F
Fabrice Bellard 已提交
1820
                    
M
Michael Niedermayer 已提交
1821 1822 1823 1824 1825 1826 1827
                /* WARNING: do no forget half pels */
                src_x = clip(src_x, -16, s->width);
                if (src_x == s->width)
                    dxy &= ~1;
                src_y = clip(src_y, -16, s->height);
                if (src_y == s->height)
                    dxy &= ~2;
F
Fabrice Bellard 已提交
1828
                    
M
Michael Niedermayer 已提交
1829 1830 1831 1832
                ptr = ref_picture[0] + (src_y * s->linesize) + (src_x);
                if(s->flags&CODEC_FLAG_EMU_EDGE){
                    if(src_x<0 || src_y<0 || src_x + (motion_x&1) + 8 > s->h_edge_pos
                                          || src_y + (motion_y&1) + 8 > s->v_edge_pos){
M
Michael Niedermayer 已提交
1833
                        ff_emulated_edge_mc(s, ptr, s->linesize, 9, 9, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
M
Michael Niedermayer 已提交
1834 1835
                        ptr= s->edge_emu_buffer;
                    }
M
Michael Niedermayer 已提交
1836
                }
M
Michael Niedermayer 已提交
1837 1838 1839 1840 1841
                dest = dest_y + ((i & 1) * 8) + (i >> 1) * 8 * s->linesize;
                pix_op[1][dxy](dest, ptr, s->linesize, 8);

                mx += s->mv[dir][i][0];
                my += s->mv[dir][i][1];
M
Michael Niedermayer 已提交
1842
            }
F
Fabrice Bellard 已提交
1843
        }
M
Michael Niedermayer 已提交
1844

M
Michael Niedermayer 已提交
1845
        if(s->flags&CODEC_FLAG_GRAY) break;
F
Fabrice Bellard 已提交
1846 1847
        /* In case of 8X8, we construct a single chroma motion vector
           with a special rounding */
1848 1849
        mx= ff_h263_round_chroma(mx);
        my= ff_h263_round_chroma(my);
F
Fabrice Bellard 已提交
1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
        dxy = ((my & 1) << 1) | (mx & 1);
        mx >>= 1;
        my >>= 1;

        src_x = mb_x * 8 + mx;
        src_y = mb_y * 8 + my;
        src_x = clip(src_x, -8, s->width/2);
        if (src_x == s->width/2)
            dxy &= ~1;
        src_y = clip(src_y, -8, s->height/2);
        if (src_y == s->height/2)
            dxy &= ~2;
        
M
Michael Niedermayer 已提交
1863
        offset = (src_y * (s->uvlinesize)) + src_x;
F
Fabrice Bellard 已提交
1864
        ptr = ref_picture[1] + offset;
M
Michael Niedermayer 已提交
1865
        if(s->flags&CODEC_FLAG_EMU_EDGE){
1866 1867
                if(src_x<0 || src_y<0 || src_x + (dxy &1) + 8 > s->h_edge_pos>>1
                                      || src_y + (dxy>>1) + 8 > s->v_edge_pos>>1){
M
Michael Niedermayer 已提交
1868
                    ff_emulated_edge_mc(s, ptr, s->uvlinesize, 9, 9, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
M
Michael Niedermayer 已提交
1869 1870 1871 1872
                    ptr= s->edge_emu_buffer;
                    emu=1;
                }
            }
M
Michael Niedermayer 已提交
1873
        pix_op[1][dxy](dest_cb, ptr, s->uvlinesize, 8);
M
Michael Niedermayer 已提交
1874

F
Fabrice Bellard 已提交
1875
        ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1876
        if(emu){
M
Michael Niedermayer 已提交
1877
            ff_emulated_edge_mc(s, ptr, s->uvlinesize, 9, 9, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
M
Michael Niedermayer 已提交
1878 1879
            ptr= s->edge_emu_buffer;
        }
M
Michael Niedermayer 已提交
1880
        pix_op[1][dxy](dest_cr, ptr, s->uvlinesize, 8);
F
Fabrice Bellard 已提交
1881 1882 1883
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906
            if(s->quarter_sample){
                /* top field */
                qpel_motion(s, dest_y, dest_cb, dest_cr, 0,
                            ref_picture, s->field_select[dir][0] ? s->linesize : 0,
                            1, pix_op, qpix_op,
                            s->mv[dir][0][0], s->mv[dir][0][1], 8);
                /* bottom field */
                qpel_motion(s, dest_y, dest_cb, dest_cr, s->linesize,
                            ref_picture, s->field_select[dir][1] ? s->linesize : 0,
                            1, pix_op, qpix_op,
                            s->mv[dir][1][0], s->mv[dir][1][1], 8);
            }else{
                /* top field */       
                mpeg_motion(s, dest_y, dest_cb, dest_cr, 0,
                            ref_picture, s->field_select[dir][0] ? s->linesize : 0,
                            1, pix_op,
                            s->mv[dir][0][0], s->mv[dir][0][1], 8);
                /* bottom field */
                mpeg_motion(s, dest_y, dest_cb, dest_cr, s->linesize,
                            ref_picture, s->field_select[dir][1] ? s->linesize : 0,
                            1, pix_op,
                            s->mv[dir][1][0], s->mv[dir][1][1], 8);
            }
F
Fabrice Bellard 已提交
1907
        } else {
1908 1909
            int offset;
            if(s->picture_structure == s->field_select[dir][0] + 1 || s->pict_type == B_TYPE || s->first_field){
M
Michael Niedermayer 已提交
1910
                offset= s->field_select[dir][0] ? s->linesize : 0;
1911 1912
            }else{
                ref_picture= s->current_picture.data;
M
Michael Niedermayer 已提交
1913
                offset= s->field_select[dir][0] ? s->linesize : -s->linesize; 
1914
            } 
F
Fabrice Bellard 已提交
1915

1916 1917 1918 1919
            mpeg_motion(s, dest_y, dest_cb, dest_cr, 0,
                        ref_picture, offset,
                        0, pix_op,
                        s->mv[dir][0][0], s->mv[dir][0][1], 16);
F
Fabrice Bellard 已提交
1920 1921 1922 1923 1924 1925 1926 1927
        }
        break;
    }
}


/* put block[] to dest[] */
static inline void put_dct(MpegEncContext *s, 
1928
                           DCTELEM *block, int i, uint8_t *dest, int line_size)
F
Fabrice Bellard 已提交
1929
{
1930
    s->dct_unquantize(s, block, i, s->qscale);
1931
    s->dsp.idct_put (dest, line_size, block);
F
Fabrice Bellard 已提交
1932 1933 1934 1935
}

/* add block[] to dest[] */
static inline void add_dct(MpegEncContext *s, 
1936
                           DCTELEM *block, int i, uint8_t *dest, int line_size)
F
Fabrice Bellard 已提交
1937
{
M
Michael Niedermayer 已提交
1938
    if (s->block_last_index[i] >= 0) {
1939
        s->dsp.idct_add (dest, line_size, block);
M
Michael Niedermayer 已提交
1940 1941
    }
}
1942

M
Michael Niedermayer 已提交
1943
static inline void add_dequant_dct(MpegEncContext *s, 
1944
                           DCTELEM *block, int i, uint8_t *dest, int line_size)
M
Michael Niedermayer 已提交
1945
{
F
Fabrice Bellard 已提交
1946
    if (s->block_last_index[i] >= 0) {
M
Michael Niedermayer 已提交
1947
        s->dct_unquantize(s, block, i, s->qscale);
1948

1949
        s->dsp.idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1950 1951 1952
    }
}

1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965
/**
 * cleans dc, ac, coded_block for the current non intra MB
 */
void ff_clean_intra_table_entries(MpegEncContext *s)
{
    int wrap = s->block_wrap[0];
    int xy = s->block_index[0];
    
    s->dc_val[0][xy           ] = 
    s->dc_val[0][xy + 1       ] = 
    s->dc_val[0][xy     + wrap] =
    s->dc_val[0][xy + 1 + wrap] = 1024;
    /* ac pred */
1966 1967
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(int16_t));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(int16_t));
1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979
    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 */
    wrap = s->block_wrap[4];
    xy = s->mb_x + 1 + (s->mb_y + 1) * wrap;
    s->dc_val[1][xy] =
    s->dc_val[2][xy] = 1024;
    /* ac pred */
1980 1981
    memset(s->ac_val[1][xy], 0, 16 * sizeof(int16_t));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(int16_t));
1982 1983 1984 1985
    
    s->mbintra_table[s->mb_x + s->mb_y*s->mb_width]= 0;
}

F
Fabrice Bellard 已提交
1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997
/* 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)
 */
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[6][64])
{
1998
    int mb_x, mb_y;
M
Michael Niedermayer 已提交
1999
    const int mb_xy = s->mb_y * s->mb_width + s->mb_x;
F
Fabrice Bellard 已提交
2000 2001 2002 2003

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

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

F
Fabrice Bellard 已提交
2006 2007
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
2008
        if (s->h263_pred || s->h263_aic) {
M
Michael Niedermayer 已提交
2009
            if(s->mbintra_table[mb_xy])
2010
                ff_clean_intra_table_entries(s);
F
Fabrice Bellard 已提交
2011
        } else {
2012 2013
            s->last_dc[0] =
            s->last_dc[1] =
F
Fabrice Bellard 已提交
2014 2015 2016
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
2017
    else if (s->h263_pred || s->h263_aic)
M
Michael Niedermayer 已提交
2018
        s->mbintra_table[mb_xy]=1;
2019

2020
    /* update motion predictor, not for B-frames as they need the motion_val from the last P/S-Frame */
M
Michael Niedermayer 已提交
2021
    if (s->out_format == FMT_H263 && s->pict_type!=B_TYPE) { //FIXME move into h263.c if possible, format specific stuff shouldnt be here
M
Michael Niedermayer 已提交
2022
        //FIXME a lot of thet is only needed for !low_delay
M
Michael Niedermayer 已提交
2023 2024
        const int wrap = s->block_wrap[0];
        const int xy = s->block_index[0];
2025
        const int mb_index= s->mb_x + s->mb_y*s->mb_width;
2026
        if(s->mv_type == MV_TYPE_8X8){
2027
            s->co_located_type_table[mb_index]= CO_LOCATED_TYPE_4MV;
2028 2029 2030 2031 2032
        } else {
            int motion_x, motion_y;
            if (s->mb_intra) {
                motion_x = 0;
                motion_y = 0;
2033 2034
                if(s->co_located_type_table)
                    s->co_located_type_table[mb_index]= 0;
2035 2036 2037
            } else if (s->mv_type == MV_TYPE_16X16) {
                motion_x = s->mv[0][0][0];
                motion_y = s->mv[0][0][1];
2038 2039
                if(s->co_located_type_table)
                    s->co_located_type_table[mb_index]= 0;
2040
            } else /*if (s->mv_type == MV_TYPE_FIELD)*/ {
2041
                int i;
2042 2043 2044
                motion_x = s->mv[0][0][0] + s->mv[0][1][0];
                motion_y = s->mv[0][0][1] + s->mv[0][1][1];
                motion_x = (motion_x>>1) | (motion_x&1);
2045 2046 2047 2048 2049 2050
                for(i=0; i<2; i++){
                    s->field_mv_table[mb_index][i][0]= s->mv[0][i][0];
                    s->field_mv_table[mb_index][i][1]= s->mv[0][i][1];
                    s->field_select_table[mb_index][i]= s->field_select[0][i];
                }
                s->co_located_type_table[mb_index]= CO_LOCATED_TYPE_FIELDMV;
2051
            }
F
Fabrice Bellard 已提交
2052
            /* no update if 8X8 because it has been done during parsing */
2053 2054 2055 2056 2057 2058 2059 2060
            s->motion_val[xy][0] = motion_x;
            s->motion_val[xy][1] = motion_y;
            s->motion_val[xy + 1][0] = motion_x;
            s->motion_val[xy + 1][1] = motion_y;
            s->motion_val[xy + wrap][0] = motion_x;
            s->motion_val[xy + wrap][1] = motion_y;
            s->motion_val[xy + 1 + wrap][0] = motion_x;
            s->motion_val[xy + 1 + wrap][1] = motion_y;
F
Fabrice Bellard 已提交
2061 2062 2063
        }
    }
    
2064
    if ((s->flags&CODEC_FLAG_PSNR) || !(s->encoding && (s->intra_only || s->pict_type==B_TYPE))) { //FIXME precalc
2065
        uint8_t *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
2066
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
2067 2068
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
M
Michael Niedermayer 已提交
2069 2070
        const int linesize= s->current_picture.linesize[0]; //not s->linesize as this woulnd be wrong for field pics
        const int uvlinesize= s->current_picture.linesize[1];
2071

M
cleanup  
Michael Niedermayer 已提交
2072
        /* avoid copy if macroblock skipped in last frame too */
2073
        if (s->pict_type != B_TYPE) {
M
cleanup  
Michael Niedermayer 已提交
2074 2075 2076 2077 2078
            s->current_picture.mbskip_table[mb_xy]= s->mb_skiped;
        }

        /* skip only during decoding as we might trash the buffers during encoding a bit */
        if(!s->encoding){
2079
            uint8_t *mbskip_ptr = &s->mbskip_table[mb_xy];
M
cleanup  
Michael Niedermayer 已提交
2080
            const int age= s->current_picture.age;
M
Michael Niedermayer 已提交
2081

M
cleanup  
Michael Niedermayer 已提交
2082 2083 2084 2085 2086 2087
            assert(age);

            if (s->mb_skiped) {
                s->mb_skiped= 0;
                assert(s->pict_type!=I_TYPE);
 
M
Michael Niedermayer 已提交
2088 2089 2090
                (*mbskip_ptr) ++; /* indicate that this time we skiped it */
                if(*mbskip_ptr >99) *mbskip_ptr= 99;

M
cleanup  
Michael Niedermayer 已提交
2091
                /* if previous was skipped too, then nothing to do !  */
2092 2093
                if (*mbskip_ptr >= age && s->current_picture.reference){
                    return;
M
cleanup  
Michael Niedermayer 已提交
2094
                }
2095 2096 2097 2098
            } 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{
2099 2100
                *mbskip_ptr = 0; /* not skipped */
            }
M
cleanup  
Michael Niedermayer 已提交
2101 2102
        }else
            s->mb_skiped= 0;
F
Fabrice Bellard 已提交
2103

2104
        if(s->pict_type==B_TYPE && s->avctx->draw_horiz_band && s->picture_structure==PICT_FRAME){ //FIXME precalc
M
cleanup  
Michael Niedermayer 已提交
2105 2106 2107
            dest_y  = s->current_picture.data[0] + mb_x * 16;
            dest_cb = s->current_picture.data[1] + mb_x * 8;
            dest_cr = s->current_picture.data[2] + mb_x * 8;
M
Michael Niedermayer 已提交
2108
        }else{
M
Michael Niedermayer 已提交
2109 2110 2111
            dest_y  = s->current_picture.data[0] + (mb_y * 16* linesize  ) + mb_x * 16;
            dest_cb = s->current_picture.data[1] + (mb_y * 8 * uvlinesize) + mb_x * 8;
            dest_cr = s->current_picture.data[2] + (mb_y * 8 * uvlinesize) + mb_x * 8;
M
Michael Niedermayer 已提交
2112
        }
F
Fabrice Bellard 已提交
2113 2114

        if (s->interlaced_dct) {
M
Michael Niedermayer 已提交
2115 2116
            dct_linesize = linesize * 2;
            dct_offset = linesize;
F
Fabrice Bellard 已提交
2117
        } else {
M
Michael Niedermayer 已提交
2118 2119
            dct_linesize = linesize;
            dct_offset = linesize * 8;
F
Fabrice Bellard 已提交
2120 2121 2122 2123
        }

        if (!s->mb_intra) {
            /* motion handling */
2124
            /* decoding or more than one mb_type (MC was allready done otherwise) */
M
Michael Niedermayer 已提交
2125
            if((!s->encoding) || (s->mb_type[mb_xy]&(s->mb_type[mb_xy]-1))){
2126
                if ((!s->no_rounding) || s->pict_type==B_TYPE){                
2127 2128
		    op_pix = s->dsp.put_pixels_tab;
                    op_qpix= s->dsp.put_qpel_pixels_tab;
2129
                }else{
2130 2131
                    op_pix = s->dsp.put_no_rnd_pixels_tab;
                    op_qpix= s->dsp.put_no_rnd_qpel_pixels_tab;
2132
                }
F
Fabrice Bellard 已提交
2133

2134
                if (s->mv_dir & MV_DIR_FORWARD) {
M
cleanup  
Michael Niedermayer 已提交
2135
                    MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix, op_qpix);
2136 2137
		    op_pix = s->dsp.avg_pixels_tab;
                    op_qpix= s->dsp.avg_qpel_pixels_tab;
2138 2139
                }
                if (s->mv_dir & MV_DIR_BACKWARD) {
M
cleanup  
Michael Niedermayer 已提交
2140
                    MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix, op_qpix);
2141
                }
F
Fabrice Bellard 已提交
2142 2143
            }

M
Michael Niedermayer 已提交
2144
            /* skip dequant / idct if we are really late ;) */
2145
            if(s->hurry_up>1) return;
M
Michael Niedermayer 已提交
2146

F
Fabrice Bellard 已提交
2147
            /* add dct residue */
2148 2149
            if(s->encoding || !(   s->mpeg2 || s->h263_msmpeg4 || s->codec_id==CODEC_ID_MPEG1VIDEO 
                                || (s->codec_id==CODEC_ID_MPEG4 && !s->mpeg_quant))){
M
Michael Niedermayer 已提交
2150 2151 2152 2153 2154
                add_dequant_dct(s, block[0], 0, dest_y, dct_linesize);
                add_dequant_dct(s, block[1], 1, dest_y + 8, dct_linesize);
                add_dequant_dct(s, block[2], 2, dest_y + dct_offset, dct_linesize);
                add_dequant_dct(s, block[3], 3, dest_y + dct_offset + 8, dct_linesize);

M
Michael Niedermayer 已提交
2155
                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
2156 2157
                    add_dequant_dct(s, block[4], 4, dest_cb, uvlinesize);
                    add_dequant_dct(s, block[5], 5, dest_cr, uvlinesize);
M
Michael Niedermayer 已提交
2158
                }
M
Michael Niedermayer 已提交
2159
            } else if(s->codec_id != CODEC_ID_WMV2){
M
Michael Niedermayer 已提交
2160 2161 2162 2163
                add_dct(s, block[0], 0, dest_y, dct_linesize);
                add_dct(s, block[1], 1, dest_y + 8, dct_linesize);
                add_dct(s, block[2], 2, dest_y + dct_offset, dct_linesize);
                add_dct(s, block[3], 3, dest_y + dct_offset + 8, dct_linesize);
F
Fabrice Bellard 已提交
2164

M
Michael Niedermayer 已提交
2165
                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
2166 2167
                    add_dct(s, block[4], 4, dest_cb, uvlinesize);
                    add_dct(s, block[5], 5, dest_cr, uvlinesize);
M
Michael Niedermayer 已提交
2168
                }
M
Michael Niedermayer 已提交
2169 2170 2171
            } 
#ifdef CONFIG_RISKY
            else{
M
Michael Niedermayer 已提交
2172
                ff_wmv2_add_mb(s, block, dest_y, dest_cb, dest_cr);
M
Michael Niedermayer 已提交
2173
            }
M
Michael Niedermayer 已提交
2174
#endif
F
Fabrice Bellard 已提交
2175 2176
        } else {
            /* dct only in intra block */
2177 2178 2179 2180 2181 2182 2183
            if(s->encoding || !(s->mpeg2 || s->codec_id==CODEC_ID_MPEG1VIDEO)){
                put_dct(s, block[0], 0, dest_y, dct_linesize);
                put_dct(s, block[1], 1, dest_y + 8, dct_linesize);
                put_dct(s, block[2], 2, dest_y + dct_offset, dct_linesize);
                put_dct(s, block[3], 3, dest_y + dct_offset + 8, dct_linesize);

                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
2184 2185
                    put_dct(s, block[4], 4, dest_cb, uvlinesize);
                    put_dct(s, block[5], 5, dest_cr, uvlinesize);
2186 2187
                }
            }else{
2188 2189 2190 2191
                s->dsp.idct_put(dest_y                 , dct_linesize, block[0]);
                s->dsp.idct_put(dest_y              + 8, dct_linesize, block[1]);
                s->dsp.idct_put(dest_y + dct_offset    , dct_linesize, block[2]);
                s->dsp.idct_put(dest_y + dct_offset + 8, dct_linesize, block[3]);
2192 2193

                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
2194 2195
                    s->dsp.idct_put(dest_cb, uvlinesize, block[4]);
                    s->dsp.idct_put(dest_cr, uvlinesize, block[5]);
2196
                }
M
Michael Niedermayer 已提交
2197
            }
F
Fabrice Bellard 已提交
2198 2199 2200 2201
        }
    }
}

2202 2203
#ifdef CONFIG_ENCODERS

2204
static inline void dct_single_coeff_elimination(MpegEncContext *s, int n, int threshold)
2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219
{
    static const char tab[64]=
        {3,2,2,1,1,1,1,1,
         1,1,1,1,1,1,1,1,
         1,1,1,1,1,1,1,1,
         0,0,0,0,0,0,0,0,
         0,0,0,0,0,0,0,0,
         0,0,0,0,0,0,0,0,
         0,0,0,0,0,0,0,0,
         0,0,0,0,0,0,0,0};
    int score=0;
    int run=0;
    int i;
    DCTELEM *block= s->block[n];
    const int last_index= s->block_last_index[n];
2220
    int skip_dc;
2221

2222 2223 2224
    if(threshold<0){
        skip_dc=0;
        threshold= -threshold;
2225 2226
    }else
        skip_dc=1;
2227

2228 2229 2230 2231
    /* are all which we could set to zero are allready zero? */
    if(last_index<=skip_dc - 1) return;

    for(i=0; i<=last_index; i++){
2232
        const int j = s->intra_scantable.permutated[i];
2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245
        const int level = ABS(block[j]);
        if(level==1){
            if(skip_dc && i==0) continue;
            score+= tab[run];
            run=0;
        }else if(level>1){
            return;
        }else{
            run++;
        }
    }
    if(score >= threshold) return;
    for(i=skip_dc; i<=last_index; i++){
2246
        const int j = s->intra_scantable.permutated[i];
2247 2248 2249 2250 2251 2252
        block[j]=0;
    }
    if(block[0]) s->block_last_index[n]= 0;
    else         s->block_last_index[n]= -1;
}

2253 2254 2255 2256 2257
static inline void clip_coeffs(MpegEncContext *s, DCTELEM *block, int last_index)
{
    int i;
    const int maxlevel= s->max_qcoeff;
    const int minlevel= s->min_qcoeff;
2258 2259 2260 2261 2262 2263 2264 2265
    
    if(s->mb_intra){
        i=1; //skip clipping of intra dc
    }else
        i=0;
    
    for(;i<=last_index; i++){
        const int j= s->intra_scantable.permutated[i];
2266 2267 2268 2269
        int level = block[j];
       
        if     (level>maxlevel) level=maxlevel;
        else if(level<minlevel) level=minlevel;
2270

M
cleanup  
Michael Niedermayer 已提交
2271
        block[j]= level;
2272 2273 2274 2275
    }
}

#if 0
2276
static int pix_vcmp16x8(uint8_t *s, int stride){ //FIXME move to dsputil & optimize
2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290
    int score=0;
    int x,y;
    
    for(y=0; y<7; y++){
        for(x=0; x<16; x+=4){
            score+= ABS(s[x  ] - s[x  +stride]) + ABS(s[x+1] - s[x+1+stride]) 
                   +ABS(s[x+2] - s[x+2+stride]) + ABS(s[x+3] - s[x+3+stride]);
        }
        s+= stride;
    }
    
    return score;
}

2291
static int pix_diff_vcmp16x8(uint8_t *s1, uint8_t*s2, int stride){ //FIXME move to dsputil & optimize
2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307
    int score=0;
    int x,y;
    
    for(y=0; y<7; y++){
        for(x=0; x<16; x++){
            score+= ABS(s1[x  ] - s2[x ] - s1[x  +stride] + s2[x +stride]);
        }
        s1+= stride;
        s2+= stride;
    }
    
    return score;
}
#else
#define SQ(a) ((a)*(a))

2308
static int pix_vcmp16x8(uint8_t *s, int stride){ //FIXME move to dsputil & optimize
2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322
    int score=0;
    int x,y;
    
    for(y=0; y<7; y++){
        for(x=0; x<16; x+=4){
            score+= SQ(s[x  ] - s[x  +stride]) + SQ(s[x+1] - s[x+1+stride]) 
                   +SQ(s[x+2] - s[x+2+stride]) + SQ(s[x+3] - s[x+3+stride]);
        }
        s+= stride;
    }
    
    return score;
}

2323
static int pix_diff_vcmp16x8(uint8_t *s1, uint8_t*s2, int stride){ //FIXME move to dsputil & optimize
2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338
    int score=0;
    int x,y;
    
    for(y=0; y<7; y++){
        for(x=0; x<16; x++){
            score+= SQ(s1[x  ] - s2[x ] - s1[x  +stride] + s2[x +stride]);
        }
        s1+= stride;
        s2+= stride;
    }
    
    return score;
}

#endif
2339

2340 2341
#endif //CONFIG_ENCODERS

2342 2343 2344 2345 2346
/**
 *
 * @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){
2347
    if (    s->avctx->draw_horiz_band 
M
Michael Niedermayer 已提交
2348
        && (s->last_picture_ptr || s->low_delay) ) {
2349
        uint8_t *src_ptr[3];
2350 2351
        int offset;
        h= FFMIN(h, s->height - y);
2352

2353
        if(s->pict_type==B_TYPE && s->picture_structure == PICT_FRAME)
2354 2355 2356 2357
            offset = 0;
        else
            offset = y * s->linesize;

M
Michael Niedermayer 已提交
2358
        if(s->pict_type==B_TYPE || s->low_delay){
M
cleanup  
Michael Niedermayer 已提交
2359 2360 2361
            src_ptr[0] = s->current_picture.data[0] + offset;
            src_ptr[1] = s->current_picture.data[1] + (offset >> 2);
            src_ptr[2] = s->current_picture.data[2] + (offset >> 2);
2362
        } else {
M
cleanup  
Michael Niedermayer 已提交
2363 2364 2365
            src_ptr[0] = s->last_picture.data[0] + offset;
            src_ptr[1] = s->last_picture.data[1] + (offset >> 2);
            src_ptr[2] = s->last_picture.data[2] + (offset >> 2);
2366
        }
M
cleanup  
Michael Niedermayer 已提交
2367 2368
        emms_c();

2369 2370 2371 2372 2373
        s->avctx->draw_horiz_band(s->avctx, src_ptr, s->linesize,
                               y, s->width, h);
    }
}

2374 2375
#ifdef CONFIG_ENCODERS

2376
static void encode_mb(MpegEncContext *s, int motion_x, int motion_y)
F
Fabrice Bellard 已提交
2377
{
2378 2379 2380
    const int mb_x= s->mb_x;
    const int mb_y= s->mb_y;
    int i;
2381
    int skip_dct[6];
2382 2383
    int dct_offset   = s->linesize*8; //default for progressive frames
    
2384
    for(i=0; i<6; i++) skip_dct[i]=0;
2385 2386
    
    if(s->adaptive_quant){
M
cleanup  
Michael Niedermayer 已提交
2387
        s->dquant= s->current_picture.qscale_table[mb_x + mb_y*s->mb_width] - s->qscale;
2388 2389

        if(s->out_format==FMT_H263){
2390 2391
            if     (s->dquant> 2) s->dquant= 2;
            else if(s->dquant<-2) s->dquant=-2;
2392 2393 2394
        }
            
        if(s->codec_id==CODEC_ID_MPEG4){        
2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405
            if(!s->mb_intra){
                assert(s->dquant==0 || s->mv_type!=MV_TYPE_8X8);

                if(s->mv_dir&MV_DIRECT)
                    s->dquant=0;
            }
        }
        s->qscale+= s->dquant;
        s->y_dc_scale= s->y_dc_scale_table[ s->qscale ];
        s->c_dc_scale= s->c_dc_scale_table[ s->qscale ];
    }
2406

2407
    if (s->mb_intra) {
2408
        uint8_t *ptr;
2409
        int wrap_y;
2410
        int emu=0;
2411

2412
        wrap_y = s->linesize;
M
cleanup  
Michael Niedermayer 已提交
2413
        ptr = s->new_picture.data[0] + (mb_y * 16 * wrap_y) + mb_x * 16;
2414

2415
        if(mb_x*16+16 > s->width || mb_y*16+16 > s->height){
M
Michael Niedermayer 已提交
2416
            ff_emulated_edge_mc(s, ptr, wrap_y, 16, 16, mb_x*16, mb_y*16, s->width, s->height);
2417 2418 2419
            ptr= s->edge_emu_buffer;
            emu=1;
        }
2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435
        
        if(s->flags&CODEC_FLAG_INTERLACED_DCT){
            int progressive_score, interlaced_score;
            
            progressive_score= pix_vcmp16x8(ptr, wrap_y  ) + pix_vcmp16x8(ptr + wrap_y*8, wrap_y );
            interlaced_score = pix_vcmp16x8(ptr, wrap_y*2) + pix_vcmp16x8(ptr + wrap_y  , wrap_y*2);
            
            if(progressive_score > interlaced_score + 100){
                s->interlaced_dct=1;
            
                dct_offset= wrap_y;
                wrap_y<<=1;
            }else
                s->interlaced_dct=0;
        }
        
2436 2437 2438 2439
	s->dsp.get_pixels(s->block[0], ptr                 , wrap_y);
        s->dsp.get_pixels(s->block[1], ptr              + 8, wrap_y);
        s->dsp.get_pixels(s->block[2], ptr + dct_offset    , wrap_y);
        s->dsp.get_pixels(s->block[3], ptr + dct_offset + 8, wrap_y);
2440

M
Michael Niedermayer 已提交
2441 2442 2443 2444
        if(s->flags&CODEC_FLAG_GRAY){
            skip_dct[4]= 1;
            skip_dct[5]= 1;
        }else{
2445
            int wrap_c = s->uvlinesize;
M
cleanup  
Michael Niedermayer 已提交
2446
            ptr = s->new_picture.data[1] + (mb_y * 8 * wrap_c) + mb_x * 8;
2447
            if(emu){
M
Michael Niedermayer 已提交
2448
                ff_emulated_edge_mc(s, ptr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2449 2450
                ptr= s->edge_emu_buffer;
            }
2451
	    s->dsp.get_pixels(s->block[4], ptr, wrap_c);
2452

M
cleanup  
Michael Niedermayer 已提交
2453
            ptr = s->new_picture.data[2] + (mb_y * 8 * wrap_c) + mb_x * 8;
2454
            if(emu){
M
Michael Niedermayer 已提交
2455
                ff_emulated_edge_mc(s, ptr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2456 2457
                ptr= s->edge_emu_buffer;
            }
2458
            s->dsp.get_pixels(s->block[5], ptr, wrap_c);
M
Michael Niedermayer 已提交
2459
        }
2460
    }else{
M
Michael Niedermayer 已提交
2461 2462
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
2463 2464
        uint8_t *dest_y, *dest_cb, *dest_cr;
        uint8_t *ptr_y, *ptr_cb, *ptr_cr;
2465
        int wrap_y, wrap_c;
2466
        int emu=0;
2467

M
cleanup  
Michael Niedermayer 已提交
2468 2469 2470
        dest_y  = s->current_picture.data[0] + (mb_y * 16 * s->linesize    ) + mb_x * 16;
        dest_cb = s->current_picture.data[1] + (mb_y * 8  * (s->uvlinesize)) + mb_x * 8;
        dest_cr = s->current_picture.data[2] + (mb_y * 8  * (s->uvlinesize)) + mb_x * 8;
2471
        wrap_y = s->linesize;
2472
        wrap_c = s->uvlinesize;
M
cleanup  
Michael Niedermayer 已提交
2473 2474 2475
        ptr_y  = s->new_picture.data[0] + (mb_y * 16 * wrap_y) + mb_x * 16;
        ptr_cb = s->new_picture.data[1] + (mb_y * 8 * wrap_c) + mb_x * 8;
        ptr_cr = s->new_picture.data[2] + (mb_y * 8 * wrap_c) + mb_x * 8;
2476

2477
        if ((!s->no_rounding) || s->pict_type==B_TYPE){
2478 2479
	    op_pix = s->dsp.put_pixels_tab;
            op_qpix= s->dsp.put_qpel_pixels_tab;
M
Michael Niedermayer 已提交
2480
        }else{
2481 2482
            op_pix = s->dsp.put_no_rnd_pixels_tab;
            op_qpix= s->dsp.put_no_rnd_qpel_pixels_tab;
2483 2484 2485
        }

        if (s->mv_dir & MV_DIR_FORWARD) {
M
cleanup  
Michael Niedermayer 已提交
2486
            MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix, op_qpix);
2487 2488
            op_pix = s->dsp.avg_pixels_tab;
            op_qpix= s->dsp.avg_qpel_pixels_tab;
2489 2490
        }
        if (s->mv_dir & MV_DIR_BACKWARD) {
M
cleanup  
Michael Niedermayer 已提交
2491
            MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix, op_qpix);
2492
        }
2493

2494
        if(mb_x*16+16 > s->width || mb_y*16+16 > s->height){
M
Michael Niedermayer 已提交
2495
            ff_emulated_edge_mc(s, ptr_y, wrap_y, 16, 16, mb_x*16, mb_y*16, s->width, s->height);
2496 2497 2498
            ptr_y= s->edge_emu_buffer;
            emu=1;
        }
2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516
        
        if(s->flags&CODEC_FLAG_INTERLACED_DCT){
            int progressive_score, interlaced_score;
            
            progressive_score= pix_diff_vcmp16x8(ptr_y           , dest_y           , wrap_y  ) 
                             + pix_diff_vcmp16x8(ptr_y + wrap_y*8, dest_y + wrap_y*8, wrap_y  );
            interlaced_score = pix_diff_vcmp16x8(ptr_y           , dest_y           , wrap_y*2)
                             + pix_diff_vcmp16x8(ptr_y + wrap_y  , dest_y + wrap_y  , wrap_y*2);
            
            if(progressive_score > interlaced_score + 600){
                s->interlaced_dct=1;
            
                dct_offset= wrap_y;
                wrap_y<<=1;
            }else
                s->interlaced_dct=0;
        }
        
2517 2518 2519 2520
	s->dsp.diff_pixels(s->block[0], ptr_y                 , dest_y                 , wrap_y);
        s->dsp.diff_pixels(s->block[1], ptr_y              + 8, dest_y              + 8, wrap_y);
        s->dsp.diff_pixels(s->block[2], ptr_y + dct_offset    , dest_y + dct_offset    , wrap_y);
        s->dsp.diff_pixels(s->block[3], ptr_y + dct_offset + 8, dest_y + dct_offset + 8, wrap_y);
M
Michael Niedermayer 已提交
2521 2522 2523 2524 2525
        
        if(s->flags&CODEC_FLAG_GRAY){
            skip_dct[4]= 1;
            skip_dct[5]= 1;
        }else{
2526
            if(emu){
M
Michael Niedermayer 已提交
2527
                ff_emulated_edge_mc(s, ptr_cb, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2528 2529
                ptr_cb= s->edge_emu_buffer;
            }
2530
            s->dsp.diff_pixels(s->block[4], ptr_cb, dest_cb, wrap_c);
2531
            if(emu){
M
Michael Niedermayer 已提交
2532
                ff_emulated_edge_mc(s, ptr_cr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2533 2534
                ptr_cr= s->edge_emu_buffer;
            }
2535
            s->dsp.diff_pixels(s->block[5], ptr_cr, dest_cr, wrap_c);
M
Michael Niedermayer 已提交
2536
        }
2537
        /* pre quantization */         
M
cleanup  
Michael Niedermayer 已提交
2538
        if(s->current_picture.mc_mb_var[s->mb_width*mb_y+ mb_x]<2*s->qscale*s->qscale){
2539
            //FIXME optimize
2540 2541 2542 2543
	    if(s->dsp.pix_abs8x8(ptr_y               , dest_y               , wrap_y) < 20*s->qscale) skip_dct[0]= 1;
            if(s->dsp.pix_abs8x8(ptr_y            + 8, dest_y            + 8, wrap_y) < 20*s->qscale) skip_dct[1]= 1;
            if(s->dsp.pix_abs8x8(ptr_y +dct_offset   , dest_y +dct_offset   , wrap_y) < 20*s->qscale) skip_dct[2]= 1;
            if(s->dsp.pix_abs8x8(ptr_y +dct_offset+ 8, dest_y +dct_offset+ 8, wrap_y) < 20*s->qscale) skip_dct[3]= 1;
2544 2545
            if(s->dsp.pix_abs8x8(ptr_cb              , dest_cb              , wrap_c) < 20*s->qscale) skip_dct[4]= 1;
            if(s->dsp.pix_abs8x8(ptr_cr              , dest_cr              , wrap_c) < 20*s->qscale) skip_dct[5]= 1;
2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561
#if 0
{
 static int stat[7];
 int num=0;
 for(i=0; i<6; i++)
  if(skip_dct[i]) num++;
 stat[num]++;
 
 if(s->mb_x==0 && s->mb_y==0){
  for(i=0; i<7; i++){
   printf("%6d %1d\n", stat[i], i);
  }
 }
}
#endif
        }
2562

2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578
    }
            
#if 0
            {
                float adap_parm;
                
                adap_parm = ((s->avg_mb_var << 1) + s->mb_var[s->mb_width*mb_y+mb_x] + 1.0) /
                            ((s->mb_var[s->mb_width*mb_y+mb_x] << 1) + s->avg_mb_var + 1.0);
            
                printf("\ntype=%c qscale=%2d adap=%0.2f dquant=%4.2f var=%4d avgvar=%4d", 
                        (s->mb_type[s->mb_width*mb_y+mb_x] > 0) ? 'I' : 'P', 
                        s->qscale, adap_parm, s->qscale*adap_parm,
                        s->mb_var[s->mb_width*mb_y+mb_x], s->avg_mb_var);
            }
#endif
    /* DCT & quantize */
2579 2580 2581
    if(s->out_format==FMT_MJPEG){
        for(i=0;i<6;i++) {
            int overflow;
2582
            s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, 8, &overflow);
2583
            if (overflow) clip_coeffs(s, s->block[i], s->block_last_index[i]);
2584 2585 2586
        }
    }else{
        for(i=0;i<6;i++) {
2587 2588
            if(!skip_dct[i]){
                int overflow;
2589
                s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, s->qscale, &overflow);
2590
            // FIXME we could decide to change to quantizer instead of clipping
2591 2592
            // JS: I don't think that would be a good idea it could lower quality instead
            //     of improve it. Just INTRADC clipping deserves changes in quantizer
2593 2594 2595
                if (overflow) clip_coeffs(s, s->block[i], s->block_last_index[i]);
            }else
                s->block_last_index[i]= -1;
2596
        }
2597 2598
        if(s->luma_elim_threshold && !s->mb_intra)
            for(i=0; i<4; i++)
2599
                dct_single_coeff_elimination(s, i, s->luma_elim_threshold);
2600 2601
        if(s->chroma_elim_threshold && !s->mb_intra)
            for(i=4; i<6; i++)
2602
                dct_single_coeff_elimination(s, i, s->chroma_elim_threshold);
2603 2604
    }

M
Michael Niedermayer 已提交
2605 2606 2607 2608
    if((s->flags&CODEC_FLAG_GRAY) && s->mb_intra){
        s->block_last_index[4]=
        s->block_last_index[5]= 0;
        s->block[4][0]=
M
bugs  
Michael Niedermayer 已提交
2609
        s->block[5][0]= (1024 + s->c_dc_scale/2)/ s->c_dc_scale;
M
Michael Niedermayer 已提交
2610 2611
    }

2612
    /* huffman encode */
M
Michael Niedermayer 已提交
2613 2614 2615
    switch(s->codec_id){ //FIXME funct ptr could be slightly faster
    case CODEC_ID_MPEG1VIDEO:
        mpeg1_encode_mb(s, s->block, motion_x, motion_y); break;
M
Michael Niedermayer 已提交
2616
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628
    case CODEC_ID_MPEG4:
        mpeg4_encode_mb(s, s->block, motion_x, motion_y); break;
    case CODEC_ID_MSMPEG4V2:
    case CODEC_ID_MSMPEG4V3:
    case CODEC_ID_WMV1:
        msmpeg4_encode_mb(s, s->block, motion_x, motion_y); break;
    case CODEC_ID_WMV2:
         ff_wmv2_encode_mb(s, s->block, motion_x, motion_y); break;
    case CODEC_ID_H263:
    case CODEC_ID_H263P:
    case CODEC_ID_RV10:
        h263_encode_mb(s, s->block, motion_x, motion_y); break;
M
Michael Niedermayer 已提交
2629 2630 2631
#endif
    case CODEC_ID_MJPEG:
        mjpeg_encode_mb(s, s->block); break;
M
Michael Niedermayer 已提交
2632 2633
    default:
        assert(0);
2634 2635 2636
    }
}

2637 2638
#endif //CONFIG_ENCODERS

M
cleanup  
Michael Niedermayer 已提交
2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667
/**
 * combines the (truncated) bitstream to a complete frame
 * @returns -1 if no complete frame could be created
 */
int ff_combine_frame( MpegEncContext *s, int next, uint8_t **buf, int *buf_size){
    ParseContext *pc= &s->parse_context;
        
    pc->last_index= pc->index;

    if(next==-1){
        pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, (*buf_size) + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);

        memcpy(&pc->buffer[pc->index], *buf, *buf_size);
        pc->index += *buf_size;
        return -1;
    }

    if(pc->index){
        pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, next + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);

        memcpy(&pc->buffer[pc->index], *buf, next + FF_INPUT_BUFFER_PADDING_SIZE );
        pc->index = 0;
        *buf= pc->buffer;
        *buf_size= pc->last_index + next;
    }

    return 0;
}

2668
#ifdef CONFIG_ENCODERS
2669
void ff_copy_bits(PutBitContext *pb, uint8_t *src, int length)
2670
{
2671 2672 2673 2674
    int bytes= length>>4;
    int bits= length&15;
    int i;

2675 2676
    if(length==0) return;

2677 2678
    for(i=0; i<bytes; i++) put_bits(pb, 16, be2me_16(((uint16_t*)src)[i]));
    put_bits(pb, bits, be2me_16(((uint16_t*)src)[i])>>(16-bits));
2679 2680
}

2681
static inline void copy_context_before_encode(MpegEncContext *d, MpegEncContext *s, int type){
2682 2683 2684 2685 2686
    int i;

    memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster then a loop?

    /* mpeg1 */
2687
    d->mb_skip_run= s->mb_skip_run;
2688 2689 2690 2691 2692 2693 2694 2695
    for(i=0; i<3; i++)
        d->last_dc[i]= s->last_dc[i];
    
    /* statistics */
    d->mv_bits= s->mv_bits;
    d->i_tex_bits= s->i_tex_bits;
    d->p_tex_bits= s->p_tex_bits;
    d->i_count= s->i_count;
2696 2697
    d->f_count= s->f_count;
    d->b_count= s->b_count;
2698 2699
    d->skip_count= s->skip_count;
    d->misc_bits= s->misc_bits;
2700
    d->last_bits= 0;
2701 2702

    d->mb_skiped= s->mb_skiped;
M
Michael Niedermayer 已提交
2703
    d->qscale= s->qscale;
2704 2705
}

2706
static inline void copy_context_after_encode(MpegEncContext *d, MpegEncContext *s, int type){
2707 2708 2709 2710 2711 2712
    int i;

    memcpy(d->mv, s->mv, 2*4*2*sizeof(int)); 
    memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster then a loop?
    
    /* mpeg1 */
2713
    d->mb_skip_run= s->mb_skip_run;
2714 2715 2716 2717 2718 2719 2720 2721
    for(i=0; i<3; i++)
        d->last_dc[i]= s->last_dc[i];
    
    /* statistics */
    d->mv_bits= s->mv_bits;
    d->i_tex_bits= s->i_tex_bits;
    d->p_tex_bits= s->p_tex_bits;
    d->i_count= s->i_count;
2722 2723
    d->f_count= s->f_count;
    d->b_count= s->b_count;
2724 2725 2726 2727
    d->skip_count= s->skip_count;
    d->misc_bits= s->misc_bits;

    d->mb_intra= s->mb_intra;
2728
    d->mb_skiped= s->mb_skiped;
2729 2730 2731
    d->mv_type= s->mv_type;
    d->mv_dir= s->mv_dir;
    d->pb= s->pb;
2732 2733 2734 2735
    if(s->data_partitioning){
        d->pb2= s->pb2;
        d->tex_pb= s->tex_pb;
    }
2736 2737 2738
    d->block= s->block;
    for(i=0; i<6; i++)
        d->block_last_index[i]= s->block_last_index[i];
2739
    d->interlaced_dct= s->interlaced_dct;
M
Michael Niedermayer 已提交
2740
    d->qscale= s->qscale;
2741 2742
}

2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772
static inline void encode_mb_hq(MpegEncContext *s, MpegEncContext *backup, MpegEncContext *best, int type, 
                           PutBitContext pb[2], PutBitContext pb2[2], PutBitContext tex_pb[2],
                           int *dmin, int *next_block, int motion_x, int motion_y)
{
    int bits_count;
    
    copy_context_before_encode(s, backup, type);

    s->block= s->blocks[*next_block];
    s->pb= pb[*next_block];
    if(s->data_partitioning){
        s->pb2   = pb2   [*next_block];
        s->tex_pb= tex_pb[*next_block];
    }

    encode_mb(s, motion_x, motion_y);

    bits_count= get_bit_count(&s->pb);
    if(s->data_partitioning){
        bits_count+= get_bit_count(&s->pb2);
        bits_count+= get_bit_count(&s->tex_pb);
    }

    if(bits_count<*dmin){
        *dmin= bits_count;
        *next_block^=1;

        copy_context_after_encode(best, s, type);
    }
}
2773 2774 2775 2776 2777 2778 2779
                
static inline int sse(MpegEncContext *s, uint8_t *src1, uint8_t *src2, int w, int h, int stride){
    uint32_t *sq = squareTbl + 256;
    int acc=0;
    int x,y;
    
    if(w==16 && h==16) 
M
Michael Niedermayer 已提交
2780 2781 2782
        return s->dsp.sse[0](NULL, src1, src2, stride);
    else if(w==8 && h==8)
        return s->dsp.sse[1](NULL, src1, src2, stride);
2783 2784 2785 2786 2787 2788
    
    for(y=0; y<h; y++){
        for(x=0; x<w; x++){
            acc+= sq[src1[x + y*stride] - src2[x + y*stride]];
        } 
    }
M
Michael Niedermayer 已提交
2789 2790 2791
    
    assert(acc>=0);
    
2792 2793
    return acc;
}
2794

2795 2796
static void encode_picture(MpegEncContext *s, int picture_number)
{
M
Michael Niedermayer 已提交
2797
    int mb_x, mb_y, pdif = 0;
2798
    int i;
2799
    int bits;
2800
    MpegEncContext best_s, backup_s;
2801 2802 2803
    uint8_t bit_buf[2][3000];
    uint8_t bit_buf2[2][3000];
    uint8_t bit_buf_tex[2][3000];
2804 2805 2806 2807 2808 2809 2810
    PutBitContext pb[2], pb2[2], tex_pb[2];

    for(i=0; i<2; i++){
        init_put_bits(&pb    [i], bit_buf    [i], 3000, NULL, NULL);
        init_put_bits(&pb2   [i], bit_buf2   [i], 3000, NULL, NULL);
        init_put_bits(&tex_pb[i], bit_buf_tex[i], 3000, NULL, NULL);
    }
F
Fabrice Bellard 已提交
2811 2812

    s->picture_number = picture_number;
2813

2814 2815 2816 2817 2818 2819 2820
    s->block_wrap[0]=
    s->block_wrap[1]=
    s->block_wrap[2]=
    s->block_wrap[3]= s->mb_width*2 + 2;
    s->block_wrap[4]=
    s->block_wrap[5]= s->mb_width + 2;
    
2821
    /* Reset the average MB variance */
M
cleanup  
Michael Niedermayer 已提交
2822 2823
    s->current_picture.mb_var_sum = 0;
    s->current_picture.mc_mb_var_sum = 0;
2824

M
Michael Niedermayer 已提交
2825
#ifdef CONFIG_RISKY
2826
    /* we need to initialize some time vars before we can encode b-frames */
2827 2828
    // RAL: Condition added for MPEG1VIDEO
    if (s->codec_id == CODEC_ID_MPEG1VIDEO || (s->h263_pred && !s->h263_msmpeg4))
2829
        ff_set_mpeg4_time(s, s->picture_number); 
M
Michael Niedermayer 已提交
2830 2831
#endif
        
2832
    s->scene_change_score=0;
2833 2834
    
    s->qscale= (int)(s->frame_qscale + 0.5); //FIXME qscale / ... stuff for ME ratedistoration
M
Michael Niedermayer 已提交
2835
    
M
Michael Niedermayer 已提交
2836 2837 2838 2839 2840
    if(s->pict_type==I_TYPE){
        if(s->msmpeg4_version) s->no_rounding=1;
        else                   s->no_rounding=0;
    }else if(s->pict_type!=B_TYPE){
        if(s->flipflop_rounding || s->codec_id == CODEC_ID_H263P || s->codec_id == CODEC_ID_MPEG4)
M
Michael Niedermayer 已提交
2841 2842
            s->no_rounding ^= 1;          
    }
M
Michael Niedermayer 已提交
2843
    
2844
    /* Estimate motion for every MB */
2845
    s->mb_intra=0; //for the rate distoration & bit compare functions
2846
    if(s->pict_type != I_TYPE){
M
Michael Niedermayer 已提交
2847 2848
        if(s->pict_type != B_TYPE){
            if((s->avctx->pre_me && s->last_non_b_pict_type==I_TYPE) || s->avctx->pre_me==2){
2849
                s->me.pre_pass=1;
M
Michael Niedermayer 已提交
2850
                s->me.dia_size= s->avctx->pre_dia_size;
2851

M
Michael Niedermayer 已提交
2852 2853 2854 2855 2856 2857 2858
                for(mb_y=s->mb_height-1; mb_y >=0 ; mb_y--) {
                    for(mb_x=s->mb_width-1; mb_x >=0 ; mb_x--) {
                        s->mb_x = mb_x;
                        s->mb_y = mb_y;
                        ff_pre_estimate_p_frame_motion(s, mb_x, mb_y);
                    }
                }
2859
                s->me.pre_pass=0;
M
Michael Niedermayer 已提交
2860 2861 2862
            }
        }

M
Michael Niedermayer 已提交
2863
        s->me.dia_size= s->avctx->dia_size;
2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875
        for(mb_y=0; mb_y < s->mb_height; mb_y++) {
            s->block_index[0]= s->block_wrap[0]*(mb_y*2 + 1) - 1;
            s->block_index[1]= s->block_wrap[0]*(mb_y*2 + 1);
            s->block_index[2]= s->block_wrap[0]*(mb_y*2 + 2) - 1;
            s->block_index[3]= s->block_wrap[0]*(mb_y*2 + 2);
            for(mb_x=0; mb_x < s->mb_width; mb_x++) {
                s->mb_x = mb_x;
                s->mb_y = mb_y;
                s->block_index[0]+=2;
                s->block_index[1]+=2;
                s->block_index[2]+=2;
                s->block_index[3]+=2;
M
Michael Niedermayer 已提交
2876
                
2877
                /* compute motion vector & mb_type and store in context */
2878 2879 2880 2881
                if(s->pict_type==B_TYPE)
                    ff_estimate_b_frame_motion(s, mb_x, mb_y);
                else
                    ff_estimate_p_frame_motion(s, mb_x, mb_y);
2882 2883
            }
        }
2884
    }else /* if(s->pict_type == I_TYPE) */{
2885 2886
        /* I-Frame */
        //FIXME do we need to zero them?
2887 2888 2889
        memset(s->motion_val[0], 0, sizeof(int16_t)*(s->mb_width*2 + 2)*(s->mb_height*2 + 2)*2);
        memset(s->p_mv_table   , 0, sizeof(int16_t)*(s->mb_width+2)*(s->mb_height+2)*2);
        memset(s->mb_type      , MB_TYPE_INTRA, sizeof(uint8_t)*s->mb_width*s->mb_height);
M
Michael Niedermayer 已提交
2890 2891 2892 2893 2894 2895 2896
        
        if(!s->fixed_qscale){
            /* finding spatial complexity for I-frame rate control */
            for(mb_y=0; mb_y < s->mb_height; mb_y++) {
                for(mb_x=0; mb_x < s->mb_width; mb_x++) {
                    int xx = mb_x * 16;
                    int yy = mb_y * 16;
M
cleanup  
Michael Niedermayer 已提交
2897
                    uint8_t *pix = s->new_picture.data[0] + (yy * s->linesize) + xx;
M
Michael Niedermayer 已提交
2898
                    int varc;
2899
		    int sum = s->dsp.pix_sum(pix, s->linesize);
M
Michael Niedermayer 已提交
2900
    
2901
		    varc = (s->dsp.pix_norm1(pix, s->linesize) - (((unsigned)(sum*sum))>>8) + 500 + 128)>>8;
M
Michael Niedermayer 已提交
2902

M
cleanup  
Michael Niedermayer 已提交
2903 2904 2905
                    s->current_picture.mb_var [s->mb_width * mb_y + mb_x] = varc;
                    s->current_picture.mb_mean[s->mb_width * mb_y + mb_x] = (sum+128)>>8;
                    s->current_picture.mb_var_sum    += varc;
M
Michael Niedermayer 已提交
2906 2907 2908
                }
            }
        }
2909
    }
2910 2911
    emms_c();

2912
    if(s->scene_change_score > 0 && s->pict_type == P_TYPE){
M
Michael Niedermayer 已提交
2913
        s->pict_type= I_TYPE;
2914
        memset(s->mb_type   , MB_TYPE_INTRA, sizeof(uint8_t)*s->mb_width*s->mb_height);
M
cleanup  
Michael Niedermayer 已提交
2915
//printf("Scene change detected, encoding as I Frame %d %d\n", s->current_picture.mb_var_sum, s->current_picture.mc_mb_var_sum);
M
Michael Niedermayer 已提交
2916
    }
M
cleanup  
Michael Niedermayer 已提交
2917

M
Michael Niedermayer 已提交
2918
    if(!s->umvplus){
2919 2920
        if(s->pict_type==P_TYPE || s->pict_type==S_TYPE) {
            s->f_code= ff_get_best_fcode(s, s->p_mv_table, MB_TYPE_INTER);
2921
        
2922 2923
            ff_fix_long_p_mvs(s);
        }
2924

2925 2926
        if(s->pict_type==B_TYPE){
            int a, b;
2927

2928 2929 2930
            a = ff_get_best_fcode(s, s->b_forw_mv_table, MB_TYPE_FORWARD);
            b = ff_get_best_fcode(s, s->b_bidir_forw_mv_table, MB_TYPE_BIDIR);
            s->f_code = FFMAX(a, b);
2931

2932 2933 2934
            a = ff_get_best_fcode(s, s->b_back_mv_table, MB_TYPE_BACKWARD);
            b = ff_get_best_fcode(s, s->b_bidir_back_mv_table, MB_TYPE_BIDIR);
            s->b_code = FFMAX(a, b);
2935

2936 2937 2938 2939 2940
            ff_fix_long_b_mvs(s, s->b_forw_mv_table, s->f_code, MB_TYPE_FORWARD);
            ff_fix_long_b_mvs(s, s->b_back_mv_table, s->b_code, MB_TYPE_BACKWARD);
            ff_fix_long_b_mvs(s, s->b_bidir_forw_mv_table, s->f_code, MB_TYPE_BIDIR);
            ff_fix_long_b_mvs(s, s->b_bidir_back_mv_table, s->b_code, MB_TYPE_BIDIR);
        }
2941
    }
2942
    
2943
    if (s->fixed_qscale) 
M
cleanup  
Michael Niedermayer 已提交
2944
        s->frame_qscale = s->current_picture.quality;
2945 2946
    else
        s->frame_qscale = ff_rate_estimate_qscale(s);
2947

2948
    if(s->adaptive_quant){
M
Michael Niedermayer 已提交
2949
#ifdef CONFIG_RISKY
2950 2951 2952 2953 2954 2955 2956 2957 2958
        switch(s->codec_id){
        case CODEC_ID_MPEG4:
            ff_clean_mpeg4_qscales(s);
            break;
        case CODEC_ID_H263:
        case CODEC_ID_H263P:
            ff_clean_h263_qscales(s);
            break;
        }
M
Michael Niedermayer 已提交
2959
#endif
2960

M
cleanup  
Michael Niedermayer 已提交
2961
        s->qscale= s->current_picture.qscale_table[0];
2962
    }else
2963 2964
        s->qscale= (int)(s->frame_qscale + 0.5);
        
F
Fabrice Bellard 已提交
2965 2966
    if (s->out_format == FMT_MJPEG) {
        /* for mjpeg, we do include qscale in the matrix */
2967
        s->intra_matrix[0] = ff_mpeg1_default_intra_matrix[0];
2968
        for(i=1;i<64;i++){
2969
            int j= s->dsp.idct_permutation[i];
2970 2971 2972

            s->intra_matrix[j] = CLAMP_TO_8BIT((ff_mpeg1_default_intra_matrix[i] * s->qscale) >> 3);
        }
2973
        convert_matrix(s, s->q_intra_matrix, s->q_intra_matrix16, 
2974
                       s->q_intra_matrix16_bias, s->intra_matrix, s->intra_quant_bias, 8, 8);
F
Fabrice Bellard 已提交
2975
    }
M
cleanup  
Michael Niedermayer 已提交
2976 2977 2978 2979 2980 2981 2982
    
    //FIXME var duplication
    s->current_picture.key_frame= s->pict_type == I_TYPE;
    s->current_picture.pict_type= s->pict_type;

    if(s->current_picture.key_frame)
        s->picture_in_gop_number=0;
F
Fabrice Bellard 已提交
2983

2984
    s->last_bits= get_bit_count(&s->pb);
F
Fabrice Bellard 已提交
2985 2986 2987 2988
    switch(s->out_format) {
    case FMT_MJPEG:
        mjpeg_picture_header(s);
        break;
M
Michael Niedermayer 已提交
2989
#ifdef CONFIG_RISKY
F
Fabrice Bellard 已提交
2990
    case FMT_H263:
M
Michael Niedermayer 已提交
2991 2992 2993
        if (s->codec_id == CODEC_ID_WMV2) 
            ff_wmv2_encode_picture_header(s, picture_number);
        else if (s->h263_msmpeg4) 
F
Fabrice Bellard 已提交
2994 2995 2996 2997 2998 2999 3000 3001
            msmpeg4_encode_picture_header(s, picture_number);
        else if (s->h263_pred)
            mpeg4_encode_picture_header(s, picture_number);
        else if (s->h263_rv10) 
            rv10_encode_picture_header(s, picture_number);
        else
            h263_encode_picture_header(s, picture_number);
        break;
M
Michael Niedermayer 已提交
3002
#endif
F
Fabrice Bellard 已提交
3003 3004 3005 3006
    case FMT_MPEG1:
        mpeg1_encode_picture_header(s, picture_number);
        break;
    }
3007 3008 3009 3010 3011 3012 3013 3014
    bits= get_bit_count(&s->pb);
    s->header_bits= bits - s->last_bits;
    s->last_bits= bits;
    s->mv_bits=0;
    s->misc_bits=0;
    s->i_tex_bits=0;
    s->p_tex_bits=0;
    s->i_count=0;
3015 3016
    s->f_count=0;
    s->b_count=0;
3017 3018
    s->skip_count=0;

3019 3020 3021 3022 3023
    for(i=0; i<3; i++){
        /* init last dc values */
        /* note: quant matrix value (8) is implied here */
        s->last_dc[i] = 128;
        
M
10l  
Michael Niedermayer 已提交
3024
        s->current_picture_ptr->error[i] = 0;
3025
    }
3026
    s->mb_skip_run = 0;
F
Fabrice Bellard 已提交
3027 3028
    s->last_mv[0][0][0] = 0;
    s->last_mv[0][0][1] = 0;
3029 3030 3031 3032
    s->last_mv[1][0][0] = 0;
    s->last_mv[1][0][1] = 0;
     
    s->last_mv_dir = 0;
F
Fabrice Bellard 已提交
3033

M
Michael Niedermayer 已提交
3034
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
3035
    if (s->codec_id==CODEC_ID_H263 || s->codec_id==CODEC_ID_H263P)
3036
        s->gob_index = ff_h263_get_gob_height(s);
3037

3038
    if(s->codec_id==CODEC_ID_MPEG4 && s->partitioned_frame)
3039
        ff_mpeg4_init_partitions(s);
M
Michael Niedermayer 已提交
3040
#endif
3041 3042 3043

    s->resync_mb_x=0;
    s->resync_mb_y=0;
3044
    s->first_slice_line = 1;
M
Michael Niedermayer 已提交
3045
    s->ptr_lastgob = s->pb.buf;
3046
    for(mb_y=0; mb_y < s->mb_height; mb_y++) {
M
Michael Niedermayer 已提交
3047 3048
        s->y_dc_scale= s->y_dc_scale_table[ s->qscale ];
        s->c_dc_scale= s->c_dc_scale_table[ s->qscale ];
3049
        
M
Michael Niedermayer 已提交
3050 3051 3052 3053 3054 3055
        s->block_index[0]= s->block_wrap[0]*(mb_y*2 + 1) - 1;
        s->block_index[1]= s->block_wrap[0]*(mb_y*2 + 1);
        s->block_index[2]= s->block_wrap[0]*(mb_y*2 + 2) - 1;
        s->block_index[3]= s->block_wrap[0]*(mb_y*2 + 2);
        s->block_index[4]= s->block_wrap[4]*(mb_y + 1)                    + s->block_wrap[0]*(s->mb_height*2 + 2);
        s->block_index[5]= s->block_wrap[4]*(mb_y + 1 + s->mb_height + 2) + s->block_wrap[0]*(s->mb_height*2 + 2);
3056
        for(mb_x=0; mb_x < s->mb_width; mb_x++) {
3057
            int mb_type= s->mb_type[mb_y * s->mb_width + mb_x];
3058
            const int xy= (mb_y+1) * (s->mb_width+2) + mb_x + 1;
3059
//            int d;
3060
            int dmin=10000000;
3061 3062 3063

            s->mb_x = mb_x;
            s->mb_y = mb_y;
M
Michael Niedermayer 已提交
3064 3065 3066 3067 3068 3069
            s->block_index[0]+=2;
            s->block_index[1]+=2;
            s->block_index[2]+=2;
            s->block_index[3]+=2;
            s->block_index[4]++;
            s->block_index[5]++;
M
Michael Niedermayer 已提交
3070 3071

            /* write gob / video packet header  */
M
Michael Niedermayer 已提交
3072
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
3073 3074 3075 3076 3077 3078 3079
            if(s->rtp_mode){
                int current_packet_size, is_gob_start;
                
                current_packet_size= pbBufPtr(&s->pb) - s->ptr_lastgob;
                is_gob_start=0;
                
                if(s->codec_id==CODEC_ID_MPEG4){
3080
                    if(current_packet_size >= s->rtp_payload_size
M
Michael Niedermayer 已提交
3081 3082
                       && s->mb_y + s->mb_x>0){

3083
                        if(s->partitioned_frame){
3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094
                            ff_mpeg4_merge_partitions(s);
                            ff_mpeg4_init_partitions(s);
                        }
                        ff_mpeg4_encode_video_packet_header(s);

                        if(s->flags&CODEC_FLAG_PASS1){
                            int bits= get_bit_count(&s->pb);
                            s->misc_bits+= bits - s->last_bits;
                            s->last_bits= bits;
                        }
                        ff_mpeg4_clean_buffers(s);
M
Michael Niedermayer 已提交
3095
                        is_gob_start=1;
3096
                    }
3097 3098 3099 3100 3101 3102 3103
                }else if(s->codec_id==CODEC_ID_MPEG1VIDEO){
                    if(   current_packet_size >= s->rtp_payload_size 
                       && s->mb_y + s->mb_x>0 && s->mb_skip_run==0){
                        ff_mpeg1_encode_slice_header(s);
                        ff_mpeg1_clean_buffers(s);
                        is_gob_start=1;
                    }
M
Michael Niedermayer 已提交
3104
                }else{
3105
                    if(current_packet_size >= s->rtp_payload_size
M
Michael Niedermayer 已提交
3106 3107 3108 3109 3110 3111 3112 3113
                       && s->mb_x==0 && s->mb_y>0 && s->mb_y%s->gob_index==0){
                       
                        h263_encode_gob_header(s, mb_y);                       
                        is_gob_start=1;
                    }
                }

                if(is_gob_start){
3114 3115 3116 3117 3118
                    s->ptr_lastgob = pbBufPtr(&s->pb);
                    s->first_slice_line=1;
                    s->resync_mb_x=mb_x;
                    s->resync_mb_y=mb_y;
                }
3119
            }
M
Michael Niedermayer 已提交
3120
#endif
3121

3122 3123 3124
            if(  (s->resync_mb_x   == s->mb_x)
               && s->resync_mb_y+1 == s->mb_y){
                s->first_slice_line=0; 
3125 3126
            }

3127
            if(mb_type & (mb_type-1)){ // more than 1 MB type possible
3128
                int next_block=0;
3129
                int pb_bits_count, pb2_bits_count, tex_pb_bits_count;
3130 3131

                copy_context_before_encode(&backup_s, s, -1);
3132 3133
                backup_s.pb= s->pb;
                best_s.data_partitioning= s->data_partitioning;
3134
                best_s.partitioned_frame= s->partitioned_frame;
3135 3136 3137 3138
                if(s->data_partitioning){
                    backup_s.pb2= s->pb2;
                    backup_s.tex_pb= s->tex_pb;
                }
3139

3140
                if(mb_type&MB_TYPE_INTER){
3141
                    s->mv_dir = MV_DIR_FORWARD;
M
Michael Niedermayer 已提交
3142
                    s->mv_type = MV_TYPE_16X16;
3143
                    s->mb_intra= 0;
3144 3145
                    s->mv[0][0][0] = s->p_mv_table[xy][0];
                    s->mv[0][0][1] = s->p_mv_table[xy][1];
3146 3147
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_INTER, pb, pb2, tex_pb, 
                                 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
F
Fabrice Bellard 已提交
3148
                }
3149
                if(mb_type&MB_TYPE_INTER4V){                 
3150
                    s->mv_dir = MV_DIR_FORWARD;
M
Michael Niedermayer 已提交
3151 3152 3153 3154 3155 3156
                    s->mv_type = MV_TYPE_8X8;
                    s->mb_intra= 0;
                    for(i=0; i<4; i++){
                        s->mv[0][i][0] = s->motion_val[s->block_index[i]][0];
                        s->mv[0][i][1] = s->motion_val[s->block_index[i]][1];
                    }
3157 3158
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_INTER4V, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
3159 3160 3161 3162 3163 3164 3165
                }
                if(mb_type&MB_TYPE_FORWARD){
                    s->mv_dir = MV_DIR_FORWARD;
                    s->mv_type = MV_TYPE_16X16;
                    s->mb_intra= 0;
                    s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
                    s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
3166 3167
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_FORWARD, pb, pb2, tex_pb, 
                                 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
3168 3169 3170 3171 3172 3173 3174
                }
                if(mb_type&MB_TYPE_BACKWARD){
                    s->mv_dir = MV_DIR_BACKWARD;
                    s->mv_type = MV_TYPE_16X16;
                    s->mb_intra= 0;
                    s->mv[1][0][0] = s->b_back_mv_table[xy][0];
                    s->mv[1][0][1] = s->b_back_mv_table[xy][1];
3175 3176
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_BACKWARD, pb, pb2, tex_pb, 
                                 &dmin, &next_block, s->mv[1][0][0], s->mv[1][0][1]);
3177 3178 3179 3180 3181 3182 3183 3184 3185
                }
                if(mb_type&MB_TYPE_BIDIR){
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
                    s->mv_type = MV_TYPE_16X16;
                    s->mb_intra= 0;
                    s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
                    s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
                    s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
                    s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
3186 3187
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_BIDIR, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
3188 3189
                }
                if(mb_type&MB_TYPE_DIRECT){
M
Michael Niedermayer 已提交
3190 3191 3192
                    int mx= s->b_direct_mv_table[xy][0];
                    int my= s->b_direct_mv_table[xy][1];
                    
3193 3194
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
                    s->mb_intra= 0;
M
Michael Niedermayer 已提交
3195
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
3196
                    ff_mpeg4_set_direct_mv(s, mx, my);
M
Michael Niedermayer 已提交
3197
#endif
3198
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_DIRECT, pb, pb2, tex_pb, 
M
Michael Niedermayer 已提交
3199
                                 &dmin, &next_block, mx, my);
M
Michael Niedermayer 已提交
3200
                }
3201
                if(mb_type&MB_TYPE_INTRA){
3202
                    s->mv_dir = 0;
M
Michael Niedermayer 已提交
3203
                    s->mv_type = MV_TYPE_16X16;
3204 3205 3206
                    s->mb_intra= 1;
                    s->mv[0][0][0] = 0;
                    s->mv[0][0][1] = 0;
3207 3208
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_INTRA, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
3209 3210 3211
                    /* force cleaning of ac/dc pred stuff if needed ... */
                    if(s->h263_pred || s->h263_aic)
                        s->mbintra_table[mb_x + mb_y*s->mb_width]=1;
M
Michael Niedermayer 已提交
3212
                }
3213
                copy_context_after_encode(s, &best_s, -1);
3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230
                
                pb_bits_count= get_bit_count(&s->pb);
                flush_put_bits(&s->pb);
                ff_copy_bits(&backup_s.pb, bit_buf[next_block^1], pb_bits_count);
                s->pb= backup_s.pb;
                
                if(s->data_partitioning){
                    pb2_bits_count= get_bit_count(&s->pb2);
                    flush_put_bits(&s->pb2);
                    ff_copy_bits(&backup_s.pb2, bit_buf2[next_block^1], pb2_bits_count);
                    s->pb2= backup_s.pb2;
                    
                    tex_pb_bits_count= get_bit_count(&s->tex_pb);
                    flush_put_bits(&s->tex_pb);
                    ff_copy_bits(&backup_s.tex_pb, bit_buf_tex[next_block^1], tex_pb_bits_count);
                    s->tex_pb= backup_s.tex_pb;
                }
3231
                s->last_bits= get_bit_count(&s->pb);
F
Fabrice Bellard 已提交
3232
            } else {
3233
                int motion_x, motion_y;
3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317
                int intra_score;
                int inter_score= s->current_picture.mb_cmp_score[mb_x + mb_y*s->mb_width];
                
              if(!(s->flags&CODEC_FLAG_HQ) && s->pict_type==P_TYPE){
                /* get luma score */
                if((s->avctx->mb_cmp&0xFF)==FF_CMP_SSE){
                    intra_score= (s->current_picture.mb_var[mb_x + mb_y*s->mb_width]<<8) - 500; //FIXME dont scale it down so we dont have to fix it
                }else{
                    uint8_t *dest_y;

                    int mean= s->current_picture.mb_mean[mb_x + mb_y*s->mb_width]; //FIXME
                    mean*= 0x01010101;
                    
                    dest_y  = s->new_picture.data[0] + (mb_y * 16 * s->linesize    ) + mb_x * 16;
                
                    for(i=0; i<16; i++){
                        *(uint32_t*)(&s->me.scratchpad[i*s->linesize+ 0]) = mean;
                        *(uint32_t*)(&s->me.scratchpad[i*s->linesize+ 4]) = mean;
                        *(uint32_t*)(&s->me.scratchpad[i*s->linesize+ 8]) = mean;
                        *(uint32_t*)(&s->me.scratchpad[i*s->linesize+12]) = mean;
                    }

                    s->mb_intra=1;
                    intra_score= s->dsp.mb_cmp[0](s, s->me.scratchpad, dest_y, s->linesize);
                                        
/*                    printf("intra:%7d inter:%7d var:%7d mc_var.%7d\n", intra_score>>8, inter_score>>8, 
                        s->current_picture.mb_var[mb_x + mb_y*s->mb_width],
                        s->current_picture.mc_mb_var[mb_x + mb_y*s->mb_width]);*/
                }
                
                /* get chroma score */
                if(s->avctx->mb_cmp&FF_CMP_CHROMA){
                    int i;
                    
                    s->mb_intra=1;
                    for(i=1; i<3; i++){
                        uint8_t *dest_c;
                        int mean;
                        
                        if(s->out_format == FMT_H263){
                            mean= (s->dc_val[i][mb_x + (mb_y+1)*(s->mb_width+2)] + 4)>>3; //FIXME not exact but simple ;)
                        }else{
                            mean= (s->last_dc[i] + 4)>>3;
                        }
                        dest_c = s->new_picture.data[i] + (mb_y * 8  * (s->uvlinesize)) + mb_x * 8;
                        
                        mean*= 0x01010101;
                        for(i=0; i<8; i++){
                            *(uint32_t*)(&s->me.scratchpad[i*s->uvlinesize+ 0]) = mean;
                            *(uint32_t*)(&s->me.scratchpad[i*s->uvlinesize+ 4]) = mean;
                        }
                        
                        intra_score+= s->dsp.mb_cmp[1](s, s->me.scratchpad, dest_c, s->uvlinesize);
                    }                
                }

                /* bias */
                switch(s->avctx->mb_cmp&0xFF){
                default:
                case FF_CMP_SAD:
                    intra_score+= 32*s->qscale;
                    break;
                case FF_CMP_SSE:
                    intra_score+= 24*s->qscale*s->qscale;
                    break;
                case FF_CMP_SATD:
                    intra_score+= 96*s->qscale;
                    break;
                case FF_CMP_DCT:
                    intra_score+= 48*s->qscale;
                    break;
                case FF_CMP_BIT:
                    intra_score+= 16;
                    break;
                case FF_CMP_PSNR:
                case FF_CMP_RD:
                    intra_score+= (s->qscale*s->qscale*109*8 + 64)>>7;
                    break;
                }

                if(intra_score < inter_score)
                    mb_type= MB_TYPE_INTRA;
              }  
                
3318
                s->mv_type=MV_TYPE_16X16;
3319
                // only one MB-Type possible
3320
                
3321 3322
                switch(mb_type){
                case MB_TYPE_INTRA:
3323
                    s->mv_dir = 0;
3324
                    s->mb_intra= 1;
3325 3326
                    motion_x= s->mv[0][0][0] = 0;
                    motion_y= s->mv[0][0][1] = 0;
3327 3328
                    break;
                case MB_TYPE_INTER:
3329 3330 3331 3332
                    s->mv_dir = MV_DIR_FORWARD;
                    s->mb_intra= 0;
                    motion_x= s->mv[0][0][0] = s->p_mv_table[xy][0];
                    motion_y= s->mv[0][0][1] = s->p_mv_table[xy][1];
3333
                    break;
3334 3335 3336 3337 3338 3339 3340 3341 3342 3343
                case MB_TYPE_INTER4V:
                    s->mv_dir = MV_DIR_FORWARD;
                    s->mv_type = MV_TYPE_8X8;
                    s->mb_intra= 0;
                    for(i=0; i<4; i++){
                        s->mv[0][i][0] = s->motion_val[s->block_index[i]][0];
                        s->mv[0][i][1] = s->motion_val[s->block_index[i]][1];
                    }
                    motion_x= motion_y= 0;
                    break;
3344
                case MB_TYPE_DIRECT:
3345 3346
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
                    s->mb_intra= 0;
3347 3348
                    motion_x=s->b_direct_mv_table[xy][0];
                    motion_y=s->b_direct_mv_table[xy][1];
M
Michael Niedermayer 已提交
3349
#ifdef CONFIG_RISKY
M
Michael Niedermayer 已提交
3350
                    ff_mpeg4_set_direct_mv(s, motion_x, motion_y);
M
Michael Niedermayer 已提交
3351
#endif
3352 3353
                    break;
                case MB_TYPE_BIDIR:
3354
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
3355
                    s->mb_intra= 0;
3356 3357 3358 3359 3360 3361
                    motion_x=0;
                    motion_y=0;
                    s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
                    s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
                    s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
                    s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
3362 3363
                    break;
                case MB_TYPE_BACKWARD:
3364 3365 3366 3367
                    s->mv_dir = MV_DIR_BACKWARD;
                    s->mb_intra= 0;
                    motion_x= s->mv[1][0][0] = s->b_back_mv_table[xy][0];
                    motion_y= s->mv[1][0][1] = s->b_back_mv_table[xy][1];
3368 3369
                    break;
                case MB_TYPE_FORWARD:
3370 3371 3372 3373 3374
                    s->mv_dir = MV_DIR_FORWARD;
                    s->mb_intra= 0;
                    motion_x= s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
                    motion_y= s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
//                    printf(" %d %d ", motion_x, motion_y);
3375 3376
                    break;
                default:
3377 3378
                    motion_x=motion_y=0; //gcc warning fix
                    printf("illegal MB type\n");
3379
                }
3380

3381
                encode_mb(s, motion_x, motion_y);
3382 3383 3384

                // RAL: Update last macrobloc type
                s->last_mv_dir = s->mv_dir;
F
Fabrice Bellard 已提交
3385
            }
3386

3387 3388 3389 3390 3391
            /* clean the MV table in IPS frames for direct mode in B frames */
            if(s->mb_intra /* && I,P,S_TYPE */){
                s->p_mv_table[xy][0]=0;
                s->p_mv_table[xy][1]=0;
            }
F
Fabrice Bellard 已提交
3392

3393
            MPV_decode_mb(s, s->block);
3394 3395 3396 3397 3398 3399 3400
            
            if(s->flags&CODEC_FLAG_PSNR){
                int w= 16;
                int h= 16;

                if(s->mb_x*16 + 16 > s->width ) w= s->width - s->mb_x*16;
                if(s->mb_y*16 + 16 > s->height) h= s->height- s->mb_y*16;
M
Michael Niedermayer 已提交
3401

M
10l  
Michael Niedermayer 已提交
3402
                s->current_picture_ptr->error[0] += sse(
3403 3404 3405 3406
                    s,
                    s->new_picture    .data[0] + s->mb_x*16 + s->mb_y*s->linesize*16,
                    s->current_picture.data[0] + s->mb_x*16 + s->mb_y*s->linesize*16,
                    w, h, s->linesize);
M
10l  
Michael Niedermayer 已提交
3407
                s->current_picture_ptr->error[1] += sse(
3408 3409 3410 3411
                    s,
                    s->new_picture    .data[1] + s->mb_x*8  + s->mb_y*s->uvlinesize*8,
                    s->current_picture.data[1] + s->mb_x*8  + s->mb_y*s->uvlinesize*8,
                    w>>1, h>>1, s->uvlinesize);
M
10l  
Michael Niedermayer 已提交
3412
                s->current_picture_ptr->error[2] += sse(
3413 3414 3415 3416 3417
                    s,
                    s->new_picture    .data[2] + s->mb_x*8  + s->mb_y*s->uvlinesize*8,
                    s->current_picture.data[2] + s->mb_x*8  + s->mb_y*s->uvlinesize*8,
                    w>>1, h>>1, s->uvlinesize);
            }
3418
//printf("MB %d %d bits\n", s->mb_x+s->mb_y*s->mb_width, get_bit_count(&s->pb));
F
Fabrice Bellard 已提交
3419 3420
        }
    }
3421
    emms_c();
3422

M
Michael Niedermayer 已提交
3423
#ifdef CONFIG_RISKY
3424
    if(s->codec_id==CODEC_ID_MPEG4 && s->partitioned_frame)
3425 3426 3427
        ff_mpeg4_merge_partitions(s);

    if (s->msmpeg4_version && s->msmpeg4_version<4 && s->pict_type == I_TYPE)
M
Michael Niedermayer 已提交
3428 3429
        msmpeg4_encode_ext_header(s);

3430 3431
    if(s->codec_id==CODEC_ID_MPEG4) 
        ff_mpeg4_stuffing(&s->pb);
M
Michael Niedermayer 已提交
3432
#endif
3433

3434 3435
    //if (s->gob_number)
    //    fprintf(stderr,"\nNumber of GOB: %d", s->gob_number);
3436 3437 3438 3439
    
    /* Send the last GOB if RTP */    
    if (s->rtp_mode) {
        flush_put_bits(&s->pb);
3440
        pdif = pbBufPtr(&s->pb) - s->ptr_lastgob;
3441 3442 3443
        /* Call the RTP callback to send the last GOB */
        if (s->rtp_callback)
            s->rtp_callback(s->ptr_lastgob, pdif, s->gob_number);
3444
        s->ptr_lastgob = pbBufPtr(&s->pb);
3445 3446
        //fprintf(stderr,"\nGOB: %2d size: %d (last)", s->gob_number, pdif);
    }
F
Fabrice Bellard 已提交
3447 3448
}

M
Michael Niedermayer 已提交
3449 3450 3451 3452
static int dct_quantize_trellis_c(MpegEncContext *s, 
                        DCTELEM *block, int n,
                        int qscale, int *overflow){
    const int *qmat;
3453
    const uint8_t *scantable= s->intra_scantable.scantable;
M
Michael Niedermayer 已提交
3454 3455 3456 3457 3458 3459
    int max=0;
    unsigned int threshold1, threshold2;
    int bias=0;
    int run_tab[65];
    int level_tab[65];
    int score_tab[65];
3460 3461 3462 3463 3464
    int last_run=0;
    int last_level=0;
    int last_score= 0;
    int last_i= 0;
    int coeff[3][64];
M
Michael Niedermayer 已提交
3465
    int coeff_count[64];
3466
    int lambda, qmul, qadd, start_i, last_non_zero, i;
M
Michael Niedermayer 已提交
3467 3468 3469
    const int esc_length= s->ac_esc_length;
    uint8_t * length;
    uint8_t * last_length;
3470 3471
    int score_limit=0;
    int left_limit= 0;
M
Michael Niedermayer 已提交
3472
        
3473
    s->dsp.fdct (block);
M
Michael Niedermayer 已提交
3474 3475 3476

    qmul= qscale*16;
    qadd= ((qscale-1)|1)*8;
3477

M
Michael Niedermayer 已提交
3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496
    if (s->mb_intra) {
        int q;
        if (!s->h263_aic) {
            if (n < 4)
                q = s->y_dc_scale;
            else
                q = s->c_dc_scale;
            q = q << 3;
        } else{
            /* For AIC we skip quant/dequant of INTRADC */
            q = 1 << 3;
            qadd=0;
        }
            
        /* note: block[0] is assumed to be positive */
        block[0] = (block[0] + (q >> 1)) / q;
        start_i = 1;
        last_non_zero = 0;
        qmat = s->q_intra_matrix[qscale];
3497
        if(s->mpeg_quant || s->codec_id== CODEC_ID_MPEG1VIDEO)
M
Michael Niedermayer 已提交
3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510
            bias= 1<<(QMAT_SHIFT-1);
        length     = s->intra_ac_vlc_length;
        last_length= s->intra_ac_vlc_last_length;
    } else {
        start_i = 0;
        last_non_zero = -1;
        qmat = s->q_inter_matrix[qscale];
        length     = s->inter_ac_vlc_length;
        last_length= s->inter_ac_vlc_last_length;
    }

    threshold1= (1<<QMAT_SHIFT) - bias - 1;
    threshold2= (threshold1<<1);
3511

M
Michael Niedermayer 已提交
3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524
    for(i=start_i; i<64; i++) {
        const int j = scantable[i];
        const int k= i-start_i;
        int level = block[j];
        level = level * qmat[j];

//        if(   bias+level >= (1<<(QMAT_SHIFT - 3))
//           || bias-level >= (1<<(QMAT_SHIFT - 3))){
        if(((unsigned)(level+threshold1))>threshold2){
            if(level>0){
                level= (bias + level)>>QMAT_SHIFT;
                coeff[0][k]= level;
                coeff[1][k]= level-1;
3525
//                coeff[2][k]= level-2;
M
Michael Niedermayer 已提交
3526 3527 3528 3529
            }else{
                level= (bias - level)>>QMAT_SHIFT;
                coeff[0][k]= -level;
                coeff[1][k]= -level+1;
3530
//                coeff[2][k]= -level+2;
M
Michael Niedermayer 已提交
3531
            }
3532
            coeff_count[k]= FFMIN(level, 2);
M
Michael Niedermayer 已提交
3533 3534 3535
            max |=level;
            last_non_zero = i;
        }else{
3536
            coeff[0][k]= (level>>31)|1;
M
Michael Niedermayer 已提交
3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547
            coeff_count[k]= 1;
        }
    }
    
    *overflow= s->max_qcoeff < max; //overflow might have happend
    
    if(last_non_zero < start_i){
        memset(block + start_i, 0, (64-start_i)*sizeof(DCTELEM));
        return last_non_zero;
    }

3548
    lambda= (qscale*qscale*64*105 + 64)>>7; //FIXME finetune
M
Michael Niedermayer 已提交
3549
        
3550
    score_tab[0]= 0;
M
Michael Niedermayer 已提交
3551 3552 3553 3554
    for(i=0; i<=last_non_zero - start_i; i++){
        int level_index, run, j;
        const int dct_coeff= block[ scantable[i + start_i] ];
        const int zero_distoration= dct_coeff*dct_coeff;
3555
        int best_score=256*256*256*120;
M
Michael Niedermayer 已提交
3556

3557
        last_score += zero_distoration;
M
Michael Niedermayer 已提交
3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570
        for(level_index=0; level_index < coeff_count[i]; level_index++){
            int distoration;
            int level= coeff[level_index][i];
            int unquant_coeff;
            
            assert(level);

            if(s->out_format == FMT_H263){
                if(level>0){
                    unquant_coeff= level*qmul + qadd;
                }else{
                    unquant_coeff= level*qmul - qadd;
                }
3571
            }else{ //MPEG1
3572
                j= s->dsp.idct_permutation[ scantable[i + start_i] ]; //FIXME optimize
3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591
                if(s->mb_intra){
                    if (level < 0) {
                        unquant_coeff = (int)((-level) * qscale * s->intra_matrix[j]) >> 3;
                        unquant_coeff = -((unquant_coeff - 1) | 1);
                    } else {
                        unquant_coeff = (int)(  level  * qscale * s->intra_matrix[j]) >> 3;
                        unquant_coeff =   (unquant_coeff - 1) | 1;
                    }
                }else{
                    if (level < 0) {
                        unquant_coeff = ((((-level) << 1) + 1) * qscale * ((int) s->inter_matrix[j])) >> 4;
                        unquant_coeff = -((unquant_coeff - 1) | 1);
                    } else {
                        unquant_coeff = (((  level  << 1) + 1) * qscale * ((int) s->inter_matrix[j])) >> 4;
                        unquant_coeff =   (unquant_coeff - 1) | 1;
                    }
                }
                unquant_coeff<<= 3;
            }
3592

M
Michael Niedermayer 已提交
3593 3594 3595
            distoration= (unquant_coeff - dct_coeff) * (unquant_coeff - dct_coeff);
            level+=64;
            if((level&(~127)) == 0){
3596
                for(run=0; run<=i - left_limit; run++){
3597
                    int score= distoration + length[UNI_AC_ENC_INDEX(run, level)]*lambda;
M
Michael Niedermayer 已提交
3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608
                    score += score_tab[i-run];
                    
                    if(score < best_score){
                        best_score= 
                        score_tab[i+1]= score;
                        run_tab[i+1]= run;
                        level_tab[i+1]= level-64;
                    }
                }

                if(s->out_format == FMT_H263){
3609
                    for(run=0; run<=i - left_limit; run++){
3610
                        int score= distoration + last_length[UNI_AC_ENC_INDEX(run, level)]*lambda;
M
Michael Niedermayer 已提交
3611
                        score += score_tab[i-run];
3612 3613 3614 3615 3616
                        if(score < last_score){
                            last_score= score;
                            last_run= run;
                            last_level= level-64;
                            last_i= i+1;
M
Michael Niedermayer 已提交
3617 3618 3619 3620 3621
                        }
                    }
                }
            }else{
                distoration += esc_length*lambda;
3622
                for(run=0; run<=i - left_limit; run++){
M
Michael Niedermayer 已提交
3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633
                    int score= distoration + score_tab[i-run];
                    
                    if(score < best_score){
                        best_score= 
                        score_tab[i+1]= score;
                        run_tab[i+1]= run;
                        level_tab[i+1]= level-64;
                    }
                }

                if(s->out_format == FMT_H263){
3634
                    for(run=0; run<=i - left_limit; run++){
M
Michael Niedermayer 已提交
3635
                        int score= distoration + score_tab[i-run];
3636 3637 3638 3639 3640
                        if(score < last_score){
                            last_score= score;
                            last_run= run;
                            last_level= level-64;
                            last_i= i+1;
M
Michael Niedermayer 已提交
3641 3642 3643 3644 3645 3646
                        }
                    }
                }
            }
        }

3647
        for(j=left_limit; j<=i; j++){
M
Michael Niedermayer 已提交
3648 3649
            score_tab[j] += zero_distoration;
        }
3650 3651 3652 3653 3654 3655
        score_limit+= zero_distoration;
        if(score_tab[i+1] < score_limit)
            score_limit= score_tab[i+1];
        
        //Note: there is a vlc code in mpeg4 which is 1 bit shorter then another one with a shorter run and the same level
        while(score_tab[ left_limit ] > score_limit + lambda) left_limit++;
M
Michael Niedermayer 已提交
3656
    }
3657 3658 3659

        //FIXME add some cbp penalty

M
Michael Niedermayer 已提交
3660
    if(s->out_format != FMT_H263){
3661
        last_score= 256*256*256*120;
3662
        for(i= left_limit; i<=last_non_zero - start_i + 1; i++){
3663
            int score= score_tab[i];
3664 3665
            if(i) score += lambda*2; //FIXME exacter?

3666 3667 3668 3669 3670 3671
            if(score < last_score){
                last_score= score;
                last_i= i;
                last_level= level_tab[i];
                last_run= run_tab[i];
            }
M
Michael Niedermayer 已提交
3672 3673 3674
        }
    }
    
3675
    last_non_zero= last_i - 1 + start_i;
M
Michael Niedermayer 已提交
3676 3677 3678 3679 3680
    memset(block + start_i, 0, (64-start_i)*sizeof(DCTELEM));
    
    if(last_non_zero < start_i)
        return last_non_zero;
    
3681 3682
    i= last_i;
    assert(last_level);
M
Michael Niedermayer 已提交
3683
//FIXME use permutated scantable
3684
    block[ s->dsp.idct_permutation[ scantable[last_non_zero] ] ]= last_level;
3685
    i -= last_run + 1;
M
Michael Niedermayer 已提交
3686 3687
    
    for(;i>0 ; i -= run_tab[i] + 1){
3688
        const int j= s->dsp.idct_permutation[ scantable[i - 1 + start_i] ];
M
Michael Niedermayer 已提交
3689 3690 3691 3692 3693 3694 3695 3696
    
        block[j]= level_tab[i];
        assert(block[j]);
    }

    return last_non_zero;
}

M
Michael Niedermayer 已提交
3697
static int dct_quantize_c(MpegEncContext *s, 
F
Fabrice Bellard 已提交
3698
                        DCTELEM *block, int n,
3699
                        int qscale, int *overflow)
F
Fabrice Bellard 已提交
3700 3701 3702
{
    int i, j, level, last_non_zero, q;
    const int *qmat;
3703
    const uint8_t *scantable= s->intra_scantable.scantable;
3704 3705 3706
    int bias;
    int max=0;
    unsigned int threshold1, threshold2;
3707

3708
    s->dsp.fdct (block);
F
Fabrice Bellard 已提交
3709 3710

    if (s->mb_intra) {
3711 3712 3713 3714 3715 3716 3717 3718 3719 3720
        if (!s->h263_aic) {
            if (n < 4)
                q = s->y_dc_scale;
            else
                q = s->c_dc_scale;
            q = q << 3;
        } else
            /* For AIC we skip quant/dequant of INTRADC */
            q = 1 << 3;
            
F
Fabrice Bellard 已提交
3721 3722 3723 3724
        /* note: block[0] is assumed to be positive */
        block[0] = (block[0] + (q >> 1)) / q;
        i = 1;
        last_non_zero = 0;
3725
        qmat = s->q_intra_matrix[qscale];
M
Michael Niedermayer 已提交
3726
        bias= s->intra_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
F
Fabrice Bellard 已提交
3727 3728 3729
    } else {
        i = 0;
        last_non_zero = -1;
3730
        qmat = s->q_inter_matrix[qscale];
M
Michael Niedermayer 已提交
3731
        bias= s->inter_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
F
Fabrice Bellard 已提交
3732
    }
M
Michael Niedermayer 已提交
3733 3734
    threshold1= (1<<QMAT_SHIFT) - bias - 1;
    threshold2= (threshold1<<1);
F
Fabrice Bellard 已提交
3735 3736

    for(;i<64;i++) {
M
Michael Niedermayer 已提交
3737
        j = scantable[i];
F
Fabrice Bellard 已提交
3738 3739 3740
        level = block[j];
        level = level * qmat[j];

3741 3742 3743 3744
//        if(   bias+level >= (1<<(QMAT_SHIFT - 3))
//           || bias-level >= (1<<(QMAT_SHIFT - 3))){
        if(((unsigned)(level+threshold1))>threshold2){
            if(level>0){
M
Michael Niedermayer 已提交
3745
                level= (bias + level)>>QMAT_SHIFT;
3746 3747
                block[j]= level;
            }else{
M
Michael Niedermayer 已提交
3748
                level= (bias - level)>>QMAT_SHIFT;
3749 3750 3751
                block[j]= -level;
            }
            max |=level;
F
Fabrice Bellard 已提交
3752
            last_non_zero = i;
3753 3754
        }else{
            block[j]=0;
F
Fabrice Bellard 已提交
3755 3756
        }
    }
3757 3758
    *overflow= s->max_qcoeff < max; //overflow might have happend
    
M
Michael Niedermayer 已提交
3759
    /* we need this permutation so that we correct the IDCT, we only permute the !=0 elements */
3760 3761
    if (s->dsp.idct_permutation_type != FF_NO_IDCT_PERM)
	ff_block_permute(block, s->dsp.idct_permutation, scantable, last_non_zero);
M
Michael Niedermayer 已提交
3762

F
Fabrice Bellard 已提交
3763 3764 3765
    return last_non_zero;
}

3766 3767
#endif //CONFIG_ENCODERS

3768 3769
static void dct_unquantize_mpeg1_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale)
F
Fabrice Bellard 已提交
3770
{
M
Michael Niedermayer 已提交
3771
    int i, level, nCoeffs;
3772
    const uint16_t *quant_matrix;
F
Fabrice Bellard 已提交
3773

3774
    nCoeffs= s->block_last_index[n];
M
Michael Niedermayer 已提交
3775
    
F
Fabrice Bellard 已提交
3776 3777 3778 3779 3780 3781 3782
    if (s->mb_intra) {
        if (n < 4) 
            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;
3783 3784
        for(i=1;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
M
Michael Niedermayer 已提交
3785
            level = block[j];
F
Fabrice Bellard 已提交
3786 3787 3788
            if (level) {
                if (level < 0) {
                    level = -level;
M
Michael Niedermayer 已提交
3789
                    level = (int)(level * qscale * quant_matrix[j]) >> 3;
F
Fabrice Bellard 已提交
3790 3791 3792
                    level = (level - 1) | 1;
                    level = -level;
                } else {
M
Michael Niedermayer 已提交
3793
                    level = (int)(level * qscale * quant_matrix[j]) >> 3;
F
Fabrice Bellard 已提交
3794 3795 3796 3797 3798 3799
                    level = (level - 1) | 1;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
M
Michael Niedermayer 已提交
3800
                block[j] = level;
F
Fabrice Bellard 已提交
3801 3802 3803 3804
            }
        }
    } else {
        i = 0;
3805
        quant_matrix = s->inter_matrix;
3806 3807
        for(;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
M
Michael Niedermayer 已提交
3808
            level = block[j];
F
Fabrice Bellard 已提交
3809 3810 3811 3812
            if (level) {
                if (level < 0) {
                    level = -level;
                    level = (((level << 1) + 1) * qscale *
M
Michael Niedermayer 已提交
3813
                             ((int) (quant_matrix[j]))) >> 4;
F
Fabrice Bellard 已提交
3814 3815 3816 3817
                    level = (level - 1) | 1;
                    level = -level;
                } else {
                    level = (((level << 1) + 1) * qscale *
M
Michael Niedermayer 已提交
3818
                             ((int) (quant_matrix[j]))) >> 4;
F
Fabrice Bellard 已提交
3819 3820 3821 3822 3823 3824
                    level = (level - 1) | 1;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
M
Michael Niedermayer 已提交
3825
                block[j] = level;
F
Fabrice Bellard 已提交
3826 3827 3828 3829
            }
        }
    }
}
3830

3831 3832 3833 3834
static void dct_unquantize_mpeg2_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
3835
    const uint16_t *quant_matrix;
3836

3837 3838
    if(s->alternate_scan) nCoeffs= 63;
    else nCoeffs= s->block_last_index[n];
3839 3840 3841 3842 3843 3844 3845
    
    if (s->mb_intra) {
        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;
3846 3847
        for(i=1;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
3848 3849 3850 3851 3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866
            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;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
                block[j] = level;
            }
        }
    } else {
        int sum=-1;
        i = 0;
3867
        quant_matrix = s->inter_matrix;
3868 3869
        for(;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893
            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;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
                block[j] = level;
                sum+=level;
            }
        }
        block[63]^=sum&1;
    }
}


3894 3895 3896 3897
static void dct_unquantize_h263_c(MpegEncContext *s, 
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
3898
    int nCoeffs;
3899
    
3900 3901 3902 3903 3904
    assert(s->block_last_index[n]>=0);
    
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
    
3905
    if (s->mb_intra) {
3906 3907 3908 3909 3910
        if (!s->h263_aic) {
            if (n < 4) 
                block[0] = block[0] * s->y_dc_scale;
            else
                block[0] = block[0] * s->c_dc_scale;
3911 3912
        }else
            qadd = 0;
3913
        i = 1;
3914
        nCoeffs= 63; //does not allways use zigzag table 
3915 3916
    } else {
        i = 0;
3917
        nCoeffs= s->inter_scantable.raster_end[ s->block_last_index[n] ];
3918 3919
    }

3920
    for(;i<=nCoeffs;i++) {
3921 3922 3923 3924 3925 3926 3927 3928 3929 3930 3931 3932 3933 3934 3935
        level = block[i];
        if (level) {
            if (level < 0) {
                level = level * qmul - qadd;
            } else {
                level = level * qmul + qadd;
            }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
            block[i] = level;
        }
    }
}
F
Fabrice Bellard 已提交
3936

3937

M
Michael Niedermayer 已提交
3938 3939 3940 3941 3942 3943
char ff_get_pict_type_char(int pict_type){
    switch(pict_type){
    case I_TYPE: return 'I'; 
    case P_TYPE: return 'P'; 
    case B_TYPE: return 'B'; 
    case S_TYPE: return 'S'; 
M
Michael Niedermayer 已提交
3944
    default:     return '?';
M
Michael Niedermayer 已提交
3945 3946 3947
    }
}

3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962
static const AVOption mpeg4_options[] =
{
    AVOPTION_CODEC_INT("bitrate", "desired video bitrate", bit_rate, 4, 240000000, 800000),
    AVOPTION_CODEC_FLAG("vhq", "very high quality", flags, CODEC_FLAG_HQ, 0),
    AVOPTION_CODEC_INT("ratetol", "number of bits the bitstream is allowed to diverge from the reference"
		       "the reference can be CBR (for CBR pass1) or VBR (for pass2)",
		       bit_rate_tolerance, 4, 240000000, 8000),
    AVOPTION_CODEC_INT("qmin", "minimum quantizer", qmin, 1, 31, 2),
    AVOPTION_CODEC_INT("qmax", "maximum quantizer", qmax, 1, 31, 31),
    AVOPTION_CODEC_STRING("rc_eq", "rate control equation",
			  rc_eq, "tex^qComp,option1,options2", 0),
    AVOPTION_CODEC_INT("rc_minrate", "rate control minimum bitrate",
		       rc_min_rate, 4, 24000000, 0),
    AVOPTION_CODEC_INT("rc_maxrate", "rate control maximum bitrate",
		       rc_max_rate, 4, 24000000, 0),
Z
Zdenek Kabelac 已提交
3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013 4014
    AVOPTION_CODEC_DOUBLE("rc_buf_aggresivity", "rate control buffer aggresivity",
			  rc_buffer_aggressivity, 4, 24000000, 0),
    AVOPTION_CODEC_DOUBLE("rc_initial_cplx", "initial complexity for pass1 ratecontrol",
			  rc_initial_cplx, 0., 9999999., 0),
    AVOPTION_CODEC_DOUBLE("i_quant_factor", "qscale factor between p and i frames",
			  i_quant_factor, 0., 0., 0),
    AVOPTION_CODEC_DOUBLE("i_quant_offset", "qscale offset between p and i frames",
			  i_quant_factor, -999999., 999999., 0),
    AVOPTION_CODEC_INT("dct_algo", "dct alghorithm",
		       dct_algo, 0, 5, 0), // fixme - "Auto,FastInt,Int,MMX,MLib,Altivec"
    AVOPTION_CODEC_DOUBLE("lumi_masking", "luminance masking",
			  lumi_masking, 0., 999999., 0),
    AVOPTION_CODEC_DOUBLE("temporal_cplx_masking", "temporary complexity masking",
			  temporal_cplx_masking, 0., 999999., 0),
    AVOPTION_CODEC_DOUBLE("spatial_cplx_masking", "spatial complexity masking",
			  spatial_cplx_masking, 0., 999999., 0),
    AVOPTION_CODEC_DOUBLE("p_masking", "p block masking",
			  p_masking, 0., 999999., 0),
    AVOPTION_CODEC_DOUBLE("dark_masking", "darkness masking",
			  dark_masking, 0., 999999., 0),
    AVOPTION_CODEC_INT("idct_algo", "idct alghorithm",
		       idct_algo, 0, 8, 0), // fixme - "Auto,Int,Simple,SimpleMMX,LibMPEG2MMX,PS2,MLib,ARM,Altivec"

    AVOPTION_CODEC_INT("mb_qmin", "minimum MB quantizer",
		       mb_qmin, 0, 8, 0),
    AVOPTION_CODEC_INT("mb_qmax", "maximum MB quantizer",
		       mb_qmin, 0, 8, 0),

    AVOPTION_CODEC_INT("me_cmp", "ME compare function",
		       me_cmp, 0, 24000000, 0),
    AVOPTION_CODEC_INT("me_sub_cmp", "subpixel ME compare function",
		       me_sub_cmp, 0, 24000000, 0),


    AVOPTION_CODEC_INT("dia_size", "ME diamond size & shape",
		       dia_size, 0, 24000000, 0),
    AVOPTION_CODEC_INT("last_predictor_count", "amount of previous MV predictors",
		       last_predictor_count, 0, 24000000, 0),

    AVOPTION_CODEC_INT("pre_me", "pre pass for ME",
		       pre_me, 0, 24000000, 0),
    AVOPTION_CODEC_INT("me_pre_cmp", "ME pre pass compare function",
		       me_pre_cmp, 0, 24000000, 0),

    AVOPTION_CODEC_INT("me_range", "maximum ME search range",
		       me_range, 0, 24000000, 0),
    AVOPTION_CODEC_INT("pre_dia_size", "ME pre pass diamod size & shape",
		       pre_dia_size, 0, 24000000, 0),
    AVOPTION_CODEC_INT("me_subpel_quality", "subpel ME quality",
		       me_subpel_quality, 0, 24000000, 0),
    AVOPTION_CODEC_INT("me_range", "maximum ME search range",
		       me_range, 0, 24000000, 0),
4015 4016 4017 4018
    AVOPTION_CODEC_FLAG("psnr", "calculate PSNR of compressed frames",
		        flags, CODEC_FLAG_PSNR, 0),
    AVOPTION_CODEC_RCOVERRIDE("rc_override", "ratecontrol override (=startframe,endframe,qscale,quality_factor)",
			      rc_override),
4019
    AVOPTION_SUB(avoptions_common),
4020 4021 4022
    AVOPTION_END()
};

4023 4024
#ifdef CONFIG_ENCODERS

F
Fabrice Bellard 已提交
4025 4026 4027 4028 4029 4030 4031 4032 4033 4034
AVCodec mpeg1video_encoder = {
    "mpeg1video",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MPEG1VIDEO,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

M
Michael Niedermayer 已提交
4035 4036
#ifdef CONFIG_RISKY

F
Fabrice Bellard 已提交
4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066
AVCodec h263_encoder = {
    "h263",
    CODEC_TYPE_VIDEO,
    CODEC_ID_H263,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

AVCodec h263p_encoder = {
    "h263p",
    CODEC_TYPE_VIDEO,
    CODEC_ID_H263P,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

AVCodec rv10_encoder = {
    "rv10",
    CODEC_TYPE_VIDEO,
    CODEC_ID_RV10,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

F
Fabrice Bellard 已提交
4067 4068
AVCodec mpeg4_encoder = {
    "mpeg4",
F
Fabrice Bellard 已提交
4069
    CODEC_TYPE_VIDEO,
F
Fabrice Bellard 已提交
4070
    CODEC_ID_MPEG4,
F
Fabrice Bellard 已提交
4071 4072 4073 4074
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
4075
    .options = mpeg4_options,
F
Fabrice Bellard 已提交
4076 4077
};

4078 4079 4080 4081 4082 4083 4084 4085
AVCodec msmpeg4v1_encoder = {
    "msmpeg4v1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MSMPEG4V1,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
Z
Zdenek Kabelac 已提交
4086
    .options = mpeg4_options,
4087 4088 4089 4090 4091 4092 4093 4094 4095 4096
};

AVCodec msmpeg4v2_encoder = {
    "msmpeg4v2",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MSMPEG4V2,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
Z
Zdenek Kabelac 已提交
4097
    .options = mpeg4_options,
4098 4099 4100
};

AVCodec msmpeg4v3_encoder = {
F
Fabrice Bellard 已提交
4101 4102
    "msmpeg4",
    CODEC_TYPE_VIDEO,
4103
    CODEC_ID_MSMPEG4V3,
F
Fabrice Bellard 已提交
4104 4105 4106 4107
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
Z
Zdenek Kabelac 已提交
4108
    .options = mpeg4_options,
F
Fabrice Bellard 已提交
4109
};
M
Michael Niedermayer 已提交
4110 4111 4112 4113 4114 4115 4116 4117 4118

AVCodec wmv1_encoder = {
    "wmv1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_WMV1,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
Z
Zdenek Kabelac 已提交
4119
    .options = mpeg4_options,
M
Michael Niedermayer 已提交
4120 4121
};

M
Michael Niedermayer 已提交
4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132
#endif

AVCodec mjpeg_encoder = {
    "mjpeg",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MJPEG,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};
4133 4134 4135

#endif //CONFIG_ENCODERS