109 lines
4.8 KiB
C#
109 lines
4.8 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Runtime.CompilerServices;
|
|
using UnityEngine.Experimental.Rendering;
|
|
using UnityEngine.Rendering.RenderGraphModule;
|
|
|
|
namespace UnityEngine.Rendering.Universal
|
|
{
|
|
/// <summary>
|
|
/// Uses a compute shader to capture the depth and normal of the pixel under the cursor.
|
|
/// </summary>
|
|
internal partial class ProbeVolumeDebugPass : ScriptableRenderPass
|
|
{
|
|
ComputeShader m_ComputeShader;
|
|
RTHandle m_DepthTexture;
|
|
RTHandle m_NormalTexture;
|
|
|
|
/// <summary>
|
|
/// Creates a new <c>ProbeVolumeDebugPass</c> instance.
|
|
/// </summary>
|
|
public ProbeVolumeDebugPass(RenderPassEvent evt, ComputeShader computeShader)
|
|
{
|
|
base.profilingSampler = new ProfilingSampler("Dispatch APV Debug");
|
|
renderPassEvent = evt;
|
|
m_ComputeShader = computeShader;
|
|
}
|
|
|
|
public void Setup(RTHandle depthBuffer, RTHandle normalBuffer)
|
|
{
|
|
m_DepthTexture = depthBuffer;
|
|
m_NormalTexture = normalBuffer;
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
[Obsolete(DeprecationMessage.CompatibilityScriptingAPIObsolete, false)]
|
|
public override void Execute(ScriptableRenderContext context, ref RenderingData renderingData)
|
|
{
|
|
if (!ProbeReferenceVolume.instance.isInitialized)
|
|
return;
|
|
|
|
ref CameraData cameraData = ref renderingData.cameraData;
|
|
if (ProbeReferenceVolume.instance.GetProbeSamplingDebugResources(cameraData.camera, out var resultBuffer, out Vector2 coords))
|
|
{
|
|
var cmd = renderingData.commandBuffer;
|
|
int kernel = m_ComputeShader.FindKernel("ComputePositionNormal");
|
|
|
|
cmd.SetComputeTextureParam(m_ComputeShader, kernel, "_CameraDepthTexture", m_DepthTexture);
|
|
cmd.SetComputeTextureParam(m_ComputeShader, kernel, "_NormalBufferTexture", m_NormalTexture);
|
|
cmd.SetComputeVectorParam(m_ComputeShader, "_positionSS", new Vector4(coords.x, coords.y, 0.0f, 0.0f));
|
|
cmd.SetComputeBufferParam(m_ComputeShader, kernel, "_ResultBuffer", resultBuffer);
|
|
cmd.DispatchCompute(m_ComputeShader, kernel, 1, 1, 1);
|
|
}
|
|
}
|
|
|
|
|
|
class WriteApvData
|
|
{
|
|
public ComputeShader computeShader;
|
|
public BufferHandle resultBuffer;
|
|
public Vector2 clickCoordinates;
|
|
public TextureHandle depthBuffer;
|
|
public TextureHandle normalBuffer;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Render graph entry point
|
|
/// </summary>
|
|
/// <param name="renderGraph"></param>
|
|
/// <param name="renderingData"></param>
|
|
/// <param name="depthPyramidBuffer"></param>
|
|
/// <param name="normalBuffer"></param>
|
|
internal void Render(RenderGraph renderGraph, ContextContainer frameData, TextureHandle depthPyramidBuffer, TextureHandle normalBuffer)
|
|
{
|
|
UniversalCameraData cameraData = frameData.Get<UniversalCameraData>();
|
|
|
|
if (!ProbeReferenceVolume.instance.isInitialized)
|
|
return;
|
|
|
|
if (ProbeReferenceVolume.instance.GetProbeSamplingDebugResources(cameraData.camera, out var resultBuffer, out Vector2 coords))
|
|
{
|
|
using (var builder = renderGraph.AddComputePass<WriteApvData>(passName, out var passData, profilingSampler))
|
|
{
|
|
passData.clickCoordinates = coords;
|
|
passData.computeShader = m_ComputeShader;
|
|
|
|
passData.resultBuffer = renderGraph.ImportBuffer(resultBuffer);
|
|
passData.depthBuffer = depthPyramidBuffer;
|
|
passData.normalBuffer = normalBuffer;
|
|
|
|
builder.UseBuffer(passData.resultBuffer, AccessFlags.Write);
|
|
builder.UseTexture(passData.depthBuffer, AccessFlags.Read);
|
|
builder.UseTexture(passData.normalBuffer, AccessFlags.Read);
|
|
|
|
builder.SetRenderFunc((WriteApvData data, ComputeGraphContext ctx) =>
|
|
{
|
|
int kernel = data.computeShader.FindKernel("ComputePositionNormal");
|
|
|
|
ctx.cmd.SetComputeTextureParam(data.computeShader, kernel, "_CameraDepthTexture", data.depthBuffer);
|
|
ctx.cmd.SetComputeTextureParam(data.computeShader, kernel, "_NormalBufferTexture", data.normalBuffer);
|
|
ctx.cmd.SetComputeVectorParam(data.computeShader, "_positionSS", new Vector4(data.clickCoordinates.x, data.clickCoordinates.y, 0.0f, 0.0f));
|
|
ctx.cmd.SetComputeBufferParam(data.computeShader, kernel, "_ResultBuffer", data.resultBuffer);
|
|
ctx.cmd.DispatchCompute(data.computeShader, kernel, 1, 1, 1);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|