blob: d3d9f6a87920b80bc275df63af6c9b292dfd16ba (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
module T1735_Help.State where
import Control.Monad (ap, liftM)
newtype StateT s m a = StateT { runStateT :: s -> m (a,s) }
instance Monad m => Monad (StateT s m) where
m >>= k = StateT $ \s -> do
~(a, s') <- runStateT m s
runStateT (k a) s'
instance MonadFail m => MonadFail (StateT s m) where
fail s = StateT $ \_ -> fail s
instance Monad m => Functor (StateT s m) where
fmap = liftM
instance Monad m => Applicative (StateT s m) where
pure a = StateT $ \s -> pure (a, s)
(<*>) = ap
get :: Monad m => StateT s m s
get = StateT $ \s -> return (s, s)
put :: Monad m => s -> StateT s m ()
put s = StateT $ \_ -> return ((), s)
|