allocationmanager.h

Go to the documentation of this file.

#ifndef ION_BASE_ALLOCATIONMANAGER_H_
#define ION_BASE_ALLOCATIONMANAGER_H_

#include "base/macros.h"
#include "ion/base/allocator.h"

namespace ion {
namespace base {

class ION_API AllocationManager {
 public:
  ~AllocationManager();

  static void SetDefaultAllocationLifetime(AllocationLifetime lifetime) {
    GetInstance()->default_allocation_lifetime_ = lifetime;
  }
  static AllocationLifetime GetDefaultAllocationLifetime() {
    return GetInstance()->default_allocation_lifetime_;
  }

  static void SetDefaultAllocatorForLifetime(AllocationLifetime lifetime,
                                             const AllocatorPtr& allocator) {
    AllocationManager* mgr = GetInstance();
    mgr->default_allocators_[lifetime] =
        allocator.Get() ? allocator : mgr->malloc_allocator_;
  }
  static const AllocatorPtr& GetDefaultAllocatorForLifetime(
      AllocationLifetime lifetime) {
    return GetInstance()->default_allocators_[lifetime];
  }

  static const AllocatorPtr& GetDefaultAllocator() {
    AllocationManager* mgr = GetInstance();
    return mgr->default_allocators_[mgr->default_allocation_lifetime_];
  }

  static const AllocatorPtr& GetMallocAllocator() {
    return GetInstance()->malloc_allocator_;
  }

  static const AllocatorPtr& GetNonNullAllocator(
      const AllocatorPtr& allocator) {
    return allocator.Get() ? allocator : GetDefaultAllocator();
  }

 private:
  class MallocAllocator;

  AllocationManager();

  static AllocationManager* GetInstance();

  AllocationLifetime default_allocation_lifetime_;

  AllocatorPtr default_allocators_[kNumAllocationLifetimes];

  AllocatorPtr malloc_allocator_;

  DISALLOW_COPY_AND_ASSIGN(AllocationManager);
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_ALLOCATIONMANAGER_H_