radiation.cpp

Go to the documentation of this file.

/**
 * \file lesson6_radiation.cpp
 * \example lesson6_radiation.cpp
 *
 * Using PipelineManager interface calculate the divergence of base functions,
 * and integral of flux on the boundary. Since the h-div space is used, volume
 * integral and boundary integral should give the same result.
 */
 
 
 
#include <MoFEM.hpp>
 
using namespace MoFEM;
 
static char help[] = "...\n\n";
 
#include <MoFEM.hpp>
 
using DomainEle = PipelineManager::FaceEle;
using DomainEleOp = DomainEle::UserDataOperator;
using EdgeEle = PipelineManager::EdgeEle;
using EdgeEleOp = EdgeEle::UserDataOperator;
using EntData = EntitiesFieldData::EntData;
using OpDomainGradGrad = FormsIntegrators<DomainEleOp>::Assembly<
    PETSC>::BiLinearForm<GAUSS>::OpGradGrad<1, 1, 2>;
using OpDomainGradTimesVec = FormsIntegrators<DomainEleOp>::Assembly<
    PETSC>::LinearForm<GAUSS>::OpGradTimesTensor<1, 1, 2>;
using OpBase = OpBaseImpl<PETSC, EdgeEleOp>;
 
// Units
// Temperature: Kelvins
// Length: 1 km
// Time: 1 s
 
constexpr double heat_conductivity = ((0.4 + 0.7) / 2) * 1e3;
 
constexpr double emissivity = 1;
constexpr double boltzmann_constant = 5.670374419e-2;
constexpr double Beta = emissivity * boltzmann_constant;
 
constexpr double T_ambient = 2.7;
struct Example {
 
  Example(MoFEM::Interface &m_field) : mField(m_field) {}
 
  MoFEMErrorCode runProblem();
 
private:
  MoFEM::Interface &mField;
 
  static int integrationRule(int, int, int p_data) { return 2 * p_data; };
 
  MoFEMErrorCode setupProblem();
  MoFEMErrorCode createCommonData();
  MoFEMErrorCode bC();
  MoFEMErrorCode OPs();
  MoFEMErrorCode kspSolve();
  MoFEMErrorCode postProcess();
  MoFEMErrorCode checkResults();
 
  boost::shared_ptr<VectorDouble> approxVals;
  boost::shared_ptr<MatrixDouble> approxGradVals;
 
  struct OpRadiationLhs : public OpBase {
 
  private:
    boost::shared_ptr<VectorDouble> approxVals;
 
  public:
    OpRadiationLhs(boost::shared_ptr<VectorDouble> &approx_vals)
        : OpBase("T", "T", OpBase::OPROWCOL), approxVals(approx_vals) {
      this->sYmm = false;
    }
 
    MoFEMErrorCode iNtegrate(EntData &row_data, EntData &col_data);
  };
 
  struct OpRadiationRhs : public OpBase {
 
  private:
    boost::shared_ptr<VectorDouble> approxVals;
 
  public:
    OpRadiationRhs(boost::shared_ptr<VectorDouble> &approx_vals)
        : OpBase("T", "T", OpBase::OPROW), approxVals(approx_vals) {}
 
    MoFEMErrorCode iNtegrate(EntData &row_data);
  };
 
  struct OpFluxRhs : public OpBase {
 
  private:
    FTensor::Index<'i', 2> i; ///< summit Index
 
  public:
    OpFluxRhs() : OpBase("T", "T", OpBase::OPROW) {}
 
    MoFEMErrorCode iNtegrate(EntData &row_data);
  };
 
  struct OpCalcSurfaceAverageTemperature : public EdgeEleOp {
 
  private:
    boost::shared_ptr<VectorDouble> approxVals;
    double &sumTemperature;
    double &surfaceArea;
 
