]> de.git.xonotic.org Git - xonotic/xonotic-data.pk3dir.git/blob - qcsrc/lib/iter.qh
Merge branch 'master' into Mirio/balance
[xonotic/xonotic-data.pk3dir.git] / qcsrc / lib / iter.qh
1 #pragma once
2
3 #if 1
4 #define ITER_CONST const
5 #else
6 #define ITER_CONST
7 #endif
8
9 #define FOREACH_ARRAY(arr, start, end, cond, body) \
10         MACRO_BEGIN \
11         { \
12                 for (int _i = start; _i < end; ++_i) \
13                 { \
14                         const noref int i = _i; \
15                         ITER_CONST noref entity it = arr[i]; \
16                         if (cond) { LAMBDA(body) } \
17                 } \
18         } MACRO_END
19
20 #define FOREACH(list, cond, body) FOREACH_LIST(list, enemy, cond, body)
21
22 #define FOREACH_LIST(list, next, cond, body) \
23         MACRO_BEGIN \
24         { \
25                 int _i = 0; \
26                 for (entity _it = list##_first; _it; (_it = _it.next, ++_i)) \
27                 { \
28                         const noref int i = _i; \
29                         ITER_CONST noref entity it = _it; \
30                         if (cond) { LAMBDA(body) } \
31                 } \
32         } MACRO_END
33
34 #define FOREACH_WORD(words, cond, body) \
35         MACRO_BEGIN \
36         { \
37                 string _words = words; \
38                 int _i = 0; \
39                 for (string _it; (_it = car(_words)); (_words = cdr(_words), ++_i)) \
40                 { \
41                         const noref int i = _i; \
42                         const noref string it = _it; \
43                         if (cond) { LAMBDA(body) } \
44                 } \
45         } MACRO_END
46
47 #define STRING_ITERATOR(this, s, i) \
48         string this##_s = s; \
49         int this##_i = i
50
51 #define STRING_ITERATOR_SET(this, s, i) \
52         MACRO_BEGIN { \
53                 this##_s = s; \
54                 this##_i = i; \
55         } MACRO_END
56
57 #define STRING_ITERATOR_GET(this) str2chr(this##_s, this##_i++)
58 #define STRING_ITERATOR_PEEK(this) str2chr(this##_s, this##_i)
59 #define STRING_ITERATOR_NEXT(this) MACRO_BEGIN ++this##_i; MACRO_END
60 #define STRING_ITERATOR_UNGET(this) MACRO_BEGIN --this##_i; MACRO_END
61 #define STRING_ITERATOR_SAVE(this) this##_i
62 #define STRING_ITERATOR_LOAD(this, n) MACRO_BEGIN this##_i = n; MACRO_END
63
64 #define FOREACH_CHAR(s, cond, body) \
65         MACRO_BEGIN \
66         { \
67                 STRING_ITERATOR(iter, s, 0); \
68                 int _it; \
69                 while ((_it = STRING_ITERATOR_GET(iter)) > 0) \
70                 { \
71                         const noref int it = _it; \
72                         if (cond) { LAMBDA(body) } \
73                 } \
74         } MACRO_END
75
76 #if defined(CSQC)
77     entity(entity start, .string fld, string match) _findstring = #18;
78         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #402;
79
80     entity(entity start, .entity fld, entity match) _findentity = #98;
81         entity(.entity fld, entity match, .entity tofield) _findchainentity_tofield = #403;
82
83     entity(entity start, .float fld, float match) _findfloat = #98;
84         entity(.float fld, float match, .entity tofield) _findchainfloat_tofield = #403;
85
86         entity(entity start, .float fld, float match) _findflags = #449;
87         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #450;
88 #elif defined(SVQC)
89     entity(entity start, .string fld, string match) _findstring = #18;
90         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #402;
91
92     entity(entity start, .entity fld, entity match) _findentity = #98;
93         entity(.entity fld, entity match, .entity tofield) _findchainentity_tofield = #403;
94
95     entity(entity start, .float fld, float match) _findfloat = #98;
96         entity(.float fld, float match, .entity tofield) _findchainfloat_tofield = #403;
97
98         entity(entity start, .float fld, float match) _findflags = #449;
99         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #450;
100 #elif defined(MENUQC)
101     entity(entity start, .string fld, string match) _findstring = #24;
102         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #26;
103
104     entity(entity start, .entity fld, entity match) _findentity = #25;
105         entity(.entity fld, entity match, .entity tofield) _findchainentity_tofield = #27;
106
107     entity(entity start, .float fld, float match) _findfloat = #25;
108         entity(.float fld, float match, .entity tofield) _findchainfloat_tofield = #27;
109
110         entity(entity start, .float fld, float match) _findflags = #87;
111         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #88;
112 #endif
113
114 #define ORDERED(F) F##_UNORDERED
115 #define _FOREACH_ENTITY_FIND_ORDERED(T, fld, match, cond, body) \
116     MACRO_BEGIN { \
117         int _i = 0; \
118         for (entity _it = NULL; (_it = _find##T(_it, fld, match)); ++_i) \
119         { \
120             const noref int i = _i; \
121             ITER_CONST noref entity it = _it; \
122             if (cond) LAMBDA(body) \
123         } \
124     } MACRO_END
125 #define _FOREACH_ENTITY_FIND_UNORDERED(id, T, fld, match, cond, body) \
126         MACRO_BEGIN { \
127                 if (_FOREACH_ENTITY_FIND_##T##_##id##mutex) LOG_SEVEREF("Loop mutex held by %s", _FOREACH_ENTITY_FIND_##T##_##id##mutex); \
128                 _FOREACH_ENTITY_FIND_##T##_##id##mutex = __FUNC__; \
129                 entity _foundchain_first = _findchain##T##_tofield(fld, match, _FOREACH_ENTITY_FIND_##T##_next##id); \
130                 FOREACH_LIST(_foundchain, _FOREACH_ENTITY_FIND_##T##_next##id, cond, body); \
131                 _FOREACH_ENTITY_FIND_##T##_##id##mutex = string_null; \
132         } MACRO_END
133
134 #define FOREACH_ENTITY(cond, body) ORDERED(FOREACH_ENTITY)(cond, body)
135 #define FOREACH_ENTITY_ORDERED(cond, body) \
136         MACRO_BEGIN { \
137                 int _i = 0; \
138                 for (entity _it = NULL; (_it = nextent(_it)); ++_i) \
139                 { \
140                         const noref int i = _i; \
141                         ITER_CONST noref entity it = _it; \
142                         if (cond) LAMBDA(body) \
143                 } \
144         } MACRO_END
145 /** marker field, always NULL */
146 .entity _FOREACH_ENTITY_fld;
147 .entity _FOREACH_ENTITY_FIND_entity_nextall; noref string _FOREACH_ENTITY_FIND_entity_allmutex;
148 #define FOREACH_ENTITY_UNORDERED(cond, body) _FOREACH_ENTITY_FIND_UNORDERED(all, entity, _FOREACH_ENTITY_fld, NULL, cond, body)
149
150 #define FOREACH_ENTITY_FLAGS(fld, match, body) ORDERED(FOREACH_ENTITY_FLAGS)(fld, match, body)
151 #define FOREACH_ENTITY_FLAGS_ORDERED(fld, match, body) _FOREACH_ENTITY_FIND_ORDERED(flags, fld, match, true, body)
152 .entity _FOREACH_ENTITY_FIND_flags_next; noref string _FOREACH_ENTITY_FIND_flags_mutex;
153 #define FOREACH_ENTITY_FLAGS_UNORDERED(fld, match, body) _FOREACH_ENTITY_FIND_UNORDERED(, flags, fld, match, true, body)
154
155 #ifndef MENUQC
156 entity(vector org, float rad, .entity tofield) _findchainradius_tofield = #22;
157 #define FOREACH_ENTITY_RADIUS(org, dist, cond, body) FOREACH_ENTITY_RADIUS_UNORDERED(org, dist, cond, body)
158 .entity _FOREACH_ENTITY_FIND_radius_next; noref string _FOREACH_ENTITY_FIND_radius_mutex;
159 #define FOREACH_ENTITY_RADIUS_UNORDERED(org, dist, cond, body) _FOREACH_ENTITY_FIND_UNORDERED(, radius, org, dist, cond, body)
160 #endif
161
162 #define FOREACH_ENTITY_FLOAT(fld, match, body) ORDERED(FOREACH_ENTITY_FLOAT)(fld, match, body)
163 #define FOREACH_ENTITY_FLOAT_ORDERED(fld, match, body) _FOREACH_ENTITY_FIND_ORDERED(float, fld, match, true, body)
164 .entity _FOREACH_ENTITY_FIND_float_next; noref string _FOREACH_ENTITY_FIND_float_mutex;
165 #define FOREACH_ENTITY_FLOAT_UNORDERED(fld, match, body) _FOREACH_ENTITY_FIND_UNORDERED(, float, fld, match, true, body)
166
167 #define FOREACH_ENTITY_ENT(fld, match, body) ORDERED(FOREACH_ENTITY_ENT)(fld, match, body)
168 #define FOREACH_ENTITY_ENT_ORDERED(fld, match, body) _FOREACH_ENTITY_FIND_ORDERED(entity, fld, match, true, body)
169 .entity _FOREACH_ENTITY_FIND_entity_next; noref string _FOREACH_ENTITY_FIND_entity_mutex;
170 #define FOREACH_ENTITY_ENT_UNORDERED(fld, match, body) _FOREACH_ENTITY_FIND_UNORDERED(, entity, fld, match, true, body)
171
172 #define FOREACH_ENTITY_STRING(fld, match, body) ORDERED(FOREACH_ENTITY_STRING)(fld, match, body)
173 #define FOREACH_ENTITY_STRING_ORDERED(fld, match, body) _FOREACH_ENTITY_FIND_ORDERED(string, fld, match, true, body)
174 .entity _FOREACH_ENTITY_FIND_string_next; noref string _FOREACH_ENTITY_FIND_string_mutex;
175 #define FOREACH_ENTITY_STRING_UNORDERED(fld, match, body) _FOREACH_ENTITY_FIND_UNORDERED(, string, fld, match, true, body)
176
177 #define FOREACH_ENTITY_CLASS(class, cond, body) ORDERED(FOREACH_ENTITY_CLASS)(class, cond, body)
178 #define FOREACH_ENTITY_CLASS_ORDERED(class, cond, body) _FOREACH_ENTITY_FIND_ORDERED(string, classname, class, cond, body)
179 .entity _FOREACH_ENTITY_FIND_string_nextclazz; noref string _FOREACH_ENTITY_FIND_string_clazzmutex;
180 #define FOREACH_ENTITY_CLASS_UNORDERED(class, cond, body) _FOREACH_ENTITY_FIND_UNORDERED(clazz, string, classname, class, cond, body)