mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-09 13:48:55 +00:00
298 lines
9.4 KiB
C
298 lines
9.4 KiB
C
/* GStreamer
|
|
* Copyright (C) 2010 Sebastian Dröge <sebastian.droege@collabora.co.uk>
|
|
*
|
|
* 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 <string.h>
|
|
|
|
#include <gst/check/gstcheck.h>
|
|
#include <gst/video/video.h>
|
|
#include <gst/app/gstappsrc.h>
|
|
|
|
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: 200\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"}
|
|
};
|
|
|
|
static const TestBuffer buf1 = {
|
|
40 * GST_MSECOND,
|
|
60 * GST_MSECOND,
|
|
{"1,,DefaultVCD, NTP,0000,0000,0000,,Some Test Blabla"}
|
|
};
|
|
|
|
static void
|
|
sink_handoff_cb_xRGB (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;
|
|
GstMapInfo map;
|
|
gboolean all_red = TRUE;
|
|
|
|
gst_buffer_map (buffer, &map, GST_MAP_READ);
|
|
|
|
fail_unless_equals_int (map.size, 640 * 480 * 4);
|
|
|
|
for (i = 0; i < 640; i++) {
|
|
for (j = 0; j < 480; j++) {
|
|
all_red = all_red && (map.data[i * 480 * 4 + j * 4 + 1] == 255 &&
|
|
map.data[i * 480 * 4 + j * 4 + 2] == 0 &&
|
|
map.data[i * 480 * 4 + j * 4 + 3] == 0);
|
|
}
|
|
}
|
|
gst_buffer_unmap (buffer, &map);
|
|
|
|
fail_unless (contains_text != all_red,
|
|
"Frame %d is incorrect (all red %d, contains text %d)", *sink_pos,
|
|
all_red, contains_text);
|
|
*sink_pos = *sink_pos + 1;
|
|
}
|
|
|
|
static void
|
|
sink_handoff_cb_I420 (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 c, i, j;
|
|
gboolean all_red = TRUE;
|
|
guint8 *comp;
|
|
gint comp_stride, comp_width, comp_height;
|
|
const guint8 color[] = { 81, 90, 240 };
|
|
GstVideoInfo info;
|
|
GstVideoFrame frame;
|
|
|
|
gst_video_info_init (&info);
|
|
gst_video_info_set_format (&info, GST_VIDEO_FORMAT_I420, 640, 480);
|
|
|
|
gst_video_frame_map (&frame, &info, buffer, GST_MAP_READ);
|
|
|
|
for (c = 0; c < 3; c++) {
|
|
comp = GST_VIDEO_FRAME_COMP_DATA (&frame, c);
|
|
comp_stride = GST_VIDEO_FRAME_COMP_STRIDE (&frame, c);
|
|
comp_width = GST_VIDEO_FRAME_COMP_WIDTH (&frame, c);
|
|
comp_height = GST_VIDEO_FRAME_COMP_HEIGHT (&frame, c);
|
|
|
|
for (i = 0; i < comp_height; i++) {
|
|
for (j = 0; j < comp_width; j++) {
|
|
all_red = all_red && (comp[i * comp_stride + j] == color[c]);
|
|
}
|
|
}
|
|
}
|
|
gst_video_frame_unmap (&frame);
|
|
|
|
fail_unless (contains_text != all_red,
|
|
"Frame %d is incorrect (all red %d, contains text %d)", *sink_pos,
|
|
all_red, contains_text);
|
|
*sink_pos = *sink_pos + 1;
|
|
}
|
|
|
|
static gulong probe_id = 0;
|
|
|
|
static GstPadProbeReturn
|
|
src_buffer_probe_cb (GstPad * pad, GstPadProbeInfo * info, gpointer user_data)
|
|
{
|
|
GstBuffer *buffer = GST_PAD_PROBE_INFO_BUFFER (info);
|
|
GstPad *otherpad = GST_PAD (user_data);
|
|
|
|
if (GST_BUFFER_TIMESTAMP (buffer) == buf1.ts)
|
|
gst_pad_remove_probe (otherpad, probe_id);
|
|
|
|
return GST_PAD_PROBE_OK;
|
|
}
|
|
|
|
#define CREATE_BASIC_TEST(format) \
|
|
GST_START_TEST (test_assrender_basic_##format) \
|
|
{ \
|
|
GstElement *pipeline; \
|
|
GstElement *appsrc, *videotestsrc, *capsfilter, *assrender, *fakesink; \
|
|
guint sink_pos = 0; \
|
|
GstCaps *video_caps; \
|
|
GstCaps *text_caps; \
|
|
GstBuffer *buf; \
|
|
GstBus *bus; \
|
|
GMainLoop *loop; \
|
|
GstPad *pad, *blocked_pad; \
|
|
guint bus_watch = 0; \
|
|
GstVideoInfo info; \
|
|
\
|
|
pipeline = gst_pipeline_new ("pipeline"); \
|
|
fail_unless (pipeline != NULL); \
|
|
\
|
|
capsfilter = gst_element_factory_make ("capsfilter", NULL); \
|
|
fail_unless (capsfilter != NULL); \
|
|
gst_video_info_init (&info); \
|
|
gst_video_info_set_format (&info, GST_VIDEO_FORMAT_##format, 640, 480); \
|
|
info.fps_n = 25; \
|
|
info.fps_d = 1; \
|
|
video_caps = gst_video_info_to_caps (&info); \
|
|
g_object_set (capsfilter, "caps", video_caps, NULL); \
|
|
gst_caps_unref (video_caps); \
|
|
blocked_pad = gst_element_get_static_pad (capsfilter, "src"); \
|
|
gst_pad_add_probe (blocked_pad, GST_PAD_PROBE_TYPE_BLOCK_DOWNSTREAM, NULL, NULL, NULL); \
|
|
\
|
|
appsrc = gst_element_factory_make ("appsrc", NULL); \
|
|
fail_unless (appsrc != NULL); \
|
|
buf = gst_buffer_new_and_alloc (strlen (buf0.buf) + 1); \
|
|
gst_buffer_fill (buf, 0, buf0.buf, strlen (buf0.buf) + 1); \
|
|
GST_BUFFER_TIMESTAMP (buf) = buf0.ts; \
|
|
GST_BUFFER_DURATION (buf) = buf0.duration; \
|
|
text_caps = \
|
|
gst_caps_new_simple ("application/x-ssa", "codec_data", GST_TYPE_BUFFER, \
|
|
buf, NULL); \
|
|
gst_buffer_unref (buf); \
|
|
gst_app_src_set_caps (GST_APP_SRC (appsrc), text_caps); \
|
|
g_object_set (appsrc, "format", GST_FORMAT_TIME, NULL); \
|
|
pad = gst_element_get_static_pad (appsrc, "src"); \
|
|
probe_id = gst_pad_add_probe (pad, GST_PAD_PROBE_TYPE_BUFFER, src_buffer_probe_cb, \
|
|
gst_object_ref (blocked_pad), (GDestroyNotify) gst_object_unref); \
|
|
gst_object_unref (blocked_pad); \
|
|
gst_object_unref (pad); \
|
|
\
|
|
videotestsrc = gst_element_factory_make ("videotestsrc", NULL); \
|
|
fail_unless (videotestsrc != NULL); \
|
|
g_object_set (videotestsrc, "num-buffers", 5, "pattern", 4, NULL); \
|
|
\
|
|
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_##format), \
|
|
&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); \
|
|
bus_watch = 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 (buf1.buf) + 1); \
|
|
gst_buffer_fill (buf, 0, buf1.buf, strlen (buf1.buf) + 1); \
|
|
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); \
|
|
g_source_remove (bus_watch); \
|
|
} \
|
|
\
|
|
GST_END_TEST;
|
|
|
|
CREATE_BASIC_TEST (xRGB);
|
|
CREATE_BASIC_TEST (I420);
|
|
|
|
static 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_xRGB);
|
|
tcase_add_test (tc_chain, test_assrender_basic_I420);
|
|
|
|
return s;
|
|
}
|
|
|
|
GST_CHECK_MAIN (assrender);
|