Skip to content

Commit f3652c5

Browse files
compiler-errorscompiler-errors
committed
Intern binders in TyKind
1 parent 531d3d5 commit f3652c5

File tree

18 files changed

+145
-127
lines changed

18 files changed

+145
-127
lines changed

‎compiler/rustc_ast_lowering/src/lib.rs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,8 @@ use rustc_errors::{DiagArgFromDisplay, DiagCtxtHandle, StashKey};
5555
use rustc_hir::def::{DefKind, LifetimeRes, Namespace, PartialRes, PerNS, Res};
5656
use rustc_hir::def_id::{CRATE_DEF_ID, LOCAL_CRATE, LocalDefId};
5757
use rustc_hir::{
58-
self as hir, ConstArg, GenericArg, HirId, ItemLocalMap, LangItem, ParamName, PredicateOrigin,
59-
TraitCandidate,
58+
self as hir, ConstArg, GenericArg, HirId, IsAnonInPath, ItemLocalMap, LangItem, ParamName,
59+
PredicateOrigin, TraitCandidate,
6060
};
6161
use rustc_index::{Idx, IndexSlice, IndexVec};
6262
use rustc_macros::extension;

‎compiler/rustc_borrowck/src/type_check/mod.rs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1641,7 +1641,7 @@ impl<'a, 'tcx> Visitor<'tcx> for TypeChecker<'a, 'tcx> {
16411641
let expected_ty = self.infcx.instantiate_binder_with_fresh_vars(
16421642
self.body().source_info(location).span,
16431643
BoundRegionConversionTime::HigherRankedType,
1644-
binder_ty.into(),
1644+
*binder_ty,
16451645
);
16461646
self.sub_types(
16471647
operand_ty,
@@ -1891,7 +1891,7 @@ impl<'a, 'tcx> Visitor<'tcx> for TypeChecker<'a, 'tcx> {
18911891
let found_ty = self.infcx.instantiate_binder_with_fresh_vars(
18921892
self.body.source_info(location).span,
18931893
BoundRegionConversionTime::HigherRankedType,
1894-
binder_ty.into(),
1894+
*binder_ty,
18951895
);
18961896
self.relate_types(
18971897
ty,

‎compiler/rustc_hir_typeck/src/expr.rs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1690,7 +1690,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
16901690
ty::UnsafeBinder(binder) => self.instantiate_binder_with_fresh_vars(
16911691
inner_expr.span,
16921692
infer::BoundRegionConversionTime::HigherRankedType,
1693-
binder.into(),
1693+
*binder,
16941694
),
16951695
ty::Error(e) => Ty::new_error(self.tcx, e),
16961696
_ => {
@@ -1727,7 +1727,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
17271727
ty::UnsafeBinder(binder) => self.instantiate_binder_with_fresh_vars(
17281728
inner_expr.span,
17291729
infer::BoundRegionConversionTime::HigherRankedType,
1730-
binder.into(),
1730+
*binder,
17311731
),
17321732
ty::Error(e) => Ty::new_error(self.tcx, e),
17331733
_ => {

‎compiler/rustc_middle/src/ty/codec.rs‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -621,3 +621,27 @@ macro_rules! implement_ty_decoder {
621621
}
622622
}
623623
}
624+
625+
impl<'tcx, E: TyEncoder<'tcx>> Encodable<E> for ty::TyBinderRef<'tcx> {
626+
fn encode(&self, encoder: &mut E) {
627+
(**self).encode(encoder)
628+
}
629+
}
630+
631+
impl<'tcx, D: TyDecoder<'tcx>> Decodable<D> for ty::TyBinderRef<'tcx> {
632+
fn decode(decoder: &mut D) -> Self {
633+
decoder.interner().mk_ty_binder(Decodable::decode(decoder))
634+
}
635+
}
636+
637+
impl<'tcx, E: TyEncoder<'tcx>> Encodable<E> for ty::SigBinderRef<'tcx> {
638+
fn encode(&self, encoder: &mut E) {
639+
(**self).encode(encoder)
640+
}
641+
}
642+
643+
impl<'tcx, D: TyDecoder<'tcx>> Decodable<D> for ty::SigBinderRef<'tcx> {
644+
fn decode(decoder: &mut D) -> Self {
645+
decoder.interner().mk_sig_binder(Decodable::decode(decoder))
646+
}
647+
}

