optimized codes about CreateMaterial

2023-06-14 16:49:19 +08:00
parent b021b05cb0
commit 82f3ad070b
8 changed files with 126 additions and 136 deletions
--- a/src/SceneGraph/Vulkan/VKDeviceMaterial.cpp
+++ b/src/SceneGraph/Vulkan/VKDeviceMaterial.cpp
@@ -35,86 +35,23 @@ namespace

        return(new DescriptorSet(device,binding_count,pld->pipeline_layout,desc_set));
    }
-
-    void CreateShaderStageList(List<VkPipelineShaderStageCreateInfo> &shader_stage_list,ShaderModuleMap *shader_maps)
-    {
-        const ShaderModule *sm;
-
-        const int shader_count=shader_maps->GetCount();
-        shader_stage_list.SetCount(shader_count);
-    
-        VkPipelineShaderStageCreateInfo *p=shader_stage_list.GetData();        
-
-        auto **itp=shader_maps->GetDataList();
-        for(int i=0;i<shader_count;i++)
-        {
-            sm=(*itp)->value;
-            hgl_cpy(p,sm->GetCreateInfo(),1);
-
-            ++p;
-            ++itp;
-        }
-    }
-
-    MaterialParameters *CreateMP(VkDevice device,VkDescriptorPool desc_pool,const MaterialDescriptorManager *desc_manager,const PipelineLayoutData *pld,const DescriptorSetType &desc_set_type)
-    {
-        if(!desc_manager||!pld)return(nullptr);
-        RANGE_CHECK_RETURN_NULLPTR(desc_set_type)
-
-        DescriptorSet *ds=CreateDS(device,desc_pool,pld,desc_set_type);
-
-        if(!ds)return(nullptr);
-
-    #ifdef _DEBUG
-        const UTF8String addr_string=HexToString<char,uint64_t>((uint64_t)(ds->GetDescriptorSet()));
-
-        LOG_INFO(U8_TEXT("Create [DescriptSets:")+addr_string+("] OK! Material Name: \"")+desc_manager->GetMaterialName()+U8_TEXT("\" Type: ")+GetDescriptorSetTypeName(desc_set_type));
-    #endif//_DEBUG
-
-        return(new MaterialParameters(desc_manager,desc_set_type,ds));
-    }
 }//namespace

-Material *GPUDevice::CreateMaterial(const UTF8String &mtl_name,ShaderModuleMap *shader_maps,MaterialDescriptorManager *desc_manager,VertexInput *vi)
+MaterialParameters *GPUDevice::CreateMP(const MaterialDescriptorManager *desc_manager,const PipelineLayoutData *pld,const DescriptorSetType &desc_set_type)
 {
-    const int shader_count=shader_maps->GetCount();
+    if(!desc_manager||!pld)return(nullptr);
+    RANGE_CHECK_RETURN_NULLPTR(desc_set_type)

-    if(shader_count<1)
-        return(nullptr);
+    DescriptorSet *ds=CreateDS(attr->device,attr->desc_pool,pld,desc_set_type);

-    PipelineLayoutData *pld=CreatePipelineLayoutData(attr->device,desc_manager);
-    
-    if(!pld)
-    {
-        delete shader_maps;
-        SAFE_CLEAR(desc_manager);
-        return(nullptr);
-    }
+    if(!ds)return(nullptr);

-    MaterialData *data=new MaterialData;
+#ifdef _DEBUG
+    const UTF8String addr_string=HexToString<char,uint64_t>((uint64_t)(ds->GetDescriptorSet()));

-    data->name          =mtl_name;
-    data->shader_maps   =shader_maps;
-    data->desc_manager  =desc_manager;
-    data->vertex_input  =vi;
+    LOG_INFO(U8_TEXT("Create [DescriptSets:")+addr_string+("] OK! Material Name: \"")+desc_manager->GetMaterialName()+U8_TEXT("\" Type: ")+GetDescriptorSetTypeName(desc_set_type));
+#endif//_DEBUG

-    CreateShaderStageList(data->shader_stage_list,shader_maps);
-
-    data->pipeline_layout_data=pld;
-
-    if(desc_manager)
-    {
-        ENUM_CLASS_FOR(DescriptorSetType,int,dst)
-        {
-            if(desc_manager->hasSet((DescriptorSetType)dst))
-                data->mp_array[dst]=CreateMP(attr->device,attr->desc_pool,desc_manager,pld,(DescriptorSetType)dst);
-            else
-                data->mp_array[dst]=nullptr;
-        }
-    }
-    else
-        hgl_zero(data->mp_array);
-
-    return(new Material(data));
+    return(new MaterialParameters(desc_manager,desc_set_type,ds));
 }
 VK_NAMESPACE_END
