X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Ftypecheck%2FTcIfaceSig.lhs;h=5295fec764f734372e3daabd66614cfa6ad124ce;hb=5bb5efc23e4cf5c8ac20076015308d57f8ebfc09;hp=2d7a66669b41bad81605af621f2340ed65cfb40d;hpb=c29022c49449b7d8862dcc2259e16cafe9461945;p=ghc-hetmet.git diff --git a/ghc/compiler/typecheck/TcIfaceSig.lhs b/ghc/compiler/typecheck/TcIfaceSig.lhs index 2d7a666..5295fec 100644 --- a/ghc/compiler/typecheck/TcIfaceSig.lhs +++ b/ghc/compiler/typecheck/TcIfaceSig.lhs @@ -1,47 +1,47 @@ % -% (c) The GRASP/AQUA Project, Glasgow University, 1992-1996 +% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998 % \section[TcIfaceSig]{Type checking of type signatures in interface files} \begin{code} -module TcIfaceSig ( tcInterfaceSigs ) where +module TcIfaceSig ( tcInterfaceSigs, + tcCoreExpr, + tcCoreLamBndrs, + tcCoreBinds ) where #include "HsVersions.h" -import HsSyn ( HsDecl(..), IfaceSig(..) ) -import TcMonad -import TcMonoType ( tcHsType, tcHsTypeKind ) -import TcEnv ( tcExtendTyVarEnv, tcExtendGlobalValEnv, - tcLookupTyConByKey, tcLookupGlobalValueMaybe, - tcExplicitLookupGlobal - ) -import TcKind ( TcKind, kindToTcKind ) +import HsSyn ( CoreDecl(..), TyClDecl(..), HsTupCon(..) ) +import TcHsSyn ( TypecheckedCoreBind ) +import TcRnTypes +import TcRnMonad +import TcMonoType ( tcIfaceType, kcHsSigType ) +import TcEnv ( tcExtendTyVarEnv, tcExtendGlobalValEnv, tcLookupGlobalId, + tcLookupDataCon ) -import RnHsSyn ( RenamedHsDecl(..) ) +import RnHsSyn ( RenamedCoreDecl, RenamedTyClDecl ) import HsCore -import HsDecls ( HsIdInfo(..), HsStrictnessInfo(..) ) import Literal ( Literal(..) ) import CoreSyn -import CoreUtils ( coreExprType ) +import CoreUtils ( exprType ) import CoreUnfold -import MagicUFs ( MagicUnfoldingFun ) -import WwLib ( mkWrapper ) -import PrimOp ( PrimOp(..) ) - -import Id ( GenId, mkImported, mkUserId, addInlinePragma, - isPrimitiveId_maybe, dataConArgTys, Id ) -import Type ( mkSynTy, splitAlgTyConApp ) -import TyVar ( mkSysTyVar ) -import Name ( Name ) -import Unique ( rationalTyConKey, uniqueOf ) -import TysWiredIn ( integerTy ) -import PragmaInfo ( PragmaInfo(..) ) -import ErrUtils ( pprBagOfErrors ) -import Maybes ( maybeToBool ) -import Outputable -import Util ( zipWithEqual ) +import CoreLint ( lintUnfolding ) +import WorkWrap ( mkWrapper ) +import Id ( Id, mkVanillaGlobal, mkLocalId ) +import MkId ( mkFCallId ) import IdInfo +import TyCon ( tyConDataCons, tyConTyVars ) +import DataCon ( DataCon, dataConWorkId, dataConExistentialTyVars, dataConArgTys ) +import Type ( mkTyVarTys, splitTyConApp ) +import TysWiredIn ( tupleCon ) +import Var ( mkTyVar, tyVarKind ) +import Name ( Name ) +import UniqSupply ( initUs_ ) +import Outputable +import Util ( zipWithEqual, dropList, equalLength ) +import HscTypes ( typeEnvIds ) +import CmdLineOpts ( DynFlag(..) ) \end{code} Ultimately, type signatures in interfaces will have pragmatic @@ -52,287 +52,384 @@ As always, we do not have to worry about user-pragmas in interface signatures. \begin{code} -tcInterfaceSigs :: TcEnv s -- Envt to use when checking unfoldings - -> [RenamedHsDecl] -- Ignore non-sig-decls in these decls - -> TcM s [Id] +tcInterfaceSigs :: [RenamedTyClDecl] -- Ignore non-sig-decls in these decls + -> TcM TcGblEnv - -tcInterfaceSigs unf_env (SigD (IfaceSig name ty id_infos src_loc) : rest) - = tcAddSrcLoc src_loc ( - tcAddErrCtxt (ifaceSigCtxt name) ( - tcHsType ty `thenTc` \ sigma_ty -> - tcIdInfo unf_env name sigma_ty noIdInfo id_infos `thenTc` \ id_info' -> - let - imp_id = mkImported name sigma_ty id_info' - sig_id | any inline_please id_infos = addInlinePragma imp_id - | otherwise = imp_id - - inline_please (HsUnfold inline _) = inline - inline_please other = False - in - returnTc sig_id - )) `thenTc` \ sig_id -> - tcInterfaceSigs unf_env rest `thenTc` \ sig_ids -> - returnTc (sig_id : sig_ids) - -tcInterfaceSigs unf_env (other_decl : rest) = tcInterfaceSigs unf_env rest - -tcInterfaceSigs unf_env [] = returnTc [] -\end{code} - -\begin{code} -tcIdInfo unf_env name ty info info_ins - = go noIdInfo info_ins +-- May 2003: +-- NOTE 1: careful about the side-effected EPS +-- in the two tcExtendGlobalValueEnv calls +-- NOTE 2: no point in tying the knot with fixM; all +-- the important knot-tying comes via the PCS global variable + +tcInterfaceSigs decls = + zapEnv (fixM (tc_interface_sigs decls)) `thenM` \ (_,sig_ids) -> + -- The zapEnv dramatically trims the environment, solely + -- to plug the space leak that would otherwise be caused + -- by a rich environment bound into lots of lazy thunks + -- The thunks are the lazily-typechecked IdInfo of the + -- imported things. + + tcExtendGlobalValEnv sig_ids getGblEnv `thenM` \ gbl_env -> + returnM gbl_env + -- We tie a knot so that the Ids read out of interfaces are in scope + -- when we read their pragmas. + -- What we rely on is that pragmas are typechecked lazily; if + -- any type errors are found (ie there's an inconsistency) + -- we silently discard the pragma + -- + -- NOTE ALSO: the knot is in two parts: + -- * Ids defined in this module are added to the typechecker envt + -- which is knot-tied by the fixM. + -- * Imported Ids are side-effected into the PCS by the + -- tcExtendGlobalValueEnv, so they will be seen there provided + -- we don't look them up too early. + -- In both cases, we must defer lookups until after the knot is tied + -- + -- We used to have a much bigger loop (in TcRnDriver), so that the + -- interface pragmas could mention variables bound in this module + -- (by mutual recn), but + -- (a) the knot is tiresomely big, and + -- (b) it black-holes when we have Template Haskell + -- + -- For (b) consider: f = $(...h....) + -- where h is imported, and calls f via an hi-boot file. + -- This is bad! But it is not seen as a staging error, because h + -- is indeed imported. We don't want the type-checker to black-hole + -- when simplifying and compiling the splice! + -- + -- Simple solution: discard any unfolding that mentions a variable + -- bound in this module (and hence not yet processed). + -- The discarding happens when forkM finds a type error. + +tc_interface_sigs decls ~(unf_env, _) + = sequenceM [do_one d | d@(IfaceSig {}) <- decls] `thenM` \ sig_ids -> + tcExtendGlobalValEnv sig_ids getGblEnv `thenM` \ gbl_env -> + returnM (gbl_env, sig_ids) where - go info_so_far [] = returnTc info_so_far - go info (HsArity arity : rest) = go (info `addArityInfo` arity) rest - go info (HsUpdate upd : rest) = go (info `addUpdateInfo` upd) rest - go info (HsFBType fb : rest) = go (info `addFBTypeInfo` fb) rest - go info (HsArgUsage au : rest) = go (info `addArgUsageInfo` au) rest - - go info (HsUnfold inline expr : rest) = tcUnfolding unf_env name expr `thenNF_Tc` \ unfold_info -> - go (info `addUnfoldInfo` unfold_info) rest - - go info (HsStrictness strict : rest) = tcStrictness unf_env ty info strict `thenTc` \ info' -> - go info' rest + in_scope_vars = typeEnvIds (tcg_type_env unf_env) + -- When we have hi-boot files, an unfolding might refer to + -- something defined in this module, so we must build a + -- suitable in-scope set. This thunk will only be poked + -- if -dcore-lint is on. + + do_one IfaceSig {tcdName = name, tcdType = ty, + tcdIdInfo = id_infos, tcdLoc = src_loc} + = addSrcLoc src_loc $ + addErrCtxt (ifaceSigCtxt name) $ + tcIfaceType ty `thenM` \ sigma_ty -> + tcIdInfo unf_env in_scope_vars name + sigma_ty id_infos `thenM` \ id_info -> + returnM (mkVanillaGlobal name sigma_ty id_info) \end{code} \begin{code} -tcStrictness unf_env ty info (HsStrictnessInfo demands maybe_worker) - = tcWorker unf_env maybe_worker `thenNF_Tc` \ maybe_worker_id -> - uniqSMToTcM (mkWrapper ty demands) `thenNF_Tc` \ wrap_fn -> - let - -- Watch out! We can't pull on maybe_worker_id too eagerly! - info' = case maybe_worker_id of - Just worker_id -> info `addUnfoldInfo` mkUnfolding NoPragmaInfo (wrap_fn worker_id) - Nothing -> info - has_worker = maybeToBool maybe_worker_id - in - returnTc (info' `addStrictnessInfo` StrictnessInfo demands has_worker) - --- Boring to write these out, but the result type differs from the arg type... -tcStrictness unf_env ty info HsBottom - = returnTc (info `addStrictnessInfo` BottomGuaranteed) +tcIdInfo unf_env in_scope_vars name ty info_ins + = setGblEnv unf_env $ + -- Use the knot-tied environment for the IdInfo + -- In particular: typechecking unfoldings and worker names + foldlM tcPrag init_info info_ins + where + -- Set the CgInfo to something sensible but uninformative before + -- we start; default assumption is that it has CAFs + init_info = vanillaIdInfo + + tcPrag info HsNoCafRefs = returnM (info `setCafInfo` NoCafRefs) + tcPrag info (HsArity arity) = returnM (info `setArityInfo` arity) + tcPrag info (HsStrictness str) = returnM (info `setAllStrictnessInfo` Just str) + tcPrag info (HsWorker nm arity) = tcWorkerInfo ty info nm arity + + tcPrag info (HsUnfold inline_prag expr) + = tcPragExpr name in_scope_vars expr `thenM` \ maybe_expr' -> + let + -- maybe_expr' doesn't get looked at if the unfolding + -- is never inspected; so the typecheck doesn't even happen + unfold_info = case maybe_expr' of + Nothing -> noUnfolding + Just expr' -> mkTopUnfolding expr' + in + returnM (info `setUnfoldingInfoLazily` unfold_info + `setInlinePragInfo` inline_prag) \end{code} \begin{code} -tcWorker unf_env Nothing = returnNF_Tc Nothing +tcWorkerInfo ty info wkr_name arity + = forkM doc (tcVar wkr_name) `thenM` \ maybe_wkr_id -> + -- Watch out! We can't pull on unf_env too eagerly! + -- Hence the forkM + + -- We return without testing maybe_wkr_id, but as soon as info is + -- looked at we will test it. That's ok, because its outside the + -- knot; and there seems no big reason to further defer the + -- tcVar lookup. (Contrast with tcPragExpr, where postponing walking + -- over the unfolding until it's actually used does seem worth while.) + newUniqueSupply `thenM` \ us -> + returnM (case maybe_wkr_id of + Nothing -> info + Just wkr_id -> info `setUnfoldingInfoLazily` mk_unfolding us wkr_id + `setWorkerInfo` HasWorker wkr_id arity) -tcWorker unf_env (Just (worker_name,_)) - = returnNF_Tc (trace_maybe maybe_worker_id) where - maybe_worker_id = tcExplicitLookupGlobal unf_env worker_name + doc = text "worker for" <+> ppr wkr_name + + mk_unfolding us wkr_id = mkTopUnfolding (initUs_ us (mkWrapper ty strict_sig) wkr_id) - -- The trace is so we can see what's getting dropped - trace_maybe Nothing = pprTrace "tcWorker failed:" (ppr worker_name) Nothing - trace_maybe (Just x) = Just x + -- We are relying here on strictness info always appearing + -- before worker info, fingers crossed .... + strict_sig = case newStrictnessInfo info of + Just sig -> sig + Nothing -> pprPanic "Worker info but no strictness for" (ppr wkr_name) \end{code} For unfoldings we try to do the job lazily, so that we never type check an unfolding that isn't going to be looked at. \begin{code} -tcUnfolding unf_env name core_expr - = forkNF_Tc ( - recoverNF_Tc no_unfolding ( - tcSetEnv unf_env $ - tcCoreExpr core_expr `thenTc` \ core_expr' -> - returnTc (mkUnfolding NoPragmaInfo core_expr') - )) +tcPragExpr :: Name -> [Id] -> UfExpr Name -> TcM (Maybe CoreExpr) +tcPragExpr name in_scope_vars expr + = forkM doc $ + tcCoreExpr expr `thenM` \ core_expr' -> + + -- Check for type consistency in the unfolding + ifOptM Opt_DoCoreLinting ( + getSrcLocM `thenM` \ src_loc -> + case lintUnfolding src_loc in_scope_vars core_expr' of + Nothing -> returnM () + Just fail_msg -> failWithTc ((doc <+> text "Failed Lint") $$ fail_msg) + ) `thenM_` + + returnM core_expr' where - -- The trace tells what wasn't available, for the benefit of - -- compiler hackers who want to improve it! - no_unfolding = getErrsTc `thenNF_Tc` \ (warns,errs) -> - returnNF_Tc (pprTrace "tcUnfolding failed with:" - (hang (ppr name) 4 (pprBagOfErrors errs)) - NoUnfolding) + doc = text "unfolding of" <+> ppr name \end{code} Variables in unfoldings ~~~~~~~~~~~~~~~~~~~~~~~ -****** Inside here we use only the Global environment, even for locally bound variables. -****** Why? Because we know all the types and want to bind them to real Ids. \begin{code} -tcVar :: Name -> TcM s Id -tcVar name - = tcLookupGlobalValueMaybe name `thenNF_Tc` \ maybe_id -> - case maybe_id of { - Just id -> returnTc id; - Nothing -> failWithTc (noDecl name) - } - -noDecl name = hsep [ptext SLIT("Warning: no binding for"), ppr name] +tcVar :: Name -> TcM Id + -- Inside here we use only the Global environment, even for locally bound variables. + -- Why? Because we know all the types and want to bind them to real Ids. +tcVar name = tcLookupGlobalId name \end{code} UfCore expressions. \begin{code} -tcCoreExpr :: UfExpr Name -> TcM s CoreExpr +tcCoreExpr :: UfExpr Name -> TcM CoreExpr + +tcCoreExpr (UfType ty) + = tcIfaceType ty `thenM` \ ty' -> + -- It might not be of kind type + returnM (Type ty') tcCoreExpr (UfVar name) - = tcVar name `thenTc` \ id -> - returnTc (Var id) + = tcVar name `thenM` \ id -> + returnM (Var id) --- rationalTy isn't built in so we have to construct it --- (the "ty" part of the incoming literal is simply bottom) -tcCoreExpr (UfLit (NoRepRational lit _)) - = tcLookupTyConByKey rationalTyConKey `thenNF_Tc` \ rational_tycon -> - let - rational_ty = mkSynTy rational_tycon [] - in - returnTc (Lit (NoRepRational lit rational_ty)) +tcCoreExpr (UfLit lit) + = returnM (Lit lit) --- Similarly for integers, except that it is wired in -tcCoreExpr (UfLit (NoRepInteger lit _)) - = returnTc (Lit (NoRepInteger lit integerTy)) +-- The dreaded lit-lits are also similar, except here the type +-- is read in explicitly rather than being implicit +tcCoreExpr (UfLitLit lit ty) + = tcIfaceType ty `thenM` \ ty' -> + returnM (Lit (MachLitLit lit ty')) -tcCoreExpr (UfLit other_lit) - = returnTc (Lit other_lit) +tcCoreExpr (UfFCall cc ty) + = tcIfaceType ty `thenM` \ ty' -> + newUnique `thenM` \ u -> + returnM (Var (mkFCallId u cc ty')) -tcCoreExpr (UfCon con args) - = tcVar con `thenTc` \ con_id -> - mapTc tcCoreArg args `thenTc` \ args' -> - returnTc (Con con_id args') +tcCoreExpr (UfTuple (HsTupCon boxity arity) args) + = mappM tcCoreExpr args `thenM` \ args' -> + let + -- Put the missing type arguments back in + con_args = map (Type . exprType) args' ++ args' + in + returnM (mkApps (Var con_id) con_args) + where + con_id = dataConWorkId (tupleCon boxity arity) + -tcCoreExpr (UfPrim prim args) - = tcCorePrim prim `thenTc` \ primop -> - mapTc tcCoreArg args `thenTc` \ args' -> - returnTc (Prim primop args') +tcCoreExpr (UfLam bndr body) + = tcCoreLamBndr bndr $ \ bndr' -> + tcCoreExpr body `thenM` \ body' -> + returnM (Lam bndr' body') tcCoreExpr (UfApp fun arg) - = tcCoreExpr fun `thenTc` \ fun' -> - tcCoreArg arg `thenTc` \ arg' -> - returnTc (App fun' arg') + = tcCoreExpr fun `thenM` \ fun' -> + tcCoreExpr arg `thenM` \ arg' -> + returnM (App fun' arg') -tcCoreExpr (UfCase scrut alts) - = tcCoreExpr scrut `thenTc` \ scrut' -> - tcCoreAlts (coreExprType scrut') alts `thenTc` \ alts' -> - returnTc (Case scrut' alts') - -tcCoreExpr (UfSCC cc expr) - = tcCoreExpr expr `thenTc` \ expr' -> - returnTc (SCC cc expr') - -tcCoreExpr(UfCoerce coercion ty body) - = tcCoercion coercion `thenTc` \ coercion' -> - tcHsTypeKind ty `thenTc` \ (_,ty') -> - tcCoreExpr body `thenTc` \ body' -> - returnTc (Coerce coercion' ty' body') - -tcCoreExpr (UfLam bndr body) - = tcCoreLamBndr bndr $ \ bndr' -> - tcCoreExpr body `thenTc` \ body' -> - returnTc (Lam bndr' body') +tcCoreExpr (UfCase scrut case_bndr alts) + = tcCoreExpr scrut `thenM` \ scrut' -> + let + scrut_ty = exprType scrut' + case_bndr' = mkLocalId case_bndr scrut_ty + in + tcExtendGlobalValEnv [case_bndr'] $ + mappM (tcCoreAlt scrut_ty) alts `thenM` \ alts' -> + returnM (Case scrut' case_bndr' alts') tcCoreExpr (UfLet (UfNonRec bndr rhs) body) - = tcCoreExpr rhs `thenTc` \ rhs' -> + = tcCoreExpr rhs `thenM` \ rhs' -> tcCoreValBndr bndr $ \ bndr' -> - tcCoreExpr body `thenTc` \ body' -> - returnTc (Let (NonRec bndr' rhs') body') + tcCoreExpr body `thenM` \ body' -> + returnM (Let (NonRec bndr' rhs') body') tcCoreExpr (UfLet (UfRec pairs) body) = tcCoreValBndrs bndrs $ \ bndrs' -> - mapTc tcCoreExpr rhss `thenTc` \ rhss' -> - tcCoreExpr body `thenTc` \ body' -> - returnTc (Let (Rec (bndrs' `zip` rhss')) body') + mappM tcCoreExpr rhss `thenM` \ rhss' -> + tcCoreExpr body `thenM` \ body' -> + returnM (Let (Rec (bndrs' `zip` rhss')) body') where (bndrs, rhss) = unzip pairs + +tcCoreExpr (UfNote note expr) + = tcCoreExpr expr `thenM` \ expr' -> + case note of + UfCoerce to_ty -> tcIfaceType to_ty `thenM` \ to_ty' -> + returnM (Note (Coerce to_ty' + (exprType expr')) expr') + UfInlineCall -> returnM (Note InlineCall expr') + UfInlineMe -> returnM (Note InlineMe expr') + UfSCC cc -> returnM (Note (SCC cc) expr') \end{code} \begin{code} tcCoreLamBndr (UfValBinder name ty) thing_inside - = tcHsType ty `thenTc` \ ty' -> + = tcIfaceType ty `thenM` \ ty' -> let - id = mkUserId name ty' NoPragmaInfo + id = mkLocalId name ty' in tcExtendGlobalValEnv [id] $ - thing_inside (ValBinder id) + thing_inside id tcCoreLamBndr (UfTyBinder name kind) thing_inside = let - tyvar = mkSysTyVar (uniqueOf name) kind + tyvar = mkTyVar name kind in - tcExtendTyVarEnv [name] [(kindToTcKind kind, tyvar)] $ - thing_inside (TyBinder tyvar) + tcExtendTyVarEnv [tyvar] (thing_inside tyvar) +tcCoreLamBndrs [] thing_inside = thing_inside [] +tcCoreLamBndrs (b:bs) thing_inside + = tcCoreLamBndr b $ \ b' -> + tcCoreLamBndrs bs $ \ bs' -> + thing_inside (b':bs') + tcCoreValBndr (UfValBinder name ty) thing_inside - = tcHsType ty `thenTc` \ ty' -> + = tcIfaceType ty `thenM` \ ty' -> let - id = mk_id name ty' + id = mkLocalId name ty' in tcExtendGlobalValEnv [id] $ thing_inside id tcCoreValBndrs bndrs thing_inside -- Expect them all to be ValBinders - = mapTc tcHsType tys `thenTc` \ tys' -> + = mappM tcIfaceType tys `thenM` \ tys' -> let - ids = zipWithEqual "tcCoreValBndr" mk_id names tys' + ids = zipWithEqual "tcCoreValBndr" mkLocalId names tys' in tcExtendGlobalValEnv ids $ thing_inside ids where - names = map (\ (UfValBinder name _) -> name) bndrs - tys = map (\ (UfValBinder _ ty) -> ty) bndrs - -mk_id name ty = mkUserId name ty NoPragmaInfo + names = [name | UfValBinder name _ <- bndrs] + tys = [ty | UfValBinder _ ty <- bndrs] \end{code} \begin{code} -tcCoreArg (UfVarArg v) = tcVar v `thenTc` \ v' -> returnTc (VarArg v') -tcCoreArg (UfTyArg ty) = tcHsTypeKind ty `thenTc` \ (_,ty') -> returnTc (TyArg ty') -tcCoreArg (UfLitArg lit) = returnTc (LitArg lit) - -tcCoreAlts scrut_ty (UfAlgAlts alts deflt) - = mapTc tc_alt alts `thenTc` \ alts' -> - tcCoreDefault scrut_ty deflt `thenTc` \ deflt' -> - returnTc (AlgAlts alts' deflt') - where - tc_alt (con, names, rhs) - = tcVar con `thenTc` \ con' -> - let - arg_tys = dataConArgTys con' inst_tys - (tycon, inst_tys, cons) = splitAlgTyConApp scrut_ty - arg_ids = zipWithEqual "tcCoreAlts" mk_id names arg_tys - in - tcExtendGlobalValEnv arg_ids $ - tcCoreExpr rhs `thenTc` \ rhs' -> - returnTc (con', arg_ids, rhs') - -tcCoreAlts scrut_ty (UfPrimAlts alts deflt) - = mapTc tc_alt alts `thenTc` \ alts' -> - tcCoreDefault scrut_ty deflt `thenTc` \ deflt' -> - returnTc (PrimAlts alts' deflt') - where - tc_alt (lit, rhs) = tcCoreExpr rhs `thenTc` \ rhs' -> - returnTc (lit, rhs') - -tcCoreDefault scrut_ty UfNoDefault = returnTc NoDefault -tcCoreDefault scrut_ty (UfBindDefault name rhs) - = let - deflt_id = mk_id name scrut_ty +tcCoreAlt scrut_ty (UfDefault, names, rhs) + = ASSERT( null names ) + tcCoreExpr rhs `thenM` \ rhs' -> + returnM (DEFAULT, [], rhs') + +tcCoreAlt scrut_ty (UfLitAlt lit, names, rhs) + = ASSERT( null names ) + tcCoreExpr rhs `thenM` \ rhs' -> + returnM (LitAlt lit, [], rhs') + +tcCoreAlt scrut_ty (UfLitLitAlt str ty, names, rhs) + = ASSERT( null names ) + tcCoreExpr rhs `thenM` \ rhs' -> + tcIfaceType ty `thenM` \ ty' -> + returnM (LitAlt (MachLitLit str ty'), [], rhs') + +-- A case alternative is made quite a bit more complicated +-- by the fact that we omit type annotations because we can +-- work them out. True enough, but its not that easy! +tcCoreAlt scrut_ty alt@(con, names, rhs) + = tcConAlt con `thenM` \ con -> + let + ex_tyvars = dataConExistentialTyVars con + (tycon, inst_tys) = splitTyConApp scrut_ty -- NB: not tcSplitTyConApp + -- We are looking at Core here + main_tyvars = tyConTyVars tycon + ex_tyvars' = [mkTyVar name (tyVarKind tv) | (name,tv) <- names `zip` ex_tyvars] + ex_tys' = mkTyVarTys ex_tyvars' + arg_tys = dataConArgTys con (inst_tys ++ ex_tys') + id_names = dropList ex_tyvars names + arg_ids +#ifdef DEBUG + | not (equalLength id_names arg_tys) + = pprPanic "tcCoreAlts" (ppr (con, names, rhs) $$ + (ppr main_tyvars <+> ppr ex_tyvars) $$ + ppr arg_tys) + | otherwise +#endif + = zipWithEqual "tcCoreAlts" mkLocalId id_names arg_tys in - tcExtendGlobalValEnv [deflt_id] $ - tcCoreExpr rhs `thenTc` \ rhs' -> - returnTc (BindDefault deflt_id rhs') - + ASSERT( con `elem` tyConDataCons tycon && equalLength inst_tys main_tyvars ) + tcExtendTyVarEnv ex_tyvars' $ + tcExtendGlobalValEnv arg_ids $ + tcCoreExpr rhs `thenM` \ rhs' -> + returnM (DataAlt con, ex_tyvars' ++ arg_ids, rhs') -tcCoercion (UfIn n) = tcVar n `thenTc` \ n' -> returnTc (CoerceIn n') -tcCoercion (UfOut n) = tcVar n `thenTc` \ n' -> returnTc (CoerceOut n') -tcCorePrim (UfOtherOp op) - = tcVar op `thenTc` \ op_id -> - case isPrimitiveId_maybe op_id of - Just prim_op -> returnTc prim_op - Nothing -> pprPanic "tcCorePrim" (ppr op_id) +tcConAlt :: UfConAlt Name -> TcM DataCon +tcConAlt (UfTupleAlt (HsTupCon boxity arity)) + = returnM (tupleCon boxity arity) -tcCorePrim (UfCCallOp str casm gc arg_tys res_ty) - = mapTc tcHsType arg_tys `thenTc` \ arg_tys' -> - tcHsType res_ty `thenTc` \ res_ty' -> - returnTc (CCallOp str casm gc arg_tys' res_ty') +tcConAlt (UfDataAlt con_name) -- When reading interface files + -- the con_name will be the real name of + -- the data con + = tcLookupDataCon con_name \end{code} +%************************************************************************ +%* * +\subsection{Core decls} +%* * +%************************************************************************ + + +\begin{code} +tcCoreBinds :: [RenamedCoreDecl] -> TcM [TypecheckedCoreBind] +-- We don't assume the bindings are in dependency order +-- So first build the environment, then check the RHSs +tcCoreBinds ls = mappM tcCoreBinder ls `thenM` \ bndrs -> + tcExtendGlobalValEnv bndrs $ + mappM (tcCoreBind bndrs) ls + +tcCoreBinder (CoreDecl nm ty _ _) + = kcHsSigType ty `thenM_` + tcIfaceType ty `thenM` \ ty' -> + returnM (mkLocalId nm ty') + +tcCoreBind bndrs (CoreDecl nm _ rhs loc) + = tcVar nm `thenM` \ id -> + tcCoreExpr rhs `thenM` \ rhs' -> + let + mb_err = lintUnfolding loc bndrs rhs' + in + (case mb_err of + Just err -> addErr err + Nothing -> returnM ()) `thenM_` + + returnM (id, rhs') +\end{code} + + \begin{code} ifaceSigCtxt sig_name = hsep [ptext SLIT("In an interface-file signature for"), ppr sig_name]