2014-07-14 19:52:23 +03:00
/*
* WebM DASH Manifest XML muxer
* Copyright ( c ) 2014 Vignesh Venkatasubramanian
*
* This file is part of FFmpeg .
*
* FFmpeg is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
* version 2.1 of the License , or ( at your option ) any later version .
*
* FFmpeg is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg ; if not , write to the Free Software
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
/*
* WebM DASH Specification :
* https : //sites.google.com/a/webmproject.org/wiki/adaptive-streaming/webm-dash-specification
*/
# include <stdint.h>
# include <string.h>
# include "avformat.h"
# include "avio_internal.h"
# include "matroska.h"
# include "libavutil/avstring.h"
# include "libavutil/dict.h"
# include "libavutil/opt.h"
typedef struct AdaptationSet {
char id [ 10 ] ;
int * streams ;
int nb_streams ;
} AdaptationSet ;
typedef struct WebMDashMuxContext {
const AVClass * class ;
char * adaptation_sets ;
AdaptationSet * as ;
int nb_as ;
2014-11-11 20:02:05 +02:00
int representation_id ;
2014-07-14 19:52:23 +03:00
} WebMDashMuxContext ;
static const char * get_codec_name ( int codec_id )
{
switch ( codec_id ) {
case AV_CODEC_ID_VP8 :
return " vp8 " ;
case AV_CODEC_ID_VP9 :
return " vp9 " ;
case AV_CODEC_ID_VORBIS :
return " vorbis " ;
case AV_CODEC_ID_OPUS :
return " opus " ;
}
return NULL ;
}
static double get_duration ( AVFormatContext * s )
{
int i = 0 ;
double max = 0.0 ;
for ( i = 0 ; i < s - > nb_streams ; i + + ) {
AVDictionaryEntry * duration = av_dict_get ( s - > streams [ i ] - > metadata ,
DURATION , NULL , 0 ) ;
2014-08-15 21:33:21 +03:00
if ( ! duration | | atof ( duration - > value ) < 0 ) continue ;
2014-07-14 19:52:23 +03:00
if ( atof ( duration - > value ) > max ) max = atof ( duration - > value ) ;
}
return max / 1000 ;
}
static void write_header ( AVFormatContext * s )
{
double min_buffer_time = 1.0 ;
avio_printf ( s - > pb , " <?xml version= \" 1.0 \" encoding= \" UTF-8 \" ?> \n " ) ;
avio_printf ( s - > pb , " <MPD \n " ) ;
avio_printf ( s - > pb , " xmlns:xsi= \" http://www.w3.org/2001/XMLSchema-instance \" \n " ) ;
avio_printf ( s - > pb , " xmlns= \" urn:mpeg:DASH:schema:MPD:2011 \" \n " ) ;
avio_printf ( s - > pb , " xsi:schemaLocation= \" urn:mpeg:DASH:schema:MPD:2011 \" \n " ) ;
avio_printf ( s - > pb , " type= \" static \" \n " ) ;
avio_printf ( s - > pb , " mediaPresentationDuration= \" PT%gS \" \n " ,
get_duration ( s ) ) ;
avio_printf ( s - > pb , " minBufferTime= \" PT%gS \" \n " ,
min_buffer_time ) ;
avio_printf ( s - > pb , " profiles= \" urn:webm:dash:profile:webm-on-demand:2012 \" " ) ;
avio_printf ( s - > pb , " > \n " ) ;
}
static void write_footer ( AVFormatContext * s )
{
2014-09-16 18:24:56 +03:00
avio_printf ( s - > pb , " </MPD> \n " ) ;
2014-07-14 19:52:23 +03:00
}
static int subsegment_alignment ( AVFormatContext * s , AdaptationSet * as ) {
int i ;
AVDictionaryEntry * gold = av_dict_get ( s - > streams [ as - > streams [ 0 ] ] - > metadata ,
CUE_TIMESTAMPS , NULL , 0 ) ;
2014-08-15 21:33:21 +03:00
if ( ! gold ) return 0 ;
2014-07-14 19:52:23 +03:00
for ( i = 1 ; i < as - > nb_streams ; i + + ) {
AVDictionaryEntry * ts = av_dict_get ( s - > streams [ as - > streams [ i ] ] - > metadata ,
CUE_TIMESTAMPS , NULL , 0 ) ;
2014-08-15 21:33:21 +03:00
if ( ! ts | | strncmp ( gold - > value , ts - > value , strlen ( gold - > value ) ) ) return 0 ;
2014-07-14 19:52:23 +03:00
}
return 1 ;
}
static int bitstream_switching ( AVFormatContext * s , AdaptationSet * as ) {
int i ;
AVDictionaryEntry * gold_track_num = av_dict_get ( s - > streams [ as - > streams [ 0 ] ] - > metadata ,
TRACK_NUMBER , NULL , 0 ) ;
AVCodecContext * gold_codec = s - > streams [ as - > streams [ 0 ] ] - > codec ;
2014-08-15 21:33:21 +03:00
if ( ! gold_track_num ) return 0 ;
2014-07-14 19:52:23 +03:00
for ( i = 1 ; i < as - > nb_streams ; i + + ) {
AVDictionaryEntry * track_num = av_dict_get ( s - > streams [ as - > streams [ i ] ] - > metadata ,
TRACK_NUMBER , NULL , 0 ) ;
AVCodecContext * codec = s - > streams [ as - > streams [ i ] ] - > codec ;
2014-08-15 21:33:21 +03:00
if ( ! track_num | |
2014-07-14 19:52:23 +03:00
strncmp ( gold_track_num - > value , track_num - > value , strlen ( gold_track_num - > value ) ) | |
gold_codec - > codec_id ! = codec - > codec_id | |
gold_codec - > extradata_size ! = codec - > extradata_size | |
memcmp ( gold_codec - > extradata , codec - > extradata , codec - > extradata_size ) ) {
return 0 ;
}
}
return 1 ;
}
2014-10-01 20:13:32 +03:00
/*
* Writes a Representation within an Adaptation Set . Returns 0 on success and
* < 0 on failure .
*/
static int write_representation ( AVFormatContext * s , AVStream * stream , int id ,
int output_width , int output_height ,
int output_sample_rate ) {
AVDictionaryEntry * irange = av_dict_get ( stream - > metadata , INITIALIZATION_RANGE , NULL , 0 ) ;
AVDictionaryEntry * cues_start = av_dict_get ( stream - > metadata , CUES_START , NULL , 0 ) ;
AVDictionaryEntry * cues_end = av_dict_get ( stream - > metadata , CUES_END , NULL , 0 ) ;
AVDictionaryEntry * filename = av_dict_get ( stream - > metadata , FILENAME , NULL , 0 ) ;
AVDictionaryEntry * bandwidth = av_dict_get ( stream - > metadata , BANDWIDTH , NULL , 0 ) ;
if ( ! irange | | cues_start = = NULL | | cues_end = = NULL | | filename = = NULL | |
! bandwidth ) {
return - 1 ;
}
avio_printf ( s - > pb , " <Representation id= \" %d \" " , id ) ;
avio_printf ( s - > pb , " bandwidth= \" %s \" " , bandwidth - > value ) ;
if ( stream - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO & & output_width )
avio_printf ( s - > pb , " width= \" %d \" " , stream - > codec - > width ) ;
if ( stream - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO & & output_height )
avio_printf ( s - > pb , " height= \" %d \" " , stream - > codec - > height ) ;
if ( stream - > codec - > codec_type = AVMEDIA_TYPE_AUDIO & & output_sample_rate )
avio_printf ( s - > pb , " audioSamplingRate= \" %d \" " , stream - > codec - > sample_rate ) ;
avio_printf ( s - > pb , " > \n " ) ;
avio_printf ( s - > pb , " <BaseURL>%s</BaseURL> \n " , filename - > value ) ;
avio_printf ( s - > pb , " <SegmentBase \n " ) ;
avio_printf ( s - > pb , " indexRange= \" %s-%s \" > \n " , cues_start - > value , cues_end - > value ) ;
avio_printf ( s - > pb , " <Initialization \n " ) ;
avio_printf ( s - > pb , " range= \" 0-%s \" /> \n " , irange - > value ) ;
avio_printf ( s - > pb , " </SegmentBase> \n " ) ;
avio_printf ( s - > pb , " </Representation> \n " ) ;
return 0 ;
}
/*
* Checks if width of all streams are the same . Returns 1 if true , 0 otherwise .
*/
static int check_matching_width ( AVFormatContext * s , AdaptationSet * as ) {
int first_width , i ;
if ( as - > nb_streams < 2 ) return 1 ;
first_width = s - > streams [ as - > streams [ 0 ] ] - > codec - > width ;
for ( i = 1 ; i < as - > nb_streams ; i + + )
if ( first_width ! = s - > streams [ as - > streams [ i ] ] - > codec - > width )
return 0 ;
return 1 ;
}
/*
* Checks if height of all streams are the same . Returns 1 if true , 0 otherwise .
*/
static int check_matching_height ( AVFormatContext * s , AdaptationSet * as ) {
int first_height , i ;
if ( as - > nb_streams < 2 ) return 1 ;
first_height = s - > streams [ as - > streams [ 0 ] ] - > codec - > height ;
for ( i = 1 ; i < as - > nb_streams ; i + + )
if ( first_height ! = s - > streams [ as - > streams [ i ] ] - > codec - > height )
return 0 ;
return 1 ;
}
/*
* Checks if sample rate of all streams are the same . Returns 1 if true , 0 otherwise .
*/
static int check_matching_sample_rate ( AVFormatContext * s , AdaptationSet * as ) {
int first_sample_rate , i ;
if ( as - > nb_streams < 2 ) return 1 ;
first_sample_rate = s - > streams [ as - > streams [ 0 ] ] - > codec - > sample_rate ;
for ( i = 1 ; i < as - > nb_streams ; i + + )
if ( first_sample_rate ! = s - > streams [ as - > streams [ i ] ] - > codec - > sample_rate )
return 0 ;
return 1 ;
}
2014-07-14 19:52:23 +03:00
/*
* Writes an Adaptation Set . Returns 0 on success and < 0 on failure .
*/
static int write_adaptation_set ( AVFormatContext * s , int as_index )
{
WebMDashMuxContext * w = s - > priv_data ;
AdaptationSet * as = & w - > as [ as_index ] ;
AVCodecContext * codec = s - > streams [ as - > streams [ 0 ] ] - > codec ;
2014-10-01 20:13:32 +03:00
AVDictionaryEntry * lang ;
2014-07-14 19:52:23 +03:00
int i ;
static const char boolean [ 2 ] [ 6 ] = { " false " , " true " } ;
int subsegmentStartsWithSAP = 1 ;
2014-10-01 20:13:32 +03:00
// Width, Height and Sample Rate will go in the AdaptationSet tag if they
// are the same for all contained Representations. otherwise, they will go
// on their respective Representation tag.
int width_in_as = 1 , height_in_as = 1 , sample_rate_in_as = 1 ;
if ( codec - > codec_type = = AVMEDIA_TYPE_VIDEO ) {
width_in_as = check_matching_width ( s , as ) ;
height_in_as = check_matching_height ( s , as ) ;
} else {
sample_rate_in_as = check_matching_sample_rate ( s , as ) ;
}
2014-07-14 19:52:23 +03:00
avio_printf ( s - > pb , " <AdaptationSet id= \" %s \" " , as - > id ) ;
avio_printf ( s - > pb , " mimeType= \" %s/webm \" " ,
codec - > codec_type = = AVMEDIA_TYPE_VIDEO ? " video " : " audio " ) ;
avio_printf ( s - > pb , " codecs= \" %s \" " , get_codec_name ( codec - > codec_id ) ) ;
lang = av_dict_get ( s - > streams [ as - > streams [ 0 ] ] - > metadata , " language " , NULL , 0 ) ;
2014-08-15 22:31:59 +03:00
if ( lang ) avio_printf ( s - > pb , " lang= \" %s \" " , lang - > value ) ;
2014-07-14 19:52:23 +03:00
2014-10-01 20:13:32 +03:00
if ( codec - > codec_type = = AVMEDIA_TYPE_VIDEO & & width_in_as )
2014-07-14 19:52:23 +03:00
avio_printf ( s - > pb , " width= \" %d \" " , codec - > width ) ;
2014-10-01 20:13:32 +03:00
if ( codec - > codec_type = = AVMEDIA_TYPE_VIDEO & & height_in_as )
2014-07-14 19:52:23 +03:00
avio_printf ( s - > pb , " height= \" %d \" " , codec - > height ) ;
2014-10-01 20:13:32 +03:00
if ( codec - > codec_type = = AVMEDIA_TYPE_AUDIO & & sample_rate_in_as )
2014-07-14 19:52:23 +03:00
avio_printf ( s - > pb , " audioSamplingRate= \" %d \" " , codec - > sample_rate ) ;
avio_printf ( s - > pb , " bitstreamSwitching= \" %s \" " ,
boolean [ bitstream_switching ( s , as ) ] ) ;
avio_printf ( s - > pb , " subsegmentAlignment= \" %s \" " ,
boolean [ subsegment_alignment ( s , as ) ] ) ;
for ( i = 0 ; i < as - > nb_streams ; i + + ) {
AVDictionaryEntry * kf = av_dict_get ( s - > streams [ as - > streams [ i ] ] - > metadata ,
CLUSTER_KEYFRAME , NULL , 0 ) ;
2014-08-15 21:33:21 +03:00
if ( ! kf | | ! strncmp ( kf - > value , " 0 " , 1 ) ) subsegmentStartsWithSAP = 0 ;
2014-07-14 19:52:23 +03:00
}
avio_printf ( s - > pb , " subsegmentStartsWithSAP= \" %d \" " , subsegmentStartsWithSAP ) ;
avio_printf ( s - > pb , " > \n " ) ;
for ( i = 0 ; i < as - > nb_streams ; i + + ) {
2014-11-11 20:02:05 +02:00
write_representation ( s , s - > streams [ as - > streams [ i ] ] , w - > representation_id + + ,
2014-10-01 20:13:32 +03:00
! width_in_as , ! height_in_as , ! sample_rate_in_as ) ;
2014-07-14 19:52:23 +03:00
}
avio_printf ( s - > pb , " </AdaptationSet> \n " ) ;
return 0 ;
}
static int to_integer ( char * p , int len )
{
int ret ;
char * q = av_malloc ( sizeof ( char ) * len ) ;
2014-08-15 21:33:21 +03:00
if ( ! q ) return - 1 ;
2014-08-29 14:02:34 +03:00
av_strlcpy ( q , p , len ) ;
2014-07-14 19:52:23 +03:00
ret = atoi ( q ) ;
av_free ( q ) ;
return ret ;
}
static int parse_adaptation_sets ( AVFormatContext * s )
{
WebMDashMuxContext * w = s - > priv_data ;
char * p = w - > adaptation_sets ;
char * q ;
enum { new_set , parsed_id , parsing_streams } state ;
// syntax id=0,streams=0,1,2 id=1,streams=3,4 and so on
state = new_set ;
while ( p < w - > adaptation_sets + strlen ( w - > adaptation_sets ) ) {
if ( * p = = ' ' )
continue ;
else if ( state = = new_set & & ! strncmp ( p , " id= " , 3 ) ) {
w - > as = av_realloc ( w - > as , sizeof ( * w - > as ) * + + w - > nb_as ) ;
if ( w - > as = = NULL ) return - 1 ;
w - > as [ w - > nb_as - 1 ] . nb_streams = 0 ;
w - > as [ w - > nb_as - 1 ] . streams = NULL ;
p + = 3 ; // consume "id="
q = w - > as [ w - > nb_as - 1 ] . id ;
while ( * p ! = ' , ' ) * q + + = * p + + ;
* q = 0 ;
p + + ;
state = parsed_id ;
} else if ( state = = parsed_id & & ! strncmp ( p , " streams= " , 8 ) ) {
p + = 8 ; // consume "streams="
state = parsing_streams ;
} else if ( state = = parsing_streams ) {
struct AdaptationSet * as = & w - > as [ w - > nb_as - 1 ] ;
q = p ;
while ( * q ! = ' \0 ' & & * q ! = ' , ' & & * q ! = ' ' ) q + + ;
as - > streams = av_realloc ( as - > streams , sizeof ( * as - > streams ) * + + as - > nb_streams ) ;
if ( as - > streams = = NULL ) return - 1 ;
2014-08-29 14:02:34 +03:00
as - > streams [ as - > nb_streams - 1 ] = to_integer ( p , q - p + 1 ) ;
2014-07-14 19:52:23 +03:00
if ( as - > streams [ as - > nb_streams - 1 ] < 0 ) return - 1 ;
if ( * q = = ' \0 ' ) break ;
if ( * q = = ' ' ) state = new_set ;
p = + + q ;
} else {
return - 1 ;
}
}
return 0 ;
}
static int webm_dash_manifest_write_header ( AVFormatContext * s )
{
int i ;
double start = 0.0 ;
WebMDashMuxContext * w = s - > priv_data ;
parse_adaptation_sets ( s ) ;
write_header ( s ) ;
avio_printf ( s - > pb , " <Period id= \" 0 \" " ) ;
avio_printf ( s - > pb , " start= \" PT%gS \" " , start ) ;
avio_printf ( s - > pb , " duration= \" PT%gS \" " , get_duration ( s ) ) ;
avio_printf ( s - > pb , " > \n " ) ;
for ( i = 0 ; i < w - > nb_as ; i + + ) {
if ( write_adaptation_set ( s , i ) < 0 ) return - 1 ;
}
avio_printf ( s - > pb , " </Period> \n " ) ;
write_footer ( s ) ;
return 0 ;
}
static int webm_dash_manifest_write_packet ( AVFormatContext * s , AVPacket * pkt )
{
return AVERROR_EOF ;
}
static int webm_dash_manifest_write_trailer ( AVFormatContext * s )
{
WebMDashMuxContext * w = s - > priv_data ;
int i ;
for ( i = 0 ; i < w - > nb_as ; i + + ) {
av_freep ( & w - > as [ i ] . streams ) ;
}
av_freep ( & w - > as ) ;
return 0 ;
}
# define OFFSET(x) offsetof(WebMDashMuxContext, x)
static const AVOption options [ ] = {
{ " adaptation_sets " , " Adaptation sets. Syntax: id=0,streams=0,1,2 id=1,streams=3,4 and so on " , OFFSET ( adaptation_sets ) , AV_OPT_TYPE_STRING , { 0 } , 0 , 0 , AV_OPT_FLAG_ENCODING_PARAM } ,
{ NULL } ,
} ;
# if CONFIG_WEBM_DASH_MANIFEST_MUXER
static const AVClass webm_dash_class = {
. class_name = " WebM DASH Manifest muxer " ,
. item_name = av_default_item_name ,
. option = options ,
. version = LIBAVUTIL_VERSION_INT ,
} ;
AVOutputFormat ff_webm_dash_manifest_muxer = {
. name = " webm_dash_manifest " ,
. long_name = NULL_IF_CONFIG_SMALL ( " WebM DASH Manifest " ) ,
. mime_type = " application/xml " ,
. extensions = " xml " ,
. priv_data_size = sizeof ( WebMDashMuxContext ) ,
. write_header = webm_dash_manifest_write_header ,
. write_packet = webm_dash_manifest_write_packet ,
. write_trailer = webm_dash_manifest_write_trailer ,
. priv_class = & webm_dash_class ,
} ;
# endif