CD_KMCSolver.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_KMCSolver_H
#define CD_KMCSolver_H
 
// Std includes
#include <vector>
#include <memory>
 
// Chombo includes
#include <REAL.H>
 
// Our includes
#include <CD_NamespaceHeader.H>
 
enum class KMCLeapPropagator
{
  TauPlain,
  TauMidpoint
};
 
template <typename R, typename State, typename T = long long>
class KMCSolver
{
public:
  using ReactionList = std::vector<std::shared_ptr<const R>>;
 
  KMCSolver() noexcept;
 
  KMCSolver(const KMCSolver&) = default;
 
  KMCSolver(const KMCSolver&&) = delete;
 
  inline KMCSolver(const ReactionList& a_reactions) noexcept;
 
  virtual ~KMCSolver() noexcept;
 
  KMCSolver&
  operator=(const KMCSolver&) = default;
 
  KMCSolver&
  operator=(const KMCSolver&&) = delete;
 
  inline void
  define(const ReactionList& a_reactions) noexcept;
 
  inline void
  setSolverParameters(const T a_numCrit, const T a_numSSA, const Real a_eps, const Real a_SSAlim) noexcept;
 
  inline std::vector<Real>
  propensities(const State& a_state) const noexcept;
 
  inline std::vector<Real>
  propensities(const State& a_state, const ReactionList& a_reactions) const noexcept;
 
  inline Real
  totalPropensity(const State& a_state) const noexcept;
 
  inline Real
  totalPropensity(const State& a_state, const ReactionList& a_reactions) const noexcept;
 
  inline std::pair<ReactionList, ReactionList>
  partitionReactions(const State& a_state) const noexcept;
 
  inline std::pair<ReactionList, ReactionList>
  partitionReactions(const State& a_state, const ReactionList& a_reactions) const noexcept;
 
  inline Real
  getCriticalTimeStep(const State& a_state) const noexcept;
 
  inline Real
  getCriticalTimeStep(const State& a_state, const ReactionList& a_criticalReactions) const noexcept;
 
  inline Real
  getCriticalTimeStep(const std::vector<Real>& a_propensities) const noexcept;
 
  inline Real
  getCriticalTimeStep(const Real& a_totalPropensity) const noexcept;
 
  inline Real
  getNonCriticalTimeStep(const State& a_state) const noexcept;
 
  inline Real
  getNonCriticalTimeStep(const State& a_state, const ReactionList& a_reactions) const noexcept;
 
  inline Real
  getNonCriticalTimeStep(const State&             a_state,
                         const ReactionList&      a_nonCriticalReactions,
                         const std::vector<Real>& a_nonCriticalPropensities) const noexcept;
 
  inline void
  stepTauPlain(State& a_state, const Real a_dt) const noexcept;
 
  inline void
  stepTauPlain(State& a_state, const ReactionList& a_reactions, const Real a_dt) const noexcept;
 
  inline void
  stepTauPlain(State&                   a_state,
               const ReactionList&      a_reactions,
               const std::vector<Real>& a_propensities,
               const Real               a_dt) const noexcept;
 
  inline void
  advanceTauPlain(State& a_state, const Real a_dt) const noexcept;
 
  inline void
  advanceTauPlain(State& a_state, const ReactionList& a_reactions, const Real a_dt) const noexcept;
 
  inline void
  stepTauMidpoint(State& a_state, const Real a_dt) const noexcept;
 
  inline void
  stepTauMidpoint(State& a_state, const ReactionList& a_reactions, const Real a_dt) const noexcept;
 
  inline void
  advanceTauMidpoint(State& a_state, const Real a_dt) const noexcept;
 
  inline void
  advanceTauMidpoint(State& a_state, const ReactionList& a_reactions, const Real a_dt) const noexcept;
 
  inline void
  stepSSA(State& a_state) const noexcept;
 
  inline void
  stepSSA(State& a_state, const ReactionList& a_reactions) const noexcept;
 
  inline void
  stepSSA(State& a_state, const ReactionList& a_reactions, const std::vector<Real>& a_propensities) const noexcept;
 
  inline void
  advanceSSA(State& a_state, const Real a_dt) const noexcept;
 
  inline void
  advanceSSA(State& a_state, const ReactionList& a_reactions, const Real a_dt) const noexcept;
 
  inline void
  advanceHybrid(State&                   a_state,
                const Real               a_dt,
                const KMCLeapPropagator& a_leapPropagator = KMCLeapPropagator::TauPlain) const noexcept;
 
  inline void
  advanceHybrid(State&                   a_state,
                const ReactionList&      a_reactions,
                const Real               a_dt,
                const KMCLeapPropagator& a_leapPropagator = KMCLeapPropagator::TauPlain) const noexcept;
 
  inline void
  advanceHybrid(
    State&                                                                               a_state,
    const ReactionList&                                                                  a_reactions,
    const Real                                                                           a_dt,
    const std::function<void(State&, const ReactionList& a_reactions, const Real a_dt)>& a_propagator) const noexcept;
 
protected:
  ReactionList m_reactions;
 
  T m_Ncrit;
 
  T m_numSSA;
 
  Real m_eps;
 
  Real m_SSAlim;
};
 
#include <CD_NamespaceFooter.H>
 
#include <CD_KMCSolverImplem.H>
 
#endif