From f3aba255f6d69ddbf255b33d0eb0f391908e60a8 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Wed, 12 Jul 2023 19:55:41 -0700
Subject: Pool inst worklists and hashsets to avoid rehashing. (#2982)

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/slang-container-pool.h | 93 +++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 source/slang/slang-container-pool.h

(limited to 'source/slang/slang-container-pool.h')

diff --git a/source/slang/slang-container-pool.h b/source/slang/slang-container-pool.h
new file mode 100644
index 000000000..addeb1607
--- /dev/null
+++ b/source/slang/slang-container-pool.h
@@ -0,0 +1,93 @@
+#ifndef SLANG_CONTAINER_POOL_H
+#define SLANG_CONTAINER_POOL_H
+
+#include "../core/slang-list.h"
+#include "../core/slang-dictionary.h"
+#include "../core/slang-virtual-object-pool.h"
+
+// A pool to allow reuse of common types of containers to avoid
+// frequent resizing and rehashing.
+
+namespace Slang
+{
+static const int kContainerPoolSize = 1024;
+
+template<typename T>
+struct ObjectPool
+{
+    ObjectPool(int maxElementCount)
+    {
+        m_pool.initPool(maxElementCount);
+        m_objects.setCount(maxElementCount);
+    }
+
+    T* getObject()
+    {
+        auto id = m_pool.alloc(1);
+        if (id == -1)
+            SLANG_ABORT_COMPILATION("container pool allocation failure.");
+        return &m_objects[id];
+    }
+
+    void freeObject(T* object)
+    {
+        auto id = (int)(object - m_objects.getBuffer());
+        m_pool.free(id, 1);
+    }
+
+    VirtualObjectPool m_pool;
+    List<T> m_objects;
+};
+
+struct ContainerPool
+{
+    ObjectPool<List<void*>> m_listPool;
+    ObjectPool<Dictionary<void*, void*>> m_dictionaryPool;
+    ObjectPool<HashSet<void*>> m_hashSetPool;
+
+    ContainerPool()
+        :m_listPool(kContainerPoolSize), m_dictionaryPool(kContainerPoolSize), m_hashSetPool(kContainerPoolSize)
+    {}
+
+    template<typename T>
+    List<T*>* getList()
+    {
+        return (List<T*>*)m_listPool.getObject();
+    }
+
+    template<typename T, typename U>
+    Dictionary<T*, U*>* getDictionary()
+    {
+        return (Dictionary<T*, U*>*)m_dictionaryPool.getObject();
+    }
+
+    template<typename T>
+    HashSet<T*>* getHashSet()
+    {
+        return (HashSet<T*>*)m_hashSetPool.getObject();
+    }
+
+    template<typename T>
+    void free(List<T*>* list)
+    {
+        list->clear();
+        m_listPool.freeObject((List<void*>*)list);
+    }
+
+    template<typename T, typename U>
+    void free(Dictionary<T*, U*>* dict)
+    {
+        dict->clear();
+        m_dictionaryPool.freeObject((Dictionary<void*, void*>*)dict);
+    }
+
+    template<typename T>
+    void free(HashSet<T*>* set)
+    {
+        set->clear();
+        m_hashSetPool.freeObject((HashSet<void*>*)set);
+    }
+};
+}
+
+#endif
-- 
cgit v1.2.3