CD_MeshODEStepperImplem.H

Go to the documentation of this file.

/* chombo-discharge
 * Copyright © 2022 SINTEF Energy Research.
 * Please refer to Copyright.txt and LICENSE in the chombo-discharge root directory.
 */
 
#ifndef CD_MeshODEStepperImplem_H
#define CD_MeshODEStepperImplem_H
 
// Our includes
#include <CD_TimeStepper.H>
#include <CD_MeshODESolver.H>
#include <CD_NamespaceHeader.H>
 
using namespace Physics::MeshODE;
 
template <size_t N>
MeshODEStepper<N>::MeshODEStepper()
{
  CH_TIME("MeshODEStepper::MeshODEStepper");
 
  m_verbosity = 10;
  m_realm     = Realm::Primal;
  m_phase     = phase::gas;
 
  this->parseOptions();
}
 
template <size_t N>
MeshODEStepper<N>::~MeshODEStepper()
{
  CH_TIME("MeshODEStepper::~MeshODEStepper");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::~MeshODEStepper" << endl;
  }
}
 
template <size_t N>
void
MeshODEStepper<N>::setupSolvers()
{
  CH_TIME("MeshODEStepper::setupSolvers");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::setupSolvers" << endl;
  }
 
  m_solver = RefCountedPtr<MeshODESolver<N>>(new MeshODESolver<N>(m_amr));
 
  m_solver->setPhase(m_phase);
  m_solver->setRealm(m_realm);
  m_solver->parseOptions();
}
 
template <size_t N>
void
MeshODEStepper<N>::allocate()
{
  CH_TIME("MeshODEStepper::allocate()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::allocate()" << endl;
  }
 
  m_solver->allocate();
}
 
template <size_t N>
void
MeshODEStepper<N>::initialData()
{
  CH_TIME("MeshODEStepper::initialData()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::initialData()" << endl;
  }
 
  m_solver->setPhi(m_initialData);
  m_solver->computeRHS(m_rhsFunction);
}
 
template <size_t N>
void
MeshODEStepper<N>::postInitialize()
{
  CH_TIME("MeshODEStepper::postInitialize()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::postInitialize()" << endl;
  }
}
 
template <size_t N>
void
MeshODEStepper<N>::postCheckpointSetup()
{
  CH_TIME("MeshODEStepper::postCheckpointSetup()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::postCheckpointSetup()" << endl;
  }
 
  m_solver->computeRHS(m_rhsFunction);
}
 
template <size_t N>
void
MeshODEStepper<N>::registerRealms()
{
  CH_TIME("MeshODEStepper::registerRealms()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::registerRealms()" << endl;
  }
 
  m_amr->registerRealm(Realm::Primal);
}
 
template <size_t N>
void
MeshODEStepper<N>::registerOperators()
{
  CH_TIME("MeshODEStepper::registerOperators()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::registerOperators()" << endl;
  }
 
  m_solver->registerOperators();
}
 
template <size_t N>
void
MeshODEStepper<N>::parseOptions()
{
  CH_TIME("MeshODEStepper::parseOptions()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::parseOptions()" << endl;
  }
 
  this->parseIntegrator();
  this->parseVerbosity();
  this->parseProblem();
}
 
template <size_t N>
void
MeshODEStepper<N>::parseRuntimeOptions()
{
  CH_TIME("MeshODEStepper::parseRuntimeOptions()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::parseRuntimeOptions()" << endl;
  }
 
  m_solver->parseRuntimeOptions();
 
  this->parseIntegrator();
  this->parseVerbosity();
}
 
