CD_EBAMRParticleMesh.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_EBAMRParticleMesh_H
#define CD_EBAMRParticleMesh_H
 
// Chombo includes
#include <RefCountedPtr.H>
#include <Copier.H>
#include <EBLevelGrid.H>
 
// Our includes
#include <CD_DepositionType.H>
#include <CD_EBAMRData.H>
#include <CD_CoarseFineDeposition.H>
#include <CD_EBCoarseFineParticleMesh.H>
#include <CD_EBParticleMesh.H>
#include <CD_ParticleContainer.H>
#include <CD_NamespaceHeader.H>
 
class EBAMRParticleMesh
{
public:
  EBAMRParticleMesh();
 
  EBAMRParticleMesh(const Vector<RefCountedPtr<EBLevelGrid>>& a_eblgs,
                    const Vector<int>&                        a_refRat,
                    const Vector<Real>&                       a_dx,
                    const RealVect&                           a_probLo,
                    const int&                                a_ghost,
                    const int                                 a_finestLevel);
 
  virtual ~EBAMRParticleMesh();
 
  void
  define(const Vector<RefCountedPtr<EBLevelGrid>>& a_eblgs,
         const Vector<int>&                        a_refRat,
         const Vector<Real>&                       a_dx,
         const RealVect&                           a_probLo,
         const int&                                a_ghost,
         const int                                 a_finestLevel);
 
  template <class P, class Ret, Ret (P::*MemberFunc)() const>
  void
  deposit(EBAMRCellData&              a_meshData,
          const ParticleContainer<P>& a_particles,
          const DepositionType        a_depositionType,
          const CoarseFineDeposition  a_coarseFineDeposition,
          const bool                  a_forceIrregNGP = false);
 
  template <class P, class Ret, Ret (P::*MemberFunc)()>
  void
  interpolate(ParticleContainer<P>& a_particles,
              const EBAMRCellData&  a_meshData,
              const DepositionType  a_interpType,
              const bool            a_forceIrregNGP = false) const;
 
  Vector<RefCountedPtr<EBCoarseFineParticleMesh>>&
  getEBCoarseFineParticleMesh() const;
 
  const EBParticleMesh&
  getEBParticleMesh(const int a_lvl, const DataIndex& a_dit) const;
 
protected:
  bool m_isDefined;
 
  bool m_verbose;
 
  RealVect m_probLo;
 
  int m_ghost;
 
  int m_finestLevel;
 
  Vector<RefCountedPtr<EBLevelGrid>> m_eblgs;
 
  Vector<int> m_refRat;
 
  Vector<Real> m_dx;
 
  Vector<RefCountedPtr<LayoutData<EBParticleMesh>>> m_ebParticleMesh;
 
  Vector<RefCountedPtr<LayoutData<EBParticleMesh>>> m_ebParticleMeshFiCo;
 
  Vector<RefCountedPtr<LevelData<BaseFab<bool>>>> m_validCells;
 
  std::map<int, Vector<RefCountedPtr<LevelData<BaseFab<bool>>>>> m_outerHaloMasks;
 
  std::map<int, Vector<RefCountedPtr<LevelData<BaseFab<bool>>>>> m_transitionMasks;
 
  mutable Vector<RefCountedPtr<EBCoarseFineParticleMesh>> m_coarseFinePM;
 
  mutable Vector<Copier> m_levelCopiers;
 
  void
  defineLevelMotion();
 
  void
  defineCoarseFineMotion();
 
  void
  defineEBParticleMesh();
 
  void
  defineOuterHaloMasks();
 
  void
  defineTransitionMasks();
 
  int
  getTransitionMaskWidth(const DepositionType a_depositionType, const int a_refRat) const;
 
  template <class P, class Ret, Ret (P::*MemberFunc)() const>
  void
  depositInterp(EBAMRCellData&              a_meshData,
                const ParticleContainer<P>& a_particles,
                const DepositionType        a_depositionType,
                const bool                  a_forceIrregNGP = false);
 
  template <class P, class Ret, Ret (P::*MemberFunc)() const>
  void
  depositHalo(EBAMRCellData&              a_meshData,
              const ParticleContainer<P>& a_particles,
              const DepositionType        a_depositionType,
              const bool                  a_forceIrregNGP = false);
 
  template <class P, class Ret, Ret (P::*MemberFunc)() const>
  void
  depositHaloNGP(EBAMRCellData&              a_meshData,
                 const ParticleContainer<P>& a_particles,
                 const DepositionType        a_depositionType,
                 const bool                  a_forceIrregNGP = false);
 
  template <class P, class Ret, Ret (P::*MemberFunc)() const>
  void
  depositTransition(EBAMRCellData&              a_meshData,
                    const ParticleContainer<P>& a_particles,
                    const DepositionType        a_depositionType,
                    const bool                  a_forceIrregNGP = false);
 
  template <class P>
  void
  transferMaskParticlesTransition(ParticleContainer<P>& a_particles, const DepositionType a_depositionType) const;
};
 
#include <CD_NamespaceFooter.H>
 
#include <CD_EBAMRParticleMeshImplem.H>
 
#endif