summaryrefslogtreecommitdiff
path: root/prb_arena.h
blob: 0a6de9b8950c7b35d0a7efb7335899ba6929479d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#define DEFAULT_ARENA_SIZE kilobytes(4)

arena_t	alloc_arena(u64 capacity)
{
	void *sys_alloc(u64 length);

	/* TODO(pryazha): Find reasonable maximum capacity through testing */
	assert(capacity <= (u64)gigabytes(16));

	if (!capacity)
		capacity = DEFAULT_ARENA_SIZE;

	/* TODO(pryazha): Use OS specific memory allocator
	 * (like VirtualAlloc on Windows or mmap on Linux)
	 */
	void *memory = sys_alloc(capacity);
	assert(memory);

	arena_t arena = {memory, capacity, 0};

	return arena;
}

void release_arena(arena_t *arena)
{
	void sys_free(void *memory, u64 length);

	assert(arena);
	sys_free(arena->memory, arena->capacity);
	arena->memory = 0;
	arena->capacity = 0;
	arena->used = 0;
}

void *push_arena(arena_t *arena, u64 size)
{
	assert(arena);
	assert(arena->used + size <= arena->capacity);
	void *memory = arena->memory + arena->used;
	arena->used += size;
	return memory;
}

void pop_arena(arena_t *arena, u64 size)
{
	assert(arena);
	arena->used -= min(size, arena->used);
}