1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
%
% (c) The GRASP/AQUA Project, Glasgow University, 1992-1994
%
\section[AnnCoreSyntax]{Annotated core syntax}
For when you want @CoreSyntax@ trees annotated at every node. Other
than that, just like @CoreSyntax@. (Important to be sure that it {\em
really is} just like @CoreSyntax@.)
\begin{code}
#include "HsVersions.h"
module AnnCoreSyn (
AnnCoreBinding(..), AnnCoreExpr(..),
AnnCoreExpr'(..), -- v sad that this must be exported
AnnCoreCaseAlternatives(..), AnnCoreCaseDefault(..),
#ifdef DPH
AnnCoreParQuals(..),
AnnCoreParCommunicate(..),
#endif {- Data Parallel Haskell -}
deAnnotate, -- we may eventually export some of the other deAnners
-- and to make the interface self-sufficient
BasicLit, Id, PrimOp, TyCon, TyVar, UniType, CostCentre
IF_ATTACK_PRAGMAS(COMMA cmpTyCon COMMA cmpTyVar)
IF_ATTACK_PRAGMAS(COMMA cmpUniType)
) where
import AbsPrel ( PrimOp(..), PrimKind
IF_ATTACK_PRAGMAS(COMMA tagOf_PrimOp)
IF_ATTACK_PRAGMAS(COMMA pprPrimOp)
)
import AbsUniType ( Id, TyVar, TyCon, UniType
IF_ATTACK_PRAGMAS(COMMA cmpTyCon COMMA cmpTyVar)
IF_ATTACK_PRAGMAS(COMMA cmpUniType)
)
import BasicLit ( BasicLit )
import CoreSyn
import Outputable
import CostCentre ( CostCentre )
#if USE_ATTACK_PRAGMAS
import Util
#endif
\end{code}
\begin{code}
data AnnCoreBinding binder bindee annot
= AnnCoNonRec binder (AnnCoreExpr binder bindee annot)
| AnnCoRec [(binder, AnnCoreExpr binder bindee annot)]
\end{code}
\begin{code}
type AnnCoreExpr binder bindee annot = (annot, AnnCoreExpr' binder bindee annot)
data AnnCoreExpr' binder bindee annot
= AnnCoVar bindee
| AnnCoLit BasicLit
| AnnCoCon Id [UniType] [CoreAtom bindee]
| AnnCoPrim PrimOp [UniType] [CoreAtom bindee]
| AnnCoLam [binder]
(AnnCoreExpr binder bindee annot)
| AnnCoTyLam TyVar
(AnnCoreExpr binder bindee annot)
| AnnCoApp (AnnCoreExpr binder bindee annot)
(CoreAtom bindee)
| AnnCoTyApp (AnnCoreExpr binder bindee annot)
UniType
| AnnCoCase (AnnCoreExpr binder bindee annot)
(AnnCoreCaseAlternatives binder bindee annot)
| AnnCoLet (AnnCoreBinding binder bindee annot)
(AnnCoreExpr binder bindee annot)
| AnnCoSCC CostCentre
(AnnCoreExpr binder bindee annot)
#ifdef DPH
| AnnCoZfExpr (AnnCoreExpr binder bindee annot)
(AnnCoreParQuals binder bindee annot)
| AnnCoParCon Id Int [UniType] [AnnCoreExpr binder bindee annot]
| AnnCoParComm
Int
(AnnCoreExpr binder bindee annot)
(AnnCoreParCommunicate binder bindee annot)
| AnnCoParZipWith
Int
(AnnCoreExpr binder bindee annot)
[AnnCoreExpr binder bindee annot]
#endif {- Data Parallel Haskell -}
\end{code}
\begin{code}
#ifdef DPH
data AnnCoreParQuals binder bindee annot
= AnnCoAndQuals (AnnCoreParQuals binder bindee annot)
(AnnCoreParQuals binder bindee annot)
| AnnCoParFilter (AnnCoreExpr binder bindee annot)
| AnnCoDrawnGen [binder]
(binder)
(AnnCoreExpr binder bindee annot)
| AnnCoIndexGen [AnnCoreExpr binder bindee annot]
(binder)
(AnnCoreExpr binder bindee annot)
#endif {- Data Parallel Haskell -}
\end{code}
\begin{code}
data AnnCoreCaseAlternatives binder bindee annot
= AnnCoAlgAlts [(Id,
[binder],
AnnCoreExpr binder bindee annot)]
(AnnCoreCaseDefault binder bindee annot)
| AnnCoPrimAlts [(BasicLit,
AnnCoreExpr binder bindee annot)]
(AnnCoreCaseDefault binder bindee annot)
#ifdef DPH
| AnnCoParAlgAlts TyCon
Int
[binder]
[(Id,
AnnCoreExpr binder bindee annot)]
(AnnCoreCaseDefault binder bindee annot)
| AnnCoParPrimAlts TyCon
Int
[(BasicLit,
AnnCoreExpr binder bindee annot)]
(AnnCoreCaseDefault binder bindee annot)
#endif {- Data Parallel Haskell -}
data AnnCoreCaseDefault binder bindee annot
= AnnCoNoDefault
| AnnCoBindDefault binder
(AnnCoreExpr binder bindee annot)
\end{code}
\begin{code}
#ifdef DPH
data AnnCoreParCommunicate binder bindee annot
= AnnCoParSend [AnnCoreExpr binder bindee annot]
| AnnCoParFetch [AnnCoreExpr binder bindee annot]
| AnnCoToPodized
| AnnCoFromPodized
#endif {- Data Parallel Haskell -}
\end{code}
\begin{code}
deAnnotate :: AnnCoreExpr bndr bdee ann -> CoreExpr bndr bdee
deAnnotate (_, AnnCoVar v) = CoVar v
deAnnotate (_, AnnCoLit lit) = CoLit lit
deAnnotate (_, AnnCoCon con tys args) = CoCon con tys args
deAnnotate (_, AnnCoPrim op tys args) = CoPrim op tys args
deAnnotate (_, AnnCoLam binders body) = CoLam binders (deAnnotate body)
deAnnotate (_, AnnCoTyLam tyvar body) = CoTyLam tyvar (deAnnotate body)
deAnnotate (_, AnnCoApp fun arg) = CoApp (deAnnotate fun) arg
deAnnotate (_, AnnCoTyApp fun ty) = CoTyApp (deAnnotate fun) ty
deAnnotate (_, AnnCoSCC lbl body) = CoSCC lbl (deAnnotate body)
deAnnotate (_, AnnCoLet bind body)
= CoLet (deAnnBind bind) (deAnnotate body)
where
deAnnBind (AnnCoNonRec var rhs) = CoNonRec var (deAnnotate rhs)
deAnnBind (AnnCoRec pairs) = CoRec [(v,deAnnotate rhs) | (v,rhs) <- pairs]
deAnnotate (_, AnnCoCase scrut alts)
= CoCase (deAnnotate scrut) (deAnnAlts alts)
where
deAnnAlts (AnnCoAlgAlts alts deflt)
= CoAlgAlts [(con,args,deAnnotate rhs) | (con,args,rhs) <- alts]
(deAnnDeflt deflt)
deAnnAlts (AnnCoPrimAlts alts deflt)
= CoPrimAlts [(lit,deAnnotate rhs) | (lit,rhs) <- alts]
(deAnnDeflt deflt)
deAnnDeflt AnnCoNoDefault = CoNoDefault
deAnnDeflt (AnnCoBindDefault var rhs) = CoBindDefault var (deAnnotate rhs)
\end{code}
|
