module Control.Arrow.Transformer.CoState(
CoStateArrow,
) where
import Control.Arrow
import Control.Arrow.Operations
newtype CoStateArrow s a b c = CST (a (s -> b) (s -> c))
instance Arrow a => Arrow (CoStateArrow s a) where
arr f = CST (arr (f .))
CST f >>> CST g = CST (f >>> g)
first (CST f) = CST (arr unzipMap >>> first f >>> arr zipMap)
zipMap :: (s -> a, s -> b) -> (s -> (a,b))
zipMap h s = (fst h s, snd h s)
unzipMap :: (s -> (a,b)) -> (s -> a, s -> b)
unzipMap h = (fst . h, snd . h)
instance ArrowLoop a => ArrowLoop (CoStateArrow s a) where
loop (CST f) = CST (loop (arr zipMap >>> f >>> arr unzipMap))
instance ArrowZero a => ArrowZero (CoStateArrow s a) where
zeroArrow = CST zeroArrow
instance ArrowPlus a => ArrowPlus (CoStateArrow s a) where
CST f <+> CST g = CST (f <+> g)