mirror of
https://github.com/idris-lang/Idris2.git
synced 2024-12-23 03:32:09 +03:00
136 lines
5.2 KiB
Idris
136 lines
5.2 KiB
Idris
||| Note: The difference to a 'strict' RWST implementation is
|
|
||| that accumulation of values does not happen in the
|
|
||| Applicative and Monad instances but when invoking `Writer`-specific
|
|
||| functions like `writer` or `listen`.
|
|
module Control.Monad.RWS.CPS
|
|
|
|
import Control.Monad.Identity
|
|
import Control.Monad.Trans
|
|
|
|
%default total
|
|
|
|
||| A monad transformer adding reading an environment of type `r`,
|
|
||| collecting an output of type `w` and updating a state of type `s`
|
|
||| to an inner monad `m`.
|
|
public export
|
|
record RWST (r : Type) (w : Type) (s : Type) (m : Type -> Type) (a : Type) where
|
|
constructor MkRWST
|
|
unRWST : r -> s -> w -> m (a, s, w)
|
|
|
|
||| Unwrap an RWST computation as a function. (The inverse of `rwsT`.)
|
|
public export %inline
|
|
runRWST : Monoid w => RWST r w s m a -> r -> s -> m (a, s, w)
|
|
runRWST m r s = unRWST m r s neutral
|
|
|
|
||| Construct an RWST computation from a function. (The inverse of `runRWST`.)
|
|
public export %inline
|
|
rwsT : (Functor m, Semigroup w) => (r -> s -> m (a, s, w)) -> RWST r w s m a
|
|
rwsT f = MkRWST $ \r,s,w => (\(a,s',w') => (a,s',w <+> w')) <$> f r s
|
|
|
|
||| Evaluate a computation with the given initial state and environment,
|
|
||| returning the final value and output, discarding the final state.
|
|
public export %inline
|
|
evalRWST : (Functor m, Monoid w) => RWST r w s m a -> r -> s -> m (a,w)
|
|
evalRWST m r s = (\(a,_,w) => (a,w)) <$> runRWST m r s
|
|
|
|
||| Evaluate a computation with the given initial state and environment,
|
|
||| returning the final state and output, discarding the final value.
|
|
public export %inline
|
|
execRWST : (Functor m, Monoid w) => RWST r w s m a -> r -> s -> m (s,w)
|
|
execRWST m r s = (\(_,s',w) => (s',w)) <$> runRWST m r s
|
|
|
|
||| Map the inner computation using the given function.
|
|
public export %inline
|
|
mapRWST : (Functor n, Monoid w, Semigroup w')
|
|
=> (m (a, s, w) -> n (b, s, w')) -> RWST r w s m a -> RWST r w' s n b
|
|
mapRWST f m = MkRWST \r,s,w =>
|
|
(\(a,s',w') => (a,s',w <+> w')) <$> f (runRWST m r s)
|
|
|
|
||| `withRWST f m` executes action `m` with an initial environment
|
|
||| and state modified by applying `f`.
|
|
public export %inline
|
|
withRWST : (r' -> s -> (r, s)) -> RWST r w s m a -> RWST r' w s m a
|
|
withRWST f m = MkRWST \r,s => uncurry (unRWST m) (f r s)
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- RWS Functions
|
|
--------------------------------------------------------------------------------
|
|
|
|
||| A monad containing an environment of type `r`, output of type `w`
|
|
||| and an updatable state of type `s`.
|
|
public export
|
|
RWS : (r : Type) -> (w : Type) -> (s : Type) -> (a : Type) -> Type
|
|
RWS r w s = RWST r w s Identity
|
|
|
|
||| Unwrap an RWS computation as a function. (The inverse of `rws`.)
|
|
public export %inline
|
|
runRWS : Monoid w => RWS r w s a -> r -> s -> (a, s, w)
|
|
runRWS m r s = runIdentity (runRWST m r s)
|
|
|
|
||| Construct an RWS computation from a function. (The inverse of `runRWS`.)
|
|
public export %inline
|
|
rws : Semigroup w => (r -> s -> (a, s, w)) -> RWS r w s a
|
|
rws f = MkRWST \r,s,w => let (a, s', w') = f r s
|
|
in Id (a, s', w <+> w')
|
|
|
|
||| Evaluate a computation with the given initial state and environment,
|
|
||| returning the final value and output, discarding the final state.
|
|
public export %inline
|
|
evalRWS : Monoid w => RWS r w s a -> r -> s -> (a,w)
|
|
evalRWS m r s = let (a,_,w) = runRWS m r s
|
|
in (a,w)
|
|
|
|
||| Evaluate a computation with the given initial state and environment,
|
|
||| returning the final state and output, discarding the final value.
|
|
public export %inline
|
|
execRWS : Monoid w => RWS r w s a -> r -> s -> (s,w)
|
|
execRWS m r s = let (_,s1,w) = runRWS m r s
|
|
in (s1,w)
|
|
|
|
||| Map the return value, final state and output of a computation using
|
|
||| the given function.
|
|
public export %inline
|
|
mapRWS : (Monoid w, Semigroup w')
|
|
=> ((a, s, w) -> (b, s, w')) -> RWS r w s a -> RWS r w' s b
|
|
mapRWS f = mapRWST \(Id p) => Id (f p)
|
|
|
|
||| `withRWS f m` executes action `m` with an initial environment
|
|
||| and state modified by applying `f`.
|
|
public export %inline
|
|
withRWS : (r' -> s -> (r, s)) -> RWS r w s a -> RWS r' w s a
|
|
withRWS = withRWST
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Implementations
|
|
--------------------------------------------------------------------------------
|
|
|
|
public export %inline
|
|
Functor m => Functor (RWST r w s m) where
|
|
map f m = MkRWST \r,s,w => (\(a,s',w') => (f a,s',w')) <$> unRWST m r s w
|
|
|
|
public export %inline
|
|
Monad m => Applicative (RWST r w s m) where
|
|
pure a = MkRWST \_,s,w => pure (a,s,w)
|
|
MkRWST mf <*> MkRWST mx =
|
|
MkRWST \r,s,w => do (f,s1,w1) <- mf r s w
|
|
(a,s2,w2) <- mx r s1 w1
|
|
pure (f a,s2,w2)
|
|
|
|
public export %inline
|
|
(Monad m, Alternative m) => Alternative (RWST r w s m) where
|
|
empty = MkRWST \_,_,_ => empty
|
|
MkRWST m <|> MkRWST n = MkRWST \r,s,w => m r s w <|> n r s w
|
|
|
|
public export %inline
|
|
Monad m => Monad (RWST r w s m) where
|
|
m >>= k = MkRWST \r,s,w => do (a,s1,w1) <- unRWST m r s w
|
|
unRWST (k a) r s1 w1
|
|
|
|
public export %inline
|
|
MonadTrans (RWST r w s) where
|
|
lift m = MkRWST \_,s,w => map (\a => (a,s,w)) m
|
|
|
|
public export %inline
|
|
HasIO m => HasIO (RWST r w s m) where
|
|
liftIO = lift . liftIO
|