#include <MoFEM.hpp>

Classes
struct	CoordsAndHandle

struct	PrismOp

struct	PrismFE

struct	Op< OP >

struct	TriFE

struct	QuadFE

struct	EdgeFE

Typedefs
using	EdgeEle = EdgeElementForcesAndSourcesCore

typedef multi_index_container< CoordsAndHandle, indexed_by< hashed_unique< composite_key< CoordsAndHandle, member< CoordsAndHandle, int, &CoordsAndHandle::x >, member< CoordsAndHandle, int, &CoordsAndHandle::y >, member< CoordsAndHandle, int, &CoordsAndHandle::z > > > > >	MapCoords

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

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

static constexpr int	precision_exponent = 5

static constexpr int	number_of_prisms_layers = 18

static constexpr double	delta

static constexpr std::array< double, 3 >	d3 = {0, 0, 0}

static constexpr std::array< double, 3 >	d4 = {0, 0, delta}

Typedef Documentation

◆ EdgeEle

using EdgeEle = EdgeElementForcesAndSourcesCore

Definition at line 23 of file prism_elements_from_surface.cpp.

◆ MapCoords

typedef multi_index_container< CoordsAndHandle, indexed_by< hashed_unique<composite_key< CoordsAndHandle, member<CoordsAndHandle, int, &CoordsAndHandle::x>, member<CoordsAndHandle, int, &CoordsAndHandle::y>, member<CoordsAndHandle, int, &CoordsAndHandle::z> > > > > MapCoords

Examples: prism_elements_from_surface.cpp.

Definition at line 48 of file prism_elements_from_surface.cpp.

Function Documentation

◆ main()

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

Examples: prism_elements_from_surface.cpp.

