mpegvideo.h 40.8 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
#include "dsputil.h"
30
#include "bitstream.h"
31

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

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

Michael Niedermayer's avatar
Michael Niedermayer committed
42 43
#define EDGE_WIDTH 16

Fabrice Bellard's avatar
Fabrice Bellard committed
44 45
#define MPEG_BUF_SIZE (16 * 1024)

Michael Niedermayer's avatar
Michael Niedermayer committed
46 47
#define QMAT_SHIFT_MMX 16
#define QMAT_SHIFT 22
48

49 50
#define MAX_FCODE 7
#define MAX_MV 2048
Michael Niedermayer's avatar
Michael Niedermayer committed
51

52 53
#define MAX_THREADS 8

54
#define MAX_PICTURE_COUNT 32
55

56 57 58 59
#define ME_MAP_SIZE 64
#define ME_MAP_SHIFT 3
#define ME_MAP_MV_BITS 11

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

Michael Niedermayer's avatar
Michael Niedermayer committed
64 65 66 67
#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
68 69
#define SI_TYPE FF_SI_TYPE  ///< Switching Intra
#define SP_TYPE FF_SP_TYPE  ///< Switching Predicted
70

71 72
#define MAX_MB_BYTES (30*16*16*3/8 + 120)

73 74 75 76 77 78
typedef struct Predictor{
    double coeff;
    double count;
    double decay;
} Predictor;

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

Michael Niedermayer's avatar
Michael Niedermayer committed
96 97 98
/**
 * rate control context.
 */
99 100
typedef struct RateControlContext{
    FILE *stats_file;
Michael Niedermayer's avatar
Michael Niedermayer committed
101
    int num_entries;              ///< number of RateControlEntries 
102
    RateControlEntry *entry;
Michael Niedermayer's avatar
Michael Niedermayer committed
103
    double buffer_index;          ///< amount of bits in the video/audio buffer 
104
    Predictor pred[5];
Michael Niedermayer's avatar
Michael Niedermayer committed
105 106 107 108
    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) 
109
    double last_qscale;
Michael Niedermayer's avatar
Michael Niedermayer committed
110
    double last_qscale_for[5];    ///< last qscale for a specific pict type, used for max_diff & ipb factor stuff 
111 112
    int last_mc_mb_var_sum;
    int last_mb_var_sum;
113 114 115 116
    uint64_t i_cplx_sum[5];
    uint64_t p_cplx_sum[5];
    uint64_t mv_bits_sum[5];
    uint64_t qscale_sum[5];
117
    int frame_count[5];
118
    int last_non_b_pict_type;
119 120
}RateControlContext;

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

Michael Niedermayer's avatar
Michael Niedermayer committed
134 135 136
/**
 * Picture.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
137
typedef struct Picture{
138
    FF_COMMON_FRAME
Michael Niedermayer's avatar
Michael Niedermayer committed
139

140 141 142 143
    /**
     * halfpel luma planes.
     */
    uint8_t *interpolated[3];
144
    int16_t (*motion_val_base[2])[2];
145
    uint32_t *mb_type_base;
146
#define MB_TYPE_INTRA MB_TYPE_INTRA4x4 //default mb_type if theres just one type
147 148
#define IS_INTRA4x4(a)   ((a)&MB_TYPE_INTRA4x4)
#define IS_INTRA16x16(a) ((a)&MB_TYPE_INTRA16x16)
149 150
#define IS_PCM(a)        ((a)&MB_TYPE_INTRA_PCM)
#define IS_INTRA(a)      ((a)&7)
151
#define IS_INTER(a)      ((a)&(MB_TYPE_16x16|MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8))
152
#define IS_SKIP(a)       ((a)&MB_TYPE_SKIP)
153 154 155
#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)
156
#define IS_GMC(a)        ((a)&MB_TYPE_GMC)
157 158 159 160 161 162 163 164
#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
165 166
#define IS_ACPRED(a)     ((a)&MB_TYPE_ACPRED)
#define IS_QUANT(a)      ((a)&MB_TYPE_QUANT)
167 168
#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
169
#define HAS_CBP(a)        ((a)&MB_TYPE_CBP)
170 171 172 173 174 175

    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
176 177
    int ref_poc[2][16];         ///< h264 POCs of the frames used as reference
    int ref_count[2];           ///< number of entries in ref_poc
178

Michael Niedermayer's avatar
Michael Niedermayer committed
179 180 181 182 183
    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
184
    int32_t *mb_cmp_score;	///< Table for MB cmp scores, for mb decission FIXME remove
