76 lines
3.9 KiB
TypeScript
76 lines
3.9 KiB
TypeScript
import { NODE_DEFAULT_SIZE } from "@/constant/canvas";
|
|
import { CanvasNodeType, type CanvasNodeData } from "@/types/canvas";
|
|
|
|
export const MEDIA_NODE_MIN_SIZE = { width: 420, height: 236 } as const;
|
|
export const VIDEO_NODE_MAX_SIZE = { width: 720, height: 520 } as const;
|
|
|
|
export function fitNodeSize(width: number, height: number, maxWidth = 720, maxHeight = 520, minWidth = MEDIA_NODE_MIN_SIZE.width, minHeight = MEDIA_NODE_MIN_SIZE.height) {
|
|
const w = Math.max(1, width);
|
|
const h = Math.max(1, height);
|
|
// 媒体节点既要保留原始比例，也要给生成状态、操作按钮留下稳定的可读空间。
|
|
const preferredScale = Math.min(1, maxWidth / w, maxHeight / h);
|
|
const minimumScale = Math.max(minWidth / w, minHeight / h);
|
|
const scale = Math.max(preferredScale, minimumScale);
|
|
return { width: w * scale, height: h * scale };
|
|
}
|
|
|
|
export function nodeSizeFromRatio(size: string, baseWidth: number, baseHeight: number) {
|
|
const raw = String(size || "").trim();
|
|
if (!raw || raw.toLowerCase() === "auto") return null;
|
|
// 支持 16:9 / 1024x576 / 16:9-2k
|
|
const match = raw.match(/^(\d+(?:\.\d+)?)(?:x|:)(\d+(?:\.\d+)?)/i);
|
|
if (!match) return null;
|
|
const width = Number(match[1]);
|
|
const height = Number(match[2]);
|
|
if (!Number.isFinite(width) || !Number.isFinite(height) || width <= 0 || height <= 0) return null;
|
|
const ratio = width / Math.max(1, height);
|
|
if (ratio < 0.25 || ratio > 4) return { width: baseWidth, height: baseHeight };
|
|
const candidateSize = ratio >= baseWidth / baseHeight ? { width: baseWidth, height: baseWidth / ratio } : { width: baseHeight * ratio, height: baseHeight };
|
|
return fitNodeSize(candidateSize.width, candidateSize.height, baseWidth, baseHeight);
|
|
}
|
|
|
|
export function ensureMediaNodeMinimumSize(node: CanvasNodeData) {
|
|
if (node.type !== CanvasNodeType.Image && node.type !== CanvasNodeType.Video) return node;
|
|
const title = node.title === "New Generation" ? "图片" : node.title === "Video" ? "视频" : node.title;
|
|
let width = node.width;
|
|
let height = node.height;
|
|
const emptyStage = NODE_DEFAULT_SIZE[node.type];
|
|
const shouldPromoteEmptyStage = !node.metadata?.content
|
|
&& !node.metadata?.freeResize
|
|
&& !node.metadata?.locked
|
|
&& width * height < emptyStage.width * emptyStage.height;
|
|
if (shouldPromoteEmptyStage) {
|
|
width = emptyStage.width;
|
|
height = emptyStage.height;
|
|
}
|
|
const naturalWidth = node.metadata?.naturalWidth || 0;
|
|
const naturalHeight = node.metadata?.naturalHeight || 0;
|
|
const requestedSize = node.type === CanvasNodeType.Image && node.metadata?.generationType === "edit"
|
|
? nodeSizeFromRatio(node.metadata.size || "auto", node.width, node.height)
|
|
: null;
|
|
const naturalRatio = naturalWidth / Math.max(1, naturalHeight);
|
|
const nodeRatio = node.width / Math.max(1, node.height);
|
|
// 修复旧版图生图无条件继承参考节点尺寸造成的比例错误，不覆盖自由拉伸或锁定布局。
|
|
if (requestedSize && naturalWidth > 0 && naturalHeight > 0 && !node.metadata?.freeResize && !node.metadata?.locked && Math.abs(naturalRatio - nodeRatio) > 0.01) {
|
|
const alignedSize = fitNodeSize(naturalWidth, naturalHeight, requestedSize.width, requestedSize.height);
|
|
width = alignedSize.width;
|
|
height = alignedSize.height;
|
|
}
|
|
if (width < MEDIA_NODE_MIN_SIZE.width || height < MEDIA_NODE_MIN_SIZE.height) {
|
|
const scale = Math.max(1, MEDIA_NODE_MIN_SIZE.width / Math.max(1, width), MEDIA_NODE_MIN_SIZE.height / Math.max(1, height));
|
|
width *= scale;
|
|
height *= scale;
|
|
}
|
|
if (width === node.width && height === node.height && title === node.title) return node;
|
|
return {
|
|
...node,
|
|
title,
|
|
position: {
|
|
x: node.position.x + node.width / 2 - width / 2,
|
|
y: node.position.y + node.height / 2 - height / 2,
|
|
},
|
|
width,
|
|
height,
|
|
};
|
|
}
|