虚竹风流灵鹫宫:VC中实现带有背景位图的树型控件

来源:百度文库 编辑:九乡新闻网 时间:2024/04/28 14:19:43
当前许多应用程序都在使用树型控件时为其添加了背景位图,增强的控件的魅力,然而对于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 
}

   四、小结 

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