rust/src/comp/middle/tstate/annotate.rs

import std::option;
import std::option::some;
import std::option::none;
import std::int;
import std::uint;
import syntax::ast::*;
import syntax::visit;
import syntax::codemap::span;
import std::map::new_str_hash;
import util::common::log_expr_err;
import util::common::log_block_err;
import util::common::log_item_err;
import util::common::log_stmt_err;
import util::common::log_expr;
import util::common::log_block;
import util::common::log_stmt;
import aux::fn_info;
import aux::fn_info_map;
import aux::num_constraints;
import aux::get_fn_info;
import aux::crate_ctxt;
import aux::add_node;
import middle::tstate::ann::empty_ann;

fn collect_ids_expr(&@expr e, @mutable node_id[] rs) {
    *rs += ~[e.id];
}

fn collect_ids_block(&blk b, @mutable node_id[] rs) {
    *rs += ~[b.node.id];
}

fn collect_ids_stmt(&@stmt s, @mutable node_id[] rs) {
    alt (s.node) {
        case (stmt_decl(_, ?id)) {
            log "node_id " + int::str(id);
            log_stmt(*s);
            *rs += ~[id];
        }
        case (stmt_expr(_, ?id)) {
            log "node_id " + int::str(id);
            log_stmt(*s);
            *rs += ~[id];
        }
        case (_) { }
    }
}

fn collect_ids_local(&@local l, @mutable node_id[] rs) {
    *rs += ~[l.node.id];
}

fn node_ids_in_fn(&_fn f, &ty_param[] tps, &span sp, &fn_ident i,
                  node_id id, @mutable node_id[] rs) {
    auto collect_ids = visit::mk_simple_visitor
        (@rec(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()));
    visit::visit_fn(f, tps, sp, i, id, (), collect_ids);
}

fn init_vecs(&crate_ctxt ccx, &node_id[] node_ids, uint len) {
    for (node_id i in node_ids) {
        log int::str(i) + " |-> " + uint::str(len);
        add_node(ccx, i, empty_ann(len));
    }
}

fn visit_fn(&crate_ctxt ccx, uint num_constraints, &_fn f, &ty_param[] tps,
            &span sp, &fn_ident i, node_id id) {
    let @mutable node_id[] node_ids = @mutable ~[];
    node_ids_in_fn(f, tps, sp, i, id, node_ids);
    auto node_id_vec = *node_ids;
    init_vecs(ccx, node_id_vec, num_constraints);
}

fn annotate_in_fn(&crate_ctxt ccx, &_fn f, &ty_param[] tps,
                  &span sp, &fn_ident i, node_id id) {
    auto f_info = get_fn_info(ccx, id);
    visit_fn(ccx, num_constraints(f_info), f, tps, sp, i, id);
}

fn annotate_crate(&crate_ctxt ccx, &crate crate) {
    auto do_ann = visit::mk_simple_visitor
        (@rec(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
// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End:
//