Grid_fwd.hpp

Go to the documentation of this file.

#ifndef MEDUSA_BITS_SPATIAL_SEARCH_GRID_FWD_HPP_
#define MEDUSA_BITS_SPATIAL_SEARCH_GRID_FWD_HPP_
 
 
#include <medusa/Config.hpp>
#include <array>
#include <vector>
#include <iosfwd>
 
namespace mm {
 
template <typename T, int dimension, typename IndexType = int,
          typename IndexArrayT = std::array<IndexType, dimension>>
class Grid {
  public:
    typedef T value_type;  
    typedef IndexType Index;  
    enum {  dim = dimension };
    typedef IndexArrayT IndexArray;   
 
  private:
    IndexArray sizes_;  
    IndexType size_;  
    std::vector<T> data_;  
 
  public:
    Grid(const IndexArray& sizes) : sizes_(sizes), size_(computeSize(sizes)), data_(size_) {}
    Grid(const IndexArray& sizes, const T& value) :
            sizes_(sizes), size_(computeSize(sizes)), data_(size_, value) {}
 
    IndexArray sizes() const { return sizes_; }
    inline Index size(int i) const;
    Index size() const { return size_; }
 
    const std::vector<T>& data() const { return data_; }
    std::vector<T>& data() { return data_; }
 
    inline const T& operator()(const IndexArray& index) const;
 
    inline T& operator()(const IndexArray& index);
 
    inline const T& operator[](const Index& index) const;
 
    inline T& operator[](const Index& index);
 
    Index linearIndex(const IndexArray& index) const { return linearIndex(index, sizes_); }
    IndexArray multiIndex(const Index& index) const { return multiIndex(index, sizes_); }
 
    bool inBounds(const IndexArray& index) const { return inBounds(index, sizes_); }
 
  private:
    static Index computeSize(const IndexArray& sizes);
 
    static bool inBounds(const IndexArray& index, const IndexArray& bounds);
 
    static Index linearIndex(const IndexArray& index, const IndexArray& bounds);
    static IndexArray multiIndex(Index index, const IndexArray& bounds);
 
  public:
    template <typename U, int D, typename I, typename IA>
    friend std::ostream& operator<<(std::ostream& os, const Grid<U, D, I, IA>& grid);
};
 
}  // namespace mm
 
#endif  // MEDUSA_BITS_SPATIAL_SEARCH_GRID_FWD_HPP_