/* GStreamer unit tests for audiorate

 *

 * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>

 *

 * 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 <config.h>

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

#include "std_log_result.h" 

#define LOG_FILENAME_LINE __FILE__, __LINE__

#include <gst/check/gstcheck.h>

void create_xml(int result)

{

    if(result)

        assert_failed = 1;

    testResultXml(xmlfile);

    close_log_file();

}

#include "libgstreamer_wsd_solution.h" 

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(buffers,gstcheck,GList*)

#define buffers (*GET_GSTREAMER_WSD_VAR_NAME(buffers,gstcheck,g)())

#else 

#endif

//---------

#if EMULATOR

static GET_GLOBAL_VAR_FROM_TLS(raised_critical,gstcheck,gboolean)

#define _gst_check_raised_critical (*GET_GSTREAMER_WSD_VAR_NAME(raised_critical,gstcheck,g)())

#else 

extern gboolean _gst_check_raised_critical ;

#endif

//gboolean _gst_check_raised_warning = FALSE;

#if EMULATOR

static GET_GLOBAL_VAR_FROM_TLS(raised_warning,gstcheck,gboolean)

#define _gst_check_raised_warning (*GET_GSTREAMER_WSD_VAR_NAME(raised_warning,gstcheck,g)())

#else 

extern gboolean _gst_check_raised_warning ;

#endif

//gboolean _gst_check_expecting_log = FALSE;

#if EMULATOR

static GET_GLOBAL_VAR_FROM_TLS(expecting_log,gstcheck,gboolean)

#define _gst_check_expecting_log (*GET_GSTREAMER_WSD_VAR_NAME(expecting_log,gstcheck,g)())

#else 

extern gboolean _gst_check_expecting_log ;

#endif

//gboolean _gst_check_expecting_log = FALSE;

#if EMULATOR

static GET_GLOBAL_VAR_FROM_TLS(threads_running,gstcheck,gboolean)

#define _gst_check_threads_running (*GET_GSTREAMER_WSD_VAR_NAME(threads_running,gstcheck,g)())

#else 

extern gboolean _gst_check_threads_running ;

#endif

#ifndef EMULATOR

int failed=0;

FILE* fp_std_log_t=NULL;

#define assert_failed *(get_assert_failed())

#define fp_std_log *(get_fp_std_log())

#endif

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(thread_list,gstcheck,GList*)

#define thread_list (*GET_GSTREAMER_WSD_VAR_NAME(thread_list,gstcheck,g)())

#else 

#endif

//GMutex *mutex;

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(mutex,gstcheck,GMutex*)

#define mutex (*GET_GSTREAMER_WSD_VAR_NAME(mutex,gstcheck,g)())

#else 

extern GMutex *mutex;

#endif

//GCond *start_cond;              /* used to notify main thread of thread startups */

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(start_cond,gstcheck,GCond*)

#define start_cond (*GET_GSTREAMER_WSD_VAR_NAME(start_cond,gstcheck,g)())

#else 

extern GCond *start_cond;

#endif

//GCond *sync_cond;               /* used to synchronize all threads and main thread */

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(sync_cond,gstcheck,GCond*)

#define sync_cond (*GET_GSTREAMER_WSD_VAR_NAME(sync_cond,gstcheck,g)())

#else 

extern GCond *sync_cond;

#endif

/* helper element to insert additional buffers overlapping with previous ones */

static gdouble injector_inject_probability = 0.0;

typedef GstElement TestInjector;

typedef GstElementClass TestInjectorClass;

GST_BOILERPLATE (TestInjector, test_injector, GstElement, GST_TYPE_ELEMENT);

#define INJECTOR_CAPS \

  "audio/x-raw-float, "                                  \

    "rate = (int) [ 1, MAX ], "                          \

    "channels = (int) [ 1, 8 ], "                        \

    "endianness = (int) BYTE_ORDER, "                    \

    "width = (int) 32;"                                  \

  "audio/x-raw-int, "                                    \

    "rate = (int) [ 1, MAX ], "                          \

    "channels = (int) [ 1, 8 ], "                        \

    "endianness = (int) { LITTLE_ENDIAN, BIG_ENDIAN }, " \

    "width = (int) { 8, 16, 32 }, "                      \

    "depth = (int) [ 1, 32 ], "                          \

    "signed = (boolean) { true, false }"

