]> git.sesse.net Git - ffmpeg/blob - libavcodec/avcodec.h
new motion estimation (epzs) not complete yet but allready pretty good :)
[ffmpeg] / libavcodec / avcodec.h
1 #ifndef AVCODEC_H
2 #define AVCODEC_H
3
4 #include "common.h"
5
6 enum CodecID {
7     CODEC_ID_NONE, 
8     CODEC_ID_MPEG1VIDEO,
9     CODEC_ID_H263,
10     CODEC_ID_RV10,
11     CODEC_ID_MP2,
12     CODEC_ID_MP3LAME,
13     CODEC_ID_AC3,
14     CODEC_ID_MJPEG,
15     CODEC_ID_MPEG4,
16     CODEC_ID_RAWVIDEO,
17     CODEC_ID_MSMPEG4,
18     CODEC_ID_H263P,
19     CODEC_ID_H263I,
20
21     /* various pcm "codecs" */
22     CODEC_ID_PCM_S16LE,
23     CODEC_ID_PCM_S16BE,
24     CODEC_ID_PCM_U16LE,
25     CODEC_ID_PCM_U16BE,
26     CODEC_ID_PCM_S8,
27     CODEC_ID_PCM_U8,
28     CODEC_ID_PCM_MULAW,
29     CODEC_ID_PCM_ALAW,
30 };
31
32 enum CodecType {
33     CODEC_TYPE_VIDEO,
34     CODEC_TYPE_AUDIO,
35 };
36
37 enum PixelFormat {
38     PIX_FMT_YUV420P,
39     PIX_FMT_YUV422,
40     PIX_FMT_RGB24,
41     PIX_FMT_BGR24,
42     PIX_FMT_YUV422P,
43     PIX_FMT_YUV444P,
44 };
45
46 /* currently unused, may be used if 24/32 bits samples ever supported */
47 enum SampleFormat {
48     SAMPLE_FMT_S16 = 0,         /* signed 16 bits */
49 };
50
51 /* in bytes */
52 #define AVCODEC_MAX_AUDIO_FRAME_SIZE 18432
53
54 /* motion estimation type */
55 extern int motion_estimation_method;
56 #define ME_ZERO   0
57 #define ME_FULL   1
58 #define ME_LOG    2
59 #define ME_PHODS  3
60 #define ME_EPZS   4
61 #define ME_X1     5
62
63 /* encoding support */
64
65 #define CODEC_FLAG_HQ     0x0001 /* high quality (non real time) encoding */
66 #define CODEC_FLAG_QSCALE 0x0002 /* use fixed qscale */
67
68 /* codec capabilities */
69
70 /* decoder can use draw_horiz_band callback */
71 #define CODEC_CAP_DRAW_HORIZ_BAND 0x0001
72
73 #define FRAME_RATE_BASE 10000
74
75 typedef struct AVCodecContext {
76     int bit_rate;
77     int bit_rate_tolerance; /* amount of +- bits (>0)*/
78     int flags;
79     int sub_id;    /* some codecs needs additionnal format info. It is
80                       stored there */
81     /* video only */
82     int frame_rate; /* frames per sec multiplied by FRAME_RATE_BASE */
83     int width, height;
84     int gop_size; /* 0 = intra only */
85     int pix_fmt;  /* pixel format, see PIX_FMT_xxx */
86
87     /* if non NULL, 'draw_horiz_band' is called by the libavcodec
88        decoder to draw an horizontal band. It improve cache usage. Not
89        all codecs can do that. You must check the codec capabilities
90        before */
91     void (*draw_horiz_band)(struct AVCodecContext *s,
92                             UINT8 **src_ptr, int linesize,
93                             int y, int width, int height);
94
95     /* audio only */
96     int sample_rate; /* samples per sec */
97     int channels;
98     int sample_fmt;  /* sample format, currenly unused */
99
100     /* the following data should not be initialized */
101     int frame_size; /* in samples, initialized when calling 'init' */
102     int frame_number; /* audio or video frame number */
103     int key_frame;    /* true if the previous compressed frame was 
104                          a key frame (intra, or seekable) */
105     int quality;      /* quality of the previous encoded frame 
106                          (between 1 (good) and 31 (bad)) */
107     float qcompress;  /* amount of qscale change between easy & hard scenes (0.0-1.0)*/
108     float qblur;      /* amount of qscale smoothing over time (0.0-1.0) */
109     int qmin;         /* min qscale */
110     int qmax;         /* max qscale */
111     int max_qdiff;    /* max qscale difference between frames */
112
113     struct AVCodec *codec;
114     void *priv_data;
115
116     /* The following data is for RTP friendly coding */
117     /* By now only H.263/H.263+ coder honours this   */
118     int rtp_mode;   /* 1 for activate RTP friendly-mode           */
119                     /* highers numbers represent more error-prone */
120                     /* enviroments, by now just "1" exist         */
121     
122     int rtp_payload_size;   /* The size of the RTP payload, the coder will  */
123                             /* do it's best to deliver a chunk with size    */
124                             /* below rtp_payload_size, the chunk will start */
125                             /* with a start code on some codecs like H.263  */
126                             /* This doesn't take account of any particular  */
127                             /* headers inside the transmited RTP payload    */
128
129     
130     /* The RTP callcack: This function is called  */
131     /* every time the encoder as a packet to send */
132     /* Depends on the encoder if the data starts  */
133     /* with a Start Code (it should) H.263 does   */
134     void (*rtp_callback)(void *data, int size, int packet_number); 
135
136     /* These are for PSNR calculation, if you set get_psnr to 1 */
137     /* after encoding you will have the PSNR on psnr_y/cb/cr    */
138     int get_psnr;
139     float psnr_y;
140     float psnr_cb;
141     float psnr_cr;
142                  
143     /* the following fields are ignored */
144     void *opaque;   /* can be used to carry app specific stuff */
145     char codec_name[32];
146     int codec_type; /* see CODEC_TYPE_xxx */
147     int codec_id; /* see CODEC_ID_xxx */
148     unsigned int codec_tag;  /* codec tag, only used if unknown codec */
149 } AVCodecContext;
150
151 typedef struct AVCodec {
152     char *name;
153     int type;
154     int id;
155     int priv_data_size;
156     int (*init)(AVCodecContext *);
157     int (*encode)(AVCodecContext *, UINT8 *buf, int buf_size, void *data);
158     int (*close)(AVCodecContext *);
159     int (*decode)(AVCodecContext *, void *outdata, int *outdata_size, 
160                   UINT8 *buf, int buf_size);
161     int capabilities;
162     struct AVCodec *next;
163 } AVCodec;
164
165 /* three components are given, that's all */
166 typedef struct AVPicture {
167     UINT8 *data[3];
168     int linesize[3];
169 } AVPicture;
170
171 extern AVCodec ac3_encoder;
172 extern AVCodec mp2_encoder;
173 extern AVCodec mp3lame_encoder;
174 extern AVCodec mpeg1video_encoder;
175 extern AVCodec h263_encoder;
176 extern AVCodec h263p_encoder;
177 extern AVCodec rv10_encoder;
178 extern AVCodec mjpeg_encoder;
179 extern AVCodec mpeg4_encoder;
180 extern AVCodec msmpeg4_encoder;
181
182 extern AVCodec h263_decoder;
183 extern AVCodec mpeg4_decoder;
184 extern AVCodec msmpeg4_decoder;
185 extern AVCodec mpeg_decoder;
186 extern AVCodec h263i_decoder;
187 extern AVCodec rv10_decoder;
188 extern AVCodec mjpeg_decoder;
189 extern AVCodec mp3_decoder;
190
191 /* pcm codecs */
192 #define PCM_CODEC(id, name) \
193 extern AVCodec name ## _decoder; \
194 extern AVCodec name ## _encoder;
195
196 PCM_CODEC(CODEC_ID_PCM_S16LE, pcm_s16le);
197 PCM_CODEC(CODEC_ID_PCM_S16BE, pcm_s16be);
198 PCM_CODEC(CODEC_ID_PCM_U16LE, pcm_u16le);
199 PCM_CODEC(CODEC_ID_PCM_U16BE, pcm_u16be);
200 PCM_CODEC(CODEC_ID_PCM_S8, pcm_s8);
201 PCM_CODEC(CODEC_ID_PCM_U8, pcm_u8);
202 PCM_CODEC(CODEC_ID_PCM_ALAW, pcm_alaw);
203 PCM_CODEC(CODEC_ID_PCM_MULAW, pcm_mulaw);
204
205 #undef PCM_CODEC
206
207 /* dummy raw video codec */
208 extern AVCodec rawvideo_codec;
209
210 /* the following codecs use external GPL libs */
211 extern AVCodec ac3_decoder;
212
213 /* resample.c */
214
215 struct ReSampleContext;
216
217 typedef struct ReSampleContext ReSampleContext;
218
219 ReSampleContext *audio_resample_init(int output_channels, int input_channels, 
220                                      int output_rate, int input_rate);
221 int audio_resample(ReSampleContext *s, short *output, short *input, int nb_samples);
222 void audio_resample_close(ReSampleContext *s);
223
224 /* YUV420 format is assumed ! */
225
226 struct ImgReSampleContext;
227
228 typedef struct ImgReSampleContext ImgReSampleContext;
229
230 ImgReSampleContext *img_resample_init(int output_width, int output_height,
231                                       int input_width, int input_height);
232 void img_resample(ImgReSampleContext *s, 
233                   AVPicture *output, AVPicture *input);
234
235 void img_resample_close(ImgReSampleContext *s);
236
237 void avpicture_fill(AVPicture *picture, UINT8 *ptr,
238                     int pix_fmt, int width, int height);
239 int avpicture_get_size(int pix_fmt, int width, int height);
240
241 /* convert among pixel formats */
242 int img_convert(AVPicture *dst, int dst_pix_fmt,
243                 AVPicture *src, int pix_fmt, 
244                 int width, int height);
245
246 /* deinterlace a picture */
247 int avpicture_deinterlace(AVPicture *dst, AVPicture *src,
248                           int pix_fmt, int width, int height);
249
250 /* external high level API */
251
252 extern AVCodec *first_avcodec;
253
254 void avcodec_init(void);
255
256 void register_avcodec(AVCodec *format);
257 AVCodec *avcodec_find_encoder(enum CodecID id);
258 AVCodec *avcodec_find_encoder_by_name(const char *name);
259 AVCodec *avcodec_find_decoder(enum CodecID id);
260 AVCodec *avcodec_find_decoder_by_name(const char *name);
261 void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode);
262
263 int avcodec_open(AVCodecContext *avctx, AVCodec *codec);
264 int avcodec_decode_audio(AVCodecContext *avctx, INT16 *samples, 
265                          int *frame_size_ptr,
266                          UINT8 *buf, int buf_size);
267 int avcodec_decode_video(AVCodecContext *avctx, AVPicture *picture, 
268                          int *got_picture_ptr,
269                          UINT8 *buf, int buf_size);
270 int avcodec_encode_audio(AVCodecContext *avctx, UINT8 *buf, int buf_size, 
271                          const short *samples);
272 int avcodec_encode_video(AVCodecContext *avctx, UINT8 *buf, int buf_size, 
273                          const AVPicture *pict);
274
275 int avcodec_close(AVCodecContext *avctx);
276
277 void avcodec_register_all(void);
278
279 #ifdef FF_POSTPROCESS
280 #ifndef MBC
281 #define MBC 48
282 #define MBR 36
283 #endif
284 extern int quant_store[MBR+1][MBC+1]; // [Review]
285 #endif
286
287 #endif /* AVCODEC_H */