wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-11 05:52:33 +08:00
Code Issues Actions 18 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").INodeOutputSlot } INodeOutputSlot
* @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 {
@ -175,6 +175,13 @@ class EzWidget {
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() {
if (!this.isConvertedToInput)
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();
}
stringify() {
return JSON.stringify(this.app.graph.serialize(), undefined, "\t");
}
/**
* @param { number | LGNode | EzNode } obj
* @returns { EzNode }
@ -359,7 +370,6 @@ export const Ez = {
*/
return function (...args) {
const ezNode = new EzNode(app, node);
const outputs = ezNode.outputs;
const inputs = ezNode.inputs;
let slot = 0;
@ -374,7 +384,7 @@ export const Ez = {
}
}
return outputs;
return ezNode;
};
},
}