rustc: Thread an item-to-type mapping throughout the typechecking and translation phases

This commit is contained in:
Patrick Walton 2011-03-30 17:23:25 -07:00
parent a154c5b656
commit cc59cea8b0
10 changed files with 393 additions and 351 deletions

View file

@ -6,6 +6,7 @@ import front.token;
import front.eval;
import middle.trans;
import middle.resolve;
import middle.ty;
import middle.typeck;
import util.common;
@ -62,8 +63,12 @@ impure fn compile_input(session.session sess,
auto crate = parse_input(sess, p, input);
crate = creader.read_crates(sess, crate, library_search_paths);
crate = resolve.resolve_crate(sess, crate);
crate = typeck.check_crate(sess, crate);
trans.trans_crate(sess, crate, output, shared);
auto typeck_result = typeck.check_crate(sess, crate);
crate = typeck_result._0;
auto type_cache = typeck_result._1;
trans.trans_crate(sess, crate, type_cache, output, shared);
}
impure fn pretty_print_input(session.session sess,

View file

@ -395,6 +395,8 @@ tag view_item_ {
view_item_export(ident);
}
type obj_def_ids = rec(def_id ty, def_id ctor);
type item = spanned[item_];
tag item_ {
item_const(ident, @ty, @expr, def_id, ann);
@ -403,7 +405,7 @@ tag item_ {
item_native_mod(ident, native_mod, def_id);
item_ty(ident, @ty, vec[ty_param], def_id, ann);
item_tag(ident, vec[variant], vec[ty_param], def_id);
item_obj(ident, _obj, vec[ty_param], def_id, ann);
item_obj(ident, _obj, vec[ty_param], obj_def_ids, ann);
}
type native_item = spanned[native_item_];

View file

@ -544,6 +544,23 @@ fn read_crates(session.session sess,
}
fn kind_has_type_params(u8 kind_ch) -> bool {
// FIXME: It'd be great if we had u8 char literals.
if (kind_ch == ('c' as u8)) { ret false; }
else if (kind_ch == ('f' as u8)) { ret true; }
else if (kind_ch == ('o' as u8)) { ret true; }
else if (kind_ch == ('t' as u8)) { ret true; }
else if (kind_ch == ('m' as u8)) { ret false; }
else if (kind_ch == ('n' as u8)) { ret false; }
else if (kind_ch == ('v' as u8)) { ret true; }
else {
log #fmt("kind_has_type_params(): unknown kind char: %d",
kind_ch as int);
fail;
}
}
// Crate metadata queries
fn lookup_def(session.session sess, int cnum, vec[ast.ident] path)
@ -567,7 +584,6 @@ fn lookup_def(session.session sess, int cnum, vec[ast.ident] path)
auto def;
if (kind_ch == ('c' as u8)) { def = ast.def_const(did); }
else if (kind_ch == ('f' as u8)) { def = ast.def_fn(did); }
else if (kind_ch == ('y' as u8)) { def = ast.def_ty(did); }
else if (kind_ch == ('o' as u8)) { def = ast.def_obj(did); }
else if (kind_ch == ('t' as u8)) { def = ast.def_ty(did); }
else if (kind_ch == ('m' as u8)) { def = ast.def_mod(did); }
@ -584,13 +600,23 @@ fn lookup_def(session.session sess, int cnum, vec[ast.ident] path)
ret some[ast.def](def);
}
fn get_type(session.session sess, ast.def_id def) -> ty.ty_params_and_ty {
fn get_type(session.session sess, ast.def_id def) -> ty.ty_params_opt_and_ty {
auto external_crate_id = def._0;
auto data = sess.get_external_crate(external_crate_id);
auto ebml_r = lookup_item(def._1, data);
auto t = get_item_type(ebml_r, external_crate_id);
auto tps = get_item_ty_params(ebml_r, external_crate_id);
ret tup(tps, t);
auto tps_opt;
auto kind_ch = get_item_kind(ebml_r);
auto has_ty_params = kind_has_type_params(kind_ch);
if (has_ty_params) {
auto tps = get_item_ty_params(ebml_r, external_crate_id);
tps_opt = some[vec[ast.def_id]](tps);
} else {
tps_opt = none[vec[ast.def_id]];
}
ret tup(tps_opt, t);
}
fn get_symbol(session.session sess, ast.def_id def) -> str {

View file

@ -1876,8 +1876,8 @@ impure fn parse_item_obj(parser p, ast.layer lyr) -> @ast.item {
methods=meths,
dtor=dtor);
auto item = ast.item_obj(ident, ob, ty_params,
p.next_def_id(), ast.ann_none);
auto odid = rec(ty=p.next_def_id(), ctor=p.next_def_id());
auto item = ast.item_obj(ident, ob, ty_params, odid, ast.ann_none);
ret @spanned(lo, hi, item);
}

View file

@ -269,7 +269,7 @@ type ast_fold[ENV] =
(fn(&ENV e, &span sp, ident ident,
&ast._obj ob,
vec[ast.ty_param] ty_params,
def_id id, ann a) -> @item) fold_item_obj,
ast.obj_def_ids odid, ann a) -> @item) fold_item_obj,
// View Item folds.
(fn(&ENV e, &span sp, ident ident,
@ -975,9 +975,9 @@ fn fold_item[ENV](&ENV env, ast_fold[ENV] fld, @item i) -> @item {
ty_params, id);
}
case (ast.item_obj(?ident, ?ob, ?tps, ?id, ?ann)) {
case (ast.item_obj(?ident, ?ob, ?tps, ?odid, ?ann)) {
let ast._obj ob_ = fold_obj[ENV](env_, fld, ob);
ret fld.fold_item_obj(env_, i.span, ident, ob_, tps, id, ann);
ret fld.fold_item_obj(env_, i.span, ident, ob_, tps, odid, ann);
}
}
@ -1429,8 +1429,8 @@ fn identity_fold_item_tag[ENV](&ENV e, &span sp, ident i,
fn identity_fold_item_obj[ENV](&ENV e, &span sp, ident i,
&ast._obj ob, vec[ast.ty_param] ty_params,
def_id id, ann a) -> @item {
ret @respan(sp, ast.item_obj(i, ob, ty_params, id, a));
ast.obj_def_ids odid, ann a) -> @item {
ret @respan(sp, ast.item_obj(i, ob, ty_params, odid, a));
}
// View Item folds.

View file

@ -30,6 +30,7 @@ const uint tag_items_type = 0x0au;
const uint tag_items_symbol = 0x0bu;
const uint tag_items_variant = 0x0cu;
const uint tag_items_tag_id = 0x0du;
const uint tag_items_obj_type_id = 0x0eu;
// Type encoding
@ -237,10 +238,11 @@ fn encode_module_item_paths(&ebml.writer ebml_w, &ast._mod module) {
encode_def_id(ebml_w, did);
ebml.end_tag(ebml_w);
}
case (ast.item_obj(?id, _, ?tps, ?did, ?ann)) {
case (ast.item_obj(?id, _, ?tps, ?odid, ?ann)) {
ebml.start_tag(ebml_w, tag_paths_item);
encode_name(ebml_w, id);
encode_def_id(ebml_w, did);
encode_def_id(ebml_w, odid.ctor);
encode_obj_type_id(ebml_w, odid.ty);
ebml.end_tag(ebml_w);
}
}
@ -301,6 +303,12 @@ fn encode_tag_id(&ebml.writer ebml_w, &ast.def_id id) {
ebml.end_tag(ebml_w);
}
fn encode_obj_type_id(&ebml.writer ebml_w, &ast.def_id id) {
ebml.start_tag(ebml_w, tag_items_obj_type_id);
ebml_w.writer.write(_str.bytes(def_to_str(id)));
ebml.end_tag(ebml_w);
}
fn encode_tag_variant_info(@trans.crate_ctxt cx, &ebml.writer ebml_w,
ast.def_id did, vec[ast.variant] variants) {
@ -367,13 +375,21 @@ fn encode_info_for_item(@trans.crate_ctxt cx, &ebml.writer ebml_w,
encode_tag_variant_info(cx, ebml_w, did, variants);
}
case (ast.item_obj(?id, _, ?tps, ?did, ?ann)) {
case (ast.item_obj(?id, _, ?tps, ?odid, ?ann)) {
ebml.start_tag(ebml_w, tag_items_item);
encode_def_id(ebml_w, did);
encode_def_id(ebml_w, odid.ctor);
encode_kind(ebml_w, 'o' as u8);
encode_type_params(ebml_w, tps);
encode_type(ebml_w, trans.node_ann_type(cx, ann));
encode_symbol(cx, ebml_w, did);
auto fn_ty = trans.node_ann_type(cx, ann);
encode_type(ebml_w, fn_ty);
encode_symbol(cx, ebml_w, odid.ctor);
ebml.end_tag(ebml_w);
ebml.start_tag(ebml_w, tag_items_item);
encode_def_id(ebml_w, odid.ty);
encode_kind(ebml_w, 'y' as u8);
encode_type_params(ebml_w, tps);
encode_type(ebml_w, ty.ty_fn_ret(fn_ty));
ebml.end_tag(ebml_w);
}
}

View file

@ -29,6 +29,11 @@ tag scope {
type env = rec(list[scope] scopes,
session.session sess);
tag namespace {
ns_value;
ns_type;
}
type import_map = std.map.hashmap[ast.def_id,def_wrap];
// A simple wrapper over defs that stores a bit more information about modules
@ -97,8 +102,8 @@ fn unwrap_def(def_wrap d) -> def {
}
}
fn lookup_name(&env e, ast.ident i) -> option.t[def] {
auto d_ = lookup_name_wrapped(e, i);
fn lookup_name(&env e, ast.ident i, namespace ns) -> option.t[def] {
auto d_ = lookup_name_wrapped(e, i, ns);
alt (d_) {
case (none[tup(@env, def_wrap)]) {
ret none[def];
@ -138,24 +143,26 @@ fn lookup_external_def(session.session sess, int cnum, vec[ident] idents)
// If used after imports are resolved, import_id is none.
fn find_final_def(&env e, import_map index,
&span sp, vec[ident] idents,
&span sp, vec[ident] idents, namespace ns,
option.t[ast.def_id] import_id) -> def_wrap {
// We are given a series of identifiers (p.q.r) and we know that
// in the environment 'e' the identifier 'p' was resolved to 'd'. We
// should return what p.q.r points to in the end.
fn found_something(&env e, import_map index,
&span sp, vec[ident] idents, def_wrap d) -> def_wrap {
&span sp, vec[ident] idents, namespace ns,
def_wrap d) -> def_wrap {
fn found_mod(&env e, &import_map index, &span sp,
vec[ident] idents, @ast.item i) -> def_wrap {
vec[ident] idents, namespace ns,
@ast.item i) -> def_wrap {
auto len = _vec.len[ident](idents);
auto rest_idents = _vec.slice[ident](idents, 1u, len);
auto empty_e = rec(scopes = nil[scope],
sess = e.sess);
auto tmp_e = update_env_for_item(empty_e, i);
auto next_i = rest_idents.(0);
auto next_ = lookup_name_wrapped(tmp_e, next_i);
auto next_ = lookup_name_wrapped(tmp_e, next_i, ns);
alt (next_) {
case (none[tup(@env, def_wrap)]) {
e.sess.span_err(sp, "unresolved name: " + next_i);
@ -164,21 +171,22 @@ fn find_final_def(&env e, import_map index,
case (some[tup(@env, def_wrap)](?next)) {
auto combined_e = update_env_for_item(e, i);
ret found_something(combined_e, index, sp,
rest_idents, next._1);
rest_idents, ns, next._1);
}
}
}
// TODO: Refactor with above.
fn found_external_mod(&env e, &import_map index, &span sp,
vec[ident] idents, ast.def_id mod_id)
vec[ident] idents, namespace ns,
ast.def_id mod_id)
-> def_wrap {
auto len = _vec.len[ident](idents);
auto rest_idents = _vec.slice[ident](idents, 1u, len);
auto empty_e = rec(scopes = nil[scope], sess = e.sess);
auto tmp_e = update_env_for_external_mod(empty_e, mod_id, idents);
auto next_i = rest_idents.(0);
auto next_ = lookup_name_wrapped(tmp_e, next_i);
auto next_ = lookup_name_wrapped(tmp_e, next_i, ns);
alt (next_) {
case (none[tup(@env, def_wrap)]) {
e.sess.span_err(sp, "unresolved name: " + next_i);
@ -189,7 +197,7 @@ fn find_final_def(&env e, import_map index,
mod_id,
idents);
ret found_something(combined_e, index, sp,
rest_idents, next._1);
rest_idents, ns, next._1);
}
}
}
@ -212,9 +220,9 @@ fn find_final_def(&env e, import_map index,
case (def_wrap_import(?imp)) {
alt (imp.node) {
case (ast.view_item_import(_, ?new_idents, ?d, _)) {
auto x = find_final_def(e, index, sp, new_idents,
some(d));
ret found_something(e, index, sp, idents, x);
auto x = find_final_def(e, index, sp, new_idents, ns,
some(d));
ret found_something(e, index, sp, idents, ns, x);
}
}
}
@ -227,16 +235,16 @@ fn find_final_def(&env e, import_map index,
}
alt (d) {
case (def_wrap_mod(?i)) {
ret found_mod(e, index, sp, idents, i);
ret found_mod(e, index, sp, idents, ns, i);
}
case (def_wrap_native_mod(?i)) {
ret found_mod(e, index, sp, idents, i);
ret found_mod(e, index, sp, idents, ns, i);
}
case (def_wrap_external_mod(?mod_id)) {
ret found_external_mod(e, index, sp, idents, mod_id);
ret found_external_mod(e, index, sp, idents, ns, mod_id);
}
case (def_wrap_external_native_mod(?mod_id)) {
ret found_external_mod(e, index, sp, idents, mod_id);
ret found_external_mod(e, index, sp, idents, ns, mod_id);
}
case (def_wrap_use(?vi)) {
alt (vi.node) {
@ -273,14 +281,14 @@ fn find_final_def(&env e, import_map index,
index.insert(option.get[ast.def_id](import_id), def_wrap_resolving);
}
auto first = idents.(0);
auto d_ = lookup_name_wrapped(e, first);
auto d_ = lookup_name_wrapped(e, first, ns);
alt (d_) {
case (none[tup(@env, def_wrap)]) {
e.sess.span_err(sp, "unresolved name: " + first);
fail;
}
case (some[tup(@env, def_wrap)](?d)) {
auto x = found_something(*d._0, index, sp, idents, d._1);
auto x = found_something(*d._0, index, sp, idents, ns, d._1);
if (import_id != none[ast.def_id]) {
index.insert(option.get[ast.def_id](import_id), x);
}
@ -289,11 +297,12 @@ fn find_final_def(&env e, import_map index,
}
}
fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
fn lookup_name_wrapped(&env e, ast.ident i, namespace ns)
-> option.t[tup(@env, def_wrap)] {
// log "resolving name " + i;
fn found_def_item(@ast.item i) -> def_wrap {
fn found_def_item(@ast.item i, namespace ns) -> def_wrap {
alt (i.node) {
case (ast.item_const(_, _, _, ?id, _)) {
ret def_wrap_other(ast.def_const(id));
@ -313,8 +322,15 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
case (ast.item_tag(_, _, _, ?id)) {
ret def_wrap_other(ast.def_ty(id));
}
case (ast.item_obj(_, _, _, ?id, _)) {
ret def_wrap_other(ast.def_obj(id));
case (ast.item_obj(_, _, _, ?odid, _)) {
alt (ns) {
case (ns_value) {
ret def_wrap_other(ast.def_obj(odid.ctor));
}
case (ns_type) {
ret def_wrap_other(ast.def_obj(odid.ty));
}
}
}
}
}
@ -330,7 +346,7 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
}
}
fn found_decl_stmt(@ast.stmt s) -> def_wrap {
fn found_decl_stmt(@ast.stmt s, namespace ns) -> def_wrap {
alt (s.node) {
case (ast.stmt_decl(?d)) {
alt (d.node) {
@ -339,7 +355,7 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
ret def_wrap_other(t);
}
case (ast.decl_item(?it)) {
ret found_def_item(it);
ret found_def_item(it, ns);
}
}
}
@ -359,7 +375,8 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
fail;
}
fn check_mod(ast.ident i, ast._mod m) -> option.t[def_wrap] {
fn check_mod(ast.ident i, ast._mod m, namespace ns)
-> option.t[def_wrap] {
alt (m.index.find(i)) {
case (some[ast.mod_index_entry](?ent)) {
alt (ent) {
@ -367,7 +384,7 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
ret some(found_def_view(view_item));
}
case (ast.mie_item(?item)) {
ret some(found_def_item(item));
ret some(found_def_item(item, ns));
}
case (ast.mie_tag_variant(?item, ?variant_idx)) {
alt (item.node) {
@ -440,12 +457,13 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
}
}
fn check_block(ast.ident i, &ast.block_ b) -> option.t[def_wrap] {
fn check_block(ast.ident i, &ast.block_ b, namespace ns)
-> option.t[def_wrap] {
alt (b.index.find(i)) {
case (some[ast.block_index_entry](?ix)) {
alt(ix) {
case (ast.bie_item(?it)) {
ret some(found_def_item(it));
ret some(found_def_item(it, ns));
}
case (ast.bie_local(?l)) {
auto t = ast.def_local(l.id);
@ -460,12 +478,12 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
}
}
fn in_scope(&session.session sess, ast.ident i, &scope s)
fn in_scope(&session.session sess, ast.ident i, &scope s, namespace ns)
-> option.t[def_wrap] {
alt (s) {
case (scope_crate(?c)) {
ret check_mod(i, c.node.module);
ret check_mod(i, c.node.module, ns);
}
case (scope_item(?it)) {
@ -496,7 +514,7 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
}
}
case (ast.item_mod(_, ?m, _)) {
ret check_mod(i, m);
ret check_mod(i, m, ns);
}
case (ast.item_native_mod(_, ?m, _)) {
ret check_native_mod(i, m);
@ -537,7 +555,7 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
}
case (scope_block(?b)) {
ret check_block(i, b.node);
ret check_block(i, b.node, ns);
}
case (scope_arm(?a)) {
@ -558,14 +576,14 @@ fn lookup_name_wrapped(&env e, ast.ident i) -> option.t[tup(@env, def_wrap)] {
ret none[tup(@env, def_wrap)];
}
case (cons[scope](?hd, ?tl)) {
auto x = in_scope(e.sess, i, hd);
auto x = in_scope(e.sess, i, hd, ns);
alt (x) {
case (some[def_wrap](?x)) {
ret some(tup(@e, x));
}
case (none[def_wrap]) {
auto outer_env = rec(scopes = *tl with e);
ret lookup_name_wrapped(outer_env, i);
ret lookup_name_wrapped(outer_env, i, ns);
}
}
}
@ -579,7 +597,8 @@ fn fold_pat_tag(&env e, &span sp, ast.path p, vec[@ast.pat] args,
auto last_id = p.node.idents.(len - 1u);
auto new_def;
auto index = new_def_hash[def_wrap]();
auto d = find_final_def(e, index, sp, p.node.idents, none[ast.def_id]);
auto d = find_final_def(e, index, sp, p.node.idents, ns_value,
none[ast.def_id]);
alt (unwrap_def(d)) {
case (ast.def_variant(?did, ?vid)) {
new_def = some[ast.variant_def](tup(did, vid));
@ -619,7 +638,8 @@ fn fold_expr_path(&env e, &span sp, &ast.path p, &option.t[def] d,
check (n_idents != 0u);
auto index = new_def_hash[def_wrap]();
auto d = find_final_def(e, index, sp, p.node.idents, none[ast.def_id]);
auto d = find_final_def(e, index, sp, p.node.idents, ns_value,
none[ast.def_id]);
let uint path_len = 0u;
alt (d) {
case (def_wrap_expr_field(?remaining, _)) {
@ -655,7 +675,7 @@ fn fold_view_item_import(&env e, &span sp,
// Produce errors for invalid imports
auto len = _vec.len[ast.ident](is);
auto last_id = is.(len - 1u);
auto d = find_final_def(e, index, sp, is, some(id));
auto d = find_final_def(e, index, sp, is, ns_value, some(id));
alt (d) {
case (def_wrap_expr_field(?remain, _)) {
auto ident = is.(len - remain);
@ -671,7 +691,8 @@ fn fold_view_item_import(&env e, &span sp,
fn fold_ty_path(&env e, &span sp, ast.path p, &option.t[def] d) -> @ast.ty {
auto index = new_def_hash[def_wrap]();
auto d = find_final_def(e, index, sp, p.node.idents, none[ast.def_id]);
auto d = find_final_def(e, index, sp, p.node.idents, ns_type,
none[ast.def_id]);
ret @fold.respan[ast.ty_](sp, ast.ty_path(p, some(unwrap_def(d))));
}

View file

@ -95,6 +95,7 @@ state type crate_ctxt = rec(session.session sess,
hashmap[ast.def_id, @ast.item] items,
hashmap[ast.def_id,
@ast.native_item] native_items,
ty.type_cache type_cache,
hashmap[ast.def_id, str] item_symbols,
// TODO: hashmap[tup(tag_id,subtys), @tag_info]
hashmap[@ty.t, uint] tag_sizes,
@ -3610,6 +3611,7 @@ fn lval_generic_fn(@block_ctxt cx,
fn trans_external_path(@block_ctxt cx, &ast.path p,
ast.def def, ast.ann a) -> lval_result {
// FIXME: This isn't generic-safe.
auto ccx = cx.fcx.ccx;
auto ty = node_ann_type(ccx, a);
auto name = creader.get_symbol(ccx.sess, ast.def_id_of_def(def));
@ -3618,20 +3620,11 @@ fn trans_external_path(@block_ctxt cx, &ast.path p,
ret lval_mem(cx, v);
}
fn def_is_external(@crate_ctxt cx, ast.def d) -> bool {
auto id = ast.def_id_of_def(d);
ret id._0 != cx.sess.get_targ_crate_num();
}
fn trans_path(@block_ctxt cx, &ast.path p, &option.t[ast.def] dopt,
&ast.ann ann) -> lval_result {
alt (dopt) {
case (some[ast.def](?def)) {
if (def_is_external(cx.fcx.ccx, def)) {
ret trans_external_path(cx, p, def, ann);
}
alt (def) {
case (ast.def_arg(?did)) {
alt (cx.fcx.llargs.find(did)) {
@ -3665,15 +3658,18 @@ fn trans_path(@block_ctxt cx, &ast.path p, &option.t[ast.def] dopt,
}
case (ast.def_fn(?did)) {
check (cx.fcx.ccx.items.contains_key(did));
auto fn_item = cx.fcx.ccx.items.get(did);
ret lval_generic_fn(cx, ty.item_ty(fn_item), did, ann);
auto tyt = ty.lookup_generic_item_type(cx.fcx.ccx.sess,
cx.fcx.ccx.type_cache, did);
ret lval_generic_fn(cx, tyt, did, ann);
}
case (ast.def_obj(?did)) {
check (cx.fcx.ccx.items.contains_key(did));
auto fn_item = cx.fcx.ccx.items.get(did);
ret lval_generic_fn(cx, ty.item_ty(fn_item), did, ann);
auto tyt = ty.lookup_generic_item_type(cx.fcx.ccx.sess,
cx.fcx.ccx.type_cache, did);
ret lval_generic_fn(cx, tyt, did, ann);
}
case (ast.def_variant(?tid, ?vid)) {
// TODO: externals
if (cx.fcx.ccx.fn_pairs.contains_key(vid)) {
check (cx.fcx.ccx.items.contains_key(tid));
auto tag_item = cx.fcx.ccx.items.get(tid);
@ -3716,14 +3712,15 @@ fn trans_path(@block_ctxt cx, &ast.path p, &option.t[ast.def] dopt,
}
}
case (ast.def_const(?did)) {
// TODO: externals
check (cx.fcx.ccx.consts.contains_key(did));
ret lval_mem(cx, cx.fcx.ccx.consts.get(did));
}
case (ast.def_native_fn(?did)) {
check (cx.fcx.ccx.native_items.contains_key(did));
auto fn_item = cx.fcx.ccx.native_items.get(did);
ret lval_generic_fn(cx, ty.native_item_ty(fn_item),
did, ann);
auto tyt = ty.lookup_generic_item_type(cx.fcx.ccx.sess,
cx.fcx.ccx.type_cache, did);
ret lval_generic_fn(cx, tyt, did, ann);
}
case (_) {
cx.fcx.ccx.sess.unimpl("def variant in trans");
@ -5829,7 +5826,7 @@ fn trans_item(@crate_ctxt cx, &ast.item item) {
auto sub_cx = @rec(obj_typarams=tps,
obj_fields=ob.fields with
*extend_path(cx, name));
trans_obj(sub_cx, ob, oid, tps, ann);
trans_obj(sub_cx, ob, oid.ctor, tps, ann);
}
case (ast.item_mod(?name, ?m, _)) {
auto sub_cx = extend_path(cx, name);
@ -6125,9 +6122,9 @@ fn collect_item_pass2(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
}
case (ast.item_obj(?name, ?ob, ?tps, ?oid, ?ann)) {
cx.items.insert(oid, i);
cx.items.insert(oid.ctor, i);
decl_fn_and_pair(extend_path(cx, name), "obj_ctor",
tps, ann, oid);
tps, ann, oid.ctor);
for (@ast.method m in ob.methods) {
cx.obj_methods.insert(m.node.id, ());
}
@ -6782,8 +6779,8 @@ fn make_common_glue(str output) {
llvm.LLVMDisposeModule(llmod);
}
fn trans_crate(session.session sess, @ast.crate crate, str output,
bool shared) {
fn trans_crate(session.session sess, @ast.crate crate,
&ty.type_cache type_cache, str output, bool shared) {
auto llmod =
llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
llvm.LLVMGetGlobalContext());
@ -6816,6 +6813,7 @@ fn trans_crate(session.session sess, @ast.crate crate, str output,
item_ids = new_def_hash[ValueRef](),
items = new_def_hash[@ast.item](),
native_items = new_def_hash[@ast.native_item](),
type_cache = type_cache,
item_symbols = new_def_hash[str](),
tag_sizes = tag_sizes,
discrims = new_def_hash[ValueRef](),

View file

@ -10,6 +10,7 @@ import std.option.some;
import driver.session;
import front.ast;
import front.ast.mutability;
import front.creader;
import util.common;
import util.common.new_def_hash;
import util.common.span;
@ -86,6 +87,11 @@ tag unify_result {
ures_err(type_err, @ty.t, @ty.t);
}
type ty_params_opt_and_ty = tup(option.t[vec[ast.def_id]], @ty.t);
type type_cache = hashmap[ast.def_id,ty_params_opt_and_ty];
// Stringification
fn path_to_str(&ast.path pth) -> str {
@ -1673,6 +1679,55 @@ fn substitute_ty_params(vec[ast.ty_param] ty_params, vec[@t] bound, @t ty)
ret replace_type_params(ty, bindings);
}
fn def_has_ty_params(&ast.def def) -> bool {
alt (def) {
case (ast.def_fn(_)) { ret true; }
case (ast.def_obj(_)) { ret true; }
case (ast.def_obj_field(_)) { ret false; }
case (ast.def_mod(_)) { ret false; }
case (ast.def_const(_)) { ret false; }
case (ast.def_arg(_)) { ret false; }
case (ast.def_local(_)) { ret false; }
case (ast.def_variant(_, _)) { ret true; }
case (ast.def_ty(_)) { ret false; }
case (ast.def_ty_arg(_)) { ret false; }
case (ast.def_binding(_)) { ret false; }
case (ast.def_use(_)) { ret false; }
case (ast.def_native_ty(_)) { ret false; }
case (ast.def_native_fn(_)) { ret true; }
}
}
// If the given item is in an external crate, looks up its type and adds it to
// the type cache. Returns the type parameters and type.
fn lookup_item_type(session.session sess, &type_cache cache,
ast.def_id did) -> ty_params_opt_and_ty {
if (did._0 == sess.get_targ_crate_num()) {
// The item is in this crate. The caller should have added it to the
// type cache already; we simply return it.
check (cache.contains_key(did));
ret cache.get(did);
}
if (cache.contains_key(did)) {
ret cache.get(did);
}
auto tyt = creader.get_type(sess, did);
cache.insert(did, tyt);
ret tyt;
}
// A convenience function to retrive type parameters and a type when it's
// known that the item supports generics (functions, variants, objects).
fn lookup_generic_item_type(session.session sess, &type_cache cache,
ast.def_id did) -> ty_params_and_ty {
auto tp_opt_and_ty = lookup_item_type(sess, cache, did);
ret tup(option.get[vec[ast.def_id]](tp_opt_and_ty._0), tp_opt_and_ty._1);
}
// Local Variables:
// mode: rust
// fill-column: 78;

View file

@ -21,6 +21,7 @@ import middle.ty.plain_ty;
import middle.ty.ty_to_str;
import middle.ty.type_is_integral;
import middle.ty.type_is_scalar;
import middle.ty.ty_params_opt_and_ty;
import std._str;
import std._uint;
@ -39,12 +40,10 @@ tag any_item {
}
type ty_item_table = hashmap[ast.def_id,any_item];
type ty_param_table = hashmap[ast.def_id,vec[ast.def_id]];
type crate_ctxt = rec(session.session sess,
@ty_table item_types,
ty.type_cache type_cache,
@ty_item_table item_items,
@ty_param_table item_ty_params,
vec[ast.obj_field] obj_fields,
mutable int next_var_id);
@ -53,7 +52,7 @@ type fn_ctxt = rec(@ty.t ret_ty,
@crate_ctxt ccx);
// Used for ast_ty_to_ty() below.
type ty_getter = fn(ast.def_id) -> ty.ty_params_and_ty;
type ty_getter = fn(ast.def_id) -> ty.ty_params_opt_and_ty;
// Replaces parameter types inside a type with type variables.
fn generalize_ty(@crate_ctxt cx, @ty.t t) -> @ty.t {
@ -128,24 +127,6 @@ fn substitute_ty_params(&@crate_ctxt ccx,
}
// Looks up the type of the given item in an external crate.
fn lookup_item_type_if_necessary(@crate_ctxt ccx, ast.def_id did) {
if (did._0 == ccx.sess.get_targ_crate_num()) {
ret; // Nothing to do; it should already be in the tables.
}
if (ccx.item_types.contains_key(did)) {
ret; // Nothing to do; we already looked up this item's type.
}
auto tyt = creader.get_type(ccx.sess, did);
ccx.item_types.insert(did, tyt._1);
ccx.item_ty_params.insert(did, tyt._0);
}
type ty_params_opt_and_ty = tup(option.t[vec[ast.def_id]], @ty.t);
// Returns the type parameters and the type for the given definition.
fn ty_params_and_ty_for_def(@fn_ctxt fcx, &ast.def defn)
-> ty_params_opt_and_ty {
@ -167,38 +148,23 @@ fn ty_params_and_ty_for_def(@fn_ctxt fcx, &ast.def defn)
ret tup(none[vec[ast.def_id]], fcx.locals.get(id));
}
case (ast.def_fn(?id)) {
lookup_item_type_if_necessary(fcx.ccx, id);
check (fcx.ccx.item_types.contains_key(id));
ret tup(some(fcx.ccx.item_ty_params.get(id)),
fcx.ccx.item_types.get(id));
ret ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache, id);
}
case (ast.def_native_fn(?id)) {
lookup_item_type_if_necessary(fcx.ccx, id);
check (fcx.ccx.item_types.contains_key(id));
ret tup(some(fcx.ccx.item_ty_params.get(id)),
fcx.ccx.item_types.get(id));
ret ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache, id);
}
case (ast.def_const(?id)) {
lookup_item_type_if_necessary(fcx.ccx, id);
check (fcx.ccx.item_types.contains_key(id));
ret tup(none[vec[ast.def_id]], fcx.ccx.item_types.get(id));
ret ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache, id);
}
case (ast.def_variant(?tag_id, ?variant_id)) {
lookup_item_type_if_necessary(fcx.ccx, tag_id);
lookup_item_type_if_necessary(fcx.ccx, variant_id);
check (fcx.ccx.item_types.contains_key(variant_id));
ret tup(some(fcx.ccx.item_ty_params.get(tag_id)),
fcx.ccx.item_types.get(variant_id));
case (ast.def_variant(_, ?vid)) {
ret ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache, vid);
}
case (ast.def_binding(?id)) {
check (fcx.locals.contains_key(id));
ret tup(none[vec[ast.def_id]], fcx.locals.get(id));
}
case (ast.def_obj(?id)) {
lookup_item_type_if_necessary(fcx.ccx, id);
check (fcx.ccx.item_types.contains_key(id));
ret tup(some(fcx.ccx.item_ty_params.get(id)),
fcx.ccx.item_types.get(id));
ret ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache, id);
}
case (ast.def_mod(_)) {
@ -266,33 +232,31 @@ fn instantiate_path(@fn_ctxt fcx, &ast.path pth, &ty_params_opt_and_ty tpt,
// Returns the type parameters and polytype of an item, if it's an item that
// supports type parameters.
//
// TODO: This function is a little silly in the presence of the new
// lookup_item_type(); remove this in favor of lookup_item_type() if possible.
fn ty_params_for_item(@crate_ctxt ccx, &ast.def d)
-> option.t[ty.ty_params_and_ty] {
auto params_id;
auto types_id;
auto did;
alt (d) {
case (ast.def_fn(?id)) { params_id = id; types_id = id; }
case (ast.def_obj(?id)) { params_id = id; types_id = id; }
case (ast.def_fn(?id)) { did = id; }
case (ast.def_obj(?id)) { did = id; }
case (ast.def_obj_field(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_mod(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_const(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_arg(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_local(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_variant(?tid, ?vid)) {
params_id = tid;
types_id = vid;
}
case (ast.def_variant(_, ?vid)) { did = vid; }
case (ast.def_ty(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_ty_arg(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_binding(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_use(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_native_ty(_)) { ret none[ty.ty_params_and_ty]; }
case (ast.def_native_fn(?id)) { params_id = id; types_id = id; }
case (ast.def_native_fn(?id)) { did = id; }
}
auto tps = ccx.item_ty_params.get(params_id);
auto polyty = ccx.item_types.get(types_id);
ret some[ty.ty_params_and_ty](tup(tps, polyty));
auto tpt = ty.lookup_generic_item_type(ccx.sess, ccx.type_cache, did);
ret some[ty.ty_params_and_ty](tpt);
}
// Parses the programmer's textual representation of a type into our internal
@ -312,18 +276,26 @@ fn ast_ty_to_ty(ty_getter getter, &@ast.ty ast_ty) -> @ty.t {
vec[@ast.ty] args) -> @ty.t {
// TODO: maybe record cname chains so we can do
// "foo = int" like OCaml?
auto ty_and_params = getter(id);
auto params = ty_and_params._0;
auto num_type_args = _vec.len[@ast.ty](args);
check(num_type_args == _vec.len[ast.def_id](params));
auto params_opt_and_ty = getter(id);
alt (params_opt_and_ty._0) {
case (none[vec[ast.def_id]]) {
// FIXME: Session error. This is completely user-unfriendly.
log "item has no type parameters";
fail;
}
case (some[vec[ast.def_id]](?params)) {
auto num_type_args = _vec.len[@ast.ty](args);
check(num_type_args == _vec.len[ast.def_id](params));
auto param_map = common.new_def_hash[@ty.t]();
for each (uint i in _uint.range(0u, num_type_args)) {
auto arg = args.(i);
auto param = params.(i);
param_map.insert(param, ast_ty_to_ty(getter, arg));
auto param_map = common.new_def_hash[@ty.t]();
for each (uint i in _uint.range(0u, num_type_args)) {
auto arg = args.(i);
auto param = params.(i);
param_map.insert(param, ast_ty_to_ty(getter, arg));
}
ret ty.replace_type_params(params_opt_and_ty._1, param_map);
}
}
ret ty.replace_type_params(ty_and_params._1, param_map);
}
auto mut = ast.imm;
@ -378,7 +350,7 @@ fn ast_ty_to_ty(ty_getter getter, &@ast.ty ast_ty) -> @ty.t {
sty = instantiate(getter, id, path.node.types).struct;
}
case (ast.def_native_ty(?id)) {
sty = instantiate(getter, id, path.node.types).struct;
sty = getter(id)._1.struct;
}
case (ast.def_obj(?id)) {
sty = instantiate(getter, id, path.node.types).struct;
@ -410,87 +382,16 @@ fn ast_ty_to_ty(ty_getter getter, &@ast.ty ast_ty) -> @ty.t {
ret @rec(struct=sty, cname=cname);
}
fn actual_type(@ty.t t, @ast.item item) -> @ty.t {
alt (item.node) {
case (ast.item_obj(_,_,_,_,_)) {
// An obj used as a type name refers to the output type of the
// item (constructor).
ret middle.ty.ty_fn_ret(t);
}
case (_) { }
}
ret t;
}
// A convenience function to use a crate_ctxt to resolve names for
// ast_ty_to_ty.
fn ast_ty_to_ty_crate(@crate_ctxt ccx, &@ast.ty ast_ty) -> @ty.t {
fn getter(@crate_ctxt ccx, ast.def_id id) -> ty.ty_params_and_ty {
if (id._0 != ccx.sess.get_targ_crate_num()) {
// This is a type we need to load in from the crate reader.
ret creader.get_type(ccx.sess, id);
}
check (ccx.item_items.contains_key(id));
check (ccx.item_types.contains_key(id));
auto it = ccx.item_items.get(id);
auto ty = ccx.item_types.get(id);
auto params;
alt (it) {
case (any_item_rust(?item)) {
ty = actual_type(ty, item);
params = ty_params_of_item(item);
}
case (any_item_native(?native_item, _)) {
params = ty_params_of_native_item(native_item);
}
}
let vec[ast.def_id] param_ids = vec();
for (ast.ty_param tp in params) {
param_ids += vec(tp.id);
}
ret tup(param_ids, ty);
fn getter(@crate_ctxt ccx, ast.def_id id) -> ty.ty_params_opt_and_ty {
ret ty.lookup_item_type(ccx.sess, ccx.type_cache, id);
}
auto f = bind getter(ccx, _);
ret ast_ty_to_ty(f, ast_ty);
}
fn ty_params_of_item(@ast.item item) -> vec[ast.ty_param] {
alt (item.node) {
case (ast.item_fn(_, _, ?p, _, _)) {
ret p;
}
case (ast.item_ty(_, _, ?p, _, _)) {
ret p;
}
case (ast.item_tag(_, _, ?p, _)) {
ret p;
}
case (ast.item_obj(_, _, ?p, _, _)) {
ret p;
}
case (_) {
let vec[ast.ty_param] r = vec();
ret r;
}
}
}
fn ty_params_of_native_item(@ast.native_item item) -> vec[ast.ty_param] {
alt (item.node) {
case (ast.native_item_fn(_, _, _, ?p, _, _)) {
ret p;
}
case (_) {
let vec[ast.ty_param] r = vec();
ret r;
}
}
}
// Item collection - a pair of bootstrap passes:
//
@ -503,41 +404,55 @@ fn ty_params_of_native_item(@ast.native_item item) -> vec[ast.ty_param] {
// We then annotate the AST with the resulting types and return the annotated
// AST, along with a table mapping item IDs to their types.
fn ty_params_to_def_ids(vec[ast.ty_param] tps) -> vec[ast.def_id] {
let vec[ast.def_id] result = vec();
for (ast.ty_param tp in tps) {
result += vec(tp.id);
}
ret result;
}
fn ty_of_fn_decl(@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
fn(&@ast.ty ast_ty) -> @ty.t convert,
fn(&ast.arg a) -> arg ty_of_arg,
&ast.fn_decl decl,
ast.proto proto,
ast.def_id def_id) -> @ty.t {
vec[ast.ty_param] ty_params,
ast.def_id def_id) -> ty.ty_params_opt_and_ty {
auto input_tys = _vec.map[ast.arg,arg](ty_of_arg, decl.inputs);
auto output_ty = convert(decl.output);
auto t_fn = plain_ty(ty.ty_fn(proto, input_tys, output_ty));
item_to_ty.insert(def_id, t_fn);
ret t_fn;
auto params_opt = some[vec[ast.def_id]](ty_params_to_def_ids(ty_params));
auto tpt = tup(params_opt, t_fn);
type_cache.insert(def_id, tpt);
ret tpt;
}
fn ty_of_native_fn_decl(@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
fn(&@ast.ty ast_ty) -> @ty.t convert,
fn(&ast.arg a) -> arg ty_of_arg,
&ast.fn_decl decl,
ast.native_abi abi,
ast.def_id def_id) -> @ty.t {
vec[ast.ty_param] ty_params,
ast.def_id def_id) -> ty.ty_params_opt_and_ty {
auto input_tys = _vec.map[ast.arg,arg](ty_of_arg, decl.inputs);
auto output_ty = convert(decl.output);
auto t_fn = plain_ty(ty.ty_native_fn(abi, input_tys, output_ty));
item_to_ty.insert(def_id, t_fn);
ret t_fn;
auto params_opt = some[vec[ast.def_id]](ty_params_to_def_ids(ty_params));
auto tpt = tup(params_opt, t_fn);
type_cache.insert(def_id, tpt);
ret tpt;
}
fn collect_item_types(session.session sess, @ast.crate crate)
-> tup(@ast.crate, @ty_table, @ty_item_table, @ty_param_table) {
-> tup(@ast.crate, ty.type_cache, @ty_item_table) {
fn getter(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ast.def_id id) -> ty.ty_params_and_ty {
ty.type_cache type_cache,
ast.def_id id) -> ty.ty_params_opt_and_ty {
if (id._0 != sess.get_targ_crate_num()) {
// This is a type we need to load in from the crate reader.
@ -547,44 +462,35 @@ fn collect_item_types(session.session sess, @ast.crate crate)
check (id_to_ty_item.contains_key(id));
auto it = id_to_ty_item.get(id);
auto ty;
auto params;
auto tpt;
alt (it) {
case (any_item_rust(?item)) {
ty = ty_of_item(sess, id_to_ty_item, item_to_ty, item);
ty = actual_type(ty, item);
params = ty_params_of_item(item);
tpt = ty_of_item(sess, id_to_ty_item, type_cache, item);
}
case (any_item_native(?native_item, ?abi)) {
ty = ty_of_native_item(sess, id_to_ty_item, item_to_ty,
native_item, abi);
params = ty_params_of_native_item(native_item);
tpt = ty_of_native_item(sess, id_to_ty_item, type_cache,
native_item, abi);
}
}
let vec[ast.def_id] param_ids = vec();
for (ast.ty_param tp in params) {
param_ids += vec(tp.id);
}
ret tup(param_ids, ty);
ret tpt;
}
fn ty_of_arg(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
&ast.arg a) -> arg {
auto f = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto f = bind getter(sess, id_to_ty_item, type_cache, _);
ret rec(mode=a.mode, ty=ast_ty_to_ty(f, a.ty));
}
fn ty_of_method(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
&@ast.method m) -> method {
auto get = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto get = bind getter(sess, id_to_ty_item, type_cache, _);
auto convert = bind ast_ty_to_ty(get, _);
auto f = bind ty_of_arg(sess, id_to_ty_item, item_to_ty, _);
auto f = bind ty_of_arg(sess, id_to_ty_item, type_cache, _);
auto inputs = _vec.map[ast.arg,arg](f, m.node.meth.decl.inputs);
auto output = convert(m.node.meth.decl.output);
ret rec(proto=m.node.meth.proto, ident=m.node.ident,
@ -593,77 +499,91 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn ty_of_obj(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
&ast.ident id,
&ast._obj obj_info) -> @ty.t {
auto f = bind ty_of_method(sess, id_to_ty_item, item_to_ty, _);
&ast._obj obj_info,
vec[ast.ty_param] ty_params) -> ty.ty_params_opt_and_ty {
auto f = bind ty_of_method(sess, id_to_ty_item, type_cache, _);
auto methods =
_vec.map[@ast.method,method](f, obj_info.methods);
auto t_obj = @rec(struct=ty.ty_obj(ty.sort_methods(methods)),
cname=some[str](id));
ret t_obj;
auto params = ty_params_to_def_ids(ty_params);
auto params_opt = some[vec[ast.def_id]](params);
ret tup(params_opt, t_obj);
}
fn ty_of_obj_ctor(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
&ast.ident id,
&ast._obj obj_info) -> @ty.t {
auto t_obj = ty_of_obj(sess, id_to_ty_item, item_to_ty,
id, obj_info);
&ast._obj obj_info,
ast.def_id obj_ty_id,
vec[ast.ty_param] ty_params)
-> ty.ty_params_opt_and_ty {
auto t_obj = ty_of_obj(sess, id_to_ty_item, type_cache,
id, obj_info, ty_params);
let vec[arg] t_inputs = vec();
for (ast.obj_field f in obj_info.fields) {
auto g = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto g = bind getter(sess, id_to_ty_item, type_cache, _);
auto t_field = ast_ty_to_ty(g, f.ty);
_vec.push[arg](t_inputs, rec(mode=ast.alias, ty=t_field));
}
auto t_fn = plain_ty(ty.ty_fn(ast.proto_fn, t_inputs, t_obj));
ret t_fn;
type_cache.insert(obj_ty_id, t_obj);
auto t_fn = plain_ty(ty.ty_fn(ast.proto_fn, t_inputs, t_obj._1));
ret tup(t_obj._0, t_fn);
}
fn ty_of_item(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
@ast.item it) -> @ty.t {
ty.type_cache type_cache,
@ast.item it) -> ty.ty_params_opt_and_ty {
auto get = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto get = bind getter(sess, id_to_ty_item, type_cache, _);
auto convert = bind ast_ty_to_ty(get, _);
alt (it.node) {
case (ast.item_const(?ident, ?t, _, ?def_id, _)) {
item_to_ty.insert(def_id, convert(t));
auto typ = convert(t);
type_cache.insert(def_id, tup(none[vec[ast.def_id]], typ));
}
case (ast.item_fn(?ident, ?fn_info, _, ?def_id, _)) {
auto f = bind ty_of_arg(sess, id_to_ty_item, item_to_ty, _);
ret ty_of_fn_decl(id_to_ty_item, item_to_ty, convert,
f, fn_info.decl, fn_info.proto, def_id);
case (ast.item_fn(?ident, ?fn_info, ?tps, ?def_id, _)) {
auto f = bind ty_of_arg(sess, id_to_ty_item, type_cache, _);
ret ty_of_fn_decl(id_to_ty_item, type_cache, convert, f,
fn_info.decl, fn_info.proto, tps, def_id);
}
case (ast.item_obj(?ident, ?obj_info, _, ?def_id, _)) {
// TODO: handle ty-params
case (ast.item_obj(?ident, ?obj_info, ?tps, ?odid, _)) {
auto t_ctor = ty_of_obj_ctor(sess,
id_to_ty_item,
item_to_ty,
type_cache,
ident,
obj_info);
item_to_ty.insert(def_id, t_ctor);
ret t_ctor;
obj_info,
odid.ty,
tps);
type_cache.insert(odid.ctor, t_ctor);
ret type_cache.get(odid.ty);
}
case (ast.item_ty(?ident, ?ty, _, ?def_id, _)) {
if (item_to_ty.contains_key(def_id)) {
case (ast.item_ty(?ident, ?ty, ?tps, ?def_id, _)) {
if (type_cache.contains_key(def_id)) {
// Avoid repeating work.
ret item_to_ty.get(def_id);
ret type_cache.get(def_id);
}
// Tell ast_ty_to_ty() that we want to perform a recursive
// call to resolve any named types.
auto ty_ = convert(ty);
item_to_ty.insert(def_id, ty_);
ret ty_;
auto typ = convert(ty);
auto params = ty_params_to_def_ids(tps);
auto params_opt = some[vec[ast.def_id]](params);
auto tpt = tup(params_opt, typ);
type_cache.insert(def_id, tpt);
ret tpt;
}
case (ast.item_tag(_, _, ?tps, ?def_id)) {
@ -673,8 +593,12 @@ fn collect_item_types(session.session sess, @ast.crate crate)
subtys += vec(plain_ty(ty.ty_param(tp.id)));
}
auto t = plain_ty(ty.ty_tag(def_id, subtys));
item_to_ty.insert(def_id, t);
ret t;
auto params = ty_params_to_def_ids(tps);
auto params_opt = some[vec[ast.def_id]](params);
auto tpt = tup(params_opt, t);
type_cache.insert(def_id, tpt);
ret tpt;
}
case (ast.item_mod(_, _, _)) { fail; }
@ -684,33 +608,35 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn ty_of_native_item(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
@ast.native_item it,
ast.native_abi abi) -> @ty.t {
ast.native_abi abi) -> ty.ty_params_opt_and_ty {
alt (it.node) {
case (ast.native_item_fn(?ident, ?lname, ?fn_decl,
?params, ?def_id, _)) {
auto get = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto get = bind getter(sess, id_to_ty_item, type_cache, _);
auto convert = bind ast_ty_to_ty(get, _);
auto f = bind ty_of_arg(sess, id_to_ty_item, item_to_ty, _);
ret ty_of_native_fn_decl(id_to_ty_item, item_to_ty,
convert, f, fn_decl, abi, def_id);
auto f = bind ty_of_arg(sess, id_to_ty_item, type_cache, _);
ret ty_of_native_fn_decl(id_to_ty_item, type_cache, convert,
f, fn_decl, abi, params, def_id);
}
case (ast.native_item_ty(_, ?def_id)) {
if (item_to_ty.contains_key(def_id)) {
if (type_cache.contains_key(def_id)) {
// Avoid repeating work.
ret item_to_ty.get(def_id);
ret type_cache.get(def_id);
}
auto x = @rec(struct=ty.ty_native, cname=none[str]);
item_to_ty.insert(def_id, x);
ret x;
auto t = @rec(struct=ty.ty_native, cname=none[str]);
auto tpt = tup(none[vec[ast.def_id]], t);
type_cache.insert(def_id, tpt);
ret tpt;
}
}
}
fn get_tag_variant_types(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
ty.type_cache type_cache,
&ast.def_id tag_id,
&vec[ast.variant] variants,
&vec[ast.ty_param] ty_params)
@ -723,6 +649,9 @@ fn collect_item_types(session.session sess, @ast.crate crate)
ty_param_tys += vec(plain_ty(ty.ty_param(tp.id)));
}
auto params = ty_params_to_def_ids(ty_params);
auto params_opt = some[vec[ast.def_id]](params);
for (ast.variant variant in variants) {
// Nullary tag constructors get turned into constants; n-ary tag
// constructors get turned into functions.
@ -732,7 +661,7 @@ fn collect_item_types(session.session sess, @ast.crate crate)
} else {
// As above, tell ast_ty_to_ty() that trans_ty_item_to_ty()
// should be called to resolve named types.
auto f = bind getter(sess, id_to_ty_item, item_to_ty, _);
auto f = bind getter(sess, id_to_ty_item, type_cache, _);
let vec[arg] args = vec();
for (ast.variant_arg va in variant.args) {
@ -743,7 +672,8 @@ fn collect_item_types(session.session sess, @ast.crate crate)
result_ty = plain_ty(ty.ty_fn(ast.proto_fn, args, tag_t));
}
item_to_ty.insert(variant.id, result_ty);
auto tpt = tup(params_opt, result_ty);
type_cache.insert(variant.id, tpt);
auto variant_t = rec(
ann=ast.ann_type(result_ty, none[vec[@ty.t]])
@ -767,8 +697,8 @@ fn collect_item_types(session.session sess, @ast.crate crate)
case (ast.item_tag(_, _, _, ?def_id)) {
id_to_ty_item.insert(def_id, any_item_rust(i));
}
case (ast.item_obj(_, _, _, ?def_id, _)) {
id_to_ty_item.insert(def_id, any_item_rust(i));
case (ast.item_obj(_, _, _, ?odid, _)) {
id_to_ty_item.insert(odid.ty, any_item_rust(i));
}
case (_) { /* empty */ }
}
@ -797,30 +727,17 @@ fn collect_item_types(session.session sess, @ast.crate crate)
// Second pass: translate the types of all items.
let @ty_table item_to_ty = @common.new_def_hash[@ty.t]();
auto item_ty_params = @common.new_def_hash[vec[ast.def_id]]();
auto type_cache = common.new_def_hash[ty.ty_params_opt_and_ty]();
type env = rec(session.session sess,
@ty_item_table id_to_ty_item,
@ty_table item_to_ty,
@ty_param_table item_ty_params,
ty.type_cache type_cache,
ast.native_abi abi);
let @env e = @rec(sess=sess,
id_to_ty_item=id_to_ty_item,
item_to_ty=item_to_ty,
item_ty_params=item_ty_params,
type_cache=type_cache,
abi=ast.native_abi_cdecl);
// Inserts the given type parameters into the type parameter table of the
// environment.
fn collect_ty_params(&@env e, &ast.def_id id, vec[ast.ty_param] tps) {
let vec[ast.def_id] result = vec();
for (ast.ty_param tp in tps) {
result += vec(tp.id);
}
e.item_ty_params.insert(id, result);
}
fn convert(&@env e, @ast.item i) -> @env {
auto abi = e.abi;
alt (i.node) {
@ -834,22 +751,22 @@ fn collect_item_types(session.session sess, @ast.crate crate)
case (_) {
// This call populates the ty_table with the converted type of
// the item in passing; we don't need to do anything else.
ty_of_item(e.sess, e.id_to_ty_item, e.item_to_ty, i);
ty_of_item(e.sess, e.id_to_ty_item, e.type_cache, i);
}
}
ret @rec(abi=abi with *e);
}
fn convert_native(&@env e, @ast.native_item i) -> @env {
ty_of_native_item(e.sess, e.id_to_ty_item, e.item_to_ty, i, e.abi);
ty_of_native_item(e.sess, e.id_to_ty_item, e.type_cache, i, e.abi);
ret e;
}
fn fold_item_const(&@env e, &span sp, ast.ident i,
@ast.ty t, @ast.expr ex,
ast.def_id id, ast.ann a) -> @ast.item {
check (e.item_to_ty.contains_key(id));
auto typ = e.item_to_ty.get(id);
check (e.type_cache.contains_key(id));
auto typ = e.type_cache.get(id)._1;
auto item = ast.item_const(i, t, ex, id,
ast.ann_type(typ, none[vec[@ty.t]]));
ret @fold.respan[ast.item_](sp, item);
@ -858,10 +775,8 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn fold_item_fn(&@env e, &span sp, ast.ident i,
&ast._fn f, vec[ast.ty_param] ty_params,
ast.def_id id, ast.ann a) -> @ast.item {
collect_ty_params(e, id, ty_params);
check (e.item_to_ty.contains_key(id));
auto typ = e.item_to_ty.get(id);
check (e.type_cache.contains_key(id));
auto typ = e.type_cache.get(id)._1;
auto item = ast.item_fn(i, f, ty_params, id,
ast.ann_type(typ, none[vec[@ty.t]]));
ret @fold.respan[ast.item_](sp, item);
@ -870,10 +785,8 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn fold_native_item_fn(&@env e, &span sp, ast.ident i, option.t[str] ln,
&ast.fn_decl d, vec[ast.ty_param] ty_params,
ast.def_id id, ast.ann a) -> @ast.native_item {
collect_ty_params(e, id, ty_params);
check (e.item_to_ty.contains_key(id));
auto typ = e.item_to_ty.get(id);
check (e.type_cache.contains_key(id));
auto typ = e.type_cache.get(id)._1;
auto item = ast.native_item_fn(i, ln, d, ty_params, id,
ast.ann_type(typ, none[vec[@ty.t]]));
ret @fold.respan[ast.native_item_](sp, item);
@ -902,11 +815,9 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn fold_item_obj(&@env e, &span sp, ast.ident i,
&ast._obj ob, vec[ast.ty_param] ty_params,
ast.def_id id, ast.ann a) -> @ast.item {
collect_ty_params(e, id, ty_params);
check (e.item_to_ty.contains_key(id));
auto t = e.item_to_ty.get(id);
ast.obj_def_ids odid, ast.ann a) -> @ast.item {
check (e.type_cache.contains_key(odid.ctor));
auto t = e.type_cache.get(odid.ctor)._1;
let vec[method] meth_tys = get_ctor_obj_methods(t);
let vec[@ast.method] methods = vec();
let vec[ast.obj_field] fields = vec();
@ -928,7 +839,7 @@ fn collect_item_types(session.session sess, @ast.crate crate)
m = @rec(node=m_ with *meth);
_vec.push[@ast.method](methods, m);
}
auto g = bind getter(e.sess, e.id_to_ty_item, e.item_to_ty, _);
auto g = bind getter(e.sess, e.id_to_ty_item, e.type_cache, _);
for (ast.obj_field fld in ob.fields) {
let @ty.t fty = ast_ty_to_ty(g, fld.ty);
let ast.obj_field f = rec(
@ -941,7 +852,7 @@ fn collect_item_types(session.session sess, @ast.crate crate)
auto ob_ = rec(methods = methods,
fields = fields
with ob);
auto item = ast.item_obj(i, ob_, ty_params, id,
auto item = ast.item_obj(i, ob_, ty_params, odid,
ast.ann_type(t, none[vec[@ty.t]]));
ret @fold.respan[ast.item_](sp, item);
}
@ -949,10 +860,8 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fn fold_item_ty(&@env e, &span sp, ast.ident i,
@ast.ty t, vec[ast.ty_param] ty_params,
ast.def_id id, ast.ann a) -> @ast.item {
collect_ty_params(e, id, ty_params);
check (e.item_to_ty.contains_key(id));
auto typ = e.item_to_ty.get(id);
check (e.type_cache.contains_key(id));
auto typ = e.type_cache.get(id)._1;
auto item = ast.item_ty(i, t, ty_params, id,
ast.ann_type(typ, none[vec[@ty.t]]));
ret @fold.respan[ast.item_](sp, item);
@ -962,11 +871,9 @@ fn collect_item_types(session.session sess, @ast.crate crate)
vec[ast.variant] variants,
vec[ast.ty_param] ty_params,
ast.def_id id) -> @ast.item {
collect_ty_params(e, id, ty_params);
auto variants_t = get_tag_variant_types(e.sess,
e.id_to_ty_item,
e.item_to_ty,
e.type_cache,
id,
variants,
ty_params);
@ -986,7 +893,7 @@ fn collect_item_types(session.session sess, @ast.crate crate)
fold_item_tag = bind fold_item_tag(_,_,_,_,_,_)
with *fld_2);
auto crate_ = fold.fold_crate[@env](e, fld_2, crate);
ret tup(crate_, item_to_ty, id_to_ty_item, item_ty_params);
ret tup(crate_, type_cache, id_to_ty_item);
}
fn unify(&@fn_ctxt fcx, @ty.t expected, @ty.t actual) -> ty.unify_result {
@ -1139,9 +1046,15 @@ fn demand_pat(&@fn_ctxt fcx, @ty.t expected, @ast.pat pat) -> @ast.pat {
case (ast.pat_tag(?id, ?subpats, ?vdef_opt, ?ann)) {
auto t = demand(fcx, pat.span, expected, ann_to_type(ann));
// FIXME: This is probably more convoluted than it has to be.
// Refactor to use the type cache.
// Figure out the type parameters of the tag.
auto tag_id = option.get[ast.variant_def](vdef_opt)._0;
auto ty_params = fcx.ccx.item_ty_params.get(tag_id);
auto tpt = ty.lookup_generic_item_type(fcx.ccx.sess,
fcx.ccx.type_cache, tag_id);
auto ty_params = tpt._0;
// Take the type parameters out of the expected type.
auto ty_param_substs;
@ -1161,7 +1074,9 @@ fn demand_pat(&@fn_ctxt fcx, @ty.t expected, @ast.pat pat) -> @ast.pat {
// variants.
auto vdef = option.get[ast.variant_def](vdef_opt);
auto variant_ty = fcx.ccx.item_types.get(vdef._1);
auto variant_ty = ty.lookup_item_type(fcx.ccx.sess,
fcx.ccx.type_cache,
vdef._1)._1;
auto subpats_len = _vec.len[@ast.pat](subpats);
alt (variant_ty.struct) {
@ -1598,13 +1513,13 @@ fn check_pat(&@fn_ctxt fcx, @ast.pat pat) -> @ast.pat {
}
case (ast.pat_tag(?p, ?subpats, ?vdef_opt, _)) {
auto vdef = option.get[ast.variant_def](vdef_opt);
auto t = fcx.ccx.item_types.get(vdef._1);
auto t = ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache,
vdef._1)._1;
auto len = _vec.len[ast.ident](p.node.idents);
auto last_id = p.node.idents.(len - 1u);
auto ty_params = fcx.ccx.item_ty_params.get(vdef._0);
auto tag_ty = fcx.ccx.item_types.get(vdef._0);
auto tpt = tup(some(ty_params), tag_ty);
auto tpt = ty.lookup_item_type(fcx.ccx.sess, fcx.ccx.type_cache,
vdef._0);
auto ann = instantiate_path(fcx, p, tpt, pat.span);
alt (t.struct) {
@ -2673,15 +2588,17 @@ fn update_obj_fields(&@crate_ctxt ccx, @ast.item i) -> @crate_ctxt {
ret ccx;
}
fn check_crate(session.session sess, @ast.crate crate) -> @ast.crate {
type typecheck_result = tup(@ast.crate, ty.type_cache);
fn check_crate(session.session sess, @ast.crate crate) -> typecheck_result {
auto result = collect_item_types(sess, crate);
let vec[ast.obj_field] fields = vec();
auto ccx = @rec(sess=sess,
item_types=result._1,
type_cache=result._1,
item_items=result._2,
item_ty_params=result._3,
obj_fields=fields,
mutable next_var_id=0);
@ -2691,7 +2608,9 @@ fn check_crate(session.session sess, @ast.crate crate) -> @ast.crate {
fold_fn = bind check_fn(_,_,_,_),
fold_item_fn = bind check_item_fn(_,_,_,_,_,_,_)
with *fld);
ret fold.fold_crate[@crate_ctxt](ccx, fld, result._0);
auto crate_1 = fold.fold_crate[@crate_ctxt](ccx, fld, result._0);
ret tup(crate_1, ccx.type_cache);
}
//