mpegvideo.h 37.2 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.
Fabrice Bellard's avatar
Fabrice Bellard committed
4
 *
5 6 7 8
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
Fabrice Bellard's avatar
Fabrice Bellard committed
9
 *
10
 * This library is distributed in the hope that it will be useful,
Fabrice Bellard's avatar
Fabrice Bellard committed
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
Fabrice Bellard's avatar
Fabrice Bellard committed
14
 *
15 16 17
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
Fabrice Bellard's avatar
Fabrice Bellard committed
18 19
 */

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

28 29
#include "dsputil.h"

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

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

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

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

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

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

49
#define MAX_PICTURE_COUNT 15
50

51 52 53 54
#define ME_MAP_SIZE 64
#define ME_MAP_SHIFT 3
#define ME_MAP_MV_BITS 11

Michael Niedermayer's avatar
Michael Niedermayer committed
55 56 57 58
/* run length table */
#define MAX_RUN    64
#define MAX_LEVEL  64

Michael Niedermayer's avatar
Michael Niedermayer committed
59 60 61 62
#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
63 64
#define SI_TYPE FF_SI_TYPE  ///< Switching Intra
#define SP_TYPE FF_SP_TYPE  ///< Switching Predicted
65

66 67 68 69 70 71
typedef struct Predictor{
    double coeff;
    double count;
    double decay;
} Predictor;

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

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

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

Michael Niedermayer's avatar
Michael Niedermayer committed
127 128 129
/**
 * Picture.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
130
typedef struct Picture{
131
    FF_COMMON_FRAME
Michael Niedermayer's avatar
Michael Niedermayer committed
132

133 134 135 136 137 138 139
    /**
     * halfpel luma planes.
     */
    uint8_t *interpolated[3];
    
    int16_t (*motion_val[2])[2];
    int8_t *ref_index[2];
140 141
    uint32_t *mb_type_base;
    uint32_t *mb_type;           ///< mb_type_base + mb_width + 2, note: only used for decoding currently
142
#define MB_TYPE_INTRA4x4   0x0001
143 144
#define MB_TYPE_INTRA16x16 0x0002 //FIXME h264 specific
#define MB_TYPE_INTRA_PCM  0x0004 //FIXME h264 specific
145 146 147 148 149 150
#define MB_TYPE_16x16      0x0008
#define MB_TYPE_16x8       0x0010
#define MB_TYPE_8x16       0x0020
#define MB_TYPE_8x8        0x0040
#define MB_TYPE_INTERLACED 0x0080
#define MB_TYPE_DIRECT2     0x0100 //FIXME
151 152 153
#define MB_TYPE_ACPRED     0x0200
#define MB_TYPE_GMC        0x0400 //FIXME mpeg4 specific
#define MB_TYPE_SKIP       0x0800
154 155 156 157
#define MB_TYPE_P0L0       0x1000
#define MB_TYPE_P1L0       0x2000
#define MB_TYPE_P0L1       0x4000
#define MB_TYPE_P1L1       0x8000
158 159 160 161 162
#define MB_TYPE_L0         (MB_TYPE_P0L0 | MB_TYPE_P1L0)
#define MB_TYPE_L1         (MB_TYPE_P0L1 | MB_TYPE_P1L1)
#define MB_TYPE_L0L1       (MB_TYPE_L0   | MB_TYPE_L1)
#define MB_TYPE_QUANT      0x00010000
//Note bits 24-31 are reserved for codec specific use (h264 ref0, mpeg1 pat, ...)
163 164 165

#define IS_INTRA4x4(a)   ((a)&MB_TYPE_INTRA4x4)
#define IS_INTRA16x16(a) ((a)&MB_TYPE_INTRA16x16)
166 167
#define IS_PCM(a)        ((a)&MB_TYPE_INTRA_PCM)
#define IS_INTRA(a)      ((a)&7)
168
#define IS_INTER(a)      ((a)&(MB_TYPE_16x16|MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8))
169
#define IS_SKIP(a)       ((a)&MB_TYPE_SKIP)
170 171 172
#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)
173
#define IS_GMC(a)        ((a)&MB_TYPE_GMC)
174 175 176 177 178 179 180 181
#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
182 183
#define IS_ACPRED(a)     ((a)&MB_TYPE_ACPRED)
#define IS_QUANT(a)      ((a)&MB_TYPE_QUANT)
184 185 186 187 188 189 190 191 192 193
#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


    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
