1 //use the camera to aid the decision to sleep.
2 //Copyright 2009 Daniel Watson
4 use a camera to prevent screensaver
5 Copyright (C) 2009 daniel watson, ozzloy@gmail.com
7 This program is free software: you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation, either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 This file is part of oble.
23 oble is free software: you can redistribute it and/or modify
24 it under the terms of the GNU General Public License as published by
25 the Free Software Foundation, either version 3 of the License, or
26 (at your option) any later version.
28 oble is distributed in the hope that it will be useful,
29 but WITHOUT ANY WARRANTY; without even the implied warranty of
30 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
31 GNU General Public License for more details.
33 You should have received a copy of the GNU General Public License
34 along with oble. If not, see <http://www.gnu.org/licenses/>.
48 static string cascade_filename
= "";
49 static int verbose
= 0;
51 // argument for cvFlip(src, dest, FLIP_TYPE)
54 CvHaarClassifierCascade
* load_object_detector(const char* cascade_path
)
56 return (CvHaarClassifierCascade
*)cvLoad(cascade_path
);
59 void detect_and_draw_objects(IplImage
* image
,
60 CvHaarClassifierCascade
* cascade
,
63 IplImage
* small_image
= image
;
64 CvMemStorage
* storage
= cvCreateMemStorage(0);
68 /* if the flag is specified, down-scale the input image to get a
69 performance boost w/o loosing quality (perhaps) */
72 small_image
= cvCreateImage(cvSize(image
->width
/2,image
->height
/2),
74 cvPyrDown(image
, small_image
, CV_GAUSSIAN_5x5
);
78 /* use the fastest variant */
79 faces
= cvHaarDetectObjects(small_image
, cascade
, storage
, 1.2, 2,
80 CV_HAAR_DO_CANNY_PRUNING
);
82 update_idle(faces
->total
);
84 /* draw all the rectangles */
85 for(i
= 0; i
< faces
->total
; i
++)
87 /* extract the rectangles only */
88 CvRect face
= *(CvRect
*)cvGetSeqElem(faces
, i
);
89 CvPoint upperLeft
= cvPoint(face
.x
* scale
, face
.y
* scale
);
90 CvPoint bottomRight
= cvPoint((face
.x
+ face
.width
) * scale
,
91 (face
.y
+ face
.height
) * scale
);
92 cvRectangle(image
, upperLeft
, bottomRight
, CV_RGB(255,0,0), 3);
95 if(small_image
!= image
)
96 cvReleaseImage(&small_image
);
97 cvReleaseMemStorage(&storage
);
100 void update_idle(int faces_total
)
102 static int saw_last_time
= 0;
106 t_current
= time(NULL
);
109 printf(":) face \n");
113 if(verbose
)printf("\t\tpoking\n");
115 system("gnome-screensaver-command --poke");
124 printf(":( no face \n");
130 void screensave(time_t t_current
)
132 static int last_elapse
= 0;
133 static int activated
= 0;
134 int elapse
= difftime(time(NULL
), t_current
);
136 if(elapse
> timeout
&& elapse
!= last_elapse
)
138 last_elapse
= elapse
;
141 printf("elapse = %d\n", elapse
);
145 printf("activated\n");
147 system("gnome-screensaver-command -a");
150 if(elapse
< timeout
&& activated
)
152 printf("deactivated\n");
154 system("gnome-screensaver-command -d");
158 int parse_opts(int argc
, char **argv
)
162 const char *options
= "vc:\x0";
163 while((c
= getopt(argc
, argv
, options
)) != -1)
167 case 'v': verbose
= 1; break;
168 case 'c': cascade_filename
= string(optarg
); break;
172 printf("option -%c requires an argument.\n", optopt
);
174 else if(isprint (optopt
))
176 printf("unknown option `-%c'.\n", optopt
);
180 printf("unknown option char `\\x%x'.\n", optopt
);
183 case '\x0': break; //ignore. not sure why this shows up. maybe zsh?
188 for(index
= optind
; index
< argc
; index
++)
189 printf("Non-option arg %s\n", argv
[index
]);
191 if(cascade_filename
== "")
193 printf("you must supply a filename with -c option. example:\n");
197 "/usr/share/opencv/haarcascades/haarcascade_frontalface_alt.xml"
201 if(verbose
) printf("SILENT ALARM ACTIVATED!!!\n");
205 void get_frames(CvCapture
* capture
, CvHaarClassifierCascade
* cascade
)
207 IplImage
* frame
= NULL
;
208 get_one_frame(capture
, frame
);
211 cvCreateImage(cvGetSize(frame
), frame
->depth
, frame
->nChannels
);
212 // Show the image captured from the camera in the window and repeat
214 if(get_one_frame(capture
, frame
)) break;
216 //flip the image so displayed right/left corresponds to physical right/left
217 cvFlip(frame
, mirrored
, MIRROR
);
219 detect_and_draw_objects(mirrored
, cascade
, 1);
220 cvShowImage("mywindow", mirrored
);
221 // Do not release the frame!
222 //screensave(t_current);
224 //If ESC key pressed, Key=0x10001B under OpenCV 0.9.7(linux version),
225 //remove higher bits using AND operator
226 if((cvWaitKey(100) & 255) == 27) break;
228 cvReleaseImage(&mirrored
);
233 CvHaarClassifierCascade
* cascade
=
234 load_object_detector(cascade_filename
.c_str());
236 CvCapture
* capture
= cvCaptureFromCAM(CV_CAP_ANY
);
238 cerr
<< "ERROR: capture is NULL " << endl
;
243 // Create a window in which the captured images will be presented
244 cvNamedWindow("mywindow", CV_WINDOW_AUTOSIZE
);
245 t_current
= time(NULL
);
246 get_frames(capture
, cascade
);
248 // Release the capture device housekeeping
249 cvReleaseCapture(&capture
);
250 cvDestroyWindow("mywindow");
254 int get_one_frame(CvCapture
* capture
, IplImage
* &frame
)
256 frame
= cvQueryFrame(capture
);
258 cerr
<< "ERROR: frame is null..." << endl
;
265 // A Simple Camera Capture Framework.
266 int main(int argc
, char** argv
)
268 if(parse_opts(argc
, argv
)) return 0;