Chris@16
|
1 // Copyright John Maddock 2012.
|
Chris@16
|
2 // Use, modification and distribution are subject to the
|
Chris@16
|
3 // Boost Software License, Version 1.0.
|
Chris@16
|
4 // (See accompanying file LICENSE_1_0.txt
|
Chris@16
|
5 // or copy at http://www.boost.org/LICENSE_1_0.txt)
|
Chris@16
|
6
|
Chris@16
|
7 #ifndef BOOST_MATH_HANKEL_HPP
|
Chris@16
|
8 #define BOOST_MATH_HANKEL_HPP
|
Chris@16
|
9
|
Chris@101
|
10 #include <boost/math/special_functions/math_fwd.hpp>
|
Chris@16
|
11 #include <boost/math/special_functions/bessel.hpp>
|
Chris@16
|
12
|
Chris@16
|
13 namespace boost{ namespace math{
|
Chris@16
|
14
|
Chris@16
|
15 namespace detail{
|
Chris@16
|
16
|
Chris@16
|
17 template <class T, class Policy>
|
Chris@16
|
18 std::complex<T> hankel_imp(T v, T x, const bessel_no_int_tag&, const Policy& pol, int sign)
|
Chris@16
|
19 {
|
Chris@16
|
20 BOOST_MATH_STD_USING
|
Chris@16
|
21 static const char* function = "boost::math::cyl_hankel_1<%1%>(%1%,%1%)";
|
Chris@16
|
22
|
Chris@16
|
23 if(x < 0)
|
Chris@16
|
24 {
|
Chris@16
|
25 bool isint_v = floor(v) == v;
|
Chris@16
|
26 T j, y;
|
Chris@16
|
27 bessel_jy(v, -x, &j, &y, need_j | need_y, pol);
|
Chris@16
|
28 std::complex<T> cx(x), cv(v);
|
Chris@16
|
29 std::complex<T> j_result, y_result;
|
Chris@16
|
30 if(isint_v)
|
Chris@16
|
31 {
|
Chris@16
|
32 int s = (iround(v) & 1) ? -1 : 1;
|
Chris@16
|
33 j_result = j * s;
|
Chris@16
|
34 y_result = T(s) * (y - (2 / constants::pi<T>()) * (log(-x) - log(cx)) * j);
|
Chris@16
|
35 }
|
Chris@16
|
36 else
|
Chris@16
|
37 {
|
Chris@16
|
38 j_result = pow(cx, v) * pow(-cx, -v) * j;
|
Chris@16
|
39 T p1 = pow(-x, v);
|
Chris@16
|
40 std::complex<T> p2 = pow(cx, v);
|
Chris@16
|
41 y_result = p1 * y / p2
|
Chris@16
|
42 + (p2 / p1 - p1 / p2) * j / tan(constants::pi<T>() * v);
|
Chris@16
|
43 }
|
Chris@16
|
44 // multiply y_result by i:
|
Chris@16
|
45 y_result = std::complex<T>(-sign * y_result.imag(), sign * y_result.real());
|
Chris@16
|
46 return j_result + y_result;
|
Chris@16
|
47 }
|
Chris@16
|
48
|
Chris@16
|
49 if(x == 0)
|
Chris@16
|
50 {
|
Chris@16
|
51 if(v == 0)
|
Chris@16
|
52 {
|
Chris@16
|
53 // J is 1, Y is -INF
|
Chris@16
|
54 return std::complex<T>(1, sign * -policies::raise_overflow_error<T>(function, 0, pol));
|
Chris@16
|
55 }
|
Chris@16
|
56 else
|
Chris@16
|
57 {
|
Chris@16
|
58 // At least one of J and Y is complex infinity:
|
Chris@16
|
59 return std::complex<T>(policies::raise_overflow_error<T>(function, 0, pol), sign * policies::raise_overflow_error<T>(function, 0, pol));
|
Chris@16
|
60 }
|
Chris@16
|
61 }
|
Chris@16
|
62
|
Chris@16
|
63 T j, y;
|
Chris@16
|
64 bessel_jy(v, x, &j, &y, need_j | need_y, pol);
|
Chris@16
|
65 return std::complex<T>(j, sign * y);
|
Chris@16
|
66 }
|
Chris@16
|
67
|
Chris@16
|
68 template <class T, class Policy>
|
Chris@16
|
69 std::complex<T> hankel_imp(int v, T x, const bessel_int_tag&, const Policy& pol, int sign);
|
Chris@16
|
70
|
Chris@16
|
71 template <class T, class Policy>
|
Chris@16
|
72 inline std::complex<T> hankel_imp(T v, T x, const bessel_maybe_int_tag&, const Policy& pol, int sign)
|
Chris@16
|
73 {
|
Chris@16
|
74 BOOST_MATH_STD_USING // ADL of std names.
|
Chris@16
|
75 int ival = detail::iconv(v, pol);
|
Chris@16
|
76 if(0 == v - ival)
|
Chris@16
|
77 {
|
Chris@16
|
78 return hankel_imp(ival, x, bessel_int_tag(), pol, sign);
|
Chris@16
|
79 }
|
Chris@16
|
80 return hankel_imp(v, x, bessel_no_int_tag(), pol, sign);
|
Chris@16
|
81 }
|
Chris@16
|
82
|
Chris@16
|
83 template <class T, class Policy>
|
Chris@16
|
84 inline std::complex<T> hankel_imp(int v, T x, const bessel_int_tag&, const Policy& pol, int sign)
|
Chris@16
|
85 {
|
Chris@16
|
86 BOOST_MATH_STD_USING
|
Chris@16
|
87 if((std::abs(v) < 200) && (x > 0))
|
Chris@16
|
88 return std::complex<T>(bessel_jn(v, x, pol), sign * bessel_yn(v, x, pol));
|
Chris@16
|
89 return hankel_imp(static_cast<T>(v), x, bessel_no_int_tag(), pol, sign);
|
Chris@16
|
90 }
|
Chris@16
|
91
|
Chris@16
|
92 template <class T, class Policy>
|
Chris@16
|
93 inline std::complex<T> sph_hankel_imp(T v, T x, const Policy& pol, int sign)
|
Chris@16
|
94 {
|
Chris@16
|
95 BOOST_MATH_STD_USING
|
Chris@16
|
96 return constants::root_half_pi<T>() * hankel_imp(v + 0.5f, x, bessel_no_int_tag(), pol, sign) / sqrt(std::complex<T>(x));
|
Chris@16
|
97 }
|
Chris@16
|
98
|
Chris@16
|
99 } // namespace detail
|
Chris@16
|
100
|
Chris@16
|
101 template <class T1, class T2, class Policy>
|
Chris@16
|
102 inline std::complex<typename detail::bessel_traits<T1, T2, Policy>::result_type> cyl_hankel_1(T1 v, T2 x, const Policy& pol)
|
Chris@16
|
103 {
|
Chris@16
|
104 BOOST_FPU_EXCEPTION_GUARD
|
Chris@16
|
105 typedef typename detail::bessel_traits<T1, T2, Policy>::result_type result_type;
|
Chris@16
|
106 typedef typename detail::bessel_traits<T1, T2, Policy>::optimisation_tag tag_type;
|
Chris@16
|
107 typedef typename policies::evaluation<result_type, Policy>::type value_type;
|
Chris@16
|
108 return policies::checked_narrowing_cast<std::complex<result_type>, Policy>(detail::hankel_imp<value_type>(v, static_cast<value_type>(x), tag_type(), pol, 1), "boost::math::cyl_hankel_1<%1%>(%1%,%1%)");
|
Chris@16
|
109 }
|
Chris@16
|
110
|
Chris@16
|
111 template <class T1, class T2>
|
Chris@16
|
112 inline std::complex<typename detail::bessel_traits<T1, T2, policies::policy<> >::result_type> cyl_hankel_1(T1 v, T2 x)
|
Chris@16
|
113 {
|
Chris@16
|
114 return cyl_hankel_1(v, x, policies::policy<>());
|
Chris@16
|
115 }
|
Chris@16
|
116
|
Chris@16
|
117 template <class T1, class T2, class Policy>
|
Chris@16
|
118 inline std::complex<typename detail::bessel_traits<T1, T2, Policy>::result_type> cyl_hankel_2(T1 v, T2 x, const Policy& pol)
|
Chris@16
|
119 {
|
Chris@16
|
120 BOOST_FPU_EXCEPTION_GUARD
|
Chris@16
|
121 typedef typename detail::bessel_traits<T1, T2, Policy>::result_type result_type;
|
Chris@16
|
122 typedef typename detail::bessel_traits<T1, T2, Policy>::optimisation_tag tag_type;
|
Chris@16
|
123 typedef typename policies::evaluation<result_type, Policy>::type value_type;
|
Chris@16
|
124 return policies::checked_narrowing_cast<std::complex<result_type>, Policy>(detail::hankel_imp<value_type>(v, static_cast<value_type>(x), tag_type(), pol, -1), "boost::math::cyl_hankel_1<%1%>(%1%,%1%)");
|
Chris@16
|
125 }
|
Chris@16
|
126
|
Chris@16
|
127 template <class T1, class T2>
|
Chris@16
|
128 inline std::complex<typename detail::bessel_traits<T1, T2, policies::policy<> >::result_type> cyl_hankel_2(T1 v, T2 x)
|
Chris@16
|
129 {
|
Chris@16
|
130 return cyl_hankel_2(v, x, policies::policy<>());
|
Chris@16
|
131 }
|
Chris@16
|
132
|
Chris@16
|
133 template <class T1, class T2, class Policy>
|
Chris@16
|
134 inline std::complex<typename detail::bessel_traits<T1, T2, Policy>::result_type> sph_hankel_1(T1 v, T2 x, const Policy&)
|
Chris@16
|
135 {
|
Chris@16
|
136 BOOST_FPU_EXCEPTION_GUARD
|
Chris@16
|
137 typedef typename detail::bessel_traits<T1, T2, Policy>::result_type result_type;
|
Chris@16
|
138 typedef typename policies::evaluation<result_type, Policy>::type value_type;
|
Chris@16
|
139 typedef typename policies::normalise<
|
Chris@16
|
140 Policy,
|
Chris@16
|
141 policies::promote_float<false>,
|
Chris@16
|
142 policies::promote_double<false>,
|
Chris@16
|
143 policies::discrete_quantile<>,
|
Chris@16
|
144 policies::assert_undefined<> >::type forwarding_policy;
|
Chris@16
|
145
|
Chris@16
|
146 return policies::checked_narrowing_cast<std::complex<result_type>, Policy>(detail::sph_hankel_imp<value_type>(static_cast<value_type>(v), static_cast<value_type>(x), forwarding_policy(), 1), "boost::math::sph_hankel_1<%1%>(%1%,%1%)");
|
Chris@16
|
147 }
|
Chris@16
|
148
|
Chris@16
|
149 template <class T1, class T2>
|
Chris@16
|
150 inline std::complex<typename detail::bessel_traits<T1, T2, policies::policy<> >::result_type> sph_hankel_1(T1 v, T2 x)
|
Chris@16
|
151 {
|
Chris@16
|
152 return sph_hankel_1(v, x, policies::policy<>());
|
Chris@16
|
153 }
|
Chris@16
|
154
|
Chris@16
|
155 template <class T1, class T2, class Policy>
|
Chris@16
|
156 inline std::complex<typename detail::bessel_traits<T1, T2, Policy>::result_type> sph_hankel_2(T1 v, T2 x, const Policy&)
|
Chris@16
|
157 {
|
Chris@16
|
158 BOOST_FPU_EXCEPTION_GUARD
|
Chris@16
|
159 typedef typename detail::bessel_traits<T1, T2, Policy>::result_type result_type;
|
Chris@16
|
160 typedef typename policies::evaluation<result_type, Policy>::type value_type;
|
Chris@16
|
161 typedef typename policies::normalise<
|
Chris@16
|
162 Policy,
|
Chris@16
|
163 policies::promote_float<false>,
|
Chris@16
|
164 policies::promote_double<false>,
|
Chris@16
|
165 policies::discrete_quantile<>,
|
Chris@16
|
166 policies::assert_undefined<> >::type forwarding_policy;
|
Chris@16
|
167
|
Chris@16
|
168 return policies::checked_narrowing_cast<std::complex<result_type>, Policy>(detail::sph_hankel_imp<value_type>(static_cast<value_type>(v), static_cast<value_type>(x), forwarding_policy(), -1), "boost::math::sph_hankel_1<%1%>(%1%,%1%)");
|
Chris@16
|
169 }
|
Chris@16
|
170
|
Chris@16
|
171 template <class T1, class T2>
|
Chris@16
|
172 inline std::complex<typename detail::bessel_traits<T1, T2, policies::policy<> >::result_type> sph_hankel_2(T1 v, T2 x)
|
Chris@16
|
173 {
|
Chris@16
|
174 return sph_hankel_2(v, x, policies::policy<>());
|
Chris@16
|
175 }
|
Chris@16
|
176
|
Chris@16
|
177 }} // namespaces
|
Chris@16
|
178
|
Chris@16
|
179 #endif // BOOST_MATH_HANKEL_HPP
|
Chris@16
|
180
|