Question

我正在尝试从字节字符串到Word16s列表进行简单转换。下面使用Data.Binary.Get的实现可行，但它是代码中的性能瓶颈。这是可以理解的，因为IO总是很慢，但我想知道是否有更有效的方法来做到这一点。

getImageData' = do
    e <- isEmpty
    if e then return []
      else do
      w <- getWord16be
      ws <- getImageData'
      return $ w : ws

Answer 1

我怀疑你与Data.Binary.Get遇到的一个大问题是解码器本质上对你的目的太严格了。这似乎也是serialise的情况，也可能是其他序列化库的情况。我认为根本问题在于，只要ByteString具有偶数个字节，你知道操作就会成功，但是库不知道。所以它必须先阅读整个ByteString才能得出结论＆＃34;啊，是的，一切都很好＆＃34;并构建您要求的列表。实际上，在你构建结果的方式中，它会在实际做任何有用的事情之前构建一系列闭包（数量与长度成比例）。

你怎么解决这个问题？只需直接使用bytestring库。最简单的方法是使用unpack，但我认为您可能会获得更好的性能：

{-# language BangPatterns #-}
module Wordy where
import qualified Data.ByteString as BS
import Data.ByteString (ByteString)
import Data.Word (Word16)
import Data.Maybe (fromMaybe)
import Data.Bits (unsafeShiftL)

toDBs :: ByteString -> Maybe [Word16]
toDBs bs0
  | odd (BS.length bs0) = Nothing
  | otherwise = Just (go bs0)
  where
    go bs = fromMaybe [] $ do
      (b1, bs') <- BS.uncons bs
      (b2, bs'') <- BS.uncons bs'
      let !res = (fromIntegral b1 `unsafeShiftL` 8) + fromIntegral b2
      Just (res : go bs'')

字节串的有效转换为[Word16]

1 个答案: