Skip to content

Commit 311e44a

Browse files
tidy(ui): clean up control layers renderers, docstrings
1 parent 848ca79 commit 311e44a

File tree

3 files changed

+282
-190
lines changed

3 files changed

+282
-190
lines changed

invokeai/frontend/web/src/features/controlLayers/components/StageComponent.tsx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import { debouncedRenderers, renderers as normalRenderers } from 'features/contr
1818
import Konva from 'konva';
1919
import type { IRect } from 'konva/lib/types';
2020
import { memo, useCallback, useLayoutEffect, useMemo, useState } from 'react';
21+
import { getImageDTO } from 'services/api/endpoints/images';
2122
import { useDevicePixelRatio } from 'use-device-pixel-ratio';
2223
import { v4 as uuidv4 } from 'uuid';
2324

@@ -160,7 +161,7 @@ const useStageRenderer = (
160161

161162
useLayoutEffect(() => {
162163
log.trace('Rendering layers');
163-
renderers.renderLayers(stage, state.layers, state.globalMaskLayerOpacity, tool, onLayerPosChanged);
164+
renderers.renderLayers(stage, state.layers, state.globalMaskLayerOpacity, tool, getImageDTO, onLayerPosChanged);
164165
}, [
165166
stage,
166167
state.layers,

invokeai/frontend/web/src/features/controlLayers/util/getLayerBlobs.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import { blobToDataURL } from 'features/canvas/util/blobToDataURL';
44
import { isRegionalGuidanceLayer, RG_LAYER_NAME } from 'features/controlLayers/store/controlLayersSlice';
55
import { renderers } from 'features/controlLayers/util/renderers';
66
import Konva from 'konva';
7+
import { getImageDTO } from 'services/api/endpoints/images';
78
import { assert } from 'tsafe';
89

910
/**
@@ -22,7 +23,7 @@ export const getRegionalPromptLayerBlobs = async (
2223
const reduxLayers = layers.filter(isRegionalGuidanceLayer);
2324
const container = document.createElement('div');
2425
const stage = new Konva.Stage({ container, width, height });
25-
renderers.renderLayers(stage, reduxLayers, 1, 'brush');
26+
renderers.renderLayers(stage, reduxLayers, 1, 'brush', getImageDTO);
2627

2728
const konvaLayers = stage.find<Konva.Layer>(`.${RG_LAYER_NAME}`);
2829
const blobs: Record<string, Blob> = {};

0 commit comments

Comments
 (0)