| 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 | // UNSUPPORTED: c++03, c++11, c++14, c++17 |
| 10 | |
| 11 | // template<bool OtherConst> |
| 12 | // requires sentinel_for<sentinel_t<Base>, iterator_t<maybe-const<OtherConst, V>>> |
| 13 | // friend constexpr bool operator==(const iterator<OtherConst>& x, const sentinel& y); |
| 14 | |
| 15 | #include <cassert> |
| 16 | #include <concepts> |
| 17 | #include <functional> |
| 18 | #include <ranges> |
| 19 | #include <type_traits> |
| 20 | |
| 21 | #include "../types.h" |
| 22 | #include "test_range.h" |
| 23 | |
| 24 | using Iterator = random_access_iterator<BufferView<int*>*>; |
| 25 | using ConstIterator = random_access_iterator<const BufferView<int*>*>; |
| 26 | |
| 27 | template <bool Const> |
| 28 | struct ConstComparableSentinel { |
| 29 | |
| 30 | using Iter = std::conditional_t<Const, ConstIterator, Iterator>; |
| 31 | Iter iter_; |
| 32 | |
| 33 | explicit ConstComparableSentinel() = default; |
| 34 | constexpr explicit ConstComparableSentinel(const Iter& it) : iter_(it) {} |
| 35 | |
| 36 | constexpr friend bool operator==(const Iterator& i, const ConstComparableSentinel& s) { |
| 37 | return base(i) == base(s.iter_); |
| 38 | } |
| 39 | |
| 40 | constexpr friend bool operator==(const ConstIterator& i, const ConstComparableSentinel& s) { |
| 41 | return base(i) == base(s.iter_); |
| 42 | } |
| 43 | }; |
| 44 | |
| 45 | struct ConstComparableView : BufferView<BufferView<int*>*> { |
| 46 | using BufferView<BufferView<int*>*>::BufferView; |
| 47 | |
| 48 | constexpr auto begin() { return Iterator(data_); } |
| 49 | constexpr auto begin() const { return ConstIterator(data_); } |
| 50 | constexpr auto end() { return ConstComparableSentinel<false>(Iterator(data_ + size_)); } |
| 51 | constexpr auto end() const { return ConstComparableSentinel<true>(ConstIterator(data_ + size_)); } |
| 52 | }; |
| 53 | |
| 54 | static_assert(weakly_equality_comparable_with<std::ranges::iterator_t<ConstComparableView>, |
| 55 | std::ranges::sentinel_t<const ConstComparableView>>); |
| 56 | static_assert(weakly_equality_comparable_with<std::ranges::iterator_t<const ConstComparableView>, |
| 57 | std::ranges::sentinel_t<ConstComparableView>>); |
| 58 | |
| 59 | constexpr bool test() { |
| 60 | int buffer[4][4] = {{1111, 2222, 3333, 4444}, {555, 666, 777, 888}, {99, 1010, 1111, 1212}, {13, 14, 15, 16}}; |
| 61 | |
| 62 | // test iterator<false> == sentinel<false> |
| 63 | { |
| 64 | ChildView children[4] = {ChildView(buffer[0]), ChildView(buffer[1]), ChildView(buffer[2]), ChildView(buffer[3])}; |
| 65 | auto jv = std::ranges::join_view(ParentView(children)); |
| 66 | assert(jv.end() == std::ranges::next(jv.begin(), 16)); |
| 67 | } |
| 68 | |
| 69 | // test iterator<false> == sentinel<true> |
| 70 | { |
| 71 | ChildView children[4] = {ChildView(buffer[0]), ChildView(buffer[1]), ChildView(buffer[2]), ChildView(buffer[3])}; |
| 72 | using ParentT = std::remove_all_extents_t<decltype(children)>; |
| 73 | auto jv = std::ranges::join_view(ForwardParentView<ParentT>(children)); |
| 74 | assert(std::as_const(jv).end() == std::ranges::next(jv.begin(), 16)); |
| 75 | } |
| 76 | |
| 77 | // test iterator<true> == sentinel<true> |
| 78 | { |
| 79 | CopyableChild children[4] = {CopyableChild(buffer[0]), CopyableChild(buffer[1]), CopyableChild(buffer[2]), |
| 80 | CopyableChild(buffer[3])}; |
| 81 | using ParentT = std::remove_all_extents_t<decltype(children)>; |
| 82 | const auto jv = std::ranges::join_view(ForwardParentView<ParentT>(children)); |
| 83 | assert(jv.end() == std::ranges::next(jv.begin(), 16)); |
| 84 | } |
| 85 | |
| 86 | // test iterator<Const> == sentinel<!Const> |
| 87 | { |
| 88 | BufferView<int*> inners[] = {buffer[0], buffer[1]}; |
| 89 | ConstComparableView outer(inners); |
| 90 | auto jv = std::ranges::join_view(outer); |
| 91 | assert(jv.end() == std::ranges::next(jv.begin(), 8)); |
| 92 | assert(std::as_const(jv).end() == std::ranges::next(jv.begin(), 8)); |
| 93 | assert(jv.end() == std::ranges::next(std::as_const(jv).begin(), 8)); |
| 94 | } |
| 95 | |
| 96 | return true; |
| 97 | } |
| 98 | |
| 99 | int main(int, char**) { |
| 100 | test(); |
| 101 | static_assert(test()); |
| 102 | |
| 103 | return 0; |
| 104 | } |
| 105 | |