Home My Page Projects Code Snippets Project Openings SML/NJ
Summary Activity Forums Tracker Lists Tasks Docs Surveys News SCM Files

SCM Repository

[smlnj] Annotation of /sml/trunk/src/compiler/FLINT/kernel/ltykernel.sml
ViewVC logotype

Annotation of /sml/trunk/src/compiler/FLINT/kernel/ltykernel.sml

Parent Directory Parent Directory | Revision Log Revision Log


Revision 65 - (view) (download)

1 : monnier 16 (* COPYRIGHT (c) 1997 YALE FLINT PROJECT *)
2 :     (* ltykernel.sml *)
3 :    
4 :     structure LtyKernel :> LTYKERNEL =
5 :     struct
6 :    
7 : monnier 45 fun bug s = ErrorMsg.impossible ("LtyKernel:" ^ s)
8 :    
9 : monnier 16 (***************************************************************************
10 :     * UTILITY FUNCTIONS FOR HASHCONSING BASICS *
11 :     ***************************************************************************)
12 :    
13 :     (** hashconsing implementation basics *)
14 :     local open SortedList
15 :     val MVAL = 10000
16 :     val BVAL = MVAL * 2 (* all index i start from 0 *)
17 :     in
18 :    
19 :     type enc_tvar = int
20 : league 53 fun tvEncode (d, i) = d * MVAL + i
21 :     fun tvDecode x = ((x div MVAL), (x mod MVAL))
22 :    
23 : monnier 16 fun exitLevel xs =
24 :     let fun h ([], x) = rev x
25 :     | h (a::r, x) = if a < BVAL then h(r, x) else h(r, (a-MVAL)::x)
26 :     in h(xs, [])
27 :     end
28 :    
29 :     datatype aux_info = AX_REG of bool * enc_tvar list
30 :     | AX_NO
31 :    
32 :     val mergeTvs = merge
33 :     val fmergeTvs = foldmerge
34 :    
35 :     type 'a hash_cell = (int * 'a * aux_info) ref
36 :    
37 :     end (* local of hashconsing implementation basics *)
38 :    
39 :     (***************************************************************************
40 :     * DATATYPE DEFINITIONS *
41 :     ***************************************************************************)
42 :    
43 :     (** definition of kinds for all the lambda tycs *)
44 :     datatype tkindI
45 :     = TK_MONO (* ground mono tycon *)
46 :     | TK_BOX (* boxed/tagged tycon *)
47 :     | TK_SEQ of tkind list (* sequence of tycons *)
48 : monnier 45 | TK_FUN of tkind list * tkind (* tycon function *)
49 : monnier 16
50 :     withtype tkind = tkindI hash_cell (* hash-consing-impl of tkind *)
51 :    
52 :     (* definitoins of named tyc variables *)
53 :     type tvar = LambdaVar.lvar (* temporary definitions *)
54 :     val mkTvar = LambdaVar.mkLvar
55 :    
56 : monnier 45 (* an special extensible token key *)
57 :     type token = int
58 :    
59 :     datatype fflag (* calling conventions *)
60 :     = FF_VAR of bool * bool (* is it fixed ? *)
61 :     | FF_FIXED (* used after rep. analysis *)
62 :    
63 :     datatype rflag = RF_TMP (* tuple kind: a template *)
64 :    
65 :     (** definitions of lambda type constructors *)
66 : monnier 16 datatype tycI
67 :     = TC_VAR of DebIndex.index * int (* tyc variables *)
68 :     | TC_NVAR of tvar * DebIndex.depth * int (* named tyc variables *)
69 :     | TC_PRIM of PrimTyc.primtyc (* primitive tyc *)
70 :    
71 :     | TC_FN of tkind list * tyc (* tyc abstraction *)
72 :     | TC_APP of tyc * tyc list (* tyc application *)
73 :     | TC_SEQ of tyc list (* tyc sequence *)
74 :     | TC_PROJ of tyc * int (* tyc projection *)
75 :    
76 :     | TC_SUM of tyc list (* sum tyc *)
77 :     | TC_FIX of (int * tyc * tyc list) * int (* recursive tyc *)
78 :    
79 : monnier 45 | TC_TUPLE of rflag * tyc list (* std record tyc *)
80 :     | TC_ARROW of fflag * tyc list * tyc list (* std function tyc *)
81 : monnier 16 | TC_PARROW of tyc * tyc (* special fun tyc, not used *)
82 :    
83 :     | TC_BOX of tyc (* boxed tyc *)
84 :     | TC_ABS of tyc (* abstract tyc, not used *)
85 : monnier 45 | TC_TOKEN of token * tyc (* extensible token tyc *)
86 : monnier 16 | TC_CONT of tyc list (* intern continuation tycon *)
87 :     | TC_IND of tyc * tycI (* indirect tyc thunk *)
88 :     | TC_ENV of tyc * int * int * tycEnv (* tyc closure *)
89 :    
90 : monnier 45 withtype tyc = tycI hash_cell (* hash-consed tyc cell *)
91 :     and tycEnv = tyc (*
92 :     * This really is (tyc list option * int) list,
93 :     * it is encoded using SEQ[(PROJ(SEQ tcs),i)]
94 :     * and SEQ[(PROJ(VOID, i))]. (ZHONG)
95 :     *)
96 :    
97 :     (** definitions of lambda types *)
98 :     datatype ltyI
99 : monnier 16 = LT_TYC of tyc (* monomorphic type *)
100 :     | LT_STR of lty list (* structure record type *)
101 :     | LT_FCT of lty list * lty list (* functor arrow type *)
102 :     | LT_POLY of tkind list * lty list (* polymorphic type *)
103 :    
104 :     | LT_PST of (int * lty) list (* partial-structure type *)
105 :     | LT_CONT of lty list (* internal cont type *)
106 :     | LT_IND of lty * ltyI (* a lty thunk and its sig *)
107 :     | LT_ENV of lty * int * int * tycEnv (* lty closure *)
108 :    
109 : monnier 45 withtype lty = ltyI hash_cell (* hash-consed lty cell *)
110 : monnier 16
111 : monnier 45 (***************************************************************************
112 :     * TOKEN TYC UTILITY FUNCTIONS *
113 :     ***************************************************************************)
114 : monnier 24
115 : monnier 45 type token_info
116 :     = {name : string,
117 :     abbrev : string,
118 :     reduce_one: token * tyc -> tyc,
119 :     is_whnm : tyc -> bool,
120 :     is_known : token * tyc -> bool}
121 : monnier 24
122 : monnier 45 local val token_key = ref 0
123 :     val token_table_size = 10
124 :     val default_token_info =
125 :     {name="TC_GARBAGE",
126 :     abbrev="GB",
127 :     reduce_one=(fn _ => bug "token not implemented"),
128 :     is_whnm=(fn _ => bug "token not implemented"),
129 :     is_known=(fn _ => bug "token not implemented")}
130 :     val token_array : token_info Array.array =
131 :     Array.array(token_table_size,default_token_info)
132 :     val token_validity_table = Array.array(token_table_size,false)
133 :     fun get_next_token () =
134 :     let val n = !token_key
135 :     in if n > token_table_size then bug "running out of tokens"
136 :     else (token_key := n+1; n)
137 :     end
138 :     fun store_token_info (x, k) = Array.update(token_array, k, x)
139 :     fun get_is_whnm k = #is_whnm (Array.sub(token_array, k))
140 :     fun get_reduce_one (z as (k, t)) =
141 :     (#reduce_one (Array.sub(token_array, k))) z
142 :     fun get_name k = #name (Array.sub(token_array, k))
143 :     fun get_abbrev k = #abbrev (Array.sub(token_array, k))
144 :     fun get_is_known (z as (k, t)) =
145 :     (#is_known (Array.sub(token_array, k))) z
146 :     fun is_valid k = Array.sub(token_validity_table, k)
147 :     fun set_valid k = Array.update(token_validity_table, k, true)
148 :     in
149 : monnier 24
150 : monnier 45 val register_token: token_info -> token =
151 :     (fn x => let val k = get_next_token ()
152 :     in store_token_info(x, k); set_valid k; k
153 :     end)
154 : monnier 24
155 : monnier 45 val token_name : token -> string = get_name
156 :     val token_abbrev : token -> string = get_abbrev
157 :     val token_whnm : token -> tyc -> bool = get_is_whnm
158 :     val token_reduce : token * tyc -> tyc = get_reduce_one
159 :     val token_isKnown : token * tyc -> bool = get_is_known
160 :     val token_isvalid : token -> bool = is_valid
161 :     val token_eq : token * token -> bool = fn (x,y) => (x=y)
162 :     val token_int : token -> int = fn x => x
163 :     val token_key : int -> token = fn x => x
164 :    
165 :     end (* end of all token-related hacks *)
166 :    
167 : monnier 16 (***************************************************************************
168 :     * HASHCONSING IMPLEMENTATIONS *
169 :     ***************************************************************************)
170 :    
171 :     (** Hash-consing implementations of tyc, tkind, lty *)
172 :    
173 :     local structure Weak = SMLofNJ.Weak
174 :     structure PT = PrimTyc
175 :     structure DI = DebIndex
176 :    
177 :     fun bug msg = ErrorMsg.impossible("LtyKernel: "^msg)
178 :    
179 :     val itow = Word.fromInt
180 :     val wtoi = Word.toIntX
181 :     val andb = Word.andb
182 :    
183 :     val N = 2048 (* 1024 *)
184 :     val NN = itow (N*N)
185 :     val P = 0w509 (* was 0w1019, a prime < 1024 so that N*N*P < maxint *)
186 :    
187 :     val tk_table : tkind Weak.weak list Array.array = Array.array(N,nil)
188 :     val tc_table : tyc Weak.weak list Array.array = Array.array(N,nil)
189 :     val lt_table : lty Weak.weak list Array.array = Array.array(N,nil)
190 :    
191 :     fun vector2list v = Vector.foldr (op ::) [] v
192 :    
193 :     fun revcat(a::rest,b) = revcat(rest,a::b)
194 :     | revcat(nil,b) = b
195 :    
196 :     fun combine [x] = itow x
197 :     | combine (a::rest) =
198 :     andb(itow a +(combine rest)*P, NN - 0w1)
199 :     | combine _ = bug "unexpected case in combine"
200 :    
201 :     (*
202 :     * Because of the "cmp" function below, it's necessary to keep
203 :     * each bucket-list in a consistent order, and not reverse
204 :     * or move-to-front or whatever.
205 :     *)
206 :     fun look(table, h, t, eq, mk) =
207 :     let val i = wtoi(andb(itow h, itow(N-1)))
208 :    
209 :     fun g(l, z as (w::rest)) =
210 :     (case Weak.strong w
211 :     of SOME (r as ref(h',t',_)) =>
212 :     if (h=h') andalso (eq {new=t, old=t'})
213 :     then (Array.update(table, i, revcat(l,z)); r)
214 :     else g(w::l, rest)
215 :     | NONE => g(l, rest))
216 :     | g(l, []) =
217 :     let val r = mk(h, t)
218 :     in Array.update(table, i, (Weak.weak r) :: rev l); r
219 :     end
220 :    
221 :     in g([], Array.sub(table, i))
222 :     end
223 :    
224 :     fun cmp(table, a as ref(ai,_,_), b as ref (bi,_,_)) =
225 :     if ai < bi then LESS
226 :     else if ai > bi then GREATER
227 :     else if a = b then EQUAL
228 :     else let val index = wtoi (andb(itow ai,itow(N-1)))
229 :     fun g [] = bug "unexpected case in cmp"
230 :     | g (w::rest) =
231 :     (case Weak.strong w
232 :     of SOME r =>
233 :     if a=r then LESS
234 :     else if b=r then GREATER
235 :     else g rest
236 :     | NONE => g rest)
237 :     in g(Array.sub(table,index))
238 :     end
239 :    
240 :    
241 :     fun getnum (ref(i,_,_)) = i
242 :     fun tagnums nil = nil
243 :     | tagnums ((i,t)::rest) = i::getnum t::tagnums rest
244 :    
245 :     fun tk_hash tk =
246 :     let fun g (TK_MONO) = 0w1
247 :     | g (TK_BOX) = 0w2
248 :     | g (TK_SEQ ks) = combine (3::map getnum ks)
249 : monnier 45 | g (TK_FUN(ks, k)) = combine (4::getnum k::(map getnum ks))
250 : monnier 16 in g tk
251 :     end
252 :    
253 :     fun tc_hash tc =
254 :     let fun g (TC_VAR(d, i)) = combine [1, (DI.di_key d)*10, i]
255 :     | g (TC_NVAR(v, d, i)) = combine[15, v, (DI.dp_key d)*13, i]
256 :     | g (TC_PRIM pt) = combine [2, PT.pt_toint pt]
257 :     | g (TC_FN(ks, t)) = combine (3::(getnum t)::(map getnum ks))
258 :     | g (TC_APP(t, ts)) = combine (4::(getnum t)::(map getnum ts))
259 :     | g (TC_SEQ ts) = combine (5::(map getnum ts))
260 :     | g (TC_PROJ(t, i)) = combine [6, (getnum t), i]
261 :     | g (TC_SUM ts) = combine (7::(map getnum ts))
262 :     | g (TC_FIX((n, t, ts), i)) =
263 :     combine (8::n::i::(getnum t)::(map getnum ts))
264 :     | g (TC_ABS t) = combine [9, getnum t]
265 :     | g (TC_BOX t) = combine [10, getnum t]
266 : monnier 45 | g (TC_TUPLE (_, ts)) = combine (11::(map getnum ts))
267 : monnier 16 | g (TC_ARROW(rw, ts1, ts2)) =
268 : monnier 45 let fun h (FF_FIXED) = 10
269 :     | h (FF_VAR(true,b2)) = if b2 then 20 else 30
270 :     | h (FF_VAR(false,b2)) = if b2 then 40 else 50
271 : monnier 16 in combine (12::(h rw)::(map getnum (ts1@ts2)))
272 :     end
273 :     | g (TC_PARROW (t1,t2)) = combine [13, getnum t1, getnum t2]
274 : monnier 45 | g (TC_TOKEN (i, tc)) = combine [14, i, getnum tc]
275 :     | g (TC_CONT ts) = combine (15::(map getnum ts))
276 : monnier 16 | g (TC_ENV(t,i,j,env)) =
277 : monnier 45 combine[16, getnum t, i, j, getnum env]
278 : monnier 16 | g (TC_IND _) = bug "unexpected TC_IND in tc_hash"
279 :    
280 :     in g tc
281 :     end
282 :    
283 :     fun lt_hash lt =
284 :     let fun g (LT_TYC t) = combine [1, getnum t]
285 :     | g (LT_STR ts) = combine (2::(map getnum ts))
286 :     | g (LT_PST ts) = combine (3::(tagnums ts))
287 :     | g (LT_FCT(ts1, ts2)) =
288 :     combine (4::(map getnum (ts1@ts2)))
289 :     | g (LT_POLY(ks, ts)) =
290 :     combine (5::((map getnum ts)@(map getnum ks)))
291 :     | g (LT_CONT ts) = combine (6::(map getnum ts))
292 :     | g (LT_ENV(t,i,j,env)) =
293 :     combine [7, getnum t, i, j, getnum env]
294 :     | g (LT_IND _) = bug "unexpected LT_IND in tc_hash"
295 :     in g lt
296 :     end
297 :    
298 :     fun tkI_eq {new: tkindI, old} = (new = old)
299 :    
300 :     (* the 1st is one being mapped; the 2nd is in the hash table *)
301 :     fun tcI_eq {new : tycI, old=TC_IND(_,s)} = tcI_eq {new=new, old=s}
302 :     | tcI_eq {new, old} = (new=old)
303 :    
304 :     fun ltI_eq {new : ltyI, old=LT_IND(_,s)} = ltI_eq {new=new, old=s}
305 :     | ltI_eq {new, old} = (new=old)
306 :    
307 :     val baseAux = AX_REG (true, [])
308 :    
309 :     fun getAux (ref(i : int, _, x)) = x
310 :    
311 :     fun mergeAux(AX_NO, _) = AX_NO
312 :     | mergeAux(_, AX_NO) = AX_NO
313 :     | mergeAux(AX_REG(b1,vs1), AX_REG(b2,vs2)) =
314 :     AX_REG(b2 andalso b1, mergeTvs(vs1, vs2))
315 :    
316 :     fun fsmerge [] = baseAux
317 :     | fsmerge [x] = getAux x
318 :     | fsmerge xs =
319 :     let fun loop([], z) = z
320 :     | loop(_, AX_NO) = AX_NO
321 :     | loop(a::r, z) = loop(r, mergeAux(getAux a, z))
322 :     in loop(xs, baseAux)
323 :     end
324 :    
325 :     fun exitAux(AX_REG(b, vs)) = AX_REG(b, exitLevel vs)
326 :     | exitAux x = x
327 :    
328 :     fun tc_aux tc =
329 : league 53 let fun g (TC_VAR(d, i)) = AX_REG(true, [tvEncode(d, i)])
330 : monnier 45 | g (TC_NVAR(v, d, i)) = baseAux (*** THIS IS WRONG ! ***)
331 : monnier 16 | g (TC_PRIM pt) = baseAux
332 :     | g (TC_APP(ref(_, TC_FN _, AX_NO), _)) = AX_NO
333 :     | g (TC_PROJ(ref(_, TC_SEQ _, AX_NO), _)) = AX_NO
334 :     | g (TC_APP(ref(_, TC_FN _, AX_REG(_,vs)), ts)) =
335 :     mergeAux(AX_REG(false, vs), fsmerge ts) (* ? *)
336 :     | g (TC_PROJ(ref(_, TC_SEQ _, AX_REG(_,vs)), _)) =
337 :     AX_REG(false, vs) (* ? *)
338 :     | g (TC_FN(ks, t)) = exitAux(getAux t)
339 :     | g (TC_APP(t, ts)) = fsmerge (t::ts)
340 :     | g (TC_SEQ ts) = fsmerge ts
341 :     | g (TC_PROJ(t, _)) = getAux t
342 :     | g (TC_SUM ts) = fsmerge ts
343 :     | g (TC_FIX((_,t,ts), _)) =
344 :     let val ax = getAux t
345 :     in case ax
346 :     of AX_REG(_,[]) => mergeAux(ax, fsmerge ts)
347 :     | _ => bug "unexpected TC_FIX freevars in tc_aux"
348 :     end
349 :     | g (TC_ABS t) = getAux t
350 :     | g (TC_BOX t) = getAux t
351 : monnier 45 | g (TC_TUPLE (_, ts)) = fsmerge ts
352 : monnier 16 | g (TC_ARROW(_, ts1, ts2)) = fsmerge (ts1@ts2)
353 :     | g (TC_PARROW(t1, t2)) = fsmerge [t1, t2]
354 : monnier 45 | g (TC_TOKEN (k, (ref(_, t, AX_NO)))) = AX_NO
355 :     | g (TC_TOKEN (k, (x as ref(_, t, AX_REG(b,vs))))) =
356 :     AX_REG((token_whnm k x) andalso b, vs)
357 : monnier 16 | g (TC_CONT ts) = fsmerge ts
358 :     | g (TC_IND _) = bug "unexpected TC_IND in tc_aux"
359 :     | g (TC_ENV _) = AX_NO
360 :     in g tc
361 :     end
362 :    
363 :     fun lt_aux lt =
364 :     let fun g (LT_TYC t) = getAux t
365 :     | g (LT_STR ts) = fsmerge ts
366 :     | g (LT_PST ts) = fsmerge (map #2 ts)
367 :     | g (LT_FCT(ts1, ts2)) = fsmerge (ts1@ts2)
368 :     | g (LT_POLY(ks, ts)) = exitAux(fsmerge ts)
369 :     | g (LT_CONT ts) = fsmerge ts
370 :     | g (LT_IND _) = bug "unexpected LT_IND in lt_aux"
371 :     | g (LT_ENV _) = AX_NO
372 :     in g lt
373 :     end
374 :    
375 :     fun tk_mk (i : int, k: tkindI) = ref (i, k, AX_NO)
376 :     fun tc_mk (i : int, tc : tycI) = ref (i, tc, tc_aux tc)
377 :     fun lt_mk (i : int, lt : ltyI) = ref (i, lt, lt_aux lt)
378 :    
379 :     in
380 :    
381 :     (** a temporary hack on getting the list of free tyvars *)
382 :     fun tc_vs (r as ref(_ : int, _ : tycI, AX_NO)) = NONE
383 :     | tc_vs (r as ref(_ : int, _ : tycI, AX_REG (_,x))) = SOME x
384 :    
385 :     fun lt_vs (r as ref(_ : int, _ : ltyI, AX_NO)) = NONE
386 :     | lt_vs (r as ref(_ : int, _ : ltyI, AX_REG (_,x))) = SOME x
387 :    
388 :     (** converting from the hash-consing reps to the standard reps *)
389 :     fun tk_outX (r as ref(_ : int, t : tkindI, _ : aux_info)) = t
390 :     fun tc_outX (r as ref(_ : int, t : tycI, _ : aux_info)) = t
391 :     fun lt_outX (r as ref(_ : int, t : ltyI, _ : aux_info)) = t
392 :    
393 :    
394 :     (** converting from the standard reps to the hash-consing reps *)
395 :     fun tk_injX t = look(tk_table, wtoi(tk_hash t), t, tkI_eq, tk_mk)
396 :     fun tc_injX t = look(tc_table, wtoi(tc_hash t), t, tcI_eq, tc_mk)
397 :     fun lt_injX t = look(lt_table, wtoi(lt_hash t), t, ltI_eq, lt_mk)
398 :    
399 :     (** key-comparison on tkind, tyc, lty *)
400 :     fun tk_cmp (k1, k2) = cmp(tk_table, k1, k2)
401 :     fun tc_cmp (t1, t2) = cmp(tc_table, t1, t2)
402 :     fun lt_cmp (t1, t2) = cmp(lt_table, t1, t2)
403 :    
404 :    
405 :     (** get the hash key of each lty, only used by reps/coerce.sml; a hack *)
406 :     fun lt_key (ref (h : int, _ : ltyI, _ : aux_info)) = h
407 :    
408 :     (***************************************************************************
409 : league 65 * UTILITY FUNCTIONS ON TKIND ENVIRONMENT *
410 :     ***************************************************************************)
411 :    
412 :     (** tkind environment: maps each tyvar, i.e., its debindex, to its kind *)
413 :     type tkindEnv = tkind list list
414 :    
415 :     (** utility functions for manipulating the tkindEnv *)
416 :     exception tkUnbound
417 :     val initTkEnv : tkindEnv = []
418 :    
419 :     fun tkLookup (kenv, i, j) =
420 :     let val ks = List.nth(kenv, i-1) handle _ => raise tkUnbound
421 :     in List.nth(ks, j) handle _ => raise tkUnbound
422 :     end
423 :    
424 :     fun tkInsert (kenv, ks) = ks::kenv
425 :    
426 :     (* strip any unused type variables out of a kenv, given a list of
427 :     * [encoded] free type variables. the result is a "parallel list" of
428 :     * the kinds of those free type variables in the environment.
429 :     * This is meant to use the same representation of a kind environment
430 :     * as in ltybasic.
431 :     * --CALeague
432 :     *)
433 :     fun tkLookupFreeVars (kenv, tyc) =
434 :     let
435 :     fun g (kenv, d, []) = []
436 :     | g (kenv, d, ftv::ftvs) =
437 :     let val (d', i') = tvDecode ftv
438 :     val kenv' = List.drop (kenv, d'-d)
439 :     handle _ => raise tkUnbound
440 :     val k = List.nth (hd kenv', i')
441 :     handle _ => raise tkUnbound
442 :     val rest = g (kenv', d', ftvs)
443 :     in
444 :     k :: rest
445 :     end
446 :    
447 :     fun h ftvs = g (kenv, 1, ftvs)
448 :     in
449 :     Option.map h (tc_vs tyc)
450 :     end
451 :    
452 :     (***************************************************************************
453 : monnier 16 * UTILITY FUNCTIONS ON TYC ENVIRONMENT *
454 :     ***************************************************************************)
455 :    
456 :     (** utility functions for manipulating the tycEnv *)
457 :     local val tc_void = tc_injX(TC_PRIM(PT.ptc_void))
458 : monnier 45 fun tc_cons (t, b) = tc_injX(TC_ARROW(FF_FIXED, [t],[b]))
459 : monnier 16 fun tc_interp x =
460 :     (case tc_outX x
461 :     of TC_PROJ(y, i) =>
462 :     (case tc_outX y of TC_SEQ ts => (SOME ts, i)
463 :     | TC_PRIM _ => (NONE, i)
464 :     | _ => bug "unexpected tycEnv1 in tc_interp")
465 :     | _ => bug "unexpected tycEnv2 in tc_interp")
466 :    
467 :     fun tc_encode(NONE, i) = tc_injX(TC_PROJ(tc_void,i))
468 :     | tc_encode(SOME ts, i) =
469 :     tc_injX(TC_PROJ(tc_injX(TC_SEQ(ts)), i))
470 :    
471 :     in
472 :    
473 :     exception tcUnbound
474 :     val initTycEnv : tycEnv = tc_void
475 :    
476 :     fun tcLookup(i, tenv : tycEnv) =
477 :     if i > 1 then
478 :     (case tc_outX tenv of TC_ARROW(_,_,[x]) => tcLookup(i-1, x)
479 :     | _ => bug "unexpected tycEnv in tcLookup")
480 :     else if i = 1 then
481 :     (case tc_outX tenv of TC_ARROW(_,[x],_) => tc_interp x
482 :     | _ => raise tcUnbound)
483 :     else bug "unexpected argument in tcLookup"
484 :    
485 :     fun tcInsert(tenv : tycEnv, et) = tc_cons(tc_encode et, tenv)
486 :    
487 :     fun tcSplit(tenv : tycEnv) =
488 :     (case tc_outX tenv of TC_ARROW(_,[x],[y]) => SOME (tc_interp x, y)
489 :     | _ => NONE)
490 :    
491 :    
492 :     end (* utililty function for tycEnv *)
493 :    
494 :    
495 :     (** checking if a tyc or an lty is in the normal form *)
496 :     fun tcp_norm ((t as ref (i, _, AX_REG(b,_))) : tyc) = b
497 :     | tcp_norm _ = false
498 :    
499 :     fun ltp_norm ((t as ref (i, _, AX_REG(b,_))) : lty) = b
500 :     | ltp_norm _ = false
501 :    
502 :    
503 :     (** utility functions for tc_env and lt_env *)
504 :     local fun tcc_env_int(x, 0, 0, te) = x
505 :     | tcc_env_int(x, i, j, te) = tc_injX(TC_ENV(x, i, j, te))
506 :    
507 :     fun ltc_env_int(x, 0, 0, te) = x
508 :     | ltc_env_int(x, i, j, te) = lt_injX(LT_ENV(x, i, j, te))
509 :    
510 :     fun withEff ([], ol, nl, tenv) = false
511 :     | withEff (a::r, ol, nl, tenv) =
512 : league 53 let val (i, j) = tvDecode a
513 : monnier 16 val neweff =
514 :     if i > ol then (ol <> nl)
515 :     else (* case tcLookup(i, tenv)
516 :     of (NONE, n) => (nl - n) <> i
517 :     | (SOME ts, n) =>
518 :     (let val y = List.nth(ts, j)
519 :     in (case tc_outX y
520 :     of TC_VAR(ni, nj) =>
521 :     ((nj <> j) orelse ((ni+nl-n) <> i))
522 :     | _ => true)
523 :     end) *) true
524 :     in neweff orelse (withEff(r, ol, nl, tenv))
525 :     end
526 :    
527 :     in
528 :    
529 :     fun tcc_env(x, ol, nl, tenv) =
530 :     let val tvs = tc_vs x
531 :     in case tvs
532 :     of NONE => tcc_env_int(x, ol, nl, tenv)
533 :     | SOME [] => x
534 :     | SOME nvs => if withEff(nvs, ol, nl, tenv)
535 :     then tcc_env_int(x, ol, nl, tenv)
536 :     else x
537 :     end
538 :    
539 :     fun ltc_env(x, ol, nl, tenv) =
540 :     let val tvs = lt_vs x
541 :     in case tvs
542 :     of NONE => ltc_env_int(x, ol, nl, tenv)
543 :     | SOME [] => x
544 :     | SOME nvs => if withEff (nvs, ol, nl, tenv)
545 :     then ltc_env_int(x, ol, nl, tenv)
546 :     else x
547 :     end
548 :    
549 :     end (* utility functions for lt_env and tc_env *)
550 :    
551 :    
552 :     (** utility functions for updating tycs and ltys *)
553 :     fun tyc_upd (tgt as ref(i : int, old : tycI, AX_NO), nt) =
554 :     (tgt := (i, TC_IND (nt, old), AX_NO))
555 :     | tyc_upd (tgt as ref(i : int, old : tycI, x as (AX_REG(false, _))), nt) =
556 :     (tgt := (i, TC_IND (nt, old), x))
557 :     | tyc_upd _ = bug "unexpected tyc_upd on already normalized tyc"
558 :    
559 :     fun lty_upd (tgt as ref(i : int, old : ltyI, AX_NO), nt) =
560 :     (tgt := (i, LT_IND (nt, old), AX_NO))
561 :     | lty_upd (tgt as ref(i : int, old : ltyI, x as (AX_REG(false, _))), nt) =
562 :     (tgt := (i, LT_IND (nt, old), x))
563 :     | lty_upd _ = bug "unexpected lty_upd on already normalized lty"
564 :    
565 :     (***************************************************************************
566 :     * UTILITY FUNCTIONS ON REASONING ABOUT REDUCTIONS *
567 :     ***************************************************************************)
568 :    
569 :     (** a list of constructor functions *)
570 :     val tcc_var = tc_injX o TC_VAR
571 :     val tcc_fn = tc_injX o TC_FN
572 :     val tcc_app = tc_injX o TC_APP
573 :     val tcc_seq = tc_injX o TC_SEQ
574 :     val tcc_proj = tc_injX o TC_PROJ
575 :     val tcc_fix = tc_injX o TC_FIX
576 :     val tcc_abs = tc_injX o TC_ABS
577 : monnier 45 val tcc_tup = tc_injX o TC_TUPLE
578 :     val tcc_parw = tc_injX o TC_PARROW
579 : monnier 24 val tcc_box = tc_injX o TC_BOX
580 : monnier 45 val tcc_real = tc_injX (TC_PRIM PT.ptc_real)
581 : monnier 16 val ltc_tyc = lt_injX o LT_TYC
582 :     val ltc_str = lt_injX o LT_STR
583 :     val ltc_pst = lt_injX o LT_PST
584 :     val ltc_fct = lt_injX o LT_FCT
585 :     val ltc_poly = lt_injX o LT_POLY
586 : monnier 45 val tcc_sum = tc_injX o TC_SUM
587 :     val tcc_token = tc_injX o TC_TOKEN
588 : monnier 16
589 :     (** the following function contains the procedure on how to
590 : monnier 45 * flatten the arguments and results of an arbitrary FLINT function
591 : monnier 16 *)
592 :     fun isKnown tc =
593 :     (case tc_outX(tc_whnm tc)
594 :     of (TC_PRIM _ | TC_ARROW _ | TC_BOX _ | TC_ABS _ | TC_PARROW _) => true
595 :     | (TC_CONT _ | TC_FIX _ | TC_SUM _ | TC_TUPLE _) => true
596 :     | TC_APP(tc, _) => isKnown tc
597 :     | TC_PROJ(tc, _) => isKnown tc
598 : monnier 45 | TC_TOKEN(k, x) => token_isKnown(k, x)
599 : monnier 16 | _ => false)
600 :    
601 :     and tc_autoflat tc =
602 :     let val ntc = tc_whnm tc
603 :     in (case tc_outX ntc
604 : monnier 45 of TC_TUPLE (_, [_]) => (* singleton record is not flattened to ensure
605 : monnier 16 isomorphism btw plambdatype and flinttype *)
606 :     (true, [ntc], false)
607 : monnier 45 | TC_TUPLE (_, ts) =>
608 :     if length ts < 10 then (true, ts, true)
609 :     else (true, [ntc], false) (* ZHONG added the magic number 10 *)
610 : monnier 16 | _ => if isKnown ntc then (true, [ntc], false)
611 :     else (false, [ntc], false))
612 :     end
613 :    
614 :     and tc_autotuple [x] = x
615 : monnier 45 | tc_autotuple xs =
616 :     if length xs < 10 then tcc_tup (RF_TMP, xs)
617 :     else bug "fatal error with tc_autotuple"
618 : monnier 16
619 :     and tcs_autoflat (flag, ts) =
620 :     if flag then (flag, ts)
621 :     else (case ts
622 : monnier 45 of [tc] => (let val ntc = tc_whnm tc
623 :     val (nraw, ntcs, _) = tc_autoflat ntc
624 : monnier 16 in (nraw, ntcs)
625 :     end)
626 :     | _ => bug "unexpected cooked multiples in tcs_autoflat")
627 :    
628 :     and lt_autoflat lt =
629 :     (case lt_outX(lt_whnm lt)
630 :     of LT_TYC tc =>
631 :     let val (raw, ts, flag) = tc_autoflat tc
632 :     in (raw, map ltc_tyc ts, flag)
633 :     end
634 :     | _ => (true, [lt], false))
635 :    
636 :     (** a special version of tcc_arw that does automatic flattening *)
637 : monnier 45 and tcc_arw (x as (FF_FIXED, _, _)) = tc_injX (TC_ARROW x)
638 :     | tcc_arw (x as (FF_VAR (true, true), _, _)) = tc_injX (TC_ARROW x)
639 :     | tcc_arw (b as (FF_VAR (b1, b2)), ts1, ts2) =
640 : monnier 16 let val (nb1, nts1) = tcs_autoflat (b1, ts1)
641 :     val (nb2, nts2) = tcs_autoflat (b2, ts2)
642 : monnier 45 in tc_injX (TC_ARROW(FF_VAR(nb1, nb2), nts1, nts2))
643 : monnier 16 end
644 :    
645 :     (** utility function to read the top-level of a tyc *)
646 :     and tc_lzrd t =
647 :     let fun g x =
648 :     (case tc_outX x
649 :     of TC_IND (tc, _) => g tc
650 :     | TC_ENV (tc, i, j, te) =>
651 :     let val ntc = g(h(tc, i, j, te))
652 :     in tyc_upd(x, ntc); ntc
653 :     end
654 :     | _ => x)
655 :    
656 :     and h (x, 0, 0, _) = g x
657 :     | h (x, ol, nl, tenv) =
658 :     let fun prop z = tcc_env(z, ol, nl, tenv)
659 :     in (case tc_outX x
660 :     of TC_VAR (i,j) =>
661 :     if (i <= ol) then
662 :     (let val et = tcLookup(i, tenv)
663 :     in case et
664 :     of (NONE, n) => tcc_var(nl - n, j)
665 :     | (SOME ts, n) =>
666 :     (let val y = List.nth(ts, j)
667 :     handle _ => raise tcUnbound
668 :     in h(y, 0, nl - n, initTycEnv)
669 :     end)
670 :     end)
671 :     else tcc_var(i-ol+nl, j)
672 :     | TC_NVAR _ => x
673 :     | TC_PRIM _ => x
674 :     | TC_FN (ks, tc) =>
675 :     let val tenv' = tcInsert(tenv, (NONE, nl))
676 :     in tcc_fn(ks, tcc_env(tc, ol+1, nl+1, tenv'))
677 :     end
678 :     | TC_APP (tc, tcs) => tcc_app(prop tc, map prop tcs)
679 :     | TC_SEQ tcs => tcc_seq (map prop tcs)
680 :     | TC_PROJ (tc, i) => tcc_proj(prop tc, i)
681 :     | TC_SUM tcs => tcc_sum (map prop tcs)
682 :     | TC_FIX ((n,tc,ts), i) =>
683 :     tcc_fix((n, prop tc, map prop ts), i)
684 :     | TC_ABS tc => tcc_abs (prop tc)
685 :     | TC_BOX tc => tcc_box (prop tc)
686 : monnier 45 | TC_TUPLE (rk, tcs) => tcc_tup (rk, map prop tcs)
687 : monnier 16 | TC_ARROW (r, ts1, ts2) =>
688 :     tcc_arw (r, map prop ts1, map prop ts2)
689 :     | TC_PARROW (t1, t2) => tcc_parw (prop t1, prop t2)
690 : monnier 45 | TC_TOKEN (k, t) => tcc_token(k, prop t)
691 : monnier 16 | TC_CONT _ => bug "unexpected TC_CONT in tc_lzrd"
692 :     | TC_IND (tc, _) => h(tc, ol, nl, tenv)
693 :     | TC_ENV(tc, ol', nl', tenv') =>
694 :     if ol = 0 then h(tc, ol', nl+nl', tenv')
695 :     else h(g x, ol, nl, tenv))
696 :     end (* function h *)
697 :     in if tcp_norm(t) then t else g t
698 :     end (* function tc_lzrd *)
699 :    
700 :     (** utility function to read the top-level of an lty *)
701 :     and lt_lzrd t =
702 :     let fun g x =
703 :     (case lt_outX x
704 :     of LT_IND (lt, _) => g lt
705 :     | LT_ENV(lt, i, j, te) =>
706 :     let val nlt = g(h(lt, i, j, te))
707 :     in lty_upd(x, nlt); nlt
708 :     end
709 :     | _ => x)
710 :    
711 :     and h (x, 0, 0, _) = g x
712 :     | h (x, ol, nl, tenv) =
713 :     let fun prop z = ltc_env(z, ol, nl, tenv)
714 :     in (case lt_outX x
715 :     of LT_TYC tc => ltc_tyc (tcc_env(tc, ol, nl, tenv))
716 :     | LT_STR ts => ltc_str (map prop ts)
717 :     | LT_PST its => ltc_pst (map (fn (i, t) => (i, prop t)) its)
718 :     | LT_FCT (ts1, ts2) => ltc_fct(map prop ts1, map prop ts2)
719 :     | LT_POLY (ks, ts) =>
720 :     let val tenv' = tcInsert(tenv, (NONE, nl))
721 :     in ltc_poly(ks,
722 :     map (fn t => ltc_env(t, ol+1, nl+1, tenv')) ts)
723 :     end
724 :     | LT_CONT _ => bug "unexpected LT_CONT in lt_lzrd"
725 :     | LT_IND (t, _) => h(t, ol, nl, tenv)
726 :     | LT_ENV (lt, ol', nl', tenv') =>
727 :     if ol = 0 then h(lt, ol', nl+nl', tenv')
728 :     else h(g x, ol, nl, tenv))
729 :     end (* function h *)
730 :     in if ltp_norm(t) then t else g t
731 :     end (* function lt_lzrd *)
732 :    
733 :     (** taking out the TC_IND indirection *)
734 :     and stripInd t = (case tc_outX t of TC_IND (x,_) => stripInd x | _ => t)
735 :    
736 :     (** normalizing an arbitrary tyc into a simple weak-head-normal-form *)
737 :     and tc_whnm t = if tcp_norm(t) then t else
738 :     let val nt = tc_lzrd t
739 :     in case (tc_outX nt)
740 :     of TC_APP(tc, tcs) =>
741 :     (let val tc' = tc_whnm tc
742 :     in case (tc_outX tc')
743 :     of TC_FN(ks, b) =>
744 :     let fun base () =
745 :     (b, 1, 0, tcInsert(initTycEnv,(SOME tcs, 0)))
746 :     val sp =
747 :     (case tc_outX b
748 :     of TC_ENV(b', ol', nl', te') =>
749 :     (case tcSplit te'
750 :     of SOME((NONE, n), te) =>
751 :     if (n = nl'-1) andalso (ol' > 0)
752 :     then (b', ol', n,
753 :     tcInsert(te, (SOME tcs, n)))
754 :     else base()
755 :     | _ => base())
756 :     | _ => base())
757 :     val res = tc_whnm(tcc_env sp)
758 :     in tyc_upd(nt, res); res
759 :     end
760 :     | ((TC_SEQ _) | (TC_TUPLE _) | (TC_ARROW _) | (TC_IND _)) =>
761 :     bug "unexpected tycs in tc_whnm-TC_APP"
762 :     | _ => let val xx = tcc_app(tc', tcs)
763 :     in stripInd xx
764 :     end
765 :     end)
766 :     | TC_PROJ(tc, i) =>
767 :     (let val tc' = tc_whnm tc
768 :     in (case (tc_outX tc')
769 :     of (TC_SEQ tcs) =>
770 :     let val res = List.nth(tcs, i)
771 :     handle _ => bug "TC_SEQ in tc_whnm"
772 :     val nres = tc_whnm res
773 :     in tyc_upd(nt, nres); nres
774 :     end
775 :     | ((TC_PRIM _) | (TC_NVAR _) | (TC_FIX _) | (TC_FN _) |
776 :     (TC_SUM _) | (TC_ARROW _) | (TC_ABS _) | (TC_BOX _) |
777 :     (TC_IND _) | (TC_TUPLE _)) =>
778 :     bug "unexpected tycs in tc_whnm-TC_PROJ"
779 :     | _ => let val xx = tcc_proj(tc', i)
780 :     in stripInd xx
781 :     end)
782 :     end)
783 : monnier 45 | TC_TOKEN(k, tc) =>
784 :     (let val tc' = tc_whnm tc
785 :     in if token_whnm k tc'
786 :     then let val xx = tcc_token(k, tc') in stripInd xx end
787 :     else let val res = token_reduce(k, tc')
788 :     val nres = tc_whnm res
789 :     in tyc_upd(nt, nres); nres
790 :     end
791 :     end)
792 : monnier 16 | TC_IND (tc, _) => tc_whnm tc
793 :     | TC_ENV _ => bug "unexpected TC_ENV in tc_whnm"
794 :     | _ => nt
795 :     end (* function tc_whnm *)
796 :    
797 :     (** normalizing an arbitrary lty into the simple weak-head-normal-form *)
798 :     and lt_whnm t = if ltp_norm(t) then t else
799 :     let val nt = lt_lzrd t
800 :     in case (lt_outX nt)
801 :     of LT_TYC tc => ltc_tyc(tc_whnm tc)
802 :     | _ => nt
803 :     end (* function lt_whnm *)
804 :    
805 :     (** normalizing an arbitrary tyc into the standard normal form *)
806 :     fun tc_norm t = if (tcp_norm t) then t else
807 :     let val nt = tc_whnm t
808 :     in if (tcp_norm nt) then nt
809 :     else
810 :     (let val res =
811 :     (case (tc_outX nt)
812 :     of TC_FN (ks, tc) => tcc_fn(ks, tc_norm tc)
813 :     | TC_APP (tc, tcs) =>
814 :     tcc_app(tc_norm tc, map tc_norm tcs)
815 :     | TC_SEQ tcs => tcc_seq(map tc_norm tcs)
816 :     | TC_PROJ (tc, i) => tcc_proj(tc_norm tc, i)
817 :     | TC_SUM tcs => tcc_sum (map tc_norm tcs)
818 :     | TC_FIX ((n,tc,ts), i) =>
819 :     tcc_fix((n, tc_norm tc, map tc_norm ts), i)
820 :     | TC_ABS tc => tcc_abs(tc_norm tc)
821 :     | TC_BOX tc => tcc_box(tc_norm tc)
822 : monnier 45 | TC_TUPLE (rk, tcs) => tcc_tup(rk, map tc_norm tcs)
823 : monnier 16 | TC_ARROW (r, ts1, ts2) =>
824 :     tcc_arw(r, map tc_norm ts1, map tc_norm ts2)
825 :     | TC_PARROW (t1, t2) => tcc_parw(tc_norm t1, tc_norm t2)
826 : monnier 45 | TC_TOKEN (k, t) => tcc_token(k, tc_norm t)
827 : monnier 16 | TC_IND (tc, _) => tc_norm tc
828 :     | TC_ENV _ => bug "unexpected tycs in tc_norm"
829 :     | _ => nt)
830 :     in tyc_upd(nt, res); res
831 :     end)
832 :     end (* function tc_norm *)
833 :    
834 :     (** normalizing an arbitrary lty into the standard normal form *)
835 :     fun lt_norm t = if (ltp_norm t) then t else
836 :     let val nt = lt_lzrd t
837 :     in if (ltp_norm nt) then nt
838 :     else
839 :     (let val res =
840 :     (case lt_outX nt
841 :     of LT_TYC tc => ltc_tyc(tc_norm tc)
842 :     | LT_STR ts => ltc_str(map lt_norm ts)
843 :     | LT_PST its =>
844 :     ltc_pst(map (fn (i, t) => (i, lt_norm t)) its)
845 :     | LT_FCT (ts1, ts2) =>
846 :     ltc_fct(map lt_norm ts1, map lt_norm ts2)
847 :     | LT_POLY (ks, ts) => ltc_poly (ks, map lt_norm ts)
848 :     | LT_IND (lt, _) => lt_norm lt
849 :     | _ => bug "unexpected ltys in lt_norm")
850 :     in lty_upd(nt, res); res
851 :     end)
852 :     end (* function lt_norm *)
853 :    
854 :     (***************************************************************************
855 : monnier 45 * REGISTER A NEW TOKEN TYC --- TC_WRAP *
856 :     ***************************************************************************)
857 :    
858 :     (** we add a new constructor named TC_RBOX through the token facility *)
859 :     local val name = "TC_WRAP"
860 :     val abbrev = "WR"
861 :     val is_known = fn _ => true (* why is this ? *)
862 :     fun tcc_tok k t = tcc_token(k, t)
863 :    
864 :     fun unknown tc =
865 :     (case tc_outX tc
866 :     of (TC_VAR _ | TC_NVAR _) => true
867 :     | (TC_APP(tc, _)) => unknown tc
868 :     | (TC_PROJ(tc, _)) => unknown tc
869 :     | _ => false)
870 :    
871 :     fun flex_tuple ts =
872 :     let fun hhh(x::r, ukn, wfree) =
873 :     let fun iswp tc =
874 :     (case tc_outX tc
875 :     of TC_TOKEN(k', t) => (* WARNING: need check k' *)
876 :     (case tc_outX t
877 :     of TC_PRIM pt => false
878 :     | _ => true)
879 :     | _ => true)
880 :     in hhh(r, (unknown x) orelse ukn, (iswp x) andalso wfree)
881 :     end
882 :     | hhh([], ukn, wfree) = ukn andalso wfree
883 :     in hhh(ts, false, true)
884 :     end
885 :    
886 :     fun is_whnm tc =
887 :     (case tc_outX tc
888 :     of (TC_ARROW(FF_FIXED, [t], _)) => (unknown t)
889 :     | (TC_TUPLE(rf, ts)) => flex_tuple ts
890 :     | (TC_PRIM pt) => PT.unboxed pt
891 :     | _ => false)
892 :    
893 :     (* invariants: tc itself is in whnm but is_whnm tc = false *)
894 :     fun reduce_one (k, tc) =
895 :     (case tc_outX tc
896 :     of TC_TUPLE (rk, ts) =>
897 :     let fun hhh (x::r, nts, ukn) =
898 :     let val nx = tc_whnm x
899 :     val b1 = unknown nx
900 :     val nnx =
901 :     (case tc_outX nx
902 :     of TC_TOKEN(k', t) =>
903 :     if token_eq(k, k') then
904 :     (case tc_outX t
905 :     of TC_PRIM _ => t
906 :     | _ => nx)
907 :     else nx
908 :     | _ => nx)
909 :     in hhh(r, nnx::nts, b1 orelse ukn)
910 :     end
911 :     | hhh ([], nts, ukn) =
912 :     let val nt = tcc_tup(rk, rev nts)
913 :     in if ukn then tcc_token(k, nt) else nt
914 :     end
915 :     in hhh(ts, [], false)
916 :     end
917 :     | TC_ARROW (FF_FIXED, [_,_], [_]) => tc
918 :     | TC_ARROW (FF_FIXED, [t1], ts2 as [_]) =>
919 :     let val nt1 = tc_whnm t1
920 :     fun ggg z =
921 :     let val nz = tc_whnm z
922 :     in (case tc_outX nz
923 :     of TC_PRIM pt =>
924 :     if PT.unboxed pt then tcc_token(k, nz)
925 :     else nz
926 :     | _ => nz)
927 :     end
928 :     val (wp, nts1) =
929 :     (case tc_outX nt1
930 :     of TC_TUPLE(_, [x,y]) => (false, [ggg x, ggg y])
931 :     | TC_TOKEN(k', x) =>
932 :     if token_eq(k, k') then
933 :     (case (tc_outX x)
934 :     of TC_TUPLE(_, [y, z]) =>
935 :     (false, [ggg y, ggg z])
936 :     | _ => (true, [nt1]))
937 :     else (false, [nt1])
938 :     | _ => (unknown nt1, [nt1]))
939 :     val nt = tcc_arw(FF_FIXED, nts1, ts2)
940 :     in if wp then tcc_token(k, nt) else nt
941 :     end
942 :     | TC_ARROW (FF_FIXED, _, _) =>
943 :     bug "unexpected reduce_one on ill-formed FF_FIX arrow types"
944 :     | TC_ARROW (FF_VAR(b1,b2), ts1, ts2) =>
945 :     bug "calling reduce_one on FF_VAR arrow types"
946 :     | TC_PRIM pt =>
947 :     if PT.unboxed pt then
948 :     bug "calling reduce_one on an already-reduced whnm"
949 :     else tc
950 :     | TC_TOKEN(k', t) =>
951 :     if token_eq(k, k') then tc
952 :     else bug "unexpected token in reduce_one"
953 :     | (TC_BOX _ | TC_ABS _ | TC_PARROW _) =>
954 :     bug "unexpected tc_box/abs/parrow in reduce_one"
955 :     | TC_ENV _ => bug "unexpected TC_ENV in reduce_one"
956 :     | TC_IND _ => bug "unexpected TC_IND in reduce_one"
957 :     | _ => tc)
958 :    
959 :     in
960 :    
961 :     val wrap_token =
962 :     register_token {name=name, abbrev=abbrev, reduce_one=reduce_one,
963 :     is_whnm=is_whnm, is_known=is_known}
964 :    
965 :     end (* end of creating the box token for "tcc_rbox" *)
966 :    
967 :     (** testing if a tyc is a unknown constructor *)
968 :     fun tc_unknown tc = not (isKnown tc)
969 :    
970 :     (***************************************************************************
971 : monnier 16 * REBINDING THE INJECTION AND PROJECTION FUNCTIONS *
972 :     ***************************************************************************)
973 :     (** converting from the standard reps to the hash-consing reps *)
974 :     val tk_inj = tk_injX
975 :     val tc_inj = tc_injX
976 :     val lt_inj = lt_injX
977 :    
978 :     (** converting from the hash-consing reps to the standard reps *)
979 :     val tk_out = tk_outX
980 :     val tc_out = tc_outX o tc_whnm
981 :     val lt_out = lt_outX o lt_whnm
982 :    
983 :     (***************************************************************************
984 :     * UTILITY FUNCTIONS ON TESTING EQUIVALENCE *
985 :     ***************************************************************************)
986 :    
987 :     (** testing the equality of values of tkind, tyc, lty *)
988 : league 60 fun eqlist p (x::xs, y::ys) = (p(x,y)) andalso (eqlist p (xs, ys))
989 :     | eqlist p ([], []) = true
990 :     | eqlist _ _ = false
991 :    
992 : monnier 16 (** testing the "pointer" equality on normalized tkind, tyc, and lty *)
993 :     fun tk_eq (x: tkind, y) = (x = y)
994 :     fun tc_eq (x: tyc, y) = (x = y)
995 :     fun lt_eq (x: lty, y) = (x = y)
996 :    
997 :     (** testing the equivalence for arbitrary tkinds, tycs and ltys *)
998 :     val tk_eqv = tk_eq (* all tkinds are normalized *)
999 :    
1000 : league 60 local (* tyc equivalence utilities *)
1001 :     (* The efficiency of checking FIX equivalence could probably be
1002 :     * improved somewhat, but it doesn't seem so bad for my purposes right
1003 :     * now. Anyway, somebody might eventually want to do some profiling
1004 :     * and improve this. --league, 24 March 1998
1005 :     *)
1006 :    
1007 :     (* Profiling code, temporary?? *)
1008 :     structure Click =
1009 :     struct
1010 :     local
1011 :     val s_unroll = Stats.makeStat "FIX unrolls"
1012 :     in
1013 :     fun unroll() = Stats.addStat s_unroll 1
1014 :     end
1015 :     end (* Click *)
1016 : monnier 16
1017 : league 60 (** unrolling a fix, tyc -> tyc *)
1018 :     fun tc_unroll_fix tyc =
1019 :     case tc_outX tyc of
1020 :     (TC_FIX((n,tc,ts),i)) => let
1021 :     fun genfix i = tcc_fix ((n,tc,ts),i)
1022 :     val fixes = List.tabulate(n, genfix)
1023 :     val mu = tc
1024 :     val mu = if null ts then mu
1025 :     else tcc_app (mu,ts)
1026 :     val mu = tcc_app (mu, fixes)
1027 :     val mu = if n=1 then mu
1028 :     else tcc_proj (mu, i)
1029 :     in
1030 :     Click.unroll();
1031 :     mu
1032 :     end
1033 :     | _ => bug "unexpected non-FIX in tc_unroll_fix"
1034 : monnier 16
1035 : league 60 (* In order to check equality of two FIXes, we need to be able to
1036 :     * unroll them once, and check equality on the unrolled version, with
1037 :     * an inductive assumption that they ARE equal. The following code
1038 :     * supports making and checking these inductive assumptions.
1039 :     * Furthermore, we need to avoid unrolling any FIX more than once.
1040 :     *)
1041 :     structure TcDict = BinaryDict
1042 :     (struct
1043 :     type ord_key = tyc
1044 :     val cmpKey = tc_cmp
1045 :     end)
1046 :     (* for each tyc in this dictionary, we store a dictionary containing
1047 :     * tycs that are assumed equivalent to it.
1048 :     *)
1049 :     type eqclass = unit TcDict.dict
1050 :     type hyp = eqclass TcDict.dict
1051 :    
1052 :     (* the null hypothesis, no assumptions about equality *)
1053 :     val empty_eqclass : eqclass = TcDict.mkDict()
1054 :     val null_hyp : hyp = TcDict.mkDict()
1055 :    
1056 :     (* add assumption t1=t2 to current hypothesis. returns composite
1057 :     * hypothesis.
1058 :     *)
1059 :     fun assume_eq' (hyp, t1, t1eqOpt, t2) = let
1060 :     val t1eq = case t1eqOpt of SOME e => e | NONE => empty_eqclass
1061 :     val t1eq' = TcDict.insert (t1eq, t2, ())
1062 :     val hyp' = TcDict.insert (hyp, t1, t1eq')
1063 :     in
1064 :     hyp'
1065 :     end
1066 :    
1067 :     fun assume_eq (hyp, t1, t1eqOpt, t2, t2eqOpt) =
1068 :     assume_eq' (assume_eq' (hyp, t1, t1eqOpt, t2),
1069 :     t2, t2eqOpt, t1)
1070 :    
1071 :     (* check whether t1=t2 according to the hypothesis *)
1072 :     val eq_by_hyp : eqclass option * tyc -> bool
1073 :     = fn (NONE, t2) => false
1074 :     | (SOME eqclass, t2) =>
1075 :     isSome (TcDict.peek (eqclass, t2))
1076 :    
1077 :     (* have we made any assumptions about `t' already? *)
1078 :     val visited : eqclass option -> bool
1079 :     = isSome
1080 :    
1081 :     (* tc_eqv_generator, invariant: t1 and t2 are in the wh-normal form
1082 :     * eqop1 is the default equality to be used for tycs
1083 :     * eqop2 is used for body of FN, arguments in APP,
1084 :     * eqop3 is used for ABS and BOX.
1085 :     * eqop4 is used for arrow arguments and results
1086 :     * Each of these first takes the set of hypotheses.
1087 :     *)
1088 :     fun tc_eqv_gen (eqop1, eqop2, eqop3, eqop4) hyp (t1, t2) =
1089 :     case (tc_outX t1, tc_outX t2) of
1090 :     (TC_FIX((n1,tc1,ts1),i1), TC_FIX((n2,tc2,ts2),i2)) =>
1091 :     if not (!Control.CG.checkDatatypes) then true
1092 :     else let
1093 :     val t1eqOpt = TcDict.peek (hyp, t1)
1094 :     in
1095 :     (* first check the induction hypothesis. we only ever
1096 :     * make hypotheses about FIX nodes, so this test is okay
1097 :     * here. if assume_eq appears in other cases, this
1098 :     * test should be lifted outside the switch.
1099 :     *)
1100 :     if eq_by_hyp (t1eqOpt, t2) then true
1101 :     (* next try structural eq on the components. i'm not sure why
1102 :     * this part is necessary, but it does seem to be... --league,
1103 :     * 23 March 1998
1104 :     *)
1105 :     else
1106 :     (n1 = n2 andalso i1 = i2 andalso
1107 :     eqop1 hyp (tc1, tc2) andalso
1108 :     eqlist (eqop2 hyp) (ts1, ts2)) orelse
1109 :     (* not equal by inspection; we have to unroll it.
1110 :     * we prevent unrolling the same FIX twice by asking
1111 :     * the `visited' function.
1112 :     *)
1113 :     if visited t1eqOpt then false
1114 :     else let
1115 :     val t2eqOpt = TcDict.peek (hyp, t2)
1116 :     in
1117 :     if visited t2eqOpt then false
1118 :     else eqop1 (assume_eq (hyp, t1, t1eqOpt,
1119 :     t2, t2eqOpt))
1120 :     (tc_unroll_fix t1, tc_unroll_fix t2)
1121 :     end
1122 :     end
1123 :     | (TC_FN(ks1, b1), TC_FN(ks2, b2)) =>
1124 :     eqlist tk_eqv (ks1, ks2) andalso eqop2 hyp (b1, b2)
1125 :     | (TC_APP(a1, b1), TC_APP(a2, b2)) =>
1126 :     eqop1 hyp (a1, a2) andalso eqlist (eqop2 hyp) (b1, b2)
1127 :     | (TC_SEQ ts1, TC_SEQ ts2) =>
1128 :     eqlist (eqop1 hyp) (ts1, ts2)
1129 :     | (TC_SUM ts1, TC_SUM ts2) =>
1130 :     eqlist (eqop1 hyp) (ts1, ts2)
1131 :     | (TC_TUPLE (_, ts1), TC_TUPLE (_, ts2)) =>
1132 :     eqlist (eqop1 hyp) (ts1, ts2)
1133 :     | (TC_ABS a, TC_ABS b) =>
1134 :     eqop1 hyp (a, b)
1135 :     | (TC_ABS a, _) =>
1136 :     eqop3 hyp (a, t2)
1137 :     | (_, TC_ABS b) =>
1138 :     eqop3 hyp (t1, b)
1139 :     | (TC_BOX a, TC_BOX b) =>
1140 :     eqop1 hyp (a, b)
1141 :     | (TC_BOX a, _) =>
1142 :     eqop3 hyp (a, t2)
1143 :     | (_, TC_BOX b) =>
1144 :     eqop3 hyp (t1, b)
1145 :     | (TC_TOKEN(k1,t1), TC_TOKEN(k2,t2)) =>
1146 :     token_eq(k1,k2) andalso eqop1 hyp (t1,t2)
1147 :     | (TC_PROJ(a1, i1), TC_PROJ(a2, i2)) =>
1148 :     i1 = i2 andalso eqop1 hyp (a1, a2)
1149 :     | (TC_ARROW(r1, a1, b1), TC_ARROW(r2, a2, b2)) =>
1150 :     r1 = r2 andalso eqop4 hyp (a1, a2) andalso eqop4 hyp (b1, b2)
1151 :     | (TC_PARROW(a1, b1), TC_PARROW(a2, b2)) =>
1152 :     eqop1 hyp (a1, a2) andalso eqop1 hyp (b1, b2)
1153 :     | (TC_CONT ts1, TC_CONT ts2) =>
1154 :     eqlist (eqop1 hyp) (ts1, ts2)
1155 :     | _ => false
1156 :    
1157 :     fun never h (t1,t2) = false
1158 :    
1159 :     (** general equality for tycs *)
1160 :     fun tc_eqv' hyp (x as ref (_, _, AX_REG(true, _)),
1161 :     y as ref (_, _, AX_REG(true, _))) = tc_eq(x,y)
1162 :     | tc_eqv' hyp (x, y) = let
1163 :     val t1 = tc_whnm x
1164 :     val t2 = tc_whnm y
1165 :     in
1166 :     if tcp_norm t1 andalso tcp_norm t2 then tc_eq (t1, t2)
1167 :     else
1168 :     tc_eqv_gen (tc_eqv', tc_eqv', never, eqlist o tc_eqv')
1169 :     hyp (t1, t2)
1170 :     end (* tc_eqv' *)
1171 :    
1172 :     (* slightly relaxed constraints (???) *)
1173 :     fun tc_eqv_x' hyp (x, y) =
1174 : monnier 45 let val t1 = tc_whnm x
1175 :     val t2 = tc_whnm y
1176 :     in (if (tcp_norm t1) andalso (tcp_norm t2) then tc_eq(t1, t2)
1177 :     else false) orelse
1178 : league 60 (tc_eqv_gen (tc_eqv_x', tc_eqv_x', never, eqlist o tc_eqv_x')
1179 :     hyp (t1, t2))
1180 : monnier 45 end (* function tc_eqv_x *)
1181 :    
1182 : monnier 16 (** testing the equivalence of two tycs with relaxed constraints *)
1183 : league 60 fun mkeqop4 eqop1 hyp (ts1, ts2) =
1184 :     eqop1 hyp (tc_autotuple ts1, tc_autotuple ts2)
1185 :    
1186 :     fun tc_eqv_bx' hyp (x : tyc, y) =
1187 : monnier 16 let val t1 = tc_whnm x
1188 :     val t2 = tc_whnm y
1189 :     in (if (tcp_norm t1) andalso (tcp_norm t2) then tc_eq(t1, t2)
1190 :     else false) orelse
1191 : league 60 (tc_eqv_gen (tc_eqv_bx', tc_eqv_sp', never,
1192 :     mkeqop4 tc_eqv_bx')
1193 :     hyp (t1, t2))
1194 : monnier 16 end (* function tc_eqv_bx *)
1195 :    
1196 : league 60 and tc_eqv_sp' hyp (x : tyc, y) =
1197 : monnier 16 let val t1 = tc_whnm x
1198 :     val t2 = tc_whnm y
1199 :     in (if (tcp_norm t1) andalso (tcp_norm t2) then tc_eq(t1, t2)
1200 :     else false) orelse
1201 : league 60 (tc_eqv_gen (tc_eqv_sp', tc_eqv_sp', tc_eqv_sp',
1202 :     mkeqop4 tc_eqv_sp')
1203 :     hyp (t1, t2))
1204 : monnier 16 end (* function tc_eqv_sp *)
1205 :    
1206 : league 60 in (* tyc equivalence utilities *)
1207 :    
1208 :     (* TODO: explain the precise difference between these three notions of
1209 :     * equality! --league
1210 : monnier 16 *)
1211 : league 60 val tc_eqv = tc_eqv' null_hyp
1212 :     val tc_eqv_x = tc_eqv_x' null_hyp
1213 :     val tc_eqv_bx = tc_eqv_bx' null_hyp
1214 : monnier 16
1215 : league 60 end (* tyc equivalence utilities *)
1216 :    
1217 : monnier 16 (** lt_eqv_generator, invariant: x and y are in the wh-normal form *)
1218 :     fun lt_eqv_gen (eqop1, eqop2) (x : lty, y) =
1219 :     let fun sp (r, []) = true
1220 :     | sp (r, (i,t)::s) =
1221 :     (if (eqop1(List.nth(r,i),t))
1222 :     then sp(r,s) else false) handle _ => false
1223 :    
1224 :     fun pp ([], _) = true
1225 :     | pp (_, []) = true
1226 :     | pp (a as ((i,t)::l), b as ((j,s)::r)) =
1227 :     if i > j then pp(a,r)
1228 :     else if i < j then pp(l,b)
1229 :     else if (eqop1(t,s)) then pp(l,r) else false
1230 :    
1231 :     (* seq should be called if t1 and t2 are weak-head normal form *)
1232 :     fun seq (t1, t2) =
1233 :     (case (lt_outX t1, lt_outX t2)
1234 :     of (LT_POLY(ks1, b1), LT_POLY(ks2, b2)) =>
1235 : league 60 (eqlist tk_eqv (ks1, ks2)) andalso (eqlist eqop1 (b1, b2))
1236 : monnier 16 | (LT_FCT(as1, bs1), LT_FCT(as2, bs2)) =>
1237 : league 60 (eqlist eqop1 (as1, as2)) andalso (eqlist eqop1 (bs1, bs2))
1238 : monnier 16 | (LT_TYC a, LT_TYC b) => eqop2(a, b)
1239 : league 60 | (LT_STR s1, LT_STR s2) => eqlist eqop1 (s1, s2)
1240 : monnier 16 | (LT_PST s1, LT_PST s2) => pp(s1, s2)
1241 :     | (LT_PST s1, LT_STR s2) => sp(s2, s1)
1242 :     | (LT_STR s1, LT_PST s2) => sp(s1, s2)
1243 : league 60 | (LT_CONT s1, LT_CONT s2) => eqlist eqop1 (s1, s2)
1244 : monnier 16 | _ => false)
1245 :     in seq(x, y)
1246 :     end (* function lt_eqv_gen *)
1247 :    
1248 :     fun lt_eqv(x : lty, y) =
1249 :     let val seq = lt_eqv_gen (lt_eqv, tc_eqv)
1250 : monnier 45 in if ((ltp_norm x) andalso (ltp_norm y)) then
1251 :     (lt_eq(x, y)) orelse (seq(x, y))
1252 : monnier 16 else (let val t1 = lt_whnm x
1253 :     val t2 = lt_whnm y
1254 :     in if (ltp_norm t1) andalso (ltp_norm t2) then
1255 : monnier 45 (lt_eq(t1, t2)) orelse (seq(t1, t2))
1256 : monnier 16 else seq(t1, t2)
1257 :     end)
1258 :     end (* function lt_eqv *)
1259 :    
1260 : monnier 45 fun lt_eqv_x(x : lty, y) =
1261 :     let val seq = lt_eqv_gen (lt_eqv_x, tc_eqv_x)
1262 :     in if ((ltp_norm x) andalso (ltp_norm y)) then
1263 :     (lt_eq(x, y)) orelse (seq(x, y))
1264 :     else (let val t1 = lt_whnm x
1265 :     val t2 = lt_whnm y
1266 :     in if (ltp_norm t1) andalso (ltp_norm t2) then
1267 :     (lt_eq(t1, t2)) orelse (seq(t1, t2))
1268 :     else seq(t1, t2)
1269 :     end)
1270 :     end (* function lt_eqv *)
1271 :    
1272 : monnier 16 fun lt_eqv_bx (x : lty, y) =
1273 :     let val seq = lt_eqv_gen (lt_eqv_bx, tc_eqv_bx)
1274 :     in if (ltp_norm x) andalso (ltp_norm y) then
1275 :     (lt_eq(x, y)) orelse (seq(x, y))
1276 :     else (let val t1 = lt_whnm x
1277 :     val t2 = lt_whnm y
1278 :     in if (ltp_norm t1) andalso (ltp_norm t2) then
1279 :     (lt_eq(t1, t2)) orelse (seq(t1, t2))
1280 :     else seq(t1, t2)
1281 :     end)
1282 :     end (* function lt_eqv_bx *)
1283 :    
1284 : monnier 45 (** testing equivalence of fflags and rflags *)
1285 :     val ff_eqv : fflag * fflag -> bool = (op =)
1286 :     val rf_eqv : rflag * rflag -> bool = (op =)
1287 :    
1288 :     (***************************************************************************
1289 :     * UTILITY FUNCTIONS ON FINDING OUT THE DEPTH OF THE FREE TYC VARIABLES *
1290 :     ***************************************************************************)
1291 :     (** finding out the innermost binding depth for a tyc's free variables *)
1292 :     fun tc_depth (x, d) =
1293 :     let val tvs = tc_vs (tc_norm x)
1294 :     (* unfortunately we have to reduce everything to the normal form
1295 :     before we can talk about its list of free type variables.
1296 :     *)
1297 :     in case tvs
1298 :     of NONE => bug "unexpected case in tc_depth"
1299 :     | SOME [] => DI.top
1300 : league 53 | SOME (a::_) => d + 1 - (#1(tvDecode a))
1301 : monnier 45 end
1302 :    
1303 :     fun tcs_depth ([], d) = DI.top
1304 :     | tcs_depth (x::r, d) = Int.max(tc_depth(x, d), tcs_depth(r, d))
1305 :    
1306 : monnier 16 end (* toplevel local *)
1307 :     end (* abstraction LtyKernel *)
1308 :    

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0