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

import std::vec;
import std::option;
import std::option::some;
import std::option::none;
import front::ast::*;
import util::common::istr;
import util::common::uistr;
import util::common::span;
import util::common::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 vec[node_id] rs) {
    vec::push(*rs, e.id);
}

fn collect_ids_block(&block b, @mutable vec[node_id] rs) {
    vec::push(*rs, b.node.id);
}

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

fn collect_ids_local(&@local l, @mutable vec[node_id] rs) {
    vec::push(*rs, l.node.id);
}

fn node_ids_in_fn(&_fn f, &vec[ty_param] tps, &span sp, &fn_ident i,
                  node_id id, @mutable vec[node_id] rs) {
    auto collect_ids = walk::default_visitor();
    collect_ids =
        rec(visit_expr_pre=bind collect_ids_expr(_, rs),
            visit_block_pre=bind collect_ids_block(_, rs),
            visit_stmt_pre=bind collect_ids_stmt(_, rs),
            visit_local_pre=bind collect_ids_local(_, rs) with collect_ids);
    walk::walk_fn(collect_ids, f, tps, sp, i, id);
}

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

fn visit_fn(&crate_ctxt ccx, uint num_constraints, &_fn f, &vec[ty_param] tps,
            &span sp, &fn_ident i, node_id id) {
    let @mutable vec[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, &vec[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 = walk::default_visitor();
    do_ann =
        rec(visit_fn_pre=bind annotate_in_fn(ccx, _, _, _, _, _) with do_ann);
    walk::walk_crate(do_ann, crate);
}
//
// 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:
//