visit.rs 22.4 KB
Newer Older
1

2 3
import ast::*;
import codemap::span;
M
Marijn Haverbeke 已提交
4

M
Marijn Haverbeke 已提交
5 6 7 8 9 10
// 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.

11
// Our typesystem doesn't do circular types, so the visitor record can not
P
Patrick Walton 已提交
12
// hold functions that take visitors. A vt enum is used to break the cycle.
P
Patrick Walton 已提交
13
enum vt<E> { mk_vt(visitor<E>), }
M
Marijn Haverbeke 已提交
14

P
Patrick Walton 已提交
15
enum fn_kind {
16 17
    fk_item_fn(ident, ~[ty_param]), //< an item declared with fn()
    fk_method(ident, ~[ty_param], @method),
18 19
    fk_anon(proto, capture_clause),  //< an anonymous function like fn@(...)
    fk_fn_block(capture_clause),     //< a block {||...}
20
    fk_ctor(ident, ~[attribute], ~[ty_param], node_id /* self id */,
T
Tim Chevalier 已提交
21
            def_id /* parent class id */), // class constructor
22
    fk_dtor(~[ty_param], ~[attribute], node_id /* self id */,
T
Tim Chevalier 已提交
23 24
            def_id /* parent class id */) // class destructor

25 26 27
}

fn name_of_fn(fk: fn_kind) -> ident {
28
    match fk {
T
Tim Chevalier 已提交
29
      fk_item_fn(name, _) | fk_method(name, _, _)
B
Brian Anderson 已提交
30 31 32
          | fk_ctor(name, _, _, _, _) => /* FIXME (#2543) */ copy name,
      fk_anon(*) | fk_fn_block(*) => @~"anon",
      fk_dtor(*)                  => @~"drop"
33 34 35
    }
}

36
fn tps_of_fn(fk: fn_kind) -> ~[ty_param] {
37
    match fk {
T
Tim Chevalier 已提交
38
      fk_item_fn(_, tps) | fk_method(_, tps, _)
B
Brian Anderson 已提交
39
          | fk_ctor(_, _, tps, _, _) | fk_dtor(tps, _, _, _) => {
40
          /* FIXME (#2543) */ copy tps
41
      }
B
Brian Anderson 已提交
42
      fk_anon(*) | fk_fn_block(*) => ~[]
43 44 45
    }
}

46
type visitor<E> =
47
    @{visit_mod: fn@(_mod, span, node_id, E, vt<E>),
48
      visit_view_item: fn@(@view_item, E, vt<E>),
49
      visit_foreign_item: fn@(@foreign_item, E, vt<E>),
50 51 52 53 54 55 56 57
      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>),
58
      visit_expr_post: fn@(@expr, E, vt<E>),
59
      visit_ty: fn@(@ty, E, vt<E>),
60
      visit_ty_params: fn@(~[ty_param], E, vt<E>),
61
      visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id, E, vt<E>),
62
      visit_ty_method: fn@(ty_method, E, vt<E>),
63
      visit_trait_method: fn@(trait_method, E, vt<E>),
P
Patrick Walton 已提交
64
      visit_struct_def: fn@(@struct_def, ident, ~[ty_param], node_id, E,
65
                            vt<E>),
66
      visit_class_item: fn@(@class_member, E, vt<E>)};
M
Marijn Haverbeke 已提交
67

68
fn default_visitor<E>() -> visitor<E> {
B
Brian Anderson 已提交
69
    return @{visit_mod: |a,b,c,d,e|visit_mod::<E>(a, b, c, d, e),
B
Brian Anderson 已提交
70 71 72 73 74 75 76 77 78 79
          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),
80
          visit_expr_post: |_a,_b,_c| (),
B
Brian Anderson 已提交
81 82 83
          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),
84
          visit_ty_method: |a,b,c|visit_ty_method::<E>(a, b, c),
85
          visit_trait_method: |a,b,c|visit_trait_method::<E>(a, b, c),
P
Patrick Walton 已提交
86 87
          visit_struct_def: |a,b,c,d,e,f|visit_struct_def::<E>(a, b, c,
                                                               d, e, f),
