dali-physics/third-party/bullet3/src/Bullet3Common/b3StackAlloc.h

   1 /*
   2 Copyright (c) 2003-2013 Gino van den Bergen / Erwin Coumans  http://bulletphysics.org
   3
   4 This software is provided 'as-is', without any express or implied warranty.
   5 In no event will the authors be held liable for any damages arising from the use of this software.
   6 Permission is granted to anyone to use this software for any purpose,
   7 including commercial applications, and to alter it and redistribute it freely,
   8 subject to the following restrictions:
   9
  10 1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
  11 2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
  12 3. This notice may not be removed or altered from any source distribution.
  13 */
  14
  15 /*
  16 StackAlloc extracted from GJK-EPA collision solver by Nathanael Presson
  17 Nov.2006
  18 */
  19
  20 #ifndef B3_STACK_ALLOC
  21 #define B3_STACK_ALLOC
  22
  23 #include "b3Scalar.h"  //for b3Assert
  24 #include "b3AlignedAllocator.h"
  25
  26 ///The b3Block class is an internal structure for the b3StackAlloc memory allocator.
  27 struct b3Block
  28 {
  29         b3Block* previous;
  30         unsigned char* address;
  31 };
  32
  33 ///The StackAlloc class provides some fast stack-based memory allocator (LIFO last-in first-out)
  34 class b3StackAlloc
  35 {
  36 public:
  37         b3StackAlloc(unsigned int size)
  38         {
  39                 ctor();
  40                 create(size);
  41         }
  42         ~b3StackAlloc() { destroy(); }
  43
  44         inline void create(unsigned int size)
  45         {
  46                 destroy();
  47                 data = (unsigned char*)b3AlignedAlloc(size, 16);
  48                 totalsize = size;
  49         }
  50         inline void destroy()
  51         {
  52                 b3Assert(usedsize == 0);
  53                 //Raise(L"StackAlloc is still in use");
  54
  55                 if (usedsize == 0)
  56                 {
  57                         if (!ischild && data)
  58                                 b3AlignedFree(data);
  59
  60                         data = 0;
  61                         usedsize = 0;
  62                 }
  63         }
  64
  65         int getAvailableMemory() const
  66         {
  67                 return static_cast<int>(totalsize - usedsize);
  68         }
  69
  70         unsigned char* allocate(unsigned int size)
  71         {
  72                 const unsigned int nus(usedsize + size);
  73                 if (nus < totalsize)
  74                 {
  75                         usedsize = nus;
  76                         return (data + (usedsize - size));
  77                 }
  78                 b3Assert(0);
  79                 //&& (L"Not enough memory"));
  80
  81                 return (0);
  82         }
  83         B3_FORCE_INLINE b3Block* beginBlock()
  84         {
  85                 b3Block* pb = (b3Block*)allocate(sizeof(b3Block));
  86                 pb->previous = current;
  87                 pb->address = data + usedsize;
  88                 current = pb;
  89                 return (pb);
  90         }
  91         B3_FORCE_INLINE void endBlock(b3Block* block)
  92         {
  93                 b3Assert(block == current);
  94                 //Raise(L"Unmatched blocks");
  95                 if (block == current)
  96                 {
  97                         current = block->previous;
  98                         usedsize = (unsigned int)((block->address - data) - sizeof(b3Block));
  99                 }
 100         }
 101
 102 private:
 103         void ctor()
 104         {
 105                 data = 0;
 106                 totalsize = 0;
 107                 usedsize = 0;
 108                 current = 0;
 109                 ischild = false;
 110         }
 111         unsigned char* data;
 112         unsigned int totalsize;
 113         unsigned int usedsize;
 114         b3Block* current;
 115         bool ischild;
 116 };
 117
 118 #endif  //B3_STACK_ALLOC