X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fstream_out%2Ftranscode.c;h=9316d66af3a8a4cc6900bf17c8532d2bb4867e96;hb=a9583b5d9c501e00463f80f994c7d2e759437114;hp=0afa0e1e3db0284d0b66c6350520411e2e0611c2;hpb=d44f9d26aec48a4af5e4eccca6fb1fc9bbc75c13;p=vlc diff --git a/modules/stream_out/transcode.c b/modules/stream_out/transcode.c index 0afa0e1e3d..9316d66af3 100644 --- a/modules/stream_out/transcode.c +++ b/modules/stream_out/transcode.c @@ -1,10 +1,11 @@ /***************************************************************************** - * transcode.c + * transcode.c: transcoding stream output module ***************************************************************************** - * Copyright (C) 2001, 2002 VideoLAN - * $Id: transcode.c,v 1.35 2003/10/08 21:01:07 gbazin Exp $ + * Copyright (C) 2003-2004 VideoLAN + * $Id$ * * Authors: Laurent Aimar + * Gildas Bazin * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -32,83 +33,275 @@ #include #include #include - -/* ffmpeg header */ -#ifdef HAVE_FFMPEG_AVCODEC_H -# include -#else -# include -#endif - -/* vorbis header */ -#ifdef HAVE_VORBIS_VORBISENC_H -# include -# ifndef OV_ECTL_RATEMANAGE_AVG -# define OV_ECTL_RATEMANAGE_AVG 0x0 -# endif -#endif - -/***************************************************************************** - * Exported prototypes - *****************************************************************************/ -static int Open ( vlc_object_t * ); -static void Close ( vlc_object_t * ); - -static sout_stream_id_t *Add ( sout_stream_t *, sout_format_t * ); -static int Del ( sout_stream_t *, sout_stream_id_t * ); -static int Send( sout_stream_t *, sout_stream_id_t *, sout_buffer_t* ); - -static int transcode_audio_ffmpeg_new ( sout_stream_t *, sout_stream_id_t * ); -static void transcode_audio_ffmpeg_close ( sout_stream_t *, sout_stream_id_t * ); -static int transcode_audio_ffmpeg_process( sout_stream_t *, sout_stream_id_t *, sout_buffer_t *, sout_buffer_t ** ); - -static int transcode_video_ffmpeg_new ( sout_stream_t *, sout_stream_id_t * ); -static void transcode_video_ffmpeg_close ( sout_stream_t *, sout_stream_id_t * ); -static int transcode_video_ffmpeg_process( sout_stream_t *, sout_stream_id_t *, sout_buffer_t *, sout_buffer_t ** ); - -static int transcode_video_ffmpeg_getframebuf( struct AVCodecContext *, AVFrame *); +#include "vlc_filter.h" +#include "osd.h" /***************************************************************************** * Module descriptor *****************************************************************************/ +#define VENC_TEXT N_("Video encoder") +#define VENC_LONGTEXT N_( \ + "Allows you to specify the video encoder to use and its associated " \ + "options." ) +#define VCODEC_TEXT N_("Destination video codec") +#define VCODEC_LONGTEXT N_( \ + "Allows you to specify the destination video codec used for the " \ + "streaming output." ) +#define VB_TEXT N_("Video bitrate") +#define VB_LONGTEXT N_( \ + "Allows you to specify the video bitrate used for the streaming " \ + "output." ) +#define SCALE_TEXT N_("Video scaling") +#define SCALE_LONGTEXT N_( \ + "Allows you to scale the video before encoding." ) +#define FPS_TEXT N_("Video frame-rate") +#define FPS_LONGTEXT N_( \ + "Allows you to specify an output frame rate for the video." ) +#define DEINTERLACE_TEXT N_("Deinterlace video") +#define DEINTERLACE_LONGTEXT N_( \ + "Allows you to deinterlace the video before encoding." ) +#define WIDTH_TEXT N_("Video width") +#define WIDTH_LONGTEXT N_( \ + "Allows you to specify the output video width." ) +#define HEIGHT_TEXT N_("Video height") +#define HEIGHT_LONGTEXT N_( \ + "Allows you to specify the output video height." ) + +#define CROPTOP_TEXT N_("Video crop top") +#define CROPTOP_LONGTEXT N_( \ + "Allows you to specify the top coordinate for the video cropping." ) +#define CROPLEFT_TEXT N_("Video crop left") +#define CROPLEFT_LONGTEXT N_( \ + "Allows you to specify the left coordinate for the video cropping." ) +#define CROPBOTTOM_TEXT N_("Video crop bottom") +#define CROPBOTTOM_LONGTEXT N_( \ + "Allows you to specify the bottom coordinate for the video cropping." ) +#define CROPRIGHT_TEXT N_("Video crop right") +#define CROPRIGHT_LONGTEXT N_( \ + "Allows you to specify the right coordinate for the video cropping." ) + +#define AENC_TEXT N_("Audio encoder") +#define AENC_LONGTEXT N_( \ + "Allows you to specify the audio encoder to use and its associated " \ + "options." ) +#define ACODEC_TEXT N_("Destination audio codec") +#define ACODEC_LONGTEXT N_( \ + "Allows you to specify the destination audio codec used for the " \ + "streaming output." ) +#define AB_TEXT N_("Audio bitrate") +#define AB_LONGTEXT N_( \ + "Allows you to specify the audio bitrate used for the streaming " \ + "output." ) +#define ARATE_TEXT N_("Audio sample rate") +#define ARATE_LONGTEXT N_( \ + "Allows you to specify the audio sample rate used for the streaming " \ + "output." ) +#define ACHANS_TEXT N_("Audio channels") +#define ACHANS_LONGTEXT N_( \ + "Allows you to specify the number of audio channels used for the " \ + "streaming output." ) + +#define SENC_TEXT N_("Subtitles encoder") +#define SENC_LONGTEXT N_( \ + "Allows you to specify the subtitles encoder to use and its associated " \ + "options." ) +#define SCODEC_TEXT N_("Destination subtitles codec") +#define SCODEC_LONGTEXT N_( \ + "Allows you to specify the destination subtitles codec used for the " \ + "streaming output." ) +#define SFILTER_TEXT N_("Subpictures filter") +#define SFILTER_LONGTEXT N_( \ + "Allows you to specify subpictures filters used during the video " \ + "transcoding. The subpictures produced by the filters will be overlayed " \ + "directly onto the video." ) + +#define THREADS_TEXT N_("Number of threads") +#define THREADS_LONGTEXT N_( \ + "Allows you to specify the number of threads used for the transcoding." ) + +#define ASYNC_TEXT N_("Synchronise on audio track") +#define ASYNC_LONGTEXT N_( \ + "This option will drop/duplicate video frames to synchronise the video " \ + "track on the audio track." ) + +static int Open ( vlc_object_t * ); +static void Close( vlc_object_t * ); + +#define SOUT_CFG_PREFIX "sout-transcode-" + vlc_module_begin(); - set_description( _("Transcode stream") ); + set_description( _("Transcode stream output") ); set_capability( "sout stream", 50 ); add_shortcut( "transcode" ); set_callbacks( Open, Close ); + + add_string( SOUT_CFG_PREFIX "venc", NULL, NULL, VENC_TEXT, + VENC_LONGTEXT, VLC_FALSE ); + add_string( SOUT_CFG_PREFIX "vcodec", NULL, NULL, VCODEC_TEXT, + VCODEC_LONGTEXT, VLC_FALSE ); + add_integer( SOUT_CFG_PREFIX "vb", 800 * 1000, NULL, VB_TEXT, + VB_LONGTEXT, VLC_FALSE ); + add_float( SOUT_CFG_PREFIX "scale", 1, NULL, SCALE_TEXT, + SCALE_LONGTEXT, VLC_FALSE ); + add_float( SOUT_CFG_PREFIX "fps", 0, NULL, FPS_TEXT, + FPS_LONGTEXT, VLC_FALSE ); + add_bool( SOUT_CFG_PREFIX "deinterlace", 0, NULL, DEINTERLACE_TEXT, + DEINTERLACE_LONGTEXT, VLC_FALSE ); + add_integer( SOUT_CFG_PREFIX "width", 0, NULL, WIDTH_TEXT, + WIDTH_LONGTEXT, VLC_TRUE ); + add_integer( SOUT_CFG_PREFIX "height", 0, NULL, HEIGHT_TEXT, + HEIGHT_LONGTEXT, VLC_TRUE ); + + add_integer( SOUT_CFG_PREFIX "croptop", 0, NULL, CROPTOP_TEXT, + CROPTOP_LONGTEXT, VLC_TRUE ); + add_integer( SOUT_CFG_PREFIX "cropleft", 0, NULL, CROPLEFT_TEXT, + CROPLEFT_LONGTEXT, VLC_TRUE ); + add_integer( SOUT_CFG_PREFIX "cropbottom", 0, NULL, CROPBOTTOM_TEXT, + CROPBOTTOM_LONGTEXT, VLC_TRUE ); + add_integer( SOUT_CFG_PREFIX "cropright", 0, NULL, CROPRIGHT_TEXT, + CROPRIGHT_LONGTEXT, VLC_TRUE ); + + add_string( SOUT_CFG_PREFIX "aenc", NULL, NULL, AENC_TEXT, + AENC_LONGTEXT, VLC_FALSE ); + add_string( SOUT_CFG_PREFIX "acodec", NULL, NULL, ACODEC_TEXT, + ACODEC_LONGTEXT, VLC_FALSE ); + add_integer( SOUT_CFG_PREFIX "ab", 64000, NULL, AB_TEXT, + AB_LONGTEXT, VLC_FALSE ); + add_integer( SOUT_CFG_PREFIX "channels", 0, NULL, ACHANS_TEXT, + ACHANS_LONGTEXT, VLC_FALSE ); + add_integer( SOUT_CFG_PREFIX "samplerate", 0, NULL, ARATE_TEXT, + ARATE_LONGTEXT, VLC_TRUE ); + + add_string( SOUT_CFG_PREFIX "senc", NULL, NULL, SENC_TEXT, + SENC_LONGTEXT, VLC_FALSE ); + add_string( SOUT_CFG_PREFIX "scodec", NULL, NULL, SCODEC_TEXT, + SCODEC_LONGTEXT, VLC_FALSE ); + add_bool( SOUT_CFG_PREFIX "soverlay", 0, NULL, SCODEC_TEXT, + SCODEC_LONGTEXT, VLC_FALSE ); + add_string( SOUT_CFG_PREFIX "sfilter", NULL, NULL, SFILTER_TEXT, + SFILTER_LONGTEXT, VLC_FALSE ); + + add_integer( SOUT_CFG_PREFIX "threads", 0, NULL, THREADS_TEXT, + THREADS_LONGTEXT, VLC_TRUE ); + + add_bool( SOUT_CFG_PREFIX "audio-sync", 0, NULL, ASYNC_TEXT, + ASYNC_LONGTEXT, VLC_FALSE ); vlc_module_end(); +static const char *ppsz_sout_options[] = { + "venc", "vcodec", "vb", "croptop", "cropbottom", "cropleft", "cropright", + "scale", "fps", "width", "height", "deinterlace", "threads", + "aenc", "acodec", "ab", "samplerate", "channels", + "senc", "scodec", "soverlay", "sfilter", + "audio-sync", NULL +}; + +/***************************************************************************** + * Exported prototypes + *****************************************************************************/ +static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * ); +static int Del ( sout_stream_t *, sout_stream_id_t * ); +static int Send( sout_stream_t *, sout_stream_id_t *, block_t* ); + +static int transcode_audio_new ( sout_stream_t *, sout_stream_id_t * ); +static void transcode_audio_close ( sout_stream_t *, sout_stream_id_t * ); +static int transcode_audio_process( sout_stream_t *, sout_stream_id_t *, + block_t *, block_t ** ); + +static aout_buffer_t *audio_new_buffer( decoder_t *, int ); +static void audio_del_buffer( decoder_t *, aout_buffer_t * ); + +static int transcode_video_new ( sout_stream_t *, sout_stream_id_t * ); +static void transcode_video_close ( sout_stream_t *, sout_stream_id_t * ); +static int transcode_video_encoder_open( sout_stream_t *, sout_stream_id_t *); +static int transcode_video_process( sout_stream_t *, sout_stream_id_t *, + block_t *, block_t ** ); + +static void video_del_buffer( vlc_object_t *, picture_t * ); +static picture_t *video_new_buffer_decoder( decoder_t * ); +static void video_del_buffer_decoder( decoder_t *, picture_t * ); +static void video_link_picture_decoder( decoder_t *, picture_t * ); +static void video_unlink_picture_decoder( decoder_t *, picture_t * ); +static picture_t *video_new_buffer_filter( filter_t * ); +static void video_del_buffer_filter( filter_t *, picture_t * ); + +static int transcode_spu_new ( sout_stream_t *, sout_stream_id_t * ); +static void transcode_spu_close ( sout_stream_t *, sout_stream_id_t * ); +static int transcode_spu_process( sout_stream_t *, sout_stream_id_t *, + block_t *, block_t ** ); + +static int EncoderThread( struct sout_stream_sys_t * p_sys ); + +static int pi_channels_maps[6] = +{ + 0, + AOUT_CHAN_CENTER, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT, + AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT, + AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT + | AOUT_CHAN_REARRIGHT, + AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER + | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT +}; + +#define PICTURE_RING_SIZE 64 +#define SUBPICTURE_RING_SIZE 20 + struct sout_stream_sys_t { - sout_stream_t *p_out; + VLC_COMMON_MEMBERS + sout_stream_t *p_out; + sout_stream_id_t *id_video; + block_t *p_buffers; + vlc_mutex_t lock_out; + vlc_cond_t cond; + picture_t * pp_pics[PICTURE_RING_SIZE]; + int i_first_pic, i_last_pic; + + /* Audio */ vlc_fourcc_t i_acodec; /* codec audio (0 if not transcode) */ + char *psz_aenc; + sout_cfg_t *p_audio_cfg; int i_sample_rate; int i_channels; int i_abitrate; - vlc_fourcc_t i_vcodec; /* " video " " " " */ + /* Video */ + vlc_fourcc_t i_vcodec; /* codec video (0 if not transcode) */ + char *psz_venc; + sout_cfg_t *p_video_cfg; int i_vbitrate; - int i_vtolerance; + double f_scale; + double f_fps; int i_width; int i_height; - int i_b_frames; - int i_key_int; - int i_qmin; - int i_qmax; - vlc_bool_t i_hq; vlc_bool_t b_deinterlace; + int i_threads; int i_crop_top; int i_crop_bottom; int i_crop_right; int i_crop_left; - mtime_t i_input_pts; - mtime_t i_output_pts; - mtime_t i_last_ref_pts; + /* SPU */ + vlc_fourcc_t i_scodec; /* codec spu (0 if not transcode) */ + char *psz_senc; + vlc_bool_t b_soverlay; + sout_cfg_t *p_spu_cfg; + spu_t *p_spu; + + /* Sync */ + vlc_bool_t b_audio_sync; + mtime_t i_master_drift; +}; - mtime_t i_buggy_pts_detect; +struct decoder_owner_sys_t +{ + picture_t *pp_pics[PICTURE_RING_SIZE]; +}; +struct filter_owner_sys_t +{ + picture_t *pp_pics[PICTURE_RING_SIZE]; }; /***************************************************************************** @@ -118,1639 +311,1596 @@ static int Open( vlc_object_t *p_this ) { sout_stream_t *p_stream = (sout_stream_t*)p_this; sout_stream_sys_t *p_sys; - char *codec; - - p_sys = malloc( sizeof( sout_stream_sys_t ) ); - p_sys->p_out = sout_stream_new( p_stream->p_sout, p_stream->psz_next ); - - p_sys->i_acodec = 0; - p_sys->i_sample_rate = 0; - p_sys->i_channels = 0; - p_sys->i_abitrate = 0; - - p_sys->i_vcodec = 0; - p_sys->i_vbitrate = 0; - p_sys->i_vtolerance = -1; - p_sys->i_width = 0; - p_sys->i_height = 0; - p_sys->i_key_int = -1; - p_sys->i_b_frames = 0; - p_sys->i_qmin = 2; - p_sys->i_qmax = 31; -#if LIBAVCODEC_BUILD >= 4673 - p_sys->i_hq = FF_MB_DECISION_SIMPLE; -#else - p_sys->i_hq = VLC_FALSE; -#endif - p_sys->b_deinterlace= VLC_FALSE; + vlc_value_t val; - p_sys->i_crop_top = 0; - p_sys->i_crop_bottom= 0; - p_sys->i_crop_right = 0; - p_sys->i_crop_left = 0; + p_sys = vlc_object_create( p_this, sizeof( sout_stream_sys_t ) ); - if( ( codec = sout_cfg_find_value( p_stream->p_cfg, "acodec" ) ) ) + p_sys->p_out = sout_StreamNew( p_stream->p_sout, p_stream->psz_next ); + if( !p_sys->p_out ) { - char fcc[4] = " "; - char *val; + msg_Err( p_stream, "cannot create chain" ); + free( p_sys ); + return VLC_EGENERIC; + } + + p_sys->i_master_drift = 0; - memcpy( fcc, codec, strlen( codec ) ); + sout_CfgParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options, + p_stream->p_cfg ); + /* Audio transcoding parameters */ + var_Get( p_stream, SOUT_CFG_PREFIX "aenc", &val ); + p_sys->psz_aenc = NULL; + p_sys->p_audio_cfg = NULL; + if( val.psz_string && *val.psz_string ) + { + char *psz_next; + psz_next = sout_CfgCreate( &p_sys->psz_aenc, &p_sys->p_audio_cfg, + val.psz_string ); + if( psz_next ) free( psz_next ); + } + if( val.psz_string ) free( val.psz_string ); + + var_Get( p_stream, SOUT_CFG_PREFIX "acodec", &val ); + p_sys->i_acodec = 0; + if( val.psz_string && *val.psz_string ) + { + char fcc[4] = " "; + memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) ); p_sys->i_acodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] ); + } + if( val.psz_string ) free( val.psz_string ); - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "samplerate" ) ) ) - { - p_sys->i_sample_rate = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "channels" ) ) ) - { - p_sys->i_channels = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "ab" ) ) ) - { - p_sys->i_abitrate = atoi( val ); - if( p_sys->i_abitrate < 4000 ) - { - p_sys->i_abitrate *= 1000; - } - } + var_Get( p_stream, SOUT_CFG_PREFIX "ab", &val ); + p_sys->i_abitrate = val.i_int; + if( p_sys->i_abitrate < 4000 ) p_sys->i_abitrate *= 1000; + + var_Get( p_stream, SOUT_CFG_PREFIX "samplerate", &val ); + p_sys->i_sample_rate = val.i_int; + + var_Get( p_stream, SOUT_CFG_PREFIX "channels", &val ); + p_sys->i_channels = val.i_int; - msg_Dbg( p_stream, "codec audio=%4.4s %dHz %d channels %dKb/s", fcc, - p_sys->i_sample_rate, p_sys->i_channels, - p_sys->i_abitrate / 1024 ); + if( p_sys->i_acodec ) + { + msg_Dbg( p_stream, "codec audio=%4.4s %dHz %d channels %dKb/s", + (char *)&p_sys->i_acodec, p_sys->i_sample_rate, + p_sys->i_channels, p_sys->i_abitrate / 1000 ); + } + + /* Video transcoding parameters */ + var_Get( p_stream, SOUT_CFG_PREFIX "venc", &val ); + p_sys->psz_venc = NULL; + p_sys->p_video_cfg = NULL; + if( val.psz_string && *val.psz_string ) + { + char *psz_next; + psz_next = sout_CfgCreate( &p_sys->psz_venc, &p_sys->p_video_cfg, + val.psz_string ); + if( psz_next ) free( psz_next ); } + if( val.psz_string ) free( val.psz_string ); - if( ( codec = sout_cfg_find_value( p_stream->p_cfg, "vcodec" ) ) ) + var_Get( p_stream, SOUT_CFG_PREFIX "vcodec", &val ); + p_sys->i_vcodec = 0; + if( val.psz_string && *val.psz_string ) { char fcc[4] = " "; - char *val; + memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) ); + p_sys->i_vcodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] ); + } + if( val.psz_string ) free( val.psz_string ); - memcpy( fcc, codec, strlen( codec ) ); + var_Get( p_stream, SOUT_CFG_PREFIX "vb", &val ); + p_sys->i_vbitrate = val.i_int; + if( p_sys->i_vbitrate < 16000 ) p_sys->i_vbitrate *= 1000; - p_sys->i_vcodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] ); + var_Get( p_stream, SOUT_CFG_PREFIX "scale", &val ); + p_sys->f_scale = val.f_float; - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "width" ) ) ) - { - p_sys->i_width = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "height" ) ) ) - { - p_sys->i_height = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "vb" ) ) ) - { - p_sys->i_vbitrate = atoi( val ); - if( p_sys->i_vbitrate < 16000 ) - { - p_sys->i_vbitrate *= 1000; - } - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "vt" ) ) ) - { - p_sys->i_vtolerance = atoi( val ); - } - if( sout_cfg_find( p_stream->p_cfg, "deinterlace" ) ) - { - p_sys->b_deinterlace = VLC_TRUE; - } - /* crop */ - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "croptop" ) ) ) - { - p_sys->i_crop_top = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "cropbottom" ) ) ) - { - p_sys->i_crop_bottom = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "cropleft" ) ) ) - { - p_sys->i_crop_left = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "cropright" ) ) ) - { - p_sys->i_crop_right = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "keyint" ) ) ) - { - p_sys->i_key_int = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "bframes" ) ) ) - { - p_sys->i_b_frames = atoi( val ); - } -#if LIBAVCODEC_BUILD >= 4673 - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "hq" ) ) ) - { - if( !strcmp( val, "rd" ) ) - { - p_sys->i_hq = FF_MB_DECISION_RD; - } - else if( !strcmp( val, "bits" ) ) - { - p_sys->i_hq = FF_MB_DECISION_BITS; - } - else if( !strcmp( val, "simple" ) ) - { - p_sys->i_hq = FF_MB_DECISION_SIMPLE; - } - else - { - p_sys->i_hq = FF_MB_DECISION_RD; - } - } -#else - if( sout_cfg_find( p_stream->p_cfg, "hq" ) ) - { - p_sys->i_hq = VLC_TRUE; - } -#endif - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "qmin" ) ) ) - { - p_sys->i_qmin = atoi( val ); - } - if( ( val = sout_cfg_find_value( p_stream->p_cfg, "qmax" ) ) ) - { - p_sys->i_qmax = atoi( val ); - } + var_Get( p_stream, SOUT_CFG_PREFIX "fps", &val ); + p_sys->f_fps = val.f_float; + + var_Get( p_stream, SOUT_CFG_PREFIX "width", &val ); + p_sys->i_width = val.i_int; + + var_Get( p_stream, SOUT_CFG_PREFIX "height", &val ); + p_sys->i_height = val.i_int; + + var_Get( p_stream, SOUT_CFG_PREFIX "deinterlace", &val ); + p_sys->b_deinterlace = val.b_bool; - msg_Dbg( p_stream, "codec video=%4.4s %dx%d %dkb/s", - fcc, - p_sys->i_width, p_sys->i_height, - p_sys->i_vbitrate / 1024 ); + var_Get( p_stream, SOUT_CFG_PREFIX "croptop", &val ); + p_sys->i_crop_top = val.i_int; + var_Get( p_stream, SOUT_CFG_PREFIX "cropbottom", &val ); + p_sys->i_crop_bottom = val.i_int; + var_Get( p_stream, SOUT_CFG_PREFIX "cropleft", &val ); + p_sys->i_crop_left = val.i_int; + var_Get( p_stream, SOUT_CFG_PREFIX "cropright", &val ); + p_sys->i_crop_right = val.i_int; + + var_Get( p_stream, SOUT_CFG_PREFIX "threads", &val ); + p_sys->i_threads = val.i_int; + + if( p_sys->i_vcodec ) + { + msg_Dbg( p_stream, "codec video=%4.4s %dx%d scaling: %f %dkb/s", + (char *)&p_sys->i_vcodec, p_sys->i_width, p_sys->i_height, + p_sys->f_scale, p_sys->i_vbitrate / 1000 ); } - if( !p_sys->p_out ) + /* Subpictures transcoding parameters */ + p_sys->p_spu = 0; + p_sys->psz_senc = NULL; + p_sys->p_spu_cfg = NULL; + p_sys->i_scodec = 0; + + var_Get( p_stream, SOUT_CFG_PREFIX "senc", &val ); + if( val.psz_string && *val.psz_string ) { - msg_Err( p_stream, "cannot create chain" ); - free( p_sys ); - return VLC_EGENERIC; + char *psz_next; + psz_next = sout_CfgCreate( &p_sys->psz_senc, &p_sys->p_spu_cfg, + val.psz_string ); + if( psz_next ) free( psz_next ); + } + if( val.psz_string ) free( val.psz_string ); + + var_Get( p_stream, SOUT_CFG_PREFIX "scodec", &val ); + if( val.psz_string && *val.psz_string ) + { + char fcc[4] = " "; + memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) ); + p_sys->i_scodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] ); } + if( val.psz_string ) free( val.psz_string ); + + if( p_sys->i_scodec ) + { + msg_Dbg( p_stream, "codec spu=%4.4s", (char *)&p_sys->i_acodec ); + } + + var_Get( p_stream, SOUT_CFG_PREFIX "soverlay", &val ); + p_sys->b_soverlay = val.b_bool; + + var_Get( p_stream, SOUT_CFG_PREFIX "sfilter", &val ); + if( val.psz_string && *val.psz_string ) + { + p_sys->p_spu = spu_Create( p_stream ); + var_Create( p_sys->p_spu, "sub-filter", VLC_VAR_STRING ); + var_Set( p_sys->p_spu, "sub-filter", val ); + spu_Init( p_sys->p_spu ); + } + if( val.psz_string ) free( val.psz_string ); + + var_Get( p_stream, SOUT_CFG_PREFIX "audio-sync", &val ); + p_sys->b_audio_sync = val.b_bool; + if( p_sys->f_fps > 0 ) p_sys->b_audio_sync = VLC_TRUE; + p_stream->pf_add = Add; p_stream->pf_del = Del; p_stream->pf_send = Send; - p_stream->p_sys = p_sys; - avcodec_init(); - avcodec_register_all(); - - /* ffmpeg needs some padding at the end of each buffer */ - p_stream->p_sout->i_padding += FF_INPUT_BUFFER_PADDING_SIZE; - return VLC_SUCCESS; } /***************************************************************************** * Close: *****************************************************************************/ - static void Close( vlc_object_t * p_this ) { sout_stream_t *p_stream = (sout_stream_t*)p_this; sout_stream_sys_t *p_sys = p_stream->p_sys; - sout_stream_delete( p_sys->p_out ); - free( p_sys ); -} + sout_StreamDelete( p_sys->p_out ); -struct sout_stream_id_t -{ - vlc_fourcc_t b_transcode; - sout_format_t f_src; /* only if transcoding */ - sout_format_t f_dst; /* " " " */ + while( p_sys->p_audio_cfg != NULL ) + { + sout_cfg_t *p_next = p_sys->p_audio_cfg->p_next; - /* id of the out stream */ - void *id; + if( p_sys->p_audio_cfg->psz_name ) + free( p_sys->p_audio_cfg->psz_name ); + if( p_sys->p_audio_cfg->psz_value ) + free( p_sys->p_audio_cfg->psz_value ); + free( p_sys->p_audio_cfg ); - /* Encoder */ - encoder_t *p_encoder; + p_sys->p_audio_cfg = p_next; + } + if( p_sys->psz_aenc ) free( p_sys->psz_aenc ); + + while( p_sys->p_video_cfg != NULL ) + { + sout_cfg_t *p_next = p_sys->p_video_cfg->p_next; - /* ffmpeg part */ - AVCodec *ff_dec; - AVCodecContext *ff_dec_c; + if( p_sys->p_video_cfg->psz_name ) + free( p_sys->p_video_cfg->psz_name ); + if( p_sys->p_video_cfg->psz_value ) + free( p_sys->p_video_cfg->psz_value ); + free( p_sys->p_video_cfg ); + p_sys->p_video_cfg = p_next; + } + if( p_sys->psz_venc ) free( p_sys->psz_venc ); + + while( p_sys->p_spu_cfg != NULL ) + { + sout_cfg_t *p_next = p_sys->p_spu_cfg->p_next; + + if( p_sys->p_spu_cfg->psz_name ) + free( p_sys->p_spu_cfg->psz_name ); + if( p_sys->p_spu_cfg->psz_value ) + free( p_sys->p_spu_cfg->psz_value ); + free( p_sys->p_spu_cfg ); - vlc_fourcc_t b_enc_inited; - AVCodec *ff_enc; - AVCodecContext *ff_enc_c; + p_sys->p_spu_cfg = p_next; + } + if( p_sys->psz_senc ) free( p_sys->psz_senc ); - mtime_t i_dts; - mtime_t i_length; + if( p_sys->p_spu ) spu_Destroy( p_sys->p_spu ); - int i_buffer_in; - int i_buffer_in_pos; - uint8_t *p_buffer_in; + vlc_object_destroy( p_sys ); +} - int i_buffer; - int i_buffer_pos; - uint8_t *p_buffer; +struct sout_stream_id_t +{ + vlc_fourcc_t b_transcode; - int i_buffer_out; - int i_buffer_out_pos; - uint8_t *p_buffer_out; + /* id of the out stream */ + void *id; - AVFrame *p_ff_pic; - AVFrame *p_ff_pic_tmp0; /* to do deinterlace */ - AVFrame *p_ff_pic_tmp1; /* to do pix conversion */ - AVFrame *p_ff_pic_tmp2; /* to do resample */ + /* Decoder */ + decoder_t *p_decoder; - ImgReSampleContext *p_vresample; + /* Filters */ + filter_t *pp_filter[10]; + int i_filter; -#ifdef HAVE_VORBIS_VORBISENC_H + /* Encoder */ + encoder_t *p_encoder; - /* Vorbis part */ - vorbis_info *p_vi; - vorbis_dsp_state *p_vd; - vorbis_block *p_vb; - vorbis_comment *p_vc; - int i_last_block_size; - int i_samples_delay; - vlc_bool_t b_headers_sent; -#endif + /* Sync */ + date_t interpolated_pts; }; -static sout_stream_id_t * Add( sout_stream_t *p_stream, sout_format_t *p_fmt ) +static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt ) { - sout_stream_sys_t *p_sys = p_stream->p_sys; - sout_stream_id_t *id; + sout_stream_sys_t *p_sys = p_stream->p_sys; + sout_stream_id_t *id; id = malloc( sizeof( sout_stream_id_t ) ); - id->i_dts = 0; + memset( id, 0, sizeof(sout_stream_id_t) ); + id->id = NULL; + id->p_decoder = NULL; id->p_encoder = NULL; - if( p_fmt->i_cat == AUDIO_ES && p_sys->i_acodec != 0 ) + /* Create decoder object */ + id->p_decoder = vlc_object_create( p_stream, VLC_OBJECT_DECODER ); + if( !id->p_decoder ) + { + msg_Err( p_stream, "out of memory" ); + goto error; + } + vlc_object_attach( id->p_decoder, p_stream ); + id->p_decoder->p_module = NULL; + id->p_decoder->fmt_in = *p_fmt; + id->p_decoder->fmt_out = *p_fmt; + id->p_decoder->fmt_out.i_extra = 0; + id->p_decoder->fmt_out.p_extra = 0; + id->p_decoder->b_pace_control = VLC_TRUE; + + /* Create encoder object */ + id->p_encoder = vlc_object_create( p_stream, VLC_OBJECT_ENCODER ); + if( !id->p_encoder ) + { + msg_Err( p_stream, "out of memory" ); + goto error; + } + vlc_object_attach( id->p_encoder, p_stream ); + id->p_encoder->p_module = NULL; + + /* Create destination format */ + es_format_Init( &id->p_encoder->fmt_out, p_fmt->i_cat, 0 ); + id->p_encoder->fmt_out.i_id = p_fmt->i_id; + id->p_encoder->fmt_out.i_group = p_fmt->i_group; + if( p_fmt->psz_language ) + id->p_encoder->fmt_out.psz_language = strdup( p_fmt->psz_language ); + + if( p_fmt->i_cat == AUDIO_ES && (p_sys->i_acodec || p_sys->psz_aenc) ) { msg_Dbg( p_stream, "creating audio transcoding from fcc=`%4.4s' to fcc=`%4.4s'", - (char*)&p_fmt->i_fourcc, - (char*)&p_sys->i_acodec ); - - /* src format */ - memcpy( &id->f_src, p_fmt, sizeof( sout_format_t ) ); - - /* create dst format */ - id->f_dst.i_cat = AUDIO_ES; - id->f_dst.i_fourcc = p_sys->i_acodec; - id->f_dst.i_sample_rate = p_sys->i_sample_rate > 0 ? p_sys->i_sample_rate : id->f_src.i_sample_rate; - id->f_dst.i_channels = p_sys->i_channels > 0 ? p_sys->i_channels : id->f_src.i_channels; - id->f_dst.i_bitrate = p_sys->i_abitrate > 0 ? p_sys->i_abitrate : 64000; - id->f_dst.i_block_align = 0; - id->f_dst.i_extra_data = 0; - id->f_dst.p_extra_data = NULL; - - /* build decoder -> filter -> encoder */ - if( transcode_audio_ffmpeg_new( p_stream, id ) ) + (char*)&p_fmt->i_codec, (char*)&p_sys->i_acodec ); + + /* Complete destination format */ + id->p_encoder->fmt_out.i_codec = p_sys->i_acodec; + id->p_encoder->fmt_out.audio.i_rate = p_sys->i_sample_rate > 0 ? + p_sys->i_sample_rate : (int)p_fmt->audio.i_rate; + id->p_encoder->fmt_out.audio.i_channels = p_sys->i_channels > 0 ? + p_sys->i_channels : p_fmt->audio.i_channels; + id->p_encoder->fmt_out.i_bitrate = p_sys->i_abitrate; + id->p_encoder->fmt_out.audio.i_bitspersample = + p_fmt->audio.i_bitspersample; + + /* Build decoder -> filter -> encoder chain */ + if( transcode_audio_new( p_stream, id ) ) { msg_Err( p_stream, "cannot create audio chain" ); - free( id ); - return NULL; + goto error; } - /* open output stream */ - id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->f_dst ); + /* Open output stream */ + id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->p_encoder->fmt_out ); id->b_transcode = VLC_TRUE; - if( id->id == NULL ) - { - free( id ); - return NULL; - } + if( !id->id ) goto error; + + date_Init( &id->interpolated_pts, p_fmt->audio.i_rate, 1 ); } - else if( p_fmt->i_cat == VIDEO_ES && p_sys->i_vcodec != 0 ) + else if( p_fmt->i_cat == VIDEO_ES && + (p_sys->i_vcodec != 0 || p_sys->psz_venc) ) { msg_Dbg( p_stream, "creating video transcoding from fcc=`%4.4s' to fcc=`%4.4s'", - (char*)&p_fmt->i_fourcc, - (char*)&p_sys->i_vcodec ); + (char*)&p_fmt->i_codec, (char*)&p_sys->i_vcodec ); + + /* Complete destination format */ + id->p_encoder->fmt_out.i_codec = p_sys->i_vcodec; + id->p_encoder->fmt_out.video.i_width = p_sys->i_width; + id->p_encoder->fmt_out.video.i_height = p_sys->i_height; + id->p_encoder->fmt_out.i_bitrate = p_sys->i_vbitrate; + + /* Build decoder -> filter -> encoder chain */ + if( transcode_video_new( p_stream, id ) ) + { + msg_Err( p_stream, "cannot create video chain" ); + goto error; + } - memcpy( &id->f_src, p_fmt, sizeof( sout_format_t ) ); + /* Stream will be added later on because we don't know + * all the characteristics of the decoded stream yet */ + id->b_transcode = VLC_TRUE; + + if( p_sys->f_fps > 0 ) + { + id->p_encoder->fmt_out.video.i_frame_rate = p_sys->f_fps * 1000; + id->p_encoder->fmt_out.video.i_frame_rate_base = 1000; + } + } + else if( p_fmt->i_cat == SPU_ES && (p_sys->i_scodec || p_sys->psz_senc) ) + { + msg_Dbg( p_stream, "creating subtitles transcoding from fcc=`%4.4s' " + "to fcc=`%4.4s'", (char*)&p_fmt->i_codec, + (char*)&p_sys->i_scodec ); - /* create dst format */ - id->f_dst.i_cat = VIDEO_ES; - id->f_dst.i_fourcc = p_sys->i_vcodec; - id->f_dst.i_width = p_sys->i_width ; /* > 0 ? p_sys->i_width : id->f_src.i_width; */ - id->f_dst.i_height = p_sys->i_height; /* > 0 ? p_sys->i_height: id->f_src.i_height; */ - id->f_dst.i_bitrate = p_sys->i_vbitrate > 0 ? p_sys->i_vbitrate : 800*1000; - id->f_dst.i_extra_data = 0; - id->f_dst.p_extra_data = NULL; + /* Complete destination format */ + id->p_encoder->fmt_out.i_codec = p_sys->i_scodec; /* build decoder -> filter -> encoder */ - if( transcode_video_ffmpeg_new( p_stream, id ) ) + if( transcode_spu_new( p_stream, id ) ) { - msg_Err( p_stream, "cannot create video chain" ); - free( id ); - return NULL; + msg_Err( p_stream, "cannot create subtitles chain" ); + goto error; } -#if 0 + /* open output stream */ - id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->f_dst ); -#endif + id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->p_encoder->fmt_out ); + id->b_transcode = VLC_TRUE; + + if( !id->id ) goto error; + } + else if( p_fmt->i_cat == SPU_ES && p_sys->b_soverlay ) + { + msg_Dbg( p_stream, "subtitles (fcc=`%4.4s') overlaying", + (char*)&p_fmt->i_codec ); + id->b_transcode = VLC_TRUE; + + /* Build decoder -> filter -> overlaying chain */ + if( transcode_spu_new( p_stream, id ) ) + { + msg_Err( p_stream, "cannot create subtitles chain" ); + goto error; + } } else { - msg_Dbg( p_stream, "not transcoding a stream (fcc=`%4.4s')", (char*)&p_fmt->i_fourcc ); + msg_Dbg( p_stream, "not transcoding a stream (fcc=`%4.4s')", + (char*)&p_fmt->i_codec ); id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt ); id->b_transcode = VLC_FALSE; - if( id->id == NULL ) - { - free( id ); - return NULL; - } + if( !id->id ) goto error; } return id; + + error: + if( id->p_decoder ) + { + vlc_object_detach( id->p_decoder ); + vlc_object_destroy( id->p_decoder ); + } + + if( id->p_encoder ) + { + vlc_object_detach( id->p_encoder ); + vlc_object_destroy( id->p_encoder ); + } + + free( id ); + return NULL; } -static int Del ( sout_stream_t *p_stream, sout_stream_id_t *id ) +static int Del( sout_stream_t *p_stream, sout_stream_id_t *id ) { - sout_stream_sys_t *p_sys = p_stream->p_sys; + sout_stream_sys_t *p_sys = p_stream->p_sys; if( id->b_transcode ) { - if( id->f_src.i_cat == AUDIO_ES ) - { - transcode_audio_ffmpeg_close( p_stream, id ); - } - else if( id->f_src.i_cat == VIDEO_ES ) + switch( id->p_decoder->fmt_in.i_cat ) { - transcode_video_ffmpeg_close( p_stream, id ); + case AUDIO_ES: + transcode_audio_close( p_stream, id ); + break; + case VIDEO_ES: + transcode_video_close( p_stream, id ); + break; + case SPU_ES: + transcode_spu_close( p_stream, id ); + break; } } if( id->id ) p_sys->p_out->pf_del( p_sys->p_out, id->id ); + + if( id->p_decoder ) + { + vlc_object_detach( id->p_decoder ); + vlc_object_destroy( id->p_decoder ); + } + + if( id->p_encoder ) + { + vlc_object_detach( id->p_encoder ); + vlc_object_destroy( id->p_encoder ); + } + free( id ); return VLC_SUCCESS; } static int Send( sout_stream_t *p_stream, sout_stream_id_t *id, - sout_buffer_t *p_buffer ) + block_t *p_buffer ) { - sout_stream_sys_t *p_sys = p_stream->p_sys; - - if( id->b_transcode ) - { - sout_buffer_t *p_buffer_out; - if( id->f_src.i_cat == AUDIO_ES ) - { - transcode_audio_ffmpeg_process( p_stream, id, p_buffer, &p_buffer_out ); - } - else if( id->f_src.i_cat == VIDEO_ES ) - { - transcode_video_ffmpeg_process( p_stream, id, p_buffer, &p_buffer_out ); - } - sout_BufferDelete( p_stream->p_sout, p_buffer ); + sout_stream_sys_t *p_sys = p_stream->p_sys; + block_t *p_out; - if( p_buffer_out ) - { - return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer_out ); - } - return VLC_SUCCESS; - } - else if( id->id != NULL ) + if( !id->b_transcode && id->id ) { return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer ); } - else + else if( !id->b_transcode ) { - sout_BufferDelete( p_stream->p_sout, p_buffer ); + block_Release( p_buffer ); return VLC_EGENERIC; } -} - -/**************************************************************************** - * ffmpeg decoder reencocdr part - ****************************************************************************/ -static struct -{ - vlc_fourcc_t i_fcc; - int i_ff_codec; -} fourcc_to_ff_code[] = -{ - /* audio */ - { VLC_FOURCC( 'm', 'p', 'g', 'a' ), CODEC_ID_MP2 }, - { VLC_FOURCC( 'm', 'p', '3', ' ' ), CODEC_ID_MP3LAME }, - { VLC_FOURCC( 'm', 'p', '4', 'a' ), CODEC_ID_AAC }, - { VLC_FOURCC( 'a', '5', '2', ' ' ), CODEC_ID_AC3 }, - { VLC_FOURCC( 'a', 'c', '3', ' ' ), CODEC_ID_AC3 }, - { VLC_FOURCC( 'w', 'm', 'a', '1' ), CODEC_ID_WMAV1 }, - { VLC_FOURCC( 'w', 'm', 'a', '2' ), CODEC_ID_WMAV2 }, - { VLC_FOURCC( 'v', 'o', 'r', 'b' ), CODEC_ID_VORBIS }, - { VLC_FOURCC( 'a', 'l', 'a', 'w' ), CODEC_ID_PCM_ALAW }, - - /* video */ - { VLC_FOURCC( 'm', 'p', 'g', 'v' ), CODEC_ID_MPEG1VIDEO }, - { VLC_FOURCC( 'm', 'p', '1', 'v' ), CODEC_ID_MPEG1VIDEO }, -#if LIBAVCODEC_BUILD >= 4676 - { VLC_FOURCC( 'm', 'p', '2', 'v' ), CODEC_ID_MPEG2VIDEO }, -#endif - { VLC_FOURCC( 'm', 'p', '4', 'v'), CODEC_ID_MPEG4 }, - { VLC_FOURCC( 'D', 'I', 'V', '1' ), CODEC_ID_MSMPEG4V1 }, - { VLC_FOURCC( 'D', 'I', 'V', '2' ), CODEC_ID_MSMPEG4V2 }, - { VLC_FOURCC( 'D', 'I', 'V', '3' ), CODEC_ID_MSMPEG4V3 }, - { VLC_FOURCC( 'H', '2', '6', '3' ), CODEC_ID_H263 }, - { VLC_FOURCC( 'I', '2', '6', '3' ), CODEC_ID_H263I }, - { VLC_FOURCC( 'h', 'u', 'f', 'f' ), CODEC_ID_HUFFYUV }, - { VLC_FOURCC( 'W', 'M', 'V', '1' ), CODEC_ID_WMV1 }, - { VLC_FOURCC( 'W', 'M', 'V', '2' ), CODEC_ID_WMV2 }, - { VLC_FOURCC( 'M', 'J', 'P', 'G' ), CODEC_ID_MJPEG }, - { VLC_FOURCC( 'm', 'j', 'p', 'b' ), CODEC_ID_MJPEGB }, - { VLC_FOURCC( 'd', 'v', 's', 'l' ), CODEC_ID_DVVIDEO }, - { VLC_FOURCC( 'S', 'V', 'Q', '1' ), CODEC_ID_SVQ1 }, -#if LIBAVCODEC_BUILD >= 4666 - { VLC_FOURCC( 'S', 'V', 'Q', '3' ), CODEC_ID_SVQ3 }, -#endif - - /* raw video code, only used for 'encoding' */ - { VLC_FOURCC( 'I', '4', '2', '0' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'I', '4', '2', '2' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'I', '4', '4', '4' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'R', 'V', '1', '5' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'R', 'V', '1', '6' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'R', 'V', '2', '4' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'R', 'V', '3', '2' ), CODEC_ID_RAWVIDEO }, - { VLC_FOURCC( 'Y', 'U', 'Y', '2' ), CODEC_ID_RAWVIDEO }, - - { VLC_FOURCC( 0, 0, 0, 0 ), 0 } -}; + switch( id->p_decoder->fmt_in.i_cat ) + { + case AUDIO_ES: + transcode_audio_process( p_stream, id, p_buffer, &p_out ); + break; -static inline int get_ff_codec( vlc_fourcc_t i_fcc ) -{ - int i; + case VIDEO_ES: + if( transcode_video_process( p_stream, id, p_buffer, &p_out ) + != VLC_SUCCESS ) + { + return VLC_EGENERIC; + } + break; - for( i = 0; fourcc_to_ff_code[i].i_fcc != 0; i++ ) - { - if( fourcc_to_ff_code[i].i_fcc == i_fcc ) + case SPU_ES: + if( transcode_spu_process( p_stream, id, p_buffer, &p_out ) != + VLC_SUCCESS ) { - return fourcc_to_ff_code[i].i_ff_codec; + return VLC_EGENERIC; } + break; + + default: + block_Release( p_buffer ); + break; } - return 0; + if( p_out ) return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_out ); + return VLC_SUCCESS; } -static inline int get_ff_chroma( vlc_fourcc_t i_chroma ) +/**************************************************************************** + * decoder reencoder part + ****************************************************************************/ +int audio_BitsPerSample( vlc_fourcc_t i_format ) { - switch( i_chroma ) - { - case VLC_FOURCC( 'I', '4', '2', '0' ): - return PIX_FMT_YUV420P; - case VLC_FOURCC( 'I', '4', '2', '2' ): - return PIX_FMT_YUV422P; - case VLC_FOURCC( 'I', '4', '4', '4' ): - return PIX_FMT_YUV444P; - case VLC_FOURCC( 'R', 'V', '1', '5' ): - return PIX_FMT_RGB555; - case VLC_FOURCC( 'R', 'V', '1', '6' ): - return PIX_FMT_RGB565; - case VLC_FOURCC( 'R', 'V', '2', '4' ): - return PIX_FMT_RGB24; - case VLC_FOURCC( 'R', 'V', '3', '2' ): - return PIX_FMT_RGBA32; - case VLC_FOURCC( 'G', 'R', 'E', 'Y' ): - return PIX_FMT_GRAY8; - case VLC_FOURCC( 'Y', 'U', 'Y', '2' ): - return PIX_FMT_YUV422; - default: - return 0; + switch( i_format ) + { + case VLC_FOURCC('u','8',' ',' '): + case VLC_FOURCC('s','8',' ',' '): + return 8; + + case VLC_FOURCC('u','1','6','l'): + case VLC_FOURCC('s','1','6','l'): + case VLC_FOURCC('u','1','6','b'): + case VLC_FOURCC('s','1','6','b'): + return 16; + + case VLC_FOURCC('u','2','4','l'): + case VLC_FOURCC('s','2','4','l'): + case VLC_FOURCC('u','2','4','b'): + case VLC_FOURCC('s','2','4','b'): + return 24; + + case VLC_FOURCC('u','3','2','l'): + case VLC_FOURCC('s','3','2','l'): + case VLC_FOURCC('u','3','2','b'): + case VLC_FOURCC('s','3','2','b'): + case VLC_FOURCC('f','l','3','2'): + case VLC_FOURCC('f','i','3','2'): + return 32; + + case VLC_FOURCC('f','l','6','4'): + return 64; } + + return 0; } -static int transcode_audio_ffmpeg_new( sout_stream_t *p_stream, - sout_stream_id_t *id ) +static int transcode_audio_new( sout_stream_t *p_stream, + sout_stream_id_t *id ) { - int i_ff_codec; + sout_stream_sys_t *p_sys = p_stream->p_sys; - if( id->f_src.i_fourcc == VLC_FOURCC('s','1','6','l') || - id->f_src.i_fourcc == VLC_FOURCC('s','1','6','b') || - id->f_src.i_fourcc == VLC_FOURCC('s','8',' ',' ') || - id->f_src.i_fourcc == VLC_FOURCC('u','8',' ',' ') ) - { - id->ff_dec = NULL; + /* + * Open decoder + */ - id->ff_dec_c = avcodec_alloc_context(); - id->ff_dec_c->sample_rate = id->f_src.i_sample_rate; - id->ff_dec_c->channels = id->f_src.i_channels; - id->ff_dec_c->block_align = id->f_src.i_block_align; - id->ff_dec_c->bit_rate = id->f_src.i_bitrate; - } + /* Initialization of decoder structures */ + id->p_decoder->pf_decode_audio = 0; + id->p_decoder->pf_aout_buffer_new = audio_new_buffer; + id->p_decoder->pf_aout_buffer_del = audio_del_buffer; + //id->p_decoder->p_cfg = p_sys->p_video_cfg; + + id->p_decoder->p_module = + module_Need( id->p_decoder, "decoder", "$codec", 0 ); + + if( !id->p_decoder->p_module ) + { + msg_Err( p_stream, "cannot find decoder" ); + return VLC_EGENERIC; + } + id->p_decoder->fmt_out.audio.i_bitspersample = + audio_BitsPerSample( id->p_decoder->fmt_out.i_codec ); + + /* + * Open encoder + */ + + /* Initialization of encoder format structures */ + es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat, + id->p_decoder->fmt_out.i_codec ); + id->p_encoder->fmt_in.audio.i_format = id->p_decoder->fmt_out.i_codec; + + /* Sanity check for audio channels */ + id->p_encoder->fmt_out.audio.i_channels = + __MIN( id->p_encoder->fmt_out.audio.i_channels, + id->p_decoder->fmt_out.audio.i_channels ); + if( id->p_decoder->fmt_out.audio.i_channels == + id->p_encoder->fmt_out.audio.i_channels ) + id->p_encoder->fmt_out.audio.i_physical_channels = + id->p_encoder->fmt_out.audio.i_original_channels = + id->p_decoder->fmt_out.audio.i_physical_channels; else + id->p_encoder->fmt_out.audio.i_physical_channels = + id->p_encoder->fmt_out.audio.i_original_channels = + pi_channels_maps[id->p_encoder->fmt_out.audio.i_channels]; + + /* Initialization of encoder format structures */ + es_format_Init( &id->p_encoder->fmt_in, AUDIO_ES, AOUT_FMT_S16_NE ); + id->p_encoder->fmt_in.audio.i_format = AOUT_FMT_S16_NE; + id->p_encoder->fmt_in.audio.i_rate = id->p_encoder->fmt_out.audio.i_rate; + id->p_encoder->fmt_in.audio.i_physical_channels = + id->p_encoder->fmt_in.audio.i_original_channels = + id->p_encoder->fmt_out.audio.i_physical_channels; + id->p_encoder->fmt_in.audio.i_channels = + id->p_encoder->fmt_out.audio.i_channels; + id->p_encoder->fmt_in.audio.i_bitspersample = + audio_BitsPerSample( id->p_encoder->fmt_in.i_codec ); + + id->p_encoder->p_cfg = p_stream->p_sys->p_audio_cfg; + + id->p_encoder->p_module = + module_Need( id->p_encoder, "encoder", p_sys->psz_aenc, VLC_TRUE ); + if( !id->p_encoder->p_module ) { - /* find decoder */ - i_ff_codec = get_ff_codec( id->f_src.i_fourcc ); - if( i_ff_codec == 0 ) - { - msg_Err( p_stream, "cannot find decoder id" ); - return VLC_EGENERIC; - } + msg_Err( p_stream, "cannot find encoder" ); + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + id->p_decoder->p_module = 0; + return VLC_EGENERIC; + } + id->p_encoder->fmt_in.audio.i_format = id->p_encoder->fmt_in.i_codec; + id->p_encoder->fmt_in.audio.i_bitspersample = + audio_BitsPerSample( id->p_encoder->fmt_in.i_codec ); + + /* Check if we need a filter for chroma conversion or resizing */ + if( id->p_decoder->fmt_out.i_codec != + id->p_encoder->fmt_in.i_codec ) + { + id->pp_filter[0] = + vlc_object_create( p_stream, VLC_OBJECT_FILTER ); + vlc_object_attach( id->pp_filter[0], p_stream ); + + id->pp_filter[0]->pf_audio_buffer_new = __block_New; - id->ff_dec = avcodec_find_decoder( i_ff_codec ); - if( !id->ff_dec ) + id->pp_filter[0]->fmt_in = id->p_decoder->fmt_out; + id->pp_filter[0]->fmt_out = id->p_encoder->fmt_in; + id->pp_filter[0]->p_module = + module_Need( id->pp_filter[0], "audio filter2", 0, 0 ); + if( id->pp_filter[0]->p_module ) id->i_filter++; + else { - msg_Err( p_stream, "cannot find decoder (avcodec)" ); + msg_Dbg( p_stream, "no audio filter found (%4.4s->%4.4s)", + (char *)&id->pp_filter[0]->fmt_in, + (char *)&id->pp_filter[0]->fmt_out ); + vlc_object_detach( id->pp_filter[0] ); + vlc_object_destroy( id->pp_filter[0] ); + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + id->p_decoder->p_module = 0; + module_Unneed( id->p_encoder, id->p_encoder->p_module ); + id->p_encoder->p_module = 0; return VLC_EGENERIC; } - id->ff_dec_c = avcodec_alloc_context(); - id->ff_dec_c->sample_rate = id->f_src.i_sample_rate; - id->ff_dec_c->channels = id->f_src.i_channels; - id->ff_dec_c->block_align = id->f_src.i_block_align; - id->ff_dec_c->bit_rate = id->f_src.i_bitrate; + id->pp_filter[0]->fmt_out.audio.i_bitspersample = + audio_BitsPerSample( id->pp_filter[0]->fmt_out.i_codec ); - id->ff_dec_c->extradata_size = id->f_src.i_extra_data; - id->ff_dec_c->extradata = id->f_src.p_extra_data; - if( avcodec_open( id->ff_dec_c, id->ff_dec ) ) + /* Try a 2 stage conversion */ + if( id->pp_filter[0]->fmt_out.i_codec != + id->p_encoder->fmt_in.i_codec ) { - msg_Err( p_stream, "cannot open decoder" ); - return VLC_EGENERIC; + id->pp_filter[1] = + vlc_object_create( p_stream, VLC_OBJECT_FILTER ); + vlc_object_attach( id->pp_filter[1], p_stream ); + + id->pp_filter[1]->pf_audio_buffer_new = __block_New; + + id->pp_filter[1]->fmt_in = id->pp_filter[0]->fmt_out; + id->pp_filter[1]->fmt_out = id->p_encoder->fmt_in; + id->pp_filter[1]->p_module = + module_Need( id->pp_filter[1], "audio filter2", 0, 0 ); + if( !id->pp_filter[1]->p_module || + id->pp_filter[1]->fmt_out.i_codec != + id->p_encoder->fmt_in.i_codec ) + { + msg_Dbg( p_stream, "no audio filter found (%4.4s->%4.4s)", + (char *)&id->pp_filter[1]->fmt_in, + (char *)&id->pp_filter[1]->fmt_out ); + module_Unneed( id->pp_filter[0], id->pp_filter[0]->p_module ); + vlc_object_detach( id->pp_filter[0] ); + vlc_object_destroy( id->pp_filter[0] ); + if( id->pp_filter[1]->p_module ) + module_Unneed( id->pp_filter[0], id->pp_filter[0]->p_module ); + vlc_object_detach( id->pp_filter[1] ); + vlc_object_destroy( id->pp_filter[1] ); + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + id->p_decoder->p_module = 0; + module_Unneed( id->p_encoder, id->p_encoder->p_module ); + id->p_encoder->p_module = 0; + return VLC_EGENERIC; + } + else id->i_filter++; } } - /* find encoder */ - id->i_buffer_in = 2 * AVCODEC_MAX_AUDIO_FRAME_SIZE; - id->i_buffer_in_pos = 0; - id->p_buffer_in = malloc( id->i_buffer_in ); + /* FIXME: Hack for mp3 transcoding support */ + if( id->p_encoder->fmt_out.i_codec == VLC_FOURCC( 'm','p','3',' ' ) ) + id->p_encoder->fmt_out.i_codec = VLC_FOURCC( 'm','p','g','a' ); - id->i_buffer = 2 * AVCODEC_MAX_AUDIO_FRAME_SIZE; - id->i_buffer_pos = 0; - id->p_buffer = malloc( id->i_buffer ); + return VLC_SUCCESS; +} - id->i_buffer_out = 2 * AVCODEC_MAX_AUDIO_FRAME_SIZE; - id->i_buffer_out_pos = 0; - id->p_buffer_out = malloc( id->i_buffer_out ); +static void transcode_audio_close( sout_stream_t *p_stream, + sout_stream_id_t *id ) +{ + int i; - /* Sanity check for audio channels */ - id->f_dst.i_channels = __MIN( id->f_dst.i_channels, id->f_src.i_channels ); + /* Close decoder */ + if( id->p_decoder->p_module ) + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + + /* Close encoder */ + if( id->p_encoder->p_module ) + module_Unneed( id->p_encoder, id->p_encoder->p_module ); -#ifdef HAVE_VORBIS_VORBISENC_H - if( id->f_dst.i_fourcc == VLC_FOURCC('v','o','r','b') ) + /* Close filters */ + for( i = 0; i < id->i_filter; i++ ) { - id->p_vi = (vorbis_info *)malloc( sizeof(vorbis_info) ); - id->p_vd = (vorbis_dsp_state *)malloc( sizeof(vorbis_dsp_state) ); - id->p_vb = (vorbis_block *)malloc( sizeof(vorbis_block) ); - id->p_vc = (vorbis_comment *)malloc( sizeof(vorbis_comment) ); + vlc_object_detach( id->pp_filter[i] ); + if( id->pp_filter[i]->p_module ) + module_Unneed( id->pp_filter[i], id->pp_filter[i]->p_module ); + vlc_object_destroy( id->pp_filter[i] ); + } +} - vorbis_info_init( id->p_vi ); +static int transcode_audio_process( sout_stream_t *p_stream, + sout_stream_id_t *id, + block_t *in, block_t **out ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; + aout_buffer_t *p_audio_buf; + block_t *p_block, *p_audio_block; + int i; + *out = NULL; - if( vorbis_encode_setup_managed( id->p_vi, id->f_dst.i_channels, - id->f_dst.i_sample_rate, -1, id->f_dst.i_bitrate, -1 ) || - vorbis_encode_ctl( id->p_vi, OV_ECTL_RATEMANAGE_AVG, NULL ) || - vorbis_encode_setup_init( id->p_vi ) ){} + while( (p_audio_buf = id->p_decoder->pf_decode_audio( id->p_decoder, + &in )) ) + { + if( p_sys->b_audio_sync ) + { + mtime_t i_dts = date_Get( &id->interpolated_pts ) + 1; + p_sys->i_master_drift = p_audio_buf->start_date - i_dts; + date_Increment( &id->interpolated_pts, p_audio_buf->i_nb_samples ); + p_audio_buf->start_date -= p_sys->i_master_drift; + p_audio_buf->end_date -= p_sys->i_master_drift; + } - /* add a comment */ - vorbis_comment_init( id->p_vc); - vorbis_comment_add_tag( id->p_vc, "ENCODER", "VLC media player"); + p_audio_block = p_audio_buf->p_sys; + p_audio_block->i_buffer = p_audio_buf->i_nb_bytes; + p_audio_block->i_dts = p_audio_block->i_pts = + p_audio_buf->start_date; + p_audio_block->i_length = p_audio_buf->end_date - + p_audio_buf->start_date; + p_audio_block->i_samples = p_audio_buf->i_nb_samples; - /* set up the analysis state and auxiliary encoding storage */ - vorbis_analysis_init( id->p_vd, id->p_vi ); - vorbis_block_init( id->p_vd, id->p_vb ); + /* Run filter chain */ + for( i = 0; i < id->i_filter; i++ ) + { + p_audio_block = + id->pp_filter[i]->pf_audio_filter( id->pp_filter[i], + p_audio_block ); + } - id->b_headers_sent = VLC_FALSE; - id->i_last_block_size = 0; - id->i_samples_delay = 0; + p_audio_buf->p_buffer = p_audio_block->p_buffer; + p_audio_buf->i_nb_bytes = p_audio_block->i_buffer; + p_audio_buf->i_nb_samples = p_audio_block->i_samples; + p_audio_buf->start_date = p_audio_block->i_dts; + p_audio_buf->end_date = p_audio_block->i_dts + p_audio_block->i_length; - return VLC_SUCCESS; + p_block = id->p_encoder->pf_encode_audio( id->p_encoder, p_audio_buf ); + block_ChainAppend( out, p_block ); + block_Release( p_audio_block ); + free( p_audio_buf ); } -#endif - i_ff_codec = get_ff_codec( id->f_dst.i_fourcc ); - if( i_ff_codec == 0 ) + return VLC_SUCCESS; +} + +static void audio_release_buffer( aout_buffer_t *p_buffer ) +{ + if( p_buffer && p_buffer->p_sys ) block_Release( p_buffer->p_sys ); + if( p_buffer ) free( p_buffer ); +} + +static aout_buffer_t *audio_new_buffer( decoder_t *p_dec, int i_samples ) +{ + aout_buffer_t *p_buffer; + block_t *p_block; + int i_size; + + if( p_dec->fmt_out.audio.i_bitspersample ) + { + i_size = i_samples * p_dec->fmt_out.audio.i_bitspersample / 8 * + p_dec->fmt_out.audio.i_channels; + } + else if( p_dec->fmt_out.audio.i_bytes_per_frame && + p_dec->fmt_out.audio.i_frame_length ) + { + i_size = i_samples * p_dec->fmt_out.audio.i_bytes_per_frame / + p_dec->fmt_out.audio.i_frame_length; + } + else + { + i_size = i_samples * 4 * p_dec->fmt_out.audio.i_channels; + } + + p_buffer = malloc( sizeof(aout_buffer_t) ); + p_buffer->pf_release = audio_release_buffer; + p_buffer->p_sys = p_block = block_New( p_dec, i_size ); + + p_buffer->p_buffer = p_block->p_buffer; + p_buffer->i_size = p_buffer->i_nb_bytes = p_block->i_buffer; + p_buffer->i_nb_samples = i_samples; + p_block->i_samples = i_samples; + + return p_buffer; +} + +static void audio_del_buffer( decoder_t *p_dec, aout_buffer_t *p_buffer ) +{ + if( p_buffer && p_buffer->p_sys ) block_Release( p_buffer->p_sys ); + if( p_buffer ) free( p_buffer ); +} + +/* + * video + */ +static int transcode_video_new( sout_stream_t *p_stream, sout_stream_id_t *id ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; + int i; + + /* + * Open decoder + */ + + /* Initialization of decoder structures */ + id->p_decoder->pf_decode_video = 0; + id->p_decoder->pf_vout_buffer_new = video_new_buffer_decoder; + id->p_decoder->pf_vout_buffer_del = video_del_buffer_decoder; + id->p_decoder->pf_picture_link = video_link_picture_decoder; + id->p_decoder->pf_picture_unlink = video_unlink_picture_decoder; + id->p_decoder->p_owner = malloc( sizeof(decoder_owner_sys_t) ); + for( i = 0; i < PICTURE_RING_SIZE; i++ ) + id->p_decoder->p_owner->pp_pics[i] = 0; + //id->p_decoder->p_cfg = p_sys->p_video_cfg; + + id->p_decoder->p_module = + module_Need( id->p_decoder, "decoder", "$codec", 0 ); + + if( !id->p_decoder->p_module ) { - msg_Err( p_stream, "cannot find encoder id" ); + msg_Err( p_stream, "cannot find decoder" ); return VLC_EGENERIC; } - id->ff_enc = avcodec_find_encoder( i_ff_codec ); - if( !id->ff_enc ) + /* + * Open encoder. + * Because some info about the decoded input will only be available + * once the first frame is decoded, we actually only test the availability + * of the encoder here. + */ + + /* Initialization of encoder format structures */ + es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat, + id->p_decoder->fmt_out.i_codec ); + id->p_encoder->fmt_in.video.i_chroma = id->p_decoder->fmt_out.i_codec; + + /* The dimensions will be set properly later on. + * Just put sensible values so we can test an encoder is available. */ + id->p_encoder->fmt_in.video.i_width = + id->p_encoder->fmt_out.video.i_width ? + id->p_encoder->fmt_out.video.i_width : + id->p_decoder->fmt_in.video.i_width ? + id->p_decoder->fmt_in.video.i_width : 16; + id->p_encoder->fmt_in.video.i_height = + id->p_encoder->fmt_out.video.i_height ? + id->p_encoder->fmt_out.video.i_height : + id->p_decoder->fmt_in.video.i_height ? + id->p_decoder->fmt_in.video.i_height : 16; + id->p_encoder->fmt_in.video.i_frame_rate = 25; + id->p_encoder->fmt_in.video.i_frame_rate_base = 1; + + id->p_encoder->i_threads = p_sys->i_threads; + id->p_encoder->p_cfg = p_sys->p_video_cfg; + + id->p_encoder->p_module = + module_Need( id->p_encoder, "encoder", p_sys->psz_venc, VLC_TRUE ); + if( !id->p_encoder->p_module ) { - msg_Err( p_stream, "cannot find encoder (avcodec)" ); + msg_Err( p_stream, "cannot find encoder" ); + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + id->p_decoder->p_module = 0; return VLC_EGENERIC; } - /* Hack for mp3 transcoding support */ - if( id->f_dst.i_fourcc == VLC_FOURCC( 'm','p','3',' ' ) ) + /* Close the encoder. + * We'll open it only when we have the first frame. */ + module_Unneed( id->p_encoder, id->p_encoder->p_module ); + id->p_encoder->p_module = NULL; + + if( p_sys->i_threads >= 1 ) { - id->f_dst.i_fourcc = VLC_FOURCC( 'm','p','g','a' ); + p_sys->id_video = id; + vlc_mutex_init( p_stream, &p_sys->lock_out ); + vlc_cond_init( p_stream, &p_sys->cond ); + memset( p_sys->pp_pics, 0, sizeof(p_sys->pp_pics) ); + p_sys->i_first_pic = 0; + p_sys->i_last_pic = 0; + p_sys->p_buffers = NULL; + p_sys->b_die = p_sys->b_error = 0; + if( vlc_thread_create( p_sys, "encoder", EncoderThread, + VLC_THREAD_PRIORITY_VIDEO, VLC_FALSE ) ) + { + msg_Err( p_stream, "cannot spawn encoder thread" ); + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + id->p_decoder->p_module = 0; + return VLC_EGENERIC; + } } - id->ff_enc_c = avcodec_alloc_context(); - id->ff_enc_c->bit_rate = id->f_dst.i_bitrate; - id->ff_enc_c->sample_rate = id->f_dst.i_sample_rate; - id->ff_enc_c->channels = id->f_dst.i_channels; + date_Set( &id->interpolated_pts, 0 ); - /* Make sure we get extradata filled by the encoder */ - id->ff_enc_c->extradata_size = 0; - id->ff_enc_c->extradata = NULL; - id->ff_enc_c->flags |= CODEC_FLAG_GLOBAL_HEADER; + return VLC_SUCCESS; +} - if( avcodec_open( id->ff_enc_c, id->ff_enc ) ) +static int transcode_video_encoder_open( sout_stream_t *p_stream, + sout_stream_id_t *id ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; + + /* Hack because of the copy packetizer which can fail to detect the + * proper size (which forces us to wait until the 1st frame + * is decoded) */ + int i_width = id->p_decoder->fmt_out.video.i_width - + p_sys->i_crop_left - p_sys->i_crop_right; + int i_height = id->p_decoder->fmt_out.video.i_height - + p_sys->i_crop_top - p_sys->i_crop_bottom; + + if( id->p_encoder->fmt_out.video.i_width <= 0 && + id->p_encoder->fmt_out.video.i_height <= 0 && p_sys->f_scale ) { - if( id->ff_enc_c->channels > 2 ) + /* Apply the scaling */ + id->p_encoder->fmt_out.video.i_width = i_width * p_sys->f_scale; + id->p_encoder->fmt_out.video.i_height = i_height * p_sys->f_scale; + } + else if( id->p_encoder->fmt_out.video.i_width > 0 && + id->p_encoder->fmt_out.video.i_height <= 0 ) + { + id->p_encoder->fmt_out.video.i_height = + id->p_encoder->fmt_out.video.i_width / (double)i_width * i_height; + } + else if( id->p_encoder->fmt_out.video.i_width <= 0 && + id->p_encoder->fmt_out.video.i_height > 0 ) + { + id->p_encoder->fmt_out.video.i_width = + id->p_encoder->fmt_out.video.i_height / (double)i_height * i_width; + } + + /* Make sure the size is at least a multiple of 2 */ + id->p_encoder->fmt_out.video.i_width = + (id->p_encoder->fmt_out.video.i_width + 1) >> 1 << 1; + id->p_encoder->fmt_out.video.i_height = + (id->p_encoder->fmt_out.video.i_height + 1) >> 1 << 1; + + id->p_encoder->fmt_in.video.i_width = + id->p_encoder->fmt_out.video.i_width; + id->p_encoder->fmt_in.video.i_height = + id->p_encoder->fmt_out.video.i_height; + + if( !id->p_encoder->fmt_out.video.i_frame_rate || + !id->p_encoder->fmt_out.video.i_frame_rate_base ) + { + if( id->p_decoder->fmt_out.video.i_frame_rate && + id->p_decoder->fmt_out.video.i_frame_rate_base ) { - id->ff_enc_c->channels = 2; - id->f_dst.i_channels = 2; - if( avcodec_open( id->ff_enc_c, id->ff_enc ) ) - { - msg_Err( p_stream, "cannot open encoder" ); - return VLC_EGENERIC; - } - msg_Warn( p_stream, "stereo mode selected (codec limitation)" ); + id->p_encoder->fmt_out.video.i_frame_rate = + id->p_decoder->fmt_out.video.i_frame_rate; + id->p_encoder->fmt_out.video.i_frame_rate_base = + id->p_decoder->fmt_out.video.i_frame_rate_base; } else { - msg_Err( p_stream, "cannot open encoder" ); - return VLC_EGENERIC; + /* Pick a sensible default value */ + id->p_encoder->fmt_out.video.i_frame_rate = 25; + id->p_encoder->fmt_out.video.i_frame_rate_base = 1; } } - id->f_dst.i_extra_data = id->ff_enc_c->extradata_size; - id->f_dst.p_extra_data = id->ff_enc_c->extradata; - id->ff_enc_c->flags &= ~CODEC_FLAG_GLOBAL_HEADER; + id->p_encoder->fmt_in.video.i_frame_rate = + id->p_encoder->fmt_out.video.i_frame_rate; + id->p_encoder->fmt_in.video.i_frame_rate_base = + id->p_encoder->fmt_out.video.i_frame_rate_base; - return VLC_SUCCESS; -} + date_Init( &id->interpolated_pts, + id->p_encoder->fmt_out.video.i_frame_rate, + id->p_encoder->fmt_out.video.i_frame_rate_base ); -static void transcode_audio_ffmpeg_close( sout_stream_t *p_stream, - sout_stream_id_t *id ) -{ - if( id->ff_dec ) + /* Check whether a particular aspect ratio was requested */ + if( !id->p_encoder->fmt_out.video.i_aspect ) { - avcodec_close( id->ff_dec_c ); + id->p_encoder->fmt_out.video.i_aspect = + id->p_decoder->fmt_out.video.i_aspect; } + id->p_encoder->fmt_in.video.i_aspect = + id->p_encoder->fmt_out.video.i_aspect; -#ifdef HAVE_VORBIS_VORBISENC_H - if( id->f_dst.i_fourcc == VLC_FOURCC('v','o','r','b') ) + id->p_encoder->p_module = + module_Need( id->p_encoder, "encoder", p_sys->psz_venc, VLC_TRUE ); + if( !id->p_encoder->p_module ) { - vorbis_block_clear( id->p_vb ); - vorbis_dsp_clear( id->p_vd ); - vorbis_comment_clear( id->p_vc ); - vorbis_info_clear( id->p_vi ); /* must be called last */ + msg_Err( p_stream, "cannot find encoder" ); + return VLC_EGENERIC; + } + + id->p_encoder->fmt_in.video.i_chroma = id->p_encoder->fmt_in.i_codec; - free( id->p_vi ); - free( id->p_vd ); - free( id->p_vb ); - free( id->p_vc ); + /* Hack for mp2v/mp1v transcoding support */ + if( id->p_encoder->fmt_out.i_codec == VLC_FOURCC('m','p','1','v') || + id->p_encoder->fmt_out.i_codec == VLC_FOURCC('m','p','2','v') ) + { + id->p_encoder->fmt_out.i_codec = VLC_FOURCC('m','p','g','v'); } - else -#endif - avcodec_close( id->ff_enc_c ); - free( id->ff_dec_c ); - if( id->f_dst.i_fourcc != VLC_FOURCC('v','o','r','b') ) - free( id->ff_enc_c ); + id->id = p_stream->p_sys->p_out->pf_add( p_stream->p_sys->p_out, + &id->p_encoder->fmt_out ); + if( !id->id ) + { + msg_Err( p_stream, "cannot add this stream" ); + return VLC_EGENERIC; + } - free( id->p_buffer_in ); - free( id->p_buffer ); - free( id->p_buffer_out ); + return VLC_SUCCESS; } -static int transcode_audio_ffmpeg_process( sout_stream_t *p_stream, - sout_stream_id_t *id, - sout_buffer_t *in, - sout_buffer_t **out ) +static void transcode_video_close( sout_stream_t *p_stream, + sout_stream_id_t *id ) { - vlc_bool_t b_again = VLC_FALSE; + int i, j; - *out = NULL; + if( p_stream->p_sys->i_threads >= 1 ) + { + vlc_mutex_lock( &p_stream->p_sys->lock_out ); + p_stream->p_sys->b_die = 1; + vlc_cond_signal( &p_stream->p_sys->cond ); + vlc_mutex_unlock( &p_stream->p_sys->lock_out ); + vlc_thread_join( p_stream->p_sys ); + vlc_mutex_destroy( &p_stream->p_sys->lock_out ); + vlc_cond_destroy( &p_stream->p_sys->cond ); + } - /* gather data into p_buffer_in */ -#ifdef HAVE_VORBIS_VORBISENC_H - if( id->f_dst.i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ) - id->i_dts = in->i_dts - - (mtime_t)1000000 * (mtime_t)id->i_samples_delay / - (mtime_t)id->f_dst.i_sample_rate; - else -#endif - id->i_dts = in->i_dts - - (mtime_t)1000000 * - (mtime_t)(id->i_buffer_pos / 2 / id->ff_dec_c->channels )/ - (mtime_t)id->ff_dec_c->sample_rate; + /* Close decoder */ + if( id->p_decoder->p_module ) + module_Unneed( id->p_decoder, id->p_decoder->p_module ); - if( id->i_buffer_in_pos + (int)in->i_size > id->i_buffer_in ) + if( id->p_decoder->p_owner ) { - /* extend buffer_in */ - id->i_buffer_in = id->i_buffer_in_pos + in->i_size + 1024; - id->p_buffer_in = realloc( id->p_buffer_in, id->i_buffer_in ); + /* Clean-up pictures ring buffer */ + for( i = 0; i < PICTURE_RING_SIZE; i++ ) + { + if( id->p_decoder->p_owner->pp_pics[i] ) + video_del_buffer( VLC_OBJECT(id->p_decoder), + id->p_decoder->p_owner->pp_pics[i] ); + } + free( id->p_decoder->p_owner ); } - memcpy( &id->p_buffer_in[id->i_buffer_in_pos], - in->p_buffer, in->i_size ); - id->i_buffer_in_pos += in->i_size; - do + /* Close encoder */ + if( id->p_encoder->p_module ) + module_Unneed( id->p_encoder, id->p_encoder->p_module ); + + /* Close filters */ + for( i = 0; i < id->i_filter; i++ ) { - int i_buffer_pos; + vlc_object_detach( id->pp_filter[i] ); + if( id->pp_filter[i]->p_module ) + module_Unneed( id->pp_filter[i], id->pp_filter[i]->p_module ); - /* decode as much data as possible */ - if( id->ff_dec ) + /* Clean-up pictures ring buffer */ + for( j = 0; j < PICTURE_RING_SIZE; j++ ) { - for( ;; ) - { - int i_buffer_size; - int i_used; - - i_buffer_size = id->i_buffer - id->i_buffer_pos; + if( id->pp_filter[i]->p_owner->pp_pics[j] ) + video_del_buffer( VLC_OBJECT(id->pp_filter[i]), + id->pp_filter[i]->p_owner->pp_pics[j] ); + } + free( id->pp_filter[i]->p_owner ); - i_used = avcodec_decode_audio( id->ff_dec_c, - (int16_t*)&id->p_buffer[id->i_buffer_pos], - &i_buffer_size, id->p_buffer_in, - id->i_buffer_in_pos ); + vlc_object_destroy( id->pp_filter[i] ); + } +} - /* msg_Warn( p_stream, "avcodec_decode_audio: %d used", - i_used ); */ - id->i_buffer_pos += i_buffer_size; +static int transcode_video_process( sout_stream_t *p_stream, + sout_stream_id_t *id, + block_t *in, block_t **out ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; + int i_duplicate = 1, i; + picture_t *p_pic; + *out = NULL; - if( i_used < 0 ) - { - msg_Warn( p_stream, "error"); - id->i_buffer_in_pos = 0; - break; - } - else if( i_used < id->i_buffer_in_pos ) - { - memmove( id->p_buffer_in, - &id->p_buffer_in[i_used], - id->i_buffer_in - i_used ); - id->i_buffer_in_pos -= i_used; - } - else - { - id->i_buffer_in_pos = 0; - break; - } + while( (p_pic = id->p_decoder->pf_decode_video( id->p_decoder, &in )) ) + { + subpicture_t *p_subpic = 0; + mtime_t i_pic_date = p_pic->date; - if( id->i_buffer_pos >= AVCODEC_MAX_AUDIO_FRAME_SIZE ) - { - /* buffer full */ - b_again = VLC_TRUE; - break; - } - } - } - else + if( p_sys->b_audio_sync ) { - int16_t *sout = (int16_t*)&id->p_buffer[id->i_buffer_pos]; - int i_used = 0; + mtime_t i_video_drift; + mtime_t i_master_drift = p_sys->i_master_drift; + mtime_t i_pts; - if( id->f_src.i_fourcc == VLC_FOURCC( 's', '8', ' ', ' ' ) ) + if( !i_master_drift ) { - int8_t *sin = (int8_t*)id->p_buffer_in; - int i_samples = __MIN( ( id->i_buffer - id->i_buffer_pos ) - / 2, id->i_buffer_in_pos ); - i_used = i_samples; - while( i_samples > 0 ) - { - *sout++ = ( *sin++ ) << 8; - i_samples--; - } + /* No audio track ? */ + p_sys->i_master_drift = i_master_drift = p_pic->date; } - else if( id->f_src.i_fourcc == VLC_FOURCC( 'u', '8', ' ', ' ' ) ) - { - int8_t *sin = (int8_t*)id->p_buffer_in; - int i_samples = __MIN( ( id->i_buffer - id->i_buffer_pos ) - / 2, id->i_buffer_in_pos ); - i_used = i_samples; - while( i_samples > 0 ) - { - *sout++ = ( *sin++ - 128 ) << 8; - i_samples--; - } - } - else if( id->f_src.i_fourcc == VLC_FOURCC( 's', '1', '6', 'l' ) ) - { - int i_samples = __MIN( ( id->i_buffer - id->i_buffer_pos ) - / 2, id->i_buffer_in_pos / 2); -#ifdef WORDS_BIGENDIAN - uint8_t *sin = (uint8_t*)id->p_buffer_in; - i_used = i_samples * 2; - while( i_samples > 0 ) - { - uint8_t tmp[2]; - tmp[1] = *sin++; - tmp[0] = *sin++; - *sout++ = *(int16_t*)tmp; - i_samples--; - } + i_pts = date_Get( &id->interpolated_pts ) + 1; + i_video_drift = p_pic->date - i_pts; + i_duplicate = 1; -#else - memcpy( sout, id->p_buffer_in, i_samples * 2 ); - sout += i_samples; - i_used = i_samples * 2; -#endif - } - else if( id->f_src.i_fourcc == VLC_FOURCC( 's', '1', '6', 'b' ) ) - { - int i_samples = __MIN( ( id->i_buffer - id->i_buffer_pos ) - / 2, id->i_buffer_in_pos / 2); -#ifdef WORDS_BIGENDIAN - memcpy( sout, id->p_buffer_in, i_samples * 2 ); - sout += i_samples; - i_used = i_samples * 2; -#else - uint8_t *sin = (uint8_t*)id->p_buffer_in; - i_used = i_samples * 2; - while( i_samples > 0 ) - { - uint8_t tmp[2]; + /* Set the pts of the frame being encoded */ + p_pic->date = i_pts; - tmp[1] = *sin++; - tmp[0] = *sin++; - *sout++ = *(int16_t*)tmp; - i_samples--; - } + if( i_video_drift < i_master_drift - 50000 ) + { +#if 0 + msg_Dbg( p_stream, "dropping frame (%i)", + (int)(i_video_drift - i_master_drift) ); #endif + return VLC_EGENERIC; } - - id->i_buffer_pos = (uint8_t*)sout - id->p_buffer; - if( i_used < id->i_buffer_in_pos ) + else if( i_video_drift > i_master_drift + 50000 ) { - memmove( id->p_buffer_in, - &id->p_buffer_in[i_used], - id->i_buffer_in - i_used ); +#if 0 + msg_Dbg( p_stream, "adding frame (%i)", + (int)(i_video_drift - i_master_drift) ); +#endif + i_duplicate = 2; } - id->i_buffer_in_pos -= i_used; } - i_buffer_pos = id->i_buffer_pos; - - /* encode as much data as possible */ - -#ifdef HAVE_VORBIS_VORBISENC_H - if( id->i_buffer_pos == 0 ); - else if( id->f_dst.i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ) + if( !id->p_encoder->p_module ) { - float **buffer; - int i, j, i_samples; - sout_buffer_t *p_out; - ogg_packet op; + if( transcode_video_encoder_open( p_stream, id ) != VLC_SUCCESS ) + { + transcode_video_close( p_stream, id ); + id->b_transcode = VLC_FALSE; + return VLC_EGENERIC; + } - if( !id->b_headers_sent ) + /* Deinterlace */ + if( p_stream->p_sys->b_deinterlace ) { - ogg_packet header[3]; - vorbis_analysis_headerout( id->p_vd, id->p_vc, - &header[0], &header[1], &header[2]); - for( i = 0; i < 3; i++ ) + id->pp_filter[id->i_filter] = + vlc_object_create( p_stream, VLC_OBJECT_FILTER ); + vlc_object_attach( id->pp_filter[id->i_filter], p_stream ); + + id->pp_filter[id->i_filter]->pf_vout_buffer_new = + video_new_buffer_filter; + id->pp_filter[id->i_filter]->pf_vout_buffer_del = + video_del_buffer_filter; + + id->pp_filter[id->i_filter]->fmt_in = id->p_decoder->fmt_out; + id->pp_filter[id->i_filter]->fmt_out = id->p_decoder->fmt_out; + id->pp_filter[id->i_filter]->p_module = + module_Need( id->pp_filter[id->i_filter], + "video filter2", "deinterlace", 0 ); + if( id->pp_filter[id->i_filter]->p_module ) { - p_out = sout_BufferNew( p_stream->p_sout, header[i].bytes); - memcpy( p_out->p_buffer, header[i].packet, - header[i].bytes ); + id->pp_filter[id->i_filter]->p_owner = + malloc( sizeof(filter_owner_sys_t) ); + for( i = 0; i < PICTURE_RING_SIZE; i++ ) + id->pp_filter[id->i_filter]->p_owner->pp_pics[i] = 0; - p_out->i_size = header[i].bytes; - p_out->i_length = 0; - - p_out->i_dts = p_out->i_pts = 0; - - sout_BufferChain( out, p_out ); + id->i_filter++; } - id->b_headers_sent = VLC_TRUE; - } - - i_samples = id->i_buffer_pos / id->f_src.i_channels / 2; - id->i_samples_delay += i_samples; - id->i_buffer_pos = 0; - - buffer = vorbis_analysis_buffer( id->p_vd, i_samples ); - - /* convert samples to float and uninterleave */ - for( i = 0; i < id->f_dst.i_channels; i++ ) - { - for( j = 0 ; j < i_samples ; j++ ) + else { - buffer[i][j]= ((float)( ((int16_t *)id->p_buffer) - [j*id->f_src.i_channels + i ] ))/ 32768.f; + msg_Dbg( p_stream, "no video filter found" ); + vlc_object_detach( id->pp_filter[id->i_filter] ); + vlc_object_destroy( id->pp_filter[id->i_filter] ); } } - vorbis_analysis_wrote( id->p_vd, i_samples ); - - while( vorbis_analysis_blockout( id->p_vd, id->p_vb ) == 1 ) + /* Check if we need a filter for chroma conversion or resizing */ + if( id->p_decoder->fmt_out.video.i_chroma != + id->p_encoder->fmt_in.video.i_chroma || + id->p_decoder->fmt_out.video.i_width != + id->p_encoder->fmt_out.video.i_width || + id->p_decoder->fmt_out.video.i_height != + id->p_encoder->fmt_out.video.i_height || + p_sys->i_crop_top > 0 || p_sys->i_crop_bottom > 0 || + p_sys->i_crop_left > 0 || p_sys->i_crop_right > 0 ) { - vorbis_analysis( id->p_vb, NULL ); - vorbis_bitrate_addblock( id->p_vb ); - - while( vorbis_bitrate_flushpacket( id->p_vd, &op ) ) + id->pp_filter[id->i_filter] = + vlc_object_create( p_stream, VLC_OBJECT_FILTER ); + vlc_object_attach( id->pp_filter[id->i_filter], p_stream ); + + id->pp_filter[id->i_filter]->pf_vout_buffer_new = + video_new_buffer_filter; + id->pp_filter[id->i_filter]->pf_vout_buffer_del = + video_del_buffer_filter; + + id->pp_filter[id->i_filter]->fmt_in = id->p_decoder->fmt_out; + id->pp_filter[id->i_filter]->fmt_out = id->p_encoder->fmt_in; + id->pp_filter[id->i_filter]->p_module = + module_Need( id->pp_filter[id->i_filter], + "video filter2", 0, 0 ); + if( id->pp_filter[id->i_filter]->p_module ) { - int i_block_size; - p_out = sout_BufferNew( p_stream->p_sout, op.bytes ); - memcpy( p_out->p_buffer, op.packet, op.bytes ); - - i_block_size = vorbis_packet_blocksize( id->p_vi, &op ); + id->pp_filter[id->i_filter]->p_owner = + malloc( sizeof(filter_owner_sys_t) ); + for( i = 0; i < PICTURE_RING_SIZE; i++ ) + id->pp_filter[id->i_filter]->p_owner->pp_pics[i] = 0; - if( i_block_size < 0 ) i_block_size = 0; - i_samples = ( id->i_last_block_size + - i_block_size ) >> 2; - id->i_last_block_size = i_block_size; - - p_out->i_size = op.bytes; - p_out->i_length = (mtime_t)1000000 * - (mtime_t)i_samples / - (mtime_t)id->f_dst.i_sample_rate; - - //msg_Err( p_stream, "i_dts: %lld", id->i_dts ); + id->i_filter++; + } + else + { + msg_Dbg( p_stream, "no video filter found" ); + vlc_object_detach( id->pp_filter[id->i_filter] ); + vlc_object_destroy( id->pp_filter[id->i_filter] ); - /* FIXME */ - p_out->i_dts = id->i_dts; - p_out->i_pts = id->i_dts; + transcode_video_close( p_stream, id ); + id->b_transcode = VLC_FALSE; + return VLC_EGENERIC; + } + } + } - id->i_samples_delay -= i_samples; + /* Run filter chain */ + for( i = 0; i < id->i_filter; i++ ) + { + p_pic = id->pp_filter[i]->pf_video_filter(id->pp_filter[i], p_pic); + } - /* update dts */ - id->i_dts += p_out->i_length; - sout_BufferChain( out, p_out ); + /* + * Encoding + */ - } - } + /* Check if we have a subpicture to overlay */ + if( p_sys->p_spu ) + { + p_subpic = spu_SortSubpictures( p_sys->p_spu, i_pic_date ); + /* TODO: get another pic */ } - else -#endif - for( ;; ) + /* Overlay subpicture */ + if( p_subpic ) { - int i_frame_size = id->ff_enc_c->frame_size * 2 * - id->ff_dec_c->channels; - int i_out_size, i, j; - sout_buffer_t *p_out; - int16_t *p_buffer = (int16_t *)(id->p_buffer + i_buffer_pos - - id->i_buffer_pos); - - if( id->i_buffer_pos < i_frame_size ) - { - break; - } + int i_scale_width, i_scale_height; + video_format_t *p_fmt; + + i_scale_width = id->p_encoder->fmt_in.video.i_width * 1000 / + id->p_decoder->fmt_out.video.i_width; + i_scale_height = id->p_encoder->fmt_in.video.i_height * 1000 / + id->p_decoder->fmt_out.video.i_height; - if( id->ff_dec_c->channels != id->ff_enc_c->channels ) + if( p_pic->i_refcount && !id->i_filter ) { - /* dumb downmixing */ - for( i = 0; i < id->ff_enc_c->frame_size; i++ ) + /* We can't modify the picture, we need to duplicate it */ + picture_t *p_tmp = video_new_buffer_decoder( id->p_decoder ); + if( p_tmp ) { - for( j = 0 ; j < id->f_dst.i_channels; j++ ) - { - p_buffer[i*id->f_dst.i_channels+j] = p_buffer[i*id->f_src.i_channels+j]; - } + vout_CopyPicture( p_stream, p_tmp, p_pic ); + p_pic->pf_release( p_pic ); + p_pic = p_tmp; } } - /* msg_Warn( p_stream, "avcodec_encode_audio: frame size%d", - i_frame_size); */ - i_out_size = avcodec_encode_audio( id->ff_enc_c, - id->p_buffer_out, id->i_buffer_out, p_buffer ); - - if( i_out_size <= 0 ) - { - break; - } - - id->i_buffer_pos -= i_frame_size; - - p_out = sout_BufferNew( p_stream->p_sout, i_out_size ); - memcpy( p_out->p_buffer, id->p_buffer_out, i_out_size ); - p_out->i_size = i_out_size; - p_out->i_length = (mtime_t)1000000 * - (mtime_t)id->ff_enc_c->frame_size / - (mtime_t)id->ff_enc_c->sample_rate; - - /* FIXME */ - p_out->i_dts = id->i_dts; - p_out->i_pts = id->i_dts; - - /* update dts */ - id->i_dts += p_out->i_length; - - /* msg_Warn( p_stream, "frame dts=%lld len %lld out=%d", - p_out->i_dts, p_out->i_length, i_out_size ); */ + if( id->i_filter ) + p_fmt = &id->pp_filter[id->i_filter -1]->fmt_out.video; + else + p_fmt = &id->p_decoder->fmt_out.video; - sout_BufferChain( out, p_out ); + spu_RenderSubpictures( p_sys->p_spu, p_fmt, p_pic, p_pic, p_subpic, + i_scale_width, i_scale_height ); } - /* Copy the remaining raw samples */ - if( id->i_buffer_pos != 0 ) + if( p_sys->i_threads >= 1 ) { - memmove( id->p_buffer, - &id->p_buffer[i_buffer_pos - id->i_buffer_pos], - id->i_buffer_pos ); + vlc_mutex_lock( &p_sys->lock_out ); + p_sys->pp_pics[p_sys->i_last_pic++] = p_pic; + p_sys->i_last_pic %= PICTURE_RING_SIZE; + *out = p_sys->p_buffers; + p_sys->p_buffers = NULL; + vlc_cond_signal( &p_sys->cond ); + vlc_mutex_unlock( &p_sys->lock_out ); } + else + { + block_t *p_block; + p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic ); + block_ChainAppend( out, p_block ); + + if( p_sys->b_audio_sync ) + date_Increment( &id->interpolated_pts, 1 ); - } while( b_again ); + if( p_sys->b_audio_sync && i_duplicate > 1 ) + { + mtime_t i_pts = date_Get( &id->interpolated_pts ) + 1; + date_Increment( &id->interpolated_pts, 1 ); + p_pic->date = i_pts; + p_block = id->p_encoder->pf_encode_video(id->p_encoder, p_pic); + block_ChainAppend( out, p_block ); + } + + p_pic->pf_release( p_pic ); + } + } return VLC_SUCCESS; } - -/* - * video - */ -static int transcode_video_ffmpeg_new( sout_stream_t *p_stream, - sout_stream_id_t *id ) +static int EncoderThread( sout_stream_sys_t *p_sys ) { - sout_stream_sys_t *p_sys = p_stream->p_sys; - - int i_ff_codec; + sout_stream_id_t *id = p_sys->id_video; + picture_t *p_pic; + int i_plane; - if( id->f_src.i_fourcc == VLC_FOURCC( 'I', '4', '2', '0' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'I', '4', '2', '2' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'I', '4', '4', '4' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'Y', 'U', 'Y', '2' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'R', 'V', '1', '5' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'R', 'V', '1', '6' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'R', 'V', '2', '4' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'R', 'V', '3', '2' ) || - id->f_src.i_fourcc == VLC_FOURCC( 'G', 'R', 'E', 'Y' ) ) + while( !p_sys->b_die && !p_sys->b_error ) { - id->ff_dec = NULL; - id->ff_dec_c = avcodec_alloc_context(); - id->ff_dec_c->width = id->f_src.i_width; - id->ff_dec_c->height = id->f_src.i_height; - id->ff_dec_c->pix_fmt = get_ff_chroma( id->f_src.i_fourcc ); - } - else - { - /* find decoder */ - i_ff_codec = get_ff_codec( id->f_src.i_fourcc ); - if( i_ff_codec == 0 ) - { - msg_Err( p_stream, "cannot find decoder" ); - return VLC_EGENERIC; - } + block_t *p_block; - id->ff_dec = avcodec_find_decoder( i_ff_codec ); - if( !id->ff_dec ) + vlc_mutex_lock( &p_sys->lock_out ); + while( p_sys->i_last_pic == p_sys->i_first_pic ) { - msg_Err( p_stream, "cannot find decoder" ); - return VLC_EGENERIC; + vlc_cond_wait( &p_sys->cond, &p_sys->lock_out ); + if( p_sys->b_die || p_sys->b_error ) break; } - - id->ff_dec_c = avcodec_alloc_context(); - id->ff_dec_c->width = id->f_src.i_width; - id->ff_dec_c->height = id->f_src.i_height; - /* id->ff_dec_c->bit_rate = id->f_src.i_bitrate; */ - id->ff_dec_c->extradata_size= id->f_src.i_extra_data; - id->ff_dec_c->extradata = id->f_src.p_extra_data; - id->ff_dec_c->workaround_bugs = FF_BUG_AUTODETECT; - id->ff_dec_c->error_resilience= -1; - id->ff_dec_c->get_buffer = transcode_video_ffmpeg_getframebuf; - id->ff_dec_c->opaque = p_sys; - - if( avcodec_open( id->ff_dec_c, id->ff_dec ) < 0 ) + if( p_sys->b_die || p_sys->b_error ) { - msg_Err( p_stream, "cannot open decoder" ); - return VLC_EGENERIC; + vlc_mutex_unlock( &p_sys->lock_out ); + break; } - if( i_ff_codec == CODEC_ID_MPEG4 && id->ff_dec_c->extradata_size > 0 ) - { - int b_gotpicture; - AVFrame frame; - uint8_t *p_vol = malloc( id->ff_dec_c->extradata_size + - FF_INPUT_BUFFER_PADDING_SIZE ); - - memcpy( p_vol, id->ff_dec_c->extradata, - id->ff_dec_c->extradata_size ); - memset( p_vol + id->ff_dec_c->extradata_size, 0, - FF_INPUT_BUFFER_PADDING_SIZE ); - - avcodec_decode_video( id->ff_dec_c, &frame, &b_gotpicture, - id->ff_dec_c->extradata, - id->ff_dec_c->extradata_size ); - free( p_vol ); - } - } + p_pic = p_sys->pp_pics[p_sys->i_first_pic++]; + p_sys->i_first_pic %= PICTURE_RING_SIZE; + vlc_mutex_unlock( &p_sys->lock_out ); + p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic ); + vlc_mutex_lock( &p_sys->lock_out ); + block_ChainAppend( &p_sys->p_buffers, p_block ); + vlc_mutex_unlock( &p_sys->lock_out ); - /* find encoder */ - id->ff_enc = NULL; - i_ff_codec = get_ff_codec( id->f_dst.i_fourcc ); - if( i_ff_codec != 0 ) - { - id->ff_enc = avcodec_find_encoder( i_ff_codec ); + for( i_plane = 0; i_plane < p_pic->i_planes; i_plane++ ) + { + free( p_pic->p[i_plane].p_pixels ); + } + free( p_pic ); } - /* Hack for external encoders */ - if( !id->ff_enc ) + while( p_sys->i_last_pic != p_sys->i_first_pic ) { - id->p_encoder = vlc_object_create( p_stream, VLC_OBJECT_ENCODER ); - id->p_encoder->i_fourcc = id->f_dst.i_fourcc; - id->p_encoder->format.video.i_width = p_sys->i_width; - id->p_encoder->format.video.i_height = p_sys->i_height; - id->p_encoder->i_bitrate = p_sys->i_vbitrate; + p_pic = p_sys->pp_pics[p_sys->i_first_pic++]; + p_sys->i_first_pic %= PICTURE_RING_SIZE; - if( id->p_encoder->format.video.i_width <= 0 ) - { - id->p_encoder->format.video.i_width = id->f_dst.i_width = - id->ff_dec_c->width - p_sys->i_crop_left - - p_sys->i_crop_right; - } - if( id->p_encoder->format.video.i_height <= 0 ) - { - id->p_encoder->format.video.i_height = id->f_dst.i_height = - id->ff_dec_c->height - p_sys->i_crop_top - - p_sys->i_crop_bottom; - } - - id->p_encoder->p_module = - module_Need( id->p_encoder, "video encoder", NULL ); - - if( !id->p_encoder->p_module ) + for( i_plane = 0; i_plane < p_pic->i_planes; i_plane++ ) { - free( id->p_encoder ); - id->p_encoder = NULL; + free( p_pic->p[i_plane].p_pixels ); } + free( p_pic ); } - /* End hack for external encoders */ - if( !id->ff_enc && !id->p_encoder ) - { - msg_Err( p_stream, "cannot find encoder" ); - return VLC_EGENERIC; - } + block_ChainRelease( p_sys->p_buffers ); - /* XXX open it only when we have the first frame */ - id->b_enc_inited = VLC_FALSE; - id->i_buffer_in = 0; - id->i_buffer_in_pos = 0; - id->p_buffer_in = NULL; - - id->i_buffer = 3*1024*1024; - id->i_buffer_pos = 0; - id->p_buffer = malloc( id->i_buffer ); - - id->i_buffer_out = 0; - id->i_buffer_out_pos = 0; - id->p_buffer_out = NULL; - - id->p_ff_pic = avcodec_alloc_frame(); - id->p_ff_pic_tmp0 = NULL; - id->p_ff_pic_tmp1 = NULL; - id->p_ff_pic_tmp2 = NULL; - id->p_vresample = NULL; - - p_sys->i_last_ref_pts = 0; - p_sys->i_buggy_pts_detect = 0; + return 0; +} - /* This is enough for external encoders */ - if( id->p_encoder->p_module ) return VLC_SUCCESS; +struct picture_sys_t +{ + vlc_object_t *p_owner; +}; - if( id->f_dst.i_fourcc == VLC_FOURCC( 'm','p','1','v' )|| - id->f_dst.i_fourcc == VLC_FOURCC( 'm','p','2','v' ) ) +static void video_release_buffer( picture_t *p_pic ) +{ + if( p_pic && !p_pic->i_refcount && p_pic->pf_release && p_pic->p_sys ) { - id->f_dst.i_fourcc = VLC_FOURCC( 'm','p','g','v' ); + video_del_buffer_decoder( (decoder_t *)p_pic->p_sys->p_owner, p_pic ); } + else if( p_pic && p_pic->i_refcount > 0 ) p_pic->i_refcount--; +} - id->ff_enc_c = avcodec_alloc_context(); - id->ff_enc_c->width = id->f_dst.i_width; - id->ff_enc_c->height = id->f_dst.i_height; - id->ff_enc_c->bit_rate = id->f_dst.i_bitrate; +static picture_t *video_new_buffer( vlc_object_t *p_this, picture_t **pp_ring ) +{ + decoder_t *p_dec = (decoder_t *)p_this; + picture_t *p_pic; + int i; - if( id->ff_dec ) + /* Find an empty space in the picture ring buffer */ + for( i = 0; i < PICTURE_RING_SIZE; i++ ) { - id->ff_enc_c->frame_rate = id->ff_dec_c->frame_rate; -#if LIBAVCODEC_BUILD >= 4662 - id->ff_enc_c->frame_rate_base= id->ff_dec_c->frame_rate_base; -#endif + if( pp_ring[i] != 0 && pp_ring[i]->i_status == DESTROYED_PICTURE ) + { + pp_ring[i]->i_status = RESERVED_PICTURE; + return pp_ring[i]; + } } - else + for( i = 0; i < PICTURE_RING_SIZE; i++ ) { -#if LIBAVCODEC_BUILD >= 4662 - id->ff_enc_c->frame_rate = 25 ; /* FIXME as it break mpeg */ - id->ff_enc_c->frame_rate_base= 1; -#else - id->ff_enc_c->frame_rate = 25 * FRAME_RATE_BASE; -#endif + if( pp_ring[i] == 0 ) break; } - id->ff_enc_c->gop_size = p_sys->i_key_int >= 0 ? p_sys->i_key_int : 50; - id->ff_enc_c->max_b_frames = __MIN( p_sys->i_b_frames, FF_MAX_B_FRAMES ); - id->ff_enc_c->b_frame_strategy = 0; - id->ff_enc_c->b_quant_factor = 2.0; - - if( p_sys->i_vtolerance >= 0 ) + if( i == PICTURE_RING_SIZE ) { - id->ff_enc_c->bit_rate_tolerance = p_sys->i_vtolerance; - } - id->ff_enc_c->qmin = p_sys->i_qmin; - id->ff_enc_c->qmax = p_sys->i_qmax; + msg_Err( p_this, "decoder/filter is leaking pictures, " + "resetting its ring buffer" ); -#if LIBAVCODEC_BUILD >= 4673 - id->ff_enc_c->mb_decision = p_sys->i_hq; -#else - if( p_sys->i_hq ) - { - id->ff_enc_c->flags |= CODEC_FLAG_HQ; - } -#endif + for( i = 0; i < PICTURE_RING_SIZE; i++ ) + { + pp_ring[i]->pf_release( pp_ring[i] ); + } - if( i_ff_codec == CODEC_ID_RAWVIDEO ) - { - id->ff_enc_c->pix_fmt = get_ff_chroma( id->f_dst.i_fourcc ); + i = 0; } - return VLC_SUCCESS; -} + p_pic = malloc( sizeof(picture_t) ); + p_dec->fmt_out.video.i_chroma = p_dec->fmt_out.i_codec; + vout_AllocatePicture( VLC_OBJECT(p_dec), p_pic, + p_dec->fmt_out.video.i_chroma, + p_dec->fmt_out.video.i_width, + p_dec->fmt_out.video.i_height, + p_dec->fmt_out.video.i_aspect ); -static void transcode_video_ffmpeg_close ( sout_stream_t *p_stream, sout_stream_id_t *id ) -{ - if( id->ff_dec ) - { - avcodec_close( id->ff_dec_c ); - } - if( id->p_encoder ) - { - /* External encoding */ - module_Unneed( p_stream, id->p_encoder->p_module ); - vlc_object_destroy( id->p_encoder->p_module ); - } - else if( id->b_enc_inited ) + if( !p_pic->i_planes ) { - avcodec_close( id->ff_enc_c ); + free( p_pic ); + return 0; } - if( id->p_ff_pic) - { - free( id->p_ff_pic ); - } + p_pic->pf_release = video_release_buffer; + p_pic->p_sys = malloc( sizeof(picture_sys_t) ); + p_pic->p_sys->p_owner = p_this; + p_pic->i_status = RESERVED_PICTURE; - if( id->p_ff_pic_tmp0 ) - { - free( id->p_ff_pic_tmp0->data[0] ); - free( id->p_ff_pic_tmp0 ); - } - if( id->p_ff_pic_tmp1) - { - free( id->p_ff_pic_tmp1->data[0] ); - free( id->p_ff_pic_tmp1 ); - } - if( id->p_ff_pic_tmp2) - { - free( id->p_ff_pic_tmp2->data[0] ); - free( id->p_ff_pic_tmp2 ); - } - if( id->p_vresample ) - { - free( id->p_vresample ); - } + pp_ring[i] = p_pic; - free( id->ff_dec_c ); - free( id->ff_enc_c ); - free( id->p_buffer ); + return p_pic; } -static int transcode_video_ffmpeg_process( sout_stream_t *p_stream, - sout_stream_id_t *id, sout_buffer_t *in, sout_buffer_t **out ) +static picture_t *video_new_buffer_decoder( decoder_t *p_dec ) { - sout_stream_sys_t *p_sys = p_stream->p_sys; - int i_used; - int i_out; - int b_gotpicture; - AVFrame *frame; - - int i_data; - uint8_t *p_data; - - *out = NULL; - - i_data = in->i_size; - p_data = in->p_buffer; - - for( ;; ) - { - /* decode frame */ - frame = id->p_ff_pic; - p_sys->i_input_pts = in->i_pts; - if( id->ff_dec ) - { - i_used = avcodec_decode_video( id->ff_dec_c, frame, - &b_gotpicture, - p_data, i_data ); - } - else - { - /* raw video */ - avpicture_fill( (AVPicture*)frame, - p_data, - id->ff_dec_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); - i_used = i_data; - b_gotpicture = 1; - - /* Set PTS */ - frame->pts = p_sys->i_input_pts; - } - - if( i_used < 0 ) - { - msg_Warn( p_stream, "error"); - return VLC_EGENERIC; - } - i_data -= i_used; - p_data += i_used; - - if( !b_gotpicture ) - { - return VLC_SUCCESS; - } - - /* Get the pts of the decoded frame if any, otherwise keep the - * interpolated one */ - if( frame->pts > 0 ) - { - p_sys->i_output_pts = frame->pts; - } - - if( !id->b_enc_inited && id->p_encoder ) - { - block_t *p_block; - - /* XXX hack because of copy packetizer and mpeg4video that can fail - * detecting size */ -#if 0 - if( id->p_encoder->i_width <= 0 ) - { - id->p_encoder->i_width = id->f_dst.i_width = - id->ff_dec_c->width - p_sys->i_crop_left - - p_sys->i_crop_right; - } - if( id->p_encoder->i_height <= 0 ) - { - id->p_encoder->i_height = id->f_dst.i_height = - id->ff_dec_c->height - p_sys->i_crop_top - - p_sys->i_crop_bottom; - } -#endif - - id->p_encoder->i_bitrate = p_sys->i_vbitrate; - - if( !( id->id = p_stream->p_sys->p_out->pf_add( p_stream->p_sys->p_out, &id->f_dst ) ) ) - { - msg_Err( p_stream, "cannot add this stream" ); - id->b_transcode = VLC_FALSE; - return VLC_EGENERIC; - } - - while( (p_block = id->p_encoder->pf_header( id->p_encoder )) ) - { - sout_buffer_t *p_out; - p_out = sout_BufferNew( p_stream->p_sout, p_block->i_buffer ); - memcpy( p_out->p_buffer, p_block->p_buffer, p_block->i_buffer); - sout_BufferChain( out, p_out ); - } - - id->b_enc_inited = VLC_TRUE; - } - else if( !id->b_enc_inited ) - { - /* XXX hack because of copy packetizer and mpeg4video that can fail - * detecting size */ - if( id->ff_enc_c->width <= 0 ) - { - id->ff_enc_c->width = - id->f_dst.i_width = id->ff_dec_c->width - p_sys->i_crop_left - p_sys->i_crop_right; - } - if( id->ff_enc_c->height <= 0 ) - { - id->ff_enc_c->height = - id->f_dst.i_height = id->ff_dec_c->height - p_sys->i_crop_top - p_sys->i_crop_bottom; - } - - /* Make sure we get extradata filled by the encoder */ - id->ff_enc_c->extradata_size = 0; - id->ff_enc_c->extradata = NULL; - id->ff_enc_c->flags |= CODEC_FLAG_GLOBAL_HEADER; - - if( avcodec_open( id->ff_enc_c, id->ff_enc ) ) - { - msg_Err( p_stream, "cannot open encoder" ); - return VLC_EGENERIC; - } - - id->f_dst.i_extra_data = id->ff_enc_c->extradata_size; - id->f_dst.p_extra_data = id->ff_enc_c->extradata; - id->ff_enc_c->flags &= ~CODEC_FLAG_GLOBAL_HEADER; - - if( !( id->id = p_stream->p_sys->p_out->pf_add( p_stream->p_sys->p_out, &id->f_dst ) ) ) - { - msg_Err( p_stream, "cannot add this stream" ); - transcode_video_ffmpeg_close( p_stream, id ); - id->b_transcode = VLC_FALSE; - return VLC_EGENERIC; - } - id->b_enc_inited = VLC_TRUE; - } + return video_new_buffer( VLC_OBJECT(p_dec), + p_dec->p_owner->pp_pics ); +} +static picture_t *video_new_buffer_filter( filter_t *p_filter ) +{ + return video_new_buffer( VLC_OBJECT(p_filter), + p_filter->p_owner->pp_pics ); +} - /* deinterlace */ - if( p_stream->p_sys->b_deinterlace ) - { - if( id->p_ff_pic_tmp0 == NULL ) - { - int i_size; - uint8_t *buf; - id->p_ff_pic_tmp0 = avcodec_alloc_frame(); - i_size = avpicture_get_size( id->ff_dec_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); +static void video_del_buffer( vlc_object_t *p_this, picture_t *p_pic ) +{ + if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig ); + if( p_pic && p_pic->p_sys ) free( p_pic->p_sys ); + if( p_pic ) free( p_pic ); +} - buf = malloc( i_size ); +static void video_del_buffer_decoder( decoder_t *p_decoder, picture_t *p_pic ) +{ + p_pic->i_refcount = 0; + p_pic->i_status = DESTROYED_PICTURE; +} - avpicture_fill( (AVPicture*)id->p_ff_pic_tmp0, buf, - id->ff_enc_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); - } +static void video_del_buffer_filter( filter_t *p_filter, picture_t *p_pic ) +{ + p_pic->i_refcount = 0; + p_pic->i_status = DESTROYED_PICTURE; +} - avpicture_deinterlace( (AVPicture*)id->p_ff_pic_tmp0, (AVPicture*)frame, - id->ff_dec_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); +static void video_link_picture_decoder( decoder_t *p_dec, picture_t *p_pic ) +{ + p_pic->i_refcount++; +} - frame = id->p_ff_pic_tmp0; - } +static void video_unlink_picture_decoder( decoder_t *p_dec, picture_t *p_pic ) +{ + video_release_buffer( p_pic ); +} - /* convert pix format */ - if( !id->p_encoder ) - if( id->ff_dec_c->pix_fmt != id->ff_enc_c->pix_fmt ) - { - if( id->p_ff_pic_tmp1 == NULL ) - { - int i_size; - uint8_t *buf; - id->p_ff_pic_tmp1 = avcodec_alloc_frame(); - i_size = avpicture_get_size( id->ff_enc_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); +/* + * SPU + */ +static subpicture_t *spu_new_buffer( decoder_t * ); +static void spu_del_buffer( decoder_t *, subpicture_t * ); - buf = malloc( i_size ); +static int transcode_spu_new( sout_stream_t *p_stream, sout_stream_id_t *id ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; - avpicture_fill( (AVPicture*)id->p_ff_pic_tmp1, buf, - id->ff_enc_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); - } + /* + * Open decoder + */ - img_convert( (AVPicture*)id->p_ff_pic_tmp1, id->ff_enc_c->pix_fmt, - (AVPicture*)frame, id->ff_dec_c->pix_fmt, - id->ff_dec_c->width, id->ff_dec_c->height ); + /* Initialization of decoder structures */ + id->p_decoder->pf_spu_buffer_new = spu_new_buffer; + id->p_decoder->pf_spu_buffer_del = spu_del_buffer; + id->p_decoder->p_owner = (decoder_owner_sys_t *)p_stream; + //id->p_decoder->p_cfg = p_sys->p_spu_cfg; - frame = id->p_ff_pic_tmp1; - } + id->p_decoder->p_module = + module_Need( id->p_decoder, "decoder", "$codec", 0 ); - /* convert size and crop */ - if( !id->p_encoder ) - if( id->ff_dec_c->width != id->ff_enc_c->width || - id->ff_dec_c->height != id->ff_enc_c->height || - p_sys->i_crop_top > 0 || p_sys->i_crop_bottom > 0 || - p_sys->i_crop_left > 0 || p_sys->i_crop_right ) - { - if( id->p_ff_pic_tmp2 == NULL ) - { - int i_size; - uint8_t *buf; - id->p_ff_pic_tmp2 = avcodec_alloc_frame(); - i_size = avpicture_get_size( id->ff_enc_c->pix_fmt, - id->ff_enc_c->width, id->ff_enc_c->height ); - - buf = malloc( i_size ); - - avpicture_fill( (AVPicture*)id->p_ff_pic_tmp2, buf, - id->ff_enc_c->pix_fmt, - id->ff_enc_c->width, id->ff_enc_c->height ); - - id->p_vresample = - img_resample_full_init( id->ff_enc_c->width, id->ff_enc_c->height, - id->ff_dec_c->width, id->ff_dec_c->height, - p_stream->p_sys->i_crop_top, - p_stream->p_sys->i_crop_bottom, - p_stream->p_sys->i_crop_left, - p_stream->p_sys->i_crop_right ); - } + if( !id->p_decoder->p_module ) + { + msg_Err( p_stream, "cannot find decoder" ); + return VLC_EGENERIC; + } - img_resample( id->p_vresample, (AVPicture*)id->p_ff_pic_tmp2, - (AVPicture*)frame ); + if( !p_sys->b_soverlay ) + { + /* + * Open encoder + */ - frame = id->p_ff_pic_tmp2; - } + /* Initialization of encoder format structures */ + es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat, + id->p_decoder->fmt_in.i_codec ); - /* Set the pts of the frame being encoded (segfaults with mpeg4!)*/ - if( id->f_dst.i_fourcc == VLC_FOURCC( 'm', 'p', 'g', 'v' ) ) - frame->pts = p_sys->i_output_pts; - else - frame->pts = 0; + id->p_encoder->p_cfg = p_sys->p_spu_cfg; - /* Interpolate the next PTS - * (needed by the mpeg video packetizer which can send pts <= 0 ) */ - if( id->ff_dec_c && id->ff_dec_c->frame_rate > 0 ) - { - p_sys->i_output_pts += I64C(1000000) * (2 + frame->repeat_pict) * - id->ff_dec_c->frame_rate_base / (2 * id->ff_dec_c->frame_rate); - } + id->p_encoder->p_module = + module_Need( id->p_encoder, "encoder", p_sys->psz_senc, VLC_TRUE ); - if( id->p_encoder ) + if( !id->p_encoder->p_module ) { - /* External encoding */ - block_t *p_block; - picture_t pic; - int i_plane; - - vout_InitPicture( VLC_OBJECT(p_stream), &pic, - id->ff_dec_c->width, id->ff_dec_c->height, - VLC_FOURCC('I','4','2','0') ); - - for( i_plane = 0; i_plane < pic.i_planes; i_plane++ ) - { - pic.p[i_plane].p_pixels = frame->data[i_plane]; - pic.p[i_plane].i_pitch = frame->linesize[i_plane]; - } - - pic.date = frame->pts; - - p_block = id->p_encoder->pf_encode_video( id->p_encoder, &pic ); - if( p_block ) - { - sout_buffer_t *p_out; - p_out = sout_BufferNew( p_stream->p_sout, p_block->i_buffer ); - memcpy( p_out->p_buffer, p_block->p_buffer, p_block->i_buffer); - sout_BufferChain( out, p_out ); - } - - return VLC_SUCCESS; + module_Unneed( id->p_decoder, id->p_decoder->p_module ); + msg_Err( p_stream, "cannot find encoder" ); + return VLC_EGENERIC; } + } - /* Let ffmpeg select the frame type */ - frame->pict_type = 0; + if( !p_sys->p_spu ) + { + p_sys->p_spu = spu_Create( p_stream ); + spu_Init( p_sys->p_spu ); + } - i_out = avcodec_encode_video( id->ff_enc_c, id->p_buffer, - id->i_buffer, frame ); - if( i_out > 0 ) - { - sout_buffer_t *p_out; - p_out = sout_BufferNew( p_stream->p_sout, i_out ); + return VLC_SUCCESS; +} - memcpy( p_out->p_buffer, id->p_buffer, i_out ); +static void transcode_spu_close( sout_stream_t *p_stream, sout_stream_id_t *id) +{ + /* Close decoder */ + if( id->p_decoder->p_module ) + module_Unneed( id->p_decoder, id->p_decoder->p_module ); - p_out->i_size = i_out; + /* Close encoder */ + if( id->p_encoder->p_module ) + module_Unneed( id->p_encoder, id->p_encoder->p_module ); +} - if( id->ff_enc_c->coded_frame->pts != 0 && - p_sys->i_buggy_pts_detect != id->ff_enc_c->coded_frame->pts ) - { - p_sys->i_buggy_pts_detect = id->ff_enc_c->coded_frame->pts; +static int transcode_spu_process( sout_stream_t *p_stream, + sout_stream_id_t *id, + block_t *in, block_t **out ) +{ + sout_stream_sys_t *p_sys = p_stream->p_sys; + subpicture_t *p_subpic; + *out = NULL; - /* FIXME, 3-2 pulldown is not handled correctly */ - p_out->i_length = in->i_length; - p_out->i_pts = id->ff_enc_c->coded_frame->pts; + p_subpic = id->p_decoder->pf_decode_sub( id->p_decoder, &in ); + if( p_subpic && p_sys->b_soverlay ) + { + spu_DisplaySubpicture( p_sys->p_spu, p_subpic ); + } - if( !id->ff_enc_c->delay || - ( id->ff_enc_c->coded_frame->pict_type != FF_I_TYPE && - id->ff_enc_c->coded_frame->pict_type != FF_P_TYPE ) ) - { - p_out->i_dts = p_out->i_pts; - } - else - { - if( p_sys->i_last_ref_pts ) - { - p_out->i_dts = p_sys->i_last_ref_pts; - } - else - { - /* Let's put something sensible */ - p_out->i_dts = p_out->i_pts; - } - - p_sys->i_last_ref_pts = p_out->i_pts; - } - } - else - { - /* Buggy libavcodec which doesn't update coded_frame->pts - * correctly */ - p_out->i_length = in->i_length; - p_out->i_dts = in->i_dts; - p_out->i_pts = in->i_dts; - } + if( p_subpic && !p_sys->b_soverlay ) + { + block_t *p_block; - sout_BufferChain( out, p_out ); - } + p_block = id->p_encoder->pf_encode_sub( id->p_encoder, p_subpic ); + spu_del_buffer( id->p_decoder, p_subpic ); - if( i_data <= 0 ) + if( p_block ) { + block_ChainAppend( out, p_block ); return VLC_SUCCESS; } } - return VLC_SUCCESS; + return VLC_EGENERIC; } -/***************************************************************************** - * transcode_video_ffmpeg_getframebuf: - * - * Callback used by ffmpeg to get a frame buffer. - * We use it to get the right PTS for each decoded picture. - *****************************************************************************/ -static int transcode_video_ffmpeg_getframebuf(struct AVCodecContext *p_context, - AVFrame *p_frame) +static subpicture_t *spu_new_buffer( decoder_t *p_dec ) { - sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_context->opaque; - - /* Set PTS */ - p_frame->pts = p_sys->i_input_pts; + sout_stream_t *p_stream = (sout_stream_t *)p_dec->p_owner; + return spu_CreateSubpicture( p_stream->p_sys->p_spu ); +} - return avcodec_default_get_buffer( p_context, p_frame ); +static void spu_del_buffer( decoder_t *p_dec, subpicture_t *p_subpic ) +{ + sout_stream_t *p_stream = (sout_stream_t *)p_dec->p_owner; + spu_DestroySubpicture( p_stream->p_sys->p_spu, p_subpic ); }