ScalarField_fwd.hpp

Go to the documentation of this file.

#ifndef MEDUSA_BITS_TYPES_SCALARFIELD_FWD_HPP_
#define MEDUSA_BITS_TYPES_SCALARFIELD_FWD_HPP_
 
#include <medusa/Config.hpp>
#include <Eigen/Core>
#include "traits.hpp"
 
namespace mm {
 
template <typename Scalar>
class ScalarField : public Eigen::Matrix<Scalar, Eigen::Dynamic, 1> {
  public:
    typedef Scalar scalar_t;  
    typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> Base;  
    using Base::Base;  // inherit constructors
    using Base::operator=;  // inherit assignment operators
    template <typename OtherDerived>
    ScalarField(const Eigen::MatrixBase<OtherDerived>& other);
    template <typename OtherDerived>
    ScalarField& operator=(const Eigen::MatrixBase<OtherDerived>& other);
 
  private:
    class ScalarFieldView {
        ScalarField& sf;  
        const indexes_t& indexes;  
        friend class ScalarField;
        ScalarFieldView(ScalarField& sf, const indexes_t& indexes) : sf(sf), indexes(indexes) {}
      public:
        void operator=(const scalar_t& s) {
            sf(indexes) = ScalarField::Constant(indexes.size(), s);
        }
        template <typename OtherDerived>
        void operator=(const Eigen::MatrixBase<OtherDerived>& other) {
            sf(indexes) = other;
        }
        void operator=(const ScalarFieldView& other) {
            sf(indexes) = other.sf(other.indexes);
        }
    };
 
  public:
    using Base::operator[];
    ScalarFieldView operator[](const indexes_t& indexes) {
        return ScalarFieldView(*this, indexes);
    }
};
 
typedef ScalarField<double> ScalarFieldd;  
typedef ScalarField<float> ScalarFieldf;  
 
template <typename T> struct scalar_type<ScalarField<T>> {
    typedef T type;  
};
 
}  // namespace mm
 
#endif  // MEDUSA_BITS_TYPES_SCALARFIELD_FWD_HPP_