simple_interface.cpp File Reference

#include <MoFEM.hpp>

Go to the source code of this file.

Classes
struct	OpVolume
struct	OpFace
struct	VolRule
	Set integration rule to volume elements. More...
struct	FaceRule
	Set integration rule to boundary elements. More...

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

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

Function Documentation

main()

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

Examples

simple_interface.cpp.

Definition at line 95 of file simple_interface.cpp.

                                 {
 
  // initialize petsc
  MoFEM::Core::Initialize(&argc, &argv, (char *)0, help);
 
  auto core_log = logging::core::get();
  core_log->add_sink(
      LogManager::createSink(LogManager::getStrmWorld(), "TEST"));
  LogManager::setLog("TEST");
  MOFEM_LOG_TAG("TEST", "atom_test");
  core_log->add_sink(
      LogManager::createSink(LogManager::getStrmSync(), "TESTSYNC"));
  LogManager::setLog("TESTSYNC");
  MOFEM_LOG_TAG("TESTSYNC", "atom_test");
 
  try {
 
    // Create MoAB database
    moab::Core moab_core;
    moab::Interface &moab = moab_core;
 
    // Create MoFEM database and link it to MoAB
    MoFEM::Core mofem_core(moab);
    MoFEM::Interface &m_field = mofem_core;
 
    // Register DM Manager
    DMType dm_name = "DMMOFEM";
    CHKERR DMRegister_MoFEM(dm_name);
 
    // Simple interface
    Simple *simple_interface;
    CHKERR m_field.getInterface(simple_interface);
    {
      // get options from command line
      CHKERR simple_interface->getOptions();
      // load mesh file
      CHKERR simple_interface->loadFile();
      // add fields
      CHKERR simple_interface->addDomainField("MESH_NODE_POSITIONS", H1,
                                              AINSWORTH_LEGENDRE_BASE, 3);
      CHKERR simple_interface->addBoundaryField("MESH_NODE_POSITIONS", H1,
                                                AINSWORTH_LEGENDRE_BASE, 3);
      CHKERR simple_interface->addSkeletonField("MESH_NODE_POSITIONS", H1,
                                                AINSWORTH_LEGENDRE_BASE, 3);
 
      CHKERR simple_interface->addMeshsetField("GLOBAL", NOFIELD, NOBASE, 3);
      CHKERR simple_interface->addMeshsetField("MESH_NODE_POSITIONS", H1,
                                               AINSWORTH_LEGENDRE_BASE, 3);
 
      // Note:: two "meshset" elements are created, one with GOLOBAL field and
      // vertices with "MESH_NODE_POSITIONS" field, and other with "GLOABL"
      // field and edges with "MESH_NODE_POSITIONS" field.
      // That us only to test API, and data structures, potential application is
      // here not known.
      Range verts;
      CHKERR m_field.get_moab().get_entities_by_type(0, MBVERTEX, verts);
      simple_interface->getMeshsetFiniteElementEntities().push_back(verts);
      Range edge;
      CHKERR m_field.get_moab().get_entities_by_type(0, MBEDGE, edge);
      simple_interface->getMeshsetFiniteElementEntities().push_back(edge);
 
      // set fields order
      CHKERR simple_interface->setFieldOrder("MESH_NODE_POSITIONS", 1);
      // setup problem
      CHKERR simple_interface->setUp();
      // Project mesh coordinate on mesh
      Projection10NodeCoordsOnField ent_method(m_field, "MESH_NODE_POSITIONS");
      CHKERR m_field.loop_dofs("MESH_NODE_POSITIONS", ent_method);
      // get dm
      auto dm = simple_interface->getDM();
      // create elements
      auto domain_fe =
          boost::make_shared<VolumeElementForcesAndSourcesCore>(m_field);
      auto boundary_fe =
          boost::make_shared<FaceElementForcesAndSourcesCore>(m_field);
 
      // set integration rule
      domain_fe->getRuleHook = VolRule();
      boundary_fe->getRuleHook = FaceRule();
      // create distributed vector to accumulate values from processors.
      int ghosts[] = {0};
 
      auto vol = createGhostVector(
          PETSC_COMM_WORLD, m_field.get_comm_rank() == 0 ? 1 : 0, 1,
          m_field.get_comm_rank() == 0 ? 0 : 1, ghosts);
      auto surf_vol = vectorDuplicate(vol);
 
      // set operator to the volume element
      auto material_grad_mat = boost::make_shared<MatrixDouble>();
      auto material_det_vec = boost::make_shared<VectorDouble>();
 
      domain_fe->getOpPtrVector().push_back(
          new OpCalculateVectorFieldGradient<3, 3>("MESH_NODE_POSITIONS",
                                                   material_grad_mat));
      domain_fe->getOpPtrVector().push_back(
          new OpInvertMatrix<3>(material_grad_mat, material_det_vec, nullptr));
      domain_fe->getOpPtrVector().push_back(
          new OpSetHOWeights(material_det_vec));
      domain_fe->getOpPtrVector().push_back(
          new OpCalculateHOCoords("MESH_NODE_POSITIONS"));
      domain_fe->getOpPtrVector().push_back(
          new OpVolume("MESH_NODE_POSITIONS", vol));
      // set operator to the face element
      boundary_fe->getOpPtrVector().push_back(
          new OpCalculateHOCoords("MESH_NODE_POSITIONS"));
      boundary_fe->getOpPtrVector().push_back(
          new OpFace("MESH_NODE_POSITIONS", surf_vol));
      // make integration in volume (here real calculations starts)
      CHKERR DMoFEMLoopFiniteElements(dm, simple_interface->getDomainFEName(),
                                      domain_fe);
      // make integration on boundary
      CHKERR DMoFEMLoopFiniteElements(dm, simple_interface->getBoundaryFEName(),
                                      boundary_fe);
 
      auto skeleton_fe = boost::make_shared<FEMethod>();
      auto A = createDMMatrix(dm);
      auto B = createDMMatrix(dm);
      auto f = createDMVector(dm);
      auto x = createDMVector(dm);
      auto x_t = createDMVector(dm);
      auto x_tt = createDMVector(dm);
      skeleton_fe->f = f;
      skeleton_fe->A = A;
      skeleton_fe->B = B;
      skeleton_fe->x = x;
      skeleton_fe->x_t = x_t;
      skeleton_fe->x_tt = x_tt;
 
      skeleton_fe->preProcessHook = [&]() {
        MoFEMFunctionBegin;
        if (f != skeleton_fe->ts_F)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        if (A != skeleton_fe->ts_A)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        if (B != skeleton_fe->ts_B)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        if (x != skeleton_fe->ts_u)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        if (x_t != skeleton_fe->ts_u_t)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        if (x_tt != skeleton_fe->ts_u_tt)
          SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Wrong ptr");
        MoFEMFunctionReturn(0);
      };
 
      skeleton_fe->postProcessHook = []() { return 0; };
      skeleton_fe->operatorHook = []() { return 0; };
 
      CHKERR DMoFEMLoopFiniteElements(dm, simple_interface->getSkeletonFEName(),
                                      skeleton_fe);
 
      // assemble volumes from processors and accumulate on processor of rank 0
      CHKERR VecAssemblyBegin(vol);
      CHKERR VecAssemblyEnd(vol);
      CHKERR VecGhostUpdateBegin(vol, ADD_VALUES, SCATTER_REVERSE);
      CHKERR VecGhostUpdateEnd(vol, ADD_VALUES, SCATTER_REVERSE);
      CHKERR VecAssemblyBegin(surf_vol);
      CHKERR VecAssemblyEnd(surf_vol);
      CHKERR VecGhostUpdateBegin(surf_vol, ADD_VALUES, SCATTER_REVERSE);
      CHKERR VecGhostUpdateEnd(surf_vol, ADD_VALUES, SCATTER_REVERSE);
      if (m_field.get_comm_rank() == 0) {
        double *a_vol;
        CHKERR VecGetArray(vol, &a_vol);
        double *a_surf_vol;
        CHKERR VecGetArray(surf_vol, &a_surf_vol);
        MOFEM_LOG("TEST", Sev::inform) << "Volume = " << a_vol[0];
        MOFEM_LOG("TEST", Sev::inform) << "Surf Volume = " << a_surf_vol[0];
        if (fabs(a_vol[0] - a_surf_vol[0]) > 1e-12) {
          SETERRQ(PETSC_COMM_SELF, MOFEM_ATOM_TEST_INVALID, "Should be zero");
        }
        CHKERR VecRestoreArray(vol, &a_vol);
        CHKERR VecRestoreArray(vol, &a_surf_vol);
      }
 
      struct MeshsetFE : public ForcesAndSourcesCore {
        using ForcesAndSourcesCore::ForcesAndSourcesCore;
        MoFEMErrorCode operator()() {
          MoFEMFunctionBegin;
          const auto type = numeredEntFiniteElementPtr->getEntType();
          if (type != MBENTITYSET) {
            SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
                    "Expected entity set as a finite element");
          }
          CHKERR loopOverOperators();
          MoFEMFunctionReturn(0);
        }
        MoFEMErrorCode preProcess() { return 0; }
        MoFEMErrorCode postProcess() { return 0; }
      };
 
      auto fe_ptr = boost::make_shared<MeshsetFE>(m_field);
 
      struct OpMeshset : ForcesAndSourcesCore::UserDataOperator {
        using OP = ForcesAndSourcesCore::UserDataOperator;
        using OP::OP;
 
        MoFEMErrorCode doWork(int side, EntityType type,
                              EntitiesFieldData::EntData &data) {
          MoFEMFunctionBegin;
          MOFEM_LOG("TESTSYNC", Sev::inform) << data.getIndices();
          MoFEMFunctionReturn(0);
        }
 
        MoFEMErrorCode doWork(int row_side, int col_side, EntityType row_type,
                              EntityType col_type,
                              EntitiesFieldData::EntData &row_data,
                              EntitiesFieldData::EntData &col_data) {
          MoFEMFunctionBegin;
          MOFEM_LOG("TESTSYNC", Sev::inform)
              << row_data.getIndices() << " " << col_data.getIndices() << endl;
          MoFEMFunctionReturn(0);
        }
      };
 
      fe_ptr->getOpPtrVector().push_back(
          new OpMeshset("GLOBAL", OpMeshset::OPROW));
      fe_ptr->getOpPtrVector().push_back(
          new OpMeshset("GLOBAL", "GLOBAL", OpMeshset::OPROWCOL));
 
      CHKERR DMoFEMLoopFiniteElementsUpAndLowRank(
          dm, simple_interface->getMeshsetFEName(), fe_ptr, 0,
          m_field.get_comm_size());
      MOFEM_LOG_SEVERITY_SYNC(m_field.get_comm(), Sev::inform);
 
      MOFEM_LOG("TESTSYNC", Sev::inform) << "PipelineManager";
      auto pip_mng = m_field.getInterface<PipelineManager>();
      pip_mng->getOpMeshsetRhsPipeline().push_back(
          new OpMeshset("GLOBAL", OpMeshset::OPROW));
      pip_mng->getOpMeshsetLhsPipeline().push_back(
          new OpMeshset("GLOBAL", "GLOBAL", OpMeshset::OPROWCOL));
      pip_mng->loopFiniteElements();
      MOFEM_LOG_SEVERITY_SYNC(m_field.get_comm(), Sev::inform);
    }
  }
  CATCH_ERRORS;
 
  // finish work cleaning memory, getting statistics, etc.
  MoFEM::Core::Finalize();
 
  return 0;
}

Variable Documentation

help

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

static

Examples

simple_interface.cpp.

Definition at line 16 of file simple_interface.cpp.