slice.hpp 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  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_ACTION_SLICE_HPP
  14. #define RANGES_V3_ACTION_SLICE_HPP
  15. #include <meta/meta.hpp>
  16. #include <range/v3/range_fwd.hpp>
  17. #include <range/v3/action/action.hpp>
  18. #include <range/v3/action/erase.hpp>
  19. #include <range/v3/functional/bind_back.hpp>
  20. #include <range/v3/iterator/concepts.hpp>
  21. #include <range/v3/iterator/operations.hpp>
  22. #include <range/v3/iterator/traits.hpp>
  23. #include <range/v3/utility/static_const.hpp>
  24. #include <range/v3/view/interface.hpp>
  25. #include <range/v3/detail/prologue.hpp>
  26. namespace ranges
  27. {
  28. /// \addtogroup group-actions
  29. /// @{
  30. namespace actions
  31. {
  32. struct slice_fn
  33. {
  34. private:
  35. template<typename D>
  36. using diff_t = range_difference_t<D>;
  37. public:
  38. // Overloads for the pipe syntax: rng | actions::slice(from, to)
  39. template(typename D)(
  40. requires integral<D>)
  41. constexpr auto operator()(D from, D to) const
  42. {
  43. return make_action_closure(bind_back(slice_fn{}, from, to));
  44. }
  45. template(typename D)(
  46. requires integral<D>)
  47. constexpr auto operator()(D from, detail::from_end_<D> to) const
  48. {
  49. return make_action_closure(bind_back(slice_fn{}, from, to));
  50. }
  51. template(typename D)(
  52. requires integral<D>)
  53. constexpr auto operator()(detail::from_end_<D> from, detail::from_end_<D> to)
  54. const
  55. {
  56. return make_action_closure(bind_back(slice_fn{}, from, to));
  57. }
  58. template(typename D)(
  59. requires integral<D>)
  60. constexpr auto operator()(D from, end_fn const & to) const
  61. {
  62. return make_action_closure(bind_back(slice_fn{}, from, to));
  63. }
  64. template(typename D)(
  65. requires integral<D>)
  66. constexpr auto operator()(detail::from_end_<D> from, end_fn const & to) const
  67. {
  68. return make_action_closure(bind_back(slice_fn{}, from, to));
  69. }
  70. template(typename Rng, typename I = iterator_t<Rng>)(
  71. requires forward_range<Rng> AND erasable_range<Rng &, I, I>)
  72. Rng operator()(Rng && rng, diff_t<Rng> from, diff_t<Rng> to) const
  73. {
  74. RANGES_EXPECT(0 <= from && 0 <= to && from <= to);
  75. RANGES_EXPECT(!sized_range<Rng> || to <= distance(rng));
  76. ranges::actions::erase(rng, begin(rng), next(begin(rng), from));
  77. ranges::actions::erase(rng, next(begin(rng), to - from), end(rng));
  78. return static_cast<Rng &&>(rng);
  79. }
  80. template(typename Rng, typename I = iterator_t<Rng>)(
  81. requires bidirectional_range<Rng> AND erasable_range<Rng &, I, I>)
  82. Rng operator()(Rng && rng,
  83. diff_t<Rng> from,
  84. detail::from_end_<diff_t<Rng>> to) const
  85. {
  86. RANGES_EXPECT(0 <= from && to.dist_ <= 0);
  87. RANGES_EXPECT(!sized_range<Rng> || from - to.dist_ <= distance(rng));
  88. ranges::actions::erase(rng, begin(rng), next(begin(rng), from));
  89. if(to.dist_ != 0)
  90. {
  91. auto const last = next(begin(rng), end(rng));
  92. ranges::actions::erase(rng, prev(last, -to.dist_), last);
  93. }
  94. return static_cast<Rng &&>(rng);
  95. }
  96. template(typename Rng, typename I = iterator_t<Rng>)(
  97. requires bidirectional_range<Rng> AND erasable_range<Rng &, I, I>)
  98. Rng operator()(Rng && rng,
  99. detail::from_end_<diff_t<Rng>> from,
  100. detail::from_end_<diff_t<Rng>> to) const
  101. {
  102. RANGES_EXPECT(from.dist_ <= 0 && to.dist_ <= 0 && from.dist_ <= to.dist_);
  103. RANGES_EXPECT(!sized_range<Rng> || 0 <= distance(rng) + from.dist_);
  104. auto last = next(begin(rng), end(rng));
  105. ranges::actions::erase(rng, prev(last, -to.dist_), last);
  106. last = next(begin(rng), end(rng));
  107. ranges::actions::erase(
  108. rng, begin(rng), prev(last, to.dist_ - from.dist_));
  109. return static_cast<Rng &&>(rng);
  110. }
  111. template(typename Rng, typename I = iterator_t<Rng>)(
  112. requires forward_range<Rng> AND erasable_range<Rng &, I, I>)
  113. Rng operator()(Rng && rng, diff_t<Rng> from, end_fn const &) const
  114. {
  115. RANGES_EXPECT(0 <= from);
  116. RANGES_EXPECT(!sized_range<Rng> || from <= distance(rng));
  117. ranges::actions::erase(rng, begin(rng), next(begin(rng), from));
  118. return static_cast<Rng &&>(rng);
  119. }
  120. template(typename Rng, typename I = iterator_t<Rng>)(
  121. requires bidirectional_range<Rng> AND erasable_range<Rng &, I, I>)
  122. Rng operator()(Rng && rng,
  123. detail::from_end_<diff_t<Rng>> from,
  124. end_fn const &) const
  125. {
  126. RANGES_EXPECT(from.dist_ <= 0);
  127. RANGES_EXPECT(!sized_range<Rng> || 0 <= distance(rng) + from.dist_);
  128. auto const last = next(begin(rng), end(rng));
  129. ranges::actions::erase(rng, begin(rng), prev(last, -from.dist_));
  130. return static_cast<Rng &&>(rng);
  131. }
  132. };
  133. /// \relates actions::slice_fn
  134. RANGES_INLINE_VARIABLE(slice_fn, slice)
  135. } // namespace actions
  136. /// @}
  137. } // namespace ranges
  138. #include <range/v3/detail/epilogue.hpp>
  139. #endif