VC中实现带有背景位图的树型控件

当前许多应用程序都在使用树型控件时为其添加了背景位图，增强的控件的魅力，然而对于Visual C++编程爱好者来说，使用Visual C++MFC提供的树型控件（CTreeCtrl）本身就是一个难点，至于如何使该控件能够带有背景位图，那就更加是一个令人困惑的问题了。本实例对CTreeCtrl类进行了增强，不仅使它带有背景位图，而且解决了在点击树型控件时背景位图闪动的问题，另外，通过在对话框中使用该控件来显示三级目录，演示了树型控件的基本使用方法。下图为程序编译后的运行效果图：

图一、带背景图的树型控件效果图

　　 一、实现方法

　　在实现树型控件的背景位图之前，我们首先介绍一下树型控件的基本使用方法。树形控件在系统中大量被使用，例如Windows资源管理器就是一个典型的例子。树形控件可以用于树形的结构，其中有一个根接点(Root)然后下面有许多子结点，而每个子结点上又允许有一个或多个或没有子结点。

　　MFC中使用CTreeCtrl类来封装树形控件的各种操作，通过调用BOOL Create( DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID );创建一个窗口，dwStyle中可以使用以下一些树形控件的专用风格：TVS_HASLINES 在父/子结点之间绘制连线；TVS_LINESATROOT 在根/子结点之间绘制连线；TVS_HASBUTTONS 在每一个结点前添加一个按钮，用于表示当前结点是否已被展开；TVS_EDITLABELS 结点的显示字符可以被编辑；TVS_SHOWSELALWAYS 在失去焦点时也显示当前选中的结点；TVS_DISABLEDRAGDROP 不允许Drag/Drop；TVS_NOTOOLTIPS 不使用ToolTip显示结点的显示字符。

　　在树形控件中每一个结点都有一个句柄（HTREEITEM），同时添加结点时必须提供的参数是该结点的父结点句柄（其中根Root结点只有一个，既不可以添加也不可以删除），利用HTREEITEM InsertItem( LPCTSTR lpszItem, HTREEITEM hParent = TVI_ROOT, HTREEITEM hInsertAfter = TVI_LAST )可以添加一个结点，pszItem为显示的字符，hParent代表父结点的句柄，当前添加的结点会排在hInsertAfter表示的结点的后面，返回值为当前创建的结点的句柄。

　　如果你希望在每个结点前添加一个小图标，就必需先调用CTreeCtrl类的成员函数CImageList* SetImageList( CImageList * pImageList, int nImageListType )，指明当前控件所使用的图像列表（ImageList），nImageListType为TVSIL_NORMAL。在调用完成后控件中使用图片以设置的ImageList中图片为准。然后调用HTREEITEM InsertItem( LPCTSTR lpszItem, int nImage, int nSelectedImage, HTREEITEM hParent = TVI_ROOT, HTREEITEM hInsertAfter = TVI_LAST)添加结点，其中参数nImage为结点没被选中时所使用图片序号，nSelectedImage为结点被选中时所使用图片序号。

　　此外CTreeCtrl还提供了一些函数用于得到/修改控件的状态：

　　·HTREEITEM GetSelectedItem( )将返回当前选中的结点的句柄；

　　·BOOL SelectItem( HTREEITEM hItem )将选中指明结点；

　　·BOOL GetItemImage( HTREEITEM hItem, int& nImage, int& nSelectedImage ) / BOOL SetItemImage( HTREEITEM hItem, int nImage, int nSelectedImage )用于得到/修改某结点所使用图标索引；

　　·CString GetItemText( HTREEITEM hItem ) /BOOL SetItemText( HTREEITEM hItem, LPCTSTR lpszItem )用于得到/修改某一结点的显示字符；

　　·BOOL DeleteItem( HTREEITEM hItem )用于删除某一结点，BOOL DeleteAllItems( )将删除所有结点。

　　此外如果想遍历树可以使用下面的函数：