static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS (INJECTOR_CAPS));

static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",

    GST_PAD_SINK,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS (INJECTOR_CAPS));

static void

test_injector_base_init (gpointer g_class)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&src_template));

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&sink_template));

}

static void

test_injector_class_init (TestInjectorClass * klass)

{

  /* nothing to do here */

}

static GstFlowReturn

test_injector_chain (GstPad * pad, GstBuffer * buf)

{

  GstFlowReturn ret;

  GstPad *srcpad;

  srcpad = gst_element_get_pad (GST_ELEMENT (GST_PAD_PARENT (pad)), "src");

  /* since we're increasing timestamp/offsets, push this one first */

  GST_LOG (" passing buffer   [t=%" GST_TIME_FORMAT "-%" GST_TIME_FORMAT

      "], offset=%" G_GINT64_FORMAT ", offset_end=%" G_GINT64_FORMAT,

      GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)),

      GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf) + GST_BUFFER_DURATION (buf)),

      GST_BUFFER_OFFSET (buf), GST_BUFFER_OFFSET_END (buf));

  gst_buffer_ref (buf);

  ret = gst_pad_push (srcpad, buf);

  if (g_random_double () < injector_inject_probability) {

    GstBuffer *ibuf;

    ibuf = gst_buffer_copy (buf);

    if (GST_BUFFER_OFFSET_IS_VALID (buf) &&

        GST_BUFFER_OFFSET_END_IS_VALID (buf)) {

      guint64 delta;

      delta = GST_BUFFER_OFFSET_END (buf) - GST_BUFFER_OFFSET (buf);

      GST_BUFFER_OFFSET (ibuf) += delta / 4;

      GST_BUFFER_OFFSET_END (ibuf) += delta / 4;

    } else {

      GST_BUFFER_OFFSET (ibuf) = GST_BUFFER_OFFSET_NONE;

      GST_BUFFER_OFFSET_END (ibuf) = GST_BUFFER_OFFSET_NONE;

    }

    if (GST_BUFFER_TIMESTAMP_IS_VALID (buf) &&

        GST_BUFFER_DURATION_IS_VALID (buf)) {

      GstClockTime delta;

      delta = GST_BUFFER_DURATION (buf);

      GST_BUFFER_TIMESTAMP (ibuf) += delta / 4;

    } else {

      GST_BUFFER_TIMESTAMP (ibuf) = GST_CLOCK_TIME_NONE;

      GST_BUFFER_DURATION (ibuf) = GST_CLOCK_TIME_NONE;

    }

    if (GST_BUFFER_TIMESTAMP_IS_VALID (ibuf) ||

        GST_BUFFER_OFFSET_IS_VALID (ibuf)) {

      GST_LOG ("injecting buffer [t=%" GST_TIME_FORMAT "-%" GST_TIME_FORMAT

          "], offset=%" G_GINT64_FORMAT ", offset_end=%" G_GINT64_FORMAT,

          GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (ibuf)),

          GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (ibuf) +

              GST_BUFFER_DURATION (ibuf)), GST_BUFFER_OFFSET (ibuf),

          GST_BUFFER_OFFSET_END (ibuf));

      if (gst_pad_push (srcpad, ibuf) != GST_FLOW_OK) {

        /* ignore return value */

      }

    } else {

      GST_WARNING ("couldn't inject buffer, no incoming timestamps or offsets");

      gst_buffer_unref (ibuf);

    }

  }

  gst_buffer_unref (buf);

  return ret;

}

static void

test_injector_init (TestInjector * injector, TestInjectorClass * klass)

{

  GstPad *pad;

  pad = gst_pad_new_from_static_template (&sink_template, "sink");

  gst_pad_set_chain_function (pad, test_injector_chain);

  gst_pad_set_getcaps_function (pad, gst_pad_proxy_getcaps);

  gst_pad_set_setcaps_function (pad, gst_pad_proxy_setcaps);

  gst_element_add_pad (GST_ELEMENT (injector), pad);

  pad = gst_pad_new_from_static_template (&src_template, "src");

  gst_pad_set_getcaps_function (pad, gst_pad_proxy_getcaps);

  gst_pad_set_setcaps_function (pad, gst_pad_proxy_setcaps);

  gst_element_add_pad (GST_ELEMENT (injector), pad);

}

