1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135 |
- using System;
- using System.Collections;
- using System.Collections.Generic;
- using System.Diagnostics;
- using System.Runtime.InteropServices;
- using System.Threading;
- using Unity.Burst;
- using Unity.Collections.LowLevel.Unsafe;
- using Unity.Jobs;
-
- namespace Unity.Collections
- {
- /// <summary>
- /// An indexable collection.
- /// </summary>
- /// <typeparam name="T">The type of the elements in the collection.</typeparam>
- public interface IIndexable<T> where T : struct
- {
- /// <summary>
- /// The current number of elements in the collection.
- /// </summary>
- /// <value>The current number of elements in the collection.</value>
- int Length { get; set; }
-
- /// <summary>
- /// Returns a reference to the element at a given index.
- /// </summary>
- /// <param name="index">The index to access. Must be in the range of [0..Length).</param>
- /// <returns>A reference to the element at the index.</returns>
- ref T ElementAt(int index);
- }
-
- /// <summary>
- /// A resizable list.
- /// </summary>
- /// <typeparam name="T">The type of the elements.</typeparam>
- public interface INativeList<T> : IIndexable<T> where T : struct
- {
- /// <summary>
- /// The number of elements that fit in the current allocation.
- /// </summary>
- /// <value>The number of elements that fit in the current allocation.</value>
- /// <param name="value">A new capacity.</param>
- int Capacity { get; set; }
-
- /// <summary>
- /// Whether this list is empty.
- /// </summary>
- /// <value>True if this list is empty.</value>
- bool IsEmpty { get; }
-
- /// <summary>
- /// The element at an index.
- /// </summary>
- /// <param name="index">An index.</param>
- /// <value>The element at the index.</value>
- /// <exception cref="IndexOutOfRangeException">Thrown if index is out of bounds.</exception>
- T this[int index] { get; set; }
-
- /// <summary>
- /// Sets the length to 0.
- /// </summary>
- /// <remarks>Does not change the capacity.</remarks>
- void Clear();
- }
-
- /// <summary>
- /// An unmanaged, resizable list.
- /// </summary>
- /// <remarks>The elements are stored contiguously in a buffer rather than as linked nodes.</remarks>
- /// <typeparam name="T">The type of the elements.</typeparam>
- [StructLayout(LayoutKind.Sequential)]
- [NativeContainer]
- [DebuggerDisplay("Length = {Length}")]
- [DebuggerTypeProxy(typeof(NativeListDebugView<>))]
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public unsafe struct NativeList<T>
- : INativeDisposable
- , INativeList<T>
- , IEnumerable<T> // Used by collection initializers.
- where T : unmanaged
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- internal AtomicSafetyHandle m_Safety;
- internal int m_SafetyIndexHint;
- internal static readonly SharedStatic<int> s_staticSafetyId = SharedStatic<int>.GetOrCreate<NativeList<T>>();
-
- #if REMOVE_DISPOSE_SENTINEL
- #else
- [NativeSetClassTypeToNullOnSchedule]
- internal DisposeSentinel m_DisposeSentinel;
- #endif
- #endif
- [NativeDisableUnsafePtrRestriction]
- internal UnsafeList<T>* m_ListData;
-
- //@TODO: Unity.Physics currently relies on the specific layout of NativeList in order to
- // workaround a bug in 19.1 & 19.2 with atomic safety handle in jobified Dispose.
- internal AllocatorManager.AllocatorHandle m_DeprecatedAllocator;
-
- /// <summary>
- /// Initializes and returns a NativeList with a capacity of one.
- /// </summary>
- /// <param name="allocator">The allocator to use.</param>
- public NativeList(AllocatorManager.AllocatorHandle allocator)
- : this(1, allocator, 2)
- {
- }
-
- /// <summary>
- /// Initializes and returns a NativeList.
- /// </summary>
- /// <param name="initialCapacity">The initial capacity of the list.</param>
- /// <param name="allocator">The allocator to use.</param>
- public NativeList(int initialCapacity, AllocatorManager.AllocatorHandle allocator)
- : this(initialCapacity, allocator, 2)
- {
- }
-
- [BurstCompatible(GenericTypeArguments = new [] { typeof(AllocatorManager.AllocatorHandle) })]
- internal void Initialize<U>(int initialCapacity, ref U allocator, int disposeSentinelStackDepth) where U : unmanaged, AllocatorManager.IAllocator
- {
- var totalSize = sizeof(T) * (long)initialCapacity;
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- CollectionHelper.CheckAllocator(allocator.Handle);
- CheckInitialCapacity(initialCapacity);
- CollectionHelper.CheckIsUnmanaged<T>();
- CheckTotalSize(initialCapacity, totalSize);
-
- #if REMOVE_DISPOSE_SENTINEL
- m_Safety = CollectionHelper.CreateSafetyHandle(allocator.Handle);
- #else
- if (allocator.IsCustomAllocator)
- {
- m_Safety = AtomicSafetyHandle.Create();
- m_DisposeSentinel = null;
- }
- else
- {
- DisposeSentinel.Create(out m_Safety, out m_DisposeSentinel, disposeSentinelStackDepth, allocator.ToAllocator);
- }
- #endif
-
- CollectionHelper.SetStaticSafetyId<NativeList<T>>(ref m_Safety, ref s_staticSafetyId.Data);
-
- m_SafetyIndexHint = (allocator.Handle).AddSafetyHandle(m_Safety);
- #endif
- m_ListData = UnsafeList<T>.Create(initialCapacity, ref allocator);
- m_DeprecatedAllocator = allocator.Handle;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.SetBumpSecondaryVersionOnScheduleWrite(m_Safety, true);
- #endif
- }
-
- [BurstCompatible(GenericTypeArguments = new [] { typeof(AllocatorManager.AllocatorHandle) })]
- internal static NativeList<T> New<U>(int initialCapacity, ref U allocator, int disposeSentinelStackDepth) where U : unmanaged, AllocatorManager.IAllocator
- {
- var nativelist = new NativeList<T>();
- nativelist.Initialize(initialCapacity, ref allocator, disposeSentinelStackDepth);
- return nativelist;
- }
-
- [BurstCompatible(GenericTypeArguments = new [] { typeof(AllocatorManager.AllocatorHandle) })]
- internal static NativeList<T> New<U>(int initialCapacity, ref U allocator) where U : unmanaged, AllocatorManager.IAllocator
- {
- return New(initialCapacity, ref allocator, 2);
- }
-
- NativeList(int initialCapacity, AllocatorManager.AllocatorHandle allocator, int disposeSentinelStackDepth)
- {
- this = default;
- AllocatorManager.AllocatorHandle temp = allocator;
- Initialize(initialCapacity, ref temp, disposeSentinelStackDepth);
- }
-
- /// <summary>
- /// The element at a given index.
- /// </summary>
- /// <param name="index">An index into this list.</param>
- /// <value>The value to store at the `index`.</value>
- /// <exception cref="IndexOutOfRangeException">Thrown if `index` is out of bounds.</exception>
- public T this[int index]
- {
- get
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckReadAndThrow(m_Safety);
- #endif
- return (*m_ListData)[index];
- }
- set
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- (*m_ListData)[index] = value;
- }
- }
-
- /// <summary>
- /// Returns a reference to the element at an index.
- /// </summary>
- /// <param name="index">An index.</param>
- /// <returns>A reference to the element at the index.</returns>
- /// <exception cref="IndexOutOfRangeException">Thrown if index is out of bounds.</exception>
- public ref T ElementAt(int index)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- return ref m_ListData->ElementAt(index);
- }
-
- /// <summary>
- /// The count of elements.
- /// </summary>
- /// <value>The current count of elements. Always less than or equal to the capacity.</value>
- /// <remarks>To decrease the memory used by a list, set <see cref="Capacity"/> after reducing the length of the list.</remarks>
- /// <param name="value>">The new length. If the new length is greater than the current capacity, the capacity is increased.
- /// Newly allocated memory is cleared.</param>
- public int Length
- {
- get
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckReadAndThrow(m_Safety);
- #endif
- return CollectionHelper.AssumePositive(m_ListData->Length);
- }
-
- set
- {
- m_ListData->Resize(value, NativeArrayOptions.ClearMemory);
- }
- }
-
- /// <summary>
- /// The number of elements that fit in the current allocation.
- /// </summary>
- /// <value>The number of elements that fit in the current allocation.</value>
- /// <param name="value">The new capacity. Must be greater or equal to the length.</param>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if the new capacity is smaller than the length.</exception>
- public int Capacity
- {
- get
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckReadAndThrow(m_Safety);
- #endif
- return m_ListData->Capacity;
- }
-
- set
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->Capacity = value;
- }
- }
-
- /// <summary>
- /// Returns the internal unsafe list.
- /// </summary>
- /// <remarks>Internally, the elements of a NativeList are stored in an UnsafeList.</remarks>
- /// <returns>The internal unsafe list.</returns>
- public UnsafeList<T>* GetUnsafeList() => m_ListData;
-
- /// <summary>
- /// Appends an element to the end of this list.
- /// </summary>
- /// <param name="value">The value to add to the end of this list.</param>
- /// <remarks>
- /// Length is incremented by 1. Will not increase the capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if incrementing the length would exceed the capacity.</exception>
- public void AddNoResize(T value)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- m_ListData->AddNoResize(value);
- }
-
- /// <summary>
- /// Appends elements from a buffer to the end of this list.
- /// </summary>
- /// <param name="ptr">The buffer to copy from.</param>
- /// <param name="count">The number of elements to copy from the buffer.</param>
- /// <remarks>
- /// Length is increased by the count. Will not increase the capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if the increased length would exceed the capacity.</exception>
- public void AddRangeNoResize(void* ptr, int count)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- CheckArgPositive(count);
- m_ListData->AddRangeNoResize(ptr, count);
- }
-
- /// <summary>
- /// Appends the elements of another list to the end of this list.
- /// </summary>
- /// <param name="list">The other list to copy from.</param>
- /// <remarks>
- /// Length is increased by the length of the other list. Will not increase the capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if the increased length would exceed the capacity.</exception>
- public void AddRangeNoResize(NativeList<T> list)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- m_ListData->AddRangeNoResize(*list.m_ListData);
- }
-
- /// <summary>
- /// Appends an element to the end of this list.
- /// </summary>
- /// <param name="value">The value to add to the end of this list.</param>
- /// <remarks>
- /// Length is incremented by 1. If necessary, the capacity is increased.
- /// </remarks>
- public void Add(in T value)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->Add(value);
- }
-
- /// <summary>
- /// Appends the elements of an array to the end of this list.
- /// </summary>
- /// <param name="array">The array to copy from.</param>
- /// <remarks>
- /// Length is increased by the number of new elements. Does not increase the capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if the increased length would exceed the capacity.</exception>
- public void AddRange(NativeArray<T> array)
- {
- AddRange(array.GetUnsafeReadOnlyPtr(), array.Length);
- }
-
- /// <summary>
- /// Appends the elements of a buffer to the end of this list.
- /// </summary>
- /// <param name="ptr">The buffer to copy from.</param>
- /// <param name="count">The number of elements to copy from the buffer.</param>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if count is negative.</exception>
- public void AddRange(void* ptr, int count)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- CheckArgPositive(count);
- m_ListData->AddRange(ptr, CollectionHelper.AssumePositive(count));
- }
-
- /// <summary>
- /// Shifts elements toward the end of this list, increasing its length.
- /// </summary>
- /// <remarks>
- /// 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.
- /// </remarks>
- /// <param name="begin">The index of the first element that will be shifted up.</param>
- /// <param name="end">The index where the first shifted element will end up.</param>
- /// <exception cref="ArgumentException">Thrown if `end < begin`.</exception>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `begin` or `end` are out of bounds.</exception>
- public void InsertRangeWithBeginEnd(int begin, int end)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->InsertRangeWithBeginEnd(CollectionHelper.AssumePositive(begin), CollectionHelper.AssumePositive(end));
- }
-
- /// <summary>
- /// Copies the last element of this list to the specified index. Decrements the length by 1.
- /// </summary>
- /// <remarks>Useful as a cheap way to remove an element from this list when you don't care about preserving order.</remarks>
- /// <param name="index">The index to overwrite with the last element.</param>
- /// <exception cref="IndexOutOfRangeException">Thrown if `index` is out of bounds.</exception>
- public void RemoveAtSwapBack(int index)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveAtSwapBack(CollectionHelper.AssumePositive(index));
- }
-
- /// <summary>
- /// Copies the last *N* elements of this list to a range in this list. Decrements the length by *N*.
- /// </summary>
- /// <remarks>
- /// 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.
- /// </remarks>
- /// <param name="index">The index of the first element to overwrite.</param>
- /// <param name="count">The number of elements to copy and remove.</param>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `index` is out of bounds, `count` is negative,
- /// or `index + count` exceeds the length.</exception>
- public void RemoveRangeSwapBack(int index, int count)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveRangeSwapBack(CollectionHelper.AssumePositive(index), CollectionHelper.AssumePositive(count));
- }
-
- /// <summary>
- /// Copies the last *N* elements of this list to a range in this list. Decrements the length by *N*.
- /// </summary>
- /// <remarks>
- /// 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.
- /// </remarks>
- /// <param name="begin">The index of the first element to overwrite.</param>
- /// <param name="end">The index one greater than the last element to overwrite.</param>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `begin` or `end` are out of bounds.</exception>
- [Obsolete("RemoveRangeSwapBackWithBeginEnd(begin, end) is deprecated, use RemoveRangeSwapBack(index, count) instead. (RemovedAfter 2021-06-02)", false)]
- public void RemoveRangeSwapBackWithBeginEnd(int begin, int end)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveRangeSwapBackWithBeginEnd(CollectionHelper.AssumePositive(begin), CollectionHelper.AssumePositive(end));
- }
-
- /// <summary>
- /// Removes the element at an index, shifting everything above it down by one. Decrements the length by 1.
- /// </summary>
- /// <param name="index">The index of the item to remove.</param>
- /// <remarks>
- /// If you don't care about preserving the order of the elements, <see cref="RemoveAtSwapBack(int)"/> is a more efficient way to remove elements.
- /// </remarks>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `index` is out of bounds.</exception>
- public void RemoveAt(int index)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveAt(CollectionHelper.AssumePositive(index));
- }
-
- /// <summary>
- /// Removes *N* elements in a range, shifting everything above the range down by *N*. Decrements the length by *N*.
- /// </summary>
- /// <param name="index">The index of the first element to remove.</param>
- /// <param name="count">The number of elements to remove.</param>
- /// <remarks>
- /// If you don't care about preserving the order of the elements, <see cref="RemoveRangeSwapBackWithBeginEnd"/>
- /// is a more efficient way to remove elements.
- /// </remarks>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `index` is out of bounds, `count` is negative,
- /// or `index + count` exceeds the length.</exception>
- public void RemoveRange(int index, int count)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveRange(index, count);
- }
-
- /// <summary>
- /// Removes *N* elements in a range, shifting everything above it down by *N*. Decrements the length by *N*.
- /// </summary>
- /// <param name="begin">The index of the first element to remove.</param>
- /// <param name="end">The index one greater than the last element to remove.</param>
- /// <remarks>
- /// If you don't care about preserving the order of the elements, <see cref="RemoveRangeSwapBackWithBeginEnd"/> is a more efficient way to remove elements.
- /// </remarks>
- /// <exception cref="ArgumentException">Thrown if `end < begin`.</exception>
- /// <exception cref="ArgumentOutOfRangeException">Thrown if `begin` or `end` are out of bounds.</exception>
- [Obsolete("RemoveRangeWithBeginEnd(begin, end) is deprecated, use RemoveRange(index, count) instead. (RemovedAfter 2021-06-02)", false)]
- public void RemoveRangeWithBeginEnd(int begin, int end)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->RemoveRangeWithBeginEnd(begin, end);
- }
-
- /// <summary>
- /// Whether this list is empty.
- /// </summary>
- /// <value>True if the list is empty or if the list has not been constructed.</value>
- public bool IsEmpty => !IsCreated || Length == 0;
-
- /// <summary>
- /// Whether this list has been allocated (and not yet deallocated).
- /// </summary>
- /// <value>True if this list has been allocated (and not yet deallocated).</value>
- public bool IsCreated => m_ListData != null;
-
- /// <summary>
- /// Releases all resources (memory and safety handles).
- /// </summary>
- public void Dispose()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- m_DeprecatedAllocator.TryRemoveSafetyHandle(m_Safety, m_SafetyIndexHint);
- #if REMOVE_DISPOSE_SENTINEL
- CollectionHelper.DisposeSafetyHandle(ref m_Safety);
- #else
- DisposeSentinel.Dispose(ref m_Safety, ref m_DisposeSentinel);
- #endif
- #endif
- UnsafeList<T>.Destroy(m_ListData);
- m_ListData = null;
- }
-
- /// <summary>
- /// Releases all resources (memory and safety handles).
- /// <typeparam name="U">The type of allocator.</typeparam>
- /// <param name="allocator">The allocator that was used to allocate this list.</param>
- /// </summary>
- [BurstCompatible(GenericTypeArguments = new[] { typeof(AllocatorManager.AllocatorHandle) })]
- internal void Dispose<U>(ref U allocator) where U : unmanaged, AllocatorManager.IAllocator
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- CheckHandleMatches(allocator.Handle);
- m_DeprecatedAllocator.TryRemoveSafetyHandle(m_Safety, m_SafetyIndexHint);
- #if REMOVE_DISPOSE_SENTINEL
- CollectionHelper.DisposeSafetyHandle(ref m_Safety);
- #else
- DisposeSentinel.Dispose(ref m_Safety, ref m_DisposeSentinel);
- #endif
- #endif
- UnsafeList<T>.Destroy(m_ListData, ref allocator);
- m_ListData = null;
- }
-
- /// <summary>
- /// Creates and schedules a job that releases all resources (memory and safety handles) of this list.
- /// </summary>
- /// <param name="inputDeps">The dependency for the new job.</param>
- /// <returns>The handle of the new job. The job depends upon `inputDeps` and releases all resources (memory and safety handles) of this list.</returns>
- [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 ENABLE_UNITY_COLLECTIONS_CHECKS
- #if REMOVE_DISPOSE_SENTINEL
- #else
- // [DeallocateOnJobCompletion] is not supported, but we want the deallocation
- // to happen in a thread. DisposeSentinel needs to be cleared on main thread.
- // AtomicSafetyHandle can be destroyed after the job was scheduled (Job scheduling
- // will check that no jobs are writing to the container).
- DisposeSentinel.Clear(ref m_DisposeSentinel);
- #endif
-
- var jobHandle = new NativeListDisposeJob { Data = new NativeListDispose { m_ListData = (UntypedUnsafeList*)m_ListData, m_Safety = m_Safety } }.Schedule(inputDeps);
-
- AtomicSafetyHandle.Release(m_Safety);
- #else
- var jobHandle = new NativeListDisposeJob { Data = new NativeListDispose { m_ListData = (UntypedUnsafeList*)m_ListData } }.Schedule(inputDeps);
- #endif
- m_ListData = null;
-
- return jobHandle;
- }
-
- /// <summary>
- /// Sets the length to 0.
- /// </summary>
- /// <remarks>Does not change the capacity.</remarks>
- public void Clear()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- #endif
- m_ListData->Clear();
- }
-
- /// <summary>
- /// Returns a native array that aliases the content of a list.
- /// </summary>
- /// <param name="nativeList">The list to alias.</param>
- /// <returns>A native array that aliases the content of the list.</returns>
- public static implicit operator NativeArray<T>(NativeList<T> nativeList)
- {
- return nativeList.AsArray();
- }
-
- /// <summary>
- /// Returns a native array that aliases the content of this list.
- /// </summary>
- /// <returns>A native array that aliases the content of this list.</returns>
- public NativeArray<T> AsArray()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckGetSecondaryDataPointerAndThrow(m_Safety);
- var arraySafety = m_Safety;
- AtomicSafetyHandle.UseSecondaryVersion(ref arraySafety);
- #endif
- var array = NativeArrayUnsafeUtility.ConvertExistingDataToNativeArray<T>(m_ListData->Ptr, m_ListData->Length, Allocator.None);
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- NativeArrayUnsafeUtility.SetAtomicSafetyHandle(ref array, arraySafety);
- #endif
- return array;
- }
-
- /// <summary>
- /// Returns an array that aliases this list. The length of the array is updated when the length of
- /// this array is updated in a prior job.
- /// </summary>
- /// <remarks>
- /// Useful when a job populates a list that is then used by another job.
- ///
- /// If you pass both jobs the same list, you have to complete the first job before you schedule the second:
- /// otherwise, the second job doesn't see the first job's changes to the list's length.
- ///
- /// If instead you pass the second job a deferred array that aliases the list, the array's length is kept in sync with
- /// the first job's changes to the list's length. Consequently, the first job doesn't have to
- /// be completed before you can schedule the second: the second job simply has to depend upon the first.
- /// </remarks>
- /// <returns>An array that aliases this list and whose length can be specially modified across jobs.</returns>
- /// <example>
- /// The following example populates a list with integers in one job and passes that data to a second job as
- /// a deferred array. If we tried to pass the list directly to the second job, that job would not see any
- /// modifications made to the list by the first job. To avoid this, we instead pass the second job a deferred array that aliases the list.
- /// <code>
- /// using UnityEngine;
- /// using Unity.Jobs;
- /// using Unity.Collections;
- ///
- /// public class DeferredArraySum : MonoBehaviour
- ///{
- /// public struct Populate : IJob
- /// {
- /// public NativeList<int> list;
- ///
- /// public void Execute()
- /// {
- /// for (int i = list.Length; i < list.Capacity; i++)
- /// {
- /// list.Add(i);
- /// }
- /// }
- /// }
- ///
- /// // Sums all numbers from deferred.
- /// public struct Sum : IJob
- /// {
- /// [ReadOnly] public NativeArray<int> deferred;
- /// public NativeArray<int> sum;
- ///
- /// public void Execute()
- /// {
- /// sum[0] = 0;
- /// for (int i = 0; i < deferred.Length; i++)
- /// {
- /// sum[0] += deferred[i];
- /// }
- /// }
- /// }
- ///
- /// void Start()
- /// {
- /// var list = new NativeList<int>(100, Allocator.TempJob);
- /// var deferred = list.AsDeferredJobArray(),
- /// var output = new NativeArray<int>(1, Allocator.TempJob);
- ///
- /// // The Populate job increases the list's length from 0 to 100.
- /// var populate = new Populate { list = list }.Schedule();
- ///
- /// // At time of scheduling, the length of the deferred array given to Sum is 0.
- /// // When Populate increases the list's length, the deferred array's length field in the
- /// // Sum job is also modified, even though it has already been scheduled.
- /// var sum = new Sum { deferred = deferred, sum = output }.Schedule(populate);
- ///
- /// sum.Complete();
- ///
- /// Debug.Log("Result: " + output[0]);
- ///
- /// list.Dispose();
- /// output.Dispose();
- /// }
- /// }
- /// </code>
- /// </example>
- public NativeArray<T> AsDeferredJobArray()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckExistsAndThrow(m_Safety);
- #endif
- byte* buffer = (byte*)m_ListData;
- // We use the first bit of the pointer to infer that the array is in list mode
- // Thus the job scheduling code will need to patch it.
- buffer += 1;
- var array = NativeArrayUnsafeUtility.ConvertExistingDataToNativeArray<T>(buffer, 0, Allocator.Invalid);
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- NativeArrayUnsafeUtility.SetAtomicSafetyHandle(ref array, m_Safety);
- #endif
-
- return array;
- }
-
- /// <summary>
- /// Returns a managed array containing a copy of this list's content.
- /// </summary>
- /// <returns>A NativeArray containing copies of all the items in the list.</returns>
- [NotBurstCompatible /* Deprecated */]
- // todo: reenable deprecation warning before release [see: DOTS-DOTS-4217]
- // [Obsolete("Please use `ToArrayNBC` from `Unity.Collections.NotBurstCompatible` namespace instead. (RemovedAfter 2021-06-22)", false)]
- public T[] ToArray() => NotBurstCompatible.Extensions.ToArrayNBC(this);
-
- /// <summary>
- /// Returns an array containing a copy of this list's content.
- /// </summary>
- /// <param name="allocator">The allocator to use.</param>
- /// <returns>An array containing a copy of this list's content.</returns>
- public NativeArray<T> ToArray(AllocatorManager.AllocatorHandle allocator)
- {
- NativeArray<T> result = CollectionHelper.CreateNativeArray<T>(Length, allocator, NativeArrayOptions.UninitializedMemory);
- result.CopyFrom(this);
- return result;
- }
-
- /// <summary>
- /// Returns an enumerator over the elements of this list.
- /// </summary>
- /// <returns>An enumerator over the elements of this list.</returns>
- public NativeArray<T>.Enumerator GetEnumerator()
- {
- var array = AsArray();
- return new NativeArray<T>.Enumerator(ref array);
- }
-
- /// <summary>
- /// This method is not implemented. Use <see cref="GetEnumerator"/> instead.
- /// </summary>
- /// <returns>Throws NotImplementedException.</returns>
- /// <exception cref="NotImplementedException">Method is not implemented.</exception>
- IEnumerator IEnumerable.GetEnumerator()
- {
- throw new NotImplementedException();
- }
-
- /// <summary>
- /// This method is not implemented. Use <see cref="GetEnumerator"/> instead.
- /// </summary>
- /// <returns>Throws NotImplementedException.</returns>
- /// <exception cref="NotImplementedException">Method is not implemented.</exception>
- IEnumerator<T> IEnumerable<T>.GetEnumerator()
- {
- throw new NotImplementedException();
- }
-
- /// <summary>
- /// Overwrites the elements of this list with the elements of an equal-length array.
- /// </summary>
- /// <param name="array">An array to copy into this list.</param>
- /// <exception cref="ArgumentException">Thrown if the array and list have unequal length.</exception>
- [NotBurstCompatible /* Deprecated */]
- [Obsolete("Please use `CopyFromNBC` from `Unity.Collections.NotBurstCompatible` namespace instead. (RemovedAfter 2021-06-22)", false)]
- public void CopyFrom(T[] array) => NotBurstCompatible.Extensions.CopyFromNBC(this, array);
-
- /// <summary>
- /// Overwrites the elements of this list with the elements of an equal-length array.
- /// </summary>
- /// <param name="array">An array to copy into this list.</param>
- /// <exception cref="ArgumentException">Thrown if the array and list have unequal length.</exception>
- public void CopyFrom(NativeArray<T> array)
- {
- Clear();
- Resize(array.Length, NativeArrayOptions.UninitializedMemory);
- NativeArray<T> thisArray = AsArray();
- thisArray.CopyFrom(array);
- }
-
- /// <summary>
- /// Sets the length of this list, increasing the capacity if necessary.
- /// </summary>
- /// <param name="length">The new length of this list.</param>
- /// <param name="options">Whether to clear any newly allocated bytes to all zeroes.</param>
- public void Resize(int length, NativeArrayOptions options)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndBumpSecondaryVersion(m_Safety);
- AtomicSafetyHandle.CheckReadAndThrow(m_Safety);
- #endif
- m_ListData->Resize(length, options);
- }
-
- /// <summary>
- /// Sets the length of this list, increasing the capacity if necessary.
- /// </summary>
- /// <remarks>Does not clear newly allocated bytes.</remarks>
- /// <param name="length">The new length of this list.</param>
- public void ResizeUninitialized(int length)
- {
- Resize(length, NativeArrayOptions.UninitializedMemory);
- }
-
- /// <summary>
- /// Sets the capacity.
- /// </summary>
- /// <param name="capacity">The new capacity.</param>
- public void SetCapacity(int capacity)
- {
- m_ListData->SetCapacity(capacity);
- }
-
- /// <summary>
- /// Sets the capacity to match the length.
- /// </summary>
- public void TrimExcess()
- {
- m_ListData->TrimExcess();
- }
-
- /// <summary>
- /// Returns a parallel reader of this list.
- /// </summary>
- /// <returns>A parallel reader of this list.</returns>
- public NativeArray<T>.ReadOnly AsParallelReader()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- return new NativeArray<T>.ReadOnly(m_ListData->Ptr, m_ListData->Length, ref m_Safety);
- #else
- return new NativeArray<T>.ReadOnly(m_ListData->Ptr, m_ListData->Length);
- #endif
- }
-
- /// <summary>
- /// Returns a parallel writer of this list.
- /// </summary>
- /// <returns>A parallel writer of this list.</returns>
- public ParallelWriter AsParallelWriter()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- return new ParallelWriter(m_ListData, ref m_Safety);
- #else
- return new ParallelWriter(m_ListData);
- #endif
- }
-
- /// <summary>
- /// A parallel writer for a NativeList.
- /// </summary>
- /// <remarks>
- /// Use <see cref="AsParallelWriter"/> to create a parallel writer for a list.
- /// </remarks>
- [NativeContainer]
- [NativeContainerIsAtomicWriteOnly]
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public unsafe struct ParallelWriter
- {
- /// <summary>
- /// The data of the list.
- /// </summary>
- public readonly void* Ptr => ListData->Ptr;
-
- /// <summary>
- /// The internal unsafe list.
- /// </summary>
- /// <value>The internal unsafe list.</value>
- [NativeDisableUnsafePtrRestriction]
- public UnsafeList<T>* ListData;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- internal AtomicSafetyHandle m_Safety;
- internal static readonly SharedStatic<int> s_staticSafetyId = SharedStatic<int>.GetOrCreate<ParallelWriter>();
-
- [BurstCompatible(CompileTarget = BurstCompatibleAttribute.BurstCompatibleCompileTarget.Editor)]
- internal unsafe ParallelWriter(UnsafeList<T>* listData, ref AtomicSafetyHandle safety)
- {
- ListData = listData;
- m_Safety = safety;
- CollectionHelper.SetStaticSafetyId<ParallelWriter>(ref m_Safety, ref s_staticSafetyId.Data);
- }
- #else
- internal unsafe ParallelWriter(UnsafeList<T>* listData)
- {
- ListData = listData;
- }
- #endif
-
- /// <summary>
- /// Appends an element to the end of this list.
- /// </summary>
- /// <param name="value">The value to add to the end of this list.</param>
- /// <remarks>
- /// Increments the length by 1 unless doing so would exceed the current capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if adding an element would exceed the capacity.</exception>
- public void AddNoResize(T value)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- var idx = Interlocked.Increment(ref ListData->m_length) - 1;
- CheckSufficientCapacity(ListData->Capacity, idx + 1);
-
- UnsafeUtility.WriteArrayElement(ListData->Ptr, idx, value);
- }
-
- /// <summary>
- /// Appends elements from a buffer to the end of this list.
- /// </summary>
- /// <param name="ptr">The buffer to copy from.</param>
- /// <param name="count">The number of elements to copy from the buffer.</param>
- /// <remarks>
- /// Increments the length by `count` unless doing so would exceed the current capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if adding the elements would exceed the capacity.</exception>
- public void AddRangeNoResize(void* ptr, int count)
- {
- CheckArgPositive(count);
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- var idx = Interlocked.Add(ref ListData->m_length, count) - count;
- CheckSufficientCapacity(ListData->Capacity, idx + count);
-
- var sizeOf = sizeof(T);
- void* dst = (byte*)ListData->Ptr + idx * sizeOf;
- UnsafeUtility.MemCpy(dst, ptr, count * sizeOf);
- }
-
- /// <summary>
- /// Appends the elements of another list to the end of this list.
- /// </summary>
- /// <param name="list">The other list to copy from.</param>
- /// <remarks>
- /// Increments the length of this list by the length of the other list unless doing so would exceed the current capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if adding the elements would exceed the capacity.</exception>
- public void AddRangeNoResize(UnsafeList<T> list)
- {
- AddRangeNoResize(list.Ptr, list.Length);
- }
-
- /// <summary>
- /// Appends the elements of another list to the end of this list.
- /// </summary>
- /// <param name="list">The other list to copy from.</param>
- /// <remarks>
- /// Increments the length of this list by the length of the other list unless doing so would exceed the current capacity.
- /// </remarks>
- /// <exception cref="Exception">Thrown if adding the elements would exceed the capacity.</exception>
- public void AddRangeNoResize(NativeList<T> list)
- {
- AddRangeNoResize(*list.m_ListData);
- }
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void CheckInitialCapacity(int initialCapacity)
- {
- if (initialCapacity < 0)
- throw new ArgumentOutOfRangeException(nameof(initialCapacity), "Capacity must be >= 0");
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void CheckTotalSize(int initialCapacity, long totalSize)
- {
- // Make sure we cannot allocate more than int.MaxValue (2,147,483,647 bytes)
- // because the underlying UnsafeUtility.Malloc is expecting a int.
- // TODO: change UnsafeUtility.Malloc to accept a UIntPtr length instead to match C++ API
- if (totalSize > int.MaxValue)
- throw new ArgumentOutOfRangeException(nameof(initialCapacity), $"Capacity * sizeof(T) cannot exceed {int.MaxValue} bytes");
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void CheckSufficientCapacity(int capacity, int length)
- {
- if (capacity < length)
- throw new Exception($"Length {length} exceeds capacity Capacity {capacity}");
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void CheckIndexInRange(int value, int length)
- {
- if (value < 0)
- throw new IndexOutOfRangeException($"Value {value} must be positive.");
-
- if ((uint)value >= (uint)length)
- throw new IndexOutOfRangeException($"Value {value} is out of range in NativeList of '{length}' Length.");
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void CheckArgPositive(int value)
- {
- if (value < 0)
- throw new ArgumentOutOfRangeException($"Value {value} must be positive.");
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- void CheckHandleMatches(AllocatorManager.AllocatorHandle handle)
- {
- if(m_ListData == null)
- throw new ArgumentOutOfRangeException($"Allocator handle {handle} can't match because container is not initialized.");
- if(m_ListData->Allocator.Index != handle.Index)
- throw new ArgumentOutOfRangeException($"Allocator handle {handle} can't match because container handle index doesn't match.");
- if(m_ListData->Allocator.Version != handle.Version)
- throw new ArgumentOutOfRangeException($"Allocator handle {handle} matches container handle index, but has different version.");
- }
- }
-
- [NativeContainer]
- [BurstCompatible]
- internal unsafe struct NativeListDispose
- {
- [NativeDisableUnsafePtrRestriction]
- public UntypedUnsafeList* m_ListData;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- internal AtomicSafetyHandle m_Safety;
- #endif
-
- public void Dispose()
- {
- var listData = (UnsafeList<int>*)m_ListData;
- UnsafeList<int>.Destroy(listData);
- }
- }
-
- [BurstCompile]
- [BurstCompatible]
- internal unsafe struct NativeListDisposeJob : IJob
- {
- internal NativeListDispose Data;
-
- public void Execute()
- {
- Data.Dispose();
- }
- }
-
- sealed class NativeListDebugView<T> where T : unmanaged
- {
- NativeList<T> m_Array;
-
- public NativeListDebugView(NativeList<T> array)
- {
- m_Array = array;
- }
-
- public T[] Items => m_Array.AsArray().ToArray();
- }
- }
-
- namespace Unity.Collections.LowLevel.Unsafe
- {
- /// <summary>
- /// Provides unsafe utility methods for NativeList.
- /// </summary>
- [BurstCompatible]
- public unsafe static class NativeListUnsafeUtility
- {
- /// <summary>
- /// Returns a pointer to this list's internal buffer.
- /// </summary>
- /// <remarks>Performs a job safety check for read-write access.</remarks>
- /// <param name="list">The list.</param>
- /// <typeparam name="T">The type of the elements.</typeparam>
- /// <returns>A pointer to this list's internal buffer.</returns>
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public static void* GetUnsafePtr<T>(this NativeList<T> list) where T : unmanaged
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(list.m_Safety);
- #endif
- return list.m_ListData->Ptr;
- }
-
- /// <summary>
- /// Returns a pointer to this list's internal buffer.
- /// </summary>
- /// <remarks>Performs a job safety check for read-only access.</remarks>
- /// <param name="list">The list.</param>
- /// <typeparam name="T">The type of the elements.</typeparam>
- /// <returns>A pointer to this list's internal buffer.</returns>
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public static unsafe void* GetUnsafeReadOnlyPtr<T>(this NativeList<T> list) where T : unmanaged
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckReadAndThrow(list.m_Safety);
- #endif
- return list.m_ListData->Ptr;
- }
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- /// <summary>
- /// Returns this list's <see cref="AtomicSafetyHandle"/>.
- /// </summary>
- /// <param name="list">The list.</param>
- /// <typeparam name="T">The type of the elements.</typeparam>
- /// <returns>The atomic safety handle for this list.</returns>
- /// <remarks>
- /// The job safety checks use a native collection's atomic safety handle to assert safety.
- ///
- /// This method is only available if the symbol `ENABLE_UNITY_COLLECTIONS_CHECKS` is defined.</remarks>
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) }, RequiredUnityDefine = "ENABLE_UNITY_COLLECTIONS_CHECKS", CompileTarget = BurstCompatibleAttribute.BurstCompatibleCompileTarget.Editor)]
- public static AtomicSafetyHandle GetAtomicSafetyHandle<T>(ref NativeList<T> list) where T : unmanaged
- {
- return list.m_Safety;
- }
-
- #endif
-
- /// <summary>
- /// Returns a pointer to this list's internal unsafe list.
- /// </summary>
- /// <remarks>Performs no job safety checks.</remarks>
- /// <param name="list">The list.</param>
- /// <typeparam name="T">The type of the elements.</typeparam>
- /// <returns>A pointer to this list's internal unsafe list.</returns>
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public static void* GetInternalListDataPtrUnchecked<T>(ref NativeList<T> list) where T : unmanaged
- {
- return list.m_ListData;
- }
- }
- }
|