Change iterator from Span* & ptrdiff_t to pointer, pointer, pointer

This commit is contained in:
Jordan Maples [MSFT] 2020-02-03 16:26:36 -08:00
parent d9d6ea8196
commit 1815791af8

View File

@ -122,140 +122,165 @@ namespace details
{ {
}; };
template <class Span, bool IsConst> template <class type>
class span_iterator class span_iterator
{ {
using element_type_ = typename Span::element_type;
public: public:
#ifdef _MSC_VER
// Tell Microsoft standard library that span_iterators are checked.
using _Unchecked_type = typename Span::pointer;
#endif
using iterator_category = std::random_access_iterator_tag; using iterator_category = std::random_access_iterator_tag;
using value_type = std::remove_cv_t<element_type_>; using value_type = std::remove_cv_t<type>;
using difference_type = typename Span::index_type; using difference_type = ptrdiff_t;
using pointer = std::add_pointer_t<type>;
using reference = std::add_lvalue_reference_t<type>;
using reference = std::conditional_t<IsConst, const element_type_, element_type_>&; #ifdef _MSC_VER
using pointer = std::add_pointer_t<reference>; using _Unchecked_type = typename pointer;
#endif
span_iterator() = default; constexpr operator span_iterator<const type>() const noexcept
constexpr span_iterator(const Span* span, difference_type idx) noexcept
: span_(span), index_(idx)
{}
friend span_iterator<Span, true>;
template <bool B, std::enable_if_t<!B && IsConst>* = nullptr>
constexpr span_iterator(const span_iterator<Span, B>& other) noexcept
: span_iterator(other.span_, other.index_)
{}
GSL_SUPPRESS(bounds.1) // NO-FORMAT: attribute
constexpr reference operator*() const
{ {
Expects(static_cast<size_t>(index_) != span_->size()); return {begin_, end_, current_};
return *(span_->data() + index_);
} }
constexpr pointer operator->() const constexpr reference operator*() const noexcept { return *operator->(); }
{
Expects(index_ != span_->size());
return span_->data() + index_;
}
constexpr pointer operator->() const noexcept
{
Expects(begin_ && current_ && end_);
Expects(current_ < end_);
return current_;
}
constexpr span_iterator& operator++() noexcept constexpr span_iterator& operator++() noexcept
{ {
Expects(0 <= index_ && static_cast<size_t>(index_) != span_->size()); Expects(begin_ && current_ && end_);
++index_; Expects(current_ < end_);
++current_;
return *this; return *this;
} }
constexpr span_iterator operator++(int) noexcept constexpr span_iterator operator++(int) noexcept
{ {
auto ret = *this; auto ret{*this};
++(*this); ++*this;
return ret; return ret;
} }
constexpr span_iterator& operator--() noexcept constexpr span_iterator& operator--() noexcept
{ {
Expects(index_ != 0 && static_cast<size_t>(index_) <= span_->size()); Expects(begin_ && current_ && end_);
--index_; Expects(current_ > begin_);
--current_;
return *this; return *this;
} }
constexpr span_iterator operator--(int) noexcept constexpr span_iterator operator--(int) noexcept
{ {
auto ret = *this; auto ret{*this};
--(*this); --*this;
return ret; return ret;
} }
constexpr span_iterator operator+(difference_type n) const noexcept constexpr span_iterator& operator+=(const difference_type n) noexcept
{ {
auto ret = *this; Expects(begin_ && current_ && end_);
if (n > 0) Expects(end_ - current_ >= n);
if (n < 0) Expects(end_ - current_ >= -n);
current_ += n;
return *this;
}
constexpr span_iterator operator+(const difference_type n) const noexcept
{
auto ret{*this};
return ret += n; return ret += n;
} }
friend constexpr span_iterator operator+(difference_type n, friend constexpr span_iterator operator+(const difference_type n,
span_iterator const& rhs) noexcept span_iterator const& rhs) noexcept
{ {
return rhs + n; return rhs + n;
} }
constexpr span_iterator& operator+=(difference_type n) noexcept constexpr span_iterator& operator-=(const difference_type n) noexcept
{ {
Expects((index_ + n) >= 0 && (index_ + n) <= span_->size()); Expects(begin_ && end_ && current_);
index_ += n; if (n > 0) Expects(end_ - current_ >= n);
if (n < 0) Expects(end_ - current_ >= -n);
current_ -= n;
return *this; return *this;
} }
constexpr span_iterator operator-(difference_type n) const noexcept constexpr span_iterator operator-(const difference_type n) const noexcept
{ {
auto ret = *this; auto ret{*this};
return ret -= n; return ret -= n;
} }
constexpr span_iterator& operator-=(difference_type n) noexcept { return *this += -n; } friend constexpr span_iterator operator-(const difference_type n,
span_iterator const& rhs) noexcept
constexpr difference_type operator-(span_iterator rhs) const noexcept
{ {
Expects(span_ == rhs.span_); return rhs - n;
return index_ - rhs.index_;
} }
constexpr reference operator[](difference_type n) const noexcept { return *(*this + n); } template <
class type2,
constexpr friend bool operator==(span_iterator lhs, span_iterator rhs) noexcept std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr difference_type operator-(const span_iterator<type2>& rhs) const noexcept
{ {
return lhs.span_ == rhs.span_ && lhs.index_ == rhs.index_; Expects(begin_ == rhs.begin_);
return current_ - rhs.current_;
} }
constexpr friend bool operator!=(span_iterator lhs, span_iterator rhs) noexcept constexpr reference operator[](const difference_type n) const noexcept
{ {
return !(lhs == rhs); return *(*this + n);
} }
constexpr friend bool operator<(span_iterator lhs, span_iterator rhs) noexcept template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator==(const span_iterator<type2>& rhs) const noexcept
{ {
return lhs.index_ < rhs.index_; return begin_ == rhs.begin_ && current_ == rhs.current_;
} }
constexpr friend bool operator<=(span_iterator lhs, span_iterator rhs) noexcept template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator!=(const span_iterator<type2>& rhs) const noexcept
{ {
return !(rhs < lhs); return !(*this == rhs);
} }
constexpr friend bool operator>(span_iterator lhs, span_iterator rhs) noexcept template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator<(const span_iterator<type2>& rhs) const noexcept
{ {
return rhs < lhs; Expects(begin_ == rhs.begin_);
return current_ < rhs.current_;
} }
constexpr friend bool operator>=(span_iterator lhs, span_iterator rhs) noexcept template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator>(const span_iterator<type2>& rhs) const noexcept
{ {
return !(rhs > lhs); return !(*this < rhs);
}
template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator<=(const span_iterator<type2>& rhs) const noexcept
{
return *this < rhs || *this == rhs;
}
template <
class type2,
std::enable_if_t<std::is_same<std::remove_cv_t<type2>, value_type>::value, int> = 0>
constexpr bool operator>=(const span_iterator<type2>& rhs) const noexcept
{
return *this > rhs || *this == rhs;
} }
#ifdef _MSC_VER #ifdef _MSC_VER
@ -264,14 +289,14 @@ namespace details
// algorithm calls // algorithm calls
friend constexpr void _Verify_range(span_iterator lhs, span_iterator rhs) noexcept friend constexpr void _Verify_range(span_iterator lhs, span_iterator rhs) noexcept
{ // test that [lhs, rhs) forms a valid range inside an STL algorithm { // test that [lhs, rhs) forms a valid range inside an STL algorithm
Expects(lhs.span_ == rhs.span_ // range spans have to match Expects(lhs.begin_ == rhs.begin_ // range spans have to match
&& lhs.index_ <= rhs.index_); // range must not be transposed && lhs.end_ <= rhs.end_); // range must not be transposed
} }
constexpr void _Verify_offset(const difference_type n) const noexcept constexpr void _Verify_offset(const difference_type n) const noexcept
{ // test that the iterator *this + n is a valid range in an STL { // test that the iterator *this + n is a valid range in an STL
// algorithm call // algorithm call
Expects((index_ + n) >= 0 && (index_ + n) <= span_->size()); Expects((current_ + n) >= begin_ && (current_ + n) <= end_);
} }
GSL_SUPPRESS(bounds.1) // NO-FORMAT: attribute GSL_SUPPRESS(bounds.1) // NO-FORMAT: attribute
@ -279,7 +304,7 @@ namespace details
{ // after seeking *this to a high water mark, or using one of the { // after seeking *this to a high water mark, or using one of the
// _Verify_xxx functions above, unwrap this span_iterator to a raw // _Verify_xxx functions above, unwrap this span_iterator to a raw
// pointer // pointer
return span_->data() + index_; return current_;
} }
// Tell the STL that span_iterator should not be unwrapped if it can't // Tell the STL that span_iterator should not be unwrapped if it can't
@ -293,13 +318,13 @@ namespace details
constexpr void _Seek_to(const pointer p) noexcept constexpr void _Seek_to(const pointer p) noexcept
{ // adjust the position of *this to previously verified location p { // adjust the position of *this to previously verified location p
// after _Unwrapped // after _Unwrapped
index_ = p - span_->data(); current_ = p;
} }
#endif #endif
protected: pointer begin_ = nullptr;
const Span* span_ = nullptr; pointer end_ = nullptr;
std::ptrdiff_t index_ = 0; pointer current_ = nullptr;
}; };
template <std::size_t Ext> template <std::size_t Ext>
@ -368,8 +393,8 @@ public:
using reference = element_type&; using reference = element_type&;
using difference_type = std::ptrdiff_t; using difference_type = std::ptrdiff_t;
using iterator = details::span_iterator<span<ElementType, Extent>, false>; using iterator = details::span_iterator<ElementType>;
using const_iterator = details::span_iterator<span<ElementType, Extent>, true>; using const_iterator = details::span_iterator<const ElementType>;
using reverse_iterator = std::reverse_iterator<iterator>; using reverse_iterator = std::reverse_iterator<iterator>;
using const_reverse_iterator = std::reverse_iterator<const_iterator>; using const_reverse_iterator = std::reverse_iterator<const_iterator>;
@ -540,11 +565,11 @@ public:
constexpr pointer data() const noexcept { return storage_.data(); } constexpr pointer data() const noexcept { return storage_.data(); }
// [span.iter], span iterator support // [span.iter], span iterator support
constexpr iterator begin() const noexcept { return {this, 0}; } constexpr iterator begin() const noexcept { return {data(), data() + size(), data()}; }
constexpr iterator end() const noexcept { return {this, size()}; } constexpr iterator end() const noexcept { return {data(), data() + size(), data() + size()}; }
constexpr const_iterator cbegin() const noexcept { return {this, 0}; } constexpr const_iterator cbegin() const noexcept { return {data(), data() + size(), data()}; }
constexpr const_iterator cend() const noexcept { return {this, size()}; } constexpr const_iterator cend() const noexcept { return {data(), data() + size(), data() + size()}; }
constexpr reverse_iterator rbegin() const noexcept { return reverse_iterator{end()}; } constexpr reverse_iterator rbegin() const noexcept { return reverse_iterator{end()}; }
constexpr reverse_iterator rend() const noexcept { return reverse_iterator{begin()}; } constexpr reverse_iterator rend() const noexcept { return reverse_iterator{begin()}; }
@ -653,7 +678,7 @@ private:
span<element_type, dynamic_extent> make_subspan(index_type offset, index_type count, span<element_type, dynamic_extent> make_subspan(index_type offset, index_type count,
subspan_selector<dynamic_extent>) const subspan_selector<dynamic_extent>) const
{ {
Expects(size() >= offset && size() != dynamic_extent); Expects(size() >= offset);
if (count == dynamic_extent) { return {KnownNotNull{data() + offset}, size() - offset}; } if (count == dynamic_extent) { return {KnownNotNull{data() + offset}, size() - offset}; }