mpegvideo.c 113.3 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 22
 
#include <ctype.h>
F
Fabrice Bellard 已提交
23 24 25
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"
26
#include "simple_idct.h"
F
Fabrice Bellard 已提交
27

28 29 30 31
#ifdef USE_FASTMEMCPY
#include "fastmemcpy.h"
#endif

32 33 34
//#undef NDEBUG
//#include <assert.h>

35 36 37
static void encode_picture(MpegEncContext *s, int picture_number);
static void dct_unquantize_mpeg1_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale);
38 39
static void dct_unquantize_mpeg2_c(MpegEncContext *s,
                                   DCTELEM *block, int n, int qscale);
40 41
static void dct_unquantize_h263_c(MpegEncContext *s, 
                                  DCTELEM *block, int n, int qscale);
M
Michael Niedermayer 已提交
42
static void draw_edges_c(UINT8 *buf, int wrap, int width, int height, int w);
43
static int dct_quantize_c(MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
M
Michael Niedermayer 已提交
44 45

void (*draw_edges)(UINT8 *buf, int wrap, int width, int height, int w)= draw_edges_c;
M
Michael Niedermayer 已提交
46 47
static void emulated_edge_mc(MpegEncContext *s, UINT8 *src, int linesize, int block_w, int block_h, 
                                    int src_x, int src_y, int w, int h);
M
Michael Niedermayer 已提交
48

F
Fabrice Bellard 已提交
49 50 51 52 53 54 55
#define EDGE_WIDTH 16

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

//#define DEBUG

56

F
Fabrice Bellard 已提交
57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
/* for jpeg fast DCT */
#define CONST_BITS 14

static const unsigned short aanscales[64] = {
    /* 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,
    8867, 12299, 11585, 10426,  8867,  6967,  4799,  2446,
    4520,  6270,  5906,  5315,  4520,  3552,  2446,  1247
};

static UINT8 h263_chroma_roundtab[16] = {
    0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2,
};

76
static UINT16 default_mv_penalty[MAX_FCODE+1][MAX_MV*2+1];
77 78
static UINT8 default_fcode_tab[MAX_MV*2+1];

79 80 81
/* default motion estimation */
int motion_estimation_method = ME_EPZS;

82
static void convert_matrix(MpegEncContext *s, int (*qmat)[64], uint16_t (*qmat16)[64], uint16_t (*qmat16_bias)[64],
83
                           const UINT16 *quant_matrix, int bias, int qmin, int qmax)
F
Fabrice Bellard 已提交
84
{
85 86
    int qscale;

87
    for(qscale=qmin; qscale<=qmax; qscale++){
88
        int i;
89 90
        if (s->fdct == ff_jpeg_fdct_islow) {
            for(i=0;i<64;i++) {
91
                const int j= s->idct_permutation[i];
92 93 94 95 96
                /* 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 */
                
M
Michael Niedermayer 已提交
97
                qmat[qscale][j] = (int)((UINT64_C(1) << QMAT_SHIFT) / 
98 99 100
                                (qscale * quant_matrix[j]));
            }
        } else if (s->fdct == fdct_ifast) {
101
            for(i=0;i<64;i++) {
102
                const int j= s->idct_permutation[i];
103 104 105 106 107
                /* 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 */
                
M
Michael Niedermayer 已提交
108
                qmat[qscale][j] = (int)((UINT64_C(1) << (QMAT_SHIFT + 14)) / 
109 110 111 112
                                (aanscales[i] * qscale * quant_matrix[j]));
            }
        } else {
            for(i=0;i<64;i++) {
113
                const int j= s->idct_permutation[i];
114 115 116 117 118 119
                /* 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
                */
                qmat  [qscale][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[i]);
120
                qmat16[qscale][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[j]);
121 122 123 124

                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 已提交
125 126 127
        }
    }
}
128 129 130 131 132 133 134 135 136
// move into common.c perhaps 
#define CHECKED_ALLOCZ(p, size)\
{\
    p= av_mallocz(size);\
    if(p==NULL){\
        perror("malloc");\
        goto fail;\
    }\
}
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
/*
static void build_end(void)
{
    int lastIndex;
    int lastIndexAfterPerm=0;
    for(lastIndex=0; lastIndex<64; lastIndex++)
    {
        if(ff_zigzag_direct[lastIndex] > lastIndexAfterPerm) 
            lastIndexAfterPerm= ff_zigzag_direct[lastIndex];
        zigzag_end[lastIndex]= lastIndexAfterPerm + 1;
    }
}
*/
void ff_init_scantable(MpegEncContext *s, ScanTable *st, const UINT8 *src_scantable){
    int i;
    int end;

    for(i=0; i<64; i++){
        int j;
        j = src_scantable[i];
        st->permutated[i] = s->idct_permutation[j];
    }
    
    end=-1;
    for(i=0; i<64; i++){
        int j;
        j = st->permutated[i];
        if(j>end) end=j;
        st->raster_end[i]= end;
    }
}

/* XXX: those functions should be suppressed ASAP when all IDCTs are
   converted */
static void ff_jref_idct_put(UINT8 *dest, int line_size, DCTELEM *block)
{
    j_rev_dct (block);
    put_pixels_clamped(block, dest, line_size);
}
static void ff_jref_idct_add(UINT8 *dest, int line_size, DCTELEM *block)
{
    j_rev_dct (block);
    add_pixels_clamped(block, dest, line_size);
}
F
Fabrice Bellard 已提交
181

182 183
/* init common dct for both encoder and decoder */
int DCT_common_init(MpegEncContext *s)
F
Fabrice Bellard 已提交
184
{
185
    int i;
F
Fabrice Bellard 已提交
186

187
    s->dct_unquantize_h263 = dct_unquantize_h263_c;
188 189
    s->dct_unquantize_mpeg1 = dct_unquantize_mpeg1_c;
    s->dct_unquantize_mpeg2 = dct_unquantize_mpeg2_c;
190 191 192 193 194
    s->dct_quantize= dct_quantize_c;

    if(s->avctx->dct_algo==FF_DCT_FASTINT)
        s->fdct = fdct_ifast;
    else
195 196 197 198 199 200 201 202 203 204 205 206 207
        s->fdct = ff_jpeg_fdct_islow; //slow/accurate/default

    if(s->avctx->idct_algo==FF_IDCT_INT){
        s->idct_put= ff_jref_idct_put;
        s->idct_add= ff_jref_idct_add;
        for(i=0; i<64; i++)
            s->idct_permutation[i]= (i & 0x38) | ((i & 6) >> 1) | ((i & 1) << 2);
    }else{ //accurate/default
        s->idct_put= simple_idct_put;
        s->idct_add= simple_idct_add;
        for(i=0; i<64; i++)
            s->idct_permutation[i]= i;
    }
208 209 210
        
#ifdef HAVE_MMX
    MPV_common_init_mmx(s);
211 212 213
#endif
#ifdef ARCH_ALPHA
    MPV_common_init_axp(s);
N
Nick Kurshev 已提交
214
#endif
215 216 217
#ifdef HAVE_MLIB
    MPV_common_init_mlib(s);
#endif
218 219 220
#ifdef HAVE_MMI
    MPV_common_init_mmi(s);
#endif
221 222 223 224
#ifdef ARCH_ARMV4L
    MPV_common_init_armv4l();
#endif

225 226 227 228 229 230 231 232

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

234 235 236 237 238 239 240 241 242 243 244
    return 0;
}

/* init common structure for both encoder and decoder */
int MPV_common_init(MpegEncContext *s)
{
    int c_size, i;
    UINT8 *pict;

    DCT_common_init(s);

F
Fabrice Bellard 已提交
245 246
    s->mb_width = (s->width + 15) / 16;
    s->mb_height = (s->height + 15) / 16;
247 248 249 250
    
    /* 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;
251 252 253 254 255 256
    
    /* convert fourcc to upper case */
    s->avctx->fourcc=   toupper( s->avctx->fourcc     &0xFF)          
                     + (toupper((s->avctx->fourcc>>8 )&0xFF)<<8 )
                     + (toupper((s->avctx->fourcc>>16)&0xFF)<<16) 
                     + (toupper((s->avctx->fourcc>>24)&0xFF)<<24);
257

258
    s->mb_num = s->mb_width * s->mb_height;
259
    if(!(s->flags&CODEC_FLAG_DR1)){
M
Michael Niedermayer 已提交
260 261
      s->linesize   = s->mb_width * 16 + 2 * EDGE_WIDTH;
      s->uvlinesize = s->mb_width * 8  +     EDGE_WIDTH;
F
Fabrice Bellard 已提交
262

263
      for(i=0;i<3;i++) {
F
Fabrice Bellard 已提交
264 265 266 267 268
        int w, h, shift, pict_start;

        w = s->linesize;
        h = s->mb_height * 16 + 2 * EDGE_WIDTH;
        shift = (i == 0) ? 0 : 1;
M
Michael Niedermayer 已提交
269 270
        c_size = (s->linesize>>shift) * (h >> shift);
        pict_start = (s->linesize>>shift) * (EDGE_WIDTH >> shift) + (EDGE_WIDTH >> shift);
F
Fabrice Bellard 已提交
271

272
        CHECKED_ALLOCZ(pict, c_size)
F
Fabrice Bellard 已提交
273 274
        s->last_picture_base[i] = pict;
        s->last_picture[i] = pict + pict_start;
M
Michael Niedermayer 已提交
275
        if(i>0) memset(s->last_picture_base[i], 128, c_size);
F
Fabrice Bellard 已提交
276
    
277
        CHECKED_ALLOCZ(pict, c_size)
F
Fabrice Bellard 已提交
278 279
        s->next_picture_base[i] = pict;
        s->next_picture[i] = pict + pict_start;
M
Michael Niedermayer 已提交
280
        if(i>0) memset(s->next_picture_base[i], 128, c_size);
281
        
282 283 284
        if (s->has_b_frames || s->codec_id==CODEC_ID_MPEG4) {
        /* Note the MPEG4 stuff is here cuz of buggy encoders which dont set the low_delay flag but 
           do low-delay encoding, so we cant allways distinguish b-frame containing streams from low_delay streams */
285
            CHECKED_ALLOCZ(pict, c_size)
F
Fabrice Bellard 已提交
286 287
            s->aux_picture_base[i] = pict;
            s->aux_picture[i] = pict + pict_start;
M
Michael Niedermayer 已提交
288
            if(i>0) memset(s->aux_picture_base[i], 128, c_size);
F
Fabrice Bellard 已提交
289
        }
290
      }
M
Michael Niedermayer 已提交
291
      s->ip_buffer_count= 2;
F
Fabrice Bellard 已提交
292
    }
293
    
294
    CHECKED_ALLOCZ(s->edge_emu_buffer, (s->width+64)*2*17*2); //(width + edge + align)*interlaced*MBsize*tolerance
295
    
296
    if (s->encoding) {
297 298
        int j;
        int mv_table_size= (s->mb_width+2)*(s->mb_height+2);
J
Juanjo 已提交
299
        
300 301
        CHECKED_ALLOCZ(s->mb_var   , s->mb_num * sizeof(INT16))
        CHECKED_ALLOCZ(s->mc_mb_var, s->mb_num * sizeof(INT16))
302
        CHECKED_ALLOCZ(s->mb_mean  , s->mb_num * sizeof(INT8))
303 304

        /* Allocate MV tables */
305 306 307 308 309 310 311 312 313 314 315 316 317
        CHECKED_ALLOCZ(s->p_mv_table            , mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_forw_mv_table       , mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_back_mv_table       , mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_bidir_forw_mv_table , mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_bidir_back_mv_table , mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_direct_forw_mv_table, mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_direct_back_mv_table, mv_table_size * 2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->b_direct_mv_table     , mv_table_size * 2 * sizeof(INT16))

        CHECKED_ALLOCZ(s->me_scratchpad,  s->linesize*16*3*sizeof(uint8_t))
        
        CHECKED_ALLOCZ(s->me_map      , ME_MAP_SIZE*sizeof(uint32_t))
        CHECKED_ALLOCZ(s->me_score_map, ME_MAP_SIZE*sizeof(uint16_t))
318

319 320 321 322 323 324 325 326 327 328 329
        if(s->max_b_frames){
            for(j=0; j<REORDER_BUFFER_SIZE; j++){
                int i;
                for(i=0;i<3;i++) {
                    int w, h, shift;

                    w = s->linesize;
                    h = s->mb_height * 16;
                    shift = (i == 0) ? 0 : 1;
                    c_size = (w >> shift) * (h >> shift);

330
                    CHECKED_ALLOCZ(pict, c_size);
331 332 333 334
                    s->picture_buffer[j][i] = pict;
                }
            }
        }
335 336 337 338 339

        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 已提交
340
        
341 342 343
        if(s->msmpeg4_version){
            CHECKED_ALLOCZ(s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int));
        }
M
Michael Niedermayer 已提交
344
        CHECKED_ALLOCZ(s->avctx->stats_out, 256);
345 346
    }
    
347
    if (s->out_format == FMT_H263 || s->encoding) {
F
Fabrice Bellard 已提交
348
        int size;
349 350 351
        /* Allocate MB type table */
        CHECKED_ALLOCZ(s->mb_type  , s->mb_num * sizeof(UINT8))

F
Fabrice Bellard 已提交
352 353
        /* MV prediction */
        size = (2 * s->mb_width + 2) * (2 * s->mb_height + 2);
354
        CHECKED_ALLOCZ(s->motion_val, size * 2 * sizeof(INT16));
355 356 357 358 359 360 361
    }

    if(s->codec_id==CODEC_ID_MPEG4){
        /* 4mv and interlaced direct mode decoding tables */
        CHECKED_ALLOCZ(s->co_located_type_table, s->mb_num * sizeof(UINT8))
        CHECKED_ALLOCZ(s->field_mv_table, s->mb_num*2*2 * sizeof(INT16))
        CHECKED_ALLOCZ(s->field_select_table, s->mb_num*2* sizeof(INT8))
F
Fabrice Bellard 已提交
362 363
    }

364
    if (s->h263_pred || s->h263_plus) {
F
Fabrice Bellard 已提交
365 366 367 368 369 370 371
        int y_size, c_size, i, size;
        
        /* dc values */

        y_size = (2 * s->mb_width + 2) * (2 * s->mb_height + 2);
        c_size = (s->mb_width + 2) * (s->mb_height + 2);
        size = y_size + 2 * c_size;
372
        CHECKED_ALLOCZ(s->dc_val[0], size * sizeof(INT16));
F
Fabrice Bellard 已提交
373 374 375 376 377 378
        s->dc_val[1] = s->dc_val[0] + y_size;
        s->dc_val[2] = s->dc_val[1] + c_size;
        for(i=0;i<size;i++)
            s->dc_val[0][i] = 1024;

        /* ac values */
379
        CHECKED_ALLOCZ(s->ac_val[0], size * sizeof(INT16) * 16);
F
Fabrice Bellard 已提交
380 381 382 383
        s->ac_val[1] = s->ac_val[0] + y_size;
        s->ac_val[2] = s->ac_val[1] + c_size;
        
        /* cbp values */
384
        CHECKED_ALLOCZ(s->coded_block, y_size);
M
Michael Niedermayer 已提交
385 386
        
        /* divx501 bitstream reorder buffer */
387 388 389 390 391
        CHECKED_ALLOCZ(s->bitstream_buffer, BITSTREAM_BUFFER_SIZE);
        
        /* cbp, ac_pred, pred_dir */
        CHECKED_ALLOCZ(s->cbp_table  , s->mb_num * sizeof(UINT8))
        CHECKED_ALLOCZ(s->pred_dir_table, s->mb_num * sizeof(UINT8))
392
    }
M
Michael Niedermayer 已提交
393 394
    CHECKED_ALLOCZ(s->qscale_table  , s->mb_num * sizeof(UINT8))
    
395 396 397 398
    /* 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 已提交
399 400
    /* default structure is frame */
    s->picture_structure = PICT_FRAME;
401
    
402
    /* init macroblock skip table */
403
    CHECKED_ALLOCZ(s->mbskip_table, s->mb_num);
404
    
405
    s->block= s->blocks[0];
406

F
Fabrice Bellard 已提交
407 408 409
    s->context_initialized = 1;
    return 0;
 fail:
410
    MPV_common_end(s);
F
Fabrice Bellard 已提交
411 412 413
    return -1;
}

414 415 416

//extern int sads;

F
Fabrice Bellard 已提交
417 418 419 420 421
/* init common structure for both encoder and decoder */
void MPV_common_end(MpegEncContext *s)
{
    int i;

422 423
    av_freep(&s->mb_type);
    av_freep(&s->mb_var);
424
    av_freep(&s->mc_mb_var);
425
    av_freep(&s->mb_mean);
426 427 428 429 430 431 432 433 434 435 436 437 438
    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_forw_mv_table);
    av_freep(&s->b_direct_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);
439 440 441
    av_freep(&s->cbp_table);
    av_freep(&s->pred_dir_table);
    av_freep(&s->qscale_table);
442
    av_freep(&s->me_scratchpad);
443 444 445
    av_freep(&s->me_map);
    av_freep(&s->me_score_map);
    
446 447
    av_freep(&s->mbskip_table);
    av_freep(&s->bitstream_buffer);
448 449
    av_freep(&s->tex_pb_buffer);
    av_freep(&s->pb2_buffer);
450
    av_freep(&s->edge_emu_buffer);
451 452 453
    av_freep(&s->co_located_type_table);
    av_freep(&s->field_mv_table);
    av_freep(&s->field_select_table);
M
Michael Niedermayer 已提交
454
    av_freep(&s->avctx->stats_out);
455
    av_freep(&s->ac_stats);
M
Michael Niedermayer 已提交
456
    
F
Fabrice Bellard 已提交
457
    for(i=0;i<3;i++) {
458
        int j;
459 460 461 462 463 464 465 466 467 468 469 470
        if(!(s->flags&CODEC_FLAG_DR1)){
            av_freep(&s->last_picture_base[i]);
            av_freep(&s->next_picture_base[i]);
            av_freep(&s->aux_picture_base[i]);
        }
        s->last_picture_base[i]=
        s->next_picture_base[i]=
        s->aux_picture_base [i] = NULL;
        s->last_picture[i]=
        s->next_picture[i]=
        s->aux_picture [i] = NULL;

471
        for(j=0; j<REORDER_BUFFER_SIZE; j++){
472
            av_freep(&s->picture_buffer[j][i]);
473
        }
F
Fabrice Bellard 已提交
474 475 476 477 478 479 480 481
    }
    s->context_initialized = 0;
}

/* init video encoder */
int MPV_encode_init(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;
482
    int i;
F
Fabrice Bellard 已提交
483

484 485
    avctx->pix_fmt = PIX_FMT_YUV420P;

F
Fabrice Bellard 已提交
486
    s->bit_rate = avctx->bit_rate;
487
    s->bit_rate_tolerance = avctx->bit_rate_tolerance;
F
Fabrice Bellard 已提交
488 489 490
    s->frame_rate = avctx->frame_rate;
    s->width = avctx->width;
    s->height = avctx->height;
491
    if(avctx->gop_size > 600){
492
        fprintf(stderr, "Warning keyframe interval too large! reducing it ...\n");
493 494
        avctx->gop_size=600;
    }
F
Fabrice Bellard 已提交
495
    s->gop_size = avctx->gop_size;
496 497
    s->rtp_mode = avctx->rtp_mode;
    s->rtp_payload_size = avctx->rtp_payload_size;
498 499
    if (avctx->rtp_callback)
        s->rtp_callback = avctx->rtp_callback;
500 501 502 503 504
    s->qmin= avctx->qmin;
    s->qmax= avctx->qmax;
    s->max_qdiff= avctx->max_qdiff;
    s->qcompress= avctx->qcompress;
    s->qblur= avctx->qblur;
505
    s->avctx = avctx;
506
    s->aspect_ratio_info= avctx->aspect_ratio_info;
507 508 509 510 511
    if (avctx->aspect_ratio_info == FF_ASPECT_EXTENDED)
    {
	s->aspected_width = avctx->aspected_width;
	s->aspected_height = avctx->aspected_height;
    }
512
    s->flags= avctx->flags;
513
    s->max_b_frames= avctx->max_b_frames;
514
    s->b_frame_strategy= avctx->b_frame_strategy;
515
    s->codec_id= avctx->codec->id;
516 517 518 519
    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;
520
    s->mpeg_quant= avctx->mpeg_quant;
521

F
Fabrice Bellard 已提交
522 523 524 525 526 527
    if (s->gop_size <= 1) {
        s->intra_only = 1;
        s->gop_size = 12;
    } else {
        s->intra_only = 0;
    }
528

529
    /* ME algorithm */
530 531 532 533 534
    if (avctx->me_method == 0)
        /* For compatibility */
        s->me_method = motion_estimation_method;
    else
        s->me_method = avctx->me_method;
535

536
    /* Fixed QSCALE */
F
Fabrice Bellard 已提交
537
    s->fixed_qscale = (avctx->flags & CODEC_FLAG_QSCALE);
538
    
539
    s->adaptive_quant= (   s->avctx->lumi_masking
540
                        || s->avctx->dark_masking
541 542 543 544
                        || s->avctx->temporal_cplx_masking 
                        || s->avctx->spatial_cplx_masking
                        || s->avctx->p_masking)
                       && !s->fixed_qscale;
545 546
    
    s->progressive_sequence= !(avctx->flags & CODEC_FLAG_INTERLACED_DCT);
547

F
Fabrice Bellard 已提交
548 549 550
    switch(avctx->codec->id) {
    case CODEC_ID_MPEG1VIDEO:
        s->out_format = FMT_MPEG1;
M
Michael Niedermayer 已提交
551
        avctx->delay=0; //FIXME not sure, should check the spec
F
Fabrice Bellard 已提交
552 553 554 555
        break;
    case CODEC_ID_MJPEG:
        s->out_format = FMT_MJPEG;
        s->intra_only = 1; /* force intra only for jpeg */
556
        s->mjpeg_write_tables = 1; /* write all tables */
557
	s->mjpeg_data_only_frames = 0; /* write all the needed headers */
558 559 560
        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; 
561
        s->mjpeg_hsample[0] = 2;
562 563
        s->mjpeg_hsample[1] = 1; 
        s->mjpeg_hsample[2] = 1; 
F
Fabrice Bellard 已提交
564 565
        if (mjpeg_init(s) < 0)
            return -1;
M
Michael Niedermayer 已提交
566
        avctx->delay=0;
F
Fabrice Bellard 已提交
567 568
        break;
    case CODEC_ID_H263:
569 570
        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 已提交
571
            return -1;
572
        }
F
Fabrice Bellard 已提交
573
        s->out_format = FMT_H263;
M
Michael Niedermayer 已提交
574
        avctx->delay=0;
F
Fabrice Bellard 已提交
575 576 577
        break;
    case CODEC_ID_H263P:
        s->out_format = FMT_H263;
578 579
        s->rtp_mode = 1;
        s->rtp_payload_size = 1200; 
F
Fabrice Bellard 已提交
580
        s->h263_plus = 1;
581
        s->unrestricted_mv = 1;
582
        s->h263_aic = 1;
583 584 585 586
        
        /* These are just to be sure */
        s->umvplus = 0;
        s->umvplus_dec = 0;
M
Michael Niedermayer 已提交
587
        avctx->delay=0;
F
Fabrice Bellard 已提交
588 589 590 591
        break;
    case CODEC_ID_RV10:
        s->out_format = FMT_H263;
        s->h263_rv10 = 1;
M
Michael Niedermayer 已提交
592
        avctx->delay=0;
F
Fabrice Bellard 已提交
593
        break;
F
Fabrice Bellard 已提交
594
    case CODEC_ID_MPEG4:
F
Fabrice Bellard 已提交
595 596 597
        s->out_format = FMT_H263;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
598
        s->has_b_frames= s->max_b_frames ? 1 : 0;
599
        s->low_delay= !s->has_b_frames;
M
Michael Niedermayer 已提交
600
        avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1); 
F
Fabrice Bellard 已提交
601
        break;
602
    case CODEC_ID_MSMPEG4V1:
F
Fabrice Bellard 已提交
603 604 605 606
        s->out_format = FMT_H263;
        s->h263_msmpeg4 = 1;
        s->h263_pred = 1;
        s->unrestricted_mv = 1;
607
        s->msmpeg4_version= 1;
M
Michael Niedermayer 已提交
608
        avctx->delay=0;
609 610 611 612 613 614 615
        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 已提交
616
        avctx->delay=0;
617 618 619 620 621 622 623
        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 已提交
624
        avctx->delay=0;
F
Fabrice Bellard 已提交
625
        break;
M
Michael Niedermayer 已提交
626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641
    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;
        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;
        break;
F
Fabrice Bellard 已提交
642 643 644
    default:
        return -1;
    }
