adamc@29
|
1 (* Copyright (c) 2008, Adam Chlipala
|
adamc@29
|
2 * All rights reserved.
|
adamc@29
|
3 *
|
adamc@29
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@29
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@29
|
6 *
|
adamc@29
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@29
|
8 * this list of conditions and the following disclaimer.
|
adamc@29
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@29
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@29
|
11 * and/or other materials provided with the distribution.
|
adamc@29
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@29
|
13 * derived from this software without specific prior written permission.
|
adamc@29
|
14 *
|
adamc@29
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@29
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@29
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@29
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@29
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@29
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@29
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@29
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@29
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@29
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@29
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@29
|
26 *)
|
adamc@29
|
27
|
adamc@29
|
28 structure Cjrize :> CJRIZE = struct
|
adamc@29
|
29
|
adamc@109
|
30 structure L = Mono
|
adamc@29
|
31 structure L' = Cjr
|
adamc@29
|
32
|
adamc@196
|
33 structure IM = IntBinaryMap
|
adamc@196
|
34
|
adamc@29
|
35 structure Sm :> sig
|
adamc@29
|
36 type t
|
adamc@29
|
37
|
adamc@29
|
38 val empty : t
|
adamc@29
|
39 val find : t * (string * L.typ) list * (string * L'.typ) list -> t * int
|
adamc@29
|
40
|
adamc@29
|
41 val declares : t -> (int * (string * L'.typ) list) list
|
adamc@29
|
42 end = struct
|
adamc@29
|
43
|
adamc@29
|
44 structure FM = BinaryMapFn(struct
|
adamc@29
|
45 type ord_key = L.typ
|
adamc@109
|
46 val compare = MonoUtil.Typ.compare
|
adamc@29
|
47 end)
|
adamc@29
|
48
|
adamc@29
|
49 type t = int * int FM.map * (int * (string * L'.typ) list) list
|
adamc@29
|
50
|
adamc@102
|
51 val empty : t = (1, FM.insert (FM.empty, (L.TRecord [], ErrorMsg.dummySpan), 0), [])
|
adamc@29
|
52
|
adamc@29
|
53 fun find ((n, m, ds), xts, xts') =
|
adamc@29
|
54 let
|
adamc@29
|
55 val t = (L.TRecord xts, ErrorMsg.dummySpan)
|
adamc@29
|
56 in
|
adamc@29
|
57 case FM.find (m, t) of
|
adamc@29
|
58 NONE => ((n+1, FM.insert (m, t, n), (n, xts') :: ds), n)
|
adamc@29
|
59 | SOME i => ((n, m, ds), i)
|
adamc@29
|
60 end
|
adamc@29
|
61
|
adamc@29
|
62 fun declares (_, _, ds) = ds
|
adamc@29
|
63
|
adamc@29
|
64 end
|
adamc@29
|
65
|
adamc@196
|
66 fun cifyTyp x =
|
adamc@196
|
67 let
|
adamc@196
|
68 fun cify dtmap ((t, loc), sm) =
|
adamc@196
|
69 case t of
|
adamc@196
|
70 L.TFun (t1, t2) =>
|
adamc@196
|
71 let
|
adamc@196
|
72 val (t1, sm) = cify dtmap (t1, sm)
|
adamc@196
|
73 val (t2, sm) = cify dtmap (t2, sm)
|
adamc@196
|
74 in
|
adamc@196
|
75 ((L'.TFun (t1, t2), loc), sm)
|
adamc@196
|
76 end
|
adamc@196
|
77 | L.TRecord xts =>
|
adamc@196
|
78 let
|
adamc@196
|
79 val old_xts = xts
|
adamc@196
|
80 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
|
adamc@196
|
81 let
|
adamc@196
|
82 val (t, sm) = cify dtmap (t, sm)
|
adamc@196
|
83 in
|
adamc@196
|
84 ((x, t), sm)
|
adamc@196
|
85 end)
|
adamc@196
|
86 sm xts
|
adamc@196
|
87 val (sm, si) = Sm.find (sm, old_xts, xts)
|
adamc@196
|
88 in
|
adamc@196
|
89 ((L'.TRecord si, loc), sm)
|
adamc@196
|
90 end
|
adamc@196
|
91 | L.TDatatype (n, ref (dk, xncs)) =>
|
adamc@196
|
92 (case IM.find (dtmap, n) of
|
adamc@196
|
93 SOME r => ((L'.TDatatype (dk, n, r), loc), sm)
|
adamc@196
|
94 | NONE =>
|
adamc@196
|
95 let
|
adamc@196
|
96 val r = ref []
|
adamc@196
|
97 val dtmap = IM.insert (dtmap, n, r)
|
adamc@196
|
98
|
adamc@196
|
99 val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
|
adamc@196
|
100 case to of
|
adamc@196
|
101 NONE => ((x, n, NONE), sm)
|
adamc@196
|
102 | SOME t =>
|
adamc@196
|
103 let
|
adamc@196
|
104 val (t, sm) = cify dtmap (t, sm)
|
adamc@196
|
105 in
|
adamc@196
|
106 ((x, n, SOME t), sm)
|
adamc@196
|
107 end)
|
adamc@196
|
108 sm xncs
|
adamc@196
|
109 in
|
adamc@196
|
110 r := xncs;
|
adamc@196
|
111 ((L'.TDatatype (dk, n, r), loc), sm)
|
adamc@196
|
112 end)
|
adamc@196
|
113 | L.TFfi mx => ((L'.TFfi mx, loc), sm)
|
adamc@288
|
114 | L.TOption t =>
|
adamc@288
|
115 let
|
adamc@288
|
116 val (t, sm) = cify dtmap (t, sm)
|
adamc@288
|
117 in
|
adamc@288
|
118 ((L'.TOption t, loc), sm)
|
adamc@288
|
119 end
|
adamc@196
|
120 in
|
adamc@196
|
121 cify IM.empty x
|
adamc@196
|
122 end
|
adamc@29
|
123
|
adamc@109
|
124 val dummye = (L'.EPrim (Prim.Int 0), ErrorMsg.dummySpan)
|
adamc@109
|
125
|
adamc@186
|
126 fun cifyPatCon (pc, sm) =
|
adamc@181
|
127 case pc of
|
adamc@186
|
128 L.PConVar n => (L'.PConVar n, sm)
|
adamc@186
|
129 | L.PConFfi {mod = m, datatyp, con, arg} =>
|
adamc@186
|
130 let
|
adamc@186
|
131 val (arg, sm) =
|
adamc@186
|
132 case arg of
|
adamc@186
|
133 NONE => (NONE, sm)
|
adamc@186
|
134 | SOME t =>
|
adamc@186
|
135 let
|
adamc@186
|
136 val (t, sm) = cifyTyp (t, sm)
|
adamc@186
|
137 in
|
adamc@186
|
138 (SOME t, sm)
|
adamc@186
|
139 end
|
adamc@186
|
140 in
|
adamc@186
|
141 (L'.PConFfi {mod = m, datatyp = datatyp, con = con, arg = arg}, sm)
|
adamc@186
|
142 end
|
adamc@181
|
143
|
adamc@182
|
144 fun cifyPat ((p, loc), sm) =
|
adamc@181
|
145 case p of
|
adamc@182
|
146 L.PWild => ((L'.PWild, loc), sm)
|
adamc@182
|
147 | L.PVar (x, t) =>
|
adamc@182
|
148 let
|
adamc@182
|
149 val (t, sm) = cifyTyp (t, sm)
|
adamc@182
|
150 in
|
adamc@182
|
151 ((L'.PVar (x, t), loc), sm)
|
adamc@182
|
152 end
|
adamc@182
|
153 | L.PPrim p => ((L'.PPrim p, loc), sm)
|
adamc@188
|
154 | L.PCon (dk, pc, NONE) =>
|
adamc@186
|
155 let
|
adamc@186
|
156 val (pc, sm) = cifyPatCon (pc, sm)
|
adamc@186
|
157 in
|
adamc@188
|
158 ((L'.PCon (dk, pc, NONE), loc), sm)
|
adamc@186
|
159 end
|
adamc@188
|
160 | L.PCon (dk, pc, SOME p) =>
|
adamc@182
|
161 let
|
adamc@186
|
162 val (pc, sm) = cifyPatCon (pc, sm)
|
adamc@182
|
163 val (p, sm) = cifyPat (p, sm)
|
adamc@182
|
164 in
|
adamc@188
|
165 ((L'.PCon (dk, pc, SOME p), loc), sm)
|
adamc@182
|
166 end
|
adamc@182
|
167 | L.PRecord xps =>
|
adamc@182
|
168 let
|
adamc@182
|
169 val (xps, sm) = ListUtil.foldlMap (fn ((x, p, t), sm) =>
|
adamc@182
|
170 let
|
adamc@182
|
171 val (p, sm) = cifyPat (p, sm)
|
adamc@182
|
172 val (t, sm) = cifyTyp (t, sm)
|
adamc@182
|
173 in
|
adamc@182
|
174 ((x, p, t), sm)
|
adamc@182
|
175 end) sm xps
|
adamc@182
|
176 in
|
adamc@182
|
177 ((L'.PRecord xps, loc), sm)
|
adamc@182
|
178 end
|
adamc@288
|
179 | L.PNone t =>
|
adamc@288
|
180 let
|
adamc@288
|
181 val (t, sm) = cifyTyp (t, sm)
|
adamc@288
|
182 in
|
adamc@288
|
183 ((L'.PNone t, loc), sm)
|
adamc@288
|
184 end
|
adamc@288
|
185 | L.PSome (t, p) =>
|
adamc@288
|
186 let
|
adamc@288
|
187 val (t, sm) = cifyTyp (t, sm)
|
adamc@288
|
188 val (p, sm) = cifyPat (p, sm)
|
adamc@288
|
189 in
|
adamc@288
|
190 ((L'.PSome (t, p), loc), sm)
|
adamc@288
|
191 end
|
adamc@288
|
192
|
adamc@181
|
193
|
adamc@280
|
194 fun cifyExp (eAll as (e, loc), sm) =
|
adamc@29
|
195 case e of
|
adamc@29
|
196 L.EPrim p => ((L'.EPrim p, loc), sm)
|
adamc@29
|
197 | L.ERel n => ((L'.ERel n, loc), sm)
|
adamc@29
|
198 | L.ENamed n => ((L'.ENamed n, loc), sm)
|
adamc@188
|
199 | L.ECon (dk, pc, eo) =>
|
adamc@181
|
200 let
|
adamc@181
|
201 val (eo, sm) =
|
adamc@181
|
202 case eo of
|
adamc@181
|
203 NONE => (NONE, sm)
|
adamc@181
|
204 | SOME e =>
|
adamc@181
|
205 let
|
adamc@181
|
206 val (e, sm) = cifyExp (e, sm)
|
adamc@181
|
207 in
|
adamc@181
|
208 (SOME e, sm)
|
adamc@181
|
209 end
|
adamc@186
|
210 val (pc, sm) = cifyPatCon (pc, sm)
|
adamc@181
|
211 in
|
adamc@188
|
212 ((L'.ECon (dk, pc, eo), loc), sm)
|
adamc@181
|
213 end
|
adamc@297
|
214 | L.ENone t =>
|
adamc@297
|
215 let
|
adamc@297
|
216 val (t, sm) = cifyTyp (t, sm)
|
adamc@297
|
217 in
|
adamc@297
|
218 ((L'.ENone t, loc), sm)
|
adamc@297
|
219 end
|
adamc@290
|
220 | L.ESome (t, e) =>
|
adamc@290
|
221 let
|
adamc@290
|
222 val (t, sm) = cifyTyp (t, sm)
|
adamc@290
|
223 val (e, sm) = cifyExp (e, sm)
|
adamc@290
|
224 in
|
adamc@290
|
225 ((L'.ESome (t, e), loc), sm)
|
adamc@290
|
226 end
|
adamc@53
|
227 | L.EFfi mx => ((L'.EFfi mx, loc), sm)
|
adamc@53
|
228 | L.EFfiApp (m, x, es) =>
|
adamc@53
|
229 let
|
adamc@53
|
230 val (es, sm) = ListUtil.foldlMap cifyExp sm es
|
adamc@53
|
231 in
|
adamc@53
|
232 ((L'.EFfiApp (m, x, es), loc), sm)
|
adamc@53
|
233 end
|
adamc@29
|
234 | L.EApp (e1, e2) =>
|
adamc@29
|
235 let
|
adamc@316
|
236 fun unravel (e, args) =
|
adamc@316
|
237 case e of
|
adamc@316
|
238 (L.EApp (e1, e2), _) => unravel (e1, e2 :: args)
|
adamc@316
|
239 | _ => (e, args)
|
adamc@316
|
240
|
adamc@316
|
241 val (f, es) = unravel (e1, [e2])
|
adamc@316
|
242
|
adamc@316
|
243 val (f, sm) = cifyExp (f, sm)
|
adamc@316
|
244 val (es, sm) = ListUtil.foldlMap cifyExp sm es
|
adamc@29
|
245 in
|
adamc@316
|
246 ((L'.EApp (f, es), loc), sm)
|
adamc@29
|
247 end
|
adamc@109
|
248 | L.EAbs _ => (ErrorMsg.errorAt loc "Anonymous function remains at code generation";
|
adamc@280
|
249 Print.prefaces' [("Function", MonoPrint.p_exp MonoEnv.empty eAll)];
|
adamc@109
|
250 (dummye, sm))
|
adamc@29
|
251
|
adamc@387
|
252 | L.EUnop (s, e1) =>
|
adamc@387
|
253 let
|
adamc@387
|
254 val (e1, sm) = cifyExp (e1, sm)
|
adamc@387
|
255 in
|
adamc@387
|
256 ((L'.EUnop (s, e1), loc), sm)
|
adamc@387
|
257 end
|
adamc@387
|
258 | L.EBinop (s, e1, e2) =>
|
adamc@387
|
259 let
|
adamc@387
|
260 val (e1, sm) = cifyExp (e1, sm)
|
adamc@387
|
261 val (e2, sm) = cifyExp (e2, sm)
|
adamc@387
|
262 in
|
adamc@387
|
263 ((L'.EBinop (s, e1, e2), loc), sm)
|
adamc@387
|
264 end
|
adamc@387
|
265
|
adamc@29
|
266 | L.ERecord xes =>
|
adamc@29
|
267 let
|
adamc@29
|
268 val old_xts = map (fn (x, _, t) => (x, t)) xes
|
adamc@29
|
269
|
adamc@29
|
270 val (xets, sm) = ListUtil.foldlMap (fn ((x, e, t), sm) =>
|
adamc@29
|
271 let
|
adamc@29
|
272 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
273 val (t, sm) = cifyTyp (t, sm)
|
adamc@29
|
274 in
|
adamc@29
|
275 ((x, e, t), sm)
|
adamc@29
|
276 end)
|
adamc@29
|
277 sm xes
|
adamc@29
|
278
|
adamc@29
|
279 val (sm, si) = Sm.find (sm, old_xts, map (fn (x, _, t) => (x, t)) xets)
|
adamc@29
|
280
|
adamc@29
|
281 val xes = map (fn (x, e, _) => (x, e)) xets
|
adamc@29
|
282 val xes = ListMergeSort.sort (fn ((x1, _), (x2, _)) => String.compare (x1, x2) = GREATER) xes
|
adamc@29
|
283 in
|
adamc@29
|
284 ((L'.ERecord (si, xes), loc), sm)
|
adamc@29
|
285 end
|
adamc@29
|
286 | L.EField (e, x) =>
|
adamc@29
|
287 let
|
adamc@29
|
288 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
289 in
|
adamc@29
|
290 ((L'.EField (e, x), loc), sm)
|
adamc@29
|
291 end
|
adamc@29
|
292
|
adamc@182
|
293 | L.ECase (e, pes, {disc, result}) =>
|
adamc@181
|
294 let
|
adamc@181
|
295 val (e, sm) = cifyExp (e, sm)
|
adamc@181
|
296 val (pes, sm) = ListUtil.foldlMap
|
adamc@181
|
297 (fn ((p, e), sm) =>
|
adamc@181
|
298 let
|
adamc@181
|
299 val (e, sm) = cifyExp (e, sm)
|
adamc@182
|
300 val (p, sm) = cifyPat (p, sm)
|
adamc@181
|
301 in
|
adamc@182
|
302 ((p, e), sm)
|
adamc@181
|
303 end) sm pes
|
adamc@182
|
304 val (disc, sm) = cifyTyp (disc, sm)
|
adamc@182
|
305 val (result, sm) = cifyTyp (result, sm)
|
adamc@181
|
306 in
|
adamc@182
|
307 ((L'.ECase (e, pes, {disc = disc, result = result}), loc), sm)
|
adamc@181
|
308 end
|
adamc@178
|
309
|
adamc@283
|
310 | L.EError (e, t) =>
|
adamc@283
|
311 let
|
adamc@283
|
312 val (e, sm) = cifyExp (e, sm)
|
adamc@283
|
313 val (t, sm) = cifyTyp (t, sm)
|
adamc@283
|
314 in
|
adamc@283
|
315 ((L'.EError (e, t), loc), sm)
|
adamc@283
|
316 end
|
adamc@283
|
317
|
adamc@180
|
318 | L.EStrcat (e1, e2) =>
|
adamc@180
|
319 let
|
adamc@180
|
320 val (e1, sm) = cifyExp (e1, sm)
|
adamc@180
|
321 val (e2, sm) = cifyExp (e2, sm)
|
adamc@180
|
322 in
|
adamc@180
|
323 ((L'.EFfiApp ("Basis", "strcat", [e1, e2]), loc), sm)
|
adamc@180
|
324 end
|
adamc@102
|
325
|
adamc@102
|
326 | L.EWrite e =>
|
adamc@102
|
327 let
|
adamc@102
|
328 val (e, sm) = cifyExp (e, sm)
|
adamc@102
|
329 in
|
adamc@102
|
330 ((L'.EWrite e, loc), sm)
|
adamc@102
|
331 end
|
adamc@102
|
332
|
adamc@106
|
333 | L.ESeq (e1, e2) =>
|
adamc@106
|
334 let
|
adamc@106
|
335 val (e1, sm) = cifyExp (e1, sm)
|
adamc@106
|
336 val (e2, sm) = cifyExp (e2, sm)
|
adamc@106
|
337 in
|
adamc@106
|
338 ((L'.ESeq (e1, e2), loc), sm)
|
adamc@106
|
339 end
|
adamc@106
|
340
|
adamc@269
|
341 | L.ELet (x, t, e1, e2) =>
|
adamc@269
|
342 let
|
adamc@269
|
343 val (t, sm) = cifyTyp (t, sm)
|
adamc@269
|
344 val (e1, sm) = cifyExp (e1, sm)
|
adamc@269
|
345 val (e2, sm) = cifyExp (e2, sm)
|
adamc@269
|
346 in
|
adamc@269
|
347 ((L'.ELet (x, t, e1, e2), loc), sm)
|
adamc@269
|
348 end
|
adamc@251
|
349
|
adamc@111
|
350 | L.EClosure _ => (ErrorMsg.errorAt loc "Nested closure remains in code generation";
|
adamc@111
|
351 (dummye, sm))
|
adamc@111
|
352
|
adamc@269
|
353 | L.EQuery {exps, tables, state, query, body, initial} =>
|
adamc@269
|
354 let
|
adamc@269
|
355 val (exps', sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
|
adamc@269
|
356 let
|
adamc@269
|
357 val (t, sm) = cifyTyp (t, sm)
|
adamc@269
|
358 in
|
adamc@269
|
359 ((x, t), sm)
|
adamc@269
|
360 end) sm exps
|
adamc@269
|
361 val (tables', sm) = ListUtil.foldlMap (fn ((x, xts), sm) =>
|
adamc@269
|
362 let
|
adamc@269
|
363 val (xts, sm) = ListUtil.foldlMap
|
adamc@269
|
364 (fn ((x, t), sm) =>
|
adamc@269
|
365 let
|
adamc@269
|
366 val (t, sm) = cifyTyp (t, sm)
|
adamc@269
|
367 in
|
adamc@269
|
368 ((x, t), sm)
|
adamc@269
|
369 end) sm xts
|
adamc@269
|
370 in
|
adamc@269
|
371 ((x, xts), sm)
|
adamc@269
|
372 end) sm tables
|
adamc@269
|
373
|
adamc@269
|
374 val row = exps @ map (fn (x, xts) => (x, (L.TRecord xts, loc))) tables
|
adamc@269
|
375 val row = ListMergeSort.sort (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER) row
|
adamc@269
|
376
|
adamc@269
|
377 val (tableRows, sm) = ListUtil.foldlMap (fn (((x, xts), (_, xts')), sm) =>
|
adamc@269
|
378 let
|
adamc@269
|
379 val (sm, rnum) = Sm.find (sm, xts, xts')
|
adamc@269
|
380 in
|
adamc@269
|
381 ((x, rnum), sm)
|
adamc@269
|
382 end)
|
adamc@269
|
383 sm (ListPair.zip (tables, tables'))
|
adamc@269
|
384 val row' = exps' @ map (fn (x, n) => (x, (L'.TRecord n, loc))) tableRows
|
adamc@269
|
385 val row' = ListMergeSort.sort (fn ((x, _), (y, _)) => String.compare (x, y) = GREATER) row'
|
adamc@269
|
386
|
adamc@269
|
387 val (sm, rnum) = Sm.find (sm, row, row')
|
adamc@269
|
388
|
adamc@269
|
389 val (state, sm) = cifyTyp (state, sm)
|
adamc@269
|
390 val (query, sm) = cifyExp (query, sm)
|
adamc@269
|
391 val (body, sm) = cifyExp (body, sm)
|
adamc@269
|
392 val (initial, sm) = cifyExp (initial, sm)
|
adamc@269
|
393 in
|
adamc@269
|
394 ((L'.EQuery {exps = exps', tables = tables', rnum = rnum, state = state,
|
adamc@282
|
395 query = query, body = body, initial = initial, prepared = NONE}, loc), sm)
|
adamc@269
|
396 end
|
adamc@269
|
397
|
adamc@307
|
398 | L.EDml e =>
|
adamc@307
|
399 let
|
adamc@307
|
400 val (e, sm) = cifyExp (e, sm)
|
adamc@307
|
401 in
|
adamc@307
|
402 ((L'.EDml {dml = e, prepared = NONE}, loc), sm)
|
adamc@307
|
403 end
|
adamc@307
|
404
|
adamc@338
|
405 | L.ENextval e =>
|
adamc@338
|
406 let
|
adamc@338
|
407 val (e, sm) = cifyExp (e, sm)
|
adamc@338
|
408 in
|
adamc@338
|
409 ((L'.ENextval {seq = e, prepared = NONE}, loc), sm)
|
adamc@338
|
410 end
|
adamc@338
|
411
|
adamc@252
|
412
|
adamc@29
|
413 fun cifyDecl ((d, loc), sm) =
|
adamc@29
|
414 case d of
|
adamc@165
|
415 L.DDatatype (x, n, xncs) =>
|
adamc@165
|
416 let
|
adamc@198
|
417 val dk = ElabUtil.classifyDatatype xncs
|
adamc@165
|
418 val (xncs, sm) = ListUtil.foldlMap (fn ((x, n, to), sm) =>
|
adamc@165
|
419 case to of
|
adamc@165
|
420 NONE => ((x, n, NONE), sm)
|
adamc@165
|
421 | SOME t =>
|
adamc@165
|
422 let
|
adamc@165
|
423 val (t, sm) = cifyTyp (t, sm)
|
adamc@165
|
424 in
|
adamc@165
|
425 ((x, n, SOME t), sm)
|
adamc@165
|
426 end) sm xncs
|
adamc@165
|
427 in
|
adamc@188
|
428 (SOME (L'.DDatatype (dk, x, n, xncs), loc), NONE, sm)
|
adamc@165
|
429 end
|
adamc@164
|
430
|
adamc@164
|
431 | L.DVal (x, n, t, e, _) =>
|
adamc@29
|
432 let
|
adamc@29
|
433 val (t, sm) = cifyTyp (t, sm)
|
adamc@109
|
434
|
adamc@109
|
435 val (d, sm) = case #1 t of
|
adamc@121
|
436 L'.TFun _ =>
|
adamc@121
|
437 let
|
adamc@121
|
438 fun unravel (tAll as (t, _), eAll as (e, _)) =
|
adamc@121
|
439 case (t, e) of
|
adamc@121
|
440 (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
|
adamc@121
|
441 let
|
adamc@121
|
442 val (args, t, e) = unravel (ran, e)
|
adamc@121
|
443 in
|
adamc@121
|
444 ((ax, dom) :: args, t, e)
|
adamc@121
|
445 end
|
adamc@121
|
446 | (L'.TFun _, _) =>
|
adamc@121
|
447 (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
|
adamc@121
|
448 ([], tAll, eAll))
|
adamc@121
|
449 | _ => ([], tAll, eAll)
|
adamc@121
|
450
|
adamc@121
|
451 val (args, ran, e) = unravel (t, e)
|
adamc@121
|
452 val (e, sm) = cifyExp (e, sm)
|
adamc@121
|
453 in
|
adamc@121
|
454 (L'.DFun (x, n, args, ran, e), sm)
|
adamc@121
|
455 end
|
adamc@121
|
456
|
adamc@109
|
457 | _ =>
|
adamc@109
|
458 let
|
adamc@109
|
459 val (e, sm) = cifyExp (e, sm)
|
adamc@109
|
460 in
|
adamc@109
|
461 (L'.DVal (x, n, t, e), sm)
|
adamc@109
|
462 end
|
adamc@29
|
463 in
|
adamc@109
|
464 (SOME (d, loc), NONE, sm)
|
adamc@29
|
465 end
|
adamc@129
|
466 | L.DValRec vis =>
|
adamc@129
|
467 let
|
adamc@129
|
468 val (vis, sm) = ListUtil.foldlMap
|
adamc@129
|
469 (fn ((x, n, t, e, _), sm) =>
|
adamc@129
|
470 let
|
adamc@129
|
471 val (t, sm) = cifyTyp (t, sm)
|
adamc@129
|
472
|
adamc@129
|
473 fun unravel (tAll as (t, _), eAll as (e, _)) =
|
adamc@129
|
474 case (t, e) of
|
adamc@129
|
475 (L'.TFun (dom, ran), L.EAbs (ax, _, _, e)) =>
|
adamc@129
|
476 let
|
adamc@129
|
477 val (args, t, e) = unravel (ran, e)
|
adamc@129
|
478 in
|
adamc@129
|
479 ((ax, dom) :: args, t, e)
|
adamc@129
|
480 end
|
adamc@129
|
481 | (L'.TFun _, _) =>
|
adamc@129
|
482 (ErrorMsg.errorAt loc "Function isn't explicit at code generation";
|
adamc@129
|
483 ([], tAll, eAll))
|
adamc@129
|
484 | _ => ([], tAll, eAll)
|
adamc@129
|
485
|
adamc@129
|
486 val (args, ran, e) = unravel (t, e)
|
adamc@129
|
487 val (e, sm) = cifyExp (e, sm)
|
adamc@129
|
488 in
|
adamc@129
|
489 ((x, n, args, ran, e), sm)
|
adamc@129
|
490 end)
|
adamc@129
|
491 sm vis
|
adamc@129
|
492 in
|
adamc@129
|
493 (SOME (L'.DFunRec vis, loc), NONE, sm)
|
adamc@129
|
494 end
|
adamc@129
|
495
|
adamc@144
|
496 | L.DExport (ek, s, n, ts) =>
|
adamc@120
|
497 let
|
adamc@120
|
498 val (ts, sm) = ListUtil.foldlMap cifyTyp sm ts
|
adamc@120
|
499 in
|
adamc@144
|
500 (NONE, SOME (ek, "/" ^ s, n, ts), sm)
|
adamc@120
|
501 end
|
adamc@29
|
502
|
adamc@273
|
503 | L.DTable (s, xts) =>
|
adamc@273
|
504 let
|
adamc@273
|
505 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
|
adamc@273
|
506 let
|
adamc@273
|
507 val (t, sm) = cifyTyp (t, sm)
|
adamc@273
|
508 in
|
adamc@273
|
509 ((x, t), sm)
|
adamc@273
|
510 end) sm xts
|
adamc@273
|
511 in
|
adamc@273
|
512 (SOME (L'.DTable (s, xts), loc), NONE, sm)
|
adamc@273
|
513 end
|
adamc@338
|
514 | L.DSequence s =>
|
adamc@338
|
515 (SOME (L'.DSequence s, loc), NONE, sm)
|
adamc@271
|
516 | L.DDatabase s => (SOME (L'.DDatabase s, loc), NONE, sm)
|
adamc@271
|
517
|
adamc@29
|
518 fun cjrize ds =
|
adamc@29
|
519 let
|
adamc@196
|
520 val (dsF, ds, ps, sm) = foldl (fn (d, (dsF, ds, ps, sm)) =>
|
adamc@196
|
521 let
|
adamc@196
|
522 val (dop, pop, sm) = cifyDecl (d, sm)
|
adamc@196
|
523 val (dsF, ds) = case dop of
|
adamc@196
|
524 NONE => (dsF, ds)
|
adamc@196
|
525 | SOME (d as (L'.DDatatype (dk, x, n, _), loc)) =>
|
adamc@196
|
526 ((L'.DDatatypeForward (dk, x, n), loc) :: dsF,
|
adamc@196
|
527 d :: ds)
|
adamc@196
|
528 | SOME d => (dsF, d :: ds)
|
adamc@196
|
529 val ps = case pop of
|
adamc@196
|
530 NONE => ps
|
adamc@196
|
531 | SOME p => p :: ps
|
adamc@196
|
532 in
|
adamc@196
|
533 (dsF, ds, ps, sm)
|
adamc@196
|
534 end)
|
adamc@196
|
535 ([], [], [], Sm.empty) ds
|
adamc@29
|
536 in
|
adamc@196
|
537 (List.revAppend (dsF,
|
adamc@196
|
538 List.revAppend (map (fn v => (L'.DStruct v, ErrorMsg.dummySpan)) (Sm.declares sm),
|
adamc@196
|
539 rev ds)),
|
adamc@101
|
540 ps)
|
adamc@29
|
541 end
|
adamc@29
|
542
|
adamc@29
|
543 end
|