summaryrefslogtreecommitdiffhomepage
path: root/misc/examples/coroutines/cointerleave.c
blob: 804941763a8c17ef06c2b46b94ca072bc1c75bfd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
// https://www.youtube.com/watch?v=8sEe-4tig_A
#include <stdio.h>
#include <stc/coroutine.h>
#define i_type IVec
#define i_key int
#include <stc/cvec.h>

struct GenValue {
    IVec *v;
    IVec_iter it;
    int cco_state;
};

static int get_value(struct GenValue* g)
{
    cco_routine(g) {
        for (g->it = IVec_begin(g->v); g->it.ref; IVec_next(&g->it))
            cco_yield_v(*g->it.ref);
    }
    return -1;
}

struct Generator {
    struct GenValue x, y;
    int cco_state;
    int value;
};

cco_result interleaved(struct Generator* g) 
{
    cco_routine(g) {
        while (!(cco_done(&g->x) & cco_done(&g->y))) {
            g->value = get_value(&g->x);
            if (!cco_done(&g->x))
                cco_yield();

            g->value = get_value(&g->y);
            if (!cco_done(&g->y))
                cco_yield();
        }
    }
    return CCO_DONE;
}

void Use(void)
{
    IVec a = c_init(IVec, {2, 4, 6, 8, 10, 11});
    IVec b = c_init(IVec, {3, 5, 7, 9});

    struct Generator g = {{&a}, {&b}};

    cco_blocking_call(interleaved(&g)) {
        printf("%d ", g.value);
    }
    puts("");
    c_drop(IVec, &a, &b);
}

int main(void)
{
    Use();
}