mpegvideo.h 39.1 KB
Newer Older
Fabrice Bellard's avatar
Fabrice Bellard committed
1 2
/*
 * Generic DCT based hybrid video encoder
3
 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
4
 * Copyright (c) 2002-2004 Michael Niedermayer
Fabrice Bellard's avatar
Fabrice Bellard committed
5
 *
6 7 8 9
 * 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.
Fabrice Bellard's avatar
Fabrice Bellard committed
10
 *
11
 * This library is distributed in the hope that it will be useful,
Fabrice Bellard's avatar
Fabrice Bellard committed
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
Fabrice Bellard's avatar
Fabrice Bellard committed
15
 *
16 17 18
 * 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
Fabrice Bellard's avatar
Fabrice Bellard committed
19 20
 */

Michael Niedermayer's avatar
Michael Niedermayer committed
21 22 23 24 25
/**
 * @file mpegvideo.h
 * mpegvideo header.
 */
 
26 27 28
#ifndef AVCODEC_MPEGVIDEO_H
#define AVCODEC_MPEGVIDEO_H

29 30
#include "dsputil.h"

Michael Niedermayer's avatar
Michael Niedermayer committed
31
#define FRAME_SKIPED 100 ///< return value for header parsers if frame is not coded
Michael Niedermayer's avatar
Michael Niedermayer committed
32

Fabrice Bellard's avatar
Fabrice Bellard committed
33 34 35
enum OutputFormat {
    FMT_MPEG1,
    FMT_H263,
36
    FMT_MJPEG, 
37
    FMT_H264,
Fabrice Bellard's avatar
Fabrice Bellard committed
38 39
};

Michael Niedermayer's avatar
Michael Niedermayer committed
40 41
#define EDGE_WIDTH 16

Fabrice Bellard's avatar
Fabrice Bellard committed
42 43
#define MPEG_BUF_SIZE (16 * 1024)

Michael Niedermayer's avatar
Michael Niedermayer committed
44 45
#define QMAT_SHIFT_MMX 16
#define QMAT_SHIFT 22
46

47 48
#define MAX_FCODE 7
#define MAX_MV 2048
Michael Niedermayer's avatar
Michael Niedermayer committed
49

50 51
#define MAX_THREADS 8

52
#define MAX_PICTURE_COUNT 15
53

54 55 56 57
#define ME_MAP_SIZE 64
#define ME_MAP_SHIFT 3
#define ME_MAP_MV_BITS 11

Michael Niedermayer's avatar
Michael Niedermayer committed
58 59 60 61
/* run length table */
#define MAX_RUN    64
#define MAX_LEVEL  64

Michael Niedermayer's avatar
Michael Niedermayer committed
62 63 64 65
#define I_TYPE FF_I_TYPE  ///< Intra
#define P_TYPE FF_P_TYPE  ///< Predicted
#define B_TYPE FF_B_TYPE  ///< Bi-dir predicted
#define S_TYPE FF_S_TYPE  ///< S(GMC)-VOP MPEG4
66 67
#define SI_TYPE FF_SI_TYPE  ///< Switching Intra
#define SP_TYPE FF_SP_TYPE  ///< Switching Predicted
68

69 70 71 72 73 74
typedef struct Predictor{
    double coeff;
    double count;
    double decay;
} Predictor;

75 76
typedef struct RateControlEntry{
    int pict_type;
77
    float qscale;
78 79 80 81
    int mv_bits;
    int i_tex_bits;
    int p_tex_bits;
    int misc_bits;
82
    uint64_t expected_bits;
83 84
    int new_pict_type;
    float new_qscale;
85 86 87 88 89
    int mc_mb_var_sum;
    int mb_var_sum;
    int i_count;
    int f_code;
    int b_code;
90 91
}RateControlEntry;

Michael Niedermayer's avatar
Michael Niedermayer committed
92 93 94
/**
 * rate control context.
 */
95 96
typedef struct RateControlContext{
    FILE *stats_file;
Michael Niedermayer's avatar
Michael Niedermayer committed
97
    int num_entries;              ///< number of RateControlEntries 
98
    RateControlEntry *entry;
Michael Niedermayer's avatar
Michael Niedermayer committed
99
    double buffer_index;          ///< amount of bits in the video/audio buffer 
100
    Predictor pred[5];
Michael Niedermayer's avatar
Michael Niedermayer committed
101 102 103 104
    double short_term_qsum;       ///< sum of recent qscales 
    double short_term_qcount;     ///< count of recent qscales 
    double pass1_rc_eq_output_sum;///< sum of the output of the rc equation, this is used for normalization  
    double pass1_wanted_bits;     ///< bits which should have been outputed by the pass1 code (including complexity init) 
105
    double last_qscale;
Michael Niedermayer's avatar
Michael Niedermayer committed
106
    double last_qscale_for[5];    ///< last qscale for a specific pict type, used for max_diff & ipb factor stuff 
107 108
    int last_mc_mb_var_sum;
    int last_mb_var_sum;
109 110 111 112
    uint64_t i_cplx_sum[5];
    uint64_t p_cplx_sum[5];
    uint64_t mv_bits_sum[5];
    uint64_t qscale_sum[5];
113
    int frame_count[5];
114
    int last_non_b_pict_type;
115 116
}RateControlContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
117 118 119
/**
 * Scantable.
 */
120
typedef struct ScanTable{
121 122 123
    const uint8_t *scantable;
    uint8_t permutated[64];
    uint8_t raster_end[64];
124
#ifdef ARCH_POWERPC
Michael Niedermayer's avatar
Michael Niedermayer committed
125
		/** Used by dct_quantise_alitvec to find last-non-zero */
126
    uint8_t __align8 inverse[64];
127
#endif
128 129
} ScanTable;

Michael Niedermayer's avatar
Michael Niedermayer committed
130 131 132
/**
 * Picture.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
133
typedef struct Picture{
134
    FF_COMMON_FRAME
Michael Niedermayer's avatar
Michael Niedermayer committed
135

136 137 138 139
    /**
     * halfpel luma planes.
     */
    uint8_t *interpolated[3];
140
    int16_t (*motion_val_base[2])[2];
141
    uint32_t *mb_type_base;
