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

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

@@ -2,12 +2,13 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { concat, defer, EMPTY, from, map, merge, mergeMap, of, toArray } from "rxjs";
-import { Id64 } from "@itwin/core-bentley";
-import { HierarchyNode } from "@itwin/presentation-hierarchies";
-import { fromWithRelease } from "../../../common/internal/Utils.js";
+import { concat, defer, EMPTY, from, map, merge, mergeAll, mergeMap, of } from "rxjs";
+import { assert } from "@itwin/core-bentley";
+import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
+import { fromWithRelease, getIdsFromChildrenTree, getParentElementsIdsPath, setDifference, setIntersection } from "../../../common/internal/Utils.js";
 import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
-import { ClassificationsTreeNode } from "../ClassificationsTreeNode.js";
+import { ClassificationsTreeNodeInternal } from "../ClassificationsTreeNodeInternal.js";
 import { ClassificationsTreeVisibilityHelper } from "./ClassificationsTreeVisibilityHelper.js";
 /**
  * Handles visibility status of classifications tree nodes.
@@ -20,22 +21,11 @@ export class ClassificationsTreeVisibilityHandler {
     #visibilityHelper;
     constructor(constructorProps) {
         this.#props = constructorProps;
-        // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.
-        // We won't need to create a custom base ids cache.
-        const baseIdsCache = {
-            getCategories: (props) => this.getCategories(props),
-            getAllCategories: () => this.getAllCategories(),
-            getElementsCount: (props) => this.getElementsCount(props),
-            getModels: (props) => this.getModels(props),
-            getSubCategories: (props) => this.getSubCategories(props),
-            getSubModels: (props) => this.getSubModels(props),
-            hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),
-        };
         this.#visibilityHelper = new ClassificationsTreeVisibilityHelper({
             viewport: this.#props.viewport,
             idsCache: this.#props.idsCache,
             alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
-            baseIdsCache,
+            baseIdsCache: this.#props.idsCache,
         });
     }
     [Symbol.dispose]() {
@@ -43,7 +33,10 @@ export class ClassificationsTreeVisibilityHandler {
     }
     changeSearchTargetsVisibilityStatus(targets, on) {
         return defer(() => {
-            const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;
+            if (this.#props.viewport.viewType !== "3d") {
+                return EMPTY;
+            }
+            const { classificationIds, classificationTableIds, elements } = targets;
             const observables = new Array();
             if (classificationTableIds?.size) {
                 observables.push(this.#visibilityHelper.changeClassificationTablesVisibilityStatus({ classificationTableIds, on }));
@@ -51,60 +44,125 @@ export class ClassificationsTreeVisibilityHandler {
             if (classificationIds?.size) {
                 observables.push(this.#visibilityHelper.changeClassificationsVisibilityStatus({ classificationIds, on }));
             }
-            if (elements2d?.length) {
-                observables.push(from(elements2d).pipe(mergeMap(({ modelId, categoryId, elements }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }))));
-            }
-            if (elements3d?.length) {
-                observables.push(from(elements3d).pipe(mergeMap(({ modelId, categoryId, elements }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }))));
+            if (elements?.length) {
+                observables.push(this.changeSearchTargetElementsVisibilityStatus({ elements, on }));
             }
             return merge(...observables);
         });
     }
+    changeSearchTargetElementsVisibilityStatus({ elements, on, }) {
+        const searchTargetElements = new Array();
+        const elementIdsSet = new Set();
+        const modelCategoryElementMap = new Map();
+        // elements is an array that stores elements grouped by:
+        // 1. Their path
+        // 2. Their modelId and categoryId
+        // When changing visibility of elements, visibility handler does not care about the path.
+        // So we can first get all elements and group them only by modelId and categoryId.
+        for (const { elements: elementsMap, categoryId, modelId } of elements) {
+            const key = `${modelId}-${categoryId}`;
+            let mapEntry = modelCategoryElementMap.get(key);
+            if (!mapEntry) {
+                mapEntry = [];
+                modelCategoryElementMap.set(key, mapEntry);
+            }
+            for (const [elementId, { isSearchTarget }] of elementsMap) {
+                mapEntry.push(elementId);
+                elementIdsSet.add(elementId);
+                if (isSearchTarget) {
+                    searchTargetElements.push(elementId);
+                }
+            }
+        }
+        // Get children for search targets, since non search targets don't have all the children present in the hierarchy.
+        return this.#props.idsCache.getChildElementsTree({ elementIds: searchTargetElements }).pipe(
+        // Need to filter out and keep only those children ids that are not part of elements that are present in search paths.
+        // Elements in search paths will have their visibility changed directly: they will be provided as elementIds to changeElementsVisibilityStatus.
+        map((childrenTree) => ({
+            childrenNotInSearchPaths: setDifference(getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 }), elementIdsSet),
+            childrenTree,
+        })), mergeMap(({ childrenNotInSearchPaths, childrenTree }) => fromWithRelease({ source: modelCategoryElementMap.entries(), size: modelCategoryElementMap.size, releaseOnCount: 50 }).pipe(mergeMap(([key, elementsInSearchPathsGroupedByModelAndCategory]) => {
+            const [modelId, categoryId] = key.split("-");
+            const childrenIds = new Set();
+            // A shared children tree was created, need to get the children for each element in the group.
+            for (const elementId of elementsInSearchPathsGroupedByModelAndCategory) {
+                const elementChildrenTree = childrenTree.get(elementId)?.children;
+                if (!elementChildrenTree) {
+                    continue;
+                }
+                for (const childId of getIdsFromChildrenTree({ tree: elementChildrenTree })) {
+                    childrenIds.add(childId);
+                }
+            }
+            return this.#visibilityHelper.changeElementsVisibilityStatus({
+                modelId,
+                categoryId,
+                elementIds: elementsInSearchPathsGroupedByModelAndCategory,
+                // Pass only those children that are not part of search paths.
+                children: setIntersection(childrenIds, childrenNotInSearchPaths),
+                on,
+            });
+        }))));
+    }
     getVisibilityStatus(node) {
-        if (!HierarchyNode.isInstancesNode(node)) {
-            return EMPTY;
+        if (this.#props.viewport.viewType !== "3d") {
+            return of(createVisibilityStatus("disabled"));
         }
-        if (ClassificationsTreeNode.isClassificationTableNode(node)) {
+        if (ClassificationsTreeNodeInternal.isClassificationTableNode(node)) {
             return this.#visibilityHelper.getClassificationTablesVisibilityStatus({
                 classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
             });
         }
-        if (ClassificationsTreeNode.isClassificationNode(node)) {
+        if (ClassificationsTreeNodeInternal.isClassificationNode(node)) {
             return this.#visibilityHelper.getClassificationsVisibilityStatus({
                 classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
             });
         }
+        assert(ClassificationsTreeNodeInternal.isGeometricElementNode(node));
+        const parentElementsIdsPath = getParentElementsIdsPath({
+            parentInstanceKeys: node.parentKeys.filter((parentKey) => HierarchyNodeKey.isInstances(parentKey)).map((parentKey) => parentKey.instanceKeys),
+            topMostParentElementId: node.extendedData.topMostParentElementId,
+        });
         return this.#visibilityHelper.getElementsVisibilityStatus({
             elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
-            modelId: ClassificationsTreeNode.getModelId(node),
-            categoryId: ClassificationsTreeNode.getCategoryId(node),
-            type: node.extendedData?.type,
+            modelId: node.extendedData.modelId,
+            categoryId: node.extendedData.categoryId,
+            parentElementsIdsPath,
+            childrenCount: node.extendedData.childrenCount,
+            categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,
         });
     }
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node, on) {
         const changeObs = defer(() => {
-            if (!HierarchyNode.isInstancesNode(node)) {
+            if (this.#props.viewport.viewType !== "3d") {
                 return EMPTY;
             }
-            if (ClassificationsTreeNode.isClassificationTableNode(node)) {
+            if (ClassificationsTreeNodeInternal.isClassificationTableNode(node)) {
                 return this.#visibilityHelper.changeClassificationTablesVisibilityStatus({
                     classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                     on,
                 });
             }
-            if (ClassificationsTreeNode.isClassificationNode(node)) {
+            if (ClassificationsTreeNodeInternal.isClassificationNode(node)) {
                 return this.#visibilityHelper.changeClassificationsVisibilityStatus({
                     classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                     on,
                 });
             }
-            return this.#visibilityHelper.changeElementsVisibilityStatus({
-                elementIds: node.key.instanceKeys.map(({ id }) => id),
-                modelId: ClassificationsTreeNode.getModelId(node),
-                categoryId: ClassificationsTreeNode.getCategoryId(node),
+            assert(ClassificationsTreeNodeInternal.isGeometricElementNode(node));
+            const elementIds = node.key.instanceKeys.map(({ id }) => id);
+            return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(map((childrenTree) => {
+                // Children tree contains provided elementIds, they are at the root of this tree.
+                // We want to skip them and only get ids of children.
+                return getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 });
+            }), mergeMap((children) => this.#visibilityHelper.changeElementsVisibilityStatus({
+                elementIds,
+                modelId: node.extendedData.modelId,
+                categoryId: node.extendedData.categoryId,
+                children: children.size > 0 ? children : undefined,
                 on,
-            });
+            })));
         });
         if (this.#props.viewport.isAlwaysDrawnExclusive) {
             return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);
@@ -112,8 +170,11 @@ export class ClassificationsTreeVisibilityHandler {
         return changeObs;
     }
     getSearchTargetsVisibilityStatus(targets) {
+        if (this.#props.viewport.viewType !== "3d") {
+            return of(createVisibilityStatus("disabled"));
+        }
         return defer(() => {
-            const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;
+            const { classificationIds, classificationTableIds, elements } = targets;
             const observables = new Array();
             if (classificationTableIds?.size) {
                 observables.push(this.#visibilityHelper.getClassificationTablesVisibilityStatus({ classificationTableIds }));
@@ -121,56 +182,66 @@ export class ClassificationsTreeVisibilityHandler {
             if (classificationIds?.size) {
                 observables.push(this.#visibilityHelper.getClassificationsVisibilityStatus({ classificationIds }));
             }
-            if (elements2d?.length) {
-                observables.push(fromWithRelease({ source: elements2d, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, categoryId, elements }) => {
-                    return fromWithRelease({ source: elements.keys(), size: elements.size, releaseOnCount: 1000 }).pipe(mergeMap((elementId) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: "GeometricElement2d" })));
-                })));
-            }
-            if (elements3d?.length) {
-                observables.push(fromWithRelease({ source: elements3d, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, categoryId, elements }) => {
-                    return fromWithRelease({ source: elements.keys(), size: elements.size, releaseOnCount: 1000 }).pipe(mergeMap((elementId) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: "GeometricElement3d" })));
-                })));
+            if (elements?.length) {
+                observables.push(this.getSearchTargetElementsVisibilityStatus({ elements }));
             }
-            return merge(...observables);
-        }).pipe(mergeVisibilityStatuses);
-    }
-    getCategories(props) {
-        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache
-            .getModelCategoryIds(modelId)
-            .pipe(map(({ spatial, drawing }) => ({ id: modelId, drawingCategories: drawing, spatialCategories: spatial })))));
-    }
-    getAllCategories() {
-        return this.#props.idsCache.getAllCategories().pipe(map(({ drawing, spatial }) => {
-            return { drawingCategories: drawing, spatialCategories: spatial };
-        }));
-    }
-    getElementsCount(props) {
-        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
-    }
-    getModels(props) {
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds, true);
-    }
-    getSubCategories(props) {
-        return this.#props.idsCache.getSubCategories(props.categoryId);
+            return from(observables).pipe(mergeAll(), mergeVisibilityStatuses());
+        });
     }
-    getSubModels(props) {
-        if ("modelIds" in props) {
-            return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => {
-                if (props.categoryId) {
-                    return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));
+    getSearchTargetElementsVisibilityStatus({ elements, }) {
+        const searchTargetElements = new Array();
+        for (const { elements: elementsMap } of elements) {
+            for (const [elementId, { isSearchTarget }] of elementsMap) {
+                if (isSearchTarget) {
+                    searchTargetElements.push(elementId);
                 }
-                return from(this.#props.idsCache.getModelCategoryIds(modelId)).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)), map((subModels) => ({ id: modelId, subModels })));
-            }));
-        }
-        if (props.modelId) {
-            return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesModeledElements(props.modelId, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels })))));
+            }
         }
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(mergeMap(({ id, models }) => {
-            if (!models) {
-                return of({ id, subModels: undefined });
+        return this.#props.idsCache.getAllChildElementsCount({ elementIds: searchTargetElements }).pipe(mergeMap((elementsChildrenCountMap) => fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, categoryId, elements: elementsMap, pathToElements, categoryOfTopMostParentElement, topMostParentElementId }) => {
+            const parentElementsIdsPath = topMostParentElementId
+                ? getParentElementsIdsPath({
+                    parentInstanceKeys: pathToElements.map((instanceKey) => [instanceKey]),
+                    topMostParentElementId,
+                })
+                : [];
+            let totalSearchTargetsChildrenCount = 0;
+            const nonSearchTargetIds = new Array();
+            const searchTargetIds = new Array();
+            for (const [elementId, { isSearchTarget }] of elementsMap) {
+                if (!isSearchTarget) {
+                    nonSearchTargetIds.push(elementId);
+                    continue;
+                }
+                searchTargetIds.push(elementId);
+                const childCount = elementsChildrenCountMap.get(elementId);
+                if (childCount) {
+                    totalSearchTargetsChildrenCount += childCount;
+                }
             }
-            return from(models).pipe(mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)), map((subModels) => ({ id, subModels })));
-        }));
+            return merge(searchTargetIds.length > 0
+                ? this.#visibilityHelper.getElementsVisibilityStatus({
+                    modelId,
+                    categoryId,
+                    elementIds: searchTargetIds,
+                    parentElementsIdsPath,
+                    childrenCount: totalSearchTargetsChildrenCount,
+                    categoryOfTopMostParentElement,
+                })
+                : EMPTY, 
+            // Set childrenCount to undefined for non search targets, as some of their child elements might be filtered out.
+            // Since childrenCount is set to undefined, these elements won't check child always/never drawn child elements status.
+            // Child always/never drawn elements will be in search paths, and their visibility status will be handled separately.
+            nonSearchTargetIds.length > 0
+                ? this.#visibilityHelper.getElementsVisibilityStatus({
+                    modelId,
+                    categoryId,
+                    elementIds: nonSearchTargetIds,
+                    parentElementsIdsPath,
+                    childrenCount: undefined,
+                    categoryOfTopMostParentElement,
+                })
+                : EMPTY).pipe(mergeVisibilityStatuses());
+        }))));
     }
 }
 //# sourceMappingURL=ClassificationsTreeVisibilityHandler.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"ClassificationsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrF,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AACxE,OAAO,EAAE,mCAAmC,EAAE,MAAM,0CAA0C,CAAC;AAiB/F;;;;;GAKG;AACH,MAAM,OAAO,oCAAoC;IACtC,MAAM,CAA4C;IAC3D,iBAAiB,CAAsC;IAEvD,YAAY,gBAA2D;QACrE,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,6EAA6E;QAC7E,mDAAmD;QACnD,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YACnD,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAC/C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAC3C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACjD,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;SAChE,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,mCAAmC,CAAC;YAC/D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAyC,EAAE,EAAW;QAC/F,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;YACtF,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC,EAAE,sBAAsB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACtH,CAAC;YAED,IAAI,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5G,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAC7C,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACrH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAC7C,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACrH,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,uBAAuB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5D,OAAO,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC;gBACpE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aACnF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,uBAAuB,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBAC/D,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aAC9E,CAAC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;YACtE,OAAO,EAAE,uBAAuB,CAAC,UAAU,CAAC,IAAI,CAAC;YACjD,UAAU,EAAE,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC;YACvD,IAAI,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI;SAC9B,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,uBAAuB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5D,OAAO,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC;oBACvE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAClF,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,uBAAuB,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAClE,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC7E,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO,EAAE,uBAAuB,CAAC,UAAU,CAAC,IAAI,CAAC;gBACjD,UAAU,EAAE,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC;gBACvD,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAyC;QAC/E,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;YACtF,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;YAC/G,CAAC;YAED,IAAI,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC;YACrG,CAAC;YACD,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC9D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAC7C,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CACjG,QAAQ,CAAC,CAAC,SAAS,EAAE,EAAE,CACrB,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC/H,CACF,CAAC;gBACJ,CAAC,CAAC,CACH,CACF,CAAC;YACJ,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC9D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAC7C,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CACjG,QAAQ,CAAC,CAAC,SAAS,EAAE,EAAE,CACrB,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC/H,CACF,CAAC;gBACJ,CAAC,CAAC,CACH,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnC,CAAC;IAEO,aAAa,CAAC,KAAmD;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CACnB,IAAI,CAAC,MAAM,CAAC,QAAQ;aACjB,mBAAmB,CAAC,OAAO,CAAC;aAC5B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAClH,CACF,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CACjD,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;YAC3B,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC;QACpE,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAEO,SAAS,CAAC,KAA+C;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAClF,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACjE,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAC5E,QAAQ,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EACrF,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CACxC,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,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 { concat, defer, EMPTY, from, map, merge, mergeMap, of, toArray } from \"rxjs\";\nimport { Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { fromWithRelease } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ClassificationsTreeNode } from \"../ClassificationsTreeNode.js\";\nimport { ClassificationsTreeVisibilityHelper } from \"./ClassificationsTreeVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { BaseIdsCache, TreeSpecificVisibilityHandler } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\nimport type { ClassificationsTreeSearchTargets } from \"./SearchResultsTree.js\";\n\n/** @internal */\nexport interface ClassificationsTreeVisibilityHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n}\n\n/**\n * Handles visibility status of classifications tree nodes.\n *\n * This handler knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class ClassificationsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ClassificationsTreeSearchTargets> {\n  readonly #props: ClassificationsTreeVisibilityHandlerProps;\n  #visibilityHelper: ClassificationsTreeVisibilityHelper;\n\n  constructor(constructorProps: ClassificationsTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.\n    // We won't need to create a custom base ids cache.\n    const baseIdsCache: BaseIdsCache = {\n      getCategories: (props) => this.getCategories(props),\n      getAllCategories: () => this.getAllCategories(),\n      getElementsCount: (props) => this.getElementsCount(props),\n      getModels: (props) => this.getModels(props),\n      getSubCategories: (props) => this.getSubCategories(props),\n      getSubModels: (props) => this.getSubModels(props),\n      hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),\n    };\n    this.#visibilityHelper = new ClassificationsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      baseIdsCache,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;\n      const observables = new Array<Observable<void>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationTablesVisibilityStatus({ classificationTableIds, on }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationsVisibilityStatus({ classificationIds, on }));\n      }\n\n      if (elements2d?.length) {\n        observables.push(\n          from(elements2d).pipe(\n            mergeMap(({ modelId, categoryId, elements }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      if (elements3d?.length) {\n        observables.push(\n          from(elements3d).pipe(\n            mergeMap(({ modelId, categoryId, elements }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return EMPTY;\n    }\n\n    if (ClassificationsTreeNode.isClassificationTableNode(node)) {\n      return this.#visibilityHelper.getClassificationTablesVisibilityStatus({\n        classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (ClassificationsTreeNode.isClassificationNode(node)) {\n      return this.#visibilityHelper.getClassificationsVisibilityStatus({\n        classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      modelId: ClassificationsTreeNode.getModelId(node),\n      categoryId: ClassificationsTreeNode.getCategoryId(node),\n      type: node.extendedData?.type,\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (!HierarchyNode.isInstancesNode(node)) {\n        return EMPTY;\n      }\n\n      if (ClassificationsTreeNode.isClassificationTableNode(node)) {\n        return this.#visibilityHelper.changeClassificationTablesVisibilityStatus({\n          classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      if (ClassificationsTreeNode.isClassificationNode(node)) {\n        return this.#visibilityHelper.changeClassificationsVisibilityStatus({\n          classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId: ClassificationsTreeNode.getModelId(node),\n        categoryId: ClassificationsTreeNode.getCategoryId(node),\n        on,\n      });\n    });\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationTablesVisibilityStatus({ classificationTableIds }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationsVisibilityStatus({ classificationIds }));\n      }\n      if (elements2d?.length) {\n        observables.push(\n          fromWithRelease({ source: elements2d, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, categoryId, elements }) => {\n              return fromWithRelease({ source: elements.keys(), size: elements.size, releaseOnCount: 1000 }).pipe(\n                mergeMap((elementId) =>\n                  this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: \"GeometricElement2d\" }),\n                ),\n              );\n            }),\n          ),\n        );\n      }\n\n      if (elements3d?.length) {\n        observables.push(\n          fromWithRelease({ source: elements3d, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, categoryId, elements }) => {\n              return fromWithRelease({ source: elements.keys(), size: elements.size, releaseOnCount: 1000 }).pipe(\n                mergeMap((elementId) =>\n                  this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: \"GeometricElement3d\" }),\n                ),\n              );\n            }),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    }).pipe(mergeVisibilityStatuses);\n  }\n\n  private getCategories(props: Parameters<BaseIdsCache[\"getCategories\"]>[0]): ReturnType<BaseIdsCache[\"getCategories\"]> {\n    return from(Id64.iterable(props.modelIds)).pipe(\n      mergeMap((modelId) =>\n        this.#props.idsCache\n          .getModelCategoryIds(modelId)\n          .pipe(map(({ spatial, drawing }) => ({ id: modelId, drawingCategories: drawing, spatialCategories: spatial }))),\n      ),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(\n      map(({ drawing, spatial }) => {\n        return { drawingCategories: drawing, spatialCategories: spatial };\n      }),\n    );\n  }\n\n  private getElementsCount(props: Parameters<BaseIdsCache[\"getElementsCount\"]>[0]): ReturnType<BaseIdsCache[\"getElementsCount\"]> {\n    return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);\n  }\n\n  private getModels(props: Parameters<BaseIdsCache[\"getModels\"]>[0]): ReturnType<BaseIdsCache[\"getModels\"]> {\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds, true);\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return this.#props.idsCache.getSubCategories(props.categoryId);\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return from(this.#props.idsCache.getModelCategoryIds(modelId)).pipe(\n            mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n            toArray(),\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(\n      mergeMap(({ id, models }) => {\n        if (!models) {\n          return of({ id, subModels: undefined });\n        }\n        return from(models).pipe(\n          mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)),\n          map((subModels) => ({ id, subModels })),\n        );\n      }),\n    );\n  }\n}\n"]}
+{"version":3,"file":"ClassificationsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AACtF,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,wBAAwB,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACtJ,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,+BAA+B,EAAE,MAAM,uCAAuC,CAAC;AACxF,OAAO,EAAE,mCAAmC,EAAE,MAAM,0CAA0C,CAAC;AAqB/F;;;;;GAKG;AACH,MAAM,OAAO,oCAAoC;IACtC,MAAM,CAA4C;IAC3D,iBAAiB,CAAsC;IAEvD,YAAY,gBAA2D;QACrE,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,IAAI,CAAC,iBAAiB,GAAG,IAAI,mCAAmC,CAAC;YAC/D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;SACnC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAyC,EAAE,EAAW;QAC/F,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAC3C,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC,EAAE,sBAAsB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACtH,CAAC;YAED,IAAI,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5G,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,0CAA0C,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACtF,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,0CAA0C,CAAC,EACjD,QAAQ,EACR,EAAE,GAIH;QACC,MAAM,oBAAoB,GAAG,IAAI,KAAK,EAAc,CAAC;QACrD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAc,CAAC;QAC5C,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAgD,CAAC;QACxF,wDAAwD;QACxD,gBAAgB;QAChB,kCAAkC;QAClC,yFAAyF;QACzF,kFAAkF;QAClF,KAAK,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,QAAQ,EAAE,CAAC;YACtE,MAAM,GAAG,GAA+B,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;YACnE,IAAI,QAAQ,GAAG,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,EAAE,CAAC;gBACd,uBAAuB,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC7C,CAAC;YACD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;gBAC1D,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACzB,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC7B,IAAI,cAAc,EAAE,CAAC;oBACnB,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;QACD,kHAAkH;QAClH,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,oBAAoB,EAAE,CAAC,CAAC,IAAI;QACzF,sHAAsH;QACtH,+IAA+I;QAC/I,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACrB,wBAAwB,EAAE,aAAa,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,EAAE,aAAa,CAAC;YAC3I,YAAY;SACb,CAAC,CAAC,EACH,QAAQ,CAAC,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,EAAE,EAAE,CACtD,eAAe,CAAC,EAAE,MAAM,EAAE,uBAAuB,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,uBAAuB,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CACzH,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,8CAA8C,CAAC,EAAE,EAAE;YACjE,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAc,CAAC;YAC1C,8FAA8F;YAC9F,KAAK,MAAM,SAAS,IAAI,8CAA8C,EAAE,CAAC;gBACvE,MAAM,mBAAmB,GAA6B,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC;gBAC5F,IAAI,CAAC,mBAAmB,EAAE,CAAC;oBACzB,SAAS;gBACX,CAAC;gBACD,KAAK,MAAM,OAAO,IAAI,sBAAsB,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE,CAAC;oBAC5E,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,OAAO;gBACP,UAAU;gBACV,UAAU,EAAE,8CAA8C;gBAC1D,8DAA8D;gBAC9D,QAAQ,EAAE,eAAe,CAAC,WAAW,EAAE,wBAAwB,CAAC;gBAChE,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC3C,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,+BAA+B,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC;gBACpE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aACnF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,+BAA+B,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/D,OAAO,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBAC/D,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aAC9E,CAAC,CAAC;QACL,CAAC;QACD,MAAM,CAAC,+BAA+B,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;QACrE,MAAM,qBAAqB,GAAG,wBAAwB,CAAC;YACrD,kBAAkB,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,gBAAgB,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC;YAC7I,sBAAsB,EAAE,IAAI,CAAC,YAAY,CAAC,sBAAsB;SACjE,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;YACtE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;YACxC,qBAAqB;YACrB,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,aAAa;YAC9C,8BAA8B,EAAE,IAAI,CAAC,YAAY,CAAC,8BAA8B;SACjF,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAC3C,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,+BAA+B,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpE,OAAO,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC;oBACvE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAClF,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,+BAA+B,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAClE,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC7E,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YACD,MAAM,CAAC,+BAA+B,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;YACrE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CACnE,GAAG,CAAC,CAAC,YAAY,EAAW,EAAE;gBAC5B,iFAAiF;gBACjF,qDAAqD;gBACrD,OAAO,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7F,CAAC,CAAC,EACF,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CACpB,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBACpD,UAAU;gBACV,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;gBAClC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,QAAQ,EAAE,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;gBAClD,EAAE;aACH,CAAC,CACH,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAyC;QAC/E,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAC3C,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;YAC/G,CAAC;YAED,IAAI,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC;YACrG,CAAC;YACD,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,uCAAuC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YAC/E,CAAC;YAED,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,uCAAuC,CAAC,EAC9C,QAAQ,GAGT;QACC,MAAM,oBAAoB,GAAG,IAAI,KAAK,EAAc,CAAC;QACrD,KAAK,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,QAAQ,EAAE,CAAC;YACjD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;gBAC1D,IAAI,cAAc,EAAE,CAAC;oBACnB,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAC7F,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE,8BAA8B,EAAE,sBAAsB,EAAE,EAAE,EAAE;YAClI,MAAM,qBAAqB,GAAG,sBAAsB;gBAClD,CAAC,CAAC,wBAAwB,CAAC;oBACvB,kBAAkB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtE,sBAAsB;iBACvB,CAAC;gBACJ,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,+BAA+B,GAAG,CAAC,CAAC;YACxC,MAAM,kBAAkB,GAAG,IAAI,KAAK,EAAc,CAAC;YACnD,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;YAChD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;gBAC1D,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACnC,SAAS;gBACX,CAAC;gBACD,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAChC,MAAM,UAAU,GAAG,wBAAwB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC3D,IAAI,UAAU,EAAE,CAAC;oBACf,+BAA+B,IAAI,UAAU,CAAC;gBAChD,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CACV,eAAe,CAAC,MAAM,GAAG,CAAC;gBACxB,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;oBACjD,OAAO;oBACP,UAAU;oBACV,UAAU,EAAE,eAAe;oBAC3B,qBAAqB;oBACrB,aAAa,EAAE,+BAA+B;oBAC9C,8BAA8B;iBAC/B,CAAC;gBACJ,CAAC,CAAC,KAAK;YACT,gHAAgH;YAChH,sHAAsH;YACtH,qHAAqH;YACrH,kBAAkB,CAAC,MAAM,GAAG,CAAC;gBAC3B,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;oBACjD,OAAO;oBACP,UAAU;oBACV,UAAU,EAAE,kBAAkB;oBAC9B,qBAAqB;oBACrB,aAAa,EAAE,SAAS;oBACxB,8BAA8B;iBAC/B,CAAC;gBACJ,CAAC,CAAC,KAAK,CACV,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC;QACpC,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,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 { concat, defer, EMPTY, from, map, merge, mergeAll, mergeMap, of } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { fromWithRelease, getIdsFromChildrenTree, getParentElementsIdsPath, setDifference, setIntersection } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ClassificationsTreeNodeInternal } from \"../ClassificationsTreeNodeInternal.js\";\nimport { ClassificationsTreeVisibilityHelper } from \"./ClassificationsTreeVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport type { AlwaysAndNeverDrawnElementInfoCache } from \"../../../common/internal/caches/AlwaysAndNeverDrawnElementInfoCache.js\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { ChildrenTree } from \"../../../common/internal/Utils.js\";\nimport type { TreeSpecificVisibilityHandler } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\nimport type { ClassificationsTreeSearchTargets } from \"./SearchResultsTree.js\";\n\n/** @internal */\nexport interface ClassificationsTreeVisibilityHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfoCache;\n}\n\n/**\n * Handles visibility status of classifications tree nodes.\n *\n * This handler knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class ClassificationsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ClassificationsTreeSearchTargets> {\n  readonly #props: ClassificationsTreeVisibilityHandlerProps;\n  #visibilityHelper: ClassificationsTreeVisibilityHelper;\n\n  constructor(constructorProps: ClassificationsTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    this.#visibilityHelper = new ClassificationsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      baseIdsCache: this.#props.idsCache,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      if (this.#props.viewport.viewType !== \"3d\") {\n        return EMPTY;\n      }\n      const { classificationIds, classificationTableIds, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationTablesVisibilityStatus({ classificationTableIds, on }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationsVisibilityStatus({ classificationIds, on }));\n      }\n\n      if (elements?.length) {\n        observables.push(this.changeSearchTargetElementsVisibilityStatus({ elements, on }));\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  private changeSearchTargetElementsVisibilityStatus({\n    elements,\n    on,\n  }: {\n    elements: Required<ClassificationsTreeSearchTargets>[\"elements\"];\n    on: boolean;\n  }): Observable<void> {\n    const searchTargetElements = new Array<Id64String>();\n    const elementIdsSet = new Set<Id64String>();\n    const modelCategoryElementMap = new Map<`${ModelId}-${CategoryId}`, Array<ElementId>>();\n    // elements is an array that stores elements grouped by:\n    // 1. Their path\n    // 2. Their modelId and categoryId\n    // When changing visibility of elements, visibility handler does not care about the path.\n    // So we can first get all elements and group them only by modelId and categoryId.\n    for (const { elements: elementsMap, categoryId, modelId } of elements) {\n      const key: `${ModelId}-${CategoryId}` = `${modelId}-${categoryId}`;\n      let mapEntry = modelCategoryElementMap.get(key);\n      if (!mapEntry) {\n        mapEntry = [];\n        modelCategoryElementMap.set(key, mapEntry);\n      }\n      for (const [elementId, { isSearchTarget }] of elementsMap) {\n        mapEntry.push(elementId);\n        elementIdsSet.add(elementId);\n        if (isSearchTarget) {\n          searchTargetElements.push(elementId);\n        }\n      }\n    }\n    // Get children for search targets, since non search targets don't have all the children present in the hierarchy.\n    return this.#props.idsCache.getChildElementsTree({ elementIds: searchTargetElements }).pipe(\n      // Need to filter out and keep only those children ids that are not part of elements that are present in search paths.\n      // Elements in search paths will have their visibility changed directly: they will be provided as elementIds to changeElementsVisibilityStatus.\n      map((childrenTree) => ({\n        childrenNotInSearchPaths: setDifference(getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 }), elementIdsSet),\n        childrenTree,\n      })),\n      mergeMap(({ childrenNotInSearchPaths, childrenTree }) =>\n        fromWithRelease({ source: modelCategoryElementMap.entries(), size: modelCategoryElementMap.size, releaseOnCount: 50 }).pipe(\n          mergeMap(([key, elementsInSearchPathsGroupedByModelAndCategory]) => {\n            const [modelId, categoryId] = key.split(\"-\");\n            const childrenIds = new Set<Id64String>();\n            // A shared children tree was created, need to get the children for each element in the group.\n            for (const elementId of elementsInSearchPathsGroupedByModelAndCategory) {\n              const elementChildrenTree: ChildrenTree | undefined = childrenTree.get(elementId)?.children;\n              if (!elementChildrenTree) {\n                continue;\n              }\n              for (const childId of getIdsFromChildrenTree({ tree: elementChildrenTree })) {\n                childrenIds.add(childId);\n              }\n            }\n            return this.#visibilityHelper.changeElementsVisibilityStatus({\n              modelId,\n              categoryId,\n              elementIds: elementsInSearchPathsGroupedByModelAndCategory,\n              // Pass only those children that are not part of search paths.\n              children: setIntersection(childrenIds, childrenNotInSearchPaths),\n              on,\n            });\n          }),\n        ),\n      ),\n    );\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (this.#props.viewport.viewType !== \"3d\") {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n    if (ClassificationsTreeNodeInternal.isClassificationTableNode(node)) {\n      return this.#visibilityHelper.getClassificationTablesVisibilityStatus({\n        classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (ClassificationsTreeNodeInternal.isClassificationNode(node)) {\n      return this.#visibilityHelper.getClassificationsVisibilityStatus({\n        classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n    assert(ClassificationsTreeNodeInternal.isGeometricElementNode(node));\n    const parentElementsIdsPath = getParentElementsIdsPath({\n      parentInstanceKeys: node.parentKeys.filter((parentKey) => HierarchyNodeKey.isInstances(parentKey)).map((parentKey) => parentKey.instanceKeys),\n      topMostParentElementId: node.extendedData.topMostParentElementId,\n    });\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      modelId: node.extendedData.modelId,\n      categoryId: node.extendedData.categoryId,\n      parentElementsIdsPath,\n      childrenCount: node.extendedData.childrenCount,\n      categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (this.#props.viewport.viewType !== \"3d\") {\n        return EMPTY;\n      }\n      if (ClassificationsTreeNodeInternal.isClassificationTableNode(node)) {\n        return this.#visibilityHelper.changeClassificationTablesVisibilityStatus({\n          classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      if (ClassificationsTreeNodeInternal.isClassificationNode(node)) {\n        return this.#visibilityHelper.changeClassificationsVisibilityStatus({\n          classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n      assert(ClassificationsTreeNodeInternal.isGeometricElementNode(node));\n      const elementIds = node.key.instanceKeys.map(({ id }) => id);\n      return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(\n        map((childrenTree): Id64Set => {\n          // Children tree contains provided elementIds, they are at the root of this tree.\n          // We want to skip them and only get ids of children.\n          return getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 });\n        }),\n        mergeMap((children) =>\n          this.#visibilityHelper.changeElementsVisibilityStatus({\n            elementIds,\n            modelId: node.extendedData.modelId,\n            categoryId: node.extendedData.categoryId,\n            children: children.size > 0 ? children : undefined,\n            on,\n          }),\n        ),\n      );\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getSearchTargetsVisibilityStatus(targets: ClassificationsTreeSearchTargets): Observable<VisibilityStatus> {\n    if (this.#props.viewport.viewType !== \"3d\") {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n    return defer(() => {\n      const { classificationIds, classificationTableIds, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationTablesVisibilityStatus({ classificationTableIds }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationsVisibilityStatus({ classificationIds }));\n      }\n      if (elements?.length) {\n        observables.push(this.getSearchTargetElementsVisibilityStatus({ elements }));\n      }\n\n      return from(observables).pipe(mergeAll(), mergeVisibilityStatuses());\n    });\n  }\n\n  private getSearchTargetElementsVisibilityStatus({\n    elements,\n  }: {\n    elements: Required<ClassificationsTreeSearchTargets>[\"elements\"];\n  }): Observable<VisibilityStatus> {\n    const searchTargetElements = new Array<Id64String>();\n    for (const { elements: elementsMap } of elements) {\n      for (const [elementId, { isSearchTarget }] of elementsMap) {\n        if (isSearchTarget) {\n          searchTargetElements.push(elementId);\n        }\n      }\n    }\n    return this.#props.idsCache.getAllChildElementsCount({ elementIds: searchTargetElements }).pipe(\n      mergeMap((elementsChildrenCountMap) =>\n        fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n          mergeMap(({ modelId, categoryId, elements: elementsMap, pathToElements, categoryOfTopMostParentElement, topMostParentElementId }) => {\n            const parentElementsIdsPath = topMostParentElementId\n              ? getParentElementsIdsPath({\n                  parentInstanceKeys: pathToElements.map((instanceKey) => [instanceKey]),\n                  topMostParentElementId,\n                })\n              : [];\n            let totalSearchTargetsChildrenCount = 0;\n            const nonSearchTargetIds = new Array<Id64String>();\n            const searchTargetIds = new Array<Id64String>();\n            for (const [elementId, { isSearchTarget }] of elementsMap) {\n              if (!isSearchTarget) {\n                nonSearchTargetIds.push(elementId);\n                continue;\n              }\n              searchTargetIds.push(elementId);\n              const childCount = elementsChildrenCountMap.get(elementId);\n              if (childCount) {\n                totalSearchTargetsChildrenCount += childCount;\n              }\n            }\n            return merge(\n              searchTargetIds.length > 0\n                ? this.#visibilityHelper.getElementsVisibilityStatus({\n                    modelId,\n                    categoryId,\n                    elementIds: searchTargetIds,\n                    parentElementsIdsPath,\n                    childrenCount: totalSearchTargetsChildrenCount,\n                    categoryOfTopMostParentElement,\n                  })\n                : EMPTY,\n              // Set childrenCount to undefined for non search targets, as some of their child elements might be filtered out.\n              // Since childrenCount is set to undefined, these elements won't check child always/never drawn child elements status.\n              // Child always/never drawn elements will be in search paths, and their visibility status will be handled separately.\n              nonSearchTargetIds.length > 0\n                ? this.#visibilityHelper.getElementsVisibilityStatus({\n                    modelId,\n                    categoryId,\n                    elementIds: nonSearchTargetIds,\n                    parentElementsIdsPath,\n                    childrenCount: undefined,\n                    categoryOfTopMostParentElement,\n                  })\n                : EMPTY,\n            ).pipe(mergeVisibilityStatuses());\n          }),\n        ),\n      ),\n    );\n  }\n}\n"]}

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

@@ -2,9 +2,9 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { merge, mergeMap, of, toArray } from "rxjs";
+import { defaultIfEmpty, mergeMap } from "rxjs";
+import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
 import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
-import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
 /**
  * Visibility status helper for classifications tree.
  *
@@ -23,7 +23,10 @@ export class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
      * Determines visibility status by checking visibility status of related categories.
      */
     getClassificationTablesVisibilityStatus(props) {
-        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap(({ drawing, spatial }) => merge(of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "DrawingCategory" }))), of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "SpatialCategory" }))))), mergeVisibilityStatuses);
+        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap((categories) => this.getCategoriesVisibilityStatus({
+            modelId: undefined,
+            categoryIds: categories,
+        })), defaultIfEmpty(createVisibilityStatus("disabled")));
     }
     /**
      * Gets visibility status of classifications.
@@ -31,7 +34,10 @@ export class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
      * Determines visibility status by checking visibility status of related categories.
      */
     getClassificationsVisibilityStatus(props) {
-        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap(({ drawing, spatial }) => merge(of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "DrawingCategory" }))), of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "SpatialCategory" }))))), mergeVisibilityStatuses);
+        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap((categories) => this.getCategoriesVisibilityStatus({
+            modelId: undefined,
+            categoryIds: categories,
+        })), defaultIfEmpty(createVisibilityStatus("disabled")));
     }
     /**
      * Changes visibility status of classification tables.
@@ -39,7 +45,11 @@ export class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
      * Does this by changing visibility status of related categories.
      */
     changeClassificationTablesVisibilityStatus(props) {
-        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })));
+        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap((categories) => this.changeCategoriesVisibilityStatus({
+            modelId: undefined,
+            categoryIds: categories,
+            on: props.on,
+        })));
     }
     /**
      * Changes visibility status of classifications.
@@ -47,7 +57,11 @@ export class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
      * Does this by changing visibility status of related categories.
      */
     changeClassificationsVisibilityStatus(props) {
-        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })));
+        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap((categories) => this.changeCategoriesVisibilityStatus({
+            modelId: undefined,
+            categoryIds: categories,
+            on: props.on,
+        })));
     }
 }
 //# sourceMappingURL=ClassificationsTreeVisibilityHelper.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"ClassificationsTreeVisibilityHelper.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,6DAA6D,CAAC;AACnG,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAatF;;;;;GAKG;AACH,MAAM,OAAO,mCAAoC,SAAQ,oBAAoB;IAC3E,MAAM,CAA2C;IACjD,YAAY,KAA+C;QACzD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,uCAAuC,CAAC,KAA0C;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAChC,KAAK,CACH,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAC7I,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAC9I,CACF,EACD,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,kCAAkC,CAAC,KAAqC;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAChC,KAAK,CACH,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAC7I,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAC9I,CACF,EACD,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,0CAA0C,CAAC,KAAuD;QACvG,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CACpH,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,qCAAqC,CAAC,KAAkD;QAC7F,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CACpH,CAAC;IACJ,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 { merge, mergeMap, of, toArray } from \"rxjs\";\nimport { BaseVisibilityHelper } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { BaseVisibilityHelperProps } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\n/** @internal */\nexport type ClassificationsTreeVisibilityHelperProps = BaseVisibilityHelperProps & {\n  idsCache: ClassificationsTreeIdsCache;\n};\n\n/**\n * Visibility status helper for classifications tree.\n *\n * It extends base visibility status helper and provides methods to get and change visibility status of classification tables and classifications.\n * @internal\n */\nexport class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {\n  #props: ClassificationsTreeVisibilityHelperProps;\n  constructor(props: ClassificationsTreeVisibilityHelperProps) {\n    super(props);\n    this.#props = props;\n  }\n\n  /**\n   * Gets visibility status of classification tables.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap(({ drawing, spatial }) =>\n        merge(\n          of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"DrawingCategory\" }))),\n          of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"SpatialCategory\" }))),\n        ),\n      ),\n      mergeVisibilityStatuses,\n    );\n  }\n\n  /**\n   * Gets visibility status of classifications.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationsVisibilityStatus(props: { classificationIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap(({ drawing, spatial }) =>\n        merge(\n          of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"DrawingCategory\" }))),\n          of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"SpatialCategory\" }))),\n        ),\n      ),\n      mergeVisibilityStatuses,\n    );\n  }\n\n  /**\n   * Changes visibility status of classification tables.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n      toArray(),\n      mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })),\n    );\n  }\n\n  /**\n   * Changes visibility status of classifications.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationsVisibilityStatus(props: { classificationIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n      toArray(),\n      mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })),\n    );\n  }\n}\n"]}
