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_INCLUSIVE_SCAN_H
11*4bdff4beSrobert #define _LIBCPP___NUMERIC_INCLUSIVE_SCAN_H
12*4bdff4beSrobert
13*4bdff4beSrobert #include <__config>
14*4bdff4beSrobert #include <__functional/operations.h>
15*4bdff4beSrobert #include <__iterator/iterator_traits.h>
16*4bdff4beSrobert #include <__utility/move.h>
17*4bdff4beSrobert
18*4bdff4beSrobert #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
19*4bdff4beSrobert # pragma GCC system_header
20*4bdff4beSrobert #endif
21*4bdff4beSrobert
22*4bdff4beSrobert _LIBCPP_BEGIN_NAMESPACE_STD
23*4bdff4beSrobert
24*4bdff4beSrobert #if _LIBCPP_STD_VER > 14
25*4bdff4beSrobert
26*4bdff4beSrobert template <class _InputIterator, class _OutputIterator, class _Tp, class _BinaryOp>
27*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
inclusive_scan(_InputIterator __first,_InputIterator __last,_OutputIterator __result,_BinaryOp __b,_Tp __init)28*4bdff4beSrobert inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b, _Tp __init) {
29*4bdff4beSrobert for (; __first != __last; ++__first, (void)++__result) {
30*4bdff4beSrobert __init = __b(__init, *__first);
31*4bdff4beSrobert *__result = __init;
32*4bdff4beSrobert }
33*4bdff4beSrobert return __result;
34*4bdff4beSrobert }
35*4bdff4beSrobert
36*4bdff4beSrobert template <class _InputIterator, class _OutputIterator, class _BinaryOp>
37*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
inclusive_scan(_InputIterator __first,_InputIterator __last,_OutputIterator __result,_BinaryOp __b)38*4bdff4beSrobert inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b) {
39*4bdff4beSrobert if (__first != __last) {
40*4bdff4beSrobert typename iterator_traits<_InputIterator>::value_type __init = *__first;
41*4bdff4beSrobert *__result++ = __init;
42*4bdff4beSrobert if (++__first != __last)
43*4bdff4beSrobert return _VSTD::inclusive_scan(__first, __last, __result, __b, __init);
44*4bdff4beSrobert }
45*4bdff4beSrobert
46*4bdff4beSrobert return __result;
47*4bdff4beSrobert }
48*4bdff4beSrobert
49*4bdff4beSrobert template <class _InputIterator, class _OutputIterator>
inclusive_scan(_InputIterator __first,_InputIterator __last,_OutputIterator __result)50*4bdff4beSrobert _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator inclusive_scan(_InputIterator __first,
51*4bdff4beSrobert _InputIterator __last,
52*4bdff4beSrobert _OutputIterator __result) {
53*4bdff4beSrobert return _VSTD::inclusive_scan(__first, __last, __result, _VSTD::plus<>());
54*4bdff4beSrobert }
55*4bdff4beSrobert
56*4bdff4beSrobert #endif // _LIBCPP_STD_VER > 14
57*4bdff4beSrobert
58*4bdff4beSrobert _LIBCPP_END_NAMESPACE_STD
59*4bdff4beSrobert
60*4bdff4beSrobert #endif // _LIBCPP___NUMERIC_INCLUSIVE_SCAN_H
61