[project @ 2001-06-01 17:14:07 by apt]
[ghc-hetmet.git] / ghc / compiler / main / HscMain.lhs
index 90ca27e..5d09d7b 100644 (file)
@@ -1,6 +1,7 @@
 %
 % (c) The GRASP/AQUA Project, Glasgow University, 1993-2000
 %
+
 \section[GHC_Main]{Main driver for Glasgow Haskell compiler}
 
 \begin{code}
@@ -21,7 +22,7 @@ import RdrHsSyn               ( RdrNameStmt )
 import Type            ( Type )
 import Id              ( Id, idName, setGlobalIdDetails )
 import IdInfo          ( GlobalIdDetails(VanillaGlobal) )
-import HscTypes                ( InteractiveContext(..), TyThing(..) )
+import HscTypes                ( InteractiveContext(..) )
 import PrelNames       ( iNTERACTIVE )
 import StringBuffer    ( stringToStringBuffer )
 #endif
@@ -34,11 +35,12 @@ import StringBuffer ( hGetStringBuffer, freeStringBuffer )
 import Parser
 import Lex             ( PState(..), ParseResult(..) )
 import SrcLoc          ( mkSrcLoc )
+import Finder          ( findModule )
 import Rename          ( checkOldIface, renameModule, closeIfaceDecls )
 import Rules           ( emptyRuleBase )
 import PrelInfo                ( wiredInThingEnv, wiredInThings )
 import PrelNames       ( vanillaSyntaxMap, knownKeyNames )
-import MkIface         ( completeIface, writeIface, pprIface )
+import MkIface         ( mkFinalIface )
 import TcModule
 import InstEnv         ( emptyInstEnv )
 import Desugar
@@ -69,14 +71,15 @@ import FiniteMap    ( FiniteMap, plusFM, emptyFM, addToFM )
 import OccName         ( OccName )
 import Name            ( Name, nameModule, nameOccName, getName, isGlobalName )
 import NameEnv         ( emptyNameEnv, mkNameEnv )
-import Module          ( Module, lookupModuleEnvByName )
-import Maybes          ( orElse )
+import Module          ( Module )
 
 import IOExts          ( newIORef, readIORef, writeIORef, unsafePerformIO )
 
 import Monad           ( when )
-import Maybe           ( isJust )
+import Maybe           ( isJust, fromJust )
 import IO
+
+import MkExternalCore  ( emitExternalCore )
 \end{code}
 
 
@@ -98,8 +101,8 @@ data HscResult
    | HscRecomp   PersistentCompilerState -- updated PCS
                  ModDetails             -- new details (HomeSymbolTable additions)
                  ModIface               -- new iface (if any compilation was done)
-                (Maybe String)          -- generated stub_h filename (in TMPDIR)
-                (Maybe String)          -- generated stub_c filename (in TMPDIR)
+                Bool                   -- stub_h exists
+                Bool                   -- stub_c exists
                 (Maybe ([UnlinkedBCO],ItblEnv)) -- interpreted code, if any
              
 
@@ -111,22 +114,24 @@ hscMain
   -> DynFlags
   -> Module
   -> ModuleLocation            -- location info
-  -> Bool                      -- source unchanged?
+  -> Bool                      -- True <=> source unchanged
+  -> Bool                      -- True <=> have an object file (for msgs only)
   -> Maybe ModIface            -- old interface, if available
   -> HomeSymbolTable           -- for home module ModDetails
   -> HomeIfaceTable
   -> PersistentCompilerState    -- IN: persistent compiler state
   -> IO HscResult
 
-hscMain ghci_mode dflags mod location source_unchanged maybe_old_iface hst hit pcs
+hscMain ghci_mode dflags mod location source_unchanged have_object 
+       maybe_old_iface hst hit pcs
  = do {
       showPass dflags ("Checking old interface for hs = " 
                        ++ show (ml_hs_file location)
                        ++ ", hspp = " ++ show (ml_hspp_file location));
 
       (pcs_ch, errs_found, (recomp_reqd, maybe_checked_iface))
-         <- checkOldIface ghci_mode dflags hit hst pcs 
-               (unJust "hscMain" (ml_hi_file location))
+         <- _scc_ "checkOldIface"
+           checkOldIface ghci_mode dflags hit hst pcs (ml_hi_file location)
                source_unchanged maybe_old_iface;
 
       if errs_found then
@@ -137,13 +142,14 @@ hscMain ghci_mode dflags mod location source_unchanged maybe_old_iface hst hit p
           what_next | recomp_reqd || no_old_iface = hscRecomp 
                     | otherwise                   = hscNoRecomp
       ;
-      what_next ghci_mode dflags mod location maybe_checked_iface
-                hst hit pcs_ch
+      what_next ghci_mode dflags have_object mod location 
+               maybe_checked_iface hst hit pcs_ch
       }}
 
 
 -- we definitely expect to have the old interface available
