| 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; | |
| }; | |
| } | |
| }, | |
| }); | |