From 36ae66b73fab2ef78bc26d3a35fa66d353b5d57f Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 19 Jul 2023 20:44:59 +0100
Subject: Add Patio code gen for returns

---
 src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/main/java')

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);
         };
-- 
cgit v1.2.3