Diagnostics.H

Go to the documentation of this file.

#ifndef WARPX_DIAGNOSTICS_H_
#define WARPX_DIAGNOSTICS_H_

#include "ParticleDiag/ParticleDiag.H"

#include "ComputeDiagFunctors/ComputeDiagFunctor_fwd.H"
#include "FlushFormats/FlushFormat_fwd.H"

#include <AMReX_IntVect.H>
#include <AMReX_REAL.H>
#include <AMReX_Vector.H>

#include <AMReX_BaseFwd.H>

#include <memory>
#include <string>
#include <vector>

class Diagnostics
{
public:
    Diagnostics (int i, std::string name);
    virtual ~Diagnostics () ;
    void ComputeAndPack ();
    virtual void Flush (int i_buffer) = 0;
    void InitData ();
    virtual void InitializeFieldFunctors (int lev) = 0;
    virtual bool DoComputeAndPack(int step, bool force_flush=false) = 0;
    virtual bool DoDump (int step, int i_buffer, bool force_flush=false) = 0;
    void NewIteration () {m_already_done = false;}
    void FilterComputePackFlush (int step, bool force_flush=false);
    bool DoDumpLastTimestep () const {return  m_dump_last_timestep;}

protected:
    bool BaseReadParameters ();
    void InitBaseData ();
    virtual void InitializeFieldBufferData (int i_buffer, int lev) = 0;
    virtual void DerivedInitData () {}
    virtual void InitializeParticleBuffer () = 0;
    virtual void PrepareFieldDataForOutput () {}
    virtual void MovingWindowAndGalileanDomainShift (int /*step*/) {}
    std::string m_diag_name;
    std::string m_file_prefix;
    int m_file_min_digits = 5;
    int m_diag_index;
    amrex::Vector< std::string > m_varnames;
    bool m_dump_last_timestep = true;
    std::string m_format = "plotfile";
    int m_already_done = false;
    std::unique_ptr<FlushFormat> m_flush_format;
    amrex::Vector< amrex::Vector< amrex::MultiFab > >  m_mf_output;

    amrex::Vector< amrex::Vector <amrex::Geometry> > m_geom_output;
    // a particle buffer here?
    int nlev; 
    int nmax_lev; 
    int nlev_output;
    std::vector< std::string > m_output_species_names;
    std::vector< std::string > m_all_species_names;
    amrex::Vector< ParticleDiag > m_output_species;
    amrex::Vector< amrex::Vector <std::unique_ptr<ComputeDiagFunctor > > > m_all_field_functors;
    amrex::IntVect m_crse_ratio = amrex::IntVect(1);
    amrex::Vector< amrex::Real> m_lo;
    amrex::Vector< amrex::Real> m_hi;
    int m_num_buffers;
    amrex::Vector<int> m_rho_per_species_index;
    virtual void TMP_ClearSpeciesDataForBTD() {}
};

#endif // WARPX_DIAGNOSTICS_H_