From 3f5511f6ece79cd4393f4ce516043b734de1b0ba Mon Sep 17 00:00:00 2001
From: Romain Guy <romainguy@curious-creature.com>
Date: Mon, 12 Aug 2024 14:56:33 -0700
Subject: [PATCH] Fix counting of branches

---
 .../kotlin/explorer/code/DataModels.kt        | 29 ++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/code/DataModels.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/code/DataModels.kt
index 0905bd27..7f05b0a1 100644
--- a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/code/DataModels.kt
+++ b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/code/DataModels.kt
@@ -49,7 +49,34 @@ enum class ISA(val branchInstructions: ScatterSet<String>, val returnInstruction
         ),
         scatterSetOf("ret")
     ),
-    Arm64(scatterSetOf("b", "bl", "cbz", "cbnz", "tbz", "tbnz"), scatterSetOf("ret"))
+    Arm64(
+        scatterSetOf(
+            "b",
+            "b.eq",
+            "b.ne",
+            "b.cs",
+            "b.hs",
+            "b.cc",
+            "b.lo",
+            "b.mi",
+            "b.pl",
+            "b.vs",
+            "b.vc",
+            "b.hi",
+            "b.ls",
+            "b.ge",
+            "b.lt",
+            "b.gt",
+            "b.le",
+            "b.al",
+            "bl",
+            "cbz",
+            "cbnz",
+            "tbz",
+            "tbnz"
+        ),
+        scatterSetOf("ret")
+    )
 }
 
 data class Class(val header: String, val methods: List<Method>)