/* GStreamer * * unit test for opus * * Copyright (C) <2011> Vincent Penquerc'h * * 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., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301, USA. */ #include #include #if G_BYTE_ORDER == G_BIG_ENDIAN #define AFORMAT "S16BE" #else #define AFORMAT "S16LE" #endif #define AUDIO_CAPS_STRING "audio/x-raw, " \ "format = (string) " AFORMAT ", "\ "layout = (string) interleaved, " \ "rate = (int) 48000, " \ "channels = (int) 1 " static const guint8 opus_ogg_id_header[19] = { 0x4f, 0x70, 0x75, 0x73, 0x48, 0x65, 0x61, 0x64, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }; static const guint8 opus_ogg_comments_header[] = { 0x4f, 0x70, 0x75, 0x73, 0x54, 0x61, 0x67, 0x73, 0x1e, 0x00, 0x00, 0x00, 0x45, 0x6e, 0x63, 0x6f, 0x64, 0x65, 0x64, 0x20, 0x77, 0x69, 0x74, 0x68, 0x20, 0x47, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x65, 0x72, 0x20, 0x4f, 0x70, 0x75, 0x73, 0x65, 0x6e, 0x63, 0x00, 0x00, 0x00, 0x00 }; /* A lot of these taken from the vorbisdec test */ /* 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 *mydecsrcpad, *mydecsinkpad; static GstPad *myencsrcpad, *myencsinkpad; static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_CAPS_ANY); static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_CAPS_ANY); static GstElement * setup_opusdec (void) { GstElement *opusdec; GST_DEBUG ("setup_opusdec"); opusdec = gst_check_setup_element ("opusdec"); mydecsrcpad = gst_check_setup_src_pad (opusdec, &srctemplate); mydecsinkpad = gst_check_setup_sink_pad (opusdec, &sinktemplate); gst_pad_set_active (mydecsrcpad, TRUE); gst_pad_set_active (mydecsinkpad, TRUE); return opusdec; } static void cleanup_opusdec (GstElement * opusdec) { GST_DEBUG ("cleanup_opusdec"); gst_element_set_state (opusdec, GST_STATE_NULL); gst_pad_set_active (mydecsrcpad, FALSE); gst_pad_set_active (mydecsinkpad, FALSE); gst_check_teardown_src_pad (opusdec); gst_check_teardown_sink_pad (opusdec); gst_check_teardown_element (opusdec); } static GstElement * setup_opusenc (void) { GstElement *opusenc; GST_DEBUG ("setup_opusenc"); opusenc = gst_check_setup_element ("opusenc"); myencsrcpad = gst_check_setup_src_pad (opusenc, &srctemplate); myencsinkpad = gst_check_setup_sink_pad (opusenc, &sinktemplate); gst_pad_set_active (myencsrcpad, TRUE); gst_pad_set_active (myencsinkpad, TRUE); return opusenc; } static void cleanup_opusenc (GstElement * opusenc) { GST_DEBUG ("cleanup_opusenc"); gst_element_set_state (opusenc, GST_STATE_NULL); gst_pad_set_active (myencsrcpad, FALSE); gst_pad_set_active (myencsinkpad, FALSE); gst_check_teardown_src_pad (opusenc); gst_check_teardown_sink_pad (opusenc); gst_check_teardown_element (opusenc); } static void check_buffers (guint expected) { GstBuffer *outbuffer; guint i, num_buffers; /* check buffers are the type we expect */ num_buffers = g_list_length (buffers); fail_unless (num_buffers >= expected); for (i = 0; i < num_buffers; ++i) { outbuffer = GST_BUFFER (buffers->data); fail_if (outbuffer == NULL); fail_if (gst_buffer_get_size (outbuffer) == 0); buffers = g_list_remove (buffers, outbuffer); ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1); gst_buffer_unref (outbuffer); outbuffer = NULL; } } GST_START_TEST (test_opus_id_header) { GstElement *opusdec; GstBuffer *inbuffer; opusdec = setup_opusdec (); fail_unless (gst_element_set_state (opusdec, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); inbuffer = gst_buffer_new_and_alloc (sizeof (opus_ogg_id_header)); gst_buffer_fill (inbuffer, 0, opus_ogg_id_header, sizeof (opus_ogg_id_header)); ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); gst_buffer_ref (inbuffer); /* pushing gives away my reference ... */ fail_unless (gst_pad_push (mydecsrcpad, inbuffer) == GST_FLOW_OK); /* ... and nothing ends up on the global buffer list */ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); gst_buffer_unref (inbuffer); check_buffers (0); /* cleanup */ cleanup_opusdec (opusdec); } GST_END_TEST; GST_START_TEST (test_opus_encode_nothing) { GstElement *opusenc; opusenc = setup_opusenc (); fail_unless (gst_element_set_state (opusenc, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE); fail_unless (gst_element_set_state (opusenc, GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS, "could not set to ready"); /* cleanup */ cleanup_opusenc (opusenc); } GST_END_TEST; GST_START_TEST (test_opus_decode_nothing) { GstElement *opusdec; opusdec = setup_opusdec (); fail_unless (gst_element_set_state (opusdec, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); fail_unless (gst_pad_push_event (mydecsrcpad, gst_event_new_eos ()) == TRUE); fail_unless (gst_element_set_state (opusdec, GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS, "could not set to ready"); /* cleanup */ cleanup_opusdec (opusdec); } GST_END_TEST; GST_START_TEST (test_opus_encode_samples) { const unsigned int nsamples = 4096; GstElement *opusenc; GstBuffer *inbuffer; GstCaps *caps; opusenc = setup_opusenc (); fail_unless (gst_element_set_state (opusenc, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); inbuffer = gst_buffer_new_and_alloc (nsamples * 2); gst_buffer_memset (inbuffer, 0, 0, nsamples * 2); GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) = 0; GST_BUFFER_DURATION (inbuffer) = GST_CLOCK_TIME_NONE; ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); caps = gst_caps_from_string (AUDIO_CAPS_STRING); fail_unless (caps != NULL); gst_pad_set_caps (myencsrcpad, caps); gst_caps_unref (caps); gst_buffer_ref (inbuffer); /* pushing gives away my reference ... */ fail_unless (gst_pad_push (myencsrcpad, inbuffer) == GST_FLOW_OK); /* ... and nothing ends up on the global buffer list */ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE); ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); gst_buffer_unref (inbuffer); fail_unless (gst_element_set_state (opusenc, GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS, "could not set to ready"); /* default frame size is 20 ms, at 48000 Hz that's 960 samples */ check_buffers ((nsamples + 959) / 960); /* cleanup */ cleanup_opusenc (opusenc); g_list_free (buffers); } GST_END_TEST; GST_START_TEST (test_opus_encode_properties) { const unsigned int nsamples = 4096; enum { steps = 20 }; GstElement *opusenc; GstBuffer *inbuffer; GstCaps *caps; unsigned int step; static const struct { const char *param; int value; } param_changes[steps] = { { "frame-size", 40}, { "inband-fec", 1}, { "complexity", 5}, { "bandwidth", 1104}, { "frame-size", 2}, { "max-payload-size", 80}, { "frame-size", 60}, { "max-payload-size", 900}, { "complexity", 1}, { "bitrate", 30000}, { "frame-size", 10}, { "bitrate", 300000}, { "inband-fec", 0}, { "frame-size", 5}, { "bandwidth", 1101}, { "frame-size", 10}, { "bitrate", 500000}, { "frame-size", 5}, { "bitrate", 80000}, { "complexity", 8},}; opusenc = setup_opusenc (); fail_unless (gst_element_set_state (opusenc, GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, "could not set to playing"); caps = gst_caps_from_string (AUDIO_CAPS_STRING); fail_unless (caps != NULL); gst_pad_set_caps (myencsrcpad, caps); for (step = 0; step < steps; ++step) { inbuffer = gst_buffer_new_and_alloc (nsamples * 2); gst_buffer_memset (inbuffer, 0, 0, nsamples * 2); GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) = 0; GST_BUFFER_DURATION (inbuffer) = GST_CLOCK_TIME_NONE; ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); gst_buffer_ref (inbuffer); /* pushing gives away my reference ... */ fail_unless (gst_pad_push (myencsrcpad, inbuffer) == GST_FLOW_OK); /* ... and nothing ends up on the global buffer list */ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE); ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1); gst_buffer_unref (inbuffer); /* change random parameters */ g_object_set (opusenc, param_changes[step].param, param_changes[step].value, NULL); check_buffers (1); fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_flush_start ()) == TRUE); fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_flush_stop (TRUE)) == TRUE); } gst_caps_unref (caps); fail_unless (gst_element_set_state (opusenc, GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS, "could not set to ready"); /* cleanup */ cleanup_opusenc (opusenc); g_list_free (buffers); } GST_END_TEST; static Suite * opus_suite (void) { Suite *s = suite_create ("opus"); TCase *tc_chain = tcase_create ("general"); suite_add_tcase (s, tc_chain); #define X if (0) tcase_add_test (tc_chain, test_opus_id_header); tcase_add_test (tc_chain, test_opus_encode_nothing); tcase_add_test (tc_chain, test_opus_decode_nothing); tcase_add_test (tc_chain, test_opus_encode_samples); tcase_add_test (tc_chain, test_opus_encode_properties); #undef X return s; } int main (int argc, char **argv) { int nf; Suite *s = opus_suite (); SRunner *sr = srunner_create (s); gst_check_init (&argc, &argv); srunner_run_all (sr, CK_NORMAL); nf = srunner_ntests_failed (sr); srunner_free (sr); return nf; }