107 lines
2.4 KiB
C++
107 lines
2.4 KiB
C++
#include<hgl/graph/VKShaderResource.h>
|
|
#include<hgl/graph/VKFormat.h>
|
|
#include<hgl/filesystem/FileSystem.h>
|
|
#include<hgl/type/Map.h>
|
|
#include<hgl/io/ConstBufferReader.h>
|
|
|
|
VK_NAMESPACE_BEGIN
|
|
|
|
namespace
|
|
{
|
|
ObjectMap<OSString,ShaderResource> shader_resource_by_filename;
|
|
|
|
const bool LoadShaderStageAttributes(ShaderAttributeArray &sa_array,io::ConstBufferReader &cbr)
|
|
{
|
|
uint count;
|
|
|
|
cbr.CastRead<uint8>(count);
|
|
|
|
if(count<=0)
|
|
return(false);
|
|
|
|
Init(&sa_array,count);
|
|
|
|
ShaderAttribute *ss=sa_array.items;
|
|
|
|
for(uint i=0;i<count;i++)
|
|
{
|
|
cbr.Read(ss->location);
|
|
cbr.CastRead<uint8>(ss->basetype);
|
|
cbr.CastRead<uint8>(ss->vec_size);
|
|
|
|
cbr.ReadTinyString(ss->name);
|
|
|
|
++ss;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}//namespcae
|
|
|
|
ShaderResource::ShaderResource(const VkShaderStageFlagBits &flag,const uint32_t *sd,const uint32 size)
|
|
{
|
|
stage_flag=flag;
|
|
spv_data=sd;
|
|
spv_size=size;
|
|
|
|
Init(stage_io);
|
|
}
|
|
|
|
ShaderResource::~ShaderResource()
|
|
{
|
|
Clear(stage_io);
|
|
}
|
|
|
|
const ShaderAttribute *ShaderResource::GetInput(const AnsiString &name) const
|
|
{
|
|
if(stage_io.input.count<=0)return(nullptr);
|
|
|
|
const ShaderAttribute *ss=stage_io.input.items;
|
|
|
|
for(uint i=0;i<stage_io.input.count;i++)
|
|
{
|
|
if(name==ss->name)
|
|
return ss;
|
|
|
|
++ss;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
const int ShaderResource::GetInputBinding(const AnsiString &name) const
|
|
{
|
|
if(stage_io.input.count<=0)return(-1);
|
|
|
|
const ShaderAttribute *ss=stage_io.input.items;
|
|
|
|
for(uint i=0;i<stage_io.input.count;i++)
|
|
{
|
|
if(name==ss->name)
|
|
return i;
|
|
|
|
++ss;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
ShaderResource *LoadShaderResource(io::ConstBufferReader &cbr)
|
|
{
|
|
VkShaderStageFlagBits flag;
|
|
uint32 spv_size;
|
|
|
|
cbr.CastRead<uint32>(flag);
|
|
cbr.Read(spv_size);
|
|
|
|
ShaderResource *sr=new ShaderResource(flag,(const uint32_t *)cbr.CurPointer(),spv_size);
|
|
|
|
cbr.Skip(spv_size);
|
|
|
|
LoadShaderStageAttributes(sr->GetInputs(),cbr);
|
|
// LoadShaderStageAttributes(sr->GetStageOutputs(),cbr);
|
|
|
|
return sr;
|
|
}
|
|
VK_NAMESPACE_END
|