7d3ae99f |
1 | (* HCoop Domtool (http://hcoop.sourceforge.net/) |
2 | * Copyright (c) 2006, Adam Chlipala |
3 | * |
4 | * This program is free software; you can redistribute it and/or |
5 | * modify it under the terms of the GNU General Public License |
6 | * as published by the Free Software Foundation; either version 2 |
7 | * of the License, or (at your option) any later version. |
8 | * |
9 | * This program is distributed in the hope that it will be useful, |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | * GNU General Public License for more details. |
13 | * |
14 | * You should have received a copy of the GNU General Public License |
15 | * along with this program; if not, write to the Free Software |
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
ae3a5b8c |
17 | *) |
7d3ae99f |
18 | |
19 | (* Pretty-printing Domtool configuration file ASTs *) |
20 | |
21 | structure Print :> PRINT = struct |
22 | |
23 | open Ast |
24 | |
25 | structure SM = TextIOPP |
26 | |
27 | structure PD = PPDescFn(SM) |
28 | open PD |
29 | |
30 | fun dBox ds = hovBox (PPS.Rel 1, ds) |
31 | fun dvBox ds = vBox (PPS.Rel 0, ds) |
32 | fun ivBox ds = vBox (PPS.Rel 1, ds) |
33 | |
34 | fun parenIf pn ds = |
35 | if pn then |
36 | dBox (string "(" :: ds @ [string ")"]) |
37 | else |
38 | dBox ds |
39 | |
40 | fun p_pred' pn (p, _) = |
41 | case p of |
42 | CRoot => string "Root" |
43 | | CConst s => string s |
44 | | CPrefix p => dBox [string "^", p_pred' true p] |
45 | | CNot p => dBox [string "!", p_pred' true p] |
46 | | CAnd (p1, p2) => |
47 | parenIf pn [p_pred' true p1, space 1, string "&", space 1, p_pred' true p2] |
48 | |
49 | val p_pred = p_pred' false |
50 | |
51 | fun p_predBoxed p = dBox [string "[", p_pred p, string "]"] |
52 | |
53 | fun p_typ' pn (t, _) = |
54 | case t of |
55 | TBase s => string s |
56 | | TList t => dBox [string "[", p_typ' false t, string "]"] |
57 | | TArrow (t1, t2) => |
58 | parenIf pn [p_typ' true t1, space 1, string "->", space 1, p_typ' true t2] |
59 | | TAction (p, r1, r2) => |
60 | parenIf pn [p_predBoxed p, space 1, p_record r1, space 1, |
e680130a |
61 | string "=>", space 1, p_record r2] |
2dc33fa4 |
62 | | TNested (p, t) => |
63 | parenIf pn [p_pred' false p, space 1, string "=>", space 1, p_typ' false t] |
64014a03 |
64 | |
65 | | TError => string "<error>" |
66 | | TUnif (_, ref (SOME t)) => p_typ' pn t |
67 | | TUnif (name, ref NONE) => string ("<" ^ name ^ ">") |
68 | |
7d3ae99f |
69 | and p_record r = |
70 | case StringMap.foldri (fn (name, t, d) => |
71 | SOME (case d of |
72 | NONE => dBox [string name, space 1, |
73 | string ":", space 1, p_typ t] |
74 | | SOME d => dBox [dBox [string name, space 1, |
75 | string ":", space 1, p_typ t], |
76 | string ",", space 1, d])) |
77 | NONE r of |
78 | NONE => string "{}" |
79 | | SOME d => dBox [string "{", d, string "}"] |
80 | |
81 | and p_typ t = p_typ' false t |
82 | |
83 | fun p_exp (e, _) = |
84 | case e of |
85 | EInt n => string (Int.toString n) |
86 | | EString s => string (String.concat ["\"", String.toString s, "\""]) |
87 | | EList es => |
88 | (case foldr (fn (e, d) => |
89 | SOME (case d of |
90 | NONE => p_exp e |
91 | | SOME d => dBox [p_exp e, string ",", space 1, d])) |
92 | NONE es of |
93 | NONE => string "[]" |
94 | | SOME d => dBox [string "[", d, string "]"]) |
95 | |
64014a03 |
96 | | ELam (x, NONE, e) => dBox [string "(\\", space 1, string x, space 1, |
97 | string "->", space 1, p_exp e, string ")"] |
98 | | ELam (x, SOME t, e) => dBox [string "(\\", space 1, string x, space 1, |
7d3ae99f |
99 | string ":", space 1, |
100 | dBox [string "(", p_typ t, string ")"], |
101 | space 1, string "->", space 1, p_exp e, string ")"] |
bf9b0bc3 |
102 | | EALam (x, c, e) => dBox [string "(\\\\", space 1, string x, space 1, |
103 | string ":", space 1, p_pred c, |
104 | space 1, string "->", space 1, p_exp e, string ")"] |
64014a03 |
105 | |
7d3ae99f |
106 | | EVar x => string x |
107 | | EApp (e1, e2) => dBox [string "(", p_exp e1, break {nsp = 1, offset = 0}, p_exp e2, string ")"] |
108 | |
e680130a |
109 | | ESkip => string "_" |
7d3ae99f |
110 | | ESet (x, e) => dBox [string x, space 1, string "=", space 1, p_exp e] |
111 | | EGet (x1, x2, e) => dBox [dBox [string x1, space 1, string "<-", |
112 | space 1, string x2, string ";", space 1], |
113 | p_exp e] |
114 | | ESeq es => dBox (valOf (foldr (fn (e, NONE) => SOME [p_exp e] |
d68ab27c |
115 | | (e, SOME ds) => SOME (dBox [p_exp e, string ";", newline] :: ds)) |
7d3ae99f |
116 | NONE es)) |
2dc33fa4 |
117 | | ELocal (e1, e2) => dBox [string "let", space 1, |
118 | p_exp e1, space 1, |
119 | string "in", space 1, |
120 | p_exp e2, space 1, |
121 | string "end"] |
e680130a |
122 | | EWith (e1, (ESkip, _)) => dBox [p_exp e1, space 1, string "with", space 1, string "end"] |
123 | | EWith (e1, e2) => dBox [p_exp e1, space 1, string "with", p_exp e2, space 1, string "end"] |
7d3ae99f |
124 | |
64014a03 |
125 | fun printd d = |
7d3ae99f |
126 | let |
127 | val myStream = SM.openOut {dst = TextIO.stdOut, |
128 | wid = 80} |
129 | in |
130 | description (myStream, d); |
131 | SM.newline myStream; |
132 | SM.closeStream myStream |
133 | end |
134 | |
2f68506c |
135 | fun preface (s, d) = printd (PD.hovBox (PD.PPS.Rel 0, |
136 | [PD.string s, PD.space 1, d])) |
137 | |
7d3ae99f |
138 | end |