e412e96a791635ed33d747e8e7333adffe175b50
[matthijs/master-project/cλash.git] / cλash / CLasH / Normalize / NormalizeTypes.hs
1 module CLasH.Normalize.NormalizeTypes where
2
3 -- Standard modules
4 import qualified Control.Monad.Trans.Writer as Writer
5 import qualified Data.Monoid as Monoid
6
7 -- GHC API
8 import qualified CoreSyn
9
10 -- Local imports
11 import CLasH.Translator.TranslatorTypes
12
13 -- Wrap a writer around a TranslatorSession, to run a single transformation
14 -- over a single expression and track if the expression was changed.
15 type TransformMonad = Writer.WriterT Monoid.Any TranslatorSession
16
17 -- | In what context does a core expression occur?
18 data CoreContext = AppFirst        -- ^ The expression is the first
19                                    -- argument of an application (i.e.,
20                                    -- it is applied)
21                  | AppSecond       -- ^ The expression is the second
22                                    --   argument of an application
23                                    --   (i.e., something is applied to it)
24                  | LetBinding      -- ^ The expression is bound in a
25                                    --   (recursive or non-recursive) let
26                                    --   expression.
27                  | LetBody         -- ^ The expression is the body of a
28                                    --   let expression
29                  | Other           -- ^ Another context
30
31 -- | Transforms a CoreExpr and keeps track if it has changed.
32 type Transform = [CoreContext] -> CoreSyn.CoreExpr -> TransformMonad CoreSyn.CoreExpr