ranges_util.h 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423
  1. // Utilities for representing and manipulating ranges -*- C++ -*-
  2. // Copyright (C) 2019-2021 Free Software Foundation, Inc.
  3. //
  4. // This file is part of the GNU ISO C++ Library. This library is free
  5. // software; you can redistribute it and/or modify it under the
  6. // terms of the GNU General Public License as published by the
  7. // Free Software Foundation; either version 3, or (at your option)
  8. // any later version.
  9. // This library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU General Public License for more details.
  13. // Under Section 7 of GPL version 3, you are granted additional
  14. // permissions described in the GCC Runtime Library Exception, version
  15. // 3.1, as published by the Free Software Foundation.
  16. // You should have received a copy of the GNU General Public License and
  17. // a copy of the GCC Runtime Library Exception along with this program;
  18. // see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
  19. // <http://www.gnu.org/licenses/>.
  20. /** @file bits/ranges_util.h
  21. * This is an internal header file, included by other library headers.
  22. * Do not attempt to use it directly. @headername{ranges}
  23. */
  24. #ifndef _RANGES_UTIL_H
  25. #define _RANGES_UTIL_H 1
  26. #if __cplusplus > 201703L
  27. # include <bits/ranges_base.h>
  28. #ifdef __cpp_lib_ranges
  29. namespace std _GLIBCXX_VISIBILITY(default)
  30. {
  31. _GLIBCXX_BEGIN_NAMESPACE_VERSION
  32. namespace ranges
  33. {
  34. // C++20 24.5 [range.utility] Range utilities
  35. namespace __detail
  36. {
  37. template<typename _Range>
  38. concept __simple_view = view<_Range> && range<const _Range>
  39. && same_as<iterator_t<_Range>, iterator_t<const _Range>>
  40. && same_as<sentinel_t<_Range>, sentinel_t<const _Range>>;
  41. template<typename _It>
  42. concept __has_arrow = input_iterator<_It>
  43. && (is_pointer_v<_It> || requires(_It __it) { __it.operator->(); });
  44. template<typename _Tp, typename _Up>
  45. concept __not_same_as
  46. = !same_as<remove_cvref_t<_Tp>, remove_cvref_t<_Up>>;
  47. } // namespace __detail
  48. /// The ranges::view_interface class template
  49. template<typename _Derived>
  50. requires is_class_v<_Derived> && same_as<_Derived, remove_cv_t<_Derived>>
  51. class view_interface : public view_base
  52. {
  53. private:
  54. constexpr _Derived& _M_derived() noexcept
  55. {
  56. static_assert(derived_from<_Derived, view_interface<_Derived>>);
  57. static_assert(view<_Derived>);
  58. return static_cast<_Derived&>(*this);
  59. }
  60. constexpr const _Derived& _M_derived() const noexcept
  61. {
  62. static_assert(derived_from<_Derived, view_interface<_Derived>>);
  63. static_assert(view<_Derived>);
  64. return static_cast<const _Derived&>(*this);
  65. }
  66. public:
  67. constexpr bool
  68. empty() requires forward_range<_Derived>
  69. { return ranges::begin(_M_derived()) == ranges::end(_M_derived()); }
  70. constexpr bool
  71. empty() const requires forward_range<const _Derived>
  72. { return ranges::begin(_M_derived()) == ranges::end(_M_derived()); }
  73. constexpr explicit
  74. operator bool() requires requires { ranges::empty(_M_derived()); }
  75. { return !ranges::empty(_M_derived()); }
  76. constexpr explicit
  77. operator bool() const requires requires { ranges::empty(_M_derived()); }
  78. { return !ranges::empty(_M_derived()); }
  79. constexpr auto
  80. data() requires contiguous_iterator<iterator_t<_Derived>>
  81. { return to_address(ranges::begin(_M_derived())); }
  82. constexpr auto
  83. data() const
  84. requires range<const _Derived>
  85. && contiguous_iterator<iterator_t<const _Derived>>
  86. { return to_address(ranges::begin(_M_derived())); }
  87. constexpr auto
  88. size()
  89. requires forward_range<_Derived>
  90. && sized_sentinel_for<sentinel_t<_Derived>, iterator_t<_Derived>>
  91. { return ranges::end(_M_derived()) - ranges::begin(_M_derived()); }
  92. constexpr auto
  93. size() const
  94. requires forward_range<const _Derived>
  95. && sized_sentinel_for<sentinel_t<const _Derived>,
  96. iterator_t<const _Derived>>
  97. { return ranges::end(_M_derived()) - ranges::begin(_M_derived()); }
  98. constexpr decltype(auto)
  99. front() requires forward_range<_Derived>
  100. {
  101. __glibcxx_assert(!empty());
  102. return *ranges::begin(_M_derived());
  103. }
  104. constexpr decltype(auto)
  105. front() const requires forward_range<const _Derived>
  106. {
  107. __glibcxx_assert(!empty());
  108. return *ranges::begin(_M_derived());
  109. }
  110. constexpr decltype(auto)
  111. back()
  112. requires bidirectional_range<_Derived> && common_range<_Derived>
  113. {
  114. __glibcxx_assert(!empty());
  115. return *ranges::prev(ranges::end(_M_derived()));
  116. }
  117. constexpr decltype(auto)
  118. back() const
  119. requires bidirectional_range<const _Derived>
  120. && common_range<const _Derived>
  121. {
  122. __glibcxx_assert(!empty());
  123. return *ranges::prev(ranges::end(_M_derived()));
  124. }
  125. template<random_access_range _Range = _Derived>
  126. constexpr decltype(auto)
  127. operator[](range_difference_t<_Range> __n)
  128. { return ranges::begin(_M_derived())[__n]; }
  129. template<random_access_range _Range = const _Derived>
  130. constexpr decltype(auto)
  131. operator[](range_difference_t<_Range> __n) const
  132. { return ranges::begin(_M_derived())[__n]; }
  133. };
  134. namespace __detail
  135. {
  136. template<class _From, class _To>
  137. concept __convertible_to_non_slicing = convertible_to<_From, _To>
  138. && !(is_pointer_v<decay_t<_From>> && is_pointer_v<decay_t<_To>>
  139. && __not_same_as<remove_pointer_t<decay_t<_From>>,
  140. remove_pointer_t<decay_t<_To>>>);
  141. template<typename _Tp>
  142. concept __pair_like
  143. = !is_reference_v<_Tp> && requires(_Tp __t)
  144. {
  145. typename tuple_size<_Tp>::type;
  146. requires derived_from<tuple_size<_Tp>, integral_constant<size_t, 2>>;
  147. typename tuple_element_t<0, remove_const_t<_Tp>>;
  148. typename tuple_element_t<1, remove_const_t<_Tp>>;
  149. { get<0>(__t) } -> convertible_to<const tuple_element_t<0, _Tp>&>;
  150. { get<1>(__t) } -> convertible_to<const tuple_element_t<1, _Tp>&>;
  151. };
  152. template<typename _Tp, typename _Up, typename _Vp>
  153. concept __pair_like_convertible_from
  154. = !range<_Tp> && __pair_like<_Tp>
  155. && constructible_from<_Tp, _Up, _Vp>
  156. && __convertible_to_non_slicing<_Up, tuple_element_t<0, _Tp>>
  157. && convertible_to<_Vp, tuple_element_t<1, _Tp>>;
  158. } // namespace __detail
  159. enum class subrange_kind : bool { unsized, sized };
  160. /// The ranges::subrange class template
  161. template<input_or_output_iterator _It, sentinel_for<_It> _Sent = _It,
  162. subrange_kind _Kind = sized_sentinel_for<_Sent, _It>
  163. ? subrange_kind::sized : subrange_kind::unsized>
  164. requires (_Kind == subrange_kind::sized || !sized_sentinel_for<_Sent, _It>)
  165. class subrange : public view_interface<subrange<_It, _Sent, _Kind>>
  166. {
  167. private:
  168. // XXX: gcc complains when using constexpr here
  169. static const bool _S_store_size
  170. = _Kind == subrange_kind::sized && !sized_sentinel_for<_Sent, _It>;
  171. _It _M_begin = _It();
  172. [[no_unique_address]] _Sent _M_end = _Sent();
  173. template<typename, bool = _S_store_size>
  174. struct _Size
  175. { };
  176. template<typename _Tp>
  177. struct _Size<_Tp, true>
  178. { __detail::__make_unsigned_like_t<_Tp> _M_size; };
  179. [[no_unique_address]] _Size<iter_difference_t<_It>> _M_size = {};
  180. public:
  181. subrange() = default;
  182. constexpr
  183. subrange(__detail::__convertible_to_non_slicing<_It> auto __i, _Sent __s)
  184. requires (!_S_store_size)
  185. : _M_begin(std::move(__i)), _M_end(__s)
  186. { }
  187. constexpr
  188. subrange(__detail::__convertible_to_non_slicing<_It> auto __i, _Sent __s,
  189. __detail::__make_unsigned_like_t<iter_difference_t<_It>> __n)
  190. requires (_Kind == subrange_kind::sized)
  191. : _M_begin(std::move(__i)), _M_end(__s)
  192. {
  193. using __detail::__to_unsigned_like;
  194. __glibcxx_assert(__n == __to_unsigned_like(ranges::distance(__i, __s)));
  195. if constexpr (_S_store_size)
  196. _M_size._M_size = __n;
  197. }
  198. template<__detail::__not_same_as<subrange> _Rng>
  199. requires borrowed_range<_Rng>
  200. && __detail::__convertible_to_non_slicing<iterator_t<_Rng>, _It>
  201. && convertible_to<sentinel_t<_Rng>, _Sent>
  202. constexpr
  203. subrange(_Rng&& __r) requires _S_store_size && sized_range<_Rng>
  204. : subrange(__r, ranges::size(__r))
  205. { }
  206. template<__detail::__not_same_as<subrange> _Rng>
  207. requires borrowed_range<_Rng>
  208. && __detail::__convertible_to_non_slicing<iterator_t<_Rng>, _It>
  209. && convertible_to<sentinel_t<_Rng>, _Sent>
  210. constexpr
  211. subrange(_Rng&& __r) requires (!_S_store_size)
  212. : subrange{ranges::begin(__r), ranges::end(__r)}
  213. { }
  214. template<borrowed_range _Rng>
  215. requires __detail::__convertible_to_non_slicing<iterator_t<_Rng>, _It>
  216. && convertible_to<sentinel_t<_Rng>, _Sent>
  217. constexpr
  218. subrange(_Rng&& __r,
  219. __detail::__make_unsigned_like_t<iter_difference_t<_It>> __n)
  220. requires (_Kind == subrange_kind::sized)
  221. : subrange{ranges::begin(__r), ranges::end(__r), __n}
  222. { }
  223. template<__detail::__not_same_as<subrange> _PairLike>
  224. requires __detail::__pair_like_convertible_from<_PairLike, const _It&,
  225. const _Sent&>
  226. constexpr
  227. operator _PairLike() const
  228. { return _PairLike(_M_begin, _M_end); }
  229. constexpr _It
  230. begin() const requires copyable<_It>
  231. { return _M_begin; }
  232. [[nodiscard]] constexpr _It
  233. begin() requires (!copyable<_It>)
  234. { return std::move(_M_begin); }
  235. constexpr _Sent end() const { return _M_end; }
  236. constexpr bool empty() const { return _M_begin == _M_end; }
  237. constexpr __detail::__make_unsigned_like_t<iter_difference_t<_It>>
  238. size() const requires (_Kind == subrange_kind::sized)
  239. {
  240. if constexpr (_S_store_size)
  241. return _M_size._M_size;
  242. else
  243. return __detail::__to_unsigned_like(_M_end - _M_begin);
  244. }
  245. [[nodiscard]] constexpr subrange
  246. next(iter_difference_t<_It> __n = 1) const &
  247. requires forward_iterator<_It>
  248. {
  249. auto __tmp = *this;
  250. __tmp.advance(__n);
  251. return __tmp;
  252. }
  253. [[nodiscard]] constexpr subrange
  254. next(iter_difference_t<_It> __n = 1) &&
  255. {
  256. advance(__n);
  257. return std::move(*this);
  258. }
  259. [[nodiscard]] constexpr subrange
  260. prev(iter_difference_t<_It> __n = 1) const
  261. requires bidirectional_iterator<_It>
  262. {
  263. auto __tmp = *this;
  264. __tmp.advance(-__n);
  265. return __tmp;
  266. }
  267. constexpr subrange&
  268. advance(iter_difference_t<_It> __n)
  269. {
  270. // _GLIBCXX_RESOLVE_LIB_DEFECTS
  271. // 3433. subrange::advance(n) has UB when n < 0
  272. if constexpr (bidirectional_iterator<_It>)
  273. if (__n < 0)
  274. {
  275. ranges::advance(_M_begin, __n);
  276. if constexpr (_S_store_size)
  277. _M_size._M_size += __detail::__to_unsigned_like(-__n);
  278. return *this;
  279. }
  280. __glibcxx_assert(__n >= 0);
  281. auto __d = __n - ranges::advance(_M_begin, __n, _M_end);
  282. if constexpr (_S_store_size)
  283. _M_size._M_size -= __detail::__to_unsigned_like(__d);
  284. return *this;
  285. }
  286. };
  287. template<input_or_output_iterator _It, sentinel_for<_It> _Sent>
  288. subrange(_It, _Sent) -> subrange<_It, _Sent>;
  289. template<input_or_output_iterator _It, sentinel_for<_It> _Sent>
  290. subrange(_It, _Sent,
  291. __detail::__make_unsigned_like_t<iter_difference_t<_It>>)
  292. -> subrange<_It, _Sent, subrange_kind::sized>;
  293. template<borrowed_range _Rng>
  294. subrange(_Rng&&)
  295. -> subrange<iterator_t<_Rng>, sentinel_t<_Rng>,
  296. (sized_range<_Rng>
  297. || sized_sentinel_for<sentinel_t<_Rng>, iterator_t<_Rng>>)
  298. ? subrange_kind::sized : subrange_kind::unsized>;
  299. template<borrowed_range _Rng>
  300. subrange(_Rng&&,
  301. __detail::__make_unsigned_like_t<range_difference_t<_Rng>>)
  302. -> subrange<iterator_t<_Rng>, sentinel_t<_Rng>, subrange_kind::sized>;
  303. template<size_t _Num, class _It, class _Sent, subrange_kind _Kind>
  304. requires (_Num < 2)
  305. constexpr auto
  306. get(const subrange<_It, _Sent, _Kind>& __r)
  307. {
  308. if constexpr (_Num == 0)
  309. return __r.begin();
  310. else
  311. return __r.end();
  312. }
  313. template<size_t _Num, class _It, class _Sent, subrange_kind _Kind>
  314. requires (_Num < 2)
  315. constexpr auto
  316. get(subrange<_It, _Sent, _Kind>&& __r)
  317. {
  318. if constexpr (_Num == 0)
  319. return __r.begin();
  320. else
  321. return __r.end();
  322. }
  323. template<input_or_output_iterator _It, sentinel_for<_It> _Sent,
  324. subrange_kind _Kind>
  325. inline constexpr bool
  326. enable_borrowed_range<subrange<_It, _Sent, _Kind>> = true;
  327. template<range _Range>
  328. using borrowed_subrange_t = conditional_t<borrowed_range<_Range>,
  329. subrange<iterator_t<_Range>>,
  330. dangling>;
  331. } // namespace ranges
  332. using ranges::get;
  333. template<typename _Iter, typename _Sent, ranges::subrange_kind _Kind>
  334. struct tuple_size<ranges::subrange<_Iter, _Sent, _Kind>>
  335. : integral_constant<size_t, 2>
  336. { };
  337. template<typename _Iter, typename _Sent, ranges::subrange_kind _Kind>
  338. struct tuple_element<0, ranges::subrange<_Iter, _Sent, _Kind>>
  339. { using type = _Iter; };
  340. template<typename _Iter, typename _Sent, ranges::subrange_kind _Kind>
  341. struct tuple_element<1, ranges::subrange<_Iter, _Sent, _Kind>>
  342. { using type = _Sent; };
  343. template<typename _Iter, typename _Sent, ranges::subrange_kind _Kind>
  344. struct tuple_element<0, const ranges::subrange<_Iter, _Sent, _Kind>>
  345. { using type = _Iter; };
  346. template<typename _Iter, typename _Sent, ranges::subrange_kind _Kind>
  347. struct tuple_element<1, const ranges::subrange<_Iter, _Sent, _Kind>>
  348. { using type = _Sent; };
  349. _GLIBCXX_END_NAMESPACE_VERSION
  350. } // namespace std
  351. #endif // library concepts
  352. #endif // C++20
  353. #endif // _RANGES_UTIL_H