]> git.proxmox.com Git - rustc.git/blame - compiler/rustc_traits/src/chalk/lowering.rs
New upstream version 1.61.0+dfsg1
[rustc.git] / compiler / rustc_traits / src / chalk / lowering.rs
CommitLineData
f9f354fc
XL
1//! Contains the logic to lower rustc types into Chalk types
2//!
3//! In many cases there is a 1:1 relationship between a rustc type and a Chalk type.
4//! For example, a `SubstsRef` maps almost directly to a `Substitution`. In some
5//! other cases, such as `Param`s, there is no Chalk type, so we have to handle
6//! accordingly.
7//!
8//! ## `Ty` lowering
9//! Much of the `Ty` lowering is 1:1 with Chalk. (Or will be eventually). A
10//! helpful table for what types lower to what can be found in the
136023e0 11//! [Chalk book](https://rust-lang.github.io/chalk/book/types/rust_types.html).
f9f354fc
XL
12//! The most notable difference lies with `Param`s. To convert from rustc to
13//! Chalk, we eagerly and deeply convert `Param`s to placeholders (in goals) or
14//! bound variables (for clause generation through functions in `db`).
15//!
16//! ## `Region` lowering
17//! Regions are handled in rustc and Chalk is quite differently. In rustc, there
18//! is a difference between "early bound" and "late bound" regions, where only
19//! the late bound regions have a `DebruijnIndex`. Moreover, in Chalk all
20//! regions (Lifetimes) have an associated index. In rustc, only `BrAnon`s have
21//! an index, whereas `BrNamed` don't. In order to lower regions to Chalk, we
22//! convert all regions into `BrAnon` late-bound regions.
23//!
24//! ## `Const` lowering
25//! Chalk doesn't handle consts currently, so consts are currently lowered to
26//! an empty tuple.
27//!
28//! ## Bound variable collection
29//! Another difference between rustc and Chalk lies in the handling of binders.
30//! Chalk requires that we store the bound parameter kinds, whereas rustc does
31//! not. To lower anything wrapped in a `Binder`, we first deeply find any bound
32//! variables from the current `Binder`.
33
29967ef6 34use rustc_ast::ast;
1b1a35ee 35use rustc_middle::traits::{ChalkEnvironmentAndGoal, ChalkRustInterner as RustInterner};
f9f354fc 36use rustc_middle::ty::fold::TypeFolder;
f035d41b 37use rustc_middle::ty::subst::{GenericArg, GenericArgKind, SubstsRef};
5099ac24 38use rustc_middle::ty::{self, Binder, Region, Ty, TyCtxt, TypeFoldable, TypeVisitor};
f9f354fc
XL
39use rustc_span::def_id::DefId;
40
1b1a35ee
XL
41use chalk_ir::{FnSig, ForeignDefId};
42use rustc_hir::Unsafety;
f9f354fc 43use std::collections::btree_map::{BTreeMap, Entry};
29967ef6 44use std::ops::ControlFlow;
f9f354fc
XL
45
46/// Essentially an `Into` with a `&RustInterner` parameter
47crate trait LowerInto<'tcx, T> {
48 /// Lower a rustc construct (e.g., `ty::TraitPredicate`) to a chalk type, consuming `self`.
a2a8927a 49 fn lower_into(self, interner: RustInterner<'tcx>) -> T;
f9f354fc
XL
50}
51
52impl<'tcx> LowerInto<'tcx, chalk_ir::Substitution<RustInterner<'tcx>>> for SubstsRef<'tcx> {
53 fn lower_into(
54 self,
a2a8927a 55 interner: RustInterner<'tcx>,
f9f354fc 56 ) -> chalk_ir::Substitution<RustInterner<'tcx>> {
1b1a35ee
XL
57 chalk_ir::Substitution::from_iter(interner, self.iter().map(|s| s.lower_into(interner)))
58 }
59}
60
61impl<'tcx> LowerInto<'tcx, SubstsRef<'tcx>> for &chalk_ir::Substitution<RustInterner<'tcx>> {
a2a8927a 62 fn lower_into(self, interner: RustInterner<'tcx>) -> SubstsRef<'tcx> {
1b1a35ee 63 interner.tcx.mk_substs(self.iter(interner).map(|subst| subst.lower_into(interner)))
f9f354fc
XL
64 }
65}
66
67impl<'tcx> LowerInto<'tcx, chalk_ir::AliasTy<RustInterner<'tcx>>> for ty::ProjectionTy<'tcx> {
a2a8927a 68 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::AliasTy<RustInterner<'tcx>> {
f9f354fc 69 chalk_ir::AliasTy::Projection(chalk_ir::ProjectionTy {
f035d41b 70 associated_ty_id: chalk_ir::AssocTypeId(self.item_def_id),
f9f354fc
XL
71 substitution: self.substs.lower_into(interner),
72 })
73 }
74}
75
76impl<'tcx> LowerInto<'tcx, chalk_ir::InEnvironment<chalk_ir::Goal<RustInterner<'tcx>>>>
77 for ChalkEnvironmentAndGoal<'tcx>
78{
79 fn lower_into(
80 self,
a2a8927a 81 interner: RustInterner<'tcx>,
f9f354fc 82 ) -> chalk_ir::InEnvironment<chalk_ir::Goal<RustInterner<'tcx>>> {
1b1a35ee 83 let clauses = self.environment.into_iter().map(|predicate| {
5869c6ff
XL
84 let (predicate, binders, _named_regions) =
85 collect_bound_vars(interner, interner.tcx, predicate.kind());
1b1a35ee 86 let consequence = match predicate {
5869c6ff 87 ty::PredicateKind::TypeWellFormedFromEnv(ty) => {
1b1a35ee 88 chalk_ir::DomainGoal::FromEnv(chalk_ir::FromEnv::Ty(ty.lower_into(interner)))
f9f354fc 89 }
94222f64 90 ty::PredicateKind::Trait(predicate) => chalk_ir::DomainGoal::FromEnv(
1b1a35ee
XL
91 chalk_ir::FromEnv::Trait(predicate.trait_ref.lower_into(interner)),
92 ),
5869c6ff 93 ty::PredicateKind::RegionOutlives(predicate) => chalk_ir::DomainGoal::Holds(
1b1a35ee
XL
94 chalk_ir::WhereClause::LifetimeOutlives(chalk_ir::LifetimeOutlives {
95 a: predicate.0.lower_into(interner),
96 b: predicate.1.lower_into(interner),
97 }),
98 ),
5869c6ff 99 ty::PredicateKind::TypeOutlives(predicate) => chalk_ir::DomainGoal::Holds(
1b1a35ee
XL
100 chalk_ir::WhereClause::TypeOutlives(chalk_ir::TypeOutlives {
101 ty: predicate.0.lower_into(interner),
102 lifetime: predicate.1.lower_into(interner),
103 }),
104 ),
5869c6ff 105 ty::PredicateKind::Projection(predicate) => chalk_ir::DomainGoal::Holds(
1b1a35ee
XL
106 chalk_ir::WhereClause::AliasEq(predicate.lower_into(interner)),
107 ),
5869c6ff
XL
108 ty::PredicateKind::WellFormed(..)
109 | ty::PredicateKind::ObjectSafe(..)
110 | ty::PredicateKind::ClosureKind(..)
111 | ty::PredicateKind::Subtype(..)
94222f64 112 | ty::PredicateKind::Coerce(..)
5869c6ff
XL
113 | ty::PredicateKind::ConstEvaluatable(..)
114 | ty::PredicateKind::ConstEquate(..) => bug!("unexpected predicate {}", predicate),
1b1a35ee
XL
115 };
116 let value = chalk_ir::ProgramClauseImplication {
117 consequence,
118 conditions: chalk_ir::Goals::empty(interner),
119 priority: chalk_ir::ClausePriority::High,
120 constraints: chalk_ir::Constraints::empty(interner),
121 };
122 chalk_ir::ProgramClauseData(chalk_ir::Binders::new(binders, value)).intern(interner)
f9f354fc
XL
123 });
124
a2a8927a 125 let goal: chalk_ir::GoalData<RustInterner<'tcx>> = self.goal.lower_into(interner);
f9f354fc
XL
126 chalk_ir::InEnvironment {
127 environment: chalk_ir::Environment {
a2a8927a 128 clauses: chalk_ir::ProgramClauses::from_iter(interner, clauses),
f9f354fc 129 },
a2a8927a 130 goal: goal.intern(interner),
f9f354fc
XL
131 }
132 }
133}
134
135impl<'tcx> LowerInto<'tcx, chalk_ir::GoalData<RustInterner<'tcx>>> for ty::Predicate<'tcx> {
a2a8927a 136 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::GoalData<RustInterner<'tcx>> {
5869c6ff
XL
137 let (predicate, binders, _named_regions) =
138 collect_bound_vars(interner, interner.tcx, self.kind());
1b1a35ee
XL
139
140 let value = match predicate {
94222f64 141 ty::PredicateKind::Trait(predicate) => {
1b1a35ee
XL
142 chalk_ir::GoalData::DomainGoal(chalk_ir::DomainGoal::Holds(
143 chalk_ir::WhereClause::Implemented(predicate.trait_ref.lower_into(interner)),
144 ))
3dfed10e 145 }
5869c6ff 146 ty::PredicateKind::RegionOutlives(predicate) => {
1b1a35ee
XL
147 chalk_ir::GoalData::DomainGoal(chalk_ir::DomainGoal::Holds(
148 chalk_ir::WhereClause::LifetimeOutlives(chalk_ir::LifetimeOutlives {
149 a: predicate.0.lower_into(interner),
150 b: predicate.1.lower_into(interner),
151 }),
152 ))
f9f354fc 153 }
5869c6ff 154 ty::PredicateKind::TypeOutlives(predicate) => {
1b1a35ee
XL
155 chalk_ir::GoalData::DomainGoal(chalk_ir::DomainGoal::Holds(
156 chalk_ir::WhereClause::TypeOutlives(chalk_ir::TypeOutlives {
157 ty: predicate.0.lower_into(interner),
158 lifetime: predicate.1.lower_into(interner),
159 }),
160 ))
f9f354fc 161 }
5869c6ff 162 ty::PredicateKind::Projection(predicate) => {
1b1a35ee
XL
163 chalk_ir::GoalData::DomainGoal(chalk_ir::DomainGoal::Holds(
164 chalk_ir::WhereClause::AliasEq(predicate.lower_into(interner)),
165 ))
3dfed10e 166 }
5869c6ff 167 ty::PredicateKind::WellFormed(arg) => match arg.unpack() {
1b1a35ee 168 GenericArgKind::Type(ty) => match ty.kind() {
f035d41b
XL
169 // FIXME(chalk): In Chalk, a placeholder is WellFormed if it
170 // `FromEnv`. However, when we "lower" Params, we don't update
171 // the environment.
1b1a35ee
XL
172 ty::Placeholder(..) => {
173 chalk_ir::GoalData::All(chalk_ir::Goals::empty(interner))
3dfed10e 174 }
1b1a35ee
XL
175
176 _ => chalk_ir::GoalData::DomainGoal(chalk_ir::DomainGoal::WellFormed(
177 chalk_ir::WellFormed::Ty(ty.lower_into(interner)),
178 )),
f035d41b
XL
179 },
180 // FIXME(chalk): handle well formed consts
181 GenericArgKind::Const(..) => {
1b1a35ee 182 chalk_ir::GoalData::All(chalk_ir::Goals::empty(interner))
f9f354fc 183 }
f035d41b 184 GenericArgKind::Lifetime(lt) => bug!("unexpect well formed predicate: {:?}", lt),
f9f354fc
XL
185 },
186
5869c6ff 187 ty::PredicateKind::ObjectSafe(t) => chalk_ir::GoalData::DomainGoal(
3dfed10e 188 chalk_ir::DomainGoal::ObjectSafe(chalk_ir::TraitId(t)),
f035d41b
XL
189 ),
190
5099ac24
FG
191 ty::PredicateKind::Subtype(ty::SubtypePredicate { a, b, a_is_expected: _ }) => {
192 chalk_ir::GoalData::SubtypeGoal(chalk_ir::SubtypeGoal {
193 a: a.lower_into(interner),
194 b: b.lower_into(interner),
195 })
196 }
197
f9f354fc
XL
198 // FIXME(chalk): other predicates
199 //
200 // We can defer this, but ultimately we'll want to express
201 // some of these in terms of chalk operations.
5869c6ff 202 ty::PredicateKind::ClosureKind(..)
94222f64 203 | ty::PredicateKind::Coerce(..)
5869c6ff
XL
204 | ty::PredicateKind::ConstEvaluatable(..)
205 | ty::PredicateKind::ConstEquate(..) => {
1b1a35ee 206 chalk_ir::GoalData::All(chalk_ir::Goals::empty(interner))
f9f354fc 207 }
5869c6ff 208 ty::PredicateKind::TypeWellFormedFromEnv(ty) => chalk_ir::GoalData::DomainGoal(
1b1a35ee
XL
209 chalk_ir::DomainGoal::FromEnv(chalk_ir::FromEnv::Ty(ty.lower_into(interner))),
210 ),
211 };
212
213 chalk_ir::GoalData::Quantified(
214 chalk_ir::QuantifierKind::ForAll,
215 chalk_ir::Binders::new(binders, value.intern(interner)),
216 )
f9f354fc
XL
217 }
218}
219
220impl<'tcx> LowerInto<'tcx, chalk_ir::TraitRef<RustInterner<'tcx>>>
221 for rustc_middle::ty::TraitRef<'tcx>
222{
a2a8927a 223 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::TraitRef<RustInterner<'tcx>> {
f9f354fc 224 chalk_ir::TraitRef {
f035d41b 225 trait_id: chalk_ir::TraitId(self.def_id),
f9f354fc
XL
226 substitution: self.substs.lower_into(interner),
227 }
228 }
229}
230
f9f354fc
XL
231impl<'tcx> LowerInto<'tcx, chalk_ir::AliasEq<RustInterner<'tcx>>>
232 for rustc_middle::ty::ProjectionPredicate<'tcx>
233{
a2a8927a 234 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::AliasEq<RustInterner<'tcx>> {
5099ac24 235 // FIXME(associated_const_equality): teach chalk about terms for alias eq.
f9f354fc 236 chalk_ir::AliasEq {
5099ac24 237 ty: self.term.ty().unwrap().lower_into(interner),
f9f354fc
XL
238 alias: self.projection_ty.lower_into(interner),
239 }
240 }
241}
242
5099ac24
FG
243/*
244// FIXME(...): Where do I add this to Chalk? I can't find it in the rustc repo anywhere.
245impl<'tcx> LowerInto<'tcx, chalk_ir::Term<RustInterner<'tcx>>> for rustc_middle::ty::Term<'tcx> {
246 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::Term<RustInterner<'tcx>> {
247 match self {
248 ty::Term::Ty(ty) => ty.lower_into(interner).into(),
249 ty::Term::Const(c) => c.lower_into(interner).into(),
250 }
251 }
252}
253*/
254
f9f354fc 255impl<'tcx> LowerInto<'tcx, chalk_ir::Ty<RustInterner<'tcx>>> for Ty<'tcx> {
a2a8927a 256 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::Ty<RustInterner<'tcx>> {
29967ef6
XL
257 let int = |i| chalk_ir::TyKind::Scalar(chalk_ir::Scalar::Int(i));
258 let uint = |i| chalk_ir::TyKind::Scalar(chalk_ir::Scalar::Uint(i));
259 let float = |f| chalk_ir::TyKind::Scalar(chalk_ir::Scalar::Float(f));
f9f354fc 260
1b1a35ee 261 match *self.kind() {
29967ef6
XL
262 ty::Bool => chalk_ir::TyKind::Scalar(chalk_ir::Scalar::Bool),
263 ty::Char => chalk_ir::TyKind::Scalar(chalk_ir::Scalar::Char),
264 ty::Int(ty) => match ty {
5869c6ff
XL
265 ty::IntTy::Isize => int(chalk_ir::IntTy::Isize),
266 ty::IntTy::I8 => int(chalk_ir::IntTy::I8),
267 ty::IntTy::I16 => int(chalk_ir::IntTy::I16),
268 ty::IntTy::I32 => int(chalk_ir::IntTy::I32),
269 ty::IntTy::I64 => int(chalk_ir::IntTy::I64),
270 ty::IntTy::I128 => int(chalk_ir::IntTy::I128),
f9f354fc 271 },
29967ef6 272 ty::Uint(ty) => match ty {
5869c6ff
XL
273 ty::UintTy::Usize => uint(chalk_ir::UintTy::Usize),
274 ty::UintTy::U8 => uint(chalk_ir::UintTy::U8),
275 ty::UintTy::U16 => uint(chalk_ir::UintTy::U16),
276 ty::UintTy::U32 => uint(chalk_ir::UintTy::U32),
277 ty::UintTy::U64 => uint(chalk_ir::UintTy::U64),
278 ty::UintTy::U128 => uint(chalk_ir::UintTy::U128),
f9f354fc 279 },
29967ef6 280 ty::Float(ty) => match ty {
5869c6ff
XL
281 ty::FloatTy::F32 => float(chalk_ir::FloatTy::F32),
282 ty::FloatTy::F64 => float(chalk_ir::FloatTy::F64),
f9f354fc 283 },
29967ef6
XL
284 ty::Adt(def, substs) => {
285 chalk_ir::TyKind::Adt(chalk_ir::AdtId(def), substs.lower_into(interner))
f035d41b 286 }
29967ef6
XL
287 ty::Foreign(def_id) => chalk_ir::TyKind::Foreign(ForeignDefId(def_id)),
288 ty::Str => chalk_ir::TyKind::Str,
289 ty::Array(ty, len) => {
290 chalk_ir::TyKind::Array(ty.lower_into(interner), len.lower_into(interner))
f035d41b 291 }
29967ef6
XL
292 ty::Slice(ty) => chalk_ir::TyKind::Slice(ty.lower_into(interner)),
293
294 ty::RawPtr(ptr) => {
295 chalk_ir::TyKind::Raw(ptr.mutbl.lower_into(interner), ptr.ty.lower_into(interner))
f035d41b 296 }
29967ef6
XL
297 ty::Ref(region, ty, mutability) => chalk_ir::TyKind::Ref(
298 mutability.lower_into(interner),
299 region.lower_into(interner),
300 ty.lower_into(interner),
f9f354fc 301 ),
29967ef6
XL
302 ty::FnDef(def_id, substs) => {
303 chalk_ir::TyKind::FnDef(chalk_ir::FnDefId(def_id), substs.lower_into(interner))
304 }
305 ty::FnPtr(sig) => {
f9f354fc 306 let (inputs_and_outputs, binders, _named_regions) =
fc512014 307 collect_bound_vars(interner, interner.tcx, sig.inputs_and_output());
29967ef6 308 chalk_ir::TyKind::Function(chalk_ir::FnPointer {
f9f354fc 309 num_binders: binders.len(interner),
1b1a35ee 310 sig: sig.lower_into(interner),
5869c6ff 311 substitution: chalk_ir::FnSubst(chalk_ir::Substitution::from_iter(
f9f354fc
XL
312 interner,
313 inputs_and_outputs.iter().map(|ty| {
f035d41b 314 chalk_ir::GenericArgData::Ty(ty.lower_into(interner)).intern(interner)
f9f354fc 315 }),
5869c6ff 316 )),
f9f354fc 317 })
f9f354fc 318 }
29967ef6 319 ty::Dynamic(predicates, region) => chalk_ir::TyKind::Dyn(chalk_ir::DynTy {
f035d41b
XL
320 bounds: predicates.lower_into(interner),
321 lifetime: region.lower_into(interner),
29967ef6
XL
322 }),
323 ty::Closure(def_id, substs) => {
324 chalk_ir::TyKind::Closure(chalk_ir::ClosureId(def_id), substs.lower_into(interner))
f9f354fc 325 }
5e7ed085
FG
326 ty::Generator(def_id, substs, _) => chalk_ir::TyKind::Generator(
327 chalk_ir::GeneratorId(def_id),
328 substs.lower_into(interner),
329 ),
29967ef6
XL
330 ty::GeneratorWitness(_) => unimplemented!(),
331 ty::Never => chalk_ir::TyKind::Never,
5e7ed085
FG
332 ty::Tuple(types) => {
333 chalk_ir::TyKind::Tuple(types.len(), types.as_substs().lower_into(interner))
334 }
29967ef6
XL
335 ty::Projection(proj) => chalk_ir::TyKind::Alias(proj.lower_into(interner)),
336 ty::Opaque(def_id, substs) => {
337 chalk_ir::TyKind::Alias(chalk_ir::AliasTy::Opaque(chalk_ir::OpaqueTy {
f035d41b
XL
338 opaque_ty_id: chalk_ir::OpaqueTyId(def_id),
339 substitution: substs.lower_into(interner),
340 }))
f035d41b 341 }
f9f354fc
XL
342 // This should have been done eagerly prior to this, and all Params
343 // should have been substituted to placeholders
29967ef6
XL
344 ty::Param(_) => panic!("Lowering Param when not expected."),
345 ty::Bound(db, bound) => chalk_ir::TyKind::BoundVar(chalk_ir::BoundVar::new(
f9f354fc
XL
346 chalk_ir::DebruijnIndex::new(db.as_u32()),
347 bound.var.index(),
29967ef6
XL
348 )),
349 ty::Placeholder(_placeholder) => {
350 chalk_ir::TyKind::Placeholder(chalk_ir::PlaceholderIndex {
351 ui: chalk_ir::UniverseIndex { counter: _placeholder.universe.as_usize() },
352 idx: _placeholder.name.as_usize(),
353 })
354 }
355 ty::Infer(_infer) => unimplemented!(),
356 ty::Error(_) => chalk_ir::TyKind::Error,
f9f354fc 357 }
29967ef6 358 .intern(interner)
f9f354fc
XL
359 }
360}
361
1b1a35ee 362impl<'tcx> LowerInto<'tcx, Ty<'tcx>> for &chalk_ir::Ty<RustInterner<'tcx>> {
a2a8927a 363 fn lower_into(self, interner: RustInterner<'tcx>) -> Ty<'tcx> {
29967ef6 364 use chalk_ir::TyKind;
1b1a35ee 365
29967ef6
XL
366 let kind = match self.kind(interner) {
367 TyKind::Adt(struct_id, substitution) => {
368 ty::Adt(struct_id.0, substitution.lower_into(interner))
369 }
370 TyKind::Scalar(scalar) => match scalar {
371 chalk_ir::Scalar::Bool => ty::Bool,
372 chalk_ir::Scalar::Char => ty::Char,
373 chalk_ir::Scalar::Int(int_ty) => match int_ty {
5869c6ff
XL
374 chalk_ir::IntTy::Isize => ty::Int(ty::IntTy::Isize),
375 chalk_ir::IntTy::I8 => ty::Int(ty::IntTy::I8),
376 chalk_ir::IntTy::I16 => ty::Int(ty::IntTy::I16),
377 chalk_ir::IntTy::I32 => ty::Int(ty::IntTy::I32),
378 chalk_ir::IntTy::I64 => ty::Int(ty::IntTy::I64),
379 chalk_ir::IntTy::I128 => ty::Int(ty::IntTy::I128),
29967ef6
XL
380 },
381 chalk_ir::Scalar::Uint(int_ty) => match int_ty {
5869c6ff
XL
382 chalk_ir::UintTy::Usize => ty::Uint(ty::UintTy::Usize),
383 chalk_ir::UintTy::U8 => ty::Uint(ty::UintTy::U8),
384 chalk_ir::UintTy::U16 => ty::Uint(ty::UintTy::U16),
385 chalk_ir::UintTy::U32 => ty::Uint(ty::UintTy::U32),
386 chalk_ir::UintTy::U64 => ty::Uint(ty::UintTy::U64),
387 chalk_ir::UintTy::U128 => ty::Uint(ty::UintTy::U128),
29967ef6
XL
388 },
389 chalk_ir::Scalar::Float(float_ty) => match float_ty {
5869c6ff
XL
390 chalk_ir::FloatTy::F32 => ty::Float(ty::FloatTy::F32),
391 chalk_ir::FloatTy::F64 => ty::Float(ty::FloatTy::F64),
1b1a35ee 392 },
1b1a35ee 393 },
29967ef6
XL
394 TyKind::Array(ty, c) => {
395 let ty = ty.lower_into(interner);
396 let c = c.lower_into(interner);
5099ac24 397 ty::Array(ty, c)
29967ef6
XL
398 }
399 TyKind::FnDef(id, substitution) => ty::FnDef(id.0, substitution.lower_into(interner)),
400 TyKind::Closure(closure, substitution) => {
401 ty::Closure(closure.0, substitution.lower_into(interner))
402 }
403 TyKind::Generator(..) => unimplemented!(),
404 TyKind::GeneratorWitness(..) => unimplemented!(),
405 TyKind::Never => ty::Never,
5e7ed085
FG
406 TyKind::Tuple(_len, substitution) => {
407 ty::Tuple(substitution.lower_into(interner).try_as_type_list().unwrap())
408 }
29967ef6
XL
409 TyKind::Slice(ty) => ty::Slice(ty.lower_into(interner)),
410 TyKind::Raw(mutbl, ty) => ty::RawPtr(ty::TypeAndMut {
411 ty: ty.lower_into(interner),
412 mutbl: mutbl.lower_into(interner),
413 }),
414 TyKind::Ref(mutbl, lifetime, ty) => ty::Ref(
415 lifetime.lower_into(interner),
416 ty.lower_into(interner),
417 mutbl.lower_into(interner),
418 ),
419 TyKind::Str => ty::Str,
420 TyKind::OpaqueType(opaque_ty, substitution) => {
421 ty::Opaque(opaque_ty.0, substitution.lower_into(interner))
422 }
423 TyKind::AssociatedType(assoc_ty, substitution) => ty::Projection(ty::ProjectionTy {
424 substs: substitution.lower_into(interner),
425 item_def_id: assoc_ty.0,
426 }),
427 TyKind::Foreign(def_id) => ty::Foreign(def_id.0),
428 TyKind::Error => return interner.tcx.ty_error(),
429 TyKind::Placeholder(placeholder) => ty::Placeholder(ty::Placeholder {
1b1a35ee
XL
430 universe: ty::UniverseIndex::from_usize(placeholder.ui.counter),
431 name: ty::BoundVar::from_usize(placeholder.idx),
432 }),
29967ef6
XL
433 TyKind::Alias(alias_ty) => match alias_ty {
434 chalk_ir::AliasTy::Projection(projection) => ty::Projection(ty::ProjectionTy {
435 item_def_id: projection.associated_ty_id.0,
436 substs: projection.substitution.lower_into(interner),
437 }),
438 chalk_ir::AliasTy::Opaque(opaque) => {
439 ty::Opaque(opaque.opaque_ty_id.0, opaque.substitution.lower_into(interner))
440 }
441 },
442 TyKind::Function(_quantified_ty) => unimplemented!(),
443 TyKind::BoundVar(_bound) => ty::Bound(
1b1a35ee
XL
444 ty::DebruijnIndex::from_usize(_bound.debruijn.depth() as usize),
445 ty::BoundTy {
446 var: ty::BoundVar::from_usize(_bound.index),
447 kind: ty::BoundTyKind::Anon,
448 },
449 ),
29967ef6
XL
450 TyKind::InferenceVar(_, _) => unimplemented!(),
451 TyKind::Dyn(_) => unimplemented!(),
1b1a35ee
XL
452 };
453 interner.tcx.mk_ty(kind)
454 }
455}
456
f9f354fc 457impl<'tcx> LowerInto<'tcx, chalk_ir::Lifetime<RustInterner<'tcx>>> for Region<'tcx> {
a2a8927a 458 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::Lifetime<RustInterner<'tcx>> {
5099ac24
FG
459 match *self {
460 ty::ReEarlyBound(_) => {
f9f354fc
XL
461 panic!("Should have already been substituted.");
462 }
5099ac24 463 ty::ReLateBound(db, br) => chalk_ir::LifetimeData::BoundVar(chalk_ir::BoundVar::new(
cdc7bbd5
XL
464 chalk_ir::DebruijnIndex::new(db.as_u32()),
465 br.var.as_usize(),
466 ))
467 .intern(interner),
5099ac24
FG
468 ty::ReFree(_) => unimplemented!(),
469 ty::ReStatic => chalk_ir::LifetimeData::Static.intern(interner),
470 ty::ReVar(_) => unimplemented!(),
471 ty::RePlaceholder(placeholder_region) => {
f9f354fc
XL
472 chalk_ir::LifetimeData::Placeholder(chalk_ir::PlaceholderIndex {
473 ui: chalk_ir::UniverseIndex { counter: placeholder_region.universe.index() },
474 idx: 0,
475 })
476 .intern(interner)
477 }
5099ac24
FG
478 ty::ReEmpty(ui) => {
479 chalk_ir::LifetimeData::Empty(chalk_ir::UniverseIndex { counter: ui.index() })
480 .intern(interner)
481 }
482 ty::ReErased => chalk_ir::LifetimeData::Erased.intern(interner),
f9f354fc
XL
483 }
484 }
485}
486
1b1a35ee 487impl<'tcx> LowerInto<'tcx, Region<'tcx>> for &chalk_ir::Lifetime<RustInterner<'tcx>> {
a2a8927a 488 fn lower_into(self, interner: RustInterner<'tcx>) -> Region<'tcx> {
1b1a35ee 489 let kind = match self.data(interner) {
5099ac24 490 chalk_ir::LifetimeData::BoundVar(var) => ty::ReLateBound(
1b1a35ee 491 ty::DebruijnIndex::from_u32(var.debruijn.depth()),
cdc7bbd5
XL
492 ty::BoundRegion {
493 var: ty::BoundVar::from_usize(var.index),
494 kind: ty::BrAnon(var.index as u32),
495 },
1b1a35ee
XL
496 ),
497 chalk_ir::LifetimeData::InferenceVar(_var) => unimplemented!(),
5099ac24
FG
498 chalk_ir::LifetimeData::Placeholder(p) => ty::RePlaceholder(ty::Placeholder {
499 universe: ty::UniverseIndex::from_usize(p.ui.counter),
500 name: ty::BoundRegionKind::BrAnon(p.idx as u32),
501 }),
502 chalk_ir::LifetimeData::Static => return interner.tcx.lifetimes.re_static,
5869c6ff 503 chalk_ir::LifetimeData::Empty(ui) => {
5099ac24 504 ty::ReEmpty(ty::UniverseIndex::from_usize(ui.counter))
5869c6ff 505 }
5099ac24
FG
506 chalk_ir::LifetimeData::Erased => return interner.tcx.lifetimes.re_erased,
507 chalk_ir::LifetimeData::Phantom(void, _) => match *void {},
1b1a35ee
XL
508 };
509 interner.tcx.mk_region(kind)
510 }
511}
512
513impl<'tcx> LowerInto<'tcx, chalk_ir::Const<RustInterner<'tcx>>> for ty::Const<'tcx> {
a2a8927a 514 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::Const<RustInterner<'tcx>> {
5099ac24
FG
515 let ty = self.ty().lower_into(interner);
516 let value = match self.val() {
1b1a35ee
XL
517 ty::ConstKind::Value(val) => {
518 chalk_ir::ConstValue::Concrete(chalk_ir::ConcreteConst { interned: val })
519 }
520 ty::ConstKind::Bound(db, bound) => chalk_ir::ConstValue::BoundVar(
521 chalk_ir::BoundVar::new(chalk_ir::DebruijnIndex::new(db.as_u32()), bound.index()),
522 ),
523 _ => unimplemented!("Const not implemented. {:?}", self),
524 };
525 chalk_ir::ConstData { ty, value }.intern(interner)
526 }
527}
528
529impl<'tcx> LowerInto<'tcx, ty::Const<'tcx>> for &chalk_ir::Const<RustInterner<'tcx>> {
a2a8927a 530 fn lower_into(self, interner: RustInterner<'tcx>) -> ty::Const<'tcx> {
1b1a35ee
XL
531 let data = self.data(interner);
532 let ty = data.ty.lower_into(interner);
533 let val = match data.value {
534 chalk_ir::ConstValue::BoundVar(var) => ty::ConstKind::Bound(
535 ty::DebruijnIndex::from_u32(var.debruijn.depth()),
536 ty::BoundVar::from_u32(var.index as u32),
537 ),
538 chalk_ir::ConstValue::InferenceVar(_var) => unimplemented!(),
539 chalk_ir::ConstValue::Placeholder(_p) => unimplemented!(),
540 chalk_ir::ConstValue::Concrete(c) => ty::ConstKind::Value(c.interned),
541 };
5099ac24 542 interner.tcx.mk_const(ty::ConstS { ty, val })
1b1a35ee
XL
543 }
544}
545
f035d41b 546impl<'tcx> LowerInto<'tcx, chalk_ir::GenericArg<RustInterner<'tcx>>> for GenericArg<'tcx> {
a2a8927a 547 fn lower_into(self, interner: RustInterner<'tcx>) -> chalk_ir::GenericArg<RustInterner<'tcx>> {
f9f354fc
XL
548 match self.unpack() {
549 ty::subst::GenericArgKind::Type(ty) => {
f035d41b 550 chalk_ir::GenericArgData::Ty(ty.lower_into(interner))
f9f354fc
XL
551 }
552 ty::subst::GenericArgKind::Lifetime(lifetime) => {
f035d41b 553 chalk_ir::GenericArgData::Lifetime(lifetime.lower_into(interner))
f9f354fc 554 }
1b1a35ee
XL
555 ty::subst::GenericArgKind::Const(c) => {
556 chalk_ir::GenericArgData::Const(c.lower_into(interner))
557 }
f9f354fc
XL
558 }
559 .intern(interner)
560 }
561}
562
1b1a35ee
XL
563impl<'tcx> LowerInto<'tcx, ty::subst::GenericArg<'tcx>>
564 for &chalk_ir::GenericArg<RustInterner<'tcx>>
565{
a2a8927a 566 fn lower_into(self, interner: RustInterner<'tcx>) -> ty::subst::GenericArg<'tcx> {
1b1a35ee
XL
567 match self.data(interner) {
568 chalk_ir::GenericArgData::Ty(ty) => {
569 let t: Ty<'tcx> = ty.lower_into(interner);
570 t.into()
571 }
572 chalk_ir::GenericArgData::Lifetime(lifetime) => {
573 let r: Region<'tcx> = lifetime.lower_into(interner);
574 r.into()
575 }
576 chalk_ir::GenericArgData::Const(c) => {
577 let c: ty::Const<'tcx> = c.lower_into(interner);
5099ac24 578 c.into()
1b1a35ee
XL
579 }
580 }
581 }
582}
583
f9f354fc
XL
584// We lower into an Option here since there are some predicates which Chalk
585// doesn't have a representation for yet (as a `WhereClause`), but are so common
586// that we just are accepting the unsoundness for now. The `Option` will
587// eventually be removed.
588impl<'tcx> LowerInto<'tcx, Option<chalk_ir::QuantifiedWhereClause<RustInterner<'tcx>>>>
589 for ty::Predicate<'tcx>
590{
591 fn lower_into(
592 self,
a2a8927a 593 interner: RustInterner<'tcx>,
f9f354fc 594 ) -> Option<chalk_ir::QuantifiedWhereClause<RustInterner<'tcx>>> {
5869c6ff
XL
595 let (predicate, binders, _named_regions) =
596 collect_bound_vars(interner, interner.tcx, self.kind());
1b1a35ee 597 let value = match predicate {
94222f64 598 ty::PredicateKind::Trait(predicate) => {
1b1a35ee 599 Some(chalk_ir::WhereClause::Implemented(predicate.trait_ref.lower_into(interner)))
f9f354fc 600 }
5869c6ff 601 ty::PredicateKind::RegionOutlives(predicate) => {
1b1a35ee
XL
602 Some(chalk_ir::WhereClause::LifetimeOutlives(chalk_ir::LifetimeOutlives {
603 a: predicate.0.lower_into(interner),
604 b: predicate.1.lower_into(interner),
605 }))
606 }
5869c6ff 607 ty::PredicateKind::TypeOutlives(predicate) => {
1b1a35ee
XL
608 Some(chalk_ir::WhereClause::TypeOutlives(chalk_ir::TypeOutlives {
609 ty: predicate.0.lower_into(interner),
610 lifetime: predicate.1.lower_into(interner),
611 }))
612 }
5869c6ff 613 ty::PredicateKind::Projection(predicate) => {
1b1a35ee 614 Some(chalk_ir::WhereClause::AliasEq(predicate.lower_into(interner)))
f035d41b 615 }
5869c6ff 616 ty::PredicateKind::WellFormed(_ty) => None,
3dfed10e 617
5869c6ff
XL
618 ty::PredicateKind::ObjectSafe(..)
619 | ty::PredicateKind::ClosureKind(..)
620 | ty::PredicateKind::Subtype(..)
94222f64 621 | ty::PredicateKind::Coerce(..)
5869c6ff
XL
622 | ty::PredicateKind::ConstEvaluatable(..)
623 | ty::PredicateKind::ConstEquate(..)
624 | ty::PredicateKind::TypeWellFormedFromEnv(..) => {
1b1a35ee
XL
625 bug!("unexpected predicate {}", &self)
626 }
627 };
628 value.map(|value| chalk_ir::Binders::new(binders, value))
f9f354fc
XL
629 }
630}
631
f035d41b 632impl<'tcx> LowerInto<'tcx, chalk_ir::Binders<chalk_ir::QuantifiedWhereClauses<RustInterner<'tcx>>>>
cdc7bbd5 633 for &'tcx ty::List<ty::Binder<'tcx, ty::ExistentialPredicate<'tcx>>>
f035d41b
XL
634{
635 fn lower_into(
636 self,
a2a8927a 637 interner: RustInterner<'tcx>,
f035d41b 638 ) -> chalk_ir::Binders<chalk_ir::QuantifiedWhereClauses<RustInterner<'tcx>>> {
29967ef6
XL
639 // `Self` has one binder:
640 // Binder<&'tcx ty::List<ty::ExistentialPredicate<'tcx>>>
641 // The return type has two:
642 // Binders<&[Binders<WhereClause<I>>]>
643 // This means that any variables that are escaping `self` need to be
644 // shifted in by one so that they are still escaping.
fc512014 645 let predicates = ty::fold::shift_vars(interner.tcx, self, 1);
29967ef6 646
1b1a35ee
XL
647 let self_ty = interner.tcx.mk_ty(ty::Bound(
648 // This is going to be wrapped in a binder
649 ty::DebruijnIndex::from_usize(1),
650 ty::BoundTy { var: ty::BoundVar::from_usize(0), kind: ty::BoundTyKind::Anon },
651 ));
fc512014
XL
652 let where_clauses = predicates.into_iter().map(|predicate| {
653 let (predicate, binders, _named_regions) =
654 collect_bound_vars(interner, interner.tcx, predicate);
655 match predicate {
656 ty::ExistentialPredicate::Trait(ty::ExistentialTraitRef { def_id, substs }) => {
657 chalk_ir::Binders::new(
658 binders.clone(),
659 chalk_ir::WhereClause::Implemented(chalk_ir::TraitRef {
660 trait_id: chalk_ir::TraitId(def_id),
661 substitution: interner
662 .tcx
663 .mk_substs_trait(self_ty, substs)
664 .lower_into(interner),
665 }),
666 )
667 }
668 ty::ExistentialPredicate::Projection(predicate) => chalk_ir::Binders::new(
669 binders.clone(),
670 chalk_ir::WhereClause::AliasEq(chalk_ir::AliasEq {
671 alias: chalk_ir::AliasTy::Projection(chalk_ir::ProjectionTy {
672 associated_ty_id: chalk_ir::AssocTypeId(predicate.item_def_id),
673 substitution: interner
674 .tcx
675 .mk_substs_trait(self_ty, predicate.substs)
676 .lower_into(interner),
677 }),
5099ac24
FG
678 // FIXME(associated_const_equality): teach chalk about terms for alias eq.
679 ty: predicate.term.ty().unwrap().lower_into(interner),
fc512014
XL
680 }),
681 ),
682 ty::ExistentialPredicate::AutoTrait(def_id) => chalk_ir::Binders::new(
29967ef6 683 binders.clone(),
f035d41b
XL
684 chalk_ir::WhereClause::Implemented(chalk_ir::TraitRef {
685 trait_id: chalk_ir::TraitId(def_id),
1b1a35ee
XL
686 substitution: interner
687 .tcx
fc512014 688 .mk_substs_trait(self_ty, &[])
1b1a35ee 689 .lower_into(interner),
f035d41b 690 }),
fc512014 691 ),
f035d41b 692 }
f035d41b 693 });
29967ef6
XL
694
695 // Binder for the bound variable representing the concrete underlying type.
696 let existential_binder = chalk_ir::VariableKinds::from1(
697 interner,
698 chalk_ir::VariableKind::Ty(chalk_ir::TyVariableKind::General),
699 );
1b1a35ee 700 let value = chalk_ir::QuantifiedWhereClauses::from_iter(interner, where_clauses);
29967ef6 701 chalk_ir::Binders::new(existential_binder, value)
f035d41b
XL
702 }
703}
704
cdc7bbd5
XL
705impl<'tcx> LowerInto<'tcx, chalk_ir::FnSig<RustInterner<'tcx>>>
706 for ty::Binder<'tcx, ty::FnSig<'tcx>>
707{
a2a8927a 708 fn lower_into(self, _interner: RustInterner<'_>) -> FnSig<RustInterner<'tcx>> {
1b1a35ee
XL
709 chalk_ir::FnSig {
710 abi: self.abi(),
711 safety: match self.unsafety() {
712 Unsafety::Normal => chalk_ir::Safety::Safe,
713 Unsafety::Unsafe => chalk_ir::Safety::Unsafe,
714 },
715 variadic: self.c_variadic(),
716 }
717 }
718}
719
29967ef6
XL
720// We lower into an Option here since there are some predicates which Chalk
721// doesn't have a representation for yet (as an `InlineBound`). The `Option` will
722// eventually be removed.
723impl<'tcx> LowerInto<'tcx, Option<chalk_solve::rust_ir::QuantifiedInlineBound<RustInterner<'tcx>>>>
724 for ty::Predicate<'tcx>
725{
726 fn lower_into(
727 self,
a2a8927a 728 interner: RustInterner<'tcx>,
29967ef6 729 ) -> Option<chalk_solve::rust_ir::QuantifiedInlineBound<RustInterner<'tcx>>> {
5869c6ff
XL
730 let (predicate, binders, _named_regions) =
731 collect_bound_vars(interner, interner.tcx, self.kind());
29967ef6 732 match predicate {
94222f64 733 ty::PredicateKind::Trait(predicate) => Some(chalk_ir::Binders::new(
29967ef6
XL
734 binders,
735 chalk_solve::rust_ir::InlineBound::TraitBound(
736 predicate.trait_ref.lower_into(interner),
737 ),
738 )),
5869c6ff 739 ty::PredicateKind::Projection(predicate) => Some(chalk_ir::Binders::new(
29967ef6
XL
740 binders,
741 chalk_solve::rust_ir::InlineBound::AliasEqBound(predicate.lower_into(interner)),
742 )),
5869c6ff
XL
743 ty::PredicateKind::TypeOutlives(_predicate) => None,
744 ty::PredicateKind::WellFormed(_ty) => None,
745
746 ty::PredicateKind::RegionOutlives(..)
747 | ty::PredicateKind::ObjectSafe(..)
748 | ty::PredicateKind::ClosureKind(..)
749 | ty::PredicateKind::Subtype(..)
94222f64 750 | ty::PredicateKind::Coerce(..)
5869c6ff
XL
751 | ty::PredicateKind::ConstEvaluatable(..)
752 | ty::PredicateKind::ConstEquate(..)
753 | ty::PredicateKind::TypeWellFormedFromEnv(..) => {
29967ef6
XL
754 bug!("unexpected predicate {}", &self)
755 }
756 }
757 }
758}
759
760impl<'tcx> LowerInto<'tcx, chalk_solve::rust_ir::TraitBound<RustInterner<'tcx>>>
761 for ty::TraitRef<'tcx>
762{
763 fn lower_into(
764 self,
a2a8927a 765 interner: RustInterner<'tcx>,
29967ef6
XL
766 ) -> chalk_solve::rust_ir::TraitBound<RustInterner<'tcx>> {
767 chalk_solve::rust_ir::TraitBound {
768 trait_id: chalk_ir::TraitId(self.def_id),
769 args_no_self: self.substs[1..].iter().map(|arg| arg.lower_into(interner)).collect(),
770 }
771 }
772}
773
774impl<'tcx> LowerInto<'tcx, chalk_ir::Mutability> for ast::Mutability {
a2a8927a 775 fn lower_into(self, _interner: RustInterner<'tcx>) -> chalk_ir::Mutability {
29967ef6
XL
776 match self {
777 rustc_ast::Mutability::Mut => chalk_ir::Mutability::Mut,
778 rustc_ast::Mutability::Not => chalk_ir::Mutability::Not,
779 }
780 }
781}
782
783impl<'tcx> LowerInto<'tcx, ast::Mutability> for chalk_ir::Mutability {
a2a8927a 784 fn lower_into(self, _interner: RustInterner<'tcx>) -> ast::Mutability {
29967ef6
XL
785 match self {
786 chalk_ir::Mutability::Mut => ast::Mutability::Mut,
787 chalk_ir::Mutability::Not => ast::Mutability::Not,
788 }
789 }
790}
791
792impl<'tcx> LowerInto<'tcx, chalk_solve::rust_ir::Polarity> for ty::ImplPolarity {
a2a8927a 793 fn lower_into(self, _interner: RustInterner<'tcx>) -> chalk_solve::rust_ir::Polarity {
29967ef6
XL
794 match self {
795 ty::ImplPolarity::Positive => chalk_solve::rust_ir::Polarity::Positive,
796 ty::ImplPolarity::Negative => chalk_solve::rust_ir::Polarity::Negative,
797 // FIXME(chalk) reservation impls
798 ty::ImplPolarity::Reservation => chalk_solve::rust_ir::Polarity::Negative,
799 }
800 }
801}
5099ac24
FG
802impl<'tcx> LowerInto<'tcx, chalk_ir::Variance> for ty::Variance {
803 fn lower_into(self, _interner: RustInterner<'tcx>) -> chalk_ir::Variance {
804 match self {
805 ty::Variance::Covariant => chalk_ir::Variance::Covariant,
806 ty::Variance::Invariant => chalk_ir::Variance::Invariant,
807 ty::Variance::Contravariant => chalk_ir::Variance::Contravariant,
808 ty::Variance::Bivariant => unimplemented!(),
809 }
810 }
811}
29967ef6
XL
812
813impl<'tcx> LowerInto<'tcx, chalk_solve::rust_ir::AliasEqBound<RustInterner<'tcx>>>
814 for ty::ProjectionPredicate<'tcx>
815{
816 fn lower_into(
817 self,
a2a8927a 818 interner: RustInterner<'tcx>,
29967ef6 819 ) -> chalk_solve::rust_ir::AliasEqBound<RustInterner<'tcx>> {
6a06907d 820 let (trait_ref, own_substs) = self.projection_ty.trait_ref_and_own_substs(interner.tcx);
29967ef6
XL
821 chalk_solve::rust_ir::AliasEqBound {
822 trait_bound: trait_ref.lower_into(interner),
823 associated_ty_id: chalk_ir::AssocTypeId(self.projection_ty.item_def_id),
6a06907d 824 parameters: own_substs.iter().map(|arg| arg.lower_into(interner)).collect(),
5099ac24 825 value: self.term.ty().unwrap().lower_into(interner),
29967ef6
XL
826 }
827 }
828}
829
f9f354fc 830/// To collect bound vars, we have to do two passes. In the first pass, we
fc512014 831/// collect all `BoundRegionKind`s and `ty::Bound`s. In the second pass, we then
f9f354fc
XL
832/// replace `BrNamed` into `BrAnon`. The two separate passes are important,
833/// since we can only replace `BrNamed` with `BrAnon`s with indices *after* all
834/// "real" `BrAnon`s.
835///
836/// It's important to note that because of prior substitution, we may have
837/// late-bound regions, even outside of fn contexts, since this is the best way
838/// to prep types for chalk lowering.
fc512014 839crate fn collect_bound_vars<'tcx, T: TypeFoldable<'tcx>>(
a2a8927a 840 interner: RustInterner<'tcx>,
f9f354fc 841 tcx: TyCtxt<'tcx>,
cdc7bbd5 842 ty: Binder<'tcx, T>,
f035d41b 843) -> (T, chalk_ir::VariableKinds<RustInterner<'tcx>>, BTreeMap<DefId, u32>) {
5099ac24 844 let mut bound_vars_collector = BoundVarsCollector::new();
f035d41b 845 ty.as_ref().skip_binder().visit_with(&mut bound_vars_collector);
f9f354fc
XL
846 let mut parameters = bound_vars_collector.parameters;
847 let named_parameters: BTreeMap<DefId, u32> = bound_vars_collector
848 .named_parameters
849 .into_iter()
850 .enumerate()
851 .map(|(i, def_id)| (def_id, (i + parameters.len()) as u32))
852 .collect();
853
854 let mut bound_var_substitutor = NamedBoundVarSubstitutor::new(tcx, &named_parameters);
fc512014 855 let new_ty = ty.skip_binder().fold_with(&mut bound_var_substitutor);
f9f354fc
XL
856
857 for var in named_parameters.values() {
f035d41b 858 parameters.insert(*var, chalk_ir::VariableKind::Lifetime);
f9f354fc
XL
859 }
860
861 (0..parameters.len()).for_each(|i| {
3dfed10e
XL
862 parameters
863 .get(&(i as u32))
fc512014 864 .or_else(|| bug!("Skipped bound var index: parameters={:?}", parameters));
f9f354fc
XL
865 });
866
1b1a35ee
XL
867 let binders =
868 chalk_ir::VariableKinds::from_iter(interner, parameters.into_iter().map(|(_, v)| v));
f9f354fc
XL
869
870 (new_ty, binders, named_parameters)
871}
872
f035d41b 873crate struct BoundVarsCollector<'tcx> {
f9f354fc 874 binder_index: ty::DebruijnIndex,
f035d41b 875 crate parameters: BTreeMap<u32, chalk_ir::VariableKind<RustInterner<'tcx>>>,
f9f354fc
XL
876 crate named_parameters: Vec<DefId>,
877}
878
f035d41b 879impl<'tcx> BoundVarsCollector<'tcx> {
5099ac24 880 crate fn new() -> Self {
f9f354fc
XL
881 BoundVarsCollector {
882 binder_index: ty::INNERMOST,
883 parameters: BTreeMap::new(),
884 named_parameters: vec![],
885 }
886 }
887}
888
f035d41b 889impl<'tcx> TypeVisitor<'tcx> for BoundVarsCollector<'tcx> {
cdc7bbd5
XL
890 fn visit_binder<T: TypeFoldable<'tcx>>(
891 &mut self,
892 t: &Binder<'tcx, T>,
893 ) -> ControlFlow<Self::BreakTy> {
f9f354fc
XL
894 self.binder_index.shift_in(1);
895 let result = t.super_visit_with(self);
896 self.binder_index.shift_out(1);
897 result
898 }
899
fc512014 900 fn visit_ty(&mut self, t: Ty<'tcx>) -> ControlFlow<Self::BreakTy> {
1b1a35ee 901 match *t.kind() {
f9f354fc
XL
902 ty::Bound(debruijn, bound_ty) if debruijn == self.binder_index => {
903 match self.parameters.entry(bound_ty.var.as_u32()) {
904 Entry::Vacant(entry) => {
29967ef6 905 entry.insert(chalk_ir::VariableKind::Ty(chalk_ir::TyVariableKind::General));
f9f354fc 906 }
f035d41b
XL
907 Entry::Occupied(entry) => match entry.get() {
908 chalk_ir::VariableKind::Ty(_) => {}
909 _ => panic!(),
910 },
f9f354fc
XL
911 }
912 }
913
914 _ => (),
915 };
916
917 t.super_visit_with(self)
918 }
919
fc512014 920 fn visit_region(&mut self, r: Region<'tcx>) -> ControlFlow<Self::BreakTy> {
5099ac24
FG
921 match *r {
922 ty::ReLateBound(index, br) if index == self.binder_index => match br.kind {
fc512014 923 ty::BoundRegionKind::BrNamed(def_id, _name) => {
c295e0f8 924 if !self.named_parameters.iter().any(|d| *d == def_id) {
fc512014 925 self.named_parameters.push(def_id);
f9f354fc
XL
926 }
927 }
928
fc512014 929 ty::BoundRegionKind::BrAnon(var) => match self.parameters.entry(var) {
f9f354fc 930 Entry::Vacant(entry) => {
f035d41b 931 entry.insert(chalk_ir::VariableKind::Lifetime);
f9f354fc 932 }
f035d41b
XL
933 Entry::Occupied(entry) => match entry.get() {
934 chalk_ir::VariableKind::Lifetime => {}
935 _ => panic!(),
936 },
f9f354fc
XL
937 },
938
cdc7bbd5 939 ty::BoundRegionKind::BrEnv => unimplemented!(),
f9f354fc
XL
940 },
941
942 ty::ReEarlyBound(_re) => {
943 // FIXME(chalk): jackh726 - I think we should always have already
944 // substituted away `ReEarlyBound`s for `ReLateBound`s, but need to confirm.
945 unimplemented!();
946 }
947
948 _ => (),
949 };
950
951 r.super_visit_with(self)
952 }
953}
954
fc512014 955/// This is used to replace `BoundRegionKind::BrNamed` with `BoundRegionKind::BrAnon`.
f9f354fc
XL
956/// Note: we assume that we will always have room for more bound vars. (i.e. we
957/// won't ever hit the `u32` limit in `BrAnon`s).
958struct NamedBoundVarSubstitutor<'a, 'tcx> {
959 tcx: TyCtxt<'tcx>,
960 binder_index: ty::DebruijnIndex,
961 named_parameters: &'a BTreeMap<DefId, u32>,
962}
963
964impl<'a, 'tcx> NamedBoundVarSubstitutor<'a, 'tcx> {
965 fn new(tcx: TyCtxt<'tcx>, named_parameters: &'a BTreeMap<DefId, u32>) -> Self {
966 NamedBoundVarSubstitutor { tcx, binder_index: ty::INNERMOST, named_parameters }
967 }
968}
969
970impl<'a, 'tcx> TypeFolder<'tcx> for NamedBoundVarSubstitutor<'a, 'tcx> {
971 fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
972 self.tcx
973 }
974
cdc7bbd5 975 fn fold_binder<T: TypeFoldable<'tcx>>(&mut self, t: Binder<'tcx, T>) -> Binder<'tcx, T> {
f9f354fc
XL
976 self.binder_index.shift_in(1);
977 let result = t.super_fold_with(self);
978 self.binder_index.shift_out(1);
979 result
980 }
981
f9f354fc 982 fn fold_region(&mut self, r: Region<'tcx>) -> Region<'tcx> {
5099ac24
FG
983 match *r {
984 ty::ReLateBound(index, br) if index == self.binder_index => match br.kind {
fc512014
XL
985 ty::BrNamed(def_id, _name) => match self.named_parameters.get(&def_id) {
986 Some(idx) => {
cdc7bbd5 987 let new_br = ty::BoundRegion { var: br.var, kind: ty::BrAnon(*idx) };
5099ac24 988 return self.tcx.mk_region(ty::ReLateBound(index, new_br));
f9f354fc 989 }
fc512014
XL
990 None => panic!("Missing `BrNamed`."),
991 },
f9f354fc 992 ty::BrEnv => unimplemented!(),
fc512014 993 ty::BrAnon(_) => {}
f9f354fc
XL
994 },
995 _ => (),
996 };
997
998 r.super_fold_with(self)
999 }
1000}
1001
1002/// Used to substitute `Param`s with placeholders. We do this since Chalk
1003/// have a notion of `Param`s.
1004crate struct ParamsSubstitutor<'tcx> {
1005 tcx: TyCtxt<'tcx>,
1006 binder_index: ty::DebruijnIndex,
1007 list: Vec<rustc_middle::ty::ParamTy>,
1b1a35ee 1008 next_ty_placeholder: usize,
f9f354fc
XL
1009 crate params: rustc_data_structures::fx::FxHashMap<usize, rustc_middle::ty::ParamTy>,
1010 crate named_regions: BTreeMap<DefId, u32>,
1011}
1012
1013impl<'tcx> ParamsSubstitutor<'tcx> {
1b1a35ee 1014 crate fn new(tcx: TyCtxt<'tcx>, next_ty_placeholder: usize) -> Self {
f9f354fc
XL
1015 ParamsSubstitutor {
1016 tcx,
1017 binder_index: ty::INNERMOST,
1018 list: vec![],
1b1a35ee 1019 next_ty_placeholder,
f9f354fc
XL
1020 params: rustc_data_structures::fx::FxHashMap::default(),
1021 named_regions: BTreeMap::default(),
1022 }
1023 }
1024}
1025
1026impl<'tcx> TypeFolder<'tcx> for ParamsSubstitutor<'tcx> {
1027 fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
1028 self.tcx
1029 }
1030
cdc7bbd5 1031 fn fold_binder<T: TypeFoldable<'tcx>>(&mut self, t: Binder<'tcx, T>) -> Binder<'tcx, T> {
f9f354fc
XL
1032 self.binder_index.shift_in(1);
1033 let result = t.super_fold_with(self);
1034 self.binder_index.shift_out(1);
1035 result
1036 }
1037
1038 fn fold_ty(&mut self, t: Ty<'tcx>) -> Ty<'tcx> {
1b1a35ee 1039 match *t.kind() {
f9f354fc 1040 ty::Param(param) => match self.list.iter().position(|r| r == &param) {
1b1a35ee 1041 Some(idx) => self.tcx.mk_ty(ty::Placeholder(ty::PlaceholderType {
f9f354fc 1042 universe: ty::UniverseIndex::from_usize(0),
1b1a35ee 1043 name: ty::BoundVar::from_usize(idx),
f9f354fc
XL
1044 })),
1045 None => {
1046 self.list.push(param);
1b1a35ee 1047 let idx = self.list.len() - 1 + self.next_ty_placeholder;
f9f354fc
XL
1048 self.params.insert(idx, param);
1049 self.tcx.mk_ty(ty::Placeholder(ty::PlaceholderType {
1050 universe: ty::UniverseIndex::from_usize(0),
1051 name: ty::BoundVar::from_usize(idx),
1052 }))
1053 }
1054 },
f9f354fc
XL
1055 _ => t.super_fold_with(self),
1056 }
1057 }
1058
1059 fn fold_region(&mut self, r: Region<'tcx>) -> Region<'tcx> {
5099ac24
FG
1060 match *r {
1061 // FIXME(chalk) - jackh726 - this currently isn't hit in any tests,
1062 // since canonicalization will already change these to canonical
1063 // variables (ty::ReLateBound).
f9f354fc 1064 ty::ReEarlyBound(_re) => match self.named_regions.get(&_re.def_id) {
fc512014 1065 Some(idx) => {
cdc7bbd5
XL
1066 let br = ty::BoundRegion {
1067 var: ty::BoundVar::from_u32(*idx),
1068 kind: ty::BrAnon(*idx),
1069 };
5099ac24 1070 self.tcx.mk_region(ty::ReLateBound(self.binder_index, br))
fc512014 1071 }
f9f354fc
XL
1072 None => {
1073 let idx = self.named_regions.len() as u32;
cdc7bbd5
XL
1074 let br =
1075 ty::BoundRegion { var: ty::BoundVar::from_u32(idx), kind: ty::BrAnon(idx) };
f9f354fc 1076 self.named_regions.insert(_re.def_id, idx);
5099ac24 1077 self.tcx.mk_region(ty::ReLateBound(self.binder_index, br))
f9f354fc
XL
1078 }
1079 },
1080
1081 _ => r.super_fold_with(self),
1082 }
1083 }
1084}
1b1a35ee 1085
5099ac24
FG
1086crate struct ReverseParamsSubstitutor<'tcx> {
1087 tcx: TyCtxt<'tcx>,
1088 params: rustc_data_structures::fx::FxHashMap<usize, rustc_middle::ty::ParamTy>,
1089}
1090
1091impl<'tcx> ReverseParamsSubstitutor<'tcx> {
1092 crate fn new(
1093 tcx: TyCtxt<'tcx>,
1094 params: rustc_data_structures::fx::FxHashMap<usize, rustc_middle::ty::ParamTy>,
1095 ) -> Self {
1096 Self { tcx, params }
1097 }
1098}
1099
1100impl<'tcx> TypeFolder<'tcx> for ReverseParamsSubstitutor<'tcx> {
1101 fn tcx<'b>(&'b self) -> TyCtxt<'tcx> {
1102 self.tcx
1103 }
1104
1105 fn fold_ty(&mut self, t: Ty<'tcx>) -> Ty<'tcx> {
1106 match *t.kind() {
1107 ty::Placeholder(ty::PlaceholderType { universe: ty::UniverseIndex::ROOT, name }) => {
1108 match self.params.get(&name.as_usize()) {
1109 Some(param) => self.tcx.mk_ty(ty::Param(*param)),
1110 None => t,
1111 }
1112 }
1113
1114 _ => t.super_fold_with(self),
1115 }
1116 }
1117}
1118
1b1a35ee
XL
1119/// Used to collect `Placeholder`s.
1120crate struct PlaceholdersCollector {
1121 universe_index: ty::UniverseIndex,
1122 crate next_ty_placeholder: usize,
1123 crate next_anon_region_placeholder: u32,
1124}
1125
1126impl PlaceholdersCollector {
1127 crate fn new() -> Self {
1128 PlaceholdersCollector {
1129 universe_index: ty::UniverseIndex::ROOT,
1130 next_ty_placeholder: 0,
1131 next_anon_region_placeholder: 0,
1132 }
1133 }
1134}
1135
1136impl<'tcx> TypeVisitor<'tcx> for PlaceholdersCollector {
fc512014 1137 fn visit_ty(&mut self, t: Ty<'tcx>) -> ControlFlow<Self::BreakTy> {
1b1a35ee
XL
1138 match t.kind() {
1139 ty::Placeholder(p) if p.universe == self.universe_index => {
1140 self.next_ty_placeholder = self.next_ty_placeholder.max(p.name.as_usize() + 1);
1141 }
1142
1143 _ => (),
1144 };
1145
1146 t.super_visit_with(self)
1147 }
1148
fc512014 1149 fn visit_region(&mut self, r: Region<'tcx>) -> ControlFlow<Self::BreakTy> {
5099ac24 1150 match *r {
1b1a35ee 1151 ty::RePlaceholder(p) if p.universe == self.universe_index => {
fc512014 1152 if let ty::BoundRegionKind::BrAnon(anon) = p.name {
1b1a35ee
XL
1153 self.next_anon_region_placeholder = self.next_anon_region_placeholder.max(anon);
1154 }
1155 }
1156
1157 _ => (),
1158 };
1159
1160 r.super_visit_with(self)
1161 }
1162}