xref: /llvm-project/libcxx/include/__cxx03/__numeric/inclusive_scan.h (revision ce7771902dc50d900de639d499a60486b83f70e0)
1e78f53d1SNikolas Klauser // -*- C++ -*-
2e78f53d1SNikolas Klauser //===----------------------------------------------------------------------===//
3e78f53d1SNikolas Klauser //
4e78f53d1SNikolas Klauser // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5e78f53d1SNikolas Klauser // See https://llvm.org/LICENSE.txt for license information.
6e78f53d1SNikolas Klauser // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7e78f53d1SNikolas Klauser //
8e78f53d1SNikolas Klauser //===----------------------------------------------------------------------===//
9e78f53d1SNikolas Klauser 
10*ce777190SNikolas Klauser #ifndef _LIBCPP___CXX03___NUMERIC_INCLUSIVE_SCAN_H
11*ce777190SNikolas Klauser #define _LIBCPP___CXX03___NUMERIC_INCLUSIVE_SCAN_H
12e78f53d1SNikolas Klauser 
1373fbae83SNikolas Klauser #include <__cxx03/__config>
1473fbae83SNikolas Klauser #include <__cxx03/__functional/operations.h>
1573fbae83SNikolas Klauser #include <__cxx03/__iterator/iterator_traits.h>
1673fbae83SNikolas Klauser #include <__cxx03/__utility/move.h>
17e78f53d1SNikolas Klauser 
18e78f53d1SNikolas Klauser #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
19e78f53d1SNikolas Klauser #  pragma GCC system_header
20e78f53d1SNikolas Klauser #endif
21e78f53d1SNikolas Klauser 
22e78f53d1SNikolas Klauser _LIBCPP_BEGIN_NAMESPACE_STD
23e78f53d1SNikolas Klauser 
24e78f53d1SNikolas Klauser #if _LIBCPP_STD_VER >= 17
25e78f53d1SNikolas Klauser 
26e78f53d1SNikolas Klauser template <class _InputIterator, class _OutputIterator, class _Tp, class _BinaryOp>
27e78f53d1SNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
28e78f53d1SNikolas Klauser inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b, _Tp __init) {
29e78f53d1SNikolas Klauser   for (; __first != __last; ++__first, (void)++__result) {
30e78f53d1SNikolas Klauser     __init    = __b(__init, *__first);
31e78f53d1SNikolas Klauser     *__result = __init;
32e78f53d1SNikolas Klauser   }
33e78f53d1SNikolas Klauser   return __result;
34e78f53d1SNikolas Klauser }
35e78f53d1SNikolas Klauser 
36e78f53d1SNikolas Klauser template <class _InputIterator, class _OutputIterator, class _BinaryOp>
37e78f53d1SNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
38e78f53d1SNikolas Klauser inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b) {
39e78f53d1SNikolas Klauser   if (__first != __last) {
40e78f53d1SNikolas Klauser     typename iterator_traits<_InputIterator>::value_type __init = *__first;
41e78f53d1SNikolas Klauser     *__result++                                                 = __init;
42e78f53d1SNikolas Klauser     if (++__first != __last)
43e78f53d1SNikolas Klauser       return std::inclusive_scan(__first, __last, __result, __b, __init);
44e78f53d1SNikolas Klauser   }
45e78f53d1SNikolas Klauser 
46e78f53d1SNikolas Klauser   return __result;
47e78f53d1SNikolas Klauser }
48e78f53d1SNikolas Klauser 
49e78f53d1SNikolas Klauser template <class _InputIterator, class _OutputIterator>
50e78f53d1SNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
51e78f53d1SNikolas Klauser inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result) {
52e78f53d1SNikolas Klauser   return std::inclusive_scan(__first, __last, __result, std::plus<>());
53e78f53d1SNikolas Klauser }
54e78f53d1SNikolas Klauser 
55e78f53d1SNikolas Klauser #endif // _LIBCPP_STD_VER >= 17
56e78f53d1SNikolas Klauser 
57e78f53d1SNikolas Klauser _LIBCPP_END_NAMESPACE_STD
58e78f53d1SNikolas Klauser 
59*ce777190SNikolas Klauser #endif // _LIBCPP___CXX03___NUMERIC_INCLUSIVE_SCAN_H
60