MoonTools.ECS/src/Collections/NativeArray.cs

using System;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;

namespace MoonTools.ECS.Collections;

public unsafe class NativeArray<T> : IDisposable where T : unmanaged
{
	private T* Elements;
	public int Count { get; private set;}
	private int Capacity;
	private int ElementSize;

	public Span<T>.Enumerator GetEnumerator() => new Span<T>(Elements, Count).GetEnumerator();

	private bool disposed;

	public NativeArray(int capacity = 16)
	{
		this.Capacity = capacity;
		ElementSize = Unsafe.SizeOf<T>();
		Elements = (T*) NativeMemory.Alloc((nuint) (capacity * ElementSize));
		Count = 0;
	}

	public ref T this[int i] => ref Elements[i];

	public void Append(T item)
	{
		if (Count >= Capacity)
		{
			Capacity *= 2;
			Elements = (T*) NativeMemory.Realloc(Elements, (nuint) (Capacity * Unsafe.SizeOf<T>()));
		}

		Elements[Count] = item;
		Count += 1;
	}

	public void RemoveLastElement()
	{
		Count -= 1;
	}

	public bool TryPop(out T element)
	{
		if (Count > 0)
		{
			element = Elements[Count - 1];
			Count -= 1;
			return true;
		}

		element = default;
		return false;
	}

	public void Clear()
	{
		Count = 0;
	}

	private void ResizeTo(int size)
	{
		Capacity = size;
		Elements = (T*) NativeMemory.Realloc((void*) Elements, (nuint) (ElementSize * Capacity));
	}

	// Fills gap by copying final element to the deleted index
	public void Delete(int index)
	{
		if (Count > 1)
		{
			NativeMemory.Copy(
				(void*) (Elements + ((Count - 1) * ElementSize)),
				(void*) (Elements + (index * ElementSize)),
				(nuint) ElementSize
			);
		}

		Count -= 1;
	}

	public void CopyTo(NativeArray<T> other)
	{
		if (Count >= other.Capacity)
		{
			other.ResizeTo(Count);
		}

		NativeMemory.Copy(
			(void*) Elements,
			(void*) other.Elements,
			(nuint) (ElementSize * Count)
		);

		other.Count = Count;
	}

	protected virtual void Dispose(bool disposing)
	{
		if (!disposed)
		{
			NativeMemory.Free(Elements);
			Elements = null;

			disposed = true;
		}
	}

	~NativeArray()
	{
		// Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method
		Dispose(disposing: false);
	}

	public void Dispose()
	{
		// Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method
		Dispose(disposing: true);
		GC.SuppressFinalize(this);
	}
}