KdTree.h

Go to the documentation of this file.

 
#ifndef _SEUTILS_KDTREE_H_
#define _SEUTILS_KDTREE_H_
 
#include <vector>
#include <memory>
#include <algorithm>
 
namespace SourceXtractor {
 
template <typename T>
struct KdTreeTraits {
  static double getCoord(const T& t, size_t index);
};
 
template<typename T, size_t N=2, size_t S=100>
class KdTree {
public:
  using Traits = KdTreeTraits<T>;
 
  struct Coord {
    double coord[N];
  };
 
  explicit KdTree(const std::vector<T>& data);
  std::vector<T> findPointsWithinRadius(Coord coord, double radius) const;
 
private:
  class Node;
  class Leaf;
  class Split;
 
  std::shared_ptr<Node> m_root;
};
 
}
 
#include "_impl/KdTree.icpp"
 
#endif /* _SEUTILS_KDTREE_H_ */