aboutsummaryrefslogtreecommitdiffstats
path: root/common/alspan.h
diff options
context:
space:
mode:
Diffstat (limited to 'common/alspan.h')
-rw-r--r--common/alspan.h211
1 files changed, 134 insertions, 77 deletions
diff --git a/common/alspan.h b/common/alspan.h
index cb34d410..1d6cdfe5 100644
--- a/common/alspan.h
+++ b/common/alspan.h
@@ -7,31 +7,26 @@
#include <iterator>
#include <type_traits>
-namespace al {
+#include "almalloc.h"
+#include "altraits.h"
-template<typename T>
-constexpr auto size(T &cont) -> decltype(cont.size())
-{ return cont.size(); }
+namespace al {
template<typename T>
-constexpr auto size(const T &cont) -> decltype(cont.size())
+constexpr auto size(const T &cont) noexcept(noexcept(cont.size())) -> decltype(cont.size())
{ return cont.size(); }
template<typename T, size_t N>
-constexpr size_t size(T (&)[N]) noexcept
+constexpr size_t size(const T (&)[N]) noexcept
{ return N; }
-template<typename T>
-constexpr size_t size(std::initializer_list<T> list) noexcept
-{ return list.size(); }
-
template<typename T>
-constexpr auto data(T &cont) -> decltype(cont.data())
+constexpr auto data(T &cont) noexcept(noexcept(cont.data())) -> decltype(cont.data())
{ return cont.data(); }
template<typename T>
-constexpr auto data(const T &cont) -> decltype(cont.data())
+constexpr auto data(const T &cont) noexcept(noexcept(cont.data())) -> decltype(cont.data())
{ return cont.data(); }
template<typename T, size_t N>
@@ -43,53 +38,52 @@ constexpr const T* data(std::initializer_list<T> list) noexcept
{ return list.begin(); }
-template<typename T, size_t E=static_cast<size_t>(-1)>
+constexpr size_t dynamic_extent{static_cast<size_t>(-1)};
+
+template<typename T, size_t E=dynamic_extent>
class span;
namespace detail_ {
template<typename... Ts>
- struct make_void { using type = void; };
- template<typename... Ts>
- using void_t = typename make_void<Ts...>::type;
+ using void_t = void;
template<typename T>
struct is_span_ : std::false_type { };
template<typename T, size_t E>
struct is_span_<span<T,E>> : std::true_type { };
template<typename T>
- using is_span = is_span_<typename std::remove_cv<T>::type>;
+ constexpr bool is_span_v = is_span_<std::remove_cv_t<T>>::value;
template<typename T>
struct is_std_array_ : std::false_type { };
template<typename T, size_t N>
struct is_std_array_<std::array<T,N>> : std::true_type { };
template<typename T>
- using is_std_array = is_std_array_<typename std::remove_cv<T>::type>;
+ constexpr bool is_std_array_v = is_std_array_<std::remove_cv_t<T>>::value;
template<typename T, typename = void>
- struct has_size_and_data : std::false_type { };
+ constexpr bool has_size_and_data = false;
template<typename T>
- struct has_size_and_data<T,
+ constexpr bool has_size_and_data<T,
void_t<decltype(al::size(std::declval<T>())), decltype(al::data(std::declval<T>()))>>
- : std::true_type { };
+ = true;
- template<typename T>
- using remove_pointer_t = typename std::remove_pointer<T>::type;
+ template<typename T, typename U>
+ constexpr bool is_array_compatible = std::is_convertible<T(*)[],U(*)[]>::value;
+
+ template<typename C, typename T>
+ constexpr bool is_valid_container = !is_span_v<C> && !is_std_array_v<C>
+ && !std::is_array<C>::value && has_size_and_data<C>
+ && is_array_compatible<std::remove_pointer_t<decltype(al::data(std::declval<C&>()))>,T>;
} // namespace detail_
-#define REQUIRES(...) bool rt_=true, typename std::enable_if<rt_ && (__VA_ARGS__),bool>::type = true
-#define IS_VALID_CONTAINER(C) \
- !detail_::is_span<C>::value && !detail_::is_std_array<C>::value && \
- !std::is_array<C>::value && detail_::has_size_and_data<C>::value && \
- std::is_convertible<detail_::remove_pointer_t<decltype(al::data(std::declval<C&>()))>(*)[],element_type(*)[]>::value
+#define REQUIRES(...) std::enable_if_t<(__VA_ARGS__),bool> = true
template<typename T, size_t E>
class span {
- static constexpr size_t dynamic_extent{static_cast<size_t>(-1)};
-
public:
using element_type = T;
- using value_type = typename std::remove_cv<T>::type;
+ using value_type = std::remove_cv_t<T>;
using index_type = size_t;
using difference_type = ptrdiff_t;
@@ -105,23 +99,36 @@ public:
static constexpr size_t extent{E};
- template<REQUIRES(extent==0)>
+ template<bool is0=(extent == 0), REQUIRES(is0)>
constexpr span() noexcept { }
- constexpr span(pointer ptr, index_type /*count*/) : mData{ptr} { }
- constexpr span(pointer first, pointer /*last*/) : mData{first} { }
- constexpr span(element_type (&arr)[E]) noexcept : span{al::data(arr), al::size(arr)} { }
+ template<typename U>
+ constexpr explicit span(U iter, index_type) : mData{to_address(iter)} { }
+ template<typename U, typename V, REQUIRES(!std::is_convertible<V,size_t>::value)>
+ constexpr explicit span(U first, V) : mData{to_address(first)} { }
+
+ constexpr span(type_identity_t<element_type> (&arr)[E]) noexcept
+ : span{al::data(arr), al::size(arr)}
+ { }
constexpr span(std::array<value_type,E> &arr) noexcept : span{al::data(arr), al::size(arr)} { }
- template<REQUIRES(std::is_const<element_type>::value)>
- constexpr span(const std::array<value_type,E> &arr) noexcept : span{al::data(arr), al::size(arr)} { }
- template<typename U, REQUIRES(IS_VALID_CONTAINER(U))>
- constexpr span(U &cont) : span{al::data(cont), al::size(cont)} { }
- template<typename U, REQUIRES(IS_VALID_CONTAINER(const U))>
- constexpr span(const U &cont) : span{al::data(cont), al::size(cont)} { }
- template<typename U, REQUIRES(!std::is_same<element_type,U>::value && std::is_convertible<U(*)[],element_type(*)[]>::value)>
- constexpr span(const span<U,E> &span_) noexcept : span{al::data(span_), al::size(span_)} { }
+ template<typename U=T, REQUIRES(std::is_const<U>::value)>
+ constexpr span(const std::array<value_type,E> &arr) noexcept
+ : span{al::data(arr), al::size(arr)}
+ { }
+
+ template<typename U, REQUIRES(detail_::is_valid_container<U, element_type>)>
+ constexpr explicit span(U&& cont) : span{al::data(cont), al::size(cont)} { }
+
+ template<typename U, index_type N, REQUIRES(!std::is_same<element_type,U>::value
+ && detail_::is_array_compatible<U,element_type> && N == dynamic_extent)>
+ constexpr explicit span(const span<U,N> &span_) noexcept
+ : span{al::data(span_), al::size(span_)}
+ { }
+ template<typename U, index_type N, REQUIRES(!std::is_same<element_type,U>::value
+ && detail_::is_array_compatible<U,element_type> && N == extent)>
+ constexpr span(const span<U,N> &span_) noexcept : span{al::data(span_), al::size(span_)} { }
constexpr span(const span&) noexcept = default;
- span& operator=(const span &rhs) noexcept = default;
+ constexpr span& operator=(const span &rhs) noexcept = default;
constexpr reference front() const { return *mData; }
constexpr reference back() const { return *(mData+E-1); }
@@ -137,10 +144,12 @@ public:
constexpr const_iterator cbegin() const noexcept { return mData; }
constexpr const_iterator cend() const noexcept { return mData+E; }
- constexpr reverse_iterator rbegin() const noexcept { return end(); }
- constexpr reverse_iterator rend() const noexcept { return begin(); }
- constexpr const_reverse_iterator crbegin() const noexcept { return cend(); }
- constexpr const_reverse_iterator crend() const noexcept { return cbegin(); }
+ constexpr reverse_iterator rbegin() const noexcept { return reverse_iterator{end()}; }
+ constexpr reverse_iterator rend() const noexcept { return reverse_iterator{begin()}; }
+ constexpr const_reverse_iterator crbegin() const noexcept
+ { return const_reverse_iterator{cend()}; }
+ constexpr const_reverse_iterator crend() const noexcept
+ { return const_reverse_iterator{cbegin()}; }
template<size_t C>
constexpr span<element_type,C> first() const
@@ -156,12 +165,19 @@ public:
return span<element_type,C>{mData+(E-C), C};
}
- template<size_t O, size_t C, size_t RealC=((C==dynamic_extent) ? E-O : C)>
- constexpr span<element_type,RealC> subspan() const
+ template<size_t O, size_t C>
+ constexpr auto subspan() const -> std::enable_if_t<C!=dynamic_extent,span<element_type,C>>
{
static_assert(E >= O, "Offset exceeds extent");
- static_assert(E-O >= RealC, "New size exceeds original capacity");
- return span<element_type,RealC>{mData+O, RealC};
+ static_assert(E-O >= C, "New size exceeds original capacity");
+ return span<element_type,C>{mData+O, C};
+ }
+
+ template<size_t O, size_t C=dynamic_extent>
+ constexpr auto subspan() const -> std::enable_if_t<C==dynamic_extent,span<element_type,E-O>>
+ {
+ static_assert(E >= O, "Offset exceeds extent");
+ return span<element_type,E-O>{mData+O, E-O};
}
/* NOTE: Can't declare objects of a specialized template class prior to
@@ -170,19 +186,18 @@ public:
*/
constexpr span<element_type,dynamic_extent> first(size_t count) const;
constexpr span<element_type,dynamic_extent> last(size_t count) const;
- constexpr span<element_type,dynamic_extent> subspan(size_t offset, size_t count=dynamic_extent) const;
+ constexpr span<element_type,dynamic_extent> subspan(size_t offset,
+ size_t count=dynamic_extent) const;
private:
pointer mData{nullptr};
};
template<typename T>
-class span<T,static_cast<size_t>(-1)> {
- static constexpr size_t dynamic_extent{static_cast<size_t>(-1)};
-
+class span<T,dynamic_extent> {
public:
using element_type = T;
- using value_type = typename std::remove_cv<T>::type;
+ using value_type = std::remove_cv_t<T>;
using index_type = size_t;
using difference_type = ptrdiff_t;
@@ -196,26 +211,37 @@ public:
using reverse_iterator = std::reverse_iterator<iterator>;
using const_reverse_iterator = std::reverse_iterator<const_iterator>;
- static constexpr size_t extent{static_cast<size_t>(-1)};
+ static constexpr size_t extent{dynamic_extent};
constexpr span() noexcept = default;
- constexpr span(pointer ptr, index_type count) : mData{ptr}, mDataEnd{ptr+count} { }
- constexpr span(pointer first, pointer last) : mData{first}, mDataEnd{last} { }
+ template<typename U>
+ constexpr span(U iter, index_type count)
+ : mData{to_address(iter)}, mDataEnd{to_address(iter)+count}
+ { }
+ template<typename U, typename V, REQUIRES(!std::is_convertible<V,size_t>::value)>
+ constexpr span(U first, V last) : span{to_address(first), static_cast<size_t>(last-first)}
+ { }
+
template<size_t N>
- constexpr span(element_type (&arr)[N]) noexcept : span{al::data(arr), al::size(arr)} { }
+ constexpr span(type_identity_t<element_type> (&arr)[N]) noexcept
+ : span{al::data(arr), al::size(arr)}
+ { }
template<size_t N>
constexpr span(std::array<value_type,N> &arr) noexcept : span{al::data(arr), al::size(arr)} { }
- template<size_t N, REQUIRES(std::is_const<element_type>::value)>
- constexpr span(const std::array<value_type,N> &arr) noexcept : span{al::data(arr), al::size(arr)} { }
- template<typename U, REQUIRES(IS_VALID_CONTAINER(U))>
- constexpr span(U &cont) : span{al::data(cont), al::size(cont)} { }
- template<typename U, REQUIRES(IS_VALID_CONTAINER(const U))>
- constexpr span(const U &cont) : span{al::data(cont), al::size(cont)} { }
- template<typename U, size_t N, REQUIRES((!std::is_same<element_type,U>::value || extent != N) && std::is_convertible<U(*)[],element_type(*)[]>::value)>
+ template<size_t N, typename U=T, REQUIRES(std::is_const<U>::value)>
+ constexpr span(const std::array<value_type,N> &arr) noexcept
+ : span{al::data(arr), al::size(arr)}
+ { }
+
+ template<typename U, REQUIRES(detail_::is_valid_container<U, element_type>)>
+ constexpr span(U&& cont) : span{al::data(cont), al::size(cont)} { }
+
+ template<typename U, size_t N, REQUIRES((!std::is_same<element_type,U>::value || extent != N)
+ && detail_::is_array_compatible<U,element_type>)>
constexpr span(const span<U,N> &span_) noexcept : span{al::data(span_), al::size(span_)} { }
constexpr span(const span&) noexcept = default;
- span& operator=(const span &rhs) noexcept = default;
+ constexpr span& operator=(const span &rhs) noexcept = default;
constexpr reference front() const { return *mData; }
constexpr reference back() const { return *(mDataEnd-1); }
@@ -232,10 +258,12 @@ public:
constexpr const_iterator cbegin() const noexcept { return mData; }
constexpr const_iterator cend() const noexcept { return mDataEnd; }
- constexpr reverse_iterator rbegin() const noexcept { return end(); }
- constexpr reverse_iterator rend() const noexcept { return begin(); }
- constexpr const_reverse_iterator crbegin() const noexcept { return cend(); }
- constexpr const_reverse_iterator crend() const noexcept { return cbegin(); }
+ constexpr reverse_iterator rbegin() const noexcept { return reverse_iterator{end()}; }
+ constexpr reverse_iterator rend() const noexcept { return reverse_iterator{begin()}; }
+ constexpr const_reverse_iterator crbegin() const noexcept
+ { return const_reverse_iterator{cend()}; }
+ constexpr const_reverse_iterator crend() const noexcept
+ { return const_reverse_iterator{cbegin()}; }
template<size_t C>
constexpr span<element_type,C> first() const
@@ -251,9 +279,13 @@ public:
constexpr span last(size_t count) const
{ return (count >= size()) ? *this : span{mDataEnd-count, mDataEnd}; }
+ template<size_t O, size_t C>
+ constexpr auto subspan() const -> std::enable_if_t<C!=dynamic_extent,span<element_type,C>>
+ { return span<element_type,C>{mData+O, C}; }
+
template<size_t O, size_t C=dynamic_extent>
- constexpr span<element_type,C> subspan() const
- { return span<element_type,C>{mData+O, (C!=dynamic_extent) ? mData+C : mDataEnd}; }
+ constexpr auto subspan() const -> std::enable_if_t<C==dynamic_extent,span<element_type,C>>
+ { return span<element_type,C>{mData+O, mDataEnd}; }
constexpr span subspan(size_t offset, size_t count=dynamic_extent) const
{
@@ -282,14 +314,39 @@ constexpr inline auto span<T,E>::last(size_t count) const -> span<element_type,d
}
template<typename T, size_t E>
-constexpr inline auto span<T,E>::subspan(size_t offset, size_t count) const -> span<element_type,dynamic_extent>
+constexpr inline auto span<T,E>::subspan(size_t offset, size_t count) const
+ -> span<element_type,dynamic_extent>
{
return (offset > size()) ? span<element_type>{} :
(count >= size()-offset) ? span<element_type>{mData+offset, mData+extent} :
span<element_type>{mData+offset, mData+offset+count};
}
-#undef IS_VALID_CONTAINER
+/* Helpers to deal with the lack of user-defined deduction guides (C++17). */
+template<typename T, typename U>
+constexpr auto as_span(T ptr, U count_or_end)
+{
+ using value_type = typename std::pointer_traits<T>::element_type;
+ return span<value_type>{ptr, count_or_end};
+}
+template<typename T, size_t N>
+constexpr auto as_span(T (&arr)[N]) noexcept { return span<T,N>{al::data(arr), al::size(arr)}; }
+template<typename T, size_t N>
+constexpr auto as_span(std::array<T,N> &arr) noexcept
+{ return span<T,N>{al::data(arr), al::size(arr)}; }
+template<typename T, size_t N>
+constexpr auto as_span(const std::array<T,N> &arr) noexcept
+{ return span<std::add_const_t<T>,N>{al::data(arr), al::size(arr)}; }
+template<typename U, REQUIRES(!detail_::is_span_v<U> && !detail_::is_std_array_v<U>
+ && !std::is_array<U>::value && detail_::has_size_and_data<U>)>
+constexpr auto as_span(U&& cont)
+{
+ using value_type = std::remove_pointer_t<decltype(al::data(std::declval<U&>()))>;
+ return span<value_type>{al::data(cont), al::size(cont)};
+}
+template<typename T, size_t N>
+constexpr auto as_span(span<T,N> span_) noexcept { return span_; }
+
#undef REQUIRES
} // namespace al