TransferManager.hpp

/*
** ETIB PROJECT, 2026
** evan
** File description:
** TransferManager
*/
 
#pragma once
 
#include "evan/EvanPlatform.hpp"
 
#include <cstdint>
#include <vector>
 
namespace evan
{
    class TransferManager
    {
        public:
        TransferManager() = default;
        ~TransferManager();
 
        TransferManager(const TransferManager &) = delete;
        TransferManager &operator=(const TransferManager &) = delete;
 
        void init(VkDevice device, VkPhysicalDevice physicalDevice,
                  VkQueue graphicsQueue, uint32_t graphicsFamily);
 
        void destroy();
 
        void uploadBuffer(const void *data, VkDeviceSize size, VkBuffer dst,
                          VkDeviceSize dstOffset = 0);
 
        void uploadBufferToImage(const void *data, VkDeviceSize size,
                                 VkImage dst,
                                 const VkBufferImageCopy &region);
 
        void transitionImageLayout(VkImage image, VkFormat format,
                                   VkImageLayout oldLayout,
                                   VkImageLayout newLayout,
                                   uint32_t mipLevels);
 
        void generateMipmaps(VkImage image, VkFormat format, uint32_t width,
                             uint32_t height, uint32_t mipLevels);
 
        void flush();
 
        void waitIdle();
 
        private:
        struct PendingBatch {
            VkFence fence = VK_NULL_HANDLE;
            VkCommandBuffer commandBuffer = VK_NULL_HANDLE;
        };
 
        void ensureBegan();
        void recycle();
        void createStagingBuffer(VkDeviceSize size);
 
        VkDevice _device = VK_NULL_HANDLE;
        VkPhysicalDevice _physicalDevice = VK_NULL_HANDLE;
        VkQueue _transferQueue = VK_NULL_HANDLE;
        uint32_t _transferFamily = 0;
        VkCommandPool _commandPool = VK_NULL_HANDLE;
        VkCommandBuffer _commandBuffer = VK_NULL_HANDLE;
        bool _recording = false;
        bool _dirty = false;
 
        VkBuffer _stagingBuffer = VK_NULL_HANDLE;
        VkDeviceMemory _stagingMemory = VK_NULL_HANDLE;
        void *_stagingMapped = nullptr;
        VkDeviceSize _stagingSize = 0;
        VkDeviceSize _stagingOffset = 0;
 
        std::vector<PendingBatch> _pending;
    };
}    // namespace evan