module Data.Vector.Serialize (
genericGetVector
, genericPutVector
, genericGetVectorWith
, genericPutVectorWith
) where
import Control.Monad
import Data.Int (Int64)
import Data.Serialize (Get, Putter, Serialize(..))
import qualified Data.Vector as V
import qualified Data.Vector.Primitive as VP
import qualified Data.Vector.Unboxed as VU
import qualified Data.Vector.Generic as VG
import qualified Data.Vector.Storable as VS
genericGetVectorWith :: (VG.Vector v a) => Get a -> Get (v a)
genericGetVectorWith getter = do
len64 <- get :: Get Int64
when (len64 > fromIntegral (maxBound :: Int)) $
fail "Host can't deserialize a Vector longer than (maxBound :: Int)"
VG.replicateM (fromIntegral len64) getter
genericPutVectorWith :: (VG.Vector v a) => Putter a -> Putter (v a)
genericPutVectorWith putter v = do
put ((fromIntegral $ VG.length v) :: Int64)
VG.mapM_ putter v
genericPutVector :: (Serialize a, VG.Vector v a) => Putter (v a)
genericPutVector = genericPutVectorWith put
genericGetVector :: (Serialize a, VG.Vector v a) => Get (v a)
genericGetVector = genericGetVectorWith get
instance (Serialize a) => Serialize (V.Vector a) where
get = genericGetVector ; put = genericPutVector
instance (Serialize a, VP.Prim a) => Serialize (VP.Vector a) where
get = genericGetVector ; put = genericPutVector
instance (Serialize a, VU.Unbox a) => Serialize (VU.Vector a) where
get = genericGetVector ; put = genericPutVector
instance (Serialize a, VS.Storable a) => Serialize (VS.Vector a) where
get = genericGetVector ; put = genericPutVector