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