90-RED
9317aa3c2a
- Orthogonal edge routing (OVG + A*, Wybrow GD'09) - Flow animation (dots/pills/arrows/ovals/dashes) - Hover highlight + select dim - Incremental rerouting with path stickiness - Floating button bar - Full ComfyUI settings panel - Subgraph compatible
55 lines
1.7 KiB
JavaScript
55 lines
1.7 KiB
JavaScript
// ComfyUI-LinkRouter — object-avoiding orthogonal link routing.
|
|
//
|
|
// Links are routed with an Orthogonal Visibility Graph + A* search
|
|
// (Wybrow et al., GD'09 — the libavoid algorithm).
|
|
//
|
|
// All options live in ComfyUI Settings under the LinkRouter category.
|
|
|
|
import { app } from "../../scripts/app.js";
|
|
import { api } from "../../scripts/api.js";
|
|
|
|
import { M } from "./state.js";
|
|
import { registerSettings, setRefreshBar } from "./settings.js";
|
|
import { drawAll } from "./draw.js";
|
|
import { buildUI, refreshBar, watchHover, linksHidden } from "./ui.js";
|
|
|
|
// Wire the refresh-bar callback (settings.js => ui.js without circular import)
|
|
setRefreshBar(refreshBar);
|
|
|
|
// --- execution-state watcher ---
|
|
|
|
function watchExecution() {
|
|
try {
|
|
api.addEventListener("execution_start", () => { M.running = true; });
|
|
const done = () => { M.running = false; };
|
|
api.addEventListener("execution_success", done);
|
|
api.addEventListener("execution_error", done);
|
|
api.addEventListener("execution_interrupted", done);
|
|
api.addEventListener("status", (ev) => {
|
|
const q = ev?.detail?.exec_info?.queue_remaining;
|
|
if (q === 0) M.running = false;
|
|
});
|
|
} catch (e) {
|
|
console.warn("[LinkRouter] execution watch unavailable", e);
|
|
}
|
|
}
|
|
|
|
// --- hook into ComfyUI ---
|
|
|
|
app.registerExtension({
|
|
name: "LinkRouter",
|
|
setup() {
|
|
registerSettings();
|
|
buildUI();
|
|
watchHover();
|
|
watchExecution();
|
|
const proto = LGraphCanvas.prototype;
|
|
const original = proto.drawConnections;
|
|
proto.drawConnections = function (ctx) {
|
|
if (linksHidden(this)) return; // official "hide links" wins
|
|
if (!M.S.enabled) return original.call(this, ctx);
|
|
if (drawAll(this, ctx) === false) return original.call(this, ctx);
|
|
};
|
|
},
|
|
});
|