[project @ 2004-11-26 16:19:45 by simonmar]

[ghc-hetmet.git] / ghc / compiler / hsSyn / HsImpExp.lhs

diff --git a/ghc/compiler/hsSyn/HsImpExp.lhs b/ghc/compiler/hsSyn/HsImpExp.lhs
index 97c23f4..220afb7 100644 (file)
--- a/ghc/compiler/hsSyn/HsImpExp.lhs
+++ b/ghc/compiler/hsSyn/HsImpExp.lhs
@@ -1,5 +1,5 @@
 %
-% (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}
 
@@ -8,10 +8,11 @@ module HsImpExp where
 
 #include "HsVersions.h"
 
-import BasicTypes      ( Module, IfaceFlavour(..) )
-import Name            ( NamedThing )
+import Module          ( Module )
 import Outputable
-import SrcLoc          ( SrcLoc )
+import FastString
+import SrcLoc          ( Located(..) )
+import Char            ( isAlpha )
 \end{code}
 
 %************************************************************************
@@ -22,37 +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 Module)              -- module name
+                 Bool                          -- True <=> {-# SOURCE #-} import
                  Bool                          -- True => qualified
-                 IfaceFlavour                  -- True => source imported module 
-                                               --    (current interpretation: ignore ufolding info)
                  (Maybe Module)                -- as Module
-                 (Maybe (Bool, [IE name]))     -- (True => hiding, names)
-                 SrcLoc
+                 (Maybe (Bool, [LIE name]))    -- (True => hiding, names)
 \end{code}
 
 \begin{code}
-instance (NamedThing name, Outputable name) => Outputable (ImportDecl name) where
-    ppr (ImportDecl mod qual as_source as spec _)
-      = hang (hsep [ptext SLIT("import"), pp_src as_source, 
-                    pp_qual qual, ptext mod, pp_as as])
+instance (Outputable name) => Outputable (ImportDecl name) where
+    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_src HiFile     = empty
-       pp_src HiBootFile = ptext SLIT("{-# SOURCE #-}")
-
        pp_qual False   = empty
        pp_qual True    = ptext SLIT("qualified")
 
        pp_as Nothing   = empty
-       pp_as (Just a)  = ptext SLIT("as ") <+> ptext a
+       pp_as (Just a)  = ptext SLIT("as ") <+> ppr a
+
+       ppr_imp True  = ptext SLIT("{-# SOURCE #-}")
+       ppr_imp False = empty
 
        pp_spec Nothing = empty
        pp_spec (Just (False, spec))
                        = parens (interpp'SP spec)
        pp_spec (Just (True, spec))
                        = ptext SLIT("hiding") <+> parens (interpp'SP spec)
+
+ideclName (ImportDecl mod_nm _ _ _ _) = mod_nm
 \end{code}
 
 %************************************************************************
@@ -62,6 +65,8 @@ instance (NamedThing name, Outputable name) => Outputable (ImportDecl name) wher
 %************************************************************************
 
 \begin{code}
+type LIE name = Located (IE name)
+
 data IE name
   = IEVar              name
   | IEThingAbs          name           -- Class/Type (can't tell)
@@ -76,16 +81,45 @@ 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 (NamedThing name, Outputable name) => Outputable (IE name) where
-    ppr (IEVar         var)    = ppr var
+instance (Outputable name) => Outputable (IE name) where
+    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 ppr withs)))
+       = ppr thing <> parens (fsep (punctuate comma (map pprHsVar withs)))
     ppr (IEModuleContents mod)
-       = ptext SLIT("module") <+> ptext 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}