comparison src/flat_env.sml @ 26:4ab19c19665f

Closure conversion
author Adam Chlipala <adamc@hcoop.net>
date Tue, 10 Jun 2008 15:56:33 -0400
parents
children d3cc191cb25f
comparison
equal deleted inserted replaced
25:0a762c73824d 26:4ab19c19665f
1 (* Copyright (c) 2008, Adam Chlipala
2 * All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *
7 * - Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 * - Redistributions in binary form must reproduce the above copyright notice,
10 * this list of conditions and the following disclaimer in the documentation
11 * and/or other materials provided with the distribution.
12 * - The names of contributors may not be used to endorse or promote products
13 * derived from this software without specific prior written permission.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25 * POSSIBILITY OF SUCH DAMAGE.
26 *)
27
28 structure FlatEnv :> FLAT_ENV = struct
29
30 open Flat
31
32 structure IM = IntBinaryMap
33
34
35 exception UnboundRel of int
36 exception UnboundNamed of int
37 exception UnboundF of int
38
39 type env = {
40 namedT : (string * typ option) IM.map,
41
42 relE : (string * typ) list,
43 namedE : (string * typ) IM.map,
44
45 F : (string * typ * typ) IM.map
46 }
47
48 val empty = {
49 namedT = IM.empty,
50
51 relE = [],
52 namedE = IM.empty,
53
54 F = IM.empty
55 }
56
57 fun pushTNamed (env : env) x n co =
58 {namedT = IM.insert (#namedT env, n, (x, co)),
59
60 relE = #relE env,
61 namedE = #namedE env,
62
63 F = #F env}
64
65 fun lookupTNamed (env : env) n =
66 case IM.find (#namedT env, n) of
67 NONE => raise UnboundNamed n
68 | SOME x => x
69
70 fun pushERel (env : env) x t =
71 {namedT = #namedT env,
72
73 relE = (x, t) :: #relE env,
74 namedE = #namedE env,
75
76 F = #F env}
77
78 fun lookupERel (env : env) n =
79 (List.nth (#relE env, n))
80 handle Subscript => raise UnboundRel n
81
82 fun listERels (env : env) = #relE env
83
84 fun pushENamed (env : env) x n t =
85 {namedT = #namedT env,
86
87 relE = #relE env,
88 namedE = IM.insert (#namedE env, n, (x, t)),
89
90 F = #F env}
91
92 fun lookupENamed (env : env) n =
93 case IM.find (#namedE env, n) of
94 NONE => raise UnboundNamed n
95 | SOME x => x
96
97 fun pushF (env : env) n x dom ran =
98 {namedT = #namedT env,
99
100 relE = #relE env,
101 namedE = #namedE env,
102
103 F = IM.insert (#F env, n, (x, dom, ran))}
104
105 fun lookupF (env : env) n =
106 case IM.find (#F env, n) of
107 NONE => raise UnboundF n
108 | SOME x => x
109
110 fun declBinds env (d, _) =
111 case d of
112 DVal (x, n, t, _) => pushENamed env x n t
113 | DFun (n, x, dom, ran, _) => pushF env n x dom ran
114
115 fun bbind env x =
116 case ElabEnv.lookupC ElabEnv.basis x of
117 ElabEnv.NotBound => raise Fail "FlatEnv.bbind: Not bound"
118 | ElabEnv.Rel _ => raise Fail "FlatEnv.bbind: Rel"
119 | ElabEnv.Named (n, _) => pushTNamed env x n NONE
120
121 val basis = empty
122 val basis = bbind basis "int"
123 val basis = bbind basis "float"
124 val basis = bbind basis "string"
125
126 end