xref: /llvm-project/libcxx/include/__numeric/exclusive_scan.h (revision 9783f28cbb155e4a8d49c12e1c60ce14dcfaf0c7)
1a0efb175SChristopher Di Bella // -*- C++ -*-
2a0efb175SChristopher Di Bella //===----------------------------------------------------------------------===//
3a0efb175SChristopher Di Bella //
4a0efb175SChristopher Di Bella // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5a0efb175SChristopher Di Bella // See https://llvm.org/LICENSE.txt for license information.
6a0efb175SChristopher Di Bella // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7a0efb175SChristopher Di Bella //
8a0efb175SChristopher Di Bella //===----------------------------------------------------------------------===//
9a0efb175SChristopher Di Bella 
10a0efb175SChristopher Di Bella #ifndef _LIBCPP___NUMERIC_EXCLUSIVE_SCAN_H
11a0efb175SChristopher Di Bella #define _LIBCPP___NUMERIC_EXCLUSIVE_SCAN_H
12a0efb175SChristopher Di Bella 
13a0efb175SChristopher Di Bella #include <__config>
14a0efb175SChristopher Di Bella #include <__functional/operations.h>
15a0efb175SChristopher Di Bella #include <__utility/move.h>
16a0efb175SChristopher Di Bella 
17a0efb175SChristopher Di Bella #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
18a0efb175SChristopher Di Bella #  pragma GCC system_header
19a0efb175SChristopher Di Bella #endif
20a0efb175SChristopher Di Bella 
2192e4d679SNicole Rabjohn _LIBCPP_PUSH_MACROS
2292e4d679SNicole Rabjohn #include <__undef_macros>
2392e4d679SNicole Rabjohn 
24a0efb175SChristopher Di Bella _LIBCPP_BEGIN_NAMESPACE_STD
25a0efb175SChristopher Di Bella 
264f15267dSNikolas Klauser #if _LIBCPP_STD_VER >= 17
27a0efb175SChristopher Di Bella 
28a0efb175SChristopher Di Bella template <class _InputIterator, class _OutputIterator, class _Tp, class _BinaryOp>
294c198542SLouis Dionne _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
exclusive_scan(_InputIterator __first,_InputIterator __last,_OutputIterator __result,_Tp __init,_BinaryOp __b)30a0efb175SChristopher Di Bella exclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _Tp __init, _BinaryOp __b) {
31a0efb175SChristopher Di Bella   if (__first != __last) {
32a0efb175SChristopher Di Bella     _Tp __tmp(__b(__init, *__first));
33a0efb175SChristopher Di Bella     while (true) {
34*77a00c0dSLouis Dionne       *__result = std::move(__init);
35a0efb175SChristopher Di Bella       ++__result;
36a0efb175SChristopher Di Bella       ++__first;
37a0efb175SChristopher Di Bella       if (__first == __last)
38a0efb175SChristopher Di Bella         break;
39*77a00c0dSLouis Dionne       __init = std::move(__tmp);
40a0efb175SChristopher Di Bella       __tmp  = __b(__init, *__first);
41a0efb175SChristopher Di Bella     }
42a0efb175SChristopher Di Bella   }
43a0efb175SChristopher Di Bella   return __result;
44a0efb175SChristopher Di Bella }
45a0efb175SChristopher Di Bella 
46a0efb175SChristopher Di Bella template <class _InputIterator, class _OutputIterator, class _Tp>
474c198542SLouis Dionne _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
exclusive_scan(_InputIterator __first,_InputIterator __last,_OutputIterator __result,_Tp __init)48a0efb175SChristopher Di Bella exclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _Tp __init) {
49*77a00c0dSLouis Dionne   return std::exclusive_scan(__first, __last, __result, __init, std::plus<>());
50a0efb175SChristopher Di Bella }
51a0efb175SChristopher Di Bella 
524f15267dSNikolas Klauser #endif // _LIBCPP_STD_VER >= 17
53a0efb175SChristopher Di Bella 
54a0efb175SChristopher Di Bella _LIBCPP_END_NAMESPACE_STD
55a0efb175SChristopher Di Bella 
5692e4d679SNicole Rabjohn _LIBCPP_POP_MACROS
5792e4d679SNicole Rabjohn 
58a0efb175SChristopher Di Bella #endif // _LIBCPP___NUMERIC_EXCLUSIVE_SCAN_H
59