Homec4science
Diffusion akantu (bugfixes/google-test-ref-changed) Browse

rAKA/src/model/solid_mechanics/material_selector.hh
55f303b4af2ebugfixes/google-test…

Last Change
Raw File

material_selector.hh
View Options

/**
* @file material_selector.hh
*
* @author Lucas Frerot <lucas.frerot@epfl.ch>
* @author Nicolas Richart <nicolas.richart@epfl.ch>
*
* @date creation: Wed Nov 13 2013
* @date last modification: Thu Dec 17 2015
*
* @brief class describing how to choose a material for a given element
*
* @section LICENSE
*
* Copyright (©) 2014, 2015 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 "mesh.hh"
#include "element.hh"
/* -------------------------------------------------------------------------- */
#include <memory>
/* -------------------------------------------------------------------------- */
#ifndef __AKANTU_MATERIAL_SELECTOR_HH__
#define __AKANTU_MATERIAL_SELECTOR_HH__
/* -------------------------------------------------------------------------- */
namespace akantu {
class SolidMechanicsModel;
/**
* main class to assign same or different materials for different
* elements
*/
class MaterialSelector {
public:
MaterialSelector() = default;
virtual ~MaterialSelector() = default;
virtual inline UInt operator()(const Element & element) {
if (fallback_selector)
return (*fallback_selector)(element);
return fallback_value;
}
inline void setFallback(UInt f) { fallback_value = f; }
inline void
setFallback(const std::shared_ptr<MaterialSelector> & fallback_selector) {
this->fallback_selector = fallback_selector;
}
inline std::shared_ptr<MaterialSelector> & getFallbackSelector() {
return this->fallback_selector;
}
inline UInt getFallbackValue() {
return this->fallback_value;
}
protected:
UInt fallback_value{0};
std::shared_ptr<MaterialSelector> fallback_selector;
};
/* -------------------------------------------------------------------------- */
/**
* class that assigns the first material to regular elements by default
*/
class DefaultMaterialSelector : public MaterialSelector {
public:
explicit DefaultMaterialSelector(
const ElementTypeMapArray<UInt> & material_index)
: material_index(material_index) {}
UInt operator()(const Element & element) override {
if (not material_index.exists(element.type, element.ghost_type))
return MaterialSelector::operator()(element);
const auto & mat_indexes = material_index(element.type, element.ghost_type);
if (element.element < mat_indexes.size()) {
auto && tmp_mat = mat_indexes(element.element);
if (tmp_mat != UInt(-1))
return tmp_mat;
}
return MaterialSelector::operator()(element);
}
private:
const ElementTypeMapArray<UInt> & material_index;
};
/* -------------------------------------------------------------------------- */
/**
* Use elemental data to assign materials
*/
template <typename T>
class ElementDataMaterialSelector : public MaterialSelector {
public:
ElementDataMaterialSelector(const ElementTypeMapArray<T> & element_data,
const SolidMechanicsModel & model,
UInt first_index = 1)
: element_data(element_data), model(model), first_index(first_index) {}
inline T elementData(const Element & element) {
DebugLevel dbl = debug::getDebugLevel();
debug::setDebugLevel(dblError);
T data = element_data(element.type, element.ghost_type)(element.element);
debug::setDebugLevel(dbl);
return data;
}
inline UInt operator()(const Element & element) override {
return MaterialSelector::operator()(element);
}
protected:
/// list of element with the specified data (i.e. tag value)
const ElementTypeMapArray<T> & element_data;
/// the model that the materials belong
const SolidMechanicsModel & model;
/// first material index: equal to 1 if none specified
UInt first_index;
};
/* -------------------------------------------------------------------------- */
/**
* class to use mesh data information to assign different materials
* where name is the tag value: tag_0, tag_1
*/
template <typename T>
class MeshDataMaterialSelector : public ElementDataMaterialSelector<T> {
public:
MeshDataMaterialSelector(const std::string & name,
const SolidMechanicsModel & model,
UInt first_index = 1);
};
} // namespace akantu
#endif /* __AKANTU_MATERIAL_SELECTOR_HH__ */
c4science · Help