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