{-|
    Compos compatibility layer. This module serves as a drop-in
    replacement in some situations for some of the Compos operations.
    Only the single-type traversals are supported, on normal
    algebraic data types. Users should also import either "Data.Generics.Uniplate.Data"
    or "Data.Generics.Uniplate.Direct".

    Compos is described in the paper: \"A Pattern for Almost Compositional Functions\"
    by Bjorn Bringert and Aarne Ranta.

    * <http://doi.acm.org/10.1145/1159803.1159834>

    * <http://www.cs.chalmers.se/~bringert/publ/composOp/composOp.pdf>
-}

module Data.Generics.Compos where

import Control.Monad
import Data.Generics.Uniplate.Operations


-- | If you want to keep an existing type class
class Uniplate a => Compos a where


-- | @composOp == 'descend'@
composOp :: Uniplate a => (a -> a) -> a -> a
composOp :: forall a. Uniplate a => (a -> a) -> a -> a
composOp = forall a. Uniplate a => (a -> a) -> a -> a
descend


-- | @composOpM == 'descendM'@
composOpM :: (Uniplate a, Applicative m) => (a -> m a) -> a -> m a
composOpM :: forall a (m :: * -> *).
(Uniplate a, Applicative m) =>
(a -> m a) -> a -> m a
composOpM = forall on (m :: * -> *).
(Uniplate on, Applicative m) =>
(on -> m on) -> on -> m on
descendM


-- | @composOpM_ == 'composOpFold' (return ()) (>>)@
composOpM_ :: (Uniplate a, Monad m) => (a -> m ()) -> a -> m ()
composOpM_ :: forall a (m :: * -> *).
(Uniplate a, Monad m) =>
(a -> m ()) -> a -> m ()
composOpM_ = forall a b. Uniplate a => b -> (b -> b -> b) -> (a -> b) -> a -> b
composOpFold (forall (m :: * -> *) a. Monad m => a -> m a
return ()) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
(>>)


-- | @composOpMonoid = 'composOpFold' mempty mappend@
composOpMonoid :: (Uniplate a, Monoid m) => (a -> m) -> a -> m
composOpMonoid :: forall a m. (Uniplate a, Monoid m) => (a -> m) -> a -> m
composOpMonoid = forall a b. Uniplate a => b -> (b -> b -> b) -> (a -> b) -> a -> b
composOpFold forall a. Monoid a => a
mempty forall a. Monoid a => a -> a -> a
mappend


-- | @composOpMPlus = 'composOpFold' mzero mplus@
composOpMPlus :: (Uniplate a, MonadPlus m) => (a -> m b) -> a -> m b
composOpMPlus :: forall a (m :: * -> *) b.
(Uniplate a, MonadPlus m) =>
(a -> m b) -> a -> m b
composOpMPlus = forall a b. Uniplate a => b -> (b -> b -> b) -> (a -> b) -> a -> b
composOpFold forall (m :: * -> *) a. MonadPlus m => m a
mzero forall (m :: * -> *) a. MonadPlus m => m a -> m a -> m a
mplus


-- | Probably replace with 'universe', perhaps 'para'
composOpFold :: Uniplate a => b -> (b -> b -> b) -> (a -> b) -> a -> b
composOpFold :: forall a b. Uniplate a => b -> (b -> b -> b) -> (a -> b) -> a -> b
composOpFold b
zero b -> b -> b
combine a -> b
f = forall (t :: * -> *) a b.
Foldable t =>
(a -> b -> b) -> b -> t a -> b
foldr b -> b -> b
combine b
zero forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
map a -> b
f forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall on. Uniplate on => on -> [on]
children