npm - @itwin/tree-widget-react - Versions diffs - 4.0.0-alpha.25 → 4.0.0-alpha.26 - Mend

@itwin/tree-widget-react 4.0.0-alpha.25 → 4.0.0-alpha.26

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (192) hide show

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

@@ -2,210 +2,26 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { defer, EMPTY, forkJoin, from, map, mergeMap, of, reduce, shareReplay, toArray } from "rxjs";
+import { defer, from, map, mergeMap, of, reduce, shareReplay } from "rxjs";
 import { Guid, Id64 } from "@itwin/core-bentley";
-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, CLASS_NAME_SubCategory, } from "../../common/internal/ClassNameDefinitions.js";
-import { ModelCategoryElementsCountCache } from "../../common/internal/ModelCategoryElementsCountCache.js";
+import { BaseIdsCacheImpl } from "../../common/internal/caches/BaseIdsCache.js";
+import { CLASS_NAME_Classification, CLASS_NAME_ClassificationSystem, CLASS_NAME_ClassificationTable, CLASS_NAME_ElementHasClassifications, CLASS_NAME_GeometricElement3d, CLASS_NAME_SpatialCategory, } from "../../common/internal/ClassNameDefinitions.js";
 import { catchBeSQLiteInterrupts } from "../../common/internal/UseErrorState.js";
 import { joinId64Arg } from "../../common/internal/Utils.js";
 /** @internal */
