Arena/include/lae_arena.h

#ifndef LAE_ARENA_H
#define LAE_ARENA_H

#include "lae_allocator.h"
#include <inttypes.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdlib.h>

typedef struct Arena Arena;

typedef enum {
    ARENA_OK = 0,
    ARENA_BAD_ALLOC,
    ARENA_OUT_OF_SPACE,
    ARENA_NULL_ARG,
    ARENA_INVALID_SIZE,
    ARENA_INVALID_ALIGN,
    ARENA_CAPACITY_OVERFLOW,
    ARENA_LOCKED,
} ArenaErr;

typedef struct {
    bool is_valid;
    union {
        ArenaErr err;
        struct {
            Arena *arena;
            size_t offset;
        };
    };
} ArenaPointer;

typedef struct {
    bool is_valid;

    union {
        ArenaErr err;
        Allocator allocator;
    };
} ArenaAllocator;

typedef struct {
    bool is_valid;
    union {
        ArenaErr err;
        size_t val;
    };
} ArenaSizeResult;

ArenaErr arena_init(Arena **arena, Allocator allocator, size_t capacity);
ArenaErr arena_destroy(Arena **arena);

ArenaPointer arena_alloc(Arena *arena, size_t size, size_t alignment);
ArenaPointer
arena_push_data(Arena *arena, void *data, size_t size, size_t alignment);
ArenaErr arena_resize(Arena *arena, size_t new_capacity);
void *arena_unwrap_pointer(ArenaPointer p);

ArenaSizeResult arena_get_align_padding(Arena *arena, size_t alignment);
// can take big sizes but of sourse there is a limit,
// its protected against SIZE_MAX but anything that
// is big may break it, still, is an edge case that
// I shouldn't cover.
ArenaErr arena_ensure_capacity(Arena *arena, size_t size, size_t alignment);

ArenaErr arena_lock_growth(Arena *arena);
ArenaErr arena_unlock_growth(Arena *arena);

ArenaAllocator arena_get_allocator(Arena *arena);

#endif // !RENA_H