mirror of
https://github.com/bolero-MURAKAMI/Sprout.git
synced 2024-12-25 21:35:41 +00:00
73 lines
1.7 KiB
C++
73 lines
1.7 KiB
C++
|
#ifndef SPROUT_MATH_SQRT_HPP
|
||
|
#define SPROUT_MATH_SQRT_HPP
|
||
|
|
||
|
#include <limits>
|
||
|
#include <type_traits>
|
||
|
#include <sprout/config.hpp>
|
||
|
#include <sprout/utility/enabler_if.hpp>
|
||
|
#if SPROUT_USE_BUILTIN_CMATH_FUNCTION
|
||
|
# include <cmath>
|
||
|
#endif
|
||
|
|
||
|
namespace sprout {
|
||
|
namespace math {
|
||
|
namespace detail {
|
||
|
template<typename T>
|
||
|
inline SPROUT_CONSTEXPR T
|
||
|
sqrt_impl_1(T x, T s, T s2) {
|
||
|
return !(s < s2) ? s2
|
||
|
: sprout::math::detail::sqrt_impl_1(
|
||
|
x,
|
||
|
(x / s + s) / 2,
|
||
|
s
|
||
|
)
|
||
|
;
|
||
|
}
|
||
|
template<typename T>
|
||
|
inline SPROUT_CONSTEXPR T
|
||
|
sqrt_impl(T x, T s) {
|
||
|
return sprout::math::detail::sqrt_impl_1(
|
||
|
x,
|
||
|
(x / s + s) / 2,
|
||
|
s
|
||
|
)
|
||
|
;
|
||
|
}
|
||
|
|
||
|
template<
|
||
|
typename FloatType,
|
||
|
typename sprout::enabler_if<std::is_floating_point<FloatType>::value>::type = sprout::enabler
|
||
|
>
|
||
|
inline SPROUT_CONSTEXPR FloatType
|
||
|
sqrt(FloatType x) {
|
||
|
typedef double type;
|
||
|
return x < 0 ? std::numeric_limits<FloatType>::quiet_NaN()
|
||
|
: x == 0 ? type(0)
|
||
|
: static_cast<FloatType>(sprout::math::detail::sqrt_impl(
|
||
|
static_cast<type>(x),
|
||
|
x > 1 ? static_cast<type>(x) : type(1)
|
||
|
));
|
||
|
}
|
||
|
|
||
|
template<
|
||
|
typename IntType,
|
||
|
typename sprout::enabler_if<std::is_integral<IntType>::value>::type = sprout::enabler
|
||
|
>
|
||
|
inline SPROUT_CONSTEXPR double
|
||
|
sqrt(IntType x) {
|
||
|
return sprout::math::detail::sqrt(static_cast<double>(x));
|
||
|
}
|
||
|
} // namespace detail
|
||
|
|
||
|
# if SPROUT_USE_BUILTIN_CMATH_FUNCTION
|
||
|
using std::sqrt;
|
||
|
# else
|
||
|
using sprout::math::detail::sqrt;
|
||
|
# endif
|
||
|
} // namespace math
|
||
|
|
||
|
using sprout::math::sqrt;
|
||
|
} // namespace sprout
|
||
|
|
||
|
#endif // #ifndef SPROUT_MATH_SQRT_HPP
|