1a8cf78c7SLouis Dionne //===----------------------------------------------------------------------===//
2a8cf78c7SLouis Dionne //
3a8cf78c7SLouis Dionne // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4a8cf78c7SLouis Dionne // See https://llvm.org/LICENSE.txt for license information.
5a8cf78c7SLouis Dionne // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6a8cf78c7SLouis Dionne //
7a8cf78c7SLouis Dionne //===----------------------------------------------------------------------===//
8a8cf78c7SLouis Dionne // UNSUPPORTED: c++03, c++11, c++14, c++17
9a8cf78c7SLouis Dionne 
10a8cf78c7SLouis Dionne // <span>
11a8cf78c7SLouis Dionne 
12a8cf78c7SLouis Dionne //   template<class It, class EndOrSize>
13a8cf78c7SLouis Dionne //     span(It, EndOrSize) -> span<remove_reference_t<iter_reference_t<_It>>>;
14a8cf78c7SLouis Dionne //
15a8cf78c7SLouis Dionne //   template<class T, size_t N>
16a8cf78c7SLouis Dionne //     span(T (&)[N]) -> span<T, N>;
17a8cf78c7SLouis Dionne //
18a8cf78c7SLouis Dionne //   template<class T, size_t N>
19a8cf78c7SLouis Dionne //     span(array<T, N>&) -> span<T, N>;
20a8cf78c7SLouis Dionne //
21a8cf78c7SLouis Dionne //   template<class T, size_t N>
22a8cf78c7SLouis Dionne //     span(const array<T, N>&) -> span<const T, N>;
23a8cf78c7SLouis Dionne //
24a8cf78c7SLouis Dionne //   template<class R>
25a8cf78c7SLouis Dionne //     span(R&&) -> span<remove_reference_t<ranges::range_reference_t<R>>>;
26a8cf78c7SLouis Dionne 
27a8cf78c7SLouis Dionne 
28a8cf78c7SLouis Dionne #include <span>
29a8cf78c7SLouis Dionne #include <array>
30a8cf78c7SLouis Dionne #include <cassert>
31*3cd4531bSNikolas Klauser #include <iterator>
32a8cf78c7SLouis Dionne #include <memory>
33a8cf78c7SLouis Dionne #include <string>
34a8cf78c7SLouis Dionne 
35a8cf78c7SLouis Dionne #include "test_macros.h"
36a8cf78c7SLouis Dionne 
test_iterator_sentinel()37a8cf78c7SLouis Dionne void test_iterator_sentinel() {
38a8cf78c7SLouis Dionne   int arr[] = {1, 2, 3};
39a8cf78c7SLouis Dionne   {
40a8cf78c7SLouis Dionne   std::span s{std::begin(arr), std::end(arr)};
41a8cf78c7SLouis Dionne   ASSERT_SAME_TYPE(decltype(s), std::span<int>);
42a8cf78c7SLouis Dionne   assert(s.size() == std::size(arr));
43a8cf78c7SLouis Dionne   assert(s.data() == std::data(arr));
44a8cf78c7SLouis Dionne   }
45a8cf78c7SLouis Dionne   {
46a8cf78c7SLouis Dionne   std::span s{std::begin(arr), 3};
47a8cf78c7SLouis Dionne   ASSERT_SAME_TYPE(decltype(s), std::span<int>);
48a8cf78c7SLouis Dionne   assert(s.size() == std::size(arr));
49a8cf78c7SLouis Dionne   assert(s.data() == std::data(arr));
50a8cf78c7SLouis Dionne   }
51a8cf78c7SLouis Dionne }
52a8cf78c7SLouis Dionne 
test_c_array()53a8cf78c7SLouis Dionne void test_c_array() {
54a8cf78c7SLouis Dionne     {
55a8cf78c7SLouis Dionne     int arr[] = {1, 2, 3};
56a8cf78c7SLouis Dionne     std::span s{arr};
57a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<int, 3>);
58a8cf78c7SLouis Dionne     assert(s.size() == std::size(arr));
59a8cf78c7SLouis Dionne     assert(s.data() == std::data(arr));
60a8cf78c7SLouis Dionne     }
61a8cf78c7SLouis Dionne 
62a8cf78c7SLouis Dionne     {
63a8cf78c7SLouis Dionne     const int arr[] = {1,2,3};
64a8cf78c7SLouis Dionne     std::span s{arr};
65a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<const int, 3>);
66a8cf78c7SLouis Dionne     assert(s.size() == std::size(arr));
67a8cf78c7SLouis Dionne     assert(s.data() == std::data(arr));
68a8cf78c7SLouis Dionne     }
69a8cf78c7SLouis Dionne }
70a8cf78c7SLouis Dionne 
test_std_array()71a8cf78c7SLouis Dionne void test_std_array() {
72a8cf78c7SLouis Dionne     {
73a8cf78c7SLouis Dionne     std::array<double, 4> arr = {1.0, 2.0, 3.0, 4.0};
74a8cf78c7SLouis Dionne     std::span s{arr};
75a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<double, 4>);
76a8cf78c7SLouis Dionne     assert(s.size() == arr.size());
77a8cf78c7SLouis Dionne     assert(s.data() == arr.data());
78a8cf78c7SLouis Dionne     }
79a8cf78c7SLouis Dionne 
80a8cf78c7SLouis Dionne     {
81a8cf78c7SLouis Dionne     const std::array<long, 5> arr = {4, 5, 6, 7, 8};
82a8cf78c7SLouis Dionne     std::span s{arr};
83a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<const long, 5>);
84a8cf78c7SLouis Dionne     assert(s.size() == arr.size());
85a8cf78c7SLouis Dionne     assert(s.data() == arr.data());
86a8cf78c7SLouis Dionne     }
87a8cf78c7SLouis Dionne }
88a8cf78c7SLouis Dionne 
test_range_std_container()89a8cf78c7SLouis Dionne void test_range_std_container() {
90a8cf78c7SLouis Dionne     {
91a8cf78c7SLouis Dionne     std::string str{"ABCDE"};
92a8cf78c7SLouis Dionne     std::span s{str};
93a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<char>);
94a8cf78c7SLouis Dionne     assert(s.size() == str.size());
95a8cf78c7SLouis Dionne     assert(s.data() == str.data());
96a8cf78c7SLouis Dionne     }
97a8cf78c7SLouis Dionne 
98a8cf78c7SLouis Dionne     {
99a8cf78c7SLouis Dionne     const std::string str{"QWERTYUIOP"};
100a8cf78c7SLouis Dionne     std::span s{str};
101a8cf78c7SLouis Dionne     ASSERT_SAME_TYPE(decltype(s), std::span<const char>);
102a8cf78c7SLouis Dionne     assert(s.size() == str.size());
103a8cf78c7SLouis Dionne     assert(s.data() == str.data());
104a8cf78c7SLouis Dionne     }
105a8cf78c7SLouis Dionne }
106a8cf78c7SLouis Dionne 
main(int,char **)107a8cf78c7SLouis Dionne int main(int, char**)
108a8cf78c7SLouis Dionne {
109a8cf78c7SLouis Dionne   test_iterator_sentinel();
110a8cf78c7SLouis Dionne   test_c_array();
111a8cf78c7SLouis Dionne   test_std_array();
112a8cf78c7SLouis Dionne   test_range_std_container();
113a8cf78c7SLouis Dionne 
114a8cf78c7SLouis Dionne   return 0;
115a8cf78c7SLouis Dionne }
116