151 lines
5.6 KiB
C++
151 lines
5.6 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// <algorithm>
|
|
|
|
// template<ForwardIterator Iter1, ForwardIterator Iter2>
|
|
// requires HasEqualTo<Iter1::value_type, Iter2::value_type>
|
|
// constexpr Iter1 // constexpr after C++17
|
|
// search(Iter1 first1, Iter1 last1, Iter2 first2, Iter2 last2);
|
|
//
|
|
// template<class ForwardIterator, class Searcher>
|
|
// ForwardIterator search(ForwardIterator first, ForwardIterator last,
|
|
// const Searcher& searcher); // C++17
|
|
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
|
|
#include "test_macros.h"
|
|
#include "test_iterators.h"
|
|
|
|
struct MySearcherC {
|
|
template <typename Iterator>
|
|
std::pair<Iterator, Iterator>
|
|
TEST_CONSTEXPR operator() (Iterator b, Iterator e) const
|
|
{
|
|
return std::make_pair(b, e);
|
|
}
|
|
};
|
|
|
|
#if TEST_STD_VER > 17
|
|
TEST_CONSTEXPR bool test_constexpr() {
|
|
int ia[] = {0, 1, 2, 3};
|
|
int ib[] = {0, 1, 5, 3};
|
|
int ic[] = {0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4};
|
|
return (std::search(std::begin(ic), std::end(ic), std::begin(ia), std::end(ia)) == ic+3)
|
|
&& (std::search(std::begin(ic), std::end(ic), std::begin(ib), std::end(ib)) == std::end(ic))
|
|
&& (std::search(std::begin(ic), std::end(ic), MySearcherC()) == std::begin(ic))
|
|
;
|
|
}
|
|
#endif
|
|
|
|
int searcher_called = 0;
|
|
|
|
struct MySearcher {
|
|
template <typename Iterator>
|
|
std::pair<Iterator, Iterator>
|
|
operator() (Iterator b, Iterator e) const
|
|
{
|
|
++searcher_called;
|
|
return std::make_pair(b, e);
|
|
}
|
|
};
|
|
|
|
namespace User {
|
|
struct S {
|
|
S(int x) : x_(x) {}
|
|
int x_;
|
|
};
|
|
bool operator==(S lhs, S rhs)
|
|
{
|
|
return lhs.x_ == rhs.x_;
|
|
}
|
|
template <class T, class U>
|
|
void make_pair(T&&, U&&) = delete;
|
|
} // namespace User
|
|
|
|
template <class Iter1, class Iter2>
|
|
void
|
|
test()
|
|
{
|
|
int ia[] = {0, 1, 2, 3, 4, 5};
|
|
const unsigned sa = sizeof(ia)/sizeof(ia[0]);
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia)) == Iter1(ia));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia+1)) == Iter1(ia));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+1), Iter2(ia+2)) == Iter1(ia+1));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+2)) == Iter1(ia));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+3)) == Iter1(ia+2));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+2), Iter2(ia+3)) == Iter1(ia+2));
|
|
assert(std::search(Iter1(ia), Iter1(ia), Iter2(ia+2), Iter2(ia+3)) == Iter1(ia));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+sa-1), Iter2(ia+sa)) == Iter1(ia+sa-1));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia+sa-3), Iter2(ia+sa)) == Iter1(ia+sa-3));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa), Iter2(ia), Iter2(ia+sa)) == Iter1(ia));
|
|
assert(std::search(Iter1(ia), Iter1(ia+sa-1), Iter2(ia), Iter2(ia+sa)) == Iter1(ia+sa-1));
|
|
assert(std::search(Iter1(ia), Iter1(ia+1), Iter2(ia), Iter2(ia+sa)) == Iter1(ia+1));
|
|
int ib[] = {0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4};
|
|
const unsigned sb = sizeof(ib)/sizeof(ib[0]);
|
|
int ic[] = {1};
|
|
assert(std::search(Iter1(ib), Iter1(ib+sb), Iter2(ic), Iter2(ic+1)) == Iter1(ib+1));
|
|
int id[] = {1, 2};
|
|
assert(std::search(Iter1(ib), Iter1(ib+sb), Iter2(id), Iter2(id+2)) == Iter1(ib+1));
|
|
int ie[] = {1, 2, 3};
|
|
assert(std::search(Iter1(ib), Iter1(ib+sb), Iter2(ie), Iter2(ie+3)) == Iter1(ib+4));
|
|
int ig[] = {1, 2, 3, 4};
|
|
assert(std::search(Iter1(ib), Iter1(ib+sb), Iter2(ig), Iter2(ig+4)) == Iter1(ib+8));
|
|
int ih[] = {0, 1, 1, 1, 1, 2, 3, 0, 1, 2, 3, 4};
|
|
const unsigned sh = sizeof(ih)/sizeof(ih[0]);
|
|
int ii[] = {1, 1, 2};
|
|
assert(std::search(Iter1(ih), Iter1(ih+sh), Iter2(ii), Iter2(ii+3)) == Iter1(ih+3));
|
|
int ij[] = {0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0};
|
|
const unsigned sj = sizeof(ij)/sizeof(ij[0]);
|
|
int ik[] = {0, 0, 0, 0, 1, 1, 1, 1, 0, 0};
|
|
const unsigned sk = sizeof(ik)/sizeof(ik[0]);
|
|
assert(std::search(Iter1(ij), Iter1(ij+sj), Iter2(ik), Iter2(ik+sk)) == Iter1(ij+6));
|
|
}
|
|
|
|
template <class Iter>
|
|
void
|
|
adl_test()
|
|
{
|
|
User::S ua[] = {1};
|
|
assert(std::search(Iter(ua), Iter(ua), Iter(ua), Iter(ua)) == Iter(ua));
|
|
}
|
|
|
|
int main(int, char**)
|
|
{
|
|
test<forward_iterator<const int*>, forward_iterator<const int*> >();
|
|
test<forward_iterator<const int*>, bidirectional_iterator<const int*> >();
|
|
test<forward_iterator<const int*>, random_access_iterator<const int*> >();
|
|
test<bidirectional_iterator<const int*>, forward_iterator<const int*> >();
|
|
test<bidirectional_iterator<const int*>, bidirectional_iterator<const int*> >();
|
|
test<bidirectional_iterator<const int*>, random_access_iterator<const int*> >();
|
|
test<random_access_iterator<const int*>, forward_iterator<const int*> >();
|
|
test<random_access_iterator<const int*>, bidirectional_iterator<const int*> >();
|
|
test<random_access_iterator<const int*>, random_access_iterator<const int*> >();
|
|
|
|
adl_test<forward_iterator<User::S*> >();
|
|
adl_test<random_access_iterator<User::S*> >();
|
|
|
|
#if TEST_STD_VER > 14
|
|
{
|
|
typedef int * RI;
|
|
static_assert((std::is_same<RI, decltype(std::search(RI(), RI(), MySearcher()))>::value), "" );
|
|
|
|
RI it(nullptr);
|
|
assert(it == std::search(it, it, MySearcher()));
|
|
assert(searcher_called == 1);
|
|
}
|
|
#endif
|
|
|
|
#if TEST_STD_VER > 17
|
|
static_assert(test_constexpr());
|
|
#endif
|
|
|
|
return 0;
|
|
}
|