@forgeax-extension/character-3d 0.1.2

package/shared/rodin-image.ts

@@ -0,0 +1,125 @@
+/** Volcengine hyper3d-gen2 rejects images outside this width/height range. */
+export const RODIN_ASPECT_MIN = 0.4;
+export const RODIN_ASPECT_MAX = 2.5;
+export const RODIN_MAX_EDGE = 4096;
+
+/** Pad slightly inside the official band so rounding cannot fail the check. */
+const PAD_MIN = 0.41;
+const PAD_MAX = 2.49;
+
+export const RODIN_ASPECT_USER_MESSAGE =
+  'Rodin 不收这张图：太瘦或太宽了（宽高比要在 0.4～2.5 之间）。请裁成更方一点，或重新用本地上传（会自动留白补齐）。';
+
+export function rodinAspectRatio(width: number, height: number): number {
+  return width / height;
+}
+
+export function fitsRodinAspect(width: number, height: number): boolean {
+  if (width <= 0 || height <= 0) return false;
+  const ratio = rodinAspectRatio(width, height);
+  return ratio >= RODIN_ASPECT_MIN && ratio <= RODIN_ASPECT_MAX;
+}
+
+export function paddedRodinSize(width: number, height: number): { width: number; height: number } {
+  if (fitsRodinAspect(width, height)) return { width, height };
+  const ratio = rodinAspectRatio(width, height);
+  if (ratio < RODIN_ASPECT_MIN) return { width: Math.ceil(height * PAD_MIN), height };
+  return { width, height: Math.ceil(width / PAD_MAX) };
+}
+
+export function clampRodinEdge(width: number, height: number): { width: number; height: number } {
+  if (width <= RODIN_MAX_EDGE && height <= RODIN_MAX_EDGE) return { width, height };
+  const scale = Math.min(RODIN_MAX_EDGE / width, RODIN_MAX_EDGE / height);
+  return {
+    width: Math.max(1, Math.floor(width * scale)),
+    height: Math.max(1, Math.floor(height * scale)),
+  };
+}
+
+export function rodinAspectRejectMessage(width: number, height: number): string {
+  const ratio = rodinAspectRatio(width, height);
+  const kind = ratio < RODIN_ASPECT_MIN ? '太瘦（太高）' : '太宽';
+  return `Rodin 不收这张图：${width}×${height}，${kind}，宽高比 ${ratio.toFixed(2)}（只要 0.4～2.5）。请裁成更方一点，或重新用本地上传（会自动留白补齐）。`;
+}
+
+export function explainRodinGatewayError(raw: string): string | null {
+  if (/aspect ratio/i.test(raw) && /0\.40|0\.4/.test(raw)) return RODIN_ASPECT_USER_MESSAGE;
+  if (/downloading image/i.test(raw) && /aspect ratio/i.test(raw)) return RODIN_ASPECT_USER_MESSAGE;
+  return null;
+}
+
+export function readImageSize(data: Uint8Array): { width: number; height: number } | null {
+  if (data.length < 24) return null;
+  if (data[0] === 0x89 && data[1] === 0x50 && data[2] === 0x4e && data[3] === 0x47) {
+    return { width: u32be(data, 16), height: u32be(data, 20) };
+  }
+  if (data[0] === 0xff && data[1] === 0xd8) return readJpegSize(data);
+  if (data[0] === 0x47 && data[1] === 0x49 && data[2] === 0x46) {
+    return { width: u16le(data, 6), height: u16le(data, 8) };
+  }
+  if (
+    data[0] === 0x52 &&
+    data[1] === 0x49 &&
+    data[2] === 0x46 &&
+    data[3] === 0x46 &&
+    data[8] === 0x57 &&
+    data[9] === 0x45 &&
+    data[10] === 0x42 &&
+    data[11] === 0x50
+  ) {
+    return readWebpSize(data);
+  }
+  return null;
+}
+
+function u16be(data: Uint8Array, i: number): number {
+  return ((data[i] ?? 0) << 8) | (data[i + 1] ?? 0);
+}
+
+function u16le(data: Uint8Array, i: number): number {
+  return (data[i] ?? 0) | ((data[i + 1] ?? 0) << 8);
+}
+
+function u32be(data: Uint8Array, i: number): number {
+  return (
+    ((data[i] ?? 0) << 24) |
+    ((data[i + 1] ?? 0) << 16) |
+    ((data[i + 2] ?? 0) << 8) |
+    (data[i + 3] ?? 0)
+  ) >>> 0;
+}
+
+function readJpegSize(data: Uint8Array): { width: number; height: number } | null {
+  let i = 2;
+  while (i + 8 < data.length) {
+    if (data[i] !== 0xff) {
+      i += 1;
+      continue;
+    }
+    const marker = data[i + 1] ?? 0;
+    if (marker === 0xc0 || marker === 0xc1 || marker === 0xc2) {
+      return { height: u16be(data, i + 5), width: u16be(data, i + 7) };
+    }
+    const len = u16be(data, i + 2);
+    if (len < 2) return null;
+    i += 2 + len;
+  }
+  return null;
+}
+
+function readWebpSize(data: Uint8Array): { width: number; height: number } | null {
+  const kind = String.fromCharCode(data[12] ?? 0, data[13] ?? 0, data[14] ?? 0, data[15] ?? 0);
+  if (kind === 'VP8X' && data.length >= 30) {
+    const width = 1 + ((data[24] ?? 0) | ((data[25] ?? 0) << 8) | ((data[26] ?? 0) << 16));
+    const height = 1 + ((data[27] ?? 0) | ((data[28] ?? 0) << 8) | ((data[29] ?? 0) << 16));
+    return { width, height };
+  }
+  if (kind === 'VP8 ' && data.length >= 30) {
+    return { width: u16le(data, 26) & 0x3fff, height: u16le(data, 28) & 0x3fff };
+  }
+  if (kind === 'VP8L' && data.length >= 25) {
+    const bits = (data[21] ?? 0) | ((data[22] ?? 0) << 8) | ((data[23] ?? 0) << 16) | ((data[24] ?? 0) << 24);
+    return { width: (bits & 0x3fff) + 1, height: ((bits >> 14) & 0x3fff) + 1 };
+  }
+  return null;
+}