bicub.hpp

Go to the documentation of this file.

#ifndef BICUB_HPP
#define BICUB_HPP
#include "space_settings.hpp"
#include "simd.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 {

    // bicub variables
    int nr, nz;                
    double rmin, dr_inv, zmin, 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
    Bicub(int eq_mr, int eq_mz, double rmin_in, double zmin_in, double rmax_in, double zmax_in, Kokkos::View<double**, Kokkos::LayoutRight, HostType>& val);

    // 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