AMReX_EBFluxRegister.H

Go to the documentation of this file.

#ifndef AMREX_EBFLUXREGISTER_H_
#define AMREX_EBFLUXREGISTER_H_
#include <AMReX_Config.H>
 
#include <AMReX_YAFluxRegister.H>
#include <AMReX_EBCellFlag.H>
 
extern "C" {
    void amrex_eb_disable_reredistribution ();
    amrex::Real amrex_eb_get_reredistribution_threshold ();
}
 
namespace amrex {
 
class EBFluxRegister
    : public YAFluxRegister
{
public:
 
    EBFluxRegister () = default;
 
    EBFluxRegister (const BoxArray& fba, const BoxArray& cba,
                    const DistributionMapping& fdm, const DistributionMapping& cdm,
                    const Geometry& fgeom, const Geometry& cgeom,
                    const IntVect& ref_ratio, int fine_lev, int nvar);
 
    void define (const BoxArray& fba, const BoxArray& cba,
                 const DistributionMapping& fdm, const DistributionMapping& cdm,
                 const Geometry& fgeom, const Geometry& cgeom,
                 const IntVect& ref_ratio, int fine_lev, int nvar);
 
    void clear ();
 
    using YAFluxRegister::CrseAdd;
    void CrseAdd (const MFIter& mfi,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& flux,
                  const Real* dx, Real dt,
                  const FArrayBox& volfrac,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& areafrac,
                  RunOn runon);
    void CrseAdd (const MFIter& mfi,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& flux,
                  const Real* dx, Real dt,
                  const FArrayBox& volfrac,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& areafrac,
                  int srccomp, int destcomp, int numcomp, RunOn runon);
 
    using YAFluxRegister::FineAdd;
    void FineAdd (const MFIter& mfi,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& flux,
                  const Real* dx, Real dt,
                  const FArrayBox& volfrac,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& areafrac,
                  const FArrayBox& dm,
                  RunOn runon);
    void FineAdd (const MFIter& mfi,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& flux,
                  const Real* dx, Real dt,
                  const FArrayBox& volfrac,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& areafrac,
                  const FArrayBox& dm,
                  int srccomp, int destcomp, int numcomp, RunOn runon);
    void FineAdd (const MFIter& mfi,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& flux,
                  const Real* dx, Real dt,
                  const FArrayBox& volfrac,
                  const std::array<FArrayBox const*, AMREX_SPACEDIM>& areafrac,
                  int srccomp, int destcomp, int numcomp, RunOn runon);
 
    void Reflux (MultiFab& crse_state, const amrex::MultiFab& crse_vfrac,
                 MultiFab& fine_state, const amrex::MultiFab& fine_vfrac);
    void Reflux (MultiFab& crse_state, const amrex::MultiFab& crse_vfrac,
                 MultiFab& fine_state, const amrex::MultiFab& fine_vfrac,
                 int srccomp, int destcomp, int numcomp);
    void Reflux (MultiFab& crse_state, const amrex::MultiFab& crse_vfrac,
                 int srccomp, int destcomp, int numcomp);
 
    FArrayBox* getCrseData (const MFIter& mfi) {
        return &(m_crse_data[mfi]);
    }
 
    const IArrayBox* getCrseFlag (const MFIter& mfi) const {
        return &(m_crse_flag[mfi]);
    }
 
private:
 
    iMultiFab m_cfp_inside_mask;
 
public: // for cuda
 
    void defineExtra (const BoxArray& fba, const DistributionMapping& fdm);
};
 
}
 
#endif