From 88bd4a8ada893b16d9c3e6f609fa84195ab036e3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= Date: Tue, 16 Feb 2010 21:41:38 +0100 Subject: [PATCH] assrender: Add simple unit test This renders a test string on top of a red video stream and checks if the text is only rendered at the correct timestamps. --- tests/check/Makefile.am | 10 ++ tests/check/elements/.gitignore | 1 + tests/check/elements/assrender.c | 220 +++++++++++++++++++++++++++++++ 3 files changed, 231 insertions(+) create mode 100644 tests/check/elements/assrender.c diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index f0449fddcf..97386fac0f 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -27,6 +27,12 @@ SUPPRESSIONS = $(top_srcdir)/common/gst.supp $(srcdir)/gst-plugins-bad.supp clean-local: clean-local-check +if USE_ASSRENDER +check_assrender = elements/assrender +else +check_assrender = +endif + if USE_FAAC check_faac = elements/faac else @@ -107,6 +113,7 @@ noinst_PROGRAMS = \ check_PROGRAMS = \ generic/states \ + $(check_assrender) \ $(check_faac) \ $(check_faad) \ $(check_mpeg2enc) \ @@ -159,4 +166,7 @@ elements_kate_LDADD = $(GST_BASE_LIBS) $(LDADD) elements_rtpmux_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS) elements_rtpmux_LDADD = $(GST_BASE_LIBS) $(LDADD) -lgstrtp-0.10 +elements_assrender_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_BASE_CFLAGS) $(AM_CFLAGS) +elements_assrender_LDADD = $(GST_BASE_LIBS) $(LDADD) -lgstvideo-0.10 -lgstapp-0.10 + EXTRA_DIST = gst-plugins-bad.supp diff --git a/tests/check/elements/.gitignore b/tests/check/elements/.gitignore index bc2dd1f16e..13e49ac7fe 100644 --- a/tests/check/elements/.gitignore +++ b/tests/check/elements/.gitignore @@ -2,6 +2,7 @@ aacparse amrparse asfmux +assrender autoconvert camerabin capssetter diff --git a/tests/check/elements/assrender.c b/tests/check/elements/assrender.c new file mode 100644 index 0000000000..bdcb06eeba --- /dev/null +++ b/tests/check/elements/assrender.c @@ -0,0 +1,220 @@ +/* GStreamer + * Copyright (C) 2010 Sebastian Dröge + * + * 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., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#include + +#include +#include +#include + +static gboolean +bus_handler (GstBus * bus, GstMessage * message, gpointer data) +{ + GMainLoop *loop = (GMainLoop *) data; + + switch (message->type) { + case GST_MESSAGE_EOS: + g_main_loop_quit (loop); + break; + case GST_MESSAGE_WARNING: + case GST_MESSAGE_ERROR:{ + GError *gerror; + gchar *debug; + + if (message->type == GST_MESSAGE_WARNING) + gst_message_parse_warning (message, &gerror, &debug); + else + gst_message_parse_error (message, &gerror, &debug); + gst_object_default_error (GST_MESSAGE_SRC (message), gerror, debug); + gst_message_unref (message); + g_error_free (gerror); + g_free (debug); + g_main_loop_quit (loop); + break; + } + default: + break; + } + + return TRUE; +} + +typedef struct +{ + GstClockTime ts; + GstClockTime duration; + const gchar buf[]; +} TestBuffer; + +static const TestBuffer buf0 = { + 0, + 0, + "[Script Info]\n" + "; This is a Sub Station Alpha v4 script.\n" + "; For Sub Station Alpha info and downloads,\n" + "; go to http://www.eswat.demon.co.uk/\n" + "Title: Some Test\n" + "Script Updated By: version 2.8.01\n" + "ScriptType: v4.00\n" + "Collisions: Normal\n" + "PlayResY: 600\n" + "PlayDepth: 0\n" + "Timer: 100,0000\n" + " \n" + "[V4 Styles]\n" + "Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, TertiaryColour, BackColour, Bold, Italic, BorderStyle, Outline, Shadow, \n" + " Alignment, MarginL, MarginR, MarginV, AlphaLevel, Encoding\n" + "Style: DefaultVCD, Arial,28,11861244,11861244,11861244,-2147483640,-1,0,1,1,2,2,30,30,30,0,0\n" + " \n" + "[Events]\n" + "Format: Marked, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n" +}; + +static const TestBuffer buf1 = { + 40 * GST_MSECOND, + 40 * GST_MSECOND, + "Dialogue: Marked=0,0:00:00.04,0:00:00.10,DefaultVCD, NTP,0000,0000,0000,,Some Test Blabla\n" +}; + +static void +sink_handoff_cb (GstElement * object, GstBuffer * buffer, GstPad * pad, + gpointer user_data) +{ + guint *sink_pos = (guint *) user_data; + gboolean contains_text = (*sink_pos == 1 || *sink_pos == 2); + guint i, j; + guint8 *data = GST_BUFFER_DATA (buffer); + gboolean all_red = TRUE; + + fail_unless_equals_int (GST_BUFFER_SIZE (buffer), 640 * 480 * 4); + for (i = 0; i < 640; i++) { + for (j = 0; j < 480; j++) { + all_red = all_red && (data[i * 480 * 4 + j * 4 + 1] == 255 && + data[i * 480 * 4 + j * 4 + 2] == 0 && + data[i * 480 * 4 + j * 4 + 3] == 0); + } + } + + fail_unless (contains_text || all_red); + *sink_pos = *sink_pos + 1; +} + +GST_START_TEST (test_assrender_basic) +{ + GstElement *pipeline; + GstElement *appsrc, *videotestsrc, *capsfilter, *assrender, *fakesink; + guint sink_pos = 0; + GstCaps *video_caps; + GstCaps *text_caps; + GstBuffer *buf; + GstBus *bus; + GMainLoop *loop; + + pipeline = gst_pipeline_new ("pipeline"); + fail_unless (pipeline != NULL); + + appsrc = gst_element_factory_make ("appsrc", NULL); + fail_unless (appsrc != NULL); + text_caps = gst_caps_new_simple ("application/x-ssa", NULL); + gst_app_src_set_caps (GST_APP_SRC (appsrc), text_caps); + g_object_set (appsrc, "format", GST_FORMAT_TIME, NULL); + + videotestsrc = gst_element_factory_make ("videotestsrc", NULL); + fail_unless (videotestsrc != NULL); + g_object_set (videotestsrc, "num-buffers", 5, "pattern", 4, NULL); + + capsfilter = gst_element_factory_make ("capsfilter", NULL); + fail_unless (capsfilter != NULL); + video_caps = + gst_video_format_new_caps (GST_VIDEO_FORMAT_xRGB, 640, 480, 25, 1, 1, 1); + g_object_set (capsfilter, "caps", video_caps, NULL); + gst_caps_unref (video_caps); + + assrender = gst_element_factory_make ("assrender", NULL); + fail_unless (assrender != NULL); + + fakesink = gst_element_factory_make ("fakesink", NULL); + fail_unless (fakesink != NULL); + g_object_set (fakesink, "signal-handoffs", TRUE, "async", FALSE, NULL); + g_signal_connect (fakesink, "handoff", G_CALLBACK (sink_handoff_cb), + &sink_pos); + + gst_bin_add_many (GST_BIN (pipeline), appsrc, videotestsrc, capsfilter, + assrender, fakesink, NULL); + + fail_unless (gst_element_link_pads (appsrc, "src", assrender, "text_sink")); + fail_unless (gst_element_link_pads (videotestsrc, "src", capsfilter, "sink")); + fail_unless (gst_element_link_pads (capsfilter, "src", assrender, + "video_sink")); + fail_unless (gst_element_link_pads (assrender, "src", fakesink, "sink")); + + loop = g_main_loop_new (NULL, TRUE); + fail_unless (loop != NULL); + + bus = gst_element_get_bus (pipeline); + fail_unless (bus != NULL); + gst_bus_add_watch (bus, bus_handler, loop); + gst_object_unref (bus); + + fail_unless_equals_int (gst_element_set_state (pipeline, GST_STATE_PLAYING), + GST_STATE_CHANGE_SUCCESS); + + buf = gst_buffer_new_and_alloc (strlen (buf0.buf) + 1); + memcpy (GST_BUFFER_DATA (buf), buf0.buf, GST_BUFFER_SIZE (buf)); + gst_buffer_set_caps (buf, text_caps); + GST_BUFFER_TIMESTAMP (buf) = buf0.ts; + GST_BUFFER_DURATION (buf) = buf0.duration; + gst_app_src_push_buffer (GST_APP_SRC (appsrc), buf); + buf = gst_buffer_new_and_alloc (strlen (buf1.buf) + 1); + memcpy (GST_BUFFER_DATA (buf), buf1.buf, GST_BUFFER_SIZE (buf)); + gst_buffer_set_caps (buf, text_caps); + GST_BUFFER_TIMESTAMP (buf) = buf1.ts; + GST_BUFFER_DURATION (buf) = buf1.duration; + gst_app_src_push_buffer (GST_APP_SRC (appsrc), buf); + gst_caps_unref (text_caps); + gst_app_src_end_of_stream (GST_APP_SRC (appsrc)); + + g_main_loop_run (loop); + + gst_element_set_state (pipeline, GST_STATE_NULL); + + fail_unless_equals_int (sink_pos, 5); + + g_object_unref (pipeline); + g_main_loop_unref (loop); +} + +GST_END_TEST; + +Suite * +assrender_suite (void) +{ + Suite *s = suite_create ("assrender"); + TCase *tc_chain = tcase_create ("linear"); + + /* time out after 120s, not the default 3 */ + tcase_set_timeout (tc_chain, 120); + + suite_add_tcase (s, tc_chain); + tcase_add_test (tc_chain, test_assrender_basic); + + return s; +} + +GST_CHECK_MAIN (assrender);