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 | // <functional> |
10 | |
11 | // UNSUPPORTED: c++03, c++11, c++14 |
12 | |
13 | // default searcher |
14 | // template<class _ForwardIterator, class _BinaryPredicate = equal_to<>> |
15 | // class default_searcher { |
16 | // public: |
17 | // default_searcher(_ForwardIterator __f, _ForwardIterator __l, |
18 | // _BinaryPredicate __p = _BinaryPredicate()) |
19 | // : __first_(__f), __last_(__l), __pred_(__p) {} |
20 | // |
21 | // template <typename _ForwardIterator2> |
22 | // pair<_ForwardIterator2, _ForwardIterator2> |
23 | // operator () (_ForwardIterator2 __f, _ForwardIterator2 __l) const { |
24 | // return std::search(__f, __l, __first_, __last_, __pred_); |
25 | // } |
26 | // |
27 | // private: |
28 | // _ForwardIterator __first_; |
29 | // _ForwardIterator __last_; |
30 | // _BinaryPredicate __pred_; |
31 | // }; |
32 | |
33 | |
34 | #include <algorithm> |
35 | #include <functional> |
36 | #include <cassert> |
37 | |
38 | #include "test_macros.h" |
39 | #include "test_iterators.h" |
40 | |
41 | struct count_equal |
42 | { |
43 | int *count; |
44 | |
45 | template <class T> |
46 | TEST_CONSTEXPR_CXX14 bool operator()(const T& x, const T& y) const |
47 | {++*count; return x == y;} |
48 | }; |
49 | |
50 | template <typename Iter1, typename Iter2> |
51 | TEST_CONSTEXPR_CXX20 |
52 | void do_search(Iter1 b1, Iter1 e1, Iter2 b2, Iter2 e2, Iter1 result) { |
53 | int count = 0; |
54 | std::default_searcher<Iter2, count_equal> s{b2, e2, count_equal{.count: &count}}; |
55 | assert(result == std::search(b1, e1, s)); |
56 | auto d1 = std::distance(b1, e1); |
57 | auto d2 = std::distance(b2, e2); |
58 | assert((count >= 1) || (d2 == 0) || (d1 < d2)); |
59 | assert((d1 < d2) || count <= d1 * (d1 - d2 + 1)); |
60 | } |
61 | |
62 | template <class Iter1, class Iter2> |
63 | TEST_CONSTEXPR_CXX20 |
64 | bool test() |
65 | { |
66 | int ia[] = {0, 1, 2, 3, 4, 5}; |
67 | const unsigned sa = sizeof(ia)/sizeof(ia[0]); |
68 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia), Iter1(ia)); |
69 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia+1), Iter1(ia)); |
70 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+1), Iter2(ia+2), Iter1(ia+1)); |
71 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+2), Iter1(ia)); |
72 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+3), Iter1(ia+2)); |
73 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+3), Iter1(ia+2)); |
74 | do_search(Iter1(ia), Iter1(ia), Iter2(ia+2), Iter2(ia+3), Iter1(ia)); |
75 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+sa-1), Iter2(ia+sa), Iter1(ia+sa-1)); |
76 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia+sa-3), Iter2(ia+sa), Iter1(ia+sa-3)); |
77 | do_search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia+sa), Iter1(ia)); |
78 | do_search(Iter1(ia), Iter1(ia+sa-1), Iter2(ia), Iter2(ia+sa), Iter1(ia+sa-1)); |
79 | do_search(Iter1(ia), Iter1(ia+1), Iter2(ia), Iter2(ia+sa), Iter1(ia+1)); |
80 | int ib[] = {0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4}; |
81 | const unsigned sb = sizeof(ib)/sizeof(ib[0]); |
82 | int ic[] = {1}; |
83 | do_search(Iter1(ib), Iter1(ib+sb), Iter2(ic), Iter2(ic+1), Iter1(ib+1)); |
84 | int id[] = {1, 2}; |
85 | do_search(Iter1(ib), Iter1(ib+sb), Iter2(id), Iter2(id+2), Iter1(ib+1)); |
86 | int ie[] = {1, 2, 3}; |
87 | do_search(Iter1(ib), Iter1(ib+sb), Iter2(ie), Iter2(ie+3), Iter1(ib+4)); |
88 | int ig[] = {1, 2, 3, 4}; |
89 | do_search(Iter1(ib), Iter1(ib+sb), Iter2(ig), Iter2(ig+4), Iter1(ib+8)); |
90 | int ih[] = {0, 1, 1, 1, 1, 2, 3, 0, 1, 2, 3, 4}; |
91 | const unsigned sh = sizeof(ih)/sizeof(ih[0]); |
92 | int ii[] = {1, 1, 2}; |
93 | do_search(Iter1(ih), Iter1(ih+sh), Iter2(ii), Iter2(ii+3), Iter1(ih+3)); |
94 | |
95 | return true; |
96 | } |
97 | |
98 | int main(int, char**) { |
99 | test<forward_iterator<const int*>, forward_iterator<const int*> >(); |
100 | test<forward_iterator<const int*>, bidirectional_iterator<const int*> >(); |
101 | test<forward_iterator<const int*>, random_access_iterator<const int*> >(); |
102 | test<bidirectional_iterator<const int*>, forward_iterator<const int*> >(); |
103 | test<bidirectional_iterator<const int*>, bidirectional_iterator<const int*> >(); |
104 | test<bidirectional_iterator<const int*>, random_access_iterator<const int*> >(); |
105 | test<random_access_iterator<const int*>, forward_iterator<const int*> >(); |
106 | test<random_access_iterator<const int*>, bidirectional_iterator<const int*> >(); |
107 | test<random_access_iterator<const int*>, random_access_iterator<const int*> >(); |
108 | |
109 | #if TEST_STD_VER >= 20 |
110 | static_assert(test<forward_iterator<const int*>, forward_iterator<const int*>>()); |
111 | static_assert(test<forward_iterator<const int*>, bidirectional_iterator<const int*>>()); |
112 | static_assert(test<forward_iterator<const int*>, random_access_iterator<const int*>>()); |
113 | static_assert(test<bidirectional_iterator<const int*>, forward_iterator<const int*>>()); |
114 | static_assert(test<bidirectional_iterator<const int*>, bidirectional_iterator<const int*>>()); |
115 | static_assert(test<bidirectional_iterator<const int*>, random_access_iterator<const int*>>()); |
116 | static_assert(test<random_access_iterator<const int*>, forward_iterator<const int*>>()); |
117 | static_assert(test<random_access_iterator<const int*>, bidirectional_iterator<const int*>>()); |
118 | static_assert(test<random_access_iterator<const int*>, random_access_iterator<const int*>>()); |
119 | #endif |
120 | |
121 | return 0; |
122 | } |
123 | |