1*4bdff4beSrobert // -*- C++ -*-
2*4bdff4beSrobert //===----------------------------------------------------------------------===//
3*4bdff4beSrobert //
4*4bdff4beSrobert // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5*4bdff4beSrobert // See https://llvm.org/LICENSE.txt for license information.
6*4bdff4beSrobert // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7*4bdff4beSrobert //
8*4bdff4beSrobert //===----------------------------------------------------------------------===//
9*4bdff4beSrobert
10*4bdff4beSrobert #ifndef _LIBCPP___NUMERIC_TRANSFORM_REDUCE_H
11*4bdff4beSrobert #define _LIBCPP___NUMERIC_TRANSFORM_REDUCE_H
12*4bdff4beSrobert
13*4bdff4beSrobert #include <__config>
14*4bdff4beSrobert #include <__functional/operations.h>
15*4bdff4beSrobert #include <__utility/move.h>
16*4bdff4beSrobert
17*4bdff4beSrobert #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
18*4bdff4beSrobert # pragma GCC system_header
19*4bdff4beSrobert #endif
20*4bdff4beSrobert
21*4bdff4beSrobert _LIBCPP_BEGIN_NAMESPACE_STD
22*4bdff4beSrobert
23*4bdff4beSrobert #if _LIBCPP_STD_VER > 14
24*4bdff4beSrobert template <class _InputIterator, class _Tp, class _BinaryOp, class _UnaryOp>
transform_reduce(_InputIterator __first,_InputIterator __last,_Tp __init,_BinaryOp __b,_UnaryOp __u)25*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _Tp transform_reduce(_InputIterator __first,
26*4bdff4beSrobert _InputIterator __last, _Tp __init,
27*4bdff4beSrobert _BinaryOp __b, _UnaryOp __u) {
28*4bdff4beSrobert for (; __first != __last; ++__first)
29*4bdff4beSrobert __init = __b(__init, __u(*__first));
30*4bdff4beSrobert return __init;
31*4bdff4beSrobert }
32*4bdff4beSrobert
33*4bdff4beSrobert template <class _InputIterator1, class _InputIterator2, class _Tp, class _BinaryOp1, class _BinaryOp2>
transform_reduce(_InputIterator1 __first1,_InputIterator1 __last1,_InputIterator2 __first2,_Tp __init,_BinaryOp1 __b1,_BinaryOp2 __b2)34*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _Tp transform_reduce(_InputIterator1 __first1,
35*4bdff4beSrobert _InputIterator1 __last1,
36*4bdff4beSrobert _InputIterator2 __first2, _Tp __init,
37*4bdff4beSrobert _BinaryOp1 __b1, _BinaryOp2 __b2) {
38*4bdff4beSrobert for (; __first1 != __last1; ++__first1, (void)++__first2)
39*4bdff4beSrobert __init = __b1(__init, __b2(*__first1, *__first2));
40*4bdff4beSrobert return __init;
41*4bdff4beSrobert }
42*4bdff4beSrobert
43*4bdff4beSrobert template <class _InputIterator1, class _InputIterator2, class _Tp>
transform_reduce(_InputIterator1 __first1,_InputIterator1 __last1,_InputIterator2 __first2,_Tp __init)44*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _Tp transform_reduce(_InputIterator1 __first1,
45*4bdff4beSrobert _InputIterator1 __last1,
46*4bdff4beSrobert _InputIterator2 __first2, _Tp __init) {
47*4bdff4beSrobert return _VSTD::transform_reduce(__first1, __last1, __first2, _VSTD::move(__init), _VSTD::plus<>(),
48*4bdff4beSrobert _VSTD::multiplies<>());
49*4bdff4beSrobert }
50*4bdff4beSrobert #endif
51*4bdff4beSrobert
52*4bdff4beSrobert _LIBCPP_END_NAMESPACE_STD
53*4bdff4beSrobert
54*4bdff4beSrobert #endif // _LIBCPP___NUMERIC_TRANSFORM_REDUCE_H
55