支持三态CheckBox的 MFC TreeCtrl 控件扩展

已于 2023-03-19 11:23:40 修改
阅读量2.8k 收藏 9
点赞数 1
CC 4.0 BY-SA版权
分类专栏： win32 经验 C++ 文章标签： mfc c++
于 2012-08-26 19:57:36 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/ZML086/article/details/7909420
本文介绍如何在MFC中实现一个支持三态CheckBox的TreeCtrl控件，包括自绘CheckBox图片、垂直居中对齐及事件处理等关键步骤。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

前言: 如果选择 MFC 来做界面, 那么, MFC 中各种功能有限的控件一定让你蛋疼不已. 比如, TreeCtrl  的 CheckBox 居然不支持三态以及 CheckBox 居然不能垂直居中对齐. 下面我将为大家介绍如何实现一个支持三态 CheckBox 的 TreeCtrl 控件, 谨作抛砖引玉.

首先, 从 CtreeCtrl 派生一个类, 姑且命名为 CExTreeCtrl.  

第二,响应 NM_CUSTOMDRAW 消息来自绘 TreeItem  (仅仅要支持三态的话不用重绘, 但是要垂直居中的话就免不了了). 另外再上今天的主角, 虚函数 DrawItem(HTREEITEM hItem, LPNMTVCUSTOMDRAW lpNMTVCD), 由它负责绘制 Item. 以下我将详细说明 OnNMCustomdraw 与 DrawItem 的实现:

void CExTreeCtrl::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
{
    LPNMCUSTOMDRAW pNMCD = reinterpret_cast<LPNMCUSTOMDRAW>(pNMHDR);
    LPNMTVCUSTOMDRAW lpNMTVCD = (LPNMTVCUSTOMDRAW) pNMCD;
    //默认处理
    *pResult = CDRF_DODEFAULT;
    // TODO: Add your control notification handler code here
    if(lpNMTVCD->nmcd.dwDrawStage == CDDS_PREPAINT)
    {//返回 CDRF_NOTIFYITEMDRAW 标志, 告诉系统通知你每个绘制的细节
        *pResult = CDRF_NOTIFYITEMDRAW;
        return;
    }
    else if(lpNMTVCD->nmcd.dwDrawStage == CDDS_ITEMPREPAINT) 
   {//在 ItemPrePaint 状态实现自绘 
        CRect rcItem(pNMCD->rc); 
        CPoint ptItem(rcItem.left + 1, rcItem.top + 1); 
        UINT uFlags = 0; 
        HTREEITEM hItem = HitTest(ptItem, &uFlags);
        //取出要绘制的 Item 的句柄 
        if(!hItem) return; DrawItem(hItem, lpNMTVCD); 
        //返回 CDRF_SKIPDEFAULT 标志, 告诉系统跳过默认处理
        *pResult = CDRF_SKIPDEFAULT; return;
    }
}



