75 lines
2.0 KiB
C++
75 lines
2.0 KiB
C++
#pragma once
|
||
|
||
#include<hgl/graph/VKShaderDescriptor.h>
|
||
#include<hgl/graph/VKDescriptorSetType.h>
|
||
#include<hgl/shadergen/ShaderGenNamespace.h>
|
||
#include<hgl/type/Map.h>
|
||
|
||
using namespace hgl;
|
||
using namespace hgl::graph;
|
||
|
||
SHADERGEN_NAMESPACE_BEGIN
|
||
/**
|
||
* 材质描述符管理</p>
|
||
* 该类使用于SHADER生成前，用于统计编号set/binding
|
||
*/
|
||
class MaterialDescriptorManager
|
||
{
|
||
struct ShaderDescriptorSet
|
||
{
|
||
DescriptorSetType set_type;
|
||
|
||
int set;
|
||
int count;
|
||
|
||
ObjectMap<AnsiString,ShaderDescriptor> descriptor_map;
|
||
|
||
public:
|
||
|
||
ShaderDescriptor *AddDescriptor(VkShaderStageFlagBits ssb,ShaderDescriptor *new_sd); ///<添加一个描述符，如果它本身存在，则返回false
|
||
};
|
||
|
||
using ShaderDescriptorSetArray=ShaderDescriptorSet[size_t(DescriptorSetType::RANGE_SIZE)];
|
||
|
||
ShaderDescriptorSetArray desc_set_array;
|
||
|
||
Map<AnsiString,AnsiString> ubo_struct_map;
|
||
Map<AnsiString,UBODescriptor *> ubo_map;
|
||
Map<AnsiString,SamplerDescriptor *> sampler_map;
|
||
|
||
public:
|
||
|
||
MaterialDescriptorManager();
|
||
~MaterialDescriptorManager()=default;
|
||
|
||
bool AddUBOStruct(const AnsiString &name,const AnsiString &code)
|
||
{
|
||
if(ubo_struct_map.KeyExist(name))
|
||
return(false);
|
||
|
||
ubo_struct_map.Add(name,code);
|
||
return(true);
|
||
}
|
||
|
||
bool GetUBOStruct(const AnsiString &name,AnsiString &code) const
|
||
{
|
||
return(ubo_struct_map.Get(name,code));
|
||
}
|
||
|
||
bool hasUBOStruct(const AnsiString &name) const
|
||
{
|
||
return(ubo_struct_map.KeyExist(name));
|
||
}
|
||
|
||
const UBODescriptor *AddUBO(VkShaderStageFlagBits ssb,DescriptorSetType set_type,UBODescriptor *sd);
|
||
const SamplerDescriptor *AddSampler(VkShaderStageFlagBits ssb,DescriptorSetType set_type,SamplerDescriptor *sd);
|
||
|
||
UBODescriptor *GetUBO(const AnsiString &name);
|
||
SamplerDescriptor *GetSampler(const AnsiString &name);
|
||
|
||
const DescriptorSetType GetSetType(const AnsiString &)const;
|
||
|
||
void Resort(); //排序产生set号与binding号
|
||
};
|
||
SHADERGEN_NAMESPACE_END
|