rbf_functions.hpp

Go to the documentation of this file.

//
// This file is subject to the terms and conditions defined in
// file 'LICENSE', which is part of this source code package.
//
 
#pragma once
 
#include <Kokkos_Core.hpp>
 
#include "common/types.hpp"
#include "utils/utils.hpp"
 
namespace PACMAN {
namespace RbfPum {

class WendlandC0 {
public:
  KOKKOS_FORCEINLINE_FUNCTION
  fp_t constexpr Eval(const fp_t &r) const {
    const fp_t p = r * mRadiusInv;
    char mask =
        static_cast<char>(r >= fp_consts::zero() && p < fp_consts::one() &&
                          p >= fp_consts::zero());
    return static_cast<fp_t>(mask) *
           ((fp_consts::one() - p) * (fp_consts::one() - p));
  }
  inline void SetRadiusInv(const fp_t &radiusInv) {
    this->mRadiusInv = radiusInv;
  }
 
private:
  fp_t mRadiusInv = fp_consts::zero();
};

class WendlandC2 {
public:
  KOKKOS_FORCEINLINE_FUNCTION
  fp_t Eval(const fp_t &r) const {
    const fp_t p = r * mRadiusInv;
    char mask =
        static_cast<char>(r >= fp_consts::zero() && p < fp_consts::one() &&
                          p >= fp_consts::zero());
    return static_cast<fp_t>(mask) *
           (Kokkos::pow(fp_consts::one() - p, 4) *
            Kokkos::fma(static_cast<fp_t>(4.0), p, fp_consts::one()));
  }
  inline void SetRadiusInv(const fp_t &radiusInv) {
    this->mRadiusInv = radiusInv;
  }
 
private:
  fp_t mRadiusInv = fp_consts::zero();
};

class WendlandC4 {
public:
  KOKKOS_FORCEINLINE_FUNCTION
  fp_t Eval(const fp_t &r) const {
    const fp_t p = r * mRadiusInv;
    char mask =
        static_cast<char>(r >= fp_consts::zero() && p < fp_consts::one() &&
                          p >= fp_consts::zero());
    return static_cast<fp_t>(mask) *
           (Kokkos::pow(fp_consts::one() - p, 6) *
            (static_cast<fp_t>(35.0) * Kokkos::pow(p, 2) +
             Kokkos::fma(static_cast<fp_t>(18.0), p, static_cast<fp_t>(3.0))));
  }
  inline void SetRadiusInv(const fp_t &radiusInv) {
    this->mRadiusInv = radiusInv;
  };
 
private:
  fp_t mRadiusInv = fp_consts::zero();
};

class WendlandC6 {
public:
  KOKKOS_FORCEINLINE_FUNCTION
  fp_t Eval(const fp_t &r) const {
    const fp_t p = r * mRadiusInv;
    char mask =
        static_cast<char>(r >= fp_consts::zero() && p < fp_consts::one() &&
                          p >= fp_consts::zero());
    return static_cast<fp_t>(mask) *
           (Kokkos::pow(fp_consts::one() - p, 8) *
            (static_cast<fp_t>(32.0) * Kokkos::pow(p, 3) +
             static_cast<fp_t>(25.0) * Kokkos::pow(p, 2) +
             Kokkos::fma(static_cast<fp_t>(8.0), p, fp_consts::one())));
  }
  inline void SetRadiusInv(const fp_t &radiusInv) {
    this->mRadiusInv = radiusInv;
  }
 
private:
  fp_t mRadiusInv = fp_consts::zero();
};

class WendlandC8 {
public:
  KOKKOS_FORCEINLINE_FUNCTION
  fp_t Eval(const fp_t &r) const {
    const fp_t p = r * mRadiusInv;
    char mask =
        static_cast<char>(r >= fp_consts::zero() && p < fp_consts::one() &&
                          p >= fp_consts::zero());
    return static_cast<fp_t>(mask) *
           (Kokkos::pow(fp_consts::one() - p, 10) *
            (static_cast<fp_t>(1287.0) * Kokkos::pow(p, 4) +
             static_cast<fp_t>(1350.0) * Kokkos::pow(p, 3) +
             static_cast<fp_t>(630.0) * Kokkos::pow(p, 2) +
             Kokkos::fma(static_cast<fp_t>(150.0), p,
                         static_cast<fp_t>(15.0))));
  }
  inline void SetRadiusInv(const fp_t &radiusInv) {
    this->mRadiusInv = radiusInv;
  }
 
private:
  fp_t mRadiusInv = fp_consts::zero();
};
 
} // namespace RbfPum
} // namespace PACMAN