X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2FhsSyn%2FHsImpExp.lhs;h=f63d86aec2af50e52ad8f6c22f86be6066b5ebe0;hb=ff845ab59d1d465d874d3908fd0cdd61b8594da2;hp=b1d462da8689139db3e6178dfc76487fe03c7039;hpb=2f51f1402e6869c0f049ffbe7b019bf6ab80558f;p=ghc-hetmet.git

diff --git a/ghc/compiler/hsSyn/HsImpExp.lhs b/ghc/compiler/hsSyn/HsImpExp.lhs
index b1d462d..f63d86a 100644
--- a/ghc/compiler/hsSyn/HsImpExp.lhs
+++ b/ghc/compiler/hsSyn/HsImpExp.lhs
@@ -1,19 +1,18 @@
 %
-% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996
+% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
 %
 \section[HsImpExp]{Abstract syntax: imports, exports, interfaces}
 
 \begin{code}
-#include "HsVersions.h"
-
 module HsImpExp where
 
-import Ubiq
+#include "HsVersions.h"
 
+import Module		( ModuleName )
 import Outputable
-import PprStyle		( PprStyle(..) )
-import Pretty
-import SrcLoc		( SrcLoc )
+import FastString
+import SrcLoc		( Located(..) )
+import Char		( isAlpha )
 \end{code}
 
 %************************************************************************
@@ -24,32 +23,39 @@ import SrcLoc		( SrcLoc )
 
 One per \tr{import} declaration in a module.
 \begin{code}
+type LImportDecl name = Located (ImportDecl name)
+
 data ImportDecl name
-  = ImportDecl	  Module			-- module name
+  = ImportDecl	  (Located ModuleName)		-- module name
+		  Bool				-- True <=> {-# SOURCE #-} import
 		  Bool				-- True => qualified
-		  (Maybe Module)		-- as Module
-		  (Maybe (Bool, [IE name]))	-- (True => hiding, names)
-		  SrcLoc
+		  (Maybe ModuleName)		-- as Module
+		  (Maybe (Bool, [LIE name]))	-- (True => hiding, names)
 \end{code}
 
 \begin{code}
 instance (Outputable name) => Outputable (ImportDecl name) where
-    ppr sty (ImportDecl mod qual as spec _)
-      = ppHang (ppCat [ppStr "import", pp_qual qual, ppPStr mod, pp_as as])
+    ppr (ImportDecl mod from qual as spec)
+      = hang (hsep [ptext SLIT("import"), ppr_imp from, 
+                    pp_qual qual, ppr mod, pp_as as])
 	     4 (pp_spec spec)
       where
-	pp_qual False   = ppNil
-	pp_qual True	= ppStr "qualified"
+	pp_qual False   = empty
+	pp_qual True	= ptext SLIT("qualified")
 
-	pp_as Nothing   = ppNil
-	pp_as (Just a)  = ppCat [ppStr "as", ppPStr a]
+	pp_as Nothing   = empty
+	pp_as (Just a)  = ptext SLIT("as ") <+> ppr a
 
-	pp_spec Nothing = ppNil
+	ppr_imp True  = ptext SLIT("{-# SOURCE #-}")
+	ppr_imp False = empty
+
+	pp_spec Nothing = empty
 	pp_spec (Just (False, spec))
-			= ppBesides [ppStr "(", interpp'SP sty spec, ppStr ")"]
+			= parens (interpp'SP spec)
 	pp_spec (Just (True, spec))
-			= ppBesides [ppStr "hiding (", interpp'SP sty spec, ppStr ")"]
+			= ptext SLIT("hiding") <+> parens (interpp'SP spec)
 
+ideclName (ImportDecl mod_nm _ _ _ _) = mod_nm
 \end{code}
 
 %************************************************************************
@@ -57,23 +63,63 @@ instance (Outputable name) => Outputable (ImportDecl name) where
 \subsection{Imported and exported entities}
 %*									*
 %************************************************************************
+
 \begin{code}
+type LIE name = Located (IE name)
+
 data IE name
   = IEVar		name
   | IEThingAbs          name		-- Class/Type (can't tell)
   | IEThingAll          name		-- Class/Type plus all methods/constructors
   | IEThingWith		name [name]	-- Class/Type plus some methods/constructors
-  | IEModuleContents    Module		-- (Export Only)
+  | IEModuleContents    ModuleName	-- (Export Only)
+\end{code}
+
+\begin{code}
+ieName :: IE name -> name
+ieName (IEVar n) 	 = n
+ieName (IEThingAbs  n)   = n
+ieName (IEThingWith n _) = n
+ieName (IEThingAll  n)   = n
+
+ieNames :: IE a -> [a]
+ieNames (IEVar            n   ) = [n]
+ieNames (IEThingAbs       n   ) = [n]
+ieNames (IEThingAll       n   ) = [n]
+ieNames (IEThingWith      n ns) = n:ns
+ieNames (IEModuleContents _   ) = []
 \end{code}
 
 \begin{code}
 instance (Outputable name) => Outputable (IE name) where
-    ppr sty (IEVar	var)	= ppr sty var
-    ppr sty (IEThingAbs	thing)	= ppr sty thing
-    ppr sty (IEThingAll	thing)
-	= ppBesides [ppr sty thing, ppStr "(..)"]
-    ppr sty (IEThingWith thing withs)
-	= ppBesides [ppr sty thing, ppLparen, ppInterleave ppComma (map (ppr sty) withs), ppRparen]
-    ppr sty (IEModuleContents mod)
-	= ppBeside (ppPStr SLIT("module ")) (ppPStr mod)
+    ppr (IEVar	        var)	= pprHsVar var
+    ppr (IEThingAbs	thing)	= ppr thing
+    ppr (IEThingAll	thing)	= hcat [ppr thing, text "(..)"]
+    ppr (IEThingWith thing withs)
+	= ppr thing <> parens (fsep (punctuate comma (map pprHsVar withs)))
+    ppr (IEModuleContents mod)
+	= ptext SLIT("module") <+> ppr mod
+\end{code}
+
+\begin{code}
+pprHsVar :: Outputable name => name -> SDoc
+pprHsVar v | isOperator ppr_v = parens ppr_v
+	   | otherwise	      = ppr_v
+	   where
+	     ppr_v = ppr v
+
+isOperator :: SDoc -> Bool
+isOperator ppr_v 
+  = case showSDocUnqual ppr_v of
+	('(':s)   -> False		-- (), (,) etc
+	('[':s)   -> False		-- []
+	('$':c:s) -> not (isAlpha c)	-- Don't treat $d as an operator
+	(':':c:s) -> not (isAlpha c)	-- Don't treat :T as an operator
+	('_':s)   -> False		-- Not an operator
+	(c:s)     -> not (isAlpha c)	-- Starts with non-alpha
+	other     -> False
+    -- We use (showSDoc (ppr v)), rather than isSymOcc (getOccName v) simply so
+    -- that we don't need NamedThing in the context of all these functions.
+    -- Gruesome, but simple.
 \end{code}
+