drop.hpp 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  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_DROP_HPP
  14. #define RANGES_V3_VIEW_DROP_HPP
  15. #include <type_traits>
  16. #include <meta/meta.hpp>
  17. #include <range/v3/range_fwd.hpp>
  18. #include <range/v3/algorithm/min.hpp>
  19. #include <range/v3/functional/bind_back.hpp>
  20. #include <range/v3/iterator/operations.hpp>
  21. #include <range/v3/iterator/traits.hpp>
  22. #include <range/v3/range/concepts.hpp>
  23. #include <range/v3/range/traits.hpp>
  24. #include <range/v3/utility/box.hpp>
  25. #include <range/v3/utility/optional.hpp>
  26. #include <range/v3/utility/static_const.hpp>
  27. #include <range/v3/view/all.hpp>
  28. #include <range/v3/view/interface.hpp>
  29. #include <range/v3/view/subrange.hpp>
  30. #include <range/v3/view/view.hpp>
  31. #include <range/v3/detail/prologue.hpp>
  32. namespace ranges
  33. {
  34. /// \addtogroup group-views
  35. /// @{
  36. template<typename Rng>
  37. struct RANGES_EMPTY_BASES drop_view
  38. : view_interface<drop_view<Rng>,
  39. is_finite<Rng>::value ? finite : range_cardinality<Rng>::value>
  40. , private detail::non_propagating_cache<iterator_t<Rng>, drop_view<Rng>,
  41. !random_access_range<Rng>>
  42. {
  43. private:
  44. using difference_type_ = range_difference_t<Rng>;
  45. Rng rng_;
  46. difference_type_ n_;
  47. template(bool Const = true)(
  48. requires Const AND range<meta::const_if_c<Const, Rng>>)
  49. iterator_t<meta::const_if_c<Const, Rng>> //
  50. get_begin_(std::true_type, std::true_type) const
  51. {
  52. CPP_assert(random_access_range<meta::const_if_c<Const, Rng>>);
  53. return next(ranges::begin(rng_), n_, ranges::end(rng_));
  54. }
  55. iterator_t<Rng> get_begin_(std::true_type, std::false_type)
  56. {
  57. CPP_assert(random_access_range<Rng>);
  58. return next(ranges::begin(rng_), n_, ranges::end(rng_));
  59. }
  60. iterator_t<Rng> get_begin_(std::false_type, detail::ignore_t)
  61. {
  62. CPP_assert(!random_access_range<Rng>);
  63. using cache_t =
  64. detail::non_propagating_cache<iterator_t<Rng>, drop_view<Rng>>;
  65. auto & begin_ = static_cast<cache_t &>(*this);
  66. if(!begin_)
  67. begin_ = next(ranges::begin(rng_), n_, ranges::end(rng_));
  68. return *begin_;
  69. }
  70. public:
  71. drop_view() = default;
  72. drop_view(Rng rng, difference_type_ n)
  73. : rng_(std::move(rng))
  74. , n_(n)
  75. {
  76. RANGES_EXPECT(n >= 0);
  77. }
  78. iterator_t<Rng> begin()
  79. {
  80. return this->get_begin_(meta::bool_<random_access_range<Rng>>{},
  81. std::false_type{});
  82. }
  83. sentinel_t<Rng> end()
  84. {
  85. return ranges::end(rng_);
  86. }
  87. template(bool Const = true)(
  88. requires Const AND random_access_range<meta::const_if_c<Const, Rng>>)
  89. iterator_t<meta::const_if_c<Const, Rng>> begin() const
  90. {
  91. return this->get_begin_(std::true_type{}, std::true_type{});
  92. }
  93. template(bool Const = true)(
  94. requires Const AND random_access_range<meta::const_if_c<Const, Rng>>)
  95. sentinel_t<meta::const_if_c<Const, Rng>> end() const
  96. {
  97. return ranges::end(rng_);
  98. }
  99. CPP_auto_member
  100. auto CPP_fun(size)()(const //
  101. requires sized_range<Rng const>)
  102. {
  103. auto const s = ranges::size(rng_);
  104. auto const n = static_cast<range_size_t<Rng const>>(n_);
  105. return s < n ? 0 : s - n;
  106. }
  107. CPP_auto_member
  108. auto CPP_fun(size)()(
  109. requires sized_range<Rng>)
  110. {
  111. auto const s = ranges::size(rng_);
  112. auto const n = static_cast<range_size_t<Rng>>(n_);
  113. return s < n ? 0 : s - n;
  114. }
  115. Rng base() const
  116. {
  117. return rng_;
  118. }
  119. };
  120. template<typename Rng>
  121. RANGES_INLINE_VAR constexpr bool enable_borrowed_range<drop_view<Rng>> = //
  122. enable_borrowed_range<Rng>;
  123. #if RANGES_CXX_DEDUCTION_GUIDES >= RANGES_CXX_DEDUCTION_GUIDES_17
  124. template<typename Rng>
  125. drop_view(Rng &&, range_difference_t<Rng>)
  126. -> drop_view<views::all_t<Rng>>;
  127. #endif
  128. namespace views
  129. {
  130. struct drop_base_fn
  131. {
  132. private:
  133. template<typename Rng>
  134. static auto impl_(Rng && rng, range_difference_t<Rng> n, input_range_tag)
  135. -> drop_view<all_t<Rng>>
  136. {
  137. return {all(static_cast<Rng &&>(rng)), n};
  138. }
  139. template(typename Rng)(
  140. requires borrowed_range<Rng> AND sized_range<Rng>)
  141. static subrange<iterator_t<Rng>, sentinel_t<Rng>> //
  142. impl_(Rng && rng, range_difference_t<Rng> n, random_access_range_tag)
  143. {
  144. return {begin(rng) + ranges::min(n, distance(rng)), end(rng)};
  145. }
  146. public:
  147. template(typename Rng)(
  148. requires viewable_range<Rng> AND input_range<Rng>)
  149. auto operator()(Rng && rng, range_difference_t<Rng> n) const
  150. {
  151. return drop_base_fn::impl_(
  152. static_cast<Rng &&>(rng), n, range_tag_of<Rng>{});
  153. }
  154. };
  155. struct drop_fn : drop_base_fn
  156. {
  157. using drop_base_fn::operator();
  158. template(typename Int)(
  159. requires detail::integer_like_<Int>)
  160. constexpr auto operator()(Int n) const
  161. {
  162. return make_view_closure(bind_back(drop_base_fn{}, n));
  163. }
  164. };
  165. /// \relates drop_fn
  166. /// \ingroup group-views
  167. RANGES_INLINE_VARIABLE(drop_fn, drop)
  168. } // namespace views
  169. namespace cpp20
  170. {
  171. namespace views
  172. {
  173. using ranges::views::drop;
  174. }
  175. template(typename Rng)(
  176. requires view_<Rng>)
  177. using drop_view = ranges::drop_view<Rng>;
  178. } // namespace cpp20
  179. /// @}
  180. } // namespace ranges
  181. #include <range/v3/detail/epilogue.hpp>
  182. #include <range/v3/detail/satisfy_boost_range.hpp>
  183. RANGES_SATISFY_BOOST_RANGE(::ranges::drop_view)
  184. #endif