B
Brian Anderson 已提交
88
          visit_class_item: |a,b,c|visit_class_item::<E>(a, b, c)};
M
Marijn Haverbeke 已提交
89 90
}

91
fn visit_crate<E>(c: crate, e: E, v: vt<E>) {
92
    v.visit_mod(c.node.module, c.span, crate_node_id, e, v);
M
Marijn Haverbeke 已提交
93 94
}

95
fn visit_crate_directive<E>(cd: @crate_directive, e: E, v: vt<E>) {
96
    match cd.node {
B
Brian Anderson 已提交
97 98 99
      cdir_src_mod(_, _) => (),
      cdir_dir_mod(_, cdirs, _) => for cdirs.each |cdir| {
        visit_crate_directive(cdir, e, v);
100
      },
B
Brian Anderson 已提交
101 102
      cdir_view_item(vi) => v.visit_view_item(vi, e, v),
      cdir_syntax(_) => ()
103 104 105
    }
}

106
fn visit_mod<E>(m: _mod, _sp: span, _id: node_id, e: E, v: vt<E>) {
B
Brian Anderson 已提交
107 108
    for m.view_items.each |vi| { v.visit_view_item(vi, e, v); }
    for m.items.each |i| { v.visit_item(i, e, v); }
M
Marijn Haverbeke 已提交
109 110
}

111
fn visit_view_item<E>(_vi: @view_item, _e: E, _v: vt<E>) { }
112

113
fn visit_local<E>(loc: @local, e: E, v: vt<E>) {
114
    v.visit_pat(loc.node.pat, e, v);
115
    v.visit_ty(loc.node.ty, e, v);
116
    match loc.node.init {
B
Brian Anderson 已提交
117 118 119
      none => (),
      some(i) => v.visit_expr(i.expr, e, v)
    }
120 121
}

122
fn visit_item<E>(i: @item, e: E, v: vt<E>) {
123
    match i.node {
B
Brian Anderson 已提交
124 125
      item_const(t, ex) => { v.visit_ty(t, e, v); v.visit_expr(ex, e, v); }
      item_fn(decl, tp, body) => {
126 127
        v.visit_fn(fk_item_fn(/* FIXME (#2543) */ copy i.ident,
                              /* FIXME (#2543) */ copy tp), decl, body,
128
                   i.span, i.id, e, v);
129
      }
B
Brian Anderson 已提交
130 131
      item_mod(m) => v.visit_mod(m, i.span, i.id, e, v),
      item_foreign_mod(nm) => {
B
Brian Anderson 已提交
132 133
        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); }
M
Marijn Haverbeke 已提交
134
      }
B
Brian Anderson 已提交
135
      item_ty(t, tps) => {
136 137 138
        v.visit_ty(t, e, v);
        v.visit_ty_params(tps, e, v);
      }
B
Brian Anderson 已提交
139
      item_enum(variants, tps) => {
140
        v.visit_ty_params(tps, e, v);
B
Brian Anderson 已提交
141
        for variants.each |vr| {
142
            match vr.node.kind {
143 144 145 146 147 148 149
                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);
                }
150
            }
M
Marijn Haverbeke 已提交
151 152
        }
      }
B
Brian Anderson 已提交
153
      item_impl(tps, traits, ty, methods) => {
154
        v.visit_ty_params(tps, e, v);
155 156 157
        for traits.each |p| {
            visit_path(p.path, e, v);
        }
158
        v.visit_ty(ty, e, v);
B
Brian Anderson 已提交
159
        for methods.each |m| {
160
            visit_method_helper(m, e, v)
161 162
        }
      }
163
      item_class(struct_def, tps) => {
164 165
        v.visit_ty_params(tps, e, v);
        v.visit_struct_def(struct_def, i.ident, tps, i.id, e, v);
166
      }
B
Brian Anderson 已提交
167
      item_trait(tps, traits, methods) => {
168
        v.visit_ty_params(tps, e, v);
169
        for traits.each |p| { visit_path(p.path, e, v); }
B
Brian Anderson 已提交
170
        for methods.each |m| {
171
            v.visit_trait_method(m, e, v);
172 173
        }
      }
