StitchATon2/Infra/Buffers/MemoryAllocator.cs

using System.Buffers;
using System.Runtime.CompilerServices;

namespace StitchATon2.Infra.Buffers;

public static class MemoryAllocator
{
    public static IBuffer<T> Allocate<T>(int count) where T : unmanaged
        => new UnmanagedMemory<T>(count);
    
    public static IMemoryOwner<T> AllocateManaged<T>(int count)
        => MemoryPool<T>.Shared.Rent(count);
    
    public static ArrayOwner<T> AllocateArray<T>(int count) where T : unmanaged
        => new(ArrayPool<T>.Shared, count);

    public static MemoryManager<T> AllocateImmovable<T>(int count) where T : unmanaged
        => new ImmovableMemory<T>(count);

    public static unsafe IBuffer<T> Clone<T>(this IBuffer<T> buffer) where T : unmanaged
    {
        if (buffer is UnmanagedMemory<T> unmanagedMemory)
        {
            var newBuffer = new UnmanagedMemory<T>(buffer.Length);
            var byteCount = (uint)(Unsafe.SizeOf<T>() * buffer.Length);
            Unsafe.CopyBlock(newBuffer.Pointer, unmanagedMemory.Pointer, byteCount);
            return newBuffer;
        }
        
        throw new NotSupportedException();
    }

    public static unsafe void Copy<T>(this IBuffer<T> source, IBuffer<T> destination, int count) where T : unmanaged
    {
        if (source is UnmanagedMemory<T> sourceBuffer && destination is UnmanagedMemory<T> destinationBuffer)
        {
            var byteCount = (uint)(Unsafe.SizeOf<T>() * count);
            Unsafe.CopyBlock(destinationBuffer.Pointer, sourceBuffer.Pointer, byteCount);
            return;
        }
        
        throw new NotSupportedException();
    }
}