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 const int STATS_ENGINE_RESERVE = 32;
  12 // must be listed in ascending order
  13 #define MAGIC_STATS(_, x) \
  14         _(x, MOVEVARS_TIMESCALE, 241) \
  15         /**/
  16
  17 int g_magic_stats_hole = 0;
  18
  19 #define MAGIC_STATS_FIX_MANUAL(it, var, id) \
  20         if (it.registered_id == "STAT_" #var) { --g_magic_stats_hole; it.m_id = id; } else
  21
  22 #define MAGIC_STATS_FIX_AUTO(it, var, id) \
  23         if (it.m_id == id) { ++g_magic_stats_hole; ++it.m_id; }
  24
  25 #define MAGIC_STATS_FIX(it) \
  26         it.m_id += g_magic_stats_hole; \
  27         MAGIC_STATS(MAGIC_STATS_FIX_MANUAL, it) { MAGIC_STATS(MAGIC_STATS_FIX_AUTO, it) }
  28
  29 #define REGISTER_STAT(...) EVAL_REGISTER_STAT(OVERLOAD(REGISTER_STAT, __VA_ARGS__))
  30 #define EVAL_REGISTER_STAT(...) __VA_ARGS__
  31 #if defined(CSQC)
  32         /** Get all stats and store them as globals, access with `STAT(ID)` */
  33         void stats_get() {}
  34         #define STAT(...) EVAL_STAT(OVERLOAD(STAT, __VA_ARGS__))
  35         #define EVAL_STAT(...) __VA_ARGS__
  36     #define STAT_1(id) (RVALUE, _STAT(id))
  37         #define STAT_2(id, cl) STAT_1(id)
  38
  39         #define getstat_int(id) getstati(id, 0, 24)
  40         #define getstat_bool(id) boolean(getstati(id))
  41         #define getstat_float(id) getstatf(id)
  42         #define getstat_vector(id) vec3(getstat_float(id + 0), getstat_float(id + 1), getstat_float(id + 2))
  43         #define getstat_vectori(id) vec3(getstat_int(id + 0), getstat_int(id + 1), getstat_int(id + 2))
  44
  45         #define _STAT(id) g_stat_##id
  46         #define REGISTER_STAT_2(id, T) \
  47                 T _STAT(id); \
  48                 /* T CAT(_STAT(id), _prev); */ \
  49                 REGISTER(Stats, STAT_##id, m_id, new_pure(stat)) \
  50                 { \
  51                         if (#T == "vector" || #T == "vectori") { \
  52                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, y); \
  53                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, z); \
  54                         } \
  55                 } \
  56                 ACCUMULATE void stats_get() \
  57                 { \
  58                         T it = getstat_##T(STAT_##id.m_id); \
  59                         /* if (it != CAT(_STAT(id), _prev)) \
  60                                 CAT(_STAT(id), _prev) = */ _STAT(id) = it; \
  61                 }
  62         #define REGISTER_STAT_3(x, T, expr) REGISTER_STAT_2(x, T)
  63 #elif defined(SVQC)
  64     /** Internal use only */
  65     entity STATS;
  66         /** Add all registered stats, access with `STAT(ID, player)` or `.type stat = _STAT(ID); player.stat` */
  67         void stats_add() {}
  68         #define STAT(...) EVAL_STAT(OVERLOAD_(STAT, __VA_ARGS__))
  69     #define EVAL_STAT(...) __VA_ARGS__
  70     #define STAT_1(id) (RVALUE, STAT_2(id, STATS))
  71         #define STAT_2(id, cl) (cl)._STAT(id)
  72
  73         #define addstat_int(id, fld) addstat(id, AS_INT, fld)
  74         #define addstat_bool(id, fld) addstat(id, AS_INT, fld)
  75         #define addstat_float(id, fld) addstat(id, AS_FLOAT, fld)
  76         #define addstat_vector(id, fld) MACRO_BEGIN { \
  77                 addstat_float(id + 0, fld##_x); \
  78                 addstat_float(id + 1, fld##_y); \
  79                 addstat_float(id + 2, fld##_z); \
  80         } MACRO_END
  81         #define addstat_vectori(id, fld) MACRO_BEGIN { \
  82                 addstat_int(id + 0, fld##_x); \
  83                 addstat_int(id + 1, fld##_y); \
  84                 addstat_int(id + 2, fld##_z); \
  85         } MACRO_END
  86         const int AS_STRING = 1;
  87         const int AS_INT = 2;
  88         const int AS_FLOAT = 8;
  89
  90         .int __stat_null;
  91         STATIC_INIT(stats)
  92         {
  93             STATS = new(stats);
  94             // Prevent engine stats being sent
  95                 int r = STATS_ENGINE_RESERVE;
  96                 for (int i = 0, n = 256 - r; i < n; ++i) {
  97                         #define X(_, name, id) if (i == id) continue;
  98                         MAGIC_STATS(X, );
  99                         #undef X
 100                         addstat(r + i, AS_INT, __stat_null);
 101                 }
 102         }
 103
 104         #define _STAT(id) stat_##id
 105         #define REGISTER_STAT_2(id, T) \
 106                 .T _STAT(id); \
 107                 REGISTER(Stats, STAT_##id, m_id, new_pure(stat)) \
 108                 { \
 109                         if (#T == "vector" || #T == "vectori") { \
 110                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, y); \
 111                                 REGISTRY_RESERVE(Stats, m_id, STAT_##id, z); \
 112                         } \
 113                 } \
 114                 ACCUMULATE void stats_add() \
 115                 { \
 116                         .T fld = _STAT(id); \
 117                         addstat_##T(STAT_##id.m_id, fld); \
 118                 }
 119         void GlobalStats_update(entity this) {}
 120     /** TODO: do we want the global copy to update? */
 121     #define REGISTER_STAT_3(id, T, expr) \
 122         REGISTER_STAT_2(id, T); \
 123         ACCUMULATE void GlobalStats_update(entity this) { STAT(id, this) = (expr); } \
 124         STATIC_INIT(worldstat_##id) { entity this = STATS; STAT(id, this) = (expr); }
 125 #else
 126         #define REGISTER_STAT_2(id, type)
 127     #define REGISTER_STAT_3(id, T, expr)
 128 #endif
 129
 130 REGISTRY(Stats, 256 - STATS_ENGINE_RESERVE)
 131 REGISTER_REGISTRY(Stats)
 132 REGISTRY_SORT(Stats)
 133 REGISTRY_CHECK(Stats)
 134 STATIC_INIT(RegisterStats_renumber)
 135 {
 136         FOREACH(Stats, true, {
 137                 it.m_id = STATS_ENGINE_RESERVE + i;
 138                 MAGIC_STATS_FIX(it);
 139         });
 140 }
 141 #ifdef SVQC
 142 STATIC_INIT(stats_add) { stats_add(); }
 143 #endif