grid_stomms.hpp

Go to the documentation of this file.

#ifndef GRID_STOMMS_HPP
#define GRID_STOMMS_HPP
 
#include "grid_files.hpp"
 
#ifdef USE_STOMMS_GRID
 
#include "xgc_reader_interface.hpp"  // Mesh reader
 
class PlaneFilesFromStommsMesh: public PlaneFilesBase{
    public:
      PlaneFilesFromStommsMesh(const std::string& stomms_mesh_file, const MPI_Comm& comm);
 
      // Functions
      int get_nnodes() const override;
      View<RZPair*, HostType> get_x() const override;
      View<int*, HostType> get_rgn() const override;
 
      int get_ntriangles() const override;
      View<Vertex*, HostType> get_nodes() const override;
 
      View<int*, HostType> get_i_x() const override; 
 
      int get_nsurfaces() const override;
      int get_nsurfaces1() const override;
      int get_nsurfaces2() const override;
      int get_nsurfaces3a() const override;
      int get_nsurfaces3b() const override;
      View<int*, HostType> get_i_surf_separatrices() const override;
      View<int*, HostType> get_nnodes_on_surface() const override; 
      int get_sum_nnodes_on_surfaces() const override;
      int get_num_non_aligned() const override;
      View<int*, HostType> get_non_aligned_vert() const override;
      View<int*, HostType> get_non_aligned_nsurf() const override;
      View<int**, CLayout, HostType> get_non_aligned_surf_idx() const override;
      View<int**, CLayout, HostType> get_surf_idx() const override;
 
      int n_specified_xpts() const override;
      View<Equil::XPoint*, HostType> get_xpts() const override;
      RZPair get_axis() const override;
      RZPair get_reference_point() const override;
      View<double*, HostType> get_psiN() const override;
 
    private:
      // Data Members
      XgcMesh xgc_mesh;
      View<int*, HostType> axis;      
 
      // Functions
      void write_nodes_in_x_rgn_format(Omega_h::Mesh omegah_mesh, View<RZPair*, HostType>& x, View<int*, HostType>& rgn, const int& plane_num);
      void write_triangles_in_node_format(Omega_h::Mesh omegah_mesh,View<Vertex*, HostType>& nodes);
      int read_node_rgn(const int& node_id, const int& plane_num);
      std::map <CurveIdType, std::vector<MeshIdType>> get_nodes_at_curves(const int& plane_num);
};
 
#endif  // end of USE_STOMMS_GRID guard
 
#endif