Rasagar/Library/PackageCache/com.unity.render-pipelines.universal/Runtime/Tiling/TileRangeExpansionJob.cs

64 lines
2.3 KiB
C#
Raw Normal View History

2024-08-26 13:07:20 -07:00
using Unity.Burst;
using Unity.Collections;
using Unity.Jobs;
using Unity.Mathematics;
namespace UnityEngine.Rendering.Universal
{
[BurstCompile(FloatMode = FloatMode.Fast, DisableSafetyChecks = true, OptimizeFor = OptimizeFor.Performance)]
struct TileRangeExpansionJob : IJobFor
{
[ReadOnly]
public NativeArray<InclusiveRange> tileRanges;
[NativeDisableParallelForRestriction]
public NativeArray<uint> tileMasks;
public int rangesPerItem;
public int itemsPerTile;
public int wordsPerTile;
public int2 tileResolution;
public void Execute(int jobIndex)
{
var rowIndex = jobIndex % tileResolution.y;
var viewIndex = jobIndex / tileResolution.y;
var compactCount = 0;
var itemIndices = new NativeArray<short>(itemsPerTile, Allocator.Temp);
var itemRanges = new NativeArray<InclusiveRange>(itemsPerTile, Allocator.Temp);
// Compact the light ranges for the current row.
for (var itemIndex = 0; itemIndex < itemsPerTile; itemIndex++)
{
var range = tileRanges[viewIndex * rangesPerItem * itemsPerTile + itemIndex * rangesPerItem + 1 + rowIndex];
if (!range.isEmpty)
{
itemIndices[compactCount] = (short)itemIndex;
itemRanges[compactCount] = range;
compactCount++;
}
}
var rowBaseMaskIndex = viewIndex * wordsPerTile * tileResolution.x * tileResolution.y + rowIndex * wordsPerTile * tileResolution.x;
for (var tileIndex = 0; tileIndex < tileResolution.x; tileIndex++)
{
var tileBaseIndex = rowBaseMaskIndex + tileIndex * wordsPerTile;
for (var i = 0; i < compactCount; i++)
{
var itemIndex = (int)itemIndices[i];
var wordIndex = itemIndex / 32;
var itemMask = 1u << (itemIndex % 32);
var range = itemRanges[i];
if (range.Contains((short)tileIndex))
{
tileMasks[tileBaseIndex + wordIndex] |= itemMask;
}
}
}
itemIndices.Dispose();
itemRanges.Dispose();
}
}
}