X-Git-Url: http://git.megacz.com/?p=ghc-hetmet.git;a=blobdiff_plain;f=compiler%2Ftypes%2FFamInstEnv.lhs;h=eb31751678daf8add4ec0422749daec5759187f6;hp=9b49f5c34393fb9ac329bd4620894c90678cd6be;hb=84923cc7de2a93c22a2f72daf9ac863959efae13;hpb=8a5d47de2b82d9cca86546a7bd89d915488934ef diff --git a/compiler/types/FamInstEnv.lhs b/compiler/types/FamInstEnv.lhs index 9b49f5c..eb31751 100644 --- a/compiler/types/FamInstEnv.lhs +++ b/compiler/types/FamInstEnv.lhs @@ -6,7 +6,8 @@ FamInstEnv: Type checked family instance declarations \begin{code} module FamInstEnv ( - FamInst(..), famInstTyCon, pprFamInst, pprFamInstHdr, pprFamInsts, + FamInst(..), famInstTyCon, famInstTyVars, + pprFamInst, pprFamInstHdr, pprFamInsts, famInstHead, mkLocalFamInst, mkImportedFamInst, FamInstEnv, emptyFamInstEnv, extendFamInstEnv, extendFamInstEnvList, @@ -32,7 +33,6 @@ import UniqFM import Outputable import Maybe -import Monad \end{code} @@ -60,6 +60,8 @@ data FamInst -- famInstTyCon :: FamInst -> TyCon famInstTyCon = fi_tycon + +famInstTyVars = fi_tvs \end{code} \begin{code} @@ -174,7 +176,7 @@ extendFamInstEnv inst_env ins_item@(FamInst {fi_fam = cls_nm, fi_tcs = mb_tcs}) add (FamIE items tyvar) _ = FamIE (ins_item:items) (ins_tyvar || tyvar) ins_tyvar = not (any isJust mb_tcs) -\end{code} +\end{code} %************************************************************************ %* * @@ -182,6 +184,52 @@ extendFamInstEnv inst_env ins_item@(FamInst {fi_fam = cls_nm, fi_tcs = mb_tcs}) %* * %************************************************************************ +@lookupFamInstEnvExact@ looks up in a @FamInstEnv@ using an exact match. +This is used when we want the @TyCon@ of a particular family instance (e.g., +during deriving classes). + +\begin{code} +{- NOT NEEDED ANY MORE +lookupFamInstEnvExact :: (FamInstEnv -- External package inst-env + ,FamInstEnv) -- Home-package inst-env + -> TyCon -> [Type] -- What we are looking for + -> Maybe FamInst +lookupFamInstEnvExact (pkg_ie, home_ie) fam tys + = home_matches `mplus` pkg_matches + where + rough_tcs = roughMatchTcs tys + all_tvs = all isNothing rough_tcs + home_matches = lookup home_ie + pkg_matches = lookup pkg_ie + + -------------- + lookup env = case lookupUFM env fam of + Nothing -> Nothing -- No instances for this class + Just (FamIE insts has_tv_insts) + -- Short cut for common case: + -- The thing we are looking up is of form (C a + -- b c), and the FamIE has no instances of + -- that form, so don't bother to search + | all_tvs && not has_tv_insts -> Nothing + | otherwise -> find insts + + -------------- + find [] = Nothing + find (item@(FamInst { fi_tcs = mb_tcs, fi_tys = tpl_tys }) : rest) + -- Fast check for no match, uses the "rough match" fields + | instanceCantMatch rough_tcs mb_tcs + = find rest + + -- Proper check + | tcEqTypes tpl_tys tys + = Just item + + -- No match => try next + | otherwise + = find rest +-} +\end{code} + @lookupFamInstEnv@ looks up in a @FamInstEnv@, using a one-way match. Multiple matches are only possible in case of type families (not data families), and then, it doesn't matter which match we choose (as the