v0.14.0
Loading...
Searching...
No Matches
Functions | Variables
build_composite_problem.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 14 of file build_composite_problem.cpp.

14 {
15
16 MoFEM::Core::Initialize(&argc, &argv, (char *)0, help);
17
18 try {
19
20 PetscBool flg = PETSC_TRUE;
21 char mesh_file_name[255];
22#if PETSC_VERSION_GE(3, 6, 4)
23 CHKERR PetscOptionsGetString(PETSC_NULL, "", "-my_file", mesh_file_name,
24 255, &flg);
25#else
26 CHKERR PetscOptionsGetString(PETSC_NULL, PETSC_NULL, "-my_file",
27 mesh_file_name, 255, &flg);
28#endif
29 if (flg != PETSC_TRUE) {
30 SETERRQ(PETSC_COMM_SELF, 1, "*** ERROR -my_file (MESH FILE NEEDED)");
31 }
32
33 // read mesh and create moab and mofem data structures
34
35 moab::Core mb_instance;
36 moab::Interface &moab = mb_instance;
37 const char *option;
38 option = "";
39 CHKERR moab.load_file(mesh_file_name, 0, option);
40
41
42 MoFEM::Core core(moab);
43 MoFEM::Interface &m_field = core;
44
45 EntityHandle root_set = moab.get_root_set();
46 Range tets;
47 CHKERR moab.get_entities_by_type(root_set, MBTET, tets, false);
48
49 ProblemsManager *prb_mng_ptr = m_field.getInterface<ProblemsManager>();
50 CommInterface *comm_interface_ptr = m_field.getInterface<CommInterface>();
51 CHKERR comm_interface_ptr->partitionMesh(
52 tets, 3, 2, m_field.get_comm_size(), NULL, NULL, NULL);
53
54 EntityHandle part_set;
55 CHKERR moab.create_meshset(MESHSET_SET, part_set);
56
57 ParallelComm *pcomm = ParallelComm::get_pcomm(&moab, MYPCOMM_INDEX);
58 if (pcomm == NULL)
59 SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
60 "Communicator should be allocated");
61
62 Tag part_tag = pcomm->part_tag();
63 Range proc_ents;
64 Range tagged_sets;
65 CHKERR m_field.get_moab().get_entities_by_type_and_tag(
66 0, MBENTITYSET, &part_tag, NULL, 1, tagged_sets,
67 moab::Interface::UNION);
68 for (Range::iterator mit = tagged_sets.begin(); mit != tagged_sets.end();
69 mit++) {
70 int part;
71 CHKERR moab.tag_get_data(part_tag, &*mit, 1, &part);
72 if (part == m_field.get_comm_rank()) {
73 // pcomm->partition_sets().insert(*mit);
74 CHKERR moab.get_entities_by_type(*mit, MBTET, proc_ents, true);
75 CHKERR moab.add_entities(part_set, proc_ents);
76 }
77 }
78
79 Skinner skin(&m_field.get_moab());
80 Range tets_skin;
81 CHKERR skin.find_skin(0, tets, false, tets_skin);
82 Range proc_ents_skin[4];
83 proc_ents_skin[3] = proc_ents;
84 CHKERR skin.find_skin(0, proc_ents, false, proc_ents_skin[2]);
85 proc_ents_skin[2] = subtract(proc_ents_skin[2], tets_skin);
86 CHKERR moab.get_adjacencies(proc_ents_skin[2], 1, false, proc_ents_skin[1],
87 moab::Interface::UNION);
88 CHKERR moab.get_connectivity(proc_ents_skin[1], proc_ents_skin[0], true);
89 for (int dd = 0; dd != 3; dd++) {
90 CHKERR moab.add_entities(part_set, proc_ents_skin[dd]);
91 }
92
93 if (0) {
94 std::ostringstream file_skin;
95 file_skin << "out_skin_" << m_field.get_comm_rank() << ".vtk";
96 EntityHandle meshset_skin;
97 CHKERR moab.create_meshset(MESHSET_SET, meshset_skin);
98 CHKERR moab.add_entities(meshset_skin, proc_ents_skin[2]);
99 CHKERR moab.add_entities(meshset_skin, proc_ents_skin[1]);
100 CHKERR moab.add_entities(meshset_skin, proc_ents_skin[0]);
101 CHKERR moab.write_file(file_skin.str().c_str(), "VTK", "", &meshset_skin,
102 1);
103 }
104
105 CHKERR pcomm->resolve_shared_ents(0, proc_ents, 3, -1, proc_ents_skin);
106 Range owned_tets = proc_ents;
107
108 if (0) {
109 std::ostringstream file_owned;
110 file_owned << "out_owned_" << m_field.get_comm_rank() << ".vtk";
111 EntityHandle meshset_owned;
112 CHKERR moab.create_meshset(MESHSET_SET, meshset_owned);
113 CHKERR moab.add_entities(meshset_owned, owned_tets);
114 CHKERR moab.write_file(file_owned.str().c_str(), "VTK", "",
115 &meshset_owned, 1);
116 }
117 Range shared_ents;
118 // Get entities shared with all other processors
119 CHKERR pcomm->get_shared_entities(-1, shared_ents);
120
121 if (0) {
122 std::ostringstream file_shared_owned;
123 file_shared_owned << "out_shared_owned_" << m_field.get_comm_rank()
124 << ".vtk";
125 EntityHandle meshset_shared_owned;
126 CHKERR moab.create_meshset(MESHSET_SET, meshset_shared_owned);
127 CHKERR moab.add_entities(meshset_shared_owned, shared_ents);
128 CHKERR moab.write_file(file_shared_owned.str().c_str(), "VTK", "",
129 &meshset_shared_owned, 1);
130 }
131
132 // set entities bit level
133 BitRefLevel bit_level0;
134 bit_level0.set(0);
135 CHKERR m_field.getInterface<BitRefManager>()->setBitRefLevelByDim(
136 part_set, 3, bit_level0);
137
138 // Fields
139 CHKERR m_field.add_field("F1", H1, AINSWORTH_LEGENDRE_BASE, 1);
140 CHKERR m_field.add_field("F2", H1, AINSWORTH_LEGENDRE_BASE, 1);
141 // add entities to field
142 CHKERR m_field.add_ents_to_field_by_type(part_set, MBTET, "F1");
143 CHKERR m_field.add_ents_to_field_by_type(part_set, MBTET, "F2");
144 int order = 4;
145 CHKERR m_field.set_field_order(part_set, MBTET, "F1", order);
146 CHKERR m_field.set_field_order(part_set, MBTRI, "F1", order);
147 CHKERR m_field.set_field_order(part_set, MBEDGE, "F1", order);
148 CHKERR m_field.set_field_order(part_set, MBVERTEX, "F1", 1);
149 CHKERR m_field.set_field_order(part_set, MBTET, "F2", order);
150 CHKERR m_field.set_field_order(part_set, MBTRI, "F2", order);
151 CHKERR m_field.set_field_order(part_set, MBEDGE, "F2", order);
152 CHKERR m_field.set_field_order(part_set, MBVERTEX, "F2", 1);
153 CHKERR m_field.build_fields();
154
155 // Elements
156 CHKERR m_field.add_finite_element("E1");
157 CHKERR m_field.add_finite_element("E2");
162 CHKERR m_field.add_ents_to_finite_element_by_type(part_set, MBTET, "E1");
163 CHKERR m_field.add_ents_to_finite_element_by_type(part_set, MBTET, "E2");
165 CHKERR m_field.build_adjacencies(bit_level0);
166
167 // Problems
168 CHKERR m_field.add_problem("P1");
169 CHKERR m_field.add_problem("P2");
170 // set refinement level for problem
171 CHKERR m_field.modify_problem_ref_level_add_bit("P1", bit_level0);
172 CHKERR m_field.modify_problem_ref_level_add_bit("P2", bit_level0);
173 CHKERR m_field.modify_problem_add_finite_element("P1", "E1");
174 CHKERR m_field.modify_problem_add_finite_element("P2", "E2");
175
176 // Build problems
177 CHKERR prb_mng_ptr->buildProblemOnDistributedMesh("P1", true, 1);
178 CHKERR prb_mng_ptr->buildProblemOnDistributedMesh("P2", true, 1);
179 CHKERR prb_mng_ptr->partitionFiniteElements("P1", true, 0,
180 m_field.get_comm_size(), 1);
181 CHKERR prb_mng_ptr->partitionGhostDofs("P1", 1);
182 CHKERR prb_mng_ptr->partitionFiniteElements("P2", true, 0,
183 m_field.get_comm_size(), 1);
184 CHKERR prb_mng_ptr->partitionGhostDofs("P2", 1);
185
186 if (0) {
189 ->createMPIAIJWithArrays<PetscGlobalIdx_mi_tag>("P1", m);
190 MatView(m, PETSC_VIEWER_DRAW_WORLD);
191 std::string wait;
192 std::cin >> wait;
193 }
194
196 ->checkMPIAIJWithArraysMatrixFillIn<PetscGlobalIdx_mi_tag>("P1", -1, -1,
197 1);
199 ->checkMPIAIJWithArraysMatrixFillIn<PetscGlobalIdx_mi_tag>("P2", -1, -1,
200 1);
201
202 // register new dm type, i.e. mofem
203 DMType dm_name = "MOFEM";
204 CHKERR DMRegister_MoFEM(dm_name);
205 // create dm instance
206 auto dm = createDM(m_field.get_comm(), dm_name);
207
208 CHKERR DMMoFEMCreateMoFEM(dm, &m_field, "COMP", bit_level0);
209 CHKERR DMSetFromOptions(dm);
210 CHKERR DMMoFEMSetIsPartitioned(dm, PETSC_TRUE);
211 CHKERR DMMoFEMAddElement(dm, "E1");
212 CHKERR DMMoFEMAddElement(dm, "E2");
215 CHKERR DMSetUp(dm);
216
217 if (0) {
220 ->createMPIAIJWithArrays<PetscGlobalIdx_mi_tag>("COMP", m);
221 MatView(m, PETSC_VIEWER_DRAW_WORLD);
222 std::string wait;
223 std::cin >> wait;
224 }
225
227 ->checkMPIAIJWithArraysMatrixFillIn<PetscGlobalIdx_mi_tag>("COMP", -1,
228 -1, 1);
229
230 }
232
233 // Finish work cleaning memory, getting statistics, etc.
235
236 return 0;
237}
static char help[]
#define CATCH_ERRORS
Catch errors.
@ AINSWORTH_LEGENDRE_BASE
Ainsworth Cole (Legendre) approx. base .
Definition definitions.h:60
@ H1
continuous field
Definition definitions.h:85
#define MYPCOMM_INDEX
default communicator number PCOMM
@ MOFEM_DATA_INCONSISTENCY
Definition definitions.h:31
#define CHKERR
Inline error check.
constexpr int order
FTensor::Index< 'm', SPACE_DIM > m
PetscErrorCode DMMoFEMSetIsPartitioned(DM dm, PetscBool is_partitioned)
Definition DMMoFEM.cpp:1109
PetscErrorCode DMMoFEMAddElement(DM dm, std::string fe_name)
add element to dm
Definition DMMoFEM.cpp:483
PetscErrorCode DMMoFEMCreateMoFEM(DM dm, MoFEM::Interface *m_field_ptr, const char problem_name[], const MoFEM::BitRefLevel bit_level, const MoFEM::BitRefLevel bit_mask=MoFEM::BitRefLevel().set())
Must be called by user to set MoFEM data structures.
Definition DMMoFEM.cpp:118
PetscErrorCode DMMoFEMAddRowCompositeProblem(DM dm, const char prb_name[])
Add problem to composite DM on row.
Definition DMMoFEM.cpp:353
PetscErrorCode DMRegister_MoFEM(const char sname[])
Register MoFEM problem.
Definition DMMoFEM.cpp:47
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 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.
MoFEMErrorCode buildProblemOnDistributedMesh(const std::string name, const bool square_matrix, int verb=VERBOSE)
build problem data structures, assuming that mesh is distributed (collective)
MoFEMErrorCode partitionGhostDofs(const std::string name, int verb=VERBOSE)
determine ghost nodes
MoFEMErrorCode partitionMesh(const Range &ents, const int dim, const int adj_dim, const int n_parts, Tag *th_vertex_weights=nullptr, Tag *th_edge_weights=nullptr, Tag *th_part_weights=nullptr, int verb=VERBOSE, const bool debug=false)
Set partition tag to each finite element in the problem.
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]
const Tensor2_symmetric_Expr< const ddTensor0< T, Dim, i, j >, typename promote< T, double >::V, Dim, i, j > dd(const Tensor0< T * > &a, const Index< i, Dim > index1, const Index< j, Dim > index2, const Tensor1< int, Dim > &d_ijk, const Tensor1< double, Dim > &d_xyz)
Definition ddTensor0.hpp:33
std::bitset< BITREFLEVEL_SIZE > BitRefLevel
Bit structure attached to each entity identifying to what mesh entity is attached.
Definition Types.hpp:40
PetscErrorCode PetscOptionsGetString(PetscOptions *, const char pre[], const char name[], char str[], size_t size, PetscBool *set)
auto createDM(MPI_Comm comm, const std::string dm_type_name)
Creates smart DM object.
Managing BitRefLevels.
Managing BitRefLevels.
virtual int get_comm_size() const =0
virtual moab::Interface & get_moab()=0
virtual MoFEMErrorCode build_adjacencies(const Range &ents, int verb=DEFAULT_VERBOSITY)=0
build adjacencies
virtual MPI_Comm & get_comm() const =0
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.
virtual int get_comm_rank() const =0
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:112
Deprecated interface functions.
Matrix manager is used to build and partition problems.
Problem manager is used to build and partition problems.
intrusive_ptr for managing petsc objects
MoFEMErrorCode getInterface(IFACE *&iface) const
Get interface refernce to pointer of interface.

Variable Documentation

◆ help

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

Definition at line 12 of file build_composite_problem.cpp.