1 /*****************************************************************************
2 * ffmpeg.c: video decoder using ffmpeg library
3 *****************************************************************************
4 * Copyright (C) 1999-2001 VideoLAN
5 * $Id: ffmpeg.c,v 1.64 2003/11/29 13:12:11 fenrir Exp $
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Gildas Bazin <gbazin@netcourrier.com>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
29 #include <vlc/decoder.h>
33 #ifdef HAVE_FFMPEG_AVCODEC_H
34 # include <ffmpeg/avcodec.h>
39 #if LIBAVCODEC_BUILD < 4655
40 # error You must have a libavcodec >= 4655 (get CVS)
46 # ifdef HAVE_POSTPROC_POSTPROCESS_H
47 # include <postproc/postprocess.h>
49 # include <libpostproc/postprocess.h>
53 /*****************************************************************************
54 * decoder_sys_t: decoder descriptor
55 *****************************************************************************/
58 /* Common part between video and audio decoder */
63 AVCodecContext *p_context;
67 /****************************************************************************
69 ****************************************************************************/
70 static int OpenDecoder( vlc_object_t * );
71 static void CloseDecoder( vlc_object_t * );
73 /*****************************************************************************
75 *****************************************************************************/
78 /* decoder main module */
79 add_category_hint( N_("ffmpeg"), NULL, VLC_FALSE );
80 set_capability( "decoder", 70 );
81 set_callbacks( OpenDecoder, CloseDecoder );
82 set_description( _("ffmpeg audio/video decoder((MS)MPEG4,SVQ1,H263,WMV,WMA)") );
84 add_bool( "ffmpeg-dr", 1, NULL, DR_TEXT, DR_TEXT, VLC_TRUE );
85 add_integer ( "ffmpeg-error-resilience", -1, NULL, ERROR_TEXT,
86 ERROR_LONGTEXT, VLC_TRUE );
87 add_integer ( "ffmpeg-workaround-bugs", 1, NULL, BUGS_TEXT, BUGS_LONGTEXT,
89 add_bool( "ffmpeg-hurry-up", 0, NULL, HURRYUP_TEXT, HURRYUP_LONGTEXT,
91 add_integer( "ffmpeg-truncated", 0, NULL, TRUNC_TEXT, TRUNC_LONGTEXT,
95 add_integer( "ffmpeg-pp-q", 0, NULL, PP_Q_TEXT, PP_Q_LONGTEXT, VLC_FALSE );
96 add_string( "ffmpeg-pp-name", "default", NULL, LIBAVCODEC_PP_TEXT,
97 LIBAVCODEC_PP_LONGTEXT, VLC_TRUE );
100 /* chroma conversion submodule */
102 set_capability( "chroma", 50 );
103 set_callbacks( E_(OpenChroma), E_(CloseChroma) );
104 set_description( _("ffmpeg chroma conversion") );
106 /* encoder submodule */
108 set_description( _("ffmpeg audio/video encoder") );
109 set_capability( "encoder", 100 );
110 set_callbacks( E_(OpenEncoder), E_(CloseEncoder) );
112 var_Create( p_module->p_libvlc, "avcodec", VLC_VAR_MUTEX );
115 /*****************************************************************************
116 * OpenDecoder: probe the decoder and return score
117 *****************************************************************************/
118 static int OpenDecoder( vlc_object_t *p_this )
120 decoder_t *p_dec = (decoder_t*) p_this;
121 int i_cat, i_codec_id, i_result;
124 AVCodecContext *p_context;
127 /* *** determine codec type *** */
128 if( !E_(GetFfmpegCodec)( p_dec->fmt_in.i_codec, &i_cat, &i_codec_id,
134 /* Initialization must be done before avcodec_find_decoder() */
135 E_(InitLibavcodec)(p_this);
137 /* *** ask ffmpeg for a decoder *** */
138 if( !( p_codec = avcodec_find_decoder( i_codec_id ) ) )
140 msg_Dbg( p_dec, "codec not found (%s)", psz_namecodec );
144 /* *** get a p_context *** */
145 p_context = avcodec_alloc_context();
147 /* Set CPU capabilities */
148 p_context->dsp_mask = 0;
149 if( !(p_dec->p_libvlc->i_cpu & CPU_CAPABILITY_MMX) )
151 p_context->dsp_mask |= FF_MM_MMX;
153 if( !(p_dec->p_libvlc->i_cpu & CPU_CAPABILITY_MMXEXT) )
155 p_context->dsp_mask |= FF_MM_MMXEXT;
157 if( !(p_dec->p_libvlc->i_cpu & CPU_CAPABILITY_3DNOW) )
159 p_context->dsp_mask |= FF_MM_3DNOW;
161 if( !(p_dec->p_libvlc->i_cpu & CPU_CAPABILITY_SSE) )
163 p_context->dsp_mask |= FF_MM_SSE;
164 p_context->dsp_mask |= FF_MM_SSE2;
170 p_dec->pf_decode_video = E_(DecodeVideo);
171 i_result = E_( InitVideoDec )( p_dec, p_context, p_codec,
172 i_codec_id, psz_namecodec );
175 p_dec->pf_decode_audio = E_(DecodeAudio);
176 i_result = E_( InitAudioDec )( p_dec, p_context, p_codec,
177 i_codec_id, psz_namecodec );
180 i_result = VLC_EGENERIC;
183 p_dec->p_sys->i_cat = i_cat;
188 /*****************************************************************************
189 * CloseDecoder: decoder destruction
190 *****************************************************************************/
191 static void CloseDecoder( vlc_object_t *p_this )
193 decoder_t *p_dec = (decoder_t *)p_this;
194 decoder_sys_t *p_sys = p_dec->p_sys;
196 if( !p_sys->p_context )
198 if( p_sys->p_context->extradata )
199 free( p_sys->p_context->extradata );
201 avcodec_close( p_sys->p_context );
202 msg_Dbg( p_dec, "ffmpeg codec (%s) stopped", p_sys->psz_namecodec );
203 free( p_sys->p_context );
206 switch( p_sys->i_cat )
209 E_( EndAudioDec )( p_dec );
212 E_( EndVideoDec )( p_dec );
219 /*****************************************************************************
221 *****************************************************************************/
222 int E_(GetFfmpegCodec)( vlc_fourcc_t i_fourcc, int *pi_cat,
223 int *pi_ffmpeg_codec, char **ppsz_name )
237 case VLC_FOURCC('m','p','1','v'):
239 i_codec = CODEC_ID_MPEG1VIDEO;
240 psz_name = "MPEG-1 Video";
244 case VLC_FOURCC('m','p','2','v'):
245 case VLC_FOURCC('m','p','g','v'):
247 #if LIBAVCODEC_BUILD >= 4676
248 i_codec = CODEC_ID_MPEG2VIDEO;
249 psz_name = "MPEG-2 Video";
251 i_codec = CODEC_ID_MPEG1VIDEO;
252 psz_name = "MPEG-1 Video";
257 case VLC_FOURCC('D','I','V','X'):
258 case VLC_FOURCC('d','i','v','x'):
259 case VLC_FOURCC('M','P','4','S'):
260 case VLC_FOURCC('m','p','4','s'):
261 case VLC_FOURCC('M','4','S','2'):
262 case VLC_FOURCC('m','4','s','2'):
263 case VLC_FOURCC('x','v','i','d'):
264 case VLC_FOURCC('X','V','I','D'):
265 case VLC_FOURCC('X','v','i','D'):
266 case VLC_FOURCC('D','X','5','0'):
267 case VLC_FOURCC('m','p','4','v'):
268 case VLC_FOURCC( 4, 0, 0, 0 ):
269 case VLC_FOURCC('m','4','c','c'):
270 case VLC_FOURCC('M','4','C','C'):
271 /* 3ivx delta 3.5 Unsupported
272 * putting it here gives extreme distorted images
273 case VLC_FOURCC('3','I','V','1'):
274 case VLC_FOURCC('3','i','v','1'): */
276 case VLC_FOURCC('3','I','V','2'):
277 case VLC_FOURCC('3','i','v','2'):
279 i_codec = CODEC_ID_MPEG4;
284 case VLC_FOURCC('D','I','V','1'):
285 case VLC_FOURCC('d','i','v','1'):
286 case VLC_FOURCC('M','P','G','4'):
287 case VLC_FOURCC('m','p','g','4'):
289 i_codec = CODEC_ID_MSMPEG4V1;
290 psz_name = "MS MPEG-4 v1";
294 case VLC_FOURCC('D','I','V','2'):
295 case VLC_FOURCC('d','i','v','2'):
296 case VLC_FOURCC('M','P','4','2'):
297 case VLC_FOURCC('m','p','4','2'):
299 i_codec = CODEC_ID_MSMPEG4V2;
300 psz_name = "MS MPEG-4 v2";
303 /* MSMPEG4 v3 / M$ mpeg4 v3 */
304 case VLC_FOURCC('M','P','G','3'):
305 case VLC_FOURCC('m','p','g','3'):
306 case VLC_FOURCC('d','i','v','3'):
307 case VLC_FOURCC('M','P','4','3'):
308 case VLC_FOURCC('m','p','4','3'):
310 case VLC_FOURCC('D','I','V','3'):
311 case VLC_FOURCC('D','I','V','4'):
312 case VLC_FOURCC('d','i','v','4'):
313 case VLC_FOURCC('D','I','V','5'):
314 case VLC_FOURCC('d','i','v','5'):
315 case VLC_FOURCC('D','I','V','6'):
316 case VLC_FOURCC('d','i','v','6'):
317 /* AngelPotion stuff */
318 case VLC_FOURCC('A','P','4','1'):
319 /* 3ivx doctered divx files */
320 case VLC_FOURCC('3','I','V','D'):
321 case VLC_FOURCC('3','i','v','d'):
323 case VLC_FOURCC('3','V','I','D'):
324 case VLC_FOURCC('3','v','i','d'):
326 i_codec = CODEC_ID_MSMPEG4V3;
327 psz_name = "MS MPEG-4 v3";
331 case VLC_FOURCC('S','V','Q','1'):
333 i_codec = CODEC_ID_SVQ1;
334 psz_name = "SVQ-1 (Sorenson Video v1)";
338 case VLC_FOURCC('S','V','Q','3'):
340 i_codec = CODEC_ID_SVQ3;
341 psz_name = "SVQ-3 (Sorenson Video v3)";
345 case VLC_FOURCC('h','2','6','4'):
346 case VLC_FOURCC('H','2','6','4'):
348 i_codec = CODEC_ID_H264;
353 /* H263(+) is also known as Real Video 1.0 */
355 /* FIXME FOURCC_H263P exist but what fourcc ? */
358 case VLC_FOURCC('H','2','6','3'):
359 case VLC_FOURCC('h','2','6','3'):
360 case VLC_FOURCC('U','2','6','3'):
362 i_codec = CODEC_ID_H263;
367 case VLC_FOURCC('I','2','6','3'):
368 case VLC_FOURCC('i','2','6','3'):
370 i_codec = CODEC_ID_H263I;
374 #if LIBAVCODEC_BUILD >= 4669
375 /* Flash (H263) variant */
376 case VLC_FOURCC('F','L','V','1'):
378 i_codec = CODEC_ID_FLV1;
379 psz_name = "Flash Video";
384 case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
385 case VLC_FOURCC( 'm', 'j', 'p', 'g' ):
386 case VLC_FOURCC( 'm', 'j', 'p', 'a' ): /* for mov file */
387 case VLC_FOURCC( 'j', 'p', 'e', 'g' ):
388 case VLC_FOURCC( 'J', 'P', 'E', 'G' ):
389 case VLC_FOURCC( 'J', 'F', 'I', 'F' ):
390 case VLC_FOURCC( 'J', 'P', 'G', 'L' ):
392 i_codec = CODEC_ID_MJPEG;
393 psz_name = "Motion JPEG";
395 case VLC_FOURCC( 'm', 'j', 'p', 'b' ): /* for mov file */
397 i_codec = CODEC_ID_MJPEGB;
398 psz_name = "Motion JPEG B";
402 case VLC_FOURCC('d','v','s','l'):
403 case VLC_FOURCC('d','v','s','d'):
404 case VLC_FOURCC('D','V','S','D'):
405 case VLC_FOURCC('d','v','h','d'):
406 case VLC_FOURCC('d','v','c',' '):
407 case VLC_FOURCC('d','v','p',' '):
408 case VLC_FOURCC('C','D','V','C'):
410 i_codec = CODEC_ID_DVVIDEO;
411 psz_name = "DV video";
414 /* Windows Media Video */
415 case VLC_FOURCC('W','M','V','1'):
417 i_codec = CODEC_ID_WMV1;
418 psz_name ="Windows Media Video 1";
420 case VLC_FOURCC('W','M','V','2'):
422 i_codec = CODEC_ID_WMV2;
423 psz_name ="Windows Media Video 2";
426 #if LIBAVCODEC_BUILD >= 4683
427 /* Microsoft Video 1 */
428 case VLC_FOURCC('M','S','V','C'):
429 case VLC_FOURCC('m','s','v','c'):
430 case VLC_FOURCC('C','R','A','M'):
431 case VLC_FOURCC('c','r','a','m'):
432 case VLC_FOURCC('W','H','A','M'):
433 case VLC_FOURCC('w','h','a','m'):
435 i_codec = CODEC_ID_MSVIDEO1;
436 psz_name = "Microsoft Video 1";
440 case VLC_FOURCC('m','r','l','e'):
441 case VLC_FOURCC(0x1,0x0,0x0,0x0):
443 i_codec = CODEC_ID_MSRLE;
444 psz_name = "Microsoft RLE";
448 #if( ( LIBAVCODEC_BUILD >= 4663 ) && ( !defined( WORDS_BIGENDIAN ) ) )
449 /* Indeo Video Codecs (Quality of this decoder on ppc is not good) */
450 case VLC_FOURCC('I','V','3','1'):
451 case VLC_FOURCC('i','v','3','1'):
452 case VLC_FOURCC('I','V','3','2'):
453 case VLC_FOURCC('i','v','3','2'):
455 i_codec = CODEC_ID_INDEO3;
456 psz_name = "Indeo v3";
461 case VLC_FOURCC('H','F','Y','U'):
463 i_codec = CODEC_ID_HUFFYUV;
464 psz_name ="Huff YUV";
468 case VLC_FOURCC('C','Y','U','V'):
470 i_codec = CODEC_ID_CYUV;
471 psz_name ="Creative YUV";
474 #if LIBAVCODEC_BUILD >= 4668
475 /* On2 VP3 Video Codecs */
476 case VLC_FOURCC('V','P','3','1'):
477 case VLC_FOURCC('v','p','3','1'):
479 i_codec = CODEC_ID_VP3;
480 psz_name = "On2's VP3 Video";
484 #if ( !defined( WORDS_BIGENDIAN ) )
485 #if LIBAVCODEC_BUILD >= 4668
486 /* Asus Video (Another thing that doesn't work on PPC) */
487 case VLC_FOURCC('A','S','V','1'):
489 i_codec = CODEC_ID_ASV1;
490 psz_name = "Asus V1";
493 #if LIBAVCODEC_BUILD >= 4677
494 case VLC_FOURCC('A','S','V','2'):
496 i_codec = CODEC_ID_ASV2;
497 psz_name = "Asus V2";
502 #if LIBAVCODEC_BUILD >= 4668
503 /* FFMPEG Video 1 (lossless codec) */
504 case VLC_FOURCC('F','F','V','1'):
506 i_codec = CODEC_ID_FFV1;
507 psz_name = "FFMpeg Video 1";
511 #if LIBAVCODEC_BUILD >= 4669
513 case VLC_FOURCC('V','C','R','1'):
515 i_codec = CODEC_ID_VCR1;
516 psz_name = "ATI VCR1";
520 #if LIBAVCODEC_BUILD >= 4672
521 /* Cirrus Logic AccuPak */
522 case VLC_FOURCC('C','L','J','R'):
524 i_codec = CODEC_ID_CLJR;
525 psz_name = "Creative Logic AccuPak";
529 #if LIBAVCODEC_BUILD >= 4683
531 case VLC_FOURCC('r','p','z','a'):
533 i_codec = CODEC_ID_RPZA;
534 psz_name = "Apple Video";
538 #if LIBAVCODEC_BUILD >= 4683
540 case VLC_FOURCC('c','v','i','d'):
542 i_codec = CODEC_ID_CINEPAK;
543 psz_name = "Cinepak";
551 /* Windows Media Audio 1 */
552 case VLC_FOURCC('W','M','A','1'):
553 case VLC_FOURCC('w','m','a','1'):
555 i_codec = CODEC_ID_WMAV1;
556 psz_name ="Windows Media Audio 1";
559 /* Windows Media Audio 2 */
560 case VLC_FOURCC('W','M','A','2'):
561 case VLC_FOURCC('w','m','a','2'):
563 i_codec = CODEC_ID_WMAV2;
564 psz_name ="Windows Media Audio 2";
568 case VLC_FOURCC('d','v','a','u'):
570 i_codec = CODEC_ID_DVAUDIO;
571 psz_name = "DV audio";
575 case VLC_FOURCC('M','A','C','3'):
577 i_codec = CODEC_ID_MACE3;
578 psz_name = "MACE-3 audio";
582 case VLC_FOURCC('M','A','C','6'):
584 i_codec = CODEC_ID_MACE6;
585 psz_name = "MACE-6 audio";
588 #if LIBAVCODEC_BUILD >= 4668
590 case VLC_FOURCC('1','4','_','4'):
592 i_codec = CODEC_ID_RA_144;
593 psz_name = "RealAudio 1.0";
597 case VLC_FOURCC('2','8','_','8'):
599 i_codec = CODEC_ID_RA_288;
600 psz_name = "RealAudio 2.0";
604 /* MPEG Audio layer 1/2/3 */
605 case VLC_FOURCC('m','p','g','a'):
607 i_codec = CODEC_ID_MP2;
608 psz_name = "MPEG Audio layer 1/2";
610 case VLC_FOURCC('m','p','3',' '):
612 #if LIBAVCODEC_BUILD >= 4678
613 i_codec = CODEC_ID_MP3;
615 i_codec = CODEC_ID_MP3LAME;
617 psz_name = "MPEG Audio layer 1/2/3";
620 /* A52 Audio (aka AC3) */
621 case VLC_FOURCC('a','5','2',' '):
622 case VLC_FOURCC('a','5','2','b'): /* VLC specific hack */
624 i_codec = CODEC_ID_AC3;
625 psz_name = "A52 Audio (aka AC3)";
629 case VLC_FOURCC('m','p','4','a'):
631 i_codec = CODEC_ID_AAC;
632 psz_name = "MPEG AAC Audio";
637 i_codec = CODEC_ID_NONE;
642 if( i_codec != CODEC_ID_NONE )
644 if( pi_cat ) *pi_cat = i_cat;
645 if( pi_ffmpeg_codec ) *pi_ffmpeg_codec = i_codec;
646 if( ppsz_name ) *ppsz_name = psz_name;
653 int E_(GetFfmpegChroma)( vlc_fourcc_t i_chroma )
657 case VLC_FOURCC( 'I', '4', '2', '0' ):
658 return PIX_FMT_YUV420P;
659 case VLC_FOURCC( 'I', '4', '2', '2' ):
660 return PIX_FMT_YUV422P;
661 case VLC_FOURCC( 'I', '4', '4', '4' ):
662 return PIX_FMT_YUV444P;
663 case VLC_FOURCC( 'R', 'V', '1', '5' ):
664 return PIX_FMT_RGB555;
665 case VLC_FOURCC( 'R', 'V', '1', '6' ):
666 return PIX_FMT_RGB565;
667 case VLC_FOURCC( 'R', 'V', '2', '4' ):
668 return PIX_FMT_RGB24;
669 case VLC_FOURCC( 'R', 'V', '3', '2' ):
670 return PIX_FMT_RGBA32;
671 case VLC_FOURCC( 'G', 'R', 'E', 'Y' ):
672 return PIX_FMT_GRAY8;
673 case VLC_FOURCC( 'Y', 'U', 'Y', '2' ):
674 return PIX_FMT_YUV422;
680 void E_(InitLibavcodec)( vlc_object_t *p_object )
682 static int b_ffmpeginit = 0;
685 var_Get( p_object->p_libvlc, "avcodec", &lockval );
686 vlc_mutex_lock( lockval.p_address );
688 /* *** init ffmpeg library (libavcodec) *** */
692 avcodec_register_all();
695 msg_Dbg( p_object, "libavcodec initialized (interface %d )",
700 msg_Dbg( p_object, "libavcodec already initialized" );
703 vlc_mutex_unlock( lockval.p_address );