From f08393c11645f29f93467f5c32fdce4abf41d2e7 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 19 Jul 2023 21:06:44 +0100
Subject: Add local-store instruction to Patio IR

---
 .../org/shedlang/compiler/patio/PatioInstructionSerializer.java  | 6 ++++++
 .../shedlang/compiler/patio/instructions/PatioInstruction.java   | 2 +-
 .../org/shedlang/compiler/patio/instructions/PatioLocalLoad.java | 1 -
 .../shedlang/compiler/patio/instructions/PatioLocalStore.java    | 9 +++++++++
 4 files changed, 16 insertions(+), 2 deletions(-)
 create mode 100644 src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalStore.java

(limited to 'src/main/java')

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioInstructionSerializer.java b/src/main/java/org/shedlang/compiler/patio/PatioInstructionSerializer.java
index 342adc3a..a729f225 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioInstructionSerializer.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioInstructionSerializer.java
@@ -43,6 +43,12 @@ public class PatioInstructionSerializer {
                     treeString(localLoad.name())
                 );
 
+            case PatioLocalStore localStore ->
+                treeList(
+                    treeIdentifier("local-store"),
+                    treeString(localStore.name())
+                );
+
             case PatioMemoryLoad memoryLoad ->
                 treeList(
                     treeIdentifier("memory-load"),
diff --git a/src/main/java/org/shedlang/compiler/patio/instructions/PatioInstruction.java b/src/main/java/org/shedlang/compiler/patio/instructions/PatioInstruction.java
index fa6fe302..f839e6e7 100644
--- a/src/main/java/org/shedlang/compiler/patio/instructions/PatioInstruction.java
+++ b/src/main/java/org/shedlang/compiler/patio/instructions/PatioInstruction.java
@@ -2,6 +2,6 @@ package org.shedlang.compiler.patio.instructions;
 
 import org.shedlang.compiler.sources.Source;
 
-public sealed interface PatioInstruction permits PatioAlloc, PatioDiscard, PatioLocalLoad, PatioMemoryLoad, PatioMemoryStore, PatioPushFunctionPointer, PatioReturn {
+public sealed interface PatioInstruction permits PatioAlloc, PatioDiscard, PatioLocalLoad, PatioLocalStore, PatioMemoryLoad, PatioMemoryStore, PatioPushFunctionPointer, PatioReturn {
     Source source();
 }
diff --git a/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalLoad.java b/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalLoad.java
index 984b9ee4..4ec34a12 100644
--- a/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalLoad.java
+++ b/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalLoad.java
@@ -5,6 +5,5 @@ import org.shedlang.compiler.sources.Source;
 /**
  * Load the local with the given name and push it onto the stack.
  */
-
 public record PatioLocalLoad(String name, Source source) implements PatioInstruction  {
 }
diff --git a/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalStore.java b/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalStore.java
new file mode 100644
index 00000000..317a1172
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/patio/instructions/PatioLocalStore.java
@@ -0,0 +1,9 @@
+package org.shedlang.compiler.patio.instructions;
+
+import org.shedlang.compiler.sources.Source;
+
+/**
+ * Pop a value from the stack and store it as the local with the given name.
+ */
+public record PatioLocalStore(String name, Source source) implements PatioInstruction  {
+}
-- 
cgit v1.2.3