adamc@2
|
1 (* Copyright (c) 2008, Adam Chlipala
|
adamc@2
|
2 * All rights reserved.
|
adamc@2
|
3 *
|
adamc@2
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@2
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@2
|
6 *
|
adamc@2
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@2
|
8 * this list of conditions and the following disclaimer.
|
adamc@2
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@2
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@2
|
11 * and/or other materials provided with the distribution.
|
adamc@2
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@2
|
13 * derived from this software without specific prior written permission.
|
adamc@2
|
14 *
|
adamc@2
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@2
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@2
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@2
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@2
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@2
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@2
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@2
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@2
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@2
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@2
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@2
|
26 *)
|
adamc@2
|
27
|
adamc@2
|
28 structure ElabUtil :> ELAB_UTIL = struct
|
adamc@2
|
29
|
adamc@2
|
30 open Elab
|
adamc@2
|
31
|
adamc@188
|
32 fun classifyDatatype xncs =
|
adamc@198
|
33 case xncs of
|
adamc@198
|
34 [(_, _, NONE), (_, _, SOME _)] => Option
|
adamc@198
|
35 | [(_, _, SOME _), (_, _, NONE)] => Option
|
adamc@198
|
36 | _ =>
|
adamc@198
|
37 if List.all (fn (_, _, NONE) => true | _ => false) xncs then
|
adamc@198
|
38 Enum
|
adamc@198
|
39 else
|
adamc@198
|
40 Default
|
adamc@188
|
41
|
adamc@2
|
42 structure S = Search
|
adamc@2
|
43
|
adamc@2
|
44 structure Kind = struct
|
adamc@2
|
45
|
adamc@6
|
46 fun mapfold f =
|
adamc@2
|
47 let
|
adamc@2
|
48 fun mfk k acc =
|
adamc@2
|
49 S.bindP (mfk' k acc, f)
|
adamc@2
|
50
|
adamc@2
|
51 and mfk' (kAll as (k, loc)) =
|
adamc@2
|
52 case k of
|
adamc@2
|
53 KType => S.return2 kAll
|
adamc@2
|
54
|
adamc@2
|
55 | KArrow (k1, k2) =>
|
adamc@2
|
56 S.bind2 (mfk k1,
|
adamc@2
|
57 fn k1' =>
|
adamc@2
|
58 S.map2 (mfk k2,
|
adamc@2
|
59 fn k2' =>
|
adamc@2
|
60 (KArrow (k1', k2'), loc)))
|
adamc@2
|
61
|
adamc@2
|
62 | KName => S.return2 kAll
|
adamc@2
|
63
|
adamc@2
|
64 | KRecord k =>
|
adamc@2
|
65 S.map2 (mfk k,
|
adamc@2
|
66 fn k' =>
|
adamc@2
|
67 (KRecord k', loc))
|
adamc@2
|
68
|
adamc@82
|
69 | KUnit => S.return2 kAll
|
adamc@82
|
70
|
adamc@207
|
71 | KTuple ks =>
|
adamc@207
|
72 S.map2 (ListUtil.mapfold mfk ks,
|
adamc@207
|
73 fn ks' =>
|
adamc@207
|
74 (KTuple ks', loc))
|
adamc@207
|
75
|
adamc@2
|
76 | KError => S.return2 kAll
|
adamc@2
|
77
|
adamc@76
|
78 | KUnif (_, _, ref (SOME k)) => mfk' k
|
adamc@2
|
79 | KUnif _ => S.return2 kAll
|
adamc@2
|
80 in
|
adamc@2
|
81 mfk
|
adamc@2
|
82 end
|
adamc@2
|
83
|
adamc@2
|
84 fun exists f k =
|
adamc@6
|
85 case mapfold (fn k => fn () =>
|
adamc@6
|
86 if f k then
|
adamc@6
|
87 S.Return ()
|
adamc@6
|
88 else
|
adamc@6
|
89 S.Continue (k, ())) k () of
|
adamc@6
|
90 S.Return _ => true
|
adamc@6
|
91 | S.Continue _ => false
|
adamc@6
|
92
|
adamc@6
|
93 end
|
adamc@6
|
94
|
adamc@6
|
95 structure Con = struct
|
adamc@6
|
96
|
adamc@11
|
97 datatype binder =
|
adamc@11
|
98 Rel of string * Elab.kind
|
adamc@329
|
99 | Named of string * int * Elab.kind
|
adamc@11
|
100
|
adamc@11
|
101 fun mapfoldB {kind = fk, con = fc, bind} =
|
adamc@6
|
102 let
|
adamc@6
|
103 val mfk = Kind.mapfold fk
|
adamc@6
|
104
|
adamc@11
|
105 fun mfc ctx c acc =
|
adamc@11
|
106 S.bindP (mfc' ctx c acc, fc ctx)
|
adamc@6
|
107
|
adamc@11
|
108 and mfc' ctx (cAll as (c, loc)) =
|
adamc@6
|
109 case c of
|
adamc@6
|
110 TFun (c1, c2) =>
|
adamc@11
|
111 S.bind2 (mfc ctx c1,
|
adamc@6
|
112 fn c1' =>
|
adamc@11
|
113 S.map2 (mfc ctx c2,
|
adamc@6
|
114 fn c2' =>
|
adamc@6
|
115 (TFun (c1', c2'), loc)))
|
adamc@6
|
116 | TCFun (e, x, k, c) =>
|
adamc@6
|
117 S.bind2 (mfk k,
|
adamc@6
|
118 fn k' =>
|
adamc@11
|
119 S.map2 (mfc (bind (ctx, Rel (x, k))) c,
|
adamc@6
|
120 fn c' =>
|
adamc@6
|
121 (TCFun (e, x, k', c'), loc)))
|
adamc@345
|
122 | CDisjoint (ai, c1, c2, c3) =>
|
adamc@85
|
123 S.bind2 (mfc ctx c1,
|
adamc@85
|
124 fn c1' =>
|
adamc@85
|
125 S.bind2 (mfc ctx c2,
|
adamc@85
|
126 fn c2' =>
|
adamc@85
|
127 S.map2 (mfc ctx c3,
|
adamc@85
|
128 fn c3' =>
|
adamc@345
|
129 (CDisjoint (ai, c1', c2', c3'), loc))))
|
adamc@6
|
130 | TRecord c =>
|
adamc@11
|
131 S.map2 (mfc ctx c,
|
adamc@6
|
132 fn c' =>
|
adamc@6
|
133 (TRecord c', loc))
|
adamc@6
|
134
|
adamc@6
|
135 | CRel _ => S.return2 cAll
|
adamc@6
|
136 | CNamed _ => S.return2 cAll
|
adamc@34
|
137 | CModProj _ => S.return2 cAll
|
adamc@6
|
138 | CApp (c1, c2) =>
|
adamc@11
|
139 S.bind2 (mfc ctx c1,
|
adamc@6
|
140 fn c1' =>
|
adamc@11
|
141 S.map2 (mfc ctx c2,
|
adamc@6
|
142 fn c2' =>
|
adamc@6
|
143 (CApp (c1', c2'), loc)))
|
adamc@8
|
144 | CAbs (x, k, c) =>
|
adamc@6
|
145 S.bind2 (mfk k,
|
adamc@6
|
146 fn k' =>
|
adamc@11
|
147 S.map2 (mfc (bind (ctx, Rel (x, k))) c,
|
adamc@6
|
148 fn c' =>
|
adamc@8
|
149 (CAbs (x, k', c'), loc)))
|
adamc@6
|
150
|
adamc@6
|
151 | CName _ => S.return2 cAll
|
adamc@6
|
152
|
adamc@6
|
153 | CRecord (k, xcs) =>
|
adamc@6
|
154 S.bind2 (mfk k,
|
adamc@6
|
155 fn k' =>
|
adamc@6
|
156 S.map2 (ListUtil.mapfold (fn (x, c) =>
|
adamc@11
|
157 S.bind2 (mfc ctx x,
|
adamc@6
|
158 fn x' =>
|
adamc@11
|
159 S.map2 (mfc ctx c,
|
adamc@6
|
160 fn c' =>
|
adamc@6
|
161 (x', c'))))
|
adamc@6
|
162 xcs,
|
adamc@6
|
163 fn xcs' =>
|
adamc@6
|
164 (CRecord (k', xcs'), loc)))
|
adamc@6
|
165 | CConcat (c1, c2) =>
|
adamc@11
|
166 S.bind2 (mfc ctx c1,
|
adamc@6
|
167 fn c1' =>
|
adamc@11
|
168 S.map2 (mfc ctx c2,
|
adamc@6
|
169 fn c2' =>
|
adamc@6
|
170 (CConcat (c1', c2'), loc)))
|
adamc@621
|
171 | CMap (k1, k2) =>
|
adamc@67
|
172 S.bind2 (mfk k1,
|
adamc@67
|
173 fn k1' =>
|
adamc@67
|
174 S.map2 (mfk k2,
|
adamc@67
|
175 fn k2' =>
|
adamc@621
|
176 (CMap (k1', k2'), loc)))
|
adamc@6
|
177
|
adamc@82
|
178 | CUnit => S.return2 cAll
|
adamc@82
|
179
|
adamc@207
|
180 | CTuple cs =>
|
adamc@207
|
181 S.map2 (ListUtil.mapfold (mfc ctx) cs,
|
adamc@207
|
182 fn cs' =>
|
adamc@207
|
183 (CTuple cs', loc))
|
adamc@207
|
184
|
adamc@207
|
185 | CProj (c, n) =>
|
adamc@207
|
186 S.map2 (mfc ctx c,
|
adamc@207
|
187 fn c' =>
|
adamc@207
|
188 (CProj (c', n), loc))
|
adamc@207
|
189
|
adamc@6
|
190 | CError => S.return2 cAll
|
adamc@76
|
191 | CUnif (_, _, _, ref (SOME c)) => mfc' ctx c
|
adamc@6
|
192 | CUnif _ => S.return2 cAll
|
adamc@6
|
193 in
|
adamc@6
|
194 mfc
|
adamc@6
|
195 end
|
adamc@6
|
196
|
adamc@11
|
197 fun mapfold {kind = fk, con = fc} =
|
adamc@11
|
198 mapfoldB {kind = fk,
|
adamc@11
|
199 con = fn () => fc,
|
adamc@11
|
200 bind = fn ((), _) => ()} ()
|
adamc@11
|
201
|
adamc@11
|
202 fun mapB {kind, con, bind} ctx c =
|
adamc@11
|
203 case mapfoldB {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@11
|
204 con = fn ctx => fn c => fn () => S.Continue (con ctx c, ()),
|
adamc@11
|
205 bind = bind} ctx c () of
|
adamc@11
|
206 S.Continue (c, ()) => c
|
adamc@34
|
207 | S.Return _ => raise Fail "ElabUtil.Con.mapB: Impossible"
|
adamc@34
|
208
|
adamc@34
|
209 fun map {kind, con} s =
|
adamc@34
|
210 case mapfold {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@34
|
211 con = fn c => fn () => S.Continue (con c, ())} s () of
|
adamc@34
|
212 S.Return () => raise Fail "ElabUtil.Con.map: Impossible"
|
adamc@34
|
213 | S.Continue (s, ()) => s
|
adamc@11
|
214
|
adamc@6
|
215 fun exists {kind, con} k =
|
adamc@6
|
216 case mapfold {kind = fn k => fn () =>
|
adamc@6
|
217 if kind k then
|
adamc@6
|
218 S.Return ()
|
adamc@6
|
219 else
|
adamc@6
|
220 S.Continue (k, ()),
|
adamc@6
|
221 con = fn c => fn () =>
|
adamc@6
|
222 if con c then
|
adamc@6
|
223 S.Return ()
|
adamc@6
|
224 else
|
adamc@6
|
225 S.Continue (c, ())} k () of
|
adamc@2
|
226 S.Return _ => true
|
adamc@2
|
227 | S.Continue _ => false
|
adamc@2
|
228
|
adamc@448
|
229 fun foldB {kind, con, bind} ctx st c =
|
adamc@448
|
230 case mapfoldB {kind = fn k => fn st => S.Continue (k, kind (k, st)),
|
adamc@448
|
231 con = fn ctx => fn c => fn st => S.Continue (c, con (ctx, c, st)),
|
adamc@448
|
232 bind = bind} ctx c st of
|
adamc@448
|
233 S.Continue (_, st) => st
|
adamc@448
|
234 | S.Return _ => raise Fail "ElabUtil.Con.foldB: Impossible"
|
adamc@448
|
235
|
adamc@2
|
236 end
|
adamc@2
|
237
|
adamc@10
|
238 structure Exp = struct
|
adamc@10
|
239
|
adamc@11
|
240 datatype binder =
|
adamc@11
|
241 RelC of string * Elab.kind
|
adamc@329
|
242 | NamedC of string * int * Elab.kind
|
adamc@11
|
243 | RelE of string * Elab.con
|
adamc@11
|
244 | NamedE of string * Elab.con
|
adamc@11
|
245
|
adamc@11
|
246 fun mapfoldB {kind = fk, con = fc, exp = fe, bind} =
|
adamc@10
|
247 let
|
adamc@10
|
248 val mfk = Kind.mapfold fk
|
adamc@10
|
249
|
adamc@11
|
250 fun bind' (ctx, b) =
|
adamc@11
|
251 let
|
adamc@11
|
252 val b' = case b of
|
adamc@11
|
253 Con.Rel x => RelC x
|
adamc@11
|
254 | Con.Named x => NamedC x
|
adamc@11
|
255 in
|
adamc@11
|
256 bind (ctx, b')
|
adamc@11
|
257 end
|
adamc@11
|
258 val mfc = Con.mapfoldB {kind = fk, con = fc, bind = bind'}
|
adamc@10
|
259
|
adamc@11
|
260 fun mfe ctx e acc =
|
adamc@11
|
261 S.bindP (mfe' ctx e acc, fe ctx)
|
adamc@11
|
262
|
adamc@11
|
263 and mfe' ctx (eAll as (e, loc)) =
|
adamc@10
|
264 case e of
|
adamc@14
|
265 EPrim _ => S.return2 eAll
|
adamc@14
|
266 | ERel _ => S.return2 eAll
|
adamc@10
|
267 | ENamed _ => S.return2 eAll
|
adamc@34
|
268 | EModProj _ => S.return2 eAll
|
adamc@10
|
269 | EApp (e1, e2) =>
|
adamc@11
|
270 S.bind2 (mfe ctx e1,
|
adamc@10
|
271 fn e1' =>
|
adamc@11
|
272 S.map2 (mfe ctx e2,
|
adamc@10
|
273 fn e2' =>
|
adamc@10
|
274 (EApp (e1', e2'), loc)))
|
adamc@26
|
275 | EAbs (x, dom, ran, e) =>
|
adamc@26
|
276 S.bind2 (mfc ctx dom,
|
adamc@26
|
277 fn dom' =>
|
adamc@26
|
278 S.bind2 (mfc ctx ran,
|
adamc@26
|
279 fn ran' =>
|
adamc@26
|
280 S.map2 (mfe (bind (ctx, RelE (x, dom'))) e,
|
adamc@26
|
281 fn e' =>
|
adamc@26
|
282 (EAbs (x, dom', ran', e'), loc))))
|
adamc@26
|
283
|
adamc@10
|
284 | ECApp (e, c) =>
|
adamc@11
|
285 S.bind2 (mfe ctx e,
|
adamc@10
|
286 fn e' =>
|
adamc@11
|
287 S.map2 (mfc ctx c,
|
adamc@10
|
288 fn c' =>
|
adamc@10
|
289 (ECApp (e', c'), loc)))
|
adamc@10
|
290 | ECAbs (expl, x, k, e) =>
|
adamc@10
|
291 S.bind2 (mfk k,
|
adamc@10
|
292 fn k' =>
|
adamc@11
|
293 S.map2 (mfe (bind (ctx, RelC (x, k))) e,
|
adamc@10
|
294 fn e' =>
|
adamc@10
|
295 (ECAbs (expl, x, k', e'), loc)))
|
adamc@10
|
296
|
adamc@12
|
297 | ERecord xes =>
|
adamc@29
|
298 S.map2 (ListUtil.mapfold (fn (x, e, t) =>
|
adamc@12
|
299 S.bind2 (mfc ctx x,
|
adamc@12
|
300 fn x' =>
|
adamc@29
|
301 S.bind2 (mfe ctx e,
|
adamc@12
|
302 fn e' =>
|
adamc@29
|
303 S.map2 (mfc ctx t,
|
adamc@29
|
304 fn t' =>
|
adamc@29
|
305 (x', e', t')))))
|
adamc@12
|
306 xes,
|
adamc@12
|
307 fn xes' =>
|
adamc@12
|
308 (ERecord xes', loc))
|
adamc@12
|
309 | EField (e, c, {field, rest}) =>
|
adamc@12
|
310 S.bind2 (mfe ctx e,
|
adamc@12
|
311 fn e' =>
|
adamc@12
|
312 S.bind2 (mfc ctx c,
|
adamc@12
|
313 fn c' =>
|
adamc@12
|
314 S.bind2 (mfc ctx field,
|
adamc@12
|
315 fn field' =>
|
adamc@12
|
316 S.map2 (mfc ctx rest,
|
adamc@12
|
317 fn rest' =>
|
adamc@12
|
318 (EField (e', c', {field = field', rest = rest'}), loc)))))
|
adamc@445
|
319 | EConcat (e1, c1, e2, c2) =>
|
adamc@339
|
320 S.bind2 (mfe ctx e1,
|
adamc@339
|
321 fn e1' =>
|
adamc@445
|
322 S.bind2 (mfc ctx c1,
|
adamc@445
|
323 fn c1' =>
|
adamc@339
|
324 S.bind2 (mfe ctx e2,
|
adamc@339
|
325 fn e2' =>
|
adamc@445
|
326 S.map2 (mfc ctx c2,
|
adamc@445
|
327 fn c2' =>
|
adamc@445
|
328 (EConcat (e1', c1', e2', c2'),
|
adamc@445
|
329 loc)))))
|
adamc@149
|
330 | ECut (e, c, {field, rest}) =>
|
adamc@149
|
331 S.bind2 (mfe ctx e,
|
adamc@149
|
332 fn e' =>
|
adamc@149
|
333 S.bind2 (mfc ctx c,
|
adamc@149
|
334 fn c' =>
|
adamc@149
|
335 S.bind2 (mfc ctx field,
|
adamc@149
|
336 fn field' =>
|
adamc@149
|
337 S.map2 (mfc ctx rest,
|
adamc@149
|
338 fn rest' =>
|
adamc@149
|
339 (ECut (e', c', {field = field', rest = rest'}), loc)))))
|
adamc@12
|
340
|
adamc@493
|
341 | ECutMulti (e, c, {rest}) =>
|
adamc@493
|
342 S.bind2 (mfe ctx e,
|
adamc@493
|
343 fn e' =>
|
adamc@493
|
344 S.bind2 (mfc ctx c,
|
adamc@493
|
345 fn c' =>
|
adamc@493
|
346 S.map2 (mfc ctx rest,
|
adamc@493
|
347 fn rest' =>
|
adamc@493
|
348 (ECutMulti (e', c', {rest = rest'}), loc))))
|
adamc@493
|
349
|
adamc@71
|
350 | EFold k =>
|
adamc@71
|
351 S.map2 (mfk k,
|
adamc@71
|
352 fn k' =>
|
adamc@71
|
353 (EFold k', loc))
|
adamc@71
|
354
|
adamc@182
|
355 | ECase (e, pes, {disc, result}) =>
|
adamc@171
|
356 S.bind2 (mfe ctx e,
|
adamc@171
|
357 fn e' =>
|
adamc@171
|
358 S.bind2 (ListUtil.mapfold (fn (p, e) =>
|
adamc@448
|
359 let
|
adamc@448
|
360 fun pb ((p, _), ctx) =
|
adamc@448
|
361 case p of
|
adamc@448
|
362 PWild => ctx
|
adamc@448
|
363 | PVar (x, t) => bind (ctx, RelE (x, t))
|
adamc@448
|
364 | PPrim _ => ctx
|
adamc@448
|
365 | PCon (_, _, _, NONE) => ctx
|
adamc@448
|
366 | PCon (_, _, _, SOME p) => pb (p, ctx)
|
adamc@448
|
367 | PRecord xps => foldl (fn ((_, p, _), ctx) =>
|
adamc@448
|
368 pb (p, ctx)) ctx xps
|
adamc@448
|
369 in
|
adamc@448
|
370 S.map2 (mfe (pb (p, ctx)) e,
|
adamc@448
|
371 fn e' => (p, e'))
|
adamc@448
|
372 end) pes,
|
adamc@171
|
373 fn pes' =>
|
adamc@182
|
374 S.bind2 (mfc ctx disc,
|
adamc@182
|
375 fn disc' =>
|
adamc@182
|
376 S.map2 (mfc ctx result,
|
adamc@182
|
377 fn result' =>
|
adamc@182
|
378 (ECase (e', pes', {disc = disc', result = result'}), loc)))))
|
adamc@171
|
379
|
adamc@10
|
380 | EError => S.return2 eAll
|
adamc@228
|
381 | EUnif (ref (SOME e)) => mfe ctx e
|
adamc@228
|
382 | EUnif _ => S.return2 eAll
|
adamc@447
|
383
|
adamc@447
|
384 | ELet (des, e) =>
|
adamc@447
|
385 let
|
adamc@447
|
386 val (des, ctx) = foldl (fn (ed, (des, ctx)) =>
|
adamc@453
|
387 let
|
adamc@453
|
388 val ctx' =
|
adamc@453
|
389 case #1 ed of
|
adamc@453
|
390 EDVal (x, t, _) => bind (ctx, RelE (x, t))
|
adamc@453
|
391 | EDValRec vis =>
|
adamc@453
|
392 foldl (fn ((x, t, _), ctx) => bind (ctx, RelE (x, t))) ctx vis
|
adamc@453
|
393 in
|
adamc@453
|
394 (S.bind2 (des,
|
adamc@453
|
395 fn des' =>
|
adamc@453
|
396 S.map2 (mfed ctx ed,
|
adamc@447
|
397 fn ed' => des' @ [ed'])),
|
adamc@453
|
398 ctx')
|
adamc@453
|
399 end)
|
adamc@447
|
400 (S.return2 [], ctx) des
|
adamc@447
|
401 in
|
adamc@447
|
402 S.bind2 (des,
|
adamc@447
|
403 fn des' =>
|
adamc@447
|
404 S.map2 (mfe ctx e,
|
adamc@447
|
405 fn e' =>
|
adamc@447
|
406 (ELet (des', e'), loc)))
|
adamc@447
|
407 end
|
adamc@447
|
408
|
adamc@447
|
409 and mfed ctx (dAll as (d, loc)) =
|
adamc@447
|
410 case d of
|
adamc@447
|
411 EDVal vi =>
|
adamc@447
|
412 S.map2 (mfvi ctx vi,
|
adamc@447
|
413 fn vi' =>
|
adamc@447
|
414 (EDVal vi', loc))
|
adamc@447
|
415 | EDValRec vis =>
|
adamc@447
|
416 let
|
adamc@453
|
417 val ctx = foldl (fn ((x, t, _), ctx) => bind (ctx, RelE (x, t))) ctx vis
|
adamc@447
|
418 in
|
adamc@447
|
419 S.map2 (ListUtil.mapfold (mfvi ctx) vis,
|
adamc@447
|
420 fn vis' =>
|
adamc@447
|
421 (EDValRec vis', loc))
|
adamc@447
|
422 end
|
adamc@447
|
423
|
adamc@447
|
424 and mfvi ctx (x, c, e) =
|
adamc@447
|
425 S.bind2 (mfc ctx c,
|
adamc@447
|
426 fn c' =>
|
adamc@447
|
427 S.map2 (mfe ctx e,
|
adamc@447
|
428 fn e' =>
|
adamc@447
|
429 (x, c', e')))
|
adamc@10
|
430 in
|
adamc@10
|
431 mfe
|
adamc@10
|
432 end
|
adamc@10
|
433
|
adamc@11
|
434 fun mapfold {kind = fk, con = fc, exp = fe} =
|
adamc@11
|
435 mapfoldB {kind = fk,
|
adamc@11
|
436 con = fn () => fc,
|
adamc@11
|
437 exp = fn () => fe,
|
adamc@11
|
438 bind = fn ((), _) => ()} ()
|
adamc@11
|
439
|
adamc@10
|
440 fun exists {kind, con, exp} k =
|
adamc@10
|
441 case mapfold {kind = fn k => fn () =>
|
adamc@10
|
442 if kind k then
|
adamc@10
|
443 S.Return ()
|
adamc@10
|
444 else
|
adamc@10
|
445 S.Continue (k, ()),
|
adamc@10
|
446 con = fn c => fn () =>
|
adamc@10
|
447 if con c then
|
adamc@10
|
448 S.Return ()
|
adamc@10
|
449 else
|
adamc@10
|
450 S.Continue (c, ()),
|
adamc@10
|
451 exp = fn e => fn () =>
|
adamc@10
|
452 if exp e then
|
adamc@10
|
453 S.Return ()
|
adamc@10
|
454 else
|
adamc@10
|
455 S.Continue (e, ())} k () of
|
adamc@10
|
456 S.Return _ => true
|
adamc@10
|
457 | S.Continue _ => false
|
adamc@10
|
458
|
adamc@211
|
459 fun mapB {kind, con, exp, bind} ctx e =
|
adamc@211
|
460 case mapfoldB {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@211
|
461 con = fn ctx => fn c => fn () => S.Continue (con ctx c, ()),
|
adamc@211
|
462 exp = fn ctx => fn e => fn () => S.Continue (exp ctx e, ()),
|
adamc@211
|
463 bind = bind} ctx e () of
|
adamc@211
|
464 S.Continue (e, ()) => e
|
adamc@211
|
465 | S.Return _ => raise Fail "ElabUtil.Exp.mapB: Impossible"
|
adamc@211
|
466
|
adamc@448
|
467 fun foldB {kind, con, exp, bind} ctx st e =
|
adamc@448
|
468 case mapfoldB {kind = fn k => fn st => S.Continue (k, kind (k, st)),
|
adamc@448
|
469 con = fn ctx => fn c => fn st => S.Continue (c, con (ctx, c, st)),
|
adamc@448
|
470 exp = fn ctx => fn e => fn st => S.Continue (e, exp (ctx, e, st)),
|
adamc@448
|
471 bind = bind} ctx e st of
|
adamc@448
|
472 S.Continue (_, st) => st
|
adamc@448
|
473 | S.Return _ => raise Fail "ElabUtil.Exp.foldB: Impossible"
|
adamc@448
|
474
|
adamc@10
|
475 end
|
adamc@10
|
476
|
adamc@34
|
477 structure Sgn = struct
|
adamc@34
|
478
|
adamc@34
|
479 datatype binder =
|
adamc@34
|
480 RelC of string * Elab.kind
|
adamc@329
|
481 | NamedC of string * int * Elab.kind
|
adamc@34
|
482 | Str of string * Elab.sgn
|
adamc@59
|
483 | Sgn of string * Elab.sgn
|
adamc@34
|
484
|
adamc@34
|
485 fun mapfoldB {kind, con, sgn_item, sgn, bind} =
|
adamc@34
|
486 let
|
adamc@34
|
487 fun bind' (ctx, b) =
|
adamc@34
|
488 let
|
adamc@34
|
489 val b' = case b of
|
adamc@34
|
490 Con.Rel x => RelC x
|
adamc@34
|
491 | Con.Named x => NamedC x
|
adamc@34
|
492 in
|
adamc@34
|
493 bind (ctx, b')
|
adamc@34
|
494 end
|
adamc@34
|
495 val con = Con.mapfoldB {kind = kind, con = con, bind = bind'}
|
adamc@34
|
496
|
adamc@34
|
497 val kind = Kind.mapfold kind
|
adamc@34
|
498
|
adamc@34
|
499 fun sgi ctx si acc =
|
adamc@34
|
500 S.bindP (sgi' ctx si acc, sgn_item ctx)
|
adamc@34
|
501
|
adamc@156
|
502 and sgi' ctx (siAll as (si, loc)) =
|
adamc@34
|
503 case si of
|
adamc@34
|
504 SgiConAbs (x, n, k) =>
|
adamc@34
|
505 S.map2 (kind k,
|
adamc@34
|
506 fn k' =>
|
adamc@34
|
507 (SgiConAbs (x, n, k'), loc))
|
adamc@34
|
508 | SgiCon (x, n, k, c) =>
|
adamc@34
|
509 S.bind2 (kind k,
|
adamc@34
|
510 fn k' =>
|
adamc@34
|
511 S.map2 (con ctx c,
|
adamc@34
|
512 fn c' =>
|
adamc@34
|
513 (SgiCon (x, n, k', c'), loc)))
|
adamc@191
|
514 | SgiDatatype (x, n, xs, xncs) =>
|
adamc@156
|
515 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@156
|
516 case c of
|
adamc@156
|
517 NONE => S.return2 (x, n, c)
|
adamc@156
|
518 | SOME c =>
|
adamc@156
|
519 S.map2 (con ctx c,
|
adamc@156
|
520 fn c' => (x, n, SOME c'))) xncs,
|
adamc@156
|
521 fn xncs' =>
|
adamc@191
|
522 (SgiDatatype (x, n, xs, xncs'), loc))
|
adamc@191
|
523 | SgiDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
|
adamc@162
|
524 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@162
|
525 case c of
|
adamc@162
|
526 NONE => S.return2 (x, n, c)
|
adamc@162
|
527 | SOME c =>
|
adamc@162
|
528 S.map2 (con ctx c,
|
adamc@162
|
529 fn c' => (x, n, SOME c'))) xncs,
|
adamc@162
|
530 fn xncs' =>
|
adamc@191
|
531 (SgiDatatypeImp (x, n, m1, ms, s, xs, xncs'), loc))
|
adamc@34
|
532 | SgiVal (x, n, c) =>
|
adamc@34
|
533 S.map2 (con ctx c,
|
adamc@34
|
534 fn c' =>
|
adamc@34
|
535 (SgiVal (x, n, c'), loc))
|
adamc@34
|
536 | SgiStr (x, n, s) =>
|
adamc@34
|
537 S.map2 (sg ctx s,
|
adamc@34
|
538 fn s' =>
|
adamc@34
|
539 (SgiStr (x, n, s'), loc))
|
adamc@59
|
540 | SgiSgn (x, n, s) =>
|
adamc@59
|
541 S.map2 (sg ctx s,
|
adamc@59
|
542 fn s' =>
|
adamc@59
|
543 (SgiSgn (x, n, s'), loc))
|
adamc@88
|
544 | SgiConstraint (c1, c2) =>
|
adamc@88
|
545 S.bind2 (con ctx c1,
|
adamc@88
|
546 fn c1' =>
|
adamc@88
|
547 S.map2 (con ctx c2,
|
adamc@88
|
548 fn c2' =>
|
adamc@88
|
549 (SgiConstraint (c1', c2'), loc)))
|
adamc@563
|
550 | SgiClassAbs (x, n, k) =>
|
adamc@563
|
551 S.map2 (kind k,
|
adamc@563
|
552 fn k' =>
|
adamc@563
|
553 (SgiClassAbs (x, n, k'), loc))
|
adamc@563
|
554 | SgiClass (x, n, k, c) =>
|
adamc@563
|
555 S.bind2 (kind k,
|
adamc@563
|
556 fn k' =>
|
adamc@563
|
557 S.map2 (con ctx c,
|
adamc@563
|
558 fn c' =>
|
adamc@563
|
559 (SgiClass (x, n, k', c'), loc)))
|
adamc@34
|
560
|
adamc@34
|
561 and sg ctx s acc =
|
adamc@34
|
562 S.bindP (sg' ctx s acc, sgn ctx)
|
adamc@34
|
563
|
adamc@34
|
564 and sg' ctx (sAll as (s, loc)) =
|
adamc@34
|
565 case s of
|
adamc@34
|
566 SgnConst sgis =>
|
adamc@34
|
567 S.map2 (ListUtil.mapfoldB (fn (ctx, si) =>
|
adamc@34
|
568 (case #1 si of
|
adamc@329
|
569 SgiConAbs (x, n, k) =>
|
adamc@329
|
570 bind (ctx, NamedC (x, n, k))
|
adamc@329
|
571 | SgiCon (x, n, k, _) =>
|
adamc@329
|
572 bind (ctx, NamedC (x, n, k))
|
adamc@191
|
573 | SgiDatatype (x, n, _, xncs) =>
|
adamc@329
|
574 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@329
|
575 | SgiDatatypeImp (x, n, _, _, _, _, _) =>
|
adamc@329
|
576 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@34
|
577 | SgiVal _ => ctx
|
adamc@34
|
578 | SgiStr (x, _, sgn) =>
|
adamc@59
|
579 bind (ctx, Str (x, sgn))
|
adamc@59
|
580 | SgiSgn (x, _, sgn) =>
|
adamc@88
|
581 bind (ctx, Sgn (x, sgn))
|
adamc@203
|
582 | SgiConstraint _ => ctx
|
adamc@563
|
583 | SgiClassAbs (x, n, k) =>
|
adamc@563
|
584 bind (ctx, NamedC (x, n, (KArrow (k, (KType, loc)), loc)))
|
adamc@563
|
585 | SgiClass (x, n, k, _) =>
|
adamc@563
|
586 bind (ctx, NamedC (x, n, (KArrow (k, (KType, loc)), loc))),
|
adamc@34
|
587 sgi ctx si)) ctx sgis,
|
adamc@34
|
588 fn sgis' =>
|
adamc@34
|
589 (SgnConst sgis', loc))
|
adamc@34
|
590
|
adamc@34
|
591 | SgnVar _ => S.return2 sAll
|
adamc@41
|
592 | SgnFun (m, n, s1, s2) =>
|
adamc@41
|
593 S.bind2 (sg ctx s1,
|
adamc@41
|
594 fn s1' =>
|
adamc@41
|
595 S.map2 (sg (bind (ctx, Str (m, s1'))) s2,
|
adamc@41
|
596 fn s2' =>
|
adamc@41
|
597 (SgnFun (m, n, s1', s2'), loc)))
|
adamc@59
|
598 | SgnProj _ => S.return2 sAll
|
adamc@42
|
599 | SgnWhere (sgn, x, c) =>
|
adamc@42
|
600 S.bind2 (sg ctx sgn,
|
adamc@42
|
601 fn sgn' =>
|
adamc@42
|
602 S.map2 (con ctx c,
|
adamc@42
|
603 fn c' =>
|
adamc@42
|
604 (SgnWhere (sgn', x, c'), loc)))
|
adamc@34
|
605 | SgnError => S.return2 sAll
|
adamc@34
|
606 in
|
adamc@34
|
607 sg
|
adamc@34
|
608 end
|
adamc@34
|
609
|
adamc@34
|
610 fun mapfold {kind, con, sgn_item, sgn} =
|
adamc@34
|
611 mapfoldB {kind = kind,
|
adamc@34
|
612 con = fn () => con,
|
adamc@34
|
613 sgn_item = fn () => sgn_item,
|
adamc@34
|
614 sgn = fn () => sgn,
|
adamc@34
|
615 bind = fn ((), _) => ()} ()
|
adamc@34
|
616
|
adamc@34
|
617 fun map {kind, con, sgn_item, sgn} s =
|
adamc@34
|
618 case mapfold {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@34
|
619 con = fn c => fn () => S.Continue (con c, ()),
|
adamc@34
|
620 sgn_item = fn si => fn () => S.Continue (sgn_item si, ()),
|
adamc@34
|
621 sgn = fn s => fn () => S.Continue (sgn s, ())} s () of
|
adamc@34
|
622 S.Return () => raise Fail "Elab_util.Sgn.map"
|
adamc@34
|
623 | S.Continue (s, ()) => s
|
adamc@34
|
624
|
adamc@2
|
625 end
|
adamc@34
|
626
|
adamc@76
|
627 structure Decl = struct
|
adamc@76
|
628
|
adamc@76
|
629 datatype binder =
|
adamc@76
|
630 RelC of string * Elab.kind
|
adamc@329
|
631 | NamedC of string * int * Elab.kind
|
adamc@76
|
632 | RelE of string * Elab.con
|
adamc@76
|
633 | NamedE of string * Elab.con
|
adamc@76
|
634 | Str of string * Elab.sgn
|
adamc@76
|
635 | Sgn of string * Elab.sgn
|
adamc@76
|
636
|
adamc@76
|
637 fun mapfoldB {kind = fk, con = fc, exp = fe, sgn_item = fsgi, sgn = fsg, str = fst, decl = fd, bind} =
|
adamc@76
|
638 let
|
adamc@76
|
639 val mfk = Kind.mapfold fk
|
adamc@76
|
640
|
adamc@76
|
641 fun bind' (ctx, b) =
|
adamc@76
|
642 let
|
adamc@76
|
643 val b' = case b of
|
adamc@76
|
644 Con.Rel x => RelC x
|
adamc@76
|
645 | Con.Named x => NamedC x
|
adamc@76
|
646 in
|
adamc@76
|
647 bind (ctx, b')
|
adamc@76
|
648 end
|
adamc@76
|
649 val mfc = Con.mapfoldB {kind = fk, con = fc, bind = bind'}
|
adamc@76
|
650
|
adamc@76
|
651 fun bind' (ctx, b) =
|
adamc@76
|
652 let
|
adamc@76
|
653 val b' = case b of
|
adamc@76
|
654 Exp.RelC x => RelC x
|
adamc@76
|
655 | Exp.NamedC x => NamedC x
|
adamc@76
|
656 | Exp.RelE x => RelE x
|
adamc@76
|
657 | Exp.NamedE x => NamedE x
|
adamc@76
|
658 in
|
adamc@76
|
659 bind (ctx, b')
|
adamc@76
|
660 end
|
adamc@76
|
661 val mfe = Exp.mapfoldB {kind = fk, con = fc, exp = fe, bind = bind'}
|
adamc@76
|
662
|
adamc@76
|
663 fun bind' (ctx, b) =
|
adamc@76
|
664 let
|
adamc@76
|
665 val b' = case b of
|
adamc@76
|
666 Sgn.RelC x => RelC x
|
adamc@76
|
667 | Sgn.NamedC x => NamedC x
|
adamc@76
|
668 | Sgn.Sgn x => Sgn x
|
adamc@76
|
669 | Sgn.Str x => Str x
|
adamc@76
|
670 in
|
adamc@76
|
671 bind (ctx, b')
|
adamc@76
|
672 end
|
adamc@76
|
673 val mfsg = Sgn.mapfoldB {kind = fk, con = fc, sgn_item = fsgi, sgn = fsg, bind = bind'}
|
adamc@76
|
674
|
adamc@76
|
675 fun mfst ctx str acc =
|
adamc@76
|
676 S.bindP (mfst' ctx str acc, fst ctx)
|
adamc@76
|
677
|
adamc@76
|
678 and mfst' ctx (strAll as (str, loc)) =
|
adamc@76
|
679 case str of
|
adamc@76
|
680 StrConst ds =>
|
adamc@76
|
681 S.map2 (ListUtil.mapfoldB (fn (ctx, d) =>
|
adamc@76
|
682 (case #1 d of
|
adamc@329
|
683 DCon (x, n, k, _) =>
|
adamc@329
|
684 bind (ctx, NamedC (x, n, k))
|
adamc@191
|
685 | DDatatype (x, n, xs, xncs) =>
|
adamc@156
|
686 let
|
adamc@329
|
687 val ctx = bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@156
|
688 in
|
adamc@156
|
689 foldl (fn ((x, _, co), ctx) =>
|
adamc@156
|
690 let
|
adamc@156
|
691 val t =
|
adamc@156
|
692 case co of
|
adamc@156
|
693 NONE => CNamed n
|
adamc@156
|
694 | SOME t => TFun (t, (CNamed n, loc))
|
adamc@191
|
695
|
adamc@191
|
696 val k = (KType, loc)
|
adamc@191
|
697 val t = (t, loc)
|
adamc@191
|
698 val t = foldr (fn (x, t) =>
|
adamc@191
|
699 (TCFun (Explicit,
|
adamc@191
|
700 x,
|
adamc@191
|
701 k,
|
adamc@191
|
702 t), loc))
|
adamc@191
|
703 t xs
|
adamc@156
|
704 in
|
adamc@191
|
705 bind (ctx, NamedE (x, t))
|
adamc@156
|
706 end)
|
adamc@156
|
707 ctx xncs
|
adamc@156
|
708 end
|
adamc@191
|
709 | DDatatypeImp (x, n, m, ms, x', _, _) =>
|
adamc@329
|
710 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@76
|
711 | DVal (x, _, c, _) =>
|
adamc@76
|
712 bind (ctx, NamedE (x, c))
|
adamc@123
|
713 | DValRec vis =>
|
adamc@123
|
714 foldl (fn ((x, _, c, _), ctx) => bind (ctx, NamedE (x, c))) ctx vis
|
adamc@76
|
715 | DSgn (x, _, sgn) =>
|
adamc@76
|
716 bind (ctx, Sgn (x, sgn))
|
adamc@76
|
717 | DStr (x, _, sgn, _) =>
|
adamc@76
|
718 bind (ctx, Str (x, sgn))
|
adamc@76
|
719 | DFfiStr (x, _, sgn) =>
|
adamc@88
|
720 bind (ctx, Str (x, sgn))
|
adamc@100
|
721 | DConstraint _ => ctx
|
adamc@203
|
722 | DExport _ => ctx
|
adamc@205
|
723 | DTable (tn, x, n, c) =>
|
adamc@338
|
724 bind (ctx, NamedE (x, (CApp ((CModProj (n, [], "sql_table"), loc),
|
adamc@213
|
725 c), loc)))
|
adamc@338
|
726 | DSequence (tn, x, n) =>
|
adamc@338
|
727 bind (ctx, NamedE (x, (CModProj (n, [], "sql_sequence"), loc)))
|
adamc@563
|
728 | DClass (x, n, k, _) =>
|
adamc@563
|
729 bind (ctx, NamedC (x, n, (KArrow (k, (KType, loc)), loc)))
|
adamc@459
|
730 | DDatabase _ => ctx
|
adamc@459
|
731 | DCookie (tn, x, n, c) =>
|
adamc@459
|
732 bind (ctx, NamedE (x, (CApp ((CModProj (n, [], "cookie"), loc),
|
adamc@459
|
733 c), loc))),
|
adamc@76
|
734 mfd ctx d)) ctx ds,
|
adamc@76
|
735 fn ds' => (StrConst ds', loc))
|
adamc@76
|
736 | StrVar _ => S.return2 strAll
|
adamc@76
|
737 | StrProj (str, x) =>
|
adamc@76
|
738 S.map2 (mfst ctx str,
|
adamc@76
|
739 fn str' =>
|
adamc@76
|
740 (StrProj (str', x), loc))
|
adamc@76
|
741 | StrFun (x, n, sgn1, sgn2, str) =>
|
adamc@76
|
742 S.bind2 (mfsg ctx sgn1,
|
adamc@76
|
743 fn sgn1' =>
|
adamc@76
|
744 S.bind2 (mfsg ctx sgn2,
|
adamc@76
|
745 fn sgn2' =>
|
adamc@76
|
746 S.map2 (mfst ctx str,
|
adamc@76
|
747 fn str' =>
|
adamc@76
|
748 (StrFun (x, n, sgn1', sgn2', str'), loc))))
|
adamc@76
|
749 | StrApp (str1, str2) =>
|
adamc@76
|
750 S.bind2 (mfst ctx str1,
|
adamc@76
|
751 fn str1' =>
|
adamc@76
|
752 S.map2 (mfst ctx str2,
|
adamc@76
|
753 fn str2' =>
|
adamc@76
|
754 (StrApp (str1', str2'), loc)))
|
adamc@76
|
755 | StrError => S.return2 strAll
|
adamc@76
|
756
|
adamc@76
|
757 and mfd ctx d acc =
|
adamc@76
|
758 S.bindP (mfd' ctx d acc, fd ctx)
|
adamc@76
|
759
|
adamc@76
|
760 and mfd' ctx (dAll as (d, loc)) =
|
adamc@76
|
761 case d of
|
adamc@76
|
762 DCon (x, n, k, c) =>
|
adamc@76
|
763 S.bind2 (mfk k,
|
adamc@76
|
764 fn k' =>
|
adamc@76
|
765 S.map2 (mfc ctx c,
|
adamc@76
|
766 fn c' =>
|
adamc@76
|
767 (DCon (x, n, k', c'), loc)))
|
adamc@191
|
768 | DDatatype (x, n, xs, xncs) =>
|
adamc@156
|
769 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@156
|
770 case c of
|
adamc@156
|
771 NONE => S.return2 (x, n, c)
|
adamc@156
|
772 | SOME c =>
|
adamc@156
|
773 S.map2 (mfc ctx c,
|
adamc@156
|
774 fn c' => (x, n, SOME c'))) xncs,
|
adamc@156
|
775 fn xncs' =>
|
adamc@191
|
776 (DDatatype (x, n, xs, xncs'), loc))
|
adamc@191
|
777 | DDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
|
adamc@162
|
778 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@162
|
779 case c of
|
adamc@162
|
780 NONE => S.return2 (x, n, c)
|
adamc@162
|
781 | SOME c =>
|
adamc@162
|
782 S.map2 (mfc ctx c,
|
adamc@162
|
783 fn c' => (x, n, SOME c'))) xncs,
|
adamc@162
|
784 fn xncs' =>
|
adamc@191
|
785 (DDatatypeImp (x, n, m1, ms, s, xs, xncs'), loc))
|
adamc@123
|
786 | DVal vi =>
|
adamc@123
|
787 S.map2 (mfvi ctx vi,
|
adamc@123
|
788 fn vi' =>
|
adamc@123
|
789 (DVal vi', loc))
|
adamc@123
|
790 | DValRec vis =>
|
adamc@123
|
791 S.map2 (ListUtil.mapfold (mfvi ctx) vis,
|
adamc@123
|
792 fn vis' =>
|
adamc@123
|
793 (DValRec vis', loc))
|
adamc@76
|
794 | DSgn (x, n, sgn) =>
|
adamc@76
|
795 S.map2 (mfsg ctx sgn,
|
adamc@76
|
796 fn sgn' =>
|
adamc@76
|
797 (DSgn (x, n, sgn'), loc))
|
adamc@76
|
798 | DStr (x, n, sgn, str) =>
|
adamc@76
|
799 S.bind2 (mfsg ctx sgn,
|
adamc@76
|
800 fn sgn' =>
|
adamc@76
|
801 S.map2 (mfst ctx str,
|
adamc@76
|
802 fn str' =>
|
adamc@76
|
803 (DStr (x, n, sgn', str'), loc)))
|
adamc@76
|
804 | DFfiStr (x, n, sgn) =>
|
adamc@76
|
805 S.map2 (mfsg ctx sgn,
|
adamc@76
|
806 fn sgn' =>
|
adamc@76
|
807 (DFfiStr (x, n, sgn'), loc))
|
adamc@88
|
808 | DConstraint (c1, c2) =>
|
adamc@88
|
809 S.bind2 (mfc ctx c1,
|
adamc@88
|
810 fn c1' =>
|
adamc@88
|
811 S.map2 (mfc ctx c2,
|
adamc@88
|
812 fn c2' =>
|
adamc@88
|
813 (DConstraint (c1', c2'), loc)))
|
adamc@109
|
814 | DExport (en, sgn, str) =>
|
adamc@109
|
815 S.bind2 (mfsg ctx sgn,
|
adamc@109
|
816 fn sgn' =>
|
adamc@109
|
817 S.map2 (mfst ctx str,
|
adamc@109
|
818 fn str' =>
|
adamc@109
|
819 (DExport (en, sgn', str'), loc)))
|
adamc@123
|
820
|
adamc@205
|
821 | DTable (tn, x, n, c) =>
|
adamc@203
|
822 S.map2 (mfc ctx c,
|
adamc@203
|
823 fn c' =>
|
adamc@205
|
824 (DTable (tn, x, n, c'), loc))
|
adamc@338
|
825 | DSequence _ => S.return2 dAll
|
adamc@203
|
826
|
adamc@563
|
827 | DClass (x, n, k, c) =>
|
adamc@563
|
828 S.bind2 (mfk k,
|
adamc@563
|
829 fn k' =>
|
adamc@563
|
830 S.map2 (mfc ctx c,
|
adamc@563
|
831 fn c' =>
|
adamc@563
|
832 (DClass (x, n, k', c'), loc)))
|
adamc@213
|
833
|
adamc@338
|
834 | DDatabase _ => S.return2 dAll
|
adamc@271
|
835
|
adamc@459
|
836 | DCookie (tn, x, n, c) =>
|
adamc@459
|
837 S.map2 (mfc ctx c,
|
adamc@459
|
838 fn c' =>
|
adamc@459
|
839 (DCookie (tn, x, n, c'), loc))
|
adamc@459
|
840
|
adamc@123
|
841 and mfvi ctx (x, n, c, e) =
|
adamc@123
|
842 S.bind2 (mfc ctx c,
|
adamc@123
|
843 fn c' =>
|
adamc@123
|
844 S.map2 (mfe ctx e,
|
adamc@123
|
845 fn e' =>
|
adamc@123
|
846 (x, n, c', e')))
|
adamc@76
|
847 in
|
adamc@76
|
848 mfd
|
adamc@76
|
849 end
|
adamc@76
|
850
|
adamc@76
|
851 fun mapfold {kind, con, exp, sgn_item, sgn, str, decl} =
|
adamc@76
|
852 mapfoldB {kind = kind,
|
adamc@76
|
853 con = fn () => con,
|
adamc@76
|
854 exp = fn () => exp,
|
adamc@76
|
855 sgn_item = fn () => sgn_item,
|
adamc@76
|
856 sgn = fn () => sgn,
|
adamc@76
|
857 str = fn () => str,
|
adamc@76
|
858 decl = fn () => decl,
|
adamc@76
|
859 bind = fn ((), _) => ()} ()
|
adamc@76
|
860
|
adamc@76
|
861 fun exists {kind, con, exp, sgn_item, sgn, str, decl} k =
|
adamc@76
|
862 case mapfold {kind = fn k => fn () =>
|
adamc@76
|
863 if kind k then
|
adamc@76
|
864 S.Return ()
|
adamc@76
|
865 else
|
adamc@76
|
866 S.Continue (k, ()),
|
adamc@76
|
867 con = fn c => fn () =>
|
adamc@76
|
868 if con c then
|
adamc@76
|
869 S.Return ()
|
adamc@76
|
870 else
|
adamc@76
|
871 S.Continue (c, ()),
|
adamc@76
|
872 exp = fn e => fn () =>
|
adamc@76
|
873 if exp e then
|
adamc@76
|
874 S.Return ()
|
adamc@76
|
875 else
|
adamc@76
|
876 S.Continue (e, ()),
|
adamc@76
|
877 sgn_item = fn sgi => fn () =>
|
adamc@76
|
878 if sgn_item sgi then
|
adamc@76
|
879 S.Return ()
|
adamc@76
|
880 else
|
adamc@76
|
881 S.Continue (sgi, ()),
|
adamc@76
|
882 sgn = fn x => fn () =>
|
adamc@76
|
883 if sgn x then
|
adamc@76
|
884 S.Return ()
|
adamc@76
|
885 else
|
adamc@76
|
886 S.Continue (x, ()),
|
adamc@76
|
887 str = fn x => fn () =>
|
adamc@76
|
888 if str x then
|
adamc@76
|
889 S.Return ()
|
adamc@76
|
890 else
|
adamc@76
|
891 S.Continue (x, ()),
|
adamc@76
|
892 decl = fn x => fn () =>
|
adamc@76
|
893 if decl x then
|
adamc@76
|
894 S.Return ()
|
adamc@76
|
895 else
|
adamc@76
|
896 S.Continue (x, ())} k () of
|
adamc@76
|
897 S.Return _ => true
|
adamc@76
|
898 | S.Continue _ => false
|
adamc@76
|
899
|
adamc@76
|
900 fun search {kind, con, exp, sgn_item, sgn, str, decl} k =
|
adamc@76
|
901 case mapfold {kind = fn x => fn () =>
|
adamc@76
|
902 case kind x of
|
adamc@76
|
903 NONE => S.Continue (x, ())
|
adamc@76
|
904 | SOME v => S.Return v,
|
adamc@76
|
905
|
adamc@76
|
906 con = fn x => fn () =>
|
adamc@76
|
907 case con x of
|
adamc@76
|
908 NONE => S.Continue (x, ())
|
adamc@76
|
909 | SOME v => S.Return v,
|
adamc@76
|
910
|
adamc@76
|
911 exp = fn x => fn () =>
|
adamc@76
|
912 case exp x of
|
adamc@76
|
913 NONE => S.Continue (x, ())
|
adamc@76
|
914 | SOME v => S.Return v,
|
adamc@76
|
915
|
adamc@76
|
916 sgn_item = fn x => fn () =>
|
adamc@76
|
917 case sgn_item x of
|
adamc@76
|
918 NONE => S.Continue (x, ())
|
adamc@76
|
919 | SOME v => S.Return v,
|
adamc@76
|
920
|
adamc@76
|
921 sgn = fn x => fn () =>
|
adamc@76
|
922 case sgn x of
|
adamc@76
|
923 NONE => S.Continue (x, ())
|
adamc@76
|
924 | SOME v => S.Return v,
|
adamc@76
|
925
|
adamc@76
|
926 str = fn x => fn () =>
|
adamc@76
|
927 case str x of
|
adamc@76
|
928 NONE => S.Continue (x, ())
|
adamc@76
|
929 | SOME v => S.Return v,
|
adamc@76
|
930
|
adamc@76
|
931 decl = fn x => fn () =>
|
adamc@76
|
932 case decl x of
|
adamc@76
|
933 NONE => S.Continue (x, ())
|
adamc@76
|
934 | SOME v => S.Return v
|
adamc@76
|
935
|
adamc@76
|
936 } k () of
|
adamc@76
|
937 S.Return x => SOME x
|
adamc@76
|
938 | S.Continue _ => NONE
|
adamc@76
|
939
|
adamc@448
|
940 fun foldMapB {kind, con, exp, sgn_item, sgn, str, decl, bind} ctx st d =
|
adamc@448
|
941 case mapfoldB {kind = fn x => fn st => S.Continue (kind (x, st)),
|
adamc@448
|
942 con = fn ctx => fn x => fn st => S.Continue (con (ctx, x, st)),
|
adamc@448
|
943 exp = fn ctx => fn x => fn st => S.Continue (exp (ctx, x, st)),
|
adamc@448
|
944 sgn_item = fn ctx => fn x => fn st => S.Continue (sgn_item (ctx, x, st)),
|
adamc@448
|
945 sgn = fn ctx => fn x => fn st => S.Continue (sgn (ctx, x, st)),
|
adamc@448
|
946 str = fn ctx => fn x => fn st => S.Continue (str (ctx, x, st)),
|
adamc@448
|
947 decl = fn ctx => fn x => fn st => S.Continue (decl (ctx, x, st)),
|
adamc@448
|
948 bind = bind} ctx d st of
|
adamc@448
|
949 S.Continue x => x
|
adamc@448
|
950 | S.Return _ => raise Fail "ElabUtil.Decl.foldMapB: Impossible"
|
adamc@448
|
951
|
adamc@448
|
952 end
|
adamc@448
|
953
|
adamc@448
|
954 structure File = struct
|
adamc@448
|
955
|
adamc@448
|
956 fun maxName ds = foldl (fn (d, count) => Int.max (maxNameDecl d, count)) 0 ds
|
adamc@448
|
957
|
adamc@448
|
958 and maxNameDecl (d, _) =
|
adamc@448
|
959 case d of
|
adamc@448
|
960 DCon (_, n, _, _) => n
|
adamc@448
|
961 | DDatatype (_, n, _, ns) =>
|
adamc@448
|
962 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
963 n ns
|
adamc@448
|
964 | DDatatypeImp (_, n1, n2, _, _, _, ns) =>
|
adamc@448
|
965 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
966 (Int.max (n1, n2)) ns
|
adamc@448
|
967 | DVal (_, n, _, _) => n
|
adamc@448
|
968 | DValRec vis => foldl (fn ((_, n, _, _), count) => Int.max (n, count)) 0 vis
|
adamc@448
|
969 | DStr (_, n, sgn, str) => Int.max (n, Int.max (maxNameSgn sgn, maxNameStr str))
|
adamc@448
|
970 | DSgn (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
971 | DFfiStr (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
972 | DConstraint _ => 0
|
adamc@563
|
973 | DClass (_, n, _, _) => n
|
adamc@448
|
974 | DExport _ => 0
|
adamc@459
|
975 | DTable (n1, _, n2, _) => Int.max (n1, n2)
|
adamc@459
|
976 | DSequence (n1, _, n2) => Int.max (n1, n2)
|
adamc@448
|
977 | DDatabase _ => 0
|
adamc@459
|
978 | DCookie (n1, _, n2, _) => Int.max (n1, n2)
|
adamc@448
|
979
|
adamc@448
|
980 and maxNameStr (str, _) =
|
adamc@448
|
981 case str of
|
adamc@448
|
982 StrConst ds => maxName ds
|
adamc@448
|
983 | StrVar n => n
|
adamc@448
|
984 | StrProj (str, _) => maxNameStr str
|
adamc@448
|
985 | StrFun (_, n, dom, ran, str) => foldl Int.max n [maxNameSgn dom, maxNameSgn ran, maxNameStr str]
|
adamc@448
|
986 | StrApp (str1, str2) => Int.max (maxNameStr str1, maxNameStr str2)
|
adamc@448
|
987 | StrError => 0
|
adamc@448
|
988
|
adamc@448
|
989 and maxNameSgn (sgn, _) =
|
adamc@448
|
990 case sgn of
|
adamc@448
|
991 SgnConst sgis => foldl (fn (sgi, count) => Int.max (maxNameSgi sgi, count)) 0 sgis
|
adamc@448
|
992 | SgnVar n => n
|
adamc@448
|
993 | SgnFun (_, n, dom, ran) => Int.max (n, Int.max (maxNameSgn dom, maxNameSgn ran))
|
adamc@448
|
994 | SgnWhere (sgn, _, _) => maxNameSgn sgn
|
adamc@448
|
995 | SgnProj (n, _, _) => n
|
adamc@448
|
996 | SgnError => 0
|
adamc@448
|
997
|
adamc@448
|
998 and maxNameSgi (sgi, _) =
|
adamc@448
|
999 case sgi of
|
adamc@448
|
1000 SgiConAbs (_, n, _) => n
|
adamc@448
|
1001 | SgiCon (_, n, _, _) => n
|
adamc@448
|
1002 | SgiDatatype (_, n, _, ns) =>
|
adamc@448
|
1003 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
1004 n ns
|
adamc@448
|
1005 | SgiDatatypeImp (_, n1, n2, _, _, _, ns) =>
|
adamc@448
|
1006 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
1007 (Int.max (n1, n2)) ns
|
adamc@448
|
1008 | SgiVal (_, n, _) => n
|
adamc@448
|
1009 | SgiStr (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
1010 | SgiSgn (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
1011 | SgiConstraint _ => 0
|
adamc@563
|
1012 | SgiClassAbs (_, n, _) => n
|
adamc@563
|
1013 | SgiClass (_, n, _, _) => n
|
adamc@448
|
1014
|
adamc@34
|
1015 end
|
adamc@76
|
1016
|
adamc@76
|
1017 end
|