summaryrefslogtreecommitdiff
path: root/Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs
diff options
context:
space:
mode:
Diffstat (limited to 'Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs')
-rw-r--r--Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs50
1 files changed, 50 insertions, 0 deletions
diff --git a/Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs b/Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs
new file mode 100644
index 0000000..f1db3ec
--- /dev/null
+++ b/Assets/UI_Extension/Scripts/Utilities/ObjectPool.cs
@@ -0,0 +1,50 @@
+using System.Collections.Generic;
+using UnityEngine.Events;
+using UnityEngine;
+
+namespace UIExt
+{
+
+ internal class ObjectPool<T> where T : new()
+ {
+ private readonly Stack<T> m_Stack = new Stack<T>();
+ private readonly UnityAction<T> m_ActionOnGet;
+ private readonly UnityAction<T> m_ActionOnRelease;
+
+ public int countAll { get; private set; }
+ public int countActive { get { return countAll - countInactive; } }
+ public int countInactive { get { return m_Stack.Count; } }
+
+ public ObjectPool(UnityAction<T> actionOnGet, UnityAction<T> actionOnRelease)
+ {
+ m_ActionOnGet = actionOnGet;
+ m_ActionOnRelease = actionOnRelease;
+ }
+
+ public T Get()
+ {
+ T element;
+ if (m_Stack.Count == 0)
+ {
+ element = new T();
+ countAll++;
+ }
+ else
+ {
+ element = m_Stack.Pop();
+ }
+ if (m_ActionOnGet != null)
+ m_ActionOnGet(element);
+ return element;
+ }
+
+ public void Release(T element)
+ {
+ if (m_Stack.Count > 0 && ReferenceEquals(m_Stack.Peek(), element))
+ Debug.LogError("Internal error. Trying to destroy object that is already released to pool.");
+ if (m_ActionOnRelease != null)
+ m_ActionOnRelease(element);
+ m_Stack.Push(element);
+ }
+ }
+}