Poisson2DLagrangeMultiplier Struct Reference

Collaboration diagram for Poisson2DLagrangeMultiplier:

Public Member Functions
	Poisson2DLagrangeMultiplier (MoFEM::Interface &m_field)
MoFEMErrorCode	runProgram ()

Private Member Functions
MoFEMErrorCode	readMesh ()
MoFEMErrorCode	setupProblem ()
MoFEMErrorCode	boundaryCondition ()
MoFEMErrorCode	assembleSystem ()
MoFEMErrorCode	setIntegrationRules ()
MoFEMErrorCode	solveSystem ()
MoFEMErrorCode	outputResults ()

Static Private Member Functions
static double	sourceTermFunction (const double x, const double y, const double z)
static double	boundaryFunction (const double x, const double y, const double z)

Private Attributes
MoFEM::Interface &	mField
Simple *	simpleInterface
std::string	domainField
std::string	boundaryField
int	oRder
boost::shared_ptr< std::vector< unsigned char > >	boundaryMarker
Range	boundaryEntitiesForFieldsplit

Detailed Description

Definition at line 15 of file poisson_2d_lagrange_multiplier.cpp.

Constructor & Destructor Documentation

Poisson2DLagrangeMultiplier()

Poisson2DLagrangeMultiplier::Poisson2DLagrangeMultiplier

(

MoFEM::Interface &

m_field

)

Definition at line 61 of file poisson_2d_lagrange_multiplier.cpp.

    : domainField("U"), boundaryField("L"), mField(m_field) {}

Member Function Documentation

assembleSystem()

MoFEMErrorCode Poisson2DLagrangeMultiplier::assembleSystem ( )

private

Definition at line 134 of file poisson_2d_lagrange_multiplier.cpp.

                                                           {
  MoFEMFunctionBegin;
 
  auto pipeline_mng = mField.getInterface<PipelineManager>();
  auto det_ptr = boost::make_shared<VectorDouble>();
  auto jac_ptr = boost::make_shared<MatrixDouble>();
  auto inv_jac_ptr = boost::make_shared<MatrixDouble>();
 
  { // Push operators to the Pipeline that is responsible for calculating LHS of
    // domain elements
    pipeline_mng->getOpDomainLhsPipeline().push_back(
        new OpCalculateHOJac<2>(jac_ptr));
    pipeline_mng->getOpDomainLhsPipeline().push_back(
        new OpInvertMatrix<2>(jac_ptr, det_ptr, inv_jac_ptr));
    pipeline_mng->getOpDomainLhsPipeline().push_back(
        new OpSetHOInvJacToScalarBases<2>(H1, inv_jac_ptr));
    pipeline_mng->getOpDomainLhsPipeline().push_back(
        new OpSetHOWeightsOnFace());
    pipeline_mng->getOpDomainLhsPipeline().push_back(
        new OpDomainLhsK(domainField, domainField, boundaryMarker));
  }
 
  { // Push operators to the Pipeline that is responsible for calculating RHS of
    // domain elements
    pipeline_mng->getOpDomainRhsPipeline().push_back(
        new OpDomainRhsF(domainField, sourceTermFunction, boundaryMarker));
  }
 
  { // Push operators to the Pipeline that is responsible for calculating LHS of
    // boundary elements (Lagrange multiplier)
    pipeline_mng->getOpBoundaryLhsPipeline().push_back(
        new OpBoundaryLhsC(domainField, boundaryField));
    pipeline_mng->getOpBoundaryLhsPipeline().push_back(
        new OpBoundaryLhsC(boundaryField, domainField));
  }
 
  { // Push operators to the Pipeline that is responsible for calculating RHS of
    // boundary elements (Lagrange multiplier)
    pipeline_mng->getOpBoundaryRhsPipeline().push_back(
        new OpBoundaryRhsG(boundaryField, boundaryFunction));
  }
 
  MoFEMFunctionReturn(0);
}

