visualize WFSA

breandan · Apr 30, 2024 · c231fed · c231fed
1 parent 182c182
commit c231fed
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/src/commonMain/kotlin/ai/hypergraph/kaliningraph/graphs/LabeledGraph.kt b/src/commonMain/kotlin/ai/hypergraph/kaliningraph/graphs/LabeledGraph.kt
@@ -75,6 +75,13 @@ open class LabeledGraph constructor(override val vertices: Set<LGVertex> = setOf
   var accumuator = mutableSetOf<String>()
   var description = ""
 
+  override fun reversed(): LabeledGraph =
+    (vertices.associateWith { setOf<LabeledEdge>() } +
+        vertices.flatMap { src ->
+          src.outgoing.map { edge -> edge.target to LabeledEdge(edge.target, src, edge.label) }
+        }.groupBy({ it.first }, { it.second }).mapValues { (_, v) -> v.toSet() })
+      .map { (k, v) -> V(k) { v } }.toSet().let { G(it) }
+
   fun S() = BooleanMatrix(vertices.size, 1) { i, j -> this[i].occupied }
 
   fun rewrite(substitution: V2<String>) =

diff --git a/src/jvmTest/kotlin/ai/hypergraph/kaliningraph/automata/WFSATest.kt b/src/jvmTest/kotlin/ai/hypergraph/kaliningraph/automata/WFSATest.kt
@@ -2,7 +2,9 @@ package ai.hypergraph.kaliningraph.automata
 
 import Grammars
 import Grammars.shortS2PParikhMap
+import ai.hypergraph.kaliningraph.graphs.LabeledGraph
 import ai.hypergraph.kaliningraph.parsing.*
+import ai.hypergraph.kaliningraph.visualization.alsoCopy
 import net.jhoogland.jautomata.*
 import net.jhoogland.jautomata.Automaton
 import net.jhoogland.jautomata.operations.*
@@ -32,6 +34,25 @@ class WFSATest {
         .also { println("Took ${it.duration} to decode ${it.value.size} best strings") }
   }
 
+  fun Automaton<String, Double>.toDot(processed: MutableSet<Any> = mutableSetOf()) =
+    LabeledGraph {
+      val stateQueue = mutableListOf<Any>()
+      initialStates().forEach { stateQueue.add(it) }
+      while (true) {
+        if (stateQueue.isEmpty()) break
+        val state = stateQueue.removeAt(0)
+        transitionsOut(state).forEach {
+          val label = label(it) + "/" + transitionWeight(it).toString().take(4)
+          val next = this@toDot.to(it)
+          state.hashCode().toString()[label] = next.hashCode().toString()
+          if (next !in processed) {
+            processed.add(next)
+            stateQueue.add(next)
+          }
+        }
+      }
+    }.toDot().replace("Mrecord\"", "Mrecord\", label=\"\"")
+
   /*
   ./gradlew jvmTest --tests "ai.hypergraph.kaliningraph.automata.WFSATest.testLBHRepair"
   */
@@ -54,7 +75,8 @@ class WFSATest {
             addTransition(s1, s2, a.root, 1.0)
           }
         }
-      ).also { println("Total: ${Automata.transitions(it).size} arcs, ${Automata.states(it).size}") }
+      )?.also { println("\n" + Operations.determinizeER(it).toDot().alsoCopy() + "\n") }
+        .also { println("Total: ${Automata.transitions(it).size} arcs, ${Automata.states(it).size}") }
        .let { Automata.bestStrings(it, 1000).map { it.label.joinToString(" ") }.toSet() }
     }.also {
       println("Found ${it.value.size} repairs by decoding WFSA")