allocator.h

Go to the documentation of this file.

#ifndef ION_BASE_ALLOCATOR_H_
#define ION_BASE_ALLOCATOR_H_

#include "base/macros.h"
#include "ion/base/allocationtracker.h"
#include "ion/base/logging.h"
#include "ion/base/shareable.h"
#include "ion/base/sharedptr.h"

namespace ion {
namespace base {

enum AllocationLifetime {
  kShortTerm,

  kMediumTerm,

  kLongTerm,
};

static const int kNumAllocationLifetimes = kLongTerm + 1;

class Allocator;
typedef SharedPtr<Allocator> AllocatorPtr;

class ION_API Allocator : public Shareable {
 public:
  void* AllocateMemory(size_t size);

  void DeallocateMemory(void* p);

  virtual const AllocatorPtr& GetAllocatorForLifetime(
      AllocationLifetime lifetime) const;

  void SetTracker(const AllocationTrackerPtr& tracker) { tracker_ = tracker; }
  const AllocationTrackerPtr& GetTracker() const { return tracker_; }

 protected:
  Allocator() {}

  ~Allocator() override;

  virtual void* Allocate(size_t size) = 0;

  virtual void Deallocate(void* p) = 0;

 private:
  AllocationTrackerPtr tracker_;

  DISALLOW_COPY_AND_ASSIGN(Allocator);
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_ALLOCATOR_H_