  public:
    OpCalcSurfaceAverageTemperature(
        boost::shared_ptr<VectorDouble> &approx_vals, double &sum_temp,
        double &surf)
        : EdgeEleOp("T", "T", OpBase::OPROW), approxVals(approx_vals),
          sumTemperature(sum_temp), surfaceArea(surf) {}
 
    MoFEMErrorCode doWork(int side, EntityType type,
                          EntitiesFieldData::EntData &data);
  };
};
 
MoFEMErrorCode Example::runProblem() {
  MoFEMFunctionBegin;
  CHKERR setupProblem();
  CHKERR createCommonData();
  CHKERR bC();
  CHKERR OPs();
  CHKERR kspSolve();
  CHKERR postProcess();
  CHKERR checkResults();
  MoFEMFunctionReturn(0);
}
 
//! [Set up problem]
MoFEMErrorCode Example::setupProblem() {
  MoFEMFunctionBegin;
  Simple *simple = mField.getInterface<Simple>();
  // Add field
  CHKERR simple->addDomainField("T", H1, AINSWORTH_LEGENDRE_BASE, 1);
  CHKERR simple->addBoundaryField("T", H1, AINSWORTH_LEGENDRE_BASE, 1);
  constexpr int order = 3;
  CHKERR simple->setFieldOrder("T", order);
  CHKERR simple->setUp();
  MoFEMFunctionReturn(0);
}
//! [Set up problem]
 
//! [Create common data]
MoFEMErrorCode Example::createCommonData() {
  MoFEMFunctionBegin;
  approxVals = boost::make_shared<VectorDouble>();
  approxGradVals = boost::make_shared<MatrixDouble>();
  MoFEMFunctionReturn(0);
}
//! [Create common data]
 
//! [Boundary condition]
MoFEMErrorCode Example::bC() {
  MoFEMFunctionBegin;
  // Set initial values
  auto set_initial_temperature = [&](VectorAdaptor &&field_data, double *xcoord,
                                     double *ycoord, double *zcoord) {
    MoFEMFunctionBeginHot;
    field_data[0] = T_ambient;
    MoFEMFunctionReturnHot(0);
  };
  FieldBlas *field_blas;
  CHKERR mField.getInterface(field_blas);
  CHKERR field_blas->setVertexDofs(set_initial_temperature, "T");
  MoFEMFunctionReturn(0);
}
//! [Boundary condition]
 
//! [Push operators to pipeline]
MoFEMErrorCode Example::OPs() {
  MoFEMFunctionBegin;
  PipelineManager *pipeline_mng = mField.getInterface<PipelineManager>();
  auto beta = [](const double r, const double, const double) {
    return heat_conductivity * (2 * M_PI * r);
  };
 
  auto det_ptr = boost::make_shared<VectorDouble>();
  auto jac_ptr = boost::make_shared<MatrixDouble>();
  auto inv_jac_ptr = boost::make_shared<MatrixDouble>();
 
  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 OpDomainGradGrad("T", "T", beta));
  CHKERR pipeline_mng->setDomainLhsIntegrationRule(integrationRule);
 
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpCalculateHOJac<2>(jac_ptr));
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpInvertMatrix<2>(jac_ptr, det_ptr, inv_jac_ptr));
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpSetHOInvJacToScalarBases<2>(H1, inv_jac_ptr));
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpSetHOWeightsOnFace());
 
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpCalculateScalarFieldGradient<2>("T", approxGradVals));
  pipeline_mng->getOpDomainRhsPipeline().push_back(
      new OpDomainGradTimesVec("T", approxGradVals, beta));
  CHKERR pipeline_mng->setDomainRhsIntegrationRule(integrationRule);
 
  pipeline_mng->getOpBoundaryRhsPipeline().push_back(
      new OpCalculateScalarFieldValues("T", approxVals));
  pipeline_mng->getOpBoundaryRhsPipeline().push_back(
      new OpRadiationRhs(approxVals));
  pipeline_mng->getOpBoundaryRhsPipeline().push_back(new OpFluxRhs());
  CHKERR pipeline_mng->setBoundaryRhsIntegrationRule(integrationRule);
 
  pipeline_mng->getOpBoundaryLhsPipeline().push_back(
      new OpCalculateScalarFieldValues("T", approxVals));
  pipeline_mng->getOpBoundaryLhsPipeline().push_back(
      new OpRadiationLhs(approxVals));
  CHKERR pipeline_mng->setBoundaryLhsIntegrationRule(integrationRule);
  MoFEMFunctionReturn(0);
}
//! [Push operators to pipeline]
 
