From 4442967b63bbd8fb6be62e5fc3a3e24c4d33dd40 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 19 Nov 2023 12:04:03 +0000
Subject: Add accessor for effect tags

---
 .../org/shedlang/compiler/patio/PatioCodeGenContext.java   | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/main/java')

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
index 155e391f..85efa16b 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
@@ -87,6 +87,16 @@ public record PatioCodeGenContext(
                 return tag;
             }
         }
+
+        public int effectTag(UserDefinedEffect effect) {
+            var tag = this.effectTags.get(effect);
+            if (tag == null) {
+                // TODO: throw better error
+                throw new RuntimeException("No tag for: " + effect);
+            } else {
+                return tag;
+            }
+        }
     }
 
     public record CompilationUnitContext(
@@ -297,6 +307,10 @@ public record PatioCodeGenContext(
         return globalContext.structTag(type);
     }
 
+    public int effectTag(UserDefinedEffect effect) {
+        return globalContext.effectTag(effect);
+    }
+
     public PatioCodeGenResult<String> allocateCompilationUnitElementName(
         PatioCodeGenCompilationUnitElementSpace space,
         String name
-- 
cgit v1.2.3