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
|
(* Js_of_ocaml compiler
* http://www.ocsigen.org/js_of_ocaml/
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, with linking exception;
* either version 2.1 of the License, or (at your option) any later version.
*
* This program 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*)
open! Stdlib
open Code
let bound_variables { blocks; _ } ~f ~params ~cont:(pc, _) =
let bound_vars = ref Var.Map.empty in
let add_var x = bound_vars := Var.Map.add x (Var.fork x) !bound_vars in
List.iter ~f:add_var (f :: params);
let rec traverse blocks pc =
Code.traverse
{ fold = fold_children }
(fun pc _ ->
let block = Addr.Map.find pc blocks in
Freevars.iter_block_bound_vars add_var block;
List.iter
~f:(fun i ->
match i with
| Let (_, Closure (params, (pc', _), _)) ->
List.iter ~f:add_var params;
traverse blocks pc'
| _ -> ())
block.body)
pc
blocks
()
in
traverse blocks pc;
!bound_vars
let rec blocks_to_rename p pc lst =
Code.traverse
{ fold = Code.fold_children }
(fun pc lst ->
let block = Addr.Map.find pc p.blocks in
List.fold_left
~f:(fun lst i ->
match i with
| Let (_, Closure (_, (pc', _), _)) -> blocks_to_rename p pc' lst
| _ -> lst)
~init:(pc :: lst)
block.body)
pc
p.blocks
lst
let closure p ~f ~params ~cont =
let s = Subst.from_map (bound_variables p ~f ~params ~cont) in
let pc, args = cont in
let blocks = blocks_to_rename p pc [] in
let free_pc, m =
List.fold_left
~f:(fun (pc', m) pc -> pc' + 1, Addr.Map.add pc pc' m)
~init:(p.free_pc, Addr.Map.empty)
blocks
in
let blocks =
List.fold_left
~f:(fun blocks pc ->
let b = Addr.Map.find pc blocks in
let b = Subst.Including_Binders.And_Continuations.block m s b in
Addr.Map.add (Addr.Map.find pc m) b blocks)
~init:p.blocks
blocks
in
let p = { p with blocks; free_pc } in
p, s f, List.map ~f:s params, (Addr.Map.find pc m, List.map ~f:s args)
|
