97 lines
3.4 KiB
C++
97 lines
3.4 KiB
C++
#pragma once
|
|
#include<hgl/type/BaseString.h>
|
|
#include<hgl/type/List.h>
|
|
#include<hgl/type/StringList.h>
|
|
#include<hgl/graph/vulkan/VKNamespace.h>
|
|
|
|
VK_NAMESPACE_BEGIN
|
|
struct ShaderStage
|
|
{
|
|
AnsiString name;
|
|
uint location;
|
|
|
|
uint base_type;
|
|
uint component;
|
|
|
|
VkFormat format;
|
|
};//struct ShaderStage
|
|
|
|
using ShaderStageList=ObjectList<ShaderStage>;
|
|
using ShaderBindingList=List<uint32_t>;
|
|
|
|
struct ShaderDescriptorList
|
|
{
|
|
AnsiStringList name_list;
|
|
ShaderBindingList binding_list;
|
|
};
|
|
|
|
#ifndef VK_DESCRIPTOR_TYPE_BEGIN_RANGE
|
|
constexpr size_t VK_DESCRIPTOR_TYPE_BEGIN_RANGE=VK_DESCRIPTOR_TYPE_SAMPLER;
|
|
#endif//VK_DESCRIPTOR_TYPE_BEGIN_RANGE
|
|
|
|
#ifndef VK_DESCRIPTOR_TYPE_END_RANGE
|
|
constexpr size_t VK_DESCRIPTOR_TYPE_END_RANGE=VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT;
|
|
#endif//VK_DESCRIPTOR_TYPE_END_RANGE
|
|
|
|
#ifndef VK_DESCRIPTOR_TYPE_RANGE_SIZE
|
|
constexpr size_t VK_DESCRIPTOR_TYPE_RANGE_SIZE=VK_DESCRIPTOR_TYPE_END_RANGE-VK_DESCRIPTOR_TYPE_BEGIN_RANGE+1;
|
|
#endif//VK_DESCRIPTOR_TYPE_RANGE_SIZE
|
|
|
|
class ShaderResource
|
|
{
|
|
const void *data;
|
|
|
|
VkShaderStageFlagBits stage_flag;
|
|
|
|
const void *spv_data;
|
|
uint32 spv_size;
|
|
|
|
ShaderStageList stage_inputs;
|
|
ShaderStageList stage_outputs;
|
|
|
|
ShaderDescriptorList descriptor_list[VK_DESCRIPTOR_TYPE_RANGE_SIZE];
|
|
|
|
public:
|
|
|
|
ShaderResource(const void *,const VkShaderStageFlagBits &,const void *,const uint32);
|
|
virtual ~ShaderResource();
|
|
|
|
const VkShaderStageFlagBits GetStage()const{return stage_flag;}
|
|
|
|
const uint32_t *GetCode ()const{return (uint32_t *)spv_data;}
|
|
const uint32_t GetCodeSize ()const{return spv_size;}
|
|
|
|
ShaderStageList &GetStageInputs(){return stage_inputs;}
|
|
ShaderStageList &GetStageOutputs(){return stage_outputs;}
|
|
|
|
const uint GetStageInputCount()const{return stage_inputs.GetCount();}
|
|
const uint GetStageOutputCount()const{return stage_outputs.GetCount();}
|
|
|
|
const ShaderStage * GetStageInput (const AnsiString &)const;
|
|
const int GetStageInputBinding(const AnsiString &)const;
|
|
|
|
const ShaderDescriptorList *GetDescriptorList()const{return descriptor_list;}
|
|
ShaderDescriptorList *GetDescriptorList(VkDescriptorType desc_type)
|
|
{
|
|
if(desc_type<VK_DESCRIPTOR_TYPE_BEGIN_RANGE
|
|
||desc_type>VK_DESCRIPTOR_TYPE_END_RANGE)return nullptr;
|
|
|
|
return descriptor_list+desc_type;
|
|
}
|
|
|
|
ShaderDescriptorList &GetUBO(){return descriptor_list[VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER];}
|
|
ShaderDescriptorList &GetSSBO(){return descriptor_list[VK_DESCRIPTOR_TYPE_STORAGE_BUFFER];}
|
|
ShaderDescriptorList &GetSampler(){return descriptor_list[VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER];}
|
|
|
|
const int GetBinding (VkDescriptorType desc_type,const AnsiString &name)const;
|
|
const ShaderBindingList * GetBindingList (VkDescriptorType desc_type)const
|
|
{
|
|
if(desc_type<VK_DESCRIPTOR_TYPE_BEGIN_RANGE
|
|
||desc_type>VK_DESCRIPTOR_TYPE_END_RANGE)return nullptr;
|
|
|
|
return &(descriptor_list[desc_type].binding_list);
|
|
}
|
|
};//class ShaderResource
|
|
|
|
ShaderResource *LoadShaderResoruce(const OSString &filename);
|
|
VK_NAMESPACE_END |