-hscNoRecomp ghci_mode dflags mod location (Just old_iface) hst hit pcs_ch
+hscNoRecomp ghci_mode dflags have_object 
+           mod location (Just old_iface) hst hit pcs_ch
  | ghci_mode == OneShot
  = do {
       hPutStrLn stderr "compilation IS NOT required";
@@ -153,7 +159,8 @@ hscNoRecomp ghci_mode dflags mod location (Just old_iface) hst hit pcs_ch
  | otherwise
  = do {
       when (verbosity dflags >= 1) $
-               hPutStrLn stderr ("Skipping  " ++ compMsg mod location);
+               hPutStrLn stderr ("Skipping  " ++ 
+                       compMsg have_object mod location);
 
       -- CLOSURE
       (pcs_cl, closure_errs, cl_hs_decls) 
@@ -173,20 +180,26 @@ hscNoRecomp ghci_mode dflags mod location (Just old_iface) hst hit pcs_ch
       return (HscNoRecomp pcs_tc new_details old_iface)
       }}}
 
-compMsg mod location =
+compMsg use_object mod location =
     mod_str ++ take (max 0 (16 - length mod_str)) (repeat ' ')
-    ++ " (" ++ unJust "hscRecomp" (ml_hs_file location) ++ ")"
+    ++" ( " ++ unJust "hscRecomp" (ml_hs_file location) ++ ", "
+    ++ (if use_object
+         then unJust "hscRecomp" (ml_obj_file location)
+         else "interpreted")
+    ++ " )"
  where mod_str = moduleUserString mod
 
 
-hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
+hscRecomp ghci_mode dflags have_object 
+         mod location maybe_checked_iface hst hit pcs_ch
  = do  {
-       ; when (verbosity dflags >= 1) $
-               hPutStrLn stderr ("Compiling " ++ compMsg mod location);
-
          -- what target are we shooting for?
        ; let toInterp = dopt_HscLang dflags == HscInterpreted
 
+       ; when (verbosity dflags >= 1) $
+               hPutStrLn stderr ("Compiling " ++ 
+                       compMsg (not toInterp) mod location);
+
            -------------------
            -- PARSE
            -------------------
@@ -233,12 +246,26 @@ hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
              <- _scc_ "DeSugar" 
                deSugar dflags pcs_tc hst this_mod print_unqualified tc_result
 
+       ; pcs_middle
+           <- if ghci_mode == OneShot 
+                 then do init_pcs <- initPersistentCompilerState
+                         init_prs <- initPersistentRenamerState
+                         let 
+                             rules   = pcs_rules pcs_tc        
+                             orig_tc = prsOrig (pcs_PRS pcs_tc)
+                             new_prs = init_prs{ prsOrig=orig_tc }
+
+                         orig_tc `seq` rules `seq` new_prs `seq`
+                           return init_pcs{ pcs_PRS = new_prs,
+                                            pcs_rules = rules }
+                 else return pcs_tc
+
            -------------------
            -- SIMPLIFY
            -------------------
        ; simpl_details
             <- _scc_     "Core2Core"
-               core2core dflags pcs_tc hst dont_discard ds_details
+               core2core dflags pcs_middle hst dont_discard ds_details
 
            -------------------
            -- TIDY
@@ -255,13 +282,22 @@ hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
                -- Meanwhile, tidyCorePgm is careful not to look at cg_info!
 
        ; (pcs_simpl, tidy_details) 
-            <- tidyCorePgm dflags this_mod pcs_tc cg_info simpl_details
+            <- _scc_ "CoreTidy"
+               tidyCorePgm dflags this_mod pcs_middle cg_info simpl_details
       
+       ; pcs_final <- if ghci_mode == OneShot then initPersistentCompilerState
+                                              else return pcs_simpl
+
+       -- alive at this point:  
+       --      tidy_details
+       --      new_iface               
+
+       ; emitExternalCore dflags new_iface tidy_details 
            -------------------
            -- PREPARE FOR CODE GENERATION
            -------------------
              -- Do saturation and convert to A-normal form
-       ; prepd_details <- corePrepPgm dflags tidy_details
+       ; prepd_details <- _scc_ "CorePrep" corePrepPgm dflags tidy_details
 
            -------------------
            -- CONVERT TO STG and COMPLETE CODE GENERATION
@@ -273,22 +309,15 @@ hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
            local_classes    = typeEnvClasses env_tc
 
            imported_module_names = map ideclName (hsModuleImports rdr_module)
-           imported_modules = map mod_name_to_Module imported_module_names
+
+           mod_name_to_Module nm
+                = do m <- findModule nm ; return (fst (fromJust m))
 
            (h_code,c_code,fe_binders) = foreign_stuff
-       
-           pit = pcs_PIT pcs_simpl
 
-           mod_name_to_Module :: ModuleName -> Module
-           mod_name_to_Module nm
-              = let str_mi = lookupModuleEnvByName hit nm `orElse`
-                             lookupModuleEnvByName pit nm `orElse`
-                             pprPanic "mod_name_to_Module: no hst or pst mapping for" 
-                               (ppr nm)
-                in  mi_module str_mi
-
-       ; (maybe_stub_h_filename, maybe_stub_c_filename,
-          maybe_bcos, final_iface )
+        ; imported_modules <- mapM mod_name_to_Module imported_module_names
+
+       ; (stub_h_exists, stub_c_exists, maybe_bcos, final_iface )
           <- if toInterp
                then do 
                    -----------------  Generate byte code ------------------
@@ -303,8 +332,7 @@ hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
                          mkFinalIface ghci_mode dflags location 
                                    maybe_checked_iface new_iface tidy_details
 
-                   return ( Nothing, Nothing, 
-                            Just (bcos,itbl_env), final_iface )
+                   return ( False, False, Just (bcos,itbl_env), final_iface )
 
                else do
                    -----------------  Convert to STG ------------------
@@ -327,51 +355,24 @@ hscRecomp ghci_mode dflags mod location maybe_checked_iface hst hit pcs_ch
                                         local_tycons stg_binds
                    
                    ------------------  Code output -----------------------
-                   (maybe_stub_h_name, maybe_stub_c_name)
+                   (stub_h_exists, stub_c_exists)
                       <- codeOutput dflags this_mod local_tycons
                             binds stg_binds
                             c_code h_code abstractC
                        
-                   return ( maybe_stub_h_name, maybe_stub_c_name, 
-                            Nothing, final_iface )
+                   return (stub_h_exists, stub_c_exists, Nothing, final_iface)
 
        ; let final_details = tidy_details {md_binds = []} 
 
 
          -- and the answer is ...
-       ; return (HscRecomp pcs_simpl
+       ; return (HscRecomp pcs_final
                            final_details
                            final_iface
-                            maybe_stub_h_filename maybe_stub_c_filename
+                            stub_h_exists stub_c_exists
                            maybe_bcos)
          }}}}}}}
 
