| import { app } from "../../../scripts/app.js";
|
|
|
| app.registerExtension({
|
| name: "pysssss.GraphArrange",
|
| setup(app) {
|
| const orig = LGraphCanvas.prototype.getCanvasMenuOptions;
|
| LGraphCanvas.prototype.getCanvasMenuOptions = function () {
|
| const options = orig.apply(this, arguments);
|
| options.push({ content: "Arrange (float left)", callback: () => graph.arrange() });
|
| options.push({
|
| content: "Arrange (float right)",
|
| callback: () => {
|
| (function () {
|
| var margin = 50;
|
| var layout;
|
|
|
| const nodes = this.computeExecutionOrder(false, true);
|
| const columns = [];
|
|
|
|
|
| for (let i = nodes.length - 1; i >= 0; i--) {
|
| const node = nodes[i];
|
| let max = null;
|
| for (const out of node.outputs || []) {
|
| if (out.links) {
|
| for (const link of out.links) {
|
| const outNode = app.graph.getNodeById(app.graph.links[link].target_id);
|
| if (!outNode) continue;
|
| var l = outNode._level - 1;
|
| if (max === null) max = l;
|
| else if (l < max) max = l;
|
| }
|
| }
|
| }
|
| if (max != null) node._level = max;
|
| }
|
|
|
| for (let i = 0; i < nodes.length; ++i) {
|
| const node = nodes[i];
|
| const col = node._level || 1;
|
| if (!columns[col]) {
|
| columns[col] = [];
|
| }
|
| columns[col].push(node);
|
| }
|
|
|
| let x = margin;
|
|
|
| for (let i = 0; i < columns.length; ++i) {
|
| const column = columns[i];
|
| if (!column) {
|
| continue;
|
| }
|
| column.sort((a, b) => {
|
| var as = !(a.type === "SaveImage" || a.type === "PreviewImage");
|
| var bs = !(b.type === "SaveImage" || b.type === "PreviewImage");
|
| var r = as - bs;
|
| if (r === 0) r = (a.inputs?.length || 0) - (b.inputs?.length || 0);
|
| if (r === 0) r = (a.outputs?.length || 0) - (b.outputs?.length || 0);
|
| return r;
|
| });
|
| let max_size = 100;
|
| let y = margin + LiteGraph.NODE_TITLE_HEIGHT;
|
| for (let j = 0; j < column.length; ++j) {
|
| const node = column[j];
|
| node.pos[0] = layout == LiteGraph.VERTICAL_LAYOUT ? y : x;
|
| node.pos[1] = layout == LiteGraph.VERTICAL_LAYOUT ? x : y;
|
| const max_size_index = layout == LiteGraph.VERTICAL_LAYOUT ? 1 : 0;
|
| if (node.size[max_size_index] > max_size) {
|
| max_size = node.size[max_size_index];
|
| }
|
| const node_size_index = layout == LiteGraph.VERTICAL_LAYOUT ? 0 : 1;
|
| y += node.size[node_size_index] + margin + LiteGraph.NODE_TITLE_HEIGHT + j;
|
| }
|
|
|
|
|
| for (let j = 0; j < column.length; ++j) {
|
| const node = column[j];
|
| node.pos[0] += max_size - node.size[0];
|
| }
|
| x += max_size + margin;
|
| }
|
|
|
| this.setDirtyCanvas(true, true);
|
| }).apply(app.graph);
|
| },
|
| });
|
| return options;
|
| };
|
| },
|
| });
|
|
|
