PprCore.lhs 10.3 KB
Newer Older
1
%
2
% (c) The AQUA Project, Glasgow University, 1996-1998
3
4
5
6
7
8
9
10
11
%
%************************************************************************
%*									*
\section[PprCore]{Printing of Core syntax, including for interfaces}
%*									*
%************************************************************************

\begin{code}
module PprCore (
12
	pprCoreExpr, pprParendExpr,
13
	pprCoreBinding, pprCoreBindings, pprCoreAlt,
14
	pprIdRules
15
16
    ) where

17
#include "HsVersions.h"
18
19

import CoreSyn
20
import CostCentre	( pprCostCentreCore )
21
import Var		( Var )
22
import Id		( Id, idType, isDataConWorkId_maybe, idLBVarInfo, idArity,
23
24
25
			  idInfo, idInlinePragma, idOccInfo,
			  globalIdDetails, isGlobalId, isExportedId, 
			  isSpecPragmaId, idNewDemandInfo
26
			)
27
import Var		( TyVar, isTyVar, tyVarKind )
28
import IdInfo		( IdInfo, megaSeqIdInfo, 
29
			  arityInfo, ppArityInfo, 
30
			  specInfo, pprNewStrictness,
31
			  workerInfo, ppWorkerInfo,
32
			  newStrictnessInfo, cafInfo, ppCafInfo,
33
34
			)

35
#ifdef OLD_STRICTNESS
36
37
import Id		( idDemandInfo )
import IdInfo		( cprInfo, ppCprInfo, strictnessInfo, ppStrictnessInfo ) 
38
#endif
39

40
41
import DataCon		( dataConTyCon )
import TyCon		( tupleTyConBoxity, isTupleTyCon )
42
import Type		( pprParendType, pprType, pprParendKind )
43
import BasicTypes	( tupleParens )
sof's avatar
sof committed
44
import Util             ( lengthIs )
45
import Outputable
46
import FastString       ( mkFastString )
47
48
49
50
51
52
53
54
55
56
57
\end{code}

%************************************************************************
%*									*
\subsection{Public interfaces for Core printing (excluding instances)}
%*									*
%************************************************************************

@pprParendCoreExpr@ puts parens around non-atomic Core expressions.

\begin{code}
58
59
60
61
pprCoreBindings :: OutputableBndr b => [Bind b] -> SDoc
pprCoreBinding  :: OutputableBndr b => Bind b  -> SDoc
pprCoreExpr     :: OutputableBndr b => Expr b  -> SDoc
pprParendExpr   :: OutputableBndr b => Expr b  -> SDoc
62

63
64
pprCoreBindings = pprTopBinds
pprCoreBinding  = pprTopBind 
65

66
67
instance OutputableBndr b => Outputable (Bind b) where
    ppr bind = ppr_bind bind
68

69
70
instance OutputableBndr b => Outputable (Expr b) where
    ppr expr = pprCoreExpr expr
71
\end{code}
72

73
74
75

%************************************************************************
%*									*
76
\subsection{The guts}
77
78
79
80
%*									*
%************************************************************************

\begin{code}
81
pprTopBinds binds = vcat (map pprTopBind binds)
82

83
84
pprTopBind (NonRec binder expr)
 = ppr_binding (binder,expr) $$ text ""
85

86
pprTopBind (Rec binds)
87
  = vcat [ptext SLIT("Rec {"),
88
	  vcat (map ppr_binding binds),
89
90
	  ptext SLIT("end Rec }"),
	  text ""]
91
92
93
\end{code}

\begin{code}
94
ppr_bind :: OutputableBndr b => Bind b -> SDoc
95

96
97
98
99
ppr_bind (NonRec val_bdr expr) = ppr_binding (val_bdr, expr)
ppr_bind (Rec binds)  	       = vcat (map pp binds)
			       where
				 pp bind = ppr_binding bind <> semi
100

101
102
103
104
ppr_binding :: OutputableBndr b => (b, Expr b) -> SDoc
ppr_binding (val_bdr, expr)
  = pprBndr LetBind val_bdr $$ 
    (ppr val_bdr <+> equals <+> pprCoreExpr expr)
105
106
107
\end{code}

\begin{code}
108
109
pprParendExpr   expr = ppr_expr parens expr
pprCoreExpr expr = ppr_expr noParens expr
110
111
112

noParens :: SDoc -> SDoc
noParens pp = pp
113
114
115
\end{code}

\begin{code}
116
ppr_expr :: OutputableBndr b => (SDoc -> SDoc) -> Expr b -> SDoc
117
118
	-- The function adds parens in context that need
	-- an atomic value (e.g. function args)
119

120
ppr_expr add_par (Type ty)  = add_par (ptext SLIT("TYPE") <+> ppr ty)	-- Wierd
121
	           
122
123
ppr_expr add_par (Var name) = ppr name
ppr_expr add_par (Lit lit)  = ppr lit
124

125
ppr_expr add_par expr@(Lam _ _)
126
  = let
127
	(bndrs, body) = collectBinders expr
128
    in