//! [Solve]
MoFEMErrorCode Example::kspSolve() {
  MoFEMFunctionBegin;
  Simple *simple = mField.getInterface<Simple>();
  PipelineManager *pipeline_mng = mField.getInterface<PipelineManager>();
  auto ts = pipeline_mng->createTSIM();
 
  double ftime = 1;
  CHKERR TSSetDuration(ts, PETSC_DEFAULT, ftime);
  CHKERR TSSetFromOptions(ts);
  CHKERR TSSetExactFinalTime(ts, TS_EXACTFINALTIME_MATCHSTEP);
 
  auto T = createDMVector(simple->getDM());
  CHKERR DMoFEMMeshToLocalVector(simple->getDM(), T, INSERT_VALUES,
                                 SCATTER_FORWARD);
 
  CHKERR TSSolve(ts, T);
  CHKERR TSGetTime(ts, &ftime);
 
  PetscInt steps, snesfails, rejects, nonlinits, linits;
  CHKERR TSGetTimeStepNumber(ts, &steps);
  CHKERR TSGetSNESFailures(ts, &snesfails);
  CHKERR TSGetStepRejections(ts, &rejects);
  CHKERR TSGetSNESIterations(ts, &nonlinits);
  CHKERR TSGetKSPIterations(ts, &linits);
  MOFEM_LOG_C("EXAMPLE", Sev::inform,
              "steps %d (%d rejected, %d SNES fails), ftime %g, nonlinits "
              "%d, linits %d",
              steps, rejects, snesfails, ftime, nonlinits, linits);
 
  MoFEMFunctionReturn(0);
}
//! [Solve]
 
//! [Postprocess results]
MoFEMErrorCode Example::postProcess() {
  MoFEMFunctionBegin;
  PipelineManager *pipeline_mng = mField.getInterface<PipelineManager>();
  pipeline_mng->getDomainLhsFE().reset();
  pipeline_mng->getBoundaryLhsFE().reset();
  pipeline_mng->getBoundaryRhsFE().reset();
  auto post_proc_fe =
      boost::make_shared<PostProcBrokenMeshInMoab<DomainEle>>(mField);
 
  auto t_ptr = boost::make_shared<VectorDouble>();
  post_proc_fe->getOpPtrVector().push_back(
      new OpCalculateScalarFieldValues("T", t_ptr));
 
  using OpPPMap = OpPostProcMapInMoab<2, 2>;
 
  post_proc_fe->getOpPtrVector().push_back(
 
      new OpPPMap(post_proc_fe->getPostProcMesh(),
                  post_proc_fe->getMapGaussPts(),
 
                  {{"T", t_ptr}},
 
                  {}, {}, {}
 
                  )
 
  );
 
  pipeline_mng->getDomainRhsFE() = post_proc_fe;
 
  pipeline_mng->getOpBoundaryRhsPipeline().push_back(
      new OpCalculateScalarFieldValues("T", approxVals));
 
  double sum_temperature;
  double surface_area;
  pipeline_mng->getOpBoundaryRhsPipeline().push_back(
      new OpCalcSurfaceAverageTemperature(approxVals, sum_temperature,
                                          surface_area));
  auto calc_surfcae_area_op = pipeline_mng->getOpBoundaryRhsPipeline().back();
 
  sum_temperature = 0;
  surface_area = 0;
  CHKERR pipeline_mng->loopFiniteElements();
  CHKERR post_proc_fe->writeFile("out_radiation.h5m");
 
  MOFEM_LOG_C("EXAMPLE", Sev::inform, "Surface area %3.4e [km]", surface_area);
  MOFEM_LOG_C("EXAMPLE", Sev::inform,
              "Average subsurface temperatute %3.4e [K]",
              sum_temperature / surface_area);
 
  MoFEMFunctionReturn(0);
}
//! [Postprocess results]
 
