rust/src/libsyntax/visit.rs

767 lines
25 KiB
Rust
Raw Normal View History

// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
use abi::AbiSet;
2012-09-04 13:37:29 -05:00
use ast::*;
use ast;
2012-09-04 13:37:29 -05:00
use codemap::span;
use parse;
use opt_vec;
use opt_vec::OptVec;
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.
2013-03-15 14:24:24 -05:00
//
// Note: it is an important invariant that the default visitor walks the body
// of a function in "execution order" (more concretely, reverse post-order
// with respect to the CFG implied by the AST), meaning that if AST node A may
// execute before AST node B, then A is visited first. The borrow checker in
// particular relies on this property.
2011-06-10 10:29:34 -05:00
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.
pub enum vt<E> { mk_vt(visitor<E>), }
2011-06-08 15:48:19 -05:00
pub enum fn_kind<'self> {
// fn foo() or extern "Abi" fn foo()
fk_item_fn(ident, &'self Generics, purity, AbiSet),
// fn foo(&self)
fk_method(ident, &'self Generics, &'self method),
// @fn(x, y) { ... }
fk_anon(ast::Sigil),
// |x, y| ...
fk_fn_block,
}
pub fn name_of_fn(fk: &fn_kind) -> ident {
match *fk {
fk_item_fn(name, _, _, _) | fk_method(name, _, _) => {
name
}
2012-07-18 18:18:02 -05:00
fk_anon(*) | fk_fn_block(*) => parse::token::special_idents::anon,
}
}
pub fn generics_of_fn(fk: &fn_kind) -> Generics {
match *fk {
fk_item_fn(_, generics, _, _) |
fk_method(_, generics, _) => {
2013-07-02 14:47:32 -05:00
(*generics).clone()
}
fk_anon(*) | fk_fn_block(*) => {
Generics {
lifetimes: opt_vec::Empty,
ty_params: opt_vec::Empty,
}
}
}
}
pub struct Visitor<E> {
visit_mod: @fn(&_mod, span, NodeId, (E, vt<E>)),
2013-07-05 03:28:53 -05:00
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(&Block, (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>)),
2013-07-05 23:57:11 -05:00
visit_ty: @fn(&Ty, (E, vt<E>)),
visit_generics: @fn(&Generics, (E, vt<E>)),
visit_fn: @fn(&fn_kind, &fn_decl, &Block, span, NodeId, (E, vt<E>)),
visit_ty_method: @fn(&TypeMethod, (E, vt<E>)),
visit_trait_method: @fn(&trait_method, (E, vt<E>)),
visit_struct_def: @fn(@struct_def, ident, &Generics, NodeId, (E, vt<E>)),
visit_struct_field: @fn(@struct_field, (E, vt<E>)),
}
pub type visitor<E> = @Visitor<E>;
2011-06-08 15:48:19 -05:00
2013-07-02 14:47:32 -05:00
pub fn default_visitor<E:Clone>() -> visitor<E> {
return @Visitor {
visit_mod: |a,b,c,d|visit_mod::<E>(a, b, c, d),
visit_view_item: |a,b|visit_view_item::<E>(a, b),
visit_foreign_item: |a,b|visit_foreign_item::<E>(a, b),
visit_item: |a,b|visit_item::<E>(a, b),
visit_local: |a,b|visit_local::<E>(a, b),
visit_block: |a,b|visit_block::<E>(a, b),
visit_stmt: |a,b|visit_stmt::<E>(a, b),
visit_arm: |a,b|visit_arm::<E>(a, b),
visit_pat: |a,b|visit_pat::<E>(a, b),
visit_decl: |a,b|visit_decl::<E>(a, b),
visit_expr: |a,b|visit_expr::<E>(a, b),
visit_expr_post: |_a,_b| (),
visit_ty: |a,b|skip_ty::<E>(a, b),
visit_generics: |a,b|visit_generics::<E>(a, b),
visit_fn: |a,b,c,d,e,f|visit_fn::<E>(a, b, c, d, e, f),
visit_ty_method: |a,b|visit_ty_method::<E>(a, b),
visit_trait_method: |a,b|visit_trait_method::<E>(a, b),
visit_struct_def: |a,b,c,d,e|visit_struct_def::<E>(a, b, c, d, e),
visit_struct_field: |a,b|visit_struct_field::<E>(a, b),
};
2011-06-08 15:48:19 -05:00
}
pub fn visit_crate<E:Clone>(c: &Crate, (e, v): (E, vt<E>)) {
(v.visit_mod)(&c.module, c.span, CRATE_NODE_ID, (e, v));
2011-06-08 15:48:19 -05:00
}
2013-07-02 14:47:32 -05:00
pub fn visit_mod<E:Clone>(m: &_mod,
_sp: span,
_id: NodeId,
2013-07-02 14:47:32 -05:00
(e, v): (E, vt<E>)) {
for m.view_items.iter().advance |vi| {
(v.visit_view_item)(vi, (e.clone(), v));
}
for m.items.iter().advance |i| {
(v.visit_item)(*i, (e.clone(), v));
}
2011-06-08 15:48:19 -05:00
}
pub fn visit_view_item<E>(_vi: &view_item, (_e, _v): (E, vt<E>)) { }
2011-06-09 08:50:20 -05:00
pub fn visit_local<E:Clone>(loc: &Local, (e, v): (E, vt<E>)) {
(v.visit_pat)(loc.pat, (e.clone(), v));
(v.visit_ty)(&loc.ty, (e.clone(), v));
match loc.init {
2012-08-20 14:23:37 -05:00
None => (),
Some(ex) => (v.visit_expr)(ex, (e, v))
2012-08-03 21:59:04 -05:00
}
}
2013-07-02 14:47:32 -05:00
fn visit_trait_ref<E:Clone>(tref: &ast::trait_ref, (e, v): (E, vt<E>)) {
2013-07-05 05:15:21 -05:00
visit_path(&tref.path, (e, v));
}
2013-07-02 14:47:32 -05:00
pub fn visit_item<E:Clone>(i: &item, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match i.node {
2013-07-05 23:57:11 -05:00
item_static(ref t, _, ex) => {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(t, (e.clone(), v));
(v.visit_expr)(ex, (e.clone(), v));
}
item_fn(ref decl, purity, abi, ref generics, ref body) => {
(v.visit_fn)(
&fk_item_fn(
i.ident,
generics,
purity,
abi
),
decl,
body,
i.span,
i.id,
(e,
v)
);
}
item_mod(ref m) => (v.visit_mod)(m, i.span, i.id, (e, v)),
item_foreign_mod(ref nm) => {
2013-07-02 14:47:32 -05:00
for nm.view_items.iter().advance |vi| {
(v.visit_view_item)(vi, (e.clone(), v));
}
for nm.items.iter().advance |ni| {
(v.visit_foreign_item)(*ni, (e.clone(), v));
}
}
2013-07-05 23:57:11 -05:00
item_ty(ref t, ref tps) => {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(t, (e.clone(), v));
(v.visit_generics)(tps, (e, v));
}
item_enum(ref enum_definition, ref tps) => {
2013-07-02 14:47:32 -05:00
(v.visit_generics)(tps, (e.clone(), v));
visit_enum_def(
enum_definition,
tps,
(e, v)
);
}
2013-07-05 23:57:11 -05:00
item_impl(ref tps, ref traits, ref ty, ref methods) => {
2013-07-02 14:47:32 -05:00
(v.visit_generics)(tps, (e.clone(), v));
for traits.iter().advance |p| {
2013-07-02 14:47:32 -05:00
visit_trait_ref(p, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
(v.visit_ty)(ty, (e.clone(), v));
for methods.iter().advance |m| {
2013-07-02 14:47:32 -05:00
visit_method_helper(*m, (e.clone(), v))
}
}
item_struct(struct_def, ref generics) => {
2013-07-02 14:47:32 -05:00
(v.visit_generics)(generics, (e.clone(), v));
(v.visit_struct_def)(struct_def, i.ident, generics, i.id, (e, v));
}
item_trait(ref generics, ref traits, ref methods) => {
2013-07-02 14:47:32 -05:00
(v.visit_generics)(generics, (e.clone(), v));
for traits.iter().advance |p| {
visit_path(&p.path, (e.clone(), v));
}
for methods.iter().advance |m| {
2013-07-02 14:47:32 -05:00
(v.visit_trait_method)(m, (e.clone(), v));
}
}
item_mac(ref m) => visit_mac(m, (e, v))
2011-06-08 15:48:19 -05:00
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_enum_def<E:Clone>(enum_definition: &ast::enum_def,
tps: &Generics,
(e, v): (E, vt<E>)) {
for enum_definition.variants.iter().advance |vr| {
match vr.node.kind {
tuple_variant_kind(ref variant_args) => {
for variant_args.iter().advance |va| {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(&va.ty, (e.clone(), v));
}
}
struct_variant_kind(struct_def) => {
(v.visit_struct_def)(struct_def, vr.node.name, tps,
2013-07-02 14:47:32 -05:00
vr.node.id, (e.clone(), v));
}
}
// Visit the disr expr if it exists
for vr.node.disr_expr.iter().advance |ex| {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(*ex, (e.clone(), v))
}
}
}
pub fn skip_ty<E>(_t: &Ty, (_e,_v): (E, vt<E>)) {}
2013-07-02 14:47:32 -05:00
pub fn visit_ty<E:Clone>(t: &Ty, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match t.node {
2013-07-05 23:57:11 -05:00
ty_box(ref mt) | ty_uniq(ref mt) |
ty_vec(ref mt) | ty_ptr(ref mt) | ty_rptr(_, ref mt) => {
(v.visit_ty)(mt.ty, (e, v));
},
ty_tup(ref ts) => {
for ts.iter().advance |tt| {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(tt, (e.clone(), v));
}
},
ty_closure(ref f) => {
2013-07-02 14:47:32 -05:00
for f.decl.inputs.iter().advance |a| {
(v.visit_ty)(&a.ty, (e.clone(), v));
}
(v.visit_ty)(&f.decl.output, (e.clone(), v));
do f.bounds.map |bounds| {
2013-07-02 14:47:32 -05:00
visit_ty_param_bounds(bounds, (e.clone(), v));
};
},
ty_bare_fn(ref f) => {
2013-07-02 14:47:32 -05:00
for f.decl.inputs.iter().advance |a| {
(v.visit_ty)(&a.ty, (e.clone(), v));
}
2013-07-05 23:57:11 -05:00
(v.visit_ty)(&f.decl.output, (e, v));
},
2013-07-05 20:38:56 -05:00
ty_path(ref p, ref bounds, _) => {
2013-07-02 14:47:32 -05:00
visit_path(p, (e.clone(), v));
do bounds.map |bounds| {
2013-07-02 14:47:32 -05:00
visit_ty_param_bounds(bounds, (e.clone(), v));
};
},
ty_fixed_length_vec(ref mt, ex) => {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(mt.ty, (e.clone(), v));
(v.visit_expr)(ex, (e.clone(), v));
},
ty_nil | ty_bot | ty_mac(_) | ty_infer => ()
2011-06-08 15:48:19 -05:00
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_path<E:Clone>(p: &Path, (e, v): (E, vt<E>)) {
for p.types.iter().advance |tp| { (v.visit_ty)(tp, (e.clone(), v)); }
}
2013-07-02 14:47:32 -05:00
pub fn visit_pat<E:Clone>(p: &pat, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match p.node {
2013-07-05 05:15:21 -05:00
pat_enum(ref path, ref children) => {
2013-07-02 14:47:32 -05:00
visit_path(path, (e.clone(), v));
for children.iter().advance |children| {
for children.iter().advance |child| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*child, (e.clone(), v));
}
}
}
2013-07-05 05:15:21 -05:00
pat_struct(ref path, ref fields, _) => {
2013-07-02 14:47:32 -05:00
visit_path(path, (e.clone(), v));
for fields.iter().advance |f| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(f.pat, (e.clone(), v));
}
}
pat_tup(ref elts) => {
for elts.iter().advance |elt| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*elt, (e.clone(), v))
}
},
pat_box(inner) | pat_uniq(inner) | pat_region(inner) => {
(v.visit_pat)(inner, (e, v))
},
2013-07-05 05:15:21 -05:00
pat_ident(_, ref path, ref inner) => {
2013-07-02 14:47:32 -05:00
visit_path(path, (e.clone(), v));
for inner.iter().advance |subpat| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*subpat, (e.clone(), v))
}
}
pat_lit(ex) => (v.visit_expr)(ex, (e, v)),
pat_range(e1, e2) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(e1, (e.clone(), v));
(v.visit_expr)(e2, (e, v));
}
pat_wild => (),
pat_vec(ref before, ref slice, ref after) => {
for before.iter().advance |elt| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*elt, (e.clone(), v));
}
for slice.iter().advance |elt| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*elt, (e.clone(), v));
}
for after.iter().advance |tail| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(*tail, (e.clone(), v));
}
}
2011-06-08 15:48:19 -05:00
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_foreign_item<E:Clone>(ni: &foreign_item, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match ni.node {
foreign_item_fn(ref fd, _, ref generics) => {
2013-07-02 14:47:32 -05:00
visit_fn_decl(fd, (e.clone(), v));
(v.visit_generics)(generics, (e, v));
}
2013-07-05 23:57:11 -05:00
foreign_item_static(ref t, _) => {
(v.visit_ty)(t, (e, v));
}
2011-06-08 15:48:19 -05:00
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_ty_param_bounds<E:Clone>(bounds: &OptVec<TyParamBound>,
(e, v): (E, vt<E>)) {
for bounds.iter().advance |bound| {
match *bound {
2013-07-02 14:47:32 -05:00
TraitTyParamBound(ref ty) => visit_trait_ref(ty, (e.clone(), v)),
RegionTyParamBound => {}
}
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_generics<E:Clone>(generics: &Generics, (e, v): (E, vt<E>)) {
for generics.ty_params.iter().advance |tp| {
2013-07-02 14:47:32 -05:00
visit_ty_param_bounds(&tp.bounds, (e.clone(), v));
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_fn_decl<E:Clone>(fd: &fn_decl, (e, v): (E, vt<E>)) {
for fd.inputs.iter().advance |a| {
2013-07-02 14:47:32 -05:00
(v.visit_pat)(a.pat, (e.clone(), v));
(v.visit_ty)(&a.ty, (e.clone(), v));
}
2013-07-05 23:57:11 -05:00
(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
2013-07-02 14:47:32 -05:00
pub fn visit_method_helper<E:Clone>(m: &method, (e, v): (E, vt<E>)) {
(v.visit_fn)(&fk_method(m.ident, &m.generics, m),
&m.decl,
&m.body,
m.span,
m.id,
(e, v));
}
pub fn visit_fn<E:Clone>(fk: &fn_kind, decl: &fn_decl, body: &Block, _sp: span,
_id: NodeId, (e, v): (E, vt<E>)) {
2013-07-02 14:47:32 -05:00
visit_fn_decl(decl, (e.clone(), v));
let generics = generics_of_fn(fk);
2013-07-02 14:47:32 -05:00
(v.visit_generics)(&generics, (e.clone(), v));
(v.visit_block)(body, (e, v));
2011-06-08 15:48:19 -05:00
}
pub fn visit_ty_method<E:Clone>(m: &TypeMethod, (e, v): (E, vt<E>)) {
2013-07-02 14:47:32 -05:00
for m.decl.inputs.iter().advance |a| {
(v.visit_ty)(&a.ty, (e.clone(), v));
}
(v.visit_generics)(&m.generics, (e.clone(), v));
2013-07-05 23:57:11 -05:00
(v.visit_ty)(&m.decl.output, (e, v));
}
2013-07-02 14:47:32 -05:00
pub fn visit_trait_method<E:Clone>(m: &trait_method, (e, v): (E, vt<E>)) {
match *m {
required(ref ty_m) => (v.visit_ty_method)(ty_m, (e, v)),
provided(m) => visit_method_helper(m, (e, v))
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_struct_def<E:Clone>(
sd: @struct_def,
_nm: ast::ident,
_generics: &Generics,
_id: NodeId,
(e, v): (E, vt<E>)
) {
for sd.fields.iter().advance |f| {
2013-07-02 14:47:32 -05:00
(v.visit_struct_field)(*f, (e.clone(), v));
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_struct_field<E:Clone>(sf: &struct_field, (e, v): (E, vt<E>)) {
2013-07-05 23:57:11 -05:00
(v.visit_ty)(&sf.node.ty, (e, v));
}
pub fn visit_block<E:Clone>(b: &Block, (e, v): (E, vt<E>)) {
for b.view_items.iter().advance |vi| {
2013-07-02 14:47:32 -05:00
(v.visit_view_item)(vi, (e.clone(), v));
}
for b.stmts.iter().advance |s| {
2013-07-02 14:47:32 -05:00
(v.visit_stmt)(*s, (e.clone(), v));
}
visit_expr_opt(b.expr, (e, v));
2011-06-08 15:48:19 -05:00
}
pub fn visit_stmt<E>(s: &stmt, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match s.node {
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)),
stmt_mac(ref mac, _) => visit_mac(mac, (e, v))
2011-06-08 15:48:19 -05:00
}
}
2013-07-02 14:47:32 -05:00
pub fn visit_decl<E:Clone>(d: &decl, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match d.node {
decl_local(ref loc) => (v.visit_local)(*loc, (e, v)),
decl_item(it) => (v.visit_item)(it, (e, v))
2011-06-08 15:48:19 -05:00
}
}
pub fn visit_expr_opt<E>(eo: Option<@expr>, (e, v): (E, vt<E>)) {
match eo { None => (), Some(ex) => (v.visit_expr)(ex, (e, v)) }
2011-06-08 15:48:19 -05:00
}
2013-07-02 14:47:32 -05:00
pub fn visit_exprs<E:Clone>(exprs: &[@expr], (e, v): (E, vt<E>)) {
for exprs.iter().advance |ex| { (v.visit_expr)(*ex, (e.clone(), v)); }
2011-06-08 15:48:19 -05:00
}
pub fn visit_mac<E>(_m: &mac, (_e, _v): (E, vt<E>)) {
/* no user-serviceable parts inside */
}
2013-07-02 14:47:32 -05:00
pub fn visit_expr<E:Clone>(ex: @expr, (e, v): (E, vt<E>)) {
2012-08-06 14:34:08 -05:00
match ex.node {
2013-07-02 14:47:32 -05:00
expr_vstore(x, _) => (v.visit_expr)(x, (e.clone(), v)),
expr_vec(ref es, _) => visit_exprs(*es, (e.clone(), v)),
expr_repeat(element, count, _) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(element, (e.clone(), v));
(v.visit_expr)(count, (e.clone(), v));
}
2013-07-05 05:15:21 -05:00
expr_struct(ref p, ref flds, base) => {
2013-07-02 14:47:32 -05:00
visit_path(p, (e.clone(), v));
for flds.iter().advance |f| {
(v.visit_expr)(f.expr, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
visit_expr_opt(base, (e.clone(), v));
}
expr_tup(ref elts) => {
2013-07-02 14:47:32 -05:00
for elts.iter().advance |el| { (v.visit_expr)(*el, (e.clone(), v)) }
}
expr_call(callee, ref args, _) => {
2013-07-02 14:47:32 -05:00
visit_exprs(*args, (e.clone(), v));
(v.visit_expr)(callee, (e.clone(), v));
}
expr_method_call(_, callee, _, ref tys, ref args, _) => {
2013-07-02 14:47:32 -05:00
visit_exprs(*args, (e.clone(), v));
for tys.iter().advance |tp| {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(tp, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
(v.visit_expr)(callee, (e.clone(), v));
}
expr_binary(_, _, a, b) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(a, (e.clone(), v));
(v.visit_expr)(b, (e.clone(), v));
}
expr_addr_of(_, x) | expr_unary(_, _, x) |
2013-07-02 14:47:32 -05:00
expr_loop_body(x) | expr_do_body(x) => (v.visit_expr)(x, (e.clone(), v)),
expr_lit(_) => (),
2013-07-05 23:57:11 -05:00
expr_cast(x, ref t) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(x, (e.clone(), v));
(v.visit_ty)(t, (e.clone(), v));
}
expr_if(x, ref b, eo) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(x, (e.clone(), v));
(v.visit_block)(b, (e.clone(), v));
visit_expr_opt(eo, (e.clone(), v));
}
expr_while(x, ref b) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(x, (e.clone(), v));
(v.visit_block)(b, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
expr_loop(ref b, _) => (v.visit_block)(b, (e.clone(), v)),
expr_match(x, ref arms) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(x, (e.clone(), v));
for arms.iter().advance |a| { (v.visit_arm)(a, (e.clone(), v)); }
}
expr_fn_block(ref decl, ref body) => {
(v.visit_fn)(
&fk_fn_block,
decl,
body,
ex.span,
ex.id,
2013-07-02 14:47:32 -05:00
(e.clone(), v)
);
}
2013-07-02 14:47:32 -05:00
expr_block(ref b) => (v.visit_block)(b, (e.clone(), v)),
expr_assign(a, b) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(b, (e.clone(), v));
(v.visit_expr)(a, (e.clone(), v));
}
expr_assign_op(_, _, a, b) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(b, (e.clone(), v));
(v.visit_expr)(a, (e.clone(), v));
}
expr_field(x, _, ref tys) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(x, (e.clone(), v));
for tys.iter().advance |tp| {
2013-07-02 14:47:32 -05:00
(v.visit_ty)(tp, (e.clone(), v));
}
}
expr_index(_, a, b) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(a, (e.clone(), v));
(v.visit_expr)(b, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
expr_path(ref p) => visit_path(p, (e.clone(), v)),
expr_self => (),
expr_break(_) => (),
expr_again(_) => (),
2013-07-02 14:47:32 -05:00
expr_ret(eo) => visit_expr_opt(eo, (e.clone(), v)),
expr_log(lv, x) => {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(lv, (e.clone(), v));
(v.visit_expr)(x, (e.clone(), v));
}
2013-07-02 14:47:32 -05:00
expr_mac(ref mac) => visit_mac(mac, (e.clone(), v)),
expr_paren(x) => (v.visit_expr)(x, (e.clone(), v)),
expr_inline_asm(ref a) => {
for a.inputs.iter().advance |&(_, in)| {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(in, (e.clone(), v));
}
for a.outputs.iter().advance |&(_, out)| {
2013-07-02 14:47:32 -05:00
(v.visit_expr)(out, (e.clone(), v));
}
}
2011-06-08 15:48:19 -05:00
}
(v.visit_expr_post)(ex, (e, v));
2011-06-08 15:48:19 -05:00
}
2013-07-02 14:47:32 -05:00
pub fn visit_arm<E:Clone>(a: &arm, (e, v): (E, vt<E>)) {
for a.pats.iter().advance |p| { (v.visit_pat)(*p, (e.clone(), v)); }
visit_expr_opt(a.guard, (e.clone(), v));
(v.visit_block)(&a.body, (e.clone(), 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.
pub struct SimpleVisitor {
visit_mod: @fn(&_mod, span, NodeId),
2013-07-05 03:28:53 -05:00
visit_view_item: @fn(&view_item),
visit_foreign_item: @fn(@foreign_item),
visit_item: @fn(@item),
visit_local: @fn(@Local),
visit_block: @fn(&Block),
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),
2013-07-05 23:57:11 -05:00
visit_ty: @fn(&Ty),
visit_generics: @fn(&Generics),
visit_fn: @fn(&fn_kind, &fn_decl, &Block, span, NodeId),
visit_ty_method: @fn(&TypeMethod),
visit_trait_method: @fn(&trait_method),
visit_struct_def: @fn(@struct_def, ident, &Generics, NodeId),
visit_struct_field: @fn(@struct_field),
visit_struct_method: @fn(@method)
}
pub type simple_visitor = @SimpleVisitor;
2011-07-26 09:35:31 -05:00
2013-07-05 23:57:11 -05:00
pub fn simple_ignore_ty(_t: &Ty) {}
pub fn default_simple_visitor() -> @SimpleVisitor {
2013-02-17 11:42:50 -06:00
@SimpleVisitor {
visit_mod: |_m, _sp, _id| { },
visit_view_item: |_vi| { },
visit_foreign_item: |_ni| { },
visit_item: |_i| { },
visit_local: |_l| { },
visit_block: |_b| { },
visit_stmt: |_s| { },
visit_arm: |_a| { },
visit_pat: |_p| { },
visit_decl: |_d| { },
visit_expr: |_e| { },
visit_expr_post: |_e| { },
visit_ty: simple_ignore_ty,
visit_generics: |_| {},
visit_fn: |_, _, _, _, _| {},
visit_ty_method: |_| {},
visit_trait_method: |_| {},
visit_struct_def: |_, _, _, _| {},
visit_struct_field: |_| {},
visit_struct_method: |_| {},
2013-02-17 11:42:50 -06:00
}
2011-07-26 09:35:31 -05:00
}
pub fn mk_simple_visitor(v: simple_visitor) -> vt<()> {
fn v_mod(
f: @fn(&_mod, span, NodeId),
m: &_mod,
sp: span,
id: NodeId,
(e, v): ((), vt<()>)
) {
f(m, sp, id);
visit_mod(m, sp, id, (e, v));
2011-07-26 09:35:31 -05:00
}
2013-07-05 03:28:53 -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::Block), bl: &ast::Block, (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);
}
2013-07-05 23:57:11 -05:00
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(&TypeMethod), ty: &TypeMethod, (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));
}
fn v_struct_def(
f: @fn(@struct_def, ident, &Generics, NodeId),
sd: @struct_def,
nm: ident,
generics: &Generics,
id: NodeId,
(e, v): ((), vt<()>)
) {
f(sd, nm, generics, id);
visit_struct_def(sd, nm, generics, id, (e, v));
}
fn v_generics(
f: @fn(&Generics),
ps: &Generics,
(e, v): ((), vt<()>)
) {
f(ps);
visit_generics(ps, (e, v));
}
fn v_fn(
f: @fn(&fn_kind, &fn_decl, &Block, span, NodeId),
fk: &fn_kind,
decl: &fn_decl,
body: &Block,
sp: span,
id: NodeId,
(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
}
2013-07-05 23:57:11 -05:00
let visit_ty: @fn(&Ty, ((), vt<()>)) =
|a,b| v_ty(v.visit_ty, a, b);
fn v_struct_field(f: @fn(@struct_field), sf: @struct_field, (e, v): ((), vt<()>)) {
f(sf);
visit_struct_field(sf, (e, v));
}
return mk_vt(@Visitor {
visit_mod: |a,b,c,d|v_mod(v.visit_mod, a, b, c, d),
visit_view_item: |a,b| v_view_item(v.visit_view_item, a, b),
visit_foreign_item:
|a,b|v_foreign_item(v.visit_foreign_item, a, b),
visit_item: |a,b|v_item(v.visit_item, a, b),
visit_local: |a,b|v_local(v.visit_local, a, b),
visit_block: |a,b|v_block(v.visit_block, a, b),
visit_stmt: |a,b|v_stmt(v.visit_stmt, a, b),
visit_arm: |a,b|v_arm(v.visit_arm, a, b),
visit_pat: |a,b|v_pat(v.visit_pat, a, b),
visit_decl: |a,b|v_decl(v.visit_decl, a, b),
visit_expr: |a,b|v_expr(v.visit_expr, a, b),
visit_expr_post: |a,b| v_expr_post(v.visit_expr_post, a, b),
visit_ty: visit_ty,
visit_generics: |a,b|
v_generics(v.visit_generics, a, b),
visit_fn: |a,b,c,d,e,f|
v_fn(v.visit_fn, a, b, c, d, e, f),
visit_ty_method: |a,b|
v_ty_method(v.visit_ty_method, a, b),
visit_trait_method: |a,b|
v_trait_method(v.visit_trait_method, a, b),
visit_struct_def: |a,b,c,d,e|
v_struct_def(v.visit_struct_def, a, b, c, d, e),
visit_struct_field: |a,b|
v_struct_field(v.visit_struct_field, a, b),
});
2011-07-26 09:35:31 -05:00
}