minimal_coverage_sweep.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_COVERAGE_SWEEP_HPP
#define MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_COVERAGE_SWEEP_HPP

namespace mlpack {
namespace tree {

template<typename SplitPolicy>
class MinimalCoverageSweep
{
 public:
  template<typename TreeType>
  struct SweepCost
  {
    typedef typename TreeType::ElemType type;
  };

  template<typename TreeType>
  static typename TreeType::ElemType SweepNonLeafNode(
      const size_t axis,
      const TreeType* node,
      typename TreeType::ElemType& axisCut);

  template<typename TreeType>
  static typename TreeType::ElemType SweepLeafNode(
      const size_t axis,
      const TreeType* node,
      typename TreeType::ElemType& axisCut);

  template<typename TreeType, typename ElemType>
  static bool CheckNonLeafSweep(const TreeType* node,
                                const size_t cutAxis,
                                const ElemType cut);

  template<typename TreeType, typename ElemType>
  static bool CheckLeafSweep(const TreeType* node,
                             const size_t cutAxis,
                             const ElemType cut);
};

} // namespace tree
} // namespace mlpack

// Include implementation
#include "minimal_coverage_sweep_impl.hpp"

#endif  //  MLPACK_CORE_TREE_RECTANGLE_TREE_MINIMAL_COVERAGE_SWEEP_HPP