Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/hir-ty/src/infer/expr.rs')
| -rw-r--r-- | crates/hir-ty/src/infer/expr.rs | 1238 |
1 files changed, 531 insertions, 707 deletions
diff --git a/crates/hir-ty/src/infer/expr.rs b/crates/hir-ty/src/infer/expr.rs index 7b404bb854..179eaccc65 100644 --- a/crates/hir-ty/src/infer/expr.rs +++ b/crates/hir-ty/src/infer/expr.rs @@ -2,12 +2,11 @@ use std::{iter::repeat_with, mem}; -use chalk_ir::{DebruijnIndex, Mutability, TyVariableKind, cast::Cast}; use either::Either; use hir_def::hir::ClosureKind; use hir_def::{ BlockId, FieldId, GenericDefId, GenericParamId, ItemContainerId, Lookup, TupleFieldId, TupleId, - expr_store::path::{GenericArg, GenericArgs, Path}, + expr_store::path::{GenericArg as HirGenericArg, GenericArgs as HirGenericArgs, Path}, hir::{ ArithOp, Array, AsmOperand, AsmOptions, BinaryOp, Expr, ExprId, ExprOrPatId, LabelId, Literal, Pat, PatId, Statement, UnaryOp, generics::GenericParamDataRef, @@ -17,19 +16,19 @@ use hir_def::{ }; use hir_expand::name::Name; use intern::sym; -use rustc_type_ir::inherent::{AdtDef, IntoKind, SliceLike, Ty as _}; -use stdx::always; +use rustc_ast_ir::Mutability; +use rustc_type_ir::{ + AliasTyKind, InferTy, Interner, + inherent::{AdtDef, GenericArgs as _, IntoKind, SliceLike, Ty as _}, +}; use syntax::ast::RangeOp; use tracing::debug; -use crate::autoderef::overloaded_deref_ty; -use crate::next_solver::infer::DefineOpaqueTypes; -use crate::next_solver::obligation_ctxt::ObligationCtxt; -use crate::next_solver::{DbInterner, ErrorGuaranteed}; use crate::{ - Adjust, Adjustment, AdtId, AutoBorrow, CallableDefId, CallableSig, DeclContext, DeclOrigin, - IncorrectGenericsLenKind, Interner, LifetimeElisionKind, Rawness, Scalar, Substitution, - TraitEnvironment, TraitRef, Ty, TyBuilder, TyExt, TyKind, consteval, + Adjust, Adjustment, AutoBorrow, CallableDefId, DeclContext, DeclOrigin, + IncorrectGenericsLenKind, Rawness, TraitEnvironment, + autoderef::overloaded_deref_ty, + consteval, generics::generics, infer::{ AllowTwoPhase, BreakableKind, @@ -38,18 +37,20 @@ use crate::{ pat::contains_explicit_ref_binding, }, lang_items::lang_items_for_bin_op, - lower::{ - ParamLoweringMode, lower_to_chalk_mutability, + lower_nextsolver::{ + LifetimeElisionKind, lower_mutability, path::{GenericArgsLowerer, TypeLikeConst, substs_from_args_and_bindings}, }, - mapping::{ToChalk, from_chalk}, method_resolution::{self, VisibleFromModule}, next_solver::{ - infer::traits::ObligationCause, - mapping::{ChalkToNextSolver, NextSolverToChalk}, + AliasTy, Const, DbInterner, ErrorGuaranteed, GenericArg, GenericArgs, TraitRef, Ty, TyKind, + TypeError, + infer::{ + DefineOpaqueTypes, InferOk, + traits::{Obligation, ObligationCause}, + }, + obligation_ctxt::ObligationCtxt, }, - primitive::{self, UintTy}, - static_lifetime, to_chalk_trait_id, traits::FnTrait, }; @@ -64,27 +65,30 @@ pub(crate) enum ExprIsRead { No, } -impl<'db> InferenceContext<'db> { +impl<'db> InferenceContext<'_, 'db> { pub(crate) fn infer_expr( &mut self, tgt_expr: ExprId, - expected: &Expectation, + expected: &Expectation<'db>, is_read: ExprIsRead, - ) -> Ty { + ) -> Ty<'db> { let ty = self.infer_expr_inner(tgt_expr, expected, is_read); if let Some(expected_ty) = expected.only_has_type(&mut self.table) { - let could_unify = self.unify(&ty, &expected_ty); + let could_unify = self.unify(ty, expected_ty); if !could_unify { - self.result.type_mismatches.insert( - tgt_expr.into(), - TypeMismatch { expected: expected_ty, actual: ty.clone() }, - ); + self.result + .type_mismatches + .insert(tgt_expr.into(), TypeMismatch { expected: expected_ty, actual: ty }); } } ty } - pub(crate) fn infer_expr_no_expect(&mut self, tgt_expr: ExprId, is_read: ExprIsRead) -> Ty { + pub(crate) fn infer_expr_no_expect( + &mut self, + tgt_expr: ExprId, + is_read: ExprIsRead, + ) -> Ty<'db> { self.infer_expr_inner(tgt_expr, &Expectation::None, is_read) } @@ -93,9 +97,9 @@ impl<'db> InferenceContext<'db> { pub(super) fn infer_expr_coerce( &mut self, expr: ExprId, - expected: &Expectation, + expected: &Expectation<'db>, is_read: ExprIsRead, - ) -> Ty { + ) -> Ty<'db> { let ty = self.infer_expr_inner(expr, expected, is_read); if let Some(target) = expected.only_has_type(&mut self.table) { let coerce_never = if self.expr_guaranteed_to_constitute_read_for_never(expr, is_read) { @@ -103,19 +107,12 @@ impl<'db> InferenceContext<'db> { } else { CoerceNever::No }; - match self.coerce( - expr.into(), - ty.to_nextsolver(self.table.interner), - target.to_nextsolver(self.table.interner), - AllowTwoPhase::No, - coerce_never, - ) { - Ok(res) => res.to_chalk(self.table.interner), + match self.coerce(expr.into(), ty, target, AllowTwoPhase::No, coerce_never) { + Ok(res) => res, Err(_) => { - self.result.type_mismatches.insert( - expr.into(), - TypeMismatch { expected: target.clone(), actual: ty.clone() }, - ); + self.result + .type_mismatches + .insert(expr.into(), TypeMismatch { expected: target, actual: ty }); target } } @@ -255,42 +252,34 @@ impl<'db> InferenceContext<'db> { fn infer_expr_coerce_never( &mut self, expr: ExprId, - expected: &Expectation, + expected: &Expectation<'db>, is_read: ExprIsRead, - ) -> Ty { + ) -> Ty<'db> { let ty = self.infer_expr_inner(expr, expected, is_read); // While we don't allow *arbitrary* coercions here, we *do* allow // coercions from `!` to `expected`. if ty.is_never() { if let Some(adjustments) = self.result.expr_adjustments.get(&expr) { return if let [Adjustment { kind: Adjust::NeverToAny, target }] = &**adjustments { - target.clone() + *target } else { self.err_ty() }; } if let Some(target) = expected.only_has_type(&mut self.table) { - self.coerce( - expr.into(), - ty.to_nextsolver(self.table.interner), - target.to_nextsolver(self.table.interner), - AllowTwoPhase::No, - CoerceNever::Yes, - ) - .expect("never-to-any coercion should always succeed") - .to_chalk(self.table.interner) + self.coerce(expr.into(), ty, target, AllowTwoPhase::No, CoerceNever::Yes) + .expect("never-to-any coercion should always succeed") } else { ty } } else { if let Some(expected_ty) = expected.only_has_type(&mut self.table) { - let could_unify = self.unify(&ty, &expected_ty); + let could_unify = self.unify(ty, expected_ty); if !could_unify { - self.result.type_mismatches.insert( - expr.into(), - TypeMismatch { expected: expected_ty, actual: ty.clone() }, - ); + self.result + .type_mismatches + .insert(expr.into(), TypeMismatch { expected: expected_ty, actual: ty }); } } ty @@ -301,9 +290,9 @@ impl<'db> InferenceContext<'db> { fn infer_expr_inner( &mut self, tgt_expr: ExprId, - expected: &Expectation, + expected: &Expectation<'db>, is_read: ExprIsRead, - ) -> Ty { + ) -> Ty<'db> { self.db.unwind_if_revision_cancelled(); let expr = &self.body[tgt_expr]; @@ -314,7 +303,7 @@ impl<'db> InferenceContext<'db> { let expected = &expected.adjust_for_branches(&mut self.table); self.infer_expr_coerce_never( condition, - &Expectation::HasType(self.result.standard_types.bool_.clone()), + &Expectation::HasType(self.types.bool), ExprIsRead::Yes, ); @@ -327,27 +316,15 @@ impl<'db> InferenceContext<'db> { coercion_sites[1] = else_branch; } let mut coerce = CoerceMany::with_coercion_sites( - expected - .coercion_target_type(&mut self.table) - .to_nextsolver(self.table.interner), + expected.coercion_target_type(&mut self.table), &coercion_sites, ); - coerce.coerce( - self, - &ObligationCause::new(), - then_branch, - then_ty.to_nextsolver(self.table.interner), - ); + coerce.coerce(self, &ObligationCause::new(), then_branch, then_ty); match else_branch { Some(else_branch) => { let else_ty = self.infer_expr_inner(else_branch, expected, ExprIsRead::Yes); let else_diverges = mem::replace(&mut self.diverges, Diverges::Maybe); - coerce.coerce( - self, - &ObligationCause::new(), - else_branch, - else_ty.to_nextsolver(self.table.interner), - ); + coerce.coerce(self, &ObligationCause::new(), else_branch, else_ty); self.diverges = condition_diverges | then_diverges & else_diverges; } None => { @@ -356,7 +333,7 @@ impl<'db> InferenceContext<'db> { } } - coerce.complete(self).to_chalk(self.table.interner) + coerce.complete(self) } &Expr::Let { pat, expr } => { let child_is_read = if self.pat_guaranteed_to_constitute_read_for_never(pat) { @@ -367,10 +344,10 @@ impl<'db> InferenceContext<'db> { let input_ty = self.infer_expr(expr, &Expectation::none(), child_is_read); self.infer_top_pat( pat, - &input_ty, + input_ty, Some(DeclContext { origin: DeclOrigin::LetExpr }), ); - self.result.standard_types.bool_.clone() + self.types.bool } Expr::Block { statements, tail, label, id } => { self.infer_block(tgt_expr, *id, statements, *tail, *label, expected) @@ -390,12 +367,12 @@ impl<'db> InferenceContext<'db> { &Expr::Loop { body, label } => { // FIXME: should be: // let ty = expected.coercion_target_type(&mut self.table); - let ty = self.table.new_type_var(); + let ty = self.table.next_ty_var(); let (breaks, ()) = self.with_breakable_ctx(BreakableKind::Loop, Some(ty), label, |this| { this.infer_expr( body, - &Expectation::HasType(TyBuilder::unit()), + &Expectation::HasType(this.types.unit), ExprIsRead::Yes, ); }); @@ -405,7 +382,7 @@ impl<'db> InferenceContext<'db> { self.diverges = Diverges::Maybe; breaks } - None => self.result.standard_types.never.clone(), + None => self.types.never, } } Expr::Closure { body, args, ret_type, arg_types, closure_kind, capture_by: _ } => self @@ -438,32 +415,30 @@ impl<'db> InferenceContext<'db> { if arms.is_empty() { self.diverges = Diverges::Always; - self.result.standard_types.never.clone() + self.types.never } else { let matchee_diverges = mem::replace(&mut self.diverges, Diverges::Maybe); let mut all_arms_diverge = Diverges::Always; for arm in arms.iter() { - let input_ty = self.table.structurally_resolve_type(&input_ty); - self.infer_top_pat(arm.pat, &input_ty, None); + let input_ty = self.table.structurally_resolve_type(input_ty); + self.infer_top_pat(arm.pat, input_ty, None); } let expected = expected.adjust_for_branches(&mut self.table); let result_ty = match &expected { // We don't coerce to `()` so that if the match expression is a // statement it's branches can have any consistent type. - Expectation::HasType(ty) if *ty != self.result.standard_types.unit => { - ty.clone() - } - _ => self.table.new_type_var(), + Expectation::HasType(ty) if *ty != self.types.unit => *ty, + _ => self.table.next_ty_var(), }; - let mut coerce = CoerceMany::new(result_ty.to_nextsolver(self.table.interner)); + let mut coerce = CoerceMany::new(result_ty); for arm in arms.iter() { if let Some(guard_expr) = arm.guard { self.diverges = Diverges::Maybe; self.infer_expr_coerce_never( guard_expr, - &Expectation::HasType(self.result.standard_types.bool_.clone()), + &Expectation::HasType(self.types.bool), ExprIsRead::Yes, ); } @@ -471,17 +446,12 @@ impl<'db> InferenceContext<'db> { let arm_ty = self.infer_expr_inner(arm.expr, &expected, ExprIsRead::Yes); all_arms_diverge &= self.diverges; - coerce.coerce( - self, - &ObligationCause::new(), - arm.expr, - arm_ty.to_nextsolver(self.table.interner), - ); + coerce.coerce(self, &ObligationCause::new(), arm.expr, arm_ty); } self.diverges = matchee_diverges | all_arms_diverge; - coerce.complete(self).to_chalk(self.table.interner) + coerce.complete(self) } } Expr::Path(p) => self.infer_expr_path(p, tgt_expr.into(), tgt_expr), @@ -493,13 +463,13 @@ impl<'db> InferenceContext<'db> { bad_value_break: false, }); }; - self.result.standard_types.never.clone() + self.types.never } &Expr::Break { expr, label } => { let val_ty = if let Some(expr) = expr { let opt_coerce_to = match find_breakable(&mut self.breakables, label) { Some(ctxt) => match &ctxt.coerce { - Some(coerce) => coerce.expected_ty().to_chalk(self.table.interner), + Some(coerce) => coerce.expected_ty(), None => { self.push_diagnostic(InferenceDiagnostic::BreakOutsideOfLoop { expr: tgt_expr, @@ -517,7 +487,7 @@ impl<'db> InferenceContext<'db> { ExprIsRead::Yes, ) } else { - TyBuilder::unit() + self.types.unit }; match find_breakable(&mut self.breakables, label) { @@ -527,7 +497,7 @@ impl<'db> InferenceContext<'db> { self, &ObligationCause::new(), expr.unwrap_or(tgt_expr), - val_ty.to_nextsolver(self.table.interner), + val_ty, ); // Avoiding borrowck @@ -546,12 +516,12 @@ impl<'db> InferenceContext<'db> { }); } } - self.result.standard_types.never.clone() + self.types.never } &Expr::Return { expr } => self.infer_expr_return(tgt_expr, expr), &Expr::Become { expr } => self.infer_expr_become(expr), Expr::Yield { expr } => { - if let Some((resume_ty, yield_ty)) = self.resume_yield_tys.clone() { + if let Some((resume_ty, yield_ty)) = self.resume_yield_tys { if let Some(expr) = expr { self.infer_expr_coerce( *expr, @@ -559,11 +529,11 @@ impl<'db> InferenceContext<'db> { ExprIsRead::Yes, ); } else { - let unit = self.result.standard_types.unit.clone(); + let unit = self.types.unit; let _ = self.coerce( tgt_expr.into(), - unit.to_nextsolver(self.table.interner), - yield_ty.to_nextsolver(self.table.interner), + unit, + yield_ty, AllowTwoPhase::No, CoerceNever::Yes, ); @@ -571,33 +541,30 @@ impl<'db> InferenceContext<'db> { resume_ty } else { // FIXME: report error (yield expr in non-coroutine) - self.result.standard_types.unknown.clone() + self.types.error } } Expr::Yeet { expr } => { if let &Some(expr) = expr { self.infer_expr_no_expect(expr, ExprIsRead::Yes); } - self.result.standard_types.never.clone() + self.types.never } Expr::RecordLit { path, fields, spread, .. } => { let (ty, def_id) = self.resolve_variant(tgt_expr.into(), path.as_deref(), false); if let Some(t) = expected.only_has_type(&mut self.table) { - self.unify(&ty, &t); + self.unify(ty, t); } - let substs = ty - .as_adt() - .map(|(_, s)| s.clone()) - .unwrap_or_else(|| Substitution::empty(Interner)); + let substs = ty.as_adt().map(|(_, s)| s).unwrap_or(self.types.empty_args); if let Some(variant) = def_id { self.write_variant_resolution(tgt_expr.into(), variant); } match def_id { _ if fields.is_empty() => {} Some(def) => { - let field_types = self.db.field_types(def); + let field_types = self.db.field_types_ns(def); let variant_data = def.fields(self.db); let visibilities = self.db.field_visibilities(def); for field in fields.iter() { @@ -628,7 +595,7 @@ impl<'db> InferenceContext<'db> { } }; let field_ty = field_def.map_or(self.err_ty(), |it| { - field_types[it].clone().substitute(Interner, &substs) + field_types[it].instantiate(self.interner(), &substs) }); // Field type might have some unknown types @@ -649,7 +616,7 @@ impl<'db> InferenceContext<'db> { } } if let Some(expr) = spread { - self.infer_expr(*expr, &Expectation::has_type(ty.clone()), ExprIsRead::Yes); + self.infer_expr(*expr, &Expectation::has_type(ty), ExprIsRead::Yes); } ty } @@ -660,21 +627,13 @@ impl<'db> InferenceContext<'db> { } Expr::Cast { expr, type_ref } => { let cast_ty = self.make_body_ty(*type_ref); - let expr_ty = self.infer_expr( - *expr, - &Expectation::Castable(cast_ty.clone()), - ExprIsRead::Yes, - ); - self.deferred_cast_checks.push(CastCheck::new( - tgt_expr, - *expr, - expr_ty, - cast_ty.clone(), - )); + let expr_ty = + self.infer_expr(*expr, &Expectation::Castable(cast_ty), ExprIsRead::Yes); + self.deferred_cast_checks.push(CastCheck::new(tgt_expr, *expr, expr_ty, cast_ty)); cast_ty } Expr::Ref { expr, rawness, mutability } => { - let mutability = lower_to_chalk_mutability(*mutability); + let mutability = lower_mutability(*mutability); let expectation = if let Some((exp_inner, exp_rawness, exp_mutability)) = expected .only_has_type(&mut self.table) .as_ref() @@ -688,24 +647,23 @@ impl<'db> InferenceContext<'db> { // FIXME: record type error - expected reference but found ptr, // which cannot be coerced } - Expectation::rvalue_hint(self, Ty::clone(exp_inner)) + Expectation::rvalue_hint(self, exp_inner) } else { Expectation::none() }; let inner_ty = self.infer_expr_inner(*expr, &expectation, ExprIsRead::Yes); match rawness { - Rawness::RawPtr => TyKind::Raw(mutability, inner_ty), + Rawness::RawPtr => Ty::new_ptr(self.interner(), inner_ty, mutability), Rawness::Ref => { - let lt = self.table.new_lifetime_var(); - TyKind::Ref(mutability, lt, inner_ty) + let lt = self.table.next_region_var(); + Ty::new_ref(self.interner(), lt, inner_ty, mutability) } } - .intern(Interner) } &Expr::Box { expr } => self.infer_expr_box(expr, expected), Expr::UnaryOp { expr, op } => { let inner_ty = self.infer_expr_inner(*expr, &Expectation::none(), ExprIsRead::Yes); - let inner_ty = self.table.structurally_resolve_type(&inner_ty); + let inner_ty = self.table.try_structurally_resolve_type(inner_ty); // FIXME: Note down method resolution her match op { UnaryOp::Deref => { @@ -716,49 +674,38 @@ impl<'db> InferenceContext<'db> { { // FIXME: this is wrong in multiple ways, subst is empty, and we emit it even for builtin deref (note that // the mutability is not wrong, and will be fixed in `self.infer_mut`). - self.write_method_resolution( - tgt_expr, - deref_fn, - Substitution::empty(Interner), - ); + self.write_method_resolution(tgt_expr, deref_fn, self.types.empty_args); } - if let Some(derefed) = - inner_ty.to_nextsolver(self.table.interner).builtin_deref(self.db, true) - { - self.table - .structurally_resolve_type(&derefed.to_chalk(self.table.interner)) + if let Some(derefed) = inner_ty.builtin_deref(self.db, true) { + self.table.try_structurally_resolve_type(derefed) } else { - let infer_ok = overloaded_deref_ty( - &self.table, - inner_ty.to_nextsolver(self.table.interner), - ); + let infer_ok = overloaded_deref_ty(&self.table, inner_ty); match infer_ok { - Some(infer_ok) => self - .table - .register_infer_ok(infer_ok) - .to_chalk(self.table.interner), + Some(infer_ok) => self.table.register_infer_ok(infer_ok), None => self.err_ty(), } } } UnaryOp::Neg => { - match inner_ty.kind(Interner) { + match inner_ty.kind() { // Fast path for builtins - TyKind::Scalar(Scalar::Int(_) | Scalar::Uint(_) | Scalar::Float(_)) - | TyKind::InferenceVar( - _, - TyVariableKind::Integer | TyVariableKind::Float, - ) => inner_ty, + TyKind::Int(_) + | TyKind::Uint(_) + | TyKind::Float(_) + | TyKind::Infer(InferTy::IntVar(_) | InferTy::FloatVar(_)) => inner_ty, // Otherwise we resolve via the std::ops::Neg trait _ => self .resolve_associated_type(inner_ty, self.resolve_ops_neg_output()), } } UnaryOp::Not => { - match inner_ty.kind(Interner) { + match inner_ty.kind() { // Fast path for builtins - TyKind::Scalar(Scalar::Bool | Scalar::Int(_) | Scalar::Uint(_)) - | TyKind::InferenceVar(_, TyVariableKind::Integer) => inner_ty, + TyKind::Bool + | TyKind::Int(_) + | TyKind::Uint(_) + | TyKind::Float(_) + | TyKind::Infer(InferTy::IntVar(_) | InferTy::FloatVar(_)) => inner_ty, // Otherwise we resolve via the std::ops::Not trait _ => self .resolve_associated_type(inner_ty, self.resolve_ops_not_output()), @@ -768,18 +715,10 @@ impl<'db> InferenceContext<'db> { } Expr::BinaryOp { lhs, rhs, op } => match op { Some(BinaryOp::LogicOp(_)) => { - let bool_ty = self.result.standard_types.bool_.clone(); - self.infer_expr_coerce( - *lhs, - &Expectation::HasType(bool_ty.clone()), - ExprIsRead::Yes, - ); + let bool_ty = self.types.bool; + self.infer_expr_coerce(*lhs, &Expectation::HasType(bool_ty), ExprIsRead::Yes); let lhs_diverges = self.diverges; - self.infer_expr_coerce( - *rhs, - &Expectation::HasType(bool_ty.clone()), - ExprIsRead::Yes, - ); + self.infer_expr_coerce(*rhs, &Expectation::HasType(bool_ty), ExprIsRead::Yes); // Depending on the LHS' value, the RHS can never execute. self.diverges = lhs_diverges; bool_ty @@ -826,14 +765,14 @@ impl<'db> InferenceContext<'db> { let is_destructuring_assignment = lhs_ty.is_none(); if let Some(lhs_ty) = lhs_ty { - self.write_pat_ty(target, lhs_ty.clone()); + self.write_pat_ty(target, lhs_ty); self.infer_expr_coerce(value, &Expectation::has_type(lhs_ty), ExprIsRead::No); } else { let rhs_ty = self.infer_expr(value, &Expectation::none(), ExprIsRead::Yes); let resolver_guard = self.resolver.update_to_inner_scope(self.db, self.owner, tgt_expr); self.inside_assignment = true; - self.infer_top_pat(target, &rhs_ty, None); + self.infer_top_pat(target, rhs_ty, None); self.inside_assignment = false; self.resolver.reset_to_guard(resolver_guard); } @@ -844,43 +783,49 @@ impl<'db> InferenceContext<'db> { // assignments into blocks. self.table.new_maybe_never_var() } else { - self.result.standard_types.unit.clone() + self.types.unit } } Expr::Range { lhs, rhs, range_type } => { let lhs_ty = lhs.map(|e| self.infer_expr_inner(e, &Expectation::none(), ExprIsRead::Yes)); - let rhs_expect = lhs_ty - .as_ref() - .map_or_else(Expectation::none, |ty| Expectation::has_type(ty.clone())); + let rhs_expect = + lhs_ty.as_ref().map_or_else(Expectation::none, |ty| Expectation::has_type(*ty)); let rhs_ty = rhs.map(|e| self.infer_expr(e, &rhs_expect, ExprIsRead::Yes)); + let single_arg_adt = |adt, ty: Ty<'db>| { + Ty::new_adt( + self.interner(), + adt, + GenericArgs::new_from_iter(self.interner(), [ty.into()]), + ) + }; match (range_type, lhs_ty, rhs_ty) { (RangeOp::Exclusive, None, None) => match self.resolve_range_full() { - Some(adt) => TyBuilder::adt(self.db, adt).build(), + Some(adt) => Ty::new_adt(self.interner(), adt, self.types.empty_args), None => self.err_ty(), }, (RangeOp::Exclusive, None, Some(ty)) => match self.resolve_range_to() { - Some(adt) => TyBuilder::adt(self.db, adt).push(ty).build(), + Some(adt) => single_arg_adt(adt, ty), None => self.err_ty(), }, (RangeOp::Inclusive, None, Some(ty)) => { match self.resolve_range_to_inclusive() { - Some(adt) => TyBuilder::adt(self.db, adt).push(ty).build(), + Some(adt) => single_arg_adt(adt, ty), None => self.err_ty(), } } (RangeOp::Exclusive, Some(_), Some(ty)) => match self.resolve_range() { - Some(adt) => TyBuilder::adt(self.db, adt).push(ty).build(), + Some(adt) => single_arg_adt(adt, ty), None => self.err_ty(), }, (RangeOp::Inclusive, Some(_), Some(ty)) => { match self.resolve_range_inclusive() { - Some(adt) => TyBuilder::adt(self.db, adt).push(ty).build(), + Some(adt) => single_arg_adt(adt, ty), None => self.err_ty(), } } (RangeOp::Exclusive, Some(ty), None) => match self.resolve_range_from() { - Some(adt) => TyBuilder::adt(self.db, adt).push(ty).build(), + Some(adt) => single_arg_adt(adt, ty), None => self.err_ty(), }, (RangeOp::Inclusive, _, None) => self.err_ty(), @@ -891,8 +836,7 @@ impl<'db> InferenceContext<'db> { let index_ty = self.infer_expr(*index, &Expectation::none(), ExprIsRead::Yes); if let Some(index_trait) = self.resolve_lang_trait(LangItem::Index) { - let canonicalized = - self.canonicalize(base_ty.clone().to_nextsolver(self.table.interner)); + let canonicalized = self.canonicalize(base_ty); let receiver_adjustments = method_resolution::resolve_indexing_op( &mut self.table, canonicalized, @@ -905,66 +849,52 @@ impl<'db> InferenceContext<'db> { // mutability will be fixed up in `InferenceContext::infer_mut`; adj.push(Adjustment::borrow( + self.interner(), Mutability::Not, - self_ty.clone(), - self.table.new_lifetime_var(), + self_ty, + self.table.next_region_var(), )); self.write_expr_adj(*base, adj.into_boxed_slice()); if let Some(func) = index_trait .trait_items(self.db) .method_by_name(&Name::new_symbol_root(sym::index)) { - let subst = TyBuilder::subst_for_def(self.db, index_trait, None); - if subst.remaining() != 2 { - return self.err_ty(); - } - let subst = subst.push(self_ty.clone()).push(index_ty.clone()).build(); + let subst = GenericArgs::new_from_iter( + self.interner(), + [self_ty.into(), index_ty.into()], + ); self.write_method_resolution(tgt_expr, func, subst); } let assoc = self.resolve_ops_index_output(); - self.resolve_associated_type_with_params( - self_ty, - assoc, - &[index_ty.cast(Interner)], - ) + self.resolve_associated_type_with_params(self_ty, assoc, &[index_ty.into()]) } else { self.err_ty() } } Expr::Tuple { exprs, .. } => { - let mut tys = match expected - .only_has_type(&mut self.table) - .as_ref() - .map(|t| t.kind(Interner)) - { - Some(TyKind::Tuple(_, substs)) => substs - .iter(Interner) - .map(|a| a.assert_ty_ref(Interner).clone()) - .chain(repeat_with(|| self.table.new_type_var())) - .take(exprs.len()) - .collect::<Vec<_>>(), - _ => (0..exprs.len()).map(|_| self.table.new_type_var()).collect(), - }; + let mut tys = + match expected.only_has_type(&mut self.table).as_ref().map(|t| t.kind()) { + Some(TyKind::Tuple(substs)) => substs + .iter() + .chain(repeat_with(|| self.table.next_ty_var())) + .take(exprs.len()) + .collect::<Vec<_>>(), + _ => (0..exprs.len()).map(|_| self.table.next_ty_var()).collect(), + }; for (expr, ty) in exprs.iter().zip(tys.iter_mut()) { - *ty = self.infer_expr_coerce( - *expr, - &Expectation::has_type(ty.clone()), - ExprIsRead::Yes, - ); + *ty = + self.infer_expr_coerce(*expr, &Expectation::has_type(*ty), ExprIsRead::Yes); } - TyKind::Tuple(tys.len(), Substitution::from_iter(Interner, tys)).intern(Interner) + Ty::new_tup(self.interner(), &tys) } Expr::Array(array) => self.infer_expr_array(array, expected), Expr::Literal(lit) => match lit { - Literal::Bool(..) => self.result.standard_types.bool_.clone(), - Literal::String(..) => { - TyKind::Ref(Mutability::Not, static_lifetime(), TyKind::Str.intern(Interner)) - .intern(Interner) - } + Literal::Bool(..) => self.types.bool, + Literal::String(..) => self.types.static_str_ref, Literal::ByteString(bs) => { - let byte_type = TyKind::Scalar(Scalar::Uint(UintTy::U8)).intern(Interner); + let byte_type = self.types.u8; let len = consteval::usize_const( self.db, @@ -972,75 +902,80 @@ impl<'db> InferenceContext<'db> { self.resolver.krate(), ); - let array_type = TyKind::Array(byte_type, len).intern(Interner); - TyKind::Ref(Mutability::Not, static_lifetime(), array_type).intern(Interner) + let array_type = Ty::new_array_with_const_len(self.interner(), byte_type, len); + Ty::new_ref(self.interner(), self.types.re_static, array_type, Mutability::Not) } - Literal::CString(..) => TyKind::Ref( - Mutability::Not, - static_lifetime(), + Literal::CString(..) => Ty::new_ref( + self.interner(), + self.types.re_static, self.resolve_lang_item(LangItem::CStr) .and_then(LangItemTarget::as_struct) .map_or_else( || self.err_ty(), |strukt| { - TyKind::Adt(AdtId(strukt.into()), Substitution::empty(Interner)) - .intern(Interner) + Ty::new_adt(self.interner(), strukt.into(), self.types.empty_args) }, ), - ) - .intern(Interner), - Literal::Char(..) => TyKind::Scalar(Scalar::Char).intern(Interner), + Mutability::Not, + ), + Literal::Char(..) => self.types.char, Literal::Int(_v, ty) => match ty { - Some(int_ty) => { - TyKind::Scalar(Scalar::Int(primitive::int_ty_from_builtin(*int_ty))) - .intern(Interner) - } + Some(int_ty) => match int_ty { + hir_def::builtin_type::BuiltinInt::Isize => self.types.isize, + hir_def::builtin_type::BuiltinInt::I8 => self.types.i8, + hir_def::builtin_type::BuiltinInt::I16 => self.types.i16, + hir_def::builtin_type::BuiltinInt::I32 => self.types.i32, + hir_def::builtin_type::BuiltinInt::I64 => self.types.i64, + hir_def::builtin_type::BuiltinInt::I128 => self.types.i128, + }, None => { let expected_ty = expected.to_option(&mut self.table); tracing::debug!(?expected_ty); - let opt_ty = match expected_ty.as_ref().map(|it| it.kind(Interner)) { - Some(TyKind::Scalar(Scalar::Int(_) | Scalar::Uint(_))) => expected_ty, - Some(TyKind::Scalar(Scalar::Char)) => { - Some(TyKind::Scalar(Scalar::Uint(UintTy::U8)).intern(Interner)) - } - Some(TyKind::Raw(..) | TyKind::FnDef(..) | TyKind::Function(..)) => { - Some(TyKind::Scalar(Scalar::Uint(UintTy::Usize)).intern(Interner)) + let opt_ty = match expected_ty.as_ref().map(|it| it.kind()) { + Some(TyKind::Int(_) | TyKind::Uint(_)) => expected_ty, + Some(TyKind::Char) => Some(self.types.u8), + Some(TyKind::RawPtr(..) | TyKind::FnDef(..) | TyKind::FnPtr(..)) => { + Some(self.types.usize) } _ => None, }; - opt_ty.unwrap_or_else(|| self.table.new_integer_var()) + opt_ty.unwrap_or_else(|| self.table.next_int_var()) } }, Literal::Uint(_v, ty) => match ty { - Some(int_ty) => { - TyKind::Scalar(Scalar::Uint(primitive::uint_ty_from_builtin(*int_ty))) - .intern(Interner) - } + Some(int_ty) => match int_ty { + hir_def::builtin_type::BuiltinUint::Usize => self.types.usize, + hir_def::builtin_type::BuiltinUint::U8 => self.types.u8, + hir_def::builtin_type::BuiltinUint::U16 => self.types.u16, + hir_def::builtin_type::BuiltinUint::U32 => self.types.u32, + hir_def::builtin_type::BuiltinUint::U64 => self.types.u64, + hir_def::builtin_type::BuiltinUint::U128 => self.types.u128, + }, None => { let expected_ty = expected.to_option(&mut self.table); - let opt_ty = match expected_ty.as_ref().map(|it| it.kind(Interner)) { - Some(TyKind::Scalar(Scalar::Int(_) | Scalar::Uint(_))) => expected_ty, - Some(TyKind::Scalar(Scalar::Char)) => { - Some(TyKind::Scalar(Scalar::Uint(UintTy::U8)).intern(Interner)) - } - Some(TyKind::Raw(..) | TyKind::FnDef(..) | TyKind::Function(..)) => { - Some(TyKind::Scalar(Scalar::Uint(UintTy::Usize)).intern(Interner)) + let opt_ty = match expected_ty.as_ref().map(|it| it.kind()) { + Some(TyKind::Int(_) | TyKind::Uint(_)) => expected_ty, + Some(TyKind::Char) => Some(self.types.u8), + Some(TyKind::RawPtr(..) | TyKind::FnDef(..) | TyKind::FnPtr(..)) => { + Some(self.types.usize) } _ => None, }; - opt_ty.unwrap_or_else(|| self.table.new_integer_var()) + opt_ty.unwrap_or_else(|| self.table.next_int_var()) } }, Literal::Float(_v, ty) => match ty { - Some(float_ty) => { - TyKind::Scalar(Scalar::Float(primitive::float_ty_from_builtin(*float_ty))) - .intern(Interner) - } + Some(float_ty) => match float_ty { + hir_def::builtin_type::BuiltinFloat::F16 => self.types.f16, + hir_def::builtin_type::BuiltinFloat::F32 => self.types.f32, + hir_def::builtin_type::BuiltinFloat::F64 => self.types.f64, + hir_def::builtin_type::BuiltinFloat::F128 => self.types.f128, + }, None => { - let opt_ty = expected.to_option(&mut self.table).filter(|ty| { - matches!(ty.kind(Interner), TyKind::Scalar(Scalar::Float(_))) - }); - opt_ty.unwrap_or_else(|| self.table.new_float_var()) + let opt_ty = expected + .to_option(&mut self.table) + .filter(|ty| matches!(ty.kind(), TyKind::Float(_))); + opt_ty.unwrap_or_else(|| self.table.next_float_var()) } }, }, @@ -1048,13 +983,10 @@ impl<'db> InferenceContext<'db> { // Underscore expression is an error, we render a specialized diagnostic // to let the user know what type is expected though. let expected = expected.to_option(&mut self.table).unwrap_or_else(|| self.err_ty()); - self.push_diagnostic(InferenceDiagnostic::TypedHole { - expr: tgt_expr, - expected: expected.clone(), - }); + self.push_diagnostic(InferenceDiagnostic::TypedHole { expr: tgt_expr, expected }); expected } - Expr::OffsetOf(_) => TyKind::Scalar(Scalar::Uint(UintTy::Usize)).intern(Interner), + Expr::OffsetOf(_) => self.types.usize, Expr::InlineAsm(asm) => { let check_expr_asm_operand = |this: &mut Self, expr, is_input: bool| { let ty = this.infer_expr_no_expect(expr, ExprIsRead::Yes); @@ -1067,27 +999,29 @@ impl<'db> InferenceContext<'db> { // allows them to be inferred based on how they are used later in the // function. if is_input { - let ty = this.table.structurally_resolve_type(&ty); - match ty.kind(Interner) { + let ty = this.table.structurally_resolve_type(ty); + match ty.kind() { TyKind::FnDef(def, parameters) => { - let fnptr_ty = TyKind::Function( - CallableSig::from_def(this.db, *def, parameters).to_fn_ptr(), - ) - .intern(Interner); + let fnptr_ty = Ty::new_fn_ptr( + this.interner(), + this.interner() + .fn_sig(def) + .instantiate(this.interner(), parameters), + ); _ = this.coerce( expr.into(), - ty.to_nextsolver(this.table.interner), - fnptr_ty.to_nextsolver(this.table.interner), + ty, + fnptr_ty, AllowTwoPhase::No, CoerceNever::Yes, ); } - TyKind::Ref(mutbl, _, base_ty) => { - let ptr_ty = TyKind::Raw(*mutbl, base_ty.clone()).intern(Interner); + TyKind::Ref(_, base_ty, mutbl) => { + let ptr_ty = Ty::new_ptr(this.interner(), base_ty, mutbl); _ = this.coerce( expr.into(), - ty.to_nextsolver(this.table.interner), - ptr_ty.to_nextsolver(this.table.interner), + ty, + ptr_ty, AllowTwoPhase::No, CoerceNever::Yes, ); @@ -1113,7 +1047,7 @@ impl<'db> InferenceContext<'db> { AsmOperand::Label(expr) => { self.infer_expr( expr, - &Expectation::HasType(self.result.standard_types.unit.clone()), + &Expectation::HasType(self.types.unit), ExprIsRead::No, ); } @@ -1123,17 +1057,13 @@ impl<'db> InferenceContext<'db> { // FIXME: `sym` should report for things that are not functions or statics. AsmOperand::Sym(_) => (), }); - if diverge { - self.result.standard_types.never.clone() - } else { - self.result.standard_types.unit.clone() - } + if diverge { self.types.never } else { self.types.unit } } }; // use a new type variable if we got unknown here let ty = self.insert_type_vars_shallow(ty); - self.write_expr_ty(tgt_expr, ty.clone()); - if self.resolve_ty_shallow(&ty).is_never() + self.write_expr_ty(tgt_expr, ty); + if self.shallow_resolve(ty).is_never() && self.expr_guaranteed_to_constitute_read_for_never(tgt_expr, is_read) { // Any expression that produces a value of type `!` must have diverged @@ -1142,7 +1072,7 @@ impl<'db> InferenceContext<'db> { ty } - fn infer_expr_path(&mut self, path: &Path, id: ExprOrPatId, scope_id: ExprId) -> Ty { + fn infer_expr_path(&mut self, path: &Path, id: ExprOrPatId, scope_id: ExprId) -> Ty<'db> { let g = self.resolver.update_to_inner_scope(self.db, self.owner, scope_id); let ty = match self.infer_path(path, id) { Some(ty) => ty, @@ -1164,32 +1094,24 @@ impl<'db> InferenceContext<'db> { id: &Option<BlockId>, statements: &[Statement], tail: &Option<ExprId>, - ) -> Ty { - let ret_ty = self.table.new_type_var(); + ) -> Ty<'db> { + let ret_ty = self.table.next_ty_var(); let prev_diverges = mem::replace(&mut self.diverges, Diverges::Maybe); - let prev_ret_ty = mem::replace(&mut self.return_ty, ret_ty.clone()); - let prev_ret_coercion = self - .return_coercion - .replace(CoerceMany::new(ret_ty.to_nextsolver(self.table.interner))); + let prev_ret_ty = mem::replace(&mut self.return_ty, ret_ty); + let prev_ret_coercion = self.return_coercion.replace(CoerceMany::new(ret_ty)); // FIXME: We should handle async blocks like we handle closures let expected = &Expectation::has_type(ret_ty); let (_, inner_ty) = self.with_breakable_ctx(BreakableKind::Border, None, None, |this| { let ty = this.infer_block(tgt_expr, *id, statements, *tail, None, expected); if let Some(target) = expected.only_has_type(&mut this.table) { - match this.coerce( - tgt_expr.into(), - ty.to_nextsolver(this.table.interner), - target.to_nextsolver(this.table.interner), - AllowTwoPhase::No, - CoerceNever::Yes, - ) { - Ok(res) => res.to_chalk(this.table.interner), + match this.coerce(tgt_expr.into(), ty, target, AllowTwoPhase::No, CoerceNever::Yes) + { + Ok(res) => res, Err(_) => { - this.result.type_mismatches.insert( - tgt_expr.into(), - TypeMismatch { expected: target.clone(), actual: ty.clone() }, - ); + this.result + .type_mismatches + .insert(tgt_expr.into(), TypeMismatch { expected: target, actual: ty }); target } } @@ -1207,57 +1129,67 @@ impl<'db> InferenceContext<'db> { pub(crate) fn lower_async_block_type_impl_trait( &mut self, - inner_ty: Ty, + inner_ty: Ty<'db>, tgt_expr: ExprId, - ) -> Ty { + ) -> Ty<'db> { // Use the first type parameter as the output type of future. // existential type AsyncBlockImplTrait<InnerType>: Future<Output = InnerType> let impl_trait_id = crate::ImplTraitId::AsyncBlockTypeImplTrait(self.owner, tgt_expr); let opaque_ty_id = self.db.intern_impl_trait_id(impl_trait_id).into(); - TyKind::OpaqueType(opaque_ty_id, Substitution::from1(Interner, inner_ty)).intern(Interner) + Ty::new_alias( + self.interner(), + AliasTyKind::Opaque, + AliasTy::new( + self.interner(), + opaque_ty_id, + GenericArgs::new_from_iter(self.interner(), [inner_ty.into()]), + ), + ) } pub(crate) fn write_fn_trait_method_resolution( &mut self, fn_x: FnTrait, - derefed_callee: &Ty, - adjustments: &mut Vec<Adjustment>, - callee_ty: &Ty, - params: &[Ty], + derefed_callee: Ty<'db>, + adjustments: &mut Vec<Adjustment<'db>>, + callee_ty: Ty<'db>, + params: &[Ty<'db>], tgt_expr: ExprId, ) { match fn_x { FnTrait::FnOnce | FnTrait::AsyncFnOnce => (), FnTrait::FnMut | FnTrait::AsyncFnMut => { - if let TyKind::Ref(Mutability::Mut, lt, inner) = derefed_callee.kind(Interner) { + if let TyKind::Ref(lt, inner, Mutability::Mut) = derefed_callee.kind() { if adjustments .last() .map(|it| matches!(it.kind, Adjust::Borrow(_))) .unwrap_or(true) { // prefer reborrow to move - adjustments - .push(Adjustment { kind: Adjust::Deref(None), target: inner.clone() }); + adjustments.push(Adjustment { kind: Adjust::Deref(None), target: inner }); adjustments.push(Adjustment::borrow( + self.interner(), Mutability::Mut, - inner.clone(), - lt.clone(), + inner, + lt, )) } } else { adjustments.push(Adjustment::borrow( + self.interner(), Mutability::Mut, - derefed_callee.clone(), - self.table.new_lifetime_var(), + derefed_callee, + self.table.next_region_var(), )); } } FnTrait::Fn | FnTrait::AsyncFn => { - if !matches!(derefed_callee.kind(Interner), TyKind::Ref(Mutability::Not, _, _)) { + if !matches!(derefed_callee.kind(), TyKind::Ref(_, _, Mutability::Not)) { adjustments.push(Adjustment::borrow( + self.interner(), Mutability::Not, - derefed_callee.clone(), - self.table.new_lifetime_var(), + derefed_callee, + self.table.next_region_var(), )); } } @@ -1267,57 +1199,48 @@ impl<'db> InferenceContext<'db> { }; let trait_data = trait_.trait_items(self.db); if let Some(func) = trait_data.method_by_name(&fn_x.method_name()) { - let subst = TyBuilder::subst_for_def(self.db, trait_, None) - .push(callee_ty.clone()) - .push(TyBuilder::tuple_with(params.iter().cloned())) - .build(); + let subst = GenericArgs::new_from_iter( + self.interner(), + [ + callee_ty.into(), + Ty::new_tup_from_iter(self.interner(), params.iter().copied()).into(), + ], + ); self.write_method_resolution(tgt_expr, func, subst); } } - fn infer_expr_array( - &mut self, - array: &Array, - expected: &Expectation, - ) -> chalk_ir::Ty<Interner> { - let elem_ty = match expected.to_option(&mut self.table).as_ref().map(|t| t.kind(Interner)) { - Some(TyKind::Array(st, _) | TyKind::Slice(st)) => st.clone(), - _ => self.table.new_type_var(), + fn infer_expr_array(&mut self, array: &Array, expected: &Expectation<'db>) -> Ty<'db> { + let elem_ty = match expected.to_option(&mut self.table).as_ref().map(|t| t.kind()) { + Some(TyKind::Array(st, _) | TyKind::Slice(st)) => st, + _ => self.table.next_ty_var(), }; let krate = self.resolver.krate(); - let expected = Expectation::has_type(elem_ty.clone()); + let expected = Expectation::has_type(elem_ty); let (elem_ty, len) = match array { Array::ElementList { elements, .. } if elements.is_empty() => { (elem_ty, consteval::usize_const(self.db, Some(0), krate)) } Array::ElementList { elements, .. } => { - let mut coerce = CoerceMany::with_coercion_sites( - elem_ty.to_nextsolver(self.table.interner), - elements, - ); + let mut coerce = CoerceMany::with_coercion_sites(elem_ty, elements); for &expr in elements.iter() { let cur_elem_ty = self.infer_expr_inner(expr, &expected, ExprIsRead::Yes); - coerce.coerce( - self, - &ObligationCause::new(), - expr, - cur_elem_ty.to_nextsolver(self.table.interner), - ); + coerce.coerce(self, &ObligationCause::new(), expr, cur_elem_ty); } ( - coerce.complete(self).to_chalk(self.table.interner), + coerce.complete(self), consteval::usize_const(self.db, Some(elements.len() as u128), krate), ) } &Array::Repeat { initializer, repeat } => { self.infer_expr_coerce( initializer, - &Expectation::has_type(elem_ty.clone()), + &Expectation::has_type(elem_ty), ExprIsRead::Yes, ); - let usize = TyKind::Scalar(Scalar::Uint(UintTy::Usize)).intern(Interner); + let usize = self.types.usize; match self.body[repeat] { Expr::Underscore => { self.write_expr_ty(repeat, usize); @@ -1325,20 +1248,12 @@ impl<'db> InferenceContext<'db> { _ => _ = self.infer_expr(repeat, &Expectation::HasType(usize), ExprIsRead::Yes), } - ( - elem_ty, - consteval::eval_to_const( - repeat, - ParamLoweringMode::Placeholder, - self, - DebruijnIndex::INNERMOST, - ), - ) + (elem_ty, consteval::eval_to_const(repeat, self)) } }; // Try to evaluate unevaluated constant, and insert variable if is not possible. let len = self.table.insert_const_vars_shallow(len); - TyKind::Array(elem_ty, len).intern(Interner) + Ty::new_array_with_const_len(self.interner(), elem_ty, len) } pub(super) fn infer_return(&mut self, expr: ExprId) { @@ -1346,21 +1261,15 @@ impl<'db> InferenceContext<'db> { .return_coercion .as_mut() .expect("infer_return called outside function body") - .expected_ty() - .to_chalk(self.table.interner); + .expected_ty(); let return_expr_ty = self.infer_expr_inner(expr, &Expectation::HasType(ret_ty), ExprIsRead::Yes); let mut coerce_many = self.return_coercion.take().unwrap(); - coerce_many.coerce( - self, - &ObligationCause::new(), - expr, - return_expr_ty.to_nextsolver(self.table.interner), - ); + coerce_many.coerce(self, &ObligationCause::new(), expr, return_expr_ty); self.return_coercion = Some(coerce_many); } - fn infer_expr_return(&mut self, ret: ExprId, expr: Option<ExprId>) -> Ty { + fn infer_expr_return(&mut self, ret: ExprId, expr: Option<ExprId>) -> Ty<'db> { match self.return_coercion { Some(_) => { if let Some(expr) = expr { @@ -1378,23 +1287,20 @@ impl<'db> InferenceContext<'db> { } } } - self.result.standard_types.never.clone() + self.types.never } - fn infer_expr_become(&mut self, expr: ExprId) -> Ty { + fn infer_expr_become(&mut self, expr: ExprId) -> Ty<'db> { match &self.return_coercion { Some(return_coercion) => { - let ret_ty = return_coercion.expected_ty().to_chalk(self.table.interner); + let ret_ty = return_coercion.expected_ty(); - let call_expr_ty = self.infer_expr_inner( - expr, - &Expectation::HasType(ret_ty.clone()), - ExprIsRead::Yes, - ); + let call_expr_ty = + self.infer_expr_inner(expr, &Expectation::HasType(ret_ty), ExprIsRead::Yes); // NB: this should *not* coerce. // tail calls don't support any coercions except lifetimes ones (like `&'static u8 -> &'a u8`). - self.unify(&call_expr_ty, &ret_ty); + self.unify(call_expr_ty, ret_ty); } None => { // FIXME: diagnose `become` outside of functions @@ -1402,10 +1308,10 @@ impl<'db> InferenceContext<'db> { } } - self.result.standard_types.never.clone() + self.types.never } - fn infer_expr_box(&mut self, inner_expr: ExprId, expected: &Expectation) -> Ty { + fn infer_expr_box(&mut self, inner_expr: ExprId, expected: &Expectation<'db>) -> Ty<'db> { if let Some(box_id) = self.resolve_boxed_box() { let table = &mut self.table; let inner_exp = expected @@ -1414,16 +1320,22 @@ impl<'db> InferenceContext<'db> { .and_then(|e| e.as_adt()) .filter(|(e_adt, _)| e_adt == &box_id) .map(|(_, subts)| { - let g = subts.at(Interner, 0); - Expectation::rvalue_hint(self, Ty::clone(g.assert_ty_ref(Interner))) + let g = subts.type_at(0); + Expectation::rvalue_hint(self, g) }) .unwrap_or_else(Expectation::none); let inner_ty = self.infer_expr_inner(inner_expr, &inner_exp, ExprIsRead::Yes); - TyBuilder::adt(self.db, box_id) - .push(inner_ty) - .fill_with_defaults(self.db, || self.table.new_type_var()) - .build() + Ty::new_adt( + self.interner(), + box_id, + GenericArgs::fill_with_defaults( + self.interner(), + box_id.into(), + [inner_ty.into()], + |_, _, id, _| self.table.next_var_for_param(id), + ), + ) } else { self.err_ty() } @@ -1435,7 +1347,7 @@ impl<'db> InferenceContext<'db> { op: BinaryOp, rhs: ExprId, tgt_expr: ExprId, - ) -> Ty { + ) -> Ty<'db> { let lhs_expectation = Expectation::none(); let is_read = if matches!(op, BinaryOp::Assignment { .. }) { ExprIsRead::Yes @@ -1443,24 +1355,24 @@ impl<'db> InferenceContext<'db> { ExprIsRead::No }; let lhs_ty = self.infer_expr(lhs, &lhs_expectation, is_read); - let rhs_ty = self.table.new_type_var(); + let rhs_ty = self.table.next_ty_var(); let trait_func = lang_items_for_bin_op(op).and_then(|(name, lang_item)| { let trait_id = self.resolve_lang_item(lang_item)?.as_trait()?; let func = trait_id.trait_items(self.db).method_by_name(&name)?; Some((trait_id, func)) }); - let (trait_, func) = match trait_func { - Some(it) => it, + let func = match trait_func { + Some((_, it)) => it, None => { // HACK: `rhs_ty` is a general inference variable with no clue at all at this // point. Passing `lhs_ty` as both operands just to check if `lhs_ty` is a builtin // type applicable to `op`. - let ret_ty = if self.is_builtin_binop(&lhs_ty, &lhs_ty, op) { + let ret_ty = if self.is_builtin_binop(lhs_ty, lhs_ty, op) { // Assume both operands are builtin so we can continue inference. No guarantee // on the correctness, rustc would complain as necessary lang items don't seem // to exist anyway. - self.enforce_builtin_binop_types(&lhs_ty, &rhs_ty, op) + self.enforce_builtin_binop_types(lhs_ty, rhs_ty, op) } else { self.err_ty() }; @@ -1473,59 +1385,53 @@ impl<'db> InferenceContext<'db> { // HACK: We can use this substitution for the function because the function itself doesn't // have its own generic parameters. - let subst = TyBuilder::subst_for_def(self.db, trait_, None); - if subst.remaining() != 2 { - return Ty::new(Interner, TyKind::Error); - } - let subst = subst.push(lhs_ty.clone()).push(rhs_ty.clone()).build(); + let args = GenericArgs::new_from_iter(self.interner(), [lhs_ty.into(), rhs_ty.into()]); - self.write_method_resolution(tgt_expr, func, subst.clone()); + self.write_method_resolution(tgt_expr, func, args); - let interner = DbInterner::new_with(self.db, None, None); - let args: crate::next_solver::GenericArgs<'_> = subst.to_nextsolver(interner); - let method_ty = - self.db.value_ty(func.into()).unwrap().instantiate(interner, args).to_chalk(interner); - self.register_obligations_for_call(&method_ty); + let method_ty = self.db.value_ty(func.into()).unwrap().instantiate(self.interner(), args); + self.register_obligations_for_call(method_ty); - self.infer_expr_coerce(rhs, &Expectation::has_type(rhs_ty.clone()), ExprIsRead::Yes); + self.infer_expr_coerce(rhs, &Expectation::has_type(rhs_ty), ExprIsRead::Yes); - let ret_ty = match method_ty.callable_sig(self.db) { + let ret_ty = match method_ty.callable_sig(self.interner()) { Some(sig) => { - let p_left = &sig.params()[0]; + let sig = sig.skip_binder(); + let p_left = sig.inputs_and_output.as_slice()[0]; if matches!(op, BinaryOp::CmpOp(..) | BinaryOp::Assignment { .. }) - && let TyKind::Ref(mtbl, lt, _) = p_left.kind(Interner) + && let TyKind::Ref(lt, _, mtbl) = p_left.kind() { self.write_expr_adj( lhs, Box::new([Adjustment { - kind: Adjust::Borrow(AutoBorrow::Ref(lt.clone(), *mtbl)), - target: p_left.clone(), + kind: Adjust::Borrow(AutoBorrow::Ref(lt, mtbl)), + target: p_left, }]), ); } - let p_right = &sig.params()[1]; + let p_right = sig.inputs_and_output.as_slice()[1]; if matches!(op, BinaryOp::CmpOp(..)) - && let TyKind::Ref(mtbl, lt, _) = p_right.kind(Interner) + && let TyKind::Ref(lt, _, mtbl) = p_right.kind() { self.write_expr_adj( rhs, Box::new([Adjustment { - kind: Adjust::Borrow(AutoBorrow::Ref(lt.clone(), *mtbl)), - target: p_right.clone(), + kind: Adjust::Borrow(AutoBorrow::Ref(lt, mtbl)), + target: p_right, }]), ); } - sig.ret().clone() + sig.output() } None => self.err_ty(), }; let ret_ty = self.process_remote_user_written_ty(ret_ty); - if self.is_builtin_binop(&lhs_ty, &rhs_ty, op) { + if self.is_builtin_binop(lhs_ty, rhs_ty, op) { // use knowledge of built-in binary ops, which can sometimes help inference - let builtin_ret = self.enforce_builtin_binop_types(&lhs_ty, &rhs_ty, op); - self.unify(&builtin_ret, &ret_ty); + let builtin_ret = self.enforce_builtin_binop_types(lhs_ty, rhs_ty, op); + self.unify(builtin_ret, ret_ty); builtin_ret } else { ret_ty @@ -1539,8 +1445,8 @@ impl<'db> InferenceContext<'db> { statements: &[Statement], tail: Option<ExprId>, label: Option<LabelId>, - expected: &Expectation, - ) -> Ty { + expected: &Expectation<'db>, + ) -> Ty<'db> { let coerce_ty = expected.coercion_target_type(&mut self.table); let g = self.resolver.update_to_inner_scope(self.db, self.owner, expr); let prev_env = block_id.map(|block_id| { @@ -1557,7 +1463,7 @@ impl<'db> InferenceContext<'db> { let decl_ty = type_ref .as_ref() .map(|&tr| this.make_body_ty(tr)) - .unwrap_or_else(|| this.table.new_type_var()); + .unwrap_or_else(|| this.table.next_ty_var()); let ty = if let Some(expr) = initializer { // If we have a subpattern that performs a read, we want to consider this @@ -1571,13 +1477,13 @@ impl<'db> InferenceContext<'db> { let ty = if contains_explicit_ref_binding(this.body, *pat) { this.infer_expr( *expr, - &Expectation::has_type(decl_ty.clone()), + &Expectation::has_type(decl_ty), target_is_read, ) } else { this.infer_expr_coerce( *expr, - &Expectation::has_type(decl_ty.clone()), + &Expectation::has_type(decl_ty), target_is_read, ) }; @@ -1590,13 +1496,13 @@ impl<'db> InferenceContext<'db> { origin: DeclOrigin::LocalDecl { has_else: else_branch.is_some() }, }; - this.infer_top_pat(*pat, &ty, Some(decl)); + this.infer_top_pat(*pat, ty, Some(decl)); if let Some(expr) = else_branch { let previous_diverges = mem::replace(&mut this.diverges, Diverges::Maybe); this.infer_expr_coerce( *expr, - &Expectation::HasType(this.result.standard_types.never.clone()), + &Expectation::HasType(this.types.never), ExprIsRead::Yes, ); this.diverges = previous_diverges; @@ -1608,7 +1514,7 @@ impl<'db> InferenceContext<'db> { } else { this.infer_expr_coerce( expr, - &Expectation::HasType(this.result.standard_types.unit.clone()), + &Expectation::HasType(this.types.unit), ExprIsRead::Yes, ); } @@ -1642,8 +1548,8 @@ impl<'db> InferenceContext<'db> { if this .coerce( expr.into(), - this.result.standard_types.unit.to_nextsolver(this.table.interner), - t.to_nextsolver(this.table.interner), + this.types.unit, + t, AllowTwoPhase::No, coerce_never, ) @@ -1651,15 +1557,12 @@ impl<'db> InferenceContext<'db> { { this.result.type_mismatches.insert( expr.into(), - TypeMismatch { - expected: t.clone(), - actual: this.result.standard_types.unit.clone(), - }, + TypeMismatch { expected: t, actual: this.types.unit }, ); } t } else { - this.result.standard_types.unit.clone() + this.types.unit } } }); @@ -1673,32 +1576,30 @@ impl<'db> InferenceContext<'db> { fn lookup_field( &mut self, - receiver_ty: &Ty, + receiver_ty: Ty<'db>, name: &Name, - ) -> Option<(Ty, Either<FieldId, TupleFieldId>, Vec<Adjustment>, bool)> { - let interner = self.table.interner; - let mut autoderef = self.table.autoderef(receiver_ty.to_nextsolver(self.table.interner)); + ) -> Option<(Ty<'db>, Either<FieldId, TupleFieldId>, Vec<Adjustment<'db>>, bool)> { + let interner = self.interner(); + let mut autoderef = self.table.autoderef(receiver_ty); let mut private_field = None; let res = autoderef.by_ref().find_map(|(derefed_ty, _)| { let (field_id, parameters) = match derefed_ty.kind() { - crate::next_solver::TyKind::Tuple(substs) => { + TyKind::Tuple(substs) => { return name.as_tuple_index().and_then(|idx| { substs.as_slice().get(idx).copied().map(|ty| { ( Either::Right(TupleFieldId { tuple: TupleId( - self.tuple_field_accesses_rev - .insert_full(substs.to_chalk(interner)) - .0 as u32, + self.tuple_field_accesses_rev.insert_full(substs).0 as u32, ), index: idx as u32, }), - ty.to_chalk(interner), + ty, ) }) }); } - crate::next_solver::TyKind::Adt(adt, parameters) => match adt.def_id().0 { + TyKind::Adt(adt, parameters) => match adt.def_id().0 { hir_def::AdtId::StructId(s) => { let local_id = s.fields(self.db).field(name)?; let field = FieldId { parent: s.into(), local_id }; @@ -1713,18 +1614,16 @@ impl<'db> InferenceContext<'db> { }, _ => return None, }; - let parameters: crate::Substitution = parameters.to_chalk(interner); let is_visible = self.db.field_visibilities(field_id.parent)[field_id.local_id] .is_visible_from(self.db, self.resolver.module()); if !is_visible { if private_field.is_none() { - private_field = Some((field_id, parameters.clone())); + private_field = Some((field_id, parameters)); } return None; } - let ty = self.db.field_types(field_id.parent)[field_id.local_id] - .clone() - .substitute(Interner, ¶meters); + let ty = self.db.field_types_ns(field_id.parent)[field_id.local_id] + .instantiate(interner, parameters); Some((Either::Left(field_id), ty)) }); @@ -1738,9 +1637,8 @@ impl<'db> InferenceContext<'db> { None => { let (field_id, subst) = private_field?; let adjustments = autoderef.adjust_steps(); - let ty = self.db.field_types(field_id.parent)[field_id.local_id] - .clone() - .substitute(Interner, &subst); + let ty = self.db.field_types_ns(field_id.parent)[field_id.local_id] + .instantiate(self.interner(), subst); let ty = self.process_remote_user_written_ty(ty); (ty, Either::Left(field_id), adjustments, false) @@ -1753,8 +1651,8 @@ impl<'db> InferenceContext<'db> { tgt_expr: ExprId, receiver: ExprId, name: &Name, - expected: &Expectation, - ) -> Ty { + expected: &Expectation<'db>, + ) -> Ty<'db> { // Field projections don't constitute reads. let receiver_ty = self.infer_expr_inner(receiver, &Expectation::none(), ExprIsRead::No); @@ -1764,7 +1662,7 @@ impl<'db> InferenceContext<'db> { return self.err_ty(); } - match self.lookup_field(&receiver_ty, name) { + match self.lookup_field(receiver_ty, name) { Some((ty, field_id, adjustments, is_public)) => { self.write_expr_adj(receiver, adjustments.into_boxed_slice()); self.result.field_resolutions.insert(tgt_expr, field_id); @@ -1780,8 +1678,7 @@ impl<'db> InferenceContext<'db> { None => { // no field found, lets attempt to resolve it like a function so that IDE things // work out while people are typing - let canonicalized_receiver = - self.canonicalize(receiver_ty.clone().to_nextsolver(self.table.interner)); + let canonicalized_receiver = self.canonicalize(receiver_ty); let resolved = method_resolution::lookup_method( self.db, &canonicalized_receiver, @@ -1792,28 +1689,24 @@ impl<'db> InferenceContext<'db> { ); self.push_diagnostic(InferenceDiagnostic::UnresolvedField { expr: tgt_expr, - receiver: receiver_ty.clone(), + receiver: receiver_ty, name: name.clone(), method_with_same_name_exists: resolved.is_some(), }); match resolved { Some((adjust, func, _)) => { let (ty, adjustments) = adjust.apply(&mut self.table, receiver_ty); - let substs = self.substs_for_method_call(tgt_expr, func.into(), None); + let args = self.substs_for_method_call(tgt_expr, func.into(), None); self.write_expr_adj(receiver, adjustments.into_boxed_slice()); - self.write_method_resolution(tgt_expr, func, substs.clone()); + self.write_method_resolution(tgt_expr, func, args); - let interner = DbInterner::new_with(self.db, None, None); - let args: crate::next_solver::GenericArgs<'_> = - substs.to_nextsolver(interner); self.check_method_call( tgt_expr, &[], self.db .value_ty(func.into()) .unwrap() - .instantiate(interner, args) - .to_chalk(interner), + .instantiate(self.interner(), args), ty, expected, ) @@ -1829,47 +1722,42 @@ impl<'db> InferenceContext<'db> { tgt_expr: ExprId, callee: ExprId, args: &[ExprId], - expected: &Expectation, - ) -> Ty { + expected: &Expectation<'db>, + ) -> Ty<'db> { let callee_ty = self.infer_expr(callee, &Expectation::none(), ExprIsRead::Yes); - let interner = self.table.interner; - let mut derefs = self.table.autoderef(callee_ty.to_nextsolver(interner)); + let interner = self.interner(); + let mut derefs = self.table.autoderef(callee_ty); let (res, derefed_callee) = loop { let Some((callee_deref_ty, _)) = derefs.next() else { - break (None, callee_ty.clone()); + break (None, callee_ty); }; - let callee_deref_ty = callee_deref_ty.to_chalk(interner); - if let Some(res) = derefs.table.callable_sig(&callee_deref_ty, args.len()) { + if let Some(res) = derefs.table.callable_sig(callee_deref_ty, args.len()) { break (Some(res), callee_deref_ty); } }; // if the function is unresolved, we use is_varargs=true to // suppress the arg count diagnostic here - let is_varargs = - derefed_callee.callable_sig(self.db).is_some_and(|sig| sig.is_varargs) || res.is_none(); + let is_varargs = derefed_callee.callable_sig(interner).is_some_and(|sig| sig.c_variadic()) + || res.is_none(); let (param_tys, ret_ty) = match res { Some((func, params, ret_ty)) => { - let params_chalk = - params.iter().map(|param| param.to_chalk(interner)).collect::<Vec<_>>(); let mut adjustments = derefs.adjust_steps(); if let Some(fn_x) = func { self.write_fn_trait_method_resolution( fn_x, - &derefed_callee, + derefed_callee, &mut adjustments, - &callee_ty, - ¶ms_chalk, + callee_ty, + ¶ms, tgt_expr, ); } - if let &TyKind::Closure(c, _) = - self.table.resolve_completely(callee_ty.clone()).kind(Interner) - { + if let TyKind::Closure(c, _) = self.table.resolve_completely(callee_ty).kind() { self.add_current_closure_dependency(c.into()); self.deferred_closures.entry(c.into()).or_default().push(( - derefed_callee.clone(), - callee_ty.clone(), - params_chalk, + derefed_callee, + callee_ty, + params.clone(), tgt_expr, )); } @@ -1879,9 +1767,9 @@ impl<'db> InferenceContext<'db> { None => { self.push_diagnostic(InferenceDiagnostic::ExpectedFunction { call_expr: tgt_expr, - found: callee_ty.clone(), + found: callee_ty, }); - (Vec::new(), crate::next_solver::Ty::new_error(interner, ErrorGuaranteed)) + (Vec::new(), Ty::new_error(interner, ErrorGuaranteed)) } }; let indices_to_skip = self.check_legacy_const_generics(derefed_callee, args); @@ -1901,14 +1789,14 @@ impl<'db> InferenceContext<'db> { &mut self, tgt_expr: ExprId, args: &[ExprId], - callee_ty: Ty, - param_tys: &[crate::next_solver::Ty<'db>], - ret_ty: crate::next_solver::Ty<'db>, + callee_ty: Ty<'db>, + param_tys: &[Ty<'db>], + ret_ty: Ty<'db>, indices_to_skip: &[u32], is_varargs: bool, - expected: &Expectation, - ) -> Ty { - self.register_obligations_for_call(&callee_ty); + expected: &Expectation<'db>, + ) -> Ty<'db> { + self.register_obligations_for_call(callee_ty); self.check_call_arguments( tgt_expr, @@ -1919,7 +1807,7 @@ impl<'db> InferenceContext<'db> { indices_to_skip, is_varargs, ); - self.table.normalize_associated_types_in_ns(ret_ty).to_chalk(self.table.interner) + self.table.normalize_associated_types_in(ret_ty) } fn infer_method_call( @@ -1928,16 +1816,13 @@ impl<'db> InferenceContext<'db> { receiver: ExprId, args: &[ExprId], method_name: &Name, - generic_args: Option<&GenericArgs>, - expected: &Expectation, - ) -> Ty { + generic_args: Option<&HirGenericArgs>, + expected: &Expectation<'db>, + ) -> Ty<'db> { let receiver_ty = self.infer_expr_inner(receiver, &Expectation::none(), ExprIsRead::Yes); - let receiver_ty = self.table.structurally_resolve_type(&receiver_ty); + let receiver_ty = self.table.try_structurally_resolve_type(receiver_ty); - if matches!( - receiver_ty.kind(Interner), - TyKind::Error | TyKind::InferenceVar(_, TyVariableKind::General) - ) { + if matches!(receiver_ty.kind(), TyKind::Error(_) | TyKind::Infer(InferTy::TyVar(_))) { // Don't probe on error type, or on a fully unresolved infer var. // FIXME: Emit an error if we're probing on an infer var (type annotations needed). for &arg in args { @@ -1947,8 +1832,7 @@ impl<'db> InferenceContext<'db> { return receiver_ty; } - let canonicalized_receiver = - self.canonicalize(receiver_ty.clone().to_nextsolver(self.table.interner)); + let canonicalized_receiver = self.canonicalize(receiver_ty); let resolved = method_resolution::lookup_method( self.db, @@ -1970,18 +1854,16 @@ impl<'db> InferenceContext<'db> { let (ty, adjustments) = adjust.apply(&mut self.table, receiver_ty); self.write_expr_adj(receiver, adjustments.into_boxed_slice()); - let substs = self.substs_for_method_call(tgt_expr, func.into(), generic_args); - self.write_method_resolution(tgt_expr, func, substs.clone()); + let gen_args = self.substs_for_method_call(tgt_expr, func.into(), generic_args); + self.write_method_resolution(tgt_expr, func, gen_args); let interner = DbInterner::new_with(self.db, None, None); - let gen_args: crate::next_solver::GenericArgs<'_> = substs.to_nextsolver(interner); self.check_method_call( tgt_expr, args, self.db .value_ty(func.into()) .expect("we have a function def") - .instantiate(interner, gen_args) - .to_chalk(interner), + .instantiate(interner, gen_args), ty, expected, ) @@ -1989,7 +1871,7 @@ impl<'db> InferenceContext<'db> { // Failed to resolve, report diagnostic and try to resolve as call to field access or // assoc function None => { - let field_with_same_name_exists = match self.lookup_field(&receiver_ty, method_name) + let field_with_same_name_exists = match self.lookup_field(receiver_ty, method_name) { Some((ty, field_id, adjustments, _public)) => { self.write_expr_adj(receiver, adjustments.into_boxed_slice()); @@ -2017,48 +1899,46 @@ impl<'db> InferenceContext<'db> { self.push_diagnostic(InferenceDiagnostic::UnresolvedMethodCall { expr: tgt_expr, - receiver: receiver_ty.clone(), + receiver: receiver_ty, name: method_name.clone(), - field_with_same_name: field_with_same_name_exists.clone(), + field_with_same_name: field_with_same_name_exists, assoc_func_with_same_name, }); let recovered = match assoc_func_with_same_name { Some(f) => { - let substs = self.substs_for_method_call(tgt_expr, f.into(), generic_args); + let args = self.substs_for_method_call(tgt_expr, f.into(), generic_args); let interner = DbInterner::new_with(self.db, None, None); - let args: crate::next_solver::GenericArgs<'_> = - substs.to_nextsolver(interner); let f = self .db .value_ty(f.into()) .expect("we have a function def") - .instantiate(interner, args) - .to_chalk(interner); - let sig = f.callable_sig(self.db).expect("we have a function def"); + .instantiate(interner, args); + let sig = f.callable_sig(self.interner()).expect("we have a function def"); Some((f, sig, true)) } None => field_with_same_name_exists.and_then(|field_ty| { - let callable_sig = field_ty.callable_sig(self.db)?; + let callable_sig = field_ty.callable_sig(self.interner())?; Some((field_ty, callable_sig, false)) }), }; match recovered { - Some((callee_ty, sig, strip_first)) => self.check_call( - tgt_expr, - args, - callee_ty, - &sig.params() - .get(strip_first as usize..) - .unwrap_or(&[]) - .iter() - .map(|param| param.to_nextsolver(self.table.interner)) - .collect::<Vec<_>>(), - sig.ret().to_nextsolver(self.table.interner), - &[], - true, - expected, - ), + Some((callee_ty, sig, strip_first)) => { + let sig = sig.skip_binder(); + self.check_call( + tgt_expr, + args, + callee_ty, + sig.inputs_and_output + .inputs() + .get(strip_first as usize..) + .unwrap_or(&[]), + sig.output(), + &[], + true, + expected, + ) + } None => { for &arg in args.iter() { self.infer_expr_no_expect(arg, ExprIsRead::Yes); @@ -2074,39 +1954,38 @@ impl<'db> InferenceContext<'db> { &mut self, tgt_expr: ExprId, args: &[ExprId], - method_ty: Ty, - receiver_ty: Ty, - expected: &Expectation, - ) -> Ty { - self.register_obligations_for_call(&method_ty); - let interner = self.table.interner; + method_ty: Ty<'db>, + receiver_ty: Ty<'db>, + expected: &Expectation<'db>, + ) -> Ty<'db> { + self.register_obligations_for_call(method_ty); let ((formal_receiver_ty, param_tys), ret_ty, is_varargs) = - match method_ty.callable_sig(self.db) { - Some(sig) => ( - if !sig.params().is_empty() { - ( - sig.params()[0].to_nextsolver(interner), - sig.params()[1..] - .iter() - .map(|param| param.to_nextsolver(interner)) - .collect(), - ) - } else { - (crate::next_solver::Ty::new_error(interner, ErrorGuaranteed), Vec::new()) - }, - sig.ret().to_nextsolver(interner), - sig.is_varargs, - ), + match method_ty.callable_sig(self.interner()) { + Some(sig) => { + let sig = sig.skip_binder(); + ( + if !sig.inputs_and_output.inputs().is_empty() { + ( + sig.inputs_and_output.as_slice()[0], + sig.inputs_and_output.inputs()[1..].to_vec(), + ) + } else { + (self.types.error, Vec::new()) + }, + sig.output(), + sig.c_variadic, + ) + } None => { let formal_receiver_ty = self.table.next_ty_var(); let ret_ty = self.table.next_ty_var(); ((formal_receiver_ty, Vec::new()), ret_ty, true) } }; - self.table.unify_ns(formal_receiver_ty, receiver_ty.to_nextsolver(interner)); + self.table.unify(formal_receiver_ty, receiver_ty); self.check_call_arguments(tgt_expr, ¶m_tys, ret_ty, expected, args, &[], is_varargs); - self.table.normalize_associated_types_in_ns(ret_ty).to_chalk(interner) + self.table.normalize_associated_types_in(ret_ty) } /// Generic function that factors out common logic from function calls, @@ -2115,18 +1994,16 @@ impl<'db> InferenceContext<'db> { &mut self, call_expr: ExprId, // Types (as defined in the *signature* of the target function) - formal_input_tys: &[crate::next_solver::Ty<'db>], - formal_output: crate::next_solver::Ty<'db>, + formal_input_tys: &[Ty<'db>], + formal_output: Ty<'db>, // Expected output from the parent expression or statement - expectation: &Expectation, + expectation: &Expectation<'db>, // The expressions for each provided argument provided_args: &[ExprId], skip_indices: &[u32], // Whether the function is variadic, for example when imported from C c_variadic: bool, ) { - let interner = self.table.interner; - // First, let's unify the formal method signature with the expectation eagerly. // We use this to guide coercion inference; it's output is "fudged" which means // any remaining type variables are assigned to new, unrelated variables. This @@ -2145,14 +2022,9 @@ impl<'db> InferenceContext<'db> { // is polymorphic) and the expected return type. // No argument expectations are produced if unification fails. let origin = ObligationCause::new(); - ocx.sup( - &origin, - self.table.trait_env.env, - expected_output.to_nextsolver(interner), - formal_output, - )?; + ocx.sup(&origin, self.table.trait_env.env, expected_output, formal_output)?; if !ocx.try_evaluate_obligations().is_empty() { - return Err(crate::next_solver::TypeError::Mismatch); + return Err(TypeError::Mismatch); } // Record all the argument types, with the args @@ -2200,9 +2072,9 @@ impl<'db> InferenceContext<'db> { // We introduce a helper function to demand that a given argument satisfy a given input // This is more complicated than just checking type equality, as arguments could be coerced // This version writes those types back so further type checking uses the narrowed types - let demand_compatible = |this: &mut InferenceContext<'db>, idx| { - let formal_input_ty: crate::next_solver::Ty<'db> = formal_input_tys[idx]; - let expected_input_ty: crate::next_solver::Ty<'db> = expected_input_tys[idx]; + let demand_compatible = |this: &mut InferenceContext<'_, 'db>, idx| { + let formal_input_ty: Ty<'db> = formal_input_tys[idx]; + let expected_input_ty: Ty<'db> = expected_input_tys[idx]; let provided_arg = provided_args[idx]; debug!("checking argument {}: {:?} = {:?}", idx, provided_arg, formal_input_ty); @@ -2210,19 +2082,14 @@ impl<'db> InferenceContext<'db> { // We're on the happy path here, so we'll do a more involved check and write back types // To check compatibility, we'll do 3 things: // 1. Unify the provided argument with the expected type - let expectation = Expectation::rvalue_hint(this, expected_input_ty.to_chalk(interner)); + let expectation = Expectation::rvalue_hint(this, expected_input_ty); - let checked_ty = this - .infer_expr_inner(provided_arg, &expectation, ExprIsRead::Yes) - .to_nextsolver(interner); + let checked_ty = this.infer_expr_inner(provided_arg, &expectation, ExprIsRead::Yes); // 2. Coerce to the most detailed type that could be coerced // to, which is `expected_ty` if `rvalue_hint` returns an // `ExpectHasType(expected_ty)`, or the `formal_ty` otherwise. - let coerced_ty = expectation - .only_has_type(&mut this.table) - .map(|it| it.to_nextsolver(interner)) - .unwrap_or(formal_input_ty); + let coerced_ty = expectation.only_has_type(&mut this.table).unwrap_or(formal_input_ty); // Cause selection errors caused by resolving a single argument to point at the // argument and not the call. This lets us customize the span pointed to in the @@ -2259,7 +2126,7 @@ impl<'db> InferenceContext<'db> { // If neither check failed, the types are compatible match formal_ty_error { - Ok(crate::next_solver::infer::InferOk { obligations, value: () }) => { + Ok(InferOk { obligations, value: () }) => { this.table.register_predicates(obligations); Ok(()) } @@ -2313,13 +2180,9 @@ impl<'db> InferenceContext<'db> { && args_count_matches { // Don't report type mismatches if there is a mismatch in args count. - self.result.type_mismatches.insert( - (*arg).into(), - TypeMismatch { - expected: expected.to_chalk(interner), - actual: found.to_chalk(interner), - }, - ); + self.result + .type_mismatches + .insert((*arg).into(), TypeMismatch { expected, actual: found }); } } } @@ -2331,14 +2194,14 @@ impl<'db> InferenceContext<'db> { &mut self, expr: ExprId, def: GenericDefId, - generic_args: Option<&GenericArgs>, - ) -> Substitution { - struct LowererCtx<'a, 'b> { - ctx: &'a mut InferenceContext<'b>, + generic_args: Option<&HirGenericArgs>, + ) -> GenericArgs<'db> { + struct LowererCtx<'a, 'b, 'db> { + ctx: &'a mut InferenceContext<'b, 'db>, expr: ExprId, } - impl GenericArgsLowerer for LowererCtx<'_, '_> { + impl<'db> GenericArgsLowerer<'db> for LowererCtx<'_, '_, 'db> { fn report_len_mismatch( &mut self, def: GenericDefId, @@ -2373,21 +2236,22 @@ impl<'db> InferenceContext<'db> { &mut self, param_id: GenericParamId, param: GenericParamDataRef<'_>, - arg: &GenericArg, - ) -> crate::GenericArg { + arg: &HirGenericArg, + ) -> GenericArg<'db> { match (param, arg) { - (GenericParamDataRef::LifetimeParamData(_), GenericArg::Lifetime(lifetime)) => { - self.ctx.make_body_lifetime(*lifetime).cast(Interner) + ( + GenericParamDataRef::LifetimeParamData(_), + HirGenericArg::Lifetime(lifetime), + ) => self.ctx.make_body_lifetime(*lifetime).into(), + (GenericParamDataRef::TypeParamData(_), HirGenericArg::Type(type_ref)) => { + self.ctx.make_body_ty(*type_ref).into() } - (GenericParamDataRef::TypeParamData(_), GenericArg::Type(type_ref)) => { - self.ctx.make_body_ty(*type_ref).cast(Interner) - } - (GenericParamDataRef::ConstParamData(_), GenericArg::Const(konst)) => { + (GenericParamDataRef::ConstParamData(_), HirGenericArg::Const(konst)) => { let GenericParamId::ConstParamId(const_id) = param_id else { unreachable!("non-const param ID for const param"); }; - let const_ty = self.ctx.db.const_param_ty(const_id); - self.ctx.make_body_const(*konst, const_ty).cast(Interner) + let const_ty = self.ctx.db.const_param_ty_ns(const_id); + self.ctx.make_body_const(*konst, const_ty).into() } _ => unreachable!("unmatching param kinds were passed to `provided_kind()`"), } @@ -2395,12 +2259,12 @@ impl<'db> InferenceContext<'db> { fn provided_type_like_const( &mut self, - const_ty: Ty, + const_ty: Ty<'db>, arg: TypeLikeConst<'_>, - ) -> crate::Const { + ) -> Const<'db> { match arg { TypeLikeConst::Path(path) => self.ctx.make_path_as_body_const(path, const_ty), - TypeLikeConst::Infer => self.ctx.table.new_const_var(const_ty), + TypeLikeConst::Infer => self.ctx.table.next_const_var(), } } @@ -2410,35 +2274,15 @@ impl<'db> InferenceContext<'db> { param_id: GenericParamId, _param: GenericParamDataRef<'_>, _infer_args: bool, - _preceding_args: &[crate::GenericArg], - ) -> crate::GenericArg { + _preceding_args: &[GenericArg<'db>], + ) -> GenericArg<'db> { // Always create an inference var, even when `infer_args == false`. This helps with diagnostics, // and I think it's also required in the presence of `impl Trait` (that must be inferred). - match param_id { - GenericParamId::TypeParamId(_) => self.ctx.table.new_type_var().cast(Interner), - GenericParamId::ConstParamId(const_id) => self - .ctx - .table - .new_const_var(self.ctx.db.const_param_ty(const_id)) - .cast(Interner), - GenericParamId::LifetimeParamId(_) => { - self.ctx.table.new_lifetime_var().cast(Interner) - } - } + self.ctx.table.next_var_for_param(param_id) } - fn parent_arg(&mut self, param_id: GenericParamId) -> crate::GenericArg { - match param_id { - GenericParamId::TypeParamId(_) => self.ctx.table.new_type_var().cast(Interner), - GenericParamId::ConstParamId(const_id) => self - .ctx - .table - .new_const_var(self.ctx.db.const_param_ty(const_id)) - .cast(Interner), - GenericParamId::LifetimeParamId(_) => { - self.ctx.table.new_lifetime_var().cast(Interner) - } - } + fn parent_arg(&mut self, param_id: GenericParamId) -> GenericArg<'db> { + self.ctx.table.next_var_for_param(param_id) } fn report_elided_lifetimes_in_path( @@ -2472,36 +2316,34 @@ impl<'db> InferenceContext<'db> { ) } - fn register_obligations_for_call(&mut self, callable_ty: &Ty) { - let callable_ty = self.table.structurally_resolve_type(callable_ty); - if let TyKind::FnDef(fn_def, parameters) = callable_ty.kind(Interner) { - let def: CallableDefId = from_chalk(self.db, *fn_def); + fn register_obligations_for_call(&mut self, callable_ty: Ty<'db>) { + let callable_ty = self.table.try_structurally_resolve_type(callable_ty); + if let TyKind::FnDef(fn_def, parameters) = callable_ty.kind() { let generic_predicates = - self.db.generic_predicates(GenericDefId::from_callable(self.db, def)); - for predicate in generic_predicates.iter() { - let (predicate, binders) = predicate - .clone() - .substitute(Interner, parameters) - .into_value_and_skipped_binders(); - always!(binders.len(Interner) == 0); // quantified where clauses not yet handled - self.push_obligation(predicate.cast(Interner)); + self.db.generic_predicates_ns(GenericDefId::from_callable(self.db, fn_def.0)); + if let Some(predicates) = generic_predicates.instantiate(self.interner(), parameters) { + let interner = self.interner(); + let param_env = self.table.trait_env.env; + self.table.register_predicates(predicates.map(|predicate| { + Obligation::new(interner, ObligationCause::new(), param_env, predicate) + })); } // add obligation for trait implementation, if this is a trait method - match def { + match fn_def.0 { CallableDefId::FunctionId(f) => { if let ItemContainerId::TraitId(trait_) = f.lookup(self.db).container { // construct a TraitRef let trait_params_len = generics(self.db, trait_.into()).len(); - let substs = Substitution::from_iter( - Interner, - // The generic parameters for the trait come after those for the - // function. - ¶meters.as_slice(Interner)[..trait_params_len], - ); - self.push_obligation( - TraitRef { trait_id: to_chalk_trait_id(trait_), substitution: substs } - .cast(Interner), + let substs = GenericArgs::new_from_iter( + self.interner(), + parameters.as_slice()[..trait_params_len].iter().copied(), ); + self.table.register_predicate(Obligation::new( + self.interner(), + ObligationCause::new(), + self.table.trait_env.env, + TraitRef::new(self.interner(), trait_.into(), substs), + )); } } CallableDefId::StructId(_) | CallableDefId::EnumVariantId(_) => {} @@ -2510,11 +2352,10 @@ impl<'db> InferenceContext<'db> { } /// Returns the argument indices to skip. - fn check_legacy_const_generics(&mut self, callee: Ty, args: &[ExprId]) -> Box<[u32]> { - let (func, subst) = match callee.kind(Interner) { - TyKind::FnDef(fn_id, subst) => { - let callable = CallableDefId::from_chalk(self.db, *fn_id); - let func = match callable { + fn check_legacy_const_generics(&mut self, callee: Ty<'db>, args: &[ExprId]) -> Box<[u32]> { + let (func, _subst) = match callee.kind() { + TyKind::FnDef(callable, subst) => { + let func = match callable.0 { CallableDefId::FunctionId(f) => f, _ => return Default::default(), }; @@ -2542,15 +2383,10 @@ impl<'db> InferenceContext<'db> { } // check legacy const parameters - for (subst_idx, arg_idx) in legacy_const_generics_indices.iter().copied().enumerate() { - let arg = match subst.at(Interner, subst_idx).constant(Interner) { - Some(c) => c, - None => continue, // not a const parameter? - }; + for arg_idx in legacy_const_generics_indices.iter().copied() { if arg_idx >= args.len() as u32 { continue; } - let _ty = arg.data(Interner).ty.clone(); let expected = Expectation::none(); // FIXME use actual const ty, when that is lowered correctly self.infer_expr(args[arg_idx as usize], &expected, ExprIsRead::Yes); // FIXME: evaluate and unify with the const @@ -2561,17 +2397,19 @@ impl<'db> InferenceContext<'db> { } /// Dereferences a single level of immutable referencing. - fn deref_ty_if_possible(&mut self, ty: &Ty) -> Ty { - let ty = self.table.structurally_resolve_type(ty); - match ty.kind(Interner) { - TyKind::Ref(Mutability::Not, _, inner) => self.table.structurally_resolve_type(inner), + fn deref_ty_if_possible(&mut self, ty: Ty<'db>) -> Ty<'db> { + let ty = self.table.try_structurally_resolve_type(ty); + match ty.kind() { + TyKind::Ref(_, inner, Mutability::Not) => { + self.table.try_structurally_resolve_type(inner) + } _ => ty, } } /// Enforces expectations on lhs type and rhs type depending on the operator and returns the /// output type of the binary op. - fn enforce_builtin_binop_types(&mut self, lhs: &Ty, rhs: &Ty, op: BinaryOp) -> Ty { + fn enforce_builtin_binop_types(&mut self, lhs: Ty<'db>, rhs: Ty<'db>, op: BinaryOp) -> Ty<'db> { // Special-case a single layer of referencing, so that things like `5.0 + &6.0f32` work (See rust-lang/rust#57447). let lhs = self.deref_ty_if_possible(lhs); let rhs = self.deref_ty_if_possible(rhs); @@ -2583,9 +2421,9 @@ impl<'db> InferenceContext<'db> { let output_ty = match op { BinaryOp::LogicOp(_) => { - let bool_ = self.result.standard_types.bool_.clone(); - self.unify(&lhs, &bool_); - self.unify(&rhs, &bool_); + let bool_ = self.types.bool; + self.unify(lhs, bool_); + self.unify(rhs, bool_); bool_ } @@ -2596,14 +2434,14 @@ impl<'db> InferenceContext<'db> { BinaryOp::ArithOp(_) => { // LHS, RHS, and result will have the same type - self.unify(&lhs, &rhs); + self.unify(lhs, rhs); lhs } BinaryOp::CmpOp(_) => { // LHS and RHS will have the same type - self.unify(&lhs, &rhs); - self.result.standard_types.bool_.clone() + self.unify(lhs, rhs); + self.types.bool } BinaryOp::Assignment { op: None } => { @@ -2614,10 +2452,10 @@ impl<'db> InferenceContext<'db> { BinaryOp::Assignment { .. } => unreachable!("handled above"), }; - if is_assign { self.result.standard_types.unit.clone() } else { output_ty } + if is_assign { self.types.unit } else { output_ty } } - fn is_builtin_binop(&mut self, lhs: &Ty, rhs: &Ty, op: BinaryOp) -> bool { + fn is_builtin_binop(&mut self, lhs: Ty<'db>, rhs: Ty<'db>, op: BinaryOp) -> bool { // Special-case a single layer of referencing, so that things like `5.0 + &6.0f32` work (See rust-lang/rust#57447). let lhs = self.deref_ty_if_possible(lhs); let rhs = self.deref_ty_if_possible(rhs); @@ -2644,27 +2482,25 @@ impl<'db> InferenceContext<'db> { BinaryOp::ArithOp(ArithOp::BitAnd | ArithOp::BitOr | ArithOp::BitXor) => { lhs.is_integral() && rhs.is_integral() || lhs.is_floating_point() && rhs.is_floating_point() - || matches!( - (lhs.kind(Interner), rhs.kind(Interner)), - (TyKind::Scalar(Scalar::Bool), TyKind::Scalar(Scalar::Bool)) - ) + || matches!((lhs.kind(), rhs.kind()), (TyKind::Bool, TyKind::Bool)) } BinaryOp::CmpOp(_) => { let is_scalar = |kind| { matches!( kind, - &TyKind::Scalar(_) + TyKind::Bool + | TyKind::Char + | TyKind::Int(_) + | TyKind::Uint(_) + | TyKind::Float(_) | TyKind::FnDef(..) - | TyKind::Function(_) - | TyKind::Raw(..) - | TyKind::InferenceVar( - _, - TyVariableKind::Integer | TyVariableKind::Float - ) + | TyKind::FnPtr(..) + | TyKind::RawPtr(..) + | TyKind::Infer(InferTy::IntVar(_) | InferTy::FloatVar(_)) ) }; - is_scalar(lhs.kind(Interner)) && is_scalar(rhs.kind(Interner)) + is_scalar(lhs.kind()) && is_scalar(rhs.kind()) } BinaryOp::Assignment { op: None } => { @@ -2679,27 +2515,15 @@ impl<'db> InferenceContext<'db> { pub(super) fn with_breakable_ctx<T>( &mut self, kind: BreakableKind, - ty: Option<Ty>, + ty: Option<Ty<'db>>, label: Option<LabelId>, cb: impl FnOnce(&mut Self) -> T, - ) -> (Option<Ty>, T) { + ) -> (Option<Ty<'db>>, T) { self.breakables.push({ - BreakableContext { - kind, - may_break: false, - coerce: ty.map(|ty| CoerceMany::new(ty.to_nextsolver(self.table.interner))), - label, - } + BreakableContext { kind, may_break: false, coerce: ty.map(CoerceMany::new), label } }); let res = cb(self); let ctx = self.breakables.pop().expect("breakable stack broken"); - ( - if ctx.may_break { - ctx.coerce.map(|ctx| ctx.complete(self).to_chalk(self.table.interner)) - } else { - None - }, - res, - ) + (if ctx.may_break { ctx.coerce.map(|ctx| ctx.complete(self)) } else { None }, res) } } |