Another round of External Core fixes
[ghc-hetmet.git] / compiler / main / HscMain.lhs
index 0152549..0b8a5a2 100644 (file)
@@ -7,8 +7,11 @@
 \begin{code}
 module HscMain
     ( newHscEnv, hscCmmFile
-    , hscFileCheck
     , hscParseIdentifier
+    , hscSimplify
+    , evalComp
+    , hscNormalIface, hscWriteIface, hscOneShot
+    , CompState (..)
 #ifdef GHCI
     , hscStmt, hscTcExpr, hscKcType
     , compileExpr
@@ -19,19 +22,24 @@ module HscMain
     , hscCompileInteractive -- :: Compiler (InteractiveStatus, ModIface, ModDetails)
     , HscStatus (..)
     , InteractiveStatus (..)
-    , HscChecked (..)
+
+    -- The new interface
+    , parseFile
+    , typecheckModule
+    , typecheckRenameModule
+    , deSugarModule
+    , makeSimpleIface
+    , makeSimpleDetails
     ) where
 
 #include "HsVersions.h"
 
 #ifdef GHCI
-import HsSyn           ( Stmt(..), LStmt, LHsType )
 import CodeOutput      ( outputForeignStubs )
 import ByteCodeGen     ( byteCodeGen, coreExprToBCOs )
 import Linker          ( HValue, linkExpr )
 import CoreTidy                ( tidyExpr )
 import CorePrep                ( corePrepExpr )
-import Flattening      ( flattenExpr )
 import Desugar          ( deSugarExpr )
 import SimplCore        ( simplifyExpr )
 import TcRnDriver      ( tcRnStmt, tcRnExpr, tcRnType ) 
@@ -47,25 +55,24 @@ import VarEnv               ( emptyTidyEnv )
 
 import Var             ( Id )
 import Module          ( emptyModuleEnv, ModLocation(..), Module )
-import RdrName         ( GlobalRdrEnv, RdrName, emptyGlobalRdrEnv )
-import HsSyn           ( HsModule, LHsBinds, HsGroup, LIE, LImportDecl, HsDoc,
-                          HaddockModInfo )
+import RdrName
+import HsSyn
 import CoreSyn
 import SrcLoc          ( Located(..) )
 import StringBuffer
 import Parser
 import Lexer
 import SrcLoc          ( mkSrcLoc )
-import TcRnDriver      ( tcRnModule, tcRnExtCore )
+import TcRnDriver      ( tcRnModule )
 import TcIface         ( typecheckIface )
 import TcRnMonad       ( initIfaceCheck, TcGblEnv(..) )
 import IfaceEnv                ( initNameCache )
 import LoadIface       ( ifaceStats, initExternalPackageState )
 import PrelInfo                ( wiredInThings, basicKnownKeyNames )
-import MkIface         ( checkOldIface, mkIface, writeIfaceFile )
+import MkIface
 import Desugar          ( deSugar )
 import SimplCore        ( core2core )
-import TidyPgm         ( tidyProgram, mkBootModDetails )
+import TidyPgm
 import CorePrep                ( corePrepPgm )
 import CoreToStg       ( coreToStg )
 import StgSyn
@@ -93,10 +100,8 @@ import Outputable
 import HscStats                ( ppSourceStats )
 import HscTypes
 import MkExternalCore  ( emitExternalCore )
-import ParserCore
-import ParserCoreUtils
 import FastString
-import UniqFM          ( emptyUFM )
+import LazyUniqFM              ( emptyUFM )
 import UniqSupply       ( initUs_ )
 import Bag             ( unitBag )
 
@@ -144,6 +149,85 @@ knownKeyNames = map getName wiredInThings
 \end{code}
 
 
