X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Ftypecheck%2FTcEnv.lhs;h=809abce82d5344e73ae46488eab5248496aa9d62;hb=788faebb40b51d37e73ed94dfc99460d39a1a811;hp=953d7fffab52496884605e1bc0bfdc946cc4fae4;hpb=525898a970c625753c33490318762c2b4c2770a9;p=ghc-hetmet.git diff --git a/ghc/compiler/typecheck/TcEnv.lhs b/ghc/compiler/typecheck/TcEnv.lhs index 953d7ff..809abce 100644 --- a/ghc/compiler/typecheck/TcEnv.lhs +++ b/ghc/compiler/typecheck/TcEnv.lhs @@ -1,229 +1,427 @@ \begin{code} module TcEnv( - TcId, TcIdSet, tcInstId, - tcLookupDataCon, + TcId, TcIdSet, + TyThing(..), TyThingDetails(..), TcTyThing(..), - TcEnv, ValueEnv, TcTyThing(..), - - initEnv, getEnvTyCons, getEnvClasses, getEnvAllTyCons, + -- Getting stuff from the environment + TcEnv, initTcEnv, + tcEnvTyCons, tcEnvClasses, tcEnvIds, tcEnvTcIds, tcEnvTyVars, + getTcGEnv, - tcExtendUVarEnv, tcLookupUVar, - - tcExtendTyVarEnv, tcExtendTyVarEnvForMeths, tcExtendTypeEnv, tcGetInScopeTyVars, - - tcLookupTy, - tcLookupTyConByKey, - tcLookupClassByKey, tcLookupClassByKey_maybe, - - tcExtendGlobalValEnv, tcExtendLocalValEnv, - tcGetValueEnv, tcSetValueEnv, - tcAddImportedIdInfo, + -- Instance environment, and InstInfo type + tcGetInstEnv, tcSetInstEnv, + InstInfo(..), pprInstInfo, + simpleInstInfoTy, simpleInstInfoTyCon, + + -- Global environment + tcExtendGlobalEnv, tcExtendGlobalValEnv, tcExtendGlobalTypeEnv, + tcLookupTyCon, tcLookupClass, tcLookupGlobalId, tcLookupDataCon, + tcLookupGlobal_maybe, tcLookupGlobal, tcLookupSyntaxId, tcLookupSyntaxName, + + -- Local environment + tcExtendKindEnv, tcLookupLocalIds, + tcExtendTyVarEnv, tcExtendTyVarEnvForMeths, + tcExtendLocalValEnv, tcLookup, tcLookup_maybe, tcLookupId, + + -- Global type variables + tcGetGlobalTyVars, tcExtendGlobalTyVars, - tcLookupValue, tcLookupValueMaybe, - tcLookupValueByKey, tcLookupValueByKeyMaybe, - explicitLookupValueByKey, explicitLookupValue, - valueEnvIds, + -- Random useful things + RecTcEnv, tcAddImportedIdInfo, tcLookupRecId, tcLookupRecId_maybe, + -- New Ids newLocalId, newSpecPragmaId, - tcGetGlobalTyVars, tcExtendGlobalTyVars, + newDFunName, - badCon, badPrimOp + -- Misc + isLocalThing, tcSetEnv ) where #include "HsVersions.h" -import HsTypes ( HsTyVarBndr, getTyVarName ) -import Id ( mkUserLocal, isDataConWrapId_maybe ) -import MkId ( mkSpecPragmaId ) -import Var ( TyVar, Id, setVarName, - idType, lazySetIdInfo, idInfo, tyVarKind, UVar, - ) -import TcType ( TcType, TcTyVar, TcTyVarSet, TcThetaType, - tcInstTyVars, zonkTcTyVars, - TcKind, kindToTcKind - ) -import VarEnv -import VarSet -import Type ( Kind, superKind, - tyVarsOfType, tyVarsOfTypes, mkTyVarTy, - splitForAllTys, splitRhoTy, splitFunTys, - splitAlgTyConApp_maybe, getTyVar - ) -import Subst ( substTy ) -import UsageSPUtils ( unannotTy ) -import DataCon ( DataCon ) -import TyCon ( TyCon, tyConKind, tyConArity, isSynTyCon ) -import Class ( Class, classTyCon ) - +import RnHsSyn ( RenamedMonoBinds, RenamedSig ) import TcMonad - -import BasicTypes ( Arity ) +import TcType ( TcKind, TcType, TcTyVar, TcTyVarSet, + zonkTcTyVarsAndFV + ) +import Id ( idName, mkSpecPragmaId, mkUserLocal, isDataConWrapId_maybe ) import IdInfo ( vanillaIdInfo ) -import Name ( Name, OccName, nameOccName, getSrcLoc, - maybeWiredInTyConName, maybeWiredInIdName, isLocallyDefined, - NamedThing(..), - NameEnv, emptyNameEnv, addToNameEnv, - extendNameEnv, lookupNameEnv, nameEnvElts +import Var ( TyVar, Id, idType, lazySetIdInfo, idInfo ) +import VarSet +import Type ( Type, ThetaType, + tyVarsOfTypes, splitDFunTy, + getDFunTyKey, tyConAppTyCon ) -import Unique ( pprUnique10, Unique, Uniquable(..) ) -import FiniteMap ( lookupFM, addToFM ) -import UniqFM -import Unique ( Uniquable(..) ) -import Util ( zipEqual, zipWith3Equal, mapAccumL ) -import Bag ( bagToList ) -import Maybes ( maybeToBool, catMaybes ) +import DataCon ( DataCon ) +import TyCon ( TyCon ) +import Class ( Class, ClassOpItem ) +import Name ( Name, OccName, NamedThing(..), + nameOccName, getSrcLoc, mkLocalName, isLocalName, + nameIsLocalOrFrom + ) +import NameEnv ( NameEnv, lookupNameEnv, nameEnvElts, + extendNameEnvList, emptyNameEnv, plusNameEnv ) +import OccName ( mkDFunOcc, occNameString ) +import HscTypes ( DFunId, + PackageTypeEnv, TypeEnv, + extendTypeEnvList, extendTypeEnvWithIds, + typeEnvTyCons, typeEnvClasses, typeEnvIds, + HomeSymbolTable + ) +import Module ( Module ) +import InstEnv ( InstEnv, emptyInstEnv ) +import HscTypes ( lookupType, TyThing(..) ) +import Util ( zipEqual ) import SrcLoc ( SrcLoc ) -import FastString ( FastString ) +import qualified PrelNames import Outputable + +import IOExts ( newIORef ) \end{code} %************************************************************************ %* * -\subsection{TcId} +\subsection{TcEnv} %* * %************************************************************************ - \begin{code} type TcId = Id -- Type may be a TcType type TcIdSet = IdSet -tcLookupDataCon :: Name -> TcM s (DataCon, [TcType], TcType) -tcLookupDataCon con_name - = tcLookupValue con_name `thenNF_Tc` \ con_id -> - case isDataConWrapId_maybe con_id of { - Nothing -> failWithTc (badCon con_id); - Just data_con -> - - tcInstId con_id `thenNF_Tc` \ (_, _, con_tau) -> - -- Ignore the con_theta; overloaded constructors only - -- behave differently when called, not when used for - -- matching. - let - (arg_tys, result_ty) = splitFunTys con_tau - in - ASSERT( maybeToBool (splitAlgTyConApp_maybe result_ty) ) - returnTc (data_con, arg_tys, result_ty) } - --- A useful function that takes an occurrence of a global thing --- and instantiates its type with fresh type variables -tcInstId :: Id - -> NF_TcM s ([TcTyVar], -- It's instantiated type - TcThetaType, -- - TcType) -- -tcInstId id - = let - (tyvars, rho) = splitForAllTys (unannotTy (idType id)) - in - tcInstTyVars tyvars `thenNF_Tc` \ (tyvars', arg_tys, tenv) -> - let - rho' = substTy tenv rho - (theta', tau') = splitRhoTy rho' - in - returnNF_Tc (tyvars', theta', tau') +data TcEnv + = TcEnv { + tcSyntaxMap :: PrelNames.SyntaxMap, -- The syntax map (usually the identity) + + tcGST :: Name -> Maybe TyThing, -- The type environment at the moment we began this compilation + + tcInsts :: InstEnv, -- All instances (both imported and in this module) + + tcGEnv :: TypeEnv, -- The global type environment we've accumulated while + {- NameEnv TyThing-} -- compiling this module: + -- types and classes (both imported and local) + -- imported Ids + -- (Ids defined in this module start in the local envt, + -- though they move to the global envt during zonking) + + tcLEnv :: NameEnv TcTyThing, -- The local type environment: Ids and TyVars + -- defined in this module + + tcTyVars :: TcRef TcTyVarSet -- The "global tyvars" + -- Namely, the in-scope TyVars bound in tcLEnv, plus the tyvars + -- mentioned in the types of Ids bound in tcLEnv + -- Why mutable? see notes with tcGetGlobalTyVars + } + \end{code} -Between the renamer and the first invocation of the UsageSP inference, -identifiers read from interface files will have usage information in -their types, whereas other identifiers will not. The unannotTy here -in @tcInstId@ prevents this information from pointlessly propagating -further prior to the first usage inference. +The Global-Env/Local-Env story +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ +During type checking, we keep in the GlobalEnv + * All types and classes + * All Ids derived from types and classes (constructors, selectors) + * Imported Ids + +At the end of type checking, we zonk the local bindings, +and as we do so we add to the GlobalEnv + * Locally defined top-level Ids + +Why? Because they are now Ids not TcIds. This final GlobalEnv is +used thus: + a) fed back (via the knot) to typechecking the + unfoldings of interface signatures + + b) used to augment the GlobalSymbolTable + + +\begin{code} +data TcTyThing + = AGlobal TyThing -- Used only in the return type of a lookup + | ATcId TcId -- Ids defined in this module + | ATyVar TyVar -- Type variables + | AThing TcKind -- Used temporarily, during kind checking +-- Here's an example of how the AThing guy is used +-- Suppose we are checking (forall a. T a Int): +-- 1. We first bind (a -> AThink kv), where kv is a kind variable. +-- 2. Then we kind-check the (T a Int) part. +-- 3. Then we zonk the kind variable. +-- 4. Now we know the kind for 'a', and we add (a -> ATyVar a::K) to the environment + +initTcEnv :: PrelNames.SyntaxMap -> HomeSymbolTable -> PackageTypeEnv -> IO TcEnv +initTcEnv syntax_map hst pte + = do { gtv_var <- newIORef emptyVarSet ; + return (TcEnv { tcSyntaxMap = syntax_map, + tcGST = lookup, + tcGEnv = emptyNameEnv, + tcInsts = emptyInstEnv, + tcLEnv = emptyNameEnv, + tcTyVars = gtv_var + })} + where + lookup name | isLocalName name = Nothing + | otherwise = lookupType hst pte name + + +tcEnvClasses env = typeEnvClasses (tcGEnv env) +tcEnvTyCons env = typeEnvTyCons (tcGEnv env) +tcEnvIds env = typeEnvIds (tcGEnv env) +tcEnvTyVars env = [tv | ATyVar tv <- nameEnvElts (tcLEnv env)] +tcEnvTcIds env = [id | ATcId id <- nameEnvElts (tcLEnv env)] + +getTcGEnv (TcEnv { tcGEnv = genv }) = genv + +-- This data type is used to help tie the knot +-- when type checking type and class declarations +data TyThingDetails = SynTyDetails Type + | DataTyDetails ThetaType [DataCon] [Id] + | ClassDetails ThetaType [Id] [ClassOpItem] DataCon +\end{code} %************************************************************************ %* * -\subsection{TcEnv} +\subsection{Basic lookups} %* * %************************************************************************ -Data type declarations -~~~~~~~~~~~~~~~~~~~~~ +\begin{code} +lookup_global :: TcEnv -> Name -> Maybe TyThing + -- Try the global envt and then the global symbol table +lookup_global env name + = case lookupNameEnv (tcGEnv env) name of + Just thing -> Just thing + Nothing -> tcGST env name + +lookup_local :: TcEnv -> Name -> Maybe TcTyThing + -- Try the local envt and then try the global +lookup_local env name + = case lookupNameEnv (tcLEnv env) name of + Just thing -> Just thing + Nothing -> case lookup_global env name of + Just thing -> Just (AGlobal thing) + Nothing -> Nothing +\end{code} \begin{code} -data TcEnv = TcEnv - UsageEnv - TypeEnv - ValueEnv - (TcTyVarSet, -- The in-scope TyVars - TcRef TcTyVarSet) -- Free type variables of the value env - -- ...why mutable? see notes with tcGetGlobalTyVars - -- Includes the in-scope tyvars - -type UsageEnv = NameEnv UVar -type TypeEnv = NameEnv (TcKind, TcTyThing) -type ValueEnv = NameEnv Id - -valueEnvIds :: ValueEnv -> [Id] -valueEnvIds ve = nameEnvElts ve - -data TcTyThing = ATyVar TcTyVar -- Mutable only so that the kind can be mutable - -- if the kind is mutable, the tyvar must be so that - -- zonking works - | ADataTyCon TyCon - | ASynTyCon TyCon Arity - | AClass Class Arity - - -initEnv :: TcRef TcTyVarSet -> TcEnv -initEnv mut = TcEnv emptyNameEnv emptyNameEnv emptyNameEnv (emptyVarSet, mut) - -getEnvClasses (TcEnv _ te _ _) = [cl | (_, AClass cl _) <- nameEnvElts te] - -getEnvTyCons (TcEnv _ te _ _) = catMaybes (map get_tc (nameEnvElts te)) - where - get_tc (_, ADataTyCon tc) = Just tc - get_tc (_, ASynTyCon tc _) = Just tc - get_tc other = Nothing - -getEnvAllTyCons te_list = catMaybes (map get_tc te_list) - -- The 'all' means 'including the tycons from class decls' - where - get_tc (_, ADataTyCon tc) = Just tc - get_tc (_, ASynTyCon tc _) = Just tc - get_tc (_, AClass cl _) = Just (classTyCon cl) - get_tc other = Nothing +type RecTcEnv = TcEnv +-- This environment is used for getting the 'right' IdInfo +-- on imported things and for looking up Ids in unfoldings +-- The environment doesn't have any local Ids in it + +tcAddImportedIdInfo :: RecTcEnv -> Id -> Id +tcAddImportedIdInfo env id + = id `lazySetIdInfo` new_info + -- The Id must be returned without a data dependency on maybe_id + where + new_info = case tcLookupRecId_maybe env (idName id) of + Nothing -> pprTrace "tcAddIdInfo" (ppr id) vanillaIdInfo + Just imported_id -> idInfo imported_id + -- ToDo: could check that types are the same + +tcLookupRecId_maybe :: RecTcEnv -> Name -> Maybe Id +tcLookupRecId_maybe env name = case lookup_global env name of + Just (AnId id) -> Just id + other -> Nothing + +tcLookupRecId :: RecTcEnv -> Name -> Id +tcLookupRecId env name = case lookup_global env name of + Just (AnId id) -> id + Nothing -> pprPanic "tcLookupRecId" (ppr name) \end{code} -The UsageEnv -~~~~~~~~~~~~ +%************************************************************************ +%* * +\subsection{Making new Ids} +%* * +%************************************************************************ -Extending the usage environment. +Constructing new Ids \begin{code} -tcExtendUVarEnv :: Name -> UVar -> TcM s r -> TcM s r -tcExtendUVarEnv uv_name uv scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - tcSetEnv (TcEnv (addToNameEnv ue uv_name uv) te ve gtvs) scope +newLocalId :: OccName -> TcType -> SrcLoc -> NF_TcM TcId +newLocalId name ty loc + = tcGetUnique `thenNF_Tc` \ uniq -> + returnNF_Tc (mkUserLocal name uniq ty loc) + +newSpecPragmaId :: Name -> TcType -> NF_TcM TcId +newSpecPragmaId name ty + = tcGetUnique `thenNF_Tc` \ uniq -> + returnNF_Tc (mkSpecPragmaId (nameOccName name) uniq ty (getSrcLoc name)) \end{code} -Looking up in the environments. +Make a name for the dict fun for an instance decl. +It's a *local* name for the moment. The CoreTidy pass +will globalise it. \begin{code} -tcLookupUVar :: Name -> NF_TcM s UVar -tcLookupUVar uv_name - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - case lookupNameEnv ue uv_name of - Just uv -> returnNF_Tc uv - Nothing -> failWithTc (uvNameOutOfScope uv_name) -\end{code} +newDFunName :: Class -> [Type] -> SrcLoc -> NF_TcM Name +newDFunName clas (ty:_) loc + = tcGetUnique `thenNF_Tc` \ uniq -> + returnNF_Tc (mkLocalName uniq (mkDFunOcc dfun_string) loc) + where + -- Any string that is somewhat unique will do + dfun_string = occNameString (getOccName clas) ++ occNameString (getDFunTyKey ty) +newDFunName clas [] loc = pprPanic "newDFunName" (ppr clas <+> ppr loc) +\end{code} -The TypeEnv -~~~~~~~~~~~~ +\begin{code} +isLocalThing :: NamedThing a => Module -> a -> Bool +isLocalThing mod thing = nameIsLocalOrFrom mod (getName thing) +\end{code} -Extending the type environment. +%************************************************************************ +%* * +\subsection{The global environment} +%* * +%************************************************************************ \begin{code} -tcExtendTyVarEnv :: [TyVar] -> TcM s r -> TcM s r -tcExtendTyVarEnv tyvars scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve (in_scope_tvs, gtvs)) -> +tcExtendGlobalEnv :: [TyThing] -> TcM r -> TcM r +tcExtendGlobalEnv things thing_inside + = tcGetEnv `thenNF_Tc` \ env -> + let + ge' = extendTypeEnvList (tcGEnv env) things + in + tcSetEnv (env {tcGEnv = ge'}) thing_inside + + +tcExtendGlobalTypeEnv :: TypeEnv -> TcM r -> TcM r +tcExtendGlobalTypeEnv extra_env thing_inside + = tcGetEnv `thenNF_Tc` \ env -> let - extend_list = [ (getName tv, (kindToTcKind (tyVarKind tv), ATyVar tv)) - | tv <- tyvars - ] - te' = extendNameEnv te extend_list - new_tv_set = mkVarSet tyvars - in_scope_tvs' = in_scope_tvs `unionVarSet` new_tv_set + ge' = tcGEnv env `plusNameEnv` extra_env + in + tcSetEnv (env {tcGEnv = ge'}) thing_inside + +tcExtendGlobalValEnv :: [Id] -> TcM a -> TcM a +tcExtendGlobalValEnv ids thing_inside + = tcGetEnv `thenNF_Tc` \ env -> + let + ge' = extendTypeEnvWithIds (tcGEnv env) ids + in + tcSetEnv (env {tcGEnv = ge'}) thing_inside +\end{code} + + +\begin{code} +tcLookupGlobal_maybe :: Name -> NF_TcM (Maybe TyThing) +tcLookupGlobal_maybe name + = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (lookup_global env name) +\end{code} + +A variety of global lookups, when we know what we are looking for. + +\begin{code} +tcLookupGlobal :: Name -> NF_TcM TyThing +tcLookupGlobal name + = tcLookupGlobal_maybe name `thenNF_Tc` \ maybe_thing -> + case maybe_thing of + Just thing -> returnNF_Tc thing + other -> notFound "tcLookupGlobal" name + +tcLookupGlobalId :: Name -> NF_TcM Id +tcLookupGlobalId name + = tcLookupGlobal_maybe name `thenNF_Tc` \ maybe_id -> + case maybe_id of + Just (AnId id) -> returnNF_Tc id + other -> notFound "tcLookupGlobalId" name + +tcLookupDataCon :: Name -> TcM DataCon +tcLookupDataCon con_name + = tcLookupGlobalId con_name `thenNF_Tc` \ con_id -> + case isDataConWrapId_maybe con_id of + Just data_con -> returnTc data_con + Nothing -> failWithTc (badCon con_id) + + +tcLookupClass :: Name -> NF_TcM Class +tcLookupClass name + = tcLookupGlobal_maybe name `thenNF_Tc` \ maybe_clas -> + case maybe_clas of + Just (AClass clas) -> returnNF_Tc clas + other -> notFound "tcLookupClass" name + +tcLookupTyCon :: Name -> NF_TcM TyCon +tcLookupTyCon name + = tcLookupGlobal_maybe name `thenNF_Tc` \ maybe_tc -> + case maybe_tc of + Just (ATyCon tc) -> returnNF_Tc tc + other -> notFound "tcLookupTyCon" name + +tcLookupId :: Name -> NF_TcM Id +tcLookupId name + = tcLookup name `thenNF_Tc` \ thing -> + case thing of + ATcId tc_id -> returnNF_Tc tc_id + AGlobal (AnId id) -> returnNF_Tc id + other -> pprPanic "tcLookupId" (ppr name) + +tcLookupLocalIds :: [Name] -> NF_TcM [TcId] +tcLookupLocalIds ns + = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (map (lookup (tcLEnv env)) ns) + where + lookup lenv name = case lookupNameEnv lenv name of + Just (ATcId id) -> id + other -> pprPanic "tcLookupLocalIds" (ppr name) + +tcLookupSyntaxId :: Name -> NF_TcM Id +-- Lookup a name like PrelNum.fromInt, and return the corresponding Id, +-- after mapping through the SyntaxMap. This may give us the Id for +-- (say) MyPrelude.fromInteger +tcLookupSyntaxId name + = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (case lookup_global env (tcSyntaxMap env name) of + Just (AnId id) -> id + other -> pprPanic "tcLookupSyntaxId" (ppr name)) + +tcLookupSyntaxName :: Name -> NF_TcM Name +tcLookupSyntaxName name + = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (tcSyntaxMap env name) +\end{code} + + +%************************************************************************ +%* * +\subsection{The local environment} +%* * +%************************************************************************ + +\begin{code} +tcLookup_maybe :: Name -> NF_TcM (Maybe TcTyThing) +tcLookup_maybe name + = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (lookup_local env name) + +tcLookup :: Name -> NF_TcM TcTyThing +tcLookup name + = tcLookup_maybe name `thenNF_Tc` \ maybe_thing -> + case maybe_thing of + Just thing -> returnNF_Tc thing + other -> notFound "tcLookup" name + -- Extract the IdInfo from an IfaceSig imported from an interface file +\end{code} + + +\begin{code} +tcExtendKindEnv :: [(Name,TcKind)] -> TcM r -> TcM r +tcExtendKindEnv pairs thing_inside + = tcGetEnv `thenNF_Tc` \ env -> + let + le' = extendNameEnvList (tcLEnv env) [(n, AThing k) | (n,k) <- pairs] + -- No need to extend global tyvars for kind checking + in + tcSetEnv (env {tcLEnv = le'}) thing_inside + +tcExtendTyVarEnv :: [TyVar] -> TcM r -> TcM r +tcExtendTyVarEnv tyvars thing_inside + = tcGetEnv `thenNF_Tc` \ env@(TcEnv {tcLEnv = le, tcTyVars = gtvs}) -> + let + le' = extendNameEnvList le [ (getName tv, ATyVar tv) | tv <- tyvars] + new_tv_set = mkVarSet tyvars in -- It's important to add the in-scope tyvars to the global tyvar set -- as well. Consider @@ -232,7 +430,7 @@ tcExtendTyVarEnv tyvars scope -- class and instance decls, when we mustn't generalise the class tyvars -- when typechecking the methods. tc_extend_gtvs gtvs new_tv_set `thenNF_Tc` \ gtvs' -> - tcSetEnv (TcEnv ue te' ve (in_scope_tvs', gtvs')) scope + tcSetEnv (env {tcLEnv = le', tcTyVars = gtvs'}) thing_inside -- This variant, tcExtendTyVarEnvForMeths, takes *two* bunches of tyvars: -- the signature tyvars contain the original names @@ -240,226 +438,112 @@ tcExtendTyVarEnv tyvars scope -- It's needed when typechecking the method bindings of class and instance decls -- It does *not* extend the global tyvars; tcMethodBind does that for itself -tcExtendTyVarEnvForMeths :: [TyVar] -> [TcTyVar] -> TcM s r -> TcM s r +tcExtendTyVarEnvForMeths :: [TyVar] -> [TcTyVar] -> TcM r -> TcM r tcExtendTyVarEnvForMeths sig_tyvars inst_tyvars thing_inside - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> + = tcGetEnv `thenNF_Tc` \ env -> let - te' = extendNameEnv te stuff + le' = extendNameEnvList (tcLEnv env) stuff + stuff = [ (getName sig_tv, ATyVar inst_tv) + | (sig_tv, inst_tv) <- zipEqual "tcMeth" sig_tyvars inst_tyvars + ] in - tcSetEnv (TcEnv ue te' ve gtvs) thing_inside - where - stuff = [ (getName sig_tv, (kindToTcKind (tyVarKind inst_tv), ATyVar inst_tv)) - | (sig_tv, inst_tv) <- zipEqual "tcMeth" sig_tyvars inst_tyvars - ] - -tcExtendGlobalTyVars extra_global_tvs scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve (in_scope,gtvs)) -> - tc_extend_gtvs gtvs extra_global_tvs `thenNF_Tc` \ gtvs' -> - tcSetEnv (TcEnv ue te ve (in_scope,gtvs')) scope - -tc_extend_gtvs gtvs extra_global_tvs - = tcReadMutVar gtvs `thenNF_Tc` \ global_tvs -> - let - new_global_tyvars = global_tvs `unionVarSet` extra_global_tvs - in - tcNewMutVar new_global_tyvars + tcSetEnv (env {tcLEnv = le'}) thing_inside \end{code} -@tcGetGlobalTyVars@ returns a fully-zonked set of tyvars free in the environment. -To improve subsequent calls to the same function it writes the zonked set back into -the environment. \begin{code} -tcGetGlobalTyVars :: NF_TcM s TcTyVarSet -tcGetGlobalTyVars - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve (_,gtvs)) -> - tcReadMutVar gtvs `thenNF_Tc` \ global_tvs -> - zonkTcTyVars (varSetElems global_tvs) `thenNF_Tc` \ global_tys' -> +tcExtendLocalValEnv :: [(Name,TcId)] -> TcM a -> TcM a +tcExtendLocalValEnv names_w_ids thing_inside + = tcGetEnv `thenNF_Tc` \ env -> let - global_tvs' = (tyVarsOfTypes global_tys') + extra_global_tyvars = tyVarsOfTypes [idType id | (name,id) <- names_w_ids] + extra_env = [(name, ATcId id) | (name,id) <- names_w_ids] + le' = extendNameEnvList (tcLEnv env) extra_env in - tcWriteMutVar gtvs global_tvs' `thenNF_Tc_` - returnNF_Tc global_tvs' - -tcGetInScopeTyVars :: NF_TcM s [TcTyVar] -tcGetInScopeTyVars - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve (in_scope_tvs, gtvs)) -> - returnNF_Tc (varSetElems in_scope_tvs) + tc_extend_gtvs (tcTyVars env) extra_global_tyvars `thenNF_Tc` \ gtvs' -> + tcSetEnv (env {tcLEnv = le', tcTyVars = gtvs'}) thing_inside \end{code} -Type constructors and classes +%************************************************************************ +%* * +\subsection{The global tyvars} +%* * +%************************************************************************ \begin{code} -tcExtendTypeEnv :: [(Name, (TcKind, TcTyThing))] -> TcM s r -> TcM s r -tcExtendTypeEnv bindings scope - = ASSERT( null [tv | (_, (_,ATyVar tv)) <- bindings] ) - -- Not for tyvars; use tcExtendTyVarEnv - tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - let - te' = extendNameEnv te bindings - in - tcSetEnv (TcEnv ue te' ve gtvs) scope -\end{code} +tcExtendGlobalTyVars extra_global_tvs thing_inside + = tcGetEnv `thenNF_Tc` \ env -> + tc_extend_gtvs (tcTyVars env) extra_global_tvs `thenNF_Tc` \ gtvs' -> + tcSetEnv (env {tcTyVars = gtvs'}) thing_inside +tc_extend_gtvs gtvs extra_global_tvs + = tcReadMutVar gtvs `thenNF_Tc` \ global_tvs -> + tcNewMutVar (global_tvs `unionVarSet` extra_global_tvs) +\end{code} -Looking up in the environments. +@tcGetGlobalTyVars@ returns a fully-zonked set of tyvars free in the environment. +To improve subsequent calls to the same function it writes the zonked set back into +the environment. \begin{code} -tcLookupTy :: Name -> NF_TcM s (TcKind, TcTyThing) -tcLookupTy name - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - case lookupNameEnv te name of { - Just thing -> returnNF_Tc thing ; - Nothing -> - - case maybeWiredInTyConName name of - Just tc | isSynTyCon tc -> returnNF_Tc (kindToTcKind (tyConKind tc), ASynTyCon tc (tyConArity tc)) - | otherwise -> returnNF_Tc (kindToTcKind (tyConKind tc), ADataTyCon tc) - - Nothing -> -- This can happen if an interface-file - -- unfolding is screwed up - failWithTc (tyNameOutOfScope name) - } - -tcLookupClassByKey :: Unique -> NF_TcM s Class -tcLookupClassByKey key - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - case lookupUFM_Directly te key of - Just (_, AClass cl _) -> returnNF_Tc cl - other -> pprPanic "tcLookupClassByKey:" (pprUnique10 key) - -tcLookupClassByKey_maybe :: Unique -> NF_TcM s (Maybe Class) -tcLookupClassByKey_maybe key - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - case lookupUFM_Directly te key of - Just (_, AClass cl _) -> returnNF_Tc (Just cl) - other -> returnNF_Tc Nothing - -tcLookupTyConByKey :: Unique -> NF_TcM s TyCon -tcLookupTyConByKey key - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - case lookupUFM_Directly te key of - Just (_, ADataTyCon tc) -> returnNF_Tc tc - Just (_, ASynTyCon tc _) -> returnNF_Tc tc - other -> pprPanic "tcLookupTyConByKey:" (pprUnique10 key) +tcGetGlobalTyVars :: NF_TcM TcTyVarSet +tcGetGlobalTyVars + = tcGetEnv `thenNF_Tc` \ (TcEnv {tcTyVars = gtv_var}) -> + tcReadMutVar gtv_var `thenNF_Tc` \ gbl_tvs -> + zonkTcTyVarsAndFV (varSetElems gbl_tvs) `thenNF_Tc` \ gbl_tvs' -> + tcWriteMutVar gtv_var gbl_tvs' `thenNF_Tc_` + returnNF_Tc gbl_tvs' \end{code} - - %************************************************************************ %* * -\subsection{The value environment} +\subsection{The instance environment} %* * %************************************************************************ \begin{code} -tcExtendGlobalValEnv :: [Id] -> TcM s a -> TcM s a -tcExtendGlobalValEnv ids scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - let - ve' = addListToUFM_Directly ve [(getUnique id, id) | id <- ids] - in - tcSetEnv (TcEnv ue te ve' gtvs) scope - -tcExtendLocalValEnv :: [(Name,TcId)] -> TcM s a -> TcM s a -tcExtendLocalValEnv names_w_ids scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve (in_scope_tvs,gtvs)) -> - tcReadMutVar gtvs `thenNF_Tc` \ global_tvs -> - let - ve' = extendNameEnv ve names_w_ids - extra_global_tyvars = tyVarsOfTypes (map (idType . snd) names_w_ids) - in - tc_extend_gtvs gtvs extra_global_tyvars `thenNF_Tc` \ gtvs' -> - tcSetEnv (TcEnv ue te ve' (in_scope_tvs,gtvs')) scope -\end{code} - +tcGetInstEnv :: NF_TcM InstEnv +tcGetInstEnv = tcGetEnv `thenNF_Tc` \ env -> + returnNF_Tc (tcInsts env) -\begin{code} -tcLookupValue :: Name -> NF_TcM s Id -- Panics if not found -tcLookupValue name - = case maybeWiredInIdName name of - Just id -> returnNF_Tc id - Nothing -> tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - returnNF_Tc (lookupWithDefaultUFM ve def name) - where - def = pprPanic "tcLookupValue:" (ppr name) - -tcLookupValueMaybe :: Name -> NF_TcM s (Maybe Id) -tcLookupValueMaybe name - = case maybeWiredInIdName name of - Just id -> returnNF_Tc (Just id) - Nothing -> tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - returnNF_Tc (lookupNameEnv ve name) - -tcLookupValueByKey :: Unique -> NF_TcM s Id -- Panics if not found -tcLookupValueByKey key - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - returnNF_Tc (explicitLookupValueByKey ve key) - -tcLookupValueByKeyMaybe :: Unique -> NF_TcM s (Maybe Id) -tcLookupValueByKeyMaybe key - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - returnNF_Tc (lookupUFM_Directly ve key) - -tcGetValueEnv :: NF_TcM s ValueEnv -tcGetValueEnv - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te ve gtvs) -> - returnNF_Tc ve - -tcSetValueEnv :: ValueEnv -> TcM s a -> TcM s a -tcSetValueEnv ve scope - = tcGetEnv `thenNF_Tc` \ (TcEnv ue te _ gtvs) -> - tcSetEnv (TcEnv ue te ve gtvs) scope - --- Non-monadic version, environment given explicitly -explicitLookupValueByKey :: ValueEnv -> Unique -> Id -explicitLookupValueByKey ve key - = lookupWithDefaultUFM_Directly ve def key - where - def = pprPanic "lookupValueByKey:" (pprUnique10 key) - -explicitLookupValue :: ValueEnv -> Name -> Maybe Id -explicitLookupValue ve name - = case maybeWiredInIdName name of - Just id -> Just id - Nothing -> lookupNameEnv ve name - - -- Extract the IdInfo from an IfaceSig imported from an interface file -tcAddImportedIdInfo :: ValueEnv -> Id -> Id -tcAddImportedIdInfo unf_env id - | isLocallyDefined id -- Don't look up locally defined Ids, because they - -- have explicit local definitions, so we get a black hole! - = id - | otherwise - = id `lazySetIdInfo` new_info - -- The Id must be returned without a data dependency on maybe_id - where - new_info = -- pprTrace "tcAdd" (ppr id) $ - case explicitLookupValue unf_env (getName id) of - Nothing -> vanillaIdInfo - Just imported_id -> idInfo imported_id - -- ToDo: could check that types are the same -\end{code} +tcSetInstEnv :: InstEnv -> TcM a -> TcM a +tcSetInstEnv ie thing_inside + = tcGetEnv `thenNF_Tc` \ env -> + tcSetEnv (env {tcInsts = ie}) thing_inside +\end{code} %************************************************************************ %* * -\subsection{Constructing new Ids} +\subsection{The InstInfo type} %* * %************************************************************************ +The InstInfo type summarises the information in an instance declaration + + instance c => k (t tvs) where b + \begin{code} -newLocalId :: OccName -> TcType -> SrcLoc -> NF_TcM s TcId -newLocalId name ty loc - = tcGetUnique `thenNF_Tc` \ uniq -> - returnNF_Tc (mkUserLocal name uniq ty loc) +data InstInfo + = InstInfo { + iDFunId :: DFunId, -- The dfun id + iBinds :: RenamedMonoBinds, -- Bindings, b + iPrags :: [RenamedSig] -- User pragmas recorded for generating specialised instances + } -newSpecPragmaId :: Name -> TcType -> NF_TcM s TcId -newSpecPragmaId name ty - = tcGetUnique `thenNF_Tc` \ uniq -> - returnNF_Tc (mkSpecPragmaId (nameOccName name) uniq ty (getSrcLoc name)) +pprInstInfo info = vcat [ptext SLIT("InstInfo:") <+> ppr (idType (iDFunId info)), + nest 4 (ppr (iBinds info))] + +simpleInstInfoTy :: InstInfo -> Type +simpleInstInfoTy info = case splitDFunTy (idType (iDFunId info)) of + (_, _, _, [ty]) -> ty + +simpleInstInfoTyCon :: InstInfo -> TyCon + -- Gets the type constructor for a simple instance declaration, + -- i.e. one of the form instance (...) => C (T a b c) where ... +simpleInstInfoTyCon inst = tyConAppTyCon (simpleInstInfoTy inst) \end{code} @@ -470,14 +554,8 @@ newSpecPragmaId name ty %************************************************************************ \begin{code} -badCon con_id - = quotes (ppr con_id) <+> ptext SLIT("is not a data constructor") -badPrimOp op - = quotes (ppr op) <+> ptext SLIT("is not a primop") - -uvNameOutOfScope name - = ptext SLIT("UVar") <+> quotes (ppr name) <+> ptext SLIT("is not in scope") +badCon con_id = quotes (ppr con_id) <+> ptext SLIT("is not a data constructor") -tyNameOutOfScope name - = quotes (ppr name) <+> ptext SLIT("is not in scope") +notFound wheRe name = failWithTc (text wheRe <> colon <+> quotes (ppr name) <+> + ptext SLIT("is not in scope")) \end{code}