@combeenation/3d-viewer 5.2.0-alpha5 → 6.1.0-beta1

package/src/api/manager/gltfExportManager.ts

@@ -1,192 +1,236 @@
-import { isMeshIncludedInExclusionList } from '../util/structureHelper';
-import { PBRMaterial } from '@babylonjs/core/Materials/PBR/pbrMaterial';
-import { Color3 } from '@babylonjs/core/Maths/math.color';
-import { AbstractMesh } from '@babylonjs/core/Meshes/abstractMesh';
-import { Mesh } from '@babylonjs/core/Meshes/mesh';
-import { GLTF2Export, IExportOptions } from '@babylonjs/serializers/glTF/2.0';
-import { merge } from 'lodash-es';
-
-export class GltfExportManager {
-  /**
-   * Constructor.
-   */
-  protected constructor(protected viewer: Viewer) {}
-
-  /**
-   * Creates an {@link GltfExportManager}.
-   */
-  public static async create(viewer: Viewer): Promise<GltfExportManager> {
-    return new GltfExportManager(viewer);
-  }
-
-  /**
-   * Exports selected nodes to a file.
-   * @param filename optional name of the exported .GLB file.
-   * @param exportOptions export options to be merged with default options.
-   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
-   */
-  public async exportGlb(
-    filename = 'glb-export.glb',
-    exportOptions: IExportOptions = {},
-    excluded?: ExcludedGeometryList
-  ): Promise<File | undefined> {
-    this.exportPreProcess();
-    const glbData = await GLTF2Export.GLBAsync(
-      this.viewer.scene,
-      'dummy',
-      this.gltfExportOptions(exportOptions, excluded)
-    );
-    this.exportPostProcess();
-    const resBlob = glbData.glTFFiles['dummy.glb']; // should be only one file for glb
-
-    // check if result is valid, according to the typings this could also be a string
-    if (resBlob instanceof Blob) {
-      if (!filename.endsWith('.glb')) {
-        filename += '.glb';
-      }
-      return new File([resBlob], filename);
-    } else {
-      // result was not a valid blob
-      return undefined;
-    }
-  }
-
-  /**
-   * Exports selected nodes to GLTF. This may result in more than one file, since textures are exported separately.
-   * @param filename name of the main (text-based) .GLTF file referring to separate texture files.
-   * @param exportOptions export options to be merged with default options.
-   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
-   */
-  public async exportGltfToFile(filename: string, exportOptions: IExportOptions = {}, excluded?: ExcludedGeometryList) {
-    this.exportPreProcess();
-    await GLTF2Export.GLTFAsync(this.viewer.scene, filename, this.gltfExportOptions(exportOptions, excluded)).then(
-      glb => {
-        glb.downloadFiles();
-      }
-    );
-    this.exportPostProcess();
-  }
-
-  /**
-   * Exports selected nodes to GLB. This results in one binary file.
-   * @param filename name of the .GLB file.
-   * @param exportOptions export options to be merged with default options.
-   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
-   */
-  public async exportGlbToFile(filename: string, exportOptions: IExportOptions = {}, excluded?: ExcludedGeometryList) {
-    this.exportPreProcess();
-    await GLTF2Export.GLBAsync(this.viewer.scene, filename, this.gltfExportOptions(exportOptions, excluded)).then(
-      glb => {
-        glb.downloadFiles();
-      }
-    );
-    this.exportPostProcess();
-  }
-
-  /**
-   * Gets predefined {@link IExportOptions } merged with given ones.
-   */
-  protected gltfExportOptions(mergeWithOptions: IExportOptions = {}, excluded?: ExcludedGeometryList): IExportOptions {
-    const defaultOptions = {
-      // includeCoordinateSystemConversionNodes: true,
-      shouldExportNode: function (node: any) {
-        if (!node.isEnabled()) {
-          return false;
-        }
-        if (node.isVisible === false) {
-          return false;
-        }
-        if (node.name === '__bounding_box__') {
-          return false;
-        }
-        if (excluded && node instanceof Mesh && isMeshIncludedInExclusionList(node, excluded)) {
-          return false;
-        }
-        return true;
-      },
-    };
-    return merge({}, defaultOptions, mergeWithOptions);
-  }
-
-  /**
-   * Stuff to be done before exporting to GLTF
-   */
-  protected exportPreProcess() {
-    this.viewer.pauseRendering();
-    this.exchangeRefractionMaterials();
-  }
-
-  /**
-   * Stuff to be done after the GLTF export
-   */
-  protected exportPostProcess() {
-    this.restoreRefractionMaterials();
-    this.viewer.resumeRendering();
-  }
-
-  /**
-   * Materials with refraction set are not exported properly.
-   * Exchange all such (relevant) materials with a more export-friendly version
-   */
-  protected exchangeRefractionMaterials() {
-    for (const n of this.viewer.scene.getNodes()) {
-      if (!(n instanceof AbstractMesh)) continue;
-      if (!(n.material instanceof PBRMaterial)) continue;
-      if (!(n.material as PBRMaterial).subSurface.isRefractionEnabled) continue;
-      if ((n.material as PBRMaterial).transparencyMode !== PBRMaterial.PBRMATERIAL_OPAQUE) continue;
-      // if we're here, we have a node holding a material with set refraction whose transparencyMode is set to
-      // PBRMATERIAL_OPAQUE
-      n.material = this.createRefractionMaterialReplacement(n.material);
-    }
-  }
-
-  /**
-   * Restore original materials with set refraction.
-   */
-  protected restoreRefractionMaterials() {
-    for (const n of this.viewer.scene.getNodes()) {
-      if (!(n instanceof AbstractMesh)) continue;
-      if (!(n.material instanceof PBRMaterial)) continue;
-      if (!this.isMaterialClonedForExport(n.material)) continue;
-      // at this point we have a pbrmaterial tagged as cloned in its metadata that's set on a mesh
-
-      // restore and dispose
-      const currMaterial = n.material;
-      const prevMaterial = this.viewer.scene.getMaterialByUniqueID(n.material.metadata.clonedFrom);
-      if (prevMaterial) {
-        n.material = prevMaterial; // restore previous material
-        currMaterial.dispose(false, true); // dispose of clone
-      }
-    }
-  }
-
-  /**
-   * Create an export-friendly replacement material for a material using refraction.
-   * @param mat Material to be replaced
-   */
-  protected createRefractionMaterialReplacement(mat: PBRMaterial): PBRMaterial {
-    // if we're dealing with a clone already, return it instead of cloning
-    if (this.isMaterialClonedForExport(mat)) return mat;
-
-    // change material according to https://www.notion.so/combeenation/Glas-materials-don-t-look-glasy-after-export-d5fda2c6515e4420a8772744d3e6b460
-    const clonedMaterial = mat.clone(mat.name); // clone material. clone uses same name
-    clonedMaterial.metadata = { ...mat.metadata, clonedFrom: mat.uniqueId }; // create shallow copy of metadata on clone. see https://forum.babylonjs.com/t/the-metadata-of-the-mesh-cloned-by-the-instantiatemodelstoscene-method-is-a-shallow-copy/21563
-    clonedMaterial.refractionTexture = null;
-    clonedMaterial.metallicReflectanceTexture = null; // is this the correct one for metallic roughness?
-    clonedMaterial.alpha = 0.7;
-    clonedMaterial.albedoColor = new Color3(0.3, 0.3, 0.3);
-    clonedMaterial.transparencyMode = PBRMaterial.PBRMATERIAL_ALPHABLEND;
-    clonedMaterial.metallic = 0.65;
-    clonedMaterial.roughness = 0.15;
-
-    return clonedMaterial;
-  }
-
-  /**
-   * Inspect if a material was temporarily cloned for GLB export
-   * @param mat Material to be inspected
-   */
-  protected isMaterialClonedForExport(mat: PBRMaterial): boolean {
-    return !!mat.metadata?.clonedFrom;
-  }
-}
+import { injectNodeMetadata } from '../util/babylonHelper';
+import { isMeshIncludedInExclusionList } from '../util/structureHelper';
+import { PBRMaterial } from '@babylonjs/core/Materials/PBR/pbrMaterial';
+import { Color3 } from '@babylonjs/core/Maths/math.color';
+import { AbstractMesh } from '@babylonjs/core/Meshes/abstractMesh';
+import { Mesh } from '@babylonjs/core/Meshes/mesh';
+import { GLTF2Export, IExportOptions } from '@babylonjs/serializers/glTF/2.0';
+import { has, merge } from 'lodash-es';
+
+export class GltfExportManager {
+  public static readonly UNIQUE_MESH_METADATA_PROPERTY = 'originalName';
+
+  /**
+   * Constructor.
+   */
+  protected constructor(protected viewer: Viewer) {}
+
+  /**
+   * Creates an {@link GltfExportManager}.
+   */
+  public static async create(viewer: Viewer): Promise<GltfExportManager> {
+    return new GltfExportManager(viewer);
+  }
+
+  /**
+   * Exports selected nodes to a file.
+   * @param filename optional name of the exported .GLB file.
+   * @param exportOptions export options to be merged with default options.\
+   *                      `exportOptions.createUniqueMeshNames` defaults to `true` if not given.
+   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
+   */
+  public async exportGlb(
+    filename = 'glb-export.glb',
+    { createUniqueMeshNames = true, ...exportOptionsRest }: IExportOptionsExtended = {},
+    excluded?: ExcludedGeometryList
+  ): Promise<File | undefined> {
+    const exportOptions = { createUniqueMeshNames, ...exportOptionsRest };
+    this.exportPreProcess(exportOptions);
+    const glbData = await GLTF2Export.GLBAsync(
+      this.viewer.scene,
+      'dummy',
+      this.gltfExportOptions(exportOptions, excluded)
+    );
+    this.exportPostProcess();
+    const resBlob = glbData.glTFFiles['dummy.glb']; // should be only one file for glb
+
+    // check if result is valid, according to the typings this could also be a string
+    if (resBlob instanceof Blob) {
+      if (!filename.endsWith('.glb')) {
+        filename += '.glb';
+      }
+      return new File([resBlob], filename);
+    } else {
+      // result was not a valid blob
+      return undefined;
+    }
+  }
+
+  /**
+   * Exports selected nodes to GLTF. This may result in more than one file, since textures are exported separately.
+   * @param filename name of the main (text-based) .GLTF file referring to separate texture files.
+   * @param exportOptions export options to be merged with default options.\
+   *                      `exportOptions.createUniqueMeshNames` defaults to `true` if not given.
+   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
+   */
+  public async exportGltfToFile(
+    filename: string,
+    { createUniqueMeshNames = true, ...exportOptionsRest }: IExportOptionsExtended = {},
+    excluded?: ExcludedGeometryList
+  ) {
+    const exportOptions = { createUniqueMeshNames, ...exportOptionsRest };
+    this.exportPreProcess(exportOptions);
+    await GLTF2Export.GLTFAsync(this.viewer.scene, filename, this.gltfExportOptions(exportOptions, excluded)).then(
+      glb => {
+        glb.downloadFiles();
+      }
+    );
+    this.exportPostProcess();
+  }
+
+  /**
+   * Exports selected nodes to GLB. This results in one binary file.
+   * @param filename name of the .GLB file.
+   * @param exportOptions export options to be merged with default options.\
+   *                      `exportOptions.createUniqueMeshNames` defaults to `true` if not given.
+   * @param excluded optional list of geometry (meshes, elements, variants, variantInstances) to be excluded from export
+   */
+  public async exportGlbToFile(
+    filename: string,
+    { createUniqueMeshNames = true, ...exportOptionsRest }: IExportOptionsExtended = {},
+    excluded?: ExcludedGeometryList
+  ) {
+    const exportOptions = { createUniqueMeshNames, ...exportOptionsRest };
+    this.exportPreProcess(exportOptions);
+    await GLTF2Export.GLBAsync(this.viewer.scene, filename, this.gltfExportOptions(exportOptions, excluded)).then(
+      glb => {
+        glb.downloadFiles();
+      }
+    );
+    this.exportPostProcess();
+  }
+
+  /**
+   * Gets predefined {@link IExportOptionsExtended } merged with given ones.
+   */
+  protected gltfExportOptions(
+    mergeWithOptions: IExportOptionsExtended = {},
+    excluded?: ExcludedGeometryList
+  ): IExportOptions {
+    const defaultOptions: IExportOptions = {
+      // includeCoordinateSystemConversionNodes: true,
+      shouldExportNode: function (node: any) {
+        if (!node.isEnabled()) {
+          return false;
+        }
+        if ((node.name as string).startsWith('__bounding_box__')) {
+          return false;
+        }
+        if (excluded && node instanceof Mesh && isMeshIncludedInExclusionList(node, excluded)) {
+          return false;
+        }
+        return true;
+      },
+    };
+    return merge({}, defaultOptions, mergeWithOptions);
+  }
+
+  /**
+   * Stuff to be done before exporting to GLTF
+   */
+  protected exportPreProcess({ createUniqueMeshNames = true }: IExportOptionsExtended) {
+    this.viewer.pauseRendering();
+    if (createUniqueMeshNames ?? true) {
+      this.setUniqueMeshNames();
+    }
+    this.exchangeRefractionMaterials();
+  }
+
+  /**
+   * Stuff to be done after the GLTF export
+   */
+  protected exportPostProcess() {
+    this.restoreRefractionMaterials();
+    this.restoreOriginalMeshNames();
+    this.viewer.resumeRendering();
+  }
+
+  /**
+   * Materials with refraction set are not exported properly.
+   * Exchange all such (relevant) materials with a more export-friendly version
+   */
+  protected exchangeRefractionMaterials() {
+    for (const n of this.viewer.scene.getNodes()) {
+      if (!(n instanceof AbstractMesh)) continue;
+      if (!(n.material instanceof PBRMaterial)) continue;
+      if (!(n.material as PBRMaterial).subSurface.isRefractionEnabled) continue;
+      if ((n.material as PBRMaterial).transparencyMode !== PBRMaterial.PBRMATERIAL_OPAQUE) continue;
+      // if we're here, we have a node holding a material with set refraction whose transparencyMode is set to
+      // PBRMATERIAL_OPAQUE
+      n.material = this.createRefractionMaterialReplacement(n.material);
+    }
+  }
+
+  /**
+   * Restore original materials with set refraction.
+   */
+  protected restoreRefractionMaterials() {
+    for (const n of this.viewer.scene.getNodes()) {
+      if (!(n instanceof AbstractMesh)) continue;
+      if (!(n.material instanceof PBRMaterial)) continue;
+      if (!this.isMaterialClonedForExport(n.material)) continue;
+      // at this point we have a pbrmaterial tagged as cloned in its metadata that's set on a mesh
+
+      // restore and dispose
+      const currMaterial = n.material;
+      const prevMaterial = this.viewer.scene.getMaterialByUniqueID(n.material.metadata.clonedFrom);
+      if (prevMaterial) {
+        n.material = prevMaterial; // restore previous material
+        currMaterial.dispose(false, true); // dispose of clone
+      }
+    }
+  }
+
+  /**
+   * Create an export-friendly replacement material for a material using refraction.
+   * @param mat Material to be replaced
+   */
+  protected createRefractionMaterialReplacement(mat: PBRMaterial): PBRMaterial {
+    // if we're dealing with a clone already, return it instead of cloning
+    if (this.isMaterialClonedForExport(mat)) return mat;
+
+    // change material according to https://www.notion.so/combeenation/Glas-materials-don-t-look-glasy-after-export-d5fda2c6515e4420a8772744d3e6b460
+    const clonedMaterial = mat.clone(mat.name); // clone material. clone uses same name
+    clonedMaterial.metadata = { ...mat.metadata, clonedFrom: mat.uniqueId }; // create shallow copy of metadata on clone. see https://forum.babylonjs.com/t/the-metadata-of-the-mesh-cloned-by-the-instantiatemodelstoscene-method-is-a-shallow-copy/21563
+    clonedMaterial.refractionTexture = null;
+    clonedMaterial.metallicReflectanceTexture = null; // is this the correct one for metallic roughness?
+    clonedMaterial.alpha = 0.7;
+    clonedMaterial.albedoColor = new Color3(0.3, 0.3, 0.3);
+    clonedMaterial.transparencyMode = PBRMaterial.PBRMATERIAL_ALPHABLEND;
+    clonedMaterial.metallic = 0.65;
+    clonedMaterial.roughness = 0.15;
+
+    return clonedMaterial;
+  }
+
+  /**
+   * Inspect if a material was temporarily cloned for GLB export
+   * @param mat Material to be inspected
+   */
+  protected isMaterialClonedForExport(mat: PBRMaterial): boolean {
+    return !!mat.metadata?.clonedFrom;
+  }
+
+  /**
+   * Set unique mesh names for GLB export
+   */
+  protected setUniqueMeshNames(): void {
+    this.viewer.scene.meshes.forEach(currMesh => {
+      injectNodeMetadata(currMesh, { [GltfExportManager.UNIQUE_MESH_METADATA_PROPERTY]: currMesh.name }, false);
+      currMesh.name = `${currMesh.name}_${currMesh.uniqueId}`;
+    });
+  }
+
+  /**
+   * Restore original mesh names for meshes that were
+   * renamed with {@link setUniqueMeshNames()}
+   */
+  protected restoreOriginalMeshNames(): void {
+    this.viewer.scene.meshes.forEach(currMesh => {
+      if (has(currMesh.metadata, GltfExportManager.UNIQUE_MESH_METADATA_PROPERTY)) {
+        currMesh.name = currMesh.metadata[GltfExportManager.UNIQUE_MESH_METADATA_PROPERTY];
+        delete currMesh.metadata[GltfExportManager.UNIQUE_MESH_METADATA_PROPERTY];
+      }
+    });
+  }
+}