2 % (c) The AQUA Project, Glasgow University, 1996-1998
4 %************************************************************************
6 \section[PprCore]{Printing of Core syntax, including for interfaces}
8 %************************************************************************
12 pprCoreExpr, pprParendExpr,
13 pprCoreBinding, pprCoreBindings, pprCoreAlt,
17 #include "HsVersions.h"
20 import CostCentre ( pprCostCentreCore )
22 import Id ( Id, idType, isDataConWorkId_maybe, idLBVarInfo, idArity,
23 idInfo, idInlinePragma, idOccInfo,
24 globalIdDetails, isGlobalId, isExportedId,
25 isSpecPragmaId, idNewDemandInfo
27 import Var ( TyVar, isTyVar, tyVarKind )
28 import IdInfo ( IdInfo, megaSeqIdInfo,
29 arityInfo, ppArityInfo,
30 specInfo, pprNewStrictness,
31 workerInfo, ppWorkerInfo,
32 newStrictnessInfo, cafInfo, ppCafInfo,
36 import Id ( idDemandInfo )
37 import IdInfo ( cprInfo, ppCprInfo, strictnessInfo, ppStrictnessInfo )
40 import DataCon ( dataConTyCon )
41 import TyCon ( tupleTyConBoxity, isTupleTyCon )
42 import Type ( pprParendType, pprType, pprParendKind )
43 import BasicTypes ( tupleParens )
44 import Util ( lengthIs )
46 import FastString ( mkFastString )
49 %************************************************************************
51 \subsection{Public interfaces for Core printing (excluding instances)}
53 %************************************************************************
55 @pprParendCoreExpr@ puts parens around non-atomic Core expressions.
58 pprCoreBindings :: OutputableBndr b => [Bind b] -> SDoc
59 pprCoreBinding :: OutputableBndr b => Bind b -> SDoc
60 pprCoreExpr :: OutputableBndr b => Expr b -> SDoc
61 pprParendExpr :: OutputableBndr b => Expr b -> SDoc
63 pprCoreBindings = pprTopBinds
64 pprCoreBinding = pprTopBind
66 instance OutputableBndr b => Outputable (Bind b) where
67 ppr bind = ppr_bind bind
69 instance OutputableBndr b => Outputable (Expr b) where
70 ppr expr = pprCoreExpr expr
74 %************************************************************************
78 %************************************************************************
81 pprTopBinds binds = vcat (map pprTopBind binds)
83 pprTopBind (NonRec binder expr)
84 = ppr_binding (binder,expr) $$ text ""
86 pprTopBind (Rec binds)
87 = vcat [ptext SLIT("Rec {"),
88 vcat (map ppr_binding binds),
89 ptext SLIT("end Rec }"),
94 ppr_bind :: OutputableBndr b => Bind b -> SDoc
96 ppr_bind (NonRec val_bdr expr) = ppr_binding (val_bdr, expr)
97 ppr_bind (Rec binds) = vcat (map pp binds)
99 pp bind = ppr_binding bind <> semi
101 ppr_binding :: OutputableBndr b => (b, Expr b) -> SDoc
102 ppr_binding (val_bdr, expr)
103 = pprBndr LetBind val_bdr $$
104 (ppr val_bdr <+> equals <+> pprCoreExpr expr)
108 pprParendExpr expr = ppr_expr parens expr
109 pprCoreExpr expr = ppr_expr noParens expr
111 noParens :: SDoc -> SDoc
116 ppr_expr :: OutputableBndr b => (SDoc -> SDoc) -> Expr b -> SDoc
117 -- The function adds parens in context that need
118 -- an atomic value (e.g. function args)
120 ppr_expr add_par (Type ty) = add_par (ptext SLIT("TYPE") <+> ppr ty) -- Wierd
122 ppr_expr add_par (Var name) = ppr name
123 ppr_expr add_par (Lit lit) = ppr lit
125 ppr_expr add_par expr@(Lam _ _)
127 (bndrs, body) = collectBinders expr
130 hang (ptext SLIT("\\") <+> sep (map (pprBndr LambdaBind) bndrs) <+> arrow)
133 ppr_expr add_par expr@(App fun arg)
134 = case collectArgs expr of { (fun, args) ->
136 pp_args = sep (map pprArg args)
137 val_args = dropWhile isTypeArg args -- Drop the type arguments for tuples
138 pp_tup_args = sep (punctuate comma (map pprArg val_args))
141 Var f -> case isDataConWorkId_maybe f of
142 -- Notice that we print the *worker*
143 -- for tuples in paren'd format.
144 Just dc | saturated && isTupleTyCon tc
145 -> tupleParens (tupleTyConBoxity tc) pp_tup_args
148 saturated = val_args `lengthIs` idArity f
150 other -> add_par (hang (ppr f) 2 pp_args)
152 other -> add_par (hang (pprParendExpr fun) 2 pp_args)
155 ppr_expr add_par (Case expr var ty [(con,args,rhs)])
157 sep [sep [ptext SLIT("case") <+> pprParendType ty <+> pprCoreExpr expr,
158 hsep [ptext SLIT("of"),
161 ppr_case_pat con args
167 ppr_bndr = pprBndr CaseBind
169 ppr_expr add_par (Case expr var ty alts)
171 sep [sep [ptext SLIT("case") <+> pprParendType ty <+> pprCoreExpr expr,
172 ptext SLIT("of") <+> ppr_bndr var <+> char '{'],
173 nest 2 (sep (punctuate semi (map pprCoreAlt alts))),
177 ppr_bndr = pprBndr CaseBind
180 -- special cases: let ... in let ...
181 -- ("disgusting" SLPJ)
184 ppr_expr add_par (Let bind@(NonRec val_bdr rhs@(Let _ _)) body)
187 hsep [ptext SLIT("let {"), (pprBndr LetBind val_bdr $$ ppr val_bndr), equals],
188 nest 2 (pprCoreExpr rhs),
193 ppr_expr add_par (Let bind@(NonRec val_bdr rhs) expr@(Let _ _))
195 (hang (ptext SLIT("let {"))
196 2 (hsep [ppr_binding (val_bdr,rhs),
201 -- general case (recursive case, too)
202 ppr_expr add_par (Let bind expr)
204 sep [hang (ptext keyword) 2 (ppr_bind bind),
205 hang (ptext SLIT("} in ")) 2 (pprCoreExpr expr)]
207 keyword = case bind of
208 Rec _ -> SLIT("__letrec {")
209 NonRec _ _ -> SLIT("let {")
211 ppr_expr add_par (Note (SCC cc) expr)
212 = add_par (sep [pprCostCentreCore cc, pprCoreExpr expr])
215 ppr_expr add_par (Note (Coerce to_ty from_ty) expr)
217 getPprStyle $ \ sty ->
218 if debugStyle sty then
219 sep [ptext SLIT("__coerce") <+>
220 sep [pprParendType to_ty, pprParendType from_ty],
223 sep [hsep [ptext SLIT("__coerce"), pprParendType to_ty],
226 ppr_expr add_par (Note (Coerce to_ty from_ty) expr)
228 sep [sep [ptext SLIT("__coerce"), nest 2 (pprParendType to_ty)],
232 ppr_expr add_par (Note InlineCall expr)
233 = add_par (ptext SLIT("__inline_call") <+> pprParendExpr expr)
235 ppr_expr add_par (Note InlineMe expr)
236 = add_par $ ptext SLIT("__inline_me") <+> pprParendExpr expr
238 ppr_expr add_par (Note (CoreNote s) expr)
240 sep [sep [ptext SLIT("__core_note"), pprHsString (mkFastString s)],
243 pprCoreAlt (con, args, rhs)
244 = hang (ppr_case_pat con args) 2 (pprCoreExpr rhs)
246 ppr_case_pat con@(DataAlt dc) args
248 = tupleParens (tupleTyConBoxity tc) (hsep (punctuate comma (map ppr_bndr args))) <+> arrow
250 ppr_bndr = pprBndr CaseBind
253 ppr_case_pat con args
254 = ppr con <+> hsep (map ppr_bndr args) <+> arrow
256 ppr_bndr = pprBndr CaseBind
258 pprArg (Type ty) = ptext SLIT("@") <+> pprParendType ty
259 pprArg expr = pprParendExpr expr
262 Other printing bits-and-bobs used with the general @pprCoreBinding@
263 and @pprCoreExpr@ functions.
266 instance OutputableBndr Var where
267 pprBndr = pprCoreBinder
269 pprCoreBinder :: BindingSite -> Var -> SDoc
270 pprCoreBinder LetBind binder
271 = vcat [sig, pprIdDetails binder, pragmas]
273 sig = pprTypedBinder binder
274 pragmas = ppIdInfo binder (idInfo binder)
276 -- Lambda bound type variables are preceded by "@"
277 pprCoreBinder LambdaBind bndr = parens (pprTypedBinder bndr)
279 -- Case bound things don't get a signature or a herald
280 pprCoreBinder CaseBind bndr = pprUntypedBinder bndr
282 pprUntypedBinder binder
283 | isTyVar binder = ptext SLIT("@") <+> ppr binder -- NB: don't print kind
284 | otherwise = pprIdBndr binder
286 pprTypedBinder binder
287 | isTyVar binder = ptext SLIT("@") <+> pprTyVarBndr binder
288 | otherwise = pprIdBndr binder <+> dcolon <+> pprType (idType binder)
290 pprTyVarBndr :: TyVar -> SDoc
292 = getPprStyle $ \ sty ->
293 if debugStyle sty then
294 hsep [ppr tyvar, dcolon, pprParendKind kind]
295 -- See comments with ppDcolon in PprCore.lhs
299 kind = tyVarKind tyvar
301 -- pprIdBndr does *not* print the type
302 -- When printing any Id binder in debug mode, we print its inline pragma and one-shot-ness
303 pprIdBndr id = ppr id <+>
304 (megaSeqIdInfo (idInfo id) `seq`
305 -- Useful for poking on black holes
306 ifPprDebug (ppr (idInlinePragma id) <+> ppr (idOccInfo id) <+>
307 #ifdef OLD_STRICTNESS
308 ppr (idDemandInfo id) <+>
310 ppr (idNewDemandInfo id) <+>
311 ppr (idLBVarInfo id)))
316 pprIdDetails :: Id -> SDoc
317 pprIdDetails id | isGlobalId id = ppr (globalIdDetails id)
318 | isExportedId id = ptext SLIT("[Exported]")
319 | isSpecPragmaId id = ptext SLIT("[SpecPrag]")
322 ppIdInfo :: Id -> IdInfo -> SDoc
324 = hsep [ ppArityInfo a,
325 ppWorkerInfo (workerInfo info),
326 ppCafInfo (cafInfo info),
327 #ifdef OLD_STRICTNESS
331 pprNewStrictness (newStrictnessInfo info),
332 vcat (map (pprCoreRule (ppr b)) (rulesRules p))
333 -- Inline pragma, occ, demand, lbvar info
334 -- printed out with all binders (when debug is on);
335 -- see PprCore.pprIdBndr
339 #ifdef OLD_STRICTNESS
340 s = strictnessInfo info
348 pprIdRules :: [IdCoreRule] -> SDoc
349 pprIdRules rules = vcat (map pprIdRule rules)
351 pprIdRule :: IdCoreRule -> SDoc
352 pprIdRule (IdCoreRule id _ rule) = pprCoreRule (ppr id) rule
354 pprCoreRule :: SDoc -> CoreRule -> SDoc
355 pprCoreRule pp_fn (BuiltinRule name _)
356 = ptext SLIT("Built in rule for") <+> pp_fn <> colon <+> doubleQuotes (ftext name)
358 pprCoreRule pp_fn (Rule name act tpl_vars tpl_args rhs)
359 = doubleQuotes (ftext name) <+> ppr act <+>
361 ptext SLIT("__forall") <+> braces (sep (map pprTypedBinder tpl_vars)),
362 nest 2 (pp_fn <+> sep (map pprArg tpl_args)),
363 nest 2 (ptext SLIT("=") <+> pprCoreExpr rhs)