ComputeDiagFunctor.H

Go to the documentation of this file.

#ifndef WARPX_COMPUTEDIAGFUNCTOR_H_
#define WARPX_COMPUTEDIAGFUNCTOR_H_
 
#include "ComputeDiagFunctor_fwd.H"
#include "Utils/TextMsg.H"
 
#include <ablastr/coarsen/sample.H>
 
#include <AMReX.H>
#include <AMReX_MultiFab.H>

class ComputeDiagFunctor
{
public:
    ComputeDiagFunctor( int ncomp, amrex::IntVect crse_ratio) :
                        m_ncomp(ncomp), m_crse_ratio(crse_ratio) {}
    //** Virtual Destructor to handle clean destruction of derived classes */
    virtual ~ComputeDiagFunctor() = default;
 
    // Default move and copy operations
    ComputeDiagFunctor(const ComputeDiagFunctor&) = default;
    ComputeDiagFunctor& operator=(const ComputeDiagFunctor&) = default;
    ComputeDiagFunctor(ComputeDiagFunctor&&) = default;
    ComputeDiagFunctor& operator=(ComputeDiagFunctor&&) = default;

    virtual void operator() (amrex::MultiFab& mf_dst, int dcomp, int i_buffer = 0) const = 0;
    [[nodiscard]] int nComp () const { return m_ncomp; }

    virtual void PrepareFunctorData ( int i_buffer, bool z_slice_in_domain,
                                      amrex::Real current_z_boost,
                                      amrex::Box buffer_box, const int k_index_zlab,
                                      const int snapshot_full) {
                                          amrex::ignore_unused(i_buffer,
                                          z_slice_in_domain,
                                          current_z_boost, buffer_box,
                                          k_index_zlab, snapshot_full);
                                      }
    virtual void InitData() {}
 
    void InterpolateMFForDiag (
        amrex::MultiFab& mf_dst, const amrex::MultiFab& mf_src, int dcomp,
        const amrex::DistributionMapping& dm, bool convertRZmodes2cartesian ) const
    {
#ifdef WARPX_DIM_RZ
        if (convertRZmodes2cartesian) {
            // In cylindrical geometry, sum real part of all modes of mf_src in
            // temporary MultiFab mf_dst_stag, and cell-center it to mf_dst
            WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
                nComp()==1,
                "The RZ averaging over modes must write into one single component");
            amrex::MultiFab mf_dst_stag( mf_src.boxArray(), dm, 1, mf_src.nGrowVect() );
            // Mode 0
            amrex::MultiFab::Copy( mf_dst_stag, mf_src, 0, 0, 1, mf_src.nGrowVect() );
            for (int ic=1 ; ic < mf_src.nComp() ; ic += 2) {
                // Real part of all modes > 0
                amrex::MultiFab::Add( mf_dst_stag, mf_src, ic, 0, 1, mf_src.nGrowVect() );
            }
            ablastr::coarsen::sample::Coarsen( mf_dst, mf_dst_stag, dcomp, 0, nComp(), 0, m_crse_ratio );
        } else {
            ablastr::coarsen::sample::Coarsen( mf_dst, mf_src, dcomp, 0, nComp(), 0, m_crse_ratio );
        }
#else
        // In Cartesian geometry, coarsen and interpolate from temporary MultiFab mf_src
        // to output diagnostic MultiFab mf_dst
        ablastr::coarsen::sample::Coarsen(mf_dst, mf_src, dcomp, 0, nComp(), mf_dst.nGrowVect(), m_crse_ratio );
        amrex::ignore_unused(convertRZmodes2cartesian, dm);
#endif
    }
 
private:
    int m_ncomp;
 
protected:
    amrex::IntVect m_crse_ratio;
};
 
#endif // WARPX_COMPUTEDIAGFUNCTOR_H_