From 112269849176761667031cef41499fc6ae236a87 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= <slomo@circular-chaos.org>
Date: Mon, 22 Jul 2013 15:22:36 +0200
Subject: [PATCH] flvdemux: Add support for group-id in the stream-start event

---
 gst/flv/gstflvdemux.c | 36 ++++++++++++++++++++++++++++++++++--
 gst/flv/gstflvdemux.h |  3 +++
 2 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/gst/flv/gstflvdemux.c b/gst/flv/gstflvdemux.c
index 0e166bc75d..83095ee4c0 100644
--- a/gst/flv/gstflvdemux.c
+++ b/gst/flv/gstflvdemux.c
@@ -635,6 +635,26 @@ cleanup:
   return ret;
 }
 
+static gboolean
+have_group_id (GstFlvDemux * demux)
+{
+  GstEvent *event;
+
+  event = gst_pad_get_sticky_event (demux->sinkpad, GST_EVENT_STREAM_START, 0);
+  if (event) {
+    if (gst_event_parse_group_id (event, &demux->group_id))
+      demux->have_group_id = TRUE;
+    else
+      demux->have_group_id = FALSE;
+    gst_event_unref (event);
+  } else if (!demux->have_group_id) {
+    demux->have_group_id = TRUE;
+    demux->group_id = gst_util_group_id_next ();
+  }
+
+  return demux->have_group_id;
+}
+
 static gboolean
 gst_flv_demux_audio_negotiate (GstFlvDemux * demux, guint32 codec_tag,
     guint32 rate, guint32 channels, guint32 width)
@@ -643,6 +663,7 @@ gst_flv_demux_audio_negotiate (GstFlvDemux * demux, guint32 codec_tag,
   gchar *codec_name = NULL;
   gboolean ret = FALSE;
   guint adjusted_rate = rate;
+  GstEvent *event;
   gchar *stream_id;
 
   switch (codec_tag) {
@@ -786,7 +807,11 @@ gst_flv_demux_audio_negotiate (GstFlvDemux * demux, guint32 codec_tag,
   stream_id =
       gst_pad_create_stream_id (demux->audio_pad, GST_ELEMENT_CAST (demux),
       "audio");
-  gst_pad_push_event (demux->audio_pad, gst_event_new_stream_start (stream_id));
+
+  event = gst_event_new_stream_start (stream_id);
+  if (have_group_id (demux))
+    gst_event_set_group_id (event, demux->group_id);
+  gst_pad_push_event (demux->audio_pad, event);
   g_free (stream_id);
   ret = gst_pad_set_caps (demux->audio_pad, caps);
 
@@ -1178,6 +1203,7 @@ gst_flv_demux_video_negotiate (GstFlvDemux * demux, guint32 codec_tag)
   gboolean ret = FALSE;
   GstCaps *caps = NULL;
   gchar *codec_name = NULL;
+  GstEvent *event;
   gchar *stream_id;
 
   /* Generate caps for that pad */
@@ -1240,7 +1266,10 @@ gst_flv_demux_video_negotiate (GstFlvDemux * demux, guint32 codec_tag)
   stream_id =
       gst_pad_create_stream_id (demux->video_pad, GST_ELEMENT_CAST (demux),
       "video");
-  gst_pad_push_event (demux->video_pad, gst_event_new_stream_start (stream_id));
+  event = gst_event_new_stream_start (stream_id);
+  if (have_group_id (demux))
+    gst_event_set_group_id (event, demux->group_id);
+  gst_pad_push_event (demux->video_pad, event);
   ret = gst_pad_set_caps (demux->video_pad, caps);
 
   if (G_LIKELY (ret)) {
@@ -1768,6 +1797,9 @@ gst_flv_demux_cleanup (GstFlvDemux * demux)
 
   demux->state = FLV_STATE_HEADER;
 
+  demux->have_group_id = FALSE;
+  demux->group_id = G_MAXUINT;
+
   demux->flushing = FALSE;
   demux->need_header = TRUE;
   demux->audio_need_segment = TRUE;
diff --git a/gst/flv/gstflvdemux.h b/gst/flv/gstflvdemux.h
index 78a82ae717..90953b4a64 100644
--- a/gst/flv/gstflvdemux.h
+++ b/gst/flv/gstflvdemux.h
@@ -59,6 +59,9 @@ struct _GstFlvDemux
   GstPad *audio_pad;
   GstPad *video_pad;
 
+  gboolean have_group_id;
+  guint group_id;
+
   /* <private> */
   
   GstIndex *index;