1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
// Copyright John Maddock 2018.
// Use, modification and distribution are subject to the
// Boost Software License, Version 1.0. (See accompanying file
// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// Tools for operator on complex as well as scalar types.
//
#ifndef BOOST_MATH_TOOLS_COMPLEX_HPP
#define BOOST_MATH_TOOLS_COMPLEX_HPP
#include <boost/math/tools/config.hpp>
#include <boost/math/tools/is_detected.hpp>
#ifdef BOOST_MATH_ENABLE_CUDA
#include <cuda/std/utility>
#include <cuda/std/complex>
namespace boost {
namespace math {
template <typename T>
using complex = cuda::std::complex<T>;
} // namespace math
} // namespace boost
#else
#include <utility>
#include <complex>
namespace boost {
namespace math {
template <typename T>
using complex = std::complex<T>;
} // namespace math
} // namespace boost
#endif
namespace boost {
namespace math {
namespace tools {
namespace detail {
template <typename T, typename = void>
struct is_complex_type_impl
{
static constexpr bool value = false;
};
#ifndef BOOST_MATH_ENABLE_CUDA
template <typename T>
struct is_complex_type_impl<T, void_t<decltype(std::declval<T>().real()),
decltype(std::declval<T>().imag())>>
{
static constexpr bool value = true;
};
#else
template <typename T>
struct is_complex_type_impl<T, void_t<decltype(cuda::std::declval<T>().real()),
decltype(cuda::std::declval<T>().imag())>>
{
static constexpr bool value = true;
};
#endif
} // Namespace detail
template <typename T>
struct is_complex_type : public detail::is_complex_type_impl<T> {};
//
// Use this trait to typecast integer literals to something
// that will interoperate with T:
//
template <class T, bool = is_complex_type<T>::value>
struct integer_scalar_type
{
typedef int type;
};
template <class T>
struct integer_scalar_type<T, true>
{
typedef typename T::value_type type;
};
template <class T, bool = is_complex_type<T>::value>
struct unsigned_scalar_type
{
typedef unsigned type;
};
template <class T>
struct unsigned_scalar_type<T, true>
{
typedef typename T::value_type type;
};
template <class T, bool = is_complex_type<T>::value>
struct scalar_type
{
typedef T type;
};
template <class T>
struct scalar_type<T, true>
{
typedef typename T::value_type type;
};
} } }
#endif // BOOST_MATH_TOOLS_COMPLEX_HPP
|
