doxy / cosmetics
[libav.git] / libavcodec / mpegvideo.h
CommitLineData
de6d9b64
FB
1/*
2 * Generic DCT based hybrid video encoder
ff4ec49e 3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
de6d9b64 4 *
ff4ec49e
FB
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
de6d9b64 9 *
ff4ec49e 10 * This library is distributed in the hope that it will be useful,
de6d9b64 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
ff4ec49e
FB
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
de6d9b64 14 *
ff4ec49e
FB
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
de6d9b64
FB
18 */
19
983e3246
MN
20/**
21 * @file mpegvideo.h
22 * mpegvideo header.
23 */
24
cd4af68a
ZK
25#ifndef AVCODEC_MPEGVIDEO_H
26#define AVCODEC_MPEGVIDEO_H
27
5c91a675
ZK
28#include "dsputil.h"
29
eec1c6b9
MN
30#define FRAME_SKIPED 100 // return value for header parsers if frame is not coded
31
de6d9b64
FB
32enum OutputFormat {
33 FMT_MPEG1,
34 FMT_H263,
9dbcbd92 35 FMT_MJPEG,
de6d9b64
FB
36};
37
1e491e29
MN
38#define EDGE_WIDTH 16
39
de6d9b64
FB
40#define MPEG_BUF_SIZE (16 * 1024)
41
ad324c93
MN
42#define QMAT_SHIFT_MMX 16
43#define QMAT_SHIFT 22
2f349de2 44
45870f57
MN
45#define MAX_FCODE 7
46#define MAX_MV 2048
1e491e29 47
0d1e9246 48#define MAX_PICTURE_COUNT 15
45870f57 49
7f2fe444
MN
50#define ME_MAP_SIZE 64
51#define ME_MAP_SHIFT 3
52#define ME_MAP_MV_BITS 11
53
0151a6f5
MN
54/* run length table */
55#define MAX_RUN 64
56#define MAX_LEVEL 64
57
140cb663
MN
58#define I_TYPE FF_I_TYPE // Intra
59#define P_TYPE FF_P_TYPE // Predicted
60#define B_TYPE FF_B_TYPE // Bi-dir predicted
61#define S_TYPE FF_S_TYPE // S(GMC)-VOP MPEG4
62
9cdd6a24
MN
63typedef struct Predictor{
64 double coeff;
65 double count;
66 double decay;
67} Predictor;
68
8b4c7dbc
MN
69typedef struct RateControlEntry{
70 int pict_type;
c5d309f2 71 float qscale;
8b4c7dbc
MN
72 int mv_bits;
73 int i_tex_bits;
74 int p_tex_bits;
75 int misc_bits;
0c1a9eda 76 uint64_t expected_bits;
8b4c7dbc
MN
77 int new_pict_type;
78 float new_qscale;
3aa102be
MN
79 int mc_mb_var_sum;
80 int mb_var_sum;
81 int i_count;
82 int f_code;
83 int b_code;
8b4c7dbc
MN
84}RateControlEntry;
85
86typedef struct RateControlContext{
87 FILE *stats_file;
3aa102be 88 int num_entries; /* number of RateControlEntries */
8b4c7dbc 89 RateControlEntry *entry;
3aa102be
MN
90 int buffer_index; /* amount of bits in the video/audio buffer */
91 Predictor pred[5];
92 double short_term_qsum; /* sum of recent qscales */
93 double short_term_qcount; /* count of recent qscales */
23e54f69 94 double pass1_rc_eq_output_sum;/* sum of the output of the rc equation, this is used for normalization */
3aa102be
MN
95 double pass1_wanted_bits; /* bits which should have been outputed by the pass1 code (including complexity init) */
96 double last_qscale;
947ac25a 97 double last_qscale_for[5]; /* last qscale for a specific pict type, used for max_diff & ipb factor stuff */
3aa102be
MN
98 int last_mc_mb_var_sum;
99 int last_mb_var_sum;
0c1a9eda
ZK
100 uint64_t i_cplx_sum[5];
101 uint64_t p_cplx_sum[5];
102 uint64_t mv_bits_sum[5];
103 uint64_t qscale_sum[5];
3aa102be 104 int frame_count[5];
947ac25a 105 int last_non_b_pict_type;
8b4c7dbc
MN
106}RateControlContext;
107
2ad1516a 108typedef struct ScanTable{
0c1a9eda
ZK
109 const uint8_t *scantable;
110 uint8_t permutated[64];
111 uint8_t raster_end[64];
05c4072b
MN
112#ifdef ARCH_POWERPC
113 /* Used by dct_quantise_alitvec to find last-non-zero */
0c1a9eda 114 uint8_t __align8 inverse[64];
05c4072b 115#endif
2ad1516a
MN
116} ScanTable;
117
1e491e29 118typedef struct Picture{
492cd3a9 119 FF_COMMON_FRAME
1e491e29
MN
120
121 int mb_var_sum; /* sum of MB variance for current frame */
122 int mc_mb_var_sum; /* motion compensated MB variance for current frame */
123 uint16_t *mb_var; /* Table for MB variances */
124 uint16_t *mc_mb_var; /* Table for motion compensated MB variances */
125 uint8_t *mb_mean; /* Table for MB luminance */
67725183 126 int32_t *mb_cmp_score; /* Table for MB cmp scores, for mb decission */
0d1e9246 127 int b_frame_score; /* */
1e491e29
MN
128} Picture;
129
d7425f59 130typedef struct ParseContext{
0c1a9eda 131 uint8_t *buffer;
d7425f59
MN
132 int index;
133 int last_index;
134 int buffer_size;
135 int state;
136 int frame_start_found;
137} ParseContext;
138
1457ab52
MN
139struct MpegEncContext;
140
141typedef struct MotionEstContext{
142 int skip; /* set if ME is skiped for the current MB */
143 int co_located_mv[4][2]; /* mv from last p frame for direct mode ME */
144 int direct_basis_mv[4][2];
145 uint8_t *scratchpad; /* data area for the me algo, so that the ME doesnt need to malloc/free */
146 uint32_t *map; /* map to avoid duplicate evaluations */
147 uint32_t *score_map; /* map to store the scores */
148 int map_generation;
826f429a 149 int pre_penalty_factor;
1457ab52
MN
150 int penalty_factor;
151 int sub_penalty_factor;
67725183 152 int mb_penalty_factor;
f931ff7b 153 int pre_pass; /* = 1 for the pre pass */
826f429a 154 int dia_size;
0c1a9eda 155 uint16_t (*mv_penalty)[MAX_MV*2+1]; /* amount of bits needed to encode a MV */
1457ab52
MN
156 int (*sub_motion_search)(struct MpegEncContext * s,
157 int *mx_ptr, int *my_ptr, int dmin,
158 int xmin, int ymin, int xmax, int ymax,
159 int pred_x, int pred_y, Picture *ref_picture,
160 int n, int size, uint16_t * const mv_penalty);
161 int (*motion_search[7])(struct MpegEncContext * s, int block,
162 int *mx_ptr, int *my_ptr,
163 int P[10][2], int pred_x, int pred_y,
826f429a
MN
164 int xmin, int ymin, int xmax, int ymax, Picture *ref_picture, int16_t (*last_mv)[2],
165 int ref_mv_scale, uint16_t * const mv_penalty);
166 int (*pre_motion_search)(struct MpegEncContext * s, int block,
167 int *mx_ptr, int *my_ptr,
168 int P[10][2], int pred_x, int pred_y,
b07a5980
MN
169 int xmin, int ymin, int xmax, int ymax, Picture *ref_picture, int16_t (*last_mv)[2],
170 int ref_mv_scale, uint16_t * const mv_penalty);
67725183
MN
171 int (*get_mb_score)(struct MpegEncContext * s, int mx, int my, int pred_x, int pred_y, Picture *ref_picture,
172 uint16_t * const mv_penalty);
1457ab52
MN
173}MotionEstContext;
174
de6d9b64 175typedef struct MpegEncContext {
58f26ba9 176 struct AVCodecContext *avctx;
de6d9b64
FB
177 /* the following parameters must be initialized before encoding */
178 int width, height; /* picture size. must be a multiple of 16 */
179 int gop_size;
180 int frame_rate; /* number of frames per second */
181 int intra_only; /* if true, only intra pictures are generated */
182 int bit_rate; /* wanted bit rate */
9cdd6a24 183 int bit_rate_tolerance; /* amount of +- bits (>0)*/
de6d9b64 184 enum OutputFormat out_format; /* output format */
d7e9533a
MN
185 int h263_pred; /* use mpeg4/h263 ac/dc predictions */
186
187/* the following codec id fields are deprecated in favor of codec_id */
de6d9b64
FB
188 int h263_plus; /* h263 plus headers */
189 int h263_rv10; /* use RV10 variation for H263 */
d7e9533a 190 int h263_msmpeg4; /* generate MSMPEG4 compatible stream (deprecated, use msmpeg4_version instead)*/
de6d9b64 191 int h263_intel; /* use I263 intel h263 header */
d7e9533a
MN
192
193 int codec_id; /* see CODEC_ID_xxx */
de6d9b64 194 int fixed_qscale; /* fixed qscale if non zero */
9cdd6a24
MN
195 float qcompress; /* amount of qscale change between easy & hard scenes (0.0-1.0) */
196 float qblur; /* amount of qscale smoothing over time (0.0-1.0) */
197 int qmin; /* min qscale */
198 int qmax; /* max qscale */
199 int max_qdiff; /* max qscale difference between frames */
3bb4e23a 200 int encoding; /* true if we are encoding (vs decoding) */
ba6802de 201 int flags; /* AVCodecContext.flags (HQ, MV4, ...) */
9dbcbd92 202 int max_b_frames; /* max number of b-frames for encoding */
8b4c7dbc 203 int b_frame_strategy;
7f2fe444
MN
204 int luma_elim_threshold;
205 int chroma_elim_threshold;
206 int strict_std_compliance; /* strictly follow the std (MPEG4, ...) */
92ba5ffb 207 int workaround_bugs; /* workaround bugs in encoders which cannot be detected automatically */
de6d9b64
FB
208 /* the following fields are managed internally by the encoder */
209
210 /* bit output */
211 PutBitContext pb;
212
213 /* sequence parameters */
214 int context_initialized;
9dbcbd92 215 int input_picture_number;
de6d9b64 216 int picture_number;
45870f57 217 int picture_in_gop_number; /* 0-> first pic in gop, ... */
9dbcbd92
MN
218 int b_frames_since_non_b; /* used for encoding, relative to not yet reordered input */
219 int mb_width, mb_height; /* number of MBs horizontally & vertically */
b5a093b3 220 int h_edge_pos, v_edge_pos;/* horizontal / vertical position of the right/bottom edge (pixel replicateion)*/
e03c341e 221 int mb_num; /* number of MBs of a picture */
de6d9b64 222 int linesize; /* line size, in bytes, may be different from width */
0fd90455 223 int uvlinesize; /* line size, for chroma in bytes, may be different from width */
1e491e29
MN
224 Picture picture[MAX_PICTURE_COUNT]; /* main picture buffer */
225 Picture *input_picture[MAX_PICTURE_COUNT]; /* next pictures on display order for encoding*/
226 Picture *reordered_input_picture[MAX_PICTURE_COUNT]; /* pointer to the next pictures in codedorder for encoding*/
227 Picture last_picture; /* previous picture */
228 Picture next_picture; /* previous picture (for bidir pred) */
229 Picture new_picture; /* source picture for encoding */
230 Picture current_picture; /* buffer to store the decompressed current picture */
9dbcbd92 231 int last_dc[3]; /* last DC values for MPEG1 */
0c1a9eda 232 int16_t *dc_val[3]; /* used for mpeg4 DC prediction, all 3 arrays must be continuous */
de6d9b64 233 int y_dc_scale, c_dc_scale;
0c1a9eda
ZK
234 uint8_t *y_dc_scale_table; /* qscale -> y_dc_scale table */
235 uint8_t *c_dc_scale_table; /* qscale -> c_dc_scale table */
236 uint8_t *coded_block; /* used for coded block pattern prediction (msmpeg4v3, wmv1)*/
237 int16_t (*ac_val[3])[16]; /* used for for mpeg4 AC prediction, all 3 arrays must be continuous */
de6d9b64 238 int ac_pred;
f943e138
MN
239 uint8_t *prev_pict_types; /* previous picture types in bitstream order, used for mb skip */
240#define PREV_PICT_TYPES_BUFFER_SIZE 256
3bb4e23a 241 int mb_skiped; /* MUST BE SET only during DECODING */
0c1a9eda 242 uint8_t *mbskip_table; /* used to avoid copy if macroblock skipped (for black regions for example)
9dbcbd92 243 and used for b-frame encoding & decoding (contains skip table of next P Frame) */
0c1a9eda
ZK
244 uint8_t *mbintra_table; /* used to avoid setting {ac, dc, cbp}-pred stuff to zero on inter MB decoding */
245 uint8_t *cbp_table; /* used to store cbp, ac_pred for partitioned decoding */
246 uint8_t *pred_dir_table; /* used to store pred_dir for partitioned decoding */
f7b47594
MN
247 uint8_t *allocated_edge_emu_buffer;
248 uint8_t *edge_emu_buffer; /* points into the middle of allocated_edge_emu_buffer */
de6d9b64 249
9dbcbd92 250 int qscale; /* QP */
1e491e29 251 float frame_qscale; /* qscale from the frame level rc FIXME remove*/
c5d309f2
MN
252 int adaptive_quant; /* use adaptive quantization */
253 int dquant; /* qscale difference to prev qscale */
9dbcbd92 254 int pict_type; /* I_TYPE, P_TYPE, B_TYPE, ... */
3aa102be 255 int last_pict_type;
8b4c7dbc 256 int last_non_b_pict_type; /* used for mpeg4 gmc b-frames & ratecontrol */
de6d9b64
FB
257 int frame_rate_index;
258 /* motion compensation */
e51d6d27 259 int unrestricted_mv; /* mv can point outside of the coded picture */
de6d9b64
FB
260 int h263_long_vectors; /* use horrible h263v1 long vector mode */
261
eb4b3dd3 262 DSPContext dsp; /* pointers for accelerated dsp fucntions */
9dbcbd92
MN
263 int f_code; /* forward MV resolution */
264 int b_code; /* backward MV resolution for B Frames (mpeg4) */
0c1a9eda
ZK
265 int16_t (*motion_val)[2]; /* used for MV prediction (4MV per MB) */
266 int16_t (*p_mv_table)[2]; /* MV table (1MV per MB) p-frame encoding */
267 int16_t (*b_forw_mv_table)[2]; /* MV table (1MV per MB) forward mode b-frame encoding */
268 int16_t (*b_back_mv_table)[2]; /* MV table (1MV per MB) backward mode b-frame encoding */
269 int16_t (*b_bidir_forw_mv_table)[2]; /* MV table (1MV per MB) bidir mode b-frame encoding */
270 int16_t (*b_bidir_back_mv_table)[2]; /* MV table (1MV per MB) bidir mode b-frame encoding */
271 int16_t (*b_direct_mv_table)[2]; /* MV table (1MV per MB) direct mode b-frame encoding */
91029be7 272 int me_method; /* ME algorithm */
1fb4890b 273 int scene_change_score;
de6d9b64
FB
274 int mv_dir;
275#define MV_DIR_BACKWARD 1
276#define MV_DIR_FORWARD 2
6f91bcd1 277#define MV_DIRECT 4 // bidirectional mode where the difference equals the MV of the last P/S/I-Frame (mpeg4)
de6d9b64
FB
278 int mv_type;
279#define MV_TYPE_16X16 0 /* 1 vector for the whole mb */
45870f57 280#define MV_TYPE_8X8 1 /* 4 vectors (h263, mpeg4 4MV) */
de6d9b64
FB
281#define MV_TYPE_16X8 2 /* 2 vectors, one per 16x8 block */
282#define MV_TYPE_FIELD 3 /* 2 vectors, one per field */
283#define MV_TYPE_DMV 4 /* 2 vectors, special mpeg2 Dual Prime Vectors */
284 /* motion vectors for a macroblock
285 first coordinate : 0 = forward 1 = backward
286 second " : depend on type
287 third " : 0 = x, 1 = y
288 */
289 int mv[2][4][2];
290 int field_select[2][2];
9dbcbd92 291 int last_mv[2][2][2]; /* last MV, used for MV prediction in MPEG1 & B-frame MPEG4 */
0c1a9eda 292 uint8_t *fcode_tab; /* smallest fcode needed for each MV */
1457ab52
MN
293
294 MotionEstContext me;
de6d9b64 295
91029be7
MN
296 int no_rounding; /* apply no rounding to motion compensation (MPEG4, msmpeg4, ...)
297 for b-frames rounding mode is allways 0 */
de6d9b64 298
2417652e
MN
299 int hurry_up; /* when set to 1 during decoding, b frames will be skiped
300 when set to 2 idct/dequant will be skipped too */
c5d309f2 301
de6d9b64
FB
302 /* macroblock layer */
303 int mb_x, mb_y;
304 int mb_incr;
305 int mb_intra;
0c1a9eda 306 uint8_t *mb_type; /* Table for MB type */
ba6802de
MN
307#define MB_TYPE_INTRA 0x01
308#define MB_TYPE_INTER 0x02
309#define MB_TYPE_INTER4V 0x04
310#define MB_TYPE_SKIPED 0x08
7f2fe444
MN
311#define MB_TYPE_GMC 0x10
312
ba6802de
MN
313#define MB_TYPE_DIRECT 0x10
314#define MB_TYPE_FORWARD 0x20
9dbcbd92 315#define MB_TYPE_BACKWARD 0x40
ba6802de 316#define MB_TYPE_BIDIR 0x80
4278e7a6 317
9dbcbd92 318 int block_index[6]; /* index to current MB in block based arrays with edges*/
4278e7a6
MN
319 int block_wrap[6];
320
de6d9b64 321 /* matrix transmitted in the bitstream */
0c1a9eda
ZK
322 uint16_t intra_matrix[64];
323 uint16_t chroma_intra_matrix[64];
324 uint16_t inter_matrix[64];
325 uint16_t chroma_inter_matrix[64];
d7e9533a
MN
326#define QUANT_BIAS_SHIFT 4
327 int intra_quant_bias; /* bias for the quantizer */
328 int inter_quant_bias; /* bias for the quantizer */
329 int min_qcoeff; /* minimum encodable coefficient */
330 int max_qcoeff; /* maximum encodable coefficient */
477ab036
MN
331 int ac_esc_length; /* num of bits needed to encode the longest esc */
332 uint8_t *intra_ac_vlc_length;
333 uint8_t *intra_ac_vlc_last_length;
334 uint8_t *inter_ac_vlc_length;
335 uint8_t *inter_ac_vlc_last_length;
67725183
MN
336 uint8_t *luma_dc_vlc_length;
337 uint8_t *chroma_dc_vlc_length;
c442d75c 338#define UNI_AC_ENC_INDEX(run,level) ((run)*128 + (level))
477ab036 339
de6d9b64 340 /* precomputed matrix (combine qscale and DCT renorm) */
05c4072b
MN
341 int __align8 q_intra_matrix[32][64];
342 int __align8 q_inter_matrix[32][64];
2f349de2 343 /* identical to the above but for MMX & these are not permutated */
0c1a9eda
ZK
344 uint16_t __align8 q_intra_matrix16[32][64];
345 uint16_t __align8 q_inter_matrix16[32][64];
346 uint16_t __align8 q_intra_matrix16_bias[32][64];
347 uint16_t __align8 q_inter_matrix16_bias[32][64];
de6d9b64 348 int block_last_index[6]; /* last non zero coefficient in block */
2ad1516a 349 /* scantables */
05c4072b 350 ScanTable __align8 intra_scantable;
2ad1516a
MN
351 ScanTable intra_h_scantable;
352 ScanTable intra_v_scantable;
353 ScanTable inter_scantable; // if inter == intra then intra should be used to reduce tha cache usage
de6d9b64
FB
354
355 void *opaque; /* private data for the user */
356
357 /* bit rate control */
9dbcbd92 358 int I_frame_bits; //FIXME used in mpeg12 ...
0c1a9eda
ZK
359 int64_t wanted_bits;
360 int64_t total_bits;
8b4c7dbc 361 int frame_bits; /* bits used for the current frame */
3aa102be 362 RateControlContext rc_context; // contains stuff only accessed in ratecontrol.c
9cdd6a24 363
098eefe1
MN
364 /* statistics, used for 2-pass encoding */
365 int mv_bits;
366 int header_bits;
367 int i_tex_bits;
368 int p_tex_bits;
369 int i_count;
66370d3f
MN
370 int f_count;
371 int b_count;
098eefe1
MN
372 int skip_count;
373 int misc_bits; // cbp, mb_type
374 int last_bits; //temp var used for calculating the above vars
7f2fe444
MN
375
376 /* error concealment / resync */
0c1a9eda 377 uint8_t *error_status_table; /* table of the error status of each MB */
4d2858de
MN
378#define VP_START 1 /* current MB is the first after a resync marker */
379#define AC_ERROR 2
380#define DC_ERROR 4
381#define MV_ERROR 8
382#define AC_END 16
383#define DC_END 32
384#define MV_END 64
385//FIXME some prefix?
386
7f2fe444
MN
387 int resync_mb_x; /* x position of last resync marker */
388 int resync_mb_y; /* y position of last resync marker */
b07a5980 389 GetBitContext last_resync_gb; /* used to search for the next resync marker */
4d2858de 390 int mb_num_left; /* number of MBs left in this video packet (for partitioned Slices only)*/
7f2fe444
MN
391 int next_p_frame_damaged; /* set if the next p frame is damaged, to avoid showing trashed b frames */
392 int error_resilience;
d7425f59
MN
393
394 ParseContext parse_context;
098eefe1 395
102d3908
J
396 /* H.263 specific */
397 int gob_number;
644d98a4 398 int gob_index;
d140623f 399
544286b3 400 /* H.263+ specific */
e51d6d27 401 int umvplus; /* == H263+ && unrestricted_mv */
d140623f
J
402 int h263_aic; /* Advanded INTRA Coding (AIC) */
403 int h263_aic_dir; /* AIC direction: 0 = left, 1 = top */
544286b3 404
de6d9b64 405 /* mpeg4 specific */
6f91bcd1 406 int time_increment_resolution;
9dbcbd92
MN
407 int time_increment_bits; /* number of bits to represent the fractional part of time */
408 int last_time_base;
409 int time_base; /* time in seconds of last I,P,S Frame */
0c1a9eda
ZK
410 int64_t time; /* time of current frame */
411 int64_t last_non_b_time;
412 uint16_t pp_time; /* time distance between the last 2 p,s,i frames */
413 uint16_t pb_time; /* time distance between the last b and p,s,i frame */
414 uint16_t pp_field_time;
415 uint16_t pb_field_time; /* like above, just for interlaced */
1a565432
FB
416 int shape;
417 int vol_sprite_usage;
cc9ba006
MN
418 int sprite_width;
419 int sprite_height;
420 int sprite_left;
421 int sprite_top;
422 int sprite_brightness_change;
73c8e514 423 int num_sprite_warping_points;
44eb4951 424 int real_sprite_warping_points;
073b013d
MN
425 int sprite_offset[2][2]; /* sprite offset[isChroma][isMVY] */
426 int sprite_delta[2][2]; /* sprite_delta [isY][isMVY] */
427 int sprite_shift[2]; /* sprite shift [isChroma] */
73c8e514 428 int mcsel;
1a565432 429 int quant_precision;
9dbcbd92 430 int quarter_sample; /* 1->qpel, 0->half pel ME/MC */
cc9ba006 431 int scalability;
0fd90455
MN
432 int hierachy_type;
433 int enhancement_type;
cc9ba006
MN
434 int new_pred;
435 int reduced_res_vop;
436 int aspect_ratio_info;
e769f053
AB
437 int aspected_width;
438 int aspected_height;
cc9ba006
MN
439 int sprite_warping_accuracy;
440 int low_latency_sprite;
4d2858de
MN
441 int data_partitioning; /* data partitioning flag from header */
442 int partitioned_frame; /* is current frame partitioned */
7f2fe444
MN
443 int rvlc; /* reversible vlc */
444 int resync_marker; /* could this stream contain resync markers*/
1ff662cc
MN
445 int low_delay; /* no reordering needed / has no b-frames */
446 int vo_type;
7aaf3b98 447 int vol_control_parameters; /* does the stream contain the low_delay flag, used to workaround buggy encoders */
7f2fe444
MN
448 PutBitContext tex_pb; /* used for data partitioned VOPs */
449 PutBitContext pb2; /* used for data partitioned VOPs */
450#define PB_BUFFER_SIZE 1024*256
451 uint8_t *tex_pb_buffer;
452 uint8_t *pb2_buffer;
d930ef19 453 int mpeg_quant;
bea669e5
MN
454#define CO_LOCATED_TYPE_4MV 1
455#define CO_LOCATED_TYPE_FIELDMV 2
0c1a9eda
ZK
456 int8_t *co_located_type_table; /* 4mv & field_mv info for next b frame */
457 int16_t (*field_mv_table)[2][2]; /* used for interlaced b frame decoding */
458 int8_t (*field_select_table)[2]; /* wtf, no really another table for interlaced b frames */
bea669e5 459 int t_frame; /* time distance of first I -> B, used for interlaced b frames */
a61f8265 460 int padding_bug_score; /* used to detect the VERY common padding bug in MPEG4 */
44eb4951
MN
461
462 /* divx specific, used to workaround (many) bugs in divx5 */
463 int divx_version;
464 int divx_build;
eec1c6b9 465#define BITSTREAM_BUFFER_SIZE 1024*256
0c1a9eda 466 uint8_t *bitstream_buffer; //Divx 5.01 puts several frames in a single one, this is used to reorder them
eec1c6b9
MN
467 int bitstream_buffer_size;
468
7da71a5c
MN
469 int xvid_build;
470
4d2858de
MN
471 /* lavc specific stuff, used to workaround bugs in libavcodec */
472 int ffmpeg_version;
473 int lavc_build;
474
de6d9b64
FB
475 /* RV10 specific */
476 int rv10_version; /* RV10 version: 0 or 3 */
477 int rv10_first_dc_coded[3];
478
479 /* MJPEG specific */
480 struct MJpegContext *mjpeg_ctx;
80e103d0
A
481 int mjpeg_vsample[3]; /* vertical sampling factors, default = {2, 1, 1} */
482 int mjpeg_hsample[3]; /* horizontal sampling factors, default = {2, 1, 1} */
483 int mjpeg_write_tables; /* do we want to have quantisation- and
484 huffmantables in the jpeg file ? */
a69b930c 485 int mjpeg_data_only_frames; /* frames only with SOI, SOS and EOI markers */
de6d9b64
FB
486
487 /* MSMPEG4 specific */
488 int mv_table_index;
489 int rl_table_index;
490 int rl_chroma_table_index;
491 int dc_table_index;
492 int use_skip_mb_code;
493 int slice_height; /* in macroblocks */
49092244 494 int first_slice_line; /* used in mpeg4 too to handle resync markers */
ae40484c 495 int flipflop_rounding;
0151a6f5
MN
496 int msmpeg4_version; /* 0=not msmpeg4, 1=mp41, 2=mp42, 3=mp43/divx3 4=wmv1/7 5=wmv2/8*/
497 int per_mb_rl_table;
498 int esc3_level_length;
499 int esc3_run_length;
0151a6f5 500 /* [mb_intra][isChroma][level][run][last] */
6b460aa3 501 int (*ac_stats)[2][MAX_LEVEL+1][MAX_RUN+1][2];
de0f2f4c 502 int inter_intra_pred;
1457ab52 503 int mspel;
0151a6f5 504
de6d9b64
FB
505 /* decompression specific */
506 GetBitContext gb;
507
1e491e29
MN
508 /* Mpeg1 specific */
509 int fake_picture_number; /* picture number at the bitstream frame rate */
510 int gop_picture_number; /* index of the first picture of a GOP based on fake_pic_num & mpeg1 specific */
63b15e55 511 int last_mv_dir; /* last mv_dir, used for b frame encoding */
1e491e29 512
de6d9b64
FB
513 /* MPEG2 specific - I wish I had not to support this mess. */
514 int progressive_sequence;
515 int mpeg_f_code[2][2];
516 int picture_structure;
517/* picture type */
518#define PICT_TOP_FIELD 1
519#define PICT_BOTTOM_FIELD 2
520#define PICT_FRAME 3
521
522 int intra_dc_precision;
523 int frame_pred_frame_dct;
524 int top_field_first;
525 int concealment_motion_vectors;
526 int q_scale_type;
527 int intra_vlc_format;
528 int alternate_scan;
529 int repeat_first_field;
530 int chroma_420_type;
531 int progressive_frame;
532 int mpeg2;
533 int full_pel[2];
534 int interlaced_dct;
de6d9b64 535 int first_slice;
dfb476cb 536 int first_field;
644d98a4
J
537
538 /* RTP specific */
81401c1f 539 /* These are explained on avcodec.h */
644d98a4
J
540 int rtp_mode;
541 int rtp_payload_size;
81401c1f 542 void (*rtp_callback)(void *data, int size, int packet_number);
0c1a9eda
ZK
543 uint8_t *ptr_lastgob;
544 uint8_t *ptr_last_mb_line;
545 uint32_t mb_line_avgsize;
644d98a4 546
ba6802de 547 DCTELEM (*block)[64]; /* points to one of the following blocks */
91029be7 548 DCTELEM blocks[2][6][64] __align8; // for HQ mode we need to keep the best block
4d2858de
MN
549 int (*decode_mb)(struct MpegEncContext *s, DCTELEM block[6][64]); // used by some codecs to avoid a switch()
550#define SLICE_OK 0
551#define SLICE_ERROR -1
552#define SLICE_END -2 //end marker found
553#define SLICE_NOEND -3 //no end marker or error found but mb count exceeded
554
9dbf1ddd 555 void (*dct_unquantize_mpeg1)(struct MpegEncContext *s,
67309e49 556 DCTELEM *block/*align 16*/, int n, int qscale);
9dbf1ddd 557 void (*dct_unquantize_mpeg2)(struct MpegEncContext *s,
67309e49 558 DCTELEM *block/*align 16*/, int n, int qscale);
3bf43d42 559 void (*dct_unquantize_h263)(struct MpegEncContext *s,
67309e49 560 DCTELEM *block/*align 16*/, int n, int qscale);
3bf43d42 561 void (*dct_unquantize)(struct MpegEncContext *s, // unquantizer to use (mpeg4 can use both)
67309e49
MN
562 DCTELEM *block/*align 16*/, int n, int qscale);
563 int (*dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
3a87ac94 564 int (*fast_dct_quantize)(struct MpegEncContext *s, DCTELEM *block/*align 16*/, int n, int qscale, int *overflow);
de6d9b64
FB
565} MpegEncContext;
566
4d2858de 567
defdfc9a 568int DCT_common_init(MpegEncContext *s);
de6d9b64
FB
569int MPV_common_init(MpegEncContext *s);
570void MPV_common_end(MpegEncContext *s);
571void MPV_decode_mb(MpegEncContext *s, DCTELEM block[6][64]);
d6db1c9c 572int MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx);
de6d9b64 573void MPV_frame_end(MpegEncContext *s);
1457ab52
MN
574int MPV_encode_init(AVCodecContext *avctx);
575int MPV_encode_end(AVCodecContext *avctx);
576int MPV_encode_picture(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data);
21af69f7
FB
577#ifdef HAVE_MMX
578void MPV_common_init_mmx(MpegEncContext *s);
579#endif
e0580f8c
FH
580#ifdef ARCH_ALPHA
581void MPV_common_init_axp(MpegEncContext *s);
582#endif
c7e07931
MO
583#ifdef HAVE_MLIB
584void MPV_common_init_mlib(MpegEncContext *s);
585#endif
5917d17c
LS
586#ifdef HAVE_MMI
587void MPV_common_init_mmi(MpegEncContext *s);
588#endif
83f238cb 589#ifdef ARCH_ARMV4L
d43b10ec 590void MPV_common_init_armv4l(MpegEncContext *s);
83f238cb 591#endif
05c4072b
MN
592#ifdef ARCH_POWERPC
593void MPV_common_init_ppc(MpegEncContext *s);
594#endif
0c1a9eda 595extern void (*draw_edges)(uint8_t *buf, int wrap, int width, int height, int w);
7f2fe444 596void ff_conceal_past_errors(MpegEncContext *s, int conceal_all);
0c1a9eda 597void ff_copy_bits(PutBitContext *pb, uint8_t *src, int length);
7f2fe444 598void ff_clean_intra_table_entries(MpegEncContext *s);
0c1a9eda 599void ff_init_scantable(MpegEncContext *s, ScanTable *st, const uint8_t *src_scantable);
4d2858de 600void ff_error_resilience(MpegEncContext *s);
640950c7 601void ff_draw_horiz_band(MpegEncContext *s, int y, int h);
0c1a9eda 602void ff_emulated_edge_mc(MpegEncContext *s, uint8_t *src, int linesize, int block_w, int block_h,
1457ab52 603 int src_x, int src_y, int w, int h);
59b571c1 604char ff_get_pict_type_char(int pict_type);
b7a22d84 605int ff_combine_frame( MpegEncContext *s, int next, uint8_t **buf, int *buf_size);
59b571c1 606
a33c7159 607extern enum PixelFormat ff_yuv420p_list[2];
de6d9b64 608
4d2858de
MN
609static inline void ff_init_block_index(MpegEncContext *s){
610 s->block_index[0]= s->block_wrap[0]*(s->mb_y*2 + 1) - 1 + s->mb_x*2;
611 s->block_index[1]= s->block_wrap[0]*(s->mb_y*2 + 1) + s->mb_x*2;
612 s->block_index[2]= s->block_wrap[0]*(s->mb_y*2 + 2) - 1 + s->mb_x*2;
613 s->block_index[3]= s->block_wrap[0]*(s->mb_y*2 + 2) + s->mb_x*2;
614 s->block_index[4]= s->block_wrap[4]*(s->mb_y + 1) + s->block_wrap[0]*(s->mb_height*2 + 2) + s->mb_x;
615 s->block_index[5]= s->block_wrap[4]*(s->mb_y + 1 + s->mb_height + 2) + s->block_wrap[0]*(s->mb_height*2 + 2) + s->mb_x;
616}
617
618static inline void ff_update_block_index(MpegEncContext *s){
619 s->block_index[0]+=2;
620 s->block_index[1]+=2;
621 s->block_index[2]+=2;
622 s->block_index[3]+=2;
623 s->block_index[4]++;
624 s->block_index[5]++;
625}
626
627
de6d9b64 628/* motion_est.c */
9dbcbd92
MN
629void ff_estimate_p_frame_motion(MpegEncContext * s,
630 int mb_x, int mb_y);
631void ff_estimate_b_frame_motion(MpegEncContext * s,
632 int mb_x, int mb_y);
633int ff_get_best_fcode(MpegEncContext * s, int16_t (*mv_table)[2], int type);
634void ff_fix_long_p_mvs(MpegEncContext * s);
635void ff_fix_long_b_mvs(MpegEncContext * s, int16_t (*mv_table)[2], int f_code, int type);
1457ab52 636void ff_init_me(MpegEncContext *s);
f5fb6b34 637int ff_pre_estimate_p_frame_motion(MpegEncContext * s, int mb_x, int mb_y);
de6d9b64 638
4d2858de 639
de6d9b64 640/* mpeg12.c */
0c1a9eda
ZK
641extern const int16_t ff_mpeg1_default_intra_matrix[64];
642extern const int16_t ff_mpeg1_default_non_intra_matrix[64];
643extern uint8_t ff_mpeg1_dc_scale_table[128];
de6d9b64
FB
644
645void mpeg1_encode_picture_header(MpegEncContext *s, int picture_number);
646void mpeg1_encode_mb(MpegEncContext *s,
647 DCTELEM block[6][64],
648 int motion_x, int motion_y);
0151a6f5 649void ff_mpeg1_encode_init(MpegEncContext *s);
de6d9b64 650
4d2858de 651
de6d9b64 652/* h263enc.c */
de6d9b64
FB
653typedef struct RLTable {
654 int n; /* number of entries of table_vlc minus 1 */
655 int last; /* number of values for last = 0 */
0c1a9eda
ZK
656 const uint16_t (*table_vlc)[2];
657 const int8_t *table_run;
658 const int8_t *table_level;
659 uint8_t *index_run[2]; /* encoding only */
660 int8_t *max_level[2]; /* encoding & decoding */
661 int8_t *max_run[2]; /* encoding & decoding */
45a82edb
MN
662 VLC vlc; /* decoding only deprected FIXME remove*/
663 RL_VLC_ELEM *rl_vlc[32]; /* decoding only */
de6d9b64
FB
664} RLTable;
665
666void init_rl(RLTable *rl);
667void init_vlc_rl(RLTable *rl);
668
3232bc88 669static inline int get_rl_index(const RLTable *rl, int last, int run, int level)
de6d9b64
FB
670{
671 int index;
672 index = rl->index_run[last][run];
673 if (index >= rl->n)
674 return rl->n;
675 if (level > rl->max_level[last][run])
676 return rl->n;
677 return index + level - 1;
678}
679
0c1a9eda
ZK
680extern uint8_t ff_mpeg4_y_dc_scale_table[32];
681extern uint8_t ff_mpeg4_c_dc_scale_table[32];
682extern const int16_t ff_mpeg4_default_intra_matrix[64];
683extern const int16_t ff_mpeg4_default_non_intra_matrix[64];
1457ab52
MN
684int ff_h263_decode_init(AVCodecContext *avctx);
685int ff_h263_decode_frame(AVCodecContext *avctx,
686 void *data, int *data_size,
0c1a9eda 687 uint8_t *buf, int buf_size);
1457ab52 688int ff_h263_decode_end(AVCodecContext *avctx);
de6d9b64
FB
689void h263_encode_mb(MpegEncContext *s,
690 DCTELEM block[6][64],
691 int motion_x, int motion_y);
d6231b9e
MN
692void mpeg4_encode_mb(MpegEncContext *s,
693 DCTELEM block[6][64],
694 int motion_x, int motion_y);
de6d9b64 695void h263_encode_picture_header(MpegEncContext *s, int picture_number);
644d98a4 696int h263_encode_gob_header(MpegEncContext * s, int mb_line);
0c1a9eda 697int16_t *h263_pred_motion(MpegEncContext * s, int block,
de6d9b64 698 int *px, int *py);
0e15384d 699void mpeg4_pred_ac(MpegEncContext * s, DCTELEM *block, int n,
de6d9b64 700 int dir);
91029be7 701void ff_set_mpeg4_time(MpegEncContext * s, int picture_number);
de6d9b64 702void mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
45870f57 703void h263_encode_init(MpegEncContext *s);
de6d9b64
FB
704void h263_decode_init_vlc(MpegEncContext *s);
705int h263_decode_picture_header(MpegEncContext *s);
4d2858de 706int ff_h263_decode_gob_header(MpegEncContext *s);
b704e742
MN
707int ff_mpeg4_decode_picture_header(MpegEncContext * s, GetBitContext *gb);
708
709
de6d9b64 710int intel_h263_decode_picture_header(MpegEncContext *s);
4d2858de
MN
711int ff_h263_decode_mb(MpegEncContext *s,
712 DCTELEM block[6][64]);
de6d9b64 713int h263_get_picture_format(int width, int height);
7f2fe444
MN
714void ff_mpeg4_encode_video_packet_header(MpegEncContext *s);
715void ff_mpeg4_clean_buffers(MpegEncContext *s);
716void ff_mpeg4_stuffing(PutBitContext * pbc);
717void ff_mpeg4_init_partitions(MpegEncContext *s);
718void ff_mpeg4_merge_partitions(MpegEncContext *s);
c5d309f2 719void ff_clean_mpeg4_qscales(MpegEncContext *s);
9e4c6948 720void ff_clean_h263_qscales(MpegEncContext *s);
4d2858de
MN
721int ff_mpeg4_decode_partitions(MpegEncContext *s);
722int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s);
723int ff_h263_resync(MpegEncContext *s);
724int ff_h263_get_gob_height(MpegEncContext *s);
1457ab52 725void ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my);
67725183 726inline int ff_h263_round_chroma(int x);
4d2858de 727
de6d9b64
FB
728
729/* rv10.c */
730void rv10_encode_picture_header(MpegEncContext *s, int picture_number);
731int rv_decode_dc(MpegEncContext *s, int n);
732
4d2858de 733
de6d9b64
FB
734/* msmpeg4.c */
735void msmpeg4_encode_picture_header(MpegEncContext * s, int picture_number);
ae40484c 736void msmpeg4_encode_ext_header(MpegEncContext * s);
de6d9b64
FB
737void msmpeg4_encode_mb(MpegEncContext * s,
738 DCTELEM block[6][64],
739 int motion_x, int motion_y);
de6d9b64 740int msmpeg4_decode_picture_header(MpegEncContext * s);
ae40484c 741int msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size);
0151a6f5
MN
742int ff_msmpeg4_decode_init(MpegEncContext *s);
743void ff_msmpeg4_encode_init(MpegEncContext *s);
1457ab52
MN
744int ff_wmv2_decode_picture_header(MpegEncContext * s);
745void ff_wmv2_add_mb(MpegEncContext *s, DCTELEM block[6][64], uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
746void ff_mspel_motion(MpegEncContext *s,
0c1a9eda
ZK
747 uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
748 uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
1457ab52
MN
749 int motion_x, int motion_y, int h);
750int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
751void ff_wmv2_encode_mb(MpegEncContext * s,
752 DCTELEM block[6][64],
753 int motion_x, int motion_y);
de6d9b64 754
4d2858de 755/* mjpegenc.c */
de6d9b64
FB
756int mjpeg_init(MpegEncContext *s);
757void mjpeg_close(MpegEncContext *s);
758void mjpeg_encode_mb(MpegEncContext *s,
759 DCTELEM block[6][64]);
760void mjpeg_picture_header(MpegEncContext *s);
761void mjpeg_picture_trailer(MpegEncContext *s);
8b4c7dbc 762
4d2858de 763
8b4c7dbc
MN
764/* rate control */
765int ff_rate_control_init(MpegEncContext *s);
c5d309f2 766float ff_rate_estimate_qscale(MpegEncContext *s);
8b4c7dbc
MN
767void ff_write_pass1_stats(MpegEncContext *s);
768void ff_rate_control_uninit(MpegEncContext *s);
5c91a675
ZK
769double ff_eval(char *s, double *const_value, const char **const_name,
770 double (**func1)(void *, double), const char **func1_name,
3aa102be
MN
771 double (**func2)(void *, double, double), char **func2_name,
772 void *opaque);
773
8b4c7dbc 774
cd4af68a 775#endif /* AVCODEC_MPEGVIDEO_H */