194 195 196 197 198 199
    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 
    int32_t *mb_cmp_score;	///< Table for MB cmp scores, for mb decission 
200
    int b_frame_score;          /* */
Michael Niedermayer's avatar
Michael Niedermayer committed
201 202
} Picture;

203
typedef struct ParseContext{
204
    uint8_t *buffer;
205 206 207
    int index;
    int last_index;
    int buffer_size;
208
    uint32_t state;             ///< contains the last few bytes in MSB order
209
    int frame_start_found;
210 211
    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
212 213
} ParseContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
214 215
struct MpegEncContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
216 217 218
/**
 * Motion estimation context.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
219
typedef struct MotionEstContext{
Michael Niedermayer's avatar
Michael Niedermayer committed
220 221
    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
222
    int direct_basis_mv[4][2];
Michael Niedermayer's avatar
Michael Niedermayer committed
223 224 225
    uint8_t *scratchpad;               ///< data area for the me algo, so that the ME doesnt need to malloc/free 
    uint32_t *map;                     ///< map to avoid duplicate evaluations 
    uint32_t *score_map;               ///< map to store the scores 
Michael Niedermayer's avatar
Michael Niedermayer committed
226
    int map_generation;  
Michael Niedermayer's avatar
Michael Niedermayer committed
227
    int pre_penalty_factor;
Michael Niedermayer's avatar
Michael Niedermayer committed
228 229
    int penalty_factor;
    int sub_penalty_factor;
230
    int mb_penalty_factor;
Michael Niedermayer's avatar
Michael Niedermayer committed
231
    int pre_pass;                      ///< = 1 for the pre pass 
Michael Niedermayer's avatar
Michael Niedermayer committed
232
    int dia_size;
233
    uint8_t (*mv_penalty)[MAX_MV*2+1];  ///< amount of bits needed to encode a MV 
Michael Niedermayer's avatar
Michael Niedermayer committed
234 235 236 237
    int (*sub_motion_search)(struct MpegEncContext * s,
				  int *mx_ptr, int *my_ptr, int dmin,
				  int xmin, int ymin, int xmax, int ymax,
                                  int pred_x, int pred_y, Picture *ref_picture, 
238
                                  int n, int size, uint8_t * const mv_penalty);
Michael Niedermayer's avatar
Michael Niedermayer committed
239 240 241
    int (*motion_search[7])(struct MpegEncContext * s, int block,
                             int *mx_ptr, int *my_ptr,
                             int P[10][2], int pred_x, int pred_y,
Michael Niedermayer's avatar
Michael Niedermayer committed
242
                             int xmin, int ymin, int xmax, int ymax, Picture *ref_picture, int16_t (*last_mv)[2], 
243
                             int ref_mv_scale, uint8_t * const mv_penalty);
Michael Niedermayer's avatar
Michael Niedermayer committed
244 245 246
    int (*pre_motion_search)(struct MpegEncContext * s, int block,
                             int *mx_ptr, int *my_ptr,
                             int P[10][2], int pred_x, int pred_y,
247
                             int xmin, int ymin, int xmax, int ymax, Picture *ref_picture, int16_t (*last_mv)[2], 
248
                             int ref_mv_scale, uint8_t * const mv_penalty);
249
    int (*get_mb_score)(struct MpegEncContext * s, int mx, int my, int pred_x, int pred_y, Picture *ref_picture, 
250
                                  uint8_t * const mv_penalty);
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 265
    int intra_only;   ///< if true, only intra pictures are generated 
    int bit_rate;     ///< wanted bit rate 
    int bit_rate_tolerance; ///< amount of +- bits (>0)
    enum OutputFormat out_format; ///< output format 
    int h263_pred;    ///< use mpeg4/h263 ac/dc predictions 
266 267

/* the following codec id fields are deprecated in favor of codec_id */
Michael Niedermayer's avatar
Michael Niedermayer committed
268 269 270 271
    int h263_plus;    ///< h263 plus headers 
    int h263_rv10;    ///< use RV10 variation for H263 
    int h263_msmpeg4; ///< generate MSMPEG4 compatible stream (deprecated, use msmpeg4_version instead)
    int h263_intel;   ///< use I263 intel h263 header 
