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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
module State
( State
, bindS, bindS_, returnS, getS, putS, modifyS, sequenceS, sequenceS_, mapS
, mapS_, runState, evalState, execState, liftS, liftS2
) where
infixl 1 `bindS`, `bindS_`
type State s a = s -> (a, s)
bindS :: State s a -> (a -> State s b) -> State s b
bindS state f s = case state s of
(x, newS) -> newS `seq` (f x newS)
bindS_ :: State s a -> State s b -> State s b
bindS_ a b = a `bindS` (\_ -> b)
returnS :: a -> State s a
returnS x s = (x, s)
getS :: State s s
getS s = (s, s)
putS :: s -> State s ()
putS newS _ = ((), newS)
modifyS :: (s -> s) -> State s ()
modifyS f s = ((), f s)
sequenceS :: [State s a] -> State s [a]
sequenceS = foldr (\s newS -> s `bindS`
(\a -> newS `bindS` (\as -> returnS (a:as))))
(returnS [])
sequenceS_ :: [State s a] -> State s ()
sequenceS_ = foldr bindS_ (returnS ())
mapS :: (a -> State s b) -> [a] -> State s [b]
mapS f = sequenceS . (map f)
mapS_ :: (a -> State s b) -> [a] -> State s ()
mapS_ f = sequenceS_ . (map f)
runState :: State s a -> s -> (a, s)
runState state s = state s
evalState :: State s a -> s -> a
evalState state s = fst (runState state s)
execState :: State s a -> s -> s
execState state s = snd (runState state s)
liftS :: (a -> b) -> State s a -> State s b
liftS f act = act `bindS` (returnS . f)
liftS2 :: (a -> b -> c) -> State s a -> State s b -> State s c
liftS2 f a b = a `bindS` (\x -> b `bindS` (\y -> returnS (f x y))) |