X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fmux%2Fasf.c;h=e6a21a77819e43a611ce62d7f696d1b19d324719;hb=16ceb0de696197f901935b5cd05e1018e34e76d8;hp=fcb7f2de9258e6b9e4571fcdc14623f5da546c41;hpb=c219dac6699f1deb0d6b1143ad44f49fcef318bf;p=vlc diff --git a/modules/mux/asf.c b/modules/mux/asf.c index fcb7f2de92..e6a21a7781 100644 --- a/modules/mux/asf.c +++ b/modules/mux/asf.c @@ -1,10 +1,11 @@ /***************************************************************************** * asf.c: asf muxer module for vlc ***************************************************************************** - * Copyright (C) 2003-2004 VideoLAN + * Copyright (C) 2003-2004, 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,19 +19,27 @@ * * 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 +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include +#include +#include +#include +#include -#include "codecs.h" +typedef GUID guid_t; + +#define MAX_ASF_TRACKS 128 +#define ASF_DATA_PACKET_SIZE 4096 // deprecated -- added sout-asf-packet-size /***************************************************************************** * Module descriptor @@ -41,44 +50,53 @@ static void Close ( vlc_object_t * ); #define SOUT_CFG_PREFIX "sout-asf-" #define TITLE_TEXT N_("Title") -#define TITLE_LONGTEXT N_("Allows you to define the title that will be put " \ - "in ASF comments.") +#define TITLE_LONGTEXT N_("Title to put in ASF comments." ) #define AUTHOR_TEXT N_("Author") -#define AUTHOR_LONGTEXT N_("Allows you to define the author that will be put "\ - "in ASF comments.") +#define AUTHOR_LONGTEXT N_("Author to put in ASF comments." ) #define COPYRIGHT_TEXT N_("Copyright") -#define COPYRIGHT_LONGTEXT N_("Allows you to define the copyright string " \ - "that will be put in ASF comments.") +#define COPYRIGHT_LONGTEXT N_("Copyright string to put in ASF comments." ) #define COMMENT_TEXT N_("Comment") -#define COMMENT_LONGTEXT N_("Allows you to define the comment that will be " \ - "put in ASF comments.") +#define COMMENT_LONGTEXT N_("Comment to put in ASF comments." ) #define RATING_TEXT N_("Rating") -#define RATING_LONGTEXT N_("Allows you to define the \"rating\" that will " \ - "be put in ASF comments.") +#define RATING_LONGTEXT N_("\"Rating\" to put in ASF comments." ) +#define PACKETSIZE_TEXT N_("Packet Size") +#define PACKETSIZE_LONGTEXT N_("ASF packet size -- default is 4096 bytes") +#define BITRATE_TEXT N_("Bitrate override") +#define BITRATE_LONGTEXT N_("Do not try to guess ASF bitrate. Setting this, allows you to control how Windows Media Player will cache streamed content. Set to audio+video bitrate in bytes") + vlc_module_begin(); - set_description( _("ASF muxer") ); + set_description( N_("ASF muxer") ); + set_category( CAT_SOUT ); + set_subcategory( SUBCAT_SOUT_MUX ); + set_shortname( "ASF" ); + set_capability( "sout mux", 5 ); add_shortcut( "asf" ); add_shortcut( "asfh" ); set_callbacks( Open, Close ); add_string( SOUT_CFG_PREFIX "title", "", NULL, TITLE_TEXT, TITLE_LONGTEXT, - VLC_TRUE ); + true ); add_string( SOUT_CFG_PREFIX "author", "", NULL, AUTHOR_TEXT, - AUTHOR_LONGTEXT, VLC_TRUE ); + AUTHOR_LONGTEXT, true ); add_string( SOUT_CFG_PREFIX "copyright","", NULL, COPYRIGHT_TEXT, - COPYRIGHT_LONGTEXT, VLC_TRUE ); + COPYRIGHT_LONGTEXT, true ); add_string( SOUT_CFG_PREFIX "comment", "", NULL, COMMENT_TEXT, - COMMENT_LONGTEXT, VLC_TRUE ); + COMMENT_LONGTEXT, true ); add_string( SOUT_CFG_PREFIX "rating", "", NULL, RATING_TEXT, - RATING_LONGTEXT, VLC_TRUE ); + RATING_LONGTEXT, true ); + add_integer( SOUT_CFG_PREFIX "packet-size", 4096, NULL, PACKETSIZE_TEXT, + PACKETSIZE_LONGTEXT, true ); + add_integer( SOUT_CFG_PREFIX "bitrate-override", 0, NULL, BITRATE_TEXT, + BITRATE_LONGTEXT, true ); + vlc_module_end(); /***************************************************************************** * Locales prototypes *****************************************************************************/ -static const char *ppsz_sout_options[] = { +static const char *const ppsz_sout_options[] = { "title", "author", "copyright", "comment", "rating", NULL }; @@ -87,15 +105,6 @@ static int AddStream( sout_mux_t *, sout_input_t * ); static int DelStream( sout_mux_t *, sout_input_t * ); static int Mux ( sout_mux_t * ); -typedef struct -{ - uint32_t v1; /* le */ - uint16_t v2; /* le */ - uint16_t v3; /* le */ - uint8_t v4[8]; - -} guid_t; - typedef struct { int i_id; @@ -104,12 +113,16 @@ typedef struct /* codec information */ uint16_t i_tag; /* for audio */ vlc_fourcc_t i_fourcc; /* for video */ - char *psz_name; /* codec name */ + const char *psz_name; /* codec name */ + int i_blockalign; /* for audio only */ + bool b_audio_correction; int i_sequence; int i_extra; - uint8_t *p_extra; + uint8_t *p_extra; + + es_format_t fmt; } asf_track_t; @@ -122,18 +135,19 @@ struct sout_mux_sys_t mtime_t i_dts_last; mtime_t i_preroll_time; int64_t i_bitrate; + int64_t i_bitrate_override; int i_track; - asf_track_t track[128]; + asf_track_t track[MAX_ASF_TRACKS]; - vlc_bool_t b_write_header; + bool b_write_header; block_t *pk; int i_pk_used; int i_pk_frame; mtime_t i_pk_dts; - vlc_bool_t b_asf_http; + bool b_asf_http; int i_seq; /* meta data */ @@ -146,15 +160,17 @@ struct sout_mux_sys_t static int MuxGetStream( sout_mux_t *, int *pi_stream, mtime_t *pi_dts ); -static block_t *asf_header_create( sout_mux_t *, vlc_bool_t b_broadcast ); +static block_t *asf_header_create( sout_mux_t *, bool ); static block_t *asf_packet_create( sout_mux_t *, asf_track_t *, block_t * ); static block_t *asf_stream_end_create( sout_mux_t *); +static block_t *asf_packet_flush( sout_mux_t * ); typedef struct { int i_buffer_size; int i_buffer; uint8_t *p_buffer; + } bo_t; static void bo_init ( bo_t *, uint8_t *, int ); @@ -164,7 +180,7 @@ static void bo_addle_u32( bo_t *, uint32_t ); static void bo_addle_u64( bo_t *, uint64_t ); static void bo_add_mem ( bo_t *, uint8_t *, int ); -static void bo_addle_str16( bo_t *, char * ); +static void bo_addle_str16( bo_t *, const char * ); /***************************************************************************** * Open: @@ -176,8 +192,8 @@ static int Open( vlc_object_t *p_this ) vlc_value_t val; int i; - msg_Dbg( p_mux, "Asf muxer opened" ); - sout_CfgParse( p_mux, SOUT_CFG_PREFIX, ppsz_sout_options, p_mux->p_cfg ); + msg_Dbg( p_mux, "asf muxer opened" ); + config_ChainParse( p_mux, SOUT_CFG_PREFIX, ppsz_sout_options, p_mux->p_cfg ); p_mux->pf_control = Control; p_mux->pf_addstream = AddStream; @@ -197,21 +213,26 @@ static int Open( vlc_object_t *p_this ) p_sys->i_dts_last = 0; p_sys->i_preroll_time = 2000; p_sys->i_bitrate = 0; + p_sys->i_bitrate_override = 0; p_sys->i_seq = 0; - p_sys->b_write_header = VLC_TRUE; - p_sys->i_track = 1; - p_sys->i_packet_size = 4096; + p_sys->b_write_header = true; + p_sys->i_track = 0; + p_sys->i_packet_size = config_GetInt( p_mux, "sout-asf-packet-size" ); + p_sys->i_bitrate_override = config_GetInt( p_mux, "sout-asf-bitrate-override" ); + msg_Dbg( p_mux, "Packet size %d", p_sys->i_packet_size); + if (p_sys->i_bitrate_override) + msg_Dbg( p_mux, "Bitrate override %"PRId64, p_sys->i_bitrate_override); p_sys->i_packet_count= 0; /* Generate a random fid */ srand( mdate() & 0xffffffff ); - p_sys->fid.v1 = 0xbabac001; - p_sys->fid.v2 = ( (uint64_t)rand() << 16 ) / RAND_MAX; - p_sys->fid.v3 = ( (uint64_t)rand() << 16 ) / RAND_MAX; + p_sys->fid.Data1 = 0xbabac001; + p_sys->fid.Data2 = ( (uint64_t)rand() << 16 ) / RAND_MAX; + p_sys->fid.Data3 = ( (uint64_t)rand() << 16 ) / RAND_MAX; for( i = 0; i < 8; i++ ) { - p_sys->fid.v4[i] = ( (uint64_t)rand() << 8 ) / RAND_MAX; + p_sys->fid.Data4[i] = ( (uint64_t)rand() << 8 ) / RAND_MAX; } /* Meta data */ @@ -230,8 +251,8 @@ static int Open( vlc_object_t *p_this ) var_Get( p_mux, SOUT_CFG_PREFIX "rating", &val ); p_sys->psz_rating = val.psz_string; - msg_Dbg( p_mux, "meta data: title='%s' author='%s' copyright='%s' " - "comment='%s' rating='%s'", + msg_Dbg( p_mux, "meta data: title='%s', author='%s', copyright='%s', " + "comment='%s', rating='%s'", p_sys->psz_title, p_sys->psz_author, p_sys->psz_copyright, p_sys->psz_comment, p_sys->psz_rating ); @@ -250,21 +271,28 @@ static void Close( vlc_object_t * p_this ) msg_Dbg( p_mux, "Asf muxer closed" ); + /* Flush last packet if any */ + if( (out = asf_packet_flush( p_mux ) ) ) + { + sout_AccessOutWrite( p_mux->p_access, out ); + } + if( ( out = asf_stream_end_create( p_mux ) ) ) { sout_AccessOutWrite( p_mux->p_access, out ); } /* rewrite header */ - if( !sout_AccessOutSeek( p_mux->p_access, 0 ) ) + if( sout_AccessOutSeek( p_mux->p_access, 0 ) == VLC_SUCCESS ) { - out = asf_header_create( p_mux, VLC_FALSE ); + out = asf_header_create( p_mux, false ); sout_AccessOutWrite( p_mux->p_access, out ); } - for( i = 1; i < p_sys->i_track; i++ ) + for( i = 0; i < p_sys->i_track; i++ ) { free( p_sys->track[i].p_extra ); + es_format_Clean( &p_sys->track[i].fmt ); } free( p_sys ); } @@ -275,20 +303,20 @@ static void Close( vlc_object_t * p_this ) static int Control( sout_mux_t *p_mux, int i_query, va_list args ) { sout_mux_sys_t *p_sys = p_mux->p_sys; - vlc_bool_t *pb_bool; + bool *pb_bool; char **ppsz; switch( i_query ) { case MUX_CAN_ADD_STREAM_WHILE_MUXING: - pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * ); - if( p_sys->b_asf_http ) *pb_bool = VLC_TRUE; - else *pb_bool = VLC_FALSE; + pb_bool = (bool*)va_arg( args, bool * ); + if( p_sys->b_asf_http ) *pb_bool = true; + else *pb_bool = false; return VLC_SUCCESS; case MUX_GET_ADD_STREAM_WAIT: - pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * ); - *pb_bool = VLC_FALSE; + pb_bool = (bool*)va_arg( args, bool * ); + *pb_bool = true; return VLC_SUCCESS; case MUX_GET_MIME: @@ -314,23 +342,24 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) bo_t bo; msg_Dbg( p_mux, "adding input" ); - if( p_sys->i_track > 127 ) + if( p_sys->i_track >= MAX_ASF_TRACKS ) { - msg_Dbg( p_mux, "cannot add this track (too much track)" ); + msg_Dbg( p_mux, "cannot add this track (too much tracks)" ); return VLC_EGENERIC; } tk = p_input->p_sys = &p_sys->track[p_sys->i_track]; - tk->i_id = p_sys->i_track; + tk->i_id = p_sys->i_track + 1; tk->i_cat = p_input->p_fmt->i_cat; tk->i_sequence = 0; + tk->b_audio_correction = 0; switch( tk->i_cat ) { case AUDIO_ES: { int i_blockalign = p_input->p_fmt->audio.i_blockalign; - int i_bitspersample = 0; + int i_bitspersample = p_input->p_fmt->audio.i_bitspersample; int i_extra = 0; switch( p_input->p_fmt->i_codec ) @@ -338,32 +367,49 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) case VLC_FOURCC( 'a', '5', '2', ' ' ): tk->i_tag = WAVE_FORMAT_A52; tk->psz_name = "A/52"; + i_bitspersample = 0; + break; + case VLC_FOURCC( 'm', 'p', '4', 'a' ): + tk->i_tag = WAVE_FORMAT_AAC; + tk->psz_name = "MPEG-4 Audio"; + i_bitspersample = 0; break; case VLC_FOURCC( 'm', 'p', 'g', 'a' ): #if 1 tk->psz_name = "MPEG Audio Layer 3"; tk->i_tag = WAVE_FORMAT_MPEGLAYER3; + i_bitspersample = 0; i_blockalign = 1; i_extra = 12; break; #else tk->psz_name = "MPEG Audio Layer 1/2"; tk->i_tag = WAVE_FORMAT_MPEG; + i_bitspersample = 0; i_blockalign = 1; i_extra = 22; break; #endif case VLC_FOURCC( 'w', 'm', 'a', '1' ): - tk->psz_name = "Windows Media Audio 1"; + tk->psz_name = "Windows Media Audio v1"; tk->i_tag = WAVE_FORMAT_WMA1; + tk->b_audio_correction = true; break; + case VLC_FOURCC( 'w', 'm', 'a', ' ' ): case VLC_FOURCC( 'w', 'm', 'a', '2' ): - tk->psz_name = "Windows Media Audio 2"; + tk->psz_name= "Windows Media Audio (v2) 7, 8 and 9 Series"; tk->i_tag = WAVE_FORMAT_WMA2; + tk->b_audio_correction = true; break; - case VLC_FOURCC( 'w', 'm', 'a', '3' ): - tk->psz_name = "Windows Media Audio 3"; - tk->i_tag = WAVE_FORMAT_WMA3; + case VLC_FOURCC( 'w', 'm', 'a', 'p' ): + tk->psz_name = "Windows Media Audio 9 Professional"; + tk->i_tag = WAVE_FORMAT_WMAP; + tk->b_audio_correction = true; + break; + case VLC_FOURCC( 'w', 'm', 'a', 'l' ): + tk->psz_name = "Windows Media Audio 9 Lossless"; + tk->i_tag = WAVE_FORMAT_WMAL; + tk->b_audio_correction = true; break; /* raw codec */ case VLC_FOURCC( 'u', '8', ' ', ' ' ): @@ -394,7 +440,6 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) return VLC_EGENERIC; } - tk->i_extra = sizeof( WAVEFORMATEX ) + p_input->p_fmt->i_extra + i_extra; tk->p_extra = malloc( tk->i_extra ); @@ -404,6 +449,7 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) bo_addle_u32( &bo, p_input->p_fmt->audio.i_rate ); bo_addle_u32( &bo, p_input->p_fmt->i_bitrate / 8 ); bo_addle_u16( &bo, i_blockalign ); + tk->i_blockalign = i_blockalign; bo_addle_u16( &bo, i_bitspersample ); if( p_input->p_fmt->i_extra > 0 ) { @@ -445,6 +491,8 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) { p_sys->i_bitrate += 512000; } + if (p_sys->i_bitrate_override) + p_sys->i_bitrate = p_sys->i_bitrate_override; break; } case VIDEO_ES: @@ -486,19 +534,24 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) } else if( p_input->p_fmt->i_codec == VLC_FOURCC('W','M','V','1') ) { - tk->psz_name = "Windows Media Video 1"; + tk->psz_name = "Windows Media Video 7"; tk->i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '1' ); } else if( p_input->p_fmt->i_codec == VLC_FOURCC('W','M','V','2') ) { - tk->psz_name = "Windows Media Video 2"; + tk->psz_name = "Windows Media Video 8"; tk->i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '2' ); } else if( p_input->p_fmt->i_codec == VLC_FOURCC('W','M','V','3') ) { - tk->psz_name = "Windows Media Video 3"; + tk->psz_name = "Windows Media Video 9"; tk->i_fourcc = VLC_FOURCC( 'W', 'M', 'V', '3' ); } + else if( p_input->p_fmt->i_codec == VLC_FOURCC('h','2','6','4') ) + { + tk->psz_name = "H.264/MPEG-4 AVC"; + tk->i_fourcc = VLC_FOURCC('h','2','6','4'); + } else { tk->psz_name = _("Unknown Video"); @@ -524,6 +577,8 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) { p_sys->i_bitrate += 1000000; } + if (p_sys->i_bitrate_override) + p_sys->i_bitrate = p_sys->i_bitrate_override; break; } default: @@ -531,6 +586,8 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) return VLC_EGENERIC; } + es_format_Copy( &tk->fmt, p_input->p_fmt ); + p_sys->i_track++; return VLC_SUCCESS; } @@ -540,6 +597,27 @@ static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input ) *****************************************************************************/ static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input ) { + /* if bitrate ain't defined in commanline, reduce it when tracks are deleted + */ + sout_mux_sys_t *p_sys = p_mux->p_sys; + asf_track_t *tk = p_input->p_sys; + if(!p_sys->i_bitrate_override) + { + if( tk->i_cat == AUDIO_ES ) + { + if( p_input->p_fmt->i_bitrate > 24000 ) + p_sys->i_bitrate -= p_input->p_fmt->i_bitrate; + else + p_sys->i_bitrate -= 512000; + } + else if(tk->i_cat == VIDEO_ES ) + { + if( p_input->p_fmt->i_bitrate > 50000 ) + p_sys->i_bitrate -= p_input->p_fmt->i_bitrate; + else + p_sys->i_bitrate -= 1000000; + } + } msg_Dbg( p_mux, "removing input" ); return VLC_SUCCESS; } @@ -553,12 +631,12 @@ static int Mux( sout_mux_t *p_mux ) if( p_sys->b_write_header ) { - block_t *out = asf_header_create( p_mux, VLC_TRUE ); + block_t *out = asf_header_create( p_mux, true ); out->i_flags |= BLOCK_FLAG_HEADER; sout_AccessOutWrite( p_mux->p_access, out ); - p_sys->b_write_header = VLC_FALSE; + p_sys->b_write_header = false; } for( ;; ) @@ -610,7 +688,7 @@ static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts ) sout_input_t *p_input = p_mux->pp_inputs[i]; block_t *p_data; - if( p_input->p_fifo->i_depth <= 0 ) + if( block_FifoCount( p_input->p_fifo ) <= 0 ) { if( p_input->p_fmt->i_cat == AUDIO_ES || p_input->p_fmt->i_cat == VIDEO_ES ) @@ -684,7 +762,7 @@ static void bo_add_mem( bo_t *p_bo, uint8_t *p_mem, int i_size ) p_bo->i_buffer += i_size; } -static void bo_addle_str16( bo_t *bo, char *str ) +static void bo_addle_str16( bo_t *bo, const char *str ) { bo_addle_u16( bo, strlen( str ) + 1 ); for( ;; ) @@ -695,7 +773,7 @@ static void bo_addle_str16( bo_t *bo, char *str ) } } -static void bo_addle_str16_nosize( bo_t *bo, char *str ) +static void bo_addle_str16_nosize( bo_t *bo, const char *str ) { for( ;; ) { @@ -706,116 +784,56 @@ static void bo_addle_str16_nosize( bo_t *bo, char *str ) } /**************************************************************************** - * guid + * GUID definitions ****************************************************************************/ static void bo_add_guid( bo_t *p_bo, const guid_t *id ) { int i; - bo_addle_u32( p_bo, id->v1 ); - bo_addle_u16( p_bo, id->v2 ); - bo_addle_u16( p_bo, id->v3 ); + bo_addle_u32( p_bo, id->Data1 ); + bo_addle_u16( p_bo, id->Data2 ); + bo_addle_u16( p_bo, id->Data3 ); for( i = 0; i < 8; i++ ) { - bo_add_u8( p_bo, id->v4[i] ); + bo_add_u8( p_bo, id->Data4[i] ); } } static const guid_t asf_object_header_guid = -{ - 0x75B22630, - 0x668E, - 0x11CF, - { 0xA6,0xD9, 0x00,0xAA,0x00,0x62,0xCE,0x6C } -}; +{0x75B22630, 0x668E, 0x11CF, {0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C}}; static const guid_t asf_object_data_guid = -{ - 0x75B22636, - 0x668E, - 0x11CF, - { 0xA6,0xD9, 0x00,0xAA,0x00,0x62,0xCE,0x6C } -}; +{0x75B22636, 0x668E, 0x11CF, {0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C}}; static const guid_t asf_object_file_properties_guid = -{ - 0x8cabdca1, - 0xa947, - 0x11cf, - { 0x8e,0xe4, 0x00,0xC0,0x0C,0x20,0x53,0x65 } - -}; +{0x8cabdca1, 0xa947, 0x11cf, {0x8e, 0xe4, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65}}; static const guid_t asf_object_stream_properties_guid = -{ - 0xB7DC0791, - 0xA9B7, - 0x11CF, - { 0x8E,0xE6, 0x00,0xC0,0x0C,0x20,0x53,0x65 } - -}; -static const guid_t asf_object_header_extention_guid = -{ - 0x5FBF03B5, - 0xA92E, - 0x11CF, - { 0x8E,0xE3, 0x00,0xC0,0x0C,0x20,0x53,0x65 } -}; +{0xB7DC0791, 0xA9B7, 0x11CF, {0x8E, 0xE6, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65}}; +static const guid_t asf_object_header_extension_guid = +{0x5FBF03B5, 0xA92E, 0x11CF, {0x8E, 0xE3, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65}}; static const guid_t asf_object_stream_type_audio = -{ - 0xF8699E40, - 0x5B4D, - 0x11CF, - { 0xA8,0xFD, 0x00,0x80,0x5F,0x5C,0x44,0x2B } -}; - +{0xF8699E40, 0x5B4D, 0x11CF, {0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B}}; static const guid_t asf_object_stream_type_video = -{ - 0xbc19efc0, - 0x5B4D, - 0x11CF, - { 0xA8,0xFD, 0x00,0x80,0x5F,0x5C,0x44,0x2B } -}; - +{0xbc19efc0, 0x5B4D, 0x11CF, {0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B}}; static const guid_t asf_guid_audio_conceal_none = -{ - 0x20FB5700, - 0x5B55, - 0x11CF, - { 0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B } -}; +{0x20FB5700, 0x5B55, 0x11CF, {0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B}}; +static const guid_t asf_guid_audio_conceal_spread = +{0xBFC3CD50, 0x618F, 0x11CF, {0x8B, 0xB2, 0x00, 0xAA, 0x00, 0xB4, 0xE2, 0x20}}; static const guid_t asf_guid_video_conceal_none = -{ - 0x20FB5700, - 0x5B55, - 0x11CF, - { 0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B } -}; +{0x20FB5700, 0x5B55, 0x11CF, {0xA8, 0xFD, 0x00, 0x80, 0x5F, 0x5C, 0x44, 0x2B}}; static const guid_t asf_guid_reserved_1 = -{ - 0xABD3D211, - 0xA9BA, - 0x11cf, - { 0x8E, 0xE6,0x00, 0xC0, 0x0C ,0x20, 0x53, 0x65 } -}; -static const guid_t asf_object_codec_comment_guid = -{ - 0x86D15240, - 0x311D, - 0x11D0, - { 0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 } -}; -static const guid_t asf_object_codec_comment_reserved_guid = -{ - 0x86D15241, - 0x311D, - 0x11D0, - { 0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 } -}; +{0xABD3D211, 0xA9BA, 0x11cf, {0x8E, 0xE6, 0x00, 0xC0, 0x0C ,0x20, 0x53, 0x65}}; +static const guid_t asf_object_codec_list_guid = +{0x86D15240, 0x311D, 0x11D0, {0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6}}; +static const guid_t asf_object_codec_list_reserved_guid = +{0x86D15241, 0x311D, 0x11D0, {0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6}}; static const guid_t asf_object_content_description_guid = -{ - 0x75B22633, - 0x668E, - 0x11CF, - { 0xa6, 0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c } -}; +{0x75B22633, 0x668E, 0x11CF, {0xa6, 0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c}}; +static const guid_t asf_object_index_guid = +{0x33000890, 0xE5B1, 0x11CF, {0x89, 0xF4, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xCB}}; +static const guid_t asf_object_metadata_guid = +{0xC5F8CBEA, 0x5BAF, 0x4877, {0x84, 0x67, 0xAA, 0x8C, 0x44, 0xFA, 0x4C, 0xCA}}; +/**************************************************************************** + * Misc + ****************************************************************************/ static void asf_chunk_add( bo_t *bo, int i_type, int i_len, int i_flags, int i_seq ) { @@ -826,18 +844,15 @@ static void asf_chunk_add( bo_t *bo, bo_addle_u16( bo, i_len + 8 ); } -static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) +static block_t *asf_header_create( sout_mux_t *p_mux, bool b_broadcast ) { sout_mux_sys_t *p_sys = p_mux->p_sys; asf_track_t *tk; - mtime_t i_duration = 0; - int i_size; - int i_ci_size; - int i_cd_size = 0; + int i_size, i_header_ext_size, i; + int i_ci_size, i_cm_size = 0, i_cd_size = 0; block_t *out; bo_t bo; - int i; msg_Dbg( p_mux, "Asf muxer creating header" ); @@ -848,22 +863,20 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) } /* calculate header size */ - i_size = 30 + 104 + 46; + i_size = 30 + 104; i_ci_size = 44; - for( i = 1; i < p_sys->i_track; i++ ) + for( i = 0; i < p_sys->i_track; i++ ) { i_size += 78 + p_sys->track[i].i_extra; i_ci_size += 8 + 2 * strlen( p_sys->track[i].psz_name ); - if( p_sys->track[i].i_cat == AUDIO_ES ) - { - i_ci_size += 4; - } - else if( p_sys->track[i].i_cat == VIDEO_ES ) - { - i_ci_size += 6; - } + if( p_sys->track[i].i_cat == AUDIO_ES ) i_ci_size += 4; + else if( p_sys->track[i].i_cat == VIDEO_ES ) i_ci_size += 6; + + /* Error correction data field */ + if( p_sys->track[i].b_audio_correction ) i_size += 8; } + /* size of the content description object */ if( *p_sys->psz_title || *p_sys->psz_author || *p_sys->psz_copyright || *p_sys->psz_comment || *p_sys->psz_rating ) { @@ -874,7 +887,18 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) strlen( p_sys->psz_rating ) + 1 ); } - i_size += i_ci_size + i_cd_size; + /* size of the metadata object */ + for( i = 0; i < p_sys->i_track; i++ ) + { + if( p_sys->track[i].i_cat == VIDEO_ES ) + { + i_cm_size = 26 + 2 * (16 + 2 * sizeof("AspectRatio?")); + break; + } + } + + i_header_ext_size = i_cm_size ? i_cm_size + 46 : 0; + i_size += i_ci_size + i_cd_size + i_header_ext_size ; if( p_sys->b_asf_http ) { @@ -891,7 +915,8 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) /* header object */ bo_add_guid ( &bo, &asf_object_header_guid ); bo_addle_u64( &bo, i_size ); - bo_addle_u32( &bo, 2 + p_sys->i_track - 1 ); + bo_addle_u32( &bo, 2 + p_sys->i_track + + (i_cd_size ? 1 : 0) + (i_cm_size ? 1 : 0) ); bo_add_u8 ( &bo, 1 ); bo_add_u8 ( &bo, 2 ); @@ -908,17 +933,57 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) bo_addle_u64( &bo, i_duration * 10 ); /* play duration (100ns) */ bo_addle_u64( &bo, i_duration * 10 ); /* send duration (100ns) */ bo_addle_u64( &bo, p_sys->i_preroll_time ); /* preroll duration (ms) */ - bo_addle_u32( &bo, b_broadcast ? 0x01 : 0x00); /* flags */ + bo_addle_u32( &bo, b_broadcast ? 0x01 : 0x02 /* seekable */ ); /* flags */ bo_addle_u32( &bo, p_sys->i_packet_size ); /* packet size min */ bo_addle_u32( &bo, p_sys->i_packet_size ); /* packet size max */ bo_addle_u32( &bo, p_sys->i_bitrate ); /* maxbitrate */ - /* header extention */ - bo_add_guid ( &bo, &asf_object_header_extention_guid ); - bo_addle_u64( &bo, 46 ); - bo_add_guid ( &bo, &asf_guid_reserved_1 ); - bo_addle_u16( &bo, 6 ); - bo_addle_u32( &bo, 0 ); + /* header extension */ + if( i_header_ext_size ) + { + bo_add_guid ( &bo, &asf_object_header_extension_guid ); + bo_addle_u64( &bo, i_header_ext_size ); + bo_add_guid ( &bo, &asf_guid_reserved_1 ); + bo_addle_u16( &bo, 6 ); + bo_addle_u32( &bo, i_header_ext_size - 46 ); + } + + /* metadata object (part of header extension) */ + if( i_cm_size ) + { + int64_t i_num, i_den; + unsigned int i_dst_num, i_dst_den; + + for( i = 0; i < p_sys->i_track; i++ ) + if( p_sys->track[i].i_cat == VIDEO_ES ) break; + + i_num = p_sys->track[i].fmt.video.i_aspect * + (int64_t)p_sys->track[i].fmt.video.i_height; + i_den = VOUT_ASPECT_FACTOR * p_sys->track[i].fmt.video.i_width; + vlc_ureduce( &i_dst_num, &i_dst_den, i_num, i_den, 0 ); + + msg_Dbg( p_mux, "pixel aspect-ratio: %i/%i", i_dst_num, i_dst_den ); + + bo_add_guid ( &bo, &asf_object_metadata_guid ); + bo_addle_u64( &bo, i_cm_size ); + bo_addle_u16( &bo, 2 ); /* description records count */ + /* 1st description record */ + bo_addle_u16( &bo, 0 ); /* reserved */ + bo_addle_u16( &bo, i + 1 ); /* stream number (0 for the whole file) */ + bo_addle_u16( &bo, 2 * sizeof("AspectRatioX") ); /* name length */ + bo_addle_u16( &bo, 0x3 /* DWORD */ ); /* data type */ + bo_addle_u32( &bo, 4 ); /* data length */ + bo_addle_str16_nosize( &bo, "AspectRatioX" ); + bo_addle_u32( &bo, i_dst_num ); /* data */ + /* 2nd description record */ + bo_addle_u16( &bo, 0 ); /* reserved */ + bo_addle_u16( &bo, i + 1 ); /* stream number (0 for the whole file) */ + bo_addle_u16( &bo, 2 * sizeof("AspectRatioY") ); /* name length */ + bo_addle_u16( &bo, 0x3 /* DWORD */ ); /* data type */ + bo_addle_u32( &bo, 4 ); /* data length */ + bo_addle_str16_nosize( &bo, "AspectRatioY" ); + bo_addle_u32( &bo, i_dst_den ); /* data */ + } /* content description header */ if( i_cd_size > 0 ) @@ -939,16 +1004,20 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) } /* stream properties */ - for( i = 1; i < p_sys->i_track; i++ ) + for( i = 0; i < p_sys->i_track; i++ ) { tk = &p_sys->track[i]; bo_add_guid ( &bo, &asf_object_stream_properties_guid ); - bo_addle_u64( &bo, 78 + tk->i_extra ); + bo_addle_u64( &bo, 78 + tk->i_extra + (tk->b_audio_correction ? 8:0) ); + if( tk->i_cat == AUDIO_ES ) { bo_add_guid( &bo, &asf_object_stream_type_audio ); - bo_add_guid( &bo, &asf_guid_audio_conceal_none ); + if( tk->b_audio_correction ) + bo_add_guid( &bo, &asf_guid_audio_conceal_spread ); + else + bo_add_guid( &bo, &asf_guid_audio_conceal_none ); } else if( tk->i_cat == VIDEO_ES ) { @@ -957,22 +1026,36 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) } bo_addle_u64( &bo, 0 ); /* time offset */ bo_addle_u32( &bo, tk->i_extra ); - bo_addle_u32( &bo, 0 ); /* 0 */ + /* correction data length */ + bo_addle_u32( &bo, tk->b_audio_correction ? 8 : 0 ); bo_addle_u16( &bo, tk->i_id ); /* stream number */ bo_addle_u32( &bo, 0 ); bo_add_mem ( &bo, tk->p_extra, tk->i_extra ); + + /* Error correction data field */ + if( tk->b_audio_correction ) + { + bo_add_u8( &bo, 0x1 ); /* span */ + bo_addle_u16( &bo, tk->i_blockalign ); /* virtual packet length */ + bo_addle_u16( &bo, tk->i_blockalign ); /* virtual chunck length */ + bo_addle_u16( &bo, 1 ); /* silence length */ + bo_add_u8( &bo, 0x0 ); /* data */ + } } /* Codec Infos */ - bo_add_guid ( &bo, &asf_object_codec_comment_guid ); + bo_add_guid ( &bo, &asf_object_codec_list_guid ); bo_addle_u64( &bo, i_ci_size ); - bo_add_guid ( &bo, &asf_object_codec_comment_reserved_guid ); - bo_addle_u32( &bo, p_sys->i_track - 1 ); - for( i = 1; i < p_sys->i_track; i++ ) + bo_add_guid ( &bo, &asf_object_codec_list_reserved_guid ); + bo_addle_u32( &bo, p_sys->i_track ); + for( i = 0; i < p_sys->i_track; i++ ) { tk = &p_sys->track[i]; - bo_addle_u16( &bo, tk->i_id ); + if( tk->i_cat == VIDEO_ES ) bo_addle_u16( &bo, 1 /* video */ ); + else if( tk->i_cat == AUDIO_ES ) bo_addle_u16( &bo, 2 /* audio */ ); + else bo_addle_u16( &bo, 0xFFFF /* unknown */ ); + bo_addle_str16( &bo, tk->psz_name ); bo_addle_u16( &bo, 0 ); if( tk->i_cat == AUDIO_ES ) @@ -984,7 +1067,6 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) { bo_addle_u16( &bo, 4 ); bo_add_mem ( &bo, (uint8_t*)&tk->i_fourcc, 4 ); - } } @@ -1001,6 +1083,41 @@ static block_t *asf_header_create( sout_mux_t *p_mux, vlc_bool_t b_broadcast ) /**************************************************************************** * ****************************************************************************/ +static block_t *asf_packet_flush( sout_mux_t *p_mux ) +{ + sout_mux_sys_t *p_sys = p_mux->p_sys; + int i_pad, i_preheader = p_sys->b_asf_http ? 12 : 0; + block_t *pk; + bo_t bo; + + if( !p_sys->pk ) return 0; + + i_pad = p_sys->i_packet_size - p_sys->i_pk_used; + memset( p_sys->pk->p_buffer + p_sys->i_pk_used, 0, i_pad ); + + bo_init( &bo, p_sys->pk->p_buffer, 14 + i_preheader ); + + if( p_sys->b_asf_http ) + asf_chunk_add( &bo, 0x4424, p_sys->i_packet_size, 0x0, p_sys->i_seq++); + + bo_add_u8 ( &bo, 0x82 ); + bo_addle_u16( &bo, 0 ); + bo_add_u8( &bo, 0x11 ); + bo_add_u8( &bo, 0x5d ); + bo_addle_u16( &bo, i_pad ); + bo_addle_u32( &bo, (p_sys->i_pk_dts - p_sys->i_dts_first) / 1000 + + p_sys->i_preroll_time ); + bo_addle_u16( &bo, 0 /* data->i_length */ ); + bo_add_u8( &bo, 0x80 | p_sys->i_pk_frame ); + + pk = p_sys->pk; + p_sys->pk = NULL; + + p_sys->i_packet_count++; + + return pk; +} + static block_t *asf_packet_create( sout_mux_t *p_mux, asf_track_t *tk, block_t *data ) { @@ -1032,12 +1149,23 @@ static block_t *asf_packet_create( sout_mux_t *p_mux, /* add payload (header size = 17) */ i_payload = __MIN( i_data - i_pos, p_sys->i_packet_size - p_sys->i_pk_used - 17 ); - bo_add_u8 ( &bo, 0x80 | tk->i_id ); + + if( tk->b_audio_correction && p_sys->i_pk_frame && i_payload < i_data ) + { + /* Don't know why yet but WMP doesn't like splitted WMA packets */ + *last = asf_packet_flush( p_mux ); + last = &(*last)->p_next; + continue; + } + + bo_add_u8 ( &bo, !(data->i_flags & BLOCK_FLAG_TYPE_P || + data->i_flags & BLOCK_FLAG_TYPE_B) ? + 0x80 | tk->i_id : tk->i_id ); bo_add_u8 ( &bo, tk->i_sequence ); bo_addle_u32( &bo, i_pos ); bo_add_u8 ( &bo, 0x08 ); /* flags */ bo_addle_u32( &bo, i_data ); - bo_addle_u32( &bo, (data->i_dts - p_sys->i_dts_first) / 1000 - + bo_addle_u32( &bo, (data->i_dts - p_sys->i_dts_first) / 1000 + p_sys->i_preroll_time ); bo_addle_u16( &bo, i_payload ); bo_add_mem ( &bo, &p_data[i_pos], i_payload ); @@ -1048,33 +1176,9 @@ static block_t *asf_packet_create( sout_mux_t *p_mux, if( p_sys->i_pk_used + 17 >= p_sys->i_packet_size ) { - /* not enough data for another payload, flush the packet */ - int i_pad = p_sys->i_packet_size - p_sys->i_pk_used; - - bo_init( &bo, p_sys->pk->p_buffer, 14 + i_preheader ); - - if( p_sys->b_asf_http ) - { - asf_chunk_add( &bo, 0x4424, p_sys->i_packet_size, - 0x00, p_sys->i_seq++); - } - bo_add_u8 ( &bo, 0x82 ); - bo_addle_u16( &bo, 0 ); - bo_add_u8( &bo, 0x11 ); - bo_add_u8( &bo, 0x5d ); - bo_addle_u16( &bo, i_pad ); - bo_addle_u32( &bo, (p_sys->i_pk_dts - p_sys->i_dts_first) / 1000 - - p_sys->i_preroll_time ); - bo_addle_u16( &bo, 0 * data->i_length / 1000 ); - bo_add_u8( &bo, 0x80 | p_sys->i_pk_frame ); - - /* append the packet */ - *last = p_sys->pk; - last = &p_sys->pk->p_next; - - p_sys->pk = NULL; - - p_sys->i_packet_count++; + /* Not enough data for another payload, flush the packet */ + *last = asf_packet_flush( p_mux ); + last = &(*last)->p_next; } } @@ -1097,6 +1201,18 @@ static block_t *asf_stream_end_create( sout_mux_t *p_mux ) bo_init( &bo, out->p_buffer, 12 ); asf_chunk_add( &bo, 0x4524, 0, 0x00, p_sys->i_seq++ ); } + else + { + /* Create index */ + out = block_New( p_mux, 56 ); + bo_init( &bo, out->p_buffer, 56 ); + bo_add_guid ( &bo, &asf_object_index_guid ); + bo_addle_u64( &bo, 56 ); + bo_add_guid ( &bo, &p_sys->fid ); + bo_addle_u64( &bo, 10000000 ); + bo_addle_u32( &bo, 5 ); + bo_addle_u32( &bo, 0 ); + } return out; }