v0.14.0
Loading...
Searching...
No Matches
serial_matrix.cpp
Go to the documentation of this file.
1/**
2 * \file serial_matrix.cpp
3 *
4 * \brief testing create serial/sequential matrix
5 */
6
7
8
9#include <MoFEM.hpp>
10
11namespace bio = boost::iostreams;
12using bio::stream;
13using bio::tee_device;
14
15using namespace MoFEM;
16
17static char help[] = "...\n\n";
18
19int main(int argc, char *argv[]) {
20
21 MoFEM::Core::Initialize(&argc, &argv, PETSC_NULL, help);
22
23 try {
24
25 moab::Core mb_instance;
26 moab::Interface &moab = mb_instance;
27
28 // Reade parameters from line command
29 PetscBool flg = PETSC_TRUE;
30 char mesh_file_name[255];
31#if PETSC_VERSION_GE(3, 6, 4)
32 CHKERR PetscOptionsGetString(PETSC_NULL, "", "-my_file", mesh_file_name,
33 255, &flg);
34#else
35 CHKERR PetscOptionsGetString(PETSC_NULL, PETSC_NULL, "-my_file",
36 mesh_file_name, 255, &flg);
37#endif
38 if (flg != PETSC_TRUE) {
39 SETERRQ(PETSC_COMM_SELF, 1, "*** ERROR -my_file (MESH FILE NEEDED)");
40 }
41 PetscInt order;
42#if PETSC_VERSION_GE(3, 6, 4)
43 CHKERR PetscOptionsGetInt(PETSC_NULL, "", "-my_order", &order, &flg);
44#else
45 CHKERR PetscOptionsGetInt(PETSC_NULL, PETSC_NULL, "-my_order", &order,
46 &flg);
47#endif
48 if (flg != PETSC_TRUE) {
49 order = 1;
50 }
51
52 // Read mesh to MOAB
53 const char *option;
54 option = ""; //"PARALLEL=BCAST;";//;DEBUG_IO";
55 CHKERR moab.load_file(mesh_file_name, 0, option);
56 ParallelComm *pcomm = ParallelComm::get_pcomm(&moab, MYPCOMM_INDEX);
57 auto moab_comm_wrap =
58 boost::make_shared<WrapMPIComm>(PETSC_COMM_WORLD, false);
59 if (pcomm == NULL)
60 pcomm = new ParallelComm(&moab, moab_comm_wrap->get_comm());
61
62 int do_for_rank = 0;
63 if (pcomm->rank() ==
64 (unsigned int)do_for_rank) { // should work only with rank 0
65
66 // Create MoFEM (Joseph) database
67 // second argument set communicator for sequential problem
68 // last argument make mofem QUIET
69 MoFEM::Core core(moab, PETSC_COMM_SELF, -1);
70 MoFEM::Interface &m_field = core;
71
72 // stl::bitset see for more details
73 BitRefLevel bit_level0;
74 bit_level0.set(0);
75 EntityHandle meshset_level0;
76 CHKERR moab.create_meshset(MESHSET_SET, meshset_level0);
78 CHKERR m_field.getInterface<BitRefManager>()->setBitRefLevelByDim(
79 0, 3, bit_level0);
80 CHKERR m_field.getInterface<BitRefManager>()->getEntitiesByRefLevel(
81 bit_level0, BitRefLevel().set(), meshset_level0);
82
83 /***/
84 // Define problem
85
86 // Fields
87 CHKERR m_field.add_field("FIELD_A", H1, AINSWORTH_LEGENDRE_BASE, 3,
88 MB_TAG_DENSE);
89 CHKERR m_field.add_field("FIELD_B", L2, AINSWORTH_LEGENDRE_BASE, 1,
90 MB_TAG_DENSE);
91
92 CHKERR m_field.add_ents_to_field_by_type(0, MBTET, "FIELD_A");
93 CHKERR m_field.add_ents_to_field_by_type(0, MBTET, "FIELD_B");
94
95 CHKERR m_field.set_field_order(0, MBTET, "FIELD_A", order);
96 CHKERR m_field.set_field_order(0, MBTRI, "FIELD_A", order);
97 CHKERR m_field.set_field_order(0, MBEDGE, "FIELD_A", order);
98 CHKERR m_field.set_field_order(0, MBVERTEX, "FIELD_A", 1);
99
100 CHKERR m_field.set_field_order(0, MBTET, "FIELD_B", order - 1);
101 CHKERR m_field.set_field_order(0, MBTRI, "FIELD_B", order - 1);
102 CHKERR m_field.set_field_order(0, MBEDGE, "FIELD_B", order - 1);
103
104 // build field
105 CHKERR m_field.build_fields();
106
107 // Element
108 CHKERR m_field.add_finite_element("FE1");
109 CHKERR m_field.add_finite_element("FE2");
110
111 // Define rows/cols and element data
112 CHKERR m_field.modify_finite_element_add_field_row("FE1", "FIELD_A");
113 CHKERR m_field.modify_finite_element_add_field_col("FE1", "FIELD_A");
114 CHKERR m_field.modify_finite_element_add_field_data("FE1", "FIELD_A");
115 CHKERR m_field.modify_finite_element_add_field_col("FE1", "FIELD_B");
116 CHKERR m_field.modify_finite_element_add_field_data("FE1", "FIELD_B");
117
118 // Define rows/cols and element data
119 CHKERR m_field.modify_finite_element_add_field_row("FE2", "FIELD_B");
120 CHKERR m_field.modify_finite_element_add_field_col("FE2", "FIELD_A");
121 CHKERR m_field.modify_finite_element_add_field_data("FE2", "FIELD_B");
122
123 // add entities to finite element
124 CHKERR m_field.add_ents_to_finite_element_by_type(0, MBTET, "FE1");
125 CHKERR m_field.add_ents_to_finite_element_by_type(0, MBTET, "FE2");
126
127 // build finite elemnts
129 // build adjacencies
130 CHKERR m_field.build_adjacencies(bit_level0);
131
132 // Problem
133 CHKERR m_field.add_problem("TEST_PROBLEM");
134
135 // set refinement level for problem
136 CHKERR m_field.modify_problem_ref_level_add_bit("TEST_PROBLEM",
137 bit_level0);
138 CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "FE1");
139 CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "FE2");
140
141 ProblemsManager *prb_mng_ptr;
142 CHKERR m_field.getInterface(prb_mng_ptr);
143 CHKERR prb_mng_ptr->buildProblem("TEST_PROBLEM", true);
144 CHKERR prb_mng_ptr->partitionSimpleProblem("TEST_PROBLEM");
145
146 // partition finite elements
147 CHKERR prb_mng_ptr->partitionFiniteElements("TEST_PROBLEM");
148 // what are ghost nodes, see Petsc Manual
149 CHKERR prb_mng_ptr->partitionGhostDofs("TEST_PROBLEM");
150
151 Vec F;
152 CHKERR m_field.getInterface<VecManager>()->vecCreateSeq("TEST_PROBLEM",
153 ROW, &F);
154 Mat A;
156 ->createMPIAIJWithArrays<PetscGlobalIdx_mi_tag>("TEST_PROBLEM", &A);
157
158 PetscViewer viewer;
159 CHKERR PetscViewerASCIIOpen(PETSC_COMM_SELF, "serial_matrix.txt",
160 &viewer);
161 CHKERR PetscViewerPushFormat(viewer, PETSC_VIEWER_ASCII_INFO);
162 CHKERR MatView(A, viewer);
163
164 CHKERR MatDestroy(&A);
165 CHKERR VecDestroy(&F);
166 }
167 }
169
171 return 0;
172}
int main()
Definition: adol-c_atom.cpp:46
@ ROW
Definition: definitions.h:123
#define CATCH_ERRORS
Catch errors.
Definition: definitions.h:372
@ AINSWORTH_LEGENDRE_BASE
Ainsworth Cole (Legendre) approx. base .
Definition: definitions.h:60
@ L2
field with C-1 continuity
Definition: definitions.h:88
@ H1
continuous field
Definition: definitions.h:85
#define MYPCOMM_INDEX
default communicator number PCOMM
Definition: definitions.h:215
#define CHKERRG(n)
Check error code of MoFEM/MOAB/PETSc function.
Definition: definitions.h:483
#define CHKERR
Inline error check.
Definition: definitions.h:535
constexpr int order
@ F
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_data(const std::string &fe_name, const std::string name_filed)=0
set finite element field data
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 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]
static MoFEMErrorCodeGeneric< moab::ErrorCode > rval
Definition: Exceptions.hpp:74
std::bitset< BITREFLEVEL_SIZE > BitRefLevel
Bit structure attached to each entity identifying to what mesh entity is attached.
Definition: Types.hpp:40
implementation of Data Operators for Forces and Sources
Definition: Common.hpp:10
PetscErrorCode PetscOptionsGetInt(PetscOptions *, const char pre[], const char name[], PetscInt *ivalue, PetscBool *set)
PetscErrorCode PetscOptionsGetString(PetscOptions *, const char pre[], const char name[], char str[], size_t size, PetscBool *set)
constexpr AssemblyType A
static char help[]
Managing BitRefLevels.
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:112
Deprecated interface functions.
Matrix manager is used to build and partition problems.
Problem manager is used to build and partition problems.
MoFEMErrorCode getInterface(IFACE *&iface) const
Get interface refernce to pointer of interface.
Vector manager is used to create vectors \mofem_vectors.
Definition: VecManager.hpp:23