class CDBLayerTree : public CTreeCtrl
CDBLayerTree派生于CTreeCtrl類
CTreeCtrl派生于CWnd類，樹形結構其實是一個窗口
添加消息響應函數(shù)：
afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
afx_msg void OnMouseMove(UINT nFlags, CPoint point);
afx_msg void OnBegindrag(NMHDR* pNMHDR, LRESULT* pResult);
afx_msg void OnTimer(UINT_PTR nIDEvent);
DECLARE_MESSAGE_MAP()
添加函數(shù)：
HTREEITEM CopyBranch(HTREEITEM htiBranch, HTREEITEM htiNewParent, HTREEITEM htiAfter);
HTREEITEM CopyItem(HTREEITEM hItem, HTREEITEM htiNewParent, HTREEITEM htiAfter);
添加變量：
protected://HPD20161103
UINT m_TimerTicks; //處理滾動的定時器所經(jīng)過的時間
UINT m_nScrollTimerID; //處理滾動的定時器
CPoint m_HoverPoint; //鼠標位置
UINT m_nHoverTimerID; //鼠標敏感定時器
DWORD m_dwDragStart; //按下鼠標左鍵那一刻的時間
BOOL m_bDragging; //標識是否正在拖動過程中
CImageList* m_pDragImage; //拖動時顯示的圖象列表
HTREEITEM m_hItemDragS; //被拖動的標簽
HTREEITEM m_hItemDragD; //接受拖動的標簽
BOOL m_bIsChange;
函數(shù)的實現(xiàn)：
cpp文件加入宏：
#define DRAG_DELAY 60
構造函數(shù)內：
m_bDragging = false;
void CDBLayerTree::OnLButtonDown(UINT nFlags, CPoint point)
{
// TODO: 在此添加消息處理程序代碼和/或調用默認值
//處理無意拖曳
m_dwDragStart = GetTickCount();
CTreeCtrl::OnLButtonDown(nFlags, point);
}
void CDBLayerTree::OnLButtonUp(UINT nFlags, CPoint point)
{
//TODO:在此添加消息處理程序代碼和/或調用默認值
CString strF =_T(""), strT =_T("");
__super::OnLButtonUp(nFlags, point);
//__super是基類的意思
if( m_bDragging )
{
m_bDragging = FALSE;
CImageList::DragLeave( this );
CImageList::EndDrag();
ReleaseCapture();
delete m_pDragImage;
SelectDropTarget( NULL );
if( m_hItemDragS == m_hItemDragD )
{
goto label;
}
HTREEITEM hParentDragS = GetParentItem(m_hItemDragS);
HTREEITEM hParentDragD = GetParentItem(m_hItemDragD);
if(hParentDragS != hParentDragD)
{
goto label;
}
long lLayerHandleS,lLayerHandleD;
lLayerHandleS = GetItemData(m_hItemDragS);
lLayerHandleD = GetItemData(m_hItemDragD);
{
long InitialPosition = -1;
long TargetPosition = -1;
long nCount = -1;
HTREEITEM hNextItem;
HTREEITEM hChildItem = GetChildItem(hParentDragS);
while (hChildItem != NULL)
{
nCount++;
hNextItem = GetNextItem(hChildItem, TVGN_NEXT);
if(hChildItem == m_hItemDragS)
{
InitialPosition = nCount;
break;
}
hChildItem = hNextItem;
}
nCount = -1;
hChildItem = GetChildItem(hParentDragS);
while (hChildItem != NULL)
{
nCount++;
hNextItem = GetNextItem(hChildItem, TVGN_NEXT);
if(hChildItem == m_hItemDragD)
{
TargetPosition = nCount;
HTREEITEM htiNew;
if( InitialPosition < TargetPosition ) //被拖動的往下拖，則顯示在接受拖動的下面
{
htiNew = CopyBranch( m_hItemDragS, GetParentItem(m_hItemDragD), m_hItemDragD );
}
else //被拖動的往上拖，則顯示在接受拖動的上面
{
if( TargetPosition == 0 )
{
htiNew = CopyBranch( m_hItemDragS, GetParentItem(m_hItemDragD), TVI_FIRST );
}
else
{
htiNew = CopyBranch( m_hItemDragS, GetParentItem(m_hItemDragD), GetPrevSiblingItem(m_hItemDragD) );
}
}
DeleteItem( m_hItemDragS );
SelectItem( htiNew );
KillTimer( m_nScrollTimerID );
return;
}
hChildItem = hNextItem;
}
}
label:
KillTimer( m_nScrollTimerID );
m_hItemDragS = NULL;
m_hItemDragD = NULL;
}
//CTreeCtrl::OnLButtonUp(nFlags, point);
}
void CDBLayerTree::OnMouseMove(UINT nFlags, CPoint point)
{
// TODO: 在此添加消息處理程序代碼和/或調用默認值
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 );
//鼠標經(jīng)過時高亮顯示
CImageList::DragShowNolock( false ); //避免鼠標經(jīng)過時留下難看的痕跡
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;
__super::OnMouseMove(nFlags, point);
}
return;
}
//開始拖動的消息響應
void CDBLayerTree::OnBegindrag(NMHDR* pNMHDR, LRESULT* pResult)
{
NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
*pResult = 0;
//如果是無意拖曳，則放棄操作
if( (GetTickCount() - m_dwDragStart) < DRAG_DELAY )
{
smartlog<<"(GetTickCount() - m_dwDragStart) < DRAG_DELAY!";
return;
}
HTREEITEM item = pNMTreeView->itemNew.hItem;
HTREEITEM hParent = GetParentItem(item);
if( !(hParent != NULL && GetParentItem(hParent) == NULL) )
{
smartlog<<"hParent != NULL && GetParentItem(hParent) == NULL!";
return;
}
m_hItemDragS = item;
m_hItemDragD = NULL;
DWORD_PTR pData = GetItemData(m_hItemDragS);
if( long(pData) >= 0 )
m_bDragging = true;
//得到用于拖動時顯示的圖象列表
if( m_bDragging == TRUE )
m_pDragImage = CreateDragImage( m_hItemDragS );
else
m_pDragImage = NULL;
if( !m_pDragImage )
{
smartlog<<"m_pDragImage為NULL!";
return;
}
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 );
}
//定時器響應函數(shù)
void CDBLayerTree::OnTimer(UINT_PTR nIDEvent)
{
// TODO: 在此添加消息處理程序代碼和/或調用默認值
//鼠標敏感節(jié)點
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
__super::OnTimer(nIDEvent);
}
//拷貝條目
HTREEITEM CDBLayerTree::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 ();
//限制拷貝條目數(shù)據(jù)和條目狀態(tài)
SetItemData( hNewItem, GetItemData(hItem) );
SetItemState( hNewItem, GetItemState(hItem, TVIS_STATEIMAGEMASK), TVIS_STATEIMAGEMASK );
return hNewItem;
}
//拷貝分支
HTREEITEM CDBLayerTree::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;
}
