From d5707930a4e75c2fa47aa3b75fea13575e269009 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 23 Jul 2023 21:17:13 +0100
Subject: Add Patio code gen for int-eq

---
 .../org/shedlang/compiler/patio/PatioCodeGen.java  | 17 +++++++++----
 .../compiler/patio/PatioCodeGenIntEqualsTests.java | 28 ++++++++++++++++++++++
 2 files changed, 41 insertions(+), 4 deletions(-)
 create mode 100644 src/test/java/org/shedlang/compiler/patio/PatioCodeGenIntEqualsTests.java

(limited to 'src')

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
index f9bfd38b..63d5cd5e 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
@@ -2,10 +2,7 @@ package org.shedlang.compiler.patio;
 
 import org.shedlang.compiler.patio.ast.PatioCompilationUnit;
 import org.shedlang.compiler.patio.ast.PatioVisibility;
-import org.shedlang.compiler.patio.ast.instructions.PatioDataPtrToInt;
-import org.shedlang.compiler.patio.ast.instructions.PatioIntAdd;
-import org.shedlang.compiler.patio.ast.instructions.PatioLocalStore;
-import org.shedlang.compiler.patio.ast.instructions.PatioReturn;
+import org.shedlang.compiler.patio.ast.instructions.*;
 import org.shedlang.compiler.patio.ast.operands.PatioInt;
 import org.shedlang.compiler.patio.ast.operands.PatioLocalRef;
 import org.shedlang.compiler.patio.ast.types.PatioTypes;
@@ -266,6 +263,18 @@ public class PatioCodeGen {
                     context
                 );
 
+            case IntEqualsInstruction intEquals ->
+                generateBinaryOperation(
+                    (target, left, right, operationContext) ->
+                        operationContext.addInstruction(new PatioIntEquals(
+                            target,
+                            left,
+                            right,
+                            intEquals.source()
+                        )),
+                    context
+                );
+
             case LocalLoadInstruction localLoad -> {
                 var local = context.stackIrLocalNameToLocal(localLoad.name());
 
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenIntEqualsTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenIntEqualsTests.java
new file mode 100644
index 00000000..cc1eaf2d
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenIntEqualsTests.java
@@ -0,0 +1,28 @@
+package org.shedlang.compiler.patio;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.shedlang.compiler.patio.ast.operands.PatioLocalRef;
+import org.shedlang.compiler.stackir.instructions.Instructions;
+import org.shedlang.compiler.testing.snapshots.SnapshotResolver;
+import org.shedlang.compiler.testing.snapshots.Snapshotter;
+
+import java.io.IOException;
+
+import static org.shedlang.compiler.patio.PatioTesting.serialize;
+
+@ExtendWith({SnapshotResolver.class})
+public class PatioCodeGenIntEqualsTests {
+    @Test
+    public void canGenerateCodeForIntEquals(Snapshotter snapshotter) throws IOException {
+        var stackIr = Instructions.intEquals();
+        var context = PatioCodeGenContext.EMPTY
+            .temporaryPush(new PatioLocalRef("a"))
+            .temporaryPush(new PatioLocalRef("b"))
+            .temporaryPush(new PatioLocalRef("c"));
+
+        var result = PatioCodeGen.generateInstruction(stackIr, context);
+
+        snapshotter.assertSnapshot(serialize(result));
+    }
+}
-- 
cgit v1.2.3