Buffer.hpp

#pragma once
 
#include "PixelBullet/Graphics/Descriptors/DescriptorSet.hpp"
 
namespace PixelBullet
{
    class Buffer
    {
    public:
        enum class Status
        {
            Reset,
            Changed,
            Normal
        };
 
        Buffer(VkDeviceSize size, VkBufferUsageFlags usage, VkMemoryPropertyFlags properties,
               const void* data = nullptr);
        virtual ~Buffer();
 
        void MapMemory(void** data) const;
        void UnmapMemory() const;
 
        VkDeviceSize GetSize() const
        {
            return m_Size;
        }
        const VkBuffer& GetBuffer() const
        {
            return m_Buffer;
        }
        const VkDeviceMemory& GetBufferMemory() const
        {
            return m_BufferMemory;
        }
 
        static uint32_t FindMemoryType(uint32_t typeFilter, const VkMemoryPropertyFlags& requiredProperties);
 
        static void InsertBufferMemoryBarrier(const CommandBuffer& commandBuffer, const VkBuffer& buffer,
                                              VkAccessFlags srcAccessMask, VkAccessFlags dstAccessMask,
                                              VkPipelineStageFlags srcStageMask, VkPipelineStageFlags dstStageMask,
                                              VkDeviceSize offset = 0, VkDeviceSize size = VK_WHOLE_SIZE);
 
    protected:
        VkDeviceSize m_Size;
        VkBuffer m_Buffer = VK_NULL_HANDLE;
        VkDeviceMemory m_BufferMemory = VK_NULL_HANDLE;
    };
} // namespace PixelBullet