wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

improve litegraph.core.js to include associated groups when 'Save Selected as Template'

Browse Source
This commit is contained in:
hku 2024-02-12 11:06:51 +08:00
parent 02409c30d9
commit c3264b85c1
1 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
web/lib/litegraph.core.js
View File

@ -7139,7 +7139,8 @@ LGraphNode.prototype.executeAction = function(action)
LGraphCanvas.prototype.copyToClipboard = function(nodes) { LGraphCanvas.prototype.copyToClipboard = function(nodes) {
var clipboard_info = { var clipboard_info = {
nodes: [], nodes: [],
links: [] links: [],
groups: []
}; };
var index = 0; var index = 0;
var selected_nodes_array = []; var selected_nodes_array = [];
@ -7153,6 +7154,8 @@ LGraphNode.prototype.executeAction = function(action)
index += 1; index += 1;
} }
var _selected_groups = [];
for (var i = 0; i < selected_nodes_array.length; ++i) { for (var i = 0; i < selected_nodes_array.length; ++i) {
var node = selected_nodes_array[i]; var node = selected_nodes_array[i];
var cloned = node.clone(); var cloned = node.clone();
@ -7162,6 +7165,15 @@ LGraphNode.prototype.executeAction = function(action)
continue; continue;
} }
clipboard_info.nodes.push(cloned.serialize()); clipboard_info.nodes.push(cloned.serialize());
for (var group of this.graph._groups) {
group.recomputeInsideNodes();
if (group._nodes?.includes(node) && (!_selected_groups.includes(group))) {
_selected_groups.push(group);
clipboard_info.groups.push(group.serialize());
}
}
if (node.inputs && node.inputs.length) { if (node.inputs && node.inputs.length) {
for (var j = 0; j < node.inputs.length; ++j) { for (var j = 0; j < node.inputs.length; ++j) {
var input = node.inputs[j]; var input = node.inputs[j];
@ -7243,6 +7255,21 @@ LGraphNode.prototype.executeAction = function(action)
nodes.push(node); nodes.push(node);
} }
} }
//restore groups
if (clipboard_info.groups) {
for (var i = 0; i < clipboard_info.groups.length; ++i) {
var group = new LiteGraph.LGraphGroup();
var group_data = clipboard_info.groups[i];
group.configure(group_data);
group.pos[0] += this.graph_mouse[0] - posMin[0]; //+= 5;
group.pos[1] += this.graph_mouse[1] - posMin[1]; //+= 5;
app.graph.add(group,{doProcessChange:false});
}
}
//create links //create links
for (var i = 0; i < clipboard_info.links.length; ++i) { for (var i = 0; i < clipboard_info.links.length; ++i) {