Compile load-store instruction to LLVM IR

4 files changed, 76 insertions, 9 deletions

diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompiler.java b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompiler.java
index b311b41d..7c74f211 100644
--- a/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompiler.java
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompiler.java
@@ -25,9 +25,14 @@ public class LlvmCompiler {
             case LocalLoadInstruction localLoadInstruction -> {
                 throw new UnsupportedOperationException("TODO");
             }
-            case LocalStoreInstruction localStoreInstruction -> {
-                throw new UnsupportedOperationException("TODO");
+            case LocalStoreInstruction localStore -> {
+                var result = context.pop();
+                var value = result.value();
+                context = result.context();
+
+                yield context.addLocal(localStore.name(), value);
             }
+
             case ModuleExportInstruction moduleExportInstruction -> {
                 throw new UnsupportedOperationException("TODO");
             }
diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerContext.java b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerContext.java
index 6d917e49..1af09f36 100644
--- a/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerContext.java
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerContext.java
@@ -1,30 +1,49 @@
 package org.shedlang.compiler.backends.llvm;
 
 import org.pcollections.Empty;
+import org.pcollections.PMap;
+import org.pcollections.PStack;
 import org.pcollections.PVector;
 import org.shedlang.compiler.backends.llvm.ir.ast.instructions.LlvmInstruction;
 import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmOperand;
 
 public record LlvmCompilerContext(
-    PVector<LlvmOperand> stack,
+    PStack<LlvmOperand> stack,
+    PMap<String, LlvmOperand> locals,
     PVector<LlvmInstruction> instructions
 ) {
     public static LlvmCompilerContext EMPTY = new LlvmCompilerContext(
-        Empty.vector(),
+        Empty.stack(),
+        Empty.map(),
         Empty.vector()
     );
 
-    public LlvmCompilerContext addInstruction(LlvmInstruction instruction) {
+    public LlvmCompilerContext push(LlvmOperand operand) {
         return new LlvmCompilerContext(
-            stack,
-            instructions.plus(instruction)
+            stack.plus(operand),
+            locals,
+            instructions
         );
     }
 
-    public LlvmCompilerContext push(LlvmOperand operand) {
+    public LlvmCompilerResult<LlvmOperand> pop() {
+        var context = new LlvmCompilerContext(stack.minus(0), locals, instructions);
+        return new LlvmCompilerResult<>(stack.get(0), context);
+    }
+
+    public LlvmCompilerContext addLocal(String name, LlvmOperand value) {
         return new LlvmCompilerContext(
-            stack.plus(operand),
+            stack,
+            locals.plus(name, value),
             instructions
         );
     }
+
+    public LlvmCompilerContext addInstruction(LlvmInstruction instruction) {
+        return new LlvmCompilerContext(
+            stack,
+            locals,
+            instructions.plus(instruction)
+        );
+    }
 }
diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerResult.java b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerResult.java
new file mode 100644
index 00000000..b0d7e1b8
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/LlvmCompilerResult.java
@@ -0,0 +1,7 @@
+package org.shedlang.compiler.backends.llvm;
+
+public record LlvmCompilerResult<T>(
+    T value,
+    LlvmCompilerContext context
+) {
+}
diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java
new file mode 100644
index 00000000..2f20ec85
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java
@@ -0,0 +1,36 @@
+package org.shedlang.compiler.backends.llvm;
+
+import org.junit.jupiter.api.Test;
+import org.shedlang.compiler.backends.llvm.ir.ast.LlvmIr;
+import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmOperand;
+import org.shedlang.compiler.stackir.instructions.Instructions;
+import org.zwobble.precisely.Matcher;
+
+import java.util.Map;
+
+import static org.shedlang.compiler.backends.llvm.ir.ast.LlvmIrMatchers.isLlvmOperandInt;
+import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.*;
+
+public class LlvmCompilerLocalStoreTests {
+    @Test
+    public void localStorePopsFromStackAndUpdatesLocals() {
+        var instruction = Instructions.localStore("x");
+        var context = LlvmCompilerContext.EMPTY
+            .push(LlvmIr.operandInt(0))
+            .push(LlvmIr.operandInt(1));
+
+        var result = LlvmCompiler.compileInstruction(instruction, context);
+
+        assertThat(result.instructions(), isSequence());
+        assertThat(result.stack(), isSequence(isLlvmOperandInt(0)));
+        assertThat(result.locals().entrySet(), isSequence(isLocal("x", isLlvmOperandInt(1))));
+    }
+
+    private static Matcher<Map.Entry<String, LlvmOperand>> isLocal(String name, Matcher<LlvmOperand> operand) {
+        return allOf(
+            has("name", x -> x.getKey(), equalTo(name)),
+            has("operand", x -> x.getValue(), operand)
+        );
+    }
+}