size.h 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. // -*- C++ -*-
  2. //===----------------------------------------------------------------------===//
  3. //
  4. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  5. // See https://llvm.org/LICENSE.txt for license information.
  6. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  7. //
  8. //===----------------------------------------------------------------------===//
  9. #ifndef _LIBCPP___RANGES_SIZE_H
  10. #define _LIBCPP___RANGES_SIZE_H
  11. #include <__concepts/class_or_enum.h>
  12. #include <__config>
  13. #include <__iterator/concepts.h>
  14. #include <__iterator/iterator_traits.h>
  15. #include <__ranges/access.h>
  16. #include <__utility/auto_cast.h>
  17. #include <concepts>
  18. #include <type_traits>
  19. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  20. # pragma GCC system_header
  21. #endif
  22. _LIBCPP_BEGIN_NAMESPACE_STD
  23. #if !defined(_LIBCPP_HAS_NO_CONCEPTS) && !defined(_LIBCPP_HAS_NO_INCOMPLETE_RANGES)
  24. namespace ranges {
  25. template<class>
  26. inline constexpr bool disable_sized_range = false;
  27. } // namespace ranges
  28. // [range.prim.size]
  29. namespace ranges {
  30. namespace __size {
  31. void size(auto&) = delete;
  32. void size(const auto&) = delete;
  33. template <class _Tp>
  34. concept __size_enabled = !disable_sized_range<remove_cvref_t<_Tp>>;
  35. template <class _Tp>
  36. concept __member_size =
  37. __size_enabled<_Tp> &&
  38. __workaround_52970<_Tp> &&
  39. requires(_Tp&& __t) {
  40. { _LIBCPP_AUTO_CAST(__t.size()) } -> __integer_like;
  41. };
  42. template <class _Tp>
  43. concept __unqualified_size =
  44. __size_enabled<_Tp> &&
  45. !__member_size<_Tp> &&
  46. __class_or_enum<remove_cvref_t<_Tp>> &&
  47. requires(_Tp&& __t) {
  48. { _LIBCPP_AUTO_CAST(size(__t)) } -> __integer_like;
  49. };
  50. template <class _Tp>
  51. concept __difference =
  52. !__member_size<_Tp> &&
  53. !__unqualified_size<_Tp> &&
  54. __class_or_enum<remove_cvref_t<_Tp>> &&
  55. requires(_Tp&& __t) {
  56. { ranges::begin(__t) } -> forward_iterator;
  57. { ranges::end(__t) } -> sized_sentinel_for<decltype(ranges::begin(declval<_Tp>()))>;
  58. };
  59. struct __fn {
  60. template <class _Tp, size_t _Sz>
  61. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr size_t operator()(_Tp (&&)[_Sz]) const noexcept {
  62. return _Sz;
  63. }
  64. template <class _Tp, size_t _Sz>
  65. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr size_t operator()(_Tp (&)[_Sz]) const noexcept {
  66. return _Sz;
  67. }
  68. template <__member_size _Tp>
  69. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr __integer_like auto operator()(_Tp&& __t) const
  70. noexcept(noexcept(_LIBCPP_AUTO_CAST(__t.size()))) {
  71. return _LIBCPP_AUTO_CAST(__t.size());
  72. }
  73. template <__unqualified_size _Tp>
  74. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr __integer_like auto operator()(_Tp&& __t) const
  75. noexcept(noexcept(_LIBCPP_AUTO_CAST(size(__t)))) {
  76. return _LIBCPP_AUTO_CAST(size(__t));
  77. }
  78. template<__difference _Tp>
  79. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr __integer_like auto operator()(_Tp&& __t) const
  80. noexcept(noexcept(ranges::end(__t) - ranges::begin(__t))) {
  81. return std::__to_unsigned_like(ranges::end(__t) - ranges::begin(__t));
  82. }
  83. };
  84. } // namespace __size
  85. inline namespace __cpo {
  86. inline constexpr auto size = __size::__fn{};
  87. } // namespace __cpo
  88. } // namespace ranges
  89. // [range.prim.ssize]
  90. namespace ranges {
  91. namespace __ssize {
  92. struct __fn {
  93. template<class _Tp>
  94. requires requires (_Tp&& __t) { ranges::size(__t); }
  95. [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr integral auto operator()(_Tp&& __t) const
  96. noexcept(noexcept(ranges::size(__t)))
  97. {
  98. using _Signed = make_signed_t<decltype(ranges::size(__t))>;
  99. if constexpr (sizeof(ptrdiff_t) > sizeof(_Signed))
  100. return static_cast<ptrdiff_t>(ranges::size(__t));
  101. else
  102. return static_cast<_Signed>(ranges::size(__t));
  103. }
  104. };
  105. } // namespace __ssize
  106. inline namespace __cpo {
  107. inline constexpr auto ssize = __ssize::__fn{};
  108. } // namespace __cpo
  109. } // namespace ranges
  110. #endif // !defined(_LIBCPP_HAS_NO_CONCEPTS) && !defined(_LIBCPP_HAS_NO_INCOMPLETE_RANGES)
  111. _LIBCPP_END_NAMESPACE_STD
  112. #endif // _LIBCPP___RANGES_SIZE_H