#include <gst/gst.h>
+#include <errno.h>
#include "gstdisksink.h"
-
+#include <string.h>
GstElementDetails gst_disksink_details = {
"Disk Sink",
enum {
ARG_0,
ARG_LOCATION,
- ARG_CLOSED/*,
- ARG_SILENT,
- ARG_BYTESPERWRITE,
- ARG_NUM_SOURCES,
- */
+ 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_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 void gst_disksink_set_arg (GtkObject *object, GtkArg *arg, guint id);
-static void gst_disksink_get_arg (GtkObject *object, GtkArg *arg, guint id);
+static gboolean gst_disksink_handle_event (GstPad *pad, GstEvent *event);
+static void gst_disksink_chain (GstPad *pad,GstBuffer *buf);
-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, /* QUESTION : why null ? otherwise coredump */
+ 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;
}
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);
+ gst_element_class_install_std_props (
+ GST_ELEMENT_CLASS (klass),
+ "location", ARG_LOCATION, G_PARAM_READWRITE,
+ NULL);
- gtk_object_add_arg_type ("GstDiskSink::location", GST_TYPE_FILENAME,
- GTK_ARG_READWRITE, ARG_LOCATION);
-/*
- gtk_object_add_arg_type ("GstDiskSink::silent", GTK_TYPE_BOOL,
- GTK_ARG_READWRITE, ARG_SILENT);
-*/
- gtk_object_add_arg_type ("GstDiskSink::closed", GTK_TYPE_BOOL,
- GTK_ARG_READWRITE, ARG_CLOSED);
-
+ 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));
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);
-
- gtk_object_class_add_signals (gtkobject_class, gst_disksink_signals,
- LAST_SIGNAL);
+ 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);
+ gobject_class->set_property = gst_disksink_set_property;
+ gobject_class->get_property = gst_disksink_get_property;
- gtkobject_class->set_arg = gst_disksink_set_arg;
- gtkobject_class->get_arg = gst_disksink_get_arg;
+ gstelement_class->change_state = gst_disksink_change_state;
}
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);
- disksink->opened = FALSE;
+
+ 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->silent = FALSE; ? what's this ? it's for output !
+ 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 in order to do this */
- g_return_if_fail (GST_STATE (sink) < GST_STATE_PLAYING);
-
- if (sink->filename) g_free (sink->filename);
-
- sink->filename = g_strdup (GTK_VALUE_STRING (*arg));
- sink->file = fopen (GTK_VALUE_STRING (*arg), "w");
- if (sink->file == NULL)
+ /* 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 (g_value_get_string (value));
+ if ( (GST_STATE (sink) == GST_STATE_PAUSED)
+ && (sink->filename != NULL))
{
- g_error ("Cannot open %s for writing !\n", GTK_VALUE_STRING (*arg));
- //exit (-2);
+ gst_disksink_close_file (sink);
+ gst_disksink_open_file (sink);
}
- else sink->opened = TRUE;
- gst_element_set_state(GST_ELEMENT(sink),GST_STATE_READY);
- break;
- /*
- case ARG_SILENT:
- sink->silent = GTK_VALUE_BOOL (*arg);
+
break;
- */
- case ARG_CLOSED:
- if (GTK_VALUE_BOOL (*arg) == TRUE)
- {
- /* close the file descriptor */
- sink->opened = FALSE;
- if (! (fclose (sink->file)))
- {
- g_warning ("Cannot close file !\n");
- }
- }
+ case ARG_MAXFILESIZE:
+ sink->maxfilesize = g_value_get_int(value);
break;
-
default:
break;
}
}
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;
sink = GST_DISKSINK (object);
- switch (id) {
- /*
- case ARG_SILENT:
- GTK_VALUE_BOOL (*arg) = sink->silent;
+ switch (prop_id) {
+ case ARG_LOCATION:
+ g_value_set_string (value, gst_disksink_getcurrentfilename(sink));
break;
- */
- case ARG_CLOSED:
- GTK_VALUE_BOOL (*arg) = !sink->opened;
+ case ARG_MAXFILESIZE:
+ g_value_set_int (value, sink->maxfilesize);
break;
- case ARG_LOCATION:
- GTK_VALUE_STRING (*arg) = sink->filename;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ }
+}
+
+static gboolean
+gst_disksink_open_file (GstDiskSink *sink)
+{
+ g_return_val_if_fail (!GST_FLAG_IS_SET (sink, GST_DISKSINK_OPEN), FALSE);
+
+ /* open the file */
+ 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("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;
+}
+
+static void
+gst_disksink_close_file (GstDiskSink *sink)
+{
+ g_return_if_fail (GST_FLAG_IS_SET (sink, GST_DISKSINK_OPEN));
+
+ if (fclose (sink->file) != 0)
+ {
+ perror ("close");
+ 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:
- arg->type = GTK_TYPE_INVALID;
+ gst_pad_event_default (pad, event);
break;
}
+
+ return TRUE;
}
/**
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));
g_return_if_fail (buf != NULL);
disksink = GST_DISKSINK (gst_pad_get_parent (pad));
-/*
- if (!disksink->silent)
- g_print("disksink: ******* (%s:%s)< \n",GST_DEBUG_PAD_NAME(pad));
-*/
- if (disksink->opened)
+
+ 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);
if (bytes_written < GST_BUFFER_SIZE (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);
}
+
+static GstElementStateReturn
+gst_disksink_change_state (GstElement *element)
+{
+ g_return_val_if_fail (GST_IS_DISKSINK (element), GST_STATE_FAILURE);
+
+ if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
+ if (GST_FLAG_IS_SET (element, GST_DISKSINK_OPEN))
+ gst_disksink_close_file (GST_DISKSINK (element));
+ } else {
+ if (!GST_FLAG_IS_SET (element, GST_DISKSINK_OPEN)) {
+ if (!gst_disksink_open_file (GST_DISKSINK (element)))
+ return GST_STATE_FAILURE;
+ }
+ }
+
+ if (GST_ELEMENT_CLASS (parent_class)->change_state)
+ return GST_ELEMENT_CLASS (parent_class)->change_state (element);
+
+ return GST_STATE_SUCCESS;
+}
+