grid_files.hpp

Go to the documentation of this file.

#ifndef GRID_FILES_HPP
#define GRID_FILES_HPP

#ifdef USE_MPI
#include "my_mpi.hpp"
#endif
#include "file_reader.hpp"
#include "magnetic_field.hpp"
#include "grid_structs.hpp"
#include "space_settings.hpp"
#include "NamelistReader.hpp"

struct PlaneFiles{
    // These are the regions designated in the .node file
    enum FileRegion{
        Inside=0,
        OnWall
    };

    // Contents of .node
    int nnodes;
    View<RZPair*, HostType> x;
    View<int*, HostType> rgn;

    // Contents of .ele
    int ntriangles;
    View<Vertex*, HostType> nodes;

    // Contents of .flx.aif
    View<int*, HostType> i_x;

    int nsurfaces;
    int nsurfaces1, nsurfaces2, nsurfaces3a, nsurfaces3b;
    View<int*, HostType> i_surf_separatrices;
    View<int*, HostType> nnodes_on_surface;
    int sum_nnodes_on_surfaces;
    int num_non_aligned;
    View<int*, HostType> non_aligned_vert;
    View<int*, HostType> non_aligned_nsurf;
    View<int**, CLayout, HostType> non_aligned_surf_idx;
    View<int**, CLayout, HostType> surf_idx;

    View<double*, HostType> psiN; 

    void convert_to_x_rgn_format(const View<double**, CLayout, HostType>& tmp_nodes, View<RZPair*, HostType>& x, View<int*, HostType>& rgn, bool is_stellarator);

    void convert_to_node_format(const View<int**, CLayout, HostType>& tmp_ele, View<Vertex*, HostType>& nodes);

    View<int*, HostType> pack_scalars(int max_nodes_on_surface, int n_xpts, int n_separatrices) const;
    void unpack_scalars(const View<int*, HostType>& scalars, int& max_nodes_on_surface, int& n_xpts, int& n_separatrices);

    // Default constructor, to use to deallocate the file arrays after use
    PlaneFiles(){}

// Should remove this USE_MPI at some point
#ifdef USE_MPI

    // Copy data from a PlaneFiles object on a neighboring rank
    PlaneFiles(const MyMPI& my_mpi, const PlaneFiles& plane_files);

    // Read the files
    PlaneFiles(const std::string& node_file, const std::string& element_file, const std::string& flx_aif_file, bool is_stellarator, const MPI_Comm& comm);
#endif

    // Options: hex vs circular
    enum GridCreateOpt{
        Hexagonal=0,
        Circular=1
    };

    void construct_hex_grid(int nshells, double raxis, double zaxis, double rscale, double zscale);

    double node_to_node_dist(RZPair a, RZPair b);

    void construct_circular_grid(int nshells, double raxis, double zaxis, double rscale, double zscale);

    static int nshells_from_nnodes(int nnodes);

    PlaneFiles(int nshells, double raxis, double zaxis, double rscale, double zscale, GridCreateOpt grid_opt=Hexagonal);

    PlaneFiles(MagneticField<HostType>& magnetic_field, int nshells, GridCreateOpt grid_opt=Hexagonal);

    int n_specified_xpts() const;

    View<Equil::XPoint*, HostType> get_xpts() const;

    RZPair get_axis() const;

    RZPair get_reference_point() const;

    View<double*, HostType> get_psiN() const;
};

struct GridFiles{
    using GridCreateOpt = PlaneFiles::GridCreateOpt;
    static constexpr GridCreateOpt Hexagonal = PlaneFiles::Hexagonal;
    static constexpr GridCreateOpt Circular = PlaneFiles::Circular;

    std::vector<PlaneFiles> plane_files_vec;

    // For test grids
    GridFiles(int nshells, double raxis, double zaxis, double rscale, double zscale, GridCreateOpt grid_opt=Hexagonal);

    GridFiles(MagneticField<HostType>& magnetic_field, int nshells, GridCreateOpt grid_opt=Hexagonal);

    std::string ifile_int2str(int ifile) const;

#ifdef USE_MPI
    GridFiles(NLReader::NamelistReader& nlr, bool is_stellarator, const MyMPI& my_mpi);
#endif

    const PlaneFiles& lplane() const;

    const PlaneFiles& midplane() const;

    const PlaneFiles& rplane() const;

    int n_planes() const;

    void clear();

    static int nshells_from_nnodes(int nnodes);
};

#endif