Update binary metadata to match the updated WARDuino --dump-info output + use it for arg counts when mocking

This way we now have much more information about primitives and the mock window can show the correct number of argument text fields for each primitive based on the metadata instead of hardcoded values.

Author: MaartenS11

src/main/kotlin/be/ugent/topl/mio/debugger/MultiverseDebugger.kt

@@ -257,10 +257,10 @@ class MultiverseDebugger(
             overrides.clear()
             for (snapshotOverrides in snapshot.overrides) {
                 // Ignore overrides that don't appear in this program. Maybe they were for another program that we hot loaded.
-                if (snapshotOverrides.fidx >= wasmBinary.metadata.primitive_fidx_mapping.size)
+                if (snapshotOverrides.fidx >= wasmBinary.metadata.primitives.size)
                     continue
 
-                val primitiveName = wasmBinary.metadata.primitive_fidx_mapping[snapshotOverrides.fidx]
+                val primitiveName = wasmBinary.metadata.primitives[snapshotOverrides.fidx].name
                 if (!overrides.containsKey(primitiveName)) {
                     overrides[primitiveName] = mutableMapOf()
                 }
@@ -291,11 +291,11 @@ class MultiverseDebugger(
         val checkpoint = newCheckpoints.last()
         if (graph.currentNode.children.isEmpty() && change > 0 && checkpoint?.fidx_called != null) {
             val newNode = if (isAfterChoicePoint(checkpoint.snapshot.pc!!)) {
-                PrimitiveNode(wasmBinary.metadata.primitive_fidx_mapping[checkpoint.fidx_called], checkpoint.args!!).apply {
+                PrimitiveNode(wasmBinary.metadata.primitives[checkpoint.fidx_called].name, checkpoint.args!!).apply {
                     values.add(checkpoint.returns!!.first())
                 }
             } else {
-                DeterministicPrimitiveNode(wasmBinary.metadata.primitive_fidx_mapping[checkpoint.fidx_called], checkpoint.args!!)
+                DeterministicPrimitiveNode(wasmBinary.metadata.primitives[checkpoint.fidx_called].name, checkpoint.args!!)
             }
 
             if (graph.currentNode.parent != null) {

src/main/kotlin/be/ugent/topl/mio/sourcemap/BinaryInfo.kt

@@ -8,7 +8,15 @@ data class WasmInfo(
     val after_choicepoints: Set<Int>,
     val primitive_calls: Set<Int>,
     val after_primitive_calls: Set<Int>,
-    val primitive_fidx_mapping: List<String>
+    val primitives: List<WasmPrimitive>
+)
+
+data class WasmPrimitive(
+    val fidx: Int,
+    val name: String,
+    val module: String,
+    val arg_types: List<String>,
+    val return_types: List<String>
 )
 data class WasmBinary(val file: File, val metadata: WasmInfo)
 

src/main/kotlin/be/ugent/topl/mio/ui/CheckpointVisualiser.kt

@@ -137,7 +137,7 @@ data class CheckpointNode(val t: Int, val binaryInfo: WasmInfo, val checkpoint:
     init {
         var str = "Checkpoint t = $t"
         if (checkpoint.fidx_called != null) {
-            str += ", fidx = ${checkpoint.fidx_called}, ${binaryInfo.primitive_fidx_mapping[checkpoint.fidx_called]}(${checkpoint.args?.joinToString(",")})"
+            str += ", fidx = ${checkpoint.fidx_called}, ${binaryInfo.primitives[checkpoint.fidx_called].name}(${checkpoint.args?.joinToString(",")})"
         }
         userObject = str
     }

src/main/kotlin/be/ugent/topl/mio/ui/InteractiveDebugger.kt

@@ -1,6 +1,7 @@
 package be.ugent.topl.mio.ui
 
 import WasmBinary
+import WasmPrimitive
 import be.ugent.topl.mio.DebuggerConfig
 import be.ugent.topl.mio.connections.Connection
 import be.ugent.topl.mio.debugger.ConstraintParser
@@ -638,8 +639,10 @@ class MultiversePanel(private val multiverseDebugger: MultiverseDebugger, config
             val currentNode = multiverseDebugger.graph.currentNode
             val mainPanel = JPanel()
             val primitiveSelector = JComboBox<String>()
-            multiverseDebugger.wasmBinary.metadata.primitive_fidx_mapping.forEach {
-                primitiveSelector.addItem(it)
+            multiverseDebugger.wasmBinary.metadata.primitives.forEach {
+                if (it.return_types.isNotEmpty()) {
+                    primitiveSelector.addItem(it.name)
+                }
             }
             mainPanel.add(primitiveSelector)
             mainPanel.add(JLabel("("))
@@ -649,11 +652,9 @@ class MultiversePanel(private val multiverseDebugger: MultiverseDebugger, config
             val argTextFields = mutableListOf<JTextField>()
             fun updateArgFields() {
                 argTextFields.clear()
-                // TODO: Determine the count based on binary info from WARDuino
-                val argCount = if (
-                    primitiveSelector.selectedItem.toString() == "display_width" ||
-                    primitiveSelector.selectedItem.toString() == "display_height" ||
-                    primitiveSelector.selectedItem.toString() == "random_int") 0 else 1
+                val argCount = multiverseDebugger.wasmBinary.metadata.primitives.find {
+                    it.name ==  primitiveSelector.selectedItem as String
+                }!!.arg_types.size
                 repeat(argCount) {
                     argTextFields.add(JTextField())
                 }

src/main/kotlin/be/ugent/topl/mio/woodstate/WOODState.kt

@@ -131,9 +131,9 @@ data class PrimitiveOverride(
         primitiveName: String,
         args: List<Int>,
         returnValue: Int
-    ) : this(metadata.primitive_fidx_mapping.indexOf(primitiveName), args, returnValue)
+    ) : this(metadata.primitives.find { it.name == primitiveName }!!.fidx, args, returnValue)
 
-    fun getPrimitiveName(metadata: WasmInfo): String = metadata.primitive_fidx_mapping[fidx]
+    fun getPrimitiveName(metadata: WasmInfo): String = metadata.primitives[fidx].name
 }
 
 data class WOODDumpResponse(