Initial commit: Unity WordConnect project

Assets/WordConnectGameToolkit/Scripts/Gameplay/Pool/PoolObject.cs

@@ -0,0 +1,118 @@
+// // ©2015 - 2025 Candy Smith
+// // All rights reserved
+// // Redistribution of this software is strictly not allowed.
+// // Copy of this software can be obtained from unity asset store only.
+// // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// // FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
+// // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+// // THE SOFTWARE.
+
+using System.Collections.Generic;
+using UnityEngine;
+using WordsToolkit.Scripts.System;
+
+namespace WordsToolkit.Scripts.Gameplay.Pool
+{
+    public class PoolObject : MonoBehaviour
+    {
+        protected static readonly Dictionary<string, PoolObject> pools = new();
+
+        public GameObject prefab;
+
+        protected Queue<GameObject> pool = new();
+
+        public virtual void Awake()
+        {
+            if (prefab != null)
+            {
+                SetPrefab(prefab);
+            }
+        }
+
+        public void SetPrefab(GameObject newPrefab)
+        {
+            pools[newPrefab.name] = this;
+            prefab = newPrefab;
+        }
+
+        private void OnDestroy()
+        {
+            pools.Clear();
+        }
+
+        protected GameObject Create()
+        {
+            var item = Instantiate(prefab, transform);
+            item.name = prefab.name;
+            return item;
+        }
+
+        private GameObject Get()
+        {
+            var item = pool.Count == 0 ? Create() : pool.Dequeue();
+            if (item.activeSelf && pool.Count > 0)
+            {
+                item = pool.Dequeue();
+            }
+
+            item.SetActive(true);
+            return item;
+        }
+
+        public static void Return(GameObject item)
+        {
+            if (item == null)
+            {
+                return;
+            }
+
+            item.SetActive(false);
+            if (pools.TryGetValue(item.name, out var pool))
+            {
+                pool.pool.Enqueue(item);
+            }
+        }
+
+        public static GameObject GetObject(GameObject prefab)
+        {
+            return GetPool(prefab);
+        }
+
+        public static GameObject GetObject(GameObject prefab, Vector3 position)
+        {
+            var item = GetPool(prefab);
+            item.transform.position = position;
+            return item;
+        }
+
+        private static GameObject GetPool(GameObject prefab)
+        {
+            var prefabName = prefab.name;
+            if (pools.TryGetValue(prefabName, out var pool))
+            {
+                return pool.Get();
+            }
+
+            var poolObject = new GameObject(prefabName).AddComponent<PoolObject>();
+            poolObject.transform.SetParent(GameObject.Find("FXPool").transform);
+            poolObject.prefab = prefab;
+            poolObject.transform.localScale = Vector3.one;
+            pools.Add(prefabName, poolObject);
+            return poolObject.Get();
+        }
+
+        public static GameObject GetObject(string prefabName)
+        {
+            if (pools.TryGetValue(prefabName, out var pool))
+            {
+                return pool.Get();
+            }
+
+            Debug.LogError($"Pool with name {prefabName} not found");
+            return null;
+        }
+    }
+}