Add GlobalVariable::ty and Type::to_ptr methods

fe-lang · Nov 26, 2024 · 69c8f76 · 69c8f76
1 parent 7f349e4
commit 69c8f76
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/crates/ir/src/global_variable.rs b/crates/ir/src/global_variable.rs
@@ -55,6 +55,12 @@ impl GlobalVariableStore {
 pub struct GlobalVariable(pub u32);
 cranelift_entity::entity_impl!(GlobalVariable);
 
+impl GlobalVariable {
+    pub fn ty(self, module: &ModuleCtx) -> Type {
+        module.with_gv_store(|s| s.ty(self))
+    }
+}
+
 impl WriteWithModule for GlobalVariable {
     fn write(&self, module: &ModuleCtx, w: &mut impl io::Write) -> io::Result<()> {
         module.with_gv_store(|s| s.gv_data(*self).write(module, w))

diff --git a/crates/ir/src/types.rs b/crates/ir/src/types.rs
@@ -167,6 +167,10 @@ impl Type {
 
         Some(ctx.with_ty_store(|s| s.resolve_compound(cmpd).clone()))
     }
+
+    pub fn to_ptr(self, ctx: &ModuleCtx) -> Type {
+        ctx.with_ty_store_mut(|s| s.make_ptr(self))
+    }
 }
 
 impl cmp::PartialOrd for Type {