[project @ 2005-10-14 11:22:41 by simonpj]
[ghc-hetmet.git] / ghc / compiler / rename / RnHsSyn.lhs
index 3dd375f..6ce0379 100644 (file)
@@ -1,58 +1,30 @@
 %
-% (c) The AQUA Project, Glasgow University, 1996
+% (c) The AQUA Project, Glasgow University, 1996-1998
 %
 \section[RnHsSyn]{Specialisations of the @HsSyn@ syntax for the renamer}
 
 \begin{code}
-module RnHsSyn where
+module RnHsSyn( 
+       -- Names
+       charTyCon_name, listTyCon_name, parrTyCon_name, tupleTyCon_name,
+       extractHsTyVars, extractHsTyNames, extractHsTyNames_s, 
+       extractFunDepNames, extractHsCtxtTyNames, extractHsPredTyNames,
+
+       -- Free variables
+       hsSigsFVs, hsSigFVs, conDeclFVs, bangTyFVs,
+       
+       maybeGenericMatch
+  ) where
 
 #include "HsVersions.h"
 
 import HsSyn
-import HsPragmas       ( InstancePragmas, GenPragmas, DataPragmas, ClassPragmas, ClassOpPragmas )
-
-import Id              ( GenId, Id )
-import BasicTypes      ( Unused, NewOrData, IfaceFlavour )
-import Name            ( Name )
-import Name            ( NameSet, unitNameSet, mkNameSet, minusNameSet, unionNameSets, emptyNameSet )
-import TyVar           ( GenTyVar )
-import Unique          ( Unique )
-import Util
-import Outputable
-\end{code}
-
-
-\begin{code}
-type RenamedArithSeqInfo       = ArithSeqInfo          Unused Name RenamedPat
-type RenamedClassDecl          = ClassDecl             Unused Name RenamedPat
-type RenamedClassOpSig         = Sig                   Name
-type RenamedConDecl            = ConDecl               Name
-type RenamedContext            = Context               Name
-type RenamedHsDecl             = HsDecl                Unused Name RenamedPat
-type RenamedSpecDataSig                = SpecDataSig           Name
-type RenamedDefaultDecl                = DefaultDecl           Name
-type RenamedFixityDecl         = FixityDecl            Name
-type RenamedGRHS               = GRHS                  Unused Name RenamedPat
-type RenamedGRHSsAndBinds      = GRHSsAndBinds         Unused Name RenamedPat
-type RenamedHsBinds            = HsBinds               Unused Name RenamedPat
-type RenamedHsExpr             = HsExpr                Unused Name RenamedPat
-type RenamedHsModule           = HsModule              Unused Name RenamedPat
-type RenamedInstDecl           = InstDecl              Unused Name RenamedPat
-type RenamedMatch              = Match                 Unused Name RenamedPat
-type RenamedMonoBinds          = MonoBinds             Unused Name RenamedPat
-type RenamedPat                        = InPat                 Name
-type RenamedHsType             = HsType                Name
-type RenamedRecordBinds                = HsRecordBinds         Unused Name RenamedPat
-type RenamedSig                        = Sig                   Name
-type RenamedSpecInstSig                = SpecInstSig           Name
-type RenamedStmt               = Stmt                  Unused Name RenamedPat
-type RenamedTyDecl             = TyDecl                Name
-
-type RenamedClassOpPragmas     = ClassOpPragmas        Name
-type RenamedClassPragmas       = ClassPragmas          Name
-type RenamedDataPragmas                = DataPragmas           Name
-type RenamedGenPragmas         = GenPragmas            Name
-type RenamedInstancePragmas    = InstancePragmas       Name
+import Class           ( FunDep )
+import TysWiredIn      ( tupleTyCon, listTyCon, parrTyCon, charTyCon )
+import Name            ( Name, getName, isTyVarName )
+import NameSet
+import BasicTypes      ( Boxity )
+import SrcLoc          ( Located(..), unLoc )
 \end{code}
 
 %************************************************************************
@@ -64,26 +36,123 @@ type RenamedInstancePragmas        = InstancePragmas       Name
 These free-variable finders returns tycons and classes too.
 
 \begin{code}
-extractHsTyNames   :: RenamedHsType -> NameSet
+charTyCon_name, listTyCon_name, parrTyCon_name :: Name
+charTyCon_name    = getName charTyCon
+listTyCon_name    = getName listTyCon
+parrTyCon_name    = getName parrTyCon
+
+tupleTyCon_name :: Boxity -> Int -> Name
+tupleTyCon_name boxity n = getName (tupleTyCon boxity n)
+
+extractHsTyVars :: LHsType Name -> NameSet
+extractHsTyVars x = filterNameSet isTyVarName (extractHsTyNames x)
+
+extractFunDepNames :: FunDep Name -> NameSet
+extractFunDepNames (ns1, ns2) = mkNameSet ns1 `unionNameSets` mkNameSet ns2
+
+extractHsTyNames   :: LHsType Name -> NameSet
 extractHsTyNames ty
-  = get ty
+  = getl ty
   where