‎compiler/rustc_middle/src/ty/context.rs‎

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ use crate::ty::{
8080
self, AdtDef, AdtDefData, AdtKind, Binder, Clause, Clauses, Const, GenericArg, GenericArgs,
8181
GenericArgsRef, GenericParamDefKind, List, ListWithCachedTypeInfo, ParamConst, ParamTy,
8282
Pattern, PatternKind, PolyExistentialPredicate, PolyFnSig, Predicate, PredicateKind,
83-
PredicatePolarity, Region, RegionKind, ReprOptions, TraitObjectVisitor, Ty, TyKind, TyVid,
84-
ValTree, ValTreeKind, Visibility,
83+
PredicatePolarity, Region, RegionKind, ReprOptions, SigBinderRef, TraitObjectVisitor, Ty,
84+
TyBinderRef, TyKind, TyVid, ValTree, ValTreeKind, Visibility,
8585
};
8686

8787
#[allow(rustc::usage_of_ty_tykind)]
@@ -97,6 +97,9 @@ impl<'tcx> Interner for TyCtxt<'tcx> {
9797
type Term = ty::Term<'tcx>;
9898
type BoundVarKinds = &'tcx List<ty::BoundVariableKind>;
9999

100+
type SigBinderRef = ty::SigBinderRef<'tcx>;
101+
type TyBinderRef = ty::TyBinderRef<'tcx>;
102+
100103
type BoundVarKind = ty::BoundVariableKind;
101104
type PredefinedOpaques = solve::PredefinedOpaques<'tcx>;
102105

@@ -813,6 +816,8 @@ pub struct CtxtInterners<'tcx> {
813816
captures: InternedSet<'tcx, List<&'tcx ty::CapturedPlace<'tcx>>>,
814817
offset_of: InternedSet<'tcx, List<(VariantIdx, FieldIdx)>>,
815818
valtree: InternedSet<'tcx, ty::ValTreeKind<'tcx>>,
819+
ty_binder: InternedSet<'tcx, ty::Binder<'tcx, Ty<'tcx>>>,
820+
sig_binder: InternedSet<'tcx, ty::Binder<'tcx, ty::FnSigTys<'tcx>>>,
816821
}
817822

818823
impl<'tcx> CtxtInterners<'tcx> {
@@ -849,6 +854,8 @@ impl<'tcx> CtxtInterners<'tcx> {
849854
captures: InternedSet::with_capacity(N),
850855
offset_of: InternedSet::with_capacity(N),
851856
valtree: InternedSet::with_capacity(N),
857+
ty_binder: InternedSet::with_capacity(N),
858+
sig_binder: InternedSet::with_capacity(N),
852859
}
853860
}
854861

@@ -2331,6 +2338,8 @@ nop_lift! { predicate; Predicate<'a> => Predicate<'tcx> }
23312338
nop_lift! { predicate; Clause<'a> => Clause<'tcx> }
23322339
nop_lift! { layout; Layout<'a> => Layout<'tcx> }
23332340
nop_lift! { valtree; ValTree<'a> => ValTree<'tcx> }
2341+
nop_lift! { ty_binder; TyBinderRef<'a> => TyBinderRef<'tcx> }
2342+
nop_lift! { sig_binder; SigBinderRef<'a> => SigBinderRef<'tcx> }
23342343

