应用SetProp等API设置HWND的属性表

做GUI时经常需要让控件或者控件的每个Item附带一些数据.ListCtrl和TreeCtrl等复杂的控件都有类似于SetItem之类的API函数可以附加自己的数据.对于窗口句柄HWND,也有类似的函数,就是SetProp、GetProp等API。具体使用MSDN有详细的介绍和例子。下面我为方便使用而对其进行简要地封装.

#ifndef   _WINDOW_PROP_H_
#define   _WINDOW_PROP_H_

/*
*  封装API函数SetProp、GetProp、EnumPropsEx、RemoveProp
*  用于管理窗口的属性
*/
class   CWindowProp
{
private :
     // 保存需要设置的窗口句柄
    HWND  m_hWnd;

public :
    CWindowProp( HWND hWnd  =  NULL )
    {
         this -> m_hWnd   =   hWnd;
    }

    CWindowProp &   operator = ( HWND hWnd )
    {
         this -> m_hWnd  =  hWnd;
         return   * this ;
    }

     /*
    * 设置属性值
    * lpszName: 属性key
    * hData: 值句柄
    * Return: 返回成功与否
     */
    BOOL  SetProperty(  LPCTSTR lpszName, HANDLE hData )
    {
        ATLASSERT( ::IsWindow(  this -> m_hWnd ) );
         return  ::SetProp(  this -> m_hWnd, lpszName, hData );
    }

     /*
    * 获取属性值
    * lpszName: 属性key
    * Return: lpszName对应的值,若不存在返回NULL
     */
    HANDLE  GetProperty( LPCTSTR lpszName )
    {
        ATLASSERT( ::IsWindow(  this -> m_hWnd ) );
         return  ::GetProp(  this -> m_hWnd, lpszName );
    }

     /*
    * 删除属性值
    * lpszName: 属性key
    * Return: lpszName对应的值,若不存在返回NULL
     */
    HANDLE  RemoveProperty( LPCTSTR lpszName )
    {
        ATLASSERT( ::IsWindow(  this -> m_hWnd ) );
         return   ::RemoveProp(   this -> m_hWnd, lpszName );
    }

     /*
    * 枚举属性列表
     */
    BOOL EnumerateProperties( PROPENUMPROCEX pfnProc, LPARAM lParam)
    {
        ATLASSERT( ::IsWindow(  this -> m_hWnd ) );
        ATLASSERT(  ! ::IsBadCodePtr( (FARPROC)pfnProc ) );
         return  ::EnumPropsEx(  this -> m_hWnd, pfnProc, lParam)  !=   - 1 ;
    }  

     /*
    *  删除所有的属性列表
     */
     void  RemoveAll()
    {
        ATLASSERT( ::IsWindow(  this -> m_hWnd ) );
        ::EnumPropsEx(  this -> m_hWnd, sRemoveAllProc,  0  ); 
    }

private :
     static  BOOL CALLBACK sRemoveAllProc( HWND hWnd, LPTSTR lpszName, HANDLE hData, ULONG_PTR dwData )
    {
        ::RemoveProp( hWnd, lpszName );
         return  TRUE;
    }

};

#endif