xref: /llvm-project/libcxx/include/__cxx03/__numeric/partial_sum.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_PARTIAL_SUM_H
11*ce777190SNikolas Klauser #define _LIBCPP___CXX03___NUMERIC_PARTIAL_SUM_H
12e78f53d1SNikolas Klauser 
1373fbae83SNikolas Klauser #include <__cxx03/__config>
1473fbae83SNikolas Klauser #include <__cxx03/__iterator/iterator_traits.h>
1573fbae83SNikolas Klauser #include <__cxx03/__utility/move.h>
16e78f53d1SNikolas Klauser 
17e78f53d1SNikolas Klauser #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
18e78f53d1SNikolas Klauser #  pragma GCC system_header
19e78f53d1SNikolas Klauser #endif
20e78f53d1SNikolas Klauser 
21e78f53d1SNikolas Klauser _LIBCPP_PUSH_MACROS
2273fbae83SNikolas Klauser #include <__cxx03/__undef_macros>
23e78f53d1SNikolas Klauser 
24e78f53d1SNikolas Klauser _LIBCPP_BEGIN_NAMESPACE_STD
25e78f53d1SNikolas Klauser 
26e78f53d1SNikolas Klauser template <class _InputIterator, class _OutputIterator>
27e78f53d1SNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
28e78f53d1SNikolas Klauser partial_sum(_InputIterator __first, _InputIterator __last, _OutputIterator __result) {
29e78f53d1SNikolas Klauser   if (__first != __last) {
30e78f53d1SNikolas Klauser     typename iterator_traits<_InputIterator>::value_type __t(*__first);
31e78f53d1SNikolas Klauser     *__result = __t;
32e78f53d1SNikolas Klauser     for (++__first, (void)++__result; __first != __last; ++__first, (void)++__result) {
33e78f53d1SNikolas Klauser #if _LIBCPP_STD_VER >= 20
34e78f53d1SNikolas Klauser       __t = std::move(__t) + *__first;
35e78f53d1SNikolas Klauser #else
36e78f53d1SNikolas Klauser       __t = __t + *__first;
37e78f53d1SNikolas Klauser #endif
38e78f53d1SNikolas Klauser       *__result = __t;
39e78f53d1SNikolas Klauser     }
40e78f53d1SNikolas Klauser   }
41e78f53d1SNikolas Klauser   return __result;
42e78f53d1SNikolas Klauser }
43e78f53d1SNikolas Klauser 
44e78f53d1SNikolas Klauser template <class _InputIterator, class _OutputIterator, class _BinaryOperation>
45e78f53d1SNikolas Klauser _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator
46e78f53d1SNikolas Klauser partial_sum(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOperation __binary_op) {
47e78f53d1SNikolas Klauser   if (__first != __last) {
48e78f53d1SNikolas Klauser     typename iterator_traits<_InputIterator>::value_type __t(*__first);
49e78f53d1SNikolas Klauser     *__result = __t;
50e78f53d1SNikolas Klauser     for (++__first, (void)++__result; __first != __last; ++__first, (void)++__result) {
51e78f53d1SNikolas Klauser #if _LIBCPP_STD_VER >= 20
52e78f53d1SNikolas Klauser       __t = __binary_op(std::move(__t), *__first);
53e78f53d1SNikolas Klauser #else
54e78f53d1SNikolas Klauser       __t = __binary_op(__t, *__first);
55e78f53d1SNikolas Klauser #endif
56e78f53d1SNikolas Klauser       *__result = __t;
57e78f53d1SNikolas Klauser     }
58e78f53d1SNikolas Klauser   }
59e78f53d1SNikolas Klauser   return __result;
60e78f53d1SNikolas Klauser }
61e78f53d1SNikolas Klauser 
62e78f53d1SNikolas Klauser _LIBCPP_END_NAMESPACE_STD
63e78f53d1SNikolas Klauser 
64e78f53d1SNikolas Klauser _LIBCPP_POP_MACROS
65e78f53d1SNikolas Klauser 
66*ce777190SNikolas Klauser #endif // _LIBCPP___CXX03___NUMERIC_PARTIAL_SUM_H
67