272
    int h263_flv;     ///< use flv h263 header 
273 274
    
    int codec_id;     /* see CODEC_ID_xxx */
Michael Niedermayer's avatar
Michael Niedermayer committed
275 276 277 278 279 280 281
    int fixed_qscale; ///< fixed qscale if non zero 
    float qcompress;  ///< amount of qscale change between easy & hard scenes (0.0-1.0) 
    float qblur;      ///< amount of qscale smoothing over time (0.0-1.0) 
    int max_qdiff;    ///< max qscale difference between frames 
    int encoding;     ///< true if we are encoding (vs decoding) 
    int flags;        ///< AVCodecContext.flags (HQ, MV4, ...) 
    int max_b_frames; ///< max number of b-frames for encoding 
282
    int b_frame_strategy;
283 284
    int luma_elim_threshold;
    int chroma_elim_threshold;
Michael Niedermayer's avatar
Michael Niedermayer committed
285 286
    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
287 288
    /* the following fields are managed internally by the encoder */

Michael Niedermayer's avatar
Michael Niedermayer committed
289
    /** bit output */
Fabrice Bellard's avatar
Fabrice Bellard committed
290 291 292 293
    PutBitContext pb;

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

Michael Niedermayer's avatar
Michael Niedermayer committed
355
    int qscale;                 ///< 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;
Michael Niedermayer's avatar
Michael Niedermayer committed
363
    int last_non_b_pict_type;   ///< used for mpeg4 gmc b-frames & ratecontrol 
Fabrice Bellard's avatar
Fabrice Bellard committed
364 365
    int frame_rate_index;
    /* motion compensation */
Michael Niedermayer's avatar
Michael Niedermayer committed
366 367
    int unrestricted_mv;        ///< mv can point outside of the coded picture 
    int h263_long_vectors;      ///< use horrible h263v1 long vector mode 
368
    int decode;                 ///< if 0 then decoding will be skiped (for encoding b frames for example)
Michael Niedermayer's avatar
Michael Niedermayer committed
369 370 371 372

    DSPContext dsp;             ///< pointers for accelerated dsp fucntions 
    int f_code;                 ///< forward MV resolution 
    int b_code;                 ///< backward MV resolution for B Frames (mpeg4) 
373 374 375 376 377 378 379
    int16_t (*motion_val)[2];
    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];
Michael Niedermayer's avatar
Michael Niedermayer committed
380 381 382 383 384 385 386
    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 
    int me_method;                       ///< ME algorithm 
387
    int scene_change_score;
Fabrice Bellard's avatar
Fabrice Bellard committed
388
    int mv_dir;
389 390
#define MV_DIR_BACKWARD  1
#define MV_DIR_FORWARD   2
Michael Niedermayer's avatar
Michael Niedermayer committed
391
#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
392
    int mv_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
393 394 395 396 397 398
#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
399 400 401 402 403 404
       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
405 406
    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
407 408
    
    MotionEstContext me;
Fabrice Bellard's avatar
Fabrice Bellard committed
409

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

Michael Niedermayer's avatar
Michael Niedermayer committed
413
    int hurry_up;     /**< when set to 1 during decoding, b frames will be skiped
414
                         when set to 2 idct/dequant will be skipped too */
415
                        
Fabrice Bellard's avatar
Fabrice Bellard committed
416 417
    /* macroblock layer */
    int mb_x, mb_y;
418
    int mb_skip_run;
Fabrice Bellard's avatar
Fabrice Bellard committed
419
    int mb_intra;
420
    uint8_t *mb_type;       ///< Table for MB type FIXME remove and use picture->mb_type
