1
0
Fork 0
mirror of https://github.com/bolero-MURAKAMI/Sprout synced 2024-11-12 21:09:01 +00:00
Sprout/sprout/utility/while_loop.hpp
2015-04-13 11:40:49 +09:00

81 lines
2.6 KiB
C++

/*=============================================================================
Copyright (c) 2011-2015 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_UTILITY_WHILE_LOOP_HPP
#define SPROUT_UTILITY_WHILE_LOOP_HPP
#include <sprout/config.hpp>
#include <sprout/utility/pair/pair.hpp>
namespace sprout {
#ifdef SPROUT_CONFIG_DISABLE_CXX14_CONSTEXPR
namespace detail {
template<typename T, typename Predicate>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
while_loop_check(T const& init, Predicate pred) {
return sprout::pair<T, bool>(init, !pred(init));
}
template<typename T, typename UnaryOperation, typename Predicate>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
while_loop_impl_1(sprout::pair<T, bool> const& current, UnaryOperation unary_op, Predicate pred, std::size_t n) {
return current.second || !pred(current.first) ? current
: n == 1 ? sprout::detail::while_loop_check<T>(unary_op(current.first), pred)
: sprout::detail::while_loop_impl_1(
sprout::detail::while_loop_impl_1(
current,
unary_op, pred, n / 2
),
unary_op, pred, n - n / 2
)
;
}
template<typename T, typename UnaryOperation, typename Predicate>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
while_loop_impl(sprout::pair<T, bool> const& current, UnaryOperation unary_op, Predicate pred, std::size_t n) {
return current.second || !pred(current.first) ? current
: sprout::detail::while_loop_impl(
sprout::detail::while_loop_impl_1(
current,
unary_op, pred, n
),
unary_op, pred, n * 2
)
;
}
} // namespace detail
//
// while_loop
//
// recursion depth:
// O(log N)
//
template<typename T, typename UnaryOperation, typename Predicate>
inline SPROUT_CONSTEXPR T
while_loop(T init, UnaryOperation unary_op, Predicate pred) {
typedef sprout::pair<T, bool> type;
return sprout::detail::while_loop_impl(type(init, false), unary_op, pred, 1).second;
}
#else
//
// while_loop
//
// recursion depth:
// 0
//
template<typename T, typename UnaryOperation, typename Predicate>
inline SPROUT_CXX14_CONSTEXPR T
while_loop(T init, UnaryOperation unary_op, Predicate pred) {
while (pred(init)) {
init = unary_op(init);
}
return init;
}
#endif
} // namespace sprout
#endif // #ifndef SPROUT_UTILITY_WHILE_LOOP_HPP