2011-06-15 13:19:50 -05:00
|
|
|
|
2011-07-05 04:48:19 -05:00
|
|
|
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 {
|
2012-01-19 19:56:05 -06:00
|
|
|
fk_item_fn(ident, [ty_param]), //< an item declared with fn()
|
|
|
|
fk_method(ident, [ty_param]),
|
|
|
|
fk_res(ident, [ty_param]),
|
|
|
|
fk_anon(proto), //< an anonymous function like fn@(...)
|
|
|
|
fk_fn_block, //< a block {||...}
|
2011-12-29 22:07:55 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
fn name_of_fn(fk: fn_kind) -> ident {
|
|
|
|
alt fk {
|
|
|
|
fk_item_fn(name, _) | fk_method(name, _) | fk_res(name, _) { name }
|
2012-01-19 00:37:22 -06:00
|
|
|
fk_anon(_) | fk_fn_block { "anon" }
|
2011-12-29 22:07:55 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn tps_of_fn(fk: fn_kind) -> [ty_param] {
|
|
|
|
alt fk {
|
|
|
|
fk_item_fn(_, tps) | fk_method(_, tps) | fk_res(_, tps) { tps }
|
2012-01-19 00:37:22 -06:00
|
|
|
fk_anon(_) | fk_fn_block { [] }
|
2011-12-29 22:07:55 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-08-12 08:36:51 -05:00
|
|
|
type visitor<E> =
|
2011-07-27 07:19:39 -05:00
|
|
|
// takes the components so that one function can be
|
|
|
|
// generic over constr and ty_constr
|
2012-01-26 10:33:12 -06:00
|
|
|
@{visit_mod: fn@(_mod, span, node_id, E, vt<E>),
|
2011-10-18 17:07:40 -05:00
|
|
|
visit_view_item: fn@(@view_item, E, vt<E>),
|
|
|
|
visit_native_item: fn@(@native_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_ty: fn@(@ty, E, vt<E>),
|
2012-01-04 09:37:39 -06:00
|
|
|
visit_ty_params: fn@([ty_param], E, vt<E>),
|
2011-11-30 06:38:38 -06:00
|
|
|
visit_constr: fn@(@path, span, node_id, E, vt<E>),
|
2012-01-31 21:30:40 -06:00
|
|
|
visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id, E, vt<E>),
|
2012-02-07 13:31:15 -06:00
|
|
|
visit_class_item: fn@(span, privacy, class_member, E, vt<E>)};
|
2011-06-08 15:48:19 -05:00
|
|
|
|
2011-08-12 08:36:51 -05:00
|
|
|
fn default_visitor<E>() -> visitor<E> {
|
2012-01-26 10:33:12 -06:00
|
|
|
ret @{visit_mod: bind visit_mod::<E>(_, _, _, _, _),
|
2011-08-13 02:09:25 -05:00
|
|
|
visit_view_item: bind visit_view_item::<E>(_, _, _),
|
|
|
|
visit_native_item: bind visit_native_item::<E>(_, _, _),
|
|
|
|
visit_item: bind visit_item::<E>(_, _, _),
|
|
|
|
visit_local: bind visit_local::<E>(_, _, _),
|
|
|
|
visit_block: bind visit_block::<E>(_, _, _),
|
|
|
|
visit_stmt: bind visit_stmt::<E>(_, _, _),
|
|
|
|
visit_arm: bind visit_arm::<E>(_, _, _),
|
|
|
|
visit_pat: bind visit_pat::<E>(_, _, _),
|
|
|
|
visit_decl: bind visit_decl::<E>(_, _, _),
|
|
|
|
visit_expr: bind visit_expr::<E>(_, _, _),
|
2011-11-22 03:57:47 -06:00
|
|
|
visit_ty: bind skip_ty::<E>(_, _, _),
|
2012-01-04 09:37:39 -06:00
|
|
|
visit_ty_params: bind visit_ty_params::<E>(_, _, _),
|
2011-08-13 02:09:25 -05:00
|
|
|
visit_constr: bind visit_constr::<E>(_, _, _, _, _),
|
2012-01-31 21:30:40 -06:00
|
|
|
visit_fn: bind visit_fn::<E>(_, _, _, _, _, _, _),
|
|
|
|
visit_class_item: bind visit_class_item::<E>(_,_,_,_,_)};
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_crate<E>(c: crate, e: E, v: vt<E>) {
|
2012-01-26 10:33:12 -06:00
|
|
|
v.visit_mod(c.node.module, c.span, crate_node_id, e, v);
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_crate_directive<E>(cd: @crate_directive, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt cd.node {
|
2011-11-21 22:31:09 -06:00
|
|
|
cdir_src_mod(_, _) { }
|
|
|
|
cdir_dir_mod(_, cdirs, _) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for cdir: @crate_directive in cdirs {
|
2011-07-27 07:19:39 -05:00
|
|
|
visit_crate_directive(cdir, e, v);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
cdir_view_item(vi) { v.visit_view_item(vi, e, v); }
|
|
|
|
cdir_syntax(_) { }
|
2011-06-09 08:50:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-01-26 10:33:12 -06:00
|
|
|
fn visit_mod<E>(m: _mod, _sp: span, _id: node_id, e: E, v: vt<E>) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for vi: @view_item in m.view_items { v.visit_view_item(vi, e, v); }
|
|
|
|
for i: @item in m.items { v.visit_item(i, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_view_item<E>(_vi: @view_item, _e: E, _v: vt<E>) { }
|
2011-06-09 08:50:20 -05:00
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_local<E>(loc: @local, e: E, v: vt<E>) {
|
2011-07-28 05:01:45 -05:00
|
|
|
v.visit_pat(loc.node.pat, e, v);
|
2011-08-10 14:51:50 -05:00
|
|
|
v.visit_ty(loc.node.ty, e, v);
|
2012-01-19 00:37:22 -06:00
|
|
|
alt loc.node.init { none { } some(i) { v.visit_expr(i.expr, e, v); } }
|
2011-06-13 19:04:15 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_item<E>(i: @item, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt i.node {
|
|
|
|
item_const(t, ex) { v.visit_ty(t, e, v); v.visit_expr(ex, e, v); }
|
2011-12-22 10:49:54 -06:00
|
|
|
item_fn(decl, tp, body) {
|
2011-12-29 22:07:55 -06:00
|
|
|
v.visit_fn(fk_item_fn(i.ident, tp), decl, body, i.span, i.id, e, v);
|
2011-12-22 10:49:54 -06:00
|
|
|
}
|
2012-01-26 10:33:12 -06:00
|
|
|
item_mod(m) { v.visit_mod(m, i.span, i.id, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
item_native_mod(nm) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for vi: @view_item in nm.view_items { v.visit_view_item(vi, e, v); }
|
|
|
|
for ni: @native_item in nm.items { v.visit_native_item(ni, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2012-01-04 09:37:39 -06:00
|
|
|
item_ty(t, tps) { v.visit_ty(t, e, v); v.visit_ty_params(tps, e, v); }
|
2011-12-22 10:49:54 -06:00
|
|
|
item_res(decl, tps, body, dtor_id, _) {
|
2011-12-29 22:07:55 -06:00
|
|
|
v.visit_fn(fk_res(i.ident, tps), decl, body, i.span,
|
|
|
|
dtor_id, e, v);
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2012-01-25 07:34:31 -06:00
|
|
|
item_enum(variants, tps) {
|
2012-01-04 09:37:39 -06:00
|
|
|
v.visit_ty_params(tps, e, v);
|
2011-08-15 23:54:52 -05:00
|
|
|
for vr: variant in variants {
|
|
|
|
for va: variant_arg in vr.node.args { v.visit_ty(va.ty, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
|
|
|
}
|
2011-12-28 10:50:12 -06:00
|
|
|
item_impl(tps, ifce, ty, methods) {
|
2012-01-04 09:37:39 -06:00
|
|
|
v.visit_ty_params(tps, e, v);
|
2011-12-20 09:33:55 -06:00
|
|
|
alt ifce { some(ty) { v.visit_ty(ty, e, v); } _ {} }
|
|
|
|
v.visit_ty(ty, e, v);
|
2011-12-13 06:19:56 -06:00
|
|
|
for m in methods {
|
2011-12-29 22:07:55 -06:00
|
|
|
v.visit_fn(fk_method(m.ident, m.tps), m.decl, m.body, m.span,
|
|
|
|
m.id, e, v);
|
2011-12-13 06:19:56 -06:00
|
|
|
}
|
|
|
|
}
|
2012-02-07 13:31:15 -06:00
|
|
|
item_class(tps, members, _, ctor_decl, ctor_blk) {
|
2012-01-31 21:30:40 -06:00
|
|
|
v.visit_ty_params(tps, e, v);
|
|
|
|
for m in members {
|
|
|
|
v.visit_class_item(m.span, m.node.privacy, m.node.decl, e, v);
|
|
|
|
}
|
|
|
|
visit_fn_decl(ctor_decl, e, v);
|
|
|
|
v.visit_block(ctor_blk, e, v);
|
|
|
|
}
|
2011-12-28 10:50:12 -06:00
|
|
|
item_iface(tps, methods) {
|
2012-01-04 09:37:39 -06:00
|
|
|
v.visit_ty_params(tps, e, v);
|
2011-12-20 09:33:55 -06:00
|
|
|
for m in methods {
|
|
|
|
for a in m.decl.inputs { v.visit_ty(a.ty, e, v); }
|
|
|
|
v.visit_ty(m.decl.output, e, v);
|
|
|
|
}
|
|
|
|
}
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-02-07 13:31:15 -06:00
|
|
|
fn visit_class_item<E>(_s: span, _p: privacy, cm: class_member,
|
2012-01-31 21:30:40 -06:00
|
|
|
e:E, v:vt<E>) {
|
2012-02-07 13:31:15 -06:00
|
|
|
alt cm {
|
2012-01-31 21:30:40 -06:00
|
|
|
instance_var(ident, t, mt, id) {
|
|
|
|
v.visit_ty(t, e, v);
|
|
|
|
}
|
|
|
|
class_method(i) {
|
|
|
|
v.visit_item(i, e, v);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-11-22 03:57:47 -06:00
|
|
|
fn skip_ty<E>(_t: @ty, _e: E, _v: vt<E>) {}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_ty<E>(t: @ty, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt t.node {
|
|
|
|
ty_box(mt) { v.visit_ty(mt.ty, e, v); }
|
2011-09-20 18:07:09 -05:00
|
|
|
ty_uniq(mt) { v.visit_ty(mt.ty, e, v); }
|
2011-08-18 16:11:06 -05:00
|
|
|
ty_vec(mt) { v.visit_ty(mt.ty, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
ty_ptr(mt) { v.visit_ty(mt.ty, e, v); }
|
|
|
|
ty_rec(flds) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for f: ty_field in flds { v.visit_ty(f.node.mt.ty, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2011-08-19 17:16:48 -05:00
|
|
|
ty_tup(ts) { for tt in ts { v.visit_ty(tt, e, v); } }
|
2011-12-29 22:07:55 -06:00
|
|
|
ty_fn(_, decl) {
|
2011-12-22 10:49:54 -06:00
|
|
|
for a in decl.inputs { v.visit_ty(a.ty, e, v); }
|
|
|
|
for c: @constr in decl.constraints {
|
2011-07-27 07:19:39 -05:00
|
|
|
v.visit_constr(c.node.path, c.span, c.node.id, e, v);
|
|
|
|
}
|
2011-12-22 10:49:54 -06:00
|
|
|
v.visit_ty(decl.output, e, v);
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2011-12-13 06:19:56 -06:00
|
|
|
ty_path(p, _) { visit_path(p, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
ty_constr(t, cs) {
|
|
|
|
v.visit_ty(t, e, v);
|
2011-11-30 06:38:38 -06:00
|
|
|
for tc: @spanned<constr_general_<@path, node_id>> in cs {
|
2011-07-27 07:19:39 -05:00
|
|
|
v.visit_constr(tc.node.path, tc.span, tc.node.id, e, v);
|
|
|
|
}
|
|
|
|
}
|
2011-12-07 14:06:12 -06:00
|
|
|
_ {}
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-11-30 06:38:38 -06:00
|
|
|
fn visit_constr<E>(_operator: @path, _sp: span, _id: node_id, _e: E,
|
2011-09-12 04:27:30 -05:00
|
|
|
_v: vt<E>) {
|
2011-06-10 21:12:42 -05:00
|
|
|
// default
|
|
|
|
}
|
|
|
|
|
2011-12-13 06:19:56 -06:00
|
|
|
fn visit_path<E>(p: @path, e: E, v: vt<E>) {
|
|
|
|
for tp: @ty in p.node.types { v.visit_ty(tp, e, v); }
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_pat<E>(p: @pat, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt p.node {
|
2012-01-25 07:34:31 -06:00
|
|
|
pat_enum(path, children) {
|
2011-12-13 06:19:56 -06:00
|
|
|
visit_path(path, e, v);
|
2011-08-15 23:54:52 -05:00
|
|
|
for child: @pat in children { v.visit_pat(child, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
|
|
|
pat_rec(fields, _) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for f: field_pat in fields { v.visit_pat(f.pat, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2011-08-19 17:16:48 -05:00
|
|
|
pat_tup(elts) { for elt in elts { v.visit_pat(elt, e, v); } }
|
2012-01-14 18:05:07 -06:00
|
|
|
pat_box(inner) | pat_uniq(inner) {
|
2011-12-08 04:56:16 -06:00
|
|
|
v.visit_pat(inner, e, v);
|
|
|
|
}
|
2012-01-14 18:05:07 -06:00
|
|
|
pat_ident(path, inner) {
|
|
|
|
visit_path(path, e, v);
|
|
|
|
option::may(inner, {|subpat| v.visit_pat(subpat, e, v)});
|
|
|
|
}
|
2011-07-27 07:19:39 -05:00
|
|
|
_ { }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_native_item<E>(ni: @native_item, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt ni.node {
|
2011-12-28 10:50:12 -06:00
|
|
|
native_item_fn(fd, tps) {
|
2012-01-04 09:37:39 -06:00
|
|
|
v.visit_ty_params(tps, e, v);
|
2011-12-28 10:50:12 -06:00
|
|
|
visit_fn_decl(fd, e, v);
|
|
|
|
}
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-12-28 10:50:12 -06:00
|
|
|
fn visit_ty_params<E>(tps: [ty_param], e: E, v: vt<E>) {
|
|
|
|
for tp in tps {
|
|
|
|
for bound in *tp.bounds {
|
|
|
|
alt bound {
|
|
|
|
bound_iface(t) { v.visit_ty(t, e, v); }
|
|
|
|
_ {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_fn_decl<E>(fd: fn_decl, e: E, v: vt<E>) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for a: arg in fd.inputs { v.visit_ty(a.ty, e, v); }
|
|
|
|
for c: @constr in fd.constraints {
|
2011-07-19 19:52:34 -05:00
|
|
|
v.visit_constr(c.node.path, c.span, c.node.id, e, v);
|
|
|
|
}
|
2011-07-01 07:05:54 -05:00
|
|
|
v.visit_ty(fd.output, e, v);
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-12-29 22:07:55 -06:00
|
|
|
fn visit_fn<E>(fk: fn_kind, decl: fn_decl, body: blk, _sp: span,
|
|
|
|
_id: node_id, e: E, v: vt<E>) {
|
2011-12-20 13:03:21 -06:00
|
|
|
visit_fn_decl(decl, e, v);
|
2012-01-04 09:37:39 -06:00
|
|
|
v.visit_ty_params(tps_of_fn(fk), e, v);
|
2011-12-20 13:03:21 -06:00
|
|
|
v.visit_block(body, e, v);
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_block<E>(b: ast::blk, e: E, v: vt<E>) {
|
2011-11-23 13:57:34 -06:00
|
|
|
for vi in b.node.view_items { v.visit_view_item(vi, e, v); }
|
|
|
|
for s in b.node.stmts { v.visit_stmt(s, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
visit_expr_opt(b.node.expr, e, v);
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_stmt<E>(s: @stmt, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt s.node {
|
|
|
|
stmt_decl(d, _) { v.visit_decl(d, e, v); }
|
|
|
|
stmt_expr(ex, _) { v.visit_expr(ex, e, v); }
|
2012-01-04 16:16:41 -06:00
|
|
|
stmt_semi(ex, _) { v.visit_expr(ex, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_decl<E>(d: @decl, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt d.node {
|
|
|
|
decl_local(locs) {
|
2012-02-10 07:33:36 -06:00
|
|
|
for loc in locs { v.visit_local(loc, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
|
|
|
decl_item(it) { v.visit_item(it, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-01-31 19:05:20 -06:00
|
|
|
fn visit_expr_opt<E>(eo: option<@expr>, e: E, v: vt<E>) {
|
2012-01-19 00:37:22 -06:00
|
|
|
alt eo { none { } some(ex) { v.visit_expr(ex, e, v); } }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_exprs<E>(exprs: [@expr], e: E, v: vt<E>) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for ex: @expr in exprs { v.visit_expr(ex, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_mac<E>(m: mac, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt m.node {
|
2012-02-01 01:20:31 -06:00
|
|
|
ast::mac_invoc(pth, arg, body) {
|
|
|
|
option::map(arg) {|arg| visit_expr(arg, e, v)}; }
|
2011-07-27 07:19:39 -05:00
|
|
|
ast::mac_embed_type(ty) { v.visit_ty(ty, e, v); }
|
|
|
|
ast::mac_embed_block(blk) { v.visit_block(blk, e, v); }
|
2012-01-19 00:37:22 -06:00
|
|
|
ast::mac_ellipsis { }
|
2012-01-25 17:38:09 -06:00
|
|
|
ast::mac_aq(_, e) { /* FIXME: maybe visit */ }
|
|
|
|
ast::mac_var(_) { }
|
2011-07-08 18:35:09 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_expr<E>(ex: @expr, e: E, v: vt<E>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
alt ex.node {
|
2011-08-18 13:37:19 -05:00
|
|
|
expr_vec(es, _) { visit_exprs(es, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
expr_rec(flds, base) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for f: field in flds { v.visit_expr(f.node.expr, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
visit_expr_opt(base, e, v);
|
|
|
|
}
|
2011-08-19 17:16:48 -05:00
|
|
|
expr_tup(elts) { for el in elts { v.visit_expr(el, e, v); } }
|
2011-10-21 07:11:24 -05:00
|
|
|
expr_call(callee, args, _) {
|
2011-07-27 07:19:39 -05:00
|
|
|
visit_exprs(args, e, v);
|
2011-08-30 10:03:00 -05:00
|
|
|
v.visit_expr(callee, e, v);
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
|
|
|
expr_bind(callee, args) {
|
|
|
|
v.visit_expr(callee, e, v);
|
2012-01-31 19:05:20 -06:00
|
|
|
for eo: option<@expr> in args { visit_expr_opt(eo, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
|
|
|
expr_binary(_, a, b) { v.visit_expr(a, e, v); v.visit_expr(b, e, v); }
|
|
|
|
expr_unary(_, a) { v.visit_expr(a, 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) {
|
|
|
|
v.visit_expr(x, e, v);
|
|
|
|
v.visit_block(b, e, v);
|
|
|
|
visit_expr_opt(eo, e, v);
|
|
|
|
}
|
|
|
|
expr_if_check(x, b, eo) {
|
|
|
|
v.visit_expr(x, e, v);
|
|
|
|
v.visit_block(b, e, v);
|
|
|
|
visit_expr_opt(eo, e, v);
|
|
|
|
}
|
|
|
|
expr_while(x, b) { v.visit_expr(x, e, v); v.visit_block(b, e, v); }
|
2011-10-21 06:42:26 -05:00
|
|
|
expr_for(dcl, x, b) {
|
2011-07-27 07:19:39 -05:00
|
|
|
v.visit_local(dcl, e, v);
|
|
|
|
v.visit_expr(x, e, v);
|
|
|
|
v.visit_block(b, e, v);
|
|
|
|
}
|
|
|
|
expr_do_while(b, x) { v.visit_block(b, e, v); v.visit_expr(x, e, v); }
|
2012-02-15 02:35:11 -06:00
|
|
|
expr_alt(x, arms, _) {
|
2011-07-27 07:19:39 -05:00
|
|
|
v.visit_expr(x, e, v);
|
2011-08-15 23:54:52 -05:00
|
|
|
for a: arm in arms { v.visit_arm(a, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
}
|
2011-12-29 22:07:55 -06:00
|
|
|
expr_fn(proto, decl, body, _) {
|
|
|
|
v.visit_fn(fk_anon(proto), decl, body, ex.span, ex.id, e, v);
|
2011-12-20 13:03:21 -06:00
|
|
|
}
|
|
|
|
expr_fn_block(decl, body) {
|
2011-12-29 22:07:55 -06:00
|
|
|
v.visit_fn(fk_fn_block, decl, body, ex.span, ex.id, e, v);
|
2011-12-20 13:03:21 -06:00
|
|
|
}
|
2011-07-27 07:19:39 -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); }
|
2011-08-15 17:35:40 -05:00
|
|
|
expr_copy(a) { v.visit_expr(a, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
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) {
|
|
|
|
v.visit_expr(b, e, v);
|
|
|
|
v.visit_expr(a, e, v);
|
|
|
|
}
|
2011-12-19 03:21:31 -06:00
|
|
|
expr_field(x, _, tys) {
|
|
|
|
v.visit_expr(x, e, v);
|
|
|
|
for tp in tys { v.visit_ty(tp, e, v); }
|
|
|
|
}
|
2011-07-27 07:19:39 -05:00
|
|
|
expr_index(a, b) { v.visit_expr(a, e, v); v.visit_expr(b, e, v); }
|
2011-12-13 06:19:56 -06:00
|
|
|
expr_path(p) { visit_path(p, e, v); }
|
2011-07-27 07:19:39 -05:00
|
|
|
expr_fail(eo) { visit_expr_opt(eo, e, v); }
|
2012-01-19 00:37:22 -06:00
|
|
|
expr_break { }
|
|
|
|
expr_cont { }
|
2011-07-27 07:19:39 -05:00
|
|
|
expr_ret(eo) { visit_expr_opt(eo, e, v); }
|
|
|
|
expr_be(x) { v.visit_expr(x, e, v); }
|
2011-12-21 16:31:31 -06:00
|
|
|
expr_log(_, lv, x) {
|
|
|
|
v.visit_expr(lv, e, v);
|
|
|
|
v.visit_expr(x, e, v);
|
|
|
|
}
|
2011-07-27 07:19:39 -05:00
|
|
|
expr_check(_, x) { v.visit_expr(x, e, v); }
|
|
|
|
expr_assert(x) { v.visit_expr(x, e, v); }
|
|
|
|
expr_mac(mac) { visit_mac(mac, e, v); }
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn visit_arm<E>(a: arm, e: E, v: vt<E>) {
|
2011-08-15 23:54:52 -05:00
|
|
|
for p: @pat in a.pats { v.visit_pat(p, e, v); }
|
2011-08-22 07:38:48 -05:00
|
|
|
visit_expr_opt(a.guard, e, v);
|
2011-08-10 17:38:41 -05:00
|
|
|
v.visit_block(a.body, e, v);
|
2011-06-08 15:48:19 -05:00
|
|
|
}
|
2011-07-01 07:05:54 -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 =
|
2011-07-27 07:19:39 -05:00
|
|
|
// takes the components so that one function can be
|
|
|
|
// generic over constr and ty_constr
|
2012-01-26 10:33:12 -06:00
|
|
|
@{visit_mod: fn@(_mod, span, node_id),
|
2011-10-18 17:07:40 -05:00
|
|
|
visit_view_item: fn@(@view_item),
|
|
|
|
visit_native_item: fn@(@native_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_ty: fn@(@ty),
|
2012-01-04 09:37:39 -06:00
|
|
|
visit_ty_params: fn@([ty_param]),
|
2011-11-30 06:38:38 -06:00
|
|
|
visit_constr: fn@(@path, span, node_id),
|
2012-01-31 21:30:40 -06:00
|
|
|
visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id),
|
2012-02-07 13:31:15 -06:00
|
|
|
visit_class_item: fn@(span, privacy, class_member)};
|
2011-07-26 09:35:31 -05:00
|
|
|
|
2011-11-22 03:57:47 -06:00
|
|
|
fn simple_ignore_ty(_t: @ty) {}
|
|
|
|
|
2011-07-26 09:35:31 -05:00
|
|
|
fn default_simple_visitor() -> simple_visitor {
|
2012-01-26 10:33:12 -06:00
|
|
|
ret @{visit_mod: fn@(_m: _mod, _sp: span, _id: node_id) { },
|
2012-01-11 11:58:05 -06:00
|
|
|
visit_view_item: fn@(_vi: @view_item) { },
|
|
|
|
visit_native_item: fn@(_ni: @native_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) { },
|
2011-11-22 03:57:47 -06:00
|
|
|
visit_ty: simple_ignore_ty,
|
2012-01-11 11:58:05 -06:00
|
|
|
visit_ty_params: fn@(_ps: [ty_param]) {},
|
|
|
|
visit_constr: fn@(_p: @path, _sp: span, _id: node_id) { },
|
|
|
|
visit_fn: fn@(_fk: fn_kind, _d: fn_decl, _b: blk, _sp: span,
|
2012-01-31 21:30:40 -06:00
|
|
|
_id: node_id) { },
|
2012-02-07 13:31:15 -06:00
|
|
|
visit_class_item: fn@(_s: span, _p: privacy, _c: class_member) {}
|
2011-12-20 13:03:21 -06:00
|
|
|
};
|
2011-07-26 09:35:31 -05:00
|
|
|
}
|
|
|
|
|
2011-09-12 04:27:30 -05:00
|
|
|
fn mk_simple_visitor(v: simple_visitor) -> vt<()> {
|
2012-01-26 10:33:12 -06:00
|
|
|
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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -05:00
|
|
|
fn v_native_item(f: fn@(@native_item), ni: @native_item, &&e: (),
|
2011-09-12 04:27:30 -05:00
|
|
|
v: vt<()>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
f(ni);
|
|
|
|
visit_native_item(ni, e, v);
|
2011-07-26 09:35:31 -05:00
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2011-10-18 17:07:40 -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
|
|
|
}
|
2012-01-04 09:37:39 -06:00
|
|
|
fn v_ty_params(f: fn@([ty_param]), ps: [ty_param], &&e: (), v: vt<()>) {
|
|
|
|
f(ps);
|
|
|
|
visit_ty_params(ps, e, v);
|
|
|
|
}
|
2011-11-30 06:38:38 -06:00
|
|
|
fn v_constr(f: fn@(@path, span, node_id), pt: @path, sp: span,
|
|
|
|
id: node_id, &&e: (), v: vt<()>) {
|
2011-07-27 07:19:39 -05:00
|
|
|
f(pt, sp, id);
|
|
|
|
visit_constr(pt, sp, id, e, v);
|
2011-07-26 09:35:31 -05:00
|
|
|
}
|
2011-12-29 22:07:55 -06:00
|
|
|
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
|
|
|
}
|
2011-11-22 03:57:47 -06:00
|
|
|
let visit_ty = if v.visit_ty == simple_ignore_ty {
|
|
|
|
bind skip_ty(_, _, _)
|
|
|
|
} else {
|
|
|
|
bind v_ty(v.visit_ty, _, _, _)
|
|
|
|
};
|
2012-02-07 13:31:15 -06:00
|
|
|
fn v_class_item(f: fn@(span, privacy, class_member),
|
|
|
|
s:span, p:privacy, cm: class_member, &&e: (),
|
2012-01-31 21:30:40 -06:00
|
|
|
v: vt<()>) {
|
|
|
|
f(s, p, cm);
|
|
|
|
visit_class_item(s, p, cm, e, v);
|
|
|
|
}
|
2012-01-26 10:33:12 -06:00
|
|
|
ret mk_vt(@{visit_mod: bind v_mod(v.visit_mod, _, _, _, _, _),
|
2011-07-27 07:19:39 -05:00
|
|
|
visit_view_item: bind v_view_item(v.visit_view_item, _, _, _),
|
|
|
|
visit_native_item:
|
|
|
|
bind v_native_item(v.visit_native_item, _, _, _),
|
|
|
|
visit_item: bind v_item(v.visit_item, _, _, _),
|
|
|
|
visit_local: bind v_local(v.visit_local, _, _, _),
|
|
|
|
visit_block: bind v_block(v.visit_block, _, _, _),
|
|
|
|
visit_stmt: bind v_stmt(v.visit_stmt, _, _, _),
|
|
|
|
visit_arm: bind v_arm(v.visit_arm, _, _, _),
|
|
|
|
visit_pat: bind v_pat(v.visit_pat, _, _, _),
|
|
|
|
visit_decl: bind v_decl(v.visit_decl, _, _, _),
|
|
|
|
visit_expr: bind v_expr(v.visit_expr, _, _, _),
|
2011-11-22 03:57:47 -06:00
|
|
|
visit_ty: visit_ty,
|
2012-01-04 09:37:39 -06:00
|
|
|
visit_ty_params: bind v_ty_params(v.visit_ty_params, _, _, _),
|
2011-07-27 07:19:39 -05:00
|
|
|
visit_constr: bind v_constr(v.visit_constr, _, _, _, _, _),
|
2012-01-31 21:30:40 -06:00
|
|
|
visit_fn: bind v_fn(v.visit_fn, _, _, _, _, _, _, _),
|
|
|
|
visit_class_item: bind v_class_item(v.visit_class_item, _, _,
|
|
|
|
_, _, _)
|
2011-12-20 13:03:21 -06:00
|
|
|
});
|
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:
|