module Ptr.Peek
where
import Ptr.Prelude hiding (take)
import qualified Ptr.PokeAndPeek as B
import qualified Ptr.Parse as C
import qualified Ptr.ParseUnbound as D
import qualified Ptr.IO as A
data Peek output =
Peek {-# UNPACK #-} !Int !(Ptr Word8 -> IO output)
instance Functor Peek where
{-# INLINE fmap #-}
fmap fn (Peek size io) =
Peek size (fmap fn . io)
instance Applicative Peek where
{-# INLINE pure #-}
pure x =
Peek 0 (const (pure x))
{-# INLINE (<*>) #-}
(<*>) (Peek leftSize leftIO) (Peek rightSize rightIO) =
Peek (leftSize + rightSize) io
where
io ptr =
leftIO ptr <*> rightIO (plusPtr ptr leftSize)
{-# INLINE word8 #-}
word8 :: Peek Word8
word8 =
{-# SCC "word8" #-}
Peek 1 A.peekWord8
{-# INLINE beWord16 #-}
beWord16 :: Peek Word16
beWord16 =
{-# SCC "beWord16" #-}
Peek 2 A.peekBEWord16
{-# INLINE beWord32 #-}
beWord32 :: Peek Word32
beWord32 =
{-# SCC "beWord32" #-}
Peek 4 A.peekBEWord32
{-# INLINE beWord64 #-}
beWord64 :: Peek Word64
beWord64 =
{-# SCC "beWord64" #-}
Peek 8 A.peekBEWord64
{-# INLINE bytes #-}
bytes :: Int -> Peek ByteString
bytes !amount =
{-# SCC "bytes" #-}
Peek amount (\ ptr -> A.peekBytes ptr amount)
{-# INLINE shortByteString #-}
shortByteString :: Int -> Peek ShortByteString
shortByteString !amount =
{-# SCC "shortByteString" #-}
Peek amount (\ ptr -> A.peekShortByteString ptr amount)
{-# INLINE pokeAndPeek #-}
pokeAndPeek :: B.PokeAndPeek input output -> Peek output
pokeAndPeek (B.PokeAndPeek size _ io) =
{-# SCC "pokeAndPeek" #-}
Peek size io
{-# INLINE parse #-}
parse :: Int -> C.Parse a -> (Int -> a) -> (Text -> a) -> Peek a
parse amount (C.Parse parseIO) eoi error =
{-# SCC "parse" #-}
Peek amount $ \ ptr ->
parseIO amount ptr (return . eoi) (return . error) (\result _ _ -> return result)
{-# INLINE parseUnbound #-}
parseUnbound :: Int -> D.ParseUnbound a -> (Int -> a) -> (Text -> a) -> Peek a
parseUnbound sizeBound (D.ParseUnbound parseIO) eoi error =
{-# SCC "parse" #-}
Peek sizeBound $ \ ptr ->
parseIO ptr (return . error)
(\ result size -> if size <= sizeBound
then return (eoi (size - sizeBound))
else return result)
{-# INLINE peekAmountAndParse #-}
peekAmountAndParse :: Peek Int -> C.Parse a -> (Int -> a) -> (Text -> a) -> Peek (Peek a)
peekAmountAndParse peekAmount parse_ eoi error =
{-# SCC "peekAmountAndParse" #-}
flip fmap peekAmount $ \amount ->
parse amount parse_ eoi error