14824e7fdSDimitry Andric // -*- C++ -*- 24824e7fdSDimitry Andric //===----------------------------------------------------------------------===// 34824e7fdSDimitry Andric // 44824e7fdSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 54824e7fdSDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 64824e7fdSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 74824e7fdSDimitry Andric // 84824e7fdSDimitry Andric //===----------------------------------------------------------------------===// 94824e7fdSDimitry Andric 104824e7fdSDimitry Andric #ifndef _LIBCPP___NUMERIC_INCLUSIVE_SCAN_H 114824e7fdSDimitry Andric #define _LIBCPP___NUMERIC_INCLUSIVE_SCAN_H 124824e7fdSDimitry Andric 134824e7fdSDimitry Andric #include <__config> 144824e7fdSDimitry Andric #include <__functional/operations.h> 154824e7fdSDimitry Andric #include <__iterator/iterator_traits.h> 164824e7fdSDimitry Andric #include <__utility/move.h> 174824e7fdSDimitry Andric 184824e7fdSDimitry Andric #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) 194824e7fdSDimitry Andric # pragma GCC system_header 204824e7fdSDimitry Andric #endif 214824e7fdSDimitry Andric 224824e7fdSDimitry Andric _LIBCPP_BEGIN_NAMESPACE_STD 234824e7fdSDimitry Andric 2406c3fb27SDimitry Andric #if _LIBCPP_STD_VER >= 17 254824e7fdSDimitry Andric 264824e7fdSDimitry Andric template <class _InputIterator, class _OutputIterator, class _Tp, class _BinaryOp> 27*5f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator 284824e7fdSDimitry Andric inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b, _Tp __init) { 294824e7fdSDimitry Andric for (; __first != __last; ++__first, (void)++__result) { 304824e7fdSDimitry Andric __init = __b(__init, *__first); 314824e7fdSDimitry Andric *__result = __init; 324824e7fdSDimitry Andric } 334824e7fdSDimitry Andric return __result; 344824e7fdSDimitry Andric } 354824e7fdSDimitry Andric 364824e7fdSDimitry Andric template <class _InputIterator, class _OutputIterator, class _BinaryOp> 37*5f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator 384824e7fdSDimitry Andric inclusive_scan(_InputIterator __first, _InputIterator __last, _OutputIterator __result, _BinaryOp __b) { 394824e7fdSDimitry Andric if (__first != __last) { 404824e7fdSDimitry Andric typename iterator_traits<_InputIterator>::value_type __init = *__first; 414824e7fdSDimitry Andric *__result++ = __init; 424824e7fdSDimitry Andric if (++__first != __last) 43*5f757f3fSDimitry Andric return std::inclusive_scan(__first, __last, __result, __b, __init); 444824e7fdSDimitry Andric } 454824e7fdSDimitry Andric 464824e7fdSDimitry Andric return __result; 474824e7fdSDimitry Andric } 484824e7fdSDimitry Andric 494824e7fdSDimitry Andric template <class _InputIterator, class _OutputIterator> 50*5f757f3fSDimitry Andric _LIBCPP_HIDE_FROM_ABI _LIBCPP_CONSTEXPR_SINCE_CXX20 _OutputIterator inclusive_scan(_InputIterator __first, 514824e7fdSDimitry Andric _InputIterator __last, 524824e7fdSDimitry Andric _OutputIterator __result) { 53*5f757f3fSDimitry Andric return std::inclusive_scan(__first, __last, __result, std::plus<>()); 544824e7fdSDimitry Andric } 554824e7fdSDimitry Andric 5606c3fb27SDimitry Andric #endif // _LIBCPP_STD_VER >= 17 574824e7fdSDimitry Andric 584824e7fdSDimitry Andric _LIBCPP_END_NAMESPACE_STD 594824e7fdSDimitry Andric 604824e7fdSDimitry Andric #endif // _LIBCPP___NUMERIC_INCLUSIVE_SCAN_H 61