12 auto core_log = logging::core::get();
15 "FieldEvaluatorWorld"));
17 "FieldEvaluatorSync"));
19 "FieldEvaluatorSelf"));
30 MOFEM_LOG(
"FieldEvaluatorWorld", Sev::noisy) <<
"Field evaluator intreface";
42 boost::shared_ptr<SetIntegrationPtsMethodData> spd_ptr,
49 CHKERR m_field.
get_moab().get_entities_by_dimension(fe_meshset,
D, entities,
55 spd_ptr->treePtr.reset(
new AdaptiveKDTree(&m_field.
get_moab()));
56 CHKERR spd_ptr->treePtr->build_tree(entities, &spd_ptr->rooTreeSet);
62 boost::shared_ptr<SetIntegrationPtsMethodData> spd_ptr,
64 return buildTreeImpl<2>(spd_ptr, finite_element,
bit, mask);
69 boost::shared_ptr<SetIntegrationPtsMethodData> spd_ptr,
71 return buildTreeImpl<3>(spd_ptr, finite_element,
bit, mask);
75 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr)
76 : dataPtr(data_ptr) {}
83 if (
auto data_ptr = dataPtr.lock()) {
85 const auto nb_eval_points = data_ptr->nbEvalPoints;
86 const auto &shape_functions = data_ptr->shapeFunctions;
87 const auto &eval_pointentity_handle = data_ptr->evalPointEntityHandle;
89 if (
auto fe_ptr = data_ptr->feMethodPtr.lock()) {
94 if (fe_ptr.get() != fe_raw_ptr)
98 const auto fe_ent = fe.numeredEntFiniteElementPtr->getEnt();
100 const auto fe_dim = moab::CN::Dimension(fe_type);
102 auto &gauss_pts = fe.gaussPts;
106 gauss_pts.resize(4, nb_eval_points,
false);
108 &shape_functions(0, 0), &shape_functions(0, 1),
109 &shape_functions(0, 2), &shape_functions(0, 3)};
111 &gauss_pts(0, 0), &gauss_pts(1, 0), &gauss_pts(2, 0)};
114 for (
int nn = 0; nn != nb_eval_points; ++nn) {
115 if (eval_pointentity_handle[nn] == fe_ent) {
116 for (
const int i : {0, 1, 2}) {
117 t_gauss_pts(
i) = t_shape(
i + 1);
119 gauss_pts(3, nb_gauss_pts) = nn;
125 gauss_pts.resize(4, nb_gauss_pts,
true);
126 }
else if (fe_dim == 2) {
127 gauss_pts.resize(3, nb_eval_points,
false);
129 &shape_functions(0, 0), &shape_functions(0, 1),
130 &shape_functions(0, 2)};
132 &gauss_pts(0, 0), &gauss_pts(1, 0)};
134 for (
int nn = 0; nn != nb_eval_points; ++nn) {
135 if (eval_pointentity_handle[nn] == fe_ent) {
136 for (
const int i : {0, 1}) {
137 t_gauss_pts(
i) = t_shape(
i + 1);
139 gauss_pts(2, nb_gauss_pts) = nn;
145 gauss_pts.resize(3, nb_gauss_pts,
true);
148 "Dimension not implemented");
153 <<
"nbEvalOPoints / nbGau_sspt_s: " << nb_eval_points <<
" / "
155 MOFEM_LOG(
"SELF", Sev::noisy) <<
"gauss pts: " << gauss_pts;
160 "Pointer to element does not exists");
164 "Pointer to data does not exists");
171 const double *
const point,
const double distance,
const std::string problem,
172 const std::string finite_element,
173 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
175 boost::shared_ptr<CacheTuple> cache_ptr,
MoFEMTypes bh,
180 std::vector<EntityHandle> leafs_out;
181 CHKERR data_ptr->treePtr->distance_search(point, distance, leafs_out);
183 for (
auto lit : leafs_out)
188 MOFEM_LOG(
"SELF", Sev::noisy) <<
"tree entities: " << tree_ents;
190 data_ptr->evalPointEntityHandle.resize(data_ptr->nbEvalPoints);
191 std::fill(data_ptr->evalPointEntityHandle.begin(),
192 data_ptr->evalPointEntityHandle.end(), 0);
194 std::vector<double> local_coords;
195 std::vector<double> shape;
197 auto nb_eval_points = data_ptr->nbEvalPoints;
199 for (
auto tet : tree_ents) {
203 "Wrong element type, Field Evaluator not implemented for Quads "
208 CHKERR m_field.
get_moab().get_connectivity(tet, conn, num_nodes,
true);
210 if constexpr (
D == 3) {
212 local_coords.resize(3 * nb_eval_points);
213 shape.resize(4 * nb_eval_points);
215 std::array<double, 12> coords;
216 CHKERR m_field.
get_moab().get_coords(conn, num_nodes, coords.data());
219 coords.data(), &data_ptr->evalPoints[0], nb_eval_points,
221 CHKERR Tools::shapeFunMBTET<3>(&shape[0], &local_coords[0],
222 &local_coords[1], &local_coords[2],
227 &shape[0], &shape[1], &shape[2], &shape[3]};
229 &data_ptr->shapeFunctions(0, 0), &data_ptr->shapeFunctions(0, 1),
230 &data_ptr->shapeFunctions(0, 2), &data_ptr->shapeFunctions(0, 3)};
234 &local_coords[0], &local_coords[1], &local_coords[2]};
236 &(data_ptr->localCoords(0, 0)), &(data_ptr->localCoords(0, 1)),
237 &(data_ptr->localCoords(0, 2))};
239 for (
int n = 0;
n != nb_eval_points; ++
n) {
241 const double eps = data_ptr->eps;
242 if (t_shape(0) >= 0 -
eps && t_shape(0) <= 1 +
eps &&
244 t_shape(1) >= 0 -
eps && t_shape(1) <= 1 +
eps &&
246 t_shape(2) >= 0 -
eps && t_shape(2) <= 1 +
eps &&
248 t_shape(3) >= 0 -
eps && t_shape(3) <= 1 +
eps) {
250 data_ptr->evalPointEntityHandle[
n] = tet;
251 t_shape_data(i4) = t_shape(i4);
252 t_local_data(j3) = t_local(j3);
262 if constexpr (
D == 2) {
264 local_coords.resize(2 * nb_eval_points);
265 shape.resize(3 * nb_eval_points);
267 std::array<double, 9> coords;
268 CHKERR m_field.
get_moab().get_coords(conn, num_nodes, coords.data());
271 coords.data(), &data_ptr->evalPoints[0], nb_eval_points,
273 CHKERR Tools::shapeFunMBTRI<2>(&shape[0], &local_coords[0],
274 &local_coords[1], nb_eval_points);
278 &shape[0], &shape[1], &shape[2]};
280 &data_ptr->shapeFunctions(0, 0), &data_ptr->shapeFunctions(0, 1),
281 &data_ptr->shapeFunctions(0, 2)};
285 &local_coords[0], &local_coords[1]};
286 data_ptr->localCoords.resize(nb_eval_points, 2);
288 &(data_ptr->localCoords(0, 0)), &(data_ptr->localCoords(0, 1))};
290 for (
int n = 0;
n != nb_eval_points; ++
n) {
292 const double eps = data_ptr->eps;
293 if (t_shape(0) >= 0 -
eps && t_shape(0) <= 1 +
eps &&
295 t_shape(1) >= 0 -
eps && t_shape(1) <= 1 +
eps &&
297 t_shape(2) >= 0 -
eps && t_shape(2) <= 1 +
eps) {
299 data_ptr->evalPointEntityHandle[
n] = tet;
300 t_shape_data(i3) = t_shape(i3);
301 t_local_data(j2) = t_local(j2);
314 boost::shared_ptr<NumeredEntFiniteElement_multiIndex> numered_fes(
318 in_tets.insert_list(data_ptr->evalPointEntityHandle.begin(),
319 data_ptr->evalPointEntityHandle.end());
320 in_tets = in_tets.subset_by_dimension(
D);
323 MOFEM_LOG(
"SELF", Sev::noisy) <<
"in tets: " << in_tets << endl;
331 for (
auto peit = in_tets.pair_begin(); peit != in_tets.pair_end(); ++peit) {
336 peit->first, (*fe_miit)->getFEUId()));
340 peit->second, (*fe_miit)->getFEUId()));
344 MOFEM_LOG(
"FieldEvaluatorSelf", Sev::noisy);
345 for (
auto it = lo; it != hi; ++it) {
346 auto fe_bit = (*it)->getBitRefLevel();
347 if ((fe_bit & mask) != fe_bit)
349 if ((fe_bit &
bit).none())
351 numered_fes->insert(*it);
353 MOFEM_LOG(
"FieldEvaluatorSelf", Sev::noisy) << **it << endl;
356 numered_fes->insert(lo, hi);
358 MOFEM_LOG(
"FieldEvaluatorSelf", Sev::noisy)
359 <<
"numbered elements:" << std::endl;
360 for (; lo != hi; ++lo)
361 MOFEM_LOG(
"FieldEvaluatorSelf", Sev::noisy) << **lo << endl;
366 if (
auto fe_ptr = data_ptr->feMethodPtr.lock()) {
369 "Number elements %d to evaluate at proc %d",
373 cache_ptr = boost::make_shared<CacheTuple>();
376 MOFEM_LOG(
"FieldEvaluatorSelf", Sev::noisy)
377 <<
"If you call that function many times in the loop consider to "
378 "set cache_ptr outside of the loop. Otherwise code can be slow.";
382 lower_rank, upper_rank, numered_fes,
383 bh, cache_ptr, verb);
387 "Pointer to element does not exists");
395 const double *
const point,
const double distance,
const std::string problem,
396 const std::string finite_element,
397 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
399 boost::shared_ptr<CacheTuple> cache_ptr,
MoFEMTypes bh,
401 return evalFEAtThePointImpl<2>(point, distance, problem, finite_element,
402 data_ptr, lower_rank, upper_rank,
bit, mask,
403 cache_ptr, bh, verb);
408 const double *
const point,
const double distance,
const std::string problem,
409 const std::string finite_element,
410 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
412 boost::shared_ptr<CacheTuple> cache_ptr,
MoFEMTypes bh,
414 return evalFEAtThePointImpl<3>(point, distance, problem, finite_element,
415 data_ptr, lower_rank, upper_rank,
bit, mask,
416 cache_ptr, bh, verb);
423 const std::string finite_element,
424 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
431 struct Op :
public BaseOP {
435 const std::string finite_element,
436 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
441 : BaseOP(
NOSPACE, BaseOP::OPSPACE),
442 matDataPts(mat_data_pts), finiteElement(finite_element),
443 dataPtr(data_ptr), lowerRank(lower_rank), upperRank(upper_rank),
444 bitRefLevel(
bit), maskRefLevel(mask),
eps(
eps), bhType(bh),
454 auto &m_field = this->getPtrFE()->mField;
457 auto nb_integration_pts = getGaussPts().size2();
459 auto prb_name = getPtrFE()->problemPtr->getName();
460 auto cache_ptr = getPtrFE()->getCacheWeakPtr();
462 auto coords_at_pts = getCoordsAtGaussPts();
463 for (
int gg = 0; gg != nb_integration_pts; ++gg) {
464 double *coords_ptr = &coords_at_pts(gg, 0);
466 dataPtr->setEvalPoints(coords_ptr, 1);
467 CHKERR field_eval_ptr->evalFEAtThePoint<
D>(
468 coords_ptr,
eps, prb_name, finiteElement, dataPtr, lowerRank,
469 upperRank, bitRefLevel, maskRefLevel, cache_ptr.lock(),
MF_EXIST,
474 auto loop_size = dataPtr->feMethodPtr.lock()->getLoopSize();
475 if (loop_size != 1) {
476 if (verbosity >=
QUIET) {
478 LogManager::BitLineID | LogManager::BitScope);
480 <<
"Field evaluator wrong number of elements found: "
485 for (
auto &
m : matDataPts) {
486 auto nb_rows =
m.first->size1();
487 m.second->resize(nb_rows, nb_integration_pts,
false);
490 for (
auto ll = 0; ll != loop_size; ++ll) {
491 for (
int dd = 0; dd != nb_rows; ++dd) {
492 (*
m.second)(dd, gg) += (*
m.first)(dd, ll) / loop_size;
505 std::string finiteElement;
506 boost::shared_ptr<SetIntegrationPtsMethodData> dataPtr;
519 CHKERR data_ptr->treePtr->get_info(data_ptr->rooTreeSet, min, max, dep);
520 double dist = std::sqrt(pow(max[0] - min[0], 2) + pow(max[1] - min[1], 2) +
521 pow(max[2] - min[2], 2));
523 return new Op( mat_data_pts, finite_element, data_ptr,
524 lower_rank, upper_rank,
bit, mask, 1e-6 * dist, bh, verb);
529 boost::shared_ptr<MoFEM::ForcesAndSourcesCore> fe_method_ptr,
530 const double *eval_points,
const int nb_eval_points,
const double eps,
532 : feMethodPtr(fe_method_ptr), evalPoints(eval_points),
533 nbEvalPoints(nb_eval_points),
eps(
eps), verb(verb) {
539 const double *ptr,
const int nb_eval_points) {
541 nbEvalPoints = nb_eval_points;
542 localCoords.resize(nbEvalPoints, 3,
false);
543 shapeFunctions.resize(nbEvalPoints, 4,
false);
548FieldEvaluatorInterface::getDataOperator<2>(
550 const std::string finite_element,
551 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
554 return getDataOperatorImpl<2>(mat_data_pts, finite_element,
555 data_ptr, lower_rank, upper_rank,
bit, mask, bh,
561FieldEvaluatorInterface::getDataOperator<3>(
563 const std::string finite_element,
564 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
567 return getDataOperatorImpl<3>(mat_data_pts, finite_element,
568 data_ptr, lower_rank, upper_rank,
bit, mask, bh,
#define MOFEM_LOG_C(channel, severity, format,...)
VERBOSITY_LEVELS
Verbosity levels.
MoFEMTypes
Those types control how functions respond on arguments, f.e. error handling.
#define MoFEMFunctionReturnHot(a)
Last executable line of each PETSc function used for error handling. Replaces return()
#define MoFEMFunctionBegin
First executable line of each MoFEM function, used for error handling. Final line of MoFEM functions ...
@ MOFEM_DATA_INCONSISTENCY
#define MoFEMFunctionReturn(a)
Last executable line of each PETSc function used for error handling. Replaces return()
#define CHKERR
Inline error check.
#define MoFEMFunctionBeginHot
First executable line of each MoFEM function, used for error handling. Final line of MoFEM functions ...
multi_index_container< boost::shared_ptr< NumeredEntFiniteElement >, indexed_by< ordered_unique< tag< Unique_mi_tag >, const_mem_fun< NumeredEntFiniteElement::interface_type_EntFiniteElement, UId, &NumeredEntFiniteElement::getLocalUniqueId > >, ordered_non_unique< tag< Part_mi_tag >, member< NumeredEntFiniteElement, unsigned int, &NumeredEntFiniteElement::part > >, ordered_non_unique< tag< Ent_mi_tag >, const_mem_fun< NumeredEntFiniteElement::interface_type_RefEntity, EntityHandle, &NumeredEntFiniteElement::getEnt > >, ordered_non_unique< tag< Composite_Name_And_Part_mi_tag >, composite_key< NumeredEntFiniteElement, const_mem_fun< NumeredEntFiniteElement::interface_type_FiniteElement, boost::string_ref, &NumeredEntFiniteElement::getNameRef >, member< NumeredEntFiniteElement, unsigned int, &NumeredEntFiniteElement::part > > > > > NumeredEntFiniteElement_multiIndex
MultiIndex for entities for NumeredEntFiniteElement.
virtual const Problem * get_problem(const std::string problem_name) const =0
Get the problem object.
virtual const FiniteElement_multiIndex * get_finite_elements() const =0
Get the finite elements object.
virtual EntityHandle get_finite_element_meshset(const std::string name) const =0
static LoggerType & setLog(const std::string channel)
Set ans resset chanel logger.
#define MOFEM_LOG(channel, severity)
Log.
#define MOFEM_LOG_TAG(channel, tag)
Tag channel.
#define MOFEM_LOG_CHANNEL(channel)
Set and reset channel.
#define MOFEM_LOG_ATTRIBUTES(channel, bit)
Add attributes to channel.
virtual MoFEMErrorCode loop_finite_elements(const std::string problem_name, const std::string &fe_name, FEMethod &method, boost::shared_ptr< NumeredEntFiniteElement_multiIndex > fe_ptr=nullptr, MoFEMTypes bh=MF_EXIST, CacheTupleWeakPtr cache_ptr=CacheTupleSharedPtr(), int verb=DEFAULT_VERBOSITY)=0
Make a loop over finite elements.
FTensor::Index< 'i', SPACE_DIM > i
const double n
refractive index of diffusive medium
PetscErrorCode MoFEMErrorCode
MoFEM/PETSc error code.
std::bitset< BITREFLEVEL_SIZE > BitRefLevel
Bit structure attached to each entity identifying to what mesh entity is attached.
implementation of Data Operators for Forces and Sources
auto type_from_handle(const EntityHandle h)
get type from entity handle
FTensor::Index< 'm', 3 > m
virtual moab::Interface & get_moab()=0
virtual MoFEMErrorCode cache_problem_entities(const std::string prb_name, CacheTupleWeakPtr cache_ptr)=0
Cache variables.
virtual int get_comm_rank() const =0
UId getLocalUniqueIdCalculate() const
Generate UId for finite element entity.
Data on single entity (This is passed as argument to DataOperator::doWork)
MatrixDouble shapeFunctions
void setEvalPoints(const double *ptr, const int nb_eval_points)
SetIntegrationPtsMethodData()=delete
SetIntegrationPtsMethod()=delete
MoFEMErrorCode operator()(ForcesAndSourcesCore *fe_raw_ptr, int order_row, int order_col, int order_data)
Field evaluator interface.
MoFEMErrorCode query_interface(boost::typeindex::type_index type_index, UnknownInterface **iface) const
std::vector< std::pair< boost::shared_ptr< MatrixDouble >, boost::shared_ptr< MatrixDouble > > > MatDataPts
MoFEMErrorCode buildTreeImpl(boost::shared_ptr< SetIntegrationPtsMethodData > spd_ptr, const std::string finite_element, BitRefLevel bit, BitRefLevel mask)
FieldEvaluatorInterface(const MoFEM::Core &core)
MoFEMErrorCode evalFEAtThePointImpl(const double *const point, const double distance, const std::string problem, const std::string finite_element, boost::shared_ptr< SetIntegrationPtsMethodData > data_ptr, int lower_rank, int upper_rank, BitRefLevel bit, BitRefLevel mask, boost::shared_ptr< CacheTuple > cache_ptr, MoFEMTypes bh=MF_EXIST, VERBOSITY_LEVELS verb=QUIET)
ForcesAndSourcesCore::UserDataOperator * getDataOperatorImpl(MatDataPts mat_data_pts, const std::string finite_element, boost::shared_ptr< SetIntegrationPtsMethodData > data_ptr, int lower_rank, int upper_rank, BitRefLevel bit, BitRefLevel mask, MoFEMTypes bh=MF_EXIST, VERBOSITY_LEVELS verb=QUIET)
structure to get information from mofem into EntitiesFieldData
static boost::shared_ptr< SinkType > createSink(boost::shared_ptr< std::ostream > stream_ptr, std::string comm_filter)
Create a sink object.
static boost::shared_ptr< std::ostream > getStrmWorld()
Get the strm world object.
static boost::shared_ptr< std::ostream > getStrmSync()
Get the strm sync object.
static bool checkIfChannelExist(const std::string channel)
Check if channel exist.
static boost::shared_ptr< std::ostream > getStrmSelf()
Get the strm self object.
keeps basic data about problem
boost::shared_ptr< NumeredEntFiniteElement_multiIndex > numeredFiniteElementsPtr
store finite elements
base class for all interface classes
MoFEMErrorCode getInterface(IFACE *&iface) const
Get interface reference to pointer of interface.