129
    add_par $
130
131
    hang (ptext SLIT("\\") <+> sep (map (pprBndr LambdaBind) bndrs) <+> arrow)
	 2 (pprCoreExpr body)
132

133
ppr_expr add_par expr@(App fun arg)
134
135
  = case collectArgs expr of { (fun, args) -> 
    let
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))
139
    in
140
    case fun of
141
	Var f -> case isDataConWorkId_maybe f of
142
143
			-- Notice that we print the *worker*
			-- for tuples in paren'd format.
144
145
146
147
		   Just dc | saturated && isTupleTyCon tc
			   -> tupleParens (tupleTyConBoxity tc) pp_tup_args
			   where
			     tc	       = dataConTyCon dc
sof's avatar
sof committed
148
			     saturated = val_args `lengthIs` idArity f
149

150
		   other -> add_par (hang (ppr f) 2 pp_args)
151

152
	other -> add_par (hang (pprParendExpr fun) 2 pp_args)
153
154
    }

155
ppr_expr add_par (Case expr var ty [(con,args,rhs)])
156
  = add_par $
157
    sep [sep [ptext SLIT("case") <+> pprParendType ty <+> pprCoreExpr expr,
158
	      hsep [ptext SLIT("of"),
159
		    ppr_bndr var, 
160
		    char '{',
161
		    ppr_case_pat con args
162
	  ]],
163
	 pprCoreExpr rhs,
164
165
	 char '}'
    ]
166
  where
167
    ppr_bndr = pprBndr CaseBind
168

169
ppr_expr add_par (Case expr var ty alts)
170
  = add_par $
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))),
174
175
176
	 char '}'
    ]
  where
177
    ppr_bndr = pprBndr CaseBind
178
 
179
180
181
182

-- special cases: let ... in let ...
-- ("disgusting" SLPJ)

183
184
{-
ppr_expr add_par (Let bind@(NonRec val_bdr rhs@(Let _ _)) body)
185
186
  = add_par $
    vcat [
187
188
      hsep [ptext SLIT("let {"), (pprBndr LetBind val_bdr $$ ppr val_bndr), equals],
      nest 2 (pprCoreExpr rhs),
sof's avatar
sof committed
189
      ptext SLIT("} in"),
190
191
      pprCoreExpr body ]
-}
192

193
ppr_expr add_par (Let bind@(NonRec val_bdr rhs) expr@(Let _ _))
194
195
  = add_par
    (hang (ptext SLIT("let {"))
196
197
	  2 (hsep [ppr_binding (val_bdr,rhs),
		   ptext SLIT("} in")])
198
     $$
199
     pprCoreExpr expr)
200
201

-- general case (recursive case, too)
202
ppr_expr add_par (Let bind expr)
203
  = add_par $
204
205
    sep [hang (ptext keyword) 2 (ppr_bind bind),
	 hang (ptext SLIT("} in ")) 2 (pprCoreExpr expr)]
206
207
  where
    keyword = case bind of
208
		Rec _      -> SLIT("__letrec {")
209
		NonRec _ _ -> SLIT("let {")
210

211
212
ppr_expr add_par (Note (SCC cc) expr)
  = add_par (sep [pprCostCentreCore cc, pprCoreExpr expr])
213

sof's avatar
sof committed
214
#ifdef DEBUG
215
ppr_expr add_par (Note (Coerce to_ty from_ty) expr)
216
217
 = add_par $
   getPprStyle $ \ sty ->
218
   if debugStyle sty then
219
220
221
      sep [ptext SLIT("__coerce") <+> 
		sep [pprParendType to_ty, pprParendType from_ty],
	   pprParendExpr expr]
sof's avatar
sof committed
222
   else
223
224
      sep [hsep [ptext SLIT("__coerce"), pprParendType to_ty],
	          pprParendExpr expr]
sof's avatar
sof committed
225
#else
226
ppr_expr add_par (Note (Coerce to_ty from_ty) expr)
227
  = add_par $
228
229
    sep [sep [ptext SLIT("__coerce"), nest 2 (pprParendType to_ty)],
	 pprParendExpr expr]
sof's avatar
sof committed
230
#endif
231

232
233
ppr_expr add_par (Note InlineCall expr)
  = add_par (ptext SLIT("__inline_call") <+> pprParendExpr expr)
234

235
236
ppr_expr add_par (Note InlineMe expr)
  = add_par $ ptext SLIT("__inline_me") <+> pprParendExpr expr
237

238
239
240
241
242
ppr_expr add_par (Note (CoreNote s) expr)
  = add_par $ 
    sep [sep [ptext SLIT("__core_note"), pprHsString (mkFastString s)],
         pprParendExpr expr]

243
244
pprCoreAlt (con, args, rhs) 
  = hang (ppr_case_pat con args) 2 (pprCoreExpr rhs)
245

246
ppr_case_pat con@(DataAlt dc) args
247
248
  | isTupleTyCon tc
  = tupleParens (tupleTyConBoxity tc) (hsep (punctuate comma (map ppr_bndr args))) <+> arrow
