source: doc/papers/concurrency/examples/Format.cpp @ aa22c60

#include "base.hpp"

struct fmt_cor {
        struct promise_type {
                char _value;
                int g, b;

                fmt_cor get_return_object() {
                        return fmt_cor(std::experimental::coroutine_handle<promise_type>::from_promise(*this));
                }

                auto initial_suspend() { return suspend_never(); }
                auto final_suspend()   { return suspend_always(); }

                void return_void() {}
                void unhandled_exception() {}
        };

        struct get {
                promise_type * _promise = nullptr;

                bool await_ready() noexcept {
                        return false;
                }

                void await_suspend(std::experimental::coroutine_handle<promise_type> _coroutine) noexcept {
                        _promise = &_coroutine.promise();
                }
                char await_resume() noexcept {
                        assert(_promise);
                        return _promise->_value;
                }
        };

        std::experimental::coroutine_handle<promise_type> _coroutine = nullptr;

        explicit fmt_cor(std::experimental::coroutine_handle<promise_type> coroutine)
                : _coroutine(coroutine)
        {}

        ~fmt_cor() {
                if(_coroutine) { _coroutine.destroy(); }
        }

        fmt_cor() = default;
        fmt_cor(fmt_cor const &) = delete;
        fmt_cor& operator=(fmt_cor const &) = delete;

        fmt_cor(fmt_cor&& other) {
                std::swap(_coroutine, other._coroutine);
        }

        fmt_cor& operator=(fmt_cor&& other) {
                if(&other != this) {
                        _coroutine = other._coroutine;
                        other._coroutine = nullptr;
                }
                return *this;
        }

        void send(char a) {
                _coroutine.promise()._value = a;
                _coroutine.resume();
        }
};

fmt_cor Fmt() {
        int g; // = co_await fmt_cor::g();
        int b; // = co_await fmt_cor::b();
        for(;;) {
                for(g = 0; g < 5; g++) {
                        for(b = 0; b < 4; b++) {
                                std::cout << co_await fmt_cor::get();
                        }
                        std::cout << "  ";
                }
                std::cout << std::endl;
        }
}

int main() {
        auto fmt = Fmt();
        for(int i = 0; i < 41; i++) {
                fmt.send('a');
        }
}