From d7bb712031ab64974ca5ddfdcc908bf81e8ea9f6 Mon Sep 17 00:00:00 2001
From: Timofey Solomko <tsolomko@gmail.com>
Date: Sat, 23 Dec 2017 13:40:34 +0300
Subject: [PATCH] Return Data instead of bytes array in
 Deflate.encodeHuffmanBlock

---
 Sources/Deflate/Deflate+Compress.swift | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Sources/Deflate/Deflate+Compress.swift b/Sources/Deflate/Deflate+Compress.swift
index 229f5b1a..dcde59e9 100644
--- a/Sources/Deflate/Deflate+Compress.swift
+++ b/Sources/Deflate/Deflate+Compress.swift
@@ -39,7 +39,7 @@ extension Deflate: CompressionAlgorithm {
             // TODO: Implement dynamic Huffman code!
             return Deflate.createUncompressedBlock(data)
         } else {
-            return Data(bytes: Deflate.encodeHuffmanBlock(bldCodes.codes))
+            return Deflate.encodeHuffmanBlock(bldCodes.codes)
         }
     }
 
@@ -95,7 +95,7 @@ extension Deflate: CompressionAlgorithm {
         return out
     }
 
-    private static func encodeHuffmanBlock(_ bldCodes: [BLDCode]) -> [UInt8] {
+    private static func encodeHuffmanBlock(_ bldCodes: [BLDCode]) -> Data {
         let bitWriter = BitWriter(bitOrder: .reversed)
 
         // Write block header.
@@ -138,7 +138,7 @@ extension Deflate: CompressionAlgorithm {
         mainLiterals.code(symbol: 256)
         bitWriter.finish()
 
-        return bitWriter.buffer
+        return Data(bytes: bitWriter.buffer)
     }
 
     private enum BLDCode {