summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--ChangeLog16
-rw-r--r--gst/flv/Makefile.am13
-rw-r--r--gst/flv/gstflvdemux.c7
-rw-r--r--gst/flv/gstflvmux.c754
-rw-r--r--gst/flv/gstflvmux.h84
5 files changed, 865 insertions, 9 deletions
diff --git a/ChangeLog b/ChangeLog
index cc3ab715..c87aea16 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,19 @@
+2008-10-09 Sebastian Dröge <sebastian.droege@collabora.co.uk>
+
+ * gst/flv/Makefile.am:
+ * gst/flv/gstflvdemux.c: (plugin_init):
+ * gst/flv/gstflvmux.c: (gst_flv_mux_base_init),
+ (gst_flv_mux_class_init), (gst_flv_mux_init),
+ (gst_flv_mux_finalize), (gst_flv_mux_reset),
+ (gst_flv_mux_handle_src_event), (gst_flv_mux_handle_sink_event),
+ (gst_flv_mux_video_pad_setcaps), (gst_flv_mux_audio_pad_setcaps),
+ (gst_flv_mux_request_new_pad), (gst_flv_mux_release_pad),
+ (gst_flv_mux_write_header), (gst_flv_mux_write_buffer),
+ (gst_flv_mux_collected), (gst_flv_mux_change_state):
+ * gst/flv/gstflvmux.h:
+ Add first version of a FLV muxer. The only missing feature is writing
+ of stream metadata.
+
2008-10-09 Stefan Kost <ensonic@users.sf.net>
* ext/amrwb/gstamrwbparse.c:
diff --git a/gst/flv/Makefile.am b/gst/flv/Makefile.am
index 7850e7ee..08018d20 100644
--- a/gst/flv/Makefile.am
+++ b/gst/flv/Makefile.am
@@ -1,9 +1,8 @@
+plugin_LTLIBRARIES = libgstflv.la
-plugin_LTLIBRARIES = libgstflvdemux.la
+libgstflv_la_CFLAGS = ${GST_CFLAGS}
+libgstflv_la_LIBADD = $(GST_BASE_LIBS)
+libgstflv_la_LDFLAGS = ${GST_PLUGIN_LDFLAGS}
+libgstflv_la_SOURCES = gstflvdemux.c gstflvparse.c gstflvmux.c
-libgstflvdemux_la_CFLAGS = ${GST_CFLAGS}
-libgstflvdemux_la_LIBADD = $(GST_BASE_LIBS)
-libgstflvdemux_la_LDFLAGS = ${GST_PLUGIN_LDFLAGS}
-libgstflvdemux_la_SOURCES = gstflvdemux.c gstflvparse.c
-
-noinst_HEADERS = gstflvdemux.h gstflvparse.h
+noinst_HEADERS = gstflvdemux.h gstflvparse.h gstflvmux.h
diff --git a/gst/flv/gstflvdemux.c b/gst/flv/gstflvdemux.c
index 0f5e3f79..896d8bf1 100644
--- a/gst/flv/gstflvdemux.c
+++ b/gst/flv/gstflvdemux.c
@@ -23,6 +23,7 @@
#include "gstflvdemux.h"
#include "gstflvparse.h"
+#include "gstflvmux.h"
#include <string.h>
@@ -1207,12 +1208,14 @@ plugin_init (GstPlugin * plugin)
GST_DEBUG_CATEGORY_INIT (flvdemux_debug, "flvdemux", 0, "FLV demuxer");
if (!gst_element_register (plugin, "flvdemux", GST_RANK_PRIMARY,
- gst_flv_demux_get_type ()))
+ gst_flv_demux_get_type ()) ||
+ !gst_element_register (plugin, "flvmux", GST_RANK_NONE,
+ gst_flv_mux_get_type ()))
return FALSE;
return TRUE;
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR,
- "flvdemux", "Element demuxing FLV stream",
+ "flv", "FLV muxing and demuxing plugin",
plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
diff --git a/gst/flv/gstflvmux.c b/gst/flv/gstflvmux.c
new file mode 100644
index 00000000..a41b03f6
--- /dev/null
+++ b/gst/flv/gstflvmux.c
@@ -0,0 +1,754 @@
+/* GStreamer
+ *
+ * Copyright (c) 2008 Sebastian Dröge <sebastian.droege@collabora.co.uk>
+ *
+ * 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.
+ */
+
+/* TODO:
+ * - Write metadata for the file, see FLV spec page 13
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <math.h>
+#include <string.h>
+
+#include "gstflvmux.h"
+
+GST_DEBUG_CATEGORY_STATIC (flvmux_debug);
+#define GST_CAT_DEFAULT flvmux_debug
+
+static GstStaticPadTemplate src_templ = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-flv")
+ );
+
+static GstStaticPadTemplate videosink_templ = GST_STATIC_PAD_TEMPLATE ("video",
+ GST_PAD_SINK,
+ GST_PAD_REQUEST,
+ GST_STATIC_CAPS ("video/x-flash-video; "
+ "video/x-flash-screen; "
+ "video/x-vp6-flash; " "video/x-vp6-alpha; " "video/x-h264;")
+ );
+
+static GstStaticPadTemplate audiosink_templ = GST_STATIC_PAD_TEMPLATE ("audio",
+ GST_PAD_SINK,
+ GST_PAD_REQUEST,
+ GST_STATIC_CAPS
+ ("audio/x-adpcm, layout = (string) swf, channels = (int) { 1, 2 }, rate = (int) { 5512, 11025, 22050, 44100 }; "
+ "audio/mpeg, mpegversion = (int) 1, layer = (int) 3, channels = (int) { 1, 2 }, rate = (int) { 5512, 8000, 11025, 22050, 44100 }; "
+ "audio/mpeg, mpegversion = (int) 4; "
+ "audio/x-nellymoser, channels = (int) { 1, 2 }, rate = (int) { 5512, 8000, 11025, 16000, 22050, 44100 }; "
+ "audio/x-raw-int, endianness = (int) LITTLE_ENDIAN, channels = (int) { 1, 2 }, width = (int) 8, depth = (int) 8, rate = (int) { 5512, 11025, 22050, 44100 }, signed = (boolean) FALSE; "
+ "audio/x-raw-int, endianness = (int) LITTLE_ENDIAN, channels = (int) { 1, 2 }, width = (int) 16, depth = (int) 16, rate = (int) { 5512, 11025, 22050, 44100 }, signed = (boolean) TRUE; "
+ "audio/x-alaw, channels = (int) { 1, 2 }, rate = (int) { 5512, 11025, 22050, 44100 }; "
+ "audio/x-mulaw, channels = (int) { 1, 2 }, rate = (int) { 5512, 11025, 22050, 44100 };")
+ );
+
+GST_BOILERPLATE (GstFlvMux, gst_flv_mux, GstElement, GST_TYPE_ELEMENT);
+
+static void gst_flv_mux_finalize (GObject * object);
+static GstFlowReturn
+gst_flv_mux_collected (GstCollectPads * pads, gpointer user_data);
+
+static gboolean gst_flv_mux_handle_src_event (GstPad * pad, GstEvent * event);
+static GstPad *gst_flv_mux_request_new_pad (GstElement * element,
+ GstPadTemplate * templ, const gchar * name);
+static void gst_flv_mux_release_pad (GstElement * element, GstPad * pad);
+
+static GstStateChangeReturn
+gst_flv_mux_change_state (GstElement * element, GstStateChange transition);
+
+static void gst_flv_mux_reset (GstElement * element);
+
+static void
+gst_flv_mux_base_init (gpointer g_class)
+{
+ GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
+
+ gst_element_class_add_pad_template (element_class,
+ gst_static_pad_template_get (&videosink_templ));
+ gst_element_class_add_pad_template (element_class,
+ gst_static_pad_template_get (&audiosink_templ));
+ gst_element_class_add_pad_template (element_class,
+ gst_static_pad_template_get (&src_templ));
+ gst_element_class_set_details_simple (element_class, "FLV muxer",
+ "Codec/Muxer",
+ "Muxes video/audio streams into a FLV stream",
+ "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
+
+ GST_DEBUG_CATEGORY_INIT (flvmux_debug, "flvmux", 0, "FLV muxer");
+}
+
+static void
+gst_flv_mux_class_init (GstFlvMuxClass * klass)
+{
+ GObjectClass *gobject_class;
+ GstElementClass *gstelement_class;
+
+ GST_DEBUG_CATEGORY_INIT (flvmux_debug, "flvmux", 0, "FLV muxer");
+
+ gobject_class = (GObjectClass *) klass;
+ gstelement_class = (GstElementClass *) klass;
+
+ gobject_class->finalize = gst_flv_mux_finalize;
+
+ gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_flv_mux_change_state);
+ gstelement_class->request_new_pad =
+ GST_DEBUG_FUNCPTR (gst_flv_mux_request_new_pad);
+ gstelement_class->release_pad = GST_DEBUG_FUNCPTR (gst_flv_mux_release_pad);
+}
+
+static void
+gst_flv_mux_init (GstFlvMux * mux, GstFlvMuxClass * g_class)
+{
+ mux->srcpad = gst_pad_new_from_static_template (&src_templ, "src");
+ gst_pad_set_event_function (mux->srcpad, gst_flv_mux_handle_src_event);
+ gst_element_add_pad (GST_ELEMENT (mux), mux->srcpad);
+
+ mux->collect = gst_collect_pads_new ();
+ gst_collect_pads_set_function (mux->collect,
+ (GstCollectPadsFunction) GST_DEBUG_FUNCPTR (gst_flv_mux_collected), mux);
+
+ gst_flv_mux_reset (GST_ELEMENT (mux));
+}
+
+static void
+gst_flv_mux_finalize (GObject * object)
+{
+ GstFlvMux *mux = GST_FLV_MUX (object);
+
+ gst_object_unref (mux->collect);
+
+ G_OBJECT_CLASS (parent_class)->finalize (object);
+}
+
+static void
+gst_flv_mux_reset (GstElement * element)
+{
+ GstFlvMux *mux = GST_FLV_MUX (element);
+ GSList *sl;
+
+ while ((sl = mux->collect->data) != NULL) {
+ GstFlvPad *cpad = (GstFlvPad *) sl->data;
+
+ if (cpad->audio_codec_data)
+ gst_buffer_unref (cpad->audio_codec_data);
+ if (cpad->video_codec_data)
+ gst_buffer_unref (cpad->video_codec_data);
+
+ gst_collect_pads_remove_pad (mux->collect, cpad->collect.pad);
+ }
+
+ mux->state = GST_FLV_MUX_STATE_HEADER;
+}
+
+static gboolean
+gst_flv_mux_handle_src_event (GstPad * pad, GstEvent * event)
+{
+ GstEventType type;
+
+ type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN;
+
+ switch (type) {
+ case GST_EVENT_SEEK:
+ /* disable seeking for now */
+ return FALSE;
+ default:
+ break;
+ }
+
+ return gst_pad_event_default (pad, event);
+}
+
+static gboolean
+gst_flv_mux_handle_sink_event (GstPad * pad, GstEvent * event)
+{
+ GstFlvMux *mux = GST_FLV_MUX (gst_pad_get_parent (pad));
+ gboolean ret = TRUE;
+
+ switch (GST_EVENT_TYPE (event)) {
+ case GST_EVENT_TAG:
+ /* TODO do something with the tags */
+ break;
+ case GST_EVENT_NEWSEGMENT:
+ /* We don't support NEWSEGMENT events */
+ ret = FALSE;
+ gst_event_unref (event);
+ break;
+ default:
+ break;
+ }
+
+ /* now GstCollectPads can take care of the rest, e.g. EOS */
+ if (ret)
+ ret = mux->collect_event (pad, event);
+ gst_object_unref (mux);
+
+ return ret;
+}
+
+static gboolean
+gst_flv_mux_video_pad_setcaps (GstPad * pad, GstCaps * caps)
+{
+ GstFlvMux *mux = GST_FLV_MUX (gst_pad_get_parent (pad));
+ GstFlvPad *cpad = (GstFlvPad *) gst_pad_get_element_private (pad);
+ gboolean ret = TRUE;
+ GstStructure *s;
+
+ s = gst_caps_get_structure (caps, 0);
+
+ if (strcmp (gst_structure_get_name (s), "video/x-flash-video") == 0) {
+ cpad->video_codec = 2;
+ } else if (strcmp (gst_structure_get_name (s), "video/x-flash-screen") == 0) {
+ cpad->video_codec = 3;
+ } else if (strcmp (gst_structure_get_name (s), "video/x-vp6-flash") == 0) {
+ cpad->video_codec = 4;
+ } else if (strcmp (gst_structure_get_name (s), "video/x-vp6-alpha") == 0) {
+ cpad->video_codec = 5;
+ } else if (strcmp (gst_structure_get_name (s), "video/x-h264") == 0) {
+ cpad->video_codec = 7;
+ } else {
+ ret = FALSE;
+ }
+
+ if (ret && gst_structure_has_field (s, "codec_data")) {
+ const GValue *val = gst_structure_get_value (s, "codec_data");
+
+ if (val) {
+ cpad->video_codec_data = gst_buffer_ref (gst_value_get_buffer (val));
+ cpad->sent_codec_data = FALSE;
+ } else {
+ cpad->sent_codec_data = TRUE;
+ }
+ } else {
+ cpad->sent_codec_data = TRUE;
+ }
+
+ gst_object_unref (mux);
+
+ return ret;
+}
+
+static gboolean
+gst_flv_mux_audio_pad_setcaps (GstPad * pad, GstCaps * caps)
+{
+ GstFlvMux *mux = GST_FLV_MUX (gst_pad_get_parent (pad));
+ GstFlvPad *cpad = (GstFlvPad *) gst_pad_get_element_private (pad);
+ gboolean ret = TRUE;
+ GstStructure *s;
+
+ s = gst_caps_get_structure (caps, 0);
+
+ if (strcmp (gst_structure_get_name (s), "audio/x-adpcm") == 0) {
+ const gchar *layout = gst_structure_get_string (s, "layout");
+ if (layout && strcmp (layout, "swf") == 0) {
+ cpad->audio_codec = 1;
+ } else {
+ ret = FALSE;
+ }
+ } else if (strcmp (gst_structure_get_name (s), "audio/mpeg") == 0) {
+ gint mpegversion;
+
+ if (gst_structure_get_int (s, "mpegversion", &mpegversion)) {
+ if (mpegversion == 1) {
+ gint layer;
+
+ if (gst_structure_get_int (s, "layer", &layer) && layer == 3) {
+ gint rate;
+
+ if (gst_structure_get_int (s, "rate", &rate) && rate == 8000)
+ cpad->audio_codec = 14;
+ else
+ cpad->audio_codec = 2;
+ } else {
+ ret = FALSE;
+ }
+ } else if (mpegversion == 4) {
+ cpad->audio_codec = 10;
+ } else {
+ ret = FALSE;
+ }
+ } else {
+ ret = FALSE;
+ }
+ } else if (strcmp (gst_structure_get_name (s), "audio/x-nellymoser") == 0) {
+ gint rate, channels;
+
+ if (gst_structure_get_int (s, "rate", &rate)
+ && gst_structure_get_int (s, "channels", &channels)) {
+ if (channels == 1 && rate == 16000)
+ cpad->audio_codec = 4;
+ else if (channels == 1 && rate == 8000)
+ cpad->audio_codec = 5;
+ } else {
+ cpad->audio_codec = 6;
+ }
+ } else if (strcmp (gst_structure_get_name (s), "audio/x-raw-int") == 0) {
+ gint endianness;
+
+ if (gst_structure_get_int (s, "endianness", &endianness)
+ && endianness == G_LITTLE_ENDIAN)
+ cpad->audio_codec = 3;
+ else
+ ret = FALSE;
+ } else if (strcmp (gst_structure_get_name (s), "audio/x-alaw") == 0) {
+ cpad->video_codec = 7;
+ } else if (strcmp (gst_structure_get_name (s), "audio/x-mulaw") == 0) {
+ cpad->video_codec = 8;
+ } else {
+ ret = FALSE;
+ }
+
+ if (ret) {
+ gint rate, channels, width;
+
+ if (gst_structure_get_int (s, "rate", &rate)) {
+ if (cpad->audio_codec == 10)
+ cpad->rate = 3;
+ else if (rate == 5512)
+ cpad->rate = 0;
+ else if (rate == 11025)
+ cpad->rate = 1;
+ else if (rate == 22050)
+ cpad->rate = 2;
+ else if (rate == 44100)
+ cpad->rate = 3;
+ else if (rate == 8000 && (cpad->audio_codec == 5
+ || cpad->audio_codec == 14))
+ cpad->rate = 0;
+ else if (rate == 16000 && cpad->audio_codec == 4)
+ cpad->rate = 0;
+ else
+ ret = FALSE;
+ } else if (cpad->audio_codec == 10) {
+ cpad->rate = 3;
+ } else {
+ ret = FALSE;
+ }
+
+ if (gst_structure_get_int (s, "channels", &channels)) {
+ if (cpad->audio_codec == 4 || cpad->audio_codec == 5
+ || cpad->audio_codec == 6)
+ cpad->channels = 0;
+ else if (cpad->audio_codec == 10)
+ cpad->channels = 1;
+ else if (channels == 1)
+ cpad->channels = 0;
+ else if (channels == 2)
+ cpad->channels = 1;
+ else
+ ret = FALSE;
+ } else if (cpad->audio_codec == 4 || cpad->audio_codec == 5
+ || cpad->audio_codec == 6) {
+ cpad->channels = 0;
+ } else if (cpad->audio_codec == 10) {
+ cpad->channels = 1;
+ } else {
+ ret = FALSE;
+ }
+
+ if (gst_structure_get_int (s, "width", &width)) {
+ if (cpad->audio_codec != 3)
+ cpad->width = 1;
+ else if (width == 8)
+ cpad->width = 0;
+ else if (width == 16)
+ cpad->width = 1;
+ else
+ ret = FALSE;
+ } else if (cpad->audio_codec != 3) {
+ cpad->width = 1;
+ } else {
+ ret = FALSE;
+ }
+ }
+
+ if (ret && gst_structure_has_field (s, "codec_data")) {
+ const GValue *val = gst_structure_get_value (s, "codec_data");
+
+ if (val) {
+ cpad->audio_codec_data = gst_buffer_ref (gst_value_get_buffer (val));
+ cpad->sent_codec_data = FALSE;
+ } else {
+ cpad->sent_codec_data = TRUE;
+ }
+ } else {
+ cpad->sent_codec_data = TRUE;
+ }
+
+ gst_object_unref (mux);
+
+ return ret;
+}
+
+static GstPad *
+gst_flv_mux_request_new_pad (GstElement * element,
+ GstPadTemplate * templ, const gchar * pad_name)
+{
+ GstElementClass *klass = GST_ELEMENT_GET_CLASS (element);
+ GstFlvMux *mux = GST_FLV_MUX (element);
+ GstFlvPad *cpad;
+ GstPad *pad = NULL;
+ const gchar *name = NULL;
+ GstPadSetCapsFunction setcapsfunc = NULL;
+ gboolean video;
+
+ if (mux->state != GST_FLV_MUX_STATE_HEADER) {
+ GST_WARNING_OBJECT (mux, "Can't request pads after writing header");
+ return NULL;
+ }
+
+ if (templ == gst_element_class_get_pad_template (klass, "audio")) {
+ if (mux->have_audio) {
+ GST_WARNING_OBJECT (mux, "Already have an audio pad");
+ return NULL;
+ }
+ mux->have_audio = TRUE;
+ name = "audio";
+ video = FALSE;
+ setcapsfunc = GST_DEBUG_FUNCPTR (gst_flv_mux_audio_pad_setcaps);
+ } else if (templ == gst_element_class_get_pad_template (klass, "video")) {
+ if (mux->have_video) {
+ GST_WARNING_OBJECT (mux, "Already have a video pad");
+ return NULL;
+ }
+ mux->have_video = TRUE;
+ name = "video";
+ video = TRUE;
+ setcapsfunc = GST_DEBUG_FUNCPTR (gst_flv_mux_video_pad_setcaps);
+ } else {
+ GST_WARNING_OBJECT (mux, "Invalid template");
+ return NULL;
+ }
+
+ pad = gst_pad_new_from_template (templ, name);
+ cpad = (GstFlvPad *)
+ gst_collect_pads_add_pad (mux->collect, pad, sizeof (GstFlvPad));
+
+ cpad->video = video;
+
+ cpad->audio_codec = G_MAXUINT;
+ cpad->rate = G_MAXUINT;
+ cpad->width = G_MAXUINT;
+ cpad->channels = G_MAXUINT;
+ cpad->audio_codec_data = NULL;
+
+ cpad->video_codec = G_MAXUINT;
+ cpad->video_codec_data = NULL;
+
+ cpad->sent_codec_data = FALSE;
+
+ /* FIXME: hacked way to override/extend the event function of
+ * GstCollectPads; because it sets its own event function giving the
+ * element no access to events.
+ */
+ mux->collect_event = (GstPadEventFunction) GST_PAD_EVENTFUNC (pad);
+ gst_pad_set_event_function (pad,
+ GST_DEBUG_FUNCPTR (gst_flv_mux_handle_sink_event));
+
+ gst_pad_set_setcaps_function (pad, setcapsfunc);
+ gst_pad_set_active (pad, TRUE);
+ gst_element_add_pad (element, pad);
+
+ return pad;
+}
+
+static void
+gst_flv_mux_release_pad (GstElement * element, GstPad * pad)
+{
+ GstFlvMux *mux = GST_FLV_MUX (GST_PAD_PARENT (pad));
+ GstFlvPad *cpad = (GstFlvPad *) gst_pad_get_element_private (pad);
+
+ if (cpad && cpad->audio_codec_data)
+ gst_buffer_unref (cpad->audio_codec_data);
+ if (cpad && cpad->video_codec_data)
+ gst_buffer_unref (cpad->video_codec_data);
+
+ gst_collect_pads_remove_pad (mux->collect, pad);
+ gst_element_remove_pad (element, pad);
+}
+
+static GstFlowReturn
+gst_flv_mux_write_header (GstFlvMux * mux)
+{
+ GstBuffer *header = gst_buffer_new_and_alloc (9 + 4);
+ guint8 *data = GST_BUFFER_DATA (header);
+
+ if (GST_PAD_CAPS (mux->srcpad) == NULL) {
+ GstCaps *caps = gst_caps_new_simple ("video/x-flv", NULL);
+
+ gst_pad_set_caps (mux->srcpad, caps);
+ gst_caps_unref (caps);
+ }
+ gst_buffer_set_caps (header, GST_PAD_CAPS (mux->srcpad));
+
+ data[0] = 'F';
+ data[1] = 'L';
+ data[2] = 'V';
+ data[3] = 0x01; /* Version */
+
+ data[4] = (mux->have_audio << 2) | mux->have_video; /* flags */
+ GST_WRITE_UINT32_BE (data + 5, 9); /* data offset */
+
+ GST_WRITE_UINT32_BE (data + 9, 0); /* previous tag size */
+
+ return gst_pad_push (mux->srcpad, header);
+}
+
+static GstFlowReturn
+gst_flv_mux_write_buffer (GstFlvMux * mux, GstFlvPad * cpad)
+{
+ GstBuffer *tag;
+ guint8 *data;
+ guint size;
+ GstBuffer *buffer =
+ gst_collect_pads_pop (mux->collect, (GstCollectData *) cpad);
+ guint32 timestamp = GST_BUFFER_TIMESTAMP (buffer) / GST_MSECOND;
+ gboolean second_run = FALSE;
+
+next:
+ size = 11;
+ if (cpad->video) {
+ size += 1;
+ if (cpad->video_codec == 7 && !cpad->sent_codec_data)
+ size += 4 + GST_BUFFER_SIZE (cpad->video_codec_data);
+ else if (cpad->video_codec == 7)
+ size += 4 + GST_BUFFER_SIZE (buffer);
+ else
+ size += GST_BUFFER_SIZE (buffer);
+ } else {
+ size += 1;
+ if (cpad->video_codec == 10 && !cpad->sent_codec_data)
+ size += 1 + GST_BUFFER_SIZE (cpad->audio_codec_data);
+ else if (cpad->audio_codec == 10)
+ size += 1 + GST_BUFFER_SIZE (buffer);
+ else
+ size += GST_BUFFER_SIZE (buffer);
+ }
+ size += 4;
+
+ tag = gst_buffer_new_and_alloc (size);
+ data = GST_BUFFER_DATA (tag);
+ memset (data, 0, size);
+
+ data[0] = (cpad->video) ? 9 : 8;
+
+ data[1] = ((size - 11 - 4) >> 16) & 0xff;
+ data[2] = ((size - 11 - 4) >> 8) & 0xff;
+ data[3] = ((size - 11 - 4) >> 0) & 0xff;
+
+ data[4] = (timestamp >> 16) & 0xff;
+ data[5] = (timestamp >> 8) & 0xff;
+ data[6] = (timestamp >> 0) & 0xff;
+ data[7] = (timestamp >> 24) & 0xff;
+
+ data[8] = data[9] = data[10] = 0;
+
+ if (cpad->video) {
+ if (GST_BUFFER_FLAG_IS_SET (buffer, GST_BUFFER_FLAG_DELTA_UNIT))
+ data[11] |= 2 << 4;
+ else
+ data[11] |= 1 << 4;
+
+ data[11] |= cpad->video_codec & 0x0f;
+
+ if (cpad->video_codec == 7 && !cpad->sent_codec_data) {
+ data[12] = 0;
+ data[13] = data[14] = data[15] = 0;
+
+ memcpy (data + 11 + 1 + 4, GST_BUFFER_DATA (cpad->video_codec_data),
+ GST_BUFFER_SIZE (cpad->video_codec_data));
+ second_run = TRUE;
+ } else if (cpad->video_codec == 7) {
+ data[12] = 1;
+
+ /* FIXME: what to do about composition time */
+ data[13] = data[14] = data[15] = 0;
+
+ memcpy (data + 11 + 1 + 4, GST_BUFFER_DATA (buffer),
+ GST_BUFFER_SIZE (buffer));
+ } else {
+ memcpy (data + 11 + 1, GST_BUFFER_DATA (buffer),
+ GST_BUFFER_SIZE (buffer));
+ }
+ } else {
+ data[11] |= (cpad->audio_codec << 4) & 0xf0;
+ data[11] |= (cpad->rate << 2) & 0x0c;
+ data[11] |= (cpad->width << 1) & 0x02;
+ data[11] |= (cpad->channels << 0) & 0x01;
+
+ if (cpad->video_codec == 10 && !cpad->sent_codec_data) {
+ data[12] = 0;
+
+ memcpy (data + 11 + 1 + 1, GST_BUFFER_DATA (cpad->audio_codec_data),
+ GST_BUFFER_SIZE (cpad->audio_codec_data));
+ second_run = TRUE;
+ } else if (cpad->audio_codec == 10) {
+ data[12] = 1;
+
+ memcpy (data + 11 + 1 + 1, GST_BUFFER_DATA (buffer),
+ GST_BUFFER_SIZE (buffer));
+ } else {
+ memcpy (data + 11 + 1, GST_BUFFER_DATA (buffer),
+ GST_BUFFER_SIZE (buffer));
+ }
+ }
+
+ GST_WRITE_UINT32_BE (data + size - 4, size - 4);
+
+ gst_buffer_set_caps (tag, GST_PAD_CAPS (mux->srcpad));
+
+ if (second_run) {
+ GstFlowReturn ret;
+
+ second_run = FALSE;
+ cpad->sent_codec_data = TRUE;
+
+ ret = gst_pad_push (mux->srcpad, tag);
+ if (ret != GST_FLOW_OK) {
+ gst_buffer_unref (buffer);
+ return ret;
+ }
+
+ tag = NULL;
+ goto next;
+ }
+
+ gst_buffer_copy_metadata (tag, buffer, GST_BUFFER_COPY_TIMESTAMPS);
+ GST_BUFFER_OFFSET (tag) = GST_BUFFER_OFFSET_END (tag) =
+ GST_BUFFER_OFFSET_NONE;
+
+ gst_buffer_unref (buffer);
+
+ return gst_pad_push (mux->srcpad, tag);
+}
+
+static GstFlowReturn
+gst_flv_mux_collected (GstCollectPads * pads, gpointer user_data)
+{
+ GstFlvMux *mux = GST_FLV_MUX (user_data);
+ GstFlvPad *best;
+ GstClockTime best_time;
+ GstFlowReturn ret;
+ GSList *sl;
+ gboolean eos = TRUE;
+
+ if (mux->state == GST_FLV_MUX_STATE_HEADER) {
+ if (mux->collect->data == NULL) {
+ GST_ELEMENT_ERROR (mux, STREAM, MUX, (NULL),
+ ("No input streams configured"));
+ return GST_FLOW_ERROR;
+ }
+
+ if (gst_pad_push_event (mux->srcpad,
+ gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES, 0, -1, 0)))
+ ret = gst_flv_mux_write_header (mux);
+ else
+ ret = GST_FLOW_ERROR;
+
+ if (ret != GST_FLOW_OK)
+ return ret;
+ mux->state = GST_FLV_MUX_STATE_DATA;
+ }
+
+ best = NULL;
+ best_time = GST_CLOCK_TIME_NONE;
+ for (sl = mux->collect->data; sl; sl = sl->next) {
+ GstFlvPad *cpad = sl->data;
+ GstBuffer *buffer = gst_collect_pads_peek (pads, (GstCollectData *) cpad);
+ GstClockTime time;
+
+ if (!buffer)
+ continue;
+
+ eos = FALSE;
+
+ time = GST_BUFFER_TIMESTAMP (buffer);
+ gst_buffer_unref (buffer);
+
+ if (!GST_CLOCK_TIME_IS_VALID (time)) {
+ GST_WARNING_OBJECT (pads, "Buffer without valid timestamp, dropping");
+
+ gst_buffer_unref (gst_collect_pads_pop (pads, (GstCollectData *) cpad));
+ continue;
+ }
+
+
+ if (best == NULL || (GST_CLOCK_TIME_IS_VALID (best_time)
+ && time < best_time)) {
+ best = cpad;
+ best_time = time;
+ }
+ }
+
+ if (GST_CLOCK_TIME_IS_VALID (best_time)
+ && best_time / GST_MSECOND > G_MAXUINT32) {
+ GST_WARNING_OBJECT (mux, "Timestamp larger than FLV supports - EOS");
+ eos = TRUE;
+ }
+
+ if (!eos && best) {
+ return gst_flv_mux_write_buffer (mux, best);
+ } else if (eos) {
+ gst_pad_push_event (mux->srcpad, gst_event_new_eos ());
+ return GST_FLOW_UNEXPECTED;
+ } else {
+ return GST_FLOW_OK;
+ }
+}
+
+static GstStateChangeReturn
+gst_flv_mux_change_state (GstElement * element, GstStateChange transition)
+{
+ GstStateChangeReturn ret;
+ GstFlvMux *mux = GST_FLV_MUX (element);
+
+ switch (transition) {
+ case GST_STATE_CHANGE_NULL_TO_READY:
+ break;
+ case GST_STATE_CHANGE_READY_TO_PAUSED:
+ gst_collect_pads_start (mux->collect);
+ break;
+ case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
+ break;
+ case GST_STATE_CHANGE_PAUSED_TO_READY:
+ gst_collect_pads_stop (mux->collect);
+ break;
+ default:
+ break;
+ }
+
+ ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
+
+ switch (transition) {
+ case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
+ break;
+ case GST_STATE_CHANGE_PAUSED_TO_READY:
+ gst_flv_mux_reset (GST_ELEMENT (mux));
+ break;
+ case GST_STATE_CHANGE_READY_TO_NULL:
+ break;
+ default:
+ break;
+ }
+
+ return ret;
+}
diff --git a/gst/flv/gstflvmux.h b/gst/flv/gstflvmux.h
new file mode 100644
index 00000000..9ec0e5df
--- /dev/null
+++ b/gst/flv/gstflvmux.h
@@ -0,0 +1,84 @@
+/* GStreamer
+ *
+ * Copyright (c) 2008 Sebastian Dröge <sebastian.droege@collabora.co.uk>
+ *
+ * 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.
+ */
+
+#ifndef __GST_FLV_MUX_H__
+#define __GST_FLV_MUX_H__
+
+#include <gst/gst.h>
+#include <gst/base/gstcollectpads.h>
+
+G_BEGIN_DECLS
+
+#define GST_TYPE_FLV_MUX \
+ (gst_flv_mux_get_type ())
+#define GST_FLV_MUX(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_FLV_MUX, GstFlvMux))
+#define GST_FLV_MUX_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_FLV_MUX, GstFlvMuxClass))
+#define GST_IS_FLV_MUX(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_FLV_MUX))
+#define GST_IS_FLV_MUX_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_FLV_MUX))
+
+typedef struct
+{
+ GstCollectData collect;
+
+ gboolean video;
+
+ guint audio_codec;
+ guint rate;
+ guint width;
+ guint channels;
+ GstBuffer *audio_codec_data;
+
+ guint video_codec;
+ GstBuffer *video_codec_data;
+
+ gboolean sent_codec_data;
+} GstFlvPad;
+
+typedef enum
+{
+ GST_FLV_MUX_STATE_HEADER,
+ GST_FLV_MUX_STATE_DATA
+} GstFlvMuxState;
+
+typedef struct _GstFlvMux {
+ GstElement element;
+
+ GstPad *srcpad;
+ GstCollectPads *collect;
+ GstPadEventFunction collect_event;
+
+ GstFlvMuxState state;
+ gboolean have_audio;
+ gboolean have_video;
+} GstFlvMux;
+
+typedef struct _GstFlvMuxClass {
+ GstElementClass parent;
+} GstFlvMuxClass;
+
+GType gst_flv_mux_get_type (void);
+
+G_END_DECLS
+
+#endif /* __GST_FLV_MUX_H__ */