Merge pull request #235 from Cysharp/hadashiA/expose-brotli-compresse…

…d-size Add method to get max compressed length to BrotliCompressor
Cysharp · Mar 14, 2024 · fb16a8b · fb16a8b
2 parents 27feb86 + 5beb2c0
commit fb16a8b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/MemoryPack.Core/Compression/BrotliCompressor.cs b/src/MemoryPack.Core/Compression/BrotliCompressor.cs
@@ -70,6 +70,12 @@ Span<byte> IBufferWriter<byte>.GetSpan(int sizeHint)
         return bufferWriter.GetSpan(sizeHint);
     }
 
+    public int GetMaxCompressedLength()
+    {
+        ThrowIfDisposed();
+        return BrotliUtils.BrotliEncoderMaxCompressedSize(bufferWriter.TotalWritten);
+    }
+
     public byte[] ToArray()
     {
         ThrowIfDisposed();

diff --git a/...k.Unity/Assets/Plugins/MemoryPack/Runtime/MemoryPack.Core/Compression/BrotliCompressor.cs b/...k.Unity/Assets/Plugins/MemoryPack/Runtime/MemoryPack.Core/Compression/BrotliCompressor.cs
@@ -79,6 +79,12 @@ Span<byte> IBufferWriter<byte>.GetSpan(int sizeHint)
         return bufferWriter.GetSpan(sizeHint);
     }
 
+    public int GetMaxCompressedLength()
+    {
+        ThrowIfDisposed();
+        return BrotliUtils.BrotliEncoderMaxCompressedSize(bufferWriter.TotalWritten);
+    }
+
     public byte[] ToArray()
     {
         ThrowIfDisposed();
@@ -334,5 +340,5 @@ internal static int BrotliEncoderMaxCompressedSize(int input_size)
         return (result < input_size) ? 0 : result;
     }
 }
-
+
 }