X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=compiler%2Ftypecheck%2FTcEnv.lhs;h=d59278a26357a881769db37943203ea7f17addf2;hb=02c48cf14cd8c7771dfb41089412f35e1eaeedd5;hp=2e3b80ba7e356008476191881a40f513de80dba1;hpb=1166c7d62f3fa9acd2084c90df6585cbbf868ceb;p=ghc-hetmet.git diff --git a/compiler/typecheck/TcEnv.lhs b/compiler/typecheck/TcEnv.lhs index 2e3b80b..d59278a 100644 --- a/compiler/typecheck/TcEnv.lhs +++ b/compiler/typecheck/TcEnv.lhs @@ -17,7 +17,7 @@ module TcEnv( tcLookupLocatedGlobal, tcLookupGlobal, tcLookupField, tcLookupTyCon, tcLookupClass, tcLookupDataCon, tcLookupLocatedGlobalId, tcLookupLocatedTyCon, - tcLookupLocatedClass, + tcLookupLocatedClass, tcLookupFamInst, -- Local environment tcExtendKindEnv, tcExtendKindEnvTvs, @@ -61,6 +61,7 @@ import VarSet import VarEnv import RdrName import InstEnv +import FamInstEnv import DataCon import TyCon import Class @@ -157,6 +158,18 @@ tcLookupLocatedClass = addLocM tcLookupClass tcLookupLocatedTyCon :: Located Name -> TcM TyCon tcLookupLocatedTyCon = addLocM tcLookupTyCon + +-- Look up the representation tycon of a family instance. +-- +tcLookupFamInst :: TyCon -> [Type] -> TcM TyCon +tcLookupFamInst tycon tys + = do { env <- getGblEnv + ; eps <- getEps + ; let instEnv = (eps_fam_inst_env eps, tcg_fam_inst_env env) + ; case lookupFamInstEnvExact instEnv tycon tys of + Nothing -> famInstNotFound tycon tys + Just famInst -> return $ famInstTyCon famInst + } \end{code} %************************************************************************ @@ -400,6 +413,9 @@ refineEnvironment :: Refinement -> TcM a -> TcM a -- I don't think I have to refine the set of global type variables in scope -- Reason: the refinement never increases that set refineEnvironment reft thing_inside + | isEmptyRefinement reft -- Common case + = thing_inside + | otherwise = do { env <- getLclEnv ; let le' = mapNameEnv refine (tcl_env env) ; setLclEnv (env {tcl_env = le'}) thing_inside } @@ -653,4 +669,9 @@ notFound name wrongThingErr expected thing name = failWithTc (pprTcTyThingCategory thing <+> quotes (ppr name) <+> ptext SLIT("used as a") <+> text expected) + +famInstNotFound tycon tys + = failWithTc (quotes famInst <+> ptext SLIT("is not in scope")) + where + famInst = ppr tycon <+> hsep (map pprParendType tys) \end{code}