/* GStreamer

 *

 * unit test for volume

 *

 * Copyright (C) <2005> Thomas Vander Stichele <thomas at apestaart dot 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.

 */

#include <unistd.h>

#include <gst/base/gstbasetransform.h>

#include <gst/check/gstcheck.h>

#include <gst/controller/gstcontroller.h>

/* For ease of programming we use globals to keep refs for our floating

 * src and sink pads we create; otherwise we always have to do get_pad,

 * get_peer, and then remove references in every test function */

static GstPad *mysrcpad, *mysinkpad;

#define VOLUME_CAPS_TEMPLATE_STRING     \

    "audio/x-raw-int, "                 \

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

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

    "endianness = (int) BYTE_ORDER, "   \

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

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

    "signed = (bool) TRUE; "            \

    "audio/x-raw-float, "               \

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

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

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) { 32, 64 }"          \

#define VOLUME_CAPS_STRING_S8           \

    "audio/x-raw-int, "                 \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 8, "                 \

    "depth = (int) 8, "                 \

    "signed = (bool) TRUE"

#define VOLUME_CAPS_STRING_S16          \

    "audio/x-raw-int, "                 \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 16, "                \

    "depth = (int) 16, "                \

    "signed = (bool) TRUE"

#define VOLUME_CAPS_STRING_S24          \

    "audio/x-raw-int, "                 \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 24, "                \

    "depth = (int) 24, "                \

    "signed = (bool) TRUE"

#define VOLUME_CAPS_STRING_S32          \

    "audio/x-raw-int, "                 \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 32, "                \

    "depth = (int) 32, "                \

    "signed = (bool) TRUE"

#define VOLUME_CAPS_STRING_F32          \

    "audio/x-raw-float, "               \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 32"

#define VOLUME_CAPS_STRING_F64          \

    "audio/x-raw-float, "               \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 64"

#define VOLUME_WRONG_CAPS_STRING        \

    "audio/x-raw-int, "                 \

    "channels = (int) 1, "              \

    "rate = (int) 44100, "              \

    "endianness = (int) BYTE_ORDER, "   \

    "width = (int) 16, "                \

    "depth = (int) 16, "                \

    "signed = (bool) FALSE"

static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",

    GST_PAD_SINK,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS (VOLUME_CAPS_TEMPLATE_STRING)

    );

static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS (VOLUME_CAPS_TEMPLATE_STRING)

    );

static GstElement *

setup_volume (void)

{

  GstElement *volume;

  GST_DEBUG ("setup_volume");

  volume = gst_check_setup_element ("volume");

  mysrcpad = gst_check_setup_src_pad (volume, &srctemplate, NULL);

  mysinkpad = gst_check_setup_sink_pad (volume, &sinktemplate, NULL);

  gst_pad_set_active (mysrcpad, TRUE);

  gst_pad_set_active (mysinkpad, TRUE);

  return volume;

}

static void

cleanup_volume (GstElement * volume)

{

  GST_DEBUG ("cleanup_volume");

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

  g_list_free (buffers);

  buffers = NULL;

  gst_pad_set_active (mysrcpad, FALSE);

  gst_pad_set_active (mysinkpad, FALSE);

  gst_check_teardown_src_pad (volume);

  gst_check_teardown_sink_pad (volume);

  gst_check_teardown_element (volume);

}

GST_START_TEST (test_unity_s8)

{

  GstElement *volume;

  GstBuffer *inbuffer, *outbuffer;

  GstCaps *caps;

  gint8 in[2] = { 64, -16 };

  gint8 *res;

  volume = setup_volume ();

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (2);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 2);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S8);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint8 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", in[0], in[1], res[0], res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 2) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_half_s8)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint8 in[2] = { 64, -16 };

  gint8 out[2] = { 32, -8 };

  gint8 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "volume", 0.5, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (2);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 2);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 2) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S8);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint8 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 2) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_double_s8)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint8 in[2] = { 64, -16 };

  gint8 out[2] = { 127, -32 };  /* notice the clamped sample */

  gint8 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "volume", 2.0, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (2);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 2);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 2) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S8);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint8 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 2) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_mute_s8)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint8 in[2] = { 64, -16 };

  gint8 out[2] = { 0, 0 };

  gint8 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "mute", TRUE, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (2);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 2);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 2) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S8);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint8 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 2) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_unity_s16)

