arena.h

Go to the documentation of this file.

#ifndef HTILS_ARENA_H
#define HTILS_ARENA_H
 
//
//
//
 
#ifdef HTILS_THREAD_SAFE
#include <htils/atomic_types.h>
#include <stdatomic.h>
#include <threads.h>
#endif
 
#include <htils/basictypes.h>
 
#define KiB(bytes) ((u64)bytes << 10)
#define MiB(bytes) ((u64)bytes << 20)
#define GiB(bytes) ((u64)bytes << 30)
 
//
//
//
 
#ifndef HTILS_THREAD_SAFE
typedef struct arena {
  u64 reserved;
  u64 committed;
  u64 pos;
  u64 commit_pos;
} arena_t;
#else
typedef struct arena {
  u64 reserved;
  u64 committed;
  atomic_u64 pos;
  u64 commit_pos;
  mtx_t commit_mtx;
} arena_t;
#endif
 
typedef struct temp_arena {
  struct arena *arena;
  u64 start_pos;
} temp_arena_t;
 
//
//
//
 
arena_t *arena_new(u64 reserve_size, u64 commit_size);
 
void arena_free(arena_t *arena);
 
//
//
//
 
void *__arena_alloc(struct arena *arena, u64 size);
 
void *__arena_alloc_zeroed(struct arena *arena, u64 size);
 
void __arena_dealloc(struct arena *arena, u64 size);
 
//
//
//
 
#define arena_alloc(arena, type, size)                                         \
  __arena_alloc(arena, sizeof(type) * size);
 
#define arena_alloc_zeroed(arena, type, size)                                  \
  __arena_alloc_zeroed(arena, sizeof(type) * size);
 
#define arena_dealloc(arena, type, size)                                       \
  __arena_dealloc(arena, sizeof(type) * size);
 
//
//
//
 
void arena_dealloc_to(arena_t *arena, u64 pos);
 
void arena_clear(arena_t *arena);
 
//
//
//
 
temp_arena_t temp_arena_new(arena_t *arena);
 
void temp_arena_free(temp_arena_t temp);
 
#endif // !HTILS_ARENA_H