More Encoding and Encoder compat extensions

LTRData · Sep 13, 2024 · dd5b787 · dd5b787
1 parent 608c43d
commit dd5b787
Showing 1 changed file with 49 additions and 1 deletion.
diff --git a/Library/DiscUtils.Streams/Util/EncodingExtensions.cs b/Library/DiscUtils.Streams/Util/EncodingExtensions.cs
@@ -64,7 +64,7 @@ public static int GetCharCount(this Decoder decoder, ReadOnlySpan<byte> bytes, b
         }
     }
 
-    public static int GetChars(this Decoder decoder, ReadOnlySpan<byte> bytes, Span<char> chars, bool flush)
+    public static int GetChars(this Decoder decoder, ReadOnlySpan<byte> bytes, Span<char> chars)
     {
         var bytesBuffer = ArrayPool<byte>.Shared.Rent(bytes.Length);
         try
@@ -88,6 +88,54 @@ public static int GetChars(this Decoder decoder, ReadOnlySpan<byte> bytes, Span<
         }
     }
 
+    public static int GetChars(this Decoder decoder, ReadOnlySpan<byte> bytes, Span<char> chars, bool flush)
+    {
+        var bytesBuffer = ArrayPool<byte>.Shared.Rent(bytes.Length);
+        try
+        {
+            var charsBuffer = ArrayPool<char>.Shared.Rent(chars.Length);
+            try
+            {
+                bytes.CopyTo(bytesBuffer);
+                var i = decoder.GetChars(bytesBuffer, 0, bytes.Length, charsBuffer, 0, flush);
+                charsBuffer.AsSpan(0, i).CopyTo(chars);
+                return i;
+            }
+            finally
+            {
+                ArrayPool<char>.Shared.Return(charsBuffer);
+            }
+        }
+        finally
+        {
+            ArrayPool<byte>.Shared.Return(bytesBuffer);
+        }
+    }
+
+    public static int GetChars(this Encoding encoding, ReadOnlySpan<byte> bytes, Span<char> chars)
+    {
+        var bytesBuffer = ArrayPool<byte>.Shared.Rent(bytes.Length);
+        try
+        {
+            var charsBuffer = ArrayPool<char>.Shared.Rent(chars.Length);
+            try
+            {
+                bytes.CopyTo(bytesBuffer);
+                var i = encoding.GetChars(bytesBuffer, 0, bytes.Length, charsBuffer, 0);
+                charsBuffer.AsSpan(0, i).CopyTo(chars);
+                return i;
+            }
+            finally
+            {
+                ArrayPool<char>.Shared.Return(charsBuffer);
+            }
+        }
+        finally
+        {
+            ArrayPool<byte>.Shared.Return(bytesBuffer);
+        }
+    }
+
     public static void Convert(this Encoder decoder, ReadOnlySpan<char> chars, Span<byte> bytes, bool flush, out int charsUsed, out int bytesUsed, out bool completed)
     {
         var bytesBuffer = ArrayPool<byte>.Shared.Rent(bytes.Length);