xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-29 01:17:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'pr/50'

Browse Source
This commit is contained in:
Kijai 2024-04-24 14:31:58 +03:00
commit 2f5264d4b6
1 changed files with 82 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
web/js/setgetnodes.js
View File

@ -41,6 +41,7 @@ app.registerExtension({
class SetNode {
defaultVisibility = true;
serialize_widgets = true;
drawConnection = false;
constructor() {
if (!this.properties) {
this.properties = {
@ -252,6 +253,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(
@ -272,6 +316,7 @@ app.registerExtension({
defaultVisibility = true;
serialize_widgets = true;
drawConnection = false;
constructor() {
if (!this.properties) {
@ -306,7 +351,7 @@ app.registerExtension({
) {
this.validateLinks();
}
this.setName = function(name) {
node.widgets[0].value = name;
node.onRename();
@ -436,6 +481,42 @@ app.registerExtension({
},
);
}
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(