, MonadFix(..)
, MonadIO(..)
+
+ , ID, runID
, liftIO1, liftIO2, liftIO3, liftIO4
-
+
+ , zipWith3M
, mapAndUnzipM, mapAndUnzip3M, mapAndUnzip4M
, mapAccumLM
, mapSndM
, concatMapM
, mapMaybeM
+ , fmapMaybeM, fmapEitherM
, anyM, allM
- , foldlM, foldrM
+ , foldlM, foldlM_, foldrM
+ , maybeMapM
) where
-----------------------------------------------------------------------------------------
+import Outputable
+
+-------------------------------------------------------------------------------
-- Detection of available libraries
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-#if __GLASGOW_HASKELL__ >= 606
-#define HAVE_APPLICATIVE 1
-#else
-#define HAVE_APPLICATIVE 0
-#endif
-- we don't depend on MTL for now
#define HAVE_MTL 0
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-- Imports
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
import Maybes
-#if HAVE_APPLICATIVE
import Control.Applicative
-#endif
#if HAVE_MTL
import Control.Monad.Trans
#endif
import Control.Monad
import Control.Monad.Fix
-----------------------------------------------------------------------------------------
--- Applicative
-----------------------------------------------------------------------------------------
-
-#if !HAVE_APPLICATIVE
-
-class Functor f => Applicative f where
- pure :: a -> f a
- (<*>) :: f (a -> b) -> f a -> f b
+-------------------------------------------------------------------------------
+-- The ID monad
+-------------------------------------------------------------------------------
-(<$>) :: Functor f => (a -> b) -> (f a -> f b)
-(<$>) = fmap
+newtype ID a = ID a
+instance Monad ID where
+ return x = ID x
+ (ID x) >>= f = f x
+ _ >> y = y
+ fail s = panic s
-infixl 4 <$>
-infixl 4 <*>
+runID :: ID a -> a
+runID (ID x) = x
-instance Applicative IO where
- pure = return
- (<*>) = ap
-
-#endif
-
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-- MTL
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
#if !HAVE_MTL
instance MonadIO IO where liftIO = id
#endif
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-- Lift combinators
-- These are used throughout the compiler
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-- | Lift an 'IO' operation with 1 argument into another monad
liftIO1 :: MonadIO m => (a -> IO b) -> a -> m b
liftIO4 :: MonadIO m => (a -> b -> c -> d -> IO e) -> a -> b -> c -> d -> m e
liftIO4 = (((.).(.)).((.).(.))) liftIO
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
-- Common functions
-- These are used throughout the compiler
-----------------------------------------------------------------------------------------
+-------------------------------------------------------------------------------
+
+zipWith3M :: Monad m => (a -> b -> c -> m d) -> [a] -> [b] -> [c] -> m [d]
+zipWith3M _ [] _ _ = return []
+zipWith3M _ _ [] _ = return []
+zipWith3M _ _ _ [] = return []
+zipWith3M f (x:xs) (y:ys) (z:zs)
+ = do { r <- f x y z
+ ; rs <- zipWith3M f xs ys zs
+ ; return $ r:rs
+ }
-- | mapAndUnzipM for triples
mapAndUnzip3M :: Monad m => (a -> m (b,c,d)) -> [a] -> m ([b],[c],[d])
mapMaybeM :: (Monad m) => (a -> m (Maybe b)) -> [a] -> m [b]
mapMaybeM f = liftM catMaybes . mapM f
+-- | Monadic version of fmap
+fmapMaybeM :: (Monad m) => (a -> m b) -> Maybe a -> m (Maybe b)
+fmapMaybeM _ Nothing = return Nothing
+fmapMaybeM f (Just x) = f x >>= (return . Just)
+
+-- | Monadic version of fmap
+fmapEitherM :: Monad m => (a -> m b) -> (c -> m d) -> Either a c -> m (Either b d)
+fmapEitherM fl _ (Left a) = fl a >>= (return . Left)
+fmapEitherM _ fr (Right b) = fr b >>= (return . Right)
+
-- | Monadic version of 'any', aborts the computation at the first @True@ value
anyM :: Monad m => (a -> m Bool) -> [a] -> m Bool
anyM _ [] = return False
foldlM :: (Monad m) => (a -> b -> m a) -> a -> [b] -> m a
foldlM = foldM
+-- | Monadic version of foldl that discards its result
+foldlM_ :: (Monad m) => (a -> b -> m a) -> a -> [b] -> m ()
+foldlM_ = foldM_
+
-- | Monadic version of foldr
foldrM :: (Monad m) => (b -> a -> m a) -> a -> [b] -> m a
foldrM _ z [] = return z
-foldrM k z (x:xs) = do { r <- foldrM k z xs; k x r }
\ No newline at end of file
+foldrM k z (x:xs) = do { r <- foldrM k z xs; k x r }
+
+-- | Monadic version of fmap specialised for Maybe
+maybeMapM :: Monad m => (a -> m b) -> (Maybe a -> m (Maybe b))
+maybeMapM _ Nothing = return Nothing
+maybeMapM m (Just x) = liftM Just $ m x