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., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, 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
 100   srcadapter = gst_adapter_new ();
 101   fail_unless (srcadapter != NULL);
 102   sinkadapter = gst_adapter_new ();
 103   fail_unless (sinkadapter != NULL);
 104
 105   pipeline = gst_pipeline_new ("pipeline");
 106   fail_unless (pipeline != NULL);
 107
 108   audiotestsrc = gst_element_factory_make ("audiotestsrc", "src");
 109   fail_unless (audiotestsrc != NULL);
 110   g_object_set (G_OBJECT (audiotestsrc), "wave", 0, "freq", 440.0,
 111       "num-buffers", 2000, NULL);
 112
 113   identity1 = gst_element_factory_make ("identity", "identity1");
 114   fail_unless (identity1 != NULL);
 115   g_object_set (G_OBJECT (identity1), "signal-handoffs", TRUE, NULL);
 116   g_signal_connect (G_OBJECT (identity1), "handoff",
 117       G_CALLBACK (identity_handoff), srcadapter);
 118
 119   wavpackenc = gst_element_factory_make ("wavpackenc", "enc");
 120   fail_unless (wavpackenc != NULL);
 121
 122   identity2 = gst_element_factory_make ("identity", "identity2");
 123   fail_unless (identity2 != NULL);
 124   g_object_set (G_OBJECT (identity2), "check-imperfect-timestamp", TRUE,
 125       "check-imperfect-offset", TRUE, NULL);
 126
 127   wavpackdec = gst_element_factory_make ("wavpackdec", "dec");
 128   fail_unless (wavpackdec != NULL);
 129
 130   identity3 = gst_element_factory_make ("identity", "identity3");
 131   fail_unless (identity3 != NULL);
 132   g_object_set (G_OBJECT (identity3), "check-imperfect-timestamp", TRUE,
 133       "check-imperfect-offset", TRUE, NULL);
 134
 135   fakesink = gst_element_factory_make ("fakesink", "sink");
 136   fail_unless (fakesink != NULL);
 137   g_object_set (G_OBJECT (fakesink), "signal-handoffs", TRUE, NULL);
 138   g_signal_connect (G_OBJECT (fakesink), "handoff",
 139       G_CALLBACK (fakesink_handoff), sinkadapter);
 140
 141   gst_bin_add_many (GST_BIN (pipeline), audiotestsrc, identity1, wavpackenc,
 142       identity2, wavpackdec, identity3, fakesink, NULL);
 143
 144   fail_unless (gst_element_link_many (audiotestsrc, identity1, wavpackenc,
 145           identity2, wavpackdec, identity3, fakesink, NULL));
 146
 147   loop = g_main_loop_new (NULL, TRUE);
 148   fail_unless (loop != NULL);
 149
 150   bus = gst_element_get_bus (pipeline);
 151   fail_unless (bus != NULL);
 152   bus_watch = gst_bus_add_watch (bus, bus_handler, loop);
 153   gst_object_unref (bus);
 154
 155   had_first_buffer = FALSE;
 156   gst_element_set_state (pipeline, GST_STATE_PLAYING);
 157   g_main_loop_run (loop);
 158   gst_element_set_state (pipeline, GST_STATE_NULL);
 159
 160   fail_unless (had_first_buffer == TRUE);
 161   fail_unless (gst_adapter_available (srcadapter) > 0);
 162   fail_unless (gst_adapter_available (sinkadapter) > 0);
 163   fail_unless_equals_int (gst_adapter_available (srcadapter),
 164       gst_adapter_available (sinkadapter));
 165
 166   in = gst_adapter_take_buffer (srcadapter, gst_adapter_available (srcadapter));
 167   fail_unless (in != NULL);
 168   out =
 169       gst_adapter_take_buffer (sinkadapter,
 170       gst_adapter_available (sinkadapter));
 171   fail_unless (out != NULL);
 172
 173   fail_unless_equals_int (GST_BUFFER_SIZE (in), GST_BUFFER_SIZE (out));
 174   fail_unless (memcmp (GST_BUFFER_DATA (in), GST_BUFFER_DATA (out),
 175           GST_BUFFER_SIZE (in)) == 0);
 176
 177   gst_buffer_unref (in);
 178   gst_buffer_unref (out);
 179   g_object_unref (pipeline);
 180   g_main_loop_unref (loop);
 181   g_object_unref (srcadapter);
 182   g_object_unref (sinkadapter);
 183   g_source_remove (bus_watch);
 184 }
 185
 186 GST_END_TEST;
 187
 188 static Suite *
 189 wavpack_suite (void)
 190 {
 191   Suite *s = suite_create ("Wavpack");
 192   TCase *tc_chain = tcase_create ("linear");
 193
 194   /* time out after 60s, not the default 3 */
 195   tcase_set_timeout (tc_chain, 60);
 196
 197   suite_add_tcase (s, tc_chain);
 198   tcase_add_test (tc_chain, test_encode_decode);
 199
 200   return s;
 201 }
 202
 203 int
 204 main (int argc, char **argv)
 205 {
 206   int nf;
 207   Suite *s = wavpack_suite ();
 208   SRunner *sr = srunner_create (s);
 209
 210   gst_check_init (&argc, &argv);
 211
 212   srunner_run_all (sr, CK_NORMAL);
 213   nf = srunner_ntests_failed (sr);
 214   srunner_free (sr);
 215
 216   return nf;
 217 }