diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am index 90677f7755..ab73bd21ed 100644 --- a/tests/check/Makefile.am +++ b/tests/check/Makefile.am @@ -136,6 +136,12 @@ else check_zbar = endif +if USE_OPENCV +check_opencv = elements/templatematch +else +check_opencv = +endif + if USE_OPUS check_opus = elements/opus else @@ -214,6 +220,7 @@ check_PROGRAMS = \ $(check_ofa) \ $(check_timidity) \ $(check_kate) \ + $(check_opencv) \ $(check_opus) \ $(check_curl) \ $(check_shm) \ diff --git a/tests/check/elements/templatematch.c b/tests/check/elements/templatematch.c new file mode 100644 index 0000000000..cce165faf4 --- /dev/null +++ b/tests/check/elements/templatematch.c @@ -0,0 +1,169 @@ +/* + * GStreamer + * + * unit test for templatematch + * + * Copyright (C) 2014 Samsung Electronics. All rights reserved. + * Author: Thiago Santos + * + * 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 + +#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; i < 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; i < 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"); + 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_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); diff --git a/tests/files/Makefile.am b/tests/files/Makefile.am index 710d286530..2807612ce4 100644 --- a/tests/files/Makefile.am +++ b/tests/files/Makefile.am @@ -1,5 +1,6 @@ EXTRA_DIST = \ barcode.png \ + blue-square.png \ cbr_stream.mp3 \ s16be-id3v2.aiff \ stream.mp2 \