1 files changed, 76 insertions, 0 deletions
diff --git a/Ryujinx.Graphics.Gpu/Shader/ShaderSpecializationList.cs b/Ryujinx.Graphics.Gpu/Shader/ShaderSpecializationList.cs
new file mode 100644
index 00000000..87e08754
--- /dev/null
+++ b/Ryujinx.Graphics.Gpu/Shader/ShaderSpecializationList.cs
@@ -0,0 +1,76 @@
+using System.Collections;
+using System.Collections.Generic;
+
+namespace Ryujinx.Graphics.Gpu.Shader
+{
+    /// <summary>
+    /// List of cached shader programs that differs only by specialization state.
+    /// </summary>
+    class ShaderSpecializationList : IEnumerable<CachedShaderProgram>
+    {
+        private readonly List<CachedShaderProgram> _entries = new List<CachedShaderProgram>();
+
+        /// <summary>
+        /// Adds a program to the list.
+        /// </summary>
+        /// <param name="program">Program to be added</param>
+        public void Add(CachedShaderProgram program)
+        {
+            _entries.Add(program);
+        }
+
+        /// <summary>
+        /// Tries to find an existing 3D program on the cache.
+        /// </summary>
+        /// <param name="channel">GPU channel</param>
+        /// <param name="poolState">Texture pool state</param>
+        /// <param name="program">Cached program, if found</param>
+        /// <returns>True if a compatible program is found, false otherwise</returns>
+        public bool TryFindForGraphics(GpuChannel channel, GpuChannelPoolState poolState, out CachedShaderProgram program)
+        {
+            foreach (var entry in _entries)
+            {
+                if (entry.SpecializationState.MatchesGraphics(channel, poolState))
+                {
+                    program = entry;
+                    return true;
+                }
+            }
+
+            program = default;
+            return false;
+        }
+
+        /// <summary>
+        /// Tries to find an existing compute program on the cache.
+        /// </summary>
+        /// <param name="channel">GPU channel</param>
+        /// <param name="poolState">Texture pool state</param>
+        /// <param name="program">Cached program, if found</param>
+        /// <returns>True if a compatible program is found, false otherwise</returns>
+        public bool TryFindForCompute(GpuChannel channel, GpuChannelPoolState poolState, out CachedShaderProgram program)
+        {
+            foreach (var entry in _entries)
+            {
+                if (entry.SpecializationState.MatchesCompute(channel, poolState))
+                {
+                    program = entry;
+                    return true;
+                }
+            }
+
+            program = default;
+            return false;
+        }
+
+        public IEnumerator<CachedShaderProgram> GetEnumerator()
+        {
+            return _entries.GetEnumerator();
+        }
+
+        IEnumerator IEnumerable.GetEnumerator()
+        {
+            return GetEnumerator();
+        }
+    }
+}
+\ No newline at end of file