🧪 实验性的增加了xmind树形输入模式但还存在一些问题（反引号深入扩展、反斜杠广度扩展）

Author: Littlefean

app/src/core/service/controlService/keyboardOnlyEngine/keyboardOnlyEngine.tsx

@@ -3,14 +3,16 @@ import { Vector } from "../../../dataStruct/Vector";
 import { EdgeRenderer } from "../../../render/canvas2d/entityRenderer/edge/EdgeRenderer";
 import { Camera } from "../../../stage/Camera";
 import { Stage } from "../../../stage/Stage";
+import { GraphMethods } from "../../../stage/stageManager/basicMethods/GraphMethods";
+import { StageAutoAlignManager } from "../../../stage/stageManager/concreteMethods/StageAutoAlignManager";
 import { StageNodeAdder } from "../../../stage/stageManager/concreteMethods/stageNodeAdder";
 import { StageManager } from "../../../stage/stageManager/StageManager";
 import { ConnectableEntity } from "../../../stage/stageObject/abstract/ConnectableEntity";
 import { EntityDashTipEffect } from "../../feedbackService/effectEngine/concrete/EntityDashTipEffect";
 import { EntityShakeEffect } from "../../feedbackService/effectEngine/concrete/EntityShakeEffect";
 import { TextRiseEffect } from "../../feedbackService/effectEngine/concrete/TextRiseEffect";
 import { Settings } from "../../Settings";
-import { editTextNode } from "../controller/concrete/utilsControl";
+import { addTextNodeByLocation, editTextNode } from "../controller/concrete/utilsControl";
 import { KeyboardOnlyDirectionController } from "./keyboardOnlyDirectionController";
 import { NewTargetLocationSelector } from "./newTargetLocationSelector";
 import { SelectChangeEngine } from "./selectChangeEngine";
@@ -52,6 +54,11 @@ export namespace KeyboardOnlyEngine {
    */
   function bindKeyEvents() {
     window.addEventListener("keydown", (event) => {
+      if (event.key === "`") {
+        onDeepGenerateNode();
+      } else if (event.key === "\\") {
+        onBroadGenerateNode();
+      }
       if (event.key === "Tab") {
         // if (isEnableVirtualCreate()) {
         //   createStart();
@@ -93,6 +100,66 @@ export namespace KeyboardOnlyEngine {
     // });
   }
 
+  function onDeepGenerateNode() {
+    const rootNode = StageManager.getConnectableEntity().find((node) => node.isSelected);
+    if (!rootNode) return;
+    Camera.clearMoveCommander();
+    Camera.speed = Vector.getZero();
+    // 在自己的右下方创建一个节点
+    const newLocation = rootNode.collisionBox.getRectangle().rightCenter.add(new Vector(100, 100));
+    addTextNodeByLocation(newLocation, true, (newUUID) => {
+      const newNode = StageManager.getTextNodeByUUID(newUUID);
+      if (!newNode) return;
+      // 连接到之前的节点
+      StageManager.connectEntity(rootNode, newNode);
+      // 重新排列树形节点
+      const rootNodeParents = GraphMethods.getRoots(rootNode);
+      if (rootNodeParents.length === 1) {
+        const rootNodeParent = rootNodeParents[0];
+        if (GraphMethods.isTree(rootNodeParent)) {
+          StageAutoAlignManager.autoLayoutSelectedFastTreeModeRight(rootNodeParent);
+          // 更新选择状态
+          rootNodeParent.isSelected = false;
+          newNode.isSelected = true;
+        }
+      }
+
+      Stage.effectMachine.addEffects(EdgeRenderer.getConnectedEffects(rootNode, newNode));
+    });
+  }
+
+  function onBroadGenerateNode() {
+    const currentSelectNode = StageManager.getConnectableEntity().find((node) => node.isSelected);
+    if (!currentSelectNode) return;
+    Camera.clearMoveCommander();
+    Camera.speed = Vector.getZero();
+    // 找到自己的父节点
+    const parents = GraphMethods.nodeParentArray(currentSelectNode);
+    if (parents.length === 0) return;
+    if (parents.length !== 1) return;
+    const parent = parents[0];
+    // 在自己的正下方创建一个节点
+    const newLocation = parent.collisionBox.getRectangle().bottomCenter.add(new Vector(0, 100));
+    addTextNodeByLocation(newLocation, true, (newUUID) => {
+      const newNode = StageManager.getTextNodeByUUID(newUUID);
+      if (!newNode) return;
+      // 连接到之前的节点
+      StageManager.connectEntity(parent, newNode);
+      // 重新排列树形节点
+      const rootNodeParents = GraphMethods.getRoots(parent);
+      if (rootNodeParents.length === 1) {
+        const rootNodeParent = rootNodeParents[0];
+        if (GraphMethods.isTree(rootNodeParent)) {
+          StageAutoAlignManager.autoLayoutSelectedFastTreeModeRight(rootNodeParent);
+          // 更新选择状态
+          rootNodeParent.isSelected = false;
+          newNode.isSelected = true;
+        }
+      }
+      Stage.effectMachine.addEffects(EdgeRenderer.getConnectedEffects(parent, newNode));
+    });
+  }
+
   function addSuccessEffect() {
     const textNodes = StageManager.getTextNodes().filter((textNode) => textNode.isSelected);
     for (const textNode of textNodes) {

app/src/core/stage/stageManager/basicMethods/GraphMethods.tsx

@@ -32,19 +32,57 @@ export namespace GraphMethods {
   }
 
   /**
-   * 获取一个节点的所有父亲节点，未排除自环
+   * 获取一个节点的所有父亲节点，排除自环
    * 性能有待优化！！
    */
   export function nodeParentArray(node: ConnectableEntity): ConnectableEntity[] {
     const res: ConnectableEntity[] = [];
     for (const edge of StageManager.getLineEdges()) {
-      if (edge.target.uuid === node.uuid) {
+      if (edge.target.uuid === node.uuid && edge.target.uuid !== edge.source.uuid) {
         res.push(edge.source);
       }
     }
     return res;
   }
 
+  /**
+   * 获取反向边集
+   * @param edges
+   */
+  function getReversedEdgeDict(): Record<string, string> {
+    const res: Record<string, string> = {};
+    for (const edge of StageManager.getLineEdges()) {
+      res[edge.target.uuid] = edge.source.uuid;
+    }
+    return res;
+  }
+
+  /**
+   * 获取自己的祖宗节点
+   * @param node 节点
+   */
+  export function getRoots(node: ConnectableEntity): ConnectableEntity[] {
+    const reverseEdges = getReversedEdgeDict();
+    let rootUUID = node.uuid;
+    const visited: Set<string> = new Set(); // 用于记录已经访问过的节点，避免重复访问
+    while (reverseEdges[rootUUID] && !visited.has(rootUUID)) {
+      visited.add(rootUUID);
+      const parentUUID = reverseEdges[rootUUID];
+      const parent = StageManager.getConnectableEntityByUUID(parentUUID);
+      if (parent) {
+        rootUUID = parentUUID;
+      } else {
+        break;
+      }
+    }
+    const root = StageManager.getConnectableEntityByUUID(rootUUID);
+    if (root) {
+      return [root];
+    } else {
+      return [];
+    }
+  }
+
   export function isConnected(node: ConnectableEntity, target: ConnectableEntity): boolean {
     for (const edge of StageManager.getLineEdges()) {
       if (edge.source === node && edge.target === target) {