Src/Common/Allocators/PooledAlloc.h

   1 // PooledAlloc.h
   2 //
   3 // Copyright (c) 2005 David Ward
   4
   5 #ifndef __PooledAlloc_h__
   6 #define __PooledAlloc_h__
   7
   8 // CPooledAlloc allocates objects T in fixed-size blocks (specified in the constructor)
   9 // Alloc returns an uninitialized T*
  10 // Free returns an object to the pool
  11
  12 #include "SimplePooledAlloc.h"
  13
  14 /////////////////////////////////////////////////////////////////////////////
  15
  16 template<typename T> class CPooledAlloc {
  17
  18 public:
  19
  20   // Construct with given block size
  21   CPooledAlloc(size_t iBlockSize);
  22   ~CPooledAlloc();
  23
  24   // Return an uninitialized object
  25   T *Alloc();
  26
  27   // Return an object to the pool
  28   void Free(T * pFree);
  29
  30 private:
  31
  32   // Use simple pooled alloc for the blocked allocation
  33   CSimplePooledAlloc < T > m_Alloc;
  34
  35   // The free list
  36   std::vector < T * >m_vpFree;
  37
  38 };
  39
  40 template<typename T> CPooledAlloc<T>::CPooledAlloc(size_t iSize):m_Alloc(iSize) {}
  41
  42 template<typename T> CPooledAlloc<T>::~CPooledAlloc() {}
  43
  44 template<typename T> T * CPooledAlloc < T >::Alloc() {
  45   if(m_vpFree.size() > 0) {
  46     T *pLast = m_vpFree.back();
  47     m_vpFree.pop_back();
  48     return pLast;
  49   }
  50   return m_Alloc.Alloc();
  51 }
  52
  53 template<typename T> void CPooledAlloc<T>::Free(T *pFree) {
  54   m_vpFree.push_back(pFree);
  55 }
  56
  57 #endif // __include__