From 47f3094508b59f030cb26d39ac4d2dccdf3414f2 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 13 Jun 2023 20:06:26 +0100
Subject: Add LLVM getelementptr instruction

---
 .../llvm/ir/LlvmIrSerializerInstructionTests.java  | 24 ++++++++++++++++++++++
 .../compiler/backends/llvm/ir/ast/LlvmIr.java      |  9 ++++++++
 2 files changed, 33 insertions(+)

(limited to 'src/test/java')

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