142
#define MB_TYPE_INTRA MB_TYPE_INTRA4x4 //default mb_type if theres just one type
143 144
#define IS_INTRA4x4(a)   ((a)&MB_TYPE_INTRA4x4)
#define IS_INTRA16x16(a) ((a)&MB_TYPE_INTRA16x16)
145 146
#define IS_PCM(a)        ((a)&MB_TYPE_INTRA_PCM)
#define IS_INTRA(a)      ((a)&7)
147
#define IS_INTER(a)      ((a)&(MB_TYPE_16x16|MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8))
148
#define IS_SKIP(a)       ((a)&MB_TYPE_SKIP)
149 150 151
#define IS_INTRA_PCM(a)  ((a)&MB_TYPE_INTRA_PCM)
#define IS_INTERLACED(a) ((a)&MB_TYPE_INTERLACED)
#define IS_DIRECT(a)     ((a)&MB_TYPE_DIRECT2)
152
#define IS_GMC(a)        ((a)&MB_TYPE_GMC)
153 154 155 156 157 158 159 160
#define IS_16X16(a)      ((a)&MB_TYPE_16x16)
#define IS_16X8(a)       ((a)&MB_TYPE_16x8)
#define IS_8X16(a)       ((a)&MB_TYPE_8x16)
#define IS_8X8(a)        ((a)&MB_TYPE_8x8)
#define IS_SUB_8X8(a)    ((a)&MB_TYPE_16x16) //note reused
#define IS_SUB_8X4(a)    ((a)&MB_TYPE_16x8)  //note reused
#define IS_SUB_4X8(a)    ((a)&MB_TYPE_8x16)  //note reused
#define IS_SUB_4X4(a)    ((a)&MB_TYPE_8x8)   //note reused
161 162
#define IS_ACPRED(a)     ((a)&MB_TYPE_ACPRED)
#define IS_QUANT(a)      ((a)&MB_TYPE_QUANT)
163 164
#define IS_DIR(a, part, list) ((a) & (MB_TYPE_P0L0<<((part)+2*(list))))
#define USES_LIST(a, list) ((a) & ((MB_TYPE_P0L0|MB_TYPE_P1L0)<<(2*(list)))) ///< does this mb use listX, note doesnt work if subMBs
165
#define HAS_CBP(a)        ((a)&MB_TYPE_CBP)
166 167 168 169 170 171 172

    int field_poc[2];           ///< h264 top/bottom POC
    int poc;                    ///< h264 frame POC
    int frame_num;              ///< h264 frame_num
    int pic_id;                 ///< h264 pic_num or long_term_pic_idx
    int long_ref;               ///< 1->long term reference 0->short term reference

Michael Niedermayer's avatar
Michael Niedermayer committed
173 174 175 176 177
    int mb_var_sum;             ///< sum of MB variance for current frame 
    int mc_mb_var_sum;          ///< motion compensated MB variance for current frame 
    uint16_t *mb_var;           ///< Table for MB variances 
    uint16_t *mc_mb_var;        ///< Table for motion compensated MB variances 
    uint8_t *mb_mean;           ///< Table for MB luminance 
Michael Niedermayer's avatar
Michael Niedermayer committed
178
    int32_t *mb_cmp_score;	///< Table for MB cmp scores, for mb decission FIXME remove
179
    int b_frame_score;          /* */
Michael Niedermayer's avatar
Michael Niedermayer committed
180 181
} Picture;

182
typedef struct ParseContext{
183
    uint8_t *buffer;
184 185 186
    int index;
    int last_index;
    int buffer_size;
187
    uint32_t state;             ///< contains the last few bytes in MSB order
188
    int frame_start_found;
189 190
    int overread;               ///< the number of bytes which where irreversibly read from the next frame
    int overread_index;         ///< the index into ParseContext.buffer of the overreaded bytes
191 192
} ParseContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
193 194
struct MpegEncContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
195 196 197
/**
 * Motion estimation context.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
198
typedef struct MotionEstContext{
Michael Niedermayer's avatar
Michael Niedermayer committed
199 200
    int skip;                          ///< set if ME is skiped for the current MB 
    int co_located_mv[4][2];           ///< mv from last p frame for direct mode ME 
Michael Niedermayer's avatar
Michael Niedermayer committed
201
    int direct_basis_mv[4][2];
Michael Niedermayer's avatar
Michael Niedermayer committed
202
    uint8_t *scratchpad;               ///< data area for the me algo, so that the ME doesnt need to malloc/free 
203 204 205 206
    uint8_t *best_mb;
    uint8_t *temp_mb[2];
    uint8_t *temp;
    int best_bits;
Michael Niedermayer's avatar
Michael Niedermayer committed
207 208
    uint32_t *map;                     ///< map to avoid duplicate evaluations 
    uint32_t *score_map;               ///< map to store the scores 
Michael Niedermayer's avatar
Michael Niedermayer committed
209
    int map_generation;  
Michael Niedermayer's avatar
Michael Niedermayer committed
210
    int pre_penalty_factor;
Michael Niedermayer's avatar
Michael Niedermayer committed
211 212
    int penalty_factor;
    int sub_penalty_factor;
213
    int mb_penalty_factor;
214 215 216
    int flags;
    int sub_flags;
    int mb_flags;
Michael Niedermayer's avatar
Michael Niedermayer committed
217
    int pre_pass;                      ///< = 1 for the pre pass 
Michael Niedermayer's avatar
Michael Niedermayer committed
218
    int dia_size;
219 220 221 222
    int xmin;
    int xmax;
    int ymin;
    int ymax;
223 224 225 226 227 228 229 230 231 232 233
    int pred_x;
    int pred_y;
    uint8_t *src[4][4];
    uint8_t *ref[4][4];
    int stride;
    int uvstride;
/*    cmp, chroma_cmp;*/
    op_pixels_func (*hpel_put)[4];
    op_pixels_func (*hpel_avg)[4];
    qpel_mc_func (*qpel_put)[16];
    qpel_mc_func (*qpel_avg)[16];
234
    uint8_t (*mv_penalty)[MAX_MV*2+1];  ///< amount of bits needed to encode a MV 
235
    uint8_t *current_mv_penalty;
Michael Niedermayer's avatar
Michael Niedermayer committed
236 237
    int (*sub_motion_search)(struct MpegEncContext * s,
				  int *mx_ptr, int *my_ptr, int dmin,
238 239
                                  int src_index, int ref_index,
                                  int size, int h);
Michael Niedermayer's avatar
Michael Niedermayer committed
240 241
}MotionEstContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
242 243 244
/**
 * MpegEncContext.
 */