B
Brian Anderson 已提交
174
      item_mac(m) => visit_mac(m, e, v)
M
Marijn Haverbeke 已提交
175 176 177
    }
}

178
fn visit_class_item<E>(cm: @class_member, e:E, v:vt<E>) {
179
    match cm.node {
B
Brian Anderson 已提交
180 181
      instance_var(_, t, _, _, _) => v.visit_ty(t, e, v),
      class_method(m) => visit_method_helper(m, e, v)
182 183 184
    }
}

185 186
fn skip_ty<E>(_t: @ty, _e: E, _v: vt<E>) {}

187
fn visit_ty<E>(t: @ty, e: E, v: vt<E>) {
188
    match t.node {
189
      ty_box(mt) | ty_uniq(mt) |
B
Brian Anderson 已提交
190
      ty_vec(mt) | ty_ptr(mt) | ty_rptr(_, mt) => {
191 192
        v.visit_ty(mt.ty, e, v);
      }
B
Brian Anderson 已提交
193 194
      ty_rec(flds) => for flds.each |f| {
        v.visit_ty(f.node.mt.ty, e, v);
195
      },
B
Brian Anderson 已提交
196 197
      ty_tup(ts) => for ts.each |tt| {
        v.visit_ty(tt, e, v);
198
      },
199
      ty_fn(_, bounds, decl) => {
B
Brian Anderson 已提交
200
        for decl.inputs.each |a| { v.visit_ty(a.ty, e, v); }
201
        visit_ty_param_bounds(bounds, e, v);
202
        v.visit_ty(decl.output, e, v);
M
Marijn Haverbeke 已提交
203
      }
B
Brian Anderson 已提交
204 205
      ty_path(p, _) => visit_path(p, e, v),
      ty_fixed_length(t, _) => v.visit_ty(t, e, v),
206 207 208
      ty_nil |
      ty_bot |
      ty_mac(_) |
B
Brian Anderson 已提交
209
      ty_infer => ()
M
Marijn Haverbeke 已提交
210 211 212
    }
}

213
fn visit_path<E>(p: @path, e: E, v: vt<E>) {
B
Brian Anderson 已提交
214
    for p.types.each |tp| { v.visit_ty(tp, e, v); }
215 216
}

217
fn visit_pat<E>(p: @pat, e: E, v: vt<E>) {
218
    match p.node {
B
Brian Anderson 已提交
219
      pat_enum(path, children) => {
220
        visit_path(path, e, v);
B
Brian Anderson 已提交
221 222
        do option::iter(children) |children| {
            for children.each |child| { v.visit_pat(child, e, v); }}
M
Marijn Haverbeke 已提交
223
      }
B
Brian Anderson 已提交
224 225
      pat_rec(fields, _) => for fields.each |f| {
        v.visit_pat(f.pat, e, v)
226
      },
227 228 229 230 231 232
      pat_struct(path, fields, _) => {
        visit_path(path, e, v);
        for fields.each |f| {
            v.visit_pat(f.pat, e, v);
        }
      }
B
Brian Anderson 已提交
233 234
      pat_tup(elts) => for elts.each |elt| {
        v.visit_pat(elt, e, v)
235
      },
B
Brian Anderson 已提交
236 237
      pat_box(inner) | pat_uniq(inner) => v.visit_pat(inner, e, v),
      pat_ident(_, path, inner) => {
238
          visit_path(path, e, v);
B
Brian Anderson 已提交
239
          do option::iter(inner) |subpat| { v.visit_pat(subpat, e, v)};
240
      }
B
Brian Anderson 已提交
241 242 243
      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 => ()
M
Marijn Haverbeke 已提交
244 245 246
    }
}

247
fn visit_foreign_item<E>(ni: @foreign_item, e: E, v: vt<E>) {
248
    match ni.node {
B
Brian Anderson 已提交
249
      foreign_item_fn(fd, tps) => {
250
        v.visit_ty_params(tps, e, v);
251 252
        visit_fn_decl(fd, e, v);
      }
M
Marijn Haverbeke 已提交
253 254 255
    }
}

