AMReX_EdgeFluxRegister.H

Go to the documentation of this file.

#ifndef AMREX_EDGE_FLUX_REGISTER_H_
#define AMREX_EDGE_FLUX_REGISTER_H_
#include <AMReX_Config.H>
 
#include <AMReX_iMultiFab.H>
#include <AMReX_LayoutData.H>
#include <AMReX_MultiFab.H>
 
namespace amrex {
 
class EdgeFluxRegister
{
public:
 
    EdgeFluxRegister () = default;
 
    EdgeFluxRegister (const BoxArray& fba, const BoxArray& cba,
                      const DistributionMapping& fdm, const DistributionMapping& cdm,
                      const Geometry& fgeom, const Geometry& cgeom,
                      int nvar = 1);
 
    void define (const BoxArray& fba, const BoxArray& cba,
                 const DistributionMapping& fdm, const DistributionMapping& cdm,
                 const Geometry& fgeom, const Geometry& cgeom,
                 int nvar = 1);
 
    void reset ();
 
    EdgeFluxRegister& plus (EdgeFluxRegister const& rhs);
 
#if (AMREX_SPACEDIM == 3)
 
    void CrseAdd (MFIter const& mfi, const Array<FArrayBox const*,3>& E_crse, Real dt_crse);
 
    void FineAdd (MFIter const& mfi, const Array<FArrayBox const*,3>& E_fine, Real dt_fine);
 
#else /* 2D */
 
    void CrseAdd (MFIter const& mfi, FArrayBox const& E_crse, Real dt_crse);
 
    void FineAdd (MFIter const& mfi, FArrayBox const& E_fine, Real dt_fine);
 
#endif
 
    void Reflux (Array<MultiFab*,AMREX_SPACEDIM> const& B_crse) const;
 
private:
 
    Geometry m_fine_geom;
    Geometry m_crse_geom;
 
    IntVect m_ratio;
    int m_ncomp;
 
#if (AMREX_SPACEDIM == 3)
 
    Array<MultiFab,AMREX_SPACEDIM> m_E_crse; // on original grids
 
    // There are AMREX_SPACEDIM*2 faces. For each face, we need to store two
    // component. For example, at the x-faces, we need to store Ey and Ez.
    Array<Array<MultiFab,2>,AMREX_SPACEDIM*2> m_E_fine;
 
    // Mask on the coarse level indicating overlap with m_E_fine
    Array<iMultiFab,AMREX_SPACEDIM> m_fine_mask;
 
#else
 
    MultiFab m_E_crse;
    Array<MultiFab,AMREX_SPACEDIM*2> m_E_fine;
    iMultiFab m_fine_mask;
 
#endif
 
    LayoutData<int> m_has_cf; // Flag on the coarse level indicating c/f interface
};
 
}
 
#endif