aboutsummaryrefslogtreecommitdiff
path: root/src/Ryujinx.Common/Pools/ObjectPool.cs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Ryujinx.Common/Pools/ObjectPool.cs')
-rw-r--r--src/Ryujinx.Common/Pools/ObjectPool.cs75
1 files changed, 75 insertions, 0 deletions
diff --git a/src/Ryujinx.Common/Pools/ObjectPool.cs b/src/Ryujinx.Common/Pools/ObjectPool.cs
new file mode 100644
index 00000000..e0bf5df6
--- /dev/null
+++ b/src/Ryujinx.Common/Pools/ObjectPool.cs
@@ -0,0 +1,75 @@
+using System;
+using System.Threading;
+
+namespace Ryujinx.Common
+{
+ public class ObjectPool<T>
+ where T : class
+ {
+ private T _firstItem;
+ private readonly T[] _items;
+
+ private readonly Func<T> _factory;
+
+ public ObjectPool(Func<T> factory, int size)
+ {
+ _items = new T[size - 1];
+ _factory = factory;
+ }
+
+ public T Allocate()
+ {
+ T instance = _firstItem;
+
+ if (instance == null || instance != Interlocked.CompareExchange(ref _firstItem, null, instance))
+ {
+ instance = AllocateInternal();
+ }
+
+ return instance;
+ }
+
+ private T AllocateInternal()
+ {
+ T[] items = _items;
+
+ for (int i = 0; i < items.Length; i++)
+ {
+ T instance = items[i];
+
+ if (instance != null && instance == Interlocked.CompareExchange(ref items[i], null, instance))
+ {
+ return instance;
+ }
+ }
+
+ return _factory();
+ }
+
+ public void Release(T obj)
+ {
+ if (_firstItem == null)
+ {
+ _firstItem = obj;
+ }
+ else
+ {
+ ReleaseInternal(obj);
+ }
+ }
+
+ private void ReleaseInternal(T obj)
+ {
+ T[] items = _items;
+
+ for (int i = 0; i < items.Length; i++)
+ {
+ if (items[i] == null)
+ {
+ items[i] = obj;
+ break;
+ }
+ }
+ }
+ }
+}