From c144752a2de4ffe3a2a22da9a8309ca2ecd85c58 Mon Sep 17 00:00:00 2001 From: Steven Fackler Date: Sun, 24 Nov 2013 23:08:53 -0800 Subject: Support multiple item macros Closes #4375 --- src/libsyntax/ext/base.rs | 3 +- src/libsyntax/ext/expand.rs | 114 ++++++++++++++++++------------------ src/libsyntax/ext/tt/macro_rules.rs | 13 +++- src/libsyntax/fold.rs | 53 +++++++---------- src/libsyntax/util/small_vector.rs | 28 +++++++++ 5 files changed, 118 insertions(+), 93 deletions(-) (limited to 'src/libsyntax') diff --git a/src/libsyntax/ext/base.rs b/src/libsyntax/ext/base.rs index ccb88e0c9c0..deaa821cd45 100644 --- a/src/libsyntax/ext/base.rs +++ b/src/libsyntax/ext/base.rs @@ -18,6 +18,7 @@ use ext::expand; use parse; use parse::token; use parse::token::{ident_to_str, intern, str_to_ident}; +use util::small_vector::SmallVector; use std::hashmap::HashMap; @@ -131,7 +132,7 @@ pub type SyntaxExpanderTTItemFunNoCtxt = pub trait AnyMacro { fn make_expr(&self) -> @ast::Expr; - fn make_item(&self) -> Option<@ast::item>; + fn make_items(&self) -> SmallVector<@ast::item>; fn make_stmt(&self) -> @ast::Stmt; } diff --git a/src/libsyntax/ext/expand.rs b/src/libsyntax/ext/expand.rs index f23e13b8931..3719362a7df 100644 --- a/src/libsyntax/ext/expand.rs +++ b/src/libsyntax/ext/expand.rs @@ -28,6 +28,7 @@ use parse::token; use parse::token::{fresh_mark, fresh_name, ident_to_str, intern}; use visit; use visit::Visitor; +use util::small_vector::SmallVector; use std::vec; @@ -310,7 +311,7 @@ pub fn expand_item(extsbox: @mut SyntaxEnv, cx: @ExtCtxt, it: @ast::item, fld: &MacroExpander) - -> Option<@ast::item> { + -> SmallVector<@ast::item> { match it.node { ast::item_mac(*) => expand_item_mac(extsbox, cx, it, fld), ast::item_mod(_) | ast::item_foreign_mod(_) => { @@ -337,7 +338,7 @@ pub fn expand_item_mac(extsbox: @mut SyntaxEnv, cx: @ExtCtxt, it: @ast::item, fld: &MacroExpander) - -> Option<@ast::item> { + -> SmallVector<@ast::item> { let (pth, tts, ctxt) = match it.node { item_mac(codemap::Spanned { node: mac_invoc_tt(ref pth, ref tts, ctxt), @@ -396,28 +397,30 @@ pub fn expand_item_mac(extsbox: @mut SyntaxEnv, it.span, format!("{}! is not legal in item position", extnamestr)) }; - let maybe_it = match expanded { + let items = match expanded { MRItem(it) => { - mark_item(it,fm) - .and_then(|i| fld.fold_item(i)) + mark_item(it,fm).move_iter() + .flat_map(|i| fld.fold_item(i).move_iter()) + .collect() } MRExpr(_) => { cx.span_fatal(pth.span, format!("expr macro in item position: {}", extnamestr)) } MRAny(any_macro) => { - any_macro.make_item() - .and_then(|i| mark_item(i,fm)) - .and_then(|i| fld.fold_item(i)) + any_macro.make_items().move_iter() + .flat_map(|i| mark_item(i, fm).move_iter()) + .flat_map(|i| fld.fold_item(i).move_iter()) + .collect() } MRDef(ref mdef) => { // yikes... no idea how to apply the mark to this. I'm afraid // we're going to have to wait-and-see on this one. insert_macro(*extsbox,intern(mdef.name), @SE((*mdef).ext)); - None + SmallVector::zero() } }; cx.bt_pop(); - return maybe_it; + return items; } @@ -442,7 +445,7 @@ pub fn expand_stmt(extsbox: @mut SyntaxEnv, cx: @ExtCtxt, s: &Stmt, fld: &MacroExpander) - -> Option<@Stmt> { + -> SmallVector<@Stmt> { // why the copying here and not in expand_expr? // looks like classic changed-in-only-one-place let (pth, tts, semi, ctxt) = match s.node { @@ -461,7 +464,7 @@ pub fn expand_stmt(extsbox: @mut SyntaxEnv, } let extname = &pth.segments[0].identifier; let extnamestr = ident_to_str(extname); - let fully_expanded: @ast::Stmt = match (*extsbox).find(&extname.name) { + let fully_expanded: SmallVector<@Stmt> = match (*extsbox).find(&extname.name) { None => { cx.span_fatal(pth.span, format!("macro undefined: '{}'", extnamestr)) } @@ -501,22 +504,15 @@ pub fn expand_stmt(extsbox: @mut SyntaxEnv, let marked_after = mark_stmt(expanded,fm); // Keep going, outside-in. - let fully_expanded = match fld.fold_stmt(marked_after) { - Some(stmt) => { - let fully_expanded = &stmt.node; - cx.bt_pop(); - @Spanned { - span: stmt.span, - node: (*fully_expanded).clone(), - } - } - None => { - cx.span_fatal(pth.span, - "macro didn't expand to a statement") - } - }; - - fully_expanded + let fully_expanded = fld.fold_stmt(marked_after); + if fully_expanded.is_empty() { + cx.span_fatal(pth.span, + "macro didn't expand to a statement"); + } + cx.bt_pop(); + fully_expanded.move_iter() + .map(|s| @Spanned { span: s.span, node: s.node.clone() }) + .collect() } _ => { @@ -525,21 +521,23 @@ pub fn expand_stmt(extsbox: @mut SyntaxEnv, } }; - match fully_expanded.node { - StmtExpr(e, stmt_id) if semi => { - Some(@Spanned { - span: fully_expanded.span, - node: StmtSemi(e, stmt_id), - }) + fully_expanded.move_iter().map(|s| { + match s.node { + StmtExpr(e, stmt_id) if semi => { + @Spanned { + span: s.span, + node: StmtSemi(e, stmt_id) + } + } + _ => s /* might already have a semi */ } - _ => Some(fully_expanded), /* might already have a semi */ - } + }).collect() } // expand a non-macro stmt. this is essentially the fallthrough for // expand_stmt, above. fn expand_non_macro_stmt(exts: SyntaxEnv, s: &Stmt, fld: &MacroExpander) - -> Option<@Stmt> { + -> SmallVector<@Stmt> { // is it a let? match s.node { StmtDecl(@Spanned { @@ -590,7 +588,7 @@ fn expand_non_macro_stmt(exts: SyntaxEnv, s: &Stmt, fld: &MacroExpander) id: id, span: span, }; - Some(@Spanned { + SmallVector::one(@Spanned { node: StmtDecl(@Spanned { node: DeclLocal(rewritten_local), span: stmt_span @@ -679,13 +677,9 @@ pub fn expand_block_elts(exts: SyntaxEnv, b: &Block, fld: &MacroExpander) let pending_renames = block_info.pending_renames; let rename_fld = renames_to_fold(pending_renames); let new_view_items = b.view_items.map(|x| fld.fold_view_item(x)); - let mut new_stmts = ~[]; - for x in b.stmts.iter() { - match fld.fold_stmt(mustbesome(rename_fld.fold_stmt(*x))) { - Some(s) => new_stmts.push(s), - None => () - } - } + let new_stmts = b.stmts.iter() + .flat_map(|x| fld.fold_stmt(mustbeone(rename_fld.fold_stmt(*x))).move_iter()) + .collect(); let new_expr = b.expr.map(|x| fld.fold_expr(rename_fld.fold_expr(x))); Block{ view_items: new_view_items, @@ -697,13 +691,12 @@ pub fn expand_block_elts(exts: SyntaxEnv, b: &Block, fld: &MacroExpander) } } -// rename_fold should never return "None". -// (basically, just .get() with a better message...) -fn mustbesome(val : Option) -> T { - match val { - Some(v) => v, - None => fail!("rename_fold returned None") +// rename_fold should never return anything other than one thing +fn mustbeone(mut val : SmallVector) -> T { + if val.len() != 1 { + fail!("rename_fold didn't return one value"); } + val.pop() } // get the (innermost) BlockInfo from an exts stack @@ -741,10 +734,11 @@ pub fn renames_to_fold(renames: @mut ~[(ast::Ident,ast::Name)]) -> @ast_fold { // perform a bunch of renames fn apply_pending_renames(folder : @ast_fold, stmt : ast::Stmt) -> @ast::Stmt { - match folder.fold_stmt(&stmt) { - Some(s) => s, - None => fail!("renaming of stmt produced None") + let mut stmts = folder.fold_stmt(&stmt); + if stmts.len() != 1 { + fail!("renaming of stmt did not produce one stmt"); } + stmts.pop() } @@ -1025,14 +1019,14 @@ impl ast_fold for MacroExpander { self) } - fn fold_item(&self, item: @ast::item) -> Option<@ast::item> { + fn fold_item(&self, item: @ast::item) -> SmallVector<@ast::item> { expand_item(self.extsbox, self.cx, item, self) } - fn fold_stmt(&self, stmt: &ast::Stmt) -> Option<@ast::Stmt> { + fn fold_stmt(&self, stmt: &ast::Stmt) -> SmallVector<@ast::Stmt> { expand_stmt(self.extsbox, self.cx, stmt, @@ -1191,11 +1185,15 @@ fn mark_expr(expr : @ast::Expr, m : Mrk) -> @ast::Expr { // apply a given mark to the given stmt. Used following the expansion of a macro. fn mark_stmt(expr : &ast::Stmt, m : Mrk) -> @ast::Stmt { - new_mark_folder(m).fold_stmt(expr).unwrap() + let mut stmts = new_mark_folder(m).fold_stmt(expr); + if stmts.len() != 1 { + fail!("marking a stmt didn't return a stmt"); + } + stmts.pop() } // apply a given mark to the given item. Used following the expansion of a macro. -fn mark_item(expr : @ast::item, m : Mrk) -> Option<@ast::item> { +fn mark_item(expr : @ast::item, m : Mrk) -> SmallVector<@ast::item> { new_mark_folder(m).fold_item(expr) } diff --git a/src/libsyntax/ext/tt/macro_rules.rs b/src/libsyntax/ext/tt/macro_rules.rs index 343116df642..3cc00ef8199 100644 --- a/src/libsyntax/ext/tt/macro_rules.rs +++ b/src/libsyntax/ext/tt/macro_rules.rs @@ -24,6 +24,7 @@ use parse::attr::parser_attr; use parse::token::{get_ident_interner, special_idents, gensym_ident, ident_to_str}; use parse::token::{FAT_ARROW, SEMI, nt_matchers, nt_tt, EOF}; use print; +use util::small_vector::SmallVector; struct ParserAnyMacro { parser: @Parser, @@ -54,9 +55,15 @@ impl AnyMacro for ParserAnyMacro { self.ensure_complete_parse(true); ret } - fn make_item(&self) -> Option<@ast::item> { - let attrs = self.parser.parse_outer_attributes(); - let ret = self.parser.parse_item(attrs); + fn make_items(&self) -> SmallVector<@ast::item> { + let mut ret = SmallVector::zero(); + loop { + let attrs = self.parser.parse_outer_attributes(); + match self.parser.parse_item(attrs) { + Some(item) => ret.push(item), + None => break + } + } self.ensure_complete_parse(false); ret } diff --git a/src/libsyntax/fold.rs b/src/libsyntax/fold.rs index b268988c3c5..4c0653a3c04 100644 --- a/src/libsyntax/fold.rs +++ b/src/libsyntax/fold.rs @@ -13,6 +13,7 @@ use ast; use codemap::{respan, Span, Spanned}; use parse::token; use opt_vec::OptVec; +use util::small_vector::SmallVector; // We may eventually want to be able to fold over type parameters, too. pub trait ast_fold { @@ -113,7 +114,7 @@ pub trait ast_fold { } } - fn fold_item(&self, i: @item) -> Option<@item> { + fn fold_item(&self, i: @item) -> SmallVector<@item> { noop_fold_item(i, self) } @@ -159,7 +160,7 @@ pub trait ast_fold { noop_fold_block(b, self) } - fn fold_stmt(&self, s: &Stmt) -> Option<@Stmt> { + fn fold_stmt(&self, s: &Stmt) -> SmallVector<@Stmt> { noop_fold_stmt(s, self) } @@ -216,23 +217,20 @@ pub trait ast_fold { } } - fn fold_decl(&self, d: @Decl) -> Option<@Decl> { + fn fold_decl(&self, d: @Decl) -> SmallVector<@Decl> { let node = match d.node { - DeclLocal(ref l) => Some(DeclLocal(self.fold_local(*l))), + DeclLocal(ref l) => SmallVector::one(DeclLocal(self.fold_local(*l))), DeclItem(it) => { - match self.fold_item(it) { - Some(it_folded) => Some(DeclItem(it_folded)), - None => None, - } + self.fold_item(it).move_iter().map(|i| DeclItem(i)).collect() } }; - node.map(|node| { + node.move_iter().map(|node| { @Spanned { node: node, span: d.span, } - }) + }).collect() } fn fold_expr(&self, e: @Expr) -> @Expr { @@ -618,13 +616,7 @@ fn fold_variant_arg_(va: &variant_arg, folder: &T) pub fn noop_fold_block(b: &Block, folder: &T) -> Block { let view_items = b.view_items.map(|x| folder.fold_view_item(x)); - let mut stmts = ~[]; - for stmt in b.stmts.iter() { - match folder.fold_stmt(*stmt) { - None => {} - Some(stmt) => stmts.push(stmt) - } - } + let stmts = b.stmts.iter().flat_map(|s| folder.fold_stmt(*s).move_iter()).collect(); ast::Block { view_items: view_items, stmts: stmts, @@ -711,7 +703,7 @@ pub fn noop_fold_mod(m: &_mod, folder: &T) -> _mod { view_items: m.view_items .iter() .map(|x| folder.fold_view_item(x)).collect(), - items: m.items.iter().filter_map(|x| folder.fold_item(*x)).collect(), + items: m.items.iter().flat_map(|x| folder.fold_item(*x).move_iter()).collect(), } } @@ -728,10 +720,10 @@ pub fn noop_fold_crate(c: Crate, folder: &T) -> Crate { } pub fn noop_fold_item(i: @ast::item, folder: &T) - -> Option<@ast::item> { + -> SmallVector<@ast::item> { let fold_attribute = |x| fold_attribute_(x, folder); - Some(@ast::item { + SmallVector::one(@ast::item { ident: folder.fold_ident(i.ident), attrs: i.attrs.map(|e| fold_attribute(*e)), id: folder.new_id(i.id), @@ -867,27 +859,26 @@ pub fn noop_fold_expr(e: @ast::Expr, folder: &T) -> @ast::Expr { } } -pub fn noop_fold_stmt(s: &Stmt, folder: &T) -> Option<@Stmt> { - let node = match s.node { +pub fn noop_fold_stmt(s: &Stmt, folder: &T) -> SmallVector<@Stmt> { + let nodes = match s.node { StmtDecl(d, nid) => { - match folder.fold_decl(d) { - Some(d) => Some(StmtDecl(d, folder.new_id(nid))), - None => None, - } + folder.fold_decl(d).move_iter() + .map(|d| StmtDecl(d, folder.new_id(nid))) + .collect() } StmtExpr(e, nid) => { - Some(StmtExpr(folder.fold_expr(e), folder.new_id(nid))) + SmallVector::one(StmtExpr(folder.fold_expr(e), folder.new_id(nid))) } StmtSemi(e, nid) => { - Some(StmtSemi(folder.fold_expr(e), folder.new_id(nid))) + SmallVector::one(StmtSemi(folder.fold_expr(e), folder.new_id(nid))) } - StmtMac(ref mac, semi) => Some(StmtMac(folder.fold_mac(mac), semi)) + StmtMac(ref mac, semi) => SmallVector::one(StmtMac(folder.fold_mac(mac), semi)) }; - node.map(|node| @Spanned { + nodes.move_iter().map(|node| @Spanned { node: node, span: folder.new_span(s.span), - }) + }).collect() } #[cfg(test)] diff --git a/src/libsyntax/util/small_vector.rs b/src/libsyntax/util/small_vector.rs index c6b223f7c15..0196a0212e2 100644 --- a/src/libsyntax/util/small_vector.rs +++ b/src/libsyntax/util/small_vector.rs @@ -73,6 +73,22 @@ impl SmallVector { } } + pub fn pop(&mut self) -> T { + match *self { + Zero => fail!("attempted to pop from an empty SmallVector"), + One(*) => { + let mut tmp = Zero; + util::swap(self, &mut tmp); + match tmp { + One(v) => v, + _ => unreachable!() + } + } + // Should this reduce to a One if possible? + Many(ref mut vs) => vs.pop() + } + } + pub fn get<'a>(&'a self, idx: uint) -> &'a T { match *self { One(ref v) if idx == 0 => v, @@ -176,6 +192,18 @@ mod test { assert_eq!(&3, v.get(2)); } + #[test] + fn test_pop() { + let mut v = SmallVector::one(1); + assert_eq!(1, v.pop()); + assert_eq!(0, v.len()); + + let mut v= SmallVector::many(~[1, 2]); + assert_eq!(2, v.pop()); + assert_eq!(1, v.pop()); + assert_eq!(0, v.len()); + } + #[test] fn test_from_iterator() { let v: SmallVector = (~[1, 2, 3]).move_iter().collect(); -- cgit 1.4.1-3-g733a5