take.hpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327
  1. /// \file
  2. // Range v3 library
  3. //
  4. // Copyright Eric Niebler 2013-present
  5. //
  6. // Use, modification and distribution is subject to the
  7. // Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at
  9. // http://www.boost.org/LICENSE_1_0.txt)
  10. //
  11. // Project home: https://github.com/ericniebler/range-v3
  12. //
  13. #ifndef RANGES_V3_VIEW_TAKE_HPP
  14. #define RANGES_V3_VIEW_TAKE_HPP
  15. #include <type_traits>
  16. #include <range/v3/range_fwd.hpp>
  17. #include <range/v3/algorithm/min.hpp>
  18. #include <range/v3/functional/bind_back.hpp>
  19. #include <range/v3/iterator/counted_iterator.hpp>
  20. #include <range/v3/iterator/default_sentinel.hpp>
  21. #include <range/v3/range/concepts.hpp>
  22. #include <range/v3/range/traits.hpp>
  23. #include <range/v3/utility/static_const.hpp>
  24. #include <range/v3/view/all.hpp>
  25. #include <range/v3/view/view.hpp>
  26. #include <range/v3/detail/prologue.hpp>
  27. namespace ranges
  28. {
  29. /// \addtogroup group-views
  30. /// @{
  31. template<typename Rng>
  32. struct take_view : view_interface<take_view<Rng>, finite>
  33. {
  34. private:
  35. CPP_assert(view_<Rng>);
  36. Rng base_ = Rng();
  37. range_difference_t<Rng> count_ = 0;
  38. template<bool Const>
  39. struct sentinel
  40. {
  41. private:
  42. using Base = meta::conditional_t<Const, Rng const, Rng>;
  43. using CI = counted_iterator<iterator_t<Base>>;
  44. sentinel_t<Base> end_ = sentinel_t<Base>();
  45. public:
  46. sentinel() = default;
  47. constexpr explicit sentinel(sentinel_t<Base> last)
  48. : end_(std::move(last))
  49. {}
  50. template(bool Other)(
  51. requires Const AND CPP_NOT(Other) AND
  52. convertible_to<sentinel_t<Rng>,
  53. sentinel_t<Base>>)
  54. constexpr sentinel(sentinel<Other> that)
  55. : end_(std::move(that.end_))
  56. {}
  57. constexpr sentinel_t<Base> base() const
  58. {
  59. return end_;
  60. }
  61. #ifdef RANGES_WORKAROUND_MSVC_756601
  62. template<typename = void>
  63. #endif // RANGES_WORKAROUND_MSVC_756601
  64. friend constexpr bool operator==(sentinel const & x, CI const & y)
  65. {
  66. return y.count() == 0 || y.base() == x.end_;
  67. }
  68. #ifdef RANGES_WORKAROUND_MSVC_756601
  69. template<typename = void>
  70. #endif // RANGES_WORKAROUND_MSVC_756601
  71. friend constexpr bool operator==(CI const & y, sentinel const & x)
  72. {
  73. return y.count() == 0 || y.base() == x.end_;
  74. }
  75. #ifdef RANGES_WORKAROUND_MSVC_756601
  76. template<typename = void>
  77. #endif // RANGES_WORKAROUND_MSVC_756601
  78. friend constexpr bool operator!=(sentinel const & x, CI const & y)
  79. {
  80. return y.count() != 0 && y.base() != x.end_;
  81. }
  82. #ifdef RANGES_WORKAROUND_MSVC_756601
  83. template<typename = void>
  84. #endif // RANGES_WORKAROUND_MSVC_756601
  85. friend constexpr bool operator!=(CI const & y, sentinel const & x)
  86. {
  87. return y.count() != 0 && y.base() != x.end_;
  88. }
  89. };
  90. #if RANGES_CXX_IF_CONSTEXPR < RANGES_CXX_IF_CONSTEXPR_17
  91. template<typename Take>
  92. static auto begin_random_access_(Take & take, std::true_type)
  93. {
  94. return ranges::begin(take.base_);
  95. }
  96. template<typename Take>
  97. static auto begin_random_access_(Take & take, std::false_type)
  98. {
  99. auto s = static_cast<range_difference_t<Rng>>(take.size());
  100. return make_counted_iterator(ranges::begin(take.base_), s);
  101. }
  102. template<typename Take>
  103. static auto begin_sized_(Take & take, std::true_type)
  104. {
  105. return begin_random_access_(
  106. take, meta::bool_<random_access_range<decltype((take.base_))>>{});
  107. }
  108. template<typename Take>
  109. static auto begin_sized_(Take & take, std::false_type)
  110. {
  111. return make_counted_iterator(ranges::begin(take.base_), take.count_);
  112. }
  113. template<typename Take>
  114. static auto end_random_access_(Take & take, std::true_type)
  115. {
  116. return ranges::begin(take.base_) +
  117. static_cast<range_difference_t<Rng>>(take.size());
  118. }
  119. static auto end_random_access_(detail::ignore_t, std::false_type)
  120. {
  121. return default_sentinel;
  122. }
  123. template<typename Take>
  124. static auto end_sized_(Take & take, std::true_type, std::false_type) // sized
  125. {
  126. return end_random_access_(
  127. take, meta::bool_<random_access_range<decltype((take.base_))>>{});
  128. }
  129. static auto end_sized_(detail::ignore_t, std::false_type,
  130. std::true_type) // infinite
  131. {
  132. return default_sentinel;
  133. }
  134. static auto end_sized_(take_view & take, std::false_type, std::false_type)
  135. {
  136. return sentinel<false>{ranges::end(take.base_)};
  137. }
  138. static auto end_sized_(take_view const & take, std::false_type, std::false_type)
  139. {
  140. return sentinel<true>{ranges::end(take.base_)};
  141. }
  142. #endif
  143. public:
  144. take_view() = default;
  145. constexpr take_view(Rng base, range_difference_t<Rng> cnt)
  146. : base_(std::move(base))
  147. , count_(cnt)
  148. {}
  149. constexpr Rng base() const
  150. {
  151. return base_;
  152. }
  153. CPP_auto_member
  154. constexpr auto CPP_fun(begin)()(
  155. requires(!simple_view<Rng>()))
  156. {
  157. #if RANGES_CXX_IF_CONSTEXPR >= RANGES_CXX_IF_CONSTEXPR_17
  158. if constexpr(sized_range<Rng>)
  159. if constexpr(random_access_range<Rng>)
  160. return ranges::begin(base_);
  161. else
  162. {
  163. // cannot always delegate to size() member on GCC with ConceptsTS
  164. #if defined(__cpp_concepts) && __cpp_concepts <= 201507
  165. auto s = ranges::min(
  166. static_cast<range_difference_t<Rng>>(count_),
  167. static_cast<range_difference_t<Rng>>(ranges::size(base_)));
  168. #else
  169. auto s = static_cast<range_difference_t<Rng>>(size());
  170. #endif
  171. return make_counted_iterator(ranges::begin(base_), s);
  172. }
  173. else
  174. return make_counted_iterator(ranges::begin(base_), count_);
  175. #else
  176. return begin_sized_(*this, meta::bool_<sized_range<Rng>>{});
  177. #endif
  178. }
  179. CPP_auto_member
  180. constexpr auto CPP_fun(begin)()(const //
  181. requires range<Rng const>)
  182. {
  183. #if RANGES_CXX_IF_CONSTEXPR >= RANGES_CXX_IF_CONSTEXPR_17
  184. if constexpr(sized_range<Rng const>)
  185. if constexpr(random_access_range<Rng const>)
  186. return ranges::begin(base_);
  187. else
  188. {
  189. auto s = static_cast<range_difference_t<Rng>>(size());
  190. return make_counted_iterator(ranges::begin(base_), s);
  191. }
  192. else
  193. return make_counted_iterator(ranges::begin(base_), count_);
  194. #else
  195. return begin_sized_(*this, meta::bool_<sized_range<Rng const>>{});
  196. #endif
  197. }
  198. CPP_auto_member
  199. constexpr auto CPP_fun(end)()(
  200. requires(!simple_view<Rng>()))
  201. {
  202. #if RANGES_CXX_IF_CONSTEXPR >= RANGES_CXX_IF_CONSTEXPR_17
  203. if constexpr(sized_range<Rng>)
  204. if constexpr(random_access_range<Rng>)
  205. return ranges::begin(base_) +
  206. static_cast<range_difference_t<Rng>>(size());
  207. else
  208. return default_sentinel;
  209. // Not to spec: Infinite ranges:
  210. else if constexpr(is_infinite<Rng>::value)
  211. return default_sentinel;
  212. else
  213. return sentinel<false>{ranges::end(base_)};
  214. #else
  215. return end_sized_(*this, meta::bool_<sized_range<Rng>>{}, is_infinite<Rng>{});
  216. #endif
  217. }
  218. CPP_auto_member
  219. constexpr auto CPP_fun(end)()(const //
  220. requires range<Rng const>)
  221. {
  222. #if RANGES_CXX_IF_CONSTEXPR >= RANGES_CXX_IF_CONSTEXPR_17
  223. if constexpr(sized_range<Rng const>)
  224. if constexpr(random_access_range<Rng const>)
  225. return ranges::begin(base_) +
  226. static_cast<range_difference_t<Rng>>(size());
  227. else
  228. return default_sentinel;
  229. // Not to spec: Infinite ranges:
  230. else if constexpr(is_infinite<Rng const>::value)
  231. return default_sentinel;
  232. else
  233. return sentinel<true>{ranges::end(base_)};
  234. #else
  235. return end_sized_(
  236. *this, meta::bool_<sized_range<Rng const>>{}, is_infinite<Rng const>{});
  237. #endif
  238. }
  239. CPP_auto_member
  240. constexpr auto CPP_fun(size)()(
  241. requires sized_range<Rng>)
  242. {
  243. auto n = ranges::size(base_);
  244. return ranges::min(n, static_cast<decltype(n)>(count_));
  245. }
  246. CPP_auto_member
  247. constexpr auto CPP_fun(size)()(const //
  248. requires sized_range<Rng const>)
  249. {
  250. auto n = ranges::size(base_);
  251. return ranges::min(n, static_cast<decltype(n)>(count_));
  252. }
  253. };
  254. template<typename Rng>
  255. RANGES_INLINE_VAR constexpr bool enable_borrowed_range<take_view<Rng>> = //
  256. enable_borrowed_range<Rng>;
  257. #if RANGES_CXX_DEDUCTION_GUIDES >= RANGES_CXX_DEDUCTION_GUIDES_17
  258. template<typename Rng>
  259. take_view(Rng &&, range_difference_t<Rng>)
  260. -> take_view<views::all_t<Rng>>;
  261. #endif
  262. namespace views
  263. {
  264. struct take_base_fn
  265. {
  266. template(typename Rng)(
  267. requires viewable_range<Rng>)
  268. take_view<all_t<Rng>> operator()(Rng && rng, range_difference_t<Rng> n) const
  269. {
  270. return {all(static_cast<Rng &&>(rng)), n};
  271. }
  272. };
  273. struct take_fn : take_base_fn
  274. {
  275. using take_base_fn::operator();
  276. template(typename Int)(
  277. requires detail::integer_like_<Int>)
  278. constexpr auto operator()(Int n) const
  279. {
  280. return make_view_closure(bind_back(take_base_fn{}, n));
  281. }
  282. };
  283. /// \relates take_fn
  284. RANGES_INLINE_VARIABLE(take_fn, take)
  285. } // namespace views
  286. namespace cpp20
  287. {
  288. namespace views
  289. {
  290. using ranges::views::take;
  291. }
  292. template(typename Rng)(
  293. requires view_<Rng>)
  294. using take_view = ranges::take_view<Rng>;
  295. } // namespace cpp20
  296. /// @}
  297. } // namespace ranges
  298. #include <range/v3/detail/epilogue.hpp>
  299. #include <range/v3/detail/satisfy_boost_range.hpp>
  300. RANGES_SATISFY_BOOST_RANGE(::ranges::take_view)
  301. #endif