Add support for large sampler arrays on Vulkan (#6489)

* Add support for large sampler arrays on Vulkan

* Shader cache version bump

* Format whitespace

* Move DescriptorSetManager to PipelineLayoutCacheEntry to allow different pool sizes per layout

* Handle array textures with different types on the same buffer

* Somewhat better caching system

* Avoid useless buffer data modification checks

* Move redundant bindings update checking to the backend

* Fix an issue where texture arrays would get the same bindings across stages on Vulkan

* Backport some fixes from part 2

* Fix typo

* PR feedback

* Format whitespace

* Add some missing XML docs

4 files changed, 138 insertions, 0 deletions

diff --git a/src/Ryujinx.Graphics.OpenGL/Image/ImageArray.cs b/src/Ryujinx.Graphics.OpenGL/Image/ImageArray.cs
new file mode 100644
index 00000000..1c5acedf
--- /dev/null
+++ b/src/Ryujinx.Graphics.OpenGL/Image/ImageArray.cs
@@ -0,0 +1,67 @@
+using OpenTK.Graphics.OpenGL;
+using Ryujinx.Graphics.GAL;
+using System;
+
+namespace Ryujinx.Graphics.OpenGL.Image
+{
+    class ImageArray : IImageArray
+    {
+        private record struct TextureRef
+        {
+            public int Handle;
+            public Format Format;
+        }
+
+        private readonly TextureRef[] _images;
+
+        public ImageArray(int size)
+        {
+            _images = new TextureRef[size];
+        }
+
+        public void SetFormats(int index, GAL.Format[] imageFormats)
+        {
+            for (int i = 0; i < imageFormats.Length; i++)
+            {
+                _images[index + i].Format = imageFormats[i];
+            }
+        }
+
+        public void SetImages(int index, ITexture[] images)
+        {
+            for (int i = 0; i < images.Length; i++)
+            {
+                ITexture image = images[i];
+
+                if (image is TextureBase imageBase)
+                {
+                    _images[index + i].Handle = imageBase.Handle;
+                }
+                else
+                {
+                    _images[index + i].Handle = 0;
+                }
+            }
+        }
+
+        public void Bind(int baseBinding)
+        {
+            for (int i = 0; i < _images.Length; i++)
+            {
+                if (_images[i].Handle == 0)
+                {
+                    GL.BindImageTexture(baseBinding + i, 0, 0, true, 0, TextureAccess.ReadWrite, SizedInternalFormat.Rgba8);
+                }
+                else
+                {
+                    SizedInternalFormat format = FormatTable.GetImageFormat(_images[i].Format);
+
+                    if (format != 0)
+                    {
+                        GL.BindImageTexture(baseBinding + i, _images[i].Handle, 0, true, 0, TextureAccess.ReadWrite, format);
+                    }
+                }
+            }
+        }
+    }
+}
diff --git a/src/Ryujinx.Graphics.OpenGL/Image/TextureArray.cs b/src/Ryujinx.Graphics.OpenGL/Image/TextureArray.cs
new file mode 100644
index 00000000..d70b0a00
--- /dev/null
+++ b/src/Ryujinx.Graphics.OpenGL/Image/TextureArray.cs
@@ -0,0 +1,52 @@
+using Ryujinx.Graphics.GAL;
+
+namespace Ryujinx.Graphics.OpenGL.Image
+{
+    class TextureArray : ITextureArray
+    {
+        private record struct TextureRef
+        {
+            public TextureBase Texture;
+            public Sampler Sampler;
+        }
+
+        private readonly TextureRef[] _textureRefs;
+
+        public TextureArray(int size)
+        {
+            _textureRefs = new TextureRef[size];
+        }
+
+        public void SetSamplers(int index, ISampler[] samplers)
+        {
+            for (int i = 0; i < samplers.Length; i++)
+            {
+                _textureRefs[index + i].Sampler = samplers[i] as Sampler;
+            }
+        }
+
+        public void SetTextures(int index, ITexture[] textures)
+        {
+            for (int i = 0; i < textures.Length; i++)
+            {
+                _textureRefs[index + i].Texture = textures[i] as TextureBase;
+            }
+        }
+
+        public void Bind(int baseBinding)
+        {
+            for (int i = 0; i < _textureRefs.Length; i++)
+            {
+                if (_textureRefs[i].Texture != null)
+                {
+                    _textureRefs[i].Texture.Bind(baseBinding + i);
+                    _textureRefs[i].Sampler?.Bind(baseBinding + i);
+                }
+                else
+                {
+                    TextureBase.ClearBinding(baseBinding + i);
+                }
+            }
+        }
+    }
+}
diff --git a/src/Ryujinx.Graphics.OpenGL/OpenGLRenderer.cs b/src/Ryujinx.Graphics.OpenGL/OpenGLRenderer.cs
index eabcb3c1..a945cbf2 100644
--- a/src/Ryujinx.Graphics.OpenGL/OpenGLRenderer.cs
+++ b/src/Ryujinx.Graphics.OpenGL/OpenGLRenderer.cs
@@ -90,6 +90,11 @@ namespace Ryujinx.Graphics.OpenGL
             throw new NotSupportedException();
         }
 
+        public IImageArray CreateImageArray(int size, bool isBuffer)
+        {
+            return new ImageArray(size);
+        }
+
         public IProgram CreateProgram(ShaderSource[] shaders, ShaderInfo info)
         {
             return new Program(shaders, info.FragmentOutputMap);
@@ -112,6 +117,11 @@ namespace Ryujinx.Graphics.OpenGL
             }
         }
 
+        public ITextureArray CreateTextureArray(int size, bool isBuffer)
+        {
+            return new TextureArray(size);
+        }
+
         public void DeleteBuffer(BufferHandle buffer)
         {
             PersistentBuffers.Unmap(buffer);
diff --git a/src/Ryujinx.Graphics.OpenGL/Pipeline.cs b/src/Ryujinx.Graphics.OpenGL/Pipeline.cs
index 0757fcd9..6d066bb6 100644
--- a/src/Ryujinx.Graphics.OpenGL/Pipeline.cs
+++ b/src/Ryujinx.Graphics.OpenGL/Pipeline.cs
@@ -958,6 +958,11 @@ namespace Ryujinx.Graphics.OpenGL
             }
         }
 
+        public void SetImageArray(ShaderStage stage, int binding, IImageArray array)
+        {
+            (array as ImageArray).Bind(binding);
+        }
+
         public void SetIndexBuffer(BufferRange buffer, IndexType type)
         {
             _elementsType = type.Convert();
@@ -1302,6 +1307,10 @@ namespace Ryujinx.Graphics.OpenGL
             }
         }
 
+        public void SetTextureArray(ShaderStage stage, int binding, ITextureArray array)
+        {
+            (array as TextureArray).Bind(binding);
+        }
 
         public void SetTransformFeedbackBuffers(ReadOnlySpan<BufferRange> buffers)
         {