traversal_info.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_TREE_TRAVERSAL_INFO_HPP
#define MLPACK_CORE_TREE_TRAVERSAL_INFO_HPP

namespace mlpack {
namespace tree {

template<typename TreeType>
class TraversalInfo
{
 public:
  TraversalInfo() :
    lastQueryNode(NULL),
    lastReferenceNode(NULL),
    lastScore(0.0),
    lastBaseCase(0.0) { /* Nothing to do. */ }

  TreeType* LastQueryNode() const { return lastQueryNode; }
  TreeType*& LastQueryNode() { return lastQueryNode; }

  TreeType* LastReferenceNode() const { return lastReferenceNode; }
  TreeType*& LastReferenceNode() { return lastReferenceNode; }

  double LastScore() const { return lastScore; }
  double& LastScore() { return lastScore; }

  double LastBaseCase() const { return lastBaseCase; }
  double& LastBaseCase() { return lastBaseCase; }

 private:
  TreeType* lastQueryNode;
  TreeType* lastReferenceNode;
  double lastScore;
  double lastBaseCase;
};

} // namespace tree
} // namespace mlpack

#endif