feat: store spatial elements within fragment group

packages/core/src/fragments/Classifier/index.ts

@@ -342,7 +342,9 @@ export class Classifier extends Component implements Disposable {
    * Classifies fragments based on their spatial structure in the IFC model.
    *
    * @param model - The FragmentsGroup containing the fragments to be classified.
-   * @param config - The configuration for the classifier. It includes "useProperties" (if false, the classification will use the expressIDs instead of the names)
+   * @param config - The configuration for the classifier. It includes "useProperties", which is true by default
+   * (if false, the classification will use the expressIDs instead of the names), and "isolate", which will make
+   * the classifier just pick the WEBIFC categories provided.
    *
    * @remarks
    * This method iterates through the relations of the fragments in the provided group,
@@ -354,7 +356,7 @@ export class Classifier extends Component implements Disposable {
    */
   async bySpatialStructure(
     model: FRAGS.FragmentsGroup,
-    config: { useProperties: boolean } = { useProperties: true },
+    config: { useProperties?: boolean; isolate?: Set<number> },
   ) {
     const indexer = this.components.get(IfcRelationsIndexer);
     const modelRelations = indexer.relationMaps[model.uuid];
@@ -364,7 +366,22 @@ export class Classifier extends Component implements Disposable {
       );
     }
     const systemName = "spatialStructures";
+
+    // If useProperties is undefined, use properties by default
+    const propsUndefined = config.useProperties === undefined;
+    const useProperties = propsUndefined || config.useProperties;
+
     for (const [expressID] of modelRelations) {
+      // E.g. if the user just wants the building storeys
+      if (config.isolate) {
+        const data = model.data.get(expressID);
+        if (!data) continue;
+        const category = data[1][1];
+        if (category === undefined || !config.isolate.has(category)) {
+          continue;
+        }
+      }
+
       const spatialRels = indexer.getEntityRelations(
         model,
         expressID,
@@ -375,7 +392,7 @@ export class Classifier extends Component implements Disposable {
       if (spatialRels) {
         for (const id of spatialRels) {
           let relName = id.toString();
-          if (config.useProperties) {
+          if (useProperties) {
             const spatialRelAttrs = await model.getProperties(id);
             if (!spatialRelAttrs) {
               continue;
@@ -398,7 +415,7 @@ export class Classifier extends Component implements Disposable {
       }
 
       let relName = expressID.toString();
-      if (config.useProperties) {
+      if (useProperties) {
         const relAttrs = await model.getProperties(expressID);
         if (!relAttrs) {
           continue;

packages/core/src/fragments/Exploder/example.ts

@@ -124,7 +124,9 @@ Before being able to use it, we will need to get the classifier to classify the
 */
 
 const classifier = components.get(OBC.Classifier);
-await classifier.bySpatialStructure(model);
+await classifier.bySpatialStructure(model, {
+  isolate: new Set([WEBIFC.IFCBUILDINGSTOREY]),
+});
 
 /* MD
   ### ⏱️ Measuring the performance (optional)

packages/core/src/fragments/IfcGeometryTiler/index.ts

@@ -8,6 +8,7 @@ import {
   SpatialStructure,
   CivilReader,
   IfcMetadataReader,
+  SpatialIdsFinder,
 } from "../IfcLoader/src";
 
 export * from "./src";
@@ -278,13 +279,15 @@ export class IfcGeometryTiler extends Component implements Disposable {
     }
 
     // Delete assets that have no geometric representation
-    const ids = group.data.keys();
-    for (const id of ids) {
-      const [keys] = group.data.get(id)!;
-      if (!keys.length) {
-        group.data.delete(id);
-      }
-    }
+    // const ids = group.data.keys();
+    // for (const id of ids) {
+    //   const [keys] = group.data.get(id)!;
+    //   if (!keys.length) {
+    //     group.data.delete(id);
+    //   }
+    // }
+
+    SpatialIdsFinder.get(group, this.webIfc);
 
     const matrix = this.webIfc.GetCoordinationMatrix(0);
     group.coordinationMatrix.fromArray(matrix);

packages/core/src/fragments/IfcLoader/index.ts

@@ -6,6 +6,7 @@ import { CivilReader, IfcFragmentSettings, IfcMetadataReader } from "./src";
 import { FragmentsManager } from "../FragmentsManager";
 import { Component, Components, Event, Disposable } from "../../core";
 import { IfcJsonExporter } from "../../ifc/IfcJsonExporter";
+import { SpatialIdsFinder } from "./src/spatial-ids-finder.ts";
 
 export * from "./src/ifc-fragment-settings";
 
@@ -126,8 +127,6 @@ export class IfcLoader extends Component implements Disposable {
     const properties = await jsonExporter.export(this.webIfc, 0);
     group.setLocalProperties(properties);
 
-    this.cleanUp();
-
     const fragments = this.components.get(FragmentsManager);
     fragments.groups.set(group.uuid, group);
 
@@ -139,7 +138,9 @@ export class IfcLoader extends Component implements Disposable {
 
     fragments.onFragmentsLoaded.trigger(group);
 
-    if (coordinate) fragments.coordinate([group]);
+    if (coordinate) {
+      fragments.coordinate([group]);
+    }
 
     for (const [expressID] of group.data) {
       const props = properties[expressID];
@@ -148,6 +149,10 @@ export class IfcLoader extends Component implements Disposable {
       group.globalToExpressIDs.set(globalID, expressID);
     }
 
+    SpatialIdsFinder.get(group, this.webIfc);
+
+    this.cleanUp();
+
     console.log(`Streaming the IFC took ${performance.now() - before} ms!`);
 
     return group;

packages/core/src/fragments/IfcLoader/src/index.ts

@@ -2,3 +2,4 @@ export * from "./ifc-fragment-settings";
 export * from "./spatial-structure";
 export * from "./civil-reader";
 export * from "./ifc-metadata-reader";
+export * from "./spatial-ids-finder";

packages/core/src/fragments/IfcLoader/src/spatial-ids-finder.ts

@@ -0,0 +1,31 @@
+import * as FRAGS from "@thatopen/fragments";
+import * as WEBIFC from "web-ifc";
+
+export class SpatialIdsFinder {
+  static get(model: FRAGS.FragmentsGroup, webIfc: WEBIFC.IfcAPI) {
+    const spatialTypes = [
+      WEBIFC.IFCPROJECT,
+      WEBIFC.IFCSITE,
+      WEBIFC.IFCBUILDING,
+      WEBIFC.IFCBUILDINGSTOREY,
+      WEBIFC.IFCSPACE,
+      WEBIFC.IFCROAD,
+      WEBIFC.IFCFACILITY,
+      WEBIFC.IFCFACILITYPART,
+      WEBIFC.IFCBRIDGE,
+    ];
+
+    const data = model.data;
+
+    for (const category of spatialTypes) {
+      const ids = webIfc.GetLineIDsWithType(0, category);
+      const size = ids.size();
+      for (let i = 0; i < size; i++) {
+        const id = ids.get(i);
+        if (!data.has(id)) {
+          data.set(id, [[], [0, category]]);
+        }
+      }
+    }
+  }
+}