PO/Library/PackageCache/com.unity.2d.spriteshape@5.1.4/Runtime/UTess2D/ArraySlice.cs

165 lines
5.9 KiB
C#

using System;
using System.Diagnostics;
using System.Runtime.InteropServices;
using Unity.Collections;
using Unity.Collections.LowLevel.Unsafe;
namespace UnityEngine.U2D
{
namespace UTess
{
[StructLayout(LayoutKind.Sequential)]
[DebuggerDisplay("Length = {Length}")]
internal unsafe struct ArraySlice<T> : System.IEquatable<ArraySlice<T>> where T : struct
{
[NativeDisableUnsafePtrRestriction] internal byte* m_Buffer;
internal int m_Stride;
internal int m_Length;
#if ENABLE_UNITY_COLLECTIONS_CHECKS
internal int m_MinIndex;
internal int m_MaxIndex;
internal AtomicSafetyHandle m_Safety;
#endif
public ArraySlice(NativeArray<T> array, int start, int length)
{
#if ENABLE_UNITY_COLLECTIONS_CHECKS
if (start < 0)
throw new ArgumentOutOfRangeException(nameof(start), $"Slice start {start} < 0.");
if (length < 0)
throw new ArgumentOutOfRangeException(nameof(length), $"Slice length {length} < 0.");
if (start + length > array.Length)
throw new ArgumentException(
$"Slice start + length ({start + length}) range must be <= array.Length ({array.Length})");
m_MinIndex = 0;
m_MaxIndex = length - 1;
m_Safety = Unity.Collections.LowLevel.Unsafe.NativeArrayUnsafeUtility.GetAtomicSafetyHandle(array);
#endif
m_Stride = UnsafeUtility.SizeOf<T>();
var ptr = (byte*) array.GetUnsafePtr() + m_Stride * start;
m_Buffer = ptr;
m_Length = length;
}
public bool Equals(ArraySlice<T> other)
{
return m_Buffer == other.m_Buffer && m_Stride == other.m_Stride && m_Length == other.m_Length;
}
public override bool Equals(object obj)
{
if (ReferenceEquals(null, obj)) return false;
return obj is ArraySlice<T> && Equals((ArraySlice<T>) obj);
}
public override int GetHashCode()
{
unchecked
{
var hashCode = (int) m_Buffer;
hashCode = (hashCode * 397) ^ m_Stride;
hashCode = (hashCode * 397) ^ m_Length;
return hashCode;
}
}
public static bool operator ==(ArraySlice<T> left, ArraySlice<T> right)
{
return left.Equals(right);
}
public static bool operator !=(ArraySlice<T> left, ArraySlice<T> right)
{
return !left.Equals(right);
}
#if ENABLE_UNITY_COLLECTIONS_CHECKS
// These are double-whammy excluded to we can elide bounds checks in the Burst disassembly view
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckReadIndex(int index)
{
#if ENABLE_UNITY_COLLECTIONS_CHECKS
if (index < m_MinIndex || index > m_MaxIndex)
FailOutOfRangeError(index);
#endif
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
void CheckWriteIndex(int index)
{
#if ENABLE_UNITY_COLLECTIONS_CHECKS
if (index < m_MinIndex || index > m_MaxIndex)
FailOutOfRangeError(index);
#endif
}
[Conditional("ENABLE_UNITY_COLLECTIONS_CHECKS")]
private void FailOutOfRangeError(int index)
{
if (index < Length && (m_MinIndex != 0 || m_MaxIndex != Length - 1))
throw new System.IndexOutOfRangeException(
$"Index {index} is out of restricted IJobParallelFor range [{m_MinIndex}...{m_MaxIndex}] in ReadWriteBuffer.\n" +
"ReadWriteBuffers are restricted to only read & write the element at the job index. " +
"You can use double buffering strategies to avoid race conditions due to " +
"reading & writing in parallel to the same elements from a job.");
throw new System.IndexOutOfRangeException($"Index {index} is out of range of '{Length}' Length.");
}
#endif
public static unsafe ArraySlice<T> ConvertExistingDataToArraySlice(void* dataPointer, int stride, int length)
{
if (length < 0)
throw new System.ArgumentException($"Invalid length of '{length}'. It must be greater than 0.",
nameof(length));
if (stride < 0)
throw new System.ArgumentException($"Invalid stride '{stride}'. It must be greater than 0.",
nameof(stride));
var newSlice = new ArraySlice<T>
{
m_Stride = stride,
m_Buffer = (byte*) dataPointer,
m_Length = length,
#if ENABLE_UNITY_COLLECTIONS_CHECKS
m_MinIndex = 0,
m_MaxIndex = length - 1,
#endif
};
return newSlice;
}
public T this[int index]
{
get
{
#if ENABLE_UNITY_COLLECTIONS_CHECKS
CheckReadIndex(index);
#endif
return UnsafeUtility.ReadArrayElementWithStride<T>(m_Buffer, index, m_Stride);
}
[WriteAccessRequired]
set
{
#if ENABLE_UNITY_COLLECTIONS_CHECKS
CheckWriteIndex(index);
#endif
UnsafeUtility.WriteArrayElementWithStride(m_Buffer, index, m_Stride, value);
}
}
public int Stride => m_Stride;
public int Length => m_Length;
}
}
}