-export class ClassificationsTreeIdsCache {
-    #categoryElementCounts;
-    #categorySubCategories;
-    #elementModelsCategories;
-    #modelWithCategoryModeledElements;
+export class ClassificationsTreeIdsCache extends BaseIdsCacheImpl {
     #classificationInfos;
     #filteredElementsData;
     #queryExecutor;
     #hierarchyConfig;
     #componentId;
     #componentName;
-    constructor(queryExecutor, hierarchyConfig, componentId) {
-        this.#queryExecutor = queryExecutor;
-        this.#hierarchyConfig = hierarchyConfig;
-        this.#componentId = componentId ?? Guid.createValue();
+    constructor(props) {
+        super(props);
+        this.#queryExecutor = props.queryExecutor;
+        this.#hierarchyConfig = props.hierarchyConfig;
+        this.#componentId = Guid.createValue();
         this.#componentName = "ClassificationsTreeIdsCache";
-        this.#categoryElementCounts = new ModelCategoryElementsCountCache(this.#queryExecutor, ["BisCore.GeometricElement2d", "BisCore.GeometricElement3d"], this.#componentId);
-    }
-    [Symbol.dispose]() {
-        this.#categoryElementCounts[Symbol.dispose]();
-    }
-    querySubCategories() {
-        return defer(() => {
-            const definitionsQuery = `
-        SELECT
-          sc.ECInstanceId id,
-          sc.Parent.Id categoryId
-        FROM
-          ${CLASS_NAME_SubCategory} sc
-        WHERE
-          NOT sc.IsPrivate
-      `;
-            return this.#queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/visible-sub-categories` });
-        }).pipe(catchBeSQLiteInterrupts, map((row) => {
-            return { id: row.id, parentId: row.categoryId };
-        }));
-    }
-    getSubCategoriesInfo() {
-        this.#categorySubCategories ??= this.querySubCategories()
-            .pipe(reduce((acc, queriedSubCategory) => {
-            const entry = acc.get(queriedSubCategory.parentId);
-            if (entry) {
-                entry.push(queriedSubCategory.id);
-            }
-            else {
-                acc.set(queriedSubCategory.parentId, [queriedSubCategory.id]);
-            }
-            return acc;
-        }, new Map()))
-            .pipe(shareReplay());
-        return this.#categorySubCategories;
-    }
-    getSubCategories(categoryId) {
-        return this.getSubCategoriesInfo().pipe(map((categorySubCategories) => categorySubCategories.get(categoryId) ?? []));
-    }
-    queryElementModelCategories() {
-        return defer(() => {
-            const query = `
-        SELECT * FROM (
-          SELECT '3d' type, this.Model.Id modelId, this.Category.Id categoryId
-          FROM BisCore.GeometricModel m
-          JOIN BisCore.GeometricElement3d this ON m.ECInstanceId = this.Model.Id
-          WHERE this.Parent.Id IS NULL AND m.IsPrivate = false
-          GROUP BY modelId, categoryId
-        )
-        UNION ALL
-        SELECT * FROM (
-          SELECT '2d' type, this.Model.Id modelId, this.Category.Id categoryId
-          FROM BisCore.GeometricModel m
-          JOIN BisCore.GeometricElement2d this ON m.ECInstanceId = this.Model.Id
-          WHERE this.Parent.Id IS NULL AND m.IsPrivate = false
-          GROUP BY modelId, categoryId
-        )
-      `;
-            return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` });
-        }).pipe(catchBeSQLiteInterrupts, map((row) => {
-            return { modelId: row.modelId, categoryId: row.categoryId, type: row.type };
-        }));
-    }
-    getElementModelsCategories() {
-        this.#elementModelsCategories ??= forkJoin({
-            modelCategories: this.queryElementModelCategories().pipe(reduce((acc, queriedCategory) => {
-                let modelEntry = acc.get(queriedCategory.modelId);
-                if (modelEntry === undefined) {
-                    modelEntry = { category2dIds: new Set(), category3dIds: new Set() };
-                    acc.set(queriedCategory.modelId, modelEntry);
-                }
-                switch (queriedCategory.type) {
-                    case "2d":
-                        modelEntry.category2dIds.add(queriedCategory.categoryId);
-                        break;
-                    case "3d":
-                        modelEntry.category3dIds.add(queriedCategory.categoryId);
-                        break;
-                }
-                return acc;
-            }, new Map())),
-            modelWithCategoryModeledElements: this.getModelWithCategoryModeledElements(),
-        }).pipe(map(({ modelCategories, modelWithCategoryModeledElements }) => {
-            const result = new Map();
-            const subModels = new Set();
-            modelWithCategoryModeledElements.forEach((categoryMap) => categoryMap.forEach((categorySubModels) => categorySubModels.forEach((subModelId) => subModels.add(subModelId))));
-            for (const [modelId, modelEntry] of modelCategories) {
-                const isSubModel = subModels.has(modelId);
-                result.set(modelId, { category2dIds: modelEntry.category2dIds, category3dIds: modelEntry.category3dIds, isSubModel });
-            }
-            return result;
-        }), shareReplay());
-        return this.#elementModelsCategories;
-    }
-    queryModeledElements() {
-        return defer(() => {
-            const query = `
-        SELECT
-          pe.ECInstanceId modeledElementId,
-          pe.Category.Id categoryId,
-          pe.Model.Id modelId
-        FROM BisCore.GeometricModel m
-        JOIN BisCore.GeometricElement3d pe ON pe.ECInstanceId = m.ModeledElement.Id
-        WHERE
-          m.IsPrivate = false
-          AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)
-        UNION ALL
-        SELECT
-          pe.ECInstanceId modeledElementId,
-          pe.Category.Id categoryId,
-          pe.Model.Id modelId
-        FROM BisCore.GeometricModel m
-        JOIN BisCore.GeometricElement2d pe ON pe.ECInstanceId = m.ModeledElement.Id
-        WHERE
-          m.IsPrivate = false
-          AND m.ECInstanceId IN (SELECT Model.Id FROM BisCore.Element)
-      `;
-            return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` });
-        }).pipe(catchBeSQLiteInterrupts, map((row) => {
-            return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };
-        }));
-    }
-    getModelWithCategoryModeledElements() {
-        this.#modelWithCategoryModeledElements ??= this.queryModeledElements().pipe(reduce((acc, { modelId, categoryId, modeledElementId }) => {
-            let modelEntry = acc.get(modelId);
-            if (!modelEntry) {
-                modelEntry = new Map();
-                acc.set(modelId, modelEntry);
-            }
-            const categoryEntry = modelEntry.get(categoryId);
-            if (!categoryEntry) {
-                modelEntry.set(categoryId, new Set([modeledElementId]));
-            }
-            else {
-                categoryEntry.add(modeledElementId);
-            }
-            return acc;
-        }, new Map()), shareReplay());
-        return this.#modelWithCategoryModeledElements;
-    }
-    getCategoriesModeledElements(modelId, categoryIds) {
-        return this.getModelWithCategoryModeledElements().pipe(mergeMap((modelWithCategoryModeledElements) => {
-            const result = new Array();
-            const categoryMap = modelWithCategoryModeledElements.get(modelId);
-            if (!categoryMap) {
-                return of(result);
-            }
-            return from(Id64.iterable(categoryIds)).pipe(map((categoryId) => categoryMap.get(categoryId)), mergeMap((elementsSet) => (elementsSet ? from(elementsSet) : EMPTY)), toArray());
-        }));
-    }
-    getCategoriesElementModels(categoryIds, includeSubModels) {
-        return this.getElementModelsCategories().pipe(mergeMap((elementModelsCategories) => from(Id64.iterable(categoryIds)).pipe(map((categoryId) => {
-            const categoryModels = new Array();
-            elementModelsCategories.forEach(({ category2dIds, category3dIds, isSubModel }, modelId) => {
-                if ((includeSubModels || !isSubModel) && (category2dIds.has(categoryId) || category3dIds.has(categoryId))) {
-                    categoryModels.push(modelId);
-                }
-            });
-            return { id: categoryId, models: categoryModels.length > 0 ? categoryModels : undefined };
-        }))));
-    }
-    getModelCategoryIds(modelId) {
-        return this.getElementModelsCategories().pipe(map((elementModelsCategories) => {
-            return {
-                drawing: Array.from(elementModelsCategories.get(modelId)?.category2dIds ?? []),
-                spatial: Array.from(elementModelsCategories.get(modelId)?.category3dIds ?? []),
-            };
-        }));
-    }
-    getAllCategories() {
-        return this.getElementModelsCategories().pipe(mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()), reduce((acc, { category2dIds, category3dIds }) => {
-            category2dIds.forEach((id) => acc.drawing.add(id));
-            category3dIds.forEach((id) => acc.spatial.add(id));
-            return acc;
-        }, { drawing: new Set(), spatial: new Set() }));
-    }
-    hasSubModel(elementId) {
-        return this.getElementModelsCategories().pipe(map((elementModelsCategories) => elementModelsCategories.has(elementId)));
-    }
-    getCategoryElementsCount(modelId, categoryId) {
-        return this.#categoryElementCounts.getCategoryElementsCount(modelId, categoryId);
     }
     queryClassifications() {
         return defer(() => {
@@ -245,14 +61,6 @@ export class ClassificationsTreeIdsCache {
           cl.ClassificationId id,
           cl.ClassificationTableId tableId,
           cl.ParentClassificationId parentId,
-          (
-            SELECT group_concat(IdToHex(cat.ECInstanceId))
-            FROM ${CLASS_NAME_GeometricElement2d} e
-            JOIN ${CLASS_NAME_DrawingCategory} cat ON cat.ECInstanceId = e.Category.Id
-            JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId
-            WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId
-            GROUP BY ehc.TargetECInstanceId
-          ) relatedCategories2d,
           (
             SELECT group_concat(IdToHex(cat.ECInstanceId))
             FROM ${CLASS_NAME_GeometricElement3d} e
@@ -260,7 +68,7 @@ export class ClassificationsTreeIdsCache {
             JOIN ${CLASS_NAME_ElementHasClassifications} ehc ON ehc.SourceECInstanceId = e.ECInstanceId
             WHERE e.Parent.Id IS NULL AND NOT cat.IsPrivate AND ehc.TargetECInstanceId = cl.ClassificationId
             GROUP BY ehc.TargetECInstanceId
-          ) relatedCategories3d
+          ) relatedCategories
         FROM ${CLASSIFICATIONS_CTE} cl
       `;
             return this.#queryExecutor.createQueryReader({ ctes, ecsql }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/classifications` });
@@ -269,23 +77,22 @@ export class ClassificationsTreeIdsCache {
                 id: row.id,
                 tableId: row.tableId,
                 parentId: row.parentId,
-                relatedCategories2d: row.relatedCategories2d ? row.relatedCategories2d.split(",") : [],
-                relatedCategories3d: row.relatedCategories3d ? row.relatedCategories3d.split(",") : [],
+                relatedCategories: row.relatedCategories ? row.relatedCategories.split(",") : [],
             };
         }));
     }
     getClassificationsInfo() {
-        this.#classificationInfos ??= this.queryClassifications().pipe(reduce((acc, { id, tableId, parentId, relatedCategories2d, relatedCategories3d }) => {
+        this.#classificationInfos ??= this.queryClassifications().pipe(reduce((acc, { id, tableId, parentId, relatedCategories }) => {
             const tableOrParentId = tableId ?? parentId;
             let parentInfo = acc.get(tableOrParentId);
             if (!parentInfo) {
-                parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [], parentClassificationOrTableId: undefined };
+                parentInfo = { childClassificationIds: [], relatedCategories: [], parentClassificationOrTableId: undefined };
                 acc.set(tableOrParentId, parentInfo);
             }
             parentInfo.childClassificationIds.push(id);
             let classificationEntry = acc.get(id);
             if (!classificationEntry) {
-                classificationEntry = { childClassificationIds: [], relatedCategories2d, relatedCategories3d, parentClassificationOrTableId: tableOrParentId };
+                classificationEntry = { childClassificationIds: [], relatedCategories, parentClassificationOrTableId: tableOrParentId };
                 acc.set(id, classificationEntry);
             }
             else {
@@ -296,7 +103,7 @@ export class ClassificationsTreeIdsCache {
         return this.#classificationInfos;
     }
     getAllContainedCategories(classificationOrTableIds) {
-        const result = { drawing: new Array(), spatial: new Array() };
+        const result = new Array();
         if (Id64.sizeOf(classificationOrTableIds) === 0) {
             return of(result);
         }
@@ -305,18 +112,16 @@ export class ClassificationsTreeIdsCache {
             if (classificationInfo === undefined) {
                 return acc;
             }
-            acc.drawing.push(...classificationInfo.relatedCategories2d);
-            acc.spatial.push(...classificationInfo.relatedCategories3d);
-            acc.childClassifications.push(...classificationInfo.childClassificationIds);
+            classificationInfo.relatedCategories.forEach((id) => acc.categories.push(id));
+            classificationInfo.childClassificationIds.forEach((id) => acc.childClassifications.push(id));
             return acc;
-        }, { drawing: new Array(), spatial: new Array(), childClassifications: new Array() }), mergeMap(({ drawing, spatial, childClassifications }) => {
+        }, { categories: new Array(), childClassifications: new Array() }), mergeMap(({ categories, childClassifications }) => {
             if (childClassifications.length === 0) {
-                return of({ drawing, spatial });
+                return of(categories);
             }
-            return this.getAllContainedCategories(childClassifications).pipe(map((childResult) => {
-                drawing.push(...childResult.drawing);
-                spatial.push(...childResult.spatial);
-                return { drawing, spatial };
+            return this.getAllContainedCategories(childClassifications).pipe(map((childCategories) => {
+                childCategories.forEach((id) => categories.push(id));
+                return categories;
             }));
         }))));
     }
