mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-10-05 18:22:23 +00:00
Code Issues 1 Projects Releases Wiki Activity

Add an imagesequencesrc element to stream sequence of images

Browse source 
See: https://gitlab.freedesktop.org/gstreamer/gst-plugins-good/issues/121
This commit is contained in:
Fabian Orccon 2016-04-10 02:25:32 +00:00 committed by GStreamer Merge Bot
parent 05e8dd42e7
commit 7511999083
2 changed files with 540 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

471
gst/multifile/gstimagesequencesrc.c Normal file
View file

@ -0,0 +1,471 @@
/* GStreamer
* Copyright (C) 2019 Cesar Fabian Orccon Chipana
*
* 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., 51 Franklin Street, Suite 500,
* Boston, MA 02110-1335, USA.
*/
/**
* SECTION:element-gstimagesequencesrc
*
* Stream image sequences from image files.
* <refsect2>
*
* |[
* gst-launch-1.0 imagesequencesrc location=%d.jpg start-index=1 stop-index=50 framerate=24/1 ! decodebin ! videoconvert ! autovideosink
* ]|
*
* </refsect2>
*/
#include <gst/gst.h>
#include <gst/base/gsttypefindhelper.h>
#include "gstimagesequencesrc.h"
static GstFlowReturn gst_image_sequence_src_create (GstPushSrc * src,
GstBuffer ** buffer);
static void gst_image_sequence_src_dispose (GObject * object);
static void gst_image_sequence_src_set_property (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec);
static void gst_image_sequence_src_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec);
static GstCaps *gst_image_sequence_src_getcaps (GstBaseSrc * src,
GstCaps * filter);
static gboolean gst_image_sequence_src_query (GstBaseSrc * src,
GstQuery * query);
static void gst_image_sequence_src_set_caps (GstImageSequenceSrc * src,
GstCaps * caps);
static void gst_image_sequence_src_set_duration (GstImageSequenceSrc * src);
static gint gst_image_sequence_src_count_frames (GstImageSequenceSrc * self);
static GstStaticPadTemplate gst_image_sequence_src_pad_template =
GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
GST_DEBUG_CATEGORY_STATIC (gst_image_sequence_src_debug);
#define GST_CAT_DEFAULT gst_image_sequence_src_debug
enum
{
PROP_0,
PROP_LOCATION,
PROP_INDEX,
PROP_START_INDEX,
PROP_STOP_INDEX,
PROP_FRAMERATE
};
#define DEFAULT_LOCATION "%05d"
#define DEFAULT_INDEX 0
#define gst_image_sequence_src_parent_class parent_class
#define _do_init \
GST_DEBUG_CATEGORY_INIT (gst_image_sequence_src_debug, "imagesequencesrc", \
0, "imagesequencesrc element");
G_DEFINE_TYPE_WITH_CODE (GstImageSequenceSrc, gst_image_sequence_src,
GST_TYPE_PUSH_SRC, _do_init);
static gboolean
is_seekable (GstBaseSrc * src)
{
GstImageSequenceSrc *self = GST_IMAGE_SEQUENCE_SRC (src);
if ((self->count_frames != 0) && (self->fps_n) && (self->fps_d))
return TRUE;
return FALSE;
}
static gboolean
do_seek (GstBaseSrc * bsrc, GstSegment * segment)
{
GstImageSequenceSrc *src;
gboolean reverse;
src = GST_IMAGE_SEQUENCE_SRC (bsrc);
reverse = segment->rate < 0;
if (reverse) {
GST_FIXME_OBJECT (src, "Handle reverse playback");
return FALSE;
}
segment->time = segment->start;
src->index = src->start_index +
segment->position * src->fps_n / (src->fps_d * GST_SECOND);
GST_DEBUG_OBJECT (src, "Seek to frame at index %d.\n", src->index);
return TRUE;
}
static void
gst_image_sequence_src_class_init (GstImageSequenceSrcClass * klass)
{
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
GstPushSrcClass *gstpushsrc_class = GST_PUSH_SRC_CLASS (klass);
GstBaseSrcClass *gstbasesrc_class = GST_BASE_SRC_CLASS (klass);
gobject_class->set_property = gst_image_sequence_src_set_property;
gobject_class->get_property = gst_image_sequence_src_get_property;
g_object_class_install_property (gobject_class, PROP_LOCATION,
g_param_spec_string ("location", "File Location",
"Pattern to create file names of input files. File names are "
"created by calling sprintf() with the pattern and the current "
"index.", DEFAULT_LOCATION,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_INDEX,
g_param_spec_int ("index", "File Index",
"Index to use with location property to create file names. The "
"index is incremented by one for each buffer read.",
0, INT_MAX, DEFAULT_INDEX,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_START_INDEX,
g_param_spec_int ("start-index", "Start Index",
"Start value of index. The initial value of index can be set "
"either by setting index or start-index. When the end of the loop "
"is reached, the index will be set to the value start-index.",
0, INT_MAX, DEFAULT_INDEX,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_STOP_INDEX,
g_param_spec_int ("stop-index", "Stop Index",
"Stop value of index. The special value -1 means no stop.",
-1, INT_MAX, DEFAULT_INDEX,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_FRAMERATE,
gst_param_spec_fraction ("framerate", "Framerate",
"Set the framerate to internal caps.",
1, 1, G_MAXINT, 1, 1, 1, G_PARAM_WRITABLE | G_PARAM_STATIC_STRINGS));
gobject_class->dispose = gst_image_sequence_src_dispose;
gstbasesrc_class->get_caps = gst_image_sequence_src_getcaps;
gstbasesrc_class->query = gst_image_sequence_src_query;
gstbasesrc_class->is_seekable = is_seekable;
gstbasesrc_class->do_seek = do_seek;
gstpushsrc_class->create = gst_image_sequence_src_create;
gst_element_class_add_static_pad_template (gstelement_class,
&gst_image_sequence_src_pad_template);
gst_element_class_set_static_metadata (gstelement_class,
"ImageSequenceSrc plugin", "Src/File",
"Creates an image-sequence video stream",
"Cesar Fabian Orccon Chipana <cfoch.fabian@gmail.com>");
}
static void
gst_image_sequence_src_init (GstImageSequenceSrc * self)
{
GstBaseSrc *bsrc;
GST_DEBUG_CATEGORY_INIT (gst_image_sequence_src_debug, "imagesequencesrc", 0,
"imagesequencesrc element");
bsrc = GST_BASE_SRC (self);
gst_base_src_set_format (bsrc, GST_FORMAT_TIME);
self->start_index = DEFAULT_INDEX;
self->index = DEFAULT_INDEX;
self->stop_index = -1;
self->filename = g_strdup (DEFAULT_LOCATION);
self->caps = NULL;
self->count_frames = 0;
self->fps_n = self->fps_d = 1;
}
static void
gst_image_sequence_src_dispose (GObject * object)
{
GstImageSequenceSrc *src = GST_IMAGE_SEQUENCE_SRC (object);
g_clear_pointer (&src->filename, g_free);
if (src->caps)
gst_clear_caps (&src->caps);
G_OBJECT_CLASS (parent_class)->dispose (object);
}
static GstCaps *
gst_image_sequence_src_getcaps (GstBaseSrc * src, GstCaps * filter)
{
GstImageSequenceSrc *self = GST_IMAGE_SEQUENCE_SRC (src);
GST_DEBUG_OBJECT (src, "returning %" GST_PTR_FORMAT, self->caps);
if (filter) {
if (self->caps)
return gst_caps_intersect_full (filter, self->caps,
GST_CAPS_INTERSECT_FIRST);
else
return gst_caps_ref (filter);
}
return gst_caps_new_any ();
}
static gboolean
gst_image_sequence_src_query (GstBaseSrc * bsrc, GstQuery * query)
{
gboolean ret;
GstImageSequenceSrc *src;
src = GST_IMAGE_SEQUENCE_SRC (bsrc);
switch (GST_QUERY_TYPE (query)) {
case GST_QUERY_DURATION:
{
GstFormat format;
gst_query_parse_duration (query, &format, NULL);
switch (format) {
case GST_FORMAT_TIME:
if (src->count_frames > 0)
gst_query_set_duration (query, format, src->duration);
ret = TRUE;
break;
default:
ret = GST_BASE_SRC_CLASS (parent_class)->query (bsrc, query);
}
break;
}
default:
ret = GST_BASE_SRC_CLASS (parent_class)->query (bsrc, query);
break;
}
return ret;
}
static gboolean
gst_image_sequence_src_set_location (GstImageSequenceSrc * 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_image_sequence_src_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstImageSequenceSrc *self = GST_IMAGE_SEQUENCE_SRC (object);
switch (prop_id) {
case PROP_LOCATION:
gst_image_sequence_src_set_location (self, g_value_get_string (value));
break;
case PROP_INDEX:
GST_OBJECT_LOCK (self);
/* index was really meant to be read-only, but for backwards-compatibility
* we set start_index to make it work as it used to */
if (!GST_OBJECT_FLAG_IS_SET (self, GST_BASE_SRC_FLAG_STARTED))
self->start_index = g_value_get_int (value);
else
self->index = g_value_get_int (value);
GST_OBJECT_UNLOCK (self);
break;
case PROP_START_INDEX:
self->start_index = g_value_get_int (value);
gst_image_sequence_src_count_frames (self);
break;
case PROP_STOP_INDEX:
self->stop_index = g_value_get_int (value);
gst_image_sequence_src_count_frames (self);
break;
case PROP_FRAMERATE:
self->fps_n = gst_value_get_fraction_numerator (value);
self->fps_d = gst_value_get_fraction_denominator (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static void
gst_image_sequence_src_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
GstImageSequenceSrc *src = GST_IMAGE_SEQUENCE_SRC (object);
switch (prop_id) {
case PROP_LOCATION:
g_value_set_string (value, src->filename);
break;
case PROP_INDEX:
g_value_set_int (value, src->index);
break;
case PROP_START_INDEX:
g_value_set_int (value, src->start_index);
break;
case PROP_STOP_INDEX:
g_value_set_int (value, src->stop_index);
break;
case PROP_FRAMERATE:
src->fps_n = gst_value_get_fraction_numerator (value);
src->fps_d = gst_value_get_fraction_denominator (value);
GST_DEBUG_OBJECT (src, "Set (framerate) property to (%d/%d)", src->fps_n,
src->fps_d);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static gint
gst_image_sequence_src_count_frames (GstImageSequenceSrc * self)
{
self->count_frames = self->stop_index - self->start_index + 1;
return self->count_frames;
}
static void
gst_image_sequence_src_set_caps (GstImageSequenceSrc * self, GstCaps * caps)
{
GstCaps *new_caps;
g_assert (caps != NULL);
new_caps = gst_caps_copy (caps);
if (self->count_frames > 0) {
GValue fps = G_VALUE_INIT;
g_value_init (&fps, GST_TYPE_FRACTION);
gst_value_set_fraction (&fps, self->fps_n, self->fps_d);
gst_caps_set_value (new_caps, "framerate", &fps);
}
gst_caps_replace (&self->caps, new_caps);
gst_pad_set_caps (GST_BASE_SRC_PAD (self), new_caps);
GST_DEBUG_OBJECT (self, "Setting new caps: %s",
gst_caps_to_string (new_caps));
}
static void
gst_image_sequence_src_set_duration (GstImageSequenceSrc * self)
{
/* Calculate duration */
self->duration =
gst_util_uint64_scale (GST_SECOND * self->count_frames, self->fps_d,
self->fps_n);
}
static gchar *
gst_image_sequence_src_get_filename (GstImageSequenceSrc * self)
{
gchar *filename;
GST_DEBUG ("Reading filename at index %d.", self->index);
filename = g_strdup_printf (self->filename, self->index);
return filename;
}
static GstFlowReturn
gst_image_sequence_src_create (GstPushSrc * src, GstBuffer ** buffer)
{
GstImageSequenceSrc *self;
gsize size;
gchar *data;
gchar *filename;
GstBuffer *buf;
gboolean ret;
GError *error = NULL;
self = GST_IMAGE_SEQUENCE_SRC (src);
if (self->index > self->stop_index)
return GST_FLOW_EOS;
if (self->index < self->start_index)
self->index = self->start_index;
g_assert (self->start_index <= self->index &&
self->index <= self->stop_index);
filename = gst_image_sequence_src_get_filename (self);
GST_DEBUG_OBJECT (self, "reading from file \"%s\".", filename);
ret = g_file_get_contents (filename, &data, &size, &error);
if (!ret)
goto handle_error;
buf = gst_buffer_new_wrapped (data, size);
if (!self->caps) {
GstCaps *caps;
caps = gst_type_find_helper_for_buffer (NULL, buf, NULL);
gst_image_sequence_src_set_caps (self, caps);
gst_image_sequence_src_set_duration (self);
gst_caps_unref (caps);
}
self->buffer_duration = GST_SECOND * self->fps_d / self->fps_n;
GST_BUFFER_PTS (buf) =
(self->index - self->start_index) * self->buffer_duration;
GST_BUFFER_DURATION (buf) = self->buffer_duration;
GST_DEBUG_OBJECT (self, "index: %d", self->index);
GST_DEBUG_OBJECT (self, "Timestamp: %" GST_TIME_FORMAT,
GST_TIME_ARGS (GST_BUFFER_PTS (buf)));
GST_DEBUG_OBJECT (self, "Buffer duration: %" GST_TIME_FORMAT,
GST_TIME_ARGS (GST_BUFFER_DURATION (buf)));
GST_BUFFER_OFFSET (buf) = self->index - self->start_index;
GST_BUFFER_OFFSET_END (buf) = self->index - self->start_index + 1;
GST_DEBUG_OBJECT (self, "read file \"%s\".", filename);
g_free (filename);
*buffer = buf;
self->index++;
return GST_FLOW_OK;
handle_error:
{
if (error != NULL) {
GST_ELEMENT_ERROR (self, RESOURCE, READ,
("Error while reading from file \"%s\".", filename),
("%s", error->message));
g_error_free (error);
} else {
GST_ELEMENT_ERROR (self, RESOURCE, READ,
("Error while reading from file \"%s\".", filename),
("%s", g_strerror (errno)));
}
g_free (filename);
return GST_FLOW_ERROR;
}
}

69
gst/multifile/gstimagesequencesrc.h Normal file
View file

@ -0,0 +1,69 @@
/* GStreamer
* Copyright (C) 2019 Cesar Fabian Orccon Chipana
*
* 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., 51 Franklin Street, Suite 500,
* Boston, MA 02110-1335, USA.
*/
#ifndef __GST_IMAGESEQUENCESRC_H__
#define __GST_IMAGESEQUENCESRC_H__
#include <gst/gst.h>
#include <gst/base/gstpushsrc.h>
G_BEGIN_DECLS
#define GST_TYPE_IMAGE_SEQUENCE_SRC \
(gst_image_sequence_src_get_type())
#define GST_IMAGE_SEQUENCE_SRC(obj) \
(G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_IMAGE_SEQUENCE_SRC,GstImageSequenceSrc))
#define GST_IMAGE_SEQUENCE_SRC_CLASS(klass) \
(G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_IMAGE_SEQUENCE_SRC,GstImageSequenceSrcClass))
#define GST_IS_IMAGE_SEQUENCE_SRC(obj) \
(G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_IMAGE_SEQUENCE_SRC))
#define GST_IS_IMAGE_SEQUENCE_SRC_CLASS(klass) \
(G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_IMAGE_SEQUENCE_SRC))
typedef struct _GstImageSequenceSrc GstImageSequenceSrc;
typedef struct _GstImageSequenceSrcClass GstImageSequenceSrcClass;
struct _GstImageSequenceSrc
{
GstPushSrc parent;
gchar *filename;
int start_index;
int stop_index;
int index;
int count_frames;
guint64 duration;
guint64 buffer_duration;
GstCaps *caps;
gint fps_n, fps_d;
};
struct _GstImageSequenceSrcClass
{
GstPushSrcClass parent_class;
};
GType gst_image_sequence_src_get_type (void);
G_END_DECLS
#endif /* __GST_IMAGESEQUENCESRC_H__ */