256 257 258 259 260 261 262 263 264
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 => ()
        }
    }
}

265
fn visit_ty_params<E>(tps: ~[ty_param], e: E, v: vt<E>) {
B
Brian Anderson 已提交
266
    for tps.each |tp| {
267
        visit_ty_param_bounds(tp.bounds, e, v);
268 269 270
    }
}

271
fn visit_fn_decl<E>(fd: fn_decl, e: E, v: vt<E>) {
B
Brian Anderson 已提交
272
    for fd.inputs.each |a| { v.visit_ty(a.ty, e, v); }
273
    v.visit_ty(fd.output, e, v);
M
Marijn Haverbeke 已提交
274 275
}

276 277 278 279 280
// 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>) {
281 282
    v.visit_fn(fk_method(/* FIXME (#2543) */ copy m.ident,
                         /* FIXME (#2543) */ copy m.tps, m),
283
               m.decl, m.body, m.span, m.id, e, v);
284 285
}

286
// Similar logic to the comment on visit_method_helper - Tim
287
fn visit_class_ctor_helper<E>(ctor: class_ctor, nm: ident, tps: ~[ty_param],
288
                              parent_id: def_id, e: E, v: vt<E>) {
289
    v.visit_fn(fk_ctor(/* FIXME (#2543) */ copy nm,
290
                       ctor.node.attrs,
291
                       /* FIXME (#2543) */ copy tps,
292 293
                       ctor.node.self_id, parent_id),
        ctor.node.dec, ctor.node.body, ctor.span, ctor.node.id, e, v)
294 295 296

}

297
fn visit_class_dtor_helper<E>(dtor: class_dtor, tps: ~[ty_param],
T
Tim Chevalier 已提交
298
                              parent_id: def_id, e: E, v: vt<E>) {
299 300
    v.visit_fn(fk_dtor(/* FIXME (#2543) */ copy tps, dtor.node.attrs,
                       dtor.node.self_id, parent_id), ast_util::dtor_dec(),
T
Tim Chevalier 已提交
301 302 303 304
               dtor.node.body, dtor.span, dtor.node.id, e, v)

}

305 306
fn visit_fn<E>(fk: fn_kind, decl: fn_decl, body: blk, _sp: span,
               _id: node_id, e: E, v: vt<E>) {
307
    visit_fn_decl(decl, e, v);
308
    v.visit_ty_params(tps_of_fn(fk), e, v);
309
    v.visit_block(body, e, v);
M
Marijn Haverbeke 已提交
310 311
}

312 313 314 315 316 317
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);
}

318
fn visit_trait_method<E>(m: trait_method, e: E, v: vt<E>) {
319
    match m {
B
Brian Anderson 已提交
320 321
      required(ty_m) => v.visit_ty_method(ty_m, e, v),
      provided(m) => visit_method_helper(m, e, v)
322 323 324
    }
}

P
Patrick Walton 已提交
325
fn visit_struct_def<E>(sd: @struct_def, nm: ast::ident, tps: ~[ty_param],
326 327 328 329 330 331 332 333 334 335 336 337 338
                       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)
    };
}

339
fn visit_block<E>(b: ast::blk, e: E, v: vt<E>) {
B
Brian Anderson 已提交
340 341
    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); }
M
Marijn Haverbeke 已提交
342 343 344
    visit_expr_opt(b.node.expr, e, v);
}

345
fn visit_stmt<E>(s: @stmt, e: E, v: vt<E>) {
346
    match s.node {
B
Brian Anderson 已提交
347 348 349
      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)
M
Marijn Haverbeke 已提交
350 351 352
    }
}

353
fn visit_decl<E>(d: @decl, e: E, v: vt<E>) {
354
    match d.node {
B
Brian Anderson 已提交
355 356
      decl_local(locs) => for locs.each |loc| {
        v.visit_local(loc, e, v)
357
      },
B
Brian Anderson 已提交
358
      decl_item(it) => v.visit_item(it, e, v)
M
Marijn Haverbeke 已提交
359 360 361
    }
}

