applicare una funzione Word32 ad un ByteString

Question

Ho un ByteString b con una lunghezza di forma 4 * n (con n intero) e si desidera utilizzare map con una funzione di f::Word32->Word32 su b (in modo che f viene applicata a "b[0..3]" , "b[4..7]", ecc.). Come può essere fatto in modo efficiente (ed elegante)?

Answer 1

È possibile pezzo fino il ByteString in modo molto efficiente, dato che B.take e B.drop sono entrambi O(1) operazioni:

import Data.ByteString (ByteString) 
import qualified Data.ByteString as B 

chunk :: Int -> ByteString -> [ByteString] 
chunk k = takeWhile (not . B.null) . map (B.take k) . iterate (B.drop k) 

poi:

\> :set -XOverloadedStrings 
\> chunk 4 "abcdefghijkl" 
["abcd","efgh","ijkl"] 

il resto sarebbe quello di mappare sulla lista la conversione da e verso il tipo desiderato e una singola chiamata a B.concat alla fine.

Una possibile fromByteString potrebbero essere attuate mediante turni bit e piega a sinistra:

import Data.Bits (Bits, shiftL, (.|.)) 

fromByteString :: (Num a, Bits a) => ByteString -> a 
fromByteString = B.foldl go 0 
    where go acc i = (acc `shiftL` 8) .|. (fromIntegral i) 

quindi: (! Che non richiede una copia)

\> map fromByteString $ chunk 4 "abcdefghijkl" :: [Word32] 
[1633837924,1701209960,1768581996] 

Answer 2

Un modo hacker ma efficiente è quello di convert to a storable vector , mappare sopra e riconvertire:

import Data.Vector.Storable.ByteString 
import qualified Data.Vector.Storable as VS 
import qualified Data.ByteString as BS 

mapBSChunks :: (VS.Storable a, VS.Storable b) 
       => (a->b) -> BS.ByteString -> BS.ByteString 
mapBSChunks f = vectorToByteString . VS.map f . byteStringToVector 

come da Michael's comment, si può facilmente definire le funzioni di conversione hacker a livello locale:

bytestringToVector bs = runST 
    (V.unsafeThaw (toByteVector bs) >>= V.unsafeFreeze . M.unsafeCast) 
vectorToByteString v = runST 
    (V.unsafeThaw v >>= fmap fromByteVector . V.unsafeFreeze . M.unsafeCast) 

anche se preferirei dipendo da una biblioteca per fornire questo, in particolare perché il casting non sicuro è un po 'di pesce.