这个例子展示了如何使用itk::TreeContainer和相关的TreeIterators。
- TreeContainer实现了树的概念,并在节点类型上模板化,因此它实际上可以处理任何对象。
- 每个节点应该只有一个父节点,因此树中不存在循环。不会安全检查是否存在循环。
首先需要包含相关的头文件:
#include <itkTreeContainer.h>
#include "itkTreeContainer.h"
#include "itkChildTreeIterator.h"
#include "itkLeafTreeIterator.h"
#include "itkLevelOrderTreeIterator.h"
#include "itkInOrderTreeIterator.h"
#include "itkPostOrderTreeIterator.h"
#include "itkPreOrderTreeIterator.h"
#include "itkRootTreeIterator.h"
#include "itkTreeIteratorClone.h"
然后,我们创建一个整数树。TreeContainer被模板化到节点的类型上。
typedef int NodeType;
typedef itk::TreeContainer<NodeType> TreeType;
TreeType::Pointer tree = TreeType::New();
接下来,我们使用SetRoot()设置根节点的值:
tree->SetRoot(0);
然后使用Add()函数向树中添加节点。第一个参数是新节点的值,第二个参数是父节点的值。如果两个节点具有相同的值,则选择第一个节点。在这种特殊情况下,最好使用迭代器来填充树。
tree->Add(1,0);
tree->Add(2,0);
tree->Add(3,0);
tree->Add(4,2);
tree->Add(5,2);
tree->Add(6,5);
tree->Add(7,1);
我们定义了一个itk::LevelOrderTreeIterator来按照级别顺序解析树。
这个特定的迭代器有三个参数。
- 第一个是实际要解析的树
- 第二个是最大深度级别
- 第三个是开始节点
GetNode() 函数的作用是:返回给定值的节点。再次返回与值对应的第一个节点:
itk::LevelOrderTreeIterator<TreeType> levelIt(tree,10,tree->GetNode(2));
levelIt.GoToBegin();
while(!levelIt.IsAtEnd())
{
std::cout << levelIt.Get() << " ("<< levelIt.GetLevel() << ")" << std::endl;
++levelIt;
}
std::cout << std::endl;
GetType()函数的作用是:返回所使用的迭代器的类型。枚举类型的列表如下:PREORDER、INORDER、POSTORDER、LEVELORDER、CHILD、ROOT和LEAF。
if(childIt.GetType() != itk::TreeIteratorBase<TreeType>::CHILD)
{
std::cout << "[FAILURE]" << std::endl;
return EXIT_FAILURE;
}
每个TreeIterator都有一个Clone()函数,该函数返回当前迭代器的副本。注意,用户应该手动删除创建的迭代器。
childIt.GoToParent();
itk::TreeIteratorBase<TreeType>* childItClone = childIt.Clone();
delete childItClone;
LeafTreeIterator迭代树的叶子:
itk::LeafTreeIterator<TreeType> leafIt(tree);
leafIt.GoToBegin();
while(!leafIt.IsAtEnd())
{
std::cout << leafIt.Get() << std::endl;;
++leafIt;
}
std::cout << std::endl;
InOrderTreeIterator按从左到右的顺序遍历树:
itk::InOrderTreeIterator<TreeType> InOrderIt(tree);
InOrderIt.GoToBegin();
while(!InOrderIt.IsAtEnd())
{
std::cout << InOrderIt.Get() << std::endl;;
++InOrderIt;
}
std::cout << std::endl;
PreOrderTreeIterator从左到右遍历树,但是执行深度优先搜索:
itk::PreOrderTreeIterator<TreeType> PreOrderIt(tree);
PreOrderIt.GoToBegin();
while(!PreOrderIt.IsAtEnd())
{
std::cout << PreOrderIt.Get() << std::endl;;
++PreOrderIt;
}
std::cout << std::endl;
PostOrderTreeIterator从左到右迭代树,但是在搜索中从叶子到根:
itk::PostOrderTreeIterator<TreeType> PostOrderIt(tree);
PostOrderIt.GoToBegin();
while(!PostOrderIt.IsAtEnd())
{
std::cout << PostOrderIt.Get() << std::endl;;
++PostOrderIt;
}
std::cout << std::endl;
树的所有节点都可以使用Clear()函数删除:
tree->Clear();
我们将展示如何使用TreeIterator通过创建节点来形成树。Add()函数用于添加节点并为其赋值。GoToChild()用于跳转到节点:
itk::PreOrderTreeIterator<TreeType> PreOrderIt2(tree);
PreOrderIt2.Add(0);
PreOrderIt2.Add(1);
PreOrderIt2.Add(2);
PreOrderIt2.Add(3);
PreOrderIt2.GoToChild(2);
PreOrderIt2.Add(4);
PreOrderIt2.Add(5);
TreeIteratorClone可以用来拥有迭代器的泛型副本:
typedef itk::TreeIteratorBase<TreeType> IteratorType;
typedef itk::TreeIteratorClone<IteratorType> IteratorCloneType;
itk::PreOrderTreeIterator<TreeType> anIterator(tree);
IteratorCloneType aClone = anIterator;