421 422 423 424
#define MB_TYPE_INTRA    0x01
#define MB_TYPE_INTER    0x02
#define MB_TYPE_INTER4V  0x04
#define MB_TYPE_SKIPED   0x08
425
//#define MB_TYPE_GMC      0x10
426

427 428
#define MB_TYPE_DIRECT   0x10
#define MB_TYPE_FORWARD  0x20
429
#define MB_TYPE_BACKWARD 0x40
430
#define MB_TYPE_BIDIR    0x80
Michael Niedermayer's avatar
Michael Niedermayer committed
431

Michael Niedermayer's avatar
Michael Niedermayer committed
432
    int block_index[6]; ///< index to current MB in block based arrays with edges
Michael Niedermayer's avatar
Michael Niedermayer committed
433
    int block_wrap[6];
434
    uint8_t *dest[3];
435 436
    
    int *mb_index2xy;        ///< mb_index -> mb_x + mb_y*mb_stride
Michael Niedermayer's avatar
Michael Niedermayer committed
437

Michael Niedermayer's avatar
Michael Niedermayer committed
438
    /** matrix transmitted in the bitstream */
439 440 441 442
    uint16_t intra_matrix[64];
    uint16_t chroma_intra_matrix[64];
    uint16_t inter_matrix[64];
    uint16_t chroma_inter_matrix[64];
443
#define QUANT_BIAS_SHIFT 8
Michael Niedermayer's avatar
Michael Niedermayer committed
444 445 446 447 448
    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 
449 450 451 452
    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;
453 454
    uint8_t *luma_dc_vlc_length;
    uint8_t *chroma_dc_vlc_length;
455
#define UNI_AC_ENC_INDEX(run,level) ((run)*128 + (level))
456

457 458
    int coded_score[6];

Michael Niedermayer's avatar
Michael Niedermayer committed
459
    /** precomputed matrix (combine qscale and DCT renorm) */
460 461
    int (*q_intra_matrix)[64];
    int (*q_inter_matrix)[64];
462 463 464
    /** 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];
Michael Niedermayer's avatar
Michael Niedermayer committed
465
    int block_last_index[6];  ///< last non zero coefficient in block
466
    /* scantables */
467
    ScanTable __align8 intra_scantable;
468 469
    ScanTable intra_h_scantable;
    ScanTable intra_v_scantable;
Michael Niedermayer's avatar
Michael Niedermayer committed
470
    ScanTable inter_scantable; ///< if inter == intra then intra should be used to reduce tha cache usage
Fabrice Bellard's avatar
Fabrice Bellard committed
471

Michael Niedermayer's avatar
Michael Niedermayer committed
472
    void *opaque;              ///< private data for the user
Fabrice Bellard's avatar
Fabrice Bellard committed
473 474

    /* bit rate control */
475
    int I_frame_bits; //FIXME used in mpeg12 ...
476 477
    int64_t wanted_bits;
    int64_t total_bits;
Michael Niedermayer's avatar
Michael Niedermayer committed
478 479
    int frame_bits;                ///< bits used for the current frame 
    RateControlContext rc_context; ///< contains stuff only accessed in ratecontrol.c
480

481 482 483 484 485 486
    /* statistics, used for 2-pass encoding */
    int mv_bits;
    int header_bits;
    int i_tex_bits;
    int p_tex_bits;
    int i_count;
487 488
    int f_count;
    int b_count;
489
    int skip_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
490 491
    int misc_bits; ///< cbp, mb_type
    int last_bits; ///< temp var used for calculating the above vars
492 493
    
    /* error concealment / resync */
494
    int error_count;
Michael Niedermayer's avatar
Michael Niedermayer committed
495 496
    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 
497 498 499 500 501 502 503 504
#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
505 506 507 508 509
    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 
510
    int error_resilience;
511 512
    
    ParseContext parse_context;
513

514 515
    /* H.263 specific */
    int gob_number;
516
    int gob_index;
517
        
