KtxLoadingCallbackManager.h

#pragma once
 
// Standard.
#include <mutex>
#include <unordered_map>
#include <format>
 
// External.
#include "VulkanMemoryAllocator/include/vk_mem_alloc.h"
#include "ktxvulkan.h"
 
namespace ne {
    class KtxLoadingCallbackManager {
    public:
        static size_t getCurrentAllocationCount();
 
        static ktxVulkanTexture_subAllocatorCallbacks getKtxSubAllocatorCallbacks();
 
        static uint64_t allocMem(
            VkMemoryAllocateInfo* pAllocationInfo,
            VkMemoryRequirements* pMemoryRequirements,
            uint64_t* pPageCount);
 
        static VkResult bindBuffer(VkBuffer pBuffer, uint64_t iAllocationId);
 
        static VkResult bindImage(VkImage pImage, uint64_t iAllocationId);
 
        static VkResult
        memoryMap(uint64_t iAllocationId, uint64_t iPageNumber, VkDeviceSize* pMapLength, void** pData);
 
        static void memoryUnmap(uint64_t iAllocationId, uint64_t iPageNumber);
 
        static void freeMem(uint64_t iAllocationId);
 
    private:
        struct Data {
            std::unordered_map<uint64_t, std::pair<VmaAllocation, VkDeviceSize>> allocations;
 
            uint64_t iAllocationId = 0;
        };
 
        static VmaAllocator getMemoryAllocator();
 
        static VkPhysicalDevice getPhysicalDevice();
 
        static std::pair<std::recursive_mutex, Data> mtxData;
    };
} // namespace ne