CTreeCtrl利用TV_ITEM插入节点与遍历节点

插入树节点

TV_ITEM tvItem;
tvItem.mask = TVIF_TEXT | TVIF_PARAM | TVIF_IMAGE | TVIF_HANDLE | TVIF_SELECTEDIMAGE | TVIS_SELECTED | TVIS_EXPANDED; 
CString TmpStr = "文本内容"
tvItem.pszText =(LPWSTR)(LPCTSTR)TmpStr;
tvItem.cchTextMax = MAX_PATH;
tvItem.lParam  =nID; //额外参数
tvItem.iImage =0;     //未被选中时显示的图片
tvItem.iSelectedImage = 1;  //选中时显示的图片


TV_INSERTSTRUCT tvInsert;
tvInsert.hParent = ParentNode; //确定父节点
tvInsert.hInsertAfter = TVI_LAST;
tvInsert.item = tvItem;       

HTREEITEM RetItm = m_tree.InsertItem(&tvInsert);

 

 

//遍历树节点,代码为获得树控件节点的item.lparam

GetChildItemID(CTreeCtrl* pCtrlTree,HTREEITEM& hNode)
{
    TVITEM item;
    item.mask = TVIF_PARAM;

    if (pCtrlTree->ItemHasChildren(hNode))//是否有孩子节点
    {
        HTREEITEM hNextItem;
        HTREEITEM hChildItem = pCtrlTree->GetChildItem(hNode); 

        while (hChildItem != NULL)
        {
            hNextItem = pCtrlTree->GetNextItem(hChildItem, TVGN_NEXT);
            item.hItem = hChildItem;
            pCtrlTree->GetItem(&item);
            //得到item.lParam
            GetChildOrgID(pCtrlTree,hChildItem);//遍历
            hChildItem = hNextItem;
        }
    }
}

 

 

 

//遍历添加节点


AddTreeItem(CTreeCtrl* pCtrlTree,HTREEITEM& ParentNode, const int nParentID)
{
    int nCount=m_OrgArr.size();//m_OrgArr是一个vector<int>类型的数组
    HTREEITEM RetItm;

    for(int i=0;i<nCount;i++)
    {
        XXX*    pObj    =    (XXX*)(int)m_OrgArr.at(i);

        if(pObj->nParentID==nParentID)//满足要求就插入了
        {
            TV_ITEM tvItem;
            tvItem.mask = TVIF_TEXT | TVIF_PARAM | TVIF_IMAGE | TVIF_HANDLE | TVIF_SELECTEDIMAGE | TVIS_SELECTED | TVIS_EXPANDED; 
            CString TmpStr = pObj->sTest; //(LPSTR)
            tvItem.pszText =(LPWSTR)(LPCTSTR)TmpStr;
            tvItem.cchTextMax = MAX_PATH;
            tvItem.lParam  =pObj->nID;
            tvItem.iImage =0;
            tvItem.iSelectedImage = 1;
            
            TV_INSERTSTRUCT tvInsert;
            tvInsert.hParent = ParentNode;
            tvInsert.hInsertAfter = TVI_LAST;
            tvInsert.item = tvItem;

            HTREEITEM RetItm = pCtrlTree->InsertItem(&tvInsert);
             AddTreeItem(pCtrlTree,RetItm,pObj->nID);
        }
    }

}