M
Michael Niedermayer 已提交
645
    
646 647 648 649 650
    { /* set up some save defaults, some codecs might override them later */
        static int done=0;
        if(!done){
            int i;
            done=1;
651
            memset(default_mv_penalty, 0, sizeof(UINT16)*(MAX_FCODE+1)*(2*MAX_MV+1));
652 653 654 655 656 657 658 659 660
            memset(default_fcode_tab , 0, sizeof(UINT8)*(2*MAX_MV+1));

            for(i=-16; i<16; i++){
                default_fcode_tab[i + MAX_MV]= 1;
            }
        }
    }
    s->mv_penalty= default_mv_penalty;
    s->fcode_tab= default_fcode_tab;
661 662 663
    s->y_dc_scale_table=
    s->c_dc_scale_table= ff_mpeg1_dc_scale_table;
 
664
    /* dont use mv_penalty table for crap MV as it would be confused */
665
    if (s->me_method < ME_EPZS) s->mv_penalty = default_mv_penalty;
666

667 668
    s->encoding = 1;

F
Fabrice Bellard 已提交
669 670 671 672
    /* init */
    if (MPV_common_init(s) < 0)
        return -1;
    
673 674 675 676 677 678 679
    if (s->out_format == FMT_H263)
        h263_encode_init(s);
    else if (s->out_format == FMT_MPEG1)
        ff_mpeg1_encode_init(s);
    if(s->msmpeg4_version)
        ff_msmpeg4_encode_init(s);

680 681
    /* init default q matrix */
    for(i=0;i<64;i++) {
682
        int j= s->idct_permutation[i];
683
        if(s->codec_id==CODEC_ID_MPEG4 && s->mpeg_quant){
684 685
            s->intra_matrix[j] = ff_mpeg4_default_intra_matrix[i];
            s->inter_matrix[j] = ff_mpeg4_default_non_intra_matrix[i];
686
        }else if(s->out_format == FMT_H263){
687 688
            s->intra_matrix[j] =
            s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
689
        }else{ /* mpeg1 */
690 691
            s->intra_matrix[j] = ff_mpeg1_default_intra_matrix[i];
            s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
692
        }
693 694 695
    }

    /* precompute matrix */
696
    /* for mjpeg, we do include qscale in the matrix */
697
    if (s->out_format != FMT_MJPEG) {
698
        convert_matrix(s, s->q_intra_matrix, s->q_intra_matrix16, s->q_intra_matrix16_bias, 
699
                       s->intra_matrix, s->intra_quant_bias, 1, 31);
700
        convert_matrix(s, s->q_inter_matrix, s->q_inter_matrix16, s->q_inter_matrix16_bias, 
701
                       s->inter_matrix, s->inter_quant_bias, 1, 31);
702 703
    }

704 705
    if(ff_rate_control_init(s) < 0)
        return -1;
F
Fabrice Bellard 已提交
706 707

    s->picture_number = 0;
708
    s->picture_in_gop_number = 0;
F
Fabrice Bellard 已提交
709 710 711
    s->fake_picture_number = 0;
    /* motion detector init */
    s->f_code = 1;
712
    s->b_code = 1;
F
Fabrice Bellard 已提交
713 714 715 716 717 718 719 720 721 722 723

    return 0;
}

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

#ifdef STATS
    print_stats();
#endif
724 725 726

    ff_rate_control_uninit(s);

F
Fabrice Bellard 已提交
727 728 729
    MPV_common_end(s);
    if (s->out_format == FMT_MJPEG)
        mjpeg_close(s);
730
      
F
Fabrice Bellard 已提交
731 732 733 734
    return 0;
}