23352344
nop_list_lift! { type_lists; Ty<'a> => Ty<'tcx> }
23362345
nop_list_lift! {
@@ -2595,6 +2604,8 @@ direct_interners! {
25952604
ExternalConstraints -> ExternalConstraints<'tcx>,
25962605
predefined_opaques_in_body: pub mk_predefined_opaques_in_body(PredefinedOpaquesData<TyCtxt<'tcx>>):
25972606
PredefinedOpaques -> PredefinedOpaques<'tcx>,
2607+
ty_binder: pub mk_ty_binder(ty::Binder<'tcx, Ty<'tcx>>): TyBinderRef -> TyBinderRef<'tcx>,
2608+
sig_binder: pub mk_sig_binder(ty::Binder<'tcx, ty::FnSigTys<'tcx>>): SigBinderRef -> SigBinderRef<'tcx>,
25982609
}
25992610

26002611
macro_rules! slice_interners {

‎compiler/rustc_middle/src/ty/flags.rs‎

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -192,15 +192,13 @@ impl FlagComputation {
192192
self.add_args(args);
193193
}
194194

195-
&ty::FnPtr(sig_tys, _) => self.bound_computation(sig_tys, |computation, sig_tys| {
195+
&ty::FnPtr(sig_tys, _) => self.bound_computation(*sig_tys, |computation, sig_tys| {
196196
computation.add_tys(sig_tys.inputs_and_output);
197197
}),
198198

199-
&ty::UnsafeBinder(bound_ty) => {
200-
self.bound_computation(bound_ty.into(), |computation, ty| {
201-
computation.add_ty(ty);
202-
})
203-
}
199+
&ty::UnsafeBinder(bound_ty) => self.bound_computation(*bound_ty, |computation, ty| {
200+
computation.add_ty(ty);
201+
}),
204202
}
205203
}
206204

‎compiler/rustc_middle/src/ty/layout.rs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -828,7 +828,7 @@ where
828828
}
829829

830830
ty::UnsafeBinder(bound_ty) => {
831-
let ty = tcx.instantiate_bound_regions_with_erased(bound_ty.into());
831+
let ty = tcx.instantiate_bound_regions_with_erased(*bound_ty);
832832
field_ty_or_layout(TyAndLayout { ty, ..this }, cx, i)
833833
}
834834

‎compiler/rustc_middle/src/ty/mod.rs‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,9 @@ pub use self::region::{
9191
pub use self::rvalue_scopes::RvalueScopes;
9292
pub use self::sty::{
9393
AliasTy, Article, Binder, BoundTy, BoundTyKind, BoundVariableKind, CanonicalPolyFnSig,
94-
CoroutineArgsExt, EarlyBinder, FnSig, InlineConstArgs, InlineConstArgsParts, ParamConst,
95-
ParamTy, PolyFnSig, TyKind, TypeAndMut, TypingMode, UpvarArgs,
94+
CoroutineArgsExt, EarlyBinder, FnSig, FnSigTys, InlineConstArgs, InlineConstArgsParts,
95+
ParamConst, ParamTy, PolyFnSig, SigBinderRef, TyBinderRef, TyKind, TypeAndMut, TypingMode,
96+
UpvarArgs,
9697
};
9798
pub use self::trait_def::TraitDef;
9899
pub use self::typeck_results::{
@@ -2224,6 +2225,6 @@ mod size_asserts {
22242225
use super::*;
22252226
// tidy-alphabetical-start
22262227
static_assert_size!(PredicateKind<'_>, 32);
2227-
static_assert_size!(WithCachedTypeInfo<TyKind<'_>>, 56);
2228+
static_assert_size!(WithCachedTypeInfo<TyKind<'_>>, 48);
22282229
// tidy-alphabetical-end
22292230
}

‎compiler/rustc_middle/src/ty/structural_impls.rs‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -720,3 +720,35 @@ impl<'tcx> TypeFoldable<TyCtxt<'tcx>> for &'tcx ty::List<PlaceElem<'tcx>> {
720720
ty::util::fold_list(self, folder, |tcx, v| tcx.mk_place_elems(v))
721721
}
722722
}
723+
724+
impl<'tcx> TypeVisitable<TyCtxt<'tcx>> for ty::TyBinderRef<'tcx> {
725+
fn visit_with<V: TypeVisitor<TyCtxt<'tcx>>>(&self, visitor: &mut V) -> V::Result {
726+
(**self).visit_with(visitor)
727+
}
728+
}
729+
730+
impl<'tcx> TypeFoldable<TyCtxt<'tcx>> for ty::TyBinderRef<'tcx> {
731+
fn try_fold_with<F: FallibleTypeFolder<TyCtxt<'tcx>>>(
732+
self,
733+
folder: &mut F,
734+
) -> Result<Self, F::Error> {
735+
let folded = (*self).try_fold_with(folder)?;
736+
Ok(if *self == folded { self } else { folder.cx().mk_ty_binder(folded) })
737+
}
738+
}
739+
740+
impl<'tcx> TypeVisitable<TyCtxt<'tcx>> for ty::SigBinderRef<'tcx> {
741+
fn visit_with<V: TypeVisitor<TyCtxt<'tcx>>>(&self, visitor: &mut V) -> V::Result {
742+
(**self).visit_with(visitor)
743+
}
744+
}
745+
746+
impl<'tcx> TypeFoldable<TyCtxt<'tcx>> for ty::SigBinderRef<'tcx> {
747+
fn try_fold_with<F: FallibleTypeFolder<TyCtxt<'tcx>>>(
748+
self,
749+
folder: &mut F,
750+
) -> Result<Self, F::Error> {
751+
let folded = (*self).try_fold_with(folder)?;
752+
Ok(if *self == folded { self } else { folder.cx().mk_sig_binder(folded) })
753+
}
754+
}

