gstreamer/gst/asfmux/gstasfmux.c
2021-04-11 16:16:55 +00:00

2459 lines
79 KiB
C

/* ASF muxer plugin for GStreamer
* Copyright (C) 2009 Thiago Santos <thiagoss@embedded.ufcg.edu.br>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
/* based on:
* - avimux (by Ronald Bultje and Mark Nauwelaerts)
* - qtmux (by Thiago Santos and Mark Nauwelaerts)
*/
/**
* SECTION:element-asfmux
* @title: asfmux
*
* Muxes media into an ASF file/stream.
*
* Pad names are either video_xx or audio_xx, where 'xx' is the
* stream number of the stream that goes through that pad. Stream numbers
* are assigned sequentially, starting from 1.
*
* ## Example launch lines
*
* (write everything in one line, without the backslash characters)
* |[
* gst-launch-1.0 videotestsrc num-buffers=250 \
* ! "video/x-raw,format=(string)I420,framerate=(fraction)25/1" ! avenc_wmv2 \
* ! asfmux name=mux ! filesink location=test.asf \
* audiotestsrc num-buffers=440 ! audioconvert \
* ! "audio/x-raw,rate=44100" ! avenc_wmav2 ! mux.
* ]| This creates an ASF file containing an WMV video stream
* with a test picture and WMA audio stream of a test sound.
*
* ## Live streaming
* asfmux and rtpasfpay are capable of generating a live asf stream.
* asfmux has to set its 'streamable' property to true, because in this
* mode it won't try to seek back to the start of the file to replace
* some fields that couldn't be known at the file start. In this mode,
* it won't also send indexes at the end of the data packets (the actual
* media content)
* the following pipelines are an example of this usage.
* (write everything in one line, without the backslash characters)
* Server (sender)
* |[
* gst-launch-1.0 -ve videotestsrc ! avenc_wmv2 ! asfmux name=mux streamable=true \
* ! rtpasfpay ! udpsink host=127.0.0.1 port=3333 \
* audiotestsrc ! avenc_wmav2 ! mux.
* ]|
* Client (receiver)
* |[
* gst-launch-1.0 udpsrc port=3333 ! "caps_from_rtpasfpay_at_sender" \
* ! rtpasfdepay ! decodebin name=d ! queue \
* ! videoconvert ! autovideosink \
* d. ! queue ! audioconvert ! autoaudiosink
* ]|
*
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string.h>
#include <stdio.h>
#include <gst/gst-i18n-plugin.h>
#include "gstasfmux.h"
#define DEFAULT_SIMPLE_INDEX_TIME_INTERVAL G_GUINT64_CONSTANT (10000000)
#define MAX_PAYLOADS_IN_A_PACKET 63
GST_DEBUG_CATEGORY_STATIC (asfmux_debug);
#define GST_CAT_DEFAULT asfmux_debug
enum
{
PROP_0,
PROP_PACKET_SIZE,
PROP_PREROLL,
PROP_MERGE_STREAM_TAGS,
PROP_PADDING,
PROP_STREAMABLE
};
/* Stores a tag list for the available/known tags
* in an ASF file
* Also stores the sizes those entries would use in a
* content description object and extended content
* description object
*/
typedef struct
{
GstTagList *tags;
guint64 cont_desc_size;
guint64 ext_cont_desc_size;
} GstAsfTags;
/* Helper struct to be used as user data
* in gst_tag_foreach function for writing
* each tag for the metadata objects
*
* stream_num is used only for stream dependent tags
*/
typedef struct
{
GstAsfMux *asfmux;
guint8 *buf;
guint16 count;
guint64 size;
guint16 stream_num;
} GstAsfExtContDescData;
typedef GstAsfExtContDescData GstAsfMetadataObjData;
#define DEFAULT_PACKET_SIZE 4800
#define DEFAULT_PREROLL 5000
#define DEFAULT_MERGE_STREAM_TAGS TRUE
#define DEFAULT_PADDING 0
#define DEFAULT_STREAMABLE FALSE
static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/x-ms-asf, " "parsed = (boolean) true")
);
static GstStaticPadTemplate video_sink_factory =
GST_STATIC_PAD_TEMPLATE ("video_%u",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_STATIC_CAPS ("video/x-wmv, wmvversion = (int) [1,3]"));
static GstStaticPadTemplate audio_sink_factory =
GST_STATIC_PAD_TEMPLATE ("audio_%u",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_STATIC_CAPS ("audio/x-wma, wmaversion = (int) [1,3]; "
"audio/mpeg, layer = (int) 3, mpegversion = (int) 1, "
"channels = (int) [1,2], rate = (int) [8000,96000]"));
static gboolean gst_asf_mux_audio_set_caps (GstPad * pad, GstCaps * caps);
static gboolean gst_asf_mux_video_set_caps (GstPad * pad, GstCaps * caps);
static GstPad *gst_asf_mux_request_new_pad (GstElement * element,
GstPadTemplate * templ, const gchar * name, const GstCaps * caps);
static void gst_asf_mux_set_property (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec);
static void gst_asf_mux_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec);
static GstStateChangeReturn gst_asf_mux_change_state (GstElement * element,
GstStateChange transition);
static gboolean gst_asf_mux_sink_event (GstCollectPads * pads,
GstCollectData * cdata, GstEvent * event, GstAsfMux * asfmux);
static void gst_asf_mux_pad_reset (GstAsfPad * data);
static GstFlowReturn gst_asf_mux_collected (GstCollectPads * collect,
gpointer data);
static GstElementClass *parent_class = NULL;
G_DEFINE_TYPE_WITH_CODE (GstAsfMux, gst_asf_mux, GST_TYPE_ELEMENT,
G_IMPLEMENT_INTERFACE (GST_TYPE_TAG_SETTER, NULL));
GST_ELEMENT_REGISTER_DEFINE (asfmux, "asfmux",
GST_RANK_PRIMARY, GST_TYPE_ASF_MUX);
static void
gst_asf_mux_reset (GstAsfMux * asfmux)
{
asfmux->state = GST_ASF_MUX_STATE_NONE;
asfmux->stream_number = 0;
asfmux->data_object_size = 0;
asfmux->data_object_position = 0;
asfmux->file_properties_object_position = 0;
asfmux->total_data_packets = 0;
asfmux->file_size = 0;
asfmux->packet_size = 0;
asfmux->first_ts = GST_CLOCK_TIME_NONE;
if (asfmux->payloads) {
GSList *walk;
for (walk = asfmux->payloads; walk; walk = g_slist_next (walk)) {
gst_asf_payload_free ((AsfPayload *) walk->data);
walk->data = NULL;
}
g_slist_free (asfmux->payloads);
}
asfmux->payloads = NULL;
asfmux->payload_data_size = 0;
asfmux->file_id.v1 = 0;
asfmux->file_id.v2 = 0;
asfmux->file_id.v3 = 0;
asfmux->file_id.v4 = 0;
gst_tag_setter_reset_tags (GST_TAG_SETTER (asfmux));
}
static void
gst_asf_mux_finalize (GObject * object)
{
GstAsfMux *asfmux;
asfmux = GST_ASF_MUX (object);
gst_asf_mux_reset (asfmux);
gst_object_unref (asfmux->collect);
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
gst_asf_mux_class_init (GstAsfMuxClass * klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
gobject_class = (GObjectClass *) klass;
gstelement_class = (GstElementClass *) klass;
parent_class = g_type_class_peek_parent (klass);
gobject_class->get_property = gst_asf_mux_get_property;
gobject_class->set_property = gst_asf_mux_set_property;
gobject_class->finalize = gst_asf_mux_finalize;
g_object_class_install_property (gobject_class, PROP_PACKET_SIZE,
g_param_spec_uint ("packet-size", "Packet size",
"The ASF packets size (bytes)",
ASF_MULTIPLE_PAYLOAD_HEADER_SIZE + 1, G_MAXUINT32,
DEFAULT_PACKET_SIZE,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_PREROLL,
g_param_spec_uint64 ("preroll", "Preroll",
"The preroll time (milliseconds)",
0, G_MAXUINT64,
DEFAULT_PREROLL,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_MERGE_STREAM_TAGS,
g_param_spec_boolean ("merge-stream-tags", "Merge Stream Tags",
"If the stream metadata (received as events in the sink) should be "
"merged to the main file metadata.",
DEFAULT_MERGE_STREAM_TAGS,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_PADDING,
g_param_spec_uint64 ("padding", "Padding",
"Size of the padding object to be added to the end of the header. "
"If this less than 24 (the smaller size of an ASF object), "
"no padding is added.",
0, G_MAXUINT64,
DEFAULT_PADDING,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_STREAMABLE,
g_param_spec_boolean ("streamable", "Streamable",
"If set to true, the output should be as if it is to be streamed "
"and hence no indexes written or duration written.",
DEFAULT_STREAMABLE,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
gstelement_class->request_new_pad =
GST_DEBUG_FUNCPTR (gst_asf_mux_request_new_pad);
gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_asf_mux_change_state);
gst_element_class_add_static_pad_template (gstelement_class, &src_factory);
gst_element_class_add_static_pad_template (gstelement_class,
&audio_sink_factory);
gst_element_class_add_static_pad_template (gstelement_class,
&video_sink_factory);
gst_element_class_set_static_metadata (gstelement_class, "ASF muxer",
"Codec/Muxer",
"Muxes audio and video into an ASF stream",
"Thiago Santos <thiagoss@embedded.ufcg.edu.br>");
GST_DEBUG_CATEGORY_INIT (asfmux_debug, "asfmux", 0, "Muxer for ASF streams");
}
static void
gst_asf_mux_init (GstAsfMux * asfmux)
{
asfmux->srcpad = gst_pad_new_from_static_template (&src_factory, "src");
gst_pad_use_fixed_caps (asfmux->srcpad);
gst_element_add_pad (GST_ELEMENT (asfmux), asfmux->srcpad);
asfmux->collect = gst_collect_pads_new ();
gst_collect_pads_set_function (asfmux->collect,
(GstCollectPadsFunction) GST_DEBUG_FUNCPTR (gst_asf_mux_collected),
asfmux);
gst_collect_pads_set_event_function (asfmux->collect,
(GstCollectPadsEventFunction) GST_DEBUG_FUNCPTR (gst_asf_mux_sink_event),
asfmux);
asfmux->payloads = NULL;
asfmux->prop_packet_size = DEFAULT_PACKET_SIZE;
asfmux->prop_preroll = DEFAULT_PREROLL;
asfmux->prop_merge_stream_tags = DEFAULT_MERGE_STREAM_TAGS;
asfmux->prop_padding = DEFAULT_PADDING;
asfmux->prop_streamable = DEFAULT_STREAMABLE;
gst_asf_mux_reset (asfmux);
}
static gboolean
gst_asf_mux_sink_event (GstCollectPads * pads, GstCollectData * cdata,
GstEvent * event, GstAsfMux * asfmux)
{
GstAsfPad *asfpad = (GstAsfPad *) cdata;
gboolean ret = TRUE;
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_CAPS:
{
GstCaps *caps;
gst_event_parse_caps (event, &caps);
if (asfpad->is_audio)
ret = gst_asf_mux_audio_set_caps (cdata->pad, caps);
else
ret = gst_asf_mux_video_set_caps (cdata->pad, caps);
gst_event_unref (event);
event = NULL;
break;
}
case GST_EVENT_TAG:{
GST_DEBUG_OBJECT (asfmux, "received tag event");
/* we discard tag events that come after we started
* writing the headers, because tags are to be in
* the headers
*/
if (asfmux->state == GST_ASF_MUX_STATE_NONE) {
GstTagList *list = NULL;
gst_event_parse_tag (event, &list);
if (asfmux->merge_stream_tags) {
GstTagSetter *setter = GST_TAG_SETTER (asfmux);
const GstTagMergeMode mode =
gst_tag_setter_get_tag_merge_mode (setter);
gst_tag_setter_merge_tags (setter, list, mode);
} else {
if (asfpad->taglist == NULL) {
asfpad->taglist = gst_tag_list_new_empty ();
}
gst_tag_list_insert (asfpad->taglist, list, GST_TAG_MERGE_REPLACE);
}
}
break;
}
default:
break;
}
if (event != NULL)
return gst_collect_pads_event_default (pads, cdata, event, FALSE);
return ret;
}
/**
* gst_asf_mux_push_buffer:
* @asfmux: #GstAsfMux that should push the buffer
* @buf: #GstBuffer to be pushed
*
* Pushes a buffer downstream and adds its size to the total file size
*
* Returns: the result of #gst_pad_push on the buffer
*/
static GstFlowReturn
gst_asf_mux_push_buffer (GstAsfMux * asfmux, GstBuffer * buf, gsize bufsize)
{
GstFlowReturn ret;
ret = gst_pad_push (asfmux->srcpad, buf);
if (ret == GST_FLOW_OK)
asfmux->file_size += bufsize;
return ret;
}
/**
* content_description_calc_size_for_tag:
* @taglist: the #GstTagList that contains the tag
* @tag: the tag's name
* @user_data: a #GstAsfTags struct for putting the results
*
* Function that has the #GstTagForEach signature and
* is used to calculate the size in bytes for each tag
* that can be contained in asf's content description object
* and extended content description object. This size is added
* to the total size for each of that objects in the #GstAsfTags
* struct passed in the user_data pointer.
*/
static void
content_description_calc_size_for_tag (const GstTagList * taglist,
const gchar * tag, gpointer user_data)
{
const gchar *asftag = gst_asf_get_asf_tag (tag);
GValue value = { 0 };
guint type;
GstAsfTags *asftags = (GstAsfTags *) user_data;
guint content_size;
if (asftag == NULL)
return;
if (!gst_tag_list_copy_value (&value, taglist, tag)) {
return;
}
type = gst_asf_get_tag_field_type (&value);
switch (type) {
case ASF_TAG_TYPE_UNICODE_STR:
{
const gchar *text;
text = g_value_get_string (&value);
/* +1 -> because of the \0 at the end
* 2* -> because we have uft8, and asf demands utf16
*/
content_size = 2 * (1 + g_utf8_strlen (text, -1));
if (gst_asf_tag_present_in_content_description (tag)) {
asftags->cont_desc_size += content_size;
}
}
break;
case ASF_TAG_TYPE_DWORD:
content_size = 4;
break;
default:
GST_WARNING ("Unhandled asf tag field type %u for tag %s", type, tag);
g_value_reset (&value);
return;
}
if (asftag) {
/* size of the tag content in utf16 +
* size of the tag name +
* 3 uint16 (size of the tag name string,
* size of the tag content string and
* type of content
*/
asftags->ext_cont_desc_size += content_size +
(g_utf8_strlen (asftag, -1) + 1) * 2 + 6;
}
gst_tag_list_add_value (asftags->tags, GST_TAG_MERGE_REPLACE, tag, &value);
g_value_reset (&value);
}
/* FIXME
* it is awful to keep track of the size here
* and get the same tags in the writing function */
/**
* gst_asf_mux_get_content_description_tags:
* @asfmux: #GstAsfMux to have its tags processed
* @asftags: #GstAsfTags to hold the results
*
* Inspects the tags received by the GstTagSetter interface
* or possibly by sink tag events and calculates the total
* size needed for the default and extended content description objects.
* This results and a copy of the #GstTagList
* are stored in the #GstAsfTags. We store a copy so that
* the sizes estimated here maintain the same until they are
* written to the asf file.
*/
static void
gst_asf_mux_get_content_description_tags (GstAsfMux * asfmux,
GstAsfTags * asftags)
{
const GstTagList *tags;
tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (asfmux));
if (tags && !gst_tag_list_is_empty (tags)) {
if (asftags->tags != NULL) {
gst_tag_list_unref (asftags->tags);
}
asftags->tags = gst_tag_list_new_empty ();
asftags->cont_desc_size = 0;
asftags->ext_cont_desc_size = 0;
GST_DEBUG_OBJECT (asfmux, "Processing tags");
gst_tag_list_foreach (tags, content_description_calc_size_for_tag, asftags);
} else {
GST_DEBUG_OBJECT (asfmux, "No tags received");
}
if (asftags->cont_desc_size > 0) {
asftags->cont_desc_size += ASF_CONTENT_DESCRIPTION_OBJECT_SIZE;
}
if (asftags->ext_cont_desc_size > 0) {
asftags->ext_cont_desc_size += ASF_EXT_CONTENT_DESCRIPTION_OBJECT_SIZE;
}
}
/**
* add_metadata_tag_size:
* @taglist: #GstTagList
* @tag: tag name
* @user_data: pointer to a guint to store the result
*
* GstTagForeachFunc implementation that accounts the size of
* each tag in the taglist and adds them to the guint pointed
* by the user_data
*/
static void
add_metadata_tag_size (const GstTagList * taglist, const gchar * tag,
gpointer user_data)
{
const gchar *asftag = gst_asf_get_asf_tag (tag);
GValue value = { 0 };
guint type;
guint content_size;
guint *total_size = (guint *) user_data;
if (asftag == NULL)
return;
if (!gst_tag_list_copy_value (&value, taglist, tag)) {
return;
}
type = gst_asf_get_tag_field_type (&value);
switch (type) {
case ASF_TAG_TYPE_UNICODE_STR:
{
const gchar *text;
text = g_value_get_string (&value);
/* +1 -> because of the \0 at the end
* 2* -> because we have uft8, and asf demands utf16
*/
content_size = 2 * (1 + g_utf8_strlen (text, -1));
}
break;
case ASF_TAG_TYPE_DWORD:
content_size = 4;
break;
default:
GST_WARNING ("Unhandled asf tag field type %u for tag %s", type, tag);
g_value_reset (&value);
return;
}
/* size of reserved (2) +
* size of stream number (2) +
* size of the tag content in utf16 +
* size of the tag name +
* 2 uint16 (size of the tag name string and type of content) +
* 1 uint32 (size of the data)
*/
*total_size +=
4 + content_size + (g_utf8_strlen (asftag, -1) + 1) * 2 + 4 + 4;
g_value_reset (&value);
}
/**
* gst_asf_mux_get_metadata_object_size:
* @asfmux: #GstAsfMux
* @asfpad: pad for which the metadata object size should be calculated
*
* Calculates the size of the metadata object for the tags of the stream
* handled by the asfpad in the parameter
*
* Returns: The size calculated
*/
static guint
gst_asf_mux_get_metadata_object_size (GstAsfMux * asfmux, GstAsfPad * asfpad)
{
guint size = ASF_METADATA_OBJECT_SIZE;
if (asfpad->taglist == NULL || gst_tag_list_is_empty (asfpad->taglist))
return 0;
gst_tag_list_foreach (asfpad->taglist, add_metadata_tag_size, &size);
return size;
}
/**
* gst_asf_mux_get_headers_size:
* @asfmux: #GstAsfMux
*
* Calculates the size of the headers of the asf stream
* to be generated by this #GstAsfMux.
* Its used for determining the size of the buffer to allocate
* to exactly fit the headers in.
* Padding and metadata objects sizes are not included.
*
* Returns: the calculated size
*/
static guint
gst_asf_mux_get_headers_size (GstAsfMux * asfmux)
{
GSList *walk;
gint stream_num = 0;
guint size = ASF_HEADER_OBJECT_SIZE +
ASF_FILE_PROPERTIES_OBJECT_SIZE + ASF_HEADER_EXTENSION_OBJECT_SIZE;
/* per stream data */
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
GstAsfPad *asfpad = (GstAsfPad *) walk->data;
if (asfpad->is_audio)
size += ASF_AUDIO_SPECIFIC_DATA_SIZE;
else
size += ASF_VIDEO_SPECIFIC_DATA_SIZE;
if (asfpad->codec_data)
size += gst_buffer_get_size (asfpad->codec_data);
stream_num++;
}
size += stream_num * (ASF_STREAM_PROPERTIES_OBJECT_SIZE +
ASF_EXTENDED_STREAM_PROPERTIES_OBJECT_SIZE);
return size;
}
/**
* gst_asf_mux_write_header_object:
* @asfmux:
* @buf: pointer to the data pointer
* @size: size of the header object
* @child_objects: number of children objects inside the main header object
*
* Writes the main asf header object start. The buffer pointer
* is incremented to the next writing position.
*/
static void
gst_asf_mux_write_header_object (GstAsfMux * asfmux, guint8 ** buf,
guint64 size, guint32 child_objects)
{
gst_asf_put_guid (*buf, guids[ASF_HEADER_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, size); /* object size */
GST_WRITE_UINT32_LE (*buf + 24, child_objects); /* # of child objects */
GST_WRITE_UINT8 (*buf + 28, 0x01); /* reserved */
GST_WRITE_UINT8 (*buf + 29, 0x02); /* reserved */
*buf += ASF_HEADER_OBJECT_SIZE;
}
/**
* gst_asf_mux_write_file_properties:
* @asfmux:
* @buf: pointer to the data pointer
*
* Writes the file properties object to the buffer. The buffer pointer
* is incremented to the next writing position.
*/
static void
gst_asf_mux_write_file_properties (GstAsfMux * asfmux, guint8 ** buf)
{
gst_asf_put_guid (*buf, guids[ASF_FILE_PROPERTIES_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, ASF_FILE_PROPERTIES_OBJECT_SIZE); /* object size */
gst_asf_put_guid (*buf + 24, asfmux->file_id);
GST_WRITE_UINT64_LE (*buf + 40, 0); /* file size - needs update */
gst_asf_put_time (*buf + 48, gst_asf_get_current_time ()); /* creation time */
GST_WRITE_UINT64_LE (*buf + 56, 0); /* data packets - needs update */
GST_WRITE_UINT64_LE (*buf + 64, 0); /* play duration - needs update */
GST_WRITE_UINT64_LE (*buf + 72, 0); /* send duration - needs update */
GST_WRITE_UINT64_LE (*buf + 80, asfmux->preroll); /* preroll */
GST_WRITE_UINT32_LE (*buf + 88, 0x1); /* flags - broadcast on */
GST_WRITE_UINT32_LE (*buf + 92, asfmux->packet_size); /* minimum data packet size */
GST_WRITE_UINT32_LE (*buf + 96, asfmux->packet_size); /* maximum data packet size */
GST_WRITE_UINT32_LE (*buf + 100, 0); /* maximum bitrate TODO */
*buf += ASF_FILE_PROPERTIES_OBJECT_SIZE;
}
/**
* gst_asf_mux_write_stream_properties:
* @asfmux:
* @buf: pointer to the data pointer
* @asfpad: Pad that handles the stream
*
* Writes the stream properties object in the buffer
* for the stream handled by the #GstAsfPad passed.
* The pointer is incremented to the next writing position
*/
static void
gst_asf_mux_write_stream_properties (GstAsfMux * asfmux, guint8 ** buf,
GstAsfPad * asfpad)
{
guint32 codec_data_length = 0;
guint32 media_specific_data_length = 0;
guint16 flags = 0;
/* codec specific data length */
if (asfpad->codec_data)
codec_data_length = gst_buffer_get_size (asfpad->codec_data);
if (asfpad->is_audio)
media_specific_data_length = ASF_AUDIO_SPECIFIC_DATA_SIZE;
else
media_specific_data_length = ASF_VIDEO_SPECIFIC_DATA_SIZE;
GST_DEBUG_OBJECT (asfmux, "Stream %" G_GUINT16_FORMAT " codec data length: %"
G_GUINT32_FORMAT ", media specific data length: %" G_GUINT32_FORMAT,
(guint16) asfpad->stream_number, codec_data_length,
media_specific_data_length);
gst_asf_put_guid (*buf, guids[ASF_STREAM_PROPERTIES_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, ASF_STREAM_PROPERTIES_OBJECT_SIZE + codec_data_length + media_specific_data_length); /* object size */
/* stream type */
if (asfpad->is_audio)
gst_asf_put_guid (*buf + 24, guids[ASF_AUDIO_MEDIA_INDEX]);
else
gst_asf_put_guid (*buf + 24, guids[ASF_VIDEO_MEDIA_INDEX]);
/* error correction */
gst_asf_put_guid (*buf + 40, guids[ASF_NO_ERROR_CORRECTION_INDEX]);
GST_WRITE_UINT64_LE (*buf + 56, 0); /* time offset */
GST_WRITE_UINT32_LE (*buf + 64, codec_data_length + media_specific_data_length); /* type specific data length */
GST_WRITE_UINT32_LE (*buf + 68, 0); /* error correction data length */
flags = (asfpad->stream_number & 0x7F);
GST_WRITE_UINT16_LE (*buf + 72, flags);
GST_WRITE_UINT32_LE (*buf + 74, 0); /* reserved */
*buf += ASF_STREAM_PROPERTIES_OBJECT_SIZE;
/* audio specific data */
if (asfpad->is_audio) {
GstAsfAudioPad *audiopad = (GstAsfAudioPad *) asfpad;
GST_WRITE_UINT16_LE (*buf, audiopad->audioinfo.format);
GST_WRITE_UINT16_LE (*buf + 2, audiopad->audioinfo.channels);
GST_WRITE_UINT32_LE (*buf + 4, audiopad->audioinfo.rate);
GST_WRITE_UINT32_LE (*buf + 8, audiopad->audioinfo.av_bps);
GST_WRITE_UINT16_LE (*buf + 12, audiopad->audioinfo.blockalign);
GST_WRITE_UINT16_LE (*buf + 14, audiopad->audioinfo.bits_per_sample);
GST_WRITE_UINT16_LE (*buf + 16, codec_data_length);
GST_DEBUG_OBJECT (asfmux,
"wave formatex values: codec_id=%" G_GUINT16_FORMAT ", channels=%"
G_GUINT16_FORMAT ", rate=%" G_GUINT32_FORMAT ", bytes_per_sec=%"
G_GUINT32_FORMAT ", block_alignment=%" G_GUINT16_FORMAT
", bits_per_sample=%" G_GUINT16_FORMAT ", codec_data_length=%u",
audiopad->audioinfo.format, audiopad->audioinfo.channels,
audiopad->audioinfo.rate, audiopad->audioinfo.av_bps,
audiopad->audioinfo.blockalign, audiopad->audioinfo.bits_per_sample,
codec_data_length);
*buf += ASF_AUDIO_SPECIFIC_DATA_SIZE;
} else {
GstAsfVideoPad *videopad = (GstAsfVideoPad *) asfpad;
GST_WRITE_UINT32_LE (*buf, (guint32) videopad->vidinfo.width);
GST_WRITE_UINT32_LE (*buf + 4, (guint32) videopad->vidinfo.height);
GST_WRITE_UINT8 (*buf + 8, 2);
/* the BITMAPINFOHEADER size + codec_data size */
GST_WRITE_UINT16_LE (*buf + 9,
ASF_VIDEO_SPECIFIC_DATA_SIZE + codec_data_length - 11);
/* BITMAPINFOHEADER */
GST_WRITE_UINT32_LE (*buf + 11,
ASF_VIDEO_SPECIFIC_DATA_SIZE + codec_data_length - 11);
gst_asf_put_i32 (*buf + 15, videopad->vidinfo.width);
gst_asf_put_i32 (*buf + 19, videopad->vidinfo.height);
GST_WRITE_UINT16_LE (*buf + 23, 1); /* reserved */
GST_WRITE_UINT16_LE (*buf + 25, videopad->vidinfo.bit_cnt);
GST_WRITE_UINT32_LE (*buf + 27, videopad->vidinfo.compression);
GST_WRITE_UINT32_LE (*buf + 31, videopad->vidinfo.width *
videopad->vidinfo.height * videopad->vidinfo.bit_cnt);
GST_WRITE_UINT32_LE (*buf + 35, videopad->vidinfo.xpels_meter);
GST_WRITE_UINT32_LE (*buf + 39, videopad->vidinfo.ypels_meter);
GST_WRITE_UINT32_LE (*buf + 43, videopad->vidinfo.num_colors);
GST_WRITE_UINT32_LE (*buf + 47, videopad->vidinfo.imp_colors);
*buf += ASF_VIDEO_SPECIFIC_DATA_SIZE;
}
if (codec_data_length > 0)
gst_buffer_extract (asfpad->codec_data, 0, *buf, codec_data_length);
*buf += codec_data_length;
}
/**
* gst_asf_mux_write_header_extension:
* @asfmux:
* @buf: pointer to the buffer pointer
* @extension_size: size of the extensions
*
* Writes the header of the header extension object. The buffer pointer
* is incremented to the next writing position (the header extension object
* childs should be written from that point)
*/
static void
gst_asf_mux_write_header_extension (GstAsfMux * asfmux, guint8 ** buf,
guint64 extension_size)
{
gst_asf_put_guid (*buf, guids[ASF_HEADER_EXTENSION_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, ASF_HEADER_EXTENSION_OBJECT_SIZE + extension_size); /* object size */
gst_asf_put_guid (*buf + 24, guids[ASF_RESERVED_1_INDEX]); /* reserved */
GST_WRITE_UINT16_LE (*buf + 40, 6); /* reserved */
GST_WRITE_UINT32_LE (*buf + 42, extension_size); /* header extension data size */
*buf += ASF_HEADER_EXTENSION_OBJECT_SIZE;
}
/**
* gst_asf_mux_write_extended_stream_properties:
* @asfmux:
* @buf: pointer to the buffer pointer
* @asfpad: Pad that handles the stream of the properties to be written
*
* Writes the extended stream properties object (that is part of the
* header extension objects) for the stream handled by asfpad
*/
static void
gst_asf_mux_write_extended_stream_properties (GstAsfMux * asfmux, guint8 ** buf,
GstAsfPad * asfpad)
{
gst_asf_put_guid (*buf, guids[ASF_EXTENDED_STREAM_PROPERTIES_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, ASF_EXTENDED_STREAM_PROPERTIES_OBJECT_SIZE);
GST_WRITE_UINT64_LE (*buf + 24, 0); /* start time */
GST_WRITE_UINT64_LE (*buf + 32, 0); /* end time */
GST_WRITE_UINT32_LE (*buf + 40, asfpad->bitrate); /* bitrate */
GST_WRITE_UINT32_LE (*buf + 44, 0); /* buffer size */
GST_WRITE_UINT32_LE (*buf + 48, 0); /* initial buffer fullness */
GST_WRITE_UINT32_LE (*buf + 52, asfpad->bitrate); /* alternate data bitrate */
GST_WRITE_UINT32_LE (*buf + 56, 0); /* alternate buffer size */
GST_WRITE_UINT32_LE (*buf + 60, 0); /* alternate initial buffer fullness */
GST_WRITE_UINT32_LE (*buf + 64, 0); /* maximum object size */
/* flags */
if (asfpad->is_audio) {
/* TODO check if audio is seekable */
GST_WRITE_UINT32_LE (*buf + 68, 0x0);
} else {
/* video has indexes, so it is seekable unless we are streaming */
if (asfmux->prop_streamable)
GST_WRITE_UINT32_LE (*buf + 68, 0x0);
else
GST_WRITE_UINT32_LE (*buf + 68, 0x2);
}
GST_WRITE_UINT16_LE (*buf + 72, asfpad->stream_number);
GST_WRITE_UINT16_LE (*buf + 74, 0); /* language index */
GST_WRITE_UINT64_LE (*buf + 76, 0); /* avg time per frame */
GST_WRITE_UINT16_LE (*buf + 84, 0); /* stream name count */
GST_WRITE_UINT16_LE (*buf + 86, 0); /* payload extension count */
*buf += ASF_EXTENDED_STREAM_PROPERTIES_OBJECT_SIZE;
}
/**
* gst_asf_mux_write_string_with_size:
* @asfmux:
* @size_buf: pointer to the memory position to write the size of the string
* @str_buf: pointer to the memory position to write the string
* @str: the string to be written (in UTF-8)
* @use32: if the string size should be written with 32 bits (if true)
* or with 16 (if false)
*
* Writes a string with its size as it is needed in many asf objects.
* The size is written to size_buf as a WORD field if use32 is false, and
* as a DWORD if use32 is true. The string is written to str_buf in UTF16-LE.
* The string should be passed in UTF-8.
*
* The string size in UTF16-LE is returned.
*/
static guint64
gst_asf_mux_write_string_with_size (GstAsfMux * asfmux,
guint8 * size_buf, guint8 * str_buf, const gchar * str, gboolean use32)
{
GError *error = NULL;
gsize str_size = 0;
gchar *str_utf16 = NULL;
GST_LOG_OBJECT (asfmux, "Writing extended content description string: "
"%s", str);
/*
* Covert the string to utf16
* Also force the last bytes to null terminated,
* tags were with extra weird characters without it.
*/
str_utf16 = g_convert (str, -1, "UTF-16LE", "UTF-8", NULL, &str_size, &error);
/* sum up the null terminating char */
str_size += 2;
if (use32)
GST_WRITE_UINT32_LE (size_buf, str_size);
else
GST_WRITE_UINT16_LE (size_buf, str_size);
if (error) {
GST_WARNING_OBJECT (asfmux, "Error converting string "
"to UTF-16: %s - %s", str, error->message);
g_error_free (error);
memset (str_buf, 0, str_size);
} else {
/* HACK: g_convert seems to add only a single byte null char to
* the end of the stream, we force the second one */
memcpy (str_buf, str_utf16, str_size - 1);
str_buf[str_size - 1] = 0;
}
g_free (str_utf16);
return str_size;
}
/**
* gst_asf_mux_write_content_description_entry:
* @asfmux:
* @tags:
* @tagname:
* @size_buf:
* @data_buf:
*
* Checks if a string tag with tagname exists in the taglist. If it
* exists it is written as an UTF-16LE to data_buf and its size in bytes
* is written to size_buf. It is used for writing content description
* object fields.
*
* Returns: the size of the string
*/
static guint16
gst_asf_mux_write_content_description_entry (GstAsfMux * asfmux,
const GstTagList * tags, const gchar * tagname,
guint8 * size_buf, guint8 * data_buf)
{
gchar *text = NULL;
guint16 text_size = 0;
if (gst_tag_list_get_string (tags, tagname, &text)) {
text_size = gst_asf_mux_write_string_with_size (asfmux, size_buf,
data_buf, text, FALSE);
g_free (text);
} else {
GST_WRITE_UINT16_LE (size_buf, 0);
}
return text_size;
}
static guint64
gst_asf_mux_write_ext_content_description_dword_entry (GstAsfMux * asfmux,
guint8 * buf, const gchar * asf_tag, const guint32 value)
{
guint64 tag_size;
GST_DEBUG_OBJECT (asfmux, "Writing extended content description tag: "
"%s (%u)", asf_tag, value);
tag_size = gst_asf_mux_write_string_with_size (asfmux, buf, buf + 2,
asf_tag, FALSE);
buf += tag_size + 2;
GST_WRITE_UINT16_LE (buf, ASF_TAG_TYPE_DWORD);
GST_WRITE_UINT16_LE (buf + 2, 4);
GST_WRITE_UINT32_LE (buf + 4, value);
/* tagsize -> string size
* 2 -> string size field size
* 4 -> dword entry
* 4 -> type of entry + entry size
*/
return tag_size + 2 + 4 + 4;
}
static guint64
gst_asf_mux_write_ext_content_description_string_entry (GstAsfMux * asfmux,
guint8 * buf, const gchar * asf_tag, const gchar * text)
{
guint64 tag_size = 0;
guint64 text_size = 0;
GST_DEBUG_OBJECT (asfmux, "Writing extended content description tag: "
"%s (%s)", asf_tag, text);
tag_size = gst_asf_mux_write_string_with_size (asfmux,
buf, buf + 2, asf_tag, FALSE);
GST_WRITE_UINT16_LE (buf + tag_size + 2, ASF_TAG_TYPE_UNICODE_STR);
buf += tag_size + 2 + 2;
text_size = gst_asf_mux_write_string_with_size (asfmux,
buf, buf + 2, text, FALSE);
/* the size of the strings in utf16-le plus the 3 WORD fields */
return tag_size + text_size + 6;
}
static void
gst_asf_mux_write_content_description (GstAsfMux * asfmux, guint8 ** buf,
const GstTagList * tags)
{
guint8 *values = (*buf) + ASF_CONTENT_DESCRIPTION_OBJECT_SIZE;
guint64 size = 0;
GST_DEBUG_OBJECT (asfmux, "Writing content description object");
gst_asf_put_guid (*buf, guids[ASF_CONTENT_DESCRIPTION_INDEX]);
values += gst_asf_mux_write_content_description_entry (asfmux, tags,
GST_TAG_TITLE, *buf + 24, values);
values += gst_asf_mux_write_content_description_entry (asfmux, tags,
GST_TAG_ARTIST, *buf + 26, values);
values += gst_asf_mux_write_content_description_entry (asfmux, tags,
GST_TAG_COPYRIGHT, *buf + 28, values);
values += gst_asf_mux_write_content_description_entry (asfmux, tags,
GST_TAG_DESCRIPTION, *buf + 30, values);
/* rating is currently not present in gstreamer tags, so we put 0 */
GST_WRITE_UINT16_LE (*buf + 32, 0);
size += values - *buf;
GST_WRITE_UINT64_LE (*buf + 16, size);
*buf += size;
}
static void
write_ext_content_description_tag (const GstTagList * taglist,
const gchar * tag, gpointer user_data)
{
const gchar *asftag = gst_asf_get_asf_tag (tag);
GValue value = { 0 };
guint type;
GstAsfExtContDescData *data = (GstAsfExtContDescData *) user_data;
if (asftag == NULL)
return;
if (!gst_tag_list_copy_value (&value, taglist, tag)) {
return;
}
type = gst_asf_get_tag_field_type (&value);
switch (type) {
case ASF_TAG_TYPE_UNICODE_STR:
{
const gchar *text;
text = g_value_get_string (&value);
data->size +=
gst_asf_mux_write_ext_content_description_string_entry (data->asfmux,
data->buf + data->size, asftag, text);
}
break;
case ASF_TAG_TYPE_DWORD:
{
guint num = g_value_get_uint (&value);
data->size +=
gst_asf_mux_write_ext_content_description_dword_entry (data->asfmux,
data->buf + data->size, asftag, num);
}
break;
default:
GST_WARNING_OBJECT (data->asfmux,
"Unhandled asf tag field type %u for tag %s", type, tag);
g_value_reset (&value);
return;
}
data->count++;
g_value_reset (&value);
}
static void
gst_asf_mux_write_ext_content_description (GstAsfMux * asfmux, guint8 ** buf,
GstTagList * tags)
{
GstAsfExtContDescData extContDesc;
extContDesc.asfmux = asfmux;
extContDesc.buf = *buf;
extContDesc.count = 0;
extContDesc.size = ASF_EXT_CONTENT_DESCRIPTION_OBJECT_SIZE;
GST_DEBUG_OBJECT (asfmux, "Writing extended content description object");
gst_asf_put_guid (*buf, guids[ASF_EXT_CONTENT_DESCRIPTION_INDEX]);
gst_tag_list_foreach (tags, write_ext_content_description_tag, &extContDesc);
GST_WRITE_UINT64_LE (*buf + 16, extContDesc.size);
GST_WRITE_UINT16_LE (*buf + 24, extContDesc.count);
*buf += extContDesc.size;
}
static void
write_metadata_tag (const GstTagList * taglist, const gchar * tag,
gpointer user_data)
{
const gchar *asftag = gst_asf_get_asf_tag (tag);
GValue value = { 0 };
guint type;
GstAsfMetadataObjData *data = (GstAsfMetadataObjData *) user_data;
guint16 tag_size;
guint32 content_size;
if (asftag == NULL)
return;
if (!gst_tag_list_copy_value (&value, taglist, tag)) {
return;
}
type = gst_asf_get_tag_field_type (&value);
switch (type) {
case ASF_TAG_TYPE_UNICODE_STR:
{
const gchar *text;
text = g_value_get_string (&value);
GST_WRITE_UINT16_LE (data->buf + data->size, 0);
GST_WRITE_UINT16_LE (data->buf + data->size + 2, data->stream_num);
data->size += 4;
tag_size = gst_asf_mux_write_string_with_size (data->asfmux,
data->buf + data->size, data->buf + data->size + 8, asftag, FALSE);
data->size += 2;
GST_WRITE_UINT16_LE (data->buf + data->size, type);
data->size += 2;
content_size = gst_asf_mux_write_string_with_size (data->asfmux,
data->buf + data->size, data->buf + data->size + tag_size + 4, text,
TRUE);
data->size += tag_size + content_size + 4;
}
break;
case ASF_TAG_TYPE_DWORD:
{
guint num = g_value_get_uint (&value);
GST_WRITE_UINT16_LE (data->buf + data->size, 0);
GST_WRITE_UINT16_LE (data->buf + data->size + 2, data->stream_num);
data->size += 4;
tag_size = gst_asf_mux_write_string_with_size (data->asfmux,
data->buf + data->size, data->buf + data->size + 8, asftag, FALSE);
data->size += 2;
GST_WRITE_UINT16_LE (data->buf + data->size, type);
data->size += 2;
/* dword length */
GST_WRITE_UINT32_LE (data->buf + data->size, 4);
data->size += 4 + tag_size;
GST_WRITE_UINT32_LE (data->buf + data->size, num);
data->size += 4;
}
break;
default:
GST_WARNING_OBJECT (data->asfmux,
"Unhandled asf tag field type %u for tag %s", type, tag);
g_value_reset (&value);
return;
}
data->count++;
g_value_reset (&value);
}
static void
gst_asf_mux_write_metadata_object (GstAsfMux * asfmux, guint8 ** buf,
GstAsfPad * asfpad)
{
GstAsfMetadataObjData metaObjData;
metaObjData.asfmux = asfmux;
metaObjData.buf = *buf;
metaObjData.count = 0;
metaObjData.size = ASF_METADATA_OBJECT_SIZE;
metaObjData.stream_num = asfpad->stream_number;
if (asfpad->taglist == NULL || gst_tag_list_is_empty (asfpad->taglist))
return;
GST_DEBUG_OBJECT (asfmux, "Writing metadata object");
gst_asf_put_guid (*buf, guids[ASF_METADATA_OBJECT_INDEX]);
gst_tag_list_foreach (asfpad->taglist, write_metadata_tag, &metaObjData);
GST_WRITE_UINT64_LE (*buf + 16, metaObjData.size);
GST_WRITE_UINT16_LE (*buf + 24, metaObjData.count);
*buf += metaObjData.size;
}
static void
gst_asf_mux_write_padding_object (GstAsfMux * asfmux, guint8 ** buf,
guint64 padding)
{
if (padding < ASF_PADDING_OBJECT_SIZE) {
return;
}
GST_DEBUG_OBJECT (asfmux, "Writing padding object of size %" G_GUINT64_FORMAT,
padding);
gst_asf_put_guid (*buf, guids[ASF_PADDING_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (*buf + 16, padding);
memset (*buf + 24, 0, padding - ASF_PADDING_OBJECT_SIZE);
*buf += padding;
}
static void
gst_asf_mux_write_data_object (GstAsfMux * asfmux, guint8 ** buf)
{
gst_asf_put_guid (*buf, guids[ASF_DATA_OBJECT_INDEX]);
/* Data object size. This is always >= ASF_DATA_OBJECT_SIZE. The standard
* specifically accepts the value 0 in live streams, but WMP is not accepting
* this while streaming using WMSP, so we default to minimum size also for
* live streams. Otherwise this field must be updated later on when we know
* the complete stream size.
*/
GST_WRITE_UINT64_LE (*buf + 16, ASF_DATA_OBJECT_SIZE);
gst_asf_put_guid (*buf + 24, asfmux->file_id);
GST_WRITE_UINT64_LE (*buf + 40, 0); /* total data packets */
GST_WRITE_UINT16_LE (*buf + 48, 0x0101); /* reserved */
*buf += ASF_DATA_OBJECT_SIZE;
}
static void
gst_asf_mux_put_buffer_in_streamheader (GValue * streamheader,
GstBuffer * buffer)
{
GValue value = { 0 };
GstBuffer *buf;
g_value_init (&value, GST_TYPE_BUFFER);
buf = gst_buffer_copy (buffer);
gst_value_set_buffer (&value, buf);
gst_buffer_unref (buf);
gst_value_array_append_value (streamheader, &value);
g_value_unset (&value);
}
static guint
gst_asf_mux_find_payload_parsing_info_size (GstAsfMux * asfmux)
{
/* Minimum payload parsing information size is 8 bytes */
guint size = 8;
if (asfmux->prop_packet_size > 65535)
size += 4;
else
size += 2;
if (asfmux->prop_padding > 65535)
size += 4;
else
size += 2;
return size;
}
/**
* gst_asf_mux_start_file:
* @asfmux: #GstAsfMux
*
* Starts the asf file/stream by creating and pushing
* the headers downstream.
*/
static GstFlowReturn
gst_asf_mux_start_file (GstAsfMux * asfmux)
{
GstBuffer *buf = NULL;
guint8 *bufdata = NULL;
GSList *walk;
guint stream_num = g_slist_length (asfmux->collect->data);
guint metadata_obj_size = 0;
GstAsfTags *asftags;
GValue streamheader = { 0 };
GstCaps *caps;
GstStructure *structure;
guint64 padding = asfmux->prop_padding;
GstSegment segment;
GstMapInfo map;
gsize bufsize;
gchar s_id[32];
if (padding < ASF_PADDING_OBJECT_SIZE)
padding = 0;
/* if not streaming, check if downstream is seekable */
if (!asfmux->prop_streamable) {
gboolean seekable;
GstQuery *query;
query = gst_query_new_seeking (GST_FORMAT_BYTES);
if (gst_pad_peer_query (asfmux->srcpad, query)) {
gst_query_parse_seeking (query, NULL, &seekable, NULL, NULL);
GST_INFO_OBJECT (asfmux, "downstream is %sseekable",
seekable ? "" : "not ");
} else {
/* assume seeking is not supported if query not handled downstream */
GST_WARNING_OBJECT (asfmux, "downstream did not handle seeking query");
seekable = FALSE;
}
if (!seekable) {
asfmux->prop_streamable = TRUE;
g_object_notify (G_OBJECT (asfmux), "streamable");
GST_WARNING_OBJECT (asfmux, "downstream is not seekable, but "
"streamable=false. Will ignore that and create streamable output "
"instead");
}
gst_query_unref (query);
}
/* from this point we started writing the headers */
GST_INFO_OBJECT (asfmux, "Writing headers");
asfmux->state = GST_ASF_MUX_STATE_HEADERS;
/* stream-start (FIXME: create id based on input ids) */
g_snprintf (s_id, sizeof (s_id), "asfmux-%08x", g_random_int ());
gst_pad_push_event (asfmux->srcpad, gst_event_new_stream_start (s_id));
caps = gst_pad_get_pad_template_caps (asfmux->srcpad);
gst_pad_set_caps (asfmux->srcpad, caps);
gst_caps_unref (caps);
/* send a BYTE format segment if we're going to seek to fix up the headers
* later, otherwise send a TIME segment */
if (asfmux->prop_streamable)
gst_segment_init (&segment, GST_FORMAT_TIME);
else
gst_segment_init (&segment, GST_FORMAT_BYTES);
gst_pad_push_event (asfmux->srcpad, gst_event_new_segment (&segment));
gst_asf_generate_file_id (&asfmux->file_id);
/* Get the metadata for content description object.
* We store our own taglist because it might get changed from now
* to the time we actually add its contents to the file, changing
* the size of the data we already calculated here.
*/
asftags = g_new0 (GstAsfTags, 1);
gst_asf_mux_get_content_description_tags (asfmux, asftags);
/* get the total metadata objects size */
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
metadata_obj_size += gst_asf_mux_get_metadata_object_size (asfmux,
(GstAsfPad *) walk->data);
}
/* alloc a buffer for all header objects */
buf = gst_buffer_new_and_alloc (gst_asf_mux_get_headers_size (asfmux) +
asftags->cont_desc_size +
asftags->ext_cont_desc_size +
metadata_obj_size + padding + ASF_DATA_OBJECT_SIZE);
gst_buffer_map (buf, &map, GST_MAP_WRITE);
bufdata = map.data;
bufsize = map.size;
gst_asf_mux_write_header_object (asfmux, &bufdata, map.size -
ASF_DATA_OBJECT_SIZE, 2 + stream_num);
/* get the position of the file properties object for
* updating it in gst_asf_mux_stop_file */
asfmux->file_properties_object_position = bufdata - map.data;
gst_asf_mux_write_file_properties (asfmux, &bufdata);
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
gst_asf_mux_write_stream_properties (asfmux, &bufdata,
(GstAsfPad *) walk->data);
}
if (asftags->cont_desc_size) {
gst_asf_mux_write_content_description (asfmux, &bufdata, asftags->tags);
}
if (asftags->ext_cont_desc_size) {
gst_asf_mux_write_ext_content_description (asfmux, &bufdata, asftags->tags);
}
if (asftags->tags)
gst_tag_list_unref (asftags->tags);
g_free (asftags);
/* writing header extension objects */
gst_asf_mux_write_header_extension (asfmux, &bufdata, stream_num *
ASF_EXTENDED_STREAM_PROPERTIES_OBJECT_SIZE + metadata_obj_size);
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
gst_asf_mux_write_extended_stream_properties (asfmux, &bufdata,
(GstAsfPad *) walk->data);
}
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
gst_asf_mux_write_metadata_object (asfmux, &bufdata,
(GstAsfPad *) walk->data);
}
gst_asf_mux_write_padding_object (asfmux, &bufdata, padding);
/* store data object position for later updating some fields */
asfmux->data_object_position = bufdata - map.data;
gst_asf_mux_write_data_object (asfmux, &bufdata);
/* set streamheader in source pad if 'streamable' */
if (asfmux->prop_streamable) {
g_value_init (&streamheader, GST_TYPE_ARRAY);
gst_asf_mux_put_buffer_in_streamheader (&streamheader, buf);
caps = gst_pad_get_current_caps (asfmux->srcpad);
caps = gst_caps_make_writable (caps);
structure = gst_caps_get_structure (caps, 0);
gst_structure_set_value (structure, "streamheader", &streamheader);
gst_pad_set_caps (asfmux->srcpad, caps);
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_HEADER);
g_value_unset (&streamheader);
gst_caps_unref (caps);
}
g_assert (bufdata - map.data == map.size);
gst_buffer_unmap (buf, &map);
return gst_asf_mux_push_buffer (asfmux, buf, bufsize);
}
/**
* gst_asf_mux_add_simple_index_entry:
* @asfmux:
* @videopad:
*
* Adds a new entry to the simple index of the stream handler by videopad.
* This functions doesn't check if the time elapsed
* is larger than the established time interval between entries. The caller
* is responsible for verifying this.
*/
static void
gst_asf_mux_add_simple_index_entry (GstAsfMux * asfmux,
GstAsfVideoPad * videopad)
{
SimpleIndexEntry *entry = NULL;
GST_DEBUG_OBJECT (asfmux, "Adding new simple index entry "
"packet number: %" G_GUINT32_FORMAT ", "
"packet count: %" G_GUINT16_FORMAT,
videopad->last_keyframe_packet, videopad->last_keyframe_packet_count);
entry = g_malloc0 (sizeof (SimpleIndexEntry));
entry->packet_number = videopad->last_keyframe_packet;
entry->packet_count = videopad->last_keyframe_packet_count;
if (entry->packet_count > videopad->max_keyframe_packet_count)
videopad->max_keyframe_packet_count = entry->packet_count;
videopad->simple_index = g_slist_append (videopad->simple_index, entry);
}
/**
* gst_asf_mux_send_packet:
* @asfmux:
* @buf: The asf data packet
*
* Pushes an asf data packet downstream. The total number
* of packets and bytes of the stream are incremented.
*
* Returns: the result of pushing the buffer downstream
*/
static GstFlowReturn
gst_asf_mux_send_packet (GstAsfMux * asfmux, GstBuffer * buf, gsize bufsize)
{
g_assert (bufsize == asfmux->packet_size);
asfmux->total_data_packets++;
GST_LOG_OBJECT (asfmux,
"Pushing a packet of size %" G_GSIZE_FORMAT " and timestamp %"
G_GUINT64_FORMAT, bufsize, GST_BUFFER_TIMESTAMP (buf));
GST_LOG_OBJECT (asfmux, "Total data packets: %" G_GUINT64_FORMAT,
asfmux->total_data_packets);
return gst_asf_mux_push_buffer (asfmux, buf, bufsize);
}
/**
* gst_asf_mux_flush_payloads:
* @asfmux: #GstAsfMux to flush the payloads from
*
* Fills an asf packet with asfmux queued payloads and
* pushes it downstream.
*
* Returns: The result of pushing the packet
*/
static GstFlowReturn
gst_asf_mux_flush_payloads (GstAsfMux * asfmux)
{
GstBuffer *buf;
guint8 payloads_count = 0; /* we only use 6 bits, max is 63 */
guint i;
GstClockTime send_ts = GST_CLOCK_TIME_NONE;
guint64 size_left;
guint8 *data;
gsize size;
GSList *walk;
GstAsfPad *pad;
gboolean has_keyframe;
AsfPayload *payload;
guint32 payload_size;
guint offset;
GstMapInfo map;
if (asfmux->payloads == NULL)
return GST_FLOW_OK; /* nothing to send is ok */
GST_LOG_OBJECT (asfmux, "Flushing payloads");
buf = gst_buffer_new_and_alloc (asfmux->packet_size);
gst_buffer_map (buf, &map, GST_MAP_WRITE);
memset (map.data, 0, asfmux->packet_size);
/* 1 for the multiple payload flags */
data = map.data + asfmux->payload_parsing_info_size + 1;
size_left = asfmux->packet_size - asfmux->payload_parsing_info_size - 1;
has_keyframe = FALSE;
walk = asfmux->payloads;
while (walk && payloads_count < MAX_PAYLOADS_IN_A_PACKET) {
payload = (AsfPayload *) walk->data;
pad = (GstAsfPad *) payload->pad;
payload_size = gst_asf_payload_get_size (payload);
if (size_left < payload_size) {
break; /* next payload doesn't fit fully */
}
if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (send_ts))) {
send_ts = GST_BUFFER_TIMESTAMP (payload->data);
}
/* adding new simple index entry (if needed) */
if (!pad->is_audio
&& GST_CLOCK_TIME_IS_VALID (GST_BUFFER_TIMESTAMP (payload->data))) {
GstAsfVideoPad *videopad = (GstAsfVideoPad *) pad;
if (videopad->has_keyframe) {
for (; videopad->next_index_time <=
ASF_MILI_TO_100NANO (payload->presentation_time);
videopad->next_index_time += videopad->time_interval) {
gst_asf_mux_add_simple_index_entry (asfmux, videopad);
}
}
}
/* serialize our payload */
GST_DEBUG_OBJECT (asfmux, "Serializing payload into packet");
GST_DEBUG_OBJECT (asfmux, "stream number: %d", pad->stream_number & 0x7F);
GST_DEBUG_OBJECT (asfmux, "media object number: %d",
(gint) payload->media_obj_num);
GST_DEBUG_OBJECT (asfmux, "offset into media object: %" G_GUINT32_FORMAT,
payload->offset_in_media_obj);
GST_DEBUG_OBJECT (asfmux, "media object size: %" G_GUINT32_FORMAT,
payload->media_object_size);
GST_DEBUG_OBJECT (asfmux, "replicated data length: %d",
(gint) payload->replicated_data_length);
GST_DEBUG_OBJECT (asfmux, "payload size: %" G_GSIZE_FORMAT,
gst_buffer_get_size (payload->data));
GST_DEBUG_OBJECT (asfmux, "presentation time: %" G_GUINT32_FORMAT " (%"
GST_TIME_FORMAT ")", payload->presentation_time,
GST_TIME_ARGS (payload->presentation_time * GST_MSECOND));
GST_DEBUG_OBJECT (asfmux, "keyframe: %s",
(payload->stream_number & 0x80 ? "yes" : "no"));
GST_DEBUG_OBJECT (asfmux, "buffer timestamp: %" GST_TIME_FORMAT,
GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (payload->data)));
GST_DEBUG_OBJECT (asfmux, "buffer duration %" GST_TIME_FORMAT,
GST_TIME_ARGS (GST_BUFFER_DURATION (payload->data)));
gst_asf_put_payload (data, payload);
if (!payload->has_packet_info) {
payload->has_packet_info = TRUE;
payload->packet_number = asfmux->total_data_packets;
}
GST_DEBUG_OBJECT (asfmux, "packet number: %" G_GUINT32_FORMAT,
payload->packet_number);
if (ASF_PAYLOAD_IS_KEYFRAME (payload)) {
has_keyframe = TRUE;
if (!pad->is_audio) {
GstAsfVideoPad *videopad = (GstAsfVideoPad *) pad;
videopad->last_keyframe_packet = payload->packet_number;
videopad->last_keyframe_packet_count = payload->packet_count;
videopad->has_keyframe = TRUE;
}
}
/* update our variables */
data += payload_size;
size_left -= payload_size;
payloads_count++;
walk = g_slist_next (walk);
}
/* remove flushed payloads */
GST_LOG_OBJECT (asfmux, "Freeing already used payloads");
for (i = 0; i < payloads_count; i++) {
GSList *aux = g_slist_nth (asfmux->payloads, 0);
AsfPayload *payload;
g_assert (aux);
payload = (AsfPayload *) aux->data;
asfmux->payloads = g_slist_remove (asfmux->payloads, payload);
asfmux->payload_data_size -=
(gst_buffer_get_size (payload->data) +
ASF_MULTIPLE_PAYLOAD_HEADER_SIZE);
gst_asf_payload_free (payload);
}
/* check if we can add part of the next payload */
if (asfmux->payloads && size_left > ASF_MULTIPLE_PAYLOAD_HEADER_SIZE) {
AsfPayload *payload =
(AsfPayload *) g_slist_nth (asfmux->payloads, 0)->data;
guint16 bytes_writen;
GST_DEBUG_OBJECT (asfmux, "Adding part of a payload to a packet");
if (ASF_PAYLOAD_IS_KEYFRAME (payload))
has_keyframe = TRUE;
if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (send_ts))) {
send_ts = GST_BUFFER_TIMESTAMP (payload->data);
}
bytes_writen = gst_asf_put_subpayload (data, payload, size_left);
if (!payload->has_packet_info) {
payload->has_packet_info = TRUE;
payload->packet_number = asfmux->total_data_packets;
}
asfmux->payload_data_size -= bytes_writen;
size_left -= (bytes_writen + ASF_MULTIPLE_PAYLOAD_HEADER_SIZE);
payloads_count++;
}
GST_LOG_OBJECT (asfmux, "Payload data size: %" G_GUINT32_FORMAT,
asfmux->payload_data_size);
/* fill payload parsing info */
data = map.data;
size = map.size;
/* flags */
GST_WRITE_UINT8 (data, (0x0 << 7) | /* no error correction */
(ASF_FIELD_TYPE_DWORD << 5) | /* packet length type */
(ASF_FIELD_TYPE_DWORD << 3) | /* padding length type */
(ASF_FIELD_TYPE_NONE << 1) | /* sequence type type */
0x1); /* multiple payloads */
offset = 1;
/* property flags - according to the spec, this should not change */
GST_WRITE_UINT8 (data + offset, (ASF_FIELD_TYPE_BYTE << 6) | /* stream number length type */
(ASF_FIELD_TYPE_BYTE << 4) | /* media obj number length type */
(ASF_FIELD_TYPE_DWORD << 2) | /* offset info media object length type */
(ASF_FIELD_TYPE_BYTE)); /* replicated data length type */
offset++;
/* Due to a limitation in WMP while streaming through WMSP we reduce the
* packet & padding size to 16bit if they are <= 65535 bytes
*/
if (asfmux->packet_size > 65535) {
GST_WRITE_UINT32_LE (data + offset, asfmux->packet_size - size_left);
offset += 4;
} else {
*data &= ~(ASF_FIELD_TYPE_MASK << 5);
*data |= ASF_FIELD_TYPE_WORD << 5;
GST_WRITE_UINT16_LE (data + offset, asfmux->packet_size - size_left);
offset += 2;
}
if (asfmux->prop_padding > 65535) {
GST_WRITE_UINT32_LE (data + offset, size_left);
offset += 4;
} else {
*data &= ~(ASF_FIELD_TYPE_MASK << 3);
*data |= ASF_FIELD_TYPE_WORD << 3;
GST_WRITE_UINT16_LE (data + offset, size_left);
offset += 2;
}
/* packet send time */
if (GST_CLOCK_TIME_IS_VALID (send_ts)) {
GST_WRITE_UINT32_LE (data + offset, (send_ts / GST_MSECOND));
GST_BUFFER_TIMESTAMP (buf) = send_ts;
}
offset += 4;
/* packet duration */
GST_WRITE_UINT16_LE (data + offset, 0); /* FIXME send duration needs to be estimated */
offset += 2;
/* multiple payloads flags */
GST_WRITE_UINT8 (data + offset, 0x2 << 6 | payloads_count);
gst_buffer_unmap (buf, &map);
if (payloads_count == 0) {
GST_WARNING_OBJECT (asfmux, "Sending packet without any payload");
}
asfmux->data_object_size += size;
if (!has_keyframe)
GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT);
return gst_asf_mux_send_packet (asfmux, buf, size);
}
/**
* stream_number_compare:
* @a: a #GstAsfPad
* @b: another #GstAsfPad
*
* Utility function to compare #GstAsfPad by their stream numbers
*
* Returns: The difference between their stream numbers
*/
static gint
stream_number_compare (gconstpointer a, gconstpointer b)
{
GstAsfPad *pad_a = (GstAsfPad *) a;
GstAsfPad *pad_b = (GstAsfPad *) b;
return pad_b->stream_number - pad_a->stream_number;
}
static GstFlowReturn
gst_asf_mux_push_simple_index (GstAsfMux * asfmux, GstAsfVideoPad * pad)
{
guint64 object_size = ASF_SIMPLE_INDEX_OBJECT_SIZE +
g_slist_length (pad->simple_index) * ASF_SIMPLE_INDEX_ENTRY_SIZE;
GstBuffer *buf;
GSList *walk;
guint8 *data;
guint32 entries_count = g_slist_length (pad->simple_index);
GstMapInfo map;
gsize bufsize;
buf = gst_buffer_new_and_alloc (object_size);
bufsize = object_size;
gst_buffer_map (buf, &map, GST_MAP_WRITE);
data = map.data;
gst_asf_put_guid (data, guids[ASF_SIMPLE_INDEX_OBJECT_INDEX]);
GST_WRITE_UINT64_LE (data + 16, object_size);
gst_asf_put_guid (data + 24, asfmux->file_id);
GST_WRITE_UINT64_LE (data + 40, pad->time_interval);
GST_WRITE_UINT32_LE (data + 48, pad->max_keyframe_packet_count);
GST_WRITE_UINT32_LE (data + 52, entries_count);
data += ASF_SIMPLE_INDEX_OBJECT_SIZE;
GST_DEBUG_OBJECT (asfmux,
"Simple index object values - size:%" G_GUINT64_FORMAT ", time interval:%"
G_GUINT64_FORMAT ", max packet count:%" G_GUINT32_FORMAT ", entries:%"
G_GUINT32_FORMAT, object_size, pad->time_interval,
pad->max_keyframe_packet_count, entries_count);
for (walk = pad->simple_index; walk; walk = g_slist_next (walk)) {
SimpleIndexEntry *entry = (SimpleIndexEntry *) walk->data;
GST_DEBUG_OBJECT (asfmux, "Simple index entry: packet_number:%"
G_GUINT32_FORMAT " packet_count:%" G_GUINT16_FORMAT,
entry->packet_number, entry->packet_count);
GST_WRITE_UINT32_LE (data, entry->packet_number);
GST_WRITE_UINT16_LE (data + 4, entry->packet_count);
data += ASF_SIMPLE_INDEX_ENTRY_SIZE;
}
GST_DEBUG_OBJECT (asfmux, "Pushing the simple index");
g_assert (data - map.data == object_size);
gst_buffer_unmap (buf, &map);
return gst_asf_mux_push_buffer (asfmux, buf, bufsize);
}
static GstFlowReturn
gst_asf_mux_write_indexes (GstAsfMux * asfmux)
{
GSList *ordered_pads;
GSList *walker;
GstFlowReturn ret = GST_FLOW_OK;
/* write simple indexes for video medias */
ordered_pads =
g_slist_sort (g_slist_copy (asfmux->collect->data),
(GCompareFunc) stream_number_compare);
for (walker = ordered_pads; walker; walker = g_slist_next (walker)) {
GstAsfPad *pad = (GstAsfPad *) walker->data;
if (!pad->is_audio) {
ret = gst_asf_mux_push_simple_index (asfmux, (GstAsfVideoPad *) pad);
if (ret != GST_FLOW_OK) {
GST_ERROR_OBJECT (asfmux, "Failed to write simple index for stream %"
G_GUINT16_FORMAT, (guint16) pad->stream_number);
goto cleanup_and_return;
}
}
}
cleanup_and_return:
g_slist_free (ordered_pads);
return ret;
}
/**
* gst_asf_mux_stop_file:
* @asfmux: #GstAsfMux
*
* Finalizes the asf stream by pushing the indexes after
* the data object. Also seeks back to the header positions
* to rewrite some fields such as the total number of bytes
* of the file, or any other that couldn't be predicted/known
* back on the header generation.
*
* Returns: GST_FLOW_OK on success
*/
static GstFlowReturn
gst_asf_mux_stop_file (GstAsfMux * asfmux)
{
GstEvent *event;
GstBuffer *buf;
GstFlowReturn ret = GST_FLOW_OK;
GSList *walk;
GstClockTime play_duration = 0;
guint32 bitrate = 0;
GstSegment segment;
GstMapInfo map;
guint8 *data;
/* write indexes */
ret = gst_asf_mux_write_indexes (asfmux);
if (ret != GST_FLOW_OK) {
GST_ERROR_OBJECT (asfmux, "Failed to write indexes");
return ret;
}
/* find max stream duration and bitrate */
for (walk = asfmux->collect->data; walk; walk = g_slist_next (walk)) {
GstAsfPad *pad = (GstAsfPad *) walk->data;
bitrate += pad->bitrate;
if (pad->play_duration > play_duration)
play_duration = pad->play_duration;
}
/* going back to file properties object to fill in
* values we didn't know back then */
GST_DEBUG_OBJECT (asfmux,
"Sending new segment to file properties object position");
gst_segment_init (&segment, GST_FORMAT_BYTES);
segment.start = segment.position =
asfmux->file_properties_object_position + 40;
event = gst_event_new_segment (&segment);
if (!gst_pad_push_event (asfmux->srcpad, event)) {
GST_ERROR_OBJECT (asfmux, "Failed to update file properties object");
return GST_FLOW_ERROR;
}
/* All file properties fields except the first 40 bytes */
buf = gst_buffer_new_and_alloc (ASF_FILE_PROPERTIES_OBJECT_SIZE - 40);
gst_buffer_map (buf, &map, GST_MAP_WRITE);
data = map.data;
GST_WRITE_UINT64_LE (data, asfmux->file_size);
gst_asf_put_time (data + 8, gst_asf_get_current_time ());
GST_WRITE_UINT64_LE (data + 16, asfmux->total_data_packets);
GST_WRITE_UINT64_LE (data + 24, (play_duration / 100) +
ASF_MILI_TO_100NANO (asfmux->preroll));
GST_WRITE_UINT64_LE (data + 32, (play_duration / 100)); /* TODO send duration */
/* if play duration is smaller then preroll, player might have problems */
if (asfmux->preroll > play_duration / GST_MSECOND) {
GST_ELEMENT_WARNING (asfmux, STREAM, MUX, (_("Generated file has a larger"
" preroll time than its streams duration")),
("Preroll time larger than streams duration, "
"try setting a smaller preroll value next time"));
}
GST_WRITE_UINT64_LE (data + 40, asfmux->preroll);
GST_WRITE_UINT32_LE (data + 48, 0x2); /* flags - seekable */
GST_WRITE_UINT32_LE (data + 52, asfmux->packet_size);
GST_WRITE_UINT32_LE (data + 56, asfmux->packet_size);
/* FIXME - we want the max instantaneous bitrate, for vbr streams, we can't
* get it this way, this would be the average, right? */
GST_WRITE_UINT32_LE (data + 60, bitrate); /* max bitrate */
gst_buffer_unmap (buf, &map);
/* we don't use gst_asf_mux_push_buffer because we are overwriting
* already sent data */
ret = gst_pad_push (asfmux->srcpad, buf);
if (ret != GST_FLOW_OK) {
GST_ERROR_OBJECT (asfmux, "Failed to update file properties object");
return ret;
}
GST_DEBUG_OBJECT (asfmux, "Seeking back to data object");
/* seek back to the data object */
segment.start = segment.position = asfmux->data_object_position + 16;
event = gst_event_new_segment (&segment);
if (!gst_pad_push_event (asfmux->srcpad, event)) {
GST_ERROR_OBJECT (asfmux, "Seek to update data object failed");
return GST_FLOW_ERROR;
}
buf = gst_buffer_new_and_alloc (32); /* qword+guid+qword */
gst_buffer_map (buf, &map, GST_MAP_WRITE);
data = map.data;
GST_WRITE_UINT64_LE (data, asfmux->data_object_size + ASF_DATA_OBJECT_SIZE);
gst_asf_put_guid (data + 8, asfmux->file_id);
GST_WRITE_UINT64_LE (data + 24, asfmux->total_data_packets);
gst_buffer_unmap (buf, &map);
return gst_pad_push (asfmux->srcpad, buf);
}
/**
* gst_asf_mux_process_buffer:
* @asfmux:
* @pad: stream of the buffer
* @buf: The buffer to be processed
*
* Processes the buffer by parsing it and
* queueing it up as an asf payload for later
* being added and pushed inside an asf packet.
*
* Returns: a #GstFlowReturn
*/
static GstFlowReturn
gst_asf_mux_process_buffer (GstAsfMux * asfmux, GstAsfPad * pad,
GstBuffer * buf)
{
guint8 keyframe;
AsfPayload *payload;
payload = g_malloc0 (sizeof (AsfPayload));
payload->pad = (GstCollectData *) pad;
payload->data = buf;
GST_LOG_OBJECT (asfmux, "Processing payload data for stream number %u",
pad->stream_number);
/* stream number */
if (GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT)) {
keyframe = 0;
} else {
keyframe = 0x1 << 7;
}
payload->stream_number = keyframe | pad->stream_number;
payload->media_obj_num = pad->media_object_number;
payload->offset_in_media_obj = 0;
payload->replicated_data_length = 8;
/* replicated data - 1) media object size */
payload->media_object_size = gst_buffer_get_size (buf);
/* replicated data - 2) presentation time */
if (!GST_CLOCK_TIME_IS_VALID (GST_BUFFER_TIMESTAMP (buf))) {
GST_ERROR_OBJECT (asfmux, "Received buffer without timestamp");
gst_asf_payload_free (payload);
return GST_FLOW_ERROR;
}
g_assert (GST_CLOCK_TIME_IS_VALID (asfmux->first_ts));
g_assert (GST_CLOCK_TIME_IS_VALID (pad->first_ts));
payload->presentation_time = asfmux->preroll +
((GST_BUFFER_TIMESTAMP (buf) - asfmux->first_ts) / GST_MSECOND);
/* update counting values */
pad->media_object_number = (pad->media_object_number + 1) % 256;
if (GST_BUFFER_DURATION (buf) != GST_CLOCK_TIME_NONE) {
pad->play_duration += GST_BUFFER_DURATION (buf);
} else {
GST_WARNING_OBJECT (asfmux, "Received buffer without duration, it will not "
"be accounted in the total file time");
}
asfmux->payloads = g_slist_append (asfmux->payloads, payload);
asfmux->payload_data_size +=
gst_buffer_get_size (buf) + ASF_MULTIPLE_PAYLOAD_HEADER_SIZE;
GST_LOG_OBJECT (asfmux, "Payload data size: %" G_GUINT32_FORMAT,
asfmux->payload_data_size);
while (asfmux->payload_data_size + asfmux->payload_parsing_info_size >=
asfmux->packet_size) {
GstFlowReturn ret = gst_asf_mux_flush_payloads (asfmux);
if (ret != GST_FLOW_OK)
return ret;
}
return GST_FLOW_OK;
}
static GstFlowReturn
gst_asf_mux_collected (GstCollectPads * collect, gpointer data)
{
GstAsfMux *asfmux = GST_ASF_MUX_CAST (data);
GstFlowReturn ret = GST_FLOW_OK;
GstAsfPad *best_pad = NULL;
GstClockTime best_time = GST_CLOCK_TIME_NONE;
GstBuffer *buf = NULL;
GSList *walk;
if (G_UNLIKELY (asfmux->state == GST_ASF_MUX_STATE_NONE)) {
ret = gst_asf_mux_start_file (asfmux);
if (ret != GST_FLOW_OK) {
GST_WARNING_OBJECT (asfmux, "Failed to send headers");
return ret;
} else {
asfmux->state = GST_ASF_MUX_STATE_DATA;
}
}
if (G_UNLIKELY (asfmux->state == GST_ASF_MUX_STATE_EOS))
return GST_FLOW_EOS;
/* select the earliest buffer */
walk = asfmux->collect->data;
while (walk) {
GstAsfPad *pad;
GstCollectData *data;
GstClockTime time;
data = (GstCollectData *) walk->data;
pad = (GstAsfPad *) data;
walk = g_slist_next (walk);
buf = gst_collect_pads_peek (collect, data);
if (buf == NULL) {
GST_LOG_OBJECT (asfmux, "Pad %s has no buffers",
GST_PAD_NAME (pad->collect.pad));
continue;
}
time = GST_BUFFER_TIMESTAMP (buf);
/* check the ts for getting the first time */
if (!GST_CLOCK_TIME_IS_VALID (pad->first_ts) &&
GST_CLOCK_TIME_IS_VALID (time)) {
GST_DEBUG_OBJECT (asfmux,
"First ts for stream number %u: %" GST_TIME_FORMAT,
pad->stream_number, GST_TIME_ARGS (time));
pad->first_ts = time;
if (!GST_CLOCK_TIME_IS_VALID (asfmux->first_ts) ||
time < asfmux->first_ts) {
GST_DEBUG_OBJECT (asfmux, "New first ts for file %" GST_TIME_FORMAT,
GST_TIME_ARGS (time));
asfmux->first_ts = time;
}
}
gst_buffer_unref (buf);
if (best_pad == NULL || !GST_CLOCK_TIME_IS_VALID (time) ||
(GST_CLOCK_TIME_IS_VALID (best_time) && time < best_time)) {
best_pad = pad;
best_time = time;
}
}
if (best_pad != NULL) {
/* we have data */
GST_LOG_OBJECT (asfmux, "selected pad %s with time %" GST_TIME_FORMAT,
GST_PAD_NAME (best_pad->collect.pad), GST_TIME_ARGS (best_time));
buf = gst_collect_pads_pop (collect, &best_pad->collect);
ret = gst_asf_mux_process_buffer (asfmux, best_pad, buf);
} else {
/* no data, let's finish it up */
while (asfmux->payloads) {
ret = gst_asf_mux_flush_payloads (asfmux);
if (ret != GST_FLOW_OK) {
return ret;
}
}
g_assert (asfmux->payloads == NULL);
g_assert (asfmux->payload_data_size == 0);
/* in not on 'streamable' mode we need to push indexes
* and update headers */
if (!asfmux->prop_streamable) {
ret = gst_asf_mux_stop_file (asfmux);
}
if (ret == GST_FLOW_OK) {
gst_pad_push_event (asfmux->srcpad, gst_event_new_eos ());
ret = GST_FLOW_EOS;
}
asfmux->state = GST_ASF_MUX_STATE_EOS;
}
return ret;
}
static void
gst_asf_mux_pad_reset (GstAsfPad * pad)
{
pad->stream_number = 0;
pad->media_object_number = 0;
pad->play_duration = (GstClockTime) 0;
pad->bitrate = 0;
if (pad->codec_data)
gst_buffer_unref (pad->codec_data);
pad->codec_data = NULL;
if (pad->taglist)
gst_tag_list_unref (pad->taglist);
pad->taglist = NULL;
pad->first_ts = GST_CLOCK_TIME_NONE;
if (pad->is_audio) {
GstAsfAudioPad *audiopad = (GstAsfAudioPad *) pad;
audiopad->audioinfo.rate = 0;
audiopad->audioinfo.channels = 0;
audiopad->audioinfo.format = 0;
audiopad->audioinfo.av_bps = 0;
audiopad->audioinfo.blockalign = 0;
audiopad->audioinfo.bits_per_sample = 0;
} else {
GstAsfVideoPad *videopad = (GstAsfVideoPad *) pad;
videopad->vidinfo.size = 0;
videopad->vidinfo.width = 0;
videopad->vidinfo.height = 0;
videopad->vidinfo.planes = 1;
videopad->vidinfo.bit_cnt = 0;
videopad->vidinfo.compression = 0;
videopad->vidinfo.image_size = 0;
videopad->vidinfo.xpels_meter = 0;
videopad->vidinfo.ypels_meter = 0;
videopad->vidinfo.num_colors = 0;
videopad->vidinfo.imp_colors = 0;
videopad->last_keyframe_packet = 0;
videopad->has_keyframe = FALSE;
videopad->last_keyframe_packet_count = 0;
videopad->max_keyframe_packet_count = 0;
videopad->next_index_time = 0;
videopad->time_interval = DEFAULT_SIMPLE_INDEX_TIME_INTERVAL;
if (videopad->simple_index) {
GSList *walk;
for (walk = videopad->simple_index; walk; walk = g_slist_next (walk)) {
g_free (walk->data);
walk->data = NULL;
}
g_slist_free (videopad->simple_index);
}
videopad->simple_index = NULL;
}
}
static gboolean
gst_asf_mux_audio_set_caps (GstPad * pad, GstCaps * caps)
{
GstAsfMux *asfmux;
GstAsfAudioPad *audiopad;
GstStructure *structure;
const gchar *caps_name;
gint channels, rate;
gchar *aux;
const GValue *codec_data;
asfmux = GST_ASF_MUX (gst_pad_get_parent (pad));
audiopad = (GstAsfAudioPad *) gst_pad_get_element_private (pad);
g_assert (audiopad);
aux = gst_caps_to_string (caps);
GST_DEBUG_OBJECT (asfmux, "%s:%s, caps=%s", GST_DEBUG_PAD_NAME (pad), aux);
g_free (aux);
structure = gst_caps_get_structure (caps, 0);
caps_name = gst_structure_get_name (structure);
if (!gst_structure_get_int (structure, "channels", &channels) ||
!gst_structure_get_int (structure, "rate", &rate))
goto refuse_caps;
audiopad->audioinfo.channels = (guint16) channels;
audiopad->audioinfo.rate = (guint32) rate;
/* taken from avimux
* codec initialization data, if any
*/
codec_data = gst_structure_get_value (structure, "codec_data");
if (codec_data) {
audiopad->pad.codec_data = gst_value_get_buffer (codec_data);
gst_buffer_ref (audiopad->pad.codec_data);
}
if (strcmp (caps_name, "audio/x-wma") == 0) {
gint version;
gint block_align = 0;
gint bitrate = 0;
if (!gst_structure_get_int (structure, "wmaversion", &version)) {
goto refuse_caps;
}
if (gst_structure_get_int (structure, "block_align", &block_align)) {
audiopad->audioinfo.blockalign = (guint16) block_align;
}
if (gst_structure_get_int (structure, "bitrate", &bitrate)) {
audiopad->pad.bitrate = (guint32) bitrate;
audiopad->audioinfo.av_bps = bitrate / 8;
}
if (version == 1) {
audiopad->audioinfo.format = GST_RIFF_WAVE_FORMAT_WMAV1;
} else if (version == 2) {
audiopad->audioinfo.format = GST_RIFF_WAVE_FORMAT_WMAV2;
} else if (version == 3) {
audiopad->audioinfo.format = GST_RIFF_WAVE_FORMAT_WMAV3;
} else {
goto refuse_caps;
}
} else if (strcmp (caps_name, "audio/mpeg") == 0) {
gint version;
gint layer;
if (!gst_structure_get_int (structure, "mpegversion", &version) ||
!gst_structure_get_int (structure, "layer", &layer)) {
goto refuse_caps;
}
if (version != 1 || layer != 3) {
goto refuse_caps;
}
audiopad->audioinfo.format = GST_RIFF_WAVE_FORMAT_MPEGL3;
} else {
goto refuse_caps;
}
gst_object_unref (asfmux);
return TRUE;
refuse_caps:
GST_WARNING_OBJECT (asfmux, "pad %s refused caps %" GST_PTR_FORMAT,
GST_PAD_NAME (pad), caps);
gst_object_unref (asfmux);
return FALSE;
}
/* TODO Read pixel aspect ratio */
static gboolean
gst_asf_mux_video_set_caps (GstPad * pad, GstCaps * caps)
{
GstAsfMux *asfmux;
GstAsfVideoPad *videopad;
GstStructure *structure;
const gchar *caps_name;
gint width, height;
gchar *aux;
const GValue *codec_data;
asfmux = GST_ASF_MUX (gst_pad_get_parent (pad));
videopad = (GstAsfVideoPad *) gst_pad_get_element_private (pad);
g_assert (videopad);
aux = gst_caps_to_string (caps);
GST_DEBUG_OBJECT (asfmux, "%s:%s, caps=%s", GST_DEBUG_PAD_NAME (pad), aux);
g_free (aux);
structure = gst_caps_get_structure (caps, 0);
caps_name = gst_structure_get_name (structure);
if (!gst_structure_get_int (structure, "width", &width) ||
!gst_structure_get_int (structure, "height", &height))
goto refuse_caps;
videopad->vidinfo.width = (gint32) width;
videopad->vidinfo.height = (gint32) height;
/* taken from avimux
* codec initialization data, if any
*/
codec_data = gst_structure_get_value (structure, "codec_data");
if (codec_data) {
videopad->pad.codec_data = gst_value_get_buffer (codec_data);
gst_buffer_ref (videopad->pad.codec_data);
}
if (strcmp (caps_name, "video/x-wmv") == 0) {
gint wmvversion;
const gchar *fstr;
videopad->vidinfo.bit_cnt = 24;
/* in case we have a format, we use it */
fstr = gst_structure_get_string (structure, "format");
if (fstr && strlen (fstr) == 4) {
videopad->vidinfo.compression = GST_STR_FOURCC (fstr);
} else if (gst_structure_get_int (structure, "wmvversion", &wmvversion)) {
if (wmvversion == 2) {
videopad->vidinfo.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '2');
} else if (wmvversion == 1) {
videopad->vidinfo.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '1');
} else if (wmvversion == 3) {
videopad->vidinfo.compression = GST_MAKE_FOURCC ('W', 'M', 'V', '3');
}
} else
goto refuse_caps;
} else {
goto refuse_caps;
}
gst_object_unref (asfmux);
return TRUE;
refuse_caps:
GST_WARNING_OBJECT (asfmux, "pad %s refused caps %" GST_PTR_FORMAT,
GST_PAD_NAME (pad), caps);
gst_object_unref (asfmux);
return FALSE;
}
static GstPad *
gst_asf_mux_request_new_pad (GstElement * element,
GstPadTemplate * templ, const gchar * req_name, const GstCaps * caps)
{
GstElementClass *klass = GST_ELEMENT_GET_CLASS (element);
GstAsfMux *asfmux = GST_ASF_MUX_CAST (element);
GstPad *newpad;
GstAsfPad *collect_pad;
gboolean is_audio;
guint collect_size = 0;
gchar *name = NULL;
const gchar *pad_name = NULL;
guint pad_id;
GST_DEBUG_OBJECT (asfmux, "Requested pad: %s", GST_STR_NULL (req_name));
if (asfmux->state != GST_ASF_MUX_STATE_NONE) {
GST_WARNING_OBJECT (asfmux, "Not providing request pad after element is at "
"paused/playing state.");
return NULL;
}
if (templ == gst_element_class_get_pad_template (klass, "audio_%u")) {
/* don't mix named and unnamed pads, if the pad already exists we fail when
* trying to add it */
if (req_name != NULL && sscanf (req_name, "audio_%u", &pad_id) == 1) {
pad_name = req_name;
} else {
name = g_strdup_printf ("audio_%u", asfmux->stream_number + 1);
pad_name = name;
}
GST_DEBUG_OBJECT (asfmux, "Adding new pad %s", name);
newpad = gst_pad_new_from_template (templ, pad_name);
is_audio = TRUE;
} else if (templ == gst_element_class_get_pad_template (klass, "video_%u")) {
/* don't mix named and unnamed pads, if the pad already exists we fail when
* trying to add it */
if (req_name != NULL && sscanf (req_name, "video_%u", &pad_id) == 1) {
pad_name = req_name;
} else {
name = g_strdup_printf ("video_%u", asfmux->stream_number + 1);
pad_name = name;
}
GST_DEBUG_OBJECT (asfmux, "Adding new pad %s", name);
newpad = gst_pad_new_from_template (templ, name);
is_audio = FALSE;
} else {
GST_WARNING_OBJECT (asfmux, "This is not our template!");
return NULL;
}
g_free (name);
/* add pad to collections */
if (is_audio) {
collect_size = sizeof (GstAsfAudioPad);
} else {
collect_size = sizeof (GstAsfVideoPad);
}
collect_pad = (GstAsfPad *)
gst_collect_pads_add_pad (asfmux->collect, newpad, collect_size,
(GstCollectDataDestroyNotify) (gst_asf_mux_pad_reset), TRUE);
/* set up pad */
collect_pad->is_audio = is_audio;
if (!is_audio)
((GstAsfVideoPad *) collect_pad)->simple_index = NULL;
collect_pad->taglist = NULL;
gst_asf_mux_pad_reset (collect_pad);
/* set pad stream number */
asfmux->stream_number += 1;
collect_pad->stream_number = asfmux->stream_number;
gst_pad_set_active (newpad, TRUE);
gst_element_add_pad (element, newpad);
return newpad;
}
static void
gst_asf_mux_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec)
{
GstAsfMux *asfmux;
asfmux = GST_ASF_MUX (object);
switch (prop_id) {
case PROP_PACKET_SIZE:
g_value_set_uint (value, asfmux->prop_packet_size);
break;
case PROP_PREROLL:
g_value_set_uint64 (value, asfmux->prop_preroll);
break;
case PROP_MERGE_STREAM_TAGS:
g_value_set_boolean (value, asfmux->prop_merge_stream_tags);
break;
case PROP_PADDING:
g_value_set_uint64 (value, asfmux->prop_padding);
break;
case PROP_STREAMABLE:
g_value_set_boolean (value, asfmux->prop_streamable);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static void
gst_asf_mux_set_property (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec)
{
GstAsfMux *asfmux;
asfmux = GST_ASF_MUX (object);
switch (prop_id) {
case PROP_PACKET_SIZE:
asfmux->prop_packet_size = g_value_get_uint (value);
break;
case PROP_PREROLL:
asfmux->prop_preroll = g_value_get_uint64 (value);
break;
case PROP_MERGE_STREAM_TAGS:
asfmux->prop_merge_stream_tags = g_value_get_boolean (value);
break;
case PROP_PADDING:
asfmux->prop_padding = g_value_get_uint64 (value);
break;
case PROP_STREAMABLE:
asfmux->prop_streamable = g_value_get_boolean (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static GstStateChangeReturn
gst_asf_mux_change_state (GstElement * element, GstStateChange transition)
{
GstAsfMux *asfmux;
GstStateChangeReturn ret;
asfmux = GST_ASF_MUX (element);
switch (transition) {
case GST_STATE_CHANGE_READY_TO_PAUSED:
/* TODO - check if it is possible to mux 2 files without going
* through here */
asfmux->payload_parsing_info_size =
gst_asf_mux_find_payload_parsing_info_size (asfmux);
asfmux->packet_size = asfmux->prop_packet_size;
asfmux->preroll = asfmux->prop_preroll;
asfmux->merge_stream_tags = asfmux->prop_merge_stream_tags;
gst_collect_pads_start (asfmux->collect);
break;
case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
break;
case GST_STATE_CHANGE_PAUSED_TO_READY:
gst_collect_pads_stop (asfmux->collect);
asfmux->state = GST_ASF_MUX_STATE_NONE;
break;
default:
break;
}
ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
if (ret == GST_STATE_CHANGE_FAILURE)
goto done;
switch (transition) {
case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
break;
case GST_STATE_CHANGE_PAUSED_TO_READY:
break;
case GST_STATE_CHANGE_READY_TO_NULL:
break;
default:
break;
}
done:
return ret;
}