/* * 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., 51 Franklin St, Fifth Floor, * Boston, MA 02110-1301, USA. */ /** * SECTION:element-opencvtextoverlay * * opencvtextoverlay renders the text on top of the video frames * * ## Example launch line * * |[ * gst-launch-1.0 videotestsrc ! videoconvert ! opencvtextoverlay text="Opencv Text Overlay " ! videoconvert ! xvimagesink * ]| */ #ifdef HAVE_CONFIG_H # include #endif #include "gsttextoverlay.h" GST_DEBUG_CATEGORY_STATIC (gst_opencv_text_overlay_debug); #define GST_CAT_DEFAULT gst_opencv_opencv_text_overlay_debug #define DEFAULT_PROP_TEXT "Opencv Text Overlay" #define DEFAULT_PROP_WIDTH 1 #define DEFAULT_PROP_HEIGHT 1 #define DEFAULT_PROP_XPOS 50 #define DEFAULT_PROP_YPOS 50 #define DEFAULT_PROP_THICKNESS 2 #define DEFAULT_PROP_COLOR 0 /* Filter signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; #define DEFAULT_WIDTH 1.0 #define DEFAULT_HEIGHT 1.0 enum { PROP_0, PROP_XPOS, PROP_YPOS, PROP_THICKNESS, PROP_COLOR_R, PROP_COLOR_G, PROP_COLOR_B, 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 (GST_VIDEO_CAPS_MAKE ("RGB")) ); static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("RGB")) ); G_DEFINE_TYPE_WITH_CODE (GstOpencvTextOverlay, gst_opencv_text_overlay, GST_TYPE_OPENCV_VIDEO_FILTER, GST_DEBUG_CATEGORY_INIT (gst_opencv_text_overlay_debug, "opencvtextoverlay", 0, "Template opencvtextoverlay"); ); GST_ELEMENT_REGISTER_DEFINE (opencvtextoverlay, "opencvtextoverlay", GST_RANK_NONE, GST_TYPE_OPENCV_TEXT_OVERLAY); static void gst_opencv_text_overlay_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec); static void gst_opencv_text_overlay_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec); static GstFlowReturn gst_opencv_text_overlay_transform_ip (GstOpencvVideoFilter * filter, GstBuffer * buf, cv::Mat img); /* Clean up */ static void gst_opencv_text_overlay_finalize (GObject * obj) { GstOpencvTextOverlay *filter = GST_OPENCV_TEXT_OVERLAY (obj); g_free (filter->textbuf); G_OBJECT_CLASS (gst_opencv_text_overlay_parent_class)->finalize (obj); } /* initialize the opencvtextoverlay's class */ static void gst_opencv_text_overlay_class_init (GstOpencvTextOverlayClass * klass) { GObjectClass *gobject_class; GstOpencvVideoFilterClass *gstopencvbasefilter_class; GstElementClass *element_class = GST_ELEMENT_CLASS (klass); gobject_class = (GObjectClass *) klass; gobject_class->finalize = GST_DEBUG_FUNCPTR (gst_opencv_text_overlay_finalize); gstopencvbasefilter_class = (GstOpencvVideoFilterClass *) klass; gstopencvbasefilter_class->cv_trans_ip_func = gst_opencv_text_overlay_transform_ip; gobject_class->set_property = gst_opencv_text_overlay_set_property; gobject_class->get_property = gst_opencv_text_overlay_get_property; g_object_class_install_property (gobject_class, PROP_TEXT, g_param_spec_string ("text", "text", "Text to be display.", DEFAULT_PROP_TEXT, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_XPOS, g_param_spec_int ("xpos", "horizontal position", "Sets the Horizontal position", 0, G_MAXINT, DEFAULT_PROP_XPOS, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_YPOS, g_param_spec_int ("ypos", "vertical position", "Sets the Vertical position", 0, G_MAXINT, DEFAULT_PROP_YPOS, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_THICKNESS, g_param_spec_int ("thickness", "font thickness", "Sets the Thickness of Font", 0, G_MAXINT, DEFAULT_PROP_THICKNESS, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_COLOR_R, g_param_spec_int ("colorR", "color -Red ", "Sets the color -R", 0, 255, DEFAULT_PROP_COLOR, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_COLOR_G, g_param_spec_int ("colorG", "color -Green", "Sets the color -G", 0, 255, DEFAULT_PROP_COLOR, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); g_object_class_install_property (gobject_class, PROP_COLOR_B, g_param_spec_int ("colorB", "color -Blue", "Sets the color -B", 0, 255, DEFAULT_PROP_COLOR, (GParamFlags) (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, (GParamFlags) (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, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); gst_element_class_set_static_metadata (element_class, "opencvtextoverlay", "Filter/Effect/Video", "Write text on the top of video", "sreerenj"); gst_element_class_add_static_pad_template (element_class, &src_factory); gst_element_class_add_static_pad_template (element_class, &sink_factory); } /* initialize the new element * instantiate pads and add them to element * set pad callback functions * initialize instance structure */ static void gst_opencv_text_overlay_init (GstOpencvTextOverlay * filter) { filter->textbuf = g_strdup (DEFAULT_PROP_TEXT); filter->width = DEFAULT_PROP_WIDTH; filter->height = DEFAULT_PROP_HEIGHT; filter->xpos = DEFAULT_PROP_XPOS; filter->ypos = DEFAULT_PROP_YPOS; filter->thickness = DEFAULT_PROP_THICKNESS; filter->colorR = DEFAULT_PROP_COLOR; filter->colorG = DEFAULT_PROP_COLOR; filter->colorB = DEFAULT_PROP_COLOR; gst_opencv_video_filter_set_in_place (GST_OPENCV_VIDEO_FILTER_CAST (filter), TRUE); } static void gst_opencv_text_overlay_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstOpencvTextOverlay *filter = GST_OPENCV_TEXT_OVERLAY (object); switch (prop_id) { case PROP_TEXT: g_free (filter->textbuf); filter->textbuf = g_value_dup_string (value); break; case PROP_XPOS: filter->xpos = g_value_get_int (value); break; case PROP_YPOS: filter->ypos = g_value_get_int (value); break; case PROP_THICKNESS: filter->thickness = g_value_get_int (value); break; case PROP_COLOR_R: filter->colorR = g_value_get_int (value); break; case PROP_COLOR_G: filter->colorG = g_value_get_int (value); break; case PROP_COLOR_B: filter->colorB = g_value_get_int (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_opencv_text_overlay_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstOpencvTextOverlay *filter = GST_OPENCV_TEXT_OVERLAY (object); switch (prop_id) { case PROP_TEXT: g_value_set_string (value, filter->textbuf); break; case PROP_XPOS: g_value_set_int (value, filter->xpos); break; case PROP_YPOS: g_value_set_int (value, filter->ypos); break; case PROP_THICKNESS: g_value_set_int (value, filter->thickness); break; case PROP_COLOR_R: g_value_set_int (value, filter->colorR); break; case PROP_COLOR_G: g_value_set_int (value, filter->colorG); break; case PROP_COLOR_B: g_value_set_int (value, filter->colorB); 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; } } /* chain function * this function does the actual processing */ static GstFlowReturn gst_opencv_text_overlay_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf, cv::Mat img) { GstOpencvTextOverlay *filter = GST_OPENCV_TEXT_OVERLAY (base); cv::putText (img, filter->textbuf, cv::Point (filter->xpos, filter->ypos), cv::FONT_HERSHEY_SIMPLEX, (filter->width + filter->height) * 0.5, cv::Scalar (filter->colorR, filter->colorG, filter->colorB), filter->thickness); return GST_FLOW_OK; }