518
    /* H.263+ specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
519 520 521
    int umvplus;                    ///< == H263+ && unrestricted_mv 
    int h263_aic;                   ///< Advanded INTRA Coding (AIC) 
    int h263_aic_dir;               ///< AIC direction: 0 = left, 1 = top 
522
    
Fabrice Bellard's avatar
Fabrice Bellard committed
523
    /* mpeg4 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
524
    int time_increment_resolution;
Michael Niedermayer's avatar
Michael Niedermayer committed
525
    int time_increment_bits;        ///< number of bits to represent the fractional part of time 
526
    int last_time_base;
Michael Niedermayer's avatar
Michael Niedermayer committed
527 528
    int time_base;                  ///< time in seconds of last I,P,S Frame 
    int64_t time;                   ///< time of current frame  
529
    int64_t last_non_b_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
530 531
    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 
532
    uint16_t pp_field_time;
Michael Niedermayer's avatar
Michael Niedermayer committed
533
    uint16_t pb_field_time;         ///< like above, just for interlaced 
Fabrice Bellard's avatar
Fabrice Bellard committed
534 535
    int shape;
    int vol_sprite_usage;
536 537 538 539 540
    int sprite_width;
    int sprite_height;
    int sprite_left;
    int sprite_top;
    int sprite_brightness_change;
541
    int num_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
542
    int real_sprite_warping_points;
Michael Niedermayer's avatar
Michael Niedermayer committed
543 544 545
    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] 
546
    int mcsel;
Fabrice Bellard's avatar
Fabrice Bellard committed
547
    int quant_precision;
Michael Niedermayer's avatar
Michael Niedermayer committed
548
    int quarter_sample;              ///< 1->qpel, 0->half pel ME/MC  
549
    int scalability;
Michael Niedermayer's avatar
Michael Niedermayer committed
550 551
    int hierachy_type;
    int enhancement_type;
552 553
    int new_pred;
    int reduced_res_vop;
554
    int aspect_ratio_info; //FIXME remove
555 556
    int sprite_warping_accuracy;
    int low_latency_sprite;
Michael Niedermayer's avatar
Michael Niedermayer committed
557 558 559 560 561
    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
562
    int vo_type;
Michael Niedermayer's avatar
Michael Niedermayer committed
563
    int vol_control_parameters;      ///< does the stream contain the low_delay flag, used to workaround buggy encoders 
564
    int intra_dc_threshold;          ///< QP above whch the ac VLC should be used for intra dc 
Michael Niedermayer's avatar
Michael Niedermayer committed
565 566
    PutBitContext tex_pb;            ///< used for data partitioned VOPs 
    PutBitContext pb2;               ///< used for data partitioned VOPs 
567 568 569
#define PB_BUFFER_SIZE 1024*256
    uint8_t *tex_pb_buffer;          
    uint8_t *pb2_buffer;
570
    int mpeg_quant;
Michael Niedermayer's avatar
Michael Niedermayer committed
571 572 573 574
    int16_t (*field_mv_table)[2][2];   ///< used for interlaced b frame decoding 
    int8_t (*field_select_table)[2];   ///< wtf, no really another table for interlaced b frames 
    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
575 576 577 578

    /* divx specific, used to workaround (many) bugs in divx5 */
    int divx_version;
    int divx_build;
579
    int divx_packed;
Michael Niedermayer's avatar
Michael Niedermayer committed
580
#define BITSTREAM_BUFFER_SIZE 1024*256
581
    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
582 583
    int bitstream_buffer_size;
    
584 585
    int xvid_build;
    
586 587 588 589
    /* lavc specific stuff, used to workaround bugs in libavcodec */
    int ffmpeg_version;
    int lavc_build;
    
Fabrice Bellard's avatar
Fabrice Bellard committed
590
    /* RV10 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
591
    int rv10_version; ///< RV10 version: 0 or 3 
Fabrice Bellard's avatar
Fabrice Bellard committed
592 593 594 595
    int rv10_first_dc_coded[3];
    
    /* MJPEG specific */
    struct MJpegContext *mjpeg_ctx;
Michael Niedermayer's avatar
Michael Niedermayer committed
596 597 598 599
    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
