adamc@29
|
1 (* Copyright (c) 2008, Adam Chlipala
|
adamc@29
|
2 * All rights reserved.
|
adamc@29
|
3 *
|
adamc@29
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@29
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@29
|
6 *
|
adamc@29
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@29
|
8 * this list of conditions and the following disclaimer.
|
adamc@29
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@29
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@29
|
11 * and/or other materials provided with the distribution.
|
adamc@29
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@29
|
13 * derived from this software without specific prior written permission.
|
adamc@29
|
14 *
|
adamc@29
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@29
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@29
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@29
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@29
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@29
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@29
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@29
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@29
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@29
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@29
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@29
|
26 *)
|
adamc@29
|
27
|
adamc@29
|
28 structure Cjrize :> CJRIZE = struct
|
adamc@29
|
29
|
adamc@29
|
30 structure L = Flat
|
adamc@29
|
31 structure L' = Cjr
|
adamc@29
|
32
|
adamc@29
|
33 structure Sm :> sig
|
adamc@29
|
34 type t
|
adamc@29
|
35
|
adamc@29
|
36 val empty : t
|
adamc@29
|
37 val find : t * (string * L.typ) list * (string * L'.typ) list -> t * int
|
adamc@29
|
38
|
adamc@29
|
39 val declares : t -> (int * (string * L'.typ) list) list
|
adamc@29
|
40 end = struct
|
adamc@29
|
41
|
adamc@29
|
42 structure FM = BinaryMapFn(struct
|
adamc@29
|
43 type ord_key = L.typ
|
adamc@29
|
44 val compare = FlatUtil.Typ.compare
|
adamc@29
|
45 end)
|
adamc@29
|
46
|
adamc@29
|
47 type t = int * int FM.map * (int * (string * L'.typ) list) list
|
adamc@29
|
48
|
adamc@29
|
49 val empty = (0, FM.empty, [])
|
adamc@29
|
50
|
adamc@29
|
51 fun find ((n, m, ds), xts, xts') =
|
adamc@29
|
52 let
|
adamc@29
|
53 val t = (L.TRecord xts, ErrorMsg.dummySpan)
|
adamc@29
|
54 in
|
adamc@29
|
55 case FM.find (m, t) of
|
adamc@29
|
56 NONE => ((n+1, FM.insert (m, t, n), (n, xts') :: ds), n)
|
adamc@29
|
57 | SOME i => ((n, m, ds), i)
|
adamc@29
|
58 end
|
adamc@29
|
59
|
adamc@29
|
60 fun declares (_, _, ds) = ds
|
adamc@29
|
61
|
adamc@29
|
62 end
|
adamc@29
|
63
|
adamc@29
|
64 fun cifyTyp ((t, loc), sm) =
|
adamc@29
|
65 case t of
|
adamc@29
|
66 L.TTop => ((L'.TTop, loc), sm)
|
adamc@29
|
67 | L.TFun (t1, t2) =>
|
adamc@29
|
68 let
|
adamc@29
|
69 val (_, sm) = cifyTyp (t1, sm)
|
adamc@29
|
70 val (_, sm) = cifyTyp (t2, sm)
|
adamc@29
|
71 in
|
adamc@29
|
72 ((L'.TFun, loc), sm)
|
adamc@29
|
73 end
|
adamc@29
|
74 | L.TCode (t1, t2) =>
|
adamc@29
|
75 let
|
adamc@29
|
76 val (t1, sm) = cifyTyp (t1, sm)
|
adamc@29
|
77 val (t2, sm) = cifyTyp (t2, sm)
|
adamc@29
|
78 in
|
adamc@29
|
79 ((L'.TCode (t1, t2), loc), sm)
|
adamc@29
|
80 end
|
adamc@29
|
81 | L.TRecord xts =>
|
adamc@29
|
82 let
|
adamc@29
|
83 val old_xts = xts
|
adamc@29
|
84 val (xts, sm) = ListUtil.foldlMap (fn ((x, t), sm) =>
|
adamc@29
|
85 let
|
adamc@29
|
86 val (t, sm) = cifyTyp (t, sm)
|
adamc@29
|
87 in
|
adamc@29
|
88 ((x, t), sm)
|
adamc@29
|
89 end)
|
adamc@29
|
90 sm xts
|
adamc@29
|
91 val (sm, si) = Sm.find (sm, old_xts, xts)
|
adamc@29
|
92 in
|
adamc@29
|
93 ((L'.TRecord si, loc), sm)
|
adamc@29
|
94 end
|
adamc@29
|
95 | L.TNamed n => ((L'.TNamed n, loc), sm)
|
adamc@52
|
96 | L.TFfi _ => raise Fail "Cjrize TFfi"
|
adamc@29
|
97
|
adamc@29
|
98 fun cifyExp ((e, loc), sm) =
|
adamc@29
|
99 case e of
|
adamc@29
|
100 L.EPrim p => ((L'.EPrim p, loc), sm)
|
adamc@29
|
101 | L.ERel n => ((L'.ERel n, loc), sm)
|
adamc@29
|
102 | L.ENamed n => ((L'.ENamed n, loc), sm)
|
adamc@52
|
103 | L.EFfi _ => raise Fail "Cjrize EFfi"
|
adamc@52
|
104 | L.EFfiApp _ => raise Fail "Cjrize EFfiApp"
|
adamc@29
|
105 | L.ECode n => ((L'.ECode n, loc), sm)
|
adamc@29
|
106 | L.EApp (e1, e2) =>
|
adamc@29
|
107 let
|
adamc@29
|
108 val (e1, sm) = cifyExp (e1, sm)
|
adamc@29
|
109 val (e2, sm) = cifyExp (e2, sm)
|
adamc@29
|
110 in
|
adamc@29
|
111 ((L'.EApp (e1, e2), loc), sm)
|
adamc@29
|
112 end
|
adamc@29
|
113
|
adamc@29
|
114 | L.ERecord xes =>
|
adamc@29
|
115 let
|
adamc@29
|
116 val old_xts = map (fn (x, _, t) => (x, t)) xes
|
adamc@29
|
117
|
adamc@29
|
118 val (xets, sm) = ListUtil.foldlMap (fn ((x, e, t), sm) =>
|
adamc@29
|
119 let
|
adamc@29
|
120 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
121 val (t, sm) = cifyTyp (t, sm)
|
adamc@29
|
122 in
|
adamc@29
|
123 ((x, e, t), sm)
|
adamc@29
|
124 end)
|
adamc@29
|
125 sm xes
|
adamc@29
|
126
|
adamc@29
|
127 val (sm, si) = Sm.find (sm, old_xts, map (fn (x, _, t) => (x, t)) xets)
|
adamc@29
|
128
|
adamc@29
|
129 val xes = map (fn (x, e, _) => (x, e)) xets
|
adamc@29
|
130 val xes = ListMergeSort.sort (fn ((x1, _), (x2, _)) => String.compare (x1, x2) = GREATER) xes
|
adamc@29
|
131 in
|
adamc@29
|
132 ((L'.ERecord (si, xes), loc), sm)
|
adamc@29
|
133 end
|
adamc@29
|
134 | L.EField (e, x) =>
|
adamc@29
|
135 let
|
adamc@29
|
136 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
137 in
|
adamc@29
|
138 ((L'.EField (e, x), loc), sm)
|
adamc@29
|
139 end
|
adamc@29
|
140
|
adamc@29
|
141 | L.ELet (xes, e) =>
|
adamc@29
|
142 let
|
adamc@29
|
143 val (xes, sm) = ListUtil.foldlMap (fn ((x, t, e), sm) =>
|
adamc@29
|
144 let
|
adamc@29
|
145 val (t, sm) = cifyTyp (t, sm)
|
adamc@29
|
146 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
147 in
|
adamc@29
|
148 ((x, t, e), sm)
|
adamc@29
|
149 end)
|
adamc@29
|
150 sm xes
|
adamc@29
|
151 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
152 in
|
adamc@29
|
153 ((L'.ELet (xes, e), loc), sm)
|
adamc@29
|
154 end
|
adamc@29
|
155
|
adamc@29
|
156 fun cifyDecl ((d, loc), sm) =
|
adamc@29
|
157 case d of
|
adamc@29
|
158 L.DVal (x, n, t, e) =>
|
adamc@29
|
159 let
|
adamc@29
|
160 val (t, sm) = cifyTyp (t, sm)
|
adamc@29
|
161 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
162 in
|
adamc@29
|
163 ((L'.DVal (x, n, t, e), loc), sm)
|
adamc@29
|
164 end
|
adamc@29
|
165 | L.DFun (n, x, dom, ran, e) =>
|
adamc@29
|
166 let
|
adamc@29
|
167 val (dom, sm) = cifyTyp (dom, sm)
|
adamc@29
|
168 val (ran, sm) = cifyTyp (ran, sm)
|
adamc@29
|
169 val (e, sm) = cifyExp (e, sm)
|
adamc@29
|
170 in
|
adamc@29
|
171 ((L'.DFun (n, x, dom, ran, e), loc), sm)
|
adamc@29
|
172 end
|
adamc@29
|
173
|
adamc@29
|
174 fun cjrize ds =
|
adamc@29
|
175 let
|
adamc@29
|
176 val (ds, sm) = ListUtil.foldlMap cifyDecl Sm.empty ds
|
adamc@29
|
177 in
|
adamc@29
|
178 List.revAppend (map (fn v => (L'.DStruct v, ErrorMsg.dummySpan)) (Sm.declares sm),
|
adamc@29
|
179 ds)
|
adamc@29
|
180 end
|
adamc@29
|
181
|
adamc@29
|
182 end
|