diff --git a/ext/opencv/Makefile.am b/ext/opencv/Makefile.am index 712582758d..e4173db25f 100644 --- a/ext/opencv/Makefile.am +++ b/ext/opencv/Makefile.am @@ -1,4 +1,4 @@ -SUBDIRS = edgedetect facedetect pyramidsegment +SUBDIRS = edgedetect facedetect pyramidsegment templatematch # plugindir is set in configure @@ -12,19 +12,22 @@ libgstopencv_la_SOURCES = gstopencv.c libgstopencv_la_CFLAGS = $(GST_CFLAGS) $(OPENCV_CFLAGS) \ -I${top_srcdir}/ext/opencv/edgedetect \ -I${top_srcdir}/ext/opencv/facedetect \ - -I${top_srcdir}/ext/opencv/pyramidsegment + -I${top_srcdir}/ext/opencv/pyramidsegment \ + -I${top_srcdir}/ext/opencv/templatematch libgstopencv_la_LIBADD = $(GST_LIBS) $(OPENCV_LIBS) \ $(top_builddir)/ext/opencv/edgedetect/libgstedgedetect.la \ $(top_builddir)/ext/opencv/facedetect/libgstfacedetect.la \ - $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la + $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la \ + $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la libgstopencv_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) libgstopencv_la_DEPENDENCIES = \ $(top_builddir)/ext/opencv/edgedetect/libgstedgedetect.la \ $(top_builddir)/ext/opencv/facedetect/libgstfacedetect.la \ - $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la + $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la \ + $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la # headers we need but don't want installed noinst_HEADERS = diff --git a/ext/opencv/gstopencv.c b/ext/opencv/gstopencv.c index 8c0364791b..6cf2676ff3 100644 --- a/ext/opencv/gstopencv.c +++ b/ext/opencv/gstopencv.c @@ -26,6 +26,7 @@ #include "gstedgedetect.h" #include "gstfacedetect.h" #include "gstpyramidsegment.h" +#include "gsttemplatematch.h" static gboolean plugin_init (GstPlugin * plugin) @@ -40,6 +41,9 @@ plugin_init (GstPlugin * plugin) if (!gst_pyramidsegment_plugin_init (plugin)) return FALSE; + if (!gst_templatematch_plugin_init (plugin)) + return FALSE; + return TRUE; } diff --git a/ext/opencv/templatematch/Makefile.am b/ext/opencv/templatematch/Makefile.am new file mode 100644 index 0000000000..af807038b6 --- /dev/null +++ b/ext/opencv/templatematch/Makefile.am @@ -0,0 +1,15 @@ +# plugindir is set in configure + +noinst_LTLIBRARIES = libgsttemplatematch.la + +# sources used to compile this plug-in +libgsttemplatematch_la_SOURCES = gsttemplatematch.c + +# flags used to compile this templatematch +# add other _CFLAGS and _LIBS as needed +libgsttemplatematch_la_CFLAGS = $(GST_CFLAGS) $(OPENCV_CFLAGS) +libgsttemplatematch_la_LIBADD = $(GST_LIBS) $(OPENCV_LIBS) +libgsttemplatematch_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) + +# headers we need but don't want installed +noinst_HEADERS = gsttemplatematch.h diff --git a/ext/opencv/templatematch/gsttemplatematch.c b/ext/opencv/templatematch/gsttemplatematch.c new file mode 100644 index 0000000000..b9f5779f9f --- /dev/null +++ b/ext/opencv/templatematch/gsttemplatematch.c @@ -0,0 +1,379 @@ +/* + * GStreamer + * Copyright (C) 2005 Thomas Vander Stichele + * Copyright (C) 2005 Ronald S. Bultje + * Copyright (C) 2008 Michael Sheldon + * Copyright (C) 2009 Noam Lewis + * + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. + * + * Alternatively, the contents of this file may be used under the + * GNU Lesser General Public License Version 2.1 (the "LGPL"), in + * which case the following provisions apply instead of the ones + * mentioned above: + * + * 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. + */ + +/** + * SECTION:element-templatematch + * + * FIXME:Describe templatematch here. + * + * + * Example launch line + * |[ + * gst-launch-0.10 videotestsrc ! decodebin ! ffmpegcolorspace ! templatematch ! ffmpegcolorspace ! xvimagesink + * ]| + * + */ + +#ifdef HAVE_CONFIG_H +# include +#endif + +#include + +#include "gsttemplatematch.h" + +GST_DEBUG_CATEGORY_STATIC (gst_templatematch_debug); +#define GST_CAT_DEFAULT gst_templatematch_debug + +#define DEFAULT_TEMPLATE NULL + +/* Filter signals and args */ +enum +{ + /* FILL ME */ + LAST_SIGNAL +}; + +enum +{ + PROP_0, + PROP_DISPLAY, + PROP_TEMPLATE +}; + +/* the capabilities of the inputs and outputs. + */ +static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ( + "video/x-raw-rgb" + ) + ); + +static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ( + "video/x-raw-rgb" + ) + ); + +GST_BOILERPLATE (GstTemplateMatch, gst_templatematch, GstElement, + GST_TYPE_ELEMENT); + +static void gst_templatematch_finalize (GObject * object); +static void gst_templatematch_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_templatematch_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); + +static gboolean gst_templatematch_set_caps (GstPad * pad, GstCaps * caps); +static GstFlowReturn gst_templatematch_chain (GstPad * pad, GstBuffer * buf); + +static void gst_templatematch_load_template (GObject * object); +static void gst_templatematch_match(IplImage *input, IplImage *template, + IplImage *dist_image, double *best_res, CvPoint *best_pos, int method); + +/* GObject vmethod implementations */ + +static void +gst_templatematch_base_init (gpointer gclass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (gclass); + + gst_element_class_set_details_simple(element_class, + "templatematch", + "Filter/Effect/Video", + "Performs template matching on videos and images, providing detected positions via bus messages", + "Noam Lewis "); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&src_factory)); + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&sink_factory)); +} + +/* initialize the templatematch's class */ +static void +gst_templatematch_class_init (GstTemplateMatchClass * klass) +{ + GObjectClass *gobject_class; + GstElementClass *gstelement_class; + + gobject_class = (GObjectClass *) klass; + gstelement_class = (GstElementClass *) klass; + + gobject_class->finalize = gst_templatematch_finalize; + gobject_class->set_property = gst_templatematch_set_property; + gobject_class->get_property = gst_templatematch_get_property; + + g_object_class_install_property (gobject_class, PROP_DISPLAY, + g_param_spec_boolean ("display", "Display", "Sets whether the detected template should be highlighted in the output", + TRUE, G_PARAM_READWRITE)); + g_object_class_install_property (gobject_class, PROP_TEMPLATE, + g_param_spec_string ("template", "Template", "Filename of template image", + DEFAULT_TEMPLATE, G_PARAM_READWRITE)); +} + +/* initialize the new element + * instantiate pads and add them to element + * set pad calback functions + * initialize instance structure + */ +static void +gst_templatematch_init (GstTemplateMatch * filter, + GstTemplateMatchClass * gclass) +{ + filter->sinkpad = gst_pad_new_from_static_template (&sink_factory, "sink"); + gst_pad_set_setcaps_function (filter->sinkpad, + GST_DEBUG_FUNCPTR(gst_templatematch_set_caps)); + gst_pad_set_getcaps_function (filter->sinkpad, + GST_DEBUG_FUNCPTR(gst_pad_proxy_getcaps)); + gst_pad_set_chain_function (filter->sinkpad, + GST_DEBUG_FUNCPTR(gst_templatematch_chain)); + + filter->srcpad = gst_pad_new_from_static_template (&src_factory, "src"); + gst_pad_set_getcaps_function (filter->srcpad, + GST_DEBUG_FUNCPTR(gst_pad_proxy_getcaps)); + + gst_element_add_pad (GST_ELEMENT (filter), filter->sinkpad); + gst_element_add_pad (GST_ELEMENT (filter), filter->srcpad); + filter->template = DEFAULT_TEMPLATE; + filter->display = TRUE; + filter->cvTemplateImage = NULL; + filter->cvDistImage = NULL; + filter->cvImage = NULL; + gst_templatematch_load_template(filter); +} + +static void +gst_templatematch_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + GstTemplateMatch *filter = GST_TEMPLATEMATCH (object); + + switch (prop_id) { + case PROP_TEMPLATE: + filter->template = g_value_get_string (value); + gst_templatematch_load_template(filter); + break; + case PROP_DISPLAY: + filter->display = g_value_get_boolean (value); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_templatematch_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) +{ + GstTemplateMatch *filter = GST_TEMPLATEMATCH (object); + + switch (prop_id) { + case PROP_TEMPLATE: + g_value_set_string (value, filter->template); + break; + case PROP_DISPLAY: + g_value_set_boolean (value, filter->display); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +/* GstElement vmethod implementations */ + +/* this function handles the link with other elements */ +static gboolean +gst_templatematch_set_caps (GstPad * pad, GstCaps * caps) +{ + GstTemplateMatch *filter; + GstPad *otherpad; + gint width, height; + GstStructure *structure; + + filter = GST_TEMPLATEMATCH (gst_pad_get_parent (pad)); + structure = gst_caps_get_structure (caps, 0); + gst_structure_get_int (structure, "width", &width); + gst_structure_get_int (structure, "height", &height); + + filter->cvImage = cvCreateImageHeader(cvSize(width, height), IPL_DEPTH_8U, 3); + + otherpad = (pad == filter->srcpad) ? filter->sinkpad : filter->srcpad; + gst_object_unref (filter); + + return gst_pad_set_caps (otherpad, caps); +} + +static void gst_templatematch_finalize (GObject * object) +{ + GstTemplateMatch *filter; + filter = GST_TEMPLATEMATCH (object); + + if (filter->cvImage) { + cvReleaseImageHeader(&filter->cvImage); + } + if (filter->cvDistImage) { + cvReleaseImage(&filter->cvDistImage); + } + if (filter->cvTemplateImage) { + cvReleaseImage(&filter->cvTemplateImage); + } +} + +/* chain function + * this function does the actual processing + */ +static GstFlowReturn +gst_templatematch_chain (GstPad * pad, GstBuffer * buf) +{ + GstTemplateMatch *filter; + CvPoint best_pos; + double best_res; + int i; + + filter = GST_TEMPLATEMATCH (GST_OBJECT_PARENT (pad)); + buf = gst_buffer_make_writable(buf); + if ((!filter) || (!buf)) { + return GST_FLOW_OK; + } + filter->cvImage->imageData = (char *) GST_BUFFER_DATA (buf); + + + if (!filter->cvDistImage) { + filter->cvDistImage = cvCreateImage( cvSize(filter->cvImage->width - filter->cvTemplateImage->width + 1, + filter->cvImage->height - filter->cvTemplateImage->height + 1), + IPL_DEPTH_32F, 1); + if (!filter->cvDistImage) { + GST_WARNING ("Couldn't create dist image."); + } + } + if (filter->cvTemplateImage) { + gst_templatematch_match(filter->cvImage, filter->cvTemplateImage, + filter->cvDistImage, &best_res, &best_pos, CV_TM_CCORR); + + GstStructure *s = gst_structure_new ("template_match", + "x", G_TYPE_UINT, best_pos.x, + "y", G_TYPE_UINT, best_pos.y, + "width", G_TYPE_UINT, filter->cvTemplateImage->width, + "height", G_TYPE_UINT, filter->cvTemplateImage->height, + "result", G_TYPE_DOUBLE, best_res, + NULL); + + GstMessage *m = gst_message_new_element (GST_OBJECT (filter), s); + gst_element_post_message (GST_ELEMENT (filter), m); + + if (filter->display) { + CvPoint corner = best_pos; + corner.x += filter->cvTemplateImage->width; + corner.y += filter->cvTemplateImage->height; + cvRectangle(filter->cvImage, best_pos, corner, CV_RGB(255, 32, 32), 3, 8, 0); + } + + } + + + gst_buffer_set_data(buf, filter->cvImage->imageData, filter->cvImage->imageSize); + + return gst_pad_push (filter->srcpad, buf); +} + + + +static void gst_templatematch_match(IplImage *input, IplImage *template, + IplImage *dist_image, double *best_res, CvPoint *best_pos, int method) +{ + double dist_min = 0, dist_max = 0; + CvPoint min_pos, max_pos; + cvMatchTemplate(input, template, dist_image, method); + cvMinMaxLoc(dist_image, &dist_min, &dist_max, &min_pos, &max_pos, NULL); + if ((CV_TM_SQDIFF_NORMED == method) || (CV_TM_SQDIFF == method)) + { + *best_res = dist_min; + *best_pos = min_pos; + if (CV_TM_SQDIFF_NORMED == method) { + *best_res = 1-*best_res; + } + } + else { + *best_res = dist_max; + *best_pos = max_pos; + } +} + + +static void gst_templatematch_load_template(GObject * object) { + GstTemplateMatch *filter = GST_TEMPLATEMATCH (object); + + if (filter->template) { + filter->cvTemplateImage = cvLoadImage(filter->template, CV_LOAD_IMAGE_COLOR); + if (!filter->cvTemplateImage) { + GST_WARNING ("Couldn't load template image: %s.", filter->template); + } + } +} + + +/* entry point to initialize the plug-in + * initialize the plug-in itself + * register the element factories and other features + */ +gboolean gst_templatematch_plugin_init (GstPlugin * templatematch) +{ + /* debug category for fltering log messages */ + GST_DEBUG_CATEGORY_INIT (gst_templatematch_debug, "templatematch", + 0, "Performs template matching on videos and images, providing detected positions via bus messages"); + + return gst_element_register (templatematch, "templatematch", GST_RANK_NONE, + GST_TYPE_TEMPLATEMATCH); +} + + diff --git a/ext/opencv/templatematch/gsttemplatematch.h b/ext/opencv/templatematch/gsttemplatematch.h new file mode 100644 index 0000000000..cb2d1661b9 --- /dev/null +++ b/ext/opencv/templatematch/gsttemplatematch.h @@ -0,0 +1,94 @@ +/* + * GStreamer + * Copyright (C) 2005 Thomas Vander Stichele + * Copyright (C) 2005 Ronald S. Bultje + * Copyright (C) 2008 Michael Sheldon + * + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. + * + * Alternatively, the contents of this file may be used under the + * GNU Lesser General Public License Version 2.1 (the "LGPL"), in + * which case the following provisions apply instead of the ones + * mentioned above: + * + * 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. + */ + +#ifndef __GST_TEMPLATEMATCH_H__ +#define __GST_TEMPLATEMATCH_H__ + +#include +#include +#include + +G_BEGIN_DECLS + +/* #defines don't like whitespacey bits */ +#define GST_TYPE_TEMPLATEMATCH \ + (gst_templatematch_get_type()) +#define GST_TEMPLATEMATCH(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_TEMPLATEMATCH,GstTemplateMatch)) +#define GST_TEMPLATEMATCH_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_TEMPLATEMATCH,GstTemplateMatchClass)) +#define GST_IS_TEMPLATEMATCH(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_TEMPLATEMATCH)) +#define GST_IS_TEMPLATEMATCH_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_TEMPLATEMATCH)) + +typedef struct _GstTemplateMatch GstTemplateMatch; +typedef struct _GstTemplateMatchClass GstTemplateMatchClass; + +struct _GstTemplateMatch +{ + GstElement element; + + GstPad *sinkpad, *srcpad; + + gboolean display; + + gchar *template; + + IplImage *cvImage, *cvGray, *cvTemplateImage, *cvDistImage; +}; + +struct _GstTemplateMatchClass +{ + GstElementClass parent_class; +}; + +GType gst_templatematch_get_type (void); + +gboolean gst_templatematch_plugin_init (GstPlugin * templatematch); + +G_END_DECLS + +#endif /* __GST_TEMPLATEMATCH_H__ */