53 lines
1.2 KiB
C
53 lines
1.2 KiB
C
/*
|
|
* This software is licensed under the terms of the MIT License.
|
|
* See COPYING for further information.
|
|
* ---
|
|
* Copyright (c) 2011-2024, Lukas Weber <laochailan@web.de>.
|
|
* Copyright (c) 2012-2024, Andrei Alexeyev <akari@taisei-project.org>.
|
|
*/
|
|
|
|
#include "mempool.h"
|
|
|
|
void mempool_init(
|
|
MemPool *pool,
|
|
const char *tag,
|
|
MemArena *arena,
|
|
size_t obj_size,
|
|
size_t obj_align
|
|
) {
|
|
*pool = (MemPool) {
|
|
.arena = arena,
|
|
.obj_size = obj_size,
|
|
.obj_align = obj_align,
|
|
.tag = tag,
|
|
};
|
|
}
|
|
|
|
void *mempool_acquire(MemPool *pool) {
|
|
MemPoolObjectHeader *obj = pool->free_objects;
|
|
|
|
if(obj) {
|
|
assert(pool->num_used < pool->num_allocated);
|
|
pool->free_objects = obj->next;
|
|
} else {
|
|
assert(pool->num_used == pool->num_allocated);
|
|
obj = marena_alloc_aligned(pool->arena, pool->obj_size, pool->obj_align);
|
|
++pool->num_allocated;
|
|
}
|
|
|
|
memset(obj, 0, pool->obj_size);
|
|
++pool->num_used;
|
|
|
|
assert(pool->num_used <= pool->num_allocated);
|
|
return obj;
|
|
}
|
|
|
|
void mempool_release(MemPool *pool, void *object) {
|
|
MemPoolObjectHeader *obj = object;
|
|
obj->next = pool->free_objects;
|
|
pool->free_objects = obj;
|
|
|
|
--pool->num_used;
|
|
assert(pool->num_used >= 0);
|
|
assert(pool->num_used <= pool->num_allocated);
|
|
}
|