foray_dualbuffer.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_managedbuffer.hpp"
#include "../foray_basics.hpp"
 
namespace foray::util {
 
    class DualBuffer : public NoMoveDefaults
    {
      public:
        void Create(core::Context* context, const core::ManagedBuffer::CreateInfo& devicebufferCreateInfo, uint32_t stageBufferCount = INFLIGHT_FRAME_COUNT);
 
        virtual void StageFullBuffer(uint32_t frameIndex, const void* data);
 
        virtual void StageSection(uint32_t frameIndex, const void* data, size_t destOffset, size_t size);
 
        virtual void CmdCopyToDevice(uint32_t frameIndex, VkCommandBuffer cmdBuffer);
 
        virtual void CmdPrepareForRead(VkCommandBuffer cmdBuffer, VkPipelineStageFlags2 dstStageMask, VkAccessFlags2 dstAccessMask) const;
 
        virtual VkBufferMemoryBarrier2 MakeBarrierPrepareForRead(VkPipelineStageFlags2 dstStageMask, VkAccessFlags2 dstAccessMask) const;
 
        inline bool Exists() const { return mDeviceBuffer.Exists(); }
 
        inline std::string GetName() const { return mDeviceBuffer.GetName(); }
        DualBuffer&        SetName(std::string_view name);
 
        inline VkDescriptorBufferInfo GetVkDescriptorInfo() const { return VkDescriptorBufferInfo{.buffer = mDeviceBuffer.GetBuffer(), .offset = 0U, .range = VK_WHOLE_SIZE}; }
        inline VkBuffer GetDeviceVkBuffer() const { return mDeviceBuffer.GetBuffer(); }
 
        void Destroy();
 
        inline virtual ~DualBuffer() { Destroy(); }
 
        FORAY_GETTER_CR(DeviceBuffer)
 
      protected:
        std::vector<void*> mStagingBufferMaps;
        std::vector<std::unique_ptr<core::ManagedBuffer>> mStagingBuffers;
        std::vector<std::vector<VkBufferCopy>> mBufferCopies;
        core::ManagedBuffer mDeviceBuffer;
    };
}  // namespace foray::util