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   template <typename T>
  35   T *Allocate(size_t Num) {
  36     return static_cast<T*>(malloc(sizeof(T)*Num));
  37   }
  38
  39   void Deallocate(void *Ptr) { free(Ptr); }
  40
  41   void PrintStats() const {}
  42 };
  43
  44 /// BumpPtrAllocator - This allocator is useful for containers that need very
  45 /// simple memory allocation strategies.  In particular, this just keeps
  46 /// allocating memory, and never deletes it until the entire block is dead. This
  47 /// makes allocation speedy, but must only be used when the trade-off is ok.
  48 class BumpPtrAllocator {
  49   BumpPtrAllocator(const BumpPtrAllocator &); // do not implement
  50   void operator=(const BumpPtrAllocator &);   // do not implement
  51
  52   void *TheMemory;
  53 public:
  54   BumpPtrAllocator();
  55   ~BumpPtrAllocator();
  56
  57   void Reset();
  58
  59   void *Allocate(size_t Size, size_t Alignment);
  60
  61   template <typename T>
  62   T *Allocate() {
  63     return static_cast<T*>(Allocate(sizeof(T),AlignOf<T>::Alignment));
  64   }
  65
  66   template <typename T>
  67   T *Allocate(size_t Num) {
  68     return static_cast<T*>(Allocate(Num * sizeof(T), AlignOf<T>::Alignment));
  69   }
  70
  71   void Deallocate(void * /*Ptr*/) {}
  72
  73   void PrintStats() const;
  74 };
  75
  76 }  // end namespace llvm
  77
  78 #endif