diagnostics.hpp

Go to the documentation of this file.

#ifndef DIAGNOSTICS_HPP
#define DIAGNOSTICS_HPP

#include "push_diagnostic.hpp"
#include "heat_diagnostics.hpp"
#include "pseudo_inverse_diag.hpp"
#include "tracer_diag.hpp"
#include "loop_voltage_diag.hpp"
#include "particle_stream.hpp"
#include <mpi.h>

/* Contains all the diagnostics */
struct Diagnostics{
    PushDiagnostic<DeviceType> push_diag;
    HeatDiagnostics<DeviceType> heat_diag;
    PseudoInvDiagnostics pseudo_inv_diag;
    TracerDiagnostics tracer_diag;
    LoopVolDiagnostics loop_vol_diag;
    ParticleStream particle_stream;
    // Make these use Diagnostic class
    bool diag_1d_on;        
    int diag_1d_period;     

    bool diag_3d_on;        
    int diag_3d_period;     

    bool diag_poincare;     

    Diagnostics(){}

    Diagnostics(NLReader::NamelistReader& nlr){
        nlr.use_namelist("diag_param");
        diag_1d_on = nlr.get<bool>("diag_1d_on",true);
        diag_1d_period = nlr.get<int>("diag_1d_period",10);

        diag_3d_on = nlr.get<bool>("diag_3d_on",true);
        diag_3d_period = nlr.get<int>("diag_3d_period",diag_1d_period);

        nlr.use_namelist("sml_param");
        int special = nlr.get<int>("sml_special",0);
        diag_poincare = (special==4);
    }

    void open_streams(int step){
        if(pseudo_inv_diag.is_on){
            XGC_IO_Mode mode = (!pseudo_inv_diag.has_been_triggered(step) ? XGC_IO_Mode::Write : XGC_IO_Mode::Append);
            pseudo_inv_diag.open_stream("xgc.pseudo_inv.bp", mode);
        }
        if(loop_vol_diag.is_on){
            XGC_IO_Mode mode = (!loop_vol_diag.has_been_triggered(step) ? XGC_IO_Mode::Write : XGC_IO_Mode::Append);
#ifdef USE_MPI
            loop_vol_diag.open_stream("xgc.loop_vol.bp", mode, MPI_COMM_SELF);
#endif
        }
        if(tracer_diag.is_on){
            tracer_diag.open_stream("xgc.particle_stream.bp", XGC_IO_Mode::Write);
        }
    }

    void close_streams(){
        if(pseudo_inv_diag.is_on) pseudo_inv_diag.close_stream();
        if(loop_vol_diag.is_on) loop_vol_diag.close_stream();
        if(tracer_diag.is_on) tracer_diag.close_stream();
    }
};

#endif