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@67
|
171 | CFold (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@67
|
176 (CFold (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@71
|
341 | EFold k =>
|
adamc@71
|
342 S.map2 (mfk k,
|
adamc@71
|
343 fn k' =>
|
adamc@71
|
344 (EFold k', loc))
|
adamc@71
|
345
|
adamc@182
|
346 | ECase (e, pes, {disc, result}) =>
|
adamc@171
|
347 S.bind2 (mfe ctx e,
|
adamc@171
|
348 fn e' =>
|
adamc@171
|
349 S.bind2 (ListUtil.mapfold (fn (p, e) =>
|
adamc@448
|
350 let
|
adamc@448
|
351 fun pb ((p, _), ctx) =
|
adamc@448
|
352 case p of
|
adamc@448
|
353 PWild => ctx
|
adamc@448
|
354 | PVar (x, t) => bind (ctx, RelE (x, t))
|
adamc@448
|
355 | PPrim _ => ctx
|
adamc@448
|
356 | PCon (_, _, _, NONE) => ctx
|
adamc@448
|
357 | PCon (_, _, _, SOME p) => pb (p, ctx)
|
adamc@448
|
358 | PRecord xps => foldl (fn ((_, p, _), ctx) =>
|
adamc@448
|
359 pb (p, ctx)) ctx xps
|
adamc@448
|
360 in
|
adamc@448
|
361 S.map2 (mfe (pb (p, ctx)) e,
|
adamc@448
|
362 fn e' => (p, e'))
|
adamc@448
|
363 end) pes,
|
adamc@171
|
364 fn pes' =>
|
adamc@182
|
365 S.bind2 (mfc ctx disc,
|
adamc@182
|
366 fn disc' =>
|
adamc@182
|
367 S.map2 (mfc ctx result,
|
adamc@182
|
368 fn result' =>
|
adamc@182
|
369 (ECase (e', pes', {disc = disc', result = result'}), loc)))))
|
adamc@171
|
370
|
adamc@10
|
371 | EError => S.return2 eAll
|
adamc@228
|
372 | EUnif (ref (SOME e)) => mfe ctx e
|
adamc@228
|
373 | EUnif _ => S.return2 eAll
|
adamc@447
|
374
|
adamc@447
|
375 | ELet (des, e) =>
|
adamc@447
|
376 let
|
adamc@447
|
377 val (des, ctx) = foldl (fn (ed, (des, ctx)) =>
|
adamc@447
|
378 (S.bind2 (des,
|
adamc@447
|
379 fn des' =>
|
adamc@447
|
380 S.map2 (mfed ctx ed,
|
adamc@447
|
381 fn ed' => des' @ [ed'])),
|
adamc@447
|
382 case #1 ed of
|
adamc@447
|
383 EDVal (x, t, _) => bind (ctx, RelE (x, t))
|
adamc@447
|
384 | EDValRec vis =>
|
adamc@447
|
385 foldl (fn ((x, t, _), env) => bind (ctx, RelE (x, t))) ctx vis))
|
adamc@447
|
386 (S.return2 [], ctx) des
|
adamc@447
|
387 in
|
adamc@447
|
388 S.bind2 (des,
|
adamc@447
|
389 fn des' =>
|
adamc@447
|
390 S.map2 (mfe ctx e,
|
adamc@447
|
391 fn e' =>
|
adamc@447
|
392 (ELet (des', e'), loc)))
|
adamc@447
|
393 end
|
adamc@447
|
394
|
adamc@447
|
395 and mfed ctx (dAll as (d, loc)) =
|
adamc@447
|
396 case d of
|
adamc@447
|
397 EDVal vi =>
|
adamc@447
|
398 S.map2 (mfvi ctx vi,
|
adamc@447
|
399 fn vi' =>
|
adamc@447
|
400 (EDVal vi', loc))
|
adamc@447
|
401 | EDValRec vis =>
|
adamc@447
|
402 let
|
adamc@447
|
403 val ctx = foldl (fn ((x, t, _), env) => bind (ctx, RelE (x, t))) ctx vis
|
adamc@447
|
404 in
|
adamc@447
|
405 S.map2 (ListUtil.mapfold (mfvi ctx) vis,
|
adamc@447
|
406 fn vis' =>
|
adamc@447
|
407 (EDValRec vis', loc))
|
adamc@447
|
408 end
|
adamc@447
|
409
|
adamc@447
|
410 and mfvi ctx (x, c, e) =
|
adamc@447
|
411 S.bind2 (mfc ctx c,
|
adamc@447
|
412 fn c' =>
|
adamc@447
|
413 S.map2 (mfe ctx e,
|
adamc@447
|
414 fn e' =>
|
adamc@447
|
415 (x, c', e')))
|
adamc@10
|
416 in
|
adamc@10
|
417 mfe
|
adamc@10
|
418 end
|
adamc@10
|
419
|
adamc@11
|
420 fun mapfold {kind = fk, con = fc, exp = fe} =
|
adamc@11
|
421 mapfoldB {kind = fk,
|
adamc@11
|
422 con = fn () => fc,
|
adamc@11
|
423 exp = fn () => fe,
|
adamc@11
|
424 bind = fn ((), _) => ()} ()
|
adamc@11
|
425
|
adamc@10
|
426 fun exists {kind, con, exp} k =
|
adamc@10
|
427 case mapfold {kind = fn k => fn () =>
|
adamc@10
|
428 if kind k then
|
adamc@10
|
429 S.Return ()
|
adamc@10
|
430 else
|
adamc@10
|
431 S.Continue (k, ()),
|
adamc@10
|
432 con = fn c => fn () =>
|
adamc@10
|
433 if con c then
|
adamc@10
|
434 S.Return ()
|
adamc@10
|
435 else
|
adamc@10
|
436 S.Continue (c, ()),
|
adamc@10
|
437 exp = fn e => fn () =>
|
adamc@10
|
438 if exp e then
|
adamc@10
|
439 S.Return ()
|
adamc@10
|
440 else
|
adamc@10
|
441 S.Continue (e, ())} k () of
|
adamc@10
|
442 S.Return _ => true
|
adamc@10
|
443 | S.Continue _ => false
|
adamc@10
|
444
|
adamc@211
|
445 fun mapB {kind, con, exp, bind} ctx e =
|
adamc@211
|
446 case mapfoldB {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@211
|
447 con = fn ctx => fn c => fn () => S.Continue (con ctx c, ()),
|
adamc@211
|
448 exp = fn ctx => fn e => fn () => S.Continue (exp ctx e, ()),
|
adamc@211
|
449 bind = bind} ctx e () of
|
adamc@211
|
450 S.Continue (e, ()) => e
|
adamc@211
|
451 | S.Return _ => raise Fail "ElabUtil.Exp.mapB: Impossible"
|
adamc@211
|
452
|
adamc@448
|
453 fun foldB {kind, con, exp, bind} ctx st e =
|
adamc@448
|
454 case mapfoldB {kind = fn k => fn st => S.Continue (k, kind (k, st)),
|
adamc@448
|
455 con = fn ctx => fn c => fn st => S.Continue (c, con (ctx, c, st)),
|
adamc@448
|
456 exp = fn ctx => fn e => fn st => S.Continue (e, exp (ctx, e, st)),
|
adamc@448
|
457 bind = bind} ctx e st of
|
adamc@448
|
458 S.Continue (_, st) => st
|
adamc@448
|
459 | S.Return _ => raise Fail "ElabUtil.Exp.foldB: Impossible"
|
adamc@448
|
460
|
adamc@10
|
461 end
|
adamc@10
|
462
|
adamc@34
|
463 structure Sgn = struct
|
adamc@34
|
464
|
adamc@34
|
465 datatype binder =
|
adamc@34
|
466 RelC of string * Elab.kind
|
adamc@329
|
467 | NamedC of string * int * Elab.kind
|
adamc@34
|
468 | Str of string * Elab.sgn
|
adamc@59
|
469 | Sgn of string * Elab.sgn
|
adamc@34
|
470
|
adamc@34
|
471 fun mapfoldB {kind, con, sgn_item, sgn, bind} =
|
adamc@34
|
472 let
|
adamc@34
|
473 fun bind' (ctx, b) =
|
adamc@34
|
474 let
|
adamc@34
|
475 val b' = case b of
|
adamc@34
|
476 Con.Rel x => RelC x
|
adamc@34
|
477 | Con.Named x => NamedC x
|
adamc@34
|
478 in
|
adamc@34
|
479 bind (ctx, b')
|
adamc@34
|
480 end
|
adamc@34
|
481 val con = Con.mapfoldB {kind = kind, con = con, bind = bind'}
|
adamc@34
|
482
|
adamc@34
|
483 val kind = Kind.mapfold kind
|
adamc@34
|
484
|
adamc@34
|
485 fun sgi ctx si acc =
|
adamc@34
|
486 S.bindP (sgi' ctx si acc, sgn_item ctx)
|
adamc@34
|
487
|
adamc@156
|
488 and sgi' ctx (siAll as (si, loc)) =
|
adamc@34
|
489 case si of
|
adamc@34
|
490 SgiConAbs (x, n, k) =>
|
adamc@34
|
491 S.map2 (kind k,
|
adamc@34
|
492 fn k' =>
|
adamc@34
|
493 (SgiConAbs (x, n, k'), loc))
|
adamc@34
|
494 | SgiCon (x, n, k, c) =>
|
adamc@34
|
495 S.bind2 (kind k,
|
adamc@34
|
496 fn k' =>
|
adamc@34
|
497 S.map2 (con ctx c,
|
adamc@34
|
498 fn c' =>
|
adamc@34
|
499 (SgiCon (x, n, k', c'), loc)))
|
adamc@191
|
500 | SgiDatatype (x, n, xs, xncs) =>
|
adamc@156
|
501 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@156
|
502 case c of
|
adamc@156
|
503 NONE => S.return2 (x, n, c)
|
adamc@156
|
504 | SOME c =>
|
adamc@156
|
505 S.map2 (con ctx c,
|
adamc@156
|
506 fn c' => (x, n, SOME c'))) xncs,
|
adamc@156
|
507 fn xncs' =>
|
adamc@191
|
508 (SgiDatatype (x, n, xs, xncs'), loc))
|
adamc@191
|
509 | SgiDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
|
adamc@162
|
510 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@162
|
511 case c of
|
adamc@162
|
512 NONE => S.return2 (x, n, c)
|
adamc@162
|
513 | SOME c =>
|
adamc@162
|
514 S.map2 (con ctx c,
|
adamc@162
|
515 fn c' => (x, n, SOME c'))) xncs,
|
adamc@162
|
516 fn xncs' =>
|
adamc@191
|
517 (SgiDatatypeImp (x, n, m1, ms, s, xs, xncs'), loc))
|
adamc@34
|
518 | SgiVal (x, n, c) =>
|
adamc@34
|
519 S.map2 (con ctx c,
|
adamc@34
|
520 fn c' =>
|
adamc@34
|
521 (SgiVal (x, n, c'), loc))
|
adamc@34
|
522 | SgiStr (x, n, s) =>
|
adamc@34
|
523 S.map2 (sg ctx s,
|
adamc@34
|
524 fn s' =>
|
adamc@34
|
525 (SgiStr (x, n, s'), loc))
|
adamc@59
|
526 | SgiSgn (x, n, s) =>
|
adamc@59
|
527 S.map2 (sg ctx s,
|
adamc@59
|
528 fn s' =>
|
adamc@59
|
529 (SgiSgn (x, n, s'), loc))
|
adamc@88
|
530 | SgiConstraint (c1, c2) =>
|
adamc@88
|
531 S.bind2 (con ctx c1,
|
adamc@88
|
532 fn c1' =>
|
adamc@88
|
533 S.map2 (con ctx c2,
|
adamc@88
|
534 fn c2' =>
|
adamc@88
|
535 (SgiConstraint (c1', c2'), loc)))
|
adamc@205
|
536 | SgiTable (tn, x, n, c) =>
|
adamc@203
|
537 S.map2 (con ctx c,
|
adamc@203
|
538 fn c' =>
|
adamc@205
|
539 (SgiTable (tn, x, n, c'), loc))
|
adamc@338
|
540 | SgiSequence _ => S.return2 siAll
|
adamc@211
|
541 | SgiClassAbs _ => S.return2 siAll
|
adamc@211
|
542 | SgiClass (x, n, c) =>
|
adamc@211
|
543 S.map2 (con ctx c,
|
adamc@211
|
544 fn c' =>
|
adamc@211
|
545 (SgiClass (x, n, c'), loc))
|
adamc@34
|
546
|
adamc@34
|
547 and sg ctx s acc =
|
adamc@34
|
548 S.bindP (sg' ctx s acc, sgn ctx)
|
adamc@34
|
549
|
adamc@34
|
550 and sg' ctx (sAll as (s, loc)) =
|
adamc@34
|
551 case s of
|
adamc@34
|
552 SgnConst sgis =>
|
adamc@34
|
553 S.map2 (ListUtil.mapfoldB (fn (ctx, si) =>
|
adamc@34
|
554 (case #1 si of
|
adamc@329
|
555 SgiConAbs (x, n, k) =>
|
adamc@329
|
556 bind (ctx, NamedC (x, n, k))
|
adamc@329
|
557 | SgiCon (x, n, k, _) =>
|
adamc@329
|
558 bind (ctx, NamedC (x, n, k))
|
adamc@191
|
559 | SgiDatatype (x, n, _, xncs) =>
|
adamc@329
|
560 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@329
|
561 | SgiDatatypeImp (x, n, _, _, _, _, _) =>
|
adamc@329
|
562 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@34
|
563 | SgiVal _ => ctx
|
adamc@34
|
564 | SgiStr (x, _, sgn) =>
|
adamc@59
|
565 bind (ctx, Str (x, sgn))
|
adamc@59
|
566 | SgiSgn (x, _, sgn) =>
|
adamc@88
|
567 bind (ctx, Sgn (x, sgn))
|
adamc@203
|
568 | SgiConstraint _ => ctx
|
adamc@211
|
569 | SgiTable _ => ctx
|
adamc@338
|
570 | SgiSequence _ => ctx
|
adamc@329
|
571 | SgiClassAbs (x, n) =>
|
adamc@329
|
572 bind (ctx, NamedC (x, n, (KArrow ((KType, loc), (KType, loc)), loc)))
|
adamc@329
|
573 | SgiClass (x, n, _) =>
|
adamc@329
|
574 bind (ctx, NamedC (x, n, (KArrow ((KType, loc), (KType, loc)), loc))),
|
adamc@34
|
575 sgi ctx si)) ctx sgis,
|
adamc@34
|
576 fn sgis' =>
|
adamc@34
|
577 (SgnConst sgis', loc))
|
adamc@34
|
578
|
adamc@34
|
579 | SgnVar _ => S.return2 sAll
|
adamc@41
|
580 | SgnFun (m, n, s1, s2) =>
|
adamc@41
|
581 S.bind2 (sg ctx s1,
|
adamc@41
|
582 fn s1' =>
|
adamc@41
|
583 S.map2 (sg (bind (ctx, Str (m, s1'))) s2,
|
adamc@41
|
584 fn s2' =>
|
adamc@41
|
585 (SgnFun (m, n, s1', s2'), loc)))
|
adamc@59
|
586 | SgnProj _ => S.return2 sAll
|
adamc@42
|
587 | SgnWhere (sgn, x, c) =>
|
adamc@42
|
588 S.bind2 (sg ctx sgn,
|
adamc@42
|
589 fn sgn' =>
|
adamc@42
|
590 S.map2 (con ctx c,
|
adamc@42
|
591 fn c' =>
|
adamc@42
|
592 (SgnWhere (sgn', x, c'), loc)))
|
adamc@34
|
593 | SgnError => S.return2 sAll
|
adamc@34
|
594 in
|
adamc@34
|
595 sg
|
adamc@34
|
596 end
|
adamc@34
|
597
|
adamc@34
|
598 fun mapfold {kind, con, sgn_item, sgn} =
|
adamc@34
|
599 mapfoldB {kind = kind,
|
adamc@34
|
600 con = fn () => con,
|
adamc@34
|
601 sgn_item = fn () => sgn_item,
|
adamc@34
|
602 sgn = fn () => sgn,
|
adamc@34
|
603 bind = fn ((), _) => ()} ()
|
adamc@34
|
604
|
adamc@34
|
605 fun map {kind, con, sgn_item, sgn} s =
|
adamc@34
|
606 case mapfold {kind = fn k => fn () => S.Continue (kind k, ()),
|
adamc@34
|
607 con = fn c => fn () => S.Continue (con c, ()),
|
adamc@34
|
608 sgn_item = fn si => fn () => S.Continue (sgn_item si, ()),
|
adamc@34
|
609 sgn = fn s => fn () => S.Continue (sgn s, ())} s () of
|
adamc@34
|
610 S.Return () => raise Fail "Elab_util.Sgn.map"
|
adamc@34
|
611 | S.Continue (s, ()) => s
|
adamc@34
|
612
|
adamc@2
|
613 end
|
adamc@34
|
614
|
adamc@76
|
615 structure Decl = struct
|
adamc@76
|
616
|
adamc@76
|
617 datatype binder =
|
adamc@76
|
618 RelC of string * Elab.kind
|
adamc@329
|
619 | NamedC of string * int * Elab.kind
|
adamc@76
|
620 | RelE of string * Elab.con
|
adamc@76
|
621 | NamedE of string * Elab.con
|
adamc@76
|
622 | Str of string * Elab.sgn
|
adamc@76
|
623 | Sgn of string * Elab.sgn
|
adamc@76
|
624
|
adamc@76
|
625 fun mapfoldB {kind = fk, con = fc, exp = fe, sgn_item = fsgi, sgn = fsg, str = fst, decl = fd, bind} =
|
adamc@76
|
626 let
|
adamc@76
|
627 val mfk = Kind.mapfold fk
|
adamc@76
|
628
|
adamc@76
|
629 fun bind' (ctx, b) =
|
adamc@76
|
630 let
|
adamc@76
|
631 val b' = case b of
|
adamc@76
|
632 Con.Rel x => RelC x
|
adamc@76
|
633 | Con.Named x => NamedC x
|
adamc@76
|
634 in
|
adamc@76
|
635 bind (ctx, b')
|
adamc@76
|
636 end
|
adamc@76
|
637 val mfc = Con.mapfoldB {kind = fk, con = fc, bind = bind'}
|
adamc@76
|
638
|
adamc@76
|
639 fun bind' (ctx, b) =
|
adamc@76
|
640 let
|
adamc@76
|
641 val b' = case b of
|
adamc@76
|
642 Exp.RelC x => RelC x
|
adamc@76
|
643 | Exp.NamedC x => NamedC x
|
adamc@76
|
644 | Exp.RelE x => RelE x
|
adamc@76
|
645 | Exp.NamedE x => NamedE x
|
adamc@76
|
646 in
|
adamc@76
|
647 bind (ctx, b')
|
adamc@76
|
648 end
|
adamc@76
|
649 val mfe = Exp.mapfoldB {kind = fk, con = fc, exp = fe, 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 Sgn.RelC x => RelC x
|
adamc@76
|
655 | Sgn.NamedC x => NamedC x
|
adamc@76
|
656 | Sgn.Sgn x => Sgn x
|
adamc@76
|
657 | Sgn.Str x => Str x
|
adamc@76
|
658 in
|
adamc@76
|
659 bind (ctx, b')
|
adamc@76
|
660 end
|
adamc@76
|
661 val mfsg = Sgn.mapfoldB {kind = fk, con = fc, sgn_item = fsgi, sgn = fsg, bind = bind'}
|
adamc@76
|
662
|
adamc@76
|
663 fun mfst ctx str acc =
|
adamc@76
|
664 S.bindP (mfst' ctx str acc, fst ctx)
|
adamc@76
|
665
|
adamc@76
|
666 and mfst' ctx (strAll as (str, loc)) =
|
adamc@76
|
667 case str of
|
adamc@76
|
668 StrConst ds =>
|
adamc@76
|
669 S.map2 (ListUtil.mapfoldB (fn (ctx, d) =>
|
adamc@76
|
670 (case #1 d of
|
adamc@329
|
671 DCon (x, n, k, _) =>
|
adamc@329
|
672 bind (ctx, NamedC (x, n, k))
|
adamc@191
|
673 | DDatatype (x, n, xs, xncs) =>
|
adamc@156
|
674 let
|
adamc@329
|
675 val ctx = bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@156
|
676 in
|
adamc@156
|
677 foldl (fn ((x, _, co), ctx) =>
|
adamc@156
|
678 let
|
adamc@156
|
679 val t =
|
adamc@156
|
680 case co of
|
adamc@156
|
681 NONE => CNamed n
|
adamc@156
|
682 | SOME t => TFun (t, (CNamed n, loc))
|
adamc@191
|
683
|
adamc@191
|
684 val k = (KType, loc)
|
adamc@191
|
685 val t = (t, loc)
|
adamc@191
|
686 val t = foldr (fn (x, t) =>
|
adamc@191
|
687 (TCFun (Explicit,
|
adamc@191
|
688 x,
|
adamc@191
|
689 k,
|
adamc@191
|
690 t), loc))
|
adamc@191
|
691 t xs
|
adamc@156
|
692 in
|
adamc@191
|
693 bind (ctx, NamedE (x, t))
|
adamc@156
|
694 end)
|
adamc@156
|
695 ctx xncs
|
adamc@156
|
696 end
|
adamc@191
|
697 | DDatatypeImp (x, n, m, ms, x', _, _) =>
|
adamc@329
|
698 bind (ctx, NamedC (x, n, (KType, loc)))
|
adamc@76
|
699 | DVal (x, _, c, _) =>
|
adamc@76
|
700 bind (ctx, NamedE (x, c))
|
adamc@123
|
701 | DValRec vis =>
|
adamc@123
|
702 foldl (fn ((x, _, c, _), ctx) => bind (ctx, NamedE (x, c))) ctx vis
|
adamc@76
|
703 | DSgn (x, _, sgn) =>
|
adamc@76
|
704 bind (ctx, Sgn (x, sgn))
|
adamc@76
|
705 | DStr (x, _, sgn, _) =>
|
adamc@76
|
706 bind (ctx, Str (x, sgn))
|
adamc@76
|
707 | DFfiStr (x, _, sgn) =>
|
adamc@88
|
708 bind (ctx, Str (x, sgn))
|
adamc@100
|
709 | DConstraint _ => ctx
|
adamc@203
|
710 | DExport _ => ctx
|
adamc@205
|
711 | DTable (tn, x, n, c) =>
|
adamc@338
|
712 bind (ctx, NamedE (x, (CApp ((CModProj (n, [], "sql_table"), loc),
|
adamc@213
|
713 c), loc)))
|
adamc@338
|
714 | DSequence (tn, x, n) =>
|
adamc@338
|
715 bind (ctx, NamedE (x, (CModProj (n, [], "sql_sequence"), loc)))
|
adamc@329
|
716 | DClass (x, n, _) =>
|
adamc@329
|
717 bind (ctx, NamedC (x, n, (KArrow ((KType, loc), (KType, loc)), loc)))
|
adamc@271
|
718 | DDatabase _ => ctx,
|
adamc@76
|
719 mfd ctx d)) ctx ds,
|
adamc@76
|
720 fn ds' => (StrConst ds', loc))
|
adamc@76
|
721 | StrVar _ => S.return2 strAll
|
adamc@76
|
722 | StrProj (str, x) =>
|
adamc@76
|
723 S.map2 (mfst ctx str,
|
adamc@76
|
724 fn str' =>
|
adamc@76
|
725 (StrProj (str', x), loc))
|
adamc@76
|
726 | StrFun (x, n, sgn1, sgn2, str) =>
|
adamc@76
|
727 S.bind2 (mfsg ctx sgn1,
|
adamc@76
|
728 fn sgn1' =>
|
adamc@76
|
729 S.bind2 (mfsg ctx sgn2,
|
adamc@76
|
730 fn sgn2' =>
|
adamc@76
|
731 S.map2 (mfst ctx str,
|
adamc@76
|
732 fn str' =>
|
adamc@76
|
733 (StrFun (x, n, sgn1', sgn2', str'), loc))))
|
adamc@76
|
734 | StrApp (str1, str2) =>
|
adamc@76
|
735 S.bind2 (mfst ctx str1,
|
adamc@76
|
736 fn str1' =>
|
adamc@76
|
737 S.map2 (mfst ctx str2,
|
adamc@76
|
738 fn str2' =>
|
adamc@76
|
739 (StrApp (str1', str2'), loc)))
|
adamc@76
|
740 | StrError => S.return2 strAll
|
adamc@76
|
741
|
adamc@76
|
742 and mfd ctx d acc =
|
adamc@76
|
743 S.bindP (mfd' ctx d acc, fd ctx)
|
adamc@76
|
744
|
adamc@76
|
745 and mfd' ctx (dAll as (d, loc)) =
|
adamc@76
|
746 case d of
|
adamc@76
|
747 DCon (x, n, k, c) =>
|
adamc@76
|
748 S.bind2 (mfk k,
|
adamc@76
|
749 fn k' =>
|
adamc@76
|
750 S.map2 (mfc ctx c,
|
adamc@76
|
751 fn c' =>
|
adamc@76
|
752 (DCon (x, n, k', c'), loc)))
|
adamc@191
|
753 | DDatatype (x, n, xs, xncs) =>
|
adamc@156
|
754 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@156
|
755 case c of
|
adamc@156
|
756 NONE => S.return2 (x, n, c)
|
adamc@156
|
757 | SOME c =>
|
adamc@156
|
758 S.map2 (mfc ctx c,
|
adamc@156
|
759 fn c' => (x, n, SOME c'))) xncs,
|
adamc@156
|
760 fn xncs' =>
|
adamc@191
|
761 (DDatatype (x, n, xs, xncs'), loc))
|
adamc@191
|
762 | DDatatypeImp (x, n, m1, ms, s, xs, xncs) =>
|
adamc@162
|
763 S.map2 (ListUtil.mapfold (fn (x, n, c) =>
|
adamc@162
|
764 case c of
|
adamc@162
|
765 NONE => S.return2 (x, n, c)
|
adamc@162
|
766 | SOME c =>
|
adamc@162
|
767 S.map2 (mfc ctx c,
|
adamc@162
|
768 fn c' => (x, n, SOME c'))) xncs,
|
adamc@162
|
769 fn xncs' =>
|
adamc@191
|
770 (DDatatypeImp (x, n, m1, ms, s, xs, xncs'), loc))
|
adamc@123
|
771 | DVal vi =>
|
adamc@123
|
772 S.map2 (mfvi ctx vi,
|
adamc@123
|
773 fn vi' =>
|
adamc@123
|
774 (DVal vi', loc))
|
adamc@123
|
775 | DValRec vis =>
|
adamc@123
|
776 S.map2 (ListUtil.mapfold (mfvi ctx) vis,
|
adamc@123
|
777 fn vis' =>
|
adamc@123
|
778 (DValRec vis', loc))
|
adamc@76
|
779 | DSgn (x, n, sgn) =>
|
adamc@76
|
780 S.map2 (mfsg ctx sgn,
|
adamc@76
|
781 fn sgn' =>
|
adamc@76
|
782 (DSgn (x, n, sgn'), loc))
|
adamc@76
|
783 | DStr (x, n, sgn, str) =>
|
adamc@76
|
784 S.bind2 (mfsg ctx sgn,
|
adamc@76
|
785 fn sgn' =>
|
adamc@76
|
786 S.map2 (mfst ctx str,
|
adamc@76
|
787 fn str' =>
|
adamc@76
|
788 (DStr (x, n, sgn', str'), loc)))
|
adamc@76
|
789 | DFfiStr (x, n, sgn) =>
|
adamc@76
|
790 S.map2 (mfsg ctx sgn,
|
adamc@76
|
791 fn sgn' =>
|
adamc@76
|
792 (DFfiStr (x, n, sgn'), loc))
|
adamc@88
|
793 | DConstraint (c1, c2) =>
|
adamc@88
|
794 S.bind2 (mfc ctx c1,
|
adamc@88
|
795 fn c1' =>
|
adamc@88
|
796 S.map2 (mfc ctx c2,
|
adamc@88
|
797 fn c2' =>
|
adamc@88
|
798 (DConstraint (c1', c2'), loc)))
|
adamc@109
|
799 | DExport (en, sgn, str) =>
|
adamc@109
|
800 S.bind2 (mfsg ctx sgn,
|
adamc@109
|
801 fn sgn' =>
|
adamc@109
|
802 S.map2 (mfst ctx str,
|
adamc@109
|
803 fn str' =>
|
adamc@109
|
804 (DExport (en, sgn', str'), loc)))
|
adamc@123
|
805
|
adamc@205
|
806 | DTable (tn, x, n, c) =>
|
adamc@203
|
807 S.map2 (mfc ctx c,
|
adamc@203
|
808 fn c' =>
|
adamc@205
|
809 (DTable (tn, x, n, c'), loc))
|
adamc@338
|
810 | DSequence _ => S.return2 dAll
|
adamc@203
|
811
|
adamc@338
|
812 | DClass (x, n, c) =>
|
adamc@213
|
813 S.map2 (mfc ctx c,
|
adamc@338
|
814 fn c' =>
|
adamc@338
|
815 (DClass (x, n, c'), loc))
|
adamc@213
|
816
|
adamc@338
|
817 | DDatabase _ => S.return2 dAll
|
adamc@271
|
818
|
adamc@123
|
819 and mfvi ctx (x, n, c, e) =
|
adamc@123
|
820 S.bind2 (mfc ctx c,
|
adamc@123
|
821 fn c' =>
|
adamc@123
|
822 S.map2 (mfe ctx e,
|
adamc@123
|
823 fn e' =>
|
adamc@123
|
824 (x, n, c', e')))
|
adamc@76
|
825 in
|
adamc@76
|
826 mfd
|
adamc@76
|
827 end
|
adamc@76
|
828
|
adamc@76
|
829 fun mapfold {kind, con, exp, sgn_item, sgn, str, decl} =
|
adamc@76
|
830 mapfoldB {kind = kind,
|
adamc@76
|
831 con = fn () => con,
|
adamc@76
|
832 exp = fn () => exp,
|
adamc@76
|
833 sgn_item = fn () => sgn_item,
|
adamc@76
|
834 sgn = fn () => sgn,
|
adamc@76
|
835 str = fn () => str,
|
adamc@76
|
836 decl = fn () => decl,
|
adamc@76
|
837 bind = fn ((), _) => ()} ()
|
adamc@76
|
838
|
adamc@76
|
839 fun exists {kind, con, exp, sgn_item, sgn, str, decl} k =
|
adamc@76
|
840 case mapfold {kind = fn k => fn () =>
|
adamc@76
|
841 if kind k then
|
adamc@76
|
842 S.Return ()
|
adamc@76
|
843 else
|
adamc@76
|
844 S.Continue (k, ()),
|
adamc@76
|
845 con = fn c => fn () =>
|
adamc@76
|
846 if con c then
|
adamc@76
|
847 S.Return ()
|
adamc@76
|
848 else
|
adamc@76
|
849 S.Continue (c, ()),
|
adamc@76
|
850 exp = fn e => fn () =>
|
adamc@76
|
851 if exp e then
|
adamc@76
|
852 S.Return ()
|
adamc@76
|
853 else
|
adamc@76
|
854 S.Continue (e, ()),
|
adamc@76
|
855 sgn_item = fn sgi => fn () =>
|
adamc@76
|
856 if sgn_item sgi then
|
adamc@76
|
857 S.Return ()
|
adamc@76
|
858 else
|
adamc@76
|
859 S.Continue (sgi, ()),
|
adamc@76
|
860 sgn = fn x => fn () =>
|
adamc@76
|
861 if sgn x then
|
adamc@76
|
862 S.Return ()
|
adamc@76
|
863 else
|
adamc@76
|
864 S.Continue (x, ()),
|
adamc@76
|
865 str = fn x => fn () =>
|
adamc@76
|
866 if str x then
|
adamc@76
|
867 S.Return ()
|
adamc@76
|
868 else
|
adamc@76
|
869 S.Continue (x, ()),
|
adamc@76
|
870 decl = fn x => fn () =>
|
adamc@76
|
871 if decl x then
|
adamc@76
|
872 S.Return ()
|
adamc@76
|
873 else
|
adamc@76
|
874 S.Continue (x, ())} k () of
|
adamc@76
|
875 S.Return _ => true
|
adamc@76
|
876 | S.Continue _ => false
|
adamc@76
|
877
|
adamc@76
|
878 fun search {kind, con, exp, sgn_item, sgn, str, decl} k =
|
adamc@76
|
879 case mapfold {kind = fn x => fn () =>
|
adamc@76
|
880 case kind x of
|
adamc@76
|
881 NONE => S.Continue (x, ())
|
adamc@76
|
882 | SOME v => S.Return v,
|
adamc@76
|
883
|
adamc@76
|
884 con = fn x => fn () =>
|
adamc@76
|
885 case con x of
|
adamc@76
|
886 NONE => S.Continue (x, ())
|
adamc@76
|
887 | SOME v => S.Return v,
|
adamc@76
|
888
|
adamc@76
|
889 exp = fn x => fn () =>
|
adamc@76
|
890 case exp x of
|
adamc@76
|
891 NONE => S.Continue (x, ())
|
adamc@76
|
892 | SOME v => S.Return v,
|
adamc@76
|
893
|
adamc@76
|
894 sgn_item = fn x => fn () =>
|
adamc@76
|
895 case sgn_item x of
|
adamc@76
|
896 NONE => S.Continue (x, ())
|
adamc@76
|
897 | SOME v => S.Return v,
|
adamc@76
|
898
|
adamc@76
|
899 sgn = fn x => fn () =>
|
adamc@76
|
900 case sgn x of
|
adamc@76
|
901 NONE => S.Continue (x, ())
|
adamc@76
|
902 | SOME v => S.Return v,
|
adamc@76
|
903
|
adamc@76
|
904 str = fn x => fn () =>
|
adamc@76
|
905 case str x of
|
adamc@76
|
906 NONE => S.Continue (x, ())
|
adamc@76
|
907 | SOME v => S.Return v,
|
adamc@76
|
908
|
adamc@76
|
909 decl = fn x => fn () =>
|
adamc@76
|
910 case decl x of
|
adamc@76
|
911 NONE => S.Continue (x, ())
|
adamc@76
|
912 | SOME v => S.Return v
|
adamc@76
|
913
|
adamc@76
|
914 } k () of
|
adamc@76
|
915 S.Return x => SOME x
|
adamc@76
|
916 | S.Continue _ => NONE
|
adamc@76
|
917
|
adamc@448
|
918 fun foldMapB {kind, con, exp, sgn_item, sgn, str, decl, bind} ctx st d =
|
adamc@448
|
919 case mapfoldB {kind = fn x => fn st => S.Continue (kind (x, st)),
|
adamc@448
|
920 con = fn ctx => fn x => fn st => S.Continue (con (ctx, x, st)),
|
adamc@448
|
921 exp = fn ctx => fn x => fn st => S.Continue (exp (ctx, x, st)),
|
adamc@448
|
922 sgn_item = fn ctx => fn x => fn st => S.Continue (sgn_item (ctx, x, st)),
|
adamc@448
|
923 sgn = fn ctx => fn x => fn st => S.Continue (sgn (ctx, x, st)),
|
adamc@448
|
924 str = fn ctx => fn x => fn st => S.Continue (str (ctx, x, st)),
|
adamc@448
|
925 decl = fn ctx => fn x => fn st => S.Continue (decl (ctx, x, st)),
|
adamc@448
|
926 bind = bind} ctx d st of
|
adamc@448
|
927 S.Continue x => x
|
adamc@448
|
928 | S.Return _ => raise Fail "ElabUtil.Decl.foldMapB: Impossible"
|
adamc@448
|
929
|
adamc@448
|
930 end
|
adamc@448
|
931
|
adamc@448
|
932 structure File = struct
|
adamc@448
|
933
|
adamc@448
|
934 fun maxName ds = foldl (fn (d, count) => Int.max (maxNameDecl d, count)) 0 ds
|
adamc@448
|
935
|
adamc@448
|
936 and maxNameDecl (d, _) =
|
adamc@448
|
937 case d of
|
adamc@448
|
938 DCon (_, n, _, _) => n
|
adamc@448
|
939 | DDatatype (_, n, _, ns) =>
|
adamc@448
|
940 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
941 n ns
|
adamc@448
|
942 | DDatatypeImp (_, n1, n2, _, _, _, ns) =>
|
adamc@448
|
943 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
944 (Int.max (n1, n2)) ns
|
adamc@448
|
945 | DVal (_, n, _, _) => n
|
adamc@448
|
946 | DValRec vis => foldl (fn ((_, n, _, _), count) => Int.max (n, count)) 0 vis
|
adamc@448
|
947 | DStr (_, n, sgn, str) => Int.max (n, Int.max (maxNameSgn sgn, maxNameStr str))
|
adamc@448
|
948 | DSgn (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
949 | DFfiStr (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
950 | DConstraint _ => 0
|
adamc@448
|
951 | DClass (_, n, _) => n
|
adamc@448
|
952 | DExport _ => 0
|
adamc@448
|
953 | DTable (n, _, _, _) => n
|
adamc@448
|
954 | DSequence (n, _, _) => n
|
adamc@448
|
955 | DDatabase _ => 0
|
adamc@448
|
956
|
adamc@448
|
957 and maxNameStr (str, _) =
|
adamc@448
|
958 case str of
|
adamc@448
|
959 StrConst ds => maxName ds
|
adamc@448
|
960 | StrVar n => n
|
adamc@448
|
961 | StrProj (str, _) => maxNameStr str
|
adamc@448
|
962 | StrFun (_, n, dom, ran, str) => foldl Int.max n [maxNameSgn dom, maxNameSgn ran, maxNameStr str]
|
adamc@448
|
963 | StrApp (str1, str2) => Int.max (maxNameStr str1, maxNameStr str2)
|
adamc@448
|
964 | StrError => 0
|
adamc@448
|
965
|
adamc@448
|
966 and maxNameSgn (sgn, _) =
|
adamc@448
|
967 case sgn of
|
adamc@448
|
968 SgnConst sgis => foldl (fn (sgi, count) => Int.max (maxNameSgi sgi, count)) 0 sgis
|
adamc@448
|
969 | SgnVar n => n
|
adamc@448
|
970 | SgnFun (_, n, dom, ran) => Int.max (n, Int.max (maxNameSgn dom, maxNameSgn ran))
|
adamc@448
|
971 | SgnWhere (sgn, _, _) => maxNameSgn sgn
|
adamc@448
|
972 | SgnProj (n, _, _) => n
|
adamc@448
|
973 | SgnError => 0
|
adamc@448
|
974
|
adamc@448
|
975 and maxNameSgi (sgi, _) =
|
adamc@448
|
976 case sgi of
|
adamc@448
|
977 SgiConAbs (_, n, _) => n
|
adamc@448
|
978 | SgiCon (_, n, _, _) => n
|
adamc@448
|
979 | SgiDatatype (_, n, _, ns) =>
|
adamc@448
|
980 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
981 n ns
|
adamc@448
|
982 | SgiDatatypeImp (_, n1, n2, _, _, _, ns) =>
|
adamc@448
|
983 foldl (fn ((_, n', _), m) => Int.max (n', m))
|
adamc@448
|
984 (Int.max (n1, n2)) ns
|
adamc@448
|
985 | SgiVal (_, n, _) => n
|
adamc@448
|
986 | SgiStr (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
987 | SgiSgn (_, n, sgn) => Int.max (n, maxNameSgn sgn)
|
adamc@448
|
988 | SgiConstraint _ => 0
|
adamc@448
|
989 | SgiTable (n, _, _, _) => n
|
adamc@448
|
990 | SgiSequence (n, _, _) => n
|
adamc@448
|
991 | SgiClassAbs (_, n) => n
|
adamc@448
|
992 | SgiClass (_, n, _) => n
|
adamc@448
|
993
|
adamc@34
|
994 end
|
adamc@76
|
995
|
adamc@76
|
996 end
|