From c3aae3dc1755b0f9881617e137f6f7c8eda0a5f5 Mon Sep 17 00:00:00 2001
From: Thiago Santos <thiago.sousa.santos@collabora.co.uk>
Date: Wed, 16 Feb 2011 16:07:49 -0300
Subject: [PATCH] oggmux: Keep track of pad's segments

https://bugzilla.gnome.org/show_bug.cgi?id=643775
---
 ext/ogg/gstoggmux.c | 36 ++++++++++++++++++++++++++++++++----
 ext/ogg/gstoggmux.h |  2 ++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/ext/ogg/gstoggmux.c b/ext/ogg/gstoggmux.c
index f2194f5119..ae6fdde253 100644
--- a/ext/ogg/gstoggmux.c
+++ b/ext/ogg/gstoggmux.c
@@ -292,6 +292,11 @@ gst_ogg_mux_ogg_pad_destroy_notify (GstCollectData * data)
     g_queue_free (oggpad->pagebuffers);
     oggpad->pagebuffers = NULL;
   }
+
+  if (oggpad->segment) {
+    gst_segment_free (oggpad->segment);
+    oggpad->segment = NULL;
+  }
 }
 
 static GstPadLinkReturn
@@ -319,11 +324,28 @@ gst_ogg_mux_sink_event (GstPad * pad, GstEvent * event)
       GST_DEBUG_PAD_NAME (pad));
 
   switch (GST_EVENT_TYPE (event)) {
-    case GST_EVENT_NEWSEGMENT:
-      /* We don't support NEWSEGMENT events */
-      gst_event_unref (event);
-      ret = FALSE;
+    case GST_EVENT_NEWSEGMENT:{
+      gboolean update;
+      gdouble rate;
+      gdouble applied_rate;
+      GstFormat format;
+      gint64 start, stop, position;
+
+      g_return_val_if_fail (ogg_pad->segment != NULL, FALSE);
+
+      gst_event_parse_new_segment_full (event, &update, &rate,
+          &applied_rate, &format, &start, &stop, &position);
+
+      /* We don't support non time NEWSEGMENT events */
+      if (format != GST_FORMAT_TIME) {
+        gst_event_unref (event);
+        return FALSE;
+      }
+      gst_segment_set_newsegment_full (ogg_pad->segment, update, rate,
+          applied_rate, format, start, stop, position);
+
       break;
+    }
     default:
       ret = TRUE;
       break;
@@ -427,6 +449,9 @@ gst_ogg_mux_request_new_pad (GstElement * element,
       oggpad->pagebuffers = g_queue_new ();
       oggpad->map.headers = NULL;
       oggpad->map.queued = NULL;
+      oggpad->segment = gst_segment_new ();
+      gst_segment_set_newsegment (oggpad->segment, FALSE, 1, GST_FORMAT_TIME,
+          0, -1, 0);
 
       oggpad->collect_event = (GstPadEventFunction) GST_PAD_EVENTFUNC (newpad);
       gst_pad_set_event_function (newpad,
@@ -1674,6 +1699,9 @@ gst_ogg_mux_init_collectpads (GstCollectPads * collect)
     oggpad->prev_delta = FALSE;
     oggpad->data_pushed = FALSE;
     oggpad->pagebuffers = g_queue_new ();
+    oggpad->segment = gst_segment_new ();
+    gst_segment_set_newsegment (oggpad->segment, FALSE, 1, GST_FORMAT_TIME,
+        0, -1, 0);
 
     walk = g_slist_next (walk);
   }
diff --git a/ext/ogg/gstoggmux.h b/ext/ogg/gstoggmux.h
index 6812f82ca2..62ac3ce5b7 100644
--- a/ext/ogg/gstoggmux.h
+++ b/ext/ogg/gstoggmux.h
@@ -53,6 +53,8 @@ typedef struct
   GstOggStream map;
   gboolean have_type;
 
+  GstSegment *segment;
+
   /* These two buffers make a very simple queue - they enter as 'next_buffer'
    * and (usually) leave as 'buffer', except at EOS, when buffer will be NULL */
   GstBuffer *buffer;            /* the first waiting buffer for the pad */