T
Tim Chevalier 已提交
362
fn visit_expr_opt<E>(eo: option<@expr>, e: E, v: vt<E>) {
363
    match eo { none => (), some(ex) => v.visit_expr(ex, e, v) }
M
Marijn Haverbeke 已提交
364 365
}

366
fn visit_exprs<E>(exprs: ~[@expr], e: E, v: vt<E>) {
B
Brian Anderson 已提交
367
    for exprs.each |ex| { v.visit_expr(ex, e, v); }
M
Marijn Haverbeke 已提交
368 369
}

370
fn visit_mac<E>(m: mac, e: E, v: vt<E>) {
371
    match m.node {
B
Brian Anderson 已提交
372
      ast::mac_invoc(pth, arg, body) => {
B
Brian Anderson 已提交
373
        option::map(arg, |arg| v.visit_expr(arg, e, v)); }
B
Brian Anderson 已提交
374 375 376 377
      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(_) => ()
378 379 380
    }
}

381
fn visit_expr<E>(ex: @expr, e: E, v: vt<E>) {
382
    match ex.node {
B
Brian Anderson 已提交
383 384
      expr_vstore(x, _) => v.visit_expr(x, e, v),
      expr_vec(es, _) => visit_exprs(es, e, v),
385 386 387 388
      expr_repeat(element, count, _) => {
        v.visit_expr(element, e, v);
        v.visit_expr(count, e, v);
      }
B
Brian Anderson 已提交
389
      expr_rec(flds, base) => {
B
Brian Anderson 已提交
390
        for flds.each |f| { v.visit_expr(f.node.expr, e, v); }
M
Marijn Haverbeke 已提交
391 392
        visit_expr_opt(base, e, v);
      }
393
      expr_struct(p, flds, base) => {
394 395
        visit_path(p, e, v);
        for flds.each |f| { v.visit_expr(f.node.expr, e, v); }
396
        visit_expr_opt(base, e, v);
397
      }
398
      expr_tup(elts) => for elts.each |el| { v.visit_expr(el, e, v); },
B
Brian Anderson 已提交
399
      expr_call(callee, args, _) => {
M
Marijn Haverbeke 已提交
400
        visit_exprs(args, e, v);
401
        v.visit_expr(callee, e, v);
M
Marijn Haverbeke 已提交
402
      }
B
Brian Anderson 已提交
403 404 405
      expr_binary(_, a, b) => {
        v.visit_expr(a, e, v); v.visit_expr(b, e, v);
      }
B
Brian Anderson 已提交
406 407
      expr_addr_of(_, x) | expr_unary(_, x) |
      expr_loop_body(x) | expr_do_body(x) |
B
Brian Anderson 已提交
408 409 410 411
      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) => {
M
Marijn Haverbeke 已提交
412 413 414 415
        v.visit_expr(x, e, v);
        v.visit_block(b, e, v);
        visit_expr_opt(eo, e, v);
      }
B
Brian Anderson 已提交
416 417
      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),
418
      expr_match(x, arms, _) => {
M
Marijn Haverbeke 已提交
419
        v.visit_expr(x, e, v);
B
Brian Anderson 已提交
420
        for arms.each |a| { v.visit_arm(a, e, v); }
M
Marijn Haverbeke 已提交
421
      }
B
Brian Anderson 已提交
422
      expr_fn(proto, decl, body, cap_clause) => {
423 424
        v.visit_fn(fk_anon(proto, cap_clause), decl, body,
                   ex.span, ex.id, e, v);
425
      }
B
Brian Anderson 已提交
426
      expr_fn_block(decl, body, cap_clause) => {
427 428
        v.visit_fn(fk_fn_block(cap_clause), decl, body,
                   ex.span, ex.id, e, v);
429
      }
B
Brian Anderson 已提交
430 431 432 433 434 435 436
      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) => {
M
Marijn Haverbeke 已提交
437 438 439
        v.visit_expr(b, e, v);
        v.visit_expr(a, e, v);
      }
