Question

我正在构建一个Conduit，它写一个二进制文件，其中包含一个标题，后跟一个Double矩阵作为行排序列表。这是代码：

import Conduit ((.|), ConduitM, mapC, sinkFileBS, yield) 
import Control.Monad.Trans.Except (ExceptT) 
import Control.Monad.Trans.Resource (ResourceT) 
import Data.ByteString (ByteString) 
import Data.ByteString.Conversion (toByteString') 
import Data.Serialize.IEEE754 (putFloat64be) 
import Data.Serialize.Put (putListOf, runPut) 
import Data.Void (Void) 
import Numeric.LinearAlgebra.Data ((><), Matrix, toLists) 
import System.FilePath (FilePath) 

type FileWriter = ResourceT (ExceptT String IO) 

matrixSink :: FilePath -> ConduitM (Matrix Double) Void FileWriter ()
matrixSink path = byteBuilder .| sinkFileBS path where
  byteBuilder = do
    yield $ toByteString' "header" 
    mapC fromDoubleMatrix

fromDoubleMatrix :: Matrix Double -> ByteString
fromDoubleMatrix matrix = runPut $
   putListOf putFloat64be (concat toLists matrix)

这几乎可行。如果我使用

测试它

runExceptT . runConduitRes $ yield matrix .| matrixSink "test.dat"
  where matrix = (2 >< 2) [1, 2, 3, 4]

我得到了预期的文件，但是在标题和双打列表之间有一个额外的字节。使用show显示时，额外字节如下所示：

"\NUL\NUL\NUL\NUL\NUL\NUL\NUL\t"

知道怎么不打印这个字节？或者，如果它是规范的分隔符或其他东西（以便我可以在阅读器中忽略它）？

编辑：问题似乎发生在putListOf的{{1}}构造中。

Answer 1

putListOf :: Putter a -> Putter [a]
putListOf pa = \l -> do
  putWord64be (fromIntegral (length l))
  mapM_ pa l

在编码各个列表元素之前，

putListOf对列表的长度进行编码。我想也许你正在处理固定的2x2矩阵，所以你不需要这个长度，你只需要：

fromDoubleMatrix :: Matrix Double -> ByteString
fromDoubleMatrix matrix = runPut $
   mapM_ putFloat64be (concat toLists matrix)

二进制文件写入器添加额外的字节

1 个答案: