1 /*****************************************************************************
2 * opencv_example.cpp : Example OpenCV internal video filter
3 * (performs face identification). Mostly taken from the facedetect.c
5 *****************************************************************************
6 * Copyright (C) 2006-2012 VLC authors and VideoLAN
7 * Copyright (C) 2012 Edward Wang
9 * Authors: Dugal Harris <dugalh@protoclea.co.za>
10 * Edward Wang <edward.c.wang@compdigitec.com>
12 * This program is free software; you can redistribute it and/or modify it
13 * under the terms of the GNU Lesser General Public License as published by
14 * the Free Software Foundation; either version 2.1 of the License, or
15 * (at your option) any later version.
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU Lesser General Public License for more details.
22 * You should have received a copy of the GNU Lesser General Public License
23 * along with this program; if not, write to the Free Software Foundation,
24 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25 *****************************************************************************/
27 /*****************************************************************************
29 *****************************************************************************/
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_filter.h>
38 #include <vlc_picture.h>
40 #include <vlc_image.h>
41 #include "filter_event_info.h"
43 #include <opencv2/core/core_c.h>
44 #include <opencv2/core/core.hpp>
45 #include <opencv2/imgproc/imgproc_c.h>
46 #include <opencv2/imgproc/imgproc.hpp>
47 #include <opencv2/objdetect/objdetect.hpp>
49 /*****************************************************************************
50 * filter_sys_t : filter descriptor
51 *****************************************************************************/
57 CvMemStorage
* p_storage
;
58 CvHaarClassifierCascade
* p_cascade
;
59 video_filter_event_info_t event_info
;
65 /****************************************************************************
67 ****************************************************************************/
68 static int OpenFilter ( vlc_object_t
* );
69 static void CloseFilter( vlc_object_t
* );
71 static picture_t
*Filter( filter_t
*, picture_t
* );
73 /*****************************************************************************
75 *****************************************************************************/
77 set_description( N_("OpenCV face detection example filter") )
78 set_shortname( N_( "OpenCV example" ))
79 set_capability( "opencv internal filter", 1 )
80 add_shortcut( "opencv_example" )
82 set_category( CAT_VIDEO
)
83 set_subcategory( SUBCAT_VIDEO_VFILTER
)
84 set_callbacks( OpenFilter
, CloseFilter
)
86 add_string( "opencv-haarcascade-file", "c:\\haarcascade_frontalface_alt.xml",
87 N_("Haar cascade filename"),
88 N_("Name of XML file containing Haar cascade description"), false);
91 /*****************************************************************************
92 * OpenFilter: probe the filter and return score
93 *****************************************************************************/
94 static int OpenFilter( vlc_object_t
*p_this
)
96 filter_t
*p_filter
= (filter_t
*)p_this
;
99 /* Allocate the memory needed to store the decoder's structure */
100 if( ( p_filter
->p_sys
= p_sys
=
101 (filter_sys_t
*)malloc(sizeof(filter_sys_t
)) ) == NULL
)
106 //init the video_filter_event_info_t struct
107 p_sys
->event_info
.i_region_size
= 0;
108 p_sys
->event_info
.p_region
= NULL
;
111 p_filter
->pf_video_filter
= Filter
;
113 //create the VIDEO_FILTER_EVENT_VARIABLE
115 if (var_Create( vlc_object_instance(p_filter
), VIDEO_FILTER_EVENT_VARIABLE
, VLC_VAR_ADDRESS
| VLC_VAR_DOINHERIT
) != VLC_SUCCESS
)
116 msg_Err( p_filter
, "Could not create %s", VIDEO_FILTER_EVENT_VARIABLE
);
118 val
.p_address
= &(p_sys
->event_info
);
119 if (var_Set( vlc_object_instance(p_filter
), VIDEO_FILTER_EVENT_VARIABLE
, val
)!=VLC_SUCCESS
)
120 msg_Err( p_filter
, "Could not set %s", VIDEO_FILTER_EVENT_VARIABLE
);
122 //OpenCV init specific to this example
123 char* filename
= var_InheritString( p_filter
, "opencv-haarcascade-file" );
124 p_sys
->p_cascade
= (CvHaarClassifierCascade
*)cvLoad( filename
, 0, 0, 0 );
125 p_sys
->p_storage
= cvCreateMemStorage(0);
131 /*****************************************************************************
132 * CloseFilter: clean up the filter
133 *****************************************************************************/
134 static void CloseFilter( vlc_object_t
*p_this
)
136 filter_t
*p_filter
= (filter_t
*)p_this
;
137 filter_sys_t
*p_sys
= static_cast<filter_sys_t
*>(p_filter
->p_sys
);
139 if( p_sys
->p_cascade
)
140 cvReleaseHaarClassifierCascade( &p_sys
->p_cascade
);
142 if( p_sys
->p_storage
)
143 cvReleaseMemStorage( &p_sys
->p_storage
);
145 free( p_sys
->event_info
.p_region
);
148 var_Destroy( vlc_object_instance(p_filter
), VIDEO_FILTER_EVENT_VARIABLE
);
151 /****************************************************************************
152 * Filter: Check for faces and raises an event when one is found.
153 ****************************************************************************/
154 static picture_t
*Filter( filter_t
*p_filter
, picture_t
*p_pic
)
156 IplImage
** p_img
= NULL
;
159 filter_sys_t
*p_sys
= static_cast<filter_sys_t
*>(p_filter
->p_sys
);
163 msg_Err( p_filter
, "no image array" );
166 //(hack) cast the picture_t to array of IplImage*
167 p_img
= (IplImage
**) p_pic
;
169 //check the image array for validity
170 if ((!p_img
[0])) //1st plane is 'I' i.e. greyscale
172 msg_Err( p_filter
, "no image" );
176 //perform face detection
177 cvClearMemStorage(p_sys
->p_storage
);
178 if( p_sys
->p_cascade
)
180 //we should make some of these params config variables
181 CvSeq
*faces
= cvHaarDetectObjects( p_img
[0], p_sys
->p_cascade
,
182 p_sys
->p_storage
, 1.15, 5,
183 CV_HAAR_DO_CANNY_PRUNING
,
185 //create the video_filter_region_info_t struct
186 if (faces
&& (faces
->total
> 0))
188 //msg_Dbg( p_filter, "Found %d face(s)", faces->total );
189 free( p_sys
->event_info
.p_region
);
190 p_sys
->event_info
.p_region
= (video_filter_region_info_t
*)
191 calloc( faces
->total
, sizeof(video_filter_region_info_t
));
192 if( !p_sys
->event_info
.p_region
)
194 p_sys
->event_info
.i_region_size
= faces
->total
;
197 //populate the video_filter_region_info_t struct
198 for( int i
= 0; i
< (faces
? faces
->total
: 0); i
++ )
200 CvRect
*r
= (CvRect
*)cvGetSeqElem( faces
, i
);
202 pt2
.x
= (r
->x
+r
->width
)*scale
;
204 pt2
.y
= (r
->y
+r
->height
)*scale
;
205 cvRectangle( p_img
[0], pt1
, pt2
, CV_RGB(0,0,0), 3, 8, 0 );
207 *(CvRect
*)(&(p_sys
->event_info
.p_region
[i
])) = *r
;
208 p_sys
->event_info
.p_region
[i
].i_id
= p_sys
->i_id
++;
209 p_sys
->event_info
.p_region
[i
].p_description
= "Face Detected";
212 if (faces
&& (faces
->total
> 0)) //raise the video filter event
213 var_TriggerCallback( vlc_object_instance(p_filter
), VIDEO_FILTER_EVENT_VARIABLE
);
216 msg_Err( p_filter
, "No cascade - is opencv-haarcascade-file valid?" );