MFC

CTreeCtrl

MFC

Posted by YiMiTuMi on July 29, 2020

CTreeCtrl常用函数

CTreeCtrl常用函数

HTREEITEM hItem=GetRootItem(); //获取根结点,可能会有多个根结点

ItemHasChildren(hParent) //判断结点是否有孩子结点

hItem=GetChildItem(hParent); //获取第一个子结点

hItem=GetNextSiblingItem(hItem); //获取下一个兄弟结点结点

BOOL CTreeCtrl::ItemHasChildren(HTREEITEM hItem); //查询hItem是否有子项

HTREEITEM CTreeCtrl::GetChildItem(HTREEITEM hItem); //取得hItem 第一个子项的句柄

HTREEITEM CTreeCtrl::GetPrevSiblingItem(HTREEITEM hItem); //查询排在hItem前兄弟项

HTREEITEM CTreeCtrl::GetNextSiblingItem(HTREEITEM hItem); //查询排在hItem后兄弟项

GetNextItem参数

TVGN_CARET 获取当前被选择的项。

TVGN_CHILD 获取第一个子项。hItem参数必须是NULL。

TVGN_DROPHILITE 获取是一次拖放操作的目标的项。

TVGN_FIRSTVISIBLE 获取第一个可见的项。

TVGN_TEXT 获取下一个兄弟项。

TVGN_NEXTVISIBLE 获取跟随在指定项之后的下一个可视项。

TVGN_PARENT 获取指定项的父项。

TVGN_PREVIOUS 获取前一个兄弟项。

TVGN_PREVIOUSVISIBLE 获取在指定项之前的第一个可视项。

TVGN_ROOT 获取根项的第一个子项，指定项是该根项的一个部分。

CTreeCtrl生成文件树状图

CTreeCtrl生成文件树状图。

void FilePath(CString strFindPath, HTREEITEM hItem/* = NULL*/, BOOL bInsertData/* = TRUE*/)
{
	BOOL bFileExist = FALSE;
	if (hItem == NULL) //一开始获取根节点
	{
		HTREEITEM hRoot = m_treePath.GetRootItem();
		if (hRoot == NULL)
		{
			hRoot = m_treePath.InsertItem(L"此电脑");
		}
		hItem = hRoot;
	}

	CString strPath;
	int iLocation = strFindPath.Find('\\');
	if (iLocation != -1)
	{
		strPath = strFindPath.Mid(0, iLocation);
		strFindPath = strFindPath.Mid(iLocation + 1);
		bFileExist = TRUE;
	}
	else
	{	
		strPath = strFindPath;
		bFileExist = FALSE;
	}

	//循环遍历当前节点下的叶子节点是否和插入的重复GetNextSiblingItem
	HTREEITEM hBrotherItem = m_treePath.GetNextSiblingItem(hItem); //获取下一个节点的兄弟节点
	HTREEITEM hNextItem = m_treePath.GetChildItem(hItem); //获取子节点

	while (bInsertData)
	{
		if (hNextItem == hBrotherItem)
		{
			bInsertData = FALSE;
			break;
		}

		if (hNextItem != NULL)
		{
			CString strItemText = m_treePath.GetItemText(hNextItem);
			if (strItemText == strPath)
			{
				break;
			}
		}

		hNextItem = m_treePath.GetNextItem(hNextItem, TVGN_NEXTVISIBLE); //这个只会获取同级别的下一个节点
	}

	if (bInsertData)
	{
		hItem = hNextItem;
	}
	else
	{
		hItem = m_treePath.InsertItem(strPath, hItem);
	}
	
	if (bFileExist)
	{
		FilePath(strFindPath, hItem, bInsertData);
	}
	return;
}

CTreeCtrl点击消息

消息：

ON_NOTIFY(NM_CLICK, IDC_TREE_xxxx, &xxxxxxx::OnNMClickTree1)

消息函数：

