/* 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/gst_global.h>

#include <gstcheck.h>

#include <gstfft.h>

#include <gstffts16.h>

#include <gstffts32.h>

#include <gstfftf32.h>

#include <gstfftf64.h>

#define LOG_FILE "c:\\logs\\gstfft_logs.txt" 

#include "std_log_result.h" 

#define LOG_FILENAME_LINE __FILE__, __LINE__

//char* xmlfile = "gstsystemclock";

void create_xml(int result)

{

    if(result)

        assert_failed = 1;

    testResultXml(xmlfile);

    close_log_file();

}

void test_next_fast_length()

{

	xmlfile = "test_next_fast_length";

  std_log(LOG_FILENAME_LINE, "Test Started 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);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s16_0hz()

{

  gint i;

  gint16 *in;

  GstFFTS16Complex *out;

  GstFFTS16 *ctx;

	xmlfile = "test_s16_0hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s16_0hz");

  in = g_new (gint16, 2048);

  out = g_new (GstFFTS16Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 32767.0 * 32767.0;

    mag = 10.0 * log10 (mag);

    if (i < 2)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -55.0);

  }

  gst_fft_s16_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s16_11025hz()

{

  gint i;

  gint16 *in;

  GstFFTS16Complex *out;

  GstFFTS16 *ctx;

 xmlfile = "test_s16_11025hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s16_11025hz");

  in = g_new (gint16, 2048);

  out = g_new (GstFFTS16Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 32767.0 * 32767.0;

    mag = 10.0 * log10 (mag);

    if (abs (512 - i) < 2)

      fail_unless (mag > -20.0);

    else

      fail_unless (mag < -55.0);

  }

  gst_fft_s16_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s16_22050hz()

{

  gint i;

  gint16 *in;

  GstFFTS16Complex *out;

  GstFFTS16 *ctx;

  xmlfile = "test_s16_22050hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s16_22050hz");

  in = g_new (gint16, 2048);

  out = g_new (GstFFTS16Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 32767.0 * 32767.0;

    mag = 10.0 * log10 (mag);

    if (i > 1022)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -55.0);

  }

  gst_fft_s16_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s32_0hz()

{

  gint i;

  gint32 *in;

  GstFFTS32Complex *out;

  GstFFTS32 *ctx;

    xmlfile = "test_s32_0hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s32_0hz");

  in = g_new (gint32, 2048);

  out = g_new (GstFFTS32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2147483647.0 * 2147483647.0;

    mag = 10.0 * log10 (mag);

    if (i < 2)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_s32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s32_11025hz()

{

  gint i;

  gint32 *in;

  GstFFTS32Complex *out;

  GstFFTS32 *ctx;

  xmlfile = "test_s32_11025hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s32_11025hz");

  in = g_new (gint32, 2048);

  out = g_new (GstFFTS32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2147483647.0 * 2147483647.0;

    mag = 10.0 * log10 (mag);

    if (abs (512 - i) < 2)

      fail_unless (mag > -20.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_s32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_s32_22050hz()

{

  gint i;

  gint32 *in;

  GstFFTS32Complex *out;

  GstFFTS32 *ctx;

    xmlfile = "test_s32_22050hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_s32_22050hz");

  in = g_new (gint32, 2048);

  out = g_new (GstFFTS32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2147483647.0 * 2147483647.0;

    mag = 10.0 * log10 (mag);

    if (i > 1022)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_s32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_f32_0hz()

{

  gint i;

  gfloat *in;

  GstFFTF32Complex *out;

  GstFFTF32 *ctx;

      xmlfile = "test_f32_0hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_f32_0hz");

  in = g_new (gfloat, 2048);

  out = g_new (GstFFTF32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2048.0 * 2048.0;

    mag = 10.0 * log10 (mag);

    if (i < 2)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_f32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_f32_11025hz()

{

  gint i;

  gfloat *in;

  GstFFTF32Complex *out;

  GstFFTF32 *ctx;

        xmlfile = "test_f32_11025hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_f32_11025hz");

  in = g_new (gfloat, 2048);

  out = g_new (GstFFTF32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2048.0 * 2048.0;

    mag = 10.0 * log10 (mag);

    if (abs (512 - i) < 2)

      fail_unless (mag > -20.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_f32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_f32_22050hz()

{

  gint i;

  gfloat *in;

  GstFFTF32Complex *out;

  GstFFTF32 *ctx;

          xmlfile = "test_f32_22050hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_f32_22050hz");

  in = g_new (gfloat, 2048);

  out = g_new (GstFFTF32Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2048.0 * 2048.0;

    mag = 10.0 * log10 (mag);

    if (i > 1022)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_f32_free (ctx);

  g_free (in);

  g_free (out);

    std_log(LOG_FILENAME_LINE, "Test Successful");

  create_xml(0);

}

void test_f64_0hz()

{

  gint i;

  gdouble *in;

  GstFFTF64Complex *out;

  GstFFTF64 *ctx;

            xmlfile = "test_f64_0hz";

  std_log(LOG_FILENAME_LINE, "Test Started test_f64_0hz");

  in = g_new (gdouble, 2048);

  out = g_new (GstFFTF64Complex, 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);

  for (i = 0; i < 1025; i++) {

    gdouble mag;

    mag = (gdouble) out[i].r * (gdouble) out[i].r;

    mag += (gdouble) out[i].i * (gdouble) out[i].i;

    mag /= 2048.0 * 2048.0;

    mag = 10.0 * log10 (mag);

    if (i < 2)

      fail_unless (mag > -15.0);

    else

      fail_unless (mag < -60.0);

  }

  gst_fft_f64_free (ctx);

  g_free (in);

  g_free (out);