-
-
-mkFinalIface ghci_mode dflags location 
-       maybe_old_iface new_iface new_details
- = case completeIface maybe_old_iface new_iface new_details of
-
-      (new_iface, Nothing) -- no change in the interfacfe
-         -> do when (dopt Opt_D_dump_hi_diffs dflags)
-                    (printDump (text "INTERFACE UNCHANGED"))
-               dumpIfSet_dyn dflags Opt_D_dump_hi
-                             "UNCHANGED FINAL INTERFACE" (pprIface new_iface)
-              return new_iface
-
-      (new_iface, Just sdoc_diffs)
-         -> do dumpIfSet_dyn dflags Opt_D_dump_hi_diffs "INTERFACE HAS CHANGED" 
-                                    sdoc_diffs
-               dumpIfSet_dyn dflags Opt_D_dump_hi "NEW FINAL INTERFACE" 
-                                    (pprIface new_iface)
-
-               -- Write the interface file, if not in interactive mode
-               when (ghci_mode /= Interactive) 
-                    (writeIface (unJust "hscRecomp:hi" (ml_hi_file location))
-                                new_iface)
-               return new_iface
-
-
 myParseModule dflags src_filename
  = do --------------------------  Parser  ----------------
       showPass dflags "Parser"
@@ -426,6 +427,8 @@ myCoreToStg dflags this_mod tidy_binds
    where
       stgBindPairs (StgNonRec _ b r) = [(b,r)]
       stgBindPairs (StgRec    _ prs) = prs
+
+
 \end{code}
 
 
@@ -621,14 +624,4 @@ initOrigNames
         grab names = foldl add emptyFM names
         add env name 
            = addToFM env (moduleName (nameModule name), nameOccName name) name
-
-
-initRules :: PackageRuleBase
-initRules = emptyRuleBase
-{- SHOULD BE (ish)
-            foldl add emptyVarEnv builtinRules
-         where
-           add env (name,rule) 
-              = extendRuleBase env name rule
--}
 \end{code}