@@ -328,7 +133,7 @@ export class ClassificationsTreeIdsCache {
         return this.getClassificationsInfo().pipe(mergeMap((classificationsInfo) => from(Id64.iterable(classificationOrTableIds)).pipe(reduce((acc, classificationOrTableId) => {
             const classificationInfo = classificationsInfo.get(classificationOrTableId);
             if (classificationInfo !== undefined) {
-                acc.push(...classificationInfo.childClassificationIds);
+                classificationInfo.childClassificationIds.forEach((id) => acc.push(id));
             }
             return acc;
         }, result))));
@@ -355,42 +160,54 @@ export class ClassificationsTreeIdsCache {
     getAllClassifications() {
         return this.getClassificationsInfo().pipe(map((classificationsInfo) => [...classificationsInfo.keys()]));
     }
-    queryFilteredElementsData({ element2dIds, element3dIds }) {
+    queryFilteredElementsData({ elementIds }) {
         return defer(() => {
-            const queries = new Array();
-            if (Id64.sizeOf(element2dIds) > 0) {
-                queries.push(`
-          SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id
-          FROM ${CLASS_NAME_GeometricElement2d}
-          WHERE ECInstanceId IN (${joinId64Arg(element2dIds, ",")})
-        `);
-            }
-            if (Id64.sizeOf(element3dIds) > 0) {
-                queries.push(`
-          SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id
-          FROM ${CLASS_NAME_GeometricElement3d}
-          WHERE ECInstanceId IN (${joinId64Arg(element3dIds, ",")})
-        `);
-            }
-            return this.#queryExecutor.createQueryReader({ ecsql: queries.join(" UNION ALL ") }, {
+            const query = `
+        SELECT
+          this.Model.Id modelId,
+          this.Category.Id categoryId,
+          this.ECInstanceId id,
+          (
+            WITH RECURSIVE
+              ParentWithCategory(id, categoryId, parentId) AS (
+                SELECT e.ECInstanceId, e.Category.Id, e.Parent.Id
+                FROM ${CLASS_NAME_GeometricElement3d} e
+                WHERE e.ECInstanceId = this.ECInstanceId
+                UNION ALL
+                SELECT p.ECInstanceId, p.Category.Id, p.Parent.Id
+                FROM ${CLASS_NAME_GeometricElement3d} p
+                JOIN ParentWithCategory c ON p.ECInstanceId = c.parentId
+              )
+            SELECT IdToHex(categoryId)
+            FROM ParentWithCategory
+            WHERE parentId IS NULL
+          ) categoryOfTopMostParentElement
+        FROM ${CLASS_NAME_GeometricElement3d} this
+        WHERE ECInstanceId IN (${joinId64Arg(elementIds, ",")})
+      `;
+            return this.#queryExecutor.createQueryReader({ ecsql: query }, {
                 rowFormat: "ECSqlPropertyNames",
                 limit: "unbounded",
                 restartToken: `${this.#componentName}/${this.#componentId}/filtered-elements/${Guid.createValue()}`,
             });
         }).pipe(catchBeSQLiteInterrupts, map((row) => {
-            return { modelId: row.modelId, id: row.id, categoryId: row.categoryId };
+            return {
+                modelId: row.modelId,
+                id: row.id,
+                categoryId: row.categoryId,
+                categoryOfTopMostParentElement: row.categoryOfTopMostParentElement,
+            };
         }));
     }
-    getFilteredElementsData({ element2dIds, element3dIds, }) {
+    getFilteredElementsData({ elementIds, }) {
         const result = new Map();
-        if (Id64.sizeOf(element2dIds) === 0 && Id64.sizeOf(element3dIds) === 0) {
+        if (Id64.sizeOf(elementIds) === 0) {
             return of(result);
         }
         this.#filteredElementsData ??= this.queryFilteredElementsData({
-            element2dIds,
-            element3dIds,
-        }).pipe(reduce((acc, { modelId, id, categoryId }) => {
-            acc.set(id, { modelId, categoryId });
+            elementIds,
+        }).pipe(reduce((acc, { modelId, id, categoryId, categoryOfTopMostParentElement }) => {
+            acc.set(id, { modelId, categoryId, categoryOfTopMostParentElement });
             return acc;
         }, result), shareReplay());
         return this.#filteredElementsData;

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

	@@ -1 +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,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrG,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,0BAA0B,EAC1B,oCAAoC,EACpC,6BAA6B,EAC7B,6BAA6B,EAC7B,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,+BAA+B,EAAE,MAAM,0DAA0D,CAAC;AAC3G,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAoB7D,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAC7B,sBAAsB,CAAkC;IACjE,sBAAsB,CAAgE;IACtF,wBAAwB,CAAgH;IACxI,iCAAiC,CAAwE;IACzG,oBAAoB,CAA4F;IAChH,qBAAqB,CAA0F;IAC/G,cAAc,CAA6B;IAC3C,gBAAgB,CAA4C;IAC5D,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,aAAyC,EAAE,eAA0D,EAAE,WAAwB;QACzI,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtD,IAAI,CAAC,cAAc,GAAG,6BAA6B,CAAC;QACpD,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAC/D,IAAI,CAAC,cAAc,EACnB,CAAC,4BAA4B,EAAE,4BAA4B,CAAC,EAC5D,IAAI,CAAC,YAAY,CAClB,CAAC;IACJ,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,kBAAkB;QACxB,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,gBAAgB,GAAG;;;;;YAKnB,sBAAsB;;;OAG3B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAC3B,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,yBAAyB,EAAE,CAC5I,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAClD,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IACO,oBAAoB;QAC1B,IAAI,CAAC,sBAAsB,KAAK,IAAI,CAAC,kBAAkB,EAAE;aACtD,IAAI,CACH,MAAM,CAAC,CAAC,GAAG,EAAE,kBAAkB,EAAE,EAAE;YACjC,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAoC,CAAC,CAChD;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACrC,CAAC;IAEM,gBAAgB,CAAC,UAAsB;QAC5C,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,qBAAqB,EAAE,EAAE,CAAC,qBAAqB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACvH,CAAC;IAEO,2BAA2B;QAKjC,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;OAgBb,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,gCAAgC,EAAE,CACnJ,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;QAC9E,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,0BAA0B;QAChC,IAAI,CAAC,wBAAwB,KAAK,QAAQ,CAAC;YACzC,eAAe,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CACtD,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;gBAC9B,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,UAAU,GAAG,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;oBACpE,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC/C,CAAC;gBACD,QAAQ,eAAe,CAAC,IAAI,EAAE,CAAC;oBAC7B,KAAK,IAAI;wBACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;wBACzD,MAAM;oBACR,KAAK,IAAI;wBACP,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;wBACzD,MAAM;gBACV,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,GAAG,EAA+D,CAAC,CAC3E;YACD,gCAAgC,EAAE,IAAI,CAAC,mCAAmC,EAAE;SAC7E,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,eAAe,EAAE,gCAAgC,EAAE,EAAE,EAAE;YAC5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAoF,CAAC;YAC3G,MAAM,SAAS,GAAG,IAAI,GAAG,EAAc,CAAC;YACxC,gCAAgC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CACvD,WAAW,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CACjH,CAAC;YACF,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,EACF,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,wBAAwB,CAAC;IACvC,CAAC;IAEO,oBAAoB;QAM1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;;;;;;;;;;;;OAoBb,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,mBAAmB,EAAE,CACtI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,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;QAC1I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,mCAAmC;QACzC,IAAI,CAAC,iCAAiC,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CACzE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAAE,EAAE;YACxD,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;gBACvB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAC/B,CAAC;YACD,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAA4C,CAAC,EACvD,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,iCAAiC,CAAC;IAChD,CAAC;IAEM,4BAA4B,CAAC,OAAmB,EAAE,WAAoB;QAC3E,OAAO,IAAI,CAAC,mCAAmC,EAAE,CAAC,IAAI,CACpD,QAAQ,CAAC,CAAC,gCAAgC,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAa,CAAC;YACtC,MAAM,WAAW,GAAG,gCAAgC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClE,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC;YACD,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAC1C,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAChD,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EACpE,OAAO,EAAE,CACV,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,0BAA0B,CAAC,WAAoB,EAAE,gBAA0B;QAChF,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,QAAQ,CAAC,CAAC,uBAAuB,EAAE,EAAE,CACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACnC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,MAAM,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;YAC5C,uBAAuB,CAAC,OAAO,CAAC,CAAC,EAAE,aAAa,EAAE,aAAa,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE;gBACxF,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,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;QAC5F,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CAAC,OAAmB;QAC5C,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE;YAC9B,OAAO;gBACL,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;gBAC9E,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;aAC/E,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAC3C,QAAQ,CAAC,CAAC,oBAAoB,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,EACjE,MAAM,CACJ,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE,EAAE,EAAE;YACxC,aAAa,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACnD,aAAa,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YACnD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,OAAO,EAAE,IAAI,GAAG,EAAc,EAAE,OAAO,EAAE,IAAI,GAAG,EAAc,EAAE,CACnE,CACF,CAAC;IACJ,CAAC;IAEM,WAAW,CAAC,SAAqB;QACtC,OAAO,IAAI,CAAC,0BAA0B,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,uBAAuB,EAAE,EAAE,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEM,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QACzE,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEO,oBAAoB;QAO1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;YAC9C,MAAM,IAAI,GAAG;gBACX;YACI,mBAAmB;;;;;mBAKZ,yBAAyB;mBACzB,8BAA8B;mBAC9B,+BAA+B;;gCAElB,IAAI,CAAC,gBAAgB,CAAC,4BAA4B;;;;;;;;;;;;gBAYlE,mBAAmB;qBACd,yBAAyB;;;;SAIrC;aACF,CAAC;YACF,MAAM,KAAK,GAAG;;;;;;;mBAOD,6BAA6B;mBAC7B,0BAA0B;mBAC1B,oCAAoC;;;;;;mBAMpC,6BAA6B;mBAC7B,0BAA0B;mBAC1B,oCAAoC;;;;eAIxC,mBAAmB;OAC3B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,IAAI,EAAE,KAAK,EAAE,EACf,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,kBAAkB,EAAE,CACrI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO;gBACL,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,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,sBAAsB;QAC5B,IAAI,CAAC,oBAAoB,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAC5D,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,EAAE,EAAE;YAClF,MAAM,eAAe,GAAG,OAAO,IAAI,QAAQ,CAAC;YAC5C,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC1C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,6BAA6B,EAAE,SAAS,EAAE,CAAC;gBACxI,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;YACvC,CAAC;YACD,UAAU,CAAC,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,IAAI,mBAAmB,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,mBAAmB,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,eAAe,EAAE,CAAC;gBAC/I,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,mBAAmB,CAAC,6BAA6B,GAAG,eAAe,CAAC;YACtE,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAgE,CAAC,EAC3E,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,yBAAyB,CAAC,wBAAiC;QAChE,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACtF,IAAI,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,CAAC;YAChD,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAC/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAChD,MAAM,CACJ,CAAC,GAAG,EAAE,uBAAuB,EAAE,EAAE;YAC/B,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO,GAAG,CAAC;YACb,CAAC;YACD,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC5D,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC5D,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;YAC5E,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,OAAO,EAAE,IAAI,KAAK,EAAc,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,CACtH,EACD,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,oBAAoB,EAAE,EAAE,EAAE;YACtD,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtC,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAClC,CAAC;YACD,OAAO,IAAI,CAAC,yBAAyB,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAC9D,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE;gBAClB,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;gBACrC,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;gBACrC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YAC9B,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,6BAA6B,CAAC,wBAAiC;QACpE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAoB,CAAC;QAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,CAAC;YAChD,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAC/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAChD,MAAM,CAAC,CAAC,GAAG,EAAE,uBAAuB,EAAE,EAAE;YACtC,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,GAAG,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,sBAAsB,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,MAAM,CAAC,CACX,CACF,CACF,CAAC;IACJ,CAAC;IAEM,yBAAyB,CAAC,iBAA0B;QACzD,OAAO,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE;YAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAChD,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,qBAAqB;QAC1B,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC,GAAG,mBAAmB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3G,CAAC;IAEO,yBAAyB,CAAC,EAAE,YAAY,EAAE,YAAY,EAAoD;QAKhH,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,OAAO,GAAG,IAAI,KAAK,EAAU,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC;;iBAEJ,6BAA6B;mCACX,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC;SACxD,CAAC,CAAC;YACL,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC;;iBAEJ,6BAA6B;mCACX,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC;SACxD,CAAC,CAAC;YACL,CAAC;YACD,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,EACtC;gBACE,SAAS,EAAE,oBAAoB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,sBAAsB,IAAI,CAAC,WAAW,EAAE,EAAE;aACpG,CACF,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAC1E,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,uBAAuB,CAAC,EAC7B,YAAY,EACZ,YAAY,GAIb;QACC,MAAM,MAAM,GAAG,IAAI,GAAG,EAA8D,CAAC;QACrF,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACvE,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,qBAAqB,KAAK,IAAI,CAAC,yBAAyB,CAAC;YAC5D,YAAY;YACZ,YAAY;SACb,CAAC,CAAC,IAAI,CACL,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;YAC1C,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;YACrC,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,MAAM,CAAC,EACV,WAAW,EAAE,CACd,CAAC;QACF,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 { defer, EMPTY, forkJoin, from, map, mergeMap, of, reduce, shareReplay, toArray } from \"rxjs\";\nimport { Guid, Id64 } 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 CLASS_NAME_SubCategory,\n} from \"../../common/internal/ClassNameDefinitions.js\";\nimport { ModelCategoryElementsCountCache } from \"../../common/internal/ModelCategoryElementsCountCache.js\";\nimport { catchBeSQLiteInterrupts } from \"../../common/internal/UseErrorState.js\";\nimport { joinId64Arg } from \"../../common/internal/Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\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\n/* @internal /\nexport class ClassificationsTreeIdsCache implements Disposable {\n readonly #categoryElementCounts: ModelCategoryElementsCountCache;\n #categorySubCategories: Observable<Map<CategoryId, Array<SubCategoryId>>> \| undefined;\n #elementModelsCategories: Observable<Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>> \| undefined;\n #modelWithCategoryModeledElements: Observable<Map<ModelId, Map<CategoryId, Set<ElementId>>>> \| undefined;\n #classificationInfos: Observable<Map<ClassificationId \| ClassificationTableId, ClassificationInfo>> \| undefined;\n #filteredElementsData: Observable<Map<ElementId, { modelId: Id64String; categoryId: Id64String }>> \| undefined;\n #queryExecutor: LimitingECSqlQueryExecutor;\n #hierarchyConfig: ClassificationsTreeHierarchyConfiguration;\n #componentId: GuidString;\n #componentName: string;\n\n constructor(queryExecutor: LimitingECSqlQueryExecutor, hierarchyConfig: ClassificationsTreeHierarchyConfiguration, componentId?: GuidString) {\n this.#queryExecutor = queryExecutor;\n this.#hierarchyConfig = hierarchyConfig;\n this.#componentId = componentId ?? Guid.createValue();\n this.#componentName = \"ClassificationsTreeIdsCache\";\n this.#categoryElementCounts = new ModelCategoryElementsCountCache(\n this.#queryExecutor,\n [\"BisCore.GeometricElement2d\", \"BisCore.GeometricElement3d\"],\n this.#componentId,\n );\n }\n\n public [Symbol.dispose]() {\n this.#categoryElementCounts[Symbol.dispose]();\n }\n\n private querySubCategories(): Observable<{ id: SubCategoryId; parentId: CategoryId }> {\n return defer(() => {\n const definitionsQuery = `\n SELECT\n sc.ECInstanceId id,\n sc.Parent.Id categoryId\n FROM\n ${CLASS_NAME_SubCategory} sc\n WHERE\n NOT sc.IsPrivate\n `;\n return this.#queryExecutor.createQueryReader(\n { ecsql: definitionsQuery },\n { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/visible-sub-categories` },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return { id: row.id, parentId: row.categoryId };\n }),\n );\n }\n private getSubCategoriesInfo() {\n this.#categorySubCategories ??= this.querySubCategories()\n .pipe(\n reduce((acc, queriedSubCategory) => {\n const entry = acc.get(queriedSubCategory.parentId);\n if (entry) {\n entry.push(queriedSubCategory.id);\n } else {\n acc.set(queriedSubCategory.parentId, [queriedSubCategory.id]);\n }\n return acc;\n }, new Map<CategoryId, Array<SubCategoryId>>()),\n )\n .pipe(shareReplay());\n return this.#categorySubCategories;\n }\n\n public getSubCategories(categoryId: Id64String): Observable<Array<SubCategoryId>> {\n return this.getSubCategoriesInfo().pipe(map((categorySubCategories) => categorySubCategories.get(categoryId) ?? []));\n }\n\n private queryElementModelCategories(): Observable<{\n modelId: Id64String;\n categoryId: Id64String;\n type: \"2d\" \| \"3d\";\n }> {\n return defer(() => {\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 return this.#queryExecutor.createQueryReader(\n { ecsql: query },\n { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return { modelId: row.modelId, categoryId: row.categoryId, type: row.type };\n }),\n );\n }\n\n private getElementModelsCategories() {\n this.#elementModelsCategories ??= forkJoin({\n modelCategories: this.queryElementModelCategories().pipe(\n reduce((acc, queriedCategory) => {\n let modelEntry = acc.get(queriedCategory.modelId);\n if (modelEntry === undefined) {\n modelEntry = { category2dIds: new Set(), category3dIds: new Set() };\n acc.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 return acc;\n }, new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set }>()),\n ),\n modelWithCategoryModeledElements: this.getModelWithCategoryModeledElements(),\n }).pipe(\n map(({ modelCategories, modelWithCategoryModeledElements }) => {\n const result = new Map<ModelId, { category2dIds: Id64Set; category3dIds: Id64Set; isSubModel: boolean }>();\n const subModels = new Set<Id64String>();\n modelWithCategoryModeledElements.forEach((categoryMap) =>\n categoryMap.forEach((categorySubModels) => categorySubModels.forEach((subModelId) => subModels.add(subModelId))),\n );\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 shareReplay(),\n );\n return this.#elementModelsCategories;\n }\n\n private queryModeledElements(): Observable<{\n modelId: Id64String;\n modeledElementId: Id64String;\n categoryId: Id64String;\n rootCategoryId: Id64String;\n }> {\n return defer(() => {\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 return this.#queryExecutor.createQueryReader(\n { ecsql: query },\n { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, rootCategoryId: row.rootCategoryId };\n }),\n );\n }\n\n private getModelWithCategoryModeledElements() {\n this.#modelWithCategoryModeledElements ??= this.queryModeledElements().pipe(\n reduce((acc, { modelId, categoryId, modeledElementId }) => {\n let modelEntry = acc.get(modelId);\n if (!modelEntry) {\n modelEntry = new Map();\n acc.set(modelId, modelEntry);\n }\n const categoryEntry = modelEntry.get(categoryId);\n if (!categoryEntry) {\n modelEntry.set(categoryId, new Set([modeledElementId]));\n } else {\n categoryEntry.add(modeledElementId);\n }\n return acc;\n }, new Map<ModelId, Map<CategoryId, Set<ElementId>>>()),\n shareReplay(),\n );\n return this.#modelWithCategoryModeledElements;\n }\n\n public getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Observable<Id64Array> {\n return this.getModelWithCategoryModeledElements().pipe(\n mergeMap((modelWithCategoryModeledElements) => {\n const result = new Array<ElementId>();\n const categoryMap = modelWithCategoryModeledElements.get(modelId);\n if (!categoryMap) {\n return of(result);\n }\n return from(Id64.iterable(categoryIds)).pipe(\n map((categoryId) => categoryMap.get(categoryId)),\n mergeMap((elementsSet) => (elementsSet ? from(elementsSet) : EMPTY)),\n toArray(),\n );\n }),\n );\n }\n\n public getCategoriesElementModels(categoryIds: Id64Arg, includeSubModels?: boolean): Observable<{ id: CategoryId; models: Array<ModelId> \| undefined }> {\n return this.getElementModelsCategories().pipe(\n mergeMap((elementModelsCategories) =>\n from(Id64.iterable(categoryIds)).pipe(\n map((categoryId) => {\n const categoryModels = new Array<ModelId>();\n elementModelsCategories.forEach(({ category2dIds, category3dIds, isSubModel }, modelId) => {\n if ((includeSubModels \|\| !isSubModel) && (category2dIds.has(categoryId) \|\| category3dIds.has(categoryId))) {\n categoryModels.push(modelId);\n }\n });\n return { id: categoryId, models: categoryModels.length > 0 ? categoryModels : undefined };\n }),\n ),\n ),\n );\n }\n\n public getModelCategoryIds(modelId: Id64String): Observable<{ drawing: Id64Array; spatial: Id64Array }> {\n return this.getElementModelsCategories().pipe(\n map((elementModelsCategories) => {\n return {\n drawing: Array.from(elementModelsCategories.get(modelId)?.category2dIds ?? []),\n spatial: Array.from(elementModelsCategories.get(modelId)?.category3dIds ?? []),\n };\n }),\n );\n }\n\n public getAllCategories(): Observable<{ drawing: Id64Set; spatial: Id64Set }> {\n return this.getElementModelsCategories().pipe(\n mergeMap((modelsCategoriesInfo) => modelsCategoriesInfo.values()),\n reduce(\n (acc, { category2dIds, category3dIds }) => {\n category2dIds.forEach((id) => acc.drawing.add(id));\n category3dIds.forEach((id) => acc.spatial.add(id));\n return acc;\n },\n { drawing: new Set<Id64String>(), spatial: new Set<Id64String>() },\n ),\n );\n }\n\n public hasSubModel(elementId: Id64String): Observable<boolean> {\n return this.getElementModelsCategories().pipe(map((elementModelsCategories) => elementModelsCategories.has(elementId)));\n }\n\n public getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number> {\n return this.#categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n }\n\n private queryClassifications(): Observable<\n {\n id: Id64String;\n relatedCategories2d: CategoryId[];\n relatedCategories3d: CategoryId[];\n } & ({ tableId: ClassificationTableId; parentId: undefined } \| { tableId: undefined; parentId: ClassificationId })\n > {\n return defer(() => {\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 return this.#queryExecutor.createQueryReader(\n { ctes, ecsql },\n { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/classifications` },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return {\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\n private getClassificationsInfo() {\n this.#classificationInfos ??= this.queryClassifications().pipe(\n reduce((acc, { id, tableId, parentId, relatedCategories2d, relatedCategories3d }) => {\n const tableOrParentId = tableId ?? parentId;\n let parentInfo = acc.get(tableOrParentId);\n if (!parentInfo) {\n parentInfo = { childClassificationIds: [], relatedCategories2d: [], relatedCategories3d: [], parentClassificationOrTableId: undefined };\n acc.set(tableOrParentId, parentInfo);\n }\n parentInfo.childClassificationIds.push(id);\n let classificationEntry = acc.get(id);\n if (!classificationEntry) {\n classificationEntry = { childClassificationIds: [], relatedCategories2d, relatedCategories3d, parentClassificationOrTableId: tableOrParentId };\n acc.set(id, classificationEntry);\n } else {\n classificationEntry.parentClassificationOrTableId = tableOrParentId;\n }\n return acc;\n }, new Map<ClassificationId \| ClassificationTableId, ClassificationInfo>()),\n shareReplay(),\n );\n return this.#classificationInfos;\n }\n\n public getAllContainedCategories(classificationOrTableIds: Id64Arg): Observable<{ drawing: Id64Array; spatial: Id64Array }> {\n const result = { drawing: new Array<CategoryId>(), spatial: new Array<CategoryId>() };\n if (Id64.sizeOf(classificationOrTableIds) === 0) {\n return of(result);\n }\n return this.getClassificationsInfo().pipe(\n mergeMap((classificationsInfo) =>\n from(Id64.iterable(classificationOrTableIds)).pipe(\n reduce(\n (acc, classificationOrTableId) => {\n const classificationInfo = classificationsInfo.get(classificationOrTableId);\n if (classificationInfo === undefined) {\n return acc;\n }\n acc.drawing.push(...classificationInfo.relatedCategories2d);\n acc.spatial.push(...classificationInfo.relatedCategories3d);\n acc.childClassifications.push(...classificationInfo.childClassificationIds);\n return acc;\n },\n { drawing: new Array<Id64String>(), spatial: new Array<Id64String>(), childClassifications: new Array<Id64String>() },\n ),\n mergeMap(({ drawing, spatial, childClassifications }) => {\n if (childClassifications.length === 0) {\n return of({ drawing, spatial });\n }\n return this.getAllContainedCategories(childClassifications).pipe(\n map((childResult) => {\n drawing.push(...childResult.drawing);\n spatial.push(...childResult.spatial);\n return { drawing, spatial };\n }),\n );\n }),\n ),\n ),\n );\n }\n\n public getDirectChildClassifications(classificationOrTableIds: Id64Arg): Observable<ClassificationId[]> {\n const result = new Array<ClassificationId>();\n if (Id64.sizeOf(classificationOrTableIds) === 0) {\n return of(result);\n }\n return this.getClassificationsInfo().pipe(\n mergeMap((classificationsInfo) =>\n from(Id64.iterable(classificationOrTableIds)).pipe(\n reduce((acc, classificationOrTableId) => {\n const classificationInfo = classificationsInfo.get(classificationOrTableId);\n if (classificationInfo !== undefined) {\n acc.push(...classificationInfo.childClassificationIds);\n }\n return acc;\n }, result),\n ),\n ),\n );\n }\n\n public getClassificationsPathObs(classificationIds: Id64Arg): Observable<HierarchyNodeIdentifiersPath> {\n return from(this.getClassificationsInfo()).pipe(\n mergeMap((classificationsInfo) => {\n return from(Id64.iterable(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 getAllClassifications(): Observable<ClassificationId[]> {\n return this.getClassificationsInfo().pipe(map((classificationsInfo) => [...classificationsInfo.keys()]));\n }\n\n private queryFilteredElementsData({ element2dIds, element3dIds }: { element2dIds: Id64Arg; element3dIds: Id64Arg }): Observable<{\n modelId: Id64String;\n id: ElementId;\n categoryId: Id64String;\n }> {\n return defer(() => {\n const queries = new Array<string>();\n if (Id64.sizeOf(element2dIds) > 0) {\n queries.push(`\n SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id\n FROM ${CLASS_NAME_GeometricElement2d}\n WHERE ECInstanceId IN (${joinId64Arg(element2dIds, \",\")})\n `);\n }\n if (Id64.sizeOf(element3dIds) > 0) {\n queries.push(`\n SELECT Model.Id modelId, Category.Id categoryId, ECInstanceId id\n FROM ${CLASS_NAME_GeometricElement3d}\n WHERE ECInstanceId IN (${joinId64Arg(element3dIds, \",\")})\n `);\n }\n return this.#queryExecutor.createQueryReader(\n { ecsql: queries.join(\" UNION ALL \") },\n {\n rowFormat: \"ECSqlPropertyNames\",\n limit: \"unbounded\",\n restartToken: `${this.#componentName}/${this.#componentId}/filtered-elements/${Guid.createValue()}`,\n },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return { modelId: row.modelId, id: row.id, categoryId: row.categoryId };\n }),\n );\n }\n\n public getFilteredElementsData({\n element2dIds,\n element3dIds,\n }: {\n element2dIds: Id64Arg;\n element3dIds: Id64Arg;\n }): Observable<Map<ElementId, { categoryId: Id64String; modelId: Id64String }>> {\n const result = new Map<ElementId, { categoryId: Id64String; modelId: Id64String }>();\n if (Id64.sizeOf(element2dIds) === 0 && Id64.sizeOf(element3dIds) === 0) {\n return of(result);\n }\n this.#filteredElementsData ??= this.queryFilteredElementsData({\n element2dIds,\n element3dIds,\n }).pipe(\n reduce((acc, { modelId, id, categoryId }) => {\n acc.set(id, { modelId, categoryId });\n return acc;\n }, result),\n shareReplay(),\n );\n return this.#filteredElementsData;\n }\n\n public clearFilteredElementsData() {\n this.#filteredElementsData = undefined;\n }\n}\n"]}
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,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AAC3E,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8CAA8C,CAAC;AAChF,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,oCAAoC,EACpC,6BAA6B,EAC7B,0BAA0B,GAC3B,MAAM,+CAA+C,CAAC;AACvD,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAoB7D,gBAAgB;AAChB,MAAM,OAAO,2BAA4B,SAAQ,gBAAgB;IAC/D,oBAAoB,CAA4F;IAChH,qBAAqB,CAAsI;IAC3J,cAAc,CAA6B;IAC3C,gBAAgB,CAA4C;IAC5D,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAY,KAAuC;QACjD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,6BAA6B,CAAC;IACtD,CAAC;IAEO,oBAAoB;QAM1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;YAC9C,MAAM,IAAI,GAAG;gBACX;YACI,mBAAmB;;;;;mBAKZ,yBAAyB;mBACzB,8BAA8B;mBAC9B,+BAA+B;;gCAElB,IAAI,CAAC,gBAAgB,CAAC,4BAA4B;;;;;;;;;;;;gBAYlE,mBAAmB;qBACd,yBAAyB;;;;SAIrC;aACF,CAAC;YACF,MAAM,KAAK,GAAG;;;;;;;mBAOD,6BAA6B;mBAC7B,0BAA0B;mBAC1B,oCAAoC;;;;eAIxC,mBAAmB;OAC3B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,IAAI,EAAE,KAAK,EAAE,EACf,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,kBAAkB,EAAE,CACrI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAE,GAAG,CAAC,iBAA4B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;aAC7F,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,sBAAsB;QAC5B,IAAI,CAAC,oBAAoB,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAC5D,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,iBAAiB,EAAE,EAAE,EAAE;YAC3D,MAAM,eAAe,GAAG,OAAO,IAAI,QAAQ,CAAC;YAC5C,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC1C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,6BAA6B,EAAE,SAAS,EAAE,CAAC;gBAC7G,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;YACvC,CAAC;YACD,UAAU,CAAC,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,IAAI,mBAAmB,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACzB,mBAAmB,GAAG,EAAE,sBAAsB,EAAE,EAAE,EAAE,iBAAiB,EAAE,6BAA6B,EAAE,eAAe,EAAE,CAAC;gBACxH,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,mBAAmB,CAAC,6BAA6B,GAAG,eAAe,CAAC;YACtE,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAgE,CAAC,EAC3E,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,yBAAyB,CAAC,wBAAiC;QAChE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,IAAI,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,CAAC;YAChD,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAC/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAChD,MAAM,CACJ,CAAC,GAAG,EAAE,uBAAuB,EAAE,EAAE;YAC/B,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO,GAAG,CAAC;YACb,CAAC;YACD,kBAAkB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YAC9E,kBAAkB,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7F,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,CACvF,EACD,QAAQ,CAAC,CAAC,EAAE,UAAU,EAAE,oBAAoB,EAAE,EAAE,EAAE;YAChD,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,IAAI,CAAC,yBAAyB,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAC9D,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;gBACtB,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;gBACrD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,6BAA6B,CAAC,wBAAiC;QACpE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAoB,CAAC;QAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,EAAE,CAAC;YAChD,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAC/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAChD,MAAM,CAAC,CAAC,GAAG,EAAE,uBAAuB,EAAE,EAAE;YACtC,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,kBAAkB,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1E,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,MAAM,CAAC,CACX,CACF,CACF,CAAC;IACJ,CAAC;IAEM,yBAAyB,CAAC,iBAA0B;QACzD,OAAO,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,mBAAmB,EAAE,EAAE;YAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAChD,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,qBAAqB;QAC1B,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC,GAAG,mBAAmB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3G,CAAC;IAEO,yBAAyB,CAAC,EAAE,UAAU,EAA2B;QAMvE,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;uBASG,6BAA6B;;;;uBAI7B,6BAA6B;;;;;;;eAOrC,6BAA6B;iCACX,WAAW,CAAC,UAAU,EAAE,GAAG,CAAC;OACtD,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB;gBACE,SAAS,EAAE,oBAAoB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,sBAAsB,IAAI,CAAC,WAAW,EAAE,EAAE;aACpG,CACF,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO;gBACL,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,8BAA8B,EAAE,GAAG,CAAC,8BAA8B;aACnE,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,uBAAuB,CAAC,EAC7B,UAAU,GAGX;QACC,MAAM,MAAM,GAAG,IAAI,GAAG,EAA0G,CAAC;QACjI,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,qBAAqB,KAAK,IAAI,CAAC,yBAAyB,CAAC;YAC5D,UAAU;SACX,CAAC,CAAC,IAAI,CACL,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE,EAAE,EAAE;YAC1E,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,8BAA8B,EAAE,CAAC,CAAC;YACrE,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,MAAM,CAAC,EACV,WAAW,EAAE,CACd,CAAC;QACF,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 { defer, from, map, mergeMap, of, reduce, shareReplay } from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { BaseIdsCacheImpl } from \"../../common/internal/caches/BaseIdsCache.js\";\nimport {\n CLASS_NAME_Classification,\n CLASS_NAME_ClassificationSystem,\n CLASS_NAME_ClassificationTable,\n CLASS_NAME_ElementHasClassifications,\n CLASS_NAME_GeometricElement3d,\n CLASS_NAME_SpatialCategory,\n} from \"../../common/internal/ClassNameDefinitions.js\";\nimport { catchBeSQLiteInterrupts } from \"../../common/internal/UseErrorState.js\";\nimport { joinId64Arg } from \"../../common/internal/Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { BaseIdsCacheImplProps } from \"../../common/internal/caches/BaseIdsCache.js\";\nimport type { CategoryId, ClassificationId, ClassificationTableId, ElementId } from \"../../common/internal/Types.js\";\nimport type { ClassificationsTreeHierarchyConfiguration } from \"../ClassificationsTreeDefinition.js\";\n\ninterface ClassificationInfo {\n parentClassificationOrTableId: ClassificationId \| ClassificationTableId \| undefined;\n childClassificationIds: ClassificationId[];\n relatedCategories: CategoryId[];\n}\n\ninterface ClassificationsTreeIdsCacheProps extends BaseIdsCacheImplProps {\n queryExecutor: LimitingECSqlQueryExecutor;\n hierarchyConfig: ClassificationsTreeHierarchyConfiguration;\n}\n\n/** @internal */\nexport class ClassificationsTreeIdsCache extends BaseIdsCacheImpl {\n #classificationInfos: Observable<Map<ClassificationId \| ClassificationTableId, ClassificationInfo>> \| undefined;\n #filteredElementsData: Observable<Map<ElementId, { modelId: Id64String; categoryId: Id64String; categoryOfTopMostParentElement: CategoryId }>> \| undefined;\n #queryExecutor: LimitingECSqlQueryExecutor;\n #hierarchyConfig: ClassificationsTreeHierarchyConfiguration;\n #componentId: GuidString;\n #componentName: string;\n\n constructor(props: ClassificationsTreeIdsCacheProps) {\n super(props);\n this.#queryExecutor = props.queryExecutor;\n this.#hierarchyConfig = props.hierarchyConfig;\n this.#componentId = Guid.createValue();\n this.#componentName = \"ClassificationsTreeIdsCache\";\n }\n\n private queryClassifications(): Observable<\n {\n id: Id64String;\n relatedCategories: CategoryId[];\n } & ({ tableId: ClassificationTableId; parentId: undefined } \| { tableId: undefined; parentId: ClassificationId })\n > {\n return defer(() => {\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_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 ) relatedCategories\n FROM ${CLASSIFICATIONS_CTE} cl\n `;\n return this.#queryExecutor.createQueryReader(\n { ctes, ecsql },\n { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/classifications` },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return {\n id: row.id,\n tableId: row.tableId,\n parentId: row.parentId,\n relatedCategories: row.relatedCategories ? (row.relatedCategories as string).split(\",\") : [],\n };\n }),\n );\n }\n\n private getClassificationsInfo() {\n this.#classificationInfos ??= this.queryClassifications().pipe(\n reduce((acc, { id, tableId, parentId, relatedCategories }) => {\n const tableOrParentId = tableId ?? parentId;\n let parentInfo = acc.get(tableOrParentId);\n if (!parentInfo) {\n parentInfo = { childClassificationIds: [], relatedCategories: [], parentClassificationOrTableId: undefined };\n acc.set(tableOrParentId, parentInfo);\n }\n parentInfo.childClassificationIds.push(id);\n let classificationEntry = acc.get(id);\n if (!classificationEntry) {\n classificationEntry = { childClassificationIds: [], relatedCategories, parentClassificationOrTableId: tableOrParentId };\n acc.set(id, classificationEntry);\n } else {\n classificationEntry.parentClassificationOrTableId = tableOrParentId;\n }\n return acc;\n }, new Map<ClassificationId \| ClassificationTableId, ClassificationInfo>()),\n shareReplay(),\n );\n return this.#classificationInfos;\n }\n\n public getAllContainedCategories(classificationOrTableIds: Id64Arg): Observable<Id64Array> {\n const result = new Array<CategoryId>();\n if (Id64.sizeOf(classificationOrTableIds) === 0) {\n return of(result);\n }\n return this.getClassificationsInfo().pipe(\n mergeMap((classificationsInfo) =>\n from(Id64.iterable(classificationOrTableIds)).pipe(\n reduce(\n (acc, classificationOrTableId) => {\n const classificationInfo = classificationsInfo.get(classificationOrTableId);\n if (classificationInfo === undefined) {\n return acc;\n }\n classificationInfo.relatedCategories.forEach((id) => acc.categories.push(id));\n classificationInfo.childClassificationIds.forEach((id) => acc.childClassifications.push(id));\n return acc;\n },\n { categories: new Array<CategoryId>(), childClassifications: new Array<Id64String>() },\n ),\n mergeMap(({ categories, childClassifications }) => {\n if (childClassifications.length === 0) {\n return of(categories);\n }\n return this.getAllContainedCategories(childClassifications).pipe(\n map((childCategories) => {\n childCategories.forEach((id) => categories.push(id));\n return categories;\n }),\n );\n }),\n ),\n ),\n );\n }\n\n public getDirectChildClassifications(classificationOrTableIds: Id64Arg): Observable<ClassificationId[]> {\n const result = new Array<ClassificationId>();\n if (Id64.sizeOf(classificationOrTableIds) === 0) {\n return of(result);\n }\n return this.getClassificationsInfo().pipe(\n mergeMap((classificationsInfo) =>\n from(Id64.iterable(classificationOrTableIds)).pipe(\n reduce((acc, classificationOrTableId) => {\n const classificationInfo = classificationsInfo.get(classificationOrTableId);\n if (classificationInfo !== undefined) {\n classificationInfo.childClassificationIds.forEach((id) => acc.push(id));\n }\n return acc;\n }, result),\n ),\n ),\n );\n }\n\n public getClassificationsPathObs(classificationIds: Id64Arg): Observable<HierarchyNodeIdentifiersPath> {\n return from(this.getClassificationsInfo()).pipe(\n mergeMap((classificationsInfo) => {\n return from(Id64.iterable(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 getAllClassifications(): Observable<ClassificationId[]> {\n return this.getClassificationsInfo().pipe(map((classificationsInfo) => [...classificationsInfo.keys()]));\n }\n\n private queryFilteredElementsData({ elementIds }: { elementIds: Id64Arg }): Observable<{\n modelId: Id64String;\n id: ElementId;\n categoryId: Id64String;\n categoryOfTopMostParentElement: Id64String;\n }> {\n return defer(() => {\n const query = `\n SELECT\n this.Model.Id modelId,\n this.Category.Id categoryId,\n this.ECInstanceId id,\n (\n WITH RECURSIVE\n ParentWithCategory(id, categoryId, parentId) AS (\n SELECT e.ECInstanceId, e.Category.Id, e.Parent.Id\n FROM ${CLASS_NAME_GeometricElement3d} e\n WHERE e.ECInstanceId = this.ECInstanceId\n UNION ALL\n SELECT p.ECInstanceId, p.Category.Id, p.Parent.Id\n FROM ${CLASS_NAME_GeometricElement3d} p\n JOIN ParentWithCategory c ON p.ECInstanceId = c.parentId\n )\n SELECT IdToHex(categoryId)\n FROM ParentWithCategory\n WHERE parentId IS NULL\n ) categoryOfTopMostParentElement\n FROM ${CLASS_NAME_GeometricElement3d} this\n WHERE ECInstanceId IN (${joinId64Arg(elementIds, \",\")})\n `;\n return this.#queryExecutor.createQueryReader(\n { ecsql: query },\n {\n rowFormat: \"ECSqlPropertyNames\",\n limit: \"unbounded\",\n restartToken: `${this.#componentName}/${this.#componentId}/filtered-elements/${Guid.createValue()}`,\n },\n );\n }).pipe(\n catchBeSQLiteInterrupts,\n map((row) => {\n return {\n modelId: row.modelId,\n id: row.id,\n categoryId: row.categoryId,\n categoryOfTopMostParentElement: row.categoryOfTopMostParentElement,\n };\n }),\n );\n }\n\n public getFilteredElementsData({\n elementIds,\n }: {\n elementIds: Id64Arg;\n }): Observable<Map<ElementId, { categoryId: Id64String; modelId: Id64String; categoryOfTopMostParentElement: CategoryId }>> {\n const result = new Map<ElementId, { categoryId: Id64String; modelId: Id64String; categoryOfTopMostParentElement: CategoryId }>();\n if (Id64.sizeOf(elementIds) === 0) {\n return of(result);\n }\n this.#filteredElementsData ??= this.queryFilteredElementsData({\n elementIds,\n }).pipe(\n reduce((acc, { modelId, id, categoryId, categoryOfTopMostParentElement }) => {\n acc.set(id, { modelId, categoryId, categoryOfTopMostParentElement });\n return acc;\n }, result),\n shareReplay(),\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/ClassificationsTreeNodeInternal.d.ts ADDED Viewed

@@ -0,0 +1,29 @@
+import type { Id64String } from "@itwin/core-bentley";
+import type { HierarchyNode, InstancesNodeKey, NonGroupingHierarchyNode } from "@itwin/presentation-hierarchies";
+/**
+ * Contains utility functions for working with Classifications Tree nodes.
+ *
+ * It is a wrapper around `ClassificationsTreeNode` that exposes some internal details that should not be exposed via public api.
+ *
+ * @internal
+ */
+export declare namespace ClassificationsTreeNodeInternal {
+    const isClassificationTableNode: (node: Pick<HierarchyNode, "extendedData">) => node is NonGroupingHierarchyNode & {
+        key: InstancesNodeKey;
+    };
+    const isClassificationNode: (node: Pick<HierarchyNode, "extendedData">) => node is NonGroupingHierarchyNode & {
+        key: InstancesNodeKey;
+    };
+    const isGeometricElementNode: (node: Pick<HierarchyNode, "extendedData">) => node is Omit<NonGroupingHierarchyNode, "extendedData"> & {
+        key: InstancesNodeKey;
+    } & {
+        extendedData: {
+            modelId: Id64String;
+            categoryId: Id64String;
+            childrenCount: number;
+            categoryOfTopMostParentElement: Id64String;
+            topMostParentElementId: Id64String;
+        };
+    };
+}
+//# sourceMappingURL=ClassificationsTreeNodeInternal.d.ts.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNodeInternal.js ADDED Viewed

@@ -0,0 +1,19 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { ClassificationsTreeNode } from "../ClassificationsTreeNode.js";
+/**
+ * Contains utility functions for working with Classifications Tree nodes.
+ *
+ * It is a wrapper around `ClassificationsTreeNode` that exposes some internal details that should not be exposed via public api.
+ *
+ * @internal
+ */
+export var ClassificationsTreeNodeInternal;
+(function (ClassificationsTreeNodeInternal) {
+    ClassificationsTreeNodeInternal.isClassificationTableNode = ClassificationsTreeNode.isClassificationTableNode;
+    ClassificationsTreeNodeInternal.isClassificationNode = ClassificationsTreeNode.isClassificationNode;
+    ClassificationsTreeNodeInternal.isGeometricElementNode = (node) => ClassificationsTreeNode.isGeometricElementNode(node);
+})(ClassificationsTreeNodeInternal || (ClassificationsTreeNodeInternal = {}));
+//# sourceMappingURL=ClassificationsTreeNodeInternal.js.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNodeInternal.js.map ADDED Viewed

@@ -0,0 +1 @@

+ {"version":3,"file":"ClassificationsTreeNodeInternal.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/ClassificationsTreeNodeInternal.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AAKxE;;;;;;GAMG;AACH,MAAM,KAAW,+BAA+B,CAgB/C;AAhBD,WAAiB,+BAA+B;IACjC,yDAAyB,GAAG,uBAAuB,CAAC,yBAAyB,CAAC;IAE9E,oDAAoB,GAAG,uBAAuB,CAAC,oBAAoB,CAAC;IAEpE,sDAAsB,GAAG,CACpC,IAAyC,EASzC,EAAE,CAAC,uBAAuB,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;AAC5D,CAAC,EAhBgB,+BAA+B,KAA/B,+BAA+B,QAgB/C","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 { ClassificationsTreeNode } from \"../ClassificationsTreeNode.js\";\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, InstancesNodeKey, NonGroupingHierarchyNode } from \"@itwin/presentation-hierarchies\";\n\n/**\n * Contains utility functions for working with Classifications Tree nodes.\n *\n * It is a wrapper around `ClassificationsTreeNode` that exposes some internal details that should not be exposed via public api.\n *\n * @internal\n */\nexport namespace ClassificationsTreeNodeInternal {\n export const isClassificationTableNode = ClassificationsTreeNode.isClassificationTableNode;\n\n export const isClassificationNode = ClassificationsTreeNode.isClassificationNode;\n\n export const isGeometricElementNode = (\n node: Pick<HierarchyNode, \"extendedData\">,\n ): node is Omit<NonGroupingHierarchyNode, \"extendedData\"> & { key: InstancesNodeKey } & {\n extendedData: {\n modelId: Id64String;\n categoryId: Id64String;\n childrenCount: number;\n categoryOfTopMostParentElement: Id64String;\n topMostParentElementId: Id64String;\n };\n } => ClassificationsTreeNode.isGeometricElementNode(node);\n}\n"]}

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

