123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693 |
- using System;
- using System.Runtime.InteropServices;
- using System.Threading;
- using Unity.Collections.LowLevel.Unsafe;
- using Unity.Burst;
- using Unity.Jobs;
- using Unity.Jobs.LowLevel.Unsafe;
- using System.Diagnostics;
-
- namespace Unity.Collections
- {
- unsafe struct NativeQueueBlockHeader
- {
- public NativeQueueBlockHeader* m_NextBlock;
- public int m_NumItems;
- }
-
- [StructLayout(LayoutKind.Sequential)]
- [BurstCompatible]
- internal unsafe struct NativeQueueBlockPoolData
- {
- internal IntPtr m_FirstBlock;
- internal int m_NumBlocks;
- internal int m_MaxBlocks;
- internal const int m_BlockSize = 16 * 1024;
- internal int m_AllocLock;
-
- public NativeQueueBlockHeader* AllocateBlock()
- {
- // There can only ever be a single thread allocating an entry from the free list since it needs to
- // access the content of the block (the next pointer) before doing the CAS.
- // If there was no lock thread A could read the next pointer, thread B could quickly allocate
- // the same block then free it with another next pointer before thread A performs the CAS which
- // leads to an invalid free list potentially causing memory corruption.
- // Having multiple threads freeing data concurrently to each other while another thread is allocating
- // is no problems since there is only ever a single thread modifying global data in that case.
- while (Interlocked.CompareExchange(ref m_AllocLock, 1, 0) != 0)
- {
- }
-
- NativeQueueBlockHeader* checkBlock = (NativeQueueBlockHeader*)m_FirstBlock;
- NativeQueueBlockHeader* block;
-
- do
- {
- block = checkBlock;
- if (block == null)
- {
- Interlocked.Exchange(ref m_AllocLock, 0);
- Interlocked.Increment(ref m_NumBlocks);
- block = (NativeQueueBlockHeader*)Memory.Unmanaged.Allocate(m_BlockSize, 16, Allocator.Persistent);
- return block;
- }
-
- checkBlock = (NativeQueueBlockHeader*)Interlocked.CompareExchange(ref m_FirstBlock, (IntPtr)block->m_NextBlock, (IntPtr)block);
- }
- while (checkBlock != block);
-
- Interlocked.Exchange(ref m_AllocLock, 0);
-
- return block;
- }
-
- public void FreeBlock(NativeQueueBlockHeader* block)
- {
- if (m_NumBlocks > m_MaxBlocks)
- {
- if (Interlocked.Decrement(ref m_NumBlocks) + 1 > m_MaxBlocks)
- {
- Memory.Unmanaged.Free(block, Allocator.Persistent);
- return;
- }
-
- Interlocked.Increment(ref m_NumBlocks);
- }
-
- NativeQueueBlockHeader* checkBlock = (NativeQueueBlockHeader*)m_FirstBlock;
- NativeQueueBlockHeader* nextPtr;
-
- do
- {
- nextPtr = checkBlock;
- block->m_NextBlock = checkBlock;
- checkBlock = (NativeQueueBlockHeader*)Interlocked.CompareExchange(ref m_FirstBlock, (IntPtr)block, (IntPtr)checkBlock);
- }
- while (checkBlock != nextPtr);
- }
- }
-
- internal unsafe class NativeQueueBlockPool
- {
- static readonly SharedStatic<IntPtr> Data = SharedStatic<IntPtr>.GetOrCreate<NativeQueueBlockPool>();
-
- internal static NativeQueueBlockPoolData* GetQueueBlockPool()
- {
- var pData = (NativeQueueBlockPoolData**)Data.UnsafeDataPointer;
- var data = *pData;
-
- if (data == null)
- {
- data = (NativeQueueBlockPoolData*)Memory.Unmanaged.Allocate(UnsafeUtility.SizeOf<NativeQueueBlockPoolData>(), 8, Allocator.Persistent);
- *pData = data;
- data->m_NumBlocks = data->m_MaxBlocks = 256;
- data->m_AllocLock = 0;
- // Allocate MaxBlocks items
- NativeQueueBlockHeader* prev = null;
-
- for (int i = 0; i < data->m_MaxBlocks; ++i)
- {
- NativeQueueBlockHeader* block = (NativeQueueBlockHeader*)Memory.Unmanaged.Allocate(NativeQueueBlockPoolData.m_BlockSize, 16, Allocator.Persistent);
- block->m_NextBlock = prev;
- prev = block;
- }
- data->m_FirstBlock = (IntPtr)prev;
-
- AppDomainOnDomainUnload();
- }
- return data;
- }
-
- [BurstDiscard]
- static void AppDomainOnDomainUnload()
- {
- #if !UNITY_DOTSRUNTIME
- AppDomain.CurrentDomain.DomainUnload += OnDomainUnload;
- #endif
- }
-
- #if !UNITY_DOTSRUNTIME
- static void OnDomainUnload(object sender, EventArgs e)
- {
- var pData = (NativeQueueBlockPoolData**)Data.UnsafeDataPointer;
- var data = *pData;
-
- while (data->m_FirstBlock != IntPtr.Zero)
- {
- NativeQueueBlockHeader* block = (NativeQueueBlockHeader*)data->m_FirstBlock;
- data->m_FirstBlock = (IntPtr)block->m_NextBlock;
- Memory.Unmanaged.Free(block, Allocator.Persistent);
- --data->m_NumBlocks;
- }
- Memory.Unmanaged.Free(data, Allocator.Persistent);
- *pData = null;
- }
- #endif
- }
-
- [StructLayout(LayoutKind.Sequential)]
- [BurstCompatible]
- internal unsafe struct NativeQueueData
- {
- public IntPtr m_FirstBlock;
- public IntPtr m_LastBlock;
- public int m_MaxItems;
- public int m_CurrentRead;
- public byte* m_CurrentWriteBlockTLS;
-
- internal NativeQueueBlockHeader* GetCurrentWriteBlockTLS(int threadIndex)
- {
- var data = (NativeQueueBlockHeader**)&m_CurrentWriteBlockTLS[threadIndex * JobsUtility.CacheLineSize];
- return *data;
- }
-
- internal void SetCurrentWriteBlockTLS(int threadIndex, NativeQueueBlockHeader* currentWriteBlock)
- {
- var data = (NativeQueueBlockHeader**)&m_CurrentWriteBlockTLS[threadIndex * JobsUtility.CacheLineSize];
- *data = currentWriteBlock;
- }
-
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public static NativeQueueBlockHeader* AllocateWriteBlockMT<T>(NativeQueueData* data, NativeQueueBlockPoolData* pool, int threadIndex) where T : struct
- {
- NativeQueueBlockHeader* currentWriteBlock = data->GetCurrentWriteBlockTLS(threadIndex);
-
- if (currentWriteBlock != null
- && currentWriteBlock->m_NumItems == data->m_MaxItems)
- {
- currentWriteBlock = null;
- }
-
- if (currentWriteBlock == null)
- {
- currentWriteBlock = pool->AllocateBlock();
- currentWriteBlock->m_NextBlock = null;
- currentWriteBlock->m_NumItems = 0;
- NativeQueueBlockHeader* prevLast = (NativeQueueBlockHeader*)Interlocked.Exchange(ref data->m_LastBlock, (IntPtr)currentWriteBlock);
-
- if (prevLast == null)
- {
- data->m_FirstBlock = (IntPtr)currentWriteBlock;
- }
- else
- {
- prevLast->m_NextBlock = currentWriteBlock;
- }
-
- data->SetCurrentWriteBlockTLS(threadIndex, currentWriteBlock);
- }
-
- return currentWriteBlock;
- }
-
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public unsafe static void AllocateQueue<T>(AllocatorManager.AllocatorHandle label, out NativeQueueData* outBuf) where T : struct
- {
- var queueDataSize = CollectionHelper.Align(UnsafeUtility.SizeOf<NativeQueueData>(), JobsUtility.CacheLineSize);
-
- var data = (NativeQueueData*)Memory.Unmanaged.Allocate(
- queueDataSize
- + JobsUtility.CacheLineSize * JobsUtility.MaxJobThreadCount
- , JobsUtility.CacheLineSize
- , label
- );
-
- data->m_CurrentWriteBlockTLS = (((byte*)data) + queueDataSize);
-
- data->m_FirstBlock = IntPtr.Zero;
- data->m_LastBlock = IntPtr.Zero;
- data->m_MaxItems = (NativeQueueBlockPoolData.m_BlockSize - UnsafeUtility.SizeOf<NativeQueueBlockHeader>()) / UnsafeUtility.SizeOf<T>();
-
- data->m_CurrentRead = 0;
- for (int threadIndex = 0; threadIndex < JobsUtility.MaxJobThreadCount; ++threadIndex)
- {
- data->SetCurrentWriteBlockTLS(threadIndex, null);
- }
-
- outBuf = data;
- }
-
- public unsafe static void DeallocateQueue(NativeQueueData* data, NativeQueueBlockPoolData* pool, AllocatorManager.AllocatorHandle allocation)
- {
- NativeQueueBlockHeader* firstBlock = (NativeQueueBlockHeader*)data->m_FirstBlock;
-
- while (firstBlock != null)
- {
- NativeQueueBlockHeader* next = firstBlock->m_NextBlock;
- pool->FreeBlock(firstBlock);
- firstBlock = next;
- }
-
- Memory.Unmanaged.Free(data, allocation);
- }
- }
-
- /// <summary>
- /// An unmanaged queue.
- /// </summary>
- /// <typeparam name="T">The type of the elements.</typeparam>
- [StructLayout(LayoutKind.Sequential)]
- [NativeContainer]
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public unsafe struct NativeQueue<T>
- : INativeDisposable
- where T : struct
- {
- [NativeDisableUnsafePtrRestriction]
- NativeQueueData* m_Buffer;
-
- [NativeDisableUnsafePtrRestriction]
- NativeQueueBlockPoolData* m_QueuePool;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle m_Safety;
-
- static readonly SharedStatic<int> s_staticSafetyId = SharedStatic<int>.GetOrCreate<NativeQueue<T>>();
-
- #if REMOVE_DISPOSE_SENTINEL
- #else
- [NativeSetClassTypeToNullOnSchedule]
- DisposeSentinel m_DisposeSentinel;
- #endif
- #endif
-
- AllocatorManager.AllocatorHandle m_AllocatorLabel;
-
- /// <summary>
- /// Initializes and returns an instance of NativeQueue.
- /// </summary>
- /// <param name="allocator">The allocator to use.</param>
- public NativeQueue(AllocatorManager.AllocatorHandle allocator)
- {
- CollectionHelper.CheckIsUnmanaged<T>();
-
- m_QueuePool = NativeQueueBlockPool.GetQueueBlockPool();
- m_AllocatorLabel = allocator;
-
- NativeQueueData.AllocateQueue<T>(allocator, out m_Buffer);
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- #if REMOVE_DISPOSE_SENTINEL
- m_Safety = CollectionHelper.CreateSafetyHandle(allocator);
- #else
- if (allocator.IsCustomAllocator)
- {
- m_Safety = AtomicSafetyHandle.Create();
- m_DisposeSentinel = null;
- }
- else
- {
- DisposeSentinel.Create(out m_Safety, out m_DisposeSentinel, 0, allocator.ToAllocator);
- }
- #endif
-
- CollectionHelper.SetStaticSafetyId<NativeQueue<T>>(ref m_Safety, ref s_staticSafetyId.Data);
- #endif
- }
-
- /// <summary>
- /// Returns true if this queue is empty.
- /// </summary>
- /// <value>True if this queue has no items or if the queue has not been constructed.</value>
- public bool IsEmpty()
- {
- if (!IsCreated)
- {
- return true;
- }
-
- CheckRead();
-
- int count = 0;
- var currentRead = m_Buffer->m_CurrentRead;
-
- for (NativeQueueBlockHeader* block = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock
- ; block != null
- ; block = block->m_NextBlock
- )
- {
- count += block->m_NumItems;
-
- if (count > currentRead)
- {
- return false;
- }
- }
-
- return count == currentRead;
- }
-
- /// <summary>
- /// Returns the current number of elements in this queue.
- /// </summary>
- /// <remarks>Note that getting the count requires traversing the queue's internal linked list of blocks.
- /// Where possible, cache this value instead of reading the property repeatedly.</remarks>
- /// <returns>The current number of elements in this queue.</returns>
- public int Count
- {
- get
- {
- CheckRead();
-
- int count = 0;
-
- for (NativeQueueBlockHeader* block = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock
- ; block != null
- ; block = block->m_NextBlock
- )
- {
- count += block->m_NumItems;
- }
-
- return count - m_Buffer->m_CurrentRead;
- }
- }
-
- internal static int PersistentMemoryBlockCount
- {
- get { return NativeQueueBlockPool.GetQueueBlockPool()->m_MaxBlocks; }
- set { Interlocked.Exchange(ref NativeQueueBlockPool.GetQueueBlockPool()->m_MaxBlocks, value); }
- }
-
- internal static int MemoryBlockSize
- {
- get { return NativeQueueBlockPoolData.m_BlockSize; }
- }
-
- /// <summary>
- /// Returns the element at the end of this queue without removing it.
- /// </summary>
- /// <returns>The element at the end of this queue.</returns>
- public T Peek()
- {
- CheckReadNotEmpty();
-
- NativeQueueBlockHeader* firstBlock = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock;
- return UnsafeUtility.ReadArrayElement<T>(firstBlock + 1, m_Buffer->m_CurrentRead);
- }
-
- /// <summary>
- /// Adds an element at the front of this queue.
- /// </summary>
- /// <param name="value">The value to be enqueued.</param>
- public void Enqueue(T value)
- {
- CheckWrite();
-
- NativeQueueBlockHeader* writeBlock = NativeQueueData.AllocateWriteBlockMT<T>(m_Buffer, m_QueuePool, 0);
- UnsafeUtility.WriteArrayElement(writeBlock + 1, writeBlock->m_NumItems, value);
- ++writeBlock->m_NumItems;
- }
-
- /// <summary>
- /// Removes and returns the element at the end of this queue.
- /// </summary>
- /// <exception cref="InvalidOperationException">Thrown if this queue is empty.</exception>
- /// <returns>The element at the end of this queue.</returns>
- public T Dequeue()
- {
- if (!TryDequeue(out T item))
- {
- ThrowEmpty();
- }
-
- return item;
- }
-
- /// <summary>
- /// Removes and outputs the element at the end of this queue.
- /// </summary>
- /// <param name="item">Outputs the removed element.</param>
- /// <returns>True if this queue was not empty.</returns>
- public bool TryDequeue(out T item)
- {
- CheckWrite();
-
- NativeQueueBlockHeader* firstBlock = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock;
-
- if (firstBlock == null)
- {
- item = default(T);
- return false;
- }
-
- var currentRead = m_Buffer->m_CurrentRead++;
- item = UnsafeUtility.ReadArrayElement<T>(firstBlock + 1, currentRead);
-
- if (m_Buffer->m_CurrentRead >= firstBlock->m_NumItems)
- {
- m_Buffer->m_CurrentRead = 0;
- m_Buffer->m_FirstBlock = (IntPtr)firstBlock->m_NextBlock;
-
- if (m_Buffer->m_FirstBlock == IntPtr.Zero)
- {
- m_Buffer->m_LastBlock = IntPtr.Zero;
- }
-
- for (int threadIndex = 0; threadIndex < JobsUtility.MaxJobThreadCount; ++threadIndex)
- {
- if (m_Buffer->GetCurrentWriteBlockTLS(threadIndex) == firstBlock)
- {
- m_Buffer->SetCurrentWriteBlockTLS(threadIndex, null);
- }
- }
-
- m_QueuePool->FreeBlock(firstBlock);
- }
-
- return true;
- }
-
- /// <summary>
- /// Returns an array containing a copy of this queue's content.
- /// </summary>
- /// <param name="allocator">The allocator to use.</param>
- /// <returns>An array containing a copy of this queue's content. The elements are ordered in the same order they were
- /// enqueued, *e.g.* the earliest enqueued element is copied to index 0 of the array.</returns>
- public NativeArray<T> ToArray(AllocatorManager.AllocatorHandle allocator)
- {
- CheckRead();
-
- NativeQueueBlockHeader* firstBlock = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock;
- var outputArray = CollectionHelper.CreateNativeArray<T>(Count, allocator);
-
- NativeQueueBlockHeader* currentBlock = firstBlock;
- var arrayPtr = (byte*)outputArray.GetUnsafePtr();
- int size = UnsafeUtility.SizeOf<T>();
- int dstOffset = 0;
- int srcOffset = m_Buffer->m_CurrentRead * size;
- int srcOffsetElements = m_Buffer->m_CurrentRead;
- while (currentBlock != null)
- {
- int bytesToCopy = (currentBlock->m_NumItems - srcOffsetElements) * size;
- UnsafeUtility.MemCpy(arrayPtr + dstOffset, (byte*)(currentBlock + 1) + srcOffset, bytesToCopy);
- srcOffset = srcOffsetElements = 0;
- dstOffset += bytesToCopy;
- currentBlock = currentBlock->m_NextBlock;
- }
-
- return outputArray;
- }
-
- /// <summary>
- /// Removes all elements of this queue.
- /// </summary>
- /// <remarks>Does not change the capacity.</remarks>
- public void Clear()
- {
- CheckWrite();
-
- NativeQueueBlockHeader* firstBlock = (NativeQueueBlockHeader*)m_Buffer->m_FirstBlock;
-
- while (firstBlock != null)
- {
- NativeQueueBlockHeader* next = firstBlock->m_NextBlock;
- m_QueuePool->FreeBlock(firstBlock);
- firstBlock = next;
- }
-
- m_Buffer->m_FirstBlock = IntPtr.Zero;
- m_Buffer->m_LastBlock = IntPtr.Zero;
- m_Buffer->m_CurrentRead = 0;
-
- for (int threadIndex = 0; threadIndex < JobsUtility.MaxJobThreadCount; ++threadIndex)
- {
- m_Buffer->SetCurrentWriteBlockTLS(threadIndex, null);
- }
- }
-
- /// <summary>
- /// Whether this queue has been allocated (and not yet deallocated).
- /// </summary>
- /// <value>True if this queue has been allocated (and not yet deallocated).</value>
- public bool IsCreated => m_Buffer != null;
-
- /// <summary>
- /// Releases all resources (memory and safety handles).
- /// </summary>
- public void Dispose()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- #if REMOVE_DISPOSE_SENTINEL
- CollectionHelper.DisposeSafetyHandle(ref m_Safety);
- #else
- DisposeSentinel.Dispose(ref m_Safety, ref m_DisposeSentinel);
- #endif
- #endif
- NativeQueueData.DeallocateQueue(m_Buffer, m_QueuePool, m_AllocatorLabel);
- m_Buffer = null;
- }
-
- /// <summary>
- /// Creates and schedules a job that releases all resources (memory and safety handles) of this queue.
- /// </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 queue.</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 NativeQueueDisposeJob { Data = new NativeQueueDispose { m_Buffer = m_Buffer, m_QueuePool = m_QueuePool, m_AllocatorLabel = m_AllocatorLabel, m_Safety = m_Safety } }.Schedule(inputDeps);
-
- AtomicSafetyHandle.Release(m_Safety);
- #else
- var jobHandle = new NativeQueueDisposeJob { Data = new NativeQueueDispose { m_Buffer = m_Buffer, m_QueuePool = m_QueuePool, m_AllocatorLabel = m_AllocatorLabel } }.Schedule(inputDeps);
- #endif
- m_Buffer = null;
-
- return jobHandle;
- }
-
- /// <summary>
- /// Returns a parallel writer for this queue.
- /// </summary>
- /// <returns>A parallel writer for this queue.</returns>
- public ParallelWriter AsParallelWriter()
- {
- ParallelWriter writer;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- writer.m_Safety = m_Safety;
- CollectionHelper.SetStaticSafetyId<ParallelWriter>(ref writer.m_Safety, ref ParallelWriter.s_staticSafetyId.Data);
- #endif
- writer.m_Buffer = m_Buffer;
- writer.m_QueuePool = m_QueuePool;
- writer.m_ThreadIndex = 0;
-
- return writer;
- }
-
- /// <summary>
- /// A parallel writer for a NativeQueue.
- /// </summary>
- /// <remarks>
- /// Use <see cref="AsParallelWriter"/> to create a parallel writer for a NativeQueue.
- /// </remarks>
- [NativeContainer]
- [NativeContainerIsAtomicWriteOnly]
- [BurstCompatible(GenericTypeArguments = new [] { typeof(int) })]
- public unsafe struct ParallelWriter
- {
- [NativeDisableUnsafePtrRestriction]
- internal NativeQueueData* m_Buffer;
-
- [NativeDisableUnsafePtrRestriction]
- internal NativeQueueBlockPoolData* m_QueuePool;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- internal AtomicSafetyHandle m_Safety;
- internal static readonly SharedStatic<int> s_staticSafetyId = SharedStatic<int>.GetOrCreate<ParallelWriter>();
- #endif
- [NativeSetThreadIndex]
- internal int m_ThreadIndex;
-
- /// <summary>
- /// Adds an element at the front of the queue.
- /// </summary>
- /// <param name="value">The value to be enqueued.</param>
- public void Enqueue(T value)
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- NativeQueueBlockHeader* writeBlock = NativeQueueData.AllocateWriteBlockMT<T>(m_Buffer, m_QueuePool, m_ThreadIndex);
- UnsafeUtility.WriteArrayElement(writeBlock + 1, writeBlock->m_NumItems, value);
- ++writeBlock->m_NumItems;
- }
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- void CheckRead()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckReadAndThrow(m_Safety);
- #endif
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- void CheckReadNotEmpty()
- {
- CheckRead();
-
- if (m_Buffer->m_FirstBlock == (IntPtr)0)
- {
- ThrowEmpty();
- }
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- void CheckWrite()
- {
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- AtomicSafetyHandle.CheckWriteAndThrow(m_Safety);
- #endif
- }
-
- [Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
- static void ThrowEmpty()
- {
- throw new InvalidOperationException("Trying to read from an empty queue.");
- }
- }
-
- [NativeContainer]
- [BurstCompatible]
- internal unsafe struct NativeQueueDispose
- {
- [NativeDisableUnsafePtrRestriction]
- internal NativeQueueData* m_Buffer;
-
- [NativeDisableUnsafePtrRestriction]
- internal NativeQueueBlockPoolData* m_QueuePool;
-
- internal AllocatorManager.AllocatorHandle m_AllocatorLabel;
-
- #if ENABLE_UNITY_COLLECTIONS_CHECKS
- internal AtomicSafetyHandle m_Safety;
- #endif
-
- public void Dispose()
- {
- NativeQueueData.DeallocateQueue(m_Buffer, m_QueuePool, m_AllocatorLabel);
- }
- }
-
- [BurstCompile]
- struct NativeQueueDisposeJob : IJob
- {
- public NativeQueueDispose Data;
-
- public void Execute()
- {
- Data.Dispose();
- }
- }
- }
|