FlushFormatSensei.H

Go to the documentation of this file.

#ifndef WARPX_FLUSHFORMATSENSEI_H_
#define WARPX_FLUSHFORMATSENSEI_H_
 
#include "FlushFormat.H"
 
#include <AMReX_AmrMesh.H>
#if defined(AMREX_USE_SENSEI_INSITU)
#  include <AMReX_AmrMeshParticleInSituBridge.H>
#else
namespace amrex {
using AmrMeshParticleInSituBridge = void;
}
#endif

class FlushFormatSensei : public FlushFormat
{
public:
    FlushFormatSensei();
    ~FlushFormatSensei() override;
 
    FlushFormatSensei(const FlushFormatSensei &)            = delete;
    FlushFormatSensei& operator=(const FlushFormatSensei &) = delete;
    FlushFormatSensei(FlushFormatSensei&&)                  = default;
    FlushFormatSensei& operator=(FlushFormatSensei&& )      = default;

    FlushFormatSensei (amrex::AmrMesh *amr_mesh, const std::string& diag_name);

    void WriteToFile (
        const amrex::Vector<std::string>& varnames,
        const amrex::Vector<amrex::MultiFab>& mf,
        amrex::Vector<amrex::Geometry>& geom,
        amrex::Vector<int> iteration, double time,
        const amrex::Vector<ParticleDiag>& particle_diags, int nlev,
        std::string prefix, int file_min_digits,
        bool plot_raw_fields,
        bool plot_raw_fields_guards,
        int verbose = 2,
        bool use_pinned_pc = false,
        bool isBTD = false, int snapshotID = -1,
        int bufferID = 1, int numBuffers = 1,
        const amrex::Geometry& full_BTD_snapshot = amrex::Geometry(),
        bool isLastBTDFlush = false) const override;

    void WriteParticles(const amrex::Vector<ParticleDiag>& particle_diags) const;
 
private:
    std::string m_insitu_config;
    int m_insitu_pin_mesh = 0;
    amrex::AmrMeshParticleInSituBridge * m_insitu_bridge = nullptr;
    amrex::AmrMesh * m_amr_mesh = nullptr;
};
 
#endif // WARPX_FLUSHFORMATSENSEI_H_