Add method to get instance instantiation arguments

celinval · celinval · commit e0a46932943a · 2023-12-20T11:07:06.000-08:00
diff --git a/compiler/rustc_smir/src/rustc_smir/context.rs b/compiler/rustc_smir/src/rustc_smir/context.rs
@@ -341,6 +341,12 @@ impl<'tcx> Context for TablesWrapper<'tcx> {
         instance.ty(tables.tcx, ParamEnv::reveal_all()).stable(&mut *tables)
     }
 
+    fn instance_args(&self, def: InstanceDef) -> GenericArgs {
+        let mut tables = self.0.borrow_mut();
+        let instance = tables.instances[def];
+        instance.args.stable(&mut *tables)
+    }
+
     fn instance_abi(&self, def: InstanceDef) -> Result<FnAbi, Error> {
         let mut tables = self.0.borrow_mut();
         let instance = tables.instances[def];
diff --git a/compiler/stable_mir/src/compiler_interface.rs b/compiler/stable_mir/src/compiler_interface.rs
@@ -125,6 +125,9 @@ pub trait Context {
     /// Get the instance type with generic substitutions applied and lifetimes erased.
     fn instance_ty(&self, instance: InstanceDef) -> Ty;
 
+    /// Get the instantiation types.
+    fn instance_args(&self, def: InstanceDef) -> GenericArgs;
+
     /// Get the instance.
     fn instance_def_id(&self, instance: InstanceDef) -> DefId;
 
diff --git a/compiler/stable_mir/src/mir/mono.rs b/compiler/stable_mir/src/mir/mono.rs
@@ -35,6 +35,11 @@ pub enum InstanceKind {
 }
 
 impl Instance {
+    /// Get the arguments this instance was instantiated with.
+    pub fn args(&self) -> GenericArgs {
+        with(|cx| cx.instance_args(self.def))
+    }
+
     /// Get the body of an Instance. The body will be eagerly monomorphized.
     pub fn body(&self) -> Option<Body> {
         with(|context| context.instance_body(self.def))
@@ -148,6 +153,7 @@ impl Debug for Instance {
         f.debug_struct("Instance")
             .field("kind", &self.kind)
             .field("def", &self.mangled_name())
+            .field("args", &self.args())
             .finish()
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,11 @@ pub enum InstanceKind {`
`35`	`35`	`}`
`36`	`36`
`37`	`37`	`impl Instance {`
	`38`	`+ /// Get the arguments this instance was instantiated with.`
	`39`	`+ pub fn args(&self) -> GenericArgs {`
	`40`	`+ with(\|cx\| cx.instance_args(self.def))`
	`41`	`+ }`
	`42`	`+`
`38`	`43`	`/// Get the body of an Instance. The body will be eagerly monomorphized.`
`39`	`44`	`pub fn body(&self) -> Option<Body> {`
`40`	`45`	`with(\|context\| context.instance_body(self.def))`
`@@ -148,6 +153,7 @@ impl Debug for Instance {`
`148`	`153`	`f.debug_struct("Instance")`
`149`	`154`	`.field("kind", &self.kind)`
`150`	`155`	`.field("def", &self.mangled_name())`
	`156`	`+ .field("args", &self.args())`
`151`	`157`	`.finish()`
`152`	`158`	`}`
`153`	`159`	`}`