qcsrc/lib/iter.qh

   1 #ifndef ITER_H
   2 #define ITER_H
   3
   4 #define FOREACH_ARRAY(arr, start, end, cond, body) \
   5         MACRO_BEGIN \
   6         { \
   7                 for (int i = start; i < end; ++i) \
   8                 { \
   9                         const noref entity it = arr[i]; \
  10                         if (cond) { LAMBDA(body) } \
  11                 } \
  12         } MACRO_END
  13
  14 #define FOREACH_LIST(list, next, cond, body) \
  15         MACRO_BEGIN \
  16         { \
  17                 int i = 0; \
  18                 for (entity it = list##_first; it; (it = it.next, ++i)) \
  19                 { \
  20                         if (cond) { LAMBDA(body) } \
  21                 } \
  22         } MACRO_END
  23
  24 #define FOREACH_WORD(words, cond, body) \
  25         MACRO_BEGIN \
  26         { \
  27                 string _words = words; \
  28                 int i = 0; \
  29                 for (string _it; (_it = car(_words)); (_words = cdr(_words), ++i)) \
  30                 { \
  31                         const noref string it = _it; \
  32                         if (cond) { LAMBDA(body) } \
  33                 } \
  34         } MACRO_END
  35
  36 #define STRING_ITERATOR(this, s, i) \
  37         string this##_s = s; \
  38         int this##_i = i
  39
  40 #define STRING_ITERATOR_SET(this, s, i) \
  41         MACRO_BEGIN { \
  42                 this##_s = s; \
  43                 this##_i = i; \
  44         } MACRO_END
  45
  46 #define STRING_ITERATOR_GET(this) str2chr(this##_s, this##_i++)
  47
  48 #define FOREACH_CHAR(s, cond, body) \
  49         MACRO_BEGIN \
  50         { \
  51                 STRING_ITERATOR(iter, s, 0); \
  52                 int it; \
  53                 while ((it = STRING_ITERATOR_GET(iter)) > 0) \
  54                 { \
  55                         if (cond) { LAMBDA(body) } \
  56                 } \
  57         } MACRO_END
  58
  59 #if defined(CSQC)
  60         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #402;
  61         entity(.entity fld, entity match, .entity tofield) findchainentity_tofield = #403;
  62         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #450;
  63 #elif defined(SVQC)
  64         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #402;
  65         entity(.entity fld, entity match, .entity tofield) findchainentity_tofield = #403;
  66         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #450;
  67 #elif defined(MENUQC)
  68         entity(.string fld, string match, .entity tofield) _findchainstring_tofield = #26;
  69         entity(.entity fld, entity match, .entity tofield) findchainentity_tofield = #27;
  70         entity(.float fld, float match, .entity tofield) _findchainflags_tofield = #88;
  71 #endif
  72
  73 .entity _FOREACH_ENTITY_fld;
  74 .entity _FOREACH_ENTITY_next;
  75
  76 #define FOREACH_ENTITY_UNORDERED(cond, body) \
  77         MACRO_BEGIN { \
  78                 int i = 0; \
  79                 for (entity it = findchainentity_tofield(_FOREACH_ENTITY_fld, NULL, _FOREACH_ENTITY_next); it; (it = it._FOREACH_ENTITY_next, ++i)) \
  80                 { \
  81                         if (cond) { LAMBDA(body) } \
  82                 } \
  83         } MACRO_END
  84
  85 #define FOREACH_ENTITY_ORDERED(cond, body) \
  86         MACRO_BEGIN { \
  87                 int i = 0; \
  88                 for (entity it = NULL; (it = nextent(it)); ++i) \
  89                 { \
  90                         if (cond) { LAMBDA(body) } \
  91                 } \
  92         } MACRO_END
  93
  94 #define FOREACH_ENTITY_FLAGS(fld, flags, body) \
  95         MACRO_BEGIN { \
  96                 int i = 0; \
  97                 for (entity it = _findchainflags_tofield(fld, flags, _FOREACH_ENTITY_next); it; (it = it._FOREACH_ENTITY_next, ++i)) \
  98                 { \
  99                         LAMBDA(body) \
 100                 } \
 101         } MACRO_END
 102
 103 #define FOREACH_ENTITY_CLASS(class, cond, body) \
 104         MACRO_BEGIN { \
 105                 int i = 0; \
 106                 for (entity it = _findchainstring_tofield(classname, class, _FOREACH_ENTITY_next); it; (it = it._FOREACH_ENTITY_next, ++i)) \
 107                 { \
 108                         if (cond) { LAMBDA(body) } \
 109                 } \
 110         } MACRO_END
 111
 112 #define FOREACH_ENTITY(cond, body) FOREACH_ENTITY_UNORDERED(cond, body)
 113
 114 #define FOREACH(list, cond, body) FOREACH_LIST(list, enemy, cond, body)
 115
 116 #endif