cub_interp.hpp

Go to the documentation of this file.

#ifndef CUB_INTERP_HPP
#define CUB_INTERP_HPP
#include "space_settings.hpp"
#include "my_mirror_view.hpp"

// An array of this struct contains the background magnetic field used for interpolation. Size 4 since it is 3rd-order spline
struct OneDCoeff {
    double coeff[4];
};

template<class Device>
class CubInterp {
    // This line enables private access between the host and device implementations of this class
    // i.e. Makes the copy from host to device more straightforward
    template<class Device2> friend class CubInterp;

    // Variables needed in the interp
    int ncoeff;    
    double max_x;  
    double min_x;  
    double dx_inv; 

    Kokkos::View<OneDCoeff*,Kokkos::LayoutRight,Device> acoef; 

    public:

    // Constructor
    CubInterp(OneDCoeff *one_d_cub_acoef_in, int ncoeff_in, double max_psi_in, double min_psi_in, double one_d_cub_dpsi_inv_in);

    // Constructor from x and y
    inline CubInterp(const View<double*, HostType>& x, const View<double*, HostType>& y);

    // Create a mirror with a different device type
    template<class Device2>
    inline CubInterp<Device2> mirror() const{
        CubInterp<Device2> m;
        m.max_x = max_x;
        m.min_x = min_x;
        m.dx_inv = dx_inv;
        m.ncoeff = ncoeff;
        m.acoef = my_mirror_view(acoef, Device2());
        mirror_copy(m.acoef, acoef);

        return m;
    }

    // Default constructor
    CubInterp(){}

    KOKKOS_INLINE_FUNCTION double value(double x) const;
    KOKKOS_INLINE_FUNCTION double derivative(double x) const;
};

#include "cub_interp.tpp"

#endif