From fa8210fe94b83bb9db0e7934b60160f846c713a1 Mon Sep 17 00:00:00 2001 From: Sreerenj Balachandran Date: Sun, 28 Feb 2010 14:32:00 +0000 Subject: [PATCH] Add simple text overlay plugin --- ext/opencv/Makefile.am | 11 +- ext/opencv/gstopencv.c | 4 + ext/opencv/textwrite/Makefile.am | 15 ++ ext/opencv/textwrite/gsttextwrite.c | 329 ++++++++++++++++++++++++++++ ext/opencv/textwrite/gsttextwrite.h | 98 +++++++++ 5 files changed, 453 insertions(+), 4 deletions(-) create mode 100644 ext/opencv/textwrite/Makefile.am create mode 100644 ext/opencv/textwrite/gsttextwrite.c create mode 100644 ext/opencv/textwrite/gsttextwrite.h diff --git a/ext/opencv/Makefile.am b/ext/opencv/Makefile.am index dbfd66b264..4857918999 100644 --- a/ext/opencv/Makefile.am +++ b/ext/opencv/Makefile.am @@ -1,4 +1,4 @@ -SUBDIRS = edgedetect faceblur facedetect pyramidsegment +SUBDIRS = edgedetect faceblur facedetect pyramidsegment templatematch textwrite # plugindir is set in configure @@ -14,14 +14,16 @@ libgstopencv_la_CFLAGS = $(GST_CFLAGS) $(OPENCV_CFLAGS) \ -I${top_srcdir}/ext/opencv/faceblur \ -I${top_srcdir}/ext/opencv/facedetect \ -I${top_srcdir}/ext/opencv/pyramidsegment \ - -I${top_srcdir}/ext/opencv/templatematch + -I${top_srcdir}/ext/opencv/templatematch \ + -I${top_srcdir}/ext/opencv/textwrite libgstopencv_la_LIBADD = $(GST_LIBS) $(OPENCV_LIBS) \ $(top_builddir)/ext/opencv/edgedetect/libgstedgedetect.la \ $(top_builddir)/ext/opencv/faceblur/libgstfaceblur.la \ $(top_builddir)/ext/opencv/facedetect/libgstfacedetect.la \ $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la \ - $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la + $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la \ + $(top_builddir)/ext/opencv/textwrite/libgsttextwrite.la libgstopencv_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) @@ -30,7 +32,8 @@ libgstopencv_la_DEPENDENCIES = \ $(top_builddir)/ext/opencv/faceblur/libgstfaceblur.la \ $(top_builddir)/ext/opencv/facedetect/libgstfacedetect.la \ $(top_builddir)/ext/opencv/pyramidsegment/libgstpyramidsegment.la \ - $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la + $(top_builddir)/ext/opencv/templatematch/libgsttemplatematch.la \ + $(top_builddir)/ext/opencv/textwrite/libgsttextwrite.la # headers we need but don't want installed noinst_HEADERS = diff --git a/ext/opencv/gstopencv.c b/ext/opencv/gstopencv.c index f08fdb1c1c..120efd8478 100644 --- a/ext/opencv/gstopencv.c +++ b/ext/opencv/gstopencv.c @@ -28,6 +28,7 @@ #include "gstfacedetect.h" #include "gstpyramidsegment.h" #include "gsttemplatematch.h" +#include "gsttextwrite.h" static gboolean plugin_init (GstPlugin * plugin) @@ -48,6 +49,9 @@ plugin_init (GstPlugin * plugin) if (!gst_templatematch_plugin_init (plugin)) return FALSE; + if (!gst_textwrite_plugin_init (plugin)) + return FALSE; + return TRUE; } diff --git a/ext/opencv/textwrite/Makefile.am b/ext/opencv/textwrite/Makefile.am new file mode 100644 index 0000000000..677224ca61 --- /dev/null +++ b/ext/opencv/textwrite/Makefile.am @@ -0,0 +1,15 @@ +# plugindir is set in configure + +noinst_LTLIBRARIES = libgsttextwrite.la + +# sources used to compile this plug-in +libgsttextwrite_la_SOURCES = gsttextwrite.c + +# flags used to compile this textwrite +# add other _CFLAGS and _LIBS as needed +libgsttextwrite_la_CFLAGS = $(GST_CFLAGS) $(OPENCV_CFLAGS) +libgsttextwrite_la_LIBADD = $(GST_LIBS) $(OPENCV_LIBS) +libgsttextwrite_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) + +# headers we need but don't want installed +noinst_HEADERS = gsttextwrite.h diff --git a/ext/opencv/textwrite/gsttextwrite.c b/ext/opencv/textwrite/gsttextwrite.c new file mode 100644 index 0000000000..3369e92ef0 --- /dev/null +++ b/ext/opencv/textwrite/gsttextwrite.c @@ -0,0 +1,329 @@ +/* + * GStreamer + * Copyright (C) 2005 Thomas Vander Stichele + * Copyright (C) 2005 Ronald S. Bultje + * Copyright (C) 2010 Sreerenj Balachandran + * + * 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-textwrite + * + * FIXME:Describe textwrite here. + * + * + * Example launch line + * |[ + * gst-launch -v -m fakesrc ! textwrite ! fakesink silent=TRUE + * ]| + * + */ + +#ifdef HAVE_CONFIG_H +# include +#endif + +#include + +#include "gsttextwrite.h" + +GST_DEBUG_CATEGORY_STATIC (gst_textwrite_debug); +#define GST_CAT_DEFAULT gst_textwrite_debug +#define DEFAULT_PROP_TEXT "" +#define DEFAULT_PROP_WIDTH 1 +#define DEFAULT_PROP_HEIGHT 1 + +/* Filter signals and args */ +enum +{ + /* FILL ME */ + LAST_SIGNAL +}; +#define DEFAULT_WIDTH 1.0 +#define DEFAULT_HEIGHT 1.0 +enum +{ + PROP_0, + PROP_TEXT, + PROP_HEIGHT, + PROP_WIDTH +}; + +/* the capabilities of the inputs and outputs. + * + * describe the real formats here. + */ +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 ("ANY") + ); + +GST_BOILERPLATE (Gsttextwrite, gst_textwrite, GstElement,GST_TYPE_ELEMENT); + +static void gst_textwrite_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_textwrite_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); + +static gboolean gst_textwrite_set_caps (GstPad * pad, GstCaps * caps); +static GstFlowReturn gst_textwrite_chain (GstPad * pad, GstBuffer * buf); + + + +/* Clean up */ +static void +gst_textwrite_finalize (GObject * obj) +{ + Gsttextwrite *filter = GST_textwrite (obj); + + if (filter->cvImage) { + cvReleaseImage (&filter->cvImage); + } + + G_OBJECT_CLASS (parent_class)->finalize (obj); +} + + + +/* GObject vmethod implementations */ + +static void +gst_textwrite_base_init (gpointer gclass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (gclass); + + gst_element_class_set_details_simple(element_class, + "textwrite", + "Filter/Effect/Video", + "Performs text writing to the video", + "sreerenj"); + + 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 textwrite's class */ +static void +gst_textwrite_class_init (GsttextwriteClass * klass) +{ + GObjectClass *gobject_class; + GstElementClass *gstelement_class; + + gobject_class = (GObjectClass *) klass; + gstelement_class = (GstElementClass *) klass; + + parent_class = g_type_class_peek_parent (klass); + gobject_class->finalize = GST_DEBUG_FUNCPTR (gst_textwrite_finalize); + + gobject_class->set_property = gst_textwrite_set_property; + gobject_class->get_property = gst_textwrite_get_property; + + + g_object_class_install_property (gobject_class, PROP_TEXT, + g_param_spec_string ("text", "text", + "Text to be display.", DEFAULT_PROP_TEXT, + G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); + + g_object_class_install_property (gobject_class, PROP_HEIGHT, + g_param_spec_double ("height", "Height", + "Sets the height of fonts",1.0,5.0, + DEFAULT_HEIGHT, G_PARAM_READWRITE|G_PARAM_STATIC_STRINGS)); + + g_object_class_install_property (gobject_class, PROP_WIDTH, + g_param_spec_double ("width", "Width", + "Sets the width of fonts",1.0,5.0, + DEFAULT_WIDTH, G_PARAM_READWRITE| G_PARAM_STATIC_STRINGS)); + +} + +/* initialize the new element + * instantiate pads and add them to element + * set pad calback functions + * initialize instance structure + */ +static void +gst_textwrite_init (Gsttextwrite * filter, + GsttextwriteClass * gclass) +{ + filter->sinkpad = gst_pad_new_from_static_template (&sink_factory, "sink"); + gst_pad_set_setcaps_function (filter->sinkpad, + GST_DEBUG_FUNCPTR(gst_textwrite_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_textwrite_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->textbuf = g_strdup (DEFAULT_PROP_TEXT); + filter->width=DEFAULT_PROP_WIDTH; + filter->height=DEFAULT_PROP_HEIGHT; + +} + +static void +gst_textwrite_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + Gsttextwrite *filter = GST_textwrite (object); + + switch (prop_id) { + case PROP_TEXT: + g_free (filter->textbuf); + filter->textbuf = g_value_dup_string (value); + break; + case PROP_HEIGHT: + filter->height = g_value_get_double(value); + break; + case PROP_WIDTH: + filter->width = g_value_get_double(value); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_textwrite_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) +{ + Gsttextwrite *filter = GST_textwrite (object); + + switch (prop_id) { + case PROP_TEXT: + g_value_set_string (value, filter->textbuf); + break; + case PROP_HEIGHT: + g_value_set_double (value, filter->height); + break; + case PROP_WIDTH: + g_value_set_double (value, filter->width); + 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_textwrite_set_caps (GstPad * pad, GstCaps * caps) +{ + Gsttextwrite *filter; + GstPad *otherpad; + + gint width, height; + GstStructure *structure; + + filter = GST_textwrite (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 = cvCreateImage (cvSize (width, height), IPL_DEPTH_8U, 3); + filter->cvStorage = cvCreateMemStorage (0); + + otherpad = (pad == filter->srcpad) ? filter->sinkpad : filter->srcpad; + gst_object_unref (filter); + + return gst_pad_set_caps (otherpad, caps); +} + +/* chain function + * this function does the actual processing + */ +static GstFlowReturn +gst_textwrite_chain (GstPad * pad, GstBuffer * buf) +{ + Gsttextwrite *filter; + + filter = GST_textwrite (GST_OBJECT_PARENT (pad)); + + filter->cvImage->imageData = (char *) GST_BUFFER_DATA (buf); + int lineWidth=1; + cvInitFont(&(filter->font),CV_FONT_VECTOR0, filter->width,filter->height,0,lineWidth,0); + + + cvPutText (filter->cvImage,filter->textbuf,cvPoint(100,100), &(filter->font), cvScalar(165,14,14,0)); + + + gst_buffer_set_data (buf, filter->cvImage->imageData,filter->cvImage->imageSize); + + + return gst_pad_push (filter->srcpad, buf); +} + + +/* entry point to initialize the plug-in + * initialize the plug-in itself + * register the element factories and other features + */ +gboolean +gst_textwrite_plugin_init (GstPlugin * plugin) +{ + /* debug category for fltering log messages + * + * exchange the string 'Template textwrite' with your description + */ + GST_DEBUG_CATEGORY_INIT (gst_textwrite_debug, "textwrite", + 0, "Template textwrite"); + + return gst_element_register (plugin, "textwrite", GST_RANK_NONE, + GST_TYPE_textwrite); +} + + diff --git a/ext/opencv/textwrite/gsttextwrite.h b/ext/opencv/textwrite/gsttextwrite.h new file mode 100644 index 0000000000..8253e91fb3 --- /dev/null +++ b/ext/opencv/textwrite/gsttextwrite.h @@ -0,0 +1,98 @@ +/* + * GStreamer + * Copyright (C) 2005 Thomas Vander Stichele + * Copyright (C) 2005 Ronald S. Bultje + * Copyright (C) 2010 root <>m + * + * 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_textwrite_H__ +#define __GST_textwrite_H__ + +#include +//sreechage +#include +#include +#include +G_BEGIN_DECLS + +/* #defines don't like whitespacey bits */ +#define GST_TYPE_textwrite \ + (gst_textwrite_get_type()) +#define GST_textwrite(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_textwrite,Gsttextwrite)) +#define GST_textwrite_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_textwrite,GsttextwriteClass)) +#define GST_IS_textwrite(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_textwrite)) +#define GST_IS_textwrite_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_textwrite)) + +typedef struct _Gsttextwrite Gsttextwrite; +typedef struct _GsttextwriteClass GsttextwriteClass; + +struct _Gsttextwrite +{ + GstElement element; + + GstPad *sinkpad, *srcpad; + + + IplImage *cvImage; + CvMemStorage *cvStorage; + CvFont font; + + gdouble height; + gdouble width; + gchar *textbuf; + +}; + +struct _GsttextwriteClass +{ + GstElementClass parent_class; +}; + +GType gst_textwrite_get_type (void); +gboolean gst_textwrite_plugin_init (GstPlugin * plugin); + +G_END_DECLS + +#endif /* __GST_textwrite_H__ */