600 601 602 603 604 605 606

    /* 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
607 608
    int slice_height;      ///< in macroblocks 
    int first_slice_line;  ///< used in mpeg4 too to handle resync markers 
609
    int flipflop_rounding;
Michael Niedermayer's avatar
Michael Niedermayer committed
610
    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
611 612 613
    int per_mb_rl_table;
    int esc3_level_length;
    int esc3_run_length;
Michael Niedermayer's avatar
Michael Niedermayer committed
614
    /** [mb_intra][isChroma][level][run][last] */
615
    int (*ac_stats)[2][MAX_LEVEL+1][MAX_RUN+1][2];
616
    int inter_intra_pred;
Michael Niedermayer's avatar
Michael Niedermayer committed
617
    int mspel;
Michael Niedermayer's avatar
Michael Niedermayer committed
618

Fabrice Bellard's avatar
Fabrice Bellard committed
619 620 621
    /* decompression specific */
    GetBitContext gb;

Michael Niedermayer's avatar
Michael Niedermayer committed
622
    /* Mpeg1 specific */
Michael Niedermayer's avatar
Michael Niedermayer committed
623 624 625
    int fake_picture_number; ///< picture number at the bitstream frame rate 
    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 
626
    int broken_link;         ///< no_output_of_prior_pics_flag
Michael Niedermayer's avatar
Michael Niedermayer committed
627
    
Fabrice Bellard's avatar
Fabrice Bellard committed
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649
    /* 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;
    int progressive_frame;
    int full_pel[2];
    int interlaced_dct;
    int first_slice;
650
    int first_field;         ///< is 1 for the first field of a field picture 0 otherwise
651 652
    
    /* RTP specific */
653
    /* These are explained on avcodec.h */
654
    int rtp_mode;
655
    int rtp_payload_size;
656
    void (*rtp_callback)(void *data, int size, int packet_number);
657
    uint8_t *ptr_lastgob;
658 659
    int swap_uv;//vcr2 codec is mpeg2 varint with UV swaped
    short * pblocks[12];
660
    
Michael Niedermayer's avatar
Michael Niedermayer committed
661
    DCTELEM (*block)[64]; ///< points to one of the following blocks 
662
    DCTELEM (*blocks)[6][64]; // for HQ mode we need to keep the best block
663 664 665
    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
666 667
#define SLICE_END       -2 ///<end marker found
#define SLICE_NOEND     -3 ///<no end marker or error found but mb count exceeded
668
    
669
    void (*dct_unquantize_mpeg1)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
670
                           DCTELEM *block/*align 16*/, int n, int qscale);
671
    void (*dct_unquantize_mpeg2)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
672
                           DCTELEM *block/*align 16*/, int n, int qscale);
673
    void (*dct_unquantize_h263)(struct MpegEncContext *s, 
Michael Niedermayer's avatar
Michael Niedermayer committed
674
                           DCTELEM *block/*align 16*/, int n, int qscale);
