/* -*- c-basic-offset: 2 -*- * GStreamer * Copyright (C) 1999-2001 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 "gstfilter.h" #include "iir.h" static GstElementDetails gst_iir_details = GST_ELEMENT_DETAILS ("Infinite Impulse Response(IIR) filter", "Filter/Effect/Audio", "IIR filter based on vorbis code", "Monty , " "Thomas "); enum { /* FILL ME */ LAST_SIGNAL }; enum { ARG_0, ARG_A, ARG_B, ARG_GAIN, ARG_STAGES }; #define GST_TYPE_IIR \ (gst_iir_get_type()) #define GST_IIR(obj) \ (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_IIR,GstIIR)) #define GST_IIR_CLASS(klass) \ (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ULAW,GstIIR)) #define GST_IS_IIR(obj) \ (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_IIR)) #define GST_IS_IIR_CLASS(obj) \ (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_IIR)) typedef struct _GstIIR GstIIR; typedef struct _GstIIRClass GstIIRClass; struct _GstIIR { GstElement element; GstPad *sinkpad, *srcpad; double A, B; double gain; int stages; IIR_state *state; }; struct _GstIIRClass { GstElementClass parent_class; }; static void gst_iir_base_init (gpointer g_class); static void gst_iir_class_init (GstIIRClass * klass); static void gst_iir_init (GstIIR * filter); static void gst_iir_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec); static void gst_iir_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec); static void gst_iir_chain (GstPad * pad, GstData * _data); static GstPadLinkReturn gst_iir_sink_connect (GstPad * pad, const GstCaps * caps); static GstElementClass *parent_class = NULL; /*static guint gst_iir_signals[LAST_SIGNAL] = { 0 }; */ GType gst_iir_get_type (void) { static GType iir_type = 0; if (!iir_type) { static const GTypeInfo iir_info = { sizeof (GstIIRClass), gst_iir_base_init, NULL, (GClassInitFunc) gst_iir_class_init, NULL, NULL, sizeof (GstIIR), 0, (GInstanceInitFunc) gst_iir_init, }; iir_type = g_type_register_static (GST_TYPE_ELEMENT, "GstIIR", &iir_info, 0); } return iir_type; } static void gst_iir_base_init (gpointer g_class) { GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); /* register src pads */ gst_element_class_add_pad_template (element_class, gst_static_pad_template_get (&gst_filter_src_template)); gst_element_class_add_pad_template (element_class, gst_static_pad_template_get (&gst_filter_sink_template)); gst_element_class_set_details (element_class, &gst_iir_details); } static void gst_iir_class_init (GstIIRClass * klass) { GObjectClass *gobject_class; GstElementClass *gstelement_class; gobject_class = (GObjectClass *) klass; gstelement_class = (GstElementClass *) klass; parent_class = g_type_class_ref (GST_TYPE_ELEMENT); g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_A, g_param_spec_double ("A", "A", "A filter coefficient", -G_MAXDOUBLE, G_MAXDOUBLE, 0, G_PARAM_READWRITE)); g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_B, g_param_spec_double ("B", "B", "B filter coefficient", -G_MAXDOUBLE, G_MAXDOUBLE, 0, G_PARAM_READWRITE)); g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_GAIN, g_param_spec_double ("gain", "Gain", "Filter gain", -G_MAXDOUBLE, G_MAXDOUBLE, 0, G_PARAM_READWRITE)); g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_STAGES, g_param_spec_int ("stages", "Stages", "Number of filter stages", 1, G_MAXINT, 1, G_PARAM_READWRITE)); gobject_class->set_property = gst_iir_set_property; gobject_class->get_property = gst_iir_get_property; } static void gst_iir_init (GstIIR * filter) { filter->sinkpad = gst_pad_new_from_template (gst_static_pad_template_get (&gst_filter_sink_template), "sink"); gst_pad_set_chain_function (filter->sinkpad, gst_iir_chain); gst_pad_set_link_function (filter->sinkpad, gst_iir_sink_connect); gst_element_add_pad (GST_ELEMENT (filter), filter->sinkpad); filter->srcpad = gst_pad_new_from_template (gst_static_pad_template_get (&gst_filter_src_template), "src"); gst_element_add_pad (GST_ELEMENT (filter), filter->srcpad); filter->A = 0.0; filter->B = 0.0; filter->gain = 1.0; /* unity gain as default */ filter->stages = 1; filter->state = NULL; } static GstPadLinkReturn gst_iir_sink_connect (GstPad * pad, const GstCaps * caps) { GstIIR *filter; GstPadLinkReturn set_retval; filter = GST_IIR (gst_pad_get_parent (pad)); set_retval = gst_pad_try_set_caps (filter->srcpad, caps); if (set_retval > 0) { /* connection works, so init the filter */ /* FIXME: remember to free it */ filter->state = (IIR_state *) g_malloc (sizeof (IIR_state)); IIR_init (filter->state, filter->stages, filter->gain, &(filter->A), &(filter->B)); } return set_retval; } static void gst_iir_chain (GstPad * pad, GstData * _data) { GstBuffer *buf = GST_BUFFER (_data); GstIIR *filter; gfloat *src; int i; filter = GST_IIR (gst_pad_get_parent (pad)); src = (gfloat *) GST_BUFFER_DATA (buf); /* get a writable buffer */ buf = gst_buffer_copy_on_write (buf); /* do an in-place edit */ for (i = 0; i < GST_BUFFER_SIZE (buf) / sizeof (gfloat); ++i) *(src + i) = (gfloat) IIR_filter (filter->state, (double) *(src + i)); gst_pad_push (filter->srcpad, GST_DATA (buf)); } static void gst_iir_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstIIR *filter; g_return_if_fail (GST_IS_IIR (object)); filter = GST_IIR (object); switch (prop_id) { case ARG_A: filter->A = g_value_get_double (value); break; case ARG_B: filter->B = g_value_get_double (value); break; case ARG_GAIN: filter->gain = g_value_get_double (value); break; case ARG_STAGES: filter->stages = g_value_get_int (value); break; default: break; } } static void gst_iir_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstIIR *filter; g_return_if_fail (GST_IS_IIR (object)); filter = GST_IIR (object); switch (prop_id) { case ARG_A: g_value_set_double (value, filter->A); break; case ARG_B: g_value_set_double (value, filter->B); break; case ARG_GAIN: g_value_set_double (value, filter->gain); break; case ARG_STAGES: g_value_set_int (value, filter->stages); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } }