#include "jit-internal.h"
void _jit_memory_pool_init(jit_memory_pool *pool, unsigned int elem_size)
{
pool->elem_size = elem_size;
pool->elems_per_block = 4000 / elem_size;
pool->elems_in_last = pool->elems_per_block;
pool->blocks = 0;
pool->free_list = 0;
}
void _jit_memory_pool_free(jit_memory_pool *pool, jit_meta_free_func func)
{
jit_pool_block_t block;
while(pool->blocks != 0)
{
block = pool->blocks;
pool->blocks = block->next;
if(func)
{
while(pool->elems_in_last > 0)
{
--(pool->elems_in_last);
(*func)(block->data + pool->elems_in_last * pool->elem_size);
}
}
jit_free(block);
pool->elems_in_last = pool->elems_per_block;
}
pool->free_list = 0;
}
void *_jit_memory_pool_alloc(jit_memory_pool *pool)
{
void *data;
if(pool->free_list)
{
data = pool->free_list;
pool->free_list = *((void **)data);
jit_memzero(data, pool->elem_size);
return data;
}
if(pool->elems_in_last >= pool->elems_per_block)
{
data = (void *)jit_calloc(1, sizeof(struct jit_pool_block) +
pool->elem_size * pool->elems_per_block - 1);
if(!data)
{
return 0;
}
((jit_pool_block_t)data)->next = pool->blocks;
pool->blocks = (jit_pool_block_t)data;
pool->elems_in_last = 0;
}
data = (void *)(pool->blocks->data +
pool->elems_in_last * pool->elem_size);
++(pool->elems_in_last);
return data;
}
void _jit_memory_pool_dealloc(jit_memory_pool *pool, void *item)
{
*((void **)item) = pool->free_list;
pool->free_list = item;
}