[project @ 2000-11-06 08:15:20 by simonpj]
[ghc-hetmet.git] / ghc / compiler / typecheck / TcTyClsDecls.lhs
index ae7e4d2..b92276e 100644 (file)
@@ -11,18 +11,18 @@ module TcTyClsDecls (
 #include "HsVersions.h"
 
 import HsSyn           ( HsDecl(..), TyClDecl(..),
-                         HsType(..), HsTyVarBndr,
-                         ConDecl(..), ConDetails(..), 
-                         Sig(..), HsPred(..), HsTupCon(..),
-                         tyClDeclName, hsTyVarNames, isClassDecl, isSynDecl, isClassOpSig, getBangType
+                         HsTyVarBndr,
+                         ConDecl(..), 
+                         Sig(..), HsPred(..), 
+                         tyClDeclName, hsTyVarNames, 
+                         isIfaceSigDecl, isClassDecl, isSynDecl, isClassOpSig
                        )
-import RnHsSyn         ( RenamedHsDecl, RenamedTyClDecl, listTyCon_name )
+import RnHsSyn         ( RenamedHsDecl, RenamedTyClDecl, tyClDeclFVs )
 import BasicTypes      ( RecFlag(..), NewOrData(..) )
 
 import TcMonad
-import TcEnv           ( TcEnv, TyThing(..), TyThingDetails(..), tyThingKind,
-                         tcExtendTypeEnv, tcExtendKindEnv, tcLookupGlobal
-                       )
+import TcEnv           ( TcEnv, RecTcEnv, TcTyThing(..), TyThing(..), TyThingDetails(..),
+                         tcExtendKindEnv, tcLookup, tcExtendGlobalEnv )
 import TcTyDecls       ( tcTyDecl1, kcConDetails, mkNewTyConRep )
 import TcClassDcl      ( tcClassDecl1 )
 import TcMonoType      ( kcHsTyVars, kcHsType, kcHsBoxedSigType, kcHsContext, mkTyClTyVars )
@@ -33,22 +33,21 @@ import TcInstDcls   ( tcAddDeclCtxt )
 import Type            ( Kind, mkArrowKind, boxedTypeKind, zipFunTys )
 import Variance         ( calcTyConArgVrcs )
 import Class           ( Class, mkClass, classTyCon )
-import TyCon           ( TyCon, ArgVrcs, AlgTyConFlavour(..), mkSynTyCon, mkAlgTyConRep, mkClassTyCon )
+import TyCon           ( TyCon, tyConKind, ArgVrcs, AlgTyConFlavour(..), 
+                         mkSynTyCon, mkAlgTyConRep, mkClassTyCon )
 import DataCon         ( isNullaryDataCon )
 import Var             ( varName )
 import FiniteMap
 import Digraph         ( stronglyConnComp, SCC(..) )
-import Name            ( Name, NamedThing(..), NameEnv, getSrcLoc, isTvOcc, nameOccName,
-                         mkNameEnv, lookupNameEnv_NF
-                       )
+import Name            ( Name, NamedThing(..), getSrcLoc, isTyVarName )
+import Name            ( NameEnv, mkNameEnv, lookupNameEnv_NF )
+import NameSet
 import Outputable
-import Maybes          ( mapMaybe, catMaybes )
-import UniqSet         ( emptyUniqSet, unitUniqSet, unionUniqSets, 
-                         unionManyUniqSets, uniqSetToList ) 
+import Maybes          ( mapMaybe )
 import ErrUtils                ( Message )
-import Unique          ( Unique, Uniquable(..) )
-import HsDecls          ( fromClassDeclNameList )
+import HsDecls          ( getClassDeclSysNames )
 import Generics         ( mkTyConGenInfo )
+import CmdLineOpts     ( DynFlags )
 \end{code}
 
 
@@ -61,7 +60,7 @@ import Generics         ( mkTyConGenInfo )
 The main function
 ~~~~~~~~~~~~~~~~~
 \begin{code}
-tcTyAndClassDecls :: TcEnv             -- Knot tying stuff
+tcTyAndClassDecls :: RecTcEnv          -- Knot tying stuff
                  -> [RenamedHsDecl]
                  -> TcM TcEnv
 
@@ -75,7 +74,7 @@ tcGroups unf_env []
 
 tcGroups unf_env (group:groups)
   = tcGroup unf_env group      `thenTc` \ env ->
-    tcSetEnv env                       $
+    tcSetEnv env               $
     tcGroups unf_env groups
 \end{code}
 
@@ -111,9 +110,10 @@ The knot-tying parameters: @rec_details_list@ is an alist mapping @Name@s to
 @TyThing@s.  @rec_vrcs@ is a finite map from @Name@s to @ArgVrcs@s.
 
 \begin{code}
-tcGroup :: TcEnv -> SCC RenamedTyClDecl -> TcM TcEnv
+tcGroup :: RecTcEnv -> SCC RenamedTyClDecl -> TcM TcEnv
 tcGroup unf_env scc
-  =    -- Step 1
+  = getDOptsTc                                                 `thenTc` \ dflags ->
+       -- Step 1
     mapNF_Tc getInitialKind decls                              `thenNF_Tc` \ initial_kinds ->
 
        -- Step 2
@@ -130,7 +130,8 @@ tcGroup unf_env scc
            rec_details = mkNameEnv rec_details_list
 
            tyclss, all_tyclss :: [(Name, TyThing)]
-           tyclss = map (buildTyConOrClass is_rec kind_env rec_vrcs rec_details) decls
+           tyclss = map (buildTyConOrClass dflags is_rec kind_env 
+                                                  rec_vrcs rec_details) decls
 
                -- Add the tycons that come from the classes
                -- We want them in the environment because 
@@ -180,11 +181,11 @@ getInitialKind (TySynonym name tyvars _ _)
    newKindVar          `thenNF_Tc` \ result_kind  ->
    returnNF_Tc (name, mk_kind arg_kinds result_kind)
 
-getInitialKind (TyData _ _ name tyvars _ _ _ _ _ _ _)
+getInitialKind (TyData _ _ name tyvars _ _ _ _ _ _)
  = kcHsTyVars tyvars   `thenNF_Tc` \ arg_kinds ->
    returnNF_Tc (name, mk_kind arg_kinds boxedTypeKind)
 
-getInitialKind (ClassDecl _ name tyvars _ _ _ _ _ _ )
+getInitialKind (ClassDecl _ name tyvars _ _ _ _ _ )
  = kcHsTyVars tyvars   `thenNF_Tc` \ arg_kinds ->
    returnNF_Tc (name, mk_kind arg_kinds boxedTypeKind)
 
@@ -220,7 +221,7 @@ kcTyClDecl decl@(TySynonym tycon_name hs_tyvars rhs loc)
     kcHsType rhs                       `thenTc` \ rhs_kind ->
     unifyKind result_kind rhs_kind
 
-kcTyClDecl decl@(TyData _ context tycon_name hs_tyvars con_decls _ _ _ loc _ _)
+kcTyClDecl decl@(TyData _ context tycon_name hs_tyvars con_decls _ _ loc _ _)
   = tcAddDeclCtxt decl                 $
     kcTyClDeclBody tycon_name hs_tyvars        $ \ result_kind ->
     kcHsContext context                        `thenTc_` 
@@ -234,7 +235,7 @@ kcTyClDecl decl@(TyData _ context tycon_name hs_tyvars con_decls _ _ _ loc _ _)
 
 kcTyClDecl decl@(ClassDecl context class_name
                           hs_tyvars fundeps class_sigs
-                          _ _ _ loc)
+                          _ _ loc)
   = tcAddDeclCtxt decl                 $
     kcTyClDeclBody class_name hs_tyvars        $ \ result_kind ->
     kcHsContext context                        `thenTc_`
