/* GStreamer unit tests for decodebin2
*
* Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>
+ * Copyright (C) 2011 Hewlett-Packard Development Company, L.P.
+ * Author: Tim-Philipp Müller <tim.muller@collabora.co.uk>, Collabora Ltd.
+ * Sebastian Dröge <sebastian.droege@collabora.co.uk>, Collabora Ltd.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
#endif
#include <gst/check/gstcheck.h>
+#include <gst/base/gstbaseparse.h>
#include <unistd.h>
static const gchar dummytext[] =
GST_END_TEST;
+/* Fake mp3 parser for test */
+typedef GstBaseParse TestMpegAudioParse;
+typedef GstBaseParseClass TestMpegAudioParseClass;
+
+static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("audio/mpeg, mpegversion=1, layer=[1,3], parsed=(b)true")
+ );
+
+static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("audio/mpeg, mpegversion=1, parsed=(bool) { false, true }")
+ );
+
+static GType test_mpeg_audio_parse_get_type (void);
+static gboolean test_mpeg_audio_parse_start (GstBaseParse * parse);
+static gboolean test_mpeg_audio_parse_stop (GstBaseParse * parse);
+static gboolean test_mpeg_audio_parse_check_valid_frame (GstBaseParse * parse,
+ GstBaseParseFrame * frame, guint * size, gint * skipsize);
+static GstFlowReturn test_mpeg_audio_parse_parse_frame (GstBaseParse * parse,
+ GstBaseParseFrame * frame);
+
+GST_BOILERPLATE (TestMpegAudioParse, test_mpeg_audio_parse, GstBaseParse,
+ GST_TYPE_BASE_PARSE);
+
+static void
+test_mpeg_audio_parse_base_init (gpointer klass)
+{
+ GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
+
+ gst_element_class_add_static_pad_template (element_class,
+ &sink_template);
+ gst_element_class_add_static_pad_template (element_class, &src_template);
+
+ gst_element_class_set_details_simple (element_class, "MPEG1 Audio Parser",
+ "Codec/Parser/Audio", "Pretends to parse mpeg1 audio stream",
+ "Foo Bar <foo@bar.com>");
+}
+
+static void
+test_mpeg_audio_parse_class_init (TestMpegAudioParseClass * klass)
+{
+ GstBaseParseClass *parse_class = GST_BASE_PARSE_CLASS (klass);
+
+ parse_class->start = test_mpeg_audio_parse_start;
+ parse_class->stop = test_mpeg_audio_parse_stop;
+ parse_class->check_valid_frame = test_mpeg_audio_parse_check_valid_frame;
+ parse_class->parse_frame = test_mpeg_audio_parse_parse_frame;
+}
+
+static gint num_parse_instances = 0;
+
+static void
+test_mpeg_audio_parse_init (TestMpegAudioParse * mp3parse,
+ TestMpegAudioParseClass * klass)
+{
+ /* catch decodebin plugging parsers in a loop early */
+ fail_unless (++num_parse_instances < 10);
+}
+
+static gboolean
+test_mpeg_audio_parse_start (GstBaseParse * parse)
+{
+ gst_base_parse_set_min_frame_size (parse, 6);
+ return TRUE;
+}
+
+static gboolean
+test_mpeg_audio_parse_stop (GstBaseParse * parse)
+{
+ return TRUE;
+}
+
+static gboolean
+test_mpeg_audio_parse_check_valid_frame (GstBaseParse * parse,
+ GstBaseParseFrame * frame, guint * framesize, gint * skipsize)
+{
+ const guint8 *data = GST_BUFFER_DATA (frame->buffer);
+
+ if ((GST_READ_UINT16_BE (data) & 0xffe0) == 0xffe0) {
+ /* this framesize is hard-coded for ../test.mp3 */
+ *framesize = 1045;
+ return TRUE;
+ } else {
+ *skipsize = 1;
+ return FALSE;
+ }
+}
+
+static GstFlowReturn
+test_mpeg_audio_parse_parse_frame (GstBaseParse * parse,
+ GstBaseParseFrame * frame)
+{
+ if (GST_BUFFER_OFFSET (frame->buffer) == 0) {
+ GstCaps *caps;
+
+ caps = gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT, 1,
+ "mpegaudioversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 3,
+ "rate", G_TYPE_INT, 44100, "channels", G_TYPE_INT, 2, NULL);
+ gst_buffer_set_caps (frame->buffer, caps);
+ gst_pad_set_caps (GST_BASE_PARSE_SRC_PAD (parse), caps);
+ gst_caps_unref (caps);
+ }
+ return GST_FLOW_OK;
+}
+
+static gboolean
+plugin_init (GstPlugin * plugin)
+{
+ return gst_element_register (plugin, "testmpegaudioparse", GST_RANK_NONE,
+ test_mpeg_audio_parse_get_type ());
+}
+
+GST_START_TEST (test_mp3_parser_loop)
+{
+ GstStateChangeReturn sret;
+ GstPluginFeature *feature;
+ GstMessage *msg;
+ GstElement *pipe, *src, *dec;
+ gchar *path;
+
+ num_parse_instances = 0;
+
+ gst_plugin_register_static (GST_VERSION_MAJOR, GST_VERSION_MINOR,
+ "fakemp3parse", "fakemp3parse", plugin_init, VERSION, "LGPL",
+ "gst-plugins-base", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);
+
+ feature = gst_default_registry_find_feature ("testmpegaudioparse",
+ GST_TYPE_ELEMENT_FACTORY);
+
+ gst_plugin_feature_set_rank (feature, GST_RANK_PRIMARY + 100);
+
+ pipe = gst_pipeline_new (NULL);
+
+ src = gst_element_factory_make ("filesrc", NULL);
+ fail_unless (src != NULL);
+
+ path = g_build_filename (GST_TEST_FILES_PATH, "test.mp3", NULL);
+ g_object_set (src, "location", path, NULL);
+ g_free (path);
+
+ dec = gst_element_factory_make ("decodebin2", NULL);
+ fail_unless (dec != NULL);
+
+ gst_bin_add_many (GST_BIN (pipe), src, dec, NULL);
+ gst_element_link_many (src, dec, NULL);
+
+ sret = gst_element_set_state (pipe, GST_STATE_PLAYING);
+ fail_unless_equals_int (sret, GST_STATE_CHANGE_ASYNC);
+
+ /* wait for unlinked error */
+ msg = gst_bus_timed_pop_filtered (GST_ELEMENT_BUS (pipe),
+ GST_CLOCK_TIME_NONE, GST_MESSAGE_ERROR);
+ gst_message_unref (msg);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_object_unref (pipe);
+
+ /* make sure out parser got plugged at all though */
+ fail_unless_equals_int (num_parse_instances, 1);
+
+ /* don't want to interfere with any other of the other tests */
+ gst_plugin_feature_set_rank (feature, GST_RANK_NONE);
+ gst_object_unref (feature);
+}
+
+GST_END_TEST;
+
+/* Fake parser/decoder for parser_negotiation test */
+static GType gst_fake_h264_parser_get_type (void);
+static GType gst_fake_h264_decoder_get_type (void);
+
+#undef parent_class
+#define parent_class fake_h264_parser_parent_class
+typedef struct _GstFakeH264Parser GstFakeH264Parser;
+typedef GstElementClass GstFakeH264ParserClass;
+
+struct _GstFakeH264Parser
+{
+ GstElement parent;
+};
+
+GST_BOILERPLATE (GstFakeH264Parser, gst_fake_h264_parser, GstElement,
+ GST_TYPE_ELEMENT);
+
+static void
+gst_fake_h264_parser_base_init (gpointer klass)
+{
+ static GstStaticPadTemplate sink_templ = GST_STATIC_PAD_TEMPLATE ("sink",
+ GST_PAD_SINK, GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-h264"));
+ static GstStaticPadTemplate src_templ = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC, GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-h264, "
+ "stream-format=(string) { avc, byte-stream }"));
+ GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
+
+ gst_element_class_add_static_pad_template (element_class, &sink_templ);
+ gst_element_class_add_static_pad_template (element_class, &src_templ);
+ gst_element_class_set_details_simple (element_class,
+ "FakeH264Parser", "Codec/Parser/Converter/Video", "yep", "me");
+}
+
+static void
+gst_fake_h264_parser_class_init (GstFakeH264ParserClass * klass)
+{
+}
+
+static gboolean
+gst_fake_h264_parser_sink_setcaps (GstPad * pad, GstCaps * caps)
+{
+ GstElement *self = GST_ELEMENT (gst_pad_get_parent (pad));
+ GstPad *otherpad = gst_element_get_static_pad (self, "src");
+ GstCaps *accepted_caps;
+ GstStructure *s;
+ const gchar *stream_format;
+
+ accepted_caps = gst_pad_get_allowed_caps (otherpad);
+ accepted_caps = gst_caps_make_writable (accepted_caps);
+ gst_caps_truncate (accepted_caps);
+
+ s = gst_caps_get_structure (accepted_caps, 0);
+ stream_format = gst_structure_get_string (s, "stream-format");
+ if (!stream_format)
+ gst_structure_set (s, "stream-format", G_TYPE_STRING, "avc", NULL);
+
+ gst_pad_set_caps (otherpad, accepted_caps);
+ gst_caps_unref (accepted_caps);
+
+ gst_object_unref (otherpad);
+ gst_object_unref (self);
+
+ return TRUE;
+}
+
+static GstFlowReturn
+gst_fake_h264_parser_sink_chain (GstPad * pad, GstBuffer * buf)
+{
+ GstElement *self = GST_ELEMENT (gst_pad_get_parent (pad));
+ GstPad *otherpad = gst_element_get_static_pad (self, "src");
+ GstFlowReturn ret = GST_FLOW_OK;
+
+ buf = gst_buffer_make_metadata_writable (buf);
+ gst_buffer_set_caps (buf, GST_PAD_CAPS (otherpad));
+
+ ret = gst_pad_push (otherpad, buf);
+
+ gst_object_unref (otherpad);
+ gst_object_unref (self);
+
+ return ret;
+}
+
+static void
+gst_fake_h264_parser_init (GstFakeH264Parser * self,
+ GstFakeH264ParserClass * klass)
+{
+ GstPad *pad;
+
+ pad =
+ gst_pad_new_from_template (gst_element_class_get_pad_template
+ (GST_ELEMENT_GET_CLASS (self), "sink"), "sink");
+ gst_pad_set_setcaps_function (pad, gst_fake_h264_parser_sink_setcaps);
+ gst_pad_set_chain_function (pad, gst_fake_h264_parser_sink_chain);
+ gst_element_add_pad (GST_ELEMENT (self), pad);
+
+ pad =
+ gst_pad_new_from_template (gst_element_class_get_pad_template
+ (GST_ELEMENT_GET_CLASS (self), "src"), "src");
+ gst_element_add_pad (GST_ELEMENT (self), pad);
+}
+
+#undef parent_class
+#define parent_class fake_h264_decoder_parent_class
+typedef struct _GstFakeH264Decoder GstFakeH264Decoder;
+typedef GstElementClass GstFakeH264DecoderClass;
+
+struct _GstFakeH264Decoder
+{
+ GstElement parent;
+};
+
+GST_BOILERPLATE (GstFakeH264Decoder, gst_fake_h264_decoder, GstElement,
+ GST_TYPE_ELEMENT);
+
+static void
+gst_fake_h264_decoder_base_init (gpointer klass)
+{
+ static GstStaticPadTemplate sink_templ = GST_STATIC_PAD_TEMPLATE ("sink",
+ GST_PAD_SINK, GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-h264, " "stream-format=(string) byte-stream"));
+ static GstStaticPadTemplate src_templ = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC, GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-raw-yuv"));
+ GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
+
+ gst_element_class_add_static_pad_template (element_class, &sink_templ);
+ gst_element_class_add_static_pad_template (element_class, &src_templ);
+ gst_element_class_set_details_simple (element_class,
+ "FakeH264Decoder", "Codec/Decoder/Video", "yep", "me");
+}
+
+static void
+gst_fake_h264_decoder_class_init (GstFakeH264DecoderClass * klass)
+{
+}
+
+static gboolean
+gst_fake_h264_decoder_sink_setcaps (GstPad * pad, GstCaps * caps)
+{
+ GstElement *self = GST_ELEMENT (gst_pad_get_parent (pad));
+ GstPad *otherpad = gst_element_get_static_pad (self, "src");
+
+ caps = gst_caps_new_simple ("video/x-raw-yuv", NULL);
+ gst_pad_set_caps (otherpad, caps);
+ gst_caps_unref (caps);
+
+ gst_object_unref (otherpad);
+ gst_object_unref (self);
+
+ return TRUE;
+}
+
+static GstFlowReturn
+gst_fake_h264_decoder_sink_chain (GstPad * pad, GstBuffer * buf)
+{
+ GstElement *self = GST_ELEMENT (gst_pad_get_parent (pad));
+ GstPad *otherpad = gst_element_get_static_pad (self, "src");
+ GstFlowReturn ret = GST_FLOW_OK;
+
+ buf = gst_buffer_make_metadata_writable (buf);
+ gst_buffer_set_caps (buf, GST_PAD_CAPS (otherpad));
+
+ ret = gst_pad_push (otherpad, buf);
+
+ gst_object_unref (otherpad);
+ gst_object_unref (self);
+
+ return ret;
+}
+
+static void
+gst_fake_h264_decoder_init (GstFakeH264Decoder * self,
+ GstFakeH264DecoderClass * klass)
+{
+ GstPad *pad;
+
+ pad =
+ gst_pad_new_from_template (gst_element_class_get_pad_template
+ (GST_ELEMENT_GET_CLASS (self), "sink"), "sink");
+ gst_pad_set_setcaps_function (pad, gst_fake_h264_decoder_sink_setcaps);
+ gst_pad_set_chain_function (pad, gst_fake_h264_decoder_sink_chain);
+ gst_element_add_pad (GST_ELEMENT (self), pad);
+
+ pad =
+ gst_pad_new_from_template (gst_element_class_get_pad_template
+ (GST_ELEMENT_GET_CLASS (self), "src"), "src");
+ gst_element_add_pad (GST_ELEMENT (self), pad);
+}
+
+static void
+parser_negotiation_pad_added_cb (GstElement * dec, GstPad * pad,
+ gpointer user_data)
+{
+ GstBin *pipe = user_data;
+ GstElement *sink;
+ GstPad *sinkpad;
+
+ sink = gst_element_factory_make ("fakesink", NULL);
+ gst_bin_add (pipe, sink);
+ gst_element_sync_state_with_parent (sink);
+ sinkpad = gst_element_get_static_pad (sink, "sink");
+ gst_pad_link (pad, sinkpad);
+ gst_object_unref (sinkpad);
+}
+
+GST_START_TEST (test_parser_negotiation)
+{
+ GstStateChangeReturn sret;
+ GstMessage *msg;
+ GstCaps *caps;
+ GstElement *pipe, *src, *filter, *dec;
+
+ gst_element_register (NULL, "fakeh264parse", GST_RANK_PRIMARY + 101,
+ gst_fake_h264_parser_get_type ());
+ gst_element_register (NULL, "fakeh264dec", GST_RANK_PRIMARY + 100,
+ gst_fake_h264_decoder_get_type ());
+
+ pipe = gst_pipeline_new (NULL);
+
+ src = gst_element_factory_make ("fakesrc", NULL);
+ fail_unless (src != NULL);
+ g_object_set (G_OBJECT (src), "num-buffers", 5, "sizetype", 2, "filltype", 2,
+ "can-activate-pull", FALSE, NULL);
+
+ filter = gst_element_factory_make ("capsfilter", NULL);
+ fail_unless (filter != NULL);
+ caps = gst_caps_from_string ("video/x-h264");
+ g_object_set (G_OBJECT (filter), "caps", caps, NULL);
+ gst_caps_unref (caps);
+
+ dec = gst_element_factory_make ("decodebin2", NULL);
+ fail_unless (dec != NULL);
+
+ g_signal_connect (dec, "pad-added",
+ G_CALLBACK (parser_negotiation_pad_added_cb), pipe);
+
+ gst_bin_add_many (GST_BIN (pipe), src, filter, dec, NULL);
+ gst_element_link_many (src, filter, dec, NULL);
+
+ sret = gst_element_set_state (pipe, GST_STATE_PLAYING);
+ fail_unless_equals_int (sret, GST_STATE_CHANGE_ASYNC);
+
+ /* wait for EOS or error */
+ msg = gst_bus_timed_pop_filtered (GST_ELEMENT_BUS (pipe),
+ GST_CLOCK_TIME_NONE, GST_MESSAGE_ERROR | GST_MESSAGE_EOS);
+ fail_unless (msg != NULL);
+ fail_unless (GST_MESSAGE_TYPE (msg) == GST_MESSAGE_EOS);
+ gst_message_unref (msg);
+
+ gst_element_set_state (pipe, GST_STATE_NULL);
+ gst_object_unref (pipe);
+}
+
+GST_END_TEST;
+
static Suite *
decodebin2_suite (void)
{
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_text_plain_streams);
tcase_add_test (tc_chain, test_reuse_without_decoders);
+ tcase_add_test (tc_chain, test_mp3_parser_loop);
+ tcase_add_test (tc_chain, test_parser_negotiation);
return s;
}