enumerate_tree.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_TREE_ENUMERATE_TREE_HPP
#define MLPACK_CORE_TREE_ENUMERATE_TREE_HPP

namespace mlpack {
namespace tree  {
namespace enumerate {

// Actual implementation of the enumeration. The problem is the unified
// detection if we're on the root, because Enter and Leave expect the
// parent being passed.
template <class TreeType, class Walker>
void EnumerateTreeImpl(TreeType* tree, Walker& walker, bool root)
{
  if (root)
    walker.Enter(tree, (const TreeType*)nullptr);

  const size_t numChildren = tree->NumChildren();
  for (size_t i = 0; i < numChildren; ++i)
  {
    TreeType* child = tree->ChildPtr(i);
    walker.Enter(child, tree);
    EnumerateTreeImpl(child, walker, false);
    walker.Leave(child, tree);
  }

  if (root)
    walker.Leave(tree, (const TreeType*)nullptr);
}

} // namespace enumerate


template <class TreeType, class Walker>
inline void EnumerateTree(TreeType* tree, Walker& walker)
{
  enumerate::EnumerateTreeImpl(tree, walker, true);
}

} // namespace tree
} // namespace mlpack


#endif // MLPACK_CORE_TREE_ENUMERATE_TREE_HPP