X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2FdeSugar%2FDsMonad.lhs;h=ecddeb4757be4027b029709f82268a16e0ee65da;hb=61663f75b09d05a083bcb2c0c3821528e129fcc2;hp=bf3f5f0878881791cee86d4960d9ce173d8d353e;hpb=7a3bd641457666e10d0a47be9f22762e03defbf0;p=ghc-hetmet.git

diff --git a/ghc/compiler/deSugar/DsMonad.lhs b/ghc/compiler/deSugar/DsMonad.lhs
index bf3f5f0..ecddeb4 100644
--- a/ghc/compiler/deSugar/DsMonad.lhs
+++ b/ghc/compiler/deSugar/DsMonad.lhs
@@ -1,49 +1,47 @@
 %
-% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996
+% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
 %
 \section[DsMonad]{@DsMonad@: monadery used in desugaring}
 
 \begin{code}
-#include "HsVersions.h"
-
 module DsMonad (
-	SYN_IE(DsM),
+	DsM,
 	initDs, returnDs, thenDs, andDs, mapDs, listDs,
-	mapAndUnzipDs, zipWithDs,
+	mapAndUnzipDs, zipWithDs, foldlDs,
 	uniqSMtoDsM,
 	newTyVarsDs, cloneTyVarsDs,
 	duplicateLocalDs, newSysLocalDs, newSysLocalsDs,
 	newFailLocalDs,
 	getSrcLocDs, putSrcLocDs,
-	getModuleAndGroupDs,
-	extendEnvDs, lookupEnvDs, lookupEnvWithDefaultDs,
-	SYN_IE(DsIdEnv),
-	lookupId,
-
-	dsShadowError,
-	DsMatchContext(..), DsMatchKind(..), pprDsWarnings
+	getModuleDs,
+	getUniqueDs,
+	getDOptsDs,
+	dsLookupGlobalValue,
+
+	dsWarn, 
+	DsWarnings,
+	DsMatchContext(..), DsMatchKind(..)
     ) where
 
-IMP_Ubiq()
-
-import Bag		( emptyBag, snocBag, bagToList )
-import CmdLineOpts	( opt_SccGroup )
-import CoreSyn		( SYN_IE(CoreExpr) )
-import CoreUtils	( substCoreExpr )
-import HsSyn		( OutPat )
-import Id		( mkSysLocal, mkIdWithNewUniq,
-			  lookupIdEnv, growIdEnvList, GenId, SYN_IE(IdEnv)
-			)
-import PprType		( GenType, GenTyVar )
-import PprStyle		( PprStyle(..) )
-import Pretty
+#include "HsVersions.h"
+
+import Bag		( emptyBag, snocBag, Bag )
+import ErrUtils 	( WarnMsg )
+import Id		( mkSysLocal, setIdUnique, Id )
+import Module		( Module )
+import Var		( TyVar, setTyVarUnique )
+import Outputable
 import SrcLoc		( noSrcLoc, SrcLoc )
-import TcHsSyn		( SYN_IE(TypecheckedPat) )
-import TyVar		( nullTyVarEnv, cloneTyVar, GenTyVar{-instance Eq-} )
-import Unique		( Unique{-instances-} )
-import UniqSupply	( splitUniqSupply, getUnique, getUniques,
-			  mapUs, thenUs, returnUs, SYN_IE(UniqSM) )
-import Util		( assoc, mapAccumL, zipWithEqual, panic )
+import TcHsSyn		( TypecheckedPat )
+import Type             ( Type )
+import UniqSupply	( initUs_, splitUniqSupply, uniqFromSupply, uniqsFromSupply,
+			  UniqSM, UniqSupply )
+import Unique		( Unique )
+import Util		( zipWithEqual )
+import Name		( Name, lookupNameEnv )
+import HscTypes		( HomeSymbolTable, PersistentCompilerState(..), 
+			  TyThing(..), TypeEnv, lookupTypeEnv )
+import CmdLineOpts	( DynFlags )
 
 infixr 9 `thenDs`
 \end{code}
@@ -53,51 +51,65 @@ a @UniqueSupply@ and some annotations, which
 presumably include source-file location information:
 \begin{code}
 type DsM result =
-	UniqSupply
-	-> SrcLoc			-- to put in pattern-matching error msgs
-	-> (FAST_STRING, FAST_STRING)	-- "module"+"group" : for SCC profiling
-	-> DsIdEnv
+	DynFlags
+	-> UniqSupply
+        -> (Name -> Id)		-- Lookup well-known Ids
+	-> SrcLoc		-- to put in pattern-matching error msgs
+	-> Module       	-- module: for SCC profiling
 	-> DsWarnings
 	-> (result, DsWarnings)
 
