ranges_find.h 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. //===----------------------------------------------------------------------===//
  2. //
  3. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  4. // See https://llvm.org/LICENSE.txt for license information.
  5. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. #ifndef _LIBCPP___ALGORITHM_RANGES_FIND_H
  9. #define _LIBCPP___ALGORITHM_RANGES_FIND_H
  10. #include <__algorithm/find.h>
  11. #include <__algorithm/ranges_find_if.h>
  12. #include <__algorithm/unwrap_range.h>
  13. #include <__config>
  14. #include <__functional/identity.h>
  15. #include <__functional/invoke.h>
  16. #include <__functional/ranges_operations.h>
  17. #include <__iterator/concepts.h>
  18. #include <__iterator/projected.h>
  19. #include <__ranges/access.h>
  20. #include <__ranges/concepts.h>
  21. #include <__ranges/dangling.h>
  22. #include <__utility/forward.h>
  23. #include <__utility/move.h>
  24. #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
  25. # pragma GCC system_header
  26. #endif
  27. _LIBCPP_PUSH_MACROS
  28. #include <__undef_macros>
  29. #if _LIBCPP_STD_VER >= 20
  30. _LIBCPP_BEGIN_NAMESPACE_STD
  31. namespace ranges {
  32. namespace __find {
  33. struct __fn {
  34. template <class _Iter, class _Sent, class _Tp, class _Proj>
  35. _LIBCPP_HIDE_FROM_ABI static constexpr _Iter
  36. __find_unwrap(_Iter __first, _Sent __last, const _Tp& __value, _Proj& __proj) {
  37. if constexpr (forward_iterator<_Iter>) {
  38. auto [__first_un, __last_un] = std::__unwrap_range(__first, std::move(__last));
  39. return std::__rewrap_range<_Sent>(
  40. std::move(__first), std::__find_impl(std::move(__first_un), std::move(__last_un), __value, __proj));
  41. } else {
  42. return std::__find_impl(std::move(__first), std::move(__last), __value, __proj);
  43. }
  44. }
  45. template <input_iterator _Ip, sentinel_for<_Ip> _Sp, class _Tp, class _Proj = identity>
  46. requires indirect_binary_predicate<ranges::equal_to, projected<_Ip, _Proj>, const _Tp*>
  47. _LIBCPP_NODISCARD_EXT _LIBCPP_HIDE_FROM_ABI constexpr _Ip
  48. operator()(_Ip __first, _Sp __last, const _Tp& __value, _Proj __proj = {}) const {
  49. return __find_unwrap(std::move(__first), std::move(__last), __value, __proj);
  50. }
  51. template <input_range _Rp, class _Tp, class _Proj = identity>
  52. requires indirect_binary_predicate<ranges::equal_to, projected<iterator_t<_Rp>, _Proj>, const _Tp*>
  53. _LIBCPP_NODISCARD_EXT _LIBCPP_HIDE_FROM_ABI constexpr borrowed_iterator_t<_Rp>
  54. operator()(_Rp&& __r, const _Tp& __value, _Proj __proj = {}) const {
  55. return __find_unwrap(ranges::begin(__r), ranges::end(__r), __value, __proj);
  56. }
  57. };
  58. } // namespace __find
  59. inline namespace __cpo {
  60. inline constexpr auto find = __find::__fn{};
  61. } // namespace __cpo
  62. } // namespace ranges
  63. _LIBCPP_END_NAMESPACE_STD
  64. #endif // _LIBCPP_STD_VER >= 20
  65. _LIBCPP_POP_MACROS
  66. #endif // _LIBCPP___ALGORITHM_RANGES_FIND_H