185
    int b_frame_score;          /* */
Michael Niedermayer's avatar
Michael Niedermayer committed
186 187
} Picture;

188
typedef struct ParseContext{
189
    uint8_t *buffer;
190 191 192
    int index;
    int last_index;
    int buffer_size;
193
    uint32_t state;             ///< contains the last few bytes in MSB order
194
    int frame_start_found;
195 196
    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
197 198
} ParseContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
199 200
struct MpegEncContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
201 202 203
/**
 * Motion estimation context.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
204
typedef struct MotionEstContext{
205
    AVCodecContext *avctx;
Michael Niedermayer's avatar
Michael Niedermayer committed
206 207
    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
208
    int direct_basis_mv[4][2];
Michael Niedermayer's avatar
Michael Niedermayer committed
209
    uint8_t *scratchpad;               ///< data area for the me algo, so that the ME doesnt need to malloc/free 
210 211 212 213
    uint8_t *best_mb;
    uint8_t *temp_mb[2];
    uint8_t *temp;
    int best_bits;
Michael Niedermayer's avatar
Michael Niedermayer committed
214 215
    uint32_t *map;                     ///< map to avoid duplicate evaluations 
    uint32_t *score_map;               ///< map to store the scores 
Michael Niedermayer's avatar
Michael Niedermayer committed
216
    int map_generation;  
Michael Niedermayer's avatar
Michael Niedermayer committed
217
    int pre_penalty_factor;
Michael Niedermayer's avatar
Michael Niedermayer committed
218 219
    int penalty_factor;
    int sub_penalty_factor;
220
    int mb_penalty_factor;
221 222 223
    int flags;
    int sub_flags;
    int mb_flags;
Michael Niedermayer's avatar
Michael Niedermayer committed
224
    int pre_pass;                      ///< = 1 for the pre pass 
Michael Niedermayer's avatar
Michael Niedermayer committed
225
    int dia_size;
226 227 228 229
    int xmin;
    int xmax;
    int ymin;
    int ymax;
230 231 232 233 234 235
    int pred_x;
    int pred_y;
    uint8_t *src[4][4];
    uint8_t *ref[4][4];
    int stride;
    int uvstride;
236 237 238 239
    /* temp variables for picture complexity calculation */
    int mc_mb_var_sum_temp;
    int mb_var_sum_temp;
    int scene_change_score;
240 241 242 243 244
/*    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];
245
    uint8_t (*mv_penalty)[MAX_MV*2+1];  ///< amount of bits needed to encode a MV 
246
    uint8_t *current_mv_penalty;
Michael Niedermayer's avatar
Michael Niedermayer committed
247 248
    int (*sub_motion_search)(struct MpegEncContext * s,
				  int *mx_ptr, int *my_ptr, int dmin,
249 250
                                  int src_index, int ref_index,
                                  int size, int h);
Michael Niedermayer's avatar
Michael Niedermayer committed
251 252
}MotionEstContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
253 254 255
/**
 * MpegEncContext.
 */
Fabrice Bellard's avatar
Fabrice Bellard committed
256
typedef struct MpegEncContext {
257
    struct AVCodecContext *avctx;
Fabrice Bellard's avatar
Fabrice Bellard committed
258
    /* the following parameters must be initialized before encoding */
Michael Niedermayer's avatar
Michael Niedermayer committed
259
    int width, height;///< picture size. must be a multiple of 16 
Fabrice Bellard's avatar
Fabrice Bellard committed
260
    int gop_size;
Michael Niedermayer's avatar
Michael Niedermayer committed
261 262 263 264
    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 
265 266

/* the following codec id fields are deprecated in favor of codec_id */
Michael Niedermayer's avatar
Michael Niedermayer committed
267 268
    int h263_plus;    ///< h263 plus headers 
    int h263_msmpeg4; ///< generate MSMPEG4 compatible stream (deprecated, use msmpeg4_version instead)
269
    int h263_flv;     ///< use flv h263 header 
270
    
Michael Niedermayer's avatar
Michael Niedermayer committed
271
    enum CodecID codec_id;     /* see CODEC_ID_xxx */
Michael Niedermayer's avatar
Michael Niedermayer committed
272 273 274
    int fixed_qscale; ///< fixed qscale if non zero 
    int encoding;     ///< true if we are encoding (vs decoding) 
    int flags;        ///< AVCodecContext.flags (HQ, MV4, ...) 
275
    int flags2;       ///< AVCodecContext.flags2
Michael Niedermayer's avatar
Michael Niedermayer committed
276
    int max_b_frames; ///< max number of b-frames for encoding 
277 278
    int luma_elim_threshold;
    int chroma_elim_threshold;
Michael Niedermayer's avatar
Michael Niedermayer committed
279 280
    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
281 282
    /* the following fields are managed internally by the encoder */

Michael Niedermayer's avatar
Michael Niedermayer committed
283
    /** bit output */
Fabrice Bellard's avatar
Fabrice Bellard committed
284 285 286 287
    PutBitContext pb;

    /* sequence parameters */
    int context_initialized;
288 289 290
    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
291 292
    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 
293
    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
294
    int mb_width, mb_height;   ///< number of MBs horizontally & vertically 
295
    int mb_stride;             ///< mb_width+1 used for some arrays to allow simple addressng of left & top MBs withoutt sig11
296 297
    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
298 299 300 301
    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 
302
    Picture *picture;          ///< main picture buffer 
303 304
    Picture **input_picture;   ///< next pictures on display order for encoding
    Picture **reordered_input_picture; ///< pointer to the next pictures in codedorder for encoding
305
    
306 307 308 309
    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];
    