void CExTreeCtrl::DrawItem( HTREEITEM hItem, LPNMTVCUSTOMDRAW lpNMTVCD )
{
    DWORD dwStyle = GetStyle();
    CRect rcItem;
    CRect rcTemp;
    CPoint ptTemp;
    UINT indent = GetIndent();
    //取得 Item 文本的矩形范围
    GetItemRect(hItem, &rcItem, TRUE);
    CDC dc;
    dc.Attach(lpNMTVCD->nmcd.hdc);

    if(lpNMTVCD->nmcd.uItemState & CDIS_SELECTED)
    {//如果 Item 被选中, 填充背景 (背景色我乱写的)
        dc.FillSolidRect(&rcItem, RGB(255, 123, 128));
    }

    dc.SetBkMode(TRANSPARENT);
    dc.SetTextColor(RGB(0, 0, 0));

    rcTemp = rcItem;
    rcTemp.left += 2;
    rcTemp.top += 2;
    //绘制文本
    dc.DrawText(GetItemText(hItem), &rcTemp, DT_LEFT | DT_SINGLELINE | DT_VCENTER);

    int nImage, nImageSelected;

    //取出与 Item 关联的 ImageList
    CImageList* imageList = GetImageList(TVSIL_NORMAL);
    if(imageList && GetItemImage(hItem, nImage, nImageSelected))
    {//绘制与 Item 相关联的 image
        rcItem.left -= 18;//据我目测,系统本身是在 18px 宽的矩形内绘制图片的
        rcItem.right = rcItem.left + 18;
        CPoint pt(rcItem.left, rcItem.top);
        IMAGEINFO imageInfo = {0};
        imageList->GetImageInfo( ((lpNMTVCD->nmcd.uItemState & CDIS_SELECTED ) ? nImageSelected : nImage), &imageInfo);
        //center alignment
        pt.y += ( rcItem.Height() - imageInfo.rcImage.bottom + imageInfo.rcImage.top ) / 2;
        //pt.x += ( rcItem.Width() - imageInfo.rcImage.right + imageInfo.rcImage.left ) / 2;
        imageList->Draw(&dc, nImage, pt, ILD_NORMAL);
        }

    //取出 CheckBox 的各个选中状态的图片,系统本身的CheckBox只支持两态, 因此得自己准备一张包含各个状态图片的位图资源, 并通过SetImageList(XX, TVSIL_STATE)将它设置到 ExTreeCtrl中, 图片规格(以索引说明): 0-空白图片,1-未选中状态图片, 2-选中状态图片, 3-部分选中状态图片
    imageList = GetImageList(TVSIL_STATE);
    if(imageList && (dwStyle & TVS_CHECKBOXES))
    {//绘制CheckBox相应状态的图片
        //将状态码变换成图片在ImageList中的索引, MFC中有一个宏是把索引变换到StateImageState的, 实际就是 <<12 操作, 因此反过来也成立
        nImage = GetItemState( hItem, TVIS_STATEIMAGEMASK ) >> 12;
        rcItem.left -= 18;//据我目测..., 
        rcItem.right = rcItem.left + 18;
        CPoint pt(rcItem.left, rcItem.top);
        IMAGEINFO imageInfo = {0};
        imageList->GetImageInfo(nImage, &imageInfo);
        //居中对齐
        pt.y += ( rcItem.Height() - imageInfo.rcImage.bottom + imageInfo.rcImage.top ) / 2;
        pt.x += ( rcItem.Width() - imageInfo.rcImage.right + imageInfo.rcImage.left ) / 2;
        imageList->Draw(&dc, nImage, pt, ILD_TRANSPARENT);
    }
    //如果要画线
    if(dwStyle & TVS_HASLINES)
    {
        //创建一个真正的点线画笔
        LOGBRUSH logBrush;
        logBrush.lbColor = RGB(0xa2, 0xa2, 0xa2);
        logBrush.lbStyle = BS_SOLID;
        CPen pen(PS_COSMETIC | PS_ALTERNATE, 1, &logBrush);
        CPen* oldPen = dc.SelectObject(&pen);

        rcTemp = rcItem;
        rcTemp.left -= indent;//从当前向左移动一个 缩进 的量
        rcTemp.right = rcTemp.left + indent;
        ptTemp = rcTemp.CenterPoint();

        //如果 Item 有父 Item 则在自己面前画一个'L'型的线,拐点正是缩进矩形的中心点
        if( GetParentItem(hItem) != NULL )
        {
            //如果 Item 有一个弟弟节点(哥哥排上面), 则在自己面前画一条竖线, 否则画半条
            dc.MoveTo( ptTemp.x - 1,rcTemp.top - 1 );
            if(GetNextSiblingItem(hItem) != NULL)
                dc.LineTo( ptTemp.x - 1, rcTemp.bottom );
            else dc.LineTo( ptTemp.x - 1, pt.y - 1 );
            dc.MoveTo( rcTemp.right, ptTemp.y - 1 );
            dc.LineTo( ptTemp.x - 1, ptTemp.y - 1 );
        }
        //依次绘制各个 Item 的父节点与叔叔节点之间被撑开的部分的连线
        HTREEITEM hItemTemp = hItem;
        while( hItemTemp = GetParentItem(hItemTemp) )
        {
            rcTemp.OffsetRect(-indent, 0);
            ptTemp = rcTemp.CenterPoint();
            if(GetNextSiblingItem(hItemTemp))
            {
                dc.MoveTo( ptTemp.x - 1, rcTemp.top );
                dc.LineTo( ptTemp.x - 1, rcTemp.bottom );
            }
        }
        //显示删除 MFC GDI 对象, 因为以前版本的 MFC 貌似有个BUG, 如果你不显示删除它就不会帮你删除, 这个BUG好像还存在
        dc.SelectObject(oldPen);
        pen.DeleteObject();
        pen.DeleteTempMap();
    }
    //绘制小'+'框
    if(dwStyle & TVS_HASBUTTONS)
    {
        rcItem.left -= indent;
        rcItem.right = rcItem.left + indent;
        //确定绘制范围
        rcTemp.SetRect(0, 0, 9, 9);
        rcTemp.OffsetRect(rcItem.CenterPoint());
        rcTemp.OffsetRect(-5, -5);
        //绘制, 因为使用的是MFC10.0, 因此实际绘制工作可交给 CMFCVisualManager 完成,其实... 自己画难度也不大
        if( ItemHasChildren(hItem) ) CMFCVisualManager::GetInstance()->OnDrawExpandingBox(&dc, rcTemp, ( GetItemState(hItem, TVIS_EXPANDED) & TVIS_EXPANDED ?TRUE:FALSE), RGB(0, 0, 0));
    }
    //完成
    dc.Detach();
}

