Sprout/sprout/detail/pow.hpp
2019-01-07 17:47:17 +09:00

72 lines
2.1 KiB
C++

/*=============================================================================
Copyright (c) 2011-2019 Bolero MURAKAMI
https://github.com/bolero-MURAKAMI/Sprout
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 SPROUT_DETAIL_POW_HPP
#define SPROUT_DETAIL_POW_HPP
#include <type_traits>
#include <sprout/config.hpp>
#include <sprout/type_traits/is_sint.hpp>
#include <sprout/type_traits/is_uint.hpp>
#include <sprout/type_traits/enabler_if.hpp>
namespace sprout {
namespace detail {
template<typename T>
inline SPROUT_CONSTEXPR T
pow2(T const& x) {
return x * x;
}
template<typename T>
inline SPROUT_CONSTEXPR T
pow3(T const& x) {
return x * x * x;
}
template<typename T>
inline SPROUT_CONSTEXPR T
pow4(T const& x) {
return x * x * x * x;
}
template<typename T>
inline SPROUT_CONSTEXPR T
pow5(T const& x) {
return x * x * x * x * x;
}
template<typename T, typename IntType>
inline SPROUT_CONSTEXPR T
pow_n_impl(T const& x, IntType n) {
return n == 1 ? x
: n % 2 ? x * sprout::detail::pow2(sprout::detail::pow_n_impl(x, n / 2))
: sprout::detail::pow2(sprout::detail::pow_n_impl(x, n / 2))
;
}
template<
typename T, typename UIntType,
typename sprout::enabler_if<sprout::is_uint<UIntType>::value>::type = sprout::enabler
>
inline SPROUT_CONSTEXPR T
pow_n(T const& x, UIntType n) {
return n == 0 ? T(1)
: sprout::detail::pow_n_impl(x, n)
;
}
template<
typename T, typename IntType,
typename sprout::enabler_if<sprout::is_sint<IntType>::value>::type = sprout::enabler
>
inline SPROUT_CONSTEXPR T
pow_n(T const& x, IntType n) {
return n == 0 ? T(1)
: n > 0 ? sprout::detail::pow_n_impl(x, n)
: T(1) / sprout::detail::pow_n_impl(x, -n)
;
}
} // namespace detail
} // namespace sprout
#endif // #ifndef SPROUT_DETAIL_POW_HPP