//! [Check results]
MoFEMErrorCode Example::checkResults() { return 0; }
//! [Check results]
 
int main(int argc, char *argv[]) {
 
  // Initialisation of MoFEM/PETSc and MOAB data structures
  const char param_file[] = "param_file.petsc";
  MoFEM::Core::Initialize(&argc, &argv, param_file, help);
 
  // Add logging channel for example
  auto core_log = logging::core::get();
  core_log->add_sink(
      LogManager::createSink(LogManager::getStrmWorld(), "EXAMPLE"));
  LogManager::setLog("EXAMPLE");
  MOFEM_LOG_TAG("EXAMPLE", "example");
 
  try {
 
    //! [Register MoFEM discrete manager in PETSc]
    DMType dm_name = "DMMOFEM";
    CHKERR DMRegister_MoFEM(dm_name);
    //! [Register MoFEM discrete manager in PETSc
 
    //! [Create MoAB]
    moab::Core mb_instance;              ///< mesh database
    moab::Interface &moab = mb_instance; ///< mesh database interface
    //! [Create MoAB]
 
    //! [Create MoFEM]
    MoFEM::Core core(moab);           ///< finite element database
    MoFEM::Interface &m_field = core; ///< finite element database insterface
    //! [Create MoFEM]
 
    //! [Load mesh]
    Simple *simple = m_field.getInterface<Simple>();
    CHKERR simple->getOptions();
    CHKERR simple->loadFile("");
    //! [Load mesh]
 
    //! [Example]
    Example ex(m_field);
    CHKERR ex.runProblem();
    //! [Example]
  }
  CATCH_ERRORS;
 
  CHKERR MoFEM::Core::Finalize();
}
 
//! [Radiation Lhs]
MoFEMErrorCode Example::OpRadiationLhs::iNtegrate(EntData &row_data,
                                                  EntData &col_data) {
  MoFEMFunctionBegin;
  // get element volume
  const double vol = this->getMeasure();
  // get integration weights
  auto t_w = getFTensor0IntegrationWeight();
  // get base function gradient on rows
  auto t_row_base = row_data.getFTensor0N();
  // gat temperature at integration points
  auto t_val = getFTensor0FromVec(*(approxVals));
  // get coordinate at integration points
  auto t_coords = getFTensor1CoordsAtGaussPts();
 
  // loop over integration points
  for (int gg = 0; gg != nbIntegrationPts; gg++) {
 
    // Cylinder radius
    const double r_cylinder = t_coords(0);
 
    // take into account Jacobean
    const double alpha = t_w * vol * Beta * (2 * M_PI * r_cylinder);
    // loop over rows base functions
    for (int rr = 0; rr != nbRows; ++rr) {
      auto t_col_base = col_data.getFTensor0N(gg, 0);
      // loop over columns
      for (int cc = 0; cc != nbCols; cc++) {
        if (std::abs(t_coords(0)) > std::numeric_limits<double>::epsilon()) {
          locMat(rr, cc) += alpha * t_row_base * t_col_base * 4 *
                            std::pow(static_cast<double>(t_val), 3);
        }
        ++t_col_base;
      }
      ++t_row_base;
    }
 
    ++t_val;
    ++t_coords;
    ++t_w; // move to another integration weight
  }
  MoFEMFunctionReturn(0);
}
//! [Radiation Lhs]
 
