www.pudn.com > c++或C下如何使用多线程.zip > MainFrm.cpp

// MainFrm.cpp : implementation of the CMainFrame class 
// 
 
#include "stdafx.h" 
#include "TrdDemo.h" 
 
#include "MainFrm.h" 
 
#include "counterthread.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) 
	ON_WM_CREATE() 
	ON_COMMAND(ID_COUNTER_START, OnCounterStart) 
	ON_UPDATE_COMMAND_UI(ID_COUNTER_START, OnUpdateCounterStart) 
	ON_COMMAND(ID_COUNTER_STOP, OnCounterStop) 
	ON_UPDATE_COMMAND_UI(ID_COUNTER_STOP, OnUpdateCounterStop) 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
static UINT indicators[] = 
{ 
	ID_SEPARATOR,           // status line indicator 
	ID_INDICATOR_CAPS, 
	ID_INDICATOR_NUM, 
	ID_INDICATOR_SCRL, 
}; 
 
///////////////////////////////////////////////////////////////////////////// 
// CMainFrame construction/destruction 
 
CMainFrame::CMainFrame() 
{ 
	m_bThreadStart = FALSE; 
	m_bThreadStop = TRUE; 
	// 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_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 
 
 
void CMainFrame::OnCounterStart()  
{ 
	StartThread(); 
	m_bThreadStart = TRUE; 
	m_bThreadStop = FALSE; 
	// TODO: Add your command handler code here 
	 
} 
 
void CMainFrame::OnUpdateCounterStart(CCmdUI* pCmdUI)  
{ 
	pCmdUI->Enable(!m_bThreadStart);	 
	// TODO: Add your command update UI handler code here 
	 
} 
 
void CMainFrame::OnCounterStop()  
{ 
	StopThread(); 
	m_bThreadStop = TRUE; 
	m_bThreadStart = FALSE; 
	// TODO: Add your command handler code here 
	 
} 
 
void CMainFrame::OnUpdateCounterStop(CCmdUI* pCmdUI)  
{ 
	pCmdUI->Enable(!m_bThreadStop);	 
	// TODO: Add your command update UI handler code here 
	 
} 
 
void CMainFrame::StartThread() 
{ 
	CBaseThread* pThread; 
 
	for (UINT i = 0; i < 13; i++) 
	{ 
	    pThread = new CCounterThread(i); 
 
		if (pThread == NULL) 
			return; 
 
		ASSERT_VALID(pThread); 
		pThread->m_pThreadParams = NULL;  
 
		if (!pThread->CreateThread(CREATE_SUSPENDED)) 
		{ 
			delete pThread; 
			return; 
		} 
 
		m_ThreadsList.AddTail(pThread); 
 
		VERIFY(pThread->SetThreadPriority(THREAD_PRIORITY_BELOW_NORMAL)); 
 
		pThread->ResumeThread(); 
	} 
 
} 
 
void CMainFrame::StopThread() 
{ 
	for (POSITION pos = m_ThreadsList.GetHeadPosition(); pos != NULL; ) 
	{ 
		CBaseThread* pThread = m_ThreadsList.GetNext(pos); 
		VERIFY(SetEvent(pThread->m_hEventKill)); 
	} 
 
	for (int nThreadsLeft = m_ThreadsList.GetCount(); nThreadsLeft != 0; ) 
	{ 
		WaitForSingleObject(CBaseThread::m_hAnotherDead, INFINITE); 
		Sleep(nThreadsLeft*2); 
		nThreadsLeft = 0; 
		for (pos = m_ThreadsList.GetHeadPosition(); pos != NULL; ) 
		{ 
			CBaseThread* pThread = m_ThreadsList.GetNext(pos); 
			if (WaitForSingleObject(pThread->m_hEventDead, 0) == WAIT_TIMEOUT) 
				++nThreadsLeft; 
		} 
	} 
 
	while (!m_ThreadsList.IsEmpty()) 
	{ 
		CBaseThread* pThread = m_ThreadsList.RemoveHead(); 
		VERIFY(WaitForSingleObject(pThread->m_hThread, INFINITE) == WAIT_OBJECT_0); 
		delete pThread; 
	} 
 
} 
 
BOOL CMainFrame::DestroyWindow()  
{ 
	// TODO: Add your specialized code here and/or call the base class 
	StopThread(); 
	 
	return CFrameWnd::DestroyWindow(); 
}