123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167 |
- ///////////////////////////////////////////////////////////////////////////////
- // Copyright 2018 John Maddock. Distributed under 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)
- #ifndef BOOST_MP_EIGEN_HPP
- #define BOOST_MP_EIGEN_HPP
- #include <boost/multiprecision/number.hpp>
- #include <Eigen/Core>
- //
- // Generic Eigen support code:
- //
- namespace Eigen {
- template <class Backend, boost::multiprecision::expression_template_option ExpressionTemplates>
- struct NumTraits<boost::multiprecision::number<Backend, ExpressionTemplates> >
- {
- using self_type = boost::multiprecision::number<Backend, ExpressionTemplates> ;
- using Real = typename boost::multiprecision::scalar_result_from_possible_complex<self_type>::type;
- using NonInteger = self_type ; // Not correct but we can't do much better??
- using Literal = double ;
- using Nested = self_type ;
- enum
- {
- IsComplex = boost::multiprecision::number_category<self_type>::value == boost::multiprecision::number_kind_complex,
- IsInteger = boost::multiprecision::number_category<self_type>::value == boost::multiprecision::number_kind_integer,
- ReadCost = 1,
- AddCost = 4,
- MulCost = 8,
- IsSigned = std::numeric_limits<self_type>::is_specialized ? std::numeric_limits<self_type>::is_signed : true,
- RequireInitialization = 1,
- };
- static Real epsilon()
- {
- return std::numeric_limits<Real>::epsilon();
- }
- static Real dummy_precision()
- {
- return 1000 * epsilon();
- }
- static Real highest()
- {
- return (std::numeric_limits<Real>::max)();
- }
- static Real lowest()
- {
- return (std::numeric_limits<Real>::min)();
- }
- static int digits10_imp(const std::integral_constant<bool, true>&)
- {
- return std::numeric_limits<Real>::digits10;
- }
- template <bool B>
- static int digits10_imp(const std::integral_constant<bool, B>&)
- {
- return Real::default_precision();
- }
- static int digits10()
- {
- return digits10_imp(std::integral_constant<bool, std::numeric_limits<Real>::digits10 && (std::numeric_limits<Real>::digits10 != INT_MAX) ? true : false > ());
- }
- };
- template <class tag, class Arg1, class Arg2, class Arg3, class Arg4>
- struct NumTraits<boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4> > : public NumTraits<typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>::result_type>
- {
- };
- #define BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(A) \
- template <class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, typename BinaryOp> \
- struct ScalarBinaryOpTraits<boost::multiprecision::number<Backend, ExpressionTemplates>, A, BinaryOp> \
- { \
- /*static_assert(boost::multiprecision::is_compatible_arithmetic_type<A, boost::multiprecision::number<Backend, ExpressionTemplates> >::value, "Interoperability with this arithmetic type is not supported.");*/ \
- using ReturnType = boost::multiprecision::number<Backend, ExpressionTemplates>; \
- }; \
- template <class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, typename BinaryOp> \
- struct ScalarBinaryOpTraits<A, boost::multiprecision::number<Backend, ExpressionTemplates>, BinaryOp> \
- { \
- /*static_assert(boost::multiprecision::is_compatible_arithmetic_type<A, boost::multiprecision::number<Backend, ExpressionTemplates> >::value, "Interoperability with this arithmetic type is not supported.");*/ \
- using ReturnType = boost::multiprecision::number<Backend, ExpressionTemplates>; \
- };
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(float)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(double)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(long double)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(char)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(unsigned char)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(signed char)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(short)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(unsigned short)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(int)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(unsigned int)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(long)
- BOOST_MP_EIGEN_SCALAR_TRAITS_DECL(unsigned long)
- #if 0
- template<class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, class Backend2, boost::multiprecision::expression_template_option ExpressionTemplates2, typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::number<Backend, ExpressionTemplates>, boost::multiprecision::number<Backend2, ExpressionTemplates2>, BinaryOp>
- {
- static_assert(
- boost::multiprecision::is_compatible_arithmetic_type<boost::multiprecision::number<Backend2, ExpressionTemplates2>, boost::multiprecision::number<Backend, ExpressionTemplates> >::value
- || boost::multiprecision::is_compatible_arithmetic_type<boost::multiprecision::number<Backend, ExpressionTemplates>, boost::multiprecision::number<Backend2, ExpressionTemplates2> >::value, "Interoperability with this arithmetic type is not supported.");
- using ReturnType = typename std::conditional<std::is_convertible<boost::multiprecision::number<Backend2, ExpressionTemplates2>, boost::multiprecision::number<Backend, ExpressionTemplates> >::value,
- boost::multiprecision::number<Backend, ExpressionTemplates>, boost::multiprecision::number<Backend2, ExpressionTemplates2> >::type;
- };
- template<unsigned D, typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::number<boost::multiprecision::backends::mpc_complex_backend<D>, boost::multiprecision::et_on>, boost::multiprecision::mpfr_float, BinaryOp>
- {
- using ReturnType = boost::multiprecision::number<boost::multiprecision::backends::mpc_complex_backend<D>, boost::multiprecision::et_on>;
- };
- template<typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::mpfr_float, boost::multiprecision::mpc_complex, BinaryOp>
- {
- using ReturnType = boost::multiprecision::number<boost::multiprecision::backends::mpc_complex_backend<0>, boost::multiprecision::et_on>;
- };
- template<class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::number<Backend, ExpressionTemplates>, boost::multiprecision::number<Backend, ExpressionTemplates>, BinaryOp>
- {
- using ReturnType = boost::multiprecision::number<Backend, ExpressionTemplates>;
- };
- #endif
- template <class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, class tag, class Arg1, class Arg2, class Arg3, class Arg4, typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::number<Backend, ExpressionTemplates>, boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>, BinaryOp>
- {
- static_assert(std::is_convertible<typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>::result_type, boost::multiprecision::number<Backend, ExpressionTemplates> >::value, "Interoperability with this arithmetic type is not supported.");
- using ReturnType = boost::multiprecision::number<Backend, ExpressionTemplates>;
- };
- template <class tag, class Arg1, class Arg2, class Arg3, class Arg4, class Backend, boost::multiprecision::expression_template_option ExpressionTemplates, typename BinaryOp>
- struct ScalarBinaryOpTraits<boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>, boost::multiprecision::number<Backend, ExpressionTemplates>, BinaryOp>
- {
- static_assert(std::is_convertible<typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>::result_type, boost::multiprecision::number<Backend, ExpressionTemplates> >::value, "Interoperability with this arithmetic type is not supported.");
- using ReturnType = boost::multiprecision::number<Backend, ExpressionTemplates>;
- };
- namespace internal {
- template <typename Scalar>
- struct conj_retval;
- template <typename Scalar, bool IsComplex>
- struct conj_impl;
- template <class tag, class Arg1, class Arg2, class Arg3, class Arg4>
- struct conj_retval<boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4> >
- {
- using type = typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>::result_type;
- };
- template <class tag, class Arg1, class Arg2, class Arg3, class Arg4>
- struct conj_impl<boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>, true>
- {
- EIGEN_DEVICE_FUNC
- static inline typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>::result_type run(const typename boost::multiprecision::detail::expression<tag, Arg1, Arg2, Arg3, Arg4>& x)
- {
- return conj(x);
- }
- };
- } // namespace internal
- } // namespace Eigen
- #endif
|