249
  where
250
    ppr_bndr = pprBndr CaseBind
251
    tc = dataConTyCon dc
252

253
ppr_case_pat con args
254
  = ppr con <+> hsep (map ppr_bndr args) <+> arrow
255
  where
256
    ppr_bndr = pprBndr CaseBind
257

258
259
pprArg (Type ty) = ptext SLIT("@") <+> pprParendType ty
pprArg expr      = pprParendExpr expr
260
261
262
263
264
265
\end{code}

Other printing bits-and-bobs used with the general @pprCoreBinding@
and @pprCoreExpr@ functions.

\begin{code}
266
267
268
269
instance OutputableBndr Var where
  pprBndr = pprCoreBinder

pprCoreBinder :: BindingSite -> Var -> SDoc
270
pprCoreBinder LetBind binder
271
  = vcat [sig, pprIdDetails binder, pragmas]
272
  where
273
    sig     = pprTypedBinder binder
274
    pragmas = ppIdInfo binder (idInfo binder)
275

276
-- Lambda bound type variables are preceded by "@"
277
pprCoreBinder LambdaBind bndr = parens (pprTypedBinder bndr)
278

279
280
-- Case bound things don't get a signature or a herald
pprCoreBinder CaseBind bndr = pprUntypedBinder bndr
281

282
pprUntypedBinder binder
283
  | isTyVar binder = ptext SLIT("@") <+> ppr binder	-- NB: don't print kind
284
285
  | otherwise      = pprIdBndr binder

286
pprTypedBinder binder
287
  | isTyVar binder  = ptext SLIT("@") <+> pprTyVarBndr binder
288
  | otherwise	    = pprIdBndr binder <+> dcolon <+> pprType (idType binder)
289

290
291
292
293
294
295
296
297
298
299
300
pprTyVarBndr :: TyVar -> SDoc
pprTyVarBndr tyvar
  = getPprStyle $ \ sty ->
    if debugStyle sty then
        hsep [ppr tyvar, dcolon, pprParendKind kind]
		-- See comments with ppDcolon in PprCore.lhs
    else
        ppr tyvar
  where
    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
304
305
306
pprIdBndr id = ppr id <+> 
	       (megaSeqIdInfo (idInfo id) `seq`
			-- Useful for poking on black holes
	        ifPprDebug (ppr (idInlinePragma id) <+> ppr (idOccInfo id) <+> 
307
#ifdef OLD_STRICTNESS
308
309
310
311
			    ppr (idDemandInfo id) <+>
#endif
			    ppr (idNewDemandInfo id) <+>
			    ppr (idLBVarInfo id)))
312
\end{code}
313
314
315


\begin{code}
316
317
318
319
320
321
pprIdDetails :: Id -> SDoc
pprIdDetails id | isGlobalId id     = ppr (globalIdDetails id)
		| isExportedId id   = ptext SLIT("[Exported]")
		| isSpecPragmaId id = ptext SLIT("[SpecPrag]")
		| otherwise	    = empty

322
323
ppIdInfo :: Id -> IdInfo -> SDoc
ppIdInfo b info
324
  = hsep [  ppArityInfo a,
325
	    ppWorkerInfo (workerInfo info),
326
	    ppCafInfo (cafInfo info),
327
#ifdef OLD_STRICTNESS
328
	    ppStrictnessInfo s,
329
            ppCprInfo m,
330
#endif
331
	    pprNewStrictness (newStrictnessInfo info),
332
	    vcat (map (pprCoreRule (ppr b)) (rulesRules p))
333
334
335
	-- Inline pragma, occ, demand, lbvar info
	-- printed out with all binders (when debug is on); 
	-- see PprCore.pprIdBndr
336
337
338
	]
  where
    a = arityInfo info
339
#ifdef OLD_STRICTNESS
340
    s = strictnessInfo info
341
    m = cprInfo info
342
#endif
343
344
345
    p = specInfo info
\end{code}

346

347
\begin{code}
348
349
pprIdRules :: [IdCoreRule] -> SDoc
pprIdRules rules = vcat (map pprIdRule rules)
350

351
pprIdRule :: IdCoreRule -> SDoc
352
pprIdRule (IdCoreRule id _ rule) = pprCoreRule (ppr id) rule
353

354
pprCoreRule :: SDoc -> CoreRule -> SDoc
355
pprCoreRule pp_fn (BuiltinRule name _)
356
  = ptext SLIT("Built in rule for") <+> pp_fn <> colon <+> doubleQuotes (ftext name)
357

358
pprCoreRule pp_fn (Rule name act tpl_vars tpl_args rhs)
359
  = doubleQuotes (ftext name) <+> ppr act <+>
360
361
    sep [
	  ptext SLIT("__forall") <+> braces (sep (map pprTypedBinder tpl_vars)),
chak's avatar
chak committed
362
363
	  nest 2 (pp_fn <+> sep (map pprArg tpl_args)),
	  nest 2 (ptext SLIT("=") <+> pprCoreExpr rhs)
364
    ] <+> semi
365
\end{code}