tests/check/pipelines/wavpack.c

   1 /* GStreamer
   2  * Copyright (C) 2008 Sebastian Dröge <slomo@circular-chaos.org>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19
  20 #include <gst/check/gstcheck.h>
  21 #include <gst/base/gstadapter.h>
  22
  23 static gboolean
  24 bus_handler (GstBus * bus, GstMessage * message, gpointer data)
  25 {
  26   GMainLoop *loop = (GMainLoop *) data;
  27
  28   switch (message->type) {
  29     case GST_MESSAGE_EOS:
  30       g_main_loop_quit (loop);
  31       break;
  32     case GST_MESSAGE_WARNING:
  33     case GST_MESSAGE_ERROR:{
  34       GError *gerror;
  35       gchar *debug;
  36
  37       gst_message_parse_error (message, &gerror, &debug);
  38       gst_object_default_error (GST_MESSAGE_SRC (message), gerror, debug);
  39       gst_message_unref (message);
  40       g_error_free (gerror);
  41       g_free (debug);
  42       g_main_loop_quit (loop);
  43       break;
  44     }
  45     case GST_MESSAGE_ELEMENT:
  46     {
  47       const GstStructure *s = gst_message_get_structure (message);
  48
  49       const gchar *name = gst_structure_get_name (s);
  50
  51       fail_unless (strcmp (name, "imperfect-timestamp") != 0);
  52       fail_unless (strcmp (name, "imperfect-offset") != 0);
  53       break;
  54     }
  55     default:
  56       break;
  57   }
  58
  59   return TRUE;
  60 }
  61
  62 static gboolean had_first_buffer = FALSE;
  63
  64 static void
  65 identity_handoff (GstElement * object, GstBuffer * buffer, gpointer user_data)
  66 {
  67   GstAdapter *adapter = GST_ADAPTER (user_data);
  68
  69   gst_adapter_push (adapter, gst_buffer_ref (buffer));
  70 }
  71
  72 static void
  73 fakesink_handoff (GstElement * object, GstBuffer * buffer, GstPad * pad,
  74     gpointer user_data)
  75 {
  76   GstAdapter *adapter = GST_ADAPTER (user_data);
  77
  78   /* Don't allow the second buffer with offset=0 as it's the decoded
  79    * rewrite of the first
  80    */
  81   if (had_first_buffer == FALSE && GST_BUFFER_OFFSET (buffer) == 0)
  82     had_first_buffer = TRUE;
  83   else if (GST_BUFFER_OFFSET (buffer) == 0)
  84     return;
  85
  86   gst_adapter_push (adapter, gst_buffer_ref (buffer));
  87 }
  88
  89 GST_START_TEST (test_encode_decode)
  90 {
  91   GstElement *pipeline;
  92   GstElement *audiotestsrc, *identity1, *wavpackenc, *identity2, *wavpackdec,
  93       *identity3, *fakesink;
  94   GstAdapter *srcadapter, *sinkadapter;
  95   GstBus *bus;
  96   GMainLoop *loop;
  97   GstBuffer *in, *out;
  98   guint bus_watch = 0;
  99   GstMapInfo map;
 100
 101   srcadapter = gst_adapter_new ();
 102   fail_unless (srcadapter != NULL);
 103   sinkadapter = gst_adapter_new ();
 104   fail_unless (sinkadapter != NULL);
 105
 106   pipeline = gst_pipeline_new ("pipeline");
 107   fail_unless (pipeline != NULL);
 108
 109   audiotestsrc = gst_element_factory_make ("audiotestsrc", "src");
 110   fail_unless (audiotestsrc != NULL);
 111   g_object_set (G_OBJECT (audiotestsrc), "wave", 0, "freq", 440.0,
 112       "num-buffers", 200, NULL);
 113
 114   identity1 = gst_element_factory_make ("identity", "identity1");
 115   fail_unless (identity1 != NULL);
 116   g_object_set (G_OBJECT (identity1), "signal-handoffs", TRUE, NULL);
 117   g_signal_connect (G_OBJECT (identity1), "handoff",
 118       G_CALLBACK (identity_handoff), srcadapter);
 119
 120   wavpackenc = gst_element_factory_make ("wavpackenc", "enc");
 121   fail_unless (wavpackenc != NULL);
 122
 123   identity2 = gst_element_factory_make ("identity", "identity2");
 124   fail_unless (identity2 != NULL);
 125   g_object_set (G_OBJECT (identity2), "check-imperfect-timestamp", TRUE,
 126       "check-imperfect-offset", TRUE, NULL);
 127
 128   wavpackdec = gst_element_factory_make ("wavpackdec", "dec");
 129   fail_unless (wavpackdec != NULL);
 130
 131   identity3 = gst_element_factory_make ("identity", "identity3");
 132   fail_unless (identity3 != NULL);
 133   g_object_set (G_OBJECT (identity3), "check-imperfect-timestamp", TRUE,
 134       "check-imperfect-offset", TRUE, NULL);
 135
 136   fakesink = gst_element_factory_make ("fakesink", "sink");
 137   fail_unless (fakesink != NULL);
 138   g_object_set (G_OBJECT (fakesink), "signal-handoffs", TRUE, NULL);
 139   g_signal_connect (G_OBJECT (fakesink), "handoff",
 140       G_CALLBACK (fakesink_handoff), sinkadapter);
 141
 142   gst_bin_add_many (GST_BIN (pipeline), audiotestsrc, identity1, wavpackenc,
 143       identity2, wavpackdec, identity3, fakesink, NULL);
 144
 145   fail_unless (gst_element_link_many (audiotestsrc, identity1, wavpackenc,
 146           identity2, wavpackdec, identity3, fakesink, NULL));
 147
 148   loop = g_main_loop_new (NULL, TRUE);
 149   fail_unless (loop != NULL);
 150
 151   bus = gst_element_get_bus (pipeline);
 152   fail_unless (bus != NULL);
 153   bus_watch = gst_bus_add_watch (bus, bus_handler, loop);
 154   gst_object_unref (bus);
 155
 156   had_first_buffer = FALSE;
 157   gst_element_set_state (pipeline, GST_STATE_PLAYING);
 158   g_main_loop_run (loop);
 159   gst_element_set_state (pipeline, GST_STATE_NULL);
 160
 161   fail_unless (had_first_buffer == TRUE);
 162   fail_unless (gst_adapter_available (srcadapter) > 0);
 163   fail_unless (gst_adapter_available (sinkadapter) > 0);
 164   fail_unless_equals_int (gst_adapter_available (srcadapter),
 165       gst_adapter_available (sinkadapter));
 166
 167   in = gst_adapter_take_buffer (srcadapter, gst_adapter_available (srcadapter));
 168   fail_unless (in != NULL);
 169   out =
 170       gst_adapter_take_buffer (sinkadapter,
 171       gst_adapter_available (sinkadapter));
 172   fail_unless (out != NULL);
 173
 174   fail_unless_equals_int (gst_buffer_get_size (in), gst_buffer_get_size (out));
 175   gst_buffer_map (out, &map, GST_MAP_READ);
 176   fail_unless (gst_buffer_memcmp (in, 0, map.data, map.size) == 0);
 177   gst_buffer_unmap (out, &map);
 178
 179   gst_buffer_unref (in);
 180   gst_buffer_unref (out);
 181   g_object_unref (pipeline);
 182   g_main_loop_unref (loop);
 183   g_object_unref (srcadapter);
 184   g_object_unref (sinkadapter);
 185   g_source_remove (bus_watch);
 186 }
 187
 188 GST_END_TEST;
 189
 190 static Suite *
 191 wavpack_suite (void)
 192 {
 193   Suite *s = suite_create ("Wavpack");
 194   TCase *tc_chain = tcase_create ("linear");
 195
 196   /* time out after 60s, not the default 3 */
 197   tcase_set_timeout (tc_chain, 60);
 198
 199   suite_add_tcase (s, tc_chain);
 200   tcase_add_test (tc_chain, test_encode_decode);
 201
 202   return s;
 203 }
 204
 205 GST_CHECK_MAIN (wavpack);