bitmap_allocator.h [plain text]
#if !defined _BITMAP_ALLOCATOR_H
#define _BITMAP_ALLOCATOR_H 1
#include <cstddef>
#include <utility>
#include <algorithm>
#include <vector>
#include <functional>
#include <new>
#include <bits/gthr.h>
#include <ext/new_allocator.h>
#include <cassert>
#define NDEBUG
namespace __gnu_cxx
{
namespace {
#if defined __GTHREADS
bool const __threads_enabled = __gthread_active_p();
#endif
}
#if defined __GTHREADS
class _Mutex {
__gthread_mutex_t _M_mut;
_Mutex (_Mutex const&);
_Mutex& operator= (_Mutex const&);
public:
_Mutex ()
{
if (__threads_enabled)
{
#if !defined __GTHREAD_MUTEX_INIT
__GTHREAD_MUTEX_INIT_FUNCTION(&_M_mut);
#else
__gthread_mutex_t __mtemp = __GTHREAD_MUTEX_INIT;
_M_mut = __mtemp;
#endif
}
}
~_Mutex ()
{
}
__gthread_mutex_t *_M_get() { return &_M_mut; }
};
class _Lock {
_Mutex* _M_pmt;
bool _M_locked;
_Lock (_Lock const&);
_Lock& operator= (_Lock const&);
public:
_Lock(_Mutex* __mptr)
: _M_pmt(__mptr), _M_locked(false)
{ this->_M_lock(); }
void _M_lock()
{
if (__threads_enabled)
{
_M_locked = true;
__gthread_mutex_lock(_M_pmt->_M_get());
}
}
void _M_unlock()
{
if (__threads_enabled)
{
if (__builtin_expect(_M_locked, true))
{
__gthread_mutex_unlock(_M_pmt->_M_get());
_M_locked = false;
}
}
}
~_Lock() { this->_M_unlock(); }
};
#endif
namespace __aux_balloc {
static const unsigned int _Bits_Per_Byte = 8;
static const unsigned int _Bits_Per_Block = sizeof(unsigned int) * _Bits_Per_Byte;
template <typename _Addr_Pair_t>
inline size_t __balloc_num_blocks (_Addr_Pair_t __ap)
{
return (__ap.second - __ap.first) + 1;
}
template <typename _Addr_Pair_t>
inline size_t __balloc_num_bit_maps (_Addr_Pair_t __ap)
{
return __balloc_num_blocks(__ap) / _Bits_Per_Block;
}
template <typename _Tp>
class _Inclusive_between : public std::unary_function<typename std::pair<_Tp, _Tp>, bool> {
typedef _Tp pointer;
pointer _M_ptr_value;
typedef typename std::pair<_Tp, _Tp> _Block_pair;
public:
_Inclusive_between (pointer __ptr) : _M_ptr_value(__ptr) { }
bool operator () (_Block_pair __bp) const throw ()
{
if (std::less_equal<pointer> ()(_M_ptr_value, __bp.second) &&
std::greater_equal<pointer> ()(_M_ptr_value, __bp.first))
return true;
else
return false;
}
};
template <typename _Functor>
class _Functor_Ref :
public std::unary_function<typename _Functor::argument_type, typename _Functor::result_type> {
_Functor& _M_fref;
public:
typedef typename _Functor::argument_type argument_type;
typedef typename _Functor::result_type result_type;
_Functor_Ref (_Functor& __fref) : _M_fref(__fref) { }
result_type operator() (argument_type __arg) { return _M_fref (__arg); }
};
template <typename _Tp, typename _Alloc>
class _Ffit_finder
: public std::unary_function<typename std::pair<_Tp, _Tp>, bool> {
typedef typename std::vector<std::pair<_Tp, _Tp>, _Alloc> _BPVector;
typedef typename _BPVector::difference_type _Counter_type;
typedef typename std::pair<_Tp, _Tp> _Block_pair;
unsigned int *_M_pbitmap;
unsigned int _M_data_offset;
public:
_Ffit_finder ()
: _M_pbitmap (0), _M_data_offset (0)
{ }
bool operator() (_Block_pair __bp) throw()
{
_Counter_type __diff = __gnu_cxx::__aux_balloc::__balloc_num_bit_maps (__bp);
assert (*(reinterpret_cast<unsigned int*>(__bp.first) - (__diff + 1)) <=
__gnu_cxx::__aux_balloc::__balloc_num_blocks (__bp));
if (*(reinterpret_cast<unsigned int*>(__bp.first) - (__diff + 1)) ==
__gnu_cxx::__aux_balloc::__balloc_num_blocks (__bp))
return false;
unsigned int *__rover = reinterpret_cast<unsigned int*>(__bp.first) - 1;
for (_Counter_type __i = 0; __i < __diff; ++__i)
{
_M_data_offset = __i;
if (*__rover)
{
_M_pbitmap = __rover;
return true;
}
--__rover;
}
return false;
}
unsigned int *_M_get () { return _M_pbitmap; }
unsigned int _M_offset () { return _M_data_offset * _Bits_Per_Block; }
};
template <typename _Tp, typename _Alloc>
class _Bit_map_counter {
typedef typename std::vector<std::pair<_Tp, _Tp>, _Alloc> _BPVector;
typedef typename _BPVector::size_type _Index_type;
typedef _Tp pointer;
_BPVector& _M_vbp;
unsigned int *_M_curr_bmap;
unsigned int *_M_last_bmap_in_block;
_Index_type _M_curr_index;
public:
_Bit_map_counter (_BPVector& Rvbp, int __index = -1)
: _M_vbp(Rvbp)
{
this->_M_reset(__index);
}
void _M_reset (int __index = -1) throw()
{
if (__index == -1)
{
_M_curr_bmap = 0;
_M_curr_index = (_Index_type)-1;
return;
}
_M_curr_index = __index;
_M_curr_bmap = reinterpret_cast<unsigned int*>(_M_vbp[_M_curr_index].first) - 1;
assert (__index <= (int)_M_vbp.size() - 1);
_M_last_bmap_in_block = _M_curr_bmap -
((_M_vbp[_M_curr_index].second - _M_vbp[_M_curr_index].first + 1) / _Bits_Per_Block - 1);
}
void _M_set_internal_bit_map (unsigned int *__new_internal_marker) throw()
{
_M_curr_bmap = __new_internal_marker;
}
bool _M_finished () const throw()
{
return (_M_curr_bmap == 0);
}
_Bit_map_counter& operator++ () throw()
{
if (_M_curr_bmap == _M_last_bmap_in_block)
{
if (++_M_curr_index == _M_vbp.size())
{
_M_curr_bmap = 0;
}
else
{
this->_M_reset (_M_curr_index);
}
}
else
{
--_M_curr_bmap;
}
return *this;
}
unsigned int *_M_get ()
{
return _M_curr_bmap;
}
pointer _M_base () { return _M_vbp[_M_curr_index].first; }
unsigned int _M_offset ()
{
return _Bits_Per_Block * ((reinterpret_cast<unsigned int*>(this->_M_base()) - _M_curr_bmap) - 1);
}
unsigned int _M_where () { return _M_curr_index; }
};
}
typedef unsigned int _Bit_map_type;
static inline unsigned int _Bit_scan_forward (register _Bit_map_type __num)
{
return static_cast<unsigned int>(__builtin_ctz(__num));
}
struct _OOM_handler {
static std::new_handler _S_old_handler;
static bool _S_handled_oom;
typedef void (*_FL_clear_proc)(void);
static _FL_clear_proc _S_oom_fcp;
_OOM_handler (_FL_clear_proc __fcp)
{
_S_oom_fcp = __fcp;
_S_old_handler = std::set_new_handler (_S_handle_oom_proc);
_S_handled_oom = false;
}
static void _S_handle_oom_proc()
{
_S_oom_fcp();
std::set_new_handler (_S_old_handler);
_S_handled_oom = true;
}
~_OOM_handler ()
{
if (!_S_handled_oom)
std::set_new_handler (_S_old_handler);
}
};
std::new_handler _OOM_handler::_S_old_handler;
bool _OOM_handler::_S_handled_oom = false;
_OOM_handler::_FL_clear_proc _OOM_handler::_S_oom_fcp = 0;
class _BA_free_list_store {
struct _LT_pointer_compare {
template <typename _Tp>
bool operator() (_Tp* __pt, _Tp const& __crt) const throw()
{
return *__pt < __crt;
}
};
#if defined __GTHREADS
static _Mutex _S_bfl_mutex;
#endif
static std::vector<unsigned int*> _S_free_list;
typedef std::vector<unsigned int*>::iterator _FLIter;
static void _S_validate_free_list(unsigned int *__addr) throw()
{
const unsigned int __max_size = 64;
if (_S_free_list.size() >= __max_size)
{
if (*__addr >= *_S_free_list.back())
{
operator delete((void*)__addr);
return;
}
else
{
operator delete((void*)_S_free_list.back());
_S_free_list.pop_back();
}
}
_FLIter __temp = std::lower_bound(_S_free_list.begin(), _S_free_list.end(),
*__addr, _LT_pointer_compare ());
_S_free_list.insert(__temp, __addr);
}
static bool _S_should_i_give(unsigned int __block_size, unsigned int __required_size) throw()
{
const unsigned int __max_wastage_percentage = 36;
if (__block_size >= __required_size &&
(((__block_size - __required_size) * 100 / __block_size) < __max_wastage_percentage))
return true;
else
return false;
}
public:
typedef _BA_free_list_store _BFL_type;
static inline void _S_insert_free_list(unsigned int *__addr) throw()
{
#if defined __GTHREADS
_Lock __bfl_lock(&_S_bfl_mutex);
#endif
_S_validate_free_list(--__addr);
}
static unsigned int *_S_get_free_list(unsigned int __sz) throw (std::bad_alloc)
{
#if defined __GTHREADS
_Lock __bfl_lock(&_S_bfl_mutex);
#endif
_FLIter __temp = std::lower_bound(_S_free_list.begin(), _S_free_list.end(),
__sz, _LT_pointer_compare());
if (__temp == _S_free_list.end() || !_S_should_i_give (**__temp, __sz))
{
_OOM_handler __set_handler(_BFL_type::_S_clear);
unsigned int *__ret_val = reinterpret_cast<unsigned int*>
(operator new (__sz + sizeof(unsigned int)));
*__ret_val = __sz;
return ++__ret_val;
}
else
{
unsigned int* __ret_val = *__temp;
_S_free_list.erase (__temp);
return ++__ret_val;
}
}
static void _S_clear()
{
#if defined __GTHREADS
_Lock __bfl_lock(&_S_bfl_mutex);
#endif
_FLIter __iter = _S_free_list.begin();
while (__iter != _S_free_list.end())
{
operator delete((void*)*__iter);
++__iter;
}
_S_free_list.clear();
}
};
#if defined __GTHREADS
_Mutex _BA_free_list_store::_S_bfl_mutex;
#endif
std::vector<unsigned int*> _BA_free_list_store::_S_free_list;
template <typename _Tp> class bitmap_allocator;
template <> class bitmap_allocator<void> {
public:
typedef void* pointer;
typedef const void* const_pointer;
typedef void value_type;
template <typename _Tp1> struct rebind { typedef bitmap_allocator<_Tp1> other; };
};
template <typename _Tp> class bitmap_allocator : private _BA_free_list_store {
public:
typedef size_t size_type;
typedef ptrdiff_t difference_type;
typedef _Tp* pointer;
typedef const _Tp* const_pointer;
typedef _Tp& reference;
typedef const _Tp& const_reference;
typedef _Tp value_type;
template <typename _Tp1> struct rebind { typedef bitmap_allocator<_Tp1> other; };
private:
static const unsigned int _Bits_Per_Byte = 8;
static const unsigned int _Bits_Per_Block = sizeof(unsigned int) * _Bits_Per_Byte;
static inline void _S_bit_allocate(unsigned int *__pbmap, unsigned int __pos) throw()
{
unsigned int __mask = 1 << __pos;
__mask = ~__mask;
*__pbmap &= __mask;
}
static inline void _S_bit_free(unsigned int *__pbmap, unsigned int __pos) throw()
{
unsigned int __mask = 1 << __pos;
*__pbmap |= __mask;
}
static inline void *_S_memory_get(size_t __sz) throw (std::bad_alloc)
{
return operator new(__sz);
}
static inline void _S_memory_put(void *__vptr) throw ()
{
operator delete(__vptr);
}
typedef typename std::pair<pointer, pointer> _Block_pair;
typedef typename __gnu_cxx::new_allocator<_Block_pair> _BPVec_allocator_type;
typedef typename std::vector<_Block_pair, _BPVec_allocator_type> _BPVector;
#if defined CHECK_FOR_ERRORS
static void _S_check_for_free_blocks() throw()
{
typedef typename __gnu_cxx::__aux_balloc::_Ffit_finder<pointer, _BPVec_allocator_type> _FFF;
_FFF __fff;
typedef typename _BPVector::iterator _BPiter;
_BPiter __bpi = std::find_if(_S_mem_blocks.begin(), _S_mem_blocks.end(),
__gnu_cxx::__aux_balloc::_Functor_Ref<_FFF>(__fff));
assert(__bpi == _S_mem_blocks.end());
}
#endif
static void _S_refill_pool() throw (std::bad_alloc)
{
#if defined CHECK_FOR_ERRORS
_S_check_for_free_blocks();
#endif
const unsigned int __num_bit_maps = _S_block_size / _Bits_Per_Block;
const unsigned int __size_to_allocate = sizeof(unsigned int) +
_S_block_size * sizeof(value_type) + __num_bit_maps*sizeof(unsigned int);
unsigned int *__temp =
reinterpret_cast<unsigned int*>(_BA_free_list_store::_S_get_free_list(__size_to_allocate));
*__temp = 0;
++__temp;
_Block_pair __bp = std::make_pair(reinterpret_cast<pointer>(__temp + __num_bit_maps),
reinterpret_cast<pointer>(__temp + __num_bit_maps)
+ _S_block_size - 1);
_S_mem_blocks.push_back(__bp);
unsigned int __bit_mask = 0; __bit_mask = ~__bit_mask;
for (unsigned int __i = 0; __i < __num_bit_maps; ++__i)
__temp[__i] = __bit_mask;
_S_block_size *= 2;
}
static _BPVector _S_mem_blocks;
static unsigned int _S_block_size;
static __gnu_cxx::__aux_balloc::_Bit_map_counter<pointer, _BPVec_allocator_type> _S_last_request;
static typename _BPVector::size_type _S_last_dealloc_index;
#if defined __GTHREADS
static _Mutex _S_mut;
#endif
static pointer _S_allocate_single_object()
{
#if defined __GTHREADS
_Lock __bit_lock(&_S_mut);
#endif
while (_S_last_request._M_finished() == false && (*(_S_last_request._M_get()) == 0))
{
_S_last_request.operator++();
}
if (__builtin_expect(_S_last_request._M_finished() == true, false))
{
typedef typename __gnu_cxx::__aux_balloc::_Ffit_finder<pointer, _BPVec_allocator_type> _FFF;
_FFF __fff;
typedef typename _BPVector::iterator _BPiter;
_BPiter __bpi = std::find_if(_S_mem_blocks.begin(), _S_mem_blocks.end(),
__gnu_cxx::__aux_balloc::_Functor_Ref<_FFF>(__fff));
if (__bpi != _S_mem_blocks.end())
{
unsigned int __nz_bit = _Bit_scan_forward(*__fff._M_get());
_S_bit_allocate(__fff._M_get(), __nz_bit);
_S_last_request._M_reset(__bpi - _S_mem_blocks.begin());
pointer __ret_val = __bpi->first + __fff._M_offset() + __nz_bit;
unsigned int *__puse_count = reinterpret_cast<unsigned int*>(__bpi->first) -
(__gnu_cxx::__aux_balloc::__balloc_num_bit_maps(*__bpi) + 1);
++(*__puse_count);
return __ret_val;
}
else
{
_S_refill_pool();
_S_last_request._M_reset(_S_mem_blocks.size() - 1);
}
}
unsigned int __nz_bit = _Bit_scan_forward(*_S_last_request._M_get());
_S_bit_allocate(_S_last_request._M_get(), __nz_bit);
pointer __ret_val = _S_last_request._M_base() + _S_last_request._M_offset() + __nz_bit;
unsigned int *__puse_count = reinterpret_cast<unsigned int*>
(_S_mem_blocks[_S_last_request._M_where()].first) -
(__gnu_cxx::__aux_balloc::__balloc_num_bit_maps(_S_mem_blocks[_S_last_request._M_where()]) + 1);
++(*__puse_count);
return __ret_val;
}
static void _S_deallocate_single_object(pointer __p) throw()
{
#if defined __GTHREADS
_Lock __bit_lock(&_S_mut);
#endif
typedef typename _BPVector::iterator _Iterator;
typedef typename _BPVector::difference_type _Difference_type;
_Difference_type __diff;
int __displacement;
assert(_S_last_dealloc_index >= 0);
if (__gnu_cxx::__aux_balloc::_Inclusive_between<pointer>(__p)(_S_mem_blocks[_S_last_dealloc_index]))
{
assert(_S_last_dealloc_index <= _S_mem_blocks.size() - 1);
__diff = _S_last_dealloc_index;
__displacement = __p - _S_mem_blocks[__diff].first;
}
else
{
_Iterator _iter = (std::find_if(_S_mem_blocks.begin(), _S_mem_blocks.end(),
__gnu_cxx::__aux_balloc::_Inclusive_between<pointer>(__p)));
assert(_iter != _S_mem_blocks.end());
__diff = _iter - _S_mem_blocks.begin();
__displacement = __p - _S_mem_blocks[__diff].first;
_S_last_dealloc_index = __diff;
}
const unsigned int __rotate = __displacement % _Bits_Per_Block;
unsigned int *__bit_mapC = reinterpret_cast<unsigned int*>(_S_mem_blocks[__diff].first) - 1;
__bit_mapC -= (__displacement / _Bits_Per_Block);
_S_bit_free(__bit_mapC, __rotate);
unsigned int *__puse_count = reinterpret_cast<unsigned int*>
(_S_mem_blocks[__diff].first) -
(__gnu_cxx::__aux_balloc::__balloc_num_bit_maps(_S_mem_blocks[__diff]) + 1);
assert(*__puse_count != 0);
--(*__puse_count);
if (__builtin_expect(*__puse_count == 0, false))
{
_S_block_size /= 2;
_Block_pair __bp = _S_mem_blocks[__diff];
_S_insert_free_list(__puse_count);
_S_mem_blocks.erase(_S_mem_blocks.begin() + __diff);
if ((_Difference_type)_S_last_request._M_where() >= __diff--)
{
_S_last_request._M_reset(__diff);
}
if (_S_last_dealloc_index >= _S_mem_blocks.size())
{
_S_last_dealloc_index =(__diff != -1 ? __diff : 0);
assert(_S_last_dealloc_index >= 0);
}
}
}
public:
bitmap_allocator() throw()
{ }
bitmap_allocator(const bitmap_allocator&) { }
template <typename _Tp1> bitmap_allocator(const bitmap_allocator<_Tp1>&) throw()
{ }
~bitmap_allocator() throw()
{ }
pointer allocate(size_type __n)
{
if (__builtin_expect(__n == 1, true))
return _S_allocate_single_object();
else
return reinterpret_cast<pointer>(_S_memory_get(__n * sizeof(value_type)));
}
pointer allocate(size_type __n, typename bitmap_allocator<void>::const_pointer)
{
return allocate(__n);
}
void deallocate(pointer __p, size_type __n) throw()
{
if (__builtin_expect(__n == 1, true))
_S_deallocate_single_object(__p);
else
_S_memory_put(__p);
}
pointer address(reference r) const { return &r; }
const_pointer address(const_reference r) const { return &r; }
size_type max_size(void) const throw() { return (size_type()-1)/sizeof(value_type); }
void construct (pointer p, const_reference __data)
{
::new(p) value_type(__data);
}
void destroy (pointer p)
{
p->~value_type();
}
};
template <typename _Tp>
typename bitmap_allocator<_Tp>::_BPVector bitmap_allocator<_Tp>::_S_mem_blocks;
template <typename _Tp>
unsigned int bitmap_allocator<_Tp>::_S_block_size = bitmap_allocator<_Tp>::_Bits_Per_Block;
template <typename _Tp>
typename __gnu_cxx::bitmap_allocator<_Tp>::_BPVector::size_type
bitmap_allocator<_Tp>::_S_last_dealloc_index = 0;
template <typename _Tp>
__gnu_cxx::__aux_balloc::_Bit_map_counter
<typename bitmap_allocator<_Tp>::pointer, typename bitmap_allocator<_Tp>::_BPVec_allocator_type>
bitmap_allocator<_Tp>::_S_last_request(_S_mem_blocks);
#if defined __GTHREADS
template <typename _Tp>
__gnu_cxx::_Mutex
bitmap_allocator<_Tp>::_S_mut;
#endif
template <typename _Tp1, typename _Tp2>
bool operator== (const bitmap_allocator<_Tp1>&, const bitmap_allocator<_Tp2>&) throw()
{
return true;
}
template <typename _Tp1, typename _Tp2>
bool operator!= (const bitmap_allocator<_Tp1>&, const bitmap_allocator<_Tp2>&) throw()
{
return false;
}
}
#endif //_BITMAP_ALLOCATOR_H