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()));
343 for (
auto it = lo; it != hi; ++it) {
344 auto fe_bit = (*it)->getBitRefLevel();
345 if ((fe_bit & mask) != fe_bit)
347 if ((fe_bit &
bit).none())
349 numered_fes->insert(*it);
352 numered_fes->insert(lo, hi);
356 std::cerr <<
"numered elements:" << std::endl;
357 for (; lo != hi; ++lo)
359 std::cerr << **lo << endl;
362 std::cerr << std::endl;
364 if (
auto fe_ptr = data_ptr->feMethodPtr.lock()) {
367 "Number elements %d to evaluate at proc %d",
372 cache_ptr = boost::make_shared<CacheTuple>();
375 MOFEM_LOG(
"FieldEvaluatorSync", Sev::noisy)
376 <<
"If you call that function many times in the loop consider to "
378 "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);
424 const std::string finite_element,
425 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
432 struct Op :
public BaseOP {
437 const std::string finite_element,
438 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
443 : BaseOP(
NOSPACE, BaseOP::OPSPACE), vecDataPts(vec_data_pts),
444 matDataPts(mat_data_pts), finiteElement(finite_element),
445 dataPtr(data_ptr), lowerRank(lower_rank), upperRank(upper_rank),
446 bitRefLevel(
bit), maskRefLevel(mask),
eps(
eps), bhType(bh),
456 auto &m_field = this->getPtrFE()->mField;
459 auto nb_integration_pts = getGaussPts().size2();
460 auto &spatial_coords = getCoordsAtGaussPts();
462 auto prb_name = getPtrFE()->problemPtr->getName();
463 auto cache_ptr = getPtrFE()->getCacheWeakPtr();
465 auto coords_at_pts = getCoordsAtGaussPts();
466 for (
int gg = 0; gg != nb_integration_pts; ++gg) {
467 double *coords_ptr = &coords_at_pts(gg, 0);
469 dataPtr->setEvalPoints(coords_ptr, 1);
470 CHKERR field_eval_ptr->evalFEAtThePoint<
D>(
471 coords_ptr,
eps, prb_name, finiteElement, dataPtr, lowerRank,
472 upperRank, bitRefLevel, maskRefLevel, cache_ptr.lock(),
MF_EXIST,
477 auto loop_size = dataPtr->feMethodPtr.lock()->getLoopSize();
478 if (loop_size != 1) {
479 if (verbosity >=
QUIET) {
481 LogManager::BitLineID | LogManager::BitScope);
483 <<
"Field evaluator wrong number of elements found %d",
488 for (
auto &
v : vecDataPts) {
489 v.second->resize(nb_integration_pts,
false);
492 for (
auto ll = 0; ll != loop_size; ++ll)
493 (*
v.second)(gg) += (*
v.first)(ll) / loop_size;
496 for (
auto &
m : matDataPts) {
497 auto nb_rows =
m.first->size1();
498 m.second->resize(nb_rows, nb_integration_pts,
false);
501 for (
auto ll = 0; ll != loop_size; ++ll) {
502 for (
int dd = 0; dd != nb_rows; ++dd) {
503 (*
m.second)(dd, gg) += (*
m.first)(dd, ll) / loop_size;
517 std::string finiteElement;
518 boost::shared_ptr<SetIntegrationPtsMethodData> dataPtr;
531 CHKERR data_ptr->treePtr->get_info(data_ptr->rooTreeSet, min, max, dep);
532 double dist = std::sqrt(pow(max[0] - min[0], 2) + pow(max[1] - min[1], 2) +
533 pow(max[2] - min[2], 2));
535 return new Op(vec_data_pts, mat_data_pts, finite_element, data_ptr,
536 lower_rank, upper_rank,
bit, mask, 1e-6 * dist, bh, verb);
541 boost::shared_ptr<MoFEM::ForcesAndSourcesCore> fe_method_ptr,
542 const double *eval_points,
const int nb_eval_points,
const double eps,
544 : feMethodPtr(fe_method_ptr), evalPoints(eval_points),
545 nbEvalPoints(nb_eval_points),
eps(
eps), verb(verb) {
551 const double *ptr,
const int nb_eval_points) {
553 nbEvalPoints = nb_eval_points;
554 localCoords.resize(nbEvalPoints, 3,
false);
555 shapeFunctions.resize(nbEvalPoints, 4,
false);
560FieldEvaluatorInterface::getDataOperator<2>(
563 const std::string finite_element,
564 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
567 return getDataOperatorImpl<2>(vec_data_pts, mat_data_pts, finite_element,
568 data_ptr, lower_rank, upper_rank,
bit, mask, bh,
574FieldEvaluatorInterface::getDataOperator<3>(
577 const std::string finite_element,
578 boost::shared_ptr<SetIntegrationPtsMethodData> data_ptr,
int lower_rank,
581 return getDataOperatorImpl<3>(vec_data_pts, mat_data_pts, finite_element,
582 data_ptr, lower_rank, upper_rank,
bit, mask, bh,
#define MOFEM_LOG_SYNCHRONISE(comm)
Synchronise "SYNC" channel.
#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 v
phase velocity of light in medium (cm/ns)
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 MPI_Comm & get_comm() const =0
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
ForcesAndSourcesCore::UserDataOperator * getDataOperatorImpl(VecDataPts vec_data_pts, 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)
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)
std::vector< std::pair< boost::shared_ptr< VectorDouble >, boost::shared_ptr< VectorDouble > > > VecDataPts
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.