@itwin/tree-widget-react 4.0.0-alpha.13 → 4.0.0-alpha.14

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeIdsCache.d.ts

@@ -1,6 +1,7 @@
+import type { Observable } from "rxjs";
 import type { Id64Array, Id64String } from "@itwin/core-bentley";
-import type { CategoryId, ModelId } from "../../common/internal/Types.js";
-import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { CategoryId, ElementId, ModelId } from "../../common/internal/Types.js";
+import type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
 import type { ClassificationsTreeHierarchyConfiguration } from "../ClassificationsTreeDefinition.js";
 /** @internal */
 export type ClassificationId = Id64String;
@@ -14,6 +15,7 @@ export declare class ClassificationsTreeIdsCache implements Disposable {
     private _elementModelsCategories;
     private _modelWithCategoryModeledElements;
     private _classificationInfos;
+    private _filteredElementsData;
     constructor(_queryExecutor: LimitingECSqlQueryExecutor, _hierarchyConfig: ClassificationsTreeHierarchyConfiguration);
     [Symbol.dispose](): void;
     private queryElementModelCategories;
@@ -21,7 +23,7 @@ export declare class ClassificationsTreeIdsCache implements Disposable {
     private queryModeledElements;
     private getModelWithCategoryModeledElements;
     getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array>;
-    getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Array<ModelId>>>;
+    getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Set<ModelId>>>;
     getModelCategoryIds(modelId: Id64String): Promise<{
         drawing: Id64Array;
         spatial: Id64Array;
@@ -35,5 +37,16 @@ export declare class ClassificationsTreeIdsCache implements Disposable {
         spatial: Id64Array;
     }>;
     getDirectChildClassifications(classificationOrTableIds: Id64Array): Promise<ClassificationId[]>;
+    getClassificationsPathObs(classificationIds: Id64Array): Observable<HierarchyNodeIdentifiersPath>;
+    getAllClassifications(): Promise<ClassificationId[]>;
+    private queryFilteredElementsData;
+    getFilteredElementsData({ element2dIds, element3dIds, }: {
+        element2dIds: Id64Array;
+        element3dIds: Id64Array;
+    }): Promise<Map<ElementId, {
+        categoryId: Id64String;
+        modelId: Id64String;
+    }>>;
+    clearFilteredElementsData(): void;
 }
 //# sourceMappingURL=ClassificationsTreeIdsCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeIdsCache.js

@@ -2,7 +2,7 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { assert } from "@itwin/core-bentley";
+import { from, map, mergeMap } from "rxjs";
 import { CLASS_NAME_Classification, CLASS_NAME_ClassificationSystem, CLASS_NAME_ClassificationTable, CLASS_NAME_DrawingCategory, CLASS_NAME_ElementHasClassifications, CLASS_NAME_GeometricElement2d, CLASS_NAME_GeometricElement3d, CLASS_NAME_SpatialCategory, } from "../../common/internal/ClassNameDefinitions.js";
 import { ModelCategoryElementsCountCache } from "../../common/internal/ModelCategoryElementsCountCache.js";
 import { getDistinctMapValues } from "../../common/internal/Utils.js";
@@ -14,6 +14,7 @@ export class ClassificationsTreeIdsCache {
     _elementModelsCategories;
     _modelWithCategoryModeledElements;
     _classificationInfos;
+    _filteredElementsData;
     constructor(_queryExecutor, _hierarchyConfig) {
         this._queryExecutor = _queryExecutor;
         this._hierarchyConfig = _hierarchyConfig;
@@ -140,10 +141,10 @@ export class ClassificationsTreeIdsCache {
                 if ((includeSubModels || !isSubModel) && (category2dIds.has(categoryId) || category3dIds.has(categoryId))) {
                     let categoryModels = result.get(categoryId);
                     if (!categoryModels) {
-                        categoryModels = new Array();
+                        categoryModels = new Set();
                         result.set(categoryId, categoryModels);
                     }
-                    categoryModels.push(modelId);
+                    categoryModels.add(modelId);
                 }
             }
         }
@@ -233,14 +234,20 @@ export class ClassificationsTreeIdsCache {
             const classificationInfos = new Map();
             for await (const { id, tableId, parentId, relatedCategories2d, relatedCategories3d } of this.queryClassifications()) {
                 const tableOrParentId = tableId ?? parentId;
-                assert(!!tableOrParentId);
                 let parentInfo = classificationInfos.get(tableOrParentId);
                 if (!parentInfo) {
-                    parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [] };
+                    parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [], parentClassificationOrTableId: undefined };
                     classificationInfos.set(tableOrParentId, parentInfo);
                 }
                 parentInfo.childClassificationIds.push(id);
-                classificationInfos.set(id, { childClassificationIds: [], relatedCategories2d, relatedCategories3d });
+                let classificationEntry = classificationInfos.get(id);
+                if (!classificationEntry) {
+                    classificationEntry = { childClassificationIds: [], relatedCategories2d, relatedCategories3d, parentClassificationOrTableId: tableOrParentId };
+                    classificationInfos.set(id, classificationEntry);
+                }
+                else {
+                    classificationEntry.parentClassificationOrTableId = tableOrParentId;
+                }
             }
             return classificationInfos;
         })();
@@ -269,14 +276,73 @@ export class ClassificationsTreeIdsCache {
             return result;
         }
         const classificationsInfo = await this.getClassificationsInfo();
-        await Promise.all(classificationOrTableIds.map(async (classificationOrTableId) => {
+        classificationOrTableIds.forEach((classificationOrTableId) => {
             const classificationInfo = classificationsInfo.get(classificationOrTableId);
             if (classificationInfo === undefined) {
                 return;
             }
             result.push(...classificationInfo.childClassificationIds);
-        }));
+        });
         return result;
     }
+    getClassificationsPathObs(classificationIds) {
+        return from(this.getClassificationsInfo()).pipe(mergeMap((classificationsInfo) => {
+            return from(classificationIds).pipe(map((classificationId) => {
+                const path = [{ id: classificationId, className: CLASS_NAME_Classification }];
+                let parentId = classificationsInfo.get(classificationId)?.parentClassificationOrTableId;
+                while (parentId !== undefined) {
+                    const parentIdOfParent = classificationsInfo.get(parentId)?.parentClassificationOrTableId;
+                    if (parentIdOfParent) {
+                        path.push({ className: CLASS_NAME_Classification, id: parentId });
+                    }
+                    else {
+                        path.push({ className: CLASS_NAME_ClassificationTable, id: parentId });
+                    }
+                    parentId = parentIdOfParent;
+                }
+                return path.reverse();
+            }));
+        }));
+    }
+    async getAllClassifications() {
+        const classificationsInfo = await this.getClassificationsInfo();
+        return [...classificationsInfo.keys()];
+    }
+    async *queryFilteredElementsData({ element2dIds, element3dIds }) {
+        const queries = new Array();
+        if (element2dIds.length > 0) {
+            queries.push(`
+        SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id
+        FROM ${CLASS_NAME_GeometricElement2d}
+        WHERE ECInstanceId IN (${element2dIds.join(", ")})
+      `);
+        }
+        if (element3dIds.length > 0) {
+            queries.push(`
+        SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id
+        FROM ${CLASS_NAME_GeometricElement3d}
+        WHERE ECInstanceId IN (${element3dIds.join(", ")})
+      `);
+        }
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: queries.join(" UNION ALL ") }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { modelId: row.modelId, id: row.id, categoryId: row.categoryId };
+        }
+    }
+    async getFilteredElementsData({ element2dIds, element3dIds, }) {
+        if (element2dIds.length === 0 && element3dIds.length === 0) {
+            return new Map();
+        }
+        this._filteredElementsData ??= (async () => {
+            const filteredElementsData = new Map();
+            for await (const { modelId, id, categoryId } of this.queryFilteredElementsData({ element2dIds, element3dIds })) {
+                filteredElementsData.set(id, { modelId, categoryId });
+            }
+            return filteredElementsData;
+        })();
+        return this._filteredElementsData;
+    }
+    clearFilteredElementsData() {
+        this._filteredElementsData = undefined;
+    }
 }
 //# sourceMappingURL=ClassificationsTreeIdsCache.js.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeIdsCache.js.map

