source: Deliverables/D2.2/8051-indexed-labels-branch/src/RTLabs/RTLabsPrinter.ml @ 1473

Last change on this file since 1473 was 1473, checked in by tranquil, 9 years ago
  • implemented partial redundancy elimination
  • added some tools for RTLabs, with a depth-first fold
  • prettier printing of RTLabs
File size: 8.3 KB
Line 
1
2let n_spaces n = String.make n ' '
3
4
5let rec print_size = function
6  | AST.SQ q -> Memory.string_of_quantity q
7  | AST.SProd l -> "struct {" ^ (print_size_list l) ^ "}"
8  | AST.SSum l -> "union {" ^ (print_size_list l) ^ "}"
9  | AST.SArray (i, se) ->
10    (print_size se) ^ "[" ^ (string_of_int i) ^ "]"
11and print_size_list l =
12  MiscPottier.string_of_list ", " print_size l
13
14let print_global n (x, size) =
15  Printf.sprintf "%s\"%s\" { %s }" (n_spaces n) x (print_size size)
16
17let print_globals n globs =
18  Printf.sprintf "%sglobals:\n%s"
19    (n_spaces n)
20    (List.fold_left (fun s g -> s ^ (print_global (n+2) g) ^ "\n") "" globs)
21
22
23let print_reg = Register.print
24
25let print_oreg = function
26  | None -> "_"
27  | Some r -> print_reg r
28
29let print_decl (r, t) =
30  (Primitive.print_type t) ^ " " ^ (Register.print r)
31
32let rec print_args args =
33  Printf.sprintf "[%s]" (MiscPottier.string_of_list ", " print_reg args)
34
35let print_result = function
36  | None -> "_"
37  | Some (r, t) -> (Primitive.print_type t) ^ " " ^ (Register.print r)
38
39let print_params r =
40  Printf.sprintf "(%s)" (MiscPottier.string_of_list ", " print_decl r)
41
42let print_locals r =
43  Printf.sprintf "%s" (MiscPottier.string_of_list ", " print_decl r)
44
45
46let print_cmp = function
47  | AST.Cmp_eq -> "="
48  | AST.Cmp_ne -> "!="
49  | AST.Cmp_gt -> ">"
50  | AST.Cmp_ge -> ">="
51  | AST.Cmp_lt -> "<"
52  | AST.Cmp_le -> "<="
53
54let rec print_size = function
55  | AST.SQ q -> Memory.string_of_quantity q
56  | AST.SProd l -> "struct {" ^ (print_size_list l) ^ "}"
57  | AST.SSum l -> "union {" ^ (print_size_list l) ^ "}"
58  | AST.SArray (i, se) ->
59    (print_size se) ^ "[" ^ (string_of_int i) ^ "]"
60and print_size_list l =
61  MiscPottier.string_of_list ", " print_size l
62
63let print_stacksize = print_size
64
65let print_offset (size, depth) =
66  (print_size size) ^ ", " ^ (string_of_int depth)
67
68let print_sizeof = print_size
69
70let print_cst = function
71  | AST.Cst_int i -> Printf.sprintf "imm_int %d" i
72  | AST.Cst_float f -> Printf.sprintf "imm_float %f" f
73  | AST.Cst_addrsymbol id -> Printf.sprintf "imm_addr \"%s\"" id
74  | AST.Cst_stack -> "imm_addr STACK"
75  | AST.Cst_offset off -> Printf.sprintf "imm_offset { %s }" (print_offset off)
76  | AST.Cst_sizeof t -> "imm_sizeof (" ^ (print_size t) ^ ")"
77
78let string_of_signedness = function
79  | AST.Signed -> "s"
80  | AST.Unsigned -> "u"
81
82let string_of_int_type (size, sign) =
83  Printf.sprintf "%d%s" size (string_of_signedness sign)
84
85let print_op1 op r = Printf.sprintf "%s %s"
86  (match op with
87  | AST.Op_cast (int_type, dest_size) ->
88    Printf.sprintf "int%sto%d" (string_of_int_type int_type) dest_size
89  | AST.Op_negint -> "-"
90  | AST.Op_notbool -> "!"
91  | AST.Op_notint -> "!i"
92  | AST.Op_id -> ""
93  | AST.Op_ptrofint -> "ptrofint"
94  | AST.Op_intofptr -> "intofptr")
95        (print_reg r)
96
97let print_op2 op r s = Printf.sprintf "%s %s %s"
98  (print_reg r)
99  (match op with
100  | AST.Op_add -> "+"
101  | AST.Op_sub -> "-"
102  | AST.Op_mul -> "*"
103  | AST.Op_div -> "/"
104  | AST.Op_divu -> "/u"
105  | AST.Op_mod -> "mod"
106  | AST.Op_modu -> "modu"
107  | AST.Op_and -> "and"
108  | AST.Op_or -> "or"
109  | AST.Op_xor -> "xor"
110  | AST.Op_shl -> "<<"
111  | AST.Op_shr -> ">>"
112  | AST.Op_shru -> ">>u"
113  | AST.Op_cmp cmp -> print_cmp cmp
114  | AST.Op_addp -> "+p"
115  | AST.Op_subp -> "-p"
116  | AST.Op_subpp -> "-pp"
117  | AST.Op_cmpp cmp -> (print_cmp cmp) ^ "p"
118  | AST.Op_cmpu cmp -> (print_cmp cmp) ^ "u")
119        (print_reg s)
120
121
122(*
123let print_addressing = function
124  | RTLabs.Aindexed off -> Printf.sprintf "{ %s }" (print_offset off)
125  | RTLabs.Aindexed2 -> "add"
126  | RTLabs.Aglobal (id, off) ->
127    Printf.sprintf "{ %s }(\"%s\")" (print_offset off) id
128  | RTLabs.Abased (id, off) ->
129    Printf.sprintf "add, { %s }(\"%s\")" (print_offset off) id
130  | RTLabs.Ainstack off -> Printf.sprintf "{ %s }(STACK)" (print_offset off)
131*)
132
133
134let rec print_table = function
135  | [] -> ""
136  | [lbl] -> lbl
137  | lbl :: tbl -> lbl ^ ", " ^ (print_table tbl)
138
139
140let print_statement = function
141  | RTLabs.St_skip lbl -> "--> " ^ lbl
142  | RTLabs.St_cost (cost_lbl, lbl) ->
143    let cost_lbl = CostLabel.string_of_cost_label ~pretty:true cost_lbl in
144    Printf.sprintf "emit %s --> %s" cost_lbl lbl
145  | RTLabs.St_ind_0 (i, lbl) ->
146    Printf.sprintf "index %d --> %s" i lbl
147  | RTLabs.St_ind_inc (i, lbl) ->
148    Printf.sprintf "increment %d --> %s" i lbl
149  | RTLabs.St_cst (destr, cst, lbl) ->
150      Printf.sprintf "%s := %s --> %s"
151        (print_reg destr)
152        (print_cst cst)
153        lbl
154  | RTLabs.St_op1 (op1, destr, srcr, lbl) ->
155      Printf.sprintf "%s := %s --> %s"
156        (print_reg destr)
157  (print_op1 op1 srcr)
158        lbl
159  | RTLabs.St_op2 (op2, destr, srcr1, srcr2, lbl) ->
160      Printf.sprintf "%s := %s --> %s"
161        (print_reg destr)
162  (print_op2 op2 srcr1 srcr2)
163        lbl
164  | RTLabs.St_load (q, addr, destr, lbl) ->
165      Printf.sprintf "%s := *(%s) %s --> %s"
166  (print_reg destr)
167        (Memory.string_of_quantity q)
168        (print_reg addr)
169        lbl
170  | RTLabs.St_store (q, addr, srcr, lbl) ->
171      Printf.sprintf "*(%s)%s := %s --> %s"
172  (Memory.string_of_quantity q)
173        (print_reg addr)
174        (print_reg srcr)
175        lbl
176  | RTLabs.St_call_id (f, args, Some r, sg, lbl) ->
177      Printf.sprintf "%s := \"%s\"(%s) : %s --> %s"
178  (print_reg r)
179  f
180        (print_args args)
181        (Primitive.print_sig sg)
182        lbl
183  | RTLabs.St_call_id (f, args, None, sg, lbl) ->
184    Printf.sprintf "\"%s\"(%s) : %s --> %s"
185    f
186    (print_args args)
187    (Primitive.print_sig sg)
188    lbl
189  | RTLabs.St_call_ptr (f, args, Some r, sg, lbl) ->
190      Printf.sprintf "%s := *%s (%s) : %s --> %s"
191  (print_reg r)
192    (print_reg f)
193    (print_args args)
194    (Primitive.print_sig sg)
195    lbl
196  | RTLabs.St_call_ptr (f, args, None, sg, lbl) ->
197    Printf.sprintf "*%s (%s) : %s --> %s"
198    (print_reg f)
199    (print_args args)
200    (Primitive.print_sig sg)
201    lbl
202  | RTLabs.St_tailcall_id (f, args, sg) ->
203      Printf.sprintf "tailcall \"%s\" (%s) : %s"
204        f
205        (print_args args)
206        (Primitive.print_sig sg)
207  | RTLabs.St_tailcall_ptr (f, args, sg) ->
208      Printf.sprintf "tailcall *%s (%s) : %s"
209        (print_reg f)
210        (print_args args)
211        (Primitive.print_sig sg)
212  | RTLabs.St_cond (r, lbl_true, lbl_false) ->
213      Printf.sprintf "%s? --> %s, %s"
214        (print_reg r)
215        lbl_true
216        lbl_false
217(*
218  | RTLabs.St_condcst (cst, t, lbl_true, lbl_false) ->
219      Printf.sprintf "(%s) %s --> %s, %s"
220        (Primitive.print_type t)
221        (print_cst cst)
222        lbl_true
223        lbl_false
224  | RTLabs.St_cond1 (op1, srcr, lbl_true, lbl_false) ->
225      Printf.sprintf "%s %s --> %s, %s"
226        (print_op1 op1)
227        (print_reg srcr)
228        lbl_true
229        lbl_false
230  | RTLabs.St_cond2 (op2, srcr1, srcr2, lbl_true, lbl_false) ->
231      Printf.sprintf "%s %s, %s --> %s, %s"
232        (print_op2 op2)
233        (print_reg srcr1)
234        (print_reg srcr2)
235        lbl_true
236        lbl_false
237*)
238  | RTLabs.St_jumptable (r, tbl) ->
239      Printf.sprintf "j_tbl %s --> %s"
240        (print_reg r)
241        (print_table tbl)
242  | RTLabs.St_return None -> Printf.sprintf "return"
243  | RTLabs.St_return (Some r) -> Printf.sprintf "return %s" (print_reg r)
244
245
246let print_graph n c entry =
247  let f lbl stmt s =
248    Printf.sprintf "%s%s: %s\n%s"
249      (n_spaces n)
250      lbl
251      (print_statement stmt)
252      s in
253        let f' lbl stmt (reach, s) =
254    (Label.Set.add lbl reach, f lbl stmt s) in
255        let (reachable, str) =
256                RTLabsUtilities.dfs_fold f' c entry (Label.Set.empty, "") in
257        let filter lbl _ = not (Label.Set.mem lbl reachable) in
258        let c_rest = Label.Map.filter filter c in
259        if Label.Map.is_empty c_rest then str else
260        let str' = Label.Map.fold f c_rest "" in
261        str ^ "DEAD NODES:\n" ^ str'
262       
263let print_internal_decl n f def =
264
265  Printf.sprintf
266    "%s\"%s\"%s\n%slocals: %s\n%sresult: %s\n%sstacksize: %s\n%sentry: %s\n%sexit: %s\n\n%s"
267    (n_spaces n)
268    f
269    (print_params def.RTLabs.f_params)
270    (n_spaces (n+2))
271    (print_locals def.RTLabs.f_locals)
272    (n_spaces (n+2))
273    (print_result def.RTLabs.f_result)
274    (n_spaces (n+2))
275    (print_stacksize def.RTLabs.f_stacksize)
276    (n_spaces (n+2))
277    def.RTLabs.f_entry
278    (n_spaces (n+2))
279    def.RTLabs.f_exit
280    (print_graph (n+2) def.RTLabs.f_graph def.RTLabs.f_entry)
281
282
283let print_external_decl n f def =
284  Printf.sprintf "%sextern \"%s\": %s\n"
285    (n_spaces n)
286    f
287    (Primitive.print_sig def.AST.ef_sig)
288
289
290let print_fun_decl n (f, def) = match def with
291  | RTLabs.F_int def -> print_internal_decl n f def
292  | RTLabs.F_ext def -> print_external_decl n f def
293
294let print_fun_decls n functs =
295  List.fold_left (fun s f -> s ^ (print_fun_decl n f) ^ "\n\n") ""
296    functs
297
298
299let print_program p =
300  Printf.sprintf "program:\n\n\n%s\n\n%s"
301    (print_globals 2 p.RTLabs.vars)
302    (print_fun_decls 2 p.RTLabs.functs)
Note: See TracBrowser for help on using the repository browser.