Add Patio code gen for returns

1 files changed, 5 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
index f19e88ba..790bf8d7 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
@@ -3,8 +3,10 @@ package org.shedlang.compiler.patio;
 import org.pcollections.PVector;
 import org.shedlang.compiler.patio.instructions.PatioDiscard;
 import org.shedlang.compiler.patio.instructions.PatioInstruction;
+import org.shedlang.compiler.patio.instructions.PatioReturn;
 import org.shedlang.compiler.stackir.instructions.DiscardInstruction;
 import org.shedlang.compiler.stackir.instructions.Instruction;
+import org.shedlang.compiler.stackir.instructions.ReturnInstruction;
 import org.shedlang.compiler.util.P;
 
 public class PatioCodeGen {
@@ -15,6 +17,9 @@ public class PatioCodeGen {
             case DiscardInstruction discard ->
                 P.vector(new PatioDiscard(discard.source()));
 
+            case ReturnInstruction discard ->
+                P.vector(new PatioReturn(discard.source()));
+
             default ->
                 throw new IllegalStateException("Unexpected value: " + stackIrInstruction);
         };