CD_BrownianWalkerStepper.H

Go to the documentation of this file.

/* chombo-discharge
 * Copyright © 2021 SINTEF Energy Research.
 * Please refer to Copyright.txt and LICENSE in the chombo-discharge root directory.
 */
 
#ifndef CD_BrownianWalkerStepper_H
#define CD_BrownianWalkerStepper_H
 
// Our includes
#include "CD_TimeStepper.H"
#include <CD_ItoSolver.H>
#include <CD_ItoLayout.H>
#include <CD_NamespaceHeader.H>
 
namespace Physics {
  namespace BrownianWalker {
 
    class BrownianWalkerStepper : public TimeStepper
    {
    public:
      BrownianWalkerStepper();
 
      BrownianWalkerStepper(RefCountedPtr<ItoSolver>& a_solver);
 
      virtual ~BrownianWalkerStepper();
 
      void
      initialData() override;
 
      void
      allocate() override;
 
      void
      postInitialize() override;
 
      bool
      loadBalanceThisRealm(const std::string a_realm) const override;
 
      Vector<long int>
      getCheckpointLoads(const std::string a_realm, const int a_level) const override;
 
      void
      loadBalanceBoxes(Vector<Vector<int>>&             a_procs,
                       Vector<Vector<Box>>&             a_boxes,
                       const std::string                a_realm,
                       const Vector<DisjointBoxLayout>& a_grids,
                       const int                        a_lmin,
                       const int                        a_finestLevel) override;
 
#ifdef CH_USE_HDF5
      void
      writeCheckpointData(HDF5Handle& a_handle, const int a_lvl) const override;
#endif
 
#ifdef CH_USE_HDF5
      void
      readCheckpointData(HDF5Handle& a_handle, const int a_lvl) override;
#endif
 
      void
      postCheckpointSetup() override;
 
      int
      getNumberOfPlotVariables() const override;
 
      Vector<std::string>
      getPlotVariableNames() const override;
 
      void
      writePlotData(LevelData<EBCellFAB>& a_output,
                    int&                  a_icomp,
                    const std::string     a_outputRealm,
                    const int             a_level) const override;
 
      Real
      computeDt() override;
 
      void
      synchronizeSolverTimes(const int a_step, const Real a_time, const Real a_dt) override;
      void
      printStepReport() override;
 
      void
      registerOperators() override;
 
      void
      registerRealms() override;
 
      void
      parseRuntimeOptions() override;
 
      bool
      needToRegrid() override;
 
      void
      preRegrid(const int a_lbase, const int a_oldFinestLevel) override;
 
      void
      setupSolvers() override;
 
      Real
      advance(const Real a_dt) override;
 
      void
      regrid(const int a_lmin, const int a_oldFinestLevel, const int a_newFinestLevel) override;
 
      void
      postRegrid() override;
 
    protected:
      enum LoadBalancingMethod
      {
        Mesh,
        Particle
      };
 
      LoadBalancingMethod m_whichLoadBalance;
 
      std::string m_realm;
 
      phase::which_phase m_phase;
 
      bool m_loadBalance;
 
      RefCountedPtr<ItoSolver> m_solver;
 
      RefCountedPtr<ItoSpecies> m_species;
 
      EBAMRCellData m_regridPPC;
 
      Real m_diffCo;
 
      Real m_mobility;
 
      Real m_omega;
 
      int m_ppc;
 
      Real m_cfl;
 
      void
      setAdvectionDiffusion();
 
      void
      setVelocity();
 
      void
      setDiffusion();
 
      void
      makeSuperParticles();
 
      void
      loadBalanceBoxesMesh(Vector<Vector<int>>&             a_procs,
                           Vector<Vector<Box>>&             a_boxes,
                           const std::string                a_realm,
                           const Vector<DisjointBoxLayout>& a_grids,
                           const int                        a_lmin,
                           const int                        a_finestLevel);
 
      void
      loadBalanceBoxesParticles(Vector<Vector<int>>&             a_procs,
                                Vector<Vector<Box>>&             a_boxes,
                                const std::string                a_realm,
                                const Vector<DisjointBoxLayout>& a_grids,
                                const int                        a_lmin,
                                const int                        a_finestLevel);
    };
  } // namespace BrownianWalker
} // namespace Physics
 
#include <CD_NamespaceFooter.H>
 
#endif