在 CTreeCtrl 中枚举系统中的所有窗口！(II)

这是实现文件 
 
// WinList.cpp : implementation file 
// 
 
#include "stdafx.h" 
#include "mfc.h" 
#include "WinList.h" 
#include "mfcApi.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CWinList dialog 
 
 
CWinList::CWinList(CWnd* pParent /*=NULL*/) 
        : CDialog(CWinList::IDD, pParent) 
{ 
        //{{AFX_DATA_INIT(CWinList) 
                // NOTE: the ClassWizard will add member initialization here 
        //}}AFX_DATA_INIT 
} 
 
 
void CWinList::DoDataExchange(CDataExchange* pDX) 
{ 
        CDialog::DoDataExchange(pDX); 
        //{{AFX_DATA_MAP(CWinList) 
        DDX_Control(pDX, IDTREE, m_Tree); 
        //}}AFX_DATA_MAP 
} 
 
 
BEGIN_MESSAGE_MAP(CWinList, CDialog) 
        //{{AFX_MSG_MAP(CWinList) 
        ON_BN_CLICKED(IDREFRESH, OnRefresh) 
        ON_NOTIFY(TVN_SELCHANGED, IDTREE, OnSelchangedTree) 
        //}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CWinList message handlers 
 
static CDWordArray WinArray; 
static unsigned long Level; 
static int Ignore[10]; 
static HWND hPrevWnd; 
 
BOOL CALLBACK CWinList::EnumChildProc(HWND hwnd,LPARAM lParam) 
{ 
        Ignore[Level]++; 
        if (Ignore[Level+1]) Ignore[Level+1]--; 
        else 
        {       WinArray.Add((unsigned long)((Level<<16)|(unsigned long)hwnd)); 
                Level++;Ignore[Level+1]=0; 
                EnumChildWindows(hwnd,EnumChildProc,(long)0); 
                Level--; 
        } 
         
        return TRUE; 
} 
 
BOOL CALLBACK CWinList::EnumWindowsProc(HWND hwnd, LPARAM lParam) 
{ 
        WinArray.Add((unsigned long)hwnd); 
        Level=1;Ignore[Level+1]=0; 
        EnumChildWindows(hwnd,EnumChildProc,(long)0); 
        return TRUE; 
} 
 
void CWinList::BuiltTree(HWND hWnd) 
{       char Temp[90]; 
        char Text[50]; 
        char Name[20]; 
 
        TreeItem.hParent=RootArray[RootPt]; 
 
        ::GetWindowText(hWnd,Text,50); 
        ::GetClassName(hWnd,Name,20); 
        sprintf(Temp,"[%04x]:{%s}:%s",hWnd,Name,Text); 
        TreeItem.item.pszText=Temp; 
        TreeItem.item.lParam=(unsigned long)hWnd; 
        HTREEITEM TempItem=m_Tree.InsertItem(&TreeItem); 
        RootArray[RootPt+1]=TempItem; 
} 
 
void CWinList::DrawFlash(HWND hWnd) 
{ 
   HDC  hDC; 
   RECT rc; 
   HPEN hPen; 
 
   ::GetWindowRect(hWnd, &rc); 
   hDC=::GetWindowDC(hWnd); 
   ::SetROP2(hDC,R2_NOT); 
   hPen=::CreatePen(PS_INSIDEFRAME,5*GetSystemMetrics(SM_CXBORDER),RGB(255,0,0)); 
   ::SelectObject(hDC,hPen); 
   ::SelectObject(hDC,GetStockObject(NULL_BRUSH)); 
   ::Rectangle(hDC,0,0,rc.right-rc.left,rc.bottom-rc.top); 
   ::ReleaseDC(hWnd,hDC); 
   ::DeleteObject(hPen); 
} 
 
void CWinList::Refresh() 
{       RootArray[0]=TVI_ROOT; 
        RootPt=0; 
 
        TreeItem.hInsertAfter=TVI_LAST; 
        TreeItem.item.mask=TVIF_TEXT|TVIF_PARAM; 
        WinArray.RemoveAll(); 
        m_Tree.SetFont(&TreeFont); 
        m_Tree.DeleteAllItems(); 
        Level=0; 
        for(int j=0;j<10;j++) Ignore[j]=0; 
        EnumWindows(EnumWindowsProc,(long)0); 
         
        for(int i=0;i<WinArray.GetSize();i++) 
        {       RootPt=(int)((WinArray[i]&0xFFFF0000)>>16); 
                BuiltTree((HWND)(WinArray[i]&0x0000FFFF)); 
        } 
} 
 
BOOL CWinList::OnInitDialog()  
{ 
        CDialog::OnInitDialog(); 
         
        // TODO: Add extra initialization here 
        m_hSelected=NULL; 
        hPrevWnd=NULL; 
        Refresh();       
        return TRUE;  // return TRUE unless you set the focus to a control 
                      // EXCEPTION: OCX Property Pages should return FALSE 
} 
 
void CWinList::OnRefresh()  
{ 
        // TODO: Add your control notification handler code here 
        Refresh();       
} 
 
void CWinList::OnSelchangedTree(NMHDR* pNMHDR, LRESULT* pResult)  
{ 
        NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR; 
        // TODO: Add your control notification handler code here 
 
        HWND hWnd=(HWND)pNMTreeView->itemNew.lParam; 
        m_hSelected=hWnd; 
        if (hPrevWnd) DrawFlash(hPrevWnd); 
        hPrevWnd=hWnd; 
        DrawFlash(hWnd); 
 
        *pResult = 0; 
} 
 
BOOL CWinList::DestroyWindow()  
{ 
        // TODO: Add your specialized code here and/or call the base class 
        if (hPrevWnd) DrawFlash(hPrevWnd); 
        return CDialog::DestroyWindow(); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
//      DLL Export Interface 
 
extern "C" 
HWND FAR PASCAL EXPORT SearchWindow() 
{        
        CWinList WinList; 
        if (WinList.DoModal()==IDOK) 
                return WinList.m_hSelected; 
        else 
                return (HWND)NULL; 
} 
 
 
-- 
┌————┐ 
│ 贝  贝 │  
└————┘