@@ -249,11 +250,12 @@ kcTyClDeclBody :: Name -> [HsTyVarBndr Name]      -- Kind of the tycon/cls and its t
 -- the kind of the tycon/class.  Give it to the thing inside, and 
 -- check the result kind matches
 kcTyClDeclBody tc_name hs_tyvars thing_inside
-  = tcLookupGlobal tc_name             `thenNF_Tc` \ thing ->
+  = tcLookup tc_name           `thenNF_Tc` \ thing ->
     let
        kind = case thing of
-                 ATyCon tc -> tyConKind tc
-                 AClass cl -> tyConKind (classTyCon cl)
+                 AGlobal (ATyCon tc) -> tyConKind tc
+                 AGlobal (AClass cl) -> tyConKind (classTyCon cl)
+                 AThing kind         -> kind
                -- For some odd reason, a class doesn't include its kind
 
        (tyvars_w_kinds, result_kind) = zipFunTys (hsTyVarNames hs_tyvars) kind
@@ -270,13 +272,14 @@ kcTyClDeclBody tc_name hs_tyvars thing_inside
 
 \begin{code}
 buildTyConOrClass 
-       :: RecFlag -> NameEnv Kind
+       :: DynFlags
+       -> RecFlag -> NameEnv Kind
        -> FiniteMap TyCon ArgVrcs -> NameEnv TyThingDetails
        -> RenamedTyClDecl -> (Name, TyThing)
        -- Can't fail; the only reason it's in the monad 
        -- is so it can zonk the kinds
 
