gstreamer/gst/asfdemux/gstasfmux.c
Iain Holmes 32ec751c90 gstasfmux - updated for new changes
Original commit message from CVS:
gstasfmux - updated for new changes
2003-11-01 11:35:26 +00:00

1398 lines
41 KiB
C

/* ASF muxer plugin for GStreamer
* Copyright (C) 2003 Ronald Bultje <rbultje@ronald.bitfreak.net>
*
* 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., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
/* based on:
* - ffmpeg ASF muxer
* - Owen's GStreamer ASF demuxer ("just reverse it"(tm)?)
* - Some own random bits and bytes and stuffies and more
* - Grolsch (oh, and Heineken) beer
* - Borrelnootjes (and chips, and stuff)
* - Why are you reading this?
*
* "The best code is written when you're drunk.
* You'll just never understand it, too."
* -- truth hurts.
*/
/* stream does NOT work on Windows Media Player (does work on
* other (Linux-based) players, because we do not specify bitrate
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include <gst/video/video.h>
/* for audio codec IDs */
#include <gst/riff/riff.h>
#include "gstasfmux.h"
/* elementfactory information */
static GstElementDetails
gst_asfmux_details =
{
"Asf multiplexer",
"Codec/Muxer",
"Muxes audio and video streams into an asf stream",
"Ronald Bultje <rbultje@ronald.bitfreak.net>",
};
/* AsfMux signals and args */
enum {
/* FILL ME */
LAST_SIGNAL
};
enum {
ARG_0,
};
GST_PAD_TEMPLATE_FACTORY (src_factory,
"src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_CAPS_NEW (
"asfmux_src_video",
"video/x-ms-asf",
NULL
)
);
GST_PAD_TEMPLATE_FACTORY (video_sink_factory,
"video_%d",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_CAPS_NEW (
"asfmux_sink_video_yuv",
"video/x-raw-yuv",
"format", GST_PROPS_LIST (
GST_PROPS_FOURCC (GST_MAKE_FOURCC('Y','U','Y','2')),
GST_PROPS_FOURCC (GST_MAKE_FOURCC('I','4','2','0'))
),
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
),
GST_CAPS_NEW (
"asfmux_sink_video_jpeg",
"video/x-jpeg",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
),
GST_CAPS_NEW (
"asfmux_sink_video_divx",
"video/x-divx",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096),
"divxversion", GST_PROPS_INT_RANGE (3, 5)
),
GST_CAPS_NEW (
"asfmux_sink_video_xvid",
"video/x-xvid",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
),
GST_CAPS_NEW (
"asfmux_sink_video_3ivx",
"video/x-3ivx",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
),
GST_CAPS_NEW (
"asfmux_sink_video_msmpeg",
"video/x-msmpeg",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096),
"msmpegversion", GST_PROPS_INT_RANGE (41, 43)
),
GST_CAPS_NEW (
"asfmux_sink_video_mpeg",
"video/mpeg",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096),
"mpegversion", GST_PROPS_INT (1),
"systemstream", GST_PROPS_BOOLEAN (FALSE)
),
GST_CAPS_NEW (
"asfmux_sink_video_h263",
"video/x-h263",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
),
GST_CAPS_NEW (
"asfmux_sink_video_dv",
"video/x-dv",
"width", GST_PROPS_INT (720),
"height", GST_PROPS_LIST (
GST_PROPS_INT (576),
GST_PROPS_INT (480)
),
"systemstream", GST_PROPS_BOOLEAN (FALSE)
),
GST_CAPS_NEW (
"asfmux_sink_video_hfyu",
"video/x-huffyuv",
"width", GST_PROPS_INT_RANGE (16, 4096),
"height", GST_PROPS_INT_RANGE (16, 4096)
)
);
GST_PAD_TEMPLATE_FACTORY (audio_sink_factory,
"audio_%d",
GST_PAD_SINK,
GST_PAD_REQUEST,
GST_CAPS_NEW (
"asfmux_sink_audio_raw",
"audio/x-raw-int",
"endianness", GST_PROPS_INT (G_LITTLE_ENDIAN),
"signed", GST_PROPS_LIST (
GST_PROPS_BOOLEAN (TRUE),
GST_PROPS_BOOLEAN (FALSE)
),
"width", GST_PROPS_LIST (
GST_PROPS_INT (8),
GST_PROPS_INT (16)
),
"depth", GST_PROPS_LIST (
GST_PROPS_INT (8),
GST_PROPS_INT (16)
),
"rate", GST_PROPS_INT_RANGE (1000, 96000),
"channels", GST_PROPS_INT_RANGE (1, 2)
),
GST_CAPS_NEW (
"asfmux_sink_audio_mpeg",
"audio/mpeg",
"layer", GST_PROPS_INT_RANGE (1, 3),
"rate", GST_PROPS_INT_RANGE (1000, 96000),
"channels", GST_PROPS_INT_RANGE (1, 2)
),
GST_CAPS_NEW (
"asfmux_sink_audio_vorbis",
"audio/x-vorbis",
"rate", GST_PROPS_INT_RANGE (1000, 96000),
"channels", GST_PROPS_INT_RANGE (1, 2)
),
GST_CAPS_NEW (
"asfmux_sink_audio_ac3",
"audio/x-ac3",
"rate", GST_PROPS_INT_RANGE (1000, 96000),
"channels", GST_PROPS_INT_RANGE (1, 6)
)
);
#define GST_ASF_PACKET_SIZE 3200
#define GST_ASF_PACKET_HEADER_SIZE 12
#define GST_ASF_FRAME_HEADER_SIZE 17
static void gst_asfmux_base_init (gpointer g_class);
static void gst_asfmux_class_init (GstAsfMuxClass *klass);
static void gst_asfmux_init (GstAsfMux *asfmux);
static void gst_asfmux_loop (GstElement *element);
static gboolean gst_asfmux_handle_event (GstPad *pad,
GstEvent *event);
static GstPad* gst_asfmux_request_new_pad (GstElement *element,
GstPadTemplate *templ,
const gchar *name);
static GstElementStateReturn gst_asfmux_change_state (GstElement *element);
static GstElementClass *parent_class = NULL;
/*static guint gst_asfmux_signals[LAST_SIGNAL] = { 0 }; */
GType
gst_asfmux_get_type (void)
{
static GType asfmux_type = 0;
if (!asfmux_type) {
static const GTypeInfo asfmux_info = {
sizeof (GstAsfMuxClass),
gst_asfmux_base_init,
NULL,
(GClassInitFunc) gst_asfmux_class_init,
NULL,
NULL,
sizeof (GstAsfMux),
0,
(GInstanceInitFunc) gst_asfmux_init,
};
asfmux_type = g_type_register_static (GST_TYPE_ELEMENT,
"GstAsfMux",
&asfmux_info, 0);
}
return asfmux_type;
}
static void
gst_asfmux_base_init (gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
gst_element_class_add_pad_template (element_class,
GST_PAD_TEMPLATE_GET (src_factory));
gst_element_class_add_pad_template (element_class,
GST_PAD_TEMPLATE_GET (audio_sink_factory));
gst_element_class_add_pad_template (element_class,
GST_PAD_TEMPLATE_GET (video_sink_factory));
gst_element_class_set_details (element_class, &gst_asfmux_details);
}
static void
gst_asfmux_class_init (GstAsfMuxClass *klass)
{
GstElementClass *gstelement_class;
gstelement_class = (GstElementClass *) klass;
parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
gstelement_class->request_new_pad = gst_asfmux_request_new_pad;
gstelement_class->change_state = gst_asfmux_change_state;
}
static const GstEventMask *
gst_asfmux_get_event_masks (GstPad *pad)
{
static const GstEventMask gst_asfmux_sink_event_masks[] = {
{ GST_EVENT_EOS, 0 },
{ 0, }
};
return gst_asfmux_sink_event_masks;
}
static void
gst_asfmux_init (GstAsfMux *asfmux)
{
gint n;
asfmux->srcpad = gst_pad_new_from_template (
GST_PAD_TEMPLATE_GET (src_factory), "src");
gst_element_add_pad (GST_ELEMENT (asfmux), asfmux->srcpad);
GST_FLAG_SET (GST_ELEMENT (asfmux), GST_ELEMENT_EVENT_AWARE);
asfmux->num_outputs = asfmux->num_video = asfmux->num_audio = 0;
memset (&asfmux->output, 0, sizeof (asfmux->output));
for (n = 0; n < MAX_ASF_OUTPUTS; n++) {
asfmux->output[n].index = n;
asfmux->output[n].connected = FALSE;
}
asfmux->write_header = TRUE;
asfmux->packet = NULL;
asfmux->num_packets = 0;
asfmux->sequence = 0;
gst_element_set_loop_function (GST_ELEMENT (asfmux), gst_asfmux_loop);
}
static GstPadLinkReturn
gst_asfmux_vidsinkconnect (GstPad *pad, GstCaps *vscaps)
{
GstAsfMux *asfmux;
GstCaps *caps;
GstAsfMuxStream *stream = NULL;
gint n;
asfmux = GST_ASFMUX (gst_pad_get_parent (pad));
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].pad == pad) {
stream = &asfmux->output[n];
break;
}
}
g_assert (n < asfmux->num_outputs);
g_assert (stream != NULL);
g_assert (stream->type == ASF_STREAM_VIDEO);
/* we are not going to act on variable caps */
if (!GST_CAPS_IS_FIXED (vscaps))
return GST_PAD_LINK_DELAYED;
GST_DEBUG ("asfmux: video sinkconnect triggered on %s",
gst_pad_get_name (pad));
for (caps = vscaps; caps != NULL; caps = caps->next) {
const gchar* mimetype = gst_caps_get_mime(caps);
gint w, h;
/* global */
gst_caps_get (caps, "width", &w,
"height", &h,
NULL);
stream->header.video.stream.width = w;
stream->header.video.stream.height = h;
stream->header.video.stream.unknown = 2;
stream->header.video.stream.size = 40;
stream->bitrate = 0; /* TODO */
if (!strcmp (mimetype, "video/x-raw-yuv")) {
guint32 format;
gst_caps_get_fourcc_int (caps, "format", &format);
stream->header.video.format.tag = format;
switch (format) {
case GST_MAKE_FOURCC ('Y','U','Y','2'):
stream->header.video.format.depth = 16;
stream->header.video.format.planes = 1;
break;
case GST_MAKE_FOURCC ('I','4','2','0'):
stream->header.video.format.depth = 12;
stream->header.video.format.planes = 3;
break;
}
goto done;
} else {
stream->header.video.format.depth = 24;
stream->header.video.format.planes = 1;
stream->header.video.format.tag = 0;
/* find format */
if (!strcmp (mimetype, "video/x-huffyuv")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('H','F','Y','U');
} else if (!strcmp (mimetype, "video/x-jpeg")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('M','J','P','G');
} else if (!strcmp (mimetype, "video/x-divx")) {
gint divxversion;
gst_caps_get_int (caps, "divxversion", &divxversion);
switch (divxversion) {
case 3:
stream->header.video.format.tag = GST_MAKE_FOURCC ('D','I','V','3');
break;
case 4:
stream->header.video.format.tag = GST_MAKE_FOURCC ('D','I','V','X');
break;
case 5:
stream->header.video.format.tag = GST_MAKE_FOURCC ('D','X','5','0');
break;
}
} else if (!strcmp (mimetype, "video/x-xvid")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('X','V','I','D');
} else if (!strcmp (mimetype, "video/x-3ivx")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('3','I','V','2');
} else if (!strcmp (mimetype, "video/x-msmpeg")) {
gint msmpegversion;
gst_caps_get_int (caps, "msmpegversion", &msmpegversion);
switch (msmpegversion) {
case 41:
stream->header.video.format.tag = GST_MAKE_FOURCC ('M','P','G','4');
break;
case 42:
stream->header.video.format.tag = GST_MAKE_FOURCC ('M','P','4','2');
break;
case 43:
stream->header.video.format.tag = GST_MAKE_FOURCC ('M','P','4','3');
break;
}
} else if (!strcmp (mimetype, "video/x-dv")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('D','V','S','D');
} else if (!strcmp (mimetype, "video/x-h263")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('H','2','6','3');
} else if (!strcmp (mimetype, "video/mpeg")) {
stream->header.video.format.tag = GST_MAKE_FOURCC ('M','P','E','G');
}
if (!stream->header.video.format.tag) {
continue;
}
goto done;
}
}
return GST_PAD_LINK_REFUSED;
done:
stream->bitrate = 1024 * 1024;
stream->header.video.format.size = stream->header.video.stream.size;
stream->header.video.format.width = stream->header.video.stream.width;
stream->header.video.format.height = stream->header.video.stream.height;
stream->header.video.format.image_size = stream->header.video.stream.width *
stream->header.video.stream.height;
stream->header.video.format.xpels_meter = 0;
stream->header.video.format.ypels_meter = 0;
stream->header.video.format.num_colors = 0;
stream->header.video.format.imp_colors = 0;
return GST_PAD_LINK_OK;
}
static GstPadLinkReturn
gst_asfmux_audsinkconnect (GstPad *pad, GstCaps *vscaps)
{
GstAsfMux *asfmux;
GstCaps *caps;
GstAsfMuxStream *stream = NULL;
gint n;
asfmux = GST_ASFMUX (gst_pad_get_parent (pad));
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].pad == pad) {
stream = &asfmux->output[n];
break;
}
}
g_assert (n < asfmux->num_outputs);
g_assert (stream != NULL);
g_assert (stream->type == ASF_STREAM_AUDIO);
/* we are not going to act on variable caps */
if (!GST_CAPS_IS_FIXED (vscaps))
return GST_PAD_LINK_DELAYED;
GST_DEBUG ("asfmux: audio sinkconnect triggered on %s",
gst_pad_get_name (pad));
for (caps = vscaps; caps != NULL; caps = caps->next) {
const gchar* mimetype = gst_caps_get_mime(caps);
gint rate, channels;
/* we want these for all */
gst_caps_get (caps, "channels", &channels,
"rate", &rate,
NULL);
stream->header.audio.sample_rate = rate;
stream->header.audio.channels = channels;
if (!strcmp (mimetype, "audio/x-raw-int")) {
gint block, size;
stream->header.audio.codec_tag = GST_RIFF_WAVE_FORMAT_PCM;
gst_caps_get (caps, "width", &block,
"depth", &size,
NULL);
stream->header.audio.block_align = block;
stream->header.audio.word_size = size;
stream->header.audio.size = 0;
/* set some more info straight */
stream->header.audio.block_align /= 8;
stream->header.audio.block_align *= stream->header.audio.channels;
stream->header.audio.byte_rate = stream->header.audio.block_align *
stream->header.audio.sample_rate;
goto done;
} else {
stream->header.audio.codec_tag = 0;
if (!strcmp (mimetype, "audio/mpeg")) {
gint layer = 3;
gst_caps_get_int(caps, "layer", &layer);
switch (layer) {
case 3:
stream->header.audio.codec_tag = GST_RIFF_WAVE_FORMAT_MPEGL3;
break;
case 1: case 2:
stream->header.audio.codec_tag = GST_RIFF_WAVE_FORMAT_MPEGL12;
break;
}
} else if (!strcmp (mimetype, "audio/x-vorbis")) {
stream->header.audio.codec_tag = GST_RIFF_WAVE_FORMAT_VORBIS3;
} else if (!strcmp (mimetype, "audio/x-ac3")) {
stream->header.audio.codec_tag = GST_RIFF_WAVE_FORMAT_A52;
}
stream->header.audio.block_align = 1;
stream->header.audio.byte_rate = 8 * 1024;
stream->header.audio.word_size = 16;
stream->header.audio.size = 0;
if (!stream->header.audio.codec_tag) {
continue;
}
goto done;
}
}
return GST_PAD_LINK_REFUSED;
done:
stream->bitrate = stream->header.audio.byte_rate * 8;
return GST_PAD_LINK_OK;
}
static void
gst_asfmux_pad_link (GstPad *pad,
GstPad *peer,
gpointer data)
{
GstAsfMux *asfmux;
GstAsfMuxStream *stream = NULL;
gint n;
asfmux = GST_ASFMUX (gst_pad_get_parent (pad));
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].pad == pad) {
stream = &asfmux->output[n];
break;
}
}
g_assert (n < asfmux->num_outputs);
g_assert (stream != NULL);
g_assert (stream->connected == FALSE);
stream->connected = TRUE;
}
static void
gst_asfmux_pad_unlink (GstPad *pad,
GstPad *peer,
gpointer data)
{
GstAsfMux *asfmux;
GstAsfMuxStream *stream = NULL;
gint n;
asfmux = GST_ASFMUX (gst_pad_get_parent (pad));
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].pad == pad) {
stream = &asfmux->output[n];
break;
}
}
g_assert (n < asfmux->num_outputs);
g_assert (stream != NULL);
g_assert (stream->connected == TRUE);
stream->connected = FALSE;
}
static GstPad*
gst_asfmux_request_new_pad (GstElement *element,
GstPadTemplate *templ,
const gchar *req_name)
{
GstAsfMux *asfmux;
GstPad *newpad;
gchar *padname;
GstPadLinkFunction linkfunc;
GstAsfMuxStream *stream;
g_return_val_if_fail (templ != NULL, NULL);
g_return_val_if_fail (templ->direction == GST_PAD_SINK, NULL);
g_return_val_if_fail (GST_IS_ASFMUX (element), NULL);
asfmux = GST_ASFMUX (element);
stream = &asfmux->output[asfmux->num_outputs++];
stream->queue = NULL;
stream->time = 0;
stream->connected = FALSE;
stream->eos = FALSE;
stream->seqnum = 0;
if (templ == GST_PAD_TEMPLATE_GET (audio_sink_factory)) {
padname = g_strdup_printf ("audio_%02d", asfmux->num_audio++);
stream->type = ASF_STREAM_AUDIO;
linkfunc = gst_asfmux_audsinkconnect;
} else if (templ == GST_PAD_TEMPLATE_GET (video_sink_factory)) {
padname = g_strdup_printf ("video_%02d", asfmux->num_video++);
stream->type = ASF_STREAM_VIDEO;
linkfunc = gst_asfmux_vidsinkconnect;
} else {
g_warning ("asfmux: this is not our template!\n");
return NULL;
}
newpad = gst_pad_new_from_template (templ, padname);
stream->pad = newpad;
g_free (padname);
g_signal_connect (newpad, "linked",
G_CALLBACK (gst_asfmux_pad_link), (gpointer) asfmux);
g_signal_connect (newpad, "unlinked",
G_CALLBACK (gst_asfmux_pad_unlink), (gpointer) asfmux);
gst_pad_set_link_function (newpad, linkfunc);
gst_element_add_pad (element, newpad);
gst_pad_set_event_function (newpad, gst_asfmux_handle_event);
gst_pad_set_event_mask_function (newpad, gst_asfmux_get_event_masks);
return newpad;
}
/* can we seek? If not, we assume we're streamable */
static gboolean
gst_asfmux_can_seek (GstAsfMux *asfmux)
{
const GstEventMask *masks = gst_pad_get_event_masks (GST_PAD_PEER (asfmux->srcpad));
/* this is for stream or file-storage */
while (masks != NULL && masks->type != 0) {
if (masks->type == GST_EVENT_SEEK) {
return TRUE;
} else {
masks++;
}
}
return FALSE;
}
static gboolean
gst_asfmux_is_stream (GstAsfMux *asfmux)
{
/* this is for RTP */
return FALSE; /*!gst_asfmux_can_seek (asfmux)*/
}
/* handle events (search) */
static gboolean
gst_asfmux_handle_event (GstPad *pad, GstEvent *event)
{
GstAsfMux *asfmux;
GstEventType type;
gint n;
asfmux = GST_ASFMUX (gst_pad_get_parent (pad));
type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN;
switch (type) {
case GST_EVENT_EOS:
/* is this allright? */
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].pad == pad) {
asfmux->output[n].eos = TRUE;
break;
}
}
if (n == asfmux->num_outputs) {
g_warning ("Unknown pad for EOS!");
}
break;
default:
break;
}
return TRUE;
}
/* fill the internal queue for each available pad */
static void
gst_asfmux_fill_queue (GstAsfMux *asfmux)
{
GstBuffer *buffer;
gint n;
for (n = 0; n < asfmux->num_outputs; n++) {
GstAsfMuxStream *stream = &asfmux->output[n];
while (stream->queue == NULL &&
stream->pad != NULL &&
stream->connected == TRUE &&
GST_PAD_IS_USABLE (stream->pad) &&
stream->eos == FALSE) {
buffer = GST_BUFFER (gst_pad_pull (stream->pad));
if (GST_IS_EVENT (buffer)) {
gst_asfmux_handle_event (stream->pad, GST_EVENT (buffer));
} else {
stream->queue = buffer;
}
}
}
}
static guint
gst_asfmux_packet_remaining (GstAsfMux *asfmux)
{
guint position;
if (asfmux->packet != NULL) {
position = GST_BUFFER_SIZE (asfmux->packet);
} else {
position = 0;
}
return GST_ASF_PACKET_SIZE - GST_ASF_PACKET_HEADER_SIZE - 2 - position;
}
static void
gst_asfmux_put_buffer (GstBuffer *packet,
guint8 *data,
guint length)
{
if ((GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet)) >= length) {
guint8 *pos = GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet);
memcpy (pos, data, length);
GST_BUFFER_SIZE (packet) += length;
} else {
g_warning ("Buffer too small");
}
}
static void
gst_asfmux_put_byte (GstBuffer *packet,
guint8 data)
{
if ((GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet)) >= sizeof (data)) {
guint8 *pos = GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet);
* (guint8 *) pos = data;
GST_BUFFER_SIZE (packet) += 1;
} else {
g_warning ("Buffer too small");
}
}
static void
gst_asfmux_put_le16 (GstBuffer *packet,
guint16 data)
{
if ((GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet)) >= sizeof (data)) {
guint8 *pos = GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet);
* (guint16 *) pos = GUINT16_TO_LE (data);
GST_BUFFER_SIZE (packet) += 2;
} else {
g_warning ("Buffer too small");
}
}
static void
gst_asfmux_put_le32 (GstBuffer *packet,
guint32 data)
{
if ((GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet)) >= sizeof (data)) {
guint8 *pos = GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet);
* (guint32 *) pos = GUINT32_TO_LE (data);
GST_BUFFER_SIZE (packet) += 4;
} else {
g_warning ("Buffer too small");
}
}
static void
gst_asfmux_put_le64 (GstBuffer *packet,
guint64 data)
{
if ((GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet)) >= sizeof (data)) {
guint8 *pos = GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet);
* (guint64 *) pos = GUINT64_TO_LE (data);
GST_BUFFER_SIZE (packet) += 8;
} else {
g_warning ("Buffer too small");
}
}
static void
gst_asfmux_put_time (GstBuffer *packet,
guint64 time)
{
gst_asfmux_put_le64 (packet, time + 116444736000000000LLU);
}
static void
gst_asfmux_put_guid (GstBuffer *packet,
ASFGuidHash *hash,
guint8 id)
{
gint n = 0;
ASFGuid *guid;
/* find GUID */
while (hash[n].obj_id != id &&
hash[n].obj_id != ASF_OBJ_UNDEFINED) {
n++;
}
guid = &hash[n].guid;
#if (G_BYTE_ORDER == G_BIG_ENDIAN)
gst_asfmux_put_le32 (packet, guid->v1);
gst_asfmux_put_le32 (packet, guid->v2);
gst_asfmux_put_le32 (packet, guid->v3);
gst_asfmux_put_le32 (packet, guid->v4);
#else
gst_asfmux_put_buffer (packet, (guint8 *) guid, 16);
#endif
}
static void
gst_asfmux_put_string (GstBuffer *packet,
const gchar *str)
{
gunichar2 *utf16_str = g_utf8_to_utf16 (str, strlen (str), NULL, NULL, NULL);
#if (G_BYTE_ORDER == G_BIG_ENDIAN)
gint i, len = strlen (str);
/* this is not an off-by-one-bug, we need the terminating /0 too */
for (i = 0; i <= len; i++) {
gst_asfmux_put_le16 (packet, utf16_str[i]);
}
#else
gst_asfmux_put_buffer (packet, (guint8 *) utf16_str, (strlen (str) + 1) * 2);
#endif
g_free (utf16_str);
}
static void
gst_asfmux_put_flush (GstAsfMux *asfmux)
{
gst_pad_push (asfmux->srcpad,
GST_DATA (gst_event_new_flush ()));
}
/* write an asf chunk (only used in streaming case) */
static void
gst_asfmux_put_chunk (GstBuffer *packet,
GstAsfMux *asfmux,
guint16 type,
guint length,
gint flags)
{
gst_asfmux_put_le16 (packet, type);
gst_asfmux_put_le16 (packet, length + 8);
gst_asfmux_put_le32 (packet, asfmux->sequence++);
gst_asfmux_put_le16 (packet, flags);
gst_asfmux_put_le16 (packet, length + 8);
}
static void
gst_asfmux_put_wav_header (GstBuffer *packet,
asf_stream_audio *hdr)
{
#if (G_BYTE_ORDER == G_BIG_ENDIAN)
gst_asfmux_put_le16 (packet, hdr->codec_tag);
gst_asfmux_put_le16 (packet, hdr->channels);
gst_asfmux_put_le32 (packet, hdr->sample_rate);
gst_asfmux_put_le32 (packet, hdr->byte_rate);
gst_asfmux_put_le16 (packet, hdr->block_align);
gst_asfmux_put_le16 (packet, hdr->word_size);
gst_asfmux_put_le16 (packet, hdr->size);
#else
gst_asfmux_put_buffer (packet, (guint8 *) hdr, 18);
#endif
}
static void
gst_asfmux_put_vid_header (GstBuffer *packet,
asf_stream_video *hdr)
{
/* why does it write hdr->size incorrectly? */
#if (G_BYTE_ORDER == G_BIG_ENDIAN)
gst_asfmux_put_le32 (packet, hdr->width);
gst_asfmux_put_le32 (packet, hdr->height);
gst_asfmux_put_byte (packet, hdr->unknown);
gst_asfmux_put_le16 (packet, hdr->size);
#else
gst_asfmux_put_buffer (packet, (guint8 *) hdr, 11);
#endif
}
static void
gst_asfmux_put_bmp_header (GstBuffer *packet,
asf_stream_video_format *hdr)
{
#if (G_BYTE_ORDER == G_BIG_ENDIAN)
gst_asfmux_put_le32 (packet, hdr->size);
gst_asfmux_put_le32 (packet, hdr->width);
gst_asfmux_put_le32 (packet, hdr->height);
gst_asfmux_put_le16 (packet, hdr->planes);
gst_asfmux_put_le16 (packet, hdr->depth);
gst_asfmux_put_le32 (packet, hdr->tag);
gst_asfmux_put_le32 (packet, hdr->image_size);
gst_asfmux_put_le32 (packet, hdr->xpels_meter);
gst_asfmux_put_le32 (packet, hdr->ypels_meter);
gst_asfmux_put_le32 (packet, hdr->num_colors);
gst_asfmux_put_le32 (packet, hdr->imp_colors);
#else
gst_asfmux_put_buffer (packet, (guint8 *) hdr, 40);
#endif
}
/* init header */
static guint
gst_asfmux_put_header (GstBuffer *packet,
ASFGuidHash *hash,
guint8 id)
{
guint pos = GST_BUFFER_SIZE (packet);
gst_asfmux_put_guid (packet, hash, id);
gst_asfmux_put_le64 (packet, 24);
return pos;
}
/* update header size */
static void
gst_asfmux_end_header (GstBuffer *packet,
guint pos)
{
guint cur = GST_BUFFER_SIZE (packet);
GST_BUFFER_SIZE (packet) = pos + sizeof (ASFGuid);
gst_asfmux_put_le64 (packet, cur - pos);
GST_BUFFER_SIZE (packet) = cur;
}
static void
gst_asfmux_file_start (GstAsfMux *asfmux,
guint64 file_size,
guint64 data_size)
{
GstBuffer *header = gst_buffer_new_and_alloc (4096);
guint bitrate;
guint header_offset, header_pos, header_size;
gint n;
guint64 duration;
bitrate = 0;
for (n = 0; n < asfmux->num_outputs; n++) {
GstAsfMuxStream *stream = &asfmux->output[n];
bitrate += stream->bitrate;
}
GST_BUFFER_SIZE (header) = 0;
if (asfmux->packet != NULL) {
duration = GST_BUFFER_DURATION (asfmux->packet) +
GST_BUFFER_TIMESTAMP (asfmux->packet);
} else {
duration = 0;
}
if (gst_asfmux_is_stream (asfmux)) {
/* start of stream (length will be patched later) */
gst_asfmux_put_chunk (header, asfmux, 0x4824, 0, 0xc00);
}
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_HEADER);
/* header length, will be patched after */
gst_asfmux_put_le64 (header, ~0);
/* number of chunks in header */
gst_asfmux_put_le32 (header, 3 + /*has_title +*/ asfmux->num_outputs);
gst_asfmux_put_byte (header, 1); /* ??? */
gst_asfmux_put_byte (header, 2); /* ??? */
/* file header */
header_offset = GST_BUFFER_SIZE (header);
header_pos = gst_asfmux_put_header (header, asf_object_guids, ASF_OBJ_FILE);
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_UNDEFINED);
gst_asfmux_put_le64 (header, file_size);
gst_asfmux_put_time (header, 0);
gst_asfmux_put_le64 (header, asfmux->num_packets); /* number of packets */
gst_asfmux_put_le64 (header, duration / (GST_SECOND / 10000000)); /* end time stamp (in 100ns units) */
gst_asfmux_put_le64 (header, duration / (GST_SECOND / 10000000)); /* duration (in 100ns units) */
gst_asfmux_put_le64 (header, 0); /* start time stamp */
gst_asfmux_put_le32 (header, gst_asfmux_can_seek (asfmux) ? 0x02 : 0x01); /* seekable or streamable */
gst_asfmux_put_le32 (header, GST_ASF_PACKET_SIZE); /* packet size */
gst_asfmux_put_le32 (header, GST_ASF_PACKET_SIZE); /* packet size */
gst_asfmux_put_le32 (header, bitrate); /* Nominal data rate in bps */
gst_asfmux_end_header (header, header_pos);
/* unknown headers */
header_pos = gst_asfmux_put_header (header, asf_object_guids, ASF_OBJ_HEAD1);
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_HEAD2);
gst_asfmux_put_le32 (header, 6);
gst_asfmux_put_le16 (header, 0);
gst_asfmux_end_header (header, header_pos);
/* title and other infos */
#if 0
if (has_title) {
header_pos = gst_asfmux_put_header (header, asf_object_guids, ASF_OBJ_COMMENT);
gst_asfmux_put_le16 (header, 2 * (strlen(title) + 1));
gst_asfmux_put_le16 (header, 2 * (strlen(author) + 1));
gst_asfmux_put_le16 (header, 2 * (strlen(copyright) + 1));
gst_asfmux_put_le16 (header, 2 * (strlen(comment) + 1));
gst_asfmux_put_le16 (header, 0); /* rating */
gst_asfmux_put_string (header, title);
gst_asfmux_put_string (header, author);
gst_asfmux_put_string (header, copyright);
gst_asfmux_put_string (header, comment);
gst_asfmux_end_header (header, header_pos);
}
#endif
/* stream headers */
for (n = 0; n < asfmux->num_outputs; n++) {
GstAsfMuxStream *stream = &asfmux->output[n];
guint obj_size = 0;
stream->seqnum = 0;
header_pos = gst_asfmux_put_header (header, asf_object_guids, ASF_OBJ_STREAM);
switch (stream->type) {
case ASF_STREAM_AUDIO:
obj_size = 18;
gst_asfmux_put_guid (header, asf_stream_guids, ASF_STREAM_AUDIO);
gst_asfmux_put_guid (header, asf_correction_guids, ASF_CORRECTION_OFF);
break;
case ASF_STREAM_VIDEO:
obj_size = 11 + 40;
gst_asfmux_put_guid (header, asf_stream_guids, ASF_STREAM_VIDEO);
gst_asfmux_put_guid (header, asf_correction_guids, ASF_CORRECTION_OFF);
break;
default:
g_assert (0);
}
gst_asfmux_put_le64 (header, 0); /* offset */
gst_asfmux_put_le32 (header, obj_size); /* wav header len */
gst_asfmux_put_le32 (header, 0); /* additional data len */
gst_asfmux_put_le16 (header, n + 1); /* stream number */
gst_asfmux_put_le32 (header, 0); /* ??? */
switch (stream->type) {
case ASF_STREAM_AUDIO:
gst_asfmux_put_wav_header (header, &stream->header.audio);
break;
case ASF_STREAM_VIDEO:
gst_asfmux_put_vid_header (header, &stream->header.video.stream);
gst_asfmux_put_bmp_header (header, &stream->header.video.format);
break;
}
gst_asfmux_end_header (header, header_pos);
}
/* media comments */
header_pos = gst_asfmux_put_header (header, asf_object_guids, ASF_OBJ_CODEC_COMMENT);
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_CODEC_COMMENT1);
gst_asfmux_put_le32 (header, asfmux->num_outputs);
for (n = 0; n < asfmux->num_outputs; n++) {
GstAsfMuxStream *stream = &asfmux->output[n];
const char *codec = "Unknown codec";
gst_asfmux_put_le16 (header, stream->index + 1);
/* Isn't this wrong? This is UTF16! */
gst_asfmux_put_le16 (header, strlen (codec) + 1);
gst_asfmux_put_string (header, codec);
gst_asfmux_put_le16 (header, 0); /* no parameters */
/* id */
switch (stream->type) {
case ASF_STREAM_AUDIO:
gst_asfmux_put_le16 (header, 2);
gst_asfmux_put_le16 (header, stream->header.audio.codec_tag);
break;
case ASF_STREAM_VIDEO:
gst_asfmux_put_le16 (header, 4);
gst_asfmux_put_le32 (header, stream->header.video.format.tag);
break;
default:
g_assert (0);
}
}
gst_asfmux_end_header (header, header_pos);
/* patch the header size fields */
header_pos = GST_BUFFER_SIZE (header);
header_size = header_pos - header_offset;
if (gst_asfmux_is_stream (asfmux)) {
header_size += 8 + 30 + 50;
GST_BUFFER_SIZE (header) = header_offset - 10 - 30;
gst_asfmux_put_le16 (header, header_size);
GST_BUFFER_SIZE (header) = header_offset - 2 - 30;
gst_asfmux_put_le16 (header, header_size);
header_size -= 8 + 30 + 50;
}
header_size += 24 + 6;
GST_BUFFER_SIZE (header) = header_offset - 14;
gst_asfmux_put_le64 (header, header_size);
GST_BUFFER_SIZE (header) = header_pos;
/* movie chunk, followed by packets of packet_size */
asfmux->data_offset = GST_BUFFER_SIZE (header);
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_DATA);
gst_asfmux_put_le64 (header, data_size);
gst_asfmux_put_guid (header, asf_object_guids, ASF_OBJ_UNDEFINED);
gst_asfmux_put_le64 (header, asfmux->num_packets); /* nb packets */
gst_asfmux_put_byte (header, 1); /* ??? */
gst_asfmux_put_byte (header, 1); /* ??? */
gst_pad_push (asfmux->srcpad, GST_DATA (header));
asfmux->write_header = FALSE;
}
static void
gst_asfmux_file_stop (GstAsfMux *asfmux)
{
if (gst_asfmux_is_stream (asfmux)) {
/* send EOS chunk */
GstBuffer *footer = gst_buffer_new_and_alloc (16);
GST_BUFFER_SIZE (footer) = 0;
gst_asfmux_put_chunk (footer, asfmux, 0x4524, 0, 0); /* end of stream */
gst_pad_push (asfmux->srcpad, GST_DATA (footer));
} else if (gst_asfmux_can_seek (asfmux)) {
/* rewrite an updated header */
guint64 filesize;
GstFormat fmt = GST_FORMAT_BYTES;
GstEvent *event;
gst_pad_query (asfmux->srcpad, GST_QUERY_POSITION,
&fmt, &filesize);
event = gst_event_new_seek (GST_SEEK_METHOD_SET | GST_FORMAT_BYTES, 0);
gst_pad_push (asfmux->srcpad, GST_DATA (event));
gst_asfmux_file_start (asfmux, filesize, filesize - asfmux->data_offset);
event = gst_event_new_seek (GST_SEEK_METHOD_SET | GST_FORMAT_BYTES, filesize);
gst_pad_push (asfmux->srcpad, GST_DATA (event));
}
gst_asfmux_put_flush (asfmux);
}
static GstBuffer *
gst_asfmux_packet_header (GstAsfMux *asfmux)
{
GstBuffer *packet = asfmux->packet, *header;
guint flags, padsize = gst_asfmux_packet_remaining (asfmux);
header = gst_buffer_new_and_alloc (GST_ASF_PACKET_HEADER_SIZE + 2 + 12);
GST_BUFFER_SIZE (header) = 0;
if (gst_asfmux_is_stream (asfmux)) {
gst_asfmux_put_chunk (header, asfmux, 0x4424, GST_ASF_PACKET_SIZE, 0);
}
gst_asfmux_put_byte (header, 0x82);
gst_asfmux_put_le16 (header, 0);
flags = 0x01; /* nb segments present */
if (padsize > 0) {
if (padsize < 256) {
flags |= 0x08;
} else {
flags |= 0x10;
}
}
gst_asfmux_put_byte (header, flags); /* flags */
gst_asfmux_put_byte (header, 0x5d);
if (flags & 0x10) {
gst_asfmux_put_le16 (header, padsize - 2);
} else if (flags & 0x08) {
gst_asfmux_put_byte (header, padsize - 1);
}
gst_asfmux_put_le32 (header, GST_BUFFER_TIMESTAMP (packet) / (GST_SECOND/1000));
gst_asfmux_put_le16 (header, GST_BUFFER_DURATION (packet) / (GST_SECOND/1000));
gst_asfmux_put_byte (header, asfmux->packet_frames | 0x80);
return header;
}
static void
gst_asfmux_frame_header (GstAsfMux *asfmux,
GstAsfMuxStream *stream,
guint position,
guint length,
guint total,
guint64 time,
gboolean key)
{
/* fill in some values for the packet */
if (!GST_CLOCK_TIME_IS_VALID (GST_BUFFER_TIMESTAMP (asfmux->packet))) {
GST_BUFFER_TIMESTAMP (asfmux->packet) = time;
}
GST_BUFFER_DURATION (asfmux->packet) =
time - GST_BUFFER_TIMESTAMP (asfmux->packet);
gst_asfmux_put_byte (asfmux->packet, (stream->index + 1) | 0x80); //(key ? 0x80 : 0));
gst_asfmux_put_byte (asfmux->packet, stream->seqnum);
gst_asfmux_put_le32 (asfmux->packet, position);
gst_asfmux_put_byte (asfmux->packet, 0x08);
gst_asfmux_put_le32 (asfmux->packet, total);
gst_asfmux_put_le32 (asfmux->packet, time / (GST_SECOND/1000)); /* time in ms */
gst_asfmux_put_le16 (asfmux->packet, length);
}
static void
gst_asfmux_frame_buffer (GstAsfMux *asfmux,
guint8 *data,
guint length)
{
gst_asfmux_put_buffer (asfmux->packet, data, length);
asfmux->packet_frames++;
}
static void
gst_asfmux_packet_flush (GstAsfMux *asfmux)
{
GstBuffer *header, *packet = asfmux->packet;
guint header_size;
/* packet header */
header = gst_asfmux_packet_header (asfmux);
header_size = GST_BUFFER_SIZE (header);
if (!gst_asfmux_can_seek (asfmux)) {
header_size -= 12; /* hack... bah */
}
/* Clear out the padding bytes */
memset (GST_BUFFER_DATA (packet) + GST_BUFFER_SIZE (packet), 0,
GST_BUFFER_MAXSIZE (packet) - GST_BUFFER_SIZE (packet));
GST_BUFFER_SIZE (packet) = GST_ASF_PACKET_SIZE - header_size;
/* send packet over */
gst_pad_push (asfmux->srcpad, GST_DATA (header));
gst_pad_push (asfmux->srcpad, GST_DATA (packet));
gst_asfmux_put_flush (asfmux);
asfmux->num_packets++;
asfmux->packet_frames = 0;
/* reset packet */
asfmux->packet = NULL;
}
static void
gst_asfmux_write_buffer (GstAsfMux *asfmux,
GstAsfMuxStream *stream,
GstBuffer *buffer)
{
guint position = 0, to_write, size = GST_BUFFER_SIZE (buffer), remaining;
while (position < size) {
remaining = gst_asfmux_packet_remaining (asfmux);
if (remaining <= GST_ASF_FRAME_HEADER_SIZE) {
gst_asfmux_packet_flush (asfmux);
continue;
} else if (remaining >= (GST_ASF_FRAME_HEADER_SIZE + size - position)) {
to_write = size - position;
} else {
to_write = remaining - GST_ASF_FRAME_HEADER_SIZE;
}
if (asfmux->packet == NULL) {
asfmux->packet_frames = 0;
asfmux->packet = gst_buffer_new_and_alloc (GST_ASF_PACKET_SIZE);
GST_BUFFER_SIZE (asfmux->packet) = 0;
}
/* write frame header plus data in this packet */
gst_asfmux_frame_header (asfmux, stream, position, to_write, size,
GST_BUFFER_TIMESTAMP (buffer),
GST_BUFFER_FLAG_IS_SET (buffer, GST_BUFFER_KEY_UNIT));
gst_asfmux_frame_buffer (asfmux, GST_BUFFER_DATA (buffer) + position, to_write);
position += to_write;
}
stream->seqnum++;
}
/* take the oldest buffer in our internal queue and push-it */
static gboolean
gst_asfmux_do_one_buffer (GstAsfMux *asfmux)
{
gint n, chosen = -1;
/* find the earliest buffer */
for (n = 0; n < asfmux->num_outputs; n++) {
if (asfmux->output[n].queue != NULL) {
if (chosen == -1 ||
GST_BUFFER_TIMESTAMP (asfmux->output[n].queue) <
GST_BUFFER_TIMESTAMP (asfmux->output[chosen].queue)) {
chosen = n;
}
}
}
if (chosen == -1) {
/* simply finish off the file and send EOS */
gst_asfmux_file_stop (asfmux);
gst_pad_push (asfmux->srcpad,
GST_DATA (gst_event_new (GST_EVENT_EOS)));
gst_element_set_eos (GST_ELEMENT(asfmux));
return FALSE;
}
/* do this buffer */
gst_asfmux_write_buffer (asfmux, &asfmux->output[chosen],
asfmux->output[chosen].queue);
/* update stream info after buffer push */
gst_buffer_unref (asfmux->output[chosen].queue);
asfmux->output[chosen].time = GST_BUFFER_TIMESTAMP (asfmux->output[chosen].queue);
asfmux->output[chosen].queue = NULL;
return TRUE;
}
static void
gst_asfmux_loop (GstElement *element)
{
GstAsfMux *asfmux;
asfmux = GST_ASFMUX (element);
/* first fill queue (some elements only set caps when
* flowing data), then write header */
gst_asfmux_fill_queue (asfmux);
if (asfmux->write_header == TRUE) {
/* indeed, these are fake values. We need this so that
* players will read the file. Without these fake values,
* the players will mark the file as invalid and stop */
gst_asfmux_file_start (asfmux, 0xFFFFFFFF, 0xFFFFFFFF);
}
gst_asfmux_do_one_buffer (asfmux);
}
static GstElementStateReturn
gst_asfmux_change_state (GstElement *element)
{
GstAsfMux *asfmux;
gint transition = GST_STATE_TRANSITION (element), n;
g_return_val_if_fail (GST_IS_ASFMUX (element), GST_STATE_FAILURE);
asfmux = GST_ASFMUX (element);
switch (transition) {
case GST_STATE_PAUSED_TO_PLAYING:
for (n = 0; n < asfmux->num_outputs; n++) {
asfmux->output[n].eos = FALSE;
}
break;
}
if (GST_ELEMENT_CLASS (parent_class)->change_state)
return GST_ELEMENT_CLASS (parent_class)->change_state (element);
return GST_STATE_SUCCESS;
}
static gboolean
plugin_init (GstPlugin *plugin)
{
if (!gst_element_register (plugin, "asfmux", GST_RANK_NONE, GST_TYPE_ASFMUX))
return FALSE;
return TRUE;
}
GST_PLUGIN_DEFINE (
GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"asfmux",
"Muxes audio and video into an ASF stream",
plugin_init,
VERSION,
"LGPL",
GST_COPYRIGHT,
GST_PACKAGE,
GST_ORIGIN)