transform_reduce.hpp

Go to the documentation of this file.

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

template <typename InputIt1, typename InputIt2, typename T, typename BinaryReductionOp, typename BinaryTransformOp>
[[nodiscard]] constexpr auto transform_reduce(
    InputIt1 first1,
    InputIt1 last1,
    InputIt2 first2,
    T init,
    BinaryReductionOp reduce,
    BinaryTransformOp transform
) -> T
{
    for (; first1 != last1; ++first1, (void)++first2) {
        init = reduce(init, transform(*first1, *first2));
    }
    return init;
}

template <typename InputIt1, typename InputIt2, typename T>
[[nodiscard]] constexpr auto transform_reduce(InputIt1 first1, InputIt1 last1, InputIt2 first2, T init) -> T
{
    return etl::transform_reduce(first1, last1, first2, init, etl::plus(), etl::multiplies());
}

template <typename InputIt, typename T, typename BinaryReductionOp, typename UnaryTransformOp>
[[nodiscard]] constexpr auto
transform_reduce(InputIt first, InputIt last, T init, BinaryReductionOp reduce, UnaryTransformOp transform) -> T
{
    for (; first != last; ++first) {
        init = reduce(init, transform(*first));
    }
    return init;
}
 
} // namespace etl
 
#endif // TETL_NUMERIC_TRANSFORM_REDUCE_HPP