　　·HTREEITEM GetRootItem( )得到根结点；

　　·HTREEITEM GetChildItem( HTREEITEM hItem )得到子结点；

　　·HTREEITEM GetPrevSiblingItem/GetNextSiblingItem( HTREEITEM hItem )得到指明结点的上/下一个兄弟结点；

　　·HTREEITEM GetParentItem( HTREEITEM hItem )得到父结点。

　　树形控件的消息映射使用ON_NOTIFY宏，形式如同：ON_NOTIFY( wNotifyCode, id, memberFxn )，wNotifyCode为通知代码，id为产生该消息的窗口ID，memberFxn为处理函数，函数的原型如同void OnXXXTree(NMHDR* pNMHDR, LRESULT* pResult)，其中pNMHDR为一数据结构，在具体使用时需要转换成其他类型的结构。对于树形控件可能取值和对应的数据结构为：

　　·TVN_SELCHANGED 在所选中的结点发生改变后发送，所用结构：NMTREEVIEW；

　　·TVN_ITEMEXPANDED 在某结点被展开后发送，所用结构：NMTREEVIEW；

　　·TVN_BEGINLABELEDIT 在开始编辑结点字符时发送，所用结构：NMTVDISPINFO；

　　·TVN_ENDLABELEDIT 在结束编辑结点字符时发送，所用结构：NMTVDISPINFO；

　　·TVN_GETDISPINFO 在需要得到某结点信息时发送，（如得到结点的显示字符）所用结构：NMTVDISPINFO；

　　对于Visual C++ MFC提供的标准树型控件CTreeCtrl来说，并不支持背景位图，所以如果需要实现背景位图就需要先让其在内存CDC对象上对TREEVIEW缺省绘图，然后在选择背景位图，与缺省位图合成，即采用贴图的方式，把标准的TREEVIEW窗口贴在底图上。这个操作在内存中完成。同时为了避免闪烁，必须重载OnItemexpanding（）和OnItemexpanded（）这两个函数。SetRedraw函数主要保证其不要在子节点弹出时重画，而是在子节点已经扩展后重画。为此，例程中定义了一个CTreeCtrl类的子类CmyTreeCtrl，并重载了以下几个成员函数：

BOOL CMyTreeCtrl::SetBKImage(LPCTSTR LpszResource)
void CMyTreeCtrl::OnPaint()
void CMyTreeCtrl::OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult)
void CMyTreeCtrl::OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult)
BOOL CMyTreeCtrl::OnEraseBkgnd(CDC* pDC)
　　 二、编程步骤

　　1、 启动Visual C++6.0，生成一个基于对话框的项目Tree，在框架上放置一个树形控制件，其ID标志符为：IDC_TREE1；

　　2、 创建CmyTreeCtrl类后，使用CLASSWIZARD为其添加消息映射：

ON_NOTIFY_REFLECT(TVN_ITEMEXPANDED,OnItemexpanded) ON_NOTIFY_REFLECT(TVN_ITEMEXPANDING, OnItemexpanding)
　　消息响应函数：

afx_msg void OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult);
afx_msg void OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult);
　　3、 将树型控件与CmyTreeCtrl类建立关联，在对话框中添加变量CMyTreeCtrl m_CtrlTree；

　　4、 制作一个准备作为树形控件背景的位图；

　　5、 修改对话框的初始化函数BOOL CTreeDlg::OnInitDialog（）；

　　6、 添加代码，编译运行程序。

三、实现代码

/////////////////////////////////////////////////
#if !defined(AFX_TREEDLG_H__D82DB384_F574_44A7_96DA_6EC9068E22B1__INCLUDED_)
#define AFX_TREEDLG_H__D82DB384_F574_44A7_96DA_6EC9068E22B1__INCLUDED_
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

