qcsrc/lib/stats.qh

   1 #pragma once
   2
   3 // TODO: rename to 'netvars'
   4
   5 #include "registry.qh"
   6 #include "sort.qh"
   7
   8 .int m_id;
   9 USING(vectori, vector);
  10
  11 #define REGISTER_STAT(...) EVAL_REGISTER_STAT(OVERLOAD(REGISTER_STAT, __VA_ARGS__))
  12 #define EVAL_REGISTER_STAT(...) __VA_ARGS__
  13 #if defined(CSQC)
  14         /** Get all stats and store them as globals, access with `STAT(ID)` */
  15         void stats_get() {}
  16         #define STAT(...) EVAL_STAT(OVERLOAD(STAT, __VA_ARGS__))
  17         #define EVAL_STAT(...) __VA_ARGS__
  18     #define STAT_1(id) STAT_2(id, NULL)
  19         #define STAT_2(id, cl) (0, _STAT(id))
  20
  21         #define getstat_int(id) getstati(id, 0, 24)
  22         #define getstat_bool(id) boolean(getstati(id))
  23         #define getstat_float(id) getstatf(id)
  24         #define getstat_vector(id) vec3(getstat_float(id + 0), getstat_float(id + 1), getstat_float(id + 2))
  25         #define getstat_vectori(id) vec3(getstat_int(id + 0), getstat_int(id + 1), getstat_int(id + 2))
  26
  27         #define _STAT(id) g_stat_##id
  28         #define REGISTER_STAT_2(id, T) \
  29                 T _STAT(id); \
  30                 T CAT(_STAT(id), _prev); \
  31                 REGISTER(Stats, STAT_##id, m_id, new_pure(stat)) \
  32                 { \
  33                         if (#T == "vector" || #T == "vectori") { \
  34                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, y); \
  35                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, z); \
  36                         } \
  37                 } \
  38                 [[accumulate]] void stats_get() \
  39                 { \
  40                         T it = getstat_##T(STAT_##id.m_id); \
  41                         if (it != CAT(_STAT(id), _prev)) \
  42                                 CAT(_STAT(id), _prev) = _STAT(id) = it; \
  43                 }
  44         #define REGISTER_STAT_3(x, T, expr) REGISTER_STAT_2(x, T)
  45 #elif defined(SVQC)
  46         /** Add all registered stats, access with `STAT(ID, player)` or `.type stat = _STAT(ID); player.stat` */
  47         void stats_add() {}
  48         #define STAT(id, cl) (cl)._STAT(id)
  49
  50         #define addstat_int(id, fld) addstat(id, AS_INT, fld)
  51         #define addstat_bool(id, fld) addstat(id, AS_INT, fld)
  52         #define addstat_float(id, fld) addstat(id, AS_FLOAT, fld)
  53         #define addstat_vector(id, fld) MACRO_BEGIN { \
  54                 addstat_float(id + 0, fld##_x); \
  55                 addstat_float(id + 1, fld##_y); \
  56                 addstat_float(id + 2, fld##_z); \
  57         } MACRO_END
  58         #define addstat_vectori(id, fld) MACRO_BEGIN { \
  59                 addstat_int(id + 0, fld##_x); \
  60                 addstat_int(id + 1, fld##_y); \
  61                 addstat_int(id + 2, fld##_z); \
  62         } MACRO_END
  63         const int AS_STRING = 1;
  64         const int AS_INT = 2;
  65         const int AS_FLOAT = 8;
  66
  67         .int __stat_null;
  68         /** Prevent engine stats being sent */
  69         STATIC_INIT(stats_clear)
  70         {
  71                 int r = 32;
  72                 for (int i = 0, n = 256 - r; i < n; ++i) {
  73                         addstat(r + i, AS_INT, __stat_null);
  74                 }
  75         }
  76
  77         #define _STAT(id) stat_##id
  78         #define REGISTER_STAT_2(id, T) \
  79                 .T _STAT(id); \
  80                 REGISTER(Stats, STAT_##id, m_id, new_pure(stat)) \
  81                 { \
  82                         if (#T == "vector" || #T == "vectori") { \
  83                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, y); \
  84                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, z); \
  85                         } \
  86                 } \
  87                 [[accumulate]] void stats_add() \
  88                 { \
  89                         .T fld = _STAT(id); \
  90                         addstat_##T(STAT_##id.m_id, fld); \
  91                 }
  92         void GlobalStats_update(entity this) {}
  93     #define REGISTER_STAT_3(id, T, expr) \
  94         REGISTER_STAT_2(id, T); \
  95         [[accumulate]] void GlobalStats_update(entity this) { STAT(id, this) = (expr); } \
  96         STATIC_INIT(worldstat_##id) { entity this = world; STAT(id, this) = (expr); }
  97 #else
  98         #define REGISTER_STAT_2(id, type)
  99     #define REGISTER_STAT_3(id, T, expr)
 100 #endif
 101
 102 const int STATS_ENGINE_RESERVE = 32;
 103
 104 REGISTRY(Stats, 256 - STATS_ENGINE_RESERVE)
 105 REGISTER_REGISTRY(Stats)
 106 REGISTRY_SORT(Stats)
 107 REGISTRY_CHECK(Stats)
 108 STATIC_INIT(RegisterStats_renumber)
 109 {
 110         FOREACH(Stats, true, it.m_id = STATS_ENGINE_RESERVE + i);
 111 }
 112 #ifdef SVQC
 113 STATIC_INIT(stats_add) { stats_add(); }
 114 #endif