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

manual set default link color

Browse Source
This commit is contained in:
EllangoK 2023-04-08 09:28:35 -04:00
parent 7bce83aa03
commit 5bb1358dd6
2 changed files with 73 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
web/extensions/core/colorPalette.js
View File

@ -238,11 +238,11 @@ app.registerExtension({
if (colorPalette.colors) {
if (colorPalette.colors.node_slot) {
Object.assign(app.canvas.default_connection_color_byType, colorPalette.colors.node_slot);
app.canvas.draw(true, true);
}
if (colorPalette.colors.litegraph_base) {
// Everything updates correctly in the loop, except the Node Title for some reason
app.canvas.node_title_color = colorPalette.colors.litegraph_base.NODE_TITLE_COLOR;
app.canvas.default_link_color = colorPalette.colors.litegraph_base.LINK_COLOR;
for (const key in colorPalette.colors.litegraph_base) {
if (colorPalette.colors.litegraph_base.hasOwnProperty(key) && LiteGraph.hasOwnProperty(key)) {
@ -250,6 +250,7 @@ app.registerExtension({
}
}
}
app.canvas.draw(true, true);
}
};

71
web/scripts/test.js Normal file
View File

@ -0,0 +1,71 @@
(function() {
var LGraphCanvas = LiteGraph.LGraphCanvas;
var LGraph = LiteGraph.LGraph;
// Save the original renderLink function
var originalRenderLink = LGraphCanvas.prototype.renderLink;
// Save the original connect function
var originalConnect = LGraph.prototype.connect;
// Override the connect function
LGraph.prototype.connect = function (
origin_slot,
target_slot,
options
) {
var origin_id = origin_slot[0];
var target_id = target_slot[0];
var origin_node = this.getNodeById(origin_id);
var target_node = this.getNodeById(target_id);
if (origin_node && target_node) {
var output_slot = origin_slot[1];
var output_slot_info = origin_node.getOutputInfo(output_slot);
console.log(output_slot_info)
if (output_slot_info) {
options = options || {};
options.color = output_slot_info.label_color || null;
}
}
return originalConnect.call(this, origin_slot, target_slot, options);
};
// Override the renderLink function
LGraphCanvas.prototype.renderLink = function (
ctx,
a,
b,
link,
skip_border,
flow,
color,
start_dir,
end_dir,
num_sublines
) {
if (link && link.color) {
color = link.color;
}
// Call the original renderLink function with the new color
originalRenderLink.call(
this,
ctx,
a,
b,
link,
skip_border,
flow,
color,
start_dir,
end_dir,
num_sublines
);
};
})();