/* draw the edges of width 'w' of an image of size width, height */
735
//FIXME check that this is ok for mpeg4 interlaced
M
Michael Niedermayer 已提交
736
static void draw_edges_c(UINT8 *buf, int wrap, int width, int height, int w)
F
Fabrice Bellard 已提交
737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763
{
    UINT8 *ptr, *last_line;
    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 */
    }
}

/* generic function for encode/decode called before a frame is coded/decoded */
764
void MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx)
F
Fabrice Bellard 已提交
765 766 767 768
{
    int i;
    UINT8 *tmp;

769
    s->mb_skiped = 0;
770
    s->decoding_error=0;
771
    avctx->mbskip_table= s->mbskip_table;
772

773 774 775
    if(avctx->flags&CODEC_FLAG_DR1){
        avctx->get_buffer_callback(avctx, s->width, s->height, s->pict_type);

M
Michael Niedermayer 已提交
776 777 778
        s->linesize  = avctx->dr_stride;
        s->uvlinesize= avctx->dr_uvstride;
        s->ip_buffer_count= avctx->dr_ip_buffer_count;
779
    }
780
    avctx->dr_ip_buffer_count= s->ip_buffer_count;
781
    
F
Fabrice Bellard 已提交
782 783
    if (s->pict_type == B_TYPE) {
        for(i=0;i<3;i++) {
784 785
            if(avctx->flags&CODEC_FLAG_DR1)
                s->aux_picture[i]= avctx->dr_buffer[i];
786 787 788 789 790 791
            
            //FIXME the following should never be needed, the decoder should drop b frames if no reference is available
            if(s->next_picture[i]==NULL)
                s->next_picture[i]= s->aux_picture[i];
            if(s->last_picture[i]==NULL)
                s->last_picture[i]= s->next_picture[i];
792

F
Fabrice Bellard 已提交
793 794 795 796 797
            s->current_picture[i] = s->aux_picture[i];
        }
    } else {
        for(i=0;i<3;i++) {
            /* swap next and last */
798 799 800 801 802
            if(avctx->flags&CODEC_FLAG_DR1)
                tmp= avctx->dr_buffer[i];
            else
                tmp = s->last_picture[i];

F
Fabrice Bellard 已提交
803 804 805
            s->last_picture[i] = s->next_picture[i];
            s->next_picture[i] = tmp;
            s->current_picture[i] = tmp;
806

807 808 809
            if(s->last_picture[i]==NULL)
                s->last_picture[i]= s->next_picture[i];

810 811 812
            s->last_dr_opaque= s->next_dr_opaque;
            s->next_dr_opaque= avctx->dr_opaque_frame;

M
Michael Niedermayer 已提交
813
            if(s->has_b_frames && s->last_dr_opaque && s->codec_id!=CODEC_ID_SVQ1)
814 815 816
                avctx->dr_opaque_frame= s->last_dr_opaque;
            else
                avctx->dr_opaque_frame= s->next_dr_opaque;
F
Fabrice Bellard 已提交
817 818
        }
    }
819 820 821 822 823 824 825 826 827
    /* 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;
F
Fabrice Bellard 已提交
828
}
829

F
Fabrice Bellard 已提交
830 831 832
/* generic function for encode/decode called after a frame has been coded/decoded */
void MPV_frame_end(MpegEncContext *s)
{
833 834
    s->avctx->key_frame   = (s->pict_type == I_TYPE);
    s->avctx->pict_type   = s->pict_type;
835

F
Fabrice Bellard 已提交
836
    /* draw edge for correct motion prediction if outside */
837
    if (s->pict_type != B_TYPE && !s->intra_only && !(s->flags&CODEC_FLAG_EMU_EDGE)) {
838 839 840
        draw_edges(s->current_picture[0], s->linesize  , s->h_edge_pos   , s->v_edge_pos   , EDGE_WIDTH  );
        draw_edges(s->current_picture[1], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
        draw_edges(s->current_picture[2], s->uvlinesize, s->h_edge_pos>>1, s->v_edge_pos>>1, EDGE_WIDTH/2);
F
Fabrice Bellard 已提交
841
    }
M
Michael Niedermayer 已提交
842
    emms_c();
843
    
M
Michael Niedermayer 已提交
844
    s->last_pict_type    = s->pict_type;
845 846
    if(s->pict_type!=B_TYPE){
        s->last_non_b_pict_type= s->pict_type;
M
Michael Niedermayer 已提交
847 848
        s->num_available_buffers++;
        if(s->num_available_buffers>2) s->num_available_buffers= 2;
849
    }
F
Fabrice Bellard 已提交
850 851
}

852 853 854 855 856 857 858 859 860 861 862 863 864
/* reorder input for encoding */
void reorder_input(MpegEncContext *s, AVPicture *pict)
{
    int i, j, index;
            
    if(s->max_b_frames > FF_MAX_B_FRAMES) s->max_b_frames= FF_MAX_B_FRAMES;

//        delay= s->max_b_frames+1; (or 0 if no b frames cuz decoder diff)

    for(j=0; j<REORDER_BUFFER_SIZE-1; j++){
        s->coded_order[j]= s->coded_order[j+1];
    }
    s->coded_order[j].picture[0]= s->coded_order[j].picture[1]= s->coded_order[j].picture[2]= NULL; //catch uninitalized buffers
M
Michael Niedermayer 已提交
865
    s->coded_order[j].pict_type=0;
866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882

    switch(s->input_pict_type){
    default: 
    case I_TYPE:
    case S_TYPE:
    case P_TYPE:
        index= s->max_b_frames - s->b_frames_since_non_b;
        s->b_frames_since_non_b=0;
        break;            
    case B_TYPE:
        index= s->max_b_frames + 1;
        s->b_frames_since_non_b++;
        break;          
    }
//printf("index:%d type:%d strides: %d %d\n", index, s->input_pict_type, pict->linesize[0], s->linesize);
    if(   (index==0 || (s->flags&CODEC_FLAG_INPUT_PRESERVED))
       && pict->linesize[0] == s->linesize
M
Michael Niedermayer 已提交
883 884
       && pict->linesize[1] == s->uvlinesize
       && pict->linesize[2] == s->uvlinesize){
885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916
//printf("ptr\n");
        for(i=0; i<3; i++){
            s->coded_order[index].picture[i]= pict->data[i];
        }
    }else{
//printf("copy\n");
        for(i=0; i<3; i++){
            uint8_t *src = pict->data[i];
            uint8_t *dest;
            int src_wrap = pict->linesize[i];
            int dest_wrap = s->linesize;
            int w = s->width;
            int h = s->height;

            if(index==0) dest= s->last_picture[i]+16; //is current_picture indeed but the switch hapens after reordering
            else         dest= s->picture_buffer[s->picture_buffer_index][i];

            if (i >= 1) {
                dest_wrap >>= 1;
                w >>= 1;
                h >>= 1;
            }

            s->coded_order[index].picture[i]= dest;
            for(j=0;j<h;j++) {
                memcpy(dest, src, w);
                dest += dest_wrap;
                src += src_wrap;
            }
        }
        if(index!=0){
            s->picture_buffer_index++;
M
Michael Niedermayer 已提交
917
            if(s->picture_buffer_index >= REORDER_BUFFER_SIZE) s->picture_buffer_index=0;
918 919 920 921 922 923 924 925 926 927 928 929 930
        }
    }
    s->coded_order[index].pict_type = s->input_pict_type;
    s->coded_order[index].qscale    = s->input_qscale;
    s->coded_order[index].force_type= s->force_input_type;
    s->coded_order[index].picture_in_gop_number= s->input_picture_in_gop_number;
    s->coded_order[index].picture_number= s->input_picture_number;

    for(i=0; i<3; i++){
        s->new_picture[i]= s->coded_order[0].picture[i];
    }
}

F
Fabrice Bellard 已提交
931 932 933 934 935 936
int MPV_encode_picture(AVCodecContext *avctx,
                       unsigned char *buf, int buf_size, void *data)
{
    MpegEncContext *s = avctx->priv_data;
    AVPicture *pict = data;

937
    s->input_qscale = avctx->quality;
F
Fabrice Bellard 已提交
938 939 940

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

941 942 943 944 945 946 947 948 949 950 951 952 953
    if(avctx->flags&CODEC_FLAG_TYPE){
        s->input_pict_type=
        s->force_input_type= avctx->key_frame ? I_TYPE : P_TYPE;
    }else if(s->flags&CODEC_FLAG_PASS2){
        s->input_pict_type=
        s->force_input_type= s->rc_context.entry[s->input_picture_number].new_pict_type;
    }else{
        s->force_input_type=0;
        if (!s->intra_only) {
            /* first picture of GOP is intra */
            if (s->input_picture_in_gop_number % s->gop_size==0){
                s->input_pict_type = I_TYPE;
            }else if(s->max_b_frames==0){
954
                s->input_pict_type = P_TYPE;
955 956 957 958 959 960 961 962
            }else{
                if(s->b_frames_since_non_b < s->max_b_frames) //FIXME more IQ
                    s->input_pict_type = B_TYPE;
                else
                    s->input_pict_type = P_TYPE;
            }
        } else {
            s->input_pict_type = I_TYPE;
963
        }
F
Fabrice Bellard 已提交
964 965
    }

966 967 968
    if(s->input_pict_type==I_TYPE)
        s->input_picture_in_gop_number=0;
    
969 970 971 972 973 974 975 976 977 978 979 980
    reorder_input(s, pict);
    
    /* output? */
    if(s->coded_order[0].picture[0]){

        s->pict_type= s->coded_order[0].pict_type;
        if (s->fixed_qscale) /* the ratecontrol needs the last qscale so we dont touch it for CBR */
            s->qscale= s->coded_order[0].qscale;
        s->force_type= s->coded_order[0].force_type;
        s->picture_in_gop_number= s->coded_order[0].picture_in_gop_number;
        s->picture_number= s->coded_order[0].picture_number;

981
        MPV_frame_start(s, avctx);
982 983

        encode_picture(s, s->picture_number);
984
        
J
Juanjo 已提交
985
        avctx->real_pict_num  = s->picture_number;
986 987 988 989 990 991
        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;
992
        avctx->p_count     = s->mb_num - s->i_count - s->skip_count; //FIXME f/b_count in avctx
993 994 995 996 997 998 999
        avctx->skip_count  = s->skip_count;

        MPV_frame_end(s);

        if (s->out_format == FMT_MJPEG)
            mjpeg_picture_trailer(s);

1000 1001
        if(!s->fixed_qscale)
            avctx->quality = s->qscale;
1002 1003 1004
        
        if(s->flags&CODEC_FLAG_PASS1)
            ff_write_pass1_stats(s);
1005
    
F
Fabrice Bellard 已提交
1006 1007
    }

1008 1009
    s->input_picture_number++;
    s->input_picture_in_gop_number++;
F
Fabrice Bellard 已提交
1010 1011

    flush_put_bits(&s->pb);
1012
    s->frame_bits  = (pbBufPtr(&s->pb) - s->pb.buf) * 8;
M
Michael Niedermayer 已提交
1013
    
1014
    s->total_bits += s->frame_bits;
1015 1016 1017
    avctx->frame_bits  = s->frame_bits;
//printf("fcode: %d, type: %d, head: %d, mv: %d, misc: %d, frame: %d, itex: %d, ptex: %d\n", 
//s->f_code, avctx->key_frame, s->header_bits, s->mv_bits, s->misc_bits, s->frame_bits, s->i_tex_bits, s->p_tex_bits);
M
Michael Niedermayer 已提交
1018 1019 1020 1021 1022 1023 1024 1025 1026 1027
#if 0 //dump some stats to stats.txt for testing/debuging
if(s->max_b_frames==0)
{
    static FILE *f=NULL;
    if(!f) f= fopen("stats.txt", "wb");
    get_psnr(pict->data, s->current_picture,
             pict->linesize, s->linesize, avctx);
    fprintf(f, "%7d, %7d, %2.4f\n", pbBufPtr(&s->pb) - s->pb.buf, s->qscale, avctx->psnr_y);
}
#endif
1028

1029 1030 1031 1032 1033
    if (avctx->get_psnr) {
        /* At this point pict->data should have the original frame   */
        /* an s->current_picture should have the coded/decoded frame */
        get_psnr(pict->data, s->current_picture,
                 pict->linesize, s->linesize, avctx);
1034
//        printf("%f\n", avctx->psnr_y);
1035
    }
1036
    return pbBufPtr(&s->pb) - s->pb.buf;
F
Fabrice Bellard 已提交
1037 1038
}

M
Michael Niedermayer 已提交
1039 1040 1041 1042 1043 1044 1045
static inline void gmc1_motion(MpegEncContext *s,
                               UINT8 *dest_y, UINT8 *dest_cb, UINT8 *dest_cr,
                               int dest_offset,
                               UINT8 **ref_picture, int src_offset,
                               int h)
{
    UINT8 *ptr;
M
Michael Niedermayer 已提交
1046
    int offset, src_x, src_y, linesize, uvlinesize;
M
Michael Niedermayer 已提交
1047
    int motion_x, motion_y;
M
Michael Niedermayer 已提交
1048
    int emu=0;
M
Michael Niedermayer 已提交
1049

1050
    if(s->real_sprite_warping_points>1) printf("more than 1 warp point isnt supported\n");
M
Michael Niedermayer 已提交
1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
    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;
    
    linesize = s->linesize;
M
Michael Niedermayer 已提交
1065
    uvlinesize = s->uvlinesize;
M
Michael Niedermayer 已提交
1066 1067 1068
    ptr = ref_picture[0] + (src_y * linesize) + src_x + src_offset;

    dest_y+=dest_offset;
M
Michael Niedermayer 已提交
1069
    if(s->flags&CODEC_FLAG_EMU_EDGE){
1070 1071 1072
        if(src_x<0 || src_y<0 || src_x + (motion_x&15) + 16 > s->h_edge_pos
                              || src_y + (motion_y&15) + h  > s->v_edge_pos){
            emulated_edge_mc(s, ptr, linesize, 17, h+1, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
M
Michael Niedermayer 已提交
1073 1074 1075 1076
            ptr= s->edge_emu_buffer;
            emu=1;
        }
    }
M
Michael Niedermayer 已提交
1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
    gmc1(dest_y  , ptr  , linesize, h, motion_x&15, motion_y&15, s->no_rounding);
    gmc1(dest_y+8, ptr+8, linesize, h, motion_x&15, motion_y&15, s->no_rounding);

    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 已提交
1093
    offset = (src_y * uvlinesize) + src_x + (src_offset>>1);
M
Michael Niedermayer 已提交
1094
    ptr = ref_picture[1] + offset;
M
Michael Niedermayer 已提交
1095
    if(emu){
1096
        emulated_edge_mc(s, ptr, uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
M
Michael Niedermayer 已提交
1097 1098
        ptr= s->edge_emu_buffer;
    }
M
Michael Niedermayer 已提交
1099
    gmc1(dest_cb + (dest_offset>>1), ptr, uvlinesize, h>>1, motion_x&15, motion_y&15, s->no_rounding);
M
Michael Niedermayer 已提交
1100
    
M
Michael Niedermayer 已提交
1101
    ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1102
    if(emu){
1103
        emulated_edge_mc(s, ptr, uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, s->v_edge_pos>>1);
M
Michael Niedermayer 已提交
1104 1105
        ptr= s->edge_emu_buffer;
    }
M
Michael Niedermayer 已提交
1106
    gmc1(dest_cr + (dest_offset>>1), ptr, uvlinesize, h>>1, motion_x&15, motion_y&15, s->no_rounding);
M
Michael Niedermayer 已提交
1107 1108 1109 1110
    
    return;
}

M
Michael Niedermayer 已提交
1111
static void emulated_edge_mc(MpegEncContext *s, UINT8 *src, int linesize, int block_w, int block_h, 
1112 1113 1114
                                    int src_x, int src_y, int w, int h){
    int x, y;
    int start_y, start_x, end_y, end_x;
M
Michael Niedermayer 已提交
1115
    UINT8 *buf= s->edge_emu_buffer;
1116
    
1117 1118 1119
    if(src_y>= h){
        src+= (h-1-src_y)*linesize;
        src_y=h-1;
M
Michael Niedermayer 已提交
1120 1121 1122
    }else if(src_y<=-block_h){
        src+= (1-block_h-src_y)*linesize;
        src_y=1-block_h;
1123 1124 1125 1126
    }
    if(src_x>= w){
        src+= (w-1-src_x);
        src_x=w-1;
M
Michael Niedermayer 已提交
1127 1128 1129
    }else if(src_x<=-block_w){
        src+= (1-block_w-src_x);
        src_x=1-block_w;
1130 1131 1132 1133 1134 1135
    }

    start_y= MAX(0, -src_y);
    start_x= MAX(0, -src_x);
    end_y= MIN(block_h, h-src_y);
    end_x= MIN(block_w, w-src_x);
M
Michael Niedermayer 已提交
1136

1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171
    // 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 已提交
1172 1173 1174 1175 1176
/* apply one mpeg motion vector to the three components */
static inline void mpeg_motion(MpegEncContext *s,
                               UINT8 *dest_y, UINT8 *dest_cb, UINT8 *dest_cr,
                               int dest_offset,
                               UINT8 **ref_picture, int src_offset,
M
Michael Niedermayer 已提交
1177
                               int field_based, op_pixels_func (*pix_op)[4],
F
Fabrice Bellard 已提交
1178 1179 1180
                               int motion_x, int motion_y, int h)
{
    UINT8 *ptr;
1181
    int dxy, offset, mx, my, src_x, src_y, height, v_edge_pos, linesize, uvlinesize;
1182
    int emu=0;
M
Michael Niedermayer 已提交
1183
#if 0    
M
Michael Niedermayer 已提交
1184 1185 1186 1187 1188
if(s->quarter_sample)
{
    motion_x>>=1;
    motion_y>>=1;
}
M
Michael Niedermayer 已提交
1189
#endif
F
Fabrice Bellard 已提交
1190 1191 1192 1193 1194 1195
    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;
1196
    v_edge_pos = s->v_edge_pos >> field_based;
F
Fabrice Bellard 已提交
1197 1198 1199 1200 1201 1202
    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;
1203 1204
    linesize   = s->linesize << field_based;
    uvlinesize = s->uvlinesize << field_based;
F
Fabrice Bellard 已提交
1205 1206
    ptr = ref_picture[0] + (src_y * linesize) + (src_x) + src_offset;
    dest_y += dest_offset;
1207 1208

    if(s->flags&CODEC_FLAG_EMU_EDGE){
1209 1210 1211
        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){
            emulated_edge_mc(s, ptr, linesize, 17, h+1, src_x, src_y, s->h_edge_pos, v_edge_pos);
1212 1213 1214 1215
            ptr= s->edge_emu_buffer;
            emu=1;
        }
    }
M
Michael Niedermayer 已提交
1216
    pix_op[0][dxy](dest_y, ptr, linesize, h);
F
Fabrice Bellard 已提交
1217

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

F
Fabrice Bellard 已提交
1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243
    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;
1244
    offset = (src_y * uvlinesize) + src_x + (src_offset >> 1);
F
Fabrice Bellard 已提交
1245
    ptr = ref_picture[1] + offset;
1246
    if(emu){
1247
        emulated_edge_mc(s, ptr, uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, v_edge_pos>>1);
1248 1249
        ptr= s->edge_emu_buffer;
    }
M
Michael Niedermayer 已提交
1250
    pix_op[1][dxy](dest_cb + (dest_offset >> 1), ptr, uvlinesize, h >> 1);
1251

F
Fabrice Bellard 已提交
1252
    ptr = ref_picture[2] + offset;
1253
    if(emu){
1254
        emulated_edge_mc(s, ptr, uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, v_edge_pos>>1);
1255 1256
        ptr= s->edge_emu_buffer;
    }
M
Michael Niedermayer 已提交
1257
    pix_op[1][dxy](dest_cr + (dest_offset >> 1), ptr, uvlinesize, h >> 1);
F
Fabrice Bellard 已提交
1258 1259
}

M
Michael Niedermayer 已提交
1260 1261 1262 1263
static inline void qpel_motion(MpegEncContext *s,
                               UINT8 *dest_y, UINT8 *dest_cb, UINT8 *dest_cr,
                               int dest_offset,
                               UINT8 **ref_picture, int src_offset,
M
Michael Niedermayer 已提交
1264 1265
                               int field_based, op_pixels_func (*pix_op)[4],
                               qpel_mc_func (*qpix_op)[16],
M
Michael Niedermayer 已提交
1266 1267 1268
                               int motion_x, int motion_y, int h)
{
    UINT8 *ptr;
1269
    int dxy, offset, mx, my, src_x, src_y, height, v_edge_pos, linesize, uvlinesize;
M
Michael Niedermayer 已提交
1270
    int emu=0;
M
Michael Niedermayer 已提交
1271 1272 1273 1274 1275 1276

    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;
1277
    v_edge_pos = s->v_edge_pos >> field_based;
M
Michael Niedermayer 已提交
1278 1279 1280 1281 1282 1283 1284
    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;
1285
    uvlinesize = s->uvlinesize << field_based;
M
Michael Niedermayer 已提交
1286 1287 1288
    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 已提交
1289 1290
    
    if(s->flags&CODEC_FLAG_EMU_EDGE){
1291 1292 1293
        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){
            emulated_edge_mc(s, ptr, linesize, 17, h+1, src_x, src_y, s->h_edge_pos, v_edge_pos);
M
Michael Niedermayer 已提交
1294 1295 1296 1297
            ptr= s->edge_emu_buffer;
            emu=1;
        }
    }
1298 1299 1300 1301 1302 1303 1304 1305
    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 已提交
1306

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

1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323
    if(field_based){
        mx= motion_x/2;
        my= motion_y>>1;
    }else if(s->divx_version){
        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);
    dxy= (mx&1) | ((my&1)<<1);
    mx>>=1;
    my>>=1;
M
Michael Niedermayer 已提交
1324 1325 1326 1327 1328 1329 1330 1331 1332 1333

    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;

1334
    offset = (src_y * uvlinesize) + src_x + (src_offset >> 1);
M
Michael Niedermayer 已提交
1335
    ptr = ref_picture[1] + offset;
M
Michael Niedermayer 已提交
1336
    if(emu){
1337
        emulated_edge_mc(s, ptr,  uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, v_edge_pos>>1);
M
Michael Niedermayer 已提交
1338 1339
        ptr= s->edge_emu_buffer;
    }
1340
    pix_op[1][dxy](dest_cb + (dest_offset >> 1), ptr,  uvlinesize, h >> 1);
M
Michael Niedermayer 已提交
1341
    
M
Michael Niedermayer 已提交
1342
    ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1343
    if(emu){
1344
        emulated_edge_mc(s, ptr,  uvlinesize, 9, (h>>1)+1, src_x, src_y, s->h_edge_pos>>1, v_edge_pos>>1);
M
Michael Niedermayer 已提交
1345 1346
        ptr= s->edge_emu_buffer;
    }
1347
    pix_op[1][dxy](dest_cr + (dest_offset >> 1), ptr,  uvlinesize, h >> 1);
M
Michael Niedermayer 已提交
1348 1349 1350
}


F
Fabrice Bellard 已提交
1351 1352 1353
static inline void MPV_motion(MpegEncContext *s, 
                              UINT8 *dest_y, UINT8 *dest_cb, UINT8 *dest_cr,
                              int dir, UINT8 **ref_picture, 
M
Michael Niedermayer 已提交
1354
                              op_pixels_func (*pix_op)[4], qpel_mc_func (*qpix_op)[16])
F
Fabrice Bellard 已提交
1355 1356 1357 1358
{
    int dxy, offset, mx, my, src_x, src_y, motion_x, motion_y;
    int mb_x, mb_y, i;
    UINT8 *ptr, *dest;
M
Michael Niedermayer 已提交
1359
    int emu=0;
F
Fabrice Bellard 已提交
1360 1361 1362 1363 1364 1365

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

    switch(s->mv_type) {
    case MV_TYPE_16X16:
M
Michael Niedermayer 已提交
1366 1367 1368 1369
        if(s->mcsel){
            gmc1_motion(s, dest_y, dest_cb, dest_cr, 0,
                        ref_picture, 0,
                        16);
M
Michael Niedermayer 已提交
1370
        }else if(s->quarter_sample){
M
Michael Niedermayer 已提交
1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
            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);
        }else{
            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 已提交
1381 1382
        break;
    case MV_TYPE_8X8:
M
Michael Niedermayer 已提交
1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
        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){
                        emulated_edge_mc(s, ptr, s->linesize, 9, 9, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
                        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 已提交
1420

M
Michael Niedermayer 已提交
1421 1422 1423
                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 已提交
1424
                    
M
Michael Niedermayer 已提交
1425 1426 1427 1428 1429 1430 1431
                /* 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 已提交
1432
                    
M
Michael Niedermayer 已提交
1433 1434 1435 1436 1437 1438 1439
                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){
                        emulated_edge_mc(s, ptr, s->linesize, 9, 9, src_x, src_y, s->h_edge_pos, s->v_edge_pos);
                        ptr= s->edge_emu_buffer;
                    }
M
Michael Niedermayer 已提交
1440
                }
M
Michael Niedermayer 已提交
1441 1442 1443 1444 1445
                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 已提交
1446
            }
F
Fabrice Bellard 已提交
1447
        }
M
Michael Niedermayer 已提交
1448

M
Michael Niedermayer 已提交
1449
        if(s->flags&CODEC_FLAG_GRAY) break;
F
Fabrice Bellard 已提交
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
        /* In case of 8X8, we construct a single chroma motion vector
           with a special rounding */
        for(i=0;i<4;i++) {
        }
        if (mx >= 0)
            mx = (h263_chroma_roundtab[mx & 0xf] + ((mx >> 3) & ~1));
        else {
            mx = -mx;
            mx = -(h263_chroma_roundtab[mx & 0xf] + ((mx >> 3) & ~1));
        }
        if (my >= 0)
            my = (h263_chroma_roundtab[my & 0xf] + ((my >> 3) & ~1));
        else {
            my = -my;
            my = -(h263_chroma_roundtab[my & 0xf] + ((my >> 3) & ~1));
        }
        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 已提交
1479
        offset = (src_y * (s->uvlinesize)) + src_x;
F
Fabrice Bellard 已提交
1480
        ptr = ref_picture[1] + offset;
M
Michael Niedermayer 已提交
1481
        if(s->flags&CODEC_FLAG_EMU_EDGE){
1482 1483 1484
                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){
                    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 已提交
1485 1486 1487 1488
                    ptr= s->edge_emu_buffer;
                    emu=1;
                }
            }
M
Michael Niedermayer 已提交
1489
        pix_op[1][dxy](dest_cb, ptr, s->uvlinesize, 8);
M
Michael Niedermayer 已提交
1490

F
Fabrice Bellard 已提交
1491
        ptr = ref_picture[2] + offset;
M
Michael Niedermayer 已提交
1492
        if(emu){
1493
            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 已提交
1494 1495
            ptr= s->edge_emu_buffer;
        }
M
Michael Niedermayer 已提交
1496
        pix_op[1][dxy](dest_cr, ptr, s->uvlinesize, 8);
F
Fabrice Bellard 已提交
1497 1498 1499
        break;
    case MV_TYPE_FIELD:
        if (s->picture_structure == PICT_FRAME) {
1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522
            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 已提交
1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535
        } else {
            

        }
        break;
    }
}


/* put block[] to dest[] */
static inline void put_dct(MpegEncContext *s, 
                           DCTELEM *block, int i, UINT8 *dest, int line_size)
{
1536
    s->dct_unquantize(s, block, i, s->qscale);
1537
    s->idct_put (dest, line_size, block);
F
Fabrice Bellard 已提交
1538 1539 1540 1541 1542 1543
}

/* add block[] to dest[] */
static inline void add_dct(MpegEncContext *s, 
                           DCTELEM *block, int i, UINT8 *dest, int line_size)
{
M
Michael Niedermayer 已提交
1544
    if (s->block_last_index[i] >= 0) {
1545
        s->idct_add (dest, line_size, block);
M
Michael Niedermayer 已提交
1546 1547
    }
}
1548

M
Michael Niedermayer 已提交
1549 1550 1551
static inline void add_dequant_dct(MpegEncContext *s, 
                           DCTELEM *block, int i, UINT8 *dest, int line_size)
{
F
Fabrice Bellard 已提交
1552
    if (s->block_last_index[i] >= 0) {
M
Michael Niedermayer 已提交
1553
        s->dct_unquantize(s, block, i, s->qscale);
1554

1555
        s->idct_add (dest, line_size, block);
F
Fabrice Bellard 已提交
1556 1557 1558
    }
}

1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591
/**
 * 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 */
    memset(s->ac_val[0][xy       ], 0, 32 * sizeof(INT16));
    memset(s->ac_val[0][xy + wrap], 0, 32 * sizeof(INT16));
    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 */
    memset(s->ac_val[1][xy], 0, 16 * sizeof(INT16));
    memset(s->ac_val[2][xy], 0, 16 * sizeof(INT16));
    
    s->mbintra_table[s->mb_x + s->mb_y*s->mb_width]= 0;
}

F
Fabrice Bellard 已提交
1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603
/* 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])
{
1604
    int mb_x, mb_y;
M
Michael Niedermayer 已提交
1605
    const int mb_xy = s->mb_y * s->mb_width + s->mb_x;
F
Fabrice Bellard 已提交
1606 1607 1608 1609

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

1610
#ifdef FF_POSTPROCESS
1611
    /* Obsolete. Exists for compatibility with mplayer only. */
1612 1613
    quant_store[mb_y][mb_x]=s->qscale;
    //printf("[%02d][%02d] %d\n",mb_x,mb_y,s->qscale);
1614
#else
M
Michael Niedermayer 已提交
1615
    /* even more obsolete, exists for mplayer xp only */
1616
    if(s->avctx->quant_store) s->avctx->quant_store[mb_y*s->avctx->qstride+mb_x] = s->qscale;
1617
#endif
M
Michael Niedermayer 已提交
1618
    s->qscale_table[mb_xy]= s->qscale;
1619

F
Fabrice Bellard 已提交
1620 1621
    /* update DC predictors for P macroblocks */
    if (!s->mb_intra) {
1622
        if (s->h263_pred || s->h263_aic) {
M
Michael Niedermayer 已提交
1623
            if(s->mbintra_table[mb_xy])
1624
                ff_clean_intra_table_entries(s);
F
Fabrice Bellard 已提交
1625
        } else {
1626 1627
            s->last_dc[0] =
            s->last_dc[1] =
F
Fabrice Bellard 已提交
1628 1629 1630
            s->last_dc[2] = 128 << s->intra_dc_precision;
        }
    }
1631
    else if (s->h263_pred || s->h263_aic)
M
Michael Niedermayer 已提交
1632
        s->mbintra_table[mb_xy]=1;
1633

1634
    /* update motion predictor, not for B-frames as they need the motion_val from the last P/S-Frame */
M
Michael Niedermayer 已提交
1635
    if (s->out_format == FMT_H263 && s->pict_type!=B_TYPE) { //FIXME move into h263.c if possible, format specific stuff shouldnt be here
F
Fabrice Bellard 已提交
1636
        
M
Michael Niedermayer 已提交
1637 1638
        const int wrap = s->block_wrap[0];
        const int xy = s->block_index[0];
1639
        const int mb_index= s->mb_x + s->mb_y*s->mb_width;
1640
        if(s->mv_type == MV_TYPE_8X8){
1641
            s->co_located_type_table[mb_index]= CO_LOCATED_TYPE_4MV;
1642 1643 1644 1645 1646
        } else {
            int motion_x, motion_y;
            if (s->mb_intra) {
                motion_x = 0;
                motion_y = 0;
1647 1648
                if(s->co_located_type_table)
                    s->co_located_type_table[mb_index]= 0;
1649 1650 1651
            } else if (s->mv_type == MV_TYPE_16X16) {
                motion_x = s->mv[0][0][0];
                motion_y = s->mv[0][0][1];
1652 1653
                if(s->co_located_type_table)
                    s->co_located_type_table[mb_index]= 0;
1654
            } else /*if (s->mv_type == MV_TYPE_FIELD)*/ {
1655
                int i;
1656 1657 1658
                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);
1659 1660 1661 1662 1663 1664
                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;
1665
            }
F
Fabrice Bellard 已提交
1666
            /* no update if 8X8 because it has been done during parsing */
1667 1668 1669 1670 1671 1672 1673 1674
            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 已提交
1675 1676 1677
        }
    }
    
1678
    if (!(s->encoding && (s->intra_only || s->pict_type==B_TYPE))) {
F
Fabrice Bellard 已提交
1679
        UINT8 *dest_y, *dest_cb, *dest_cr;
M
Michael Niedermayer 已提交
1680
        int dct_linesize, dct_offset;
M
Michael Niedermayer 已提交
1681 1682
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
1683

1684 1685 1686
        /* avoid copy if macroblock skipped in last frame too 
           dont touch it for B-frames as they need the skip info from the next p-frame */
        if (s->pict_type != B_TYPE) {
M
Michael Niedermayer 已提交
1687
            UINT8 *mbskip_ptr = &s->mbskip_table[mb_xy];
1688 1689
            if (s->mb_skiped) {
                s->mb_skiped = 0;
M
Michael Niedermayer 已提交
1690 1691 1692 1693

                (*mbskip_ptr) ++; /* indicate that this time we skiped it */
                if(*mbskip_ptr >99) *mbskip_ptr= 99;

1694 1695
                /* if previous was skipped too, then nothing to do ! 
                   skip only during decoding as we might trash the buffers during encoding a bit */
M
Michael Niedermayer 已提交
1696
                if (*mbskip_ptr >= s->ip_buffer_count  && !s->encoding) 
1697 1698 1699 1700 1701
                    goto the_end;
            } else {
                *mbskip_ptr = 0; /* not skipped */
            }
        }
F
Fabrice Bellard 已提交
1702

M
Michael Niedermayer 已提交
1703 1704 1705 1706 1707 1708 1709 1710 1711
        if(s->pict_type==B_TYPE && s->avctx->draw_horiz_band){
            dest_y = s->current_picture [0] + mb_x * 16;
            dest_cb = s->current_picture[1] + mb_x * 8;
            dest_cr = s->current_picture[2] + mb_x * 8;
        }else{
            dest_y = s->current_picture [0] + (mb_y * 16* s->linesize  ) + mb_x * 16;
            dest_cb = s->current_picture[1] + (mb_y * 8 * s->uvlinesize) + mb_x * 8;
            dest_cr = s->current_picture[2] + (mb_y * 8 * s->uvlinesize) + mb_x * 8;
        }
F
Fabrice Bellard 已提交
1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722

        if (s->interlaced_dct) {
            dct_linesize = s->linesize * 2;
            dct_offset = s->linesize;
        } else {
            dct_linesize = s->linesize;
            dct_offset = s->linesize * 8;
        }

        if (!s->mb_intra) {
            /* motion handling */
1723
            /* decoding or more than one mb_type (MC was allready done otherwise) */
M
Michael Niedermayer 已提交
1724
            if((!s->encoding) || (s->mb_type[mb_xy]&(s->mb_type[mb_xy]-1))){
1725
                if ((!s->no_rounding) || s->pict_type==B_TYPE){                
1726
                    op_pix = put_pixels_tab;
M
Michael Niedermayer 已提交
1727
                    op_qpix= put_qpel_pixels_tab;
1728 1729
                }else{
                    op_pix = put_no_rnd_pixels_tab;
M
Michael Niedermayer 已提交
1730
                    op_qpix= put_no_rnd_qpel_pixels_tab;
1731
                }
F
Fabrice Bellard 已提交
1732

1733 1734
                if (s->mv_dir & MV_DIR_FORWARD) {
                    MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture, op_pix, op_qpix);
M
Michael Niedermayer 已提交
1735 1736
                    op_pix = avg_pixels_tab;
                    op_qpix= avg_qpel_pixels_tab;
1737 1738 1739 1740
                }
                if (s->mv_dir & MV_DIR_BACKWARD) {
                    MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture, op_pix, op_qpix);
                }
F
Fabrice Bellard 已提交
1741 1742
            }

M
Michael Niedermayer 已提交
1743 1744 1745
            /* skip dequant / idct if we are really late ;) */
            if(s->hurry_up>1) goto the_end;

F
Fabrice Bellard 已提交
1746
            /* add dct residue */
1747 1748
            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 已提交
1749 1750 1751 1752 1753
                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 已提交
1754
                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1755 1756
                    add_dequant_dct(s, block[4], 4, dest_cb, s->uvlinesize);
                    add_dequant_dct(s, block[5], 5, dest_cr, s->uvlinesize);
M
Michael Niedermayer 已提交
1757
                }
