自动追呼系统:VC程序中树型控件节点拖动的完美实现

Visual C++中提供的MFC类CtreeCtrl（树型控件）用来显示具有一定层次结构的数据项时方便、直观，所以它已经被广泛地应用在各种软件中，如资源管理器中的磁盘目录就用的是树型控件，我们在编程中也会经常用到这个控件 ，但是这个控件也有缺陷，那就是它并不直接支持拖动节点等高级特性，这使得程序员在编程时使用它受到了很大限制，同时又给软件用户带来了一些不便。为此，本实例通过从 CTreeCtrl 中派生了一个类 CXTreeCtrl ，实现树型控件中节点的拖动。这个类具有如下的功能：⑴ 基本项目条拖动的实现；⑵ 处理项目条的无意拖动；⑶ 能处理项目条拖动过程中的滚动问题；⑷ 拖动过程中节点会智能展开。程序编译运行后的效果如图所示：

图一：树型控件节点拖动示例

一、实现方法

我们针对上述自定义类的实现功能，介绍实现思路和方法。

（1）基本项目条拖动的实现

当我们要拖动树型控件的一个项目条时，树型控件会给它的父窗口发送一个TVN_BEGINDRAG通知消息，所以可以在此消息的响应函数中，调用 CTreeCtrl ：：CreateDragImage （）函数创建表示当前项目条正处在拖动操作中的图象，该函数创建的图象由项目条的图象和标签文本组成。创建了拖动图象后，调用CImageList：：BeginDrag（）函数指定拖动图象的热点位置，然后调用CImageList：：DragEnter（）函数显示拖动图象。接下来处理 WM_MOUSEMOVE 消息用于更新拖动图象，我们想让移动中的图象经过某些项目时高亮度显示，这可以调用 CTreeCtrl ：：SelectDropTarget（） 来实现。在调用 SelectDropTarget（）函数之前，需要先调用CImageList：：DragShowNolock ( false )函数来隐藏图象列表，然后再调用CImageList：：DragShowNolock ( true ) 函数来恢复图象列表的显示，这样就不会在拖动过程中留下难看的轨迹。最后我们处理 WM_LBUTTONUP 消息用于完成拖动操作，在该消息响应函数中，我们需要完成结束拖动图象的显示、删除拖动图象、释放鼠标、节点的拷贝/删除等操作。在节点的拷贝/删除操作中，如果是父节点拖到子节点上，我们可以先将父节点拷到根结点下的临时节点中，再从临时结点处拷到子节点，然后将根结点下的临时节点删除，这样做的目的是防止产生异常。

（2）处理项目条的无意拖动

如果在鼠标按下时不小心移动了鼠标，这时系统就认为产生了一个移动操作，这就产生了误操作。解决这个问题的方法是设置时间延迟，也就是说当用户按下鼠标后必须在原位置停留一段时间，才能激活拖动操作。

（3）处理拖动过程中的滚动问题

当我们拖动树型控件的项目条时，如果目的节点不可见，则需要拖动滚动条或收拢其它一些节点以使得目的节点显示出来，无疑，这会给我们带来很大的不便。为此就要给树型控件添加自动滚动支持。首先设置一个定时器，在 WM_TIMER 消息中检测鼠标的位置，如果靠近树型控件的下边缘，则使得控件向下滚动。靠近上边缘则向上滚动。滚动速度根据鼠标的位置确定。

（4）拖动过程中节点的智能展开

为了实现在拖动过程中鼠标停留在某个节点上一段时间后，该节点会自动展开的功能。设置一个定时器，当鼠标在拖动过程中停止在某个节点上时，定时器被启动，再设置一变量保存当前的鼠标位置。

二、编程步骤

1、 新建一对话框工程DragTree，编辑资源，在对话框中加入一树型控件IDC_TREE ，属性设置为：Has Buttons、Has Lines、Lines at root、Edit Labels、Border；

2、 使用Class Wizard给该控件添加一个成员变量 m_wndTree ，在代码部分将该控件的类型修改为CXTreeCtrl。

3、 在对话框的OnInitDialog（）函数中添加代码，初始化树型控件的项目条；

4、 制作一个图像资源（ID为IDB_TREEIMAGE），其中包含两个小图标，用来作为树型控件项目条的显示图标；

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

三、程序代码

// XTreeCtrl.h : header file

#if !defined(AFX_XTREECTRL_H__3EF12526_EF66_4FD9_A572_59476441D79A__INCLUDED_)

