From 090e353e9682024803747174241baf46b566b949 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 19 Nov 2023 12:05:47 +0000 Subject: Add accessor for user defined effect infos --- .../java/org/shedlang/compiler/patio/PatioCodeGenContext.java | 10 ++++++++++ 1 file changed, 10 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 85efa16b..fd45f29d 100644 --- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java +++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java @@ -88,6 +88,12 @@ public record PatioCodeGenContext( } } + public UserDefinedEffectInfo userDefinedEffectInfo(UserDefinedEffect effect) { + return typeInfos.userDefinedEffectInfo(effect) + // TODO: throw better error + .orElseThrow(() -> new RuntimeException("No effect info for: " + effect)); + } + public int effectTag(UserDefinedEffect effect) { var tag = this.effectTags.get(effect); if (tag == null) { @@ -307,6 +313,10 @@ public record PatioCodeGenContext( return globalContext.structTag(type); } + public UserDefinedEffectInfo userDefinedEffectInfo(UserDefinedEffect effect) { + return this.globalContext.userDefinedEffectInfo(effect); + } + public int effectTag(UserDefinedEffect effect) { return globalContext.effectTag(effect); } -- cgit v1.2.3