90 lines
3.2 KiB
TypeScript
90 lines
3.2 KiB
TypeScript
import type { CanvasNodeData } from "@/types/canvas";
|
||
|
||
export const MEDIA_CONVERSION_SCHEMA_VERSION = 1 as const;
|
||
|
||
export const MEDIA_CONVERSION_NODE_TYPE = "media-conversion" as const;
|
||
|
||
export type MediaConversionOperation = "grayscale" | "edge-canny" | "lineart" | "depth" | "pose" | "cutout";
|
||
export type MediaConversionStatus = "idle" | "processing" | "completed" | "stale" | "skipped" | "unavailable" | "error";
|
||
export type MediaConversionOutputKind = "image" | "video";
|
||
|
||
export type MediaConversionNodeState = {
|
||
schemaVersion: typeof MEDIA_CONVERSION_SCHEMA_VERSION;
|
||
operation: MediaConversionOperation;
|
||
status: MediaConversionStatus;
|
||
sourceNodeId?: string;
|
||
sourceFingerprint?: string;
|
||
outputKind?: MediaConversionOutputKind;
|
||
resultStorageKey?: string;
|
||
resultWidth?: number;
|
||
resultHeight?: number;
|
||
detectedPeople?: number;
|
||
errorCode?: string;
|
||
errorMessage?: string;
|
||
startedAt?: string;
|
||
updatedAt?: string;
|
||
};
|
||
|
||
export const MEDIA_CONVERSION_OPERATION_LABELS: Record<MediaConversionOperation, string> = {
|
||
grayscale: "灰度图",
|
||
"edge-canny": "Canny 边缘",
|
||
lineart: "AI 线稿",
|
||
depth: "深度图",
|
||
pose: "姿态骨架",
|
||
cutout: "透明抠图",
|
||
};
|
||
|
||
export const MEDIA_CONVERSION_OPERATION_DESCRIPTIONS: Record<MediaConversionOperation, string> = {
|
||
grayscale: "普通图像算法，适合先检查明暗层次",
|
||
"edge-canny": "普通图像算法，提取清晰轮廓",
|
||
lineart: "本地 ControlNet Aux 线稿预处理，首次运行需要加载模型",
|
||
depth: "本地 Depth Anything V2 Small，首次运行需要加载模型",
|
||
pose: "本地 OpenPose 人体骨架预处理，未检测到人物时会跳过",
|
||
cutout: "需要安装并验证本地抠图模型",
|
||
};
|
||
|
||
export function createDefaultMediaConversionState(): MediaConversionNodeState {
|
||
return {
|
||
schemaVersion: MEDIA_CONVERSION_SCHEMA_VERSION,
|
||
operation: "edge-canny",
|
||
status: "idle",
|
||
};
|
||
}
|
||
|
||
export function mediaConversionSourceFingerprint(node: CanvasNodeData) {
|
||
const metadata = node.metadata;
|
||
const source = [
|
||
node.id,
|
||
metadata?.storageKey || "",
|
||
compactSource(metadata?.content || metadata?.previewContent || ""),
|
||
metadata?.mimeType || "",
|
||
metadata?.bytes || "",
|
||
metadata?.naturalWidth || "",
|
||
metadata?.naturalHeight || "",
|
||
metadata?.durationMs || "",
|
||
].join("|");
|
||
let hash = 2166136261;
|
||
for (let index = 0; index < source.length; index += 1) {
|
||
hash ^= source.charCodeAt(index);
|
||
hash = Math.imul(hash, 16777619);
|
||
}
|
||
return `v1-${(hash >>> 0).toString(16).padStart(8, "0")}`;
|
||
}
|
||
|
||
export function mediaConversionOperationLabel(operation: MediaConversionOperation) {
|
||
return MEDIA_CONVERSION_OPERATION_LABELS[operation];
|
||
}
|
||
|
||
export function mediaConversionOperationDescription(operation: MediaConversionOperation) {
|
||
return MEDIA_CONVERSION_OPERATION_DESCRIPTIONS[operation];
|
||
}
|
||
|
||
export function isLocalImageOperation(operation: MediaConversionOperation) {
|
||
return operation === "grayscale" || operation === "edge-canny";
|
||
}
|
||
|
||
function compactSource(value: string) {
|
||
if (value.length <= 2048) return value;
|
||
return `${value.slice(0, 1024)}|${value.slice(-1024)}|${value.length}`;
|
||
}
|