B
Brian Anderson 已提交
440
      expr_field(x, _, tys) => {
441
        v.visit_expr(x, e, v);
B
Brian Anderson 已提交
442
        for tys.each |tp| { v.visit_ty(tp, e, v); }
443
      }
B
Brian Anderson 已提交
444 445 446 447 448 449 450
      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) => {
451 452 453
        v.visit_expr(lv, e, v);
        v.visit_expr(x, e, v);
      }
B
Brian Anderson 已提交
454
      expr_mac(mac) => visit_mac(mac, e, v),
M
Marijn Haverbeke 已提交
455
    }
456
    v.visit_expr_post(ex, e, v);
M
Marijn Haverbeke 已提交
457 458
}

459
fn visit_arm<E>(a: arm, e: E, v: vt<E>) {
B
Brian Anderson 已提交
460
    for a.pats.each |p| { v.visit_pat(p, e, v); }
M
Marijn Haverbeke 已提交
461
    visit_expr_opt(a.guard, e, v);
462
    v.visit_block(a.body, e, v);
M
Marijn Haverbeke 已提交
463
}
464

465 466 467 468
// Simpler, non-context passing interface. Always walks the whole tree, simply
// calls the given functions on the nodes.

type simple_visitor =
469
    @{visit_mod: fn@(_mod, span, node_id),
470
      visit_view_item: fn@(@view_item),
471
      visit_foreign_item: fn@(@foreign_item),
472 473 474 475 476 477 478 479
      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),
480
      visit_expr_post: fn@(@expr),
481
      visit_ty: fn@(@ty),
482
      visit_ty_params: fn@(~[ty_param]),
483
      visit_fn: fn@(fn_kind, fn_decl, blk, span, node_id),
484
      visit_ty_method: fn@(ty_method),
485
      visit_trait_method: fn@(trait_method),
P
Patrick Walton 已提交
486
      visit_struct_def: fn@(@struct_def, ident, ~[ty_param], node_id),
487
      visit_class_item: fn@(@class_member)};
488

489 490
fn simple_ignore_ty(_t: @ty) {}

491
fn default_simple_visitor() -> simple_visitor {
B
Brian Anderson 已提交
492
    return @{visit_mod: fn@(_m: _mod, _sp: span, _id: node_id) { },
493
          visit_view_item: fn@(_vi: @view_item) { },
494
          visit_foreign_item: fn@(_ni: @foreign_item) { },
495 496 497 498 499 500 501 502
          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) { },
503
          visit_expr_post: fn@(_e: @expr) { },
504
          visit_ty: simple_ignore_ty,
505
          visit_ty_params: fn@(_ps: ~[ty_param]) {},
506
          visit_fn: fn@(_fk: fn_kind, _d: fn_decl, _b: blk, _sp: span,
507
                        _id: node_id) { },
508
          visit_ty_method: fn@(_m: ty_method) { },
509
          visit_trait_method: fn@(_m: trait_method) { },
P
Patrick Walton 已提交
510
          visit_struct_def: fn@(_sd: @struct_def, _nm: ident,
511
                                _tps: ~[ty_param], _id: node_id) { },
512
          visit_class_item: fn@(_c: @class_member) {}
513
         };
514 515
}

516
fn mk_simple_visitor(v: simple_visitor) -> vt<()> {
517 518 519 520
    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);
521
    }
522
    fn v_view_item(f: fn@(@view_item), vi: @view_item, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
523 524
        f(vi);
        visit_view_item(vi, e, v);
525
    }
526
    fn v_foreign_item(f: fn@(@foreign_item), ni: @foreign_item, &&e: (),
527
                     v: vt<()>) {
M
Marijn Haverbeke 已提交
528
        f(ni);
529
        visit_foreign_item(ni, e, v);
530
    }
531
    fn v_item(f: fn@(@item), i: @item, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
532 533
        f(i);
        visit_item(i, e, v);
534
    }
535
    fn v_local(f: fn@(@local), l: @local, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
536 537
        f(l);
        visit_local(l, e, v);
538
    }
539
    fn v_block(f: fn@(ast::blk), bl: ast::blk, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
540 541
        f(bl);
        visit_block(bl, e, v);
542
    }
