Win32编程点滴之PreTranslateMessage与消息循环

当我们非常熟练得用着MFC/ATL/WTL的时候，是否还记得如何用SDK方式编写程序？本博客将关注与用MFC/ATL/WTL时容易忽略的问题，容易犯的错误，以及一些技巧。

作为第一篇，我们先来讨论一下最基础的一个东西，消息循环（Message loop）。

第一个版本

首先让我们来写一个最容易让人想到的消息循环的形式：

MSG msg
while ( GetMessage( & msg,NULL, 0 , 0 ) )
{
    TranslateMessage( & msg);
    DispatchMessage( & msg);
}

GetMessage函数第一个参数是用来获取MSG结构的指针。第二个参数是一个窗口句柄（HWND），用来获取指定窗口的消息，填NULL表示获取当前线程所有窗口的消息或者线程消息（Thread message）。最后两个参数是wMsgFilterMin和wMsgFilterMax，用来获取指定的消息，当都填0则表示获取所有的消息。

TranslateMessage函数根据WM_KEYUP,WM_KEYDOWN之类的时间，生成相应的WM_CHAR之类的消息。

DispatchMessage函数将窗口消息，交给相应的窗口过程（WindowProc）来处理。

以上的这个消息循环，在大部分情况下都能工作得很好，尤其是一般写点小程序，写成上面的形式完全没有问题。不过偶尔可能会出些问题，所以请继续往下看，还有哪些改进的余地。

第二个版本

如果我们仔细的看一下MSDN上关于GetMessage函数的说明，那么就可以看到MSDN指出了 while( GetMessage(...) ) 的方式是错误的，并给出了如下的形式：

BOOL bRet;

while ( (bRet  =  GetMessage(  & msg, NULL,  0 ,  0  ))  !=   0 )
{ 
     if  (bRet  ==   - 1 )
    {
         //  handle the error and possibly exit
    }
     else
    {
        TranslateMessage( & msg); 
        DispatchMessage( & msg); 
    }
}

原来GetMessage除了在收到WM_QUIT消息的时候返回0之外，在发生错误的时候返回的是-1。在大部分情况下，即使发生了错误，msg也保存了上一次的消息，一个消息处理了两次，我想大部分人都不会察觉到吧。不过，如果我们想自己写一个类似于MFC之类的框架程序或者严于律己的人来说，这点程序的健壮性还是不容忽略的。

 

在继续下一个版本的改进以前，现在模仿MFC或WTL，做一个PreTranslateMessage：

BOOL PreTranslateMessage(LPMSG pMsg)
{
     return  FALSE;
}

BOOL bRet;
while ( (bRet  =  GetMessage(  & msg, NULL,  0 ,  0  ))  !=   0 )
{ 
     if  (bRet  ==   - 1 )
    {
          //  handle the error and possibly exit
    }
     else   if  ( ! PreTranslaeMessage( & msg))
    {
        TranslateMessage( & msg); 
        DispatchMessage( & msg); 
    }
}

PreTranslateMessage函数的作用是非常重要的。由于DispatchMessage函数是将消息分发给各个窗口过程（WindowProc）处理，我觉得有3种情况要放在PreTranslateMessage里处理：

• 全局性的东西，不适宜或不方便放在窗口过程中处理的东西。
• 要处理某些第三方或通用控件的消息。
• 线程消息。

下面要讲的东西，都将放在PreTranslateMessage函数中。

第三个版本

首先要放在PreTraslateMessage函数中的，属于上面3中情况的第一条，全局性的东西，快捷键。

HACCEL hAcc  =  LoadAccelerator(hInst,MAKEINTRESOURCE(IDA_XXX));

BOOL PreTranslateMessage(LPMSG pMsg)
{
     if  (TranslateAccelerator(hWnd,hAcc,pMsg)) 
         return  TRUE;
     return  FALSE;
}

这里要注意的是，除了自己程序定义的快捷键之外，很多ActiveX控件，都暴露出了有TranslateAccelerator的接口。如果没有在PreTranslateMessage中调用的话，那程序一定会缺乏某些使人不方便的行为，比如tab键导航。尤其是嵌入了webbrowser控件的程序,如果想让用户舒适得使用这个内嵌的浏览器的话，一定要调用下面类似的代码：

IWebBrowser2  *  m_pBrowser;

BOOL PreTranslateMessage(LPMSG pMsg)
{
    IOleInPlaceActiveObject  *  pObj;
     if  ( SUCCESSED(m_pBrowser -> QueryInterface(IID_IOleInPlaceActiveObject, & pObj))  &&  
    S_OK  ==  pObj -> TranslateAccelerator(pMsg))
    {
         return  TRUE;
    }
     return  FALSE;
}

第四个版本

如果不查MSDN，是否能立刻说出IsDialogMessage函数的作用呢？IsDialogMessage函数并不仅仅是一个IsXXX的函数，它的作用是：判断一个消息是否为一个对话框的消息，如果是，就处理它。所以，代码应该如下：

BOOL PreTranslateMessage(LPMSG pMsg)
{
     if  (IsDialogMessage(hDlg,pMsg))
         return  TRUE;
     return  FALSE;
}

IsDialogMessage是用来处理对话框上面控件的键盘导航的。例如：当焦点在一个按钮上面的时候，按下tab键，这时应该将焦点设到下一个控件上面，而由于焦点在这个按钮上面，所以只有这个按钮才收得到这个tab键的键盘消息，因此我们需要在消息循环中也就是PreTranslateMessage中调用IsDialogMessage来处理这样的消息。

