| /* |
| * GStreamer |
| * |
| * unit test for (audio) parser |
| * |
| * Copyright (C) 2008 Nokia Corporation. All rights reserved. |
| * |
| * Contact: Stefan Kost <stefan.kost@nokia.com> |
| * |
| * 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 <gst/check/gstcheck.h> |
| #include "elements/parser.h" |
| |
| |
| /* context state variables */ |
| const gchar *ctx_factory; |
| GstStaticPadTemplate *ctx_sink_template; |
| GstStaticPadTemplate *ctx_src_template; |
| GstCaps *ctx_input_caps; |
| GstCaps *ctx_output_caps; |
| guint ctx_discard = 0; |
| datablob ctx_headers[MAX_HEADERS] = { {NULL, 0}, }; |
| |
| gboolean ctx_no_metadata = FALSE; |
| |
| /* helper variables */ |
| GList *current_buf = NULL; |
| |
| GstPad *srcpad, *sinkpad; |
| guint dataoffset = 0; |
| GstClockTime ts_counter = 0; |
| gint64 offset_counter = 0; |
| guint buffer_counter = 0; |
| |
| typedef struct |
| { |
| guint discard; |
| guint buffers_before_offset_skip; |
| guint offset_skip_amount; |
| const guint8 *data_to_verify; |
| guint data_to_verify_size; |
| GstCaps *caps; |
| gboolean no_metadata; |
| } buffer_verify_data_s; |
| |
| /* takes a copy of the passed buffer data */ |
| static GstBuffer * |
| buffer_new (const unsigned char *buffer_data, guint size) |
| { |
| GstBuffer *buffer; |
| |
| buffer = gst_buffer_new_and_alloc (size); |
| if (buffer_data) { |
| gst_buffer_fill (buffer, 0, buffer_data, size); |
| } else { |
| guint i; |
| GstMapInfo map; |
| |
| gst_buffer_map (buffer, &map, GST_MAP_WRITE); |
| /* Create a recognizable pattern (loop 0x00 -> 0xff) in the data block */ |
| for (i = 0; i < map.size; i++) { |
| map.data[i] = i % 0x100; |
| } |
| gst_buffer_unmap (buffer, &map); |
| } |
| |
| /* gst_buffer_set_caps (buffer, GST_PAD_CAPS (srcpad)); */ |
| GST_BUFFER_OFFSET (buffer) = dataoffset; |
| dataoffset += size; |
| return buffer; |
| } |
| |
| /* |
| * Adds buffer sizes together. |
| */ |
| static void |
| buffer_count_size (void *buffer, void *user_data) |
| { |
| guint *sum = (guint *) user_data; |
| *sum += gst_buffer_get_size (buffer); |
| } |
| |
| /* |
| * Verify that given buffer contains predefined ADTS frame. |
| */ |
| static void |
| buffer_verify_data (void *buffer, void *user_data) |
| { |
| buffer_verify_data_s *vdata; |
| |
| if (!user_data) { |
| return; |
| } |
| |
| vdata = (buffer_verify_data_s *) user_data; |
| |
| GST_DEBUG ("discard: %d", vdata->discard); |
| if (vdata->discard) { |
| buffer_counter++; |
| if (buffer_counter == vdata->discard) { |
| buffer_counter = 0; |
| vdata->discard = 0; |
| } |
| return; |
| } |
| |
| fail_unless (gst_buffer_get_size (buffer) == vdata->data_to_verify_size); |
| fail_unless (gst_buffer_memcmp (buffer, 0, vdata->data_to_verify, |
| vdata->data_to_verify_size) == 0); |
| |
| if (vdata->buffers_before_offset_skip) { |
| /* This is for skipping the garbage in some test cases */ |
| if (buffer_counter == vdata->buffers_before_offset_skip) { |
| offset_counter += vdata->offset_skip_amount; |
| } |
| } |
| if (!vdata->no_metadata) { |
| fail_unless (GST_BUFFER_TIMESTAMP (buffer) == ts_counter); |
| fail_unless (GST_BUFFER_DURATION (buffer) != 0); |
| fail_unless (GST_BUFFER_OFFSET (buffer) == offset_counter); |
| } |
| |
| ts_counter += GST_BUFFER_DURATION (buffer); |
| offset_counter += gst_buffer_get_size (buffer); |
| buffer_counter++; |
| } |
| |
| static GstElement * |
| setup_element (const gchar * factory, GstStaticPadTemplate * sink_template, |
| GstCaps * sink_caps, GstStaticPadTemplate * src_template, |
| GstCaps * src_caps) |
| { |
| GstElement *element; |
| GstBus *bus; |
| gchar *caps_str = NULL; |
| |
| element = gst_check_setup_element (factory); |
| srcpad = gst_check_setup_src_pad (element, src_template); |
| if (sink_caps) { |
| caps_str = gst_caps_to_string (sink_caps); |
| sink_template->static_caps.string = caps_str; |
| } |
| sinkpad = gst_check_setup_sink_pad (element, sink_template); |
| gst_pad_set_active (srcpad, TRUE); |
| gst_check_setup_events (srcpad, element, src_caps, GST_FORMAT_BYTES); |
| gst_pad_set_active (sinkpad, TRUE); |
| |
| bus = gst_bus_new (); |
| gst_element_set_bus (element, bus); |
| |
| fail_unless (gst_element_set_state (element, |
| GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE, |
| "could not set to playing"); |
| |
| ts_counter = offset_counter = buffer_counter = 0; |
| buffers = NULL; |
| g_free (caps_str); |
| return element; |
| } |
| |
| static void |
| cleanup_element (GstElement * element) |
| { |
| GstBus *bus; |
| |
| /* Free parsed buffers */ |
| gst_check_drop_buffers (); |
| |
| bus = GST_ELEMENT_BUS (element); |
| gst_bus_set_flushing (bus, TRUE); |
| gst_object_unref (bus); |
| |
| gst_pad_set_active (srcpad, FALSE); |
| gst_pad_set_active (sinkpad, FALSE); |
| gst_check_teardown_src_pad (element); |
| gst_check_teardown_sink_pad (element); |
| gst_check_teardown_element (element); |
| } |
| |
| /* inits a standard test */ |
| void |
| gst_parser_test_init (GstParserTest * ptest, guint8 * data, guint size, |
| guint num) |
| { |
| /* need these */ |
| fail_unless (ctx_factory != NULL); |
| fail_unless (ctx_src_template != NULL); |
| fail_unless (ctx_sink_template != NULL); |
| |
| /* basics */ |
| memset (ptest, 0, sizeof (*ptest)); |
| ptest->factory = ctx_factory; |
| ptest->sink_template = ctx_sink_template; |
| ptest->src_template = ctx_src_template; |
| ptest->framed = TRUE; |
| /* could be NULL if not relevant/needed */ |
| ptest->src_caps = ctx_input_caps; |
| ptest->sink_caps = ctx_output_caps; |
| memcpy (ptest->headers, ctx_headers, sizeof (ptest->headers)); |
| ptest->discard = ctx_discard; |
| /* some data that pleases caller */ |
| ptest->series[0].data = data; |
| ptest->series[0].size = size; |
| ptest->series[0].num = num; |
| ptest->series[0].fpb = 1; |
| ptest->series[1].fpb = 1; |
| ptest->series[2].fpb = 1; |
| ptest->no_metadata = ctx_no_metadata; |
| } |
| |
| /* |
| * Test if the parser pushes clean data properly. |
| */ |
| void |
| gst_parser_test_run (GstParserTest * test, GstCaps ** out_caps) |
| { |
| buffer_verify_data_s vdata = { 0, 0, 0, NULL, 0, NULL, FALSE }; |
| GstElement *element; |
| GstBuffer *buffer = NULL; |
| GstCaps *src_caps; |
| guint i, j, k; |
| guint frames = 0, size = 0; |
| |
| element = setup_element (test->factory, test->sink_template, NULL, |
| test->src_template, test->src_caps); |
| |
| /* push some setup headers */ |
| for (j = 0; j < G_N_ELEMENTS (test->headers) && test->headers[j].data; j++) { |
| buffer = buffer_new (test->headers[j].data, test->headers[j].size); |
| fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK); |
| } |
| |
| for (j = 0; j < 3; j++) { |
| for (i = 0; i < test->series[j].num; i++) { |
| /* sanity enforcing */ |
| for (k = 0; k < MAX (1, test->series[j].fpb); k++) { |
| if (!k) |
| buffer = buffer_new (test->series[j].data, test->series[j].size); |
| else { |
| buffer = gst_buffer_append (buffer, |
| buffer_new (test->series[j].data, test->series[j].size)); |
| } |
| } |
| fail_unless_equals_int (gst_pad_push (srcpad, buffer), GST_FLOW_OK); |
| if (j == 0) |
| vdata.buffers_before_offset_skip++; |
| else if (j == 1) |
| vdata.offset_skip_amount += test->series[j].size * test->series[j].fpb; |
| if (j != 1) { |
| frames += test->series[j].fpb; |
| size += test->series[j].size * test->series[j].fpb; |
| } |
| } |
| } |
| gst_pad_push_event (srcpad, gst_event_new_eos ()); |
| |
| if (G_LIKELY (test->framed)) |
| fail_unless_equals_int (g_list_length (buffers) - test->discard, frames); |
| |
| /* if all frames are identical, do extended test, |
| * otherwise only verify total data size */ |
| if (test->series[0].data && (!test->series[2].size || |
| (test->series[0].size == test->series[2].size && test->series[2].data |
| && !memcmp (test->series[0].data, test->series[2].data, |
| test->series[0].size)))) { |
| vdata.data_to_verify = test->series[0].data; |
| vdata.data_to_verify_size = test->series[0].size; |
| vdata.caps = test->sink_caps; |
| vdata.discard = test->discard; |
| vdata.no_metadata = test->no_metadata; |
| g_list_foreach (buffers, buffer_verify_data, &vdata); |
| } else { |
| guint datasum = 0; |
| |
| g_list_foreach (buffers, buffer_count_size, &datasum); |
| size -= test->dropped; |
| fail_unless_equals_int (datasum, size); |
| } |
| |
| src_caps = gst_pad_get_current_caps (sinkpad); |
| GST_LOG ("output caps: %" GST_PTR_FORMAT, src_caps); |
| |
| if (test->sink_caps) { |
| GST_LOG ("%" GST_PTR_FORMAT " = %" GST_PTR_FORMAT " ?", src_caps, |
| test->sink_caps); |
| fail_unless (gst_caps_is_equal (src_caps, test->sink_caps)); |
| } |
| |
| if (out_caps) |
| *out_caps = src_caps; |
| else |
| gst_caps_unref (src_caps); |
| |
| cleanup_element (element); |
| } |
| |
| /* |
| * Test if the parser pushes clean data properly. |
| */ |
| void |
| gst_parser_test_normal (guint8 * data, guint size) |
| { |
| GstParserTest ptest; |
| |
| gst_parser_test_init (&ptest, data, size, 10); |
| gst_parser_test_run (&ptest, NULL); |
| } |
| |
| /* |
| * Test if parser drains its buffers properly. Even one single frame |
| * should be drained and pushed forward when EOS occurs. This single frame |
| * case is special, since normally the parser needs more data to be sure |
| * about stream format. But it should still push the frame forward in EOS. |
| */ |
| void |
| gst_parser_test_drain_single (guint8 * data, guint size) |
| { |
| GstParserTest ptest; |
| |
| gst_parser_test_init (&ptest, data, size, 1); |
| gst_parser_test_run (&ptest, NULL); |
| } |
| |
| /* |
| * Make sure that parser does not drain garbage when EOS occurs. |
| */ |
| void |
| gst_parser_test_drain_garbage (guint8 * data, guint size, guint8 * garbage, |
| guint gsize) |
| { |
| GstParserTest ptest; |
| |
| /* provide enough initial frames since it may take some parsers some |
| * time to be convinced of proper sync */ |
| gst_parser_test_init (&ptest, data, size, 10); |
| ptest.series[1].data = garbage; |
| ptest.series[1].size = gsize; |
| ptest.series[1].num = 1; |
| gst_parser_test_run (&ptest, NULL); |
| } |
| |
| /* |
| * Test if parser splits a buffer that contains two frames into two |
| * separate buffers properly. |
| */ |
| void |
| gst_parser_test_split (guint8 * data, guint size) |
| { |
| GstParserTest ptest; |
| |
| gst_parser_test_init (&ptest, data, size, 10); |
| ptest.series[0].fpb = 2; |
| gst_parser_test_run (&ptest, NULL); |
| } |
| |
| /* |
| * Test if the parser skips garbage between frames properly. |
| */ |
| void |
| gst_parser_test_skip_garbage (guint8 * data, guint size, guint8 * garbage, |
| guint gsize) |
| { |
| GstParserTest ptest; |
| |
| gst_parser_test_init (&ptest, data, size, 10); |
| ptest.series[1].data = garbage; |
| ptest.series[1].size = gsize; |
| ptest.series[1].num = 1; |
| ptest.series[2].data = data; |
| ptest.series[2].size = size; |
| ptest.series[2].num = 10; |
| gst_parser_test_run (&ptest, NULL); |
| } |
| |
| /* |
| * Test if the src caps are set according to stream format. |
| */ |
| void |
| gst_parser_test_output_caps (guint8 * data, guint size, |
| const gchar * input_caps, const gchar * output_caps) |
| { |
| GstParserTest ptest; |
| |
| gst_parser_test_init (&ptest, data, size, 10); |
| if (input_caps) { |
| ptest.src_caps = gst_caps_from_string (input_caps); |
| fail_unless (ptest.src_caps != NULL); |
| } |
| if (output_caps) { |
| ptest.sink_caps = gst_caps_from_string (output_caps); |
| fail_unless (ptest.sink_caps != NULL); |
| } |
| gst_parser_test_run (&ptest, NULL); |
| if (ptest.sink_caps) |
| gst_caps_unref (ptest.sink_caps); |
| if (ptest.src_caps) |
| gst_caps_unref (ptest.src_caps); |
| } |
| |
| /* |
| * Test if the src caps are set according to stream format. |
| */ |
| GstCaps * |
| gst_parser_test_get_output_caps (guint8 * data, guint size, |
| const gchar * input_caps) |
| { |
| GstParserTest ptest; |
| GstCaps *out_caps; |
| |
| gst_parser_test_init (&ptest, data, size, 10); |
| if (input_caps) { |
| ptest.src_caps = gst_caps_from_string (input_caps); |
| fail_unless (ptest.src_caps != NULL); |
| } |
| gst_parser_test_run (&ptest, &out_caps); |
| if (ptest.src_caps) |
| gst_caps_unref (ptest.src_caps); |
| |
| return out_caps; |
| } |