field_map_base.hh
No OneTemporary
Actions

Subscribers

None

File Metadata

Created: Thu, Nov 21, 12:20

field_map_base.hh
View Options

	/**
	* @file field_map.hh
	*
	* @author Till Junge <till.junge@epfl.ch>
	*
	* @date 12 Sep 2017
	*
	* @brief Defined a strongly defines proxy that iterates efficiently over a field
	*
	* Copyright © 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_MAP_H
	#define FIELD_MAP_H

	#include "common/field.hh"
	#include "field_collection_base.hh"
	#include "common/common.hh"

	#include <unsupported/Eigen/CXX11/Tensor>

	#include <array>
	#include <string>
	#include <memory>
	#include <type_traits>

	namespace muSpectre {

	namespace internal {
	//----------------------------------------------------------------------------//
	//! little helper to automate creation of const maps without duplication
	template<class T, bool isConst>
	struct const_corrector {
	//! non-const type
	using type = typename T::reference;
	};

	//! specialisation for constant case
	template<class T>
	struct const_corrector<T, true> {
	//! const type
	using type = typename T::const_reference;
	};

	//! convenience alias
	template<class T, bool isConst>
	using const_corrector_t = typename const_corrector<T, isConst>::type;

	//----------------------------------------------------------------------------//
	template <class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	class FieldMap
	{
	public:
	//! number of scalars per entry
	constexpr static auto nb_components{NbComponents};
	using TypedField_nc = TypedSizedFieldBase
	<FieldCollection, T, NbComponents>; //!< non-constant version of field
	//! field type as seen from iterator
	using TypedField = std::conditional_t<ConstField,
	const TypedField_nc,
	TypedField_nc>;
	using Field = typename TypedField::Base; //!< iterated field type
	using size_type = std::size_t; //!< stl conformance
	using pointer = std::conditional_t<ConstField,
	const T*,
	T*>; //!< stl conformance
	//! Default constructor
	FieldMap() = delete;

	//! constructor
	template <bool isntConst=!ConstField>
	FieldMap(std::enable_if_t<isntConst, Field &> field);
	//! constructor
	template <bool isConst=ConstField>
	FieldMap(std::enable_if_t<isConst, const Field &> field);

	//! constructor with run-time cost (for python and debugging)
	template<class FC, typename T2, Dim_t NbC>
	FieldMap(TypedSizedFieldBase<FC, T2, NbC> & field);

	//! Copy constructor
	FieldMap(const FieldMap &other) = default;

	//! Move constructor
	FieldMap(FieldMap &&other) = default;

	//! Destructor
	virtual ~FieldMap() = default;

	//! Copy assignment operator
	FieldMap& operator=(const FieldMap &other) = delete;

	//! Move assignment operator
	FieldMap& operator=(FieldMap &&other) = delete;

	//! give human-readable field map type
	virtual std::string info_string() const = 0;

	//! return field name
	inline const std::string & get_name() const;

	//! return my collection (for iterating)
	inline const FieldCollection & get_collection() const;

	//! member access needs to be implemented by inheriting classes
	//inline value_type operator[](size_t index);
	//inline value_type operator[](Ccoord ccord);

	//! check compatibility (must be called by all inheriting classes at the
	//! end of their constructors
	inline void check_compatibility();

	//! convenience call to collection's size method
	inline size_t size() const;

	//! compile-time compatibility check
	template<class TypedField>
	struct is_compatible;

	/**
	* iterates over all pixels in the `muSpectre::FieldCollection`
	* and dereferences to an Eigen map to the currently used field.
	*/
	template <class FullyTypedFieldMap, bool ConstIter=false>
	class iterator
	{
	static_assert(!((ConstIter==false) && (ConstField==true)),
	"You can't have a non-const iterator over a const "
	"field");
	public:
	//! stl conformance
	using value_type =
	const_corrector_t<FullyTypedFieldMap, ConstIter>;
	//! stl conformance
	using const_value_type =
	const_corrector_t<FullyTypedFieldMap, true>;
	//! stl conformance
	using pointer = typename FullyTypedFieldMap::pointer;
	//! stl conformance
	using difference_type = std::ptrdiff_t;
	//! stl conformance
	using iterator_category = std::random_access_iterator_tag;
	//! cell coordinates type
	using Ccoord = typename FieldCollection::Ccoord;
	//! stl conformance
	using reference = typename FullyTypedFieldMap::reference;
	//! fully typed reference as seen by the iterator
	using TypedRef = std::conditional_t<ConstIter,
	const FullyTypedFieldMap &,
	FullyTypedFieldMap>;

	//! Default constructor
	iterator() = delete;

	//! constructor
	inline iterator(TypedRef fieldmap, bool begin=true);

	//! constructor for random access
	inline iterator(TypedRef fieldmap, size_t index);

	//! Copy constructor
	iterator(const iterator &other)= default;

	//! Move constructor
	iterator(iterator &&other) = default;

	//! Destructor
	virtual ~iterator() = default;

	//! Copy assignment operator
	iterator& operator=(const iterator &other) = default;

	//! Move assignment operator
	iterator& operator=(iterator &&other) = default;

	//! pre-increment
	inline iterator & operator++();
	//! post-increment
	inline iterator operator++(int);
	//! dereference
	inline value_type operator*();
	//! dereference
	inline const_value_type operator*() const;
	//! member of pointer
	inline pointer operator->();
	//! pre-decrement
	inline iterator & operator--();
	//! post-decrement
	inline iterator operator--(int);
	//! access subscripting
	inline value_type operator[](difference_type diff);
	//! access subscripting
	inline const_value_type operator[](const difference_type diff) const;
	//! equality
	inline bool operator==(const iterator & other) const;
	//! inequality
	inline bool operator!=(const iterator & other) const;
	//! div. comparisons
	inline bool operator<(const iterator & other) const;
	//! div. comparisons
	inline bool operator<=(const iterator & other) const;
	//! div. comparisons
	inline bool operator>(const iterator & other) const;
	//! div. comparisons
	inline bool operator>=(const iterator & other) const;
	//! additions, subtractions and corresponding assignments
	inline iterator operator+(difference_type diff) const;
	//! additions, subtractions and corresponding assignments
	inline iterator operator-(difference_type diff) const;
	//! additions, subtractions and corresponding assignments
	inline iterator& operator+=(difference_type diff);
	//! additions, subtractions and corresponding assignments
	inline iterator& operator-=(difference_type diff);

	//! get pixel coordinates
	inline Ccoord get_ccoord() const;

	//! ostream operator (mainly for debug
	friend std::ostream & operator<<(std::ostream & os,
	const iterator& it) {
	if (ConstIter) {
	os << "const ";
	}
	os << "iterator on field '"
	<< it.fieldmap.get_name()
	<< "', entry " << it.index;
	return os;
	}

	protected:
	const FieldCollection & collection; //!< collection of the field
	TypedRef fieldmap; //!< ref to the field itself
	size_t index; //!< index of currently pointed-to pixel
	private:
	};


	protected:
	//! raw pointer to entry (for Eigen Map)
	inline pointer get_ptr_to_entry(size_t index);
	//! raw pointer to entry (for Eigen Map)
	inline const T* get_ptr_to_entry(size_t index) const;
	const FieldCollection & collection; //!< collection holding Field
	TypedField & field; //!< mapped Field
	private:
	};
	} // internal


	namespace internal {

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template <bool isntConst>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	FieldMap(std::enable_if_t<isntConst, Field &> field)
	:collection(field.get_collection()), field(static_cast<TypedField&>(field)) {
	static_assert(NbComponents>0,
	"Only fields with more than 0 components allowed");
	}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template <bool isConst>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	FieldMap(std::enable_if_t<isConst, const Field &> field)
	:collection(field.get_collection()), field(static_cast<const TypedField&>(field)) {
	static_assert(NbComponents>0,
	"Only fields with more than 0 components allowed");
	}

	/* ---------------------------------------------------------------------- */
	template <class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template <class FC, typename T2, Dim_t NbC>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	FieldMap(TypedSizedFieldBase<FC, T2, NbC> & field)
	:collection(field.get_collection()), field(static_cast<TypedField&>(field)) {
	static_assert(std::is_same<FC, FieldCollection>::value,
	"The field does not have the expected FieldCollection type");
	static_assert(std::is_same<T2, T>::value,
	"The field does not have the expected Scalar type");
	static_assert((NbC == NbComponents),
	"The field does not have the expected number of components");
	}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	void
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	check_compatibility() {
	if (typeid(T).hash_code() !=
	this->field.get_stored_typeid().hash_code()) {
	std::string err{"Cannot create a Map of type '" +
	this->info_string() +
	"' for field '" + this->field.get_name() + "' of type '" +
	this->field.get_stored_typeid().name() + "'"};
	throw FieldInterpretationError
	(err);
	}
	//check size compatibility
	if (NbComponents != this->field.get_nb_components()) {
	throw FieldInterpretationError
	("Cannot create a Map of type '" +
	this->info_string() +
	"' for field '" + this->field.get_name() + "' with " +
	std::to_string(this->field.get_nb_components()) + " components");
	}
	}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	size_t
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	size() const {
	return this->collection.size();
	}

	/* ---------------------------------------------------------------------- */
	template <class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template <class myField>
	struct FieldMap<FieldCollection, T, NbComponents, ConstField>::is_compatible {
	//! creates a more readable compile error
	constexpr static bool explain() {
	static_assert
	(std::is_same<typename myField::collection_t, FieldCollection>::value,
	"The field does not have the expected FieldCollection type");
	static_assert
	(std::is_same<typename myField::Scalar, T>::value,
	"The // field does not have the expected Scalar type");
	static_assert((TypedField::nb_components == NbComponents),
	"The field does not have the expected number of components");
	//The static asserts wouldn't pass in the incompatible case, so this is it
	return true;
	}
	//! evaluated compatibility
	constexpr static bool value{std::is_base_of<TypedField, myField>::value};
	};

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	const std::string &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	get_name() const {
	return this->field.get_name();
	}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	const FieldCollection &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	get_collection() const {
	return this->collection;
	}

	/* ---------------------------------------------------------------------- */
	/* ---------------------------------------------------------------------- */
	// Iterator implementations
	//! constructor
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::iterator
	<FullyTypedFieldMap, ConstIter>::
	iterator(TypedRef fieldmap, bool begin)
	:collection(fieldmap.get_collection()), fieldmap(fieldmap),
	index(begin ? 0 : fieldmap.field.size()) {}

	/* ---------------------------------------------------------------------- */
	//! constructor for random access
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::iterator<FullyTypedFieldMap, ConstIter>::
	iterator(TypedRef fieldmap, size_t index)
	:collection(fieldmap.collection), fieldmap(fieldmap),
	index(index) {}

	/* ---------------------------------------------------------------------- */
	//! pre-increment
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter> &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::iterator<FullyTypedFieldMap, ConstIter>::
	operator++() {
	this->index++;
	return *this;
	}

	/* ---------------------------------------------------------------------- */
	//! post-increment
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::iterator<FullyTypedFieldMap, ConstIter>::
	operator++(int) {
	iterator current = *this;
	this->index++;
	return current;
	}

	/* ---------------------------------------------------------------------- */
	//! dereference
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::
	template iterator<FullyTypedFieldMap, ConstIter>::value_type
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator*() {
	return this->fieldmap.operator[](this->index);
	}

	/* ---------------------------------------------------------------------- */
	//! dereference
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::
	template iterator<FullyTypedFieldMap, ConstIter>::const_value_type
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator*() const {
	return this->fieldmap.operator[](this->index);
	}

	/* ---------------------------------------------------------------------- */
	//! member of pointer
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FullyTypedFieldMap::pointer
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator->() {
	return this->fieldmap.ptr_to_val_t(this->index);
	}

	/* ---------------------------------------------------------------------- */
	//! pre-decrement
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter> &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator--() {
	this->index--;
	return *this;
	}

	/* ---------------------------------------------------------------------- */
	//! post-decrement
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator--(int) {
	iterator current = *this;
	this->index--;
	return current;
	}

	/* ---------------------------------------------------------------------- */
	//! Access subscripting
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::
	template iterator<FullyTypedFieldMap, ConstIter>::value_type
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator[](difference_type diff) {
	return this->fieldmap[this->index+diff];
	}

	/* ---------------------------------------------------------------------- */
	//! Access subscripting
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template iterator<FullyTypedFieldMap, ConstIter>::const_value_type
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator[](const difference_type diff) const {
	return this->fieldmap[this->index+diff];
	}

	/* ---------------------------------------------------------------------- */
	//! equality
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator==(const iterator & other) const {
	return (this->index == other.index);
	}

	/* ---------------------------------------------------------------------- */
	//! inquality
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator!=(const iterator & other) const {
	return !(*this == other);
	}

	/* ---------------------------------------------------------------------- */
	//! div. comparisons
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator<(const iterator & other) const {
	return (this->index < other.index);
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator<=(const iterator & other) const {
	return (this->index <= other.index);
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator>(const iterator & other) const {
	return (this->index > other.index);
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	bool
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator>=(const iterator & other) const {
	return (this->index >= other.index);
	}

	/* ---------------------------------------------------------------------- */
	//! additions, subtractions and corresponding assignments
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator+(difference_type diff) const {
	return iterator(this->fieldmap, this->index + diff);
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter>
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator-(difference_type diff) const {
	return iterator(this->fieldmap, this->index - diff);
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter> &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator+=(difference_type diff) {
	this->index += diff;
	return *this;
	}
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::template
	iterator<FullyTypedFieldMap, ConstIter> &
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	operator-=(difference_type diff) {
	this->index -= diff;
	return *this;
	}

	/* ---------------------------------------------------------------------- */
	//! get pixel coordinates
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	template<class FullyTypedFieldMap, bool ConstIter>
	typename FieldCollection::Ccoord
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	iterator<FullyTypedFieldMap, ConstIter>::
	get_ccoord() const {
	return this->collection.get_ccoord(this->index);
	}

	////----------------------------------------------------------------------------//
	//template<class FieldCollection, typename T, Dim_t NbComponents, class FullyTypedFieldMap>
	//std::ostream & operator <<
	//(std::ostream &os,
	// const typename FieldMap<FieldCollection, T, NbComponents, ConstField>::
	// template iterator<FullyTypedFieldMap, ConstIter> & it) {
	// os << "iterator on field '"
	// << it.field.get_name()
	// << "', entry " << it.index;
	// return os;
	//}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	typename FieldMap<FieldCollection, T, NbComponents, ConstField>::pointer
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	get_ptr_to_entry(size_t index) {
	return this->field.get_ptr_to_entry(std::move(index));
	}

	/* ---------------------------------------------------------------------- */
	template<class FieldCollection, typename T, Dim_t NbComponents, bool ConstField>
	const T*
	FieldMap<FieldCollection, T, NbComponents, ConstField>::
	get_ptr_to_entry(size_t index) const {
	return this->field.get_ptr_to_entry(std::move(index));
	}

	} // internal


	} // muSpectre

	#endif /* FIELD_MAP_H */

field_map_base.hhNo OneTemporaryActions

File Metadata

field_map_base.hhView Options

Event Timeline

field_map_base.hh
No OneTemporary
Actions

field_map_base.hh
View Options