-buildTyConOrClass is_rec kenv rec_vrcs rec_details
+buildTyConOrClass dflags is_rec kenv rec_vrcs rec_details
                  (TySynonym tycon_name tyvar_names rhs src_loc)
   = (tycon_name, ATyCon tycon)
   where
@@ -287,15 +290,15 @@ buildTyConOrClass is_rec kenv rec_vrcs rec_details
        SynTyDetails rhs_ty = lookupNameEnv_NF rec_details tycon_name
         argvrcs                    = lookupWithDefaultFM rec_vrcs bogusVrcs tycon
 
-buildTyConOrClass is_rec kenv rec_vrcs  rec_details
-                 (TyData data_or_new context tycon_name tyvar_names _ nconstrs _ _ src_loc name1 name2)
+buildTyConOrClass dflags is_rec kenv rec_vrcs  rec_details
+                 (TyData data_or_new context tycon_name tyvar_names _ nconstrs _ src_loc name1 name2)
   = (tycon_name, ATyCon tycon)
   where
        tycon = mkAlgTyConRep tycon_name tycon_kind tyvars ctxt argvrcs
                           data_cons nconstrs
                           derived_classes
                           flavour is_rec gen_info
-       gen_info = mkTyConGenInfo tycon name1 name2
+       gen_info = mkTyConGenInfo dflags tycon name1 name2
 
        DataTyDetails ctxt data_cons derived_classes = lookupNameEnv_NF rec_details tycon_name
 
@@ -308,13 +311,13 @@ buildTyConOrClass is_rec kenv rec_vrcs  rec_details
                        DataType | all isNullaryDataCon data_cons -> EnumTyCon
                                 | otherwise                      -> DataTyCon
 
-buildTyConOrClass is_rec kenv rec_vrcs  rec_details
+buildTyConOrClass dflags is_rec kenv rec_vrcs  rec_details
                   (ClassDecl context class_name
-                            tyvar_names fundeps class_sigs def_methods pragmas
+                            tyvar_names fundeps class_sigs def_methods
                             name_list src_loc)
   = (class_name, AClass clas)
   where
-        (tycon_name, _, _, _) = fromClassDeclNameList name_list
+        (tycon_name, _, _, _) = getClassDeclSysNames name_list
        clas = mkClass class_name tyvars fds
                       sc_theta sc_sel_ids op_items
                       tycon
@@ -358,7 +361,7 @@ Dependency analysis
 sortByDependency :: [RenamedHsDecl] -> TcM [SCC RenamedTyClDecl]
 sortByDependency decls
   = let                -- CHECK FOR CLASS CYCLES
-       cls_sccs   = stronglyConnComp (mapMaybe mk_cls_edges tycl_decls)
+       cls_sccs   = stronglyConnComp (mapMaybe mkClassEdges tycl_decls)
        cls_cycles = [ decls | CyclicSCC decls <- cls_sccs]
     in
     checkTc (null cls_cycles) (classCycleErr cls_cycles)       `thenTc_`
@@ -376,8 +379,8 @@ sortByDependency decls
     in
     returnTc decl_sccs
   where
-    tycl_decls = [d | TyClD d <- decls]
-    edges      = map mk_edges tycl_decls
+    tycl_decls = [d | TyClD d <- decls, not (isIfaceSigDecl d)]
+    edges      = map mkEdges tycl_decls
     
     is_syn_decl (d, _, _) = isSynDecl d
 \end{code}
@@ -386,84 +389,25 @@ Edges in Type/Class decls
 ~~~~~~~~~~~~~~~~~~~~~~~~~
 
 \begin{code}
+tyClDeclFTVs :: RenamedTyClDecl -> [Name]
+tyClDeclFTVs d = foldNameSet add [] (tyClDeclFVs d)
+              where
+                add n fvs | isTyVarName n = fvs
+                          | otherwise     = n : fvs
+
 ----------------------------------------------------
 -- mk_cls_edges looks only at the context of class decls
 -- Its used when we are figuring out if there's a cycle in the
 -- superclass hierarchy
 
