From 6d00525334903adddb9133b4a795b1ba5cba1c1a Mon Sep 17 00:00:00 2001 From: Jack Wines Date: Mon, 15 Apr 2024 20:36:05 -0700 Subject: [PATCH] type signature cleanup --- src/Main.hs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/Main.hs b/src/Main.hs index c161d09..7944151 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -89,13 +89,15 @@ testCompression _ bs = . (compress :: BS.ByteString -> Maybe ([TreeDir], HuffmanTree a)) $ bs -encodeCompressed :: (C.Serialize a) => Maybe ([TreeDir], HuffmanTree a) -> (BS.ByteString) +encodeCompressed :: (C.Serialize a) => Maybe ([TreeDir], HuffmanTree a) -> BS.ByteString encodeCompressed = C.encode . fmap (Bi.first encodeTreeDirs) where encodeTreeDirs = cloneToByteStringWithLen . V.fromList . map (BV.Bit . (== R)) +cloneToByteStringWithLen :: V.Vector BV.Bit -> (BS.ByteString, Int) cloneToByteStringWithLen vec = (BV.cloneToByteString vec, V.length vec) +cloneFromByteStringWithLen :: (BS.ByteString, Int) -> V.Vector BV.Bit cloneFromByteStringWithLen (bs, len) = V.take len . BV.cloneFromByteString $ bs decodeTreeDirs :: (BS.ByteString, Int) -> [TreeDir]