mirror of
https://github.com/bolero-MURAKAMI/Sprout
synced 2024-11-12 21:09:01 +00:00
247 lines
9.9 KiB
C++
247 lines
9.9 KiB
C++
#ifndef SPROUT_NUMERIC_FIXED_PARTIAL_SUM_HPP
|
|
#define SPROUT_NUMERIC_FIXED_PARTIAL_SUM_HPP
|
|
|
|
#include <type_traits>
|
|
#include <sprout/config.hpp>
|
|
#include <sprout/fixed_container/traits.hpp>
|
|
#include <sprout/fixed_container/functions.hpp>
|
|
#include <sprout/iterator/operation.hpp>
|
|
#include <sprout/algorithm/fixed/result_of.hpp>
|
|
|
|
namespace sprout {
|
|
namespace fixed {
|
|
namespace detail {
|
|
template<typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_3(
|
|
Result const& result,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_3(
|
|
Result const& result,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return partial_sum_impl_3(result, args..., *sprout::next(sprout::fixed_begin(result), sizeof...(Args)));
|
|
}
|
|
template<typename InputIterator, typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_2(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
typename sprout::fixed_container_traits<Result>::value_type const& value,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename InputIterator, typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_2(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
typename sprout::fixed_container_traits<Result>::value_type const& value,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return first != last && sizeof...(Args) < static_cast<std::size_t>(offset)
|
|
? partial_sum_impl_2(sprout::next(first), last, result, value + *first, offset, args..., value)
|
|
: partial_sum_impl_3(result, args..., value)
|
|
;
|
|
}
|
|
template<typename InputIterator, typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_1(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename InputIterator, typename Result, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_1(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sizeof...(Args) < static_cast<std::size_t>(offset)
|
|
? partial_sum_impl_1(first, last, result, offset, args..., *sprout::next(sprout::fixed_begin(result), sizeof...(Args)))
|
|
: first != last
|
|
? partial_sum_impl_2(sprout::next(first), last, result, *first, offset + sprout::size(result), args...)
|
|
: partial_sum_impl_3(result, args...)
|
|
;
|
|
}
|
|
template<typename InputIterator, typename Result>
|
|
SPROUT_CONSTEXPR inline typename sprout::fixed::result_of::algorithm<Result>::type partial_sum_impl(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result
|
|
)
|
|
{
|
|
return partial_sum_impl_1(first, last, result, sprout::fixed_begin_offset(result));
|
|
}
|
|
} // namespace detail
|
|
//
|
|
// partial_sum
|
|
//
|
|
template<typename InputIterator, typename Result>
|
|
SPROUT_CONSTEXPR inline typename sprout::fixed::result_of::algorithm<Result>::type partial_sum(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result
|
|
)
|
|
{
|
|
return sprout::fixed::detail::partial_sum_impl(first, last, result);
|
|
}
|
|
|
|
namespace detail {
|
|
template<typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_3(
|
|
Result const& result,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_3(
|
|
Result const& result,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return partial_sum_impl_3(result, args..., *sprout::next(sprout::fixed_begin(result), sizeof...(Args)));
|
|
}
|
|
template<typename InputIterator, typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_2(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op,
|
|
typename sprout::fixed_container_traits<Result>::value_type const& value,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename InputIterator, typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_2(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op,
|
|
typename sprout::fixed_container_traits<Result>::value_type const& value,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return first != last && sizeof...(Args) < static_cast<std::size_t>(offset)
|
|
? partial_sum_impl_2(sprout::next(first), last, result, binary_op, binary_op(value, *first), offset, args..., value)
|
|
: partial_sum_impl_3(result, args..., value)
|
|
;
|
|
}
|
|
template<typename InputIterator, typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size == sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_1(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sprout::remake_clone<Result, Result>(result, sprout::size(result), args...);
|
|
}
|
|
template<typename InputIterator, typename Result, typename BinaryOperation, typename... Args>
|
|
SPROUT_CONSTEXPR inline typename std::enable_if<
|
|
sprout::fixed_container_traits<Result>::fixed_size != sizeof...(Args),
|
|
typename sprout::fixed::result_of::algorithm<Result>::type
|
|
>::type partial_sum_impl_1(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op,
|
|
typename sprout::fixed_container_traits<Result>::difference_type offset,
|
|
Args const&... args
|
|
)
|
|
{
|
|
return sizeof...(Args) < static_cast<std::size_t>(offset)
|
|
? partial_sum_impl_1(first, last, result, binary_op, offset, args..., *sprout::next(sprout::fixed_begin(result), sizeof...(Args)))
|
|
: first != last
|
|
? partial_sum_impl_2(sprout::next(first), last, result, binary_op, *first, offset + sprout::size(result), args...)
|
|
: partial_sum_impl_3(result, args...)
|
|
;
|
|
}
|
|
template<typename InputIterator, typename Result, typename BinaryOperation>
|
|
SPROUT_CONSTEXPR inline typename sprout::fixed::result_of::algorithm<Result>::type partial_sum_impl(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op
|
|
)
|
|
{
|
|
return partial_sum_impl_1(first, last, result, binary_op, sprout::fixed_begin_offset(result));
|
|
}
|
|
} // namespace detail
|
|
//
|
|
// partial_sum
|
|
//
|
|
template<typename InputIterator, typename Result, typename BinaryOperation>
|
|
SPROUT_CONSTEXPR inline typename sprout::fixed::result_of::algorithm<Result>::type partial_sum(
|
|
InputIterator first,
|
|
InputIterator last,
|
|
Result const& result,
|
|
BinaryOperation binary_op
|
|
)
|
|
{
|
|
return sprout::fixed::detail::partial_sum_impl(first, last, result, binary_op);
|
|
}
|
|
} // namespace fixed
|
|
|
|
using sprout::fixed::partial_sum;
|
|
} // namespace sprout
|
|
|
|
#endif // #ifndef SPROUT_NUMERIC_FIXED_PARTIAL_SUM_HPP
|