From f012ab67cc31fa73972485c5cfc193fd25323f44 Mon Sep 17 00:00:00 2001
From: Zaheer Abbas Merali <zaheerabbas@merali.org>
Date: Sun, 10 Oct 2010 14:43:58 +0100
Subject: [PATCH] matroskamux: set offsets on outgoing buffers

---
 gst/matroska/ebml-write.c | 7 ++++++-
 gst/matroska/ebml-write.h | 1 +
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/gst/matroska/ebml-write.c b/gst/matroska/ebml-write.c
index 5665ca2c16..8a7209f8c9 100644
--- a/gst/matroska/ebml-write.c
+++ b/gst/matroska/ebml-write.c
@@ -65,7 +65,7 @@ gst_ebml_write_init (GstEbmlWrite * ebml, GstEbmlWriteClass * klass)
   ebml->streamheader = NULL;
   ebml->streamheader_pos = 0;
   ebml->writing_streamheader = FALSE;
-
+  ebml->current_offset = 0;
   ebml->caps = NULL;
 }
 
@@ -135,6 +135,7 @@ gst_ebml_write_reset (GstEbmlWrite * ebml)
   ebml->last_write_result = GST_FLOW_OK;
   ebml->timestamp = GST_CLOCK_TIME_NONE;
   ebml->need_newsegment = TRUE;
+  ebml->current_offset = 0;
 }
 
 
@@ -223,6 +224,8 @@ gst_ebml_write_flush_cache (GstEbmlWrite * ebml, gboolean is_keyframe)
   ebml->cache = NULL;
   GST_DEBUG ("Flushing cache of size %d", GST_BUFFER_SIZE (buffer));
   gst_buffer_set_caps (buffer, ebml->caps);
+  GST_BUFFER_OFFSET (buffer) = ebml->current_offset++;
+  GST_BUFFER_OFFSET_END (buffer) = ebml->current_offset;
   if (ebml->last_write_result == GST_FLOW_OK) {
     if (ebml->need_newsegment) {
       GstEvent *ev;
@@ -402,6 +405,8 @@ gst_ebml_write_element_push (GstEbmlWrite * ebml, GstBuffer * buf)
     }
     buf = gst_buffer_make_metadata_writable (buf);
     gst_buffer_set_caps (buf, ebml->caps);
+    GST_BUFFER_OFFSET (buf) = ebml->current_offset++;
+    GST_BUFFER_OFFSET_END (buf) = ebml->current_offset;
     if (ebml->writing_streamheader) {
       GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_IN_CAPS);
     }
diff --git a/gst/matroska/ebml-write.h b/gst/matroska/ebml-write.h
index 2240f4f969..852d6c8e3e 100644
--- a/gst/matroska/ebml-write.h
+++ b/gst/matroska/ebml-write.h
@@ -59,6 +59,7 @@ typedef struct _GstEbmlWrite {
   gboolean writing_streamheader;
   GstByteWriter *streamheader;
   guint64 streamheader_pos;
+  guint64 current_offset;
 
   GstCaps *caps;
 } GstEbmlWrite;