You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
6.6 KiB
126 lines
6.6 KiB
using FidelityFX;
|
|
using FidelityFX.FSR2;
|
|
using UnityEngine.Experimental.Rendering;
|
|
|
|
namespace UnityEngine.Rendering.PostProcessing
|
|
{
|
|
internal class FSR2Upscaler: Upscaler
|
|
{
|
|
public static bool IsSupported => SystemInfo.supportsComputeShaders;
|
|
|
|
private Fsr2Context _fsrContext;
|
|
|
|
private readonly Fsr2.DispatchDescription _dispatchDescription = new();
|
|
private readonly Fsr2.GenerateReactiveDescription _genReactiveDescription = new();
|
|
|
|
public override void CreateContext(PostProcessRenderContext context, Upscaling config)
|
|
{
|
|
// Initialize FSR2 context
|
|
Fsr2.InitializationFlags flags = 0;
|
|
if (context.camera.allowHDR) flags |= Fsr2.InitializationFlags.EnableHighDynamicRange;
|
|
if (config.exposureSource == Upscaling.ExposureSource.Auto) flags |= Fsr2.InitializationFlags.EnableAutoExposure;
|
|
if (RuntimeUtilities.IsDynamicResolutionEnabled(context.camera)) flags |= Fsr2.InitializationFlags.EnableDynamicResolution;
|
|
|
|
_fsrContext = Fsr2.CreateContext(config.UpscaleSize, config.MaxRenderSize, context.resources.computeShaders.fsr2Upscaler, flags);
|
|
}
|
|
|
|
public override void DestroyContext()
|
|
{
|
|
base.DestroyContext();
|
|
|
|
if (_fsrContext != null)
|
|
{
|
|
_fsrContext.Destroy();
|
|
_fsrContext = null;
|
|
}
|
|
}
|
|
|
|
public override void Render(PostProcessRenderContext context, Upscaling config)
|
|
{
|
|
var cmd = context.command;
|
|
cmd.BeginSample("FSR2");
|
|
|
|
SetupDispatchDescription(context, config);
|
|
|
|
if (config.autoGenerateReactiveMask)
|
|
{
|
|
SetupAutoReactiveDescription(context, config);
|
|
|
|
var scaledRenderSize = _genReactiveDescription.RenderSize;
|
|
cmd.GetTemporaryRT(Fsr2ShaderIDs.UavAutoReactive, scaledRenderSize.x, scaledRenderSize.y, 0, default, GraphicsFormat.R8_UNorm, 1, true);
|
|
_fsrContext.GenerateReactiveMask(_genReactiveDescription, cmd);
|
|
_dispatchDescription.Reactive = new ResourceView(Fsr2ShaderIDs.UavAutoReactive);
|
|
}
|
|
|
|
_fsrContext.Dispatch(_dispatchDescription, cmd);
|
|
|
|
cmd.EndSample("FSR2");
|
|
}
|
|
|
|
private void SetupDispatchDescription(PostProcessRenderContext context, Upscaling config)
|
|
{
|
|
var camera = context.camera;
|
|
|
|
// Set up the main FSR2 dispatch parameters
|
|
_dispatchDescription.Color = new ResourceView(context.source);
|
|
_dispatchDescription.Depth = new ResourceView(Upscaling.GetDepthTexture(context.camera), RenderTextureSubElement.Depth);
|
|
_dispatchDescription.MotionVectors = new ResourceView(BuiltinRenderTextureType.MotionVectors);
|
|
_dispatchDescription.Exposure = ResourceView.Unassigned;
|
|
_dispatchDescription.Reactive = ResourceView.Unassigned;
|
|
_dispatchDescription.TransparencyAndComposition = ResourceView.Unassigned;
|
|
|
|
if (config.exposureSource == Upscaling.ExposureSource.Manual && config.exposure != null) _dispatchDescription.Exposure = new ResourceView(config.exposure);
|
|
if (config.exposureSource == Upscaling.ExposureSource.Unity) _dispatchDescription.Exposure = new ResourceView(context.autoExposureTexture);
|
|
if (config.reactiveMask != null) _dispatchDescription.Reactive = new ResourceView(config.reactiveMask);
|
|
if (config.transparencyAndCompositionMask != null) _dispatchDescription.TransparencyAndComposition = new ResourceView(config.transparencyAndCompositionMask);
|
|
|
|
var scaledRenderSize = config.GetScaledRenderSize(context.camera);
|
|
|
|
_dispatchDescription.Output = new ResourceView(context.destination);
|
|
_dispatchDescription.PreExposure = config.preExposure;
|
|
_dispatchDescription.EnableSharpening = config.performSharpenPass;
|
|
_dispatchDescription.Sharpness = config.sharpness;
|
|
_dispatchDescription.JitterOffset = config.JitterOffset;
|
|
_dispatchDescription.MotionVectorScale.x = -scaledRenderSize.x;
|
|
_dispatchDescription.MotionVectorScale.y = -scaledRenderSize.y;
|
|
_dispatchDescription.RenderSize = scaledRenderSize;
|
|
_dispatchDescription.InputResourceSize = scaledRenderSize;
|
|
_dispatchDescription.FrameTimeDelta = Time.unscaledDeltaTime;
|
|
_dispatchDescription.CameraNear = camera.nearClipPlane;
|
|
_dispatchDescription.CameraFar = camera.farClipPlane;
|
|
_dispatchDescription.CameraFovAngleVertical = camera.fieldOfView * Mathf.Deg2Rad;
|
|
_dispatchDescription.ViewSpaceToMetersFactor = 1.0f; // 1 unit is 1 meter in Unity
|
|
_dispatchDescription.Reset = config.Reset;
|
|
|
|
// Set up the parameters for the optional experimental auto-TCR feature
|
|
_dispatchDescription.EnableAutoReactive = config.autoGenerateTransparencyAndComposition;
|
|
if (config.autoGenerateTransparencyAndComposition)
|
|
{
|
|
_dispatchDescription.ColorOpaqueOnly = new ResourceView(config.ColorOpaqueOnly);
|
|
_dispatchDescription.AutoTcThreshold = config.generateTransparencyAndCompositionParameters.autoTcThreshold;
|
|
_dispatchDescription.AutoTcScale = config.generateTransparencyAndCompositionParameters.autoTcScale;
|
|
_dispatchDescription.AutoReactiveScale = config.generateTransparencyAndCompositionParameters.autoReactiveScale;
|
|
_dispatchDescription.AutoReactiveMax = config.generateTransparencyAndCompositionParameters.autoReactiveMax;
|
|
}
|
|
|
|
if (SystemInfo.usesReversedZBuffer)
|
|
{
|
|
// Swap the near and far clip plane distances as FSR2 expects this when using inverted depth
|
|
(_dispatchDescription.CameraNear, _dispatchDescription.CameraFar) = (_dispatchDescription.CameraFar, _dispatchDescription.CameraNear);
|
|
}
|
|
}
|
|
|
|
private void SetupAutoReactiveDescription(PostProcessRenderContext context, Upscaling config)
|
|
{
|
|
// Set up the parameters to auto-generate a reactive mask
|
|
_genReactiveDescription.ColorOpaqueOnly = new ResourceView(config.ColorOpaqueOnly);
|
|
_genReactiveDescription.ColorPreUpscale = new ResourceView(context.source);
|
|
_genReactiveDescription.OutReactive = new ResourceView(Fsr2ShaderIDs.UavAutoReactive);
|
|
_genReactiveDescription.RenderSize = config.GetScaledRenderSize(context.camera);
|
|
_genReactiveDescription.Scale = config.generateReactiveParameters.scale;
|
|
_genReactiveDescription.CutoffThreshold = config.generateReactiveParameters.cutoffThreshold;
|
|
_genReactiveDescription.BinaryValue = config.generateReactiveParameters.binaryValue;
|
|
_genReactiveDescription.Flags = config.generateReactiveParameters.flags;
|
|
}
|
|
}
|
|
}
|