| import { app } from "../../scripts/app.js"; |
| import { ComfyWidgets } from "../../scripts/widgets.js"; |
| import { RgthreeBaseServerNode } from "./base_node.js"; |
| import { rgthree } from "./rgthree.js"; |
| import { addConnectionLayoutSupport } from "./utils.js"; |
| import { NodeTypesString } from "./constants.js"; |
| const LAST_SEED_BUTTON_LABEL = "♻️ (Use Last Queued Seed)"; |
| const SPECIAL_SEED_RANDOM = -1; |
| const SPECIAL_SEED_INCREMENT = -2; |
| const SPECIAL_SEED_DECREMENT = -3; |
| const SPECIAL_SEEDS = [SPECIAL_SEED_RANDOM, SPECIAL_SEED_INCREMENT, SPECIAL_SEED_DECREMENT]; |
| class RgthreeSeed extends RgthreeBaseServerNode { |
| constructor(title = RgthreeSeed.title) { |
| super(title); |
| this.serialize_widgets = true; |
| this.logger = rgthree.newLogSession(`[Seed]`); |
| this.lastSeed = undefined; |
| this.serializedCtx = {}; |
| this.lastSeedValue = null; |
| this.randMax = 1125899906842624; |
| this.randMin = 0; |
| this.randomRange = 1125899906842624; |
| this.handleApiHijackingBound = this.handleApiHijacking.bind(this); |
| rgthree.addEventListener("comfy-api-queue-prompt-before", this.handleApiHijackingBound); |
| } |
| onRemoved() { |
| rgthree.addEventListener("comfy-api-queue-prompt-before", this.handleApiHijackingBound); |
| } |
| configure(info) { |
| var _a; |
| super.configure(info); |
| if ((_a = this.properties) === null || _a === void 0 ? void 0 : _a["showLastSeed"]) { |
| this.addLastSeedValue(); |
| } |
| } |
| async handleAction(action) { |
| if (action === "Randomize Each Time") { |
| this.seedWidget.value = SPECIAL_SEED_RANDOM; |
| } |
| else if (action === "Use Last Queued Seed") { |
| this.seedWidget.value = this.lastSeed != null ? this.lastSeed : this.seedWidget.value; |
| this.lastSeedButton.name = LAST_SEED_BUTTON_LABEL; |
| this.lastSeedButton.disabled = true; |
| } |
| } |
| onNodeCreated() { |
| var _a; |
| (_a = super.onNodeCreated) === null || _a === void 0 ? void 0 : _a.call(this); |
| for (const [i, w] of this.widgets.entries()) { |
| if (w.name === "seed") { |
| this.seedWidget = w; |
| this.seedWidget.value = SPECIAL_SEED_RANDOM; |
| } |
| else if (w.name === "control_after_generate") { |
| this.widgets.splice(i, 1); |
| } |
| } |
| let step = this.seedWidget.options.step || 1; |
| this.randMax = Math.min(1125899906842624, this.seedWidget.options.max); |
| this.randMin = Math.max(0, this.seedWidget.options.min); |
| this.randomRange = (this.randMax - Math.max(0, this.randMin)) / (step / 10); |
| this.addWidget("button", "🎲 Randomize Each Time", null, () => { |
| this.seedWidget.value = SPECIAL_SEED_RANDOM; |
| }, { serialize: false }); |
| this.addWidget("button", "🎲 New Fixed Random", null, () => { |
| this.seedWidget.value = |
| Math.floor(Math.random() * this.randomRange) * (step / 10) + this.randMin; |
| }, { serialize: false }); |
| this.lastSeedButton = this.addWidget("button", LAST_SEED_BUTTON_LABEL, null, () => { |
| this.seedWidget.value = this.lastSeed != null ? this.lastSeed : this.seedWidget.value; |
| this.lastSeedButton.name = LAST_SEED_BUTTON_LABEL; |
| this.lastSeedButton.disabled = true; |
| }, { width: 50, serialize: false }); |
| this.lastSeedButton.disabled = true; |
| } |
| getExtraMenuOptions(canvas, options) { |
| var _a; |
| (_a = super.getExtraMenuOptions) === null || _a === void 0 ? void 0 : _a.apply(this, [...arguments]); |
| options.splice(options.length - 1, 0, { |
| content: "Show/Hide Last Seed Value", |
| callback: (_value, _options, _event, _parentMenu, _node) => { |
| this.properties["showLastSeed"] = !this.properties["showLastSeed"]; |
| if (this.properties["showLastSeed"]) { |
| this.addLastSeedValue(); |
| } |
| else { |
| this.removeLastSeedValue(); |
| } |
| }, |
| }); |
| } |
| addLastSeedValue() { |
| if (this.lastSeedValue) |
| return; |
| this.lastSeedValue = ComfyWidgets["STRING"](this, "last_seed", ["STRING", { multiline: true }], app).widget; |
| this.lastSeedValue.inputEl.readOnly = true; |
| this.lastSeedValue.inputEl.style.fontSize = "0.75rem"; |
| this.lastSeedValue.inputEl.style.textAlign = "center"; |
| this.computeSize(); |
| } |
| removeLastSeedValue() { |
| if (!this.lastSeedValue) |
| return; |
| this.lastSeedValue.inputEl.remove(); |
| this.widgets.splice(this.widgets.indexOf(this.lastSeedValue), 1); |
| this.lastSeedValue = null; |
| this.computeSize(); |
| } |
| handleApiHijacking(e) { |
| var _a, _b, _c, _d; |
| if (this.mode === LiteGraph.NEVER || this.mode === 4) { |
| return; |
| } |
| const workflow = e.detail.workflow; |
| const output = e.detail.output; |
| let workflowNode = (_b = (_a = workflow === null || workflow === void 0 ? void 0 : workflow.nodes) === null || _a === void 0 ? void 0 : _a.find((n) => n.id === this.id)) !== null && _b !== void 0 ? _b : null; |
| let outputInputs = (_c = output === null || output === void 0 ? void 0 : output[this.id]) === null || _c === void 0 ? void 0 : _c.inputs; |
| if (!workflowNode || |
| !outputInputs || |
| outputInputs[this.seedWidget.name || "seed"] === undefined) { |
| const [n, v] = this.logger.warnParts(`Node ${this.id} not found in prompt data sent to server. This may be fine if only ` + |
| `queuing part of the workflow. If not, then this could be a bug.`); |
| (_d = console[n]) === null || _d === void 0 ? void 0 : _d.call(console, ...v); |
| return; |
| } |
| const seedToUse = this.getSeedToUse(); |
| const seedWidgetndex = this.widgets.indexOf(this.seedWidget); |
| workflowNode.widgets_values[seedWidgetndex] = seedToUse; |
| outputInputs[this.seedWidget.name || "seed"] = seedToUse; |
| this.lastSeed = seedToUse; |
| if (seedToUse != this.seedWidget.value) { |
| this.lastSeedButton.name = `♻️ ${this.lastSeed}`; |
| this.lastSeedButton.disabled = false; |
| } |
| else { |
| this.lastSeedButton.name = LAST_SEED_BUTTON_LABEL; |
| this.lastSeedButton.disabled = true; |
| } |
| if (this.lastSeedValue) { |
| this.lastSeedValue.value = `Last Seed: ${this.lastSeed}`; |
| } |
| } |
| getSeedToUse() { |
| const inputSeed = this.seedWidget.value; |
| let seedToUse = null; |
| if (SPECIAL_SEEDS.includes(inputSeed)) { |
| if (typeof this.lastSeed === "number" && !SPECIAL_SEEDS.includes(this.lastSeed)) { |
| if (inputSeed === SPECIAL_SEED_INCREMENT) { |
| seedToUse = this.lastSeed + 1; |
| } |
| else if (inputSeed === SPECIAL_SEED_DECREMENT) { |
| seedToUse = this.lastSeed - 1; |
| } |
| } |
| if (seedToUse == null || SPECIAL_SEEDS.includes(seedToUse)) { |
| seedToUse = |
| Math.floor(Math.random() * this.randomRange) * |
| ((this.seedWidget.options.step || 1) / 10) + |
| this.randMin; |
| } |
| } |
| return seedToUse !== null && seedToUse !== void 0 ? seedToUse : inputSeed; |
| } |
| static setUp(comfyClass, nodeData) { |
| RgthreeBaseServerNode.registerForOverride(comfyClass, nodeData, RgthreeSeed); |
| } |
| static onRegisteredForOverride(comfyClass, ctxClass) { |
| addConnectionLayoutSupport(RgthreeSeed, app, [ |
| ["Left", "Right"], |
| ["Right", "Left"], |
| ]); |
| setTimeout(() => { |
| RgthreeSeed.category = comfyClass.category; |
| }); |
| } |
| } |
| RgthreeSeed.title = NodeTypesString.SEED; |
| RgthreeSeed.type = NodeTypesString.SEED; |
| RgthreeSeed.comfyClass = NodeTypesString.SEED; |
| RgthreeSeed.exposedActions = ["Randomize Each Time", "Use Last Queued Seed"]; |
| app.registerExtension({ |
| name: "rgthree.Seed", |
| async beforeRegisterNodeDef(nodeType, nodeData) { |
| if (nodeData.name === RgthreeSeed.type) { |
| RgthreeSeed.setUp(nodeType, nodeData); |
| } |
| }, |
| }); |
|
|
