add RegionDef

Author: ouz-a

compiler/rustc_smir/src/rustc_internal/mod.rs

@@ -135,6 +135,10 @@ impl<'tcx> Tables<'tcx> {
         stable_mir::ty::ImplDef(self.create_def_id(did))
     }
 
+    pub fn region_def(&mut self, did: DefId) -> stable_mir::ty::RegionDef {
+        stable_mir::ty::RegionDef(self.create_def_id(did))
+    }
+
     pub fn prov(&mut self, aid: AllocId) -> stable_mir::ty::Prov {
         stable_mir::ty::Prov(self.create_alloc_id(aid))
     }

compiler/rustc_smir/src/rustc_smir/mod.rs

@@ -10,7 +10,7 @@
 use crate::rustc_internal::{self, opaque};
 use crate::stable_mir::mir::{CopyNonOverlapping, UserTypeProjection, VariantIdx};
 use crate::stable_mir::ty::{
-    FloatTy, GenericParamDef, IntTy, Movability, RigidTy, Span, TyKind, UintTy,
+    EarlyBoundRegion, FloatTy, GenericParamDef, IntTy, Movability, RigidTy, Span, TyKind, UintTy,
 };
 use crate::stable_mir::{self, CompilerError, Context};
 use rustc_hir as hir;
@@ -1514,6 +1514,27 @@ impl<'tcx> Stable<'tcx> for ty::Region<'tcx> {
     }
 }
 
+impl<'tcx> Stable<'tcx> for ty::RegionKind<'tcx> {
+    type T = stable_mir::ty::RegionKind;
+
+    fn stable(&self, tables: &mut Tables<'tcx>) -> Self::T {
+        match self {
+            ty::ReEarlyBound(early_reg) => RegionKind::ReEarlyBound(EarlyBoundRegion {
+                def_id: tables.region_def(early_reg.def_id),
+                index: early_reg.index,
+                name: early_reg.name.to_string(),
+            }),
+            ty::ReLateBound(_, _) => todo!(),
+            ty::ReFree(_) => todo!(),
+            ty::ReStatic => todo!(),
+            ty::ReVar(_) => todo!(),
+            ty::RePlaceholder(_) => todo!(),
+            ty::ReErased => todo!(),
+            ty::ReError(_) => todo!(),
+        }
+    }
+}
+
 impl<'tcx> Stable<'tcx> for rustc_span::Span {
     type T = stable_mir::ty::Span;
 

compiler/rustc_smir/src/stable_mir/ty.rs

@@ -38,7 +38,7 @@ pub(crate) struct Region {
     kind: RegionKind,
 }
 
-enum RegionKind {
+pub enum RegionKind {
     ReEarlyBound(EarlyBoundRegion),
     ReLateBound(DebruijnIndex, BoundRegion),
     ReFree(FreeRegion),
@@ -52,7 +52,7 @@ enum RegionKind {
 pub(crate) type DebruijnIndex = u32;
 
 pub struct EarlyBoundRegion {
-    pub def_id: DefId,
+    pub def_id: RegionDef,
     pub index: u32,
     pub name: Symbol,
 }
@@ -65,7 +65,7 @@ pub struct BoundRegion {
 }
 
 pub struct FreeRegion {
-    pub scope: DefId,
+    pub scope: RegionDef,
     pub bound_region: BoundRegionKind,
 }
 
@@ -196,6 +196,9 @@ pub struct ConstDef(pub(crate) DefId);
 #[derive(Clone, PartialEq, Eq, Debug)]
 pub struct ImplDef(pub(crate) DefId);
 
+#[derive(Clone, PartialEq, Eq, Debug)]
+pub struct RegionDef(pub(crate) DefId);
+
 #[derive(Clone, Debug)]
 pub struct GenericArgs(pub Vec<GenericArgKind>);