+{"version":3,"file":"ClassificationsTreeVisibilityHelper.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,MAAM,CAAC;AAChD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,oBAAoB,EAAE,MAAM,6DAA6D,CAAC;AAanG;;;;;GAKG;AACH,MAAM,OAAO,mCAAoC,SAAQ,oBAAoB;IAC3E,MAAM,CAA2C;IACjD,YAAY,KAA+C;QACzD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,uCAAuC,CAAC,KAA0C;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,6BAA6B,CAAC;YACjC,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE,UAAU;SACxB,CAAC,CACH,EACD,cAAc,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CACnD,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,kCAAkC,CAAC,KAAqC;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,6BAA6B,CAAC;YACjC,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE,UAAU;SACxB,CAAC,CACH,EACD,cAAc,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CACnD,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,0CAA0C,CAAC,KAAuD;QACvG,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,gCAAgC,CAAC;YACpC,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE,UAAU;YACvB,EAAE,EAAE,KAAK,CAAC,EAAE;SACb,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,qCAAqC,CAAC,KAAkD;QAC7F,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,gCAAgC,CAAC;YACpC,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE,UAAU;YACvB,EAAE,EAAE,KAAK,CAAC,EAAE;SACb,CAAC,CACH,CACF,CAAC;IACJ,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 { defaultIfEmpty, mergeMap } from \"rxjs\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { BaseVisibilityHelper } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { BaseVisibilityHelperProps } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\n/** @internal */\nexport type ClassificationsTreeVisibilityHelperProps = BaseVisibilityHelperProps & {\n  idsCache: ClassificationsTreeIdsCache;\n};\n\n/**\n * Visibility status helper for classifications tree.\n *\n * It extends base visibility status helper and provides methods to get and change visibility status of classification tables and classifications.\n * @internal\n */\nexport class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {\n  #props: ClassificationsTreeVisibilityHelperProps;\n  constructor(props: ClassificationsTreeVisibilityHelperProps) {\n    super(props);\n    this.#props = props;\n  }\n\n  /**\n   * Gets visibility status of classification tables.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap((categories) =>\n        this.getCategoriesVisibilityStatus({\n          modelId: undefined,\n          categoryIds: categories,\n        }),\n      ),\n      defaultIfEmpty(createVisibilityStatus(\"disabled\")),\n    );\n  }\n\n  /**\n   * Gets visibility status of classifications.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationsVisibilityStatus(props: { classificationIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap((categories) =>\n        this.getCategoriesVisibilityStatus({\n          modelId: undefined,\n          categoryIds: categories,\n        }),\n      ),\n      defaultIfEmpty(createVisibilityStatus(\"disabled\")),\n    );\n  }\n\n  /**\n   * Changes visibility status of classification tables.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap((categories) =>\n        this.changeCategoriesVisibilityStatus({\n          modelId: undefined,\n          categoryIds: categories,\n          on: props.on,\n        }),\n      ),\n    );\n  }\n\n  /**\n   * Changes visibility status of classifications.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationsVisibilityStatus(props: { classificationIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap((categories) =>\n        this.changeCategoriesVisibilityStatus({\n          modelId: undefined,\n          categoryIds: categories,\n          on: props.on,\n        }),\n      ),\n    );\n  }\n}\n"]}

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

