ULRE/inc/hgl/graph/vulkan/VKBuffer.h

103 lines
2.2 KiB
C++

#ifndef HGL_GRAPH_VULKAN_BUFFER_INCLUDE
#define HGL_GRAPH_VULKAN_BUFFER_INCLUDE
#include"VK.h"
VK_NAMESPACE_BEGIN
struct VulkanBuffer
{
VkBuffer buffer;
VkDeviceMemory memory;
VkDescriptorBufferInfo info;
};//struct VulkanBuffer
class Buffer
{
protected:
VkDevice device;
VulkanBuffer buf;
private:
friend class Device;
friend class VertexBuffer;
friend class IndexBuffer;
Buffer(VkDevice d,const VulkanBuffer &vb)
{
device=d;
buf=vb;
}
public:
virtual ~Buffer();
operator VkBuffer (){return buf.buffer;}
operator VkDeviceMemory (){return buf.memory;}
operator VkDescriptorBufferInfo * (){return &buf.info;}
virtual uint8_t *Map(uint32_t start=0,uint32_t size=0);
void Unmap();
bool Write(const void *ptr,uint32_t start,uint32_t size);
bool Write(const void *ptr){return Write(ptr,0,buf.info.range);}
};//class Buffer
class VertexBuffer:public Buffer
{
VkFormat format; ///<数据格式
uint32_t stride; ///<单个数据字节数
uint32_t count; ///<数据数量
private:
friend class Device;
VertexBuffer(VkDevice d,const VulkanBuffer &vb,VkFormat fmt,uint32_t _stride,uint32_t _count):Buffer(d,vb)
{
format=fmt;
stride=_stride;
count=_count;
}
public:
~VertexBuffer()=default;
const VkFormat GetFormat()const { return format; }
const uint32_t GetStride()const { return stride; }
const uint32_t GetCount ()const { return count; }
uint8_t *Map(uint32_t start=0,uint32_t size=0) override
{
return Buffer::Map(start*stride,size*stride);
}
};//class VertexBuffer:public Buffer
class IndexBuffer:public Buffer
{
VkIndexType index_type;
uint32_t count;
private:
friend class Device;
IndexBuffer(VkDevice d,const VulkanBuffer &vb,VkIndexType it,uint32_t _count):Buffer(d,vb)
{
index_type=it;
count=_count;
}
public:
~IndexBuffer()=default;
const VkIndexType GetType ()const{return index_type;}
const uint32 GetCount()const{return count;}
};//class IndexBuffer:public Buffer
VK_NAMESPACE_END
#endif//HGL_GRAPH_VULKAN_BUFFER_INCLUDE