[project @ 2001-01-26 17:37:27 by simonmar]
[ghc-hetmet.git] / ghc / compiler / compMan / CompManager.lhs
index 8041e61..9948d3d 100644 (file)
@@ -4,9 +4,9 @@
 \section[CompManager]{The Compilation Manager}
 
 \begin{code}
-module CompManager ( cmInit, cmLoadModule,
+module CompManager ( cmInit, cmLoadModule, cmUnload,
 #ifdef GHCI
-                     cmGetExpr, cmTypeExpr, cmRunExpr,
+                     cmGetExpr, cmRunExpr,
 #endif
                      CmState, emptyCmState  -- abstract
                    )
@@ -17,37 +17,34 @@ where
 import CmLink
 import CmTypes
 import HscTypes
-import Module          ( ModuleName, moduleName,
-                         isHomeModule, moduleEnvElts,
-                         moduleNameUserString )
+import Module          ( Module, ModuleName, moduleName, isHomeModule,
+                         mkModuleName, moduleNameUserString )
 import CmStaticInfo    ( GhciMode(..) )
 import DriverPipeline
 import GetImports
 import HscTypes                ( HomeSymbolTable, HomeIfaceTable, 
                          PersistentCompilerState, ModDetails(..) )
-import Type            ( Type )
-import Name            ( lookupNameEnv )
-import Module
-import PrelNames       ( mainName )
 import HscMain         ( initPersistentCompilerState )
 import Finder
 import UniqFM          ( emptyUFM, lookupUFM, addToUFM, delListFromUFM,
                          UniqFM, listToUFM )
 import Unique          ( Uniquable )
-import Digraph         ( SCC(..), stronglyConnComp )
+import Digraph         ( SCC(..), stronglyConnComp, flattenSCC )
 import DriverFlags     ( getDynFlags )
 import DriverPhases
-import DriverUtil      ( BarfKind(..), splitFilename3 )
+import DriverUtil      ( splitFilename3 )
 import ErrUtils                ( showPass )
 import Util
+import DriverUtil
 import Outputable
-import Panic           ( panic )
+import Panic
 import CmdLineOpts     ( DynFlags(..) )
+import IOExts
 
 #ifdef GHCI
 import Interpreter     ( HValue )
-import HscMain         ( hscExpr, hscTypeExpr )
-import RdrName
+import HscMain         ( hscExpr )
+import Type            ( Type )
 import PrelGHC         ( unsafeCoerce# )
 #endif
 
@@ -60,7 +57,7 @@ import Directory        ( getModificationTime, doesFileExist )
 import IO
 import Monad
 import List            ( nub )
-import Maybe           ( catMaybes, fromMaybe, isJust )
+import Maybe           ( catMaybes, fromMaybe, maybeToList, isJust )
 \end{code}
 
 
@@ -72,35 +69,23 @@ cmInit gmode
 #ifdef GHCI
 cmGetExpr :: CmState
          -> DynFlags
-          -> ModuleName
+         -> Bool       -- True <=> wrap in 'print' to get an IO-typed result
+          -> Module
           -> String
-          -> IO (CmState, Maybe HValue)
-cmGetExpr cmstate dflags modname expr
-   = do (new_pcs, maybe_unlinked_iexpr) <- 
-          hscExpr dflags hst hit pcs (mkHomeModule modname) expr
-        case maybe_unlinked_iexpr of
+          -> IO (CmState, Maybe (HValue, PrintUnqualified, Type))
+cmGetExpr cmstate dflags wrap_io mod expr
+   = do (new_pcs, maybe_stuff) <- 
+          hscExpr dflags wrap_io hst hit pcs mod expr
+        case maybe_stuff of
           Nothing     -> return (cmstate{ pcs=new_pcs }, Nothing)
-          Just uiexpr -> do
-               hValue <- linkExpr pls uiexpr
-               return (cmstate{ pcs=new_pcs }, Just hValue)
+          Just (bcos, print_unqual, ty) -> do
+               hValue <- linkExpr pls bcos
+               return (cmstate{ pcs=new_pcs }, 
+                       Just (hValue, print_unqual, ty))
 
    -- ToDo: check that the module we passed in is sane/exists?
    where
-       CmState{ pcs=pcs, pcms=pcms, pls=pls } = cmstate
-       PersistentCMState{ hst=hst, hit=hit } = pcms
-
-cmTypeExpr :: CmState
-         -> DynFlags
-          -> ModuleName
-          -> String
-          -> IO (CmState, Maybe (PrintUnqualified, Type))
-cmTypeExpr cmstate dflags modname expr
-   = do (new_pcs, expr_type) <- 
-          hscTypeExpr dflags hst hit pcs (mkHomeModule modname) expr
-        return (cmstate{ pcs=new_pcs }, expr_type)
-   where
-       CmState{ pcs=pcs, pcms=pcms, pls=pls } = cmstate
-       PersistentCMState{ hst=hst, hit=hit } = pcms
+       CmState{ hst=hst, hit=hit, pcs=pcs, pls=pls } = cmstate
 
 -- The HValue should represent a value of type IO () (Perhaps IO a?)
 cmRunExpr :: HValue -> IO ()
@@ -109,43 +94,33 @@ cmRunExpr hval
        -- putStrLn "done."
 #endif
 
--- Persistent state just for CM, excluding link & compile subsystems
-data PersistentCMState
-   = PersistentCMState {
-        hst   :: HomeSymbolTable,    -- home symbol table
-        hit   :: HomeIfaceTable,     -- home interface table
-        ui    :: UnlinkedImage,      -- the unlinked images
-        mg    :: ModuleGraph,        -- the module graph
-        gmode :: GhciMode            -- NEVER CHANGES
-     }
-
-emptyPCMS :: GhciMode -> PersistentCMState
-emptyPCMS gmode
-  = PersistentCMState { hst = emptyHST, hit = emptyHIT,
-                        ui  = emptyUI,  mg  = emptyMG, 
-                        gmode = gmode }
-
 emptyHIT :: HomeIfaceTable
 emptyHIT = emptyUFM
 emptyHST :: HomeSymbolTable
 emptyHST = emptyUFM
 
-
-
 -- Persistent state for the entire system
 data CmState
    = CmState {
-        pcms   :: PersistentCMState,       -- CM's persistent state
+        hst   :: HomeSymbolTable,    -- home symbol table
+        hit   :: HomeIfaceTable,     -- home interface table
+        ui    :: UnlinkedImage,      -- the unlinked images
+        mg    :: ModuleGraph,        -- the module graph
+        gmode :: GhciMode,           -- NEVER CHANGES
+
         pcs    :: PersistentCompilerState, -- compile's persistent state
         pls    :: PersistentLinkerState    -- link's persistent state
      }
 
 emptyCmState :: GhciMode -> IO CmState
 emptyCmState gmode
-    = do let pcms = emptyPCMS gmode
-         pcs     <- initPersistentCompilerState
+    = do pcs     <- initPersistentCompilerState
          pls     <- emptyPLS
-         return (CmState { pcms   = pcms,
+         return (CmState { hst = emptyHST,
+                           hit = emptyHIT,
+                           ui  = emptyUI,
+                           mg  = emptyMG, 
+                           gmode = gmode,
                            pcs    = pcs,
                            pls    = pls })
 
@@ -160,6 +135,21 @@ emptyMG = []
 
 \end{code}
 
+Unload the compilation manager's state: everything it knows about the
+current collection of modules in the Home package.
+
+\begin{code}
+cmUnload :: CmState -> IO CmState
+cmUnload state 
+ = do -- Throw away the old home dir cache
+      emptyHomeDirCache
+      -- Throw away the HIT and the HST
+      return state{ hst=new_hst, hit=new_hit }
+   where
+     CmState{ hst=hst, hit=hit } = state
+     (new_hst, new_hit) = retainInTopLevelEnvs [] (hst,hit)
+\end{code}
+
 The real business of the compilation manager: given a system state and
 a module name, try and bring the module up to date, probably changing
 the system state at the same time.
@@ -169,42 +159,38 @@ cmLoadModule :: CmState
              -> FilePath
              -> IO (CmState,           -- new state
                    Bool,               -- was successful
-                   [ModuleName])       -- list of modules loaded
+                   [Module])           -- list of modules loaded
 
 cmLoadModule cmstate1 rootname
    = do -- version 1's are the original, before downsweep
-        let pcms1     = pcms   cmstate1
         let pls1      = pls    cmstate1
         let pcs1      = pcs    cmstate1
-        let mg1       = mg     pcms1
-        let hst1      = hst    pcms1
-        let hit1      = hit    pcms1
-        let ui1       = ui     pcms1
+        let hst1      = hst    cmstate1
+        let hit1      = hit    cmstate1
+       -- similarly, ui1 is the (complete) set of linkables from
+       -- the previous pass, if any.
+        let ui1       = ui     cmstate1
    
-        let ghci_mode = gmode pcms1 -- this never changes
+        let ghci_mode = gmode cmstate1 -- this never changes
 
         -- Do the downsweep to reestablish the module graph
-        -- then generate version 2's by removing from HIT,HST,UI any
-        -- modules in the old MG which are not in the new one.
-
-        -- Throw away the old home dir cache
-        emptyHomeDirCache
+        -- then generate version 2's by retaining in HIT,HST,UI a
+        -- stable set S of modules, as defined below.
 
        dflags <- getDynFlags
         let verb = verbosity dflags
 
        showPass dflags "Chasing dependencies"
         when (verb >= 1 && ghci_mode == Batch) $
-           hPutStrLn stderr ("ghc: chasing modules from: " ++ rootname)
+           hPutStrLn stderr (progName ++ ": chasing modules from: " ++ rootname)
 
-        mg2unsorted <- downsweep [rootname]
+        (mg2unsorted, a_root_is_Main) <- downsweep [rootname]
+        let mg2unsorted_names = map name_of_summary mg2unsorted
 
-        let modnames1   = map name_of_summary mg1
-        let modnames2   = map name_of_summary mg2unsorted
-        let mods_to_zap = filter (`notElem` modnames2) modnames1
-
-        let (hst2, hit2, ui2)
-               = removeFromTopLevelEnvs mods_to_zap (hst1, hit1, ui1)
+        -- reachable_from follows source as well as normal imports
+        let reachable_from :: ModuleName -> [ModuleName]
+            reachable_from = downwards_closure_of_module mg2unsorted
         -- should be cycle free; ignores 'import source's
         let mg2 = topological_sort False mg2unsorted
         -- ... whereas this takes them into account.  Used for
@@ -212,9 +198,49 @@ cmLoadModule cmstate1 rootname
         -- upsweep, and for removing from hst/hit all the modules
         -- not in strict downwards closure, during calls to compile.
         let mg2_with_srcimps = topological_sort True mg2unsorted
-      
-        let reachable_from :: ModuleName -> [ModuleName]
-            reachable_from = downwards_closure_of_module mg2unsorted
+
+       -- Sort out which linkables we wish to keep in the unlinked image.
+       -- For each module, we take:
+       --
+       --      - the old in-core linkable, if available
+       --      - an on-disk linkable, if available
+       --
+       -- and we take the youngest of these, provided it is younger than the
+       -- source file.
+       --
+       -- If a module has a valid linkable, then it may be STABLE (see below),
+       -- and it is classified as SOURCE UNCHANGED for the purposes of calling
+       -- compile.
+       valid_linkables <- getValidLinkables ui1 mg2unsorted
+
+        -- Figure out a stable set of modules which can be retained
+        -- the top level envs, to avoid upsweeping them.  Goes to a
+        -- bit of trouble to avoid upsweeping module cycles.
+        --
+        -- Construct a set S of stable modules like this:
+        -- Travel upwards, over the sccified graph.  For each scc
+        -- of modules ms, add ms to S only if:
+        -- 1.  All home imports of ms are either in ms or S
+        -- 2.  A valid linkable exists for each module in ms
+
+        stable_mods
+           <- preUpsweep valid_linkables mg2unsorted_names [] mg2_with_srcimps
+
+        let stable_summaries
+               = concatMap (findInSummaries mg2unsorted) stable_mods
+
+        when (verb >= 2) $
+           putStrLn (showSDoc (text "STABLE MODULES:" 
+                               <+> sep (map (text.moduleNameUserString) stable_mods)))
+
+        -- We could at this point detect cycles which aren't broken by
+        -- a source-import, and complain immediately, but it seems better
+        -- to let upsweep_mods do this, so at least some useful work gets
+        -- done before the upsweep is abandoned.
+        let upsweep_these
+               = filter (\scc -> any (`notElem` stable_mods) 
+                                     (map name_of_summary (flattenSCC scc)))
+                        mg2
 
         --hPutStrLn stderr "after tsort:\n"
         --hPutStrLn stderr (showSDoc (vcat (map ppr mg2)))
@@ -227,18 +253,27 @@ cmLoadModule cmstate1 rootname
         -- Now do the upsweep, calling compile for each module in
         -- turn.  Final result is version 3 of everything.
 
-        let threaded2 = CmThreaded pcs1 hst2 hit2
+        let threaded2 = CmThreaded pcs1 hst1 hit1
 
-        (upsweep_complete_success, threaded3, modsDone, newLis)
-           <- upsweep_mods ghci_mode ui2 reachable_from threaded2 mg2
+        (upsweep_complete_success, threaded3, modsUpswept, newLis)
+           <- upsweep_mods ghci_mode dflags valid_linkables reachable_from 
+                           threaded2 upsweep_these
 
-        let ui3 = add_to_ui ui2 newLis
+        let ui3 = add_to_ui valid_linkables newLis
         let (CmThreaded pcs3 hst3 hit3) = threaded3
 
-        -- At this point, modsDone and newLis should have the same
+        -- At this point, modsUpswept and newLis should have the same
         -- length, so there is one new (or old) linkable for each 
         -- mod which was processed (passed to compile).
 
+       -- Make modsDone be the summaries for each home module now
+       -- available; this should equal the domains of hst3 and hit3.
+       -- (NOT STRICTLY TRUE if an interactive session was started
+       --  with some object on disk ???)
+        -- Get in in a roughly top .. bottom order (hence reverse).
+
+        let modsDone = reverse modsUpswept ++ stable_summaries
+
         -- Try and do linking in some form, depending on whether the
         -- upsweep was completely or only partially successful.
 
@@ -247,21 +282,20 @@ cmLoadModule cmstate1 rootname
          then 
            -- Easy; just relink it all.
            do when (verb >= 2) $ 
-               hPutStrLn stderr "Upsweep completely successful."
+                hPutStrLn stderr "Upsweep completely successful."
               linkresult 
-                 <- link ghci_mode dflags 
-                       (any exports_main (moduleEnvElts hst3)) 
-                        newLis pls1
+                 <- link ghci_mode dflags a_root_is_Main ui3 pls1
               case linkresult of
                  LinkErrs _ _
                     -> panic "cmLoadModule: link failed (1)"
                  LinkOK pls3 
-                    -> do let pcms3 = PersistentCMState { hst=hst3, hit=hit3, 
-                                                          ui=ui3, mg=modsDone, 
-                                                          gmode=ghci_mode }
-                          let cmstate3 
-                                 = CmState { pcms=pcms3, pcs=pcs3, pls=pls3 }
-                          return (cmstate3, True, map name_of_summary modsDone)
+                    -> do let cmstate3 
+                                 = CmState { hst=hst3, hit=hit3, 
+                                             ui=ui3, mg=modsDone, 
+                                             gmode=ghci_mode,
+                                            pcs=pcs3, pls=pls3 }
+                          return (cmstate3, True, 
+                                  map ms_mod modsDone)
 
          else 
            -- Tricky.  We need to back out the effects of compiling any
@@ -291,12 +325,130 @@ cmLoadModule cmstate1 rootname
                  LinkErrs _ _
                     -> panic "cmLoadModule: link failed (2)"
                  LinkOK pls4
-                    -> do let pcms4 = PersistentCMState { hst=hst4, hit=hit4, 
-                                                          ui=ui4, mg=mods_to_keep,
-                                                          gmode=ghci_mode }
-                          let cmstate4 
-                                 = CmState { pcms=pcms4, pcs=pcs3, pls=pls4 }
-                          return (cmstate4, False, mods_to_keep_names)
+                    -> do let cmstate4 
+                                 = CmState { hst=hst4, hit=hit4, 
+                                             ui=ui4, mg=mods_to_keep,
+                                             gmode=ghci_mode, pcs=pcs3, pls=pls4 }
+                          return (cmstate4, False, 
+                                  map ms_mod mods_to_keep)
+
+
+-----------------------------------------------------------------------------
+-- getValidLinkables
+
+getValidLinkables
+       :: [Linkable]                   -- old linkables
+       -> [ModSummary]                 -- all modules in the program
+       -> IO [Linkable]                -- still-valid linkables 
+
+getValidLinkables old_linkables summaries
+  = do lis <- mapM (getValidLinkable old_linkables) summaries
+       return (concat lis)
+
+getValidLinkable old_linkables summary
+  = do let mod_name = moduleName (ms_mod summary)
+       maybe_disk_linkable
+           <- case ml_obj_file (ms_location summary) of
+                 Nothing -> return Nothing
+                 Just obj_fn -> maybe_getFileLinkable mod_name obj_fn
+
+       -- find an old in-core linkable if we have one. (forget about
+       -- on-disk linkables for now, we'll check again whether there's
+       -- one here below, just in case a new one has popped up recently).
+       let old_linkable = findModuleLinkable_maybe old_linkables mod_name
+           maybe_old_linkable =
+               case old_linkable of
+                   Just (LM _ _ ls) | all isInterpretable ls -> old_linkable
+                   _ -> Nothing      
+
+       -- The most recent of the old UI linkable or whatever we could
+       -- find on disk.  Is returned as the linkable if compile
+       -- doesn't think we need to recompile.        
+       let linkable_list
+               = case (maybe_old_linkable, maybe_disk_linkable) of
+                    (Nothing, Nothing) -> []
+                    (Nothing, Just di) -> [di]
+                    (Just ui, Nothing) -> [ui]
+                    (Just ui, Just di)
+                       | linkableTime ui >= linkableTime di -> [ui]
+                       | otherwise                          -> [di]
+
+       -- only linkables newer than the source code are valid
+       let maybe_src_date = ms_hs_date summary
+
+          valid_linkable_list
+             = case maybe_src_date of
+                 Nothing -> panic "valid_linkable_list"
+                 Just src_date 
+                     -> filter (\li -> linkableTime li > src_date) linkable_list
+
+       return valid_linkable_list
+
+
+
+maybe_getFileLinkable :: ModuleName -> FilePath -> IO (Maybe Linkable)
+maybe_getFileLinkable mod_name obj_fn
+   = do obj_exist <- doesFileExist obj_fn
+        if not obj_exist 
+         then return Nothing 
+         else 
+         do let stub_fn = case splitFilename3 obj_fn of
+                             (dir, base, ext) -> dir ++ "/" ++ base ++ ".stub_o"
+            stub_exist <- doesFileExist stub_fn
+            obj_time <- getModificationTime obj_fn
+            if stub_exist
+             then return (Just (LM obj_time mod_name [DotO obj_fn, DotO stub_fn]))
+             else return (Just (LM obj_time mod_name [DotO obj_fn]))
+
+
+-----------------------------------------------------------------------------
+-- Do a pre-upsweep without use of "compile", to establish a 
+-- (downward-closed) set of stable modules which can be retained
+-- in the top-level environments.  Also return linkables for those 
+-- modules determined to be stable, since (in Batch mode, at least)
+-- there's no other way for them to get into UI.
+
+preUpsweep :: [Linkable]       -- valid linkables
+           -> [ModuleName]      -- names of all mods encountered in downsweep
+           -> [ModuleName]      -- accumulating stable modules
+           -> [SCC ModSummary]  -- scc-ified mod graph, including src imps
+           -> IO [ModuleName]  -- stable modules
+
+preUpsweep valid_lis all_home_mods stable [] 
+   = return stable
+preUpsweep valid_lis all_home_mods stable (scc0:sccs)
+   = do let scc = flattenSCC scc0
+            scc_allhomeimps :: [ModuleName]
+            scc_allhomeimps 
+               = nub (filter (`elem` all_home_mods) (concatMap ms_allimps scc))
+            all_imports_in_scc_or_stable
+               = all in_stable_or_scc scc_allhomeimps
+            scc_names
+               = map name_of_summary scc
+            in_stable_or_scc m
+               = --trace (showSDoc (text "ISOS" <+> ppr m <+> ppr scc_names <+> ppr stable)) (
+                 m `elem` scc_names || m `elem` stable
+                 --)
+        all_scc_stable
+           <- if   not all_imports_in_scc_or_stable
+               then do --putStrLn ("PART1 fail " ++ showSDoc (ppr scc_allhomeimps <+> ppr (filter (not.in_stable_or_scc) scc_allhomeimps)))
+                       return False
+               else do --when (not (and bools)) (putStrLn ("PART2 fail: " ++ showSDoc (ppr scc_names)))
+                       return (all is_stable scc)
+        if not all_scc_stable
+         then preUpsweep valid_lis all_home_mods stable sccs
+         else preUpsweep valid_lis all_home_mods (scc_names++stable) sccs
+
+   where is_stable new_summary
+           = isJust (findModuleLinkable_maybe valid_lis (name_of_summary new_summary))
+
+
+-- Helper for preUpsweep.  Assuming that new_summary's imports are all
+-- stable (in the sense of preUpsweep), determine if new_summary is itself
+-- stable, and, if so, in batch mode, return its linkable.
+findInSummaries :: [ModSummary] -> ModuleName -> [ModSummary]
+findInSummaries old_summaries mod_name
+   = [s | s <- old_summaries, name_of_summary s == mod_name]
 
 
 -- Return (names of) all those in modsDone who are part of a cycle
@@ -321,9 +473,9 @@ findPartiallyCompletedCycles modsDone theGraph
 
 
 -- Does this ModDetails export Main.main?
-exports_main :: ModDetails -> Bool
-exports_main md
-   = isJust (lookupNameEnv (md_types md) mainName)
+--exports_main :: ModDetails -> Bool
+--exports_main md
+--   = isJust (lookupNameEnv (md_types md) mainName)
 
 
 -- Add the given (LM-form) Linkables to the UI, overwriting previous
@@ -350,7 +502,8 @@ data CmThreaded  -- stuff threaded through individual module compilations
 -- Compile multiple modules, stopping as soon as an error appears.
 -- There better had not be any cyclic groups here -- we check for them.
 upsweep_mods :: GhciMode
-             -> UnlinkedImage         -- old linkables
+            -> DynFlags
+             -> UnlinkedImage         -- valid linkables
              -> (ModuleName -> [ModuleName])  -- to construct downward closures
              -> CmThreaded            -- PCS & HST & HIT
              -> [SCC ModSummary]      -- mods to do (the worklist)
@@ -360,26 +513,30 @@ upsweep_mods :: GhciMode
                     [ModSummary],     -- mods which succeeded
                     [Linkable])       -- new linkables
 
-upsweep_mods ghci_mode oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
      []
    = return (True, threaded, [], [])
 
-upsweep_mods ghci_mode oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
      ((CyclicSCC ms):_)
    = do hPutStrLn stderr ("Module imports form a cycle for modules:\n\t" ++
                           unwords (map (moduleNameUserString.name_of_summary) ms))
         return (False, threaded, [], [])
 
-upsweep_mods ghci_mode oldUI reachable_from threaded 
+upsweep_mods ghci_mode dflags oldUI reachable_from threaded 
      ((AcyclicSCC mod):mods)
-   = do (threaded1, maybe_linkable) 
-           <- upsweep_mod ghci_mode oldUI threaded mod 
-                          (reachable_from (name_of_summary mod)) 
+   = do --case threaded of
+        --   CmThreaded pcsz hstz hitz
+        --      -> putStrLn ("UPSWEEP_MOD: hit = " ++ show (map (moduleNameUserString.moduleName.mi_module) (eltsUFM hitz)))
+
+        (threaded1, maybe_linkable) 
+           <- upsweep_mod ghci_mode dflags oldUI threaded mod 
+                          (reachable_from (name_of_summary mod))
         case maybe_linkable of
            Just linkable 
               -> -- No errors; do the rest
                  do (restOK, threaded2, modOKs, linkables) 
-                       <- upsweep_mods ghci_mode oldUI reachable_from 
+                       <- upsweep_mods ghci_mode dflags oldUI reachable_from 
                                        threaded1 mods
                     return (restOK, threaded2, mod:modOKs, linkable:linkables)
            Nothing -- we got a compilation error; give up now
@@ -388,75 +545,39 @@ upsweep_mods ghci_mode oldUI reachable_from threaded
 
 -- Compile a single module.  Always produce a Linkable for it if 
 -- successful.  If no compilation happened, return the old Linkable.
-maybe_getFileLinkable :: ModuleName -> FilePath -> IO (Maybe Linkable)
-maybe_getFileLinkable mod_name obj_fn
-   = do obj_exist <- doesFileExist obj_fn
-        if not obj_exist 
-         then return Nothing 
-         else 
-         do let stub_fn = case splitFilename3 obj_fn of
-                             (dir, base, ext) -> dir ++ "/" ++ base ++ ".stub_o"
-            stub_exist <- doesFileExist stub_fn
-            obj_time <- getModificationTime obj_fn
-            if stub_exist
-             then return (Just (LM obj_time mod_name [DotO obj_fn, DotO stub_fn]))
-             else return (Just (LM obj_time mod_name [DotO obj_fn]))
-
-
 upsweep_mod :: GhciMode 
+           -> DynFlags
             -> UnlinkedImage
             -> CmThreaded
             -> ModSummary
             -> [ModuleName]
             -> IO (CmThreaded, Maybe Linkable)
 
-upsweep_mod ghci_mode oldUI threaded1 summary1 reachable_from_here
-   = do hPutStr stderr ("ghc: module " 
-                        ++ moduleNameUserString (name_of_summary summary1) ++ ": ")
+upsweep_mod ghci_mode dflags oldUI threaded1 summary1 reachable_from_here
+   = do 
         let mod_name = name_of_summary summary1
+       let verb = verbosity dflags
+
+        when (verb == 1) $
+          if (ghci_mode == Batch)
+               then hPutStr stderr (progName ++ ": module " 
+                               ++ moduleNameUserString mod_name
+                       ++ ": ")
+               else hPutStr stderr ("Compiling "
+                       ++ moduleNameUserString mod_name
+                       ++ " ... ")
+
         let (CmThreaded pcs1 hst1 hit1) = threaded1
-        let old_iface = lookupUFM hit1 (name_of_summary summary1)
-
-        -- We *have* to compile it if we're in batch mode and we can't see
-        -- a previous linkable for it on disk.
-        compilation_mandatory 
-           <- if ghci_mode /= Batch then return False 
-              else case ml_obj_file (ms_location summary1) of
-                      Nothing     -> do --putStrLn "cmcm: object?!"
-                                        return True
-                      Just obj_fn -> do --putStrLn ("cmcm: old obj " ++ obj_fn)
-                                        b <- doesFileExist obj_fn
-                                        return (not b)
-
-        let maybe_oldUI_linkable = findModuleLinkable_maybe oldUI mod_name
-        maybe_oldDisk_linkable
-           <- case ml_obj_file (ms_location summary1) of
-                 Nothing -> return Nothing
-                 Just obj_fn -> maybe_getFileLinkable mod_name obj_fn
+        let old_iface = lookupUFM hit1 mod_name
 
-        -- The most recent of the old UI linkable or whatever we could
-        -- find on disk.  Is returned as the linkable if compile
-        -- doesn't think we need to recompile.        
-        let maybe_old_linkable
-               = case (maybe_oldUI_linkable, maybe_oldDisk_linkable) of
-                    (Nothing, Nothing) -> Nothing
-                    (Nothing, Just di) -> Just di
-                    (Just ui, Nothing) -> Just ui
-                    (Just ui, Just di)
-                       | linkableTime ui >= linkableTime di -> Just ui
-                       | otherwise                          -> Just di
-
-        let compilation_mandatory
-               = case maybe_old_linkable of
-                    Nothing -> True
-                    Just li -> case ms_hs_date summary1 of
-                                  Nothing -> panic "compilation_mandatory:no src date"
-                                  Just src_date -> src_date >= linkableTime li
-            source_unchanged
-               = not compilation_mandatory
+        let maybe_old_linkable = findModuleLinkable_maybe oldUI mod_name
+
+            source_unchanged = isJust maybe_old_linkable
 
             (hst1_strictDC, hit1_strictDC)
-               = retainInTopLevelEnvs reachable_from_here (hst1,hit1)
+               = retainInTopLevelEnvs 
+                    (filter (/= (name_of_summary summary1)) reachable_from_here)
+                    (hst1,hit1)
 
             old_linkable 
                = unJust "upsweep_mod:old_linkable" maybe_old_linkable
@@ -470,25 +591,42 @@ upsweep_mod ghci_mode oldUI threaded1 summary1 reachable_from_here
            -- linkable, meaning that compilation wasn't needed, and the
            -- new details were manufactured from the old iface.
            CompOK pcs2 new_details new_iface Nothing
-              -> let hst2         = addToUFM hst1 mod_name new_details
-                     hit2         = addToUFM hit1 mod_name new_iface
-                     threaded2    = CmThreaded pcs2 hst2 hit2
-                 in  return (threaded2, Just old_linkable)
+              -> do let hst2         = addToUFM hst1 mod_name new_details
+                        hit2         = addToUFM hit1 mod_name new_iface
+                        threaded2    = CmThreaded pcs2 hst2 hit2
+
+                   if ghci_mode == Interactive && verb >= 1 then
+                     -- if we're using an object file, tell the user
+                     case maybe_old_linkable of
+                       Just (LM _ _ objs@(DotO _:_))
+                          -> do hPutStr stderr (showSDoc (space <> 
+                                  parens (hsep (text "using": 
+                                       punctuate comma 
+                                         [ text o | DotO o <- objs ]))))
+                                when (verb > 1) $ hPutStrLn stderr ""
+                       _ -> return ()
+                     else
+                       return ()
+
+                   when (verb == 1) $ hPutStrLn stderr ""
+                    return (threaded2, Just old_linkable)
 
            -- Compilation really did happen, and succeeded.  A new
            -- details, iface and linkable are returned.
            CompOK pcs2 new_details new_iface (Just new_linkable)
-              -> let hst2      = addToUFM hst1 mod_name new_details
-                     hit2      = addToUFM hit1 mod_name new_iface
-                     threaded2 = CmThreaded pcs2 hst2 hit2
-                 in  return (threaded2, Just new_linkable)
+              -> do let hst2      = addToUFM hst1 mod_name new_details
+                        hit2      = addToUFM hit1 mod_name new_iface
+                        threaded2 = CmThreaded pcs2 hst2 hit2
+
+                   when (verb == 1) $ hPutStrLn stderr ""
+                   return (threaded2, Just new_linkable)
 
            -- Compilation failed.  compile may still have updated
            -- the PCS, tho.
            CompErrs pcs2
-              -> let threaded2 = CmThreaded pcs2 hst1 hit1
-                 in  return (threaded2, Nothing)
-
+             -> do let threaded2 = CmThreaded pcs2 hst1 hit1
+                   when (verb == 1) $ hPutStrLn stderr ""
+                    return (threaded2, Nothing)
 
 -- Remove unwanted modules from the top level envs (HST, HIT, UI).
 removeFromTopLevelEnvs :: [ModuleName]
@@ -518,8 +656,7 @@ retainInTopLevelEnvs keep_these (hst, hit)
 downwards_closure_of_module :: [ModSummary] -> ModuleName -> [ModuleName]
 downwards_closure_of_module summaries root
    = let toEdge :: ModSummary -> (ModuleName,[ModuleName])
-         toEdge summ
-             = (name_of_summary summ, ms_srcimps summ ++ ms_imps summ)
+         toEdge summ = (name_of_summary summ, ms_allimps summ)
          res = simple_transitive_closure (map toEdge summaries) [root]             
      in
          --trace (showSDoc (text "DC of mod" <+> ppr root
@@ -567,11 +704,17 @@ topological_sort include_source_imports summaries
 
 -- Chase downwards from the specified root set, returning summaries
 -- for all home modules encountered.  Only follow source-import
--- links.
-downsweep :: [FilePath] -> IO [ModSummary]
+-- links.  Also returns a Bool to indicate whether any of the roots
+-- are module Main.
+downsweep :: [FilePath] -> IO ([ModSummary], Bool)
 downsweep rootNm
    = do rootSummaries <- mapM getRootSummary rootNm
-        loop (filter (isHomeModule.ms_mod) rootSummaries)
+        let a_root_is_Main 
+               = any ((=="Main").moduleNameUserString.name_of_summary) 
+                     rootSummaries
+        all_summaries
+           <- loop (filter (isHomeModule.ms_mod) rootSummaries)
+        return (all_summaries, a_root_is_Main)
      where
        getRootSummary :: FilePath -> IO ModSummary
        getRootSummary file