| [17c6c1c3] | 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 | } | 
|---|