91 lines
3.2 KiB
JavaScript
91 lines
3.2 KiB
JavaScript
/**
|
|
* XB Qwen 图像编辑编码 — 动态图片输入槽位管理
|
|
* ==============================================
|
|
* 适用于 XB_TextEncodeQwenImageEdit / XB_TextEncodeQwenImageEditPlus。
|
|
* 默认1个图片输入，接入后自动新增，最多9个。
|
|
*/
|
|
|
|
import { app } from "../../scripts/app.js";
|
|
|
|
const NODE_TYPES = new Set(["XB_TextEncodeQwenImageEdit", "XB_TextEncodeQwenImageEditPlus"]);
|
|
const MAX_IMAGES = 9;
|
|
const FIXED_INPUT_NAMES = new Set(["clip", "prompt", "vae", "image1"]);
|
|
|
|
function isImageInput(inp) {
|
|
return !FIXED_INPUT_NAMES.has(inp.name) && (inp.type === "IMAGE" || inp.type === "*");
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "XB_ToolBox.QwenImageEdit",
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData) {
|
|
if (!NODE_TYPES.has(nodeData?.name)) return;
|
|
|
|
const orig = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = orig?.apply(this, arguments);
|
|
const self = this;
|
|
|
|
// Ensure one dynamic image slot exists
|
|
const hasDyn = (self.inputs || []).some(inp => isImageInput(inp));
|
|
if (!hasDyn) self.addInput("image2", "IMAGE");
|
|
|
|
const stabilize = () => {
|
|
if (!self.graph || self.__xb_removed) return;
|
|
let dirty = false;
|
|
const inputs = self.inputs;
|
|
|
|
const dynIndices = [];
|
|
for (let i = 0; i < inputs.length; i++) {
|
|
if (isImageInput(inputs[i])) dynIndices.push(i);
|
|
}
|
|
|
|
const count = dynIndices.length;
|
|
if (count > 0 && count < MAX_IMAGES) {
|
|
const lastDyn = inputs[dynIndices[count - 1]];
|
|
if (lastDyn?.link != null) {
|
|
self.addInput(`image${count + 2}`, "IMAGE");
|
|
dirty = true;
|
|
}
|
|
}
|
|
|
|
for (let di = dynIndices.length - 2; di >= 0; di--) {
|
|
const idx = dynIndices[di];
|
|
if (inputs[idx].link == null) {
|
|
self.removeInput(idx);
|
|
dirty = true;
|
|
}
|
|
}
|
|
|
|
if (dirty) self.graph.setDirtyCanvas(true, true);
|
|
if (!self.__xb_removed) {
|
|
self.__xb_timer = setTimeout(stabilize, 300);
|
|
}
|
|
};
|
|
|
|
const origConn = self.onConnectionsChange;
|
|
self.onConnectionsChange = function () {
|
|
if (origConn) origConn.apply(this, arguments);
|
|
if (self.__xb_timer) clearTimeout(self.__xb_timer);
|
|
self.__xb_timer = setTimeout(stabilize, 80);
|
|
};
|
|
|
|
const origRem = self.onRemoved;
|
|
self.onRemoved = function () {
|
|
self.__xb_removed = true;
|
|
if (self.__xb_timer) { clearTimeout(self.__xb_timer); self.__xb_timer = null; }
|
|
if (origRem) origRem.apply(this, arguments);
|
|
};
|
|
|
|
stabilize();
|
|
return r;
|
|
};
|
|
},
|
|
|
|
async loadedGraphNode(node) {
|
|
if (!NODE_TYPES.has(node.type)) return;
|
|
const hasDyn = (node.inputs || []).some(inp => isImageInput(inp));
|
|
if (!hasDyn) node.addInput("image2", "IMAGE");
|
|
},
|
|
});
|