v0.15.0
Loading...
Searching...
No Matches
forces_and_sources_testing_volume_element.cpp File Reference
#include <MoFEM.hpp>

Go to the source code of this file.

Functions

int main (int argc, char *argv[])
 

Variables

static char help [] = "...\n\n"
 

Function Documentation

◆ main()

int main ( int argc,
char * argv[] )

Definition at line 13 of file forces_and_sources_testing_volume_element.cpp.

13 {
14
15 MoFEM::Core::Initialize(&argc, &argv, (char *)0, help);
16
17 try {
18
19 moab::Core mb_instance;
20 moab::Interface &moab = mb_instance;
21 int rank;
22 MPI_Comm_rank(PETSC_COMM_WORLD, &rank);
23
24 PetscBool flg = PETSC_TRUE;
25 char mesh_file_name[255];
26#if PETSC_VERSION_GE(3, 6, 4)
27 CHKERR PetscOptionsGetString(PETSC_NULLPTR, "", "-my_file", mesh_file_name,
28 255, &flg);
29#else
30 CHKERR PetscOptionsGetString(PETSC_NULLPTR, PETSC_NULLPTR, "-my_file",
31 mesh_file_name, 255, &flg);
32#endif
33 if (flg != PETSC_TRUE) {
34 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
35 "*** ERROR -my_file (MESH FILE NEEDED)");
36 }
37
38 // Create MoFEM (Joseph) database
39 MoFEM::Core core(moab);
40 MoFEM::Interface &m_field = core;
41
42 const char *option;
43 option = "";
44 CHKERR moab.load_file(mesh_file_name, 0, option);
45
46 // set entitities bit level
47 BitRefLevel bit_level0;
48 bit_level0.set(0);
49 EntityHandle meshset_level0;
50 CHKERR moab.create_meshset(MESHSET_SET, meshset_level0);
51 CHKERR m_field.getInterface<BitRefManager>()->setBitRefLevelByDim(
52 0, 3, bit_level0);
53
54 // Fields
55 CHKERR m_field.add_field("FIELD1", H1, AINSWORTH_LEGENDRE_BASE, 1);
56 CHKERR m_field.add_field("FIELD2", H1, AINSWORTH_LEGENDRE_BASE, 3);
57 CHKERR m_field.add_field("FIELD3", NOFIELD, NOBASE, 3);
58 CHKERR m_field.add_field("MESH_NODE_POSITIONS", H1, AINSWORTH_LEGENDRE_BASE,
59 3);
60
61 {
62 // Creating and adding no field entities.
63 const double coords[] = {0, 0, 0};
64 EntityHandle no_field_vertex;
65 CHKERR m_field.get_moab().create_vertex(coords, no_field_vertex);
66 Range range_no_field_vertex;
67 range_no_field_vertex.insert(no_field_vertex);
68 CHKERR m_field.getInterface<BitRefManager>()->setBitRefLevel(
69 range_no_field_vertex, BitRefLevel().set());
70 EntityHandle meshset = m_field.get_field_meshset("FIELD3");
71 CHKERR m_field.get_moab().add_entities(meshset, range_no_field_vertex);
72 }
73
74 // FE
75 CHKERR m_field.add_finite_element("TEST_FE1");
76 CHKERR m_field.add_finite_element("TEST_FE2");
77
78 // Define rows/cols and element data
79 CHKERR m_field.modify_finite_element_add_field_row("TEST_FE1", "FIELD1");
80 CHKERR m_field.modify_finite_element_add_field_col("TEST_FE1", "FIELD2");
81 CHKERR m_field.modify_finite_element_add_field_data("TEST_FE1", "FIELD1");
82 CHKERR m_field.modify_finite_element_add_field_data("TEST_FE1", "FIELD2");
84 "MESH_NODE_POSITIONS");
85
86 CHKERR m_field.modify_finite_element_add_field_row("TEST_FE2", "FIELD3");
87 CHKERR m_field.modify_finite_element_add_field_col("TEST_FE2", "FIELD1");
88 CHKERR m_field.modify_finite_element_add_field_data("TEST_FE2", "FIELD1");
89 CHKERR m_field.modify_finite_element_add_field_data("TEST_FE2", "FIELD3");
91 "MESH_NODE_POSITIONS");
92
93 // Problem
94 CHKERR m_field.add_problem("TEST_PROBLEM");
95
96 // set finite elements for problem
97 CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM",
98 "TEST_FE1");
99 CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM",
100 "TEST_FE2");
101 // set refinement level for problem
102 CHKERR m_field.modify_problem_ref_level_add_bit("TEST_PROBLEM", bit_level0);
103
104 // meshset consisting all entities in mesh
105 EntityHandle root_set = moab.get_root_set();
106 // add entities to field
107 CHKERR m_field.add_ents_to_field_by_type(root_set, MBTET, "FIELD1");
108 CHKERR m_field.add_ents_to_field_by_type(root_set, MBTET, "FIELD2");
109 CHKERR m_field.add_ents_to_field_by_type(root_set, MBTET,
110 "MESH_NODE_POSITIONS");
111
112 // add entities to finite element
113 CHKERR m_field.add_ents_to_finite_element_by_type(root_set, MBTET,
114 "TEST_FE1");
115 CHKERR m_field.add_ents_to_finite_element_by_type(root_set, MBTET,
116 "TEST_FE2");
117
118 // set app. order
119 // see Hierarchic Finite Element Bases on Unstructured Tetrahedral Meshes
120 // (Mark Ainsworth & Joe Coyle)
121 int order = 4;
122 CHKERR m_field.set_field_order(root_set, MBTET, "FIELD1", order);
123 CHKERR m_field.set_field_order(root_set, MBTRI, "FIELD1", order);
124 CHKERR m_field.set_field_order(root_set, MBEDGE, "FIELD1", order);
125 CHKERR m_field.set_field_order(root_set, MBVERTEX, "FIELD1", 1);
126 CHKERR m_field.set_field_order(root_set, MBTET, "FIELD2", order);
127 CHKERR m_field.set_field_order(root_set, MBTRI, "FIELD2", order);
128 CHKERR m_field.set_field_order(root_set, MBEDGE, "FIELD2", order);
129 CHKERR m_field.set_field_order(root_set, MBVERTEX, "FIELD2", 1);
130
131 CHKERR m_field.set_field_order(root_set, MBTET, "MESH_NODE_POSITIONS", 2);
132 CHKERR m_field.set_field_order(root_set, MBTRI, "MESH_NODE_POSITIONS", 2);
133 CHKERR m_field.set_field_order(root_set, MBEDGE, "MESH_NODE_POSITIONS", 2);
134 CHKERR m_field.set_field_order(root_set, MBVERTEX, "MESH_NODE_POSITIONS",
135 1);
136
137 /****/
138 // build database
139 // build field
140 CHKERR m_field.build_fields();
141 // build finite elemnts
143 // build adjacencies
144 CHKERR m_field.build_adjacencies(bit_level0);
145 // build problem
146 ProblemsManager *prb_mng_ptr;
147 CHKERR m_field.getInterface(prb_mng_ptr);
148 // const Problem_multiIndex *problems_ptr;
149 CHKERR prb_mng_ptr->buildProblem("TEST_PROBLEM", false);
150
151 /****/
152 // mesh partitioning
153 // partition
154 CHKERR prb_mng_ptr->partitionSimpleProblem("TEST_PROBLEM");
155 CHKERR prb_mng_ptr->partitionFiniteElements("TEST_PROBLEM");
156 // what are ghost nodes, see Petsc Manual
157 CHKERR prb_mng_ptr->partitionGhostDofs("TEST_PROBLEM");
158
159 // set from positions of 10 node tets
160 Projection10NodeCoordsOnField ent_method(m_field, "MESH_NODE_POSITIONS");
161 CHKERR m_field.loop_dofs("MESH_NODE_POSITIONS", ent_method);
162
163 typedef tee_device<std::ostream, std::ofstream> TeeDevice;
164 typedef stream<TeeDevice> TeeStream;
165
166 std::ofstream ofs("forces_and_sources_testing_volume_element.txt");
167 TeeDevice my_tee(std::cout, ofs);
168 TeeStream my_split(my_tee);
169
170 struct MyOp1 : public VolumeElementForcesAndSourcesCore::UserDataOperator {
171
172 TeeStream &my_split;
173 MyOp1(TeeStream &_my_split, char type)
175 "FIELD2", type),
176 my_split(_my_split) {}
177
178 MoFEMErrorCode doWork(int side, EntityType type,
181 my_split << "NH1" << std::endl;
182 my_split << "side: " << side << " type: " << type << std::endl;
183 my_split << data << std::endl;
184 my_split << std::setprecision(3) << getVolume() << std::endl;
185 my_split << std::setprecision(3) << getCoords() << std::endl;
186 my_split << std::setprecision(3) << getCoordsAtGaussPts() << std::endl;
188 }
189
190 MoFEMErrorCode doWork(int row_side, int col_side, EntityType row_type,
191 EntityType col_type,
193 EntitiesFieldData::EntData &col_data) {
195 my_split << "NH1NH1" << std::endl;
196 my_split << "row side: " << row_side << " row_type: " << row_type
197 << std::endl;
198 my_split << row_data << std::endl;
199 my_split << "NH1NH1" << std::endl;
200 my_split << "col side: " << col_side << " col_type: " << col_type
201 << std::endl;
202 my_split << col_data << std::endl;
203
204 VectorInt row_indices, col_indices;
205 CHKERR getProblemRowIndices("FIELD1", row_type, row_side, row_indices);
206 CHKERR getProblemColIndices("FIELD2", col_type, col_side, col_indices);
207
208 if (row_indices.size() != row_data.getIndices().size()) {
209 std::cerr << row_indices << std::endl;
210 std::cerr << row_data.getIndices() << std::endl;
211 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
212 "row inconsistency %d != %d", row_indices.size(),
213 row_data.getIndices().size());
214 }
215
216 if (col_indices.size() != col_data.getIndices().size()) {
217 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
218 "row inconsistency %d != %d", col_indices.size(),
219 col_data.getIndices().size());
220 }
221
222 for (unsigned int rr = 0; rr < row_indices.size(); rr++) {
223 if (row_indices[rr] != row_data.getIndices()[rr]) {
224 std::cerr << row_indices << std::endl;
225 std::cerr << row_data.getIndices() << std::endl;
226 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
227 "row inconsistency");
228 }
229 }
230
231 for (unsigned int cc = 0; cc < col_indices.size(); cc++) {
232 if (col_indices[cc] != col_data.getIndices()[cc]) {
233 std::cerr << col_indices << std::endl;
234 std::cerr << col_data.getIndices() << std::endl;
235 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
236 "row inconsistency");
237 }
238 }
239
241 }
242 };
243
244 struct MyOp2 : public VolumeElementForcesAndSourcesCore::UserDataOperator {
245
246 TeeStream &my_split;
247 MyOp2(TeeStream &_my_split, char type)
249 "FIELD1", type),
250 my_split(_my_split) {
251 sYmm = 0;
252 }
253
254 MoFEMErrorCode doWork(int side, EntityType type,
257
258 my_split << "NOFIELD" << std::endl;
259 my_split << "side: " << side << " type: " << type << std::endl;
260 my_split << data << std::endl;
262 }
263
264 MoFEMErrorCode doWork(int row_side, int col_side, EntityType row_type,
265 EntityType col_type,
267 EntitiesFieldData::EntData &col_data) {
269
270 unSetSymm();
271
272 my_split << "NOFILEDH1" << std::endl;
273 my_split << "row side: " << row_side << " row_type: " << row_type
274 << std::endl;
275 my_split << row_data << std::endl;
276 my_split << "col side: " << col_side << " col_type: " << col_type
277 << std::endl;
278 my_split << col_data << std::endl;
279
281 }
282 };
283
285
286 fe1.getOpPtrVector().push_back(
287 new MyOp1(my_split, ForcesAndSourcesCore::UserDataOperator::OPROW));
288 fe1.getOpPtrVector().push_back(
289 new MyOp1(my_split, ForcesAndSourcesCore::UserDataOperator::OPROWCOL));
290
292 fe2.getOpPtrVector().push_back(
293 new MyOp2(my_split, ForcesAndSourcesCore::UserDataOperator::OPROW));
294 fe2.getOpPtrVector().push_back(
295 new MyOp2(my_split, ForcesAndSourcesCore::UserDataOperator::OPROWCOL));
296
297 CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "TEST_FE1", fe1);
298 CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "TEST_FE2", fe2);
299 }
301
303
304 return 0;
305}
ForcesAndSourcesCore::UserDataOperator UserDataOperator
#define CATCH_ERRORS
Catch errors.
@ AINSWORTH_LEGENDRE_BASE
Ainsworth Cole (Legendre) approx. base nme:nme847.
Definition definitions.h:60
@ NOBASE
Definition definitions.h:59
#define MoFEMFunctionReturnHot(a)
Last executable line of each PETSc function used for error handling. Replaces return()
@ NOFIELD
scalar or vector of scalars describe (no true field)
Definition definitions.h:84
@ H1
continuous field
Definition definitions.h:85
@ MOFEM_DATA_INCONSISTENCY
Definition definitions.h:31
#define CHKERR
Inline error check.
#define MoFEMFunctionBeginHot
First executable line of each MoFEM function, used for error handling. Final line of MoFEM functions ...
constexpr int order
tee_device< std::ostream, std::ofstream > TeeDevice
virtual MoFEMErrorCode add_finite_element(const std::string &fe_name, enum MoFEMTypes bh=MF_EXCL, int verb=DEFAULT_VERBOSITY)=0
add finite element
virtual MoFEMErrorCode build_finite_elements(int verb=DEFAULT_VERBOSITY)=0
Build finite elements.
virtual MoFEMErrorCode modify_finite_element_add_field_col(const std::string &fe_name, const std::string name_row)=0
set field col which finite element use
virtual MoFEMErrorCode add_ents_to_finite_element_by_type(const EntityHandle entities, const EntityType type, const std::string name, const bool recursive=true)=0
add entities to finite element
virtual MoFEMErrorCode modify_finite_element_add_field_row(const std::string &fe_name, const std::string name_row)=0
set field row which finite element use
virtual MoFEMErrorCode modify_finite_element_add_field_data(const std::string &fe_name, const std::string name_field)=0
set finite element field data
virtual MoFEMErrorCode build_fields(int verb=DEFAULT_VERBOSITY)=0
virtual MoFEMErrorCode set_field_order(const EntityHandle meshset, const EntityType type, const std::string &name, const ApproximationOrder order, int verb=DEFAULT_VERBOSITY)=0
Set order approximation of the entities in the field.
virtual MoFEMErrorCode add_ents_to_field_by_type(const Range &ents, const EntityType type, const std::string &name, int verb=DEFAULT_VERBOSITY)=0
Add entities to field meshset.
virtual MoFEMErrorCode loop_dofs(const Problem *problem_ptr, const std::string &field_name, RowColData rc, DofMethod &method, int lower_rank, int upper_rank, int verb=DEFAULT_VERBOSITY)=0
Make a loop over dofs.
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.
MoFEMErrorCode partitionGhostDofs(const std::string name, int verb=VERBOSE)
determine ghost nodes
MoFEMErrorCode partitionSimpleProblem(const std::string name, int verb=VERBOSE)
partition problem dofs
MoFEMErrorCode buildProblem(const std::string name, const bool square_matrix, int verb=VERBOSE)
build problem data structures
MoFEMErrorCode partitionFiniteElements(const std::string name, bool part_from_moab=false, int low_proc=-1, int hi_proc=-1, int verb=VERBOSE)
partition finite elements
virtual MoFEMErrorCode add_problem(const std::string &name, enum MoFEMTypes bh=MF_EXCL, int verb=DEFAULT_VERBOSITY)=0
Add problem.
virtual MoFEMErrorCode modify_problem_ref_level_add_bit(const std::string &name_problem, const BitRefLevel &bit)=0
add ref level to problem
virtual MoFEMErrorCode modify_problem_add_finite_element(const std::string name_problem, const std::string &fe_name)=0
add finite element to problem, this add entities assigned to finite element to a particular problem
char mesh_file_name[255]
PetscErrorCode MoFEMErrorCode
MoFEM/PETSc error code.
std::bitset< BITREFLEVEL_SIZE > BitRefLevel
Bit structure attached to each entity identifying to what mesh entity is attached.
Definition Types.hpp:40
UBlasVector< int > VectorInt
Definition Types.hpp:67
PetscErrorCode PetscOptionsGetString(PetscOptions *, const char pre[], const char name[], char str[], size_t size, PetscBool *set)
Managing BitRefLevels.
virtual moab::Interface & get_moab()=0
virtual EntityHandle get_field_meshset(const std::string name) const =0
get field meshset
virtual MoFEMErrorCode build_adjacencies(const Range &ents, int verb=DEFAULT_VERBOSITY)=0
build adjacencies
virtual MoFEMErrorCode add_field(const std::string name, const FieldSpace space, const FieldApproximationBase base, const FieldCoefficientsNumber nb_of_coefficients, const TagType tag_type=MB_TAG_SPARSE, const enum MoFEMTypes bh=MF_EXCL, int verb=DEFAULT_VERBOSITY)=0
Add field.
Core (interface) class.
Definition Core.hpp:82
static MoFEMErrorCode Initialize(int *argc, char ***args, const char file[], const char help[])
Initializes the MoFEM database PETSc, MOAB and MPI.
Definition Core.cpp:72
static MoFEMErrorCode Finalize()
Checks for options to be called at the conclusion of the program.
Definition Core.cpp:118
Deprecated interface functions.
Data on single entity (This is passed as argument to DataOperator::doWork)
const VectorInt & getIndices() const
Get global indices of dofs on entity.
Problem manager is used to build and partition problems.
Projection of edge entities with one mid-node on hierarchical basis.
MoFEMErrorCode getInterface(IFACE *&iface) const
Get interface reference to pointer of interface.

Variable Documentation

◆ help

char help[] = "...\n\n"
static