535 lines
24 KiB
C#
535 lines
24 KiB
C#
|
using System;
|
||
|
using System.Collections.Generic;
|
||
|
using System.Linq;
|
||
|
using NUnit.Framework.Interfaces;
|
||
|
using NUnit.Framework.Internal;
|
||
|
using Unity.Mathematics;
|
||
|
|
||
|
namespace Burst.Compiler.IL.Tests.Helpers
|
||
|
{
|
||
|
[Flags]
|
||
|
internal enum DataRange
|
||
|
{
|
||
|
// Standard Test (Zero, Minus100To100, Inf, Nan)
|
||
|
Standard = Zero | Minus100To100 | Inf | NaN,
|
||
|
|
||
|
// Standard Test (Zero, ZeroExclusiveTo100, Inf, Nan)
|
||
|
StandardPositive = Zero | ZeroExclusiveTo100 | Inf | NaN,
|
||
|
|
||
|
// Standard Between -1 and 1 (Zero, MinusOneInclusiveToOneInclusive, Inf, Nan)
|
||
|
Standard11 = Zero | MinusOneInclusiveToOneInclusive | Inf | NaN,
|
||
|
|
||
|
Zero = 1 << 1,
|
||
|
ZeroExclusiveToOneInclusive = 1 << 2,
|
||
|
MinusOneInclusiveToOneInclusive = 1 << 3,
|
||
|
Minus100To100 = 1 << 4,
|
||
|
ZeroExclusiveTo100 = 1 << 5,
|
||
|
Inf = 1 << 6,
|
||
|
NaN = 1 << 7,
|
||
|
}
|
||
|
|
||
|
internal static class DataRangeExtensions
|
||
|
{
|
||
|
private const int VectorsCount = 6;
|
||
|
|
||
|
public static IEnumerable<object> ExpandRange(this DataRange dataRange, Type type, int seed)
|
||
|
{
|
||
|
if (type == typeof(bool))
|
||
|
{
|
||
|
yield return true;
|
||
|
yield return false;
|
||
|
}
|
||
|
else if (type == typeof(int))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
yield return (int)(double)value;
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(uint))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
var d = (double) value;
|
||
|
if (d >= 0.0)
|
||
|
{
|
||
|
yield return (uint) d;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(long))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
yield return (long)(double)value;
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(ulong))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
var d = (double)value;
|
||
|
if (d >= 0.0)
|
||
|
{
|
||
|
yield return (ulong)d;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(short))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
yield return (short)(double)value;
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(ushort))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf|DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
var d = (double)value;
|
||
|
if (d >= 0.0)
|
||
|
{
|
||
|
yield return (ushort)d;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(sbyte))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
var d = (double)value;
|
||
|
yield return (sbyte) d;
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(ushort))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange & ~(DataRange.Inf | DataRange.NaN), typeof(double), seed))
|
||
|
{
|
||
|
var d = (double)value;
|
||
|
if (d >= 0.0)
|
||
|
{
|
||
|
yield return (byte)d;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(float))
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange, typeof(double), seed))
|
||
|
{
|
||
|
var d = (double) value;
|
||
|
if (double.IsNaN(d))
|
||
|
{
|
||
|
yield return float.NaN;
|
||
|
}
|
||
|
else if (double.IsPositiveInfinity(d))
|
||
|
{
|
||
|
yield return float.PositiveInfinity;
|
||
|
}
|
||
|
else if (double.IsNegativeInfinity(d))
|
||
|
{
|
||
|
yield return float.NegativeInfinity;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
yield return (float) (double) value;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (type == typeof(double))
|
||
|
{
|
||
|
if ((dataRange & (DataRange.Minus100To100)) != 0)
|
||
|
{
|
||
|
yield return -100.0;
|
||
|
yield return -77.9;
|
||
|
yield return -50.0;
|
||
|
yield return -36.5;
|
||
|
yield return -9.1;
|
||
|
if ((dataRange & (DataRange.Zero)) != 0)
|
||
|
{
|
||
|
yield return 0.0;
|
||
|
}
|
||
|
yield return 5.1;
|
||
|
yield return 43.5;
|
||
|
yield return 50.0;
|
||
|
yield return 76.8;
|
||
|
yield return 100.0;
|
||
|
|
||
|
if ((dataRange & (DataRange.NaN)) != 0)
|
||
|
{
|
||
|
yield return double.NaN;
|
||
|
}
|
||
|
|
||
|
if ((dataRange & (DataRange.Inf)) != 0)
|
||
|
{
|
||
|
yield return double.PositiveInfinity;
|
||
|
yield return double.NegativeInfinity;
|
||
|
}
|
||
|
}
|
||
|
else if ((dataRange & DataRange.ZeroExclusiveTo100) != 0)
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange | DataRange.Minus100To100, typeof(double), seed))
|
||
|
{
|
||
|
var d = (double)value;
|
||
|
if (double.IsNaN(d) || double.IsInfinity(d))
|
||
|
{
|
||
|
yield return d;
|
||
|
}
|
||
|
else if (d != 0.0)
|
||
|
{
|
||
|
d = d * 0.5 + 50.1;
|
||
|
if (d > 100.0)
|
||
|
{
|
||
|
d = 100.0;
|
||
|
}
|
||
|
yield return d;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ((dataRange & (DataRange.Zero)) != 0)
|
||
|
{
|
||
|
yield return 0.0;
|
||
|
}
|
||
|
}
|
||
|
else if ((dataRange & (DataRange.MinusOneInclusiveToOneInclusive)) != 0)
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange | DataRange.Minus100To100, typeof(double), seed))
|
||
|
{
|
||
|
var d = (double) value;
|
||
|
// Return nan/inf as-is
|
||
|
if (double.IsNaN(d) || double.IsInfinity(d) || d == 0.0)
|
||
|
{
|
||
|
yield return d;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
yield return d / 100.0;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if ((dataRange & (DataRange.ZeroExclusiveToOneInclusive)) != 0)
|
||
|
{
|
||
|
foreach (var value in ExpandRange(dataRange | DataRange.ZeroExclusiveTo100, typeof(double), seed))
|
||
|
{
|
||
|
var d = (double) value;
|
||
|
if (double.IsNaN(d) || double.IsInfinity(d) || d == 0.0)
|
||
|
{
|
||
|
yield return d;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
yield return d / 100.0;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
throw new NotSupportedException($"Invalid datarange `{dataRange}`: missing either Minus100To100 | MinusOneInclusiveToOneInclusive | ZeroExclusiveToOneInclusive`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Namespace == "Unity.Mathematics")
|
||
|
{
|
||
|
if (type.IsByRef)
|
||
|
{
|
||
|
type = type.GetElementType();
|
||
|
}
|
||
|
|
||
|
if (type.Name.StartsWith("bool"))
|
||
|
{
|
||
|
var size = (uint)(type.Name["bool".Length] - '0');
|
||
|
var bools = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(bool), seed).OfType<bool>().ToArray();
|
||
|
var indices = Enumerable.Range(0, bools.Length).ToList();
|
||
|
var originalIndices = new List<int>(indices);
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var y = bools[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new bool2(x, y);
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var y = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var z = bools[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new bool3(x, y, z);
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var y = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var z = bools[NextIndex(random, indices, originalIndices)];
|
||
|
var w = bools[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new bool4(x, y, z, w);
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Name.StartsWith("int"))
|
||
|
{
|
||
|
var size = (uint) (type.Name["int".Length] - '0');
|
||
|
var ints = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(int), seed).OfType<int>().ToArray();
|
||
|
var indices = Enumerable.Range(0, ints.Length).ToList();
|
||
|
var originalIndices = new List<int>(indices);
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = ints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new int2(x, y);
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var z = ints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new int3(x, y, z);
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var z = ints[NextIndex(random, indices, originalIndices)];
|
||
|
var w = ints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new int4(x, y, z, w);
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Name.StartsWith("uint"))
|
||
|
{
|
||
|
var size = (uint)(type.Name["uint".Length] - '0');
|
||
|
var uints = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(uint), seed).OfType<uint>().ToArray();
|
||
|
var indices = Enumerable.Range(0, uints.Length).ToList();
|
||
|
var originalIndices = new List<int>(indices);
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = uints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new uint2(x, y);
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var z = uints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new uint3(x, y, z);
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var y = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var z = uints[NextIndex(random, indices, originalIndices)];
|
||
|
var w = uints[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new uint4(x, y, z, w);
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Name.StartsWith("half"))
|
||
|
{
|
||
|
var size = (uint)(type.Name["half".Length] - '0');
|
||
|
var floats = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(float), seed).OfType<float>().ToList();
|
||
|
var originalIndices = Enumerable.Range(0, floats.Count).ToList();
|
||
|
var indices = new List<int>(originalIndices);
|
||
|
// We only put NaN and Inf in the first set of values
|
||
|
if ((dataRange & DataRange.NaN) != 0)
|
||
|
{
|
||
|
indices.Add(floats.Count);
|
||
|
floats.Add(float.NaN);
|
||
|
}
|
||
|
if ((dataRange & DataRange.Inf) != 0)
|
||
|
{
|
||
|
indices.Add(floats.Count);
|
||
|
floats.Add(float.PositiveInfinity);
|
||
|
}
|
||
|
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new half2(new float2(x, y));
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var z = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new half3(new float3(x, y, z));
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var z = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var w = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new half4(new float4(x, y, z, w));
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Name.StartsWith("float"))
|
||
|
{
|
||
|
var size = (uint) (type.Name["float".Length] - '0');
|
||
|
var floats = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(float), seed).OfType<float>().ToList();
|
||
|
var originalIndices = Enumerable.Range(0, floats.Count).ToList();
|
||
|
var indices = new List<int>(originalIndices);
|
||
|
// We only put NaN and Inf in the first set of values
|
||
|
if ((dataRange & DataRange.NaN) != 0)
|
||
|
{
|
||
|
indices.Add(floats.Count);
|
||
|
floats.Add(float.NaN);
|
||
|
}
|
||
|
if ((dataRange & DataRange.Inf) != 0)
|
||
|
{
|
||
|
indices.Add(floats.Count);
|
||
|
floats.Add(float.PositiveInfinity);
|
||
|
}
|
||
|
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new float2(x, y);
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var z = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new float3(x, y, z);
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var y = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var z = floats[NextIndex(random, indices, originalIndices)];
|
||
|
var w = floats[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new float4(x, y, z, w);
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else if (type.Name.StartsWith("double"))
|
||
|
{
|
||
|
var size = (uint)(type.Name["double".Length] - '0');
|
||
|
var doubles = ExpandRange(dataRange & ~(DataRange.NaN | DataRange.Inf), typeof(double), seed).OfType<double>().ToList();
|
||
|
var originalIndices = Enumerable.Range(0, doubles.Count).ToList();
|
||
|
var indices = new List<int>(originalIndices);
|
||
|
// We only put NaN and Inf in the first set of values
|
||
|
if ((dataRange & DataRange.NaN) != 0)
|
||
|
{
|
||
|
indices.Add(doubles.Count);
|
||
|
doubles.Add(double.NaN);
|
||
|
}
|
||
|
if ((dataRange & DataRange.Inf) != 0)
|
||
|
{
|
||
|
indices.Add(doubles.Count);
|
||
|
doubles.Add(double.PositiveInfinity);
|
||
|
}
|
||
|
|
||
|
var random = new System.Random(seed);
|
||
|
switch (size)
|
||
|
{
|
||
|
case 2:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var y = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new double2(x, y);
|
||
|
}
|
||
|
break;
|
||
|
case 3:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var y = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var z = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new double3(x, y, z);
|
||
|
}
|
||
|
break;
|
||
|
case 4:
|
||
|
for (int i = 0; i < VectorsCount; i++)
|
||
|
{
|
||
|
var x = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var y = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var z = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
var w = doubles[NextIndex(random, indices, originalIndices)];
|
||
|
yield return new double4(x, y, z, w);
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
throw new NotSupportedException($"Unsupported DataRange type `{type}`");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private static int NextIndex(System.Random random, List<int> indices, List<int> originalIndices)
|
||
|
{
|
||
|
var id = random.Next(0, indices.Count - 1);
|
||
|
var index = indices[id];
|
||
|
indices.RemoveAt(id);
|
||
|
if (indices.Count == 0)
|
||
|
{
|
||
|
indices.AddRange(originalIndices);
|
||
|
}
|
||
|
return index;
|
||
|
}
|
||
|
}
|
||
|
}
|