MemTypeAllocator.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <OsoMemoryProfiler/CogsMemoryProfile.h>
 
namespace Cogs::Memory
{
 
  template<class T, Cogs::MemBlockType memType>
  struct MemTypeAllocator
  {
    std::allocator<T> alloc;
 
    using value_type = T;
    using pointer = T*;
    using size_type = typename std::allocator<T>::size_type;
 
    MemTypeAllocator() = default;
 
    MemTypeAllocator(const MemTypeAllocator& other) : alloc(std::allocator<T>(other.alloc)) { }
 
    template<typename U>
    MemTypeAllocator(const MemTypeAllocator<U, memType>& other) : alloc(other.alloc) {}
 
    pointer allocate(size_type n)
    {
      pointer p = alloc.allocate(n);
      if (p) {
        OsoMP_Allocate(uint8_t(memType), p, n, nullptr, 0);
      }
      return p;
    }
 
    void deallocate(T* p, size_type n)
    {
      if (p) {
        OsoMP_Free(uint8_t(memType), p, nullptr, 0);
      }
      alloc.deallocate(p, n);
    }
 
    template<typename U>
    struct rebind { typedef MemTypeAllocator<U, memType> other; };
  };
 
  template<class T1, Cogs::MemBlockType memType1, class T2, Cogs::MemBlockType memType2>
  bool operator==(const MemTypeAllocator<T1, memType1>& lhs, const MemTypeAllocator<T2, memType2>& rhs) noexcept
  {
    return  memType1 == memType2 && lhs.alloc == rhs.alloc;
  }
}