From 6ab508dfa56f92c86c3b379b6cfe4ff28f2551e4 Mon Sep 17 00:00:00 2001
From: David Schleef <ds@schleef.org>
Date: Sun, 2 Jan 2011 19:19:27 -0800
Subject: [PATCH] multifilesink: send stream headers in key-frame mode

---
 gst/multifile/gstmultifilesink.c | 63 ++++++++++++++++++++++++++++++++++++++++
 gst/multifile/gstmultifilesink.h |  3 ++
 2 files changed, 66 insertions(+)

diff --git a/gst/multifile/gstmultifilesink.c b/gst/multifile/gstmultifilesink.c
index 2d4ab5b..b333e80 100644
--- a/gst/multifile/gstmultifilesink.c
+++ b/gst/multifile/gstmultifilesink.c
@@ -146,6 +146,8 @@ static void gst_multi_file_sink_get_property (GObject * object, guint prop_id,
 static gboolean gst_multi_file_sink_stop (GstBaseSink * sink);
 static GstFlowReturn gst_multi_file_sink_render (GstBaseSink * sink,
     GstBuffer * buffer);
+static gboolean gst_multi_file_sink_set_caps (GstBaseSink * sink,
+    GstCaps * caps);
 
 #define GST_TYPE_MULTI_FILE_SINK_NEXT (gst_multi_file_sink_next_get_type ())
 static GType
@@ -237,6 +239,8 @@ gst_multi_file_sink_class_init (GstMultiFileSinkClass * klass)
   gstbasesink_class->get_times = NULL;
   gstbasesink_class->stop = GST_DEBUG_FUNCPTR (gst_multi_file_sink_stop);
   gstbasesink_class->render = GST_DEBUG_FUNCPTR (gst_multi_file_sink_render);
+  gstbasesink_class->set_caps =
+      GST_DEBUG_FUNCPTR (gst_multi_file_sink_set_caps);
 }
 
 static void
@@ -327,6 +331,7 @@ static gboolean
 gst_multi_file_sink_stop (GstBaseSink * sink)
 {
   GstMultiFileSink *multifilesink;
+  int i;
 
   multifilesink = GST_MULTI_FILE_SINK (sink);
 
@@ -335,6 +340,13 @@ gst_multi_file_sink_stop (GstBaseSink * sink)
     multifilesink->file = NULL;
   }
 
+  if (multifilesink->streamheaders) {
+    for (i = 0; i < multifilesink->n_streamheaders; i++) {
+      gst_buffer_unref (multifilesink->streamheaders[i]);
+    }
+    g_free (multifilesink->streamheaders);
+  }
+
   return TRUE;
 }
 
@@ -462,6 +474,8 @@ gst_multi_file_sink_render (GstBaseSink * sink, GstBuffer * buffer)
       }
 
       if (multifilesink->file == NULL) {
+        int i;
+
         filename = g_strdup_printf (multifilesink->filename,
             multifilesink->index);
         multifilesink->file = g_fopen (filename, "wb");
@@ -469,6 +483,16 @@ gst_multi_file_sink_render (GstBaseSink * sink, GstBuffer * buffer)
 
         if (multifilesink->file == NULL)
           goto stdio_write_error;
+
+        if (multifilesink->streamheaders) {
+          for (i = 0; i < multifilesink->n_streamheaders; i++) {
+            ret = fwrite (GST_BUFFER_DATA (multifilesink->streamheaders[i]),
+                GST_BUFFER_SIZE (multifilesink->streamheaders[i]), 1,
+                multifilesink->file);
+            if (ret != 1)
+              goto stdio_write_error;
+          }
+        }
       }
 
       ret = fwrite (GST_BUFFER_DATA (buffer), GST_BUFFER_SIZE (buffer), 1,
@@ -508,3 +532,42 @@ stdio_write_error:
       ("Error while writing to file."), (NULL));
   return GST_FLOW_ERROR;
 }
+
+static gboolean
+gst_multi_file_sink_set_caps (GstBaseSink * sink, GstCaps * caps)
+{
+  GstMultiFileSink *multifilesink;
+  GstStructure *structure;
+
+  multifilesink = GST_MULTI_FILE_SINK (sink);
+
+  structure = gst_caps_get_structure (caps, 0);
+  if (structure) {
+    const GValue *value;
+
+    value = gst_structure_get_value (structure, "streamheader");
+
+    if (GST_VALUE_HOLDS_ARRAY (value)) {
+      int i;
+
+      if (multifilesink->streamheaders) {
+        for (i = 0; i < multifilesink->n_streamheaders; i++) {
+          gst_buffer_unref (multifilesink->streamheaders[i]);
+        }
+        g_free (multifilesink->streamheaders);
+      }
+
+      multifilesink->n_streamheaders = gst_value_array_get_size (value);
+      multifilesink->streamheaders =
+          g_malloc (sizeof (GstBuffer *) * multifilesink->n_streamheaders);
+
+      for (i = 0; i < multifilesink->n_streamheaders; i++) {
+        multifilesink->streamheaders[i] =
+            gst_buffer_ref (gst_value_get_buffer (gst_value_array_get_value
+                (value, i)));
+      }
+    }
+  }
+
+  return TRUE;
+}
diff --git a/gst/multifile/gstmultifilesink.h b/gst/multifile/gstmultifilesink.h
index cc4a4dd..c551570 100644
--- a/gst/multifile/gstmultifilesink.h
+++ b/gst/multifile/gstmultifilesink.h
@@ -69,6 +69,9 @@ struct _GstMultiFileSink
   FILE *file;
 
   gint64 next_segment;
+
+  int n_streamheaders;
+  GstBuffer **streamheaders;
 };
 
 struct _GstMultiFileSinkClass
-- 
2.7.4