21 lines
956 B
JavaScript
21 lines
956 B
JavaScript
import { app } from '../../scripts/app.js';
|
|
import { bindDimensionPresets, extractResolutionFromPreset, swapDimensions } from './ui/dimensions.js';
|
|
import { mountDimensionSwap } from './ui/dimension-swap.js';
|
|
|
|
app.registerExtension({
|
|
name: 'comfyassets.WidthHeightSelector',
|
|
async beforeRegisterNodeDef(nodeType, nodeData) {
|
|
if (nodeData.name !== 'WidthHeightSelector') return;
|
|
nodeType.prototype.createSwapButton = function () { return mountDimensionSwap(this, app); };
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const result = onNodeCreated?.apply(this, arguments);
|
|
this.extractResolutionFromPreset = extractResolutionFromPreset;
|
|
this.swapDimensions = () => swapDimensions(this, app);
|
|
bindDimensionPresets(this, app);
|
|
this.createSwapButton();
|
|
return result;
|
|
};
|
|
},
|
|
});
|