GLTF Resource Manager: added GetVertexPoolIndex method

Author: TheMostDiligent

AssetLoader/interface/GLTFLoader.hpp

@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2023 Diligent Graphics LLC
+ *  Copyright 2019-2024 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -1047,6 +1047,7 @@ struct Model
         std::vector<Uint32>                  Strides;
         std::vector<RefCntAutoPtr<IBuffer>>  Buffers;
         RefCntAutoPtr<IVertexPoolAllocation> pAllocation;
+        Uint32                               PoolId                = 0; // Vertex pool allocation Id
         Uint32                               EnabledAttributeFlags = 0;
     };
     VertexDataInfo VertexData;

AssetLoader/interface/GLTFResourceManager.hpp

@@ -234,6 +234,9 @@ class ResourceManager final : public ObjectBase<IObject>
     /// Returns all vertex pools for the given key.
     std::vector<IVertexPool*> GetVertexPools(const VertexLayoutKey& Key);
 
+    /// Returns index of the vertex pool with the give key.
+    /// If the pool does not exist, InvalidIndex (0xFFFFFFFF) is returned.
+    Uint32 GetVertexPoolIndex(const VertexLayoutKey& Key, IVertexPool* pPool);
 
     /// Updates the atlas texture for the given format.
     /// If the atlas does not exist, null is returned.

AssetLoader/src/GLTFBuilder.cpp

@@ -318,6 +318,8 @@ void ModelBuilder::InitVertexBuffers(IRenderDevice* pDevice)
             auto pBuffInitData  = BufferInitData::Create();
             pBuffInitData->Data = std::move(m_VertexData);
             m_Model.VertexData.pAllocation->SetUserData(pBuffInitData);
+            m_Model.VertexData.PoolId = m_CI.pResourceManager->GetVertexPoolIndex(LayoutKey, m_Model.VertexData.pAllocation->GetPool());
+            VERIFY_EXPR(m_Model.VertexData.PoolId != ~0u);
         }
         else
         {

AssetLoader/src/GLTFResourceManager.cpp

@@ -394,6 +394,23 @@ std::vector<IVertexPool*> ResourceManager::GetVertexPools(const VertexLayoutKey&
     return Pools;
 }
 
+Uint32 ResourceManager::GetVertexPoolIndex(const VertexLayoutKey& Key, IVertexPool* pPool)
+{
+    std::lock_guard<std::mutex> Guard{m_VertexPoolsMtx};
+
+    const auto pools_it = m_VertexPools.find(Key);
+    if (pools_it != m_VertexPools.end())
+    {
+        for (Uint32 i = 0; i < pools_it->second.size(); ++i)
+        {
+            if (pPool == pools_it->second[i])
+                return i;
+        }
+    }
+
+    return ~0u;
+}
+
 ITexture* ResourceManager::UpdateTexture(TEXTURE_FORMAT Fmt, IRenderDevice* pDevice, IDeviceContext* pContext)
 {
     decltype(m_Atlases)::iterator cache_it; // NB: can't initialize it without locking the mutex