BTDiagnostics.H

Go to the documentation of this file.

#ifndef WARPX_BTDIAGNOSTICS_H_
#define WARPX_BTDIAGNOSTICS_H_

#include "Diagnostics.H"

class
BTDiagnostics final : public Diagnostics
{
public:

    BTDiagnostics (int i, std::string name);

private:
    void ReadParameters ();
    void Flush (int i_buffer) override;
    bool DoDump (int step, int i_buffer, bool force_flush=false) override;
    bool DoComputeAndPack (int step, bool force_flush=false) override;
    void DerivedInitData () override;
    void InitializeFieldFunctors (int lev) override;
    void InitializeParticleBuffer () override {}
    void PrepareFieldDataForOutput () override;

    bool GetZSliceInDomainFlag (const int i_buffer, const int lev);

    void InitializeFieldBufferData ( int i_buffer , int lev) override;
    bool m_do_back_transformed_fields = true;
    bool m_do_back_transformed_particles = true;

    amrex::Real m_gamma_boost;
    amrex::Real m_beta_boost;
    int m_moving_window_dir;

    int m_num_snapshots_lab = std::numeric_limits<int>::lowest();
    amrex::Real m_dt_snapshots_lab = std::numeric_limits<amrex::Real>::lowest();
    amrex::Real m_dz_snapshots_lab = 0.0;

    int m_buffer_size = 256;
    int m_max_box_size = 256;

    amrex::Vector<amrex::Real> m_t_lab;
    amrex::Vector<amrex::RealBox> m_prob_domain_lab;
    amrex::Vector<amrex::RealBox> m_buffer_domain_lab;
    amrex::Vector<amrex::IntVect> m_buffer_ncells_lab;
    amrex::Vector<amrex::Box> m_buffer_box;
    amrex::Vector<amrex::Real> m_current_z_lab;
    amrex::Vector<amrex::Real> m_current_z_boost;
    amrex::Vector<int> m_buffer_counter;
    amrex::Vector<std::unique_ptr<amrex::MultiFab> > m_cell_centered_data;
    std::vector<std::string> m_file_name;
    amrex::Vector< amrex::Vector <std::unique_ptr<ComputeDiagFunctor const> > > m_cell_center_functors;

    void TMP_writeMetaData ();
    void TMP_createLabFrameDirectories (int i_buffer, int lev);
    void TMP_writeLabFrameHeader (int i_buffer);
    void TMP_FlushLabFrameData ( int i_buffer );
    void DefineCellCenteredMultiFab(int lev);
    void DefineFieldBufferMultiFab (const int i_buffer, const int lev);
    amrex::Real UpdateCurrentZBoostCoordinate(amrex::Real t_lab, amrex::Real t_boost)
    {
        amrex::Real current_z_boost = (t_lab / m_gamma_boost - t_boost) * PhysConst::c / m_beta_boost;
        return current_z_boost;
    }
    amrex::Real UpdateCurrentZLabCoordinate(amrex::Real t_lab, amrex::Real t_boost)
    {
        amrex::Real current_z_lab = (t_lab - t_boost / m_gamma_boost ) * PhysConst::c / m_beta_boost;
        return current_z_lab;
    }
    amrex::Real dz_lab (amrex::Real dt, amrex::Real ref_ratio);
    int k_index_zlab (int i_buffer, int lev);
    bool buffer_full (int i_buffer) {
        return ( m_buffer_counter[i_buffer] == m_buffer_size );
    }

    bool buffer_empty (int i_buffer) {
        return ( m_buffer_counter[i_buffer] == 0) ;
    }

    void ResetBufferCounter(int i_buffer) {
        m_buffer_counter[i_buffer] = 0;
    }
    amrex::Vector< std::string > m_cellcenter_varnames = {"Ex", "Ey", "Ez",
                                                          "Bx", "By", "Bz",
                                                          "jx", "jy", "jz", "rho"};

};

#endif // WARPX_BTDIAGNOSTICS_H_