如何实现三态选择树

示例代码运行效果图如下：

有时候我们经常需要实现树的多态选择，本文就介绍一种三态选择树的具体实现。

步骤一：生成一个对话框工程。

步骤二：添加树控件，设置所需的属性。

步骤三：做三态选择框图标，且在OnInitDialog()中初始化树，使树控件与选择框图标所构成的图像列表控件联系，具体实现看源代码。

步骤四：实现树的左击消息，实现三态选择。

01. void CMutiTreeDemoDlg::OnClickTree1(NMHDR* pNMHDR,LRESULT* pResult)
02. {
03. UINT uFlags=0;
04. HTREEITEM hPItem;
05. CPoint point = GetCurrentMessage()->pt;
06. ScreenToClient(&point);
07. point.Offset(-8,-16);
08. HTREEITEM hItem =m_TripleTree.HitTest(point, &uFlags);
09. if ( (uFlags&TVHT_ONITEM) )
10. {
11. hPItem=m_TripleTree.GetParentItem(hItem);
12. int m,n,m1,n1;
13. m_TripleTree.GetItemImage(hItem,m,n);
14. if(m!=1)
15. {
16. m_TripleTree.SetItemImage(hItem,1,1);
17. m_TripleTree.GetItemImage(hItem,m1,n1);
18. TravelChild(hItem, m1);
19. if(hPItem!=NULL)
20. TravelSiblingAndParent(hItem,m1);
21. }
22. else if(m==1)
23. {
24. m_TripleTree.SetItemImage(hItem,0,0);
25. m_TripleTree.GetItemImage(hItem,m1,n1);
26. TravelChild(hItem, m1);
27. if(hPItem!=NULL)
28. TravelSiblingAndParent(hItem,m1);
29. }
30. //  MessageBox("xxxx");
31. }
32. *pResult = 0;  
33. }

这其中，最难实现的是树的遍历，我用递归

      TravelChild(HTREEITEM hItem, int m1);

      TravelSiblingAndParent(HTREEITEM hItem,int m1); 来实现。

代码如下：

view source print ?
01. //递归
02. void CMutiTreeDemoDlg::TravelChild(HTREEITEM hItem,int m)
03. {
04. HTREEITEM hChildItem,hBrotherItem;
05. hChildItem=m_TripleTree.GetChildItem(hItem);
06. m_TripleTree.SetItemImage(hChildItem,m,m);
07. if(hChildItem!=NULL)
08. TravelChild(hChildItem, m);
09. hBrotherItem=m_TripleTree.GetNextSiblingItem(hChildItem);
10. m_TripleTree.SetItemImage(hBrotherItem,m,m);
11. if(hBrotherItem!=NULL)
12. TravelChild(hBrotherItem, m);
13. }
14.  
15. void CMutiTreeDemoDlg::TravelSiblingAndParent(HTREEITEM hItem, int m)
16. {
17. int m1,n1;
18. HTREEITEM hNextSiblingItem,hPrevSiblingItem,hParentItem;
19. m1=m;
20. hNextSiblingItem=m_TripleTree.GetNextSiblingItem(hItem);
21. while(hNextSiblingItem!=NULL)
22. {
23. m_TripleTree.GetItemImage(hNextSiblingItem,m1,n1);
24. if(m1!=m)
25. break;
26. else
27. {
28. hNextSiblingItem=m_TripleTree.GetNextSiblingItem(hNextSiblingItem);
29. if(hNextSiblingItem!=NULL)
30. m_TripleTree.GetItemImage(hNextSiblingItem,m1,n1);
31. }
32. }
33.  
34. hPrevSiblingItem=m_TripleTree.GetPrevSiblingItem(hItem);
35. while(hPrevSiblingItem!=NULL)
36. {
37. m_TripleTree.GetItemImage(hPrevSiblingItem,m1,n1);
38. if(m1!=m)
39. break;
40. else
41. {
42. hPrevSiblingItem=m_TripleTree.GetPrevSiblingItem(hPrevSiblingItem);
43. if(hPrevSiblingItem!=NULL)
44. m_TripleTree.GetItemImage(hPrevSiblingItem,m1,n1);
45. }
46. }
47. if(m1==m)
48. {
49. hParentItem=m_TripleTree.GetParentItem(hItem);
50. if(hParentItem!=NULL)
51. {
52. m_TripleTree.SetItemImage(hParentItem,m1,m1);
53. TravelSiblingAndParent(hParentItem,m1);
54. }
55. }  
56.  
57. if(m1!=m)
58. {
59.  
60. hParentItem=m_TripleTree.GetParentItem(hItem);
61. while(hParentItem!=NULL)
62. {
63. m_TripleTree.SetItemImage(hParentItem,2,2);
64. hParentItem=m_TripleTree.GetParentItem(hParentItem);
65. }
66. }
67.  
68. }