Austin Mroz
5610657a56
Allow for autosizing to be configured by setting a hidden field instead of dummy widget. If autosizing isn't specified for a node, instead set the initial width to the greater of the title length, or the original base node width used by ComfyUI
54 lines
1.8 KiB
JavaScript
54 lines
1.8 KiB
JavaScript
import { app } from '../../../scripts/app.js'
|
|
|
|
function addResizeHook(node, padding, useOldMin=false) {
|
|
let origOnCreated = node.onNodeCreated
|
|
node.onNodeCreated = function() {
|
|
let r = origOnCreated?.apply(this, arguments)
|
|
let size = this.computeSize();
|
|
size[0] += padding || 0;
|
|
if (useOldMin) {
|
|
//equal to LiteGraph.NODE_WIDTH*1.5*1.5
|
|
size[0] = Math.max(size[0], 315)
|
|
}
|
|
this.setSize(size);
|
|
return r
|
|
}
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "AdvancedControlNet.autosize",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
//since python_module is based off folder path,
|
|
//it could be changed by users and should only be used as fallback
|
|
if (nodeData?.name?.startsWith("ACN_")
|
|
|| nodeData.python_module == 'custom_nodes.ComfyUI-Advanced-ControlNet') {
|
|
if (nodeData?.input?.hidden?.autosize) {
|
|
addResizeHook(nodeType.prototype, nodeData.input.hidden.autosize[1]?.padding)
|
|
} else if (!nodeData?.input?.optional?.autosize) {
|
|
addResizeHook(nodeType.prototype, 0, true)
|
|
}
|
|
}
|
|
},
|
|
async getCustomWidgets() {
|
|
return {
|
|
ACNAUTOSIZE(node, inputName, inputData) {
|
|
let w = {
|
|
name : inputName,
|
|
type : "ACN.AUTOSIZE",
|
|
value : "",
|
|
options : {"serialize": false},
|
|
computeSize : function(width) {
|
|
return [0, -4];
|
|
}
|
|
}
|
|
if (!node.widgets) {
|
|
node.widgets = []
|
|
}
|
|
node.widgets.push(w)
|
|
addResizeHook(node, inputData[1].padding);
|
|
return w;
|
|
}
|
|
}
|
|
}
|
|
});
|