90-RED
8246479bee
Major features:
- Adaptive drag mode with 5 fixed modes (Normal, Freeze+Check, Freeze, Hide, Heavy Deferred)
- Heavy Deferred for lag-free dragging in large workflows (250+ nodes)
- Three drag phases: moving → held → released, with priority-based re-routing during pause
- Node hard-obstacle avoidance — links can never pass through nodes
Performance:
- Progressive routing with frame-budgeted progressive batches
- Fast graph signature for geometry-change detection
- OVG rebuild deferred during large drags
- Collision re-check limited to affected links only
Fixes:
- Hover no longer triggers full-workflow redraw
- Krea2 / Ultimate SD Upscale link disappearance during drag
- Stale sticky paths now validated before reuse
- Drag-endpoint dogleg for straight-line links
New modules: adaptive-policy, drag-policy, draw-policy, profiler, progressive, ui-policy
Tests: 39 automated tests covering routing, adaptive, drag, profiler
Floating bar: Adaptive 🧠 toggle, 5 fixed-mode buttons, debug row, Nodes 2.0 switch
82 lines
2.9 KiB
JavaScript
82 lines
2.9 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();
|
|
// Capture-phase tracking still sees gestures captured by Nodes 2.0.
|
|
const beginPointer = () => {
|
|
M._pointerDown = true;
|
|
M._nodeDragActive = false;
|
|
M._dragAdaptiveMode = null;
|
|
M._dragHeavyActive = null;
|
|
M._dragLastFastSig = M.routeFastSig || "";
|
|
M._dragHiddenLinkIds.clear();
|
|
M._dragAffectedLinkIds.clear();
|
|
M._dragPauseActive = false;
|
|
M._dragPausePending = false;
|
|
M._dragPauseQueue = null;
|
|
M._dragPauseCleanupLinkIds.clear();
|
|
M._dragPauseAttemptedLinkIds.clear();
|
|
M._dragPauseCompletedLinkIds.clear();
|
|
M._dragInterruptedBatch = false;
|
|
};
|
|
const releasePointer = () => { M._pointerDown = false; };
|
|
window.addEventListener("pointerdown", beginPointer, true);
|
|
window.addEventListener("pointermove", (ev) => {
|
|
// Pointer capture can occasionally hide the original down event from an
|
|
// extension, but the held-button state remains reliable.
|
|
if (ev.buttons & 1) M._pointerDown = true;
|
|
}, true);
|
|
window.addEventListener("pointerup", releasePointer, true);
|
|
window.addEventListener("pointercancel", releasePointer, true);
|
|
window.addEventListener("blur", releasePointer);
|
|
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);
|
|
};
|
|
},
|
|
});
|