Public API Reference

csutil/allocator.h

Go to the documentation of this file.

/*
    Copyright (C) 2006 by Frank Richter

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the Free
    Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#ifndef __CS_CSUTIL_ALLOCATOR_H__
#define __CS_CSUTIL_ALLOCATOR_H__

#include "csutil/alignedalloc.h"
#include "csutil/memdebug.h"

namespace CS
{
  namespace Memory
  {
    class AllocatorMalloc
    {
    #ifdef CS_MEMORY_TRACKER
      csMemTrackerInfo* mti;
    #endif
    public:
    #ifdef CS_MEMORY_TRACKER
      AllocatorMalloc() : mti (0) {}
    #endif
      void* Alloc (const size_t n)
      {
      #ifdef CS_MEMORY_TRACKER
        size_t* p = (size_t*)cs_malloc (n + sizeof (size_t));
        *p = n;
        p++;
        if (mti) mtiUpdateAmount (mti, 1, int (n));
        return p;
      #else
        return cs_malloc (n);
      #endif
      }
      void Free (void* p)
      {
      #ifdef CS_MEMORY_TRACKER
        size_t* x = (size_t*)p;
        x--;
        size_t allocSize = *x;
        cs_free (x);
        if (mti) mtiUpdateAmount (mti, -1, -int (allocSize));
      #else
        cs_free (p);
      #endif
      }
      void* Realloc (void* p, size_t newSize)
      {
      #ifdef CS_MEMORY_TRACKER
        if (p == 0) return Alloc (newSize);
        size_t* x = (size_t*)p;
        x--;
        if (mti) mtiUpdateAmount (mti, -1, -int (*x));
        size_t* np = (size_t*)cs_realloc (x, newSize + sizeof (size_t));
        *np = newSize;
        np++;
        if (mti) mtiUpdateAmount (mti, 1, int (newSize));
        return np;
      #else
        return cs_realloc (p, newSize);
      #endif
      }
      void SetMemTrackerInfo (const char* info)
      {
      #ifdef CS_MEMORY_TRACKER
        mti = mtiRegister (info);
      #else
        (void)info;
      #endif
      }
    };
    
    template<typename T, size_t N, class ExcessAllocator = AllocatorMalloc>
    class LocalBufferAllocator : public ExcessAllocator
    {
      static const size_t localSize = N * sizeof (T);
      uint8 localBuf[localSize];
    #ifdef CS_DEBUG
      bool allocation;
    #endif
    public:
    #ifdef CS_DEBUG
      LocalBufferAllocator () : allocation (false) {}
      LocalBufferAllocator (const ExcessAllocator& xalloc) : 
        ExcessAllocator (xalloc), allocation (false) {}
    #else
      LocalBufferAllocator () {}
      LocalBufferAllocator (const ExcessAllocator& xalloc) : 
        ExcessAllocator (xalloc) {}
    #endif
      T* Alloc (size_t allocSize)
      {
      #ifdef CS_DEBUG
        CS_ASSERT_MSG("LocalBufferAllocator only allows one allocation a time!",
          !allocation);
        allocation = true;
      #endif
        if (allocSize <= localSize)
          return (T*)localBuf;
        else
          return (T*)ExcessAllocator::Alloc (allocSize);
      }
    
      void Free (T* mem)
      {
      #ifdef CS_DEBUG
        CS_ASSERT_MSG("Free() without prior allocation",
          allocation);
        allocation = false;
      #endif
        if (mem != (T*)localBuf) ExcessAllocator::Free (mem);
      }
    
      // The 'relevantcount' parameter should be the number of items
      // in the old array that are initialized.
      void* Realloc (void* p, size_t newSize)
      {
      #ifdef CS_DEBUG
        CS_ASSERT_MSG("Realloc() without prior allocation",
          allocation);
      #endif
        if (p == localBuf)
        {
          if (newSize <= localSize)
            return p;
          else
          {
            p = ExcessAllocator::Alloc (newSize);
            memcpy (p, localBuf, localSize);
            return p;
          }
        }
        else
        {
          if (newSize <= localSize)
          {
            memcpy (localBuf, p, newSize);
            ExcessAllocator::Free (p);
            return localBuf;
          }
          else
            return ExcessAllocator::Realloc (p, newSize);
        }
      }
      using ExcessAllocator::SetMemTrackerInfo;
    };
    
    template <size_t A = 1>
    class AllocatorAlign
    {
    public:
      static inline void* Alloc (size_t size) 
      {
        return AlignedMalloc (size, A);
      }

      static inline void Free (void* p)
      {
        AlignedFree (p);
      }

      void* Realloc (void* p, size_t newSize)
      {
        return AlignedRealloc (p, newSize, A);
      }

      void SetMemTrackerInfo (const char* info)
      {
        (void)info;
      }
    };

    template<typename T, typename Allocator>
    struct AllocatorPointerWrapper : public Allocator
    {
      T* p;

      AllocatorPointerWrapper () {}
      AllocatorPointerWrapper (const Allocator& alloc) : 
        Allocator (alloc) {}
      AllocatorPointerWrapper (T* p) : p (p) {}
      AllocatorPointerWrapper (const Allocator& alloc, T* p) : 
        Allocator (alloc), p (p) {}
    };
  } // namespace Memory
} // namespace CS

#endif // __CS_CSUTIL_ALLOCATOR_H__

---