#define AFX_XTREECTRL_H__3EF12526_EF66_4FD9_A572_59476441D79A__INCLUDED_

#if _MSC_VER > 1000

#pragma once

#endif // _MSC_VER > 1000

class CXTreeCtrl : public CTreeCtrl

{

　// Construction

　public:

CXTreeCtrl();

// Attributes

　public:

// Operations

　public:

// Overrides

// ClassWizard generated virtual function overrides

//{{AFX_VIRTUAL(CXTreeCtrl)

//}}AFX_VIRTUAL

// Implementation

　public:

virtual ~CXTreeCtrl();

// Generated message map functions

　protected:

UINT m_TimerTicks; //处理滚动的定时器所经过的时间

UINT m_nScrollTimerID; //处理滚动的定时器

CPoint m_HoverPoint; //鼠标位置

UINT m_nHoverTimerID; //鼠标敏感定时器

DWORD m_dwDragStart; //按下鼠标左键那一刻的时间

BOOL m_bDragging; //标识是否正在拖动过程中

CImageList* m_pDragImage; //拖动时显示的图象列表

HTREEITEM m_hItemDragS; //被拖动的标签

HTREEITEM m_hItemDragD; //接受拖动的标签

//{{AFX_MSG(CXTreeCtrl)

　afx_msg void OnBegindrag(NMHDR* pNMHDR, LRESULT* pResult);

　afx_msg void OnMouseMove(UINT nFlags, CPoint point);

　afx_msg void OnLButtonUp(UINT nFlags, CPoint point);

　afx_msg void OnLButtonDown(UINT nFlags, CPoint point);

　afx_msg void OnTimer(UINT nIDEvent);

//}}AFX_MSG

DECLARE_MESSAGE_MAP()

　private:

HTREEITEM CopyBranch(HTREEITEM htiBranch,HTREEITEM htiNewParent,HTREEITEM htiAfter);

HTREEITEM CopyItem(HTREEITEM hItem,HTREEITEM htiNewParent,HTREEITEM htiAfter);

};

#endif

//////////////////////////////////////////////////////////// CXTreeCtrl

#include "stdafx.h"

#include "DragTree.h"

#include "XTreeCtrl.h"

#ifdef _DEBUG

#define new DEBUG_NEW

#undef THIS_FILE

static char THIS_FILE[] = __FILE__;

#endif

#define DRAG_DELAY 60

CXTreeCtrl::CXTreeCtrl()

{

　m_bDragging = false;

}

CXTreeCtrl::~CXTreeCtrl()

{}

BEGIN_MESSAGE_MAP(CXTreeCtrl, CTreeCtrl)

//{{AFX_MSG_MAP(CXTreeCtrl)

　ON_NOTIFY_REFLECT(TVN_BEGINDRAG, OnBegindrag)

　ON_WM_MOUSEMOVE()

　ON_WM_LBUTTONUP()

　ON_WM_LBUTTONDOWN()

　ON_WM_TIMER()

//}}AFX_MSG_MAP

END_MESSAGE_MAP()

void CXTreeCtrl::OnBegindrag(NMHDR* pNMHDR, LRESULT* pResult)

{

　NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;

　*pResult = 0;

　//如果是无意拖曳，则放弃操作

　if( (GetTickCount() - m_dwDragStart) < DRAG_DELAY )

return;

　m_hItemDragS = pNMTreeView->itemNew.hItem;

　m_hItemDragD = NULL;

　//得到用于拖动时显示的图象列表

　m_pDragImage = CreateDragImage( m_hItemDragS );

　if( !m_pDragImage )

return;

　m_bDragging = true;

　m_pDragImage->BeginDrag ( 0,CPoint(8,8) );

　CPoint pt = pNMTreeView->ptDrag;

　ClientToScreen( &pt );

　m_pDragImage->DragEnter ( this,pt ); //"this"将拖曳动作限制在该窗口

　SetCapture();

　m_nScrollTimerID = SetTimer( 2,40,NULL );

}

void CXTreeCtrl::OnMouseMove(UINT nFlags, CPoint point)

