1 /*
2  * GStreamer
3  * Copyright (C) 2010 Thiago Santos <thiago.sousa.santos@collabora.co.uk>
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in
13  * all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21  * DEALINGS IN THE SOFTWARE.
22  *
23  * Alternatively, the contents of this file may be used under the
24  * GNU Lesser General Public License Version 2.1 (the "LGPL"), in
25  * which case the following provisions apply instead of the ones
26  * mentioned above:
27  *
28  * This library is free software; you can redistribute it and/or
29  * modify it under the terms of the GNU Library General Public
30  * License as published by the Free Software Foundation; either
31  * version 2 of the License, or (at your option) any later version.
32  *
33  * This library is distributed in the hope that it will be useful,
34  * but WITHOUT ANY WARRANTY; without even the implied warranty of
35  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
36  * Library General Public License for more details.
37  *
38  * You should have received a copy of the GNU Library General Public
39  * License along with this library; if not, write to the
40  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
41  * Boston, MA 02110-1301, USA.
42  */
43 
44 /**
45  * SECTION:element-cverode
46  *
47  * Erodes the image with the cvErode OpenCV function.
48  *
49  * <refsect2>
50  * <title>Example launch line</title>
51  * |[
52  * gst-launch-1.0 videotestsrc ! cverode ! videoconvert ! autovideosink
53  * ]|
54  * </refsect2>
55  */
56 
57 #ifdef HAVE_CONFIG_H
58 #  include <config.h>
59 #endif
60 
61 #include "gstcverode.h"
62 #include <opencv2/imgproc.hpp>
63 
64 
65 GST_DEBUG_CATEGORY_STATIC (gst_cv_erode_debug);
66 #define GST_CAT_DEFAULT gst_cv_erode_debug
67 
68 G_DEFINE_TYPE (GstCvErode, gst_cv_erode, GST_TYPE_CV_DILATE_ERODE);
69 
70 static GstFlowReturn gst_cv_erode_transform_ip (GstOpencvVideoFilter *
71     filter, GstBuffer * buf, cv::Mat img);
72 
73 /* initialize the cverode's class */
74 static void
gst_cv_erode_class_init(GstCvErodeClass * klass)75 gst_cv_erode_class_init (GstCvErodeClass * klass)
76 {
77   GstOpencvVideoFilterClass *gstopencvbasefilter_class;
78   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
79 
80   gstopencvbasefilter_class = (GstOpencvVideoFilterClass *) klass;
81 
82   gstopencvbasefilter_class->cv_trans_ip_func = gst_cv_erode_transform_ip;
83   gst_element_class_set_static_metadata (element_class,
84       "cverode",
85       "Transform/Effect/Video",
86       "Applies cvErode OpenCV function to the image",
87       "Thiago Santos<thiago.sousa.santos@collabora.co.uk>");
88 }
89 
90 /* initialize the new element
91  * instantiate pads and add them to element
92  * set pad calback functions
93  * initialize instance structure
94  */
95 static void
gst_cv_erode_init(GstCvErode * filter)96 gst_cv_erode_init (GstCvErode * filter)
97 {
98 }
99 
100 static GstFlowReturn
gst_cv_erode_transform_ip(GstOpencvVideoFilter * base,GstBuffer * buf,cv::Mat img)101 gst_cv_erode_transform_ip (GstOpencvVideoFilter * base, GstBuffer * buf,
102     cv::Mat img)
103 {
104   GstCvDilateErode *filter = GST_CV_DILATE_ERODE (base);
105 
106   cv::erode (img, img, cv::Mat (), cv::Point (-1, -1), filter->iterations);
107 
108   return GST_FLOW_OK;
109 }
110 
111 gboolean
gst_cv_erode_plugin_init(GstPlugin * plugin)112 gst_cv_erode_plugin_init (GstPlugin * plugin)
113 {
114   GST_DEBUG_CATEGORY_INIT (gst_cv_erode_debug, "cverode", 0, "cverode");
115 
116   return gst_element_register (plugin, "cverode", GST_RANK_NONE,
117       GST_TYPE_CV_ERODE);
118 }
119