M
Michael Niedermayer 已提交
1758 1759 1760 1761 1762
            } else {
                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 已提交
1763

M
Michael Niedermayer 已提交
1764
                if(!(s->flags&CODEC_FLAG_GRAY)){
M
Michael Niedermayer 已提交
1765 1766
                    add_dct(s, block[4], 4, dest_cb, s->uvlinesize);
                    add_dct(s, block[5], 5, dest_cr, s->uvlinesize);
M
Michael Niedermayer 已提交
1767
                }
M
Michael Niedermayer 已提交
1768
            }
F
Fabrice Bellard 已提交
1769 1770
        } else {
            /* dct only in intra block */
1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
            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)){
                    put_dct(s, block[4], 4, dest_cb, s->uvlinesize);
                    put_dct(s, block[5], 5, dest_cr, s->uvlinesize);
                }
            }else{
                s->idct_put(dest_y                 , dct_linesize, block[0]);
                s->idct_put(dest_y              + 8, dct_linesize, block[1]);
                s->idct_put(dest_y + dct_offset    , dct_linesize, block[2]);
                s->idct_put(dest_y + dct_offset + 8, dct_linesize, block[3]);

                if(!(s->flags&CODEC_FLAG_GRAY)){
                    s->idct_put(dest_cb, s->uvlinesize, block[4]);
                    s->idct_put(dest_cr, s->uvlinesize, block[5]);
                }
M
Michael Niedermayer 已提交
1791
            }
