Diagnostics.H

Go to the documentation of this file.

#ifndef WARPX_DIAGNOSTICS_H_
#define WARPX_DIAGNOSTICS_H_

#include "FlushFormats/FlushFormat.H"
#include "ComputeDiagFunctors/ComputeDiagFunctor.H"
#include "ParticleDiag/ParticleDiag.H"
#include "Utils/IntervalsParser.H"
#include <AMReX_Vector.H>
#include <AMReX_MultiFab.H>

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);

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 () {}
    std::string m_diag_name;
    std::string m_file_prefix;
    int m_diag_index;
    amrex::Vector< std::string > m_varnames;
    std::string m_format = "plotfile";
    int m_already_done = false;
    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_species_names;
    amrex::Vector< ParticleDiag > m_all_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;
};

#endif // WARPX_DIAGNOSTICS_H_