RaggedShapeStorage_fwd.hpp

Go to the documentation of this file.

#ifndef MEDUSA_BITS_OPERATORS_RAGGEDSHAPESTORAGE_FWD_HPP_
#define MEDUSA_BITS_OPERATORS_RAGGEDSHAPESTORAGE_FWD_HPP_
 
#include <medusa/Config.hpp>
#include <medusa/bits/operators/ShapeStorage_fwd.hpp>
#include <medusa/bits/operators/shape_flags.hpp>
#include <Eigen/Core>
 
namespace mm {
 
template <typename vec_t, typename OpFamilies =
        std::tuple<Lap<vec_t::dim>, Der1s<vec_t::dim>, Der2s<vec_t::dim>>>
class RaggedShapeStorage :
        public ShapeStorage<RaggedShapeStorage<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:
    Range<int> support_sizes_;
    Range<int> support_starts_;
    int total_size_;
 
    typedef ShapeStorage<RaggedShapeStorage<vec_t, OpFamilies>, vec_t, OpFamilies> base_t;
    friend base_t;  
    using base_t::domain_size_;
    using base_t::support_;
    using base_t::shapes_;
 
  public:
    using base_t::num_operators;  
    using base_t::size;  
 
    RaggedShapeStorage() : total_size_(0) {}
 
    int supportSize(int node) const { return support_sizes_[node]; }
 
  private:
    void resize_(const std::vector<int>& support_sizes);
 
    template <typename T> T* access(std::vector<T>& v, int i, int j) const {
        return v.data() + i*total_size_ + support_starts_[j]; }
    template <typename T> T* access(std::vector<T>& v, int i) const {
        return v.data() + support_starts_[i]; }
    template <typename T> const T* access(const std::vector<T>& v, int i, int j) const {
        return v.data() + i*total_size_ + support_starts_[j]; }
    template <typename T> const T* access(const std::vector<T>& v, int i) const {
        return v.data() + support_starts_[i]; }
};
 
}  // namespace mm
 
#endif  // MEDUSA_BITS_OPERATORS_RAGGEDSHAPESTORAGE_FWD_HPP_