node param widget visibility handler registration

2024-06-29 00:57:21 +02:00
parent 75452536eb
commit ffe0fad4d4
1 changed files with 69 additions and 0 deletions
@@ -0,0 +1,69 @@
+//@ts-check
+// import { app } from "../../../web/scripts/app.js";
+import { app } from "../../scripts/app.js";
+
+let origProps = {};
+let initialized = false;
+const HIDDEN_TAG = "tschide";
+
+function handleParamVisibility(node, countValue) {
+    for (let i = 1; i <= 50; i++) {
+        const pname = `param_${i}`;
+        const paramWidget = node.widgets.find((w) => w.name === pname);
+        if (!paramWidget) continue;
+
+        if (!origProps[pname]) {
+            origProps[pname] = { origType: paramWidget.type, origComputeSize: paramWidget.computeSize };
+        }
+
+        if (i <= countValue) {
+            paramWidget.type = origProps[paramWidget.name].origType;
+            paramWidget.computeSize = origProps[paramWidget.name].origComputeSize;
+        }
+        else {
+            paramWidget.type = HIDDEN_TAG;
+            paramWidget.computeSize = () => [0, -4];
+        }
+    }
+    const newHeight = node.computeSize()[1];
+    node.setSize([node.size[0], newHeight]);
+}
+
+function registerIndustrialMagickParamVisibility(node) {
+    for (const w of node.widgets || []) {
+        if (w.name !== "param_count") continue;
+        let countValue = w.value;
+        // Store the original descriptor if it exists
+        let originalDescriptor = Object.getOwnPropertyDescriptor(w, 'value');
+        handleParamVisibility(node, countValue);
+        Object.defineProperty(w, 'value', {
+            get() {
+                // If there's an original getter, use it. Otherwise, return widgetValue.
+                let valueToReturn = originalDescriptor && originalDescriptor.get
+                    ? originalDescriptor.get.call(w)
+                    : countValue;
+
+                return valueToReturn;
+            },
+            set(newVal) {
+                // If there's an original setter, use it. Otherwise, set widgetValue.
+                if (originalDescriptor && originalDescriptor.set) {
+                    originalDescriptor.set.call(w, newVal);
+                } else {
+                    countValue = newVal;
+                }
+                handleParamVisibility(node, countValue);
+            }
+        });
+    }
+}
+
+app.registerExtension({
+    name: "industrialmagick.paramvisibility",
+    nodeCreated(node) {
+        if (node.comfyClass === "IndustrialMagick")
+            registerIndustrialMagickParamVisibility(node);
+        setTimeout(() => { initialized = true; }, 500);
+    }
+});
+