/////////////////////////////////////////// CTreeDlg dialog
#include "MyTreeCtrl.h"
class CTreeDlg : public CDialog
{
　// Construction
　public:
　　CTreeDlg(CWnd* pParent = NULL); // standard constructor

　　// Dialog Data
　　//{{AFX_DATA(CTreeDlg)
　　　enum { IDD = IDD_TREE_DIALOG };
　　　CMyTreeCtrl m_CtrlTree;
　　//}}AFX_DATA

　　// ClassWizard generated virtual function overrides
　　//{{AFX_VIRTUAL(CTreeDlg)
　　　protected:
　　　　virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
　　//}}AFX_VIRTUAL

　　// Implementation
　protected:
　　HICON m_hIcon;

　　// Generated message map functions
　　//{{AFX_MSG(CTreeDlg)
　　　virtual BOOL OnInitDialog();
　　　afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
　　　afx_msg void OnPaint();
　　　afx_msg HCURSOR OnQueryDragIcon();
　　//}}AFX_MSG
　　DECLARE_MESSAGE_MAP()
};

#endif

////////////////////////////////// MyTreeCtrl.cpp : implementation file

#include "StdAfx.h"
#include "Tree.h"
#include "MyTreeCtrl.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

////////////////////////////// CMyTreeCtrl

CMyTreeCtrl::CMyTreeCtrl()
{}

CMyTreeCtrl::~CMyTreeCtrl()
{}

BEGIN_MESSAGE_MAP(CMyTreeCtrl, CTreeCtrl)
　//{{AFX_MSG_MAP(CMyTreeCtrl)
　　ON_WM_PAINT()
　　ON_WM_ERASEBKGND()
　　ON_NOTIFY_REFLECT(TVN_ITEMEXPANDED, OnItemexpanded)
　　ON_NOTIFY_REFLECT(TVN_ITEMEXPANDING, OnItemexpanding)
　//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMyTreeCtrl message handlers
BOOL CMyTreeCtrl::SetBKImage(LPCTSTR LpszResource)
{
　// if this is not the first call then delete gdi objects
　if( m_bitmap.m_hObject != NULL )
　　m_bitmap.DeleteObject();

　　HBITMAP hbmp = (HBITMAP)::LoadImage(AfxGetInstanceHandle(),
LpszResource, IMAGE_BITMAP, 0, 0, LR_CREATEDIBSECTION|LR_LOADFROMFILE);

　if( hbmp == NULL )
　　return FALSE;
　m_bitmap.Attach( hbmp );
　return TRUE;
}

LRESULT CMyTreeCtrl::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
　// TODO: Add your specialized code here and/or call the base class
　return CTreeCtrl::WindowProc(message, wParam, lParam);
}

void CMyTreeCtrl::OnPaint()
{
　CPaintDC dc(this); // device context for painting
　CRect rcclient;
　GetClientRect(&rcclient);

　// create a compatible memory dc
　CDC memdc;
　memdc.CreateCompatibleDC(&dc);
　CBitmap bitmap;
　bitmap.CreateCompatibleBitmap(&dc, rcclient.Width(), rcclient.Height());
　memdc.SelectObject( &bitmap );
　CWnd::DefWindowProc(WM_PAINT, (WPARAM)memdc.m_hDC , 0);
　CDC maskdc;
　maskdc.CreateCompatibleDC(&dc);
　CBitmap maskbitmap;
　maskbitmap.CreateBitmap(rcclient.Width(), rcclient.Height(), 1, 1, NULL);
　maskdc.SelectObject( &maskbitmap );
　maskdc.BitBlt( 0, 0, rcclient.Width(), rcclient.Height(), &memdc,
　rcclient.left, rcclient.top, SRCCOPY);
　CBrush brush;
　brush.CreatePatternBrush(&m_bitmap);
　dc.FillRect(rcclient, &brush);
　memdc.SetBkColor(RGB(0,0,0));
　memdc.SetTextColor(RGB(255,255,255));
　memdc.BitBlt(rcclient.left, rcclient.top, rcclient.Width(), rcclient.Height(), &maskdc, rcclient.left, 　rcclient.top, SRCAND);
　dc.SetBkColor(RGB(255,255,255));
　dc.SetTextColor(RGB(0,0,0));
　dc.BitBlt(rcclient.left, rcclient.top, rcclient.Width(), rcclient.Height(), &maskdc, rcclient.left, rcclient.top, SRCAND);
　dc.BitBlt(rcclient.left, rcclient.top, rcclient.Width(), rcclient.Height(), &memdc, rcclient.left, rcclient.top,SRCPAINT);
　brush.DeleteObject();
}