F
Fabrice Bellard 已提交
1792 1793
        }
    }
1794
 the_end:
1795
    emms_c(); //FIXME remove
F
Fabrice Bellard 已提交
1796 1797
}

1798
static inline void dct_single_coeff_elimination(MpegEncContext *s, int n, int threshold)
1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813
{
    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];
1814
    int skip_dc;
1815

1816 1817 1818
    if(threshold<0){
        skip_dc=0;
        threshold= -threshold;
1819 1820
    }else
        skip_dc=1;
1821

1822 1823 1824 1825
    /* 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++){
1826
        const int j = s->intra_scantable.permutated[i];
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839
        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++){
1840
        const int j = s->intra_scantable.permutated[i];
1841 1842 1843 1844 1845 1846
        block[j]=0;
    }
    if(block[0]) s->block_last_index[n]= 0;
    else         s->block_last_index[n]= -1;
}

1847 1848 1849 1850 1851
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;
1852 1853 1854 1855 1856 1857 1858 1859
    
    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];
1860 1861 1862 1863 1864 1865 1866
        int level = block[j];
       
        if     (level>maxlevel) level=maxlevel;
        else if(level<minlevel) level=minlevel;
        block[j]= level;
    }
}
1867

1868 1869 1870
static inline void requantize_coeffs(MpegEncContext *s, DCTELEM block[64], int oldq, int newq, int n)
{
    int i;
1871 1872 1873 1874

    if(s->mb_intra){
        i=1; //skip clipping of intra dc
         //FIXME requantize, note (mpeg1/h263/h263p-aic dont need it,...)
1875 1876 1877 1878
    }else
        i=0;
    
    for(;i<=s->block_last_index[n]; i++){
1879
        const int j = s->intra_scantable.permutated[i];
1880 1881 1882 1883 1884 1885
        int level = block[j];
        
        block[j]= ROUNDED_DIV(level*oldq, newq);
    }

    for(i=s->block_last_index[n]; i>=0; i--){
1886
        const int j = s->intra_scantable.permutated[i];
1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901
        if(block[j]) break;
    }
    s->block_last_index[n]= i;
}

static inline void auto_requantize_coeffs(MpegEncContext *s, DCTELEM block[6][64])
{
    int i,n, newq;
    const int maxlevel= s->max_qcoeff;
    const int minlevel= s->min_qcoeff;
    int largest=0, smallest=0;

    assert(s->adaptive_quant);
    
    for(n=0; n<6; n++){
1902 1903 1904 1905 1906
        if(s->mb_intra){
            i=1; //skip clipping of intra dc
             //FIXME requantize, note (mpeg1/h263/h263p-aic dont need it,...)
        }else
            i=0;
1907 1908

        for(;i<=s->block_last_index[n]; i++){
1909
            const int j = s->intra_scantable.permutated[i];
1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999
            int level = block[n][j];
            if(largest  < level) largest = level;
            if(smallest > level) smallest= level;
        }
    }
    
    for(newq=s->qscale+1; newq<32; newq++){
        if(   ROUNDED_DIV(smallest*s->qscale, newq) >= minlevel
           && ROUNDED_DIV(largest *s->qscale, newq) <= maxlevel) 
            break;
    }
        
    if(s->out_format==FMT_H263){
        /* h263 like formats cannot change qscale by more than 2 easiely */
        if(s->avctx->qmin + 2 < newq)
            newq= s->avctx->qmin + 2;
    }

    for(n=0; n<6; n++){
        requantize_coeffs(s, block[n], s->qscale, newq, n);
        clip_coeffs(s, block[n], s->block_last_index[n]);
    }
     
    s->dquant+= newq - s->qscale;
    s->qscale= newq;
}
#if 0
static int pix_vcmp16x8(UINT8 *s, int stride){ //FIXME move to dsputil & optimize
    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;
}

static int pix_diff_vcmp16x8(UINT8 *s1, UINT8*s2, int stride){ //FIXME move to dsputil & optimize
    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))

static int pix_vcmp16x8(UINT8 *s, int stride){ //FIXME move to dsputil & optimize
    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;
}

