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);
    }
}

有一點需要解釋的是，考慮到非葉子節點一般都是作為分類來使用的，所以添加的時候只添加葉子節點。

至此這個小程序的功能就完成了，但是有點不完美的是不能實現父節點與子節點的級聯選中或者取消選中。

這個功能在我的下篇博客中實現。