{

　HTREEITEM hItem;

　UINT flags;

　//检测鼠标敏感定时器是否存在,如果存在则删除,删除后再定时

　if( m_nHoverTimerID )

　{

KillTimer( m_nHoverTimerID );

m_nHoverTimerID = 0;

　}

　m_nHoverTimerID = SetTimer( 1,800,NULL ); //定时为 0.8 秒则自动展开

　m_HoverPoint = point;

　if( m_bDragging )

　{

CPoint pt = point;

CImageList::DragMove( pt );

//鼠标经过时高亮显示

CImageList::DragShowNolock( false ); //避免鼠标经过时留下难看的痕迹

if( (hItem = HitTest(point,&flags)) != NULL )

{

　SelectDropTarget( hItem );

　m_hItemDragD = hItem;

}

CImageList::DragShowNolock( true );

//当条目被拖曳到左边缘时，将条目放在根下

CRect rect;

GetClientRect( &rect );

if( point.x < rect.left + 20 )

　m_hItemDragD = NULL;

　}

　CTreeCtrl::OnMouseMove(nFlags, point);

}

void CXTreeCtrl::OnLButtonUp(UINT nFlags, CPoint point)

{

　CTreeCtrl::OnLButtonUp(nFlags, point);

　if( m_bDragging )

　{

m_bDragging = FALSE;

CImageList::DragLeave( this );

CImageList::EndDrag();

ReleaseCapture();

delete m_pDragImage;

SelectDropTarget( NULL );

if( m_hItemDragS == m_hItemDragD )

{

　KillTimer( m_nScrollTimerID );

　return;

}

Expand( m_hItemDragD,TVE_EXPAND );

HTREEITEM htiParent = m_hItemDragD;

while( (htiParent = GetParentItem(htiParent)) != NULL )

{

　if( htiParent == m_hItemDragS )

　{

HTREEITEM htiNewTemp = CopyBranch( m_hItemDragS,NULL,TVI_LAST );

HTREEITEM htiNew = CopyBranch( htiNewTemp,m_hItemDragD,TVI_LAST );

DeleteItem( htiNewTemp );

SelectItem( htiNew );

KillTimer( m_nScrollTimerID );

return;

　}

}

HTREEITEM htiNew = CopyBranch( m_hItemDragS,m_hItemDragD,TVI_LAST );

DeleteItem( m_hItemDragS );

SelectItem( htiNew );

KillTimer( m_nScrollTimerID );

　}

}

HTREEITEM CXTreeCtrl::CopyItem(HTREEITEM hItem, HTREEITEM htiNewParent, HTREEITEM htiAfter) //拷贝条目

{

　TV_INSERTSTRUCT tvstruct;

　HTREEITEM hNewItem;

　CString sText;

　//得到源条目的信息

　tvstruct.item.hItem = hItem;

　tvstruct.item.mask=TVIF_CHILDREN|TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE;

　GetItem( &tvstruct.item );

　sText = GetItemText( hItem );

　tvstruct.item.cchTextMax = sText.GetLength ();

　tvstruct.item.pszText = sText.LockBuffer ();

　//将条目插入到合适的位置

　tvstruct.hParent = htiNewParent;

　tvstruct.hInsertAfter = htiAfter;

　tvstruct.item.mask = TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_TEXT;

　hNewItem = InsertItem( &tvstruct );

　sText.ReleaseBuffer ();

　//限制拷贝条目数据和条目状态

　SetItemData( hNewItem,GetItemData(hItem) );

　SetItemState( hNewItem,GetItemState(hItem,TVIS_STATEIMAGEMASK),TVIS_STATEIMAGEMASK);

　return hNewItem;

}

HTREEITEM CXTreeCtrl::CopyBranch(HTREEITEM htiBranch, HTREEITEM htiNewParent, HTREEITEM htiAfter) //拷贝分支

{

　HTREEITEM hChild;

　HTREEITEM hNewItem = CopyItem( htiBranch,htiNewParent,htiAfter );

　hChild = GetChildItem( htiBranch );

　while( hChild != NULL )

　{

CopyBranch( hChild,hNewItem,htiAfter );

hChild = GetNextSiblingItem( hChild );

　}

　return hNewItem;

}

void CXTreeCtrl::OnLButtonDown(UINT nFlags, CPoint point) //处理无意拖曳

{

　m_dwDragStart = GetTickCount();

　CTreeCtrl::OnLButtonDown(nFlags, point);

}

void CXTreeCtrl::OnTimer(UINT nIDEvent)

