NumCpp  2.12.1
A Templatized Header Only C++ Implementation of the Python NumPy Library
comp_ellint_3.hpp
Go to the documentation of this file.
1
28#pragma once
29
30#include <cmath>
31
32#if defined(__cpp_lib_math_special_functions) || !defined(NUMCPP_NO_USE_BOOST)
33
37#include "NumCpp/NdArray.hpp"
38
39#ifndef __cpp_lib_math_special_functions
40#include "boost/math/special_functions/ellint_3.hpp"
41#endif
42
43#include <type_traits>
44
45namespace nc::special
46{
47 //============================================================================
48 // Method Description:
57 template<typename dtype1, typename dtype2>
58 auto comp_ellint_3(dtype1 inK, dtype2 inV)
59 {
62
63#ifdef __cpp_lib_math_special_functions
64 return std::comp_ellint_3(inK, inV);
65#else
66 return boost::math::ellint_3(inK, inV);
67#endif
68 }
69
70 //============================================================================
71 // Method Description:
80 template<typename dtype1, typename dtype2>
81 auto comp_ellint_3(const NdArray<dtype1>& inArrayK, const NdArray<dtype2>& inArrayV)
82 {
83 if (inArrayK.size() != inArrayV.size())
84 {
85 THROW_INVALID_ARGUMENT_ERROR("Shapes of inArrayk and inArrayV must match.");
86 }
87
88 NdArray<decltype(comp_ellint_3(dtype1{ 0 }, dtype2{ 0 }))> returnArray(inArrayK.shape());
89
91 inArrayK.cend(),
92 inArrayV.cbegin(),
93 returnArray.begin(),
94 [](dtype1 inK, dtype2 inV) -> auto { return comp_ellint_3(inK, inV); });
95
96 return returnArray;
97 }
98} // namespace nc::special
99
100#endif // #if defined(__cpp_lib_math_special_functions) || !defined(NUMCPP_NO_USE_BOOST)
#define THROW_INVALID_ARGUMENT_ERROR(msg)
Definition: Error.hpp:37
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:39
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:139
size_type size() const noexcept
Definition: NdArrayCore.hpp:4524
const_iterator cbegin() const noexcept
Definition: NdArrayCore.hpp:1365
const Shape & shape() const noexcept
Definition: NdArrayCore.hpp:4511
const_iterator cend() const noexcept
Definition: NdArrayCore.hpp:1673
Definition: airy_ai.hpp:39
auto comp_ellint_3(const NdArray< dtype1 > &inArrayK, const NdArray< dtype2 > &inArrayV)
Definition: comp_ellint_3.hpp:81
auto comp_ellint_3(dtype1 inK, dtype2 inV)
Definition: comp_ellint_3.hpp:58
auto ellint_3(dtype1 inK, dtype2 inV, dtype3 inP)
Definition: ellint_3.hpp:59
OutputIt transform(InputIt first, InputIt last, OutputIt destination, UnaryOperation unaryFunction)
Definition: StlAlgorithms.hpp:775