接下来处理事件, 让支持三态 CheckBox 的TreeCtrl 能够根据自己的状态改变父节点与子节点的状态, 因此, 再添加几个辅助函数来帮助我们完成. 

1, TVM_SETITEM 的响应函数, 让它处理State变化的情况. 当然, 重载SetItemState也是可以的, 但不完美, 因为有些程序员就是喜欢用 SendMessage(TVM_SETITEM, XX,XXX) 来破坏你精心设计好的重载. 因此, 响应该消息就能解决问题. 以下是函数的实现:

LRESULT CExTreeCtrl::DoSetItem( WPARAM wParam, LPARAM lParam )
{

    LPTVITEM lpItem = (LPTVITEM)lParam;
    UINT uItemState = 0;
    //如果是设置 State 就自定义处理
    if(lpItem && (lpItem->mask & TVIF_STATE)) 
    { 
        uItemState = lpItem->state >> 12; 
        //设置父节点与子节点的状态 
        SetParentState(lpItem->hItem, uItemState);
        SetChildrenState(lpItem->hItem, uItemState); 
    } 
    //调用默认处理 
    return DefWindowProc(TVM_SETITEM, wParam, lParam);
}

2, SetParentState 函数, 用来设置父节点状态, 如果本层所有节点状态一致, 则父节点与本层状态一致, 否则父节点应该是部分选中的状态. 以下具体实现:

void CExTreeCtrl::SetParentState( HTREEITEM hCurrentItem, UINT nState )
{
    if(hCurrentItem == NULL) return;
    //如果状态未改变, 则直接返回
    if( (GetItemState(hCurrentItem, TVIS_STATEIMAGEMASK) >> 12) == nState ) return;

    HTREEITEM hParentItem = GetParentItem(hCurrentItem);
    if(hParentItem == NULL) return;
    //从当前节点开始循环遍历, 直到遇到与当前节点状态不一致的节点
    for(HTREEITEM hItemItr = GetNextSiblingItem(hCurrentItem); hItemItr != hCurrentItem; )
    {

        if( (hItemItr != NULL) && (nState != GetItemState(hItemItr, TVIS_STATEIMAGEMASK) >> 12) )
        {//不一致, 则父节点应该为部分选中状态
            nState = 3;
            break;
        }
        hItemItr = GetNextSiblingItem(hItemItr);
        if(hItemItr == NULL) hItemItr = GetChildItem(hParentItem);
    }

    //如果父节点状态与要设置的状态相同, 则返回
    if( (GetItemState(hParentItem, TVIS_STATEIMAGEMASK) >> 12) == nState ) return;
    //递归
    SetParentState(hParentItem, nState);

    TVITEM item;
    item.hItem = hParentItem;
    item.mask = TVIF_STATE;
    item.pszText = (LPTSTR) 0;
    item.iImage = 0;
    item.iSelectedImage = 0;
    item.state = INDEXTOSTATEIMAGEMASK(nState);
    item.stateMask = TVIS_STATEIMAGEMASK;
    item.lParam = 0;
    //调用默认处理过程
    DefWindowProc(TVM_SETITEM, 0, (LPARAM)&item);
}

