Add method to get instance instantiation arguments

celinval · celinval · commit 2fd4c26ed1ca · 2023-12-19T14:08:48.000-08:00
diff --git a/compiler/rustc_smir/src/rustc_smir/context.rs b/compiler/rustc_smir/src/rustc_smir/context.rs
@@ -335,6 +335,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_def_id(&self, def: InstanceDef) -> stable_mir::DefId {
         let mut tables = self.0.borrow_mut();
         let def_id = tables.instances[def].def_id();
diff --git a/compiler/stable_mir/src/compiler_interface.rs b/compiler/stable_mir/src/compiler_interface.rs
@@ -124,6 +124,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
@@ -34,6 +34,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))
@@ -142,6 +147,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
`@@ -34,6 +34,11 @@ pub enum InstanceKind {`
`34`	`34`	`}`
`35`	`35`
`36`	`36`	`impl Instance {`
	`37`	`+ /// Get the arguments this instance was instantiated with.`
	`38`	`+ pub fn args(&self) -> GenericArgs {`
	`39`	`+ with(\|cx\| cx.instance_args(self.def))`
	`40`	`+ }`
	`41`	`+`
`37`	`42`	`/// Get the body of an Instance. The body will be eagerly monomorphized.`
`38`	`43`	`pub fn body(&self) -> Option<Body> {`
`39`	`44`	`with(\|context\| context.instance_body(self.def))`
`@@ -142,6 +147,7 @@ impl Debug for Instance {`
`142`	`147`	`f.debug_struct("Instance")`
`143`	`148`	`.field("kind", &self.kind)`
`144`	`149`	`.field("def", &self.mangled_name())`
	`150`	`+ .field("args", &self.args())`
`145`	`151`	`.finish()`
`146`	`152`	`}`
`147`	`153`	`}`