OpenE2K/rust
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Eliminate a use of map::get in freevars

Browse Source 
map::get is evil -- just say no!
This commit is contained in:
Tim Chevalier 2012-01-05 11:33:22 -08:00
parent 4f3171ea81
commit da519c8587
1 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/comp/middle/freevars.rs
View File

@ -1,6 +1,7 @@
// A pass that annotates for each loops and functions with the free
// variables that they contain.
import syntax::print::pprust::path_to_str;
import int;
import std::map::*;
import option::*;
@ -47,21 +48,27 @@ fn collect_freevars(def_map: resolve::def_map, blk: ast::blk)
visit::visit_expr(expr, depth + 1, v);
}
ast::expr_path(path) {
let def = def_map.get(expr.id), i = 0;
while i < depth {
alt copy def {
ast::def_upvar(_, inner, _) { def = *inner; }
_ { break; }
let i = 0;
alt def_map.find(expr.id) {
none. { fail ("Not found: " + path_to_str(path)) }
some(df) {
let def = df;
while i < depth {
alt copy def {
ast::def_upvar(_, inner, _) { def = *inner; }
_ { break; }
}
i += 1;
}
if i == depth { // Made it to end of loop
let dnum = ast_util::def_id_of_def(def).node;
if !seen.contains_key(dnum) {
*refs += [@{def:def, span:expr.span}];
seen.insert(dnum, ());
}
}
}
i += 1;
}
if i == depth { // Made it to end of loop
let dnum = ast_util::def_id_of_def(def).node;
if !seen.contains_key(dnum) {
*refs += [@{def:def, span:expr.span}];
seen.insert(dnum, ());
}
}
}
}
_ { visit::visit_expr(expr, depth, v); }
}