xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-14 15:24:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

when selected, display links between GetNodes & SetNode

Browse Source
This commit is contained in:
Maoxie 2024-04-19 18:41:07 +08:00
parent 85724229f4
commit fafaa60f37
1 changed files with 82 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
web/js/setgetnodes.js
View File

@ -41,6 +41,7 @@ app.registerExtension({
class SetNode { class SetNode {
defaultVisibility = true; defaultVisibility = true;
serialize_widgets = true; serialize_widgets = true;
drawConnection = false;
constructor() { constructor() {
if (!this.properties) { if (!this.properties) {
this.properties = { this.properties = {
@ -213,6 +214,49 @@ app.registerExtension({
} }
}) })
} }
onSelected() {
this.drawConnection = true;
}
onDeselected() {
this.drawConnection = false;
}
onDrawForeground(ctx, lGraphCanvas) {
if (this.drawConnection) {
this._drawVirtualLinks(lGraphCanvas, ctx);
}
}
onDrawCollapsed(ctx, lGraphCanvas) {
if (this.drawConnection) {
this._drawVirtualLinks(lGraphCanvas, ctx);
}
}
_drawVirtualLinks(lGraphCanvas, ctx) {
const getters = this.findGetters(this.graph);
if (!getters?.length) return;
// draw the virtual connection from SetNode to GetNode
let start_node_slotpos = [
this.size[0],
LiteGraph.NODE_TITLE_HEIGHT * 0.5,
];
for (const getter of getters) {
let end_node_slotpos = this.getConnectionPos(false, 0);
end_node_slotpos = [
getter.pos[0] - end_node_slotpos[0] + this.size[0],
getter.pos[1] - end_node_slotpos[1],
];
lGraphCanvas.renderLink(
ctx,
start_node_slotpos,
end_node_slotpos,
null,
false,
null,
"#FFF",
LiteGraph.RIGHT,
LiteGraph.LEFT
);
}
}
} }
LiteGraph.registerNodeType( LiteGraph.registerNodeType(
@ -233,6 +277,7 @@ app.registerExtension({
defaultVisibility = true; defaultVisibility = true;
serialize_widgets = true; serialize_widgets = true;
drawConnection = false;
constructor() { constructor() {
if (!this.properties) { if (!this.properties) {
@ -337,6 +382,42 @@ app.registerExtension({
} }
onAdded(graph) { onAdded(graph) {
} }
onSelected() {
this.drawConnection = true;
}
onDeselected() {
this.drawConnection = false;
}
onDrawForeground(ctx, lGraphCanvas) {
if (this.drawConnection) {
this._drawVirtualLink(lGraphCanvas, ctx);
}
}
onDrawCollapsed(ctx, lGraphCanvas) {
if (this.drawConnection) {
this._drawVirtualLink(lGraphCanvas, ctx);
}
}
_drawVirtualLink(lGraphCanvas, ctx) {
const setter = this.findSetter(this.graph);
if (!setter) return;
// draw the virtual connection from SetNode to GetNode
let start_node_slotpos = setter.getConnectionPos(false, 0);
start_node_slotpos = [
start_node_slotpos[0] - this.pos[0],
start_node_slotpos[1] - this.pos[1],
];
let end_node_slotpos = [0, -LiteGraph.NODE_TITLE_HEIGHT * 0.5];
lGraphCanvas.renderLink(
ctx,
start_node_slotpos,
end_node_slotpos,
null,
false,
null,
"#FFF"
);
}
} }
LiteGraph.registerNodeType( LiteGraph.registerNodeType(