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

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

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { firstValueFrom } from "rxjs";
 import { assert } from "@itwin/core-bentley";
-import { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement2d } from "../../../common/internal/ClassNameDefinitions.js";
+import { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement3d } from "../../../common/internal/ClassNameDefinitions.js";
 import { createSearchResultsTree, SearchResultsNodesHandler } from "../../../common/internal/visibility/BaseSearchResultsTree.js";
 /** @internal */
 export async function createClassificationsSearchResultsTree(props) {
@@ -21,34 +21,28 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
         this.#props = props;
     }
     async getProcessedSearchResultsNodes() {
-        const searchResultsTemporary2dElements = new Map();
-        const searchResultsTemporary3dElements = new Map();
+        const searchResultsTemporaryElements = new Map();
         const result = {
-            searchResults2dElements: new Map(),
-            searchResults3dElements: new Map(),
+            searchResultsElements: new Map(),
         };
         for (const node of this.searchResultsNodesArr) {
-            if (node.type === "element2d") {
-                searchResultsTemporary2dElements.set(node.id, node);
-            }
-            else if (node.type === "element3d") {
-                searchResultsTemporary3dElements.set(node.id, node);
+            if (node.type === "element") {
+                searchResultsTemporaryElements.set(node.id, node);
             }
         }
         const searchResultsElementsModels = await firstValueFrom(this.#props.idsCache.getFilteredElementsData({
-            element2dIds: [...searchResultsTemporary2dElements.keys()],
-            element3dIds: [...searchResultsTemporary3dElements.keys()],
+            elementIds: [...searchResultsTemporaryElements.keys()],
         }));
-        searchResultsTemporary2dElements.forEach((element, id) => {
-            const entry = searchResultsElementsModels.get(element.id);
-            assert(entry !== undefined);
-            result.searchResults2dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });
-        });
-        searchResultsTemporary3dElements.forEach((element, id) => {
+        for (const [id, element] of searchResultsTemporaryElements) {
             const entry = searchResultsElementsModels.get(element.id);
             assert(entry !== undefined);
-            result.searchResults3dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });
-        });
+            result.searchResultsElements.set(id, {
+                ...element,
+                modelId: entry.modelId,
+                categoryId: entry.categoryId,
+                categoryOfTopMostParentElement: entry.categoryOfTopMostParentElement,
+            });
+        }
         return result;
     }
     convertNodesToSearchTargets(searchResultsNodes, processedSearchResultsNodes) {
@@ -56,39 +50,51 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
         searchResultsNodes.forEach((searchResultsNode) => this.collectSearchTargets(searchTargets, searchResultsNode, processedSearchResultsNodes));
         return this.convertInternalSearchTargets(searchTargets);
     }
+    convertInternalSearchTargetElementsRecursively(searchTargetsInternalElements, currentPath) {
+        const result = [];
+        // Internal search target elements are stored in a tree structure, need to convert that to array structure.
+        for (const [identifierAsString, entry] of searchTargetsInternalElements) {
+            const identifier = this.convertSearchResultsNodeIdentifierStringToHierarchyNodeIdentifier(identifierAsString);
+            if (entry.modelCategoryElements) {
+                for (const [modelCategoryKey, { elementsMap: elements, categoryOfTopMostParentElement }] of entry.modelCategoryElements) {
+                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
+                    result.push({
+                        pathToElements: [...currentPath, identifier],
+                        modelId,
+                        categoryId,
+                        elements,
+                        categoryOfTopMostParentElement,
+                        topMostParentElementId: entry.topMostParentElementId,
+                    });
+                }
+            }
+            if (!entry.children) {
+                continue;
+            }
+            for (const childValue of this.convertInternalSearchTargetElementsRecursively(entry.children, [...currentPath, identifier])) {
+                result.push(childValue);
+            }
+        }
+        return result;
+    }
     convertInternalSearchTargets(searchTargets) {
-        if (!searchTargets.classificationIds && !searchTargets.classificationIds && !searchTargets.elements2d && !searchTargets.elements3d) {
+        if (!searchTargets.classificationIds && !searchTargets.classificationIds && !searchTargets.elements) {
             return undefined;
         }
         return {
             classificationIds: searchTargets.classificationIds,
             classificationTableIds: searchTargets.classificationIds,
-            elements2d: searchTargets.elements2d
-                ? [...searchTargets.elements2d?.entries()].map(([modelCategoryKey, elements]) => {
-                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
-                    return { modelId, categoryId, elements };
-                })
-                : undefined,
-            elements3d: searchTargets.elements3d
-                ? [...searchTargets.elements3d?.entries()].map(([modelCategoryKey, elements]) => {
-                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
-                    return { modelId, categoryId, elements };
-                })
-                : undefined,
+            elements: searchTargets.elements ? this.convertInternalSearchTargetElementsRecursively(searchTargets.elements, []) : undefined,
         };
     }
     collectSearchTargets(searchTargets, node, processedSearchResultsNodes) {
-        const searchResultsNode = node.type === "element2d"
-            ? processedSearchResultsNodes.searchResults2dElements.get(node.id)
-            : node.type === "element3d"
-                ? processedSearchResultsNodes.searchResults3dElements.get(node.id)
-                : node;
+        const searchResultsNode = node.type === "element" ? processedSearchResultsNodes.searchResultsElements.get(node.id) : node;
         assert(searchResultsNode !== undefined);
         if (searchResultsNode.isSearchTarget) {
             this.addTarget(searchTargets, searchResultsNode);
             return;
         }
-        if (searchResultsNode.type === "element2d" || searchResultsNode.type === "element3d") {
+        if (searchResultsNode.type === "element") {
             // need to add parent ids as search target will be an element
             this.addTarget(searchTargets, searchResultsNode);
         }
@@ -99,6 +105,47 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
             this.collectSearchTargets(searchTargets, child, processedSearchResultsNodes);
         }
     }
+    addTargetElement(searchTargets, node) {
+        // Internal search target elements need to have path saved in some way.
+        // For this, a tree structure is used, where keys are stringified identifiers of parent nodes depending on the hierarchy.
+        const modelCategoryKey = this.createModelCategoryKey(node.modelId, node.categoryId);
+        if (!searchTargets.elements) {
+            searchTargets.elements = new Map();
+        }
+        const searchTargetElementsMap = searchTargets.elements;
+        assert(searchTargetElementsMap !== undefined);
+        let entry = searchTargetElementsMap;
+        let topMostParentElementId;
+        for (let i = 0; i < node.pathToNode.length; ++i) {
+            if (topMostParentElementId === undefined && node.type === "element") {
+                topMostParentElementId = node.pathToNode[i].id;
+            }
+            const identifierAsString = this.convertSearchResultsNodeIdentifierToString(node.pathToNode[i]);
+            let identifierEntry = entry.get(identifierAsString);
+            // create a new entry for parent node if it does not exist
+            if (!identifierEntry) {
+                identifierEntry = { topMostParentElementId };
+                entry.set(identifierAsString, identifierEntry);
+            }
+            // last entry in the path don't need to have children
+            if (i < node.pathToNode.length - 1) {
+                identifierEntry.children ??= new Map();
+                entry = identifierEntry.children;
+                continue;
+            }
+            const elements = (identifierEntry.modelCategoryElements ??= new Map()).get(modelCategoryKey);
+            // Add elements who share the same path to the modelCategoryElements map
+            if (elements) {
+                elements.set(node.id, { isSearchTarget: node.isSearchTarget });
+            }
+            else {
+                identifierEntry.modelCategoryElements.set(modelCategoryKey, {
+                    elementsMap: new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]),
+                    categoryOfTopMostParentElement: node.categoryOfTopMostParentElement,
+                });
+            }
+        }
+    }
     addTarget(searchTargets, node) {
         switch (node.type) {
             case "classificationTable":
@@ -107,25 +154,8 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
             case "classification":
                 (searchTargets.classificationIds ??= new Set()).add(node.id);
                 return;
-            case "element2d":
-                const element2dKey = this.createModelCategoryKey(node.modelId, node.categoryId);
-                const elements2d = (searchTargets.elements2d ??= new Map()).get(element2dKey);
-                if (elements2d) {
-                    elements2d.set(node.id, { isSearchTarget: node.isSearchTarget });
-                }
-                else {
-                    searchTargets.elements2d.set(element2dKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));
-                }
-                return;
-            case "element3d":
-                const element3dKey = this.createModelCategoryKey(node.modelId, node.categoryId);
-                const elements3d = (searchTargets.elements3d ??= new Map()).get(element3dKey);
-                if (elements3d) {
-                    elements3d.set(node.id, { isSearchTarget: node.isSearchTarget });
-                }
-                else {
-                    searchTargets.elements3d.set(element3dKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));
-                }
+            case "element":
+                this.addTargetElement(searchTargets, node);
                 return;
         }
     }
@@ -136,20 +166,24 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
         const [modelId, categoryId] = key.split("-");
         return { modelId, categoryId };
     }
