Fix shifting of binders in FnPointer
- don't shift in/out for Chalk mapping (we want to have the same binders now) - do shift in when creating the signature for a closure (though it shouldn't matter much) - do shift in when lowering a `fn()` type - correctly deal with the implied binder in TypeWalk
This commit is contained in:
parent
edc59d897d
commit
1ae967bf8e
@ -23,7 +23,7 @@
|
||||
traits::{chalk::from_chalk, FnTrait},
|
||||
utils::{generics, variant_data, Generics},
|
||||
AdtId, Binders, CallableDefId, FnPointer, FnSig, FnSubst, InEnvironment, Interner,
|
||||
ProjectionTyExt, Rawness, Scalar, Substitution, TraitRef, Ty, TyBuilder, TyKind,
|
||||
ProjectionTyExt, Rawness, Scalar, Substitution, TraitRef, Ty, TyBuilder, TyKind, TypeWalk,
|
||||
};
|
||||
|
||||
use super::{
|
||||
@ -262,7 +262,9 @@ fn infer_expr_inner(&mut self, tgt_expr: ExprId, expected: &Expectation) -> Ty {
|
||||
let sig_ty = TyKind::Function(FnPointer {
|
||||
num_binders: 0,
|
||||
sig: FnSig { abi: (), safety: chalk_ir::Safety::Safe, variadic: false },
|
||||
substitution: FnSubst(Substitution::from_iter(&Interner, sig_tys.clone())),
|
||||
substitution: FnSubst(
|
||||
Substitution::from_iter(&Interner, sig_tys.clone()).shifted_in(&Interner),
|
||||
),
|
||||
})
|
||||
.intern(&Interner);
|
||||
let closure_id = self.db.intern_closure((self.owner, tgt_expr)).into();
|
||||
|
@ -178,8 +178,9 @@ pub fn lower_ty_ext(&self, type_ref: &TypeRef) -> (Ty, Option<TypeNs>) {
|
||||
}
|
||||
TypeRef::Placeholder => TyKind::Error.intern(&Interner),
|
||||
TypeRef::Fn(params, is_varargs) => {
|
||||
let substs =
|
||||
Substitution::from_iter(&Interner, params.iter().map(|tr| self.lower_ty(tr)));
|
||||
let substs = self.with_shifted_in(DebruijnIndex::ONE, |ctx| {
|
||||
Substitution::from_iter(&Interner, params.iter().map(|tr| ctx.lower_ty(tr)))
|
||||
});
|
||||
TyKind::Function(FnPointer {
|
||||
num_binders: 0, // FIXME lower `for<'a> fn()` correctly
|
||||
sig: FnSig { abi: (), safety: Safety::Safe, variadic: *is_varargs },
|
||||
|
@ -3,7 +3,7 @@
|
||||
//! Chalk (in both directions); plus some helper functions for more specialized
|
||||
//! conversions.
|
||||
|
||||
use chalk_ir::{cast::Cast, fold::shift::Shift, interner::HasInterner, LifetimeData};
|
||||
use chalk_ir::{cast::Cast, interner::HasInterner, LifetimeData};
|
||||
use chalk_solve::rust_ir;
|
||||
|
||||
use base_db::salsa::InternKey;
|
||||
@ -25,7 +25,7 @@ fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::Ty<Interner> {
|
||||
TyKind::Ref(m, ty) => ref_to_chalk(db, m, ty),
|
||||
TyKind::Array(ty) => array_to_chalk(db, ty),
|
||||
TyKind::Function(FnPointer { sig, substitution: substs, .. }) => {
|
||||
let substitution = chalk_ir::FnSubst(substs.0.to_chalk(db).shifted_in(&Interner));
|
||||
let substitution = chalk_ir::FnSubst(substs.0.to_chalk(db));
|
||||
chalk_ir::TyKind::Function(chalk_ir::FnPointer {
|
||||
num_binders: 0,
|
||||
sig,
|
||||
@ -132,10 +132,7 @@ fn from_chalk(db: &dyn HirDatabase, chalk: chalk_ir::Ty<Interner>) -> Self {
|
||||
..
|
||||
}) => {
|
||||
assert_eq!(num_binders, 0);
|
||||
let substs = crate::FnSubst(from_chalk(
|
||||
db,
|
||||
substitution.0.shifted_out(&Interner).expect("fn ptr should have no binders"),
|
||||
));
|
||||
let substs = crate::FnSubst(from_chalk(db, substitution.0));
|
||||
TyKind::Function(FnPointer { num_binders, sig, substitution: substs })
|
||||
}
|
||||
chalk_ir::TyKind::BoundVar(idx) => TyKind::BoundVar(idx),
|
||||
|
@ -92,6 +92,13 @@ fn subst_bound_vars_at_depth(mut self, substs: &Substitution, depth: DebruijnInd
|
||||
self
|
||||
}
|
||||
|
||||
fn shifted_in(self, _interner: &Interner) -> Self
|
||||
where
|
||||
Self: Sized,
|
||||
{
|
||||
self.shifted_in_from(DebruijnIndex::ONE)
|
||||
}
|
||||
|
||||
/// Shifts up debruijn indices of `TyKind::Bound` vars by `n`.
|
||||
fn shifted_in_from(self, n: DebruijnIndex) -> Self
|
||||
where
|
||||
@ -149,6 +156,9 @@ fn walk(&self, f: &mut impl FnMut(&Ty)) {
|
||||
TyKind::Slice(ty) | TyKind::Array(ty) | TyKind::Ref(_, ty) | TyKind::Raw(_, ty) => {
|
||||
ty.walk(f);
|
||||
}
|
||||
TyKind::Function(fn_pointer) => {
|
||||
fn_pointer.substitution.0.walk(f);
|
||||
}
|
||||
_ => {
|
||||
if let Some(substs) = self.substs() {
|
||||
for t in substs.iter(&Interner) {
|
||||
@ -180,6 +190,9 @@ fn walk_mut_binders(
|
||||
TyKind::Slice(ty) | TyKind::Array(ty) | TyKind::Ref(_, ty) | TyKind::Raw(_, ty) => {
|
||||
ty.walk_mut_binders(f, binders);
|
||||
}
|
||||
TyKind::Function(fn_pointer) => {
|
||||
fn_pointer.substitution.0.walk_mut_binders(f, binders.shifted_in());
|
||||
}
|
||||
_ => {
|
||||
if let Some(substs) = self.substs_mut() {
|
||||
substs.walk_mut_binders(f, binders);
|
||||
|
Loading…
Reference in New Issue
Block a user