@@ -1,24 +1,20 @@
 import type { Id64Set, Id64String } from "@itwin/core-bentley";
 import type { HierarchySearchPath } from "@itwin/presentation-hierarchies";
-import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
-import type { ElementId } from "../../../common/internal/Types.js";
+import type { ECClassHierarchyInspector, InstanceKey } from "@itwin/presentation-shared";
+import type { CategoryId, ElementId } from "../../../common/internal/Types.js";
 import type { SearchResultsTree } from "../../../common/internal/visibility/BaseSearchResultsTree.js";
 import type { ClassificationsTreeIdsCache } from "../ClassificationsTreeIdsCache.js";
 /** @internal */
 export interface ClassificationsTreeSearchTargets {
-    elements2d?: Array<{
-        modelId: Id64String;
-        categoryId: Id64String;
-        elements: Map<ElementId, {
-            isSearchTarget: boolean;
-        }>;
-    }>;
-    elements3d?: Array<{
+    elements?: Array<{
+        pathToElements: InstanceKey[];
         modelId: Id64String;
         categoryId: Id64String;
+        topMostParentElementId?: Id64String;
         elements: Map<ElementId, {
             isSearchTarget: boolean;
         }>;
+        categoryOfTopMostParentElement: CategoryId;
     }>;
     classificationTableIds?: Id64Set;
     classificationIds?: Id64Set;