@@ -1 +1 @@
-{"version":3,"file":"ClassificationsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,0BAA0B,EAC1B,oCAAoC,EACpC,6BAA6B,EAC7B,6BAA6B,EAC7B,0BAA0B,GAC3B,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AAmBtE,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAO5B;IACA;IAPO,sBAAsB,CAAkC;IACjE,wBAAwB,CAA6G;IACrI,iCAAiC,CAA6D;IAC9F,oBAAoB,CAAyF;IAErH,YACU,cAA0C,EAC1C,gBAA2D;QAD3D,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAA2C;QAEnE,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,cAAc,EAAE,CAAC,4BAA4B,EAAE,4BAA4B,CAAC,CAAC,CAAC;IAClJ,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B;QAKxC,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;KAgBb,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,sEAAsE,EAAE,CAC9I,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;QAC7E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,IAAI,CAAC,wBAAwB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,CAAC,eAAe,EAAE,gCAAgC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5E,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAA+D,CAAC;oBACvG,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE,CAAC;wBACvE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;wBACtE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;4BAC7B,UAAU,GAAG,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;4BACpE,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;wBACnE,CAAC;wBACD,QAAQ,eAAe,CAAC,IAAI,EAAE,CAAC;4BAC7B,KAAK,IAAI;gCACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gCACzD,MAAM;4BACR,KAAK,IAAI;gCACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gCACzD,MAAM;wBACV,CAAC;oBACH,CAAC;oBACD,OAAO,uBAAuB,CAAC;gBACjC,CAAC,CAAC,EAAE;gBACJ,IAAI,CAAC,mCAAmC,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoF,CAAC;YAC3G,MAAM,SAAS,GAAG,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YACzE,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,CAAC,aAAa,EAAE,aAAa,EAAE,UAAU,CAAC,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;YACxH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;;;;;KAoBb,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,yDAAyD,EAAE,CACjI,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QACzI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAoC,CAAC;YACrF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAsB;QACnF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;QACtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,WAAsB,EAAE,gBAA0B;QACxF,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAA8B,CAAC;QACrD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBAC9F,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;oBAC1G,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE,CAAC;wBACpB,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;wBACtC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBACzC,CAAC;oBACD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,OAAmB;QAClD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;YAC9E,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;SAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAOjC,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;QAC9C,MAAM,IAAI,GAAG;YACX;UACI,mBAAmB;;;;;iBAKZ,yBAAyB;iBACzB,8BAA8B;iBAC9B,+BAA+B;;8BAElB,IAAI,CAAC,gBAAgB,CAAC,4BAA4B;;;;;;;;;;;;cAYlE,mBAAmB;mBACd,yBAAyB;;;;OAIrC;SACF,CAAC;QACF,MAAM,KAAK,GAAG;;;;;;;iBAOD,6BAA6B;iBAC7B,0BAA0B;iBAC1B,oCAAoC;;;;;;iBAMpC,6BAA6B;iBAC7B,0BAA0B;iBAC1B,oCAAoC;;;;aAIxC,mBAAmB;KAC3B,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,IAAI,EAAE,KAAK,EAAE,EACf,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,wDAAwD,EAAE,CAChI,EAAE,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,mBAAmB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAE,GAAG,CAAC,mBAA8B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;gBAClG,mBAAmB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAE,GAAG,CAAC,mBAA8B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;aACnG,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAClC,IAAI,CAAC,oBAAoB,KAAK,CAAC,KAAK,IAAI,EAAE;YACxC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAgE,CAAC;YACpG,IAAI,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBACpH,MAAM,eAAe,GAAG,OAAO,IAAI,QAAQ,CAAC;gBAC5C,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;gBAC1B,IAAI,UAAU,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;gBAC1D,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,CAAC;oBAC9F,mBAAmB,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;gBACvD,CAAC;gBACD,UAAU,CAAC,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAE3C,mBAAmB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACxG,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,wBAAmC;QACxE,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACtF,IAAI,wBAAwB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChE,MAAM,OAAO,CAAC,GAAG,CACf,wBAAwB,CAAC,GAAG,CAAC,KAAK,EAAE,uBAAuB,EAAE,EAAE;YAC7D,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,MAAM,0BAA0B,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;YACnH,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,EAAE,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;YACtG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,EAAE,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACxG,CAAC,CAAC,CACH,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,wBAAmC;QAC5E,MAAM,MAAM,GAAG,IAAI,KAAK,EAAoB,CAAC;QAC7C,IAAI,wBAAwB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChE,MAAM,OAAO,CAAC,GAAG,CACf,wBAAwB,CAAC,GAAG,CAAC,KAAK,EAAE,uBAAuB,EAAE,EAAE;YAC7D,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;QAC5D,CAAC,CAAC,CACH,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { assert } from \"@itwin/core-bentley\";\nimport {\n  CLASS_NAME_Classification,\n  CLASS_NAME_ClassificationSystem,\n  CLASS_NAME_ClassificationTable,\n  CLASS_NAME_DrawingCategory,\n  CLASS_NAME_ElementHasClassifications,\n  CLASS_NAME_GeometricElement2d,\n  CLASS_NAME_GeometricElement3d,\n  CLASS_NAME_SpatialCategory,\n} from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getDistinctMapValues } from \"../../common/internal/Utils.js\";\n\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { CategoryId, ElementId, ModelId } from \"../../common/internal/Types.js\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { ClassificationsTreeHierarchyConfiguration } from \"../ClassificationsTreeDefinition.js\";\n/** @internal */\nexport type ClassificationId = Id64String;\n/** @internal */\nexport type ClassificationTableId = Id64String;\n\ninterface ClassificationInfo {\n  childClassificationIds: ClassificationId[];\n  relatedCategories2d: CategoryId[];\n  relatedCategories3d: CategoryId[];\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class ClassificationsTreeIdsCache implements Disposable {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _elementModelsCategories: Promise<Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>> | undefined;\n  private _modelWithCategoryModeledElements: Promise<Map<ModelCategoryKey, Set<ElementId>>> | undefined;\n  private _classificationInfos: Promise<Map<ClassificationId | ClassificationTableId, ClassificationInfo>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ClassificationsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, [\"BisCore.GeometricElement2d\", \"BisCore.GeometricElement3d\"]);\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *queryElementModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n    type: \"2d\" | \"3d\";\n  }> {\n    const query = `\n      SELECT * FROM (\n        SELECT '3d' type, this.Model.Id modelId, this.Category.Id categoryId\n        FROM BisCore.GeometricModel m\n        JOIN BisCore.GeometricElement3d this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      )\n      UNION ALL\n      SELECT * FROM (\n        SELECT '2d' type, this.Model.Id modelId, this.Category.Id categoryId\n        FROM BisCore.GeometricModel m\n        JOIN BisCore.GeometricElement2d this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      )\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/element-models-and-categories-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, type: row.type };\n    }\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([\n        (async () => {\n          const elementModelsCategories = new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set }>();\n          for await (const queriedCategory of this.queryElementModelCategories()) {\n            let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n            if (modelEntry === undefined) {\n              modelEntry = { category2dIds: new Set(), category3dIds: new Set() };\n              elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n            }\n            switch (queriedCategory.type) {\n              case \"2d\":\n                modelEntry.category2dIds.add(queriedCategory.categoryId);\n                break;\n              case \"3d\":\n                modelEntry.category3dIds.add(queriedCategory.categoryId);\n                break;\n            }\n          }\n          return elementModelsCategories;\n        })(),\n        this.getModelWithCategoryModeledElements(),\n      ]);\n      const result = new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>();\n      const subModels = getDistinctMapValues(modelWithCategoryModeledElements);\n      for (const [modelId, modelEntry] of modelCategories) {\n        const isSubModel = subModels.has(modelId);\n        result.set(modelId, { category2dIds: modelEntry.category2dIds, category3dIds: modelEntry.category3dIds, isSubModel });\n      }\n      return result;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async *queryModeledElements(): AsyncIterableIterator<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.GeometricModel m\n      JOIN BisCore.GeometricElement3d pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)\n      UNION ALL\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.GeometricModel m\n      JOIN BisCore.GeometricElement2d pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/modeled-elements-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<ModelCategoryKey, Set<ElementId>>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key: ModelCategoryKey = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<ElementId>();\n    for (const categoryId of categoryIds) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Array<ModelId>>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<CategoryId, Array<ModelId>>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, { category2dIds, category3dIds, isSubModel }] of elementModelsCategories) {\n        if ((includeSubModels || !isSubModel) && (category2dIds.has(categoryId) || category3dIds.has(categoryId))) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Array<ModelId>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.push(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getModelCategoryIds(modelId: Id64String): Promise<{ drawing: Id64Array; spatial: Id64Array }> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return {\n      drawing: Array.from(elementModelsCategories.get(modelId)?.category2dIds ?? []),\n      spatial: Array.from(elementModelsCategories.get(modelId)?.category3dIds ?? []),\n    };\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return elementModelsCategories.has(elementId);\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  private async *queryClassifications(): AsyncIterableIterator<{\n    id: Id64String;\n    tableId?: ClassificationTableId;\n    parentId?: ClassificationId;\n    relatedCategories2d: CategoryId[];\n    relatedCategories3d: CategoryId[];\n  }> {\n    const CLASSIFICATIONS_CTE = \"Classifications\";\n    const ctes = [\n      `\n        ${CLASSIFICATIONS_CTE}(ClassificationId, ClassificationTableId, ParentClassificationId) AS (\n          SELECT\n            cl.ECInstanceId,\n            ct.ECInstanceId,\n            NULL\n          FROM ${CLASS_NAME_Classification} cl\n          JOIN ${CLASS_NAME_ClassificationTable} ct ON ct.ECInstanceId = cl.Model.Id\n          JOIN ${CLASS_NAME_ClassificationSystem} cs ON cs.ECInstanceId = ct.Parent.Id\n          WHERE\n            cs.CodeValue = '${this._hierarchyConfig.rootClassificationSystemCode}'\n            AND NOT ct.IsPrivate\n            AND NOT cl.IsPrivate\n            AND cl.Parent.Id IS NULL\n\n          UNION ALL\n\n          SELECT\n            cl.ECInstanceId,\n            NULL,\n            cl.Parent.Id\n          FROM\n            ${CLASSIFICATIONS_CTE} cte\n            JOIN ${CLASS_NAME_Classification} cl ON cl.Parent.Id = cte.ClassificationId\n          WHERE\n            NOT cl.IsPrivate\n        )\n      `,\n    ];\n    const ecsql = `\n      SELECT\n        cl.ClassificationId id,\n        cl.ClassificationTableId tableId,\n        cl.ParentClassificationId parentId,\n        (\n          SELECT group_concat(IdToHex(cat.ECInstanceId))\n          FROM ${CLASS_NAME_GeometricElement2d} e\n          JOIN ${CLASS_NAME_DrawingCategory} cat ON cat.ECInstanceId = e.Category.Id\n          JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId\n          WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId\n          GROUP BY ehc.TargetECInstanceId\n        ) relatedCategories2d,\n        (\n          SELECT group_concat(IdToHex(cat.ECInstanceId))\n          FROM ${CLASS_NAME_GeometricElement3d} e\n          JOIN ${CLASS_NAME_SpatialCategory} cat ON cat.ECInstanceId = e.Category.Id\n          JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId\n          WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId\n          GROUP BY ehc.TargetECInstanceId\n        ) relatedCategories3d\n      FROM ${CLASSIFICATIONS_CTE} cl\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ctes, ecsql },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/classifications-query\" },\n    )) {\n      yield {\n        id: row.id,\n        tableId: row.tableId,\n        parentId: row.parentId,\n        relatedCategories2d: row.relatedCategories2d ? (row.relatedCategories2d as string).split(\",\") : [],\n        relatedCategories3d: row.relatedCategories3d ? (row.relatedCategories3d as string).split(\",\") : [],\n      };\n    }\n  }\n\n  private async getClassificationsInfo() {\n    this._classificationInfos ??= (async () => {\n      const classificationInfos = new Map<ClassificationId | ClassificationTableId, ClassificationInfo>();\n      for await (const { id, tableId, parentId, relatedCategories2d, relatedCategories3d } of this.queryClassifications()) {\n        const tableOrParentId = tableId ?? parentId;\n        assert(!!tableOrParentId);\n        let parentInfo = classificationInfos.get(tableOrParentId);\n        if (!parentInfo) {\n          parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [] };\n          classificationInfos.set(tableOrParentId, parentInfo);\n        }\n        parentInfo.childClassificationIds.push(id);\n\n        classificationInfos.set(id, { childClassificationIds: [], relatedCategories2d, relatedCategories3d });\n      }\n      return classificationInfos;\n    })();\n    return this._classificationInfos;\n  }\n\n  public async getAllContainedCategories(classificationOrTableIds: Id64Array): Promise<{ drawing: Id64Array; spatial: Id64Array }> {\n    const result = { drawing: new Array<CategoryId>(), spatial: new Array<CategoryId>() };\n    if (classificationOrTableIds.length === 0) {\n      return result;\n    }\n    const classificationsInfo = await this.getClassificationsInfo();\n    await Promise.all(\n      classificationOrTableIds.map(async (classificationOrTableId) => {\n        const classificationInfo = classificationsInfo.get(classificationOrTableId);\n        if (classificationInfo === undefined) {\n          return;\n        }\n        const childClassificationsResult = await this.getAllContainedCategories(classificationInfo.childClassificationIds);\n        result.drawing.push(...classificationInfo.relatedCategories2d, ...childClassificationsResult.drawing);\n        result.spatial.push(...classificationInfo.relatedCategories3d, ...childClassificationsResult.spatial);\n      }),\n    );\n    return result;\n  }\n\n  public async getDirectChildClassifications(classificationOrTableIds: Id64Array): Promise<ClassificationId[]> {\n    const result = new Array<ClassificationId>();\n    if (classificationOrTableIds.length === 0) {\n      return result;\n    }\n    const classificationsInfo = await this.getClassificationsInfo();\n    await Promise.all(\n      classificationOrTableIds.map(async (classificationOrTableId) => {\n        const classificationInfo = classificationsInfo.get(classificationOrTableId);\n        if (classificationInfo === undefined) {\n          return;\n        }\n        result.push(...classificationInfo.childClassificationIds);\n      }),\n    );\n    return result;\n  }\n}\n"]}
+{"version":3,"file":"ClassificationsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAC3C,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,0BAA0B,EAC1B,oCAAoC,EACpC,6BAA6B,EAC7B,6BAA6B,EAC7B,0BAA0B,GAC3B,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AAsBtE,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAQ5B;IACA;IARO,sBAAsB,CAAkC;IACjE,wBAAwB,CAA6G;IACrI,iCAAiC,CAA6D;IAC9F,oBAAoB,CAAyF;IAC7G,qBAAqB,CAAuF;IAEpH,YACU,cAA0C,EAC1C,gBAA2D;QAD3D,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAA2C;QAEnE,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,cAAc,EAAE,CAAC,4BAA4B,EAAE,4BAA4B,CAAC,CAAC,CAAC;IAClJ,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,2BAA2B;QAKxC,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;KAgBb,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,sEAAsE,EAAE,CAC9I,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;QAC7E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,IAAI,CAAC,wBAAwB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC5C,MAAM,CAAC,eAAe,EAAE,gCAAgC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5E,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAA+D,CAAC;oBACvG,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,2BAA2B,EAAE,EAAE,CAAC;wBACvE,IAAI,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;wBACtE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;4BAC7B,UAAU,GAAG,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;4BACpE,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;wBACnE,CAAC;wBACD,QAAQ,eAAe,CAAC,IAAI,EAAE,CAAC;4BAC7B,KAAK,IAAI;gCACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gCACzD,MAAM;4BACR,KAAK,IAAI;gCACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gCACzD,MAAM;wBACV,CAAC;oBACH,CAAC;oBACD,OAAO,uBAAuB,CAAC;gBACjC,CAAC,CAAC,EAAE;gBACJ,IAAI,CAAC,mCAAmC,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoF,CAAC;YAC3G,MAAM,SAAS,GAAG,oBAAoB,CAAC,gCAAgC,CAAC,CAAC;YACzE,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC1C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,CAAC,aAAa,EAAE,aAAa,EAAE,UAAU,CAAC,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;YACxH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAMjC,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;;;;;KAoBb,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,yDAAyD,EAAE,CACjI,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,CAAC;QACzI,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,CAAC,iCAAiC,KAAK,CAAC,KAAK,IAAI,EAAE;YACrD,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAoC,CAAC;YACrF,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBAC1F,MAAM,GAAG,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;gBACzD,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,gCAAgC,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACzE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,OAAO,gCAAgC,CAAC;QAC1C,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,4BAA4B,CAAC,OAAmB,EAAE,WAAsB;QACnF,MAAM,gCAAgC,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;QACtC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,gCAAgC,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;YAC/E,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,WAAsB,EAAE,gBAA0B;QACxF,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,MAAM,MAAM,GAAG,IAAI,GAAG,EAA4B,CAAC;QACnD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,IAAI,uBAAuB,EAAE,CAAC;gBAC9F,IAAI,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;oBAC1G,IAAI,cAAc,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5C,IAAI,CAAC,cAAc,EAAE,CAAC;wBACpB,cAAc,GAAG,IAAI,GAAG,EAAW,CAAC;wBACpC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBACzC,CAAC;oBACD,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,OAAmB;QAClD,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;YAC9E,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;SAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAqB;QAC5C,MAAM,uBAAuB,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACxE,OAAO,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QAOjC,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;QAC9C,MAAM,IAAI,GAAG;YACX;UACI,mBAAmB;;;;;iBAKZ,yBAAyB;iBACzB,8BAA8B;iBAC9B,+BAA+B;;8BAElB,IAAI,CAAC,gBAAgB,CAAC,4BAA4B;;;;;;;;;;;;cAYlE,mBAAmB;mBACd,yBAAyB;;;;OAIrC;SACF,CAAC;QACF,MAAM,KAAK,GAAG;;;;;;;iBAOD,6BAA6B;iBAC7B,0BAA0B;iBAC1B,oCAAoC;;;;;;iBAMpC,6BAA6B;iBAC7B,0BAA0B;iBAC1B,oCAAoC;;;;aAIxC,mBAAmB;KAC3B,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,IAAI,EAAE,KAAK,EAAE,EACf,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,wDAAwD,EAAE,CAChI,EAAE,CAAC;YACF,MAAM;gBACJ,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,mBAAmB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAE,GAAG,CAAC,mBAA8B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;gBAClG,mBAAmB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAE,GAAG,CAAC,mBAA8B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;aACnG,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAClC,IAAI,CAAC,oBAAoB,KAAK,CAAC,KAAK,IAAI,EAAE;YACxC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAgE,CAAC;YACpG,IAAI,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;gBACpH,MAAM,eAAe,GAAG,OAAO,IAAI,QAAQ,CAAC;gBAC5C,IAAI,UAAU,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;gBAC1D,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,6BAA6B,EAAE,SAAS,EAAE,CAAC;oBACxI,mBAAmB,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;gBACvD,CAAC;gBACD,UAAU,CAAC,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC3C,IAAI,mBAAmB,GAAG,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACtD,IAAI,CAAC,mBAAmB,EAAE,CAAC;oBACzB,mBAAmB,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,eAAe,EAAE,CAAC;oBAC/I,mBAAmB,CAAC,GAAG,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC;gBACnD,CAAC;qBAAM,CAAC;oBACN,mBAAmB,CAAC,6BAA6B,GAAG,eAAe,CAAC;gBACtE,CAAC;YACH,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,wBAAmC;QACxE,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACtF,IAAI,wBAAwB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChE,MAAM,OAAO,CAAC,GAAG,CACf,wBAAwB,CAAC,GAAG,CAAC,KAAK,EAAE,uBAAuB,EAAE,EAAE;YAC7D,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,MAAM,0BAA0B,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;YACnH,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,EAAE,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;YACtG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,EAAE,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACxG,CAAC,CAAC,CACH,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,wBAAmC;QAC5E,MAAM,MAAM,GAAG,IAAI,KAAK,EAAoB,CAAC;QAC7C,IAAI,wBAAwB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChE,wBAAwB,CAAC,OAAO,CAAC,CAAC,uBAAuB,EAAE,EAAE;YAC3D,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,yBAAyB,CAAC,iBAA4B;QAC3D,OAAO,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE;YAC/B,OAAO,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjC,GAAG,CAAC,CAAC,gBAAgB,EAAE,EAAE;gBACvB,MAAM,IAAI,GAAiC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAE,SAAS,EAAE,yBAAyB,EAAE,CAAC,CAAC;gBAC5G,IAAI,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,6BAA6B,CAAC;gBACxF,OAAO,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC9B,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,6BAA6B,CAAC;oBAC1F,IAAI,gBAAgB,EAAE,CAAC;wBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;oBACpE,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,8BAA8B,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;oBACzE,CAAC;oBACD,QAAQ,GAAG,gBAAgB,CAAC;gBAC9B,CAAC;gBACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACxB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,qBAAqB;QAChC,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAChE,OAAO,CAAC,GAAG,mBAAmB,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,EAAE,YAAY,EAAE,YAAY,EAAwD;QAK3H,MAAM,OAAO,GAAG,IAAI,KAAK,EAAU,CAAC;QACpC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CAAC;;eAEJ,6BAA6B;iCACX,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;OACjD,CAAC,CAAC;QACL,CAAC;QACD,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CAAC;;eAEJ,6BAA6B;iCACX,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;OACjD,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC3D,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,EACtC,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,EAAE,CAAC;YACF,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QACzE,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,uBAAuB,CAAC,EACnC,YAAY,EACZ,YAAY,GAIb;QACC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3D,OAAO,IAAI,GAAG,EAAE,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAE,CAAC;YACvC,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,yBAAyB,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;gBAC/G,oBAAoB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,oBAAoB,CAAC;QAC9B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEM,yBAAyB;QAC9B,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC;IACzC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { from, map, mergeMap } from \"rxjs\";\nimport {\n  CLASS_NAME_Classification,\n  CLASS_NAME_ClassificationSystem,\n  CLASS_NAME_ClassificationTable,\n  CLASS_NAME_DrawingCategory,\n  CLASS_NAME_ElementHasClassifications,\n  CLASS_NAME_GeometricElement2d,\n  CLASS_NAME_GeometricElement3d,\n  CLASS_NAME_SpatialCategory,\n} from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { getDistinctMapValues } from \"../../common/internal/Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { CategoryId, ElementId, ModelId } from \"../../common/internal/Types.js\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { ClassificationsTreeHierarchyConfiguration } from \"../ClassificationsTreeDefinition.js\";\n\n/** @internal */\nexport type ClassificationId = Id64String;\n/** @internal */\nexport type ClassificationTableId = Id64String;\n\ninterface ClassificationInfo {\n  parentClassificationOrTableId: ClassificationId | ClassificationTableId | undefined;\n  childClassificationIds: ClassificationId[];\n  relatedCategories2d: CategoryId[];\n  relatedCategories3d: CategoryId[];\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class ClassificationsTreeIdsCache implements Disposable {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _elementModelsCategories: Promise<Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>> | undefined;\n  private _modelWithCategoryModeledElements: Promise<Map<ModelCategoryKey, Set<ElementId>>> | undefined;\n  private _classificationInfos: Promise<Map<ClassificationId | ClassificationTableId, ClassificationInfo>> | undefined;\n  private _filteredElementsData: Promise<Map<ElementId, { modelId: Id64String; categoryId: Id64String }>> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ClassificationsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(_queryExecutor, [\"BisCore.GeometricElement2d\", \"BisCore.GeometricElement3d\"]);\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *queryElementModelCategories(): AsyncIterableIterator<{\n    modelId: Id64String;\n    categoryId: Id64String;\n    type: \"2d\" | \"3d\";\n  }> {\n    const query = `\n      SELECT * FROM (\n        SELECT '3d' type, this.Model.Id modelId, this.Category.Id categoryId\n        FROM BisCore.GeometricModel m\n        JOIN BisCore.GeometricElement3d this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      )\n      UNION ALL\n      SELECT * FROM (\n        SELECT '2d' type, this.Model.Id modelId, this.Category.Id categoryId\n        FROM BisCore.GeometricModel m\n        JOIN BisCore.GeometricElement2d this ON m.ECInstanceId = this.Model.Id\n        WHERE this.Parent.Id IS NULL AND m.IsPrivate = false\n        GROUP BY modelId, categoryId\n      )\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/element-models-and-categories-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, type: row.type };\n    }\n  }\n\n  private async getElementModelsCategories() {\n    this._elementModelsCategories ??= (async () => {\n      const [modelCategories, modelWithCategoryModeledElements] = await Promise.all([\n        (async () => {\n          const elementModelsCategories = new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set }>();\n          for await (const queriedCategory of this.queryElementModelCategories()) {\n            let modelEntry = elementModelsCategories.get(queriedCategory.modelId);\n            if (modelEntry === undefined) {\n              modelEntry = { category2dIds: new Set(), category3dIds: new Set() };\n              elementModelsCategories.set(queriedCategory.modelId, modelEntry);\n            }\n            switch (queriedCategory.type) {\n              case \"2d\":\n                modelEntry.category2dIds.add(queriedCategory.categoryId);\n                break;\n              case \"3d\":\n                modelEntry.category3dIds.add(queriedCategory.categoryId);\n                break;\n            }\n          }\n          return elementModelsCategories;\n        })(),\n        this.getModelWithCategoryModeledElements(),\n      ]);\n      const result = new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>();\n      const subModels = getDistinctMapValues(modelWithCategoryModeledElements);\n      for (const [modelId, modelEntry] of modelCategories) {\n        const isSubModel = subModels.has(modelId);\n        result.set(modelId, { category2dIds: modelEntry.category2dIds, category3dIds: modelEntry.category3dIds, isSubModel });\n      }\n      return result;\n    })();\n    return this._elementModelsCategories;\n  }\n\n  private async *queryModeledElements(): AsyncIterableIterator<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    rootCategoryId: Id64String;\n  }> {\n    const query = `\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.GeometricModel m\n      JOIN BisCore.GeometricElement3d pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)\n      UNION ALL\n      SELECT\n        pe.ECInstanceId modeledElementId,\n        pe.Category.Id categoryId,\n        pe.Model.Id modelId\n      FROM BisCore.GeometricModel m\n      JOIN BisCore.GeometricElement2d pe ON pe.ECInstanceId = m.ModeledElement.Id\n      WHERE\n        m.IsPrivate = false\n        AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: query },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/modeled-elements-query\" },\n    )) {\n      yield { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n    }\n  }\n\n  private async getModelWithCategoryModeledElements() {\n    this._modelWithCategoryModeledElements ??= (async () => {\n      const modelWithCategoryModeledElements = new Map<ModelCategoryKey, Set<ElementId>>();\n      for await (const { modelId, categoryId, modeledElementId } of this.queryModeledElements()) {\n        const key: ModelCategoryKey = `${modelId}-${categoryId}`;\n        const entry = modelWithCategoryModeledElements.get(key);\n        if (entry === undefined) {\n          modelWithCategoryModeledElements.set(key, new Set([modeledElementId]));\n        } else {\n          entry.add(modeledElementId);\n        }\n      }\n      return modelWithCategoryModeledElements;\n    })();\n    return this._modelWithCategoryModeledElements;\n  }\n\n  public async getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Array): Promise<Id64Array> {\n    const modelWithCategoryModeledElements = await this.getModelWithCategoryModeledElements();\n    const result = new Array<ElementId>();\n    for (const categoryId of categoryIds) {\n      const entry = modelWithCategoryModeledElements.get(`${modelId}-${categoryId}`);\n      if (entry !== undefined) {\n        result.push(...entry);\n      }\n    }\n    return result;\n  }\n\n  public async getCategoriesElementModels(categoryIds: Id64Array, includeSubModels?: boolean): Promise<Map<CategoryId, Set<ModelId>>> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    const result = new Map<CategoryId, Set<ModelId>>();\n    for (const categoryId of categoryIds) {\n      for (const [modelId, { category2dIds, category3dIds, isSubModel }] of elementModelsCategories) {\n        if ((includeSubModels || !isSubModel) && (category2dIds.has(categoryId) || category3dIds.has(categoryId))) {\n          let categoryModels = result.get(categoryId);\n          if (!categoryModels) {\n            categoryModels = new Set<ModelId>();\n            result.set(categoryId, categoryModels);\n          }\n          categoryModels.add(modelId);\n        }\n      }\n    }\n    return result;\n  }\n\n  public async getModelCategoryIds(modelId: Id64String): Promise<{ drawing: Id64Array; spatial: Id64Array }> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return {\n      drawing: Array.from(elementModelsCategories.get(modelId)?.category2dIds ?? []),\n      spatial: Array.from(elementModelsCategories.get(modelId)?.category3dIds ?? []),\n    };\n  }\n\n  public async hasSubModel(elementId: Id64String): Promise<boolean> {\n    const elementModelsCategories = await this.getElementModelsCategories();\n    return elementModelsCategories.has(elementId);\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  private async *queryClassifications(): AsyncIterableIterator<\n    {\n      id: Id64String;\n      relatedCategories2d: CategoryId[];\n      relatedCategories3d: CategoryId[];\n    } & ({ tableId: ClassificationTableId; parentId: undefined } | { tableId: undefined; parentId: ClassificationId })\n  > {\n    const CLASSIFICATIONS_CTE = \"Classifications\";\n    const ctes = [\n      `\n        ${CLASSIFICATIONS_CTE}(ClassificationId, ClassificationTableId, ParentClassificationId) AS (\n          SELECT\n            cl.ECInstanceId,\n            ct.ECInstanceId,\n            NULL\n          FROM ${CLASS_NAME_Classification} cl\n          JOIN ${CLASS_NAME_ClassificationTable} ct ON ct.ECInstanceId = cl.Model.Id\n          JOIN ${CLASS_NAME_ClassificationSystem} cs ON cs.ECInstanceId = ct.Parent.Id\n          WHERE\n            cs.CodeValue = '${this._hierarchyConfig.rootClassificationSystemCode}'\n            AND NOT ct.IsPrivate\n            AND NOT cl.IsPrivate\n            AND cl.Parent.Id IS NULL\n\n          UNION ALL\n\n          SELECT\n            cl.ECInstanceId,\n            NULL,\n            cl.Parent.Id\n          FROM\n            ${CLASSIFICATIONS_CTE} cte\n            JOIN ${CLASS_NAME_Classification} cl ON cl.Parent.Id = cte.ClassificationId\n          WHERE\n            NOT cl.IsPrivate\n        )\n      `,\n    ];\n    const ecsql = `\n      SELECT\n        cl.ClassificationId id,\n        cl.ClassificationTableId tableId,\n        cl.ParentClassificationId parentId,\n        (\n          SELECT group_concat(IdToHex(cat.ECInstanceId))\n          FROM ${CLASS_NAME_GeometricElement2d} e\n          JOIN ${CLASS_NAME_DrawingCategory} cat ON cat.ECInstanceId = e.Category.Id\n          JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId\n          WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId\n          GROUP BY ehc.TargetECInstanceId\n        ) relatedCategories2d,\n        (\n          SELECT group_concat(IdToHex(cat.ECInstanceId))\n          FROM ${CLASS_NAME_GeometricElement3d} e\n          JOIN ${CLASS_NAME_SpatialCategory} cat ON cat.ECInstanceId = e.Category.Id\n          JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId\n          WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId\n          GROUP BY ehc.TargetECInstanceId\n        ) relatedCategories3d\n      FROM ${CLASSIFICATIONS_CTE} cl\n    `;\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ctes, ecsql },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: \"tree-widget/classifications-tree/classifications-query\" },\n    )) {\n      yield {\n        id: row.id,\n        tableId: row.tableId,\n        parentId: row.parentId,\n        relatedCategories2d: row.relatedCategories2d ? (row.relatedCategories2d as string).split(\",\") : [],\n        relatedCategories3d: row.relatedCategories3d ? (row.relatedCategories3d as string).split(\",\") : [],\n      };\n    }\n  }\n\n  private async getClassificationsInfo() {\n    this._classificationInfos ??= (async () => {\n      const classificationInfos = new Map<ClassificationId | ClassificationTableId, ClassificationInfo>();\n      for await (const { id, tableId, parentId, relatedCategories2d, relatedCategories3d } of this.queryClassifications()) {\n        const tableOrParentId = tableId ?? parentId;\n        let parentInfo = classificationInfos.get(tableOrParentId);\n        if (!parentInfo) {\n          parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [], parentClassificationOrTableId: undefined };\n          classificationInfos.set(tableOrParentId, parentInfo);\n        }\n        parentInfo.childClassificationIds.push(id);\n        let classificationEntry = classificationInfos.get(id);\n        if (!classificationEntry) {\n          classificationEntry = { childClassificationIds: [], relatedCategories2d, relatedCategories3d, parentClassificationOrTableId: tableOrParentId };\n          classificationInfos.set(id, classificationEntry);\n        } else {\n          classificationEntry.parentClassificationOrTableId = tableOrParentId;\n        }\n      }\n      return classificationInfos;\n    })();\n    return this._classificationInfos;\n  }\n\n  public async getAllContainedCategories(classificationOrTableIds: Id64Array): Promise<{ drawing: Id64Array; spatial: Id64Array }> {\n    const result = { drawing: new Array<CategoryId>(), spatial: new Array<CategoryId>() };\n    if (classificationOrTableIds.length === 0) {\n      return result;\n    }\n    const classificationsInfo = await this.getClassificationsInfo();\n    await Promise.all(\n      classificationOrTableIds.map(async (classificationOrTableId) => {\n        const classificationInfo = classificationsInfo.get(classificationOrTableId);\n        if (classificationInfo === undefined) {\n          return;\n        }\n        const childClassificationsResult = await this.getAllContainedCategories(classificationInfo.childClassificationIds);\n        result.drawing.push(...classificationInfo.relatedCategories2d, ...childClassificationsResult.drawing);\n        result.spatial.push(...classificationInfo.relatedCategories3d, ...childClassificationsResult.spatial);\n      }),\n    );\n    return result;\n  }\n\n  public async getDirectChildClassifications(classificationOrTableIds: Id64Array): Promise<ClassificationId[]> {\n    const result = new Array<ClassificationId>();\n    if (classificationOrTableIds.length === 0) {\n      return result;\n    }\n    const classificationsInfo = await this.getClassificationsInfo();\n    classificationOrTableIds.forEach((classificationOrTableId) => {\n      const classificationInfo = classificationsInfo.get(classificationOrTableId);\n      if (classificationInfo === undefined) {\n        return;\n      }\n      result.push(...classificationInfo.childClassificationIds);\n    });\n    return result;\n  }\n\n  public getClassificationsPathObs(classificationIds: Id64Array): Observable<HierarchyNodeIdentifiersPath> {\n    return from(this.getClassificationsInfo()).pipe(\n      mergeMap((classificationsInfo) => {\n        return from(classificationIds).pipe(\n          map((classificationId) => {\n            const path: HierarchyNodeIdentifiersPath = [{ id: classificationId, className: CLASS_NAME_Classification }];\n            let parentId = classificationsInfo.get(classificationId)?.parentClassificationOrTableId;\n            while (parentId !== undefined) {\n              const parentIdOfParent = classificationsInfo.get(parentId)?.parentClassificationOrTableId;\n              if (parentIdOfParent) {\n                path.push({ className: CLASS_NAME_Classification, id: parentId });\n              } else {\n                path.push({ className: CLASS_NAME_ClassificationTable, id: parentId });\n              }\n              parentId = parentIdOfParent;\n            }\n            return path.reverse();\n          }),\n        );\n      }),\n    );\n  }\n\n  public async getAllClassifications(): Promise<ClassificationId[]> {\n    const classificationsInfo = await this.getClassificationsInfo();\n    return [...classificationsInfo.keys()];\n  }\n\n  private async *queryFilteredElementsData({ element2dIds, element3dIds }: { element2dIds: Id64Array; element3dIds: Id64Array }): AsyncIterableIterator<{\n    modelId: Id64String;\n    id: ElementId;\n    categoryId: Id64String;\n  }> {\n    const queries = new Array<string>();\n    if (element2dIds.length > 0) {\n      queries.push(`\n        SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id\n        FROM ${CLASS_NAME_GeometricElement2d}\n        WHERE ECInstanceId IN (${element2dIds.join(\", \")})\n      `);\n    }\n    if (element3dIds.length > 0) {\n      queries.push(`\n        SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id\n        FROM ${CLASS_NAME_GeometricElement3d}\n        WHERE ECInstanceId IN (${element3dIds.join(\", \")})\n      `);\n    }\n    for await (const row of this._queryExecutor.createQueryReader(\n      { ecsql: queries.join(\" UNION ALL \") },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n    )) {\n      yield { modelId: row.modelId, id: row.id, categoryId: row.categoryId };\n    }\n  }\n\n  public async getFilteredElementsData({\n    element2dIds,\n    element3dIds,\n  }: {\n    element2dIds: Id64Array;\n    element3dIds: Id64Array;\n  }): Promise<Map<ElementId, { categoryId: Id64String; modelId: Id64String }>> {\n    if (element2dIds.length === 0 && element3dIds.length === 0) {\n      return new Map();\n    }\n\n    this._filteredElementsData ??= (async () => {\n      const filteredElementsData = new Map();\n      for await (const { modelId, id, categoryId } of this.queryFilteredElementsData({ element2dIds, element3dIds })) {\n        filteredElementsData.set(id, { modelId, categoryId });\n      }\n      return filteredElementsData;\n    })();\n    return this._filteredElementsData;\n  }\n\n  public clearFilteredElementsData() {\n    this._filteredElementsData = undefined;\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNode.d.ts

@@ -1,4 +1,4 @@
-import type { Id64String } from "@itwin/core-bentley";
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
 import type { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
 interface ClassificationsTreeNode {
     key: HierarchyNodeKey;
@@ -11,7 +11,7 @@ export declare namespace ClassificationsTreeNode {
     const isClassificationTableNode: (node: Pick<ClassificationsTreeNode, "extendedData">) => boolean;
     const isClassificationNode: (node: Pick<ClassificationsTreeNode, "extendedData">) => boolean;
     const isGeometricElementNode: (node: Pick<ClassificationsTreeNode, "extendedData">) => boolean;
-    const getClassificationIds: (node: Pick<ClassificationsTreeNode, "extendedData">) => Id64String[];
+    const getClassificationIds: (node: Pick<ClassificationsTreeNode, "extendedData">) => Id64Array;
     const getModelId: (node: Pick<ClassificationsTreeNode, "extendedData">) => Id64String;
     const getCategoryId: (node: Pick<ClassificationsTreeNode, "extendedData">) => Id64String;
 }

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNode.js.map

@@ -1 +1 @@
-{"version":3,"file":"ClassificationsTreeNode.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNode.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAGhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AASxE,gBAAgB;AAChB,MAAM,KAAW,uBAAuB,CA0BvC;AA1BD,WAAiB,uBAAuB;IACzB,iDAAyB,GAAG,CAAC,IAAmD,EAAE,EAAE,CAC/F,WAAW,CAAC,IAAI,CAAC,KAAK,qBAAqB,CAAC;IAEjC,4CAAoB,GAAG,CAAC,IAAmD,EAAE,EAAE,CAC1F,WAAW,CAAC,IAAI,CAAC,KAAK,gBAAgB,CAAC;IAE5B,8CAAsB,GAAG,CAAC,IAAmD,EAAE,EAAE;QAC5F,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;QAC/B,OAAO,IAAI,KAAK,oBAAoB,IAAI,IAAI,KAAK,oBAAoB,CAAC;IACxE,CAAC,CAAA;IAEY,4CAAoB,GAAG,CAAC,IAAmD,EAAgB,EAAE;QACxG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,IAAI,mBAAmB,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;QACxE,OAAO,sBAAsB,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;IACrE,CAAC,CAAA;IAEY,kCAAU,GAAG,CAAC,IAAmD,EAAc,EAAE;QAC5F,MAAM,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IACnC,CAAC,CAAA;IAEY,qCAAa,GAAG,CAAC,IAAmD,EAAc,EAAE;QAC/F,MAAM,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC;IACtC,CAAC,CAAA;AACH,CAAC,EA1BgB,uBAAuB,KAAvB,uBAAuB,QA0BvC;AAED,SAAS,WAAW,CAAC,IAAmD;IACtE,OAAO,IAAI,CAAC,YAAY,IAAI,MAAM,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/F,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { parseIdsSelectorResult } from \"../../common/internal/Utils.js\";\n\nimport type { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\n\ninterface ClassificationsTreeNode {\n  key: HierarchyNodeKey;\n  extendedData?: { [id: string]: any };\n}\n\n/** @internal */\nexport namespace ClassificationsTreeNode {\n  export const isClassificationTableNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) =>\n    getNodeType(node) === \"ClassificationTable\";\n\n  export const isClassificationNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) =>\n    getNodeType(node) === \"Classification\";\n\n  export const isGeometricElementNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) => {\n    const type = getNodeType(node);\n    return type === \"GeometricElement3d\" || type === \"GeometricElement2d\";\n  }\n\n  export const getClassificationIds = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64String[] => {\n    assert(!!node.extendedData && \"classificationIds\" in node.extendedData);\n    return parseIdsSelectorResult(node.extendedData.classificationIds);\n  }\n\n  export const getModelId = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64String => {\n    assert(typeof node.extendedData?.modelId === \"string\");\n    return node.extendedData.modelId;\n  }\n\n  export const getCategoryId = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64String => {\n    assert(typeof node.extendedData?.categoryId === \"string\");\n    return node.extendedData.categoryId;\n  }\n}\n\nfunction getNodeType(node: Pick<ClassificationsTreeNode, \"extendedData\">) {\n  return node.extendedData && \"type\" in node.extendedData ? node.extendedData.type : undefined;\n}\n"]}
+{"version":3,"file":"ClassificationsTreeNode.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNode.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAGhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AASxE,gBAAgB;AAChB,MAAM,KAAW,uBAAuB,CAwBvC;AAxBD,WAAiB,uBAAuB;IACzB,iDAAyB,GAAG,CAAC,IAAmD,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,qBAAqB,CAAC;IAEjI,4CAAoB,GAAG,CAAC,IAAmD,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,gBAAgB,CAAC;IAEvH,8CAAsB,GAAG,CAAC,IAAmD,EAAE,EAAE;QAC5F,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;QAC/B,OAAO,IAAI,KAAK,oBAAoB,IAAI,IAAI,KAAK,oBAAoB,CAAC;IACxE,CAAC,CAAC;IAEW,4CAAoB,GAAG,CAAC,IAAmD,EAAa,EAAE;QACrG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,IAAI,mBAAmB,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;QACxE,OAAO,sBAAsB,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;IACrE,CAAC,CAAC;IAEW,kCAAU,GAAG,CAAC,IAAmD,EAAc,EAAE;QAC5F,MAAM,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IACnC,CAAC,CAAC;IAEW,qCAAa,GAAG,CAAC,IAAmD,EAAc,EAAE;QAC/F,MAAM,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC;IACtC,CAAC,CAAC;AACJ,CAAC,EAxBgB,uBAAuB,KAAvB,uBAAuB,QAwBvC;AAED,SAAS,WAAW,CAAC,IAAmD;IACtE,OAAO,IAAI,CAAC,YAAY,IAAI,MAAM,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/F,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { parseIdsSelectorResult } from \"../../common/internal/Utils.js\";\n\nimport type { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\n\ninterface ClassificationsTreeNode {\n  key: HierarchyNodeKey;\n  extendedData?: { [id: string]: any };\n}\n\n/** @internal */\nexport namespace ClassificationsTreeNode {\n  export const isClassificationTableNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) => getNodeType(node) === \"ClassificationTable\";\n\n  export const isClassificationNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) => getNodeType(node) === \"Classification\";\n\n  export const isGeometricElementNode = (node: Pick<ClassificationsTreeNode, \"extendedData\">) => {\n    const type = getNodeType(node);\n    return type === \"GeometricElement3d\" || type === \"GeometricElement2d\";\n  };\n\n  export const getClassificationIds = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64Array => {\n    assert(!!node.extendedData && \"classificationIds\" in node.extendedData);\n    return parseIdsSelectorResult(node.extendedData.classificationIds);\n  };\n\n  export const getModelId = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64String => {\n    assert(typeof node.extendedData?.modelId === \"string\");\n    return node.extendedData.modelId;\n  };\n\n  export const getCategoryId = (node: Pick<ClassificationsTreeNode, \"extendedData\">): Id64String => {\n    assert(typeof node.extendedData?.categoryId === \"string\");\n    return node.extendedData.categoryId;\n  };\n}\n\nfunction getNodeType(node: Pick<ClassificationsTreeNode, \"extendedData\">) {\n  return node.extendedData && \"type\" in node.extendedData ? node.extendedData.type : undefined;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeVisibilityHandler.d.ts

@@ -1,3 +1,4 @@
+import type { HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
 import type { Viewport } from "@itwin/core-frontend";
 import type { HierarchyVisibilityHandler } from "../../common/UseHierarchyVisibility.js";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
@@ -10,6 +11,7 @@ export interface ClassificationsTreeVisibilityHandlerProps {
     viewport: Viewport;
     idsCache: ClassificationsTreeIdsCache;
     imodelAccess: ECClassHierarchyInspector;
+    filteredPaths?: HierarchyFilteringPath[];
 }
 /**
  * Creates an instance if `ClassificationsTreeVisibilityHandler`.

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeVisibilityHandler.js

@@ -9,11 +9,12 @@ import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies
 import { AlwaysAndNeverDrawnElementInfo } from "../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
 import { toVoidPromise } from "../../common/internal/Rxjs.js";
 import { createVisibilityStatus } from "../../common/internal/Tooltip.js";
-import { setDifference, setIntersection } from "../../common/internal/Utils.js";
+import { getDistinctMapValues, releaseMainThreadOnItemsCount, setDifference, setIntersection } from "../../common/internal/Utils.js";
 import { createVisibilityChangeEventListener } from "../../common/internal/VisibilityChangeEventListener.js";
 import { changeElementStateNoChildrenOperator, enableCategoryDisplay, getElementOverriddenVisibility, getElementVisibility, getSubModeledElementsVisibilityStatus, getVisibilityFromAlwaysAndNeverDrawnElementsImpl, mergeVisibilityStatuses, } from "../../common/internal/VisibilityUtils.js";
 import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility.js";
 import { ClassificationsTreeNode } from "./ClassificationsTreeNode.js";
+import { createFilteredTree } from "./FilteredTree.js";
 /**
  * Creates an instance if `ClassificationsTreeVisibilityHandler`.
  * @internal
@@ -26,6 +27,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
     _eventListener;
     _alwaysAndNeverDrawnElements;
     _idsCache;
+    _filteredTree;
     _elementChangeQueue = new Subject();
     _subscriptions = [];
     _changeRequest = new Subject();
@@ -41,6 +43,13 @@ class ClassificationsTreeVisibilityHandlerImpl {
         });
         this._alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo(_props.viewport);
         this._idsCache = this._props.idsCache;
+        if (_props.filteredPaths) {
+            this._filteredTree = createFilteredTree({
+                idsCache: this._idsCache,
+                filteringPaths: _props.filteredPaths,
+                imodelAccess: this._props.imodelAccess,
+            });
+        }
         this._subscriptions.push(this._elementChangeQueue.pipe(concatAll()).subscribe());
     }
     [Symbol.dispose]() {
@@ -80,6 +89,9 @@ class ClassificationsTreeVisibilityHandlerImpl {
         return toVoidPromise(changeObservable);
     }
     getVisibilityStatusObs(node) {
+        if (node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.getFilteredNodeVisibility({ node });
+        }
         if (!HierarchyNode.isInstancesNode(node)) {
             return EMPTY;
         }
@@ -103,6 +115,32 @@ class ClassificationsTreeVisibilityHandlerImpl {
             elementType: node.extendedData?.type,
         });
     }
+    getFilteredNodeVisibility(props) {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ classificationIds, classificationTableIds, elements2d, elements3d }) => {
+            const observables = new Array();
+            if (classificationTableIds?.size) {
+                observables.push(this.getClassificationTableDisplayStatus({ classificationTableIds: [...classificationTableIds] }));
+            }
+            if (classificationIds?.size) {
+                observables.push(this.getClassificationDisplayStatus({ classificationIds: [...classificationIds] }));
+            }
+            if (elements2d?.length) {
+                observables.push(from(elements2d).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, categoryId, elementIds }) => {
+                    return from(elementIds).pipe(releaseMainThreadOnItemsCount(1000), mergeMap((elementId) => this.getGeometricElementDisplayStatus({ modelId, categoryId, elementId, elementType: "2d" })));
+                })));
+            }
+            if (elements3d?.length) {
+                observables.push(from(elements3d).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, categoryId, elementIds }) => {
+                    return from(elementIds).pipe(releaseMainThreadOnItemsCount(1000), mergeMap((elementId) => this.getGeometricElementDisplayStatus({ modelId, categoryId, elementId, elementType: "3d" })));
+                })));
+            }
+            return merge(...observables);
+        }), mergeVisibilityStatuses);
+    }
+    async getVisibilityChangeTargets({ node }) {
+        const filteredTree = await this._filteredTree;
+        return filteredTree ? filteredTree.getVisibilityChangeTargets(node) : {};
+    }
     getClassificationTableDisplayStatus(props) {
         const result = defer(() => {
             return from(this._idsCache.getAllContainedCategories(props.classificationTableIds)).pipe(mergeMap(({ drawing, spatial }) => merge(of(drawing).pipe(mergeMap((categoryIds) => this.getCategoryDisplayStatus({ categoryIds, type: "DrawingCategory" }))), of(spatial).pipe(mergeMap((categoryIds) => this.getCategoryDisplayStatus({ categoryIds, type: "SpatialCategory" }))))), mergeVisibilityStatuses);
@@ -163,7 +201,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
     getDefaultCategoryVisibilityStatus({ categoryIds }) {
         const result = from(categoryIds).pipe(mergeMap(async (categoryId) => {
             let visibility = "unknown";
-            const categoryModels = [...(await this._idsCache.getCategoriesElementModels([categoryId], true)).values()].flat();
+            const categoryModels = getDistinctMapValues(await this._idsCache.getCategoriesElementModels([categoryId], true));
             let nonDefaultModelDisplayStatesCount = 0;
             for (const modelId of categoryModels) {
                 if (!this._props.viewport.view.viewsModel(modelId)) {
@@ -192,7 +230,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
                     continue;
                 }
             }
-            if (categoryModels.length > 0 && nonDefaultModelDisplayStatesCount === categoryModels.length) {
+            if (categoryModels.size > 0 && nonDefaultModelDisplayStatesCount === categoryModels.size) {
                 assert(visibility === "visible" || visibility === "hidden");
                 return createVisibilityStatus(visibility);
             }
@@ -216,7 +254,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
                 return of(createVisibilityStatus("disabled"));
             }
             const modelsObservable = props.modelId
-                ? of(new Map(props.categoryIds.map((id) => [id, [props.modelId]])))
+                ? of(new Map(props.categoryIds.map((id) => [id, new Set([props.modelId])])))
                 : from(this._idsCache.getCategoriesElementModels(props.categoryIds));
             return merge(
             // get visibility status from always and never drawn elements
@@ -308,7 +346,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
             }));
         }
         const { modelId, categoryIds } = props.queryProps;
-        const totalCount = (modelId ? of(new Map(categoryIds.map((categoryId) => [categoryId, [modelId]]))) : from(this._idsCache.getCategoriesElementModels(categoryIds))).pipe(mergeMap((categoriesMap) => from(categoriesMap)), mergeMap(([categoryId, modelIds]) => {
+        const totalCount = (modelId ? of(new Map(categoryIds.map((categoryId) => [categoryId, new Set([modelId])]))) : from(this._idsCache.getCategoriesElementModels(categoryIds))).pipe(mergeMap((categoriesMap) => from(categoriesMap)), mergeMap(([categoryId, modelIds]) => {
             return from(modelIds).pipe(mergeMap((modelOfCategory) => from(this._idsCache.getCategoryElementsCount(modelOfCategory, categoryId))), reduce((acc, specificModelCategoryCount) => acc + specificModelCategoryCount, 0));
         }));
         return forkJoin({
@@ -341,6 +379,9 @@ class ClassificationsTreeVisibilityHandlerImpl {
         return createVisibilityHandlerResult(this, props, result, undefined);
     }
     changeVisibilityObs(node, on) {
+        if (node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.changeFilteredNodeVisibility({ node, on });
+        }
         if (!HierarchyNode.isInstancesNode(node)) {
             return EMPTY;
         }
@@ -363,6 +404,28 @@ class ClassificationsTreeVisibilityHandlerImpl {
             on,
         });
     }
+    changeFilteredNodeVisibility({ on, ...props }) {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ classificationTableIds, classificationIds, elements2d, elements3d }) => {
+            const observables = new Array();
+            if (classificationTableIds?.size) {
+                observables.push(this.changeClassificationTableDisplayState({ classificationTableIds: [...classificationTableIds], on }));
+            }
+            if (classificationIds?.size) {
+                observables.push(this.changeClassificationDisplayState({ classificationIds: [...classificationIds], on }));
+            }
+            if (elements2d?.length) {
+                observables.push(from(elements2d).pipe(mergeMap(({ modelId, categoryId, elementIds }) => {
+                    return this.changeGeometricElementsDisplayState({ modelId, categoryId, elementIds, on });
+                })));
+            }
+            if (elements3d?.length) {
+                observables.push(from(elements3d).pipe(mergeMap(({ modelId, categoryId, elementIds }) => {
+                    return this.changeGeometricElementsDisplayState({ modelId, categoryId, elementIds, on });
+                })));
+            }
+            return merge(...observables);
+        }));
+    }
     changeClassificationTableDisplayState(props) {
         const result = defer(() => {
             return from(this._idsCache.getAllContainedCategories(props.classificationTableIds)).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => {
@@ -437,7 +500,7 @@ class ClassificationsTreeVisibilityHandlerImpl {
             const { modelId, categoryIds, on } = props;
             const viewport = this._props.viewport;
             const modelIdsObservable = modelId
-                ? of(new Map(categoryIds.map((id) => [id, [modelId]])))
+                ? of(new Map(categoryIds.map((id) => [id, new Set([modelId])])))
                 : from(this._idsCache.getCategoriesElementModels(categoryIds, true));
             return concat(modelId ? EMPTY : from(enableCategoryDisplay(viewport, categoryIds, on)), on
                 ? modelIdsObservable.pipe(mergeMap((categoriesMap) => from(categoriesMap.values())), mergeAll(), filter((modelIdToCheck) => !viewport.view.viewsModel(modelIdToCheck)), mergeMap((modelIdToCheck) => this.showModelWithoutAnyCategoriesOrElements(modelIdToCheck)))