VulkanStagingRing.h

#pragma once
#include "VulkanDevice.h"
namespace NDEVR
{
    struct CopyStep
    {
        Buffer<VkBufferCopy> regions;   
        bool is_src;                    
        CopyStep(bool is_src)
            : is_src(is_src)
        {}
    };

    struct CopyOperations
    {
        void add(bool is_src, VkBufferCopy item);
        Buffer<CopyStep> ops;   
    };

    class VulkanStagingRing
    {
        enum CMDBufferState
        {
              e_has_pending     
            , e_is_running      
            , e_empty           
        };
    public:
        VulkanStagingRing(const VulkanDevice* device);
        ~VulkanStagingRing();
        void ensureSize(VkDeviceSize size);
        void setSize(VkDeviceSize size);
        void* mapData(VkDeviceSize size);
        void setData(const void* data, VkDeviceSize byte_size, VkDeviceSize alignment = 1);
        VkCommandBuffer getCommandBuffer();
        void cleanup();
        void begin();
        void setDeviceLost();
        void end();
        void sendData(VkBuffer dst, VkDeviceSize dst_offset_bytes);
        void sendData(VkImage dst, Vector<2, uint04> image_size, uint04 mip_level, uint04 image_layer, VkCommandBuffer command_buffer);
        void removePending(VkBuffer buffer);
        void moveData(VkBuffer target, VkDeviceSize src_offset, VkDeviceSize dst_offset, VkDeviceSize size);
        void ensureComplete();
    private:
        void record(uint04 idx);
        void submit();
        void waitForComplete();
    private:
        void* m_ptr = nullptr;                                      
        VkCommandPool m_command_pool = VK_NULL_HANDLE;              
        VkBuffer m_local_buffer = VK_NULL_HANDLE;                   
        VkDeviceMemory m_memory = VK_NULL_HANDLE;                   
        VkDeviceSize m_size = 0;                                    
        VkDeviceSize m_head = 0;                                    
        VkDeviceSize m_atom = 1;                                    
        VkDeviceSize m_pending_size = 0;                            
        Dictionary<VkBuffer, CopyOperations> m_regions_to_copy;     
        Dictionary<VkBuffer, CopyOperations> m_regions_copying;     
        Buffer<VkCommandBuffer> m_command_buffers;                  
        Buffer<CMDBufferState> m_cmd_buffer_states;                 
        VkFence m_fence = VK_NULL_HANDLE;                           
        uint04 m_next_cmd_buffer = 0;                               
        const VulkanDevice* m_device;                               
        bool m_coherent = true;                                     
        bool m_needs_fence_wait = false;                            
        bool m_is_running = false;                                  
#ifdef _DEBUG
        uint04 m_start_thread = Constant<uint04>::Invalid;
#endif
    };
}