void xxxx::OnNMClickTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
	CTreeCtrl *ctreectrl = (CTreeCtrl *)GetDlgItem(IDC_TREE1);
	
	CPoint pt = GetCurrentMessage()->pt;
	ctreectrl->ScreenToClient(&pt); 
	
	UINT uFlags = 0;  
	HTREEITEM hItem = ctreectrl->HitTest(pt, &uFlags);

	if ((hItem != NULL) && (TVHT_ONITEM & uFlags))//如果点击的位置是在节点位置上
	{  
		HTREEITEM hRoot = m_treePath.GetRootItem();  //获取根节点

	}

	*pResult = 0;
}

获取我的电脑、我的文档、桌面等图标添加

构造函数：

//获取计算机图标
SHFILEINFO sfi_root;
LPITEMIDLIST pidl = NULL;
SHGetSpecialFolderLocation(NULL, CSIDL_DRIVES, &pidl);
SHGetFileInfo((LPCTSTR) pidl, 0, &sfi_root, sizeof(sfi_root), SHGFI_DISPLAYNAME|SHGFI_ICON | SHGFI_PIDL);//如果只获取图标，可以去掉				SHGFI_DISPLAYNAME
HICON hIcon  = sfi_root.hIcon;
ASSERT(hIcon);
m_ilTreeIcons.Add(sfi_root.hIcon); //0

SHFILEINFO sfi_drive;
//获取驱动器图标
wchar_t driver_name[4] = L"C:\\";
::SHGetFileInfo(driver_name,0, &sfi_drive, sizeof(sfi_drive), SHGFI_DISPLAYNAME|SHGFI_ICON);
m_ilTreeIcons.Add(sfi_drive.hIcon); //1
	
//路径
hIcon = AfxGetApp()->LoadIcon(IDI_ICON_OPEN_FOLDER);//2
m_ilTreeIcons.Add (hIcon);
	
//桌面
wchar_t path_Desktop[512]={0};
SHGetSpecialFolderPath(0,path_Desktop,CSIDL_DESKTOPDIRECTORY,0); 
SHFILEINFO sfi_DeskTop;
::SHGetFileInfo(path_Desktop,0, &sfi_DeskTop, sizeof(sfi_DeskTop),SHGFI_ICON);
m_ilTreeIcons.Add(sfi_DeskTop.hIcon);  //3
	
//我的文档
wchar_t path_Document[512] = {0};
SHGetSpecialFolderPath(0,path_Document,CSIDL_MYDOCUMENTS,0); 
SHFILEINFO sfi_Document;
::SHGetFileInfo(path_Document,0, &sfi_Document, sizeof(sfi_Document),SHGFI_ICON);
m_ilTreeIcons.Add(sfi_Document.hIcon); //4

初始化函数：

m_TreePath.ModifyStyle(0, WS_VISIBLE|TVS_HASBUTTONS|TVS_HASLINES|TVS_LINESATROOT|TVS_SHOWSELALWAYS|TVS_TRACKSELECT);
m_TreePath.SetImageList(&m_ilTreeIcons, TVSIL_NORMAL);

//插入
hRoot = m_TreePath.InsertItem(L"此电脑");
//m_TreePath.InsertItem(L"桌面", hRoot);
//m_TreePath.InsertItem(L"我的文档",hRoot);
m_TreePath.InsertItem(L"桌面", 3, 3, hRoot, TVI_LAST);
m_TreePath.InsertItem(L"我的文档", 4, 4, hRoot, TVI_LAST);

展开消息事件

Tree.h

afx_msg void OnTvnGetdispinfoTree(NMHDR *pNMHDR, LRESULT *pResult);
afx_msg void OnTvnItemexpandingTree(NMHDR *pNMHDR, LRESULT *pResult);

class CLASS_TREE_DATA_INFO
{
public:
	BOOL bIsExpanded;
	CLASS_TREE_DATA_INFO()
	{
	     bIsExpanded = FALSE;
	}
};

消息

ON_NOTIFY(TVN_GETDISPINFO, IDC_TREE_xxx, &Tree::OnTvnGetdispinfoTree)
ON_NOTIFY(TVN_ITEMEXPANDING, IDC_TREE_xxx, &Tree::OnTvnItemexpandingTree)

Tree.cpp