Fabrice Bellard's avatar
Fabrice Bellard committed
245
typedef struct MpegEncContext {
246
    struct AVCodecContext *avctx;
Fabrice Bellard's avatar
Fabrice Bellard committed
247
    /* the following parameters must be initialized before encoding */
Michael Niedermayer's avatar
Michael Niedermayer committed
248
    int width, height;///< picture size. must be a multiple of 16 
Fabrice Bellard's avatar
Fabrice Bellard committed
249
    int gop_size;
Michael Niedermayer's avatar
Michael Niedermayer committed
250 251 252 253
    int intra_only;   ///< if true, only intra pictures are generated 
    int bit_rate;     ///< wanted bit rate 
    enum OutputFormat out_format; ///< output format 
    int h263_pred;    ///< use mpeg4/h263 ac/dc predictions 
254 255

/* the following codec id fields are deprecated in favor of codec_id */
Michael Niedermayer's avatar
Michael Niedermayer committed
256 257
    int h263_plus;    ///< h263 plus headers 
    int h263_msmpeg4; ///< generate MSMPEG4 compatible stream (deprecated, use msmpeg4_version instead)
258
    int h263_flv;     ///< use flv h263 header 
259 260
    
    int codec_id;     /* see CODEC_ID_xxx */
Michael Niedermayer's avatar
Michael Niedermayer committed
261 262 263
    int fixed_qscale; ///< fixed qscale if non zero 
    int encoding;     ///< true if we are encoding (vs decoding) 
    int flags;        ///< AVCodecContext.flags (HQ, MV4, ...) 
264
    int flags2;       ///< AVCodecContext.flags2
Michael Niedermayer's avatar
Michael Niedermayer committed
265
    int max_b_frames; ///< max number of b-frames for encoding 
266 267
    int luma_elim_threshold;
    int chroma_elim_threshold;
Michael Niedermayer's avatar
Michael Niedermayer committed
268 269
    int strict_std_compliance; ///< strictly follow the std (MPEG4, ...) 
    int workaround_bugs;       ///< workaround bugs in encoders which cannot be detected automatically 
Fabrice Bellard's avatar
Fabrice Bellard committed
270 271
    /* the following fields are managed internally by the encoder */

Michael Niedermayer's avatar
Michael Niedermayer committed
272
    /** bit output */
Fabrice Bellard's avatar
Fabrice Bellard committed
273 274 275 276
    PutBitContext pb;

    /* sequence parameters */
    int context_initialized;
277 278 279
    int input_picture_number;  ///< used to set pic->display_picture_number, shouldnt be used for/by anything else
    int coded_picture_number;  ///< used to set pic->coded_picture_number, shouldnt be used for/by anything else
    int picture_number;       //FIXME remove, unclear definition
Michael Niedermayer's avatar
Michael Niedermayer committed
280 281
    int picture_in_gop_number; ///< 0-> first pic in gop, ... 
    int b_frames_since_non_b;  ///< used for encoding, relative to not yet reordered input 
282
    int64_t user_specified_pts;///< last non zero pts from AVFrame which was passed into avcodec_encode_video()
Michael Niedermayer's avatar
Michael Niedermayer committed
283
    int mb_width, mb_height;   ///< number of MBs horizontally & vertically 
284
    int mb_stride;             ///< mb_width+1 used for some arrays to allow simple addressng of left & top MBs withoutt sig11
285 286
    int b8_stride;             ///< 2*mb_width+1 used for some 8x8 block arrays to allow simple addressng
    int b4_stride;             ///< 4*mb_width+1 used for some 4x4 block arrays to allow simple addressng
Michael Niedermayer's avatar
Michael Niedermayer committed
287 288 289 290
    int h_edge_pos, v_edge_pos;///< horizontal / vertical position of the right/bottom edge (pixel replicateion)
    int mb_num;                ///< number of MBs of a picture 
    int linesize;              ///< line size, in bytes, may be different from width 
    int uvlinesize;            ///< line size, for chroma in bytes, may be different from width 
291
    Picture *picture;          ///< main picture buffer 
292 293
    Picture **input_picture;   ///< next pictures on display order for encoding
    Picture **reordered_input_picture; ///< pointer to the next pictures in codedorder for encoding
294
    
295 296 297 298
    int start_mb_y;            ///< start mb_y of this thread (so current thread should process start_mb_y <= row < end_mb_y)
    int end_mb_y;              ///< end   mb_y of this thread (so current thread should process start_mb_y <= row < end_mb_y)
    struct MpegEncContext *thread_context[MAX_THREADS];
    
299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
    /** 
     * copy of the previous picture structure.
     * note, linesize & data, might not match the previous picture (for field pictures)
     */
    Picture last_picture;       
    
    /** 
     * copy of the next picture structure.
     * note, linesize & data, might not match the next picture (for field pictures)
     */
    Picture next_picture;
    
    /** 
     * copy of the source picture structure for encoding.
     * note, linesize & data, might not match the source picture (for field pictures)
     */
    Picture new_picture;
    
    /** 
     * copy of the current picture structure.
     * note, linesize & data, might not match the current picture (for field pictures)
     */
Michael Niedermayer's avatar
Michael Niedermayer committed
321
    Picture current_picture;    ///< buffer to store the decompressed current picture 
322 323 324 325
    
    Picture *last_picture_ptr;     ///< pointer to the previous picture.
    Picture *next_picture_ptr;     ///< pointer to the next picture (for bidir pred) 
    Picture *current_picture_ptr;  ///< pointer to the current picture
326
    uint8_t *visualization_buffer[3]; //< temporary buffer vor MV visualization
Michael Niedermayer's avatar
Michael Niedermayer committed
327
    int last_dc[3];                ///< last DC values for MPEG1 
328
    int16_t *dc_val_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
329
    int16_t *dc_val[3];            ///< used for mpeg4 DC prediction, all 3 arrays must be continuous 
330
    int16_t dc_cache[4*5];
Fabrice Bellard's avatar
Fabrice Bellard committed
331
    int y_dc_scale, c_dc_scale;
Michael Niedermayer's avatar
Michael Niedermayer committed
332 333
    uint8_t *y_dc_scale_table;     ///< qscale -> y_dc_scale table 
    uint8_t *c_dc_scale_table;     ///< qscale -> c_dc_scale table 
Michael Niedermayer's avatar
Michael Niedermayer committed
334
    const uint8_t *chroma_qscale_table;  ///< qscale -> chroma_qscale (h263)
335
    uint8_t *coded_block_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
336
    uint8_t *coded_block;          ///< used for coded block pattern prediction (msmpeg4v3, wmv1)
337
    int16_t (*ac_val_base)[16];
Michael Niedermayer's avatar
Michael Niedermayer committed
338
    int16_t (*ac_val[3])[16];      ///< used for for mpeg4 AC prediction, all 3 arrays must be continuous 
Fabrice Bellard's avatar
Fabrice Bellard committed
339
    int ac_pred;
Michael Niedermayer's avatar
Michael Niedermayer committed
340
    uint8_t *prev_pict_types;     ///< previous picture types in bitstream order, used for mb skip 
341
#define PREV_PICT_TYPES_BUFFER_SIZE 256
Michael Niedermayer's avatar
Michael Niedermayer committed
342 343
    int mb_skiped;                ///< MUST BE SET only during DECODING 
    uint8_t *mbskip_table;        /**< used to avoid copy if macroblock skipped (for black regions for example) 
344
                                   and used for b-frame encoding & decoding (contains skip table of next P Frame) */
Michael Niedermayer's avatar
Michael Niedermayer committed
345 346 347
    uint8_t *mbintra_table;       ///< used to avoid setting {ac, dc, cbp}-pred stuff to zero on inter MB decoding 
    uint8_t *cbp_table;           ///< used to store cbp, ac_pred for partitioned decoding 
    uint8_t *pred_dir_table;      ///< used to store pred_dir for partitioned decoding 
348
    uint8_t *allocated_edge_emu_buffer;
349 350 351 352
    uint8_t *edge_emu_buffer;     ///< points into the middle of allocated_edge_emu_buffer
    uint8_t *rd_scratchpad;       ///< scartchpad for rate distortion mb decission
    uint8_t *obmc_scratchpad;
    uint8_t *b_scratchpad;        ///< scratchpad used for writing into write only buffers
Fabrice Bellard's avatar
Fabrice Bellard committed
353

Michael Niedermayer's avatar
Michael Niedermayer committed
354
    int qscale;                 ///< QP 
Michael Niedermayer's avatar
Michael Niedermayer committed
355
    int chroma_qscale;          ///< chroma QP 
356 357 358
    int lambda;                 ///< lagrange multipler used in rate distortion
    int lambda2;                ///< (lambda*lambda) >> FF_LAMBDA_SHIFT 
    int *lambda_table;
Michael Niedermayer's avatar
Michael Niedermayer committed
359 360 361
    int adaptive_quant;         ///< use adaptive quantization 
    int dquant;                 ///< qscale difference to prev qscale  
    int pict_type;              ///< I_TYPE, P_TYPE, B_TYPE, ... 
362
    int last_pict_type; //FIXME removes
Michael Niedermayer's avatar
Michael Niedermayer committed
363
    int last_non_b_pict_type;   ///< used for mpeg4 gmc b-frames & ratecontrol 
364
    int dropable;
Fabrice Bellard's avatar
Fabrice Bellard committed
365
    int frame_rate_index;
366 367 368
    int frame_rate_ext_n;       ///< MPEG-2 specific framerate modificators (numerator)
    int frame_rate_ext_d;       ///< MPEG-2 specific framerate modificators (denominator)

Fabrice Bellard's avatar
Fabrice Bellard committed
369
    /* motion compensation */
Michael Niedermayer's avatar
Michael Niedermayer committed
370 371
    int unrestricted_mv;        ///< mv can point outside of the coded picture 
    int h263_long_vectors;      ///< use horrible h263v1 long vector mode 
372
    int decode;                 ///< if 0 then decoding will be skiped (for encoding b frames for example)
Michael Niedermayer's avatar
Michael Niedermayer committed
373 374 375 376

    DSPContext dsp;             ///< pointers for accelerated dsp fucntions 
    int f_code;                 ///< forward MV resolution 
    int b_code;                 ///< backward MV resolution for B Frames (mpeg4) 
377 378 379 380 381 382
    int16_t (*p_mv_table_base)[2];
    int16_t (*b_forw_mv_table_base)[2];
    int16_t (*b_back_mv_table_base)[2];
    int16_t (*b_bidir_forw_mv_table_base)[2]; 
    int16_t (*b_bidir_back_mv_table_base)[2]; 
    int16_t (*b_direct_mv_table_base)[2];
383 384
    int16_t (*p_field_mv_table_base[2][2])[2];
    int16_t (*b_field_mv_table_base[2][2][2])[2];
Michael Niedermayer's avatar
Michael Niedermayer committed
385 386 387 388 389 390
    int16_t (*p_mv_table)[2];            ///< MV table (1MV per MB) p-frame encoding 
    int16_t (*b_forw_mv_table)[2];       ///< MV table (1MV per MB) forward mode b-frame encoding 
    int16_t (*b_back_mv_table)[2];       ///< MV table (1MV per MB) backward mode b-frame encoding 
    int16_t (*b_bidir_forw_mv_table)[2]; ///< MV table (1MV per MB) bidir mode b-frame encoding 
    int16_t (*b_bidir_back_mv_table)[2]; ///< MV table (1MV per MB) bidir mode b-frame encoding 
    int16_t (*b_direct_mv_table)[2];     ///< MV table (1MV per MB) direct mode b-frame encoding 
391 392 393 394
    int16_t (*p_field_mv_table[2][2])[2];   ///< MV table (2MV per MB) interlaced p-frame encoding
    int16_t (*b_field_mv_table[2][2][2])[2];///< MV table (4MV per MB) interlaced b-frame encoding
    uint8_t (*p_field_select_table[2]);
    uint8_t (*b_field_select_table[2][2]);
Michael Niedermayer's avatar
Michael Niedermayer committed
395
    int me_method;                       ///< ME algorithm 
396
    int scene_change_score;
Fabrice Bellard's avatar
Fabrice Bellard committed
397
    int mv_dir;
398 399
#define MV_DIR_BACKWARD  1
#define MV_DIR_FORWARD   2
Michael Niedermayer's avatar
Michael Niedermayer committed
400
#define MV_DIRECT        4 ///< bidirectional mode where the difference equals the MV of the last P/S/I-Frame (mpeg4)
Fabrice Bellard's avatar
Fabrice Bellard committed
401
    int mv_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
402 403 404 405 406 407
#define MV_TYPE_16X16       0   ///< 1 vector for the whole mb 
#define MV_TYPE_8X8         1   ///< 4 vectors (h263, mpeg4 4MV) 
#define MV_TYPE_16X8        2   ///< 2 vectors, one per 16x8 block  
#define MV_TYPE_FIELD       3   ///< 2 vectors, one per field  
#define MV_TYPE_DMV         4   ///< 2 vectors, special mpeg2 Dual Prime Vectors 
    /**motion vectors for a macroblock 
Fabrice Bellard's avatar
Fabrice Bellard committed
408 409 410 411 412 413
       first coordinate : 0 = forward 1 = backward
       second "         : depend on type
       third  "         : 0 = x, 1 = y
    */
    int mv[2][4][2];
    int field_select[2][2];
Michael Niedermayer's avatar
Michael Niedermayer committed
414 415
    int last_mv[2][2][2];             ///< last MV, used for MV prediction in MPEG1 & B-frame MPEG4 
    uint8_t *fcode_tab;               ///< smallest fcode needed for each MV 
Michael Niedermayer's avatar
Michael Niedermayer committed
416 417
    
    MotionEstContext me;
Fabrice Bellard's avatar
Fabrice Bellard committed
418

Michael Niedermayer's avatar
Michael Niedermayer committed
419
    int no_rounding;  /**< apply no rounding to motion compensation (MPEG4, msmpeg4, ...) 
420
                        for b-frames rounding mode is allways 0 */
Fabrice Bellard's avatar
Fabrice Bellard committed
421

Michael Niedermayer's avatar
Michael Niedermayer committed
422
    int hurry_up;     /**< when set to 1 during decoding, b frames will be skiped
423
                         when set to 2 idct/dequant will be skipped too */
424
                        
Fabrice Bellard's avatar
Fabrice Bellard committed
425 426
    /* macroblock layer */
    int mb_x, mb_y;
427
    int mb_skip_run;
Fabrice Bellard's avatar
Fabrice Bellard committed
428
    int mb_intra;
429 430 431 432 433
    uint16_t *mb_type;           ///< Table for candidate MB types for encoding
#define CANDIDATE_MB_TYPE_INTRA    0x01
#define CANDIDATE_MB_TYPE_INTER    0x02
#define CANDIDATE_MB_TYPE_INTER4V  0x04
#define CANDIDATE_MB_TYPE_SKIPED   0x08
434
//#define MB_TYPE_GMC      0x10
435

436 437 438 439 440 441 442 443 444
#define CANDIDATE_MB_TYPE_DIRECT   0x10
#define CANDIDATE_MB_TYPE_FORWARD  0x20
#define CANDIDATE_MB_TYPE_BACKWARD 0x40
#define CANDIDATE_MB_TYPE_BIDIR    0x80

#define CANDIDATE_MB_TYPE_INTER_I    0x100
#define CANDIDATE_MB_TYPE_FORWARD_I  0x200
#define CANDIDATE_MB_TYPE_BACKWARD_I 0x400
#define CANDIDATE_MB_TYPE_BIDIR_I    0x800
Michael Niedermayer's avatar
Michael Niedermayer committed
445

Michael Niedermayer's avatar
Michael Niedermayer committed
446
    int block_index[6]; ///< index to current MB in block based arrays with edges
Michael Niedermayer's avatar
Michael Niedermayer committed
447
    int block_wrap[6];
448
    uint8_t *dest[3];
449 450
    
    int *mb_index2xy;        ///< mb_index -> mb_x + mb_y*mb_stride
Michael Niedermayer's avatar
Michael Niedermayer committed
451

Michael Niedermayer's avatar
Michael Niedermayer committed
452
    /** matrix transmitted in the bitstream */
453 454 455 456
    uint16_t intra_matrix[64];
    uint16_t chroma_intra_matrix[64];
    uint16_t inter_matrix[64];
    uint16_t chroma_inter_matrix[64];
457
#define QUANT_BIAS_SHIFT 8
Michael Niedermayer's avatar
Michael Niedermayer committed
458 459 460 461 462
    int intra_quant_bias;    ///< bias for the quantizer 
    int inter_quant_bias;    ///< bias for the quantizer 
    int min_qcoeff;          ///< minimum encodable coefficient 
    int max_qcoeff;          ///< maximum encodable coefficient 
    int ac_esc_length;       ///< num of bits needed to encode the longest esc 
463 464 465 466
    uint8_t *intra_ac_vlc_length;
    uint8_t *intra_ac_vlc_last_length;
    uint8_t *inter_ac_vlc_length;
    uint8_t *inter_ac_vlc_last_length;
467 468
    uint8_t *luma_dc_vlc_length;
    uint8_t *chroma_dc_vlc_length;
469
#define UNI_AC_ENC_INDEX(run,level) ((run)*128 + (level))
470

471 472
    int coded_score[6];

Michael Niedermayer's avatar
Michael Niedermayer committed
473
    /** precomputed matrix (combine qscale and DCT renorm) */
474 475
    int (*q_intra_matrix)[64];
    int (*q_inter_matrix)[64];
476 477 478
    /** identical to the above but for MMX & these are not permutated, second 64 entries are bias*/
    uint16_t (*q_intra_matrix16)[2][64];
    uint16_t (*q_inter_matrix16)[2][64];
479
    int block_last_index[12];  ///< last non zero coefficient in block
480
    /* scantables */
481
    ScanTable __align8 intra_scantable;
482 483
    ScanTable intra_h_scantable;
    ScanTable intra_v_scantable;
Michael Niedermayer's avatar
Michael Niedermayer committed
484
    ScanTable inter_scantable; ///< if inter == intra then intra should be used to reduce tha cache usage
485 486 487 488 489
    
    /* noise reduction */
    int (*dct_error_sum)[64];
    int dct_count[2];
    uint16_t (*dct_offset)[64];
Fabrice Bellard's avatar
Fabrice Bellard committed
490

Michael Niedermayer's avatar
Michael Niedermayer committed
491
    void *opaque;              ///< private data for the user
Fabrice Bellard's avatar
Fabrice Bellard committed
492 493

    /* bit rate control */
494 495
    int64_t wanted_bits;
    int64_t total_bits;
Michael Niedermayer's avatar
Michael Niedermayer committed
496 497
    int frame_bits;                ///< bits used for the current frame 
    RateControlContext rc_context; ///< contains stuff only accessed in ratecontrol.c
498

499 500 501 502 503 504
    /* statistics, used for 2-pass encoding */
    int mv_bits;
    int header_bits;
    int i_tex_bits;
    int p_tex_bits;
    int i_count;
505 506
    int f_count;
    int b_count;
507
    int skip_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
508 509
    int misc_bits; ///< cbp, mb_type
    int last_bits; ///< temp var used for calculating the above vars
510
    
511 512 513 514
    /* temp variables for picture complexity calculation */
    int mc_mb_var_sum_temp;
    int mb_var_sum_temp;

515
    /* error concealment / resync */
516
    int error_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
517 518
    uint8_t *error_status_table;       ///< table of the error status of each MB  
#define VP_START            1          ///< current MB is the first after a resync marker 
519 520 521 522 523 524 525 526
#define AC_ERROR            2
#define DC_ERROR            4
#define MV_ERROR            8
#define AC_END              16
#define DC_END              32
#define MV_END              64
//FIXME some prefix?
    
Michael Niedermayer's avatar
Michael Niedermayer committed
527 528 529 530 531
    int resync_mb_x;                 ///< x position of last resync marker 
    int resync_mb_y;                 ///< y position of last resync marker 
    GetBitContext last_resync_gb;    ///< used to search for the next resync marker 
    int mb_num_left;                 ///< number of MBs left in this video packet (for partitioned Slices only)
    int next_p_frame_damaged;        ///< set if the next p frame is damaged, to avoid showing trashed b frames 
532
    int error_resilience;
533 534
    
    ParseContext parse_context;
535

536
    /* H.263 specific */
537
    int gob_index;
538
    int obmc;                       ///< overlapped block motion compensation
539
        
540
    /* H.263+ specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
541 542
    int umvplus;                    ///< == H263+ && unrestricted_mv 
    int h263_aic;                   ///< Advanded INTRA Coding (AIC) 
543 544
    int h263_aic_dir;               ///< AIC direction: 0 = left, 1 = top
    int h263_slice_structured;
545
    int alt_inter_vlc;              ///< alternative inter vlc
Michael Niedermayer's avatar
Michael Niedermayer committed
546
    int modified_quant;
547
    int loop_filter;    
548
    int custom_pcf;
549
    
Fabrice Bellard's avatar
Fabrice Bellard committed
550
    /* mpeg4 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
551
    int time_increment_resolution;
Michael Niedermayer's avatar
Michael Niedermayer committed
552
    int time_increment_bits;        ///< number of bits to represent the fractional part of time 
553
    int last_time_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
554 555
    int time_base;                  ///< time in seconds of last I,P,S Frame 
    int64_t time;                   ///< time of current frame  
556
    int64_t last_non_b_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
557 558
    uint16_t pp_time;               ///< time distance between the last 2 p,s,i frames 
    uint16_t pb_time;               ///< time distance between the last b and p,s,i frame 
559
    uint16_t pp_field_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
560
    uint16_t pb_field_time;         ///< like above, just for interlaced 
Fabrice Bellard's avatar
Fabrice Bellard committed
561 562
    int shape;
    int vol_sprite_usage;
563 564 565 566 567
    int sprite_width;
    int sprite_height;
    int sprite_left;
    int sprite_top;
    int sprite_brightness_change;
568
    int num_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
569
    int real_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
570 571 572
    int sprite_offset[2][2];         ///< sprite offset[isChroma][isMVY] 
    int sprite_delta[2][2];          ///< sprite_delta [isY][isMVY]  
    int sprite_shift[2];             ///< sprite shift [isChroma] 
573
    int mcsel;
Fabrice Bellard's avatar
Fabrice Bellard committed
574
    int quant_precision;
Michael Niedermayer's avatar
Michael Niedermayer committed
575
    int quarter_sample;              ///< 1->qpel, 0->half pel ME/MC  
576
    int scalability;
Michael Niedermayer's avatar
Michael Niedermayer committed
577 578
    int hierachy_type;
    int enhancement_type;
579 580
    int new_pred;
    int reduced_res_vop;
581
    int aspect_ratio_info; //FIXME remove
582 583
    int sprite_warping_accuracy;
    int low_latency_sprite;
Michael Niedermayer's avatar
Michael Niedermayer committed
584 585 586 587 588
    int data_partitioning;           ///< data partitioning flag from header 
    int partitioned_frame;           ///< is current frame partitioned 
    int rvlc;                        ///< reversible vlc 
    int resync_marker;               ///< could this stream contain resync markers
    int low_delay;                   ///< no reordering needed / has no b-frames 
Michael Niedermayer's avatar
Michael Niedermayer committed
589
    int vo_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
590
    int vol_control_parameters;      ///< does the stream contain the low_delay flag, used to workaround buggy encoders 
591
    int intra_dc_threshold;          ///< QP above whch the ac VLC should be used for intra dc 
Michael Niedermayer's avatar
Michael Niedermayer committed
592 593
    PutBitContext tex_pb;            ///< used for data partitioned VOPs 
    PutBitContext pb2;               ///< used for data partitioned VOPs 
594
    int mpeg_quant;
Michael Niedermayer's avatar
Michael Niedermayer committed
595 596
    int t_frame;                       ///< time distance of first I -> B, used for interlaced b frames 
    int padding_bug_score;             ///< used to detect the VERY common padding bug in MPEG4 
Michael Niedermayer's avatar
Michael Niedermayer committed
597 598 599 600

    /* divx specific, used to workaround (many) bugs in divx5 */
    int divx_version;
    int divx_build;
601
    int divx_packed;
Michael Niedermayer's avatar
Michael Niedermayer committed
602
#define BITSTREAM_BUFFER_SIZE 1024*256
603
    uint8_t *bitstream_buffer; //Divx 5.01 puts several frames in a single one, this is used to reorder them
Michael Niedermayer's avatar
Michael Niedermayer committed
604 605
    int bitstream_buffer_size;
    
606 607
    int xvid_build;
    
608 609 610 611
    /* lavc specific stuff, used to workaround bugs in libavcodec */
    int ffmpeg_version;
    int lavc_build;
    
Fabrice Bellard's avatar
Fabrice Bellard committed
612
    /* RV10 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
613
    int rv10_version; ///< RV10 version: 0 or 3 
Fabrice Bellard's avatar
Fabrice Bellard committed
614 615 616 617
    int rv10_first_dc_coded[3];
    
    /* MJPEG specific */
    struct MJpegContext *mjpeg_ctx;
Michael Niedermayer's avatar
Michael Niedermayer committed
618 619 620 621
    int mjpeg_vsample[3];       ///< vertical sampling factors, default = {2, 1, 1} 
    int mjpeg_hsample[3];       ///< horizontal sampling factors, default = {2, 1, 1} 
    int mjpeg_write_tables;     ///< do we want to have quantisation- and huffmantables in the jpeg file ? 
    int mjpeg_data_only_frames; ///< frames only with SOI, SOS and EOI markers 
Fabrice Bellard's avatar
Fabrice Bellard committed
622 623 624 625 626 627 628

    /* MSMPEG4 specific */
    int mv_table_index;
    int rl_table_index;
    int rl_chroma_table_index;
    int dc_table_index;
    int use_skip_mb_code;
Michael Niedermayer's avatar
Michael Niedermayer committed
629 630
    int slice_height;      ///< in macroblocks 
    int first_slice_line;  ///< used in mpeg4 too to handle resync markers 
631
    int flipflop_rounding;
Michael Niedermayer's avatar
Michael Niedermayer committed
632
    int msmpeg4_version;   ///< 0=not msmpeg4, 1=mp41, 2=mp42, 3=mp43/divx3 4=wmv1/7 5=wmv2/8
Michael Niedermayer's avatar
Michael Niedermayer committed
633 634 635
    int per_mb_rl_table;
    int esc3_level_length;
    int esc3_run_length;
Michael Niedermayer's avatar
Michael Niedermayer committed
636
    /** [mb_intra][isChroma][level][run][last] */
637
    int (*ac_stats)[2][MAX_LEVEL+1][MAX_RUN+1][2];
638
    int inter_intra_pred;
Michael Niedermayer's avatar
Michael Niedermayer committed
639
    int mspel;
Michael Niedermayer's avatar
Michael Niedermayer committed
640

Fabrice Bellard's avatar
Fabrice Bellard committed
641 642 643
    /* decompression specific */
    GetBitContext gb;

Michael Niedermayer's avatar
Michael Niedermayer committed
644
    /* Mpeg1 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
645 646
    int gop_picture_number;  ///< index of the first picture of a GOP based on fake_pic_num & mpeg1 specific 
    int last_mv_dir;         ///< last mv_dir, used for b frame encoding 
647
    int broken_link;         ///< no_output_of_prior_pics_flag
Michael Niedermayer's avatar
Michael Niedermayer committed
648
    uint8_t *vbv_delay_ptr;  ///< pointer to vbv_delay in the bitstream 
Michael Niedermayer's avatar
Michael Niedermayer committed
649
    
Fabrice Bellard's avatar
Fabrice Bellard committed
650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667
    /* MPEG2 specific - I wish I had not to support this mess. */
    int progressive_sequence;
    int mpeg_f_code[2][2];
    int picture_structure;
/* picture type */
#define PICT_TOP_FIELD     1
#define PICT_BOTTOM_FIELD  2
#define PICT_FRAME         3

    int intra_dc_precision;
    int frame_pred_frame_dct;
    int top_field_first;
    int concealment_motion_vectors;
    int q_scale_type;
    int intra_vlc_format;
    int alternate_scan;
    int repeat_first_field;
    int chroma_420_type;
668 669 670 671 672
    int chroma_format;
#define CHROMA_420 1
#define CHROMA_422 2
#define CHROMA_444 3

Fabrice Bellard's avatar
Fabrice Bellard committed
673 674 675 676
    int progressive_frame;
    int full_pel[2];
    int interlaced_dct;
    int first_slice;
677
    int first_field;         ///< is 1 for the first field of a field picture 0 otherwise
678

679
    /* RTP specific */
680
    int rtp_mode;
681
    
682
    uint8_t *ptr_lastgob;
683 684
    int swap_uv;//vcr2 codec is mpeg2 varint with UV swaped
    short * pblocks[12];
685
    
Michael Niedermayer's avatar
Michael Niedermayer committed
686
    DCTELEM (*block)[64]; ///< points to one of the following blocks 
687
    DCTELEM (*blocks)[6][64]; // for HQ mode we need to keep the best block
688 689 690
    int (*decode_mb)(struct MpegEncContext *s, DCTELEM block[6][64]); // used by some codecs to avoid a switch()
#define SLICE_OK         0
#define SLICE_ERROR     -1
691 692
#define SLICE_END       -2 ///<end marker found
#define SLICE_NOEND     -3 ///<no end marker or error found but mb count exceeded
693
    
694
    void (*dct_unquantize_mpeg1_intra)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
695
                           DCTELEM *block/*align 16*/, int n, int qscale);
