UniformShapeStorage_fwd.hpp

Go to the documentation of this file.

#ifndef MEDUSA_BITS_OPERATORS_UNIFORMSHAPESTORAGE_FWD_HPP_
#define MEDUSA_BITS_OPERATORS_UNIFORMSHAPESTORAGE_FWD_HPP_
 
#include <medusa/Config.hpp>
#include <vector>
#include <medusa/bits/operators/shape_flags.hpp>
#include <iosfwd>
#include <Eigen/Core>
#include "ShapeStorage_fwd.hpp"
 
namespace mm {
 
template <typename vec_t, typename OpFamilies =
        std::tuple<Lap<vec_t::dim>, Der1s<vec_t::dim>, Der2s<vec_t::dim>>>
class UniformShapeStorage :
        public ShapeStorage<UniformShapeStorage<vec_t, OpFamilies>, vec_t, OpFamilies> {
  public:
    typedef vec_t vector_t;  
    typedef typename vec_t::scalar_t scalar_t;  
    enum {  dim = vec_t::dim };
 
  private:
    typedef ShapeStorage<UniformShapeStorage<vec_t, OpFamilies>, vec_t, OpFamilies> base_t;
    friend base_t;  
    using base_t::domain_size_;
    using base_t::support_;
    using base_t::shapes_;
 
    int support_size_;
 
  public:
    using base_t::num_operators;  
    using base_t::size;  
 
    UniformShapeStorage() : support_size_(0) {}
 
    int supportSize(int /* node */) const { return support_size_; }
 
  private:
    void resize_(const std::vector<int>& support_sizes);
 
    template <typename T> T* access(std::vector<T>& v, int op, int node) const {
        return v.data() + op*domain_size_*support_size_ + node*support_size_; }
    template <typename T> T* access(std::vector<T>& v, int node) const {
        return v.data() + node*support_size_; }
    template <typename T> const T* access(const std::vector<T>& v, int op, int node) const {
        return v.data() + op*domain_size_*support_size_ + node*support_size_; }
    template <typename T> const T* access(const std::vector<T>& v, int node) const {
        return v.data() + node*support_size_; }
};
 
}  // namespace mm
 
#endif  // MEDUSA_BITS_OPERATORS_UNIFORMSHAPESTORAGE_FWD_HPP_