#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;  ///<基本类型(如vec4中的vec)
        uint        component;  ///<成份数量(如vec4中的4)

        VkFormat    format;     ///<对应的Vulkan格式(如vec4对应的FMT_RGBA32F)

        uint        binding;
    };//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