wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-11 14:02:37 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

created Export option for nodes

Browse Source
This commit is contained in:
Sammy Franklin 2023-10-06 20:29:48 -07:00
parent ae3e4e9ad8
commit 0f39cfe403
1 changed files with 68 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12949,6 +12949,69 @@ LGraphNode.prototype.executeAction = function(action)
return false;
};
LGraphCanvas.onMenuNodeExport = function(value, options, e, menu, node) {
if (!node) {
throw "no node passed";
}
const getOption = ({ name, type }, pinType) => {
const color = LGraphCanvas.link_type_colors[type];
const innerHtml = `<span style='display: block; padding-left: 4px; background-color: ${color}; color: white'>${name}</span>`;
return {
name,
type: pinType,
content: innerHtml
};
};
let inputs = node.inputs ?? [];
let outputs = node.outputs ?? [];
if (node.properties?.exports?.inputs) {
inputs = inputs.filter(({name}) => !node.properties.exports.inputs.includes(name));
}
if (node.properties?.exports?.outputs) {
outputs = outputs.filter(({name}) => !node.properties.exports.outputs.includes(name));
}
const exportableVars = [
...inputs.map((node) => getOption(node, "input")),
null,
...outputs.map((node) => getOption(node, "output")),
];
new LiteGraph.ContextMenu(exportableVars, {
event: e,
callback: inner_clicked,
parentMenu: menu,
node: node
});
function inner_clicked(v) {
if (!node) {
return;
}
node.graph.beforeChange(/*?*/); //node
const exportVar = () => {
if (!node.properties.exports) {
node.properties.exports = { inputs: [], outputs: [] };
}
console.log(v);
if (v.type == "input") {
node.properties.exports.inputs.push(v.name);
} else if (v.type == "output") {
node.properties.exports.outputs.push(v.name);
}
};
exportVar();
node.graph.afterChange(/*?*/); //node
node.setDirtyCanvas(true);
}
return false;
};
LGraphCanvas.onMenuNodeRemove = function(value, options, e, menu, node) {
if (!node) {
throw "no node passed";
@ -13158,6 +13221,11 @@ LGraphNode.prototype.executeAction = function(action)
has_submenu: true,
callback: LGraphCanvas.onMenuNodeShapes
},
{
content: "Export",
has_submenu: true,
callback: LGraphCanvas.onMenuNodeExport
},
null
);
}