internal void RemovePendingAsset(ProbeVolumeAsset asset)
{
var key = asset.GetSerializedFullPath();
if (m_ActiveAssets.ContainsKey(key))
{
m_ActiveAssets.Remove(key);
}
// Remove bricks and empty cells
foreach (var cell in asset.cells)
{
if (cells.ContainsKey(cell.index))
{
cells.Remove(cell.index);
}
}
// Unload brick data
if (m_AssetPathToBricks.ContainsKey(key))
{
var regIds = m_AssetPathToBricks[key];
foreach (var regId in regIds)
{
ReleaseBricks(regId);
}
m_AssetPathToBricks.Remove(key);
}
}
private void OnValidate()
{
if (!enabled || !gameObject.activeSelf)
{
return;
}
if (ShaderConfig.s_EnableProbeVolumes == 1)
{
if (m_Profile != null)
{
bool hasIndexDimensionChangedOnProfileSwitch = m_PrevProfile == null || (m_PrevProfile != null && m_PrevProfile.indexDimensions != m_Profile.indexDimensions);
if (hasIndexDimensionChangedOnProfileSwitch)
{
var refVol = ProbeReferenceVolume.instance;
refVol.AddPendingIndexDimensionChange(indexDimensions);
}
m_PrevProfile = m_Profile;
QueueAssetLoading();
}
if (volumeAsset != m_PrevAsset && m_PrevAsset != null)
{
ProbeReferenceVolume.instance.AddPendingAssetRemoval(m_PrevAsset);
}
m_PrevAsset = volumeAsset;
}
}
internal void OnProbesBakeCompleted()
{
if (this.gameObject == null || !this.gameObject.activeInHierarchy)
{
return;
}
int numProbes = parameters.resolutionX * parameters.resolutionY * parameters.resolutionZ;
var sh = new NativeArray <SphericalHarmonicsL2>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var validity = new NativeArray <float>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
// TODO: Currently, we need to always allocate and pass this octahedralDepth array into GetAdditionalBakedProbes().
// In the future, we should add an API call for GetAdditionalBakedProbes() without octahedralDepth required.
var octahedralDepth = new NativeArray <float>(numProbes * 8 * 8, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
if (UnityEditor.Experimental.Lightmapping.GetAdditionalBakedProbes(GetID(), sh, validity, octahedralDepth))
{
if (!probeVolumeAsset || GetID() != probeVolumeAsset.instanceID)
{
probeVolumeAsset = ProbeVolumeAsset.CreateAsset(GetID());
}
probeVolumeAsset.instanceID = GetID();
probeVolumeAsset.resolutionX = parameters.resolutionX;
probeVolumeAsset.resolutionY = parameters.resolutionY;
probeVolumeAsset.resolutionZ = parameters.resolutionZ;
ProbeVolumePayload.Ensure(ref probeVolumeAsset.payload, numProbes);
// Always serialize L0, L1 and L2 coefficients, even if atlas is configured to only store L1.
// In the future we will strip the L2 coefficients from the project at build time if L2 is never used.
for (int i = 0, iLen = sh.Length; i < iLen; ++i)
{
ProbeVolumePayload.SetSphericalHarmonicsL2FromIndex(ref probeVolumeAsset.payload, sh[i], i);
}
validity.CopyTo(probeVolumeAsset.payload.dataValidity);
if (ShaderConfig.s_ProbeVolumesBilateralFilteringMode == ProbeVolumesBilateralFilteringModes.OctahedralDepth)
{
octahedralDepth.CopyTo(probeVolumeAsset.payload.dataOctahedralDepth);
}
if (UnityEditor.Lightmapping.giWorkflowMode != UnityEditor.Lightmapping.GIWorkflowMode.Iterative)
{
UnityEditor.EditorUtility.SetDirty(probeVolumeAsset);
}
UnityEditor.AssetDatabase.Refresh();
dataUpdated = true;
}
sh.Dispose();
validity.Dispose();
octahedralDepth.Dispose();
}
internal void AddPendingAssetRemoval(ProbeVolumeAsset asset)
{
var key = asset.GetSerializedFullPath();
if (m_PendingAssetsToBeUnloaded.ContainsKey(key))
{
m_PendingAssetsToBeUnloaded.Remove(key);
}
m_PendingAssetsToBeUnloaded.Add(asset.GetSerializedFullPath(), asset);
}
internal void AddPendingAssetLoading(ProbeVolumeAsset asset)
{
var key = asset.GetSerializedFullPath();
if (m_PendingAssetsToBeLoaded.ContainsKey(key))
{
m_PendingAssetsToBeLoaded.Remove(key);
}
m_PendingAssetsToBeLoaded.Add(asset.GetSerializedFullPath(), asset);
m_NeedLoadAsset = true;
}
private void LoadAsset(ProbeVolumeAsset asset)
{
var path = asset.GetSerializedFullPath();
m_AssetPathToBricks[path] = new List <RegId>();
foreach (var cell in asset.cells)
{
// Push data to HDRP
bool compressed = false;
var dataLocation = ProbeBrickPool.CreateDataLocation(cell.sh.Length, compressed);
ProbeBrickPool.FillDataLocation(ref dataLocation, cell.sh);
// TODO register ID of brick list
List <ProbeBrickIndex.Brick> brickList = new List <ProbeBrickIndex.Brick>();
brickList.AddRange(cell.bricks);
var regId = AddBricks(brickList, dataLocation);
cells[cell.index] = cell;
m_AssetPathToBricks[path].Add(regId);
}
}
private static void OnAdditionalProbesBakeCompleted()
{
UnityEditor.Experimental.Lightmapping.additionalBakedProbesCompleted -= OnAdditionalProbesBakeCompleted;
var numCells = bakingCells.Count;
// Fetch results of all cells
for (int c = 0; c < numCells; ++c)
{
var cell = bakingCells[c];
if (cell.probePositions == null)
{
continue;
}
int numProbes = cell.probePositions.Length;
Debug.Assert(numProbes > 0);
var sh = new NativeArray <SphericalHarmonicsL2>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var validity = new NativeArray <float>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var bakedProbeOctahedralDepth = new NativeArray <float>(numProbes * 64, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
UnityEditor.Experimental.Lightmapping.GetAdditionalBakedProbes(cell.index, sh, validity, bakedProbeOctahedralDepth);
cell.sh = new SphericalHarmonicsL1[numProbes];
cell.validity = new float[numProbes];
for (int i = 0; i < numProbes; ++i)
{
Vector4[] channels = new Vector4[3];
// compare to SphericalHarmonicsL2::GetShaderConstantsFromNormalizedSH
channels[0] = new Vector4(sh[i][0, 3], sh[i][0, 1], sh[i][0, 2], sh[i][0, 0]);
channels[1] = new Vector4(sh[i][1, 3], sh[i][1, 1], sh[i][1, 2], sh[i][1, 0]);
channels[2] = new Vector4(sh[i][2, 3], sh[i][2, 1], sh[i][2, 2], sh[i][2, 0]);
// It can be shown that |L1_i| <= |2*L0|
// Precomputed Global Illumination in Frostbite by Yuriy O'Donnell.
// https://media.contentapi.ea.com/content/dam/eacom/frostbite/files/gdc2018-precomputedgiobalilluminationinfrostbite.pdf
//
// So divide by L0 brings us to [-2, 2],
// divide by 4 brings us to [-0.5, 0.5],
// and plus by 0.5 brings us to [0, 1].
for (int channel = 0; channel < 3; ++channel)
{
var l0 = channels[channel][3];
if (l0 != 0.0f)
{
for (int axis = 0; axis < 3; ++axis)
{
channels[channel][axis] = channels[channel][axis] / (l0 * 4.0f) + 0.5f;
Debug.Assert(channels[channel][axis] >= 0.0f && channels[channel][axis] <= 1.0f);
}
}
}
SphericalHarmonicsL1 sh1 = new SphericalHarmonicsL1();
sh1.shAr = channels[0];
sh1.shAg = channels[1];
sh1.shAb = channels[2];
cell.sh[i] = sh1;
cell.validity[i] = validity[i];
}
// Reset index
UnityEditor.Experimental.Lightmapping.SetAdditionalBakedProbes(cell.index, null);
DilateInvalidProbes(cell.probePositions, cell.bricks, cell.sh, cell.validity, bakingReferenceVolumeAuthoring.GetDilationSettings());
ProbeReferenceVolume.instance.cells[cell.index] = cell;
}
// Map from each scene to an existing reference volume
var scene2RefVol = new Dictionary <Scene, ProbeReferenceVolumeAuthoring>();
foreach (var refVol in GameObject.FindObjectsOfType <ProbeReferenceVolumeAuthoring>())
{
if (refVol.enabled)
{
scene2RefVol[refVol.gameObject.scene] = refVol;
}
}
// Map from each reference volume to its asset
var refVol2Asset = new Dictionary <ProbeReferenceVolumeAuthoring, ProbeVolumeAsset>();
foreach (var refVol in scene2RefVol.Values)
{
refVol2Asset[refVol] = ProbeVolumeAsset.CreateAsset(refVol.gameObject.scene);
}
// Put cells into the respective assets
foreach (var cell in ProbeReferenceVolume.instance.cells.Values)
{
foreach (var scene in cellIndex2SceneReferences[cell.index])
{
// This scene has a reference volume authoring component in it?
ProbeReferenceVolumeAuthoring refVol = null;
if (scene2RefVol.TryGetValue(scene, out refVol))
{
var asset = refVol2Asset[refVol];
asset.cells.Add(cell);
}
}
}
// Connect the assets to their components
foreach (var pair in refVol2Asset)
{
var refVol = pair.Key;
var asset = pair.Value;
refVol.volumeAsset = asset;
if (UnityEditor.Lightmapping.giWorkflowMode != UnityEditor.Lightmapping.GIWorkflowMode.Iterative)
{
UnityEditor.EditorUtility.SetDirty(refVol);
UnityEditor.EditorUtility.SetDirty(refVol.volumeAsset);
}
}
UnityEditor.AssetDatabase.SaveAssets();
UnityEditor.AssetDatabase.Refresh();
foreach (var refVol in refVol2Asset.Keys)
{
if (refVol.enabled && refVol.gameObject.activeSelf)
{
refVol.QueueAssetLoading();
}
}
}
internal void OnLightingDataCleared()
{
probeVolumeAsset = null;
dataUpdated = true;
BakeKeyClear();
}
internal void OnProbesBakeCompleted()
{
if (this.gameObject == null || !this.gameObject.activeInHierarchy)
{
return;
}
int numProbes = parameters.resolutionX * parameters.resolutionY * parameters.resolutionZ;
SphericalHarmonicsL1[] data = new SphericalHarmonicsL1[numProbes];
float[] dataValidity = new float[numProbes];
float[] dataOctahedralDepth = new float[numProbes * 8 * 8];
var sh = new NativeArray <SphericalHarmonicsL2>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var validity = new NativeArray <float>(numProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var octahedralDepth = new NativeArray <float>(numProbes * 8 * 8, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
if (UnityEditor.Experimental.Lightmapping.GetAdditionalBakedProbes(GetID(), sh, validity, octahedralDepth))
{
// TODO: Remove this data copy.
for (int i = 0, iLen = data.Length; i < iLen; ++i)
{
data[i].shAr = new Vector4(sh[i][0, 3], sh[i][0, 1], sh[i][0, 2], sh[i][0, 0] - sh[i][0, 6]);
data[i].shAg = new Vector4(sh[i][1, 3], sh[i][1, 1], sh[i][1, 2], sh[i][1, 0] - sh[i][1, 6]);
data[i].shAb = new Vector4(sh[i][2, 3], sh[i][2, 1], sh[i][2, 2], sh[i][2, 0] - sh[i][2, 6]);
dataValidity[i] = validity[i];
for (int j = 0; j < 64; ++j)
{
dataOctahedralDepth[i * 64 + j] = octahedralDepth[i * 64 + j];
}
}
if (!probeVolumeAsset || GetID() != probeVolumeAsset.instanceID)
{
probeVolumeAsset = ProbeVolumeAsset.CreateAsset(GetID());
}
probeVolumeAsset.instanceID = GetID();
probeVolumeAsset.dataSH = data;
probeVolumeAsset.dataValidity = dataValidity;
probeVolumeAsset.dataOctahedralDepth = dataOctahedralDepth;
probeVolumeAsset.resolutionX = parameters.resolutionX;
probeVolumeAsset.resolutionY = parameters.resolutionY;
probeVolumeAsset.resolutionZ = parameters.resolutionZ;
if (UnityEditor.Lightmapping.giWorkflowMode != UnityEditor.Lightmapping.GIWorkflowMode.Iterative)
{
UnityEditor.EditorUtility.SetDirty(probeVolumeAsset);
}
UnityEditor.AssetDatabase.Refresh();
dataUpdated = true;
}
sh.Dispose();
validity.Dispose();
octahedralDepth.Dispose();
}
