MFC入门示例之树控件(CTreeControl)

//增加按钮
void CMFCApplication8Dlg::OnBnClickedButtonAdd()
{
    //树中添加节点
    CString strText;
    GetDlgItemText(IDC_EDIT1, strText);
    if (strText.GetLength() == 0) {
        AfxMessageBox(TEXT("请输入文本"));
        return;
    }
    HTREEITEM hItem = m_tree.GetSelectedItem();//获取当前被选中的节点
    if (hItem == NULL) 
        hItem = TVI_ROOT;
    
    TVINSERTSTRUCT ts = { 0 };
    ts.hParent = hItem;
    ts.hInsertAfter = TVI_LAST;
    ts.item.pszText = strText.GetBuffer();
    ts.item.mask = TVIF_TEXT;
    HTREEITEM hNewItem = m_tree.InsertItem(&ts);
    m_tree.SelectItem(hNewItem);        //选中新加入的节点
    m_tree.EnsureVisible(hNewItem);        //树太高看不见问题
}

//删除按钮
void CMFCApplication8Dlg::OnBnClickedButtonDel()
{
    //获取当前被选中的节点
    HTREEITEM hItem = m_tree.GetSelectedItem();//获取当前被选中的节点
    if (hItem == NULL) {
        AfxMessageBox(TEXT("请选择一个节点"));
        return;
    }
    HTREEITEM hParent = m_tree.GetParentItem(hItem);
    m_tree.DeleteItem(hItem);
    m_tree.SelectItem(hParent);
}

//修改按钮
void CMFCApplication8Dlg::OnBnClickedButtonUpdate()
{
    //获取当前被选中的节点
    HTREEITEM hItem = m_tree.GetSelectedItem();//获取当前被选中的节点
    if (hItem == NULL) {
        AfxMessageBox(TEXT("请选择一个节点"));
        return;
    }
    CString strText;
    GetDlgItemText(IDC_EDIT1, strText);
    if (strText.GetLength() == 0) {
        AfxMessageBox(TEXT("请输入文本"));
        return;
    }
    m_tree.SetItemText(hItem, strText);
}

//选中新节点事件
void CMFCApplication8Dlg::OnTvnSelchangedTree1(NMHDR *pNMHDR, LRESULT *pResult)
{
    LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
    //获取当前被选中的节点
    HTREEITEM hItem = m_tree.GetSelectedItem();//获取当前被选中的节点
    if (hItem != NULL) {
        CString strText = m_tree.GetItemText(hItem);
        SetDlgItemText(IDC_EDIT1, strText);        //选中节点文本添加到编辑框中
    }

    *pResult = 0;
}