Enhance memory management and performance benchmarks
Added a new configuration setting in `.editorconfig` to sort system directives last and increased the maximum line length to 400 characters. Added a new static class `MathUtilities` in `MathUtilities.cs` with a method `CeilPow2` for computing powers of two. Added a new benchmark class `CollectionBenchmark` in `CollectionBenchmark.cs` to measure performance of standard versus unsafe arrays. Added a new benchmark class `HashCodeBenchmark` in `HashCodeBenchmark.cs` to evaluate hash code generation performance. Added new utility methods in `UnsafeUtilities.cs` for memory allocation and deallocation, including `Malloc`, `AlignedAlloc`, `Realloc`, and `Free`. Added a new `AllocationType` enum in `AllocationType.cs` to specify memory allocation types. Changed the project file `Misaki.HighPerformance.Mathematics.csproj` to target .NET 9.0 and enable implicit usings and nullable reference types. Changed the `ParallelNoiseBenchmark` class in `ParallelNoiseBenchmark.cs` to improve memory allocation strategies and performance. Changed memory management in `Arena.cs` and `DynamicArena.cs` to use custom `Malloc` and `Free` functions. Changed the `IUnsafeCollection` interface in `IUnsafeCollection.cs` to include new methods for resizing collections and obtaining unsafe pointers. Changed the `UnsafeArray.cs` to improve management of unsafe arrays, including constructor and method updates. Changed the `UnsafeHashMap` and `UnsafeHashSet` classes to enhance performance and memory management. Changed the `UnsafeCollectionExtensions` class to provide additional methods for copying elements and converting collections. Changed the `ObjectPool` class in `ObjectPool.cs` to simplify cleanup and remove auto-cleanup functionality. Changed job scheduling and worker classes in `JobExtensions.cs` and `JobWorker.cs` to improve job scheduling in a thread pool. Removed commented-out code in `Program.cs` related to previous testing methods. Removed auto-cleanup functionality from the `ObjectPool` class.
This commit is contained in:
2
.editorconfig
Normal file
2
.editorconfig
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
dotnet_sort_system_directives_first = false
|
||||||
|
max_line_length = 400
|
||||||
19
Misaki.HighPerformance.Math/MathUtilities.cs
Normal file
19
Misaki.HighPerformance.Math/MathUtilities.cs
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Mathematics;
|
||||||
|
|
||||||
|
public static class MathUtilities
|
||||||
|
{
|
||||||
|
/// <summary>Returns the smallest power of two that is greater than or equal to the specified number.</summary>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static int CeilPow2(int x)
|
||||||
|
{
|
||||||
|
x -= 1;
|
||||||
|
x |= x >> 1;
|
||||||
|
x |= x >> 2;
|
||||||
|
x |= x >> 4;
|
||||||
|
x |= x >> 8;
|
||||||
|
x |= x >> 16;
|
||||||
|
return x + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
<Project Sdk="Microsoft.NET.Sdk">
|
||||||
|
|
||||||
|
<PropertyGroup>
|
||||||
|
<TargetFramework>net9.0</TargetFramework>
|
||||||
|
<ImplicitUsings>enable</ImplicitUsings>
|
||||||
|
<Nullable>enable</Nullable>
|
||||||
|
</PropertyGroup>
|
||||||
|
|
||||||
|
</Project>
|
||||||
41
Misaki.HighPerformance.Test/CollectionBenchmark.cs
Normal file
41
Misaki.HighPerformance.Test/CollectionBenchmark.cs
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
using BenchmarkDotNet.Attributes;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Collections.Services;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Test;
|
||||||
|
|
||||||
|
[MemoryDiagnoser]
|
||||||
|
public class CollectionBenchmark
|
||||||
|
{
|
||||||
|
[Params(10, 100, 1000)]
|
||||||
|
public int count = 100;
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public void Array()
|
||||||
|
{
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
var array = new int[count];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public void UnsafeArray()
|
||||||
|
{
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
var array = new UnsafeArray<int>(count, Allocator.Temp);
|
||||||
|
for (var j = 0; j < count; j++)
|
||||||
|
{
|
||||||
|
array[j] = j;
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (var item in array)
|
||||||
|
{
|
||||||
|
Console.WriteLine(item);
|
||||||
|
}
|
||||||
|
|
||||||
|
AllocationManager.Reset();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
95
Misaki.HighPerformance.Test/HashCodeBenchmark.cs
Normal file
95
Misaki.HighPerformance.Test/HashCodeBenchmark.cs
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
using BenchmarkDotNet.Attributes;
|
||||||
|
using System.Numerics;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Test;
|
||||||
|
|
||||||
|
public class HashCodeBenchmark
|
||||||
|
{
|
||||||
|
private struct Component
|
||||||
|
{
|
||||||
|
public int Value;
|
||||||
|
public int Value2;
|
||||||
|
public float Value3;
|
||||||
|
public Guid Value4;
|
||||||
|
public Matrix4x4 Value5;
|
||||||
|
public Vector4 Value6;
|
||||||
|
}
|
||||||
|
|
||||||
|
[Params(100, 1000, 10000)]
|
||||||
|
public int count;
|
||||||
|
|
||||||
|
private Component _component = new Component()
|
||||||
|
{
|
||||||
|
Value = 0,
|
||||||
|
Value2 = 1,
|
||||||
|
Value3 = 2,
|
||||||
|
Value4 = Guid.NewGuid(),
|
||||||
|
Value5 = Matrix4x4.Identity,
|
||||||
|
Value6 = Vector4.One
|
||||||
|
};
|
||||||
|
|
||||||
|
private Dictionary<Type, int> _hashCache = new();
|
||||||
|
//private UnsafeHashMap<Guid, int> _hashMap = new(16);
|
||||||
|
|
||||||
|
private bool _disposed;
|
||||||
|
|
||||||
|
//~HashCodeBenchmark()
|
||||||
|
//{
|
||||||
|
// Dispose();
|
||||||
|
//}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public void Hash()
|
||||||
|
{
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
_ = _component.GetHashCode();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
[Benchmark]
|
||||||
|
public void HashWithCache()
|
||||||
|
{
|
||||||
|
for (var i = 0; i < count; i++)
|
||||||
|
{
|
||||||
|
var type = typeof(Component);
|
||||||
|
if (!_hashCache.TryGetValue(type, out var hash))
|
||||||
|
{
|
||||||
|
hash = type.GetHashCode();
|
||||||
|
_hashCache[type] = hash;
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = hash;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//[Benchmark]
|
||||||
|
//public void HashWithUnsafeHashMap()
|
||||||
|
//{
|
||||||
|
// for (var i = 0; i < count; i++)
|
||||||
|
// {
|
||||||
|
// var type = _component.GetType();
|
||||||
|
// var guid = type.GUID;
|
||||||
|
// if (!_hashMap.TryGetValue(guid, out var hash))
|
||||||
|
// {
|
||||||
|
// hash = type.GetHashCode();
|
||||||
|
// _hashMap.Add(guid, hash);
|
||||||
|
// _hashMap.Test(ref _hashMap._hashMap);
|
||||||
|
// }
|
||||||
|
|
||||||
|
// _ = hash;
|
||||||
|
// }
|
||||||
|
//}
|
||||||
|
|
||||||
|
//public void Dispose()
|
||||||
|
//{
|
||||||
|
// if (_disposed)
|
||||||
|
// {
|
||||||
|
// return;
|
||||||
|
// }
|
||||||
|
|
||||||
|
// _hashMap.Dispose();
|
||||||
|
|
||||||
|
// GC.SuppressFinalize(this);
|
||||||
|
//}
|
||||||
|
}
|
||||||
@@ -6,6 +6,7 @@
|
|||||||
<ImplicitUsings>enable</ImplicitUsings>
|
<ImplicitUsings>enable</ImplicitUsings>
|
||||||
<Nullable>enable</Nullable>
|
<Nullable>enable</Nullable>
|
||||||
<PublishAot>True</PublishAot>
|
<PublishAot>True</PublishAot>
|
||||||
|
<AllowUnsafeBlocks>True</AllowUnsafeBlocks>
|
||||||
</PropertyGroup>
|
</PropertyGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
|
|||||||
@@ -21,12 +21,6 @@ public class ParallelNoiseBenchmark
|
|||||||
return x - MathF.Truncate(x);
|
return x - MathF.Truncate(x);
|
||||||
}
|
}
|
||||||
|
|
||||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
||||||
private static float Lerp(float a, float b, float t)
|
|
||||||
{
|
|
||||||
return a + t * (b - a);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static Vector2 GradientNoiseDirect(Vector2 uv)
|
private static Vector2 GradientNoiseDirect(Vector2 uv)
|
||||||
{
|
{
|
||||||
uv.X %= 289;
|
uv.X %= 289;
|
||||||
@@ -48,7 +42,7 @@ public class ParallelNoiseBenchmark
|
|||||||
var d11 = Vector2.Dot(GradientNoiseDirect(ip + new Vector2(1, 1)), fp - new Vector2(1, 1));
|
var d11 = Vector2.Dot(GradientNoiseDirect(ip + new Vector2(1, 1)), fp - new Vector2(1, 1));
|
||||||
|
|
||||||
fp = fp * fp * fp * (fp * (fp * new Vector2(6.0f) - new Vector2(15.0f)) + new Vector2(10.0f));
|
fp = fp * fp * fp * (fp * (fp * new Vector2(6.0f) - new Vector2(15.0f)) + new Vector2(10.0f));
|
||||||
return Lerp(Lerp(d00, d10, fp.Y), Lerp(d01, d11, fp.Y), fp.X);
|
return float.Lerp(float.Lerp(d00, d10, fp.Y), float.Lerp(d01, d11, fp.Y), fp.X);
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Execute(int index)
|
public void Execute(int index)
|
||||||
@@ -65,9 +59,9 @@ public class ParallelNoiseBenchmark
|
|||||||
private const int _LENGTH = _WIDTH * _HEIGHT;
|
private const int _LENGTH = _WIDTH * _HEIGHT;
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public void JobSystem()
|
public static void JobSystem()
|
||||||
{
|
{
|
||||||
using var buffers = new UnsafeArray<float>(_LENGTH, AllocationType.UnInitialized);
|
using var buffers = new UnsafeArray<float>(_LENGTH, Allocator.Persistent, AllocationType.UnInitialized);
|
||||||
var job = new NoiseJob()
|
var job = new NoiseJob()
|
||||||
{
|
{
|
||||||
buffers = buffers,
|
buffers = buffers,
|
||||||
@@ -80,9 +74,9 @@ public class ParallelNoiseBenchmark
|
|||||||
}
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public void ParallelFor()
|
public static void ParallelFor()
|
||||||
{
|
{
|
||||||
using var buffers = new UnsafeArray<float>(_LENGTH, AllocationType.UnInitialized);
|
using var buffers = new UnsafeArray<float>(_LENGTH, Allocator.Persistent, AllocationType.UnInitialized);
|
||||||
|
|
||||||
Parallel.For(0, _LENGTH, i =>
|
Parallel.For(0, _LENGTH, i =>
|
||||||
{
|
{
|
||||||
@@ -94,9 +88,9 @@ public class ParallelNoiseBenchmark
|
|||||||
}
|
}
|
||||||
|
|
||||||
[Benchmark]
|
[Benchmark]
|
||||||
public void For()
|
public static void For()
|
||||||
{
|
{
|
||||||
using var buffers = new UnsafeArray<float>(_LENGTH, AllocationType.UnInitialized);
|
using var buffers = new UnsafeArray<float>(_LENGTH, Allocator.Persistent, AllocationType.UnInitialized);
|
||||||
for (var i = 0; i < _LENGTH; i++)
|
for (var i = 0; i < _LENGTH; i++)
|
||||||
{
|
{
|
||||||
var x = i % _WIDTH;
|
var x = i % _WIDTH;
|
||||||
|
|||||||
@@ -1,37 +1,7 @@
|
|||||||
//using BenchmarkDotNet.Running;
|
using Misaki.HighPerformance.Test;
|
||||||
//using Misaki.HighPerformance.Test;
|
using Misaki.HighPerformance.Unsafe.Collections.Services;
|
||||||
|
|
||||||
//BenchmarkRunner.Run<ParallelNoiseBenchmark>();
|
AllocationManager.Initialize(512_000);
|
||||||
|
var test = new CollectionBenchmark();
|
||||||
using Misaki.HighPerformance.Unsafe.Collections;
|
test.UnsafeArray();
|
||||||
|
AllocationManager.Dispose();
|
||||||
using var test = new UnsafeArray<Test>(10, AllocationType.UnInitialized);
|
|
||||||
|
|
||||||
for (var i = 0; i < 10; i++)
|
|
||||||
{
|
|
||||||
var t = new Test();
|
|
||||||
t.buffers[0] = i;
|
|
||||||
test[i] = t;
|
|
||||||
}
|
|
||||||
|
|
||||||
test.ReAlloc(20);
|
|
||||||
|
|
||||||
for (var i = 0; i < 10; i++)
|
|
||||||
{
|
|
||||||
Console.WriteLine(test[i].buffers[0]);
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Test : IDisposable
|
|
||||||
{
|
|
||||||
public UnsafeArray<float> buffers;
|
|
||||||
|
|
||||||
public Test()
|
|
||||||
{
|
|
||||||
buffers = new UnsafeArray<float>(1, AllocationType.UnInitialized);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void Dispose()
|
|
||||||
{
|
|
||||||
buffers.Dispose();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ namespace Misaki.HighPerformance.Unsafe.Buffer;
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
public unsafe struct Arena : IDisposable
|
public unsafe struct Arena : IDisposable
|
||||||
{
|
{
|
||||||
private void* _buffer;
|
private byte* _buffer;
|
||||||
private uint _size;
|
private uint _size;
|
||||||
private uint _offset;
|
private uint _offset;
|
||||||
|
|
||||||
@@ -16,7 +16,7 @@ public unsafe struct Arena : IDisposable
|
|||||||
|
|
||||||
public Arena(uint size)
|
public Arena(uint size)
|
||||||
{
|
{
|
||||||
_buffer = NativeMemory.Alloc(size);
|
_buffer = (byte*)Malloc(size);
|
||||||
_size = size;
|
_size = size;
|
||||||
_offset = 0;
|
_offset = 0;
|
||||||
}
|
}
|
||||||
@@ -70,7 +70,7 @@ public unsafe struct Arena : IDisposable
|
|||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
NativeMemory.Free(_buffer);
|
Free(_buffer);
|
||||||
|
|
||||||
_buffer = null;
|
_buffer = null;
|
||||||
_size = 0;
|
_size = 0;
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
using Misaki.HighPerformance.Unsafe.Collections;
|
using Misaki.HighPerformance.Unsafe.Collections;
|
||||||
using System.Runtime.InteropServices;
|
|
||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Buffer;
|
namespace Misaki.HighPerformance.Unsafe.Buffer;
|
||||||
|
|
||||||
@@ -27,18 +26,17 @@ public unsafe struct DynamicArena : IDisposable
|
|||||||
public DynamicArena(uint initialSize)
|
public DynamicArena(uint initialSize)
|
||||||
{
|
{
|
||||||
_initialSize = initialSize;
|
_initialSize = initialSize;
|
||||||
_root = (ArenaNode*)NativeMemory.Alloc(SizeOf<ArenaNode>());
|
_root = (ArenaNode*)Malloc(SizeOf<ArenaNode>());
|
||||||
_root->arena = new Arena(initialSize);
|
_root->arena = new Arena(initialSize);
|
||||||
_root->next = null;
|
_root->next = null;
|
||||||
_current = _root;
|
_current = _root;
|
||||||
_disposed = false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private bool CreateNewNode(uint size)
|
private bool CreateNewNode(uint size)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
var newNode = (ArenaNode*)NativeMemory.Alloc(SizeOf<ArenaNode>());
|
var newNode = (ArenaNode*)Malloc(SizeOf<ArenaNode>());
|
||||||
newNode->arena = new Arena(size);
|
newNode->arena = new Arena(size);
|
||||||
newNode->next = null;
|
newNode->next = null;
|
||||||
|
|
||||||
@@ -70,10 +68,14 @@ public unsafe struct DynamicArena : IDisposable
|
|||||||
{
|
{
|
||||||
result = current->arena.Allocate(size, alignSize, allocationType);
|
result = current->arena.Allocate(size, alignSize, allocationType);
|
||||||
if (result != null)
|
if (result != null)
|
||||||
|
{
|
||||||
return result;
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
if (current->next == null && !CreateNewNode(Math.Max(size, _initialSize)))
|
if (current->next == null && !CreateNewNode(Math.Max(size, _initialSize)))
|
||||||
|
{
|
||||||
return null;
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
current = current->next;
|
current = current->next;
|
||||||
}
|
}
|
||||||
@@ -107,14 +109,16 @@ public unsafe struct DynamicArena : IDisposable
|
|||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
if (_disposed)
|
if (_disposed)
|
||||||
|
{
|
||||||
return;
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
var current = _root;
|
var current = _root;
|
||||||
while (current != null)
|
while (current != null)
|
||||||
{
|
{
|
||||||
var next = current->next;
|
var next = current->next;
|
||||||
current->arena.Dispose();
|
current->arena.Dispose();
|
||||||
NativeMemory.Free(current);
|
Free(current);
|
||||||
current = next;
|
current = next;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -4,4 +4,10 @@ public enum AllocationType
|
|||||||
{
|
{
|
||||||
UnInitialized,
|
UnInitialized,
|
||||||
Clear
|
Clear
|
||||||
|
}
|
||||||
|
|
||||||
|
public enum Allocator
|
||||||
|
{
|
||||||
|
Temp,
|
||||||
|
Persistent
|
||||||
}
|
}
|
||||||
@@ -1,21 +1,38 @@
|
|||||||
namespace Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
namespace Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
||||||
|
|
||||||
public unsafe interface IUnsafeCollection<T> : IDisposable
|
public unsafe interface IUnsafeCollection<T> : IEnumerable<T>, IDisposable
|
||||||
where T : unmanaged
|
where T : unmanaged
|
||||||
{
|
{
|
||||||
public T* Buffer
|
/// <summary>
|
||||||
|
/// Gets the number of elements in a collection. The value is read-only.
|
||||||
|
/// </summary>
|
||||||
|
public int Count
|
||||||
{
|
{
|
||||||
get;
|
get;
|
||||||
}
|
}
|
||||||
|
|
||||||
public int Size
|
/// <summary>
|
||||||
|
/// Indicates whether the object has been created. Returns true if the object is created, otherwise false.
|
||||||
|
/// </summary>
|
||||||
|
public bool IsCreated
|
||||||
{
|
{
|
||||||
get;
|
get;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ref T this[int index] { get; }
|
/// <summary>
|
||||||
|
/// Removes all elements from the collection. The collection will be empty after this operation.
|
||||||
|
/// </summary>
|
||||||
public void Clear();
|
public void Clear();
|
||||||
public void ReAlloc(int newSize);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Changes the size of a collection or array to the specified value.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="newSize">Specifies the new size to which the collection or array should be adjusted.</param>
|
||||||
|
public void Resize(int newSize);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns a pointer to an unmanaged memory location. This pointer can be used for low-level memory operations.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>The method returns a void pointer to the unsafe memory location.</returns>
|
||||||
|
public void* GetUnsafePtr();
|
||||||
|
}
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
using Misaki.HighPerformance.Unsafe.Buffer;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Unsafe.Collections.Services;
|
||||||
|
|
||||||
|
public static unsafe class AllocationManager
|
||||||
|
{
|
||||||
|
private static DynamicArena _arena;
|
||||||
|
private static bool _initialized;
|
||||||
|
|
||||||
|
private static readonly Lock _lock = new();
|
||||||
|
|
||||||
|
public static void Initialize(uint initialSize)
|
||||||
|
{
|
||||||
|
_arena = new DynamicArena(initialSize);
|
||||||
|
_initialized = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static T* Allocate<T>(uint size, uint alignSize, Allocator allocator, AllocationType allocationType)
|
||||||
|
where T : unmanaged
|
||||||
|
{
|
||||||
|
if (!_initialized)
|
||||||
|
{
|
||||||
|
throw new InvalidOperationException("The AllocationManager has not been initialized.");
|
||||||
|
}
|
||||||
|
|
||||||
|
lock (_lock)
|
||||||
|
{
|
||||||
|
return allocator switch
|
||||||
|
{
|
||||||
|
Allocator.Temp => (T*)_arena.Allocate(size * (uint)sizeof(T), alignSize, allocationType),
|
||||||
|
Allocator.Persistent => (T*)AlignedAlloc((nuint)(size * sizeof(T)), alignSize),
|
||||||
|
_ => throw new ArgumentOutOfRangeException(nameof(allocator), "Invalid allocator type."),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void Reset(bool clear = false)
|
||||||
|
{
|
||||||
|
if (!_initialized)
|
||||||
|
{
|
||||||
|
throw new InvalidOperationException("The AllocationManager has not been initialized.");
|
||||||
|
}
|
||||||
|
|
||||||
|
_arena.Reset(clear);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void Dispose()
|
||||||
|
{
|
||||||
|
_arena.Dispose();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,21 +1,25 @@
|
|||||||
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Collections.Services;
|
||||||
using Misaki.HighPerformance.Unsafe.Helpers;
|
using Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
using System.Collections;
|
using System.Collections;
|
||||||
using System.Runtime.CompilerServices;
|
using System.Runtime.CompilerServices;
|
||||||
using System.Runtime.InteropServices;
|
|
||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Collections;
|
namespace Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
|
||||||
public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
/// <summary>
|
||||||
|
/// A structure for managing an array of unmanaged types with unsafe memory operations.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents a type that can be stored in an unmanaged memory context.</typeparam>
|
||||||
|
public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>
|
||||||
where T : unmanaged
|
where T : unmanaged
|
||||||
{
|
{
|
||||||
private struct Enumerator : IEnumerator<T>
|
private struct Enumerator : IEnumerator<T>
|
||||||
{
|
{
|
||||||
private UnsafeArray<T> _collection;
|
private UnsafeArray<T>* _collection;
|
||||||
private int _index;
|
private int _index;
|
||||||
private T _value;
|
private T _value;
|
||||||
|
|
||||||
public Enumerator(ref UnsafeArray<T> collection)
|
public Enumerator(UnsafeArray<T>* collection)
|
||||||
{
|
{
|
||||||
_collection = collection;
|
_collection = collection;
|
||||||
_index = -1;
|
_index = -1;
|
||||||
@@ -26,9 +30,9 @@ public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
public bool MoveNext()
|
public bool MoveNext()
|
||||||
{
|
{
|
||||||
_index++;
|
_index++;
|
||||||
if (_index < _collection.Size)
|
if (_index < _collection->_count)
|
||||||
{
|
{
|
||||||
_value = UnsafeUtilities.ReadArrayElement<T>(_collection.Buffer, _index);
|
_value = UnsafeUtilities.ReadArrayElement<T>(_collection->_buffer, _index);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -42,22 +46,16 @@ public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Let NativeArray indexer check for out of range.
|
// Let NativeArray indexer check for out of range.
|
||||||
public T Current
|
public readonly T Current
|
||||||
{
|
{
|
||||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
get
|
get => _value;
|
||||||
{
|
|
||||||
return _value;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
object IEnumerator.Current
|
readonly object IEnumerator.Current
|
||||||
{
|
{
|
||||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
get
|
get => Current;
|
||||||
{
|
|
||||||
return Current;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
@@ -66,10 +64,9 @@ public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
}
|
}
|
||||||
|
|
||||||
private T* _buffer;
|
private T* _buffer;
|
||||||
private int _size;
|
private int _count;
|
||||||
|
|
||||||
public readonly T* Buffer => _buffer;
|
public readonly int Count => _count;
|
||||||
public readonly int Size => _size;
|
|
||||||
|
|
||||||
public readonly ref T this[int index]
|
public readonly ref T this[int index]
|
||||||
{
|
{
|
||||||
@@ -77,14 +74,31 @@ public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
get => ref UnsafeUtilities.ReadArrayElementRef<T>(_buffer, index);
|
get => ref UnsafeUtilities.ReadArrayElementRef<T>(_buffer, index);
|
||||||
}
|
}
|
||||||
|
|
||||||
public IEnumerator<T> GetEnumerator() => new Enumerator(ref this);
|
public readonly bool IsCreated
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _buffer != null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public IEnumerator<T> GetEnumerator() => new Enumerator((UnsafeArray<T>*)UnsafeUtilities.AddressOf(ref this));
|
||||||
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
||||||
|
|
||||||
public UnsafeArray(int size, AllocationType allocationType)
|
/// <summary>
|
||||||
|
/// Initializes a new instance of UnsafeArray with a specified number of elements and an allocation type. It
|
||||||
|
/// allocates memory and optionally clears it.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="count">Specifies the number of elements to allocate in the array, which must be greater than zero.</param>
|
||||||
|
/// <param name="allocationType">Determines how the allocated memory should be initialized, either uninitialized or cleared.</param>
|
||||||
|
/// <exception cref="ArgumentOutOfRangeException">Thrown when the specified number of elements is less than or equal to zero.</exception>
|
||||||
|
public UnsafeArray(int count, Allocator allocator, AllocationType allocationType = AllocationType.UnInitialized)
|
||||||
{
|
{
|
||||||
_size = size;
|
if (count <= 0)
|
||||||
_buffer = (T*)NativeMemory.AlignedAlloc((nuint)(size * sizeof(T)), AlignOf<T>());
|
{
|
||||||
|
throw new ArgumentOutOfRangeException(nameof(count), "Count must be greater than zero.");
|
||||||
|
}
|
||||||
|
|
||||||
|
_buffer = AllocationManager.Allocate<T>((uint)count, (uint)AlignOf<T>(), allocator, allocationType);
|
||||||
|
_count = count;
|
||||||
|
|
||||||
if (allocationType == AllocationType.Clear)
|
if (allocationType == AllocationType.Clear)
|
||||||
{
|
{
|
||||||
@@ -92,29 +106,47 @@ public unsafe struct UnsafeArray<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ReAlloc(int newSize)
|
/// <summary>
|
||||||
|
/// Initializes an UnsafeArray with a pointer to a buffer and a count of elements. The count is adjusted based on
|
||||||
|
/// the size of the type T.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="buffer">A pointer to the memory location that holds the elements of the array.</param>
|
||||||
|
/// <param name="count">The total size of the data in bytes, which is divided by the size of type T to determine the number of elements.</param>
|
||||||
|
public UnsafeArray(void* buffer, int count)
|
||||||
{
|
{
|
||||||
if (newSize == _size)
|
_buffer = (T*)buffer;
|
||||||
|
_count = count;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Resize(int newSize)
|
||||||
|
{
|
||||||
|
if (newSize == _count)
|
||||||
{
|
{
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
_buffer = (T*)NativeMemory.AlignedRealloc(_buffer, (nuint)(newSize * sizeof(T)), AlignOf<T>());
|
_buffer = (T*)AlignedRealloc(_buffer, (nuint)(newSize * sizeof(T)), AlignOf<T>());
|
||||||
_size = newSize;
|
_count = newSize;
|
||||||
}
|
}
|
||||||
|
|
||||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
public void Clear()
|
public readonly void Clear()
|
||||||
{
|
{
|
||||||
MemClear(_buffer, (uint)(_size * sizeof(T)));
|
MemClear(_buffer, (uint)(_count * sizeof(T)));
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly void* GetUnsafePtr()
|
||||||
|
{
|
||||||
|
return _buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
NativeMemory.AlignedFree(_buffer);
|
AlignedFree(_buffer);
|
||||||
|
|
||||||
_buffer = null;
|
_buffer = null;
|
||||||
_size = 0;
|
_count = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
205
Misaki.HighPerformance.Unsafe/Collections/UnsafeHashMap.cs
Normal file
205
Misaki.HighPerformance.Unsafe/Collections/UnsafeHashMap.cs
Normal file
@@ -0,0 +1,205 @@
|
|||||||
|
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
using System.Collections;
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
|
||||||
|
public unsafe struct UnsafeHashMap<TKey, TValue> : IUnsafeCollection<KeyValuePair<TKey, TValue>> where TKey : unmanaged, IEquatable<TKey> where TValue : unmanaged
|
||||||
|
{
|
||||||
|
private struct Enumerator : IEnumerator<KeyValuePair<TKey, TValue>>
|
||||||
|
{
|
||||||
|
internal HashMapHelper<TKey>.Enumerator _enumerator;
|
||||||
|
|
||||||
|
public Enumerator(HashMapHelper<TKey>* data)
|
||||||
|
{
|
||||||
|
_enumerator = new HashMapHelper<TKey>.Enumerator(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// The current key-value pair.
|
||||||
|
/// </summary>
|
||||||
|
/// <value>The current key-value pair.</value>
|
||||||
|
public KeyValuePair<TKey, TValue> Current
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _enumerator.GetCurrent<TValue>();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gets the element at the current position of the enumerator in the container.
|
||||||
|
/// </summary>
|
||||||
|
object IEnumerator.Current => Current;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Advances the enumerator to the next key-value pair.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>True if <see cref="Current"/> is valid to read after the call.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public bool MoveNext() => _enumerator.MoveNext();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Resets the enumerator to its initial state.
|
||||||
|
/// </summary>
|
||||||
|
public void Reset() => _enumerator.Reset();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Does nothing.
|
||||||
|
/// </summary>
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private HashMapHelper<TKey> _hashMap;
|
||||||
|
|
||||||
|
public readonly int Count => _hashMap.Count;
|
||||||
|
public readonly int Capacity => _hashMap.Capacity;
|
||||||
|
public readonly bool IsCreated => _hashMap.IsCreated;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gets and sets values by key.
|
||||||
|
/// </summary>
|
||||||
|
/// <remarks>Getting a key that is not present will throw. Setting a key that is not already present will add the key.</remarks>
|
||||||
|
/// <param name="key">The key to look up.</param>
|
||||||
|
/// <value>The value associated with the key.</value>
|
||||||
|
/// <exception cref="ArgumentException">For getting, thrown if the key was not present.</exception>
|
||||||
|
public TValue this[TKey key]
|
||||||
|
{
|
||||||
|
get
|
||||||
|
{
|
||||||
|
if (!_hashMap.TryGetValue<TValue>(key, out var result))
|
||||||
|
{
|
||||||
|
throw new ArgumentException($"Key: {key} is not present.");
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
set
|
||||||
|
{
|
||||||
|
var idx = _hashMap.Find(key);
|
||||||
|
if (-1 != idx)
|
||||||
|
{
|
||||||
|
UnsafeUtilities.WriteArrayElement(_hashMap.Buffer, idx, value);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
TryAdd(key, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public IEnumerator<KeyValuePair<TKey, TValue>> GetEnumerator() => new Enumerator((HashMapHelper<TKey>*)UnsafeUtilities.AddressOf(ref _hashMap));
|
||||||
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
||||||
|
|
||||||
|
public UnsafeHashMap(int capacity)
|
||||||
|
{
|
||||||
|
_hashMap = new HashMapHelper<TKey>(capacity, sizeof(TValue), HashMapHelper<TKey>.MINIMAL_CAPACITY);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds a new key-value pair.
|
||||||
|
/// </summary>
|
||||||
|
/// <remarks>If the key is already present, this method returns false without modifying the hash map.</remarks>
|
||||||
|
/// <param name="key">The key to add.</param>
|
||||||
|
/// <param name="item">The value to add.</param>
|
||||||
|
/// <returns>True if the key-value pair was added.</returns>
|
||||||
|
public bool TryAdd(TKey key, TValue item)
|
||||||
|
{
|
||||||
|
var idx = _hashMap.TryAdd(key);
|
||||||
|
if (idx != -1)
|
||||||
|
{
|
||||||
|
UnsafeUtilities.WriteArrayElement(_hashMap.Buffer, idx, item);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds a new key-value pair.
|
||||||
|
/// </summary>
|
||||||
|
/// <remarks>If the key is already present, this method throws without modifying the hash map.</remarks>
|
||||||
|
/// <param name="key">The key to add.</param>
|
||||||
|
/// <param name="item">The value to add.</param>
|
||||||
|
/// <exception cref="ArgumentException">Thrown if the key was already present.</exception>
|
||||||
|
public void Add(TKey key, TValue item)
|
||||||
|
{
|
||||||
|
var result = TryAdd(key, item);
|
||||||
|
if (!result)
|
||||||
|
{
|
||||||
|
throw new ArgumentException($"An item with the same key has already been added: {key}");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns the value associated with a key.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">The key to look up.</param>
|
||||||
|
/// <param name="item">Outputs the value associated with the key. Outputs default if the key was not present.</param>
|
||||||
|
/// <returns>True if the key was present.</returns>
|
||||||
|
public bool TryGetValue(TKey key, out TValue item)
|
||||||
|
{
|
||||||
|
return _hashMap.TryGetValue(key, out item);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns true if a given key is present in this hash map.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">The key to look up.</param>
|
||||||
|
/// <returns>True if the key was present.</returns>
|
||||||
|
public bool ContainsKey(TKey key)
|
||||||
|
{
|
||||||
|
return -1 != _hashMap.Find(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Sets the capacity to match what it would be if it had been originally initialized with all its entries.
|
||||||
|
/// </summary>
|
||||||
|
public void TrimExcess() => _hashMap.TrimExcess();
|
||||||
|
|
||||||
|
public void Resize(int newSize)
|
||||||
|
{
|
||||||
|
_hashMap.Resize(newSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Clear()
|
||||||
|
{
|
||||||
|
_hashMap.Clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Retrieves an array of keys from the hash map.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>An array containing the keys stored in the hash map.</returns>
|
||||||
|
public UnsafeArray<TKey> GetKeyArray(Allocator allocator) => _hashMap.GetKeyArray(allocator);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Retrieves an array of values from the underlying hash map.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>An UnsafeArray containing the values stored in the hash map.</returns>
|
||||||
|
public UnsafeArray<TValue> GetValueArray(Allocator allocator) => _hashMap.GetValueArray<TValue>(allocator);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Retrieves an array of key-value pairs from the hash map. The keys are of type TKey and the values are of type
|
||||||
|
/// TValue.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>Returns an UnsafeArray containing KeyValuePair objects.</returns>
|
||||||
|
public UnsafeArray<KeyValuePair<TKey, TValue>> GetKeyValueArrays(Allocator allocator) => _hashMap.GetKeyValueArrays<TValue>(allocator);
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly void* GetUnsafePtr()
|
||||||
|
{
|
||||||
|
return _hashMap.Buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
_hashMap.Dispose();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Test(ref HashMapHelper<TKey> t)
|
||||||
|
{
|
||||||
|
Console.WriteLine(t.Equals(_hashMap));
|
||||||
|
}
|
||||||
|
}
|
||||||
122
Misaki.HighPerformance.Unsafe/Collections/UnsafeHashSet.cs
Normal file
122
Misaki.HighPerformance.Unsafe/Collections/UnsafeHashSet.cs
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
using System.Collections;
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// A collection that provides fast, unsafe operations for managing a set of unmanaged types. It supports adding,
|
||||||
|
/// removing, and checking for values.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents an unmanaged type that can be compared for equality.</typeparam>
|
||||||
|
public unsafe struct UnsafeHashSet<T> : IUnsafeCollection<T>, IEnumerable<T>
|
||||||
|
where T : unmanaged, IEquatable<T>
|
||||||
|
{
|
||||||
|
private struct Enumerator : IEnumerator<T>
|
||||||
|
{
|
||||||
|
internal HashMapHelper<T>.Enumerator _enumerator;
|
||||||
|
|
||||||
|
public Enumerator(HashMapHelper<T>* hashMap)
|
||||||
|
{
|
||||||
|
_enumerator = new HashMapHelper<T>.Enumerator(hashMap);
|
||||||
|
}
|
||||||
|
|
||||||
|
public T Current
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _enumerator.buffer->_keys[_enumerator.index];
|
||||||
|
}
|
||||||
|
|
||||||
|
object IEnumerator.Current => Current;
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public bool MoveNext() => _enumerator.MoveNext();
|
||||||
|
|
||||||
|
public void Reset() => _enumerator.Reset();
|
||||||
|
|
||||||
|
public readonly void Dispose()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private HashMapHelper<T> _hashMap;
|
||||||
|
|
||||||
|
public readonly int Count => _hashMap.Count;
|
||||||
|
public readonly int Capacity => _hashMap.Capacity;
|
||||||
|
public readonly bool IsCreated => _hashMap.IsCreated;
|
||||||
|
|
||||||
|
public IEnumerator<T> GetEnumerator() => new Enumerator((HashMapHelper<T>*)UnsafeUtilities.AddressOf(ref _hashMap));
|
||||||
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
||||||
|
|
||||||
|
public UnsafeHashSet(int capacity)
|
||||||
|
{
|
||||||
|
_hashMap = new HashMapHelper<T>(capacity, 0, HashMapHelper<T>.MINIMAL_CAPACITY);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds a new value (unless it is already present).
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="item">The value to add.</param>
|
||||||
|
/// <returns>True if the value was not already present.</returns>
|
||||||
|
public bool Add(T item)
|
||||||
|
{
|
||||||
|
return -1 != _hashMap.TryAdd(item);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Removes a particular value.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="item">The value to remove.</param>
|
||||||
|
/// <returns>True if the value was present.</returns>
|
||||||
|
public bool Remove(T item)
|
||||||
|
{
|
||||||
|
return -1 != _hashMap.TryRemove(item);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns true if a particular value is present.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="item">The value to check for.</param>
|
||||||
|
/// <returns>True if the value was present.</returns>
|
||||||
|
public bool Contains(T item)
|
||||||
|
{
|
||||||
|
return -1 != _hashMap.Find(item);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Sets the capacity to match what it would be if it had been originally initialized with all its entries.
|
||||||
|
/// </summary>
|
||||||
|
public void TrimExcess() => _hashMap.TrimExcess();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns an array with a copy of this set's values (in no particular order).
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="allocator">The allocator to use.</param>
|
||||||
|
/// <returns>An array with a copy of the set's values.</returns>
|
||||||
|
public UnsafeArray<T> ToNativeArray(Allocator allocator)
|
||||||
|
{
|
||||||
|
return _hashMap.GetKeyArray(allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Resize(int newSize)
|
||||||
|
{
|
||||||
|
_hashMap.Resize(newSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Clear()
|
||||||
|
{
|
||||||
|
_hashMap.Clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly void* GetUnsafePtr()
|
||||||
|
{
|
||||||
|
return _hashMap.Buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
_hashMap.Dispose();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -5,16 +5,20 @@ using System.Runtime.CompilerServices;
|
|||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Collections;
|
namespace Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
|
||||||
public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
/// <summary>
|
||||||
|
/// A collection that allows for unsafe operations on a list of unmanaged types.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents a type that can be stored in the collection, constrained to unmanaged types for performance and safety.</typeparam>
|
||||||
|
public unsafe struct UnsafeList<T> : IUnsafeCollection<T>
|
||||||
where T : unmanaged
|
where T : unmanaged
|
||||||
{
|
{
|
||||||
private struct Enumerator : IEnumerator<T>
|
private struct Enumerator : IEnumerator<T>
|
||||||
{
|
{
|
||||||
private UnsafeList<T> _collection;
|
private UnsafeList<T>* _collection;
|
||||||
private int _index;
|
private int _index;
|
||||||
private T _value;
|
private T _value;
|
||||||
|
|
||||||
public Enumerator(ref UnsafeList<T> collection)
|
public Enumerator(UnsafeList<T>* collection)
|
||||||
{
|
{
|
||||||
_collection = collection;
|
_collection = collection;
|
||||||
_index = -1;
|
_index = -1;
|
||||||
@@ -25,9 +29,9 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
public bool MoveNext()
|
public bool MoveNext()
|
||||||
{
|
{
|
||||||
_index++;
|
_index++;
|
||||||
if (_index < _collection.Size)
|
if (_index < _collection->_count)
|
||||||
{
|
{
|
||||||
_value = UnsafeUtilities.ReadArrayElement<T>(_collection.Buffer, _index);
|
_value = UnsafeUtilities.ReadArrayElement<T>(_collection->_array.GetUnsafePtr(), _index);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -88,9 +92,9 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
/// <param name="value">The value to be added to the collection.</param>
|
/// <param name="value">The value to be added to the collection.</param>
|
||||||
public void AddNoResize(T value)
|
public void AddNoResize(T value)
|
||||||
{
|
{
|
||||||
var idx = Interlocked.Increment(ref listData->_size) - 1;
|
var idx = Interlocked.Increment(ref listData->_count) - 1;
|
||||||
listData->CheckNoResizeCapacity(idx, 1);
|
listData->CheckNoResizeCapacity(idx, 1);
|
||||||
UnsafeUtilities.WriteArrayElement(listData->Buffer, idx, value);
|
UnsafeUtilities.WriteArrayElement(listData->_array.GetUnsafePtr(), idx, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -100,19 +104,19 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
/// <param name="count">Indicates the number of elements to be added from the source data.</param>
|
/// <param name="count">Indicates the number of elements to be added from the source data.</param>
|
||||||
public void AddRangeNoResize(T* ptr, int count)
|
public void AddRangeNoResize(T* ptr, int count)
|
||||||
{
|
{
|
||||||
var idx = Interlocked.Add(ref listData->_size, count) - count;
|
var idx = Interlocked.Add(ref listData->_count, count) - count;
|
||||||
listData->CheckNoResizeCapacity(idx, count);
|
listData->CheckNoResizeCapacity(idx, count);
|
||||||
MemCpy(listData->Buffer + idx, ptr, (uint)(count * sizeof(T)));
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(listData->_array.GetUnsafePtr(), idx), ptr, (uint)(count * sizeof(T)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private UnsafeArray<T> _array;
|
private UnsafeArray<T> _array;
|
||||||
|
|
||||||
private int _size;
|
private int _count;
|
||||||
|
|
||||||
public readonly T* Buffer => _array.Buffer;
|
public readonly int Count => _count;
|
||||||
public readonly int Size => _size;
|
public readonly int Capacity => _array.Count;
|
||||||
public readonly int Capacity => _array.Size;
|
public readonly bool IsCreated => _array.IsCreated;
|
||||||
|
|
||||||
public readonly ref T this[int index]
|
public readonly ref T this[int index]
|
||||||
{
|
{
|
||||||
@@ -120,17 +124,15 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
get => ref _array[index];
|
get => ref _array[index];
|
||||||
}
|
}
|
||||||
|
|
||||||
public IEnumerator<T> GetEnumerator() => new Enumerator(ref this);
|
public IEnumerator<T> GetEnumerator() => new Enumerator((UnsafeList<T>*)UnsafeUtilities.AddressOf(ref this));
|
||||||
|
|
||||||
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
||||||
|
|
||||||
public ParallelWriter AsParallelWriter() =>
|
public ParallelWriter AsParallelWriter() => new((UnsafeList<T>*)UnsafeUtilities.AddressOf(ref this));
|
||||||
new((UnsafeList<T>*)UnsafeUtilities.AddressOf(ref this));
|
|
||||||
|
|
||||||
public UnsafeList(int capacity, AllocationType allocationType)
|
public UnsafeList(int capacity, Allocator allocator, AllocationType allocationType = AllocationType.UnInitialized)
|
||||||
{
|
{
|
||||||
_array = new UnsafeArray<T>(capacity, allocationType);
|
_array = new UnsafeArray<T>(capacity, allocator, allocationType);
|
||||||
_size = 0;
|
_count = 0;
|
||||||
|
|
||||||
if (allocationType == AllocationType.Clear)
|
if (allocationType == AllocationType.Clear)
|
||||||
{
|
{
|
||||||
@@ -140,7 +142,7 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
|
|
||||||
private readonly void CheckNoResizeCapacity(int count)
|
private readonly void CheckNoResizeCapacity(int count)
|
||||||
{
|
{
|
||||||
CheckNoResizeCapacity(count, Size);
|
CheckNoResizeCapacity(count, Count);
|
||||||
}
|
}
|
||||||
|
|
||||||
private readonly void CheckNoResizeCapacity(int index, int count)
|
private readonly void CheckNoResizeCapacity(int index, int count)
|
||||||
@@ -148,7 +150,7 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
if (index + count > Capacity)
|
if (index + count > Capacity)
|
||||||
{
|
{
|
||||||
throw new Exception(
|
throw new Exception(
|
||||||
$"AddNoResize assumes that list capacity is sufficient (Capacity {Capacity}, Size {Size}), requested count {count}!"
|
$"AddNoResize assumes that list capacity is sufficient (Capacity {Capacity}, Size {Count}), requested count {count}!"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -165,12 +167,12 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
throw new ArgumentOutOfRangeException($"Value for index {index} must be positive.");
|
throw new ArgumentOutOfRangeException($"Value for index {index} must be positive.");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (index > Size)
|
if (index > Count)
|
||||||
{
|
{
|
||||||
throw new ArgumentOutOfRangeException($"Value for index {index} is out of bounds.");
|
throw new ArgumentOutOfRangeException($"Value for index {index} is out of bounds.");
|
||||||
}
|
}
|
||||||
|
|
||||||
if (index + count > Size)
|
if (index + count > Count)
|
||||||
{
|
{
|
||||||
throw new ArgumentOutOfRangeException($"Value for count {count} is out of bounds.");
|
throw new ArgumentOutOfRangeException($"Value for count {count} is out of bounds.");
|
||||||
}
|
}
|
||||||
@@ -178,37 +180,37 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
|
|
||||||
public void Add(T value)
|
public void Add(T value)
|
||||||
{
|
{
|
||||||
if (_size >= Capacity)
|
if (_count >= Capacity)
|
||||||
{
|
{
|
||||||
ReAlloc(Capacity + (int)(Capacity * 0.5f));
|
Resize(Capacity + (int)(Capacity * 0.5f));
|
||||||
}
|
}
|
||||||
|
|
||||||
UnsafeUtilities.WriteArrayElement(Buffer, _size, value);
|
UnsafeUtilities.WriteArrayElement(_array.GetUnsafePtr(), _count, value);
|
||||||
_size++;
|
_count++;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void AddNoResize(T value)
|
public void AddNoResize(T value)
|
||||||
{
|
{
|
||||||
CheckNoResizeCapacity(1);
|
CheckNoResizeCapacity(1);
|
||||||
|
|
||||||
UnsafeUtilities.WriteArrayElement(Buffer, _size, value);
|
UnsafeUtilities.WriteArrayElement(_array.GetUnsafePtr(), _count, value);
|
||||||
_size++;
|
_count++;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void AddRange(Span<T> values, int count)
|
public void AddRange(Span<T> values, int count)
|
||||||
{
|
{
|
||||||
var newSize = _size + count;
|
var newSize = _count + count;
|
||||||
if (newSize > Capacity)
|
if (newSize > Capacity)
|
||||||
{
|
{
|
||||||
ReAlloc(Capacity + count);
|
Resize(Capacity + count);
|
||||||
}
|
}
|
||||||
|
|
||||||
fixed (T* ptr = values)
|
fixed (T* ptr = values)
|
||||||
{
|
{
|
||||||
MemCpy(_array.Buffer + _size, ptr, (uint)(count * sizeof(T)));
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), _count), ptr, (uint)(count * sizeof(T)));
|
||||||
}
|
}
|
||||||
|
|
||||||
_size += count;
|
_count += count;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void AddRangeNoResize(ReadOnlySpan<T> values)
|
public void AddRangeNoResize(ReadOnlySpan<T> values)
|
||||||
@@ -217,18 +219,18 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
|
|
||||||
fixed (T* ptr = values)
|
fixed (T* ptr = values)
|
||||||
{
|
{
|
||||||
MemCpy(_array.Buffer + _size, ptr, (uint)(values.Length * sizeof(T)));
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), _count), ptr, (uint)(values.Length * sizeof(T)));
|
||||||
}
|
}
|
||||||
|
|
||||||
_size += values.Length;
|
_count += values.Length;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void AddRangeNoResize(T* ptr, int count)
|
public void AddRangeNoResize(T* ptr, int count)
|
||||||
{
|
{
|
||||||
CheckNoResizeCapacity(count);
|
CheckNoResizeCapacity(count);
|
||||||
|
|
||||||
MemCpy(_array.Buffer + _size, ptr, (uint)(count * sizeof(T)));
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), _count), ptr, (uint)(count * sizeof(T)));
|
||||||
_size += count;
|
_count += count;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void RemoveRange(int start, int length)
|
public void RemoveRange(int start, int length)
|
||||||
@@ -240,13 +242,12 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
var copyFrom = Math.Min(start + length, _size);
|
var copyFrom = Math.Min(start + length, _count);
|
||||||
MemCpy(
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), start),
|
||||||
_array.Buffer + start,
|
UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), copyFrom),
|
||||||
_array.Buffer + copyFrom,
|
(uint)((_count - copyFrom) * sizeof(T))
|
||||||
(uint)((_size - copyFrom) * sizeof(T))
|
|
||||||
);
|
);
|
||||||
_size -= length;
|
_count -= length;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void RemoveAt(int index)
|
public void RemoveAt(int index)
|
||||||
@@ -263,13 +264,12 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
var copyFrom = Math.Min(_size - length, start + length);
|
var copyFrom = Math.Min(_count - length, start + length);
|
||||||
MemCpy(
|
MemCpy(UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), start),
|
||||||
_array.Buffer + start,
|
UnsafeUtilities.ReadArrayElementUnsafe<T>(_array.GetUnsafePtr(), copyFrom),
|
||||||
_array.Buffer + copyFrom,
|
(uint)((_count - copyFrom) * sizeof(T))
|
||||||
(uint)((_size - copyFrom) * sizeof(T))
|
|
||||||
);
|
);
|
||||||
_size -= length;
|
_count -= length;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void RemoveAtSwapBack(int index)
|
public void RemoveAtSwapBack(int index)
|
||||||
@@ -277,26 +277,47 @@ public unsafe struct UnsafeList<T> : IUnsafeCollection<T>, IEnumerable<T>
|
|||||||
RemoveRangeSwapBack(index, 1);
|
RemoveRangeSwapBack(index, 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ReAlloc(int newSize)
|
public void Resize(int newSize)
|
||||||
{
|
{
|
||||||
_array.ReAlloc(newSize);
|
_array.Resize(newSize);
|
||||||
|
|
||||||
if (_size > newSize)
|
if (_count > newSize)
|
||||||
{
|
{
|
||||||
_size = newSize;
|
_count = newSize;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Clear()
|
public void Clear()
|
||||||
{
|
{
|
||||||
_array.Clear();
|
_array.Clear();
|
||||||
_size = 0;
|
_count = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Returns a pointer to the underlying data of the array in an unsafe manner. This method is optimized for
|
||||||
|
/// performance.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>A pointer to the array's data.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly void* GetUnsafePtr()
|
||||||
|
{
|
||||||
|
return _array.GetUnsafePtr();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Converts the current array to an UnsafeArray representation using its pointer and count.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>Returns a new UnsafeArray instance initialized with the array's unsafe pointer and its count.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly UnsafeArray<T> AsUnsafeArray()
|
||||||
|
{
|
||||||
|
return new UnsafeArray<T>(_array.GetUnsafePtr(), _count);
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
_array.Dispose();
|
_array.Dispose();
|
||||||
_size = 0;
|
_count = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,20 +1,75 @@
|
|||||||
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
using Misaki.HighPerformance.Unsafe.Collections.Contracts;
|
||||||
using Misaki.HighPerformance.Unsafe.Helpers;
|
using Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
using System.Collections;
|
||||||
using System.Diagnostics.CodeAnalysis;
|
using System.Diagnostics.CodeAnalysis;
|
||||||
using System.Runtime.CompilerServices;
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Collections;
|
namespace Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// A structure that implements a queue using unmanaged types for efficient memory management.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents the type of elements stored in the queue, which must be an unmanaged type for performance and safety.</typeparam>
|
||||||
public unsafe struct UnsafeQueue<T> : IUnsafeCollection<T>
|
public unsafe struct UnsafeQueue<T> : IUnsafeCollection<T>
|
||||||
where T : unmanaged
|
where T : unmanaged
|
||||||
{
|
{
|
||||||
|
private struct Enumerator : IEnumerator<T>
|
||||||
|
{
|
||||||
|
private UnsafeQueue<T>* _collection;
|
||||||
|
private int _index;
|
||||||
|
private T _value;
|
||||||
|
|
||||||
|
public Enumerator(UnsafeQueue<T>* collection)
|
||||||
|
{
|
||||||
|
_collection = collection;
|
||||||
|
_index = -1;
|
||||||
|
_value = default;
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public bool MoveNext()
|
||||||
|
{
|
||||||
|
_index++;
|
||||||
|
if (_index < _collection->_count)
|
||||||
|
{
|
||||||
|
_value = UnsafeUtilities.ReadArrayElement<T>(_collection->_array.GetUnsafePtr(), _index);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
_value = default;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Reset()
|
||||||
|
{
|
||||||
|
_index = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Let NativeArray indexer check for out of range.
|
||||||
|
public readonly T Current
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _value;
|
||||||
|
}
|
||||||
|
|
||||||
|
readonly object IEnumerator.Current
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => Current;
|
||||||
|
}
|
||||||
|
|
||||||
|
public readonly void Dispose()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private UnsafeArray<T> _array;
|
private UnsafeArray<T> _array;
|
||||||
private int _size;
|
private int _count;
|
||||||
private int _offset;
|
private int _offset;
|
||||||
|
|
||||||
public readonly T* Buffer => _array.Buffer;
|
public readonly int Count => _count;
|
||||||
public readonly int Size => _size;
|
public readonly int Capacity => _array.Count;
|
||||||
public readonly int Capacity => _array.Size;
|
public readonly bool IsCreated => _array.IsCreated;
|
||||||
|
|
||||||
public readonly ref T this[int index]
|
public readonly ref T this[int index]
|
||||||
{
|
{
|
||||||
@@ -22,10 +77,13 @@ public unsafe struct UnsafeQueue<T> : IUnsafeCollection<T>
|
|||||||
get => ref _array[index];
|
get => ref _array[index];
|
||||||
}
|
}
|
||||||
|
|
||||||
public UnsafeQueue(int capacity, AllocationType allocationType)
|
public IEnumerator<T> GetEnumerator() => new Enumerator((UnsafeQueue<T>*)UnsafeUtilities.AddressOf(ref this));
|
||||||
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
||||||
|
|
||||||
|
public UnsafeQueue(int capacity, Allocator allocator, AllocationType allocationType = AllocationType.UnInitialized)
|
||||||
{
|
{
|
||||||
_array = new UnsafeArray<T>(capacity, allocationType);
|
_array = new UnsafeArray<T>(capacity, allocator, allocationType);
|
||||||
_size = 0;
|
_count = 0;
|
||||||
_offset = 0;
|
_offset = 0;
|
||||||
|
|
||||||
if (allocationType == AllocationType.Clear)
|
if (allocationType == AllocationType.Clear)
|
||||||
@@ -34,34 +92,49 @@ public unsafe struct UnsafeQueue<T> : IUnsafeCollection<T>
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds an element to the end of a collection, resizing if the current capacity is reached. The new element is
|
||||||
|
/// stored in a circular buffer.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="value">The item to be added to the collection.</param>
|
||||||
public void Enqueue(T value)
|
public void Enqueue(T value)
|
||||||
{
|
{
|
||||||
if (_size >= Capacity)
|
if (_count >= Capacity)
|
||||||
{
|
{
|
||||||
ReAlloc(Capacity + (int)(Capacity * 0.5f));
|
Resize(Capacity + (int)(Capacity * 0.5f));
|
||||||
}
|
}
|
||||||
|
|
||||||
UnsafeUtilities.WriteArrayElement(Buffer, (_offset + _size) % Capacity, value);
|
UnsafeUtilities.WriteArrayElement(_array.GetUnsafePtr(), (_offset + _count) % Capacity, value);
|
||||||
_size++;
|
_count++;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Removes and returns the element at the front of the queue. If the queue is empty, an exception is thrown.
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>The element that was removed from the front of the queue.</returns>
|
||||||
|
/// <exception cref="InvalidOperationException">Thrown when attempting to dequeue from an empty queue.</exception>
|
||||||
public T Dequeue()
|
public T Dequeue()
|
||||||
{
|
{
|
||||||
if (_size == 0)
|
if (_count == 0)
|
||||||
{
|
{
|
||||||
throw new InvalidOperationException("Queue is empty.");
|
throw new InvalidOperationException("Queue is empty.");
|
||||||
}
|
}
|
||||||
|
|
||||||
var value = UnsafeUtilities.ReadArrayElement<T>(Buffer, _offset);
|
var value = UnsafeUtilities.ReadArrayElement<T>(_array.GetUnsafePtr(), _offset);
|
||||||
_offset = (_offset + 1) % Capacity;
|
_offset = (_offset + 1) % Capacity;
|
||||||
_size--;
|
_count--;
|
||||||
|
|
||||||
return value;
|
return value;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Attempts to remove and return an item from a collection. Returns a boolean indicating success or failure.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="value">The output variable that will hold the dequeued item if the operation is successful.</param>
|
||||||
|
/// <returns>True if an item was successfully dequeued, otherwise false.</returns>
|
||||||
public bool TryDequeue([MaybeNullWhen(false)] out T value)
|
public bool TryDequeue([MaybeNullWhen(false)] out T value)
|
||||||
{
|
{
|
||||||
if (_size == 0)
|
if (_count == 0)
|
||||||
{
|
{
|
||||||
value = default;
|
value = default;
|
||||||
return false;
|
return false;
|
||||||
@@ -71,27 +144,33 @@ public unsafe struct UnsafeQueue<T> : IUnsafeCollection<T>
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ReAlloc(int newSize)
|
public void Resize(int newSize)
|
||||||
{
|
{
|
||||||
_array.ReAlloc(newSize);
|
_array.Resize(newSize);
|
||||||
|
|
||||||
if (_size > newSize)
|
if (_count > newSize)
|
||||||
{
|
{
|
||||||
_size = newSize;
|
_count = newSize;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Clear()
|
public void Clear()
|
||||||
{
|
{
|
||||||
_array.Clear();
|
_array.Clear();
|
||||||
_size = 0;
|
_count = 0;
|
||||||
_offset = 0;
|
_offset = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public readonly void* GetUnsafePtr()
|
||||||
|
{
|
||||||
|
return _array.GetUnsafePtr();
|
||||||
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
_array.Dispose();
|
_array.Dispose();
|
||||||
_size = 0;
|
_count = 0;
|
||||||
_offset = 0;
|
_offset = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
491
Misaki.HighPerformance.Unsafe/Helpers/HashMapHelper.cs
Normal file
491
Misaki.HighPerformance.Unsafe/Helpers/HashMapHelper.cs
Normal file
@@ -0,0 +1,491 @@
|
|||||||
|
using Misaki.HighPerformance.Mathematics;
|
||||||
|
using Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
using System.Numerics;
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
|
||||||
|
public unsafe struct HashMapHelper<TKey> : IDisposable
|
||||||
|
where TKey : unmanaged, IEquatable<TKey>
|
||||||
|
{
|
||||||
|
internal unsafe struct Enumerator
|
||||||
|
{
|
||||||
|
public HashMapHelper<TKey>* buffer;
|
||||||
|
public int index;
|
||||||
|
public int bucketIndex;
|
||||||
|
public int nextIndex;
|
||||||
|
|
||||||
|
public unsafe Enumerator(HashMapHelper<TKey>* data)
|
||||||
|
{
|
||||||
|
buffer = data;
|
||||||
|
index = -1;
|
||||||
|
bucketIndex = 0;
|
||||||
|
nextIndex = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public bool MoveNext()
|
||||||
|
{
|
||||||
|
return buffer->MoveNext(ref bucketIndex, ref nextIndex, out index);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Reset()
|
||||||
|
{
|
||||||
|
index = -1;
|
||||||
|
bucketIndex = 0;
|
||||||
|
nextIndex = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public KeyValuePair<TKey, TValue> GetCurrent<TValue>()
|
||||||
|
where TValue : unmanaged
|
||||||
|
{
|
||||||
|
return new KeyValuePair<TKey, TValue>(buffer->_keys[index], UnsafeUtilities.ReadArrayElement<TValue>(buffer->_buffer, index));
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public TKey GetCurrentKey()
|
||||||
|
{
|
||||||
|
if (index != -1)
|
||||||
|
{
|
||||||
|
return buffer->_keys[index];
|
||||||
|
}
|
||||||
|
|
||||||
|
return default;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// This buffer has 4 parts: TValue, TKey, Next, Buckets.
|
||||||
|
private byte* _buffer;
|
||||||
|
|
||||||
|
internal TKey* _keys;
|
||||||
|
internal int* _next;
|
||||||
|
internal int* _buckets;
|
||||||
|
|
||||||
|
private int _count;
|
||||||
|
private int _capacity;
|
||||||
|
private int _bucketCapacity;
|
||||||
|
private int _allocatedIndex;
|
||||||
|
private int _firstFreeIndex;
|
||||||
|
|
||||||
|
private readonly int _sizeOfTValue;
|
||||||
|
private readonly int _log2MinGrowth;
|
||||||
|
|
||||||
|
public const int MINIMAL_CAPACITY = 64;
|
||||||
|
|
||||||
|
public readonly byte* Buffer
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public readonly int Count
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _count;
|
||||||
|
}
|
||||||
|
|
||||||
|
public readonly int Capacity
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _capacity;
|
||||||
|
}
|
||||||
|
|
||||||
|
public readonly bool IsCreated
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => _buffer != null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public readonly bool IsEmpty
|
||||||
|
{
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
get => !IsCreated || _count == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
private readonly int CalcCapacityCeilPow2(int capacity)
|
||||||
|
{
|
||||||
|
capacity = Math.Max(Math.Max(1, _count), capacity);
|
||||||
|
var newCapacity = Math.Max(capacity, 1 << _log2MinGrowth);
|
||||||
|
var result = MathUtilities.CeilPow2(newCapacity);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static int CalculateDataSize(int capacity, int bucketCapacity, int sizeOfTValue, out int outKeyOffset, out int outNextOffset, out int outBucketOffset)
|
||||||
|
{
|
||||||
|
var sizeOfTKey = sizeof(TKey);
|
||||||
|
var sizeOfInt = sizeof(int);
|
||||||
|
|
||||||
|
var valuesSize = sizeOfTValue * capacity;
|
||||||
|
var keysSize = sizeOfTKey * capacity;
|
||||||
|
var nextSize = sizeOfInt * capacity;
|
||||||
|
var bucketSize = sizeOfInt * bucketCapacity;
|
||||||
|
var totalSize = valuesSize + keysSize + nextSize + bucketSize;
|
||||||
|
|
||||||
|
outKeyOffset = 0 + valuesSize;
|
||||||
|
outNextOffset = outKeyOffset + keysSize;
|
||||||
|
outBucketOffset = outNextOffset + nextSize;
|
||||||
|
|
||||||
|
return totalSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
public HashMapHelper(int capacity, int sizeOfTValue, uint minGrowth)
|
||||||
|
{
|
||||||
|
if (capacity <= 0)
|
||||||
|
{
|
||||||
|
throw new ArgumentOutOfRangeException(nameof(capacity), "Capacity must be greater than zero.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sizeOfTValue <= 0)
|
||||||
|
{
|
||||||
|
throw new ArgumentOutOfRangeException(nameof(sizeOfTValue), "Size of TValue must be greater than zero.");
|
||||||
|
}
|
||||||
|
|
||||||
|
_capacity = CalcCapacityCeilPow2(capacity);
|
||||||
|
_bucketCapacity = _capacity * 2;
|
||||||
|
|
||||||
|
var totalSize = CalculateDataSize(_capacity, _bucketCapacity, sizeOfTValue,
|
||||||
|
out var keyOffset, out var nextOffset, out var bucketOffset);
|
||||||
|
|
||||||
|
_buffer = (byte*)Malloc((nuint)totalSize);
|
||||||
|
_keys = (TKey*)(_buffer + keyOffset);
|
||||||
|
_next = (int*)(_buffer + nextOffset);
|
||||||
|
_buckets = (int*)(_buffer + bucketOffset);
|
||||||
|
|
||||||
|
Clear();
|
||||||
|
|
||||||
|
_sizeOfTValue = sizeOfTValue;
|
||||||
|
_log2MinGrowth = BitOperations.Log2(minGrowth);
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
private readonly int GetBucket(in TKey key)
|
||||||
|
{
|
||||||
|
return (int)((uint)key.GetHashCode() & _bucketCapacity - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
private readonly void CheckIndexOutOfBounds(int idx)
|
||||||
|
{
|
||||||
|
if ((uint)idx >= (uint)_capacity)
|
||||||
|
{
|
||||||
|
throw new InvalidOperationException($"Internal HashMap error. idx {idx}");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
internal void ResizeExact(int newCapacity, int newBucketCapacity)
|
||||||
|
{
|
||||||
|
var totalSize = CalculateDataSize(newCapacity, newBucketCapacity, _sizeOfTValue,
|
||||||
|
out var keyOffset, out var nextOffset, out var bucketOffset);
|
||||||
|
|
||||||
|
var oldBuffer = _buffer;
|
||||||
|
var oldKeys = _keys;
|
||||||
|
var oldNext = _next;
|
||||||
|
var oldBuckets = _buckets;
|
||||||
|
var oldBucketCapacity = _bucketCapacity;
|
||||||
|
|
||||||
|
_buffer = (byte*)Malloc((nuint)totalSize);
|
||||||
|
_keys = (TKey*)(_buffer + keyOffset);
|
||||||
|
_next = (int*)(_buffer + nextOffset);
|
||||||
|
_buckets = (int*)(_buffer + bucketOffset);
|
||||||
|
_capacity = newCapacity;
|
||||||
|
_bucketCapacity = newBucketCapacity;
|
||||||
|
|
||||||
|
Clear();
|
||||||
|
|
||||||
|
for (int i = 0, num = oldBucketCapacity; i < num; ++i)
|
||||||
|
{
|
||||||
|
for (var idx = oldBuckets[i]; idx != -1; idx = oldNext[idx])
|
||||||
|
{
|
||||||
|
var newIdx = TryAdd(oldKeys[idx]);
|
||||||
|
MemCpy(_buffer + _sizeOfTValue * newIdx, oldBuffer + _sizeOfTValue * idx, (nuint)_sizeOfTValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Free(oldBuffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
internal void Resize(int newCapacity)
|
||||||
|
{
|
||||||
|
newCapacity = Math.Max(newCapacity, _count);
|
||||||
|
var newBucketCapacity = MathUtilities.CeilPow2(newCapacity * 2);
|
||||||
|
|
||||||
|
if (_capacity == newCapacity && _bucketCapacity == newBucketCapacity)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
ResizeExact(newCapacity, newBucketCapacity);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void TrimExcess()
|
||||||
|
{
|
||||||
|
var capacity = CalcCapacityCeilPow2(_count);
|
||||||
|
ResizeExact(capacity, capacity * 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
public int Find(TKey key)
|
||||||
|
{
|
||||||
|
if (_allocatedIndex <= 0)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// First find the slot based on the hash
|
||||||
|
var bucket = GetBucket(key);
|
||||||
|
var entryIdx = _buckets[bucket];
|
||||||
|
|
||||||
|
if ((uint)entryIdx < (uint)_capacity)
|
||||||
|
{
|
||||||
|
var nextPtrs = _next;
|
||||||
|
var test = UnsafeUtilities.ReadArrayElement<TKey>(_keys, entryIdx);
|
||||||
|
var test2 = UnsafeUtilities.ReadArrayElement<TKey>(_next, 0);
|
||||||
|
while (!UnsafeUtilities.ReadArrayElement<TKey>(_keys, entryIdx).Equals(key))
|
||||||
|
{
|
||||||
|
entryIdx = nextPtrs[entryIdx];
|
||||||
|
if ((uint)entryIdx >= (uint)_capacity)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return entryIdx;
|
||||||
|
}
|
||||||
|
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int TryAdd(in TKey key)
|
||||||
|
{
|
||||||
|
if (Find(key) != -1)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Allocate an entry from the free list
|
||||||
|
int idx;
|
||||||
|
int* next;
|
||||||
|
|
||||||
|
if (_allocatedIndex >= _capacity && _firstFreeIndex < 0)
|
||||||
|
{
|
||||||
|
var newCap = Math.Min(MINIMAL_CAPACITY, CalcCapacityCeilPow2(_capacity + (1 << _log2MinGrowth)));
|
||||||
|
Resize(newCap);
|
||||||
|
}
|
||||||
|
|
||||||
|
idx = _firstFreeIndex;
|
||||||
|
|
||||||
|
if (idx >= 0)
|
||||||
|
{
|
||||||
|
_firstFreeIndex = _next[idx];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
idx = _allocatedIndex++;
|
||||||
|
}
|
||||||
|
|
||||||
|
CheckIndexOutOfBounds(idx);
|
||||||
|
|
||||||
|
UnsafeUtilities.WriteArrayElement(_keys, idx, key);
|
||||||
|
var bucket = GetBucket(key);
|
||||||
|
|
||||||
|
// Add the index to the hash-map
|
||||||
|
next = _next;
|
||||||
|
next[idx] = _buckets[bucket];
|
||||||
|
_buckets[bucket] = idx;
|
||||||
|
_count++;
|
||||||
|
|
||||||
|
return idx;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int TryRemove(TKey key)
|
||||||
|
{
|
||||||
|
if (_capacity == 0)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
var removed = 0;
|
||||||
|
|
||||||
|
// First find the slot based on the hash
|
||||||
|
var bucket = GetBucket(key);
|
||||||
|
|
||||||
|
var prevEntry = -1;
|
||||||
|
var entryIdx = _buckets[bucket];
|
||||||
|
|
||||||
|
while (entryIdx >= 0 && entryIdx < _capacity)
|
||||||
|
{
|
||||||
|
if (UnsafeUtilities.ReadArrayElement<TKey>(_keys, entryIdx).Equals(key))
|
||||||
|
{
|
||||||
|
++removed;
|
||||||
|
|
||||||
|
// Found matching element, remove it
|
||||||
|
if (prevEntry < 0)
|
||||||
|
{
|
||||||
|
_buckets[bucket] = _next[entryIdx];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_next[prevEntry] = _next[entryIdx];
|
||||||
|
}
|
||||||
|
|
||||||
|
// And free the index
|
||||||
|
var nextIdx = _next[entryIdx];
|
||||||
|
_next[entryIdx] = _firstFreeIndex;
|
||||||
|
_firstFreeIndex = entryIdx;
|
||||||
|
entryIdx = nextIdx;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
prevEntry = entryIdx;
|
||||||
|
entryIdx = _next[entryIdx];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_count -= removed;
|
||||||
|
return 0 != removed ? removed : -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool TryGetValue<TValue>(TKey key, out TValue item)
|
||||||
|
where TValue : unmanaged
|
||||||
|
{
|
||||||
|
var idx = Find(key);
|
||||||
|
|
||||||
|
if (idx != -1)
|
||||||
|
{
|
||||||
|
item = UnsafeUtilities.ReadArrayElement<TValue>(_buffer, idx);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
item = default;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool MoveNextSearch(ref int bucketIndex, ref int nextIndex, out int index)
|
||||||
|
{
|
||||||
|
for (int i = bucketIndex, num = _bucketCapacity; i < num; ++i)
|
||||||
|
{
|
||||||
|
var idx = _buckets[i];
|
||||||
|
|
||||||
|
if (idx != -1)
|
||||||
|
{
|
||||||
|
index = idx;
|
||||||
|
bucketIndex = i + 1;
|
||||||
|
nextIndex = _next[idx];
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
index = -1;
|
||||||
|
bucketIndex = _bucketCapacity;
|
||||||
|
nextIndex = -1;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public bool MoveNext(ref int bucketIndex, ref int nextIndex, out int index)
|
||||||
|
{
|
||||||
|
if (nextIndex != -1)
|
||||||
|
{
|
||||||
|
index = nextIndex;
|
||||||
|
nextIndex = _next[nextIndex];
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return MoveNextSearch(ref bucketIndex, ref nextIndex, out index);
|
||||||
|
}
|
||||||
|
|
||||||
|
internal UnsafeArray<TKey> GetKeyArray(Allocator allocator)
|
||||||
|
{
|
||||||
|
var result = new UnsafeArray<TKey>(_count, allocator, AllocationType.UnInitialized);
|
||||||
|
|
||||||
|
for (int i = 0, count = 0, max = result.Count, capacity = _bucketCapacity; i < capacity && count < max; i++)
|
||||||
|
{
|
||||||
|
var bucket = _buckets[i];
|
||||||
|
|
||||||
|
while (bucket != -1)
|
||||||
|
{
|
||||||
|
result[count++] = UnsafeUtilities.ReadArrayElement<TKey>(_keys, bucket);
|
||||||
|
bucket = _next[bucket];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
internal UnsafeArray<TValue> GetValueArray<TValue>(Allocator allocator)
|
||||||
|
where TValue : unmanaged
|
||||||
|
{
|
||||||
|
var result = new UnsafeArray<TValue>(_count, allocator, AllocationType.UnInitialized);
|
||||||
|
|
||||||
|
for (int i = 0, count = 0, max = result.Count, capacity = _bucketCapacity; i < capacity && count < max; ++i)
|
||||||
|
{
|
||||||
|
var bucket = _buckets[i];
|
||||||
|
|
||||||
|
while (bucket != -1)
|
||||||
|
{
|
||||||
|
result[count++] = UnsafeUtilities.ReadArrayElement<TValue>(_buffer, bucket);
|
||||||
|
bucket = _next[bucket];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
public UnsafeArray<KeyValuePair<TKey, TValue>> GetKeyValueArrays<TValue>(Allocator allocator)
|
||||||
|
where TValue : unmanaged
|
||||||
|
{
|
||||||
|
var result = new UnsafeArray<KeyValuePair<TKey, TValue>>(_count, allocator, AllocationType.UnInitialized);
|
||||||
|
|
||||||
|
for (int i = 0, count = 0, max = result.Count, capacity = _bucketCapacity; i < capacity && count < max; i++)
|
||||||
|
{
|
||||||
|
var bucket = _buckets[i];
|
||||||
|
|
||||||
|
while (bucket != -1)
|
||||||
|
{
|
||||||
|
result[count] = new(UnsafeUtilities.ReadArrayElement<TKey>(_keys, bucket),
|
||||||
|
UnsafeUtilities.ReadArrayElement<TValue>(_buffer, bucket));
|
||||||
|
|
||||||
|
count++;
|
||||||
|
bucket = _next[bucket];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Clear()
|
||||||
|
{
|
||||||
|
MemSet(_buckets, 0xff, (nuint)_bucketCapacity * sizeof(int));
|
||||||
|
MemSet(_next, 0xff, (nuint)_capacity * sizeof(int));
|
||||||
|
|
||||||
|
_count = 0;
|
||||||
|
_firstFreeIndex = -1;
|
||||||
|
_allocatedIndex = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Dispose()
|
||||||
|
{
|
||||||
|
if (IsCreated)
|
||||||
|
{
|
||||||
|
Free(_buffer);
|
||||||
|
|
||||||
|
_buffer = null;
|
||||||
|
_keys = null;
|
||||||
|
_next = null;
|
||||||
|
_buckets = null;
|
||||||
|
|
||||||
|
_count = 0;
|
||||||
|
_capacity = 0;
|
||||||
|
_bucketCapacity = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -13,6 +13,76 @@ public static unsafe class MemoryUtilities
|
|||||||
public T data;
|
public T data;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Allocates a block of memory of the specified size in bytes.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="size">Specifies the number of bytes to allocate in memory.</param>
|
||||||
|
/// <returns>Returns a pointer to the allocated memory block.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void* Malloc(nuint size)
|
||||||
|
{
|
||||||
|
return NativeMemory.Alloc(size);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Allocates a block of memory with a specified size and alignment.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="size">Specifies the total number of bytes to allocate for the memory block.</param>
|
||||||
|
/// <param name="alignment">Defines the required alignment for the allocated memory address.</param>
|
||||||
|
/// <returns>Returns a pointer to the allocated memory block.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void* AlignedAlloc(nuint size, nuint alignment)
|
||||||
|
{
|
||||||
|
return NativeMemory.AlignedAlloc(size, alignment);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Resizes a previously allocated memory block to a new size. It returns a pointer to the reallocated memory.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="ptr">The pointer to the memory block that needs to be resized.</param>
|
||||||
|
/// <param name="size">The new size for the memory block after resizing.</param>
|
||||||
|
/// <returns>A pointer to the reallocated memory block, or null if the operation fails.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void* Realloc(void* ptr, nuint size)
|
||||||
|
{
|
||||||
|
return NativeMemory.Realloc(ptr, size);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Reallocates memory to a specified size with a given alignment. It returns a pointer to the newly allocated
|
||||||
|
/// memory.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="ptr">The pointer to the existing memory block that needs to be reallocated.</param>
|
||||||
|
/// <param name="size">The new size for the memory allocation.</param>
|
||||||
|
/// <param name="alignment">The required alignment for the new memory allocation.</param>
|
||||||
|
/// <returns>A pointer to the reallocated memory block, or null if the allocation fails.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void* AlignedRealloc(void* ptr, nuint size, nuint alignment)
|
||||||
|
{
|
||||||
|
return NativeMemory.AlignedRealloc(ptr, size, alignment);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Releases the allocated memory pointed to by the given pointer. This helps in managing memory usage effectively.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="ptr">The pointer to the memory block that needs to be freed.</param>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void Free(void* ptr)
|
||||||
|
{
|
||||||
|
NativeMemory.Free(ptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Releases memory that was allocated with alignment requirements. It ensures proper deallocation of aligned memory
|
||||||
|
/// blocks.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="ptr">The pointer to the memory block that needs to be freed.</param>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static void AlignedFree(void* ptr)
|
||||||
|
{
|
||||||
|
NativeMemory.AlignedFree(ptr);
|
||||||
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Clears a block of memory by setting it to zero. It initializes a specified number of bytes at a given memory
|
/// Clears a block of memory by setting it to zero. It initializes a specified number of bytes at a given memory
|
||||||
/// address.
|
/// address.
|
||||||
@@ -32,7 +102,7 @@ public static unsafe class MemoryUtilities
|
|||||||
/// <param name="size">The number of bytes to set to the specified value.</param>
|
/// <param name="size">The number of bytes to set to the specified value.</param>
|
||||||
/// <param name="value">The byte value to which the memory block will be initialized.</param>
|
/// <param name="value">The byte value to which the memory block will be initialized.</param>
|
||||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
public static void MemSet(void* ptr, nuint size, byte value)
|
public static void MemSet(void* ptr, byte value, nuint size)
|
||||||
{
|
{
|
||||||
NativeMemory.Fill(ptr, size, value);
|
NativeMemory.Fill(ptr, size, value);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,10 @@
|
|||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Helpers;
|
namespace Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Provides extension methods for copying elements between unsafe collections and spans, converting collections to
|
||||||
|
/// arrays or lists, and searching for values.
|
||||||
|
/// </summary>
|
||||||
public unsafe static class UnsafeCollectionExtensions
|
public unsafe static class UnsafeCollectionExtensions
|
||||||
{
|
{
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -13,14 +17,37 @@ public unsafe static class UnsafeCollectionExtensions
|
|||||||
/// <exception cref="ArgumentException">Thrown when the sizes of the source collection and destination span do not match.</exception>
|
/// <exception cref="ArgumentException">Thrown when the sizes of the source collection and destination span do not match.</exception>
|
||||||
public static void CopyTo<T>(this IUnsafeCollection<T> source, Span<T> destination) where T : unmanaged
|
public static void CopyTo<T>(this IUnsafeCollection<T> source, Span<T> destination) where T : unmanaged
|
||||||
{
|
{
|
||||||
if (source.Size > destination.Length)
|
if (source.Count > destination.Length)
|
||||||
{
|
{
|
||||||
throw new ArgumentException("Source collection is larger than the destination span.");
|
throw new ArgumentException("Source collection is larger than the destination span.");
|
||||||
}
|
}
|
||||||
|
|
||||||
fixed (T* ptr = destination)
|
fixed (T* ptr = destination)
|
||||||
{
|
{
|
||||||
SystemUnsfae.CopyBlock(ptr, source.Buffer, (uint)(source.Size * sizeof(T)));
|
SystemUnsfae.CopyBlock(ptr, source.GetUnsafePtr(), (uint)(source.Count * sizeof(T)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Copies a range of elements from a source collection to a destination span, ensuring both are adequately sized.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Specifies the type of elements being copied, which must be a value type.</typeparam>
|
||||||
|
/// <param name="source">The collection from which elements are copied.</param>
|
||||||
|
/// <param name="destination">The span where the elements will be copied to.</param>
|
||||||
|
/// <param name="sourceIndex">The starting index in the source collection for the copy operation.</param>
|
||||||
|
/// <param name="destinationIndex">The starting index in the destination span where the elements will be placed.</param>
|
||||||
|
/// <param name="length">The number of elements to copy from the source to the destination.</param>
|
||||||
|
/// <exception cref="ArgumentException">Thrown when the specified range exceeds the bounds of the source collection or destination span.</exception>
|
||||||
|
public static void CopyTo<T>(this IUnsafeCollection<T> source, Span<T> destination, int sourceIndex, int destinationIndex, int length) where T : unmanaged
|
||||||
|
{
|
||||||
|
if (sourceIndex + length > source.Count || destinationIndex + length > destination.Length)
|
||||||
|
{
|
||||||
|
throw new ArgumentException("Source collection or destination span is too small for the specified range.");
|
||||||
|
}
|
||||||
|
|
||||||
|
fixed (T* ptr = destination)
|
||||||
|
{
|
||||||
|
SystemUnsfae.CopyBlock(ptr + destinationIndex, (byte*)source.GetUnsafePtr() + (sourceIndex * sizeof(T)), (uint)(length * sizeof(T)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -33,14 +60,37 @@ public unsafe static class UnsafeCollectionExtensions
|
|||||||
/// <exception cref="ArgumentException">Thrown when the source span and destination collection have different sizes.</exception>
|
/// <exception cref="ArgumentException">Thrown when the source span and destination collection have different sizes.</exception>
|
||||||
public static void CopyFrom<T>(this IUnsafeCollection<T> destination, Span<T> source) where T : unmanaged
|
public static void CopyFrom<T>(this IUnsafeCollection<T> destination, Span<T> source) where T : unmanaged
|
||||||
{
|
{
|
||||||
if (destination.Size > source.Length)
|
if (destination.Count > source.Length)
|
||||||
{
|
{
|
||||||
throw new ArgumentException("Destination collection is larger than the source span.");
|
throw new ArgumentException("Destination collection is larger than the source span.");
|
||||||
}
|
}
|
||||||
|
|
||||||
fixed (T* ptr = source)
|
fixed (T* ptr = source)
|
||||||
{
|
{
|
||||||
SystemUnsfae.CopyBlock(destination.Buffer, ptr, (uint)(source.Length * sizeof(T)));
|
SystemUnsfae.CopyBlock(destination.GetUnsafePtr(), ptr, (uint)(source.Length * sizeof(T)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Copies a specified range of elements from a source span to a destination collection.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents the type of elements being copied, which must be unmanaged.</typeparam>
|
||||||
|
/// <param name="destination">The collection where elements will be copied to.</param>
|
||||||
|
/// <param name="source">The span containing the elements to be copied.</param>
|
||||||
|
/// <param name="sourceIndex">The starting index in the source span from which to begin copying.</param>
|
||||||
|
/// <param name="destinationIndex">The starting index in the destination collection where the elements will be placed.</param>
|
||||||
|
/// <param name="length">The number of elements to copy from the source span to the destination collection.</param>
|
||||||
|
/// <exception cref="ArgumentException">Thrown when the specified range exceeds the bounds of the source span or destination collection.</exception>
|
||||||
|
public static void CopyFrom<T>(this IUnsafeCollection<T> destination, Span<T> source, int sourceIndex, int destinationIndex, int length) where T : unmanaged
|
||||||
|
{
|
||||||
|
if (sourceIndex + length > source.Length || destinationIndex + length > destination.Count)
|
||||||
|
{
|
||||||
|
throw new ArgumentException("Source span or destination collection is too small for the specified range.");
|
||||||
|
}
|
||||||
|
|
||||||
|
fixed (T* ptr = source)
|
||||||
|
{
|
||||||
|
SystemUnsfae.CopyBlock((byte*)destination.GetUnsafePtr() + (destinationIndex * sizeof(T)), ptr + sourceIndex, (uint)(length * sizeof(T)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -52,10 +102,10 @@ public unsafe static class UnsafeCollectionExtensions
|
|||||||
/// <returns>A new collection containing the elements from the UnsafeCollection.</returns>
|
/// <returns>A new collection containing the elements from the UnsafeCollection.</returns>
|
||||||
public static T[] ToArray<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
public static T[] ToArray<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
||||||
{
|
{
|
||||||
var array = new T[source.Size];
|
var array = new T[source.Count];
|
||||||
fixed (T* ptr = array)
|
fixed (T* ptr = array)
|
||||||
{
|
{
|
||||||
SystemUnsfae.CopyBlock(ptr, source.Buffer, (uint)(source.Size * sizeof(T)));
|
SystemUnsfae.CopyBlock(ptr, source.GetUnsafePtr(), (uint)(source.Count * sizeof(T)));
|
||||||
}
|
}
|
||||||
|
|
||||||
return array;
|
return array;
|
||||||
@@ -69,10 +119,10 @@ public unsafe static class UnsafeCollectionExtensions
|
|||||||
/// <returns>A list containing the elements from the specified unmanaged collection.</returns>
|
/// <returns>A list containing the elements from the specified unmanaged collection.</returns>
|
||||||
public static List<T> ToList<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
public static List<T> ToList<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
||||||
{
|
{
|
||||||
var list = new List<T>(source.Size);
|
var list = new List<T>(source.Count);
|
||||||
fixed (T* ptr = list.ToArray())
|
fixed (T* ptr = list.ToArray())
|
||||||
{
|
{
|
||||||
SystemUnsfae.CopyBlock(ptr, source.Buffer, (uint)(source.Size * sizeof(T)));
|
SystemUnsfae.CopyBlock(ptr, source.GetUnsafePtr(), (uint)(source.Count * sizeof(T)));
|
||||||
}
|
}
|
||||||
return list;
|
return list;
|
||||||
}
|
}
|
||||||
@@ -85,6 +135,39 @@ public unsafe static class UnsafeCollectionExtensions
|
|||||||
/// <returns>A Span that provides a view over the elements of the UnsafeCollection.</returns>
|
/// <returns>A Span that provides a view over the elements of the UnsafeCollection.</returns>
|
||||||
public static Span<T> AsSpan<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
public static Span<T> AsSpan<T>(this IUnsafeCollection<T> source) where T : unmanaged
|
||||||
{
|
{
|
||||||
return new(source.Buffer, source.Size);
|
return new(source.GetUnsafePtr(), source.Count);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Finds the index of a specified value in a collection. Returns -1 if the value is not found.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">The type of elements in the collection, which must support equality comparison.</typeparam>
|
||||||
|
/// <param name="source">The collection to search for the specified value.</param>
|
||||||
|
/// <param name="value">The value to locate within the collection.</param>
|
||||||
|
/// <param name="index">Outputs the index of the found value or -1 if not found.</param>
|
||||||
|
public static void IndexOf<T>(this IUnsafeCollection<T> source, T value, out int index) where T : unmanaged, IEquatable<T>
|
||||||
|
{
|
||||||
|
for (var i = 0; i < source.Count; i++)
|
||||||
|
{
|
||||||
|
if (UnsafeUtilities.ReadArrayElement<T>(source.GetUnsafePtr(), i).Equals(value))
|
||||||
|
{
|
||||||
|
index = i;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
index = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Checks if a specified value exists within an unsafe collection of unmanaged types.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Represents a type that is unmanaged and supports equality comparison.</typeparam>
|
||||||
|
/// <param name="source">The collection being searched for the specified value.</param>
|
||||||
|
/// <param name="value">The value being searched for within the collection.</param>
|
||||||
|
/// <returns>Returns true if the value is found; otherwise, returns false.</returns>
|
||||||
|
public static bool Conations<T>(this IUnsafeCollection<T> source, T value) where T : unmanaged, IEquatable<T>
|
||||||
|
{
|
||||||
|
source.IndexOf(value, out var index);
|
||||||
|
return index != -1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
using System.Runtime.CompilerServices;
|
using Misaki.HighPerformance.Unsafe.Collections;
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
namespace Misaki.HighPerformance.Unsafe.Helpers;
|
namespace Misaki.HighPerformance.Unsafe.Helpers;
|
||||||
|
|
||||||
@@ -79,4 +80,18 @@ public static unsafe class UnsafeUtilities
|
|||||||
{
|
{
|
||||||
*ReadArrayElementUnsafe<T>(ptr, index) = value;
|
*ReadArrayElementUnsafe<T>(ptr, index) = value;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Converts an UnsafeArray of one unmanaged type to another unmanaged type without copying the elements.
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="TIn">Represents the type of elements in the input array.</typeparam>
|
||||||
|
/// <typeparam name="TOut">Represents the type of elements in the output array.</typeparam>
|
||||||
|
/// <param name="array">The input array containing elements of the specified input type.</param>
|
||||||
|
/// <returns>An UnsafeArray containing elements of the specified output type.</returns>
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
public static UnsafeArray<TOut> CastArray<TIn, TOut>(UnsafeArray<TIn> array)
|
||||||
|
where TIn : unmanaged where TOut : unmanaged
|
||||||
|
{
|
||||||
|
return new UnsafeArray<TOut>(array.GetUnsafePtr(), array.Count * sizeof(TIn) / sizeof(TOut));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -15,4 +15,8 @@
|
|||||||
<IsAotCompatible>True</IsAotCompatible>
|
<IsAotCompatible>True</IsAotCompatible>
|
||||||
</PropertyGroup>
|
</PropertyGroup>
|
||||||
|
|
||||||
|
<ItemGroup>
|
||||||
|
<ProjectReference Include="..\Misaki.HighPerformance.Math\Misaki.HighPerformance.Mathematics.csproj" />
|
||||||
|
</ItemGroup>
|
||||||
|
|
||||||
</Project>
|
</Project>
|
||||||
|
|||||||
@@ -9,6 +9,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Misaki.HighPerformance.Unsa
|
|||||||
EndProject
|
EndProject
|
||||||
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Misaki.HighPerformance.Test", "Misaki.HighPerformance.Test\Misaki.HighPerformance.Test.csproj", "{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}"
|
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Misaki.HighPerformance.Test", "Misaki.HighPerformance.Test\Misaki.HighPerformance.Test.csproj", "{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}"
|
||||||
EndProject
|
EndProject
|
||||||
|
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Misaki.HighPerformance.Mathematics", "Misaki.HighPerformance.Math\Misaki.HighPerformance.Mathematics.csproj", "{861F9574-2063-4B54-8D6A-AA32B26B6583}"
|
||||||
|
EndProject
|
||||||
Global
|
Global
|
||||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||||
Debug|Any CPU = Debug|Any CPU
|
Debug|Any CPU = Debug|Any CPU
|
||||||
@@ -27,6 +29,10 @@ Global
|
|||||||
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||||
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||||
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Release|Any CPU.Build.0 = Release|Any CPU
|
{90EFF5B8-22CD-4B6A-83AB-48E0E97610EA}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||||
|
{861F9574-2063-4B54-8D6A-AA32B26B6583}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||||
|
{861F9574-2063-4B54-8D6A-AA32B26B6583}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||||
|
{861F9574-2063-4B54-8D6A-AA32B26B6583}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||||
|
{861F9574-2063-4B54-8D6A-AA32B26B6583}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||||
EndGlobalSection
|
EndGlobalSection
|
||||||
GlobalSection(SolutionProperties) = preSolution
|
GlobalSection(SolutionProperties) = preSolution
|
||||||
HideSolutionNode = FALSE
|
HideSolutionNode = FALSE
|
||||||
|
|||||||
@@ -8,9 +8,6 @@ namespace Misaki.HighPerformance.Buffer
|
|||||||
private readonly Func<T> _factory;
|
private readonly Func<T> _factory;
|
||||||
private readonly ConcurrentQueue<T> _objects = new();
|
private readonly ConcurrentQueue<T> _objects = new();
|
||||||
|
|
||||||
private readonly bool _autoCleanup;
|
|
||||||
private readonly int _autoCleanupInterval;
|
|
||||||
|
|
||||||
private bool _disposed;
|
private bool _disposed;
|
||||||
|
|
||||||
public uint InitialSize
|
public uint InitialSize
|
||||||
@@ -24,13 +21,10 @@ namespace Misaki.HighPerformance.Buffer
|
|||||||
private set;
|
private set;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ObjectPool(Func<T> factory, uint initialSize = uint.MinValue, uint maxSize = uint.MaxValue, bool autoCleanup = false, int autoCleanupInterval = 1000 * 60 * 5)
|
public ObjectPool(Func<T> factory, uint initialSize = uint.MinValue, uint maxSize = uint.MaxValue)
|
||||||
{
|
{
|
||||||
_factory = factory;
|
_factory = factory;
|
||||||
|
|
||||||
_autoCleanup = autoCleanup;
|
|
||||||
_autoCleanupInterval = autoCleanupInterval;
|
|
||||||
|
|
||||||
InitialSize = initialSize;
|
InitialSize = initialSize;
|
||||||
MaxSize = maxSize;
|
MaxSize = maxSize;
|
||||||
|
|
||||||
@@ -41,40 +35,11 @@ namespace Misaki.HighPerformance.Buffer
|
|||||||
_objects.Enqueue(_factory());
|
_objects.Enqueue(_factory());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
SetupAutoCleanup();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private void PoolCleanup()
|
~ObjectPool()
|
||||||
{
|
{
|
||||||
foreach (var obj in _objects)
|
Dispose();
|
||||||
{
|
|
||||||
if (obj is IDisposable disposable)
|
|
||||||
{
|
|
||||||
disposable.Dispose();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
_objects.Clear();
|
|
||||||
|
|
||||||
GC.Collect();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void SetupAutoCleanup()
|
|
||||||
{
|
|
||||||
if (!_autoCleanup)
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
Task.Run(async () =>
|
|
||||||
{
|
|
||||||
while (true)
|
|
||||||
{
|
|
||||||
await Task.Delay(_autoCleanupInterval);
|
|
||||||
PoolCleanup();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public bool TryRent([MaybeNullWhen(false)] out T obj)
|
public bool TryRent([MaybeNullWhen(false)] out T obj)
|
||||||
@@ -100,11 +65,32 @@ namespace Misaki.HighPerformance.Buffer
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public void Reset()
|
||||||
|
{
|
||||||
|
foreach (var obj in _objects)
|
||||||
|
{
|
||||||
|
if (obj is IDisposable disposable)
|
||||||
|
{
|
||||||
|
disposable.Dispose();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_objects.Clear();
|
||||||
|
GC.Collect();
|
||||||
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
{
|
{
|
||||||
PoolCleanup();
|
if (_disposed)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Reset();
|
||||||
|
|
||||||
_disposed = true;
|
_disposed = true;
|
||||||
|
|
||||||
|
GC.SuppressFinalize(this);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,7 +2,8 @@
|
|||||||
|
|
||||||
public static class JobExtensions
|
public static class JobExtensions
|
||||||
{
|
{
|
||||||
public static JobHandle Schedule<T>(this T job, bool preferLocal = false) where T : struct, IJob
|
public static JobHandle Schedule<T>(this T job, bool preferLocal = false)
|
||||||
|
where T : struct, IJob
|
||||||
{
|
{
|
||||||
var handle = new JobHandle(1);
|
var handle = new JobHandle(1);
|
||||||
var worker = new JobWorker<T>(job, handle);
|
var worker = new JobWorker<T>(job, handle);
|
||||||
@@ -11,7 +12,8 @@ public static class JobExtensions
|
|||||||
return handle;
|
return handle;
|
||||||
}
|
}
|
||||||
|
|
||||||
public static JobHandle Schedule<T>(this T job, ReadOnlySpan<JobHandle> dependencies, bool preferLocal = false) where T : struct, IJob
|
public static JobHandle Schedule<T>(this T job, ReadOnlySpan<JobHandle> dependencies, bool preferLocal = false)
|
||||||
|
where T : struct, IJob
|
||||||
{
|
{
|
||||||
foreach (var dependency in dependencies)
|
foreach (var dependency in dependencies)
|
||||||
{
|
{
|
||||||
@@ -21,7 +23,8 @@ public static class JobExtensions
|
|||||||
return job.Schedule(preferLocal);
|
return job.Schedule(preferLocal);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static JobHandle Schedule<T>(this T job, int length, int batchCount, bool preferLocal = false) where T : struct, IJobParallelFor
|
public static JobHandle Schedule<T>(this T job, int length, int batchCount, bool preferLocal = false)
|
||||||
|
where T : struct, IJobParallelFor
|
||||||
{
|
{
|
||||||
var batchSize = (length + batchCount - 1) / batchCount;
|
var batchSize = (length + batchCount - 1) / batchCount;
|
||||||
var handle = new JobHandle(batchCount);
|
var handle = new JobHandle(batchCount);
|
||||||
@@ -37,7 +40,8 @@ public static class JobExtensions
|
|||||||
return handle;
|
return handle;
|
||||||
}
|
}
|
||||||
|
|
||||||
public static JobHandle Schedule<T>(this T job, int length, int batchCount, ReadOnlySpan<JobHandle> dependencies, bool preferLocal = false) where T : struct, IJobParallelFor
|
public static JobHandle Schedule<T>(this T job, int length, int batchCount, ReadOnlySpan<JobHandle> dependencies, bool preferLocal = false)
|
||||||
|
where T : struct, IJobParallelFor
|
||||||
{
|
{
|
||||||
foreach (var dependency in dependencies)
|
foreach (var dependency in dependencies)
|
||||||
{
|
{
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
namespace Misaki.HighPerformance.Jobs;
|
namespace Misaki.HighPerformance.Jobs;
|
||||||
|
|
||||||
internal readonly struct JobWorker<T>(T job, JobHandle handle) : IThreadPoolWorkItem where T : struct, IJob
|
internal readonly struct JobWorker<T>(T job, JobHandle handle) : IThreadPoolWorkItem
|
||||||
|
where T : struct, IJob
|
||||||
{
|
{
|
||||||
public void Execute()
|
public void Execute()
|
||||||
{
|
{
|
||||||
@@ -9,7 +10,8 @@ internal readonly struct JobWorker<T>(T job, JobHandle handle) : IThreadPoolWork
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
internal readonly struct ParallelJobWorker<T>(T job, JobHandle handle, int start, int end) : IThreadPoolWorkItem where T : struct, IJobParallelFor
|
internal readonly struct ParallelJobWorker<T>(T job, JobHandle handle, int start, int end) : IThreadPoolWorkItem
|
||||||
|
where T : struct, IJobParallelFor
|
||||||
{
|
{
|
||||||
public void Execute()
|
public void Execute()
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user