challenge-bot.com Git - ozzloy@gmail.com/oble/blob - hello-world.cpp

   1 //use the camera to aid the decision to sleep.
   2 //Copyright 2009 Daniel Watson
   3 /*
   4     This file is part of oble.
   5
   6     oble is free software: you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation, either version 3 of the License, or
   9     (at your option) any later version.
  10
  11     oble is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with oble.  If not, see <http://www.gnu.org/licenses/>.
  18
  19  */
  20 #include <cv.h>
  21 #include <highgui.h>
  22 #include <stdio.h>
  23
  24 // argument for cvFlip(src, dest, FLIP_TYPE)
  25 #define VERTICAL_FLIP 1
  26
  27 CvHaarClassifierCascade* load_object_detector(const char* cascade_path)
  28 {
  29         return (CvHaarClassifierCascade*)cvLoad(cascade_path);
  30 }
  31
  32 void detect_and_draw_objects(IplImage* image,
  33                 CvHaarClassifierCascade* cascade,
  34                 int do_pyramids)
  35 {
  36         IplImage* small_image = image;
  37         CvMemStorage* storage = cvCreateMemStorage(0);
  38         CvSeq* faces;
  39         int i, scale = 1;
  40
  41         /* if the flag is specified, down-scale the input image to get a
  42                  performance boost w/o loosing quality (perhaps) */
  43         if(do_pyramids)
  44         {
  45                 small_image = cvCreateImage(cvSize(image->width/2,image->height/2),
  46                                 IPL_DEPTH_8U, 3);
  47                 cvPyrDown(image, small_image, CV_GAUSSIAN_5x5);
  48                 scale = 2;
  49         }
  50
  51         /* use the fastest variant */
  52         faces = cvHaarDetectObjects(small_image, cascade, storage, 1.2, 2,
  53                         CV_HAAR_DO_CANNY_PRUNING);
  54
  55         /* draw all the rectangles */
  56         for(i = 0; i < faces->total; i++)
  57         {
  58                 /* extract the rectangles only */
  59                 CvRect face = *(CvRect*)cvGetSeqElem(faces, i);
  60                 CvPoint upperLeft = cvPoint(face.x*scale,face.y*scale);
  61                 CvPoint bottomRight = cvPoint((face.x+face.width)*scale,
  62                                 (face.y+face.height)*scale);
  63                 cvRectangle(image, upperLeft, bottomRight, CV_RGB(255,0,0), 3);
  64         }
  65
  66         if(small_image != image)
  67                 cvReleaseImage(&small_image);
  68         cvReleaseMemStorage(&storage);
  69 }
  70
  71 // A Simple Camera Capture Framework.
  72 int main(int argc, char** argv)
  73 {
  74         CvHaarClassifierCascade* cascade = load_object_detector(argv[1]);
  75
  76         CvCapture* capture = cvCaptureFromCAM(CV_CAP_ANY);
  77         if(!capture) {
  78                 fprintf(stderr, "ERROR: capture is NULL \n");
  79                 getchar();
  80                 return -1;
  81         }
  82
  83         // Create a window in which the captured images will be presented
  84         cvNamedWindow("mywindow", CV_WINDOW_AUTOSIZE);
  85
  86         IplImage* frame = cvQueryFrame(capture);
  87         if(!frame) {
  88                 fprintf(stderr, "ERROR: frame is null...\n");
  89                 getchar();
  90                 return -1;
  91         }
  92         IplImage* flipped =
  93                 cvCreateImage(cvGetSize(frame), frame->depth, frame->nChannels);
  94         // Show the image captured from the camera in the window and repeat
  95         while(1) {
  96                 // Get one frame
  97                 frame = cvQueryFrame(capture);
  98                 if(!frame) {
  99                         fprintf(stderr, "ERROR: frame is null...\n");
 100                         getchar();
 101                         break;
 102                 }
 103
 104                 //flip the image so displayed right/left corresponds to physical right/left
 105                 cvFlip(frame, flipped, VERTICAL_FLIP);
 106
 107                 detect_and_draw_objects(flipped, cascade, 1);
 108                 cvShowImage("mywindow", flipped);
 109                 // Do not release the frame!
 110
 111                 //If ESC key pressed, Key=0x10001B under OpenCV 0.9.7(linux version),
 112                 //remove higher bits using AND operator
 113                 if((cvWaitKey(10) & 255) == 27) break;
 114         }
 115
 116         // Release the capture device housekeeping
 117         cvReleaseCapture(&capture);
 118         cvDestroyWindow("mywindow");
 119         cvReleaseImage(&flipped);
 120         return 0;
 121 }