Diffusion akantu (bugfixes/python_package_test) Browse

python_package…

fragment_manager.cc
View Options

			/**
			* @file fragment_manager.cc
			*
			* @author Aurelia Isabel Cuba Ramos <aurelia.cubaramos@epfl.ch>
			* @author Marco Vocialta <marco.vocialta@epfl.ch>
			*
			* @date creation: Thu Jan 23 2014
			* @date last modification: Tue Feb 20 2018
			*
			* @brief Group manager to handle fragments
			*
			*
			* 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/>.
			*
			*/

			/* -------------------------------------------------------------------------- */
			#include "fragment_manager.hh"
			#include "aka_iterators.hh"
			#include "communicator.hh"
			#include "element_synchronizer.hh"
			#include "material_cohesive.hh"
			#include "mesh_iterators.hh"
			#include "solid_mechanics_model_cohesive.hh"
			/* -------------------------------------------------------------------------- */
			#include <algorithm>
			#include <functional>
			#include <numeric>
			/* -------------------------------------------------------------------------- */

			namespace akantu {

			/* -------------------------------------------------------------------------- */
			FragmentManager::FragmentManager(SolidMechanicsModelCohesive & model,
			bool dump_data, const ID & id)
			: GroupManager(model.getMesh(), id), model(model),
			mass_center(0, model.getSpatialDimension(), "mass_center"),
			mass(0, model.getSpatialDimension(), "mass"),
			velocity(0, model.getSpatialDimension(), "velocity"),
			inertia_moments(0, model.getSpatialDimension(), "inertia_moments"),
			principal_directions(
			0, model.getSpatialDimension() * model.getSpatialDimension(),
			"principal_directions"),
			quad_coordinates("quad_coordinates", id),
			mass_density("mass_density", id),
			nb_elements_per_fragment(0, 1, "nb_elements_per_fragment"),
			dump_data(dump_data) {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = mesh.getSpatialDimension();

			/// compute quadrature points' coordinates
			quad_coordinates.initialize(mesh, _nb_component = spatial_dimension,
			_spatial_dimension = spatial_dimension,
			_ghost_type = _not_ghost);
			// mesh.initElementTypeMapArray(quad_coordinates, spatial_dimension,
			// spatial_dimension, _not_ghost);

			model.getFEEngine().interpolateOnIntegrationPoints(model.getMesh().getNodes(),
			quad_coordinates);

			/// store mass density per quadrature point
			mass_density.initialize(mesh, _spatial_dimension = spatial_dimension,
			_ghost_type = _not_ghost);
			// mesh.initElementTypeMapArray(mass_density, 1, spatial_dimension,
			// _not_ghost);

			storeMassDensityPerIntegrationPoint();

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			class CohesiveElementFilter : public GroupManager::ClusteringFilter {
			public:
			CohesiveElementFilter(const SolidMechanicsModelCohesive & model,
			const Real max_damage = 1.)
			: model(model), is_unbroken(max_damage) {}

			bool operator()(const Element & el) const override {
			if (Mesh::getKind(el.type) == _ek_regular) {
			return true;
			}

			const Array<UInt> & mat_indexes =
			model.getMaterialByElement(el.type, el.ghost_type);
			const Array<UInt> & mat_loc_num =
			model.getMaterialLocalNumbering(el.type, el.ghost_type);

			const auto & mat = static_cast<const MaterialCohesive &>(
			model.getMaterial(mat_indexes(el.element)));

			UInt el_index = mat_loc_num(el.element);
			UInt nb_quad_per_element =
			model.getFEEngine("CohesiveFEEngine")
			.getNbIntegrationPoints(el.type, el.ghost_type);

			const Array<Real> & damage_array = mat.getDamage(el.type, el.ghost_type);

			AKANTU_DEBUG_ASSERT(nb_quad_per_element * el_index < damage_array.size(),
			"This quadrature point is out of range");

			const Real * element_damage =
			damage_array.storage() + nb_quad_per_element * el_index;

			UInt unbroken_quads = std::count_if(
			element_damage, element_damage + nb_quad_per_element, is_unbroken);

			return (unbroken_quads > 0);
			}

			private:
			struct IsUnbrokenFunctor {
			IsUnbrokenFunctor(const Real & max_damage) : max_damage(max_damage) {}
			bool operator()(const Real & x) const { return x < max_damage; }
			const Real max_damage;
			};

			const SolidMechanicsModelCohesive & model;
			const IsUnbrokenFunctor is_unbroken;
			};

			/* -------------------------------------------------------------------------- */
			void FragmentManager::buildFragments(Real damage_limit) {
			AKANTU_DEBUG_IN();

			if (mesh.isDistributed()) {
			auto & cohesive_synchronizer = model.getCohesiveSynchronizer();
			cohesive_synchronizer.synchronize(model, SynchronizationTag::_smmc_damage);
			}

			auto & mesh_facets = mesh.getMeshFacets();

			UInt spatial_dimension = model.getSpatialDimension();
			std::string fragment_prefix("fragment");

			/// generate fragments
			global_nb_fragment =
			createClusters(spatial_dimension, mesh_facets, fragment_prefix,
			CohesiveElementFilter(model, damage_limit));

			nb_fragment = getNbElementGroups(spatial_dimension);
			fragment_index.resize(nb_fragment);

			/// loop over fragments
			for (auto && data : zip(iterateElementGroups(), fragment_index)) {
			auto name = std::get<0>(data).getName();
			/// get fragment index
			std::string fragment_index_string = name.substr(fragment_prefix.size() + 1);
			std::get<1>(data) = std::stoul(fragment_index_string);
			}

			/// compute fragments' mass
			computeMass();

			if (dump_data) {
			createDumpDataArray(fragment_index, "fragments", true);
			createDumpDataArray(mass, "fragments mass");
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::computeMass() {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();

			/// create a unit field per quadrature point, since to compute mass
			/// it's neccessary to integrate only density
			ElementTypeMapArray<Real> unit_field("unit_field", id);
			unit_field.initialize(model.getFEEngine(), _nb_component = spatial_dimension,
			_spatial_dimension = spatial_dimension,
			_ghost_type = _not_ghost, _default_value = 1.);

			integrateFieldOnFragments(unit_field, mass);

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::computeCenterOfMass() {
			AKANTU_DEBUG_IN();

			/// integrate position multiplied by density
			integrateFieldOnFragments(quad_coordinates, mass_center);

			/// divide it by the fragments' mass
			Real * mass_storage = mass.storage();
			Real * mass_center_storage = mass_center.storage();

			UInt total_components = mass_center.size() * mass_center.getNbComponent();

			for (UInt i = 0; i < total_components; ++i) {
			mass_center_storage[i] /= mass_storage[i];
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::computeVelocity() {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();

			/// compute velocity per quadrature point
			ElementTypeMapArray<Real> velocity_field("velocity_field", id);
			velocity_field.initialize(
			model.getFEEngine(), _nb_component = spatial_dimension,
			_spatial_dimension = spatial_dimension, _ghost_type = _not_ghost);

			model.getFEEngine().interpolateOnIntegrationPoints(model.getVelocity(),
			velocity_field);

			/// integrate on fragments
			integrateFieldOnFragments(velocity_field, velocity);

			/// divide it by the fragments' mass
			Real * mass_storage = mass.storage();
			Real * velocity_storage = velocity.storage();

			UInt total_components = velocity.size() * velocity.getNbComponent();

			for (UInt i = 0; i < total_components; ++i) {
			velocity_storage[i] /= mass_storage[i];
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			/**
			* Given the distance @f$ \mathbf{r} @f$ between a quadrature point
			* and its center of mass, the moment of inertia is computed as \f[
			* I_\mathrm{CM} = \mathrm{tr}(\mathbf{r}\mathbf{r}^\mathrm{T})
			* \mathbf{I} - \mathbf{r}\mathbf{r}^\mathrm{T} \f] for more
			* information check Wikipedia
			* (http://en.wikipedia.org/wiki/Moment_of_inertia#Identities_for_a_skew-symmetric_matrix)
			*
			*/

			void FragmentManager::computeInertiaMoments() {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();

			computeCenterOfMass();

			/// compute local coordinates products with respect to the center of match
			ElementTypeMapArray<Real> moments_coords("moments_coords", id);
			moments_coords.initialize(model.getFEEngine(),
			_nb_component =
			spatial_dimension * spatial_dimension,
			_spatial_dimension = spatial_dimension,
			_ghost_type = _not_ghost, _default_value = 1.);

			/// loop over fragments
			for (auto && data :
			zip(iterateElementGroups(), make_view(mass_center, spatial_dimension))) {
			const auto & el_list = std::get<0>(data).getElements();
			auto & mass_center = std::get<1>(data);

			/// loop over elements of the fragment
			for (auto type :
			el_list.elementTypes(spatial_dimension, _not_ghost, _ek_regular)) {
			auto nb_quad_per_element =
			model.getFEEngine().getNbIntegrationPoints(type);

			auto & moments_coords_array = moments_coords(type);
			const auto & quad_coordinates_array = quad_coordinates(type);
			const auto & el_list_array = el_list(type);

			auto moments_begin =
			moments_coords_array.begin(spatial_dimension, spatial_dimension);
			auto quad_coordinates_begin =
			quad_coordinates_array.begin(spatial_dimension);

			Vector<Real> relative_coords(spatial_dimension);

			for (UInt el = 0; el < el_list_array.size(); ++el) {
			UInt global_el = el_list_array(el);

			/// loop over quadrature points
			for (UInt q = 0; q < nb_quad_per_element; ++q) {
			UInt global_q = global_el * nb_quad_per_element + q;
			Matrix<Real> moments_matrix = moments_begin[global_q];
			const Vector<Real> & quad_coord_vector =
			quad_coordinates_begin[global_q];

			/// to understand this read the documentation written just
			/// before this function
			relative_coords = quad_coord_vector;
			relative_coords -= mass_center;

			moments_matrix.outerProduct(relative_coords, relative_coords);
			Real trace = moments_matrix.trace();
			moments_matrix *= -1.;
			moments_matrix += Matrix<Real>::eye(spatial_dimension, trace);
			}
			}
			}
			}

			/// integrate moments
			Array<Real> integrated_moments(global_nb_fragment,
			spatial_dimension * spatial_dimension);

			integrateFieldOnFragments(moments_coords, integrated_moments);

			/// compute and store principal moments
			inertia_moments.resize(global_nb_fragment);
			principal_directions.resize(global_nb_fragment);

			auto integrated_moments_it =
			integrated_moments.begin(spatial_dimension, spatial_dimension);
			auto inertia_moments_it = inertia_moments.begin(spatial_dimension);
			auto principal_directions_it =
			principal_directions.begin(spatial_dimension, spatial_dimension);

			for (UInt frag = 0; frag < global_nb_fragment; ++frag,
			++integrated_moments_it, ++inertia_moments_it,
			++principal_directions_it) {
			integrated_moments_it->eig(inertia_moments_it, principal_directions_it);
			}

			if (dump_data) {
			createDumpDataArray(inertia_moments, "moments of inertia");
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::computeAllData(Real damage_limit) {
			AKANTU_DEBUG_IN();

			buildFragments(damage_limit);
			computeVelocity();
			computeInertiaMoments();
			computeNbElementsPerFragment();

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::storeMassDensityPerIntegrationPoint() {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();

			for (auto type : mesh.elementTypes(_spatial_dimension = spatial_dimension,
			_element_kind = _ek_regular)) {
			Array<Real> & mass_density_array = mass_density(type);

			UInt nb_element = mesh.getNbElement(type);
			UInt nb_quad_per_element = model.getFEEngine().getNbIntegrationPoints(type);
			mass_density_array.resize(nb_element * nb_quad_per_element);

			const Array<UInt> & mat_indexes = model.getMaterialByElement(type);

			Real * mass_density_it = mass_density_array.storage();

			/// store mass_density for each element and quadrature point
			for (UInt el = 0; el < nb_element; ++el) {
			Material & mat = model.getMaterial(mat_indexes(el));

			for (UInt q = 0; q < nb_quad_per_element; ++q, ++mass_density_it) {
			*mass_density_it = mat.getRho();
			}
			}
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::integrateFieldOnFragments(
			ElementTypeMapArray<Real> & field, Array<Real> & output) {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();
			UInt nb_component = output.getNbComponent();

			/// integration part
			output.resize(global_nb_fragment);
			output.zero();

			auto output_begin = output.begin(nb_component);

			/// loop over fragments
			for (auto && data : zip(iterateElementGroups(), fragment_index)) {
			const auto & el_list = std::get<0>(data).getElements();
			auto fragment_index = std::get<1>(data);

			/// loop over elements of the fragment
			for (auto type :
			el_list.elementTypes(spatial_dimension, _not_ghost, _ek_regular)) {

			UInt nb_quad_per_element =
			model.getFEEngine().getNbIntegrationPoints(type);

			const Array<Real> & density_array = mass_density(type);
			Array<Real> & field_array = field(type);
			const Array<UInt> & elements = el_list(type);

			/// generate array to be integrated by filtering fragment's elements
			Array<Real> integration_array(elements.size() * nb_quad_per_element,
			nb_component);

			auto field_array_begin = field_array.begin_reinterpret(
			nb_quad_per_element, nb_component,
			field_array.size() / nb_quad_per_element);
			auto density_array_begin = density_array.begin_reinterpret(
			nb_quad_per_element, density_array.size() / nb_quad_per_element);

			for (auto && data : enumerate(make_view(
			integration_array, nb_quad_per_element, nb_component))) {
			UInt global_el = elements(std::get<0>(data));
			auto & int_array = std::get<1>(data);
			int_array = field_array_begin[global_el];

			/// multiply field by density
			const Vector<Real> & density_vector = density_array_begin[global_el];

			for (UInt i = 0; i < nb_quad_per_element; ++i) {
			for (UInt j = 0; j < nb_component; ++j) {
			int_array(i, j) *= density_vector(i);
			}
			}
			}

			/// integrate the field over the fragment
			Array<Real> integrated_array(elements.size(), nb_component);
			model.getFEEngine().integrate(integration_array, integrated_array,
			nb_component, type, _not_ghost, elements);

			/// sum over all elements and store the result
			Vector<Real> output_tmp(output_begin[fragment_index]);
			output_tmp += std::accumulate(integrated_array.begin(nb_component),
			integrated_array.end(nb_component),
			Vector<Real>(nb_component));
			}
			}

			/// sum output over all processors
			const Communicator & comm = mesh.getCommunicator();
			comm.allReduce(output, SynchronizerOperation::_sum);

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			void FragmentManager::computeNbElementsPerFragment() {
			AKANTU_DEBUG_IN();

			UInt spatial_dimension = model.getSpatialDimension();
			nb_elements_per_fragment.resize(global_nb_fragment);
			nb_elements_per_fragment.zero();

			/// loop over fragments
			for (auto && data : zip(iterateElementGroups(), fragment_index)) {
			const auto & el_list = std::get<0>(data).getElements();
			auto fragment_index = std::get<1>(data);

			/// loop over elements of the fragment
			for (auto type :
			el_list.elementTypes(spatial_dimension, _not_ghost, _ek_regular)) {
			UInt nb_element = el_list(type).size();

			nb_elements_per_fragment(fragment_index) += nb_element;
			}
			}

			/// sum values over all processors
			const auto & comm = mesh.getCommunicator();
			comm.allReduce(nb_elements_per_fragment, SynchronizerOperation::_sum);

			if (dump_data) {
			createDumpDataArray(nb_elements_per_fragment, "elements per fragment");
			}

			AKANTU_DEBUG_OUT();
			}

			/* -------------------------------------------------------------------------- */
			template <typename T>
			void FragmentManager::createDumpDataArray(Array<T> & data, std::string name,
			bool fragment_index_output) {
			AKANTU_DEBUG_IN();

			if (data.empty()) {
			return;
			}

			auto & mesh_not_const = const_cast<Mesh &>(mesh);

			auto && spatial_dimension = mesh.getSpatialDimension();
			auto && nb_component = data.getNbComponent();
			auto && data_begin = data.begin(nb_component);

			/// loop over fragments
			for (auto && data : zip(iterateElementGroups(), fragment_index)) {
			const auto & fragment = std::get<0>(data);
			auto fragment_idx = std::get<1>(data);

			/// loop over cluster types
			for (auto && type : fragment.elementTypes(spatial_dimension)) {
			/// init mesh data
			auto & mesh_data = mesh_not_const.getDataPointer<T>(
			name, type, _not_ghost, nb_component);

			auto mesh_data_begin = mesh_data.begin(nb_component);

			/// fill mesh data
			for (const auto & elem : fragment.getElements(type)) {
			Vector<T> md_tmp = mesh_data_begin[elem];
			if (fragment_index_output) {
			md_tmp(0) = fragment_idx;
			} else {
			md_tmp = data_begin[fragment_idx];
			}
			}
			}
			}

			AKANTU_DEBUG_OUT();
			}

			} // namespace akantu

rAKA/src/model/solid_mechanics/solid_mechanics_model_cohesive/fragment_manager.cc2c2ac9d01138bugfixes/python_package…

fragment_manager.ccView Options

rAKA/src/model/solid_mechanics/solid_mechanics_model_cohesive/fragment_manager.cc
2c2ac9d01138bugfixes/python_package…

fragment_manager.cc
View Options