113 lines
4.3 KiB
C++
113 lines
4.3 KiB
C++
#ifndef __SCOP_GPU_BUFFER__
|
|
#define __SCOP_GPU_BUFFER__
|
|
|
|
#include <kvf.h>
|
|
#include <Renderer/Enums.h>
|
|
#include <Core/Logs.h>
|
|
#include <Renderer/RenderCore.h>
|
|
#include <Utils/Buffer.h>
|
|
#include <Renderer/Memory/Block.h>
|
|
|
|
namespace Scop
|
|
{
|
|
class GPUBuffer
|
|
{
|
|
public:
|
|
GPUBuffer() = default;
|
|
|
|
void Init(BufferType type, VkDeviceSize size, VkBufferUsageFlags usage, CPUBuffer data, std::string_view name = {}, bool dedicated_alloc = false);
|
|
void Destroy() noexcept;
|
|
|
|
bool CopyFrom(const GPUBuffer& buffer, std::size_t src_offset = 0, std::size_t dst_offset = 0) noexcept;
|
|
|
|
void Swap(GPUBuffer& buffer) noexcept;
|
|
|
|
[[nodiscard]] inline void* GetMap() const noexcept { return m_memory.map; }
|
|
[[nodiscard]] inline VkBuffer operator()() const noexcept { return m_buffer; }
|
|
[[nodiscard]] inline VkBuffer Get() const noexcept { return m_buffer; }
|
|
[[nodiscard]] inline VkDeviceMemory GetMemory() const noexcept { return m_memory.memory; }
|
|
[[nodiscard]] inline VkDeviceSize GetSize() const noexcept { return m_size; }
|
|
[[nodiscard]] inline VkDeviceSize GetOffset() const noexcept { return 0; }
|
|
|
|
[[nodiscard]] inline static std::size_t GetBufferCount() noexcept { return s_buffer_count; }
|
|
|
|
[[nodiscard]] inline bool IsInit() const noexcept { return m_buffer != VK_NULL_HANDLE; }
|
|
|
|
~GPUBuffer() = default;
|
|
|
|
protected:
|
|
void PushToGPU() noexcept;
|
|
|
|
protected:
|
|
VkBuffer m_buffer = VK_NULL_HANDLE;
|
|
MemoryBlock m_memory = NULL_MEMORY_BLOCK;
|
|
VkDeviceSize m_size = 0;
|
|
|
|
private:
|
|
void CreateBuffer(VkDeviceSize size, VkBufferUsageFlags usage, VkMemoryPropertyFlags properties, std::string_view name, bool dedicated_alloc);
|
|
|
|
private:
|
|
inline static std::size_t s_buffer_count = 0;
|
|
|
|
std::string m_name;
|
|
|
|
VkBufferUsageFlags m_usage = 0;
|
|
VkMemoryPropertyFlags m_flags = 0;
|
|
bool m_is_dedicated_alloc = false;
|
|
};
|
|
|
|
class VertexBuffer : public GPUBuffer
|
|
{
|
|
public:
|
|
inline void Init(std::uint32_t size, VkBufferUsageFlags additional_flags = 0, std::string_view name = {}) { GPUBuffer::Init(BufferType::LowDynamic, size, VK_BUFFER_USAGE_VERTEX_BUFFER_BIT | additional_flags, {}, std::move(name)); }
|
|
void SetData(CPUBuffer data);
|
|
inline void Bind(VkCommandBuffer cmd) const noexcept { VkDeviceSize offset = 0; RenderCore::Get().vkCmdBindVertexBuffers(cmd, 0, 1, &m_buffer, &offset); }
|
|
};
|
|
|
|
class IndexBuffer : public GPUBuffer
|
|
{
|
|
public:
|
|
inline void Init(std::uint32_t size, VkBufferUsageFlags additional_flags = 0, std::string_view name = {}) { GPUBuffer::Init(BufferType::LowDynamic, size, VK_BUFFER_USAGE_INDEX_BUFFER_BIT | additional_flags, {}, std::move(name)); }
|
|
void SetData(CPUBuffer data);
|
|
inline void Bind(VkCommandBuffer cmd) const noexcept { RenderCore::Get().vkCmdBindIndexBuffer(cmd, m_buffer, 0, VK_INDEX_TYPE_UINT32); }
|
|
};
|
|
|
|
class MeshBuffer : public GPUBuffer
|
|
{
|
|
public:
|
|
inline void Init(std::uint32_t vertex_size, std::uint32_t index_size, VkBufferUsageFlags additional_flags = 0, CPUBuffer data = {}, std::string_view name = {})
|
|
{
|
|
m_vertex_offset = 0;
|
|
m_index_offset = vertex_size;
|
|
GPUBuffer::Init(BufferType::LowDynamic, vertex_size + index_size, VK_BUFFER_USAGE_VERTEX_BUFFER_BIT | VK_BUFFER_USAGE_INDEX_BUFFER_BIT | additional_flags, std::move(data), std::move(name), false);
|
|
}
|
|
void SetVertexData(CPUBuffer data);
|
|
void SetIndexData(CPUBuffer data);
|
|
inline void BindVertex(VkCommandBuffer cmd) const noexcept { RenderCore::Get().vkCmdBindVertexBuffers(cmd, 0, 1, &m_buffer, &m_vertex_offset); }
|
|
inline void BindIndex(VkCommandBuffer cmd) const noexcept { RenderCore::Get().vkCmdBindIndexBuffer(cmd, m_buffer, m_index_offset, VK_INDEX_TYPE_UINT32); }
|
|
|
|
private:
|
|
VkDeviceSize m_vertex_offset;
|
|
VkDeviceSize m_index_offset;
|
|
};
|
|
|
|
class UniformBuffer
|
|
{
|
|
public:
|
|
void Init(std::uint32_t size, std::string_view name = {});
|
|
void SetData(CPUBuffer data, std::size_t frame_index);
|
|
void Destroy() noexcept;
|
|
|
|
inline VkDeviceSize GetSize(int i) const noexcept { return m_buffers[i].GetSize(); }
|
|
inline VkDeviceSize GetOffset(int i) const noexcept { return m_buffers[i].GetOffset(); }
|
|
inline VkBuffer GetVk(int i) const noexcept { return m_buffers[i].Get(); }
|
|
inline GPUBuffer& Get(int i) noexcept { return m_buffers[i]; }
|
|
|
|
private:
|
|
std::array<GPUBuffer, MAX_FRAMES_IN_FLIGHT> m_buffers;
|
|
std::array<void*, MAX_FRAMES_IN_FLIGHT> m_maps;
|
|
};
|
|
}
|
|
|
|
#endif
|