CTreeCtrl 父结点联动子结点CheckBox

网上搜索了一番，发现没有一个符合我的要求的，于是自己写。

程序功能实现：在按下父结点的CheckBox后，其子结点全部跟随父结点的CheckBox属性设置。

实现步骤：

1.重载CTreeCtrl的单击事件

void CXMLTree::OnNMClickTree(NMHDR *pNMHDR, LRESULT *pResult)
{
    NM_TREEVIEW* pHdr = (NM_TREEVIEW*)pNMHDR;
    CPoint point;
    UINT uFlag;
    GetCursorPos(&point);
    ScreenToClient(&point);
    HTREEITEM hItem = m_TreeCtrl.HitTest(point, &uFlag);
    if((hItem) && (TVHT_ONITEMSTATEICON & uFlag))   
    {   
        // 发送消息，在进行父节点和子节点关联之前判断CheckBox前后的状态
        PostMessage(UM_CHECKSTATECHANGE, 0, (LPARAM)hItem);
    }
    else
    {
        *pResult = 0;
    }
}

2.重载双击时间，屏蔽掉因双击展开结点而改变CheckBox的状态

void CXMLTree::OnNMDblclkTree(NMHDR *pNMHDR, LRESULT *pResult)
{
          *pResult = TRUE;
}

3.写自己Post的Message函数

#define UM_CHECKSTATECHANGE (WM_USER + 100)

//Message Map里面加这句
ON_MESSAGE(UM_CHECKSTATECHANGE, &CXMLTree::OnMyMessage)

afx_msg LRESULT CXMLTree::OnMyMessage(WPARAM wParam, LPARAM lParam)
{
    HTREEITEM hItem = (HTREEITEM)lParam;

    DWORD data = m_TreeCtrl.GetItemData(hItem);
    BOOL bCheck = m_TreeCtrl.GetCheck(hItem);
    CString cStrTemp = m_TreeCtrl.GetItemText(hItem);
    //AfxMessageBox(cStrTemp);

    if(data==1 && !bCheck) 
    {
        //AfxMessageBox(_T("Check状态从check到uncheck"));
        //SetChildCheck(hItem, 0); 
        SetParentChildNodeCheck(hItem, 0);  //按下CheckBox后触发的动作
        m_TreeCtrl.SetItemData(hItem, 0);
    }
    if(data==0 && bCheck) {
        //AfxMessageBox(_T("Check状态从uncheck到check"));
        //SetChildCheck(hItem, 1); 
        SetParentChildNodeCheck(hItem, 1); //按下CheckBox后触发的动作

        m_TreeCtrl.SetItemData(hItem, 1);
    }
    return 0L;
}

4.写递归函数，实现父结点关联子结点

BOOL CXMLTree::SetParentChildNodeCheck(HTREEITEM hItem, BOOL bCheck)
{
    if (hItem==NULL)
        return FALSE;

    HTREEITEM hChild = m_TreeCtrl.GetChildItem(hItem);
    SetChildCheck(hChild, bCheck); 
    m_TreeCtrl.SetCheck(hItem, bCheck);
    return TRUE;
}



//递归函数
void CXMLTree::SetChildCheck(HTREEITEM hItem, BOOL bCheck)
{
    if (hItem == NULL) return;

    m_TreeCtrl.SetCheck(hItem, bCheck); 

    HTREEITEM hSibling = m_TreeCtrl.GetNextItem(hItem, TVGN_NEXT);
    while (hSibling)
    {
        m_TreeCtrl.SetCheck(hSibling, bCheck); 
        HTREEITEM hSibChild = m_TreeCtrl.GetChildItem(hSibling);

        if (hSibChild)
        {
            SetChildCheck(hSibChild, bCheck);
        }

        hSibling = m_TreeCtrl.GetNextItem(hSibling, TVGN_NEXT);
    }
    HTREEITEM hChild = m_TreeCtrl.GetChildItem(hItem);
    if (hChild!=NULL)
    {
        SetChildCheck(hChild, bCheck);
    }
}