310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331
    /** 
     * 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
332
    Picture current_picture;    ///< buffer to store the decompressed current picture 
333 334 335 336
    
    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
337
    uint8_t *visualization_buffer[3]; //< temporary buffer vor MV visualization
Michael Niedermayer's avatar
Michael Niedermayer committed
338
    int last_dc[3];                ///< last DC values for MPEG1 
339
    int16_t *dc_val_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
340
    int16_t *dc_val[3];            ///< used for mpeg4 DC prediction, all 3 arrays must be continuous 
341
    int16_t dc_cache[4*5];
Fabrice Bellard's avatar
Fabrice Bellard committed
342
    int y_dc_scale, c_dc_scale;
Michael Niedermayer's avatar
Michael Niedermayer committed
343 344
    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
345
    const uint8_t *chroma_qscale_table;  ///< qscale -> chroma_qscale (h263)
346
    uint8_t *coded_block_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
347
    uint8_t *coded_block;          ///< used for coded block pattern prediction (msmpeg4v3, wmv1)
348
    int16_t (*ac_val_base)[16];
Michael Niedermayer's avatar
Michael Niedermayer committed
349
    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
350
    int ac_pred;
Michael Niedermayer's avatar
Michael Niedermayer committed
351
    uint8_t *prev_pict_types;     ///< previous picture types in bitstream order, used for mb skip 
352
#define PREV_PICT_TYPES_BUFFER_SIZE 256
Michael Niedermayer's avatar
Michael Niedermayer committed
353 354
    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) 
355
                                   and used for b-frame encoding & decoding (contains skip table of next P Frame) */
Michael Niedermayer's avatar
Michael Niedermayer committed
356 357 358
    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 
359
    uint8_t *allocated_edge_emu_buffer;
360 361 362 363
    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
364

Michael Niedermayer's avatar
Michael Niedermayer committed
365
    int qscale;                 ///< QP 
Michael Niedermayer's avatar
Michael Niedermayer committed
366
    int chroma_qscale;          ///< chroma QP 
367 368 369
    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
370 371 372
    int adaptive_quant;         ///< use adaptive quantization 
    int dquant;                 ///< qscale difference to prev qscale  
    int pict_type;              ///< I_TYPE, P_TYPE, B_TYPE, ... 
373
    int last_pict_type; //FIXME removes
Michael Niedermayer's avatar
Michael Niedermayer committed
374
    int last_non_b_pict_type;   ///< used for mpeg4 gmc b-frames & ratecontrol 
375
    int dropable;
Fabrice Bellard's avatar
Fabrice Bellard committed
376
    int frame_rate_index;
377

Fabrice Bellard's avatar
Fabrice Bellard committed
378
    /* motion compensation */
Michael Niedermayer's avatar
Michael Niedermayer committed
379 380
    int unrestricted_mv;        ///< mv can point outside of the coded picture 
    int h263_long_vectors;      ///< use horrible h263v1 long vector mode 
381
    int decode;                 ///< if 0 then decoding will be skiped (for encoding b frames for example)
Michael Niedermayer's avatar
Michael Niedermayer committed
382 383 384 385

    DSPContext dsp;             ///< pointers for accelerated dsp fucntions 
    int f_code;                 ///< forward MV resolution 
    int b_code;                 ///< backward MV resolution for B Frames (mpeg4) 
386 387 388 389 390 391
    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];
392 393
    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
