MFC树控件节点拖拽的实现（附带教程和源码）

最新推荐文章于 2024-04-29 16:37:38 发布

Ðоп丶ÇОг1の0иΕ

最新推荐文章于 2024-04-29 16:37:38 发布

阅读量1.7k

点赞数 4

文章标签： mfc c++

原文链接：http://blog.chinaunix.net/uid-7535303-id-2615506.html

版权

本文转自VC程序中树型控件节点拖动的完美实现-yuanma-ChinaUnix博客

原文中源代码已无法下载，于是在CSDN资源中搜索并下载到源码经过测试成功。

下面附上教程和源码，具体功能实现的原理和参考原博客

首先创建MFC应用程序，保持默认设置就够了，添加一个树控件在对话框中

为树控件添加变量m_TreeCtrl

头文件添加新建项XTreeCtrl.h，源文件添加新建项XTreeCtrl.cpp，这是实现节点拖拽的关键代码

XTreeCtrl.h代码如下

#pragma once
#include "afxcmn.h"

class CXTreeCtrl : public CTreeCtrl
{
public:
	CXTreeCtrl();

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 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);

	DECLARE_MESSAGE_MAP()
private:
	HTREEITEM CopyBranch(HTREEITEM htiBranch,HTREEITEM htiNewParent,HTREEITEM htiAfter);
	HTREEITEM CopyItem(HTREEITEM hItem,HTREEITEM htiNewParent,HTREEITEM htiAfter);
};

XTreeCtrl.cpp代码如下

// XTreeCtrl.cpp : implementation file

#include "stdafx.h"
#include "TreeDrag.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)
	ON_NOTIFY_REFLECT(TVN_BEGINDRAG, OnBegindrag)
	ON_WM_MOUSEMOVE()
	ON_WM_LBUTTONUP()
	ON_WM_LBUTTONDOWN()
	ON_WM_TIMER()
END_MESSAGE_MAP()

/
// CXTreeCtrl message handlers

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<nCount-1 ; i++ )
					hItem = GetNextVisibleItem( hItem );
				if( hItem )
					SelectDropTarget( hItem );
				m_hItemDragD = hItem;
				CImageList::DragShowNolock ( true );
			}
		}
	}
	else
		CTreeCtrl::OnTimer(nIDEvent);
}

将以上文件添加好之后，找到声明树控件对象m_TreeCtrl的头文件

将CTreeCtrl改为CXTreeCtrl，并在此头文件中添加#include "XTreeCtrl.h"

添加三个Icon资源，不做修改即可

下面初始化树控件开始测试

首先将树控件的以下属性改为True

添加CImageList m_ImageList; //声明一个图像列表类的对象，用于存放加载的一系列图像，后面需要进行创建

OnInitDialog函数中添加以下代码

    HICON hIcon[3];           //图标句柄数组

	hIcon[0] = AfxGetApp()->LoadIcon(IDI_ICON1);  //加载ico文件并将返回句柄存入数组中
	hIcon[1] = AfxGetApp()->LoadIcon(IDI_ICON2);
	hIcon[2] = AfxGetApp()->LoadIcon(IDI_ICON3);

	m_ImageList.Create(32,32,ILC_COLOR32,3,3); //创建图像列表的对象（大小为32X32像素）
	for (int i = 0; i < 3; i++){m_ImageList.Add(hIcon[i]); }//往图像列表中添加图像
	m_TreeCtrl.SetImageList(&m_ImageList,TVSIL_NORMAL); //为TreeCtrl设置图像列表

	HTREEITEM hRoot;                    //根节点的句柄
	HTREEITEM hChild;                   //一级子节点的句柄
	HTREEITEM hCChild;                  //二级子节点的句柄
	hRoot = m_TreeCtrl.InsertItem(_T("根节点0"),0,0);                        //插入根节点0
	hChild = m_TreeCtrl.InsertItem(_T("一级子节点"), 1, 1, hRoot, TVI_LAST); //插入一级子节点
	m_TreeCtrl.InsertItem(_T("二级子节点"), 2, 2, hChild, TVI_LAST);         //插入二级子节点
	m_TreeCtrl.Expand(hRoot, TVE_EXPAND);			                        //展开根节点
	hRoot = m_TreeCtrl.InsertItem(_T("根节点1"), 0, 0);                      //插入根节点1
	hChild = m_TreeCtrl.InsertItem(_T("一级子节点"), 1, 1, hRoot, TVI_LAST); //插入一级子节点
	m_TreeCtrl.InsertItem(_T("二级子节点"), 2, 2, hChild, TVI_LAST);         //插入二级子节点
	m_TreeCtrl.Expand(hRoot, TVE_EXPAND);		//展开根节点

树控件初始化完成。F5开始调试

测试有一个小bug当移动一级子节点至其二级子节点时，一级子节点不会消失。

以上就是树控件节点拖拽实现的全部过程了

Ðоп丶ÇОг1の0иΕ

关注

4
点赞
踩
16

收藏

觉得还不错? 一键收藏
2
评论
MFC树控件节点拖拽的实现（附带教程和源码）

本文转自VC程序中树型控件节点拖动的完美实现-yuanma-ChinaUnix博客原文中源代码已无法下载，于是在CSDN资源中搜索并下载到源码经过测试成功。下面附上教程和源码，具体功能实现的原理和参考原博客首先创建MFC应用程序，保持默认设置就够了，添加一个树控件在对话框中为树控件添加变量m_TreeCtrl头文件添加新建项XTreeCtrl.h，源文件添加新建项XTreeCtrl.cpp，这是实现节点拖拽的关键代码XTreeCtrl.h代码如下#pragma on..
复制链接

扫一扫