From e5a857e78ab99018544e7dfbd98316c9ecfab29a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= Date: Tue, 29 Mar 2011 10:39:42 +0200 Subject: [PATCH] funnel: Import funnel element from farsight2 --- plugins/elements/gstfunnel.c | 435 ++++++++++++++++++++++++++++++++++ plugins/elements/gstfunnel.h | 69 ++++++ tests/check/elements/funnel.c | 185 +++++++++++++++ 3 files changed, 689 insertions(+) create mode 100644 plugins/elements/gstfunnel.c create mode 100644 plugins/elements/gstfunnel.h create mode 100644 tests/check/elements/funnel.c diff --git a/plugins/elements/gstfunnel.c b/plugins/elements/gstfunnel.c new file mode 100644 index 0000000000..96d0871257 --- /dev/null +++ b/plugins/elements/gstfunnel.c @@ -0,0 +1,435 @@ +/* + * Farsight2 - Farsight Funnel element + * + * Copyright 2007 Collabora Ltd. + * @author: Olivier Crete + * Copyright 2007 Nokia Corp. + * + * fs-funnel.c: Simple Funnel element + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +/** + * SECTION:element-fsfunnel + * @short_description: N-to-1 simple funnel + * + * Takes packets from various input sinks into one output source + */ + +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif + +#include "fs-funnel.h" + +GST_DEBUG_CATEGORY_STATIC (fs_funnel_debug); +#define GST_CAT_DEFAULT fs_funnel_debug + +static const GstElementDetails fs_funnel_details = +GST_ELEMENT_DETAILS ("Farsight Funnel pipe fitting", + "Generic", + "N-to-1 pipe fitting", + "Olivier Crete "); + +static GstStaticPadTemplate funnel_sink_template = +GST_STATIC_PAD_TEMPLATE ("sink%d", + GST_PAD_SINK, + GST_PAD_REQUEST, + GST_STATIC_CAPS_ANY); + +static GstStaticPadTemplate funnel_src_template = +GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS_ANY); + + +static void +_do_init (GType type) +{ + GST_DEBUG_CATEGORY_INIT (fs_funnel_debug, "fsfunnel", 0, "fsfunnel element"); +} + +GST_BOILERPLATE_FULL (FsFunnel, fs_funnel, GstElement, GST_TYPE_ELEMENT, + _do_init); + + + +static GstStateChangeReturn fs_funnel_change_state (GstElement * element, + GstStateChange transition); + +static GstPad *fs_funnel_request_new_pad (GstElement * element, + GstPadTemplate * templ, const gchar * name); +static void fs_funnel_release_pad (GstElement * element, GstPad * pad); + +static GstFlowReturn fs_funnel_buffer_alloc (GstPad * pad, guint64 offset, + guint size, GstCaps * caps, GstBuffer ** buf); +static GstFlowReturn fs_funnel_chain (GstPad * pad, GstBuffer * buffer); +static gboolean fs_funnel_event (GstPad * pad, GstEvent * event); +static gboolean fs_funnel_src_event (GstPad * pad, GstEvent * event); +static GstCaps *fs_funnel_getcaps (GstPad * pad); + + +typedef struct +{ + GstSegment segment; +} FsFunnelPadPrivate; + +static void +fs_funnel_base_init (gpointer g_class) +{ + GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class); + + gst_element_class_set_details (gstelement_class, &fs_funnel_details); + + gst_element_class_add_pad_template (gstelement_class, + gst_static_pad_template_get (&funnel_sink_template)); + gst_element_class_add_pad_template (gstelement_class, + gst_static_pad_template_get (&funnel_src_template)); +} + + +static void +fs_funnel_dispose (GObject * object) +{ + GList *item; + +restart: + for (item = GST_ELEMENT_PADS (object); item; item = g_list_next (item)) { + GstPad *pad = GST_PAD (item->data); + + if (GST_PAD_IS_SINK (pad)) { + gst_element_release_request_pad (GST_ELEMENT (object), pad); + goto restart; + } + } + + G_OBJECT_CLASS (parent_class)->dispose (object); +} + +static void +fs_funnel_class_init (FsFunnelClass * klass) +{ + GObjectClass *gobject_class = G_OBJECT_CLASS (klass); + GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass); + + gobject_class->dispose = GST_DEBUG_FUNCPTR (fs_funnel_dispose); + + gstelement_class->request_new_pad = + GST_DEBUG_FUNCPTR (fs_funnel_request_new_pad); + gstelement_class->release_pad = GST_DEBUG_FUNCPTR (fs_funnel_release_pad); + gstelement_class->change_state = GST_DEBUG_FUNCPTR (fs_funnel_change_state); +} + +static void +fs_funnel_init (FsFunnel * funnel, FsFunnelClass * g_class) +{ + funnel->srcpad = gst_pad_new_from_static_template (&funnel_src_template, + "src"); + gst_pad_set_event_function (funnel->srcpad, fs_funnel_src_event); + gst_pad_use_fixed_caps (funnel->srcpad); + gst_element_add_pad (GST_ELEMENT (funnel), funnel->srcpad); +} + +static GstFlowReturn +fs_funnel_buffer_alloc (GstPad * pad, guint64 offset, guint size, + GstCaps * caps, GstBuffer ** buf) +{ + FsFunnel *funnel = FS_FUNNEL (gst_pad_get_parent_element (pad)); + GstFlowReturn ret = GST_FLOW_OK; + + ret = gst_pad_alloc_buffer (funnel->srcpad, offset, size, caps, buf); + + gst_object_unref (funnel); + + return ret; +} + + +static GstPad * +fs_funnel_request_new_pad (GstElement * element, GstPadTemplate * templ, + const gchar * name) +{ + GstPad *sinkpad; + FsFunnelPadPrivate *priv = g_slice_alloc0 (sizeof (FsFunnelPadPrivate)); + + GST_DEBUG_OBJECT (element, "requesting pad"); + + sinkpad = gst_pad_new_from_template (templ, name); + + gst_pad_set_chain_function (sinkpad, GST_DEBUG_FUNCPTR (fs_funnel_chain)); + gst_pad_set_event_function (sinkpad, GST_DEBUG_FUNCPTR (fs_funnel_event)); + gst_pad_set_getcaps_function (sinkpad, GST_DEBUG_FUNCPTR (fs_funnel_getcaps)); + gst_pad_set_bufferalloc_function (sinkpad, + GST_DEBUG_FUNCPTR (fs_funnel_buffer_alloc)); + + gst_segment_init (&priv->segment, GST_FORMAT_UNDEFINED); + gst_pad_set_element_private (sinkpad, priv); + + gst_pad_set_active (sinkpad, TRUE); + + gst_element_add_pad (element, sinkpad); + + return sinkpad; +} + +static void +fs_funnel_release_pad (GstElement * element, GstPad * pad) +{ + FsFunnel *funnel = FS_FUNNEL (element); + FsFunnelPadPrivate *priv = gst_pad_get_element_private (pad); + + GST_DEBUG_OBJECT (funnel, "releasing pad"); + + gst_pad_set_active (pad, FALSE); + + if (priv) + g_slice_free1 (sizeof (FsFunnelPadPrivate), priv); + + gst_element_remove_pad (GST_ELEMENT_CAST (funnel), pad); +} + +static GstCaps * +fs_funnel_getcaps (GstPad * pad) +{ + FsFunnel *funnel = FS_FUNNEL (gst_pad_get_parent (pad)); + GstCaps *caps; + + caps = gst_pad_peer_get_caps_reffed (funnel->srcpad); + if (caps == NULL) + caps = gst_caps_copy (gst_pad_get_pad_template_caps (pad)); + + gst_object_unref (funnel); + + return caps; +} + +static GstFlowReturn +fs_funnel_chain (GstPad * pad, GstBuffer * buffer) +{ + GstFlowReturn res; + FsFunnel *funnel = FS_FUNNEL (gst_pad_get_parent (pad)); + FsFunnelPadPrivate *priv = gst_pad_get_element_private (pad); + GstEvent *event = NULL; + GstClockTime newts; + GstCaps *padcaps; + + GST_DEBUG_OBJECT (funnel, "received buffer %p", buffer); + + GST_OBJECT_LOCK (funnel); + if (priv->segment.format == GST_FORMAT_UNDEFINED) { + GST_WARNING_OBJECT (funnel, "Got buffer without segment," + " setting segment [0,inf["); + gst_segment_set_newsegment_full (&priv->segment, FALSE, 1.0, 1.0, + GST_FORMAT_TIME, 0, -1, 0); + } + + if (GST_CLOCK_TIME_IS_VALID (GST_BUFFER_TIMESTAMP (buffer))) + gst_segment_set_last_stop (&priv->segment, priv->segment.format, + GST_BUFFER_TIMESTAMP (buffer)); + + newts = gst_segment_to_running_time (&priv->segment, + priv->segment.format, GST_BUFFER_TIMESTAMP (buffer)); + if (newts != GST_BUFFER_TIMESTAMP (buffer)) { + buffer = gst_buffer_make_metadata_writable (buffer); + GST_BUFFER_TIMESTAMP (buffer) = newts; + } + + if (!funnel->has_segment) { + event = gst_event_new_new_segment_full (FALSE, 1.0, 1.0, GST_FORMAT_TIME, + 0, -1, 0); + funnel->has_segment = TRUE; + } + GST_OBJECT_UNLOCK (funnel); + + if (event) { + if (!gst_pad_push_event (funnel->srcpad, event)) + GST_WARNING_OBJECT (funnel, "Could not push out newsegment event"); + } + + + GST_OBJECT_LOCK (pad); + padcaps = GST_PAD_CAPS (funnel->srcpad); + GST_OBJECT_UNLOCK (pad); + + if (GST_BUFFER_CAPS (buffer) && GST_BUFFER_CAPS (buffer) != padcaps) { + if (!gst_pad_set_caps (funnel->srcpad, GST_BUFFER_CAPS (buffer))) { + res = GST_FLOW_NOT_NEGOTIATED; + goto out; + } + } + + res = gst_pad_push (funnel->srcpad, buffer); + + GST_LOG_OBJECT (funnel, "handled buffer %s", gst_flow_get_name (res)); + +out: + gst_object_unref (funnel); + + return res; +} + +static gboolean +fs_funnel_event (GstPad * pad, GstEvent * event) +{ + FsFunnel *funnel = FS_FUNNEL (gst_pad_get_parent (pad)); + FsFunnelPadPrivate *priv = gst_pad_get_element_private (pad); + gboolean forward = TRUE; + gboolean res = TRUE; + + switch (GST_EVENT_TYPE (event)) { + case GST_EVENT_NEWSEGMENT: + { + gboolean update; + gdouble rate, arate; + GstFormat format; + gint64 start; + gint64 stop; + gint64 time; + + gst_event_parse_new_segment_full (event, &update, &rate, &arate, + &format, &start, &stop, &time); + + + GST_OBJECT_LOCK (funnel); + gst_segment_set_newsegment_full (&priv->segment, update, rate, arate, + format, start, stop, time); + GST_OBJECT_UNLOCK (funnel); + + forward = FALSE; + gst_event_unref (event); + } + break; + case GST_EVENT_FLUSH_STOP: + { + GST_OBJECT_LOCK (funnel); + gst_segment_init (&priv->segment, GST_FORMAT_UNDEFINED); + GST_OBJECT_UNLOCK (funnel); + } + break; + default: + break; + } + + + if (forward) + res = gst_pad_push_event (funnel->srcpad, event); + + gst_object_unref (funnel); + + return res; +} + +static gboolean +fs_funnel_src_event (GstPad * pad, GstEvent * event) +{ + GstElement *funnel; + GstIterator *iter; + GstPad *sinkpad; + gboolean result = FALSE; + gboolean done = FALSE; + + funnel = gst_pad_get_parent_element (pad); + g_return_val_if_fail (funnel != NULL, FALSE); + + iter = gst_element_iterate_sink_pads (funnel); + + while (!done) { + switch (gst_iterator_next (iter, (gpointer) & sinkpad)) { + case GST_ITERATOR_OK: + gst_event_ref (event); + result |= gst_pad_push_event (sinkpad, event); + gst_object_unref (sinkpad); + break; + case GST_ITERATOR_RESYNC: + gst_iterator_resync (iter); + result = FALSE; + break; + case GST_ITERATOR_ERROR: + GST_WARNING_OBJECT (funnel, "Error iterating sinkpads"); + case GST_ITERATOR_DONE: + done = TRUE; + break; + } + } + gst_iterator_free (iter); + gst_object_unref (funnel); + gst_event_unref (event); + + return result; +} + +static void +reset_pad (gpointer data, gpointer user_data) +{ + GstPad *pad = data; + FsFunnelPadPrivate *priv = gst_pad_get_element_private (pad); + + GST_OBJECT_LOCK (pad); + gst_segment_init (&priv->segment, GST_FORMAT_UNDEFINED); + GST_OBJECT_UNLOCK (pad); + gst_object_unref (pad); +} + +static GstStateChangeReturn +fs_funnel_change_state (GstElement * element, GstStateChange transition) +{ + FsFunnel *funnel = FS_FUNNEL (element); + GstStateChangeReturn ret; + + switch (transition) { + case GST_STATE_CHANGE_READY_TO_PAUSED: + { + GstIterator *iter = gst_element_iterate_sink_pads (element); + GstIteratorResult res; + + do { + res = gst_iterator_foreach (iter, reset_pad, NULL); + } while (res == GST_ITERATOR_RESYNC); + + gst_iterator_free (iter); + + if (res == GST_ITERATOR_ERROR) + return GST_STATE_CHANGE_FAILURE; + + GST_OBJECT_LOCK (funnel); + funnel->has_segment = FALSE; + GST_OBJECT_UNLOCK (funnel); + } + break; + default: + break; + } + + ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition); + + return ret; +} + + +static gboolean +plugin_init (GstPlugin * plugin) +{ + return gst_element_register (plugin, "fsfunnel", + GST_RANK_NONE, FS_TYPE_FUNNEL); +} + +GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "fsfunnel", + "Farsight Funnel plugin", + plugin_init, + VERSION, "LGPL", "Farsight", "http://farsight.freedesktop.org/") diff --git a/plugins/elements/gstfunnel.h b/plugins/elements/gstfunnel.h new file mode 100644 index 0000000000..c1d5094e46 --- /dev/null +++ b/plugins/elements/gstfunnel.h @@ -0,0 +1,69 @@ +/* + * Farsight2 - Farsight Funnel element + * + * Copyright 2007 Collabora Ltd. + * @author: Olivier Crete + * Copyright 2007 Nokia Corp. + * + * fs-funnel.h: Simple Funnel element + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + + +#ifndef __FS_FUNNEL_H__ +#define __FS_FUNNEL_H__ + +#include + +G_BEGIN_DECLS + +#define FS_TYPE_FUNNEL \ + (fs_funnel_get_type ()) +#define FS_FUNNEL(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),FS_TYPE_FUNNEL,FsFunnel)) +#define FS_FUNNEL_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),FS_TYPE_FUNNEL,FsFunnelClass)) +#define FS_IS_FUNNEL(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),FS_TYPE_FUNNEL)) +#define FS_IS_FUNNEL_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),FS_TYPE_FUNNEL)) + +typedef struct _FsFunnel FsFunnel; +typedef struct _FsFunnelClass FsFunnelClass; + +/** + * FsFunnel: + * + * Opaque #FsFunnel data structure. + */ +struct _FsFunnel { + GstElement element; + + /*< private >*/ + GstPad *srcpad; + + gboolean has_segment; +}; + +struct _FsFunnelClass { + GstElementClass parent_class; +}; + +GType fs_funnel_get_type (void); + +G_END_DECLS + +#endif /* __FS_FUNNEL_H__ */ diff --git a/tests/check/elements/funnel.c b/tests/check/elements/funnel.c new file mode 100644 index 0000000000..8da1c3b1cf --- /dev/null +++ b/tests/check/elements/funnel.c @@ -0,0 +1,185 @@ +/* Farsight 2 unit tests for the fsfunnel + * + * Copyright (C) 2008 Collabora, Nokia + * @author: Olivier Crete + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +*/ + + +#ifdef HAVE_CONFIG_H +# include +#endif + +#include + +struct TestData +{ + GstElement *funnel; + GstPad *funnelsrc, *funnelsink11, *funnelsink22; + GstPad *mysink, *mysrc1, *mysrc2; + GstCaps *mycaps; +}; + +static void +setup_test_objects (struct TestData *td, GstPadChainFunction chain_func, + GstPadBufferAllocFunction alloc_func) +{ + td->mycaps = gst_caps_new_simple ("test/test", NULL); + + td->funnel = gst_element_factory_make ("fsfunnel", NULL); + + td->funnelsrc = gst_element_get_static_pad (td->funnel, "src"); + fail_unless (td->funnelsrc != NULL); + + td->funnelsink11 = gst_element_get_request_pad (td->funnel, "sink11"); + fail_unless (td->funnelsink11 != NULL); + fail_unless (!strcmp (GST_OBJECT_NAME (td->funnelsink11), "sink11")); + + td->funnelsink22 = gst_element_get_request_pad (td->funnel, "sink22"); + fail_unless (td->funnelsink22 != NULL); + fail_unless (!strcmp (GST_OBJECT_NAME (td->funnelsink22), "sink22")); + + fail_unless (gst_element_set_state (td->funnel, GST_STATE_PLAYING) == + GST_STATE_CHANGE_SUCCESS); + + td->mysink = gst_pad_new ("sink", GST_PAD_SINK); + gst_pad_set_chain_function (td->mysink, chain_func); + gst_pad_set_bufferalloc_function (td->mysink, alloc_func); + gst_pad_set_active (td->mysink, TRUE); + gst_pad_set_caps (td->mysink, td->mycaps); + + td->mysrc1 = gst_pad_new ("src1", GST_PAD_SRC); + gst_pad_set_active (td->mysrc1, TRUE); + gst_pad_set_caps (td->mysrc1, td->mycaps); + + td->mysrc2 = gst_pad_new ("src2", GST_PAD_SRC); + gst_pad_set_active (td->mysrc2, TRUE); + gst_pad_set_caps (td->mysrc2, td->mycaps); + + fail_unless (GST_PAD_LINK_SUCCESSFUL (gst_pad_link (td->funnelsrc, + td->mysink))); + + fail_unless (GST_PAD_LINK_SUCCESSFUL (gst_pad_link (td->mysrc1, + td->funnelsink11))); + + fail_unless (GST_PAD_LINK_SUCCESSFUL (gst_pad_link (td->mysrc2, + td->funnelsink22))); + +} + +static void +release_test_objects (struct TestData *td) +{ + gst_pad_set_active (td->mysink, FALSE); + gst_pad_set_active (td->mysrc1, FALSE); + gst_pad_set_active (td->mysrc1, FALSE); + + gst_object_unref (td->mysink); + gst_object_unref (td->mysrc1); + gst_object_unref (td->mysrc2); + + fail_unless (gst_element_set_state (td->funnel, GST_STATE_NULL) == + GST_STATE_CHANGE_SUCCESS); + + gst_object_unref (td->funnelsrc); + gst_object_unref (td->funnelsink11); + gst_element_release_request_pad (td->funnel, td->funnelsink11); + gst_object_unref (td->funnelsink22); + gst_element_release_request_pad (td->funnel, td->funnelsink22); + + gst_caps_unref (td->mycaps); + gst_object_unref (td->funnel); +} + +static gint bufcount = 0; +static gint alloccount = 0; + +static GstFlowReturn +chain_ok (GstPad * pad, GstBuffer * buffer) +{ + bufcount++; + + gst_buffer_unref (buffer); + + return GST_FLOW_OK; +} + +static GstFlowReturn +alloc_ok (GstPad * pad, + guint64 offset, guint size, GstCaps * caps, GstBuffer ** buffer) +{ + alloccount++; + + fail_unless (buffer != NULL); + fail_unless (*buffer == NULL); + + *buffer = gst_buffer_new_and_alloc (size); + gst_buffer_set_caps (*buffer, caps); + GST_BUFFER_OFFSET (*buffer) = offset; + + return GST_FLOW_OK; +} + +GST_START_TEST (test_funnel_simple) +{ + struct TestData td; + GstBuffer *buf1 = NULL; + GstBuffer *buf2 = NULL; + + setup_test_objects (&td, chain_ok, alloc_ok); + + bufcount = 0; + alloccount = 0; + + fail_unless (gst_pad_push (td.mysrc1, gst_buffer_new ()) == GST_FLOW_OK); + fail_unless (gst_pad_push (td.mysrc2, gst_buffer_new ()) == GST_FLOW_OK); + + fail_unless (bufcount == 2); + + fail_unless (gst_pad_alloc_buffer (td.mysrc1, 0, 1024, td.mycaps, + &buf1) == GST_FLOW_OK); + fail_unless (gst_pad_alloc_buffer (td.mysrc2, 1024, 1024, td.mycaps, + &buf2) == GST_FLOW_OK); + + fail_unless (alloccount == 2); + + gst_buffer_unref (buf1); + gst_buffer_unref (buf2); + + release_test_objects (&td); +} + +GST_END_TEST; + +static Suite * +funnel_suite (void) +{ + Suite *s = suite_create ("funnel"); + TCase *tc_chain; + GLogLevelFlags fatal_mask; + + fatal_mask = g_log_set_always_fatal (G_LOG_FATAL_MASK); + fatal_mask |= G_LOG_LEVEL_WARNING | G_LOG_LEVEL_CRITICAL; + g_log_set_always_fatal (fatal_mask); + + tc_chain = tcase_create ("funnel simple"); + tcase_add_test (tc_chain, test_funnel_simple); + suite_add_tcase (s, tc_chain); + + return s; +} + +GST_CHECK_MAIN (funnel);