blob: 56d2edcf301552b940d5de2ff0466f17cd1326b8 [file] [log] [blame]
// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fsyntax-only -verify -std=c++20 -I%S/Inputs -Wno-unused -Wno-uninitialized -Wunsequenced %s
// expected-no-diagnostics
#include "std-coroutine.h"
typedef __PTRDIFF_TYPE__ ptrdiff_t;
using namespace std;
template<class T>
struct Task {
struct promise_type {
Task<T> get_return_object() noexcept;
suspend_always initial_suspend() noexcept;
suspend_always final_suspend() noexcept;
void return_value(T);
void unhandled_exception();
auto yield_value(Task<T>) noexcept { return final_suspend(); }
};
bool await_ready() noexcept { return false; }
void await_suspend(coroutine_handle<>) noexcept {}
T await_resume();
};
template<>
struct Task<void> {
struct promise_type {
Task<void> get_return_object() noexcept;
suspend_always initial_suspend() noexcept;
suspend_always final_suspend() noexcept;
void return_void() noexcept;
void unhandled_exception() noexcept;
auto yield_value(Task<void>) noexcept { return final_suspend(); }
};
bool await_ready() noexcept { return false; }
void await_suspend(coroutine_handle<>) noexcept {}
void await_resume() noexcept {}
};
template <typename T>
class generator
{
struct Promise
{
auto get_return_object() { return generator{*this}; }
auto initial_suspend() { return suspend_never{}; }
auto final_suspend() noexcept { return suspend_always{}; }
void unhandled_exception() {}
void return_void() {}
auto yield_value(T value)
{
value_ = std::move(value);
return suspend_always{};
}
T value_;
};
using Handle = coroutine_handle<Promise>;
struct sentinel{};
struct iterator
{
using iterator_category = input_iterator_tag;
using value_type = T;
using difference_type = ptrdiff_t;
using reference = T &;
using const_reference = const T &;
using pointer = T *;
iterator &operator++()
{
h_.resume();
return *this;
}
const_reference &operator*() const { return h_.promise().value_; }
bool operator!=(sentinel) { return !h_.done(); }
Handle h_;
};
explicit generator(Promise &p) : h_(Handle::from_promise(p)) {}
Handle h_;
public:
using promise_type = Promise;
auto begin() { return iterator{h_}; }
auto end() { return sentinel{}; }
};
Task<void> c(int i) {
co_await (i = 0, std::suspend_always{});
}
generator<int> range(int start, int end)
{
while (start < end)
co_yield start++;
}
Task<int> go(int const& val);
Task<int> go1(int x) {
co_return co_await go(++x);
}