ictfddlg.cpp

该文件夹下的代码是我在做人脸检测时的VC源代码

// IctFdDlg.cpp : implementation file
//

#include "stdafx.h"
#include "IctFd.h"
#include "IctFdDlg.h"
#include "FdHaar.h"
#include "count_time.h"
#include <time.h>
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
extern CvMemStorage* storage;
CvCapture* capture = 0;
extern CvHaarClassifierCascade* cascade;
class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CIctFdDlg dialog

CIctFdDlg::CIctFdDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CIctFdDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CIctFdDlg)
	m_time = 0;
	m_filepath = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CIctFdDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CIctFdDlg)
	DDX_Text(pDX, IDC_TIME, m_time);
	DDX_Text(pDX, IDC_FilePath, m_filepath);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CIctFdDlg, CDialog)
	//{{AFX_MSG_MAP(CIctFdDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_FD, OnFd)
	ON_BN_CLICKED(IDC_OpenPic, OnOpenPic)
	ON_BN_CLICKED(IDC_OpenVidio, OnOpenVidio)
	ON_BN_CLICKED(IDC_CLEAR, OnClear)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CIctFdDlg message handlers

BOOL CIctFdDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CIctFdDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CIctFdDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CIctFdDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CIctFdDlg::OnFd() 
{
	// TODO: Add your control notification handler code here
	clock_t first,finish;   
    IplImage *frame, *frame_copy = 0;
    int optlen = strlen("--cascade=");
//    const char* input_name;
    const char* cascade_name ="C:\\OpenCV\\data\\haarcascades\\haarcascade_frontalface_alt.xml";
    cascade = (CvHaarClassifierCascade*)cvLoad( cascade_name, 0, 0, 0 );
    if( !cascade )
    {
		cascade_name ="C:\\Program Files\\OpenCV\\data\\haarcascades\\haarcascade_frontalface_alt.xml";
		cascade = (CvHaarClassifierCascade*)cvLoad( cascade_name, 0, 0, 0 );
		 if( !cascade )
		 {
			 MessageBox("ERROR: Could not load classifier cascade\n" );
			 return;		 
		 }
        
    }
	 //确定内存大小默认64k
    storage = cvCreateMemStorage(0);

    if(capture )
    {
        for(;;)
        {
            if( !cvGrabFrame( capture ))
                break;
            frame = cvRetrieveFrame( capture );
            if( !frame )
                break;
            if( !frame_copy )
                frame_copy = cvCreateImage( cvSize(frame->width,frame->height),
                                            IPL_DEPTH_8U, frame->nChannels );
            if( frame->origin == IPL_ORIGIN_TL )
                cvCopy( frame, frame_copy, 0 );
            else
                cvFlip( frame, frame_copy, 0 );
           // first=clock(); 
            detect_and_draw( frame_copy );
			//finish=clock();  
            if( cvWaitKey( 10 ) >= 0 )
                break;
        }

        cvReleaseImage( &frame_copy );
        cvReleaseCapture( &capture );
    }
    else
    {
		if(m_filepath=="")
		{
			MessageBox("please open a picture!");
			return;
		}
		IplImage* image = cvLoadImage( m_filepath, 1 );
        if( image )
        {
			first=clock();  
		//	timerBegin();
            detect_and_draw( image );
			finish=clock(); m_time=(finish-first);
		//	double dTime=getTime();	m_time=(long)(dTime*1000);
			
			UpdateData(false);
            cvWaitKey(0);
            cvReleaseImage( &image );
        }
        else
        {
            /* assume it is a text file containing the list of the image filenames to be processed - one per line */
            FILE* f = fopen( m_filepath, "rt" );
            if( f )
            {
                char buf[1000+1];
                while( fgets( buf, 1000, f ) )
                {
                    int len = (int)strlen(buf);
                    while( len > 0 && isspace(buf[len-1]) )
                        len--;
                    buf[len] = '\0';
                    image = cvLoadImage( buf, 1 );
                    if( image )
                    {
                        detect_and_draw( image );
                        cvWaitKey(0);
                        cvReleaseImage( &image );
                    }
                }
                fclose(f);
            }
        }

    }
    capture = 0;
    cvDestroyWindow("result");
}

void CIctFdDlg::OnOpenPic() 
{
	// TODO: Add your control notification handler code here
	IplImage* pImg;

	char szFilter[]="All Files(*.*)|*.*||";
	CFileDialog filedlg(TRUE,NULL,NULL,OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,szFilter);
	if(filedlg.DoModal()!=IDOK)
	{	
		return;
	}
	m_filepath=filedlg.GetPathName();
	UpdateData(false);
	pImg = cvLoadImage( m_filepath, 1);
	cvNamedWindow( "result", 1 );
	cvShowImage( "result", pImg );	
}

void CIctFdDlg::OnOpenVidio() 
{
	// TODO: Add your control notification handler code here
//	CvCapture* capture = 0;
    capture = cvCaptureFromCAM(-1);
    if( !capture )
    {
        MessageBox("Open Vedio error! ");
        return;
    }       
    cvNamedWindow( "result", 0 );
    for(;;)
    {
        IplImage* frame = 0;
        frame = cvQueryFrame( capture );
        if( !frame )
            break;
		cvShowImage("result", frame );
        if( cvWaitKey(10) >= 0 )
            break;
    }
	cvReleaseCapture( &capture );
    cvDestroyWindow("result");
	capture = 0;
}

void CIctFdDlg::OnClear() 
{
	// TODO: Add your control notification handler code here
	m_filepath="";
	m_time=0;
	UpdateData(false);
}