| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116 |
- // 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
|