gstreamer/tests/check/libs/fft.c

505 lines
11 KiB
C
Raw Normal View History

Add libgstfft, a FFT library based on Kiss FFT which is Original commit message from CVS: Reviewed by: Stefan Kost <ensonic@users.sf.net> * configure.ac: * gst-libs/gst/Makefile.am: * gst-libs/gst/fft/Makefile.am: * gst-libs/gst/fft/_kiss_fft_guts_f32.h: * gst-libs/gst/fft/_kiss_fft_guts_f64.h: * gst-libs/gst/fft/_kiss_fft_guts_s16.h: * gst-libs/gst/fft/_kiss_fft_guts_s32.h: * gst-libs/gst/fft/gstfft.c: (gst_fft_next_fast_length): * gst-libs/gst/fft/gstfft.h: * gst-libs/gst/fft/gstfftf32.c: (gst_fft_f32_new), (gst_fft_f32_fft), (gst_fft_f32_inverse_fft), (gst_fft_f32_free), (gst_fft_f32_window), (gst_fft_f32_magnitude), (gst_fft_f32_phase): * gst-libs/gst/fft/gstfftf32.h: * gst-libs/gst/fft/gstfftf64.c: (gst_fft_f64_new), (gst_fft_f64_fft), (gst_fft_f64_inverse_fft), (gst_fft_f64_free), (gst_fft_f64_window), (gst_fft_f64_magnitude), (gst_fft_f64_phase): * gst-libs/gst/fft/gstfftf64.h: * gst-libs/gst/fft/gstffts16.c: (gst_fft_s16_new), (gst_fft_s16_fft), (gst_fft_s16_inverse_fft), (gst_fft_s16_free), (gst_fft_s16_window), (gst_fft_s16_magnitude), (gst_fft_s16_phase): * gst-libs/gst/fft/gstffts16.h: * gst-libs/gst/fft/gstffts32.c: (gst_fft_s32_new), (gst_fft_s32_fft), (gst_fft_s32_inverse_fft), (gst_fft_s32_free), (gst_fft_s32_window), (gst_fft_s32_magnitude), (gst_fft_s32_phase): * gst-libs/gst/fft/gstffts32.h: * gst-libs/gst/fft/kiss_fft_f32.c: (kf_bfly2), (kf_bfly4), (kf_bfly3), (kf_bfly5), (kf_bfly_generic), (kf_work), (kf_factor), (kiss_fft_f32_alloc), (kiss_fft_f32_stride), (kiss_fft_f32), (kiss_fft_f32_cleanup), (kiss_fft_f32_next_fast_size): * gst-libs/gst/fft/kiss_fft_f32.h: * gst-libs/gst/fft/kiss_fft_f64.c: (kf_bfly2), (kf_bfly4), (kf_bfly3), (kf_bfly5), (kf_bfly_generic), (kf_work), (kf_factor), (kiss_fft_f64_alloc), (kiss_fft_f64_stride), (kiss_fft_f64), (kiss_fft_f64_cleanup), (kiss_fft_f64_next_fast_size): * gst-libs/gst/fft/kiss_fft_f64.h: * gst-libs/gst/fft/kiss_fft_s16.c: (kf_bfly2), (kf_bfly4), (kf_bfly3), (kf_bfly5), (kf_bfly_generic), (kf_work), (kf_factor), (kiss_fft_s16_alloc), (kiss_fft_s16_stride), (kiss_fft_s16), (kiss_fft_s16_cleanup), (kiss_fft_s16_next_fast_size): * gst-libs/gst/fft/kiss_fft_s16.h: * gst-libs/gst/fft/kiss_fft_s32.c: (kf_bfly2), (kf_bfly4), (kf_bfly3), (kf_bfly5), (kf_bfly_generic), (kf_work), (kf_factor), (kiss_fft_s32_alloc), (kiss_fft_s32_stride), (kiss_fft_s32), (kiss_fft_s32_cleanup), (kiss_fft_s32_next_fast_size): * gst-libs/gst/fft/kiss_fft_s32.h: * gst-libs/gst/fft/kiss_fftr_f32.c: (kiss_fftr_f32_alloc), (kiss_fftr_f32), (kiss_fftri_f32): * gst-libs/gst/fft/kiss_fftr_f32.h: * gst-libs/gst/fft/kiss_fftr_f64.c: (kiss_fftr_f64_alloc), (kiss_fftr_f64), (kiss_fftri_f64): * gst-libs/gst/fft/kiss_fftr_f64.h: * gst-libs/gst/fft/kiss_fftr_s16.c: (kiss_fftr_s16_alloc), (kiss_fftr_s16), (kiss_fftri_s16): * gst-libs/gst/fft/kiss_fftr_s16.h: * gst-libs/gst/fft/kiss_fftr_s32.c: (kiss_fftr_s32_alloc), (kiss_fftr_s32), (kiss_fftri_s32): * gst-libs/gst/fft/kiss_fftr_s32.h: * gst-libs/gst/fft/kiss_version: * pkgconfig/gstreamer-plugins-base-uninstalled.pc.in: * pkgconfig/gstreamer-plugins-base.pc.in: Add libgstfft, a FFT library based on Kiss FFT which is BSD licensed. Supported sample formats are int16, int32, float and double. For those formats a real FFT and IFFT can be done, different windowing functions can be applied and functions for extracting the magnitude and phase exist. Fixes #468619. * docs/libs/Makefile.am: * docs/libs/gst-plugins-base-libs-docs.sgml: * docs/libs/gst-plugins-base-libs-sections.txt: Integrate libgstfft into the docs. * tests/check/Makefile.am: * tests/check/libs/fft.c: (GST_START_TEST), (fft_suite), (main): Add unit tests for libgstfft, currently only testing the FFT. Unit tests for IFFT will follow soon.
2007-09-06 07:00:36 +00:00
/* GStreamer
*
* unit test for FFT library
*
* Copyright (C) 2007 Sebastian Dröge <slomo@circular-chaos.org>
*
* 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 <gst/check/gstcheck.h>
#include <gst/fft/gstfft.h>
#include <gst/fft/gstffts16.h>
#include <gst/fft/gstffts32.h>
#include <gst/fft/gstfftf32.h>
#include <gst/fft/gstfftf64.h>
GST_START_TEST (test_next_fast_length)
{
fail_unless_equals_int (gst_fft_next_fast_length (13), 16);
fail_unless_equals_int (gst_fft_next_fast_length (30), 30);
fail_unless_equals_int (gst_fft_next_fast_length (31), 32);
fail_unless_equals_int (gst_fft_next_fast_length (1), 2);
}
GST_END_TEST;
GST_START_TEST (test_s16_0hz)
{
gint i;
gint16 *in;
GstFFTS16Complex *out;
gdouble *mag;
GstFFTS16 *ctx;
in = g_new (gint16, 2048);
out = g_new (GstFFTS16Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s16_new (2048, FALSE);
for (i = 0; i < 2048; i++)
in[i] = G_MAXINT16;
gst_fft_s16_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s16_fft (ctx, in, out);
gst_fft_s16_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i < 2 && mag[i] < -15.0);
fail_if (i >= 2 && mag[i] > -60.0);
}
gst_fft_s16_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_s16_11025hz)
{
gint i;
gint16 *in;
GstFFTS16Complex *out;
gdouble *mag;
GstFFTS16 *ctx;
in = g_new (gint16, 2048);
out = g_new (GstFFTS16Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s16_new (2048, FALSE);
for (i = 0; i < 2048; i += 4) {
in[i] = 0;
in[i + 1] = G_MAXINT16;
in[i + 2] = 0;
in[i + 3] = G_MININT16;
}
gst_fft_s16_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s16_fft (ctx, in, out);
gst_fft_s16_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (abs (512 - i) < 2 && mag[i] < -20.0);
fail_if (abs (512 - i) >= 2 && mag[i] > -60.0);
}
gst_fft_s16_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_s16_22050hz)
{
gint i;
gint16 *in;
GstFFTS16Complex *out;
gdouble *mag;
GstFFTS16 *ctx;
in = g_new (gint16, 2048);
out = g_new (GstFFTS16Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s16_new (2048, FALSE);
for (i = 0; i < 2048; i += 2) {
in[i] = G_MAXINT16;
in[i + 1] = G_MININT16;
}
gst_fft_s16_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s16_fft (ctx, in, out);
gst_fft_s16_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i > 1022 && mag[i] < -20.0);
fail_if (i <= 1022 && mag[i] > -60.0);
}
gst_fft_s16_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_s32_0hz)
{
gint i;
gint32 *in;
GstFFTS32Complex *out;
gdouble *mag;
GstFFTS32 *ctx;
in = g_new (gint32, 2048);
out = g_new (GstFFTS32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s32_new (2048, FALSE);
for (i = 0; i < 2048; i++)
in[i] = 2147483647;
gst_fft_s32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s32_fft (ctx, in, out);
gst_fft_s32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i < 2 && mag[i] < -15.0);
fail_if (i >= 2 && mag[i] > -60.0);
}
gst_fft_s32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_s32_11025hz)
{
gint i;
gint32 *in;
GstFFTS32Complex *out;
gdouble *mag;
GstFFTS32 *ctx;
in = g_new (gint32, 2048);
out = g_new (GstFFTS32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s32_new (2048, FALSE);
for (i = 0; i < 2048; i += 4) {
in[i] = 0;
in[i + 1] = G_MAXINT32;
in[i + 2] = 0;
in[i + 3] = G_MININT32;
}
gst_fft_s32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s32_fft (ctx, in, out);
gst_fft_s32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (abs (512 - i) < 2 && mag[i] < -20.0);
fail_if (abs (512 - i) >= 2 && mag[i] > -60.0);
}
gst_fft_s32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_s32_22050hz)
{
gint i;
gint32 *in;
GstFFTS32Complex *out;
gdouble *mag;
GstFFTS32 *ctx;
in = g_new (gint32, 2048);
out = g_new (GstFFTS32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_s32_new (2048, FALSE);
for (i = 0; i < 2048; i += 2) {
in[i] = G_MAXINT32;
in[i + 1] = G_MININT32;
}
gst_fft_s32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_s32_fft (ctx, in, out);
gst_fft_s32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i > 1022 && mag[i] < -20.0);
fail_if (i <= 1022 && mag[i] > -60.0);
}
gst_fft_s32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f32_0hz)
{
gint i;
gfloat *in;
GstFFTF32Complex *out;
gdouble *mag;
GstFFTF32 *ctx;
in = g_new (gfloat, 2048);
out = g_new (GstFFTF32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f32_new (2048, FALSE);
for (i = 0; i < 2048; i++)
in[i] = 1.0;
gst_fft_f32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f32_fft (ctx, in, out);
gst_fft_f32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i < 2 && mag[i] < -15.0);
fail_if (i >= 2 && mag[i] > -60.0);
}
gst_fft_f32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f32_11025hz)
{
gint i;
gfloat *in;
GstFFTF32Complex *out;
gdouble *mag;
GstFFTF32 *ctx;
in = g_new (gfloat, 2048);
out = g_new (GstFFTF32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f32_new (2048, FALSE);
for (i = 0; i < 2048; i += 4) {
in[i] = 0.0;
in[i + 1] = 1.0;
in[i + 2] = 0.0;
in[i + 3] = -1.0;
}
gst_fft_f32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f32_fft (ctx, in, out);
gst_fft_f32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (abs (512 - i) < 2 && mag[i] < -20.0);
fail_if (abs (512 - i) >= 2 && mag[i] > -60.0);
}
gst_fft_f32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f32_22050hz)
{
gint i;
gfloat *in;
GstFFTF32Complex *out;
gdouble *mag;
GstFFTF32 *ctx;
in = g_new (gfloat, 2048);
out = g_new (GstFFTF32Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f32_new (2048, FALSE);
for (i = 0; i < 2048; i += 2) {
in[i] = 1.0;
in[i + 1] = -1.0;
}
gst_fft_f32_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f32_fft (ctx, in, out);
gst_fft_f32_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i > 1022 && mag[i] < -20.0);
fail_if (i <= 1022 && mag[i] > -60.0);
}
gst_fft_f32_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f64_0hz)
{
gint i;
gdouble *in;
GstFFTF64Complex *out;
gdouble *mag;
GstFFTF64 *ctx;
in = g_new (gdouble, 2048);
out = g_new (GstFFTF64Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f64_new (2048, FALSE);
for (i = 0; i < 2048; i++)
in[i] = 1.0;
gst_fft_f64_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f64_fft (ctx, in, out);
gst_fft_f64_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i < 2 && mag[i] < -15.0);
fail_if (i >= 2 && mag[i] > -60.0);
}
gst_fft_f64_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f64_11025hz)
{
gint i;
gdouble *in;
GstFFTF64Complex *out;
gdouble *mag;
GstFFTF64 *ctx;
in = g_new (gdouble, 2048);
out = g_new (GstFFTF64Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f64_new (2048, FALSE);
for (i = 0; i < 2048; i += 4) {
in[i] = 0.0;
in[i + 1] = 1.0;
in[i + 2] = 0.0;
in[i + 3] = -1.0;
}
gst_fft_f64_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f64_fft (ctx, in, out);
gst_fft_f64_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (abs (512 - i) < 2 && mag[i] < -20.0);
fail_if (abs (512 - i) >= 2 && mag[i] > -60.0);
}
gst_fft_f64_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
GST_START_TEST (test_f64_22050hz)
{
gint i;
gdouble *in;
GstFFTF64Complex *out;
gdouble *mag;
GstFFTF64 *ctx;
in = g_new (gdouble, 2048);
out = g_new (GstFFTF64Complex, 1025);
mag = g_new (gdouble, 1025);
ctx = gst_fft_f64_new (2048, FALSE);
for (i = 0; i < 2048; i += 2) {
in[i] = 1.0;
in[i + 1] = -1.0;
}
gst_fft_f64_window (ctx, in, GST_FFT_WINDOW_HAMMING);
gst_fft_f64_fft (ctx, in, out);
gst_fft_f64_magnitude (ctx, out, mag, TRUE);
for (i = 0; i < 1025; i++) {
fail_if (i > 1022 && mag[i] < -20.0);
fail_if (i <= 1022 && mag[i] > -60.0);
}
gst_fft_f64_free (ctx);
g_free (in);
g_free (out);
g_free (mag);
}
GST_END_TEST;
static Suite *
fft_suite (void)
{
Suite *s = suite_create ("fft library");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_next_fast_length);
tcase_add_test (tc_chain, test_s16_0hz);
tcase_add_test (tc_chain, test_s16_11025hz);
tcase_add_test (tc_chain, test_s16_22050hz);
tcase_add_test (tc_chain, test_s32_0hz);
tcase_add_test (tc_chain, test_s32_11025hz);
tcase_add_test (tc_chain, test_s32_22050hz);
tcase_add_test (tc_chain, test_f32_0hz);
tcase_add_test (tc_chain, test_f32_11025hz);
tcase_add_test (tc_chain, test_f32_22050hz);
tcase_add_test (tc_chain, test_f64_0hz);
tcase_add_test (tc_chain, test_f64_11025hz);
tcase_add_test (tc_chain, test_f64_22050hz);
return s;
}
int
main (int argc, char **argv)
{
int nf;
Suite *s = fft_suite ();
SRunner *sr = srunner_create (s);
gst_check_init (&argc, &argv);
srunner_run_all (sr, CK_NORMAL);
nf = srunner_ntests_failed (sr);
srunner_free (sr);
return nf;
}