X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Frename%2FRnSource.lhs;h=9150440aeee654159d32276a54069cba77c03a52;hb=f85903abe9103e545ea5b1dc6fdd6b672da4f3f2;hp=1175d107d77ea7663b9dc3739cc6c1447ccdbcca;hpb=dbc254c3dcd64761015a3d1c191ac742caafbf4c;p=ghc-hetmet.git diff --git a/ghc/compiler/rename/RnSource.lhs b/ghc/compiler/rename/RnSource.lhs index 1175d10..9150440 100644 --- a/ghc/compiler/rename/RnSource.lhs +++ b/ghc/compiler/rename/RnSource.lhs @@ -5,60 +5,44 @@ \begin{code} module RnSource ( - rnSrcDecls, rnExtCoreDecls, checkModDeprec, - rnTyClDecl, rnIfaceRuleDecl, rnInstDecl, - rnBinds, rnBindsAndThen, rnStats, + rnSrcDecls, addTcgDUs, + rnTyClDecls, checkModDeprec, + rnSplice, checkTH ) where #include "HsVersions.h" +import {-# SOURCE #-} RnExpr( rnLExpr ) + import HsSyn -import RdrName ( RdrName, isRdrDataCon, elemRdrEnv ) -import RdrHsSyn ( RdrNameConDecl, RdrNameTyClDecl, RdrNameHsDecl, - RdrNameDeprecation, RdrNameFixitySig, - RdrNameHsBinds, - extractGenericPatTyVars - ) +import RdrName ( RdrName, isRdrDataCon, elemLocalRdrEnv, globalRdrEnvElts, + GlobalRdrElt(..), isLocalGRE ) +import RdrHsSyn ( extractGenericPatTyVars, extractHsRhoRdrTyVars ) import RnHsSyn -import HsCore - -import RnNames ( importsFromLocalDecls ) -import RnExpr ( rnExpr ) -import RnTypes ( rnHsType, rnHsSigType, rnHsTypeFVs, rnContext ) - -import RnBinds ( rnTopMonoBinds, rnMonoBinds, rnMethodBinds, - rnMonoBindsAndThen, renameSigs, checkSigs ) -import RnEnv ( lookupTopBndrRn, lookupOccRn, lookupSysBndr, - newLocalsRn, lookupGlobalOccRn, - bindLocalsFVRn, bindPatSigTyVars, +import RnTypes ( rnLHsType, rnLHsTypes, rnHsSigType, rnHsTypeFVs, rnContext ) +import RnBinds ( rnTopBinds, rnMethodBinds, renameSigs ) +import RnEnv ( lookupLocalDataTcNames, + lookupLocatedTopBndrRn, lookupLocatedOccRn, + lookupOccRn, newLocalsRn, + bindLocatedLocalsFV, bindPatSigTyVarsFV, bindTyVarsRn, extendTyVarEnvFVRn, - bindCoreLocalRn, bindCoreLocalsRn, bindLocalNames, - checkDupOrQualNames, checkDupNames, mapFvRn, - lookupTopSrcBndr_maybe, lookupTopSrcBndr, - dataTcOccs, unknownNameErr, - plusGlobalRdrEnv + bindLocalNames, checkDupNames, mapFvRn ) import TcRnMonad -import BasicTypes ( FixitySig(..) ) -import HscTypes ( ExternalPackageState(..), FixityEnv, - Deprecations(..), plusDeprecs ) -import Module ( moduleEnvElts ) -import Class ( FunDep, DefMeth (..) ) -import TyCon ( DataConDetails(..), visibleDataCons ) -import Name ( Name ) +import HscTypes ( FixityEnv, FixItem(..), + Deprecations, Deprecs(..), DeprecTxt, plusDeprecs ) +import Class ( FunDep ) +import Name ( Name, nameOccName ) import NameSet import NameEnv -import ErrUtils ( dumpIfSet ) -import PrelNames ( newStablePtrName, bindIOName, returnIOName ) -import List ( partition ) -import Bag ( bagToList ) +import OccName ( occEnvElts ) import Outputable -import SrcLoc ( SrcLoc ) -import CmdLineOpts ( DynFlag(..) ) - -- Warn of unused for-all'd tyvars -import Maybes ( maybeToBool, seqMaybe ) -import Maybe ( maybe, catMaybes, isNothing ) +import SrcLoc ( Located(..), unLoc, getLoc, noLoc ) +import DynFlags ( DynFlag(..) ) +import Maybes ( seqMaybe ) +import Maybe ( isNothing ) +import BasicTypes ( Boxity(..) ) \end{code} @rnSourceDecl@ `renames' declarations. @@ -78,48 +62,84 @@ Checks the @(..)@ etc constraints in the export list. \begin{code} -rnSrcDecls :: [RdrNameHsDecl] -> RnM (TcGblEnv, [RenamedHsDecl], FreeVars) - -rnSrcDecls decls - = do { (rdr_env, imports) <- importsFromLocalDecls decls ; - updGblEnv (\gbl -> gbl { tcg_rdr_env = rdr_env `plusGlobalRdrEnv` - tcg_rdr_env gbl, - tcg_imports = imports `plusImportAvails` - tcg_imports gbl }) - $ do { - - -- Deal with deprecations (returns only the extra deprecations) - deprecs <- rnSrcDeprecDecls [d | DeprecD d <- decls] ; +rnSrcDecls :: HsGroup RdrName -> RnM (TcGblEnv, HsGroup Name) + +rnSrcDecls (HsGroup { hs_valds = val_decls, + hs_tyclds = tycl_decls, + hs_instds = inst_decls, + hs_fixds = fix_decls, + hs_depds = deprec_decls, + hs_fords = foreign_decls, + hs_defds = default_decls, + hs_ruleds = rule_decls }) + + = do { -- Deal with deprecations (returns only the extra deprecations) + deprecs <- rnSrcDeprecDecls deprec_decls ; updGblEnv (\gbl -> gbl { tcg_deprecs = tcg_deprecs gbl `plusDeprecs` deprecs }) $ do { -- Deal with top-level fixity decls -- (returns the total new fixity env) - fix_env <- rnSrcFixityDecls decls ; + fix_env <- rnSrcFixityDeclsEnv fix_decls ; + rn_fix_decls <- rnSrcFixityDecls fix_decls ; updGblEnv (\gbl -> gbl { tcg_fix_env = fix_env }) $ do { - failIfErrsM ; -- No point in continuing if (say) we have duplicate declarations - - -- Rename remaining declarations - (rn_src_decls, src_fvs) <- rn_src_decls decls ; - + -- Rename other declarations + traceRn (text "Start rnmono") ; + (rn_val_decls, bind_dus) <- rnTopBinds val_decls ; + traceRn (text "finish rnmono" <+> ppr rn_val_decls) ; + + -- You might think that we could build proper def/use information + -- for type and class declarations, but they can be involved + -- in mutual recursion across modules, and we only do the SCC + -- analysis for them in the type checker. + -- So we content ourselves with gathering uses only; that + -- means we'll only report a declaration as unused if it isn't + -- mentioned at all. Ah well. + (rn_tycl_decls, src_fvs1) + <- mapFvRn (wrapLocFstM rnTyClDecl) tycl_decls ; + (rn_inst_decls, src_fvs2) + <- mapFvRn (wrapLocFstM rnSrcInstDecl) inst_decls ; + (rn_rule_decls, src_fvs3) + <- mapFvRn (wrapLocFstM rnHsRuleDecl) rule_decls ; + (rn_foreign_decls, src_fvs4) + <- mapFvRn (wrapLocFstM rnHsForeignDecl) foreign_decls ; + (rn_default_decls, src_fvs5) + <- mapFvRn (wrapLocFstM rnDefaultDecl) default_decls ; + + let { + rn_group = HsGroup { hs_valds = rn_val_decls, + hs_tyclds = rn_tycl_decls, + hs_instds = rn_inst_decls, + hs_fixds = rn_fix_decls, + hs_depds = [], + hs_fords = rn_foreign_decls, + hs_defds = rn_default_decls, + hs_ruleds = rn_rule_decls } ; + + other_fvs = plusFVs [src_fvs1, src_fvs2, src_fvs3, + src_fvs4, src_fvs5] ; + src_dus = bind_dus `plusDU` usesOnly other_fvs + -- Note: src_dus will contain *uses* for locally-defined types + -- and classes, but no *defs* for them. (Because rnTyClDecl + -- returns only the uses.) This is a little + -- surprising but it doesn't actually matter at all. + } ; + + traceRn (text "finish rnSrc" <+> ppr rn_group) ; + traceRn (text "finish Dus" <+> ppr src_dus ) ; tcg_env <- getGblEnv ; - return (tcg_env, rn_src_decls, src_fvs) - }}}} + return (tcg_env `addTcgDUs` src_dus, rn_group) + }}} -rnExtCoreDecls :: [RdrNameHsDecl] -> RnM ([RenamedHsDecl], FreeVars) -rnExtCoreDecls decls = rn_src_decls decls +rnTyClDecls :: [LTyClDecl RdrName] -> RnM [LTyClDecl Name] +rnTyClDecls tycl_decls = do + (decls', fvs) <- mapFvRn (wrapLocFstM rnTyClDecl) tycl_decls + return decls' -rn_src_decls decls -- Declarartions get reversed, but no matter - = go emptyFVs [] decls - where - -- Fixity and deprecations have been dealt with already; ignore them - go fvs ds' [] = returnM (ds', fvs) - go fvs ds' (FixD _:ds) = go fvs ds' ds - go fvs ds' (DeprecD _:ds) = go fvs ds' ds - go fvs ds' (d:ds) = rnSrcDecl d `thenM` \(d', fvs') -> - go (fvs `plusFV` fvs') (d':ds') ds +addTcgDUs :: TcGblEnv -> DefUses -> TcGblEnv +addTcgDUs tcg_env dus = tcg_env { tcg_dus = tcg_dus tcg_env `plusDU` dus } \end{code} @@ -130,45 +150,53 @@ rn_src_decls decls -- Declarartions get reversed, but no matter %********************************************************* \begin{code} -rnSrcFixityDecls :: [RdrNameHsDecl] -> TcRn m FixityEnv -rnSrcFixityDecls decls +rnSrcFixityDecls :: [LFixitySig RdrName] -> RnM [LFixitySig Name] +rnSrcFixityDecls fix_decls + = do fix_decls <- mapM rnFixityDecl fix_decls + return (concat fix_decls) + +rnFixityDecl :: LFixitySig RdrName -> RnM [LFixitySig Name] +rnFixityDecl (L loc (FixitySig (L nameLoc rdr_name) fixity)) + = do names <- lookupLocalDataTcNames rdr_name + return [ L loc (FixitySig (L nameLoc name) fixity) + | name <- names ] + +rnSrcFixityDeclsEnv :: [LFixitySig RdrName] -> RnM FixityEnv +rnSrcFixityDeclsEnv fix_decls = getGblEnv `thenM` \ gbl_env -> - foldlM rnFixityDecl (tcg_fix_env gbl_env) - fix_decls `thenM` \ fix_env -> - traceRn (text "fixity env" <+> ppr fix_env) `thenM_` + foldlM rnFixityDeclEnv (tcg_fix_env gbl_env) + fix_decls `thenM` \ fix_env -> + traceRn (text "fixity env" <+> pprFixEnv fix_env) `thenM_` returnM fix_env - where - fix_decls = foldr get_fix_sigs [] decls - - -- Get fixities from top level decls, and from class decl sigs too - get_fix_sigs (FixD fix) acc = fix:acc - get_fix_sigs (TyClD (ClassDecl { tcdSigs = sigs})) acc - = [sig | FixSig sig <- sigs] ++ acc - get_fix_sigs other_decl acc = acc -rnFixityDecl :: FixityEnv -> RdrNameFixitySig -> TcRn m FixityEnv -rnFixityDecl fix_env (FixitySig rdr_name fixity loc) - = -- GHC extension: look up both the tycon and data con +rnFixityDeclEnv :: FixityEnv -> LFixitySig RdrName -> RnM FixityEnv +rnFixityDeclEnv fix_env (L loc (FixitySig rdr_name fixity)) + = setSrcSpan loc $ + -- GHC extension: look up both the tycon and data con -- for con-like things -- If neither are in scope, report an error; otherwise -- add both to the fixity env - mappM lookupTopSrcBndr_maybe (dataTcOccs rdr_name) `thenM` \ maybe_ns -> - case catMaybes maybe_ns of - [] -> addSrcLoc loc $ - addErr (unknownNameErr rdr_name) `thenM_` - returnM fix_env - ns -> foldlM add fix_env ns + addLocM lookupLocalDataTcNames rdr_name `thenM` \ names -> + foldlM add fix_env names where - add fix_env name + add fix_env name = case lookupNameEnv fix_env name of - Just (FixitySig _ _ loc') -> addErr (dupFixityDecl rdr_name loc loc') `thenM_` - returnM fix_env - Nothing -> returnM (extendNameEnv fix_env name (FixitySig name fixity loc)) - -dupFixityDecl rdr_name loc1 loc2 + Just (FixItem _ _ loc') + -> addLocErr rdr_name (dupFixityDecl loc') `thenM_` + returnM fix_env + Nothing -> returnM (extendNameEnv fix_env name fix_item) + where + fix_item = FixItem (nameOccName name) fixity (getLoc rdr_name) + +pprFixEnv :: FixityEnv -> SDoc +pprFixEnv env + = pprWithCommas (\ (FixItem n f _) -> ppr f <+> ppr n) + (nameEnvElts env) + +dupFixityDecl loc rdr_name = vcat [ptext SLIT("Multiple fixity declarations for") <+> quotes (ppr rdr_name), - ptext SLIT("at ") <+> ppr loc1, - ptext SLIT("and") <+> ppr loc2] + ptext SLIT("also at ") <+> ppr loc + ] \end{code} @@ -183,27 +211,22 @@ It's only imported deprecations, dealt with in RnIfaces, that we gather them together. \begin{code} -rnSrcDeprecDecls :: [RdrNameDeprecation] -> TcRn m Deprecations +rnSrcDeprecDecls :: [LDeprecDecl RdrName] -> RnM Deprecations rnSrcDeprecDecls [] = returnM NoDeprecs rnSrcDeprecDecls decls - = mappM rn_deprec decls `thenM` \ pairs -> - returnM (DeprecSome (mkNameEnv (catMaybes pairs))) + = mappM (addLocM rn_deprec) decls `thenM` \ pairs_s -> + returnM (DeprecSome (mkNameEnv (concat pairs_s))) where - rn_deprec (Deprecation rdr_name txt loc) - = addSrcLoc loc $ - lookupTopSrcBndr rdr_name `thenM` \ name -> - returnM (Just (name, (name,txt))) + rn_deprec (Deprecation rdr_name txt) + = lookupLocalDataTcNames rdr_name `thenM` \ names -> + returnM [(name, (nameOccName name, txt)) | name <- names] checkModDeprec :: Maybe DeprecTxt -> Deprecations -- Check for a module deprecation; done once at top level checkModDeprec Nothing = NoDeprecs -checkModdeprec (Just txt) = DeprecAll txt - -badDeprec d - = sep [ptext SLIT("Illegal deprecation when whole module is deprecated"), - nest 4 (ppr d)] +checkModDeprec (Just txt) = DeprecAll txt \end{code} %********************************************************* @@ -213,105 +236,29 @@ badDeprec d %********************************************************* \begin{code} -rnSrcDecl :: RdrNameHsDecl -> RnM (RenamedHsDecl, FreeVars) - -rnSrcDecl (ValD binds) = rnTopBinds binds `thenM` \ (new_binds, fvs) -> - returnM (ValD new_binds, fvs) - -rnSrcDecl (TyClD tycl_decl) - = rnTyClDecl tycl_decl `thenM` \ new_decl -> - finishSourceTyClDecl tycl_decl new_decl `thenM` \ (new_decl', fvs) -> - returnM (TyClD new_decl', fvs `plusFV` tyClDeclFVs new_decl') - -rnSrcDecl (InstD inst) - = rnInstDecl inst `thenM` \ new_inst -> - finishSourceInstDecl inst new_inst `thenM` \ (new_inst', fvs) -> - returnM (InstD new_inst', fvs `plusFV` instDeclFVs new_inst') - -rnSrcDecl (RuleD rule) - = rnHsRuleDecl rule `thenM` \ (new_rule, fvs) -> - returnM (RuleD new_rule, fvs) - -rnSrcDecl (ForD ford) - = rnHsForeignDecl ford `thenM` \ (new_ford, fvs) -> - returnM (ForD new_ford, fvs) - -rnSrcDecl (DefD (DefaultDecl tys src_loc)) - = addSrcLoc src_loc $ - mapFvRn (rnHsTypeFVs doc_str) tys `thenM` \ (tys', fvs) -> - returnM (DefD (DefaultDecl tys' src_loc), fvs) +rnDefaultDecl (DefaultDecl tys) + = mapFvRn (rnHsTypeFVs doc_str) tys `thenM` \ (tys', fvs) -> + returnM (DefaultDecl tys', fvs) where doc_str = text "In a `default' declaration" - - -rnSrcDecl (CoreD (CoreDecl name ty rhs loc)) - = addSrcLoc loc $ - lookupTopBndrRn name `thenM` \ name' -> - rnHsTypeFVs doc_str ty `thenM` \ (ty', ty_fvs) -> - rnCoreExpr rhs `thenM` \ rhs' -> - returnM (CoreD (CoreDecl name' ty' rhs' loc), - ty_fvs `plusFV` ufExprFVs rhs') - where - doc_str = text "In the Core declaration for" <+> quotes (ppr name) \end{code} %********************************************************* %* * - Bindings -%* * -%********************************************************* - -These chaps are here, rather than in TcBinds, so that there -is just one hi-boot file (for RnSource). rnSrcDecls is part -of the loop too, and it must be defined in this module. - -\begin{code} -rnTopBinds :: RdrNameHsBinds -> RnM (RenamedHsBinds, FreeVars) -rnTopBinds EmptyBinds = returnM (EmptyBinds, emptyFVs) -rnTopBinds (MonoBind bind sigs _) = rnTopMonoBinds bind sigs - -- The parser doesn't produce other forms - -rnBinds :: RdrNameHsBinds -> RnM (RenamedHsBinds, FreeVars) --- This version assumes that the binders are already in scope -rnBinds EmptyBinds = returnM (EmptyBinds, emptyFVs) -rnBinds (MonoBind bind sigs _) = rnMonoBinds bind sigs - -- The parser doesn't produce other forms - -rnBindsAndThen :: RdrNameHsBinds - -> (RenamedHsBinds -> RnM (result, FreeVars)) - -> RnM (result, FreeVars) --- This version (a) assumes that the binding vars are not already in scope --- (b) removes the binders from the free vars of the thing inside -rnBindsAndThen EmptyBinds thing_inside = thing_inside EmptyBinds -rnBindsAndThen (MonoBind bind sigs _) thing_inside = rnMonoBindsAndThen bind sigs thing_inside - -- The parser doesn't produce other forms -\end{code} - - -%********************************************************* -%* * \subsection{Foreign declarations} %* * %********************************************************* \begin{code} -rnHsForeignDecl (ForeignImport name ty spec isDeprec src_loc) - = addSrcLoc src_loc $ - lookupTopBndrRn name `thenM` \ name' -> +rnHsForeignDecl (ForeignImport name ty spec isDeprec) + = lookupLocatedTopBndrRn name `thenM` \ name' -> rnHsTypeFVs (fo_decl_msg name) ty `thenM` \ (ty', fvs) -> - returnM (ForeignImport name' ty' spec isDeprec src_loc, - fvs `plusFV` extras spec) - where - extras (CImport _ _ _ _ CWrapper) = mkFVs [newStablePtrName, - bindIOName, returnIOName] - extras _ = emptyFVs - -rnHsForeignDecl (ForeignExport name ty spec isDeprec src_loc) - = addSrcLoc src_loc $ - lookupOccRn name `thenM` \ name' -> - rnHsTypeFVs (fo_decl_msg name) ty `thenM` \ (ty', fvs) -> - returnM (ForeignExport name' ty' spec isDeprec src_loc, - mkFVs [name', bindIOName, returnIOName] `plusFV` fvs ) + returnM (ForeignImport name' ty' spec isDeprec, fvs) + +rnHsForeignDecl (ForeignExport name ty spec isDeprec) + = lookupLocatedOccRn name `thenM` \ name' -> + rnHsTypeFVs (fo_decl_msg name) ty `thenM` \ (ty', fvs) -> + returnM (ForeignExport name' ty' spec isDeprec, fvs ) -- NB: a foreign export is an *occurrence site* for name, so -- we add it to the free-variable list. It might, for example, -- be imported from another module @@ -327,42 +274,24 @@ fo_decl_msg name = ptext SLIT("In the foreign declaration for") <+> ppr name %********************************************************* \begin{code} -rnInstDecl (InstDecl inst_ty mbinds uprags maybe_dfun_rdr_name src_loc) +rnSrcInstDecl (InstDecl inst_ty mbinds uprags) -- Used for both source and interface file decls - = addSrcLoc src_loc $ - rnHsSigType (text "an instance decl") inst_ty `thenM` \ inst_ty' -> - - (case maybe_dfun_rdr_name of - Nothing -> returnM Nothing - Just dfun_rdr_name -> lookupGlobalOccRn dfun_rdr_name `thenM` \ dfun_name -> - returnM (Just dfun_name) - ) `thenM` \ maybe_dfun_name -> - - -- The typechecker checks that all the bindings are for the right class. - returnM (InstDecl inst_ty' EmptyMonoBinds [] maybe_dfun_name src_loc) - --- Compare finishSourceTyClDecl -finishSourceInstDecl (InstDecl _ mbinds uprags _ _ ) - (InstDecl inst_ty _ _ maybe_dfun_name src_loc) - -- Used for both source decls only - = ASSERT( not (maybeToBool maybe_dfun_name) ) -- Source decl! + = rnHsSigType (text "an instance decl") inst_ty `thenM` \ inst_ty' -> + + -- Rename the bindings + -- The typechecker (not the renamer) checks that all + -- the bindings are for the right class let meth_doc = text "In the bindings in an instance declaration" - meth_names = collectLocatedMonoBinders mbinds - (inst_tyvars, _, cls,_) = splitHsInstDeclTy inst_ty - -- (Slightly strangely) the forall-d tyvars scope over - -- the method bindings too + meth_names = collectHsBindLocatedBinders mbinds + (inst_tyvars, _, cls,_) = splitHsInstDeclTy (unLoc inst_ty') in - - -- Rename the bindings - -- NB meth_names can be qualified! - checkDupNames meth_doc meth_names `thenM_` + checkDupNames meth_doc meth_names `thenM_` extendTyVarEnvForMethodBinds inst_tyvars ( + -- (Slightly strangely) the forall-d tyvars scope over + -- the method bindings too rnMethodBinds cls [] mbinds ) `thenM` \ (mbinds', meth_fvs) -> - let - binders = collectMonoBinders mbinds' - in -- Rename the prags and signatures. -- Note that the type variables are not in scope here, -- so that instance Eq a => Eq (T a) where @@ -370,13 +299,30 @@ finishSourceInstDecl (InstDecl _ mbinds uprags _ _ ) -- works OK. -- -- But the (unqualified) method names are in scope - bindLocalNames binders (renameSigs uprags) `thenM` \ uprags' -> - checkSigs okInstDclSig (mkNameSet binders) uprags' `thenM_` + let + binders = collectHsBindBinders mbinds' + ok_sig = okInstDclSig (mkNameSet binders) + in + bindLocalNames binders (renameSigs ok_sig uprags) `thenM` \ uprags' -> - returnM (InstDecl inst_ty mbinds' uprags' maybe_dfun_name src_loc, - meth_fvs `plusFV` hsSigsFVs uprags') + returnM (InstDecl inst_ty' mbinds' uprags', + meth_fvs `plusFV` hsSigsFVs uprags' + `plusFV` extractHsTyNames inst_ty') \end{code} +For the method bindings in class and instance decls, we extend the +type variable environment iff -fglasgow-exts + +\begin{code} +extendTyVarEnvForMethodBinds tyvars thing_inside + = doptM Opt_GlasgowExts `thenM` \ opt_GlasgowExts -> + if opt_GlasgowExts then + extendTyVarEnvFVRn (map hsLTyVarName tyvars) thing_inside + else + thing_inside +\end{code} + + %********************************************************* %* * \subsection{Rules} @@ -384,47 +330,36 @@ finishSourceInstDecl (InstDecl _ mbinds uprags _ _ ) %********************************************************* \begin{code} -rnIfaceRuleDecl (IfaceRule rule_name act vars fn args rhs src_loc) - = addSrcLoc src_loc $ - lookupOccRn fn `thenM` \ fn' -> - rnCoreBndrs vars $ \ vars' -> - mappM rnCoreExpr args `thenM` \ args' -> - rnCoreExpr rhs `thenM` \ rhs' -> - returnM (IfaceRule rule_name act vars' fn' args' rhs' src_loc) - -rnIfaceRuleDecl (IfaceRuleOut fn rule) -- Builtin rules come this way - = lookupOccRn fn `thenM` \ fn' -> - returnM (IfaceRuleOut fn' rule) - -rnHsRuleDecl (HsRule rule_name act vars lhs rhs src_loc) - = addSrcLoc src_loc $ - bindPatSigTyVars (collectRuleBndrSigTys vars) $ - - bindLocalsFVRn doc (map get_var vars) $ \ ids -> +rnHsRuleDecl (HsRule rule_name act vars lhs fv_lhs rhs fv_rhs) + = bindPatSigTyVarsFV (collectRuleBndrSigTys vars) $ + + bindLocatedLocalsFV doc (map get_var vars) $ \ ids -> mapFvRn rn_var (vars `zip` ids) `thenM` \ (vars', fv_vars) -> - rnExpr lhs `thenM` \ (lhs', fv_lhs) -> - rnExpr rhs `thenM` \ (rhs', fv_rhs) -> + rnLExpr lhs `thenM` \ (lhs', fv_lhs') -> + rnLExpr rhs `thenM` \ (rhs', fv_rhs') -> let mb_bad = validRuleLhs ids lhs' in checkErr (isNothing mb_bad) (badRuleLhsErr rule_name lhs' mb_bad) `thenM_` let - bad_vars = [var | var <- ids, not (var `elemNameSet` fv_lhs)] + bad_vars = [var | var <- ids, not (var `elemNameSet` fv_lhs')] in mappM (addErr . badRuleVar rule_name) bad_vars `thenM_` - returnM (HsRule rule_name act vars' lhs' rhs' src_loc, - fv_vars `plusFV` fv_lhs `plusFV` fv_rhs) + returnM (HsRule rule_name act vars' lhs' fv_lhs' rhs' fv_rhs', + fv_vars `plusFV` fv_lhs' `plusFV` fv_rhs') where doc = text "In the transformation rule" <+> ftext rule_name get_var (RuleBndr v) = v get_var (RuleBndrSig v _) = v - rn_var (RuleBndr v, id) = returnM (RuleBndr id, emptyFVs) - rn_var (RuleBndrSig v t, id) = rnHsTypeFVs doc t `thenM` \ (t', fvs) -> - returnM (RuleBndrSig id t', fvs) + rn_var (RuleBndr (L loc v), id) + = returnM (RuleBndr (L loc id), emptyFVs) + rn_var (RuleBndrSig (L loc v) t, id) + = rnHsTypeFVs doc t `thenM` \ (t', fvs) -> + returnM (RuleBndrSig (L loc id) t', fvs) \end{code} Check the shape of a transformation rule LHS. Currently @@ -436,30 +371,46 @@ applications. (E.g. a case expression is not allowed: too elaborate.) NB: if you add new cases here, make sure you add new ones to TcRule.ruleLhsTvs \begin{code} -validRuleLhs :: [Name] -> RenamedHsExpr -> Maybe RenamedHsExpr +validRuleLhs :: [Name] -> LHsExpr Name -> Maybe (HsExpr Name) -- Nothing => OK -- Just e => Not ok, and e is the offending expression validRuleLhs foralls lhs - = check lhs + = checkl lhs where - check (OpApp e1 op _ e2) = check op `seqMaybe` check_e e1 `seqMaybe` check_e e2 - check (HsApp e1 e2) = check e1 `seqMaybe` check_e e2 + checkl (L loc e) = check e + + check (OpApp e1 op _ e2) = checkl op `seqMaybe` checkl_e e1 `seqMaybe` checkl_e e2 + check (HsApp e1 e2) = checkl e1 `seqMaybe` checkl_e e2 check (HsVar v) | v `notElem` foralls = Nothing check other = Just other -- Failure + checkl_e (L loc e) = check_e e + check_e (HsVar v) = Nothing - check_e (HsPar e) = check_e e + check_e (HsPar e) = checkl_e e check_e (HsLit e) = Nothing check_e (HsOverLit e) = Nothing - check_e (OpApp e1 op _ e2) = check_e e1 `seqMaybe` check_e op `seqMaybe` check_e e2 - check_e (HsApp e1 e2) = check_e e1 `seqMaybe` check_e e2 - check_e (NegApp e _) = check_e e - check_e (ExplicitList _ es) = check_es es - check_e (ExplicitTuple es _) = check_es es + check_e (OpApp e1 op _ e2) = checkl_e e1 `seqMaybe` checkl_e op `seqMaybe` checkl_e e2 + check_e (HsApp e1 e2) = checkl_e e1 `seqMaybe` checkl_e e2 + check_e (NegApp e _) = checkl_e e + check_e (ExplicitList _ es) = checkl_es es + check_e (ExplicitTuple es _) = checkl_es es check_e other = Just other -- Fails - check_es es = foldr (seqMaybe . check_e) Nothing es + checkl_es es = foldr (seqMaybe . checkl_e) Nothing es + +badRuleLhsErr name lhs (Just bad_e) + = sep [ptext SLIT("Rule") <+> ftext name <> colon, + nest 4 (vcat [ptext SLIT("Illegal expression:") <+> ppr bad_e, + ptext SLIT("in left-hand side:") <+> ppr lhs])] + $$ + ptext SLIT("LHS must be of form (f e1 .. en) where f is not forall'd") + +badRuleVar name var + = sep [ptext SLIT("Rule") <+> doubleQuotes (ftext name) <> colon, + ptext SLIT("Forall'd variable") <+> quotes (ppr var) <+> + ptext SLIT("does not appear on left hand side")] \end{code} @@ -483,123 +434,97 @@ and then go over it again to rename the tyvars! However, we can also do some scoping checks at the same time. \begin{code} -rnTyClDecl (IfaceSig {tcdName = name, tcdType = ty, tcdIdInfo = id_infos, tcdLoc = loc}) - = addSrcLoc loc $ - lookupTopBndrRn name `thenM` \ name' -> - rnHsType doc_str ty `thenM` \ ty' -> - mappM rnIdInfo id_infos `thenM` \ id_infos' -> - returnM (IfaceSig {tcdName = name', tcdType = ty', tcdIdInfo = id_infos', tcdLoc = loc}) - where - doc_str = text "In the interface signature for" <+> quotes (ppr name) - -rnTyClDecl (ForeignType {tcdName = name, tcdFoType = fo_type, tcdExtName = ext_name, tcdLoc = loc}) - = addSrcLoc loc $ - lookupTopBndrRn name `thenM` \ name' -> - returnM (ForeignType {tcdName = name', tcdFoType = fo_type, tcdExtName = ext_name, tcdLoc = loc}) - -rnTyClDecl (TyData {tcdND = new_or_data, tcdCtxt = context, tcdName = tycon, - tcdTyVars = tyvars, tcdCons = condecls, tcdGeneric = want_generic, - tcdDerivs = derivs, tcdLoc = src_loc}) - = addSrcLoc src_loc $ - lookupTopBndrRn tycon `thenM` \ tycon' -> +rnTyClDecl (ForeignType {tcdLName = name, tcdFoType = fo_type, tcdExtName = ext_name}) + = lookupLocatedTopBndrRn name `thenM` \ name' -> + returnM (ForeignType {tcdLName = name', tcdFoType = fo_type, tcdExtName = ext_name}, + emptyFVs) + +rnTyClDecl (TyData {tcdND = new_or_data, tcdCtxt = context, tcdLName = tycon, + tcdTyVars = tyvars, tcdCons = condecls, + tcdKindSig = sig, tcdDerivs = derivs}) + | is_vanilla -- Normal Haskell data type decl + = ASSERT( isNothing sig ) -- In normal H98 form, kind signature on the + -- data type is syntactically illegal bindTyVarsRn data_doc tyvars $ \ tyvars' -> - rnContext data_doc context `thenM` \ context' -> - rn_derivs derivs `thenM` \ derivs' -> - checkDupOrQualNames data_doc con_names `thenM_` - - rnConDecls tycon' condecls `thenM` \ condecls' -> - returnM (TyData {tcdND = new_or_data, tcdCtxt = context', tcdName = tycon', - tcdTyVars = tyvars', tcdCons = condecls', tcdGeneric = want_generic, - tcdDerivs = derivs', tcdLoc = src_loc}) + do { tycon' <- lookupLocatedTopBndrRn tycon + ; context' <- rnContext data_doc context + ; (derivs', deriv_fvs) <- rn_derivs derivs + ; checkDupNames data_doc con_names + ; condecls' <- rnConDecls (unLoc tycon') condecls + ; returnM (TyData {tcdND = new_or_data, tcdCtxt = context', tcdLName = tycon', + tcdTyVars = tyvars', tcdKindSig = Nothing, tcdCons = condecls', + tcdDerivs = derivs'}, + delFVs (map hsLTyVarName tyvars') $ + extractHsCtxtTyNames context' `plusFV` + plusFVs (map conDeclFVs condecls') `plusFV` + deriv_fvs) } + + | otherwise -- GADT + = do { tycon' <- lookupLocatedTopBndrRn tycon + ; checkTc (null (unLoc context)) (badGadtStupidTheta tycon) + ; tyvars' <- bindTyVarsRn data_doc tyvars + (\ tyvars' -> return tyvars') + -- For GADTs, the type variables in the declaration + -- do not scope over the constructor signatures + -- data T a where { T1 :: forall b. b-> b } + ; (derivs', deriv_fvs) <- rn_derivs derivs + ; checkDupNames data_doc con_names + ; condecls' <- rnConDecls (unLoc tycon') condecls + ; returnM (TyData {tcdND = new_or_data, tcdCtxt = noLoc [], tcdLName = tycon', + tcdTyVars = tyvars', tcdCons = condecls', tcdKindSig = sig, + tcdDerivs = derivs'}, + plusFVs (map conDeclFVs condecls') `plusFV` deriv_fvs) } + where + is_vanilla = case condecls of -- Yuk + [] -> True + L _ (ConDecl { con_res = ResTyH98 }) : _ -> True + other -> False + data_doc = text "In the data type declaration for" <+> quotes (ppr tycon) - con_names = map conDeclName (visibleDataCons condecls) + con_names = map con_names_helper condecls + + con_names_helper (L _ c) = con_name c - rn_derivs Nothing = returnM Nothing - rn_derivs (Just ds) = rnContext data_doc ds `thenM` \ ds' -> returnM (Just ds') + rn_derivs Nothing = returnM (Nothing, emptyFVs) + rn_derivs (Just ds) = rnLHsTypes data_doc ds `thenM` \ ds' -> + returnM (Just ds', extractHsTyNames_s ds') -rnTyClDecl (TySynonym {tcdName = name, tcdTyVars = tyvars, tcdSynRhs = ty, tcdLoc = src_loc}) - = addSrcLoc src_loc $ - lookupTopBndrRn name `thenM` \ name' -> +rnTyClDecl (TySynonym {tcdLName = name, tcdTyVars = tyvars, tcdSynRhs = ty}) + = lookupLocatedTopBndrRn name `thenM` \ name' -> bindTyVarsRn syn_doc tyvars $ \ tyvars' -> - rnHsType syn_doc ty `thenM` \ ty' -> - returnM (TySynonym {tcdName = name', tcdTyVars = tyvars', tcdSynRhs = ty', tcdLoc = src_loc}) + rnHsTypeFVs syn_doc ty `thenM` \ (ty', fvs) -> + returnM (TySynonym {tcdLName = name', tcdTyVars = tyvars', + tcdSynRhs = ty'}, + delFVs (map hsLTyVarName tyvars') fvs) where syn_doc = text "In the declaration for type synonym" <+> quotes (ppr name) -rnTyClDecl (ClassDecl {tcdCtxt = context, tcdName = cname, +rnTyClDecl (ClassDecl {tcdCtxt = context, tcdLName = cname, tcdTyVars = tyvars, tcdFDs = fds, tcdSigs = sigs, - tcdLoc = src_loc}) - -- Used for both source and interface file decls - = addSrcLoc src_loc $ - - lookupTopBndrRn cname `thenM` \ cname' -> + tcdMeths = mbinds}) + = lookupLocatedTopBndrRn cname `thenM` \ cname' -> -- Tyvars scope over superclass context and method signatures - bindTyVarsRn cls_doc tyvars $ \ tyvars' -> - - -- Check the superclasses - rnContext cls_doc context `thenM` \ context' -> - - -- Check the functional dependencies - rnFds cls_doc fds `thenM` \ fds' -> + bindTyVarsRn cls_doc tyvars ( \ tyvars' -> + rnContext cls_doc context `thenM` \ context' -> + rnFds cls_doc fds `thenM` \ fds' -> + renameSigs okClsDclSig sigs `thenM` \ sigs' -> + returnM (tyvars', context', fds', sigs') + ) `thenM` \ (tyvars', context', fds', sigs') -> -- Check the signatures -- First process the class op sigs (op_sigs), then the fixity sigs (non_op_sigs). let - (op_sigs, non_op_sigs) = partition isClassOpSig sigs - sig_rdr_names_w_locs = [(op,locn) | ClassOpSig op _ _ locn <- sigs] + sig_rdr_names_w_locs = [op | L _ (TypeSig op _) <- sigs] in - checkDupOrQualNames sig_doc sig_rdr_names_w_locs `thenM_` - mappM (rnClassOp cname' fds') op_sigs `thenM` \ sigs' -> - let - binders = mkNameSet [ nm | (ClassOpSig nm _ _ _) <- sigs' ] - in - renameSigs non_op_sigs `thenM` \ non_ops' -> - checkSigs okClsDclSig binders non_ops' `thenM_` + checkDupNames sig_doc sig_rdr_names_w_locs `thenM_` -- Typechecker is responsible for checking that we only -- give default-method bindings for things in this class. -- The renamer *could* check this for class decls, but can't -- for instance decls. - returnM (ClassDecl { tcdCtxt = context', tcdName = cname', tcdTyVars = tyvars', - tcdFDs = fds', tcdSigs = non_ops' ++ sigs', tcdMeths = Nothing, - tcdLoc = src_loc}) - where - cls_doc = text "In the declaration for class" <+> ppr cname - sig_doc = text "In the signatures for class" <+> ppr cname - -rnClassOp clas clas_fds sig@(ClassOpSig op dm_stuff ty locn) - = addSrcLoc locn $ - lookupTopBndrRn op `thenM` \ op_name -> - - -- Check the signature - rnHsSigType (quotes (ppr op)) ty `thenM` \ new_ty -> - - -- Make the default-method name - (case dm_stuff of - DefMeth dm_rdr_name - -> -- Imported class that has a default method decl - lookupSysBndr dm_rdr_name `thenM` \ dm_name -> - returnM (DefMeth dm_name) - -- An imported class decl for a class decl that had an explicit default - -- method, mentions, rather than defines, - -- the default method, so we must arrange to pull it in - - GenDefMeth -> returnM GenDefMeth - NoDefMeth -> returnM NoDefMeth - ) `thenM` \ dm_stuff' -> - - returnM (ClassOpSig op_name dm_stuff' new_ty locn) - -finishSourceTyClDecl :: RdrNameTyClDecl -> RenamedTyClDecl -> RnM (RenamedTyClDecl, FreeVars) - -- Used for source file decls only - -- Renames the default-bindings of a class decl -finishSourceTyClDecl (ClassDecl {tcdMeths = Just mbinds, tcdLoc = src_loc}) -- Get mbinds from here - rn_cls_decl@(ClassDecl {tcdName = cls, tcdTyVars = tyvars}) -- Everything else is here - -- There are some default-method bindings (abeit possibly empty) so - -- this is a source-code class declaration - = -- The newLocals call is tiresome: given a generic class decl + -- The newLocals call is tiresome: given a generic class decl -- class C a where -- op :: a -> a -- op {| x+y |} (Inl a) = ... @@ -607,48 +532,36 @@ finishSourceTyClDecl (ClassDecl {tcdMeths = Just mbinds, tcdLoc = src_loc}) -- G -- op {| a*b |} (a*b) = ... -- we want to name both "x" tyvars with the same unique, so that they are -- easy to group together in the typechecker. - -- Hence the - addSrcLoc src_loc $ - extendTyVarEnvForMethodBinds tyvars $ - getLocalRdrEnv `thenM` \ name_env -> - let - meth_rdr_names_w_locs = collectLocatedMonoBinders mbinds - gen_rdr_tyvars_w_locs = [(tv,src_loc) | tv <- extractGenericPatTyVars mbinds, - not (tv `elemRdrEnv` name_env)] - in - checkDupOrQualNames meth_doc meth_rdr_names_w_locs `thenM_` - newLocalsRn gen_rdr_tyvars_w_locs `thenM` \ gen_tyvars -> - rnMethodBinds cls gen_tyvars mbinds `thenM` \ (mbinds', meth_fvs) -> - returnM (rn_cls_decl {tcdMeths = Just mbinds'}, meth_fvs) + extendTyVarEnvForMethodBinds tyvars' ( + getLocalRdrEnv `thenM` \ name_env -> + let + meth_rdr_names_w_locs = collectHsBindLocatedBinders mbinds + gen_rdr_tyvars_w_locs = + [ tv | tv <- extractGenericPatTyVars mbinds, + not (unLoc tv `elemLocalRdrEnv` name_env) ] + in + checkDupNames meth_doc meth_rdr_names_w_locs `thenM_` + newLocalsRn gen_rdr_tyvars_w_locs `thenM` \ gen_tyvars -> + rnMethodBinds (unLoc cname') gen_tyvars mbinds + ) `thenM` \ (mbinds', meth_fvs) -> + + returnM (ClassDecl { tcdCtxt = context', tcdLName = cname', tcdTyVars = tyvars', + tcdFDs = fds', tcdSigs = sigs', tcdMeths = mbinds'}, + delFVs (map hsLTyVarName tyvars') $ + extractHsCtxtTyNames context' `plusFV` + plusFVs (map extractFunDepNames (map unLoc fds')) `plusFV` + hsSigsFVs sigs' `plusFV` + meth_fvs) where - meth_doc = text "In the default-methods for class" <+> ppr (tcdName rn_cls_decl) - -finishSourceTyClDecl _ tycl_decl@(TyData {tcdDerivs = derivings}) - -- Derivings are returned here so that they don't form part of the tyClDeclFVs. - -- This is important, because tyClDeclFVs should contain only the - -- FVs that are `needed' by the interface file declaration, and - -- derivings do not appear in this. It also means that the tcGroups - -- are smaller, which turned out to be important for the usage inference. KSW 2002-02. - = returnM (tycl_decl, - maybe emptyFVs extractHsCtxtTyNames derivings) - -finishSourceTyClDecl _ tycl_decl = returnM (tycl_decl, emptyFVs) - -- Not a class declaration -\end{code} - -For the method bindings in class and instance decls, we extend the -type variable environment iff -fglasgow-exts + meth_doc = text "In the default-methods for class" <+> ppr cname + cls_doc = text "In the declaration for class" <+> ppr cname + sig_doc = text "In the signatures for class" <+> ppr cname -\begin{code} -extendTyVarEnvForMethodBinds tyvars thing_inside - = doptM Opt_GlasgowExts `thenM` \ opt_GlasgowExts -> - if opt_GlasgowExts then - extendTyVarEnvFVRn (map hsTyVarName tyvars) thing_inside - else - thing_inside +badGadtStupidTheta tycon + = vcat [ptext SLIT("No context is allowed on a GADT-style data declaration"), + ptext SLIT("(You can put a context on each contructor, though.)")] \end{code} - %********************************************************* %* * \subsection{Support code for type/data declarations} @@ -656,64 +569,67 @@ extendTyVarEnvForMethodBinds tyvars thing_inside %********************************************************* \begin{code} -conDeclName :: RdrNameConDecl -> (RdrName, SrcLoc) -conDeclName (ConDecl n _ _ _ l) = (n,l) - -rnConDecls :: Name -> DataConDetails RdrNameConDecl -> RnM (DataConDetails RenamedConDecl) -rnConDecls tycon Unknown = returnM Unknown -rnConDecls tycon (HasCons n) = returnM (HasCons n) -rnConDecls tycon (DataCons condecls) - = -- Check that there's at least one condecl, - -- or else we're reading an interface file, or -fglasgow-exts - (if null condecls then - doptM Opt_GlasgowExts `thenM` \ glaExts -> - getModeRn `thenM` \ mode -> - checkErr (glaExts || isInterfaceMode mode) - (emptyConDeclsErr tycon) - else returnM () - ) `thenM_` - - mappM rnConDecl condecls `thenM` \ condecls' -> - returnM (DataCons condecls') - -rnConDecl :: RdrNameConDecl -> RnM RenamedConDecl -rnConDecl (ConDecl name tvs cxt details locn) - = addSrcLoc locn $ - checkConName name `thenM_` - lookupTopBndrRn name `thenM` \ new_name -> - - bindTyVarsRn doc tvs $ \ new_tyvars -> - rnContext doc cxt `thenM` \ new_context -> - rnConDetails doc locn details `thenM` \ new_details -> - returnM (ConDecl new_name new_tyvars new_context new_details locn) +rnConDecls :: Name -> [LConDecl RdrName] -> RnM [LConDecl Name] +rnConDecls tycon condecls + = mappM (wrapLocM rnConDecl) condecls + +rnConDecl :: ConDecl RdrName -> RnM (ConDecl Name) +rnConDecl (ConDecl name expl tvs cxt details res_ty) + = do { addLocM checkConName name + + ; new_name <- lookupLocatedTopBndrRn name + ; name_env <- getLocalRdrEnv + + -- For H98 syntax, the tvs are the existential ones + -- For GADT syntax, the tvs are all the quantified tyvars + -- Hence the 'filter' in the ResTyH98 case only + ; let not_in_scope = not . (`elemLocalRdrEnv` name_env) . unLoc + arg_tys = hsConArgs details + implicit_tvs = case res_ty of + ResTyH98 -> filter not_in_scope $ + get_rdr_tvs arg_tys + ResTyGADT ty -> get_rdr_tvs (ty : arg_tys) + tvs' = case expl of + Explicit -> tvs + Implicit -> userHsTyVarBndrs implicit_tvs + + ; bindTyVarsRn doc tvs' $ \new_tyvars -> do + { new_context <- rnContext doc cxt + ; new_details <- rnConDetails doc details + ; new_res_ty <- rnConResult doc res_ty + ; let rv = ConDecl new_name expl new_tyvars new_context new_details new_res_ty + ; traceRn (text "****** - autrijus" <> ppr rv) + ; return rv } } where doc = text "In the definition of data constructor" <+> quotes (ppr name) + get_rdr_tvs tys = extractHsRhoRdrTyVars cxt (noLoc (HsTupleTy Boxed tys)) -rnConDetails doc locn (PrefixCon tys) - = mappM (rnBangTy doc) tys `thenM` \ new_tys -> +rnConResult _ ResTyH98 = return ResTyH98 +rnConResult doc (ResTyGADT ty) = do + ty' <- rnHsSigType doc ty + return $ ResTyGADT ty' + +rnConDetails doc (PrefixCon tys) + = mappM (rnLHsType doc) tys `thenM` \ new_tys -> returnM (PrefixCon new_tys) -rnConDetails doc locn (InfixCon ty1 ty2) - = rnBangTy doc ty1 `thenM` \ new_ty1 -> - rnBangTy doc ty2 `thenM` \ new_ty2 -> +rnConDetails doc (InfixCon ty1 ty2) + = rnLHsType doc ty1 `thenM` \ new_ty1 -> + rnLHsType doc ty2 `thenM` \ new_ty2 -> returnM (InfixCon new_ty1 new_ty2) -rnConDetails doc locn (RecCon fields) - = checkDupOrQualNames doc field_names `thenM_` +rnConDetails doc (RecCon fields) + = checkDupNames doc field_names `thenM_` mappM (rnField doc) fields `thenM` \ new_fields -> returnM (RecCon new_fields) where - field_names = [(fld, locn) | (fld, _) <- fields] + field_names = [fld | (fld, _) <- fields] rnField doc (name, ty) - = lookupTopBndrRn name `thenM` \ new_name -> - rnBangTy doc ty `thenM` \ new_ty -> + = lookupLocatedTopBndrRn name `thenM` \ new_name -> + rnLHsType doc ty `thenM` \ new_ty -> returnM (new_name, new_ty) -rnBangTy doc (BangType s ty) - = rnHsType doc ty `thenM` \ new_ty -> - returnM (BangType s new_ty) - -- This data decl will parse OK -- data T = a Int -- treating "a" as the constructor. @@ -724,8 +640,10 @@ rnBangTy doc (BangType s ty) -- data T = :% Int Int -- from interface files, which always print in prefix form -checkConName name - = checkErr (isRdrDataCon name) (badDataCon name) +checkConName name = checkErr (isRdrDataCon name) (badDataCon name) + +badDataCon name + = hsep [ptext SLIT("Illegal data constructor name"), quotes (ppr name)] \end{code} @@ -736,10 +654,10 @@ checkConName name %********************************************************* \begin{code} -rnFds :: SDoc -> [FunDep RdrName] -> RnM [FunDep Name] +rnFds :: SDoc -> [Located (FunDep RdrName)] -> RnM [Located (FunDep Name)] rnFds doc fds - = mappM rn_fds fds + = mappM (wrapLocM rn_fds) fds where rn_fds (tys1, tys2) = rnHsTyVars doc tys1 `thenM` \ tys1' -> @@ -750,214 +668,55 @@ rnHsTyVars doc tvs = mappM (rnHsTyvar doc) tvs rnHsTyvar doc tyvar = lookupOccRn tyvar \end{code} -%********************************************************* -%* * -\subsection{IdInfo} -%* * -%********************************************************* - -\begin{code} -rnIdInfo (HsWorker worker arity) - = lookupOccRn worker `thenM` \ worker' -> - returnM (HsWorker worker' arity) - -rnIdInfo (HsUnfold inline expr) = rnCoreExpr expr `thenM` \ expr' -> - returnM (HsUnfold inline expr') -rnIdInfo (HsStrictness str) = returnM (HsStrictness str) -rnIdInfo (HsArity arity) = returnM (HsArity arity) -rnIdInfo HsNoCafRefs = returnM HsNoCafRefs -\end{code} - -@UfCore@ expressions. - -\begin{code} -rnCoreExpr (UfType ty) - = rnHsType (text "unfolding type") ty `thenM` \ ty' -> - returnM (UfType ty') - -rnCoreExpr (UfVar v) - = lookupOccRn v `thenM` \ v' -> - returnM (UfVar v') - -rnCoreExpr (UfLit l) - = returnM (UfLit l) - -rnCoreExpr (UfLitLit l ty) - = rnHsType (text "litlit") ty `thenM` \ ty' -> - returnM (UfLitLit l ty') - -rnCoreExpr (UfFCall cc ty) - = rnHsType (text "ccall") ty `thenM` \ ty' -> - returnM (UfFCall cc ty') - -rnCoreExpr (UfTuple (HsTupCon boxity arity) args) - = mappM rnCoreExpr args `thenM` \ args' -> - returnM (UfTuple (HsTupCon boxity arity) args') - -rnCoreExpr (UfApp fun arg) - = rnCoreExpr fun `thenM` \ fun' -> - rnCoreExpr arg `thenM` \ arg' -> - returnM (UfApp fun' arg') - -rnCoreExpr (UfCase scrut bndr alts) - = rnCoreExpr scrut `thenM` \ scrut' -> - bindCoreLocalRn bndr $ \ bndr' -> - mappM rnCoreAlt alts `thenM` \ alts' -> - returnM (UfCase scrut' bndr' alts') - -rnCoreExpr (UfNote note expr) - = rnNote note `thenM` \ note' -> - rnCoreExpr expr `thenM` \ expr' -> - returnM (UfNote note' expr') - -rnCoreExpr (UfLam bndr body) - = rnCoreBndr bndr $ \ bndr' -> - rnCoreExpr body `thenM` \ body' -> - returnM (UfLam bndr' body') - -rnCoreExpr (UfLet (UfNonRec bndr rhs) body) - = rnCoreExpr rhs `thenM` \ rhs' -> - rnCoreBndr bndr $ \ bndr' -> - rnCoreExpr body `thenM` \ body' -> - returnM (UfLet (UfNonRec bndr' rhs') body') - -rnCoreExpr (UfLet (UfRec pairs) body) - = rnCoreBndrs bndrs $ \ bndrs' -> - mappM rnCoreExpr rhss `thenM` \ rhss' -> - rnCoreExpr body `thenM` \ body' -> - returnM (UfLet (UfRec (bndrs' `zip` rhss')) body') - where - (bndrs, rhss) = unzip pairs -\end{code} - -\begin{code} -rnCoreBndr (UfValBinder name ty) thing_inside - = rnHsType doc ty `thenM` \ ty' -> - bindCoreLocalRn name $ \ name' -> - thing_inside (UfValBinder name' ty') - where - doc = text "unfolding id" - -rnCoreBndr (UfTyBinder name kind) thing_inside - = bindCoreLocalRn name $ \ name' -> - thing_inside (UfTyBinder name' kind) - -rnCoreBndrs [] thing_inside = thing_inside [] -rnCoreBndrs (b:bs) thing_inside = rnCoreBndr b $ \ name' -> - rnCoreBndrs bs $ \ names' -> - thing_inside (name':names') -\end{code} - -\begin{code} -rnCoreAlt (con, bndrs, rhs) - = rnUfCon con `thenM` \ con' -> - bindCoreLocalsRn bndrs $ \ bndrs' -> - rnCoreExpr rhs `thenM` \ rhs' -> - returnM (con', bndrs', rhs') - -rnNote (UfCoerce ty) - = rnHsType (text "unfolding coerce") ty `thenM` \ ty' -> - returnM (UfCoerce ty') - -rnNote (UfSCC cc) = returnM (UfSCC cc) -rnNote UfInlineCall = returnM UfInlineCall -rnNote UfInlineMe = returnM UfInlineMe - - -rnUfCon UfDefault - = returnM UfDefault - -rnUfCon (UfTupleAlt tup_con) - = returnM (UfTupleAlt tup_con) - -rnUfCon (UfDataAlt con) - = lookupOccRn con `thenM` \ con' -> - returnM (UfDataAlt con') - -rnUfCon (UfLitAlt lit) - = returnM (UfLitAlt lit) - -rnUfCon (UfLitLitAlt lit ty) - = rnHsType (text "litlit") ty `thenM` \ ty' -> - returnM (UfLitLitAlt lit ty') -\end{code} %********************************************************* %* * -\subsection{Statistics} + Splices %* * %********************************************************* -\begin{code} -rnStats :: [RenamedHsDecl] -- Imported decls - -> TcRn m () -rnStats imp_decls - = doptM Opt_D_dump_rn_trace `thenM` \ dump_rn_trace -> - doptM Opt_D_dump_rn_stats `thenM` \ dump_rn_stats -> - doptM Opt_D_dump_rn `thenM` \ dump_rn -> - getEps `thenM` \ eps -> - - ioToTcRn (dumpIfSet (dump_rn_trace || dump_rn_stats || dump_rn) - "Renamer statistics" - (getRnStats eps imp_decls)) `thenM_` - returnM () - -getRnStats :: ExternalPackageState -> [RenamedHsDecl] -> SDoc -getRnStats eps imported_decls - = hcat [text "Renamer stats: ", stats] - where - n_mods = length [() | _ <- moduleEnvElts (eps_PIT eps)] - -- This is really only right for a one-shot compile - - (decls_map, n_decls_slurped) = eps_decls eps - - n_decls_left = length [decl | (avail, True, (_,decl)) <- nameEnvElts decls_map - -- Data, newtype, and class decls are in the decls_fm - -- under multiple names; the tycon/class, and each - -- constructor/class op too. - -- The 'True' selects just the 'main' decl - ] - - (insts_left, n_insts_slurped) = eps_insts eps - n_insts_left = length (bagToList insts_left) - - (rules_left, n_rules_slurped) = eps_rules eps - n_rules_left = length (bagToList rules_left) - - stats = vcat - [int n_mods <+> text "interfaces read", - hsep [ int n_decls_slurped, text "type/class/variable imported, out of", - int (n_decls_slurped + n_decls_left), text "read"], - hsep [ int n_insts_slurped, text "instance decls imported, out of", - int (n_insts_slurped + n_insts_left), text "read"], - hsep [ int n_rules_slurped, text "rule decls imported, out of", - int (n_rules_slurped + n_rules_left), text "read"] - ] -\end{code} - -%********************************************************* -%* * -\subsection{Errors} -%* * -%********************************************************* +Note [Splices] +~~~~~~~~~~~~~~ +Consider + f = ... + h = ...$(thing "f")... + +The splice can expand into literally anything, so when we do dependency +analysis we must assume that it might mention 'f'. So we simply treat +all locally-defined names as mentioned by any splice. This is terribly +brutal, but I don't see what else to do. For example, it'll mean +that every locally-defined thing will appear to be used, so no unused-binding +warnings. But if we miss the dependency, then we might typecheck 'h' before 'f', +and that will crash the type checker because 'f' isn't in scope. + +Currently, I'm not treating a splice as also mentioning every import, +which is a bit inconsistent -- but there are a lot of them. We might +thereby get some bogus unused-import warnings, but we won't crash the +type checker. Not very satisfactory really. \begin{code} -badDataCon name - = hsep [ptext SLIT("Illegal data constructor name"), quotes (ppr name)] - -badRuleLhsErr name lhs (Just bad_e) - = sep [ptext SLIT("Rule") <+> ftext name <> colon, - nest 4 (vcat [ptext SLIT("Illegal expression:") <+> ppr bad_e, - ptext SLIT("in left-hand side:") <+> ppr lhs])] - $$ - ptext SLIT("LHS must be of form (f e1 .. en) where f is not forall'd") - -badRuleVar name var - = sep [ptext SLIT("Rule") <+> doubleQuotes (ftext name) <> colon, - ptext SLIT("Forall'd variable") <+> quotes (ppr var) <+> - ptext SLIT("does not appear on left hand side")] - -emptyConDeclsErr tycon - = sep [quotes (ppr tycon) <+> ptext SLIT("has no constructors"), - nest 4 (ptext SLIT("(-fglasgow-exts permits this)"))] +rnSplice :: HsSplice RdrName -> RnM (HsSplice Name, FreeVars) +rnSplice (HsSplice n expr) + = do { checkTH expr "splice" + ; loc <- getSrcSpanM + ; [n'] <- newLocalsRn [L loc n] + ; (expr', fvs) <- rnLExpr expr + + -- Ugh! See Note [Splices] above + ; lcl_rdr <- getLocalRdrEnv + ; gbl_rdr <- getGlobalRdrEnv + ; let gbl_names = mkNameSet [gre_name gre | gre <- globalRdrEnvElts gbl_rdr, + isLocalGRE gre] + lcl_names = mkNameSet (occEnvElts lcl_rdr) + + ; return (HsSplice n' expr', fvs `plusFV` lcl_names `plusFV` gbl_names) } + +#ifdef GHCI +checkTH e what = returnM () -- OK +#else +checkTH e what -- Raise an error in a stage-1 compiler + = addErr (vcat [ptext SLIT("Template Haskell") <+> text what <+> + ptext SLIT("illegal in a stage-1 compiler"), + nest 2 (ppr e)]) +#endif \end{code}