X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fmux%2Fogg.c;h=95bda99d98a98bff7e7373e1b162bae1363dd540;hb=3921a7f87de03160ecd18bcfeb68fbac459f2094;hp=e9997b7c8f6c23f8d45f428068d4ede653a49cbb;hpb=949a4ce261b3f115b283a4804b289a7ddd00b7a5;p=vlc diff --git a/modules/mux/ogg.c b/modules/mux/ogg.c index e9997b7c8f..95bda99d98 100644 --- a/modules/mux/ogg.c +++ b/modules/mux/ogg.c @@ -1,10 +1,11 @@ /***************************************************************************** - * ogg.c + * ogg.c: ogg muxer module for vlc ***************************************************************************** - * Copyright (C) 2001, 2002 VideoLAN - * $Id: ogg.c,v 1.3 2003/03/11 19:02:30 fenrir Exp $ + * Copyright (C) 2001, 2002, 2006 the VideoLAN team + * $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 @@ -18,70 +19,91 @@ * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA. + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************/ /***************************************************************************** * Preamble *****************************************************************************/ -#include -#include -#include -#include -#include -#include +#ifdef HAVE_TIME_H +# include +#endif -#include "codecs.h" +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif -#include +#include +#include +#include +#include +#include -/***************************************************************************** - * Exported prototypes - *****************************************************************************/ -static int Open ( vlc_object_t * ); -static void Close ( vlc_object_t * ); - -static int Capability(sout_mux_t *, int, void *, void * ); -static int AddStream( sout_mux_t *, sout_input_t * ); -static int DelStream( sout_mux_t *, sout_input_t * ); -static int Mux ( sout_mux_t * ); +#include /***************************************************************************** * Module descriptor *****************************************************************************/ +static int Open ( vlc_object_t * ); +static void Close ( vlc_object_t * ); + vlc_module_begin(); - set_description( _("Ogg/ogm muxer") ); + set_description( N_("Ogg/OGM muxer") ); set_capability( "sout mux", 10 ); + set_category( CAT_SOUT ); + set_subcategory( SUBCAT_SOUT_MUX ); add_shortcut( "ogg" ); add_shortcut( "ogm" ); set_callbacks( Open, Close ); vlc_module_end(); + /***************************************************************************** - * + * Exported prototypes *****************************************************************************/ -#define FREE( p ) if( p ) { free( p ); (p) = NULL; } +static int Control ( sout_mux_t *, int, va_list ); +static int AddStream( sout_mux_t *, sout_input_t * ); +static int DelStream( sout_mux_t *, sout_input_t * ); +static int Mux ( sout_mux_t * ); +static int MuxBlock ( sout_mux_t *, sout_input_t * ); + +static block_t *OggCreateHeader( sout_mux_t * ); +static block_t *OggCreateFooter( sout_mux_t * ); + +/***************************************************************************** + * Misc declarations + *****************************************************************************/ + +/* Structures used for OggDS headers used in ogm files */ #define PACKET_TYPE_HEADER 0x01 #define PACKET_TYPE_COMMENT 0x03 +#define PACKET_IS_SYNCPOINT 0x08 -#define PACKET_IS_SYNCPOINT 0x08 - -typedef struct __attribute__((__packed__)) +typedef struct +#ifdef HAVE_ATTRIBUTE_PACKED + __attribute__((__packed__)) +#endif { int32_t i_width; int32_t i_height; -} ogg_stream_header_video_t; +} oggds_header_video_t; -typedef struct __attribute__((__packed__)) +typedef struct +#ifdef HAVE_ATTRIBUTE_PACKED + __attribute__((__packed__)) +#endif { int16_t i_channels; int16_t i_block_align; int32_t i_avgbytespersec; -} ogg_stream_header_audio_t; +} oggds_header_audio_t; -typedef struct __attribute__((__packed__)) +typedef struct +#ifdef HAVE_ATTRIBUTE_PACKED + __attribute__((__packed__)) +#endif { uint8_t i_packet_type; @@ -96,65 +118,97 @@ typedef struct __attribute__((__packed__)) int32_t i_buffer_size; int16_t i_bits_per_sample; - int16_t i_padding_0; // hum hum + + int16_t i_padding_0; /* Because the original is using MSVC packing style */ + union { - ogg_stream_header_video_t video; - ogg_stream_header_audio_t audio; + oggds_header_video_t video; + oggds_header_audio_t audio; } header; -} ogg_stream_header_t; + int32_t i_padding_1; /* Because the original is using MSVC packing style */ + +} oggds_header_t; + +/* + * TODO move this function to src/stream_output.c (used by nearly all muxers) + */ +static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts ) +{ + mtime_t i_dts; + int i_stream, i; + for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ ) + { + block_fifo_t *p_fifo; + p_fifo = p_mux->pp_inputs[i]->p_fifo; + + /* We don't really need to have anything in the SPU fifo */ + if( p_mux->pp_inputs[i]->p_fmt->i_cat == SPU_ES && + block_FifoCount( p_fifo ) == 0 ) continue; + + if( block_FifoCount( p_fifo ) ) + { + block_t *p_buf; + + p_buf = block_FifoShow( p_fifo ); + if( i_stream < 0 || p_buf->i_dts < i_dts ) + { + i_dts = p_buf->i_dts; + i_stream = i; + } + } + else return -1; + + } + if( pi_stream ) *pi_stream = i_stream; + if( pi_dts ) *pi_dts = i_dts; + return i_stream; +} + +/***************************************************************************** + * Definitions of structures and functions used by this plugins + *****************************************************************************/ typedef struct { int i_cat; int i_fourcc; - ogg_stream_header_t header; + int b_new; - int i_packet_no; + mtime_t i_dts; + mtime_t i_length; + int i_packet_no; + int i_serial_no; + int i_keyframe_granule_shift; /* Theora only */ + ogg_stream_state os; - mtime_t i_dts; - mtime_t i_length; - ogg_stream_state os; + oggds_header_t *p_oggds_header; } ogg_stream_t; struct sout_mux_sys_t { - int b_write_header; - int i_streams; + int i_streams; -}; + mtime_t i_start_dts; + int i_next_serial_no; -#define SetWLE( p, v ) _SetWLE( (uint8_t*)p, v) -static void _SetWLE( uint8_t *p, uint16_t i_dw ) -{ - p[1] = ( i_dw >> 8 )&0xff; - p[0] = ( i_dw )&0xff; -} + /* number of logical streams pending to be added */ + int i_add_streams; -#define SetDWLE( p, v ) _SetDWLE( (uint8_t*)p, v) -static void _SetDWLE( uint8_t *p, uint32_t i_dw ) -{ - p[3] = ( i_dw >> 24 )&0xff; - p[2] = ( i_dw >> 16 )&0xff; - p[1] = ( i_dw >> 8 )&0xff; - p[0] = ( i_dw )&0xff; -} -#define SetQWLE( p, v ) _SetQWLE( (uint8_t*)p, v) -static void _SetQWLE( uint8_t *p, uint64_t i_qw ) -{ - SetDWLE( p, i_qw&0xffffffff ); - SetDWLE( p+4, ( i_qw >> 32)&0xffffffff ); -} + /* logical streams pending to be deleted */ + int i_del_streams; + ogg_stream_t **pp_del_streams; +}; -static void OggSetDate( sout_buffer_t *, mtime_t , mtime_t ); -static sout_buffer_t *OggStreamFlush( sout_mux_t *, ogg_stream_state *, mtime_t ); +static void OggSetDate( block_t *, mtime_t , mtime_t ); +static block_t *OggStreamFlush( sout_mux_t *, ogg_stream_state *, mtime_t ); /***************************************************************************** - * Open: + * Open: Open muxer *****************************************************************************/ static int Open( vlc_object_t *p_this ) { @@ -165,339 +219,521 @@ static int Open( vlc_object_t *p_this ) p_sys = malloc( sizeof( sout_mux_sys_t ) ); p_sys->i_streams = 0; - p_sys->b_write_header = VLC_TRUE; + p_sys->i_add_streams = 0; + p_sys->i_del_streams = 0; + p_sys->pp_del_streams = 0; p_mux->p_sys = p_sys; - p_mux->pf_capacity = Capability; + p_mux->pf_control = Control; p_mux->pf_addstream = AddStream; p_mux->pf_delstream = DelStream; p_mux->pf_mux = Mux; - p_mux->i_preheader = 1; + + /* First serial number is random. + * (Done like this because on win32 you need to seed the random number + * generator once per thread). */ + srand( (unsigned int)time( NULL ) ); + p_sys->i_next_serial_no = rand(); return VLC_SUCCESS; } /***************************************************************************** - * Close: + * Close: Finalize ogg bitstream and close muxer *****************************************************************************/ - static void Close( vlc_object_t * p_this ) { - sout_mux_t *p_mux = (sout_mux_t*)p_this; - sout_mux_sys_t *p_sys = p_mux->p_sys; + sout_mux_t *p_mux = (sout_mux_t*)p_this; + sout_mux_sys_t *p_sys = p_mux->p_sys; msg_Info( p_mux, "Close" ); + if( p_sys->i_del_streams ) + { + block_t *p_og = NULL; + mtime_t i_dts = -1; + int i; + + /* Close the current ogg stream */ + msg_Dbg( p_mux, "writing footer" ); + block_ChainAppend( &p_og, OggCreateFooter( p_mux ) ); + + /* Remove deleted logical streams */ + for( i = 0; i < p_sys->i_del_streams; i++ ) + { + i_dts = p_sys->pp_del_streams[i]->i_dts; + ogg_stream_clear( &p_sys->pp_del_streams[i]->os ); + FREENULL( p_sys->pp_del_streams[i]->p_oggds_header ); + FREENULL( p_sys->pp_del_streams[i] ); + } + FREENULL( p_sys->pp_del_streams ); + p_sys->i_streams -= p_sys->i_del_streams; + + /* Write footer */ + OggSetDate( p_og, i_dts, 0 ); + sout_AccessOutWrite( p_mux->p_access, p_og ); + } + free( p_sys ); } -static int Capability( sout_mux_t *p_mux, int i_query, void *p_args, void *p_answer ) +/***************************************************************************** + * Control: + *****************************************************************************/ +static int Control( sout_mux_t *p_mux, int i_query, va_list args ) { + VLC_UNUSED(p_mux); + bool *pb_bool; + char **ppsz; + switch( i_query ) { - case SOUT_MUX_CAP_GET_ADD_STREAM_ANY_TIME: - *(vlc_bool_t*)p_answer = VLC_FALSE; - return( SOUT_MUX_CAP_ERR_OK ); + case MUX_CAN_ADD_STREAM_WHILE_MUXING: + pb_bool = (bool*)va_arg( args, bool * ); + *pb_bool = true; + return VLC_SUCCESS; + + case MUX_GET_ADD_STREAM_WAIT: + pb_bool = (bool*)va_arg( args, bool * ); + *pb_bool = true; + return VLC_SUCCESS; + + case MUX_GET_MIME: + ppsz = (char**)va_arg( args, char ** ); + *ppsz = strdup( "application/ogg" ); + return VLC_SUCCESS; + default: - return( SOUT_MUX_CAP_ERR_UNIMPLEMENTED ); + return VLC_EGENERIC; } } - +/***************************************************************************** + * AddStream: Add an elementary stream to the muxed stream + *****************************************************************************/ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) { - sout_mux_sys_t *p_sys = p_mux->p_sys; - ogg_stream_t *p_stream; - - BITMAPINFOHEADER *p_bih; - WAVEFORMATEX *p_wf; + sout_mux_sys_t *p_sys = p_mux->p_sys; + ogg_stream_t *p_stream; + uint16_t i_tag; msg_Dbg( p_mux, "adding input" ); - p_input->p_sys = (void*)p_stream = malloc( sizeof( ogg_stream_t ) ); - p_stream->i_cat = p_input->input_format.i_cat; - p_stream->i_fourcc = p_input->input_format.i_fourcc; + p_input->p_sys = p_stream = malloc( sizeof( ogg_stream_t ) ); + + p_stream->i_cat = p_input->p_fmt->i_cat; + p_stream->i_fourcc = p_input->p_fmt->i_codec; + p_stream->i_serial_no = p_sys->i_next_serial_no++; p_stream->i_packet_no = 0; - p_stream->header.i_packet_type = PACKET_TYPE_HEADER; - switch( p_input->input_format.i_cat ) + p_stream->p_oggds_header = 0; + + switch( p_input->p_fmt->i_cat ) { - case VIDEO_ES: - p_bih = (BITMAPINFOHEADER*)p_input->input_format.p_format; - switch( p_input->input_format.i_fourcc ) + case VIDEO_ES: + if( !p_input->p_fmt->video.i_frame_rate || + !p_input->p_fmt->video.i_frame_rate_base ) + { + msg_Warn( p_mux, "Missing frame rate, assuming 25fps" ); + p_input->p_fmt->video.i_frame_rate = 25; + p_input->p_fmt->video.i_frame_rate_base = 1; + } + + switch( p_stream->i_fourcc ) + { + case VLC_FOURCC( 'm', 'p', 'g', 'v' ): + case VLC_FOURCC( 'm', 'p', '4', 'v' ): + case VLC_FOURCC( 'D', 'I', 'V', '3' ): + case VLC_FOURCC( 'M', 'J', 'P', 'G' ): + case VLC_FOURCC( 'W', 'M', 'V', '1' ): + case VLC_FOURCC( 'W', 'M', 'V', '2' ): + case VLC_FOURCC( 'W', 'M', 'V', '3' ): + case VLC_FOURCC( 'S', 'N', 'O', 'W' ): + case VLC_FOURCC( 'd', 'r', 'a', 'c' ): + p_stream->p_oggds_header = malloc( sizeof(oggds_header_t) ); + memset( p_stream->p_oggds_header, 0, sizeof(oggds_header_t) ); + p_stream->p_oggds_header->i_packet_type = PACKET_TYPE_HEADER; + + memcpy( p_stream->p_oggds_header->stream_type, "video", 5 ); + if( p_stream->i_fourcc == VLC_FOURCC( 'm', 'p', '4', 'v' ) ) { - case VLC_FOURCC( 'm', 'p','4', 'v' ): - case VLC_FOURCC( 'D', 'I','V', '3' ): - memcpy( p_stream->header.stream_type, - "video ", - 8 ); - if( p_input->input_format.i_fourcc == VLC_FOURCC( 'm', 'p','4', 'v' ) ) - { - memcpy( p_stream->header.sub_type, "XVID", 4 ); - } - else if( p_input->input_format.i_fourcc == VLC_FOURCC( 'D', 'I','V', '3' ) ) - { - memcpy( p_stream->header.sub_type, "DIV3", 4 ); - } - SetDWLE( &p_stream->header.i_size, sizeof( ogg_stream_header_t ) - 1); - /* XXX this won't make mplayer happy, but vlc can read that without any problem so...*/ - SetQWLE( &p_stream->header.i_time_unit, 10*1000 );//(int64_t)10*1000*1000/(int64_t)25 ); // FIXME (25fps) - SetQWLE( &p_stream->header.i_samples_per_unit, 1 ); - SetDWLE( &p_stream->header.i_default_len, 0 ); /* ??? */ - SetDWLE( &p_stream->header.i_buffer_size, 1024*1024 ); - SetWLE( &p_stream->header.i_bits_per_sample, 0 ); - if( p_bih ) - { - SetDWLE( &p_stream->header.header.video.i_width, p_bih->biWidth ); - SetDWLE( &p_stream->header.header.video.i_height, p_bih->biHeight ); - } - else - { - SetDWLE( &p_stream->header.header.video.i_width, 0 ); - SetDWLE( &p_stream->header.header.video.i_height, 0 ); - } - break; - default: - FREE( p_input->p_sys ); - return( VLC_EGENERIC ); + memcpy( p_stream->p_oggds_header->sub_type, "XVID", 4 ); } - break; - case AUDIO_ES: - p_wf = (WAVEFORMATEX*)p_input->input_format.p_format; - switch( p_input->input_format.i_fourcc ) + else if( p_stream->i_fourcc == VLC_FOURCC( 'D', 'I', 'V', '3' ) ) { - case VLC_FOURCC( 'm', 'p','g', 'a' ): - case VLC_FOURCC( 'a', '5','2', ' ' ): - memcpy( p_stream->header.stream_type, - "audio ", - 8 ); - if( p_input->input_format.i_fourcc == VLC_FOURCC( 'm', 'p','g', 'a' ) ) - { - memcpy( p_stream->header.sub_type, "55 ", 4 ); - } - else if( p_input->input_format.i_fourcc == VLC_FOURCC( 'a', '5','2', ' ' ) ) - { - memcpy( p_stream->header.sub_type, "2000", 4 ); - } - SetDWLE( &p_stream->header.i_size, sizeof( ogg_stream_header_t ) - 1); - SetQWLE( &p_stream->header.i_time_unit, 1000000 ); /* is it used ? */ - SetDWLE( &p_stream->header.i_default_len, 0 ); /* ??? */ - SetDWLE( &p_stream->header.i_buffer_size, 30*1024 ); - if( p_wf ) - { - SetQWLE( &p_stream->header.i_samples_per_unit, p_wf->nSamplesPerSec ); - SetWLE( &p_stream->header.i_bits_per_sample, p_wf->wBitsPerSample ); - SetDWLE( &p_stream->header.header.audio.i_channels, p_wf->nChannels ); - SetDWLE( &p_stream->header.header.audio.i_block_align, p_wf->nBlockAlign ); - SetDWLE( &p_stream->header.header.audio.i_avgbytespersec, p_wf->nAvgBytesPerSec ); - } - else - { - /* perhaps it's better to fail */ - SetQWLE( &p_stream->header.i_samples_per_unit, 44100 ); - SetWLE( &p_stream->header.i_bits_per_sample, 0 ); - SetDWLE( &p_stream->header.header.audio.i_channels, 2 ); - SetDWLE( &p_stream->header.header.audio.i_block_align, 0 ); - SetDWLE( &p_stream->header.header.audio.i_avgbytespersec, 0 ); - } - break; - case VLC_FOURCC( 'v', 'o', 'r', 'b' ): - default: - FREE( p_input->p_sys ); - return( VLC_EGENERIC ); + memcpy( p_stream->p_oggds_header->sub_type, "DIV3", 4 ); } + else + { + memcpy( p_stream->p_oggds_header->sub_type, + &p_stream->i_fourcc, 4 ); + } + SetDWLE( &p_stream->p_oggds_header->i_size, + sizeof( oggds_header_t ) - 1 ); + SetQWLE( &p_stream->p_oggds_header->i_time_unit, + INT64_C(10000000) * p_input->p_fmt->video.i_frame_rate_base / + (int64_t)p_input->p_fmt->video.i_frame_rate ); + SetQWLE( &p_stream->p_oggds_header->i_samples_per_unit, 1 ); + SetDWLE( &p_stream->p_oggds_header->i_default_len, 1 ); /* ??? */ + SetDWLE( &p_stream->p_oggds_header->i_buffer_size, 1024*1024 ); + SetWLE( &p_stream->p_oggds_header->i_bits_per_sample, 0 ); + SetDWLE( &p_stream->p_oggds_header->header.video.i_width, + p_input->p_fmt->video.i_width ); + SetDWLE( &p_stream->p_oggds_header->header.video.i_height, + p_input->p_fmt->video.i_height ); + msg_Dbg( p_mux, "%4.4s stream", (char *)&p_stream->i_fourcc ); break; + + case VLC_FOURCC( 't', 'h', 'e', 'o' ): + msg_Dbg( p_mux, "theora stream" ); + break; + default: - FREE( p_input->p_sys ); - return( VLC_EGENERIC ); - } + FREENULL( p_input->p_sys ); + return VLC_EGENERIC; + } + break; - ogg_stream_init (&p_stream->os, rand ()); + case AUDIO_ES: + switch( p_stream->i_fourcc ) + { + case VLC_FOURCC( 'v', 'o', 'r', 'b' ): + msg_Dbg( p_mux, "vorbis stream" ); + break; - p_sys->i_streams++; - return( VLC_SUCCESS ); -} + case VLC_FOURCC( 's', 'p', 'x', ' ' ): + msg_Dbg( p_mux, "speex stream" ); + break; -static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input ) -{ - ogg_stream_t *p_stream = (ogg_stream_t*)p_input->p_sys; - sout_buffer_t *p_og; + case VLC_FOURCC( 'f', 'l', 'a', 'c' ): + msg_Dbg( p_mux, "flac stream" ); + break; - msg_Dbg( p_mux, "removing input" ); + default: + fourcc_to_wf_tag( p_stream->i_fourcc, &i_tag ); + if( i_tag == WAVE_FORMAT_UNKNOWN ) + { + FREENULL( p_input->p_sys ); + return VLC_EGENERIC; + } - /* flush all remaining data */ + p_stream->p_oggds_header = + malloc( sizeof(oggds_header_t) + p_input->p_fmt->i_extra ); + memset( p_stream->p_oggds_header, 0, sizeof(oggds_header_t) ); + p_stream->p_oggds_header->i_packet_type = PACKET_TYPE_HEADER; - p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); - if( p_og ) - { - OggSetDate( p_og, p_stream->i_dts, p_stream->i_length ); + SetDWLE( &p_stream->p_oggds_header->i_size, + sizeof( oggds_header_t ) - 1 + p_input->p_fmt->i_extra ); - sout_AccessOutWrite( p_mux->p_access, p_og ); + if( p_input->p_fmt->i_extra ) + { + memcpy( &p_stream->p_oggds_header[1], + p_input->p_fmt->p_extra, p_input->p_fmt->i_extra ); + } + + memcpy( p_stream->p_oggds_header->stream_type, "audio", 5 ); + + memset( p_stream->p_oggds_header->sub_type, 0, 4 ); + sprintf( p_stream->p_oggds_header->sub_type, "%-x", i_tag ); + + SetQWLE( &p_stream->p_oggds_header->i_time_unit, INT64_C(10000000) ); + SetDWLE( &p_stream->p_oggds_header->i_default_len, 1 ); + SetDWLE( &p_stream->p_oggds_header->i_buffer_size, 30*1024 ); + SetQWLE( &p_stream->p_oggds_header->i_samples_per_unit, + p_input->p_fmt->audio.i_rate ); + SetWLE( &p_stream->p_oggds_header->i_bits_per_sample, + p_input->p_fmt->audio.i_bitspersample ); + SetDWLE( &p_stream->p_oggds_header->header.audio.i_channels, + p_input->p_fmt->audio.i_channels ); + SetDWLE( &p_stream->p_oggds_header->header.audio.i_block_align, + p_input->p_fmt->audio.i_blockalign ); + SetDWLE( &p_stream->p_oggds_header->header.audio.i_avgbytespersec, + p_input->p_fmt->i_bitrate / 8); + msg_Dbg( p_mux, "%4.4s stream", (char *)&p_stream->i_fourcc ); + break; + } + break; + + case SPU_ES: + switch( p_stream->i_fourcc ) + { + case VLC_FOURCC( 's', 'u','b', 't' ): + p_stream->p_oggds_header = malloc( sizeof(oggds_header_t) ); + memset( p_stream->p_oggds_header, 0, sizeof(oggds_header_t) ); + p_stream->p_oggds_header->i_packet_type = PACKET_TYPE_HEADER; + + memcpy( p_stream->p_oggds_header->stream_type, "text", 4 ); + msg_Dbg( p_mux, "subtitles stream" ); + break; + + default: + FREENULL( p_input->p_sys ); + return VLC_EGENERIC; + } + break; + default: + FREENULL( p_input->p_sys ); + return VLC_EGENERIC; } - ogg_stream_clear( &p_stream->os ); + p_stream->b_new = true; - FREE( p_input->p_sys ); - return( 0 ); + p_sys->i_add_streams++; + + return VLC_SUCCESS; } -/* - * TODO move this function to src/stream_output.c (used by nearly all muxers) - */ -static int MuxGetStream( sout_mux_t *p_mux, - int *pi_stream, - mtime_t *pi_dts ) +/***************************************************************************** + * DelStream: Delete an elementary stream from the muxed stream + *****************************************************************************/ +static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input ) { - mtime_t i_dts; - int i_stream; - int i; + sout_mux_sys_t *p_sys = p_mux->p_sys; + ogg_stream_t *p_stream = (ogg_stream_t*)p_input->p_sys; + block_t *p_og; - for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ ) - { - sout_fifo_t *p_fifo; + msg_Dbg( p_mux, "removing input" ); - p_fifo = p_mux->pp_inputs[i]->p_fifo; + /* flush all remaining data */ + if( p_input->p_sys ) + { + if( !p_stream->b_new ) + { + while( block_FifoCount( p_input->p_fifo ) ) + MuxBlock( p_mux, p_input ); + } - if( p_fifo->i_depth > 1 ) + if( !p_stream->b_new && + ( p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ) ) ) { - sout_buffer_t *p_buf; + OggSetDate( p_og, p_stream->i_dts, p_stream->i_length ); + sout_AccessOutWrite( p_mux->p_access, p_og ); + } - p_buf = sout_FifoShow( p_fifo ); - if( i_stream < 0 || p_buf->i_dts < i_dts ) - { - i_dts = p_buf->i_dts; - i_stream = i; - } + /* move input in delete queue */ + if( !p_stream->b_new ) + { + p_sys->pp_del_streams = realloc( p_sys->pp_del_streams, + (p_sys->i_del_streams + 1) * + sizeof(ogg_stream_t *) ); + p_sys->pp_del_streams[p_sys->i_del_streams++] = p_stream; } else { - return( -1 ); // wait that all fifo have at least 2 packets + /* wasn't already added so get rid of it */ + FREENULL( p_stream->p_oggds_header ); + FREENULL( p_stream ); + p_sys->i_add_streams--; } } - if( pi_stream ) - { - *pi_stream = i_stream; - } - if( pi_dts ) - { - *pi_dts = i_dts; - } - return( i_stream ); -} + p_input->p_sys = NULL; + + return 0; +} -static sout_buffer_t *OggStreamFlush( sout_mux_t *p_mux, - ogg_stream_state *p_os, - mtime_t i_pts ) +/***************************************************************************** + * Ogg bitstream manipulation routines + *****************************************************************************/ +static block_t *OggStreamFlush( sout_mux_t *p_mux, + ogg_stream_state *p_os, mtime_t i_pts ) { - sout_buffer_t *p_og, *p_og_first = NULL; - ogg_page og; + block_t *p_og, *p_og_first = NULL; + ogg_page og; - for( ;; ) + while( ogg_stream_flush( p_os, &og ) ) { - /* flush all data */ - int i_result; - int i_size; - if( ( i_result = ogg_stream_flush( p_os, &og ) ) == 0 ) - { - break; - } - i_size = og.header_len + og.body_len; - p_og = sout_BufferNew( p_mux->p_sout, i_size); - - memcpy( p_og->p_buffer, - og.header, - og.header_len ); - memcpy( p_og->p_buffer + og.header_len, - og.body, - og.body_len ); - p_og->i_size = i_size; + /* Flush all data */ + p_og = block_New( p_mux, og.header_len + og.body_len ); + + memcpy( p_og->p_buffer, og.header, og.header_len ); + memcpy( p_og->p_buffer + og.header_len, og.body, og.body_len ); p_og->i_dts = 0; p_og->i_pts = i_pts; p_og->i_length = 0; - i_pts = 0; // write it only once + i_pts = 0; // write it only once - sout_BufferChain( &p_og_first, p_og ); + block_ChainAppend( &p_og_first, p_og ); } - return( p_og_first ); + return p_og_first; } -static sout_buffer_t *OggStreamPageOut( sout_mux_t *p_mux, - ogg_stream_state *p_os, - mtime_t i_pts ) + +static block_t *OggStreamPageOut( sout_mux_t *p_mux, + ogg_stream_state *p_os, mtime_t i_pts ) { - sout_buffer_t *p_og, *p_og_first = NULL; - ogg_page og; + block_t *p_og, *p_og_first = NULL; + ogg_page og; - for( ;; ) + while( ogg_stream_pageout( p_os, &og ) ) { - /* flush all data */ - int i_result; - int i_size; - if( ( i_result = ogg_stream_pageout( p_os, &og ) ) == 0 ) - { - break; - } - i_size = og.header_len + og.body_len; - p_og = sout_BufferNew( p_mux->p_sout, i_size); - - memcpy( p_og->p_buffer, - og.header, - og.header_len ); - memcpy( p_og->p_buffer + og.header_len, - og.body, - og.body_len ); - p_og->i_size = i_size; + /* Flush all data */ + p_og = block_New( p_mux, og.header_len + og.body_len ); + + memcpy( p_og->p_buffer, og.header, og.header_len ); + memcpy( p_og->p_buffer + og.header_len, og.body, og.body_len ); p_og->i_dts = 0; p_og->i_pts = i_pts; p_og->i_length = 0; - i_pts = 0; // write them only once + i_pts = 0; // write them only once - sout_BufferChain( &p_og_first, p_og ); + block_ChainAppend( &p_og_first, p_og ); } - return( p_og_first ); + return p_og_first; } -static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts ) -{ - sout_buffer_t *p_hdr = NULL; - sout_buffer_t *p_og; - ogg_packet op; - int i; - /* write header for each stream */ +static block_t *OggCreateHeader( sout_mux_t *p_mux ) +{ + block_t *p_hdr = NULL; + block_t *p_og = NULL; + ogg_packet op; + uint8_t *p_extra; + int i, i_extra; + + /* Write header for each stream. All b_o_s (beginning of stream) packets + * must appear first in the ogg stream so we take care of them first. */ for( i = 0; i < p_mux->i_nb_inputs; i++ ) { - ogg_stream_t *p_stream; + sout_input_t *p_input = p_mux->pp_inputs[i]; + ogg_stream_t *p_stream = (ogg_stream_t*)p_input->p_sys; + p_stream->b_new = false; + + msg_Dbg( p_mux, "creating header for %4.4s", + (char *)&p_stream->i_fourcc ); - p_stream = (ogg_stream_t*)p_mux->pp_inputs[i]->p_sys; + ogg_stream_init( &p_stream->os, p_stream->i_serial_no ); + p_stream->i_packet_no = 0; - if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) ) + if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) || + p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) || + p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) ) { - /* special case */ + /* First packet in order: vorbis/speex/theora info */ + p_extra = p_input->p_fmt->p_extra; + i_extra = p_input->p_fmt->i_extra; + + op.bytes = *(p_extra++) << 8; + op.bytes |= (*(p_extra++) & 0xFF); + op.packet = p_extra; + i_extra -= (op.bytes + 2); + if( i_extra < 0 ) + { + msg_Err( p_mux, "header data corrupted"); + op.bytes += i_extra; + } + + op.b_o_s = 1; + op.e_o_s = 0; + op.granulepos = 0; + op.packetno = p_stream->i_packet_no++; + ogg_stream_packetin( &p_stream->os, &op ); + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + + /* Get keyframe_granule_shift for theora granulepos calculation */ + if( p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) ) + { + int i_keyframe_frequency_force = + 1 << ((op.packet[40] << 6 >> 3) | (op.packet[41] >> 5)); + + /* granule_shift = i_log( frequency_force -1 ) */ + p_stream->i_keyframe_granule_shift = 0; + i_keyframe_frequency_force--; + while( i_keyframe_frequency_force ) + { + p_stream->i_keyframe_granule_shift++; + i_keyframe_frequency_force >>= 1; + } + } } - else + else if( p_stream->i_fourcc == VLC_FOURCC( 'f', 'l', 'a', 'c' ) ) + { + /* flac stream marker (yeah, only that in the 1st packet) */ + op.packet = (unsigned char *)"fLaC"; + op.bytes = 4; + op.b_o_s = 1; + op.e_o_s = 0; + op.granulepos = 0; + op.packetno = p_stream->i_packet_no++; + ogg_stream_packetin( &p_stream->os, &op ); + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + } + else if( p_stream->p_oggds_header ) { /* ds header */ -#if 0 - uint8_t com[128]; - int i_com; -#endif - - /* header */ - op.packet = (uint8_t*)&p_stream->header; - op.bytes = sizeof( ogg_stream_t ); + op.packet = (uint8_t*)p_stream->p_oggds_header; + op.bytes = p_stream->p_oggds_header->i_size + 1; op.b_o_s = 1; op.e_o_s = 0; op.granulepos = 0; op.packetno = p_stream->i_packet_no++; ogg_stream_packetin( &p_stream->os, &op ); -#if 0 - /* I don't know why but this produce broken file */ + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + } + + block_ChainAppend( &p_hdr, p_og ); + } + + /* Take care of the non b_o_s headers */ + for( i = 0; i < p_mux->i_nb_inputs; i++ ) + { + sout_input_t *p_input = p_mux->pp_inputs[i]; + ogg_stream_t *p_stream = (ogg_stream_t*)p_input->p_sys; + + if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) || + p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) || + p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) ) + { + /* Special case, headers are already there in the incoming stream. + * We need to gather them an mark them as headers. */ + int j = 2; + + if( p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) ) j = 1; + + p_extra = p_input->p_fmt->p_extra; + i_extra = p_input->p_fmt->i_extra; + + /* Skip 1 header */ + op.bytes = *(p_extra++) << 8; + op.bytes |= (*(p_extra++) & 0xFF); + op.packet = p_extra; + p_extra += op.bytes; + i_extra -= (op.bytes + 2); + + while( j-- ) + { + op.bytes = *(p_extra++) << 8; + op.bytes |= (*(p_extra++) & 0xFF); + op.packet = p_extra; + p_extra += op.bytes; + i_extra -= (op.bytes + 2); + if( i_extra < 0 ) + { + msg_Err( p_mux, "header data corrupted"); + op.bytes += i_extra; + } + + op.b_o_s = 0; + op.e_o_s = 0; + op.granulepos = 0; + op.packetno = p_stream->i_packet_no++; + ogg_stream_packetin( &p_stream->os, &op ); + + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + block_ChainAppend( &p_hdr, p_og ); + } + } + else if( p_stream->i_fourcc != VLC_FOURCC( 'f', 'l', 'a', 'c' ) ) + { + uint8_t com[128]; + int i_com; + /* comment */ com[0] = PACKET_TYPE_COMMENT; - i_com = snprintf( &com[1], 128, "VLC 0.5.x stream output" ) + 1; + i_com = snprintf( (char *)(com+1), 127, + PACKAGE_VERSION" stream output" ) + + 1; op.packet = com; op.bytes = i_com; op.b_o_s = 0; @@ -505,26 +741,110 @@ static sout_buffer_t *OggCreateHeader( sout_mux_t *p_mux, mtime_t i_dts ) op.granulepos = 0; op.packetno = p_stream->i_packet_no++; ogg_stream_packetin( &p_stream->os, &op ); -#endif + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + block_ChainAppend( &p_hdr, p_og ); } - p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); - sout_BufferChain( &p_hdr, p_og ); + /* Special case for mp4v and flac */ + if( ( p_stream->i_fourcc == VLC_FOURCC( 'm', 'p', '4', 'v' ) || + p_stream->i_fourcc == VLC_FOURCC( 'f', 'l', 'a', 'c' ) ) && + p_input->p_fmt->i_extra ) + { + /* Send a packet with the VOL data for mp4v + * or STREAMINFO for flac */ + msg_Dbg( p_mux, "writing extra data" ); + op.bytes = p_input->p_fmt->i_extra; + op.packet = p_input->p_fmt->p_extra; + if( p_stream->i_fourcc == VLC_FOURCC( 'f', 'l', 'a', 'c' ) ) + { + /* Skip the flac stream marker */ + op.bytes -= 4; + op.packet+= 4; + } + op.b_o_s = 0; + op.e_o_s = 0; + op.granulepos = 0; + op.packetno = p_stream->i_packet_no++; + ogg_stream_packetin( &p_stream->os, &op ); + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + block_ChainAppend( &p_hdr, p_og ); + } } /* set HEADER flag */ for( p_og = p_hdr; p_og != NULL; p_og = p_og->p_next ) { - p_og->i_flags |= SOUT_BUFFER_FLAGS_HEADER; + p_og->i_flags |= BLOCK_FLAG_HEADER; } - return( p_hdr ); + return p_hdr; } +static block_t *OggCreateFooter( sout_mux_t *p_mux ) +{ + sout_mux_sys_t *p_sys = p_mux->p_sys; + block_t *p_hdr = NULL; + block_t *p_og; + ogg_packet op; + int i; + + /* flush all remaining data */ + for( i = 0; i < p_mux->i_nb_inputs; i++ ) + { + ogg_stream_t *p_stream = p_mux->pp_inputs[i]->p_sys; -static void OggSetDate( sout_buffer_t *p_og, mtime_t i_dts, mtime_t i_length ) + /* skip newly added streams */ + if( p_stream->b_new ) continue; + + if( ( p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ) ) ) + { + OggSetDate( p_og, p_stream->i_dts, p_stream->i_length ); + sout_AccessOutWrite( p_mux->p_access, p_og ); + } + } + + /* Write eos packets for each stream. */ + for( i = 0; i < p_mux->i_nb_inputs; i++ ) + { + ogg_stream_t *p_stream = p_mux->pp_inputs[i]->p_sys; + + /* skip newly added streams */ + if( p_stream->b_new ) continue; + + op.packet = NULL; + op.bytes = 0; + op.b_o_s = 0; + op.e_o_s = 1; + op.granulepos = -1; + op.packetno = p_stream->i_packet_no++; + ogg_stream_packetin( &p_stream->os, &op ); + + p_og = OggStreamFlush( p_mux, &p_stream->os, 0 ); + block_ChainAppend( &p_hdr, p_og ); + ogg_stream_clear( &p_stream->os ); + } + + for( i = 0; i < p_sys->i_del_streams; i++ ) + { + op.packet = NULL; + op.bytes = 0; + op.b_o_s = 0; + op.e_o_s = 1; + op.granulepos = -1; + op.packetno = p_sys->pp_del_streams[i]->i_packet_no++; + ogg_stream_packetin( &p_sys->pp_del_streams[i]->os, &op ); + + p_og = OggStreamFlush( p_mux, &p_sys->pp_del_streams[i]->os, 0 ); + block_ChainAppend( &p_hdr, p_og ); + ogg_stream_clear( &p_sys->pp_del_streams[i]->os ); + } + + return p_hdr; +} + +static void OggSetDate( block_t *p_og, mtime_t i_dts, mtime_t i_length ) { int i_count; - sout_buffer_t *p_tmp; + block_t *p_tmp; mtime_t i_delta; for( p_tmp = p_og, i_count = 0; p_tmp != NULL; p_tmp = p_tmp->p_next ) @@ -542,85 +862,157 @@ static void OggSetDate( sout_buffer_t *p_og, mtime_t i_dts, mtime_t i_length ) } } -static int Mux ( sout_mux_t *p_mux ) +/***************************************************************************** + * Mux: multiplex available data in input fifos into the Ogg bitstream + *****************************************************************************/ +static int Mux( sout_mux_t *p_mux ) { - sout_mux_sys_t *p_sys = p_mux->p_sys; - sout_buffer_t *p_og = NULL; - int i_stream; - mtime_t i_dts; + sout_mux_sys_t *p_sys = p_mux->p_sys; + block_t *p_og = NULL; + int i_stream; + mtime_t i_dts; - if( p_sys->b_write_header ) + if( p_sys->i_add_streams || p_sys->i_del_streams ) { + /* Open new ogg stream */ if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 ) { - return( VLC_SUCCESS ); + msg_Dbg( p_mux, "waiting for data..." ); + return VLC_SUCCESS; } - sout_BufferChain( &p_og, OggCreateHeader( p_mux, i_dts ) ); - p_sys->b_write_header = VLC_FALSE; - } - - for( ;; ) - { - sout_input_t *p_input; - ogg_stream_t *p_stream; - sout_buffer_t *p_data; - ogg_packet op; - if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 ) + if( p_sys->i_streams ) { - return( VLC_SUCCESS ); + /* Close current ogg stream */ + int i; + + msg_Dbg( p_mux, "writing footer" ); + block_ChainAppend( &p_og, OggCreateFooter( p_mux ) ); + + /* Remove deleted logical streams */ + for( i = 0; i < p_sys->i_del_streams; i++ ) + { + FREENULL( p_sys->pp_del_streams[i]->p_oggds_header ); + FREENULL( p_sys->pp_del_streams[i] ); + } + FREENULL( p_sys->pp_del_streams ); + p_sys->i_streams = 0; } - p_input = p_mux->pp_inputs[i_stream]; - p_stream = (ogg_stream_t*)p_input->p_sys; + msg_Dbg( p_mux, "writing header" ); + p_sys->i_start_dts = i_dts; + p_sys->i_streams = p_mux->i_nb_inputs; + p_sys->i_del_streams = 0; + p_sys->i_add_streams = 0; + block_ChainAppend( &p_og, OggCreateHeader( p_mux ) ); + + /* Write header and/or footer */ + OggSetDate( p_og, i_dts, 0 ); + sout_AccessOutWrite( p_mux->p_access, p_og ); + p_og = NULL; + } + + for( ;; ) + { + if( MuxGetStream( p_mux, &i_stream, 0 ) < 0 ) return VLC_SUCCESS; + MuxBlock( p_mux, p_mux->pp_inputs[i_stream] ); + } - p_data = sout_FifoGet( p_input->p_fifo ); + return VLC_SUCCESS; +} - sout_BufferReallocFromPreHeader( p_mux->p_sout, p_data, 1 ); +static int MuxBlock( sout_mux_t *p_mux, sout_input_t *p_input ) +{ + sout_mux_sys_t *p_sys = p_mux->p_sys; + ogg_stream_t *p_stream = (ogg_stream_t*)p_input->p_sys; + block_t *p_data = block_FifoGet( p_input->p_fifo ); + block_t *p_og = NULL; + ogg_packet op; + + if( p_stream->i_fourcc != VLC_FOURCC( 'v', 'o', 'r', 'b' ) && + p_stream->i_fourcc != VLC_FOURCC( 'f', 'l', 'a', 'c' ) && + p_stream->i_fourcc != VLC_FOURCC( 's', 'p', 'x', ' ' ) && + p_stream->i_fourcc != VLC_FOURCC( 't', 'h', 'e', 'o' ) ) + { + p_data = block_Realloc( p_data, 1, p_data->i_buffer ); p_data->p_buffer[0] = PACKET_IS_SYNCPOINT; // FIXME + } - op.packet = p_data->p_buffer; - op.bytes = p_data->i_size; - op.b_o_s = 0; - op.e_o_s = 0; - if( p_stream->i_cat == AUDIO_ES ) + op.packet = p_data->p_buffer; + op.bytes = p_data->i_buffer; + op.b_o_s = 0; + op.e_o_s = 0; + op.packetno = p_stream->i_packet_no++; + + if( p_stream->i_cat == AUDIO_ES ) + { + if( p_stream->i_fourcc == VLC_FOURCC( 'v', 'o', 'r', 'b' ) || + p_stream->i_fourcc == VLC_FOURCC( 'f', 'l', 'a', 'c' ) || + p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) ) + { + /* number of sample from begining + current packet */ + op.granulepos = + ( p_data->i_dts - p_sys->i_start_dts + p_data->i_length ) * + (mtime_t)p_input->p_fmt->audio.i_rate / INT64_C(1000000); + } + else if( p_stream->p_oggds_header ) { /* number of sample from begining */ - op.granulepos = i_dts * p_stream->header.i_samples_per_unit / (int64_t)1000000; + op.granulepos = ( p_data->i_dts - p_sys->i_start_dts ) * + p_stream->p_oggds_header->i_samples_per_unit / INT64_C(1000000); } - else if( p_stream->i_cat == VIDEO_ES ) + } + else if( p_stream->i_cat == VIDEO_ES ) + { + if( p_stream->i_fourcc == VLC_FOURCC( 't', 'h', 'e', 'o' ) ) { - op.granulepos = i_dts/1000;//p_stream->i_packet_no; + /* FIXME, we assume only keyframes */ + op.granulepos = ( ( p_data->i_dts - p_sys->i_start_dts ) * + p_input->p_fmt->video.i_frame_rate / + p_input->p_fmt->video.i_frame_rate_base / + INT64_C(1000000) ) << p_stream->i_keyframe_granule_shift; } - op.packetno = p_stream->i_packet_no++; - ogg_stream_packetin( &p_stream->os, &op ); + else if( p_stream->p_oggds_header ) + op.granulepos = ( p_data->i_dts - p_sys->i_start_dts ) * INT64_C(10) / + p_stream->p_oggds_header->i_time_unit; + } + else if( p_stream->i_cat == SPU_ES ) + { + /* granulepos is in milisec */ + op.granulepos = ( p_data->i_dts - p_sys->i_start_dts ) / 1000; + } - sout_BufferChain( &p_og, - OggStreamPageOut( p_mux, - &p_stream->os, - p_data->i_pts ) ); + ogg_stream_packetin( &p_stream->os, &op ); - if( p_og ) - { - OggSetDate( p_og, p_stream->i_dts, p_stream->i_length ); - p_stream->i_dts = -1; - p_stream->i_length = 0;; - sout_AccessOutWrite( p_mux->p_access, p_og ); + if( p_stream->i_cat == SPU_ES || + p_stream->i_fourcc == VLC_FOURCC( 's', 'p', 'x', ' ' ) ) + { + /* Subtitles or Speex packets are quite small so they + * need to be flushed to be sent on time */ + p_og = OggStreamFlush( p_mux, &p_stream->os, p_data->i_dts ); + } + else + { + p_og = OggStreamPageOut( p_mux, &p_stream->os, p_data->i_dts ); + } - p_og = NULL; - } - else + if( p_og ) + { + OggSetDate( p_og, p_stream->i_dts, p_stream->i_length ); + p_stream->i_dts = -1; + p_stream->i_length = 0; + + sout_AccessOutWrite( p_mux->p_access, p_og ); + } + else + { + if( p_stream->i_dts < 0 ) { - if( p_stream->i_dts < 0 ) - { - p_stream->i_dts = p_data->i_dts; - } - p_stream->i_length += p_data->i_length; + p_stream->i_dts = p_data->i_dts; } - - sout_BufferDelete( p_mux->p_sout, p_data ); + p_stream->i_length += p_data->i_length; } - return( VLC_SUCCESS ); + block_Release( p_data ); + return VLC_SUCCESS; } -