public DataForDilation(ProbeReferenceVolume.Cell cell, float defaultThreshold)
{
this.cell = cell;
int probeCount = cell.probePositions.Length;
positionBuffer = new ComputeBuffer(probeCount, System.Runtime.InteropServices.Marshal.SizeOf <Vector3>());
outputProbes = new ComputeBuffer(probeCount, System.Runtime.InteropServices.Marshal.SizeOf <DilatedProbe>());
needDilatingBuffer = new ComputeBuffer(probeCount, sizeof(int));
// Init with pre-dilated SH so we don't need to re-fill from sampled data from texture (that might be less precise).
dilatedProbes = new DilatedProbe[probeCount];
int[] needDilating = new int[probeCount];
for (int i = 0; i < probeCount; ++i)
{
dilatedProbes[i].FromSphericalHarmonicsShaderConstants(cell, i);
needDilating[i] = s_CustomDilationThresh.ContainsKey(i) ?
(cell.validity[i] > s_CustomDilationThresh[i] ? 1 : 0) : (cell.validity[i] > defaultThreshold ? 1 : 0);
}
outputProbes.SetData(dilatedProbes);
positionBuffer.SetData(cell.probePositions);
needDilatingBuffer.SetData(needDilating);
}
internal void ToSphericalHarmonicsShaderConstants(ProbeReferenceVolume.Cell cell, int probeIdx)
{
var sh = new SphericalHarmonicsL2();
ToSphericalHarmonicsL2(ref sh);
WriteToShaderCoeffsL0L1(sh, cell.bakingScenario.shL0L1RxData, cell.bakingScenario.shL1GL1RyData, cell.bakingScenario.shL1BL1RzData, probeIdx * 4);
if (cell.shBands == ProbeVolumeSHBands.SphericalHarmonicsL2)
{
WriteToShaderCoeffsL2(sh, cell.bakingScenario.shL2Data_0, cell.bakingScenario.shL2Data_1, cell.bakingScenario.shL2Data_2, cell.bakingScenario.shL2Data_3, probeIdx * 4);
}
}
static void PerformDilation(ProbeReferenceVolume.Cell cell, ProbeDilationSettings settings)
{
InitDilationShaders();
DataForDilation data = new DataForDilation(cell, settings.dilationValidityThreshold);
var cmd = CommandBufferPool.Get("Cell Dilation");
cmd.SetComputeBufferParam(dilationShader, dilationKernel, _ProbePositionsBuffer, data.positionBuffer);
cmd.SetComputeBufferParam(dilationShader, dilationKernel, _OutputProbes, data.outputProbes);
cmd.SetComputeBufferParam(dilationShader, dilationKernel, _NeedDilating, data.needDilatingBuffer);
int probeCount = cell.probePositions.Length;
cmd.SetComputeVectorParam(dilationShader, _DilationParameters, new Vector4(probeCount, settings.dilationValidityThreshold, settings.dilationDistance, ProbeReferenceVolume.instance.MinBrickSize()));
cmd.SetComputeVectorParam(dilationShader, _DilationParameters2, new Vector4(settings.squaredDistWeighting ? 1 : 0, 0, 0, 0));
var refVolume = ProbeReferenceVolume.instance;
ProbeReferenceVolume.RuntimeResources rr = refVolume.GetRuntimeResources();
bool validResources = rr.index != null && rr.L0_L1rx != null && rr.L1_G_ry != null && rr.L1_B_rz != null;
if (validResources)
{
cmd.SetGlobalBuffer(_APVResIndex, rr.index);
cmd.SetGlobalBuffer(_APVResCellIndices, rr.cellIndices);
cmd.SetGlobalTexture(_APVResL0_L1Rx, rr.L0_L1rx);
cmd.SetGlobalTexture(_APVResL1G_L1Ry, rr.L1_G_ry);
cmd.SetGlobalTexture(_APVResL1B_L1Rz, rr.L1_B_rz);
cmd.SetGlobalTexture(_APVResL2_0, rr.L2_0);
cmd.SetGlobalTexture(_APVResL2_1, rr.L2_1);
cmd.SetGlobalTexture(_APVResL2_2, rr.L2_2);
cmd.SetGlobalTexture(_APVResL2_3, rr.L2_3);
}
ProbeVolumeShadingParameters parameters;
parameters.normalBias = 0;
parameters.viewBias = 0;
parameters.scaleBiasByMinDistanceBetweenProbes = false;
parameters.samplingNoise = 0;
parameters.weight = 1f;
parameters.leakReductionMode = APVLeakReductionMode.None;
parameters.occlusionWeightContribution = 0.0f;
parameters.minValidNormalWeight = 0.0f;
parameters.frameIndexForNoise = 0;
parameters.reflNormalizationLowerClamp = 0.1f;
parameters.reflNormalizationUpperClamp = 1.0f;
ProbeReferenceVolume.instance.UpdateConstantBuffer(cmd, parameters);
int groupCount = (probeCount + 63) / 64;
cmd.DispatchCompute(dilationShader, dilationKernel, groupCount, 1, 1);
cmd.WaitAllAsyncReadbackRequests();
Graphics.ExecuteCommandBuffer(cmd);
data.ExtractDilatedProbes();
data.Dispose();
}
