rz_bounds.hpp

Go to the documentation of this file.

#ifndef RZ_BOUNDS_HPP
#define RZ_BOUNDS_HPP

struct RZBounds{
    double min_r;
    double max_r;
    double min_z;
    double max_z;

    KOKKOS_INLINE_FUNCTION RZBounds(){}

    RZBounds(double rmin, double rmax, double zmin, double zmax)
     : min_r(rmin),
       max_r(rmax),
       min_z(zmin),
       max_z(zmax)
    {}

    KOKKOS_INLINE_FUNCTION bool is_outside(double r, double z) const{
        return ( r<min_r || r>max_r ||
                 z<min_z || z>max_z    );
    }

    KOKKOS_INLINE_FUNCTION double get_r_width() const{
        return (max_r - min_r);
    }

    KOKKOS_INLINE_FUNCTION double get_z_width() const{
        return (max_z - min_z);
    }

    KOKKOS_INLINE_FUNCTION double bound_r(double r) const{
        return max(min(r,max_r),min_r);
    }

    KOKKOS_INLINE_FUNCTION double bound_z(double z) const{
        return max(min(z,max_z),min_z);
    }

    inline double area() const{
        return get_r_width()*get_z_width();
    }

    inline double dVolume_dphi() const{
        return area()*(min_r + 0.5*get_r_width());
    }
};

#endif