VirtualAllocator.hpp

Go to the documentation of this file.

#pragma once
#include <Async/Future.hpp>
#include <Core/Core.hpp>
#include <Core/Pool.hpp>
#include <vk_mem_alloc.h>
namespace Foundation::Rendering
{
    using namespace Core;
    using VirtualAllocation = uint32_t;
    // [Raw Offset, Size]
    constexpr VirtualAllocation kInvalidVirtualAllocation = ~0u;
    class VirtualAllocator
    {
        const size_t mSize;
        VmaVirtualBlock mBlock{};
        Pool<VirtualAllocation, Tuple<size_t, size_t, VmaVirtualAllocation>> mAllocs;
        // VMA virtual allocs are not thread-safe. Need guards.
        Async::Mutex mMutex;
 
    public:
        VirtualAllocator(size_t size, Allocator* alloc);
        VirtualAllocation Allocate(size_t size, size_t alignment);
        void Free(VirtualAllocation handle);
        Pair<size_t, size_t> Query(VirtualAllocation handle);
        size_t QuerySize(size_t handle)
        {
            auto [off, sz] = Query(handle);
            return sz;
        };
        size_t QueryOffset(size_t handle)
        {
            auto [off, sz] = Query(handle);
            return off;
        }
        ~VirtualAllocator();
    };
} // namespace Foundation::Rendering