-mk_cls_edges :: RenamedTyClDecl -> Maybe (RenamedTyClDecl, Unique, [Unique])
-
-mk_cls_edges decl@(ClassDecl ctxt name _ _ _ _ _ _ _)
-  = Just (decl, getUnique name, map getUnique (catMaybes (map get_clas ctxt)))
-mk_cls_edges other_decl
-  = Nothing
-
-----------------------------------------------------
-mk_edges :: RenamedTyClDecl -> (RenamedTyClDecl, Unique, [Unique])
-
-mk_edges decl@(TyData _ ctxt name _ condecls _ derivs _ _ _ _)
-  = (decl, getUnique name, uniqSetToList (get_ctxt ctxt `unionUniqSets`
-                                        get_cons condecls `unionUniqSets`
-                                        get_deriv derivs))
-
-mk_edges decl@(TySynonym name _ rhs _)
-  = (decl, getUnique name, uniqSetToList (get_ty rhs))
-
-mk_edges decl@(ClassDecl ctxt name _ _ sigs _ _ _ _)
-  = (decl, getUnique name, uniqSetToList (get_ctxt ctxt `unionUniqSets`
-                                        get_sigs sigs))
-
-
-----------------------------------------------------
-get_ctxt ctxt = unionManyUniqSets (map set_name (catMaybes (map get_clas ctxt)))
-get_clas (HsPClass clas _) = Just clas
-get_clas _                 = Nothing
-
-----------------------------------------------------
-get_deriv Nothing     = emptyUniqSet
-get_deriv (Just clss) = unionManyUniqSets (map set_name clss)
+mkClassEdges :: RenamedTyClDecl -> Maybe (RenamedTyClDecl, Name, [Name])
 
-----------------------------------------------------
-get_cons cons = unionManyUniqSets (map get_con cons)
-
-----------------------------------------------------
-get_con (ConDecl _ _ _ ctxt details _) 
-  = get_ctxt ctxt `unionUniqSets` get_con_details details
-
-----------------------------------------------------
-get_con_details (VanillaCon btys)    = unionManyUniqSets (map get_bty btys)
-get_con_details (InfixCon bty1 bty2) = unionUniqSets (get_bty bty1) (get_bty bty2)
-get_con_details (RecCon nbtys)       = unionManyUniqSets (map (get_bty.snd) nbtys)
-
-----------------------------------------------------
-get_bty bty = get_ty (getBangType bty)
-
-----------------------------------------------------
-get_ty (HsTyVar name) | isTvOcc (nameOccName name) = emptyUniqSet 
-                     | otherwise                  = set_name name
-get_ty (HsAppTy ty1 ty2)             = unionUniqSets (get_ty ty1) (get_ty ty2)
-get_ty (HsFunTy ty1 ty2)             = unionUniqSets (get_ty ty1) (get_ty ty2)
-get_ty (HsListTy ty)                 = set_name listTyCon_name `unionUniqSets` get_ty ty
-get_ty (HsTupleTy (HsTupCon n _) tys) = set_name n `unionUniqSets` get_tys tys
-get_ty (HsUsgTy _ ty)                = get_ty ty
-get_ty (HsUsgForAllTy _ ty)          = get_ty ty
-get_ty (HsForAllTy _ ctxt mty)               = get_ctxt ctxt `unionUniqSets` get_ty mty
-get_ty (HsPredTy (HsPClass name _))   = set_name name
-get_ty (HsPredTy (HsPIParam _ _))     = emptyUniqSet   -- I think
-
-----------------------------------------------------
-get_tys tys = unionManyUniqSets (map get_ty tys)
-
-----------------------------------------------------
-get_sigs sigs
-  = unionManyUniqSets (map get_sig sigs)
-  where 
-    get_sig (ClassOpSig _ _ ty _) = get_ty ty
-    get_sig (FixSig _)           = emptyUniqSet
-    get_sig other = panic "TcTyClsDecls:get_sig"
+mkClassEdges decl@(ClassDecl ctxt name _ _ _ _ _ _) = Just (decl, name, [c | HsPClass c _ <- ctxt])
+mkClassEdges other_decl                                    = Nothing
 
 ----------------------------------------------------
-set_name name = unitUniqSet (getUnique name)
+mkEdges :: RenamedTyClDecl -> (RenamedTyClDecl, Name, [Name])
+mkEdges decl = (decl, tyClDeclName decl, tyClDeclFTVs decl)
 \end{code}