static gboolean

probe_cb (GstPad * pad, GstBuffer * buf, gdouble * drop_probability)

{

  if (g_random_double () < *drop_probability) {

    GST_LOG ("dropping buffer [t=%" GST_TIME_FORMAT "-%" GST_TIME_FORMAT "], "

        "offset=%" G_GINT64_FORMAT ", offset_end=%" G_GINT64_FORMAT,

        GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)),

        GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf) + GST_BUFFER_DURATION (buf)),

        GST_BUFFER_OFFSET (buf), GST_BUFFER_OFFSET_END (buf));

    return FALSE;               /* drop buffer */

  }

  return TRUE;                  /* don't drop buffer */

}

static void

got_buf (GstElement * fakesink, GstBuffer * buf, GstPad * pad, GList ** p_bufs)

{

  *p_bufs = g_list_append (*p_bufs, gst_buffer_ref (buf));

}

static void

do_perfect_stream_test (guint rate, guint width, gdouble drop_probability,

    gdouble inject_probability)

{

  GstElement *pipe, *src, *conv, *filter, *injector, *audiorate, *sink;

  GstMessage *msg;

  GstCaps *caps;

  GstPad *srcpad;

  GList *l, *bufs = NULL;

  GstClockTime next_time = GST_CLOCK_TIME_NONE;

  guint64 next_offset = GST_BUFFER_OFFSET_NONE;

  caps = gst_caps_new_simple ("audio/x-raw-int", "rate", G_TYPE_INT,

      rate, "width", G_TYPE_INT, width, NULL);

  GST_INFO ("-------- drop=%.0f%% caps = %" GST_PTR_FORMAT " ---------- ",

      drop_probability * 100.0, caps);

  g_assert (drop_probability >= 0.0 && drop_probability <= 1.0);

  g_assert (inject_probability >= 0.0 && inject_probability <= 1.0);

  g_assert (width > 0 && (width % 8) == 0);

  pipe = gst_pipeline_new ("pipeline");

  fail_unless (pipe != NULL);

  src = gst_element_factory_make ("audiotestsrc", "audiotestsrc");

  fail_unless (src != NULL);

  g_object_set (src, "num-buffers", 100, NULL);

  conv = gst_element_factory_make ("audioconvert", "audioconvert");

  fail_unless (conv != NULL);

  filter = gst_element_factory_make ("capsfilter", "capsfilter");

   fail_unless (filter != NULL);

   g_object_set (filter, "caps", caps, NULL);

  injector_inject_probability = inject_probability;

  injector = GST_ELEMENT (g_object_new (test_injector_get_type (), NULL));

  srcpad = gst_element_get_pad (injector, "src");

  fail_unless (srcpad != NULL);

   gst_pad_add_buffer_probe (srcpad, G_CALLBACK (probe_cb), &drop_probability);

  gst_object_unref (srcpad);

         audiorate = gst_element_factory_make ("audiorate", "audiorate");

         fail_unless (audiorate != NULL);

   sink = gst_element_factory_make ("fakesink", "fakesink");

  fail_unless (sink != NULL);

   g_object_set (sink, "signal-handoffs", TRUE, NULL);

   g_signal_connect (sink, "handoff", G_CALLBACK (got_buf), &bufs);

  gst_bin_add_many (GST_BIN (pipe), src, conv, filter, injector, audiorate,

      sink, NULL);

  gst_element_link_many (src, conv, filter, injector, audiorate, sink, NULL);

  fail_unless_equals_int (gst_element_set_state (pipe, GST_STATE_PLAYING),

      GST_STATE_CHANGE_ASYNC);

  fail_unless_equals_int (gst_element_get_state (pipe, NULL, NULL, -1),

      GST_STATE_CHANGE_SUCCESS);

  msg = gst_bus_poll (GST_ELEMENT_BUS (pipe),

      GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);

  fail_unless_equals_string (GST_MESSAGE_TYPE_NAME (msg), "eos");

  for (l = bufs; l != NULL; l = l->next) {

    GstBuffer *buf = GST_BUFFER (l->data);

    guint num_samples;

    fail_unless (GST_BUFFER_TIMESTAMP_IS_VALID (buf));

    fail_unless (GST_BUFFER_DURATION_IS_VALID (buf));

    fail_unless (GST_BUFFER_OFFSET_IS_VALID (buf));

    fail_unless (GST_BUFFER_OFFSET_END_IS_VALID (buf));

    GST_LOG ("buffer: ts=%" GST_TIME_FORMAT ", end_ts=%" GST_TIME_FORMAT

        " off=%" G_GINT64_FORMAT ", end_off=%" G_GINT64_FORMAT,

        GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)),

        GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf) + GST_BUFFER_DURATION (buf)),

        GST_BUFFER_OFFSET (buf), GST_BUFFER_OFFSET_END (buf));

    if (GST_CLOCK_TIME_IS_VALID (next_time)) {

      fail_unless_equals_uint64 (next_time, GST_BUFFER_TIMESTAMP (buf));

    }

    if (next_offset != GST_BUFFER_OFFSET_NONE) {

      fail_unless_equals_uint64 (next_offset, GST_BUFFER_OFFSET (buf));

    }

    /* check buffer size for sanity */

    fail_unless_equals_int (GST_BUFFER_SIZE (buf) % (width / 8), 0);

    /* check there is actually as much data as there should be */

    num_samples = GST_BUFFER_OFFSET_END (buf) - GST_BUFFER_OFFSET (buf);

    fail_unless_equals_int (GST_BUFFER_SIZE (buf), num_samples * (width / 8));

    next_time = GST_BUFFER_TIMESTAMP (buf) + GST_BUFFER_DURATION (buf);

    next_offset = GST_BUFFER_OFFSET_END (buf);

  }

  gst_message_unref (msg);

  gst_element_set_state (pipe, GST_STATE_NULL);

  gst_object_unref (pipe);

  g_list_foreach (bufs, (GFunc) gst_mini_object_unref, NULL);

  g_list_free (bufs);

  gst_caps_unref (caps);

}

