| 本文將給出一個來源程式,實現在一個對話方塊中的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;
} |
