xiaoyang-sde fa79e0a400
[libc++][ranges] implement ranges::elements_of (#91414)
## Introduction

This patch implements `ranges::elements_of` from
[P2502R2](https://wg21.link/P2502R2). Specializations of `elements_of`
encapsulate a range and act as a tag in overload sets to disambiguate
when a range should be treated as a sequence rather than a single value.

```cpp
template <bool YieldElements>
std::generator<std::any> f(std::ranges::input_range auto &&r) {
  if constexpr (YieldElements) {
    co_yield std::ranges::elements_of(r);
  } else {
    co_yield r;
  }
}
```

## Reference

- [P2502R2: `std::generator`: Synchronous Coroutine Generator for
Ranges](https://wg21.link/P2502R2)
- [[range.elementsof]](https://eel.is/c++draft/range.elementsof)

Partially addresses #105226

---------

Co-authored-by: Louis Dionne <ldionne.2@gmail.com>
Co-authored-by: A. Jiang <de34@live.cn>
2025-12-13 19:44:45 +08:00

101 lines
2.8 KiB
C++

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++03, c++11, c++14, c++17, c++20
// <ranges>
//
// template<range R, class Allocator = allocator<byte>>
// struct std::ranges::elements_of;
//
// template<class R, class Allocator = allocator<byte>>
// elements_of(R&&, Allocator = Allocator()) -> elements_of<R&&, Allocator>;
#include <concepts>
#include <cstddef>
#include <memory>
#include <ranges>
#include <utility>
#include "min_allocator.h"
#include "test_allocator.h"
#include "test_iterators.h"
template <class Allocator, class Range>
constexpr void test_impl() {
Allocator a;
// With a lvalue range
{
Range r;
std::ranges::elements_of elements(r);
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&, std::allocator<std::byte>>>);
}
// With a rvalue range
{
Range r;
std::ranges::elements_of elements(std::move(r));
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&&, std::allocator<std::byte>>>);
}
// With lvalue range and allocator
{
Range r;
std::ranges::elements_of elements(r, a);
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&, Allocator>>);
}
// With rvalue range and allocator
{
Range r;
std::ranges::elements_of elements(std::move(r), Allocator());
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&&, Allocator>>);
}
// Ensure we can use designated initializers
{
// lvalues
{
Range r;
std::ranges::elements_of elements{.range = r, .allocator = a};
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&, Allocator>>);
}
// rvalues
{
Range r;
std::ranges::elements_of elements{.range = std::move(r), .allocator = Allocator()};
static_assert(std::same_as<decltype(elements), std::ranges::elements_of<Range&&, Allocator>>);
}
}
}
template <class Iterator>
struct Range {
Iterator begin() const;
sentinel_wrapper<Iterator> end() const;
};
constexpr bool test() {
types::for_each(types::type_list<std::allocator<std::byte>, min_allocator<std::byte>, test_allocator<std::byte>>{},
[]<class Allocator> {
types::for_each(types::cpp20_input_iterator_list<int*>{}, []<class Iterator> {
test_impl<Allocator, Range<Iterator>>();
});
});
return true;
}
int main(int, char**) {
test();
static_assert(test());
return 0;
}