1.樹視圖風格:TVS_HASBUTTONS; //在父項旁邊顯示(+)和(-) TVS_HASLINES; //使用線條顯示各項之間的層次 TVS_LINESATROOT;//使用線條鏈接樹視圖控件根部各項2.處理單擊事件:TVN_SELCHANGEDvoid CTreeCtrlDlg::OnTvnSelchangedTree1(NMHDR *pNMHDR, LRESULT *pResult)
{
LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR); // TODO: 在此添加控件通知處理程序代碼 HTREEITEM ht=m_treeCtrl.GetSelectedItem(); CString strSelect=m_treeCtrl.GetItemText(ht); m_strTreeVal=strSelect; UpdateData(FALSE); *pResult = 0;
}3.同時讓自己派生的CMyTreeCtrl類和對話框處理TVN_SELCHANGED消息:只須在CMyTreeCtrl中處理以下消息,並返回FALSE就OK了ON_NOTIFY_REFLECT_EX(TVN_SELCHANGED, OnTvnSelchanged) OnTvnSelchanged的簽名如下BOOL CMyTreeCtrl::OnTvnSelchanged(NMHDR *pNMHDR,LRESULT *pResult)4.編輯標籤:要允許編輯樹視圖控件中的文本,可以設置以下三個步驟 1.設置樹視圖的TVS_EDITLABELS風格TVS_EDITLABE可以通過資源編輯框內部修改(Edit labels),也可以用代碼的方式修改,如下long lStyle=::GetWindowLong(m_treeCtrl.GetSafeHwnd(),GWL_STYLE); lStyle|=TVS_EDITLABELS; ::SetWindowLong(m_treeCtrl.GetSafeHwnd(),GWL_STYLE,lStyle) 2.處理TVN_BEGINLABELEDIT通知消息 //設置相關限制,如限制編輯框字符串長度CEdit*pEdit=GetEditControl(); //獲取當前選中結點編輯框 ASSERT(pEdit); if (pEdit) { pEdit->LimitText(15);//設置編輯框文本長度爲15個字符串 *pResult = 0; } 3.處理TVN_ENDLABLEEDIT通知消息 HTREEITEM hParent=GetParentItem(pTVDispInfo->item.hItem); //獲取父結點 HTREEITEM hChild=GetChildItem(hParent?hParent:TVI_ROOT); //獲取第一個根結點hChild=GetNextSiblingItem(hChild); //獲取下一個兄弟節點if (pTVDispInfo->item.hItem!=hChild) //判斷是否與當前節點相同pTVDispInfo->item.pszText //獲取當前節點的字符串CString strText=GetItemText(hChild); //獲取節點的字符串3.讓樹視圖處理Esc和Enter鍵重載PreTranslateMessage函數BOOL bHandleMsg=FALSE; switch(pMsg->message) { case VK_ESCAPE: case VK_RETURN: if (::GetKeyState(VK_CONTROL)&0x8000)
{
break;
} if (GetEditControl()) { ::TranslateMessage(pMsg); ::DispatchMessage(pMsg); bHandleMsg=TRUE; } break; }4.實現上下文菜單在WM_RBUTTONDOWN消息處理函數上實現上下文菜單5.展開和收起樹視圖結點:HTREEITEM hItem=GetRootItem(); //獲取根結點,可能會有多個根結點ItemHasChildren(hParent) //判斷結點是否有孩子結點hItem=GetChildItem(hParent); //獲取第一個子結點hItem=GetNextSiblingItem(hItem)); //獲取下一個兄弟結點結點Expand(hItem,bExpand?TVE_EXPAND:TVE_COLLAPSE);//展開/疊起結點Select(hItem,TVGN_FIRSTVISIBLE); //設置選中結點CString str=GetItemText(hChild); //獲取結點字符串信息HTREEITEM hCurrSel = GetSelectedItem(); //獲取當前選中結點SelectItem(hNewSel);HTREEITEM hNewSel = HitTest(pt, &nFlags); //判斷座標是否在當前結點範圍內HTREEITEM hItem=InsertItem(dlg.m_strItemText,hItemParent); //插入結點#pragma once//定義文件MyTreeCtrl.h// CMyTreeCtrlclass CMyTreeCtrl : public CTreeCtrl{ DECLARE_DYNAMIC(CMyTreeCtrl)public: CMyTreeCtrl(); virtual ~CMyTreeCtrl();protected: DECLARE_MESSAGE_MAP() void ExpandBranch(HTREEITEM hItem,BOOL bExpand =TRUE);public: void ExpandAllBranches(BOOL bExpand =TRUE); BOOL DoesItemExist(HTREEITEM hItemParent, CString const& strItem); afx_msg void OnRButtonDown(UINT nFlags, CPoint point); afx_msg void OnAddItem(); virtual BOOL PreTranslateMessage(MSG* pMsg); afx_msg void OnTvnBeginlabeledit(NMHDR *pNMHDR, LRESULT *pResult); afx_msg void OnTvnEndlabeledit(NMHDR *pNMHDR, LRESULT *pResult); afx_msg BOOL OnTvnSelchanged(NMHDR *pNMHDR, LRESULT *pResult);};// MyTreeCtrl.cpp : 實現文件#include "stdafx.h"#include "TreeCtrl.h"#include "MyTreeCtrl.h"#include ".mytreectrl.h"#include "AddItemDlg.h" // CMyTreeCtrlIMPLEMENT_DYNAMIC(CMyTreeCtrl, CTreeCtrl)CMyTreeCtrl::CMyTreeCtrl(){}CMyTreeCtrl::~CMyTreeCtrl(){}BEGIN_MESSAGE_MAP(CMyTreeCtrl, CTreeCtrl) ON_WM_RBUTTONDOWN() ON_COMMAND(IDR_ADD_ITEM, OnAddItem) ON_NOTIFY_REFLECT(TVN_ENDLABELEDIT, OnTvnEndlabeledit) ON_NOTIFY_REFLECT(TVN_BEGINLABELEDIT, OnTvnBeginlabeledit) ON_NOTIFY_REFLECT_EX(TVN_SELCHANGED, OnTvnSelchanged)END_MESSAGE_MAP()// CMyTreeCtrl 消息處理程序void CMyTreeCtrl::ExpandBranch(HTREEITEM hItem,BOOL bExpand /* =TRUE */)//展開{ if (ItemHasChildren(hItem)) //判斷是否有孩子結點 { Expand(hItem,bExpand?TVE_EXPAND:TVE_COLLAPSE); //展開/疊起結點 hItem=GetChildItem(hItem);//獲取第一個子結點 do { ExpandBranch(hItem); } while(hItem=GetNextSiblingItem(hItem));//獲取兄弟結點 }}void CMyTreeCtrl::ExpandAllBranches(BOOL bExpand /* =TRUE */){ HTREEITEM hItem=GetRootItem();//獲取根結點,可能會有多個根結點 do { ExpandBranch(hItem,bExpand); } while(hItem=GetNextSiblingItem(hItem)); Select(hItem,TVGN_FIRSTVISIBLE);//設置選中結點}BOOL CMyTreeCtrl::DoesItemExist(HTREEITEM hItemParent, CString const& strItem){ BOOL bDoesItemExist=FALSE; ASSERT(strItem.GetLength()); HTREEITEM hChild=GetChildItem(hItemParent?hItemParent:TVI_ROOT); while (NULL!=hChild&&!bDoesItemExist) { CString str=GetItemText(hChild);//獲取結點字符串信息 if (0==str.CompareNoCase(strItem)) { bDoesItemExist=TRUE; } else { hChild=GetNextSiblingItem(hChild); } } return bDoesItemExist;}void CMyTreeCtrl::OnRButtonDown(UINT nFlags, CPoint point){ // TODO: 在此添加消息處理程序代碼和/或調用默認值 // set focus to the tree control SetFocus(); // map the point that is passed to the // function from client coordinates // to screen coordinates ClientToScreen(&point); // Get the currently selected item HTREEITEM hCurrSel = GetSelectedItem();//獲取當前選中結點 // Figure out which item was right clicked CPoint pt(0, 0); ::GetCursorPos(&pt); ScreenToClient (&pt); HTREEITEM hNewSel = HitTest(pt, &nFlags); // Set the selection to the item that the // mouse was over when the user right // clicked if (NULL == hNewSel) { SelectItem(NULL); } else if (hCurrSel != hNewSel) { SelectItem(hNewSel); SetFocus(); } // Load the context menu CMenu Menu; if (Menu.LoadMenu(IDM_CONTEXT_MENU)) { CMenu* pSubMenu = Menu.GetSubMenu(0); if (pSubMenu!=NULL) { // Display the context menu pSubMenu->TrackPopupMenu( TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, this); } } }void CMyTreeCtrl::OnAddItem()//添加上下文菜單{ // TODO: 在此添加命令處理程序代碼 HTREEITEM hItemParent=GetSelectedItem(); //獲取當前選中結點 CAddItemDlg dlg; if (dlg.DoModal()==IDOK) { if (!DoesItemExist(hItemParent,dlg.m_strItemText)) { HTREEITEM hItem=InsertItem(dlg.m_strItemText,hItemParent); //插入結點 SelectItem(hItem); } else { AfxMessageBox("已存在相同結點"); } } }BOOL CMyTreeCtrl::PreTranslateMessage(MSG* pMsg){ // TODO: 在此添加專用代碼和/或調用基類 BOOL bHandledMsg = FALSE; switch (pMsg->message) { case WM_KEYDOWN: { switch (pMsg->wParam) { case VK_ESCAPE: case VK_RETURN: if (::GetKeyState(VK_CONTROL) & 0x8000) { break; } if (GetEditControl()) { ::TranslateMessage(pMsg); ::DispatchMessage(pMsg); bHandledMsg = TRUE; } break; default: break; } // switch (pMsg->wParam) } // WM_KEYDOWN break; default: break; } // switch (pMsg->message) // continue normal translation and dispatching return (bHandledMsg ?TRUE : CTreeCtrl::PreTranslateMessage(pMsg));}void CMyTreeCtrl::OnTvnBeginlabeledit(NMHDR *pNMHDR, LRESULT *pResult){ LPNMTVDISPINFO pTVDispInfo = reinterpret_cast<LPNMTVDISPINFO>(pNMHDR); // TODO: 在此添加控件通知處理程序代碼 *pResult=1; CEdit*pEdit=GetEditControl(); ASSERT(pEdit); if (pEdit) { pEdit->LimitText(15); *pResult=0; } }void CMyTreeCtrl::OnTvnEndlabeledit(NMHDR *pNMHDR, LRESULT *pResult){ LPNMTVDISPINFO pTVDispInfo = reinterpret_cast<LPNMTVDISPINFO>(pNMHDR); // TODO: 在此添加控件通知處理程序代碼 BOOL bValidItem=FALSE; CString strItem=pTVDispInfo->item.pszText; if (0<strItem.GetLength()) { HTREEITEM hParent=GetParentItem(pTVDispInfo->item.hItem); bValidItem=!DoesItemExist(hParent,strItem); } *pResult = bValidItem;} BOOL CMyTreeCtrl::OnTvnSelchanged(NMHDR *pNMHDR, LRESULT *pResult){ LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR); // TODO: 在此添加控件通知處理程序代碼 TRACE(GetItemText(pNMTreeView->itemNew.hItem)); TRACE(" "); *pResult = 0; return FALSE; //返回FALSE可以讓父窗口進行進一步的處理}