1use std::cell::{Cell, RefCell};
2use std::fmt;
3
4pub use at::DefineOpaqueTypes;
5use free_regions::RegionRelations;
6pub use freshen::TypeFreshener;
7use lexical_region_resolve::LexicalRegionResolutions;
8pub use lexical_region_resolve::RegionResolutionError;
9pub use opaque_types::{OpaqueTypeStorage, OpaqueTypeStorageEntries, OpaqueTypeTable};
10use region_constraints::{
11 GenericKind, RegionConstraintCollector, RegionConstraintStorage, VarInfos, VerifyBound,
12};
13pub use relate::StructurallyRelateAliases;
14pub use relate::combine::PredicateEmittingRelation;
15use rustc_data_structures::fx::{FxHashSet, FxIndexMap};
16use rustc_data_structures::undo_log::{Rollback, UndoLogs};
17use rustc_data_structures::unify as ut;
18use rustc_errors::{DiagCtxtHandle, ErrorGuaranteed};
19use rustc_hir as hir;
20use rustc_hir::def_id::{DefId, LocalDefId};
21use rustc_macros::extension;
22pub use rustc_macros::{TypeFoldable, TypeVisitable};
23use rustc_middle::bug;
24use rustc_middle::infer::canonical::{CanonicalQueryInput, CanonicalVarValues};
25use rustc_middle::mir::ConstraintCategory;
26use rustc_middle::traits::select;
27use rustc_middle::traits::solve::Goal;
28use rustc_middle::ty::error::{ExpectedFound, TypeError};
29use rustc_middle::ty::{
30 self, BoundVarReplacerDelegate, ConstVid, FloatVid, GenericArg, GenericArgKind, GenericArgs,
31 GenericArgsRef, GenericParamDefKind, InferConst, IntVid, OpaqueHiddenType, OpaqueTypeKey,
32 PseudoCanonicalInput, Term, TermKind, Ty, TyCtxt, TyVid, TypeFoldable, TypeFolder,
33 TypeSuperFoldable, TypeVisitable, TypeVisitableExt, TypingEnv, TypingMode, fold_regions,
34};
35use rustc_span::{DUMMY_SP, Span, Symbol};
36use snapshot::undo_log::InferCtxtUndoLogs;
37use tracing::{debug, instrument};
38use type_variable::TypeVariableOrigin;
39
40use crate::infer::snapshot::undo_log::UndoLog;
41use crate::infer::unify_key::{ConstVariableOrigin, ConstVariableValue, ConstVidKey};
42use crate::traits::{
43 self, ObligationCause, ObligationInspector, PredicateObligation, PredicateObligations,
44 TraitEngine,
45};
46
47pub mod at;
48pub mod canonical;
49mod context;
50mod free_regions;
51mod freshen;
52mod lexical_region_resolve;
53mod opaque_types;
54pub mod outlives;
55mod projection;
56pub mod region_constraints;
57pub mod relate;
58pub mod resolve;
59pub(crate) mod snapshot;
60mod type_variable;
61mod unify_key;
62
63#[must_use]
71#[derive(Debug)]
72pub struct InferOk<'tcx, T> {
73 pub value: T,
74 pub obligations: PredicateObligations<'tcx>,
75}
76pub type InferResult<'tcx, T> = Result<InferOk<'tcx, T>, TypeError<'tcx>>;
77
78pub(crate) type FixupResult<T> = Result<T, FixupError>; pub(crate) type UnificationTable<'a, 'tcx, T> = ut::UnificationTable<
81 ut::InPlace<T, &'a mut ut::UnificationStorage<T>, &'a mut InferCtxtUndoLogs<'tcx>>,
82>;
83
84#[derive(Clone)]
89pub struct InferCtxtInner<'tcx> {
90 undo_log: InferCtxtUndoLogs<'tcx>,
91
92 projection_cache: traits::ProjectionCacheStorage<'tcx>,
96
97 type_variable_storage: type_variable::TypeVariableStorage<'tcx>,
101
102 const_unification_storage: ut::UnificationTableStorage<ConstVidKey<'tcx>>,
104
105 int_unification_storage: ut::UnificationTableStorage<ty::IntVid>,
107
108 float_unification_storage: ut::UnificationTableStorage<ty::FloatVid>,
110
111 region_constraint_storage: Option<RegionConstraintStorage<'tcx>>,
118
119 region_obligations: Vec<TypeOutlivesConstraint<'tcx>>,
152
153 region_assumptions: Vec<ty::ArgOutlivesPredicate<'tcx>>,
159
160 hir_typeck_potentially_region_dependent_goals: Vec<PredicateObligation<'tcx>>,
166
167 opaque_type_storage: OpaqueTypeStorage<'tcx>,
169}
170
171impl<'tcx> InferCtxtInner<'tcx> {
172 fn new() -> InferCtxtInner<'tcx> {
173 InferCtxtInner {
174 undo_log: InferCtxtUndoLogs::default(),
175
176 projection_cache: Default::default(),
177 type_variable_storage: Default::default(),
178 const_unification_storage: Default::default(),
179 int_unification_storage: Default::default(),
180 float_unification_storage: Default::default(),
181 region_constraint_storage: Some(Default::default()),
182 region_obligations: Default::default(),
183 region_assumptions: Default::default(),
184 hir_typeck_potentially_region_dependent_goals: Default::default(),
185 opaque_type_storage: Default::default(),
186 }
187 }
188
189 #[inline]
190 pub fn region_obligations(&self) -> &[TypeOutlivesConstraint<'tcx>] {
191 &self.region_obligations
192 }
193
194 #[inline]
195 pub fn region_assumptions(&self) -> &[ty::ArgOutlivesPredicate<'tcx>] {
196 &self.region_assumptions
197 }
198
199 #[inline]
200 pub fn projection_cache(&mut self) -> traits::ProjectionCache<'_, 'tcx> {
201 self.projection_cache.with_log(&mut self.undo_log)
202 }
203
204 #[inline]
205 fn try_type_variables_probe_ref(
206 &self,
207 vid: ty::TyVid,
208 ) -> Option<&type_variable::TypeVariableValue<'tcx>> {
209 self.type_variable_storage.eq_relations_ref().try_probe_value(vid)
212 }
213
214 #[inline]
215 fn type_variables(&mut self) -> type_variable::TypeVariableTable<'_, 'tcx> {
216 self.type_variable_storage.with_log(&mut self.undo_log)
217 }
218
219 #[inline]
220 pub fn opaque_types(&mut self) -> opaque_types::OpaqueTypeTable<'_, 'tcx> {
221 self.opaque_type_storage.with_log(&mut self.undo_log)
222 }
223
224 #[inline]
225 fn int_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ty::IntVid> {
226 self.int_unification_storage.with_log(&mut self.undo_log)
227 }
228
229 #[inline]
230 fn float_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ty::FloatVid> {
231 self.float_unification_storage.with_log(&mut self.undo_log)
232 }
233
234 #[inline]
235 fn const_unification_table(&mut self) -> UnificationTable<'_, 'tcx, ConstVidKey<'tcx>> {
236 self.const_unification_storage.with_log(&mut self.undo_log)
237 }
238
239 #[inline]
240 pub fn unwrap_region_constraints(&mut self) -> RegionConstraintCollector<'_, 'tcx> {
241 self.region_constraint_storage
242 .as_mut()
243 .expect("region constraints already solved")
244 .with_log(&mut self.undo_log)
245 }
246}
247
248pub struct InferCtxt<'tcx> {
249 pub tcx: TyCtxt<'tcx>,
250
251 typing_mode: TypingMode<'tcx>,
254
255 pub considering_regions: bool,
259 pub in_hir_typeck: bool,
279
280 skip_leak_check: bool,
285
286 pub inner: RefCell<InferCtxtInner<'tcx>>,
287
288 lexical_region_resolutions: RefCell<Option<LexicalRegionResolutions<'tcx>>>,
290
291 pub selection_cache: select::SelectionCache<'tcx, ty::ParamEnv<'tcx>>,
294
295 pub evaluation_cache: select::EvaluationCache<'tcx, ty::ParamEnv<'tcx>>,
298
299 pub reported_trait_errors:
302 RefCell<FxIndexMap<Span, (Vec<Goal<'tcx, ty::Predicate<'tcx>>>, ErrorGuaranteed)>>,
303
304 pub reported_signature_mismatch: RefCell<FxHashSet<(Span, Option<Span>)>>,
305
306 tainted_by_errors: Cell<Option<ErrorGuaranteed>>,
314
315 universe: Cell<ty::UniverseIndex>,
325
326 next_trait_solver: bool,
327
328 pub obligation_inspector: Cell<Option<ObligationInspector<'tcx>>>,
329}
330
331#[derive(Clone, Copy, Debug, PartialEq, Eq, TypeFoldable, TypeVisitable)]
333pub enum ValuePairs<'tcx> {
334 Regions(ExpectedFound<ty::Region<'tcx>>),
335 Terms(ExpectedFound<ty::Term<'tcx>>),
336 Aliases(ExpectedFound<ty::AliasTerm<'tcx>>),
337 TraitRefs(ExpectedFound<ty::TraitRef<'tcx>>),
338 PolySigs(ExpectedFound<ty::PolyFnSig<'tcx>>),
339 ExistentialTraitRef(ExpectedFound<ty::PolyExistentialTraitRef<'tcx>>),
340 ExistentialProjection(ExpectedFound<ty::PolyExistentialProjection<'tcx>>),
341}
342
343impl<'tcx> ValuePairs<'tcx> {
344 pub fn ty(&self) -> Option<(Ty<'tcx>, Ty<'tcx>)> {
345 if let ValuePairs::Terms(ExpectedFound { expected, found }) = self
346 && let Some(expected) = expected.as_type()
347 && let Some(found) = found.as_type()
348 {
349 Some((expected, found))
350 } else {
351 None
352 }
353 }
354}
355
356#[derive(Clone, Debug)]
361pub struct TypeTrace<'tcx> {
362 pub cause: ObligationCause<'tcx>,
363 pub values: ValuePairs<'tcx>,
364}
365
366#[derive(Clone, Debug)]
370pub enum SubregionOrigin<'tcx> {
371 Subtype(Box<TypeTrace<'tcx>>),
373
374 RelateObjectBound(Span),
377
378 RelateParamBound(Span, Ty<'tcx>, Option<Span>),
381
382 RelateRegionParamBound(Span, Option<Ty<'tcx>>),
385
386 Reborrow(Span),
388
389 ReferenceOutlivesReferent(Ty<'tcx>, Span),
391
392 CompareImplItemObligation {
395 span: Span,
396 impl_item_def_id: LocalDefId,
397 trait_item_def_id: DefId,
398 },
399
400 CheckAssociatedTypeBounds {
402 parent: Box<SubregionOrigin<'tcx>>,
403 impl_item_def_id: LocalDefId,
404 trait_item_def_id: DefId,
405 },
406
407 AscribeUserTypeProvePredicate(Span),
408}
409
410#[cfg(target_pointer_width = "64")]
412rustc_data_structures::static_assert_size!(SubregionOrigin<'_>, 32);
413
414impl<'tcx> SubregionOrigin<'tcx> {
415 pub fn to_constraint_category(&self) -> ConstraintCategory<'tcx> {
416 match self {
417 Self::Subtype(type_trace) => type_trace.cause.to_constraint_category(),
418 Self::AscribeUserTypeProvePredicate(span) => ConstraintCategory::Predicate(*span),
419 _ => ConstraintCategory::BoringNoLocation,
420 }
421 }
422}
423
424#[derive(Clone, Copy, Debug)]
426pub enum BoundRegionConversionTime {
427 FnCall,
429
430 HigherRankedType,
432
433 AssocTypeProjection(DefId),
435}
436
437#[derive(Copy, Clone, Debug)]
441pub enum RegionVariableOrigin {
442 Misc(Span),
446
447 PatternRegion(Span),
449
450 BorrowRegion(Span),
452
453 Autoref(Span),
455
456 Coercion(Span),
458
459 RegionParameterDefinition(Span, Symbol),
464
465 BoundRegion(Span, ty::BoundRegionKind, BoundRegionConversionTime),
468
469 UpvarRegion(ty::UpvarId, Span),
470
471 Nll(NllRegionVariableOrigin),
474}
475
476#[derive(Copy, Clone, Debug)]
477pub enum NllRegionVariableOrigin {
478 FreeRegion,
482
483 Placeholder(ty::PlaceholderRegion),
486
487 Existential {
488 name: Option<Symbol>,
489 },
490}
491
492#[derive(Copy, Clone, Debug)]
493pub struct FixupError {
494 unresolved: TyOrConstInferVar,
495}
496
497impl fmt::Display for FixupError {
498 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
499 match self.unresolved {
500 TyOrConstInferVar::TyInt(_) => write!(
501 f,
502 "cannot determine the type of this integer; \
503 add a suffix to specify the type explicitly"
504 ),
505 TyOrConstInferVar::TyFloat(_) => write!(
506 f,
507 "cannot determine the type of this number; \
508 add a suffix to specify the type explicitly"
509 ),
510 TyOrConstInferVar::Ty(_) => write!(f, "unconstrained type"),
511 TyOrConstInferVar::Const(_) => write!(f, "unconstrained const value"),
512 }
513 }
514}
515
516#[derive(Clone, Debug)]
518pub struct TypeOutlivesConstraint<'tcx> {
519 pub sub_region: ty::Region<'tcx>,
520 pub sup_type: Ty<'tcx>,
521 pub origin: SubregionOrigin<'tcx>,
522}
523
524pub struct InferCtxtBuilder<'tcx> {
526 tcx: TyCtxt<'tcx>,
527 considering_regions: bool,
528 in_hir_typeck: bool,
529 skip_leak_check: bool,
530 next_trait_solver: bool,
533}
534
535#[extension(pub trait TyCtxtInferExt<'tcx>)]
536impl<'tcx> TyCtxt<'tcx> {
537 fn infer_ctxt(self) -> InferCtxtBuilder<'tcx> {
538 InferCtxtBuilder {
539 tcx: self,
540 considering_regions: true,
541 in_hir_typeck: false,
542 skip_leak_check: false,
543 next_trait_solver: self.next_trait_solver_globally(),
544 }
545 }
546}
547
548impl<'tcx> InferCtxtBuilder<'tcx> {
549 pub fn with_next_trait_solver(mut self, next_trait_solver: bool) -> Self {
550 self.next_trait_solver = next_trait_solver;
551 self
552 }
553
554 pub fn ignoring_regions(mut self) -> Self {
555 self.considering_regions = false;
556 self
557 }
558
559 pub fn in_hir_typeck(mut self) -> Self {
560 self.in_hir_typeck = true;
561 self
562 }
563
564 pub fn skip_leak_check(mut self, skip_leak_check: bool) -> Self {
565 self.skip_leak_check = skip_leak_check;
566 self
567 }
568
569 pub fn build_with_canonical<T>(
577 mut self,
578 span: Span,
579 input: &CanonicalQueryInput<'tcx, T>,
580 ) -> (InferCtxt<'tcx>, T, CanonicalVarValues<'tcx>)
581 where
582 T: TypeFoldable<TyCtxt<'tcx>>,
583 {
584 let infcx = self.build(input.typing_mode);
585 let (value, args) = infcx.instantiate_canonical(span, &input.canonical);
586 (infcx, value, args)
587 }
588
589 pub fn build_with_typing_env(
590 mut self,
591 TypingEnv { typing_mode, param_env }: TypingEnv<'tcx>,
592 ) -> (InferCtxt<'tcx>, ty::ParamEnv<'tcx>) {
593 (self.build(typing_mode), param_env)
594 }
595
596 pub fn build(&mut self, typing_mode: TypingMode<'tcx>) -> InferCtxt<'tcx> {
597 let InferCtxtBuilder {
598 tcx,
599 considering_regions,
600 in_hir_typeck,
601 skip_leak_check,
602 next_trait_solver,
603 } = *self;
604 InferCtxt {
605 tcx,
606 typing_mode,
607 considering_regions,
608 in_hir_typeck,
609 skip_leak_check,
610 inner: RefCell::new(InferCtxtInner::new()),
611 lexical_region_resolutions: RefCell::new(None),
612 selection_cache: Default::default(),
613 evaluation_cache: Default::default(),
614 reported_trait_errors: Default::default(),
615 reported_signature_mismatch: Default::default(),
616 tainted_by_errors: Cell::new(None),
617 universe: Cell::new(ty::UniverseIndex::ROOT),
618 next_trait_solver,
619 obligation_inspector: Cell::new(None),
620 }
621 }
622}
623
624impl<'tcx, T> InferOk<'tcx, T> {
625 pub fn into_value_registering_obligations<E: 'tcx>(
627 self,
628 infcx: &InferCtxt<'tcx>,
629 fulfill_cx: &mut dyn TraitEngine<'tcx, E>,
630 ) -> T {
631 let InferOk { value, obligations } = self;
632 fulfill_cx.register_predicate_obligations(infcx, obligations);
633 value
634 }
635}
636
637impl<'tcx> InferOk<'tcx, ()> {
638 pub fn into_obligations(self) -> PredicateObligations<'tcx> {
639 self.obligations
640 }
641}
642
643impl<'tcx> InferCtxt<'tcx> {
644 pub fn dcx(&self) -> DiagCtxtHandle<'_> {
645 self.tcx.dcx().taintable_handle(&self.tainted_by_errors)
646 }
647
648 pub fn next_trait_solver(&self) -> bool {
649 self.next_trait_solver
650 }
651
652 #[inline(always)]
653 pub fn typing_mode(&self) -> TypingMode<'tcx> {
654 self.typing_mode
655 }
656
657 pub fn freshen<T: TypeFoldable<TyCtxt<'tcx>>>(&self, t: T) -> T {
658 t.fold_with(&mut self.freshener())
659 }
660
661 pub fn type_var_origin(&self, vid: TyVid) -> TypeVariableOrigin {
665 self.inner.borrow_mut().type_variables().var_origin(vid)
666 }
667
668 pub fn const_var_origin(&self, vid: ConstVid) -> Option<ConstVariableOrigin> {
672 match self.inner.borrow_mut().const_unification_table().probe_value(vid) {
673 ConstVariableValue::Known { .. } => None,
674 ConstVariableValue::Unknown { origin, .. } => Some(origin),
675 }
676 }
677
678 pub fn freshener<'b>(&'b self) -> TypeFreshener<'b, 'tcx> {
679 freshen::TypeFreshener::new(self)
680 }
681
682 pub fn unresolved_variables(&self) -> Vec<Ty<'tcx>> {
683 let mut inner = self.inner.borrow_mut();
684 let mut vars: Vec<Ty<'_>> = inner
685 .type_variables()
686 .unresolved_variables()
687 .into_iter()
688 .map(|t| Ty::new_var(self.tcx, t))
689 .collect();
690 vars.extend(
691 (0..inner.int_unification_table().len())
692 .map(|i| ty::IntVid::from_usize(i))
693 .filter(|&vid| inner.int_unification_table().probe_value(vid).is_unknown())
694 .map(|v| Ty::new_int_var(self.tcx, v)),
695 );
696 vars.extend(
697 (0..inner.float_unification_table().len())
698 .map(|i| ty::FloatVid::from_usize(i))
699 .filter(|&vid| inner.float_unification_table().probe_value(vid).is_unknown())
700 .map(|v| Ty::new_float_var(self.tcx, v)),
701 );
702 vars
703 }
704
705 #[instrument(skip(self), level = "debug")]
706 pub fn sub_regions(
707 &self,
708 origin: SubregionOrigin<'tcx>,
709 a: ty::Region<'tcx>,
710 b: ty::Region<'tcx>,
711 ) {
712 self.inner.borrow_mut().unwrap_region_constraints().make_subregion(origin, a, b);
713 }
714
715 pub fn coerce_predicate(
731 &self,
732 cause: &ObligationCause<'tcx>,
733 param_env: ty::ParamEnv<'tcx>,
734 predicate: ty::PolyCoercePredicate<'tcx>,
735 ) -> Result<InferResult<'tcx, ()>, (TyVid, TyVid)> {
736 let subtype_predicate = predicate.map_bound(|p| ty::SubtypePredicate {
737 a_is_expected: false, a: p.a,
739 b: p.b,
740 });
741 self.subtype_predicate(cause, param_env, subtype_predicate)
742 }
743
744 pub fn subtype_predicate(
745 &self,
746 cause: &ObligationCause<'tcx>,
747 param_env: ty::ParamEnv<'tcx>,
748 predicate: ty::PolySubtypePredicate<'tcx>,
749 ) -> Result<InferResult<'tcx, ()>, (TyVid, TyVid)> {
750 let r_a = self.shallow_resolve(predicate.skip_binder().a);
764 let r_b = self.shallow_resolve(predicate.skip_binder().b);
765 match (r_a.kind(), r_b.kind()) {
766 (&ty::Infer(ty::TyVar(a_vid)), &ty::Infer(ty::TyVar(b_vid))) => {
767 self.sub_unify_ty_vids_raw(a_vid, b_vid);
768 return Err((a_vid, b_vid));
769 }
770 _ => {}
771 }
772
773 self.enter_forall(predicate, |ty::SubtypePredicate { a_is_expected, a, b }| {
774 if a_is_expected {
775 Ok(self.at(cause, param_env).sub(DefineOpaqueTypes::Yes, a, b))
776 } else {
777 Ok(self.at(cause, param_env).sup(DefineOpaqueTypes::Yes, b, a))
778 }
779 })
780 }
781
782 pub fn num_ty_vars(&self) -> usize {
784 self.inner.borrow_mut().type_variables().num_vars()
785 }
786
787 pub fn next_ty_vid(&self, span: Span) -> TyVid {
788 self.next_ty_vid_with_origin(TypeVariableOrigin { span, param_def_id: None })
789 }
790
791 pub fn next_ty_vid_with_origin(&self, origin: TypeVariableOrigin) -> TyVid {
792 self.inner.borrow_mut().type_variables().new_var(self.universe(), origin)
793 }
794
795 pub fn next_ty_vid_in_universe(&self, span: Span, universe: ty::UniverseIndex) -> TyVid {
796 let origin = TypeVariableOrigin { span, param_def_id: None };
797 self.inner.borrow_mut().type_variables().new_var(universe, origin)
798 }
799
800 pub fn next_ty_var(&self, span: Span) -> Ty<'tcx> {
801 self.next_ty_var_with_origin(TypeVariableOrigin { span, param_def_id: None })
802 }
803
804 pub fn next_ty_var_with_origin(&self, origin: TypeVariableOrigin) -> Ty<'tcx> {
805 let vid = self.next_ty_vid_with_origin(origin);
806 Ty::new_var(self.tcx, vid)
807 }
808
809 pub fn next_ty_var_in_universe(&self, span: Span, universe: ty::UniverseIndex) -> Ty<'tcx> {
810 let vid = self.next_ty_vid_in_universe(span, universe);
811 Ty::new_var(self.tcx, vid)
812 }
813
814 pub fn next_const_var(&self, span: Span) -> ty::Const<'tcx> {
815 self.next_const_var_with_origin(ConstVariableOrigin { span, param_def_id: None })
816 }
817
818 pub fn next_const_var_with_origin(&self, origin: ConstVariableOrigin) -> ty::Const<'tcx> {
819 let vid = self
820 .inner
821 .borrow_mut()
822 .const_unification_table()
823 .new_key(ConstVariableValue::Unknown { origin, universe: self.universe() })
824 .vid;
825 ty::Const::new_var(self.tcx, vid)
826 }
827
828 pub fn next_const_var_in_universe(
829 &self,
830 span: Span,
831 universe: ty::UniverseIndex,
832 ) -> ty::Const<'tcx> {
833 let origin = ConstVariableOrigin { span, param_def_id: None };
834 let vid = self
835 .inner
836 .borrow_mut()
837 .const_unification_table()
838 .new_key(ConstVariableValue::Unknown { origin, universe })
839 .vid;
840 ty::Const::new_var(self.tcx, vid)
841 }
842
843 pub fn next_int_var(&self) -> Ty<'tcx> {
844 let next_int_var_id =
845 self.inner.borrow_mut().int_unification_table().new_key(ty::IntVarValue::Unknown);
846 Ty::new_int_var(self.tcx, next_int_var_id)
847 }
848
849 pub fn next_float_var(&self) -> Ty<'tcx> {
850 let next_float_var_id =
851 self.inner.borrow_mut().float_unification_table().new_key(ty::FloatVarValue::Unknown);
852 Ty::new_float_var(self.tcx, next_float_var_id)
853 }
854
855 pub fn next_region_var(&self, origin: RegionVariableOrigin) -> ty::Region<'tcx> {
859 self.next_region_var_in_universe(origin, self.universe())
860 }
861
862 pub fn next_region_var_in_universe(
866 &self,
867 origin: RegionVariableOrigin,
868 universe: ty::UniverseIndex,
869 ) -> ty::Region<'tcx> {
870 let region_var =
871 self.inner.borrow_mut().unwrap_region_constraints().new_region_var(universe, origin);
872 ty::Region::new_var(self.tcx, region_var)
873 }
874
875 pub fn next_term_var_of_kind(&self, term: ty::Term<'tcx>, span: Span) -> ty::Term<'tcx> {
876 match term.kind() {
877 ty::TermKind::Ty(_) => self.next_ty_var(span).into(),
878 ty::TermKind::Const(_) => self.next_const_var(span).into(),
879 }
880 }
881
882 pub fn universe_of_region(&self, r: ty::Region<'tcx>) -> ty::UniverseIndex {
888 self.inner.borrow_mut().unwrap_region_constraints().universe(r)
889 }
890
891 pub fn num_region_vars(&self) -> usize {
893 self.inner.borrow_mut().unwrap_region_constraints().num_region_vars()
894 }
895
896 #[instrument(skip(self), level = "debug")]
898 pub fn next_nll_region_var(&self, origin: NllRegionVariableOrigin) -> ty::Region<'tcx> {
899 self.next_region_var(RegionVariableOrigin::Nll(origin))
900 }
901
902 #[instrument(skip(self), level = "debug")]
904 pub fn next_nll_region_var_in_universe(
905 &self,
906 origin: NllRegionVariableOrigin,
907 universe: ty::UniverseIndex,
908 ) -> ty::Region<'tcx> {
909 self.next_region_var_in_universe(RegionVariableOrigin::Nll(origin), universe)
910 }
911
912 pub fn var_for_def(&self, span: Span, param: &ty::GenericParamDef) -> GenericArg<'tcx> {
913 match param.kind {
914 GenericParamDefKind::Lifetime => {
915 self.next_region_var(RegionVariableOrigin::RegionParameterDefinition(
918 span, param.name,
919 ))
920 .into()
921 }
922 GenericParamDefKind::Type { .. } => {
923 let ty_var_id = self.inner.borrow_mut().type_variables().new_var(
932 self.universe(),
933 TypeVariableOrigin { param_def_id: Some(param.def_id), span },
934 );
935
936 Ty::new_var(self.tcx, ty_var_id).into()
937 }
938 GenericParamDefKind::Const { .. } => {
939 let origin = ConstVariableOrigin { param_def_id: Some(param.def_id), span };
940 let const_var_id = self
941 .inner
942 .borrow_mut()
943 .const_unification_table()
944 .new_key(ConstVariableValue::Unknown { origin, universe: self.universe() })
945 .vid;
946 ty::Const::new_var(self.tcx, const_var_id).into()
947 }
948 }
949 }
950
951 pub fn fresh_args_for_item(&self, span: Span, def_id: DefId) -> GenericArgsRef<'tcx> {
954 GenericArgs::for_item(self.tcx, def_id, |param, _| self.var_for_def(span, param))
955 }
956
957 #[must_use = "this method does not have any side effects"]
963 pub fn tainted_by_errors(&self) -> Option<ErrorGuaranteed> {
964 self.tainted_by_errors.get()
965 }
966
967 pub fn set_tainted_by_errors(&self, e: ErrorGuaranteed) {
970 debug!("set_tainted_by_errors(ErrorGuaranteed)");
971 self.tainted_by_errors.set(Some(e));
972 }
973
974 pub fn region_var_origin(&self, vid: ty::RegionVid) -> RegionVariableOrigin {
975 let mut inner = self.inner.borrow_mut();
976 let inner = &mut *inner;
977 inner.unwrap_region_constraints().var_origin(vid)
978 }
979
980 pub fn get_region_var_infos(&self) -> VarInfos {
983 let inner = self.inner.borrow();
984 assert!(!UndoLogs::<UndoLog<'_>>::in_snapshot(&inner.undo_log));
985 let storage = inner.region_constraint_storage.as_ref().expect("regions already resolved");
986 assert!(storage.data.is_empty(), "{:#?}", storage.data);
987 storage.var_infos.clone()
991 }
992
993 pub fn has_opaque_types_in_storage(&self) -> bool {
994 !self.inner.borrow().opaque_type_storage.is_empty()
995 }
996
997 #[instrument(level = "debug", skip(self), ret)]
998 pub fn take_opaque_types(&self) -> Vec<(OpaqueTypeKey<'tcx>, OpaqueHiddenType<'tcx>)> {
999 self.inner.borrow_mut().opaque_type_storage.take_opaque_types().collect()
1000 }
1001
1002 #[instrument(level = "debug", skip(self), ret)]
1003 pub fn clone_opaque_types(&self) -> Vec<(OpaqueTypeKey<'tcx>, OpaqueHiddenType<'tcx>)> {
1004 self.inner.borrow_mut().opaque_type_storage.iter_opaque_types().collect()
1005 }
1006
1007 #[inline(always)]
1008 pub fn can_define_opaque_ty(&self, id: impl Into<DefId>) -> bool {
1009 debug_assert!(!self.next_trait_solver());
1010 match self.typing_mode() {
1011 TypingMode::Analysis {
1012 defining_opaque_types_and_generators: defining_opaque_types,
1013 }
1014 | TypingMode::Borrowck { defining_opaque_types } => {
1015 id.into().as_local().is_some_and(|def_id| defining_opaque_types.contains(&def_id))
1016 }
1017 TypingMode::Coherence
1021 | TypingMode::PostBorrowckAnalysis { .. }
1022 | TypingMode::PostAnalysis => false,
1023 }
1024 }
1025
1026 pub fn push_hir_typeck_potentially_region_dependent_goal(
1027 &self,
1028 goal: PredicateObligation<'tcx>,
1029 ) {
1030 let mut inner = self.inner.borrow_mut();
1031 inner.undo_log.push(UndoLog::PushHirTypeckPotentiallyRegionDependentGoal);
1032 inner.hir_typeck_potentially_region_dependent_goals.push(goal);
1033 }
1034
1035 pub fn take_hir_typeck_potentially_region_dependent_goals(
1036 &self,
1037 ) -> Vec<PredicateObligation<'tcx>> {
1038 assert!(!self.in_snapshot(), "cannot take goals in a snapshot");
1039 std::mem::take(&mut self.inner.borrow_mut().hir_typeck_potentially_region_dependent_goals)
1040 }
1041
1042 pub fn ty_to_string(&self, t: Ty<'tcx>) -> String {
1043 self.resolve_vars_if_possible(t).to_string()
1044 }
1045
1046 pub fn probe_ty_var(&self, vid: TyVid) -> Result<Ty<'tcx>, ty::UniverseIndex> {
1049 use self::type_variable::TypeVariableValue;
1050
1051 match self.inner.borrow_mut().type_variables().probe(vid) {
1052 TypeVariableValue::Known { value } => Ok(value),
1053 TypeVariableValue::Unknown { universe } => Err(universe),
1054 }
1055 }
1056
1057 pub fn shallow_resolve(&self, ty: Ty<'tcx>) -> Ty<'tcx> {
1058 if let ty::Infer(v) = *ty.kind() {
1059 match v {
1060 ty::TyVar(v) => {
1061 let known = self.inner.borrow_mut().type_variables().probe(v).known();
1074 known.map_or(ty, |t| self.shallow_resolve(t))
1075 }
1076
1077 ty::IntVar(v) => {
1078 match self.inner.borrow_mut().int_unification_table().probe_value(v) {
1079 ty::IntVarValue::IntType(ty) => Ty::new_int(self.tcx, ty),
1080 ty::IntVarValue::UintType(ty) => Ty::new_uint(self.tcx, ty),
1081 ty::IntVarValue::Unknown => ty,
1082 }
1083 }
1084
1085 ty::FloatVar(v) => {
1086 match self.inner.borrow_mut().float_unification_table().probe_value(v) {
1087 ty::FloatVarValue::Known(ty) => Ty::new_float(self.tcx, ty),
1088 ty::FloatVarValue::Unknown => ty,
1089 }
1090 }
1091
1092 ty::FreshTy(_) | ty::FreshIntTy(_) | ty::FreshFloatTy(_) => ty,
1093 }
1094 } else {
1095 ty
1096 }
1097 }
1098
1099 pub fn shallow_resolve_const(&self, ct: ty::Const<'tcx>) -> ty::Const<'tcx> {
1100 match ct.kind() {
1101 ty::ConstKind::Infer(infer_ct) => match infer_ct {
1102 InferConst::Var(vid) => self
1103 .inner
1104 .borrow_mut()
1105 .const_unification_table()
1106 .probe_value(vid)
1107 .known()
1108 .unwrap_or(ct),
1109 InferConst::Fresh(_) => ct,
1110 },
1111 ty::ConstKind::Param(_)
1112 | ty::ConstKind::Bound(_, _)
1113 | ty::ConstKind::Placeholder(_)
1114 | ty::ConstKind::Unevaluated(_)
1115 | ty::ConstKind::Value(_)
1116 | ty::ConstKind::Error(_)
1117 | ty::ConstKind::Expr(_) => ct,
1118 }
1119 }
1120
1121 pub fn shallow_resolve_term(&self, term: ty::Term<'tcx>) -> ty::Term<'tcx> {
1122 match term.kind() {
1123 ty::TermKind::Ty(ty) => self.shallow_resolve(ty).into(),
1124 ty::TermKind::Const(ct) => self.shallow_resolve_const(ct).into(),
1125 }
1126 }
1127
1128 pub fn root_var(&self, var: ty::TyVid) -> ty::TyVid {
1129 self.inner.borrow_mut().type_variables().root_var(var)
1130 }
1131
1132 pub fn sub_unify_ty_vids_raw(&self, a: ty::TyVid, b: ty::TyVid) {
1133 self.inner.borrow_mut().type_variables().sub_unify(a, b);
1134 }
1135
1136 pub fn sub_unification_table_root_var(&self, var: ty::TyVid) -> ty::TyVid {
1137 self.inner.borrow_mut().type_variables().sub_unification_table_root_var(var)
1138 }
1139
1140 pub fn root_const_var(&self, var: ty::ConstVid) -> ty::ConstVid {
1141 self.inner.borrow_mut().const_unification_table().find(var).vid
1142 }
1143
1144 pub fn opportunistic_resolve_int_var(&self, vid: ty::IntVid) -> Ty<'tcx> {
1147 let mut inner = self.inner.borrow_mut();
1148 let value = inner.int_unification_table().probe_value(vid);
1149 match value {
1150 ty::IntVarValue::IntType(ty) => Ty::new_int(self.tcx, ty),
1151 ty::IntVarValue::UintType(ty) => Ty::new_uint(self.tcx, ty),
1152 ty::IntVarValue::Unknown => {
1153 Ty::new_int_var(self.tcx, inner.int_unification_table().find(vid))
1154 }
1155 }
1156 }
1157
1158 pub fn opportunistic_resolve_float_var(&self, vid: ty::FloatVid) -> Ty<'tcx> {
1161 let mut inner = self.inner.borrow_mut();
1162 let value = inner.float_unification_table().probe_value(vid);
1163 match value {
1164 ty::FloatVarValue::Known(ty) => Ty::new_float(self.tcx, ty),
1165 ty::FloatVarValue::Unknown => {
1166 Ty::new_float_var(self.tcx, inner.float_unification_table().find(vid))
1167 }
1168 }
1169 }
1170
1171 pub fn resolve_vars_if_possible<T>(&self, value: T) -> T
1178 where
1179 T: TypeFoldable<TyCtxt<'tcx>>,
1180 {
1181 if let Err(guar) = value.error_reported() {
1182 self.set_tainted_by_errors(guar);
1183 }
1184 if !value.has_non_region_infer() {
1185 return value;
1186 }
1187 let mut r = resolve::OpportunisticVarResolver::new(self);
1188 value.fold_with(&mut r)
1189 }
1190
1191 pub fn resolve_numeric_literals_with_default<T>(&self, value: T) -> T
1192 where
1193 T: TypeFoldable<TyCtxt<'tcx>>,
1194 {
1195 if !value.has_infer() {
1196 return value; }
1198 let mut r = InferenceLiteralEraser { tcx: self.tcx };
1199 value.fold_with(&mut r)
1200 }
1201
1202 pub fn probe_const_var(&self, vid: ty::ConstVid) -> Result<ty::Const<'tcx>, ty::UniverseIndex> {
1203 match self.inner.borrow_mut().const_unification_table().probe_value(vid) {
1204 ConstVariableValue::Known { value } => Ok(value),
1205 ConstVariableValue::Unknown { origin: _, universe } => Err(universe),
1206 }
1207 }
1208
1209 pub fn fully_resolve<T: TypeFoldable<TyCtxt<'tcx>>>(&self, value: T) -> FixupResult<T> {
1217 match resolve::fully_resolve(self, value) {
1218 Ok(value) => {
1219 if value.has_non_region_infer() {
1220 bug!("`{value:?}` is not fully resolved");
1221 }
1222 if value.has_infer_regions() {
1223 let guar = self.dcx().delayed_bug(format!("`{value:?}` is not fully resolved"));
1224 Ok(fold_regions(self.tcx, value, |re, _| {
1225 if re.is_var() { ty::Region::new_error(self.tcx, guar) } else { re }
1226 }))
1227 } else {
1228 Ok(value)
1229 }
1230 }
1231 Err(e) => Err(e),
1232 }
1233 }
1234
1235 pub fn instantiate_binder_with_fresh_vars<T>(
1243 &self,
1244 span: Span,
1245 lbrct: BoundRegionConversionTime,
1246 value: ty::Binder<'tcx, T>,
1247 ) -> T
1248 where
1249 T: TypeFoldable<TyCtxt<'tcx>> + Copy,
1250 {
1251 if let Some(inner) = value.no_bound_vars() {
1252 return inner;
1253 }
1254
1255 let bound_vars = value.bound_vars();
1256 let mut args = Vec::with_capacity(bound_vars.len());
1257
1258 for bound_var_kind in bound_vars {
1259 let arg: ty::GenericArg<'_> = match bound_var_kind {
1260 ty::BoundVariableKind::Ty(_) => self.next_ty_var(span).into(),
1261 ty::BoundVariableKind::Region(br) => {
1262 self.next_region_var(RegionVariableOrigin::BoundRegion(span, br, lbrct)).into()
1263 }
1264 ty::BoundVariableKind::Const => self.next_const_var(span).into(),
1265 };
1266 args.push(arg);
1267 }
1268
1269 struct ToFreshVars<'tcx> {
1270 args: Vec<ty::GenericArg<'tcx>>,
1271 }
1272
1273 impl<'tcx> BoundVarReplacerDelegate<'tcx> for ToFreshVars<'tcx> {
1274 fn replace_region(&mut self, br: ty::BoundRegion) -> ty::Region<'tcx> {
1275 self.args[br.var.index()].expect_region()
1276 }
1277 fn replace_ty(&mut self, bt: ty::BoundTy) -> Ty<'tcx> {
1278 self.args[bt.var.index()].expect_ty()
1279 }
1280 fn replace_const(&mut self, bc: ty::BoundConst) -> ty::Const<'tcx> {
1281 self.args[bc.var.index()].expect_const()
1282 }
1283 }
1284 let delegate = ToFreshVars { args };
1285 self.tcx.replace_bound_vars_uncached(value, delegate)
1286 }
1287
1288 pub(crate) fn verify_generic_bound(
1290 &self,
1291 origin: SubregionOrigin<'tcx>,
1292 kind: GenericKind<'tcx>,
1293 a: ty::Region<'tcx>,
1294 bound: VerifyBound<'tcx>,
1295 ) {
1296 debug!("verify_generic_bound({:?}, {:?} <: {:?})", kind, a, bound);
1297
1298 self.inner
1299 .borrow_mut()
1300 .unwrap_region_constraints()
1301 .verify_generic_bound(origin, kind, a, bound);
1302 }
1303
1304 pub fn closure_kind(&self, closure_ty: Ty<'tcx>) -> Option<ty::ClosureKind> {
1308 let unresolved_kind_ty = match *closure_ty.kind() {
1309 ty::Closure(_, args) => args.as_closure().kind_ty(),
1310 ty::CoroutineClosure(_, args) => args.as_coroutine_closure().kind_ty(),
1311 _ => bug!("unexpected type {closure_ty}"),
1312 };
1313 let closure_kind_ty = self.shallow_resolve(unresolved_kind_ty);
1314 closure_kind_ty.to_opt_closure_kind()
1315 }
1316
1317 pub fn universe(&self) -> ty::UniverseIndex {
1318 self.universe.get()
1319 }
1320
1321 pub fn create_next_universe(&self) -> ty::UniverseIndex {
1324 let u = self.universe.get().next_universe();
1325 debug!("create_next_universe {u:?}");
1326 self.universe.set(u);
1327 u
1328 }
1329
1330 pub fn typing_env(&self, param_env: ty::ParamEnv<'tcx>) -> ty::TypingEnv<'tcx> {
1334 let typing_mode = match self.typing_mode() {
1335 ty::TypingMode::Analysis { defining_opaque_types_and_generators: _ }
1340 | ty::TypingMode::Borrowck { defining_opaque_types: _ } => {
1341 TypingMode::non_body_analysis()
1342 }
1343 mode @ (ty::TypingMode::Coherence
1344 | ty::TypingMode::PostBorrowckAnalysis { .. }
1345 | ty::TypingMode::PostAnalysis) => mode,
1346 };
1347 ty::TypingEnv { typing_mode, param_env }
1348 }
1349
1350 pub fn pseudo_canonicalize_query<V>(
1354 &self,
1355 param_env: ty::ParamEnv<'tcx>,
1356 value: V,
1357 ) -> PseudoCanonicalInput<'tcx, V>
1358 where
1359 V: TypeVisitable<TyCtxt<'tcx>>,
1360 {
1361 debug_assert!(!value.has_infer());
1362 debug_assert!(!value.has_placeholders());
1363 debug_assert!(!param_env.has_infer());
1364 debug_assert!(!param_env.has_placeholders());
1365 self.typing_env(param_env).as_query_input(value)
1366 }
1367
1368 #[inline]
1371 pub fn is_ty_infer_var_definitely_unchanged(&self) -> impl Fn(TyOrConstInferVar) -> bool {
1372 let inner = self.inner.try_borrow();
1374
1375 move |infer_var: TyOrConstInferVar| match (infer_var, &inner) {
1376 (TyOrConstInferVar::Ty(ty_var), Ok(inner)) => {
1377 use self::type_variable::TypeVariableValue;
1378
1379 matches!(
1380 inner.try_type_variables_probe_ref(ty_var),
1381 Some(TypeVariableValue::Unknown { .. })
1382 )
1383 }
1384 _ => false,
1385 }
1386 }
1387
1388 #[inline(always)]
1398 pub fn ty_or_const_infer_var_changed(&self, infer_var: TyOrConstInferVar) -> bool {
1399 match infer_var {
1400 TyOrConstInferVar::Ty(v) => {
1401 use self::type_variable::TypeVariableValue;
1402
1403 match self.inner.borrow_mut().type_variables().inlined_probe(v) {
1406 TypeVariableValue::Unknown { .. } => false,
1407 TypeVariableValue::Known { .. } => true,
1408 }
1409 }
1410
1411 TyOrConstInferVar::TyInt(v) => {
1412 self.inner.borrow_mut().int_unification_table().inlined_probe_value(v).is_known()
1416 }
1417
1418 TyOrConstInferVar::TyFloat(v) => {
1419 self.inner.borrow_mut().float_unification_table().probe_value(v).is_known()
1424 }
1425
1426 TyOrConstInferVar::Const(v) => {
1427 match self.inner.borrow_mut().const_unification_table().probe_value(v) {
1432 ConstVariableValue::Unknown { .. } => false,
1433 ConstVariableValue::Known { .. } => true,
1434 }
1435 }
1436 }
1437 }
1438
1439 pub fn attach_obligation_inspector(&self, inspector: ObligationInspector<'tcx>) {
1441 debug_assert!(
1442 self.obligation_inspector.get().is_none(),
1443 "shouldn't override a set obligation inspector"
1444 );
1445 self.obligation_inspector.set(Some(inspector));
1446 }
1447}
1448
1449#[derive(Copy, Clone, Debug)]
1452pub enum TyOrConstInferVar {
1453 Ty(TyVid),
1455 TyInt(IntVid),
1457 TyFloat(FloatVid),
1459
1460 Const(ConstVid),
1462}
1463
1464impl<'tcx> TyOrConstInferVar {
1465 pub fn maybe_from_generic_arg(arg: GenericArg<'tcx>) -> Option<Self> {
1469 match arg.kind() {
1470 GenericArgKind::Type(ty) => Self::maybe_from_ty(ty),
1471 GenericArgKind::Const(ct) => Self::maybe_from_const(ct),
1472 GenericArgKind::Lifetime(_) => None,
1473 }
1474 }
1475
1476 pub fn maybe_from_term(term: Term<'tcx>) -> Option<Self> {
1480 match term.kind() {
1481 TermKind::Ty(ty) => Self::maybe_from_ty(ty),
1482 TermKind::Const(ct) => Self::maybe_from_const(ct),
1483 }
1484 }
1485
1486 fn maybe_from_ty(ty: Ty<'tcx>) -> Option<Self> {
1489 match *ty.kind() {
1490 ty::Infer(ty::TyVar(v)) => Some(TyOrConstInferVar::Ty(v)),
1491 ty::Infer(ty::IntVar(v)) => Some(TyOrConstInferVar::TyInt(v)),
1492 ty::Infer(ty::FloatVar(v)) => Some(TyOrConstInferVar::TyFloat(v)),
1493 _ => None,
1494 }
1495 }
1496
1497 fn maybe_from_const(ct: ty::Const<'tcx>) -> Option<Self> {
1500 match ct.kind() {
1501 ty::ConstKind::Infer(InferConst::Var(v)) => Some(TyOrConstInferVar::Const(v)),
1502 _ => None,
1503 }
1504 }
1505}
1506
1507struct InferenceLiteralEraser<'tcx> {
1510 tcx: TyCtxt<'tcx>,
1511}
1512
1513impl<'tcx> TypeFolder<TyCtxt<'tcx>> for InferenceLiteralEraser<'tcx> {
1514 fn cx(&self) -> TyCtxt<'tcx> {
1515 self.tcx
1516 }
1517
1518 fn fold_ty(&mut self, ty: Ty<'tcx>) -> Ty<'tcx> {
1519 match ty.kind() {
1520 ty::Infer(ty::IntVar(_) | ty::FreshIntTy(_)) => self.tcx.types.i32,
1521 ty::Infer(ty::FloatVar(_) | ty::FreshFloatTy(_)) => self.tcx.types.f64,
1522 _ => ty.super_fold_with(self),
1523 }
1524 }
1525}
1526
1527impl<'tcx> TypeTrace<'tcx> {
1528 pub fn span(&self) -> Span {
1529 self.cause.span
1530 }
1531
1532 pub fn types(cause: &ObligationCause<'tcx>, a: Ty<'tcx>, b: Ty<'tcx>) -> TypeTrace<'tcx> {
1533 TypeTrace {
1534 cause: cause.clone(),
1535 values: ValuePairs::Terms(ExpectedFound::new(a.into(), b.into())),
1536 }
1537 }
1538
1539 pub fn trait_refs(
1540 cause: &ObligationCause<'tcx>,
1541 a: ty::TraitRef<'tcx>,
1542 b: ty::TraitRef<'tcx>,
1543 ) -> TypeTrace<'tcx> {
1544 TypeTrace { cause: cause.clone(), values: ValuePairs::TraitRefs(ExpectedFound::new(a, b)) }
1545 }
1546
1547 pub fn consts(
1548 cause: &ObligationCause<'tcx>,
1549 a: ty::Const<'tcx>,
1550 b: ty::Const<'tcx>,
1551 ) -> TypeTrace<'tcx> {
1552 TypeTrace {
1553 cause: cause.clone(),
1554 values: ValuePairs::Terms(ExpectedFound::new(a.into(), b.into())),
1555 }
1556 }
1557}
1558
1559impl<'tcx> SubregionOrigin<'tcx> {
1560 pub fn span(&self) -> Span {
1561 match *self {
1562 SubregionOrigin::Subtype(ref a) => a.span(),
1563 SubregionOrigin::RelateObjectBound(a) => a,
1564 SubregionOrigin::RelateParamBound(a, ..) => a,
1565 SubregionOrigin::RelateRegionParamBound(a, _) => a,
1566 SubregionOrigin::Reborrow(a) => a,
1567 SubregionOrigin::ReferenceOutlivesReferent(_, a) => a,
1568 SubregionOrigin::CompareImplItemObligation { span, .. } => span,
1569 SubregionOrigin::AscribeUserTypeProvePredicate(span) => span,
1570 SubregionOrigin::CheckAssociatedTypeBounds { ref parent, .. } => parent.span(),
1571 }
1572 }
1573
1574 pub fn from_obligation_cause<F>(cause: &traits::ObligationCause<'tcx>, default: F) -> Self
1575 where
1576 F: FnOnce() -> Self,
1577 {
1578 match *cause.code() {
1579 traits::ObligationCauseCode::ReferenceOutlivesReferent(ref_type) => {
1580 SubregionOrigin::ReferenceOutlivesReferent(ref_type, cause.span)
1581 }
1582
1583 traits::ObligationCauseCode::CompareImplItem {
1584 impl_item_def_id,
1585 trait_item_def_id,
1586 kind: _,
1587 } => SubregionOrigin::CompareImplItemObligation {
1588 span: cause.span,
1589 impl_item_def_id,
1590 trait_item_def_id,
1591 },
1592
1593 traits::ObligationCauseCode::CheckAssociatedTypeBounds {
1594 impl_item_def_id,
1595 trait_item_def_id,
1596 } => SubregionOrigin::CheckAssociatedTypeBounds {
1597 impl_item_def_id,
1598 trait_item_def_id,
1599 parent: Box::new(default()),
1600 },
1601
1602 traits::ObligationCauseCode::AscribeUserTypeProvePredicate(span) => {
1603 SubregionOrigin::AscribeUserTypeProvePredicate(span)
1604 }
1605
1606 traits::ObligationCauseCode::ObjectTypeBound(ty, _reg) => {
1607 SubregionOrigin::RelateRegionParamBound(cause.span, Some(ty))
1608 }
1609
1610 _ => default(),
1611 }
1612 }
1613}
1614
1615impl RegionVariableOrigin {
1616 pub fn span(&self) -> Span {
1617 match *self {
1618 RegionVariableOrigin::Misc(a)
1619 | RegionVariableOrigin::PatternRegion(a)
1620 | RegionVariableOrigin::BorrowRegion(a)
1621 | RegionVariableOrigin::Autoref(a)
1622 | RegionVariableOrigin::Coercion(a)
1623 | RegionVariableOrigin::RegionParameterDefinition(a, ..)
1624 | RegionVariableOrigin::BoundRegion(a, ..)
1625 | RegionVariableOrigin::UpvarRegion(_, a) => a,
1626 RegionVariableOrigin::Nll(..) => bug!("NLL variable used with `span`"),
1627 }
1628 }
1629}
1630
1631impl<'tcx> InferCtxt<'tcx> {
1632 pub fn find_block_span(&self, block: &'tcx hir::Block<'tcx>) -> Span {
1635 let block = block.innermost_block();
1636 if let Some(expr) = &block.expr {
1637 expr.span
1638 } else if let Some(stmt) = block.stmts.last() {
1639 stmt.span
1641 } else {
1642 block.span
1644 }
1645 }
1646
1647 pub fn find_block_span_from_hir_id(&self, hir_id: hir::HirId) -> Span {
1650 match self.tcx.hir_node(hir_id) {
1651 hir::Node::Block(blk)
1652 | hir::Node::Expr(&hir::Expr { kind: hir::ExprKind::Block(blk, _), .. }) => {
1653 self.find_block_span(blk)
1654 }
1655 hir::Node::Expr(e) => e.span,
1656 _ => DUMMY_SP,
1657 }
1658 }
1659}