static int pix_diff_vcmp16x8(UINT8 *s1, UINT8*s2, int stride){ //FIXME move to dsputil & optimize
    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
2000
static void encode_mb(MpegEncContext *s, int motion_x, int motion_y)
F
Fabrice Bellard 已提交
2001
{
2002 2003 2004
    const int mb_x= s->mb_x;
    const int mb_y= s->mb_y;
    int i;
2005
    int skip_dct[6];
2006 2007
    int dct_offset   = s->linesize*8; //default for progressive frames
    
2008
    for(i=0; i<6; i++) skip_dct[i]=0;
2009 2010 2011
    
    if(s->adaptive_quant){
        s->dquant= s->qscale_table[mb_x + mb_y*s->mb_width] - s->qscale;
2012 2013

        if(s->out_format==FMT_H263){
2014 2015
            if     (s->dquant> 2) s->dquant= 2;
            else if(s->dquant<-2) s->dquant=-2;
2016 2017 2018
        }
            
        if(s->codec_id==CODEC_ID_MPEG4){        
2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029
            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 ];
    }
2030

2031 2032
    if (s->mb_intra) {
        UINT8 *ptr;
2033
        int wrap_y;
2034
        int emu=0;
2035

2036 2037 2038
        wrap_y = s->linesize;
        ptr = s->new_picture[0] + (mb_y * 16 * wrap_y) + mb_x * 16;

2039
        if(mb_x*16+16 > s->width || mb_y*16+16 > s->height){
2040
            emulated_edge_mc(s, ptr, wrap_y, 16, 16, mb_x*16, mb_y*16, s->width, s->height);
2041 2042 2043
            ptr= s->edge_emu_buffer;
            emu=1;
        }
2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063
        
        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;
        }
        
        get_pixels(s->block[0], ptr                 , wrap_y);
        get_pixels(s->block[1], ptr              + 8, wrap_y);
        get_pixels(s->block[2], ptr + dct_offset    , wrap_y);
        get_pixels(s->block[3], ptr + dct_offset + 8, wrap_y);
2064

M
Michael Niedermayer 已提交
2065 2066 2067 2068
        if(s->flags&CODEC_FLAG_GRAY){
            skip_dct[4]= 1;
            skip_dct[5]= 1;
        }else{
2069 2070
            int wrap_c = s->uvlinesize;
            ptr = s->new_picture[1] + (mb_y * 8 * wrap_c) + mb_x * 8;
2071
            if(emu){
2072
                emulated_edge_mc(s, ptr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2073 2074
                ptr= s->edge_emu_buffer;
            }
2075
            get_pixels(s->block[4], ptr, wrap_c);
2076

2077
            ptr = s->new_picture[2] + (mb_y * 8 * wrap_c) + mb_x * 8;
2078
            if(emu){
2079
                emulated_edge_mc(s, ptr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
2080 2081
                ptr= s->edge_emu_buffer;
            }
2082
            get_pixels(s->block[5], ptr, wrap_c);
M
Michael Niedermayer 已提交
2083
        }
2084
    }else{
M
Michael Niedermayer 已提交
2085 2086
        op_pixels_func (*op_pix)[4];
        qpel_mc_func (*op_qpix)[16];
2087
        UINT8 *dest_y, *dest_cb, *dest_cr;
2088 2089
        UINT8 *ptr_y, *ptr_cb, *ptr_cr;
        int wrap_y, wrap_c;
2090
        int emu=0;
2091 2092

        dest_y  = s->current_picture[0] + (mb_y * 16 * s->linesize       ) + mb_x * 16;
M
Michael Niedermayer 已提交
2093 2094
        dest_cb = s->current_picture[1] + (mb_y * 8  * (s->uvlinesize)) + mb_x * 8;
        dest_cr = s->current_picture[2] + (mb_y * 8  * (s->uvlinesize)) + mb_x * 8;
2095
        wrap_y = s->linesize;
2096
        wrap_c = s->uvlinesize;
2097 2098 2099
        ptr_y  = s->new_picture[0] + (mb_y * 16 * wrap_y) + mb_x * 16;
        ptr_cb = s->new_picture[1] + (mb_y * 8 * wrap_c) + mb_x * 8;
        ptr_cr = s->new_picture[2] + (mb_y * 8 * wrap_c) + mb_x * 8;
2100

2101
        if ((!s->no_rounding) || s->pict_type==B_TYPE){
2102
            op_pix = put_pixels_tab;
M
Michael Niedermayer 已提交
2103
            op_qpix= put_qpel_pixels_tab;
M
Michael Niedermayer 已提交
2104
        }else{
2105
            op_pix = put_no_rnd_pixels_tab;
M
Michael Niedermayer 已提交
2106
            op_qpix= put_no_rnd_qpel_pixels_tab;
2107 2108 2109 2110
        }

        if (s->mv_dir & MV_DIR_FORWARD) {
            MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture, op_pix, op_qpix);
M
Michael Niedermayer 已提交
2111 2112
            op_pix = avg_pixels_tab;
            op_qpix= avg_qpel_pixels_tab;
2113 2114 2115
        }
        if (s->mv_dir & MV_DIR_BACKWARD) {
            MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture, op_pix, op_qpix);
2116
        }
2117

2118 2119 2120 2121 2122
        if(mb_x*16+16 > s->width || mb_y*16+16 > s->height){
            emulated_edge_mc(s, ptr_y, wrap_y, 16, 16, mb_x*16, mb_y*16, s->width, s->height);
            ptr_y= s->edge_emu_buffer;
            emu=1;
        }
2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140
        
        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;
        }
        
2141 2142
        diff_pixels(s->block[0], ptr_y                 , dest_y                 , wrap_y);
        diff_pixels(s->block[1], ptr_y              + 8, dest_y              + 8, wrap_y);
2143 2144
        diff_pixels(s->block[2], ptr_y + dct_offset    , dest_y + dct_offset    , wrap_y);
        diff_pixels(s->block[3], ptr_y + dct_offset + 8, dest_y + dct_offset + 8, wrap_y);
M
Michael Niedermayer 已提交
2145 2146 2147 2148 2149
        
        if(s->flags&CODEC_FLAG_GRAY){
            skip_dct[4]= 1;
            skip_dct[5]= 1;
        }else{
2150 2151 2152 2153
            if(emu){
                emulated_edge_mc(s, ptr_cb, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
                ptr_cb= s->edge_emu_buffer;
            }
M
Michael Niedermayer 已提交
2154
            diff_pixels(s->block[4], ptr_cb, dest_cb, wrap_c);
2155 2156 2157 2158
            if(emu){
                emulated_edge_mc(s, ptr_cr, wrap_c, 8, 8, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
                ptr_cr= s->edge_emu_buffer;
            }
M
Michael Niedermayer 已提交
2159 2160 2161
            diff_pixels(s->block[5], ptr_cr, dest_cr, wrap_c);
        }

2162 2163
        /* pre quantization */         
        if(s->mc_mb_var[s->mb_width*mb_y+ mb_x]<2*s->qscale*s->qscale){
2164
            //FIXME optimize
2165 2166
            if(pix_abs8x8(ptr_y               , dest_y               , wrap_y) < 20*s->qscale) skip_dct[0]= 1;
            if(pix_abs8x8(ptr_y            + 8, dest_y            + 8, wrap_y) < 20*s->qscale) skip_dct[1]= 1;
2167 2168
            if(pix_abs8x8(ptr_y +dct_offset   , dest_y +dct_offset   , wrap_y) < 20*s->qscale) skip_dct[2]= 1;
            if(pix_abs8x8(ptr_y +dct_offset+ 8, dest_y +dct_offset+ 8, wrap_y) < 20*s->qscale) skip_dct[3]= 1;
2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186
            if(pix_abs8x8(ptr_cb              , dest_cb              , wrap_y) < 20*s->qscale) skip_dct[4]= 1;
            if(pix_abs8x8(ptr_cr              , dest_cr              , wrap_y) < 20*s->qscale) skip_dct[5]= 1;
#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
        }
2187

2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203
    }
            
#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 */
2204 2205 2206
    if(s->out_format==FMT_MJPEG){
        for(i=0;i<6;i++) {
            int overflow;
2207
            s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, 8, &overflow);
2208
            if (overflow) clip_coeffs(s, s->block[i], s->block_last_index[i]);
2209 2210 2211
        }
    }else{
        for(i=0;i<6;i++) {
2212 2213
            if(!skip_dct[i]){
                int overflow;
2214
                s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, s->qscale, &overflow);
2215
            // FIXME we could decide to change to quantizer instead of clipping
2216 2217
            // 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
2218 2219 2220
                if (overflow) clip_coeffs(s, s->block[i], s->block_last_index[i]);
            }else
                s->block_last_index[i]= -1;
2221
        }
2222 2223
        if(s->luma_elim_threshold && !s->mb_intra)
            for(i=0; i<4; i++)
2224
                dct_single_coeff_elimination(s, i, s->luma_elim_threshold);
2225 2226
        if(s->chroma_elim_threshold && !s->mb_intra)
            for(i=4; i<6; i++)
2227
                dct_single_coeff_elimination(s, i, s->chroma_elim_threshold);
2228 2229
    }

M
Michael Niedermayer 已提交
2230 2231 2232 2233 2234 2235 2236
    if((s->flags&CODEC_FLAG_GRAY) && s->mb_intra){
        s->block_last_index[4]=
        s->block_last_index[5]= 0;
        s->block[4][0]=
        s->block[5][0]= 128;
    }

2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255
    /* huffman encode */
    switch(s->out_format) {
    case FMT_MPEG1:
        mpeg1_encode_mb(s, s->block, motion_x, motion_y);
        break;
    case FMT_H263:
        if (s->h263_msmpeg4)
            msmpeg4_encode_mb(s, s->block, motion_x, motion_y);
        else if(s->h263_pred)
            mpeg4_encode_mb(s, s->block, motion_x, motion_y);
        else
            h263_encode_mb(s, s->block, motion_x, motion_y);
        break;
    case FMT_MJPEG:
        mjpeg_encode_mb(s, s->block);
        break;
    }
}

2256
void ff_copy_bits(PutBitContext *pb, UINT8 *src, int length)
2257
{
2258 2259 2260 2261
    int bytes= length>>4;
    int bits= length&15;
    int i;

2262 2263
    if(length==0) return;

2264 2265
    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));
2266 2267
}

2268
static inline void copy_context_before_encode(MpegEncContext *d, MpegEncContext *s, int type){
2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282
    int i;

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

    /* mpeg1 */
    d->mb_incr= s->mb_incr;
    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;
2283 2284
    d->f_count= s->f_count;
    d->b_count= s->b_count;
2285 2286
    d->skip_count= s->skip_count;
    d->misc_bits= s->misc_bits;
2287
    d->last_bits= 0;
2288 2289

    d->mb_skiped= s->mb_skiped;
2290 2291
}

2292
static inline void copy_context_after_encode(MpegEncContext *d, MpegEncContext *s, int type){
2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307
    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 */
    d->mb_incr= s->mb_incr;
    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;
2308 2309
    d->f_count= s->f_count;
    d->b_count= s->b_count;
2310 2311 2312 2313
    d->skip_count= s->skip_count;
    d->misc_bits= s->misc_bits;

    d->mb_intra= s->mb_intra;
2314
    d->mb_skiped= s->mb_skiped;
2315 2316 2317
    d->mv_type= s->mv_type;
    d->mv_dir= s->mv_dir;
    d->pb= s->pb;
2318 2319 2320 2321
    if(s->data_partitioning){
        d->pb2= s->pb2;
        d->tex_pb= s->tex_pb;
    }
2322 2323 2324 2325 2326
    d->block= s->block;
    for(i=0; i<6; i++)
        d->block_last_index[i]= s->block_last_index[i];
}

2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356
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);
    }
}
2357

2358 2359 2360 2361
static void encode_picture(MpegEncContext *s, int picture_number)
{
    int mb_x, mb_y, last_gob, pdif = 0;
    int i;
2362
    int bits;
2363
    MpegEncContext best_s, backup_s;
2364 2365 2366 2367 2368 2369 2370 2371 2372 2373
    UINT8 bit_buf[2][3000];
    UINT8 bit_buf2[2][3000];
    UINT8 bit_buf_tex[2][3000];
    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 已提交
2374 2375

    s->picture_number = picture_number;
2376

2377 2378 2379 2380 2381 2382 2383
    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;
    
2384
    /* Reset the average MB variance */
2385 2386
    s->mb_var_sum = 0;
    s->mc_mb_var_sum = 0;
2387 2388 2389 2390 2391

    /* we need to initialize some time vars before we can encode b-frames */
    if (s->h263_pred && !s->h263_msmpeg4)
        ff_set_mpeg4_time(s, s->picture_number); 

2392
    s->scene_change_score=0;
2393 2394
    
    s->qscale= (int)(s->frame_qscale + 0.5); //FIXME qscale / ... stuff for ME ratedistoration
2395

2396
    /* Estimate motion for every MB */
2397
    if(s->pict_type != I_TYPE){
2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411
        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;

                /* compute motion vector & mb_type and store in context */
2412 2413 2414 2415
                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);
2416
//                s->mb_type[mb_y*s->mb_width + mb_x]=MB_TYPE_INTER;
2417 2418
            }
        }
2419
        emms_c();
2420
    }else /* if(s->pict_type == I_TYPE) */{
2421 2422 2423
        /* I-Frame */
        //FIXME do we need to zero them?
        memset(s->motion_val[0], 0, sizeof(INT16)*(s->mb_width*2 + 2)*(s->mb_height*2 + 2)*2);
2424
        memset(s->p_mv_table   , 0, sizeof(INT16)*(s->mb_width+2)*(s->mb_height+2)*2);
2425
        memset(s->mb_type      , MB_TYPE_INTRA, sizeof(UINT8)*s->mb_width*s->mb_height);
M
Michael Niedermayer 已提交
2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439
        
        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;
                    uint8_t *pix = s->new_picture[0] + (yy * s->linesize) + xx;
                    int varc;
                    int sum = pix_sum(pix, s->linesize);
    
                    sum= (sum+8)>>4;
                    varc = (pix_norm1(pix, s->linesize) - sum*sum + 500 + 128)>>8;

2440 2441
                    s->mb_var [s->mb_width * mb_y + mb_x] = varc;
                    s->mb_mean[s->mb_width * mb_y + mb_x] = (sum+7)>>4;
M
Michael Niedermayer 已提交
2442 2443 2444 2445
                    s->mb_var_sum    += varc;
                }
            }
        }
2446
    }