-type DsWarnings = Bag DsMatchContext	-- The desugarer reports matches which are
-					-- completely shadowed
+type DsWarnings = Bag WarnMsg           -- The desugarer reports matches which are
+					-- completely shadowed or incomplete patterns
+
 {-# INLINE andDs #-}
 {-# INLINE thenDs #-}
 {-# INLINE returnDs #-}
 
 -- initDs returns the UniqSupply out the end (not just the result)
 
-initDs  :: UniqSupply
-	-> DsIdEnv
-	-> FAST_STRING -- module name: for profiling; (group name: from switches)
+initDs  :: DynFlags
+	-> UniqSupply
+	-> (HomeSymbolTable, PersistentCompilerState, TypeEnv)
+	-> Module   -- module name: for profiling
 	-> DsM a
 	-> (a, DsWarnings)
 
-initDs init_us env mod_name action
-  = action init_us noSrcLoc module_and_group env emptyBag
+initDs dflags init_us (hst,pcs,local_type_env) mod action
+  = action dflags init_us lookup noSrcLoc mod emptyBag
   where
-    module_and_group = (mod_name, grp_name)
-    grp_name  = case opt_SccGroup of
-		    Just xx -> _PK_ xx
-		    Nothing -> mod_name	-- default: module name
+	-- This lookup is used for well-known Ids, 
+	-- such as fold, build, cons etc, so the chances are
+	-- it'll be found in the package symbol table.  That's
+	-- why we don't merge all these tables
+    pst = pcs_PST pcs
+    lookup n = case lookupTypeEnv pst n of {
+		 Just (AnId v) -> v ;
+		 other -> 
+	       case lookupTypeEnv hst n of {
+		 Just (AnId v) -> v ;
+		 other -> 
+	       case lookupNameEnv local_type_env n of
+		 Just (AnId v) -> v ;
+		 other	       -> pprPanic "initDS: lookup:" (ppr n)
+               }}
 
 thenDs :: DsM a -> (a -> DsM b) -> DsM b
 andDs  :: (a -> a -> a) -> DsM a -> DsM a -> DsM a
 
-thenDs m1 m2 us loc mod_and_grp env warns
+thenDs m1 m2 dflags us genv loc mod warns
   = case splitUniqSupply us		    of { (s1, s2) ->
-    case (m1 s1 loc mod_and_grp env warns)  of { (result, warns1) ->
-    m2 result s2 loc mod_and_grp env warns1}}
+    case (m1 dflags s1 genv loc mod warns)  of { (result, warns1) ->
+    m2 result dflags s2 genv loc mod warns1}}
 
-andDs combiner m1 m2 us loc mod_and_grp env warns
+andDs combiner m1 m2 dflags us genv loc mod warns
   = case splitUniqSupply us		    of { (s1, s2) ->
-    case (m1 s1 loc mod_and_grp env warns)  of { (result1, warns1) ->
-    case (m2 s2 loc mod_and_grp env warns1) of { (result2, warns2) ->
+    case (m1 dflags s1 genv loc mod warns)  of { (result1, warns1) ->
+    case (m2 dflags s2 genv loc mod warns1) of { (result2, warns2) ->
     (combiner result1 result2, warns2) }}}
 
 returnDs :: a -> DsM a
-returnDs result us loc mod_and_grp env warns = (result, warns)
+returnDs result dflags us genv loc mod warns = (result, warns)
 
 listDs :: [DsM a] -> DsM [a]
 listDs []     = returnDs []
@@ -114,6 +126,12 @@ mapDs f (x:xs)
     mapDs f xs	`thenDs` \ rs ->
     returnDs (r:rs)
 
+foldlDs :: (a -> b -> DsM a) -> a -> [b] -> DsM a
+
+foldlDs k z []     = returnDs z
+foldlDs k z (x:xs) = k z x `thenDs` \ r ->
+		     foldlDs k r xs
+
 mapAndUnzipDs :: (a -> DsM (b, c)) -> [a] -> DsM ([b], [c])
 
 mapAndUnzipDs f []     = returnDs ([], [])
@@ -124,45 +142,57 @@ mapAndUnzipDs f (x:xs)
 
 zipWithDs :: (a -> b -> DsM c) -> [a] -> [b] -> DsM [c]
 
-zipWithDs f []	   [] = returnDs []
+zipWithDs f []	   ys = returnDs []
 zipWithDs f (x:xs) (y:ys)
   = f x y		`thenDs` \ r  ->
     zipWithDs f xs ys	`thenDs` \ rs ->
     returnDs (r:rs)
--- Note: crashes if lists not equal length (like zipWithEqual)
 \end{code}
 
 And all this mysterious stuff is so we can occasionally reach out and
 grab one or more names.  @newLocalDs@ isn't exported---exported
 functions are defined with it.  The difference in name-strings makes
 it easier to read debugging output.
+
 \begin{code}
-newLocalDs :: FAST_STRING -> Type -> DsM Id
-newLocalDs nm ty us loc mod_and_grp env warns
-  = case (getUnique us) of { assigned_uniq ->
-    (mkSysLocal nm assigned_uniq ty loc, warns) }
+newSysLocalDs, newFailLocalDs :: Type -> DsM Id
+newSysLocalDs ty dflags us genv loc mod warns
+  = case uniqFromSupply us of { assigned_uniq ->
+    (mkSysLocal SLIT("ds") assigned_uniq ty, warns) }
+
+newSysLocalsDs tys = mapDs newSysLocalDs tys
 
-newSysLocalDs	    = newLocalDs SLIT("ds")
-newSysLocalsDs tys  = mapDs (newLocalDs SLIT("ds")) tys
-newFailLocalDs	    = newLocalDs SLIT("fail")
+newFailLocalDs ty dflags us genv loc mod warns
+  = case uniqFromSupply us of { assigned_uniq ->
+    (mkSysLocal SLIT("fail") assigned_uniq ty, warns) }
+	-- The UserLocal bit just helps make the code a little clearer
+
+getUniqueDs :: DsM Unique
+getUniqueDs dflags us genv loc mod warns
+  = case (uniqFromSupply us) of { assigned_uniq ->
+    (assigned_uniq, warns) }
+
+getDOptsDs :: DsM DynFlags
+getDOptsDs dflags us genv loc mod warns
+  = (dflags, warns)
 
 duplicateLocalDs :: Id -> DsM Id
-duplicateLocalDs old_local us loc mod_and_grp env warns
-  = case (getUnique us) of { assigned_uniq ->
-    (mkIdWithNewUniq old_local assigned_uniq, warns) }
+duplicateLocalDs old_local dflags us genv loc mod warns
+  = case uniqFromSupply us of { assigned_uniq ->
+    (setIdUnique old_local assigned_uniq, warns) }
 
 cloneTyVarsDs :: [TyVar] -> DsM [TyVar]
-cloneTyVarsDs tyvars us loc mod_and_grp env warns
-  = case (getUniques (length tyvars) us) of { uniqs ->
-    (zipWithEqual "cloneTyVarsDs" cloneTyVar tyvars uniqs, warns) }
+cloneTyVarsDs tyvars dflags us genv loc mod warns
+  = case uniqsFromSupply (length tyvars) us of { uniqs ->
+    (zipWithEqual "cloneTyVarsDs" setTyVarUnique tyvars uniqs, warns) }
 \end{code}
 
 \begin{code}
 newTyVarsDs :: [TyVar] -> DsM [TyVar]
 
-newTyVarsDs tyvar_tmpls us loc mod_and_grp env warns
-  = case (getUniques (length tyvar_tmpls) us) of { uniqs ->
-    (zipWithEqual "newTyVarsDs" cloneTyVar tyvar_tmpls uniqs, warns) }
+newTyVarsDs tyvar_tmpls dflags us genv loc mod warns
+  = case uniqsFromSupply (length tyvar_tmpls) us of { uniqs ->
+    (zipWithEqual "newTyVarsDs" setTyVarUnique tyvar_tmpls uniqs, warns) }
 \end{code}
 
 We can also reach out and either set/grab location information from
@@ -170,69 +200,37 @@ the @SrcLoc@ being carried around.
 \begin{code}
 uniqSMtoDsM :: UniqSM a -> DsM a
 
-uniqSMtoDsM u_action us loc mod_and_grp env warns
-  = (u_action us, warns)
+uniqSMtoDsM u_action dflags us genv loc mod warns
+  = (initUs_ us u_action, warns)
 
 getSrcLocDs :: DsM SrcLoc
-getSrcLocDs us loc mod_and_grp env warns
+getSrcLocDs dflags us genv loc mod warns
   = (loc, warns)
 
 putSrcLocDs :: SrcLoc -> DsM a -> DsM a
-putSrcLocDs new_loc expr us old_loc mod_and_grp env warns
-  = expr us new_loc mod_and_grp env warns
+putSrcLocDs new_loc expr dflags us genv old_loc mod warns
+  = expr dflags us genv new_loc mod warns
+
+dsWarn :: WarnMsg -> DsM ()
+dsWarn warn dflags us genv loc mod warns = ((), warns `snocBag` warn)
 
-dsShadowError :: DsMatchContext -> DsM ()
-dsShadowError cxt us loc mod_and_grp env warns
-  = ((), warns `snocBag` cxt)
 \end{code}
 
 \begin{code}
-getModuleAndGroupDs :: DsM (FAST_STRING, FAST_STRING)
-getModuleAndGroupDs us loc mod_and_grp env warns
-  = (mod_and_grp, warns)
+getModuleDs :: DsM Module
+getModuleDs dflags us genv loc mod warns = (mod, warns)
 \end{code}
 
 \begin{code}
-type DsIdEnv = IdEnv CoreExpr
-
-extendEnvDs :: [(Id, CoreExpr)] -> DsM a -> DsM a
-
-extendEnvDs pairs then_do us loc mod_and_grp old_env warns
-  = case splitUniqSupply us 	    of { (s1, s2) ->
-    let
-	revised_pairs = subst_all pairs s1
-    in
-    then_do s2 loc mod_and_grp (growIdEnvList old_env revised_pairs) warns
-    }
-  where
-    subst_all pairs = mapUs subst pairs
-
-    subst (v, expr)
-      = substCoreExpr old_env nullTyVarEnv expr `thenUs` \ new_expr ->
-	returnUs (v, new_expr)
-
-lookupEnvDs :: Id -> DsM (Maybe CoreExpr)
-lookupEnvDs id us loc mod_and_grp env warns
-  = (lookupIdEnv env id, warns)
-  -- Note: we don't assert anything about the Id
-  -- being looked up.  There's not really anything
-  -- much to say about it. (WDP 94/06)
-
-lookupEnvWithDefaultDs :: Id -> CoreExpr -> DsM CoreExpr
-lookupEnvWithDefaultDs id deflt us loc mod_and_grp env warns
-  = (case (lookupIdEnv env id) of
-      Nothing -> deflt
-      Just xx -> xx,
-     warns)
-
-lookupId :: [(Id, a)] -> Id -> a
-lookupId env id
-  = assoc "lookupId" env id
+dsLookupGlobalValue :: Name -> DsM Id
+dsLookupGlobalValue name dflags us genv loc mod warns
+  = (genv name, warns)
 \end{code}
 
+
 %************************************************************************
 %*									*
-%* type synonym EquationInfo and access functions for its pieces	*
+\subsection{Type synonym @EquationInfo@ and access functions for its pieces}
 %*									*
 %************************************************************************
 
@@ -240,6 +238,7 @@ lookupId env id
 data DsMatchContext
   = DsMatchContext DsMatchKind [TypecheckedPat] SrcLoc
   | NoMatchContext
+  deriving ()
 
 data DsMatchKind
   = FunMatch Id
@@ -247,36 +246,8 @@ data DsMatchKind
   | LambdaMatch
   | PatBindMatch
   | DoBindMatch
-
-pprDsWarnings :: PprStyle -> Bag DsMatchContext -> Pretty
-pprDsWarnings sty warns
-  = ppAboves (map pp_cxt (bagToList warns))
-  where
-    pp_cxt NoMatchContext = ppPStr SLIT("Some match is shadowed; I don't know what")
-    pp_cxt (DsMatchContext kind pats loc)
-      = ppHang (ppBesides [ppr PprForUser loc, ppPStr SLIT(": ")])
-	     4 (ppHang (ppPStr SLIT("Pattern match(es) completely overlapped:"))
-		     4 (pp_match kind pats))
-
-    pp_match (FunMatch fun) pats
-      = ppHang (ppr sty fun)
-	4 (ppSep [ppSep (map (ppr sty) pats), ppPStr SLIT("= ...")])
-
-    pp_match CaseMatch pats
-      = ppHang (ppPStr SLIT("in a case alternative:"))
-	4 (ppSep [ppSep (map (ppr sty) pats), pp_arrow_dotdotdot])
-
-    pp_match PatBindMatch pats
-      = ppHang (ppPStr SLIT("in a pattern binding:"))
-	4 (ppSep [ppSep (map (ppr sty) pats), pp_arrow_dotdotdot])
-
-    pp_match LambdaMatch pats
-      = ppHang (ppPStr SLIT("in a lambda abstraction:"))
-	4 (ppSep [ppSep (map (ppr sty) pats), pp_arrow_dotdotdot])
-
-    pp_match DoBindMatch pats
-      = ppHang (ppPStr SLIT("in a `do' pattern binding:"))
-	4 (ppSep [ppSep (map (ppr sty) pats), pp_arrow_dotdotdot])
-
-    pp_arrow_dotdotdot = ppPStr SLIT("-> ...")
+  | ListCompMatch
+  | LetMatch
+  | RecUpdMatch
+  deriving ()
 \end{code}