Memory.h

#pragma once
#include <UnCompute/Memory/Ptr.h>
#include <UnCompute/Memory/SystemAllocator.h>
 
namespace UN
{
    template<class T, class TAllocator, class... Args>
    inline T* AllocateObjectEx(TAllocator* pAllocator, Args&&... args)
    {
        USize counterSize = AlignUp(sizeof(ReferenceCounter), alignof(T));
        USize wholeSize   = sizeof(T) + counterSize;
 
        auto* ptr     = static_cast<UInt8*>(pAllocator->Allocate(wholeSize, alignof(T)));
        auto* counter = new (ptr) ReferenceCounter(pAllocator);
 
        T* object = new (ptr + counterSize) T(std::forward<Args>(args)...);
        object->AttachRefCounter(counter);
        return object;
    }
 
    template<class T, class... Args>
    inline T* AllocateObject(Args&&... args)
    {
        return AllocateObjectEx<T, SystemAllocator>(SystemAllocator::Get(), std::forward<Args>(args)...);
    }
} // namespace UN