boundaryCondition()

MoFEMErrorCode Poisson2DLagrangeMultiplier::boundaryCondition ( )

private

Definition at line 93 of file poisson_2d_lagrange_multiplier.cpp.

                                                              {
  MoFEMFunctionBegin;
 
  // Get boundary edges marked in block named "BOUNDARY_CONDITION"
  auto get_ents_on_mesh = [&]() {
    Range boundary_entities;
    for (_IT_CUBITMESHSETS_BY_SET_TYPE_FOR_LOOP_(mField, BLOCKSET, it)) {
      std::string entity_name = it->getName();
      if (entity_name.compare(0, 18, "BOUNDARY_CONDITION") == 0) {
        CHKERR it->getMeshsetIdEntitiesByDimension(mField.get_moab(), 1,
                                                   boundary_entities, true);
      }
    }
    // Add vertices to boundary entities
    Range boundary_vertices;
    CHKERR mField.get_moab().get_connectivity(boundary_entities,
                                              boundary_vertices, true);
    boundary_entities.merge(boundary_vertices);
 
    // Store entities for fieldsplit (block) solver
    boundaryEntitiesForFieldsplit = boundary_entities;
 
    return boundary_entities;
  };
 
  auto mark_boundary_dofs = [&](Range &&skin_edges) {
    auto problem_manager = mField.getInterface<ProblemsManager>();
    auto marker_ptr = boost::make_shared<std::vector<unsigned char>>();
    problem_manager->markDofs(simpleInterface->getProblemName(), ROW,
                              skin_edges, *marker_ptr);
    return marker_ptr;
  };
 
  // Get global local vector of marked DOFs. Is global, since is set for all
  // DOFs on processor. Is local since only DOFs on processor are in the
  // vector. To access DOFs use local indices.
  boundaryMarker = mark_boundary_dofs(get_ents_on_mesh());
 
  MoFEMFunctionReturn(0);
}

boundaryFunction()

static double Poisson2DLagrangeMultiplier::boundaryFunction ( const double  x, const double  y, const double  z  )

inlinestaticprivate

Definition at line 39 of file poisson_2d_lagrange_multiplier.cpp.

                                                 {
    return sin(x * 10.) * cos(y * 10.);
    // return 0;
  }

outputResults()

MoFEMErrorCode Poisson2DLagrangeMultiplier::outputResults ( )

private

Definition at line 249 of file poisson_2d_lagrange_multiplier.cpp.

                                                          {
  MoFEMFunctionBegin;
 
  auto pipeline_mng = mField.getInterface<PipelineManager>();
  pipeline_mng->getDomainLhsFE().reset();
  pipeline_mng->getBoundaryLhsFE().reset();
 
  auto d_ptr = boost::make_shared<VectorDouble>();
  auto l_ptr = boost::make_shared<VectorDouble>();
 
  using OpPPMap = OpPostProcMapInMoab<2, 2>;
 
  auto post_proc_domain_fe = boost::make_shared<PostProcFaceEle>(mField);
  post_proc_domain_fe->getOpPtrVector().push_back(
      new OpCalculateScalarFieldValues(domainField, d_ptr));
  post_proc_domain_fe->getOpPtrVector().push_back(
      new OpPPMap(post_proc_domain_fe->getPostProcMesh(),
                  post_proc_domain_fe->getMapGaussPts(), {{domainField, d_ptr}},
                  {}, {}, {}));
  pipeline_mng->getDomainRhsFE() = post_proc_domain_fe;
 
  auto post_proc_boundary_fe = boost::make_shared<PostProcEdgeEle>(mField);
  post_proc_boundary_fe->getOpPtrVector().push_back(
      new OpCalculateScalarFieldValues(boundaryField, l_ptr));
  post_proc_boundary_fe->getOpPtrVector().push_back(
      new OpPPMap(post_proc_boundary_fe->getPostProcMesh(),
                  post_proc_boundary_fe->getMapGaussPts(),
                  {{boundaryField, l_ptr}}, {}, {}, {}));
  pipeline_mng->getBoundaryRhsFE() = post_proc_boundary_fe;
 
  CHKERR pipeline_mng->loopFiniteElements();
  CHKERR post_proc_domain_fe->writeFile("out_result_domain.h5m");
  CHKERR post_proc_boundary_fe->writeFile("out_result_boundary.h5m");
 
  MoFEMFunctionReturn(0);
}

readMesh()

MoFEMErrorCode Poisson2DLagrangeMultiplier::readMesh ( )

private

Definition at line 65 of file poisson_2d_lagrange_multiplier.cpp.

                                                     {
  MoFEMFunctionBegin;
 
  CHKERR mField.getInterface(simpleInterface);
  CHKERR simpleInterface->getOptions();
  CHKERR simpleInterface->loadFile();
 
  MoFEMFunctionReturn(0);
}

runProgram()

MoFEMErrorCode Poisson2DLagrangeMultiplier::runProgram

(

)

Definition at line 286 of file poisson_2d_lagrange_multiplier.cpp.

                                                       {
  MoFEMFunctionBegin;
 
  readMesh();
  setupProblem();
  boundaryCondition();
  assembleSystem();
  setIntegrationRules();
  solveSystem();
  outputResults();
 
  MoFEMFunctionReturn(0);
}

setIntegrationRules()

MoFEMErrorCode Poisson2DLagrangeMultiplier::setIntegrationRules ( )

private

Definition at line 179 of file poisson_2d_lagrange_multiplier.cpp.

                                                                {
  MoFEMFunctionBegin;
 
  auto pipeline_mng = mField.getInterface<PipelineManager>();
 
  auto domain_rule_lhs = [](int, int, int p) -> int { return 2 * (p - 1); };
  auto domain_rule_rhs = [](int, int, int p) -> int { return 2 * (p - 1); };
  CHKERR pipeline_mng->setDomainLhsIntegrationRule(domain_rule_lhs);
  CHKERR pipeline_mng->setDomainRhsIntegrationRule(domain_rule_rhs);
 
  auto boundary_rule_lhs = [](int, int, int p) -> int { return 2 * p; };
  auto boundary_rule_rhs = [](int, int, int p) -> int { return 2 * p; };
  CHKERR pipeline_mng->setBoundaryLhsIntegrationRule(boundary_rule_lhs);
  CHKERR pipeline_mng->setBoundaryLhsIntegrationRule(boundary_rule_rhs);
 
  MoFEMFunctionReturn(0);
}

setupProblem()

MoFEMErrorCode Poisson2DLagrangeMultiplier::setupProblem ( )

private

Definition at line 75 of file poisson_2d_lagrange_multiplier.cpp.

                                                         {
  MoFEMFunctionBegin;
 
  CHKERR simpleInterface->addDomainField(domainField, H1,
                                         AINSWORTH_BERNSTEIN_BEZIER_BASE, 1);
  CHKERR simpleInterface->addBoundaryField(boundaryField, H1,
                                           AINSWORTH_BERNSTEIN_BEZIER_BASE, 1);
 
  int oRder = 3;
  CHKERR PetscOptionsGetInt(PETSC_NULL, "", "-order", &oRder, PETSC_NULL);
  CHKERR simpleInterface->setFieldOrder(domainField, oRder);
  CHKERR simpleInterface->setFieldOrder(boundaryField, oRder);
 
  CHKERR simpleInterface->setUp();
 
  MoFEMFunctionReturn(0);
}

solveSystem()

MoFEMErrorCode Poisson2DLagrangeMultiplier::solveSystem ( )

private

