nodes/room3dev-ComfyUI-PixelForge
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: full real-time refresh support including orientation and loading

Browse Source
This commit is contained in:
Your Name
2026-01-20 10:45:29 +07:00
parent 8505e89542
commit 8f8ed13d0f
1 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/js/pixel_forge.js
+26 -8
View File
@@ -18,19 +18,20 @@ app.registerExtension({
const onNodeCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = function () {
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
console.log("PixelForge: Node created, initializing dropdown logic");
const aspectWidget = this.widgets.find(w => w.name === "aspect_ratio");
const orientationWidget = this.widgets.find(w => w.name === "orientation");
const divisibleWidget = this.widgets.find(w => w.name === "divisible_by");
const mpLimitWidget = this.widgets.find(w => w.name === "max_megapixels");
const resolutionWidget = this.widgets.find(w => w.name === "resolution");
resolutionWidget.options = { values: [] };
if (!resolutionWidget.options) resolutionWidget.options = { values: [] };
const updateResolutions = () => {
const aspect = aspectWidget.value;
const orientation = orientationWidget.value;
const div = parseInt(divisibleWidget.value);
const maxMp = parseInt(mpLimitWidget.value.split(" ")[0]);
console.log("PixelForge: Updating resolutions for aspect:", aspect, "divisible by:", div, "max MP:", maxMp);
const ratio = ASPECT_RATIOS[aspect];
if (!ratio) return;
@@ -42,12 +43,16 @@ app.registerExtension({
let k = div;
while (true) {
const w = ratio_w * k;
const h = ratio_h * k;
let w = ratio_w * k;
let h = ratio_h * k;
const total = w * h;
if (total > maxPixels) break;
if (orientation === "portrait") {
[w, h] = [h, w];
}
resolutions.push(`${w}×${h}`);
k += div;
}
@@ -59,7 +64,7 @@ app.registerExtension({
// Update the widget options
resolutionWidget.options.values = resolutions;
// If current value is not in the new list, pick the closest one or the first one
// If current value is not in the new list, pick the first one
if (!resolutions.includes(resolutionWidget.value)) {
resolutionWidget.value = resolutions[0];
}
@@ -70,8 +75,9 @@ app.registerExtension({
};
// Add callbacks to trigger update
const widgetsToWatch = [aspectWidget, divisibleWidget, mpLimitWidget];
const widgetsToWatch = [aspectWidget, orientationWidget, divisibleWidget, mpLimitWidget];
widgetsToWatch.forEach(widget => {
if (!widget) return;
const callback = widget.callback;
widget.callback = function () {
const result = callback ? callback.apply(this, arguments) : undefined;
@@ -81,10 +87,22 @@ app.registerExtension({
});
// Initial update
updateResolutions();
setTimeout(() => updateResolutions(), 10);
return r;
};
// Ensure it updates when workflow is loaded
nodeType.prototype.onConfigure = function () {
setTimeout(() => {
const resolutionWidget = this.widgets.find(w => w.name === "resolution");
if (resolutionWidget && this.widgets.find(w => w.name === "aspect_ratio")) {
// Trigger update after a short delay to ensure all values are loaded
const updateResolutions = this.widgets.find(w => w.name === "aspect_ratio").callback;
if (updateResolutions) updateResolutions();
}
}, 100);
};
}
}
});