@d5techs/3dgs-lib 1.2.0 → 1.3.0

package/dist/3dgs-lib.cjs

@@ -4413,7 +4413,8 @@ function readProperty$1(dataView, baseOffset, prop, littleEndian) {
 function sigmoid$1(x) {
   return 1 / (1 + Math.exp(-x));
 }
-async function loadPLY(url) {
+async function loadPLY(url, options = {}) {
+  const { coordinateSystem = "blender" } = options;
   const response = await fetch(url);
   if (!response.ok) {
     throw new Error(`无法加载 PLY 文件: ${url}`);
@@ -4425,6 +4426,7 @@ async function loadPLY(url) {
     throw new Error("不支持 ASCII 格式的 PLY 文件，请使用 binary_little_endian 或 binary_big_endian 格式");
   }
   const littleEndian = format === "binary_little_endian";
+  const swapYZ = coordinateSystem === "blender";
   const propMap = buildPropertyMap(properties);
   const props = {
     x: propMap.get("x"),
@@ -4490,9 +4492,14 @@ async function loadPLY(url) {
       shRest[dstBase + 2] = srcB < shRestProps.length ? readProperty$1(dataView, base, shRestProps[srcB], littleEndian) : 0;
     }
     splats[i] = {
-      mean: [x, y, z],
-      scale: [scale_0, scale_1, scale_2],
-      rotation: [
+      mean: swapYZ ? [x, z, y] : [x, y, z],
+      scale: swapYZ ? [scale_0, scale_2, scale_1] : [scale_0, scale_1, scale_2],
+      rotation: swapYZ ? [
+        -rot_0 * qnorm,
+        rot_1 * qnorm,
+        rot_3 * qnorm,
+        rot_2 * qnorm
+      ] : [
         rot_0 * qnorm,
         rot_1 * qnorm,
         rot_2 * qnorm,
@@ -4708,7 +4715,8 @@ async function parsePLYBuffer(buffer, options = {}) {
   const {
     maxSplats = 2e5,
     loadSH = false,
-    onProgress
+    onProgress,
+    coordinateSystem = "blender"
   } = options;
   const seed = options.seed ?? buffer.byteLength;
   const { headerText, dataOffset } = extractHeader(buffer);
@@ -4798,27 +4806,41 @@ async function parsePLYBuffer(buffer, options = {}) {
   const opacities = new Float32Array(actualCount);
   const shCoeffs = loadSH ? new Float32Array(actualCount * 45) : void 0;
   const dataView = new DataView(buffer, dataOffset);
+  const swapYZ = coordinateSystem === "blender";
   let outputIdx = 0;
   let lastProgress = 0;
   for (let i = 0; i < actualCount; i++) {
     const srcIdx = sampleIndices ? sampleIndices[i] : i;
     const base = srcIdx * stride;
-    positions[outputIdx * 3 + 0] = offsets.x >= 0 ? readProperty(dataView, base + offsets.x, types.x, littleEndian) : 0;
-    positions[outputIdx * 3 + 1] = offsets.y >= 0 ? readProperty(dataView, base + offsets.y, types.y, littleEndian) : 0;
-    positions[outputIdx * 3 + 2] = offsets.z >= 0 ? readProperty(dataView, base + offsets.z, types.z, littleEndian) : 0;
-    scales[outputIdx * 3 + 0] = offsets.scale_0 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_0, types.scale_0, littleEndian)) : 1;
-    scales[outputIdx * 3 + 1] = offsets.scale_1 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_1, types.scale_1, littleEndian)) : 1;
-    scales[outputIdx * 3 + 2] = offsets.scale_2 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_2, types.scale_2, littleEndian)) : 1;
+    const px = offsets.x >= 0 ? readProperty(dataView, base + offsets.x, types.x, littleEndian) : 0;
+    const py = offsets.y >= 0 ? readProperty(dataView, base + offsets.y, types.y, littleEndian) : 0;
+    const pz = offsets.z >= 0 ? readProperty(dataView, base + offsets.z, types.z, littleEndian) : 0;
+    positions[outputIdx * 3 + 0] = px;
+    positions[outputIdx * 3 + 1] = swapYZ ? pz : py;
+    positions[outputIdx * 3 + 2] = swapYZ ? py : pz;
+    const sx = offsets.scale_0 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_0, types.scale_0, littleEndian)) : 1;
+    const sy = offsets.scale_1 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_1, types.scale_1, littleEndian)) : 1;
+    const sz = offsets.scale_2 >= 0 ? Math.exp(readProperty(dataView, base + offsets.scale_2, types.scale_2, littleEndian)) : 1;
+    scales[outputIdx * 3 + 0] = sx;
+    scales[outputIdx * 3 + 1] = swapYZ ? sz : sy;
+    scales[outputIdx * 3 + 2] = swapYZ ? sy : sz;
     const rot_0 = offsets.rot_0 >= 0 ? readProperty(dataView, base + offsets.rot_0, types.rot_0, littleEndian) : 1;
     const rot_1 = offsets.rot_1 >= 0 ? readProperty(dataView, base + offsets.rot_1, types.rot_1, littleEndian) : 0;
     const rot_2 = offsets.rot_2 >= 0 ? readProperty(dataView, base + offsets.rot_2, types.rot_2, littleEndian) : 0;
     const rot_3 = offsets.rot_3 >= 0 ? readProperty(dataView, base + offsets.rot_3, types.rot_3, littleEndian) : 0;
     const qlen = Math.sqrt(rot_0 * rot_0 + rot_1 * rot_1 + rot_2 * rot_2 + rot_3 * rot_3);
     const qnorm = qlen > 0 ? 1 / qlen : 1;
-    rotations[outputIdx * 4 + 0] = rot_0 * qnorm;
-    rotations[outputIdx * 4 + 1] = rot_1 * qnorm;
-    rotations[outputIdx * 4 + 2] = rot_2 * qnorm;
-    rotations[outputIdx * 4 + 3] = rot_3 * qnorm;
+    if (swapYZ) {
+      rotations[outputIdx * 4 + 0] = -rot_0 * qnorm;
+      rotations[outputIdx * 4 + 1] = rot_1 * qnorm;
+      rotations[outputIdx * 4 + 2] = rot_3 * qnorm;
+      rotations[outputIdx * 4 + 3] = rot_2 * qnorm;
+    } else {
+      rotations[outputIdx * 4 + 0] = rot_0 * qnorm;
+      rotations[outputIdx * 4 + 1] = rot_1 * qnorm;
+      rotations[outputIdx * 4 + 2] = rot_2 * qnorm;
+      rotations[outputIdx * 4 + 3] = rot_3 * qnorm;
+    }
     const f_dc_0 = offsets.f_dc_0 >= 0 ? readProperty(dataView, base + offsets.f_dc_0, types.f_dc_0, littleEndian) : 0;
     const f_dc_1 = offsets.f_dc_1 >= 0 ? readProperty(dataView, base + offsets.f_dc_1, types.f_dc_1, littleEndian) : 0;
     const f_dc_2 = offsets.f_dc_2 >= 0 ? readProperty(dataView, base + offsets.f_dc_2, types.f_dc_2, littleEndian) : 0;
@@ -13970,8 +13992,12 @@ class App {
   }
   /**
    * 加载 PLY 文件 (3D Gaussian Splatting)
+   * @param urlOrBuffer PLY 文件 URL 或 ArrayBuffer
+   * @param onProgress 进度回调
+   * @param isLocalFile 是否为本地文件
+   * @param coordinateSystem 源数据坐标系，默认 'blender'（Z-up → Y-up 自动转换）
    */
-  async addPLY(urlOrBuffer, onProgress, isLocalFile = false) {
+  async addPLY(urlOrBuffer, onProgress, isLocalFile = false, coordinateSystem = "blender") {
     try {
       const isMobile = isMobileDevice();
       let buffer;
@@ -14000,7 +14026,8 @@ class App {
         const compactData = await this.parsePLYBuffer(buffer, {
           maxSplats: Infinity,
           loadSH: false,
-          onProgress: parseProgressCallback
+          onProgress: parseProgressCallback,
+          coordinateSystem
         });
         if (onProgress) onProgress(90, "upload");
         gsRenderer.setCompactData(compactData);
@@ -14014,7 +14041,8 @@ class App {
         const compactData = await this.parsePLYBuffer(buffer, {
           maxSplats: Infinity,
           loadSH: true,
-          onProgress: parseProgressCallback
+          onProgress: parseProgressCallback,
+          coordinateSystem
         });
         if (onProgress) onProgress(90, "upload");
         gsRenderer.setCompactData(compactData);