-
Notifications
You must be signed in to change notification settings - Fork 125
/
imperativeScope.ml
126 lines (94 loc) · 2.87 KB
/
imperativeScope.ml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
(*
Copyright © 2011 MLstate
This file is part of OPA.
OPA is free software: you can redistribute it and/or modify it under the
terms of the GNU Affero General Public License, version 3, as published by
the Free Software Foundation.
OPA is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
more details.
You should have received a copy of the GNU Affero General Public License
along with OPA. If not, see <http://www.gnu.org/licenses/>.
*)
(**
Management of imperative scope
@author Vincent Benayoun
*)
module type IMPERATIVE_SCOPE =
sig
(**
The mutable type of the instance
*)
type 'a t
type elt
(** create a new structure given a size for blocks *)
val create : int -> 'a t
(** reset the structure to its initial state *)
val reset : 'a t -> unit
(** push a new block *)
val push : 'a t -> unit
(** pop the top block *)
val pop : 'a t -> unit
(** fold the top block *)
val fold : (elt -> 'a -> 'acc -> 'acc) -> 'a t -> 'acc -> 'acc
(** bind the elt to 'a in the top block *)
val bind : 'a t -> elt -> 'a -> unit
(** unbind the elt from the top block *)
val unbind : 'a t -> elt -> unit
(** find the first binded data to the elt (top down) *)
val find_opt : 'a t -> elt -> 'a option
end
module type ARG =
sig
type elt
type 'a block
val create : int -> 'a block
val fold : (elt -> 'a -> 'acc -> 'acc) -> 'a block -> 'acc -> 'acc
val bind : 'a block -> elt -> 'a -> unit
val unbind : 'a block -> elt -> unit
val find_opt: 'a block -> elt -> 'a option
end
module Make(Arg : ARG) : IMPERATIVE_SCOPE with type elt = Arg.elt =
struct
type 'a t = 'a Arg.block list ref
type elt = Arg.elt
let block_size = ref 0
let new_stack() = [Arg.create !block_size]
let create n = block_size := n; ref (new_stack())
let reset s = s := new_stack()
let push s = s := (Arg.create !block_size)::!s
let pop s =
match !s with
| [_] -> s := new_stack()
| _ -> s := List.tl !s
let fold f s = Arg.fold f (List.hd !s)
let bind s e v = Arg.bind (List.hd !s) e v
let unbind s e = Arg.unbind (List.hd !s) e
let rec find_opt s e =
let rec aux l =
match l with
| [] -> None
| hd::tl ->
let found = Arg.find_opt hd e in
match found with
| Some _ -> found
| None -> aux tl
in
aux !s
end
module Default (Arg : sig type elt end) = Make(
struct
type elt = Arg.elt
type 'a block = (elt, 'a) Hashtbl.t
let create n = Hashtbl.create n
let fold f = Hashtbl.fold f
let bind b e v = Hashtbl.add b e v
let unbind b e = Hashtbl.remove b e
let find_opt b e =
try
Some (Hashtbl.find b e)
with
| Not_found -> None
end
)