Add LLVM getelementptr instruction

5 files changed, 61 insertions, 1 deletions

diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializer.java b/src/main/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializer.java
index cef8f005..11569f74 100644
--- a/src/main/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializer.java
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializer.java
@@ -41,6 +41,17 @@ public class LlvmIrSerializer {
             }
             case LlvmCall call -> {
             }
+            case LlvmGetElementPtr getElementPtr -> {
+                serializeVariable(getElementPtr.target(), builder);
+                builder.append(" = getelementptr ");
+                serializeType(getElementPtr.pointerType().elementType(), builder);
+                builder.append(", ptr ");
+                serializeOperand(getElementPtr.pointer(), builder);
+                for (var index : getElementPtr.indexes()) {
+                    builder.append(", ");
+                    serializeTypedOperand(index, builder);
+                }
+            }
         }
     }
 
diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmGetElementPtr.java b/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmGetElementPtr.java
new file mode 100644
index 00000000..ad9f1ced
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmGetElementPtr.java
@@ -0,0 +1,16 @@
+package org.shedlang.compiler.backends.llvm.ir.ast.instructions;
+
+import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmOperand;
+import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmTypedOperand;
+import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmVariable;
+import org.shedlang.compiler.backends.llvm.ir.ast.types.LlvmTypePointer;
+
+import java.util.List;
+
+public record LlvmGetElementPtr(
+    LlvmVariable target,
+    LlvmTypePointer pointerType,
+    LlvmOperand pointer,
+    List<LlvmTypedOperand> indexes
+) implements LlvmInstruction {
+}
diff --git a/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmInstruction.java b/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmInstruction.java
index 2e0ffb23..48d90ed4 100644
--- a/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmInstruction.java
+++ b/src/main/java/org/shedlang/compiler/backends/llvm/ir/ast/instructions/LlvmInstruction.java
@@ -1,4 +1,4 @@
 package org.shedlang.compiler.backends.llvm.ir.ast.instructions;
 
-public sealed interface LlvmInstruction permits LlvmAdd, LlvmCall {
+public sealed interface LlvmInstruction permits LlvmAdd, LlvmCall, LlvmGetElementPtr {
 }
diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializerInstructionTests.java b/src/test/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializerInstructionTests.java
index caebe5b4..1617d454 100644
--- a/src/test/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializerInstructionTests.java
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/ir/LlvmIrSerializerInstructionTests.java
@@ -4,6 +4,8 @@ import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.backends.llvm.ir.ast.LlvmIr;
 import org.shedlang.compiler.backends.llvm.ir.ast.types.LlvmTypes;
 
+import java.util.List;
+
 import static org.shedlang.compiler.backends.llvm.ir.LlvmIrSerializerTesting.serializeToString;
 import static org.zwobble.precisely.AssertThat.assertThat;
 import static org.zwobble.precisely.Matchers.equalTo;
@@ -22,4 +24,26 @@ public class LlvmIrSerializerInstructionTests {
 
         assertThat(result, equalTo("$x = add i64 $y, $z"));
     }
+
+    @Test
+    public void getElementPtrIsSerialized() {
+        var instruction = LlvmIr.getElementPtr(
+            LlvmIr.variableLocal("x"),
+            LlvmTypes.pointer(
+                LlvmTypes.array(
+                    0,
+                    LlvmTypes.structure(List.of(LlvmTypes.INT_64, LlvmTypes.INT_32))
+                )
+            ),
+            LlvmIr.variableLocal("y"),
+            List.of(
+                LlvmIr.typedOperand(LlvmTypes.INT_64, LlvmIr.operandInt(3)),
+                LlvmIr.typedOperand(LlvmTypes.INT_32, LlvmIr.operandInt(1))
+            )
+        );
+
+        var result = serializeToString(instruction, LlvmIrSerializer::serializeInstruction);
+
+        assertThat(result, equalTo("$x = getelementptr [0 x {i64, i32}], ptr $y, i64 3, i32 1"));
+    }
 }
diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java b/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
index 7ded22d8..1c1b298b 100644
--- a/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
@@ -36,6 +36,15 @@ public class LlvmIr {
         return new LlvmBasicBlock(Optional.of(name), instructions, terminator);
     }
 
+    public static LlvmGetElementPtr getElementPtr(
+        LlvmVariable target,
+        LlvmTypePointer pointerType,
+        LlvmOperand pointer,
+        List<LlvmTypedOperand> indexes
+    ) {
+        return new LlvmGetElementPtr(target, pointerType, pointer, indexes);
+    }
+
     public static LlvmOperand operandArray(
         LlvmType type,
         List<LlvmOperand> elements