+\begin{code}
+-- | parse a file, returning the abstract syntax
+parseFile :: HscEnv -> ModSummary -> IO (Maybe (Located (HsModule RdrName)))
+parseFile hsc_env mod_summary
+ = do 
+       maybe_parsed <- myParseModule dflags hspp_file hspp_buf
+       case maybe_parsed of
+         Left err
+             -> do printBagOfErrors dflags (unitBag err)
+                   return Nothing
+         Right rdr_module
+             -> return (Just rdr_module)
+  where
+           dflags    = hsc_dflags hsc_env
+           hspp_file = ms_hspp_file mod_summary
+           hspp_buf  = ms_hspp_buf  mod_summary
+
+-- | Rename and typecheck a module
+typecheckModule :: HscEnv -> ModSummary -> Located (HsModule RdrName)
+                -> IO (Maybe TcGblEnv)
+typecheckModule hsc_env mod_summary rdr_module
+ = do 
+        (tc_msgs, maybe_tc_result) 
+                <- {-# SCC "Typecheck-Rename" #-}
+                   tcRnModule hsc_env (ms_hsc_src mod_summary) False rdr_module
+        printErrorsAndWarnings dflags tc_msgs
+        return maybe_tc_result
+  where
+        dflags = hsc_dflags hsc_env
+
+type RenamedStuff = 
+        (Maybe (HsGroup Name, [LImportDecl Name], Maybe [LIE Name],
+                Maybe (HsDoc Name), HaddockModInfo Name))
+
+-- | Rename and typecheck a module, additinoally returning the renamed syntax
+typecheckRenameModule :: HscEnv -> ModSummary -> Located (HsModule RdrName)
+                -> IO (Maybe (TcGblEnv, RenamedStuff))
+typecheckRenameModule hsc_env mod_summary rdr_module
+ = do 
+        (tc_msgs, maybe_tc_result) 
+                <- {-# SCC "Typecheck-Rename" #-}
+                   tcRnModule hsc_env (ms_hsc_src mod_summary) True rdr_module
+        printErrorsAndWarnings dflags tc_msgs
+        case maybe_tc_result of
+           Nothing -> return Nothing
+           Just tc_result -> do
+              let rn_info = do decl <- tcg_rn_decls tc_result
+                               imports <- tcg_rn_imports tc_result
+                               let exports = tcg_rn_exports tc_result
+                              let doc = tcg_doc tc_result
+                              let hmi = tcg_hmi tc_result
+                               return (decl,imports,exports,doc,hmi)
+              return (Just (tc_result, rn_info))
+  where
+        dflags = hsc_dflags hsc_env
+
+-- | Convert a typechecked module to Core
+deSugarModule :: HscEnv -> ModSummary -> TcGblEnv -> IO (Maybe ModGuts)
+deSugarModule hsc_env mod_summary tc_result
+ = deSugar hsc_env (ms_location mod_summary) tc_result
+
+-- | Make a 'ModIface' from the results of typechecking.  Used when
+-- not optimising, and the interface doesn't need to contain any
+-- unfoldings or other cross-module optimisation info.
+-- ToDo: the old interface is only needed to get the version numbers,
+-- we should use fingerprint versions instead.
+makeSimpleIface :: HscEnv -> Maybe ModIface -> TcGblEnv -> ModDetails
+                -> IO (ModIface,Bool)
+makeSimpleIface hsc_env maybe_old_iface tc_result details = do
+  mkIfaceTc hsc_env maybe_old_iface details tc_result
+
+-- | Make a 'ModDetails' from the results of typechecking.  Used when
+-- typechecking only, as opposed to full compilation.
+makeSimpleDetails :: HscEnv -> TcGblEnv -> IO ModDetails
+makeSimpleDetails hsc_env tc_result = mkBootModDetailsTc hsc_env tc_result
+
+-- deSugarModule :: HscEnv -> TcGblEnv -> IO Core
+\end{code}
+
 %************************************************************************
 %*                                                                     *
                The main compiler pipeline
@@ -183,18 +267,6 @@ error. This is the only thing that isn't caught by the type-system.
 
 \begin{code}
 
-data HscChecked
-    = HscChecked
-        -- parsed
-        (Located (HsModule RdrName))
-        -- renamed
-        (Maybe (HsGroup Name, [LImportDecl Name], Maybe [LIE Name],
-                Maybe (HsDoc Name), HaddockModInfo Name))
-        -- typechecked
-        (Maybe (LHsBinds Id, GlobalRdrEnv, ModDetails))
-        -- desugared
-        (Maybe [CoreBind])
-
 -- Status of a compilation to hard-code or nothing.
 data HscStatus
     = HscNoRecomp
@@ -256,41 +328,6 @@ type Compiler result =  HscEnv
                      -> Maybe (Int,Int)     -- Just (i,n) <=> module i of n (for msgs)
                      -> IO (Maybe result)
 
-
--- This functions checks if recompilation is necessary and
--- then combines the FrontEnd and BackEnd to a working compiler.
-hscMkCompiler :: NoRecomp result         -- What to do when recompilation isn't required.
-              -> (Maybe (Int,Int) -> Bool -> Comp ())
-              -> Comp (Maybe ModGuts)       -- Front end
-              -> (ModGuts -> Comp result)   -- Backend.
-              -> Compiler result
-hscMkCompiler norecomp messenger frontend backend
-              hsc_env mod_summary source_unchanged
-              mbOldIface mbModIndex
-    = flip evalComp (CompState hsc_env mod_summary mbOldIface) $
-      do (recomp_reqd, mbCheckedIface)
-             <- {-# SCC "checkOldIface" #-}
-                liftIO $ checkOldIface hsc_env mod_summary
-                              source_unchanged mbOldIface
-        -- save the interface that comes back from checkOldIface.
-        -- In one-shot mode we don't have the old iface until this
-        -- point, when checkOldIface reads it from the disk.
-        modify (\s -> s{ compOldIface = mbCheckedIface })
-         case mbCheckedIface of 
-           Just iface | not recomp_reqd
-               -> do messenger mbModIndex False
-                     result <- norecomp iface
-                     return (Just result)
-           _otherwise
-               -> do messenger mbModIndex True
-                     mbCore <- frontend
-                     case mbCore of
-                       Nothing
-                           -> return Nothing
-                       Just core
-                           -> do result <- backend core
-                                 return (Just result)
-
 --------------------------------------------------------------
 -- Compilers
 --------------------------------------------------------------
@@ -298,49 +335,91 @@ hscMkCompiler norecomp messenger frontend backend
 -- Compile Haskell, boot and extCore in OneShot mode.
 hscCompileOneShot :: Compiler HscStatus
 hscCompileOneShot
-   = hscCompiler norecompOneShot oneShotMsg backend boot_backend
+   = hscCompiler norecompOneShot oneShotMsg (genComp backend boot_backend)
    where
      backend inp  = hscSimplify inp >>= hscNormalIface >>= hscWriteIface >>= hscOneShot
-     boot_backend inp = hscSimpleIface inp >>= hscWriteIface >> return (HscRecomp False)
+     boot_backend inp = hscSimpleIface inp >>= hscWriteIface >> return (Just (HscRecomp False))
 
 -- Compile Haskell, boot and extCore in batch mode.
 hscCompileBatch :: Compiler (HscStatus, ModIface, ModDetails)
 hscCompileBatch
-   = hscCompiler norecompBatch batchMsg backend boot_backend
+   = hscCompiler norecompBatch batchMsg (genComp backend boot_backend)
    where
      backend inp  = hscSimplify inp >>= hscNormalIface >>= hscWriteIface >>= hscBatch
      boot_backend inp = hscSimpleIface inp >>= hscWriteIface >>= hscNothing
 
--- Type-check Haskell, boot and extCore.
--- Does it make sense to compile extCore to nothing?
-hscCompileNothing :: Compiler (HscStatus, ModIface, ModDetails)
-hscCompileNothing
-   = hscCompiler norecompBatch batchMsg backend backend
-   where
-     backend inp = hscSimpleIface inp >>= hscIgnoreIface >>= hscNothing
-
 -- Compile Haskell, extCore to bytecode.
 hscCompileInteractive :: Compiler (InteractiveStatus, ModIface, ModDetails)
 hscCompileInteractive
-   = hscCompiler norecompInteractive batchMsg backend boot_backend
+   = hscCompiler norecompInteractive batchMsg (genComp backend boot_backend)
    where
      backend inp = hscSimplify inp >>= hscNormalIface >>= hscIgnoreIface >>= hscInteractive
-     boot_backend = panic "hscCompileInteractive: can't do boot files here"
+     boot_backend _ = panic "hscCompileInteractive: HsBootFile"
 
+-- Type-check Haskell and .hs-boot only (no external core)
+hscCompileNothing :: Compiler (HscStatus, ModIface, ModDetails)
+hscCompileNothing
+   = hscCompiler norecompBatch batchMsg comp
+   where
+     backend tc = hscSimpleIface tc >>= hscIgnoreIface >>= hscNothing
+
+     comp = do   -- genComp doesn't fit here, because we want to omit
+                 -- desugaring and for the backend to take a TcGblEnv
+        mod_summary <- gets compModSummary
+        case ms_hsc_src mod_summary of
+           ExtCoreFile -> panic "hscCompileNothing: cannot do external core"
+           _other -> do
+                mb_tc <- hscFileFrontEnd
+                case mb_tc of
+                  Nothing -> return Nothing
+                  Just tc_result -> backend tc_result
+        
 hscCompiler
-        :: NoRecomp result                                  -- No recomp necessary
-        -> (Maybe (Int,Int) -> Bool -> Comp ())             -- Message callback
-        -> (ModGuts -> Comp result)  -- Compile normal file
-        -> (ModGuts -> Comp result) -- Compile boot file
+        :: NoRecomp result                       -- No recomp necessary
+        -> (Maybe (Int,Int) -> Bool -> Comp ())  -- Message callback
+        -> Comp (Maybe result)
         -> Compiler result
-hscCompiler norecomp msg nonBootComp bootComp hsc_env mod_summary =
-    hscMkCompiler norecomp msg frontend backend hsc_env mod_summary
-    where
-          (frontend,backend)
-              = case ms_hsc_src mod_summary of
-                ExtCoreFile -> (hscCoreFrontEnd, nonBootComp)
-                HsSrcFile   -> (hscFileFrontEnd, nonBootComp)
-                HsBootFile  -> (hscFileFrontEnd, bootComp)
+hscCompiler norecomp messenger recomp hsc_env mod_summary 
+            source_unchanged mbOldIface mbModIndex
+    = flip evalComp (CompState hsc_env mod_summary mbOldIface) $
+      do (recomp_reqd, mbCheckedIface)
+             <- {-# SCC "checkOldIface" #-}
+                liftIO $ checkOldIface hsc_env mod_summary
+                              source_unchanged mbOldIface
+        -- save the interface that comes back from checkOldIface.
+        -- In one-shot mode we don't have the old iface until this
+        -- point, when checkOldIface reads it from the disk.
+        modify (\s -> s{ compOldIface = mbCheckedIface })
+         case mbCheckedIface of 
+           Just iface | not recomp_reqd
+               -> do messenger mbModIndex False
+                     result <- norecomp iface
+                     return (Just result)
+           _otherwise
+               -> do messenger mbModIndex True
+                     recomp
+
+-- the usual way to build the Comp (Maybe result) to pass to hscCompiler
+genComp :: (ModGuts  -> Comp (Maybe a))
+        -> (TcGblEnv -> Comp (Maybe a))
+        -> Comp (Maybe a)
+genComp backend boot_backend = do
+    mod_summary <- gets compModSummary
+    case ms_hsc_src mod_summary of
+       ExtCoreFile -> do
+          panic "GHC does not currently support reading External Core files"
+       _not_core -> do
+          mb_tc <- hscFileFrontEnd
+          case mb_tc of
+            Nothing -> return Nothing
+            Just tc_result -> 
+              case ms_hsc_src mod_summary of
+                HsBootFile -> boot_backend tc_result
+                _other     -> do
+                  mb_guts <- hscDesugar tc_result
+                  case mb_guts of
+                    Nothing -> return Nothing
+                    Just guts -> backend guts
 
 --------------------------------------------------------------
 -- NoRecomp handlers
@@ -362,7 +441,6 @@ norecompInteractive = norecompWorker InteractiveNoRecomp True
 norecompWorker :: a -> Bool -> NoRecomp (a, ModIface, ModDetails)
 norecompWorker a _isInterp old_iface
     = do hsc_env <- gets compHscEnv
-         _mod_summary <- gets compModSummary
          liftIO $ do
          new_details <- {-# SCC "tcRnIface" #-}
                         initIfaceCheck hsc_env $
@@ -400,33 +478,7 @@ batchMsg mb_mod_index recomp
 --------------------------------------------------------------
 -- FrontEnds
 --------------------------------------------------------------
-
-hscCoreFrontEnd :: Comp (Maybe ModGuts)
-hscCoreFrontEnd =
-    do hsc_env <- gets compHscEnv
-       mod_summary <- gets compModSummary
-       liftIO $ do
-            -------------------
-            -- PARSE
-            -------------------
-       inp <- readFile (ms_hspp_file mod_summary)
-       case parseCore inp 1 of
-         FailP s
-             -> do errorMsg (hsc_dflags hsc_env) (text s{-ToDo: wrong-})
-                   return Nothing
-         OkP rdr_module
-             -------------------
-             -- RENAME and TYPECHECK
-             -------------------
-             -> do (tc_msgs, maybe_tc_result) <- {-# SCC "TypeCheck" #-}
-                                                 tcRnExtCore hsc_env rdr_module
-                   printErrorsAndWarnings (hsc_dflags hsc_env) tc_msgs
-                   case maybe_tc_result of
-                     Nothing       -> return Nothing
-                     Just mod_guts -> return (Just mod_guts)         -- No desugaring to do!
-
-        
-hscFileFrontEnd :: Comp (Maybe ModGuts)
+hscFileFrontEnd :: Comp (Maybe TcGblEnv)
 hscFileFrontEnd =
     do hsc_env <- gets compHscEnv
        mod_summary <- gets compModSummary
@@ -450,14 +502,23 @@ hscFileFrontEnd =
                        <- {-# SCC "Typecheck-Rename" #-}
                           tcRnModule hsc_env (ms_hsc_src mod_summary) False rdr_module
                    printErrorsAndWarnings dflags tc_msgs
-                   case maybe_tc_result of
-                     Nothing
-                         -> return Nothing
-                     Just tc_result
-                         -------------------
-                         -- DESUGAR
-                         -------------------
-                         -> {-# SCC "DeSugar" #-} deSugar hsc_env (ms_location mod_summary) tc_result
+                   return maybe_tc_result
+
+--------------------------------------------------------------
+-- Desugaring
+--------------------------------------------------------------
+
+hscDesugar :: TcGblEnv -> Comp (Maybe ModGuts)
+hscDesugar tc_result
+  = do mod_summary <- gets compModSummary
+       hsc_env <- gets compHscEnv
+       liftIO $ do
+          -------------------
+          -- DESUGAR
+          -------------------
+       ds_result   <- {-# SCC "DeSugar" #-} 
+                      deSugar hsc_env (ms_location mod_summary) tc_result
+       return ds_result
 
 --------------------------------------------------------------
 -- Simplifiers
@@ -481,19 +542,18 @@ hscSimplify ds_result
 -- HACK: we return ModGuts even though we know it's not gonna be used.
 --       We do this because the type signature needs to be identical
 --       in structure to the type of 'hscNormalIface'.
-hscSimpleIface :: ModGuts -> Comp (ModIface, Bool, ModDetails, ModGuts)
-hscSimpleIface ds_result
+hscSimpleIface :: TcGblEnv -> Comp (ModIface, Bool, ModDetails, TcGblEnv)
+hscSimpleIface tc_result
   = do hsc_env <- gets compHscEnv
-       _mod_summary <- gets compModSummary
        maybe_old_iface <- gets compOldIface
        liftIO $ do
-       details <- mkBootModDetails hsc_env ds_result
+       details <- mkBootModDetailsTc hsc_env tc_result
        (new_iface, no_change) 
            <- {-# SCC "MkFinalIface" #-}
-              mkIface hsc_env maybe_old_iface ds_result details
+              mkIfaceTc hsc_env maybe_old_iface details tc_result
        -- And the answer is ...
        dumpIfaceStats hsc_env
-       return (new_iface, no_change, details, ds_result)
+       return (new_iface, no_change, details, tc_result)
 
 hscNormalIface :: ModGuts -> Comp (ModIface, Bool, ModDetails, CgGuts)
 hscNormalIface simpl_result
@@ -515,9 +575,12 @@ hscNormalIface simpl_result
            -- until after code output
        (new_iface, no_change)
                <- {-# SCC "MkFinalIface" #-}
-                  mkIface hsc_env maybe_old_iface simpl_result details
+                  mkIface hsc_env maybe_old_iface details simpl_result
        -- Emit external core
-       emitExternalCore (hsc_dflags hsc_env) (availsToNameSet (mg_exports simpl_result)) cg_guts -- Move this? --Lemmih 03/07/2006
+       -- This should definitely be here and not after CorePrep,
+       -- because CorePrep produces unqualified constructor wrapper declarations,
+       -- so its output isn't valid External Core (without some preprocessing).
+       emitExternalCore (hsc_dflags hsc_env) cg_guts 
        dumpIfaceStats hsc_env
 
            -------------------
@@ -543,21 +606,21 @@ hscIgnoreIface (iface, _no_change, details, a)
     = return (iface, details, a)
 
 -- Don't output any code.
-hscNothing :: (ModIface, ModDetails, a) -> Comp (HscStatus, ModIface, ModDetails)
+hscNothing :: (ModIface, ModDetails, a) -> Comp (Maybe (HscStatus, ModIface, ModDetails))
 hscNothing (iface, details, _)
-    = return (HscRecomp False, iface, details)
+    = return (Just (HscRecomp False, iface, details))
 
 -- Generate code and return both the new ModIface and the ModDetails.
-hscBatch :: (ModIface, ModDetails, CgGuts) -> Comp (HscStatus, ModIface, ModDetails)
+hscBatch :: (ModIface, ModDetails, CgGuts) -> Comp (Maybe (HscStatus, ModIface, ModDetails))
 hscBatch (iface, details, cgguts)
     = do hasStub <- hscCompile cgguts
-         return (HscRecomp hasStub, iface, details)
+         return (Just (HscRecomp hasStub, iface, details))
 
 -- Here we don't need the ModIface and ModDetails anymore.
-hscOneShot :: (ModIface, ModDetails, CgGuts) -> Comp HscStatus
+hscOneShot :: (ModIface, ModDetails, CgGuts) -> Comp (Maybe HscStatus)
 hscOneShot (_, _, cgguts)
     = do hasStub <- hscCompile cgguts
-         return (HscRecomp hasStub)
+         return (Just (HscRecomp hasStub))
 
 -- Compile to hard-code.
 hscCompile :: CgGuts -> Comp Bool
@@ -594,17 +657,9 @@ hscCompile cgguts
                       codeGen dflags this_mod data_tycons
                               dir_imps cost_centre_info
                               stg_binds hpc_info
-         --------  Optionally convert to and from zipper ------
-         cmms <-
-             if dopt Opt_ConvertToZipCfgAndBack dflags
-             then mapM (testCmmConversion dflags) cmms
-             else return cmms
-         ------------  Optionally convert to CPS --------------
-         cmms <-
-             if not (dopt Opt_ConvertToZipCfgAndBack dflags) &&
-                dopt Opt_RunCPSZ dflags
-             then cmmCPS dflags cmms
-             else return cmms
+         --- Optionally run experimental Cmm transformations ---
+         cmms <- optionallyConvertAndOrCPS dflags cmms
+                 -- ^ unless certain dflags are on, the identity function
          ------------------  Code output -----------------------
          rawcmms <- cmmToRawCmm cmms
          (_stub_h_exists, stub_c_exists)
@@ -613,7 +668,7 @@ hscCompile cgguts
          return stub_c_exists
 
 hscInteractive :: (ModIface, ModDetails, CgGuts)
-               -> Comp (InteractiveStatus, ModIface, ModDetails)
+               -> Comp (Maybe (InteractiveStatus, ModIface, ModDetails))
 #ifdef GHCI
 hscInteractive (iface, details, cgguts)
     = do hsc_env <- gets compHscEnv
@@ -642,85 +697,22 @@ hscInteractive (iface, details, cgguts)
          ------------------ Create f-x-dynamic C-side stuff ---
          (_istub_h_exists, istub_c_exists) 
              <- outputForeignStubs dflags this_mod location foreign_stubs
-         return (InteractiveRecomp istub_c_exists comp_bc mod_breaks, iface, details)
+         return (Just (InteractiveRecomp istub_c_exists comp_bc mod_breaks, iface, details))
 #else
 hscInteractive _ = panic "GHC not compiled with interpreter"
 #endif
 
 ------------------------------
 
-hscFileCheck :: HscEnv -> ModSummary -> Bool -> IO (Maybe HscChecked)
-hscFileCheck hsc_env mod_summary compileToCore = do {
-           -------------------
-           -- PARSE
-           -------------------
-       ; let dflags    = hsc_dflags hsc_env
-             hspp_file = ms_hspp_file mod_summary
-             hspp_buf  = ms_hspp_buf  mod_summary
-
-       ; maybe_parsed <- myParseModule dflags hspp_file hspp_buf
-
-       ; case maybe_parsed of {
-            Left err -> do { printBagOfErrors dflags (unitBag err)
-                           ; return Nothing } ;
-            Right rdr_module -> do {
-
-           -------------------
-           -- RENAME and TYPECHECK
-           -------------------
-         (tc_msgs, maybe_tc_result) 
-               <- {-# SCC "Typecheck-Rename" #-}
-                  tcRnModule hsc_env (ms_hsc_src mod_summary) 
-                       True{-save renamed syntax-}
-                       rdr_module
-
-       ; printErrorsAndWarnings dflags tc_msgs
-       ; case maybe_tc_result of {
-            Nothing -> return (Just (HscChecked rdr_module Nothing Nothing Nothing));
-            Just tc_result -> do
-               let type_env = tcg_type_env tc_result
-                   md = ModDetails { 
-                               md_types     = type_env,
-                               md_exports   = tcg_exports   tc_result,
-                               md_insts     = tcg_insts     tc_result,
-                               md_fam_insts = tcg_fam_insts tc_result,
-                               md_rules     = [panic "no rules"],
-                                  -- Rules are CoreRules, not the
-                                  -- RuleDecls we get out of the typechecker
-                                md_vect_info = noVectInfo
-                                   -- VectInfo is added by the Core 
-                                   -- vectorisation pass
-                          }
-                    rnInfo = do decl <- tcg_rn_decls tc_result
-                                imports <- tcg_rn_imports tc_result
-                                let exports = tcg_rn_exports tc_result
-                               let doc = tcg_doc tc_result
-                                   hmi = tcg_hmi tc_result
-                                return (decl,imports,exports,doc,hmi)
-               maybeModGuts <- 
-                 if compileToCore then
-                   deSugar hsc_env (ms_location mod_summary) tc_result
-                 else
-                   return Nothing
-                return (Just (HscChecked rdr_module 
-                                   rnInfo
-                                  (Just (tcg_binds tc_result,
-                                         tcg_rdr_env tc_result,
-                                         md))
-                                   (fmap mg_binds maybeModGuts)))
-       }}}}
-
-
 hscCmmFile :: DynFlags -> FilePath -> IO Bool
 hscCmmFile dflags filename = do
   maybe_cmm <- parseCmmFile dflags filename
   case maybe_cmm of
     Nothing -> return False
     Just cmm -> do
-        cmm <- testCmmConversion dflags cmm
-        --continuationC <- cmmCPS dflags cmm >>= cmmToRawCmm
-        continuationC <- cmmToRawCmm [cmm]
-       codeOutput dflags no_mod no_loc NoStubs [] continuationC
+        cmms <- optionallyConvertAndOrCPS dflags [cmm]
+        rawCmms <- cmmToRawCmm cmms
+       codeOutput dflags no_mod no_loc NoStubs [] rawCmms
        return True
   where
        no_mod = panic "hscCmmFile: no_mod"
@@ -728,6 +720,20 @@ hscCmmFile dflags filename = do
                               ml_hi_file  = panic "hscCmmFile: no hi file",
                               ml_obj_file = panic "hscCmmFile: no obj file" }
 
+optionallyConvertAndOrCPS :: DynFlags -> [Cmm] -> IO [Cmm]
+optionallyConvertAndOrCPS dflags cmms =
+    do   --------  Optionally convert to and from zipper ------
+       cmms <- if dopt Opt_ConvertToZipCfgAndBack dflags
+               then mapM (testCmmConversion dflags) cmms
+               else return cmms
+         ---------  Optionally convert to CPS (MDA) -----------
+       cmms <- if not (dopt Opt_ConvertToZipCfgAndBack dflags) &&
+                  dopt Opt_RunCPSZ dflags
+               then cmmCPS dflags cmms
+               else return cmms
+       return cmms
+
+
 testCmmConversion :: DynFlags -> Cmm -> IO Cmm
 testCmmConversion dflags cmm =
     do showPass dflags "CmmToCmm"
@@ -957,11 +963,8 @@ compileExpr hsc_env srcspan ds_expr
   = do { let { dflags  = hsc_dflags hsc_env ;
                lint_on = dopt Opt_DoCoreLinting dflags }
              
-               -- Flatten it
-       ; flat_expr <- flattenExpr hsc_env ds_expr
-
                -- Simplify it
-       ; simpl_expr <- simplifyExpr dflags flat_expr
+       ; simpl_expr <- simplifyExpr dflags ds_expr
 
                -- Tidy it (temporary, until coreSat does cloning)
        ; let tidy_expr = tidyExpr emptyTidyEnv simpl_expr