#include "cv.h"
#include "highgui.h"
#include <iostream>
//// Constants
//const char * DISPLAY_WINDOW = "DisplayWindow";
#define OPENCV_ROOT "C:\\OpenCV2.1"
const char * DISPLAY_WINDOW = "Eyes tracker";
//const CFIndex g_cascade_name_len = 2048;
//char g_cascade_name[g_cascade_name_len] = "abc.xml";
using namespace std;
int main (int argc, char * const argv[]) {
const int scale = 2,
refreshDelay = 10;
// create window, open camera stream, load cascade and create a buffer
cvNamedWindow(DISPLAY_WINDOW, CV_WINDOW_AUTOSIZE);
CvCapture *camera = cvCreateCameraCapture (CV_CAP_ANY);
if (!camera)
abort();
CvHaarClassifierCascade *cascade = (CvHaarClassifierCascade *)cvLoad(OPENCV_ROOT
"/data/haarcascades/haarcascade_eye.xml", 0, 0, 0);
if (!cascade)
abort();
CvMemStorage *storage = cvCreateMemStorage(0);
assert(storage);
// get an initial frame and duplicate it for later work
IplImage *current_frame = cvQueryFrame(camera);
IplImage *draw_image = cvCreateImage(cvSize(current_frame->width, current_frame->height), IPL_DEPTH_8U, 3);
IplImage *gray_image = cvCreateImage(cvSize(current_frame->width, current_frame->height), IPL_DEPTH_8U, 1);
IplImage *small_image = cvCreateImage(cvSize(current_frame->width/scale, current_frame->height/scale), IPL_DEPTH_8U, 1);
assert (current_frame && gray_image && draw_image);
// as long as there are images ...
while (current_frame = cvQueryFrame(camera)) {
// convert to gray and downsize
cvCvtColor(current_frame, gray_image, CV_BGR2GRAY);
cvResize(gray_image, small_image, CV_INTER_LINEAR);
// detect
CvSeq* eyes = cvHaarDetectObjects(small_image, cascade, storage, 1.1, 2,
CV_HAAR_DO_CANNY_PRUNING, cvSize(30, 30));
// draw areas
cvFlip(current_frame, draw_image, 1);
for (int i = 0; i < (eyes ? eyes->total : 0); i++) {
CvRect *r = (CvRect *)cvGetSeqElem(eyes, i);
CvPoint center;
center.x = cvRound((small_image->width - r->width*0.5 - r->x) *scale);
center.y = cvRound((r->y + r->height*0.5)*scale);
int radius = cvRound((r->width + r->height)*0.25*scale);
cvCircle(draw_image, center, radius, CV_RGB(0,255,0), 3, 8, 0);
}
// just show the image
cvShowImage(DISPLAY_WINDOW, draw_image);
// wait a tenth of a second for keypress and window drawing
int key = cvWaitKey(refreshDelay);
if (key == 'q' || key == 'Q')
break;
}
return 0;
}
