Normalize the types of fields we project out of a struct or tuple struct.

Fixes #20954.
This commit is contained in:
Niko Matsakis 2015-01-11 13:52:40 -05:00
parent 2127e0d56d
commit 06e798a881
3 changed files with 111 additions and 28 deletions

View File

@ -2239,6 +2239,34 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
obligations.map_move(|o| self.register_predicate(o));
}
// Only for fields! Returns <none> for methods>
// Indifferent to privacy flags
pub fn lookup_field_ty(&self,
span: Span,
class_id: ast::DefId,
items: &[ty::field_ty],
fieldname: ast::Name,
substs: &subst::Substs<'tcx>)
-> Option<Ty<'tcx>>
{
let o_field = items.iter().find(|f| f.name == fieldname);
o_field.map(|f| ty::lookup_field_type(self.tcx(), class_id, f.id, substs))
.map(|t| self.normalize_associated_types_in(span, &t))
}
pub fn lookup_tup_field_ty(&self,
span: Span,
class_id: ast::DefId,
items: &[ty::field_ty],
idx: uint,
substs: &subst::Substs<'tcx>)
-> Option<Ty<'tcx>>
{
let o_field = if idx < items.len() { Some(&items[idx]) } else { None };
o_field.map(|f| ty::lookup_field_type(self.tcx(), class_id, f.id, substs))
.map(|t| self.normalize_associated_types_in(span, &t))
}
}
impl<'a, 'tcx> RegionScope for FnCtxt<'a, 'tcx> {
@ -2953,30 +2981,6 @@ pub fn impl_self_ty<'a, 'tcx>(fcx: &FnCtxt<'a, 'tcx>,
TypeAndSubsts { substs: substs, ty: substd_ty }
}
// Only for fields! Returns <none> for methods>
// Indifferent to privacy flags
pub fn lookup_field_ty<'tcx>(tcx: &ty::ctxt<'tcx>,
class_id: ast::DefId,
items: &[ty::field_ty],
fieldname: ast::Name,
substs: &subst::Substs<'tcx>)
-> Option<Ty<'tcx>> {
let o_field = items.iter().find(|f| f.name == fieldname);
o_field.map(|f| ty::lookup_field_type(tcx, class_id, f.id, substs))
}
pub fn lookup_tup_field_ty<'tcx>(tcx: &ty::ctxt<'tcx>,
class_id: ast::DefId,
items: &[ty::field_ty],
idx: uint,
substs: &subst::Substs<'tcx>)
-> Option<Ty<'tcx>> {
let o_field = if idx < items.len() { Some(&items[idx]) } else { None };
o_field.map(|f| ty::lookup_field_type(tcx, class_id, f.id, substs))
}
// Controls whether the arguments are automatically referenced. This is useful
// for overloaded binary and unary operators.
#[derive(Copy, PartialEq)]
@ -3398,8 +3402,8 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
ty::ty_struct(base_id, substs) => {
debug!("struct named {}", ppaux::ty_to_string(tcx, base_t));
let fields = ty::lookup_struct_fields(tcx, base_id);
lookup_field_ty(tcx, base_id, &fields[],
field.node.name, &(*substs))
fcx.lookup_field_ty(expr.span, base_id, &fields[],
field.node.name, &(*substs))
}
_ => None
}
@ -3461,8 +3465,8 @@ fn check_expr_with_unifier<'a, 'tcx, F>(fcx: &FnCtxt<'a, 'tcx>,
if tuple_like {
debug!("tuple struct named {}", ppaux::ty_to_string(tcx, base_t));
let fields = ty::lookup_struct_fields(tcx, base_id);
lookup_tup_field_ty(tcx, base_id, &fields[],
idx.node, &(*substs))
fcx.lookup_tup_field_ty(expr.span, base_id, &fields[],
idx.node, &(*substs))
} else {
None
}

View File

@ -0,0 +1,41 @@
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
// Test that we correctly normalize the type of a struct field
// which has an associated type.
pub trait UnifyKey {
type Value;
}
pub struct Node<K:UnifyKey> {
pub key: K,
pub value: K::Value,
}
fn foo<K : UnifyKey<Value=Option<V>>,V : Clone>(node: &Node<K>) -> Option<V> {
node.value.clone()
}
impl UnifyKey for i32 {
type Value = Option<u32>;
}
impl UnifyKey for u32 {
type Value = Option<i32>;
}
pub fn main() {
let node: Node<i32> = Node { key: 1, value: Some(22) };
assert_eq!(foo(&node), Some(22_u32));
let node: Node<u32> = Node { key: 1, value: Some(22) };
assert_eq!(foo(&node), Some(22_i32));
}

View File

@ -0,0 +1,38 @@
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
// Test that we correctly normalize the type of a struct field
// which has an associated type.
pub trait UnifyKey {
type Value;
}
pub struct Node<K:UnifyKey>(K, K::Value);
fn foo<K : UnifyKey<Value=Option<V>>,V : Clone>(node: &Node<K>) -> Option<V> {
node.1.clone()
}
impl UnifyKey for i32 {
type Value = Option<u32>;
}
impl UnifyKey for u32 {
type Value = Option<i32>;
}
pub fn main() {
let node: Node<i32> = Node(1, Some(22));
assert_eq!(foo(&node), Some(22_u32));
let node: Node<u32> = Node(1, Some(22));
assert_eq!(foo(&node), Some(22_i32));
}