-    get (MonoTyApp ty1 ty2)      = get ty1 `unionNameSets` get ty2
-    get (MonoListTy tc ty)       = unitNameSet tc `unionNameSets` get ty
-    get (MonoTupleTy tc tys)     = unitNameSet tc `unionNameSets` extractHsTyNames_s tys
-    get (MonoFunTy ty1 ty2)      = get ty1 `unionNameSets` get ty2
-    get (MonoDictTy cls tys)     = unitNameSet cls `unionNameSets` extractHsTyNames_s tys
-    get (MonoTyVar tv)          = unitNameSet tv
-    get (HsForAllTy tvs ctxt ty) = (extractHsCtxtTyNames ctxt `unionNameSets` get ty)
+    getl (L _ ty) = get ty
+
+    get (HsAppTy ty1 ty2)      = getl ty1 `unionNameSets` getl ty2
+    get (HsListTy ty)          = unitNameSet listTyCon_name `unionNameSets` getl ty
+    get (HsPArrTy ty)          = unitNameSet parrTyCon_name `unionNameSets` getl ty
+    get (HsTupleTy con tys)    = extractHsTyNames_s tys
+    get (HsFunTy ty1 ty2)      = getl ty1 `unionNameSets` getl ty2
+    get (HsPredTy p)          = extractHsPredTyNames p
+    get (HsOpTy ty1 op ty2)    = getl ty1 `unionNameSets` getl ty2 `unionNameSets` unitNameSet (unLoc op)
+    get (HsParTy ty)           = getl ty
+    get (HsBangTy _ ty)        = getl ty
+    get (HsNumTy n)            = emptyNameSet
+    get (HsTyVar tv)          = unitNameSet tv
+    get (HsSpliceTy _)         = emptyNameSet  -- Type splices mention no type variables
+    get (HsKindSig ty k)       = getl ty
+    get (HsForAllTy _ tvs 
+                   ctxt ty)   = (extractHsCtxtTyNames ctxt
+                                        `unionNameSets` getl ty)
                                            `minusNameSet`
-                                   mkNameSet (map getTyVarName tvs)
+                                 mkNameSet (hsLTyVarNames tvs)
 
-extractHsTyNames_s  :: [RenamedHsType] -> NameSet
+extractHsTyNames_s  :: [LHsType Name] -> NameSet
 extractHsTyNames_s tys = foldr (unionNameSets . extractHsTyNames) emptyNameSet tys
 
-extractHsCtxtTyNames :: RenamedContext -> NameSet
-extractHsCtxtTyNames ctxt = foldr (unionNameSets . get) emptyNameSet ctxt
-  where
-    get (cls, tys) = unitNameSet cls `unionNameSets` extractHsTyNames_s tys
+extractHsCtxtTyNames :: LHsContext Name -> NameSet
+extractHsCtxtTyNames (L _ ctxt)
+  = foldr (unionNameSets . extractHsPredTyNames . unLoc) emptyNameSet ctxt
+
+-- You don't import or export implicit parameters,
+-- so don't mention the IP names
+extractHsPredTyNames (HsClassP cls tys)
+  = unitNameSet cls `unionNameSets` extractHsTyNames_s tys
+extractHsPredTyNames (HsIParam n ty)
+  = extractHsTyNames ty
 \end{code}
 
+
+%************************************************************************
+%*                                                                     *
+\subsection{Free variables of declarations}
+%*                                                                     *
+%************************************************************************
+
+Return the Names that must be in scope if we are to use this declaration.
+In all cases this is set up for interface-file declarations:
+       - for class decls we ignore the bindings
+       - for instance decls likewise, plus the pragmas
+       - for rule decls, we ignore HsRules
+        - for data decls, we ignore derivings
+
+       *** See "THE NAMING STORY" in HsDecls ****
+
+\begin{code}
+----------------
+hsSigsFVs :: [LSig Name] -> FreeVars
+hsSigsFVs sigs = plusFVs (map (hsSigFVs.unLoc) sigs)
+
+hsSigFVs (Sig v ty)        = extractHsTyNames ty
+hsSigFVs (SpecInstSig ty)   = extractHsTyNames ty
+hsSigFVs (SpecSig v ty)     = extractHsTyNames ty
+hsSigFVs other             = emptyFVs
+
+----------------
+-- XXX - autrijus - handle return type for GADT
+conDeclFVs (L _ (ConDecl _ _ tyvars context details _))
+  = delFVs (map hsLTyVarName tyvars) $
+    extractHsCtxtTyNames context         `plusFV`
+    conDetailsFVs details
+
+{-
+-- gaw 2004
+conDeclFVs (L _ (GadtDecl _ ty)) 
+  = extractHsTyNames ty
+-}
+
+conDetailsFVs (PrefixCon btys)     = plusFVs (map bangTyFVs btys)
+conDetailsFVs (InfixCon bty1 bty2) = bangTyFVs bty1 `plusFV` bangTyFVs bty2
+conDetailsFVs (RecCon flds)       = plusFVs [bangTyFVs bty | (_, bty) <- flds]
+
+bangTyFVs bty = extractHsTyNames (getBangType bty)
+\end{code}
+
+
+%************************************************************************
+%*                                                                     *
+\subsection{A few functions on generic defintions
+%*                                                                     *
+%************************************************************************
+
+These functions on generics are defined over Matches Name, which is
+why they are here and not in HsMatches.
+
+\begin{code}
+maybeGenericMatch :: LMatch Name -> Maybe (HsType Name, LMatch Name)
+  -- Tells whether a Match is for a generic definition
+  -- and extract the type from a generic match and put it at the front
+
+maybeGenericMatch (L loc (Match (L _ (TypePat (L _ ty)) : pats) sig_ty grhss))
+  = Just (ty, L loc (Match pats sig_ty grhss))
+
+maybeGenericMatch other_match = Nothing
+\end{code}