xref: /llvm-project/libcxx/include/__algorithm/generate.h (revision 7ed7d4ccb8991e2b5b95334b508f8cec2faee737)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef _LIBCPP___ALGORITHM_GENERATE_H
10 #define _LIBCPP___ALGORITHM_GENERATE_H
11 
12 #include <__config>
13 #include <type_traits>
14 
15 #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
16 #pragma GCC system_header
17 #endif
18 
19 _LIBCPP_PUSH_MACROS
20 #include <__undef_macros>
21 
22 _LIBCPP_BEGIN_NAMESPACE_STD
23 
24 template <class _ForwardIterator, class _Generator>
25 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
26 void
27 generate(_ForwardIterator __first, _ForwardIterator __last, _Generator __gen)
28 {
29     for (; __first != __last; ++__first)
30         *__first = __gen();
31 }
32 
33 // generate_n
34 
35 template <class _OutputIterator, class _Size, class _Generator>
36 inline _LIBCPP_INLINE_VISIBILITY _LIBCPP_CONSTEXPR_AFTER_CXX17
37 _OutputIterator
38 generate_n(_OutputIterator __first, _Size __orig_n, _Generator __gen)
39 {
40     typedef decltype(_VSTD::__convert_to_integral(__orig_n)) _IntegralSize;
41     _IntegralSize __n = __orig_n;
42     for (; __n > 0; ++__first, (void) --__n)
43         *__first = __gen();
44     return __first;
45 }
46 
47 _LIBCPP_END_NAMESPACE_STD
48 
49 _LIBCPP_POP_MACROS
50 
51 #endif // _LIBCPP___ALGORITHM_GENERATE_H
52