testss

Library/PackageCache/com.unity.render-pipelines.universal@11.0.0/Runtime/ShaderData.cs

@@ -0,0 +1,82 @@
+using System;
+using System.Runtime.InteropServices;
+
+namespace UnityEngine.Rendering.Universal
+{
+    class ShaderData : IDisposable
+    {
+        static ShaderData m_Instance = null;
+        ComputeBuffer m_LightDataBuffer = null;
+        ComputeBuffer m_LightIndicesBuffer = null;
+
+        ComputeBuffer m_AdditionalLightShadowParamsStructuredBuffer = null;
+        ComputeBuffer m_AdditionalLightShadowSliceMatricesStructuredBuffer = null;
+
+        ShaderData()
+        {
+        }
+
+        internal static ShaderData instance
+        {
+            get
+            {
+                if (m_Instance == null)
+                    m_Instance = new ShaderData();
+
+                return m_Instance;
+            }
+        }
+
+        public void Dispose()
+        {
+            DisposeBuffer(ref m_LightDataBuffer);
+            DisposeBuffer(ref m_LightIndicesBuffer);
+            DisposeBuffer(ref m_AdditionalLightShadowParamsStructuredBuffer);
+            DisposeBuffer(ref m_AdditionalLightShadowSliceMatricesStructuredBuffer);
+        }
+
+        internal ComputeBuffer GetLightDataBuffer(int size)
+        {
+            return GetOrUpdateBuffer<ShaderInput.LightData>(ref m_LightDataBuffer, size);
+        }
+
+        internal ComputeBuffer GetLightIndicesBuffer(int size)
+        {
+            return GetOrUpdateBuffer<int>(ref m_LightIndicesBuffer, size);
+        }
+
+        internal ComputeBuffer GetAdditionalLightShadowParamsStructuredBuffer(int size)
+        {
+            return GetOrUpdateBuffer<Vector4>(ref m_AdditionalLightShadowParamsStructuredBuffer, size);
+        }
+
+        internal ComputeBuffer GetAdditionalLightShadowSliceMatricesStructuredBuffer(int size)
+        {
+            return GetOrUpdateBuffer<Matrix4x4>(ref m_AdditionalLightShadowSliceMatricesStructuredBuffer, size);
+        }
+
+        ComputeBuffer GetOrUpdateBuffer<T>(ref ComputeBuffer buffer, int size) where T : struct
+        {
+            if (buffer == null)
+            {
+                buffer = new ComputeBuffer(size, Marshal.SizeOf<T>());
+            }
+            else if (size > buffer.count)
+            {
+                buffer.Dispose();
+                buffer = new ComputeBuffer(size, Marshal.SizeOf<T>());
+            }
+
+            return buffer;
+        }
+
+        void DisposeBuffer(ref ComputeBuffer buffer)
+        {
+            if (buffer != null)
+            {
+                buffer.Dispose();
+                buffer = null;
+            }
+        }
+    }
+}