diff --git a/gst/flv/gstflvmux.c b/gst/flv/gstflvmux.c index c1bd120526..1b717ed7a5 100644 --- a/gst/flv/gstflvmux.c +++ b/gst/flv/gstflvmux.c @@ -31,10 +31,6 @@ * */ -/* TODO: - * - Write metadata for the file, see FLV spec page 13 - */ - #ifdef HAVE_CONFIG_H #include "config.h" #endif @@ -76,7 +72,19 @@ static GstStaticPadTemplate audiosink_templ = GST_STATIC_PAD_TEMPLATE ("audio", "audio/x-speex, channels = (int) { 1, 2 }, rate = (int) { 5512, 11025, 22050, 44100 };") ); -GST_BOILERPLATE (GstFlvMux, gst_flv_mux, GstElement, GST_TYPE_ELEMENT); +#define _do_init(type) \ + G_STMT_START{ \ + static const GInterfaceInfo tag_setter_info = { \ + NULL, \ + NULL, \ + NULL \ + }; \ + g_type_add_interface_static (type, GST_TYPE_TAG_SETTER, \ + &tag_setter_info); \ + }G_STMT_END + +GST_BOILERPLATE_FULL (GstFlvMux, gst_flv_mux, GstElement, GST_TYPE_ELEMENT, + _do_init); static void gst_flv_mux_finalize (GObject * object); static GstFlowReturn @@ -171,6 +179,10 @@ gst_flv_mux_reset (GstElement * element) gst_collect_pads_remove_pad (mux->collect, cpad->collect.pad); } + if (mux->tags) + gst_tag_list_free (mux->tags); + mux->tags = NULL; + mux->state = GST_FLV_MUX_STATE_HEADER; } @@ -199,9 +211,20 @@ gst_flv_mux_handle_sink_event (GstPad * pad, GstEvent * event) gboolean ret = TRUE; switch (GST_EVENT_TYPE (event)) { - case GST_EVENT_TAG: - /* TODO do something with the tags */ + case GST_EVENT_TAG:{ + GstTagList *tags; + + if (!mux->tags) + mux->tags = gst_tag_list_new (); + + gst_event_parse_tag (event, &tags); + if (tags) { + gst_tag_list_insert (mux->tags, tags, + gst_tag_setter_get_tag_merge_mode (GST_TAG_SETTER (mux))); + } + break; + } case GST_EVENT_NEWSEGMENT: /* We don't support NEWSEGMENT events */ ret = FALSE; @@ -504,11 +527,201 @@ gst_flv_mux_release_pad (GstElement * element, GstPad * pad) gst_element_remove_pad (element, pad); } +static GstFlowReturn +gst_flv_mux_write_metadata (GstFlvMux * mux) +{ + GstTagList *merged_tags; + const GstTagList *user_tags; + GstFlowReturn ret = GST_FLOW_OK; + GstBuffer *script_tag, *tmp; + guint8 *data; + gint i, n_tags, tags_written = 0; + + user_tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (mux)); + GST_DEBUG_OBJECT (mux, "upstream tags = %" GST_PTR_FORMAT, mux->tags); + GST_DEBUG_OBJECT (mux, "user-set tags = %" GST_PTR_FORMAT, user_tags); + + merged_tags = gst_tag_list_merge (user_tags, mux->tags, + gst_tag_setter_get_tag_merge_mode (GST_TAG_SETTER (mux))); + + if (!merged_tags) { + GST_DEBUG_OBJECT (mux, "No tags to write"); + return GST_FLOW_OK; + } + + GST_DEBUG_OBJECT (mux, "merged tags = %" GST_PTR_FORMAT, merged_tags); + + script_tag = gst_buffer_new_and_alloc (11); + data = GST_BUFFER_DATA (script_tag); + + data[0] = 18; + + /* Data size, unknown for now */ + data[1] = 0; + data[2] = 0; + data[3] = 0; + + /* Timestamp */ + data[4] = data[5] = data[6] = data[7] = 0; + + /* Stream ID */ + data[8] = data[9] = data[10] = 0; + + tmp = gst_buffer_new_and_alloc (13); + data = GST_BUFFER_DATA (tmp); + data[0] = 2; /* string */ + data[1] = 0; + data[2] = 0x0a; /* length 10 */ + memcpy (&data[3], "onMetaData", sizeof ("onMetaData")); + + script_tag = gst_buffer_join (script_tag, tmp); + + n_tags = gst_structure_n_fields ((GstStructure *) merged_tags); + tmp = gst_buffer_new_and_alloc (5); + data = GST_BUFFER_DATA (tmp); + data[0] = 8; /* ECMA array */ + GST_WRITE_UINT32_BE (data + 1, n_tags); + script_tag = gst_buffer_join (script_tag, tmp); + + for (i = 0; i < n_tags; i++) { + const gchar *tag_name = + gst_structure_nth_field_name ((const GstStructure *) merged_tags, i); + if (!strcmp (tag_name, GST_TAG_DURATION)) { + gdouble d; + guint64 dur; + + if (!gst_tag_list_get_uint64 (merged_tags, GST_TAG_DURATION, &dur)) + continue; + + d = gst_guint64_to_gdouble (dur); + d /= (gdouble) GST_SECOND; + + tmp = gst_buffer_new_and_alloc (2 + 8 + 1 + 8); + data = GST_BUFFER_DATA (tmp); + data[0] = 0; /* 8 bytes name */ + data[1] = 8; + memcpy (&data[2], "duration", sizeof ("duration")); + data[10] = 0; /* double */ + GST_WRITE_DOUBLE_BE (data + 11, d); + script_tag = gst_buffer_join (script_tag, tmp); + tags_written++; + } else if (!strcmp (tag_name, GST_TAG_ARTIST) || + !strcmp (tag_name, GST_TAG_TITLE)) { + gchar *s; + const gchar *t; + + if (!strcmp (tag_name, GST_TAG_ARTIST)) + t = "creator"; + else if (!strcmp (tag_name, GST_TAG_TITLE)) + t = "title"; + + if (!gst_tag_list_get_string (merged_tags, tag_name, &s)) + continue; + + tmp = gst_buffer_new_and_alloc (2 + strlen (t) + 1 + 2 + strlen (s)); + data = GST_BUFFER_DATA (tmp); + data[0] = 0; /* tag name length */ + data[1] = strlen (t); + memcpy (&data[2], t, strlen (t)); + data[2 + strlen (t)] = 2; /* string */ + data[3 + strlen (t)] = (strlen (s) >> 8) & 0xff; + data[4 + strlen (t)] = (strlen (s)) & 0xff; + memcpy (&data[5 + strlen (t)], s, strlen (s)); + script_tag = gst_buffer_join (script_tag, tmp); + + g_free (s); + tags_written++; + } + } + + { + const gchar *s = "GStreamer FLV muxer"; + + tmp = gst_buffer_new_and_alloc (2 + 15 + 1 + 2 + strlen (s)); + data = GST_BUFFER_DATA (tmp); + data[0] = 0; /* 15 bytes name */ + data[1] = 15; + memcpy (&data[2], "metadatacreator", sizeof ("metadatacreator")); + data[17] = 2; /* string */ + data[18] = (strlen (s) >> 8) & 0xff; + data[19] = (strlen (s)) & 0xff; + memcpy (&data[20], s, strlen (s)); + script_tag = gst_buffer_join (script_tag, tmp); + + tags_written++; + } + + { + GTimeVal tv = { 0, }; + time_t secs; + struct tm *tm; + gchar *s; + static const gchar *weekdays[] = { + "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat" + }; + static const gchar *months[] = { + "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", + "Aug", "Sep", "Oct", "Nov", "Dec" + }; + + g_get_current_time (&tv); + secs = tv.tv_sec; + tm = gmtime (&secs); + + s = g_strdup_printf ("%s %s %d %d:%d:%d %d", weekdays[tm->tm_wday], + months[tm->tm_mon], tm->tm_mday, tm->tm_hour, tm->tm_min, tm->tm_sec, + tm->tm_year + 1900); + + tmp = gst_buffer_new_and_alloc (2 + 12 + 1 + 2 + strlen (s)); + data = GST_BUFFER_DATA (tmp); + data[0] = 0; /* 12 bytes name */ + data[1] = 12; + memcpy (&data[2], "creationdate", sizeof ("creationdate")); + data[14] = 2; /* string */ + data[15] = (strlen (s) >> 8) & 0xff; + data[16] = (strlen (s)) & 0xff; + memcpy (&data[17], s, strlen (s)); + script_tag = gst_buffer_join (script_tag, tmp); + + g_free (s); + tags_written++; + } + + tmp = gst_buffer_new_and_alloc (2 + 0 + 1); + data = GST_BUFFER_DATA (tmp); + data[0] = 0; /* 0 byte size */ + data[1] = 0; + data[2] = 9; /* end marker */ + script_tag = gst_buffer_join (script_tag, tmp); + tags_written++; + + + tmp = gst_buffer_new_and_alloc (4); + data = GST_BUFFER_DATA (tmp); + GST_WRITE_UINT32_BE (data, GST_BUFFER_SIZE (script_tag)); + script_tag = gst_buffer_join (script_tag, tmp); + + data = GST_BUFFER_DATA (script_tag); + data[1] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 16) & 0xff; + data[2] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 8) & 0xff; + data[3] = ((GST_BUFFER_SIZE (script_tag) - 11 - 4) >> 0) & 0xff; + + GST_WRITE_UINT32_BE (data + 11 + 13 + 1, tags_written); + + gst_buffer_set_caps (script_tag, GST_PAD_CAPS (mux->srcpad)); + ret = gst_pad_push (mux->srcpad, script_tag); + + gst_tag_list_free (merged_tags); + + return ret; +} + static GstFlowReturn gst_flv_mux_write_header (GstFlvMux * mux) { GstBuffer *header = gst_buffer_new_and_alloc (9 + 4); guint8 *data = GST_BUFFER_DATA (header); + GstFlowReturn ret; if (GST_PAD_CAPS (mux->srcpad) == NULL) { GstCaps *caps = gst_caps_new_simple ("video/x-flv", NULL); @@ -528,7 +741,11 @@ gst_flv_mux_write_header (GstFlvMux * mux) GST_WRITE_UINT32_BE (data + 9, 0); /* previous tag size */ - return gst_pad_push (mux->srcpad, header); + ret = gst_pad_push (mux->srcpad, header); + if (ret != GST_FLOW_OK) + return ret; + + return gst_flv_mux_write_metadata (mux); } static GstFlowReturn diff --git a/gst/flv/gstflvmux.h b/gst/flv/gstflvmux.h index 3d634a4c52..02df089d45 100644 --- a/gst/flv/gstflvmux.h +++ b/gst/flv/gstflvmux.h @@ -74,6 +74,8 @@ typedef struct _GstFlvMux { GstFlvMuxState state; gboolean have_audio; gboolean have_video; + + GstTagList *tags; } GstFlvMux; typedef struct _GstFlvMuxClass {