diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/check/Makefile.am | 2 | ||||
| -rw-r--r-- | tests/check/elements/audiofirfilter.c | 169 | ||||
| -rw-r--r-- | tests/check/elements/audioiirfilter.c | 179 | ||||
| -rw-r--r-- | tests/examples/Makefile.am | 4 | ||||
| -rw-r--r-- | tests/examples/audiofx/Makefile.am | 7 | ||||
| -rw-r--r-- | tests/examples/audiofx/firfilter-example.c | 161 | ||||
| -rw-r--r-- | tests/examples/audiofx/iirfilter-example.c | 137 |
7 files changed, 657 insertions, 2 deletions
diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index d1a55197..5cc63dbb 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -72,10 +72,12 @@ check_PROGRAMS = \ elements/audioinvert \ elements/audiochebband \ elements/audiocheblimit \ + elements/audioiirfilter \ elements/audioamplify \ elements/audiodynamic \ elements/audiowsincband \ elements/audiowsinclimit \ + elements/audiofirfilter \ elements/avimux \ elements/avisubtitle \ elements/deinterleave \ diff --git a/tests/check/elements/audiofirfilter.c b/tests/check/elements/audiofirfilter.c new file mode 100644 index 00000000..c4eb7238 --- /dev/null +++ b/tests/check/elements/audiofirfilter.c @@ -0,0 +1,169 @@ +/* GStreamer + * + * Copyright (C) 2009 Sebastian Dröge <sebastian.droege@collabora.co.uk> + * + * 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 + */ + +#include <gst/gst.h> +#include <gst/check/gstcheck.h> + +static gboolean have_eos = FALSE; + +static gboolean +on_message (GstBus * bus, GstMessage * message, gpointer user_data) +{ + GMainLoop *loop = (GMainLoop *) user_data; + + switch (GST_MESSAGE_TYPE (message)) { + case GST_MESSAGE_ERROR: + case GST_MESSAGE_WARNING: + g_assert_not_reached (); + g_main_loop_quit (loop); + break; + + case GST_MESSAGE_EOS: + have_eos = TRUE; + g_main_loop_quit (loop); + break; + default: + break; + } + + return TRUE; +} + +static void +on_rate_changed (GstElement * element, gint rate, gpointer user_data) +{ + GValueArray *va; + GValue v = { 0, }; + + fail_unless (rate > 0); + + va = g_value_array_new (6); + + g_value_init (&v, G_TYPE_DOUBLE); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 1.0); + g_value_array_append (va, &v); + g_value_reset (&v); + + g_object_set (G_OBJECT (element), "kernel", va, NULL); + + g_value_array_free (va); +} + +static gboolean have_data = FALSE; + +static void +on_handoff (GstElement * object, GstBuffer * buffer, GstPad * pad, + gpointer user_data) +{ + if (!have_data) { + gdouble *data = (gdouble *) GST_BUFFER_DATA (buffer); + + fail_unless (GST_BUFFER_SIZE (buffer) > 5 * sizeof (gdouble)); + fail_unless (data[0] == 0.0); + fail_unless (data[1] == 0.0); + fail_unless (data[2] == 0.0); + fail_unless (data[3] == 0.0); + fail_unless (data[4] == 0.0); + fail_unless (data[5] != 0.0); + have_data = TRUE; + } +} + +GST_START_TEST (test_pipeline) +{ + GstElement *pipeline, *src, *filter, *sink; + GstBus *bus; + GMainLoop *loop; + + have_data = FALSE; + have_eos = FALSE; + + pipeline = gst_element_factory_make ("pipeline", NULL); + fail_unless (pipeline != NULL); + + src = gst_element_factory_make ("audiotestsrc", NULL); + fail_unless (src != NULL); + g_object_set (G_OBJECT (src), "num-buffers", 1000, NULL); + + filter = gst_element_factory_make ("audiofirfilter", NULL); + fail_unless (filter != NULL); + g_signal_connect (G_OBJECT (filter), "rate-changed", + G_CALLBACK (on_rate_changed), NULL); + + sink = gst_element_factory_make ("fakesink", NULL); + fail_unless (sink != NULL); + g_object_set (G_OBJECT (sink), "signal-handoffs", TRUE, NULL); + g_signal_connect (G_OBJECT (sink), "handoff", G_CALLBACK (on_handoff), NULL); + + gst_bin_add_many (GST_BIN (pipeline), src, filter, sink, NULL); + fail_unless (gst_element_link_many (src, filter, sink, NULL)); + + loop = g_main_loop_new (NULL, FALSE); + + bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); + gst_bus_add_signal_watch (bus); + g_signal_connect (G_OBJECT (bus), "message", G_CALLBACK (on_message), loop); + gst_object_unref (GST_OBJECT (bus)); + + fail_if (gst_element_set_state (pipeline, + GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE); + + g_main_loop_run (loop); + + fail_unless (have_data); + fail_unless (have_eos); + + fail_unless (gst_element_set_state (pipeline, + GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS); + + g_main_loop_unref (loop); + gst_object_unref (pipeline); +} + +GST_END_TEST; + +static Suite * +audiofirfilter_suite (void) +{ + Suite *s = suite_create ("audiofirfilter"); + TCase *tc_chain = tcase_create ("general"); + + suite_add_tcase (s, tc_chain); + tcase_add_test (tc_chain, test_pipeline); + + return s; +} + +GST_CHECK_MAIN (audiofirfilter); diff --git a/tests/check/elements/audioiirfilter.c b/tests/check/elements/audioiirfilter.c new file mode 100644 index 00000000..d144792b --- /dev/null +++ b/tests/check/elements/audioiirfilter.c @@ -0,0 +1,179 @@ +/* GStreamer + * + * Copyright (C) 2009 Sebastian Dröge <sebastian.droege@collabora.co.uk> + * + * 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 + */ + +#include <gst/gst.h> +#include <gst/check/gstcheck.h> + +static gboolean have_eos = FALSE; + +static gboolean +on_message (GstBus * bus, GstMessage * message, gpointer user_data) +{ + GMainLoop *loop = (GMainLoop *) user_data; + + switch (GST_MESSAGE_TYPE (message)) { + case GST_MESSAGE_ERROR: + case GST_MESSAGE_WARNING: + g_assert_not_reached (); + g_main_loop_quit (loop); + break; + + case GST_MESSAGE_EOS: + have_eos = TRUE; + g_main_loop_quit (loop); + break; + default: + break; + } + + return TRUE; +} + +static void +on_rate_changed (GstElement * element, gint rate, gpointer user_data) +{ + GValueArray *va; + GValue v = { 0, }; + + fail_unless (rate > 0); + + va = g_value_array_new (6); + + g_value_init (&v, G_TYPE_DOUBLE); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + g_value_set_double (&v, 1.0); + g_value_array_append (va, &v); + g_value_reset (&v); + + g_object_set (G_OBJECT (element), "a", va, NULL); + + g_value_array_free (va); + + va = g_value_array_new (6); + + g_value_set_double (&v, 0.0); + g_value_array_append (va, &v); + g_value_reset (&v); + + g_object_set (G_OBJECT (element), "b", va, NULL); + + g_value_array_free (va); +} + +static gboolean have_data = FALSE; + +static void +on_handoff (GstElement * object, GstBuffer * buffer, GstPad * pad, + gpointer user_data) +{ + if (!have_data) { + gdouble *data = (gdouble *) GST_BUFFER_DATA (buffer); + + fail_unless (GST_BUFFER_SIZE (buffer) > 5 * sizeof (gdouble)); + fail_unless (data[0] == 0.0); + fail_unless (data[1] == 0.0); + fail_unless (data[2] == 0.0); + fail_unless (data[3] == 0.0); + fail_unless (data[4] == 0.0); + fail_unless (data[5] != 0.0); + have_data = TRUE; + } +} + +GST_START_TEST (test_pipeline) +{ + GstElement *pipeline, *src, *filter, *sink; + GstBus *bus; + GMainLoop *loop; + + have_data = FALSE; + have_eos = FALSE; + + pipeline = gst_element_factory_make ("pipeline", NULL); + fail_unless (pipeline != NULL); + + src = gst_element_factory_make ("audiotestsrc", NULL); + fail_unless (src != NULL); + g_object_set (G_OBJECT (src), "num-buffers", 1000, NULL); + + filter = gst_element_factory_make ("audioiirfilter", NULL); + fail_unless (filter != NULL); + g_signal_connect (G_OBJECT (filter), "rate-changed", + G_CALLBACK (on_rate_changed), NULL); + + sink = gst_element_factory_make ("fakesink", NULL); + fail_unless (sink != NULL); + g_object_set (G_OBJECT (sink), "signal-handoffs", TRUE, NULL); + g_signal_connect (G_OBJECT (sink), "handoff", G_CALLBACK (on_handoff), NULL); + + gst_bin_add_many (GST_BIN (pipeline), src, filter, sink, NULL); + fail_unless (gst_element_link_many (src, filter, sink, NULL)); + + loop = g_main_loop_new (NULL, FALSE); + + bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); + gst_bus_add_signal_watch (bus); + g_signal_connect (G_OBJECT (bus), "message", G_CALLBACK (on_message), loop); + gst_object_unref (GST_OBJECT (bus)); + + fail_if (gst_element_set_state (pipeline, + GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE); + + g_main_loop_run (loop); + + fail_unless (have_data); + fail_unless (have_eos); + + fail_unless (gst_element_set_state (pipeline, + GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS); + + g_main_loop_unref (loop); + gst_object_unref (pipeline); +} + +GST_END_TEST; + +static Suite * +audioiirfilter_suite (void) +{ + Suite *s = suite_create ("audioiirfilter"); + TCase *tc_chain = tcase_create ("general"); + + suite_add_tcase (s, tc_chain); + tcase_add_test (tc_chain, test_pipeline); + + return s; +} + +GST_CHECK_MAIN (audioiirfilter); diff --git a/tests/examples/Makefile.am b/tests/examples/Makefile.am index aad6dc18..1199543e 100644 --- a/tests/examples/Makefile.am +++ b/tests/examples/Makefile.am @@ -1,3 +1,3 @@ -SUBDIRS = equalizer level rtp spectrum +SUBDIRS = audiofx equalizer level rtp spectrum -DIST_SUBDIRS = equalizer level rtp spectrum +DIST_SUBDIRS = audiofx equalizer level rtp spectrum diff --git a/tests/examples/audiofx/Makefile.am b/tests/examples/audiofx/Makefile.am new file mode 100644 index 00000000..efcf785b --- /dev/null +++ b/tests/examples/audiofx/Makefile.am @@ -0,0 +1,7 @@ +noinst_PROGRAMS = firfilter-example iirfilter-example + +firfilter_example_CFLAGS = $(GST_CFLAGS) $(GST_PLUGINS_BASE_CFLAGS) +firfilter_example_LDADD = $(GST_LIBS) $(GST_PLUGINS_BASE_LIBS) -lgstfft-@GST_MAJORMINOR@ $(LIBM) + +iirfilter_example_CFLAGS = $(GST_CFLAGS) +iirfilter_example_LDADD = $(GST_LIBS) $(LIBM) diff --git a/tests/examples/audiofx/firfilter-example.c b/tests/examples/audiofx/firfilter-example.c new file mode 100644 index 00000000..c5d56da1 --- /dev/null +++ b/tests/examples/audiofx/firfilter-example.c @@ -0,0 +1,161 @@ +/* GStreamer + * Copyright (C) 2009 Sebastian Droege <sebastian.droege@collabora.co.uk> + * + * 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. + */ + +/* This small sample application creates a bandpass FIR filter + * by transforming the frequency response to the filter kernel. + */ + +#include <string.h> +#include <math.h> + +#include <gst/gst.h> +#include <gst/fft/gstfftf64.h> + +static gboolean +on_message (GstBus * bus, GstMessage * message, gpointer user_data) +{ + GMainLoop *loop = (GMainLoop *) user_data; + + switch (GST_MESSAGE_TYPE (message)) { + case GST_MESSAGE_ERROR: + g_error ("Got ERROR"); + g_main_loop_quit (loop); + break; + case GST_MESSAGE_WARNING: + g_warning ("Got WARNING"); + g_main_loop_quit (loop); + break; + case GST_MESSAGE_EOS: + g_main_loop_quit (loop); + break; + default: + break; + } + + return TRUE; +} + +static void +on_rate_changed (GstElement * element, gint rate, gpointer user_data) +{ + GValueArray *va; + GValue v = { 0, }; + GstFFTF64 *fft; + GstFFTF64Complex frequency_response[17]; + gdouble tmp[32]; + gdouble filter_kernel[32]; + guint i; + + /* Create the frequency response: zero outside + * a small frequency band */ + for (i = 0; i < 17; i++) { + if (i < 5 || i > 11) + frequency_response[i].r = 0.0; + else + frequency_response[i].r = 1.0; + + frequency_response[i].i = 0.0; + } + + /* Calculate the inverse FT of the frequency response */ + fft = gst_fft_f64_new (32, TRUE); + gst_fft_f64_inverse_fft (fft, frequency_response, tmp); + gst_fft_f64_free (fft); + + /* Shift the inverse FT of the frequency response by 16, + * i.e. the half of the kernel length to get the + * impulse response. See http://www.dspguide.com/ch17/1.htm + * for more information. + */ + for (i = 0; i < 32; i++) + filter_kernel[i] = tmp[(i + 16) % 32]; + + /* Apply the hamming window to the impulse response to get + * a better result than given from the rectangular window + */ + for (i = 0; i < 32; i++) + filter_kernel[i] *= (0.54 - 0.46 * cos (2 * M_PI * i / 32)); + + va = g_value_array_new (1); + + g_value_init (&v, G_TYPE_DOUBLE); + for (i = 0; i < 32; i++) { + g_value_set_double (&v, filter_kernel[i]); + g_value_array_append (va, &v); + g_value_reset (&v); + } + g_object_set (G_OBJECT (element), "kernel", va, NULL); + /* Latency is 1/2 of the kernel length for this method of + * calculating a filter kernel from the frequency response + */ + g_object_set (G_OBJECT (element), "latency", 32 / 2, NULL); + g_value_array_free (va); +} + +gint +main (gint argc, gchar * argv[]) +{ + GstElement *pipeline, *src, *filter, *conv, *sink; + GstBus *bus; + GMainLoop *loop; + + gst_init (NULL, NULL); + + pipeline = gst_element_factory_make ("pipeline", NULL); + + src = gst_element_factory_make ("audiotestsrc", NULL); + g_object_set (G_OBJECT (src), "wave", 5, NULL); + + filter = gst_element_factory_make ("audiofirfilter", NULL); + g_signal_connect (G_OBJECT (filter), "rate-changed", + G_CALLBACK (on_rate_changed), NULL); + + conv = gst_element_factory_make ("audioconvert", NULL); + + sink = gst_element_factory_make ("autoaudiosink", NULL); + g_return_val_if_fail (sink != NULL, -1); + + gst_bin_add_many (GST_BIN (pipeline), src, filter, conv, sink, NULL); + if (!gst_element_link_many (src, filter, conv, sink, NULL)) { + g_error ("Failed to link elements"); + return -2; + } + + loop = g_main_loop_new (NULL, FALSE); + + bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); + gst_bus_add_signal_watch (bus); + g_signal_connect (G_OBJECT (bus), "message", G_CALLBACK (on_message), loop); + gst_object_unref (GST_OBJECT (bus)); + + if (gst_element_set_state (pipeline, + GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE) { + g_error ("Failed to go into PLAYING state"); + return -3; + } + + g_main_loop_run (loop); + + gst_element_set_state (pipeline, GST_STATE_NULL); + + g_main_loop_unref (loop); + gst_object_unref (pipeline); + + return 0; +} diff --git a/tests/examples/audiofx/iirfilter-example.c b/tests/examples/audiofx/iirfilter-example.c new file mode 100644 index 00000000..8ccffb74 --- /dev/null +++ b/tests/examples/audiofx/iirfilter-example.c @@ -0,0 +1,137 @@ +/* GStreamer + * Copyright (C) 2009 Sebastian Droege <sebastian.droege@collabora.co.uk> + * + * 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. + */ + +/* This small sample application creates a lowpass IIR filter + * and applies it to white noise. + * See http://www.dspguide.com/ch19/2.htm for a description + * of the IIR filter that is used. + */ + +#include <string.h> +#include <math.h> + +#include <gst/gst.h> + +/* Cutoff of 4000 Hz */ +#define CUTOFF (4000.0) + +static gboolean +on_message (GstBus * bus, GstMessage * message, gpointer user_data) +{ + GMainLoop *loop = (GMainLoop *) user_data; + + switch (GST_MESSAGE_TYPE (message)) { + case GST_MESSAGE_ERROR: + g_error ("Got ERROR"); + g_main_loop_quit (loop); + break; + case GST_MESSAGE_WARNING: + g_warning ("Got WARNING"); + g_main_loop_quit (loop); + break; + case GST_MESSAGE_EOS: + g_main_loop_quit (loop); + break; + default: + break; + } + + return TRUE; +} + +static void +on_rate_changed (GstElement * element, gint rate, gpointer user_data) +{ + GValueArray *va; + GValue v = { 0, }; + gdouble x; + + if (rate / 2.0 > CUTOFF) + x = exp (-2.0 * M_PI * (CUTOFF / rate)); + else + x = 0.0; + + va = g_value_array_new (1); + + g_value_init (&v, G_TYPE_DOUBLE); + g_value_set_double (&v, 1.0 - x); + g_value_array_append (va, &v); + g_value_reset (&v); + g_object_set (G_OBJECT (element), "a", va, NULL); + g_value_array_free (va); + + va = g_value_array_new (1); + g_value_set_double (&v, x); + g_value_array_append (va, &v); + g_value_reset (&v); + g_object_set (G_OBJECT (element), "b", va, NULL); + g_value_array_free (va); +} + +gint +main (gint argc, gchar * argv[]) +{ + GstElement *pipeline, *src, *filter, *conv, *sink; + GstBus *bus; + GMainLoop *loop; + + gst_init (NULL, NULL); + + pipeline = gst_element_factory_make ("pipeline", NULL); + + src = gst_element_factory_make ("audiotestsrc", NULL); + g_object_set (G_OBJECT (src), "wave", 5, NULL); + + filter = gst_element_factory_make ("audioiirfilter", NULL); + g_signal_connect (G_OBJECT (filter), "rate-changed", + G_CALLBACK (on_rate_changed), NULL); + + conv = gst_element_factory_make ("audioconvert", NULL); + + sink = gst_element_factory_make ("autoaudiosink", NULL); + g_return_val_if_fail (sink != NULL, -1); + + gst_bin_add_many (GST_BIN (pipeline), src, filter, conv, sink, NULL); + if (!gst_element_link_many (src, filter, conv, sink, NULL)) { + g_error ("Failed to link elements"); + return -2; + } + + loop = g_main_loop_new (NULL, FALSE); + + bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); + gst_bus_add_signal_watch (bus); + g_signal_connect (G_OBJECT (bus), "message", G_CALLBACK (on_message), loop); + gst_object_unref (GST_OBJECT (bus)); + + if (gst_element_set_state (pipeline, + GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE) { + g_error ("Failed to go into PLAYING state"); + return -3; + } + + g_main_loop_run (loop); + + gst_element_set_state (pipeline, GST_STATE_NULL); + + g_main_loop_unref (loop); + gst_object_unref (pipeline); + + return 0; +} |