/* GStreamer * Copyright (C) <1999> Erik Walthinsen * * 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. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include /* GstVideodrop signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; enum { ARG_0, ARG_SPEED /* FILL ME */ }; static GstStaticPadTemplate gst_videodrop_src_template = GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{ YUY2, I420, YV12, YUYV, UYVY }") ) ); static GstStaticPadTemplate gst_videodrop_sink_template = GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{ YUY2, I420, YV12, YUYV, UYVY }") ) ); static void gst_videodrop_base_init (gpointer g_class); static void gst_videodrop_class_init (GstVideodropClass * klass); static void gst_videodrop_init (GstVideodrop * videodrop); static void gst_videodrop_chain (GstPad * pad, GstData * _data); static void gst_videodrop_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec); static void gst_videodrop_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec); static GstStateChangeReturn gst_videodrop_change_state (GstElement * element, GstStateChange transition); static GstElementClass *parent_class = NULL; /*static guint gst_videodrop_signals[LAST_SIGNAL] = { 0 }; */ GType gst_videodrop_get_type (void) { static GType videodrop_type = 0; if (!videodrop_type) { static const GTypeInfo videodrop_info = { sizeof (GstVideodropClass), gst_videodrop_base_init, NULL, (GClassInitFunc) gst_videodrop_class_init, NULL, NULL, sizeof (GstVideodrop), 0, (GInstanceInitFunc) gst_videodrop_init, }; videodrop_type = g_type_register_static (GST_TYPE_ELEMENT, "GstVideodrop", &videodrop_info, 0); } return videodrop_type; } static void gst_videodrop_base_init (gpointer g_class) { GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); gst_element_class_set_details_simple (element_class, "Frame dropper", "Filter/Effect/Video", "Re-FPS'es video by dropping frames", "Ronald Bultje "); gst_element_class_add_pad_template (element_class, gst_static_pad_template_get (&gst_videodrop_sink_template)); gst_element_class_add_pad_template (element_class, gst_static_pad_template_get (&gst_videodrop_src_template)); } static void gst_videodrop_class_init (GstVideodropClass * klass) { GObjectClass *object_class = G_OBJECT_CLASS (klass); GstElementClass *element_class = GST_ELEMENT_CLASS (klass); parent_class = g_type_class_peek_parent (klass); g_object_class_install_property (object_class, ARG_SPEED, g_param_spec_float ("speed", "Speed", "Output speed (relative to input)", 0.01, 100, 1, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); object_class->set_property = gst_videodrop_set_property; object_class->get_property = gst_videodrop_get_property; element_class->change_state = gst_videodrop_change_state; } static GstCaps * gst_videodrop_getcaps (GstPad * pad) { GstVideodrop *videodrop; GstPad *otherpad; GstCaps *caps, *copy, *copy2 = NULL; int i; gdouble otherfps; GstStructure *structure; gboolean negotiated; videodrop = GST_VIDEODROP (gst_pad_get_parent (pad)); otherpad = (pad == videodrop->srcpad) ? videodrop->sinkpad : videodrop->srcpad; negotiated = gst_pad_is_negotiated (otherpad); otherfps = (pad == videodrop->srcpad) ? videodrop->from_fps : videodrop->to_fps; caps = gst_pad_get_allowed_caps (otherpad); copy = gst_caps_copy (caps); if (negotiated) { copy2 = gst_caps_copy (caps); } for (i = 0; i < gst_caps_get_size (caps); i++) { structure = gst_caps_get_structure (caps, i); gst_structure_set (structure, "framerate", GST_TYPE_DOUBLE_RANGE, 0.0, G_MAXDOUBLE, NULL); } if ((negotiated) && (videodrop->speed != 1.0)) { for (i = 0; i < gst_caps_get_size (copy2); i++) { structure = gst_caps_get_structure (copy2, i); gst_structure_set (structure, "framerate", G_TYPE_DOUBLE, otherfps * videodrop->speed, NULL); } gst_caps_append (copy2, copy); copy = copy2; } gst_caps_append (copy, caps); return copy; } static GstPadLinkReturn gst_videodrop_link (GstPad * pad, const GstCaps * caps) { GstVideodrop *videodrop; GstStructure *structure; gboolean ret; double fps; GstPad *otherpad; videodrop = GST_VIDEODROP (gst_pad_get_parent (pad)); otherpad = (pad == videodrop->srcpad) ? videodrop->sinkpad : videodrop->srcpad; structure = gst_caps_get_structure (caps, 0); ret = gst_structure_get_double (structure, "framerate", &fps); if (!ret) return GST_PAD_LINK_REFUSED; if (pad == videodrop->srcpad) { videodrop->to_fps = fps; } else { videodrop->from_fps = fps; } if (gst_pad_is_negotiated (otherpad)) { /* * Ensure that the other side talks the format we're trying to set */ GstCaps *newcaps = gst_caps_copy (caps); if (pad == videodrop->srcpad) { gst_caps_set_simple (newcaps, "framerate", G_TYPE_DOUBLE, videodrop->from_fps, NULL); } else { gst_caps_set_simple (newcaps, "framerate", G_TYPE_DOUBLE, videodrop->to_fps, NULL); } ret = gst_pad_try_set_caps (otherpad, newcaps); if (GST_PAD_LINK_FAILED (ret)) { return GST_PAD_LINK_REFUSED; } } return GST_PAD_LINK_OK; } static void gst_videodrop_init (GstVideodrop * videodrop) { GST_OBJECT_FLAG_SET (videodrop, GST_ELEMENT_EVENT_AWARE); GST_DEBUG ("gst_videodrop_init"); videodrop->sinkpad = gst_pad_new_from_static_template (&gst_videodrop_sink_template, "sink"); gst_element_add_pad (GST_ELEMENT (videodrop), videodrop->sinkpad); gst_pad_set_chain_function (videodrop->sinkpad, gst_videodrop_chain); gst_pad_set_getcaps_function (videodrop->sinkpad, gst_videodrop_getcaps); gst_pad_set_link_function (videodrop->sinkpad, gst_videodrop_link); videodrop->srcpad = gst_pad_new_from_static_template (&gst_videodrop_src_template, "src"); gst_element_add_pad (GST_ELEMENT (videodrop), videodrop->srcpad); gst_pad_set_getcaps_function (videodrop->srcpad, gst_videodrop_getcaps); gst_pad_set_link_function (videodrop->srcpad, gst_videodrop_link); videodrop->inited = FALSE; videodrop->total = videodrop->pass = 0; videodrop->speed = 1.; videodrop->time_adjust = 0; } static void gst_videodrop_chain (GstPad * pad, GstData * data) { GstVideodrop *videodrop = GST_VIDEODROP (gst_pad_get_parent (pad)); GstBuffer *buf; if (GST_IS_EVENT (data)) { GstEvent *event = GST_EVENT (data); if (GST_EVENT_TYPE (event) == GST_EVENT_DISCONTINUOUS) { /* since we rely on timestamps of the source, we need to handle * changes in time carefully. */ gint64 time; if (gst_event_discont_get_value (event, GST_FORMAT_TIME, &time)) { videodrop->total = videodrop->pass = 0; videodrop->time_adjust = time; } else { GST_ELEMENT_ERROR (videodrop, STREAM, TOO_LAZY, (NULL), ("Received discont, but no time information")); gst_event_unref (event); return; } /* FIXME: increase timestamp / speed */ } gst_pad_event_default (pad, event); return; } buf = GST_BUFFER (data); videodrop->total++; GST_DEBUG ("Received buffer at %u:%02u:%02u:%09u, fps=%lf, pass=%" G_GUINT64_FORMAT " of " G_GUINT64_FORMAT ", speed=%lf", (guint) (GST_BUFFER_TIMESTAMP (buf) / (GST_SECOND * 60 * 60)), (guint) ((GST_BUFFER_TIMESTAMP (buf) / (GST_SECOND * 60)) % 60), (guint) ((GST_BUFFER_TIMESTAMP (buf) / GST_SECOND) % 60), (guint) (GST_BUFFER_TIMESTAMP (buf) % GST_SECOND), videodrop->to_fps, videodrop->total, videodrop->pass, videodrop->speed); while (((GST_BUFFER_TIMESTAMP (buf) - videodrop->time_adjust) / videodrop->speed * videodrop->to_fps / GST_SECOND) >= videodrop->pass) { /* since we write to the struct (time/duration), we need a new struct, * but we don't want to copy around data - a subbuffer is the easiest * way to accomplish that... */ GstBuffer *copy = gst_buffer_create_sub (buf, 0, GST_BUFFER_SIZE (buf)); /* adjust timestamp/duration and push forward */ GST_BUFFER_TIMESTAMP (copy) = (videodrop->time_adjust / videodrop->speed) + GST_SECOND * videodrop->pass / videodrop->to_fps; GST_BUFFER_DURATION (copy) = GST_SECOND / videodrop->to_fps; GST_DEBUG ("Sending out buffer from out %u:%02u:%02u:%09u", (guint) (GST_BUFFER_TIMESTAMP (copy) / (GST_SECOND * 60 * 60)), (guint) ((GST_BUFFER_TIMESTAMP (copy) / (GST_SECOND * 60)) % 60), (guint) ((GST_BUFFER_TIMESTAMP (copy) / GST_SECOND) % 60), (guint) (GST_BUFFER_TIMESTAMP (copy) % GST_SECOND)); gst_pad_push (videodrop->srcpad, GST_DATA (copy)); videodrop->pass++; } gst_buffer_unref (buf); } static void gst_videodrop_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstVideodrop *videodrop = GST_VIDEODROP (object); switch (prop_id) { case ARG_SPEED: videodrop->speed = g_value_get_float (value); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_videodrop_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstVideodrop *videodrop = GST_VIDEODROP (object); switch (prop_id) { case ARG_SPEED: g_value_set_float (value, videodrop->speed); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static GstStateChangeReturn gst_videodrop_change_state (GstElement * element, GstStateChange transition) { GstVideodrop *videodrop = GST_VIDEODROP (element); switch (transition) { case GST_STATE_CHANGE_PAUSED_TO_READY: videodrop->inited = FALSE; videodrop->time_adjust = 0; videodrop->total = videodrop->pass = 0; break; default: break; } if (parent_class->change_state) return parent_class->change_state (element, transition); return GST_STATE_CHANGE_SUCCESS; } static gboolean plugin_init (GstPlugin * plugin) { return gst_element_register (plugin, "videodrop", GST_RANK_NONE, GST_TYPE_VIDEODROP); } GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR, videodrop, "Re-FPS'es video", plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)