arena.h

Go to the documentation of this file.

 
#pragma once
 
#include "def.h"
#ifndef PS_INITIAL_ARENA_MMAP
    #define PS_INITIAL_ARENA_MMAP 0x2000000ULL
#endif
 
struct ps_arena_allocator;
#ifndef _PS_PUBLIC_SRC
struct ps_arena_allocator {
    usize page_size;
    usize offset;
    void* start;
};
#endif
 
enum ps_arena_domain { PS_ARENA_AST, PS_ARENA_IR, __PS_ARENA_COUNT };
 
void ps_arena_open(enum ps_arena_domain domain);
 
void ps_arena_close(enum ps_arena_domain domain);
 
void* ps_arena_allocate(enum ps_arena_domain domain, usize n);
void* ps_arena_allocate_debug(enum ps_arena_domain domain, usize n,
    const char* file, const char* func, usize line);
 
#if defined(PS_DEBUG) && defined(PS_DEBUG_ARENA)
    #define arena_alloc(arena, n)                                              \
        ps_arena_allocate_debug(arena, n, __FILE__, __func__, __LINE__)
    #define astalloc(n) arena_alloc(PS_ARENA_AST, n)
    #define iralloc(n) arena_alloc(IR_Arena, n)
#else
    #define arena_alloc(arena, n) ps_arena_allocate(arena, n)
    #define astalloc(n) arena_alloc(PS_ARENA_AST, n)
    #define iralloc(n) arena_alloc(PS_ARENA_IR, n)
#endif
 
char* arena_strndup(enum ps_arena_domain arena, STR str, usize length);
 
#define ast_strndup(str, len) arena_strndup(PS_ARENA_AST, str, len)
 
#define arena_sprintf(arena, fmt, ...)                                         \
    ({                                                                         \
        char* str = NULL;                                                      \
        int length = snprintf(NULL, 0, fmt, __VA_ARGS__);                      \
        if (length >= 0) {                                                     \
            str = arena_alloc(arena, length + 1);                              \
            if (str) {                                                         \
                snprintf(str, length + 1, fmt, __VA_ARGS__);                   \
            }                                                                  \
        }                                                                      \
        str;                                                                   \
    })
#define ast_sprintf(fmt, ...) arena_sprintf(PS_ARENA_AST, fmt, __VA_ARGS__)