magnetic_equil_files.hpp

Go to the documentation of this file.

#ifndef MAGNETIC_EQUIL_FILES_HPP
#define MAGNETIC_EQUIL_FILES_HPP

#include "space_settings.hpp"
#include "NamelistReader.hpp"
#include "globals.hpp"
#include "rz_bounds.hpp"
#include "file_reader.hpp"

struct MagneticEquilFiles{
    using Ptr = std::shared_ptr<MagneticEquilFiles>;

    virtual ~MagneticEquilFiles() = default;

    static Ptr eq_files_ptr_from_namelist(NLReader::NamelistReader& nlr, bool is_3D);
};

struct MagneticEquilFiles3D : public MagneticEquilFiles{
    RZBounds bounds;
    View<double*, HostType> rgrid;
    View<double*, HostType> zgrid;
    View<double*, HostType> phigrid;
    View<double***, CLayout, HostType> psi;
    View<double***, CLayout, HostType> Br;
    View<double***, CLayout, HostType> Bz;
    View<double***, CLayout, HostType> Bphi;

    void read_eq_dims(FileReader& file_reader, int& mr, int& mz, int& mphi, int& wedge_n);

    void read_eq(FileReader& file_reader, int iphi);

    MagneticEquilFiles3D(const std::string& eq_filebase);

    static MagneticEquilFiles3D eq_files_from_namelist(NLReader::NamelistReader& nlr);
};

struct MagneticEquilFiles2D : public MagneticEquilFiles{
    RZBounds bounds;
    double axis_r;
    double axis_z;
    double axis_b;
    double x_psi;
    double x_r;
    double x_z;
    View<double*, CLayout, HostType> rgrid;
    View<double*, CLayout, HostType> zgrid;
    View<double*, CLayout, HostType> psi_grid;
    View<double*, CLayout, HostType> I;
    View<double**, CLayout, HostType> psi_rz;

    void read_eq_dims(FileReader& file_reader, int& mr, int& mz, int& mpsi);

    void read_eq(FileReader& file_reader);

    MagneticEquilFiles2D(bool read_m3dc1, const std::string& eq_filename);

    MagneticEquilFiles2D(NLReader::NamelistReader& nlr);

    static MagneticEquilFiles2D eq_files_from_namelist(NLReader::NamelistReader& nlr);
};

#endif