Jak przekonwertować między ByteString a zapasowym wektorem?

Question

Jaki jest najlepszy sposób przekonwertowania między Storable.Vector Word8 a restrykcyjnym ByteString?

Oczywiście, najbardziej pożądany byłby sposób niekopiowania (no-op).

Czy powinienem po prostu unsafeCoerce lub czy istnieje funkcja biblioteki do tego (nie mogłem znaleźć)?

Co więcej, podejście będzie takie samo dla modelu Unboxed.Vector Word8?

Answer 1

Prosty unsafeCoerce nie będzie działać, jak układ konstruktorów danych jest inna:

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

vs.

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

Można importować Data.Array.Storable.Internals i Data.ByteString.Internal aby uzyskać dostęp do surowych konstruktorów a następnie skonstruuj jedno z pozostałych bez kopiowania danych:

> 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)) 

(Usunąłem raczej długi znak zachęty Prelude Data.Array.Storable.Internals Data.ByteString.Internal Data.ByteString>).

To nie zadziała dla Data.Vector.Unboxed, ponieważ tutaj dane są na stercie Haskella i zarządzane przez środowisko wykonawcze GHC, podczas gdy pozostałe dwa zarządzają danymi poza stertą Haskella.

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