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