95 lines
3.8 KiB
TypeScript
95 lines
3.8 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
import type { ImageDocument } from "@core/document";
|
|
import type { GenerateSettings } from "@editor/tools";
|
|
import { createInitialAppState } from "@editor/initial-state";
|
|
import { resolveGeneratedOutputPlacement } from "./outputPlacement";
|
|
|
|
describe("generated output placement", () => {
|
|
test("centers text-to-image output in the active artboard", () => {
|
|
const placement = resolveGeneratedOutputPlacement({
|
|
document: document(),
|
|
selection: { artboardId: "artboard", layerIds: [] },
|
|
settings: settings("text-to-image"),
|
|
intrinsicSize: { w: 400, h: 200 },
|
|
});
|
|
|
|
expect(placement).toEqual({
|
|
artboardId: "artboard",
|
|
layerName: "Generated image",
|
|
transform: { position: { x: 310, y: 270 }, scale: { x: 1, y: 1 }, rotation: 0 },
|
|
});
|
|
});
|
|
|
|
test("matches image-to-image output to the selected layer bounds", () => {
|
|
const placement = resolveGeneratedOutputPlacement({
|
|
document: document(),
|
|
selection: selected(),
|
|
settings: settings("image-to-image"),
|
|
intrinsicSize: { w: 200, h: 250 },
|
|
});
|
|
|
|
expect(placement.transform).toEqual({
|
|
position: { x: 80, y: 90 },
|
|
scale: { x: 0.75, y: 1 },
|
|
rotation: 12,
|
|
});
|
|
});
|
|
|
|
test("places an inpaint crop at its recorded document bounds", () => {
|
|
const placement = resolveGeneratedOutputPlacement({
|
|
document: document(),
|
|
selection: selected(),
|
|
settings: settings("inpaint"),
|
|
intrinsicSize: { w: 128, h: 64 },
|
|
inpaintBundle: {
|
|
inputImage: "input", maskImage: "mask", width: 256, height: 128,
|
|
targetLayerId: "source", maskLayerId: "mask", sourceAssetId: "source-asset", maskAssetId: "mask-asset",
|
|
crop: { assetBounds: { x: 0, y: 0, w: 256, h: 128 }, documentBounds: { x: 140, y: 150, w: 384, h: 64 }, padding: 16, maskedAreaOnly: true },
|
|
mask: { polarity: "hidden", activeBounds: { x: 20, y: 20, w: 40, h: 40 } },
|
|
placement: { artboardId: "artboard", layerName: "Source inpaint", transform: { position: { x: 140, y: 150 }, scale: { x: 1.5, y: 0.5 }, rotation: 12 } },
|
|
backend: { growMaskBy: 6, maskedContent: "neutral", maskBlur: 0, maskFeather: 0, maskExpand: 0, cropPadding: 16 },
|
|
},
|
|
});
|
|
|
|
expect(placement.transform).toEqual({ position: { x: 140, y: 150 }, scale: { x: 3, y: 1 }, rotation: 12 });
|
|
});
|
|
|
|
test("extends outpaint output left and top while preserving source pixel scale", () => {
|
|
const outpaint = { ...settings("outpaint"), outpaint: { left: 40, top: 20, right: 10, bottom: 30, feathering: 8 } };
|
|
const placement = resolveGeneratedOutputPlacement({
|
|
document: document(),
|
|
selection: selected(),
|
|
settings: outpaint,
|
|
intrinsicSize: { w: 150, h: 550 },
|
|
});
|
|
|
|
expect(placement.transform).toEqual({
|
|
position: { x: 20, y: 80 },
|
|
scale: { x: 1.5, y: 0.5 },
|
|
rotation: 12,
|
|
});
|
|
});
|
|
});
|
|
|
|
function settings(mode: GenerateSettings["mode"]): GenerateSettings {
|
|
return { ...createInitialAppState("Test").editor.tools.generate, mode };
|
|
}
|
|
|
|
function selected() {
|
|
return { artboardId: "artboard", layerIds: ["source"] };
|
|
}
|
|
|
|
function document(): ImageDocument {
|
|
return {
|
|
id: "document", name: "Test", version: 1,
|
|
assets: [
|
|
{ id: "source-asset", name: "Source", mimeType: "image/png", source: "source", intrinsicSize: { w: 100, h: 500 } },
|
|
{ id: "mask-asset", name: "Mask", mimeType: "image/png", source: "mask", intrinsicSize: { w: 100, h: 500 } },
|
|
],
|
|
artboards: [{
|
|
id: "artboard", name: "Artboard", bounds: { x: 10, y: 20, w: 1000, h: 700 }, backgroundColor: "transparent", visible: true, locked: false,
|
|
layers: [{ id: "source", type: "raster", name: "Source", visible: true, locked: false, opacity: 1, assetId: "source-asset", transform: { position: { x: 80, y: 90 }, scale: { x: 1.5, y: 0.5 }, rotation: 12 } }],
|
|
}],
|
|
};
|
|
}
|