696
    void (*dct_unquantize_mpeg1_inter)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
697
                           DCTELEM *block/*align 16*/, int n, int qscale);
698
    void (*dct_unquantize_mpeg2_intra)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
699
                           DCTELEM *block/*align 16*/, int n, int qscale);
700 701 702 703 704 705 706 707 708
    void (*dct_unquantize_mpeg2_inter)(struct MpegEncContext *s, 
                           DCTELEM *block/*align 16*/, int n, int qscale);
    void (*dct_unquantize_h263_intra)(struct MpegEncContext *s, 
                           DCTELEM *block/*align 16*/, int n, int qscale);
    void (*dct_unquantize_h263_inter)(struct MpegEncContext *s, 
                           DCTELEM *block/*align 16*/, int n, int qscale);
    void (*dct_unquantize_intra)(struct MpegEncContext *s, // unquantizer to use (mpeg4 can use both)
                           DCTELEM *block/*align 16*/, int n, int qscale);
    void (*dct_unquantize_inter)(struct MpegEncContext *s, // unquantizer to use (mpeg4 can use both)
Michael Niedermayer's avatar
Michael Niedermayer committed
709 710
                           DCTELEM *block/*align 16*/, int n, int qscale);
    int (*dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
711
    int (*fast_dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
Michael Niedermayer's avatar
Michael Niedermayer committed
712
    void (*denoise_dct)(struct MpegEncContext *s, DCTELEM *block);
Fabrice Bellard's avatar
Fabrice Bellard committed
713 714
} MpegEncContext;

715

716
int DCT_common_init(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
717
void MPV_decode_defaults(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
718 719 720
int MPV_common_init(MpegEncContext *s);
void MPV_common_end(MpegEncContext *s);
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[6][64]);
721
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
722
void MPV_frame_end(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
723 724 725
int MPV_encode_init(AVCodecContext *avctx);
int MPV_encode_end(AVCodecContext *avctx);
int MPV_encode_picture(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data);
726 727 728
#ifdef HAVE_MMX
void MPV_common_init_mmx(MpegEncContext *s);
#endif
729 730 731
#ifdef ARCH_ALPHA
void MPV_common_init_axp(MpegEncContext *s);
#endif
732 733 734
#ifdef HAVE_MLIB
void MPV_common_init_mlib(MpegEncContext *s);
#endif
735 736 737
#ifdef HAVE_MMI
void MPV_common_init_mmi(MpegEncContext *s);
#endif
738
#ifdef ARCH_ARMV4L
Zdenek Kabelac's avatar
Zdenek Kabelac committed
739
void MPV_common_init_armv4l(MpegEncContext *s);
740
#endif
741 742 743
#ifdef ARCH_POWERPC
void MPV_common_init_ppc(MpegEncContext *s);
#endif
744 745
extern void (*draw_edges)(uint8_t *buf, int wrap, int width, int height, int w);
void ff_copy_bits(PutBitContext *pb, uint8_t *src, int length);
746
void ff_clean_intra_table_entries(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
747
void ff_init_scantable(uint8_t *, ScanTable *st, const uint8_t *src_scantable);
748
void ff_draw_horiz_band(MpegEncContext *s, int y, int h);
749
void ff_emulated_edge_mc(uint8_t *buf, uint8_t *src, int linesize, int block_w, int block_h, 
Michael Niedermayer's avatar
Michael Niedermayer committed
750
                                    int src_x, int src_y, int w, int h);
751
#define END_NOT_FOUND -100
Michael Niedermayer's avatar
Michael Niedermayer committed
752
int ff_combine_frame( MpegEncContext *s, int next, uint8_t **buf, int *buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
753
void ff_mpeg_flush(AVCodecContext *avctx);
754
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict);
755
void ff_write_quant_matrix(PutBitContext *pb, int16_t *matrix);
756
int ff_find_unused_picture(MpegEncContext *s, int shared);
757
void ff_denoise_dct(MpegEncContext *s, DCTELEM *block);
758
void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src);
Michael Niedermayer's avatar
Michael Niedermayer committed
759

760 761 762 763 764
void ff_er_frame_start(MpegEncContext *s);
void ff_er_frame_end(MpegEncContext *s);
void ff_er_add_slice(MpegEncContext *s, int startx, int starty, int endx, int endy, int status);


Michael Niedermayer's avatar
Michael Niedermayer committed
765
extern enum PixelFormat ff_yuv420p_list[2];
Fabrice Bellard's avatar
Fabrice Bellard committed
766

767
void ff_init_block_index(MpegEncContext *s);
768 769 770 771 772 773 774 775

static inline void ff_update_block_index(MpegEncContext *s){
    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]++;
776 777 778
    s->dest[0]+= 16;
    s->dest[1]+= 8;
    s->dest[2]+= 8;
779 780
}

781
static inline int get_bits_diff(MpegEncContext *s){
782
    const int bits= put_bits_count(&s->pb);
783 784 785 786 787 788
    const int last= s->last_bits;

    s->last_bits = bits;

    return bits - last;
}
789

Fabrice Bellard's avatar
Fabrice Bellard committed
790
/* motion_est.c */
791 792 793 794 795 796
void ff_estimate_p_frame_motion(MpegEncContext * s,
                             int mb_x, int mb_y);
void ff_estimate_b_frame_motion(MpegEncContext * s,
                             int mb_x, int mb_y);
int ff_get_best_fcode(MpegEncContext * s, int16_t (*mv_table)[2], int type);
void ff_fix_long_p_mvs(MpegEncContext * s);
797 798
void ff_fix_long_mvs(MpegEncContext * s, uint8_t *field_select_table, int field_select,
                     int16_t (*mv_table)[2], int f_code, int type, int truncate);
Michael Niedermayer's avatar
Michael Niedermayer committed
799
void ff_init_me(MpegEncContext *s);
800
int ff_pre_estimate_p_frame_motion(MpegEncContext * s, int mb_x, int mb_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
801

802

Fabrice Bellard's avatar
Fabrice Bellard committed
803
/* mpeg12.c */
804 805 806
extern const int16_t ff_mpeg1_default_intra_matrix[64];
extern const int16_t ff_mpeg1_default_non_intra_matrix[64];
extern uint8_t ff_mpeg1_dc_scale_table[128];
Fabrice Bellard's avatar
Fabrice Bellard committed
807 808 809 810 811

void mpeg1_encode_picture_header(MpegEncContext *s, int picture_number);
void mpeg1_encode_mb(MpegEncContext *s,
                     DCTELEM block[6][64],
                     int motion_x, int motion_y);
Michael Niedermayer's avatar
Michael Niedermayer committed
812
void ff_mpeg1_encode_init(MpegEncContext *s);
813 814
void ff_mpeg1_encode_slice_header(MpegEncContext *s);
void ff_mpeg1_clean_buffers(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
815

816

Michael Niedermayer's avatar
Michael Niedermayer committed
817
/** RLTable. */
Fabrice Bellard's avatar
Fabrice Bellard committed
818
typedef struct RLTable {
Michael Niedermayer's avatar
Michael Niedermayer committed
819 820
    int n;                         ///< number of entries of table_vlc minus 1 
    int last;                      ///< number of values for last = 0 
821 822 823
    const uint16_t (*table_vlc)[2];
    const int8_t *table_run;
    const int8_t *table_level;
Michael Niedermayer's avatar
Michael Niedermayer committed
824 825 826 827 828
    uint8_t *index_run[2];         ///< encoding only 
    int8_t *max_level[2];          ///< encoding & decoding 
    int8_t *max_run[2];            ///< encoding & decoding 
    VLC vlc;                       ///< decoding only deprected FIXME remove
    RL_VLC_ELEM *rl_vlc[32];       ///< decoding only 
Fabrice Bellard's avatar
Fabrice Bellard committed
829 830 831 832 833
} RLTable;

void init_rl(RLTable *rl);
void init_vlc_rl(RLTable *rl);

834
static inline int get_rl_index(const RLTable *rl, int last, int run, int level)
Fabrice Bellard's avatar
Fabrice Bellard committed
835 836 837 838 839 840 841 842 843 844
{
    int index;
    index = rl->index_run[last][run];
    if (index >= rl->n)
        return rl->n;
    if (level > rl->max_level[last][run])
        return rl->n;
    return index + level - 1;
}

845 846
extern uint8_t ff_mpeg4_y_dc_scale_table[32];
extern uint8_t ff_mpeg4_c_dc_scale_table[32];
Michael Niedermayer's avatar
Michael Niedermayer committed
847
extern uint8_t ff_aic_dc_scale_table[32];
848 849
extern const int16_t ff_mpeg4_default_intra_matrix[64];
extern const int16_t ff_mpeg4_default_non_intra_matrix[64];
Michael Niedermayer's avatar
Michael Niedermayer committed
850 851
extern const uint8_t ff_h263_chroma_qscale_table[32];
extern const uint8_t ff_h263_loop_filter_strength[32];
Michael Niedermayer's avatar
Michael Niedermayer committed
852

Michael Niedermayer's avatar
Michael Niedermayer committed
853

Michael Niedermayer's avatar
Michael Niedermayer committed
854 855 856
int ff_h263_decode_init(AVCodecContext *avctx);
int ff_h263_decode_frame(AVCodecContext *avctx, 
                             void *data, int *data_size,
857
                             uint8_t *buf, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
858
int ff_h263_decode_end(AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
859 860 861
void h263_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
862 863 864
void mpeg4_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
865
void h263_encode_picture_header(MpegEncContext *s, int picture_number);
866
void ff_flv_encode_picture_header(MpegEncContext *s, int picture_number);
867
void h263_encode_gob_header(MpegEncContext * s, int mb_line);
868
int16_t *h263_pred_motion(MpegEncContext * s, int block, int dir,
Fabrice Bellard's avatar
Fabrice Bellard committed
869
                        int *px, int *py);
870
void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n, 
Fabrice Bellard's avatar
Fabrice Bellard committed
871
                   int dir);
872
void ff_set_mpeg4_time(MpegEncContext * s, int picture_number);
Fabrice Bellard's avatar
Fabrice Bellard committed
873
void mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
874
void h263_encode_init(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
875 876
void h263_decode_init_vlc(MpegEncContext *s);
int h263_decode_picture_header(MpegEncContext *s);
877
int ff_h263_decode_gob_header(MpegEncContext *s);
878
int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb);
879
void ff_h263_update_motion_val(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
880
void ff_h263_loop_filter(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
881
void ff_set_qscale(MpegEncContext * s, int qscale);
882 883
int ff_h263_decode_mba(MpegEncContext *s);
void ff_h263_encode_mba(MpegEncContext *s);
884

Fabrice Bellard's avatar
Fabrice Bellard committed
885
int intel_h263_decode_picture_header(MpegEncContext *s);
886
int flv_h263_decode_picture_header(MpegEncContext *s);
887 888
int ff_h263_decode_mb(MpegEncContext *s,
                      DCTELEM block[6][64]);
889 890
int ff_mpeg4_decode_mb(MpegEncContext *s,
                      DCTELEM block[6][64]);
Fabrice Bellard's avatar
Fabrice Bellard committed
891
int h263_get_picture_format(int width, int height);
892 893 894 895 896
void ff_mpeg4_encode_video_packet_header(MpegEncContext *s);
void ff_mpeg4_clean_buffers(MpegEncContext *s);
void ff_mpeg4_stuffing(PutBitContext * pbc);
void ff_mpeg4_init_partitions(MpegEncContext *s);
void ff_mpeg4_merge_partitions(MpegEncContext *s);
897
void ff_clean_mpeg4_qscales(MpegEncContext *s);
898
void ff_clean_h263_qscales(MpegEncContext *s);
899 900 901 902
int ff_mpeg4_decode_partitions(MpegEncContext *s);
int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s);
int ff_h263_resync(MpegEncContext *s);
int ff_h263_get_gob_height(MpegEncContext *s);
903
int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my);
904
inline int ff_h263_round_chroma(int x);
905

Fabrice Bellard's avatar
Fabrice Bellard committed
906 907 908 909 910

/* rv10.c */
void rv10_encode_picture_header(MpegEncContext *s, int picture_number);
int rv_decode_dc(MpegEncContext *s, int n);

911

Fabrice Bellard's avatar
Fabrice Bellard committed
912 913
/* msmpeg4.c */
void msmpeg4_encode_picture_header(MpegEncContext * s, int picture_number);
914
void msmpeg4_encode_ext_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
915 916 917 918
void msmpeg4_encode_mb(MpegEncContext * s, 
                       DCTELEM block[6][64],
                       int motion_x, int motion_y);
int msmpeg4_decode_picture_header(MpegEncContext * s);
919
int msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
920 921
int ff_msmpeg4_decode_init(MpegEncContext *s);
void ff_msmpeg4_encode_init(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
922
int ff_wmv2_decode_picture_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
923
int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
924 925
void ff_wmv2_add_mb(MpegEncContext *s, DCTELEM block[6][64], uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
void ff_mspel_motion(MpegEncContext *s,
926 927
                               uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
                               uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
Michael Niedermayer's avatar
Michael Niedermayer committed
928 929 930 931 932
                               int motion_x, int motion_y, int h);
int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
void ff_wmv2_encode_mb(MpegEncContext * s, 
                       DCTELEM block[6][64],
                       int motion_x, int motion_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
933

934
/* mjpeg.c */
Fabrice Bellard's avatar
Fabrice Bellard committed
935 936 937 938 939 940
int mjpeg_init(MpegEncContext *s);
void mjpeg_close(MpegEncContext *s);
void mjpeg_encode_mb(MpegEncContext *s, 
                     DCTELEM block[6][64]);
void mjpeg_picture_header(MpegEncContext *s);
void mjpeg_picture_trailer(MpegEncContext *s);
941
void ff_mjpeg_stuffing(PutBitContext * pbc);
942

943

944 945
/* rate control */
int ff_rate_control_init(MpegEncContext *s);
946
float ff_rate_estimate_qscale(MpegEncContext *s);
947 948
void ff_write_pass1_stats(MpegEncContext *s);
void ff_rate_control_uninit(MpegEncContext *s);
949 950
double ff_eval(char *s, double *const_value, const char **const_name,
               double (**func1)(void *, double), const char **func1_name,
951 952
               double (**func2)(void *, double, double), char **func2_name,
               void *opaque);
953
int ff_vbv_update(MpegEncContext *s, int frame_size);
954

955

956
#endif /* AVCODEC_MPEGVIDEO_H */