Rasagar/Library/PackageCache/com.unity.render-pipelines.universal/Runtime/Decal/Entities/DecalEntityManager.cs

542 lines
21 KiB
C#
Raw Normal View History

2024-08-26 13:07:20 -07:00
using System;
using System.Collections.Generic;
using Unity.Collections;
using Unity.Mathematics;
using UnityEngine.Assertions;
using UnityEngine.Jobs;
namespace UnityEngine.Rendering.Universal
{
internal class DecalEntityIndexer
{
public struct DecalEntityItem
{
public int chunkIndex;
public int arrayIndex;
public int version;
}
private List<DecalEntityItem> m_Entities = new List<DecalEntityItem>();
private Queue<int> m_FreeIndices = new Queue<int>();
public bool IsValid(DecalEntity decalEntity)
{
if (m_Entities.Count <= decalEntity.index)
return false;
return m_Entities[decalEntity.index].version == decalEntity.version;
}
public DecalEntity CreateDecalEntity(int arrayIndex, int chunkIndex)
{
// Reuse
if (m_FreeIndices.Count != 0)
{
int entityIndex = m_FreeIndices.Dequeue();
int newVersion = m_Entities[entityIndex].version + 1;
m_Entities[entityIndex] = new DecalEntityItem()
{
arrayIndex = arrayIndex,
chunkIndex = chunkIndex,
version = newVersion,
};
return new DecalEntity()
{
index = entityIndex,
version = newVersion,
};
}
// Create new one
{
int entityIndex = m_Entities.Count;
int version = 1;
m_Entities.Add(new DecalEntityItem()
{
arrayIndex = arrayIndex,
chunkIndex = chunkIndex,
version = version,
});
return new DecalEntity()
{
index = entityIndex,
version = version,
};
}
}
public void DestroyDecalEntity(DecalEntity decalEntity)
{
Assert.IsTrue(IsValid(decalEntity));
m_FreeIndices.Enqueue(decalEntity.index);
// Update version that everything that points to it will have outdated version
var item = m_Entities[decalEntity.index];
item.version++;
m_Entities[decalEntity.index] = item;
}
public DecalEntityItem GetItem(DecalEntity decalEntity)
{
Assert.IsTrue(IsValid(decalEntity));
return m_Entities[decalEntity.index];
}
public void UpdateIndex(DecalEntity decalEntity, int newArrayIndex)
{
Assert.IsTrue(IsValid(decalEntity));
var item = m_Entities[decalEntity.index];
item.arrayIndex = newArrayIndex;
item.version = decalEntity.version;
m_Entities[decalEntity.index] = item;
}
public void RemapChunkIndices(List<int> remaper)
{
for (int i = 0; i < m_Entities.Count; ++i)
{
int newChunkIndex = remaper[m_Entities[i].chunkIndex];
var item = m_Entities[i];
item.chunkIndex = newChunkIndex;
m_Entities[i] = item;
}
}
public void Clear()
{
m_Entities.Clear();
m_FreeIndices.Clear();
}
}
internal struct DecalEntity
{
public int index;
public int version;
}
/// <summary>
/// Contains <see cref="DecalEntity"/> and shared material.
/// </summary>
internal class DecalEntityChunk : DecalChunk
{
public Material material;
public NativeArray<DecalEntity> decalEntities;
public DecalProjector[] decalProjectors;
public TransformAccessArray transformAccessArray;
public override void Push()
{
count++;
}
public override void RemoveAtSwapBack(int entityIndex)
{
RemoveAtSwapBack(ref decalEntities, entityIndex, count);
RemoveAtSwapBack(ref decalProjectors, entityIndex, count);
transformAccessArray.RemoveAtSwapBack(entityIndex);
count--;
}
public override void SetCapacity(int newCapacity)
{
decalEntities.ResizeArray(newCapacity);
ResizeNativeArray(ref transformAccessArray, decalProjectors, newCapacity);
ArrayExtensions.ResizeArray(ref decalProjectors, newCapacity);
capacity = newCapacity;
}
public override void Dispose()
{
if (capacity == 0)
return;
decalEntities.Dispose();
transformAccessArray.Dispose();
decalProjectors = null;
count = 0;
capacity = 0;
}
}
/// <summary>
/// Manages lifetime between <see cref="DecalProjector"></see> and <see cref="DecalEntity"/>.
/// Contains all <see cref="DecalChunk"/>.
/// </summary>
internal class DecalEntityManager : IDisposable
{
public List<DecalEntityChunk> entityChunks = new List<DecalEntityChunk>();
public List<DecalCachedChunk> cachedChunks = new List<DecalCachedChunk>();
public List<DecalCulledChunk> culledChunks = new List<DecalCulledChunk>();
public List<DecalDrawCallChunk> drawCallChunks = new List<DecalDrawCallChunk>();
public int chunkCount;
private ProfilingSampler m_AddDecalSampler;
private ProfilingSampler m_ResizeChunks;
private ProfilingSampler m_SortChunks;
private DecalEntityIndexer m_DecalEntityIndexer = new DecalEntityIndexer();
private Dictionary<Material, int> m_MaterialToChunkIndex = new Dictionary<Material, int>();
private struct CombinedChunks
{
public DecalEntityChunk entityChunk;
public DecalCachedChunk cachedChunk;
public DecalCulledChunk culledChunk;
public DecalDrawCallChunk drawCallChunk;
public int previousChunkIndex;
public bool valid;
}
private List<CombinedChunks> m_CombinedChunks = new List<CombinedChunks>();
private List<int> m_CombinedChunkRemmap = new List<int>();
private Material m_ErrorMaterial;
public Material errorMaterial
{
get
{
if (m_ErrorMaterial == null)
m_ErrorMaterial = CoreUtils.CreateEngineMaterial(Shader.Find("Hidden/InternalErrorShader"));
return m_ErrorMaterial;
}
}
private Mesh m_DecalProjectorMesh;
public Mesh decalProjectorMesh
{
get
{
if (m_DecalProjectorMesh == null)
m_DecalProjectorMesh = CoreUtils.CreateCubeMesh(new Vector4(-0.5f, -0.5f, -0.5f, 1.0f), new Vector4(0.5f, 0.5f, 0.5f, 1.0f));
return m_DecalProjectorMesh;
}
}
public DecalEntityManager()
{
m_AddDecalSampler = new ProfilingSampler("DecalEntityManager.CreateDecalEntity");
m_ResizeChunks = new ProfilingSampler("DecalEntityManager.ResizeChunks");
m_SortChunks = new ProfilingSampler("DecalEntityManager.SortChunks");
}
public bool IsValid(DecalEntity decalEntity)
{
return m_DecalEntityIndexer.IsValid(decalEntity);
}
public DecalEntity CreateDecalEntity(DecalProjector decalProjector)
{
var material = decalProjector.material;
if (material == null)
material = errorMaterial;
using (new ProfilingScope(m_AddDecalSampler))
{
int chunkIndex = CreateChunkIndex(material);
int entityIndex = entityChunks[chunkIndex].count;
DecalEntity entity = m_DecalEntityIndexer.CreateDecalEntity(entityIndex, chunkIndex);
DecalEntityChunk entityChunk = entityChunks[chunkIndex];
DecalCachedChunk cachedChunk = cachedChunks[chunkIndex];
DecalCulledChunk culledChunk = culledChunks[chunkIndex];
DecalDrawCallChunk drawCallChunk = drawCallChunks[chunkIndex];
// Make sure we have space to add new entity
if (entityChunks[chunkIndex].capacity == entityChunks[chunkIndex].count)
{
using (new ProfilingScope(m_ResizeChunks))
{
int newCapacity = entityChunks[chunkIndex].capacity + entityChunks[chunkIndex].capacity;
newCapacity = math.max(8, newCapacity);
entityChunk.SetCapacity(newCapacity);
cachedChunk.SetCapacity(newCapacity);
culledChunk.SetCapacity(newCapacity);
drawCallChunk.SetCapacity(newCapacity);
}
}
entityChunk.Push();
cachedChunk.Push();
culledChunk.Push();
drawCallChunk.Push();
entityChunk.decalProjectors[entityIndex] = decalProjector;
entityChunk.decalEntities[entityIndex] = entity;
entityChunk.transformAccessArray.Add(decalProjector.transform);
UpdateDecalEntityData(entity, decalProjector);
return entity;
}
}
private int CreateChunkIndex(Material material)
{
if (!m_MaterialToChunkIndex.TryGetValue(material, out int chunkIndex))
{
var propertyBlock = new MaterialPropertyBlock();
// In order instanced and non instanced rendering to work with _NormalToWorld and _DecalLayerMaskFromDecal
// We need to make sure the array are created with maximum size
propertyBlock.SetMatrixArray("_NormalToWorld", new Matrix4x4[DecalDrawSystem.MaxBatchSize]);
propertyBlock.SetFloatArray("_DecalLayerMaskFromDecal", new float[DecalDrawSystem.MaxBatchSize]);
entityChunks.Add(new DecalEntityChunk() { material = material });
cachedChunks.Add(new DecalCachedChunk()
{
propertyBlock = propertyBlock,
});
culledChunks.Add(new DecalCulledChunk());
drawCallChunks.Add(new DecalDrawCallChunk() { subCallCounts = new NativeArray<int>(1, Allocator.Persistent) });
m_CombinedChunks.Add(new CombinedChunks());
m_CombinedChunkRemmap.Add(0);
m_MaterialToChunkIndex.Add(material, chunkCount);
return chunkCount++;
}
return chunkIndex;
}
public void UpdateAllDecalEntitiesData()
{
foreach (var entityChunk in entityChunks)
{
for (int i = 0; i < entityChunk.count; ++i)
{
var decalProjector = entityChunk.decalProjectors[i];
if (decalProjector == null)
continue;
var entity = entityChunk.decalEntities[i];
if (!IsValid(entity))
continue;
UpdateDecalEntityData(entity, decalProjector);
}
}
}
public void UpdateDecalEntityData(DecalEntity decalEntity, DecalProjector decalProjector)
{
var decalItem = m_DecalEntityIndexer.GetItem(decalEntity);
int chunkIndex = decalItem.chunkIndex;
int arrayIndex = decalItem.arrayIndex;
DecalCachedChunk cachedChunk = cachedChunks[chunkIndex];
cachedChunk.sizeOffsets[arrayIndex] = Matrix4x4.Translate(decalProjector.decalOffset) * Matrix4x4.Scale(decalProjector.decalSize);
float drawDistance = decalProjector.drawDistance;
float fadeScale = decalProjector.fadeScale;
float startAngleFade = decalProjector.startAngleFade;
float endAngleFade = decalProjector.endAngleFade;
Vector4 uvScaleBias = decalProjector.uvScaleBias;
int layerMask = decalProjector.gameObject.layer;
ulong sceneLayerMask = decalProjector.gameObject.sceneCullingMask;
float fadeFactor = decalProjector.fadeFactor;
cachedChunk.drawDistances[arrayIndex] = new Vector2(drawDistance, fadeScale);
// In the shader to remap from cosine -1 to 1 to new range 0..1 (with 0 - 0 degree and 1 - 180 degree)
// we do 1.0 - (dot() * 0.5 + 0.5) => 0.5 * (1 - dot())
// we actually square that to get smoother result => x = (0.5 - 0.5 * dot())^2
// Do a remap in the shader. 1.0 - saturate((x - start) / (end - start))
// After simplification => saturate(a + b * dot() * (dot() - 2.0))
// a = 1.0 - (0.25 - start) / (end - start), y = - 0.25 / (end - start)
if (startAngleFade == 180.0f) // angle fade is disabled
{
cachedChunk.angleFades[arrayIndex] = new Vector2(0.0f, 0.0f);
}
else
{
float angleStart = startAngleFade / 180.0f;
float angleEnd = endAngleFade / 180.0f;
var range = Mathf.Max(0.0001f, angleEnd - angleStart);
cachedChunk.angleFades[arrayIndex] = new Vector2(1.0f - (0.25f - angleStart) / range, -0.25f / range);
}
cachedChunk.uvScaleBias[arrayIndex] = uvScaleBias;
cachedChunk.layerMasks[arrayIndex] = layerMask;
cachedChunk.sceneLayerMasks[arrayIndex] = sceneLayerMask;
cachedChunk.fadeFactors[arrayIndex] = fadeFactor;
cachedChunk.scaleModes[arrayIndex] = decalProjector.scaleMode;
cachedChunk.renderingLayerMasks[arrayIndex] = RenderingLayerUtils.ToValidRenderingLayers(decalProjector.renderingLayerMask);
cachedChunk.positions[arrayIndex] = decalProjector.transform.position;
cachedChunk.rotation[arrayIndex] = decalProjector.transform.rotation;
cachedChunk.scales[arrayIndex] = decalProjector.transform.lossyScale;
cachedChunk.dirty[arrayIndex] = true;
}
public void DestroyDecalEntity(DecalEntity decalEntity)
{
if (!m_DecalEntityIndexer.IsValid(decalEntity))
return;
var decalItem = m_DecalEntityIndexer.GetItem(decalEntity);
m_DecalEntityIndexer.DestroyDecalEntity(decalEntity);
int chunkIndex = decalItem.chunkIndex;
int arrayIndex = decalItem.arrayIndex;
DecalEntityChunk entityChunk = entityChunks[chunkIndex];
DecalCachedChunk cachedChunk = cachedChunks[chunkIndex];
DecalCulledChunk culledChunk = culledChunks[chunkIndex];
DecalDrawCallChunk drawCallChunk = drawCallChunks[chunkIndex];
int lastArrayIndex = entityChunk.count - 1;
if (arrayIndex != lastArrayIndex)
m_DecalEntityIndexer.UpdateIndex(entityChunk.decalEntities[lastArrayIndex], arrayIndex);
entityChunk.RemoveAtSwapBack(arrayIndex);
cachedChunk.RemoveAtSwapBack(arrayIndex);
culledChunk.RemoveAtSwapBack(arrayIndex);
drawCallChunk.RemoveAtSwapBack(arrayIndex);
}
public void Update()
{
using (new ProfilingScope(m_SortChunks))
{
for (int i = 0; i < chunkCount; ++i)
{
if (entityChunks[i].material == null)
entityChunks[i].material = errorMaterial;
}
// Combine chunks into single array
for (int i = 0; i < chunkCount; ++i)
{
m_CombinedChunks[i] = new CombinedChunks()
{
entityChunk = entityChunks[i],
cachedChunk = cachedChunks[i],
culledChunk = culledChunks[i],
drawCallChunk = drawCallChunks[i],
previousChunkIndex = i,
valid = entityChunks[i].count != 0,
};
}
// Sort
m_CombinedChunks.Sort((a, b) =>
{
if (a.valid && !b.valid)
return -1;
if (!a.valid && b.valid)
return 1;
if (a.cachedChunk.drawOrder < b.cachedChunk.drawOrder)
return -1;
if (a.cachedChunk.drawOrder > b.cachedChunk.drawOrder)
return 1;
return a.entityChunk.material.GetHashCode().CompareTo(b.entityChunk.material.GetHashCode());
});
// Early out if nothing changed
bool dirty = false;
for (int i = 0; i < chunkCount; ++i)
{
if (m_CombinedChunks[i].previousChunkIndex != i || !m_CombinedChunks[i].valid)
{
dirty = true;
break;
}
}
if (!dirty)
return;
// Update chunks
int count = 0;
m_MaterialToChunkIndex.Clear();
for (int i = 0; i < chunkCount; ++i)
{
var combinedChunk = m_CombinedChunks[i];
// Destroy invalid chunk for cleanup
if (!m_CombinedChunks[i].valid)
{
combinedChunk.entityChunk.currentJobHandle.Complete();
combinedChunk.cachedChunk.currentJobHandle.Complete();
combinedChunk.culledChunk.currentJobHandle.Complete();
combinedChunk.drawCallChunk.currentJobHandle.Complete();
combinedChunk.entityChunk.Dispose();
combinedChunk.cachedChunk.Dispose();
combinedChunk.culledChunk.Dispose();
combinedChunk.drawCallChunk.Dispose();
continue;
}
entityChunks[i] = combinedChunk.entityChunk;
cachedChunks[i] = combinedChunk.cachedChunk;
culledChunks[i] = combinedChunk.culledChunk;
drawCallChunks[i] = combinedChunk.drawCallChunk;
if (!m_MaterialToChunkIndex.ContainsKey(entityChunks[i].material))
m_MaterialToChunkIndex.Add(entityChunks[i].material, i);
m_CombinedChunkRemmap[combinedChunk.previousChunkIndex] = i;
count++;
}
// In case some chunks where destroyed resize the arrays
if (chunkCount > count)
{
entityChunks.RemoveRange(count, chunkCount - count);
cachedChunks.RemoveRange(count, chunkCount - count);
culledChunks.RemoveRange(count, chunkCount - count);
drawCallChunks.RemoveRange(count, chunkCount - count);
m_CombinedChunks.RemoveRange(count, chunkCount - count);
chunkCount = count;
}
// Remap entities chunk index with new sorted ones
m_DecalEntityIndexer.RemapChunkIndices(m_CombinedChunkRemmap);
}
}
public void Dispose()
{
CoreUtils.Destroy(m_ErrorMaterial);
CoreUtils.Destroy(m_DecalProjectorMesh);
foreach (var entityChunk in entityChunks)
entityChunk.currentJobHandle.Complete();
foreach (var cachedChunk in cachedChunks)
cachedChunk.currentJobHandle.Complete();
foreach (var culledChunk in culledChunks)
culledChunk.currentJobHandle.Complete();
foreach (var drawCallChunk in drawCallChunks)
drawCallChunk.currentJobHandle.Complete();
foreach (var entityChunk in entityChunks)
entityChunk.Dispose();
foreach (var cachedChunk in cachedChunks)
cachedChunk.Dispose();
foreach (var culledChunk in culledChunks)
culledChunk.Dispose();
foreach (var drawCallChunk in drawCallChunks)
drawCallChunk.Dispose();
m_DecalEntityIndexer.Clear();
m_MaterialToChunkIndex.Clear();
entityChunks.Clear();
cachedChunks.Clear();
culledChunks.Clear();
drawCallChunks.Clear();
m_CombinedChunks.Clear();
chunkCount = 0;
}
}
}