Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-03-10 15:40:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Refactor] LGraphNode.drawCollapsedSlots (#500)

Browse Source
This commit is contained in:
Chenlei Hu
2025-02-09 11:02:20 -05:00
committed by GitHub
parent 01b8ae5bb8
commit 120c606f07
2 changed files with 50 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/LGraphCanvas.ts
View File

@@ -4782,55 +4782,7 @@ export class LGraphCanvas implements ConnectionColorContext {
this.drawNodeWidgets(node, max_y, ctx)
} else if (this.render_collapsed_slots) {
// if collapsed
let input_slot = null
let output_slot = null
let slot
// get first connected slot to render
if (node.inputs) {
for (let i = 0; i < node.inputs.length; i++) {
slot = node.inputs[i]
if (slot.link == null) {
continue
}
input_slot = slot
break
}
}
if (node.outputs) {
for (let i = 0; i < node.outputs.length; i++) {
slot = node.outputs[i]
if (!slot.links || !slot.links.length) {
continue
}
output_slot = slot
}
}
if (input_slot) {
let x = 0
let y = LiteGraph.NODE_TITLE_HEIGHT * -0.5 // center
if (horizontal) {
x = node._collapsed_width * 0.5
y = -LiteGraph.NODE_TITLE_HEIGHT
}
toClass(NodeInputSlot, input_slot).drawCollapsed(ctx, {
pos: [x, y],
})
}
if (output_slot) {
let x = node._collapsed_width
let y = LiteGraph.NODE_TITLE_HEIGHT * -0.5 // center
if (horizontal) {
x = node._collapsed_width * 0.5
y = 0
}
toClass(NodeOutputSlot, output_slot).drawCollapsed(ctx, {
pos: [x, y],
})
}
node.drawCollapsedSlots(ctx)
}
if (node.clip_area) {

49
src/LGraphNode.ts
View File

@@ -3146,4 +3146,53 @@ export class LGraphNode implements Positionable, IPinnable {
}
ctx.restore()
}
/**
* When {@link LGraphNode.collapsed} is `true`, this method draws the node's collapsed slots.
*/
drawCollapsedSlots(ctx: CanvasRenderingContext2D): void {
// if collapsed
let input_slot: INodeInputSlot | null = null
let output_slot: INodeOutputSlot | null = null
// get first connected slot to render
for (const slot of this.inputs ?? []) {
if (slot.link == null) {
continue
}
input_slot = slot
break
}
for (const slot of this.outputs ?? []) {
if (!slot.links || !slot.links.length) {
continue
}
output_slot = slot
break
}
if (input_slot) {
let x = 0
let y = LiteGraph.NODE_TITLE_HEIGHT * -0.5 // center
if (this.horizontal) {
x = this._collapsed_width * 0.5
y = -LiteGraph.NODE_TITLE_HEIGHT
}
toClass(NodeInputSlot, input_slot).drawCollapsed(ctx, {
pos: [x, y],
})
}
if (output_slot) {
let x = this._collapsed_width
let y = LiteGraph.NODE_TITLE_HEIGHT * -0.5 // center
if (this.horizontal) {
x = this._collapsed_width * 0.5
y = 0
}
toClass(NodeOutputSlot, output_slot).drawCollapsed(ctx, {
pos: [x, y],
})
}
}
}