Prep-work to add TAA support in SRPs

Author: Chman

PostProcessing/Runtime/Effects/TemporalAntialiasing.cs

@@ -74,84 +74,25 @@ Vector2 GenerateRandomOffset()
             return offset;
         }
 
-        // Adapted heavily from PlayDead's TAA code
-        // https://github.com/playdeadgames/temporal/blob/master/Assets/Scripts/Extensions.cs
-        Matrix4x4 GetPerspectiveProjectionMatrix(Camera camera, Vector2 offset)
-        {
-            float vertical = Mathf.Tan(0.5f * Mathf.Deg2Rad * camera.fieldOfView);
-            float horizontal = vertical * camera.aspect;
-            float near = camera.nearClipPlane;
-            float far = camera.farClipPlane;
-
-            offset.x *= horizontal / (0.5f * camera.pixelWidth);
-            offset.y *= vertical / (0.5f * camera.pixelHeight);
-
-            float left = (offset.x - horizontal) * near;
-            float right = (offset.x + horizontal) * near;
-            float top = (offset.y + vertical) * near;
-            float bottom = (offset.y - vertical) * near;
-
-            var matrix = new Matrix4x4();
-
-            matrix[0, 0] = (2f * near) / (right - left);
-            matrix[0, 1] = 0f;
-            matrix[0, 2] = (right + left) / (right - left);
-            matrix[0, 3] = 0f;
-
-            matrix[1, 0] = 0f;
-            matrix[1, 1] = (2f * near) / (top - bottom);
-            matrix[1, 2] = (top + bottom) / (top - bottom);
-            matrix[1, 3] = 0f;
-
-            matrix[2, 0] = 0f;
-            matrix[2, 1] = 0f;
-            matrix[2, 2] = -(far + near) / (far - near);
-            matrix[2, 3] = -(2f * far * near) / (far - near);
-
-            matrix[3, 0] = 0f;
-            matrix[3, 1] = 0f;
-            matrix[3, 2] = -1f;
-            matrix[3, 3] = 0f;
-
-            return matrix;
-        }
-
-        Matrix4x4 GetOrthographicProjectionMatrix(Camera camera, Vector2 offset)
-        {
-            float vertical = camera.orthographicSize;
-            float horizontal = vertical * camera.aspect;
-
-            offset.x *= horizontal / (0.5f * camera.pixelWidth);
-            offset.y *= vertical / (0.5f * camera.pixelHeight);
-
-            float left = offset.x - horizontal;
-            float right = offset.x + horizontal;
-            float top = offset.y + vertical;
-            float bottom = offset.y - vertical;
-
-            return Matrix4x4.Ortho(left, right, bottom, top, camera.nearClipPlane, camera.farClipPlane);
-        }
-
-        public void SetProjectionMatrix(Camera camera)
+        public Matrix4x4 GetJitteredProjectionMatrix(Camera camera)
         {
+            Matrix4x4 cameraProj;
             jitter = GenerateRandomOffset();
             jitter *= jitterSpread;
 
-            camera.nonJitteredProjectionMatrix = camera.projectionMatrix;
-
             if (jitteredMatrixFunc != null)
             {
-                camera.projectionMatrix = jitteredMatrixFunc(camera, jitter);
+                cameraProj = jitteredMatrixFunc(camera, jitter);
             }
             else
             {
-                camera.projectionMatrix = camera.orthographic
-                    ? GetOrthographicProjectionMatrix(camera, jitter)
-                    : GetPerspectiveProjectionMatrix(camera, jitter);
+                cameraProj = camera.orthographic
+                    ? RuntimeUtilities.GetJitteredOrthographicProjectionMatrix(camera, jitter)
+                    : RuntimeUtilities.GetJitteredPerspectiveProjectionMatrix(camera, jitter);
             }
 
-            camera.useJitteredProjectionMatrixForTransparentRendering = false;
             jitter = new Vector2(jitter.x / camera.pixelWidth, jitter.y / camera.pixelHeight);
+            return cameraProj;
         }
 
         RenderTexture CheckHistory(int id, PostProcessRenderContext context)

PostProcessing/Runtime/PostProcessLayer.cs

@@ -518,7 +518,13 @@ public void Render(PostProcessRenderContext context)
             int lastTarget = -1;
             if (context.IsTemporalAntialiasingActive())
             {
-                temporalAntialiasing.SetProjectionMatrix(context.camera);
+                if (!RuntimeUtilities.scriptableRenderPipelineActive)
+                {
+                    var camera = context.camera;
+                    camera.nonJitteredProjectionMatrix = camera.projectionMatrix;
+                    camera.projectionMatrix = temporalAntialiasing.GetJitteredProjectionMatrix(camera);
+                    camera.useJitteredProjectionMatrixForTransparentRendering = false;
+                }
 
                 lastTarget = m_TargetPool.Get();
                 var finalDestination = context.destination;

PostProcessing/Runtime/Utils/RuntimeUtilities.cs

@@ -294,6 +294,64 @@ public static float Exp2(float x)
             return Mathf.Exp(x * 0.69314718055994530941723212145818f);
         }
 