template <size_t N>
void
MeshODEStepper<N>::parseProblem()
{
  CH_TIME("MeshODEStepper::parseProblem()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::parseProblem()" << endl;
  }
 
  ParmParse pp("MeshODEStepper");
 
  Real initPhi   = 0.0;
  Real frequency = 0.0;
 
  pp.get("init_phi", initPhi);
  pp.get("frequency", frequency);
  pp.get("dt", m_dt);
 
  m_initialData = [initPhi](const RealVect& a_pos) -> std::array<Real, N> {
    std::array<Real, N> Y;
 
    for (auto& y : Y) {
      y = initPhi;
    }
 
    return Y;
  };
 
  m_rhsFunction = [frequency](const std::array<Real, N> Y, const Real t) -> std::array<Real, N> {
    std::array<Real, N> rhs;
 
    for (auto& r : rhs) {
      r = cos(2 * M_PI * frequency * t);
    }
 
    return rhs;
  };
}
 
#ifdef CH_USE_HDF5
template <size_t N>
void
MeshODEStepper<N>::writeCheckpointData(HDF5Handle& a_handle, const int a_lvl) const
{
  CH_TIME("MeshODEStepper::writeCheckpointData(HDF5Handle, int)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::writeCheckpointData(HDF5Handle, int)" << endl;
  }
 
  m_solver->writeCheckpointLevel(a_handle, a_lvl);
}
#endif
 
#ifdef CH_USE_HDF5
template <size_t N>
void
MeshODEStepper<N>::readCheckpointData(HDF5Handle& a_handle, const int a_lvl)
{
  CH_TIME("MeshODEStepper::readCheckpointData(HDF5Handle, int)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::readCheckpointData(HDF5Handle, int)" << endl;
  }
 
  m_solver->readCheckpointLevel(a_handle, a_lvl);
}
#endif
 
template <size_t N>
int
MeshODEStepper<N>::getNumberOfPlotVariables() const
{
  CH_TIME("MeshODEStepper::getNumberOfPlotVariables()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::getNumberOfPlotVariables()" << endl;
  }
 
  return m_solver->getNumberOfPlotVariables();
}
 
template <size_t N>
Vector<std::string>
MeshODEStepper<N>::getPlotVariableNames() const
{
  CH_TIME("MeshODEStepper::getPlotVariableNames()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::getPlotVariableNames()" << endl;
  }
 
  return m_solver->getPlotVariableNames();
}
 
template <size_t N>
void
MeshODEStepper<N>::writePlotData(LevelData<EBCellFAB>& a_output,
                                 int&                  a_icomp,
                                 const std::string     a_outputRealm,
                                 const int             a_level) const
{
  CH_TIME("MeshODEStepper::writePlotData");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::writePlotData" << endl;
  }
 
  m_solver->writePlotData(a_output, a_icomp, a_outputRealm, a_level);
}
 
template <size_t N>
Real
MeshODEStepper<N>::computeDt()
{
  CH_TIME("MeshODEStepper::computeDt()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::computeDt()" << endl;
  }
 
  return m_dt;
}
 
template <size_t N>
Real
MeshODEStepper<N>::advance(const Real a_dt)
{
  CH_TIME("MeshODEStepper::avdvance(Real)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::advance(Real)" << endl;
  }
 
  switch (m_algorithm) {
  case IntegrationAlgorithm::Euler: {
    this->advanceEuler(a_dt);
 
    break;
  }
  case IntegrationAlgorithm::RK2: {
    this->advanceRK2(a_dt);
 
    break;
  }
  case IntegrationAlgorithm::RK4: {
    this->advanceRK4(a_dt);
 
    break;
  }
  default: {
    MayDay::Error("MeshODEStepper::advance -- logic bust");
 
    break;
  }
  }
 
  // Update the source term.
  m_solver->computeRHS(m_rhsFunction);
 
  return a_dt;
}
 
template <size_t N>
void
MeshODEStepper<N>::synchronizeSolverTimes(const int a_step, const Real a_time, const Real a_dt)
{
  CH_TIME("MeshODEStepper::synchronizeSolverTimes(int, Real, Real)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::synchronizeSolverTimes(int, Real, Real)" << endl;
  }
 
  m_timeStep = a_step;
  m_time     = a_time;
 
  m_solver->setTime(a_step, a_time, a_dt);
}
 