BOOL CMyTreeCtrl::OnEraseBkgnd(CDC* pDC)
{
　// TODO: Add your message handler code here and/or call default
　return TRUE;
}

void CMyTreeCtrl::OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult)
{
　NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
　// TODO: Add your control notification handler code here
　Invalidate();
　SetRedraw(TRUE);
　*pResult = 0;
}

void CMyTreeCtrl::OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult)
{
　NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
　// TODO: Add your control notification handler code here
　SetRedraw(FALSE);

　*pResult = 0;
}

///////////////////////////////////////////////////////
BOOL CTreeDlg::OnInitDialog()
{
　CDialog::OnInitDialog();
　// Add "About..." menu item to system menu.
　// IDM_ABOUTBOX must be in the system command range.
　ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
　ASSERT(IDM_ABOUTBOX < 0xF000);
　CMenu* pSysMenu = GetSystemMenu(FALSE);
　if (pSysMenu != NULL)
　{
　　CString strAboutMenu;
　　strAboutMenu.LoadString(IDS_ABOUTBOX);
　　if (!strAboutMenu.IsEmpty())
　　{
　　　pSysMenu-> AppendMenu(MF_SEPARATOR);
　　　pSysMenu-> AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
　　}
　}
　// Set the icon for this dialog. The framework does this automatically
　// when the application's main window is not a dialog
　SetIcon(m_hIcon, TRUE); // Set big icon
　// TODO: Add extra initialization here
　m_CtrlTree.SetBKImage("IDB_BITMAP1");
　SetIcon(m_hIcon, FALSE); // Set small icon
　TVINSERTSTRUCT tvInsert;
　tvInsert.hParent = NULL;
　tvInsert.hInsertAfter = NULL;
　tvInsert.item.mask = TVIF_TEXT;
　tvInsert.item.pszText = _T("Visual C++编程实例");
　HTREEITEM hCountry = m_CtrlTree.InsertItem(&tvInsert);
　HTREEITEM hPA = m_CtrlTree.InsertItem(TVIF_TEXT,
　_T("文章中心"), 0, 0, 0, 0, 0, hCountry, NULL);
　HTREEITEM hWA = m_CtrlTree.InsertItem(_T("代码中心"),0, 0, hCountry, hPA);
　m_CtrlTree.InsertItem(_T("全屏幕程序的实现"), hPA, TVI_SORT);
　m_CtrlTree.InsertItem(_T("实现窗口的任意分割"), hPA, TVI_SORT);
　m_CtrlTree.InsertItem(_T("实现菜单的自绘"), hPA, TVI_SORT);
　m_CtrlTree.InsertItem(_T("实现全屏幕显示的代码"), hWA, TVI_SORT);
　m_CtrlTree.InsertItem(_T("窗口任意分割的代码"), hWA, TVI_SORT);
　m_CtrlTree.InsertItem(_T("菜单自绘代码"), hWA, TVI_SORT);
　m_CtrlTree.Expand(hCountry,TVE_EXPAND);
　return TRUE; // return TRUE unless you set the focus to a control
}
　　 四、小结

　　到此为止，本例通过实现树形控件的背景位图介绍了一些树视图控件编程方法，包括树视图控件的建立、节点值的赋予等。当然，它应用的方面很广，使用方法也很多。这里仅仅是涉及到了构建树视图控件的基本框架，读者朋友们可以在此基础上，可进行扩展，从而完成更强大的功能，感兴趣的读者不妨自己扩展该控件试试。