2447
    if(s->scene_change_score > 0 && s->pict_type == P_TYPE){
M
Michael Niedermayer 已提交
2448
        s->pict_type= I_TYPE;
2449
        memset(s->mb_type   , MB_TYPE_INTRA, sizeof(UINT8)*s->mb_width*s->mb_height);
2450 2451 2452 2453
        if(s->max_b_frames==0){
            s->input_pict_type= I_TYPE;
            s->input_picture_in_gop_number=0;
        }
M
Michael Niedermayer 已提交
2454
//printf("Scene change detected, encoding as I Frame %d %d\n", s->mb_var_sum, s->mc_mb_var_sum);
M
Michael Niedermayer 已提交
2455
    }
2456 2457 2458 2459 2460 2461 2462
    
    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);
        ff_fix_long_p_mvs(s);
    if(s->pict_type==B_TYPE){
        s->f_code= ff_get_best_fcode(s, s->b_forw_mv_table, MB_TYPE_FORWARD);
        s->b_code= ff_get_best_fcode(s, s->b_back_mv_table, MB_TYPE_BACKWARD);
2463 2464 2465 2466 2467

        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);
2468
    }
2469
    
2470 2471 2472 2473
    if (s->fixed_qscale) 
        s->frame_qscale = s->avctx->quality;
    else
        s->frame_qscale = ff_rate_estimate_qscale(s);
2474

2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485
    if(s->adaptive_quant){
        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;
        }

2486
        s->qscale= s->qscale_table[0];
2487
    }else
2488 2489
        s->qscale= (int)(s->frame_qscale + 0.5);
        
F
Fabrice Bellard 已提交
2490 2491
    if (s->out_format == FMT_MJPEG) {
        /* for mjpeg, we do include qscale in the matrix */
2492
        s->intra_matrix[0] = ff_mpeg1_default_intra_matrix[0];
2493 2494 2495 2496 2497
        for(i=1;i<64;i++){
            int j= s->idct_permutation[i];

            s->intra_matrix[j] = CLAMP_TO_8BIT((ff_mpeg1_default_intra_matrix[i] * s->qscale) >> 3);
        }
2498
        convert_matrix(s, s->q_intra_matrix, s->q_intra_matrix16, 
2499
                       s->q_intra_matrix16_bias, s->intra_matrix, s->intra_quant_bias, 8, 8);
F
Fabrice Bellard 已提交
2500 2501
    }

2502
    s->last_bits= get_bit_count(&s->pb);
F
Fabrice Bellard 已提交
2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520
    switch(s->out_format) {
    case FMT_MJPEG:
        mjpeg_picture_header(s);
        break;
    case FMT_H263:
        if (s->h263_msmpeg4) 
            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;
    case FMT_MPEG1:
        mpeg1_encode_picture_header(s, picture_number);
        break;
    }
2521 2522 2523 2524 2525 2526 2527 2528
    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;
2529 2530
    s->f_count=0;
    s->b_count=0;
2531 2532
    s->skip_count=0;

F
Fabrice Bellard 已提交
2533 2534 2535 2536 2537 2538 2539 2540 2541
    /* init last dc values */
    /* note: quant matrix value (8) is implied here */
    s->last_dc[0] = 128;
    s->last_dc[1] = 128;
    s->last_dc[2] = 128;
    s->mb_incr = 1;
    s->last_mv[0][0][0] = 0;
    s->last_mv[0][0][1] = 0;

2542
    /* Get the GOB height based on picture height */
2543
    if (s->out_format == FMT_H263 && !s->h263_pred && !s->h263_msmpeg4) {
2544 2545 2546 2547 2548 2549
        if (s->height <= 400)
            s->gob_index = 1;
        else if (s->height <= 800)
            s->gob_index = 2;
        else
            s->gob_index = 4;
2550 2551
    }else if(s->codec_id==CODEC_ID_MPEG4){
        s->gob_index = 1;
2552
    }
2553 2554 2555 2556 2557 2558

    if(s->codec_id==CODEC_ID_MPEG4 && s->data_partitioning && s->pict_type!=B_TYPE)
        ff_mpeg4_init_partitions(s);

    s->resync_mb_x=0;
    s->resync_mb_y=0;
2559
    for(mb_y=0; mb_y < s->mb_height; mb_y++) {
2560
        /* Put GOB header based on RTP MTU for formats which support it per line (H263*)*/
2561 2562 2563 2564 2565 2566
        /* TODO: Put all this stuff in a separate generic function */
        if (s->rtp_mode) {
            if (!mb_y) {
                s->ptr_lastgob = s->pb.buf;
                s->ptr_last_mb_line = s->pb.buf;
            } else if (s->out_format == FMT_H263 && !s->h263_pred && !s->h263_msmpeg4 && !(mb_y % s->gob_index)) {
2567
                // MN: we could move the space check from h263 -> here, as its not h263 specific
2568 2569
                last_gob = h263_encode_gob_header(s, mb_y);
                if (last_gob) {
2570 2571 2572 2573 2574
                    s->first_slice_line = 1;
                }else{
                    /*MN: we reset it here instead at the end of each line cuz mpeg4 can have 
                          slice lines starting & ending in the middle*/
                    s->first_slice_line = 0;
2575 2576 2577
                }
            }
        }
M
Michael Niedermayer 已提交
2578 2579 2580

        s->y_dc_scale= s->y_dc_scale_table[ s->qscale ];
        s->c_dc_scale= s->c_dc_scale_table[ s->qscale ];
2581
        
M
Michael Niedermayer 已提交
2582 2583 2584 2585 2586 2587
        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);
2588
        for(mb_x=0; mb_x < s->mb_width; mb_x++) {
2589 2590
            const int mb_type= s->mb_type[mb_y * s->mb_width + mb_x];
            const int xy= (mb_y+1) * (s->mb_width+2) + mb_x + 1;
2591
//            int d;
2592
            int dmin=10000000;
2593 2594 2595

            s->mb_x = mb_x;
            s->mb_y = mb_y;
M
Michael Niedermayer 已提交
2596 2597 2598 2599 2600 2601
            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]++;
2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634
            
            /* write gob / video packet header for formats which support it at any MB (MPEG4) */
            if(s->rtp_mode && s->mb_y>0 && s->codec_id==CODEC_ID_MPEG4){
                int pdif= pbBufPtr(&s->pb) - s->ptr_lastgob;

                //the *2 is there so we stay below the requested size
                if(pdif + s->mb_line_avgsize/s->mb_width >= s->rtp_payload_size){ 
                    if(s->codec_id==CODEC_ID_MPEG4){
                        if(s->data_partitioning && s->pict_type!=B_TYPE){
                            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);
                    }
                    s->ptr_lastgob = pbBufPtr(&s->pb);
                    s->first_slice_line=1;
                    s->resync_mb_x=mb_x;
                    s->resync_mb_y=mb_y;
                }

                if(  (s->resync_mb_x   == s->mb_x)
                   && s->resync_mb_y+1 == s->mb_y){
                    s->first_slice_line=0; 
                }
            }

2635
            if(mb_type & (mb_type-1)){ // more than 1 MB type possible
2636
                int next_block=0;
2637
                int pb_bits_count, pb2_bits_count, tex_pb_bits_count;
2638 2639

                copy_context_before_encode(&backup_s, s, -1);
2640 2641 2642 2643 2644 2645
                backup_s.pb= s->pb;
                best_s.data_partitioning= s->data_partitioning;
                if(s->data_partitioning){
                    backup_s.pb2= s->pb2;
                    backup_s.tex_pb= s->tex_pb;
                }
2646

2647
                if(mb_type&MB_TYPE_INTER){
2648
                    s->mv_dir = MV_DIR_FORWARD;
M
Michael Niedermayer 已提交
2649
                    s->mv_type = MV_TYPE_16X16;
2650
                    s->mb_intra= 0;
2651 2652
                    s->mv[0][0][0] = s->p_mv_table[xy][0];
                    s->mv[0][0][1] = s->p_mv_table[xy][1];
2653 2654
                    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 已提交
2655
                }
2656
                if(mb_type&MB_TYPE_INTER4V){                 
2657
                    s->mv_dir = MV_DIR_FORWARD;
M
Michael Niedermayer 已提交
2658 2659 2660 2661 2662 2663
                    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];
                    }
2664 2665
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_INTER4V, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
2666 2667 2668 2669 2670 2671 2672
                }
                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];
2673 2674
                    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]);
2675 2676 2677 2678 2679 2680 2681
                }
                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];
2682 2683
                    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]);
2684 2685 2686 2687 2688 2689 2690 2691 2692
                }
                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];
2693 2694
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_BIDIR, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
2695 2696 2697 2698 2699 2700 2701 2702 2703
                }
                if(mb_type&MB_TYPE_DIRECT){
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
                    s->mv_type = MV_TYPE_16X16; //FIXME
                    s->mb_intra= 0;
                    s->mv[0][0][0] = s->b_direct_forw_mv_table[xy][0];
                    s->mv[0][0][1] = s->b_direct_forw_mv_table[xy][1];
                    s->mv[1][0][0] = s->b_direct_back_mv_table[xy][0];
                    s->mv[1][0][1] = s->b_direct_back_mv_table[xy][1];
2704 2705
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_DIRECT, pb, pb2, tex_pb, 
                                 &dmin, &next_block, s->b_direct_mv_table[xy][0], s->b_direct_mv_table[xy][1]);
M
Michael Niedermayer 已提交
2706
                }
2707
                if(mb_type&MB_TYPE_INTRA){
2708
                    s->mv_dir = MV_DIR_FORWARD;
M
Michael Niedermayer 已提交
2709
                    s->mv_type = MV_TYPE_16X16;
2710 2711 2712
                    s->mb_intra= 1;
                    s->mv[0][0][0] = 0;
                    s->mv[0][0][1] = 0;
2713 2714
                    encode_mb_hq(s, &backup_s, &best_s, MB_TYPE_INTRA, pb, pb2, tex_pb, 
                                 &dmin, &next_block, 0, 0);
2715 2716 2717
                    /* 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 已提交
2718
                }
2719
                copy_context_after_encode(s, &best_s, -1);
2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736
                
                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;
                }
2737
                s->last_bits= get_bit_count(&s->pb);
F
Fabrice Bellard 已提交
2738
            } else {
2739 2740
                int motion_x, motion_y;
                s->mv_type=MV_TYPE_16X16;
2741
                // only one MB-Type possible
2742 2743
                switch(mb_type){
                case MB_TYPE_INTRA:
2744
                    s->mv_dir = MV_DIR_FORWARD;
2745
                    s->mb_intra= 1;
2746 2747
                    motion_x= s->mv[0][0][0] = 0;
                    motion_y= s->mv[0][0][1] = 0;
2748 2749
                    break;
                case MB_TYPE_INTER:
2750 2751 2752 2753
                    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];
2754
                    break;
2755 2756 2757 2758 2759 2760 2761 2762 2763 2764
                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;
2765
                case MB_TYPE_DIRECT:
2766 2767
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
                    s->mb_intra= 0;
2768 2769 2770 2771 2772 2773 2774 2775
                    motion_x=s->b_direct_mv_table[xy][0];
                    motion_y=s->b_direct_mv_table[xy][1];
                    s->mv[0][0][0] = s->b_direct_forw_mv_table[xy][0];
                    s->mv[0][0][1] = s->b_direct_forw_mv_table[xy][1];
                    s->mv[1][0][0] = s->b_direct_back_mv_table[xy][0];
                    s->mv[1][0][1] = s->b_direct_back_mv_table[xy][1];
                    break;
                case MB_TYPE_BIDIR:
2776
                    s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
2777
                    s->mb_intra= 0;
2778 2779 2780 2781 2782 2783
                    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];
2784 2785
                    break;
                case MB_TYPE_BACKWARD:
2786 2787 2788 2789
                    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];
2790 2791
                    break;
                case MB_TYPE_FORWARD:
2792 2793 2794 2795 2796
                    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);
2797 2798
                    break;
                default:
2799 2800
                    motion_x=motion_y=0; //gcc warning fix
                    printf("illegal MB type\n");
2801
                }
2802
                encode_mb(s, motion_x, motion_y);
F
Fabrice Bellard 已提交
2803
            }
2804 2805 2806 2807 2808
            /* 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 已提交
2809

2810
            MPV_decode_mb(s, s->block);
2811
//printf("MB %d %d bits\n", s->mb_x+s->mb_y*s->mb_width, get_bit_count(&s->pb));
F
Fabrice Bellard 已提交
2812
        }
2813 2814


2815 2816 2817
        /* Obtain average GOB size for RTP */
        if (s->rtp_mode) {
            if (!mb_y)
2818
                s->mb_line_avgsize = pbBufPtr(&s->pb) - s->ptr_last_mb_line;
2819
            else if (!(mb_y % s->gob_index)) {    
2820 2821
                s->mb_line_avgsize = (s->mb_line_avgsize + pbBufPtr(&s->pb) - s->ptr_last_mb_line) >> 1;
                s->ptr_last_mb_line = pbBufPtr(&s->pb);
2822 2823 2824
            }
            //fprintf(stderr, "\nMB line: %d\tSize: %u\tAvg. Size: %u", s->mb_y, 
            //                    (s->pb.buf_ptr - s->ptr_last_mb_line), s->mb_line_avgsize);
2825
            if(s->codec_id!=CODEC_ID_MPEG4) s->first_slice_line = 0; //FIXME clean
2826
        }
F
Fabrice Bellard 已提交
2827
    }
2828
    emms_c();
2829

2830 2831 2832 2833
    if(s->codec_id==CODEC_ID_MPEG4 && s->data_partitioning && s->pict_type!=B_TYPE)
        ff_mpeg4_merge_partitions(s);

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

2836 2837 2838
    if(s->codec_id==CODEC_ID_MPEG4) 
        ff_mpeg4_stuffing(&s->pb);

2839 2840
    //if (s->gob_number)
    //    fprintf(stderr,"\nNumber of GOB: %d", s->gob_number);
2841 2842 2843 2844
    
    /* Send the last GOB if RTP */    
    if (s->rtp_mode) {
        flush_put_bits(&s->pb);
2845
        pdif = pbBufPtr(&s->pb) - s->ptr_lastgob;
2846 2847 2848
        /* Call the RTP callback to send the last GOB */
        if (s->rtp_callback)
            s->rtp_callback(s->ptr_lastgob, pdif, s->gob_number);
2849
        s->ptr_lastgob = pbBufPtr(&s->pb);
2850 2851
        //fprintf(stderr,"\nGOB: %2d size: %d (last)", s->gob_number, pdif);
    }