-    createSearchResultsTreeNode({ type, id, isSearchTarget, }) {
-        if (type === "element2d" || type === "element3d") {
+    createSearchResultsTreeNode({ type, id, isSearchTarget, parent, }) {
+        const pathToNode = "pathToNode" in parent ? [...parent.pathToNode, { type: parent.type, id: parent.id }] : [];
+        if (type === "element") {
             return {
                 id,
                 isSearchTarget,
                 type,
                 modelId: undefined,
                 categoryId: undefined,
+                categoryOfTopMostParentElement: undefined,
+                pathToNode,
             };
         }
         return {
             id,
             isSearchTarget,
             type,
+            pathToNode,
         };
     }
     async getType(className) {
@@ -159,10 +193,17 @@ class ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHan
         if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Classification)) {
             return "classification";
         }
-        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_GeometricElement2d)) {
-            return "element2d";
+        return "element";
+    }
+    getClassName(type) {
+        switch (type) {
+            case "classificationTable":
+                return CLASS_NAME_ClassificationTable;
+            case "classification":
+                return CLASS_NAME_Classification;
+            case "element":
+                return CLASS_NAME_GeometricElement3d;
         }
-        return "element3d";
     }
 }
 //# sourceMappingURL=SearchResultsTree.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"SearchResultsTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/SearchResultsTree.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,8BAA8B,EAAE,6BAA6B,EAAE,MAAM,kDAAkD,CAAC;AAC5J,OAAO,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,8DAA8D,CAAC;AA6DlI,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,sCAAsC,CAAC,KAI5D;IACC,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACtD,OAAO,uBAAuB,CAAC;QAC7B,yBAAyB,EAAE,IAAI,4CAA4C,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACvG,WAAW;KACZ,CAAC,CAAC;AACL,CAAC;AAqBD,MAAM,4CAA6C,SAAQ,yBAI1D;IACU,MAAM,CAAoD;IACnE,YAAY,KAAwD;QAClE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,8BAA8B;QACzC,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAqE,CAAC;QACtH,MAAM,gCAAgC,GAAG,IAAI,GAAG,EAAqE,CAAC;QACtH,MAAM,MAAM,GAAgC;YAC1C,uBAAuB,EAAE,IAAI,GAAG,EAAE;YAClC,uBAAuB,EAAE,IAAI,GAAG,EAAE;SACnC,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC9C,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC9B,gCAAgC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACtD,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrC,gCAAgC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,MAAM,2BAA2B,GAAG,MAAM,cAAc,CACtD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YAC3C,YAAY,EAAE,CAAC,GAAG,gCAAgC,CAAC,IAAI,EAAE,CAAC;YAC1D,YAAY,EAAE,CAAC,GAAG,gCAAgC,CAAC,IAAI,EAAE,CAAC;SAC3D,CAAC,CACH,CAAC;QACF,gCAAgC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE;YACvD,MAAM,KAAK,GAAG,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC5B,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QAC/G,CAAC,CAAC,CAAC;QACH,gCAAgC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE;YACvD,MAAM,KAAK,GAAG,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC5B,MAAM,CAAC,uBAAuB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QAC/G,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,2BAA2B,CAChC,kBAAoD,EACpD,2BAAwD;QAExD,MAAM,aAAa,GAA0B,EAAE,CAAC;QAEhD,kBAAkB,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,CAAC;QAE5I,OAAO,IAAI,CAAC,4BAA4B,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC;IAEO,4BAA4B,CAAC,aAAoC;QACvE,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YACnI,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,iBAAiB,EAAE,aAAa,CAAC,iBAAiB;YAClD,sBAAsB,EAAE,aAAa,CAAC,iBAAiB;YACvD,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBAC5E,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAC3C,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBAC5E,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAC3C,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAC1B,aAAoC,EACpC,IAAoC,EACpC,2BAAwD;QAExD,MAAM,iBAAiB,GACrB,IAAI,CAAC,IAAI,KAAK,WAAW;YACvB,CAAC,CAAC,2BAA2B,CAAC,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAClE,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW;gBACzB,CAAC,CAAC,2BAA2B,CAAC,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClE,CAAC,CAAC,IAAI,CAAC;QACb,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACxC,IAAI,iBAAiB,CAAC,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,iBAAiB,CAAC,IAAI,KAAK,WAAW,IAAI,iBAAiB,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrF,6DAA6D;YAC7D,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,KAAK,EAAE,2BAA2B,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,aAAoC,EAAE,IAA2B;QACjF,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,qBAAqB;gBACxB,CAAC,aAAa,CAAC,sBAAsB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClE,OAAO;YACT,KAAK,gBAAgB;gBACnB,CAAC,aAAa,CAAC,iBAAiB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7D,OAAO;YACT,KAAK,WAAW;gBACd,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChF,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9E,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5G,CAAC;gBACD,OAAO;YACT,KAAK,WAAW;gBACd,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChF,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9E,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5G,CAAC;gBACD,OAAO;QACX,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,OAAmB,EAAE,UAAsB;QACxE,OAAO,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;IACpC,CAAC;IAEO,qBAAqB,CAAC,GAAqB;QACjD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IACjC,CAAC;IAEM,2BAA2B,CAAC,EACjC,IAAI,EACJ,EAAE,EACF,cAAc,GAKf;QACC,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACjD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,SAAS;gBAClB,UAAU,EAAE,SAAS;aACtB,CAAC;QACJ,CAAC;QACD,OAAO;YACL,EAAE;YACF,cAAc;YACd,IAAI;SACL,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,8BAA8B,CAAC,EAAE,CAAC;YAC/F,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,yBAAyB,CAAC,EAAE,CAAC;YAC1F,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,6BAA6B,CAAC,EAAE,CAAC;YAC9F,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,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 { firstValueFrom } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement2d } from \"../../../common/internal/ClassNameDefinitions.js\";\nimport { createSearchResultsTree, SearchResultsNodesHandler } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\n\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { BaseSearchResultsTreeNode, SearchResultsTree, SearchResultsTreeNodeChildren } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\ninterface ClassificationTableSearchResultsTreeNode extends BaseSearchResultsTreeNode<ClassificationTableSearchResultsTreeNode> {\n  type: \"classificationTable\";\n}\n\ninterface ClassificationSearchResultsTreeNode extends BaseSearchResultsTreeNode<ClassificationSearchResultsTreeNode> {\n  type: \"classification\";\n}\n\ninterface Element2dSearchResultsTreeNode extends BaseSearchResultsTreeNode<Element2dSearchResultsTreeNode> {\n  type: \"element2d\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ninterface Element3dSearchResultsTreeNode extends BaseSearchResultsTreeNode<Element3dSearchResultsTreeNode> {\n  type: \"element3d\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ntype SearchResultsTreeNode =\n  | ClassificationTableSearchResultsTreeNode\n  | ClassificationSearchResultsTreeNode\n  | Element2dSearchResultsTreeNode\n  | Element3dSearchResultsTreeNode;\n\ntype TemporaryElement2dSearchResultsNode = Omit<Element2dSearchResultsTreeNode, \"modelId\" | \"categoryId\" | \"children\"> & {\n  modelId: string | undefined;\n  categoryId: string | undefined;\n  children?: SearchResultsTreeNodeChildren<TemporaryElement2dSearchResultsNode>;\n};\n\ntype TemporaryElement3dSearchResultsNode = Omit<Element3dSearchResultsTreeNode, \"modelId\" | \"categoryId\" | \"children\"> & {\n  modelId: string | undefined;\n  categoryId: string | undefined;\n  children?: SearchResultsTreeNodeChildren<TemporaryElement3dSearchResultsNode>;\n};\n\ntype TemporarySearchResultsTreeNode =\n  | ClassificationTableSearchResultsTreeNode\n  | ClassificationSearchResultsTreeNode\n  | TemporaryElement2dSearchResultsNode\n  | TemporaryElement3dSearchResultsNode;\n\n/** @internal */\nexport interface ClassificationsTreeSearchTargets {\n  elements2d?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isSearchTarget: boolean }> }>;\n  elements3d?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isSearchTarget: boolean }> }>;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\n/** @internal */\nexport async function createClassificationsSearchResultsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n  idsCache: ClassificationsTreeIdsCache;\n}): Promise<SearchResultsTree<ClassificationsTreeSearchTargets>> {\n  const { imodelAccess, searchPaths, idsCache } = props;\n  return createSearchResultsTree({\n    searchResultsNodesHandler: new ClassificationsTreeSearchResultsNodesHandler({ idsCache, imodelAccess }),\n    searchPaths,\n  });\n}\n\ninterface SearchTargetsInternal {\n  elements2d?: Map<ModelCategoryKey, Map<ElementId, { isSearchTarget: boolean }>>;\n  elements3d?: Map<ModelCategoryKey, Map<ElementId, { isSearchTarget: boolean }>>;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\ninterface ClassificationsTreeSearchResultsNodesHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\ninterface ProcessedSearchResultsNodes {\n  searchResults2dElements: Map<Id64String, Omit<Element2dSearchResultsTreeNode, \"children\">>;\n  searchResults3dElements: Map<Id64String, Omit<Element3dSearchResultsTreeNode, \"children\">>;\n}\n\nclass ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHandler<\n  ProcessedSearchResultsNodes,\n  ClassificationsTreeSearchTargets,\n  TemporarySearchResultsTreeNode\n> {\n  readonly #props: ClassificationsTreeSearchResultsNodesHandlerProps;\n  constructor(props: ClassificationsTreeSearchResultsNodesHandlerProps) {\n    super();\n    this.#props = props;\n  }\n\n  public async getProcessedSearchResultsNodes(): Promise<ProcessedSearchResultsNodes> {\n    const searchResultsTemporary2dElements = new Map<Id64String, Omit<TemporaryElement2dSearchResultsNode, \"children\">>();\n    const searchResultsTemporary3dElements = new Map<Id64String, Omit<TemporaryElement3dSearchResultsNode, \"children\">>();\n    const result: ProcessedSearchResultsNodes = {\n      searchResults2dElements: new Map(),\n      searchResults3dElements: new Map(),\n    };\n    for (const node of this.searchResultsNodesArr) {\n      if (node.type === \"element2d\") {\n        searchResultsTemporary2dElements.set(node.id, node);\n      } else if (node.type === \"element3d\") {\n        searchResultsTemporary3dElements.set(node.id, node);\n      }\n    }\n\n    const searchResultsElementsModels = await firstValueFrom(\n      this.#props.idsCache.getFilteredElementsData({\n        element2dIds: [...searchResultsTemporary2dElements.keys()],\n        element3dIds: [...searchResultsTemporary3dElements.keys()],\n      }),\n    );\n    searchResultsTemporary2dElements.forEach((element, id) => {\n      const entry = searchResultsElementsModels.get(element.id);\n      assert(entry !== undefined);\n      result.searchResults2dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });\n    });\n    searchResultsTemporary3dElements.forEach((element, id) => {\n      const entry = searchResultsElementsModels.get(element.id);\n      assert(entry !== undefined);\n      result.searchResults3dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });\n    });\n    return result;\n  }\n\n  public convertNodesToSearchTargets(\n    searchResultsNodes: TemporarySearchResultsTreeNode[],\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ): ClassificationsTreeSearchTargets | undefined {\n    const searchTargets: SearchTargetsInternal = {};\n\n    searchResultsNodes.forEach((searchResultsNode) => this.collectSearchTargets(searchTargets, searchResultsNode, processedSearchResultsNodes));\n\n    return this.convertInternalSearchTargets(searchTargets);\n  }\n\n  private convertInternalSearchTargets(searchTargets: SearchTargetsInternal): ClassificationsTreeSearchTargets | undefined {\n    if (!searchTargets.classificationIds && !searchTargets.classificationIds && !searchTargets.elements2d && !searchTargets.elements3d) {\n      return undefined;\n    }\n\n    return {\n      classificationIds: searchTargets.classificationIds,\n      classificationTableIds: searchTargets.classificationIds,\n      elements2d: searchTargets.elements2d\n        ? [...searchTargets.elements2d?.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n      elements3d: searchTargets.elements3d\n        ? [...searchTargets.elements3d?.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n    };\n  }\n\n  private collectSearchTargets(\n    searchTargets: SearchTargetsInternal,\n    node: TemporarySearchResultsTreeNode,\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ) {\n    const searchResultsNode =\n      node.type === \"element2d\"\n        ? processedSearchResultsNodes.searchResults2dElements.get(node.id)\n        : node.type === \"element3d\"\n          ? processedSearchResultsNodes.searchResults3dElements.get(node.id)\n          : node;\n    assert(searchResultsNode !== undefined);\n    if (searchResultsNode.isSearchTarget) {\n      this.addTarget(searchTargets, searchResultsNode);\n      return;\n    }\n\n    if (searchResultsNode.type === \"element2d\" || searchResultsNode.type === \"element3d\") {\n      // need to add parent ids as search target will be an element\n      this.addTarget(searchTargets, searchResultsNode);\n    }\n\n    if (!node.children) {\n      return;\n    }\n\n    for (const child of node.children.values()) {\n      this.collectSearchTargets(searchTargets, child, processedSearchResultsNodes);\n    }\n  }\n\n  private addTarget(searchTargets: SearchTargetsInternal, node: SearchResultsTreeNode) {\n    switch (node.type) {\n      case \"classificationTable\":\n        (searchTargets.classificationTableIds ??= new Set()).add(node.id);\n        return;\n      case \"classification\":\n        (searchTargets.classificationIds ??= new Set()).add(node.id);\n        return;\n      case \"element2d\":\n        const element2dKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements2d = (searchTargets.elements2d ??= new Map()).get(element2dKey);\n        if (elements2d) {\n          elements2d.set(node.id, { isSearchTarget: node.isSearchTarget });\n        } else {\n          searchTargets.elements2d.set(element2dKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));\n        }\n        return;\n      case \"element3d\":\n        const element3dKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements3d = (searchTargets.elements3d ??= new Map()).get(element3dKey);\n        if (elements3d) {\n          elements3d.set(node.id, { isSearchTarget: node.isSearchTarget });\n        } else {\n          searchTargets.elements3d.set(element3dKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));\n        }\n        return;\n    }\n  }\n\n  private createModelCategoryKey(modelId: Id64String, categoryId: Id64String): ModelCategoryKey {\n    return `${modelId}-${categoryId}`;\n  }\n\n  private parseModelCategoryKey(key: ModelCategoryKey): { modelId: Id64String; categoryId: Id64String } {\n    const [modelId, categoryId] = key.split(\"-\");\n    return { modelId, categoryId };\n  }\n\n  public createSearchResultsTreeNode({\n    type,\n    id,\n    isSearchTarget,\n  }: {\n    type: SearchResultsTreeNode[\"type\"];\n    id: Id64String;\n    isSearchTarget: boolean;\n  }): TemporarySearchResultsTreeNode {\n    if (type === \"element2d\" || type === \"element3d\") {\n      return {\n        id,\n        isSearchTarget,\n        type,\n        modelId: undefined,\n        categoryId: undefined,\n      };\n    }\n    return {\n      id,\n      isSearchTarget,\n      type,\n    };\n  }\n\n  public async getType(className: string): Promise<TemporarySearchResultsTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_ClassificationTable)) {\n      return \"classificationTable\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Classification)) {\n      return \"classification\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_GeometricElement2d)) {\n      return \"element2d\";\n    }\n    return \"element3d\";\n  }\n}\n"]}