void Tree::OnTvnGetdispinfoTree(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMTVDISPINFO pTVDispInfo = reinterpret_cast<LPNMTVDISPINFO>(pNMHDR);
	// TODO: 在此添加控件通知处理程序代码
	pTVDispInfo->item.cChildren = 1;
	*pResult = 0;
}

void Tree::OnTvnItemexpandingTree(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
	// TODO: 在此添加控件通知处理程序代码
	//树形节点展开时会调用该函数
	HTREEITEM hItem = pNMTreeView->itemNew.hItem;

	//通过判断节点是否有数据判断是否展开过，产开过就返回
	CLASS_TREE_DATA_INFO *pInfo = (CLASS_TREE_DATA_INFO*)m_Tree.GetItemData(hItem);
	if (pInfo == NULL)
	{
		*pResult = 0;
		return;
	}
	if (pInfo->bIsExpanded)
	{
		*pResult = 0;
		return;
	}

	pInfo->bIsExpanded = TRUE;
	m_Tree.SetItemData(hItem,DWORD_PTR(pInfo));

	//根据层级获取信息
	int iChildrenCount = 0;

	{
		//再此添加事件代码
		
	}

	//if (iChildrenCount)
	//{
	//	TVITEM item;	
	//	item.hItem = hItem;
	//	item.mask = TVIF_HANDLE|TVIF_CHILDREN|TVIF_DI_SETITEM;
	//	//m_tree.GetItem(&item);
	//	item.mask = TVIF_CHILDREN;
	//	item.cChildren = 0;
	//	m_TreePath.SetItem(&item);
	//}

	//SetRedraw(TRUE);

	Invalidate();
	*pResult = 0;
}

建一个树行表用来展示软件分类等

建表语句：

IF NOT EXISTS (SELECT * FROM sys.objects WHERE object_id = OBJECT_ID(N'[dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_]') AND type in (N'U'))
BEGIN
CREATE TABLE [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_](
	[index] [int] IDENTITY(1,1) NOT NULL,
	[GroupName] [nvarchar](64) NULL,/*组名*/
	[ParentIndex] int NULL,/*父级的index -1*/
	[Comment] [nvarchar](64) NULL,/*备注信息*/
 CONSTRAINT [PK_CLIENT_SOFTWARE_INFO_GROUP_TABLE_] PRIMARY KEY CLUSTERED 
(
	[index] ASC
)WITH (PAD_INDEX  = OFF, IGNORE_DUP_KEY = OFF) ON [PRIMARY]
) ON [PRIMARY]

INSERT INTO [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_] ([GroupName], [ParentIndex]) VALUES (N'软件分类', -1)
INSERT INTO [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_] ([GroupName], [ParentIndex]) VALUES (N'办公软件', 1)
INSERT INTO [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_] ([GroupName], [ParentIndex]) VALUES (N'聊条软件', 1)
INSERT INTO [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_] ([GroupName], [ParentIndex]) VALUES (N'媒体工具', 1)
INSERT INTO [dbo].[CLIENT_SOFTWARE_INFO_GROUP_TABLE_] ([GroupName], [ParentIndex]) VALUES (N'下载工具', 1)
END

根据数据建立树形图：

class CGroupInfo
{
public:
	int iIndex;//表中索引值
	int iParentIndex;//父组的索引
	CString strGroupName;//父组名
	CGroupInfo()
	{
		iIndex = 0;
		iParentIndex = 0;
	}
};

map<int, HTREEITEM> m_mapTreeGroupItem; //保存index对应的节点句柄
map<int, CGroupInfo*> m_mapGroupInfo; //保存数据库数据
MyTreeCtrl m_treeCtrl; //树控件