Definition at line 132 of file prism_elements_from_surface.cpp.

                                  {
  
   MoFEM::Core::Initialize(&argc, &argv, (char *)0, help);
  
   try {
  
     moab::Core mb_instance;
     moab::Interface &moab = mb_instance;
     int rank;
     MPI_Comm_rank(PETSC_COMM_WORLD, &rank);
  
     // Read parameters from line command
     PetscBool flg = PETSC_TRUE;
     char mesh_file_name[255];
     CHKERR PetscOptionsGetString(PETSC_NULL, "", "-my_file", mesh_file_name,
                                  255, &flg);
     if (flg != PETSC_TRUE)
       SETERRQ(PETSC_COMM_SELF, MOFEM_INVALID_DATA,
               "error -my_file (MESH FILE NEEDED)");
  
     // Read mesh to MOAB
     const char *option;
     option = ""; //"PARALLEL=BCAST;";//;DEBUG_IO";
     CHKERR moab.load_file(mesh_file_name, 0, option);
  
     Range tris;
     CHKERR moab.get_entities_by_type(0, MBTRI, tris, false);
     Range tris_verts;
     CHKERR moab.get_connectivity(tris, tris_verts);
     MatrixDouble tri_coords(tris_verts.size(), 3);
     CHKERR moab.get_coords(tris_verts, &tri_coords(0, 0));
  
     MoFEM::Core core(moab);
     MoFEM::Interface &m_field = core;
  
     std::array<Range, 3> edge_block;
     for (auto b : {1, 2, 3})
       CHKERR m_field.getInterface<MeshsetsManager>()->getEntitiesByDimension(
           b, BLOCKSET, 1, edge_block[b - 1]);
  
     PrismsFromSurfaceInterface *prisms_from_surface_interface;
     CHKERR m_field.getInterface(prisms_from_surface_interface);
  
     Range prisms;
     CHKERR prisms_from_surface_interface->createPrisms(
         tris, PrismsFromSurfaceInterface::NO_SWAP, prisms);
     prisms_from_surface_interface->setThickness(prisms, d3.data(), d4.data());
     Range add_prims_layer;
     Range extrude_prisms = prisms;
  
     for (int ll = 1; ll != number_of_prisms_layers; ++ll) {
       prisms_from_surface_interface->createdVertices.clear();
       CHKERR prisms_from_surface_interface->createPrismsFromPrisms(
           extrude_prisms, false, add_prims_layer);
       prisms_from_surface_interface->setThickness(add_prims_layer, d3.data(),
                                                   d4.data());
       extrude_prisms = add_prims_layer;
       prisms.merge(add_prims_layer);
       add_prims_layer.clear();
     }
  
     EntityHandle meshset;
     CHKERR moab.create_meshset(MESHSET_SET | MESHSET_TRACK_OWNER, meshset);
     CHKERR moab.add_entities(meshset, prisms);
  
     MapCoords map_coords;
     Range verts;
     CHKERR moab.get_connectivity(prisms, verts);
     MatrixDouble coords(verts.size(), 3);
     CHKERR moab.get_coords(verts, &coords(0, 0));
  
     for (size_t v = 0; v != verts.size(); ++v)
       map_coords.insert(CoordsAndHandle(&coords(v, 0), verts[v]));
  
     EntityHandle one_prism_meshset;
     CHKERR moab.create_meshset(MESHSET_SET | MESHSET_TRACK_OWNER,
                                one_prism_meshset);
  
     std::array<double, 18> one_prism_coords = {0, 0, 0, 1, 0, 0, 0, 1, 0,
                                                0, 0, 1, 1, 0, 1, 0, 1, 1};
     std::array<EntityHandle, 6> one_prism_nodes;
     for (int n = 0; n != 6; ++n)
       CHKERR moab.create_vertex(&one_prism_coords[3 * n], one_prism_nodes[n]);
     EntityHandle one_prism;
     CHKERR m_field.get_moab().create_element(MBPRISM, one_prism_nodes.data(), 6,
                                              one_prism);
     Range one_prism_range;
     one_prism_range.insert(one_prism);
     CHKERR moab.add_entities(one_prism_meshset, one_prism_range);
     Range one_prism_verts;
     CHKERR moab.get_connectivity(one_prism_range, one_prism_verts);
     CHKERR moab.add_entities(one_prism_meshset, one_prism_verts);
     Range one_prism_adj_ents;
     for (int d = 1; d != 3; ++d)
       CHKERR moab.get_adjacencies(one_prism_range, d, true, one_prism_adj_ents,
                                   moab::Interface::UNION);
     CHKERR moab.add_entities(one_prism_meshset, one_prism_adj_ents);
  
     BitRefLevel bit_level0;
     bit_level0.set(0);
     CHKERR m_field.getInterface<BitRefManager>()->setEntitiesBitRefLevel(
         one_prism_range, bit_level0);
     CHKERR prisms_from_surface_interface->seedPrismsEntities(one_prism_range,
                                                              bit_level0);
  
     // Fields
     CHKERR m_field.add_field("FIELD1", H1, AINSWORTH_LEGENDRE_BASE, 1);
     CHKERR m_field.add_ents_to_field_by_type(one_prism_meshset, MBPRISM,
                                              "FIELD1", VERY_NOISY);
  
     CHKERR m_field.set_field_order(one_prism_meshset, MBVERTEX, "FIELD1", 1);
     CHKERR m_field.set_field_order(one_prism_meshset, MBEDGE, "FIELD1", 5,
                                    VERY_NOISY);
     CHKERR m_field.set_field_order(one_prism_meshset, MBTRI, "FIELD1", 5);
     CHKERR m_field.set_field_order(one_prism_meshset, MBQUAD, "FIELD1", 5,
                                    VERY_NOISY);
     CHKERR m_field.set_field_order(one_prism_meshset, MBPRISM, "FIELD1", 7,
                                    VERY_NOISY);
     CHKERR m_field.build_fields(VERY_NOISY);
  
     // FE
     CHKERR m_field.add_finite_element("PRISM");
     CHKERR m_field.add_finite_element("EDGE");
     CHKERR m_field.add_finite_element("TRI");
     CHKERR m_field.add_finite_element("QUAD");
  
     // Define rows/cols and element data
     CHKERR m_field.modify_finite_element_add_field_row("PRISM", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_col("PRISM", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_data("PRISM", "FIELD1");
     // Define rows/cols and element data
     CHKERR m_field.modify_finite_element_add_field_row("EDGE", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_col("EDGE", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_data("EDGE", "FIELD1");
     // Define rows/cols and element data
     CHKERR m_field.modify_finite_element_add_field_row("TRI", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_col("TRI", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_data("TRI", "FIELD1");
     // Define rows/cols and element data
     CHKERR m_field.modify_finite_element_add_field_row("QUAD", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_col("QUAD", "FIELD1");
     CHKERR m_field.modify_finite_element_add_field_data("QUAD", "FIELD1");
  
     CHKERR m_field.add_ents_to_finite_element_by_type(one_prism_meshset, MBEDGE,
                                                       "EDGE");
     CHKERR m_field.add_ents_to_finite_element_by_type(one_prism_meshset, MBTRI,
                                                       "TRI");
     CHKERR m_field.add_ents_to_finite_element_by_type(one_prism_meshset, MBQUAD,
                                                       "QUAD");
     CHKERR m_field.add_ents_to_finite_element_by_type(one_prism_meshset,
                                                       MBPRISM, "PRISM");
  
     // build finite elemnts
     CHKERR m_field.build_finite_elements();
     // //build adjacencies
     CHKERR m_field.build_adjacencies(bit_level0);
  
     // Problem
     CHKERR m_field.add_problem("TEST_PROBLEM");
  
     // set finite elements for problem
     CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "PRISM");
     CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "EDGE");
     CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "TRI");
     CHKERR m_field.modify_problem_add_finite_element("TEST_PROBLEM", "QUAD");
     // set refinement level for problem
     CHKERR m_field.modify_problem_ref_level_add_bit("TEST_PROBLEM", bit_level0);
  
     // build problem
     ProblemsManager *prb_mng_ptr;
     CHKERR m_field.getInterface(prb_mng_ptr);
     CHKERR prb_mng_ptr->buildProblem("TEST_PROBLEM", true);
     // partition
     CHKERR prb_mng_ptr->partitionSimpleProblem("TEST_PROBLEM");
     CHKERR prb_mng_ptr->partitionFiniteElements("TEST_PROBLEM");
     // what are ghost nodes, see Petsc Manual
     CHKERR prb_mng_ptr->partitionGhostDofs("TEST_PROBLEM");
  
     PrismFE fe_prism(m_field, tri_coords);
     fe_prism.getOpPtrVector().push_back(new PrismOp(moab, map_coords));
     CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "PRISM", fe_prism);
  
     EdgeFE fe_edge(m_field, edge_block, one_prism);
     fe_edge.getOpPtrVector().push_back(
         new Op<EdgeElementForcesAndSourcesCore::UserDataOperator>(
             moab, map_coords, one_prism));
     CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "EDGE", fe_edge);
  
     TriFE fe_tri(m_field, tri_coords, one_prism);
     fe_tri.getOpPtrVector().push_back(
         new Op<FaceElementForcesAndSourcesCore::UserDataOperator>(
             moab, map_coords, one_prism));
     CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "TRI", fe_tri);
  
     QuadFE fe_quad(m_field, edge_block, one_prism);
     fe_quad.getOpPtrVector().push_back(
         new Op<FaceElementForcesAndSourcesCore::UserDataOperator>(
             moab, map_coords, one_prism));
     CHKERR m_field.loop_finite_elements("TEST_PROBLEM", "QUAD", fe_quad);
  
     CHKERR moab.write_file("prism_mesh.vtk", "VTK", "", &meshset, 1);
     CHKERR moab.write_file("one_prism_mesh.vtk", "VTK", "", &one_prism_meshset,
                            1);
   }
   CATCH_ERRORS;
  
   MoFEM::Core::Finalize();
   return 0;
 }