static const guint rates[] = { 8000, 11025, 16000, 22050, 32000, 44100,

  48000, 3333, 33333, 66666, 9999

};

void test_perfect_stream_drop0()

{

  guint i;

  //xmlfile = "test_perfect_stream_drop0";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_drop0");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.0, 0.0);

    do_perfect_stream_test (rates[i], 16, 0.0, 0.0);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_drop10()

{

  guint i;

  //xmlfile = "test_perfect_stream_drop10";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_drop10");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.10, 0.0);

    do_perfect_stream_test (rates[i], 16, 0.10, 0.0);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_drop50()

{

  guint i;

  //xmlfile = "test_perfect_stream_drop50";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_drop50");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.50, 0.0);

    do_perfect_stream_test (rates[i], 16, 0.50, 0.0);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_drop90()

{

  guint i;

  //xmlfile = "test_perfect_stream_drop90";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_drop90");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.90, 0.0);

    do_perfect_stream_test (rates[i], 16, 0.90, 0.0);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_inject10()

{

  guint i;

  //xmlfile = "test_perfect_stream_inject10";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_inject10");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.0, 0.10);

    do_perfect_stream_test (rates[i], 16, 0.0, 0.10);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_inject90()

{

  guint i;

  //xmlfile = "test_perfect_stream_inject90";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_inject90");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.0, 0.90);

    do_perfect_stream_test (rates[i], 16, 0.0, 0.90);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

void test_perfect_stream_drop45_inject25()

{

  guint i;

  //xmlfile = "test_perfect_stream_drop45_inject25";

      std_log(LOG_FILENAME_LINE, "Test Started test_perfect_stream_drop45_inject25");

  for (i = 0; i < G_N_ELEMENTS (rates); ++i) {

    do_perfect_stream_test (rates[i], 8, 0.45, 0.25);

    do_perfect_stream_test (rates[i], 16, 0.45, 0.25);

  }

  std_log(LOG_FILENAME_LINE, "Test Successful");

      create_xml(0);

}

/* TODO: also do all tests with channels=1 and channels=2 */

static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,