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