xref: /freebsd-src/contrib/llvm-project/libcxx/include/__algorithm/ranges_generate_n.h (revision b3edf4467982447620505a28fc82e38a414c07dc)
1753f127fSDimitry Andric //===----------------------------------------------------------------------===//
2753f127fSDimitry Andric //
3753f127fSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4753f127fSDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5753f127fSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6753f127fSDimitry Andric //
7753f127fSDimitry Andric //===----------------------------------------------------------------------===//
8753f127fSDimitry Andric 
9753f127fSDimitry Andric #ifndef _LIBCPP___ALGORITHM_RANGES_GENERATE_N_H
10753f127fSDimitry Andric #define _LIBCPP___ALGORITHM_RANGES_GENERATE_N_H
11753f127fSDimitry Andric 
12753f127fSDimitry Andric #include <__concepts/constructible.h>
13753f127fSDimitry Andric #include <__concepts/invocable.h>
14753f127fSDimitry Andric #include <__config>
15753f127fSDimitry Andric #include <__functional/identity.h>
16753f127fSDimitry Andric #include <__functional/invoke.h>
17753f127fSDimitry Andric #include <__iterator/concepts.h>
18753f127fSDimitry Andric #include <__iterator/incrementable_traits.h>
19753f127fSDimitry Andric #include <__iterator/iterator_traits.h>
20753f127fSDimitry Andric #include <__ranges/access.h>
21753f127fSDimitry Andric #include <__ranges/concepts.h>
22753f127fSDimitry Andric #include <__utility/move.h>
23753f127fSDimitry Andric 
24753f127fSDimitry Andric #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
25753f127fSDimitry Andric #  pragma GCC system_header
26753f127fSDimitry Andric #endif
27753f127fSDimitry Andric 
28*b3edf446SDimitry Andric _LIBCPP_PUSH_MACROS
29*b3edf446SDimitry Andric #include <__undef_macros>
30*b3edf446SDimitry Andric 
3106c3fb27SDimitry Andric #if _LIBCPP_STD_VER >= 20
32753f127fSDimitry Andric 
33753f127fSDimitry Andric _LIBCPP_BEGIN_NAMESPACE_STD
34753f127fSDimitry Andric 
35753f127fSDimitry Andric namespace ranges {
36753f127fSDimitry Andric namespace __generate_n {
37753f127fSDimitry Andric 
38753f127fSDimitry Andric struct __fn {
39753f127fSDimitry Andric   template <input_or_output_iterator _OutIter, copy_constructible _Func>
40753f127fSDimitry Andric     requires invocable<_Func&> && indirectly_writable<_OutIter, invoke_result_t<_Func&>>
4106c3fb27SDimitry Andric   _LIBCPP_HIDE_FROM_ABI constexpr _OutIter
operator__fn4206c3fb27SDimitry Andric   operator()(_OutIter __first, iter_difference_t<_OutIter> __n, _Func __gen) const {
43972a253aSDimitry Andric     for (; __n > 0; --__n) {
44972a253aSDimitry Andric       *__first = __gen();
45972a253aSDimitry Andric       ++__first;
46972a253aSDimitry Andric     }
47972a253aSDimitry Andric 
48972a253aSDimitry Andric     return __first;
49753f127fSDimitry Andric   }
50753f127fSDimitry Andric };
51753f127fSDimitry Andric 
52753f127fSDimitry Andric } // namespace __generate_n
53753f127fSDimitry Andric 
54753f127fSDimitry Andric inline namespace __cpo {
55753f127fSDimitry Andric inline constexpr auto generate_n = __generate_n::__fn{};
56753f127fSDimitry Andric } // namespace __cpo
57753f127fSDimitry Andric } // namespace ranges
58753f127fSDimitry Andric 
59753f127fSDimitry Andric _LIBCPP_END_NAMESPACE_STD
60753f127fSDimitry Andric 
6106c3fb27SDimitry Andric #endif // _LIBCPP_STD_VER >= 20
62753f127fSDimitry Andric 
63*b3edf446SDimitry Andric _LIBCPP_POP_MACROS
64*b3edf446SDimitry Andric 
65753f127fSDimitry Andric #endif // _LIBCPP___ALGORITHM_RANGES_GENERATE_N_H
66