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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
|
#ifndef KIT_ASYNC_FUNCTION_H
#define KIT_ASYNC_FUNCTION_H
#include "types.h"
#include <string.h>
#ifdef __cplusplus
extern "C" {
#endif
typedef struct {
i32 _;
} kit_af_void;
typedef void (*kit_af_state_machine)(void *self_void_);
#define KIT_AF_STATE_DATA \
struct { \
i32 _index; \
i32 _id; \
kit_af_state_machine _state_machine; \
}
typedef struct {
KIT_AF_STATE_DATA;
} kit_af_type_void;
#if defined(__GNUC__) || defined(__clang__)
# pragma GCC diagnostic push
# pragma GCC diagnostic ignored "-Wunused-function"
# pragma GCC diagnostic ignored "-Wunknown-pragmas"
#endif
#define KIT_AF_INTERNAL(coro_) (*((kit_af_type_void *) (coro_)))
#ifdef KIT_ENABLE_CUSTOM_ASYNC_FUNCTION_DISPATCH
// Application should implement this function if custom async
// function dispatch is enabled.
//
// See KIT_ENABLE_CUSTOM_ASYNC_FUNCTION_DISPATCH macro.
//
void kit_async_function_dispatch(void *promise);
#else
static void kit_async_function_dispatch(void *promise) {
// Dynamic dispatch by default.
//
KIT_AF_INTERNAL(promise)._state_machine(promise);
}
#endif
#if defined(__GNUC__) || defined(__clang__)
# pragma GCC diagnostic pop
#endif
#define KIT_AF_STATE(ret_type_, name_, ...) \
struct name_##_coro_state_ { \
KIT_AF_STATE_DATA; \
ret_type_ return_value; \
__VA_ARGS__ \
}
#define KIT_AF_DECL(name_) void name_(void *self_void_)
#define KIT_CORO_IMPL(name_) \
KIT_AF_DECL(name_) { \
struct name_##_coro_state_ *self = \
(struct name_##_coro_state_ *) self_void_; \
switch (self->_index) { \
case 0:;
#define KIT_AF_LINE_() __LINE__
#define KIT_CORO_END \
} \
self->_index = -1; \
}
#define KIT_CORO_DECL(ret_type_, name_, ...) \
KIT_AF_STATE(ret_type_, name_, __VA_ARGS__); \
KIT_AF_DECL(name_)
#define KIT_CORO(ret_type_, name_, ...) \
KIT_AF_STATE(ret_type_, name_, __VA_ARGS__); \
KIT_CORO_IMPL(name_)
#define KIT_CORO_DECL_VOID(name_, ...) \
KIT_CORO_DECL(kit_af_void, name_, __VA_ARGS__)
#define KIT_CORO_VOID(name_, ...) \
KIT_CORO(kit_af_void, name_, __VA_ARGS__)
#define KIT_STATIC_CORO(ret_type_, name_, ...) \
KIT_AF_STATE(ret_type_, name_, __VA_ARGS__); \
static KIT_CORO_IMPL(name_)
#define KIT_STATIC_CORO_VOID(name_, ...) \
KIT_STATIC_CORO(kit_af_void, name_, __VA_ARGS__)
#define KIT_AF_EXECUTE(promise_) \
kit_async_function_dispatch(&(promise_))
#define KIT_AF_NEXT(promise_) \
(kit_async_function_dispatch(&(promise_)), (promise_).return_value)
#define KIT_AF_YIELD(...) \
do { \
self->_index = KIT_AF_LINE_(); \
self->return_value = __VA_ARGS__; \
return; \
case KIT_AF_LINE_():; \
} while (0)
#define KIT_AF_YIELD_VOID \
do { \
self->_index = KIT_AF_LINE_(); \
return; \
case KIT_AF_LINE_():; \
} while (0)
#define KIT_AF_RETURN(...) \
do { \
self->_index = -1; \
self->return_value = __VA_ARGS__; \
return; \
} while (0)
#define KIT_AF_RETURN_VOID \
do { \
self->_index = -1; \
return; \
} while (0)
#define KIT_AF_AWAIT(promise_) \
do { \
case KIT_AF_LINE_(): \
if ((promise_)._index != -1) { \
self->_index = KIT_AF_LINE_(); \
kit_async_function_dispatch(&(promise_)); \
} \
if ((promise_)._index != -1) \
return; \
} while (0)
#define KIT_AF_YIELD_AWAIT(promise_) \
do { \
case KIT_AF_LINE_(): \
if ((promise_)._index != -1) { \
self->_index = KIT_AF_LINE_(); \
kit_async_function_dispatch(&(promise_)); \
self->return_value = (promise_).return_value; \
return; \
} \
} while (0)
#define KIT_AF_TYPE(coro_) struct coro_##_coro_state_
#define KIT_AF_INITIAL(id_, coro_) \
._index = 0, ._id = (id_), ._state_machine = (coro_)
#define KIT_AF_CREATE(promise_, coro_, ...) \
KIT_AF_TYPE(coro_) \
promise_ = { KIT_AF_INITIAL(0, coro_), __VA_ARGS__ }
#define KIT_AF_CREATE_ID(promise_, id_, ...) \
KIT_AF_TYPE(coro_) \
promise_ = { KIT_AF_INITIAL(id_, NULL), __VA_ARGS__ }
#define KIT_AF_INIT(promise_, coro_, ...) \
do { \
KIT_AF_CREATE(kit_af_temp_, coro_, __VA_ARGS__); \
memcpy(&(promise_), &kit_af_temp_, sizeof kit_af_temp_); \
} while (0)
#define KIT_AF_INIT_ID(promise_, id_, ...) \
do { \
KIT_AF_CREATE_ID(kit_af_temp_, id_, __VA_ARGS__); \
memcpy(&(promise_), &kit_af_temp_, sizeof kit_af_temp_); \
} while (0)
#define KIT_AF_FINISHED(promise_) ((promise_)._index == -1)
#define KIT_AF_FINISHED_N(return_, promises_, size_) \
do { \
i32 kit_af_index_; \
(return_) = 1; \
for (kit_af_index_ = 0; kit_af_index_ < (size_); \
kit_af_index_++) \
if (!KIT_AF_FINISHED((promises_)[kit_af_index_])) { \
(return_) = 0; \
break; \
} \
} while (0)
#define KIT_AF_FINISHED_ALL(return_, promises_) \
KIT_AF_FINISHED_N((return_), (promises_), \
sizeof(promises_) / sizeof((promises_)[0]))
#ifdef __cplusplus
}
#endif
#ifndef KIT_DISABLE_SHORT_NAMES
# define af_void kit_af_void
# define af_state_machine kit_af_state_machine
# define af_type_void kit_af_type_void
# define AF_STATE_DATA KIT_AF_STATE_DATA
# define AF_STATE KIT_AF_STATE
# define AF_DECL KIT_AF_DECL
# define CORO_IMPL KIT_CORO_IMPL
# define CORO_END KIT_CORO_END
# define CORO_DECL KIT_CORO_DECL
# define CORO KIT_CORO
# define CORO_DECL_VOID KIT_CORO_DECL_VOID
# define STATIC_CORO KIT_STATIC_CORO
# define STATIC_CORO_VOID KIT_STATIC_CORO_VOID
# define CORO_VOID KIT_CORO_VOID
# define AF_EXECUTE KIT_AF_EXECUTE
# define AF_NEXT KIT_AF_NEXT
# define AF_YIELD KIT_AF_YIELD
# define AF_YIELD_VOID KIT_AF_YIELD_VOID
# define AF_RETURN KIT_AF_RETURN
# define AF_RETURN_VOID KIT_AF_RETURN_VOID
# define AF_AWAIT KIT_AF_AWAIT
# define AF_YIELD_AWAIT KIT_AF_YIELD_AWAIT
# define AF_TYPE KIT_AF_TYPE
# define AF_INITIAL KIT_AF_INITIAL
# define AF_CREATE KIT_AF_CREATE
# define AF_INIT KIT_AF_INIT
# define AF_FINISHED KIT_AF_FINISHED
# define AF_FINISHED_N KIT_AF_FINISHED_N
# define AF_FINISHED_ALL KIT_AF_FINISHED_ALL
#endif
#endif
|
