Skip to content
GitLab
Commit e8687b3c authored by twanvl's avatar twanvl
Browse files

Monadify rename/RnBinds: use do, return and standard monad functions

parent b9e1d789
Hide whitespace changes
Inline Side-by-side
compiler/rename/RnBinds.lhs
View file @ e8687b3c
......@@ -62,7 +62,7 @@ import Bag
import Outputable
import Maybes ( orElse )
import Util ( filterOut )
import Monad ( foldM )
import Monad ( foldM, unless )
\end{code}
-- ToDo: Put the annotations into the monad, so that they arrive in the proper
......@@ -235,19 +235,19 @@ rnLocalBindsAndThen (HsValBinds val_binds) thing_inside
= rnValBindsAndThen val_binds $ \ val_binds' ->
thing_inside (HsValBinds val_binds')
rnLocalBindsAndThen (HsIPBinds binds) thing_inside
= rnIPBinds binds `thenM` \ (binds',fv_binds) ->
thing_inside (HsIPBinds binds') `thenM` \ (thing, fvs_thing) ->
returnM (thing, fvs_thing `plusFV` fv_binds)
rnLocalBindsAndThen (HsIPBinds binds) thing_inside = do
(binds',fv_binds) <- rnIPBinds binds
(thing, fvs_thing) <- thing_inside (HsIPBinds binds')
return (thing, fvs_thing `plusFV` fv_binds)
rnIPBinds (IPBinds ip_binds _no_dict_binds)
= do { (ip_binds', fvs_s) <- mapAndUnzipM (wrapLocFstM rnIPBind) ip_binds
; return (IPBinds ip_binds' emptyLHsBinds, plusFVs fvs_s) }
rnIPBinds (IPBinds ip_binds _no_dict_binds) = do
(ip_binds', fvs_s) <- mapAndUnzipM (wrapLocFstM rnIPBind) ip_binds
return (IPBinds ip_binds' emptyLHsBinds, plusFVs fvs_s)
rnIPBind (IPBind n expr)
= newIPNameRn n `thenM` \ name ->
rnLExpr expr `thenM` \ (expr',fvExpr) ->
rnIPBind (IPBind n expr) = do
name <- newIPNameRn n
(expr',fvExpr) <- rnLExpr expr
return (IPBind name expr', fvExpr)
\end{code}
......@@ -313,8 +313,7 @@ rnValBindsLHSFromDoc :: NameMaker
-> SDoc -- doc string for dup names and shadowing
-> HsValBinds RdrName
-> RnM (HsValBindsLR Name RdrName)
rnValBindsLHSFromDoc topP original_bndrs doc binds@(ValBindsIn mbinds sigs)
= do
rnValBindsLHSFromDoc topP original_bndrs doc binds@(ValBindsIn mbinds sigs) = do
-- rename the LHSes
mbinds' <- mapBagM (rnBindLHS topP doc) mbinds
return $ ValBindsIn mbinds' sigs
......@@ -330,8 +329,8 @@ rnValBindsRHSGen :: (FreeVars -> FreeVars) -- for trimming free var sets
-> HsValBindsLR Name RdrName
-> RnM (HsValBinds Name, DefUses)
rnValBindsRHSGen trim bound_names binds@(ValBindsIn mbinds sigs)
= do -- rename the sigs
rnValBindsRHSGen trim bound_names binds@(ValBindsIn mbinds sigs) = do
-- rename the sigs
sigs' <- rename_sigs sigs
-- rename the RHSes
binds_w_dus <- mapBagM (rnBind (mkSigTvFn sigs') trim) mbinds
......@@ -341,7 +340,7 @@ rnValBindsRHSGen trim bound_names binds@(ValBindsIn mbinds sigs)
-- We do the check-sigs after renaming the bindings,
-- so that we have convenient access to the binders
check_sigs (okBindSig (duDefs anal_dus)) sigs'
returnM (valbind', valbind'_dus)
return (valbind', valbind'_dus)
-- wrapper for local binds
--
......@@ -628,24 +627,24 @@ rnMethodBinds cls sig_fn gen_tyvars binds
rnMethodBind cls sig_fn gen_tyvars (L loc (FunBind { fun_id = name, fun_infix = inf,
fun_matches = MatchGroup matches _ }))
= setSrcSpan loc $
lookupInstDeclBndr cls name `thenM` \ sel_name ->
let plain_name = unLoc sel_name in
-- We use the selector name as the binder
bindSigTyVarsFV (sig_fn plain_name) $
mapFvRn (rn_match plain_name) matches `thenM` \ (new_matches, fvs) ->
let
new_group = MatchGroup new_matches placeHolderType
in
checkPrecMatch inf plain_name new_group `thenM_`
returnM (unitBag (L loc (FunBind {
fun_id = sel_name, fun_infix = inf,
fun_matches = new_group,
bind_fvs = fvs, fun_co_fn = idHsWrapper,
fun_tick = Nothing })),
fvs `addOneFV` plain_name)
-- The 'fvs' field isn't used for method binds
= setSrcSpan loc $ do
sel_name <- lookupInstDeclBndr cls name
let plain_name = unLoc sel_name
-- We use the selector name as the binder
bindSigTyVarsFV (sig_fn plain_name) $ do
(new_matches, fvs) <- mapFvRn (rn_match plain_name) matches
let
new_group = MatchGroup new_matches placeHolderType
checkPrecMatch inf plain_name new_group
return (unitBag (L loc (FunBind {
fun_id = sel_name, fun_infix = inf,
fun_matches = new_group,
bind_fvs = fvs, fun_co_fn = idHsWrapper,
fun_tick = Nothing })),
fvs `addOneFV` plain_name)
-- The 'fvs' field isn't used for method binds
where
-- Truly gruesome; bring into scope the correct members of the generic
-- type variables. See comments in RnSource.rnSourceDecl(ClassDecl)
......@@ -660,9 +659,9 @@ rnMethodBind cls sig_fn gen_tyvars (L loc (FunBind { fun_id = name, fun_infix =
-- Can't handle method pattern-bindings which bind multiple methods.
rnMethodBind cls sig_fn gen_tyvars mbind@(L loc (PatBind other_pat _ _ _))
= addLocErr mbind methodBindErr `thenM_`
returnM (emptyBag, emptyFVs)
rnMethodBind cls sig_fn gen_tyvars mbind@(L loc (PatBind other_pat _ _ _)) = do
addLocErr mbind methodBindErr
return (emptyBag, emptyFVs)
\end{code}
......@@ -686,25 +685,24 @@ signatures. We'd only need this if we wanted to report unused tyvars.
\begin{code}
renameSigs :: (LSig Name -> Bool) -> [LSig RdrName] -> RnM [LSig Name]
-- Renames the signatures and performs error checks
renameSigs ok_sig sigs
renameSigs ok_sig sigs
= do { sigs' <- rename_sigs sigs
; check_sigs ok_sig sigs'
; return sigs' }
----------------------
rename_sigs :: [LSig RdrName] -> RnM [LSig Name]
rename_sigs sigs = mappM (wrapLocM renameSig) sigs
rename_sigs sigs = mapM (wrapLocM renameSig) sigs
----------------------
check_sigs :: (LSig Name -> Bool) -> [LSig Name] -> RnM ()
-- Used for class and instance decls, as well as regular bindings
check_sigs ok_sig sigs
check_sigs ok_sig sigs = do
-- Check for (a) duplicate signatures
-- (b) signatures for things not in this group
= do {
traceRn (text "SIGS" <+> ppr sigs)
; mappM_ unknownSigErr (filter (not . ok_sig) sigs')
; mappM_ dupSigDeclErr (findDupsEq eqHsSig sigs') }
-- (b) signatures for things not in this group = do
traceRn (text "SIGS" <+> ppr sigs)
mapM_ unknownSigErr (filter (not . ok_sig) sigs')
mapM_ dupSigDeclErr (findDupsEq eqHsSig sigs')
where
-- Don't complain about an unbound name again
sigs' = filterOut bad_name sigs
......@@ -723,27 +721,27 @@ check_sigs ok_sig sigs
renameSig :: Sig RdrName -> RnM (Sig Name)
-- FixitSig is renamed elsewhere.
renameSig (TypeSig v ty)
= lookupLocatedSigOccRn v `thenM` \ new_v ->
rnHsSigType (quotes (ppr v)) ty `thenM` \ new_ty ->
returnM (TypeSig new_v new_ty)
renameSig (SpecInstSig ty)
= rnLHsType (text "A SPECIALISE instance pragma") ty `thenM` \ new_ty ->
returnM (SpecInstSig new_ty)
renameSig (SpecSig v ty inl)
= lookupLocatedSigOccRn v `thenM` \ new_v ->
rnHsSigType (quotes (ppr v)) ty `thenM` \ new_ty ->
returnM (SpecSig new_v new_ty inl)
renameSig (InlineSig v s)
= lookupLocatedSigOccRn v `thenM` \ new_v ->
returnM (InlineSig new_v s)
renameSig (FixSig (FixitySig v f))
= lookupLocatedSigOccRn v `thenM` \ new_v ->
returnM (FixSig (FixitySig new_v f))
renameSig (TypeSig v ty) = do
new_v <- lookupLocatedSigOccRn v
new_ty <- rnHsSigType (quotes (ppr v)) ty
return (TypeSig new_v new_ty)
renameSig (SpecInstSig ty) = do
new_ty <- rnLHsType (text "A SPECIALISE instance pragma") ty
return (SpecInstSig new_ty)
renameSig (SpecSig v ty inl) = do
new_v <- lookupLocatedSigOccRn v
new_ty <- rnHsSigType (quotes (ppr v)) ty
return (SpecSig new_v new_ty inl)
renameSig (InlineSig v s) = do
new_v <- lookupLocatedSigOccRn v
return (InlineSig new_v s)
renameSig (FixSig (FixitySig v f)) = do
new_v <- lookupLocatedSigOccRn v
return (FixSig (FixitySig new_v f))
\end{code}
......@@ -755,9 +753,9 @@ renameSig (FixSig (FixitySig v f))
\begin{code}
rnMatchGroup :: HsMatchContext Name -> MatchGroup RdrName -> RnM (MatchGroup Name, FreeVars)
rnMatchGroup ctxt (MatchGroup ms _)
= mapFvRn (rnMatch ctxt) ms `thenM` \ (new_ms, ms_fvs) ->
returnM (MatchGroup new_ms placeHolderType, ms_fvs)
rnMatchGroup ctxt (MatchGroup ms _) = do
(new_ms, ms_fvs) <- mapFvRn (rnMatch ctxt) ms
return (MatchGroup new_ms placeHolderType, ms_fvs)
rnMatch :: HsMatchContext Name -> LMatch RdrName -> RnM (LMatch Name, FreeVars)
rnMatch ctxt = wrapLocFstM (rnMatch' ctxt)
......@@ -765,22 +763,22 @@ rnMatch ctxt = wrapLocFstM (rnMatch' ctxt)
rnMatch' ctxt match@(Match pats maybe_rhs_sig grhss)
=
-- Deal with the rhs type signature
bindPatSigTyVarsFV rhs_sig_tys $
doptM Opt_PatternSignatures `thenM` \ opt_PatternSignatures ->
(case maybe_rhs_sig of
Nothing -> returnM (Nothing, emptyFVs)
Just ty | opt_PatternSignatures -> rnHsTypeFVs doc_sig ty `thenM` \ (ty', ty_fvs) ->
returnM (Just ty', ty_fvs)
| otherwise -> addLocErr ty patSigErr `thenM_`
returnM (Nothing, emptyFVs)
) `thenM` \ (maybe_rhs_sig', ty_fvs) ->
-- Now the main event
bindPatSigTyVarsFV rhs_sig_tys $ do
opt_PatternSignatures <- doptM Opt_PatternSignatures
(maybe_rhs_sig', ty_fvs) <-
case maybe_rhs_sig of
Nothing -> return (Nothing, emptyFVs)
Just ty | opt_PatternSignatures -> do (ty', ty_fvs) <- rnHsTypeFVs doc_sig ty
return (Just ty', ty_fvs)
| otherwise -> do addLocErr ty patSigErr
return (Nothing, emptyFVs)
-- Now the main event
-- note that there are no local ficity decls for matches
rnPatsAndThen_LocalRightwards ctxt pats $ \ pats' ->
rnGRHSs ctxt grhss `thenM` \ (grhss', grhss_fvs) ->
rnPatsAndThen_LocalRightwards ctxt pats $ \ pats' -> do
(grhss', grhss_fvs) <- rnGRHSs ctxt grhss
returnM (Match pats' maybe_rhs_sig' grhss', grhss_fvs `plusFV` ty_fvs)
return (Match pats' maybe_rhs_sig' grhss', grhss_fvs `plusFV` ty_fvs)
-- The bindPatSigTyVarsFV and rnPatsAndThen will remove the bound FVs
where
rhs_sig_tys = case maybe_rhs_sig of
......@@ -800,9 +798,9 @@ rnMatch' ctxt match@(Match pats maybe_rhs_sig grhss)
rnGRHSs :: HsMatchContext Name -> GRHSs RdrName -> RnM (GRHSs Name, FreeVars)
rnGRHSs ctxt (GRHSs grhss binds)
= rnLocalBindsAndThen binds $ \ binds' ->
mapFvRn (rnGRHS ctxt) grhss `thenM` \ (grhss', fvGRHSs) ->
returnM (GRHSs grhss' binds', fvGRHSs)
= rnLocalBindsAndThen binds $ \ binds' -> do
(grhss', fvGRHSs) <- mapFvRn (rnGRHS ctxt) grhss
return (GRHSs grhss' binds', fvGRHSs)
rnGRHS :: HsMatchContext Name -> LGRHS RdrName -> RnM (LGRHS Name, FreeVars)
rnGRHS ctxt = wrapLocFstM (rnGRHS' ctxt)
......@@ -812,7 +810,7 @@ rnGRHS' ctxt (GRHS guards rhs)
; ((guards', rhs'), fvs) <- rnStmts (PatGuard ctxt) guards $
rnLExpr rhs
; checkM (pattern_guards_allowed || is_standard_guard guards')
; unless (pattern_guards_allowed || is_standard_guard guards')
(addWarn (nonStdGuardErr guards'))
; return (GRHS guards' rhs', fvs) }
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment