#pragma once
#include "base_core.h"


typedef U32 ArenaFlags; // no flags defined yets

typedef struct ArenaParams ArenaParams;
struct ArenaParams {
    U32 size;
    ArenaFlags flags; // No flags defined yet
};

typedef struct Arena Arena;
struct Arena {
    Arena *current;
    U8 *base;
    U32 size;
    U32 pos;          // start of next allocation
    U32 base_pos;     // absolute position of base relative to the first linked arena
    ArenaFlags flags;
    Arena *prev;
    S32 checkpoint_level;
};

typedef struct ArenaCheckPoint ArenaCheckPoint;
struct ArenaCheckPoint {
    Arena *arena;
    U32 absolute_pos;
};

StaticAssert(sizeof(Arena) <= 64);
#define ARENA_HEADER_SZ 64u


static INLINE U32 arena_available(const Arena *arena)
{
    return arena->size - arena->pos;
}

Arena *arena_alloc(const ArenaParams *params);
void arena_free(Arena *arena);
void *arena_push(Arena *arena, U32 size, U32 align);

#define arena_push_type(arena, type) arena_push(arena, (U32)sizeof(type), alignof(type))
#define arena_push_array_of(arena, type, len) arena_push(arena, (U32) (sizeof(type) * (len)), alignof(type))

ArenaCheckPoint arena_temp_begin(Arena *arena);
void arena_temp_end(ArenaCheckPoint *checkpoint);
void arena_pop_to(Arena *arena, U32 abs_pos);
void arena_reset(Arena *arena);