Sprout/sprout/utility/until_loop.hpp
2016-03-29 19:36:57 +09:00

78 lines
2.5 KiB
C++

/*=============================================================================
Copyright (c) 2011-2016 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_UNTIL_LOOP_HPP
#define SPROUT_UTILITY_UNTIL_LOOP_HPP
#include <sprout/config.hpp>
#include <sprout/utility/pair/pair.hpp>
namespace sprout {
namespace detail {
template<typename T, typename Predicate>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
until_loop_check(T const& init, Predicate pred) {
return sprout::pair<T, bool>(init, pred(init));
}
template<typename T, typename Predicate, typename UnaryOperation>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
until_loop_impl_1(sprout::pair<T, bool> const& current, Predicate pred, UnaryOperation unary_op, std::size_t n) {
return current.second || pred(current.first) ? current
: n == 1 ? sprout::detail::until_loop_check<T>(unary_op(current.first), pred)
: sprout::detail::until_loop_impl_1(
sprout::detail::until_loop_impl_1(
current,
pred, unary_op, n / 2
),
pred, unary_op, n - n / 2
)
;
}
template<typename T, typename Predicate, typename UnaryOperation>
inline SPROUT_CONSTEXPR sprout::pair<T, bool>
until_loop_impl(sprout::pair<T, bool> const& current, Predicate pred, UnaryOperation unary_op, std::size_t n) {
return current.second || pred(current.first) ? current
: sprout::detail::until_loop_impl(
sprout::detail::until_loop_impl_1(
current,
pred, unary_op, n
),
pred, unary_op, n * 2
)
;
}
} // namespace detail
//
// until_loop
//
// recursion depth:
// O(log N)
//
template<typename T, typename Predicate, typename UnaryOperation>
inline SPROUT_CONSTEXPR T
until_loop(T init, Predicate pred, UnaryOperation unary_op) {
typedef sprout::pair<T, bool> type;
return sprout::detail::until_loop_impl(type(init, false), pred, unary_op, 1).second;
}
//
// flat_until_loop
//
// recursion depth:
// 0
//
template<typename T, typename Predicate, typename UnaryOperation>
inline SPROUT_CXX14_CONSTEXPR T
flat_until_loop(T init, Predicate pred, UnaryOperation unary_op) {
for (; !pred(init); init = unary_op(init))
;
return init;
}
} // namespace sprout
#endif // #ifndef SPROUT_UTILITY_UNTIL_LOOP_HPP