675
    void (*dct_unquantize)(struct MpegEncContext *s, // unquantizer to use (mpeg4 can use both)
Michael Niedermayer's avatar
Michael Niedermayer committed
676 677
                           DCTELEM *block/*align 16*/, int n, int qscale);
    int (*dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
678
    int (*fast_dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
Fabrice Bellard's avatar
Fabrice Bellard committed
679 680
} MpegEncContext;

681

682
int DCT_common_init(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
683 684 685
int MPV_common_init(MpegEncContext *s);
void MPV_common_end(MpegEncContext *s);
void MPV_decode_mb(MpegEncContext *s, DCTELEM block[6][64]);
686
int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
687
void MPV_frame_end(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
688 689 690
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);
691 692 693
#ifdef HAVE_MMX
void MPV_common_init_mmx(MpegEncContext *s);
#endif
694 695 696
#ifdef ARCH_ALPHA
void MPV_common_init_axp(MpegEncContext *s);
#endif
697 698 699
#ifdef HAVE_MLIB
void MPV_common_init_mlib(MpegEncContext *s);
#endif
700 701 702
#ifdef HAVE_MMI
void MPV_common_init_mmi(MpegEncContext *s);
#endif
703
#ifdef ARCH_ARMV4L
Zdenek Kabelac's avatar
Zdenek Kabelac committed
704
void MPV_common_init_armv4l(MpegEncContext *s);
705
#endif
706 707 708
#ifdef ARCH_POWERPC
void MPV_common_init_ppc(MpegEncContext *s);
#endif
709 710
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);
711
void ff_clean_intra_table_entries(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
712
void ff_init_scantable(uint8_t *, ScanTable *st, const uint8_t *src_scantable);
713
void ff_draw_horiz_band(MpegEncContext *s, int y, int h);
714
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
715
                                    int src_x, int src_y, int w, int h);
716
#define END_NOT_FOUND -100
Michael Niedermayer's avatar
Michael Niedermayer committed
717
int ff_combine_frame( MpegEncContext *s, int next, uint8_t **buf, int *buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
718
void ff_mpeg_flush(AVCodecContext *avctx);
719
void ff_print_debug_info(MpegEncContext *s, Picture *pict);
720
void ff_write_quant_matrix(PutBitContext *pb, int16_t *matrix);
721
int ff_find_unused_picture(MpegEncContext *s, int shared);
Michael Niedermayer's avatar
Michael Niedermayer committed
722

723 724 725 726 727
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
728
extern enum PixelFormat ff_yuv420p_list[2];
Fabrice Bellard's avatar
Fabrice Bellard committed
729

730
void ff_init_block_index(MpegEncContext *s);
731 732 733 734 735 736 737 738

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]++;
739 740 741
    s->dest[0]+= 16;
    s->dest[1]+= 8;
    s->dest[2]+= 8;
742 743
}

744 745 746 747 748 749 750 751
static inline int get_bits_diff(MpegEncContext *s){
    const int bits= get_bit_count(&s->pb);
    const int last= s->last_bits;

    s->last_bits = bits;

    return bits - last;
}
752