+{"version":3,"file":"SearchResultsTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/SearchResultsTree.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,8BAA8B,EAAE,6BAA6B,EAAE,MAAM,kDAAkD,CAAC;AAC5J,OAAO,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,8DAA8D,CAAC;AAuDlI,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,sCAAsC,CAAC,KAI5D;IACC,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACtD,OAAO,uBAAuB,CAAC;QAC7B,yBAAyB,EAAE,IAAI,4CAA4C,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACvG,WAAW;KACZ,CAAC,CAAC;AACL,CAAC;AA2BD,MAAM,4CAA6C,SAAQ,yBAI1D;IACU,MAAM,CAAoD;IACnE,YAAY,KAAwD;QAClE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,8BAA8B;QACzC,MAAM,8BAA8B,GAAG,IAAI,GAAG,EAAmE,CAAC;QAClH,MAAM,MAAM,GAAgC;YAC1C,qBAAqB,EAAE,IAAI,GAAG,EAAE;SACjC,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC9C,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC5B,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;QAED,MAAM,2BAA2B,GAAG,MAAM,cAAc,CACtD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YAC3C,UAAU,EAAE,CAAC,GAAG,8BAA8B,CAAC,IAAI,EAAE,CAAC;SACvD,CAAC,CACH,CAAC;QACF,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,8BAA8B,EAAE,CAAC;YAC3D,MAAM,KAAK,GAAG,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC5B,MAAM,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,EAAE;gBACnC,GAAG,OAAO;gBACV,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,8BAA8B,EAAE,KAAK,CAAC,8BAA8B;aACrE,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,2BAA2B,CAChC,kBAAoD,EACpD,2BAAwD;QAExD,MAAM,aAAa,GAA0B,EAAE,CAAC;QAEhD,kBAAkB,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,CAAC;QAE5I,OAAO,IAAI,CAAC,4BAA4B,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC;IAEO,8CAA8C,CACpD,6BAA4D,EAC5D,WAA0B;QAE1B,MAAM,MAAM,GAA2D,EAAE,CAAC;QAC1E,2GAA2G;QAC3G,KAAK,MAAM,CAAC,kBAAkB,EAAE,KAAK,CAAC,IAAI,6BAA6B,EAAE,CAAC;YACxE,MAAM,UAAU,GAAG,IAAI,CAAC,iEAAiE,CAAC,kBAAkB,CAAC,CAAC;YAC9G,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;gBAChC,KAAK,MAAM,CAAC,gBAAgB,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,8BAA8B,EAAE,CAAC,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;oBACxH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,MAAM,CAAC,IAAI,CAAC;wBACV,cAAc,EAAE,CAAC,GAAG,WAAW,EAAE,UAAU,CAAC;wBAC5C,OAAO;wBACP,UAAU;wBACV,QAAQ;wBACR,8BAA8B;wBAC9B,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;qBACrD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACpB,SAAS;YACX,CAAC;YACD,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,8CAA8C,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,WAAW,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC;gBAC3H,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,4BAA4B,CAAC,aAAoC;QACvE,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;YACpG,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,iBAAiB,EAAE,aAAa,CAAC,iBAAiB;YAClD,sBAAsB,EAAE,aAAa,CAAC,iBAAiB;YACvD,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,8CAA8C,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;SAC/H,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAC1B,aAAoC,EACpC,IAAoC,EACpC,2BAAwD;QAExD,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,2BAA2B,CAAC,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1H,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACxC,IAAI,iBAAiB,CAAC,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QAED,IAAI,iBAAiB,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACzC,6DAA6D;YAC7D,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,KAAK,EAAE,2BAA2B,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,aAAoC,EAAE,IAAkC;QAC/F,uEAAuE;QACvE,yHAAyH;QACzH,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACpF,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;YAC5B,aAAa,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QACrC,CAAC;QACD,MAAM,uBAAuB,GAAG,aAAa,CAAC,QAAQ,CAAC;QACvD,MAAM,CAAC,uBAAuB,KAAK,SAAS,CAAC,CAAC;QAC9C,IAAI,KAAK,GAAG,uBAAuB,CAAC;QACpC,IAAI,sBAA8C,CAAC;QACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YAChD,IAAI,sBAAsB,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBACpE,sBAAsB,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACjD,CAAC;YACD,MAAM,kBAAkB,GAAG,IAAI,CAAC,0CAA0C,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/F,IAAI,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;YACpD,0DAA0D;YAC1D,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,eAAe,GAAG,EAAE,sBAAsB,EAAE,CAAC;gBAC7C,KAAK,CAAC,GAAG,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC;YACjD,CAAC;YACD,qDAAqD;YACrD,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,eAAe,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC;gBACvC,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,CAAC,eAAe,CAAC,qBAAqB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAC7F,wEAAwE;YACxE,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,eAAe,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE;oBAC1D,WAAW,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;oBAC1E,8BAA8B,EAAE,IAAI,CAAC,8BAA8B;iBACpE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,aAAoC,EAAE,IAA2B;QACjF,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,qBAAqB;gBACxB,CAAC,aAAa,CAAC,sBAAsB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClE,OAAO;YACT,KAAK,gBAAgB;gBACnB,CAAC,aAAa,CAAC,iBAAiB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7D,OAAO;YACT,KAAK,SAAS;gBACZ,IAAI,CAAC,gBAAgB,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;gBAC3C,OAAO;QACX,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,OAAmB,EAAE,UAAsB;QACxE,OAAO,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;IACpC,CAAC;IAEO,qBAAqB,CAAC,GAAqB;QACjD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IACjC,CAAC;IAEM,2BAA2B,CAAC,EACjC,IAAI,EACJ,EAAE,EACF,cAAc,EACd,MAAM,GAMP;QACC,MAAM,UAAU,GAAG,YAAY,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9G,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,SAAS;gBAClB,UAAU,EAAE,SAAS;gBACrB,8BAA8B,EAAE,SAAS;gBACzC,UAAU;aACX,CAAC;QACJ,CAAC;QACD,OAAO;YACL,EAAE;YACF,cAAc;YACd,IAAI;YACJ,UAAU;SACX,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,8BAA8B,CAAC,EAAE,CAAC;YAC/F,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,yBAAyB,CAAC,EAAE,CAAC;YAC1F,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,YAAY,CAAC,IAA4C;QAC9D,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,qBAAqB;gBACxB,OAAO,8BAA8B,CAAC;YACxC,KAAK,gBAAgB;gBACnB,OAAO,yBAAyB,CAAC;YACnC,KAAK,SAAS;gBACZ,OAAO,6BAA6B,CAAC;QACzC,CAAC;IACH,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 { firstValueFrom } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement3d } from \"../../../common/internal/ClassNameDefinitions.js\";\nimport { createSearchResultsTree, SearchResultsNodesHandler } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\n\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type {\n  BaseSearchResultsTreeNode,\n  SearchResultsNodeIdentifierAsString,\n  SearchResultsTree,\n  SearchResultsTreeNodeChildren,\n  SearchResultsTreeRootNode,\n} from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\ninterface ClassificationTableSearchResultsTreeNode extends BaseSearchResultsTreeNode<ClassificationTableSearchResultsTreeNode> {\n  type: \"classificationTable\";\n}\n\ninterface ClassificationSearchResultsTreeNode extends BaseSearchResultsTreeNode<ClassificationSearchResultsTreeNode> {\n  type: \"classification\";\n}\n\ninterface ElementSearchResultsTreeNode extends BaseSearchResultsTreeNode<ElementSearchResultsTreeNode> {\n  type: \"element\";\n  categoryId: Id64String;\n  modelId: Id64String;\n  categoryOfTopMostParentElement: CategoryId;\n}\n\ntype SearchResultsTreeNode = ClassificationTableSearchResultsTreeNode | ClassificationSearchResultsTreeNode | ElementSearchResultsTreeNode;\n\ntype TemporaryElementSearchResultsNode = Omit<ElementSearchResultsTreeNode, \"modelId\" | \"categoryId\" | \"categoryOfTopMostParentElement\" | \"children\"> & {\n  modelId: Id64String | undefined;\n  categoryId: Id64String | undefined;\n  categoryOfTopMostParentElement: CategoryId | undefined;\n  children?: SearchResultsTreeNodeChildren<TemporaryElementSearchResultsNode>;\n};\n\ntype TemporarySearchResultsTreeNode = ClassificationTableSearchResultsTreeNode | ClassificationSearchResultsTreeNode | TemporaryElementSearchResultsNode;\n\n/** @internal */\nexport interface ClassificationsTreeSearchTargets {\n  elements?: Array<{\n    pathToElements: InstanceKey[];\n    modelId: Id64String;\n    categoryId: Id64String;\n    topMostParentElementId?: Id64String;\n    elements: Map<ElementId, { isSearchTarget: boolean }>;\n    categoryOfTopMostParentElement: CategoryId;\n  }>;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\n/** @internal */\nexport async function createClassificationsSearchResultsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n  idsCache: ClassificationsTreeIdsCache;\n}): Promise<SearchResultsTree<ClassificationsTreeSearchTargets>> {\n  const { imodelAccess, searchPaths, idsCache } = props;\n  return createSearchResultsTree({\n    searchResultsNodesHandler: new ClassificationsTreeSearchResultsNodesHandler({ idsCache, imodelAccess }),\n    searchPaths,\n  });\n}\n\ntype SearchTargetsInternalElements = Map<\n  SearchResultsNodeIdentifierAsString,\n  {\n    children?: SearchTargetsInternalElements;\n    topMostParentElementId?: Id64String;\n    modelCategoryElements?: Map<ModelCategoryKey, { elementsMap: Map<ElementId, { isSearchTarget: boolean }>; categoryOfTopMostParentElement: CategoryId }>;\n  }\n>;\ninterface SearchTargetsInternal {\n  elements?: SearchTargetsInternalElements;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\ninterface ClassificationsTreeSearchResultsNodesHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\ninterface ProcessedSearchResultsNodes {\n  searchResultsElements: Map<Id64String, Omit<ElementSearchResultsTreeNode, \"children\">>;\n}\n\nclass ClassificationsTreeSearchResultsNodesHandler extends SearchResultsNodesHandler<\n  ProcessedSearchResultsNodes,\n  ClassificationsTreeSearchTargets,\n  TemporarySearchResultsTreeNode\n> {\n  readonly #props: ClassificationsTreeSearchResultsNodesHandlerProps;\n  constructor(props: ClassificationsTreeSearchResultsNodesHandlerProps) {\n    super();\n    this.#props = props;\n  }\n\n  public async getProcessedSearchResultsNodes(): Promise<ProcessedSearchResultsNodes> {\n    const searchResultsTemporaryElements = new Map<Id64String, Omit<TemporaryElementSearchResultsNode, \"children\">>();\n    const result: ProcessedSearchResultsNodes = {\n      searchResultsElements: new Map(),\n    };\n    for (const node of this.searchResultsNodesArr) {\n      if (node.type === \"element\") {\n        searchResultsTemporaryElements.set(node.id, node);\n      }\n    }\n\n    const searchResultsElementsModels = await firstValueFrom(\n      this.#props.idsCache.getFilteredElementsData({\n        elementIds: [...searchResultsTemporaryElements.keys()],\n      }),\n    );\n    for (const [id, element] of searchResultsTemporaryElements) {\n      const entry = searchResultsElementsModels.get(element.id);\n      assert(entry !== undefined);\n      result.searchResultsElements.set(id, {\n        ...element,\n        modelId: entry.modelId,\n        categoryId: entry.categoryId,\n        categoryOfTopMostParentElement: entry.categoryOfTopMostParentElement,\n      });\n    }\n    return result;\n  }\n\n  public convertNodesToSearchTargets(\n    searchResultsNodes: TemporarySearchResultsTreeNode[],\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ): ClassificationsTreeSearchTargets | undefined {\n    const searchTargets: SearchTargetsInternal = {};\n\n    searchResultsNodes.forEach((searchResultsNode) => this.collectSearchTargets(searchTargets, searchResultsNode, processedSearchResultsNodes));\n\n    return this.convertInternalSearchTargets(searchTargets);\n  }\n\n  private convertInternalSearchTargetElementsRecursively(\n    searchTargetsInternalElements: SearchTargetsInternalElements,\n    currentPath: InstanceKey[],\n  ): Required<ClassificationsTreeSearchTargets>[\"elements\"] {\n    const result: Required<ClassificationsTreeSearchTargets>[\"elements\"] = [];\n    // Internal search target elements are stored in a tree structure, need to convert that to array structure.\n    for (const [identifierAsString, entry] of searchTargetsInternalElements) {\n      const identifier = this.convertSearchResultsNodeIdentifierStringToHierarchyNodeIdentifier(identifierAsString);\n      if (entry.modelCategoryElements) {\n        for (const [modelCategoryKey, { elementsMap: elements, categoryOfTopMostParentElement }] of entry.modelCategoryElements) {\n          const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n          result.push({\n            pathToElements: [...currentPath, identifier],\n            modelId,\n            categoryId,\n            elements,\n            categoryOfTopMostParentElement,\n            topMostParentElementId: entry.topMostParentElementId,\n          });\n        }\n      }\n      if (!entry.children) {\n        continue;\n      }\n      for (const childValue of this.convertInternalSearchTargetElementsRecursively(entry.children, [...currentPath, identifier])) {\n        result.push(childValue);\n      }\n    }\n    return result;\n  }\n\n  private convertInternalSearchTargets(searchTargets: SearchTargetsInternal): ClassificationsTreeSearchTargets | undefined {\n    if (!searchTargets.classificationIds && !searchTargets.classificationIds && !searchTargets.elements) {\n      return undefined;\n    }\n\n    return {\n      classificationIds: searchTargets.classificationIds,\n      classificationTableIds: searchTargets.classificationIds,\n      elements: searchTargets.elements ? this.convertInternalSearchTargetElementsRecursively(searchTargets.elements, []) : undefined,\n    };\n  }\n\n  private collectSearchTargets(\n    searchTargets: SearchTargetsInternal,\n    node: TemporarySearchResultsTreeNode,\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ) {\n    const searchResultsNode = node.type === \"element\" ? processedSearchResultsNodes.searchResultsElements.get(node.id) : node;\n    assert(searchResultsNode !== undefined);\n    if (searchResultsNode.isSearchTarget) {\n      this.addTarget(searchTargets, searchResultsNode);\n      return;\n    }\n\n    if (searchResultsNode.type === \"element\") {\n      // need to add parent ids as search target will be an element\n      this.addTarget(searchTargets, searchResultsNode);\n    }\n\n    if (!node.children) {\n      return;\n    }\n\n    for (const child of node.children.values()) {\n      this.collectSearchTargets(searchTargets, child, processedSearchResultsNodes);\n    }\n  }\n\n  private addTargetElement(searchTargets: SearchTargetsInternal, node: ElementSearchResultsTreeNode) {\n    // Internal search target elements need to have path saved in some way.\n    // For this, a tree structure is used, where keys are stringified identifiers of parent nodes depending on the hierarchy.\n    const modelCategoryKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n    if (!searchTargets.elements) {\n      searchTargets.elements = new Map();\n    }\n    const searchTargetElementsMap = searchTargets.elements;\n    assert(searchTargetElementsMap !== undefined);\n    let entry = searchTargetElementsMap;\n    let topMostParentElementId: Id64String | undefined;\n    for (let i = 0; i < node.pathToNode.length; ++i) {\n      if (topMostParentElementId === undefined && node.type === \"element\") {\n        topMostParentElementId = node.pathToNode[i].id;\n      }\n      const identifierAsString = this.convertSearchResultsNodeIdentifierToString(node.pathToNode[i]);\n      let identifierEntry = entry.get(identifierAsString);\n      // create a new entry for parent node if it does not exist\n      if (!identifierEntry) {\n        identifierEntry = { topMostParentElementId };\n        entry.set(identifierAsString, identifierEntry);\n      }\n      // last entry in the path don't need to have children\n      if (i < node.pathToNode.length - 1) {\n        identifierEntry.children ??= new Map();\n        entry = identifierEntry.children;\n        continue;\n      }\n\n      const elements = (identifierEntry.modelCategoryElements ??= new Map()).get(modelCategoryKey);\n      // Add elements who share the same path to the modelCategoryElements map\n      if (elements) {\n        elements.set(node.id, { isSearchTarget: node.isSearchTarget });\n      } else {\n        identifierEntry.modelCategoryElements.set(modelCategoryKey, {\n          elementsMap: new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]),\n          categoryOfTopMostParentElement: node.categoryOfTopMostParentElement,\n        });\n      }\n    }\n  }\n\n  private addTarget(searchTargets: SearchTargetsInternal, node: SearchResultsTreeNode) {\n    switch (node.type) {\n      case \"classificationTable\":\n        (searchTargets.classificationTableIds ??= new Set()).add(node.id);\n        return;\n      case \"classification\":\n        (searchTargets.classificationIds ??= new Set()).add(node.id);\n        return;\n      case \"element\":\n        this.addTargetElement(searchTargets, node);\n        return;\n    }\n  }\n\n  private createModelCategoryKey(modelId: Id64String, categoryId: Id64String): ModelCategoryKey {\n    return `${modelId}-${categoryId}`;\n  }\n\n  private parseModelCategoryKey(key: ModelCategoryKey): { modelId: Id64String; categoryId: Id64String } {\n    const [modelId, categoryId] = key.split(\"-\");\n    return { modelId, categoryId };\n  }\n\n  public createSearchResultsTreeNode({\n    type,\n    id,\n    isSearchTarget,\n    parent,\n  }: {\n    type: SearchResultsTreeNode[\"type\"];\n    id: Id64String;\n    isSearchTarget: boolean;\n    parent: SearchResultsTreeNode | SearchResultsTreeRootNode<SearchResultsTreeNode>;\n  }): TemporarySearchResultsTreeNode {\n    const pathToNode = \"pathToNode\" in parent ? [...parent.pathToNode, { type: parent.type, id: parent.id }] : [];\n    if (type === \"element\") {\n      return {\n        id,\n        isSearchTarget,\n        type,\n        modelId: undefined,\n        categoryId: undefined,\n        categoryOfTopMostParentElement: undefined,\n        pathToNode,\n      };\n    }\n    return {\n      id,\n      isSearchTarget,\n      type,\n      pathToNode,\n    };\n  }\n\n  public async getType(className: string): Promise<TemporarySearchResultsTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_ClassificationTable)) {\n      return \"classificationTable\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Classification)) {\n      return \"classification\";\n    }\n    return \"element\";\n  }\n\n  public getClassName(type: TemporarySearchResultsTreeNode[\"type\"]): string {\n    switch (type) {\n      case \"classificationTable\":\n        return CLASS_NAME_ClassificationTable;\n      case \"classification\":\n        return CLASS_NAME_Classification;\n      case \"element\":\n        return CLASS_NAME_GeometricElement3d;\n    }\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/SharedTreeContextProvider.d.ts

@@ -0,0 +1,7 @@
+import type { PropsWithChildren } from "react";
+/**
+ * A React context provider for setting up shared tree context.
+ * @beta
+ */
+export declare function SharedTreeContextProvider({ children }: PropsWithChildren<{}>): import("react/jsx-runtime").JSX.Element;
+//# sourceMappingURL=SharedTreeContextProvider.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/SharedTreeContextProvider.js

@@ -0,0 +1,14 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { SharedTreeContextProviderInternal } from "./internal/SharedTreeContextProviderInternal.js";
+/**
+ * A React context provider for setting up shared tree context.
+ * @beta
+ */
+export function SharedTreeContextProvider({ children }) {
+    return _jsx(SharedTreeContextProviderInternal, { children: children });
+}
+//# sourceMappingURL=SharedTreeContextProvider.js.map

package/lib/esm/tree-widget-react/components/trees/common/SharedTreeContextProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SharedTreeContextProvider.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/SharedTreeContextProvider.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,iCAAiC,EAAE,MAAM,iDAAiD,CAAC;AAIpG;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CAAC,EAAE,QAAQ,EAAyB;IAC3E,OAAO,KAAC,iCAAiC,cAAE,QAAQ,GAAqC,CAAC;AAC3F,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { SharedTreeContextProviderInternal } from \"./internal/SharedTreeContextProviderInternal.js\";\n\nimport type { PropsWithChildren } from \"react\";\n\n/**\n * A React context provider for setting up shared tree context.\n * @beta\n */\nexport function SharedTreeContextProvider({ children }: PropsWithChildren<{}>) {\n  return <SharedTreeContextProviderInternal>{children}</SharedTreeContextProviderInternal>;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/TreeWidgetViewport.js

@@ -24,7 +24,7 @@ export function createTreeWidgetViewport(viewport) {
             viewport.changeModelDisplay(props.modelIds, false);
         },
         viewsCategory: (categoryId) => viewport.view.viewsCategory(categoryId),
-        changeCategoryDisplay: (props) => viewport.changeCategoryDisplay(props.categoryIds, props.display, props.enableAllSubCategories),
+        changeCategoryDisplay: (props) => viewport.changeCategoryDisplay(props.categoryIds, props.display, props.enableAllSubCategories, true),
         viewsSubCategory: (subCategoryId) => viewport.isSubCategoryVisible(subCategoryId),
         changeSubCategoryDisplay: (props) => viewport.changeSubCategoryDisplay(props.subCategoryId, props.display),
         get perModelCategoryOverrides() {

package/lib/esm/tree-widget-react/components/trees/common/TreeWidgetViewport.js.map

@@ -1 +1 @@
-{"version":3,"file":"TreeWidgetViewport.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/TreeWidgetViewport.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAQlE;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAkB;IACzD,OAAO;QACL,IAAI,QAAQ;YACV,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;QACtF,CAAC;QACD,IAAI,MAAM;YACR,OAAO,QAAQ,CAAC,MAAM,CAAC;QACzB,CAAC;QACD,UAAU,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QAC1D,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,KAAK,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC9C,OAAO;YACT,CAAC;YACD,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC;QACD,aAAa,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;QACtE,qBAAqB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,sBAAsB,CAAC;QAChI,gBAAgB,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,aAAa,CAAC;QACjF,wBAAwB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC;QAC1G,IAAI,yBAAyB;YAC3B,OAAO,QAAQ,CAAC,0BAA0B,CAAC;QAC7C,CAAC;QACD,2BAA2B,EAAE,CAAC,KAAK,EAAE,EAAE;YACrC,MAAM,QAAQ,GACZ,KAAK,CAAC,QAAQ,KAAK,MAAM;gBACvB,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI;gBAC1C,CAAC,CAAC,KAAK,CAAC,QAAQ,KAAK,MAAM;oBACzB,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI;oBAC1C,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjD,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC/F,CAAC;QACD,2BAA2B,EAAE,CAAC,KAAK,EAAE,EAAE;YACrC,MAAM,QAAQ,GAAG,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YAClG,OAAO,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAClJ,CAAC;QACD,8BAA8B,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC,cAAc,CAAC,KAAK,EAAE,QAAQ,CAAC;QAC9G,IAAI,UAAU;YACZ,OAAO,QAAQ,CAAC,UAAU,CAAC;QAC7B,CAAC;QACD,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,CAAC;QAClE,eAAe,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,eAAe,EAAE;QACjD,IAAI,WAAW;YACb,OAAO,QAAQ,CAAC,WAAW,CAAC;QAC9B,CAAC;QACD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,SAAS,CAAC;QACrF,gBAAgB,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,EAAE;QACnD,IAAI,sBAAsB;YACxB,OAAO,QAAQ,CAAC,sBAAsB,CAAC;QACzC,CAAC;QACD,IAAI,oBAAoB;YACtB,OAAO,QAAQ,CAAC,oBAAoB,CAAC;QACvC,CAAC;QACD,IAAI,mBAAmB;YACrB,OAAO,QAAQ,CAAC,mBAAmB,CAAC;QACtC,CAAC;QACD,IAAI,qBAAqB;YACvB,OAAO,QAAQ,CAAC,qBAAqB,CAAC;QACxC,CAAC;QACD,IAAI,wBAAwB;YAC1B,OAAO,QAAQ,CAAC,qBAAqB,CAAC;QACxC,CAAC;QACD,IAAI,4BAA4B;YAC9B,OAAO,QAAQ,CAAC,yBAAyB,CAAC;QAC5C,CAAC;QACD,IAAI,oCAAoC;YACtC,OAAO,QAAQ,CAAC,iCAAiC,CAAC;QACpD,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\n\nimport type { BeEvent, Id64Arg, Id64String } from \"@itwin/core-bentley\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\n\n/** @public */\nexport type PerModelCategoryOverride = \"show\" | \"hide\" | \"none\";\n\n/**\n * Creates a `TreeWidgetViewport`.\n * @beta\n */\nexport function createTreeWidgetViewport(viewport: Viewport): TreeWidgetViewport {\n  return {\n    get viewType() {\n      return viewport.view.is2d() ? \"2d\" : viewport.view.isSpatialView() ? \"3d\" : \"other\";\n    },\n    get iModel() {\n      return viewport.iModel;\n    },\n    viewsModel: (modelId) => viewport.view.viewsModel(modelId),\n    changeModelDisplay: (props) => {\n      if (props.display) {\n        void viewport.addViewedModels(props.modelIds);\n        return;\n      }\n      viewport.changeModelDisplay(props.modelIds, false);\n    },\n    viewsCategory: (categoryId) => viewport.view.viewsCategory(categoryId),\n    changeCategoryDisplay: (props) => viewport.changeCategoryDisplay(props.categoryIds, props.display, props.enableAllSubCategories),\n    viewsSubCategory: (subCategoryId) => viewport.isSubCategoryVisible(subCategoryId),\n    changeSubCategoryDisplay: (props) => viewport.changeSubCategoryDisplay(props.subCategoryId, props.display),\n    get perModelCategoryOverrides() {\n      return viewport.perModelCategoryVisibility;\n    },\n    setPerModelCategoryOverride: (props) => {\n      const override =\n        props.override === \"show\"\n          ? PerModelCategoryVisibility.Override.Show\n          : props.override === \"hide\"\n            ? PerModelCategoryVisibility.Override.Hide\n            : PerModelCategoryVisibility.Override.None;\n      viewport.perModelCategoryVisibility.setOverride(props.modelIds, props.categoryIds, override);\n    },\n    getPerModelCategoryOverride: (props) => {\n      const override = viewport.perModelCategoryVisibility.getOverride(props.modelId, props.categoryId);\n      return override === PerModelCategoryVisibility.Override.Show ? \"show\" : override === PerModelCategoryVisibility.Override.Hide ? \"hide\" : \"none\";\n    },\n    clearPerModelCategoryOverrides: (props) => viewport.perModelCategoryVisibility.clearOverrides(props?.modelIds),\n    get neverDrawn() {\n      return viewport.neverDrawn;\n    },\n    setNeverDrawn: (props) => viewport.setNeverDrawn(props.elementIds),\n    clearNeverDrawn: () => viewport.clearNeverDrawn(),\n    get alwaysDrawn() {\n      return viewport.alwaysDrawn;\n    },\n    setAlwaysDrawn: (props) => viewport.setAlwaysDrawn(props.elementIds, props.exclusive),\n    clearAlwaysDrawn: () => viewport.clearAlwaysDrawn(),\n    get isAlwaysDrawnExclusive() {\n      return viewport.isAlwaysDrawnExclusive;\n    },\n    get onAlwaysDrawnChanged() {\n      return viewport.onAlwaysDrawnChanged;\n    },\n    get onNeverDrawnChanged() {\n      return viewport.onNeverDrawnChanged;\n    },\n    get onDisplayStyleChanged() {\n      return viewport.onDisplayStyleChanged;\n    },\n    get onDisplayedModelsChanged() {\n      return viewport.onViewedModelsChanged;\n    },\n    get onDisplayedCategoriesChanged() {\n      return viewport.onViewedCategoriesChanged;\n    },\n    get onPerModelCategoriesOverridesChanged() {\n      return viewport.onViewedCategoriesPerModelChanged;\n    },\n  };\n}\n\n/**\n * A simplified interface for a Viewport that is used by the Tree Widget to control and determine visibility.\n *\n * Viewport should only display elements. Since elements have model, category and sub-category assigned to them, each of these can affect element visibility.\n *\n * The order of precedence for visibility is:\n * 1. Model visibility - if model is not visible, elements from that model should never be displayed.\n * 2. `neverDrawn` set - elements in that set should never be displayed.\n * 3. `alwaysDrawn` set - elements in that set should always be displayed.\n * 4. Per-model category visibility overrides:\n * -  if a per-model-category has `Hide` override, elements which have that category and model should not be displayed.\n * -  if a per-model-category has `Show` override, elements which have that category and model should be displayed.\n * 5. Category and sub-category visibility - if element's category or sub-category is turned off, it should not be displayed.\n *\n * Based on this order of precedence, element can only be displayed in these scenarios:\n * - Model is visible *AND* element is in `alwaysDrawn` set.\n * - Model is visible *AND* element is not in `neverDrawn` set *AND* per-model category override is set to show `Show`.\n * - Model is visible *AND* element is not in `neverDrawn` set *AND* per-model category override is not set to `Hide` *AND* category and sub-category are visible.\n * @public\n */\nexport interface TreeWidgetViewport {\n  /**\n   * The type of the view. Generally it should be either `3d` or `2d`.\n   *\n   * The `other` value is reserved for edge cases when viewport doesn't show any elements, e.g. a blank viewport.\n   */\n  viewType: \"2d\" | \"3d\" | \"other\";\n  /** The iModel of this Viewport. */\n  iModel: IModelConnection;\n  /**\n   * Should return true if model specified by `modelId` is visible in the viewport.\n   *\n   * Model display should not take into account its' child elements display.\n   *\n   * When model is not displayed, all elements that have that model should not be shown in the viewport.\n   */\n  viewsModel: (modelId: Id64String) => boolean;\n  /**\n   * Should add or remove a set of models from those models currently displayed in this viewport.\n   *\n   * **NOTE** When turning model display to:\n   * - `true`, model display should be turned on. This does not mean that any elements should be turned on, it only means that\n   * model visibility should not interfere with elements visibility.\n   * - `false`, model display should be turned off. All elements which have that model should not be displayed in the viewport.\n   */\n  changeModelDisplay: (props: { modelIds: Id64Arg; display: boolean }) => void;\n  /**\n   * Should return true if category specified by `categoryId` is visible in the viewport.\n   *\n   * Category display should not take into account its' child elements or sub-categories display.\n   *\n   * When category is not displayed, all elements and sub-categories which have that category should not be shown in the viewport.\n   *\n   * **NOTE** Elements might still be visible if per-model category `Show` override or `alwaysDrawn` set makes them visible.\n   */\n  viewsCategory: (categoryId: Id64String) => boolean;\n  /**\n   * Should enable or disable display of elements belonging to a set of categories specified by Id.\n   *\n   * Visibility of individual sub-categories belonging to a category can be controlled separately through the use of `changeSubCategoryDisplay`.\n   *\n   * By default, enabling display of a category does not affect display of sub-categories thereof which have been set to be invisible.\n   *\n   * Changing category visibility should not change per-model category overrides.\n   *\n   * **NOTE** When turning category display to:\n   * - `true`, category display should be turned on. All elements which have specified category should be turned on if model is visible,\n   * category does not have per-model `hide` override and element is not in the `neverDrawn` set.\n   * - `false`, category and its' sub-categories display should be turned off. All elements which have that category should not be displayed in the viewport,\n   * unless element has per-model category `Show` override, or is in the `alwaysDrawn` set.\n   */\n  changeCategoryDisplay: (props: { categoryIds: Id64Arg; display: boolean; enableAllSubCategories?: boolean }) => void;\n  /**\n   * Should return true if sub-category specified by `subCategoryId` is visible in the viewport.\n   *\n   * Sub-category display should not take into account its' child elements.\n   *\n   * When sub-category is not displayed, all elements that have that sub-category should not be shown in the viewport, unless they have per-model category `Show` override or are in the `alwaysDrawn` list.\n   */\n  viewsSubCategory: (subCategoryId: Id64String) => boolean;\n  /**\n   * Should enable or disable display of elements belonging to sub-category specified by Id.\n   *\n   * Changing sub-category display should not change category display.\n   *\n   * **NOTE** When turning sub-category display to:\n   * - `true`, sub-category display should be turned on. All elements which have specified sub-category should be turned on if model and category are visible,\n   * category does not have per-model `hide` override and element is not in the `neverDrawn` set.\n   * - `false`, sub-category display should be turned off. All elements that have that sub-category should not be displayed in the viewport,\n   * unless element has per-model category `Show` override, or is in the `alwaysDrawn` set.\n   */\n  changeSubCategoryDisplay: (props: { subCategoryId: Id64String; display: boolean }) => void;\n  /** Returns an iterable of per-model-category overrides. */\n  perModelCategoryOverrides: Readonly<Iterable<{ modelId: Id64String; categoryId: Id64String; visible: boolean }>>;\n  /** Should change the override state of one or more categories for one or more models. */\n  setPerModelCategoryOverride: (props: { modelIds: Id64Arg; categoryIds: Id64Arg; override: PerModelCategoryOverride }) => void;\n  /** Should return the per-model category override. */\n  getPerModelCategoryOverride: (props: { modelId: Id64String; categoryId: Id64String }) => PerModelCategoryOverride;\n  /** Should remove per-model category override for the specified models. */\n  clearPerModelCategoryOverrides: (props?: { modelIds?: Id64Arg }) => void;\n  /**\n   * Ids of elements which should not be displayed.\n   *\n   * This set takes precedence over category and sub-category display, per-model category overrides, and the `alwaysDrawn` set.\n   */\n  neverDrawn: ReadonlySet<Id64String> | undefined;\n  /** Ids of elements which should not be displayed in the viewport, regardless of category and sub-category visibility. */\n  setNeverDrawn: (props: { elementIds: Set<Id64String> }) => void;\n  /** Should clear the set of never-drawn elements. */\n  clearNeverDrawn: () => void;\n  /**\n   * Ids of elements which should be displayed.\n   *\n   * Elements in this set should be displayed only when model is displayed as well.\n   *\n   * This set takes precedence over category and sub-category display and per-model category override.\n   */\n  alwaysDrawn: ReadonlySet<Id64String> | undefined;\n  /**\n   * Ids of elements which should be displayed in the viewport, regardless of category and sub-category visibility.\n   *\n   * These elements should be displayed only when their models are displayed as well.\n   *\n   * When the `exclusive` prop is set to `true`, only the specified elements should be drawn.\n   */\n  setAlwaysDrawn: (props: { elementIds: Set<Id64String>; exclusive?: boolean }) => void;\n  /** Should clear the set of always-drawn elements. */\n  clearAlwaysDrawn: () => void;\n  /**\n   * Should return true if elements in the [[alwaysDrawn]] set are the *only* elements that are displayed when their models are visible.\n   *\n   * Should be set to true when `setAlwaysDrawn` is called with `exclusive` flag. Should be set to false\n   * when `setAlwaysDrawn` is called with `exclusive` set to `false` or `undefined`, or when `clearAlwaysDrawn` is called.\n   */\n  readonly isAlwaysDrawnExclusive: boolean;\n  /** Event that should be raised when set of always-drawn elements changes. */\n  onAlwaysDrawnChanged: BeEvent<() => void>;\n  /** Event that should be raised when set of never-drawn elements changes. */\n  onNeverDrawnChanged: BeEvent<() => void>;\n  /** Event that should be raised when per-model category overrides changes. */\n  onPerModelCategoriesOverridesChanged: BeEvent<() => void>;\n  /** Event that should be raised when category display changes. */\n  onDisplayedCategoriesChanged: BeEvent<() => void>;\n  /** Event that should be raised when model display changes. */\n  onDisplayedModelsChanged: BeEvent<() => void>;\n  /** Event that should be raised when view changes between `2d`, `3d` or `other`. */\n  onDisplayStyleChanged: BeEvent<() => void>;\n}\n"]}
+{"version":3,"file":"TreeWidgetViewport.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/TreeWidgetViewport.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAQlE;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAkB;IACzD,OAAO;QACL,IAAI,QAAQ;YACV,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;QACtF,CAAC;QACD,IAAI,MAAM;YACR,OAAO,QAAQ,CAAC,MAAM,CAAC;QACzB,CAAC;QACD,UAAU,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QAC1D,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,KAAK,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC9C,OAAO;YACT,CAAC;YACD,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC;QACD,aAAa,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;QACtE,qBAAqB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,sBAAsB,EAAE,IAAI,CAAC;QACtI,gBAAgB,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,aAAa,CAAC;QACjF,wBAAwB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC;QAC1G,IAAI,yBAAyB;YAC3B,OAAO,QAAQ,CAAC,0BAA0B,CAAC;QAC7C,CAAC;QACD,2BAA2B,EAAE,CAAC,KAAK,EAAE,EAAE;YACrC,MAAM,QAAQ,GACZ,KAAK,CAAC,QAAQ,KAAK,MAAM;gBACvB,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI;gBAC1C,CAAC,CAAC,KAAK,CAAC,QAAQ,KAAK,MAAM;oBACzB,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI;oBAC1C,CAAC,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjD,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC/F,CAAC;QACD,2BAA2B,EAAE,CAAC,KAAK,EAAE,EAAE;YACrC,MAAM,QAAQ,GAAG,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YAClG,OAAO,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,KAAK,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAClJ,CAAC;QACD,8BAA8B,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC,cAAc,CAAC,KAAK,EAAE,QAAQ,CAAC;QAC9G,IAAI,UAAU;YACZ,OAAO,QAAQ,CAAC,UAAU,CAAC;QAC7B,CAAC;QACD,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,CAAC;QAClE,eAAe,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,eAAe,EAAE;QACjD,IAAI,WAAW;YACb,OAAO,QAAQ,CAAC,WAAW,CAAC;QAC9B,CAAC;QACD,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,SAAS,CAAC;QACrF,gBAAgB,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,EAAE;QACnD,IAAI,sBAAsB;YACxB,OAAO,QAAQ,CAAC,sBAAsB,CAAC;QACzC,CAAC;QACD,IAAI,oBAAoB;YACtB,OAAO,QAAQ,CAAC,oBAAoB,CAAC;QACvC,CAAC;QACD,IAAI,mBAAmB;YACrB,OAAO,QAAQ,CAAC,mBAAmB,CAAC;QACtC,CAAC;QACD,IAAI,qBAAqB;YACvB,OAAO,QAAQ,CAAC,qBAAqB,CAAC;QACxC,CAAC;QACD,IAAI,wBAAwB;YAC1B,OAAO,QAAQ,CAAC,qBAAqB,CAAC;QACxC,CAAC;QACD,IAAI,4BAA4B;YAC9B,OAAO,QAAQ,CAAC,yBAAyB,CAAC;QAC5C,CAAC;QACD,IAAI,oCAAoC;YACtC,OAAO,QAAQ,CAAC,iCAAiC,CAAC;QACpD,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\n\nimport type { BeEvent, Id64Arg, Id64String } from \"@itwin/core-bentley\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\n\n/** @public */\nexport type PerModelCategoryOverride = \"show\" | \"hide\" | \"none\";\n\n/**\n * Creates a `TreeWidgetViewport`.\n * @beta\n */\nexport function createTreeWidgetViewport(viewport: Viewport): TreeWidgetViewport {\n  return {\n    get viewType() {\n      return viewport.view.is2d() ? \"2d\" : viewport.view.isSpatialView() ? \"3d\" : \"other\";\n    },\n    get iModel() {\n      return viewport.iModel;\n    },\n    viewsModel: (modelId) => viewport.view.viewsModel(modelId),\n    changeModelDisplay: (props) => {\n      if (props.display) {\n        void viewport.addViewedModels(props.modelIds);\n        return;\n      }\n      viewport.changeModelDisplay(props.modelIds, false);\n    },\n    viewsCategory: (categoryId) => viewport.view.viewsCategory(categoryId),\n    changeCategoryDisplay: (props) => viewport.changeCategoryDisplay(props.categoryIds, props.display, props.enableAllSubCategories, true),\n    viewsSubCategory: (subCategoryId) => viewport.isSubCategoryVisible(subCategoryId),\n    changeSubCategoryDisplay: (props) => viewport.changeSubCategoryDisplay(props.subCategoryId, props.display),\n    get perModelCategoryOverrides() {\n      return viewport.perModelCategoryVisibility;\n    },\n    setPerModelCategoryOverride: (props) => {\n      const override =\n        props.override === \"show\"\n          ? PerModelCategoryVisibility.Override.Show\n          : props.override === \"hide\"\n            ? PerModelCategoryVisibility.Override.Hide\n            : PerModelCategoryVisibility.Override.None;\n      viewport.perModelCategoryVisibility.setOverride(props.modelIds, props.categoryIds, override);\n    },\n    getPerModelCategoryOverride: (props) => {\n      const override = viewport.perModelCategoryVisibility.getOverride(props.modelId, props.categoryId);\n      return override === PerModelCategoryVisibility.Override.Show ? \"show\" : override === PerModelCategoryVisibility.Override.Hide ? \"hide\" : \"none\";\n    },\n    clearPerModelCategoryOverrides: (props) => viewport.perModelCategoryVisibility.clearOverrides(props?.modelIds),\n    get neverDrawn() {\n      return viewport.neverDrawn;\n    },\n    setNeverDrawn: (props) => viewport.setNeverDrawn(props.elementIds),\n    clearNeverDrawn: () => viewport.clearNeverDrawn(),\n    get alwaysDrawn() {\n      return viewport.alwaysDrawn;\n    },\n    setAlwaysDrawn: (props) => viewport.setAlwaysDrawn(props.elementIds, props.exclusive),\n    clearAlwaysDrawn: () => viewport.clearAlwaysDrawn(),\n    get isAlwaysDrawnExclusive() {\n      return viewport.isAlwaysDrawnExclusive;\n    },\n    get onAlwaysDrawnChanged() {\n      return viewport.onAlwaysDrawnChanged;\n    },\n    get onNeverDrawnChanged() {\n      return viewport.onNeverDrawnChanged;\n    },\n    get onDisplayStyleChanged() {\n      return viewport.onDisplayStyleChanged;\n    },\n    get onDisplayedModelsChanged() {\n      return viewport.onViewedModelsChanged;\n    },\n    get onDisplayedCategoriesChanged() {\n      return viewport.onViewedCategoriesChanged;\n    },\n    get onPerModelCategoriesOverridesChanged() {\n      return viewport.onViewedCategoriesPerModelChanged;\n    },\n  };\n}\n\n/**\n * A simplified interface for a Viewport that is used by the Tree Widget to control and determine visibility.\n *\n * Viewport should only display elements. Since elements have model, category and sub-category assigned to them, each of these can affect element visibility.\n *\n * The order of precedence for visibility is:\n * 1. Model visibility - if model is not visible, elements from that model should never be displayed.\n * 2. `neverDrawn` set - elements in that set should never be displayed.\n * 3. `alwaysDrawn` set - elements in that set should always be displayed.\n * 4. Per-model category visibility overrides:\n * -  if a per-model-category has `Hide` override, elements which have that category and model should not be displayed.\n * -  if a per-model-category has `Show` override, elements which have that category and model should be displayed.\n * 5. Category and sub-category visibility - if element's category or sub-category is turned off, it should not be displayed.\n *\n * Based on this order of precedence, element can only be displayed in these scenarios:\n * - Model is visible *AND* element is in `alwaysDrawn` set.\n * - Model is visible *AND* element is not in `neverDrawn` set *AND* per-model category override is set to show `Show`.\n * - Model is visible *AND* element is not in `neverDrawn` set *AND* per-model category override is not set to `Hide` *AND* category and sub-category are visible.\n * @public\n */\nexport interface TreeWidgetViewport {\n  /**\n   * The type of the view. Generally it should be either `3d` or `2d`.\n   *\n   * The `other` value is reserved for edge cases when viewport doesn't show any elements, e.g. a blank viewport.\n   */\n  viewType: \"2d\" | \"3d\" | \"other\";\n  /** The iModel of this Viewport. */\n  iModel: IModelConnection;\n  /**\n   * Should return true if model specified by `modelId` is visible in the viewport.\n   *\n   * Model display should not take into account its' child elements display.\n   *\n   * When model is not displayed, all elements that have that model should not be shown in the viewport.\n   */\n  viewsModel: (modelId: Id64String) => boolean;\n  /**\n   * Should add or remove a set of models from those models currently displayed in this viewport.\n   *\n   * **NOTE** When turning model display to:\n   * - `true`, model display should be turned on. This does not mean that any elements should be turned on, it only means that\n   * model visibility should not interfere with elements visibility.\n   * - `false`, model display should be turned off. All elements which have that model should not be displayed in the viewport.\n   */\n  changeModelDisplay: (props: { modelIds: Id64Arg; display: boolean }) => void;\n  /**\n   * Should return true if category specified by `categoryId` is visible in the viewport.\n   *\n   * Category display should not take into account its' child elements or sub-categories display.\n   *\n   * When category is not displayed, all elements and sub-categories which have that category should not be shown in the viewport.\n   *\n   * **NOTE** Elements might still be visible if per-model category `Show` override or `alwaysDrawn` set makes them visible.\n   */\n  viewsCategory: (categoryId: Id64String) => boolean;\n  /**\n   * Should enable or disable display of elements belonging to a set of categories specified by Id.\n   *\n   * Visibility of individual sub-categories belonging to a category can be controlled separately through the use of `changeSubCategoryDisplay`.\n   *\n   * By default, enabling display of a category does not affect display of sub-categories thereof which have been set to be invisible.\n   *\n   * Changing category visibility should not change per-model category overrides.\n   *\n   * **NOTE** When turning category display to:\n   * - `true`, category display should be turned on. All elements which have specified category should be turned on if model is visible,\n   * category does not have per-model `hide` override and element is not in the `neverDrawn` set.\n   * - `false`, category and its' sub-categories display should be turned off. All elements which have that category should not be displayed in the viewport,\n   * unless element has per-model category `Show` override, or is in the `alwaysDrawn` set.\n   */\n  changeCategoryDisplay: (props: { categoryIds: Id64Arg; display: boolean; enableAllSubCategories?: boolean }) => void;\n  /**\n   * Should return true if sub-category specified by `subCategoryId` is visible in the viewport.\n   *\n   * Sub-category display should not take into account its' child elements.\n   *\n   * When sub-category is not displayed, all elements that have that sub-category should not be shown in the viewport, unless they have per-model category `Show` override or are in the `alwaysDrawn` list.\n   */\n  viewsSubCategory: (subCategoryId: Id64String) => boolean;\n  /**\n   * Should enable or disable display of elements belonging to sub-category specified by Id.\n   *\n   * Changing sub-category display should not change category display.\n   *\n   * **NOTE** When turning sub-category display to:\n   * - `true`, sub-category display should be turned on. All elements which have specified sub-category should be turned on if model and category are visible,\n   * category does not have per-model `hide` override and element is not in the `neverDrawn` set.\n   * - `false`, sub-category display should be turned off. All elements that have that sub-category should not be displayed in the viewport,\n   * unless element has per-model category `Show` override, or is in the `alwaysDrawn` set.\n   */\n  changeSubCategoryDisplay: (props: { subCategoryId: Id64String; display: boolean }) => void;\n  /** Returns an iterable of per-model-category overrides. */\n  perModelCategoryOverrides: Readonly<Iterable<{ modelId: Id64String; categoryId: Id64String; visible: boolean }>>;\n  /** Should change the override state of one or more categories for one or more models. */\n  setPerModelCategoryOverride: (props: { modelIds: Id64Arg; categoryIds: Id64Arg; override: PerModelCategoryOverride }) => void;\n  /** Should return the per-model category override. */\n  getPerModelCategoryOverride: (props: { modelId: Id64String; categoryId: Id64String }) => PerModelCategoryOverride;\n  /** Should remove per-model category override for the specified models. */\n  clearPerModelCategoryOverrides: (props?: { modelIds?: Id64Arg }) => void;\n  /**\n   * Ids of elements which should not be displayed.\n   *\n   * This set takes precedence over category and sub-category display, per-model category overrides, and the `alwaysDrawn` set.\n   */\n  neverDrawn: ReadonlySet<Id64String> | undefined;\n  /** Ids of elements which should not be displayed in the viewport, regardless of category and sub-category visibility. */\n  setNeverDrawn: (props: { elementIds: Set<Id64String> }) => void;\n  /** Should clear the set of never-drawn elements. */\n  clearNeverDrawn: () => void;\n  /**\n   * Ids of elements which should be displayed.\n   *\n   * Elements in this set should be displayed only when model is displayed as well.\n   *\n   * This set takes precedence over category and sub-category display and per-model category override.\n   */\n  alwaysDrawn: ReadonlySet<Id64String> | undefined;\n  /**\n   * Ids of elements which should be displayed in the viewport, regardless of category and sub-category visibility.\n   *\n   * These elements should be displayed only when their models are displayed as well.\n   *\n   * When the `exclusive` prop is set to `true`, only the specified elements should be drawn.\n   */\n  setAlwaysDrawn: (props: { elementIds: Set<Id64String>; exclusive?: boolean }) => void;\n  /** Should clear the set of always-drawn elements. */\n  clearAlwaysDrawn: () => void;\n  /**\n   * Should return true if elements in the [[alwaysDrawn]] set are the *only* elements that are displayed when their models are visible.\n   *\n   * Should be set to true when `setAlwaysDrawn` is called with `exclusive` flag. Should be set to false\n   * when `setAlwaysDrawn` is called with `exclusive` set to `false` or `undefined`, or when `clearAlwaysDrawn` is called.\n   */\n  readonly isAlwaysDrawnExclusive: boolean;\n  /** Event that should be raised when set of always-drawn elements changes. */\n  onAlwaysDrawnChanged: BeEvent<() => void>;\n  /** Event that should be raised when set of never-drawn elements changes. */\n  onNeverDrawnChanged: BeEvent<() => void>;\n  /** Event that should be raised when per-model category overrides changes. */\n  onPerModelCategoriesOverridesChanged: BeEvent<() => void>;\n  /** Event that should be raised when category display changes. */\n  onDisplayedCategoriesChanged: BeEvent<() => void>;\n  /** Event that should be raised when model display changes. */\n  onDisplayedModelsChanged: BeEvent<() => void>;\n  /** Event that should be raised when view changes between `2d`, `3d` or `other`. */\n  onDisplayStyleChanged: BeEvent<() => void>;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/UseHierarchyVisibility.js

@@ -26,9 +26,9 @@ export function useHierarchyVisibility({ visibilityHandlerFactory }) {
             calculate.next(node);
         };
         const resetCache = () => {
-            visibilityStatusMap.current.forEach((value) => {
+            for (const value of visibilityStatusMap.current.values()) {
                 value.needsRefresh = true;
-            });
+            }
             visibilityChanged.next();
         };
         const triggerCheckboxUpdate = () => {

package/lib/esm/tree-widget-react/components/trees/common/UseHierarchyVisibility.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,YAAY,EAAE,MAAM,MAAM,CAAC;AAC3J,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAoC/D,gBAAgB;AAChB,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,mBAAmB,GAAG,MAAM,CAAC,IAAI,GAAG,EAA6G,CAAC,CAAC;IACzJ,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAqD;QACrF,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QACrD,uBAAuB,EAAE,GAAG,EAAE,GAAE,CAAC;QACjC,cAAc,EAAE,GAAG,EAAE,GAAE,CAAC;KACzB,CAAC,CAAC;IACH,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAChD,MAAM,aAAa,GAAG,aAAa,EAAE,CAAC;IAEtC,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,wBAAwB,EAAE,CAAC;QAE3C,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAQ,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,OAAO,EAA6B,CAAC;QAC3D,MAAM,mBAAmB,GAAG,CAAC,IAA+B,EAAE,EAAE;YAC9D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,GAAG,EAAE;YACtB,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC5C,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YACH,iBAAiB,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,wBAAwB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;aACtF,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,SAAS,CAAC,cAAc,CAAC,EACzB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAChB,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChE,GAAG,CAAC;YACF,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvC,IAAI;oBACJ,MAAM,EAAE;wBACN,GAAG,MAAM;wBACT,OAAO,EAAE,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC;qBACrC;oBACD,YAAY,EAAE,KAAK;iBACpB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,EACF,SAAS,CAAC,iBAAiB,CAAC,EAC5B,UAAU,CAAC,CAAC,KAAK,EAAE,EAAE;YACnB,aAAa,CAAC,KAAK,CAAC,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CACH,CACF,EACD,YAAY,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CACjE;aACA,SAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE;gBACT,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAiD,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE;YAC/F,aAAa,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,gGAAgG;YAChG,MAAM,EAAE,GAAG,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YACxD,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,IAAI,CAAC;oBACH,MAAM,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACpD,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;YACZ,CAAC,CAAC,EAAE,CAAC;YACL,MAAM,KAAK,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,KAAK,CAAC,MAAM,GAAG;gBACb,GAAG,KAAK,CAAC,MAAM;gBACf,KAAK,EAAE,eAAe;gBACtB,OAAO,EAAE,aAAa,CAAC,aAAa,CAAC;aACtC,CAAC;YACF,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,uBAAuB,EAAE,gBAAgB;YACzC,wBAAwB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YACrF,cAAc,EAAE,GAAG,EAAE;gBACnB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,UAAU,EAAE,CAAC;YACb,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,YAAY,CAAC,WAAW,EAAE,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC;IAE7D,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CACxB,GAAqI,EACrI,mBAA8D;IAE9D,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC;AA0BD,gBAAgB;AAChB,MAAM,OAAO,kCAAkC;IAC7C,YAAY,CAA6B;IACzC,YAAY,WAAuC;QACjD,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;IAClC,CAAC;IACM,6BAA6B,CAA0B,KAI7D;QACC,MAAM,EAAE,mBAAmB,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAC/D,OAAO,QAAQ;YACb,CAAC,CAAC,IAAI,CACF,QAAQ,CAAC;gBACP,GAAG,aAAa;gBAChB,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC,mBAAmB,EAAE,EAAE,YAAY,EAAE,SAAoB,EAAE,CAAC;gBAC9G,OAAO,EAAE,IAAI,CAAC,YAAY;aAC3B,CAAC,CACH;YACH,CAAC,CAAC,mBAAmB,CAAC;IAC1B,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 { useEffect, useRef, useState } from \"react\";\nimport { asyncScheduler, catchError, defer, distinct, EMPTY, from, lastValueFrom, mergeMap, observeOn, Subject, takeUntil, tap, throttleTime } from \"rxjs\";\nimport { createTooltip } from \"./internal/Tooltip.js\";\nimport { useErrorState } from \"./internal/UseErrorState.js\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { MutableRefObject } from \"react\";\nimport type { Observable } from \"rxjs\";\nimport type { BeEvent } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeItemVisibilityButtonState, VisibilityContext } from \"./components/TreeNodeVisibilityButton.js\";\n\n/**\n * Data structure that describes instance visibility status.\n * @beta\n */\nexport interface VisibilityStatus {\n  /** Instance visibility state. */\n  state: \"visible\" | \"partial\" | \"hidden\";\n  /** Specifies whether visibility changing is disabled or not. */\n  isDisabled?: boolean;\n}\n\n/**\n * Handler that can be used to determine and change visibility of instances represented by tree nodes.\n * @beta\n */\nexport interface HierarchyVisibilityHandler extends Disposable {\n  /** Event used to notify tree about visibility changes from outside. */\n  readonly onVisibilityChange: BeEvent<() => void>;\n  /** Returns current visibility status for tree node. */\n  getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;\n  /** Changes visibility of the instance represented by tree node. */\n  changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\n/** @internal */\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): VisibilityContext & { triggerRefresh: () => void } {\n  const visibilityStatusMap = useRef(new Map<string, { node: PresentationHierarchyNode; status: TreeItemVisibilityButtonState; needsRefresh: boolean }>());\n  const [state, setState] = useState<VisibilityContext & { triggerRefresh: () => void }>({\n    getVisibilityButtonState: () => ({ isLoading: true }),\n    onVisibilityButtonClick: () => {},\n    triggerRefresh: () => {},\n  });\n  const { onFeatureUsed } = useTelemetryContext();\n  const setErrorState = useErrorState();\n\n  useEffect(() => {\n    visibilityStatusMap.current.clear();\n    const handler = visibilityHandlerFactory();\n\n    const visibilityChanged = new Subject<void>();\n    const calculate = new Subject<PresentationHierarchyNode>();\n    const calculateNodeStatus = (node: PresentationHierarchyNode) => {\n      calculate.next(node);\n    };\n\n    const resetCache = () => {\n      visibilityStatusMap.current.forEach((value) => {\n        value.needsRefresh = true;\n      });\n      visibilityChanged.next();\n    };\n\n    const triggerCheckboxUpdate = () => {\n      setState((prev) => ({\n        ...prev,\n        getVisibilityButtonState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      }));\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        observeOn(asyncScheduler),\n        mergeMap((node) =>\n          defer(async () => handler.getVisibilityStatus(node.nodeData)).pipe(\n            tap({\n              next: (status) => {\n                visibilityStatusMap.current.set(node.id, {\n                  node,\n                  status: {\n                    ...status,\n                    tooltip: createTooltip(status.state),\n                  },\n                  needsRefresh: false,\n                });\n              },\n            }),\n            takeUntil(visibilityChanged),\n            catchError((error) => {\n              setErrorState(error);\n              return EMPTY;\n            }),\n          ),\n        ),\n        throttleTime(100, undefined, { leading: false, trailing: true }),\n      )\n      .subscribe({\n        next: () => {\n          triggerCheckboxUpdate();\n        },\n      });\n\n    const changeVisibility: VisibilityContext[\"onVisibilityButtonClick\"] = (node, visibilityState) => {\n      onFeatureUsed({ featureId: \"visibility-change\", reportInteraction: true });\n      // visible should become hidden, partial and hidden should become visible TODO: redo for clarity\n      const on = visibilityState === \"visible\" ? false : true;\n      void (async () => {\n        try {\n          await handler.changeVisibility(node.nodeData, on);\n        } catch {}\n      })();\n      const entry = visibilityStatusMap.current.get(node.id);\n      if (!entry) {\n        return;\n      }\n      entry.status = {\n        ...entry.status,\n        state: visibilityState,\n        tooltip: createTooltip(\"determining\"),\n      };\n      triggerCheckboxUpdate();\n    };\n\n    setState({\n      onVisibilityButtonClick: changeVisibility,\n      getVisibilityButtonState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      triggerRefresh: () => {\n        resetCache();\n        triggerCheckboxUpdate();\n      },\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      resetCache();\n      triggerCheckboxUpdate();\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler[Symbol.dispose]();\n    };\n  }, [visibilityHandlerFactory, onFeatureUsed, setErrorState]);\n\n  return state;\n}\n\nfunction createStateGetter(\n  map: MutableRefObject<Map<string, { node: PresentationHierarchyNode; status: TreeItemVisibilityButtonState; needsRefresh: boolean }>>,\n  calculateVisibility: (node: PresentationHierarchyNode) => void,\n): VisibilityContext[\"getVisibilityButtonState\"] {\n  return (node) => {\n    const entry = map.current.get(node.id);\n    if (entry === undefined) {\n      calculateVisibility(node);\n      return { isLoading: true };\n    }\n\n    if (entry.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    return entry.status;\n  };\n}\n\n/**\n * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult\n  ? TProps & {\n      /** A callback that produces the value from the original implementation. */\n      readonly originalImplementation: () => TResult;\n      /**\n       * Reference to the hierarchy based handler.\n       * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.\n       */\n      readonly handler: HierarchyVisibilityHandler;\n    }\n  : never;\n\n/**\n * Function type for an overridden method of `HierarchyVisibilityHandler`.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult\n  ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult\n  : never;\n\n/** @internal */\nexport class HierarchyVisibilityOverrideHandler {\n  #baseHandler: HierarchyVisibilityHandler;\n  constructor(baseHandler: HierarchyVisibilityHandler) {\n    this.#baseHandler = baseHandler;\n  }\n  public createVisibilityHandlerResult<TResult, TOverrideProps>(props: {\n    nonOverriddenResult: Observable<TResult>;\n    override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined;\n    overrideProps: TOverrideProps;\n  }): Observable<TResult> {\n    const { nonOverriddenResult, override, overrideProps } = props;\n    return override\n      ? from(\n          override({\n            ...overrideProps,\n            originalImplementation: async () => lastValueFrom(nonOverriddenResult, { defaultValue: undefined as TResult }),\n            handler: this.#baseHandler,\n          }),\n        )\n      : nonOverriddenResult;\n  }\n}\n"]}
+{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,YAAY,EAAE,MAAM,MAAM,CAAC;AAC3J,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAoC/D,gBAAgB;AAChB,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,mBAAmB,GAAG,MAAM,CAAC,IAAI,GAAG,EAA4F,CAAC,CAAC;IACxI,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAqD;QACrF,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QACrD,uBAAuB,EAAE,GAAG,EAAE,GAAE,CAAC;QACjC,cAAc,EAAE,GAAG,EAAE,GAAE,CAAC;KACzB,CAAC,CAAC;IACH,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAChD,MAAM,aAAa,GAAG,aAAa,EAAE,CAAC;IAEtC,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,wBAAwB,EAAE,CAAC;QAE3C,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAQ,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,OAAO,EAAY,CAAC;QAC1C,MAAM,mBAAmB,GAAG,CAAC,IAAc,EAAE,EAAE;YAC7C,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,GAAG,EAAE;YACtB,KAAK,MAAM,KAAK,IAAI,mBAAmB,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzD,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC;YACD,iBAAiB,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,wBAAwB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;aACtF,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,SAAS,CAAC,cAAc,CAAC,EACzB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAChB,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChE,GAAG,CAAC;YACF,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvC,IAAI;oBACJ,MAAM,EAAE;wBACN,GAAG,MAAM;wBACT,OAAO,EAAE,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC;qBACrC;oBACD,YAAY,EAAE,KAAK;iBACpB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,EACF,SAAS,CAAC,iBAAiB,CAAC,EAC5B,UAAU,CAAC,CAAC,KAAK,EAAE,EAAE;YACnB,aAAa,CAAC,KAAK,CAAC,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CACH,CACF,EACD,YAAY,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CACjE;aACA,SAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE;gBACT,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAiD,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE;YAC/F,aAAa,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,gGAAgG;YAChG,MAAM,EAAE,GAAG,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YACxD,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,IAAI,CAAC;oBACH,MAAM,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACpD,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;YACZ,CAAC,CAAC,EAAE,CAAC;YACL,MAAM,KAAK,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,KAAK,CAAC,MAAM,GAAG;gBACb,GAAG,KAAK,CAAC,MAAM;gBACf,KAAK,EAAE,eAAe;gBACtB,OAAO,EAAE,aAAa,CAAC,aAAa,CAAC;aACtC,CAAC;YACF,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,uBAAuB,EAAE,gBAAgB;YACzC,wBAAwB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YACrF,cAAc,EAAE,GAAG,EAAE;gBACnB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,UAAU,EAAE,CAAC;YACb,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,YAAY,CAAC,WAAW,EAAE,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC;IAE7D,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CACxB,GAAoH,EACpH,mBAA6C;IAE7C,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC;AA0BD,gBAAgB;AAChB,MAAM,OAAO,kCAAkC;IAC7C,YAAY,CAA6B;IACzC,YAAY,WAAuC;QACjD,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;IAClC,CAAC;IACM,6BAA6B,CAA0B,KAI7D;QACC,MAAM,EAAE,mBAAmB,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAC/D,OAAO,QAAQ;YACb,CAAC,CAAC,IAAI,CACF,QAAQ,CAAC;gBACP,GAAG,aAAa;gBAChB,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC,mBAAmB,EAAE,EAAE,YAAY,EAAE,SAAoB,EAAE,CAAC;gBAC9G,OAAO,EAAE,IAAI,CAAC,YAAY;aAC3B,CAAC,CACH;YACH,CAAC,CAAC,mBAAmB,CAAC;IAC1B,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 { useEffect, useRef, useState } from \"react\";\nimport { asyncScheduler, catchError, defer, distinct, EMPTY, from, lastValueFrom, mergeMap, observeOn, Subject, takeUntil, tap, throttleTime } from \"rxjs\";\nimport { createTooltip } from \"./internal/Tooltip.js\";\nimport { useErrorState } from \"./internal/UseErrorState.js\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { MutableRefObject } from \"react\";\nimport type { Observable } from \"rxjs\";\nimport type { BeEvent } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, TreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeItemVisibilityButtonState, VisibilityContext } from \"./components/TreeNodeVisibilityButton.js\";\n\n/**\n * Data structure that describes instance visibility status.\n * @beta\n */\nexport interface VisibilityStatus {\n  /** Instance visibility state. */\n  state: \"visible\" | \"partial\" | \"hidden\";\n  /** Specifies whether visibility changing is disabled or not. */\n  isDisabled?: boolean;\n}\n\n/**\n * Handler that can be used to determine and change visibility of instances represented by tree nodes.\n * @beta\n */\nexport interface HierarchyVisibilityHandler extends Disposable {\n  /** Event used to notify tree about visibility changes from outside. */\n  readonly onVisibilityChange: BeEvent<() => void>;\n  /** Returns current visibility status for tree node. */\n  getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;\n  /** Changes visibility of the instance represented by tree node. */\n  changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\n/** @internal */\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): VisibilityContext & { triggerRefresh: () => void } {\n  const visibilityStatusMap = useRef(new Map<string, { node: TreeNode; status: TreeItemVisibilityButtonState; needsRefresh: boolean }>());\n  const [state, setState] = useState<VisibilityContext & { triggerRefresh: () => void }>({\n    getVisibilityButtonState: () => ({ isLoading: true }),\n    onVisibilityButtonClick: () => {},\n    triggerRefresh: () => {},\n  });\n  const { onFeatureUsed } = useTelemetryContext();\n  const setErrorState = useErrorState();\n\n  useEffect(() => {\n    visibilityStatusMap.current.clear();\n    const handler = visibilityHandlerFactory();\n\n    const visibilityChanged = new Subject<void>();\n    const calculate = new Subject<TreeNode>();\n    const calculateNodeStatus = (node: TreeNode) => {\n      calculate.next(node);\n    };\n\n    const resetCache = () => {\n      for (const value of visibilityStatusMap.current.values()) {\n        value.needsRefresh = true;\n      }\n      visibilityChanged.next();\n    };\n\n    const triggerCheckboxUpdate = () => {\n      setState((prev) => ({\n        ...prev,\n        getVisibilityButtonState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      }));\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        observeOn(asyncScheduler),\n        mergeMap((node) =>\n          defer(async () => handler.getVisibilityStatus(node.nodeData)).pipe(\n            tap({\n              next: (status) => {\n                visibilityStatusMap.current.set(node.id, {\n                  node,\n                  status: {\n                    ...status,\n                    tooltip: createTooltip(status.state),\n                  },\n                  needsRefresh: false,\n                });\n              },\n            }),\n            takeUntil(visibilityChanged),\n            catchError((error) => {\n              setErrorState(error);\n              return EMPTY;\n            }),\n          ),\n        ),\n        throttleTime(100, undefined, { leading: false, trailing: true }),\n      )\n      .subscribe({\n        next: () => {\n          triggerCheckboxUpdate();\n        },\n      });\n\n    const changeVisibility: VisibilityContext[\"onVisibilityButtonClick\"] = (node, visibilityState) => {\n      onFeatureUsed({ featureId: \"visibility-change\", reportInteraction: true });\n      // visible should become hidden, partial and hidden should become visible TODO: redo for clarity\n      const on = visibilityState === \"visible\" ? false : true;\n      void (async () => {\n        try {\n          await handler.changeVisibility(node.nodeData, on);\n        } catch {}\n      })();\n      const entry = visibilityStatusMap.current.get(node.id);\n      if (!entry) {\n        return;\n      }\n      entry.status = {\n        ...entry.status,\n        state: visibilityState,\n        tooltip: createTooltip(\"determining\"),\n      };\n      triggerCheckboxUpdate();\n    };\n\n    setState({\n      onVisibilityButtonClick: changeVisibility,\n      getVisibilityButtonState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      triggerRefresh: () => {\n        resetCache();\n        triggerCheckboxUpdate();\n      },\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      resetCache();\n      triggerCheckboxUpdate();\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler[Symbol.dispose]();\n    };\n  }, [visibilityHandlerFactory, onFeatureUsed, setErrorState]);\n\n  return state;\n}\n\nfunction createStateGetter(\n  map: MutableRefObject<Map<string, { node: TreeNode; status: TreeItemVisibilityButtonState; needsRefresh: boolean }>>,\n  calculateVisibility: (node: TreeNode) => void,\n): VisibilityContext[\"getVisibilityButtonState\"] {\n  return (node) => {\n    const entry = map.current.get(node.id);\n    if (entry === undefined) {\n      calculateVisibility(node);\n      return { isLoading: true };\n    }\n\n    if (entry.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    return entry.status;\n  };\n}\n\n/**\n * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult\n  ? TProps & {\n      /** A callback that produces the value from the original implementation. */\n      readonly originalImplementation: () => TResult;\n      /**\n       * Reference to the hierarchy based handler.\n       * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.\n       */\n      readonly handler: HierarchyVisibilityHandler;\n    }\n  : never;\n\n/**\n * Function type for an overridden method of `HierarchyVisibilityHandler`.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult\n  ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult\n  : never;\n\n/** @internal */\nexport class HierarchyVisibilityOverrideHandler {\n  #baseHandler: HierarchyVisibilityHandler;\n  constructor(baseHandler: HierarchyVisibilityHandler) {\n    this.#baseHandler = baseHandler;\n  }\n  public createVisibilityHandlerResult<TResult, TOverrideProps>(props: {\n    nonOverriddenResult: Observable<TResult>;\n    override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined;\n    overrideProps: TOverrideProps;\n  }): Observable<TResult> {\n    const { nonOverriddenResult, override, overrideProps } = props;\n    return override\n      ? from(\n          override({\n            ...overrideProps,\n            originalImplementation: async () => lastValueFrom(nonOverriddenResult, { defaultValue: undefined as TResult }),\n            handler: this.#baseHandler,\n          }),\n        )\n      : nonOverriddenResult;\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/Utils.js

@@ -78,7 +78,7 @@ export function areAllModelsVisible(models, viewport) {
 export function joinHierarchySearchPaths(subTreePaths, searchPaths) {
     const result = new Array();
     const searchPathsToIncludeIndexes = new Set();
-    subTreePaths.forEach((subTreePath) => {
+    for (const subTreePath of subTreePaths) {
         let options;
         let addSubTreePathToResult = false;
         for (let i = 0; i < searchPaths.length; ++i) {
@@ -115,7 +115,7 @@ export function joinHierarchySearchPaths(subTreePaths, searchPaths) {
                 options,
             });
         }
-    });
+    }
     for (const index of searchPathsToIncludeIndexes) {
         result.push(searchPaths[index]);
     }

package/lib/esm/tree-widget-react/components/trees/common/Utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/Utils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC/F,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAMlG;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,YAAY,CAAC;AAK9C;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,MAAgB,EAAE,QAA4B;IAC1E,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AACpE,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,KAO7B;IACC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC5D,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,iBAAiB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC;SAAM,CAAC;QACN,MAAM,aAAa,GAAG,MAAM,0BAA0B,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC9E,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACzE,MAAM,qBAAqB,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;IACnD,CAAC;IACD,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IACjE,QAAQ,CAAC,eAAe,EAAE,CAAC;IAC3B,QAAQ,CAAC,gBAAgB,EAAE,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,MAAiB,EAAE,QAA4B;IAC7E,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;IAChD,MAAM,YAAY,GAAG,IAAI,KAAK,EAAc,CAAC;IAC7C,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACzB,IAAI,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IACH,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1E,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,MAAgB,EAAE,MAAe,EAAE,QAA4B;IAC1F,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;IACT,CAAC;IACD,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;AACrE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAgB,EAAE,QAA4B;IAChF,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AACrF,CAAC;AAKD,gBAAgB;AAChB,MAAM,UAAU,wBAAwB,CACtC,YAA4C,EAC5C,WAA4C;IAE5C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAiC,CAAC;IAC1D,MAAM,2BAA2B,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtD,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;QACnC,IAAI,OAA+C,CAAC;QACpD,IAAI,sBAAsB,GAAG,KAAK,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC5C,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;gBAClC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,CAAC;oBAClG,MAAM;gBACR,CAAC;gBAED,wGAAwG;gBACxG,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACrC,sBAAsB,GAAG,IAAI,CAAC;oBAC9B,yGAAyG;oBACzG,sCAAsC;oBACtC,OAAO;wBACL,UAAU,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI;4BACjC,CAAC,CAAC,mBAAmB,CAAC,YAAY,CAAC,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC;4BAC/D,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,EAAE,CAAC;oBAC9D,MAAM;gBACR,CAAC;gBAED,iFAAiF;gBACjF,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACjC,sBAAsB,GAAG,IAAI,CAAC;oBAC9B,2BAA2B,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,sBAAsB,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,WAAW;gBACjB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IACH,KAAK,MAAM,KAAK,IAAI,2BAA2B,EAAE,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { HierarchyNodeIdentifier, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport { showAllCategories } from \"./CategoriesVisibilityUtils.js\";\nimport { enableCategoryDisplay, loadCategoriesFromViewport } from \"./internal/VisibilityUtils.js\";\n\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, HierarchySearchPathOptions } from \"@itwin/presentation-hierarchies\";\nimport type { TreeWidgetViewport } from \"./TreeWidgetViewport.js\";\n\n/**\n * This is a logging namespace for public log messages that may be interesting to consumers.\n * @internal\n */\nexport const LOGGING_NAMESPACE = \"TreeWidget\";\n\n/** @beta */\nexport type FunctionProps<THook extends (props: any) => any> = Parameters<THook>[0];\n\n/**\n * Disables display of all given models.\n * @public\n */\nexport function hideAllModels(models: string[], viewport: TreeWidgetViewport) {\n  viewport.changeModelDisplay({ modelIds: models, display: false });\n}\n\n/**\n * Enables display of all given models. Also enables display of all categories and clears always and\n * never drawn lists in the viewport.\n * @public\n */\nexport async function showAll(props: {\n  /** ID's of models to enable */\n  models: Id64Array;\n  /** ID's of categories to enable, if set to undefined, all categories will be enabled */\n  categories?: Id64Array;\n  viewport: TreeWidgetViewport;\n  componentId?: GuidString;\n}) {\n  const { models, categories, viewport, componentId } = props;\n  if (categories) {\n    await showAllCategories(categories, viewport);\n  } else {\n    const categoryInfos = await loadCategoriesFromViewport(viewport, componentId);\n    if (categoryInfos.length === 0) {\n      return;\n    }\n    const ids = categoryInfos.map((categoryInfo) => categoryInfo.categoryId);\n    await enableCategoryDisplay(viewport, ids, true);\n  }\n  viewport.changeModelDisplay({ modelIds: models, display: true });\n  viewport.clearNeverDrawn();\n  viewport.clearAlwaysDrawn();\n}\n\n/**\n * Inverts display of all given models.\n * @public\n */\nexport function invertAllModels(models: Id64Array, viewport: TreeWidgetViewport) {\n  const notViewedModels = new Array<Id64String>();\n  const viewedModels = new Array<Id64String>();\n  models.forEach((modelId) => {\n    if (viewport.viewsModel(modelId)) {\n      viewedModels.push(modelId);\n    } else {\n      notViewedModels.push(modelId);\n    }\n  });\n  viewport.changeModelDisplay({ modelIds: notViewedModels, display: true });\n  viewport.changeModelDisplay({ modelIds: viewedModels, display: false });\n}\n\n/**\n * Based on the value of `enable` argument, either enables or disables display of given models.\n * @public\n */\nexport function toggleModels(models: string[], enable: boolean, viewport: TreeWidgetViewport) {\n  if (!models) {\n    return;\n  }\n  viewport.changeModelDisplay({ modelIds: models, display: enable });\n}\n\n/**\n * Checks if all given models are displayed in given viewport.\n * @public\n */\nexport function areAllModelsVisible(models: string[], viewport: TreeWidgetViewport) {\n  return models.length !== 0 ? models.every((id) => viewport.viewsModel(id)) : false;\n}\n\n/** @public */\nexport type NormalizedHierarchySearchPath = ReturnType<(typeof HierarchySearchPath)[\"normalize\"]>;\n\n/** @internal */\nexport function joinHierarchySearchPaths(\n  subTreePaths: HierarchyNodeIdentifiersPath[],\n  searchPaths: NormalizedHierarchySearchPath[],\n): NormalizedHierarchySearchPath[] {\n  const result = new Array<NormalizedHierarchySearchPath>();\n  const searchPathsToIncludeIndexes = new Set<number>();\n\n  subTreePaths.forEach((subTreePath) => {\n    let options: HierarchySearchPathOptions | undefined;\n    let addSubTreePathToResult = false;\n\n    for (let i = 0; i < searchPaths.length; ++i) {\n      const searchPath = searchPaths[i];\n      if (searchPath.path.length === 0) {\n        continue;\n      }\n\n      for (let j = 0; j < subTreePath.length; ++j) {\n        const identifier = subTreePath[j];\n        if (searchPath.path.length <= j || !HierarchyNodeIdentifier.equal(searchPath.path[j], identifier)) {\n          break;\n        }\n\n        // search paths that are shorter or equal than subTree paths length don't need to be added to the result\n        if (searchPath.path.length === j + 1) {\n          addSubTreePathToResult = true;\n          // If search path has reveal set to true, it means that we should expand only to the targeted search node\n          // This is done by setting depthInPath\n          options =\n            searchPath.options?.reveal !== true\n              ? HierarchySearchPath.mergeOptions(options, searchPath.options)\n              : { reveal: { depthInPath: searchPath.path.length - 1 } };\n          break;\n        }\n\n        // search paths that are longer than subTree paths need to be added to the result\n        if (subTreePath.length === j + 1) {\n          addSubTreePathToResult = true;\n          searchPathsToIncludeIndexes.add(i);\n        }\n      }\n    }\n\n    if (addSubTreePathToResult) {\n      result.push({\n        path: subTreePath,\n        options,\n      });\n    }\n  });\n  for (const index of searchPathsToIncludeIndexes) {\n    result.push(searchPaths[index]);\n  }\n  return result;\n}\n"]}
+{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/Utils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC/F,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAMlG;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,YAAY,CAAC;AAK9C;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,MAAgB,EAAE,QAA4B;IAC1E,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AACpE,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,KAO7B;IACC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC5D,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,iBAAiB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC;SAAM,CAAC;QACN,MAAM,aAAa,GAAG,MAAM,0BAA0B,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC9E,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACzE,MAAM,qBAAqB,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;IACnD,CAAC;IACD,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IACjE,QAAQ,CAAC,eAAe,EAAE,CAAC;IAC3B,QAAQ,CAAC,gBAAgB,EAAE,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,MAAiB,EAAE,QAA4B;IAC7E,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;IAChD,MAAM,YAAY,GAAG,IAAI,KAAK,EAAc,CAAC;IAC7C,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACzB,IAAI,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IACH,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1E,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,MAAgB,EAAE,MAAe,EAAE,QAA4B;IAC1F,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;IACT,CAAC;IACD,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;AACrE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAgB,EAAE,QAA4B;IAChF,OAAO,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AACrF,CAAC;AAKD,gBAAgB;AAChB,MAAM,UAAU,wBAAwB,CACtC,YAA4C,EAC5C,WAA4C;IAE5C,MAAM,MAAM,GAAG,IAAI,KAAK,EAAiC,CAAC;IAC1D,MAAM,2BAA2B,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtD,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACvC,IAAI,OAA+C,CAAC;QACpD,IAAI,sBAAsB,GAAG,KAAK,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC5C,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;gBAClC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,CAAC;oBAClG,MAAM;gBACR,CAAC;gBAED,wGAAwG;gBACxG,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACrC,sBAAsB,GAAG,IAAI,CAAC;oBAC9B,yGAAyG;oBACzG,sCAAsC;oBACtC,OAAO;wBACL,UAAU,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI;4BACjC,CAAC,CAAC,mBAAmB,CAAC,YAAY,CAAC,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC;4BAC/D,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,EAAE,CAAC;oBAC9D,MAAM;gBACR,CAAC;gBAED,iFAAiF;gBACjF,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACjC,sBAAsB,GAAG,IAAI,CAAC;oBAC9B,2BAA2B,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,sBAAsB,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,WAAW;gBACjB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,KAAK,MAAM,KAAK,IAAI,2BAA2B,EAAE,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { HierarchyNodeIdentifier, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport { showAllCategories } from \"./CategoriesVisibilityUtils.js\";\nimport { enableCategoryDisplay, loadCategoriesFromViewport } from \"./internal/VisibilityUtils.js\";\n\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, HierarchySearchPathOptions } from \"@itwin/presentation-hierarchies\";\nimport type { TreeWidgetViewport } from \"./TreeWidgetViewport.js\";\n\n/**\n * This is a logging namespace for public log messages that may be interesting to consumers.\n * @internal\n */\nexport const LOGGING_NAMESPACE = \"TreeWidget\";\n\n/** @beta */\nexport type FunctionProps<THook extends (props: any) => any> = Parameters<THook>[0];\n\n/**\n * Disables display of all given models.\n * @public\n */\nexport function hideAllModels(models: string[], viewport: TreeWidgetViewport) {\n  viewport.changeModelDisplay({ modelIds: models, display: false });\n}\n\n/**\n * Enables display of all given models. Also enables display of all categories and clears always and\n * never drawn lists in the viewport.\n * @public\n */\nexport async function showAll(props: {\n  /** ID's of models to enable */\n  models: Id64Array;\n  /** ID's of categories to enable, if set to undefined, all categories will be enabled */\n  categories?: Id64Array;\n  viewport: TreeWidgetViewport;\n  componentId?: GuidString;\n}) {\n  const { models, categories, viewport, componentId } = props;\n  if (categories) {\n    await showAllCategories(categories, viewport);\n  } else {\n    const categoryInfos = await loadCategoriesFromViewport(viewport, componentId);\n    if (categoryInfos.length === 0) {\n      return;\n    }\n    const ids = categoryInfos.map((categoryInfo) => categoryInfo.categoryId);\n    await enableCategoryDisplay(viewport, ids, true);\n  }\n  viewport.changeModelDisplay({ modelIds: models, display: true });\n  viewport.clearNeverDrawn();\n  viewport.clearAlwaysDrawn();\n}\n\n/**\n * Inverts display of all given models.\n * @public\n */\nexport function invertAllModels(models: Id64Array, viewport: TreeWidgetViewport) {\n  const notViewedModels = new Array<Id64String>();\n  const viewedModels = new Array<Id64String>();\n  models.forEach((modelId) => {\n    if (viewport.viewsModel(modelId)) {\n      viewedModels.push(modelId);\n    } else {\n      notViewedModels.push(modelId);\n    }\n  });\n  viewport.changeModelDisplay({ modelIds: notViewedModels, display: true });\n  viewport.changeModelDisplay({ modelIds: viewedModels, display: false });\n}\n\n/**\n * Based on the value of `enable` argument, either enables or disables display of given models.\n * @public\n */\nexport function toggleModels(models: string[], enable: boolean, viewport: TreeWidgetViewport) {\n  if (!models) {\n    return;\n  }\n  viewport.changeModelDisplay({ modelIds: models, display: enable });\n}\n\n/**\n * Checks if all given models are displayed in given viewport.\n * @public\n */\nexport function areAllModelsVisible(models: string[], viewport: TreeWidgetViewport) {\n  return models.length !== 0 ? models.every((id) => viewport.viewsModel(id)) : false;\n}\n\n/** @public */\nexport type NormalizedHierarchySearchPath = ReturnType<(typeof HierarchySearchPath)[\"normalize\"]>;\n\n/** @internal */\nexport function joinHierarchySearchPaths(\n  subTreePaths: HierarchyNodeIdentifiersPath[],\n  searchPaths: NormalizedHierarchySearchPath[],\n): NormalizedHierarchySearchPath[] {\n  const result = new Array<NormalizedHierarchySearchPath>();\n  const searchPathsToIncludeIndexes = new Set<number>();\n\n  for (const subTreePath of subTreePaths) {\n    let options: HierarchySearchPathOptions | undefined;\n    let addSubTreePathToResult = false;\n\n    for (let i = 0; i < searchPaths.length; ++i) {\n      const searchPath = searchPaths[i];\n      if (searchPath.path.length === 0) {\n        continue;\n      }\n\n      for (let j = 0; j < subTreePath.length; ++j) {\n        const identifier = subTreePath[j];\n        if (searchPath.path.length <= j || !HierarchyNodeIdentifier.equal(searchPath.path[j], identifier)) {\n          break;\n        }\n\n        // search paths that are shorter or equal than subTree paths length don't need to be added to the result\n        if (searchPath.path.length === j + 1) {\n          addSubTreePathToResult = true;\n          // If search path has reveal set to true, it means that we should expand only to the targeted search node\n          // This is done by setting depthInPath\n          options =\n            searchPath.options?.reveal !== true\n              ? HierarchySearchPath.mergeOptions(options, searchPath.options)\n              : { reveal: { depthInPath: searchPath.path.length - 1 } };\n          break;\n        }\n\n        // search paths that are longer than subTree paths need to be added to the result\n        if (subTreePath.length === j + 1) {\n          addSubTreePathToResult = true;\n          searchPathsToIncludeIndexes.add(i);\n        }\n      }\n    }\n\n    if (addSubTreePathToResult) {\n      result.push({\n        path: subTreePath,\n        options,\n      });\n    }\n  }\n  for (const index of searchPathsToIncludeIndexes) {\n    result.push(searchPaths[index]);\n  }\n  return result;\n}\n"]}