volumes_and_areas.hpp

Go to the documentation of this file.

#ifndef VOLUMES_AND_AREAS_HPP
#define VOLUMES_AND_AREAS_HPP

#include "globals.hpp"

extern "C" void init_fortran_vols_and_areas(double* tr_area_h, double* tr_vol_h,
#ifdef XGC1
                    double* node_vol_ff_h,
#endif
                    double* node_vol_nearest_h, double* node_vol_h, double* node_area_h, double* psi00_h, int npsi00);


// Volume- and Area-related Host Views. Should find a different location for these so that these unused views aren't sent to GPU
struct VolumesAndAreas{
    View<double*,CLayout,HostType> tr_area_h;
    View<double*,CLayout,HostType> tr_vol_h;
    View<double**,CLayout,HostType> node_vol_ff_h; // Default to midplane, for now
    View<double**,CLayout,HostType> node_vol_ff_l_h; // Field following volume for left plane nodes
    View<double**,CLayout,HostType> node_vol_ff_r_h; // Field following volume for right plane nnodes
    View<double*,CLayout,HostType> node_vol_nearest_h;
    View<double*,CLayout,HostType> node_vol_h;
    View<double*,CLayout,HostType> node_area_h;
    View<double*,CLayout,HostType> vol00_h;

    VolumesAndAreas(){}

    VolumesAndAreas(int ntriangle_in, int nnode_in, int npsi00_in)
      : tr_area_h(NoInit("tr_area"), ntriangle_in),
        tr_vol_h(NoInit("tr_vol"), ntriangle_in),
#ifdef XGC1
        node_vol_ff_h(NoInit("node_vol_ff"),2, nnode_in),
#ifdef STELLARATOR
        node_vol_ff_l_h(NoInit("node_vol_ff_l"),2, nnode_in),
        node_vol_ff_r_h(NoInit("node_vol_ff_r"),2, nnode_in),
#else
        node_vol_ff_l_h(node_vol_ff_h), // Shallow copy of midplane node_vol_ff
        node_vol_ff_r_h(node_vol_ff_h),
#endif
#endif
        node_vol_nearest_h(NoInit("node_vol_nearest"),nnode_in),
        node_vol_h(NoInit("node_vol"),nnode_in),
        node_area_h(NoInit("node_area"),nnode_in),
        vol00_h(NoInit("vol00_h"), npsi00_in)
    {}

    void set_fortran_ptrs(){
#ifndef NO_FORTRAN_MODULES
        int npsi00 = vol00_h.size();
        init_fortran_vols_and_areas(tr_area_h.data(), tr_vol_h.data(),
#ifdef XGC1
                node_vol_ff_h.data(),
#endif
                node_vol_nearest_h.data(), node_vol_h.data(), node_area_h.data(),
                vol00_h.data(), npsi00);
#endif
    }
};

#endif