--- a/src/SceneGraph/Vulkan/VKMaterial.cpp
+++ b/src/SceneGraph/Vulkan/VKMaterial.cpp
@@ -4,16 +4,33 @@
 #include<hgl/graph/VKVertexInput.h>
 #include"VKPipelineLayoutData.h"
 VK_NAMESPACE_BEGIN
+MaterialData::MaterialData(const AnsiString &n)
+{
+    name=n;
+
+    vertex_input=nullptr;
+    shader_maps=new ShaderModuleMap;
+    desc_manager=nullptr;
+    pipeline_layout_data=nullptr;
+
+    hgl_zero(mp_array);
+
+    mi_data_bytes=0;
+    mi_max_count=0;
+}
+
 MaterialData::~MaterialData()
 {
-    for(int i=0;i<DESCRIPTOR_SET_TYPE_COUNT;i++)
-        if(mp_array[i])
-            delete mp_array[i];
-
-    delete shader_maps;
+    SAFE_CLEAR(vertex_input);
+    delete shader_maps;             //不用SAFE_CLEAR是因为这个一定会有
    SAFE_CLEAR(desc_manager);
+    SAFE_CLEAR(pipeline_layout_data);

-    delete vertex_input;
+    for(int i=0;i<DESCRIPTOR_SET_TYPE_COUNT;i++)
+        SAFE_CLEAR(mp_array[i]);
+
+    mi_data_bytes=0;
+    mi_max_count=0;
 }

 Material::~Material()
--- a/src/SceneGraph/Vulkan/VKPipelineLayoutData.cpp
+++ b/src/SceneGraph/Vulkan/VKPipelineLayoutData.cpp
@@ -4,7 +4,7 @@
 #include<hgl/graph/VKMaterialDescriptorManager.h>

 VK_NAMESPACE_BEGIN
