94 lines
2.3 KiB
TypeScript
94 lines
2.3 KiB
TypeScript
import { TreeGraphData } from "@antv/g6/lib/types"
|
|
import { BehaviorNodeModel, GraphNodeModel } from "./BehaviorTreeModel";
|
|
import Settings from "../main-process/Settings";
|
|
import { remote } from "electron";
|
|
|
|
export const cloneNodeData = (nodeData: GraphNodeModel) => {
|
|
const newData: GraphNodeModel = {
|
|
id: nodeData.id + "new",
|
|
name: nodeData.name,
|
|
}
|
|
return newData;
|
|
}
|
|
|
|
export const refreshNodeId = (nodeData: GraphNodeModel, id?: number) => {
|
|
if (!id) {
|
|
id = 1;
|
|
}
|
|
nodeData.id = (id++).toString();
|
|
if (nodeData.children) {
|
|
nodeData.children.forEach(child => {
|
|
id = refreshNodeId(child, id);
|
|
})
|
|
}
|
|
return id;
|
|
}
|
|
|
|
export const calcTreeNodeSize = (treeNode: GraphNodeModel) => {
|
|
var height = 40;
|
|
const updateHeight = (arr: any) => {
|
|
if(arr && arr.length > 0) {
|
|
height += 30;
|
|
}
|
|
}
|
|
updateHeight(treeNode.args);
|
|
updateHeight(treeNode.input);
|
|
updateHeight(treeNode.output);
|
|
return [150, height];
|
|
}
|
|
|
|
export const createTreeData = (bNode: BehaviorNodeModel) => {
|
|
const treeData: GraphNodeModel = {
|
|
id: bNode.id.toString(),
|
|
name: bNode.name,
|
|
desc: bNode.desc,
|
|
args: bNode.args,
|
|
input: bNode.input,
|
|
output: bNode.output,
|
|
}
|
|
treeData.size = calcTreeNodeSize(treeData);
|
|
if (bNode.children) {
|
|
treeData.children = [];
|
|
bNode.children.forEach(child => {
|
|
treeData.children.push(createTreeData(child));
|
|
})
|
|
}
|
|
calcTreeNodeSize(treeData);
|
|
return treeData;
|
|
}
|
|
|
|
export const findParent = (node: TreeGraphData, id: string): TreeGraphData | null => {
|
|
if (node.children) {
|
|
for (let child of node.children) {
|
|
if (child.id == id) {
|
|
return node;
|
|
} else {
|
|
let parent = findParent(child, id);
|
|
if (parent) {
|
|
return parent;
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
export const findFromAllChildren = (node: TreeGraphData, id: string): TreeGraphData | null => {
|
|
if (node.id == id) {
|
|
return node;
|
|
}
|
|
if (node.children) {
|
|
for (let child of node.children) {
|
|
let found = findFromAllChildren(child, id);
|
|
if (found) {
|
|
return found;
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
export const getRemoteSettings = () => {
|
|
return remote.getGlobal("settings") as Settings;
|
|
} |