bicub.hpp

Go to the documentation of this file.

#ifndef BICUB_HPP
#define BICUB_HPP
#include "space_settings.hpp"
#include "simd.hpp"
#include "my_mirror_view.hpp"

// The dimensions of this were (0:ndeg, 0:ndeg) - hardcoded to (4,4) here
struct BicubCoeff {
    double coeff[4][4];
};

// Bicubic spline class
template<class Device>
class Bicub {
    // 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 Bicub;

    // bicub variables
    int nr, nz;            
    double rmin, zmin;     
    double dr_inv, dz_inv; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> rc; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> zc; 
    Kokkos::View<BicubCoeff**,Kokkos::LayoutRight,Device> acoeff_all;  

    // Compute value
    KOKKOS_INLINE_FUNCTION void eval_bicub_0(double x, double y, double xc, double yc, const double (&acoeff)[4][4],
                                         double &f00) const;

    // Compute first derivatives
    KOKKOS_INLINE_FUNCTION void eval_bicub_1(double x, double y, double xc, double yc, const double (&acoeff)[4][4],
                                         double &f00, double &f10, double &f01) const;

    // Compute first and second derivatives
    KOKKOS_INLINE_FUNCTION void eval_bicub_2(double x, double y, double xc, double yc, const double (&acoeff)[4][4],
                                         double &f00, double &f10, double &f01,
                                         double &f11, double &f20, double &f02) const;

    public:

    // Constructor
    Bicub(double *rc_in, double *zc_in, BicubCoeff *acoeff_in, int nr_in, int nz_in,
                double rmin_in, double zmin_in, double dr_inv_in, double dz_inv_in);

    // Constructor from grid on r,z
    inline Bicub(const View<double*, HostType>& r, const View<double*, HostType>& z, const View<double**, CLayout, HostType>& val);

    // Create a mirror with a different device type
    template<class Device2>
    inline Bicub<Device2> mirror() const{
        Bicub<Device2> m;
        m.nr = nr;
        m.nz = nz;
        m.rmin = rmin;
        m.dr_inv = dr_inv;
        m.zmin = zmin;
        m.dz_inv = dz_inv;
        m.rc = my_mirror_view(rc, Device2());
        mirror_copy(m.rc, rc);
        m.zc = my_mirror_view(zc, Device2());
        mirror_copy(m.zc, zc);
        m.acoeff_all = my_mirror_view(acoeff_all, Device2());
        mirror_copy(m.acoeff_all, acoeff_all);

        return m;
    }

    // Default constructor
    Bicub(){}

    // Get values
    KOKKOS_INLINE_FUNCTION void der_zero(double x, double y, double &f00) const;

    // Get first derivatives
    KOKKOS_INLINE_FUNCTION void der_one(double x, double y, double &f00, double &f10, double &f01) const;

    // Get first and second derivatives
    KOKKOS_INLINE_FUNCTION void der_all(double x, double y, double &f00, double &f10, double &f01,
            double &f11, double &f20, double &f02) const;
};

#include "bicub.tpp"

#endif