RhoFunctor.H

Go to the documentation of this file.

#ifndef WARPX_RHOFUNCTOR_H_
#define WARPX_RHOFUNCTOR_H_

#include "ComputeDiagFunctor.H"

class
RhoFunctor final : public ComputeDiagFunctor
{

public:

    RhoFunctor (const int lev,
                const amrex::IntVect crse_ratio,
                const int species_index = -1,
                bool convertRZmodes2cartesian = true,
                const int ncomp = 1);

    virtual void operator() ( amrex::MultiFab& mf_dst, const int dcomp, const int /*i_buffer=0*/ ) const override;

private:

    // Level on which source MultiFab mf_src is defined in RZ geometry
    int const m_lev;

    // Species index to dump rho per species
    const int m_species_index;

    // Whether to average all modes into one component in RZ geometry
    bool m_convertRZmodes2cartesian;
};

#endif // WARPX_RHOFUNCTOR_H_