54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
|
/*
|
||
|
Copyright (c) 2014 Microsoft Corporation. All rights reserved.
|
||
|
Released under Apache 2.0 license as described in the file LICENSE.
|
||
|
|
||
|
Author: Leonardo de Moura
|
||
|
*/
|
||
|
#include <vector>
|
||
|
#include "util/thread.h"
|
||
|
#include "util/memory_pool.h"
|
||
|
|
||
|
namespace lean {
|
||
|
memory_pool::~memory_pool() {
|
||
|
while (m_free_list != nullptr) {
|
||
|
void * r = m_free_list;
|
||
|
m_free_list = *(reinterpret_cast<void **>(r));
|
||
|
free(r);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void * memory_pool::allocate() {
|
||
|
if (m_free_list != nullptr) {
|
||
|
void * r = m_free_list;
|
||
|
m_free_list = *(reinterpret_cast<void **>(r));
|
||
|
return r;
|
||
|
} else {
|
||
|
return malloc(m_size);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
typedef std::vector<memory_pool*> memory_pools;
|
||
|
LEAN_THREAD_PTR(memory_pools, g_thread_pools);
|
||
|
|
||
|
static void thread_finalize_memory_pool() {
|
||
|
if (g_thread_pools) {
|
||
|
unsigned i = g_thread_pools->size();
|
||
|
while (i > 0) {
|
||
|
--i;
|
||
|
delete (*g_thread_pools)[i];
|
||
|
}
|
||
|
delete g_thread_pools;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
memory_pool * allocate_thread_memory_pool(unsigned sz) {
|
||
|
if (!g_thread_pools) {
|
||
|
register_post_thread_finalizer(thread_finalize_memory_pool);
|
||
|
g_thread_pools = new std::vector<memory_pool*>();
|
||
|
}
|
||
|
memory_pool * r = new memory_pool(sz);
|
||
|
g_thread_pools->push_back(r);
|
||
|
return r;
|
||
|
}
|
||
|
}
|