MachinePainting
271b24849c
- Aligned pyproject.toml and requirements.txt to opencv-python-headless (resolves install conflicts on headless servers) - Replaced deprecated Image.LANCZOS with Image.Resampling.LANCZOS for Pillow 11+ compatibility - Pinned numpy<3 to avoid breakage when NumPy 3 ships - Standardized JS frontend imports to absolute /scripts/app.js paths - Added emoji icon to [tool.comfy] registry metadata
79 lines
2.8 KiB
JavaScript
79 lines
2.8 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
|
|
// Selective Color Pro - Reset button
|
|
app.registerExtension({
|
|
name: "MachinePainting.SelectiveColorPro",
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name !== "SelectiveColorPro") return;
|
|
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
onNodeCreated?.apply(this, arguments);
|
|
|
|
// Set initial node size with room for button
|
|
this.size[0] = Math.max(this.size[0], 300);
|
|
this.size[1] = Math.max(this.size[1], 320);
|
|
};
|
|
|
|
const onDrawForeground = nodeType.prototype.onDrawForeground;
|
|
nodeType.prototype.onDrawForeground = function(ctx) {
|
|
onDrawForeground?.apply(this, arguments);
|
|
|
|
if (this.flags.collapsed) return;
|
|
|
|
const nodeWidth = this.size[0];
|
|
const nodeHeight = this.size[1];
|
|
|
|
// Reset button at bottom of node with padding
|
|
const btnW = 100;
|
|
const btnH = 26;
|
|
const btnX = (nodeWidth - btnW) / 2;
|
|
const btnY = nodeHeight - btnH - 15;
|
|
|
|
this.resetBtn = { x: btnX, y: btnY, w: btnW, h: btnH };
|
|
|
|
// Button background
|
|
ctx.fillStyle = "#4a4a4a";
|
|
ctx.beginPath();
|
|
ctx.roundRect(btnX, btnY, btnW, btnH, 5);
|
|
ctx.fill();
|
|
|
|
// Button border
|
|
ctx.strokeStyle = "#666";
|
|
ctx.lineWidth = 1;
|
|
ctx.stroke();
|
|
|
|
// Button text
|
|
ctx.fillStyle = "#ddd";
|
|
ctx.font = "12px Arial";
|
|
ctx.textAlign = "center";
|
|
ctx.textBaseline = "middle";
|
|
ctx.fillText("Reset All", btnX + btnW / 2, btnY + btnH / 2);
|
|
};
|
|
|
|
const onMouseDown = nodeType.prototype.onMouseDown;
|
|
nodeType.prototype.onMouseDown = function(e, pos, canvas) {
|
|
if (this.resetBtn) {
|
|
const btn = this.resetBtn;
|
|
const mx = pos[0], my = pos[1];
|
|
|
|
if (mx >= btn.x && mx <= btn.x + btn.w && my >= btn.y && my <= btn.y + btn.h) {
|
|
// Reset all sliders to 0
|
|
if (this.widgets) {
|
|
for (const w of this.widgets) {
|
|
if (["cyan", "magenta", "yellow", "black"].includes(w.name)) {
|
|
w.value = 0;
|
|
}
|
|
}
|
|
}
|
|
this.setDirtyCanvas(true, true);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return onMouseDown?.apply(this, arguments);
|
|
};
|
|
}
|
|
});
|