{

  GstElement *volume;

  GstBuffer *inbuffer, *outbuffer;

  GstCaps *caps;

  gint16 in[2] = { 16384, -256 };

  gint16 *res;

  volume = setup_volume ();

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (4);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 4);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S16);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint16 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", in[0], in[1], res[0], res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 4) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_half_s16)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint16 in[2] = { 16384, -256 };

  gint16 out[2] = { 8192, -128 };

  gint16 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "volume", 0.5, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (4);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 4);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 4) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S16);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint16 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 4) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_double_s16)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint16 in[2] = { 16384, -256 };

  gint16 out[2] = { 32767, -512 };      /* notice the clamped sample */

  gint16 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "volume", 2.0, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (4);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 4);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 4) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S16);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint16 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 4) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

GST_START_TEST (test_mute_s16)

{

  GstElement *volume;

  GstBuffer *inbuffer;

  GstBuffer *outbuffer;

  GstCaps *caps;

  gint16 in[2] = { 16384, -256 };

  gint16 out[2] = { 0, 0 };

  gint16 *res;

  volume = setup_volume ();

  g_object_set (G_OBJECT (volume), "mute", TRUE, NULL);

  fail_unless (gst_element_set_state (volume,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,

      "could not set to playing");

  inbuffer = gst_buffer_new_and_alloc (4);

  memcpy (GST_BUFFER_DATA (inbuffer), in, 4);

  fail_unless (memcmp (GST_BUFFER_DATA (inbuffer), in, 4) == 0);

  caps = gst_caps_from_string (VOLUME_CAPS_STRING_S16);

  gst_buffer_set_caps (inbuffer, caps);

  gst_caps_unref (caps);

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  /* FIXME: reffing the inbuffer should make the transformation not be

   * inplace

   gst_buffer_ref (inbuffer);

   */

  /* pushing gives away my reference ... */

  fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK);

  /* ... but it ends up being modified inplace and

   * collected on the global buffer list */

  ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);

  fail_unless_equals_int (g_list_length (buffers), 1);

  fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);

  fail_unless (inbuffer == outbuffer);

  res = (gint16 *) GST_BUFFER_DATA (outbuffer);

  GST_INFO ("expected %+5d %+5d  real %+5d %+5d", out[0], out[1], res[0],

      res[1]);

  fail_unless (memcmp (GST_BUFFER_DATA (outbuffer), out, 4) == 0);

  /* cleanup */

  cleanup_volume (volume);

}

GST_END_TEST;

#if (G_BYTE_ORDER == G_LITTLE_ENDIAN)

#define get_unaligned_i24(_x) ( (((guint8*)_x)[0]) | ((((guint8*)_x)[1]) << 8) | ((((gint8*)_x)[2]) << 16) )

#define write_unaligned_u24(_x,samp) do { (((guint8*)_x)[0]) = samp & 0xFF; (((guint8*)_x)[1]) = (samp >> 8) & 0xFF; (((guint8*)_x)[2]) = (samp >> 16) & 0xFF; } while (0)

#else /* BIG ENDIAN */

#define get_unaligned_i24(_x) ( (((guint8*)_x)[2]) | ((((guint8*)_x)[1]) << 8) | ((((gint8*)_x)[0]) << 16) )

#define write_unaligned_u24(_x,samp) do { (((guint8*)_x)[0]) = (samp >> 16) & 0xFF; (((guint8*)_x)[1]) = (samp >> 8) & 0xFF; (((guint8*)_x)[2]) = samp & 0xFF; } while (0)

#endif

GST_START_TEST (test_unity_s24)

{

  GstElement *volume;

  GstBuffer *inbuffer, *outbuffer;

  GstCaps *caps;

  gint32 in_32[2] = { 4194304, -4096 };

  guint8 in[6];

  guint8 *res;

  gint32 res_32[2];

  write_unaligned_u24 (in, in_32[0]);

  write_unaligned_u24 (in + 3, in_32[1]);