%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /proc/985914/root/lib/python3/dist-packages/pythran/pythonic/scipy/special/
Upload File :
Create Path :
Current File : //proc/985914/root/lib/python3/dist-packages/pythran/pythonic/scipy/special/spherical_jn.hpp

#ifndef PYTHONIC_SCIPY_SPECIAL_SPHERICAL_JN_HPP
#define PYTHONIC_SCIPY_SPECIAL_SPHERICAL_JN_HPP

#include "pythonic/include/scipy/special/spherical_jn.hpp"

#include "pythonic/types/ndarray.hpp"
#include "pythonic/utils/functor.hpp"
#include "pythonic/utils/numpy_traits.hpp"

#define BOOST_MATH_THREAD_LOCAL thread_local
#include <boost/math/special_functions/bessel.hpp>
#include <boost/math/special_functions/bessel_prime.hpp>

PYTHONIC_NS_BEGIN

namespace scipy
{
  namespace special
  {
    namespace details
    {
      template <class T0, class T1>
      double spherical_jn(T0 v, T1 x, bool derivative)
      {
        assert(v == (long)v &&
               "only supported for integral value as first arg");
        using namespace boost::math::policies;
        if (derivative) {
          return boost::math::sph_bessel_prime(
              v, x, make_policy(promote_double<true>()));
        } else {
          return boost::math::sph_bessel(v, x,
                                         make_policy(promote_double<true>()));
        }
      }
    }

#define NUMPY_NARY_FUNC_NAME spherical_jn
#define NUMPY_NARY_FUNC_SYM details::spherical_jn
#include "pythonic/types/numpy_nary_expr.hpp"
  }
}
PYTHONIC_NS_END

#endif

Zerion Mini Shell 1.0