Fabrice Bellard's avatar
Fabrice Bellard committed
753
/* motion_est.c */
754 755 756 757 758 759 760
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);
void ff_fix_long_b_mvs(MpegEncContext * s, int16_t (*mv_table)[2], int f_code, int type);
Michael Niedermayer's avatar
Michael Niedermayer committed
761
void ff_init_me(MpegEncContext *s);
762
int ff_pre_estimate_p_frame_motion(MpegEncContext * s, int mb_x, int mb_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
763

764

Fabrice Bellard's avatar
Fabrice Bellard committed
765
/* mpeg12.c */
766 767 768
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
769 770 771 772 773

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
774
void ff_mpeg1_encode_init(MpegEncContext *s);
775 776
void ff_mpeg1_encode_slice_header(MpegEncContext *s);
void ff_mpeg1_clean_buffers(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
777

778

Michael Niedermayer's avatar
Michael Niedermayer committed
779
/** RLTable. */
Fabrice Bellard's avatar
Fabrice Bellard committed
780
typedef struct RLTable {
Michael Niedermayer's avatar
Michael Niedermayer committed
781 782
    int n;                         ///< number of entries of table_vlc minus 1 
    int last;                      ///< number of values for last = 0 
783 784 785
    const uint16_t (*table_vlc)[2];
    const int8_t *table_run;
    const int8_t *table_level;
Michael Niedermayer's avatar
Michael Niedermayer committed
786 787 788 789 790
    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
791 792 793 794 795
} RLTable;

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

796
static inline int get_rl_index(const RLTable *rl, int last, int run, int level)
Fabrice Bellard's avatar
Fabrice Bellard committed
797 798 799 800 801 802 803 804 805 806
{
    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;
}

807 808 809 810
extern uint8_t ff_mpeg4_y_dc_scale_table[32];
extern uint8_t ff_mpeg4_c_dc_scale_table[32];
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
811 812 813
int ff_h263_decode_init(AVCodecContext *avctx);
int ff_h263_decode_frame(AVCodecContext *avctx, 
                             void *data, int *data_size,
814
                             uint8_t *buf, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
815
int ff_h263_decode_end(AVCodecContext *avctx);
Fabrice Bellard's avatar
Fabrice Bellard committed
816 817 818
void h263_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
819 820 821
void mpeg4_encode_mb(MpegEncContext *s, 
                    DCTELEM block[6][64],
                    int motion_x, int motion_y);
Fabrice Bellard's avatar
Fabrice Bellard committed
822
void h263_encode_picture_header(MpegEncContext *s, int picture_number);
823
void ff_flv_encode_picture_header(MpegEncContext *s, int picture_number);
824
int h263_encode_gob_header(MpegEncContext * s, int mb_line);
825
int16_t *h263_pred_motion(MpegEncContext * s, int block, 
Fabrice Bellard's avatar
Fabrice Bellard committed
826
                        int *px, int *py);
827
void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n, 
Fabrice Bellard's avatar
Fabrice Bellard committed
828
                   int dir);
829
void ff_set_mpeg4_time(MpegEncContext * s, int picture_number);
Fabrice Bellard's avatar
Fabrice Bellard committed
830
void mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
831
void h263_encode_init(MpegEncContext *s);
Fabrice Bellard's avatar
Fabrice Bellard committed
832 833
void h263_decode_init_vlc(MpegEncContext *s);
int h263_decode_picture_header(MpegEncContext *s);
834
int ff_h263_decode_gob_header(MpegEncContext *s);
835
int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb);
836
void ff_h263_update_motion_val(MpegEncContext * s);
837 838


Fabrice Bellard's avatar
Fabrice Bellard committed
839
int intel_h263_decode_picture_header(MpegEncContext *s);
840
int flv_h263_decode_picture_header(MpegEncContext *s);
841 842
int ff_h263_decode_mb(MpegEncContext *s,
                      DCTELEM block[6][64]);
Fabrice Bellard's avatar
Fabrice Bellard committed
843
int h263_get_picture_format(int width, int height);
844 845 846 847 848
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);
849
void ff_clean_mpeg4_qscales(MpegEncContext *s);
850
void ff_clean_h263_qscales(MpegEncContext *s);
851 852 853 854
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);
855
int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my);
856
inline int ff_h263_round_chroma(int x);
857

Fabrice Bellard's avatar
Fabrice Bellard committed
858 859 860 861 862

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

863

Fabrice Bellard's avatar
Fabrice Bellard committed
864 865
/* msmpeg4.c */
void msmpeg4_encode_picture_header(MpegEncContext * s, int picture_number);
866
void msmpeg4_encode_ext_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
867 868 869 870
void msmpeg4_encode_mb(MpegEncContext * s, 
                       DCTELEM block[6][64],
                       int motion_x, int motion_y);
int msmpeg4_decode_picture_header(MpegEncContext * s);
871
int msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size);
Michael Niedermayer's avatar
Michael Niedermayer committed
872 873
int ff_msmpeg4_decode_init(MpegEncContext *s);
void ff_msmpeg4_encode_init(MpegEncContext *s);
Michael Niedermayer's avatar
Michael Niedermayer committed
874
int ff_wmv2_decode_picture_header(MpegEncContext * s);
Fabrice Bellard's avatar
Fabrice Bellard committed
875
int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
Michael Niedermayer's avatar
Michael Niedermayer committed
876 877
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,
878 879
                               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
880 881 882 883 884
                               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
885

886
/* mjpeg.c */
Fabrice Bellard's avatar
Fabrice Bellard committed
887 888 889 890 891 892
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);
893

894

895 896
/* rate control */
int ff_rate_control_init(MpegEncContext *s);
897
float ff_rate_estimate_qscale(MpegEncContext *s);
898 899
void ff_write_pass1_stats(MpegEncContext *s);
void ff_rate_control_uninit(MpegEncContext *s);
900 901
double ff_eval(char *s, double *const_value, const char **const_name,
               double (**func1)(void *, double), const char **func1_name,
902 903 904
               double (**func2)(void *, double, double), char **func2_name,
               void *opaque);

905

906
#endif /* AVCODEC_MPEGVIDEO_H */