{-# OPTIONS_GHC -fno-warn-orphans #-}
module Data.Vector.Binary (
genericGetVector
, genericGetVectorWith
, genericPutVector
, genericPutVectorWith
) where
import Data.Binary
import qualified Data.Vector.Generic as G
import qualified Data.Vector.Generic.Mutable as GM
import qualified Data.Vector.Unboxed as U
import qualified Data.Vector.Storable as S
import qualified Data.Vector.Primitive as P
import Data.Vector (Vector)
import System.IO.Unsafe
import Foreign.Storable (Storable)
instance Binary a => Binary (Vector a) where
put :: Vector a -> Put
put = Vector a -> Put
forall (v :: * -> *) a. (Vector v a, Binary a) => v a -> Put
genericPutVector
get :: Get (Vector a)
get = Get (Vector a)
forall (v :: * -> *) a. (Vector v a, Binary a) => Get (v a)
genericGetVector
{-# INLINE get #-}
instance (U.Unbox a, Binary a) => Binary (U.Vector a) where
put :: Vector a -> Put
put = Vector a -> Put
forall (v :: * -> *) a. (Vector v a, Binary a) => v a -> Put
genericPutVector
get :: Get (Vector a)
get = Get (Vector a)
forall (v :: * -> *) a. (Vector v a, Binary a) => Get (v a)
genericGetVector
{-# INLINE get #-}
instance (P.Prim a, Binary a) => Binary (P.Vector a) where
put :: Vector a -> Put
put = Vector a -> Put
forall (v :: * -> *) a. (Vector v a, Binary a) => v a -> Put
genericPutVector
get :: Get (Vector a)
get = Get (Vector a)
forall (v :: * -> *) a. (Vector v a, Binary a) => Get (v a)
genericGetVector
{-# INLINE get #-}
instance (Storable a, Binary a) => Binary (S.Vector a) where
put :: Vector a -> Put
put = Vector a -> Put
forall (v :: * -> *) a. (Vector v a, Binary a) => v a -> Put
genericPutVector
get :: Get (Vector a)
get = Get (Vector a)
forall (v :: * -> *) a. (Vector v a, Binary a) => Get (v a)
genericGetVector
{-# INLINE get #-}
genericGetVectorWith :: G.Vector v a
=> Get Int
-> Get a
-> Get (v a)
{-# INLINE genericGetVectorWith #-}
genericGetVectorWith :: forall (v :: * -> *) a. Vector v a => Get Int -> Get a -> Get (v a)
genericGetVectorWith Get Int
getN Get a
getA = do
n <- Get Int
getN
v <- return $ unsafePerformIO $ GM.unsafeNew n
let go Int
0 = () -> Get ()
forall a. a -> Get a
forall (m :: * -> *) a. Monad m => a -> m a
return ()
go Int
i = do x <- Get a
getA
() <- return $ unsafePerformIO $ GM.unsafeWrite v (n-i) x
go (i-1)
() <- go n
return $ unsafePerformIO $ G.unsafeFreeze v
genericPutVectorWith :: G.Vector v a
=> (Int -> Put)
-> (a -> Put)
-> v a -> Put
{-# INLINE genericPutVectorWith #-}
genericPutVectorWith :: forall (v :: * -> *) a.
Vector v a =>
(Int -> Put) -> (a -> Put) -> v a -> Put
genericPutVectorWith Int -> Put
putN a -> Put
putA v a
v = do
Int -> Put
putN (v a -> Int
forall (v :: * -> *) a. Vector v a => v a -> Int
G.length v a
v)
(a -> Put) -> v a -> Put
forall (m :: * -> *) (v :: * -> *) a b.
(Monad m, Vector v a) =>
(a -> m b) -> v a -> m ()
G.mapM_ a -> Put
putA v a
v
genericGetVector :: (G.Vector v a, Binary a) => Get (v a)
{-# INLINE genericGetVector #-}
genericGetVector :: forall (v :: * -> *) a. (Vector v a, Binary a) => Get (v a)
genericGetVector = Get Int -> Get a -> Get (v a)
forall (v :: * -> *) a. Vector v a => Get Int -> Get a -> Get (v a)
genericGetVectorWith Get Int
forall t. Binary t => Get t
get Get a
forall t. Binary t => Get t
get
genericPutVector :: (G.Vector v a, Binary a) => v a -> Put
{-# INLINE genericPutVector #-}
genericPutVector :: forall (v :: * -> *) a. (Vector v a, Binary a) => v a -> Put
genericPutVector = (Int -> Put) -> (a -> Put) -> v a -> Put
forall (v :: * -> *) a.
Vector v a =>
(Int -> Put) -> (a -> Put) -> v a -> Put
genericPutVectorWith Int -> Put
forall t. Binary t => t -> Put
put a -> Put
forall t. Binary t => t -> Put
put