[project @ 2002-04-05 15:18:25 by sof]
[ghc-hetmet.git] / ghc / compiler / main / HscMain.lhs
index 5f82eaf..5739163 100644 (file)
@@ -5,7 +5,7 @@
 \section[GHC_Main]{Main driver for Glasgow Haskell compiler}
 
 \begin{code}
-module HscMain ( HscResult(..), hscMain, 
+module HscMain ( HscResult(..), hscMain,
 #ifdef GHCI
                 hscStmt, hscThing, hscModuleContents,
 #endif
@@ -16,9 +16,9 @@ module HscMain ( HscResult(..), hscMain,
 #ifdef GHCI
 import Interpreter
 import ByteCodeGen     ( byteCodeGen )
-import CoreTidy                ( tidyCoreExpr )
+import TidyPgm         ( tidyCoreExpr )
 import CorePrep                ( corePrepExpr )
-import Rename          ( renameStmt, renameRdrName, slurpIface )
+import Rename          ( renameStmt,    renameRdrName, slurpIface )
 import RdrName          ( rdrNameOcc, setRdrNameOcc )
 import RdrHsSyn                ( RdrNameStmt )
 import OccName          ( dataName, tcClsName, 
@@ -26,13 +26,12 @@ import OccName          ( dataName, tcClsName,
 import Type            ( Type )
 import Id              ( Id, idName, setGlobalIdDetails )
 import IdInfo          ( GlobalIdDetails(VanillaGlobal) )
-import Name            ( isLocalName )
+import Name            ( isInternalName )
 import NameEnv         ( lookupNameEnv )
 import Module          ( lookupModuleEnv )
 import RdrName         ( rdrEnvElts )
 import PrelNames       ( iNTERACTIVE )
 import StringBuffer    ( stringToStringBuffer )
-import FastString       ( mkFastString )
 import Maybes          ( catMaybes )
 
 import List            ( nub )
@@ -48,11 +47,12 @@ import Parser
 import Lex             ( ParseResult(..), ExtFlags(..), mkPState )
 import SrcLoc          ( mkSrcLoc )
 import Finder          ( findModule )
-import Rename          ( checkOldIface, renameModule, closeIfaceDecls )
+import Rename          ( checkOldIface, renameModule, renameExtCore, 
+                         closeIfaceDecls, RnResult(..) )
 import Rules           ( emptyRuleBase )
 import PrelInfo                ( wiredInThingEnv, wiredInThings )
 import PrelRules       ( builtinRules )
-import PrelNames       ( knownKeyNames )
+import PrelNames       ( knownKeyNames, gHC_PRIM_Name )
 import MkIface         ( mkFinalIface )
 import TcModule
 import InstEnv         ( emptyInstEnv )
@@ -60,17 +60,18 @@ import Desugar
 import Flattening       ( flatten, flattenExpr )
 import SimplCore
 import CoreUtils       ( coreBindsSize )
-import CoreTidy                ( tidyCorePgm )
+import TidyPgm         ( tidyCorePgm )
 import CorePrep                ( corePrepPgm )
 import StgSyn
 import CoreToStg       ( coreToStg )
 import SimplStg                ( stg2stg )
 import CodeGen         ( codeGen )
-import CodeOutput      ( codeOutput )
+import CodeOutput      ( codeOutput, outputForeignStubs )
 
 import Module          ( ModuleName, moduleName, mkHomeModule )
 import CmdLineOpts
 import DriverState     ( v_HCHeader )
+import DriverPhases     ( isExtCore_file )
 import ErrUtils                ( dumpIfSet_dyn, showPass, printError )
 import Util            ( unJust )
 import UniqSupply      ( mkSplitUniqSupply )
@@ -93,6 +94,9 @@ import Maybe          ( isJust, fromJust )
 import IO
 
 import MkExternalCore  ( emitExternalCore )
+import ParserCore
+import ParserCoreUtils
+
 \end{code}
 
 
@@ -202,50 +206,23 @@ hscRecomp ghci_mode dflags have_object
          mod location maybe_checked_iface hst hit pcs_ch
  = do  {
          -- what target are we shooting for?
-       ; let toInterp = dopt_HscLang dflags == HscInterpreted
+       ; let toInterp  = dopt_HscLang dflags == HscInterpreted
        ; let toNothing = dopt_HscLang dflags == HscNothing
+       ; let toCore    = isJust (ml_hs_file location) &&
+                         isExtCore_file (fromJust (ml_hs_file location))
 
        ; when (ghci_mode /= OneShot && verbosity dflags >= 1) $
                hPutStrLn stderr ("Compiling " ++ 
                        showModMsg (not toInterp) mod location);
-
-           -------------------
-           -- PARSE
-           -------------------
-       ; maybe_parsed <- myParseModule dflags 
-                             (unJust "hscRecomp:hspp" (ml_hspp_file location))
-       ; case maybe_parsed of {
-            Nothing -> return (HscFail pcs_ch);
-            Just rdr_module -> do {
-       ; let this_mod = mkHomeModule (hsModuleName rdr_module)
-    
-           -------------------
-           -- RENAME
-           -------------------
-       ; (pcs_rn, print_unqual, maybe_rn_result) 
-            <- _scc_ "Rename" 
-                renameModule dflags ghci_mode hit hst pcs_ch this_mod rdr_module
-       ; case maybe_rn_result of {
-            Nothing -> return (HscFail pcs_ch);
-            Just (dont_discard, new_iface, rn_result) -> do {
-
-           -------------------
-           -- TYPECHECK
-           -------------------
-       ; maybe_tc_result 
-           <- _scc_ "TypeCheck" 
-              typecheckModule dflags pcs_rn hst print_unqual rn_result
-       ; case maybe_tc_result of {
-            Nothing -> return (HscFail pcs_ch);
-            Just (pcs_tc, tc_result) -> do {
-    
-           -------------------
-           -- DESUGAR
-           -------------------
-       ; (ds_details, foreign_stuff) 
-             <- _scc_ "DeSugar" 
-               deSugar dflags pcs_tc hst this_mod print_unqual tc_result
-
+                       
+       ; front_res <- 
+               (if toCore then hscCoreFrontEnd else hscFrontEnd)
+                  ghci_mode dflags location hst hit pcs_ch
+       ; case front_res of
+           Left flure -> return flure;
+           Right (this_mod, rdr_module, 
+                  Just (dont_discard, new_iface, rn_result), 
+                  pcs_tc, ds_details, foreign_stuff) -> do {
            -------------------
            -- FLATTENING
            -------------------
@@ -327,7 +304,11 @@ hscRecomp ghci_mode dflags have_object
            local_tycons     = typeEnvTyCons  env_tc
            local_classes    = typeEnvClasses env_tc
 
-           imported_module_names = map ideclName (hsModuleImports rdr_module)
+           imported_module_names = 
+               filter (/= gHC_PRIM_Name) $
+               map ideclName (hsModuleImports rdr_module)
+               -- eek! doesn't this keep rdr_module live until code generation?
+               -- SDM 3/2002
 
            mod_name_to_Module nm
                 = do m <- findModule nm ; return (fst (fromJust m))
@@ -368,7 +349,12 @@ hscRecomp ghci_mode dflags have_object
                          mkFinalIface ghci_mode dflags location 
                                    maybe_checked_iface new_iface tidy_details
 
-                   return ( False, False, Just (bcos,itbl_env), final_iface )
+                   ------------------ Create f-x-dynamic C-side stuff ---
+                    (istub_h_exists, istub_c_exists) 
+                       <- outputForeignStubs dflags c_code h_code
+
+                   return ( istub_h_exists, istub_c_exists, 
+                             Just (bcos,itbl_env), final_iface )
 #else
                then error "GHC not compiled with interpreter"
 #endif
@@ -410,14 +396,93 @@ hscRecomp ghci_mode dflags have_object
                            final_iface
                             stub_h_exists stub_c_exists
                            maybe_bcos)
-         }}}}}}}
+        }}
+
+hscCoreFrontEnd ghci_mode dflags location hst hit pcs_ch = do {
+           -------------------
+           -- PARSE
+           -------------------
+       ; inp <- readFile (unJust "hscCoreFrontEnd:hspp" (ml_hspp_file location))
+       ; case parseCore inp 1 of
+           FailP s        -> hPutStrLn stderr s >> return (Left (HscFail pcs_ch));
+           OkP rdr_module -> do {
+       ; let this_mod = mkHomeModule (hsModuleName rdr_module)
+    
+           -------------------
+           -- RENAME
+           -------------------
+       ; (pcs_rn, print_unqual, maybe_rn_result) 
+            <- renameExtCore dflags hit hst pcs_ch this_mod rdr_module
+       ; case maybe_rn_result of {
+            Nothing -> return (Left (HscFail pcs_ch));
+            Just (dont_discard, new_iface, rn_result) -> do {
+
+           -------------------
+           -- TYPECHECK
+           -------------------
+       ; maybe_tc_result 
+           <- _scc_ "TypeCheck" 
+              typecheckCoreModule dflags pcs_rn hst new_iface (rr_decls rn_result)
+       ; case maybe_tc_result of {
+            Nothing -> return (Left (HscFail pcs_ch));
+            Just (pcs_tc, ty_env, core_binds) -> do {
+    
+           -------------------
+           -- DESUGAR
+           -------------------
+       ; (ds_details, foreign_stuff) <- deSugarCore ty_env core_binds
+       ; return (Right (this_mod, rdr_module, maybe_rn_result, 
+                        pcs_tc, ds_details, foreign_stuff))
+       }}}}}}
+        
+
+hscFrontEnd ghci_mode dflags location hst hit pcs_ch = do {
+           -------------------
+           -- PARSE
+           -------------------
+       ; maybe_parsed <- myParseModule dflags 
+                             (unJust "hscRecomp:hspp" (ml_hspp_file location))
+       ; case maybe_parsed of {
+            Nothing -> return (Left (HscFail pcs_ch));
+            Just rdr_module -> do {
+       ; let this_mod = mkHomeModule (hsModuleName rdr_module)
+    
+           -------------------
+           -- RENAME
+           -------------------
+       ; (pcs_rn, print_unqual, maybe_rn_result) 
+            <- _scc_ "Rename" 
+                renameModule dflags ghci_mode hit hst pcs_ch this_mod rdr_module
+       ; case maybe_rn_result of {
+            Nothing -> return (Left (HscFail pcs_ch));
+            Just (dont_discard, new_iface, rn_result) -> do {
+
+           -------------------
+           -- TYPECHECK
+           -------------------
+       ; maybe_tc_result 
+           <- _scc_ "TypeCheck" 
+              typecheckModule dflags pcs_rn hst print_unqual rn_result
+       ; case maybe_tc_result of {
+            Nothing -> return (Left (HscFail pcs_ch));
+            Just (pcs_tc, tc_result) -> do {
+    
+           -------------------
+           -- DESUGAR
+           -------------------
+       ; (ds_details, foreign_stuff) 
+             <- _scc_ "DeSugar" 
+               deSugar dflags pcs_tc hst this_mod print_unqual tc_result
+       ; return (Right (this_mod, rdr_module, maybe_rn_result, 
+                        pcs_tc, ds_details, foreign_stuff))
+       }}}}}}}
+
 
 myParseModule dflags src_filename
  = do --------------------------  Parser  ----------------
       showPass dflags "Parser"
       _scc_  "Parser" do
-
-      buf <- hGetStringBuffer True{-expand tabs-} src_filename
+      buf <- hGetStringBuffer src_filename
 
       let exts = ExtFlags {glasgowExtsEF = dopt Opt_GlasgowExts dflags,
                           parrEF        = dopt Opt_PArr        dflags}
@@ -665,7 +730,7 @@ hscThing dflags hst hit pcs0 ic str
             Nothing -> return (pcs, []);
             Just pcs ->
                let do_lookup n
-                       | isLocalName n = lookupNameEnv (ic_type_env ic) n
+                       | isInternalName n = lookupNameEnv (ic_type_env ic) n
                        | otherwise     = lookupType hst (pcs_PTE pcs) n
                
                    maybe_ty_things = map do_lookup names