include/llvm/Support/Allocator.h

   1 //===--- Allocator.h - Simple memory allocation abstraction -----*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file defines the MallocAllocator and BumpPtrAllocator interfaces.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_SUPPORT_ALLOCATOR_H
  15 #define LLVM_SUPPORT_ALLOCATOR_H
  16
  17 #include "llvm/Support/AlignOf.h"
  18 #include <cstdlib>
  19
  20 namespace llvm {
  21
  22 class MallocAllocator {
  23 public:
  24   MallocAllocator() {}
  25   ~MallocAllocator() {}
  26
  27   void Reset() {}
  28
  29   void *Allocate(size_t Size, size_t /*Alignment*/) { return malloc(Size); }
  30
  31   template <typename T>
  32   T *Allocate() { return static_cast<T*>(malloc(sizeof(T))); }
  33
  34   void Deallocate(void *Ptr) { free(Ptr); }
  35
  36   void PrintStats() const {}
  37 };
  38
  39 /// BumpPtrAllocator - This allocator is useful for containers that need very
  40 /// simple memory allocation strategies.  In particular, this just keeps
  41 /// allocating memory, and never deletes it until the entire block is dead. This
  42 /// makes allocation speedy, but must only be used when the trade-off is ok.
  43 class BumpPtrAllocator {
  44   BumpPtrAllocator(const BumpPtrAllocator &); // do not implement
  45   void operator=(const BumpPtrAllocator &);   // do not implement
  46
  47   void *TheMemory;
  48 public:
  49   BumpPtrAllocator();
  50   ~BumpPtrAllocator();
  51
  52   void Reset();
  53
  54   void *Allocate(size_t Size, size_t Alignment);
  55
  56   template <typename T>
  57   T *Allocate() {
  58     return static_cast<T*>(Allocate(sizeof(T),AlignOf<T>::Alignment));
  59   }
  60
  61   void Deallocate(void * /*Ptr*/) {}
  62
  63   void PrintStats() const;
  64 };
  65
  66 }  // end namespace llvm
  67
  68 #endif