source: proiecte/PPPP/eigenface/TrackFaces.c @ 28

// TrackFaces.c - by Robin Hewitt, 2007
// http://www.cognotics.com/opencv/downloads/camshift_wrapper
// This is free software. See License.txt, in the download
// package, for details.
//

///////////////////////////////////////////////////////////////////////////////
// An example program that It detects a face in live video, and automatically
// begins tracking it, using the Simple Camshift Wrapper.

#include "cv.h"
#include "highgui.h"
#include <stdio.h>
#include "capture.h"
#include "facedet.h"
#include "camshift_wrapper.h"
#include "eigenface.h"

//// Constants
const char * DISPLAY_WINDOW = "DisplayWindow";
const char * DISPLAY_WINDOW2 = "FaceWindow";
#define OPENCV_ROOT  "/home/marius/OpenCV-2.0.0"


//// Global variables
IplImage  * pVideoFrameCopy = 0;


//// Function definitions
int initAll();
void exitProgram(int code);
void captureVideoFrame();
IplImage* convert_grayscale(IplImage *img);

//////////////////////////////////
// main()
//
void main( int argc, char** argv )
{
        CvRect * pFaceRect = 0;

        // validate that an input was specified
        if( argc != 2 )
        {
                printUsage();
                return;
        }

        if( !strcmp(argv[1], "train") ) learn();
        else if( !strcmp(argv[1], "test") )
        {
                if( !initAll() ) exitProgram(-1);

                // Capture and display video frames until a face
                // is detected
                while( 1 )
                {
                        // Look for a face in the next video frame
                        captureVideoFrame();
                        pFaceRect = detectFace(pVideoFrameCopy);

                        // Show the display image
                        cvShowImage( DISPLAY_WINDOW, pVideoFrameCopy );
                        if( (char)27==cvWaitKey(20) ) exitProgram(0);

                        // exit loop when a face is detected
                        if(pFaceRect) break;
                }

                // initialize tracking
                startTracking(pVideoFrameCopy, pFaceRect);
                CvMat * trainPersonNumMat = 0;
                if( !loadTrainingData( &trainPersonNumMat ) ) return;

                // Track the detected face using CamShift
                while( 1 )
                {
                        CvBox2D faceBox;
                        CvRect r;
                        // get the next video frame
                        captureVideoFrame();

                        // track the face in the new video frame
                        r = track(pVideoFrameCopy);

                        cvSetImageROI(pVideoFrameCopy, r);
                        IplImage * pFaceImg =
                          cvCreateImage( cvSize(92,112), pVideoFrameCopy->depth, pVideoFrameCopy->nChannels );
                        cvResize(pVideoFrameCopy, pFaceImg, CV_INTER_AREA );
                                        cvResetImageROI(pVideoFrameCopy);
                        pFaceImg = convert_grayscale(pFaceImg);

                        CvPoint pt1 = { r.x, r.y };
                        CvPoint pt2 = { r.x + r.width, r.y + r.height };
                        cvRectangle(pVideoFrameCopy, pt1, pt2, CV_RGB(0,255,0), 3, 4, 0);

                        recognize(pFaceImg, trainPersonNumMat);
        //              faceBox = track(pVideoFrameCopy);
        //
        //              // outline face ellipse
        //              cvEllipseBox(pVideoFrameCopy, faceBox,
        //                           CV_RGB(255,0,0), 3, CV_AA, 0 );
                        cvShowImage( DISPLAY_WINDOW, pVideoFrameCopy );
                        cvShowImage( DISPLAY_WINDOW2, pFaceImg );
                        if( (char)27==cvWaitKey(20) ) break;
                }
        }
        else
        {
                printf("Unknown command: %s\n", argv[1]);
                printUsage();
        }

        exitProgram(0);
}

IplImage* convert_grayscale(IplImage *img)
{
        IplImage *gray_image,*gray_img0;
        gray_image=cvCloneImage(img);

        //check image is it gray or not if nor convert it to the gray
        if(img->nChannels!=1){
        //convert original image to gray_scale image
                gray_img0 = cvCreateImage(cvSize( gray_image->width,
                                                                                        gray_image->height), 8, 1 );
                cvCvtColor(gray_image, gray_img0, CV_RGB2GRAY );
                gray_image = cvCloneImage( gray_img0 );
        }

        return gray_image;
}

//////////////////////////////////
// initAll()
//
int initAll()
{
        if( !initCapture() ) return 0;
        if( !initFaceDet(OPENCV_ROOT
                "/data/haarcascades/haarcascade_frontalface_default.xml"))
                return 0;

        // Startup message tells user how to begin and how to exit
        printf( "\n********************************************\n"
                "To exit, click inside the video display,\n"
                "then press the ESC key\n\n"
                        "Press <ENTER> to begin"
                        "\n********************************************\n" );
        fgetc(stdin);

        // Create the display window
        cvNamedWindow( DISPLAY_WINDOW, 1 );
        cvNamedWindow( DISPLAY_WINDOW2, 1 );
        //cvSetMouseCallback( DISPLAY_WINDOW, on_mouse, 0 );
        // Initialize tracker
        captureVideoFrame();
        if( !createTracker(pVideoFrameCopy) ) return 0;

        // Set Camshift parameters
        setVmin(70);
        setSmin(30);

        return 1;
}


//////////////////////////////////
// exitProgram()
//
void exitProgram(int code)
{
        // Release resources allocated in this file
        cvDestroyWindow( DISPLAY_WINDOW );
        cvReleaseImage( &pVideoFrameCopy );

        // Release resources allocated in other project files
        closeCapture();
        closeFaceDet();
        releaseTracker();

        exit(code);
}


//////////////////////////////////
// captureVideoFrame()
//
void captureVideoFrame()
{
        // Capture the next frame
        IplImage  * pVideoFrame = nextVideoFrame();
        if( !pVideoFrame ) exitProgram(-1);

        // Copy it to the display image, inverting it if needed
        if( !pVideoFrameCopy )
                pVideoFrameCopy = cvCreateImage( cvGetSize(pVideoFrame), 8, 3 );
        cvCopy( pVideoFrame, pVideoFrameCopy, 0 );
        pVideoFrameCopy->origin = pVideoFrame->origin;

        if( 1 == pVideoFrameCopy->origin ) // 1 means the image is inverted
        {
                cvFlip( pVideoFrameCopy, 0, 0 );
                pVideoFrameCopy->origin = 0;
        }
}