程序博客网 > 阿里云终端管理

三态树的实现

来源：互联网 发布：阿里云终端管理 编辑：程序博客网 时间：2024/05/17 06:38

    三态树节点有三种状态，分别对应：子节点全部选中、子节点部分选中、没有一个子节点被选中。

   

一.消息响应函数

    这里需要对三个消息进行响应，分别是：NM_CLICK、WM_LBUTTONDOWN、WM_KEYDOWN，对应的响应函数分别是OnStateIconClick、OnLButtonDown和OnKeyDown。

    1.OnLButtonDown

     void Three_State_Tree_Ctrl::OnLButtonDown(UINT nFlags, CPoint point)
    {
         HTREEITEM hItem =HitTest(point, &uFlags_);
         if ( (uFlags_&TVHT_ONITEMSTATEICON ))
        {
             //nState: 0->无选择钮 1->没有选择 2->部分选择 3->全部选择
            UINT nState = GetItemState( hItem, TVIS_STATEIMAGEMASK ) >> 12;
            nState=(nState==3)?1:3;
            SetItemState( hItem, INDEXTOSTATEIMAGEMASK(nState), TVIS_STATEIMAGEMASK );
         }
 
         CTreeCtrl::OnLButtonDown(nFlags, point);
    }

    解释：

    int HitTest( CPoint pt, UINT* pFlags = NULL ) const;第二个参数msdn的解释是：Variable that receives information about the results of a hit test.（接收函数返回的信息）。LVHT_ONITEMSTATEICON：The position is over the state image of a list-view item.（位置处于状态图片上时就会得到LVHT_ONITEMSTATEICON）。

    GetItemState函数的返回值。首先，每个Item对应一个LVITEM结构，这个结构是如下定义的：

        

                          typedef struct _LVITEM {                                                    UINT mask;                                                    int iItem;                                                    int iSubItem;                                                    UINT state;                                                    UINT stateMask;                                                    LPTSTR pszText;                                                    int cchTextMax;                                                    int iImage;                                                    LPARAM lParam;                                           #if (_WIN32_IE >= 0x0300)                                                   int iIndent;                                           #endif                                           #if (_WIN32_IE >= 0x560)                                                   int iGroupId;                                                   UINT cColumns; // tile view columns                                                   PUINT puColumns;                                           #endif                                          } LVITEM, *LPLVITEM; 
只需关注里面的state变量就行了，GetItemState函数正是返回某Item的state参数值。关于state参数值，msdn有注解：Bits 12 
through 15 of this member specify the state image index. If these bits are zero, the item has no state image. To
isolate these bits, use the LVIS_STATEIMAGEMASK mask. To set the state image index, use the INDEXTOSTATEIMAGEMASK 
macro. The state image index specifies the index of the image in the state image list that should be drawn. 
大意是：从12位到15位是留给状态图标索引（在状态图片list中，每个索引对应于一个图片）用的。要想设置或者得到状态信息，使用
 LVIS_STATEIMAGEMASK，设置状态时，需要使用INDEXTOSTATEIMAGEMASK 宏。如果这些为为0的话，该item没有状态图片。
     2.OnStateIconClick
      void Three_State_Tree_Ctrl::OnStateIconClick(NMHDR* pNMHDR, LRESULT* pResult) 
     {
           if(uFlags_&TVHT_ONITEMSTATEICON) *pResult=1;
           else *pResult = 0;
     }
     关于NM_CLICK，当LRESULT为0时，表示单击treectrl时，执行系统默认的处理（默认处理是什么，我还不晓得）；1代表不执行。
所以这段代码是说，当鼠标在treectrl的state image上左键单击的话，不执行默认处理（因为该做的，我们在OnLButtonDown里已
经做了）。
       3.OnKeyDown
       void Three_State_Tree_Ctrl::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags) 
      {
            //处理空格键
            if(nChar==0x20)
            {
                  HTREEITEM hItem =GetSelectedItem();
                  UINT nState = GetItemState( hItem, TVIS_STATEIMAGEMASK ) >> 12;
                  if(nState!=0)
                  {
                       nState=(nState==3)?1:3;
                       SetItemState( hItem, INDEXTOSTATEIMAGEMASK(nState), TVIS_STATEIMAGEMASK );
                  }
             }
            else CTreeCtrl::OnKeyDown(nChar, nRepCnt, nFlags);
         }
         这段代码就是当空格键被按下的时候，改变state image。和OnLButtonDown差不多。
二.  SetItemState函数
      这个函数被重写了。代码如下：
      BOOL Three_State_Tree_Ctrl::SetItemState(HTREEITEM hItem, UINT nState, UINT nStateMask, BOOL bSearch)
     {
           BOOL bReturn=CTreeCtrl::SetItemState( hItem, nState, nStateMask );
           UINT iState = nState >> 12;
           if(iState!=0)
          {
              if(bSearch) TravelChild(hItem, iState);
              TravelSiblingAndParent(hItem,iState);
          }
          return bReturn;
      }
      TravelChild函数做的是：当一个节点被选中的时候，当然要把它的子节点全部设为选中了，太天经地义了。TravelSiblingAndParent
函数做的是：一个节点被选中，就需要检查它的兄弟节点，判断是否大家都被选中了，都选中了，就需要设置它们的父节点被选中，介个也是
天经地义的。
三.  TravelChild和TravelSiblingAndParent函数
       由于代码注释得已经够清楚，给出代码足矣。
