52 lines
1.4 KiB
C
52 lines
1.4 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include<hgl/graph/VKShaderDescriptor.h>
|
|||
|
|
#include<hgl/graph/VKDescriptorSetType.h>
|
|||
|
|
#include<hgl/shadergen/ShaderCommon.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<UTF8String,ShaderDescriptor> descriptor_map;
|
|||
|
|
|
|||
|
|
List<const UBODescriptor *> ubo_list;
|
|||
|
|
List<const SamplerDescriptor *> sampler_list;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
const ShaderDescriptor *AddDescriptor(VkShaderStageFlagBits ssb,ShaderDescriptor *new_sd); ///<添加一个描述符，如果它本身存在，则返回false
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
using ShaderDescriptorSetArray=ShaderDescriptorSet[size_t(DescriptorSetType::RANGE_SIZE)];
|
|||
|
|
|
|||
|
|
ShaderDescriptorSetArray desc_set_array;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
MaterialDescriptorManager();
|
|||
|
|
~MaterialDescriptorManager()=default;
|
|||
|
|
|
|||
|
|
const UBODescriptor *AddUBO(VkShaderStageFlagBits ssb,DescriptorSetType type,UBODescriptor *sd);
|
|||
|
|
const SamplerDescriptor *AddSampler(VkShaderStageFlagBits ssb,DescriptorSetType type,SamplerDescriptor *sd);
|
|||
|
|
|
|||
|
|
const DescriptorSetType GetSetType(const AnsiString &)const;
|
|||
|
|
|
|||
|
|
void Resort(); //排序产生set号与binding号
|
|||
|
|
};
|
|||
|
|
SHADERGEN_NAMESPACE_END
|