298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Made node non collapsable

Browse Source 
Fixed color of first input link
This commit is contained in:
pythongosssss 2023-03-16 14:08:11 +00:00
parent 97f3c23036
commit 37b70d7987
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
web/extensions/core/rerouteNode.js
View File

@ -34,6 +34,7 @@ app.registerExtension({
let currentNode = this;
let updateNodes = [];
let inputType = null;
let inputNode = null;
while (currentNode) {
updateNodes.unshift(currentNode);
const linkId = currentNode.inputs[0].link;
@ -46,6 +47,7 @@ app.registerExtension({
currentNode = node;
} else {
// We've found the end
inputNode = currentNode;
inputType = node.outputs[link.origin_slot].type;
break;
}
@ -93,6 +95,7 @@ app.registerExtension({
}
const displayType = inputType || outputType || "*";
const color = LGraphCanvas.link_type_colors[displayType];
// Update the types of each node
for (const node of updateNodes) {
@ -103,11 +106,14 @@ app.registerExtension({
node.outputs[0].name = node.properties.showOutputText ? displayType : "";
node.size = node.computeSize();
const color = LGraphCanvas.link_type_colors[displayType];
for (const l of node.outputs[0].links || []) {
app.graph.links[l].color = color;
}
}
if (inputNode) {
app.graph.links[inputNode.inputs[0].link].color = color;
}
};
this.clone = function () {
@ -173,6 +179,7 @@ app.registerExtension({
Object.assign(RerouteNode, {
title_mode: LiteGraph.NO_TITLE,
title: "Reroute",
collapsable: false,
})
);