Sprout/sprout/utility/until_loop.hpp

/*=============================================================================
  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_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).first;
	}

	//
	// 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