@@ -1,6 +1,6 @@
 import type { Observable } from "rxjs";
 import type { HierarchyNode } from "@itwin/presentation-hierarchies";
-import type { AlwaysAndNeverDrawnElementInfo } from "../../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import type { AlwaysAndNeverDrawnElementInfoCache } from "../../../common/internal/caches/AlwaysAndNeverDrawnElementInfoCache.js";
 import type { TreeSpecificVisibilityHandler } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
 import type { TreeWidgetViewport } from "../../../common/TreeWidgetViewport.js";
 import type { VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
@@ -10,7 +10,7 @@ import type { ClassificationsTreeSearchTargets } from "./SearchResultsTree.js";
 export interface ClassificationsTreeVisibilityHandlerProps {
     idsCache: ClassificationsTreeIdsCache;
     viewport: TreeWidgetViewport;
-    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;
+    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfoCache;
 }
 /**
  * Handles visibility status of classifications tree nodes.
@@ -23,15 +23,11 @@ export declare class ClassificationsTreeVisibilityHandler implements Disposable,
     constructor(constructorProps: ClassificationsTreeVisibilityHandlerProps);
     [Symbol.dispose](): void;
     changeSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets, on: boolean): Observable<void>;
+    private changeSearchTargetElementsVisibilityStatus;
     getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus>;
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void>;
     getSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets): Observable<VisibilityStatus>;
-    private getCategories;
-    private getAllCategories;
-    private getElementsCount;
-    private getModels;
-    private getSubCategories;
-    private getSubModels;
+    private getSearchTargetElementsVisibilityStatus;
 }
 //# sourceMappingURL=ClassificationsTreeVisibilityHandler.d.ts.map