aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorMarijn Haverbeke <[email protected]>2011-03-24 16:33:20 +0100
committerGraydon Hoare <[email protected]>2011-03-31 14:41:39 +0000
commite7e6f396d888574e7184119818ac3300f10aacbf (patch)
tree143c2c918a5123d6ed07b8ee61909981d3aa47e5 /src
parentTweak build command on rustc. (diff)
downloadrust-e7e6f396d888574e7184119818ac3300f10aacbf.tar.xz
rust-e7e6f396d888574e7184119818ac3300f10aacbf.zip
Preserve comments when pretty-printing.
The patch also includes a number of smaller fixes to the pretty-printer that were encountered on the way.
Diffstat (limited to 'src')
-rw-r--r--src/comp/driver/rustc.rs2
-rw-r--r--src/comp/front/lexer.rs80
-rw-r--r--src/comp/front/parser.rs4
-rw-r--r--src/comp/pretty/pp.rs221
-rw-r--r--src/comp/pretty/pprust.rs1541
5 files changed, 1060 insertions, 788 deletions
diff --git a/src/comp/driver/rustc.rs b/src/comp/driver/rustc.rs
index 4c6da407..9ab21622 100644
--- a/src/comp/driver/rustc.rs
+++ b/src/comp/driver/rustc.rs
@@ -77,7 +77,7 @@ impure fn pretty_print_input(session.session sess,
auto def = tup(0, 0);
auto p = front.parser.new_parser(sess, env, def, input);
auto crate = front.parser.parse_crate_from_source_file(p);
- pretty.pprust.print_ast(crate.node.module, std.io.stdout());
+ pretty.pprust.print_file(crate.node.module, input, std.io.stdout());
}
fn warn_wrong_compiler() {
diff --git a/src/comp/front/lexer.rs b/src/comp/front/lexer.rs
index 86b872da..2229a4e6 100644
--- a/src/comp/front/lexer.rs
+++ b/src/comp/front/lexer.rs
@@ -1,5 +1,6 @@
import std.io;
import std._str;
+import std._vec;
import std._int;
import std.map;
import std.map.hashmap;
@@ -781,6 +782,85 @@ impure fn next_token(reader rdr) -> token.token {
fail;
}
+tag cmnt_ {
+ cmnt_line(str);
+ cmnt_block(vec[str]);
+}
+type cmnt = rec(cmnt_ val, common.pos pos, bool space_after);
+
+impure fn consume_whitespace(reader rdr) -> uint {
+ auto lines = 0u;
+ while (is_whitespace(rdr.curr())) {
+ if (rdr.curr() == '\n') {lines += 1u;}
+ rdr.bump();
+ }
+ ret lines;
+}
+
+impure fn read_line_comment(reader rdr) -> cmnt {
+ auto p = rdr.get_curr_pos();
+ rdr.bump(); rdr.bump();
+ consume_whitespace(rdr);
+ auto val = "";
+ while (rdr.curr() != '\n') {
+ _str.push_char(val, rdr.curr());
+ rdr.bump();
+ }
+ ret rec(val=cmnt_line(val),
+ pos=p,
+ space_after=consume_whitespace(rdr) > 1u);
+}
+
+impure fn read_block_comment(reader rdr) -> cmnt {
+ auto p = rdr.get_curr_pos();
+ rdr.bump(); rdr.bump();
+ consume_whitespace(rdr);
+ let vec[str] lines = vec();
+ auto val = "";
+ auto level = 1;
+ while (true) {
+ if (rdr.curr() == '\n') {
+ _vec.push[str](lines, val);
+ val = "";
+ consume_whitespace(rdr);
+ } else {
+ if (rdr.curr() == '*' && rdr.next() == '/') {
+ level -= 1;
+ if (level == 0) {
+ rdr.bump(); rdr.bump();
+ _vec.push[str](lines, val);
+ break;
+ }
+ } else if (rdr.curr() == '/' && rdr.next() == '*') {
+ level += 1;
+ }
+ _str.push_char(val, rdr.curr());
+ rdr.bump();
+ }
+ }
+ ret rec(val=cmnt_block(lines),
+ pos=p,
+ space_after=consume_whitespace(rdr) > 1u);
+}
+
+impure fn gather_comments(str path) -> vec[cmnt] {
+ auto srdr = io.file_reader(path);
+ auto rdr = lexer.new_reader(srdr, path);
+ let vec[cmnt] comments = vec();
+ while (!rdr.is_eof()) {
+ while (true) {
+ consume_whitespace(rdr);
+ if (rdr.curr() == '/' && rdr.next() == '/') {
+ _vec.push[cmnt](comments, read_line_comment(rdr));
+ } else if (rdr.curr() == '/' && rdr.next() == '*') {
+ _vec.push[cmnt](comments, read_block_comment(rdr));
+ } else { break; }
+ }
+ next_token(rdr);
+ }
+ ret comments;
+}
+
//
// Local Variables:
diff --git a/src/comp/front/parser.rs b/src/comp/front/parser.rs
index a247c824..82f7712f 100644
--- a/src/comp/front/parser.rs
+++ b/src/comp/front/parser.rs
@@ -116,6 +116,8 @@ impure fn new_parser(session.session sess,
}
auto srdr = io.file_reader(path);
auto rdr = lexer.new_reader(srdr, path);
+ // Make sure npos points at first actual token.
+ lexer.consume_any_whitespace(rdr);
auto npos = rdr.get_curr_pos();
ret stdio_parser(sess, env, ftype, lexer.next_token(rdr),
npos, npos, initial_def._1, UNRESTRICTED, initial_def._0,
@@ -1748,8 +1750,8 @@ impure fn parse_block(parser p) -> ast.block {
}
}
- p.bump();
auto hi = p.get_span();
+ p.bump();
auto bloc = index_block(stmts, expr);
ret spanned[ast.block_](lo, hi, bloc);
diff --git a/src/comp/pretty/pp.rs b/src/comp/pretty/pp.rs
index 69eb01bc..b7b0c64a 100644
--- a/src/comp/pretty/pp.rs
+++ b/src/comp/pretty/pp.rs
@@ -4,9 +4,11 @@ import std._str;
tag boxtype {box_h; box_v; box_hv; box_align;}
tag contexttype {cx_h; cx_v;}
+tag scantype {scan_hv; scan_h; scan_none;}
tag token {
brk(uint);
+ hardbrk;
word(str);
cword(str); // closing token
open(boxtype, uint);
@@ -18,29 +20,45 @@ type context = rec(contexttype tp, uint indent);
type ps = @rec(mutable vec[context] context,
uint width,
io.writer out,
- mutable vec[token] buffered,
- mutable uint scandepth,
- mutable uint bufferedcol,
mutable uint col,
- mutable bool start_of_line);
+ mutable uint spaces,
+ mutable vec[token] buffered,
+ mutable scantype scanning,
+ mutable vec[boxtype] scandepth,
+ mutable uint scancol,
+ mutable bool start_of_line,
+ mutable bool start_of_box,
+ mutable bool potential_brk);
fn mkstate(io.writer out, uint width) -> ps {
let vec[context] stack = vec(rec(tp=cx_v, indent=0u));
let vec[token] buff = vec();
+ let vec[boxtype] sd = vec();
ret @rec(mutable context=stack,
width=width,
out=out,
- mutable buffered=buff,
- mutable scandepth=0u,
- mutable bufferedcol=0u,
mutable col=0u,
- mutable start_of_line=true);
+ mutable spaces=0u,
+ mutable buffered=buff,
+ mutable scanning=scan_none,
+ mutable scandepth=sd,
+ mutable scancol=0u,
+ mutable start_of_line=true,
+ mutable start_of_box=true,
+ mutable potential_brk=false);
+}
+
+impure fn write_spaces(ps p, uint i) {
+ while (i > 0u) {
+ i -= 1u;
+ p.out.write_str(" ");
+ }
}
impure fn push_context(ps p, contexttype tp, uint indent) {
before_print(p, false);
- _vec.push[context](p.context, rec(tp=tp, indent=base_indent(p)
- + indent));
+ _vec.push[context](p.context, rec(tp=tp, indent=indent));
+ p.start_of_box = true;
}
fn pop_context(ps p) {
@@ -49,15 +67,13 @@ fn pop_context(ps p) {
impure fn add_token(ps p, token tok) {
if (p.width == 0u) {direct_token(p, tok);}
- else if (p.scandepth == 0u) {do_token(p, tok);}
+ else if (p.scanning == scan_none) {do_token(p, tok);}
else {buffer_token(p, tok);}
}
impure fn direct_token(ps p, token tok) {
alt (tok) {
- case (brk(?sz)) {
- while (sz > 0u) {p.out.write_str(" "); sz -= 1u;}
- }
+ case (brk(?sz)) {write_spaces(p, sz);}
case (word(?w)) {p.out.write_str(w);}
case (cword(?w)) {p.out.write_str(w);}
case (_) {}
@@ -66,78 +82,75 @@ impure fn direct_token(ps p, token tok) {
impure fn buffer_token(ps p, token tok) {
p.buffered += vec(tok);
- p.bufferedcol += token_size(tok);
- alt (p.buffered.(0)) {
- case (brk(_)) {
- alt (tok) {
- case (brk(_)) {
- if (p.scandepth == 1u) {finish_break_scan(p);}
+ auto col = p.scancol;
+ p.scancol = col + token_size(tok);
+ if (p.scancol > p.width) {
+ finish_scan(p, false);
+ } else {
+ alt (tok) {
+ case (open(?tp,_)) {
+ _vec.push[boxtype](p.scandepth, tp);
+ if (p.scanning == scan_h) {
+ if (tp == box_h) {
+ check_potential_brk(p);
+ }
}
- case (open(box_h,_)) {p.scandepth += 1u;}
- case (open(_,_)) {finish_break_scan(p);}
- case (close) {
- p.scandepth -= 1u;
- if (p.scandepth == 0u) {finish_break_scan(p);}
+ }
+ case (close) {
+ _vec.pop[boxtype](p.scandepth);
+ if (_vec.len[boxtype](p.scandepth) == 0u) {
+ finish_scan(p, true);
}
- case (_) {}
}
- }
- case (open(_,_)) {
- if (p.bufferedcol > p.width) {finish_block_scan(p, cx_v);}
- else {
- alt (tok) {
- case (open(_,_)) {p.scandepth += 1u;}
- case (close) {
- p.scandepth -= 1u;
- if (p.scandepth == 0u) {finish_block_scan(p, cx_h);}
+ case (brk(_)) {
+ if (p.scanning == scan_h) {
+ if (p.scandepth.(_vec.len[boxtype](p.scandepth)-1u) == box_v) {
+ finish_scan(p, true);
}
- case (_) {}
}
}
+ case (_) {}
}
}
}
-impure fn finish_block_scan(ps p, contexttype tp) {
+impure fn check_potential_brk(ps p) {
+ for (boxtype tp in p.scandepth) {
+ if (tp != box_h) {ret;}
+ }
+ p.potential_brk = true;
+}
+
+impure fn finish_scan(ps p, bool fits) {
auto buf = p.buffered;
auto front = _vec.shift[token](buf);
- auto indent;
- alt (front){
- case (open(box_hv,?ind)) {
- indent = ind;
+ auto chosen_tp = cx_h;
+ if (!fits) {chosen_tp = cx_v;}
+ alt (front) {
+ case (open(box_hv, ?ind)) {
+ push_context(p, chosen_tp, base_indent(p) + ind);
}
case (open(box_align, _)) {
- indent = p.col - base_indent(p);
+ push_context(p, chosen_tp, p.col);
+ }
+ case (open(box_h, ?ind)) {
+ if (!fits && !p.start_of_box && !p.start_of_line && !p.potential_brk) {
+ line_break(p);
+ }
+ push_context(p, cx_h, base_indent(p) + ind);
}
}
- p.scandepth = 0u;
- p.buffered = vec();
- push_context(p, tp, indent);
+ p.scandepth = vec();
+ p.scanning = scan_none;
for (token t in buf) { add_token(p, t); }
}
-impure fn finish_break_scan(ps p) {
- auto buf = p.buffered;
- auto front = _vec.shift[token](buf);
- if (p.bufferedcol > p.width) {
- line_break(p);
- }
- else {
- auto width;
- alt (front) {case(brk(?w)) {width = w;}}
- auto i = 0u;
- while (i < width) {p.out.write_str(" "); i+=1u;}
- p.col += width;
- }
- p.scandepth = 0u;
+impure fn start_scan(ps p, token tok, scantype tp) {
p.buffered = vec();
- for (token t in buf) { add_token(p, t); }
-}
-
-impure fn start_scan(ps p, token tok) {
- p.buffered = vec(tok);
- p.scandepth = 1u;
- p.bufferedcol = p.col;
+ p.scancol = p.col;
+ p.scanning = tp;
+ buffer_token(p, tok);
+ p.potential_brk = false;
}
fn cur_context(ps p) -> context {
@@ -152,64 +165,91 @@ fn base_indent(ps p) -> uint {
}
}
+fn cx_is(contexttype a, contexttype b) -> bool {
+ if (a == b) {ret true;}
+ else {ret false;}
+}
+fn box_is(boxtype a, boxtype b) -> bool {
+ if (a == b) {ret true;}
+ else {ret false;}
+}
+
impure fn do_token(ps p, token tok) {
+ auto start_of_box = p.start_of_box;
+ p.start_of_box = false;
alt (tok) {
case (brk(?sz)) {
- alt (cur_context(p).tp) {
- case (cx_h) {
- before_print(p, false);
- start_scan(p, tok);
- }
- case (cx_v) {
- line_break(p);
- }
+ if (cx_is(cur_context(p).tp, cx_v) || sz + p.col > p.width) {
+ line_break(p);
+ }
+ else {
+ p.spaces += sz;
}
}
+ case (hardbrk) {
+ line_break(p);
+ }
case (word(?w)) {
+ auto len = _str.char_len(w);
+ if (len + p.col + p.spaces > p.width && !start_of_box &&
+ !p.start_of_line) {
+ line_break(p);
+ }
before_print(p, false);
p.out.write_str(w);
- p.col += _str.byte_len(w); // TODO char_len
+ p.col += len;
}
case (cword(?w)) {
before_print(p, true);
p.out.write_str(w);
- p.col += _str.byte_len(w); // TODO char_len
+ p.col += _str.char_len(w);
}
case (open(?tp, ?indent)) {
- alt (tp) {
- case (box_hv) {start_scan(p, tok);}
- case (box_align) {start_scan(p, tok);}
- case (box_h) {push_context(p, cx_h, indent);}
- case (box_v) {push_context(p, cx_v, indent);}
+ if (tp == box_v) {
+ push_context(p, cx_v, base_indent(p) + indent);
+ } else if (box_is(tp, box_h) && cx_is(cur_context(p).tp, cx_v)) {
+ push_context(p, cx_h, base_indent(p) + indent);
+ } else if (tp == box_h) {
+ p.start_of_box = start_of_box;
+ start_scan(p, tok, scan_h);
+ } else {
+ p.start_of_box = start_of_box;
+ start_scan(p, tok, scan_hv);
}
}
- case (close) {pop_context(p);}
+ case (close) {
+ pop_context(p);
+ }
}
}
impure fn line_break(ps p) {
p.out.write_str("\n");
p.col = 0u;
+ p.spaces = cur_context(p).indent;
p.start_of_line = true;
}
impure fn before_print(ps p, bool closing) {
if (p.start_of_line) {
p.start_of_line = false;
- auto ind;
- if (closing) {ind = base_indent(p);}
- else {ind = cur_context(p).indent;}
- p.col = ind;
- while (ind > 0u) {p.out.write_str(" "); ind -= 1u;}
+ if (closing) {p.spaces = base_indent(p);}
+ else {p.spaces = cur_context(p).indent;}
+ }
+ if (p.spaces > 0u) {
+ write_spaces(p, p.spaces);
+ p.col += p.spaces;
+ p.spaces = 0u;
}
}
fn token_size(token tok) -> uint {
alt (tok) {
case (brk(?sz)) {ret sz;}
- case (word(?w)) {ret _str.byte_len(w);}
- case (cword(?w)) {ret _str.byte_len(w);}
- case (open(_, _)) {ret 0u;} // TODO exception for V blocks?
+ case (hardbrk) {ret 0xFFFFFFu;}
+ case (word(?w)) {ret _str.char_len(w);}
+ case (cword(?w)) {ret _str.char_len(w);}
+ case (open(_, _)) {ret 0u;}
case (close) {ret 0u;}
}
}
@@ -224,3 +264,4 @@ impure fn cwrd(ps p, str wrd) {add_token(p, cword(wrd));}
impure fn space(ps p) {add_token(p, brk(1u));}
impure fn spaces(ps p, uint n) {add_token(p, brk(n));}
impure fn line(ps p) {add_token(p, brk(0u));}
+impure fn hardbreak(ps p) {add_token(p, hardbrk);}
diff --git a/src/comp/pretty/pprust.rs b/src/comp/pretty/pprust.rs
index 589ee0c3..ac5cec31 100644
--- a/src/comp/pretty/pprust.rs
+++ b/src/comp/pretty/pprust.rs
@@ -1,800 +1,949 @@
import std._vec;
import std._str;
+import std.io;
import std.option;
import front.ast;
-import pp.box; import pp.abox; import pp.vbox;
+import front.lexer;
import pp.end; import pp.wrd; import pp.space; import pp.line;
-import pp.ps;
-import foo = std.io;
-
-const uint indent_unit = 2u;
+const uint indent_unit = 4u;
const int as_prec = 5;
+const uint default_columns = 78u;
+
+type ps = @rec(pp.ps s,
+ option.t[vec[lexer.cmnt]] comments,
+ mutable uint cur_cmnt);
-impure fn print_ast(ast._mod _mod, std.io.writer out) {
- auto s = pp.mkstate(out, 80u);
- for (@ast.view_item vitem in _mod.view_items) {print_view_item(s, vitem);}
- line(s);
- for (@ast.item item in _mod.items) {print_item(s, item);}
+impure fn print_file(ast._mod _mod, str filename, io.writer out) {
+ auto cmnts = lexer.gather_comments(filename);
+ auto s = @rec(s=pp.mkstate(out, default_columns),
+ comments=option.some[vec[lexer.cmnt]](cmnts),
+ mutable cur_cmnt=0u);
+ print_mod(s, _mod);
}
fn ty_to_str(&@ast.ty ty) -> str {
- auto writer = std.io.string_writer();
- print_type(pp.mkstate(writer.get_writer(), 0u), ty);
- ret writer.get_str();
+ auto writer = io.string_writer();
+ auto s = @rec(s=pp.mkstate(writer.get_writer(), 0u),
+ comments=option.none[vec[lexer.cmnt]],
+ mutable cur_cmnt=0u);
+ print_type(s, ty);
+ ret writer.get_str();
}
impure fn hbox(ps s) {
- pp.hbox(s, indent_unit);
+ pp.hbox(s.s, indent_unit);
}
impure fn wrd1(ps s, str word) {
- wrd(s, word);
- space(s);
+ wrd(s.s, word);
+ space(s.s);
}
impure fn popen(ps s) {
- wrd(s, "(");
- abox(s);
+ wrd(s.s, "(");
+ pp.abox(s.s);
+}
+impure fn popen_h(ps s) {
+ wrd(s.s, "(");
+ pp.hbox(s.s, 0u);
}
impure fn pclose(ps s) {
- end(s);
- wrd(s, ")");
+ end(s.s);
+ wrd(s.s, ")");
}
impure fn bopen(ps s) {
- wrd1(s, "{");
- vbox(s, indent_unit);
- line(s);
+ wrd(s.s, "{");
+ pp.vbox(s.s, indent_unit);
+ line(s.s);
}
impure fn bclose(ps s) {
- end(s);
- pp.cwrd(s, "}");
+ end(s.s);
+ pp.cwrd(s.s, "}");
+}
+impure fn bclose_c(ps s, util.common.span span) {
+ maybe_print_comment(s, span.hi);
+ bclose(s);
}
impure fn commasep[IN](ps s, vec[IN] elts, impure fn (ps, &IN) op) {
- auto first = true;
- for (IN elt in elts) {
- if (first) {first = false;}
- else {wrd1(s, ",");}
- op(s, elt);
- }
+ auto first = true;
+ for (IN elt in elts) {
+ if (first) {first = false;}
+ else {wrd1(s, ",");}
+ op(s, elt);
+ }
}
-impure fn print_mt(ps s, &ast.mt mt) {
- alt (mt.mut) {
- case (ast.mut) { wrd1(s, "mutable"); }
- case (ast.maybe_mut) { wrd1(s, "mutable?"); }
- case (ast.imm) { /* nothing */ }
- }
- print_type(s, mt.ty);
+impure fn print_mod(ps s, ast._mod _mod) {
+ for (@ast.view_item vitem in _mod.view_items) {print_view_item(s, vitem);}
+ line(s.s);
+ for (@ast.item item in _mod.items) {print_item(s, item);}
+ print_remaining_comments(s);
}
impure fn print_type(ps s, &@ast.ty ty) {
- hbox(s);
- alt (ty.node) {
- case (ast.ty_nil) {wrd(s, "()");}
- case (ast.ty_bool) {wrd(s, "bool");}
- case (ast.ty_int) {wrd(s, "int");}
- case (ast.ty_uint) {wrd(s, "uint");}
- case (ast.ty_machine(?tm)) {wrd(s, util.common.ty_mach_to_str(tm));}
- case (ast.ty_char) {wrd(s, "char");}
- case (ast.ty_str) {wrd(s, "str");}
- case (ast.ty_box(?mt)) {wrd(s, "@"); print_mt(s, mt);}
- case (ast.ty_vec(?mt)) {wrd(s, "vec["); print_mt(s, mt); wrd(s, "]");}
- case (ast.ty_port(?t)) {wrd(s, "port["); print_type(s, t); wrd(s, "]");}
- case (ast.ty_chan(?t)) {wrd(s, "chan["); print_type(s, t); wrd(s, "]");}
- case (ast.ty_type) {wrd(s, "type");}
- case (ast.ty_tup(?elts)) {
- wrd(s, "tup");
- popen(s);
- auto f = print_mt;
- commasep[ast.mt](s, elts, f);
- pclose(s);
- }
- case (ast.ty_rec(?fields)) {
- wrd(s, "rec");
- popen(s);
- impure fn print_field(ps s, &ast.ty_field f) {
- hbox(s);
- print_mt(s, f.mt);
- space(s);
- wrd(s, f.ident);
- end(s);
- }
- auto f = print_field;
- commasep[ast.ty_field](s, fields, f);
- pclose(s);
- }
- case (ast.ty_obj(?methods)) {
- wrd1(s, "obj");
- bopen(s);
- for (ast.ty_method m in methods) {
- hbox(s);
- print_ty_fn(s, m.proto, option.some[str](m.ident),
- m.inputs, m.output);
- wrd(s, ";");
- end(s);
- line(s);
- }
- bclose(s);
- }
- case (ast.ty_fn(?proto,?inputs,?output)) {
- print_ty_fn(s, proto, option.none[str], inputs, output);
- }
- case (ast.ty_path(?path,_)) {
- print_path(s, path);
+ maybe_print_comment(s, ty.span.lo);
+ hbox(s);
+ alt (ty.node) {
+ case (ast.ty_nil) {wrd(s.s, "()");}
+ case (ast.ty_bool) {wrd(s.s, "bool");}
+ case (ast.ty_int) {wrd(s.s, "int");}
+ case (ast.ty_uint) {wrd(s.s, "uint");}
+ case (ast.ty_machine(?tm)) {wrd(s.s, util.common.ty_mach_to_str(tm));}
+ case (ast.ty_char) {wrd(s.s, "char");}
+ case (ast.ty_str) {wrd(s.s, "str");}
+ case (ast.ty_box(?mt)) {wrd(s.s, "@"); print_mt(s, mt);}
+ case (ast.ty_vec(?mt)) {
+ wrd(s.s, "vec["); print_mt(s, mt); wrd(s.s, "]");
+ }
+ case (ast.ty_port(?t)) {
+ wrd(s.s, "port["); print_type(s, t); wrd(s.s, "]");
+ }
+ case (ast.ty_chan(?t)) {
+ wrd(s.s, "chan["); print_type(s, t); wrd(s.s, "]");
+ }
+ case (ast.ty_type) {wrd(s.s, "type");}
+ case (ast.ty_tup(?elts)) {
+ wrd(s.s, "tup");
+ popen(s);
+ auto f = print_mt;
+ commasep[ast.mt](s, elts, f);
+ pclose(s);
+ }
+ case (ast.ty_rec(?fields)) {
+ wrd(s.s, "rec");
+ popen(s);
+ impure fn print_field(ps s, &ast.ty_field f) {
+ hbox(s);
+ print_mt(s, f.mt);
+ space(s.s);
+ wrd(s.s, f.ident);
+ end(s.s);
+ }
+ auto f = print_field;
+ commasep[ast.ty_field](s, fields, f);
+ pclose(s);
+ }
+ case (ast.ty_obj(?methods)) {
+ wrd1(s, "obj");
+ bopen(s);
+ for (ast.ty_method m in methods) {
+ hbox(s);
+ print_ty_fn(s, m.proto, option.some[str](m.ident),
+ m.inputs, m.output);
+ wrd(s.s, ";");
+ end(s.s);
+ line(s.s);
+ }
+ bclose_c(s, ty.span);
+ }
+ case (ast.ty_fn(?proto,?inputs,?output)) {
+ print_ty_fn(s, proto, option.none[str], inputs, output);
+ }
+ case (ast.ty_path(?path,_)) {
+ print_path(s, path);
+ }
}
- }
- end(s);
+ end(s.s);
}
impure fn print_item(ps s, @ast.item item) {
- hbox(s);
- alt (item.node) {
- case (ast.item_const(?id, ?ty, ?expr, _, _)) {
- wrd1(s, "const");
- print_type(s, ty);
- space(s);
- wrd1(s, id);
- wrd1(s, "=");
- print_expr(s, expr);
- wrd(s, ";");
- }
- case (ast.item_fn(?name,?_fn,?typarams,_,_)) {
- print_fn(s, _fn.decl, name, typarams);
- space(s);
- print_block(s, _fn.body);
- }
- case (ast.item_mod(?id,?_mod,_)) {
- wrd1(s, "mod");
- wrd1(s, id);
- bopen(s);
- for (@ast.item itm in _mod.items) {print_item(s, itm);}
- bclose(s);
- }
- case (ast.item_native_mod(?id,?nmod,_)) {
- wrd1(s, "native");
- alt (nmod.abi) {
- case (ast.native_abi_rust) {wrd1(s, "\"rust\"");}
- case (ast.native_abi_cdecl) {wrd1(s, "\"cdecl\"");}
- }
- wrd1(s, "mod");
- wrd1(s, id);
- bopen(s);
- for (@ast.native_item item in nmod.items) {
- hbox(s);
- alt (item.node) {
- case (ast.native_item_ty(?id,_)) {
+ maybe_print_comment(s, item.span.lo);
+ hbox(s);
+ alt (item.node) {
+ case (ast.item_const(?id, ?ty, ?expr, _, _)) {
+ wrd1(s, "const");
+ print_type(s, ty);
+ space(s.s);
+ wrd1(s, id);
+ wrd1(s, "=");
+ print_expr(s, expr);
+ wrd(s.s, ";");
+ }
+ case (ast.item_fn(?name,?_fn,?typarams,_,_)) {
+ print_fn(s, _fn.decl, name, typarams);
+ space(s.s);
+ print_block(s, _fn.body);
+ }
+ case (ast.item_mod(?id,?_mod,_)) {
+ wrd1(s, "mod");
+ wrd1(s, id);
+ bopen(s);
+ for (@ast.item itm in _mod.items) {print_item(s, itm);}
+ bclose_c(s, item.span);
+ }
+ case (ast.item_native_mod(?id,?nmod,_)) {
+ wrd1(s, "native");
+ alt (nmod.abi) {
+ case (ast.native_abi_rust) {wrd1(s, "\"rust\"");}
+ case (ast.native_abi_cdecl) {wrd1(s, "\"cdecl\"");}
+ }
+ wrd1(s, "mod");
+ wrd1(s, id);
+ bopen(s);
+ for (@ast.native_item item in nmod.items) {
+ hbox(s);
+ maybe_print_comment(s, item.span.lo);
+ alt (item.node) {
+ case (ast.native_item_ty(?id,_)) {
+ wrd1(s, "type");
+ wrd(s.s, id);
+ }
+ case (ast.native_item_fn(?id,?lname,?decl,
+ ?typarams,_,_)) {
+ print_fn(s, decl, id, typarams);
+ alt (lname) {
+ case (option.none[str]) {}
+ case (option.some[str](?ss)) {
+ print_string(s, ss);
+ }
+ }
+ }
+ }
+ wrd(s.s, ";");
+ end(s.s);
+ }
+ bclose_c(s, item.span);
+ }
+ case (ast.item_ty(?id,?ty,?params,_,_)) {
wrd1(s, "type");
- wrd(s, id);
- }
- case (ast.native_item_fn(?id,?lname,?decl,?typarams,_,_)) {
- print_fn(s, decl, id, typarams);
- alt (lname) {
- case (option.none[str]) {}
- case (option.some[str](?ss)) {print_string(s,ss);}
+ wrd(s.s, id);
+ print_type_params(s, params);
+ space(s.s);
+ wrd1(s, "=");
+ print_type(s, ty);
+ wrd(s.s, ";");
+ }
+ case (ast.item_tag(?id,?variants,?params,_)) {
+ wrd1(s, "tag");
+ wrd(s.s, id);
+ print_type_params(s, params);
+ space(s.s);
+ bopen(s);
+ for (ast.variant v in variants) {
+ wrd(s.s, v.name);
+ if (_vec.len[ast.variant_arg](v.args) > 0u) {
+ popen(s);
+ impure fn print_variant_arg(ps s, &ast.variant_arg arg) {
+ print_type(s, arg.ty);
+ }
+ auto f = print_variant_arg;
+ commasep[ast.variant_arg](s, v.args, f);
+ pclose(s);
+ }
+ wrd(s.s, ";");
+ line(s.s);
}
- }
+ bclose_c(s, item.span);
}
- wrd(s, ";");
- end(s);
- }
- bclose(s);
- }
- case (ast.item_ty(?id,?ty,?params,_,_)) {
- wrd1(s, "type");
- wrd(s, id);
- print_type_params(s, params);
- space(s);
- wrd1(s, "=");
- print_type(s, ty);
- wrd(s, ";");
- }
- case (ast.item_tag(?id,?variants,?params,_)) {
- wrd1(s, "tag");
- wrd(s, id);
- print_type_params(s, params);
- space(s);
- bopen(s);
- for (ast.variant v in variants) {
- wrd(s, v.name);
- if (_vec.len[ast.variant_arg](v.args) > 0u) {
- popen(s);
- impure fn print_variant_arg(ps s, &ast.variant_arg arg) {
- print_type(s, arg.ty);
- }
- auto f = print_variant_arg;
- commasep[ast.variant_arg](s, v.args, f);
- pclose(s);
- }
- wrd(s, ";");
- line(s);
- }
- bclose(s);
- }
- case (ast.item_obj(?id,?_obj,?params,_,_)) {
- wrd1(s, "obj");
- wrd(s, id);
- print_type_params(s, params);
- popen(s);
- impure fn print_field(ps s, &ast.obj_field field) {
- hbox(s);
- print_type(s, field.ty);
- space(s);
- wrd(s, field.ident);
- end(s);
- }
- auto f = print_field;
- commasep[ast.obj_field](s, _obj.fields, f);
- pclose(s);
- space(s);
- bopen(s);
- for (@ast.method meth in _obj.methods) {
- hbox(s);
- let vec[ast.ty_param] typarams = vec();
- print_fn(s, meth.node.meth.decl, meth.node.ident, typarams);
- space(s);
- print_block(s, meth.node.meth.body);
- end(s);
- line(s);
- }
- alt (_obj.dtor) {
- case (option.some[ast.block](?dtor)) {
- hbox(s);
- wrd1(s, "close");
- print_block(s, dtor);
- end(s);
- line(s);
+ case (ast.item_obj(?id,?_obj,?params,_,_)) {
+ wrd1(s, "obj");
+ wrd(s.s, id);
+ print_type_params(s, params);
+ popen(s);
+ impure fn print_field(ps s, &ast.obj_field field) {
+ hbox(s);
+ print_type(s, field.ty);
+ space(s.s);
+ wrd(s.s, field.ident);
+ end(s.s);
+ }
+ auto f = print_field;
+ commasep[ast.obj_field](s, _obj.fields, f);
+ pclose(s);
+ space(s.s);
+ bopen(s);
+ for (@ast.method meth in _obj.methods) {
+ hbox(s);
+ let vec[ast.ty_param] typarams = vec();
+ maybe_print_comment(s, meth.span.lo);
+ print_fn(s, meth.node.meth.decl, meth.node.ident, typarams);
+ space(s.s);
+ print_block(s, meth.node.meth.body);
+ end(s.s);
+ line(s.s);
+ }
+ alt (_obj.dtor) {
+ case (option.some[ast.block](?dtor)) {
+ hbox(s);
+ wrd1(s, "close");
+ print_block(s, dtor);
+ end(s.s);
+ line(s.s);
+ }
+ case (_) {}
+ }
+ bclose_c(s, item.span);
}
- case (_) {}
- }
- bclose(s);
}
- }
- end(s);
- line(s);
- line(s);
+ end(s.s);
+ line(s.s);
+ line(s.s);
}
impure fn print_block(ps s, ast.block blk) {
- bopen(s);
- for (@ast.stmt st in blk.node.stmts) {
- alt (st.node) {
- case (ast.stmt_decl(?decl)) {print_decl(s, decl);}
- case (ast.stmt_expr(?expr)) {print_expr(s, expr);}
+ auto cur_line = 0u;
+ maybe_print_comment(s, blk.span.lo);
+ bopen(s);
+ for (@ast.stmt st in blk.node.stmts) {
+ if (cur_line != 0u && blk.span.lo.line > cur_line + 1u) {
+ line(s.s);
+ }
+ cur_line = blk.span.hi.line;
+ maybe_print_comment(s, st.span.lo);
+ alt (st.node) {
+ case (ast.stmt_decl(?decl)) {print_decl(s, decl);}
+ case (ast.stmt_expr(?expr)) {print_expr(s, expr);}
+ }
+ if (front.parser.stmt_ends_with_semi(st)) {wrd(s.s, ";");}
+ if (!maybe_print_stmt_comment(s, st.span)) {line(s.s);}
}
- if (front.parser.stmt_ends_with_semi(st)) {wrd(s, ";");}
- line(s);
- }
- alt (blk.node.expr) {
- case (option.some[@ast.expr](?expr)) {
- print_expr(s, expr);
- line(s);
+ alt (blk.node.expr) {
+ case (option.some[@ast.expr](?expr)) {
+ if (cur_line != 0u && blk.span.lo.line > cur_line + 1u) {
+ line(s.s);
+ }
+ print_expr(s, expr);
+ if (!maybe_print_stmt_comment(s, expr.span)) {line(s.s);}
+ }
+ case (_) {}
}
- case (_) {}
- }
- bclose(s);
+ bclose_c(s, blk.span);
}
impure fn print_literal(ps s, @ast.lit lit) {
- alt (lit.node) {
- case (ast.lit_str(?st)) {print_string(s, st);}
- case (ast.lit_char(?ch)) {
- wrd(s, "'" + escape_str(_str.from_bytes(vec(ch as u8)), '\'') + "'");
- }
- case (ast.lit_int(?val)) {
- wrd(s, util.common.istr(val));
- }
- case (ast.lit_uint(?val)) { // TODO clipping? uistr?
- wrd(s, util.common.istr(val as int) + "u");
- }
- case (ast.lit_float(?fstr)) {
- wrd(s, fstr);
- }
- case (ast.lit_mach_int(?mach,?val)) {
- wrd(s, util.common.istr(val as int));
- wrd(s, util.common.ty_mach_to_str(mach));
- }
- case (ast.lit_nil) {wrd(s, "()");}
- case (ast.lit_bool(?val)) {
- if (val) {wrd(s, "true");} else {wrd(s, "false");}
+ maybe_print_comment(s, lit.span.lo);
+ alt (lit.node) {
+ case (ast.lit_str(?st)) {print_string(s, st);}
+ case (ast.lit_char(?ch)) {
+ wrd(s.s, "'" + escape_str(_str.from_bytes(vec(ch as u8)), '\'')
+ + "'");
+ }
+ case (ast.lit_int(?val)) {
+ wrd(s.s, util.common.istr(val));
+ }
+ case (ast.lit_uint(?val)) { // FIXME clipping? uistr?
+ wrd(s.s, util.common.istr(val as int) + "u");
+ }
+ case (ast.lit_float(?fstr)) {
+ wrd(s.s, fstr);
+ }
+ case (ast.lit_mach_int(?mach,?val)) {
+ wrd(s.s, util.common.istr(val as int));
+ wrd(s.s, util.common.ty_mach_to_str(mach));
+ }
+ case (ast.lit_nil) {wrd(s.s, "()");}
+ case (ast.lit_bool(?val)) {
+ if (val) {wrd(s.s, "true");} else {wrd(s.s, "false");}
+ }
}
- }
}
impure fn print_expr(ps s, &@ast.expr expr) {
- auto pe = print_expr;
- hbox(s);
- alt (expr.node) {
- case (ast.expr_vec(?exprs,?mut,_)) {
- if (mut == ast.mut) {
- wrd1(s, "mutable");
- }
- wrd(s, "vec");
- popen(s);
- commasep[@ast.expr](s, exprs, pe);
- pclose(s);
- }
- case (ast.expr_tup(?exprs,_)) {
- impure fn printElt(ps s, &ast.elt elt) {
- hbox(s);
- if (elt.mut == ast.mut) {wrd1(s, "mutable");}
- print_expr(s, elt.expr);
- end(s);
- }
- wrd(s, "tup");
- popen(s);
- auto f = printElt;
- commasep[ast.elt](s, exprs, f);
- pclose(s);
- }
- case (ast.expr_rec(?fields,?wth,_)) {
- impure fn print_field(ps s, &ast.field field) {
- hbox(s);
- if (field.mut == ast.mut) {wrd1(s, "mutable");}
- wrd(s, field.ident);
- wrd(s, "=");
- print_expr(s, field.expr);
- end(s);
- }
- wrd(s, "rec");
- popen(s);
- auto f = print_field;
- commasep[ast.field](s, fields, f);
- alt (wth) {
- case (option.some[@ast.expr](?expr)) {
- if (_vec.len[ast.field](fields) > 0u) {space(s);}
- hbox(s);
- wrd1(s, "with");
- print_expr(s, expr);
- end(s);
+ maybe_print_comment(s, expr.span.lo);
+ auto pe = print_expr;
+ hbox(s);
+ alt (expr.node) {
+ case (ast.expr_vec(?exprs,?mut,_)) {
+ if (mut == ast.mut) {
+ wrd1(s, "mutable");
+ }
+ wrd(s.s, "vec");
+ popen(s);
+ commasep[@ast.expr](s, exprs, pe);
+ pclose(s);
}
- case (_) {}
- }
- pclose(s);
- }
- case (ast.expr_call(?func,?args,_)) {
- print_expr(s, func);
- popen(s);
- commasep[@ast.expr](s, args, pe);
- pclose(s);
- }
- case (ast.expr_bind(?func,?args,_)) {
- impure fn print_opt(ps s, &option.t[@ast.expr] expr) {
- alt (expr) {
- case (option.some[@ast.expr](?expr)) {
+ case (ast.expr_tup(?exprs,_)) {
+ impure fn printElt(ps s, &ast.elt elt) {
+ hbox(s);
+ if (elt.mut == ast.mut) {wrd1(s, "mutable");}
+ print_expr(s, elt.expr);
+ end(s.s);
+ }
+ wrd(s.s, "tup");
+ popen(s);
+ auto f = printElt;
+ commasep[ast.elt](s, exprs, f);
+ pclose(s);
+ }
+ case (ast.expr_rec(?fields,?wth,_)) {
+ impure fn print_field(ps s, &ast.field field) {
+ hbox(s);
+ if (field.mut == ast.mut) {wrd1(s, "mutable");}
+ wrd(s.s, field.ident);
+ wrd(s.s, "=");
+ print_expr(s, field.expr);
+ end(s.s);
+ }
+ wrd(s.s, "rec");
+ popen(s);
+ auto f = print_field;
+ commasep[ast.field](s, fields, f);
+ alt (wth) {
+ case (option.some[@ast.expr](?expr)) {
+ if (_vec.len[ast.field](fields) > 0u) {space(s.s);}
+ hbox(s);
+ wrd1(s, "with");
+ print_expr(s, expr);
+ end(s.s);
+ }
+ case (_) {}
+ }
+ pclose(s);
+ }
+ case (ast.expr_call(?func,?args,_)) {
+ print_expr(s, func);
+ popen(s);
+ commasep[@ast.expr](s, args, pe);
+ pclose(s);
+ }
+ case (ast.expr_bind(?func,?args,_)) {
+ impure fn print_opt(ps s, &option.t[@ast.expr] expr) {
+ alt (expr) {
+ case (option.some[@ast.expr](?expr)) {
+ print_expr(s, expr);
+ }
+ case (_) {wrd(s.s, "_");}
+ }
+ }
+ wrd1(s, "bind");
+ print_expr(s, func);
+ popen(s);
+ auto f = print_opt;
+ commasep[option.t[@ast.expr]](s, args, f);
+ pclose(s);
+ }
+ case (ast.expr_binary(?op,?lhs,?rhs,_)) {
+ auto prec = operator_prec(op);
+ print_maybe_parens(s, lhs, prec);
+ space(s.s);
+ wrd1(s, ast.binop_to_str(op));
+ print_maybe_parens(s, rhs, prec + 1);
+ }
+ case (ast.expr_unary(?op,?expr,_)) {
+ wrd(s.s, ast.unop_to_str(op));
print_expr(s, expr);
- }
- case (_) {wrd(s, "_");}
- }
- }
- wrd1(s, "bind");
- print_expr(s, func);
- popen(s);
- auto f = print_opt;
- commasep[option.t[@ast.expr]](s, args, f);
- pclose(s);
- }
- case (ast.expr_binary(?op,?lhs,?rhs,_)) {
- auto prec = operator_prec(op);
- print_maybe_parens(s, lhs, prec);
- space(s);
- wrd1(s, ast.binop_to_str(op));
- print_maybe_parens(s, rhs, prec + 1);
- }
- case (ast.expr_unary(?op,?expr,_)) {
- wrd(s, ast.unop_to_str(op));
- print_expr(s, expr);
- }
- case (ast.expr_lit(?lit,_)) {
- print_literal(s, lit);
- }
- case (ast.expr_cast(?expr,?ty,_)) {
- print_maybe_parens(s, expr, as_prec);
- space(s);
- wrd1(s, "as");
- print_type(s, ty);
- }
- case (ast.expr_if(?test,?block,?elseopt,_)) {
- wrd1(s, "if");
- popen(s);
- print_expr(s, test);
- pclose(s);
- space(s);
- print_block(s, block);
- alt (elseopt) {
- case (option.some[@ast.expr](?_else)) {
- space(s);
- wrd1(s, "else");
- print_expr(s, _else);
- }
- case (_) { /* fall through */ }
- }
- }
- case (ast.expr_while(?test,?block,_)) {
- wrd1(s, "while");
- popen(s);
- print_expr(s, test);
- pclose(s);
- space(s);
- print_block(s, block);
- }
- case (ast.expr_for(?decl,?expr,?block,_)) {
- wrd1(s, "for");
- popen(s);
- print_decl(s, decl);
- space(s);
- wrd1(s, "in");
- print_expr(s, expr);
- pclose(s);
- space(s);
- print_block(s, block);
- }
- case (ast.expr_for_each(?decl,?expr,?block,_)) {
- wrd1(s, "for each");
- popen(s);
- print_decl(s, decl);
- space(s);
- wrd1(s, "in");
- print_expr(s, expr);
- space(s);
- print_block(s, block);
- }
- case (ast.expr_do_while(?block,?expr,_)) {
- wrd1(s, "do");
- space(s);
- print_block(s, block);
- space(s);
- wrd1(s, "while");
- popen(s);
- print_expr(s, expr);
- pclose(s);
- }
- case (ast.expr_alt(?expr,?arms,_)) {
- wrd1(s, "alt");
- popen(s);
- print_expr(s, expr);
- pclose(s);
- space(s);
- bopen(s);
- for (ast.arm arm in arms) {
- hbox(s);
- wrd1(s, "case");
- popen(s);
- print_pat(s, arm.pat);
- pclose(s);
- space(s);
- print_block(s, arm.block);
- end(s);
- line(s);
- }
- bclose(s);
- }
- case (ast.expr_block(?block,_)) {
- print_block(s, block);
- }
- case (ast.expr_assign(?lhs,?rhs,_)) {
- print_expr(s, lhs);
- space(s);
- wrd1(s, "=");
- print_expr(s, rhs);
- }
- case (ast.expr_assign_op(?op,?lhs,?rhs,_)) {
- print_expr(s, lhs);
- space(s);
- wrd(s, ast.binop_to_str(op));
- wrd1(s, "=");
- print_expr(s, rhs);
- }
- case (ast.expr_send(?lhs, ?rhs, _)) {
- print_expr(s, lhs);
- space(s);
- wrd1(s, "<|");
- print_expr(s, rhs);
- }
- case (ast.expr_recv(?lhs, ?rhs, _)) {
- print_expr(s, lhs);
- space(s);
- wrd1(s, "<-");
- print_expr(s, rhs);
- }
- case (ast.expr_field(?expr,?id,_)) {
- print_expr(s, expr);
- wrd(s, ".");
- wrd(s, id);
- }
- case (ast.expr_index(?expr,?index,_)) {
- print_expr(s, expr);
- wrd(s, ".");
- popen(s);
- print_expr(s, index);
- pclose(s);
- }
- case (ast.expr_path(?path,_,_)) {
- print_path(s, path);
- }
- case (ast.expr_fail) {
- wrd(s, "fail");
- }
- case (ast.expr_ret(?result)) {
- wrd(s, "ret");
- alt (result) {
- case (option.some[@ast.expr](?expr)) {
- space(s);
- print_expr(s, expr);
}
- case (_) {}
- }
- }
- case (ast.expr_put(?result)) {
- wrd(s, "put");
- alt (result) {
- case (option.some[@ast.expr](?expr)) {
- space(s);
- print_expr(s, expr);
+ case (ast.expr_lit(?lit,_)) {
+ print_literal(s, lit);
+ }
+ case (ast.expr_cast(?expr,?ty,_)) {
+ print_maybe_parens(s, expr, as_prec);
+ space(s.s);
+ wrd1(s, "as");
+ print_type(s, ty);
+ }
+ case (ast.expr_if(?test,?block,?elseopt,_)) {
+ wrd1(s, "if");
+ popen_h(s);
+ print_expr(s, test);
+ pclose(s);
+ space(s.s);
+ print_block(s, block);
+ alt (elseopt) {
+ case (option.some[@ast.expr](?_else)) {
+ space(s.s);
+ wrd1(s, "else");
+ print_expr(s, _else);
+ }
+ case (_) { /* fall through */ }
+ }
+ }
+ case (ast.expr_while(?test,?block,_)) {
+ wrd1(s, "while");
+ popen_h(s);
+ print_expr(s, test);
+ pclose(s);
+ space(s.s);
+ print_block(s, block);
+ }
+ case (ast.expr_for(?decl,?expr,?block,_)) {
+ wrd1(s, "for");
+ popen_h(s);
+ print_for_decl(s, decl);
+ space(s.s);
+ wrd1(s, "in");
+ print_expr(s, expr);
+ pclose(s);
+ space(s.s);
+ print_block(s, block);
+ }
+ case (ast.expr_for_each(?decl,?expr,?block,_)) {
+ wrd1(s, "for each");
+ popen_h(s);
+ print_for_decl(s, decl);
+ space(s.s);
+ wrd1(s, "in");
+ print_expr(s, expr);
+ pclose(s);
+ space(s.s);
+ print_block(s, block);
+ }
+ case (ast.expr_do_while(?block,?expr,_)) {
+ wrd1(s, "do");
+ space(s.s);
+ print_block(s, block);
+ space(s.s);
+ wrd1(s, "while");
+ popen_h(s);
+ print_expr(s, expr);
+ pclose(s);
+ }
+ case (ast.expr_alt(?expr,?arms,_)) {
+ wrd1(s, "alt");
+ popen_h(s);
+ print_expr(s, expr);
+ pclose(s);
+ space(s.s);
+ bopen(s);
+ for (ast.arm arm in arms) {
+ hbox(s);
+ wrd1(s, "case");
+ popen_h(s);
+ print_pat(s, arm.pat);
+ pclose(s);
+ space(s.s);
+ print_block(s, arm.block);
+ end(s.s);
+ line(s.s);
+ }
+ bclose_c(s, expr.span);
+ }
+ case (ast.expr_block(?block,_)) {
+ print_block(s, block);
+ }
+ case (ast.expr_assign(?lhs,?rhs,_)) {
+ print_expr(s, lhs);
+ space(s.s);
+ wrd1(s, "=");
+ print_expr(s, rhs);
+ }
+ case (ast.expr_assign_op(?op,?lhs,?rhs,_)) {
+ print_expr(s, lhs);
+ space(s.s);
+ wrd(s.s, ast.binop_to_str(op));
+ wrd1(s, "=");
+ print_expr(s, rhs);
+ }
+ case (ast.expr_send(?lhs, ?rhs, _)) {
+ print_expr(s, lhs);
+ space(s.s);
+ wrd1(s, "<|");
+ print_expr(s, rhs);
+ }
+ case (ast.expr_recv(?lhs, ?rhs, _)) {
+ print_expr(s, lhs);
+ space(s.s);
+ wrd1(s, "<-");
+ print_expr(s, rhs);
+ }
+ case (ast.expr_field(?expr,?id,_)) {
+ print_expr(s, expr);
+ wrd(s.s, ".");
+ wrd(s.s, id);
+ }
+ case (ast.expr_index(?expr,?index,_)) {
+ print_expr(s, expr);
+ wrd(s.s, ".");
+ popen_h(s);
+ print_expr(s, index);
+ pclose(s);
+ }
+ case (ast.expr_path(?path,_,_)) {
+ print_path(s, path);
+ }
+ case (ast.expr_fail) {
+ wrd(s.s, "fail");
+ }
+ case (ast.expr_ret(?result)) {
+ wrd(s.s, "ret");
+ alt (result) {
+ case (option.some[@ast.expr](?expr)) {
+ space(s.s);
+ print_expr(s, expr);
+ }
+ case (_) {}
+ }
+ }
+ case (ast.expr_put(?result)) {
+ wrd(s.s, "put");
+ alt (result) {
+ case (option.some[@ast.expr](?expr)) {
+ space(s.s);
+ print_expr(s, expr);
+ }
+ case (_) {}
+ }
+ }
+ case (ast.expr_be(?result)) {
+ wrd1(s, "be");
+ print_expr(s, result);
+ }
+ case (ast.expr_log(?expr)) {
+ wrd1(s, "log");
+ print_expr(s, expr);
+ }
+ case (ast.expr_check_expr(?expr)) {
+ wrd1(s, "check");
+ popen_h(s);
+ print_expr(s, expr);
+ pclose(s);
+ }
+ case (ast.expr_ext(?path, ?args, ?body, _, _)) {
+ wrd(s.s, "#");
+ print_path(s, path);
+ if (_vec.len[@ast.expr](args) > 0u) {
+ popen(s);
+ commasep[@ast.expr](s, args, pe);
+ pclose(s);
+ }
+ // FIXME: extension 'body'
+ }
+ case (ast.expr_port(_)) {
+ wrd(s.s, "port");
+ popen_h(s);
+ pclose(s);
+ }
+ case (ast.expr_chan(?expr, _)) {
+ wrd(s.s, "chan");
+ popen_h(s);
+ print_expr(s, expr);
+ pclose(s);
}
- case (_) {}
- }
- }
- case (ast.expr_be(?result)) {
- wrd1(s, "be");
- print_expr(s, result);
- }
- case (ast.expr_log(?expr)) {
- wrd1(s, "log");
- print_expr(s, expr);
- }
- case (ast.expr_check_expr(?expr)) {
- wrd1(s, "check");
- print_expr(s, expr);
- }
- case (ast.expr_ext(?path, ?args, ?body, _, _)) {
- wrd(s, "#");
- print_path(s, path);
- if (_vec.len[@ast.expr](args) > 0u) {
- popen(s);
- commasep[@ast.expr](s, args, pe);
- pclose(s);
- }
- // TODO: extension 'body'
- }
- case (ast.expr_port(_)) {
- wrd(s, "port");
- popen(s);
- pclose(s);
- }
- case (ast.expr_chan(?expr, _)) {
- wrd(s, "chan");
- popen(s);
- print_expr(s, expr);
- pclose(s);
}
- }
- end(s);
+ end(s.s);
}
impure fn print_decl(ps s, @ast.decl decl) {
- hbox(s);
- alt (decl.node) {
- case (ast.decl_local(?loc)) {
- alt (loc.ty) {
- case (option.some[@ast.ty](?ty)) {
- wrd1(s, "let");
- print_type(s, ty);
- space(s);
- }
- case (_) {
- wrd1(s, "auto");
- }
- }
- wrd(s, loc.ident);
- alt (loc.init) {
- case (option.some[ast.initializer](?init)) {
- space(s);
- alt (init.op) {
- case (ast.init_assign) {
- wrd1(s, "=");
+ maybe_print_comment(s, decl.span.lo);
+ hbox(s);
+ alt (decl.node) {
+ case (ast.decl_local(?loc)) {
+ alt (loc.ty) {
+ case (option.some[@ast.ty](?ty)) {
+ wrd1(s, "let");
+ print_type(s, ty);
+ space(s.s);
+ }
+ case (_) {
+ wrd1(s, "auto");
+ }
}
- case (ast.init_recv) {
- wrd1(s, "<-");
+ wrd(s.s, loc.ident);
+ alt (loc.init) {
+ case (option.some[ast.initializer](?init)) {
+ space(s.s);
+ alt (init.op) {
+ case (ast.init_assign) {
+ wrd1(s, "=");
+ }
+ case (ast.init_recv) {
+ wrd1(s, "<-");
+ }
+ }
+ print_expr(s, init.expr);
+ }
+ case (_) {}
}
- }
- print_expr(s, init.expr);
}
- case (_) {}
- }
+ case (ast.decl_item(?item)) {
+ print_item(s, item);
+ }
}
- case (ast.decl_item(?item)) {
- print_item(s, item);
+ end(s.s);
+}
+
+impure fn print_for_decl(ps s, @ast.decl decl) {
+ alt (decl.node) {
+ case (ast.decl_local(?loc)) {
+ print_type(s, option.get[@ast.ty](loc.ty));
+ space(s.s);
+ wrd(s.s, loc.ident);
+ }
}
- }
- end(s);
}
impure fn print_path(ps s, ast.path path) {
- auto first = true;
- for (str id in path.node.idents) {
- if (first) {first = false;}
- else {wrd(s, ".");}
- wrd(s, id);
- }
- if (_vec.len[@ast.ty](path.node.types) > 0u) {
- wrd(s, "[");
- auto f = print_type;
- commasep[@ast.ty](s, path.node.types, f);
- wrd(s, "]");
- }
+ maybe_print_comment(s, path.span.lo);
+ auto first = true;
+ for (str id in path.node.idents) {
+ if (first) {first = false;}
+ else {wrd(s.s, ".");}
+ wrd(s.s, id);
+ }
+ if (_vec.len[@ast.ty](path.node.types) > 0u) {
+ wrd(s.s, "[");
+ auto f = print_type;
+ commasep[@ast.ty](s, path.node.types, f);
+ wrd(s.s, "]");
+ }
}
impure fn print_pat(ps s, &@ast.pat pat) {
- alt (pat.node) {
- case (ast.pat_wild(_)) {wrd(s, "_");}
- case (ast.pat_bind(?id,_,_)) {wrd(s, "?" + id);}
- case (ast.pat_lit(?lit,_)) {print_literal(s, lit);}
- case (ast.pat_tag(?path,?args,_,_)) {
- print_path(s, path);
- if (_vec.len[@ast.pat](args) > 0u) {
- popen(s);
- auto f = print_pat;
- commasep[@ast.pat](s, args, f);
- pclose(s);
- }
+ maybe_print_comment(s, pat.span.lo);
+ alt (pat.node) {
+ case (ast.pat_wild(_)) {wrd(s.s, "_");}
+ case (ast.pat_bind(?id,_,_)) {wrd(s.s, "?" + id);}
+ case (ast.pat_lit(?lit,_)) {print_literal(s, lit);}
+ case (ast.pat_tag(?path,?args,_,_)) {
+ print_path(s, path);
+ if (_vec.len[@ast.pat](args) > 0u) {
+ popen_h(s);
+ auto f = print_pat;
+ commasep[@ast.pat](s, args, f);
+ pclose(s);
+ }
+ }
}
- }
}
impure fn print_fn(ps s, ast.fn_decl decl, str name,
vec[ast.ty_param] typarams) {
- alt (decl.effect) {
- case (ast.eff_impure) {wrd1(s, "impure");}
- case (ast.eff_unsafe) {wrd1(s, "unsafe");}
- case (_) {}
- }
- wrd1(s, "fn");
- wrd(s, name);
- print_type_params(s, typarams);
- popen(s);
- impure fn print_arg(ps s, &ast.arg x) {
- hbox(s);
- print_type(s, x.ty);
- space(s);
- wrd(s, x.ident);
- end(s);
- }
- auto f = print_arg;
- commasep[ast.arg](s, decl.inputs, f);
- pclose(s);
- if (decl.output.node != ast.ty_nil) {
- space(s);
- hbox(s);
- wrd1(s, "->");
- print_type(s, decl.output);
- end(s);
- }
+ alt (decl.effect) {
+ case (ast.eff_impure) {wrd1(s, "impure");}
+ case (ast.eff_unsafe) {wrd1(s, "unsafe");}
+ case (_) {}
+ }
+ wrd1(s, "fn");
+ wrd(s.s, name);
+ print_type_params(s, typarams);
+ popen(s);
+ impure fn print_arg(ps s, &ast.arg x) {
+ hbox(s);
+ if (x.mode == ast.alias) {wrd(s.s, "&");}
+ print_type(s, x.ty);
+ space(s.s);
+ wrd(s.s, x.ident);
+ end(s.s);
+ }
+ auto f = print_arg;
+ commasep[ast.arg](s, decl.inputs, f);
+ pclose(s);
+ maybe_print_comment(s, decl.output.span.lo);
+ if (decl.output.node != ast.ty_nil) {
+ space(s.s);
+ hbox(s);
+ wrd1(s, "->");
+ print_type(s, decl.output);
+ end(s.s);
+ }
}
impure fn print_type_params(ps s, vec[ast.ty_param] params) {
- if (_vec.len[ast.ty_param](params) > 0u) {
- wrd(s, "[");
- impure fn printParam(ps s, &ast.ty_param param) {wrd(s, param.ident);}
- auto f = printParam;
- commasep[ast.ty_param](s, params, f);
- wrd(s, "]");
- }
+ if (_vec.len[ast.ty_param](params) > 0u) {
+ wrd(s.s, "[");
+ impure fn printParam(ps s, &ast.ty_param param) {
+ wrd(s.s, param.ident);
+ }
+ auto f = printParam;
+ commasep[ast.ty_param](s, params, f);
+ wrd(s.s, "]");
+ }
}
impure fn print_view_item(ps s, @ast.view_item item) {
- hbox(s);
- alt (item.node) {
- case (ast.view_item_use(?id,?mta,_,_)) {
- wrd1(s, "use");
- wrd(s, id);
- if (_vec.len[@ast.meta_item](mta) > 0u) {
- popen(s);
- impure fn print_meta(ps s, &@ast.meta_item item) {
- hbox(s);
- wrd1(s, item.node.name);
- wrd1(s, "=");
- print_string(s, item.node.value);
- end(s);
- }
- auto f = print_meta;
- commasep[@ast.meta_item](s, mta, f);
- pclose(s);
- }
- }
- case (ast.view_item_import(?id,?ids,_,_)) {
- wrd1(s, "import");
- if (!_str.eq(id, ids.(_vec.len[str](ids)-1u))) {
- wrd1(s, id);
- wrd1(s, "=");
- }
- auto first = true;
- for (str elt in ids) {
- if (first) {first = false;}
- else {wrd(s, ".");}
- wrd(s, elt);
- }
- }
- case (ast.view_item_export(?id)) {
- wrd1(s, "export");
- wrd(s, id);
+ maybe_print_comment(s, item.span.lo);
+ hbox(s);
+ alt (item.node) {
+ case (ast.view_item_use(?id,?mta,_,_)) {
+ wrd1(s, "use");
+ wrd(s.s, id);
+ if (_vec.len[@ast.meta_item](mta) > 0u) {
+ popen(s);
+ impure fn print_meta(ps s, &@ast.meta_item item) {
+ hbox(s);
+ wrd1(s, item.node.name);
+ wrd1(s, "=");
+ print_string(s, item.node.value);
+ end(s.s);
+ }
+ auto f = print_meta;
+ commasep[@ast.meta_item](s, mta, f);
+ pclose(s);
+ }
+ }
+ case (ast.view_item_import(?id,?ids,_,_)) {
+ wrd1(s, "import");
+ if (!_str.eq(id, ids.(_vec.len[str](ids)-1u))) {
+ wrd1(s, id);
+ wrd1(s, "=");
+ }
+ auto first = true;
+ for (str elt in ids) {
+ if (first) {first = false;}
+ else {wrd(s.s, ".");}
+ wrd(s.s, elt);
+ }
+ }
+ case (ast.view_item_export(?id)) {
+ wrd1(s, "export");
+ wrd(s.s, id);
+ }
}
- }
- end(s);
- wrd(s, ";");
- line(s);
+ end(s.s);
+ wrd(s.s, ";");
+ line(s.s);
}
// FIXME: The fact that this builds up the table anew for every call is
// not good. Eventually, table should be a const.
fn operator_prec(ast.binop op) -> int {
- for (front.parser.op_spec spec in front.parser.prec_table()) {
- if (spec.op == op) {ret spec.prec;}
- }
- fail;
+ for (front.parser.op_spec spec in front.parser.prec_table()) {
+ if (spec.op == op) {ret spec.prec;}
+ }
+ fail;
}
impure fn print_maybe_parens(ps s, @ast.expr expr, int outer_prec) {
- auto add_them;
- alt (expr.node) {
- case (ast.expr_binary(?op,_,_,_)) {
- add_them = operator_prec(op) < outer_prec;
- }
- case (ast.expr_cast(_,_,_)) {
- add_them = as_prec < outer_prec;
- }
- case (_) {
- add_them = false;
+ auto add_them;
+ alt (expr.node) {
+ case (ast.expr_binary(?op,_,_,_)) {
+ add_them = operator_prec(op) < outer_prec;
+ }
+ case (ast.expr_cast(_,_,_)) {
+ add_them = as_prec < outer_prec;
+ }
+ case (_) {
+ add_them = false;
+ }
}
- }
- if (add_them) {popen(s);}
- print_expr(s, expr);
- if (add_them) {pclose(s);}
+ if (add_them) {popen(s);}
+ print_expr(s, expr);
+ if (add_them) {pclose(s);}
}
fn escape_str(str st, char to_escape) -> str {
- let str out = "";
- auto len = _str.byte_len(st);
- auto i = 0u;
- while (i < len) {
- alt (st.(i) as char) {
- case ('\n') {out += "\\n";}
- case ('\t') {out += "\\t";}
- case ('\r') {out += "\\r";}
- case ('\\') {out += "\\\\";}
- case (?cur) {
- if (cur == to_escape) {out += "\\";}
- _str.push_byte(out, cur as u8);
- }
+ let str out = "";
+ auto len = _str.byte_len(st);
+ auto i = 0u;
+ while (i < len) {
+ alt (st.(i) as char) {
+ case ('\n') {out += "\\n";}
+ case ('\t') {out += "\\t";}
+ case ('\r') {out += "\\r";}
+ case ('\\') {out += "\\\\";}
+ case (?cur) {
+ if (cur == to_escape) {out += "\\";}
+ _str.push_byte(out, cur as u8);
+ }
+ }
+ i += 1u;
}
- i += 1u;
- }
- ret out;
+ ret out;
+}
+
+impure fn print_mt(ps s, &ast.mt mt) {
+ alt (mt.mut) {
+ case (ast.mut) { wrd1(s, "mutable"); }
+ case (ast.maybe_mut) { wrd1(s, "mutable?"); }
+ case (ast.imm) { /* nothing */ }
+ }
+ print_type(s, mt.ty);
}
impure fn print_string(ps s, str st) {
- wrd(s, "\""); wrd(s, escape_str(st, '"')); wrd(s, "\"");
+ wrd(s.s, "\""); wrd(s.s, escape_str(st, '"')); wrd(s.s, "\"");
}
impure fn print_ty_fn(ps s, ast.proto proto, option.t[str] id,
vec[ast.ty_arg] inputs, @ast.ty output) {
- if (proto == ast.proto_fn) {wrd(s, "fn");}
- else {wrd(s, "iter");}
- alt (id) {
- case (option.some[str](?id)) {space(s); wrd(s, id);}
- case (_) {}
- }
- popen(s);
- impure fn print_arg(ps s, &ast.ty_arg input) {
- if (middle.ty.mode_is_alias(input.mode)) {wrd(s, "&");}
- print_type(s, input.ty);
- }
- auto f = print_arg;
- commasep[ast.ty_arg](s, inputs, f);
- pclose(s);
- if (output.node != ast.ty_nil) {
- space(s);
- hbox(s);
- wrd1(s, "->");
- print_type(s, output);
- end(s);
- }
+ if (proto == ast.proto_fn) {wrd(s.s, "fn");}
+ else {wrd(s.s, "iter");}
+ alt (id) {
+ case (option.some[str](?id)) {space(s.s); wrd(s.s, id);}
+ case (_) {}
+ }
+ popen_h(s);
+ impure fn print_arg(ps s, &ast.ty_arg input) {
+ if (middle.ty.mode_is_alias(input.mode)) {wrd(s.s, "&");}
+ print_type(s, input.ty);
+ }
+ auto f = print_arg;
+ commasep[ast.ty_arg](s, inputs, f);
+ pclose(s);
+ maybe_print_comment(s, output.span.lo);
+ if (output.node != ast.ty_nil) {
+ space(s.s);
+ hbox(s);
+ wrd1(s, "->");
+ print_type(s, output);
+ end(s.s);
+ }
+}
+
+fn next_comment(ps s) -> option.t[lexer.cmnt] {
+ alt (s.comments) {
+ case (option.some[vec[lexer.cmnt]](?cmnts)) {
+ if (s.cur_cmnt < _vec.len[lexer.cmnt](cmnts)) {
+ ret option.some[lexer.cmnt](cmnts.(s.cur_cmnt));
+ } else {ret option.none[lexer.cmnt];}
+ }
+ case (_) {ret option.none[lexer.cmnt];}
+ }
+}
+
+impure fn maybe_print_comment(ps s, util.common.pos pos) {
+ while (true) {
+ alt (next_comment(s)) {
+ case (option.some[lexer.cmnt](?cmnt)) {
+ if (cmnt.pos.line < pos.line ||
+ (cmnt.pos.line == pos.line && cmnt.pos.col < pos.col)) {
+ print_comment(s, cmnt.val);
+ if (cmnt.space_after) {line(s.s);}
+ s.cur_cmnt += 1u;
+ } else { break; }
+ }
+ case (_) {break;}
+ }
+ }
+}
+
+impure fn maybe_print_stmt_comment(ps s, util.common.span span) -> bool {
+ alt (next_comment(s)) {
+ case (option.some[lexer.cmnt](?cmnt)) {
+ if (span.hi.line == cmnt.pos.line) {
+ wrd(s.s, " ");
+ print_comment(s, cmnt.val);
+ s.cur_cmnt += 1u;
+ ret true;
+ }
+ }
+ case (_) {}
+ }
+ ret false;
+}
+
+impure fn print_remaining_comments(ps s) {
+ while (true) {
+ alt (next_comment(s)) {
+ case (option.some[lexer.cmnt](?cmnt)) {
+ print_comment(s, cmnt.val);
+ if (cmnt.space_after) {line(s.s);}
+ s.cur_cmnt += 1u;
+ }
+ case (_) {break;}
+ }
+ }
+}
+
+impure fn print_comment(ps s, lexer.cmnt_ cmnt) {
+ alt (cmnt) {
+ case (lexer.cmnt_line(?val)) {
+ wrd(s.s, "// " + val);
+ pp.hardbreak(s.s);
+ }
+ case (lexer.cmnt_block(?lines)) {
+ pp.abox(s.s);
+ wrd(s.s, "/* ");
+ pp.abox(s.s);
+ auto first = true;
+ for (str ln in lines) {
+ if (first) {first = false;}
+ else {pp.hardbreak(s.s);}
+ wrd(s.s, ln);
+ }
+ end(s.s);
+ wrd(s.s, "*/");
+ end(s.s);
+ line(s.s);
+ }
+ }
}