在CTreeCtrl中枚举系统中的所有窗口

最新推荐文章于 2025-05-09 10:00:09 发布

转载最新推荐文章于 2025-05-09 10:00:09 发布 · 900 阅读

文章标签：

#initialization #callback #dialog #construction #null #constructor

VC经典文章专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一个使用CTreeCtrl组件在对话框中以树状形式枚举系统中所有窗口的方法。通过递归调用枚举函数，可以获取到窗口及其子窗口的信息，并在界面上展示出来。

在CTreeCtrl中枚举系统中的所有窗口

作者：unknown 更新时间： 2005-05-07

本文将给出一个源程序,实现在一个对话框中的TreeCtrl中用树的形式枚举窗口。

这是头文件

// WinList.h : header file
//

/////////////////////////////////////////////////////////////////////////////

// CWinList dialog

class CWinList : public CDialog
{
// Construction
public:
HWND m_hSelected;
CWinList(CWnd* pParent = NULL); // standard constructor
static BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam);
static BOOL CALLBACK EnumChildProc(HWND hwnd,LPARAM lParam);
void BuiltTree(HWND hWnd);

// Dialog Data
//{{AFX_DATA(CWinList)
enum { IDD = ID_WINLIST };
CTreeCtrl m_Tree;
//}}AFX_DATA

// Overrides
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CWinList)
public:
virtual BOOL DestroyWindow();
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV
support
//}}AFX_VIRTUAL

// Implementation
protected:
void DrawFlash(HWND hWnd);
void Refresh();

// Generated message map functions
//{{AFX_MSG(CWinList)
virtual BOOL OnInitDialog();
afx_msg void OnRefresh();
afx_msg void OnSelchangedTree(NMHDR* pNMHDR, LRESULT* pResult);
//}}AFX_MSG
DECLARE_MESSAGE_MAP()
private:
CFont TreeFont;
TV_INSERTSTRUCT TreeItem;
int RootPt;
HTREEITEM RootArray[10];
};

下面是实现文件

// 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/td>
{ 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;
}