Spaces:
Running
on
Zero
Running
on
Zero
import { app } from "../../../scripts/app.js"; | |
app.registerExtension({ | |
name: "pysssss.SwapResolution", | |
async beforeRegisterNodeDef(nodeType, nodeData) { | |
const inputs = { ...nodeData.input?.required, ...nodeData.input?.optional }; | |
if (inputs.width && inputs.height) { | |
const origGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions; | |
nodeType.prototype.getExtraMenuOptions = function (_, options) { | |
const r = origGetExtraMenuOptions?.apply?.(this, arguments); | |
options.push( | |
{ | |
content: "Swap width/height", | |
callback: () => { | |
const w = this.widgets.find((w) => w.name === "width"); | |
const h = this.widgets.find((w) => w.name === "height"); | |
const a = w.value; | |
w.value = h.value; | |
h.value = a; | |
app.graph.setDirtyCanvas(true); | |
}, | |
}, | |
null | |
); | |
return r; | |
}; | |
} | |
}, | |
}); | |