F
Fabrice Bellard 已提交
2852 2853
}

M
Michael Niedermayer 已提交
2854
static int dct_quantize_c(MpegEncContext *s, 
F
Fabrice Bellard 已提交
2855
                        DCTELEM *block, int n,
2856
                        int qscale, int *overflow)
F
Fabrice Bellard 已提交
2857 2858 2859
{
    int i, j, level, last_non_zero, q;
    const int *qmat;
2860 2861 2862
    int bias;
    int max=0;
    unsigned int threshold1, threshold2;
2863
    
2864
    s->fdct (block);
F
Fabrice Bellard 已提交
2865

2866
#ifndef ARCH_ALPHA              /* Alpha uses unpermuted matrix */
F
Fabrice Bellard 已提交
2867 2868
    /* we need this permutation so that we correct the IDCT
       permutation. will be moved into DCT code */
2869
    block_permute(block, s->idct_permutation); //FIXME remove
2870
#endif
F
Fabrice Bellard 已提交
2871

F
Fabrice Bellard 已提交
2872
    if (s->mb_intra) {
2873 2874 2875 2876 2877 2878 2879 2880 2881 2882
        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 已提交
2883 2884 2885 2886
        /* note: block[0] is assumed to be positive */
        block[0] = (block[0] + (q >> 1)) / q;
        i = 1;
        last_non_zero = 0;
2887
        qmat = s->q_intra_matrix[qscale];
M
Michael Niedermayer 已提交
2888
        bias= s->intra_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
F
Fabrice Bellard 已提交
2889 2890 2891
    } else {
        i = 0;
        last_non_zero = -1;
2892
        qmat = s->q_inter_matrix[qscale];
M
Michael Niedermayer 已提交
2893
        bias= s->inter_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
F
Fabrice Bellard 已提交
2894
    }
M
Michael Niedermayer 已提交
2895 2896
    threshold1= (1<<QMAT_SHIFT) - bias - 1;
    threshold2= (threshold1<<1);
F
Fabrice Bellard 已提交
2897 2898

    for(;i<64;i++) {
2899
        j = s->intra_scantable.permutated[i];
F
Fabrice Bellard 已提交
2900 2901 2902
        level = block[j];
        level = level * qmat[j];

2903 2904 2905 2906
//        if(   bias+level >= (1<<(QMAT_SHIFT - 3))
//           || bias-level >= (1<<(QMAT_SHIFT - 3))){
        if(((unsigned)(level+threshold1))>threshold2){
            if(level>0){
M
Michael Niedermayer 已提交
2907
                level= (bias + level)>>QMAT_SHIFT;
2908 2909
                block[j]= level;
            }else{
M
Michael Niedermayer 已提交
2910
                level= (bias - level)>>QMAT_SHIFT;
2911 2912 2913
                block[j]= -level;
            }
            max |=level;
F
Fabrice Bellard 已提交
2914
            last_non_zero = i;
2915 2916
        }else{
            block[j]=0;
F
Fabrice Bellard 已提交
2917 2918
        }
    }
2919 2920
    *overflow= s->max_qcoeff < max; //overflow might have happend
    
F
Fabrice Bellard 已提交
2921 2922 2923
    return last_non_zero;
}

2924 2925
static void dct_unquantize_mpeg1_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale)
F
Fabrice Bellard 已提交
2926
{
M
Michael Niedermayer 已提交
2927
    int i, level, nCoeffs;
F
Fabrice Bellard 已提交
2928 2929
    const UINT16 *quant_matrix;

2930
    nCoeffs= s->block_last_index[n];
M
Michael Niedermayer 已提交
2931
    
F
Fabrice Bellard 已提交
2932 2933 2934 2935 2936 2937 2938
    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;
2939 2940
        for(i=1;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
M
Michael Niedermayer 已提交
2941
            level = block[j];
F
Fabrice Bellard 已提交
2942 2943 2944
            if (level) {
                if (level < 0) {
                    level = -level;
M
Michael Niedermayer 已提交
2945
                    level = (int)(level * qscale * quant_matrix[j]) >> 3;
F
Fabrice Bellard 已提交
2946 2947 2948
                    level = (level - 1) | 1;
                    level = -level;
                } else {
M
Michael Niedermayer 已提交
2949
                    level = (int)(level * qscale * quant_matrix[j]) >> 3;
F
Fabrice Bellard 已提交
2950 2951 2952 2953 2954 2955
                    level = (level - 1) | 1;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
M
Michael Niedermayer 已提交
2956
                block[j] = level;
F
Fabrice Bellard 已提交
2957 2958 2959 2960
            }
        }
    } else {
        i = 0;
2961
        quant_matrix = s->inter_matrix;
2962 2963
        for(;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
M
Michael Niedermayer 已提交
2964
            level = block[j];
F
Fabrice Bellard 已提交
2965 2966 2967 2968
            if (level) {
                if (level < 0) {
                    level = -level;
                    level = (((level << 1) + 1) * qscale *
M
Michael Niedermayer 已提交
2969
                             ((int) (quant_matrix[j]))) >> 4;
F
Fabrice Bellard 已提交
2970 2971 2972 2973
                    level = (level - 1) | 1;
                    level = -level;
                } else {
                    level = (((level << 1) + 1) * qscale *
M
Michael Niedermayer 已提交
2974
                             ((int) (quant_matrix[j]))) >> 4;
F
Fabrice Bellard 已提交
2975 2976 2977 2978 2979 2980
                    level = (level - 1) | 1;
                }
#ifdef PARANOID
                if (level < -2048 || level > 2047)
                    fprintf(stderr, "unquant error %d %d\n", i, level);
#endif
M
Michael Niedermayer 已提交
2981
                block[j] = level;
F
Fabrice Bellard 已提交
2982 2983 2984 2985
            }
        }
    }
}
2986

2987 2988 2989 2990 2991 2992
static void dct_unquantize_mpeg2_c(MpegEncContext *s, 
                                   DCTELEM *block, int n, int qscale)
{
    int i, level, nCoeffs;
    const UINT16 *quant_matrix;

2993 2994
    if(s->alternate_scan) nCoeffs= 63;
    else nCoeffs= s->block_last_index[n];
2995 2996 2997 2998 2999 3000 3001
    
    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;
3002 3003
        for(i=1;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022
            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;
3023
        quant_matrix = s->inter_matrix;
3024 3025
        for(;i<=nCoeffs;i++) {
            int j= s->intra_scantable.permutated[i];
3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049
            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;
    }
}


3050 3051 3052 3053
static void dct_unquantize_h263_c(MpegEncContext *s, 
                                  DCTELEM *block, int n, int qscale)
{
    int i, level, qmul, qadd;
M
Michael Niedermayer 已提交
3054
    int nCoeffs;
3055
    
3056 3057 3058 3059 3060
    assert(s->block_last_index[n]>=0);
    
    qadd = (qscale - 1) | 1;
    qmul = qscale << 1;
    
3061
    if (s->mb_intra) {
3062 3063 3064 3065 3066
        if (!s->h263_aic) {
            if (n < 4) 
                block[0] = block[0] * s->y_dc_scale;
            else
                block[0] = block[0] * s->c_dc_scale;
3067 3068
        }else
            qadd = 0;
3069
        i = 1;
3070
        nCoeffs= 63; //does not allways use zigzag table 
3071 3072
    } else {
        i = 0;
3073
        nCoeffs= s->intra_scantable.raster_end[ s->block_last_index[n] ];
3074 3075
    }

3076
    for(;i<=nCoeffs;i++) {
3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091
        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 已提交
3092

3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109
static void remove_ac(MpegEncContext *s, uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr, int mb_x, int mb_y)
{
    int dc, dcb, dcr, y, i;
    for(i=0; i<4; i++){
        dc= s->dc_val[0][mb_x*2+1 + (i&1) + (mb_y*2+1 + (i>>1))*(s->mb_width*2+2)];
        for(y=0; y<8; y++){
            int x;
            for(x=0; x<8; x++){
                dest_y[x + (i&1)*8 + (y + (i>>1)*8)*s->linesize]= dc/8;
            }
        }
    }
    dcb = s->dc_val[1][mb_x+1 + (mb_y+1)*(s->mb_width+2)];
    dcr= s->dc_val[2][mb_x+1 + (mb_y+1)*(s->mb_width+2)];
    for(y=0; y<8; y++){
        int x;
        for(x=0; x<8; x++){
M
Michael Niedermayer 已提交
3110 3111
            dest_cb[x + y*(s->uvlinesize)]= dcb/8;
            dest_cr[x + y*(s->uvlinesize)]= dcr/8;
3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127
        }
    }
}

/**
 * will conceal past errors, and allso drop b frames if needed
 *
 */
void ff_conceal_past_errors(MpegEncContext *s, int unknown_pos)
{
    int mb_x= s->mb_x;
    int mb_y= s->mb_y;
    int mb_dist=0;
    int i, intra_count=0, inter_count=0;
    int intra_conceal= s->msmpeg4_version ? 50 : 50; //FIXME finetune
    int inter_conceal= s->msmpeg4_version ? 50 : 50;
M
Michael Niedermayer 已提交
3128

3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165
    // for last block
    if(mb_x>=s->mb_width)  mb_x= s->mb_width -1;
    if(mb_y>=s->mb_height) mb_y= s->mb_height-1;

    if(s->decoding_error==0 && unknown_pos){
        if(s->data_partitioning && s->pict_type!=B_TYPE)
                s->decoding_error= DECODING_AC_LOST;
        else
                s->decoding_error= DECODING_DESYNC;
    }

    if(s->decoding_error==DECODING_DESYNC && s->pict_type!=B_TYPE) s->next_p_frame_damaged=1;

    for(i=mb_x + mb_y*s->mb_width; i>=0; i--){
        if(s->mbintra_table[i]) intra_count++;
        else                    inter_count++;
    }
    
    if(s->decoding_error==DECODING_AC_LOST){
        intra_conceal*=2;
        inter_conceal*=2;
    }else if(s->decoding_error==DECODING_ACDC_LOST){
        intra_conceal*=2;
        inter_conceal*=2;
    }

    if(unknown_pos && (intra_count<inter_count)){
        intra_conceal= inter_conceal= s->mb_num; 
//        printf("%d %d\n",intra_count, inter_count);
    }

    fprintf(stderr, "concealing errors\n");

    /* for all MBs from the current one back until the last resync marker */
    for(; mb_y>=0 && mb_y>=s->resync_mb_y; mb_y--){
        for(; mb_x>=0; mb_x--){
            uint8_t *dest_y  = s->current_picture[0] + (mb_y * 16*  s->linesize      ) + mb_x * 16;
M
Michael Niedermayer 已提交
3166 3167
            uint8_t *dest_cb = s->current_picture[1] + (mb_y * 8 * (s->uvlinesize)) + mb_x * 8;
            uint8_t *dest_cr = s->current_picture[2] + (mb_y * 8 * (s->uvlinesize)) + mb_x * 8;
3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205
            int mb_x_backup= s->mb_x; //FIXME pass xy to mpeg_motion
            int mb_y_backup= s->mb_y;
            s->mb_x=mb_x;
            s->mb_y=mb_y;
            if(s->mbintra_table[mb_y*s->mb_width + mb_x] && mb_dist<intra_conceal){
                if(s->decoding_error==DECODING_AC_LOST){
                    remove_ac(s, dest_y, dest_cb, dest_cr, mb_x, mb_y);
//                    printf("remove ac to %d %d\n", mb_x, mb_y);
                }else{
                    mpeg_motion(s, dest_y, dest_cb, dest_cr, 0, 
                                s->last_picture, 0, 0, put_pixels_tab,
                                0/*mx*/, 0/*my*/, 16);
                }
            }
            else if(!s->mbintra_table[mb_y*s->mb_width + mb_x] && mb_dist<inter_conceal){
                int mx=0;
                int my=0;

                if(s->decoding_error!=DECODING_DESYNC){
                    int xy= mb_x*2+1 + (mb_y*2+1)*(s->mb_width*2+2);
                    mx= s->motion_val[ xy ][0];
                    my= s->motion_val[ xy ][1];
                }

                mpeg_motion(s, dest_y, dest_cb, dest_cr, 0, 
                            s->last_picture, 0, 0, put_pixels_tab,
                            mx, my, 16);
            }
            s->mb_x= mb_x_backup;
            s->mb_y= mb_y_backup;

            if(mb_x== s->resync_mb_x && mb_y== s->resync_mb_y) return;
            if(!s->mbskip_table[mb_x + mb_y*s->mb_width]) mb_dist++;
        }
        mb_x=s->mb_width-1;
    }
}

F
Fabrice Bellard 已提交
3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255
AVCodec mpeg1video_encoder = {
    "mpeg1video",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MPEG1VIDEO,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

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,
};

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

F
Fabrice Bellard 已提交
3256 3257
AVCodec mpeg4_encoder = {
    "mpeg4",
F
Fabrice Bellard 已提交
3258
    CODEC_TYPE_VIDEO,
F
Fabrice Bellard 已提交
3259
    CODEC_ID_MPEG4,
F
Fabrice Bellard 已提交
3260 3261 3262 3263 3264 3265
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286
AVCodec msmpeg4v1_encoder = {
    "msmpeg4v1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MSMPEG4V1,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

AVCodec msmpeg4v2_encoder = {
    "msmpeg4v2",
    CODEC_TYPE_VIDEO,
    CODEC_ID_MSMPEG4V2,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

AVCodec msmpeg4v3_encoder = {
F
Fabrice Bellard 已提交
3287 3288
    "msmpeg4",
    CODEC_TYPE_VIDEO,
3289
    CODEC_ID_MSMPEG4V3,
F
Fabrice Bellard 已提交
3290 3291 3292 3293 3294
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};
M
Michael Niedermayer 已提交
3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314

AVCodec wmv1_encoder = {
    "wmv1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_WMV1,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};

AVCodec wmv2_encoder = {
    "wmv2",
    CODEC_TYPE_VIDEO,
    CODEC_ID_WMV2,
    sizeof(MpegEncContext),
    MPV_encode_init,
    MPV_encode_picture,
    MPV_encode_end,
};