From: Ross Paterson <ross@soi.city.ac.uk>
Date: Mon, 15 Jan 2007 17:45:10 +0000 (+0000)
Subject: Applicative and Monad instances for Tree
X-Git-Url: http://git.megacz.com/?a=commitdiff_plain;h=2e927974423bf1f48385a829f95b7ffc6a271b2d;p=ghc-base.git

Applicative and Monad instances for Tree
---

diff --git a/Data/Tree.hs b/Data/Tree.hs
index a7aab07..c159a74 100644
--- a/Data/Tree.hs
+++ b/Data/Tree.hs
@@ -66,6 +66,16 @@ INSTANCE_TYPEABLE1(Tree,treeTc,"Tree")
 instance Functor Tree where
   fmap f (Node x ts) = Node (f x) (map (fmap f) ts)
 
+instance Applicative Tree where
+  pure x = Node x []
+  Node f tfs <*> tx@(Node x txs) =
+    Node (f x) (map (f <$>) txs ++ map (<*> tx) tfs)
+
+instance Monad Tree where
+  return x = Node x []
+  Node x ts >>= f = Node x' (ts' ++ map (>>= f) ts)
+    where Node x' ts' = f x
+
 instance Traversable Tree where
   traverse f (Node x ts) = Node <$> f x <*> traverse (traverse f) ts