Eliminate a few allocations in ColorSpace transformers

UglyToad · Apr 12, 2024 · 4fcc5c6 · 4fcc5c6
1 parent 6d54355
commit 4fcc5c6
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/UglyToad.PdfPig/Graphics/Colors/ColorSpaceDetails.cs b/src/UglyToad.PdfPig/Graphics/Colors/ColorSpaceDetails.cs
@@ -4,6 +4,7 @@
     using System.Collections.Concurrent;
     using System.Collections.Generic;
     using System.Linq;
+    using System.Runtime.InteropServices;
     using Tokens;
     using UglyToad.PdfPig.Content;
     using UglyToad.PdfPig.Functions;
@@ -572,7 +573,12 @@ internal override ReadOnlySpan<byte> Transform(ReadOnlySpan<byte> decoded)
                     transformed.Add(ConvertToByte(colors[c]));
                 }
             }
+
+#if NET8_0_OR_GREATER
+            return CollectionsMarshal.AsSpan(transformed);
+#else
             return transformed.ToArray();
+#endif
         }
 
         /// <inheritdoc/>
@@ -721,7 +727,7 @@ public override IColor GetColor(params double[] values)
         internal override ReadOnlySpan<byte> Transform(ReadOnlySpan<byte> values)
         {
             var cache = new Dictionary<int, double[]>();
-            var transformed = new List<byte>();
+            var transformed = new List<byte>(values.Length * 3);
             for (var i = 0; i < values.Length; i += 3)
             {
                 byte b = values[i++];
@@ -737,7 +743,11 @@ internal override ReadOnlySpan<byte> Transform(ReadOnlySpan<byte> values)
                 }
             }
 
+#if NET8_0_OR_GREATER
+            return CollectionsMarshal.AsSpan(transformed);
+#else
             return transformed.ToArray();
+#endif
         }
 
         /// <inheritdoc/>