深入剖析WTL——如何封装Windows界面程序

CAppModule _Module;
int WINAPI _tWinMain(HINSTANCE hInstance, HINSTANCE /*hPrevInstance*/, 
LPTSTR lpstrCmdLine, int nCmdShow)
{
HRESULT hRes = ::CoInitialize(NULL);
// If you are running on NT 4.0 or higher you can use 
// the following call instead to 
// make the EXE free threaded. This means that calls 
// come in on a random RPC thread.
// HRESULT hRes = ::CoInitializeEx(NULL, COINIT_MULTITHREADED);
ATLASSERT(SUCCEEDED(hRes));
// this resolves ATL window thunking problem when Microsoft
//  Layer for Unicode (MSLU) is used
::DefWindowProc(NULL, 0, 0, 0L);
AtlInitCommonControls(ICC_COOL_CLASSES | ICC_BAR_CLASSES); 
// add flags to support other controls
    //初始化模块
 hRes = _Module.Init(NULL, hInstance);
 ATLASSERT(SUCCEEDED(hRes));
    //程序逻辑，调用全局函数Run()
 int nRet = Run(lpstrCmdLine, nCmdShow);
     //终止模块
 _Module.Term();
 ::CoUninitialize();
 return nRet;
}

int Run(LPTSTR /*lpstrCmdLine*/ = NULL, int nCmdShow = SW_SHOWDEFAULT)
{
 CMessageLoop theLoop;
 _Module.AddMessageLoop(&theLoop);
 CMainFrame wndMain;
 if(wndMain.CreateEx() == NULL)
 {
  ATLTRACE(_T("Main window creation failed!\n"));
  return 0;
 }
 wndMain.ShowWindow(nCmdShow);
 int nRet = theLoop.Run();
 _Module.RemoveMessageLoop();
 return nRet;
}

class CThreadManager {
public:
 // thread init param
 struct _RunData
 {
  LPTSTR lpstrCmdLine;
  int nCmdShow;
 };
static DWORD WINAPI RunThread(LPVOID lpData);
DWORD m_dwCount;  //count of threads
 HANDLE m_arrThreadHandles[MAXIMUM_WAIT_OBJECTS - 1];
 CThreadManager() : m_dwCount(0) {}
 DWORD AddThread(LPTSTR lpstrCmdLine, int nCmdShow);
 void RemoveThread(DWORD dwIndex);
 int Run(LPTSTR lpstrCmdLine, int nCmdShow);
};

static DWORD WINAPI RunThread(LPVOID lpData)
 {
  CMessageLoop theLoop;
  _Module.AddMessageLoop(&theLoop);
  _RunData* pData = (_RunData*)lpData;
  CMainFrame wndFrame;
  if(wndFrame.CreateEx() == NULL)
  {
   ATLTRACE(_T("Frame window creation failed!\n"));
   return 0;
  }
  wndFrame.ShowWindow(pData->nCmdShow);
  ::SetForegroundWindow(wndFrame); // Win95 needs this
  delete pData;
  int nRet = theLoop.Run();
  _Module.RemoveMessageLoop();
  return nRet;
 }

DWORD AddThread(LPTSTR lpstrCmdLine, int nCmdShow)
 {
  if(m_dwCount == (MAXIMUM_WAIT_OBJECTS - 1))
  {
::MessageBox(NULL, _T("ERROR: Cannot create ANY MORE threads!!!"), 
_T("test2"), MB_OK);
  return 0;
  }
  _RunData* pData = new _RunData;
  pData->lpstrCmdLine = lpstrCmdLine;
  pData->nCmdShow = nCmdShow;
  DWORD dwThreadID;
  HANDLE hThread = ::CreateThread(NULL, 0, RunThread, pData, 0, 
&dwThreadID);
  if(hThread == NULL)
  {
::MessageBox(NULL, _T("ERROR: Cannot create thread!!!"), _T("Application"), MB_OK);
   return 0;
  }
  m_arrThreadHandles[m_dwCount] = hThread;
  m_dwCount++;
  return dwThreadID;
 }

HANDLE hThread = ::CreateThread(NULL, 0, RunThread, pData, 0, &dwThreadID);

int WINAPI _tWinMain(HINSTANCE hInstance, HINSTANCE /*hPrevInstance*/, 
LPTSTR lpstrCmdLine, int nCmdShow)
{
 HRESULT hRes = ::CoInitialize(NULL);
// If you are running on NT 4.0 or higher you can use the 
// following call instead to 
// make the EXE free threaded. This means that calls come in on 
// a random RPC thread.
// HRESULT hRes = ::CoInitializeEx(NULL, COINIT_MULTITHREADED);
 ATLASSERT(SUCCEEDED(hRes));
 // this resolves ATL window thunking problem when Microsoft 
//  Layer for Unicode (MSLU) is used
 ::DefWindowProc(NULL, 0, 0, 0L);
 AtlInitCommonControls(ICC_COOL_CLASSES | ICC_BAR_CLASSES); 
// add flags to support other controls
 hRes = _Module.Init(NULL, hInstance);
 ATLASSERT(SUCCEEDED(hRes));
 int nRet = 0;
 // BLOCK: Run application
     // 注意该处和前面应用程序的差别。
     // 这里创建一个CThreadManager类的实例，然后调用该类的Run()函数。
 {
  CThreadManager mgr;
  nRet = mgr.Run(lpstrCmdLine, nCmdShow);
 }
 _Module.Term();
 ::CoUninitialize();
 return nRet;
}

int Run(LPTSTR lpstrCmdLine, int nCmdShow)
{
  MSG msg;
  // force message queue to be created
  ::PeekMessage(&msg, NULL, WM_USER, WM_USER, PM_NOREMOVE);
         // 创建一个界面线程。为该应用程序的第一个界面线程。该界面线程的主体
// 逻辑RunThread()是创建主窗口，并创建消息循环对象。
  AddThread(lpstrCmdLine, nCmdShow);
  int nRet = m_dwCount;
  DWORD dwRet;
  while(m_dwCount > 0)
  {
  dwRet = ::MsgWaitForMultipleObjects(m_dwCount,
 m_arrThreadHandles, FALSE, INFINITE, QS_ALLINPUT);
   if(dwRet == 0xFFFFFFFF)
    ::MessageBox(NULL, 
_T("ERROR: Wait for multiple objects failed!!!"),
 _T("test2"), MB_OK);
   else if(dwRet >= WAIT_OBJECT_0 && 
dwRet <= (WAIT_OBJECT_0 + m_dwCount - 1))
    RemoveThread(dwRet - WAIT_OBJECT_0);
   else if(dwRet == (WAIT_OBJECT_0 + m_dwCount))
   {
    ::GetMessage(&msg, NULL, 0, 0);
    if(msg.message == WM_USER)
     AddThread("", SW_SHOWNORMAL);
    else
     ::MessageBeep((UINT)-1);
   }
   else
    ::MessageBeep((UINT)-1);
  }
  return nRet;
 }