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 SUPPORT_FROM_RANGE_HELPERS_H |
10 | #define SUPPORT_FROM_RANGE_HELPERS_H |
11 | |
12 | #include <array> |
13 | #include <cstddef> |
14 | #include <iterator> |
15 | #include <type_traits> |
16 | #include <vector> |
17 | |
18 | #include "min_allocator.h" |
19 | #include "test_allocator.h" |
20 | #include "test_iterators.h" |
21 | #include "test_macros.h" |
22 | #include "type_algorithms.h" |
23 | |
24 | struct Empty {}; |
25 | |
26 | template <class T> |
27 | struct InputRange { |
28 | cpp20_input_iterator<T*> begin(); |
29 | sentinel_wrapper<cpp20_input_iterator<T*>> end(); |
30 | }; |
31 | |
32 | template <class Iter, class Sent, std::ranges::input_range Range> |
33 | constexpr auto wrap_input(Range&& input) { |
34 | auto b = Iter(std::ranges::begin(input)); |
35 | auto e = Sent(Iter(std::ranges::end(input))); |
36 | return std::ranges::subrange(std::move(b), std::move(e)); |
37 | } |
38 | |
39 | template <class Iter, class Sent, class T, std::size_t N> |
40 | constexpr auto wrap_input(std::array<T, N>& input) { |
41 | auto b = Iter(input.data()); |
42 | auto e = Sent(Iter(input.data() + input.size())); |
43 | return std::ranges::subrange(std::move(b), std::move(e)); |
44 | } |
45 | |
46 | template <class Iter, class Sent, class T> |
47 | constexpr auto wrap_input(std::vector<T>& input) { |
48 | auto b = Iter(input.data()); |
49 | auto e = Sent(Iter(input.data() + input.size())); |
50 | return std::ranges::subrange(std::move(b), std::move(e)); |
51 | } |
52 | |
53 | struct KeyValue { |
54 | int key; // Only the key is considered for equality comparison. |
55 | char value; // Allows distinguishing equivalent instances. |
56 | |
57 | bool operator<(const KeyValue& other) const { return key < other.key; } |
58 | bool operator==(const KeyValue& other) const { return key == other.key; } |
59 | }; |
60 | |
61 | template <> |
62 | struct std::hash<KeyValue> { |
63 | std::size_t operator()(const KeyValue& kv) const { return kv.key; } |
64 | }; |
65 | |
66 | #if !defined(TEST_HAS_NO_EXCEPTIONS) |
67 | |
68 | template <class T> |
69 | struct ThrowingAllocator { |
70 | using value_type = T; |
71 | using char_type = T; |
72 | using is_always_equal = std::false_type; |
73 | |
74 | ThrowingAllocator() = default; |
75 | |
76 | template <class U> |
77 | ThrowingAllocator(const ThrowingAllocator<U>&) {} |
78 | |
79 | T* allocate(std::size_t) { throw 1; } |
80 | void deallocate(T*, std::size_t) {} |
81 | |
82 | template <class U> |
83 | friend bool operator==(const ThrowingAllocator&, const ThrowingAllocator<U>&) { |
84 | return true; |
85 | } |
86 | }; |
87 | #endif |
88 | |
89 | template <class T, class Func> |
90 | constexpr void for_all_iterators_and_allocators(Func f) { |
91 | using Iterators = |
92 | types::type_list< cpp20_input_iterator<T*>, |
93 | forward_iterator<T*>, |
94 | bidirectional_iterator<T*>, |
95 | random_access_iterator<T*>, |
96 | contiguous_iterator<T*>, |
97 | T* >; |
98 | |
99 | types::for_each(Iterators{}, [=]<class Iter>() { |
100 | f.template operator()<Iter, sentinel_wrapper<Iter>, std::allocator<T>>(); |
101 | f.template operator()<Iter, sentinel_wrapper<Iter>, test_allocator<T>>(); |
102 | f.template operator()<Iter, sentinel_wrapper<Iter>, min_allocator<T>>(); |
103 | f.template operator()<Iter, sentinel_wrapper<Iter>, safe_allocator<T>>(); |
104 | |
105 | if constexpr (std::sentinel_for<Iter, Iter>) { |
106 | f.template operator()<Iter, Iter, std::allocator<T>>(); |
107 | f.template operator()<Iter, Iter, test_allocator<T>>(); |
108 | f.template operator()<Iter, Iter, min_allocator<T>>(); |
109 | f.template operator()<Iter, Iter, safe_allocator<T>>(); |
110 | } |
111 | }); |
112 | } |
113 | |
114 | #endif // SUPPORT_FROM_RANGE_HELPERS_H |
115 | |