5e4b548ad4
- Implemented ComfyGenerateRequest type and associated functions for generating images using various architectures and modes. - Added functions for listing generation options and handling image uploads. - Created workflows for different generation modes including SDXL, Z-Image, Z-Image Turbo, and Anima. - Introduced GenerationJobStatus component to display the status of ongoing generation jobs. - Developed MaskControls for managing mask operations and displaying mask analysis. - Created palette items for tool selection, layer management, and generation settings.
77 lines
2.6 KiB
TypeScript
77 lines
2.6 KiB
TypeScript
import type { ImageDocument } from "@core/document";
|
|
import type { Rect } from "@core/geometry";
|
|
import type { Layer } from "@core/layer";
|
|
import type { ArtboardId, LayerId } from "@core/id";
|
|
import type { TransformTarget } from "./transform";
|
|
|
|
export function resolveTransformTargetBounds(document: ImageDocument, target: TransformTarget): Rect | undefined {
|
|
switch (target.type) {
|
|
case "artboard":
|
|
return document.artboards.find((artboard) => artboard.id === target.id)?.bounds;
|
|
case "layer": {
|
|
const layer = findLayer(document, target.id);
|
|
return layer ? resolveLayerBounds(document, layer) : undefined;
|
|
}
|
|
}
|
|
}
|
|
|
|
export function selectedTransformTarget(document: ImageDocument, selection: { artboardId?: ArtboardId; layerIds: LayerId[] }): TransformTarget | undefined {
|
|
if (selection.layerIds.length === 1 && selection.layerIds[0]) return { type: "layer", id: selection.layerIds[0] };
|
|
if (selection.artboardId) return { type: "artboard", id: selection.artboardId };
|
|
return undefined;
|
|
}
|
|
|
|
function findLayer(document: ImageDocument, layerId: LayerId): Layer | undefined {
|
|
for (const artboard of document.artboards) {
|
|
const layer = findLayerInTree(artboard.layers, layerId);
|
|
if (layer) return layer;
|
|
}
|
|
|
|
return undefined;
|
|
}
|
|
|
|
function findLayerInTree(layers: Layer[], layerId: LayerId): Layer | undefined {
|
|
for (const layer of layers) {
|
|
if (layer.id === layerId) return layer;
|
|
if (layer.type === "group") {
|
|
const child = findLayerInTree(layer.children, layerId);
|
|
if (child) return child;
|
|
}
|
|
}
|
|
|
|
return undefined;
|
|
}
|
|
|
|
function resolveLayerBounds(document: ImageDocument, layer: Layer): Rect | undefined {
|
|
switch (layer.type) {
|
|
case "group":
|
|
return unionRects(layer.children.flatMap((child) => {
|
|
const bounds = resolveLayerBounds(document, child);
|
|
return bounds ? [bounds] : [];
|
|
}));
|
|
case "image":
|
|
case "raster": {
|
|
const asset = document.assets.find((candidate) => candidate.id === layer.assetId);
|
|
if (!asset) return undefined;
|
|
|
|
return {
|
|
x: layer.transform.position.x,
|
|
y: layer.transform.position.y,
|
|
w: asset.intrinsicSize.w * layer.transform.scale.x,
|
|
h: asset.intrinsicSize.h * layer.transform.scale.y,
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
function unionRects(rects: Rect[]): Rect | undefined {
|
|
if (rects.length === 0) return undefined;
|
|
|
|
const minX = Math.min(...rects.map((rect) => rect.x));
|
|
const minY = Math.min(...rects.map((rect) => rect.y));
|
|
const maxX = Math.max(...rects.map((rect) => rect.x + rect.w));
|
|
const maxY = Math.max(...rects.map((rect) => rect.y + rect.h));
|
|
|
|
return { x: minX, y: minY, w: maxX - minX, h: maxY - minY };
|
|
}
|