‎compiler/rustc_middle/src/ty/sty.rs‎

Lines changed: 29 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
use std::assert_matches::debug_assert_matches;
66
use std::borrow::Cow;
77
use std::iter;
8-
use std::ops::{ControlFlow, Range};
8+
use std::ops::{ControlFlow, Deref, Range};
99

1010
use hir::def::{CtorKind, DefKind};
1111
use rustc_abi::{ExternAbi, FIRST_VARIANT, FieldIdx, VariantIdx};
@@ -37,6 +37,7 @@ pub type FnSig<'tcx> = ir::FnSig<TyCtxt<'tcx>>;
3737
pub type Binder<'tcx, T> = ir::Binder<TyCtxt<'tcx>, T>;
3838
pub type EarlyBinder<'tcx, T> = ir::EarlyBinder<TyCtxt<'tcx>, T>;
3939
pub type TypingMode<'tcx> = ir::TypingMode<TyCtxt<'tcx>>;
40+
pub type FnSigTys<'tcx> = ir::FnSigTys<TyCtxt<'tcx>>;
4041

4142
pub trait Article {
4243
fn article(&self) -> &'static str;
@@ -704,12 +705,12 @@ impl<'tcx> Ty<'tcx> {
704705
#[inline]
705706
pub fn new_fn_ptr(tcx: TyCtxt<'tcx>, fty: PolyFnSig<'tcx>) -> Ty<'tcx> {
706707
let (sig_tys, hdr) = fty.split();
707-
Ty::new(tcx, FnPtr(sig_tys, hdr))
708+
Ty::new(tcx, FnPtr(tcx.mk_sig_binder(sig_tys), hdr))
708709
}
709710

710711
#[inline]
711712
pub fn new_unsafe_binder(tcx: TyCtxt<'tcx>, b: Binder<'tcx, Ty<'tcx>>) -> Ty<'tcx> {
712-
Ty::new(tcx, UnsafeBinder(b.into()))
713+
Ty::new(tcx, UnsafeBinder(tcx.mk_ty_binder(b)))
713714
}
714715

715716
#[inline]
@@ -2090,6 +2091,30 @@ mod size_asserts {
20902091
use super::*;
20912092
// tidy-alphabetical-start
20922093
static_assert_size!(ty::RegionKind<'_>, 24);
2093-
static_assert_size!(ty::TyKind<'_>, 32);
2094+
static_assert_size!(ty::TyKind<'_>, 24);
20942095
// tidy-alphabetical-end
20952096
}
2097+
2098+
// FIXME: this is a distinct type because we need to define `Encode`/`Decode` impls.
2099+
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, HashStable)]
2100+
pub struct TyBinderRef<'tcx>(pub ty::Interned<'tcx, ty::Binder<'tcx, Ty<'tcx>>>);
2101+
2102+
impl<'tcx> Deref for TyBinderRef<'tcx> {
2103+
type Target = ty::Binder<'tcx, Ty<'tcx>>;
2104+
2105+
fn deref(&self) -> &Self::Target {
2106+
&self.0
2107+
}
2108+
}
2109+
2110+
// FIXME: this is a distinct type because we need to define `Encode`/`Decode` impls.
2111+
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, HashStable)]
2112+
pub struct SigBinderRef<'tcx>(pub ty::Interned<'tcx, ty::Binder<'tcx, FnSigTys<'tcx>>>);
2113+
2114+
impl<'tcx> Deref for SigBinderRef<'tcx> {
2115+
type Target = ty::Binder<'tcx, FnSigTys<'tcx>>;
2116+
2117+
fn deref(&self) -> &Self::Target {
2118+
&self.0
2119+
}
2120+
}

0 commit comments

Comments
 (0)