wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-13 23:12:35 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Few tweaks

Browse Source
This commit is contained in:
pythongosssss 2023-10-12 20:15:16 +01:00
parent 89579c03ba
commit 68ef595de4
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests-ui/utils/ezgraph.js
View File

@ -9,7 +9,7 @@
* @typedef { import("../../web/types/litegraph").INodeInputSlot } INodeInputSlot * @typedef { import("../../web/types/litegraph").INodeInputSlot } INodeInputSlot
* @typedef { import("../../web/types/litegraph").INodeOutputSlot } INodeOutputSlot * @typedef { import("../../web/types/litegraph").INodeOutputSlot } INodeOutputSlot
* @typedef { InstanceType<LG["LGraphNode"]> & { widgets?: Array<IWidget> } } LGNode * @typedef { InstanceType<LG["LGraphNode"]> & { widgets?: Array<IWidget> } } LGNode
* @typedef { (...args: EzOutput[] | [...EzOutput[], Record<string, unknown>]) => Array<EzOutput> & { $: EzNode, node: LG["LGraphNode"]} } EzNodeFactory * @typedef { (...args: EzOutput[] | [...EzOutput[], Record<string, unknown>]) => EzNode } EzNodeFactory
*/ */
class EzConnection { class EzConnection {
@ -175,6 +175,13 @@ class EzWidget {
return this.widget.type === "converted-widget"; return this.widget.type === "converted-widget";
} }
getConvertedInput() {
if (!this.isConvertedToInput)
throw new Error(`Widget ${this.widget.name} is not converted to input.`);
return this.node.inputs.find(inp => inp.input["widget"]?.name === this.widget.name);
}
convertToWidget() { convertToWidget() {
if (!this.isConvertedToInput) if (!this.isConvertedToInput)
throw new Error(`Widget ${this.widget.name} cannot be converted as it is already a widget.`); throw new Error(`Widget ${this.widget.name} cannot be converted as it is already a widget.`);
@ -279,6 +286,10 @@ class EzGraph {
this.app.graph.arrange(); this.app.graph.arrange();
} }
stringify() {
return JSON.stringify(this.app.graph.serialize(), undefined, "\t");
}
/** /**
* @param { number | LGNode | EzNode } obj * @param { number | LGNode | EzNode } obj
* @returns { EzNode } * @returns { EzNode }
@ -359,7 +370,6 @@ export const Ez = {
*/ */
return function (...args) { return function (...args) {
const ezNode = new EzNode(app, node); const ezNode = new EzNode(app, node);
const outputs = ezNode.outputs;
const inputs = ezNode.inputs; const inputs = ezNode.inputs;
let slot = 0; let slot = 0;
@ -374,7 +384,7 @@ export const Ez = {
} }
} }
return outputs; return ezNode;
}; };
}, },
} }