Page MenuHomec4science
Files F87999681

field_iterator_matrixlike.hh
No OneTemporary
Actions

File Metadata

Created
Wed, Oct 16, 05:59

field_iterator_matrixlike.hh
View Options

/**
* file field_iterator_matrixlike.hh
*
* @author Till Junge <till.junge@epfl.ch>
*
* @date 26 Sep 2017
*
* @brief Eigen-Matrix and -Array iterators over strongly typed fields
*
* @section LICENCE
*
* Copyright (C) 2017 Till Junge
*
* µSpectre is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation, either version 3, or (at
* your option) any later version.
*
* µSpectre is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Emacs; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifndef FIELD_ITERATOR_MATRIXLIKE_H
#define FIELD_ITERATOR_MATRIXLIKE_H
#include <Eigen/Dense>
#include "system/field_iterator_base.hh"
namespace muSpectre {
namespace internal {
/* ---------------------------------------------------------------------- */
enum class Map_t{Matrix, Array};
template<Map_t map_type>
struct NameWrapper {
const static std::string field_info_root;
};
template<>
const std::string NameWrapper<Map_t::Array>::field_info_root{"Array"};
template<>
const std::string NameWrapper<Map_t::Matrix>::field_info_root{"Matrix"};
/* ---------------------------------------------------------------------- */
template <class FieldCollection, class EigenArray, Map_t map_type>
class MatrixLikeFieldMap: public FieldMap<FieldCollection,
typename EigenArray::Scalar,
EigenArray::SizeAtCompileTime>
{
public:
using Ccoord = Ccoord_t<FieldCollection::spatial_dim()>;
using value_type = EigenArray;
using parent = FieldMap<FieldCollection,
typename EigenArray::Scalar,
EigenArray::SizeAtCompileTime>;
using TypedField = typename parent::TypedField;
using Field = typename TypedField::parent;
using iterator = typename parent::template iterator<MatrixLikeFieldMap>;
friend iterator;
//! Default constructor
MatrixLikeFieldMap() = delete;
MatrixLikeFieldMap(Field & field);
//! Copy constructor
MatrixLikeFieldMap(const MatrixLikeFieldMap &other) = delete;
//! Move constructor
MatrixLikeFieldMap(MatrixLikeFieldMap &&other) noexcept = delete;
//! Destructor
virtual ~MatrixLikeFieldMap() noexcept = default;
//! Copy assignment operator
MatrixLikeFieldMap& operator=(const MatrixLikeFieldMap &other) = delete;
//! Move assignment operator
MatrixLikeFieldMap& operator=(MatrixLikeFieldMap &&other) noexcept = delete;
//! give human-readable field map type
inline std::string info_string() const override final;
//! member access
inline value_type operator[](Uint index);
inline value_type operator[](Ccoord ccoord);
//! return an iterator to head of field for ranges
inline iterator begin(){return iterator(*this);}
//! return an iterator to tail of field for ranges
inline iterator end(){return iterator(*this, false);};
protected:
const static std::string field_info_root;
private:
};
template <class FieldCollection, class EigenArray, Map_t map_type>
const std::string MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::
field_info_root{NameWrapper<map_type>::field_info_root};
/* ---------------------------------------------------------------------- */
template<class FieldCollection, class EigenArray, Map_t map_type>
MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::
MatrixLikeFieldMap(Field & field)
:parent(field) {
this->check_compatibility();
}
/* ---------------------------------------------------------------------- */
//! human-readable field map type
template<class FieldCollection, class EigenArray, Map_t map_type>
std::string
MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::
info_string() const {
std::stringstream info;
info << field_info_root << "("
<< typeid(typename EigenArray::value_type).name() << ", "
<< EigenArray::RowsAtCompileTime << "x"
<< EigenArray::ColsAtCompileTime << ")";
return info.str();
}
/* ---------------------------------------------------------------------- */
//! member access
template<class FieldCollection, class EigenArray, Map_t map_type>
typename MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::value_type
MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::
operator[](Uint index) {
return value_type(this->get_ptr_to_entry(index));
}
template<class FieldCollection, class EigenArray, Map_t map_type>
typename MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::value_type
MatrixLikeFieldMap<FieldCollection, EigenArray, map_type>::
operator[](Ccoord ccoord) {
auto && index = this->collection.get_index(ccoord);
return value_type(this->get_ptr_to_entry(index));
}
} // internal
/* ---------------------------------------------------------------------- */
//! short-hand for an Eigen matrix map as iterate
template <class FieldCollection, typename T, Dim_t NbRows, Dim_t NbCols>
using MatrixFieldMap = internal::MatrixLikeFieldMap
<FieldCollection,
Eigen::Map<Eigen::Matrix<T, NbRows, NbCols>>,
internal::Map_t::Matrix>;
/* ---------------------------------------------------------------------- */
//! short-hand for an Eigen array map as iterate
template <class FieldCollection, typename T, Dim_t NbRows, Dim_t NbCols>
using ArrayFieldMap = internal::MatrixLikeFieldMap
<FieldCollection,
Eigen::Map<Eigen::Array<T, NbRows, NbCols>>,
internal::Map_t::Array>;
} // muSpectre
#endif /* FIELD_ITERATOR_MATRIXLIKE_H */

Event Timeline

c4science · Help