diff options
Diffstat (limited to 'src/Ryujinx.Common/Pools')
| -rw-r--r-- | src/Ryujinx.Common/Pools/ObjectPool.cs | 75 | ||||
| -rw-r--r-- | src/Ryujinx.Common/Pools/SharedPools.cs | 17 | ||||
| -rw-r--r-- | src/Ryujinx.Common/Pools/ThreadStaticArray.cs | 20 |
3 files changed, 112 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; + } + } + } + } +} diff --git a/src/Ryujinx.Common/Pools/SharedPools.cs b/src/Ryujinx.Common/Pools/SharedPools.cs new file mode 100644 index 00000000..b4860b85 --- /dev/null +++ b/src/Ryujinx.Common/Pools/SharedPools.cs @@ -0,0 +1,17 @@ +namespace Ryujinx.Common +{ + public static class SharedPools + { + private static class DefaultPool<T> + where T : class, new() + { + public static readonly ObjectPool<T> Instance = new ObjectPool<T>(() => new T(), 20); + } + + public static ObjectPool<T> Default<T>() + where T : class, new() + { + return DefaultPool<T>.Instance; + } + } +} diff --git a/src/Ryujinx.Common/Pools/ThreadStaticArray.cs b/src/Ryujinx.Common/Pools/ThreadStaticArray.cs new file mode 100644 index 00000000..21434a02 --- /dev/null +++ b/src/Ryujinx.Common/Pools/ThreadStaticArray.cs @@ -0,0 +1,20 @@ +using System; + +namespace Ryujinx.Common.Pools +{ + public static class ThreadStaticArray<T> + { + [ThreadStatic] + private static T[] _array; + + public static ref T[] Get() + { + if (_array == null) + { + _array = new T[1]; + } + + return ref _array; + } + } +} |
