Backend: Graph Editor Split & Dissolve (#2466)

1 files changed, 72 insertions, 5 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/test/GraphEditor.kt b/src/main/java/at/hannibal2/skyhanni/test/GraphEditor.kt
index f5acf90fb..8757f19bf 100644
--- a/src/main/java/at/hannibal2/skyhanni/test/GraphEditor.kt
+++ b/src/main/java/at/hannibal2/skyhanni/test/GraphEditor.kt
@@ -55,7 +55,18 @@ object GraphEditor {
     private val edges = mutableListOf<GraphingEdge>()
 
     private var activeNode: GraphingNode? = null
+        set(value) {
+            field = value
+            selectedEdge = findEdgeBetweenActiveAndClosed()
+            checkDissolve()
+        }
     private var closedNode: GraphingNode? = null
+        set(value) {
+            field = value
+            selectedEdge = findEdgeBetweenActiveAndClosed()
+        }
+
+    private var selectedEdge: GraphingEdge? = null
     private var ghostPosition: LorenzVec? = null
 
     private var seeThroughBlocks = true
@@ -86,6 +97,7 @@ object GraphEditor {
 
     private val edgeColor = LorenzColor.GOLD.addOpacity(150)
     private val edgeDijkstraColor = LorenzColor.DARK_BLUE.addOpacity(150)
+    private val edgeSelectedColor = LorenzColor.DARK_RED.addOpacity(150)
 
     val scrollValue = ScrollValue()
     var nodesDisplay = emptyList<Renderable>()
@@ -127,7 +139,11 @@ object GraphEditor {
             add("§eTutorial: §6${KeyboardManager.getKeyName(config.tutorialKey)}")
             add("§eToggle Ghost Position: §6${KeyboardManager.getKeyName(config.toggleGhostPosition)}")
             add(" ")
-            if (activeNode != null) add("§eText: §6${KeyboardManager.getKeyName(config.textKey)}")
+            if (activeNode != null) {
+                add("§eText: §6${KeyboardManager.getKeyName(config.textKey)}")
+                if (dissolvePossible) add("§eDissolve: §6${KeyboardManager.getKeyName(config.dissolveKey)}")
+                if (selectedEdge != null) add("§eSplit: §6${KeyboardManager.getKeyName(config.splitKey)}")
+            }
         }
 
         if (!inTextMode) {
@@ -152,6 +168,18 @@ object GraphEditor {
         }
     }
 
+    private var dissolvePossible = false
+
+    private fun findEdgeBetweenActiveAndClosed(): GraphingEdge? = getEdgeIndex(activeNode, closedNode)?.let { edges[it] }
+
+    private fun checkDissolve() {
+        if (activeNode == null) {
+            dissolvePossible = false
+            return
+        }
+        dissolvePossible = edges.count { it.isInEdge(activeNode) } == 2
+    }
+
     @SubscribeEvent
     fun onGuiRender(event: GuiRenderEvent) {
         if (!isEnabled()) return
@@ -207,7 +235,7 @@ object GraphEditor {
         )
     }
 
-    private fun createTagName(
+private fun createTagName(
         name: String,
         tag: GraphNodeTag,
         node: GraphingNode,
@@ -333,7 +361,11 @@ object GraphEditor {
     private fun LorenzRenderWorldEvent.drawEdge(edge: GraphingEdge) = this.draw3DLine_nea(
         edge.node1.position.add(0.5, 0.5, 0.5),
         edge.node2.position.add(0.5, 0.5, 0.5),
-        if (edge !in highlightedEdges) edgeColor else edgeDijkstraColor,
+        when {
+            selectedEdge == edge -> edgeSelectedColor
+            edge in highlightedEdges -> edgeDijkstraColor
+            else -> edgeColor
+        },
         7,
         !seeThroughBlocks,
     )
@@ -449,6 +481,8 @@ object GraphEditor {
                 feedBackInTutorial("Added new edge.")
             } else {
                 this.edges.removeAt(edge)
+                checkDissolve()
+                selectedEdge = findEdgeBetweenActiveAndClosed()
                 feedBackInTutorial("Removed edge.")
             }
         }
@@ -466,6 +500,29 @@ object GraphEditor {
             inTutorialMode = !inTutorialMode
             ChatUtils.chat("Tutorial mode is now ${if (inTutorialMode) "active" else "inactive"}.")
         }
+        if (selectedEdge != null && config.splitKey.isKeyClicked()) {
+            val edge = selectedEdge ?: return
+            feedBackInTutorial("Split Edge into a Node and two edges.")
+            val middle = edge.node1.position.middle(edge.node2.position).roundLocationToBlock()
+            val node = GraphingNode(id++, middle)
+            nodes.add(node)
+            edges.remove(edge)
+            addEdge(node, edge.node1)
+            addEdge(node, edge.node2)
+            activeNode = node
+        }
+        if (dissolvePossible && config.dissolveKey.isKeyClicked()) {
+            feedBackInTutorial("Dissolved the node, now it is gone.")
+            val edgePair = edges.filter { it.isInEdge(activeNode) }
+            val edge1 = edgePair[0]
+            val edge2 = edgePair[1]
+            val neighbors1 = if (edge1.node1 == activeNode) edge1.node2 else edge1.node1
+            val neighbors2 = if (edge2.node1 == activeNode) edge2.node2 else edge2.node1
+            edges.removeAll(edgePair)
+            nodes.remove(activeNode)
+            activeNode = null
+            addEdge(neighbors1, neighbors2)
+        }
     }
 
     private fun save() {
@@ -561,7 +618,14 @@ object GraphEditor {
         else null
 
     private fun addEdge(node1: GraphingNode?, node2: GraphingNode?) =
-        if (node1 != null && node2 != null && node1 != node2) edges.add(GraphingEdge(node1, node2)) else false
+        if (node1 != null && node2 != null && node1 != node2) {
+            val edge = GraphingEdge(node1, node2)
+            if (edge.isInEdge(activeNode)) {
+                checkDissolve()
+                selectedEdge = findEdgeBetweenActiveAndClosed()
+            }
+            edges.add(edge)
+        } else false
 
     /** Has a side effect on the graphing graph, since it runs [prune] on the graphing graph*/
     private fun compileGraph(): Graph {
@@ -597,6 +661,8 @@ object GraphEditor {
             }.flatten().distinct(),
         )
         id = nodes.lastOrNull()?.id?.plus(1) ?: 0
+        checkDissolve()
+        selectedEdge = findEdgeBetweenActiveAndClosed()
     }
 
     private val highlightedNodes = mutableSetOf<GraphingNode>()
@@ -630,6 +696,7 @@ object GraphEditor {
         edges.clear()
         activeNode = null
         closedNode = null
+        dissolvePossible = false
         ghostPosition = null
     }
 
@@ -671,7 +738,7 @@ private class GraphingNode(
 
 private class GraphingEdge(val node1: GraphingNode, val node2: GraphingNode) {
 
-    fun isInEdge(node: GraphingNode) = node1 == node || node2 == node
+    fun isInEdge(node: GraphingNode?) = node1 == node || node2 == node
 
     override fun equals(other: Any?): Boolean {
         if (this === other) return true