th

/

#include "stdafx.h"
#include "ThreadSynch.h"
#include "ThreadSynchDlg.h"

//头文件
#include <afxmt.h>

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

//我们使用MFC类来操作事件和互斥对象,也可以使用API创建HANDLE
CEvent g_event;
CMutex g_mutex;

//这里使用API创建HANDLE和CRITICAL_SECTION,也可以使用MFC类
CRITICAL_SECTION g_hCritial;
HANDLE g_hSema = CreateSemaphore(NULL,1,1,NULL);

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

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()

/////////////////////////////////////////////////////////////////////////////
// CThreadSynchDlg dialog

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

 m_nCountThread = 0;
}

void CThreadSynchDlg::DoDataExchange(CDataExchange* pDX)
{
 CDialog::DoDataExchange(pDX);
 //{{AFX_DATA_MAP(CThreadSynchDlg)
 DDX_Radio(pDX, IDC_RADIO_SYNCH, m_nSynch);
 //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CThreadSynchDlg, CDialog)
 //{{AFX_MSG_MAP(CThreadSynchDlg)
 ON_WM_SYSCOMMAND()
 ON_WM_PAINT()
 ON_WM_QUERYDRAGICON()
 ON_BN_CLICKED(IDC_BTN_A, OnBtnA)
 ON_BN_CLICKED(IDC_BTN_B, OnBtnB)
 ON_BN_CLICKED(IDC_BTN_MYHELP, OnBtnMyhelp)
 ON_BN_CLICKED(IDC_RADIO_CRITIAL, OnRadioCritial)
 ON_WM_DESTROY()
 //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CThreadSynchDlg message handlers

BOOL CThreadSynchDlg::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
 

 //初始化
 g_event.SetEvent();
 InitializeCriticalSection(&g_hCritial);

 return TRUE;  // return TRUE  unless you set the focus to a control
}

void CThreadSynchDlg::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 CThreadSynchDlg::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 CThreadSynchDlg::OnQueryDragIcon()
{
 return (HCURSOR) m_hIcon;
}

void CThreadSynchDlg::FunThread(char Id, CThreadSynchDlg *pDlg)
{
 //根据选择处理线程同步
 switch(pDlg->m_nSynch)
 {
 case 1:
  g_event.Lock();break;
 case 2:
  g_mutex.Lock();break;
 case 3:
  EnterCriticalSection(&g_hCritial);break;
 case 4:
  WaitForSingleObject(g_hSema, INFINITE);break;
 }

 //打印字符
 for(int i = 0; i < 25; i++)
 {
  CString str;
  CEdit *pEdit = (CEdit *)pDlg->GetDlgItem(IDC_EDIT_SHOW);
  pEdit->GetWindowText(str);
  str += Id;
  pEdit->SetWindowText(str);
  Sleep(200);
 }

 //释放同步对象
 g_event.SetEvent();
 g_mutex.Unlock();
 LeaveCriticalSection(&g_hCritial);
 ReleaseSemaphore(g_hSema, 1, NULL);
}

UINT CThreadSynchDlg::ThreadA(LPVOID lParam)
{
 //线程A
 CThreadSynchDlg *pDlg = (CThreadSynchDlg *)lParam;

 pDlg->m_nCountThread++;

 pDlg->GetDlgItem(IDC_BTN_A)->SetWindowText(_T("线程 A 已启动..."));
 pDlg->GetDlgItem(IDC_BTN_A)->EnableWindow(FALSE);
 
 FunThread('A', pDlg);

 pDlg->m_nCountThread--;
 pDlg->GetDlgItem(IDC_BTN_A)->SetWindowText(_T("启动一个线程写入 A"));
 if(pDlg->m_nCountThread == 0)
 {
  pDlg->GetDlgItem(IDC_BTN_A)->EnableWindow(TRUE);
  pDlg->GetDlgItem(IDC_BTN_B)->EnableWindow(TRUE);
 }
 return 0;
}

UINT CThreadSynchDlg::ThreadB(LPVOID lParam)
{
 //线程B
 CThreadSynchDlg *pDlg = (CThreadSynchDlg *)lParam;

 pDlg->m_nCountThread++;

 pDlg->GetDlgItem(IDC_BTN_B)->SetWindowText(_T("线程 B 已启动..."));
 pDlg->GetDlgItem(IDC_BTN_B)->EnableWindow(FALSE);

 FunThread('B', pDlg);

 pDlg->m_nCountThread--;
 pDlg->GetDlgItem(IDC_BTN_B)->SetWindowText(_T("启动一个线程写入 B"));
 if(pDlg->m_nCountThread == 0)
 {
  pDlg->GetDlgItem(IDC_BTN_A)->EnableWindow(TRUE);
  pDlg->GetDlgItem(IDC_BTN_B)->EnableWindow(TRUE);
 }
 return 0;
}

void CThreadSynchDlg::OnBtnA()
{
 UpdateData();
 AfxBeginThread(ThreadA, this); 
}

void CThreadSynchDlg::OnBtnB()
{
 UpdateData();
 AfxBeginThread(ThreadB, this); 
}

void CThreadSynchDlg::OnBtnMyhelp()
{
 CAboutDlg dlg;
 dlg.DoModal(); 
}

//每次都要重新初始化关键代码段
void CThreadSynchDlg::OnRadioCritial()
{
 InitializeCriticalSection(&g_hCritial); 
}

//关闭对话框时应该释放掉同步对象,使用MFC类则有析构函数完成
void CThreadSynchDlg::OnDestroy()
{
 CDialog::OnDestroy();
 
 CloseHandle(g_hSema);
 DeleteCriticalSection(&g_hCritial);
}