alspan.h 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463
  1. #ifndef AL_SPAN_H
  2. #define AL_SPAN_H
  3. #include <cassert>
  4. #include <cstddef>
  5. #include <iterator>
  6. #include <stdexcept>
  7. #include <tuple>
  8. #include <type_traits>
  9. #include <utility>
  10. #include "alassert.h"
  11. #include "almalloc.h"
  12. #include "altraits.h"
  13. namespace al {
  14. /* This is here primarily to help ensure proper behavior for span's iterators,
  15. * being an actual object with member functions instead of a raw pointer (which
  16. * has requirements like + and - working with ptrdiff_t). This also helps
  17. * silence clang-tidy's pointer arithmetic warnings for span and FlexArray
  18. * iterators. It otherwise behaves like a plain pointer and should optimize
  19. * accordingly.
  20. *
  21. * Shouldn't be needed once we use std::span in C++20.
  22. */
  23. template<typename T>
  24. class ptr_wrapper {
  25. static_assert(std::is_pointer_v<T>);
  26. T mPointer{};
  27. public:
  28. using value_type = std::remove_pointer_t<T>;
  29. using size_type = std::size_t;
  30. using difference_type = std::ptrdiff_t;
  31. using pointer = value_type*;
  32. using reference = value_type&;
  33. using iterator_category = std::random_access_iterator_tag;
  34. explicit constexpr ptr_wrapper(T ptr) : mPointer{ptr} { }
  35. /* NOLINTBEGIN(cppcoreguidelines-pro-bounds-pointer-arithmetic) */
  36. constexpr auto operator++() noexcept -> ptr_wrapper& { ++mPointer; return *this; }
  37. constexpr auto operator--() noexcept -> ptr_wrapper& { --mPointer; return *this; }
  38. constexpr auto operator++(int) noexcept -> ptr_wrapper
  39. {
  40. auto temp = *this;
  41. ++*this;
  42. return temp;
  43. }
  44. constexpr auto operator--(int) noexcept -> ptr_wrapper
  45. {
  46. auto temp = *this;
  47. --*this;
  48. return temp;
  49. }
  50. constexpr
  51. auto operator+=(std::ptrdiff_t n) noexcept -> ptr_wrapper& { mPointer += n; return *this; }
  52. constexpr
  53. auto operator-=(std::ptrdiff_t n) noexcept -> ptr_wrapper& { mPointer -= n; return *this; }
  54. [[nodiscard]] constexpr auto operator*() const noexcept -> value_type& { return *mPointer; }
  55. [[nodiscard]] constexpr auto operator->() const noexcept -> value_type* { return mPointer; }
  56. [[nodiscard]] constexpr
  57. auto operator[](std::size_t idx) const noexcept -> value_type& {return mPointer[idx];}
  58. [[nodiscard]] friend constexpr
  59. auto operator+(const ptr_wrapper &lhs, std::ptrdiff_t n) noexcept -> ptr_wrapper
  60. { return ptr_wrapper{lhs.mPointer + n}; }
  61. [[nodiscard]] friend constexpr
  62. auto operator+(std::ptrdiff_t n, const ptr_wrapper &rhs) noexcept -> ptr_wrapper
  63. { return ptr_wrapper{n + rhs.mPointer}; }
  64. [[nodiscard]] friend constexpr
  65. auto operator-(const ptr_wrapper &lhs, std::ptrdiff_t n) noexcept -> ptr_wrapper
  66. { return ptr_wrapper{lhs.mPointer - n}; }
  67. [[nodiscard]] friend constexpr
  68. auto operator-(const ptr_wrapper &lhs, const ptr_wrapper &rhs)noexcept->std::ptrdiff_t
  69. { return lhs.mPointer - rhs.mPointer; }
  70. [[nodiscard]] friend constexpr
  71. auto operator==(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  72. { return lhs.mPointer == rhs.mPointer; }
  73. [[nodiscard]] friend constexpr
  74. auto operator!=(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  75. { return lhs.mPointer != rhs.mPointer; }
  76. [[nodiscard]] friend constexpr
  77. auto operator<=(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  78. { return lhs.mPointer <= rhs.mPointer; }
  79. [[nodiscard]] friend constexpr
  80. auto operator>=(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  81. { return lhs.mPointer >= rhs.mPointer; }
  82. [[nodiscard]] friend constexpr
  83. auto operator<(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  84. { return lhs.mPointer < rhs.mPointer; }
  85. [[nodiscard]] friend constexpr
  86. auto operator>(const ptr_wrapper &lhs, const ptr_wrapper &rhs) noexcept -> bool
  87. { return lhs.mPointer > rhs.mPointer; }
  88. /* NOLINTEND(cppcoreguidelines-pro-bounds-pointer-arithmetic) */
  89. };
  90. inline constexpr std::size_t dynamic_extent{static_cast<std::size_t>(-1)};
  91. template<typename T, std::size_t E=dynamic_extent>
  92. class span;
  93. namespace detail_ {
  94. template<typename T>
  95. struct is_span_ : std::false_type { };
  96. template<typename T, std::size_t E>
  97. struct is_span_<span<T,E>> : std::true_type { };
  98. template<typename T>
  99. inline constexpr bool is_span_v = is_span_<std::remove_cv_t<T>>::value;
  100. template<typename T>
  101. struct is_std_array_ : std::false_type { };
  102. template<typename T, std::size_t N>
  103. struct is_std_array_<std::array<T,N>> : std::true_type { };
  104. template<typename T>
  105. inline constexpr bool is_std_array_v = is_std_array_<std::remove_cv_t<T>>::value;
  106. template<typename T, typename = void>
  107. inline constexpr bool has_size_and_data = false;
  108. template<typename T>
  109. inline constexpr bool has_size_and_data<T,
  110. std::void_t<decltype(std::size(std::declval<T>())),decltype(std::data(std::declval<T>()))>>
  111. = true;
  112. template<typename C>
  113. inline constexpr bool is_valid_container_type = !is_span_v<C> && !is_std_array_v<C>
  114. && !std::is_array<C>::value && has_size_and_data<C>;
  115. template<typename T, typename U>
  116. inline constexpr bool is_array_compatible = std::is_convertible<T(*)[],U(*)[]>::value; /* NOLINT(*-avoid-c-arrays) */
  117. template<typename C, typename T>
  118. inline constexpr bool is_valid_container = is_valid_container_type<C>
  119. && is_array_compatible<std::remove_pointer_t<decltype(std::data(std::declval<C&>()))>,T>;
  120. } // namespace detail_
  121. #define REQUIRES(...) std::enable_if_t<(__VA_ARGS__),bool> = true
  122. /* NOLINTBEGIN(google-explicit-constructor) This largely follows std::span's
  123. * constructor behavior, and should be replaced once C++20 is used.
  124. */
  125. template<typename T, std::size_t E>
  126. class span {
  127. public:
  128. using element_type = T;
  129. using value_type = std::remove_cv_t<T>;
  130. using size_type = std::size_t;
  131. using difference_type = std::ptrdiff_t;
  132. using pointer = T*;
  133. using const_pointer = const T*;
  134. using reference = T&;
  135. using const_reference = const T&;
  136. using iterator = ptr_wrapper<pointer>;
  137. using const_iterator = ptr_wrapper<const_pointer>;
  138. using reverse_iterator = std::reverse_iterator<iterator>;
  139. using const_reverse_iterator = std::reverse_iterator<const_iterator>;
  140. static constexpr std::size_t extent{E};
  141. template<bool is0=(extent == 0), REQUIRES(is0)>
  142. constexpr span() noexcept { }
  143. template<typename U>
  144. constexpr explicit span(U iter, size_type size_) : mData{::al::to_address(iter)}
  145. { alassert(size_ == extent); }
  146. template<typename U, typename V, REQUIRES(!std::is_convertible<V,std::size_t>::value)>
  147. constexpr explicit span(U first, V last) : mData{::al::to_address(first)}
  148. { alassert(static_cast<std::size_t>(last-first) == extent); }
  149. template<std::size_t N>
  150. constexpr span(type_identity_t<element_type> (&arr)[N]) noexcept /* NOLINT(*-avoid-c-arrays) */
  151. : mData{std::data(arr)}
  152. { static_assert(N == extent); }
  153. template<std::size_t N>
  154. constexpr span(std::array<value_type,N> &arr) noexcept : mData{std::data(arr)}
  155. { static_assert(N == extent); }
  156. template<typename U=T, std::size_t N, REQUIRES(std::is_const<U>::value)>
  157. constexpr span(const std::array<value_type,N> &arr) noexcept : mData{std::data(arr)}
  158. { static_assert(N == extent); }
  159. template<typename U, REQUIRES(detail_::is_valid_container<U, element_type>)>
  160. constexpr explicit span(U&& cont) : span{std::data(cont), std::size(cont)} { }
  161. template<typename U, std::size_t N, REQUIRES(!std::is_same<element_type,U>::value
  162. && detail_::is_array_compatible<U,element_type> && N == dynamic_extent)>
  163. constexpr explicit span(const span<U,N> &span_) noexcept : mData{std::data(span_)}
  164. { alassert(std::size(span_) == extent); }
  165. template<typename U, std::size_t N, REQUIRES(!std::is_same<element_type,U>::value
  166. && detail_::is_array_compatible<U,element_type> && N == extent)>
  167. constexpr span(const span<U,N> &span_) noexcept : mData{std::data(span_)} { }
  168. constexpr span(const span&) noexcept = default;
  169. constexpr span& operator=(const span &rhs) noexcept = default;
  170. [[nodiscard]] constexpr auto front() const -> reference { return mData[0]; }
  171. [[nodiscard]] constexpr auto back() const -> reference { return mData[E-1]; }
  172. [[nodiscard]] constexpr auto operator[](size_type idx) const -> reference { return mData[idx]; }
  173. [[nodiscard]] constexpr auto data() const noexcept -> pointer { return mData; }
  174. [[nodiscard]] constexpr auto size() const noexcept -> size_type { return E; }
  175. [[nodiscard]] constexpr auto size_bytes() const noexcept -> size_type { return E * sizeof(value_type); }
  176. [[nodiscard]] constexpr auto empty() const noexcept -> bool { return E == 0; }
  177. [[nodiscard]] constexpr auto begin() const noexcept -> iterator { return iterator{mData}; }
  178. [[nodiscard]] constexpr auto end() const noexcept -> iterator { return iterator{mData+E}; }
  179. [[nodiscard]] constexpr
  180. auto cbegin() const noexcept -> const_iterator { return const_iterator{mData}; }
  181. [[nodiscard]] constexpr
  182. auto cend() const noexcept -> const_iterator { return const_iterator{mData+E}; }
  183. [[nodiscard]] constexpr
  184. auto rbegin() const noexcept -> reverse_iterator { return reverse_iterator{end()}; }
  185. [[nodiscard]] constexpr
  186. auto rend() const noexcept -> reverse_iterator { return reverse_iterator{begin()}; }
  187. [[nodiscard]] constexpr
  188. auto crbegin() const noexcept -> const_reverse_iterator { return cend(); }
  189. [[nodiscard]] constexpr
  190. auto crend() const noexcept -> const_reverse_iterator { return cbegin(); }
  191. template<std::size_t C>
  192. [[nodiscard]] constexpr auto first() const noexcept -> span<element_type,C>
  193. {
  194. static_assert(E >= C, "New size exceeds original capacity");
  195. return span<element_type,C>{mData, C};
  196. }
  197. template<std::size_t C>
  198. [[nodiscard]] constexpr auto last() const noexcept -> span<element_type,C>
  199. {
  200. static_assert(E >= C, "New size exceeds original capacity");
  201. return span<element_type,C>{mData+(E-C), C};
  202. }
  203. template<std::size_t O, std::size_t C>
  204. [[nodiscard]] constexpr
  205. auto subspan() const noexcept -> std::enable_if_t<C!=dynamic_extent,span<element_type,C>>
  206. {
  207. static_assert(E >= O, "Offset exceeds extent");
  208. static_assert(E-O >= C, "New size exceeds original capacity");
  209. return span<element_type,C>{mData+O, C};
  210. }
  211. template<std::size_t O, std::size_t C=dynamic_extent>
  212. [[nodiscard]] constexpr
  213. auto subspan() const noexcept -> std::enable_if_t<C==dynamic_extent,span<element_type,E-O>>
  214. {
  215. static_assert(E >= O, "Offset exceeds extent");
  216. return span<element_type,E-O>{mData+O, E-O};
  217. }
  218. /* NOTE: Can't declare objects of a specialized template class prior to
  219. * defining the specialization. As a result, these methods need to be
  220. * defined later.
  221. */
  222. [[nodiscard]] constexpr
  223. auto first(std::size_t count) const noexcept -> span<element_type,dynamic_extent>;
  224. [[nodiscard]] constexpr
  225. auto last(std::size_t count) const noexcept -> span<element_type,dynamic_extent>;
  226. [[nodiscard]] constexpr
  227. auto subspan(std::size_t offset, std::size_t count=dynamic_extent) const noexcept
  228. -> span<element_type,dynamic_extent>;
  229. private:
  230. pointer mData{nullptr};
  231. };
  232. template<typename T>
  233. class span<T,dynamic_extent> {
  234. public:
  235. using element_type = T;
  236. using value_type = std::remove_cv_t<T>;
  237. using size_type = std::size_t;
  238. using difference_type = ptrdiff_t;
  239. using pointer = T*;
  240. using const_pointer = const T*;
  241. using reference = T&;
  242. using const_reference = const T&;
  243. using iterator = ptr_wrapper<pointer>;
  244. using const_iterator = ptr_wrapper<const_pointer>;
  245. using reverse_iterator = std::reverse_iterator<iterator>;
  246. using const_reverse_iterator = std::reverse_iterator<const_iterator>;
  247. static constexpr std::size_t extent{dynamic_extent};
  248. constexpr span() noexcept = default;
  249. template<typename U>
  250. constexpr span(U iter, size_type count) : mData{::al::to_address(iter)}, mDataLength{count}
  251. { }
  252. template<typename U, typename V, REQUIRES(!std::is_convertible<V,std::size_t>::value)>
  253. constexpr span(U first, V last)
  254. : span{::al::to_address(first), static_cast<std::size_t>(last-first)}
  255. { }
  256. template<std::size_t N>
  257. constexpr span(type_identity_t<element_type> (&arr)[N]) noexcept /* NOLINT(*-avoid-c-arrays) */
  258. : mData{std::data(arr)}, mDataLength{std::size(arr)}
  259. { }
  260. template<std::size_t N>
  261. constexpr span(std::array<value_type,N> &arr) noexcept
  262. : mData{std::data(arr)}, mDataLength{std::size(arr)}
  263. { }
  264. template<std::size_t N, typename U=T, REQUIRES(std::is_const<U>::value)>
  265. constexpr span(const std::array<value_type,N> &arr) noexcept
  266. : mData{std::data(arr)}, mDataLength{std::size(arr)}
  267. { }
  268. template<typename U, REQUIRES(detail_::is_valid_container<U, element_type>)>
  269. constexpr span(U&& cont) : span{std::data(cont), std::size(cont)} { }
  270. template<typename U, std::size_t N, REQUIRES(detail_::is_array_compatible<U,element_type>
  271. && (!std::is_same<element_type,U>::value || extent != N))>
  272. constexpr span(const span<U,N> &span_) noexcept : span{std::data(span_), std::size(span_)} { }
  273. constexpr span(const span&) noexcept = default;
  274. constexpr span& operator=(const span &rhs) noexcept = default;
  275. [[nodiscard]] constexpr auto front() const -> reference { return mData[0]; }
  276. [[nodiscard]] constexpr auto back() const -> reference { return mData[mDataLength-1]; }
  277. [[nodiscard]] constexpr auto operator[](size_type idx) const -> reference {return mData[idx];}
  278. [[nodiscard]] constexpr auto data() const noexcept -> pointer { return mData; }
  279. [[nodiscard]] constexpr auto size() const noexcept -> size_type { return mDataLength; }
  280. [[nodiscard]] constexpr
  281. auto size_bytes() const noexcept -> size_type { return mDataLength * sizeof(value_type); }
  282. [[nodiscard]] constexpr auto empty() const noexcept -> bool { return mDataLength == 0; }
  283. [[nodiscard]] constexpr auto begin() const noexcept -> iterator { return iterator{mData}; }
  284. [[nodiscard]] constexpr
  285. auto end() const noexcept -> iterator { return iterator{mData+mDataLength}; }
  286. [[nodiscard]] constexpr
  287. auto cbegin() const noexcept -> const_iterator { return const_iterator{mData}; }
  288. [[nodiscard]] constexpr
  289. auto cend() const noexcept -> const_iterator { return const_iterator{mData+mDataLength}; }
  290. [[nodiscard]] constexpr
  291. auto rbegin() const noexcept -> reverse_iterator { return reverse_iterator{end()}; }
  292. [[nodiscard]] constexpr
  293. auto rend() const noexcept -> reverse_iterator { return reverse_iterator{begin()}; }
  294. [[nodiscard]] constexpr
  295. auto crbegin() const noexcept -> const_reverse_iterator { return cend(); }
  296. [[nodiscard]] constexpr
  297. auto crend() const noexcept -> const_reverse_iterator { return cbegin(); }
  298. template<std::size_t C>
  299. [[nodiscard]] constexpr auto first() const noexcept -> span<element_type,C>
  300. {
  301. assert(C <= mDataLength);
  302. return span<element_type,C>{mData, C};
  303. }
  304. [[nodiscard]] constexpr auto first(std::size_t count) const noexcept -> span
  305. {
  306. assert(count <= mDataLength);
  307. return span{mData, count};
  308. }
  309. template<std::size_t C>
  310. [[nodiscard]] constexpr auto last() const noexcept -> span<element_type,C>
  311. {
  312. assert(C <= mDataLength);
  313. return span<element_type,C>{mData+mDataLength-C, C};
  314. }
  315. [[nodiscard]] constexpr auto last(std::size_t count) const noexcept -> span
  316. {
  317. assert(count <= mDataLength);
  318. return span{mData+mDataLength-count, count};
  319. }
  320. template<std::size_t O, std::size_t C>
  321. [[nodiscard]] constexpr
  322. auto subspan() const noexcept -> std::enable_if_t<C!=dynamic_extent,span<element_type,C>>
  323. {
  324. assert(O <= mDataLength);
  325. assert(C <= mDataLength-O);
  326. return span<element_type,C>{mData+O, C};
  327. }
  328. template<std::size_t O, std::size_t C=dynamic_extent>
  329. [[nodiscard]] constexpr
  330. auto subspan() const noexcept -> std::enable_if_t<C==dynamic_extent,span<element_type,C>>
  331. {
  332. assert(O <= mDataLength);
  333. return span<element_type,C>{mData+O, mDataLength-O};
  334. }
  335. [[nodiscard]] constexpr
  336. auto subspan(std::size_t offset, std::size_t count=dynamic_extent) const noexcept -> span
  337. {
  338. assert(offset <= mDataLength);
  339. if(count != dynamic_extent)
  340. {
  341. assert(count <= mDataLength-offset);
  342. return span{mData+offset, count};
  343. }
  344. return span{mData+offset, mDataLength-offset};
  345. }
  346. private:
  347. pointer mData{nullptr};
  348. size_type mDataLength{0};
  349. };
  350. template<typename T, std::size_t E>
  351. [[nodiscard]] constexpr
  352. auto span<T,E>::first(std::size_t count) const noexcept -> span<element_type,dynamic_extent>
  353. {
  354. assert(count <= size());
  355. return span<element_type>{mData, count};
  356. }
  357. template<typename T, std::size_t E>
  358. [[nodiscard]] constexpr
  359. auto span<T,E>::last(std::size_t count) const noexcept -> span<element_type,dynamic_extent>
  360. {
  361. assert(count <= size());
  362. return span<element_type>{mData+size()-count, count};
  363. }
  364. template<typename T, std::size_t E>
  365. [[nodiscard]] constexpr
  366. auto span<T,E>::subspan(std::size_t offset, std::size_t count) const noexcept
  367. -> span<element_type,dynamic_extent>
  368. {
  369. assert(offset <= size());
  370. if(count != dynamic_extent)
  371. {
  372. assert(count <= size()-offset);
  373. return span<element_type>{mData+offset, count};
  374. }
  375. return span<element_type>{mData+offset, size()-offset};
  376. }
  377. /* NOLINTEND(google-explicit-constructor) */
  378. template<typename T, typename EndOrSize>
  379. span(T, EndOrSize) -> span<std::remove_reference_t<decltype(*std::declval<T&>())>>;
  380. template<typename T, std::size_t N>
  381. span(T (&)[N]) -> span<T, N>; /* NOLINT(*-avoid-c-arrays) */
  382. template<typename T, std::size_t N>
  383. span(std::array<T, N>&) -> span<T, N>;
  384. template<typename T, std::size_t N>
  385. span(const std::array<T, N>&) -> span<const T, N>;
  386. template<typename C, REQUIRES(detail_::is_valid_container_type<C>)>
  387. span(C&&) -> span<std::remove_pointer_t<decltype(std::data(std::declval<C&>()))>>;
  388. #undef REQUIRES
  389. } // namespace al
  390. #endif /* AL_SPAN_H */