blob: 32cf828c0fa51541aa6cd427ec011dd582c41474 [file] [log] [blame]
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001///////////////////////////////////////////////////////////////////////////////
2//
3// Copyright (c) 2015 Microsoft Corporation. All rights reserved.
4//
5// This code is licensed under the MIT License (MIT).
6//
7// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
8// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
9// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
10// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
11// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
12// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
13// THE SOFTWARE.
14//
15///////////////////////////////////////////////////////////////////////////////
16
17#pragma once
18
Neil MacIntoshb63ec942015-11-04 12:42:27 -080019#ifndef GSL_SPAN_H
20#define GSL_SPAN_H
Treb Connell51da1362015-09-24 18:08:34 -070021
Olaf van der Spek550361c2015-11-12 10:23:56 +010022#include <algorithm>
23#include <array>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070024#include <cstddef>
25#include <cstdint>
Olaf van der Spek550361c2015-11-12 10:23:56 +010026#include <functional>
27#include <iterator>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070028#include <limits>
Olaf van der Spek550361c2015-11-12 10:23:56 +010029#include <new>
Anna Gringauze2cdedda2015-10-15 13:19:24 -070030#include <numeric>
Olaf van der Spek550361c2015-11-12 10:23:56 +010031#include <stdexcept>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070032#include <type_traits>
33#include <utility>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070034#include "fail_fast.h"
35
Neil MacIntoshd5316802015-09-30 21:54:08 -070036#ifdef _MSC_VER
37
38// No MSVC does constexpr fully yet
Gabriel Dos Reis6554e832015-09-28 05:10:44 -070039#pragma push_macro("constexpr")
40#define constexpr /* nothing */
Neil MacIntoshd5316802015-09-30 21:54:08 -070041
42
43// VS 2013 workarounds
44#if _MSC_VER <= 1800
45
Neil MacIntoshe9a96022015-11-03 18:56:55 -080046#pragma push_macro("GSL_MSVC_HAS_VARIADIC_CTOR_BUG")
47#define GSL_MSVC_HAS_VARIADIC_CTOR_BUG
48
49
Neil MacIntoshd5316802015-09-30 21:54:08 -070050// noexcept is not understood
51#ifndef GSL_THROWS_FOR_TESTING
52#define noexcept /* nothing */
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070053#endif
54
Neil MacIntoshd5316802015-09-30 21:54:08 -070055// turn off some misguided warnings
Neil MacIntosh9a297122015-09-14 15:11:07 -070056#pragma warning(push)
57#pragma warning(disable: 4351) // warns about newly introduced aggregate initializer behavior
Neil MacIntoshd5316802015-09-30 21:54:08 -070058
Neil MacIntosh9a297122015-09-14 15:11:07 -070059#endif // _MSC_VER <= 1800
60
Neil MacIntoshd5316802015-09-30 21:54:08 -070061#endif // _MSC_VER
62
63// In order to test the library, we need it to throw exceptions that we can catch
64#ifdef GSL_THROWS_FOR_TESTING
65#define noexcept /* nothing */
66#endif // GSL_THROWS_FOR_TESTING
67
68
Neil MacIntoshef626fd2015-09-29 16:41:37 -070069namespace gsl {
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070070
71/*
72** begin definitions of index and bounds
73*/
74namespace details
75{
Neil MacIntosh68064d62015-11-03 19:17:11 -080076 template <typename SizeType>
77 struct SizeTypeTraits
78 {
79 static const SizeType max_value = std::numeric_limits<SizeType>::max();
80 };
Anna Gringauze1c208b32015-10-16 17:40:57 -070081
82
Neil MacIntosh68064d62015-11-03 19:17:11 -080083 template<typename... Ts>
84 class are_integral : public std::integral_constant<bool, true> {};
Anna Gringauze1c208b32015-10-16 17:40:57 -070085
Neil MacIntosh68064d62015-11-03 19:17:11 -080086 template<typename T, typename... Ts>
87 class are_integral<T, Ts...> : public std::integral_constant<bool, std::is_integral<T>::value && are_integral<Ts...>::value> {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070088}
89
Neil MacIntoshf45fedb2015-10-15 14:29:35 -070090template <size_t Rank>
Anna Gringauzedb384972015-10-05 12:34:23 -070091class index final
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070092{
Neil MacIntosh68064d62015-11-03 19:17:11 -080093 static_assert(Rank > 0, "Rank must be greater than 0!");
Anna Gringauzedb384972015-10-05 12:34:23 -070094
Neil MacIntosh68064d62015-11-03 19:17:11 -080095 template <size_t OtherRank>
96 friend class index;
Anna Gringauzedb384972015-10-05 12:34:23 -070097
Neil MacIntosha9dcbe02015-08-20 18:09:14 -070098public:
Neil MacIntosh68064d62015-11-03 19:17:11 -080099 static const size_t rank = Rank;
100 using value_type = std::ptrdiff_t;
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700101 using size_type = value_type;
Neil MacIntosh68064d62015-11-03 19:17:11 -0800102 using reference = std::add_lvalue_reference_t<value_type>;
103 using const_reference = std::add_lvalue_reference_t<std::add_const_t<value_type>>;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700104
Neil MacIntosh68064d62015-11-03 19:17:11 -0800105 constexpr index() noexcept
106 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700107
Neil MacIntosh68064d62015-11-03 19:17:11 -0800108 constexpr index(const value_type(&values)[Rank]) noexcept
109 {
110 std::copy(values, values + Rank, elems);
111 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700112
Neil MacIntoshe9a96022015-11-03 18:56:55 -0800113#ifdef GSL_MSVC_HAS_VARIADIC_CTOR_BUG
Neil MacIntosh68064d62015-11-03 19:17:11 -0800114 template<typename T, typename... Ts,
Neil MacIntoshe9a96022015-11-03 18:56:55 -0800115 typename = std::enable_if_t<((sizeof...(Ts) + 1) == Rank) &&
116 std::is_integral<T>::value &&
117 details::are_integral<Ts...>::value>>
118 constexpr index(T t, Ts... ds) : index({ static_cast<value_type>(t), static_cast<value_type>(ds)... })
119 {}
120#else
Neil MacIntosh68064d62015-11-03 19:17:11 -0800121 template<typename... Ts,
Neil MacIntoshe9a96022015-11-03 18:56:55 -0800122 typename = std::enable_if_t<(sizeof...(Ts) == Rank) && details::are_integral<Ts...>::value>>
Neil MacIntosh68064d62015-11-03 19:17:11 -0800123 constexpr index(Ts... ds) noexcept : elems{ static_cast<value_type>(ds)... }
Neil MacIntoshe9a96022015-11-03 18:56:55 -0800124 {}
125#endif
Anna Gringauzedb384972015-10-05 12:34:23 -0700126
Neil MacIntosh68064d62015-11-03 19:17:11 -0800127 constexpr index(const index& other) noexcept = default;
Anna Gringauzedb384972015-10-05 12:34:23 -0700128
Neil MacIntosh68064d62015-11-03 19:17:11 -0800129 constexpr index& operator=(const index& rhs) noexcept = default;
Anna Gringauzedb384972015-10-05 12:34:23 -0700130
Neil MacIntosh68064d62015-11-03 19:17:11 -0800131 // Preconditions: component_idx < rank
132 constexpr reference operator[](size_t component_idx)
133 {
134 fail_fast_assert(component_idx < Rank, "Component index must be less than rank");
135 return elems[component_idx];
136 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700137
Neil MacIntosh68064d62015-11-03 19:17:11 -0800138 // Preconditions: component_idx < rank
139 constexpr const_reference operator[](size_t component_idx) const noexcept
140 {
141 fail_fast_assert(component_idx < Rank, "Component index must be less than rank");
142 return elems[component_idx];
143 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700144
Neil MacIntosh68064d62015-11-03 19:17:11 -0800145 constexpr bool operator==(const index& rhs) const noexcept
146 {
147 return std::equal(elems, elems + rank, rhs.elems);
148 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700149
Neil MacIntosh68064d62015-11-03 19:17:11 -0800150 constexpr bool operator!=(const index& rhs) const noexcept
151 {
152 return !(this == rhs);
153 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700154
Neil MacIntosh68064d62015-11-03 19:17:11 -0800155 constexpr index operator+() const noexcept
156 {
157 return *this;
158 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700159
Neil MacIntosh68064d62015-11-03 19:17:11 -0800160 constexpr index operator-() const noexcept
161 {
162 index ret = *this;
163 std::transform(ret, ret + rank, ret, std::negate<value_type>{});
164 return ret;
165 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700166
Neil MacIntosh68064d62015-11-03 19:17:11 -0800167 constexpr index operator+(const index& rhs) const noexcept
168 {
169 index ret = *this;
170 ret += rhs;
171 return ret;
172 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700173
Neil MacIntosh68064d62015-11-03 19:17:11 -0800174 constexpr index operator-(const index& rhs) const noexcept
175 {
176 index ret = *this;
177 ret -= rhs;
178 return ret;
179 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700180
Neil MacIntosh68064d62015-11-03 19:17:11 -0800181 constexpr index& operator+=(const index& rhs) noexcept
182 {
183 std::transform(elems, elems + rank, rhs.elems, elems, std::plus<value_type>{});
184 return *this;
185 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700186
Neil MacIntosh68064d62015-11-03 19:17:11 -0800187 constexpr index& operator-=(const index& rhs) noexcept
188 {
189 std::transform(elems, elems + rank, rhs.elems, elems, std::minus<value_type>{});
190 return *this;
191 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700192
Neil MacIntosh68064d62015-11-03 19:17:11 -0800193 constexpr index operator*(value_type v) const noexcept
194 {
195 index ret = *this;
196 ret *= v;
197 return ret;
198 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700199
Neil MacIntosh68064d62015-11-03 19:17:11 -0800200 constexpr index operator/(value_type v) const noexcept
201 {
202 index ret = *this;
203 ret /= v;
204 return ret;
205 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700206
Neil MacIntosh68064d62015-11-03 19:17:11 -0800207 friend constexpr index operator*(value_type v, const index& rhs) noexcept
208 {
209 return rhs * v;
210 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700211
Neil MacIntosh68064d62015-11-03 19:17:11 -0800212 constexpr index& operator*=(value_type v) noexcept
213 {
214 std::transform(elems, elems + rank, elems, [v](value_type x) { return std::multiplies<value_type>{}(x, v); });
215 return *this;
216 }
Anna Gringauzedb384972015-10-05 12:34:23 -0700217
Neil MacIntosh68064d62015-11-03 19:17:11 -0800218 constexpr index& operator/=(value_type v) noexcept
219 {
220 std::transform(elems, elems + rank, elems, [v](value_type x) { return std::divides<value_type>{}(x, v); });
221 return *this;
222 }
Anna Gringauze546f8cc2015-10-05 21:04:56 -0700223
Anna Gringauzedb384972015-10-05 12:34:23 -0700224private:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800225 value_type elems[Rank] = {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700226};
227
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700228#ifndef _MSC_VER
229
230struct static_bounds_dynamic_range_t
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700231{
Neil MacIntosh68064d62015-11-03 19:17:11 -0800232 template <typename T, typename Dummy = std::enable_if_t<std::is_integral<T>::value>>
233 constexpr operator T() const noexcept
234 {
235 return static_cast<T>(-1);
236 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700237
Neil MacIntosh68064d62015-11-03 19:17:11 -0800238 template <typename T, typename Dummy = std::enable_if_t<std::is_integral<T>::value>>
239 constexpr bool operator ==(T other) const noexcept
240 {
241 return static_cast<T>(-1) == other;
242 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700243
Neil MacIntosh68064d62015-11-03 19:17:11 -0800244 template <typename T, typename Dummy = std::enable_if_t<std::is_integral<T>::value>>
245 constexpr bool operator !=(T other) const noexcept
246 {
247 return static_cast<T>(-1) != other;
248 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700249
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700250};
251
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700252template <typename T, typename Dummy = std::enable_if_t<std::is_integral<T>::value>>
253constexpr bool operator ==(T left, static_bounds_dynamic_range_t right) noexcept
254{
Neil MacIntosh68064d62015-11-03 19:17:11 -0800255 return right == left;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700256}
257
258template <typename T, typename Dummy = std::enable_if_t<std::is_integral<T>::value>>
259constexpr bool operator !=(T left, static_bounds_dynamic_range_t right) noexcept
260{
Neil MacIntosh68064d62015-11-03 19:17:11 -0800261 return right != left;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700262}
263
264constexpr static_bounds_dynamic_range_t dynamic_range{};
265#else
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700266const std::ptrdiff_t dynamic_range = -1;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700267#endif
268
269struct generalized_mapping_tag {};
270struct contiguous_mapping_tag : generalized_mapping_tag {};
271
272namespace details
273{
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700274
Neil MacIntosh68064d62015-11-03 19:17:11 -0800275 template <std::ptrdiff_t Left, std::ptrdiff_t Right>
276 struct LessThan
277 {
278 static const bool value = Left < Right;
279 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700280
Neil MacIntosh68064d62015-11-03 19:17:11 -0800281 template <std::ptrdiff_t... Ranges>
282 struct BoundsRanges {
283 using size_type = std::ptrdiff_t;
284 static const size_type Depth = 0;
285 static const size_type DynamicNum = 0;
286 static const size_type CurrentRange = 1;
287 static const size_type TotalSize = 1;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700288
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700289 // TODO : following signature is for work around VS bug
290 template <typename OtherRange>
Neil MacIntosh68064d62015-11-03 19:17:11 -0800291 BoundsRanges(const OtherRange&, bool /* firstLevel */)
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700292 {}
Neil MacIntosh68064d62015-11-03 19:17:11 -0800293
294 BoundsRanges (const BoundsRanges&) = default;
Vladislav Yaroslavlev995cfdf2015-11-12 10:46:21 +0300295 BoundsRanges& operator=(const BoundsRanges&) = default;
Neil MacIntosh68064d62015-11-03 19:17:11 -0800296 BoundsRanges(const std::ptrdiff_t* const) { }
297 BoundsRanges() = default;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700298
299
Neil MacIntosh68064d62015-11-03 19:17:11 -0800300 template <typename T, size_t Dim>
301 void serialize(T&) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700302 {}
303
Neil MacIntosh68064d62015-11-03 19:17:11 -0800304 template <typename T, size_t Dim>
305 size_type linearize(const T&) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700306 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800307 return 0;
308 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700309
Neil MacIntosh68064d62015-11-03 19:17:11 -0800310 template <typename T, size_t Dim>
311 bool contains(const T&) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700312 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800313 return 0;
314 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700315
Neil MacIntosh68064d62015-11-03 19:17:11 -0800316 size_type totalSize() const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700317 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800318 return TotalSize;
319 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700320
Neil MacIntosh68064d62015-11-03 19:17:11 -0800321 bool operator==(const BoundsRanges&) const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700322 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800323 return true;
324 }
325 };
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700326
Neil MacIntosh68064d62015-11-03 19:17:11 -0800327 template <std::ptrdiff_t... RestRanges>
328 struct BoundsRanges <dynamic_range, RestRanges...> : BoundsRanges<RestRanges...>{
329 using Base = BoundsRanges <RestRanges... >;
330 using size_type = std::ptrdiff_t;
331 static const size_t Depth = Base::Depth + 1;
332 static const size_t DynamicNum = Base::DynamicNum + 1;
333 static const size_type CurrentRange = dynamic_range;
334 static const size_type TotalSize = dynamic_range;
335 const size_type m_bound;
336
337 BoundsRanges (const BoundsRanges&) = default;
338
339 BoundsRanges(const std::ptrdiff_t* const arr) : Base(arr + 1), m_bound(*arr * this->Base::totalSize())
340 {
341 fail_fast_assert(0 <= *arr);
342 }
343
344 BoundsRanges() : m_bound(0) {}
345
346 template <std::ptrdiff_t OtherRange, std::ptrdiff_t... RestOtherRanges>
347 BoundsRanges(const BoundsRanges<OtherRange, RestOtherRanges...>& other, bool /* firstLevel */ = true) :
348 Base(static_cast<const BoundsRanges<RestOtherRanges...>&>(other), false), m_bound(other.totalSize())
349 {}
350
351 template <typename T, size_t Dim = 0>
352 void serialize(T& arr) const
353 {
354 arr[Dim] = elementNum();
355 this->Base::template serialize<T, Dim + 1>(arr);
356 }
357
358 template <typename T, size_t Dim = 0>
359 size_type linearize(const T& arr) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700360 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800361 const size_type index = this->Base::totalSize() * arr[Dim];
362 fail_fast_assert(index < m_bound);
363 return index + this->Base::template linearize<T, Dim + 1>(arr);
364 }
365
366 template <typename T, size_t Dim = 0>
367 size_type contains(const T & arr) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700368 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800369 const ptrdiff_t last = this->Base::template contains<T, Dim + 1>(arr);
370 if (last == -1)
371 return -1;
372 const ptrdiff_t cur = this->Base::totalSize() * arr[Dim];
373 return cur < m_bound ? cur + last : -1;
374 }
375
376 size_type totalSize() const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700377 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800378 return m_bound;
379 }
380
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700381 size_type elementNum() const noexcept
382 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800383 return totalSize() / this->Base::totalSize();
384 }
385
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700386 size_type elementNum(size_t dim) const noexcept
387 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800388 if (dim > 0)
389 return this->Base::elementNum(dim - 1);
390 else
391 return elementNum();
392 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700393
Neil MacIntosh68064d62015-11-03 19:17:11 -0800394 bool operator == (const BoundsRanges & rhs) const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700395 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800396 return m_bound == rhs.m_bound && static_cast<const Base&>(*this) == static_cast<const Base&>(rhs);
397 }
398 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700399
Neil MacIntosh68064d62015-11-03 19:17:11 -0800400 template <std::ptrdiff_t CurRange, std::ptrdiff_t... RestRanges>
401 struct BoundsRanges <CurRange, RestRanges...> : BoundsRanges<RestRanges...>
402 {
403 using Base = BoundsRanges <RestRanges... >;
404 using size_type = std::ptrdiff_t;
405 static const size_t Depth = Base::Depth + 1;
406 static const size_t DynamicNum = Base::DynamicNum;
407 static const size_type CurrentRange = CurRange;
408 static const size_type TotalSize = Base::TotalSize == dynamic_range ? dynamic_range : CurrentRange * Base::TotalSize;
409
410 BoundsRanges (const BoundsRanges&) = default;
411 BoundsRanges(const std::ptrdiff_t* const arr) : Base(arr) { }
412 BoundsRanges() = default;
413
414 template <std::ptrdiff_t OtherRange, std::ptrdiff_t... RestOtherRanges>
415 BoundsRanges(const BoundsRanges<OtherRange, RestOtherRanges...>&other, bool firstLevel = true) : Base(static_cast<const BoundsRanges<RestOtherRanges...>&>(other), false)
416 {
417 fail_fast_assert((firstLevel && totalSize() <= other.totalSize()) || totalSize() == other.totalSize());
418 }
419
420 template <typename T, size_t Dim = 0>
421 void serialize(T& arr) const
422 {
423 arr[Dim] = elementNum();
424 this->Base::template serialize<T, Dim + 1>(arr);
425 }
426
427 template <typename T, size_t Dim = 0>
428 size_type linearize(const T& arr) const
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700429 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800430 fail_fast_assert(arr[Dim] < CurrentRange, "Index is out of range");
431 return this->Base::totalSize() * arr[Dim] + this->Base::template linearize<T, Dim + 1>(arr);
432 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700433
Neil MacIntosh68064d62015-11-03 19:17:11 -0800434 template <typename T, size_t Dim = 0>
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700435 size_type contains(const T& arr) const
436 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800437 if (arr[Dim] >= CurrentRange)
438 return -1;
439 const size_type last = this->Base::template contains<T, Dim + 1>(arr);
440 if (last == -1)
441 return -1;
442 return this->Base::totalSize() * arr[Dim] + last;
443 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700444
Neil MacIntosh68064d62015-11-03 19:17:11 -0800445 size_type totalSize() const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700446 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800447 return CurrentRange * this->Base::totalSize();
448 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700449
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700450 size_type elementNum() const noexcept
451 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800452 return CurrentRange;
453 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700454
Neil MacIntosh68064d62015-11-03 19:17:11 -0800455 size_type elementNum(size_t dim) const noexcept
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700456 {
Neil MacIntosh68064d62015-11-03 19:17:11 -0800457 if (dim > 0)
458 return this->Base::elementNum(dim - 1);
459 else
460 return elementNum();
461 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700462
Neil MacIntosh68064d62015-11-03 19:17:11 -0800463 bool operator== (const BoundsRanges& rhs) const noexcept
464 {
465 return static_cast<const Base &>(*this) == static_cast<const Base &>(rhs);
466 }
467 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700468
Neil MacIntosh68064d62015-11-03 19:17:11 -0800469 template <typename SourceType, typename TargetType, size_t Rank>
470 struct BoundsRangeConvertible2;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700471
Neil MacIntosh68064d62015-11-03 19:17:11 -0800472 template <size_t Rank, typename SourceType, typename TargetType, typename Ret = BoundsRangeConvertible2<typename SourceType::Base, typename TargetType::Base, Rank>>
473 auto helpBoundsRangeConvertible(SourceType, TargetType, std::true_type) -> Ret;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700474
Neil MacIntosh68064d62015-11-03 19:17:11 -0800475 template <size_t Rank, typename SourceType, typename TargetType>
476 auto helpBoundsRangeConvertible(SourceType, TargetType, ...) -> std::false_type;
477
478 template <typename SourceType, typename TargetType, size_t Rank>
479 struct BoundsRangeConvertible2 : decltype(helpBoundsRangeConvertible<Rank - 1>(SourceType(), TargetType(),
480 std::integral_constant<bool, SourceType::Depth == TargetType::Depth
481 && (SourceType::CurrentRange == TargetType::CurrentRange || TargetType::CurrentRange == dynamic_range || SourceType::CurrentRange == dynamic_range)>()))
482 {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700483
Neil MacIntosh68064d62015-11-03 19:17:11 -0800484 template <typename SourceType, typename TargetType>
485 struct BoundsRangeConvertible2<SourceType, TargetType, 0> : std::true_type {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700486
Neil MacIntosh68064d62015-11-03 19:17:11 -0800487 template <typename SourceType, typename TargetType, std::ptrdiff_t Rank = TargetType::Depth>
488 struct BoundsRangeConvertible : decltype(helpBoundsRangeConvertible<Rank - 1>(SourceType(), TargetType(),
489 std::integral_constant<bool, SourceType::Depth == TargetType::Depth
490 && (!LessThan<SourceType::CurrentRange, TargetType::CurrentRange>::value || TargetType::CurrentRange == dynamic_range || SourceType::CurrentRange == dynamic_range)>()))
491 {};
492 template <typename SourceType, typename TargetType>
493 struct BoundsRangeConvertible<SourceType, TargetType, 0> : std::true_type {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700494
Neil MacIntosh68064d62015-11-03 19:17:11 -0800495 template <typename TypeChain>
496 struct TypeListIndexer
497 {
498 const TypeChain & obj;
499 TypeListIndexer(const TypeChain & obj) :obj(obj){}
500 template<size_t N>
501 const TypeChain & getObj(std::true_type)
502 {
503 return obj;
504 }
505 template<size_t N, typename MyChain = TypeChain, typename MyBase = typename MyChain::Base>
506 auto getObj(std::false_type) -> decltype(TypeListIndexer<MyBase>(static_cast<const MyBase &>(obj)).template get<N>())
507 {
508 return TypeListIndexer<MyBase>(static_cast<const MyBase &>(obj)).template get<N>();
509 }
510 template <size_t N>
511 auto get() -> decltype(getObj<N - 1>(std::integral_constant<bool, true>()))
512 {
513 return getObj<N - 1>(std::integral_constant<bool, N == 0>());
514 }
515 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700516
Neil MacIntosh68064d62015-11-03 19:17:11 -0800517 template <typename TypeChain>
518 TypeListIndexer<TypeChain> createTypeListIndexer(const TypeChain &obj)
519 {
520 return TypeListIndexer<TypeChain>(obj);
521 }
Anna Gringauzefdf86432015-10-14 10:46:22 -0700522
Neil MacIntosh68064d62015-11-03 19:17:11 -0800523 template <size_t Rank, bool Enabled = (Rank > 1), typename Ret = std::enable_if_t<Enabled, index<Rank - 1>>>
524 constexpr Ret shift_left(const index<Rank>& other) noexcept
525 {
526 Ret ret{};
527 for (size_t i = 0; i < Rank - 1; ++i)
528 {
529 ret[i] = other[i + 1];
530 }
531 return ret;
532 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700533}
534
535template <typename IndexType>
536class bounds_iterator;
537
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700538template <std::ptrdiff_t... Ranges>
539class static_bounds
540{
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700541public:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800542 static_bounds(const details::BoundsRanges<Ranges...>&) {
543 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700544};
545
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700546template <std::ptrdiff_t FirstRange, std::ptrdiff_t... RestRanges>
547class static_bounds<FirstRange, RestRanges...>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700548{
Neil MacIntosh68064d62015-11-03 19:17:11 -0800549 using MyRanges = details::BoundsRanges<FirstRange, RestRanges... >;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700550
Neil MacIntosh68064d62015-11-03 19:17:11 -0800551 MyRanges m_ranges;
552 constexpr static_bounds(const MyRanges& range) : m_ranges(range)
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700553 {}
Neil MacIntosh68064d62015-11-03 19:17:11 -0800554
555 template <std::ptrdiff_t... OtherRanges>
556 friend class static_bounds;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700557
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700558public:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800559 static const size_t rank = MyRanges::Depth;
560 static const size_t dynamic_rank = MyRanges::DynamicNum;
561 static const std::ptrdiff_t static_size = MyRanges::TotalSize;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700562
Neil MacIntosh68064d62015-11-03 19:17:11 -0800563 using size_type = std::ptrdiff_t;
564 using index_type = index<rank>;
Neil MacIntosha4fa2b32015-10-28 16:53:53 -0700565 using const_index_type = std::add_const_t<index_type>;
566 using iterator = bounds_iterator<const_index_type>;
567 using const_iterator = bounds_iterator<const_index_type>;
568 using difference_type = std::ptrdiff_t;
Neil MacIntosh68064d62015-11-03 19:17:11 -0800569 using sliced_type = static_bounds<RestRanges...>;
570 using mapping_type = contiguous_mapping_tag;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700571
Neil MacIntosh68064d62015-11-03 19:17:11 -0800572 constexpr static_bounds(const static_bounds&) = default;
573
574 template <std::ptrdiff_t... Ranges, typename Dummy = std::enable_if_t<
575 details::BoundsRangeConvertible<details::BoundsRanges<Ranges...>, details::BoundsRanges <FirstRange, RestRanges... >>::value>>
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700576 constexpr static_bounds(const static_bounds<Ranges...>& other) : m_ranges(other.m_ranges)
577 {}
Neil MacIntosha4fa2b32015-10-28 16:53:53 -0700578
Neil MacIntosh68064d62015-11-03 19:17:11 -0800579 constexpr static_bounds(std::initializer_list<size_type> il) : m_ranges((const std::ptrdiff_t*)il.begin())
580 {
581 fail_fast_assert((MyRanges::DynamicNum == 0 && il.size() == 1 && *il.begin() == static_size) || MyRanges::DynamicNum == il.size(), "Size of the initializer list must match the rank of the array");
582 fail_fast_assert(m_ranges.totalSize() <= PTRDIFF_MAX, "Size of the range is larger than the max element of the size type");
583 }
584
585 constexpr static_bounds() = default;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700586
Neil MacIntosh68064d62015-11-03 19:17:11 -0800587 constexpr static_bounds& operator=(const static_bounds& otherBounds)
588 {
589 new(&m_ranges) MyRanges (otherBounds.m_ranges);
590 return *this;
591 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700592
Neil MacIntosh68064d62015-11-03 19:17:11 -0800593 constexpr sliced_type slice() const noexcept
594 {
595 return sliced_type{static_cast<const details::BoundsRanges<RestRanges...> &>(m_ranges)};
596 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700597
Neil MacIntosh68064d62015-11-03 19:17:11 -0800598 constexpr size_type stride() const noexcept
599 {
600 return rank > 1 ? slice().size() : 1;
601 }
602
603 constexpr size_type size() const noexcept
604 {
605 return m_ranges.totalSize();
606 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -0700607
Neil MacIntosh68064d62015-11-03 19:17:11 -0800608 constexpr size_type total_size() const noexcept
609 {
610 return m_ranges.totalSize();
611 }
612
613 constexpr size_type linearize(const index_type & idx) const
614 {
615 return m_ranges.linearize(idx);
616 }
617
618 constexpr bool contains(const index_type& idx) const noexcept
619 {
620 return m_ranges.contains(idx) != -1;
621 }
622
623 constexpr size_type operator[](size_t index) const noexcept
624 {
625 return m_ranges.elementNum(index);
626 }
627
628 template <size_t Dim = 0>
629 constexpr size_type extent() const noexcept
630 {
631 static_assert(Dim < rank, "dimension should be less than rank (dimension count starts from 0)");
632 return details::createTypeListIndexer(m_ranges).template get<Dim>().elementNum();
633 }
634
635 constexpr index_type index_bounds() const noexcept
636 {
637 size_type extents[rank] = {};
638 m_ranges.serialize(extents);
639 return{ extents };
640 }
641
642 template <std::ptrdiff_t... Ranges>
643 constexpr bool operator == (const static_bounds<Ranges...>& rhs) const noexcept
644 {
645 return this->size() == rhs.size();
646 }
647
648 template <std::ptrdiff_t... Ranges>
649 constexpr bool operator != (const static_bounds<Ranges...>& rhs) const noexcept
650 {
651 return !(*this == rhs);
652 }
653
654 constexpr const_iterator begin() const noexcept
655 {
656 return const_iterator(*this, index_type{});
657 }
658
659 constexpr const_iterator end() const noexcept
660 {
661 return const_iterator(*this, this->index_bounds());
662 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700663};
664
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700665template <size_t Rank>
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700666class strided_bounds
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700667{
Neil MacIntosh68064d62015-11-03 19:17:11 -0800668 template <size_t OtherRank>
669 friend class strided_bounds;
Anna Gringauze17ed5c32015-08-30 23:30:15 -0700670
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700671public:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800672 static const size_t rank = Rank;
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700673 using value_type = std::ptrdiff_t;
Neil MacIntosh68064d62015-11-03 19:17:11 -0800674 using reference = std::add_lvalue_reference_t<value_type>;
675 using const_reference = std::add_const_t<reference>;
676 using size_type = value_type;
677 using difference_type = value_type;
678 using index_type = index<rank>;
Neil MacIntosha4fa2b32015-10-28 16:53:53 -0700679 using const_index_type = std::add_const_t<index_type>;
680 using iterator = bounds_iterator<const_index_type>;
681 using const_iterator = bounds_iterator<const_index_type>;
682 static const value_type dynamic_rank = rank;
Neil MacIntosh68064d62015-11-03 19:17:11 -0800683 static const value_type static_size = dynamic_range;
684 using sliced_type = std::conditional_t<rank != 0, strided_bounds<rank - 1>, void>;
685 using mapping_type = generalized_mapping_tag;
Neil MacIntoshd0f09e72015-10-15 16:38:53 -0700686
Neil MacIntosh68064d62015-11-03 19:17:11 -0800687 constexpr strided_bounds(const strided_bounds &) noexcept = default;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700688
Vladislav Yaroslavlev557e6692015-11-12 10:44:41 +0300689 constexpr strided_bounds & operator=(const strided_bounds &) noexcept = default;
690
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700691 constexpr strided_bounds(const value_type(&values)[rank], index_type strides)
Neil MacIntoshace9ab92015-10-23 19:49:17 -0700692 : m_extents(values), m_strides(std::move(strides))
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700693 {}
694
Neil MacIntosh68064d62015-11-03 19:17:11 -0800695 constexpr strided_bounds(const index_type &extents, const index_type &strides) noexcept
696 : m_extents(extents), m_strides(strides)
697 {}
Neil MacIntoshd0f09e72015-10-15 16:38:53 -0700698
Neil MacIntosh68064d62015-11-03 19:17:11 -0800699 constexpr index_type strides() const noexcept
700 {
701 return m_strides;
702 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700703
Neil MacIntosh68064d62015-11-03 19:17:11 -0800704 constexpr size_type total_size() const noexcept
705 {
706 size_type ret = 0;
707 for (size_t i = 0; i < rank; ++i)
708 {
709 ret += (m_extents[i] - 1) * m_strides[i];
710 }
711 return ret + 1;
712 }
713
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700714 constexpr size_type size() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -0800715 {
716 size_type ret = 1;
717 for (size_t i = 0; i < rank; ++i)
718 {
719 ret *= m_extents[i];
720 }
721 return ret;
722 }
723
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700724 constexpr bool contains(const index_type& idx) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -0800725 {
726 for (size_t i = 0; i < rank; ++i)
727 {
728 if (idx[i] < 0 || idx[i] >= m_extents[i])
729 return false;
730 }
731 return true;
732 }
Neil MacIntoshd0f09e72015-10-15 16:38:53 -0700733
Neil MacIntosh68064d62015-11-03 19:17:11 -0800734 constexpr size_type linearize(const index_type& idx) const noexcept
735 {
736 size_type ret = 0;
737 for (size_t i = 0; i < rank; i++)
738 {
739 fail_fast_assert(idx[i] < m_extents[i], "index is out of bounds of the array");
740 ret += idx[i] * m_strides[i];
741 }
742 return ret;
743 }
744
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700745 constexpr size_type stride() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -0800746 {
747 return m_strides[0];
748 }
749
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700750 template <bool Enabled = (rank > 1), typename Ret = std::enable_if_t<Enabled, sliced_type>>
Neil MacIntosh68064d62015-11-03 19:17:11 -0800751 constexpr sliced_type slice() const
752 {
753 return{ details::shift_left(m_extents), details::shift_left(m_strides) };
754 }
755
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700756 template <size_t Dim = 0>
Neil MacIntosh68064d62015-11-03 19:17:11 -0800757 constexpr size_type extent() const noexcept
758 {
759 static_assert(Dim < Rank, "dimension should be less than rank (dimension count starts from 0)");
760 return m_extents[Dim];
761 }
762
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700763 constexpr index_type index_bounds() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -0800764 {
765 return m_extents;
766 }
767 constexpr const_iterator begin() const noexcept
768 {
769 return const_iterator{ *this, index_type{} };
770 }
771
772 constexpr const_iterator end() const noexcept
773 {
774 return const_iterator{ *this, index_bounds() };
775 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700776
Anna Gringauze17ed5c32015-08-30 23:30:15 -0700777private:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800778 index_type m_extents;
779 index_type m_strides;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700780};
781
782template <typename T>
783struct is_bounds : std::integral_constant<bool, false> {};
Neil MacIntoshf45fedb2015-10-15 14:29:35 -0700784template <std::ptrdiff_t... Ranges>
785struct is_bounds<static_bounds<Ranges...>> : std::integral_constant<bool, true> {};
786template <size_t Rank>
787struct is_bounds<strided_bounds<Rank>> : std::integral_constant<bool, true> {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700788
789template <typename IndexType>
Anna Gringauzea4654a42015-10-16 12:15:22 -0700790class bounds_iterator: public std::iterator<std::random_access_iterator_tag, IndexType>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700791{
792private:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800793 using Base = std::iterator <std::random_access_iterator_tag, IndexType>;
Anna Gringauzea4654a42015-10-16 12:15:22 -0700794
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700795public:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800796 static const size_t rank = IndexType::rank;
797 using typename Base::reference;
798 using typename Base::pointer;
799 using typename Base::difference_type;
800 using typename Base::value_type;
801 using index_type = value_type;
802 using index_size_type = typename IndexType::value_type;
803 template <typename Bounds>
804 explicit bounds_iterator(const Bounds& bnd, value_type curr) noexcept
805 : boundary(bnd.index_bounds()), curr(std::move(curr))
806 {
807 static_assert(is_bounds<Bounds>::value, "Bounds type must be provided");
808 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700809
Neil MacIntosh68064d62015-11-03 19:17:11 -0800810 constexpr reference operator*() const noexcept
811 {
812 return curr;
813 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700814
Neil MacIntosh68064d62015-11-03 19:17:11 -0800815 constexpr pointer operator->() const noexcept
816 {
817 return &curr;
818 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700819
Neil MacIntosh68064d62015-11-03 19:17:11 -0800820 constexpr bounds_iterator& operator++() noexcept
821 {
822 for (size_t i = rank; i-- > 0;)
823 {
824 if (curr[i] < boundary[i] - 1)
825 {
826 curr[i]++;
827 return *this;
828 }
829 curr[i] = 0;
830 }
831 // If we're here we've wrapped over - set to past-the-end.
832 curr = boundary;
833 return *this;
834 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700835
Neil MacIntosh68064d62015-11-03 19:17:11 -0800836 constexpr bounds_iterator operator++(int) noexcept
837 {
838 auto ret = *this;
839 ++(*this);
840 return ret;
841 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700842
Neil MacIntosh68064d62015-11-03 19:17:11 -0800843 constexpr bounds_iterator& operator--() noexcept
844 {
845 if (!less(curr, boundary))
846 {
847 // if at the past-the-end, set to last element
848 for (size_t i = 0; i < rank; ++i)
849 {
850 curr[i] = boundary[i] - 1;
851 }
852 return *this;
853 }
854 for (size_t i = rank; i-- > 0;)
855 {
856 if (curr[i] >= 1)
857 {
858 curr[i]--;
859 return *this;
860 }
861 curr[i] = boundary[i] - 1;
862 }
863 // If we're here the preconditions were violated
864 // "pre: there exists s such that r == ++s"
865 fail_fast_assert(false);
866 return *this;
867 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700868
Neil MacIntosh68064d62015-11-03 19:17:11 -0800869 constexpr bounds_iterator operator--(int) noexcept
870 {
871 auto ret = *this;
872 --(*this);
873 return ret;
874 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700875
Neil MacIntosh68064d62015-11-03 19:17:11 -0800876 constexpr bounds_iterator operator+(difference_type n) const noexcept
877 {
878 bounds_iterator ret{ *this };
879 return ret += n;
880 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700881
Neil MacIntosh68064d62015-11-03 19:17:11 -0800882 constexpr bounds_iterator& operator+=(difference_type n) noexcept
883 {
884 auto linear_idx = linearize(curr) + n;
885 std::remove_const_t<value_type> stride = 0;
886 stride[rank - 1] = 1;
887 for (size_t i = rank - 1; i-- > 0;)
888 {
889 stride[i] = stride[i + 1] * boundary[i + 1];
890 }
891 for (size_t i = 0; i < rank; ++i)
892 {
893 curr[i] = linear_idx / stride[i];
894 linear_idx = linear_idx % stride[i];
895 }
896 fail_fast_assert(!less(curr, index_type{}) && !less(boundary, curr), "index is out of bounds of the array");
897 return *this;
898 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700899
Neil MacIntosh68064d62015-11-03 19:17:11 -0800900 constexpr bounds_iterator operator-(difference_type n) const noexcept
901 {
902 bounds_iterator ret{ *this };
903 return ret -= n;
904 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700905
Neil MacIntosh68064d62015-11-03 19:17:11 -0800906 constexpr bounds_iterator& operator-=(difference_type n) noexcept
907 {
908 return *this += -n;
909 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700910
Neil MacIntosh68064d62015-11-03 19:17:11 -0800911 constexpr difference_type operator-(const bounds_iterator& rhs) const noexcept
912 {
913 return linearize(curr) - linearize(rhs.curr);
914 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700915
Neil MacIntosh68064d62015-11-03 19:17:11 -0800916 constexpr value_type operator[](difference_type n) const noexcept
917 {
918 return *(*this + n);
919 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700920
Neil MacIntosh68064d62015-11-03 19:17:11 -0800921 constexpr bool operator==(const bounds_iterator& rhs) const noexcept
922 {
923 return curr == rhs.curr;
924 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700925
Neil MacIntosh68064d62015-11-03 19:17:11 -0800926 constexpr bool operator!=(const bounds_iterator& rhs) const noexcept
927 {
928 return !(*this == rhs);
929 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700930
Neil MacIntosh68064d62015-11-03 19:17:11 -0800931 constexpr bool operator<(const bounds_iterator& rhs) const noexcept
932 {
933 return less(curr, rhs.curr);
934 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700935
Neil MacIntosh68064d62015-11-03 19:17:11 -0800936 constexpr bool operator<=(const bounds_iterator& rhs) const noexcept
937 {
938 return !(rhs < *this);
939 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700940
Neil MacIntosh68064d62015-11-03 19:17:11 -0800941 constexpr bool operator>(const bounds_iterator& rhs) const noexcept
942 {
943 return rhs < *this;
944 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700945
Neil MacIntosh68064d62015-11-03 19:17:11 -0800946 constexpr bool operator>=(const bounds_iterator& rhs) const noexcept
947 {
948 return !(rhs > *this);
949 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700950
Neil MacIntosh68064d62015-11-03 19:17:11 -0800951 void swap(bounds_iterator& rhs) noexcept
952 {
953 std::swap(boundary, rhs.boundary);
954 std::swap(curr, rhs.curr);
955 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700956private:
Neil MacIntosh68064d62015-11-03 19:17:11 -0800957 constexpr bool less(index_type& one, index_type& other) const noexcept
958 {
959 for (size_t i = 0; i < rank; ++i)
960 {
961 if (one[i] < other[i])
962 return true;
963 }
964 return false;
965 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700966
Neil MacIntosh68064d62015-11-03 19:17:11 -0800967 constexpr index_size_type linearize(const value_type& idx) const noexcept
968 {
969 // TODO: Smarter impl.
970 // Check if past-the-end
971 index_size_type multiplier = 1;
972 index_size_type res = 0;
973 if (!less(idx, boundary))
974 {
975 res = 1;
976 for (size_t i = rank; i-- > 0;)
977 {
978 res += (idx[i] - 1) * multiplier;
979 multiplier *= boundary[i];
980 }
981 }
982 else
983 {
984 for (size_t i = rank; i-- > 0;)
985 {
986 res += idx[i] * multiplier;
987 multiplier *= boundary[i];
988 }
989 }
990 return res;
991 }
Anna Gringauzea4654a42015-10-16 12:15:22 -0700992
Neil MacIntosh68064d62015-11-03 19:17:11 -0800993 value_type boundary;
994 std::remove_const_t<value_type> curr;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700995};
996
997template <typename IndexType>
Neil MacIntoshd5316802015-09-30 21:54:08 -0700998bounds_iterator<IndexType> operator+(typename bounds_iterator<IndexType>::difference_type n, const bounds_iterator<IndexType>& rhs) noexcept
Neil MacIntosha9dcbe02015-08-20 18:09:14 -0700999{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001000 return rhs + n;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001001}
1002
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001003//
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001004// begin definitions of basic_span
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001005//
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001006namespace details
1007{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001008 template <typename Bounds>
1009 constexpr std::enable_if_t<std::is_same<typename Bounds::mapping_type, generalized_mapping_tag>::value, typename Bounds::index_type> make_stride(const Bounds& bnd) noexcept
1010 {
1011 return bnd.strides();
1012 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001013
Neil MacIntosh68064d62015-11-03 19:17:11 -08001014 // Make a stride vector from bounds, assuming contiguous memory.
1015 template <typename Bounds>
1016 constexpr std::enable_if_t<std::is_same<typename Bounds::mapping_type, contiguous_mapping_tag>::value, typename Bounds::index_type> make_stride(const Bounds& bnd) noexcept
1017 {
1018 auto extents = bnd.index_bounds();
1019 typename Bounds::size_type stride[Bounds::rank] = {};
Anna Gringauzedb384972015-10-05 12:34:23 -07001020
Neil MacIntosh68064d62015-11-03 19:17:11 -08001021 stride[Bounds::rank - 1] = 1;
1022 for (size_t i = 1; i < Bounds::rank; ++i)
1023 {
1024 stride[Bounds::rank - i - 1] = stride[Bounds::rank - i] * extents[Bounds::rank - i];
1025 }
1026 return{ stride };
1027 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001028
Neil MacIntosh68064d62015-11-03 19:17:11 -08001029 template <typename BoundsSrc, typename BoundsDest>
1030 void verifyBoundsReshape(const BoundsSrc &src, const BoundsDest &dest)
1031 {
1032 static_assert(is_bounds<BoundsSrc>::value && is_bounds<BoundsDest>::value, "The src type and dest type must be bounds");
1033 static_assert(std::is_same<typename BoundsSrc::mapping_type, contiguous_mapping_tag>::value, "The source type must be a contiguous bounds");
1034 static_assert(BoundsDest::static_size == dynamic_range || BoundsSrc::static_size == dynamic_range || BoundsDest::static_size == BoundsSrc::static_size, "The source bounds must have same size as dest bounds");
1035 fail_fast_assert(src.size() == dest.size());
1036 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001037
1038
1039} // namespace details
1040
1041template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001042class contiguous_span_iterator;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001043template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001044class general_span_iterator;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001045enum class byte : std::uint8_t {};
1046
1047template <typename ValueType, typename BoundsType>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001048class basic_span
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001049{
1050public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001051 static const size_t rank = BoundsType::rank;
1052 using bounds_type = BoundsType;
1053 using size_type = typename bounds_type::size_type;
1054 using index_type = typename bounds_type::index_type;
1055 using value_type = ValueType;
1056 using const_value_type = std::add_const_t<value_type>;
1057 using pointer = ValueType*;
1058 using reference = ValueType&;
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001059 using iterator = std::conditional_t<std::is_same<typename BoundsType::mapping_type, contiguous_mapping_tag>::value, contiguous_span_iterator<basic_span>, general_span_iterator<basic_span>>;
1060 using const_iterator = std::conditional_t<std::is_same<typename BoundsType::mapping_type, contiguous_mapping_tag>::value, contiguous_span_iterator<basic_span<const_value_type, BoundsType>>, general_span_iterator<basic_span<const_value_type, BoundsType>>>;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001061 using reverse_iterator = std::reverse_iterator<iterator>;
1062 using const_reverse_iterator = std::reverse_iterator<const_iterator>;
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001063 using sliced_type = std::conditional_t<rank == 1, value_type, basic_span<value_type, typename BoundsType::sliced_type>>;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001064
1065private:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001066 pointer m_pdata;
1067 bounds_type m_bounds;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001068
1069public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001070 constexpr bounds_type bounds() const noexcept
1071 {
1072 return m_bounds;
1073 }
1074 template <size_t Dim = 0>
1075 constexpr size_type extent() const noexcept
1076 {
1077 static_assert(Dim < rank, "dimension should be less than rank (dimension count starts from 0)");
1078 return m_bounds.template extent<Dim>();
1079 }
1080 constexpr size_type size() const noexcept
1081 {
1082 return m_bounds.size();
1083 }
1084 constexpr reference operator[](const index_type& idx) const
1085 {
1086 return m_pdata[m_bounds.linearize(idx)];
1087 }
1088 constexpr pointer data() const noexcept
1089 {
1090 return m_pdata;
1091 }
1092 template <bool Enabled = (rank > 1), typename Ret = std::enable_if_t<Enabled, sliced_type>>
1093 constexpr Ret operator[](size_type idx) const
1094 {
1095 fail_fast_assert(idx < m_bounds.size(), "index is out of bounds of the array");
1096 const size_type ridx = idx * m_bounds.stride();
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001097
Neil MacIntosh68064d62015-11-03 19:17:11 -08001098 fail_fast_assert(ridx < m_bounds.total_size(), "index is out of bounds of the underlying data");
1099 return Ret {m_pdata + ridx, m_bounds.slice()};
1100 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001101
Neil MacIntosh68064d62015-11-03 19:17:11 -08001102 constexpr operator bool () const noexcept
1103 {
1104 return m_pdata != nullptr;
1105 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001106
Neil MacIntosh68064d62015-11-03 19:17:11 -08001107 constexpr iterator begin() const
1108 {
1109 return iterator {this, true};
1110 }
1111 constexpr iterator end() const
1112 {
1113 return iterator {this, false};
1114 }
1115 constexpr const_iterator cbegin() const
1116 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001117 return const_iterator {reinterpret_cast<const basic_span<const value_type, bounds_type> *>(this), true};
Neil MacIntosh68064d62015-11-03 19:17:11 -08001118 }
1119 constexpr const_iterator cend() const
1120 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001121 return const_iterator {reinterpret_cast<const basic_span<const value_type, bounds_type> *>(this), false};
Neil MacIntosh68064d62015-11-03 19:17:11 -08001122 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001123
Neil MacIntosh68064d62015-11-03 19:17:11 -08001124 constexpr reverse_iterator rbegin() const
1125 {
1126 return reverse_iterator {end()};
1127 }
1128 constexpr reverse_iterator rend() const
1129 {
1130 return reverse_iterator {begin()};
1131 }
1132 constexpr const_reverse_iterator crbegin() const
1133 {
1134 return const_reverse_iterator {cend()};
1135 }
1136 constexpr const_reverse_iterator crend() const
1137 {
1138 return const_reverse_iterator {cbegin()};
1139 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001140
Neil MacIntosh68064d62015-11-03 19:17:11 -08001141 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001142 constexpr bool operator== (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001143 {
1144 return m_bounds.size() == other.m_bounds.size() &&
1145 (m_pdata == other.m_pdata || std::equal(this->begin(), this->end(), other.begin()));
1146 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001147
Neil MacIntosh68064d62015-11-03 19:17:11 -08001148 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001149 constexpr bool operator!= (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001150 {
1151 return !(*this == other);
1152 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001153
Neil MacIntosh68064d62015-11-03 19:17:11 -08001154 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001155 constexpr bool operator< (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001156 {
1157 return std::lexicographical_compare(this->begin(), this->end(), other.begin(), other.end());
1158 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001159
Neil MacIntosh68064d62015-11-03 19:17:11 -08001160 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001161 constexpr bool operator<= (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001162 {
1163 return !(other < *this);
1164 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001165
Neil MacIntosh68064d62015-11-03 19:17:11 -08001166 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001167 constexpr bool operator> (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001168 {
1169 return (other < *this);
1170 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001171
Neil MacIntosh68064d62015-11-03 19:17:11 -08001172 template <typename OtherValueType, typename OtherBoundsType, typename Dummy = std::enable_if_t<std::is_same<std::remove_cv_t<value_type>, std::remove_cv_t<OtherValueType>>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001173 constexpr bool operator>= (const basic_span<OtherValueType, OtherBoundsType> & other) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001174 {
1175 return !(*this < other);
1176 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001177
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001178public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001179 template <typename OtherValueType, typename OtherBounds,
1180 typename Dummy = std::enable_if_t<std::is_convertible<OtherValueType(*)[], value_type(*)[]>::value
1181 && std::is_convertible<OtherBounds, bounds_type>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001182 constexpr basic_span(const basic_span<OtherValueType, OtherBounds> & other ) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001183 : m_pdata(other.m_pdata), m_bounds(other.m_bounds)
1184 {
1185 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001186protected:
1187
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001188 constexpr basic_span(pointer data, bounds_type bound) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001189 : m_pdata(data)
1190 , m_bounds(std::move(bound))
1191 {
1192 fail_fast_assert((m_bounds.size() > 0 && data != nullptr) || m_bounds.size() == 0);
1193 }
1194 template <typename T>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001195 constexpr basic_span(T *data, std::enable_if_t<std::is_same<value_type, std::remove_all_extents_t<T>>::value, bounds_type> bound) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001196 : m_pdata(reinterpret_cast<pointer>(data))
1197 , m_bounds(std::move(bound))
1198 {
1199 fail_fast_assert((m_bounds.size() > 0 && data != nullptr) || m_bounds.size() == 0);
1200 }
1201 template <typename DestBounds>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001202 constexpr basic_span<value_type, DestBounds> as_span(const DestBounds &bounds)
Neil MacIntosh68064d62015-11-03 19:17:11 -08001203 {
1204 details::verifyBoundsReshape(m_bounds, bounds);
1205 return {m_pdata, bounds};
1206 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001207private:
1208
Neil MacIntosh68064d62015-11-03 19:17:11 -08001209 friend iterator;
1210 friend const_iterator;
1211 template <typename ValueType2, typename BoundsType2>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001212 friend class basic_span;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001213};
1214
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001215template <std::ptrdiff_t DimSize = dynamic_range>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001216struct dim
1217{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001218 static const std::ptrdiff_t value = DimSize;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001219};
1220template <>
1221struct dim<dynamic_range>
1222{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001223 static const std::ptrdiff_t value = dynamic_range;
1224 const std::ptrdiff_t dvalue;
1225 dim(std::ptrdiff_t size) : dvalue(size) {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001226};
1227
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001228template <typename ValueType, std::ptrdiff_t FirstDimension = dynamic_range, std::ptrdiff_t... RestDimensions>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001229class span;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001230
1231template <typename ValueType, size_t Rank>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001232class strided_span;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001233
1234namespace details
1235{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001236 template <typename T, typename = std::true_type>
1237 struct ArrayViewTypeTraits
1238 {
1239 using value_type = T;
1240 using size_type = size_t;
1241 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001242
Neil MacIntosh68064d62015-11-03 19:17:11 -08001243 template <typename Traits>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001244 struct ArrayViewTypeTraits<Traits, typename std::is_reference<typename Traits::span_traits &>::type>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001245 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001246 using value_type = typename Traits::span_traits::value_type;
1247 using size_type = typename Traits::span_traits::size_type;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001248 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001249
Neil MacIntosh68064d62015-11-03 19:17:11 -08001250 template <typename T, std::ptrdiff_t... Ranks>
1251 struct ArrayViewArrayTraits {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001252 using type = span<T, Ranks...>;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001253 using value_type = T;
1254 using bounds_type = static_bounds<Ranks...>;
1255 using pointer = T*;
1256 using reference = T&;
1257 };
1258 template <typename T, std::ptrdiff_t N, std::ptrdiff_t... Ranks>
1259 struct ArrayViewArrayTraits<T[N], Ranks...> : ArrayViewArrayTraits<T, Ranks..., N> {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001260
Neil MacIntosh68064d62015-11-03 19:17:11 -08001261 template <typename BoundsType>
1262 BoundsType newBoundsHelperImpl(std::ptrdiff_t totalSize, std::true_type) // dynamic size
1263 {
1264 fail_fast_assert(totalSize <= PTRDIFF_MAX);
1265 return BoundsType{totalSize};
1266 }
1267 template <typename BoundsType>
1268 BoundsType newBoundsHelperImpl(std::ptrdiff_t totalSize, std::false_type) // static size
1269 {
1270 fail_fast_assert(BoundsType::static_size == totalSize);
1271 return {};
1272 }
1273 template <typename BoundsType>
1274 BoundsType newBoundsHelper(std::ptrdiff_t totalSize)
1275 {
1276 static_assert(BoundsType::dynamic_rank <= 1, "dynamic rank must less or equal to 1");
1277 return newBoundsHelperImpl<BoundsType>(totalSize, std::integral_constant<bool, BoundsType::dynamic_rank == 1>());
1278 }
1279
1280 struct Sep{};
1281
1282 template <typename T, typename... Args>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001283 T static_as_span_helper(Sep, Args... args)
Neil MacIntosh68064d62015-11-03 19:17:11 -08001284 {
1285 return T{static_cast<typename T::size_type>(args)...};
1286 }
1287 template <typename T, typename Arg, typename... Args>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001288 std::enable_if_t<!std::is_same<Arg, dim<dynamic_range>>::value && !std::is_same<Arg, Sep>::value, T> static_as_span_helper(Arg, Args... args)
Neil MacIntosh68064d62015-11-03 19:17:11 -08001289 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001290 return static_as_span_helper<T>(args...);
Neil MacIntosh68064d62015-11-03 19:17:11 -08001291 }
1292 template <typename T, typename... Args>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001293 T static_as_span_helper(dim<dynamic_range> val, Args ... args)
Neil MacIntosh68064d62015-11-03 19:17:11 -08001294 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001295 return static_as_span_helper<T>(args..., val.dvalue);
Neil MacIntosh68064d62015-11-03 19:17:11 -08001296 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001297
Neil MacIntosh68064d62015-11-03 19:17:11 -08001298 template <typename ...Dimensions>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001299 struct static_as_span_static_bounds_helper
Neil MacIntosh68064d62015-11-03 19:17:11 -08001300 {
1301 using type = static_bounds<(Dimensions::value)...>;
1302 };
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001303
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001304 template <typename T>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001305 struct is_span_oracle : std::false_type
Neil MacIntosh68064d62015-11-03 19:17:11 -08001306 {};
1307
1308 template <typename ValueType, std::ptrdiff_t FirstDimension, std::ptrdiff_t... RestDimensions>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001309 struct is_span_oracle<span<ValueType, FirstDimension, RestDimensions...>> : std::true_type
Neil MacIntosh68064d62015-11-03 19:17:11 -08001310 {};
1311
1312 template <typename ValueType, std::ptrdiff_t Rank>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001313 struct is_span_oracle<strided_span<ValueType, Rank>> : std::true_type
Neil MacIntosh68064d62015-11-03 19:17:11 -08001314 {};
1315
1316 template <typename T>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001317 struct is_span : is_span_oracle<std::remove_cv_t<T>>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001318 {};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001319
1320}
1321
1322
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001323template <typename ValueType, std::ptrdiff_t FirstDimension, std::ptrdiff_t... RestDimensions>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001324class span : public basic_span <ValueType, static_bounds <FirstDimension, RestDimensions...>>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001325{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001326 template <typename ValueType2, std::ptrdiff_t FirstDimension2,
1327 std::ptrdiff_t... RestDimensions2>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001328 friend class span;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001329
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001330 using Base = basic_span<ValueType, static_bounds<FirstDimension, RestDimensions...>>;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001331
1332public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001333 using typename Base::bounds_type;
1334 using typename Base::size_type;
1335 using typename Base::pointer;
1336 using typename Base::value_type;
1337 using typename Base::index_type;
1338 using typename Base::iterator;
1339 using typename Base::const_iterator;
1340 using typename Base::reference;
1341 using Base::rank;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001342
1343public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001344 // basic
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001345 constexpr span(pointer ptr, size_type size) : Base(ptr, bounds_type{ size })
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001346 {}
1347
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001348 constexpr span(pointer ptr, bounds_type bounds) : Base(ptr, std::move(bounds))
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001349 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001350
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001351 constexpr span(std::nullptr_t) : Base(nullptr, bounds_type{})
Neil MacIntosh68064d62015-11-03 19:17:11 -08001352 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001353
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001354 constexpr span(std::nullptr_t, size_type size) : Base(nullptr, bounds_type{})
Neil MacIntosh68064d62015-11-03 19:17:11 -08001355 {
1356 fail_fast_assert(size == 0);
1357 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001358
Neil MacIntosh68064d62015-11-03 19:17:11 -08001359 // default
1360 template <std::ptrdiff_t DynamicRank = bounds_type::dynamic_rank, typename = std::enable_if_t<DynamicRank != 0>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001361 constexpr span() : Base(nullptr, bounds_type())
Neil MacIntosh68064d62015-11-03 19:17:11 -08001362 {}
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001363
Neil MacIntosh68064d62015-11-03 19:17:11 -08001364 // from n-dimensions dynamic array (e.g. new int[m][4]) (precedence will be lower than the 1-dimension pointer)
1365 template <typename T, typename Helper = details::ArrayViewArrayTraits<T, dynamic_range>
Neil MacIntoshace9ab92015-10-23 19:49:17 -07001366 /*typename Dummy = std::enable_if_t<std::is_convertible<Helper::value_type (*)[], typename Base::value_type (*)[]>::value>*/>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001367 constexpr span(T* const& data, size_type size) : Base(data, typename Helper::bounds_type{size})
Neil MacIntosh68064d62015-11-03 19:17:11 -08001368 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001369
Neil MacIntosh68064d62015-11-03 19:17:11 -08001370 // from n-dimensions static array
1371 template <typename T, size_t N, typename Helper = details::ArrayViewArrayTraits<T, N>,
Neil MacIntosh41517ff2015-11-04 02:11:49 +00001372 typename = std::enable_if_t<std::is_convertible<typename Helper::value_type(*)[], typename Base::value_type(*)[]>::value>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001373 constexpr span (T (&arr)[N]) : Base(arr, typename Helper::bounds_type())
Neil MacIntosh68064d62015-11-03 19:17:11 -08001374 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001375
Neil MacIntosh68064d62015-11-03 19:17:11 -08001376 // from n-dimensions static array with size
1377 template <typename T, size_t N, typename Helper = details::ArrayViewArrayTraits<T, N>,
1378 typename = std::enable_if_t<std::is_convertible<typename Helper::value_type(*)[], typename Base::value_type(*)[]>::value>
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001379 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001380 constexpr span(T(&arr)[N], size_type size) : Base(arr, typename Helper::bounds_type{size})
Neil MacIntosh68064d62015-11-03 19:17:11 -08001381 {
1382 fail_fast_assert(size <= N);
1383 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001384
Neil MacIntosh68064d62015-11-03 19:17:11 -08001385 // from std array
1386 template <size_t N,
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001387 typename Dummy = std::enable_if_t<std::is_convertible<static_bounds<N>, typename Base::bounds_type>::value>
1388 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001389 constexpr span (std::array<std::remove_const_t<value_type>, N> & arr) : Base(arr.data(), static_bounds<N>())
Neil MacIntosh68064d62015-11-03 19:17:11 -08001390 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001391
Neil MacIntosh68064d62015-11-03 19:17:11 -08001392 template <size_t N,
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001393 typename Dummy = std::enable_if_t<std::is_convertible<static_bounds<N>, typename Base::bounds_type>::value
1394 && std::is_const<value_type>::value>
1395 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001396 constexpr span (const std::array<std::remove_const_t<value_type>, N> & arr) : Base(arr.data(), static_bounds<N>())
Neil MacIntosh68064d62015-11-03 19:17:11 -08001397 {}
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001398
Neil MacIntosh68064d62015-11-03 19:17:11 -08001399 // from begin, end pointers. We don't provide iterator pair since no way to guarantee the contiguity
1400 template <typename Ptr,
1401 typename Dummy = std::enable_if_t<std::is_convertible<Ptr, pointer>::value
1402 && details::LessThan<Base::bounds_type::dynamic_rank, 2>::value>
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001403 > // remove literal 0 case
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001404 constexpr span (pointer begin, Ptr end) : Base(begin, details::newBoundsHelper<typename Base::bounds_type>(static_cast<pointer>(end) - begin))
Neil MacIntosh68064d62015-11-03 19:17:11 -08001405 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001406
Neil MacIntosh68064d62015-11-03 19:17:11 -08001407 // from containers. It must has .size() and .data() two function signatures
1408 template <typename Cont, typename DataType = typename Cont::value_type,
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001409 typename Dummy = std::enable_if_t<!details::is_span<Cont>::value
Neil MacIntosh68064d62015-11-03 19:17:11 -08001410 && std::is_convertible<DataType (*)[], typename Base::value_type (*)[]>::value
1411 && std::is_same<std::decay_t<decltype(std::declval<Cont>().size(), *std::declval<Cont>().data())>, DataType>::value>
1412 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001413 constexpr span (Cont& cont) : Base(static_cast<pointer>(cont.data()), details::newBoundsHelper<typename Base::bounds_type>(cont.size()))
Neil MacIntosh68064d62015-11-03 19:17:11 -08001414 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001415
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001416 constexpr span(const span &) = default;
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001417
Neil MacIntosh68064d62015-11-03 19:17:11 -08001418 // convertible
1419 template <typename OtherValueType, std::ptrdiff_t... OtherDimensions,
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001420 typename BaseType = basic_span<ValueType, static_bounds<FirstDimension, RestDimensions...>>,
1421 typename OtherBaseType = basic_span<OtherValueType, static_bounds<OtherDimensions...>>,
Neil MacIntosh68064d62015-11-03 19:17:11 -08001422 typename Dummy = std::enable_if_t<std::is_convertible<OtherBaseType, BaseType>::value>
1423 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001424 constexpr span(const span<OtherValueType, OtherDimensions...> &av)
1425 : Base(static_cast<const typename span<OtherValueType, OtherDimensions...>::Base&>(av))
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001426 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001427
Neil MacIntosh68064d62015-11-03 19:17:11 -08001428 // reshape
Neil MacIntosh14d50a62015-11-03 12:44:09 -08001429 // DimCount here is a workaround for a bug in MSVC 2015
1430 template <typename... Dimensions2, size_t DimCount = sizeof...(Dimensions2), typename = std::enable_if_t<(DimCount > 0)>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001431 constexpr span<ValueType, Dimensions2::value...> as_span(Dimensions2... dims)
Neil MacIntosh68064d62015-11-03 19:17:11 -08001432 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001433 using BoundsType = typename span<ValueType, (Dimensions2::value)...>::bounds_type;
1434 auto tobounds = details::static_as_span_helper<BoundsType>(dims..., details::Sep{});
Neil MacIntosh68064d62015-11-03 19:17:11 -08001435 details::verifyBoundsReshape(this->bounds(), tobounds);
1436 return {this->data(), tobounds};
1437 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001438
Neil MacIntosh68064d62015-11-03 19:17:11 -08001439 // to bytes array
1440 template <bool Enabled = std::is_standard_layout<std::decay_t<ValueType>>::value>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001441 auto as_bytes() const noexcept -> span<const byte>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001442 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001443 static_assert(Enabled, "The value_type of span must be standarded layout");
Neil MacIntosh68064d62015-11-03 19:17:11 -08001444 return { reinterpret_cast<const byte*>(this->data()), this->bytes() };
1445 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001446
Neil MacIntosh68064d62015-11-03 19:17:11 -08001447 template <bool Enabled = std::is_standard_layout<std::decay_t<ValueType>>::value>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001448 auto as_writeable_bytes() const noexcept -> span<byte>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001449 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001450 static_assert(Enabled, "The value_type of span must be standarded layout");
Neil MacIntosh68064d62015-11-03 19:17:11 -08001451 return { reinterpret_cast<byte*>(this->data()), this->bytes() };
1452 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001453
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001454 // from bytes array
Neil MacIntosh68064d62015-11-03 19:17:11 -08001455 template<typename U, bool IsByte = std::is_same<value_type, const byte>::value, typename = std::enable_if_t<IsByte && sizeof...(RestDimensions) == 0>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001456 constexpr auto as_span() const noexcept -> span<const U, (Base::bounds_type::static_size != dynamic_range ? static_cast<std::ptrdiff_t>(static_cast<size_t>(Base::bounds_type::static_size) / sizeof(U)) : dynamic_range)>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001457 {
1458 static_assert(std::is_standard_layout<U>::value && (Base::bounds_type::static_size == dynamic_range || Base::bounds_type::static_size % static_cast<size_type>(sizeof(U)) == 0),
1459 "Target type must be standard layout and its size must match the byte array size");
1460 fail_fast_assert((this->bytes() % sizeof(U)) == 0 && (this->bytes() / sizeof(U)) < PTRDIFF_MAX);
1461 return { reinterpret_cast<const U*>(this->data()), this->bytes() / static_cast<size_type>(sizeof(U)) };
1462 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001463
Neil MacIntosh68064d62015-11-03 19:17:11 -08001464 template<typename U, bool IsByte = std::is_same<value_type, byte>::value, typename = std::enable_if_t<IsByte && sizeof...(RestDimensions) == 0>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001465 constexpr auto as_span() const noexcept -> span<U, (Base::bounds_type::static_size != dynamic_range ? static_cast<ptrdiff_t>(static_cast<size_t>(Base::bounds_type::static_size) / sizeof(U)) : dynamic_range)>
Neil MacIntosh68064d62015-11-03 19:17:11 -08001466 {
1467 static_assert(std::is_standard_layout<U>::value && (Base::bounds_type::static_size == dynamic_range || Base::bounds_type::static_size % static_cast<size_t>(sizeof(U)) == 0),
1468 "Target type must be standard layout and its size must match the byte array size");
1469 fail_fast_assert((this->bytes() % sizeof(U)) == 0);
1470 return { reinterpret_cast<U*>(this->data()), this->bytes() / static_cast<size_type>(sizeof(U)) };
1471 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001472
Neil MacIntosh68064d62015-11-03 19:17:11 -08001473 // section on linear space
1474 template<std::ptrdiff_t Count>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001475 constexpr span<ValueType, Count> first() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001476 {
1477 static_assert(bounds_type::static_size == dynamic_range || Count <= bounds_type::static_size, "Index is out of bound");
1478 fail_fast_assert(bounds_type::static_size != dynamic_range || Count <= this->size()); // ensures we only check condition when needed
1479 return { this->data(), Count };
1480 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001481
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001482 constexpr span<ValueType, dynamic_range> first(size_type count) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001483 {
1484 fail_fast_assert(count <= this->size());
1485 return { this->data(), count };
1486 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001487
Neil MacIntosh68064d62015-11-03 19:17:11 -08001488 template<std::ptrdiff_t Count>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001489 constexpr span<ValueType, Count> last() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001490 {
1491 static_assert(bounds_type::static_size == dynamic_range || Count <= bounds_type::static_size, "Index is out of bound");
1492 fail_fast_assert(bounds_type::static_size != dynamic_range || Count <= this->size());
1493 return { this->data() + this->size() - Count, Count };
1494 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001495
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001496 constexpr span<ValueType, dynamic_range> last(size_type count) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001497 {
1498 fail_fast_assert(count <= this->size());
1499 return { this->data() + this->size() - count, count };
1500 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001501
Neil MacIntosh68064d62015-11-03 19:17:11 -08001502 template<std::ptrdiff_t Offset, std::ptrdiff_t Count>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001503 constexpr span<ValueType, Count> sub() const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001504 {
1505 static_assert(bounds_type::static_size == dynamic_range || ((Offset == 0 || Offset <= bounds_type::static_size) && Offset + Count <= bounds_type::static_size), "Index is out of bound");
1506 fail_fast_assert(bounds_type::static_size != dynamic_range || ((Offset == 0 || Offset <= this->size()) && Offset + Count <= this->size()));
1507 return { this->data() + Offset, Count };
1508 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001509
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001510 constexpr span<ValueType, dynamic_range> sub(size_type offset, size_type count = dynamic_range) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001511 {
1512 fail_fast_assert((offset == 0 || offset <= this->size()) && (count == dynamic_range || (offset + count) <= this->size()));
1513 return { this->data() + offset, count == dynamic_range ? this->length() - offset : count };
1514 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001515
Neil MacIntosh68064d62015-11-03 19:17:11 -08001516 // size
1517 constexpr size_type length() const noexcept
1518 {
1519 return this->size();
1520 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001521
Neil MacIntosh68064d62015-11-03 19:17:11 -08001522 constexpr size_type used_length() const noexcept
1523 {
1524 return length();
1525 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001526
Neil MacIntosh68064d62015-11-03 19:17:11 -08001527 constexpr size_type bytes() const noexcept
1528 {
1529 return sizeof(value_type) * this->size();
1530 }
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001531
Neil MacIntosh68064d62015-11-03 19:17:11 -08001532 constexpr size_type used_bytes() const noexcept
1533 {
1534 return bytes();
1535 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001536
Neil MacIntosh68064d62015-11-03 19:17:11 -08001537 // section
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001538 constexpr strided_span<ValueType, rank> section(index_type origin, index_type extents) const
Neil MacIntosh68064d62015-11-03 19:17:11 -08001539 {
1540 size_type size = this->bounds().total_size() - this->bounds().linearize(origin);
1541 return{ &this->operator[](origin), size, strided_bounds<rank> {extents, details::make_stride(Base::bounds())} };
1542 }
1543
1544 constexpr reference operator[](const index_type& idx) const
1545 {
1546 return Base::operator[](idx);
1547 }
1548
1549 template <bool Enabled = (rank > 1), typename Dummy = std::enable_if_t<Enabled>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001550 constexpr span<ValueType, RestDimensions...> operator[](size_type idx) const
Neil MacIntosh68064d62015-11-03 19:17:11 -08001551 {
1552 auto ret = Base::operator[](idx);
1553 return{ ret.data(), ret.bounds() };
1554 }
Neil MacIntosh9f9fad92015-08-27 18:13:49 -07001555
Neil MacIntosh68064d62015-11-03 19:17:11 -08001556 using Base::operator==;
1557 using Base::operator!=;
1558 using Base::operator<;
1559 using Base::operator<=;
1560 using Base::operator>;
1561 using Base::operator>=;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001562};
1563
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001564template <typename T, std::ptrdiff_t... Dimensions>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001565constexpr auto as_span(T* const& ptr, dim<Dimensions>... args) -> span<std::remove_all_extents_t<T>, Dimensions...>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001566{
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001567 return {reinterpret_cast<std::remove_all_extents_t<T>*>(ptr), details::static_as_span_helper<static_bounds<Dimensions...>>(args..., details::Sep{})};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001568}
1569
1570template <typename T>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001571constexpr auto as_span (T* arr, std::ptrdiff_t len) -> typename details::ArrayViewArrayTraits<T, dynamic_range>::type
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001572{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001573 return {reinterpret_cast<std::remove_all_extents_t<T>*>(arr), len};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001574}
1575
1576template <typename T, size_t N>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001577constexpr auto as_span (T (&arr)[N]) -> typename details::ArrayViewArrayTraits<T, N>::type
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001578{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001579 return {arr};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001580}
1581
1582template <typename T, size_t N>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001583constexpr span<const T, N> as_span(const std::array<T, N> &arr)
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001584{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001585 return {arr};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001586}
1587
1588template <typename T, size_t N>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001589constexpr span<const T, N> as_span(const std::array<T, N> &&) = delete;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001590
1591template <typename T, size_t N>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001592constexpr span<T, N> as_span(std::array<T, N> &arr)
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001593{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001594 return {arr};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001595}
1596
1597template <typename T>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001598constexpr span<T, dynamic_range> as_span(T *begin, T *end)
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001599{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001600 return {begin, end};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001601}
1602
1603template <typename Cont>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001604constexpr auto as_span(Cont &arr) -> std::enable_if_t<!details::is_span<std::decay_t<Cont>>::value,
1605 span<std::remove_reference_t<decltype(arr.size(), *arr.data())>, dynamic_range>>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001606{
Neil MacIntoshace9ab92015-10-23 19:49:17 -07001607 fail_fast_assert(arr.size() < PTRDIFF_MAX);
Neil MacIntosh68064d62015-11-03 19:17:11 -08001608 return {arr.data(), static_cast<std::ptrdiff_t>(arr.size())};
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001609}
1610
1611template <typename Cont>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001612constexpr auto as_span(Cont &&arr) -> std::enable_if_t<!details::is_span<std::decay_t<Cont>>::value,
1613 span<std::remove_reference_t<decltype(arr.size(), *arr.data())>, dynamic_range>> = delete;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001614
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001615template <typename ValueType, size_t Rank>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001616class strided_span : public basic_span<ValueType, strided_bounds<Rank>>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001617{
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001618 using Base = basic_span<ValueType, strided_bounds<Rank>>;
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001619
Neil MacIntosh68064d62015-11-03 19:17:11 -08001620 template<typename OtherValue, size_t OtherRank>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001621 friend class strided_span;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001622
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001623public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001624 using Base::rank;
1625 using typename Base::bounds_type;
1626 using typename Base::size_type;
1627 using typename Base::pointer;
1628 using typename Base::value_type;
1629 using typename Base::index_type;
1630 using typename Base::iterator;
1631 using typename Base::const_iterator;
1632 using typename Base::reference;
1633
1634 // from static array of size N
1635 template<size_type N>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001636 strided_span(value_type(&values)[N], bounds_type bounds) : Base(values, std::move(bounds))
Neil MacIntosh68064d62015-11-03 19:17:11 -08001637 {
1638 fail_fast_assert(this->bounds().total_size() <= N, "Bounds cross data boundaries");
1639 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001640
Neil MacIntosh68064d62015-11-03 19:17:11 -08001641 // from raw data
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001642 strided_span(pointer ptr, size_type size, bounds_type bounds): Base(ptr, std::move(bounds))
Neil MacIntosh68064d62015-11-03 19:17:11 -08001643 {
1644 fail_fast_assert(this->bounds().total_size() <= size, "Bounds cross data boundaries");
1645 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001646
Neil MacIntosh68064d62015-11-03 19:17:11 -08001647 // from array view
1648 template <std::ptrdiff_t... Dimensions, typename Dummy = std::enable_if<sizeof...(Dimensions) == Rank>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001649 strided_span(span<ValueType, Dimensions...> av, bounds_type bounds) : Base(av.data(), std::move(bounds))
Neil MacIntosh68064d62015-11-03 19:17:11 -08001650 {
1651 fail_fast_assert(this->bounds().total_size() <= av.bounds().total_size(), "Bounds cross data boundaries");
1652 }
1653
1654 // convertible
1655 template <typename OtherValueType,
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001656 typename BaseType = basic_span<ValueType, strided_bounds<Rank>>,
1657 typename OtherBaseType = basic_span<OtherValueType, strided_bounds<Rank>>,
Neil MacIntosh68064d62015-11-03 19:17:11 -08001658 typename Dummy = std::enable_if_t<std::is_convertible<OtherBaseType, BaseType>::value>
1659 >
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001660 constexpr strided_span(const strided_span<OtherValueType, Rank> &av) : Base(static_cast<const typename strided_span<OtherValueType, Rank>::Base &>(av)) // static_cast is required
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001661 {}
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001662
Neil MacIntosh68064d62015-11-03 19:17:11 -08001663 // convert from bytes
1664 template <typename OtherValueType>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001665 strided_span<typename std::enable_if<std::is_same<value_type, const byte>::value, OtherValueType>::type, rank> as_strided_span() const
Neil MacIntosh68064d62015-11-03 19:17:11 -08001666 {
1667 static_assert((sizeof(OtherValueType) >= sizeof(value_type)) && (sizeof(OtherValueType) % sizeof(value_type) == 0), "OtherValueType should have a size to contain a multiple of ValueTypes");
1668 auto d = static_cast<size_type>(sizeof(OtherValueType) / sizeof(value_type));
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001669
Neil MacIntosh68064d62015-11-03 19:17:11 -08001670 size_type size = this->bounds().total_size() / d;
1671 return{ (OtherValueType*)this->data(), size, bounds_type{ resize_extent(this->bounds().index_bounds(), d), resize_stride(this->bounds().strides(), d)} };
1672 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001673
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001674 strided_span section(index_type origin, index_type extents) const
Neil MacIntosh68064d62015-11-03 19:17:11 -08001675 {
1676 size_type size = this->bounds().total_size() - this->bounds().linearize(origin);
1677 return { &this->operator[](origin), size, bounds_type {extents, details::make_stride(Base::bounds())}};
1678 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001679
Neil MacIntosh68064d62015-11-03 19:17:11 -08001680 constexpr reference operator[](const index_type& idx) const
1681 {
1682 return Base::operator[](idx);
1683 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001684
Neil MacIntosh68064d62015-11-03 19:17:11 -08001685 template <bool Enabled = (rank > 1), typename Dummy = std::enable_if_t<Enabled>>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001686 constexpr strided_span<value_type, rank-1> operator[](size_type idx) const
Neil MacIntosh68064d62015-11-03 19:17:11 -08001687 {
1688 auto ret = Base::operator[](idx);
1689 return{ ret.data(), ret.bounds().total_size(), ret.bounds() };
1690 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001691
1692private:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001693 static index_type resize_extent(const index_type& extent, std::ptrdiff_t d)
1694 {
1695 fail_fast_assert(extent[rank - 1] >= d && (extent[rank-1] % d == 0), "The last dimension of the array needs to contain a multiple of new type elements");
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001696
Neil MacIntosh68064d62015-11-03 19:17:11 -08001697 index_type ret = extent;
1698 ret[rank - 1] /= d;
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001699
Neil MacIntosh68064d62015-11-03 19:17:11 -08001700 return ret;
1701 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001702
Neil MacIntosh68064d62015-11-03 19:17:11 -08001703 template <bool Enabled = (rank == 1), typename Dummy = std::enable_if_t<Enabled>>
1704 static index_type resize_stride(const index_type& strides, std::ptrdiff_t , void * = 0)
1705 {
1706 fail_fast_assert(strides[rank - 1] == 1, "Only strided arrays with regular strides can be resized");
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001707
Neil MacIntosh68064d62015-11-03 19:17:11 -08001708 return strides;
1709 }
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001710
Neil MacIntosh68064d62015-11-03 19:17:11 -08001711 template <bool Enabled = (rank > 1), typename Dummy = std::enable_if_t<Enabled>>
1712 static index_type resize_stride(const index_type& strides, std::ptrdiff_t d)
1713 {
1714 fail_fast_assert(strides[rank - 1] == 1, "Only strided arrays with regular strides can be resized");
1715 fail_fast_assert(strides[rank - 2] >= d && (strides[rank - 2] % d == 0), "The strides must have contiguous chunks of memory that can contain a multiple of new type elements");
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001716
Neil MacIntosh68064d62015-11-03 19:17:11 -08001717 for (size_t i = rank - 1; i > 0; --i)
1718 fail_fast_assert((strides[i-1] >= strides[i]) && (strides[i-1] % strides[i] == 0), "Only strided arrays with regular strides can be resized");
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001719
Neil MacIntosh68064d62015-11-03 19:17:11 -08001720 index_type ret = strides / d;
1721 ret[rank - 1] = 1;
Anna Gringauze17ed5c32015-08-30 23:30:15 -07001722
Neil MacIntosh68064d62015-11-03 19:17:11 -08001723 return ret;
1724 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001725};
1726
1727template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001728class contiguous_span_iterator : public std::iterator<std::random_access_iterator_tag, typename ArrayView::value_type>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001729{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001730 using Base = std::iterator<std::random_access_iterator_tag, typename ArrayView::value_type>;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001731public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001732 using typename Base::reference;
1733 using typename Base::pointer;
1734 using typename Base::difference_type;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001735
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001736private:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001737 template <typename ValueType, typename Bounds>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001738 friend class basic_span;
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001739
1740 pointer m_pdata;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001741 const ArrayView * m_validator;
1742 void validateThis() const
1743 {
1744 fail_fast_assert(m_pdata >= m_validator->m_pdata && m_pdata < m_validator->m_pdata + m_validator->size(), "iterator is out of range of the array");
1745 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001746 contiguous_span_iterator (const ArrayView *container, bool isbegin) :
Neil MacIntosh68064d62015-11-03 19:17:11 -08001747 m_pdata(isbegin ? container->m_pdata : container->m_pdata + container->size()), m_validator(container) {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001748public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001749 reference operator*() const noexcept
1750 {
1751 validateThis();
1752 return *m_pdata;
1753 }
1754 pointer operator->() const noexcept
1755 {
1756 validateThis();
1757 return m_pdata;
1758 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001759 contiguous_span_iterator& operator++() noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001760 {
1761 ++m_pdata;
1762 return *this;
1763 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001764 contiguous_span_iterator operator++(int)noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001765 {
1766 auto ret = *this;
1767 ++(*this);
1768 return ret;
1769 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001770 contiguous_span_iterator& operator--() noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001771 {
1772 --m_pdata;
1773 return *this;
1774 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001775 contiguous_span_iterator operator--(int)noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001776 {
1777 auto ret = *this;
1778 --(*this);
1779 return ret;
1780 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001781 contiguous_span_iterator operator+(difference_type n) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001782 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001783 contiguous_span_iterator ret{ *this };
Neil MacIntosh68064d62015-11-03 19:17:11 -08001784 return ret += n;
1785 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001786 contiguous_span_iterator& operator+=(difference_type n) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001787 {
1788 m_pdata += n;
1789 return *this;
1790 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001791 contiguous_span_iterator operator-(difference_type n) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001792 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001793 contiguous_span_iterator ret{ *this };
Neil MacIntosh68064d62015-11-03 19:17:11 -08001794 return ret -= n;
1795 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001796 contiguous_span_iterator& operator-=(difference_type n) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001797 {
1798 return *this += -n;
1799 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001800 difference_type operator-(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001801 {
1802 fail_fast_assert(m_validator == rhs.m_validator);
1803 return m_pdata - rhs.m_pdata;
1804 }
1805 reference operator[](difference_type n) const noexcept
1806 {
1807 return *(*this + n);
1808 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001809 bool operator==(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001810 {
1811 fail_fast_assert(m_validator == rhs.m_validator);
1812 return m_pdata == rhs.m_pdata;
1813 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001814 bool operator!=(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001815 {
1816 return !(*this == rhs);
1817 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001818 bool operator<(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001819 {
1820 fail_fast_assert(m_validator == rhs.m_validator);
1821 return m_pdata < rhs.m_pdata;
1822 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001823 bool operator<=(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001824 {
1825 return !(rhs < *this);
1826 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001827 bool operator>(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001828 {
1829 return rhs < *this;
1830 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001831 bool operator>=(const contiguous_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001832 {
1833 return !(rhs > *this);
1834 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001835 void swap(contiguous_span_iterator& rhs) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001836 {
1837 std::swap(m_pdata, rhs.m_pdata);
1838 std::swap(m_validator, rhs.m_validator);
1839 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001840};
1841
1842template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001843contiguous_span_iterator<ArrayView> operator+(typename contiguous_span_iterator<ArrayView>::difference_type n, const contiguous_span_iterator<ArrayView>& rhs) noexcept
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001844{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001845 return rhs + n;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001846}
1847
1848template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001849class general_span_iterator : public std::iterator<std::random_access_iterator_tag, typename ArrayView::value_type>
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001850{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001851 using Base = std::iterator<std::random_access_iterator_tag, typename ArrayView::value_type>;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001852public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001853 using typename Base::reference;
1854 using typename Base::pointer;
1855 using typename Base::difference_type;
1856 using typename Base::value_type;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001857private:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001858 template <typename ValueType, typename Bounds>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001859 friend class basic_span;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001860
Neil MacIntoshf45fedb2015-10-15 14:29:35 -07001861 const ArrayView * m_container;
Neil MacIntosh68064d62015-11-03 19:17:11 -08001862 typename ArrayView::bounds_type::iterator m_itr;
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001863 general_span_iterator(const ArrayView *container, bool isbegin) :
Neil MacIntosh68064d62015-11-03 19:17:11 -08001864 m_container(container), m_itr(isbegin ? m_container->bounds().begin() : m_container->bounds().end())
1865 {}
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001866public:
Neil MacIntosh68064d62015-11-03 19:17:11 -08001867 reference operator*() noexcept
1868 {
1869 return (*m_container)[*m_itr];
1870 }
1871 pointer operator->() noexcept
1872 {
1873 return &(*m_container)[*m_itr];
1874 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001875 general_span_iterator& operator++() noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001876 {
1877 ++m_itr;
1878 return *this;
1879 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001880 general_span_iterator operator++(int)noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001881 {
1882 auto ret = *this;
1883 ++(*this);
1884 return ret;
1885 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001886 general_span_iterator& operator--() noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001887 {
1888 --m_itr;
1889 return *this;
1890 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001891 general_span_iterator operator--(int)noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001892 {
1893 auto ret = *this;
1894 --(*this);
1895 return ret;
1896 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001897 general_span_iterator operator+(difference_type n) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001898 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001899 general_span_iterator ret{ *this };
Neil MacIntosh68064d62015-11-03 19:17:11 -08001900 return ret += n;
1901 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001902 general_span_iterator& operator+=(difference_type n) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001903 {
1904 m_itr += n;
1905 return *this;
1906 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001907 general_span_iterator operator-(difference_type n) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001908 {
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001909 general_span_iterator ret{ *this };
Neil MacIntosh68064d62015-11-03 19:17:11 -08001910 return ret -= n;
1911 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001912 general_span_iterator& operator-=(difference_type n) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001913 {
1914 return *this += -n;
1915 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001916 difference_type operator-(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001917 {
1918 fail_fast_assert(m_container == rhs.m_container);
1919 return m_itr - rhs.m_itr;
1920 }
1921 value_type operator[](difference_type n) const noexcept
1922 {
1923 return (*m_container)[m_itr[n]];;
1924 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001925 bool operator==(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001926 {
1927 fail_fast_assert(m_container == rhs.m_container);
1928 return m_itr == rhs.m_itr;
1929 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001930 bool operator !=(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001931 {
1932 return !(*this == rhs);
1933 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001934 bool operator<(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001935 {
1936 fail_fast_assert(m_container == rhs.m_container);
1937 return m_itr < rhs.m_itr;
1938 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001939 bool operator<=(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001940 {
1941 return !(rhs < *this);
1942 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001943 bool operator>(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001944 {
1945 return rhs < *this;
1946 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001947 bool operator>=(const general_span_iterator& rhs) const noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001948 {
1949 return !(rhs > *this);
1950 }
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001951 void swap(general_span_iterator& rhs) noexcept
Neil MacIntosh68064d62015-11-03 19:17:11 -08001952 {
1953 std::swap(m_itr, rhs.m_itr);
1954 std::swap(m_container, rhs.m_container);
1955 }
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001956};
1957
1958template <typename ArrayView>
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001959general_span_iterator<ArrayView> operator+(typename general_span_iterator<ArrayView>::difference_type n, const general_span_iterator<ArrayView>& rhs) noexcept
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001960{
Neil MacIntosh68064d62015-11-03 19:17:11 -08001961 return rhs + n;
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001962}
1963
Neil MacIntoshef626fd2015-09-29 16:41:37 -07001964} // namespace gsl
Neil MacIntosha9dcbe02015-08-20 18:09:14 -07001965
Neil MacIntoshd5316802015-09-30 21:54:08 -07001966#ifdef _MSC_VER
1967
Gabriel Dos Reis6554e832015-09-28 05:10:44 -07001968#undef constexpr
1969#pragma pop_macro("constexpr")
Gabriel Dos Reis6554e832015-09-28 05:10:44 -07001970
Neil MacIntoshd5316802015-09-30 21:54:08 -07001971#if _MSC_VER <= 1800
Neil MacIntosh9a297122015-09-14 15:11:07 -07001972#pragma warning(pop)
Neil MacIntoshd5316802015-09-30 21:54:08 -07001973
1974#ifndef GSL_THROWS_FOR_TESTING
1975#pragma undef noexcept
1976#endif // GSL_THROWS_FOR_TESTING
1977
Neil MacIntoshe9a96022015-11-03 18:56:55 -08001978#undef GSL_MSVC_HAS_VARIADIC_CTOR_BUG
1979#pragma pop_macro("GSL_MSVC_HAS_VARIADIC_CTOR_BUG")
1980
1981
Neil MacIntosh9a297122015-09-14 15:11:07 -07001982#endif // _MSC_VER <= 1800
Neil MacIntosh9a297122015-09-14 15:11:07 -07001983
Neil MacIntoshd5316802015-09-30 21:54:08 -07001984#endif // _MSC_VER
1985
1986#if defined(GSL_THROWS_FOR_TESTING)
1987#undef noexcept
1988#endif // GSL_THROWS_FOR_TESTING
1989
Treb Connell51da1362015-09-24 18:08:34 -07001990
Neil MacIntoshb63ec942015-11-04 12:42:27 -08001991#endif // GSL_SPAN_H