Permalink
7f58697 Dec 19, 2012
26 lines (21 sloc) 884 Bytes
{-# LANGUAGE MultiParamTypeClasses, DeriveGeneric, DeriveDataTypeable #-}
import Control.Applicative
import Control.Lens
import GHC.Generics
import Data.Data
import Data.Data.Lens
data Expr = Var Int | Pos Expr String | Neg Expr | Add Expr Expr deriving (Eq,Ord,Show,Read,Generic,Data,Typeable)
data Stmt = Seq [Stmt] | Sel [Expr] | Let String Expr deriving (Eq,Ord,Show,Read,Generic,Data,Typeable)
instance Plated Expr where
plate f (Var x ) = pure (Var x)
plate f (Pos x y) = Pos <$> f x <*> pure y
plate f (Neg x ) = Neg <$> f x
plate f (Add x y) = Add <$> f x <*> f y
instance Plated Stmt where
plate f (Seq xs) = Seq <$> traverse f xs
plate f (Sel xs) = pure (Sel xs)
plate f (Let x y) = pure (Let x y)
exprs :: Traversal' Stmt Expr
exprs f (Seq xs) = Seq <$> traverse (exprs f) xs
exprs f (Sel xs) = Sel <$> traverse f xs
exprs f (Let x y) = Let x <$> f y