www.pudn.com > AdaBoost_weaklearner_1.rar > MainFrm.cpp

// MainFrm.cpp : implementation of the CMainFrame class 
// 
#include  
#include  
#include  
 
#include "stdafx.h" 
#include "AdaBoost.h" 
#include "TrainDlg.h" 
#include "MainFrm.h" 
#include "Adaboost_train_main.h" 
#include "Adaboost_common.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CMainFrame 
 
IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd) 
 
BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd) 
	//{{AFX_MSG_MAP(CMainFrame) 
		// NOTE - the ClassWizard will add and remove mapping macros here. 
		//    DO NOT EDIT what you see in these blocks of generated code ! 
	ON_WM_CREATE() 
	ON_COMMAND(ID_FILE_TRAIN, OnFileTrain) 
	ON_MESSAGE(WM_COMPUTATION_FINISH,OnComputationFinish) 
	ON_UPDATE_COMMAND_UI(ID_INDICATOR_TIME,OnUpdateTime) 
	ON_MESSAGE(WM_COMPUTATION_FINISH,OnComputationFinish) 
	ON_WM_TIMER() 
	ON_COMMAND(ID_CLEAR_TEXT, OnClearText) 
	ON_WM_CLOSE() 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
static UINT indicators[] = 
{ 
	ID_SEPARATOR,           // status line indicator 
	ID_INDICATOR_CAPS, 
	ID_INDICATOR_TIME, 
	//ID_INDICATOR_NUM, 
	//ID_INDICATOR_SCRL, 
}; 
 
///////////////////////////////////////////////////////////////////////////// 
// CMainFrame construction/destruction 
 
CMainFrame::CMainFrame() 
{ 
	// TODO: add member initialization code here 
	 
} 
 
CMainFrame::~CMainFrame() 
{ 
} 
 
int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{ 
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1) 
		return -1; 
	 
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP 
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) || 
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME)) 
	{ 
		TRACE0("Failed to create toolbar\n"); 
		return -1;      // fail to create 
	} 
 
	if (!m_wndStatusBar.Create(this) || 
		!m_wndStatusBar.SetIndicators(indicators, 
		  sizeof(indicators)/sizeof(UINT))) 
	{ 
		TRACE0("Failed to create status bar\n"); 
		return -1;      // fail to create 
	} 
 
	// TODO: Delete these three lines if you don't want the toolbar to 
	//  be dockable 
	m_wndStatusBar.SetPaneInfo(1,ID_INDICATOR_TIME,SBPS_NORMAL,150); 
	m_nTimeIndex=m_wndStatusBar.CommandToIndex(ID_INDICATOR_TIME); 
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY); 
	EnableDocking(CBRS_ALIGN_ANY); 
	DockControlBar(&m_wndToolBar); 
 
	return 0; 
} 
 
BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs) 
{ 
	if( !CFrameWnd::PreCreateWindow(cs) ) 
		return FALSE; 
	// TODO: Modify the Window class or styles here by modifying 
	//  the CREATESTRUCT cs 
 
	return TRUE; 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CMainFrame diagnostics 
 
#ifdef _DEBUG 
void CMainFrame::AssertValid() const 
{ 
	CFrameWnd::AssertValid(); 
} 
 
void CMainFrame::Dump(CDumpContext& dc) const 
{ 
	CFrameWnd::Dump(dc); 
} 
 
#endif //_DEBUG 
 
///////////////////////////////////////////////////////////////////////////// 
// CMainFrame message handlers 
UINT ComputeThreadProc(LPVOID pParam) 
{	 
//	char errorTrainfile[200]; 
//	char errorTestfile[200]; 
	Adaboost_train_main(); 
	if (com_pro.show_openfile)  
	{ 
		char* cmdline; 
		char  oldline[200]="notepad "; 
		cmdline=strcat(oldline, com_param.resultfile); 
		WinExec(cmdline,3); 
	}  
	::PostMessage((HWND)pParam,WM_COMPUTATION_FINISH,0,0);  
     
    return 0; 
} 
void CMainFrame::OnFileTrain()  
{ 
	CLearnDlg Ld; 
	Ld.m_strTrain=com_param.trainfile; 
	Ld.m_strTest=com_param.trainfile; 
	Ld.m_strResult=com_param.resultfile; 
	if (Ld.DoModal()==IDOK)  
	{ 
		com_pro.show_openfile=Ld.m_bOpen; 
		strcpy(com_param.trainfile,Ld.m_strTrain); 
        strcpy(com_param.testfile,Ld.m_strTest); 
		strcpy(com_param.resultfile,Ld.m_strResult); 
		GetActiveView()->SetWindowText(NULL); 
		BeginComputation();		 
		CWinThread* pThread= 
			AfxBeginThread(ComputeThreadProc,GetSafeHwnd(),THREAD_PRIORITY_NORMAL); 
	} 
	 
} 
 
void CMainFrame::BeginComputation() 
{ 
	com_param.Running=TRUE; 
	com_param.Finished=FALSE; 
	temp_pro.Copy(com_pro); 
	m_nTimer=SetTimer(1001,500,NULL); 
	m_tmBegin=CTime::GetCurrentTime(); 
 
} 
 
void CMainFrame::OnComputationFinish(WPARAM wParam,LPARAM lParam) 
{ 
  CeaseComputation(); 
} 
 
void CMainFrame::CeaseComputation() 
{ 
	 
	if (com_param.Running==FALSE) 
		printm("----Computation was stop by user."); 
	com_param.Running=FALSE; 
	com_param.Finished=TRUE; 
	com_pro.Copy(temp_pro); 
 
	KillTimer(m_nTimer); 
	m_wndStatusBar.SetPaneText(1,"Waiting..."); 
	if (com_param.Close) 
		CFrameWnd::OnClose(); 
} 
 
void CMainFrame::OnUpdateTime(CCmdUI* pCmdUI) 
{ 
	pCmdUI->Enable(); 
	pCmdUI->SetText(m_strTime); 
} 
 
void CMainFrame::OnTimer(UINT nIDEvent)  
{ 
	CTimeSpan t=CTime::GetCurrentTime()-m_tmBegin; 
	m_strTime=t.Format("Computing %H:%M:%S already"); 
	m_wndStatusBar.SetPaneText(1,m_strTime,TRUE); 
	CFrameWnd::OnTimer(nIDEvent); 
} 
 
void CMainFrame::OnClearText()  
{ 
 	GetActiveView()->SetWindowText(NULL); 
} 
 
void CMainFrame::OnClose()  
{ 
	if (!com_param.Finished) 
	{ 
		if( AfxMessageBox("Still computing ,Do you want to exit?",MB_YESNO)==IDYES) 
		{ 
			com_param.Running=FALSE; 
			com_param.Close=TRUE; 
			return ; 
		} 
		else return ; 
 
	} 
	else CFrameWnd::OnClose(); 
}