-PipelineLayoutData *CreatePipelineLayoutData(VkDevice device,const MaterialDescriptorManager *desc_manager)
+PipelineLayoutData *GPUDevice::CreatePipelineLayoutData(const MaterialDescriptorManager *desc_manager)
 {
    PipelineLayoutData *pld=hgl_zero_new<PipelineLayoutData>();

@@ -18,9 +18,9 @@ PipelineLayoutData *CreatePipelineLayoutData(VkDevice device,const MaterialDescr
                continue;

            if(pld->layouts[i])
-                vkDestroyDescriptorSetLayout(device,pld->layouts[i],nullptr);
+                vkDestroyDescriptorSetLayout(attr->device,pld->layouts[i],nullptr);

-            if(vkCreateDescriptorSetLayout(device,dslci,nullptr,pld->layouts+i)!=VK_SUCCESS)
+            if(vkCreateDescriptorSetLayout(attr->device,dslci,nullptr,pld->layouts+i)!=VK_SUCCESS)
            {
                delete pld;
                return(nullptr);
@@ -56,9 +56,9 @@ PipelineLayoutData *CreatePipelineLayoutData(VkDevice device,const MaterialDescr
    pPipelineLayoutCreateInfo.pushConstantRangeCount    = 0;//1;
    pPipelineLayoutCreateInfo.pPushConstantRanges       = nullptr;//&push_constant_range;

-    pld->device=device;
+    pld->device=attr->device;

-    if(vkCreatePipelineLayout(device,&pPipelineLayoutCreateInfo,nullptr,&(pld->pipeline_layout))!=VK_SUCCESS)
+    if(vkCreatePipelineLayout(attr->device,&pPipelineLayoutCreateInfo,nullptr,&(pld->pipeline_layout))!=VK_SUCCESS)
    {
        delete pld;
        return(nullptr);
--- a/src/SceneGraph/Vulkan/VKRenderResourceMaterial.cpp
+++ b/src/SceneGraph/Vulkan/VKRenderResourceMaterial.cpp
@@ -12,6 +12,30 @@
 #include<hgl/shadergen/ShaderDescriptorInfo.h>

 VK_NAMESPACE_BEGIN
+
+namespace
+{
+    void CreateShaderStageList(List<VkPipelineShaderStageCreateInfo> &shader_stage_list,ShaderModuleMap *shader_maps)
+    {
+        const ShaderModule *sm;
+
+        const int shader_count=shader_maps->GetCount();
+        shader_stage_list.SetCount(shader_count);
+    
+        VkPipelineShaderStageCreateInfo *p=shader_stage_list.GetData();        
+
+        auto **itp=shader_maps->GetDataList();
+        for(int i=0;i<shader_count;i++)
+        {
+            sm=(*itp)->value;
+            hgl_cpy(p,sm->GetCreateInfo(),1);
+
+            ++p;
+            ++itp;
+        }
+    }
+}//namespace
+
 const ShaderModule *RenderResource::CreateShaderModule(const AnsiString &sm_name,const ShaderCreateInfo *sci)
 {
    if(!device)return(nullptr);
@@ -62,64 +86,66 @@ Material *RenderResource::CreateMaterial(const mtl::MaterialCreateInfo *mci)
    if(material_by_name.Get(mtl_name,mtl))
        return mtl;

-    const uint count=GetShaderCountByBits(mci->GetShaderStage());
-    const ShaderModule *sm;
+    const ShaderCreateInfoMap &sci_map=mci->GetShaderMap();
+    const uint sci_count=sci_map.GetCount();

-    ShaderModuleMap *smm=new ShaderModuleMap;
-    VertexInput *vertex_input=nullptr;
+    if(sci_count<2)
+        return(nullptr);

-    const ShaderCreateInfoVertex *vert=mci->GetVS();
+    if(!mci->GetFS())
+        return(nullptr);
+
+    AutoDelete<MaterialData> data=new MaterialData(mtl_name);

-    if(vert)
    {
-        sm=CreateShaderModule(mtl_name,vert);
+        const ShaderModule *sm;

-        if(!sm)
-            return(false);
+        auto **sci=sci_map.GetDataList();

-        if(smm->Add(sm))
-            vertex_input=new VertexInput(vert->sdm->GetShaderStageIO().input);
+        for(uint i=0;i<sci_count;i++)
+        {
+            sm=CreateShaderModule(mtl_name,(*sci)->value);
+
+            if(!sm)
+                return(nullptr);
+
+            data->shader_maps->Add(sm);
+
+            if((*sci)->key==VK_SHADER_STAGE_VERTEX_BIT)
+                data->vertex_input=new VertexInput((*sci)->value->sdm->GetShaderStageIO().input);
+
+            ++sci;
+        }
+
+        CreateShaderStageList(data->shader_stage_list,data->shader_maps);
    }

-    const ShaderCreateInfoGeometry *geom=mci->GetGS();
-
-    if(geom)
-    {
-        sm=CreateShaderModule(mtl_name,geom);
-
-        smm->Add(sm);
-    }
-
-    const ShaderCreateInfoFragment *frag=mci->GetFS();
-
-    if(frag)
-    {
-        sm=CreateShaderModule(mtl_name,frag);
-
-        smm->Add(sm);
-    }
-
-    MaterialDescriptorManager *mdm=nullptr;
-
    {
        const auto &mdi=mci->GetMDI();

        if(mdi.GetCount()>0)
-            mdm=new MaterialDescriptorManager(mci->GetName(),mdi.Get());
+            data->desc_manager=new MaterialDescriptorManager(mci->GetName(),mdi.Get());
    }

-    mtl=device->CreateMaterial(mci->GetName(),smm,mdm,vertex_input);
+    data->pipeline_layout_data=device->CreatePipelineLayoutData(data->desc_manager);

-    if(!mtl)
+    if(data->desc_manager)
    {
-        delete mdm;
-        delete smm;
-    }
-    else
-    {
-        Add(mtl);
+        ENUM_CLASS_FOR(DescriptorSetType,int,dst)
+        {
+            if(data->desc_manager->hasSet((DescriptorSetType)dst))
+                data->mp_array[dst]=device->CreateMP(data->desc_manager,data->pipeline_layout_data,(DescriptorSetType)dst);
+        }
    }

+    data->mi_data_bytes =mci->GetMIDataBytes();
+    data->mi_max_count  =mci->GetMIMaxCount();
+
+    mtl=new Material(data);
+    data.Discard();     //mtl已经接管了data的内容，这里不需要再释放
+
+    Add(mtl);
+
    material_by_name.Add(mtl_name,mtl);
    return mtl;
 }