388 lines
17 KiB
C#
388 lines
17 KiB
C#
|
using UnityEngine;
|
|||
|
|
using UnityEngine.Rendering;
|
|||
|
|
using UnityEngine.Rendering.Universal;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
#if UNIVERSAL_RP_17_0_OR_NEWER
|
|||
|
|
using UnityEngine.Experimental.Rendering;
|
|||
|
|
using UnityEngine.Rendering.RenderGraphModule;
|
|||
|
|
using UnityEngine.Rendering.RenderGraphModule.Util;
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
namespace NBShader
|
|||
|
|
{
|
|||
|
|
public class DisturbanceMaskRenderPass : ScriptableRenderPass
|
|||
|
|
{
|
|||
|
|
private ProfilingSampler _profilingSampler;
|
|||
|
|
#if UNIVERSAL_RP_13_1_2_OR_NEWER
|
|||
|
|
private RTHandle _DisturbanceMaskRTHandle;
|
|||
|
|
private RTHandle _cameraDepthRTHandle;
|
|||
|
|
private RTHandle _DownRT;
|
|||
|
|
#else
|
|||
|
|
private static readonly int _DisturbanceMaskRTID = Shader.PropertyToID("_DisturbanceMaskRT");
|
|||
|
|
private static readonly int _DownRTID = Shader.PropertyToID("_DisturbanceMaskTex");
|
|||
|
|
private RenderTargetIdentifier _DisturbanceMaskRTHandle = new RenderTargetIdentifier (_DisturbanceMaskRTID);
|
|||
|
|
private RenderTargetIdentifier _cameraDepthRTHandle;
|
|||
|
|
private RenderTargetIdentifier _DownRT = new RenderTargetIdentifier (_DownRTID);
|
|||
|
|
#endif
|
|||
|
|
private Material tempMat;
|
|||
|
|
|
|||
|
|
private readonly Downsampling _downSampling;
|
|||
|
|
|
|||
|
|
private Material _renderMaskMat ;
|
|||
|
|
// public LayerMask _DisturbanceMaskLayer = 1 << 25;
|
|||
|
|
private FilteringSettings _Filtering;
|
|||
|
|
private static readonly int CameraTexture = Shader.PropertyToID("_CameraTexture");
|
|||
|
|
private static readonly int SampleOffset = Shader.PropertyToID("_SampleOffset");
|
|||
|
|
private static readonly int DisturbanceMaskTex = Shader.PropertyToID("_DisturbanceMaskTex");
|
|||
|
|
private readonly Color _clearDisturbanceMaskColor = new Color(0f, 0f, 0f, 1f);
|
|||
|
|
|
|||
|
|
private readonly List<ShaderTagId> _shaderTag = new List<ShaderTagId>()
|
|||
|
|
{
|
|||
|
|
// new ShaderTagId("UniversalForward"),
|
|||
|
|
// new ShaderTagId("SRPDefaultUnlit"),
|
|||
|
|
// new ShaderTagId("UniversalForwardOnly")
|
|||
|
|
new ShaderTagId("NBDeferredDistortPass")
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
public DisturbanceMaskRenderPass(ProfilingSampler profilingSampler,Material DisturbanceMaskMat, Downsampling downSampling)
|
|||
|
|
{
|
|||
|
|
_profilingSampler = profilingSampler;
|
|||
|
|
_renderMaskMat = DisturbanceMaskMat;
|
|||
|
|
_downSampling = downSampling;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#if UNIVERSAL_RP_17_0_OR_NEWER
|
|||
|
|
private class RenderGraphMaskPassData
|
|||
|
|
{
|
|||
|
|
public RendererListHandle rendererListHandle;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private class GlobalTexturePassData
|
|||
|
|
{
|
|||
|
|
public TextureHandle texture;
|
|||
|
|
public int nameID;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static bool IsSupportedCamera(CameraType cameraType)
|
|||
|
|
{
|
|||
|
|
return cameraType == CameraType.Game || cameraType == CameraType.SceneView;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static void SetGlobalTextureAfterPass(RenderGraph renderGraph, TextureHandle texture, int nameID, string passName)
|
|||
|
|
{
|
|||
|
|
using (var builder = renderGraph.AddRasterRenderPass<GlobalTexturePassData>(passName, out var passData))
|
|||
|
|
{
|
|||
|
|
passData.texture = texture;
|
|||
|
|
passData.nameID = nameID;
|
|||
|
|
builder.UseTexture(texture, AccessFlags.Read);
|
|||
|
|
builder.AllowPassCulling(false);
|
|||
|
|
builder.AllowGlobalStateModification(true);
|
|||
|
|
builder.SetGlobalTextureAfterPass(texture, nameID);
|
|||
|
|
builder.SetRenderFunc(static (GlobalTexturePassData data, RasterGraphContext context) =>
|
|||
|
|
{
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void ApplyDownsampling(ref TextureDesc descriptor)
|
|||
|
|
{
|
|||
|
|
switch (_downSampling)
|
|||
|
|
{
|
|||
|
|
case Downsampling._2xBilinear:
|
|||
|
|
descriptor.width = Mathf.Max(1, descriptor.width / 2);
|
|||
|
|
descriptor.height = Mathf.Max(1, descriptor.height / 2);
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBilinear:
|
|||
|
|
case Downsampling._4xBox:
|
|||
|
|
descriptor.width = Mathf.Max(1, descriptor.width / 4);
|
|||
|
|
descriptor.height = Mathf.Max(1, descriptor.height / 4);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private int GetDownsamplePassIndex()
|
|||
|
|
{
|
|||
|
|
return _downSampling == Downsampling._4xBox ? 1 : 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override void RecordRenderGraph(RenderGraph renderGraph, ContextContainer frameData)
|
|||
|
|
{
|
|||
|
|
UniversalCameraData cameraData = frameData.Get<UniversalCameraData>();
|
|||
|
|
if (!IsSupportedCamera(cameraData.cameraType) || _renderMaskMat == null)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
UniversalResourceData resourceData = frameData.Get<UniversalResourceData>();
|
|||
|
|
if (!resourceData.activeColorTexture.IsValid())
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
UniversalRenderingData universalRenderingData = frameData.Get<UniversalRenderingData>();
|
|||
|
|
UniversalLightData lightData = frameData.Get<UniversalLightData>();
|
|||
|
|
|
|||
|
|
TextureDesc maskDescriptor = renderGraph.GetTextureDesc(resourceData.activeColorTexture);
|
|||
|
|
maskDescriptor.name = "DisturbanceMaskRT";
|
|||
|
|
maskDescriptor.colorFormat = GraphicsFormat.R16G16_SFloat;
|
|||
|
|
maskDescriptor.depthBufferBits = DepthBits.None;
|
|||
|
|
maskDescriptor.clearBuffer = true;
|
|||
|
|
maskDescriptor.clearColor = _clearDisturbanceMaskColor;
|
|||
|
|
|
|||
|
|
TextureHandle maskTexture = renderGraph.CreateTexture(maskDescriptor);
|
|||
|
|
using (var builder = renderGraph.AddRasterRenderPass<RenderGraphMaskPassData>("DisturbanceRender", out var passData))
|
|||
|
|
{
|
|||
|
|
DrawingSettings drawingSettings = RenderingUtils.CreateDrawingSettings(
|
|||
|
|
_shaderTag,
|
|||
|
|
universalRenderingData,
|
|||
|
|
cameraData,
|
|||
|
|
lightData,
|
|||
|
|
cameraData.defaultOpaqueSortFlags);
|
|||
|
|
FilteringSettings filteringSettings = new FilteringSettings(RenderQueueRange.all);
|
|||
|
|
RendererListParams rendererListParameters = new RendererListParams(
|
|||
|
|
universalRenderingData.cullResults,
|
|||
|
|
drawingSettings,
|
|||
|
|
filteringSettings);
|
|||
|
|
|
|||
|
|
passData.rendererListHandle = renderGraph.CreateRendererList(rendererListParameters);
|
|||
|
|
builder.UseRendererList(passData.rendererListHandle);
|
|||
|
|
builder.SetRenderAttachment(maskTexture, 0, AccessFlags.ReadWrite);
|
|||
|
|
|
|||
|
|
if (resourceData.activeDepthTexture.IsValid())
|
|||
|
|
builder.SetRenderAttachmentDepth(resourceData.activeDepthTexture, AccessFlags.ReadWrite);
|
|||
|
|
|
|||
|
|
if (_downSampling == Downsampling.None)
|
|||
|
|
builder.SetGlobalTextureAfterPass(maskTexture, DisturbanceMaskTex);
|
|||
|
|
|
|||
|
|
builder.SetRenderFunc(static (RenderGraphMaskPassData data, RasterGraphContext context) =>
|
|||
|
|
{
|
|||
|
|
context.cmd.DrawRendererList(data.rendererListHandle);
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (_downSampling == Downsampling.None)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
TextureDesc downsampleDescriptor = maskDescriptor;
|
|||
|
|
downsampleDescriptor.name = "MaskDownCopyRT";
|
|||
|
|
downsampleDescriptor.clearBuffer = false;
|
|||
|
|
ApplyDownsampling(ref downsampleDescriptor);
|
|||
|
|
|
|||
|
|
TextureHandle downsampleTexture = renderGraph.CreateTexture(downsampleDescriptor);
|
|||
|
|
|
|||
|
|
if (_downSampling == Downsampling._4xBox)
|
|||
|
|
_renderMaskMat.SetFloat(SampleOffset, 2);
|
|||
|
|
|
|||
|
|
RenderGraphUtils.BlitMaterialParameters blitParameters =
|
|||
|
|
new RenderGraphUtils.BlitMaterialParameters(maskTexture, downsampleTexture, _renderMaskMat, GetDownsamplePassIndex());
|
|||
|
|
blitParameters.sourceTexturePropertyID = CameraTexture;
|
|||
|
|
|
|||
|
|
renderGraph.AddBlitPass(blitParameters, "DisturbanceMaskDownsample");
|
|||
|
|
SetGlobalTextureAfterPass(renderGraph, downsampleTexture, DisturbanceMaskTex, "Set Disturbance Mask");
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
#if UNIVERSAL_RP_13_1_2_OR_NEWER
|
|||
|
|
|
|||
|
|
public void SetUp ( RTHandle cameraRTHandle )
|
|||
|
|
{
|
|||
|
|
#if !UNITY_6000_3_OR_NEWER || (URP_COMPATIBILITY_MODE && !UNITY_6000_4_OR_NEWER)
|
|||
|
|
#pragma warning disable CS0618
|
|||
|
|
|
|||
|
|
RenderTextureDescriptor descrip = cameraRTHandle.rt.descriptor;
|
|||
|
|
|
|||
|
|
descrip.colorFormat = RenderTextureFormat.RGHalf;
|
|||
|
|
descrip.depthBufferBits = 0;
|
|||
|
|
RenderingUtils.ReAllocateIfNeeded(ref _DisturbanceMaskRTHandle, descrip, name: "DisturbanceMaskRT");
|
|||
|
|
|
|||
|
|
|
|||
|
|
switch (_downSampling)
|
|||
|
|
{
|
|||
|
|
case Downsampling._2xBilinear:
|
|||
|
|
descrip.width /= 2;
|
|||
|
|
descrip.height /= 2;
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBilinear:
|
|||
|
|
descrip.width /= 4;
|
|||
|
|
descrip.height /= 4;
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBox:
|
|||
|
|
descrip.width /= 4;
|
|||
|
|
descrip.height /= 4;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
if (_downSampling != Downsampling.None)
|
|||
|
|
RenderingUtils.ReAllocateIfNeeded(ref _DownRT, descrip, name:"MaskDownCopyRT");
|
|||
|
|
#pragma warning restore CS0618
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
#else
|
|||
|
|
#if !UNITY_6000_3_OR_NEWER || (URP_COMPATIBILITY_MODE && !UNITY_6000_4_OR_NEWER)
|
|||
|
|
//在AddPass之前触发
|
|||
|
|
|
|||
|
|
public void SetUpDisturbanceMask(RenderTextureDescriptor descrip ,CommandBuffer cmd)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
descrip.colorFormat = RenderTextureFormat.RGHalf;
|
|||
|
|
descrip.depthBufferBits = 0;
|
|||
|
|
cmd.GetTemporaryRT(_DisturbanceMaskRTID, descrip,FilterMode.Bilinear);
|
|||
|
|
|
|||
|
|
switch (_downSampling)
|
|||
|
|
{
|
|||
|
|
case Downsampling._2xBilinear:
|
|||
|
|
descrip.width /= 2;
|
|||
|
|
descrip.height /= 2;
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBilinear:
|
|||
|
|
descrip.width /= 4;
|
|||
|
|
descrip.height /= 4;
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBox:
|
|||
|
|
descrip.width /= 4;
|
|||
|
|
descrip.height /= 4;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
if (_downSampling != Downsampling.None)
|
|||
|
|
cmd.GetTemporaryRT(_DownRTID, descrip,FilterMode.Bilinear);
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
#if !UNITY_6000_3_OR_NEWER || (URP_COMPATIBILITY_MODE && !UNITY_6000_4_OR_NEWER)
|
|||
|
|
#pragma warning disable CS0618, CS0672
|
|||
|
|
public override void OnCameraSetup(CommandBuffer cmd, ref RenderingData renderingData)
|
|||
|
|
{
|
|||
|
|
_Filtering = new FilteringSettings(RenderQueueRange.all);
|
|||
|
|
#if UNIVERSAL_RP_13_1_2_OR_NEWER
|
|||
|
|
_cameraDepthRTHandle = renderingData.cameraData.renderer.cameraDepthTargetHandle;
|
|||
|
|
#else
|
|||
|
|
_cameraDepthRTHandle = renderingData.cameraData.renderer.cameraDepthTarget;
|
|||
|
|
SetUpDisturbanceMask(renderingData.cameraData.cameraTargetDescriptor,cmd);
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override void Configure(CommandBuffer cmd, RenderTextureDescriptor cameraTextureDescriptor)
|
|||
|
|
{
|
|||
|
|
#if UNIVERSAL_RP_13_1_2_OR_NEWER
|
|||
|
|
ConfigureTarget(_DisturbanceMaskRTHandle, _cameraDepthRTHandle);
|
|||
|
|
#else
|
|||
|
|
ConfigureTarget(_DisturbanceMaskRTID, _cameraDepthRTHandle);
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
//将RT清空
|
|||
|
|
ConfigureClear(ClearFlag.Color, _clearDisturbanceMaskColor);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override void Execute(ScriptableRenderContext context, ref RenderingData renderingData)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
if (!(renderingData.cameraData.cameraType == CameraType.Game || renderingData.cameraData.cameraType == CameraType.SceneView))
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
if (!_renderMaskMat)
|
|||
|
|
{
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var DisturbanceDraw = CreateDrawingSettings(_shaderTag, ref renderingData, renderingData.cameraData.defaultOpaqueSortFlags);
|
|||
|
|
|
|||
|
|
CommandBuffer cmd = CommandBufferPool.Get();
|
|||
|
|
using (new ProfilingScope(cmd, _profilingSampler))
|
|||
|
|
{
|
|||
|
|
context.DrawRenderers(renderingData.cullResults, ref DisturbanceDraw, ref _Filtering);
|
|||
|
|
|
|||
|
|
#if UNIVERSAL_RP_13_1_2_OR_NEWER
|
|||
|
|
if (_downSampling == Downsampling.None)
|
|||
|
|
{
|
|||
|
|
cmd.SetGlobalTexture(DisturbanceMaskTex, _DisturbanceMaskRTHandle);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
_renderMaskMat.SetTexture(CameraTexture, _DisturbanceMaskRTHandle);
|
|||
|
|
cmd.SetRenderTarget((RenderTargetIdentifier)_DownRT);
|
|||
|
|
|
|||
|
|
switch (_downSampling)
|
|||
|
|
{
|
|||
|
|
case Downsampling._2xBilinear:
|
|||
|
|
Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, Vector2.one, _renderMaskMat, 0);
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBilinear:
|
|||
|
|
Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, Vector2.one, _renderMaskMat, 0);
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBox:
|
|||
|
|
_renderMaskMat.SetFloat(SampleOffset, 2);
|
|||
|
|
Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, Vector2.one, _renderMaskMat, 1);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
cmd.SetGlobalTexture(DisturbanceMaskTex, _DownRT);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#else
|
|||
|
|
|
|||
|
|
if (_downSampling == Downsampling.None)
|
|||
|
|
{
|
|||
|
|
cmd.SetGlobalTexture(DisturbanceMaskTex, _DisturbanceMaskRTHandle);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
// cmd.SetGlobalTexture(_DisturbanceMaskRTID, _DisturbanceMaskRTHandle);
|
|||
|
|
//Bug:在非播放状态时，_DisturbanceMaskRTID在这里会经常丢失。造成画面闪烁。但是，只要有一个DistortObject在Scene中，并LockInspector，就不会闪烁，非常奇怪。
|
|||
|
|
cmd.SetGlobalTexture(CameraTexture, _DisturbanceMaskRTHandle);
|
|||
|
|
// _renderMaskMat.SetTexture(CameraTexture, Shader.GetGlobalTexture(_DisturbanceMaskRTID));
|
|||
|
|
cmd.SetRenderTarget(_DownRT);
|
|||
|
|
switch (_downSampling)
|
|||
|
|
{
|
|||
|
|
case Downsampling._2xBilinear:
|
|||
|
|
// Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 0);
|
|||
|
|
// cmd.Blit(_DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 0);
|
|||
|
|
// cmd.DrawMesh(RenderingUtils.fullscreenMesh, Matrix4x4.identity, _renderMaskMat, 0, 2);
|
|||
|
|
Blit(cmd, _DisturbanceMaskRTHandle,_DownRT,_renderMaskMat,0);
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBilinear:
|
|||
|
|
// Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 0);
|
|||
|
|
// cmd.Blit(_DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 0);
|
|||
|
|
// cmd.DrawMesh(RenderingUtils.fullscreenMesh, Matrix4x4.identity, _renderMaskMat, 0, 2);
|
|||
|
|
Blit(cmd, _DisturbanceMaskRTHandle,_DownRT,_renderMaskMat,0);
|
|||
|
|
break;
|
|||
|
|
case Downsampling._4xBox:
|
|||
|
|
_renderMaskMat.SetFloat(SampleOffset, 2);
|
|||
|
|
// Blitter.BlitTexture(cmd, _DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 1);
|
|||
|
|
// cmd.Blit(_DisturbanceMaskRTHandle, _DownRT, _renderMaskMat, 1);
|
|||
|
|
// cmd.DrawMesh(RenderingUtils.fullscreenMesh, Matrix4x4.identity, _renderMaskMat, 0, 3);
|
|||
|
|
Blit(cmd, _DisturbanceMaskRTHandle,_DownRT,_renderMaskMat,1);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
cmd.SetGlobalTexture(DisturbanceMaskTex, _DownRT);
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
context.ExecuteCommandBuffer(cmd);
|
|||
|
|
CommandBufferPool.Release(cmd);
|
|||
|
|
}
|
|||
|
|
#pragma warning restore CS0618, CS0672
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
#if !UNIVERSAL_RP_13_1_2_OR_NEWER && (!UNITY_6000_3_OR_NEWER || (URP_COMPATIBILITY_MODE && !UNITY_6000_4_OR_NEWER))
|
|||
|
|
// Cleanup any allocated resources that were created during the execution of this render pass.
|
|||
|
|
public override void OnCameraCleanup(CommandBuffer cmd)
|
|||
|
|
{
|
|||
|
|
cmd.ReleaseTemporaryRT(_DisturbanceMaskRTID);
|
|||
|
|
if (_downSampling != Downsampling.None)
|
|||
|
|
cmd.ReleaseTemporaryRT(_DownRTID);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//JustForSimple
|
|||
|
|
public void Dispose()
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
#else
|
|||
|
|
|
|||
|
|
public void Dispose()
|
|||
|
|
{
|
|||
|
|
_DisturbanceMaskRTHandle?.Release();
|
|||
|
|
_DownRT?.Release();
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
}
|