KDGrid_fwd.hpp

Go to the documentation of this file.

#ifndef MEDUSA_BITS_SPATIAL_SEARCH_KDGRID_FWD_HPP_
#define MEDUSA_BITS_SPATIAL_SEARCH_KDGRID_FWD_HPP_
 
#include "Grid_fwd.hpp"
#include <iosfwd>
#include <vector>
 
namespace mm {
 
template <typename vec_t>
class KDGrid {
  public:
    typedef vec_t vector_t;   
    typedef typename vector_t::scalar_t scalar_t;   
    enum {  dim = vec_t::dim };
 
  private:
    vec_t bot_;  
    vec_t top_;  
    scalar_t cell_size_;  
    Grid<int, dim> grid_;  
    std::vector<vec_t> points_;  
    typedef typename Grid<int, dim>::IndexArray IndexArray;   
 
  public:
    KDGrid(const vec_t& bot, const vec_t& top, scalar_t cell_size) :
            bot_(bot), top_(top), cell_size_(cell_size),
            grid_(compute_size(bot, top, cell_size), -1),
            points_() {}
 
    const vec_t& bottom() const { return bot_; }
    const vec_t& top() const { return top_; }
    scalar_t cellSize() const { return cell_size_; }
    const Grid<int, dim>& grid() const { return grid_; }
    const std::vector<vec_t>& points() const { return points_; }
    const vec_t& point(int idx) const { return points_[idx]; }
 
    int size() const { return points_.size(); }
 
    int insert(const vec_t& p) { return insert(p, compute_index(p)); }
 
    void insert(const std::vector<vec_t>& pts) {
        for (const vec_t& p : pts) insert(p);
    }
 
    bool existsPointInSphere(const vec_t& p, scalar_t r) {
        return existsPointInSphere(p, r, compute_index(p));
    }
 
    template <typename V>
    friend std::ostream& operator<<(std::ostream& os, const KDGrid<V>& search);
 
  private:
    static IndexArray compute_size(const vec_t& bot, const vec_t& top, scalar_t cell_size);
 
    IndexArray compute_index(const vec_t& p);
 
    bool existsPointInSphere(const vec_t& p, scalar_t r, const IndexArray& index);
 
    int insert(const vec_t& p, const IndexArray& index);
 
    bool increment(IndexArray& cur, int span, const IndexArray& ref);
};
 
}  // namespace mm
 
#endif  // MEDUSA_BITS_SPATIAL_SEARCH_KDGRID_FWD_HPP_