mmmmh

arena.h

@@ -0,0 +1,65 @@
+#pragma once
+
+#include "collatypes.h"
+
+#if COLLA_TCC
+#define alignof __alignof__
+#else
+#define alignof _Alignof
+#endif
+
+typedef enum {
+    ARENA_VIRTUAL,
+    ARENA_MALLOC,
+    ARENA_STATIC,
+} arena_type_e;
+
+typedef enum {
+    ALLOC_FLAGS_NONE = 0,
+    ALLOC_NOZERO     = 1 << 0,
+    ALLOC_SOFT_FAIL  = 1 << 1,
+} alloc_flags_e;
+
+typedef struct arena_t {
+    uint8 *start;
+    uint8 *current;
+    uint8 *end;
+    arena_type_e type;
+} arena_t;
+
+typedef struct {
+    arena_type_e type;
+    usize allocation;
+    byte *static_buffer;
+} arena_desc_t;
+
+typedef struct {
+    arena_t *arena;
+    usize count;
+    alloc_flags_e flags;
+    usize align;
+    usize size;
+} arena_alloc_desc_t;
+
+#ifndef ARENA_NO_SIZE_HELPERS
+#define KB(count) (  (count) * 1024)
+#define MB(count) (KB(count) * 1024)
+#define GB(count) (MB(count) * 1024)
+#endif
+
+// arena_type_e type, usize allocation, [ byte *static_buffer ]
+#define arenaMake(...) arenaInit(&(arena_desc_t){ __VA_ARGS__ })
+
+// arena_t *arena, T type, [ usize count, alloc_flags_e flags, usize align, usize size ]
+#define alloc(arenaptr, type, ...) arenaAlloc(&(arena_alloc_desc_t){ .size = sizeof(type), .count = 1, .align = alignof(type), .arena = arenaptr, __VA_ARGS__ })
+
+arena_t arenaInit(const arena_desc_t *desc);
+void arenaCleanup(arena_t *arena);
+
+arena_t arenaScratch(arena_t *arena, usize size);
+
+void *arenaAlloc(const arena_alloc_desc_t *desc);
+usize arenaTell(arena_t *arena);
+usize arenaRemaining(arena_t *arena);
+void arenaRewind(arena_t *arena, usize from_start);
+void arenaPop(arena_t *arena, usize amount);