1*8a48e6ddSzoecarver //===----------------------------------------------------------------------===//
2*8a48e6ddSzoecarver //
3*8a48e6ddSzoecarver // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*8a48e6ddSzoecarver // See https://llvm.org/LICENSE.txt for license information.
5*8a48e6ddSzoecarver // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*8a48e6ddSzoecarver //
7*8a48e6ddSzoecarver //===----------------------------------------------------------------------===//
8*8a48e6ddSzoecarver 
9*8a48e6ddSzoecarver // UNSUPPORTED: c++03, c++11, c++14, c++17
10*8a48e6ddSzoecarver 
11*8a48e6ddSzoecarver // constexpr decltype(auto) operator[](iter_difference_t<I> n) const
12*8a48e6ddSzoecarver //   requires random_access_iterator<I>;
13*8a48e6ddSzoecarver 
14*8a48e6ddSzoecarver #include <iterator>
15*8a48e6ddSzoecarver 
16*8a48e6ddSzoecarver #include "test_macros.h"
17*8a48e6ddSzoecarver #include "test_iterators.h"
18*8a48e6ddSzoecarver 
19*8a48e6ddSzoecarver template<class Iter>
20*8a48e6ddSzoecarver concept SubscriptEnabled = requires(Iter& iter) {
21*8a48e6ddSzoecarver   iter[1];
22*8a48e6ddSzoecarver };
23*8a48e6ddSzoecarver 
test()24*8a48e6ddSzoecarver constexpr bool test() {
25*8a48e6ddSzoecarver   int buffer[8] = {1, 2, 3, 4, 5, 6, 7, 8};
26*8a48e6ddSzoecarver 
27*8a48e6ddSzoecarver   {
28*8a48e6ddSzoecarver     std::counted_iterator iter(random_access_iterator<int*>{buffer}, 8);
29*8a48e6ddSzoecarver     for (int i = 1; i < 9; ++i)
30*8a48e6ddSzoecarver       assert(iter[i - 1] == i);
31*8a48e6ddSzoecarver   }
32*8a48e6ddSzoecarver   {
33*8a48e6ddSzoecarver     std::counted_iterator iter(contiguous_iterator<int*>{buffer}, 8);
34*8a48e6ddSzoecarver     for (int i = 1; i < 9; ++i)
35*8a48e6ddSzoecarver       assert(iter[i - 1] == i);
36*8a48e6ddSzoecarver   }
37*8a48e6ddSzoecarver 
38*8a48e6ddSzoecarver   {
39*8a48e6ddSzoecarver     const std::counted_iterator iter(random_access_iterator<int*>{buffer}, 8);
40*8a48e6ddSzoecarver     assert(iter[0] == 1);
41*8a48e6ddSzoecarver   }
42*8a48e6ddSzoecarver   {
43*8a48e6ddSzoecarver     const std::counted_iterator iter(contiguous_iterator<int*>{buffer}, 8);
44*8a48e6ddSzoecarver     assert(iter[7] == 8);
45*8a48e6ddSzoecarver   }
46*8a48e6ddSzoecarver 
47*8a48e6ddSzoecarver   {
48*8a48e6ddSzoecarver       static_assert( SubscriptEnabled<std::counted_iterator<random_access_iterator<int*>>>);
49*8a48e6ddSzoecarver       static_assert(!SubscriptEnabled<std::counted_iterator<bidirectional_iterator<int*>>>);
50*8a48e6ddSzoecarver   }
51*8a48e6ddSzoecarver 
52*8a48e6ddSzoecarver   return true;
53*8a48e6ddSzoecarver }
54*8a48e6ddSzoecarver 
main(int,char **)55*8a48e6ddSzoecarver int main(int, char**) {
56*8a48e6ddSzoecarver   test();
57*8a48e6ddSzoecarver   static_assert(test());
58*8a48e6ddSzoecarver 
59*8a48e6ddSzoecarver   return 0;
60*8a48e6ddSzoecarver }
61