3, SetChildrenState, 设置子节点的状态. 子节点的状态应该与自身状态一致.

void CExTreeCtrl::SetChildrenState( HTREEITEM hCurrentItem, UINT nState )
{
    if(hCurrentItem == NULL) return;
    //自身状态未改变, 则退出
    if( (GetItemState(hCurrentItem, TVIS_STATEIMAGEMASK) >> 12) == nState ) return;
    //遍历所有子节点, 并处理
    TVITEM item;
    for(HTREEITEM hItemItr = GetChildItem(hCurrentItem); hItemItr != NULL; hItemItr = GetNextSiblingItem(hItemItr))
    {
        SetChildrenState(hItemItr, nState);

        item.hItem = hItemItr;
        item.mask = TVIF_STATE;
        item.pszText = (LPTSTR) 0;
        item.iImage = 0;
        item.iSelectedImage = 0;
        item.state = INDEXTOSTATEIMAGEMASK(nState);
        item.stateMask = TVIS_STATEIMAGEMASK;
        item.lParam = 0;
        //调用默认处理过程
        DefWindowProc(TVM_SETITEM, 0, (LPARAM)&item);
    }
}

最后处理鼠标事件, 因为部分选中状态并不一个用户直接可以用鼠标操作就能设置的状态, 因此要屏蔽掉. 先在CExTreeCtrl中添加一个成员变量 CPoint m_ptClickPoint, 然后响应三个消息: WM_LBUTTONDOWN, WM_LBUTTONDBCLK 和 NM_CLICK, 下以是这三个函数的具体实现:

void CExTreeCtrl::OnNMClick(NMHDR *pNMHDR, LRESULT *pResult)
{
    // TODO: Add your control notification handler code here
    UINT uFlags = 0;
    //取得单击时的Item
    HTREEITEM hItem = HitTest(m_ptClickPoint, &uFlags);
    if(uFlags & TVHT_ONITEMSTATEICON)
    {
        //1: 未选中, 2: 选中, 3: 部分选中
        UINT uItemState = CTreeCtrl::GetItemState(hItem, TVIS_STATEIMAGEMASK) >> 12;
        uItemState = uItemState >=2 ? 1 : 2;
        CTreeCtrl::SetItemState(hItem, INDEXTOSTATEIMAGEMASK(uItemState), TVIS_STATEIMAGEMASK);
    }

    *pResult = 0;
}
void CExTreeCtrl::OnLButtonDblClk(UINT nFlags, CPoint point)
{
    // TODO: Add your message handler code here and/or call default
    UINT uHTFlags = 0;
    HitTest(point, &uHTFlags);
    //如果点击到了CheckBox, 则阻止原来的折叠动作
    if(uHTFlags & TVHT_ONITEMSTATEICON)
    {
        SendMessage(WM_LBUTTONDOWN, uHTFlags, MAKELONG(point.x, point.y));
        return;
    }

    CTreeCtrl::OnLButtonDblClk(nFlags, point);
}

void CLayertree::OnLButtonDown( UINT nFlags, CPoint point )
{
    //记录点击的位置
    m_ptClickPoint = point;
    CTreeCtrl::OnLButtonDown(nFlags, point);
}

好, 至此, 一个基本可用的支持三态CheckBox的TreeCtrl已经完成, 当然其中还有一些细节问题需要处理, 使用方式与原来的控件没有区别, 以下是范例:

CRect rectDummy;
rectDummy.SetRectEmpty();