Definition at line 197 of file poisson_2d_lagrange_multiplier.cpp.

                                                        {
  MoFEMFunctionBegin;
 
  auto pipeline_mng = mField.getInterface<PipelineManager>();
 
  auto ksp_solver = pipeline_mng->createKSP();
  CHKERR KSPSetFromOptions(ksp_solver);
 
  // Create RHS and solution vectors
  auto dm = simpleInterface->getDM();
  auto F = createDMVector(dm);
  auto D = vectorDuplicate(F);
 
  // Setup fieldsplit (block) solver - optional: yes/no
  if (1) {
    PC pc;
    CHKERR KSPGetPC(ksp_solver, &pc);
    PetscBool is_pcfs = PETSC_FALSE;
    PetscObjectTypeCompare((PetscObject)pc, PCFIELDSPLIT, &is_pcfs);
 
    // Set up FIELDSPLIT, only when user set -pc_type fieldsplit
    // Identify the index for boundary entities, remaining will be for domain
    // Then split the fields for boundary and domain for solving
    if (is_pcfs == PETSC_TRUE) {
      IS is_domain, is_boundary;
      cerr << "Running FIELDSPLIT..." << endl;
      const MoFEM::Problem *problem_ptr;
      CHKERR DMMoFEMGetProblemPtr(dm, &problem_ptr);
      CHKERR mField.getInterface<ISManager>()->isCreateProblemFieldAndRank(
          problem_ptr->getName(), ROW, domainField, 0, 1, &is_boundary,
          &boundaryEntitiesForFieldsplit);
      // CHKERR ISView(is_boundary, PETSC_VIEWER_STDOUT_SELF);
 
      CHKERR PCFieldSplitSetIS(pc, NULL, is_boundary);
 
      CHKERR ISDestroy(&is_boundary);
    }
  }
 
  CHKERR KSPSetUp(ksp_solver);
 
  // Solve the system
  CHKERR KSPSolve(ksp_solver, F, D);
 
  // Scatter result data on the mesh
  CHKERR VecGhostUpdateBegin(D, INSERT_VALUES, SCATTER_FORWARD);
  CHKERR VecGhostUpdateEnd(D, INSERT_VALUES, SCATTER_FORWARD);
  CHKERR DMoFEMMeshToLocalVector(dm, D, INSERT_VALUES, SCATTER_REVERSE);
 
  MoFEMFunctionReturn(0);
}

sourceTermFunction()

static double Poisson2DLagrangeMultiplier::sourceTermFunction ( const double  x, const double  y, const double  z  )

inlinestaticprivate

Definition at line 33 of file poisson_2d_lagrange_multiplier.cpp.

                                                   {
    return 200 * sin(x * 10.) * cos(y * 10.);
    // return 1;
  }

Member Data Documentation

boundaryEntitiesForFieldsplit

Range Poisson2DLagrangeMultiplier::boundaryEntitiesForFieldsplit

private

Definition at line 58 of file poisson_2d_lagrange_multiplier.cpp.

boundaryField

std::string Poisson2DLagrangeMultiplier::boundaryField

private

Definition at line 51 of file poisson_2d_lagrange_multiplier.cpp.

boundaryMarker

boost::shared_ptr<std::vector<unsigned char> > Poisson2DLagrangeMultiplier::boundaryMarker

private

Definition at line 55 of file poisson_2d_lagrange_multiplier.cpp.

domainField

std::string Poisson2DLagrangeMultiplier::domainField

private

Definition at line 50 of file poisson_2d_lagrange_multiplier.cpp.

mField

MoFEM::Interface& Poisson2DLagrangeMultiplier::mField

private

Definition at line 46 of file poisson_2d_lagrange_multiplier.cpp.

oRder

int Poisson2DLagrangeMultiplier::oRder

private

Definition at line 52 of file poisson_2d_lagrange_multiplier.cpp.

simpleInterface

Simple* Poisson2DLagrangeMultiplier::simpleInterface

private

Definition at line 47 of file poisson_2d_lagrange_multiplier.cpp.

---

The documentation for this struct was generated from the following file:

• users_modules/tutorials/scl-3/poisson_2d_lagrange_multiplier.cpp