123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246 |
- ///////////////////////////////////////////////////////////////
- // Copyright 2012 John Maddock. Distributed under the Boost
- // Software License, Version 1.0. (See accompanying file
- // LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt
- #ifndef BOOST_MP_INTEGER_HPP
- #define BOOST_MP_INTEGER_HPP
- #include <boost/multiprecision/cpp_int.hpp>
- #include <boost/multiprecision/detail/bitscan.hpp>
- namespace boost {
- namespace multiprecision {
- template <class Integer, class I2>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value && boost::multiprecision::detail::is_integral<I2>::value, Integer&>::type
- multiply(Integer& result, const I2& a, const I2& b)
- {
- return result = static_cast<Integer>(a) * static_cast<Integer>(b);
- }
- template <class Integer, class I2>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value && boost::multiprecision::detail::is_integral<I2>::value, Integer&>::type
- add(Integer& result, const I2& a, const I2& b)
- {
- return result = static_cast<Integer>(a) + static_cast<Integer>(b);
- }
- template <class Integer, class I2>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value && boost::multiprecision::detail::is_integral<I2>::value, Integer&>::type
- subtract(Integer& result, const I2& a, const I2& b)
- {
- return result = static_cast<Integer>(a) - static_cast<Integer>(b);
- }
- template <class Integer>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value>::type divide_qr(const Integer& x, const Integer& y, Integer& q, Integer& r)
- {
- q = x / y;
- r = x % y;
- }
- template <class I1, class I2>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<I1>::value && boost::multiprecision::detail::is_integral<I2>::value, I2>::type integer_modulus(const I1& x, I2 val)
- {
- return static_cast<I2>(x % val);
- }
- namespace detail {
- //
- // Figure out the kind of integer that has twice as many bits as some builtin
- // integer type I. Use a native type if we can (including types which may not
- // be recognised by boost::int_t because they're larger than boost::long_long_type),
- // otherwise synthesize a cpp_int to do the job.
- //
- template <class I>
- struct double_integer
- {
- static constexpr const unsigned int_t_digits =
- 2 * sizeof(I) <= sizeof(boost::long_long_type) ? std::numeric_limits<I>::digits * 2 : 1;
- using type = typename std::conditional<
- 2 * sizeof(I) <= sizeof(boost::long_long_type),
- typename std::conditional<
- boost::multiprecision::detail::is_signed<I>::value && boost::multiprecision::detail::is_integral<I>::value,
- typename boost::int_t<int_t_digits>::least,
- typename boost::uint_t<int_t_digits>::least>::type,
- typename std::conditional<
- 2 * sizeof(I) <= sizeof(double_limb_type),
- typename std::conditional<
- boost::multiprecision::detail::is_signed<I>::value && boost::multiprecision::detail::is_integral<I>::value,
- signed_double_limb_type,
- double_limb_type>::type,
- number<cpp_int_backend<sizeof(I) * CHAR_BIT * 2, sizeof(I) * CHAR_BIT * 2, (boost::multiprecision::detail::is_signed<I>::value ? signed_magnitude : unsigned_magnitude), unchecked, void> > >::type>::type;
- };
- } // namespace detail
- template <class I1, class I2, class I3>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<I1>::value && boost::multiprecision::detail::is_unsigned<I2>::value && is_integral<I3>::value, I1>::type
- powm(const I1& a, I2 b, I3 c)
- {
- using double_type = typename detail::double_integer<I1>::type;
- I1 x(1), y(a);
- double_type result(0);
- while (b > 0)
- {
- if (b & 1)
- {
- multiply(result, x, y);
- x = integer_modulus(result, c);
- }
- multiply(result, y, y);
- y = integer_modulus(result, c);
- b >>= 1;
- }
- return x % c;
- }
- template <class I1, class I2, class I3>
- inline BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<I1>::value && boost::multiprecision::detail::is_signed<I2>::value && boost::multiprecision::detail::is_integral<I2>::value && boost::multiprecision::detail::is_integral<I3>::value, I1>::type
- powm(const I1& a, I2 b, I3 c)
- {
- if (b < 0)
- {
- BOOST_THROW_EXCEPTION(std::runtime_error("powm requires a positive exponent."));
- }
- return powm(a, static_cast<typename boost::multiprecision::detail::make_unsigned<I2>::type>(b), c);
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, unsigned>::type lsb(const Integer& val)
- {
- if (val <= 0)
- {
- if (val == 0)
- {
- BOOST_THROW_EXCEPTION(std::domain_error("No bits were set in the operand."));
- }
- else
- {
- BOOST_THROW_EXCEPTION(std::domain_error("Testing individual bits in negative values is not supported - results are undefined."));
- }
- }
- return detail::find_lsb(val);
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, unsigned>::type msb(Integer val)
- {
- if (val <= 0)
- {
- if (val == 0)
- {
- BOOST_THROW_EXCEPTION(std::domain_error("No bits were set in the operand."));
- }
- else
- {
- BOOST_THROW_EXCEPTION(std::domain_error("Testing individual bits in negative values is not supported - results are undefined."));
- }
- }
- return detail::find_msb(val);
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, bool>::type bit_test(const Integer& val, unsigned index)
- {
- Integer mask = 1;
- if (index >= sizeof(Integer) * CHAR_BIT)
- return 0;
- if (index)
- mask <<= index;
- return val & mask ? true : false;
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, Integer&>::type bit_set(Integer& val, unsigned index)
- {
- Integer mask = 1;
- if (index >= sizeof(Integer) * CHAR_BIT)
- return val;
- if (index)
- mask <<= index;
- val |= mask;
- return val;
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, Integer&>::type bit_unset(Integer& val, unsigned index)
- {
- Integer mask = 1;
- if (index >= sizeof(Integer) * CHAR_BIT)
- return val;
- if (index)
- mask <<= index;
- val &= ~mask;
- return val;
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, Integer&>::type bit_flip(Integer& val, unsigned index)
- {
- Integer mask = 1;
- if (index >= sizeof(Integer) * CHAR_BIT)
- return val;
- if (index)
- mask <<= index;
- val ^= mask;
- return val;
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, Integer>::type sqrt(const Integer& x, Integer& r)
- {
- //
- // This is slow bit-by-bit integer square root, see for example
- // http://en.wikipedia.org/wiki/Methods_of_computing_square_roots#Binary_numeral_system_.28base_2.29
- // There are better methods such as http://hal.inria.fr/docs/00/07/28/54/PDF/RR-3805.pdf
- // and http://hal.inria.fr/docs/00/07/21/13/PDF/RR-4475.pdf which should be implemented
- // at some point.
- //
- Integer s = 0;
- if (x == 0)
- {
- r = 0;
- return s;
- }
- int g = msb(x);
- if (g == 0)
- {
- r = 1;
- return s;
- }
- Integer t = 0;
- r = x;
- g /= 2;
- bit_set(s, g);
- bit_set(t, 2 * g);
- r = x - t;
- --g;
- do
- {
- t = s;
- t <<= g + 1;
- bit_set(t, 2 * g);
- if (t <= r)
- {
- bit_set(s, g);
- r -= t;
- }
- --g;
- } while (g >= 0);
- return s;
- }
- template <class Integer>
- BOOST_MP_CXX14_CONSTEXPR typename std::enable_if<boost::multiprecision::detail::is_integral<Integer>::value, Integer>::type sqrt(const Integer& x)
- {
- Integer r(0);
- return sqrt(x, r);
- }
- }} // namespace boost::multiprecision
- #endif
|