void Three_State_Tree_Ctrl::TravelChild(HTREEITEM hItem, int nState)
{
      HTREEITEM hChildItem,hBrotherItem;
 
     //查找子节点，没有就结束
     hChildItem=GetChildItem(hItem);
     if(hChildItem!=NULL)
    {
        //设置子节点的状态与当前节点的状态一致
        CTreeCtrl::SetItemState( hChildItem, INDEXTOSTATEIMAGEMASK(nState), TVIS_STATEIMAGEMASK );
        TravelChild(hChildItem, nState);
  
        //处理子节点的兄弟节点
        hBrotherItem=GetNextSiblingItem(hChildItem);
        while (hBrotherItem)
       {
            //设置子节点的兄弟节点状态与当前节点的状态一致
            int nState1 = GetItemState( hBrotherItem, TVIS_STATEIMAGEMASK ) >> 12;
            if(nState1!=0)
           {
               CTreeCtrl::SetItemState( hBrotherItem, INDEXTOSTATEIMAGEMASK(nState), TVIS_STATEIMAGEMASK );
           }
           TravelChild(hBrotherItem, nState);
           hBrotherItem=GetNextSiblingItem(hBrotherItem);
        }
     }
}
void Three_State_Tree_Ctrl::TravelSiblingAndParent(HTREEITEM hItem, int nState)
{
      HTREEITEM hNextSiblingItem,hPrevSiblingItem,hParentItem;
 
      //查找父节点，没有就结束
      hParentItem=GetParentItem(hItem);
      if(hParentItem!=NULL)
     {
          int nState1=nState;//设初始值，防止没有兄弟节点时出错
  
         //查找当前节点下面的兄弟节点的状态
         hNextSiblingItem=GetNextSiblingItem(hItem);
         while(hNextSiblingItem!=NULL)
         {
              nState1 = GetItemState( hNextSiblingItem, TVIS_STATEIMAGEMASK ) >> 12;
              if(nState1!=nState && nState1!=0) break;
              else hNextSiblingItem=GetNextSiblingItem(hNextSiblingItem);
          }
  
          if(nState1==nState)
         {
             //查找当前节点上面的兄弟节点的状态
             hPrevSiblingItem=GetPrevSiblingItem(hItem);
             while(hPrevSiblingItem!=NULL)
             {
                  nState1 = GetItemState( hPrevSiblingItem, TVIS_STATEIMAGEMASK ) >> 12;
                  if(nState1!=nState && nState1!=0) break;
                  else hPrevSiblingItem=GetPrevSiblingItem(hPrevSiblingItem);
              }
          }
  
          if(nState1==nState || nState1==0)
          {
                nState1 = GetItemState( hParentItem, TVIS_STATEIMAGEMASK ) >> 12;
                if(nState1!=0)
                {
                    //如果状态一致，则父节点的状态与当前节点的状态一致
                    CTreeCtrl::SetItemState( hParentItem, INDEXTOSTATEIMAGEMASK(nState), TVIS_STATEIMAGEMASK );
                }
                //再递归处理父节点的兄弟节点和其父节点
                TravelSiblingAndParent(hParentItem,nState);
          }
          else
          {
               //状态不一致，则当前节点的父节点、父节点的父节点……状态均为第三态
               hParentItem=GetParentItem(hItem);
               while(hParentItem!=NULL)
              {
                    nState1 = GetItemState( hParentItem, TVIS_STATEIMAGEMASK ) >> 12;
                    if(nState1!=0)
                   {
                         CTreeCtrl::SetItemState( hParentItem, INDEXTOSTATEIMAGEMASK(2), TVIS_STATEIMAGEMASK );
                   }
                   hParentItem=GetParentItem(hParentItem);
              }
          }
     } 
}
四. 声明
        代码基本是看做项目时直接从网上找的，并非本人自己所写，觉得有必要总结一下，顾写篇笔记而已。
阿里云终端管理 阿里云终端管理
原创粉丝点击
热门IT博客
everybody网络中文谐音everybody网络中文谐音
soap apache axissoap apache axis
大公教育和知满天
国外网络流行模式
网络信息安全知识竞赛
拓普康全站仪数据
管理系统制作软件
淘宝 一千零一夜视频
怎么把对象传进js函数
python与java效率
sql select case语句
f35c战斗机性能数据
阿里云域名证书在哪里
长得像混血知乎
落叶能知易水寒下句
合成歌曲的软件
溪谷软件
软件企业退税政策
域名绑定时出现未解析
驾校教练 知乎
热门问题 老师的惩罚 人脸识别 我在镇武司摸鱼那些年 重生之率土为王 我在大康的咸鱼生活 盘龙之生命进化 天生仙种 凡人之先天五行 春回大明朝 姑娘不必设防，我是瞎子 led硬灯条 led灯怎么修 欧普照明led灯带 led灯具图片 led灯条 led灯带一米多少瓦 家用led照明灯 led灯带价格 led灯是什么意思 大功率投光灯 筒灯的价格 led灯与白炽灯 雷士led吸顶灯 东风17射程 巨浪2射程 鹰击18射程 有效射程 东风5b射程 火炮射程 坦克射程 东风16射程 东风100射程 射程 钢珠一枪射程200米 东风41射程多远 东风射程多远 50米射程打药喷雾机 周子瑜射箭 赵露思射箭 射箭姿势 射箭馆 射箭技巧 射箭图片 射箭比赛 射箭的好处 成都射箭馆 射箭的技巧 射箭大作战 骑马射箭 射箭俱乐部 射箭对战

程序博客网，程序员的互联网技术博客家园。csdn论坛精品 msdn技术资料都在这里