Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-20 14:54:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Convert i/o slot to class in LGraphNode.configure (#506)

Browse Source
This commit is contained in:
Chenlei Hu
2025-02-09 15:23:38 -05:00
committed by GitHub
parent 997d673bd9
commit c75157e86e
2 changed files with 59 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/LGraphNode.ts
View File

@@ -596,44 +596,40 @@ export class LGraphNode implements Positionable, IPinnable {
this.title = this.constructor.title
}
if (this.inputs) {
for (let i = 0; i < this.inputs.length; ++i) {
const input = this.inputs[i]
const link = this.graph ? this.graph._links.get(input.link) : null
this.onConnectionsChange?.(NodeSlotType.INPUT, i, true, link, input)
this.onInputAdded?.(input)
}
this.inputs ??= []
this.inputs = this.inputs.map(input => toClass(NodeInputSlot, input))
for (const [i, input] of this.inputs.entries()) {
const link = this.graph ? this.graph._links.get(input.link) : null
this.onConnectionsChange?.(NodeSlotType.INPUT, i, true, link, input)
this.onInputAdded?.(input)
}
if (this.outputs) {
for (let i = 0; i < this.outputs.length; ++i) {
const output = this.outputs[i]
if (!output.links) {
continue
}
for (let j = 0; j < output.links.length; ++j) {
const link = this.graph
? this.graph._links.get(output.links[j])
: null
this.onConnectionsChange?.(NodeSlotType.OUTPUT, i, true, link, output)
}
this.onOutputAdded?.(output)
this.outputs ??= []
this.outputs = this.outputs.map(output => toClass(NodeOutputSlot, output))
for (const [i, output] of this.outputs.entries()) {
if (!output.links) {
continue
}
for (const linkId of output.links) {
const link = this.graph
? this.graph._links.get(linkId)
: null
this.onConnectionsChange?.(NodeSlotType.OUTPUT, i, true, link, output)
}
this.onOutputAdded?.(output)
}
if (this.widgets) {
for (let i = 0; i < this.widgets.length; ++i) {
const w = this.widgets[i]
for (const w of this.widgets) {
if (!w) continue
if (w.options?.property && this.properties[w.options.property] != undefined)
w.value = JSON.parse(JSON.stringify(this.properties[w.options.property]))
}
if (info.widgets_values) {
for (let i = 0; i < info.widgets_values.length; ++i) {
if (this.widgets[i]) {
this.widgets[i].value = info.widgets_values[i]
}
for (const [i, value] of (info.widgets_values ?? []).entries()) {
if (this.widgets[i]) {
this.widgets[i].value = value
}
}
}