@d5techs/3dgs-lib 1.4.23 → 1.4.25

package/dist/3dgs-lib.js

@@ -6624,12 +6624,14 @@ async function encodeWebP(width, height, pixelData) {
   }
   return new Uint8Array(await blob.arrayBuffer());
 }
+const yieldToUI = () => new Promise((r) => setTimeout(r, 0));
 async function serializeSOG(data, coordinateSystem = "blender", options = {}, onProgress) {
   const { maxSHBands = 3, iterations = 10, paletteSize = 1024 } = options;
   const { count, colors, opacities } = data;
   let { positions, scales, rotations } = data;
   let shCoeffs = data.shCoeffs ? new Float32Array(data.shCoeffs) : void 0;
-  onProgress == null ? void 0 : onProgress("坐标系变换", 0);
+  onProgress == null ? void 0 : onProgress("coord_transform", 0);
+  await yieldToUI();
   if (coordinateSystem === "blender") {
     positions = new Float32Array(positions);
     scales = new Float32Array(scales);
@@ -6652,7 +6654,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
       }
     }
   }
-  onProgress == null ? void 0 : onProgress("Morton 排序", 0.05);
+  onProgress == null ? void 0 : onProgress("morton_sort", 0.05);
+  await yieldToUI();
   const sortOrder = mortonSort(positions, count);
   const sPos = new Float32Array(count * 3);
   const sSca = new Float32Array(count * 3);
@@ -6685,7 +6688,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
   }
   const { width, height } = calcImageDimensions(count);
   const totalPixels = width * height;
-  onProgress == null ? void 0 : onProgress("编码位置", 0.1);
+  onProgress == null ? void 0 : onProgress("encode_position", 0.1);
+  await yieldToUI();
   const logPos = new Float32Array(count * 3);
   for (let i = 0; i < count * 3; i++) logPos[i] = symLog(sPos[i]);
   const mins = [Infinity, Infinity, Infinity];
@@ -6713,7 +6717,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
     meansLData[off + 3] = 255;
     meansUData[off + 3] = 255;
   }
-  onProgress == null ? void 0 : onProgress("编码缩放", 0.2);
+  onProgress == null ? void 0 : onProgress("encode_scale", 0.2);
+  await yieldToUI();
   const logScales = new Float32Array(count * 3);
   for (let i = 0; i < count * 3; i++) logScales[i] = Math.log(Math.max(1e-8, sSca[i]));
   const scaleCodebook = buildCodebook(logScales, 256);
@@ -6726,7 +6731,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
     scalesData[off + 2] = findNearest(scaleCodebook, logScales[i * 3 + 2]);
     scalesData[off + 3] = 255;
   }
-  onProgress == null ? void 0 : onProgress("编码四元数", 0.3);
+  onProgress == null ? void 0 : onProgress("encode_quaternion", 0.3);
+  await yieldToUI();
   const quatsData = new Uint8ClampedArray(totalPixels * 4);
   quatsData.fill(255);
   const SQRT2 = Math.SQRT2;
@@ -6779,7 +6785,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
     quatsData[off + 2] = Math.round(Math.max(0, Math.min(255, (r2 / SQRT2 + 0.5) * 255)));
     quatsData[off + 3] = largest + 252;
   }
-  onProgress == null ? void 0 : onProgress("编码颜色", 0.4);
+  onProgress == null ? void 0 : onProgress("encode_color", 0.4);
+  await yieldToUI();
   const dcValues = new Float32Array(count * 3);
   for (let i = 0; i < count; i++) {
     dcValues[i * 3] = (sCol[i * 3] - 0.5) / SH_C0;
@@ -6808,7 +6815,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
     encodeWebP(width, height, sh0Data)
   ];
   if (actualBands > 0 && sSH) {
-    onProgress == null ? void 0 : onProgress("压缩球谐系数 (GPU)", 0.5);
+    onProgress == null ? void 0 : onProgress("sh_compress", 0.5);
+    await yieldToUI();
     const pointsFlat = new Float32Array(count * shDim);
     for (let i = 0; i < count; i++) {
       const b = i * 45;
@@ -6823,10 +6831,11 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
       iterations,
       (iter, total) => {
         const p = 0.5 + 0.35 * (iter / total);
-        onProgress == null ? void 0 : onProgress(`球谐聚类 ${iter}/${total}`, p);
+        onProgress == null ? void 0 : onProgress("sh_cluster", p, { iter, total });
       }
     );
-    onProgress == null ? void 0 : onProgress("量化球谐 codebook", 0.88);
+    onProgress == null ? void 0 : onProgress("sh_quantize", 0.88);
+    await yieldToUI();
     const allCoeffValues = new Float32Array(effectivePaletteSize * shDim);
     for (let c = 0; c < effectivePaletteSize; c++) {
       for (let d = 0; d < shDim; d++) allCoeffValues[c * shDim + d] = centroids[c * shDim + d];
@@ -6868,7 +6877,8 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
       files: ["sh_centroids.webp", "sh_labels.webp"]
     };
   }
-  onProgress == null ? void 0 : onProgress("编码 WebP 图像", 0.92);
+  onProgress == null ? void 0 : onProgress("encode_webp", 0.92);
+  await yieldToUI();
   const webps = await Promise.all(webpPromises);
   const [meansLWebP, meansUWebP, scalesWebP, quatsWebP, sh0WebP] = webps;
   const meta = {
@@ -6894,9 +6904,10 @@ async function serializeSOG(data, coordinateSystem = "blender", options = {}, on
     zipEntries["sh_centroids.webp"] = webps[5];
     zipEntries["sh_labels.webp"] = webps[6];
   }
-  onProgress == null ? void 0 : onProgress("打包 ZIP", 0.98);
+  onProgress == null ? void 0 : onProgress("pack_zip", 0.98);
+  await yieldToUI();
   const zipData = zipSync(zipEntries, { level: 0 });
-  onProgress == null ? void 0 : onProgress("完成", 1);
+  onProgress == null ? void 0 : onProgress("done", 1);
   return zipData.buffer;
 }
 const WORKGROUP_SIZE$1 = 256;