{-|
Some convenience definitions.
-}
{-# LANGUAGE TypeOperators #-}
module Common where
import qualified Data.Map as M
import qualified Data.Set as S
import Control.Monad (liftM)
-- * Type aliases
-- | Convenient alias for a partial map
type a :⇀ b = M.Map a b
infixl 8 :⇀
-- | Convenient alias for a product type
type a :× b = (a,b)
infixl 9 :×
-- * Partial map functions
-- | A function used to write map updates resembling the mathematical syntax m [ k ↦ v] as m `upd` [k ↦ v]
upd :: (Ord k) => k :⇀a -> [k :× a] -> k :⇀ a
upd map list = M.union (M.fromList list) map
-- | A functorial variant of 'upd'
upd' :: (Ord k, Functor f) => f (k :⇀ a) -> [k :× a] -> f (k :⇀ a)
upd' map list = fmap (`upd` list) map
-- | For use in the argument list of 'upd'
(↦) :: k -> v -> k :× v
k ↦ v = (k,v)
-- | A monadically lifted unions
unionsM :: (Monad m, Ord k) => [m (k :⇀ a)] -> m (k :⇀ a)
unionsM = liftM M.unions . sequence