83 lines
1.7 KiB
C++
83 lines
1.7 KiB
C++
/** @example coroutine2.cc
|
|
*
|
|
* An example of using coroutines as generators as well as nested resume/yield.
|
|
*/
|
|
|
|
#include <ostd/io.hh>
|
|
#include <ostd/coroutine.hh>
|
|
|
|
using namespace ostd;
|
|
|
|
int main() {
|
|
auto f = [](auto yield) {
|
|
for (int i: range(5, 26, 5)) {
|
|
yield(i);
|
|
}
|
|
};
|
|
|
|
writeln("generator test");
|
|
for (int i: generator<int>{f}) {
|
|
writefln("generated: %s", i);
|
|
}
|
|
|
|
generator<int> x{f};
|
|
/* coroutine_context exists as a base type for any coroutine */
|
|
coroutine_context *ctx = &x;
|
|
|
|
writefln(
|
|
"generator is coroutine<int()>: %s",
|
|
bool(dynamic_cast<coroutine<int()> *>(ctx))
|
|
);
|
|
writefln(
|
|
"generator is generator<int>: %s",
|
|
bool(dynamic_cast<generator<int> *>(ctx))
|
|
);
|
|
|
|
generator<int> &gr = dynamic_cast<generator<int> &>(*ctx);
|
|
writefln("value of cast back generator: %s", gr.value());
|
|
|
|
writeln("-- nested coroutine test --");
|
|
|
|
coroutine<void()> c1 = [](auto yield) {
|
|
coroutine<void()> c2 = [&yield](auto) {
|
|
writeln("inside c2 1");
|
|
yield();
|
|
writeln("inside c2 2");
|
|
yield();
|
|
writeln("inside c2 3");
|
|
};
|
|
writeln("inside c1 1");
|
|
c2();
|
|
writeln("inside c1 2");
|
|
};
|
|
writeln("outside 1");
|
|
c1();
|
|
writeln("outside 2");
|
|
c1();
|
|
writeln("outside 3");
|
|
c1();
|
|
writeln("outside exit");
|
|
}
|
|
|
|
/*
|
|
generator test
|
|
generated: 5
|
|
generated: 10
|
|
generated: 15
|
|
generated: 20
|
|
generated: 25
|
|
generator is coroutine<int()>: false
|
|
generator is generator<int>: true
|
|
value of cast back generator: 5
|
|
-- nested coroutine test --
|
|
outside 1
|
|
inside c1 1
|
|
inside c2 1
|
|
outside 2
|
|
inside c2 2
|
|
outside 3
|
|
inside c2 3
|
|
inside c1 2
|
|
outside exit
|
|
*/
|