CTreeCtrl中的根节点被选中则子树全选或全清

      今天，我用CTreeCtrl创建一树形结构，想在点根节点时，则其下的所有子树全部被选中或者全情。我找到了两种方法来实现这一功能：

      第一种方法，就是重载CTreeCtrl下的OnLbuttonDown(UINT nFlags, CPoint point)方法，如下所示：

1. void CTreeCtrlCh::OnLButtonDown(UINT nFlags, CPoint point)
2. {
3.     bool bCheck ;
4.           //找到所点击的节点位置
5.     HTREEITEM hItem = HitTest(point, &nFlags) ;
6.     if (hItem == NULL)
7.     {
8.         return ;
9.     }
10.     CTreeCtrl::OnLButtonDown(nFlags, point) ;
11.     bCheck = GetCheck(hItem) ;
13.          //根节点
14.     HTREEITEM hRoot = GetRootItem() ;
15.           //如果所选节点就是根节点
16.     if (hRoot == hItem)
17.     {        //如果节点状态==true,则其下子树全是选中状态
18.         if (bCheck)
19.         {
20.             hRoot = GetChildItem(hRoot) ;
21.             while (hRoot != NULL)
22.             {
23.                 SetCheck(hRoot, TRUE) ;
24.                 hRoot = GetNextItem(hRoot, TVGN_NEXT) ;
25.             }
26.         }
27.         else
28.         {
29.                       //否则，子树全清
30.             hRoot = GetChildItem(hRoot) ;
31.             while (hRoot != NULL)
32.             {
33.                 SetCheck(hRoot, FALSE) ;
34.                 hRoot = GetNextItem(hRoot, TVGN_NEXT) ;
35.             }
36.         }
37.     }
39. }

 

      第二种方法：添加CTreeCtrl的NM_CLICK单击事件，然后在其事件中进行处理，我的Tree包括了“Has buttons”，“Has lines”，“Lines at root”，所以在事件要对“根节点打开”和”checkbox选中“的操作做出判断，然后在进行子树的全选或全清，示例代码如下所示：

   

1. void CDlgPerfPut::OnClickPutPerfTree(NMHDR* pNMHDR, LRESULT* pResult) 
2. {
3.     BOOL bCheck ;
4.     TVHITTESTINFO ht ;
5.     
6.     GetCursorPos(&ht.pt) ;
7.     m_wndTree.ScreenToClient(&ht.pt) ;
8.     ht.flags = TVHT_ABOVE ;
9.     
10.          //获取鼠标点击的节点的消息
11.     HTREEITEM hItem = m_wndTree.HitTest(&ht) ;
12.     if (hItem == NULL)
13.     {
14.         return ;
15.     }
16.     
17.     HTREEITEM hRoot = m_wndTree.GetRootItem() ;
19.          //选中树的checkbox
20.     if (ht.flags & TVHT_ONITEMSTATEICON)
21.     {
22.               //根节点
23.          if (hItem == hRoot)
24.           {
25.         bCheck = m_wndTree.GetCheck(hRoot) ;
26.                   //根节点被选中
27.         if (!bCheck)
28.         {
29.              hItem = m_wndTree.GetChildItem(hRoot) ;
30.              while (hItem != NULL)
31.              {
32.                 m_wndTree.SetCheck(hItem, TRUE) ;
33.                 hItem = m_wndTree.GetNextItem(hItem, TVGN_NEXT) ;
34.               }
35.          }
36.          else
37.          {
38.              hItem = m_wndTree.GetChildItem(hRoot) ;
39.              while (hItem != NULL)
40.                   {
41.                            m_wndTree.SetCheck(hItem, FALSE) ;
42.                  hItem = m_wndTree.GetNextItem(hItem, TVGN_NEXT) ;
43.               }
44.           }
45.         }
46.       }
47.     
48.     *pResult = 0;
49. }

        以上就是我想到的两种方法了。