blob: 0dae18672aff3986909cadd195b62530b24c99d2 [file] [log] [blame]
// Components for manipulating non-owning sequences of objects -*- C++ -*-
// Copyright (C) 2019-2020 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library. This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 3, or (at your option)
// any later version.
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// Under Section 7 of GPL version 3, you are granted additional
// permissions described in the GCC Runtime Library Exception, version
// 3.1, as published by the Free Software Foundation.
// You should have received a copy of the GNU General Public License and
// a copy of the GCC Runtime Library Exception along with this program;
// see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
// <http://www.gnu.org/licenses/>.
/** @file span
* This is a Standard C++ Library header.
*/
//
// P0122 span library
// Contributed by ThePhD
//
#ifndef _GLIBCXX_SPAN
#define _GLIBCXX_SPAN 1
#pragma GCC system_header
#if __cplusplus > 201703L
#include <type_traits>
#include <array>
#include <bits/stl_iterator.h>
#include <bits/range_access.h>
#if __cpp_lib_concepts
namespace std _GLIBCXX_VISIBILITY(default)
{
_GLIBCXX_BEGIN_NAMESPACE_VERSION
#define __cpp_lib_span 201902L
inline constexpr size_t dynamic_extent = static_cast<size_t>(-1);
template<typename _Type, size_t _Extent>
class span;
namespace __detail
{
template<typename _Tp>
struct __is_std_span : false_type { };
template<typename _Tp, size_t _Num>
struct __is_std_span<span<_Tp, _Num>> : true_type { };
template<typename _Tp>
struct __is_std_array : false_type { };
template<typename _Tp, size_t _Num>
struct __is_std_array<_GLIBCXX_STD_C::array<_Tp, _Num>> : true_type { };
#ifdef _GLIBCXX_DEBUG
template<typename _Tp, size_t _Num>
struct __is_std_array<__debug::array<_Tp, _Num>> : true_type { };
#endif
template<size_t _Extent>
class __extent_storage
{
public:
constexpr
__extent_storage(size_t) noexcept
{ }
static constexpr size_t
_M_extent() noexcept
{ return _Extent; }
};
template<>
class __extent_storage<dynamic_extent>
{
public:
constexpr
__extent_storage(size_t __extent) noexcept
: _M_extent_value(__extent)
{ }
constexpr size_t
_M_extent() const noexcept
{ return this->_M_extent_value; }
private:
size_t _M_extent_value;
};
} // namespace __detail
template<typename _Type, size_t _Extent = dynamic_extent>
class span
{
template<size_t _Offset, size_t _Count>
static constexpr size_t
_S_subspan_extent()
{
if constexpr (_Count != dynamic_extent)
return _Count;
else if constexpr (extent != dynamic_extent)
return _Extent - _Offset;
else
return dynamic_extent;
}
// _GLIBCXX_RESOLVE_LIB_DEFECTS
// 3255. span's array constructor is too strict
template<typename _Tp, size_t _ArrayExtent>
using __is_compatible_array = __and_<
bool_constant<(_Extent == dynamic_extent || _ArrayExtent == _Extent)>,
__is_array_convertible<_Type, _Tp>>;
template<typename _Iter, typename _Ref = iter_reference_t<_Iter>>
using __is_compatible_iterator = __and_<
bool_constant<contiguous_iterator<_Iter>>,
is_lvalue_reference<iter_reference_t<_Iter>>,
is_same<iter_value_t<_Iter>, remove_cvref_t<_Ref>>,
__is_array_convertible<_Type, remove_reference_t<_Ref>>>;
template<typename _Range>
using __is_compatible_range
= __is_compatible_iterator<ranges::iterator_t<_Range>>;
public:
// member types
using value_type = remove_cv_t<_Type>;
using element_type = _Type;
using size_type = size_t;
using reference = element_type&;
using const_reference = const element_type&;
using pointer = _Type*;
using const_pointer = const _Type*;
using iterator
= __gnu_cxx::__normal_iterator<pointer, span>;
using const_iterator
= __gnu_cxx::__normal_iterator<const_pointer, span>;
using reverse_iterator = std::reverse_iterator<iterator>;
using const_reverse_iterator = std::reverse_iterator<const_iterator>;
using difference_type = ptrdiff_t;
// member constants
static inline constexpr size_t extent = _Extent;
// constructors
constexpr
span() noexcept
requires ((_Extent + 1u) <= 1u)
: _M_extent(0), _M_ptr(nullptr)
{ }
constexpr
span(const span&) noexcept = default;
template<typename _Tp, size_t _ArrayExtent>
requires (__is_compatible_array<_Tp, _ArrayExtent>::value)
constexpr
span(_Tp (&__arr)[_ArrayExtent]) noexcept
: span(static_cast<pointer>(__arr), _ArrayExtent)
{ }
template<typename _Tp, size_t _ArrayExtent>
requires (__is_compatible_array<_Tp, _ArrayExtent>::value)
constexpr
span(array<_Tp, _ArrayExtent>& __arr) noexcept
: span(static_cast<pointer>(__arr.data()), _ArrayExtent)
{ }
template<typename _Tp, size_t _ArrayExtent>
requires (__is_compatible_array<const _Tp, _ArrayExtent>::value)
constexpr
span(const array<_Tp, _ArrayExtent>& __arr) noexcept
: span(static_cast<pointer>(__arr.data()), _ArrayExtent)
{ }
public:
template<ranges::contiguous_range _Range>
requires (_Extent == dynamic_extent)
&& (!__detail::__is_std_span<remove_cvref_t<_Range>>::value)
&& (!__detail::__is_std_array<remove_cvref_t<_Range>>::value)
&& (!is_array_v<remove_reference_t<_Range>>)
&& (__is_compatible_range<_Range>::value)
constexpr
span(_Range&& __range)
noexcept(noexcept(ranges::data(__range))
&& noexcept(ranges::size(__range)))
: span(ranges::data(__range), ranges::size(__range))
{ }
template<contiguous_iterator _ContiguousIterator,
sized_sentinel_for<_ContiguousIterator> _Sentinel>
requires (__is_compatible_iterator<_ContiguousIterator>::value)
&& (!is_convertible_v<_Sentinel, size_type>)
constexpr
span(_ContiguousIterator __first, _Sentinel __last)
noexcept(noexcept(__last - __first))
: _M_extent(static_cast<size_type>(__last - __first)),
_M_ptr(std::to_address(__first))
{
if (_Extent != dynamic_extent)
__glibcxx_assert((__last - __first) == _Extent);
}
template<contiguous_iterator _ContiguousIterator>
requires (__is_compatible_iterator<_ContiguousIterator>::value)
constexpr
span(_ContiguousIterator __first, size_type __count)
noexcept
: _M_extent(__count), _M_ptr(std::to_address(__first))
{ __glibcxx_assert(_Extent == dynamic_extent || __count == _Extent); }
template<typename _OType, size_t _OExtent>
requires (_Extent == dynamic_extent || _Extent == _OExtent)
&& (__is_array_convertible<_Type, _OType>::value)
constexpr
span(const span<_OType, _OExtent>& __s) noexcept
: _M_extent(__s.size()), _M_ptr(__s.data())
{ }
// assignment
constexpr span&
operator=(const span&) noexcept = default;
// observers
constexpr size_type
size() const noexcept
{ return this->_M_extent._M_extent(); }
constexpr size_type
size_bytes() const noexcept
{ return this->_M_extent._M_extent() * sizeof(element_type); }
[[nodiscard]] constexpr bool
empty() const noexcept
{ return size() == 0; }
// element access
constexpr reference
front() const noexcept
{
static_assert(extent != 0);
__glibcxx_assert(!empty());
return *this->_M_ptr;
}
constexpr reference
back() const noexcept
{
static_assert(extent != 0);
__glibcxx_assert(!empty());
return *(this->_M_ptr + (size() - 1));
}
constexpr reference
operator[](size_type __idx) const noexcept
{
static_assert(extent != 0);
__glibcxx_assert(__idx < size());
return *(this->_M_ptr + __idx);
}
constexpr pointer
data() const noexcept
{ return this->_M_ptr; }
// iterator support
constexpr iterator
begin() const noexcept
{ return iterator(this->_M_ptr); }
constexpr const_iterator
cbegin() const noexcept
{ return const_iterator(this->_M_ptr); }
constexpr iterator
end() const noexcept
{ return iterator(this->_M_ptr + this->size()); }
constexpr const_iterator
cend() const noexcept
{ return const_iterator(this->_M_ptr + this->size()); }
constexpr reverse_iterator
rbegin() const noexcept
{ return reverse_iterator(this->end()); }
constexpr const_reverse_iterator
crbegin() const noexcept
{ return const_reverse_iterator(this->cend()); }
constexpr reverse_iterator
rend() const noexcept
{ return reverse_iterator(this->begin()); }
constexpr const_reverse_iterator
crend() const noexcept
{ return const_reverse_iterator(this->cbegin()); }
// subviews
template<size_t _Count>
constexpr span<element_type, _Count>
first() const noexcept
{
if constexpr (_Extent == dynamic_extent)
__glibcxx_assert(_Count <= size());
else
static_assert(_Count <= extent);
return { this->data(), _Count };
}
constexpr span<element_type, dynamic_extent>
first(size_type __count) const noexcept
{
__glibcxx_assert(__count <= size());
return { this->data(), __count };
}
template<size_t _Count>
constexpr span<element_type, _Count>
last() const noexcept
{
if constexpr (_Extent == dynamic_extent)
__glibcxx_assert(_Count <= size());
else
static_assert(_Count <= extent);
return { this->data() + (this->size() - _Count), _Count };
}
constexpr span<element_type, dynamic_extent>
last(size_type __count) const noexcept
{
__glibcxx_assert(__count <= size());
return { this->data() + (this->size() - __count), __count };
}
template<size_t _Offset, size_t _Count = dynamic_extent>
constexpr auto
subspan() const noexcept
-> span<element_type, _S_subspan_extent<_Offset, _Count>()>
{
if constexpr (_Extent == dynamic_extent)
__glibcxx_assert(_Offset <= size());
else
static_assert(_Offset <= extent);
if constexpr (_Count == dynamic_extent)
return { this->data() + _Offset, this->size() - _Offset };
else
{
if constexpr (_Extent == dynamic_extent)
{
__glibcxx_assert(_Count <= size());
__glibcxx_assert(_Count <= (size() - _Offset));
}
else
{
static_assert(_Count <= extent);
static_assert(_Count <= (extent - _Offset));
}
return { this->data() + _Offset, _Count };
}
}
constexpr span<element_type, dynamic_extent>
subspan(size_type __offset, size_type __count = dynamic_extent) const
noexcept
{
__glibcxx_assert(__offset <= size());
if (__count == dynamic_extent)
__count = this->size() - __offset;
else
{
__glibcxx_assert(__count <= size());
__glibcxx_assert(__offset + __count <= size());
}
return {this->data() + __offset, __count};
}
private:
[[no_unique_address]] __detail::__extent_storage<extent> _M_extent;
pointer _M_ptr;
};
// deduction guides
template<typename _Type, size_t _ArrayExtent>
span(_Type(&)[_ArrayExtent]) -> span<_Type, _ArrayExtent>;
template<typename _Type, size_t _ArrayExtent>
span(array<_Type, _ArrayExtent>&) -> span<_Type, _ArrayExtent>;
template<typename _Type, size_t _ArrayExtent>
span(const array<_Type, _ArrayExtent>&)
-> span<const _Type, _ArrayExtent>;
template<contiguous_iterator _Iter, typename _Sentinel>
span(_Iter, _Sentinel)
-> span<remove_reference_t<ranges::range_reference_t<_Iter>>>;
template<typename _Range>
span(_Range &&)
-> span<remove_reference_t<ranges::range_reference_t<_Range&>>>;
template<typename _Type, size_t _Extent>
inline
span<const byte, _Extent == dynamic_extent
? dynamic_extent : _Extent * sizeof(_Type)>
as_bytes(span<_Type, _Extent> __sp) noexcept
{
return {reinterpret_cast<const byte*>(__sp.data()), __sp.size_bytes()};
}
template<typename _Type, size_t _Extent>
inline
span<byte, _Extent == dynamic_extent
? dynamic_extent : _Extent * sizeof(_Type)>
as_writable_bytes(span<_Type, _Extent> __sp) noexcept
{
return {reinterpret_cast<byte*>(__sp.data()), __sp.size_bytes()};
}
// tuple helpers
template<size_t _Index, typename _Type, size_t _Extent>
constexpr _Type&
get(span<_Type, _Extent> __sp) noexcept
{
static_assert(_Extent != dynamic_extent && _Index < _Extent,
"get<I> can only be used with a span of non-dynamic (fixed) extent");
return __sp[_Index];
}
template<typename _Tp> struct tuple_size;
template<size_t __i, typename _Tp> struct tuple_element;
template<typename _Type, size_t _Extent>
struct tuple_size<span<_Type, _Extent>>
: public integral_constant<size_t, _Extent>
{
static_assert(_Extent != dynamic_extent, "tuple_size can only "
"be used with a span of non-dynamic (fixed) extent");
};
template<size_t _Index, typename _Type, size_t _Extent>
struct tuple_element<_Index, span<_Type, _Extent>>
{
static_assert(_Extent != dynamic_extent, "tuple_element can only "
"be used with a span of non-dynamic (fixed) extent");
static_assert(_Index < _Extent, "Index is less than Extent");
using type = _Type;
};
namespace ranges
{
template<typename> extern inline const bool enable_safe_range;
// Opt-in to safe_range concept
template<typename _ElementType, size_t _Extent>
inline constexpr bool
enable_safe_range<span<_ElementType, _Extent>> = true;
}
_GLIBCXX_END_NAMESPACE_VERSION
} // namespace std
#endif // concepts
#endif // C++20
#endif // _GLIBCXX_SPAN