CTreeCtrl 控件实现多选并取得选中项

刚开始以为实现起来很难，所以就在网上寻找实现的扩展控件，到最后才发现只要把CTreeCtrl 控件的Check Boxes 属性设为真就可以了，会在每个树形节点前添加一个CheckBox。

多选已经实现了，下面就是取得选中的项了，在网上搜了以下也没有太好的办法，于是就自己动手写。大概思路是，遍历整个树形控件，访问每个节点并判断该节点是否被选中，使用CTreeCtrl类的成员函数GetCheck()来判断，返回真假。

实例的界面如下：

代码如下：

首先是插入测试节点函数，在OnInitDialog()函数中调用

void CCheckBoxTreeDlg::InsertNode()
{
    HTREEITEM hRootItem, hSubItem;
    CString strTest("TestNode");
    for(int i = 0;i < 10;i++)
    {
        hSubItem = m_treeTest.InsertItem(strTest + TCHAR(48 + i), 0, 0, TVI_ROOT, TVI_LAST);
        m_treeTest.InsertItem(strTest + TCHAR(48 + i) + "child", 0, 0, hSubItem, TVI_LAST);
    }

    hRootItem = m_treeTest.GetRootItem();
    if(NULL != hRootItem)
    {
        hSubItem = m_treeTest.GetNextSiblingItem(hRootItem);
        CString strText = m_treeTest.GetItemText(hSubItem);
    }
}

接着是在增加按钮的点击消息处理函数中获取选中的节点并插入到右边的CListBox控件中

// 双击添加按钮
void CCheckBoxTreeDlg::OnBnClickedBtnAdd()
{
    // TODO: 在此添加控件通知处理程序代码
    HTREEITEM hRoot = m_treeTest.GetRootItem();

    while(NULL != hRoot)
    {
        CheckAndAddNode(hRoot);
        hRoot = m_treeTest.GetNextSiblingItem(hRoot);
    }
}
// 递归遍历树形控件
void CCheckBoxTreeDlg::CheckAndAddNode(HTREEITEM hRoot)
{
    // 如果不是叶子节点
    if(TRUE == m_treeTest.ItemHasChildren(hRoot))
    {
        if(TRUE == m_treeTest.GetCheck(hRoot))
        {
            // 插入所有页节点
            InsertAllLeaves(hRoot);
        }
        else
        {
            // 查询所有节点，递归
            HTREEITEM hChild = m_treeTest.GetChildItem(hRoot);
            while(NULL != hChild)
            {
                CheckAndAddNode(hChild);
                hChild = m_treeTest.GetNextSiblingItem(hChild);
            }
        }
    }
    else // 是叶子节点
    {
        // 被选中
        if(TRUE == m_treeTest.GetCheck(hRoot))
        {
            m_listClients.InsertString(m_listClients.GetCount(), m_treeTest.GetItemText(hRoot));
        }
    }
}
// 插入所有页节点
void CCheckBoxTreeDlg::InsertAllLeaves(HTREEITEM hRoot)
{
    HTREEITEM hChild = m_treeTest.GetChildItem(hRoot);
    while(NULL != hChild)
    {
        if(TRUE == m_treeTest.ItemHasChildren(hChild))
        {
            InsertAllLeaves(hChild);
        }
        else
        {
            m_listClients.InsertString(m_listClients.GetCount(), m_treeTest.GetItemText(hChild));
        }
        hChild = m_treeTest.GetNextSiblingItem(hChild);
    }
}

有一点需要解释的是，考虑到非叶子节点一般都是作为分类来使用的，所以添加的时候只添加叶子节点。

至此这个小程序的功能就完成了，但是有点不完美的是不能实现父节点与子节点的级联选中或者取消选中。

这个功能在我的下篇博客中实现。