Sepehr Taghdisian
ago%!(EXTRA string=11 years)
parent
e9b697698a
commit
df5a06f119
3 changed files with 19 additions and 237 deletions
@ -1,210 +0,0 @@ |
||||
/***********************************************************************************
|
||||
* Author: Sepehr Taghdisian (sep.tagh@gmail.com) |
||||
*/ |
||||
|
||||
#ifndef IMMEM_H |
||||
#define IMMEM_H |
||||
|
||||
// Note: This file is meant to included only inside imgui.cpp
|
||||
|
||||
#ifndef IMGUI_INTERNAL_USE |
||||
#error "This file is intented for internal use only (as include)" |
||||
#endif |
||||
|
||||
/* Linked-List ************************************************************************************/ |
||||
struct LinkedList |
||||
{ |
||||
LinkedList *next; |
||||
LinkedList *prev; |
||||
void *data; |
||||
|
||||
LinkedList() : next(NULL), prev(NULL) {} |
||||
}; |
||||
|
||||
|
||||
static void list_add(struct LinkedList** plist, struct LinkedList* item, void* data) |
||||
{ |
||||
item->next = (*plist); |
||||
item->prev = NULL; |
||||
if (*plist != NULL) |
||||
(*plist)->prev = item; |
||||
*plist = item; |
||||
item->data = data; |
||||
} |
||||
|
||||
static void list_addlast(struct LinkedList** plist, struct LinkedList* item, void* data) |
||||
{ |
||||
if (*plist != NULL) { |
||||
struct LinkedList* last = *plist; |
||||
while (last->next != NULL) last = last->next; |
||||
last->next = item; |
||||
item->prev = last; |
||||
item->next = NULL; |
||||
} else { |
||||
*plist = item; |
||||
item->prev = item->next = NULL; |
||||
} |
||||
|
||||
item->data = data; |
||||
} |
||||
|
||||
static void list_remove(struct LinkedList** plist, struct LinkedList* item) |
||||
{ |
||||
if (item->next != NULL) item->next->prev = item->prev; |
||||
if (item->prev != NULL) item->prev->next = item->next; |
||||
if (*plist == item) *plist = item->next; |
||||
item->next = item->prev = NULL; |
||||
} |
||||
|
||||
/* PoolAlloc **************************************************************************************/ |
||||
template <typename T> |
||||
class PoolAlloc |
||||
{ |
||||
private: |
||||
LinkedList *m_blocks; /* first node of m_blocks */ |
||||
int m_block_cnt; |
||||
int m_items_max; /* maximum number of items allowed (per block) */ |
||||
ImGui_MallocCallback m_malloc; |
||||
ImGui_FreeCallback m_free; |
||||
|
||||
private: |
||||
struct Block |
||||
{ |
||||
LinkedList node; /* linked-list node */ |
||||
unsigned char *buffer; /* memory buffer that holds all objects */ |
||||
void **ptrs; /* pointer references to the buffer */ |
||||
int iter; /* iterator for current buffer position */ |
||||
}; |
||||
|
||||
private: |
||||
Block* create_block(int block_size) |
||||
{ |
||||
// Allocate in one call
|
||||
size_t total_sz = |
||||
sizeof(Block) + |
||||
sizeof(T)*block_size + |
||||
sizeof(void*)*block_size; |
||||
unsigned char *buff = (unsigned char*)m_malloc(total_sz); |
||||
if (buff == NULL) |
||||
return NULL; |
||||
memset(buff, 0x00, total_sz); |
||||
|
||||
Block *block = (Block*)buff; |
||||
buff += sizeof(Block); |
||||
block->buffer = buff; |
||||
buff += sizeof(T)*block_size; |
||||
block->ptrs = (void**)buff; |
||||
|
||||
// Assign pointer refs
|
||||
for (int i = 0; i < block_size; i++) |
||||
block->ptrs[block_size-i-1] = block->buffer + i*sizeof(T); |
||||
block->iter = block_size; |
||||
|
||||
/* add to linked-list of the pool */ |
||||
list_addlast(&m_blocks, &block->node, block); |
||||
m_block_cnt++; |
||||
return block; |
||||
} |
||||
|
||||
void destroy_block(Block *block) |
||||
{ |
||||
list_remove(&m_blocks, &block->node); |
||||
m_free(block); |
||||
m_block_cnt--; |
||||
} |
||||
|
||||
public: |
||||
PoolAlloc() |
||||
{ |
||||
m_blocks = NULL; |
||||
m_block_cnt = 0; |
||||
m_items_max = 0; |
||||
m_malloc = NULL; |
||||
m_free = NULL; |
||||
} |
||||
|
||||
bool create(int block_sz, ImGui_MallocCallback malloc_fn, ImGui_FreeCallback free_fn) |
||||
{ |
||||
m_items_max = block_sz; |
||||
m_malloc = malloc_fn; |
||||
m_free = free_fn; |
||||
|
||||
// First block
|
||||
Block *block = create_block(block_sz); |
||||
if (block == NULL) { |
||||
destroy(); |
||||
return false; |
||||
} |
||||
|
||||
return true; |
||||
} |
||||
|
||||
void destroy() |
||||
{ |
||||
LinkedList* node = m_blocks; |
||||
while (node != NULL) { |
||||
LinkedList* next = node->next; |
||||
destroy_block((Block*)node->data); |
||||
node = next; |
||||
} |
||||
} |
||||
|
||||
T* alloc() |
||||
{ |
||||
LinkedList* node = m_blocks; |
||||
|
||||
while (node != NULL) { |
||||
Block *block = (Block*)node->data; |
||||
if (block->iter > 0) |
||||
return (T*)block->ptrs[--block->iter]; |
||||
|
||||
node = node->next; |
||||
} |
||||
|
||||
/* couldn't find a free block, create a new one */ |
||||
Block *block = create_block(m_items_max); |
||||
if (block == NULL) |
||||
return NULL; |
||||
|
||||
return (T*)block->ptrs[--block->iter]; |
||||
} |
||||
|
||||
void free(T *ptr) |
||||
{ |
||||
// find the block that pointer belongs to, and free the pointer from that block
|
||||
LinkedList *node = m_blocks; |
||||
int buffer_sz = m_items_max*sizeof(T); |
||||
unsigned char *u8ptr = (unsigned char*)ptr; |
||||
|
||||
while (node != NULL) { |
||||
Block *block = (Block*)node->data; |
||||
if (u8ptr >= block->buffer && u8ptr < (block->buffer + buffer_sz)) { |
||||
IM_ASSERT(block->iter != m_items_max); |
||||
block->ptrs[block->iter++] = ptr; |
||||
return; |
||||
} |
||||
node = node->next; |
||||
} |
||||
|
||||
// Memory block does not belong to the pool?!
|
||||
IM_ASSERT(0); |
||||
} |
||||
|
||||
void clear() |
||||
{ |
||||
int block_size = m_items_max; |
||||
LinkedList* node = m_blocks; |
||||
while (node != NULL) { |
||||
Block *block = (Block*)node->data; |
||||
|
||||
/* only re-assign pointer references to buffer */ |
||||
for (int i = 0; i < block_size; i++) |
||||
block->ptrs[block_size-i-1] = block->buffer + i*sizeof(T); |
||||
block->iter = block_size; |
||||
|
||||
node = node->next; |
||||
} |
||||
} |
||||
}; |
||||
|
||||
#endif // IMMEM_H
|
||||
Loading…
Reference in New Issue