diff options
Diffstat (limited to 'gst/multifile/gstmultifilesrc.c')
-rw-r--r-- | gst/multifile/gstmultifilesrc.c | 322 |
1 files changed, 322 insertions, 0 deletions
diff --git a/gst/multifile/gstmultifilesrc.c b/gst/multifile/gstmultifilesrc.c new file mode 100644 index 00000000..c9ae5caa --- /dev/null +++ b/gst/multifile/gstmultifilesrc.c @@ -0,0 +1,322 @@ +/* GStreamer + * Copyright (C) 2006 David A. Schleef <ds@schleef.org> + * + * gstmultifilesrc.c: + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ +/** + * SECTION:element-multifilesrc + * @short_description: reads buffers from sequentially-named files + * @see_also: #GstFileSrc + * + * Reads buffers from sequentially named files. + */ + +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include <gst/gst.h> +#include <gst/base/gstpushsrc.h> + +#define GST_TYPE_MULTI_FILE_SRC \ + (gst_multi_file_src_get_type()) +#define GST_MULTI_FILE_SRC(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_MULTI_FILE_SRC,GstMultiFileSrc)) +#define GST_MULTI_FILE_SRC_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_MULTI_FILE_SRC,GstMultiFileSrcClass)) +#define GST_IS_MULTI_FILE_SRC(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_MULTI_FILE_SRC)) +#define GST_IS_MULTI_FILE_SRC_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_MULTI_FILE_SRC)) + +typedef struct _GstMultiFileSrc GstMultiFileSrc; +typedef struct _GstMultiFileSrcClass GstMultiFileSrcClass; + +struct _GstMultiFileSrc +{ + GstPushSrc parent; + + gchar *filename; + int index; + + int offset; + + GstCaps *caps; +}; + +struct _GstMultiFileSrcClass +{ + GstPushSrcClass parent_class; +}; + + +static GstFlowReturn gst_multi_file_src_create (GstPushSrc * src, + GstBuffer ** buffer); + +static void gst_multi_file_src_dispose (GObject * object); + +static void gst_multi_file_src_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_multi_file_src_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); +static GstCaps *gst_multi_file_src_getcaps (GstBaseSrc * src); + + +static GstStaticPadTemplate gst_multi_file_src_pad_template = +GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS_ANY); + +GST_DEBUG_CATEGORY_STATIC (gst_multi_file_src_debug); +#define GST_CAT_DEFAULT gst_multi_file_src_debug + +static const GstElementDetails gst_multi_file_src_details = +GST_ELEMENT_DETAILS ("Multi-File Source", + "Source/File", + "Read stream from files", + "David Schleef <ds@schleef.org>"); + +enum +{ + ARG_0, + ARG_LOCATION, + ARG_INDEX, + ARG_CAPS +}; + + +GST_BOILERPLATE (GstMultiFileSrc, gst_multi_file_src, GstPushSrc, + GST_TYPE_PUSH_SRC); + +static void +gst_multi_file_src_base_init (gpointer g_class) +{ + GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class); + + GST_DEBUG_CATEGORY_INIT (gst_multi_file_src_debug, "multifilesrc", 0, + "multifilesrc element"); + + gst_element_class_add_pad_template (gstelement_class, + gst_static_pad_template_get (&gst_multi_file_src_pad_template)); + gst_element_class_set_details (gstelement_class, &gst_multi_file_src_details); +} + +static void +gst_multi_file_src_class_init (GstMultiFileSrcClass * klass) +{ + GObjectClass *gobject_class = G_OBJECT_CLASS (klass); + GstPushSrcClass *gstpushsrc_class = GST_PUSH_SRC_CLASS (klass); + GstBaseSrcClass *gstbasesrc_class = GST_BASE_SRC_CLASS (klass); + + gobject_class->set_property = gst_multi_file_src_set_property; + gobject_class->get_property = gst_multi_file_src_get_property; + + g_object_class_install_property (gobject_class, ARG_LOCATION, + g_param_spec_string ("location", "File Location", + "Location of the file to read", NULL, G_PARAM_READWRITE)); + g_object_class_install_property (gobject_class, ARG_INDEX, + g_param_spec_int ("index", "File Index", + "Index of file to read", 0, INT_MAX, 0, G_PARAM_READWRITE)); + g_object_class_install_property (gobject_class, ARG_CAPS, + g_param_spec_boxed ("caps", "Caps", + "Caps of source pad", GST_TYPE_CAPS, G_PARAM_READWRITE)); + + gobject_class->dispose = gst_multi_file_src_dispose; + + gstpushsrc_class->create = gst_multi_file_src_create; + + gstbasesrc_class->get_caps = gst_multi_file_src_getcaps; + + if (sizeof (off_t) < 8) { + GST_LOG ("No large file support, sizeof (off_t) = %u", sizeof (off_t)); + } +} + +static void +gst_multi_file_src_init (GstMultiFileSrc * filesrc, + GstMultiFileSrcClass * g_class) +{ + GstPad *pad; + + pad = GST_BASE_SRC_PAD (filesrc); + + filesrc->filename = g_strdup ("output-%05d"); +} + +static void +gst_multi_file_src_dispose (GObject * object) +{ + GstMultiFileSrc *src = GST_MULTI_FILE_SRC (object); + + G_OBJECT_CLASS (parent_class)->dispose (object); + + g_free (src->filename); + src->filename = NULL; + if (src->caps) + gst_caps_unref (src->caps); +} + +static GstCaps * +gst_multi_file_src_getcaps (GstBaseSrc * src) +{ + GstMultiFileSrc *multi_file_src = GST_MULTI_FILE_SRC (src); + + GST_DEBUG ("returning %" GST_PTR_FORMAT, multi_file_src->caps); + + if (multi_file_src->caps) { + return gst_caps_ref (multi_file_src->caps); + } else { + return gst_caps_new_any (); + } +} + +static gboolean +gst_multi_file_src_set_location (GstMultiFileSrc * src, const gchar * location) +{ + g_free (src->filename); + if (location != NULL) { + src->filename = g_strdup (location); + } else { + src->filename = NULL; + } + + return TRUE; +} +static void +gst_multi_file_src_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + GstMultiFileSrc *src = GST_MULTI_FILE_SRC (object); + + switch (prop_id) { + case ARG_LOCATION: + gst_multi_file_src_set_location (src, g_value_get_string (value)); + break; + case ARG_INDEX: + src->index = g_value_get_int (value); + break; + case ARG_CAPS: + { + const GstCaps *caps = gst_value_get_caps (value); + GstCaps *new_caps; + + if (caps == NULL) { + new_caps = gst_caps_new_any (); + } else { + new_caps = gst_caps_copy (caps); + } + gst_caps_replace (&src->caps, new_caps); + gst_pad_set_caps (GST_BASE_SRC_PAD (src), new_caps); + } + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_multi_file_src_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) +{ + GstMultiFileSrc *src = GST_MULTI_FILE_SRC (object); + + switch (prop_id) { + case ARG_LOCATION: + g_value_set_string (value, src->filename); + break; + case ARG_INDEX: + g_value_set_int (value, src->index); + break; + case ARG_CAPS: + gst_value_set_caps (value, src->caps); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static gchar * +gst_multi_file_src_get_filename (GstMultiFileSrc * filesrc) +{ + gchar *filename; + + filename = g_strdup_printf (filesrc->filename, filesrc->index); + + return filename; +} + +static GstFlowReturn +gst_multi_file_src_create (GstPushSrc * src, GstBuffer ** buffer) +{ + GstMultiFileSrc *filesrc; + guint size; + gchar *filename; + FILE *file; + GstBuffer *buf; + int ret; + + filesrc = GST_MULTI_FILE_SRC (src); + + filename = gst_multi_file_src_get_filename (filesrc); + + file = fopen (filename, "rb"); + if (!file) { + goto handle_error; + } + + fseek (file, 0, SEEK_END); + size = ftell (file); + fseek (file, 0, SEEK_SET); + + buf = gst_buffer_new_and_alloc (size); + + ret = fread (GST_BUFFER_DATA (buf), size, 1, file); + if (ret < 1) { + goto handle_error; + } + + filesrc->index++; + + GST_BUFFER_SIZE (buf) = size; + GST_BUFFER_OFFSET (buf) = filesrc->offset; + GST_BUFFER_OFFSET (buf) = filesrc->offset + size; + filesrc->offset += size; + gst_buffer_set_caps (buf, filesrc->caps); + + fclose (file); + + g_free (filename); + *buffer = buf; + return GST_FLOW_OK; + +handle_error: + { + switch (errno) { + default:{ + GST_ELEMENT_ERROR (filesrc, RESOURCE, READ, + ("Error while reading from file \"%s\".", filename), + ("%s", g_strerror (errno))); + } + } + g_free (filename); + return GST_FLOW_ERROR; + } +} |