mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 10:25:16 +00:00
manual set default link color
This commit is contained in:
parent
7bce83aa03
commit
5bb1358dd6
@ -238,11 +238,11 @@ app.registerExtension({
|
|||||||
if (colorPalette.colors) {
|
if (colorPalette.colors) {
|
||||||
if (colorPalette.colors.node_slot) {
|
if (colorPalette.colors.node_slot) {
|
||||||
Object.assign(app.canvas.default_connection_color_byType, 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) {
|
if (colorPalette.colors.litegraph_base) {
|
||||||
// Everything updates correctly in the loop, except the Node Title for some reason
|
// 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.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) {
|
for (const key in colorPalette.colors.litegraph_base) {
|
||||||
if (colorPalette.colors.litegraph_base.hasOwnProperty(key) && LiteGraph.hasOwnProperty(key)) {
|
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
71
web/scripts/test.js
Normal 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
|
||||||
|
);
|
||||||
|
};
|
||||||
|
})();
|
Loading…
Reference in New Issue
Block a user