D:/Projekte/Super-Yano/Facedetect/Facedetect/Facedetect.cpp

gehe zur Dokumentation dieser Datei

#include "facedetect.h"

DLLEXPORT void requestVideoImage(HDC hdc)
{
        destinationHDC = hdc;
        isImageRequested = 1;
}

DLLEXPORT void initiateVideo(int camNumber)
{
        capture = cvCaptureFromCAM( camNumber );

        cascade = (CvHaarClassifierCascade*)cvLoad( "./haarcascade_frontalface_alt2.xml", NULL, NULL, NULL );
        if( !cascade )
    {
                fprintf( stderr, "ERROR: Could not load classifier cascade\n" );
                isInitiated = 0;
        return;
    }
        
    storage = cvCreateMemStorage(0);
        
        isInitiated = 1;
}

DLLEXPORT int getVideoSizeX()
{
        return videoSizeX;
}

DLLEXPORT int getVideoSizeY()
{
        return videoSizeY;
}

DLLEXPORT int getMajorFacePosX()
{
        return majorFacePosX;
}

DLLEXPORT int getMajorFacePosY()
{
        return majorFacePosY;
}

DLLEXPORT double getMajorFaceRelativeX()
{
        if((!image)|| (majorFacePosX < 0))
        {
                return -1.0;
        }
        else 
        {
                return ((double)majorFacePosX / (double)image->width);
        }
}

DLLEXPORT double getMajorFaceRelativeY()
{
        if((!image)|| (majorFacePosY < 0))
        {
                return -1.0;
        }
        else 
        {
                return ((double)majorFacePosY / (double)image->height);
        }
}

DLLEXPORT void processVideoFrame()
{
        if (isInitiated)
        {
                if( !capture )
                {
                        fprintf(stderr,"Could not initialize capturing...\n");
                        isInitiated = 0;
                        return;
                }                       

                frame = cvQueryFrame( capture );
                if( frame )
                {
                        if( !image )
                        {
                                // allocate all the buffers 
                                image = cvCreateImage( cvGetSize(frame), 8, 3 );
                                image->origin = frame->origin;
        
                                videoSizeX = image->width;
                                videoSizeY = image->height;
                                panelImage.Create(image->width, image->height, 8, image->origin);
                        }
                


                        cvCopy( frame, image, 0 );
                        isBusy = 1;
                        detect_and_draw( image );
                        isBusy = 0;


                        if (isImageRequested)
                        {
                                panelImage.CopyOf(image, -1);
                                RECT r;
                                if (frame->origin)
                                {
                                        r.bottom = 0;
                                        r.left = 0;
                                        r.top = 240;
                                        r.right = 320;
                                }
                                else
                                {
                                        r.bottom = 240;
                                        r.left = 0;
                                        r.top = 0;
                                        r.right = 320;
                                }

                                panelImage.DrawToHDC(destinationHDC, &r);
                                
                                isImageRequested = 0;
                        }

                } // if frame

        }//if isInitiated
}

DLLEXPORT void closeVideo()
{
        if (capture) cvReleaseCapture( &capture );
        if (image) cvReleaseImage( &image );
        if (&panelImage) panelImage.~CvvImage();
        if (storage) cvReleaseMemStorage( &storage );
        if (cascade) cvReleaseHaarClassifierCascade( &cascade );
}

DLLEXPORT int getNumberOfFaces()
{
        if (isBusy) return -1;
        else return numberOfFaces;
}


void detect_and_draw( IplImage* img )
{
    double scale = 4.0;
    IplImage* gray = cvCreateImage( cvSize(img->width,img->height), 8, 1 );
    IplImage* small_img = cvCreateImage( cvSize( cvRound (img->width/scale),
                         cvRound (img->height/scale)),
                     8, 1 );
    int i;

    cvCvtColor( img, gray, CV_BGR2GRAY );
    cvResize( gray, small_img, CV_INTER_LINEAR );
    cvEqualizeHist( small_img, small_img );
    cvClearMemStorage( storage );

    if( cascade )
    {
     
                if(img->origin) cvFlip(small_img, small_img, 0);

                faces = cvHaarDetectObjects( small_img, cascade, storage,
                                            1.1, 2, 0,
                                            cvSize(30, 30) );

                faceSize = 0;
                majorFacePosX = -1;
                majorFacePosY = -1;

                numberOfFaces = faces->total;


                for( i = 0; i < (faces ? faces->total : 0); i++ )
        { 
                        CvRect* r = (CvRect*)cvGetSeqElem( faces, i );
                        

            CvPoint center;
                        int height;
                        int width;
            center.x = cvRound(((double)r->x + (double)r->width  * 0.5) * scale);
            center.y = cvRound(((double)r->y + (double)r->height * 0.5) * scale);

                        int newSize = cvRound(r->height * r->width * scale);
                        if (newSize > faceSize)
                        {
                                faceSize = newSize;
                                majorFacePosX = center.x;
                                if (img->origin) majorFacePosY = img->height - center.y;
                                else majorFacePosY = center.y;
                        }

                        width = cvRound(r->width*scale * 0.7) ;
                        height = cvRound(r->height*scale * 1.2);

                        if(img->origin)
                        {
                                cvRectangle( img, cvPoint(center.x-width/2,img->height-(center.y-height/2)),
                                                                cvPoint(center.x+width/2,img->height-(center.y+height/2)),
                                                                cvScalar(0,255,0), 2, 8, 0 );
                        }
                        else
                        {
                                cvRectangle( img, cvPoint(center.x-width/2,center.y-height/2),
                              cvPoint(center.x+width/2,center.y+height/2),
                              cvScalar(0,255,0), 2, 8, 0 );
                        }


        }

        }


    cvReleaseImage( &gray );
    cvReleaseImage( &small_img );


}

---