{

　//鼠标敏感节点

　if( nIDEvent == m_nHoverTimerID )

　{

KillTimer( m_nHoverTimerID );

m_nHoverTimerID = 0;

HTREEITEM trItem = 0;

UINT uFlag = 0;

trItem = HitTest( m_HoverPoint,&uFlag );

if( trItem && m_bDragging )

{

　SelectItem( trItem );

　Expand( trItem,TVE_EXPAND );

}

　}

　//处理拖曳过程中的滚动问题

　else if( nIDEvent == m_nScrollTimerID )

　{

m_TimerTicks++;

CPoint pt;

GetCursorPos( &pt );

CRect rect;

GetClientRect( &rect );

ClientToScreen( &rect );

HTREEITEM hItem = GetFirstVisibleItem();

if( pt.y < rect.top +10 )

{

　//向上滚动

　int slowscroll = 6 - (rect.top + 10 - pt.y )/20;

　if( 0 == (m_TimerTicks % ((slowscroll > 0) ? slowscroll : 1)) )

　{

CImageList::DragShowNolock ( false );

SendMessage( WM_VSCROLL,SB_LINEUP );

SelectDropTarget( hItem );

m_hItemDragD = hItem;

CImageList::DragShowNolock ( true );

　}

}

else if( pt.y > rect.bottom - 10 )

{

　//向下滚动

　int slowscroll = 6 - (pt.y - rect.bottom + 10)/20;

　if( 0 == (m_TimerTicks % ((slowscroll > 0) ? slowscroll : 1)) )

　{

CImageList::DragShowNolock ( false );

SendMessage( WM_VSCROLL,SB_LINEDOWN );

int nCount = GetVisibleCount();

for( int i=0 ; i

　hItem = GetNextVisibleItem( hItem );

　if( hItem )

SelectDropTarget( hItem );

　m_hItemDragD = hItem;

　CImageList::DragShowNolock ( true );

　}

}

　}

　else

CTreeCtrl::OnTimer(nIDEvent);

}

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

BOOL CDragTreeDlg::OnInitDialog()

{

　CDialog::OnInitDialog();

　…………………….//此处代码省略

　// TODO: Add extra initialization here

　m_image.Create ( IDB_TREEIMAGE,16,1,RGB(255,255,255) );

　m_wndTree.SetImageList ( &m_image,TVSIL_NORMAL );

　HTREEITEM hti1 = m_wndTree.InsertItem ( _T("唐詩"),0,1 );

　HTREEITEM hti2 = m_wndTree.InsertItem ( _T("宋詞"),0,1 );

　HTREEITEM hti3 = m_wndTree.InsertItem ( _T("元曲"),0,1 );

　HTREEITEM hti4 = m_wndTree.InsertItem ( _T("李白"),0,1,hti1 );

　m_wndTree.InsertItem ( _T("靜夜思(床前明月光)"),0,1,hti4 );

　m_wndTree.InsertItem ( _T("將進酒(君不見黃河之水天上來)"),0,1,hti4 );

　m_wndTree.InsertItem ( _T("望廬山瀑布(日照香爐生紫煙)"),0,1,hti4 );

　m_wndTree.InsertItem ( _T("蜀道難(噫吁戲，危乎高哉)"),0,1,hti4 );

　HTREEITEM hti5 = m_wndTree.InsertItem ( _T("杜甫"),0,1,hti1 );

　m_wndTree.InsertItem ( _T("蜀相(丞相祠堂何處尋)"),0,1,hti5 );

　m_wndTree.InsertItem ( _T("春望(國破山河在)"),0,1,hti5 );

　m_wndTree.InsertItem ( _T("茅屋為秋風所破歌(八月秋高風怒號)"),0,1,hti5 );

　HTREEITEM hti6 = m_wndTree.InsertItem ( _T("白居易"),0,1,hti1 );

　m_wndTree.InsertItem ( _T("長恨歌(漢皇重色思傾國)"),0,1,hti6 );

　m_wndTree.InsertItem ( _T("琵琶行並序(潯陽江頭夜送客)"),0,1,hti6 );

　m_wndTree.InsertItem ( _T("李清照"),0,1,hti2 );

　m_wndTree.InsertItem ( _T("柳永"),0,1,hti2 );

　return TRUE; // return TRUE unless you set the focus to a control

}

四、小结

本实例介绍了树型控件如何实现项目条的拖动，它的实现思路主要是利用了定时器，不同的定时器来实现不同的功能，另外，读者朋友在学习中要主要掌握CimageList类的几个成员函数的用法。