| 1 | #include <algorithm> | 
|---|
| 2 | #include <iostream> | 
|---|
| 3 | #include <iterator> | 
|---|
| 4 | #include <vector> | 
|---|
| 5 |  | 
|---|
| 6 | #include <experimental/coroutine> | 
|---|
| 7 |  | 
|---|
| 8 | template<typename T> | 
|---|
| 9 | struct cor_range { | 
|---|
| 10 | struct promise_type { | 
|---|
| 11 | T _value; | 
|---|
| 12 |  | 
|---|
| 13 | cor_range get_return_object() { | 
|---|
| 14 | return cor_range(std::experimental::coroutine_handle<promise_type>::from_promise(*this)); | 
|---|
| 15 | } | 
|---|
| 16 |  | 
|---|
| 17 | auto initial_suspend() { return std::experimental::suspend_always(); } | 
|---|
| 18 | auto final_suspend()   { return std::experimental::suspend_always(); } | 
|---|
| 19 |  | 
|---|
| 20 | void return_value(T value) { | 
|---|
| 21 | _value = value; | 
|---|
| 22 | } | 
|---|
| 23 |  | 
|---|
| 24 | auto yield_value(T value) { | 
|---|
| 25 | _value = value; | 
|---|
| 26 | return std::experimental::suspend_always(); | 
|---|
| 27 | } | 
|---|
| 28 |  | 
|---|
| 29 | void unhandled_exception() {} | 
|---|
| 30 | }; | 
|---|
| 31 |  | 
|---|
| 32 | std::experimental::coroutine_handle<promise_type> _coroutine = nullptr; | 
|---|
| 33 |  | 
|---|
| 34 | explicit cor_range(std::experimental::coroutine_handle<promise_type> coroutine) | 
|---|
| 35 | : _coroutine(coroutine) | 
|---|
| 36 | {} | 
|---|
| 37 |  | 
|---|
| 38 | ~cor_range() { | 
|---|
| 39 | if(_coroutine) { _coroutine.destroy(); } | 
|---|
| 40 | } | 
|---|
| 41 |  | 
|---|
| 42 | cor_range() = default; | 
|---|
| 43 | cor_range(cor_range const &) = delete; | 
|---|
| 44 | cor_range& operator=(cor_range const &) = delete; | 
|---|
| 45 |  | 
|---|
| 46 | cor_range(cor_range&& other) { | 
|---|
| 47 | std::swap(_coroutine, other._coroutine); | 
|---|
| 48 | } | 
|---|
| 49 |  | 
|---|
| 50 | cor_range& operator=(cor_range&& other) { | 
|---|
| 51 | if(&other != this) { | 
|---|
| 52 | _coroutine = other._coroutine; | 
|---|
| 53 | other._coroutine = nullptr; | 
|---|
| 54 | } | 
|---|
| 55 | return *this; | 
|---|
| 56 | } | 
|---|
| 57 |  | 
|---|
| 58 | T next() { | 
|---|
| 59 | _coroutine.resume(); | 
|---|
| 60 | return _coroutine.promise()._value; | 
|---|
| 61 | } | 
|---|
| 62 |  | 
|---|
| 63 | struct iterator : std::iterator<std::input_iterator_tag, T> { | 
|---|
| 64 | std::experimental::coroutine_handle<promise_type> _coroutine = nullptr; | 
|---|
| 65 |  | 
|---|
| 66 | iterator() = default; | 
|---|
| 67 | explicit iterator(std::experimental::coroutine_handle<promise_type> coroutine) | 
|---|
| 68 | : _coroutine(coroutine) | 
|---|
| 69 | {} | 
|---|
| 70 |  | 
|---|
| 71 | iterator& operator++() { | 
|---|
| 72 | _coroutine.resume(); | 
|---|
| 73 | return *this; | 
|---|
| 74 | } | 
|---|
| 75 |  | 
|---|
| 76 | T const & operator*() const { | 
|---|
| 77 | return _coroutine.promise()._value; | 
|---|
| 78 | } | 
|---|
| 79 | }; | 
|---|
| 80 |  | 
|---|
| 81 | iterator begin() { | 
|---|
| 82 | if(_coroutine) { | 
|---|
| 83 | _coroutine.resume(); | 
|---|
| 84 | if(_coroutine.done()) { return end(); } | 
|---|
| 85 | } | 
|---|
| 86 |  | 
|---|
| 87 | return iterator{ _coroutine }; | 
|---|
| 88 | } | 
|---|
| 89 |  | 
|---|
| 90 | iterator end() { return iterator{}; } | 
|---|
| 91 | }; | 
|---|
| 92 |  | 
|---|
| 93 | cor_range<int> fib() { | 
|---|
| 94 | int fn1 = 0, fn = 1; | 
|---|
| 95 | for(;;) { | 
|---|
| 96 | co_yield fn1; | 
|---|
| 97 | int temp = fn1; | 
|---|
| 98 | fn1 = fn; | 
|---|
| 99 | fn = temp + fn; | 
|---|
| 100 | } | 
|---|
| 101 | } | 
|---|
| 102 |  | 
|---|
| 103 | int main() { | 
|---|
| 104 | { | 
|---|
| 105 | auto f1 = fib(); | 
|---|
| 106 | auto f2 = fib(); | 
|---|
| 107 | for(int i = 0; i < 10; i++) { | 
|---|
| 108 | std::cout << f1.next() << " " << f2.next() << std::endl; | 
|---|
| 109 | } | 
|---|
| 110 | } | 
|---|
| 111 |  | 
|---|
| 112 | { | 
|---|
| 113 | auto f1 = fib(); | 
|---|
| 114 | std::vector<int> fibs; | 
|---|
| 115 | std::copy_n(f1.begin(), 10, std::back_inserter(fibs)); | 
|---|
| 116 |  | 
|---|
| 117 | for(auto i : fibs) { | 
|---|
| 118 | std::cout << i << std::endl; | 
|---|
| 119 | } | 
|---|
| 120 | } | 
|---|
| 121 | } | 
|---|