Upstream progressbar rendering (#871)

Upstreams progress bar rendering from ComfyUI_frontend.
This commit is contained in:
Chenlei Hu
2025-03-30 17:14:09 -04:00
committed by GitHub
parent 8a9b1c7071
commit 30a7f19a64
5 changed files with 34 additions and 0 deletions

View File

@@ -262,6 +262,11 @@ export class LGraphNode implements Positionable, IPinnable, IColorable {
*/
strokeStyles: Record<string, (this: LGraphNode) => IDrawBoundingOptions | undefined>
/**
* The progress of node execution. Used to render a progress bar. Value between 0 and 1.
*/
progress?: number
exec_version?: number
action_call?: string
execute_triggered?: number
@@ -3657,4 +3662,22 @@ export class LGraphNode implements Positionable, IPinnable, IColorable {
this.layoutSlot(actualSlot, { slotIndex: slot.index })
}
}
/**
* Draws a progress bar on the node.
* @param ctx The canvas context to draw on
*/
drawProgressBar(ctx: CanvasRenderingContext2D): void {
if (!this.progress) return
const originalFillStyle = ctx.fillStyle
ctx.fillStyle = "green"
ctx.fillRect(
0,
0,
this.width * this.progress,
6,
)
ctx.fillStyle = originalFillStyle
}
}