X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2FsimplStg%2FSimplStg.lhs;h=dc945f52bef43ce7ed3cce2023b6f6b67a0b4d9d;hb=fc6e8220115637e4610ef4ac1c0aa55fe4ca529f;hp=a06915c321fbcc260e987715356e9b57ee297009;hpb=6a3f5f6beed9cec42c4b3a1b7cabc1809c838562;p=ghc-hetmet.git

diff --git a/ghc/compiler/simplStg/SimplStg.lhs b/ghc/compiler/simplStg/SimplStg.lhs
index a06915c..dc945f5 100644
--- a/ghc/compiler/simplStg/SimplStg.lhs
+++ b/ghc/compiler/simplStg/SimplStg.lhs
@@ -10,72 +10,49 @@ module SimplStg ( stg2stg ) where
 
 import StgSyn
 
-import LambdaLift	( liftProgram )
-import CostCentre       ( CostCentre, CostCentreStack )
+import CostCentre       ( CollectedCCs )
 import SCCfinal		( stgMassageForProfiling )
 import StgLint		( lintStgBindings )
 import StgStats	        ( showStgStats )
-import StgVarInfo	( setStgVarInfo )
 import SRT		( computeSRTs )
 
 import CmdLineOpts	( DynFlags, DynFlag(..), dopt, 
-			  opt_StgDoLetNoEscapes,
-			  StgToDo(..)
+			  StgToDo(..), dopt_StgToDo
 			)
 import Id		( Id )
-import Module		( Module, moduleString )
-import ErrUtils		( doIfSet_dyn, dumpIfSet_dyn )
-import UniqSupply	( splitUniqSupply, UniqSupply )
-import IO		( hPutStr, stdout )
+import Module		( Module )
+import ErrUtils		( doIfSet_dyn, dumpIfSet_dyn, showPass )
+import UniqSupply	( mkSplitUniqSupply, splitUniqSupply )
 import Outputable
 \end{code}
 
 \begin{code}
-stg2stg :: DynFlags
-	-> [StgToDo]		-- spec of what stg-to-stg passes to do
-	-> Module		-- module name (profiling only)
-	-> UniqSupply		-- a name supply
-	-> [StgBinding]		-- input...
-	-> IO
-	    ([(StgBinding,[Id])],  -- output program...
-	     ([CostCentre],	   -- local cost-centres that need to be decl'd
-	      [CostCentre],	   -- "extern" cost-centres
-	      [CostCentreStack]))  -- pre-defined "singleton" cost centre stacks
-
-stg2stg dflags stg_todos module_name us binds
-  = case (splitUniqSupply us)	of { (us4now, us4later) ->
-
-    doIfSet_dyn dflags Opt_D_verbose_stg2stg (printErrs (text "VERBOSE STG-TO-STG:")) >>
-
-    end_pass us4now "Core2Stg" ([],[],[]) binds
-		>>= \ (binds', us, ccs) ->
-
-	-- Do the main business!
-    foldl_mn do_stg_pass (binds', us, ccs) stg_todos
-		>>= \ (processed_binds, _, cost_centres) ->
-
-	-- 	Do essential wind-up
-
-	-- Essential wind-up: part (b), do setStgVarInfo. It has to
-	-- happen regardless, because the code generator uses its
-	-- decorations.
-	--
-	-- Why does it have to happen last?  Because earlier passes
-	-- may move things around, which would change the live-var
-	-- info.  Also, setStgVarInfo decides about let-no-escape
-	-- things, which in turn do a better job if arities are
-	-- correct, which is done by satStgRhs.
-	--
-
-    let
-	annotated_binds = setStgVarInfo opt_StgDoLetNoEscapes processed_binds
-	srt_binds       = computeSRTs annotated_binds
-    in
-
-    dumpIfSet_dyn dflags Opt_D_dump_stg "STG syntax:" 
-	      (pprStgBindingsWithSRTs srt_binds)	>>
-
-    return (srt_binds, cost_centres)
+stg2stg :: DynFlags		     -- includes spec of what stg-to-stg passes to do
+	-> Module		     -- module name (profiling only)
+	-> [StgBinding]		     -- input...
+	-> IO ( [(StgBinding,[(Id,[Id])])]  -- output program...
+	      , CollectedCCs)        -- cost centre information (declared and used)
+
+stg2stg dflags module_name binds
+  = do	{ showPass dflags "Stg2Stg"
+	; us <- mkSplitUniqSupply 'g'
+
+	; doIfSet_dyn dflags Opt_D_verbose_stg2stg 
+		      (printDump (text "VERBOSE STG-TO-STG:"))
+
+	; (binds', us', ccs) <- end_pass us "Stg2Stg" ([],[],[]) binds
+
+		-- Do the main business!
+	; (processed_binds, _, cost_centres) 
+		<- foldl_mn do_stg_pass (binds', us', ccs)
+			    (dopt_StgToDo dflags)
+
+	; let srt_binds = computeSRTs processed_binds
+
+	; dumpIfSet_dyn dflags Opt_D_dump_stg "STG syntax:" 
+	     		(pprStgBindingsWithSRTs srt_binds)
+
+	; return (srt_binds, cost_centres)
    }
 
   where
@@ -89,20 +66,10 @@ stg2stg dflags stg_todos module_name us binds
 	    (us1, us2) = splitUniqSupply us
 	in
 	case to_do of
-	  StgDoStaticArgs ->  panic "STG static argument transformation deleted"
-
 	  D_stg_stats ->
 	     trace (showStgStats binds)
 	     end_pass us2 "StgStats" ccs binds
 
-	  StgDoLambdaLift ->
-	     _scc_ "StgLambdaLift"
-		-- NB We have to do setStgVarInfo first!
-	     let
-		binds3 = liftProgram module_name us1 (setStgVarInfo opt_StgDoLetNoEscapes binds)
-	     in
-	     end_pass us2 "LambdaLift" ccs binds3
-
 	  StgDoMassageForProfiling ->
 	     _scc_ "ProfMassage"
 	     let
@@ -112,16 +79,11 @@ stg2stg dflags stg_todos module_name us binds
 	     end_pass us2 "ProfMassage" collected_CCs binds3
 
     end_pass us2 what ccs binds2
-      = -- report verbosely, if required
-	(if dopt Opt_D_verbose_stg2stg dflags then
-	    hPutStr stdout (showSDoc
-	      (text ("*** "++what++":") $$ vcat (map ppr binds2)
-	    ))
-	 else return ()) >>
-	let
-	    linted_binds = stg_linter what binds2
-	in
-	return (linted_binds, us2, ccs)
+      = do -- report verbosely, if required
+	   dumpIfSet_dyn dflags Opt_D_verbose_stg2stg what
+	      (vcat (map ppr binds2))
+	   let linted_binds = stg_linter what binds2
+	   return (linted_binds, us2, ccs)
 	    -- return: processed binds
 	    -- 	       UniqueSupply for the next guy to use
 	    --	       cost-centres to be declared/registered (specialised)