registry.qh

Go to the documentation of this file.

#pragma once

#include "oo.qh"

#if 1
    #define _R_MAP(r, max) ArrayList r; STATIC_INIT(r) { AL_NEW(r, max, NULL, e); }
    #define _R_GET(r, i) AL_gete(r, i)
    #define _R_SET(r, i, e) AL_sete(r, i, e)
    #define _R_DEL(r) AL_DELETE(r)
#else
    #define _R_MAP(r, max) entity r[max]
    #define _R_GET(r, i) r[i]
    #define _R_SET(r, i, e) r[i] = e
    #define _R_DEL(r)
#endif

#define REGISTRY_MAX(id) id##_MAX
#define REGISTRY_COUNT(id) id##_COUNT

#define REGISTRY(id, max) \
    void Register##id(); \
    ACCUMULATE void REGISTRY_DEPENDS_(id) {} \
    REGISTRY_BEGIN(id) {} \
    REGISTRY_END(id) {} \
    void _Register##id() {} \
    int id##_state = 0; \
    void Register##id() { if (id##_state) return; id##_state = 1; REGISTRY_DEPENDS_(id); REGISTRY_BEGIN_(id); _Register##id(); id##_state = 2; REGISTRY_END_(id); } \
    const int id##_MAX = max; \
    int id##_COUNT; \
    noref entity id##_first, id##_last; \
    _R_MAP(_##id, id##_MAX); \
    SHUTDOWN(id) { _R_DEL(_##id); } \

#define REGISTRY_DEFINE_GET(id, null) \
    entity id##_from(int i) { if (i >= 0 && i < id##_COUNT) { entity e = _R_GET(_##id, i); if (e) return e; } return null; }

#define REGISTRY_GET(id, i) id##_from(i)

#define REGISTRY_DEPENDS(id, dep) void Register##dep(); void REGISTRY_DEPENDS_(id) { Register##dep(); }
#define REGISTRY_DEPENDS_(id) Register##id##_Depends()

#define REGISTRY_BEGIN(id) ACCUMULATE void REGISTRY_BEGIN_(id) { noref void() f = Register##id; } void REGISTRY_BEGIN_(id)
#define REGISTRY_BEGIN_(id) Register##id##_First()

#define REGISTRY_END(id) ACCUMULATE void REGISTRY_END_(id) { noref void() f = Register##id; } void REGISTRY_END_(id)
#define REGISTRY_END_(id) Register##id##_Done()

REGISTRY(Registries, BITS(8))


.string registered_id;

void _regCheck(int i, int _max)
{
    // this is inside a function to avoid expanding it on compilation everytime
    // (this very long line would be repeated literally thousands times!)
    if (i >= _max)
        LOG_FATALF("Registry capacity exceeded (%d)", _max);
}

#define REGISTER(...) EVAL_REGISTER(OVERLOAD_(REGISTER, __VA_ARGS__))
#define EVAL_REGISTER(...) __VA_ARGS__
#define REGISTER_5(registry, ns, id, fld, inst) REGISTER_4(registry, ns##_##id, fld, inst)
#define REGISTER_4(registry, id, fld, inst) \
    entity id; \
    REGISTER_INIT(id) {} \
    void Register_##id() \
    { \
        entity this = id; \
        if (this == NULL) { \
            _regCheck(registry##_COUNT, registry##_MAX); \
            this = id = inst; \
            this.registered_id = #id; \
            REGISTRY_PUSH(registry, fld, this); \
        } \
        Register_##id##_init(this); \
    } \
    ACCUMULATE_FUNCTION(_Register##registry, Register_##id) \
    REGISTER_INIT(id)

#define REGISTRY_PUSH(registry, fld, it) MACRO_BEGIN \
    it.fld = registry##_COUNT; \
    _R_SET(_##registry, registry##_COUNT, it); \
    ++registry##_COUNT; \
    if (!registry##_first) registry##_first = it; \
    if (registry##_last)   registry##_last.REGISTRY_NEXT = it; \
    registry##_last = it; \
MACRO_END

#define REGISTRY_RESERVE(registry, fld, id, suffix) MACRO_BEGIN \
    entity e = new_pure(registry_reserved); \
    e.registered_id = #id "/" #suffix; \
    REGISTRY_PUSH(registry, fld, e); \
MACRO_END

#define REGISTER_INIT(id) ACCUMULATE void Register_##id##_init(entity this)

#define REGISTRY_NEXT enemy
.entity REGISTRY_NEXT;

#define REGISTRY_SORT(...) EVAL_REGISTRY_SORT(OVERLOAD(REGISTRY_SORT, __VA_ARGS__))
#define EVAL_REGISTRY_SORT(...) __VA_ARGS__
#define REGISTRY_SORT_1(id) REGISTRY_SORT_2(id, 0)
#define REGISTRY_SORT_2(id, skip) \
    void _REGISTRY_SWAP_##id(int i, int j, entity pass) \
    { \
        i += skip; j += skip; \
        \
        entity a = _R_GET(_##id, i), b = _R_GET(_##id, j); \
        _R_SET(_##id, i, b); \
        _R_SET(_##id, j, a); \
        \
        entity a_next = a.REGISTRY_NEXT, b_next = b.REGISTRY_NEXT; \
        a.REGISTRY_NEXT = b_next; \
        b.REGISTRY_NEXT = a_next; \
        \
        if (i == 0) id##_first = b; \
        else _R_GET(_##id, i - 1).REGISTRY_NEXT = b; \
        \
        if (j == 0) id##_first = a; \
        else _R_GET(_##id, j - 1).REGISTRY_NEXT = a; \
    } \
    int _REGISTRY_CMP_##id(int i, int j, entity pass) \
    { \
        i += skip; j += skip; \
        string a = _R_GET(_##id, i).registered_id; \
        string b = _R_GET(_##id, j).registered_id; \
        return strcmp(a, b); \
    } \
    STATIC_INIT(Registry_sort_##id) \
    { \
        heapsort(id##_COUNT - (skip), _REGISTRY_SWAP_##id, _REGISTRY_CMP_##id, NULL); \
    }

#define REGISTRY_HASH(id) Registry_hash_##id

ERASEABLE
ACCUMULATE void Registry_check(string r, string sv) { }
ERASEABLE
ACCUMULATE void Registry_send_all() { }

#ifdef SVQC
void Registry_send(string id, string hash);
#else
#define Registry_send(id, hash)
#endif

#define REGISTRY_CHECK(id) \
    string REGISTRY_HASH(id); \
    STATIC_INIT(Registry_check_##id) \
    { \
        /* Note: SHA256 isn't always available, use MD4 instead */ \
        string s = ""; \
        FOREACH(id, true, s = strcat(s, ":", it.registered_id)); \
        s = substring(s, 1, -1); /* remove initial ":" */ \
        string h = REGISTRY_HASH(id) = strzone(digest_hex("MD4", s)); \
        LOG_DEBUGF(#id ": %s\n[%s]", h, s); \
    } \
    void Registry_check(string r, string sv) \
    { \
        if (r == #id) \
        { \
            string cl = REGISTRY_HASH(id); \
            if (cl != sv) \
            { \
                LOG_FATALF("client/server mismatch (%s).\nCL: %s\nSV: %s", r, cl, sv); \
            } \
        } \
    } \
    void Registry_send_all() { Registry_send(#id, REGISTRY_HASH(id)); } \

#define _REGISTER_REGISTRY(id, str) \
    ACCUMULATE_FUNCTION(__static_init_1, Register##id) \
    CLASS(id##Registry, Object) \
        ATTRIB(id##Registry, m_name, string, str); \
        ATTRIB(id##Registry, REGISTRY_NEXT, entity, id##_first); \
        METHOD(id##Registry, m_reload, void()); \
    ENDCLASS(id##Registry) \
    REGISTER(Registries, REGISTRY, id, m_id, NEW(id##Registry)); \
    METHOD(id##Registry, m_reload, void()) { \
        id##_state = 0; \
        Register##id(); \
    }

#define REGISTER_REGISTRY(id) _REGISTER_REGISTRY(id, #id)