298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restore original size after add/remove input

Browse Source
This commit is contained in:
pythongosssss 2023-03-24 16:36:11 +00:00
parent b13539c3dd
commit 97198f5b74
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
web/extensions/core/widgetInputs.js
View File

@ -54,14 +54,22 @@ function convertToInput(node, widget, config) {
const { linkType } = getWidgetType(config); const { linkType } = getWidgetType(config);
// Add input and store widget config for creating on primitive node // Add input and store widget config for creating on primitive node
const sz = node.size;
node.addInput(widget.name, linkType, { node.addInput(widget.name, linkType, {
widget: { name: widget.name, config }, widget: { name: widget.name, config },
}); });
// Restore original size but grow if needed
node.setSize([Math.max(sz[0], node.size[0]), Math.max(sz[1], node.size[1])]);
} }
function convertToWidget(node, widget) { function convertToWidget(node, widget) {
showWidget(widget); showWidget(widget);
const sz = node.size;
node.removeInput(node.inputs.findIndex((i) => i.widget?.name === widget.name)); node.removeInput(node.inputs.findIndex((i) => i.widget?.name === widget.name));
// Restore original size but grow if needed
node.setSize([Math.max(sz[0], node.size[0]), Math.max(sz[1], node.size[1])]);
} }
function getWidgetType(config) { function getWidgetType(config) {