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; } } } }