s/gst_element_install_std_props/gst_element_class_install_std_props/ -- it just makes...
[platform/upstream/gstreamer.git] / gst / elements / gstdisksink.c
index 0861149..93efdf3 100644 (file)
@@ -22,8 +22,9 @@
 
 
 #include <gst/gst.h>
+#include <errno.h>
 #include "gstdisksink.h"
-
+#include <string.h>
 
 GstElementDetails gst_disksink_details = {
   "Disk Sink",
@@ -45,42 +46,46 @@ enum {
 enum {
   ARG_0,
   ARG_LOCATION,
+  ARG_MAXFILESIZE,
 };
 
 
-static void    gst_disksink_class_init (GstDiskSinkClass *klass);
-static void    gst_disksink_init       (GstDiskSink *disksink);
+static void    gst_disksink_class_init         (GstDiskSinkClass *klass);
+static void    gst_disksink_init               (GstDiskSink *disksink);
 
-static void    gst_disksink_set_arg    (GtkObject *object, GtkArg *arg, guint id);
-static void    gst_disksink_get_arg    (GtkObject *object, GtkArg *arg, guint id);
+static void    gst_disksink_set_property       (GObject *object, guint prop_id, 
+                                                const GValue *value, GParamSpec *pspec);
+static void    gst_disksink_get_property       (GObject *object, guint prop_id, 
+                                                GValue *value, GParamSpec *pspec);
 
-static gboolean gst_disksink_open_file         (GstDiskSink *sink);
-static void    gst_disksink_close_file (GstDiskSink *sink);
+static gboolean gst_disksink_open_file                 (GstDiskSink *sink);
+static void    gst_disksink_close_file         (GstDiskSink *sink);
 
-static void    gst_disksink_chain      (GstPad *pad,GstBuffer *buf);
+static gboolean gst_disksink_handle_event       (GstPad *pad, GstEvent *event);
+static void    gst_disksink_chain              (GstPad *pad,GstBuffer *buf);
 
 static GstElementStateReturn gst_disksink_change_state (GstElement *element);
 
 static GstElementClass *parent_class = NULL;
 static guint gst_disksink_signals[LAST_SIGNAL] = { 0 };
 
-GtkType
+GType
 gst_disksink_get_type (void) 
 {
-  static GtkType disksink_type = 0;
+  static GType disksink_type = 0;
 
   if (!disksink_type) {
-    static const GtkTypeInfo disksink_info = {
-      "GstDiskSink",
+    static const GTypeInfo disksink_info = {
+      sizeof(GstDiskSinkClass),      NULL,
+      NULL,
+      (GClassInitFunc)gst_disksink_class_init,
+      NULL,
+      NULL,
       sizeof(GstDiskSink),
-      sizeof(GstDiskSinkClass),
-      (GtkClassInitFunc)gst_disksink_class_init,
-      (GtkObjectInitFunc)gst_disksink_init,
-      (GtkArgSetFunc)gst_disksink_set_arg,
-      (GtkArgGetFunc)gst_disksink_get_arg,
-      (GtkClassInitFunc)NULL,  /* deprecated, do not use ! */
+      0,
+      (GInstanceInitFunc)gst_disksink_init,
     };
-    disksink_type = gtk_type_unique (GST_TYPE_ELEMENT, &disksink_info);
+    disksink_type = g_type_register_static (GST_TYPE_ELEMENT, "GstDiskSink", &disksink_info, 0);
   }
   return disksink_type;
 }
@@ -88,27 +93,30 @@ gst_disksink_get_type (void)
 static void
 gst_disksink_class_init (GstDiskSinkClass *klass) 
 {
-  GtkObjectClass *gtkobject_class;
+  GObjectClass *gobject_class;
   GstElementClass *gstelement_class;
 
-  gtkobject_class = (GtkObjectClass*)klass;
+  gobject_class = (GObjectClass*)klass;
   gstelement_class = (GstElementClass*)klass;
 
-  parent_class = gtk_type_class (GST_TYPE_ELEMENT);
+  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
 
-  gtk_object_add_arg_type ("GstDiskSink::location", GST_TYPE_FILENAME,
-                           GTK_ARG_READWRITE, ARG_LOCATION);
+  gst_element_class_install_std_props (
+         GST_ELEMENT_CLASS (klass),
+         "location", ARG_LOCATION, G_PARAM_READWRITE,
+         NULL);
 
-  gst_disksink_signals[SIGNAL_HANDOFF] =
-    gtk_signal_new ("handoff", GTK_RUN_LAST, gtkobject_class->type,
-                    GTK_SIGNAL_OFFSET (GstDiskSinkClass, handoff),
-                    gtk_marshal_NONE__NONE, GTK_TYPE_NONE, 0);
+  g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_MAXFILESIZE,
+    g_param_spec_int("maxfilesize","MaxFileSize","Maximum Size Per File",
+    G_MININT,G_MAXINT,0,G_PARAM_READWRITE));
 
-  gtk_object_class_add_signals (gtkobject_class, gst_disksink_signals,
-                                    LAST_SIGNAL);
+  gst_disksink_signals[SIGNAL_HANDOFF] =
+    g_signal_new ("handoff", G_TYPE_FROM_CLASS(klass), G_SIGNAL_RUN_LAST,
+                    G_STRUCT_OFFSET (GstDiskSinkClass, handoff), NULL, NULL,
+                    g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0);
 
-  gtkobject_class->set_arg = gst_disksink_set_arg;
-  gtkobject_class->get_arg = gst_disksink_get_arg;
+  gobject_class->set_property = gst_disksink_set_property;
+  gobject_class->get_property = gst_disksink_get_property;
 
   gstelement_class->change_state = gst_disksink_change_state;
 }
@@ -117,27 +125,66 @@ static void
 gst_disksink_init (GstDiskSink *disksink) 
 {
   GstPad *pad;
+
   pad = gst_pad_new ("sink", GST_PAD_SINK);
   gst_element_add_pad (GST_ELEMENT (disksink), pad);
   gst_pad_set_chain_function (pad, gst_disksink_chain);
 
+  GST_FLAG_SET (GST_ELEMENT(disksink), GST_ELEMENT_EVENT_AWARE);
+  gst_pad_set_event_function(pad, gst_disksink_handle_event);
+
   disksink->filename = NULL;
   disksink->file = NULL;
+  disksink->filenum = 0;
+
+  disksink->maxfilesize = -1;
+}
+
+static char *
+gst_disksink_getcurrentfilename (GstDiskSink *disksink)
+{
+  g_return_val_if_fail(disksink != NULL, NULL);
+  g_return_val_if_fail(GST_IS_DISKSINK(disksink), NULL);
+  g_return_val_if_fail(disksink->filename != NULL, NULL);
+  g_return_val_if_fail(disksink->filenum >= 0, NULL);
+
+  if (!strstr(disksink->filename, "%"))
+  {
+    if (!disksink->filenum)
+      return g_strdup(disksink->filename);
+    else
+      return NULL;
+  }
+
+  return g_strdup_printf(disksink->filename, disksink->filenum);
 }
 
 static void
-gst_disksink_set_arg (GtkObject *object, GtkArg *arg, guint id)
+gst_disksink_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
 {
   GstDiskSink *sink;
 
   /* it's not null if we got it, but it might not be ours */
   sink = GST_DISKSINK (object);
 
-  switch(id) {
+  switch (prop_id) {
     case ARG_LOCATION:
+      /* the element must be stopped or paused in order to do this */
+      g_return_if_fail ((GST_STATE (sink) < GST_STATE_PLAYING)
+                      || (GST_STATE (sink) == GST_STATE_PAUSED));
       if (sink->filename)
        g_free (sink->filename);
-      sink->filename = g_strdup (GTK_VALUE_STRING (*arg));
+      sink->filename = g_strdup (g_value_get_string (value));
+      if ( (GST_STATE (sink) == GST_STATE_PAUSED) 
+        && (sink->filename != NULL))
+      {
+              gst_disksink_close_file (sink);
+              gst_disksink_open_file (sink);   
+      }
+      break;
+    case ARG_MAXFILESIZE:
+      sink->maxfilesize = g_value_get_int(value);
       break;
     default:
       break;
@@ -145,7 +192,7 @@ gst_disksink_set_arg (GtkObject *object, GtkArg *arg, guint id)
 }
 
 static void   
-gst_disksink_get_arg (GtkObject *object, GtkArg *arg, guint id)
+gst_disksink_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 {
   GstDiskSink *sink;
  
@@ -154,12 +201,15 @@ gst_disksink_get_arg (GtkObject *object, GtkArg *arg, guint id)
  
   sink = GST_DISKSINK (object);
   
-  switch (id) {
+  switch (prop_id) {
     case ARG_LOCATION:
-      GTK_VALUE_STRING (*arg) = sink->filename;
+      g_value_set_string (value, gst_disksink_getcurrentfilename(sink));
+      break;
+    case ARG_MAXFILESIZE:
+      g_value_set_int (value, sink->maxfilesize);
       break;
     default:
-      arg->type = GTK_TYPE_INVALID;
+      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }
 }
@@ -170,15 +220,24 @@ gst_disksink_open_file (GstDiskSink *sink)
   g_return_val_if_fail (!GST_FLAG_IS_SET (sink, GST_DISKSINK_OPEN), FALSE);
 
   /* open the file */
-  sink->file = fopen (sink->filename, "w");
+  if (!gst_disksink_getcurrentfilename(sink))
+  {
+    /* Out of files */
+    gst_element_set_eos(GST_ELEMENT(sink));
+    return FALSE;
+  }
+  sink->file = fopen (gst_disksink_getcurrentfilename(sink), "w");
   if (sink->file == NULL) {
     perror ("open");
-    gst_element_error (GST_ELEMENT (sink), g_strconcat("opening file \"", sink->filename, "\"", NULL));
+    gst_element_error (GST_ELEMENT (sink), g_strconcat("Error opening file \"",
+      gst_disksink_getcurrentfilename(sink), "\": ", sys_errlist[errno], NULL));
     return FALSE;
   } 
 
   GST_FLAG_SET (sink, GST_DISKSINK_OPEN);
 
+  sink->data_written = 0;
+
   return TRUE;
 }
 
@@ -190,13 +249,69 @@ gst_disksink_close_file (GstDiskSink *sink)
   if (fclose (sink->file) != 0)
   {
     perror ("close");
-    gst_element_error (GST_ELEMENT (sink), g_strconcat("closing file \"", sink->filename, "\"", NULL));
+    gst_element_error (GST_ELEMENT (sink), g_strconcat("Error closing file \"",
+      gst_disksink_getcurrentfilename(sink), "\": ", sys_errlist[errno], NULL));
   }
   else {
     GST_FLAG_UNSET (sink, GST_DISKSINK_OPEN);
   }
 }
 
+/* handle events (search) */
+static gboolean
+gst_disksink_handle_event (GstPad *pad, GstEvent *event)
+{
+  GstEventType type;
+  GstDiskSink *disksink;
+
+  disksink = GST_DISKSINK (gst_pad_get_parent (pad));
+
+  type = event ? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN;
+
+  switch (type) {
+    case GST_EVENT_SEEK:
+      /* we need to seek */
+      if (GST_EVENT_SEEK_FLUSH(event))
+        if (fflush(disksink->file))
+          gst_element_error(GST_ELEMENT(disksink),
+            "Error flushing the buffer cache of file \'%s\' to disk: %s",
+            gst_disksink_getcurrentfilename(disksink), sys_errlist[errno]);
+      switch (GST_EVENT_SEEK_TYPE(event))
+      {
+        case GST_SEEK_BYTEOFFSET_SET:
+          fseek(disksink->file, GST_EVENT_SEEK_OFFSET(event), SEEK_SET);
+          break;
+        case GST_SEEK_BYTEOFFSET_CUR:
+          fseek(disksink->file, GST_EVENT_SEEK_OFFSET(event), SEEK_CUR);
+          break;
+        case GST_SEEK_BYTEOFFSET_END:
+          fseek(disksink->file, GST_EVENT_SEEK_OFFSET(event), SEEK_END);
+          break;
+        default:
+          g_warning("Any other then byte-offset seeking is not supported!\n");
+          break;
+      }
+      break;
+    case GST_EVENT_NEW_MEDIA:
+      /* we need to open a new file! */
+      gst_disksink_close_file(disksink);
+      disksink->filenum++;
+      if (!gst_disksink_open_file(disksink)) return FALSE;
+      break;
+    case GST_EVENT_FLUSH:
+      if (fflush(disksink->file))
+        gst_element_error(GST_ELEMENT(disksink),
+          "Error flushing the buffer cache of file \'%s\' to disk: %s",
+          gst_disksink_getcurrentfilename(disksink), sys_errlist[errno]);
+      break;
+    default:
+      gst_pad_event_default (pad, event);
+      break;
+  }
+
+  return TRUE;
+}
+
 /**
  * gst_disksink_chain:
  * @pad: the pad this disksink is connected to
@@ -208,7 +323,7 @@ static void
 gst_disksink_chain (GstPad *pad, GstBuffer *buf) 
 {
   GstDiskSink *disksink;
-  guint16 bytes_written = 0;
+  gint bytes_written = 0;
 
   g_return_if_fail (pad != NULL);
   g_return_if_fail (GST_IS_PAD (pad));
@@ -216,6 +331,25 @@ gst_disksink_chain (GstPad *pad, GstBuffer *buf)
 
   disksink = GST_DISKSINK (gst_pad_get_parent (pad));
 
+  if (GST_IS_EVENT(buf))
+  {
+    gst_disksink_handle_event(pad, GST_EVENT(buf));
+    return;
+  }
+
+  if (disksink->maxfilesize > 0)
+  {
+    if ((disksink->data_written + GST_BUFFER_SIZE(buf))/(1024*1024) > disksink->maxfilesize)
+    {
+      if (GST_ELEMENT_IS_EVENT_AWARE(GST_ELEMENT(disksink)))
+      {
+        GstEvent *event;
+        event = gst_event_new(GST_EVENT_NEW_MEDIA);
+        gst_pad_send_event(pad, event);
+      }
+    }
+  }
+
   if (GST_FLAG_IS_SET (disksink, GST_DISKSINK_OPEN))
   {
     bytes_written = fwrite (GST_BUFFER_DATA (buf), 1, GST_BUFFER_SIZE (buf), disksink->file);
@@ -225,9 +359,11 @@ gst_disksink_chain (GstPad *pad, GstBuffer *buf)
                  GST_BUFFER_SIZE (buf), bytes_written);
     }
   }
+  disksink->data_written += GST_BUFFER_SIZE(buf);
+
   gst_buffer_unref (buf);
 
-  gtk_signal_emit (GTK_OBJECT (disksink), gst_disksink_signals[SIGNAL_HANDOFF],
+  g_signal_emit (G_OBJECT (disksink), gst_disksink_signals[SIGNAL_HANDOFF], 0,
                              disksink);
 }