BOOL initializationCTree()
{
	//查找数据库
	while (TRUE)
	{
		CGroupInfo *pInfo = new CGroupInfo;

		/* 将数据保存到map
		RS_GET_VALUE_RET_FALSE((&pRs), _T("index"), pInfo->iIndex);
		RS_GET_VALUE_RET_FALSE((&pRs), _T("GroupName"), pInfo->strGroupName);
		RS_GET_VALUE_RET_FALSE((&pRs), _T("ParentIndex"), pInfo->iParentIndex);
		*/

		m_mapGroupInfo.insert(std::make_pair(pInfo->iIndex, pInfo));
		
	}
	
	//建立树
	map<int, CGroupInfo*>::iterator iter = m_mapGroupInfo.begin();
	for (; iter != m_mapGroupInfo.end(); iter++)
	{
		CGroupInfo *pInfo = iter->second;
		FindAndAddParentGroup(*pInfo);
	}

	//选中根节点
	HTREEITEM hItemRoot = m_treeCtrl.GetRootItem();
	if (hItemRoot != NULL)
	{
		
		m_treeCtrl.SelectItem(hItemRoot);

		m_treeCtrl.SelectSetFirstVisible(hItemRoot);
		
	}
	m_treeCtrl.Expand(hItemRoot, TVE_EXPAND);


}

//建树函数

//查找组,如果没有,那么就添加
HTREEITEM FindAndAddParentGroup(CGroupInfo &Info)
{
	HTREEITEM htree = FindGroup(Info.iIndex);
	if (htree)
	{
		return htree;
	}

	//添加
	if (Info.iParentIndex < 0)
	{
		HTREEITEM hTempItem = m_treeCtrl.InsertItem(Info.strGroupName, TVI_ROOT);
		TRACE("iterm iserted:%p", hTempItem);
		m_treeCtrl.SetItemData(hTempItem, DWORD_PTR(&Info));
		htree = hTempItem;
		m_mapTreeGroupItem.insert(std::make_pair(Info.iIndex, hTempItem));

		//m_treeCtrl.Expand(TVI_ROOT, TVE_EXPAND );
	}
	else
	{
		map<int, CGroupInfo*>::iterator iter = m_mapGroupInfo.find(Info.iParentIndex);
		if (iter != m_mapGroupInfo.end())
		{
			//
			CGroupInfo *pInfo = iter->second;
			htree = FindGroup(*pInfo);
			if (htree)
			{
				HTREEITEM hTempItem = m_treeCtrl.InsertItem(Info.strGroupName, htree, TVI_SORT);
				m_mapTreeGroupItem.insert(std::make_pair(Info.iIndex, hTempItem));
				m_treeCtrl.SetItemData(hTempItem, DWORD_PTR(&Info));
				//m_treeCtrl.Expand(hTempItem, TVE_EXPAND );
				htree = hTempItem;
			}
			else
			{
				htree = FindAndAddParentGroup(*pInfo);
				if (htree)
				{
					HTREEITEM hTempItem = m_treeCtrl.InsertItem(Info.strGroupName, htree, TVI_SORT);
					m_mapTreeGroupItem.insert(std::make_pair(Info.iIndex, hTempItem));
					m_treeCtrl.SetItemData(hTempItem, DWORD_PTR(&Info));
					//m_treeCtrl.Expand(htree, TVE_EXPAND );
					htree = hTempItem;
				}
			}
		}
		else
		{
			return NULL;
		}
	}

	return htree;
}


HTREEITEM FindGroup(int iGroupIndex)
{
	map<int, CGroupInfo*>::iterator iter = m_mapTreeGroupItem.find(iGroupIndex);
	if (iter != m_mapTreeGroupItem.end())
	{
		return iter->second;
	}
	else
	{
		return NULL;
	}
}

先设定根节点，数据库查表时根据index从小到大排序，新插入的节点index肯定比父节点的index大，所以从头遍历表进行插入。CGroupInfo new出来的记得释放：

map<int, CGroupInfo*>::iterator iter = m_mapGroupInfo.begin();
	for (; iter != m_mapGroupInfo.end(); iter++)
	{
		CEncryptGroupLevelInfo *pInfo = iter->second;
		delete pInfo;
		pInfo = NULL;
	}

圣诞蔷薇 – 追忆的爱情

CATALOG
    FEATURED TAGS
    MySQL 网络协议 windows QT编程 MFC 汇编语言 c++ windows原理 Driver PE文件 Objective-C 数据结构与算法
    FRIENDS