CTreeCtrl 功能扩展

老狼主

于 2020-03-26 12:48:33 发布

阅读量353

点赞数 1

分类专栏：工作随笔篇 VC++（日积月累篇）

本文链接：https://blog.csdn.net/u012156872/article/details/105116193

版权

VC++（日积月累篇）同时被 2 个专栏收录

137 篇文章 19 订阅

订阅专栏

工作随笔篇

135 篇文章 14 订阅

订阅专栏

1、遍历下一个节点

HTREEITEM getNextItem(CTreeCtrl* pTreeCtrl, HTREEITEM hSelItem)
{
	if (pTreeCtrl->GetChildItem(hSelItem))
		return pTreeCtrl->GetChildItem(hSelItem);
	else if (pTreeCtrl->GetNextSiblingItem(hSelItem))
		return pTreeCtrl->GetNextSiblingItem(hSelItem);
	else if (pTreeCtrl->GetParentItem(hSelItem))
	{
		if (pTreeCtrl->GetNextSiblingItem(pTreeCtrl->GetParentItem(hSelItem)))
			return pTreeCtrl->GetNextSiblingItem(pTreeCtrl->GetParentItem(hSelItem));
		else
		{
			HTREEITEM hItem = pTreeCtrl->GetParentItem(hSelItem);
			while (hItem && !(pTreeCtrl->GetNextSiblingItem(hItem)))
			{
				hItem = pTreeCtrl->GetParentItem(hItem);
			}

			return pTreeCtrl->GetNextSiblingItem(hItem);
		}
	}

	return NULL;
}

2、计算子节点个数

int getChildItemCount(CTreeCtrl* pTreeCtrl, HTREEITEM hItem)
{
	int nRef = 0;
	HTREEITEM hNext = pTreeCtrl->GetChildItem(hItem);
	while (hNext)
	{
		nRef++;
		hNext = pTreeCtrl->GetNextSiblingItem(hNext);
	}

	return nRef;
}