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);
}
}
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;
// Compute the max index dimension from all the loaded assets + assets we need to load
Vector3Int indexDimension = Vector3Int.zero;
foreach (var a in m_PendingAssetsToBeLoaded.Values)
{
indexDimension = Vector3Int.Max(indexDimension, a.maxCellIndex);
}
foreach (var a in m_ActiveAssets.Values)
{
indexDimension = Vector3Int.Max(indexDimension, a.maxCellIndex);
}
m_PendingIndexDimChange = indexDimension;
m_NeedsIndexDimChange = true;
}
internal void AddPendingAssetRemoval(ProbeVolumeAsset asset)
{
var key = asset.GetSerializedFullPath();
if (m_PendingAssetsToBeUnloaded.ContainsKey(key))
{
m_PendingAssetsToBeUnloaded.Remove(key);
}
m_PendingAssetsToBeUnloaded.Add(asset.GetSerializedFullPath(), asset);
}
public static ProbeVolumeAsset CreateAsset(ProbeVolumePerSceneData data)
{
ProbeVolumeAsset asset = CreateInstance <ProbeVolumeAsset>();
if (data.asset != null)
{
asset.m_AssetFullPath = UnityEditor.AssetDatabase.GetAssetPath(data.asset);
}
if (string.IsNullOrEmpty(asset.m_AssetFullPath))
{
asset.m_AssetFullPath = GetPath(data.gameObject.scene);
}
UnityEditor.AssetDatabase.CreateAsset(asset, asset.m_AssetFullPath);
return(asset);
}
internal void QueueAssetRemoval()
{
if (volumeAsset == null)
{
return;
}
#if UNITY_EDITOR
brickGizmos?.Dispose();
brickGizmos = null;
cellGizmo?.Dispose();
cellGizmo = null;
m_PrevAsset = null;
#endif
ProbeReferenceVolume.instance.AddPendingAssetRemoval(volumeAsset);
}
internal void QueueAssetRemoval()
{
if (volumeAsset == null)
{
return;
}
#if UNITY_EDITOR
foreach (var meshGizmo in brickGizmos.Values)
{
meshGizmo.Dispose();
}
brickGizmos.Clear();
cellGizmo?.Dispose();
m_PrevAsset = null;
#endif
ProbeReferenceVolume.instance.AddPendingAssetRemoval(volumeAsset);
}
void OnValidate()
{
if (!enabled || !gameObject.activeSelf)
{
return;
}
if (m_Profile != null)
{
m_PrevProfile = m_Profile;
}
if (volumeAsset != m_PrevAsset && m_PrevAsset != null)
{
ProbeReferenceVolume.instance.AddPendingAssetRemoval(m_PrevAsset);
}
if (volumeAsset != m_PrevAsset)
{
QueueAssetLoading();
}
m_PrevAsset = volumeAsset;
}
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, ProbeVolumeSHBands.SphericalHarmonicsL2);
ProbeBrickPool.FillDataLocation(ref dataLocation, cell.sh, ProbeVolumeSHBands.SphericalHarmonicsL2);
// 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);
dataLocation.Cleanup();
}
}
internal bool CompatibleWith(ProbeVolumeAsset otherAsset)
{
return((maxSubdivision == otherAsset.maxSubdivision) && (minBrickSize == otherAsset.minBrickSize) && (cellSizeInBricks == otherAsset.cellSizeInBricks));
}
static void OnAdditionalProbesBakeCompleted()
{
UnityEditor.Experimental.Lightmapping.additionalBakedProbesCompleted -= OnAdditionalProbesBakeCompleted;
var bakingCells = m_BakingBatch.cells;
var numCells = bakingCells.Count;
int numUniqueProbes = m_BakingBatch.uniqueProbeCount;
var sh = new NativeArray <SphericalHarmonicsL2>(numUniqueProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var validity = new NativeArray <float>(numUniqueProbes, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
var bakedProbeOctahedralDepth = new NativeArray <float>(numUniqueProbes * 64, Allocator.Temp, NativeArrayOptions.UninitializedMemory);
UnityEditor.Experimental.Lightmapping.GetAdditionalBakedProbes(m_BakingBatch.index, sh, validity, bakedProbeOctahedralDepth);
// Fetch results of all cells
for (int c = 0; c < numCells; ++c)
{
var cell = bakingCells[c].cell;
if (cell.probePositions == null)
{
continue;
}
int numProbes = cell.probePositions.Length;
Debug.Assert(numProbes > 0);
cell.sh = new SphericalHarmonicsL2[numProbes];
cell.validity = new float[numProbes];
for (int i = 0; i < numProbes; ++i)
{
int j = bakingCells[c].probeIndices[i];
SphericalHarmonicsL2 shv = sh[j];
// Compress the range of all coefficients but the DC component to [0..1]
// Upper bounds taken from http://ppsloan.org/publications/Sig20_Advances.pptx
// Divide each coefficient by DC*f to get to [-1,1] where f is from slide 33
for (int rgb = 0; rgb < 3; ++rgb)
{
var l0 = sh[j][rgb, 0];
if (l0 == 0.0f)
{
continue;
}
// TODO: We're working on irradiance instead of radiance coefficients
// Add safety margin 2 to avoid out-of-bounds values
float l1scale = 2.0f; // Should be: 3/(2*sqrt(3)) * 2, but rounding to 2 to issues we are observing.
float l2scale = 3.5777088f; // 4/sqrt(5) * 2
// L_1^m
shv[rgb, 1] = sh[j][rgb, 1] / (l0 * l1scale * 2.0f) + 0.5f;
shv[rgb, 2] = sh[j][rgb, 2] / (l0 * l1scale * 2.0f) + 0.5f;
shv[rgb, 3] = sh[j][rgb, 3] / (l0 * l1scale * 2.0f) + 0.5f;
// L_2^-2
shv[rgb, 4] = sh[j][rgb, 4] / (l0 * l2scale * 2.0f) + 0.5f;
shv[rgb, 5] = sh[j][rgb, 5] / (l0 * l2scale * 2.0f) + 0.5f;
shv[rgb, 6] = sh[j][rgb, 6] / (l0 * l2scale * 2.0f) + 0.5f;
shv[rgb, 7] = sh[j][rgb, 7] / (l0 * l2scale * 2.0f) + 0.5f;
shv[rgb, 8] = sh[j][rgb, 8] / (l0 * l2scale * 2.0f) + 0.5f;
for (int coeff = 1; coeff < 9; ++coeff)
{
Debug.Assert(shv[rgb, coeff] >= 0.0f && shv[rgb, coeff] <= 1.0f);
}
}
SphericalHarmonicsL2Utils.SetL0(ref cell.sh[i], new Vector3(shv[0, 0], shv[1, 0], shv[2, 0]));
SphericalHarmonicsL2Utils.SetL1R(ref cell.sh[i], new Vector3(shv[0, 3], shv[0, 1], shv[0, 2]));
SphericalHarmonicsL2Utils.SetL1G(ref cell.sh[i], new Vector3(shv[1, 3], shv[1, 1], shv[1, 2]));
SphericalHarmonicsL2Utils.SetL1B(ref cell.sh[i], new Vector3(shv[2, 3], shv[2, 1], shv[2, 2]));
SphericalHarmonicsL2Utils.SetCoefficient(ref cell.sh[i], 4, new Vector3(shv[0, 4], shv[1, 4], shv[2, 4]));
SphericalHarmonicsL2Utils.SetCoefficient(ref cell.sh[i], 5, new Vector3(shv[0, 5], shv[1, 5], shv[2, 5]));
SphericalHarmonicsL2Utils.SetCoefficient(ref cell.sh[i], 6, new Vector3(shv[0, 6], shv[1, 6], shv[2, 6]));
SphericalHarmonicsL2Utils.SetCoefficient(ref cell.sh[i], 7, new Vector3(shv[0, 7], shv[1, 7], shv[2, 7]));
SphericalHarmonicsL2Utils.SetCoefficient(ref cell.sh[i], 8, new Vector3(shv[0, 8], shv[1, 8], shv[2, 8]));
cell.validity[i] = validity[j];
}
DilateInvalidProbes(cell.probePositions, cell.bricks, cell.sh, cell.validity, m_BakingReferenceVolumeAuthoring.GetDilationSettings());
ProbeReferenceVolume.instance.cells[cell.index] = cell;
}
m_BakingBatchIndex = 0;
// Reset index
UnityEditor.Experimental.Lightmapping.SetAdditionalBakedProbes(m_BakingBatch.index, null);
// 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 m_BakingBatch.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);
if (hasFoundBounds)
{
Vector3Int cellsInDir;
int cellSizeInMeters = Mathf.CeilToInt((float)refVol.profile.cellSizeInBricks * refVol.profile.brickSize);
CellCountInDirections(out cellsInDir, cellSizeInMeters);
asset.maxCellIndex.x = cellsInDir.x * (int)refVol.profile.cellSizeInBricks;
asset.maxCellIndex.y = cellsInDir.y * (int)refVol.profile.cellSizeInBricks;
asset.maxCellIndex.z = cellsInDir.z * (int)refVol.profile.cellSizeInBricks;
}
else
{
foreach (var p in cell.probePositions)
{
float x = Mathf.Abs((float)p.x + refVol.transform.position.x) / refVol.profile.brickSize;
float y = Mathf.Abs((float)p.y + refVol.transform.position.y) / refVol.profile.brickSize;
float z = Mathf.Abs((float)p.z + refVol.transform.position.z) / refVol.profile.brickSize;
asset.maxCellIndex.x = Mathf.Max(asset.maxCellIndex.x, (int)(x * 2));
asset.maxCellIndex.y = Mathf.Max(asset.maxCellIndex.y, (int)(y * 2));
asset.maxCellIndex.z = Mathf.Max(asset.maxCellIndex.z, (int)(z * 2));
}
}
}
}
}
// 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);
}
}
var probeVolumes = GameObject.FindObjectsOfType <ProbeVolume>();
foreach (var probeVolume in probeVolumes)
{
probeVolume.OnBakeCompleted();
}
UnityEditor.AssetDatabase.SaveAssets();
UnityEditor.AssetDatabase.Refresh();
ProbeReferenceVolume.instance.clearAssetsOnVolumeClear = false;
foreach (var refVol in refVol2Asset.Keys)
{
if (refVol.enabled && refVol.gameObject.activeSelf)
{
refVol.QueueAssetLoading();
}
}
}
