generate.hpp 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. /// \file
  2. // Range v3 library
  3. //
  4. // Copyright Eric Niebler 2014-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_GENERATE_HPP
  14. #define RANGES_V3_VIEW_GENERATE_HPP
  15. #include <type_traits>
  16. #include <utility>
  17. #include <meta/meta.hpp>
  18. #include <range/v3/range_fwd.hpp>
  19. #include <range/v3/functional/invoke.hpp>
  20. #include <range/v3/iterator/unreachable_sentinel.hpp>
  21. #include <range/v3/range/access.hpp>
  22. #include <range/v3/range/primitives.hpp>
  23. #include <range/v3/range/traits.hpp>
  24. #include <range/v3/utility/optional.hpp>
  25. #include <range/v3/utility/semiregular_box.hpp>
  26. #include <range/v3/utility/static_const.hpp>
  27. #include <range/v3/view/facade.hpp>
  28. #include <range/v3/detail/prologue.hpp>
  29. namespace ranges
  30. {
  31. /// \addtogroup group-views
  32. /// @{
  33. template<typename G>
  34. struct generate_view : view_facade<generate_view<G>, infinite>
  35. {
  36. private:
  37. friend range_access;
  38. using result_t = invoke_result_t<G &>;
  39. semiregular_box_t<G> gen_;
  40. detail::non_propagating_cache<result_t> val_;
  41. struct cursor
  42. {
  43. private:
  44. generate_view * view_;
  45. public:
  46. cursor() = default;
  47. explicit cursor(generate_view * view)
  48. : view_(view)
  49. {}
  50. result_t && read() const
  51. {
  52. if(!view_->val_)
  53. view_->val_.emplace(view_->gen_());
  54. return static_cast<result_t &&>(static_cast<result_t &>(*view_->val_));
  55. }
  56. void next()
  57. {
  58. if(view_->val_)
  59. view_->val_.reset();
  60. else
  61. static_cast<void>(view_->gen_());
  62. }
  63. };
  64. cursor begin_cursor()
  65. {
  66. return cursor{this};
  67. }
  68. unreachable_sentinel_t end_cursor() const
  69. {
  70. return {};
  71. }
  72. public:
  73. generate_view() = default;
  74. explicit generate_view(G g)
  75. : gen_(std::move(g))
  76. {}
  77. result_t & cached()
  78. {
  79. return *val_;
  80. }
  81. };
  82. namespace views
  83. {
  84. struct generate_fn
  85. {
  86. template(typename G)(
  87. requires invocable<G &> AND copy_constructible<G> AND
  88. std::is_object<detail::decay_t<invoke_result_t<G &>>>::value AND
  89. constructible_from<detail::decay_t<invoke_result_t<G &>>,
  90. invoke_result_t<G &>> AND
  91. assignable_from<detail::decay_t<invoke_result_t<G &>> &,
  92. invoke_result_t<G &>>)
  93. generate_view<G> operator()(G g) const
  94. {
  95. return generate_view<G>{std::move(g)};
  96. }
  97. };
  98. /// \relates generate_fn
  99. /// \ingroup group-views
  100. RANGES_INLINE_VARIABLE(generate_fn, generate)
  101. } // namespace views
  102. /// \@}
  103. } // namespace ranges
  104. #include <range/v3/detail/epilogue.hpp>
  105. #include <range/v3/detail/satisfy_boost_range.hpp>
  106. RANGES_SATISFY_BOOST_RANGE(::ranges::generate_view)
  107. #endif