template <size_t N>
void
MeshODEStepper<N>::preRegrid(const int a_lmin, const int a_oldFinestLevel)
{
  CH_TIME("MeshODEStepper::preRegrid(int, int)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::preRegrid(int, int)" << endl;
  }
 
  m_solver->preRegrid(a_lmin, a_oldFinestLevel);
}
 
template <size_t N>
void
MeshODEStepper<N>::regrid(const int a_lmin, const int a_oldFinestLevel, const int a_newFinestLevel)
{
  CH_TIME("MeshODEStepper::regrid(int, int, int)");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::regrid(int, int, int)" << endl;
  }
 
  m_solver->regrid(a_lmin, a_oldFinestLevel, a_newFinestLevel);
}
 
template <size_t N>
void
MeshODEStepper<N>::postRegrid()
{
  CH_TIME("MeshODEStepper::postRegrid()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::postRegrid()" << endl;
  }
 
  // Compute new right-hand side.
  m_solver->computeRHS(m_rhsFunction);
}
 
template <size_t N>
void
MeshODEStepper<N>::parseIntegrator()
{
  CH_TIME("MeshODEStepper::parseIntegrator()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::parseIntegrator()" << endl;
  }
 
  ParmParse pp("MeshODEStepper");
 
  std::string str;
 
  pp.get("integration", str);
  if (str == "euler") {
    m_algorithm = IntegrationAlgorithm::Euler;
  }
  else if (str == "rk2") {
    m_algorithm = IntegrationAlgorithm::RK2;
  }
  else if (str == "rk4") {
    m_algorithm = IntegrationAlgorithm::RK4;
  }
  else {
    MayDay::Error("MeshODEStepper::parseIntegrator -- logic bust");
  }
}
 
template <size_t N>
void
MeshODEStepper<N>::parseVerbosity()
{
  CH_TIME("MeshODEStepper::parseVerbosity()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::parseVerbosity()" << endl;
  }
 
  ParmParse pp("MeshODEStepper");
 
  pp.get("verbosity", m_verbosity);
}
 
template <size_t N>
void
MeshODEStepper<N>::advanceEuler(const Real a_dt)
{
  CH_TIME("MeshODEStepper::advanceEuler()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::advanceEuler()" << endl;
  }
 
  // Let Y = Y + S * dt
  EBAMRCellData& phi = m_solver->getPhi();
  EBAMRCellData& rhs = m_solver->getRHS();
 
  m_solver->computeRHS(rhs, m_rhsFunction);
 
  DataOps::incr(phi, rhs, a_dt);
}
 
template <size_t N>
void
MeshODEStepper<N>::advanceRK2(const Real a_dt)
{
  CH_TIME("MeshODEStepper::advanceRK2()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::advanceRK2()" << endl;
  }
 
  EBAMRCellData& phi = m_solver->getPhi();
 
  EBAMRCellData k1;
  EBAMRCellData k2;
 
  m_amr->allocate(k1, m_realm, m_phase, N);
  m_amr->allocate(k2, m_realm, m_phase, N);
 
  m_solver->computeRHS(k1, m_rhsFunction);
  DataOps::incr(phi, k1, a_dt);
 
  m_solver->computeRHS(k2, m_rhsFunction);
  DataOps::incr(phi, k2, 0.5 * a_dt);
  DataOps::incr(phi, k1, -0.5 * a_dt);
}
 
template <size_t N>
void
MeshODEStepper<N>::advanceRK4(const Real a_dt)
{
  CH_TIME("MeshODEStepper::advanceRK4()");
  if (m_verbosity > 5) {
    pout() << "MeshODEStepper::advanceRK4()" << endl;
  }
 
  MayDay::Error("MeshODEStepper::advanceRK4 -- not implemented");
}
 
#include <CD_NamespaceFooter.H>
 
#endif