reference_wrapper.hpp 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184
  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_FUNCTIONAL_REFERENCE_WRAPPER_HPP
  14. #define RANGES_V3_FUNCTIONAL_REFERENCE_WRAPPER_HPP
  15. #include <type_traits>
  16. #include <utility>
  17. #include <meta/meta.hpp>
  18. #include <concepts/concepts.hpp>
  19. #include <range/v3/functional/invoke.hpp>
  20. #include <range/v3/utility/addressof.hpp>
  21. #include <range/v3/utility/static_const.hpp>
  22. #include <range/v3/detail/prologue.hpp>
  23. namespace ranges
  24. {
  25. /// \addtogroup group-functional
  26. /// @{
  27. /// \cond
  28. namespace detail
  29. {
  30. template<typename T>
  31. struct reference_wrapper_
  32. {
  33. T * t_ = nullptr;
  34. constexpr reference_wrapper_() = default;
  35. constexpr reference_wrapper_(T & t) noexcept
  36. : t_(detail::addressof(t))
  37. {}
  38. constexpr reference_wrapper_(T &&) = delete;
  39. constexpr T & get() const noexcept
  40. {
  41. return *t_;
  42. }
  43. };
  44. template<typename T>
  45. struct reference_wrapper_<T &> : reference_wrapper_<T>
  46. {
  47. using reference_wrapper_<T>::reference_wrapper_;
  48. };
  49. template<typename T>
  50. struct reference_wrapper_<T &&>
  51. {
  52. T * t_ = nullptr;
  53. constexpr reference_wrapper_() = default;
  54. constexpr reference_wrapper_(T && t) noexcept
  55. : t_(detail::addressof(t))
  56. {}
  57. constexpr T && get() const noexcept
  58. {
  59. return static_cast<T &&>(*t_);
  60. }
  61. };
  62. } // namespace detail
  63. /// \endcond
  64. // Can be used to store rvalue references in addition to lvalue references.
  65. // Also, see: https://wg21.link/lwg2993
  66. template<typename T>
  67. struct reference_wrapper : private detail::reference_wrapper_<T>
  68. {
  69. private:
  70. using base_ = detail::reference_wrapper_<T>;
  71. using base_::t_;
  72. public:
  73. using type = meta::_t<std::remove_reference<T>>;
  74. using reference = meta::if_<std::is_reference<T>, T, T &>;
  75. constexpr reference_wrapper() = default;
  76. template(typename U)(
  77. requires (!same_as<uncvref_t<U>, reference_wrapper>) AND
  78. constructible_from<base_, U>)
  79. constexpr reference_wrapper(U && u) noexcept(
  80. std::is_nothrow_constructible<base_, U>::value)
  81. : detail::reference_wrapper_<T>{static_cast<U &&>(u)}
  82. {}
  83. constexpr reference get() const noexcept
  84. {
  85. return this->base_::get();
  86. }
  87. constexpr operator reference() const noexcept
  88. {
  89. return get();
  90. }
  91. template(typename...)(
  92. requires (!std::is_rvalue_reference<T>::value)) //
  93. constexpr operator std::reference_wrapper<type>() const noexcept
  94. {
  95. return {get()};
  96. }
  97. // clang-format off
  98. template<typename ...Args>
  99. constexpr auto CPP_auto_fun(operator())(Args &&...args) (const)
  100. (
  101. return invoke(static_cast<reference>(*t_), static_cast<Args &&>(args)...)
  102. )
  103. // clang-format on
  104. };
  105. struct ref_fn
  106. {
  107. template(typename T)(
  108. requires (!is_reference_wrapper_v<T>)) //
  109. constexpr reference_wrapper<T> operator()(T & t) const
  110. {
  111. return {t};
  112. }
  113. /// \overload
  114. template<typename T>
  115. constexpr reference_wrapper<T> operator()(reference_wrapper<T> t) const
  116. {
  117. return t;
  118. }
  119. /// \overload
  120. template<typename T>
  121. constexpr reference_wrapper<T> operator()(std::reference_wrapper<T> t) const
  122. {
  123. return {t.get()};
  124. }
  125. };
  126. /// \ingroup group-functional
  127. /// \sa `ref_fn`
  128. RANGES_INLINE_VARIABLE(ref_fn, ref)
  129. template<typename T>
  130. using ref_t = decltype(ref(std::declval<T>()));
  131. struct unwrap_reference_fn
  132. {
  133. template<typename T>
  134. constexpr T && operator()(T && t) const noexcept
  135. {
  136. return static_cast<T &&>(t);
  137. }
  138. /// \overload
  139. template<typename T>
  140. constexpr typename reference_wrapper<T>::reference operator()(reference_wrapper<T> t) const
  141. noexcept
  142. {
  143. return t.get();
  144. }
  145. /// \overload
  146. template<typename T>
  147. constexpr T & operator()(std::reference_wrapper<T> t) const noexcept
  148. {
  149. return t.get();
  150. }
  151. /// \overload
  152. template<typename T>
  153. constexpr T & operator()(ref_view<T> t) const noexcept
  154. {
  155. return t.base();
  156. }
  157. };
  158. /// \ingroup group-functional
  159. /// \sa `unwrap_reference_fn`
  160. RANGES_INLINE_VARIABLE(unwrap_reference_fn, unwrap_reference)
  161. template<typename T>
  162. using unwrap_reference_t = decltype(unwrap_reference(std::declval<T>()));
  163. /// @}
  164. } // namespace ranges
  165. #include <range/v3/detail/epilogue.hpp>
  166. #endif