Add files via upload

2025-08-11 10:25:43 -07:00
parent f4ecb32f11
commit 93b3f6296d
2 changed files with 124 additions and 0 deletions
@@ -17,6 +17,9 @@ const NODE_COLORS = {
    "FashionSegmentAccessories": "segment",
    "FashionSegmentClothing": "segment",
    "BiRefNetRMBG": "segment",
+    "SAM2Segment": "segment",
+    "SAM2SegmentDiscovery": "segment",
+    "SAM2SegmentDiscoveryAdv": "segment",

    // Utility nodes
    "AILab_LoadImage": "utility",
@@ -0,0 +1,121 @@
+/**
+ * COLOR Widget for ComfyUI
+ *
+ * This integration script is licensed under the GNU General Public License v3.0 (GPL-3.0).
+ * If you incorporate or modify this code, please credit AILab as the original source:
+ * https://github.com/1038lab
+ */
+
+import { app } from "/scripts/app.js";
+
+const getContrastTextColor = (hexColor) => {
+    const hex = hexColor.replace('#', '');
+    const r = parseInt(hex.substr(0, 2), 16);
+    const g = parseInt(hex.substr(2, 2), 16);
+    const b = parseInt(hex.substr(4, 2), 16);
+    const luminance = (0.299 * r + 0.587 * g + 0.114 * b) / 255;
+    return luminance > 0.5 ? '#333333' : '#cccccc';
+};
+
+const AILabColorWidget = {
+    COLOR: (key, val) => {
+        const widget = {};
+        widget.y = 0;
+        widget.name = key;
+        widget.type = 'COLOR';
+        widget.options = { default: '#222222' };
+        widget.value = val || '#222222';
+
+        widget.draw = function (ctx, node, widgetWidth, widgetY, height) {
+            const hide = this.type !== 'COLOR' && app.canvas.ds.scale > 0.5;
+            if (hide) {
+                return;
+            }
+
+            const margin = 15;
+            const radius = 12;
+
+            ctx.fillStyle = this.value;
+            ctx.beginPath();
+            const x = margin;
+            const y = widgetY;
+            const w = widgetWidth - margin * 2;
+            const h = height;
+            ctx.moveTo(x + radius, y);
+            ctx.lineTo(x + w - radius, y);
+            ctx.quadraticCurveTo(x + w, y, x + w, y + radius);
+            ctx.lineTo(x + w, y + h - radius);
+            ctx.quadraticCurveTo(x + w, y + h, x + w - radius, y + h);
+            ctx.lineTo(x + radius, y + h);
+            ctx.quadraticCurveTo(x, y + h, x, y + h - radius);
+            ctx.lineTo(x, y + radius);
+            ctx.quadraticCurveTo(x, y, x + radius, y);
+            ctx.closePath();
+            ctx.fill();
+
+            ctx.strokeStyle = '#555';
+            ctx.lineWidth = 1;
+            ctx.stroke();
+
+            ctx.fillStyle = getContrastTextColor(this.value);
+            ctx.font = '12px sans-serif';
+            ctx.textAlign = 'center';
+
+            const text = `background_color (${this.value})`;
+            ctx.fillText(text, widgetWidth * 0.5, widgetY + height * 0.65);
+        };
+
+        widget.mouse = function (e, pos, node) {
+            if (e.type === 'pointerdown') {
+                const margin = 15;
+
+                if (pos[0] >= margin && pos[0] <= node.size[0] - margin) {
+                    const picker = document.createElement('input');
+                    picker.type = 'color';
+                    picker.value = this.value;
+
+                    picker.style.position = 'absolute';
+                    picker.style.left = '999999px';
+                    picker.style.top = '999999px';
+
+                    document.body.appendChild(picker);
+
+                    picker.addEventListener('change', () => {
+                        this.value = picker.value;
+                        node.graph._version++;
+                        node.setDirtyCanvas(true, true);
+                        picker.remove();
+                    });
+
+                    picker.click();
+                    return true;
+                }
+            }
+            return false;
+        };
+
+        widget.computeSize = function (width) {
+            return [width, 32];
+        };
+
+        return widget;
+    }
+};
+
+app.registerExtension({
+    name: "AILab.colorWidget",
+
+    getCustomWidgets() {
+        return {
+            COLOR: (node, inputName, inputData) => {
+                return {
+                    widget: node.addCustomWidget(
+                        AILabColorWidget.COLOR(inputName, inputData[1]?.default || '#222222')
+                    ),
+                    minWidth: 150,
+                    minHeight: 32,
+                };
+            }
+        };
+    }
+});