aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_hir_ty/src/traits/chalk.rs
diff options
context:
space:
mode:
authorFlorian Diebold <[email protected]>2020-05-22 16:14:53 +0100
committerFlorian Diebold <[email protected]>2020-05-22 18:52:06 +0100
commit1d0e27254d6376e980dc19e67dfcc598c7e97231 (patch)
treeb3b0a66c3f2cf28ec82a091c29659e81adc10b9a /crates/ra_hir_ty/src/traits/chalk.rs
parentb38f9a5810f28563c45acb1ae8b82438649227c9 (diff)
Split up chalk module a bit
Diffstat (limited to 'crates/ra_hir_ty/src/traits/chalk.rs')
-rw-r--r--crates/ra_hir_ty/src/traits/chalk.rs970
1 files changed, 12 insertions, 958 deletions
diff --git a/crates/ra_hir_ty/src/traits/chalk.rs b/crates/ra_hir_ty/src/traits/chalk.rs
index 7d3ad6eb4..6c286ead7 100644
--- a/crates/ra_hir_ty/src/traits/chalk.rs
+++ b/crates/ra_hir_ty/src/traits/chalk.rs
@@ -1,325 +1,29 @@
1//! Conversion code from/to Chalk. 1//! Conversion code from/to Chalk.
2use std::{fmt, sync::Arc}; 2use std::sync::Arc;
3 3
4use log::debug; 4use log::debug;
5 5
6use chalk_ir::{ 6use chalk_ir::{fold::shift::Shift, GenericArg, TypeName};
7 cast::Cast, fold::shift::Shift, interner::HasInterner, GenericArg, Goal, GoalData,
8 PlaceholderIndex, Scalar, TypeName, UniverseIndex,
9};
10 7
11use hir_def::{ 8use hir_def::{
12 lang_item::{lang_attr, LangItemTarget}, 9 lang_item::{lang_attr, LangItemTarget},
13 type_ref::Mutability, 10 AssocContainerId, AssocItemId, HasModule, Lookup, TypeAliasId,
14 AssocContainerId, AssocItemId, GenericDefId, HasModule, Lookup, TypeAliasId,
15};
16use ra_db::{
17 salsa::{InternId, InternKey},
18 CrateId,
19}; 11};
12use ra_db::{salsa::InternKey, CrateId};
20 13
21use super::{builtin, AssocTyValue, Canonical, ChalkContext, Impl, Obligation}; 14use super::{builtin, AssocTyValue, ChalkContext, Impl};
22use crate::{ 15use crate::{
23 db::HirDatabase, 16 db::HirDatabase, display::HirDisplay, method_resolution::TyFingerprint, utils::generics,
24 display::HirDisplay, 17 DebruijnIndex, GenericPredicate, Substs, Ty, TypeCtor,
25 method_resolution::TyFingerprint,
26 primitive::{FloatBitness, FloatTy, IntBitness, IntTy, Signedness, Uncertain},
27 utils::generics,
28 ApplicationTy, DebruijnIndex, GenericPredicate, ProjectionTy, Substs, TraitRef, Ty, TypeCtor,
29}; 18};
30use chalk_rust_ir::WellKnownTrait; 19use chalk_rust_ir::WellKnownTrait;
20use mapping::{convert_where_clauses, generic_predicate_to_inline_bound, make_binders};
31 21
32pub(super) mod tls; 22pub use self::interner::*;
33
34#[derive(Debug, Copy, Clone, Hash, PartialOrd, Ord, PartialEq, Eq)]
35pub struct Interner;
36
37impl chalk_ir::interner::Interner for Interner {
38 type InternedType = Box<chalk_ir::TyData<Self>>; // FIXME use Arc?
39 type InternedLifetime = chalk_ir::LifetimeData<Self>;
40 type InternedConst = Arc<chalk_ir::ConstData<Self>>;
41 type InternedConcreteConst = ();
42 type InternedGenericArg = chalk_ir::GenericArgData<Self>;
43 type InternedGoal = Arc<GoalData<Self>>;
44 type InternedGoals = Vec<Goal<Self>>;
45 type InternedSubstitution = Vec<GenericArg<Self>>;
46 type InternedProgramClause = chalk_ir::ProgramClauseData<Self>;
47 type InternedProgramClauses = Arc<[chalk_ir::ProgramClause<Self>]>;
48 type InternedQuantifiedWhereClauses = Vec<chalk_ir::QuantifiedWhereClause<Self>>;
49 type InternedVariableKinds = Vec<chalk_ir::VariableKind<Self>>;
50 type InternedCanonicalVarKinds = Vec<chalk_ir::CanonicalVarKind<Self>>;
51 type DefId = InternId;
52 type InternedAdtId = InternId;
53 type Identifier = TypeAliasId;
54
55 fn debug_adt_id(type_kind_id: StructId, fmt: &mut fmt::Formatter<'_>) -> Option<fmt::Result> {
56 tls::with_current_program(|prog| Some(prog?.debug_struct_id(type_kind_id, fmt)))
57 }
58
59 fn debug_trait_id(type_kind_id: TraitId, fmt: &mut fmt::Formatter<'_>) -> Option<fmt::Result> {
60 tls::with_current_program(|prog| Some(prog?.debug_trait_id(type_kind_id, fmt)))
61 }
62
63 fn debug_assoc_type_id(id: AssocTypeId, fmt: &mut fmt::Formatter<'_>) -> Option<fmt::Result> {
64 tls::with_current_program(|prog| Some(prog?.debug_assoc_type_id(id, fmt)))
65 }
66
67 fn debug_alias(
68 alias: &chalk_ir::AliasTy<Interner>,
69 fmt: &mut fmt::Formatter<'_>,
70 ) -> Option<fmt::Result> {
71 tls::with_current_program(|prog| Some(prog?.debug_alias(alias, fmt)))
72 }
73
74 fn debug_projection_ty(
75 proj: &chalk_ir::ProjectionTy<Interner>,
76 fmt: &mut fmt::Formatter<'_>,
77 ) -> Option<fmt::Result> {
78 tls::with_current_program(|prog| Some(prog?.debug_projection_ty(proj, fmt)))
79 }
80
81 fn debug_opaque_ty(
82 opaque_ty: &chalk_ir::OpaqueTy<Interner>,
83 fmt: &mut fmt::Formatter<'_>,
84 ) -> Option<fmt::Result> {
85 tls::with_current_program(|prog| Some(prog?.debug_opaque_ty(opaque_ty, fmt)))
86 }
87
88 fn debug_opaque_ty_id(
89 opaque_ty_id: chalk_ir::OpaqueTyId<Self>,
90 fmt: &mut fmt::Formatter<'_>,
91 ) -> Option<fmt::Result> {
92 tls::with_current_program(|prog| Some(prog?.debug_opaque_ty_id(opaque_ty_id, fmt)))
93 }
94
95 fn debug_ty(ty: &chalk_ir::Ty<Interner>, fmt: &mut fmt::Formatter<'_>) -> Option<fmt::Result> {
96 tls::with_current_program(|prog| Some(prog?.debug_ty(ty, fmt)))
97 }
98
99 fn debug_lifetime(
100 lifetime: &chalk_ir::Lifetime<Interner>,
101 fmt: &mut fmt::Formatter<'_>,
102 ) -> Option<fmt::Result> {
103 tls::with_current_program(|prog| Some(prog?.debug_lifetime(lifetime, fmt)))
104 }
105
106 fn debug_generic_arg(
107 parameter: &GenericArg<Interner>,
108 fmt: &mut fmt::Formatter<'_>,
109 ) -> Option<fmt::Result> {
110 tls::with_current_program(|prog| Some(prog?.debug_generic_arg(parameter, fmt)))
111 }
112
113 fn debug_goal(goal: &Goal<Interner>, fmt: &mut fmt::Formatter<'_>) -> Option<fmt::Result> {
114 tls::with_current_program(|prog| Some(prog?.debug_goal(goal, fmt)))
115 }
116
117 fn debug_goals(
118 goals: &chalk_ir::Goals<Interner>,
119 fmt: &mut fmt::Formatter<'_>,
120 ) -> Option<fmt::Result> {
121 tls::with_current_program(|prog| Some(prog?.debug_goals(goals, fmt)))
122 }
123
124 fn debug_program_clause_implication(
125 pci: &chalk_ir::ProgramClauseImplication<Interner>,
126 fmt: &mut fmt::Formatter<'_>,
127 ) -> Option<fmt::Result> {
128 tls::with_current_program(|prog| Some(prog?.debug_program_clause_implication(pci, fmt)))
129 }
130
131 fn debug_application_ty(
132 application_ty: &chalk_ir::ApplicationTy<Interner>,
133 fmt: &mut fmt::Formatter<'_>,
134 ) -> Option<fmt::Result> {
135 tls::with_current_program(|prog| Some(prog?.debug_application_ty(application_ty, fmt)))
136 }
137
138 fn debug_substitution(
139 substitution: &chalk_ir::Substitution<Interner>,
140 fmt: &mut fmt::Formatter<'_>,
141 ) -> Option<fmt::Result> {
142 tls::with_current_program(|prog| Some(prog?.debug_substitution(substitution, fmt)))
143 }
144
145 fn debug_separator_trait_ref(
146 separator_trait_ref: &chalk_ir::SeparatorTraitRef<Interner>,
147 fmt: &mut fmt::Formatter<'_>,
148 ) -> Option<fmt::Result> {
149 tls::with_current_program(|prog| {
150 Some(prog?.debug_separator_trait_ref(separator_trait_ref, fmt))
151 })
152 }
153
154 fn intern_ty(&self, ty: chalk_ir::TyData<Self>) -> Box<chalk_ir::TyData<Self>> {
155 Box::new(ty)
156 }
157
158 fn ty_data<'a>(&self, ty: &'a Box<chalk_ir::TyData<Self>>) -> &'a chalk_ir::TyData<Self> {
159 ty
160 }
161
162 fn intern_lifetime(
163 &self,
164 lifetime: chalk_ir::LifetimeData<Self>,
165 ) -> chalk_ir::LifetimeData<Self> {
166 lifetime
167 }
168
169 fn lifetime_data<'a>(
170 &self,
171 lifetime: &'a chalk_ir::LifetimeData<Self>,
172 ) -> &'a chalk_ir::LifetimeData<Self> {
173 lifetime
174 }
175
176 fn intern_const(&self, constant: chalk_ir::ConstData<Self>) -> Arc<chalk_ir::ConstData<Self>> {
177 Arc::new(constant)
178 }
179
180 fn const_data<'a>(
181 &self,
182 constant: &'a Arc<chalk_ir::ConstData<Self>>,
183 ) -> &'a chalk_ir::ConstData<Self> {
184 constant
185 }
186
187 fn const_eq(&self, _ty: &Box<chalk_ir::TyData<Self>>, _c1: &(), _c2: &()) -> bool {
188 true
189 }
190
191 fn intern_generic_arg(
192 &self,
193 parameter: chalk_ir::GenericArgData<Self>,
194 ) -> chalk_ir::GenericArgData<Self> {
195 parameter
196 }
197
198 fn generic_arg_data<'a>(
199 &self,
200 parameter: &'a chalk_ir::GenericArgData<Self>,
201 ) -> &'a chalk_ir::GenericArgData<Self> {
202 parameter
203 }
204
205 fn intern_goal(&self, goal: GoalData<Self>) -> Arc<GoalData<Self>> {
206 Arc::new(goal)
207 }
208
209 fn intern_goals<E>(
210 &self,
211 data: impl IntoIterator<Item = Result<Goal<Self>, E>>,
212 ) -> Result<Self::InternedGoals, E> {
213 data.into_iter().collect()
214 }
215
216 fn goal_data<'a>(&self, goal: &'a Arc<GoalData<Self>>) -> &'a GoalData<Self> {
217 goal
218 }
219
220 fn goals_data<'a>(&self, goals: &'a Vec<Goal<Interner>>) -> &'a [Goal<Interner>] {
221 goals
222 }
223
224 fn intern_substitution<E>(
225 &self,
226 data: impl IntoIterator<Item = Result<GenericArg<Self>, E>>,
227 ) -> Result<Vec<GenericArg<Self>>, E> {
228 data.into_iter().collect()
229 }
230
231 fn substitution_data<'a>(
232 &self,
233 substitution: &'a Vec<GenericArg<Self>>,
234 ) -> &'a [GenericArg<Self>] {
235 substitution
236 }
237
238 fn intern_program_clause(
239 &self,
240 data: chalk_ir::ProgramClauseData<Self>,
241 ) -> chalk_ir::ProgramClauseData<Self> {
242 data
243 }
244
245 fn program_clause_data<'a>(
246 &self,
247 clause: &'a chalk_ir::ProgramClauseData<Self>,
248 ) -> &'a chalk_ir::ProgramClauseData<Self> {
249 clause
250 }
251
252 fn intern_program_clauses<E>(
253 &self,
254 data: impl IntoIterator<Item = Result<chalk_ir::ProgramClause<Self>, E>>,
255 ) -> Result<Arc<[chalk_ir::ProgramClause<Self>]>, E> {
256 data.into_iter().collect()
257 }
258
259 fn program_clauses_data<'a>(
260 &self,
261 clauses: &'a Arc<[chalk_ir::ProgramClause<Self>]>,
262 ) -> &'a [chalk_ir::ProgramClause<Self>] {
263 &clauses
264 }
265
266 fn intern_quantified_where_clauses<E>(
267 &self,
268 data: impl IntoIterator<Item = Result<chalk_ir::QuantifiedWhereClause<Self>, E>>,
269 ) -> Result<Self::InternedQuantifiedWhereClauses, E> {
270 data.into_iter().collect()
271 }
272
273 fn quantified_where_clauses_data<'a>(
274 &self,
275 clauses: &'a Self::InternedQuantifiedWhereClauses,
276 ) -> &'a [chalk_ir::QuantifiedWhereClause<Self>] {
277 clauses
278 }
279
280 fn intern_generic_arg_kinds<E>(
281 &self,
282 data: impl IntoIterator<Item = Result<chalk_ir::VariableKind<Self>, E>>,
283 ) -> Result<Self::InternedVariableKinds, E> {
284 data.into_iter().collect()
285 }
286
287 fn variable_kinds_data<'a>(
288 &self,
289 parameter_kinds: &'a Self::InternedVariableKinds,
290 ) -> &'a [chalk_ir::VariableKind<Self>] {
291 &parameter_kinds
292 }
293
294 fn intern_canonical_var_kinds<E>(
295 &self,
296 data: impl IntoIterator<Item = Result<chalk_ir::CanonicalVarKind<Self>, E>>,
297 ) -> Result<Self::InternedCanonicalVarKinds, E> {
298 data.into_iter().collect()
299 }
300
301 fn canonical_var_kinds_data<'a>(
302 &self,
303 canonical_var_kinds: &'a Self::InternedCanonicalVarKinds,
304 ) -> &'a [chalk_ir::CanonicalVarKind<Self>] {
305 &canonical_var_kinds
306 }
307}
308 23
309impl chalk_ir::interner::HasInterner for Interner { 24pub(super) mod tls;
310 type Interner = Self; 25mod interner;
311} 26mod mapping;
312
313pub type AssocTypeId = chalk_ir::AssocTypeId<Interner>;
314pub type AssociatedTyDatum = chalk_rust_ir::AssociatedTyDatum<Interner>;
315pub type TraitId = chalk_ir::TraitId<Interner>;
316pub type TraitDatum = chalk_rust_ir::TraitDatum<Interner>;
317pub type StructId = chalk_ir::AdtId<Interner>;
318pub type StructDatum = chalk_rust_ir::AdtDatum<Interner>;
319pub type ImplId = chalk_ir::ImplId<Interner>;
320pub type ImplDatum = chalk_rust_ir::ImplDatum<Interner>;
321pub type AssociatedTyValueId = chalk_rust_ir::AssociatedTyValueId<Interner>;
322pub type AssociatedTyValue = chalk_rust_ir::AssociatedTyValue<Interner>;
323 27
324pub(super) trait ToChalk { 28pub(super) trait ToChalk {
325 type Chalk; 29 type Chalk;
@@ -334,656 +38,6 @@ where
334 T::from_chalk(db, chalk) 38 T::from_chalk(db, chalk)
335} 39}
336 40
337impl ToChalk for Ty {
338 type Chalk = chalk_ir::Ty<Interner>;
339 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::Ty<Interner> {
340 match self {
341 Ty::Apply(apply_ty) => {
342 if let TypeCtor::Ref(m) = apply_ty.ctor {
343 return ref_to_chalk(db, m, apply_ty.parameters);
344 }
345 let name = apply_ty.ctor.to_chalk(db);
346 let substitution = apply_ty.parameters.to_chalk(db);
347 chalk_ir::ApplicationTy { name, substitution }.cast(&Interner).intern(&Interner)
348 }
349 Ty::Projection(proj_ty) => {
350 let associated_ty_id = proj_ty.associated_ty.to_chalk(db);
351 let substitution = proj_ty.parameters.to_chalk(db);
352 chalk_ir::AliasTy::Projection(chalk_ir::ProjectionTy {
353 associated_ty_id,
354 substitution,
355 })
356 .cast(&Interner)
357 .intern(&Interner)
358 }
359 Ty::Placeholder(id) => {
360 let interned_id = db.intern_type_param_id(id);
361 PlaceholderIndex {
362 ui: UniverseIndex::ROOT,
363 idx: interned_id.as_intern_id().as_usize(),
364 }
365 .to_ty::<Interner>(&Interner)
366 }
367 Ty::Bound(idx) => chalk_ir::TyData::BoundVar(idx).intern(&Interner),
368 Ty::Infer(_infer_ty) => panic!("uncanonicalized infer ty"),
369 Ty::Dyn(predicates) => {
370 let where_clauses = chalk_ir::QuantifiedWhereClauses::from(
371 &Interner,
372 predicates.iter().filter(|p| !p.is_error()).cloned().map(|p| p.to_chalk(db)),
373 );
374 let bounded_ty = chalk_ir::DynTy { bounds: make_binders(where_clauses, 1) };
375 chalk_ir::TyData::Dyn(bounded_ty).intern(&Interner)
376 }
377 Ty::Opaque(_) | Ty::Unknown => {
378 let substitution = chalk_ir::Substitution::empty(&Interner);
379 let name = TypeName::Error;
380 chalk_ir::ApplicationTy { name, substitution }.cast(&Interner).intern(&Interner)
381 }
382 }
383 }
384 fn from_chalk(db: &dyn HirDatabase, chalk: chalk_ir::Ty<Interner>) -> Self {
385 match chalk.data(&Interner).clone() {
386 chalk_ir::TyData::Apply(apply_ty) => match apply_ty.name {
387 TypeName::Error => Ty::Unknown,
388 TypeName::Ref(m) => ref_from_chalk(db, m, apply_ty.substitution),
389 _ => {
390 let ctor = from_chalk(db, apply_ty.name);
391 let parameters = from_chalk(db, apply_ty.substitution);
392 Ty::Apply(ApplicationTy { ctor, parameters })
393 }
394 },
395 chalk_ir::TyData::Placeholder(idx) => {
396 assert_eq!(idx.ui, UniverseIndex::ROOT);
397 let interned_id = crate::db::GlobalTypeParamId::from_intern_id(
398 crate::salsa::InternId::from(idx.idx),
399 );
400 Ty::Placeholder(db.lookup_intern_type_param_id(interned_id))
401 }
402 chalk_ir::TyData::Alias(chalk_ir::AliasTy::Projection(proj)) => {
403 let associated_ty = from_chalk(db, proj.associated_ty_id);
404 let parameters = from_chalk(db, proj.substitution);
405 Ty::Projection(ProjectionTy { associated_ty, parameters })
406 }
407 chalk_ir::TyData::Alias(chalk_ir::AliasTy::Opaque(_)) => unimplemented!(),
408 chalk_ir::TyData::Function(_) => unimplemented!(),
409 chalk_ir::TyData::BoundVar(idx) => Ty::Bound(idx),
410 chalk_ir::TyData::InferenceVar(_iv) => Ty::Unknown,
411 chalk_ir::TyData::Dyn(where_clauses) => {
412 assert_eq!(where_clauses.bounds.binders.len(&Interner), 1);
413 let predicates = where_clauses
414 .bounds
415 .skip_binders()
416 .iter(&Interner)
417 .map(|c| from_chalk(db, c.clone()))
418 .collect();
419 Ty::Dyn(predicates)
420 }
421 }
422 }
423}
424
425const LIFETIME_PLACEHOLDER: PlaceholderIndex =
426 PlaceholderIndex { ui: UniverseIndex::ROOT, idx: usize::MAX };
427
428/// We currently don't model lifetimes, but Chalk does. So, we have to insert a
429/// fake lifetime here, because Chalks built-in logic may expect it to be there.
430fn ref_to_chalk(
431 db: &dyn HirDatabase,
432 mutability: Mutability,
433 subst: Substs,
434) -> chalk_ir::Ty<Interner> {
435 let arg = subst[0].clone().to_chalk(db);
436 let lifetime = LIFETIME_PLACEHOLDER.to_lifetime(&Interner);
437 chalk_ir::ApplicationTy {
438 name: TypeName::Ref(mutability.to_chalk(db)),
439 substitution: chalk_ir::Substitution::from(
440 &Interner,
441 vec![lifetime.cast(&Interner), arg.cast(&Interner)],
442 ),
443 }
444 .intern(&Interner)
445}
446
447/// Here we remove the lifetime from the type we got from Chalk.
448fn ref_from_chalk(
449 db: &dyn HirDatabase,
450 mutability: chalk_ir::Mutability,
451 subst: chalk_ir::Substitution<Interner>,
452) -> Ty {
453 let tys = subst
454 .iter(&Interner)
455 .filter_map(|p| Some(from_chalk(db, p.ty(&Interner)?.clone())))
456 .collect();
457 Ty::apply(TypeCtor::Ref(from_chalk(db, mutability)), Substs(tys))
458}
459
460impl ToChalk for Substs {
461 type Chalk = chalk_ir::Substitution<Interner>;
462
463 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::Substitution<Interner> {
464 chalk_ir::Substitution::from(&Interner, self.iter().map(|ty| ty.clone().to_chalk(db)))
465 }
466
467 fn from_chalk(db: &dyn HirDatabase, parameters: chalk_ir::Substitution<Interner>) -> Substs {
468 let tys = parameters
469 .iter(&Interner)
470 .map(|p| match p.ty(&Interner) {
471 Some(ty) => from_chalk(db, ty.clone()),
472 None => unimplemented!(),
473 })
474 .collect();
475 Substs(tys)
476 }
477}
478
479impl ToChalk for TraitRef {
480 type Chalk = chalk_ir::TraitRef<Interner>;
481
482 fn to_chalk(self: TraitRef, db: &dyn HirDatabase) -> chalk_ir::TraitRef<Interner> {
483 let trait_id = self.trait_.to_chalk(db);
484 let substitution = self.substs.to_chalk(db);
485 chalk_ir::TraitRef { trait_id, substitution }
486 }
487
488 fn from_chalk(db: &dyn HirDatabase, trait_ref: chalk_ir::TraitRef<Interner>) -> Self {
489 let trait_ = from_chalk(db, trait_ref.trait_id);
490 let substs = from_chalk(db, trait_ref.substitution);
491 TraitRef { trait_, substs }
492 }
493}
494
495impl ToChalk for hir_def::TraitId {
496 type Chalk = TraitId;
497
498 fn to_chalk(self, _db: &dyn HirDatabase) -> TraitId {
499 chalk_ir::TraitId(self.as_intern_id())
500 }
501
502 fn from_chalk(_db: &dyn HirDatabase, trait_id: TraitId) -> hir_def::TraitId {
503 InternKey::from_intern_id(trait_id.0)
504 }
505}
506
507impl ToChalk for TypeCtor {
508 type Chalk = TypeName<Interner>;
509
510 fn to_chalk(self, db: &dyn HirDatabase) -> TypeName<Interner> {
511 match self {
512 TypeCtor::AssociatedType(type_alias) => {
513 let type_id = type_alias.to_chalk(db);
514 TypeName::AssociatedType(type_id)
515 }
516
517 TypeCtor::Bool => TypeName::Scalar(Scalar::Bool),
518 TypeCtor::Char => TypeName::Scalar(Scalar::Char),
519 TypeCtor::Int(Uncertain::Known(int_ty)) => TypeName::Scalar(int_ty_to_chalk(int_ty)),
520 TypeCtor::Float(Uncertain::Known(FloatTy { bitness: FloatBitness::X32 })) => {
521 TypeName::Scalar(Scalar::Float(chalk_ir::FloatTy::F32))
522 }
523 TypeCtor::Float(Uncertain::Known(FloatTy { bitness: FloatBitness::X64 })) => {
524 TypeName::Scalar(Scalar::Float(chalk_ir::FloatTy::F64))
525 }
526
527 TypeCtor::Tuple { cardinality } => TypeName::Tuple(cardinality.into()),
528 TypeCtor::RawPtr(mutability) => TypeName::Raw(mutability.to_chalk(db)),
529 TypeCtor::Slice => TypeName::Slice,
530 TypeCtor::Ref(mutability) => TypeName::Ref(mutability.to_chalk(db)),
531 TypeCtor::Str => TypeName::Str,
532
533 TypeCtor::Int(Uncertain::Unknown)
534 | TypeCtor::Float(Uncertain::Unknown)
535 | TypeCtor::Adt(_)
536 | TypeCtor::Array
537 | TypeCtor::FnDef(_)
538 | TypeCtor::FnPtr { .. }
539 | TypeCtor::Never
540 | TypeCtor::Closure { .. } => {
541 // other TypeCtors get interned and turned into a chalk StructId
542 let struct_id = db.intern_type_ctor(self).into();
543 TypeName::Adt(struct_id)
544 }
545 }
546 }
547
548 fn from_chalk(db: &dyn HirDatabase, type_name: TypeName<Interner>) -> TypeCtor {
549 match type_name {
550 TypeName::Adt(struct_id) => db.lookup_intern_type_ctor(struct_id.into()),
551 TypeName::AssociatedType(type_id) => TypeCtor::AssociatedType(from_chalk(db, type_id)),
552 TypeName::OpaqueType(_) => unreachable!(),
553
554 TypeName::Scalar(Scalar::Bool) => TypeCtor::Bool,
555 TypeName::Scalar(Scalar::Char) => TypeCtor::Char,
556 TypeName::Scalar(Scalar::Int(int_ty)) => TypeCtor::Int(Uncertain::Known(IntTy {
557 signedness: Signedness::Signed,
558 bitness: bitness_from_chalk_int(int_ty),
559 })),
560 TypeName::Scalar(Scalar::Uint(uint_ty)) => TypeCtor::Int(Uncertain::Known(IntTy {
561 signedness: Signedness::Unsigned,
562 bitness: bitness_from_chalk_uint(uint_ty),
563 })),
564 TypeName::Scalar(Scalar::Float(chalk_ir::FloatTy::F32)) => {
565 TypeCtor::Float(Uncertain::Known(FloatTy { bitness: FloatBitness::X32 }))
566 }
567 TypeName::Scalar(Scalar::Float(chalk_ir::FloatTy::F64)) => {
568 TypeCtor::Float(Uncertain::Known(FloatTy { bitness: FloatBitness::X64 }))
569 }
570 TypeName::Tuple(cardinality) => TypeCtor::Tuple { cardinality: cardinality as u16 },
571 TypeName::Raw(mutability) => TypeCtor::RawPtr(from_chalk(db, mutability)),
572 TypeName::Slice => TypeCtor::Slice,
573 TypeName::Ref(mutability) => TypeCtor::Ref(from_chalk(db, mutability)),
574 TypeName::Str => TypeCtor::Str,
575
576 TypeName::FnDef(_) => unreachable!(),
577
578 TypeName::Error => {
579 // this should not be reached, since we don't represent TypeName::Error with TypeCtor
580 unreachable!()
581 }
582 }
583 }
584}
585
586fn bitness_from_chalk_uint(uint_ty: chalk_ir::UintTy) -> IntBitness {
587 use chalk_ir::UintTy;
588
589 match uint_ty {
590 UintTy::Usize => IntBitness::Xsize,
591 UintTy::U8 => IntBitness::X8,
592 UintTy::U16 => IntBitness::X16,
593 UintTy::U32 => IntBitness::X32,
594 UintTy::U64 => IntBitness::X64,
595 UintTy::U128 => IntBitness::X128,
596 }
597}
598
599fn bitness_from_chalk_int(int_ty: chalk_ir::IntTy) -> IntBitness {
600 use chalk_ir::IntTy;
601
602 match int_ty {
603 IntTy::Isize => IntBitness::Xsize,
604 IntTy::I8 => IntBitness::X8,
605 IntTy::I16 => IntBitness::X16,
606 IntTy::I32 => IntBitness::X32,
607 IntTy::I64 => IntBitness::X64,
608 IntTy::I128 => IntBitness::X128,
609 }
610}
611
612fn int_ty_to_chalk(int_ty: IntTy) -> Scalar {
613 use chalk_ir::{IntTy, UintTy};
614
615 match int_ty.signedness {
616 Signedness::Signed => Scalar::Int(match int_ty.bitness {
617 IntBitness::Xsize => IntTy::Isize,
618 IntBitness::X8 => IntTy::I8,
619 IntBitness::X16 => IntTy::I16,
620 IntBitness::X32 => IntTy::I32,
621 IntBitness::X64 => IntTy::I64,
622 IntBitness::X128 => IntTy::I128,
623 }),
624 Signedness::Unsigned => Scalar::Uint(match int_ty.bitness {
625 IntBitness::Xsize => UintTy::Usize,
626 IntBitness::X8 => UintTy::U8,
627 IntBitness::X16 => UintTy::U16,
628 IntBitness::X32 => UintTy::U32,
629 IntBitness::X64 => UintTy::U64,
630 IntBitness::X128 => UintTy::U128,
631 }),
632 }
633}
634
635impl ToChalk for Mutability {
636 type Chalk = chalk_ir::Mutability;
637 fn to_chalk(self, _db: &dyn HirDatabase) -> Self::Chalk {
638 match self {
639 Mutability::Shared => chalk_ir::Mutability::Not,
640 Mutability::Mut => chalk_ir::Mutability::Mut,
641 }
642 }
643 fn from_chalk(_db: &dyn HirDatabase, chalk: Self::Chalk) -> Self {
644 match chalk {
645 chalk_ir::Mutability::Mut => Mutability::Mut,
646 chalk_ir::Mutability::Not => Mutability::Shared,
647 }
648 }
649}
650
651impl ToChalk for Impl {
652 type Chalk = ImplId;
653
654 fn to_chalk(self, db: &dyn HirDatabase) -> ImplId {
655 db.intern_chalk_impl(self).into()
656 }
657
658 fn from_chalk(db: &dyn HirDatabase, impl_id: ImplId) -> Impl {
659 db.lookup_intern_chalk_impl(impl_id.into())
660 }
661}
662
663impl ToChalk for TypeAliasId {
664 type Chalk = AssocTypeId;
665
666 fn to_chalk(self, _db: &dyn HirDatabase) -> AssocTypeId {
667 chalk_ir::AssocTypeId(self.as_intern_id())
668 }
669
670 fn from_chalk(_db: &dyn HirDatabase, type_alias_id: AssocTypeId) -> TypeAliasId {
671 InternKey::from_intern_id(type_alias_id.0)
672 }
673}
674
675impl ToChalk for AssocTyValue {
676 type Chalk = AssociatedTyValueId;
677
678 fn to_chalk(self, db: &dyn HirDatabase) -> AssociatedTyValueId {
679 db.intern_assoc_ty_value(self).into()
680 }
681
682 fn from_chalk(db: &dyn HirDatabase, assoc_ty_value_id: AssociatedTyValueId) -> AssocTyValue {
683 db.lookup_intern_assoc_ty_value(assoc_ty_value_id.into())
684 }
685}
686
687impl ToChalk for GenericPredicate {
688 type Chalk = chalk_ir::QuantifiedWhereClause<Interner>;
689
690 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::QuantifiedWhereClause<Interner> {
691 match self {
692 GenericPredicate::Implemented(trait_ref) => {
693 let chalk_trait_ref = trait_ref.to_chalk(db);
694 let chalk_trait_ref = chalk_trait_ref.shifted_in(&Interner);
695 make_binders(chalk_ir::WhereClause::Implemented(chalk_trait_ref), 0)
696 }
697 GenericPredicate::Projection(projection_pred) => {
698 let ty = projection_pred.ty.to_chalk(db).shifted_in(&Interner);
699 let projection = projection_pred.projection_ty.to_chalk(db).shifted_in(&Interner);
700 let alias = chalk_ir::AliasTy::Projection(projection);
701 make_binders(chalk_ir::WhereClause::AliasEq(chalk_ir::AliasEq { alias, ty }), 0)
702 }
703 GenericPredicate::Error => panic!("tried passing GenericPredicate::Error to Chalk"),
704 }
705 }
706
707 fn from_chalk(
708 db: &dyn HirDatabase,
709 where_clause: chalk_ir::QuantifiedWhereClause<Interner>,
710 ) -> GenericPredicate {
711 // we don't produce any where clauses with binders and can't currently deal with them
712 match where_clause
713 .skip_binders()
714 .shifted_out(&Interner)
715 .expect("unexpected bound vars in where clause")
716 {
717 chalk_ir::WhereClause::Implemented(tr) => {
718 GenericPredicate::Implemented(from_chalk(db, tr))
719 }
720 chalk_ir::WhereClause::AliasEq(projection_eq) => {
721 let projection_ty = from_chalk(
722 db,
723 match projection_eq.alias {
724 chalk_ir::AliasTy::Projection(p) => p,
725 _ => unimplemented!(),
726 },
727 );
728 let ty = from_chalk(db, projection_eq.ty);
729 GenericPredicate::Projection(super::ProjectionPredicate { projection_ty, ty })
730 }
731 }
732 }
733}
734
735impl ToChalk for ProjectionTy {
736 type Chalk = chalk_ir::ProjectionTy<Interner>;
737
738 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::ProjectionTy<Interner> {
739 chalk_ir::ProjectionTy {
740 associated_ty_id: self.associated_ty.to_chalk(db),
741 substitution: self.parameters.to_chalk(db),
742 }
743 }
744
745 fn from_chalk(
746 db: &dyn HirDatabase,
747 projection_ty: chalk_ir::ProjectionTy<Interner>,
748 ) -> ProjectionTy {
749 ProjectionTy {
750 associated_ty: from_chalk(db, projection_ty.associated_ty_id),
751 parameters: from_chalk(db, projection_ty.substitution),
752 }
753 }
754}
755
756impl ToChalk for super::ProjectionPredicate {
757 type Chalk = chalk_ir::AliasEq<Interner>;
758
759 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::AliasEq<Interner> {
760 chalk_ir::AliasEq {
761 alias: chalk_ir::AliasTy::Projection(self.projection_ty.to_chalk(db)),
762 ty: self.ty.to_chalk(db),
763 }
764 }
765
766 fn from_chalk(_db: &dyn HirDatabase, _normalize: chalk_ir::AliasEq<Interner>) -> Self {
767 unimplemented!()
768 }
769}
770
771impl ToChalk for Obligation {
772 type Chalk = chalk_ir::DomainGoal<Interner>;
773
774 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::DomainGoal<Interner> {
775 match self {
776 Obligation::Trait(tr) => tr.to_chalk(db).cast(&Interner),
777 Obligation::Projection(pr) => pr.to_chalk(db).cast(&Interner),
778 }
779 }
780
781 fn from_chalk(_db: &dyn HirDatabase, _goal: chalk_ir::DomainGoal<Interner>) -> Self {
782 unimplemented!()
783 }
784}
785
786impl<T> ToChalk for Canonical<T>
787where
788 T: ToChalk,
789 T::Chalk: HasInterner<Interner = Interner>,
790{
791 type Chalk = chalk_ir::Canonical<T::Chalk>;
792
793 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::Canonical<T::Chalk> {
794 let parameter = chalk_ir::CanonicalVarKind::new(
795 chalk_ir::VariableKind::Ty,
796 chalk_ir::UniverseIndex::ROOT,
797 );
798 let value = self.value.to_chalk(db);
799 chalk_ir::Canonical {
800 value,
801 binders: chalk_ir::CanonicalVarKinds::from(&Interner, vec![parameter; self.num_vars]),
802 }
803 }
804
805 fn from_chalk(db: &dyn HirDatabase, canonical: chalk_ir::Canonical<T::Chalk>) -> Canonical<T> {
806 Canonical {
807 num_vars: canonical.binders.len(&Interner),
808 value: from_chalk(db, canonical.value),
809 }
810 }
811}
812
813impl ToChalk for Arc<super::TraitEnvironment> {
814 type Chalk = chalk_ir::Environment<Interner>;
815
816 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::Environment<Interner> {
817 let mut clauses = Vec::new();
818 for pred in &self.predicates {
819 if pred.is_error() {
820 // for env, we just ignore errors
821 continue;
822 }
823 let program_clause: chalk_ir::ProgramClause<Interner> =
824 pred.clone().to_chalk(db).cast(&Interner);
825 clauses.push(program_clause.into_from_env_clause(&Interner));
826 }
827 chalk_ir::Environment::new(&Interner).add_clauses(&Interner, clauses)
828 }
829
830 fn from_chalk(
831 _db: &dyn HirDatabase,
832 _env: chalk_ir::Environment<Interner>,
833 ) -> Arc<super::TraitEnvironment> {
834 unimplemented!()
835 }
836}
837
838impl<T: ToChalk> ToChalk for super::InEnvironment<T>
839where
840 T::Chalk: chalk_ir::interner::HasInterner<Interner = Interner>,
841{
842 type Chalk = chalk_ir::InEnvironment<T::Chalk>;
843
844 fn to_chalk(self, db: &dyn HirDatabase) -> chalk_ir::InEnvironment<T::Chalk> {
845 chalk_ir::InEnvironment {
846 environment: self.environment.to_chalk(db),
847 goal: self.value.to_chalk(db),
848 }
849 }
850
851 fn from_chalk(
852 db: &dyn HirDatabase,
853 in_env: chalk_ir::InEnvironment<T::Chalk>,
854 ) -> super::InEnvironment<T> {
855 super::InEnvironment {
856 environment: from_chalk(db, in_env.environment),
857 value: from_chalk(db, in_env.goal),
858 }
859 }
860}
861
862impl ToChalk for builtin::BuiltinImplData {
863 type Chalk = ImplDatum;
864
865 fn to_chalk(self, db: &dyn HirDatabase) -> ImplDatum {
866 let impl_type = chalk_rust_ir::ImplType::External;
867 let where_clauses = self.where_clauses.into_iter().map(|w| w.to_chalk(db)).collect();
868
869 let impl_datum_bound =
870 chalk_rust_ir::ImplDatumBound { trait_ref: self.trait_ref.to_chalk(db), where_clauses };
871 let associated_ty_value_ids =
872 self.assoc_ty_values.into_iter().map(|v| v.to_chalk(db)).collect();
873 chalk_rust_ir::ImplDatum {
874 binders: make_binders(impl_datum_bound, self.num_vars),
875 impl_type,
876 polarity: chalk_rust_ir::Polarity::Positive,
877 associated_ty_value_ids,
878 }
879 }
880
881 fn from_chalk(_db: &dyn HirDatabase, _data: ImplDatum) -> Self {
882 unimplemented!()
883 }
884}
885
886impl ToChalk for builtin::BuiltinImplAssocTyValueData {
887 type Chalk = AssociatedTyValue;
888
889 fn to_chalk(self, db: &dyn HirDatabase) -> AssociatedTyValue {
890 let ty = self.value.to_chalk(db);
891 let value_bound = chalk_rust_ir::AssociatedTyValueBound { ty };
892
893 chalk_rust_ir::AssociatedTyValue {
894 associated_ty_id: self.assoc_ty_id.to_chalk(db),
895 impl_id: self.impl_.to_chalk(db),
896 value: make_binders(value_bound, self.num_vars),
897 }
898 }
899
900 fn from_chalk(
901 _db: &dyn HirDatabase,
902 _data: AssociatedTyValue,
903 ) -> builtin::BuiltinImplAssocTyValueData {
904 unimplemented!()
905 }
906}
907
908fn make_binders<T>(value: T, num_vars: usize) -> chalk_ir::Binders<T>
909where
910 T: HasInterner<Interner = Interner>,
911{
912 chalk_ir::Binders::new(
913 chalk_ir::VariableKinds::from(
914 &Interner,
915 std::iter::repeat(chalk_ir::VariableKind::Ty).take(num_vars),
916 ),
917 value,
918 )
919}
920
921fn convert_where_clauses(
922 db: &dyn HirDatabase,
923 def: GenericDefId,
924 substs: &Substs,
925) -> Vec<chalk_ir::QuantifiedWhereClause<Interner>> {
926 let generic_predicates = db.generic_predicates(def);
927 let mut result = Vec::with_capacity(generic_predicates.len());
928 for pred in generic_predicates.iter() {
929 if pred.value.is_error() {
930 // skip errored predicates completely
931 continue;
932 }
933 result.push(pred.clone().subst(substs).to_chalk(db));
934 }
935 result
936}
937
938fn generic_predicate_to_inline_bound(
939 db: &dyn HirDatabase,
940 pred: &GenericPredicate,
941 self_ty: &Ty,
942) -> Option<chalk_rust_ir::InlineBound<Interner>> {
943 // An InlineBound is like a GenericPredicate, except the self type is left out.
944 // We don't have a special type for this, but Chalk does.
945 match pred {
946 GenericPredicate::Implemented(trait_ref) => {
947 if &trait_ref.substs[0] != self_ty {
948 // we can only convert predicates back to type bounds if they
949 // have the expected self type
950 return None;
951 }
952 let args_no_self = trait_ref.substs[1..]
953 .iter()
954 .map(|ty| ty.clone().to_chalk(db).cast(&Interner))
955 .collect();
956 let trait_bound =
957 chalk_rust_ir::TraitBound { trait_id: trait_ref.trait_.to_chalk(db), args_no_self };
958 Some(chalk_rust_ir::InlineBound::TraitBound(trait_bound))
959 }
960 GenericPredicate::Projection(proj) => {
961 if &proj.projection_ty.parameters[0] != self_ty {
962 return None;
963 }
964 let trait_ = match proj.projection_ty.associated_ty.lookup(db.upcast()).container {
965 AssocContainerId::TraitId(t) => t,
966 _ => panic!("associated type not in trait"),
967 };
968 let args_no_self = proj.projection_ty.parameters[1..]
969 .iter()
970 .map(|ty| ty.clone().to_chalk(db).cast(&Interner))
971 .collect();
972 let alias_eq_bound = chalk_rust_ir::AliasEqBound {
973 value: proj.ty.clone().to_chalk(db),
974 trait_bound: chalk_rust_ir::TraitBound {
975 trait_id: trait_.to_chalk(db),
976 args_no_self,
977 },
978 associated_ty_id: proj.projection_ty.associated_ty.to_chalk(db),
979 parameters: Vec::new(), // FIXME we don't support generic associated types yet
980 };
981 Some(chalk_rust_ir::InlineBound::AliasEqBound(alias_eq_bound))
982 }
983 GenericPredicate::Error => None,
984 }
985}
986
987impl<'a> chalk_solve::RustIrDatabase<Interner> for ChalkContext<'a> { 41impl<'a> chalk_solve::RustIrDatabase<Interner> for ChalkContext<'a> {
988 fn associated_ty_data(&self, id: AssocTypeId) -> Arc<AssociatedTyDatum> { 42 fn associated_ty_data(&self, id: AssocTypeId) -> Arc<AssociatedTyDatum> {
989 self.db.associated_ty_data(id) 43 self.db.associated_ty_data(id)