rust/src/libsyntax/visit.rs

654 lines
23 KiB
Rust
Raw Normal View History

import ast::*;
import codemap::span;
2011-06-08 15:48:19 -05:00
2011-06-10 10:29:34 -05:00
// Context-passing AST walker. Each overridden visit method has full control
// over what happens with its node, it can do its own traversal of the node's
// children (potentially passing in different contexts to each), call
// visit::visit_* to apply the default traversal algorithm (again, it can
// override the context), or prevent deeper traversal by doing nothing.
2011-07-26 09:35:31 -05:00
// Our typesystem doesn't do circular types, so the visitor record can not
2012-01-19 16:24:03 -06:00
// hold functions that take visitors. A vt enum is used to break the cycle.
2012-01-19 19:56:05 -06:00
enum vt<E> { mk_vt(visitor<E>), }
2011-06-08 15:48:19 -05:00
2012-01-19 16:24:03 -06:00
enum fn_kind {
fk_item_fn(ident, ~[ty_param]), //< an item declared with fn()
fk_method(ident, ~[ty_param], @method),
fk_anon(proto, capture_clause), //< an anonymous function like fn@(...)
fk_fn_block(capture_clause), //< a block {||...}
fk_ctor(ident, ~[attribute], ~[ty_param], node_id /* self id */,
def_id /* parent class id */), // class constructor
fk_dtor(~[ty_param], ~[attribute], node_id /* self id */,
def_id /* parent class id */) // class destructor
}
fn name_of_fn(fk: fn_kind) -> ident {
2012-08-06 14:34:08 -05:00
match fk {
fk_item_fn(name, _) | fk_method(name, _, _)
2012-08-03 21:59:04 -05:00
| fk_ctor(name, _, _, _, _) => /* FIXME (#2543) */ copy name,
fk_anon(*) | fk_fn_block(*) => @~"anon",
fk_dtor(*) => @~"drop"
}
}
fn tps_of_fn(fk: fn_kind) -> ~[ty_param] {
2012-08-06 14:34:08 -05:00
match fk {
fk_item_fn(_, tps) | fk_method(_, tps, _)
2012-08-03 21:59:04 -05:00
| fk_ctor(_, _, tps, _, _) | fk_dtor(tps, _, _, _) => {
/* FIXME (#2543) */ copy tps
2012-06-07 23:53:47 -05:00
}
2012-08-03 21:59:04 -05:00
fk_anon(*) | fk_fn_block(*) => ~[]
}
}
type visitor<E> =
@{visit_mod: fn@(_mod, span, node_id, E, vt<E>),
visit_view_item: fn@(@view_item, E, vt<E>),
visit_foreign_item: fn@(@foreign_item, E, vt<E>),
visit_item: fn@(@item, E, vt<E>),
visit_local: fn@(@local, E, vt<E>),
visit_block: fn@(ast::blk, E, vt<E>),
visit_stmt: fn@(@stmt, E, vt<E>),
visit_arm: fn@(arm, E, vt<E>),
visit_pat: fn@(@pat, E, vt<E>),
visit_decl: fn@(@decl, E, vt<E>),
visit_expr: fn@(@expr, E, vt<E>),
visit_expr_post: fn@(@expr, E, vt<E>),
visit_ty: fn@(@ty, E, vt<E>),
visit_ty_params: fn@(~[ty_param], E, vt<E>),
visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id, E, vt<E>),
visit_ty_method: fn@(ty_method, E, vt<E>),
visit_trait_method: fn@(trait_method, E, vt<E>),
2012-08-07 19:46:07 -05:00
visit_struct_def: fn@(@struct_def, ident, ~[ty_param], node_id, E,
vt<E>),
visit_class_item: fn@(@class_member, E, vt<E>)};
2011-06-08 15:48:19 -05:00
fn default_visitor<E>() -> visitor<E> {
2012-08-01 19:30:05 -05:00
return @{visit_mod: |a,b,c,d,e|visit_mod::<E>(a, b, c, d, e),
2012-06-30 18:19:07 -05:00
visit_view_item: |a,b,c|visit_view_item::<E>(a, b, c),
visit_foreign_item: |a,b,c|visit_foreign_item::<E>(a, b, c),
visit_item: |a,b,c|visit_item::<E>(a, b, c),
visit_local: |a,b,c|visit_local::<E>(a, b, c),
visit_block: |a,b,c|visit_block::<E>(a, b, c),
visit_stmt: |a,b,c|visit_stmt::<E>(a, b, c),
visit_arm: |a,b,c|visit_arm::<E>(a, b, c),
visit_pat: |a,b,c|visit_pat::<E>(a, b, c),
visit_decl: |a,b,c|visit_decl::<E>(a, b, c),
visit_expr: |a,b,c|visit_expr::<E>(a, b, c),
visit_expr_post: |_a,_b,_c| (),
2012-06-30 18:19:07 -05:00
visit_ty: |a,b,c|skip_ty::<E>(a, b, c),
visit_ty_params: |a,b,c|visit_ty_params::<E>(a, b, c),
visit_fn: |a,b,c,d,e,f,g|visit_fn::<E>(a, b, c, d, e, f, g),
visit_ty_method: |a,b,c|visit_ty_method::<E>(a, b, c),
visit_trait_method: |a,b,c|visit_trait_method::<E>(a, b, c),
2012-08-07 19:46:07 -05:00
visit_struct_def: |a,b,c,d,e,f|visit_struct_def::<E>(a, b, c,
d, e, f),
2012-06-30 18:19:07 -05:00
visit_class_item: |a,b,c|visit_class_item::<E>(a, b, c)};
2011-06-08 15:48:19 -05:00
}
fn visit_crate<E>(c: crate, e: E, v: vt<E>) {
v.visit_mod(c.node.module, c.span, crate_node_id, e, v);
2011-06-08 15:48:19 -05:00
}
fn visit_crate_directive<E>(cd: @crate_directive, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match cd.node {
2012-08-03 21:59:04 -05:00
cdir_src_mod(_, _) => (),
cdir_dir_mod(_, cdirs, _) => for cdirs.each |cdir| {
visit_crate_directive(cdir, e, v);
},
2012-08-03 21:59:04 -05:00
cdir_view_item(vi) => v.visit_view_item(vi, e, v),
cdir_syntax(_) => ()
2011-06-09 08:50:20 -05:00
}
}
fn visit_mod<E>(m: _mod, _sp: span, _id: node_id, e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for m.view_items.each |vi| { v.visit_view_item(vi, e, v); }
for m.items.each |i| { v.visit_item(i, e, v); }
2011-06-08 15:48:19 -05:00
}
fn visit_view_item<E>(_vi: @view_item, _e: E, _v: vt<E>) { }
2011-06-09 08:50:20 -05:00
fn visit_local<E>(loc: @local, e: E, v: vt<E>) {
v.visit_pat(loc.node.pat, e, v);
v.visit_ty(loc.node.ty, e, v);
2012-08-06 14:34:08 -05:00
match loc.node.init {
2012-08-03 21:59:04 -05:00
none => (),
some(i) => v.visit_expr(i.expr, e, v)
}
}
fn visit_item<E>(i: @item, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match i.node {
2012-08-03 21:59:04 -05:00
item_const(t, ex) => { v.visit_ty(t, e, v); v.visit_expr(ex, e, v); }
item_fn(decl, tp, body) => {
v.visit_fn(fk_item_fn(/* FIXME (#2543) */ copy i.ident,
/* FIXME (#2543) */ copy tp), decl, body,
2012-06-07 23:53:47 -05:00
i.span, i.id, e, v);
}
2012-08-03 21:59:04 -05:00
item_mod(m) => v.visit_mod(m, i.span, i.id, e, v),
item_foreign_mod(nm) => {
2012-06-30 18:19:07 -05:00
for nm.view_items.each |vi| { v.visit_view_item(vi, e, v); }
for nm.items.each |ni| { v.visit_foreign_item(ni, e, v); }
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
item_ty(t, tps) => {
v.visit_ty(t, e, v);
v.visit_ty_params(tps, e, v);
}
2012-08-03 21:59:04 -05:00
item_enum(variants, tps) => {
v.visit_ty_params(tps, e, v);
visit_variants(variants, tps, e, v);
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
item_impl(tps, traits, ty, methods) => {
v.visit_ty_params(tps, e, v);
for traits.each |p| {
visit_path(p.path, e, v);
}
v.visit_ty(ty, e, v);
2012-06-30 18:19:07 -05:00
for methods.each |m| {
visit_method_helper(m, e, v)
}
}
item_class(struct_def, tps) => {
v.visit_ty_params(tps, e, v);
v.visit_struct_def(struct_def, i.ident, tps, i.id, e, v);
}
2012-08-03 21:59:04 -05:00
item_trait(tps, traits, methods) => {
v.visit_ty_params(tps, e, v);
for traits.each |p| { visit_path(p.path, e, v); }
2012-06-30 18:19:07 -05:00
for methods.each |m| {
v.visit_trait_method(m, e, v);
}
}
2012-08-03 21:59:04 -05:00
item_mac(m) => visit_mac(m, e, v)
2011-06-08 15:48:19 -05:00
}
}
fn visit_variants<E>(variants: ~[ast::variant], tps: ~[ast::ty_param], e: E,
v: vt<E>) {
for variants.each |vr| {
match vr.node.kind {
tuple_variant_kind(variant_args) => {
for variant_args.each |va| { v.visit_ty(va.ty, e, v); }
}
struct_variant_kind(struct_def) => {
v.visit_struct_def(struct_def, vr.node.name, tps,
vr.node.id, e, v);
}
enum_variant_kind(variants) => {
visit_variants(variants, tps, e, v);
}
}
}
}
fn visit_class_item<E>(cm: @class_member, e:E, v:vt<E>) {
2012-08-06 14:34:08 -05:00
match cm.node {
2012-08-03 21:59:04 -05:00
instance_var(_, t, _, _, _) => v.visit_ty(t, e, v),
class_method(m) => visit_method_helper(m, e, v)
}
}
fn skip_ty<E>(_t: @ty, _e: E, _v: vt<E>) {}
fn visit_ty<E>(t: @ty, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match t.node {
2012-04-09 19:32:49 -05:00
ty_box(mt) | ty_uniq(mt) |
2012-08-03 21:59:04 -05:00
ty_vec(mt) | ty_ptr(mt) | ty_rptr(_, mt) => {
2012-04-09 19:32:49 -05:00
v.visit_ty(mt.ty, e, v);
}
2012-08-03 21:59:04 -05:00
ty_rec(flds) => for flds.each |f| {
v.visit_ty(f.node.mt.ty, e, v);
},
2012-08-03 21:59:04 -05:00
ty_tup(ts) => for ts.each |tt| {
v.visit_ty(tt, e, v);
},
ty_fn(_, bounds, decl) => {
2012-06-30 18:19:07 -05:00
for decl.inputs.each |a| { v.visit_ty(a.ty, e, v); }
visit_ty_param_bounds(bounds, e, v);
v.visit_ty(decl.output, e, v);
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
ty_path(p, _) => visit_path(p, e, v),
ty_fixed_length(t, _) => v.visit_ty(t, e, v),
ty_nil |
ty_bot |
ty_mac(_) |
2012-08-03 21:59:04 -05:00
ty_infer => ()
2011-06-08 15:48:19 -05:00
}
}
fn visit_path<E>(p: @path, e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for p.types.each |tp| { v.visit_ty(tp, e, v); }
}
fn visit_pat<E>(p: @pat, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match p.node {
2012-08-03 21:59:04 -05:00
pat_enum(path, children) => {
visit_path(path, e, v);
2012-06-30 18:19:07 -05:00
do option::iter(children) |children| {
for children.each |child| { v.visit_pat(child, e, v); }}
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
pat_rec(fields, _) => for fields.each |f| {
v.visit_pat(f.pat, e, v)
},
pat_struct(path, fields, _) => {
visit_path(path, e, v);
for fields.each |f| {
v.visit_pat(f.pat, e, v);
}
}
2012-08-03 21:59:04 -05:00
pat_tup(elts) => for elts.each |elt| {
v.visit_pat(elt, e, v)
},
2012-08-03 21:59:04 -05:00
pat_box(inner) | pat_uniq(inner) => v.visit_pat(inner, e, v),
pat_ident(_, path, inner) => {
visit_path(path, e, v);
2012-06-30 18:19:07 -05:00
do option::iter(inner) |subpat| { v.visit_pat(subpat, e, v)};
}
2012-08-03 21:59:04 -05:00
pat_lit(ex) => v.visit_expr(ex, e, v),
pat_range(e1, e2) => { v.visit_expr(e1, e, v); v.visit_expr(e2, e, v); }
pat_wild => ()
2011-06-08 15:48:19 -05:00
}
}
fn visit_foreign_item<E>(ni: @foreign_item, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match ni.node {
2012-08-03 21:59:04 -05:00
foreign_item_fn(fd, tps) => {
v.visit_ty_params(tps, e, v);
visit_fn_decl(fd, e, v);
}
2011-06-08 15:48:19 -05:00
}
}
fn visit_ty_param_bounds<E>(bounds: @~[ty_param_bound], e: E, v: vt<E>) {
for vec::each(*bounds) |bound| {
match bound {
bound_trait(t) => v.visit_ty(t, e, v),
bound_copy | bound_send | bound_const | bound_owned => ()
}
}
}
fn visit_ty_params<E>(tps: ~[ty_param], e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for tps.each |tp| {
visit_ty_param_bounds(tp.bounds, e, v);
}
}
fn visit_fn_decl<E>(fd: fn_decl, e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for fd.inputs.each |a| { v.visit_ty(a.ty, e, v); }
v.visit_ty(fd.output, e, v);
2011-06-08 15:48:19 -05:00
}
// Note: there is no visit_method() method in the visitor, instead override
// visit_fn() and check for fk_method(). I named this visit_method_helper()
// because it is not a default impl of any method, though I doubt that really
// clarifies anything. - Niko
fn visit_method_helper<E>(m: @method, e: E, v: vt<E>) {
v.visit_fn(fk_method(/* FIXME (#2543) */ copy m.ident,
/* FIXME (#2543) */ copy m.tps, m),
2012-06-07 23:53:47 -05:00
m.decl, m.body, m.span, m.id, e, v);
}
// Similar logic to the comment on visit_method_helper - Tim
fn visit_class_ctor_helper<E>(ctor: class_ctor, nm: ident, tps: ~[ty_param],
parent_id: def_id, e: E, v: vt<E>) {
v.visit_fn(fk_ctor(/* FIXME (#2543) */ copy nm,
ctor.node.attrs,
/* FIXME (#2543) */ copy tps,
ctor.node.self_id, parent_id),
ctor.node.dec, ctor.node.body, ctor.span, ctor.node.id, e, v)
}
fn visit_class_dtor_helper<E>(dtor: class_dtor, tps: ~[ty_param],
parent_id: def_id, e: E, v: vt<E>) {
v.visit_fn(fk_dtor(/* FIXME (#2543) */ copy tps, dtor.node.attrs,
dtor.node.self_id, parent_id), ast_util::dtor_dec(),
dtor.node.body, dtor.span, dtor.node.id, e, v)
}
fn visit_fn<E>(fk: fn_kind, decl: fn_decl, body: blk, _sp: span,
_id: node_id, e: E, v: vt<E>) {
visit_fn_decl(decl, e, v);
v.visit_ty_params(tps_of_fn(fk), e, v);
v.visit_block(body, e, v);
2011-06-08 15:48:19 -05:00
}
fn visit_ty_method<E>(m: ty_method, e: E, v: vt<E>) {
for m.decl.inputs.each |a| { v.visit_ty(a.ty, e, v); }
v.visit_ty_params(m.tps, e, v);
v.visit_ty(m.decl.output, e, v);
}
fn visit_trait_method<E>(m: trait_method, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match m {
2012-08-03 21:59:04 -05:00
required(ty_m) => v.visit_ty_method(ty_m, e, v),
provided(m) => visit_method_helper(m, e, v)
}
}
2012-08-07 19:46:07 -05:00
fn visit_struct_def<E>(sd: @struct_def, nm: ast::ident, tps: ~[ty_param],
id: node_id, e: E, v: vt<E>) {
for sd.members.each |m| {
v.visit_class_item(m, e, v);
}
for sd.traits.each |p| { visit_path(p.path, e, v); }
do option::iter(sd.ctor) |ctor| {
visit_class_ctor_helper(ctor, nm, tps, ast_util::local_def(id), e, v);
};
do option::iter(sd.dtor) |dtor| {
visit_class_dtor_helper(dtor, tps, ast_util::local_def(id), e, v)
};
}
fn visit_block<E>(b: ast::blk, e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for b.node.view_items.each |vi| { v.visit_view_item(vi, e, v); }
for b.node.stmts.each |s| { v.visit_stmt(s, e, v); }
2011-06-08 15:48:19 -05:00
visit_expr_opt(b.node.expr, e, v);
}
fn visit_stmt<E>(s: @stmt, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match s.node {
2012-08-03 21:59:04 -05:00
stmt_decl(d, _) => v.visit_decl(d, e, v),
stmt_expr(ex, _) => v.visit_expr(ex, e, v),
stmt_semi(ex, _) => v.visit_expr(ex, e, v)
2011-06-08 15:48:19 -05:00
}
}
fn visit_decl<E>(d: @decl, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match d.node {
2012-08-03 21:59:04 -05:00
decl_local(locs) => for locs.each |loc| {
v.visit_local(loc, e, v)
},
2012-08-03 21:59:04 -05:00
decl_item(it) => v.visit_item(it, e, v)
2011-06-08 15:48:19 -05:00
}
}
fn visit_expr_opt<E>(eo: option<@expr>, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match eo { none => (), some(ex) => v.visit_expr(ex, e, v) }
2011-06-08 15:48:19 -05:00
}
fn visit_exprs<E>(exprs: ~[@expr], e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for exprs.each |ex| { v.visit_expr(ex, e, v); }
2011-06-08 15:48:19 -05:00
}
fn visit_mac<E>(m: mac, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match m.node {
2012-08-03 21:59:04 -05:00
ast::mac_invoc(pth, arg, body) => {
2012-06-30 18:19:07 -05:00
option::map(arg, |arg| v.visit_expr(arg, e, v)); }
2012-08-03 21:59:04 -05:00
ast::mac_invoc_tt(pth, tt) => { /* no user-serviceable parts inside */ }
ast::mac_ellipsis => (),
ast::mac_aq(_, e) => { /* FIXME: maybe visit (Issue #2340) */ }
ast::mac_var(_) => ()
}
}
fn visit_expr<E>(ex: @expr, e: E, v: vt<E>) {
2012-08-06 14:34:08 -05:00
match ex.node {
2012-08-03 21:59:04 -05:00
expr_vstore(x, _) => v.visit_expr(x, e, v),
expr_vec(es, _) => visit_exprs(es, e, v),
expr_repeat(element, count, _) => {
v.visit_expr(element, e, v);
v.visit_expr(count, e, v);
}
2012-08-03 21:59:04 -05:00
expr_rec(flds, base) => {
2012-06-30 18:19:07 -05:00
for flds.each |f| { v.visit_expr(f.node.expr, e, v); }
2011-07-27 07:19:39 -05:00
visit_expr_opt(base, e, v);
}
expr_struct(p, flds, base) => {
visit_path(p, e, v);
for flds.each |f| { v.visit_expr(f.node.expr, e, v); }
visit_expr_opt(base, e, v);
}
expr_tup(elts) => for elts.each |el| { v.visit_expr(el, e, v); },
2012-08-03 21:59:04 -05:00
expr_call(callee, args, _) => {
2011-07-27 07:19:39 -05:00
visit_exprs(args, e, v);
v.visit_expr(callee, e, v);
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
expr_binary(_, a, b) => {
v.visit_expr(a, e, v); v.visit_expr(b, e, v);
}
2012-06-18 19:42:09 -05:00
expr_addr_of(_, x) | expr_unary(_, x) |
expr_loop_body(x) | expr_do_body(x) |
2012-08-03 21:59:04 -05:00
expr_assert(x) => v.visit_expr(x, e, v),
expr_lit(_) => (),
expr_cast(x, t) => { v.visit_expr(x, e, v); v.visit_ty(t, e, v); }
expr_if(x, b, eo) => {
2011-07-27 07:19:39 -05:00
v.visit_expr(x, e, v);
v.visit_block(b, e, v);
visit_expr_opt(eo, e, v);
}
2012-08-03 21:59:04 -05:00
expr_while(x, b) => { v.visit_expr(x, e, v); v.visit_block(b, e, v); }
expr_loop(b) => v.visit_block(b, e, v),
2012-08-07 15:35:51 -05:00
expr_match(x, arms, _) => {
2011-07-27 07:19:39 -05:00
v.visit_expr(x, e, v);
2012-06-30 18:19:07 -05:00
for arms.each |a| { v.visit_arm(a, e, v); }
2011-07-27 07:19:39 -05:00
}
2012-08-03 21:59:04 -05:00
expr_fn(proto, decl, body, cap_clause) => {
v.visit_fn(fk_anon(proto, cap_clause), decl, body,
ex.span, ex.id, e, v);
}
2012-08-03 21:59:04 -05:00
expr_fn_block(decl, body, cap_clause) => {
v.visit_fn(fk_fn_block(cap_clause), decl, body,
ex.span, ex.id, e, v);
}
2012-08-03 21:59:04 -05:00
expr_block(b) => v.visit_block(b, e, v),
expr_assign(a, b) => { v.visit_expr(b, e, v); v.visit_expr(a, e, v); }
expr_copy(a) => v.visit_expr(a, e, v),
expr_unary_move(a) => v.visit_expr(a, e, v),
expr_move(a, b) => { v.visit_expr(b, e, v); v.visit_expr(a, e, v); }
expr_swap(a, b) => { v.visit_expr(a, e, v); v.visit_expr(b, e, v); }
expr_assign_op(_, a, b) => {
2011-07-27 07:19:39 -05:00
v.visit_expr(b, e, v);
v.visit_expr(a, e, v);
}
2012-08-03 21:59:04 -05:00
expr_field(x, _, tys) => {
v.visit_expr(x, e, v);
2012-06-30 18:19:07 -05:00
for tys.each |tp| { v.visit_ty(tp, e, v); }
}
2012-08-03 21:59:04 -05:00
expr_index(a, b) => { v.visit_expr(a, e, v); v.visit_expr(b, e, v); }
expr_path(p) => visit_path(p, e, v),
expr_fail(eo) => visit_expr_opt(eo, e, v),
expr_break => (),
expr_again => (),
expr_ret(eo) => visit_expr_opt(eo, e, v),
expr_log(_, lv, x) => {
v.visit_expr(lv, e, v);
v.visit_expr(x, e, v);
}
2012-08-03 21:59:04 -05:00
expr_mac(mac) => visit_mac(mac, e, v),
2011-06-08 15:48:19 -05:00
}
v.visit_expr_post(ex, e, v);
2011-06-08 15:48:19 -05:00
}
fn visit_arm<E>(a: arm, e: E, v: vt<E>) {
2012-06-30 18:19:07 -05:00
for a.pats.each |p| { v.visit_pat(p, e, v); }
visit_expr_opt(a.guard, e, v);
v.visit_block(a.body, e, v);
2011-06-08 15:48:19 -05:00
}
2011-07-26 09:35:31 -05:00
// Simpler, non-context passing interface. Always walks the whole tree, simply
// calls the given functions on the nodes.
type simple_visitor =
@{visit_mod: fn@(_mod, span, node_id),
visit_view_item: fn@(@view_item),
visit_foreign_item: fn@(@foreign_item),
visit_item: fn@(@item),
visit_local: fn@(@local),
visit_block: fn@(ast::blk),
visit_stmt: fn@(@stmt),
visit_arm: fn@(arm),
visit_pat: fn@(@pat),
visit_decl: fn@(@decl),
visit_expr: fn@(@expr),
visit_expr_post: fn@(@expr),
visit_ty: fn@(@ty),
visit_ty_params: fn@(~[ty_param]),
visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id),
visit_ty_method: fn@(ty_method),
visit_trait_method: fn@(trait_method),
2012-08-07 19:46:07 -05:00
visit_struct_def: fn@(@struct_def, ident, ~[ty_param], node_id),
visit_class_item: fn@(@class_member)};
2011-07-26 09:35:31 -05:00
fn simple_ignore_ty(_t: @ty) {}
2011-07-26 09:35:31 -05:00
fn default_simple_visitor() -> simple_visitor {
2012-08-01 19:30:05 -05:00
return @{visit_mod: fn@(_m: _mod, _sp: span, _id: node_id) { },
visit_view_item: fn@(_vi: @view_item) { },
visit_foreign_item: fn@(_ni: @foreign_item) { },
visit_item: fn@(_i: @item) { },
visit_local: fn@(_l: @local) { },
visit_block: fn@(_b: ast::blk) { },
visit_stmt: fn@(_s: @stmt) { },
visit_arm: fn@(_a: arm) { },
visit_pat: fn@(_p: @pat) { },
visit_decl: fn@(_d: @decl) { },
visit_expr: fn@(_e: @expr) { },
visit_expr_post: fn@(_e: @expr) { },
visit_ty: simple_ignore_ty,
visit_ty_params: fn@(_ps: ~[ty_param]) {},
visit_fn: fn@(_fk: fn_kind, _d: fn_decl, _b: blk, _sp: span,
_id: node_id) { },
visit_ty_method: fn@(_m: ty_method) { },
visit_trait_method: fn@(_m: trait_method) { },
2012-08-07 19:46:07 -05:00
visit_struct_def: fn@(_sd: @struct_def, _nm: ident,
_tps: ~[ty_param], _id: node_id) { },
visit_class_item: fn@(_c: @class_member) {}
};
2011-07-26 09:35:31 -05:00
}
fn mk_simple_visitor(v: simple_visitor) -> vt<()> {
fn v_mod(f: fn@(_mod, span, node_id), m: _mod, sp: span, id: node_id,
&&e: (), v: vt<()>) {
f(m, sp, id);
visit_mod(m, sp, id, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_view_item(f: fn@(@view_item), vi: @view_item, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(vi);
visit_view_item(vi, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_foreign_item(f: fn@(@foreign_item), ni: @foreign_item, &&e: (),
v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(ni);
visit_foreign_item(ni, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_item(f: fn@(@item), i: @item, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(i);
visit_item(i, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_local(f: fn@(@local), l: @local, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(l);
visit_local(l, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_block(f: fn@(ast::blk), bl: ast::blk, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(bl);
visit_block(bl, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_stmt(f: fn@(@stmt), st: @stmt, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(st);
visit_stmt(st, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_arm(f: fn@(arm), a: arm, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(a);
visit_arm(a, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_pat(f: fn@(@pat), p: @pat, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(p);
visit_pat(p, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_decl(f: fn@(@decl), d: @decl, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(d);
visit_decl(d, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_expr(f: fn@(@expr), ex: @expr, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(ex);
visit_expr(ex, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_expr_post(f: fn@(@expr), ex: @expr, &&_e: (), _v: vt<()>) {
f(ex);
}
fn v_ty(f: fn@(@ty), ty: @ty, &&e: (), v: vt<()>) {
2011-07-27 07:19:39 -05:00
f(ty);
visit_ty(ty, e, v);
2011-07-26 09:35:31 -05:00
}
fn v_ty_method(f: fn@(ty_method), ty: ty_method, &&e: (), v: vt<()>) {
f(ty);
visit_ty_method(ty, e, v);
}
fn v_trait_method(f: fn@(trait_method), m: trait_method, &&e: (),
v: vt<()>) {
f(m);
visit_trait_method(m, e, v);
}
2012-08-07 19:46:07 -05:00
fn v_struct_def(f: fn@(@struct_def, ident, ~[ty_param], node_id),
sd: @struct_def, nm: ident, tps: ~[ty_param], id: node_id,
&&e: (), v: vt<()>) {
f(sd, nm, tps, id);
visit_struct_def(sd, nm, tps, id, e, v);
}
fn v_ty_params(f: fn@(~[ty_param]),
ps: ~[ty_param],
&&e: (), v: vt<()>) {
f(ps);
visit_ty_params(ps, e, v);
}
fn v_fn(f: fn@(fn_kind, fn_decl, blk, span, node_id),
fk: fn_kind, decl: fn_decl, body: blk, sp: span,
id: node_id, &&e: (), v: vt<()>) {
f(fk, decl, body, sp, id);
visit_fn(fk, decl, body, sp, id, e, v);
2011-07-26 09:35:31 -05:00
}
let visit_ty = if v.visit_ty == simple_ignore_ty {
2012-06-30 18:19:07 -05:00
|a,b,c| skip_ty(a, b, c)
} else {
2012-06-30 18:19:07 -05:00
|a,b,c| v_ty(v.visit_ty, a, b, c)
};
fn v_class_item(f: fn@(@class_member),
cm: @class_member, &&e: (),
v: vt<()>) {
f(cm);
visit_class_item(cm, e, v);
}
2012-08-01 19:30:05 -05:00
return mk_vt(@{visit_mod: |a,b,c,d,e|v_mod(v.visit_mod, a, b, c, d, e),
2012-06-30 18:19:07 -05:00
visit_view_item: |a,b,c|
v_view_item(v.visit_view_item, a, b, c),
visit_foreign_item:
2012-06-30 18:19:07 -05:00
|a,b,c|v_foreign_item(v.visit_foreign_item, a, b, c),
visit_item: |a,b,c|v_item(v.visit_item, a, b, c),
visit_local: |a,b,c|v_local(v.visit_local, a, b, c),
visit_block: |a,b,c|v_block(v.visit_block, a, b, c),
visit_stmt: |a,b,c|v_stmt(v.visit_stmt, a, b, c),
visit_arm: |a,b,c|v_arm(v.visit_arm, a, b, c),
visit_pat: |a,b,c|v_pat(v.visit_pat, a, b, c),
visit_decl: |a,b,c|v_decl(v.visit_decl, a, b, c),
visit_expr: |a,b,c|v_expr(v.visit_expr, a, b, c),
visit_expr_post: |a,b,c| v_expr_post(v.visit_expr_post,
a, b, c),
visit_ty: visit_ty,
2012-06-30 18:19:07 -05:00
visit_ty_params: |a,b,c|
v_ty_params(v.visit_ty_params, a, b, c),
visit_fn: |a,b,c,d,e,f,g|
v_fn(v.visit_fn, a, b, c, d, e, f, g),
visit_ty_method: |a,b,c|
v_ty_method(v.visit_ty_method, a, b, c),
visit_trait_method: |a,b,c|
v_trait_method(v.visit_trait_method, a, b, c),
visit_struct_def: |a,b,c,d,e,f|
v_struct_def(v.visit_struct_def, a, b, c, d, e, f),
2012-06-30 18:19:07 -05:00
visit_class_item: |a,b,c|
2012-06-19 21:34:01 -05:00
v_class_item(v.visit_class_item, a, b, c)
});
2011-07-26 09:35:31 -05:00
}
2011-06-08 15:48:19 -05:00
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End: