collect pcfg triples

breandan · Feb 20, 2024 · c8f0880 · c8f0880
1 parent 6f371c4
commit c8f0880
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/commonMain/kotlin/ai/hypergraph/kaliningraph/parsing/Tree.kt b/src/commonMain/kotlin/ai/hypergraph/kaliningraph/parsing/Tree.kt
@@ -3,6 +3,7 @@ package ai.hypergraph.kaliningraph.parsing
 import ai.hypergraph.kaliningraph.graphs.LGVertex
 import ai.hypergraph.kaliningraph.graphs.LabeledGraph
 import ai.hypergraph.kaliningraph.tensor.FreeMatrix
+import ai.hypergraph.kaliningraph.types.Π3A
 
 typealias TreeMatrix = FreeMatrix<Forest>
 typealias Forest = Set<Tree>
@@ -27,6 +28,11 @@ class Tree constructor(
     if (children.isEmpty()) "()"
     else children.joinToString("", prefix = "(", postfix = ")") { it.structureEncode() }
 
+  fun triples(): List<Π3A<Σᐩ>> =
+    if (children.size != 2) listOf()
+    else listOf(Π3A(root, children[0].root, children[1].root)) +
+      children.flatMap { it.triples() }
+
   fun toGraph(j: Σᐩ = "0"): LabeledGraph =
     LabeledGraph { LGVertex(root, "$root.$j").let { it - it } } +
       children.foldIndexed(