//! [Radiation Lhs]
MoFEMErrorCode Example::OpRadiationRhs::iNtegrate(EntData &row_data) {
  MoFEMFunctionBegin;
  // get element volume
  const double vol = getMeasure();
  // get integration weights
  auto t_w = getFTensor0IntegrationWeight();
  // get base function gradient on rows
  auto t_row_base = row_data.getFTensor0N();
  // gat temperature at integration points
  auto t_val = getFTensor0FromVec(*(approxVals));
  // get coordinate at integration points
  auto t_coords = getFTensor1CoordsAtGaussPts();
 
  // loop over integration points
  for (int gg = 0; gg != nbIntegrationPts; gg++) {
 
    // Cylinder radius
    const double r_cylinder = t_coords(0);
 
    // take into account Jacobean
    const double alpha = t_w * vol * Beta * (2 * M_PI * r_cylinder);
    // loop over rows base functions
    for (int rr = 0; rr != nbRows; ++rr) {
      if (std::abs(t_coords(0)) > std::numeric_limits<double>::epsilon()) {
        locF[rr] += alpha * t_row_base *
                    (std::pow(static_cast<double>(t_val), 4) -
                     std::pow(T_ambient, 4));
      }
      ++t_row_base;
    }
    ++t_coords;
    ++t_val;
    ++t_w; // move to another integration weight
  }
 
  MoFEMFunctionReturn(0);
}
//! [Radiation Lhs]
 
//! [Flux Rhs]
MoFEMErrorCode Example::OpFluxRhs::iNtegrate(EntData &row_data) {
  MoFEMFunctionBegin;
  // get element volume
  const double vol = getMeasure();
  // get integration weights
  auto t_w = getFTensor0IntegrationWeight();
  // get base function gradient on rows
  auto t_row_base = row_data.getFTensor0N();
  // get coordinate at integration points
  auto t_coords = getFTensor1CoordsAtGaussPts();
  // // get time
  const double time = getFEMethod()->ts_t;
 
  // Look to https://doi.org/10.1016/j.icarus.2014.12.028s
  constexpr double flux_p = -0.03e6;
  constexpr double flux_c = -0.23e6;
 
  // loop over integration points
  for (int gg = 0; gg != nbIntegrationPts; gg++) {
 
    // Cylinder radius
    const double r_cylinder = t_coords(0);
 
    const double r = std::sqrt(t_coords(i) * t_coords(i));
    const double s = std::abs(t_coords(1)) / r;
 
    // take into account Jacobean
    const double alpha = t_w * vol * (2 * M_PI * r_cylinder);
    // loop over rows base functions
    for (int rr = 0; rr != nbRows; ++rr) {
      locF[rr] += alpha * t_row_base * (s * flux_p + flux_c) * time;
      ++t_row_base;
    }
    ++t_coords;
    ++t_w; // move to another integration weight
  }
 
  MoFEMFunctionReturn(0);
}
//! [Flux Rhs]
 
//! [Ave Temp]
MoFEMErrorCode Example::OpCalcSurfaceAverageTemperature::doWork(
    int side, EntityType type, EntitiesFieldData::EntData &data) {
 
  MoFEMFunctionBegin;
  if (type == MBVERTEX) {
    // get element volume
    const double vol = getMeasure();
    // get integration weights
    auto t_w = getFTensor0IntegrationWeight();
    // gat temperature at integration points
    auto t_val = getFTensor0FromVec(*(approxVals));
    // get coordinate at integration points
    auto t_coords = getFTensor1CoordsAtGaussPts();
    // number of integration pts
    size_t nb_integration_pts = getGaussPts().size2();
 
    // loop over integration points
    for (auto gg = 0; gg != nb_integration_pts; ++gg) {
 
      // Cylinder radius
      const double r_cylinder = t_coords(0);
 
      // take into account Jacobean
      const double alpha = t_w * vol * (2 * M_PI * r_cylinder);
 
      sumTemperature += alpha * t_val;
      surfaceArea += alpha;
 
      ++t_coords;
      ++t_val;
      ++t_w; // move to another integration weight
    }
  }
  MoFEMFunctionReturn(0);
}
 
//! [Ave Temp]