Monitor.hpp

Go to the documentation of this file.

/**
 * @brief Monitor class for post-processing results
 * 
 */
 
struct Monitor : public FEMethod {
  Monitor(SmartPetscObj<DM> dm,
          std::pair<boost::shared_ptr<PostProcEleDomain>,
                    boost::shared_ptr<PostProcEleBdy>>
              pair_post_proc,
          boost::shared_ptr<DomainEle> reaction_fe)
      : dM(dm), postProcDomain(pair_post_proc.first),
        postProcSkin(pair_post_proc.second), reactionFe(reaction_fe) {
    CHK_THROW_MESSAGE(PetscOptionsGetInt(PETSC_NULLPTR, "", "-save_every",
                                         &saveEvery, PETSC_NULLPTR),
                      "Cannot get -save_every option");
  }
 
  MoFEMErrorCode /** postProcess */ postProcess(); 
 
private:
  SmartPetscObj<DM> dM;
  boost::shared_ptr<PostProcEleDomain> postProcDomain;
  boost::shared_ptr<PostProcEleBdy> postProcSkin;
  boost::shared_ptr<DomainEle> reactionFe;
  PetscInt saveEvery = 1;
};
 
MoFEMErrorCode Monitor::postProcess() {
  MoFEMFunctionBegin;
 
  auto make_vtk = [&]() {
    MoFEMFunctionBegin;
    if (postProcDomain) {
      CHKERR DMoFEMLoopFiniteElements(dM, "dFE", postProcDomain,
                                      getCacheWeakPtr());
      CHKERR postProcDomain->writeFile(
          "out_step_" + boost::lexical_cast<std::string>(ts_step) + ".h5m");
    }
    if (postProcSkin) {
      CHKERR DMoFEMLoopFiniteElements(dM, "bFE", postProcSkin,
                                      getCacheWeakPtr());
      CHKERR postProcSkin->writeFile(
          "out_skin_" + boost::lexical_cast<std::string>(ts_step) + ".h5m");
    }
    MoFEMFunctionReturn(0);
  };
 
  auto calculate_reaction = [&]() {
    MoFEMFunctionBegin;
    auto r = createDMVector(dM);
    reactionFe->f = r;
    CHKERR VecZeroEntries(r);
    CHKERR DMoFEMLoopFiniteElements(dM, "dFE", reactionFe, getCacheWeakPtr());
    MoFEMFunctionReturn(0);
  };
 
  if (!(ts_step % saveEvery)) {
    CHKERR make_vtk();
  }
  if (reactionFe)
    CHKERR calculate_reaction();
 
  MoFEMFunctionReturn(0);
}