bspline.h

// -*-c++-*-
#ifndef STORMM_BSPLINE_H
#define STORMM_BSPLINE_H
 
#include "copyright.h"
#include "Accelerator/hybrid.h"
#include "DataTypes/common_types.h"
 
namespace stormm {
namespace stmath {
 
using card::Hybrid;
using data_types::isSignedIntegralScalarType;

template <typename Tcoord, typename Tcalc>
void bSplineInputChecks(const std::vector<Tcoord> &xcrd, const std::vector<Tcoord> &ycrd,
            const std::vector<Tcoord> &zcrd, const int order,
            std::vector<Tcalc> *a_coefs, std::vector<Tcalc> *b_coefs,
            std::vector<Tcalc> *c_coefs, std::vector<int> *a_init,
            std::vector<int> *b_init, std::vector<int> *c_init);
  
template <typename T> void bSpline(T x, int order, T* coefs, T* dcoefs = nullptr);
 
template <typename T> std::vector<T> bSpline(T x, int order);
  
template <typename Tcoord, typename Tcalc>
void bSpline(const Tcoord* xcrd, const Tcoord* ycrd, const Tcoord* zcrd, int order,
             const Tcalc* umat_cell, const Tcoord* invu_cell, int mesh_na, int mesh_nb,
             int mesh_nc, const Tcoord* invu_mesh, Tcalc* a_coefs, Tcalc* b_coefs, Tcalc* c_coefs,
             int* a_init, int* b_init, int* c_init, Tcalc* da_coefs = nullptr,
             Tcalc* db_coefs = nullptr, Tcalc* dc_coefs = nullptr, Tcalc coordinate_scale = 1.0);
 
template <typename Tcoord, typename Tcalc>
void bSpline(const std::vector<Tcoord> &xcrd, const std::vector<Tcoord> &ycrd,
             const std::vector<Tcoord> &zcrd, int order, const Tcalc* umat_cell,
             const Tcoord* invu_cell, int mesh_na, int mesh_nb, int mesh_nc,
             const Tcoord* invu_mesh, std::vector<Tcalc> *a_coefs, std::vector<Tcalc> *b_coefs,
             std::vector<Tcalc> *c_coefs, std::vector<int> *a_init, std::vector<int> *b_init,
             std::vector<int> *c_init, std::vector<Tcalc> *da_coefs = nullptr,
             std::vector<Tcalc> *db_coefs = nullptr, std::vector<Tcalc> *dc_coefs = nullptr,
             Tcalc coordinate_scale = 1.0);
 
template <typename Tcoord, typename Tcalc>
void bSpline(const Hybrid<Tcoord> &xcrd, const Hybrid<Tcoord> &ycrd, const Hybrid<Tcoord> &zcrd,
             int order, const Tcalc* umat_cell, const Tcoord* invu_cell, int mesh_na, int mesh_nb,
             int mesh_nc, const Tcoord* invu_mesh, Hybrid<Tcalc> *a_coefs, Hybrid<Tcalc> *b_coefs,
             Hybrid<Tcalc> *c_coefs, Hybrid<int> *a_init, Hybrid<int> *b_init,
             Hybrid<int> *c_init, Hybrid<Tcalc> *da_coefs = nullptr,
             Hybrid<Tcalc> *db_coefs = nullptr, Hybrid<Tcalc> *dc_coefs = nullptr,
             Tcalc coordinate_scale = 1.0);

template <typename T> std::vector<T> bSplineNoUnity(T x, int order);

template <typename T> std::vector<T> dBSpline(T x, int order, bool exploit_partition = true);
 
} // namespace stmath
} // namespace stormm
 
#include "bspline.tpp"
 
#endif