543
    fn v_stmt(f: fn@(@stmt), st: @stmt, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
544 545
        f(st);
        visit_stmt(st, e, v);
546
    }
547
    fn v_arm(f: fn@(arm), a: arm, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
548 549
        f(a);
        visit_arm(a, e, v);
550
    }
551
    fn v_pat(f: fn@(@pat), p: @pat, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
552 553
        f(p);
        visit_pat(p, e, v);
554
    }
555
    fn v_decl(f: fn@(@decl), d: @decl, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
556 557
        f(d);
        visit_decl(d, e, v);
558
    }
559
    fn v_expr(f: fn@(@expr), ex: @expr, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
560 561
        f(ex);
        visit_expr(ex, e, v);
562
    }
563 564 565
    fn v_expr_post(f: fn@(@expr), ex: @expr, &&_e: (), _v: vt<()>) {
        f(ex);
    }
566
    fn v_ty(f: fn@(@ty), ty: @ty, &&e: (), v: vt<()>) {
M
Marijn Haverbeke 已提交
567 568
        f(ty);
        visit_ty(ty, e, v);
569
    }
570 571 572 573
    fn v_ty_method(f: fn@(ty_method), ty: ty_method, &&e: (), v: vt<()>) {
        f(ty);
        visit_ty_method(ty, e, v);
    }
574 575 576 577 578
    fn v_trait_method(f: fn@(trait_method), m: trait_method, &&e: (),
                      v: vt<()>) {
        f(m);
        visit_trait_method(m, e, v);
    }
P
Patrick Walton 已提交
579 580
    fn v_struct_def(f: fn@(@struct_def, ident, ~[ty_param], node_id),
                    sd: @struct_def, nm: ident, tps: ~[ty_param], id: node_id,
581 582 583 584
                    &&e: (), v: vt<()>) {
        f(sd, nm, tps, id);
        visit_struct_def(sd, nm, tps, id, e, v);
    }
585 586
    fn v_ty_params(f: fn@(~[ty_param]),
                   ps: ~[ty_param],
587
                   &&e: (), v: vt<()>) {
588 589 590
        f(ps);
        visit_ty_params(ps, e, v);
    }
591 592 593 594 595
    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);
596
    }
597
    let visit_ty = if v.visit_ty == simple_ignore_ty {
B
Brian Anderson 已提交
598
        |a,b,c| skip_ty(a, b, c)
599
    } else {
B
Brian Anderson 已提交
600
        |a,b,c| v_ty(v.visit_ty, a, b, c)
601
    };
602 603
    fn v_class_item(f: fn@(@class_member),
                    cm: @class_member, &&e: (),
604
                    v: vt<()>) {
605 606
        f(cm);
        visit_class_item(cm, e, v);
607
    }
B
Brian Anderson 已提交
608
    return mk_vt(@{visit_mod: |a,b,c,d,e|v_mod(v.visit_mod, a, b, c, d, e),
B
Brian Anderson 已提交
609 610
                visit_view_item: |a,b,c|
                    v_view_item(v.visit_view_item, a, b, c),
611
                visit_foreign_item:
B
Brian Anderson 已提交
612 613 614 615 616 617 618 619 620
                    |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),
621 622
                visit_expr_post: |a,b,c| v_expr_post(v.visit_expr_post,
                                                     a, b, c),
623
                visit_ty: visit_ty,
B
Brian Anderson 已提交
624 625 626 627
                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),
628 629
                visit_ty_method: |a,b,c|
                    v_ty_method(v.visit_ty_method, a, b, c),
630 631
                visit_trait_method: |a,b,c|
                    v_trait_method(v.visit_trait_method, a, b, c),
632 633
                visit_struct_def: |a,b,c,d,e,f|
                    v_struct_def(v.visit_struct_def, a, b, c, d, e, f),
B
Brian Anderson 已提交
634
                visit_class_item: |a,b,c|
B
Brian Anderson 已提交
635
                    v_class_item(v.visit_class_item, a, b, c)
636
               });
637 638
}

M
Marijn Haverbeke 已提交
639 640 641 642 643 644 645
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End: