Layout codes

This commit is contained in:
hyzboy 2024-03-26 01:22:58 +08:00
parent a60ffbb7da
commit 8492b886c9

View File

@ -1,24 +1,25 @@
#include "MaterialFileData.h" #include"MaterialFileData.h"
#include "Std3DMaterial.h" #include"Std3DMaterial.h"
#include <hgl/shadergen/MaterialCreateInfo.h> #include<hgl/shadergen/MaterialCreateInfo.h>
STD_MTL_NAMESPACE_BEGIN STD_MTL_NAMESPACE_BEGIN
const char *GetUBOCodes(const AccumMemoryManager::Block *block); const char *GetUBOCodes(const AccumMemoryManager::Block *block);
namespace { namespace
class Std3DMaterialLoader : public Std3DMaterial
{ {
protected: class Std3DMaterialLoader: public Std3DMaterial
{
protected:
material_file::MaterialFileData *mfd; material_file::MaterialFileData *mfd;
public: public:
Std3DMaterialLoader(material_file::MaterialFileData* data, const Material3DCreateConfig* cfg) Std3DMaterialLoader(material_file::MaterialFileData *data,const Material3DCreateConfig *cfg)
: Std3DMaterial(cfg) : Std3DMaterial(cfg)
{ {
mfd = data; mfd=data;
} }
~Std3DMaterialLoader() ~Std3DMaterialLoader()
@ -28,7 +29,7 @@ public:
bool BeginCustomShader() override bool BeginCustomShader() override
{ {
if (!Std3DMaterial::BeginCustomShader()) if(!Std3DMaterial::BeginCustomShader())
return (false); return (false);
for(const auto ubo:mfd->ubo_list) for(const auto ubo:mfd->ubo_list)
@ -43,9 +44,9 @@ public:
ubo.name); ubo.name);
} }
if (mfd->mi.mi_bytes > 0) if(mfd->mi.mi_bytes>0)
{ {
mci->SetMaterialInstance( mfd->mi.code, mci->SetMaterialInstance(mfd->mi.code,
mfd->mi.mi_bytes, mfd->mi.mi_bytes,
mfd->mi.shader_stage_flag_bits); mfd->mi.shader_stage_flag_bits);
} }
@ -53,72 +54,72 @@ public:
return true; return true;
} }
material_file::ShaderData* CommonProc(VkShaderStageFlagBits ss, ShaderCreateInfo* sc) material_file::ShaderData *CommonProc(VkShaderStageFlagBits ss,ShaderCreateInfo *sc)
{ {
material_file::ShaderData* sd = mfd->shader[ss]; material_file::ShaderData *sd=mfd->shader[ss];
if (!sd) if(!sd)
return (nullptr); return (nullptr);
for (auto& ua : sd->output) for(auto &ua:sd->output)
sc->AddOutput(ua.vat, ua.name); sc->AddOutput(ua.vat,ua.name);
for (auto& s : sd->sampler) for(auto &s:sd->sampler)
mci->AddSampler(ss, DescriptorSetType::PerMaterial, s.type, s.name); mci->AddSampler(ss,DescriptorSetType::PerMaterial,s.type,s.name);
sc->SetMain(sd->code, sd->code_length); // 这里会产生复制这个string但我不希望产生这个。未来想办法改进 sc->SetMain(sd->code,sd->code_length); // 这里会产生复制这个string但我不希望产生这个。未来想办法改进
return sd; return sd;
} }
bool CustomVertexShader(ShaderCreateInfoVertex* vsc) override bool CustomVertexShader(ShaderCreateInfoVertex *vsc) override
{ {
for (auto& ua : mfd->vi) for(auto &ua:mfd->vi)
vsc->AddInput(ua.vat, ua.name); vsc->AddInput(ua.vat,ua.name);
if(mfd->mi.mi_bytes>0) if(mfd->mi.mi_bytes>0)
vsc->AddMaterialInstanceID(); vsc->AddMaterialInstanceID();
if (!Std3DMaterial::CustomVertexShader(vsc)) if(!Std3DMaterial::CustomVertexShader(vsc))
return (false); return (false);
return CommonProc(VK_SHADER_STAGE_VERTEX_BIT, vsc); return CommonProc(VK_SHADER_STAGE_VERTEX_BIT,vsc);
} }
bool CustomGeometryShader(ShaderCreateInfoGeometry* gsc) override bool CustomGeometryShader(ShaderCreateInfoGeometry *gsc) override
{ {
material_file::GeometryShaderData* sd = (material_file::GeometryShaderData*)CommonProc(VK_SHADER_STAGE_GEOMETRY_BIT, gsc); material_file::GeometryShaderData *sd=(material_file::GeometryShaderData *)CommonProc(VK_SHADER_STAGE_GEOMETRY_BIT,gsc);
if (!sd) if(!sd)
return (false); return (false);
gsc->SetGeom(sd->input_prim, sd->output_prim, sd->max_vertices); gsc->SetGeom(sd->input_prim,sd->output_prim,sd->max_vertices);
return true; return true;
} }
bool CustomFragmentShader(ShaderCreateInfoFragment* fsc) override bool CustomFragmentShader(ShaderCreateInfoFragment *fsc) override
{ {
return CommonProc(VK_SHADER_STAGE_FRAGMENT_BIT, fsc); return CommonProc(VK_SHADER_STAGE_FRAGMENT_BIT,fsc);
} }
}; // class Std3DMaterialLoader:public Std3DMaterial }; // class Std3DMaterialLoader:public Std3DMaterial
} // namespace } // namespace
material_file::MaterialFileData* LoadMaterialDataFromFile(const AnsiString& mtl_filename); material_file::MaterialFileData *LoadMaterialDataFromFile(const AnsiString &mtl_filename);
MaterialCreateInfo* LoadMaterialFromFile(const AnsiString& name, Material3DCreateConfig* cfg) MaterialCreateInfo *LoadMaterialFromFile(const AnsiString &name,Material3DCreateConfig *cfg)
{ {
if (name.IsEmpty() || !cfg) if(name.IsEmpty()||!cfg)
return (nullptr); return (nullptr);
material_file::MaterialFileData* mfd = LoadMaterialDataFromFile(name); material_file::MaterialFileData *mfd=LoadMaterialDataFromFile(name);
if (!mfd) if(!mfd)
return nullptr; return nullptr;
cfg->shader_stage_flag_bit = mfd->shader_stage_flag_bit; cfg->shader_stage_flag_bit=mfd->shader_stage_flag_bit;
Std3DMaterialLoader mtl(mfd, cfg); Std3DMaterialLoader mtl(mfd,cfg);
return mtl.Create(); return mtl.Create();
} }