Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-04-30 03:01:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Switch to onDrawForeground

Browse Source
This commit is contained in:
Benjamin Lu
2025-09-18 14:44:08 -07:00
parent da3a467f61
commit 20eb500020
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/renderer/core/linkInteractions/slotLinkPreviewRenderer.ts
View File

@@ -31,12 +31,12 @@ function buildContext(canvas: LGraphCanvas): LinkRenderContext {
} }
export function attachSlotLinkPreviewRenderer(canvas: LGraphCanvas) { export function attachSlotLinkPreviewRenderer(canvas: LGraphCanvas) {
const originalOnRender = canvas.onRender?.bind(canvas) const originalOnDrawForeground = canvas.onDrawForeground?.bind(canvas)
const patched = ( const patched = (
canvasElement: HTMLCanvasElement, ctx: CanvasRenderingContext2D,
ctx: CanvasRenderingContext2D area: LGraphCanvas['visible_area']
) => { ) => {
originalOnRender?.(canvasElement, ctx) originalOnDrawForeground?.(ctx, area)
const { state } = useSlotLinkDragState() const { state } = useSlotLinkDragState()
if (!state.active || !state.source) return if (!state.active || !state.source) return
@@ -59,7 +59,6 @@ export function attachSlotLinkPreviewRenderer(canvas: LGraphCanvas) {
const colour = resolveConnectingLinkColor(sourceSlot?.type) const colour = resolveConnectingLinkColor(sourceSlot?.type)
ctx.save() ctx.save()
canvas.ds.toCanvasContext(ctx)
linkRenderer.renderDraggingLink( linkRenderer.renderDraggingLink(
ctx, ctx,
@@ -74,7 +73,7 @@ export function attachSlotLinkPreviewRenderer(canvas: LGraphCanvas) {
ctx.restore() ctx.restore()
} }
canvas.onRender = patched canvas.onDrawForeground = patched
} }
function resolveSourceSlot( function resolveSourceSlot(