Come convertire tra ByteString e Storable Vector?

Question

Qual è il modo migliore per convertire tra Storable.Vector Word8 e un ByteString rigoroso?

Naturalmente un modo non di copiatura (non opzionale) sarebbe più apprezzato.

Devo solo unsafeCoerce o c'è una funzione di libreria per quello (non riuscivo a trovarne uno)?

Inoltre, l'approccio sarà lo stesso per un Unboxed.Vector Word8?

Answer 1

Un semplice unsafeCoerce non funzionerà, come la disposizione dei costruttori di dati è diverso:

data StorableArray i e = StorableArray !i !i Int !(ForeignPtr e) 

vs.

data ByteString = PS {-# UNPACK #-} !(ForeignPtr Word8) -- payload 
        {-# UNPACK #-} !Int    -- offset 
        {-# UNPACK #-} !Int    -- length 

È possibile importare Data.Array.Storable.Internals e Data.ByteString.Internal per ottenere l'accesso ai costruttori prime e quindi costruirne uno senza copiare i dati:

> let bs = pack [1,2,3] 
> bs 
"\SOH\STX\ETX" 
> let sa = case bs of (PS ptr 0 n) -> StorableArray 0 (n-1) n ptr 
> :t sa 
sa :: StorableArray Int GHC.Word.Word8 
> Data.Array.MArray.readArray sa 1 
2 
> Data.Array.MArray.readArray sa 0 
1 
> Data.Array.MArray.readArray sa 3 
*** Exception: Ix{Int}.index: Index (3) out of range ((0,2)) 

(Ho rimosso il prompt piuttosto lungo di Prelude Data.Array.Storable.Internals Data.ByteString.Internal Data.ByteString>).

Questo non funzionerà per Data.Vector.Unboxed, perché qui i dati sono nell'heap di Haskell e gestiti dal runtime di GHC, mentre gli altri due gestiscono i dati all'esterno dell'heap di Haskell.

Answer 2

byteStringToVector :: (Storable a) => BS.ByteString -> V.Vector a 
byteStringToVector bs = vec where 
    vec = V.unsafeFromForeignPtr (castForeignPtr fptr) (scale off) (scale len) 
    (fptr, off, len) = BS.toForeignPtr bs 
    scale = (`div` sizeOfElem vec) 

sizeOfElem vec = sizeOf (undefined `asTypeOf` V.head vec) 

http://hackage.haskell.org/packages/archive/spool/0.1/doc/html/Data-Vector-Storable-ByteString.html