diff options
| author | Patrick Walton <[email protected]> | 2011-03-30 17:23:25 -0700 |
|---|---|---|
| committer | Patrick Walton <[email protected]> | 2011-03-30 17:28:06 -0700 |
| commit | cc59cea8b055cd94279c167e99b1176751702d36 (patch) | |
| tree | 9a7c33168ac8845f81e62c43a6a1ae8ee46ace87 /src | |
| parent | Um, that'd be, align the word *before* retpc. Addresses point to the low part... (diff) | |
| download | rust-cc59cea8b055cd94279c167e99b1176751702d36.tar.xz rust-cc59cea8b055cd94279c167e99b1176751702d36.zip | |
rustc: Thread an item-to-type mapping throughout the typechecking and translation phases
Diffstat (limited to 'src')
| -rw-r--r-- | src/comp/driver/rustc.rs | 9 | ||||
| -rw-r--r-- | src/comp/front/ast.rs | 4 | ||||
| -rw-r--r-- | src/comp/front/creader.rs | 34 | ||||
| -rw-r--r-- | src/comp/front/parser.rs | 4 | ||||
| -rw-r--r-- | src/comp/middle/fold.rs | 10 | ||||
| -rw-r--r-- | src/comp/middle/metadata.rs | 28 | ||||
| -rw-r--r-- | src/comp/middle/resolve.rs | 99 | ||||
| -rw-r--r-- | src/comp/middle/trans.rs | 40 | ||||
| -rw-r--r-- | src/comp/middle/ty.rs | 55 | ||||
| -rw-r--r-- | src/comp/middle/typeck.rs | 465 |
10 files changed, 395 insertions, 353 deletions
diff --git a/src/comp/driver/rustc.rs b/src/comp/driver/rustc.rs index 69644425..4c6da407 100644 --- a/src/comp/driver/rustc.rs +++ b/src/comp/driver/rustc.rs @@ -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, diff --git a/src/comp/front/ast.rs b/src/comp/front/ast.rs index 9e7c3cc1..95a5cceb 100644 --- a/src/comp/front/ast.rs +++ b/src/comp/front/ast.rs @@ -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_]; diff --git a/src/comp/front/creader.rs b/src/comp/front/creader.rs index f79d6160..ea6581e0 100644 --- a/src/comp/front/creader.rs +++ b/src/comp/front/creader.rs @@ -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 { diff --git a/src/comp/front/parser.rs b/src/comp/front/parser.rs index 25c9f4a5..a247c824 100644 --- a/src/comp/front/parser.rs +++ b/src/comp/front/parser.rs @@ -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); } diff --git a/src/comp/middle/fold.rs b/src/comp/middle/fold.rs index 3756f987..d805c3cb 100644 --- a/src/comp/middle/fold.rs +++ b/src/comp/middle/fold.rs @@ -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. diff --git a/src/comp/middle/metadata.rs b/src/comp/middle/metadata.rs index 588a7d1c..4ace1f4a 100644 --- a/src/comp/middle/metadata.rs +++ b/src/comp/middle/metadata.rs @@ -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); } } diff --git a/src/comp/middle/resolve.rs b/src/comp/middle/resolve.rs index e171a600..1ceb7e66 100644 --- a/src/comp/middle/resolve.rs +++ b/src/comp/middle/resolve.rs @@ -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)))); } diff --git a/src/comp/middle/trans.rs b/src/comp/middle/trans.rs index 180c6afe..eea22185 100644 --- a/src/comp/middle/trans.rs +++ b/src/comp/middle/trans.rs @@ -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](), diff --git a/src/comp/middle/ty.rs b/src/comp/middle/ty.rs index d1d7c5e3..4e4cf34c 100644 --- a/src/comp/middle/ty.rs +++ b/src/comp/middle/ty.rs @@ -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; diff --git a/src/comp/middle/typeck.rs b/src/comp/middle/typeck.rs index daab4b49..bf1b6c39 100644 --- a/src/comp/middle/typeck.rs +++ b/src/comp/middle/typeck.rs @@ -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 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(ty_and_params._1, param_map); + 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)); + } + ret ty.replace_type_params(params_opt_and_ty._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); } // |