105 lines
3.0 KiB
C++
105 lines
3.0 KiB
C++
#include"VKVertexInput.h"
|
||
#include"VKBuffer.h"
|
||
|
||
VK_NAMESPACE_BEGIN
|
||
|
||
int VertexInputState::Add(const UTF8String &name,const uint32_t shader_location,const VkFormat format,uint32_t offset,bool instance)
|
||
{
|
||
const int binding_index=binding_list.GetCount(); //参考opengl vab,binding_index必须从0开始，紧密排列。对应在vkCmdBindVertexBuffer中的缓冲区索引
|
||
|
||
binding_list.SetCount(binding_index+1);
|
||
attribute_list.SetCount(binding_index+1);
|
||
|
||
VkVertexInputBindingDescription *binding=binding_list.GetData()+binding_index;
|
||
VkVertexInputAttributeDescription *attrib=attribute_list.GetData()+binding_index;
|
||
|
||
binding->binding=binding_index;
|
||
binding->stride=GetStrideByFormat(format);
|
||
binding->inputRate=instance?VK_VERTEX_INPUT_RATE_INSTANCE:VK_VERTEX_INPUT_RATE_VERTEX;
|
||
|
||
//实际使用可以一个binding绑多个attrib，但我们仅支持1v1。
|
||
//一个binding是指在vertex shader中，由一个vertex输入流输入数据，attrib指其中的数据成分
|
||
//比如在一个流中传递{pos,color}这样两个数据，就需要两个attrib
|
||
//但我们在一个流中，仅支持一个attrib传递
|
||
|
||
attrib->binding=binding_index;
|
||
attrib->location=shader_location;
|
||
attrib->format=format;
|
||
attrib->offset=offset;
|
||
|
||
stage_input_locations.Add(name,StageInput(binding_index,shader_location,format));
|
||
|
||
return binding_index;
|
||
}
|
||
|
||
const int VertexInputState::GetLocation(const UTF8String &name)const
|
||
{
|
||
if(name.IsEmpty())return -1;
|
||
|
||
StageInput si;
|
||
|
||
if(!stage_input_locations.Get(name,si))
|
||
return -1;
|
||
|
||
return si.location;
|
||
}
|
||
|
||
const int VertexInputState::GetBinding(const UTF8String &name)const
|
||
{
|
||
if(name.IsEmpty())return -1;
|
||
|
||
StageInput si;
|
||
|
||
if(!stage_input_locations.Get(name,si))
|
||
return -1;
|
||
|
||
return si.binding;
|
||
}
|
||
|
||
void VertexInputState::Write(VkPipelineVertexInputStateCreateInfo &vis) const
|
||
{
|
||
vis.sType = VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_STATE_CREATE_INFO;
|
||
|
||
vis.vertexBindingDescriptionCount = binding_list.GetCount();
|
||
vis.pVertexBindingDescriptions = binding_list.GetData();
|
||
|
||
vis.vertexAttributeDescriptionCount = attribute_list.GetCount();
|
||
vis.pVertexAttributeDescriptions = attribute_list.GetData();
|
||
}
|
||
|
||
VertexInput::VertexInput(VertexInputState *state)
|
||
{
|
||
vis=state;
|
||
|
||
if(!vis)
|
||
return;
|
||
|
||
buf_count=vis->GetCount();
|
||
|
||
buf_list=hgl_zero_new<VkBuffer>(buf_count);
|
||
buf_offset=hgl_zero_new<VkDeviceSize>(buf_count);
|
||
}
|
||
|
||
VertexInput::~VertexInput()
|
||
{
|
||
delete[] buf_offset;
|
||
delete[] buf_list;
|
||
}
|
||
|
||
bool VertexInput::Set(int index,VertexBuffer *buf,VkDeviceSize offset)
|
||
{
|
||
if(index<0||index>=buf_count)return(false);
|
||
|
||
VkVertexInputBindingDescription *desc=vis->GetDesc(index);
|
||
VkVertexInputAttributeDescription *attr=vis->GetAttr(index);
|
||
|
||
if(buf->GetFormat()!=attr->format)return(false);
|
||
if(buf->GetStride()!=desc->stride)return(false);
|
||
|
||
buf_list[index]=*buf;
|
||
buf_offset[index]=offset;
|
||
|
||
return(true);
|
||
}
|
||
VK_NAMESPACE_END
|