HnTextureRegistry: recycle texture IDs

DiligentGraphics · Oct 19, 2024 · 428e277 · 428e277
1 parent c872a2c
commit 428e277
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/Hydrogent/include/HnTextureRegistry.hpp b/Hydrogent/include/HnTextureRegistry.hpp
@@ -221,6 +221,9 @@ class HnTextureRegistry final : public std::enable_shared_from_this<HnTextureReg
     std::mutex                             m_AsyncTasksMtx;
     std::vector<RefCntAutoPtr<IAsyncTask>> m_AsyncTasks;
 
+    std::mutex          m_RecycledTextureIdsMtx;
+    std::vector<Uint32> m_RecycledTextureIds;
+
     std::atomic<Uint32> m_NextTextureId{0};
     std::atomic<Int32>  m_NumTexturesLoading{0};
     std::atomic<Int64>  m_LoadingTexDataSize{0};

diff --git a/Hydrogent/src/HnTextureRegistry.cpp b/Hydrogent/src/HnTextureRegistry.cpp
@@ -339,7 +339,21 @@ HnTextureRegistry::TextureHandleSharedPtr HnTextureRegistry::Allocate(const pxr:
         [&]() {
             m_NumTexturesLoading.fetch_add(+1);
 
-            std::shared_ptr<TextureHandle> TexHandle = std::make_shared<TextureHandle>(*this, m_NextTextureId.fetch_add(1));
+            Uint32 TextureId = 0;
+            {
+                std::lock_guard<std::mutex> Lock{m_RecycledTextureIdsMtx};
+                if (!m_RecycledTextureIds.empty())
+                {
+                    TextureId = m_RecycledTextureIds.back();
+                    m_RecycledTextureIds.pop_back();
+                }
+                else
+                {
+                    TextureId = m_NextTextureId.fetch_add(1);
+                }
+            }
+
+            std::shared_ptr<TextureHandle> TexHandle = std::make_shared<TextureHandle>(*this, TextureId);
 
             if (IsAsync && m_pThreadPool)
             {
@@ -416,6 +430,11 @@ Uint32 HnTextureRegistry::GetDataVersion() const
 
 void HnTextureRegistry::OnHandleDestroyed(const TextureHandle& Handle)
 {
+    {
+        std::lock_guard<std::mutex> Guard{m_RecycledTextureIdsMtx};
+        m_RecycledTextureIds.push_back(Handle.GetId());
+    }
+
     if (Handle.m_pAtlasSuballocation)
     {
         m_AtlasDataSize.fetch_add(-static_cast<Int64>(Handle.DataSize));