一般而言，上面的hDlg参数，是一个当前存在的非模态窗口。当然，如MSDN所说，如果一个普通的窗口上面的控件需要使用键盘导航的话，也可以调用IsDialogMessage来处理。那么，为什么上面指定的是非模态窗口，模态窗口不需要了吗？是的，因为模态窗口自带消息循环，用不着我们自己的消息循环。

第五个版本？

我想我暂时是想不出第五个版本了，即便是一个简简单单的消息循环，也还有很多深层次的东西可以挖掘。我们平时用惯了MFC/ATL/WTL之类的框架，它们已经将消息循环封装的很好了，很多东西都已经自动处理了。我想，虽然有些东西我们不需要亲自处理，但是还是需要对此有一定了解的。

最后，就已WTL的消息循环的源代码，结束这篇文章吧：

// /
//  CMessageLoop - message loop implementation

class  CMessageLoop
{
public :
    ATL::CSimpleArray < CMessageFilter *>  m_aMsgFilter;
    ATL::CSimpleArray < CIdleHandler *>  m_aIdleHandler;
    MSG m_msg;

//  Message filter operations
    BOOL AddMessageFilter(CMessageFilter *  pMessageFilter)
    {
         return  m_aMsgFilter.Add(pMessageFilter);
    }

    BOOL RemoveMessageFilter(CMessageFilter *  pMessageFilter)
    {
         return  m_aMsgFilter.Remove(pMessageFilter);
    }

//  Idle handler operations
    BOOL AddIdleHandler(CIdleHandler *  pIdleHandler)
    {
         return  m_aIdleHandler.Add(pIdleHandler);
    }

    BOOL RemoveIdleHandler(CIdleHandler *  pIdleHandler)
    {
         return  m_aIdleHandler.Remove(pIdleHandler);
    }

#ifndef _ATL_NO_OLD_NAMES
     //  for compatilibility with old names only
    BOOL AddUpdateUI(CIdleHandler *  pIdleHandler)
    {
        ATLTRACE2(atlTraceUI,  0 , _T( " CUpdateUIObject and AddUpdateUI are deprecated. Please change your code to use CIdleHandler and OnIdle\n " ));
         return  AddIdleHandler(pIdleHandler);
    }

    BOOL RemoveUpdateUI(CIdleHandler *  pIdleHandler)
    {
        ATLTRACE2(atlTraceUI,  0 , _T( " CUpdateUIObject and RemoveUpdateUI are deprecated. Please change your code to use CIdleHandler and OnIdle\n " ));
         return  RemoveIdleHandler(pIdleHandler);
    }
#endif   //  !_ATL_NO_OLD_NAMES

//  message loop
     int  Run()
    {
        BOOL bDoIdle  =  TRUE;
         int  nIdleCount  =   0 ;
        BOOL bRet;

         for (;;)
        {
             while (bDoIdle  &&   ! ::PeekMessage( & m_msg, NULL,  0 ,  0 , PM_NOREMOVE))
            {
                 if ( ! OnIdle(nIdleCount ++ ))
                    bDoIdle  =  FALSE;
            }

            bRet  =  ::GetMessage( & m_msg, NULL,  0 ,  0 );

             if (bRet  ==   - 1 )
            {
                ATLTRACE2(atlTraceUI,  0 , _T( " ::GetMessage returned -1 (error)\n " ));
                 continue ;    //  error, don't process
            }
             else   if ( ! bRet)
            {
                ATLTRACE2(atlTraceUI,  0 , _T( " CMessageLoop::Run - exiting\n " ));
                 break ;    //  WM_QUIT, exit message loop
            }

             if ( ! PreTranslateMessage( & m_msg))
            {
                ::TranslateMessage( & m_msg);
                ::DispatchMessage( & m_msg);
            }

             if (IsIdleMessage( & m_msg))
            {
                bDoIdle  =  TRUE;
                nIdleCount  =   0 ;
            }
        }

         return  ( int )m_msg.wParam;
    }

     static  BOOL IsIdleMessage(MSG *  pMsg)
    {
         //  These messages should NOT cause idle processing
         switch (pMsg -> message)
        {
         case  WM_MOUSEMOVE:
#ifndef _WIN32_WCE
         case  WM_NCMOUSEMOVE:
#endif   //  !_WIN32_WCE
         case  WM_PAINT:
         case   0x0118 :     //  WM_SYSTIMER (caret blink)
             return  FALSE;
        }

         return  TRUE;
    }

//  Overrideables
     //  Override to change message filtering
     virtual  BOOL PreTranslateMessage(MSG *  pMsg)
    {
         //  loop backwards
         for ( int  i  =  m_aMsgFilter.GetSize()  -   1 ; i  >=   0 ; i -- )
        {
            CMessageFilter *  pMessageFilter  =  m_aMsgFilter[i];
             if (pMessageFilter  !=  NULL  &&  pMessageFilter -> PreTranslateMessage(pMsg))
                 return  TRUE;
        }
         return  FALSE;    //  not translated
    }

     //  override to change idle processing
     virtual  BOOL OnIdle( int   /* nIdleCount */ )
    {
         for ( int  i  =   0 ; i  <  m_aIdleHandler.GetSize(); i ++ )
        {
            CIdleHandler *  pIdleHandler  =  m_aIdleHandler[i];
             if (pIdleHandler  !=  NULL)
                pIdleHandler -> OnIdle();
        }
         return  FALSE;    //  don't continue
    }
};