From 6d722d83bae78221301f188f7e094de21621de33 Mon Sep 17 00:00:00 2001
From: Isaac Marovitz <isaacryu@icloud.com>
Date: Wed, 2 Aug 2023 15:57:57 -0400
Subject: [PATCH] SetData

---
 src/Ryujinx.Graphics.Metal/Texture.cs | 56 ++++++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 2 deletions(-)

diff --git a/src/Ryujinx.Graphics.Metal/Texture.cs b/src/Ryujinx.Graphics.Metal/Texture.cs
index 205546eb2..2a1e0a48c 100644
--- a/src/Ryujinx.Graphics.Metal/Texture.cs
+++ b/src/Ryujinx.Graphics.Metal/Texture.cs
@@ -157,9 +157,61 @@ namespace Ryujinx.Graphics.Metal
             throw new NotImplementedException();
         }
 
-        public void SetData(SpanOrArray<byte> data)
+        // TODO: Handle array formats
+        public unsafe void SetData(SpanOrArray<byte> data)
         {
-            Logger.Warning?.Print(LogClass.Gpu, "Not Implemented!");
+            MTLBlitCommandEncoder blitCommandEncoder;
+
+            if (_pipeline.CurrentEncoder is MTLBlitCommandEncoder encoder)
+            {
+                blitCommandEncoder = encoder;
+            }
+            else
+            {
+                blitCommandEncoder = _pipeline.BeginBlitPass();
+            }
+
+            var dataSpan = data.Span;
+            var mtlBuffer = _device.NewBuffer((ulong)dataSpan.Length, MTLResourceOptions.ResourceStorageModeShared);
+            var bufferSpan = new Span<byte>(mtlBuffer.Contents.ToPointer(), dataSpan.Length);
+            dataSpan.CopyTo(bufferSpan);
+
+            int width = Info.Width;
+            int height = Info.Height;
+            int depth = Info.Depth;
+            int levels = Info.GetLevelsClamped();
+
+            int offset = 0;
+
+            for (int level = 0; level < levels; level++)
+            {
+                int mipSize = Info.GetMipSize(level);
+
+                int endOffset = offset + mipSize;
+
+                if ((uint)endOffset > (uint)dataSpan.Length)
+                {
+                    return;
+                }
+
+                blitCommandEncoder.CopyFromBuffer(
+                    mtlBuffer,
+                    (ulong)offset,
+                    (ulong)Info.GetMipStride(level),
+                    (ulong)mipSize,
+                    new MTLSize { width = (ulong)width, height = (ulong)height, depth = (ulong)depth },
+                    MTLTexture,
+                    0,
+                    (ulong)level,
+                    new MTLOrigin()
+                );
+
+                offset += mipSize;
+
+                width = Math.Max(1, width >> 1);
+                height = Math.Max(1, height >> 1);
+                depth = Math.Max(1, depth >> 1);
+            }
         }
 
         public void SetData(SpanOrArray<byte> data, int layer, int level)