gstreamer/subprojects/gst-plugins-bad/tests/check/elements/templatematch.c

171 lines
5 KiB
C

/*
* GStreamer
*
* unit test for templatematch
*
* Copyright (C) 2014 Samsung Electronics. All rights reserved.
* Author: Thiago Santos <ts.santos@sisa.samsung.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>
#define CAPS_TMPL "video/x-raw, format=(string)BGR"
GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (CAPS_TMPL)
);
GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (CAPS_TMPL)
);
/* Create a 16x16 buffer split in 4 equal squares
* BG
* Rb
*
* B=Blue, G=Green, R=Red, b=black
*/
static GstBuffer *
create_input_buffer (void)
{
guint8 *data;
gsize size;
gint i, j, base;
size = 3 * 16 * 16; /* BGR 16x16 */
data = g_malloc0 (size);
/* blue and green */
for (j = 0; j < 8; j++) {
for (i = 0; i < 8; i++) {
base = j * 16;
data[base + i] = 255;
data[base + i + 1] = 0;
data[base + i + 2] = 0;
data[base + 8 + i] = 0;
data[base + 8 + i + 1] = 255;
data[base + 8 + i + 2] = 0;
}
}
/* red */
for (j = 0; j < 8; j++) {
for (i = 0; i < 8; i++) {
base = 8 * 8 + j * 16;
data[base + i] = 0;
data[base + i + 1] = 0;
data[base + i + 2] = 255;
}
}
return gst_buffer_new_wrapped (data, size);
}
/* Test to make sure that we are using the same rgb format as opencv expects for
* template matching (BGR).
* Will use a blue 8x8 square as the template and as input a 16x16 frame divided
* evenly in 4 squares: Blue (top-left), Green (top-right), Red (bottom-left) and
* Black (bottom-right)
*
* https://bugzilla.gnome.org/show_bug.cgi?id=678485
*/
GST_START_TEST (test_match_blue_square)
{
GstElement *element;
GstPad *sinkpad, *srcpad;
GstCaps *caps =
gst_caps_from_string (CAPS_TMPL
", width=(int)16, height=(int)16, framerate=1/1");
GstBus *bus;
GstMessage *msg;
const GstStructure *structure;
gchar *path;
GstBuffer *buf;
guint x, y, width, height;
element = gst_check_setup_element ("templatematch");
srcpad = gst_check_setup_src_pad (element, &srctemplate);
sinkpad = gst_check_setup_sink_pad (element, &sinktemplate);
gst_pad_set_active (srcpad, TRUE);
gst_check_setup_events (srcpad, element, caps, GST_FORMAT_TIME);
gst_pad_set_active (sinkpad, TRUE);
bus = gst_bus_new ();
gst_element_set_bus (element, bus);
path = g_build_filename (GST_TEST_FILES_PATH, "blue-square.png", NULL);
g_object_set (element, "template", path, NULL);
g_free (path);
fail_unless (gst_element_set_state (element,
GST_STATE_PLAYING) != GST_STATE_CHANGE_FAILURE,
"could not set to playing");
buf = create_input_buffer ();
fail_unless (gst_pad_push (srcpad, buf) == GST_FLOW_OK);
/* make sure that the template match message was posted, detecting the
* blue area in the top left corner */
msg = gst_bus_pop_filtered (bus, GST_MESSAGE_ELEMENT);
fail_unless (msg != NULL);
fail_unless (GST_MESSAGE_SRC (msg) == GST_OBJECT_CAST (element));
structure = gst_message_get_structure (msg);
fail_unless (gst_structure_has_name (structure, "template_match"));
fail_unless (gst_structure_get_uint (structure, "x", &x));
fail_unless (gst_structure_get_uint (structure, "y", &y));
fail_unless (gst_structure_get_uint (structure, "width", &width));
fail_unless (gst_structure_get_uint (structure, "height", &height));
fail_unless (x == 0);
fail_unless (y == 0);
fail_unless (width == 8);
fail_unless (height == 8);
gst_message_unref (msg);
gst_element_set_state (element, GST_STATE_NULL);
gst_bus_set_flushing (bus, TRUE);
gst_object_unref (bus);
gst_caps_unref (caps);
gst_check_drop_buffers ();
gst_pad_set_active (srcpad, FALSE);
gst_pad_set_active (sinkpad, FALSE);
gst_object_unref (srcpad);
gst_object_unref (sinkpad);
gst_check_teardown_src_pad (element);
gst_check_teardown_sink_pad (element);
gst_check_teardown_element (element);
}
GST_END_TEST;
static Suite *
templatematch_suite (void)
{
Suite *s = suite_create ("templatematch");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
tcase_add_test (tc_chain, test_match_blue_square);
return s;
}
GST_CHECK_MAIN (templatematch);