partial_sum.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSL-1.0
#ifndef TETL_NUMERIC_PARTIAL_SUM_HPP
#define TETL_NUMERIC_PARTIAL_SUM_HPP
 
#include <etl/_functional/plus.hpp>
#include <etl/_utility/move.hpp>
 
namespace etl {

template <typename InputIt, typename OutputIt, typename BinaryOperation>
constexpr auto partial_sum(InputIt first, InputIt last, OutputIt destination, BinaryOperation op) -> OutputIt
{
    if (first == last) {
        return destination;
    }
 
    auto sum     = *first;
    *destination = sum;
 
    while (++first != last) {
        sum            = op(etl::move(sum), *first);
        *++destination = sum;
    }
 
    return ++destination;
}

template <typename InputIt, typename OutputIt>
constexpr auto partial_sum(InputIt first, InputIt last, OutputIt destination) -> OutputIt
{
    return etl::partial_sum(first, last, destination, etl::plus<>());
}
 
} // namespace etl
 
#endif // TETL_NUMERIC_PARTIAL_SUM_HPP