Diffusion akantu (bugfixes/bug_fix_in_synchronization_of_mat_vec_mul_product_when_assembled_to_residual) Browse

bug_fix_in_syn…

dumper_compute.hh
View Options

			/**
			* @file dumper_compute.hh
			*
			* @author Guillaume Anciaux <guillaume.anciaux@epfl.ch>
			*
			* @date creation: Tue Sep 02 2014
			* @date last modification: Sun Dec 03 2017
			*
			* @brief Field that map a function to another field
			*
			* @section LICENSE
			*
			* Copyright (©) 2014-2018 EPFL (Ecole Polytechnique Fédérale de Lausanne)
			* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
			*
			* Akantu is free software: you can redistribute it and/or modify it under the
			* terms of the GNU Lesser General Public License as published by the Free
			* Software Foundation, either version 3 of the License, or (at your option) any
			* later version.
			*
			* Akantu 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 Lesser General Public License for more
			* details.
			*
			* You should have received a copy of the GNU Lesser General Public License
			* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
			*
			*/

			#ifndef __AKANTU_DUMPER_COMPUTE_HH__
			#define __AKANTU_DUMPER_COMPUTE_HH__
			/* -------------------------------------------------------------------------- */
			#include "aka_common.hh"
			#include "dumper_field.hh"
			#include "dumper_iohelper.hh"
			#include "dumper_type_traits.hh"
			#include <io_helper.hh>

			/* -------------------------------------------------------------------------- */

			namespace akantu {
			__BEGIN_AKANTU_DUMPER__

			class ComputeFunctorInterface {
			public:
			virtual ~ComputeFunctorInterface() = default;

			virtual UInt getDim() = 0;
			virtual UInt getNbComponent(UInt old_nb_comp) = 0;
			};

			/* -------------------------------------------------------------------------- */

			template <typename return_type>
			class ComputeFunctorOutput : public ComputeFunctorInterface {
			public:
			ComputeFunctorOutput() = default;
			~ComputeFunctorOutput() override = default;
			};

			/* -------------------------------------------------------------------------- */
			template <typename input_type, typename return_type>
			class ComputeFunctor : public ComputeFunctorOutput<return_type> {
			public:
			ComputeFunctor() = default;
			~ComputeFunctor() override = default;

			virtual return_type func(const input_type & d, Element global_index) = 0;
			};

			/* -------------------------------------------------------------------------- */
			template <typename SubFieldCompute, typename _return_type>
			class FieldCompute : public Field {
			/* ------------------------------------------------------------------------ */
			/* Typedefs */
			/* ------------------------------------------------------------------------ */
			public:
			using sub_iterator = typename SubFieldCompute::iterator;
			using sub_types = typename SubFieldCompute::types;
			using sub_return_type = typename sub_types::return_type;
			using return_type = _return_type;
			using data_type = typename sub_types::data_type;

			using types =
			TypeTraits<data_type, return_type, ElementTypeMapArray<data_type>>;

			class iterator {
			public:
			iterator(const sub_iterator & it,
			ComputeFunctor<sub_return_type, return_type> & func)
			: it(it), func(func) {}

			bool operator!=(const iterator & it) const { return it.it != this->it; }
			iterator operator++() {
			++this->it;
			return *this;
			}

			UInt currentGlobalIndex() { return this->it.currentGlobalIndex(); }

			return_type operator() { return func.func(it, it.getCurrentElement()); }

			Element getCurrentElement() { return this->it.getCurrentElement(); }

			UInt element_type() { return this->it.element_type(); }

			protected:
			sub_iterator it;
			ComputeFunctor<sub_return_type, return_type> & func;
			};

			/* ------------------------------------------------------------------------ */
			/* Constructors/Destructors */
			/* ------------------------------------------------------------------------ */
			public:
			FieldCompute(SubFieldCompute & cont,
			std::unique_ptr<ComputeFunctorInterface> func)
			: sub_field(aka::as_type<SubFieldCompute>(cont.shared_from_this())),
			func(aka::as_type<ComputeFunctor<sub_return_type, return_type>>(
			func.release())) {
			this->checkHomogeneity();
			};

			~FieldCompute() override = default;

			void registerToDumper(const std::string & id,
			iohelper::Dumper & dumper) override {
			dumper.addElemDataField(id, *this);
			}

			/* ------------------------------------------------------------------------ */
			/* Class Members */
			/* ------------------------------------------------------------------------ */
			public:
			iterator begin() { return iterator(sub_field->begin(), *func); }
			iterator end() { return iterator(sub_field->end(), *func); }

			UInt getDim() { return func->getDim(); }

			UInt size() {
			throw;
			// return Functor::size();
			return 0;
			}

			void checkHomogeneity() override { this->homogeneous = true; };

			iohelper::DataType getDataType() {
			return iohelper::getDataType<data_type>();
			}

			/// get the number of components of the hosted field
			ElementTypeMap<UInt>
			getNbComponents(UInt dim = _all_dimensions, GhostType ghost_type = _not_ghost,
			ElementKind kind = _ek_not_defined) override {
			ElementTypeMap<UInt> nb_components;
			const auto & old_nb_components =
			this->sub_field->getNbComponents(dim, ghost_type, kind);

			for (auto type : old_nb_components.elementTypes(dim, ghost_type, kind)) {
			UInt nb_comp = old_nb_components(type, ghost_type);
			nb_components(type, ghost_type) = func->getNbComponent(nb_comp);
			}
			return nb_components;
			};

			/// for connection to a FieldCompute
			inline std::shared_ptr<Field> connect(FieldComputeProxy & proxy) override;

			/// for connection to a FieldCompute
			std::unique_ptr<ComputeFunctorInterface>
			connect(HomogenizerProxy & proxy) override;

			/* ------------------------------------------------------------------------ */
			/* Class Members */
			/* ------------------------------------------------------------------------ */
			public:
			std::shared_ptr<SubFieldCompute> sub_field;
			std::unique_ptr<ComputeFunctor<sub_return_type, return_type>> func;
			};

			/* -------------------------------------------------------------------------- */

			/* -------------------------------------------------------------------------- */

			class FieldComputeProxy {
			/* ------------------------------------------------------------------------ */
			/* Constructors/Destructors */
			/* ------------------------------------------------------------------------ */
			public:
			FieldComputeProxy(std::unique_ptr<ComputeFunctorInterface> func)
			: func(std::move(func)){};

			inline static std::shared_ptr<Field>
			createFieldCompute(std::shared_ptr<Field> field,
			std::unique_ptr<ComputeFunctorInterface> func) {
			FieldComputeProxy compute_proxy(std::move(func));
			return field->connect(compute_proxy);
			}

			template <typename T> std::shared_ptr<Field> connectToField(T * ptr) {
			if (aka::is_of_type<ComputeFunctorOutput<Vector<Real>>>(func)) {
			return this->connectToFunctor<Vector<Real>>(ptr);
			} else if (aka::is_of_type<ComputeFunctorOutput<Vector<UInt>>>(func)) {
			return this->connectToFunctor<Vector<UInt>>(ptr);
			} else if (aka::is_of_type<ComputeFunctorOutput<Matrix<UInt>>>(func)) {
			return this->connectToFunctor<Matrix<UInt>>(ptr);
			} else if (aka::is_of_type<ComputeFunctorOutput<Matrix<Real>>>(func)) {
			return this->connectToFunctor<Matrix<Real>>(ptr);
			} else {
			throw;
			}
			}

			template <typename output, typename T>
			std::shared_ptr<Field> connectToFunctor(T * ptr) {
			return std::make_shared<FieldCompute<T, output>>(*ptr, std::move(func));
			}

			template <typename output, typename SubFieldCompute, typename return_type1,
			typename return_type2>
			std::shared_ptr<Field>
			connectToFunctor(__attribute__((unused))
			FieldCompute<FieldCompute<SubFieldCompute, return_type1>,
			return_type2> * ptr) {
			throw; // return new FieldCompute<T,output>(*ptr,func);
			return nullptr;
			}

			template <typename output, typename SubFieldCompute, typename return_type1,
			typename return_type2, typename return_type3, typename return_type4>
			std::shared_ptr<Field> connectToFunctor(
			__attribute__((unused)) FieldCompute<
			FieldCompute<FieldCompute<FieldCompute<SubFieldCompute, return_type1>,
			return_type2>,
			return_type3>,
			return_type4> * ptr) {
			throw; // return new FieldCompute<T,output>(*ptr,func);
			return nullptr;
			}

			/* ------------------------------------------------------------------------ */
			/* Class Members */
			/* ------------------------------------------------------------------------ */
			public:
			std::unique_ptr<ComputeFunctorInterface> func;
			};

			/* -------------------------------------------------------------------------- */
			/// for connection to a FieldCompute
			template <typename SubFieldCompute, typename return_type>
			inline std::shared_ptr<Field>
			FieldCompute<SubFieldCompute, return_type>::connect(FieldComputeProxy & proxy) {
			return proxy.connectToField(this);
			}

			/* -------------------------------------------------------------------------- */
			__END_AKANTU_DUMPER__
			} // namespace akantu

			#endif /* __AKANTU_DUMPER_COMPUTE_HH__ */

rAKA/src/io/dumper/dumper_compute.hh5cf490a6c1b7bugfixes/bug_fix_in_syn…

dumper_compute.hhView Options

rAKA/src/io/dumper/dumper_compute.hh
5cf490a6c1b7bugfixes/bug_fix_in_syn…

dumper_compute.hh
View Options