+        // Adapted heavily from PlayDead's TAA code
+        // https://github.com/playdeadgames/temporal/blob/master/Assets/Scripts/Extensions.cs
+        public static Matrix4x4 GetJitteredPerspectiveProjectionMatrix(Camera camera, Vector2 offset)
+        {
+            float vertical = Mathf.Tan(0.5f * Mathf.Deg2Rad * camera.fieldOfView);
+            float horizontal = vertical * camera.aspect;
+            float near = camera.nearClipPlane;
+            float far = camera.farClipPlane;
+
+            offset.x *= horizontal / (0.5f * camera.pixelWidth);
+            offset.y *= vertical / (0.5f * camera.pixelHeight);
+
+            float left = (offset.x - horizontal) * near;
+            float right = (offset.x + horizontal) * near;
+            float top = (offset.y + vertical) * near;
+            float bottom = (offset.y - vertical) * near;
+
+            var matrix = new Matrix4x4();
+
+            matrix[0, 0] = (2f * near) / (right - left);
+            matrix[0, 1] = 0f;
+            matrix[0, 2] = (right + left) / (right - left);
+            matrix[0, 3] = 0f;
+
+            matrix[1, 0] = 0f;
+            matrix[1, 1] = (2f * near) / (top - bottom);
+            matrix[1, 2] = (top + bottom) / (top - bottom);
+            matrix[1, 3] = 0f;
+
+            matrix[2, 0] = 0f;
+            matrix[2, 1] = 0f;
+            matrix[2, 2] = -(far + near) / (far - near);
+            matrix[2, 3] = -(2f * far * near) / (far - near);
+
+            matrix[3, 0] = 0f;
+            matrix[3, 1] = 0f;
+            matrix[3, 2] = -1f;
+            matrix[3, 3] = 0f;
+
+            return matrix;
+        }
+
+        public static Matrix4x4 GetJitteredOrthographicProjectionMatrix(Camera camera, Vector2 offset)
+        {
+            float vertical = camera.orthographicSize;
+            float horizontal = vertical * camera.aspect;
+
+            offset.x *= horizontal / (0.5f * camera.pixelWidth);
+            offset.y *= vertical / (0.5f * camera.pixelHeight);
+
+            float left = offset.x - horizontal;
+            float right = offset.x + horizontal;
+            float top = offset.y + vertical;
+            float bottom = offset.y - vertical;
+
+            return Matrix4x4.Ortho(left, right, bottom, top, camera.nearClipPlane, camera.farClipPlane);
+        }
+
         #endregion
 
         #region Reflection

PostProcessing/Shaders/Builtins/TemporalAntialiasing.shader

@@ -1,12 +1,18 @@
 Shader "Hidden/PostProcessing/TemporalAntialiasing"
 {
     HLSLINCLUDE
-        
+
         #pragma exclude_renderers gles
         #include "../StdLib.hlsl"
         #include "../Colors.hlsl"
 
-        TEXTURE2D_SAMPLER2D(_MainTex, sampler_MainTex);
+    #if UNITY_VERSION >= 201710
+        #define _MainTexSampler sampler_LinearClamp
+    #else
+        #define _MainTexSampler sampler_MainTex
+    #endif
+
+        TEXTURE2D_SAMPLER2D(_MainTex, _MainTexSampler);
         float4 _MainTex_TexelSize;
 
         TEXTURE2D_SAMPLER2D(_HistoryTex, sampler_HistoryTex);
@@ -86,10 +92,10 @@ Shader "Hidden/PostProcessing/TemporalAntialiasing"
             const float2 k = _MainTex_TexelSize.xy;
             float2 uv = texcoord - _Jitter;
 
-            float4 color = SAMPLE_TEXTURE2D(_MainTex, sampler_MainTex, uv);
+            float4 color = SAMPLE_TEXTURE2D(_MainTex, _MainTexSampler, uv);
 
-            float4 topLeft = SAMPLE_TEXTURE2D(_MainTex, sampler_MainTex, uv - k * 0.5);
-            float4 bottomRight = SAMPLE_TEXTURE2D(_MainTex, sampler_MainTex, uv + k * 0.5);
+            float4 topLeft = SAMPLE_TEXTURE2D(_MainTex, _MainTexSampler, uv - k * 0.5);
+            float4 bottomRight = SAMPLE_TEXTURE2D(_MainTex, _MainTexSampler, uv + k * 0.5);
 
             float4 corners = 4.0 * (topLeft + bottomRight) - 2.0 * color;