Mercurial > mplayer.hg
annotate libmpcodecs/vd_ffmpeg.c @ 6729:62b33bad8fca
Added DVB to the HTML docs.
Parts of this may be outdated, somebody please review.
author | diego |
---|---|
date | Sat, 13 Jul 2002 17:54:04 +0000 |
parents | 8898dd6c0302 |
children | 3b1f37fc0693 |
rev | line source |
---|---|
4952 | 1 #include <stdio.h> |
2 #include <stdlib.h> | |
3 #include <assert.h> | |
4 | |
5 #include "config.h" | |
6 #include "mp_msg.h" | |
7 #include "help_mp.h" | |
8 | |
9 #ifdef USE_LIBAVCODEC | |
10 | |
11 #include "bswap.h" | |
12 | |
13 #include "vd_internal.h" | |
14 | |
15 static vd_info_t info = { | |
16 "FFmpeg's libavcodec codec family", | |
17 "ffmpeg", | |
18 VFM_FFMPEG, | |
19 "A'rpi", | |
20 "http://ffmpeg.sf.net", | |
21 "native codecs" | |
22 }; | |
23 | |
24 LIBVD_EXTERN(ffmpeg) | |
25 | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
26 #include "../postproc/rgb2rgb.h" |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
27 |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
28 |
4952 | 29 #ifdef USE_LIBAVCODEC_SO |
30 #include <libffmpeg/avcodec.h> | |
31 #else | |
32 #include "libavcodec/avcodec.h" | |
33 #endif | |
34 | |
35 int avcodec_inited=0; | |
36 | |
5517 | 37 #ifdef FF_POSTPROCESS |
38 int quant_store[MBR+1][MBC+1]; | |
39 #endif | |
40 | |
5280 | 41 typedef struct { |
42 AVCodecContext *avctx; | |
5482 | 43 int last_aspect; |
44 int do_slices; | |
45 int vo_inited; | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
46 int convert; |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
47 int yuy2_support; |
6681 | 48 #if LIBAVCODEC_BUILD >= 4615 |
49 int yvu9_support; | |
50 #endif | |
5280 | 51 } vd_ffmpeg_ctx; |
52 | |
4952 | 53 //#ifdef FF_POSTPROCESS |
54 //unsigned int lavc_pp=0; | |
55 //#endif | |
56 | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
57 #include "cfgparser.h" |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
58 |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
59 static int lavc_param_workaround_bugs=0; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
60 static int lavc_param_error_resilience=0; |
6355 | 61 static int lavc_param_gray=0; |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
62 |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
63 struct config lavc_decode_opts_conf[]={ |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
64 #if LIBAVCODEC_BUILD >= 4611 |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
65 {"bug", &lavc_param_workaround_bugs, CONF_TYPE_INT, CONF_RANGE, 0, 99, NULL}, |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
66 {"ver", &lavc_param_error_resilience, CONF_TYPE_INT, CONF_RANGE, -1, 99, NULL}, |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
67 #endif |
6355 | 68 #if LIBAVCODEC_BUILD >= 4614 |
69 {"gray", &lavc_param_gray, CONF_TYPE_FLAG, 0, 0, CODEC_FLAG_PART, NULL}, | |
70 #endif | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
71 {NULL, NULL, 0, 0, 0, 0, NULL} |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
72 }; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
73 |
4952 | 74 // to set/get/query special features/parameters |
75 static int control(sh_video_t *sh,int cmd,void* arg,...){ | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
76 vd_ffmpeg_ctx *ctx = sh->context; |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
77 switch(cmd){ |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
78 case VDCTRL_QUERY_FORMAT: |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
79 if( (*((int*)arg)) == IMGFMT_YV12 ) return CONTROL_TRUE; |
5707 | 80 if( (*((int*)arg)) == IMGFMT_IYUV ) return CONTROL_TRUE; |
81 if( (*((int*)arg)) == IMGFMT_I420 ) return CONTROL_TRUE; | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
82 if( (*((int*)arg)) == IMGFMT_YUY2 && ctx->yuy2_support ) return CONTROL_TRUE; |
6681 | 83 #if LIBAVCODEC_BUILD >= 4615 |
84 if( (*((int*)arg)) == IMGFMT_YVU9 && ctx->yvu9_support ) return CONTROL_TRUE; | |
85 #endif | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
86 return CONTROL_FALSE; |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
87 } |
4952 | 88 return CONTROL_UNKNOWN; |
89 } | |
90 | |
91 // init driver | |
92 static int init(sh_video_t *sh){ | |
5280 | 93 AVCodecContext *avctx; |
94 vd_ffmpeg_ctx *ctx; | |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
95 AVCodec *lavc_codec; |
4952 | 96 |
97 if(!avcodec_inited){ | |
98 avcodec_init(); | |
99 avcodec_register_all(); | |
100 avcodec_inited=1; | |
101 } | |
5280 | 102 |
103 ctx = sh->context = malloc(sizeof(vd_ffmpeg_ctx)); | |
104 if (!ctx) | |
105 return(0); | |
106 memset(ctx, 0, sizeof(vd_ffmpeg_ctx)); | |
4952 | 107 |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
108 lavc_codec = (AVCodec *)avcodec_find_decoder_by_name(sh->codec->dll); |
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
109 if(!lavc_codec){ |
4952 | 110 mp_msg(MSGT_DECVIDEO,MSGL_ERR,MSGTR_MissingLAVCcodec,sh->codec->dll); |
111 return 0; | |
112 } | |
5482 | 113 |
5494
1c45b1484ffb
i just fixed b-frames & slices in libavcodec :) ... iam too tired for benchmarks now ...
michael
parents:
5482
diff
changeset
|
114 if(vd_use_slices && lavc_codec->capabilities&CODEC_CAP_DRAW_HORIZ_BAND) |
1c45b1484ffb
i just fixed b-frames & slices in libavcodec :) ... iam too tired for benchmarks now ...
michael
parents:
5482
diff
changeset
|
115 ctx->do_slices=1; |
4952 | 116 |
5280 | 117 ctx->avctx = malloc(sizeof(AVCodecContext)); |
118 memset(ctx->avctx, 0, sizeof(AVCodecContext)); | |
119 avctx = ctx->avctx; | |
4952 | 120 |
5280 | 121 avctx->width = sh->disp_w; |
122 avctx->height= sh->disp_h; | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
123 #if LIBAVCODEC_BUILD >= 4611 |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
124 avctx->workaround_bugs= lavc_param_workaround_bugs; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
125 avctx->error_resilience= lavc_param_error_resilience; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
126 #endif |
6355 | 127 #if LIBAVCODEC_BUILD >= 4614 |
128 if(lavc_param_gray) avctx->flags|= CODEC_FLAG_GRAY; | |
129 #endif | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
130 |
5280 | 131 mp_dbg(MSGT_DECVIDEO,MSGL_DBG2,"libavcodec.size: %d x %d\n",avctx->width,avctx->height); |
4952 | 132 if (sh->format == mmioFOURCC('R', 'V', '1', '3')) |
5280 | 133 avctx->sub_id = 3; |
5939 | 134 #if LIBAVCODEC_BUILD >= 4605 |
135 /* AVRn stores huffman table in AVI header */ | |
136 /* Pegasus MJPEG stores it also in AVI header, but it uses the common | |
137 MJPG fourcc :( */ | |
5940 | 138 if (sh->bih && (sh->bih->biSize != sizeof(BITMAPINFOHEADER)) && |
5939 | 139 (sh->format == mmioFOURCC('A','V','R','n') || |
140 sh->format == mmioFOURCC('M','J','P','G'))) | |
141 { | |
142 avctx->flags |= CODEC_FLAG_EXTERN_HUFF; | |
143 avctx->extradata_size = sh->bih->biSize-sizeof(BITMAPINFOHEADER); | |
144 avctx->extradata = malloc(avctx->extradata_size); | |
145 memcpy(avctx->extradata, sh->bih+sizeof(BITMAPINFOHEADER), | |
146 avctx->extradata_size); | |
147 | |
148 #if 0 | |
149 { | |
150 int x; | |
151 uint8_t *p = avctx->extradata; | |
152 | |
153 for (x=0; x<avctx->extradata_size; x++) | |
154 printf("[%x] ", p[x]); | |
155 printf("\n"); | |
156 } | |
157 #endif | |
158 } | |
159 #endif | |
4952 | 160 /* open it */ |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
161 if (avcodec_open(avctx, lavc_codec) < 0) { |
4952 | 162 mp_msg(MSGT_DECVIDEO,MSGL_ERR, MSGTR_CantOpenCodec); |
163 return 0; | |
164 } | |
165 mp_msg(MSGT_DECVIDEO,MSGL_V,"INFO: libavcodec init OK!\n"); | |
5482 | 166 ctx->last_aspect=-3; |
5510 | 167 return 1; //mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,IMGFMT_YV12); |
4952 | 168 } |
169 | |
170 // uninit driver | |
171 static void uninit(sh_video_t *sh){ | |
5280 | 172 vd_ffmpeg_ctx *ctx = sh->context; |
173 AVCodecContext *avctx = ctx->avctx; | |
174 | |
175 if (avcodec_close(avctx) < 0) | |
4952 | 176 mp_msg(MSGT_DECVIDEO,MSGL_ERR, MSGTR_CantCloseCodec); |
5939 | 177 |
178 #if LIBAVCODEC_BUILD >= 4605 | |
179 if (avctx->extradata_size) | |
180 free(avctx->extradata); | |
181 #endif | |
182 | |
5280 | 183 if (avctx) |
184 free(avctx); | |
185 if (ctx) | |
186 free(ctx); | |
4952 | 187 } |
188 | |
5482 | 189 static void draw_slice(struct AVCodecContext *s, |
190 UINT8 **src, int linesize, | |
191 int y, int width, int height){ | |
6710 | 192 sh_video_t * sh = s->opaque; |
5482 | 193 int stride[3]; |
194 | |
195 stride[0]=linesize; | |
196 stride[1]=stride[2]=stride[0]/2; | |
197 | |
6710 | 198 mpcodecs_draw_slice (sh,src, stride, width, height, 0, y); |
199 | |
5482 | 200 } |
4952 | 201 |
202 // decode a frame | |
203 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){ | |
204 int got_picture=0; | |
205 int ret; | |
206 AVPicture lavc_picture; | |
5280 | 207 vd_ffmpeg_ctx *ctx = sh->context; |
208 AVCodecContext *avctx = ctx->avctx; | |
5482 | 209 mp_image_t* mpi=NULL; |
4952 | 210 |
211 if(len<=0) return NULL; // skipped frame | |
212 | |
5613 | 213 avctx->draw_horiz_band=NULL; |
214 if(ctx->vo_inited && !ctx->convert && !(flags&3)){ | |
5482 | 215 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, MP_IMGFLAG_PRESERVE | |
216 (ctx->do_slices?MP_IMGFLAG_DRAW_CALLBACK:0), | |
217 sh->disp_w, sh->disp_h); | |
218 if(mpi && mpi->flags&MP_IMGFLAG_DRAW_CALLBACK){ | |
219 // vd core likes slices! | |
220 avctx->draw_horiz_band=draw_slice; | |
6710 | 221 avctx->opaque=sh; |
5613 | 222 } |
5482 | 223 } |
5875 | 224 |
225 #if LIBAVCODEC_BUILD > 4603 | |
226 avctx->hurry_up=(flags&3)?((flags&2)?2:1):0; | |
227 #endif | |
228 | |
5280 | 229 ret = avcodec_decode_video(avctx, &lavc_picture, |
4952 | 230 &got_picture, data, len); |
231 if(ret<0) mp_msg(MSGT_DECVIDEO,MSGL_WARN, "Error while decoding frame!\n"); | |
232 if(!got_picture) return NULL; // skipped image | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
233 |
5482 | 234 if (avctx->aspect_ratio_info != ctx->last_aspect || |
235 avctx->width != sh->disp_w || | |
236 avctx->height != sh->disp_h || | |
237 !ctx->vo_inited) | |
5280 | 238 { |
239 ctx->last_aspect = avctx->aspect_ratio_info; | |
240 switch(avctx->aspect_ratio_info) | |
241 { | |
242 case FF_ASPECT_4_3_625: | |
243 case FF_ASPECT_4_3_525: | |
244 sh->aspect = 4.0/3.0; | |
245 break; | |
246 case FF_ASPECT_16_9_625: | |
247 case FF_ASPECT_16_9_525: | |
248 sh->aspect = 16.0/9.0; | |
249 break; | |
250 case FF_ASPECT_SQUARE: | |
251 sh->aspect = 0.0; | |
252 break; | |
253 } | |
5482 | 254 sh->disp_w = avctx->width; |
255 sh->disp_h = avctx->height; | |
256 ctx->vo_inited=1; | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
257 ctx->yuy2_support=(avctx->pix_fmt==PIX_FMT_YUV422P); |
6681 | 258 #if LIBAVCODEC_BUILD >= 4615 |
259 ctx->yvu9_support=(avctx->pix_fmt==PIX_FMT_YUV410P); | |
260 #endif | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
261 if (!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h, |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
262 ctx->yuy2_support ? IMGFMT_YUY2 : IMGFMT_YV12)) |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
263 return NULL; |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
264 ctx->convert=(sh->codec->outfmt[sh->outfmtidx]==IMGFMT_YUY2); |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
265 } |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
266 |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
267 if(!mpi && ctx->convert){ |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
268 // do yuv422p -> yuy2 conversion: |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
269 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE, |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
270 avctx->width, avctx->height); |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
271 if(!mpi) return NULL; |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
272 yuv422ptoyuy2(lavc_picture.data[0],lavc_picture.data[1],lavc_picture.data[2], |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
273 mpi->planes[0],avctx->width,avctx->height, |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
274 lavc_picture.linesize[0],lavc_picture.linesize[1],mpi->stride[0]); |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
275 return mpi; |
5280 | 276 } |
4952 | 277 |
5482 | 278 if(!mpi) |
4952 | 279 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, MP_IMGFLAG_PRESERVE, |
5280 | 280 avctx->width, avctx->height); |
4952 | 281 if(!mpi){ // temporary! |
5207
d337cc4ab0ee
config vo if resolution changed (after decoded image read the dimensions out of lavc context)
alex
parents:
5124
diff
changeset
|
282 printf("couldn't allocate image for codec\n"); |
4952 | 283 return NULL; |
284 } | |
285 | |
286 mpi->planes[0]=lavc_picture.data[0]; | |
287 mpi->planes[1]=lavc_picture.data[1]; | |
288 mpi->planes[2]=lavc_picture.data[2]; | |
289 mpi->stride[0]=lavc_picture.linesize[0]; | |
290 mpi->stride[1]=lavc_picture.linesize[1]; | |
291 mpi->stride[2]=lavc_picture.linesize[2]; | |
292 | |
5280 | 293 if(avctx->pix_fmt==PIX_FMT_YUV422P){ |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
294 // we have 422p but user wants yv12 (420p) |
4952 | 295 mpi->stride[1]*=2; |
296 mpi->stride[2]*=2; | |
297 } | |
298 | |
6665 | 299 /* to comfirm with newer lavc style */ |
300 #if !defined(FF_POSTPROCESS) && (LIBAVCODEC_BUILD > 4612) | |
301 mpi->qscale=avctx->quant_store; | |
302 #if LIBAVCODEC_BUILD > 4613 | |
303 mpi->qstride=avctx->qstride; | |
304 #else | |
305 mpi->qstride=MBC+1; | |
306 #endif | |
307 #else ifdef FF_POSTPROCESS | |
5517 | 308 mpi->qscale=&quant_store[0][0]; |
309 mpi->qstride=MBC+1; | |
310 #endif | |
311 | |
4952 | 312 return mpi; |
313 } | |
314 | |
315 #endif | |
316 |