81 lines
2.4 KiB
Rust
81 lines
2.4 KiB
Rust
|
|
import core::{int, uint};
|
|
import syntax::ast::*;
|
|
import syntax::ast_util::pat_binding_ids;
|
|
import syntax::visit;
|
|
import syntax::codemap::span;
|
|
import util::common::{log_stmt};
|
|
import aux::{num_constraints, get_fn_info, crate_ctxt, add_node};
|
|
import middle::tstate::ann::empty_ann;
|
|
|
|
fn collect_ids_expr(e: @expr, rs: @mutable [node_id]) { *rs += [e.id]; }
|
|
|
|
fn collect_ids_block(b: blk, rs: @mutable [node_id]) { *rs += [b.node.id]; }
|
|
|
|
fn collect_ids_stmt(s: @stmt, rs: @mutable [node_id]) {
|
|
alt s.node {
|
|
stmt_decl(_, id) {
|
|
log(debug, "node_id " + int::str(id));
|
|
log_stmt(*s);;
|
|
*rs += [id];
|
|
}
|
|
stmt_expr(_, id) {
|
|
log(debug, "node_id " + int::str(id));
|
|
log_stmt(*s);;
|
|
*rs += [id];
|
|
}
|
|
_ { }
|
|
}
|
|
}
|
|
|
|
fn collect_ids_local(l: @local, rs: @mutable [node_id]) {
|
|
*rs += pat_binding_ids(l.node.pat);
|
|
}
|
|
|
|
fn node_ids_in_fn(body: blk, rs: @mutable [node_id]) {
|
|
let collect_ids =
|
|
visit::mk_simple_visitor(@{visit_expr: bind collect_ids_expr(_, rs),
|
|
visit_block: bind collect_ids_block(_, rs),
|
|
visit_stmt: bind collect_ids_stmt(_, rs),
|
|
visit_local: bind collect_ids_local(_, rs)
|
|
with *visit::default_simple_visitor()});
|
|
collect_ids.visit_block(body, (), collect_ids);
|
|
}
|
|
|
|
fn init_vecs(ccx: crate_ctxt, node_ids: [node_id], len: uint) {
|
|
for i: node_id in node_ids {
|
|
log(debug, int::str(i) + " |-> " + uint::str(len));
|
|
add_node(ccx, i, empty_ann(len));
|
|
}
|
|
}
|
|
|
|
fn visit_fn(ccx: crate_ctxt, num_constraints: uint, body: blk) {
|
|
let node_ids: @mutable [node_id] = @mutable [];
|
|
node_ids_in_fn(body, node_ids);
|
|
let node_id_vec = *node_ids;
|
|
init_vecs(ccx, node_id_vec, num_constraints);
|
|
}
|
|
|
|
fn annotate_in_fn(ccx: crate_ctxt, _fk: visit::fn_kind, _decl: fn_decl,
|
|
body: blk, _sp: span, id: node_id) {
|
|
let f_info = get_fn_info(ccx, id);
|
|
visit_fn(ccx, num_constraints(f_info), body);
|
|
}
|
|
|
|
fn annotate_crate(ccx: crate_ctxt, crate: crate) {
|
|
let do_ann =
|
|
visit::mk_simple_visitor(
|
|
@{visit_fn: bind annotate_in_fn(ccx, _, _, _, _, _)
|
|
with *visit::default_simple_visitor()});
|
|
visit::visit_crate(crate, (), do_ann);
|
|
}
|
|
//
|
|
// Local Variables:
|
|
// mode: rust
|
|
// fill-column: 78;
|
|
// indent-tabs-mode: nil
|
|
// c-basic-offset: 4
|
|
// buffer-file-coding-system: utf-8-unix
|
|
// End:
|
|
//
|