394 395 396 397 398 399
    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 
400 401 402 403
    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
404
    int me_method;                       ///< ME algorithm 
Fabrice Bellard's avatar
Fabrice Bellard committed
405
    int mv_dir;
406 407
#define MV_DIR_BACKWARD  1
#define MV_DIR_FORWARD   2
Michael Niedermayer's avatar
Michael Niedermayer committed
408
#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
409
    int mv_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
410 411 412 413 414 415
#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
416 417 418 419 420 421
       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
422 423
    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
424 425
    
    MotionEstContext me;
Fabrice Bellard's avatar
Fabrice Bellard committed
426

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

Michael Niedermayer's avatar
Michael Niedermayer committed
430
    int hurry_up;     /**< when set to 1 during decoding, b frames will be skiped
431
                         when set to 2 idct/dequant will be skipped too */
432
                        
Fabrice Bellard's avatar
Fabrice Bellard committed
433 434
    /* macroblock layer */
    int mb_x, mb_y;
435
    int mb_skip_run;
Fabrice Bellard's avatar
Fabrice Bellard committed
436
    int mb_intra;
437 438 439 440 441
    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
442
//#define MB_TYPE_GMC      0x10
443

444 445 446 447 448 449 450 451 452
#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
453

Michael Niedermayer's avatar
Michael Niedermayer committed
454
    int block_index[6]; ///< index to current MB in block based arrays with edges
Michael Niedermayer's avatar
Michael Niedermayer committed
455
    int block_wrap[6];
456
    uint8_t *dest[3];
457 458
    
    int *mb_index2xy;        ///< mb_index -> mb_x + mb_y*mb_stride
Michael Niedermayer's avatar
Michael Niedermayer committed
459

Michael Niedermayer's avatar
Michael Niedermayer committed
460
    /** matrix transmitted in the bitstream */
461 462 463 464
    uint16_t intra_matrix[64];
    uint16_t chroma_intra_matrix[64];
    uint16_t inter_matrix[64];
    uint16_t chroma_inter_matrix[64];
465
#define QUANT_BIAS_SHIFT 8
Michael Niedermayer's avatar
Michael Niedermayer committed
466 467 468 469 470
    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 
471 472 473 474
    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;
475 476
    uint8_t *luma_dc_vlc_length;
    uint8_t *chroma_dc_vlc_length;
477
#define UNI_AC_ENC_INDEX(run,level) ((run)*128 + (level))
478

479 480
    int coded_score[6];

Michael Niedermayer's avatar
Michael Niedermayer committed
481
    /** precomputed matrix (combine qscale and DCT renorm) */
482 483
    int (*q_intra_matrix)[64];
    int (*q_inter_matrix)[64];
484 485 486
    /** 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];
487
    int block_last_index[12];  ///< last non zero coefficient in block
488
    /* scantables */
489
    ScanTable __align8 intra_scantable;
490 491
    ScanTable intra_h_scantable;
    ScanTable intra_v_scantable;
Michael Niedermayer's avatar
Michael Niedermayer committed
492
    ScanTable inter_scantable; ///< if inter == intra then intra should be used to reduce tha cache usage
493 494 495 496 497
    
    /* noise reduction */
    int (*dct_error_sum)[64];
    int dct_count[2];
    uint16_t (*dct_offset)[64];
Fabrice Bellard's avatar
Fabrice Bellard committed
498

Michael Niedermayer's avatar
Michael Niedermayer committed
499
    void *opaque;              ///< private data for the user
Fabrice Bellard's avatar
Fabrice Bellard committed
500 501

    /* bit rate control */
502 503
    int64_t wanted_bits;
    int64_t total_bits;
Michael Niedermayer's avatar
Michael Niedermayer committed
504 505
    int frame_bits;                ///< bits used for the current frame 
    RateControlContext rc_context; ///< contains stuff only accessed in ratecontrol.c
506

507 508 509 510 511 512
    /* statistics, used for 2-pass encoding */
    int mv_bits;
    int header_bits;
    int i_tex_bits;
    int p_tex_bits;
    int i_count;
513 514
    int f_count;
    int b_count;
515
    int skip_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
516 517
    int misc_bits; ///< cbp, mb_type
    int last_bits; ///< temp var used for calculating the above vars
518 519
    
    /* error concealment / resync */
520
    int error_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
521 522
    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 
523 524 525 526 527 528 529 530
#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
531 532 533 534 535
    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 
536
    int error_resilience;
537 538
    
    ParseContext parse_context;
539

540
    /* H.263 specific */
541
    int gob_index;
542
    int obmc;                       ///< overlapped block motion compensation
543
        
544
    /* H.263+ specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
545 546
    int umvplus;                    ///< == H263+ && unrestricted_mv 
    int h263_aic;                   ///< Advanded INTRA Coding (AIC) 
547 548
    int h263_aic_dir;               ///< AIC direction: 0 = left, 1 = top
    int h263_slice_structured;
549
    int alt_inter_vlc;              ///< alternative inter vlc
Michael Niedermayer's avatar
Michael Niedermayer committed
550
    int modified_quant;
551
    int loop_filter;    
552
    int custom_pcf;
553
    
Fabrice Bellard's avatar
Fabrice Bellard committed
554
    /* mpeg4 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
555
    int time_increment_resolution;
Michael Niedermayer's avatar
Michael Niedermayer committed
556
    int time_increment_bits;        ///< number of bits to represent the fractional part of time 
557
    int last_time_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
558 559
    int time_base;                  ///< time in seconds of last I,P,S Frame 
    int64_t time;                   ///< time of current frame  
560
    int64_t last_non_b_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
561 562
    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 
563
    uint16_t pp_field_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
564
    uint16_t pb_field_time;         ///< like above, just for interlaced 
Fabrice Bellard's avatar
Fabrice Bellard committed
565 566
    int shape;
    int vol_sprite_usage;
567 568 569 570 571
    int sprite_width;
    int sprite_height;
    int sprite_left;
    int sprite_top;
    int sprite_brightness_change;
572
    int num_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
573
    int real_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
574 575 576
    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] 
577
    int mcsel;
Fabrice Bellard's avatar
Fabrice Bellard committed
578
    int quant_precision;
Michael Niedermayer's avatar
Michael Niedermayer committed
579
    int quarter_sample;              ///< 1->qpel, 0->half pel ME/MC  
580
    int scalability;
Michael Niedermayer's avatar
Michael Niedermayer committed
581 582
    int hierachy_type;
    int enhancement_type;
583 584
    int new_pred;
    int reduced_res_vop;
585
    int aspect_ratio_info; //FIXME remove
586 587
    int sprite_warping_accuracy;
    int low_latency_sprite;
Michael Niedermayer's avatar
Michael Niedermayer committed
588 589 590 591 592
    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
593
    int vo_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
594
    int vol_control_parameters;      ///< does the stream contain the low_delay flag, used to workaround buggy encoders 
595
    int intra_dc_threshold;          ///< QP above whch the ac VLC should be used for intra dc 
Michael Niedermayer's avatar
Michael Niedermayer committed
596 597
    PutBitContext tex_pb;            ///< used for data partitioned VOPs 
    PutBitContext pb2;               ///< used for data partitioned VOPs 
598
    int mpeg_quant;
Michael Niedermayer's avatar
Michael Niedermayer committed
599 600
    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
601 602 603 604

    /* divx specific, used to workaround (many) bugs in divx5 */
    int divx_version;
    int divx_build;
605
    int divx_packed;
606
    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
607
    int bitstream_buffer_size;
608
    int allocated_bitstream_buffer_size;
Michael Niedermayer's avatar
Michael Niedermayer committed
609
    
610 611
    int xvid_build;
    
612 613 614 615
    /* lavc specific stuff, used to workaround bugs in libavcodec */
    int ffmpeg_version;
    int lavc_build;
    
Fabrice Bellard's avatar
Fabrice Bellard committed
616
    /* RV10 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
617
    int rv10_version; ///< RV10 version: 0 or 3 
Fabrice Bellard's avatar
Fabrice Bellard committed
618 619 620 621
    int rv10_first_dc_coded[3];
    
    /* MJPEG specific */
    struct MJpegContext *mjpeg_ctx;
Michael Niedermayer's avatar
Michael Niedermayer committed
622 623 624 625
    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
626 627 628 629 630 631 632

    /* 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
633 634
    int slice_height;      ///< in macroblocks 
    int first_slice_line;  ///< used in mpeg4 too to handle resync markers 
635
    int flipflop_rounding;
Michael Niedermayer's avatar
Michael Niedermayer committed
636
    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
637 638 639
    int per_mb_rl_table;
    int esc3_level_length;
    int esc3_run_length;
Michael Niedermayer's avatar
Michael Niedermayer committed
640
    /** [mb_intra][isChroma][level][run][last] */
641
    int (*ac_stats)[2][MAX_LEVEL+1][MAX_RUN+1][2];
642
    int inter_intra_pred;
Michael Niedermayer's avatar
Michael Niedermayer committed
643
    int mspel;
Michael Niedermayer's avatar
Michael Niedermayer committed
644

Fabrice Bellard's avatar
Fabrice Bellard committed
645 646 647
    /* decompression specific */
    GetBitContext gb;

Michael Niedermayer's avatar
Michael Niedermayer committed
648
    /* Mpeg1 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
649 650
    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 
651
    int broken_link;         ///< no_output_of_prior_pics_flag
Michael Niedermayer's avatar
Michael Niedermayer committed
652
    uint8_t *vbv_delay_ptr;  ///< pointer to vbv_delay in the bitstream 
Michael Niedermayer's avatar
Michael Niedermayer committed
653
    
Fabrice Bellard's avatar
Fabrice Bellard committed
654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
    /* 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;
672 673 674 675
    int chroma_format;
#define CHROMA_420 1
#define CHROMA_422 2
#define CHROMA_444 3
676 677
    int chroma_x_shift;//depend on pix_format, that depend on chroma_format
    int chroma_y_shift;
678

Fabrice Bellard's avatar
Fabrice Bellard committed
679 680 681 682
    int progressive_frame;
    int full_pel[2];
    int interlaced_dct;
    int first_slice;
683
    int first_field;         ///< is 1 for the first field of a field picture 0 otherwise
684

685
    /* RTP specific */
686
    int rtp_mode;
687
    
688
    uint8_t *ptr_lastgob;
689 690
    int swap_uv;//vcr2 codec is mpeg2 varint with UV swaped
    short * pblocks[12];
691
    
Michael Niedermayer's avatar
Michael Niedermayer committed
692
    DCTELEM (*block)[64]; ///< points to one of the following blocks 
693
    DCTELEM (*blocks)[6][64]; // for HQ mode we need to keep the best block
694 695 696
    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
697 698
#define SLICE_END       -2 ///<end marker found
#define SLICE_NOEND     -3 ///<no end marker or error found but mb count exceeded
699
    
700
    void (*dct_unquantize_mpeg1_intra)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
701
                           DCTELEM *block/*align 16*/, int n, int qscale);
702
    void (*dct_unquantize_mpeg1_inter)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
703
                           DCTELEM *block/*align 16*/, int n, int qscale);
704
    void (*dct_unquantize_mpeg2_intra)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
705
                           DCTELEM *block/*align 16*/, int n, int qscale);
706 707 708 709 710 711
    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);
712 713 714 715
    void (*dct_unquantize_h261_intra)(struct MpegEncContext *s, 
                           DCTELEM *block/*align 16*/, int n, int qscale);
    void (*dct_unquantize_h261_inter)(struct MpegEncContext *s, 
                           DCTELEM *block/*align 16*/, int n, int qscale);
716 717 718
    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
719 720
                           DCTELEM *block/*align 16*/, int n, int qscale);
    int (*dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
721
    int (*fast_dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
Michael Niedermayer's avatar
Michael Niedermayer committed
722
    void (*denoise_dct)(struct MpegEncContext *s, DCTELEM *block);
Fabrice Bellard's avatar
Fabrice Bellard committed
723 724
} MpegEncContext;

725

726
int DCT_common_init(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
727
void MPV_decode_defaults(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
728 729
int MPV_common_init(MpegEncContext *s);
void MPV_common_end(MpegEncContext *s);
730
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]);
731
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
732
void MPV_frame_end(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
733 734 735
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);
736 737 738
#ifdef HAVE_MMX
void MPV_common_init_mmx(MpegEncContext *s);
#endif
739 740 741
#ifdef ARCH_ALPHA
void MPV_common_init_axp(MpegEncContext *s);
#endif
742 743 744
#ifdef HAVE_MLIB
void MPV_common_init_mlib(MpegEncContext *s);
#endif
745 746 747
#ifdef HAVE_MMI
void MPV_common_init_mmi(MpegEncContext *s);
#endif
748
#ifdef ARCH_ARMV4L
Zdenek Kabelac's avatar
Zdenek Kabelac committed
749
void MPV_common_init_armv4l(MpegEncContext *s);
750
#endif
751 752 753
#ifdef ARCH_POWERPC
void MPV_common_init_ppc(MpegEncContext *s);
#endif
754 755
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);
756
void ff_clean_intra_table_entries(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
757
void ff_init_scantable(uint8_t *, ScanTable *st, const uint8_t *src_scantable);
758
void ff_draw_horiz_band(MpegEncContext *s, int y, int h);
759
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
760
                                    int src_x, int src_y, int w, int h);
761
#define END_NOT_FOUND -100
762 763
int ff_combine_frame(ParseContext *pc, int next, uint8_t **buf, int *buf_size);
void ff_parse_close(AVCodecParserContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
764
void ff_mpeg_flush(AVCodecContext *avctx);
765
void ff_print_debug_info(MpegEncContext *s, AVFrame *pict);
766
void ff_write_quant_matrix(PutBitContext *pb, int16_t *matrix);
767
int ff_find_unused_picture(MpegEncContext *s, int shared);
768
void ff_denoise_dct(MpegEncContext *s, DCTELEM *block);
769
void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src);
Michael Niedermayer's avatar
Michael Niedermayer committed
770

771 772 773 774 775
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
776
extern enum PixelFormat ff_yuv420p_list[2];
Fabrice Bellard's avatar
Fabrice Bellard committed
777

778
void ff_init_block_index(MpegEncContext *s);
779 780

static inline void ff_update_block_index(MpegEncContext *s){
781 782
    const int block_size= 8>>s->avctx->lowres;

783 784 785 786 787 788
    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]++;
789 790 791
    s->dest[0]+= 2*block_size;
    s->dest[1]+= block_size;
    s->dest[2]+= block_size;
792 793
}

794
static inline int get_bits_diff(MpegEncContext *s){
795
    const int bits= put_bits_count(&s->pb);
796 797 798 799 800 801
    const int last= s->last_bits;

    s->last_bits = bits;

    return bits - last;
}
802

Fabrice Bellard's avatar
Fabrice Bellard committed
803
/* motion_est.c */
804 805 806 807 808 809
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);
810 811
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
812
void ff_init_me(MpegEncContext *s);
813
int ff_pre_estimate_p_frame_motion(MpegEncContext * s, int mb_x, int mb_y);
814 815 816 817 818
inline int ff_epzs_motion_search(MpegEncContext * s, int *mx_ptr, int *my_ptr,
                             int P[10][2], int src_index, int ref_index, int16_t (*last_mv)[2], 
                             int ref_mv_scale, int size, int h);                             
int inline ff_get_mb_score(MpegEncContext * s, int mx, int my, int src_index,
                               int ref_index, int size, int h, int add_rate);
819

Fabrice Bellard's avatar
Fabrice Bellard committed
820
/* mpeg12.c */
821 822 823
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
824 825 826 827 828

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
829
void ff_mpeg1_encode_init(MpegEncContext *s);
830 831
void ff_mpeg1_encode_slice_header(MpegEncContext *s);
void ff_mpeg1_clean_buffers(MpegEncContext *s);
832
int ff_mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size);
Fabrice Bellard's avatar
Fabrice Bellard committed
833

834

Michael Niedermayer's avatar
Michael Niedermayer committed
835
/** RLTable. */
Fabrice Bellard's avatar
Fabrice Bellard committed
836
typedef struct RLTable {
Michael Niedermayer's avatar
Michael Niedermayer committed
837 838
    int n;                         ///< number of entries of table_vlc minus 1 
    int last;                      ///< number of values for last = 0 
839 840 841
    const uint16_t (*table_vlc)[2];
    const int8_t *table_run;
    const int8_t *table_level;
Michael Niedermayer's avatar
Michael Niedermayer committed
842 843 844 845 846
    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
847 848
} RLTable;

849 850
void init_rl(RLTable *rl, int use_static);
void init_vlc_rl(RLTable *rl, int use_static);
Fabrice Bellard's avatar
Fabrice Bellard committed
851

852
static inline int get_rl_index(const RLTable *rl, int last, int run, int level)
Fabrice Bellard's avatar
Fabrice Bellard committed
853 854 855 856 857 858 859 860 861 862
{
    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;
}

863 864
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
865
extern uint8_t ff_aic_dc_scale_table[32];
866 867
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
868 869
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
870

871 872
/* h261.c */
void ff_h261_loop_filter(MpegEncContext *s);
873 874 875 876 877 878
void ff_h261_reorder_mb_index(MpegEncContext* s);
void ff_h261_encode_mb(MpegEncContext *s,
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
void ff_h261_encode_picture_header(MpegEncContext * s, int picture_number);
void ff_h261_encode_init(MpegEncContext *s);
879

Michael Niedermayer's avatar
Michael Niedermayer committed
880

881
/* h263.c, h263dec.c */
Michael Niedermayer's avatar
Michael Niedermayer committed
882 883 884
int ff_h263_decode_init(AVCodecContext *avctx);
int ff_h263_decode_frame(AVCodecContext *avctx, 
                             void *data, int *data_size,
885
                             uint8_t *buf, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
886
int ff_h263_decode_end(AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
887 888 889
void h263_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
890 891 892
void mpeg4_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
893
void h263_encode_picture_header(MpegEncContext *s, int picture_number);
894
void ff_flv_encode_picture_header(MpegEncContext *s, int picture_number);
895
void h263_encode_gob_header(MpegEncContext * s, int mb_line);
896
int16_t *h263_pred_motion(MpegEncContext * s, int block, int dir,
Fabrice Bellard's avatar
Fabrice Bellard committed
897
                        int *px, int *py);
898
void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n, 
Fabrice Bellard's avatar
Fabrice Bellard committed
899
                   int dir);
900
void ff_set_mpeg4_time(MpegEncContext * s, int picture_number);
Fabrice Bellard's avatar
Fabrice Bellard committed
901
void mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
902
void h263_encode_init(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
903 904
void h263_decode_init_vlc(MpegEncContext *s);
int h263_decode_picture_header(MpegEncContext *s);
905
int ff_h263_decode_gob_header(MpegEncContext *s);
906
int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb);
907
void ff_h263_update_motion_val(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
908
void ff_h263_loop_filter(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
909
void ff_set_qscale(MpegEncContext * s, int qscale);
910 911
int ff_h263_decode_mba(MpegEncContext *s);
void ff_h263_encode_mba(MpegEncContext *s);
912

Fabrice Bellard's avatar
Fabrice Bellard committed
913
int intel_h263_decode_picture_header(MpegEncContext *s);
914
int flv_h263_decode_picture_header(MpegEncContext *s);
915 916
int ff_h263_decode_mb(MpegEncContext *s,
                      DCTELEM block[6][64]);
917 918
int ff_mpeg4_decode_mb(MpegEncContext *s,
                      DCTELEM block[6][64]);
Fabrice Bellard's avatar
Fabrice Bellard committed
919
int h263_get_picture_format(int width, int height);
920 921 922 923 924
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);
925
void ff_clean_mpeg4_qscales(MpegEncContext *s);
926
void ff_clean_h263_qscales(MpegEncContext *s);
927 928 929 930
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);
931
int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my);
932
int ff_h263_round_chroma(int x);
933
void ff_h263_encode_motion(MpegEncContext * s, int val, int f_code);
934
int ff_mpeg4_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size);
935

Fabrice Bellard's avatar
Fabrice Bellard committed
936 937 938 939

/* rv10.c */
void rv10_encode_picture_header(MpegEncContext *s, int picture_number);
int rv_decode_dc(MpegEncContext *s, int n);
Michael Niedermayer's avatar
Michael Niedermayer committed
940
void rv20_encode_picture_header(MpegEncContext *s, int picture_number);
Fabrice Bellard's avatar
Fabrice Bellard committed
941

942

Fabrice Bellard's avatar
Fabrice Bellard committed
943 944
/* msmpeg4.c */
void msmpeg4_encode_picture_header(MpegEncContext * s, int picture_number);
945
void msmpeg4_encode_ext_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
946 947 948 949
void msmpeg4_encode_mb(MpegEncContext * s, 
                       DCTELEM block[6][64],
                       int motion_x, int motion_y);
int msmpeg4_decode_picture_header(MpegEncContext * s);
950
int msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
951 952
int ff_msmpeg4_decode_init(MpegEncContext *s);
void ff_msmpeg4_encode_init(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
953
int ff_wmv2_decode_picture_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
954
int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
955 956
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,
957 958
                               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
959 960 961 962 963
                               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
964

965
/* mjpeg.c */
Fabrice Bellard's avatar
Fabrice Bellard committed
966 967 968 969 970 971
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);
972
void ff_mjpeg_stuffing(PutBitContext * pbc);
973

974

975 976
/* rate control */
int ff_rate_control_init(MpegEncContext *s);
977
float ff_rate_estimate_qscale(MpegEncContext *s);
978 979
void ff_write_pass1_stats(MpegEncContext *s);
void ff_rate_control_uninit(MpegEncContext *s);
980 981
double ff_eval(char *s, double *const_value, const char **const_name,
               double (**func1)(void *, double), const char **func1_name,
982 983
               double (**func2)(void *, double, double), char **func2_name,
               void *opaque);
984
int ff_vbv_update(MpegEncContext *s, int frame_size);
985

986

987
#endif /* AVCODEC_MPEGVIDEO_H */