diff options
Diffstat (limited to 'crates/ra_hir_ty/src/infer')
-rw-r--r-- | crates/ra_hir_ty/src/infer/coerce.rs | 354 | ||||
-rw-r--r-- | crates/ra_hir_ty/src/infer/expr.rs | 686 | ||||
-rw-r--r-- | crates/ra_hir_ty/src/infer/pat.rs | 186 | ||||
-rw-r--r-- | crates/ra_hir_ty/src/infer/path.rs | 270 | ||||
-rw-r--r-- | crates/ra_hir_ty/src/infer/unify.rs | 162 |
5 files changed, 1658 insertions, 0 deletions
diff --git a/crates/ra_hir_ty/src/infer/coerce.rs b/crates/ra_hir_ty/src/infer/coerce.rs new file mode 100644 index 000000000..d66a21932 --- /dev/null +++ b/crates/ra_hir_ty/src/infer/coerce.rs | |||
@@ -0,0 +1,354 @@ | |||
1 | //! Coercion logic. Coercions are certain type conversions that can implicitly | ||
2 | //! happen in certain places, e.g. weakening `&mut` to `&` or deref coercions | ||
3 | //! like going from `&Vec<T>` to `&[T]`. | ||
4 | //! | ||
5 | //! See: https://doc.rust-lang.org/nomicon/coercions.html | ||
6 | |||
7 | use hir_def::{ | ||
8 | lang_item::LangItemTarget, | ||
9 | resolver::{HasResolver, Resolver}, | ||
10 | type_ref::Mutability, | ||
11 | AdtId, | ||
12 | }; | ||
13 | use rustc_hash::FxHashMap; | ||
14 | use test_utils::tested_by; | ||
15 | |||
16 | use crate::{autoderef, db::HirDatabase, Substs, TraitRef, Ty, TypeCtor, TypeWalk}; | ||
17 | |||
18 | use super::{InEnvironment, InferTy, InferenceContext, TypeVarValue}; | ||
19 | |||
20 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | ||
21 | /// Unify two types, but may coerce the first one to the second one | ||
22 | /// using "implicit coercion rules" if needed. | ||
23 | pub(super) fn coerce(&mut self, from_ty: &Ty, to_ty: &Ty) -> bool { | ||
24 | let from_ty = self.resolve_ty_shallow(from_ty).into_owned(); | ||
25 | let to_ty = self.resolve_ty_shallow(to_ty); | ||
26 | self.coerce_inner(from_ty, &to_ty) | ||
27 | } | ||
28 | |||
29 | /// Merge two types from different branches, with possible implicit coerce. | ||
30 | /// | ||
31 | /// Note that it is only possible that one type are coerced to another. | ||
32 | /// Coercing both types to another least upper bound type is not possible in rustc, | ||
33 | /// which will simply result in "incompatible types" error. | ||
34 | pub(super) fn coerce_merge_branch<'t>(&mut self, ty1: &Ty, ty2: &Ty) -> Ty { | ||
35 | if self.coerce(ty1, ty2) { | ||
36 | ty2.clone() | ||
37 | } else if self.coerce(ty2, ty1) { | ||
38 | ty1.clone() | ||
39 | } else { | ||
40 | tested_by!(coerce_merge_fail_fallback); | ||
41 | // For incompatible types, we use the latter one as result | ||
42 | // to be better recovery for `if` without `else`. | ||
43 | ty2.clone() | ||
44 | } | ||
45 | } | ||
46 | |||
47 | pub(super) fn init_coerce_unsized_map( | ||
48 | db: &'a D, | ||
49 | resolver: &Resolver, | ||
50 | ) -> FxHashMap<(TypeCtor, TypeCtor), usize> { | ||
51 | let krate = resolver.krate().unwrap(); | ||
52 | let impls = match db.lang_item(krate.into(), "coerce_unsized".into()) { | ||
53 | Some(LangItemTarget::TraitId(trait_)) => { | ||
54 | db.impls_for_trait(krate.into(), trait_.into()) | ||
55 | } | ||
56 | _ => return FxHashMap::default(), | ||
57 | }; | ||
58 | |||
59 | impls | ||
60 | .iter() | ||
61 | .filter_map(|&impl_id| { | ||
62 | let impl_data = db.impl_data(impl_id); | ||
63 | let resolver = impl_id.resolver(db); | ||
64 | let target_ty = Ty::from_hir(db, &resolver, &impl_data.target_type); | ||
65 | |||
66 | // `CoerseUnsized` has one generic parameter for the target type. | ||
67 | let trait_ref = TraitRef::from_hir( | ||
68 | db, | ||
69 | &resolver, | ||
70 | impl_data.target_trait.as_ref()?, | ||
71 | Some(target_ty), | ||
72 | )?; | ||
73 | let cur_from_ty = trait_ref.substs.0.get(0)?; | ||
74 | let cur_to_ty = trait_ref.substs.0.get(1)?; | ||
75 | |||
76 | match (&cur_from_ty, cur_to_ty) { | ||
77 | (ty_app!(ctor1, st1), ty_app!(ctor2, st2)) => { | ||
78 | // FIXME: We return the first non-equal bound as the type parameter to coerce to unsized type. | ||
79 | // This works for smart-pointer-like coercion, which covers all impls from std. | ||
80 | st1.iter().zip(st2.iter()).enumerate().find_map(|(i, (ty1, ty2))| { | ||
81 | match (ty1, ty2) { | ||
82 | (Ty::Param { idx: p1, .. }, Ty::Param { idx: p2, .. }) | ||
83 | if p1 != p2 => | ||
84 | { | ||
85 | Some(((*ctor1, *ctor2), i)) | ||
86 | } | ||
87 | _ => None, | ||
88 | } | ||
89 | }) | ||
90 | } | ||
91 | _ => None, | ||
92 | } | ||
93 | }) | ||
94 | .collect() | ||
95 | } | ||
96 | |||
97 | fn coerce_inner(&mut self, mut from_ty: Ty, to_ty: &Ty) -> bool { | ||
98 | match (&from_ty, to_ty) { | ||
99 | // Never type will make type variable to fallback to Never Type instead of Unknown. | ||
100 | (ty_app!(TypeCtor::Never), Ty::Infer(InferTy::TypeVar(tv))) => { | ||
101 | let var = self.new_maybe_never_type_var(); | ||
102 | self.var_unification_table.union_value(*tv, TypeVarValue::Known(var)); | ||
103 | return true; | ||
104 | } | ||
105 | (ty_app!(TypeCtor::Never), _) => return true, | ||
106 | |||
107 | // Trivial cases, this should go after `never` check to | ||
108 | // avoid infer result type to be never | ||
109 | _ => { | ||
110 | if self.unify_inner_trivial(&from_ty, &to_ty) { | ||
111 | return true; | ||
112 | } | ||
113 | } | ||
114 | } | ||
115 | |||
116 | // Pointer weakening and function to pointer | ||
117 | match (&mut from_ty, to_ty) { | ||
118 | // `*mut T`, `&mut T, `&T`` -> `*const T` | ||
119 | // `&mut T` -> `&T` | ||
120 | // `&mut T` -> `*mut T` | ||
121 | (ty_app!(c1@TypeCtor::RawPtr(_)), ty_app!(c2@TypeCtor::RawPtr(Mutability::Shared))) | ||
122 | | (ty_app!(c1@TypeCtor::Ref(_)), ty_app!(c2@TypeCtor::RawPtr(Mutability::Shared))) | ||
123 | | (ty_app!(c1@TypeCtor::Ref(_)), ty_app!(c2@TypeCtor::Ref(Mutability::Shared))) | ||
124 | | (ty_app!(c1@TypeCtor::Ref(Mutability::Mut)), ty_app!(c2@TypeCtor::RawPtr(_))) => { | ||
125 | *c1 = *c2; | ||
126 | } | ||
127 | |||
128 | // Illegal mutablity conversion | ||
129 | ( | ||
130 | ty_app!(TypeCtor::RawPtr(Mutability::Shared)), | ||
131 | ty_app!(TypeCtor::RawPtr(Mutability::Mut)), | ||
132 | ) | ||
133 | | ( | ||
134 | ty_app!(TypeCtor::Ref(Mutability::Shared)), | ||
135 | ty_app!(TypeCtor::Ref(Mutability::Mut)), | ||
136 | ) => return false, | ||
137 | |||
138 | // `{function_type}` -> `fn()` | ||
139 | (ty_app!(TypeCtor::FnDef(_)), ty_app!(TypeCtor::FnPtr { .. })) => { | ||
140 | match from_ty.callable_sig(self.db) { | ||
141 | None => return false, | ||
142 | Some(sig) => { | ||
143 | let num_args = sig.params_and_return.len() as u16 - 1; | ||
144 | from_ty = | ||
145 | Ty::apply(TypeCtor::FnPtr { num_args }, Substs(sig.params_and_return)); | ||
146 | } | ||
147 | } | ||
148 | } | ||
149 | |||
150 | _ => {} | ||
151 | } | ||
152 | |||
153 | if let Some(ret) = self.try_coerce_unsized(&from_ty, &to_ty) { | ||
154 | return ret; | ||
155 | } | ||
156 | |||
157 | // Auto Deref if cannot coerce | ||
158 | match (&from_ty, to_ty) { | ||
159 | // FIXME: DerefMut | ||
160 | (ty_app!(TypeCtor::Ref(_), st1), ty_app!(TypeCtor::Ref(_), st2)) => { | ||
161 | self.unify_autoderef_behind_ref(&st1[0], &st2[0]) | ||
162 | } | ||
163 | |||
164 | // Otherwise, normal unify | ||
165 | _ => self.unify(&from_ty, to_ty), | ||
166 | } | ||
167 | } | ||
168 | |||
169 | /// Coerce a type using `from_ty: CoerceUnsized<ty_ty>` | ||
170 | /// | ||
171 | /// See: https://doc.rust-lang.org/nightly/std/marker/trait.CoerceUnsized.html | ||
172 | fn try_coerce_unsized(&mut self, from_ty: &Ty, to_ty: &Ty) -> Option<bool> { | ||
173 | let (ctor1, st1, ctor2, st2) = match (from_ty, to_ty) { | ||
174 | (ty_app!(ctor1, st1), ty_app!(ctor2, st2)) => (ctor1, st1, ctor2, st2), | ||
175 | _ => return None, | ||
176 | }; | ||
177 | |||
178 | let coerce_generic_index = *self.coerce_unsized_map.get(&(*ctor1, *ctor2))?; | ||
179 | |||
180 | // Check `Unsize` first | ||
181 | match self.check_unsize_and_coerce( | ||
182 | st1.0.get(coerce_generic_index)?, | ||
183 | st2.0.get(coerce_generic_index)?, | ||
184 | 0, | ||
185 | ) { | ||
186 | Some(true) => {} | ||
187 | ret => return ret, | ||
188 | } | ||
189 | |||
190 | let ret = st1 | ||
191 | .iter() | ||
192 | .zip(st2.iter()) | ||
193 | .enumerate() | ||
194 | .filter(|&(idx, _)| idx != coerce_generic_index) | ||
195 | .all(|(_, (ty1, ty2))| self.unify(ty1, ty2)); | ||
196 | |||
197 | Some(ret) | ||
198 | } | ||
199 | |||
200 | /// Check if `from_ty: Unsize<to_ty>`, and coerce to `to_ty` if it holds. | ||
201 | /// | ||
202 | /// It should not be directly called. It is only used by `try_coerce_unsized`. | ||
203 | /// | ||
204 | /// See: https://doc.rust-lang.org/nightly/std/marker/trait.Unsize.html | ||
205 | fn check_unsize_and_coerce(&mut self, from_ty: &Ty, to_ty: &Ty, depth: usize) -> Option<bool> { | ||
206 | if depth > 1000 { | ||
207 | panic!("Infinite recursion in coercion"); | ||
208 | } | ||
209 | |||
210 | match (&from_ty, &to_ty) { | ||
211 | // `[T; N]` -> `[T]` | ||
212 | (ty_app!(TypeCtor::Array, st1), ty_app!(TypeCtor::Slice, st2)) => { | ||
213 | Some(self.unify(&st1[0], &st2[0])) | ||
214 | } | ||
215 | |||
216 | // `T` -> `dyn Trait` when `T: Trait` | ||
217 | (_, Ty::Dyn(_)) => { | ||
218 | // FIXME: Check predicates | ||
219 | Some(true) | ||
220 | } | ||
221 | |||
222 | // `(..., T)` -> `(..., U)` when `T: Unsize<U>` | ||
223 | ( | ||
224 | ty_app!(TypeCtor::Tuple { cardinality: len1 }, st1), | ||
225 | ty_app!(TypeCtor::Tuple { cardinality: len2 }, st2), | ||
226 | ) => { | ||
227 | if len1 != len2 || *len1 == 0 { | ||
228 | return None; | ||
229 | } | ||
230 | |||
231 | match self.check_unsize_and_coerce( | ||
232 | st1.last().unwrap(), | ||
233 | st2.last().unwrap(), | ||
234 | depth + 1, | ||
235 | ) { | ||
236 | Some(true) => {} | ||
237 | ret => return ret, | ||
238 | } | ||
239 | |||
240 | let ret = st1[..st1.len() - 1] | ||
241 | .iter() | ||
242 | .zip(&st2[..st2.len() - 1]) | ||
243 | .all(|(ty1, ty2)| self.unify(ty1, ty2)); | ||
244 | |||
245 | Some(ret) | ||
246 | } | ||
247 | |||
248 | // Foo<..., T, ...> is Unsize<Foo<..., U, ...>> if: | ||
249 | // - T: Unsize<U> | ||
250 | // - Foo is a struct | ||
251 | // - Only the last field of Foo has a type involving T | ||
252 | // - T is not part of the type of any other fields | ||
253 | // - Bar<T>: Unsize<Bar<U>>, if the last field of Foo has type Bar<T> | ||
254 | ( | ||
255 | ty_app!(TypeCtor::Adt(AdtId::StructId(struct1)), st1), | ||
256 | ty_app!(TypeCtor::Adt(AdtId::StructId(struct2)), st2), | ||
257 | ) if struct1 == struct2 => { | ||
258 | let field_tys = self.db.field_types((*struct1).into()); | ||
259 | let struct_data = self.db.struct_data(*struct1); | ||
260 | |||
261 | let mut fields = struct_data.variant_data.fields().iter(); | ||
262 | let (last_field_id, _data) = fields.next_back()?; | ||
263 | |||
264 | // Get the generic parameter involved in the last field. | ||
265 | let unsize_generic_index = { | ||
266 | let mut index = None; | ||
267 | let mut multiple_param = false; | ||
268 | field_tys[last_field_id].walk(&mut |ty| match ty { | ||
269 | &Ty::Param { idx, .. } => { | ||
270 | if index.is_none() { | ||
271 | index = Some(idx); | ||
272 | } else if Some(idx) != index { | ||
273 | multiple_param = true; | ||
274 | } | ||
275 | } | ||
276 | _ => {} | ||
277 | }); | ||
278 | |||
279 | if multiple_param { | ||
280 | return None; | ||
281 | } | ||
282 | index? | ||
283 | }; | ||
284 | |||
285 | // Check other fields do not involve it. | ||
286 | let mut multiple_used = false; | ||
287 | fields.for_each(|(field_id, _data)| { | ||
288 | field_tys[field_id].walk(&mut |ty| match ty { | ||
289 | &Ty::Param { idx, .. } if idx == unsize_generic_index => { | ||
290 | multiple_used = true | ||
291 | } | ||
292 | _ => {} | ||
293 | }) | ||
294 | }); | ||
295 | if multiple_used { | ||
296 | return None; | ||
297 | } | ||
298 | |||
299 | let unsize_generic_index = unsize_generic_index as usize; | ||
300 | |||
301 | // Check `Unsize` first | ||
302 | match self.check_unsize_and_coerce( | ||
303 | st1.get(unsize_generic_index)?, | ||
304 | st2.get(unsize_generic_index)?, | ||
305 | depth + 1, | ||
306 | ) { | ||
307 | Some(true) => {} | ||
308 | ret => return ret, | ||
309 | } | ||
310 | |||
311 | // Then unify other parameters | ||
312 | let ret = st1 | ||
313 | .iter() | ||
314 | .zip(st2.iter()) | ||
315 | .enumerate() | ||
316 | .filter(|&(idx, _)| idx != unsize_generic_index) | ||
317 | .all(|(_, (ty1, ty2))| self.unify(ty1, ty2)); | ||
318 | |||
319 | Some(ret) | ||
320 | } | ||
321 | |||
322 | _ => None, | ||
323 | } | ||
324 | } | ||
325 | |||
326 | /// Unify `from_ty` to `to_ty` with optional auto Deref | ||
327 | /// | ||
328 | /// Note that the parameters are already stripped the outer reference. | ||
329 | fn unify_autoderef_behind_ref(&mut self, from_ty: &Ty, to_ty: &Ty) -> bool { | ||
330 | let canonicalized = self.canonicalizer().canonicalize_ty(from_ty.clone()); | ||
331 | let to_ty = self.resolve_ty_shallow(&to_ty); | ||
332 | // FIXME: Auto DerefMut | ||
333 | for derefed_ty in autoderef::autoderef( | ||
334 | self.db, | ||
335 | self.resolver.krate(), | ||
336 | InEnvironment { | ||
337 | value: canonicalized.value.clone(), | ||
338 | environment: self.trait_env.clone(), | ||
339 | }, | ||
340 | ) { | ||
341 | let derefed_ty = canonicalized.decanonicalize_ty(derefed_ty.value); | ||
342 | match (&*self.resolve_ty_shallow(&derefed_ty), &*to_ty) { | ||
343 | // Stop when constructor matches. | ||
344 | (ty_app!(from_ctor, st1), ty_app!(to_ctor, st2)) if from_ctor == to_ctor => { | ||
345 | // It will not recurse to `coerce`. | ||
346 | return self.unify_substs(st1, st2, 0); | ||
347 | } | ||
348 | _ => {} | ||
349 | } | ||
350 | } | ||
351 | |||
352 | false | ||
353 | } | ||
354 | } | ||
diff --git a/crates/ra_hir_ty/src/infer/expr.rs b/crates/ra_hir_ty/src/infer/expr.rs new file mode 100644 index 000000000..2f9ca4bbb --- /dev/null +++ b/crates/ra_hir_ty/src/infer/expr.rs | |||
@@ -0,0 +1,686 @@ | |||
1 | //! Type inference for expressions. | ||
2 | |||
3 | use std::iter::{repeat, repeat_with}; | ||
4 | use std::sync::Arc; | ||
5 | |||
6 | use hir_def::{ | ||
7 | builtin_type::Signedness, | ||
8 | expr::{Array, BinaryOp, Expr, ExprId, Literal, Statement, UnaryOp}, | ||
9 | generics::GenericParams, | ||
10 | path::{GenericArg, GenericArgs}, | ||
11 | resolver::resolver_for_expr, | ||
12 | AdtId, ContainerId, Lookup, StructFieldId, | ||
13 | }; | ||
14 | use hir_expand::name::{self, Name}; | ||
15 | |||
16 | use crate::{ | ||
17 | autoderef, db::HirDatabase, method_resolution, op, traits::InEnvironment, utils::variant_data, | ||
18 | CallableDef, InferTy, IntTy, Mutability, Obligation, ProjectionPredicate, ProjectionTy, Substs, | ||
19 | TraitRef, Ty, TypeCtor, TypeWalk, Uncertain, | ||
20 | }; | ||
21 | |||
22 | use super::{BindingMode, Expectation, InferenceContext, InferenceDiagnostic, TypeMismatch}; | ||
23 | |||
24 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | ||
25 | pub(super) fn infer_expr(&mut self, tgt_expr: ExprId, expected: &Expectation) -> Ty { | ||
26 | let ty = self.infer_expr_inner(tgt_expr, expected); | ||
27 | let could_unify = self.unify(&ty, &expected.ty); | ||
28 | if !could_unify { | ||
29 | self.result.type_mismatches.insert( | ||
30 | tgt_expr, | ||
31 | TypeMismatch { expected: expected.ty.clone(), actual: ty.clone() }, | ||
32 | ); | ||
33 | } | ||
34 | let ty = self.resolve_ty_as_possible(&mut vec![], ty); | ||
35 | ty | ||
36 | } | ||
37 | |||
38 | /// Infer type of expression with possibly implicit coerce to the expected type. | ||
39 | /// Return the type after possible coercion. | ||
40 | fn infer_expr_coerce(&mut self, expr: ExprId, expected: &Expectation) -> Ty { | ||
41 | let ty = self.infer_expr_inner(expr, &expected); | ||
42 | let ty = if !self.coerce(&ty, &expected.ty) { | ||
43 | self.result | ||
44 | .type_mismatches | ||
45 | .insert(expr, TypeMismatch { expected: expected.ty.clone(), actual: ty.clone() }); | ||
46 | // Return actual type when type mismatch. | ||
47 | // This is needed for diagnostic when return type mismatch. | ||
48 | ty | ||
49 | } else if expected.ty == Ty::Unknown { | ||
50 | ty | ||
51 | } else { | ||
52 | expected.ty.clone() | ||
53 | }; | ||
54 | |||
55 | self.resolve_ty_as_possible(&mut vec![], ty) | ||
56 | } | ||
57 | |||
58 | fn infer_expr_inner(&mut self, tgt_expr: ExprId, expected: &Expectation) -> Ty { | ||
59 | let body = Arc::clone(&self.body); // avoid borrow checker problem | ||
60 | let ty = match &body[tgt_expr] { | ||
61 | Expr::Missing => Ty::Unknown, | ||
62 | Expr::If { condition, then_branch, else_branch } => { | ||
63 | // if let is desugared to match, so this is always simple if | ||
64 | self.infer_expr(*condition, &Expectation::has_type(Ty::simple(TypeCtor::Bool))); | ||
65 | |||
66 | let then_ty = self.infer_expr_inner(*then_branch, &expected); | ||
67 | let else_ty = match else_branch { | ||
68 | Some(else_branch) => self.infer_expr_inner(*else_branch, &expected), | ||
69 | None => Ty::unit(), | ||
70 | }; | ||
71 | |||
72 | self.coerce_merge_branch(&then_ty, &else_ty) | ||
73 | } | ||
74 | Expr::Block { statements, tail } => self.infer_block(statements, *tail, expected), | ||
75 | Expr::TryBlock { body } => { | ||
76 | let _inner = self.infer_expr(*body, expected); | ||
77 | // FIXME should be std::result::Result<{inner}, _> | ||
78 | Ty::Unknown | ||
79 | } | ||
80 | Expr::Loop { body } => { | ||
81 | self.infer_expr(*body, &Expectation::has_type(Ty::unit())); | ||
82 | // FIXME handle break with value | ||
83 | Ty::simple(TypeCtor::Never) | ||
84 | } | ||
85 | Expr::While { condition, body } => { | ||
86 | // while let is desugared to a match loop, so this is always simple while | ||
87 | self.infer_expr(*condition, &Expectation::has_type(Ty::simple(TypeCtor::Bool))); | ||
88 | self.infer_expr(*body, &Expectation::has_type(Ty::unit())); | ||
89 | Ty::unit() | ||
90 | } | ||
91 | Expr::For { iterable, body, pat } => { | ||
92 | let iterable_ty = self.infer_expr(*iterable, &Expectation::none()); | ||
93 | |||
94 | let pat_ty = match self.resolve_into_iter_item() { | ||
95 | Some(into_iter_item_alias) => { | ||
96 | let pat_ty = self.new_type_var(); | ||
97 | let projection = ProjectionPredicate { | ||
98 | ty: pat_ty.clone(), | ||
99 | projection_ty: ProjectionTy { | ||
100 | associated_ty: into_iter_item_alias, | ||
101 | parameters: Substs::single(iterable_ty), | ||
102 | }, | ||
103 | }; | ||
104 | self.obligations.push(Obligation::Projection(projection)); | ||
105 | self.resolve_ty_as_possible(&mut vec![], pat_ty) | ||
106 | } | ||
107 | None => Ty::Unknown, | ||
108 | }; | ||
109 | |||
110 | self.infer_pat(*pat, &pat_ty, BindingMode::default()); | ||
111 | self.infer_expr(*body, &Expectation::has_type(Ty::unit())); | ||
112 | Ty::unit() | ||
113 | } | ||
114 | Expr::Lambda { body, args, arg_types } => { | ||
115 | assert_eq!(args.len(), arg_types.len()); | ||
116 | |||
117 | let mut sig_tys = Vec::new(); | ||
118 | |||
119 | for (arg_pat, arg_type) in args.iter().zip(arg_types.iter()) { | ||
120 | let expected = if let Some(type_ref) = arg_type { | ||
121 | self.make_ty(type_ref) | ||
122 | } else { | ||
123 | Ty::Unknown | ||
124 | }; | ||
125 | let arg_ty = self.infer_pat(*arg_pat, &expected, BindingMode::default()); | ||
126 | sig_tys.push(arg_ty); | ||
127 | } | ||
128 | |||
129 | // add return type | ||
130 | let ret_ty = self.new_type_var(); | ||
131 | sig_tys.push(ret_ty.clone()); | ||
132 | let sig_ty = Ty::apply( | ||
133 | TypeCtor::FnPtr { num_args: sig_tys.len() as u16 - 1 }, | ||
134 | Substs(sig_tys.into()), | ||
135 | ); | ||
136 | let closure_ty = Ty::apply_one( | ||
137 | TypeCtor::Closure { def: self.owner.into(), expr: tgt_expr }, | ||
138 | sig_ty, | ||
139 | ); | ||
140 | |||
141 | // Eagerly try to relate the closure type with the expected | ||
142 | // type, otherwise we often won't have enough information to | ||
143 | // infer the body. | ||
144 | self.coerce(&closure_ty, &expected.ty); | ||
145 | |||
146 | self.infer_expr(*body, &Expectation::has_type(ret_ty)); | ||
147 | closure_ty | ||
148 | } | ||
149 | Expr::Call { callee, args } => { | ||
150 | let callee_ty = self.infer_expr(*callee, &Expectation::none()); | ||
151 | let (param_tys, ret_ty) = match callee_ty.callable_sig(self.db) { | ||
152 | Some(sig) => (sig.params().to_vec(), sig.ret().clone()), | ||
153 | None => { | ||
154 | // Not callable | ||
155 | // FIXME: report an error | ||
156 | (Vec::new(), Ty::Unknown) | ||
157 | } | ||
158 | }; | ||
159 | self.register_obligations_for_call(&callee_ty); | ||
160 | self.check_call_arguments(args, ¶m_tys); | ||
161 | let ret_ty = self.normalize_associated_types_in(ret_ty); | ||
162 | ret_ty | ||
163 | } | ||
164 | Expr::MethodCall { receiver, args, method_name, generic_args } => self | ||
165 | .infer_method_call(tgt_expr, *receiver, &args, &method_name, generic_args.as_ref()), | ||
166 | Expr::Match { expr, arms } => { | ||
167 | let input_ty = self.infer_expr(*expr, &Expectation::none()); | ||
168 | |||
169 | let mut result_ty = self.new_maybe_never_type_var(); | ||
170 | |||
171 | for arm in arms { | ||
172 | for &pat in &arm.pats { | ||
173 | let _pat_ty = self.infer_pat(pat, &input_ty, BindingMode::default()); | ||
174 | } | ||
175 | if let Some(guard_expr) = arm.guard { | ||
176 | self.infer_expr( | ||
177 | guard_expr, | ||
178 | &Expectation::has_type(Ty::simple(TypeCtor::Bool)), | ||
179 | ); | ||
180 | } | ||
181 | |||
182 | let arm_ty = self.infer_expr_inner(arm.expr, &expected); | ||
183 | result_ty = self.coerce_merge_branch(&result_ty, &arm_ty); | ||
184 | } | ||
185 | |||
186 | result_ty | ||
187 | } | ||
188 | Expr::Path(p) => { | ||
189 | // FIXME this could be more efficient... | ||
190 | let resolver = resolver_for_expr(self.db, self.owner.into(), tgt_expr); | ||
191 | self.infer_path(&resolver, p, tgt_expr.into()).unwrap_or(Ty::Unknown) | ||
192 | } | ||
193 | Expr::Continue => Ty::simple(TypeCtor::Never), | ||
194 | Expr::Break { expr } => { | ||
195 | if let Some(expr) = expr { | ||
196 | // FIXME handle break with value | ||
197 | self.infer_expr(*expr, &Expectation::none()); | ||
198 | } | ||
199 | Ty::simple(TypeCtor::Never) | ||
200 | } | ||
201 | Expr::Return { expr } => { | ||
202 | if let Some(expr) = expr { | ||
203 | self.infer_expr(*expr, &Expectation::has_type(self.return_ty.clone())); | ||
204 | } | ||
205 | Ty::simple(TypeCtor::Never) | ||
206 | } | ||
207 | Expr::RecordLit { path, fields, spread } => { | ||
208 | let (ty, def_id) = self.resolve_variant(path.as_ref()); | ||
209 | if let Some(variant) = def_id { | ||
210 | self.write_variant_resolution(tgt_expr.into(), variant); | ||
211 | } | ||
212 | |||
213 | self.unify(&ty, &expected.ty); | ||
214 | |||
215 | let substs = ty.substs().unwrap_or_else(Substs::empty); | ||
216 | let field_types = | ||
217 | def_id.map(|it| self.db.field_types(it.into())).unwrap_or_default(); | ||
218 | let variant_data = def_id.map(|it| variant_data(self.db, it)); | ||
219 | for (field_idx, field) in fields.iter().enumerate() { | ||
220 | let field_def = | ||
221 | variant_data.as_ref().and_then(|it| match it.field(&field.name) { | ||
222 | Some(local_id) => { | ||
223 | Some(StructFieldId { parent: def_id.unwrap(), local_id }) | ||
224 | } | ||
225 | None => { | ||
226 | self.push_diagnostic(InferenceDiagnostic::NoSuchField { | ||
227 | expr: tgt_expr, | ||
228 | field: field_idx, | ||
229 | }); | ||
230 | None | ||
231 | } | ||
232 | }); | ||
233 | if let Some(field_def) = field_def { | ||
234 | self.result.record_field_resolutions.insert(field.expr, field_def); | ||
235 | } | ||
236 | let field_ty = field_def | ||
237 | .map_or(Ty::Unknown, |it| field_types[it.local_id].clone()) | ||
238 | .subst(&substs); | ||
239 | self.infer_expr_coerce(field.expr, &Expectation::has_type(field_ty)); | ||
240 | } | ||
241 | if let Some(expr) = spread { | ||
242 | self.infer_expr(*expr, &Expectation::has_type(ty.clone())); | ||
243 | } | ||
244 | ty | ||
245 | } | ||
246 | Expr::Field { expr, name } => { | ||
247 | let receiver_ty = self.infer_expr(*expr, &Expectation::none()); | ||
248 | let canonicalized = self.canonicalizer().canonicalize_ty(receiver_ty); | ||
249 | let ty = autoderef::autoderef( | ||
250 | self.db, | ||
251 | self.resolver.krate(), | ||
252 | InEnvironment { | ||
253 | value: canonicalized.value.clone(), | ||
254 | environment: self.trait_env.clone(), | ||
255 | }, | ||
256 | ) | ||
257 | .find_map(|derefed_ty| match canonicalized.decanonicalize_ty(derefed_ty.value) { | ||
258 | Ty::Apply(a_ty) => match a_ty.ctor { | ||
259 | TypeCtor::Tuple { .. } => name | ||
260 | .as_tuple_index() | ||
261 | .and_then(|idx| a_ty.parameters.0.get(idx).cloned()), | ||
262 | TypeCtor::Adt(AdtId::StructId(s)) => { | ||
263 | self.db.struct_data(s).variant_data.field(name).map(|local_id| { | ||
264 | let field = StructFieldId { parent: s.into(), local_id }.into(); | ||
265 | self.write_field_resolution(tgt_expr, field); | ||
266 | self.db.field_types(s.into())[field.local_id] | ||
267 | .clone() | ||
268 | .subst(&a_ty.parameters) | ||
269 | }) | ||
270 | } | ||
271 | // FIXME: | ||
272 | TypeCtor::Adt(AdtId::UnionId(_)) => None, | ||
273 | _ => None, | ||
274 | }, | ||
275 | _ => None, | ||
276 | }) | ||
277 | .unwrap_or(Ty::Unknown); | ||
278 | let ty = self.insert_type_vars(ty); | ||
279 | self.normalize_associated_types_in(ty) | ||
280 | } | ||
281 | Expr::Await { expr } => { | ||
282 | let inner_ty = self.infer_expr(*expr, &Expectation::none()); | ||
283 | let ty = match self.resolve_future_future_output() { | ||
284 | Some(future_future_output_alias) => { | ||
285 | let ty = self.new_type_var(); | ||
286 | let projection = ProjectionPredicate { | ||
287 | ty: ty.clone(), | ||
288 | projection_ty: ProjectionTy { | ||
289 | associated_ty: future_future_output_alias, | ||
290 | parameters: Substs::single(inner_ty), | ||
291 | }, | ||
292 | }; | ||
293 | self.obligations.push(Obligation::Projection(projection)); | ||
294 | self.resolve_ty_as_possible(&mut vec![], ty) | ||
295 | } | ||
296 | None => Ty::Unknown, | ||
297 | }; | ||
298 | ty | ||
299 | } | ||
300 | Expr::Try { expr } => { | ||
301 | let inner_ty = self.infer_expr(*expr, &Expectation::none()); | ||
302 | let ty = match self.resolve_ops_try_ok() { | ||
303 | Some(ops_try_ok_alias) => { | ||
304 | let ty = self.new_type_var(); | ||
305 | let projection = ProjectionPredicate { | ||
306 | ty: ty.clone(), | ||
307 | projection_ty: ProjectionTy { | ||
308 | associated_ty: ops_try_ok_alias, | ||
309 | parameters: Substs::single(inner_ty), | ||
310 | }, | ||
311 | }; | ||
312 | self.obligations.push(Obligation::Projection(projection)); | ||
313 | self.resolve_ty_as_possible(&mut vec![], ty) | ||
314 | } | ||
315 | None => Ty::Unknown, | ||
316 | }; | ||
317 | ty | ||
318 | } | ||
319 | Expr::Cast { expr, type_ref } => { | ||
320 | let _inner_ty = self.infer_expr(*expr, &Expectation::none()); | ||
321 | let cast_ty = self.make_ty(type_ref); | ||
322 | // FIXME check the cast... | ||
323 | cast_ty | ||
324 | } | ||
325 | Expr::Ref { expr, mutability } => { | ||
326 | let expectation = | ||
327 | if let Some((exp_inner, exp_mutability)) = &expected.ty.as_reference() { | ||
328 | if *exp_mutability == Mutability::Mut && *mutability == Mutability::Shared { | ||
329 | // FIXME: throw type error - expected mut reference but found shared ref, | ||
330 | // which cannot be coerced | ||
331 | } | ||
332 | Expectation::has_type(Ty::clone(exp_inner)) | ||
333 | } else { | ||
334 | Expectation::none() | ||
335 | }; | ||
336 | // FIXME reference coercions etc. | ||
337 | let inner_ty = self.infer_expr(*expr, &expectation); | ||
338 | Ty::apply_one(TypeCtor::Ref(*mutability), inner_ty) | ||
339 | } | ||
340 | Expr::Box { expr } => { | ||
341 | let inner_ty = self.infer_expr(*expr, &Expectation::none()); | ||
342 | if let Some(box_) = self.resolve_boxed_box() { | ||
343 | Ty::apply_one(TypeCtor::Adt(box_), inner_ty) | ||
344 | } else { | ||
345 | Ty::Unknown | ||
346 | } | ||
347 | } | ||
348 | Expr::UnaryOp { expr, op } => { | ||
349 | let inner_ty = self.infer_expr(*expr, &Expectation::none()); | ||
350 | match op { | ||
351 | UnaryOp::Deref => match self.resolver.krate() { | ||
352 | Some(krate) => { | ||
353 | let canonicalized = self.canonicalizer().canonicalize_ty(inner_ty); | ||
354 | match autoderef::deref( | ||
355 | self.db, | ||
356 | krate, | ||
357 | InEnvironment { | ||
358 | value: &canonicalized.value, | ||
359 | environment: self.trait_env.clone(), | ||
360 | }, | ||
361 | ) { | ||
362 | Some(derefed_ty) => { | ||
363 | canonicalized.decanonicalize_ty(derefed_ty.value) | ||
364 | } | ||
365 | None => Ty::Unknown, | ||
366 | } | ||
367 | } | ||
368 | None => Ty::Unknown, | ||
369 | }, | ||
370 | UnaryOp::Neg => { | ||
371 | match &inner_ty { | ||
372 | Ty::Apply(a_ty) => match a_ty.ctor { | ||
373 | TypeCtor::Int(Uncertain::Unknown) | ||
374 | | TypeCtor::Int(Uncertain::Known(IntTy { | ||
375 | signedness: Signedness::Signed, | ||
376 | .. | ||
377 | })) | ||
378 | | TypeCtor::Float(..) => inner_ty, | ||
379 | _ => Ty::Unknown, | ||
380 | }, | ||
381 | Ty::Infer(InferTy::IntVar(..)) | Ty::Infer(InferTy::FloatVar(..)) => { | ||
382 | inner_ty | ||
383 | } | ||
384 | // FIXME: resolve ops::Neg trait | ||
385 | _ => Ty::Unknown, | ||
386 | } | ||
387 | } | ||
388 | UnaryOp::Not => { | ||
389 | match &inner_ty { | ||
390 | Ty::Apply(a_ty) => match a_ty.ctor { | ||
391 | TypeCtor::Bool | TypeCtor::Int(_) => inner_ty, | ||
392 | _ => Ty::Unknown, | ||
393 | }, | ||
394 | Ty::Infer(InferTy::IntVar(..)) => inner_ty, | ||
395 | // FIXME: resolve ops::Not trait for inner_ty | ||
396 | _ => Ty::Unknown, | ||
397 | } | ||
398 | } | ||
399 | } | ||
400 | } | ||
401 | Expr::BinaryOp { lhs, rhs, op } => match op { | ||
402 | Some(op) => { | ||
403 | let lhs_expectation = match op { | ||
404 | BinaryOp::LogicOp(..) => Expectation::has_type(Ty::simple(TypeCtor::Bool)), | ||
405 | _ => Expectation::none(), | ||
406 | }; | ||
407 | let lhs_ty = self.infer_expr(*lhs, &lhs_expectation); | ||
408 | // FIXME: find implementation of trait corresponding to operation | ||
409 | // symbol and resolve associated `Output` type | ||
410 | let rhs_expectation = op::binary_op_rhs_expectation(*op, lhs_ty); | ||
411 | let rhs_ty = self.infer_expr(*rhs, &Expectation::has_type(rhs_expectation)); | ||
412 | |||
413 | // FIXME: similar as above, return ty is often associated trait type | ||
414 | op::binary_op_return_ty(*op, rhs_ty) | ||
415 | } | ||
416 | _ => Ty::Unknown, | ||
417 | }, | ||
418 | Expr::Index { base, index } => { | ||
419 | let _base_ty = self.infer_expr(*base, &Expectation::none()); | ||
420 | let _index_ty = self.infer_expr(*index, &Expectation::none()); | ||
421 | // FIXME: use `std::ops::Index::Output` to figure out the real return type | ||
422 | Ty::Unknown | ||
423 | } | ||
424 | Expr::Tuple { exprs } => { | ||
425 | let mut tys = match &expected.ty { | ||
426 | ty_app!(TypeCtor::Tuple { .. }, st) => st | ||
427 | .iter() | ||
428 | .cloned() | ||
429 | .chain(repeat_with(|| self.new_type_var())) | ||
430 | .take(exprs.len()) | ||
431 | .collect::<Vec<_>>(), | ||
432 | _ => (0..exprs.len()).map(|_| self.new_type_var()).collect(), | ||
433 | }; | ||
434 | |||
435 | for (expr, ty) in exprs.iter().zip(tys.iter_mut()) { | ||
436 | self.infer_expr_coerce(*expr, &Expectation::has_type(ty.clone())); | ||
437 | } | ||
438 | |||
439 | Ty::apply(TypeCtor::Tuple { cardinality: tys.len() as u16 }, Substs(tys.into())) | ||
440 | } | ||
441 | Expr::Array(array) => { | ||
442 | let elem_ty = match &expected.ty { | ||
443 | ty_app!(TypeCtor::Array, st) | ty_app!(TypeCtor::Slice, st) => { | ||
444 | st.as_single().clone() | ||
445 | } | ||
446 | _ => self.new_type_var(), | ||
447 | }; | ||
448 | |||
449 | match array { | ||
450 | Array::ElementList(items) => { | ||
451 | for expr in items.iter() { | ||
452 | self.infer_expr_coerce(*expr, &Expectation::has_type(elem_ty.clone())); | ||
453 | } | ||
454 | } | ||
455 | Array::Repeat { initializer, repeat } => { | ||
456 | self.infer_expr_coerce( | ||
457 | *initializer, | ||
458 | &Expectation::has_type(elem_ty.clone()), | ||
459 | ); | ||
460 | self.infer_expr( | ||
461 | *repeat, | ||
462 | &Expectation::has_type(Ty::simple(TypeCtor::Int(Uncertain::Known( | ||
463 | IntTy::usize(), | ||
464 | )))), | ||
465 | ); | ||
466 | } | ||
467 | } | ||
468 | |||
469 | Ty::apply_one(TypeCtor::Array, elem_ty) | ||
470 | } | ||
471 | Expr::Literal(lit) => match lit { | ||
472 | Literal::Bool(..) => Ty::simple(TypeCtor::Bool), | ||
473 | Literal::String(..) => { | ||
474 | Ty::apply_one(TypeCtor::Ref(Mutability::Shared), Ty::simple(TypeCtor::Str)) | ||
475 | } | ||
476 | Literal::ByteString(..) => { | ||
477 | let byte_type = Ty::simple(TypeCtor::Int(Uncertain::Known(IntTy::u8()))); | ||
478 | let slice_type = Ty::apply_one(TypeCtor::Slice, byte_type); | ||
479 | Ty::apply_one(TypeCtor::Ref(Mutability::Shared), slice_type) | ||
480 | } | ||
481 | Literal::Char(..) => Ty::simple(TypeCtor::Char), | ||
482 | Literal::Int(_v, ty) => Ty::simple(TypeCtor::Int((*ty).into())), | ||
483 | Literal::Float(_v, ty) => Ty::simple(TypeCtor::Float((*ty).into())), | ||
484 | }, | ||
485 | }; | ||
486 | // use a new type variable if we got Ty::Unknown here | ||
487 | let ty = self.insert_type_vars_shallow(ty); | ||
488 | let ty = self.resolve_ty_as_possible(&mut vec![], ty); | ||
489 | self.write_expr_ty(tgt_expr, ty.clone()); | ||
490 | ty | ||
491 | } | ||
492 | |||
493 | fn infer_block( | ||
494 | &mut self, | ||
495 | statements: &[Statement], | ||
496 | tail: Option<ExprId>, | ||
497 | expected: &Expectation, | ||
498 | ) -> Ty { | ||
499 | let mut diverges = false; | ||
500 | for stmt in statements { | ||
501 | match stmt { | ||
502 | Statement::Let { pat, type_ref, initializer } => { | ||
503 | let decl_ty = | ||
504 | type_ref.as_ref().map(|tr| self.make_ty(tr)).unwrap_or(Ty::Unknown); | ||
505 | |||
506 | // Always use the declared type when specified | ||
507 | let mut ty = decl_ty.clone(); | ||
508 | |||
509 | if let Some(expr) = initializer { | ||
510 | let actual_ty = | ||
511 | self.infer_expr_coerce(*expr, &Expectation::has_type(decl_ty.clone())); | ||
512 | if decl_ty == Ty::Unknown { | ||
513 | ty = actual_ty; | ||
514 | } | ||
515 | } | ||
516 | |||
517 | let ty = self.resolve_ty_as_possible(&mut vec![], ty); | ||
518 | self.infer_pat(*pat, &ty, BindingMode::default()); | ||
519 | } | ||
520 | Statement::Expr(expr) => { | ||
521 | if let ty_app!(TypeCtor::Never) = self.infer_expr(*expr, &Expectation::none()) { | ||
522 | diverges = true; | ||
523 | } | ||
524 | } | ||
525 | } | ||
526 | } | ||
527 | |||
528 | let ty = if let Some(expr) = tail { | ||
529 | self.infer_expr_coerce(expr, expected) | ||
530 | } else { | ||
531 | self.coerce(&Ty::unit(), &expected.ty); | ||
532 | Ty::unit() | ||
533 | }; | ||
534 | if diverges { | ||
535 | Ty::simple(TypeCtor::Never) | ||
536 | } else { | ||
537 | ty | ||
538 | } | ||
539 | } | ||
540 | |||
541 | fn infer_method_call( | ||
542 | &mut self, | ||
543 | tgt_expr: ExprId, | ||
544 | receiver: ExprId, | ||
545 | args: &[ExprId], | ||
546 | method_name: &Name, | ||
547 | generic_args: Option<&GenericArgs>, | ||
548 | ) -> Ty { | ||
549 | let receiver_ty = self.infer_expr(receiver, &Expectation::none()); | ||
550 | let canonicalized_receiver = self.canonicalizer().canonicalize_ty(receiver_ty.clone()); | ||
551 | let resolved = method_resolution::lookup_method( | ||
552 | &canonicalized_receiver.value, | ||
553 | self.db, | ||
554 | method_name, | ||
555 | &self.resolver, | ||
556 | ); | ||
557 | let (derefed_receiver_ty, method_ty, def_generics) = match resolved { | ||
558 | Some((ty, func)) => { | ||
559 | let ty = canonicalized_receiver.decanonicalize_ty(ty); | ||
560 | self.write_method_resolution(tgt_expr, func); | ||
561 | (ty, self.db.value_ty(func.into()), Some(self.db.generic_params(func.into()))) | ||
562 | } | ||
563 | None => (receiver_ty, Ty::Unknown, None), | ||
564 | }; | ||
565 | let substs = self.substs_for_method_call(def_generics, generic_args, &derefed_receiver_ty); | ||
566 | let method_ty = method_ty.apply_substs(substs); | ||
567 | let method_ty = self.insert_type_vars(method_ty); | ||
568 | self.register_obligations_for_call(&method_ty); | ||
569 | let (expected_receiver_ty, param_tys, ret_ty) = match method_ty.callable_sig(self.db) { | ||
570 | Some(sig) => { | ||
571 | if !sig.params().is_empty() { | ||
572 | (sig.params()[0].clone(), sig.params()[1..].to_vec(), sig.ret().clone()) | ||
573 | } else { | ||
574 | (Ty::Unknown, Vec::new(), sig.ret().clone()) | ||
575 | } | ||
576 | } | ||
577 | None => (Ty::Unknown, Vec::new(), Ty::Unknown), | ||
578 | }; | ||
579 | // Apply autoref so the below unification works correctly | ||
580 | // FIXME: return correct autorefs from lookup_method | ||
581 | let actual_receiver_ty = match expected_receiver_ty.as_reference() { | ||
582 | Some((_, mutability)) => Ty::apply_one(TypeCtor::Ref(mutability), derefed_receiver_ty), | ||
583 | _ => derefed_receiver_ty, | ||
584 | }; | ||
585 | self.unify(&expected_receiver_ty, &actual_receiver_ty); | ||
586 | |||
587 | self.check_call_arguments(args, ¶m_tys); | ||
588 | let ret_ty = self.normalize_associated_types_in(ret_ty); | ||
589 | ret_ty | ||
590 | } | ||
591 | |||
592 | fn check_call_arguments(&mut self, args: &[ExprId], param_tys: &[Ty]) { | ||
593 | // Quoting https://github.com/rust-lang/rust/blob/6ef275e6c3cb1384ec78128eceeb4963ff788dca/src/librustc_typeck/check/mod.rs#L3325 -- | ||
594 | // We do this in a pretty awful way: first we type-check any arguments | ||
595 | // that are not closures, then we type-check the closures. This is so | ||
596 | // that we have more information about the types of arguments when we | ||
597 | // type-check the functions. This isn't really the right way to do this. | ||
598 | for &check_closures in &[false, true] { | ||
599 | let param_iter = param_tys.iter().cloned().chain(repeat(Ty::Unknown)); | ||
600 | for (&arg, param_ty) in args.iter().zip(param_iter) { | ||
601 | let is_closure = match &self.body[arg] { | ||
602 | Expr::Lambda { .. } => true, | ||
603 | _ => false, | ||
604 | }; | ||
605 | |||
606 | if is_closure != check_closures { | ||
607 | continue; | ||
608 | } | ||
609 | |||
610 | let param_ty = self.normalize_associated_types_in(param_ty); | ||
611 | self.infer_expr_coerce(arg, &Expectation::has_type(param_ty.clone())); | ||
612 | } | ||
613 | } | ||
614 | } | ||
615 | |||
616 | fn substs_for_method_call( | ||
617 | &mut self, | ||
618 | def_generics: Option<Arc<GenericParams>>, | ||
619 | generic_args: Option<&GenericArgs>, | ||
620 | receiver_ty: &Ty, | ||
621 | ) -> Substs { | ||
622 | let (parent_param_count, param_count) = | ||
623 | def_generics.as_ref().map_or((0, 0), |g| (g.count_parent_params(), g.params.len())); | ||
624 | let mut substs = Vec::with_capacity(parent_param_count + param_count); | ||
625 | // Parent arguments are unknown, except for the receiver type | ||
626 | if let Some(parent_generics) = def_generics.and_then(|p| p.parent_params.clone()) { | ||
627 | for param in &parent_generics.params { | ||
628 | if param.name == name::SELF_TYPE { | ||
629 | substs.push(receiver_ty.clone()); | ||
630 | } else { | ||
631 | substs.push(Ty::Unknown); | ||
632 | } | ||
633 | } | ||
634 | } | ||
635 | // handle provided type arguments | ||
636 | if let Some(generic_args) = generic_args { | ||
637 | // if args are provided, it should be all of them, but we can't rely on that | ||
638 | for arg in generic_args.args.iter().take(param_count) { | ||
639 | match arg { | ||
640 | GenericArg::Type(type_ref) => { | ||
641 | let ty = self.make_ty(type_ref); | ||
642 | substs.push(ty); | ||
643 | } | ||
644 | } | ||
645 | } | ||
646 | }; | ||
647 | let supplied_params = substs.len(); | ||
648 | for _ in supplied_params..parent_param_count + param_count { | ||
649 | substs.push(Ty::Unknown); | ||
650 | } | ||
651 | assert_eq!(substs.len(), parent_param_count + param_count); | ||
652 | Substs(substs.into()) | ||
653 | } | ||
654 | |||
655 | fn register_obligations_for_call(&mut self, callable_ty: &Ty) { | ||
656 | if let Ty::Apply(a_ty) = callable_ty { | ||
657 | if let TypeCtor::FnDef(def) = a_ty.ctor { | ||
658 | let generic_predicates = self.db.generic_predicates(def.into()); | ||
659 | for predicate in generic_predicates.iter() { | ||
660 | let predicate = predicate.clone().subst(&a_ty.parameters); | ||
661 | if let Some(obligation) = Obligation::from_predicate(predicate) { | ||
662 | self.obligations.push(obligation); | ||
663 | } | ||
664 | } | ||
665 | // add obligation for trait implementation, if this is a trait method | ||
666 | match def { | ||
667 | CallableDef::FunctionId(f) => { | ||
668 | if let ContainerId::TraitId(trait_) = f.lookup(self.db).container { | ||
669 | // construct a TraitDef | ||
670 | let substs = a_ty.parameters.prefix( | ||
671 | self.db | ||
672 | .generic_params(trait_.into()) | ||
673 | .count_params_including_parent(), | ||
674 | ); | ||
675 | self.obligations.push(Obligation::Trait(TraitRef { | ||
676 | trait_: trait_.into(), | ||
677 | substs, | ||
678 | })); | ||
679 | } | ||
680 | } | ||
681 | CallableDef::StructId(_) | CallableDef::EnumVariantId(_) => {} | ||
682 | } | ||
683 | } | ||
684 | } | ||
685 | } | ||
686 | } | ||
diff --git a/crates/ra_hir_ty/src/infer/pat.rs b/crates/ra_hir_ty/src/infer/pat.rs new file mode 100644 index 000000000..1ebb36239 --- /dev/null +++ b/crates/ra_hir_ty/src/infer/pat.rs | |||
@@ -0,0 +1,186 @@ | |||
1 | //! Type inference for patterns. | ||
2 | |||
3 | use std::iter::repeat; | ||
4 | use std::sync::Arc; | ||
5 | |||
6 | use hir_def::{ | ||
7 | expr::{BindingAnnotation, Pat, PatId, RecordFieldPat}, | ||
8 | path::Path, | ||
9 | type_ref::Mutability, | ||
10 | }; | ||
11 | use hir_expand::name::Name; | ||
12 | use test_utils::tested_by; | ||
13 | |||
14 | use super::{BindingMode, InferenceContext}; | ||
15 | use crate::{db::HirDatabase, utils::variant_data, Substs, Ty, TypeCtor, TypeWalk}; | ||
16 | |||
17 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | ||
18 | fn infer_tuple_struct_pat( | ||
19 | &mut self, | ||
20 | path: Option<&Path>, | ||
21 | subpats: &[PatId], | ||
22 | expected: &Ty, | ||
23 | default_bm: BindingMode, | ||
24 | ) -> Ty { | ||
25 | let (ty, def) = self.resolve_variant(path); | ||
26 | let var_data = def.map(|it| variant_data(self.db, it)); | ||
27 | self.unify(&ty, expected); | ||
28 | |||
29 | let substs = ty.substs().unwrap_or_else(Substs::empty); | ||
30 | |||
31 | let field_tys = def.map(|it| self.db.field_types(it.into())).unwrap_or_default(); | ||
32 | |||
33 | for (i, &subpat) in subpats.iter().enumerate() { | ||
34 | let expected_ty = var_data | ||
35 | .as_ref() | ||
36 | .and_then(|d| d.field(&Name::new_tuple_field(i))) | ||
37 | .map_or(Ty::Unknown, |field| field_tys[field].clone()) | ||
38 | .subst(&substs); | ||
39 | let expected_ty = self.normalize_associated_types_in(expected_ty); | ||
40 | self.infer_pat(subpat, &expected_ty, default_bm); | ||
41 | } | ||
42 | |||
43 | ty | ||
44 | } | ||
45 | |||
46 | fn infer_record_pat( | ||
47 | &mut self, | ||
48 | path: Option<&Path>, | ||
49 | subpats: &[RecordFieldPat], | ||
50 | expected: &Ty, | ||
51 | default_bm: BindingMode, | ||
52 | id: PatId, | ||
53 | ) -> Ty { | ||
54 | let (ty, def) = self.resolve_variant(path); | ||
55 | let var_data = def.map(|it| variant_data(self.db, it)); | ||
56 | if let Some(variant) = def { | ||
57 | self.write_variant_resolution(id.into(), variant); | ||
58 | } | ||
59 | |||
60 | self.unify(&ty, expected); | ||
61 | |||
62 | let substs = ty.substs().unwrap_or_else(Substs::empty); | ||
63 | |||
64 | let field_tys = def.map(|it| self.db.field_types(it.into())).unwrap_or_default(); | ||
65 | for subpat in subpats { | ||
66 | let matching_field = var_data.as_ref().and_then(|it| it.field(&subpat.name)); | ||
67 | let expected_ty = | ||
68 | matching_field.map_or(Ty::Unknown, |field| field_tys[field].clone()).subst(&substs); | ||
69 | let expected_ty = self.normalize_associated_types_in(expected_ty); | ||
70 | self.infer_pat(subpat.pat, &expected_ty, default_bm); | ||
71 | } | ||
72 | |||
73 | ty | ||
74 | } | ||
75 | |||
76 | pub(super) fn infer_pat( | ||
77 | &mut self, | ||
78 | pat: PatId, | ||
79 | mut expected: &Ty, | ||
80 | mut default_bm: BindingMode, | ||
81 | ) -> Ty { | ||
82 | let body = Arc::clone(&self.body); // avoid borrow checker problem | ||
83 | |||
84 | let is_non_ref_pat = match &body[pat] { | ||
85 | Pat::Tuple(..) | ||
86 | | Pat::TupleStruct { .. } | ||
87 | | Pat::Record { .. } | ||
88 | | Pat::Range { .. } | ||
89 | | Pat::Slice { .. } => true, | ||
90 | // FIXME: Path/Lit might actually evaluate to ref, but inference is unimplemented. | ||
91 | Pat::Path(..) | Pat::Lit(..) => true, | ||
92 | Pat::Wild | Pat::Bind { .. } | Pat::Ref { .. } | Pat::Missing => false, | ||
93 | }; | ||
94 | if is_non_ref_pat { | ||
95 | while let Some((inner, mutability)) = expected.as_reference() { | ||
96 | expected = inner; | ||
97 | default_bm = match default_bm { | ||
98 | BindingMode::Move => BindingMode::Ref(mutability), | ||
99 | BindingMode::Ref(Mutability::Shared) => BindingMode::Ref(Mutability::Shared), | ||
100 | BindingMode::Ref(Mutability::Mut) => BindingMode::Ref(mutability), | ||
101 | } | ||
102 | } | ||
103 | } else if let Pat::Ref { .. } = &body[pat] { | ||
104 | tested_by!(match_ergonomics_ref); | ||
105 | // When you encounter a `&pat` pattern, reset to Move. | ||
106 | // This is so that `w` is by value: `let (_, &w) = &(1, &2);` | ||
107 | default_bm = BindingMode::Move; | ||
108 | } | ||
109 | |||
110 | // Lose mutability. | ||
111 | let default_bm = default_bm; | ||
112 | let expected = expected; | ||
113 | |||
114 | let ty = match &body[pat] { | ||
115 | Pat::Tuple(ref args) => { | ||
116 | let expectations = match expected.as_tuple() { | ||
117 | Some(parameters) => &*parameters.0, | ||
118 | _ => &[], | ||
119 | }; | ||
120 | let expectations_iter = expectations.iter().chain(repeat(&Ty::Unknown)); | ||
121 | |||
122 | let inner_tys = args | ||
123 | .iter() | ||
124 | .zip(expectations_iter) | ||
125 | .map(|(&pat, ty)| self.infer_pat(pat, ty, default_bm)) | ||
126 | .collect(); | ||
127 | |||
128 | Ty::apply(TypeCtor::Tuple { cardinality: args.len() as u16 }, Substs(inner_tys)) | ||
129 | } | ||
130 | Pat::Ref { pat, mutability } => { | ||
131 | let expectation = match expected.as_reference() { | ||
132 | Some((inner_ty, exp_mut)) => { | ||
133 | if *mutability != exp_mut { | ||
134 | // FIXME: emit type error? | ||
135 | } | ||
136 | inner_ty | ||
137 | } | ||
138 | _ => &Ty::Unknown, | ||
139 | }; | ||
140 | let subty = self.infer_pat(*pat, expectation, default_bm); | ||
141 | Ty::apply_one(TypeCtor::Ref(*mutability), subty) | ||
142 | } | ||
143 | Pat::TupleStruct { path: p, args: subpats } => { | ||
144 | self.infer_tuple_struct_pat(p.as_ref(), subpats, expected, default_bm) | ||
145 | } | ||
146 | Pat::Record { path: p, args: fields } => { | ||
147 | self.infer_record_pat(p.as_ref(), fields, expected, default_bm, pat) | ||
148 | } | ||
149 | Pat::Path(path) => { | ||
150 | // FIXME use correct resolver for the surrounding expression | ||
151 | let resolver = self.resolver.clone(); | ||
152 | self.infer_path(&resolver, &path, pat.into()).unwrap_or(Ty::Unknown) | ||
153 | } | ||
154 | Pat::Bind { mode, name: _, subpat } => { | ||
155 | let mode = if mode == &BindingAnnotation::Unannotated { | ||
156 | default_bm | ||
157 | } else { | ||
158 | BindingMode::convert(*mode) | ||
159 | }; | ||
160 | let inner_ty = if let Some(subpat) = subpat { | ||
161 | self.infer_pat(*subpat, expected, default_bm) | ||
162 | } else { | ||
163 | expected.clone() | ||
164 | }; | ||
165 | let inner_ty = self.insert_type_vars_shallow(inner_ty); | ||
166 | |||
167 | let bound_ty = match mode { | ||
168 | BindingMode::Ref(mutability) => { | ||
169 | Ty::apply_one(TypeCtor::Ref(mutability), inner_ty.clone()) | ||
170 | } | ||
171 | BindingMode::Move => inner_ty.clone(), | ||
172 | }; | ||
173 | let bound_ty = self.resolve_ty_as_possible(&mut vec![], bound_ty); | ||
174 | self.write_pat_ty(pat, bound_ty); | ||
175 | return inner_ty; | ||
176 | } | ||
177 | _ => Ty::Unknown, | ||
178 | }; | ||
179 | // use a new type variable if we got Ty::Unknown here | ||
180 | let ty = self.insert_type_vars_shallow(ty); | ||
181 | self.unify(&ty, expected); | ||
182 | let ty = self.resolve_ty_as_possible(&mut vec![], ty); | ||
183 | self.write_pat_ty(pat, ty.clone()); | ||
184 | ty | ||
185 | } | ||
186 | } | ||
diff --git a/crates/ra_hir_ty/src/infer/path.rs b/crates/ra_hir_ty/src/infer/path.rs new file mode 100644 index 000000000..e6676e1aa --- /dev/null +++ b/crates/ra_hir_ty/src/infer/path.rs | |||
@@ -0,0 +1,270 @@ | |||
1 | //! Path expression resolution. | ||
2 | |||
3 | use hir_def::{ | ||
4 | path::{Path, PathKind, PathSegment}, | ||
5 | resolver::{HasResolver, ResolveValueResult, Resolver, TypeNs, ValueNs}, | ||
6 | AssocItemId, ContainerId, Lookup, | ||
7 | }; | ||
8 | use hir_expand::name::Name; | ||
9 | |||
10 | use crate::{db::HirDatabase, method_resolution, Substs, Ty, TypeWalk, ValueTyDefId}; | ||
11 | |||
12 | use super::{ExprOrPatId, InferenceContext, TraitRef}; | ||
13 | |||
14 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | ||
15 | pub(super) fn infer_path( | ||
16 | &mut self, | ||
17 | resolver: &Resolver, | ||
18 | path: &Path, | ||
19 | id: ExprOrPatId, | ||
20 | ) -> Option<Ty> { | ||
21 | let ty = self.resolve_value_path(resolver, path, id)?; | ||
22 | let ty = self.insert_type_vars(ty); | ||
23 | let ty = self.normalize_associated_types_in(ty); | ||
24 | Some(ty) | ||
25 | } | ||
26 | |||
27 | fn resolve_value_path( | ||
28 | &mut self, | ||
29 | resolver: &Resolver, | ||
30 | path: &Path, | ||
31 | id: ExprOrPatId, | ||
32 | ) -> Option<Ty> { | ||
33 | let (value, self_subst) = if let PathKind::Type(type_ref) = &path.kind { | ||
34 | if path.segments.is_empty() { | ||
35 | // This can't actually happen syntax-wise | ||
36 | return None; | ||
37 | } | ||
38 | let ty = self.make_ty(type_ref); | ||
39 | let remaining_segments_for_ty = &path.segments[..path.segments.len() - 1]; | ||
40 | let ty = Ty::from_type_relative_path(self.db, resolver, ty, remaining_segments_for_ty); | ||
41 | self.resolve_ty_assoc_item( | ||
42 | ty, | ||
43 | &path.segments.last().expect("path had at least one segment").name, | ||
44 | id, | ||
45 | )? | ||
46 | } else { | ||
47 | let value_or_partial = resolver.resolve_path_in_value_ns(self.db, &path)?; | ||
48 | |||
49 | match value_or_partial { | ||
50 | ResolveValueResult::ValueNs(it) => (it, None), | ||
51 | ResolveValueResult::Partial(def, remaining_index) => { | ||
52 | self.resolve_assoc_item(def, path, remaining_index, id)? | ||
53 | } | ||
54 | } | ||
55 | }; | ||
56 | |||
57 | let typable: ValueTyDefId = match value { | ||
58 | ValueNs::LocalBinding(pat) => { | ||
59 | let ty = self.result.type_of_pat.get(pat)?.clone(); | ||
60 | let ty = self.resolve_ty_as_possible(&mut vec![], ty); | ||
61 | return Some(ty); | ||
62 | } | ||
63 | ValueNs::FunctionId(it) => it.into(), | ||
64 | ValueNs::ConstId(it) => it.into(), | ||
65 | ValueNs::StaticId(it) => it.into(), | ||
66 | ValueNs::StructId(it) => it.into(), | ||
67 | ValueNs::EnumVariantId(it) => it.into(), | ||
68 | }; | ||
69 | |||
70 | let mut ty = self.db.value_ty(typable); | ||
71 | if let Some(self_subst) = self_subst { | ||
72 | ty = ty.subst(&self_subst); | ||
73 | } | ||
74 | let substs = Ty::substs_from_path(self.db, &self.resolver, path, typable); | ||
75 | let ty = ty.subst(&substs); | ||
76 | Some(ty) | ||
77 | } | ||
78 | |||
79 | fn resolve_assoc_item( | ||
80 | &mut self, | ||
81 | def: TypeNs, | ||
82 | path: &Path, | ||
83 | remaining_index: usize, | ||
84 | id: ExprOrPatId, | ||
85 | ) -> Option<(ValueNs, Option<Substs>)> { | ||
86 | assert!(remaining_index < path.segments.len()); | ||
87 | // there may be more intermediate segments between the resolved one and | ||
88 | // the end. Only the last segment needs to be resolved to a value; from | ||
89 | // the segments before that, we need to get either a type or a trait ref. | ||
90 | |||
91 | let resolved_segment = &path.segments[remaining_index - 1]; | ||
92 | let remaining_segments = &path.segments[remaining_index..]; | ||
93 | let is_before_last = remaining_segments.len() == 1; | ||
94 | |||
95 | match (def, is_before_last) { | ||
96 | (TypeNs::TraitId(trait_), true) => { | ||
97 | let segment = | ||
98 | remaining_segments.last().expect("there should be at least one segment here"); | ||
99 | let trait_ref = TraitRef::from_resolved_path( | ||
100 | self.db, | ||
101 | &self.resolver, | ||
102 | trait_.into(), | ||
103 | resolved_segment, | ||
104 | None, | ||
105 | ); | ||
106 | self.resolve_trait_assoc_item(trait_ref, segment, id) | ||
107 | } | ||
108 | (def, _) => { | ||
109 | // Either we already have a type (e.g. `Vec::new`), or we have a | ||
110 | // trait but it's not the last segment, so the next segment | ||
111 | // should resolve to an associated type of that trait (e.g. `<T | ||
112 | // as Iterator>::Item::default`) | ||
113 | let remaining_segments_for_ty = &remaining_segments[..remaining_segments.len() - 1]; | ||
114 | let ty = Ty::from_partly_resolved_hir_path( | ||
115 | self.db, | ||
116 | &self.resolver, | ||
117 | def, | ||
118 | resolved_segment, | ||
119 | remaining_segments_for_ty, | ||
120 | ); | ||
121 | if let Ty::Unknown = ty { | ||
122 | return None; | ||
123 | } | ||
124 | |||
125 | let ty = self.insert_type_vars(ty); | ||
126 | let ty = self.normalize_associated_types_in(ty); | ||
127 | |||
128 | let segment = | ||
129 | remaining_segments.last().expect("there should be at least one segment here"); | ||
130 | |||
131 | self.resolve_ty_assoc_item(ty, &segment.name, id) | ||
132 | } | ||
133 | } | ||
134 | } | ||
135 | |||
136 | fn resolve_trait_assoc_item( | ||
137 | &mut self, | ||
138 | trait_ref: TraitRef, | ||
139 | segment: &PathSegment, | ||
140 | id: ExprOrPatId, | ||
141 | ) -> Option<(ValueNs, Option<Substs>)> { | ||
142 | let trait_ = trait_ref.trait_; | ||
143 | let item = self | ||
144 | .db | ||
145 | .trait_data(trait_) | ||
146 | .items | ||
147 | .iter() | ||
148 | .map(|(_name, id)| (*id).into()) | ||
149 | .find_map(|item| match item { | ||
150 | AssocItemId::FunctionId(func) => { | ||
151 | if segment.name == self.db.function_data(func).name { | ||
152 | Some(AssocItemId::FunctionId(func)) | ||
153 | } else { | ||
154 | None | ||
155 | } | ||
156 | } | ||
157 | |||
158 | AssocItemId::ConstId(konst) => { | ||
159 | if self.db.const_data(konst).name.as_ref().map_or(false, |n| n == &segment.name) | ||
160 | { | ||
161 | Some(AssocItemId::ConstId(konst)) | ||
162 | } else { | ||
163 | None | ||
164 | } | ||
165 | } | ||
166 | AssocItemId::TypeAliasId(_) => None, | ||
167 | })?; | ||
168 | let def = match item { | ||
169 | AssocItemId::FunctionId(f) => ValueNs::FunctionId(f), | ||
170 | AssocItemId::ConstId(c) => ValueNs::ConstId(c), | ||
171 | AssocItemId::TypeAliasId(_) => unreachable!(), | ||
172 | }; | ||
173 | let substs = Substs::build_for_def(self.db, item) | ||
174 | .use_parent_substs(&trait_ref.substs) | ||
175 | .fill_with_params() | ||
176 | .build(); | ||
177 | |||
178 | self.write_assoc_resolution(id, item); | ||
179 | Some((def, Some(substs))) | ||
180 | } | ||
181 | |||
182 | fn resolve_ty_assoc_item( | ||
183 | &mut self, | ||
184 | ty: Ty, | ||
185 | name: &Name, | ||
186 | id: ExprOrPatId, | ||
187 | ) -> Option<(ValueNs, Option<Substs>)> { | ||
188 | if let Ty::Unknown = ty { | ||
189 | return None; | ||
190 | } | ||
191 | |||
192 | let canonical_ty = self.canonicalizer().canonicalize_ty(ty.clone()); | ||
193 | |||
194 | method_resolution::iterate_method_candidates( | ||
195 | &canonical_ty.value, | ||
196 | self.db, | ||
197 | &self.resolver.clone(), | ||
198 | Some(name), | ||
199 | method_resolution::LookupMode::Path, | ||
200 | move |_ty, item| { | ||
201 | let (def, container) = match item { | ||
202 | AssocItemId::FunctionId(f) => { | ||
203 | (ValueNs::FunctionId(f), f.lookup(self.db).container) | ||
204 | } | ||
205 | AssocItemId::ConstId(c) => (ValueNs::ConstId(c), c.lookup(self.db).container), | ||
206 | AssocItemId::TypeAliasId(_) => unreachable!(), | ||
207 | }; | ||
208 | let substs = match container { | ||
209 | ContainerId::ImplId(_) => self.find_self_types(&def, ty.clone()), | ||
210 | ContainerId::TraitId(trait_) => { | ||
211 | // we're picking this method | ||
212 | let trait_substs = Substs::build_for_def(self.db, trait_) | ||
213 | .push(ty.clone()) | ||
214 | .fill(std::iter::repeat_with(|| self.new_type_var())) | ||
215 | .build(); | ||
216 | let substs = Substs::build_for_def(self.db, item) | ||
217 | .use_parent_substs(&trait_substs) | ||
218 | .fill_with_params() | ||
219 | .build(); | ||
220 | self.obligations.push(super::Obligation::Trait(TraitRef { | ||
221 | trait_, | ||
222 | substs: trait_substs, | ||
223 | })); | ||
224 | Some(substs) | ||
225 | } | ||
226 | ContainerId::ModuleId(_) => None, | ||
227 | }; | ||
228 | |||
229 | self.write_assoc_resolution(id, item.into()); | ||
230 | Some((def, substs)) | ||
231 | }, | ||
232 | ) | ||
233 | } | ||
234 | |||
235 | fn find_self_types(&self, def: &ValueNs, actual_def_ty: Ty) -> Option<Substs> { | ||
236 | if let ValueNs::FunctionId(func) = *def { | ||
237 | // We only do the infer if parent has generic params | ||
238 | let gen = self.db.generic_params(func.into()); | ||
239 | if gen.count_parent_params() == 0 { | ||
240 | return None; | ||
241 | } | ||
242 | |||
243 | let impl_id = match func.lookup(self.db).container { | ||
244 | ContainerId::ImplId(it) => it, | ||
245 | _ => return None, | ||
246 | }; | ||
247 | let resolver = impl_id.resolver(self.db); | ||
248 | let impl_data = self.db.impl_data(impl_id); | ||
249 | let impl_block = Ty::from_hir(self.db, &resolver, &impl_data.target_type); | ||
250 | let impl_block_substs = impl_block.substs()?; | ||
251 | let actual_substs = actual_def_ty.substs()?; | ||
252 | |||
253 | let mut new_substs = vec![Ty::Unknown; gen.count_parent_params()]; | ||
254 | |||
255 | // The following code *link up* the function actual parma type | ||
256 | // and impl_block type param index | ||
257 | impl_block_substs.iter().zip(actual_substs.iter()).for_each(|(param, pty)| { | ||
258 | if let Ty::Param { idx, .. } = param { | ||
259 | if let Some(s) = new_substs.get_mut(*idx as usize) { | ||
260 | *s = pty.clone(); | ||
261 | } | ||
262 | } | ||
263 | }); | ||
264 | |||
265 | Some(Substs(new_substs.into())) | ||
266 | } else { | ||
267 | None | ||
268 | } | ||
269 | } | ||
270 | } | ||
diff --git a/crates/ra_hir_ty/src/infer/unify.rs b/crates/ra_hir_ty/src/infer/unify.rs new file mode 100644 index 000000000..f3a875678 --- /dev/null +++ b/crates/ra_hir_ty/src/infer/unify.rs | |||
@@ -0,0 +1,162 @@ | |||
1 | //! Unification and canonicalization logic. | ||
2 | |||
3 | use super::{InferenceContext, Obligation}; | ||
4 | use crate::{ | ||
5 | db::HirDatabase, utils::make_mut_slice, Canonical, InEnvironment, InferTy, ProjectionPredicate, | ||
6 | ProjectionTy, Substs, TraitRef, Ty, TypeWalk, | ||
7 | }; | ||
8 | |||
9 | impl<'a, D: HirDatabase> InferenceContext<'a, D> { | ||
10 | pub(super) fn canonicalizer<'b>(&'b mut self) -> Canonicalizer<'a, 'b, D> | ||
11 | where | ||
12 | 'a: 'b, | ||
13 | { | ||
14 | Canonicalizer { ctx: self, free_vars: Vec::new(), var_stack: Vec::new() } | ||
15 | } | ||
16 | } | ||
17 | |||
18 | pub(super) struct Canonicalizer<'a, 'b, D: HirDatabase> | ||
19 | where | ||
20 | 'a: 'b, | ||
21 | { | ||
22 | ctx: &'b mut InferenceContext<'a, D>, | ||
23 | free_vars: Vec<InferTy>, | ||
24 | /// A stack of type variables that is used to detect recursive types (which | ||
25 | /// are an error, but we need to protect against them to avoid stack | ||
26 | /// overflows). | ||
27 | var_stack: Vec<super::TypeVarId>, | ||
28 | } | ||
29 | |||
30 | pub(super) struct Canonicalized<T> { | ||
31 | pub value: Canonical<T>, | ||
32 | free_vars: Vec<InferTy>, | ||
33 | } | ||
34 | |||
35 | impl<'a, 'b, D: HirDatabase> Canonicalizer<'a, 'b, D> | ||
36 | where | ||
37 | 'a: 'b, | ||
38 | { | ||
39 | fn add(&mut self, free_var: InferTy) -> usize { | ||
40 | self.free_vars.iter().position(|&v| v == free_var).unwrap_or_else(|| { | ||
41 | let next_index = self.free_vars.len(); | ||
42 | self.free_vars.push(free_var); | ||
43 | next_index | ||
44 | }) | ||
45 | } | ||
46 | |||
47 | fn do_canonicalize_ty(&mut self, ty: Ty) -> Ty { | ||
48 | ty.fold(&mut |ty| match ty { | ||
49 | Ty::Infer(tv) => { | ||
50 | let inner = tv.to_inner(); | ||
51 | if self.var_stack.contains(&inner) { | ||
52 | // recursive type | ||
53 | return tv.fallback_value(); | ||
54 | } | ||
55 | if let Some(known_ty) = | ||
56 | self.ctx.var_unification_table.inlined_probe_value(inner).known() | ||
57 | { | ||
58 | self.var_stack.push(inner); | ||
59 | let result = self.do_canonicalize_ty(known_ty.clone()); | ||
60 | self.var_stack.pop(); | ||
61 | result | ||
62 | } else { | ||
63 | let root = self.ctx.var_unification_table.find(inner); | ||
64 | let free_var = match tv { | ||
65 | InferTy::TypeVar(_) => InferTy::TypeVar(root), | ||
66 | InferTy::IntVar(_) => InferTy::IntVar(root), | ||
67 | InferTy::FloatVar(_) => InferTy::FloatVar(root), | ||
68 | InferTy::MaybeNeverTypeVar(_) => InferTy::MaybeNeverTypeVar(root), | ||
69 | }; | ||
70 | let position = self.add(free_var); | ||
71 | Ty::Bound(position as u32) | ||
72 | } | ||
73 | } | ||
74 | _ => ty, | ||
75 | }) | ||
76 | } | ||
77 | |||
78 | fn do_canonicalize_trait_ref(&mut self, mut trait_ref: TraitRef) -> TraitRef { | ||
79 | for ty in make_mut_slice(&mut trait_ref.substs.0) { | ||
80 | *ty = self.do_canonicalize_ty(ty.clone()); | ||
81 | } | ||
82 | trait_ref | ||
83 | } | ||
84 | |||
85 | fn into_canonicalized<T>(self, result: T) -> Canonicalized<T> { | ||
86 | Canonicalized { | ||
87 | value: Canonical { value: result, num_vars: self.free_vars.len() }, | ||
88 | free_vars: self.free_vars, | ||
89 | } | ||
90 | } | ||
91 | |||
92 | fn do_canonicalize_projection_ty(&mut self, mut projection_ty: ProjectionTy) -> ProjectionTy { | ||
93 | for ty in make_mut_slice(&mut projection_ty.parameters.0) { | ||
94 | *ty = self.do_canonicalize_ty(ty.clone()); | ||
95 | } | ||
96 | projection_ty | ||
97 | } | ||
98 | |||
99 | fn do_canonicalize_projection_predicate( | ||
100 | &mut self, | ||
101 | projection: ProjectionPredicate, | ||
102 | ) -> ProjectionPredicate { | ||
103 | let ty = self.do_canonicalize_ty(projection.ty); | ||
104 | let projection_ty = self.do_canonicalize_projection_ty(projection.projection_ty); | ||
105 | |||
106 | ProjectionPredicate { ty, projection_ty } | ||
107 | } | ||
108 | |||
109 | // FIXME: add some point, we need to introduce a `Fold` trait that abstracts | ||
110 | // over all the things that can be canonicalized (like Chalk and rustc have) | ||
111 | |||
112 | pub(crate) fn canonicalize_ty(mut self, ty: Ty) -> Canonicalized<Ty> { | ||
113 | let result = self.do_canonicalize_ty(ty); | ||
114 | self.into_canonicalized(result) | ||
115 | } | ||
116 | |||
117 | pub(crate) fn canonicalize_obligation( | ||
118 | mut self, | ||
119 | obligation: InEnvironment<Obligation>, | ||
120 | ) -> Canonicalized<InEnvironment<Obligation>> { | ||
121 | let result = match obligation.value { | ||
122 | Obligation::Trait(tr) => Obligation::Trait(self.do_canonicalize_trait_ref(tr)), | ||
123 | Obligation::Projection(pr) => { | ||
124 | Obligation::Projection(self.do_canonicalize_projection_predicate(pr)) | ||
125 | } | ||
126 | }; | ||
127 | self.into_canonicalized(InEnvironment { | ||
128 | value: result, | ||
129 | environment: obligation.environment, | ||
130 | }) | ||
131 | } | ||
132 | } | ||
133 | |||
134 | impl<T> Canonicalized<T> { | ||
135 | pub fn decanonicalize_ty(&self, mut ty: Ty) -> Ty { | ||
136 | ty.walk_mut_binders( | ||
137 | &mut |ty, binders| match ty { | ||
138 | &mut Ty::Bound(idx) => { | ||
139 | if idx as usize >= binders && (idx as usize - binders) < self.free_vars.len() { | ||
140 | *ty = Ty::Infer(self.free_vars[idx as usize - binders]); | ||
141 | } | ||
142 | } | ||
143 | _ => {} | ||
144 | }, | ||
145 | 0, | ||
146 | ); | ||
147 | ty | ||
148 | } | ||
149 | |||
150 | pub fn apply_solution( | ||
151 | &self, | ||
152 | ctx: &mut InferenceContext<'_, impl HirDatabase>, | ||
153 | solution: Canonical<Vec<Ty>>, | ||
154 | ) { | ||
155 | // the solution may contain new variables, which we need to convert to new inference vars | ||
156 | let new_vars = Substs((0..solution.num_vars).map(|_| ctx.new_type_var()).collect()); | ||
157 | for (i, ty) in solution.value.into_iter().enumerate() { | ||
158 | let var = self.free_vars[i]; | ||
159 | ctx.unify(&Ty::Infer(var), &ty.subst_bound_vars(&new_vars)); | ||
160 | } | ||
161 | } | ||
162 | } | ||