// Create view:
const DWORD dwViewStyle = WS_CHILD | WS_VISIBLE | TVS_LINESATROOT | TVS_HASBUTTONS | TVS_CHECKBOXES | TVS_SHOWSELALWAYS | TVS_HASLINES | 0;

if (!m_wndTree.Create(dwViewStyle, rectDummy, this, 4))
{
TRACE0("Failed to create file view\n");
return -1;      // fail to create
}

// Load view images:
m_LayertreeImges.Create(IDB_LAYRETREE, 16, 0, RGB(255, 0, 255));
m_CheckStateImages.Create(IDB_CHECK_STATE, 16, 0, RGB(255, 0, 255));
m_wndTree.SetImageList(&m_LayertreeImges, TVSIL_NORMAL);
m_wndTree.SetImageList(&m_CheckStateImages, TVSIL_STATE);
m_wndLayertree.SetItemHeight(24);

最后, 再说一句, C++本来就不擅长界面, 而随着PC硬件水平的提高, 采用C++来做界面所带来的性能优势会越来越不明显..

MFC三态树控件自绘
04-23 2707
前言     在项目中想实现一个三态树，在网上找了一些例子，主要参考了《三态选择树实现终结者 》，其中作者讲解清晰，非常容易理解。其具体实现的思想是以Windows标准的CTreeCtrl类为基类派生一个类CMutiTreeCtrl，我们可以直接使用，这里主要是对使用的过程做下笔记。
MFC中CtreeCtrl控件实现三态复选的总结
qingmoshu的专栏
03-30 3506
对网上现有资料整理总结 网上所能查到的所有关于树形控件三态显示的做法都是添加一个已写好的CTreeCtrl类的派生类，但经过Q姐琢磨和测试，根本无需再创建一个派生类就能实现树形控件的三态复选，下面分享具体做法。 1.给树形控件绑定变量 CTreeCtrl m_tc; 2.添加三态显示框bmp资源，如下 3.添加树形控件三态显示效果相关函数 //.h文件中：  BOOL SetSt
MFC CTreeCtrl复选框3态实现效果
u011269801的专栏
10-22 659
CTreeCtrl复选框3态实现效果
MFC三态树控件
07-02
在很多情况下，我们经常需要实现树的多态选择，如上图所示，当全部子节点选中的情况下，当前节点才被选中，当子节点部分选中时，当前节点处于第三态当全部子节点未选中时，当前节点处于未选中的状态。本资源包括源代码和文档说明，介绍这种三态选择树的具体实现方法。
MFC checkbox使用方法
热门推荐
dearmango
05-25 1万+
MFC中复选框checkbox控件，至少有三种方法对其进行操作，他们是利用Cbutton成员函数GetCheck和SetCheck，第二种是利用CWnd成员函数IsDlgButtonChecked，最后就是把checkbox复选框控件与BOOL型变量相关联。 一、用CButton成员函数 MFC中复选框CheckBox的基类是CButton。那么就可以用GetDlgItem来获取复选框的指针，
MFC使用TreeCtrl控件
思依_vr 的专栏
12-16 2088
在开发过程中TreeCtrl控件的使用也比较广泛，比如我们的资源浏览视图，我这里为大家分享一下我使用TreeCtrl控件的一些技巧。 1.我们新建一个Dlg对话框MFC程序 2.拖入一个TreeCtrl控件，控件ID修改为IDC_TREE，把控件的Has Lines属性选择为True（这样能让根与根下的Item有连接线），把控件的Has Buttons属性选择为True（这样可以在父项
MFC TreeCtrl
weixin_30399821的博客
07-05 204
添加子节点 1 switch(nCurDate) 2 { 3 case 1: 4 m_showDetails.DeleteAllItems( ); 5 hRoot = m_showDetails.InsertItem( "A区", TVI_ROOT); 6 hfirstChild = m_show...
MFC实现树形控件TreeCtrl三态选择树
11-27
在MFC中，我们可以创建一个派生自CTreeCtrl的类，然后重写一些关键函数，如OnSelChanging、OnSelChanged等，以支持三态选择。这些函数会在用户改变节点选择时被调用，我们可以在这些函数中添加逻辑来判断并设置节点...
MFC树控件中实现二态复选框
最新发布
07-18
我们在MFC树控件（CTreeCtrl）中实现二态复选框（Tri-state Checkbox）。二态复选框通常有三种状态：选中（checked）、未选中（unchecked）和部分选中（indeterminate）。这在树控件中特别有用，例如当父节点的子...
MFC 树控件如何隐藏相应节点的复选框
zhonglixianyun的博客
08-03 3474
有时候并不需要树控件所有的节点都显示复选框，这时候就需要隐藏相应节点的复选框，代码如下： m_TreeControl是树控件所关联的CTreeCtrl变量，NrootItem是HTREEITEM定义的根节点； m_TreeControl.SetItemState(NrootItem,INDEXTOSTATEIMAGEMASK(0),TVIS_STATEIMAGEMASK);//隐藏根节点的
CTreeCtrl 使用
jianxiangyi的专栏
11-19 810
CTreeCtrl 使用 原文地址：http://blog.163.com/bluesky_hebo/blog/static/81365302010231105228673/ 用SetItemData()： 1、BOOL CTreeCtrl::SetItemData(HTREEITEM hItem, DWORD dwData); 2、对应的Get函数就是GetItemDat
MFC-三个基本控件EditControl、CheckBoxControl以及ComBoxControl
艺术人生的专栏
03-05 2912
MFC中三个最基本的控件的介绍，它们分别是：EditControl、CheckBoxControl以及ComBoxControl。1. EditControl实现的效果如下所示： 关键代码如下： 点击统计按钮的代码void CEditDemoDlg::OnBnClickedBtnCount() { CString strInput; if (0 == GetDlgItemTe
【MFC】常用控件：树形控件Tree Control
Cain Xcy的博客
09-21 6779
01、目录 文章目录01、目录02、树形控件简介03、树形控件的通知消息04、 树形控件的相关数据结构05、树形控件的创建06、CTreeCtrl类的主要成员函数07、树形控件的应用实例08、小结 02、树形控件简介 树形控件在Windows系统中是很常见的，例如资源管理器左侧的窗口中就有用来显示目录的树形视图。树形视图中以分层结构显示数据，每层的缩进不同，层次越低缩进越多。树形控件的节点一般都由标签和图标两部分组成，图标用来抽象的描述数据，能够使树形控件的层次关系更加清晰。 树形控件在插入新的树节点时
MFC的Tree-view Control树控件Checkbox复选框上下级关联勾选状态
a_dev的专栏
12-04 3248
对带复选框的树控件，有个常见的需求，就是当我们勾选了某个父节点时，如果其下有子节点，则所有子节点也应当处于勾选状态。相应地，某个子节点连同其所有兄弟节点被勾选时，相应地其父节点也应当自动被选中。 如此，当我们对树控件的某个节点进行复选操作时，需要处理两个递归逻辑： 1.判断其是否具有子节点，如果有，则将所有子节点都设置为相同的复选状态。子节点的子节点进入递归。 2.判断其同级所有兄弟节点状态...
MFC控件之TreeCtrl
airbigboy的专栏
02-21 1324
当无法可视化设置控件属性的时候，需要手动对控件属性进行设置。此时需要对控件属性十分了解。在此看一下TreeCtrl控件的属性设置。 一、TreeCtrl 不管是作用TreeCtrl控件还是ListCtrl控件，都得用到几个结构体，如果对这些结构体不了解，那在使用控件时就糊理糊涂的，胡乱修改结构体的值，设置不出想要的结果，也不知改哪里，结果出来了也不知怎么出来的。所以必须对我们所要认知的东西
MFC控件篇之TabCtrl
JamesXing的专栏
08-03 2641
Tab CtrTab属性页控件可以在一个窗口中添加不同的页面，然后在页选择发生改变时得到通知。MFC中使用CTabCtrl类来封装属性页控件的各种操作。通过调用BOOL Create( DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID );创建一个窗口，dwStyle中可以使用以下一些属性页控件的专用风格： TCS_BUTT
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值