using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Runtime.InteropServices;
using System.Threading;
using Unity.Burst;
using Unity.Jobs;
using Unity.Mathematics;
namespace Unity.Collections.LowLevel.Unsafe
{
[BurstCompile]
internal unsafe struct UnsafeDisposeJob : IJob
{
[NativeDisableUnsafePtrRestriction]
public void* Ptr;
public AllocatorManager.AllocatorHandle Allocator;
public void Execute()
{
AllocatorManager.Free(Allocator, Ptr);
}
}
internal unsafe struct UntypedUnsafeList
{
#pragma warning disable 169
[NativeDisableUnsafePtrRestriction]
public void* Ptr;
public int m_length;
public int m_capacity;
public AllocatorManager.AllocatorHandle Allocator;
internal int obsolete_length;
internal int obsolete_capacity;
#pragma warning restore 169
}
///
/// An unmanaged, resizable list.
///
/// The type of the elements.
[DebuggerDisplay("Length = {Length}, Capacity = {Capacity}, IsCreated = {IsCreated}, IsEmpty = {IsEmpty}")]
[DebuggerTypeProxy(typeof(UnsafeListTDebugView<>))]
[StructLayout(LayoutKind.Sequential)]
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public unsafe struct UnsafeList
: INativeDisposable
, INativeList
, IEnumerable // Used by collection initializers.
where T : unmanaged
{
//
// 'Header' of this struct must binary match 'UntypedUnsafeList' struct
// Fields must match UntypedUnsafeList structure, please don't reorder and don't insert anything in between first 4 fields
///
/// The internal buffer of this list.
///
[NativeDisableUnsafePtrRestriction]
public T* Ptr;
///
/// The number of elements.
///
public int m_length;
///
/// The number of elements that can fit in the internal buffer.
///
public int m_capacity;
///
/// The allocator used to create the internal buffer.
///
public AllocatorManager.AllocatorHandle Allocator;
[Obsolete("Use Length property (UnityUpgradable) -> Length", true)]
public int length;
[Obsolete("Use Capacity property (UnityUpgradable) -> Capacity", true)]
public int capacity;
///
/// The number of elements.
///
/// The number of elements.
public int Length
{
get
{
return CollectionHelper.AssumePositive(m_length);
}
set
{
if (value > Capacity)
{
Resize(value);
}
else
{
m_length = value;
}
}
}
///
/// The number of elements that can fit in the internal buffer.
///
/// The number of elements that can fit in the internal buffer.
public int Capacity
{
get
{
return CollectionHelper.AssumePositive(m_capacity);
}
set
{
SetCapacity(value);
}
}
///
/// The element at an index.
///
/// An index.
/// The element at the index.
public T this[int index]
{
get
{
CollectionHelper.CheckIndexInRange(index, Length);
return Ptr[CollectionHelper.AssumePositive(index)];
}
set
{
CollectionHelper.CheckIndexInRange(index, Length);
Ptr[CollectionHelper.AssumePositive(index)] = value;
}
}
///
/// Returns a reference to the element at a given index.
///
/// The index to access. Must be in the range of [0..Length).
/// A reference to the element at the index.
public ref T ElementAt(int index)
{
CollectionHelper.CheckIndexInRange(index, Length);
return ref Ptr[CollectionHelper.AssumePositive(index)];
}
///
/// Initializes and returns an instance of UnsafeList.
///
/// An existing byte array to set as the internal buffer.
/// The length.
public UnsafeList(T* ptr, int length) : this()
{
Ptr = ptr;
this.m_length = length;
m_capacity = 0;
Allocator = AllocatorManager.None;
}
///
/// Initializes and returns an instance of UnsafeList.
///
/// The initial capacity of the list.
/// The allocator to use.
/// Whether newly allocated bytes should be zeroed out.
public UnsafeList(int initialCapacity, AllocatorManager.AllocatorHandle allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) : this()
{
Ptr = null;
m_length = 0;
m_capacity = 0;
Allocator = allocator;
if (initialCapacity != 0)
{
SetCapacity(initialCapacity);
}
if (options == NativeArrayOptions.ClearMemory && Ptr != null)
{
var sizeOf = sizeof(T);
UnsafeUtility.MemClear(Ptr, Capacity * sizeOf);
}
}
[BurstCompatible(GenericTypeArguments = new [] { typeof(AllocatorManager.AllocatorHandle) })]
internal void Initialize(int initialCapacity, ref U allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) where U : unmanaged, AllocatorManager.IAllocator
{
Ptr = null;
m_length = 0;
m_capacity = 0;
Allocator = AllocatorManager.None;
Initialize(initialCapacity, ref allocator, options);
}
[BurstCompatible(GenericTypeArguments = new [] { typeof(AllocatorManager.AllocatorHandle) })]
internal static UnsafeList New(int initialCapacity, ref U allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) where U : unmanaged, AllocatorManager.IAllocator
{
UnsafeList instance = default;
instance.Initialize(initialCapacity, ref allocator, options);
return instance;
}
[BurstCompatible(GenericTypeArguments = new[] { typeof(AllocatorManager.AllocatorHandle) })]
internal static UnsafeList* Create(int initialCapacity, ref U allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) where U : unmanaged, AllocatorManager.IAllocator
{
UnsafeList* listData = allocator.Allocate(default(UnsafeList), 1);
UnsafeUtility.MemClear(listData, sizeof(UnsafeList));
listData->Allocator = allocator.Handle;
if (initialCapacity != 0)
{
listData->SetCapacity(ref allocator, initialCapacity);
}
if (options == NativeArrayOptions.ClearMemory
&& listData->Ptr != null)
{
var sizeOf = sizeof(T);
UnsafeUtility.MemClear(listData->Ptr, listData->Capacity * sizeOf);
}
return listData;
}
[BurstCompatible(GenericTypeArguments = new[] { typeof(AllocatorManager.AllocatorHandle) })]
internal static void Destroy(UnsafeList* listData, ref U allocator) where U : unmanaged, AllocatorManager.IAllocator
{
CheckNull(listData);
listData->Dispose(ref allocator);
allocator.Free(listData, sizeof(UnsafeList), UnsafeUtility.AlignOf>(), 1);
}
///
/// Returns a new list.
///
/// The initial capacity of the list.
/// The allocator to use.
/// Whether newly allocated bytes should be zeroed out.
/// A pointer to the new list.
public static UnsafeList* Create(int initialCapacity, AllocatorManager.AllocatorHandle allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory)
{
UnsafeList* listData = AllocatorManager.Allocate>(allocator);
*listData = new UnsafeList(initialCapacity, allocator, options);
return listData;
}
///
/// Destroys the list.
///
/// The list to destroy.
public static void Destroy(UnsafeList* listData)
{
CheckNull(listData);
var allocator = listData->Allocator;
listData->Dispose();
AllocatorManager.Free(allocator, listData);
}
///
/// Whether the list is empty.
///
/// True if the list is empty or the list has not been constructed.
public bool IsEmpty => !IsCreated || m_length == 0;
///
/// Whether this list has been allocated (and not yet deallocated).
///
/// True if this list has been allocated (and not yet deallocated).
public bool IsCreated => Ptr != null;
[BurstCompatible(GenericTypeArguments = new[] { typeof(AllocatorManager.AllocatorHandle) })]
internal void Dispose(ref U allocator) where U : unmanaged, AllocatorManager.IAllocator
{
allocator.Free(Ptr, m_length);
Ptr = null;
m_length = 0;
m_capacity = 0;
}
///
/// Releases all resources (memory).
///
public void Dispose()
{
if (CollectionHelper.ShouldDeallocate(Allocator))
{
AllocatorManager.Free(Allocator, Ptr);
Allocator = AllocatorManager.Invalid;
}
Ptr = null;
m_length = 0;
m_capacity = 0;
}
///
/// Creates and schedules a job that frees the memory of this list.
///
/// The dependency for the new job.
/// The handle of the new job. The job depends upon `inputDeps` and frees the memory of this list.
[NotBurstCompatible /* This is not burst compatible because of IJob's use of a static IntPtr. Should switch to IJobBurstSchedulable in the future */]
public JobHandle Dispose(JobHandle inputDeps)
{
if (CollectionHelper.ShouldDeallocate(Allocator))
{
var jobHandle = new UnsafeDisposeJob { Ptr = Ptr, Allocator = Allocator }.Schedule(inputDeps);
Ptr = null;
Allocator = AllocatorManager.Invalid;
return jobHandle;
}
Ptr = null;
return inputDeps;
}
///
/// Sets the length to 0.
///
/// Does not change the capacity.
public void Clear()
{
m_length = 0;
}
///
/// Sets the length, expanding the capacity if necessary.
///
/// The new length.
/// Whether newly allocated bytes should be zeroed out.
public void Resize(int length, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory)
{
var oldLength = m_length;
if (length > Capacity)
{
SetCapacity(length);
}
m_length = length;
if (options == NativeArrayOptions.ClearMemory && oldLength < length)
{
var num = length - oldLength;
byte* ptr = (byte*)Ptr;
var sizeOf = sizeof(T);
UnsafeUtility.MemClear(ptr + oldLength * sizeOf, num * sizeOf);
}
}
void Realloc(ref U allocator, int newCapacity) where U : unmanaged, AllocatorManager.IAllocator
{
CollectionHelper.CheckAllocator(Allocator);
T* newPointer = null;
var alignOf = UnsafeUtility.AlignOf();
var sizeOf = sizeof(T);
if (newCapacity > 0)
{
newPointer = (T*)allocator.Allocate(sizeOf, alignOf, newCapacity);
if (m_capacity > 0)
{
var itemsToCopy = math.min(newCapacity, Capacity);
var bytesToCopy = itemsToCopy * sizeOf;
UnsafeUtility.MemCpy(newPointer, Ptr, bytesToCopy);
}
}
allocator.Free(Ptr, Capacity);
Ptr = newPointer;
m_capacity = newCapacity;
m_length = math.min(m_length, newCapacity);
}
void Realloc(int capacity)
{
Realloc(ref Allocator, capacity);
}
void SetCapacity(ref U allocator, int capacity) where U : unmanaged, AllocatorManager.IAllocator
{
CollectionHelper.CheckCapacityInRange(capacity, Length);
var sizeOf = sizeof(T);
var newCapacity = math.max(capacity, 64 / sizeOf);
newCapacity = math.ceilpow2(newCapacity);
if (newCapacity == Capacity)
{
return;
}
Realloc(ref allocator, newCapacity);
}
///
/// Sets the capacity.
///
/// The new capacity.
public void SetCapacity(int capacity)
{
SetCapacity(ref Allocator, capacity);
}
///
/// Sets the capacity to match the length.
///
public void TrimExcess()
{
if (Capacity != m_length)
{
Realloc(m_length);
}
}
///
/// Adds an element to the end of this list.
///
///
/// Increments the length by 1. Never increases the capacity.
///
/// The value to add to the end of the list.
/// Thrown if incrementing the length would exceed the capacity.
public void AddNoResize(T value)
{
CheckNoResizeHasEnoughCapacity(1);
UnsafeUtility.WriteArrayElement(Ptr, m_length, value);
m_length += 1;
}
///
/// Copies elements from a buffer to the end of this list.
///
///
/// Increments the length by `count`. Never increases the capacity.
///
/// The buffer to copy from.
/// The number of elements to copy from the buffer.
/// Thrown if the increased length would exceed the capacity.
public void AddRangeNoResize(void* ptr, int count)
{
CheckNoResizeHasEnoughCapacity(count);
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + m_length * sizeOf;
UnsafeUtility.MemCpy(dst, ptr, count * sizeOf);
m_length += count;
}
///
/// Copies the elements of another list to the end of this list.
///
/// The other list to copy from.
///
/// Increments the length by the length of the other list. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public void AddRangeNoResize(UnsafeList list)
{
AddRangeNoResize(list.Ptr, CollectionHelper.AssumePositive(list.m_length));
}
///
/// Adds an element to the end of the list.
///
/// The value to add to the end of this list.
///
/// Increments the length by 1. Increases the capacity if necessary.
///
public void Add(in T value)
{
var idx = m_length;
if (m_length + 1 > Capacity)
{
Resize(idx + 1);
}
else
{
m_length += 1;
}
UnsafeUtility.WriteArrayElement(Ptr, idx, value);
}
///
/// Copies the elements of a buffer to the end of this list.
///
/// The buffer to copy from.
/// The number of elements to copy from the buffer.
///
/// Increments the length by `count`. Increases the capacity if necessary.
///
public void AddRange(void* ptr, int count)
{
var idx = m_length;
if (m_length + count > Capacity)
{
Resize(m_length + count);
}
else
{
m_length += count;
}
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + idx * sizeOf;
UnsafeUtility.MemCpy(dst, ptr, count * sizeOf);
}
///
/// Copies the elements of another list to the end of the list.
///
/// The list to copy from.
///
/// The length is increased by the length of the other list. Increases the capacity if necessary.
///
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public void AddRange(UnsafeList list)
{
AddRange(list.Ptr, list.Length);
}
///
/// Shifts elements toward the end of this list, increasing its length.
///
///
/// Right-shifts elements in the list so as to create 'free' slots at the beginning or in the middle.
///
/// The length is increased by `end - begin`. If necessary, the capacity will be increased accordingly.
///
/// If `end` equals `begin`, the method does nothing.
///
/// The element at index `begin` will be copied to index `end`, the element at index `begin + 1` will be copied to `end + 1`, and so forth.
///
/// The indexes `begin` up to `end` are not cleared: they will contain whatever values they held prior.
///
/// The index of the first element that will be shifted up.
/// The index where the first shifted element will end up.
/// Thrown if `end < begin`.
/// Thrown if `begin` or `end` are out of bounds.
public void InsertRangeWithBeginEnd(int begin, int end)
{
CheckBeginEnd(begin, end);
int items = end - begin;
if (items < 1)
{
return;
}
var oldLength = m_length;
if (m_length + items > Capacity)
{
Resize(m_length + items);
}
else
{
m_length += items;
}
var itemsToCopy = oldLength - begin;
if (itemsToCopy < 1)
{
return;
}
var sizeOf = sizeof(T);
var bytesToCopy = itemsToCopy * sizeOf;
unsafe
{
byte* ptr = (byte*)Ptr;
byte* dest = ptr + end * sizeOf;
byte* src = ptr + begin * sizeOf;
UnsafeUtility.MemMove(dest, src, bytesToCopy);
}
}
///
/// Copies the last element of this list to the specified index. Decrements the length by 1.
///
/// Useful as a cheap way to remove an element from this list when you don't care about preserving order.
/// The index to overwrite with the last element.
/// Thrown if `index` is out of bounds.
public void RemoveAtSwapBack(int index)
{
RemoveRangeSwapBack(index, 1);
}
///
/// Copies the last *N* elements of this list to a range in this list. Decrements the length by *N*.
///
///
/// Copies the last `count` elements to the indexes `index` up to `index + count`.
///
/// Useful as a cheap way to remove elements from a list when you don't care about preserving order.
///
/// The index of the first element to overwrite.
/// The number of elements to copy and remove.
/// Thrown if `index` is out of bounds, `count` is negative,
/// or `index + count` exceeds the length.
public void RemoveRangeSwapBack(int index, int count)
{
CheckIndexCount(index, count);
if (count > 0)
{
int copyFrom = math.max(m_length - count, index + count);
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + index * sizeOf;
void* src = (byte*)Ptr + copyFrom * sizeOf;
UnsafeUtility.MemCpy(dst, src, (m_length - copyFrom) * sizeOf);
m_length -= count;
}
}
///
/// Copies the last *N* elements of this list to a range in this list. Decrements the length by *N*.
///
///
/// Copies the last `end - begin` elements to the indexes `begin` up to `end`.
///
/// Useful as a cheap way to remove elements from a list when you don't care about preserving order.
///
/// Does nothing if `end - begin` is less than 1.
///
/// The index of the first element to overwrite.
/// The index one greater than the last element to overwrite.
/// Thrown if `begin` or `end` are out of bounds.
[Obsolete("RemoveRangeSwapBackWithBeginEnd(begin, end) is deprecated, use RemoveRangeSwapBack(index, count) instead. (RemovedAfter 2021-06-02)", false)]
public void RemoveRangeSwapBackWithBeginEnd(int begin, int end)
{
CheckBeginEnd(begin, end);
int itemsToRemove = end - begin;
if (itemsToRemove > 0)
{
int copyFrom = math.max(m_length - itemsToRemove, end);
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + begin * sizeOf;
void* src = (byte*)Ptr + copyFrom * sizeOf;
UnsafeUtility.MemCpy(dst, src, (m_length - copyFrom) * sizeOf);
m_length -= itemsToRemove;
}
}
///
/// Removes the element at an index, shifting everything above it down by one. Decrements the length by 1.
///
/// The index of the element to remove.
///
/// If you don't care about preserving the order of the elements, is a more efficient way to remove elements.
///
/// Thrown if `index` is out of bounds.
public void RemoveAt(int index)
{
RemoveRange(index, 1);
}
///
/// Removes *N* elements in a range, shifting everything above the range down by *N*. Decrements the length by *N*.
///
/// The index of the first element to remove.
/// The number of elements to remove.
///
/// If you don't care about preserving the order of the elements,
/// is a more efficient way to remove elements.
///
/// Thrown if `index` is out of bounds, `count` is negative,
/// or `index + count` exceeds the length.
public void RemoveRange(int index, int count)
{
CheckIndexCount(index, count);
if (count > 0)
{
int copyFrom = math.min(index + count, m_length);
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + index * sizeOf;
void* src = (byte*)Ptr + copyFrom * sizeOf;
UnsafeUtility.MemCpy(dst, src, (m_length - copyFrom) * sizeOf);
m_length -= count;
}
}
///
/// Removes *N* elements in a range, shifting everything above it down by *N*. Decrements the length by *N*.
///
/// The index of the first element to remove.
/// The index one greater than the last element to remove.
///
/// If you don't care about preserving the order of the elements, is a more efficient way to remove elements.
///
/// Thrown if `end < begin`.
/// Thrown if `begin` or `end` are out of bounds.
[Obsolete("RemoveRangeWithBeginEnd(begin, end) is deprecated, use RemoveRange(index, count) instead. (RemovedAfter 2021-06-02)", false)]
public void RemoveRangeWithBeginEnd(int begin, int end)
{
CheckBeginEnd(begin, end);
int itemsToRemove = end - begin;
if (itemsToRemove > 0)
{
int copyFrom = math.min(begin + itemsToRemove, m_length);
var sizeOf = sizeof(T);
void* dst = (byte*)Ptr + begin * sizeOf;
void* src = (byte*)Ptr + copyFrom * sizeOf;
UnsafeUtility.MemCpy(dst, src, (m_length - copyFrom) * sizeOf);
m_length -= itemsToRemove;
}
}
///
/// Returns a parallel reader of this list.
///
/// A parallel reader of this list.
public ParallelReader AsParallelReader()
{
return new ParallelReader(Ptr, Length);
}
///
/// A parallel reader for an UnsafeList<T>.
///
///
/// Use to create a parallel reader for a list.
///
[BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
public unsafe struct ParallelReader
{
///
/// The internal buffer of the list.
///
[NativeDisableUnsafePtrRestriction]
public readonly T* Ptr;
///
/// The number of elements.
///
public readonly int Length;
internal ParallelReader(T* ptr, int length)
{
Ptr = ptr;
Length = length;
}
}
///
/// Returns a parallel writer of this list.
///
/// A parallel writer of this list.
public ParallelWriter AsParallelWriter()
{
return new ParallelWriter((UnsafeList*)UnsafeUtility.AddressOf(ref this));
}
///
/// A parallel writer for an UnsafeList<T>.
///
///
/// Use to create a parallel writer for a list.
///
[BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
public unsafe struct ParallelWriter
{
///
/// The data of the list.
///
public readonly void* Ptr => ListData->Ptr;
///
/// The UnsafeList to write to.
///
[NativeDisableUnsafePtrRestriction]
public UnsafeList* ListData;
internal unsafe ParallelWriter(UnsafeList* listData)
{
ListData = listData;
}
///
/// Adds an element to the end of the list.
///
/// The value to add to the end of the list.
///
/// Increments the length by 1. Never increases the capacity.
///
/// Thrown if incrementing the length would exceed the capacity.
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public void AddNoResize(T value)
{
var idx = Interlocked.Increment(ref ListData->m_length) - 1;
ListData->CheckNoResizeHasEnoughCapacity(idx, 1);
UnsafeUtility.WriteArrayElement(ListData->Ptr, idx, value);
}
///
/// Copies elements from a buffer to the end of the list.
///
/// The buffer to copy from.
/// The number of elements to copy from the buffer.
///
/// Increments the length by `count`. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public void AddRangeNoResize(void* ptr, int count)
{
var idx = Interlocked.Add(ref ListData->m_length, count) - count;
ListData->CheckNoResizeHasEnoughCapacity(idx, count);
void* dst = (byte*)ListData->Ptr + idx * sizeof(T);
UnsafeUtility.MemCpy(dst, ptr, count * sizeof(T));
}
///
/// Copies the elements of another list to the end of this list.
///
/// The other list to copy from.
///
/// Increments the length by the length of the other list. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public void AddRangeNoResize(UnsafeList list)
{
AddRangeNoResize(list.Ptr, list.Length);
}
}
///
/// Overwrites the elements of this list with the elements of an equal-length array.
///
/// An array to copy into this list.
public void CopyFrom(UnsafeList array)
{
Resize(array.Length);
UnsafeUtility.MemCpy(Ptr, array.Ptr, UnsafeUtility.SizeOf() * Length);
}
///
/// Returns an enumerator over the elements of the list.
///
/// An enumerator over the elements of the list.
public Enumerator GetEnumerator()
{
return new Enumerator { m_Ptr = Ptr, m_Length = Length, m_Index = -1 };
}
///
/// This method is not implemented. Use instead.
///
/// Throws NotImplementedException.
/// Method is not implemented.
IEnumerator IEnumerable.GetEnumerator()
{
throw new NotImplementedException();
}
///
/// This method is not implemented. Use instead.
///
/// Throws NotImplementedException.
/// Method is not implemented.
IEnumerator IEnumerable.GetEnumerator()
{
throw new NotImplementedException();
}
///
/// An enumerator over the elements of a list.
///
///
/// In an enumerator's initial state, is invalid.
/// The first call advances the enumerator to the first element of the list.
///
public struct Enumerator : IEnumerator
{
internal T* m_Ptr;
internal int m_Length;
internal int m_Index;
///
/// Does nothing.
///
public void Dispose() { }
///
/// Advances the enumerator to the next element of the list.
///
///
/// The first `MoveNext` call advances the enumerator to the first element of the list. Before this call, `Current` is not valid to read.
///
/// True if `Current` is valid to read after the call.
public bool MoveNext() => ++m_Index < m_Length;
///
/// Resets the enumerator to its initial state.
///
public void Reset() => m_Index = -1;
///
/// The current element.
///
/// The current element.
public T Current => m_Ptr[m_Index];
object IEnumerator.Current => Current;
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
internal static void CheckNull(void* listData)
{
if (listData == null)
{
throw new Exception("UnsafeList has yet to be created or has been destroyed!");
}
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckIndexCount(int index, int count)
{
if (count < 0)
{
throw new ArgumentOutOfRangeException($"Value for cound {count} must be positive.");
}
if (index < 0)
{
throw new ArgumentOutOfRangeException($"Value for index {index} must be positive.");
}
if (index > Length)
{
throw new ArgumentOutOfRangeException($"Value for index {index} is out of bounds.");
}
if (index+count > Length)
{
throw new ArgumentOutOfRangeException($"Value for count {count} is out of bounds.");
}
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckBeginEnd(int begin, int end)
{
if (begin > end)
{
throw new ArgumentException($"Value for begin {begin} index must less or equal to end {end}.");
}
if (begin < 0)
{
throw new ArgumentOutOfRangeException($"Value for begin {begin} must be positive.");
}
if (begin > Length)
{
throw new ArgumentOutOfRangeException($"Value for begin {begin} is out of bounds.");
}
if (end > Length)
{
throw new ArgumentOutOfRangeException($"Value for end {end} is out of bounds.");
}
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckNoResizeHasEnoughCapacity(int length)
{
CheckNoResizeHasEnoughCapacity(length, Length);
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckNoResizeHasEnoughCapacity(int length, int index)
{
if (Capacity < index + length)
{
throw new Exception($"AddNoResize assumes that list capacity is sufficient (Capacity {Capacity}, Length {Length}), requested length {length}!");
}
}
}
///
/// Provides extension methods for UnsafeList.
///
[BurstCompatible]
public unsafe static class UnsafeListExtensions
{
///
/// Finds the index of the first occurrence of a particular value in this list.
///
/// The type of elements in this list.
/// The type of value to locate.
/// This list.
/// A value to locate.
/// The zero-based index of the first occurrence of the value if it is found. Returns -1 if no occurrence is found.
[BurstCompatible(GenericTypeArguments = new [] { typeof(int), typeof(int) })]
public static int IndexOf(this UnsafeList list, U value) where T : unmanaged, IEquatable
{
return NativeArrayExtensions.IndexOf(list.Ptr, list.Length, value);
}
///
/// Returns true if a particular value is present in this list.
///
/// The type of elements in the list.
/// The type of value to locate.
/// This list.
/// The value to locate.
/// True if the value is present in this list.
[BurstCompatible(GenericTypeArguments = new [] { typeof(int), typeof(int) })]
public static bool Contains(this UnsafeList list, U value) where T : unmanaged, IEquatable
{
return list.IndexOf(value) != -1;
}
///
/// Finds the index of the first occurrence of a particular value in the list.
///
/// The type of elements in the list.
/// The type of value to locate.
/// This reader of the list.
/// A value to locate.
/// The zero-based index of the first occurrence of the value if it is found. Returns -1 if no occurrence is found.
[BurstCompatible(GenericTypeArguments = new [] { typeof(int), typeof(int) })]
public static int IndexOf(this UnsafeList.ParallelReader list, U value) where T : unmanaged, IEquatable
{
return NativeArrayExtensions.IndexOf(list.Ptr, list.Length, value);
}
///
/// Returns true if a particular value is present in the list.
///
/// The type of elements in the list.
/// The type of value to locate.
/// This reader of the list.
/// The value to locate.
/// True if the value is present in the list.
[BurstCompatible(GenericTypeArguments = new [] { typeof(int), typeof(int) })]
public static bool Contains(this UnsafeList.ParallelReader list, U value) where T : unmanaged, IEquatable
{
return list.IndexOf(value) != -1;
}
///
/// Returns true if this array and another have equal length and content.
///
/// The type of the source array's elements.
/// The array to compare for equality.
/// The other array to compare for equality.
/// True if the arrays have equal length and content.
[BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
public static bool ArraysEqual(this UnsafeList array, UnsafeList other) where T : unmanaged, IEquatable
{
if (array.Length != other.Length)
return false;
for (int i = 0; i != array.Length; i++)
{
if (!array[i].Equals(other[i]))
return false;
}
return true;
}
}
internal sealed class UnsafeListTDebugView
where T : unmanaged
{
UnsafeList Data;
public UnsafeListTDebugView(UnsafeList data)
{
Data = data;
}
public unsafe T[] Items
{
get
{
T[] result = new T[Data.Length];
for (var i = 0; i < result.Length; ++i)
{
result[i] = Data.Ptr[i];
}
return result;
}
}
}
///
/// An unmanaged, resizable list of pointers.
///
/// The type of pointer element.
[DebuggerDisplay("Length = {Length}, Capacity = {Capacity}, IsCreated = {IsCreated}, IsEmpty = {IsEmpty}")]
[DebuggerTypeProxy(typeof(UnsafePtrListTDebugView<>))]
[StructLayout(LayoutKind.Sequential)]
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public unsafe struct UnsafePtrList
: INativeDisposable
// IIndexable and INativeList can't be implemented because this[index] and ElementAt return T* instead of T.
, IEnumerable // Used by collection initializers.
where T : unmanaged
{
///
/// The internal buffer of this list.
///
[NativeDisableUnsafePtrRestriction]
public readonly T** Ptr;
///
/// The number of elements.
///
public readonly int m_length;
///
/// The number of elements that can fit in the internal buffer.
///
public readonly int m_capacity;
///
/// The allocator used to create the internal buffer.
///
public readonly AllocatorManager.AllocatorHandle Allocator;
[Obsolete("Use Length property (UnityUpgradable) -> Length", true)]
public int length;
[Obsolete("Use Capacity property (UnityUpgradable) -> Capacity", true)]
public int capacity;
///
/// The number of elements.
///
/// The number of elements.
public int Length
{
get
{
return this.ListData().Length;
}
set
{
this.ListData().Length = value;
}
}
///
/// The number of elements that can fit in the internal buffer.
///
/// The number of elements that can fit in the internal buffer.
public int Capacity
{
get
{
return this.ListData().Capacity;
}
set
{
this.ListData().Capacity = value;
}
}
///
/// The element at an index.
///
/// An index.
/// The element at the index.
public T* this[int index]
{
get
{
CollectionHelper.CheckIndexInRange(index, Length);
return Ptr[CollectionHelper.AssumePositive(index)];
}
set
{
CollectionHelper.CheckIndexInRange(index, Length);
Ptr[CollectionHelper.AssumePositive(index)] = value;
}
}
///
/// Returns a reference to the element at a given index.
///
/// The index to access. Must be in the range of [0..Length).
/// A reference to the element at the index.
public ref T* ElementAt(int index)
{
CollectionHelper.CheckIndexInRange(index, Length);
return ref Ptr[CollectionHelper.AssumePositive(index)];
}
///
/// Initializes and returns an instance of UnsafePtrList.
///
/// An existing pointer array to set as the internal buffer.
/// The length.
public unsafe UnsafePtrList(T** ptr, int length) : this()
{
Ptr = ptr;
this.m_length = length;
this.m_capacity = length;
Allocator = AllocatorManager.None;
}
///
/// Initializes and returns an instance of UnsafePtrList.
///
/// The initial capacity of the list.
/// The allocator to use.
/// Whether newly allocated bytes should be zeroed out.
public unsafe UnsafePtrList(int initialCapacity, AllocatorManager.AllocatorHandle allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) : this()
{
Ptr = null;
m_length = 0;
m_capacity = 0;
Allocator = AllocatorManager.None;
this.ListData() = new UnsafeList(initialCapacity, allocator, options);
}
///
/// Returns a new list of pointers.
///
/// An existing pointer array to set as the internal buffer.
/// The length.
/// A pointer to the new list.
public static UnsafePtrList* Create(T** ptr, int length)
{
UnsafePtrList* listData = AllocatorManager.Allocate>(AllocatorManager.Persistent);
*listData = new UnsafePtrList(ptr, length);
return listData;
}
///
/// Returns a new list of pointers.
///
/// The initial capacity of the list.
/// The allocator to use.
/// Whether newly allocated bytes should be zeroed out.
/// A pointer to the new list.
public static UnsafePtrList* Create(int initialCapacity, AllocatorManager.AllocatorHandle allocator, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory)
{
UnsafePtrList* listData = AllocatorManager.Allocate>(allocator);
*listData = new UnsafePtrList(initialCapacity, allocator, options);
return listData;
}
///
/// Destroys the list.
///
/// The list to destroy.
public static void Destroy(UnsafePtrList* listData)
{
UnsafeList.CheckNull(listData);
var allocator = listData->ListData().Allocator.Value == AllocatorManager.Invalid.Value
? AllocatorManager.Persistent
: listData->ListData().Allocator
;
listData->Dispose();
AllocatorManager.Free(allocator, listData);
}
///
/// Whether the list is empty.
///
/// True if the list is empty or the list has not been constructed.
public bool IsEmpty => !IsCreated || Length == 0;
///
/// Whether this list has been allocated (and not yet deallocated).
///
/// True if this list has been allocated (and not yet deallocated).
public bool IsCreated => Ptr != null;
///
/// Releases all resources (memory).
///
public void Dispose()
{
this.ListData().Dispose();
}
///
/// Creates and schedules a job that frees the memory of this list.
///
/// The dependency for the new job.
/// The handle of the new job. The job depends upon `inputDeps` and frees the memory of this list.
[NotBurstCompatible /* This is not burst compatible because of IJob's use of a static IntPtr. Should switch to IJobBurstSchedulable in the future */]
public JobHandle Dispose(JobHandle inputDeps) => this.ListData().Dispose(inputDeps);
///
/// Sets the length to 0.
///
/// Does not change the capacity.
public void Clear() => this.ListData().Clear();
///
/// Sets the length, expanding the capacity if necessary.
///
/// The new length.
/// Whether newly allocated bytes should be zeroed out.
public void Resize(int length, NativeArrayOptions options = NativeArrayOptions.UninitializedMemory) => this.ListData().Resize(length, options);
///
/// Sets the capacity.
///
/// The new capacity.
public void SetCapacity(int capacity) => this.ListData().SetCapacity(capacity);
///
/// Sets the capacity to match the length.
///
public void TrimExcess() => this.ListData().TrimExcess();
///
/// Returns the index of the first occurrence of a specific pointer in the list.
///
/// The pointer to search for in the list.
/// The index of the first occurrence of the pointer. Returns -1 if it is not found in the list.
public int IndexOf(void* ptr)
{
for (int i = 0; i < Length; ++i)
{
if (Ptr[i] == ptr) return i;
}
return -1;
}
///
/// Returns true if the list contains at least one occurrence of a specific pointer.
///
/// The pointer to search for in the list.
/// True if the list contains at least one occurrence of the pointer.
public bool Contains(void* ptr)
{
return IndexOf(ptr) != -1;
}
///
/// Adds a pointer to the end of this list.
///
///
/// Increments the length by 1. Never increases the capacity.
///
/// The pointer to add to the end of the list.
/// Thrown if incrementing the length would exceed the capacity.
public void AddNoResize(void* value)
{
this.ListData().AddNoResize((IntPtr)value);
}
///
/// Copies pointers from a buffer to the end of this list.
///
///
/// Increments the length by `count`. Never increases the capacity.
///
/// The buffer to copy from.
/// The number of pointers to copy from the buffer.
/// Thrown if the increased length would exceed the capacity.
public void AddRangeNoResize(void** ptr, int count) => this.ListData().AddRangeNoResize(ptr, count);
///
/// Copies the pointers of another list to the end of this list.
///
/// The other list to copy from.
///
/// Increments the length by the length of the other list. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
public void AddRangeNoResize(UnsafePtrList list) => this.ListData().AddRangeNoResize(list.Ptr, list.Length);
///
/// Adds a pointer to the end of the list.
///
/// The pointer to add to the end of this list.
///
/// Increments the length by 1. Increases the capacity if necessary.
///
public void Add(in IntPtr value)
{
this.ListData().Add(value);
}
///
/// Adds a pointer to the end of the list.
///
/// The pointer to add to the end of this list.
///
/// Increments the length by 1. Increases the capacity if necessary.
///
public void Add(void* value)
{
this.ListData().Add((IntPtr)value);
}
///
/// Adds elements from a buffer to this list.
///
/// A pointer to the buffer.
/// The number of elements to add to the list.
public void AddRange(void* ptr, int length) => this.ListData().AddRange(ptr, length);
///
/// Copies the elements of another list to the end of this list.
///
/// The other list to copy from.
///
/// Increments the length by the length of the other list. Increases the capacity if necessary.
///
public void AddRange(UnsafePtrList list) => this.ListData().AddRange(list.ListData());
///
/// Shifts pointers toward the end of this list, increasing its length.
///
///
/// Right-shifts pointers in the list so as to create 'free' slots at the beginning or in the middle.
///
/// The length is increased by `end - begin`. If necessary, the capacity will be increased accordingly.
///
/// If `end` equals `begin`, the method does nothing.
///
/// The pointer at index `begin` will be copied to index `end`, the pointer at index `begin + 1` will be copied to `end + 1`, and so forth.
///
/// The indexes `begin` up to `end` are not cleared: they will contain whatever pointers they held prior.
///
/// The index of the first pointer that will be shifted up.
/// The index where the first shifted pointer will end up.
/// Thrown if `end < begin`.
/// Thrown if `begin` or `end` are out of bounds.
public void InsertRangeWithBeginEnd(int begin, int end) => this.ListData().InsertRangeWithBeginEnd(begin, end);
///
/// Copies the last pointer of this list to the specified index. Decrements the length by 1.
///
/// Useful as a cheap way to remove a pointer from this list when you don't care about preserving order.
/// The index to overwrite with the last pointer.
/// Thrown if `index` is out of bounds.
public void RemoveAtSwapBack(int index) => this.ListData().RemoveAtSwapBack(index);
///
/// Copies the last *N* pointer of this list to a range in this list. Decrements the length by *N*.
///
///
/// Copies the last `count` pointers to the indexes `index` up to `index + count`.
///
/// Useful as a cheap way to remove pointers from a list when you don't care about preserving order.
///
/// The index of the first pointer to overwrite.
/// The number of pointers to copy and remove.
/// Thrown if `index` is out of bounds, `count` is negative,
/// or `index + count` exceeds the length.
public void RemoveRangeSwapBack(int index, int count) => this.ListData().RemoveRangeSwapBack(index, count);
///
/// Copies the last *N* pointers of this list to a range in this list. Decrements the length by *N*.
///
///
/// Copies the last `end - begin` pointers to the indexes `begin` up to `end`.
///
/// Useful as a cheap way to remove pointers from a list when you don't care about preserving order.
///
/// Does nothing if `end - begin` is less than 1.
///
/// The index of the first pointers to overwrite.
/// The index one greater than the last pointers to overwrite.
/// Thrown if `begin` or `end` are out of bounds.
[Obsolete("RemoveRangeSwapBackWithBeginEnd(begin, end) is deprecated, use RemoveRangeSwapBack(index, count) instead. (RemovedAfter 2021-06-02)", false)]
public void RemoveRangeSwapBackWithBeginEnd(int begin, int end) => this.ListData().RemoveRangeSwapBackWithBeginEnd(begin, end);
///
/// Removes the pointer at an index, shifting everything above it down by one. Decrements the length by 1.
///
/// The index of the pointer to remove.
///
/// If you don't care about preserving the order of the pointers, is a more efficient way to remove pointers.
///
/// Thrown if `index` is out of bounds.
public void RemoveAt(int index) => this.ListData().RemoveAt(index);
///
/// Removes *N* pointers in a range, shifting everything above the range down by *N*. Decrements the length by *N*.
///
/// The index of the first pointer to remove.
/// The number of pointers to remove.
///
/// If you don't care about preserving the order of the pointers,
/// is a more efficient way to remove pointers.
///
/// Thrown if `index` is out of bounds, `count` is negative,
/// or `index + count` exceeds the length.
public void RemoveRange(int index, int count) => this.ListData().RemoveRange(index, count);
///
/// Removes *N* pointers in a range, shifting everything above it down by *N*. Decrements the length by *N*.
///
/// The index of the first pointer to remove.
/// The index one greater than the last pointer to remove.
///
/// If you don't care about preserving the order of the pointers, is a more efficient way to remove pointers.
///
/// Thrown if `end < begin`.
/// Thrown if `begin` or `end` are out of bounds.
[Obsolete("RemoveRangeWithBeginEnd(begin, end) is deprecated, use RemoveRange(index, count) instead. (RemovedAfter 2021-06-02)", false)]
public void RemoveRangeWithBeginEnd(int begin, int end) => this.ListData().RemoveRangeWithBeginEnd(begin, end);
///
/// This method is not implemented. It will throw NotImplementedException if it is used.
///
/// Use Enumerator GetEnumerator() instead.
/// Throws NotImplementedException.
/// Method is not implemented.
IEnumerator IEnumerable.GetEnumerator()
{
throw new NotImplementedException();
}
///
/// This method is not implemented. It will throw NotImplementedException if it is used.
///
/// Use Enumerator GetEnumerator() instead.
/// Throws NotImplementedException.
/// Method is not implemented.
IEnumerator IEnumerable.GetEnumerator()
{
throw new NotImplementedException();
}
///
/// Returns a parallel reader of this list.
///
/// A parallel reader of this list.
public ParallelReader AsParallelReader()
{
return new ParallelReader(Ptr, Length);
}
///
/// A parallel reader for an UnsafePtrList<T>.
///
///
/// Use to create a parallel reader for a list.
///
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public unsafe struct ParallelReader
{
///
/// The internal buffer of the list.
///
[NativeDisableUnsafePtrRestriction]
public readonly T** Ptr;
///
/// The number of elements.
///
public readonly int Length;
internal ParallelReader(T** ptr, int length)
{
Ptr = ptr;
Length = length;
}
///
/// Returns the index of the first occurrence of a specific pointer in the list.
///
/// The pointer to search for in the list.
/// The index of the first occurrence of the pointer. Returns -1 if it is not found in the list.
public int IndexOf(void* ptr)
{
for (int i = 0; i < Length; ++i)
{
if (Ptr[i] == ptr) return i;
}
return -1;
}
///
/// Returns true if the list contains at least one occurrence of a specific pointer.
///
/// The pointer to search for in the list.
/// True if the list contains at least one occurrence of the pointer.
public bool Contains(void* ptr)
{
return IndexOf(ptr) != -1;
}
}
///
/// Returns a parallel writer of this list.
///
/// A parallel writer of this list.
public ParallelWriter AsParallelWriter()
{
return new ParallelWriter(Ptr, (UnsafeList*)UnsafeUtility.AddressOf(ref this));
}
///
/// A parallel writer for an UnsafePtrList<T>.
///
///
/// Use to create a parallel writer for a list.
///
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public unsafe struct ParallelWriter
{
///
/// The data of the list.
///
[NativeDisableUnsafePtrRestriction]
public readonly T** Ptr;
///
/// The UnsafeList to write to.
///
[NativeDisableUnsafePtrRestriction]
public UnsafeList* ListData;
internal unsafe ParallelWriter(T** ptr, UnsafeList* listData)
{
Ptr = ptr;
ListData = listData;
}
///
/// Adds a pointer to the end of the list.
///
/// The pointer to add to the end of the list.
///
/// Increments the length by 1. Never increases the capacity.
///
/// Thrown if incrementing the length would exceed the capacity.
public void AddNoResize(T* value) => ListData->AddNoResize((IntPtr)value);
///
/// Copies pointers from a buffer to the end of the list.
///
/// The buffer to copy from.
/// The number of pointers to copy from the buffer.
///
/// Increments the length by `count`. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
public void AddRangeNoResize(T** ptr, int count) => ListData->AddRangeNoResize(ptr, count);
///
/// Copies the pointers of another list to the end of this list.
///
/// The other list to copy from.
///
/// Increments the length by the length of the other list. Never increases the capacity.
///
/// Thrown if the increased length would exceed the capacity.
public void AddRangeNoResize(UnsafePtrList list) => ListData->AddRangeNoResize(list.Ptr, list.Length);
}
}
[BurstCompatible]
internal static class UnsafePtrListTExtensions
{
[BurstCompatible(GenericTypeArguments = new[] { typeof(int) })]
public static ref UnsafeList ListData(ref this UnsafePtrList from) where T : unmanaged => ref UnsafeUtility.As, UnsafeList>(ref from);
}
internal sealed class UnsafePtrListTDebugView
where T : unmanaged
{
UnsafePtrList Data;
public UnsafePtrListTDebugView(UnsafePtrList data)
{
Data = data;
}
public unsafe T*[] Items
{
get
{
T*[] result = new T*[Data.Length];
for (var i = 0; i < result.Length; ++i)
{
result[i] = Data.Ptr[i];
}
return result;
}
}
}
}