mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-03 15:06:34 +00:00
tests: templatematch: add test to check that we use the correct rgb format
templatematch should use the same RGB format that opencv uses (BGR), make sure we keep it that way with this test. https://bugzilla.gnome.org/show_bug.cgi?id=678485
This commit is contained in:
parent
f5992a6cdd
commit
06243b7eee
3 changed files with 177 additions and 0 deletions
|
@ -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) \
|
||||
|
|
169
tests/check/elements/templatematch.c
Normal file
169
tests/check/elements/templatematch.c
Normal file
|
@ -0,0 +1,169 @@
|
|||
/*
|
||||
* 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; 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);
|
|
@ -1,5 +1,6 @@
|
|||
EXTRA_DIST = \
|
||||
barcode.png \
|
||||
blue-square.png \
|
||||
cbr_stream.mp3 \
|
||||
s16be-id3v2.aiff \
|
||||
stream.mp2 \
|
||||
|
|
Loading…
Reference in a new issue