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

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHelper.js

@@ -2,9 +2,11 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { concat, EMPTY, from, map, mergeMap, reduce } from "rxjs";
+import { concat, EMPTY, from, map, mergeAll, mergeMap, toArray } from "rxjs";
+import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { getIdsFromChildrenTree, getParentElementsIdsPath } from "../../../common/internal/Utils.js";
 import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
-import { enableCategoryDisplay, mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
+import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
 /**
  * Visibility status helper for categories tree.
  *
@@ -31,18 +33,24 @@ export class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper {
             .pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({
             categoryIds,
             modelId: undefined,
-            type: this.#props.viewport.viewType === "2d" ? "DrawingCategory" : "SpatialCategory",
         })));
     }
     /** Gets grouped elements visibility status. */
     getGroupedElementsVisibilityStatus(props) {
-        const { modelElementsMap, categoryId } = props;
-        return from(modelElementsMap).pipe(mergeMap(([modelId, elementIds]) => this.getElementsVisibilityStatus({
+        const { modelElementsMap, categoryId, topMostParentElementId } = props;
+        return from(modelElementsMap).pipe(mergeMap(([modelId, { elementIds, categoryOfTopMostParentElement }]) => this.getElementsVisibilityStatus({
             elementIds,
             modelId,
             categoryId,
-            type: this.#props.viewport.viewType === "2d" ? "GeometricElement2d" : "GeometricElement3d",
-        })), mergeVisibilityStatuses);
+            parentElementsIdsPath: topMostParentElementId
+                ? getParentElementsIdsPath({
+                    parentInstanceKeys: props.parentKeys.filter((key) => HierarchyNodeKey.isInstances(key)).map((key) => key.instanceKeys),
+                    topMostParentElementId,
+                })
+                : [],
+            childrenCount: props.childrenCount,
+            categoryOfTopMostParentElement,
+        })), mergeVisibilityStatuses());
     }
     /**
      * Changes visibility status of definition containers.
@@ -65,27 +73,38 @@ export class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper {
     changeSubCategoriesVisibilityStatus(props) {
         return concat(
         // make sure parent category and models are enabled
-        props.on
-            ? concat(from(enableCategoryDisplay(this.#props.viewport, props.categoryId, props.on, false)), this.enableCategoriesElementModelsVisibilityStatus(props.categoryId))
-            : EMPTY, from(props.subCategoryIds).pipe(map((subCategoryId) => this.#props.viewport.changeSubCategoryDisplay({ subCategoryId, display: props.on }))));
+        props.on ? this.enableCategoryWithoutEnablingOtherCategories(props.categoryId) : EMPTY, from(props.subCategoryIds).pipe(map((subCategoryId) => this.#props.viewport.changeSubCategoryDisplay({ subCategoryId, display: props.on }))));
     }
     /** Changes grouped elements visibility status. */
     changeGroupedElementsVisibilityStatus(props) {
-        return from(props.modelElementsMap).pipe(mergeMap(([modelId, elementIds]) => {
-            return this.changeElementsVisibilityStatus({ modelId, elementIds, categoryId: props.categoryId, on: props.on });
-        }));
+        const elementIds = new Array();
+        for (const { elementIds: ids } of props.modelElementsMap.values()) {
+            for (const id of ids) {
+                elementIds.push(id);
+            }
+        }
+        return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(map((childrenTree) => getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 })), mergeMap((children) => from(props.modelElementsMap).pipe(mergeMap(([modelId, { elementIds: modelElementIds }]) => {
+            return this.changeElementsVisibilityStatus({ modelId, elementIds: modelElementIds, categoryId: props.categoryId, on: props.on, children });
+        }))));
     }
-    /** Turns on visibility status of models (that are not yet turned on) that are related to categories. */
-    enableCategoriesElementModelsVisibilityStatus(categoryIds) {
-        return this.#props.idsCache.getCategoriesElementModels(categoryIds, true).pipe(reduce((acc, { models }) => {
-            models?.forEach((modelId) => {
-                if (!this.#props.viewport.viewsModel(modelId)) {
-                    acc.add(modelId);
-                }
-            });
-            return acc;
-        }, new Set()), map((hiddenModels) => {
-            if (hiddenModels.size > 0) {
+    /** Turns on category and its' related models. Does not turn on other categories contained in those models.*/
+    enableCategoryWithoutEnablingOtherCategories(categoryId) {
+        this.#props.viewport.changeCategoryDisplay({ categoryIds: categoryId, display: true });
+        return this.#props.idsCache.getModels({ categoryId, includeSubModels: true }).pipe(mergeAll(), mergeMap((modelId) => {
+            this.#props.viewport.setPerModelCategoryOverride({ modelIds: modelId, categoryIds: categoryId, override: "none" });
+            return this.#props.viewport.viewsModel(modelId)
+                ? EMPTY
+                : this.#props.idsCache.getCategories({ modelId }).pipe(map((allModelCategories) => {
+                    // Add 'Hide' override to categories that were hidden before model is turned on
+                    for (const modelCategoryId of allModelCategories) {
+                        if (modelCategoryId !== categoryId) {
+                            this.#props.viewport.setPerModelCategoryOverride({ modelIds: modelId, categoryIds: modelCategoryId, override: "hide" });
+                        }
+                    }
+                    return modelId;
+                }));
+        }), toArray(), map((hiddenModels) => {
+            if (hiddenModels.length > 0) {
                 this.#props.viewport.changeModelDisplay({ modelIds: hiddenModels, display: true });
             }
         }));

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeVisibilityHelper.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAClE,OAAO,EAAE,oBAAoB,EAAE,MAAM,6DAA6D,CAAC;AACnG,OAAO,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAgB7G;;;;;GAKG;AACH,MAAM,OAAO,8BAA+B,SAAQ,oBAAoB;IACtE,MAAM,CAAsC;IAC5C,YAAY,KAA0C;QACpD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,uCAAuC,CAAC,KAA0C;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ;aACxB,yBAAyB,CAAC;YACzB,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;YACpD,sBAAsB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB;SACxE,CAAC;aACD,IAAI,CACH,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CACvB,IAAI,CAAC,6BAA6B,CAAC;YACjC,WAAW;YACX,OAAO,EAAE,SAAS;YAClB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,iBAAiB;SACrF,CAAC,CACH,CACF,CAAC;IACN,CAAC;IAED,+CAA+C;IACxC,kCAAkC,CAAC,KAAiF;QACzH,MAAM,EAAE,gBAAgB,EAAE,UAAU,EAAE,GAAG,KAAK,CAAC;QAC/C,OAAO,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAChC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,EAAE,CACjC,IAAI,CAAC,2BAA2B,CAAC;YAC/B,UAAU;YACV,OAAO;YACP,UAAU;YACV,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,oBAAoB;SAC3F,CAAC,CACH,EACD,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,0CAA0C,CAAC,KAAuD;QACvG,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ;aACxB,yBAAyB,CAAC;YACzB,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;YACpD,sBAAsB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB;SACxE,CAAC;aACD,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/H,CAAC;IAED;;;;OAIG;IACI,mCAAmC,CAAC,KAAuE;QAChH,OAAO,MAAM;QACX,mDAAmD;QACnD,KAAK,CAAC,EAAE;YACN,CAAC,CAAC,MAAM,CACJ,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,EACpF,IAAI,CAAC,6CAA6C,CAAC,KAAK,CAAC,UAAU,CAAC,CACrE;YACH,CAAC,CAAC,KAAK,EACT,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAC7I,CAAC;IACJ,CAAC;IAED,kDAAkD;IAC3C,qCAAqC,CAAC,KAI5C;QACC,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,CACtC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,EAAE;YACjC,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;QAClH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,wGAAwG;IAChG,6CAA6C,CAAC,WAAoB;QACxE,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,IAAI,CAC5E,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YACzB,MAAM,EAAE,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC9C,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,EACzB,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;YACnB,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACrF,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, EMPTY, from, map, mergeMap, reduce } from \"rxjs\";\nimport { BaseVisibilityHelper } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport { enableCategoryDisplay, mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg, Id64String } from \"@itwin/core-bentley\";\nimport type { ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { BaseVisibilityHelperProps } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"../../CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport type CategoriesTreeVisibilityHelperProps = BaseVisibilityHelperProps & {\n  idsCache: CategoriesTreeIdsCache;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n};\n\n/**\n * Visibility status helper for categories tree.\n *\n * It extends base visibility status helper and provides methods to get and change visibility status of definition containers and grouped elements.\n * @internal\n */\nexport class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper {\n  #props: CategoriesTreeVisibilityHelperProps;\n  constructor(props: CategoriesTreeVisibilityHelperProps) {\n    super(props);\n    this.#props = props;\n  }\n\n  /**\n   * Gets visibility status of definition containers.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getDefinitionContainersVisibilityStatus(props: { definitionContainerIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache\n      .getAllContainedCategories({\n        definitionContainerIds: props.definitionContainerIds,\n        includeEmptyCategories: this.#props.hierarchyConfig.showEmptyCategories,\n      })\n      .pipe(\n        mergeMap((categoryIds) =>\n          this.getCategoriesVisibilityStatus({\n            categoryIds,\n            modelId: undefined,\n            type: this.#props.viewport.viewType === \"2d\" ? \"DrawingCategory\" : \"SpatialCategory\",\n          }),\n        ),\n      );\n  }\n\n  /** Gets grouped elements visibility status. */\n  public getGroupedElementsVisibilityStatus(props: { modelElementsMap: Map<ModelId, Set<ElementId>>; categoryId: Id64String }): Observable<VisibilityStatus> {\n    const { modelElementsMap, categoryId } = props;\n    return from(modelElementsMap).pipe(\n      mergeMap(([modelId, elementIds]) =>\n        this.getElementsVisibilityStatus({\n          elementIds,\n          modelId,\n          categoryId,\n          type: this.#props.viewport.viewType === \"2d\" ? \"GeometricElement2d\" : \"GeometricElement3d\",\n        }),\n      ),\n      mergeVisibilityStatuses,\n    );\n  }\n\n  /**\n   * Changes visibility status of definition containers.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeDefinitionContainersVisibilityStatus(props: { definitionContainerIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache\n      .getAllContainedCategories({\n        definitionContainerIds: props.definitionContainerIds,\n        includeEmptyCategories: this.#props.hierarchyConfig.showEmptyCategories,\n      })\n      .pipe(mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ categoryIds, modelId: undefined, on: props.on })));\n  }\n\n  /**\n   * Changes visibility status of sub-categories.\n   *\n   * Also, enables parent categories if `on` is true.\n   */\n  public changeSubCategoriesVisibilityStatus(props: { categoryId: Id64String; subCategoryIds: Id64Arg; on: boolean }): Observable<void> {\n    return concat(\n      // make sure parent category and models are enabled\n      props.on\n        ? concat(\n            from(enableCategoryDisplay(this.#props.viewport, props.categoryId, props.on, false)),\n            this.enableCategoriesElementModelsVisibilityStatus(props.categoryId),\n          )\n        : EMPTY,\n      from(props.subCategoryIds).pipe(map((subCategoryId) => this.#props.viewport.changeSubCategoryDisplay({ subCategoryId, display: props.on }))),\n    );\n  }\n\n  /** Changes grouped elements visibility status. */\n  public changeGroupedElementsVisibilityStatus(props: {\n    modelElementsMap: Map<ModelId, Set<ElementId>>;\n    categoryId: Id64String;\n    on: boolean;\n  }): Observable<void> {\n    return from(props.modelElementsMap).pipe(\n      mergeMap(([modelId, elementIds]) => {\n        return this.changeElementsVisibilityStatus({ modelId, elementIds, categoryId: props.categoryId, on: props.on });\n      }),\n    );\n  }\n\n  /** Turns on visibility status of models (that are not yet turned on) that are related to categories. */\n  private enableCategoriesElementModelsVisibilityStatus(categoryIds: Id64Arg): Observable<void> {\n    return this.#props.idsCache.getCategoriesElementModels(categoryIds, true).pipe(\n      reduce((acc, { models }) => {\n        models?.forEach((modelId) => {\n          if (!this.#props.viewport.viewsModel(modelId)) {\n            acc.add(modelId);\n          }\n        });\n        return acc;\n      }, new Set<Id64String>()),\n      map((hiddenModels) => {\n        if (hiddenModels.size > 0) {\n          this.#props.viewport.changeModelDisplay({ modelIds: hiddenModels, display: true });\n        }\n      }),\n    );\n  }\n}\n"]}
+{"version":3,"file":"CategoriesTreeVisibilityHelper.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,wBAAwB,EAAE,MAAM,mCAAmC,CAAC;AACrG,OAAO,EAAE,oBAAoB,EAAE,MAAM,6DAA6D,CAAC;AACnG,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAgBtF;;;;;GAKG;AACH,MAAM,OAAO,8BAA+B,SAAQ,oBAAoB;IACtE,MAAM,CAAsC;IAC5C,YAAY,KAA0C;QACpD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,uCAAuC,CAAC,KAA0C;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ;aACxB,yBAAyB,CAAC;YACzB,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;YACpD,sBAAsB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB;SACxE,CAAC;aACD,IAAI,CACH,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CACvB,IAAI,CAAC,6BAA6B,CAAC;YACjC,WAAW;YACX,OAAO,EAAE,SAAS;SACnB,CAAC,CACH,CACF,CAAC;IACN,CAAC;IAED,+CAA+C;IACxC,kCAAkC,CAAC,KAMzC;QACC,MAAM,EAAE,gBAAgB,EAAE,UAAU,EAAE,sBAAsB,EAAE,GAAG,KAAK,CAAC;QACvE,OAAO,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAChC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,8BAA8B,EAAE,CAAC,EAAE,EAAE,CACrE,IAAI,CAAC,2BAA2B,CAAC;YAC/B,UAAU;YACV,OAAO;YACP,UAAU;YACV,qBAAqB,EAAE,sBAAsB;gBAC3C,CAAC,CAAC,wBAAwB,CAAC;oBACvB,kBAAkB,EAAE,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC;oBACtH,sBAAsB;iBACvB,CAAC;gBACJ,CAAC,CAAC,EAAE;YACN,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,8BAA8B;SAC/B,CAAC,CACH,EACD,uBAAuB,EAAE,CAC1B,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,0CAA0C,CAAC,KAAuD;QACvG,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ;aACxB,yBAAyB,CAAC;YACzB,sBAAsB,EAAE,KAAK,CAAC,sBAAsB;YACpD,sBAAsB,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB;SACxE,CAAC;aACD,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/H,CAAC;IAED;;;;OAIG;IACI,mCAAmC,CAAC,KAAuE;QAChH,OAAO,MAAM;QACX,mDAAmD;QACnD,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,4CAA4C,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,EACtF,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAC7I,CAAC;IACJ,CAAC;IAED,kDAAkD;IAC3C,qCAAqC,CAAC,KAI5C;QACC,MAAM,UAAU,GAAG,IAAI,KAAK,EAAa,CAAC;QAC1C,KAAK,MAAM,EAAE,UAAU,EAAE,GAAG,EAAE,IAAI,KAAK,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC;YAClE,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;gBACrB,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CACnE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC,EAC1G,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CACpB,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAC/B,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,EAAE,EAAE;YACtD,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7I,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAED,6GAA6G;IACrG,4CAA4C,CAAC,UAAsB;QACzE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAChF,QAAQ,EAAE,EACV,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACnH,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC7C,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,CAClD,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE;oBACzB,+EAA+E;oBAC/E,KAAK,MAAM,eAAe,IAAI,kBAAkB,EAAE,CAAC;wBACjD,IAAI,eAAe,KAAK,UAAU,EAAE,CAAC;4BACnC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;wBAC1H,CAAC;oBACH,CAAC;oBACD,OAAO,OAAO,CAAC;gBACjB,CAAC,CAAC,CACH,CAAC;QACR,CAAC,CAAC,EACF,OAAO,EAAE,EACT,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE;YACnB,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACrF,CAAC;QACH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, EMPTY, from, map, mergeAll, mergeMap, toArray } from \"rxjs\";\nimport { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { getIdsFromChildrenTree, getParentElementsIdsPath } from \"../../../common/internal/Utils.js\";\nimport { BaseVisibilityHelper } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg, Id64String } from \"@itwin/core-bentley\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { BaseVisibilityHelperProps } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"../../CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport type CategoriesTreeVisibilityHelperProps = BaseVisibilityHelperProps & {\n  idsCache: CategoriesTreeIdsCache;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n};\n\n/**\n * Visibility status helper for categories tree.\n *\n * It extends base visibility status helper and provides methods to get and change visibility status of definition containers and grouped elements.\n * @internal\n */\nexport class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper {\n  #props: CategoriesTreeVisibilityHelperProps;\n  constructor(props: CategoriesTreeVisibilityHelperProps) {\n    super(props);\n    this.#props = props;\n  }\n\n  /**\n   * Gets visibility status of definition containers.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getDefinitionContainersVisibilityStatus(props: { definitionContainerIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache\n      .getAllContainedCategories({\n        definitionContainerIds: props.definitionContainerIds,\n        includeEmptyCategories: this.#props.hierarchyConfig.showEmptyCategories,\n      })\n      .pipe(\n        mergeMap((categoryIds) =>\n          this.getCategoriesVisibilityStatus({\n            categoryIds,\n            modelId: undefined,\n          }),\n        ),\n      );\n  }\n\n  /** Gets grouped elements visibility status. */\n  public getGroupedElementsVisibilityStatus(props: {\n    modelElementsMap: Map<ModelId, { elementIds: Set<ElementId>; categoryOfTopMostParentElement: CategoryId }>;\n    categoryId: Id64String;\n    parentKeys: HierarchyNodeKey[];\n    childrenCount: number;\n    topMostParentElementId?: ElementId;\n  }): Observable<VisibilityStatus> {\n    const { modelElementsMap, categoryId, topMostParentElementId } = props;\n    return from(modelElementsMap).pipe(\n      mergeMap(([modelId, { elementIds, categoryOfTopMostParentElement }]) =>\n        this.getElementsVisibilityStatus({\n          elementIds,\n          modelId,\n          categoryId,\n          parentElementsIdsPath: topMostParentElementId\n            ? getParentElementsIdsPath({\n                parentInstanceKeys: props.parentKeys.filter((key) => HierarchyNodeKey.isInstances(key)).map((key) => key.instanceKeys),\n                topMostParentElementId,\n              })\n            : [],\n          childrenCount: props.childrenCount,\n          categoryOfTopMostParentElement,\n        }),\n      ),\n      mergeVisibilityStatuses(),\n    );\n  }\n\n  /**\n   * Changes visibility status of definition containers.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeDefinitionContainersVisibilityStatus(props: { definitionContainerIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache\n      .getAllContainedCategories({\n        definitionContainerIds: props.definitionContainerIds,\n        includeEmptyCategories: this.#props.hierarchyConfig.showEmptyCategories,\n      })\n      .pipe(mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ categoryIds, modelId: undefined, on: props.on })));\n  }\n\n  /**\n   * Changes visibility status of sub-categories.\n   *\n   * Also, enables parent categories if `on` is true.\n   */\n  public changeSubCategoriesVisibilityStatus(props: { categoryId: Id64String; subCategoryIds: Id64Arg; on: boolean }): Observable<void> {\n    return concat(\n      // make sure parent category and models are enabled\n      props.on ? this.enableCategoryWithoutEnablingOtherCategories(props.categoryId) : EMPTY,\n      from(props.subCategoryIds).pipe(map((subCategoryId) => this.#props.viewport.changeSubCategoryDisplay({ subCategoryId, display: props.on }))),\n    );\n  }\n\n  /** Changes grouped elements visibility status. */\n  public changeGroupedElementsVisibilityStatus(props: {\n    modelElementsMap: Map<ModelId, { elementIds: Set<ElementId> }>;\n    categoryId: Id64String;\n    on: boolean;\n  }): Observable<void> {\n    const elementIds = new Array<ElementId>();\n    for (const { elementIds: ids } of props.modelElementsMap.values()) {\n      for (const id of ids) {\n        elementIds.push(id);\n      }\n    }\n    return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(\n      map((childrenTree) => getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 })),\n      mergeMap((children) =>\n        from(props.modelElementsMap).pipe(\n          mergeMap(([modelId, { elementIds: modelElementIds }]) => {\n            return this.changeElementsVisibilityStatus({ modelId, elementIds: modelElementIds, categoryId: props.categoryId, on: props.on, children });\n          }),\n        ),\n      ),\n    );\n  }\n\n  /** Turns on category and its' related models. Does not turn on other categories contained in those models.*/\n  private enableCategoryWithoutEnablingOtherCategories(categoryId: Id64String): Observable<void> {\n    this.#props.viewport.changeCategoryDisplay({ categoryIds: categoryId, display: true });\n    return this.#props.idsCache.getModels({ categoryId, includeSubModels: true }).pipe(\n      mergeAll(),\n      mergeMap((modelId) => {\n        this.#props.viewport.setPerModelCategoryOverride({ modelIds: modelId, categoryIds: categoryId, override: \"none\" });\n        return this.#props.viewport.viewsModel(modelId)\n          ? EMPTY\n          : this.#props.idsCache.getCategories({ modelId }).pipe(\n              map((allModelCategories) => {\n                // Add 'Hide' override to categories that were hidden before model is turned on\n                for (const modelCategoryId of allModelCategories) {\n                  if (modelCategoryId !== categoryId) {\n                    this.#props.viewport.setPerModelCategoryOverride({ modelIds: modelId, categoryIds: modelCategoryId, override: \"hide\" });\n                  }\n                }\n                return modelId;\n              }),\n            );\n      }),\n      toArray(),\n      map((hiddenModels) => {\n        if (hiddenModels.length > 0) {\n          this.#props.viewport.changeModelDisplay({ modelIds: hiddenModels, display: true });\n        }\n      }),\n    );\n  }\n}\n"]}

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

@@ -1,6 +1,6 @@
 import type { Id64Set, Id64String } from "@itwin/core-bentley";
 import type { HierarchySearchPath } from "@itwin/presentation-hierarchies";
-import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { ECClassHierarchyInspector, InstanceKey } from "@itwin/presentation-shared";
 import type { ElementId } from "../../../common/internal/Types.js";
 import type { SearchResultsTree } from "../../../common/internal/visibility/BaseSearchResultsTree.js";
 import type { CategoriesTreeIdsCache } from "../CategoriesTreeIdsCache.js";
@@ -11,11 +11,13 @@ export interface CategoriesTreeSearchTargets {
         categoryIds: Id64Set;
     }>;
     elements?: Array<{
+        pathToElements: InstanceKey[];
         modelId: Id64String;
         categoryId: Id64String;
         elements: Map<ElementId, {
             isSearchTarget: boolean;
         }>;
+        topMostParentElementId?: ElementId;
     }>;
     definitionContainerIds?: Id64Set;
     modelIds?: Id64Set;

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

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { firstValueFrom } from "rxjs";
 import { assert } from "@itwin/core-bentley";
-import { CLASS_NAME_SubCategory } from "../../../common/internal/ClassNameDefinitions.js";
+import { CLASS_NAME_DefinitionContainer, CLASS_NAME_SubCategory } from "../../../common/internal/ClassNameDefinitions.js";
 import { createSearchResultsTree, SearchResultsNodesHandler } from "../../../common/internal/visibility/BaseSearchResultsTree.js";
 /** @internal */
 export async function createCategoriesSearchResultsTree(props) {
@@ -31,17 +31,17 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
         const result = {
             searchResultsElements: new Map(),
         };
-        this.searchResultsNodesArr.forEach((node) => {
+        for (const node of this.searchResultsNodesArr) {
             if (node.type === "element") {
                 searchResultsTemporaryElements.set(node.id, node);
             }
-        });
+        }
         const searchResultsElementsModels = await firstValueFrom(this.#props.idsCache.getFilteredElementsModels([...searchResultsTemporaryElements.keys()]));
-        searchResultsTemporaryElements.forEach((element, id) => {
+        for (const [id, element] of searchResultsTemporaryElements) {
             const modelId = searchResultsElementsModels.get(element.id);
             assert(modelId !== undefined);
             result.searchResultsElements.set(id, { ...element, modelId });
-        });
+        }
         return result;
     }
     convertNodesToSearchTargets(searchResultsNodes, processedSearchResultsNodes) {
@@ -49,6 +49,26 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
         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, elements] of entry.modelCategoryElements) {
+                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
+                    result.push({ pathToElements: [...currentPath, identifier], modelId, categoryId, elements, 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.categories &&
             !searchTargets.definitionContainerIds &&
@@ -63,12 +83,7 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                     return { modelId, categoryIds };
                 })
                 : undefined,
-            elements: searchTargets.elements
-                ? [...searchTargets.elements.entries()].map(([modelCategoryKey, elements]) => {
-                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
-                    return { modelId, categoryId, elements };
-                })
-                : undefined,
+            elements: searchTargets.elements ? this.convertInternalSearchTargetElementsRecursively(searchTargets.elements, []) : undefined,
             definitionContainerIds: searchTargets.definitionContainerIds,
             modelIds: searchTargets.modelIds,
             subCategories: searchTargets.subCategories
@@ -121,13 +136,37 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                 searchTargets.categories.set(node.modelId, new Set([node.id]));
                 return;
             case "element":
+                // 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);
-                const elements = (searchTargets.elements ??= new Map()).get(modelCategoryKey);
-                if (elements) {
-                    elements.set(node.id, { isSearchTarget: node.isSearchTarget });
-                }
-                else {
-                    searchTargets.elements.set(modelCategoryKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));
+                searchTargets.elements ??= new Map();
+                let entry = searchTargets.elements;
+                let topMostParentElementId;
+                for (let i = 0; i < node.pathToNode.length; ++i) {
+                    if (topMostParentElementId === undefined && node.pathToNode[i].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, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));
+                    }
                 }
         }
     }
@@ -139,11 +178,13 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
         return { modelId, categoryId };
     }
     createSearchResultsTreeNode({ type, id, isSearchTarget, parent, }) {
+        const pathToNode = "pathToNode" in parent ? [...parent.pathToNode, { type: parent.type, id: parent.id }] : [];
         if (type === "definitionContainer") {
             return {
                 id,
                 isSearchTarget,
                 type,
+                pathToNode,
             };
         }
         if (type === "subCategory") {
@@ -153,6 +194,7 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                 isSearchTarget,
                 type,
                 categoryId: parent.id,
+                pathToNode,
             };
         }
         if (type === "category") {
@@ -162,12 +204,14 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                     isSearchTarget,
                     type,
                     modelId: parent.id,
+                    pathToNode,
                 };
             }
             return {
                 id,
                 isSearchTarget,
                 type,
+                pathToNode,
             };
         }
         if (type === "model") {
@@ -177,6 +221,7 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                 isSearchTarget,
                 type,
                 categoryId: parent.type === "category" ? parent.id : undefined,
+                pathToNode,
             };
         }
         if ("type" in parent) {
@@ -187,6 +232,7 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                     type,
                     categoryId: parent.id,
                     modelId: undefined,
+                    pathToNode,
                 };
             }
             assert(parent.type === "element");
@@ -196,6 +242,7 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
                 type,
                 categoryId: parent.categoryId,
                 modelId: undefined,
+                pathToNode,
             };
         }
         throw new Error("Invalid parent node type");
@@ -215,5 +262,19 @@ class CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler
         }
         return "definitionContainer";
     }
+    getClassName(type) {
+        switch (type) {
+            case "definitionContainer":
+                return CLASS_NAME_DefinitionContainer;
+            case "subCategory":
+                return CLASS_NAME_SubCategory;
+            case "category":
+                return this.#props.categoryClassName;
+            case "model":
+                return this.#props.categoryModelClassName;
+            default:
+                return this.#props.categoryElementClassName;
+        }
+    }
 }
 //# sourceMappingURL=SearchResultsTree.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"SearchResultsTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-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,sBAAsB,EAAE,MAAM,kDAAkD,CAAC;AAC1F,OAAO,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,8DAA8D,CAAC;AAkElI,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,iCAAiC,CAAC,KAOvD;IACC,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3H,OAAO,uBAAuB,CAAC;QAC7B,yBAAyB,EAAE,IAAI,uCAAuC,CAAC;YACrE,QAAQ;YACR,YAAY;YACZ,iBAAiB;YACjB,wBAAwB;YACxB,sBAAsB;SACvB,CAAC;QACF,WAAW;KACZ,CAAC,CAAC;AACL,CAAC;AAwBD,MAAM,uCAAwC,SAAQ,yBAIrD;IACU,MAAM,CAA+C;IAC9D,YAAY,KAAmD;QAC7D,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,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC5B,8BAA8B,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACpD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,2BAA2B,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,GAAG,8BAA8B,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrJ,8BAA8B,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE;YACrD,MAAM,OAAO,GAAG,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC;YAC9B,MAAM,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAChE,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,IACE,CAAC,aAAa,CAAC,UAAU;YACzB,CAAC,aAAa,CAAC,sBAAsB;YACrC,CAAC,aAAa,CAAC,QAAQ;YACvB,CAAC,aAAa,CAAC,QAAQ;YACvB,CAAC,aAAa,CAAC,aAAa,EAC5B,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE;oBACrE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;gBAClC,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAC9B,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBACzE,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,sBAAsB,EAAE,aAAa,CAAC,sBAAsB;YAC5D,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,aAAa,EAAE,aAAa,CAAC,aAAa;gBACxC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE;oBAC9E,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;gBACxC,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAC1B,aAAoC,EACpC,IAAoC,EACpC,2BAAwD;QAExD,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,2BAA2B,CAAC,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,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,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,OAAO;gBACV,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACpD,OAAO;YACT,KAAK,aAAa;gBAChB,MAAM,aAAa,GAAG,CAAC,aAAa,CAAC,aAAa,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACvF,IAAI,aAAa,EAAE,CAAC;oBAClB,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC3B,OAAO;gBACT,CAAC;gBACD,aAAa,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACrE,OAAO;YACT,KAAK,UAAU;gBACb,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9E,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACxB,OAAO;gBACT,CAAC;gBACD,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC/D,OAAO;YACT,KAAK,SAAS;gBACZ,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpF,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9E,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACjE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9G,CAAC;QACL,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,IAAI,IAAI,KAAK,qBAAqB,EAAE,CAAC;YACnC,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;aACL,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;YACvB,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,EAAE;aACtB,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACxB,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAChD,OAAO;oBACL,EAAE;oBACF,cAAc;oBACd,IAAI;oBACJ,OAAO,EAAE,MAAM,CAAC,EAAE;iBACnB,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;aACL,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;YACvB,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;aAC/D,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACrB,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,OAAO;oBACL,EAAE;oBACF,cAAc;oBACd,IAAI;oBACJ,UAAU,EAAE,MAAM,CAAC,EAAE;oBACrB,OAAO,EAAE,SAAS;iBACnB,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAClC,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,SAAS;aACnB,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC9C,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,sBAAsB,CAAC,EAAE,CAAC;YACvF,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACrG,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC9F,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACnG,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,qBAAqB,CAAC;IAC/B,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_SubCategory } 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, SubCategoryId } from \"../../../common/internal/Types.js\";\nimport type {\n  BaseSearchResultsTreeNode,\n  SearchResultsTree,\n  SearchResultsTreeNodeChildren,\n  SearchResultsTreeRootNode,\n} from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport interface CategoriesTreeSearchTargets {\n  categories?: Array<{ modelId: Id64String | undefined; categoryIds: Id64Set }>;\n  elements?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isSearchTarget: boolean }> }>;\n  definitionContainerIds?: Id64Set;\n  modelIds?: Id64Set;\n  subCategories?: Array<{ categoryId: Id64String; subCategoryIds: Id64Set }>;\n}\n\ninterface CategorySearchResultsTreeNode extends BaseSearchResultsTreeNode<CategorySearchResultsTreeNode> {\n  type: \"category\";\n  modelId?: Id64String;\n}\n\ninterface ModelSearchResultsTreeNode extends BaseSearchResultsTreeNode<ModelSearchResultsTreeNode> {\n  type: \"model\";\n  categoryId?: Id64String;\n}\ninterface SubCategorySearchResultsTreeNode extends BaseSearchResultsTreeNode<SubCategorySearchResultsTreeNode> {\n  type: \"subCategory\";\n  categoryId: Id64String;\n}\n\ninterface DefinitionContainerSearchResultsTreeNode extends BaseSearchResultsTreeNode<DefinitionContainerSearchResultsTreeNode> {\n  type: \"definitionContainer\";\n}\n\ninterface ElementSearchResultsTreeNode extends BaseSearchResultsTreeNode<ElementSearchResultsTreeNode> {\n  type: \"element\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ntype SearchResultsTreeNode =\n  | DefinitionContainerSearchResultsTreeNode\n  | SubCategorySearchResultsTreeNode\n  | CategorySearchResultsTreeNode\n  | ElementSearchResultsTreeNode\n  | ModelSearchResultsTreeNode;\n\ntype TemporaryElementSearchResultsNode = Omit<ElementSearchResultsTreeNode, \"modelId\" | \"children\"> & {\n  modelId: string | undefined;\n  children?: SearchResultsTreeNodeChildren<TemporaryElementSearchResultsNode>;\n};\n\ntype TemporarySearchResultsTreeNode =\n  | DefinitionContainerSearchResultsTreeNode\n  | SubCategorySearchResultsTreeNode\n  | CategorySearchResultsTreeNode\n  | TemporaryElementSearchResultsNode\n  | ModelSearchResultsTreeNode;\n\n/** @internal */\nexport async function createCategoriesSearchResultsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n  categoryClassName: string;\n  categoryElementClassName: string;\n  categoryModelClassName: string;\n  idsCache: CategoriesTreeIdsCache;\n}): Promise<SearchResultsTree<CategoriesTreeSearchTargets>> {\n  const { imodelAccess, searchPaths, categoryClassName, categoryElementClassName, categoryModelClassName, idsCache } = props;\n  return createSearchResultsTree({\n    searchResultsNodesHandler: new CategoriesTreeSearchResultsNodesHandler({\n      idsCache,\n      imodelAccess,\n      categoryClassName,\n      categoryElementClassName,\n      categoryModelClassName,\n    }),\n    searchPaths,\n  });\n}\n\ninterface SearchTargetsInternal {\n  elements?: Map<ModelCategoryKey, Map<ElementId, { isSearchTarget: boolean }>>;\n  categories?: Map<ModelId | undefined, Set<CategoryId>>;\n  definitionContainerIds?: Id64Set;\n  modelIds?: Id64Set;\n  subCategories?: Map<CategoryId, Set<SubCategoryId>>;\n}\n\ninterface CategoriesTreeSearchResultsNodesHandlerProps {\n  idsCache: CategoriesTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  categoryClassName: string;\n  categoryElementClassName: string;\n  categoryModelClassName: string;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\ninterface ProcessedSearchResultsNodes {\n  searchResultsElements: Map<Id64String, Omit<ElementSearchResultsTreeNode, \"children\">>;\n}\n\nclass CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler<\n  ProcessedSearchResultsNodes,\n  CategoriesTreeSearchTargets,\n  TemporarySearchResultsTreeNode\n> {\n  readonly #props: CategoriesTreeSearchResultsNodesHandlerProps;\n  constructor(props: CategoriesTreeSearchResultsNodesHandlerProps) {\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    this.searchResultsNodesArr.forEach((node) => {\n      if (node.type === \"element\") {\n        searchResultsTemporaryElements.set(node.id, node);\n      }\n    });\n\n    const searchResultsElementsModels = await firstValueFrom(this.#props.idsCache.getFilteredElementsModels([...searchResultsTemporaryElements.keys()]));\n    searchResultsTemporaryElements.forEach((element, id) => {\n      const modelId = searchResultsElementsModels.get(element.id);\n      assert(modelId !== undefined);\n      result.searchResultsElements.set(id, { ...element, modelId });\n    });\n    return result;\n  }\n\n  public convertNodesToSearchTargets(\n    searchResultsNodes: TemporarySearchResultsTreeNode[],\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ): CategoriesTreeSearchTargets | 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): CategoriesTreeSearchTargets | undefined {\n    if (\n      !searchTargets.categories &&\n      !searchTargets.definitionContainerIds &&\n      !searchTargets.elements &&\n      !searchTargets.modelIds &&\n      !searchTargets.subCategories\n    ) {\n      return undefined;\n    }\n    return {\n      categories: searchTargets.categories\n        ? [...searchTargets.categories.entries()].map(([modelId, categoryIds]) => {\n            return { modelId, categoryIds };\n          })\n        : undefined,\n      elements: searchTargets.elements\n        ? [...searchTargets.elements.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n      definitionContainerIds: searchTargets.definitionContainerIds,\n      modelIds: searchTargets.modelIds,\n      subCategories: searchTargets.subCategories\n        ? [...searchTargets.subCategories.entries()].map(([categoryId, subCategoryIds]) => {\n            return { categoryId, subCategoryIds };\n          })\n        : undefined,\n    };\n  }\n\n  private collectSearchTargets(\n    searchTargets: SearchTargetsInternal,\n    node: TemporarySearchResultsTreeNode,\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ) {\n    const searchResultsNode = node.type !== \"element\" ? node : processedSearchResultsNodes.searchResultsElements.get(node.id);\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 addTarget(searchTargets: SearchTargetsInternal, node: SearchResultsTreeNode) {\n    switch (node.type) {\n      case \"definitionContainer\":\n        (searchTargets.definitionContainerIds ??= new Set()).add(node.id);\n        return;\n      case \"model\":\n        (searchTargets.modelIds ??= new Set()).add(node.id);\n        return;\n      case \"subCategory\":\n        const subCategories = (searchTargets.subCategories ??= new Map()).get(node.categoryId);\n        if (subCategories) {\n          subCategories.add(node.id);\n          return;\n        }\n        searchTargets.subCategories.set(node.categoryId, new Set([node.id]));\n        return;\n      case \"category\":\n        const categories = (searchTargets.categories ??= new Map()).get(node.modelId);\n        if (!categories) {\n          categories.add(node.id);\n          return;\n        }\n        searchTargets.categories.set(node.modelId, new Set([node.id]));\n        return;\n      case \"element\":\n        const modelCategoryKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements = (searchTargets.elements ??= new Map()).get(modelCategoryKey);\n        if (elements) {\n          elements.set(node.id, { isSearchTarget: node.isSearchTarget });\n        } else {\n          searchTargets.elements.set(modelCategoryKey, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));\n        }\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: TemporarySearchResultsTreeNode | SearchResultsTreeRootNode<TemporarySearchResultsTreeNode>;\n  }): TemporarySearchResultsTreeNode {\n    if (type === \"definitionContainer\") {\n      return {\n        id,\n        isSearchTarget,\n        type,\n      };\n    }\n    if (type === \"subCategory\") {\n      assert(\"id\" in parent);\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.id,\n      };\n    }\n    if (type === \"category\") {\n      if (\"type\" in parent && parent.type === \"model\") {\n        return {\n          id,\n          isSearchTarget,\n          type,\n          modelId: parent.id,\n        };\n      }\n      return {\n        id,\n        isSearchTarget,\n        type,\n      };\n    }\n    if (type === \"model\") {\n      assert(\"id\" in parent);\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.type === \"category\" ? parent.id : undefined,\n      };\n    }\n\n    if (\"type\" in parent) {\n      if (parent.type === \"category\") {\n        return {\n          id,\n          isSearchTarget,\n          type,\n          categoryId: parent.id,\n          modelId: undefined,\n        };\n      }\n      assert(parent.type === \"element\");\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.categoryId,\n        modelId: undefined,\n      };\n    }\n\n    throw new Error(\"Invalid parent node type\");\n  }\n\n  public async getType(className: string): Promise<TemporarySearchResultsTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_SubCategory)) {\n      return \"subCategory\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryElementClassName)) {\n      return \"element\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryClassName)) {\n      return \"category\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryModelClassName)) {\n      return \"model\";\n    }\n    return \"definitionContainer\";\n  }\n}\n"]}
+{"version":3,"file":"SearchResultsTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-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,8BAA8B,EAAE,sBAAsB,EAAE,MAAM,kDAAkD,CAAC;AAC1H,OAAO,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,8DAA8D,CAAC;AAyElI,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,iCAAiC,CAAC,KAOvD;IACC,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAC3H,OAAO,uBAAuB,CAAC;QAC7B,yBAAyB,EAAE,IAAI,uCAAuC,CAAC;YACrE,QAAQ;YACR,YAAY;YACZ,iBAAiB;YACjB,wBAAwB;YACxB,sBAAsB;SACvB,CAAC;QACF,WAAW;KACZ,CAAC,CAAC;AACL,CAAC;AAiCD,MAAM,uCAAwC,SAAQ,yBAIrD;IACU,MAAM,CAA+C;IAC9D,YAAY,KAAmD;QAC7D,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,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,GAAG,8BAA8B,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrJ,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,8BAA8B,EAAE,CAAC;YAC3D,MAAM,OAAO,GAAG,2BAA2B,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC;YAC9B,MAAM,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAChE,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,GAAsD,EAAE,CAAC;QACrE,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,QAAQ,CAAC,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;oBACvE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,MAAM,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,CAAC,GAAG,WAAW,EAAE,UAAU,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,sBAAsB,EAAE,KAAK,CAAC,sBAAsB,EAAE,CAAC,CAAC;gBACrJ,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,IACE,CAAC,aAAa,CAAC,UAAU;YACzB,CAAC,aAAa,CAAC,sBAAsB;YACrC,CAAC,aAAa,CAAC,QAAQ;YACvB,CAAC,aAAa,CAAC,QAAQ;YACvB,CAAC,aAAa,CAAC,aAAa,EAC5B,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE;oBACrE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;gBAClC,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,8CAA8C,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;YAC9H,sBAAsB,EAAE,aAAa,CAAC,sBAAsB;YAC5D,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,aAAa,EAAE,aAAa,CAAC,aAAa;gBACxC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE;oBAC9E,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;gBACxC,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAC1B,aAAoC,EACpC,IAAoC,EACpC,2BAAwD;QAExD,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,2BAA2B,CAAC,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,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,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,OAAO;gBACV,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACpD,OAAO;YACT,KAAK,aAAa;gBAChB,MAAM,aAAa,GAAG,CAAC,aAAa,CAAC,aAAa,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACvF,IAAI,aAAa,EAAE,CAAC;oBAClB,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC3B,OAAO;gBACT,CAAC;gBACD,aAAa,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACrE,OAAO;YACT,KAAK,UAAU;gBACb,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9E,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACxB,OAAO;gBACT,CAAC;gBACD,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC/D,OAAO;YACT,KAAK,SAAS;gBACZ,uEAAuE;gBACvE,yHAAyH;gBACzH,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpF,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC;gBACrC,IAAI,KAAK,GAAG,aAAa,CAAC,QAAQ,CAAC;gBACnC,IAAI,sBAA8C,CAAC;gBACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;oBAChD,IAAI,sBAAsB,KAAK,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;wBAClF,sBAAsB,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACjD,CAAC;oBACD,MAAM,kBAAkB,GAAG,IAAI,CAAC,0CAA0C,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC/F,IAAI,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBACpD,0DAA0D;oBAC1D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,eAAe,GAAG,EAAE,sBAAsB,EAAE,CAAC;wBAC7C,KAAK,CAAC,GAAG,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC;oBACjD,CAAC;oBACD,qDAAqD;oBACrD,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACnC,eAAe,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC;wBACvC,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC;wBACjC,SAAS;oBACX,CAAC;oBAED,MAAM,QAAQ,GAAG,CAAC,eAAe,CAAC,qBAAqB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;oBAC7F,wEAAwE;oBACxE,IAAI,QAAQ,EAAE,CAAC;wBACb,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;oBACjE,CAAC;yBAAM,CAAC;wBACN,eAAe,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7H,CAAC;gBACH,CAAC;QACL,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,qBAAqB,EAAE,CAAC;YACnC,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU;aACX,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;YACvB,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,EAAE;gBACrB,UAAU;aACX,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACxB,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAChD,OAAO;oBACL,EAAE;oBACF,cAAc;oBACd,IAAI;oBACJ,OAAO,EAAE,MAAM,CAAC,EAAE;oBAClB,UAAU;iBACX,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU;aACX,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;YACvB,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;gBAC9D,UAAU;aACX,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACrB,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,OAAO;oBACL,EAAE;oBACF,cAAc;oBACd,IAAI;oBACJ,UAAU,EAAE,MAAM,CAAC,EAAE;oBACrB,OAAO,EAAE,SAAS;oBAClB,UAAU;iBACX,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAClC,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,SAAS;gBAClB,UAAU;aACX,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC9C,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,sBAAsB,CAAC,EAAE,CAAC;YACvF,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACrG,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC9F,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACnG,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,qBAAqB,CAAC;IAC/B,CAAC;IAEM,YAAY,CAAC,IAA4C;QAC9D,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,qBAAqB;gBACxB,OAAO,8BAA8B,CAAC;YACxC,KAAK,aAAa;gBAChB,OAAO,sBAAsB,CAAC;YAChC,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;YACvC,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC;YAC5C;gBACE,OAAO,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC;QAChD,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_DefinitionContainer, CLASS_NAME_SubCategory } 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, SubCategoryId } 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 { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport interface CategoriesTreeSearchTargets {\n  categories?: Array<{ modelId: Id64String | undefined; categoryIds: Id64Set }>;\n  elements?: Array<{\n    pathToElements: InstanceKey[];\n    modelId: Id64String;\n    categoryId: Id64String;\n    elements: Map<ElementId, { isSearchTarget: boolean }>;\n    topMostParentElementId?: ElementId;\n  }>;\n  definitionContainerIds?: Id64Set;\n  modelIds?: Id64Set;\n  subCategories?: Array<{ categoryId: Id64String; subCategoryIds: Id64Set }>;\n}\n\ninterface CategorySearchResultsTreeNode extends BaseSearchResultsTreeNode<CategorySearchResultsTreeNode> {\n  type: \"category\";\n  modelId?: Id64String;\n}\n\ninterface ModelSearchResultsTreeNode extends BaseSearchResultsTreeNode<ModelSearchResultsTreeNode> {\n  type: \"model\";\n  categoryId?: Id64String;\n}\ninterface SubCategorySearchResultsTreeNode extends BaseSearchResultsTreeNode<SubCategorySearchResultsTreeNode> {\n  type: \"subCategory\";\n  categoryId: Id64String;\n}\n\ninterface DefinitionContainerSearchResultsTreeNode extends BaseSearchResultsTreeNode<DefinitionContainerSearchResultsTreeNode> {\n  type: \"definitionContainer\";\n}\n\ninterface ElementSearchResultsTreeNode extends BaseSearchResultsTreeNode<ElementSearchResultsTreeNode> {\n  type: \"element\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ntype SearchResultsTreeNode =\n  | DefinitionContainerSearchResultsTreeNode\n  | SubCategorySearchResultsTreeNode\n  | CategorySearchResultsTreeNode\n  | ElementSearchResultsTreeNode\n  | ModelSearchResultsTreeNode;\n\ntype TemporaryElementSearchResultsNode = Omit<ElementSearchResultsTreeNode, \"modelId\" | \"children\"> & {\n  modelId: string | undefined;\n  children?: SearchResultsTreeNodeChildren<TemporaryElementSearchResultsNode>;\n};\n\ntype TemporarySearchResultsTreeNode =\n  | DefinitionContainerSearchResultsTreeNode\n  | SubCategorySearchResultsTreeNode\n  | CategorySearchResultsTreeNode\n  | TemporaryElementSearchResultsNode\n  | ModelSearchResultsTreeNode;\n\n/** @internal */\nexport async function createCategoriesSearchResultsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n  categoryClassName: string;\n  categoryElementClassName: string;\n  categoryModelClassName: string;\n  idsCache: CategoriesTreeIdsCache;\n}): Promise<SearchResultsTree<CategoriesTreeSearchTargets>> {\n  const { imodelAccess, searchPaths, categoryClassName, categoryElementClassName, categoryModelClassName, idsCache } = props;\n  return createSearchResultsTree({\n    searchResultsNodesHandler: new CategoriesTreeSearchResultsNodesHandler({\n      idsCache,\n      imodelAccess,\n      categoryClassName,\n      categoryElementClassName,\n      categoryModelClassName,\n    }),\n    searchPaths,\n  });\n}\n\ntype SearchTargetsInternalElements = Map<\n  SearchResultsNodeIdentifierAsString,\n  {\n    children?: SearchTargetsInternalElements;\n    topMostParentElementId?: Id64String;\n    modelCategoryElements?: Map<ModelCategoryKey, Map<ElementId, { isSearchTarget: boolean }>>;\n  }\n>;\n\ninterface SearchTargetsInternal {\n  elements?: SearchTargetsInternalElements;\n  categories?: Map<ModelId | undefined, Set<CategoryId>>;\n  definitionContainerIds?: Id64Set;\n  modelIds?: Id64Set;\n  subCategories?: Map<CategoryId, Set<SubCategoryId>>;\n}\n\ninterface CategoriesTreeSearchResultsNodesHandlerProps {\n  idsCache: CategoriesTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  categoryClassName: string;\n  categoryElementClassName: string;\n  categoryModelClassName: string;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\ninterface ProcessedSearchResultsNodes {\n  searchResultsElements: Map<Id64String, Omit<ElementSearchResultsTreeNode, \"children\">>;\n}\n\nclass CategoriesTreeSearchResultsNodesHandler extends SearchResultsNodesHandler<\n  ProcessedSearchResultsNodes,\n  CategoriesTreeSearchTargets,\n  TemporarySearchResultsTreeNode\n> {\n  readonly #props: CategoriesTreeSearchResultsNodesHandlerProps;\n  constructor(props: CategoriesTreeSearchResultsNodesHandlerProps) {\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(this.#props.idsCache.getFilteredElementsModels([...searchResultsTemporaryElements.keys()]));\n    for (const [id, element] of searchResultsTemporaryElements) {\n      const modelId = searchResultsElementsModels.get(element.id);\n      assert(modelId !== undefined);\n      result.searchResultsElements.set(id, { ...element, modelId });\n    }\n    return result;\n  }\n\n  public convertNodesToSearchTargets(\n    searchResultsNodes: TemporarySearchResultsTreeNode[],\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ): CategoriesTreeSearchTargets | 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<CategoriesTreeSearchTargets>[\"elements\"] {\n    const result: Required<CategoriesTreeSearchTargets>[\"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, elements] of entry.modelCategoryElements) {\n          const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n          result.push({ pathToElements: [...currentPath, identifier], modelId, categoryId, elements, topMostParentElementId: entry.topMostParentElementId });\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): CategoriesTreeSearchTargets | undefined {\n    if (\n      !searchTargets.categories &&\n      !searchTargets.definitionContainerIds &&\n      !searchTargets.elements &&\n      !searchTargets.modelIds &&\n      !searchTargets.subCategories\n    ) {\n      return undefined;\n    }\n    return {\n      categories: searchTargets.categories\n        ? [...searchTargets.categories.entries()].map(([modelId, categoryIds]) => {\n            return { modelId, categoryIds };\n          })\n        : undefined,\n      elements: searchTargets.elements ? this.convertInternalSearchTargetElementsRecursively(searchTargets.elements, []) : undefined,\n      definitionContainerIds: searchTargets.definitionContainerIds,\n      modelIds: searchTargets.modelIds,\n      subCategories: searchTargets.subCategories\n        ? [...searchTargets.subCategories.entries()].map(([categoryId, subCategoryIds]) => {\n            return { categoryId, subCategoryIds };\n          })\n        : undefined,\n    };\n  }\n\n  private collectSearchTargets(\n    searchTargets: SearchTargetsInternal,\n    node: TemporarySearchResultsTreeNode,\n    processedSearchResultsNodes: ProcessedSearchResultsNodes,\n  ) {\n    const searchResultsNode = node.type !== \"element\" ? node : processedSearchResultsNodes.searchResultsElements.get(node.id);\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 addTarget(searchTargets: SearchTargetsInternal, node: SearchResultsTreeNode) {\n    switch (node.type) {\n      case \"definitionContainer\":\n        (searchTargets.definitionContainerIds ??= new Set()).add(node.id);\n        return;\n      case \"model\":\n        (searchTargets.modelIds ??= new Set()).add(node.id);\n        return;\n      case \"subCategory\":\n        const subCategories = (searchTargets.subCategories ??= new Map()).get(node.categoryId);\n        if (subCategories) {\n          subCategories.add(node.id);\n          return;\n        }\n        searchTargets.subCategories.set(node.categoryId, new Set([node.id]));\n        return;\n      case \"category\":\n        const categories = (searchTargets.categories ??= new Map()).get(node.modelId);\n        if (!categories) {\n          categories.add(node.id);\n          return;\n        }\n        searchTargets.categories.set(node.modelId, new Set([node.id]));\n        return;\n      case \"element\":\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        searchTargets.elements ??= new Map();\n        let entry = searchTargets.elements;\n        let topMostParentElementId: Id64String | undefined;\n        for (let i = 0; i < node.pathToNode.length; ++i) {\n          if (topMostParentElementId === undefined && node.pathToNode[i].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, new Map([[node.id, { isSearchTarget: node.isSearchTarget }]]));\n          }\n        }\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: TemporarySearchResultsTreeNode | SearchResultsTreeRootNode<TemporarySearchResultsTreeNode>;\n  }): TemporarySearchResultsTreeNode {\n    const pathToNode = \"pathToNode\" in parent ? [...parent.pathToNode, { type: parent.type, id: parent.id }] : [];\n    if (type === \"definitionContainer\") {\n      return {\n        id,\n        isSearchTarget,\n        type,\n        pathToNode,\n      };\n    }\n    if (type === \"subCategory\") {\n      assert(\"id\" in parent);\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.id,\n        pathToNode,\n      };\n    }\n    if (type === \"category\") {\n      if (\"type\" in parent && parent.type === \"model\") {\n        return {\n          id,\n          isSearchTarget,\n          type,\n          modelId: parent.id,\n          pathToNode,\n        };\n      }\n      return {\n        id,\n        isSearchTarget,\n        type,\n        pathToNode,\n      };\n    }\n    if (type === \"model\") {\n      assert(\"id\" in parent);\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.type === \"category\" ? parent.id : undefined,\n        pathToNode,\n      };\n    }\n\n    if (\"type\" in parent) {\n      if (parent.type === \"category\") {\n        return {\n          id,\n          isSearchTarget,\n          type,\n          categoryId: parent.id,\n          modelId: undefined,\n          pathToNode,\n        };\n      }\n      assert(parent.type === \"element\");\n      return {\n        id,\n        isSearchTarget,\n        type,\n        categoryId: parent.categoryId,\n        modelId: undefined,\n        pathToNode,\n      };\n    }\n\n    throw new Error(\"Invalid parent node type\");\n  }\n\n  public async getType(className: string): Promise<TemporarySearchResultsTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_SubCategory)) {\n      return \"subCategory\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryElementClassName)) {\n      return \"element\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryClassName)) {\n      return \"category\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, this.#props.categoryModelClassName)) {\n      return \"model\";\n    }\n    return \"definitionContainer\";\n  }\n\n  public getClassName(type: TemporarySearchResultsTreeNode[\"type\"]): string {\n    switch (type) {\n      case \"definitionContainer\":\n        return CLASS_NAME_DefinitionContainer;\n      case \"subCategory\":\n        return CLASS_NAME_SubCategory;\n      case \"category\":\n        return this.#props.categoryClassName;\n      case \"model\":\n        return this.#props.categoryModelClassName;\n      default:\n        return this.#props.categoryElementClassName;\n    }\n  }\n}\n"]}

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

@@ -14,6 +14,8 @@ interface ClassificationsTreeComponentProps extends Pick<ClassificationsTreeProp
 }
 /**
  * A component that renders `ClassificationsTree` with active iModel and viewport.
+ *
+ * **Note:** Wrap tree components with a single `SharedTreeContextProvider` to improve trees' performance.
  * @alpha
  */
 export declare const ClassificationsTreeComponent: {

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

@@ -7,11 +7,14 @@ import { useActiveIModelConnection } from "@itwin/appui-react";
 import { SchemaKey, SchemaMatchType } from "@itwin/ecschema-metadata";
 import { TreeWidget } from "../../../TreeWidget.js";
 import { SelectableTree } from "../../tree-header/SelectableTree.js";
+import { SharedTreeContextProviderInternal } from "../common/internal/SharedTreeContextProviderInternal.js";
 import { useActiveTreeWidgetViewport } from "../common/internal/UseActiveTreeWidgetViewport.js";
 import { TelemetryContextProvider } from "../common/UseTelemetryContext.js";
 import { ClassificationsTree } from "./ClassificationsTree.js";
 /**
  * A component that renders `ClassificationsTree` with active iModel and viewport.
+ *
+ * **Note:** Wrap tree components with a single `SharedTreeContextProvider` to improve trees' performance.
  * @alpha
  */
 export const ClassificationsTreeComponent = (props) => {
@@ -20,7 +23,7 @@ export const ClassificationsTreeComponent = (props) => {
     if (!iModel || !viewport) {
         return null;
     }
-    return _jsx(ClassificationsTreeComponentImpl, { ...props, iModel: iModel, viewport: viewport });
+    return (_jsx(SharedTreeContextProviderInternal, { showWarning: true, children: _jsx(ClassificationsTreeComponentImpl, { ...props, iModel: iModel, viewport: viewport }) }));
 };
 /**
  * Id of the component. May be used when a creating a `TreeDefinition`.

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

@@ -1 +1 @@
-{"version":3,"file":"ClassificationsTreeComponent.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/classifications-tree/ClassificationsTreeComponent.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,2BAA2B,EAAE,MAAM,mDAAmD,CAAC;AAChG,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAiC/D;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,KAAwC,EAAE,EAAE;IACvF,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,2BAA2B,CAAC,EAAE,kBAAkB,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAErF,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAC,gCAAgC,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAI,CAAC;AAC7F,CAAC,CAAC;AAEF;;;GAGG;AACH,4BAA4B,CAAC,EAAE,GAAG,sBAAsB,CAAC;AAEzD;;;GAGG;AACH,4BAA4B,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;AAEhG,aAAa;AACb,4BAA4B,CAAC,mBAAmB,GAAG,KAAK,EAAE,MAAwB,EAAE,EAAE;IACpF,IAAI,CAAC;QACH,OAAO,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,SAAS,CAAC,uBAAuB,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;IACtH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC,CAAC;AAEF,SAAS,gCAAgC,CAAC,EACxC,MAAM,EACN,QAAQ,EACR,qBAAqB,EACrB,aAAa,EACb,UAAU,EACV,SAAS,EACT,GAAG,SAAS,EACmF;IAC/F,OAAO,CACL,KAAC,wBAAwB,IAAC,mBAAmB,EAAE,4BAA4B,CAAC,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE,qBAAqB,EAAE,qBAAqB,YACxJ,KAAC,cAAc,cACb,KAAC,mBAAmB,OAAK,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,GAAI,GAC3G,GACQ,CAC5B,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 { useActiveIModelConnection } from \"@itwin/appui-react\";\nimport { SchemaKey, SchemaMatchType } from \"@itwin/ecschema-metadata\";\nimport { TreeWidget } from \"../../../TreeWidget.js\";\nimport { SelectableTree } from \"../../tree-header/SelectableTree.js\";\nimport { useActiveTreeWidgetViewport } from \"../common/internal/UseActiveTreeWidgetViewport.js\";\nimport { TelemetryContextProvider } from \"../common/UseTelemetryContext.js\";\nimport { ClassificationsTree } from \"./ClassificationsTree.js\";\n\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { TreeWidgetViewport } from \"../common/TreeWidgetViewport.js\";\nimport type { ClassificationsTreeProps } from \"./ClassificationsTree.js\";\n\n/** @alpha */\ninterface ClassificationsTreeComponentProps\n  extends Pick<\n    ClassificationsTreeProps,\n    | \"selectionStorage\"\n    | \"hierarchyLevelConfig\"\n    | \"selectionMode\"\n    | \"searchText\"\n    | \"emptyTreeContent\"\n    | \"getInlineActions\"\n    | \"getMenuActions\"\n    | \"getContextMenuActions\"\n    | \"getTreeItemProps\"\n    | \"hierarchyConfig\"\n    | \"getEditingProps\"\n    | \"treeLabel\"\n  > {\n  /**\n   * Viewport used for visibility controls.\n   *\n   * When viewport is not provided, `IModelApp.viewManager.selectedView` will be used.\n   */\n  viewport?: TreeWidgetViewport;\n  onPerformanceMeasured?: (featureId: string, duration: number) => void;\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * A component that renders `ClassificationsTree` with active iModel and viewport.\n * @alpha\n */\nexport const ClassificationsTreeComponent = (props: ClassificationsTreeComponentProps) => {\n  const iModel = useActiveIModelConnection();\n  const viewport = useActiveTreeWidgetViewport({ treeWidgetViewport: props.viewport });\n\n  if (!iModel || !viewport) {\n    return null;\n  }\n\n  return <ClassificationsTreeComponentImpl {...props} iModel={iModel} viewport={viewport} />;\n};\n\n/**\n * Id of the component. May be used when a creating a `TreeDefinition`.\n * @alpha\n */\nClassificationsTreeComponent.id = \"classifications-tree\";\n\n/**\n * Label of the component. May be used when a creating a `TreeDefinition`.\n * @alpha\n */\nClassificationsTreeComponent.getLabel = () => TreeWidget.translate(\"classificationsTree.label\");\n\n/** @alpha */\nClassificationsTreeComponent.isSupportedByIModel = async (imodel: IModelConnection) => {\n  try {\n    return !!(await imodel.schemaContext.getSchemaInfo(new SchemaKey(\"ClassificationSystems\"), SchemaMatchType.Latest));\n  } catch {\n    return false;\n  }\n};\n\nfunction ClassificationsTreeComponentImpl({\n  iModel,\n  viewport,\n  onPerformanceMeasured,\n  onFeatureUsed,\n  searchText,\n  treeLabel,\n  ...treeProps\n}: ClassificationsTreeComponentProps & { iModel: IModelConnection; viewport: TreeWidgetViewport }) {\n  return (\n    <TelemetryContextProvider componentIdentifier={ClassificationsTreeComponent.id} onFeatureUsed={onFeatureUsed} onPerformanceMeasured={onPerformanceMeasured}>\n      <SelectableTree>\n        <ClassificationsTree {...treeProps} imodel={iModel} activeView={viewport} searchText={searchText} treeLabel={treeLabel} />\n      </SelectableTree>\n    </TelemetryContextProvider>\n  );\n}\n"]}
+{"version":3,"file":"ClassificationsTreeComponent.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/classifications-tree/ClassificationsTreeComponent.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,iCAAiC,EAAE,MAAM,yDAAyD,CAAC;AAC5G,OAAO,EAAE,2BAA2B,EAAE,MAAM,mDAAmD,CAAC;AAChG,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAgC/D;;;;;GAKG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,KAAwC,EAAE,EAAE;IACvF,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,2BAA2B,CAAC,EAAE,kBAAkB,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAErF,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CACL,KAAC,iCAAiC,IAAC,WAAW,EAAE,IAAI,YAClD,KAAC,gCAAgC,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAI,GACjD,CACrC,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,4BAA4B,CAAC,EAAE,GAAG,sBAAsB,CAAC;AAEzD;;;GAGG;AACH,4BAA4B,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC;AAEhG,aAAa;AACb,4BAA4B,CAAC,mBAAmB,GAAG,KAAK,EAAE,MAAwB,EAAE,EAAE;IACpF,IAAI,CAAC;QACH,OAAO,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,SAAS,CAAC,uBAAuB,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC;IACtH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC,CAAC;AAEF,SAAS,gCAAgC,CAAC,EACxC,MAAM,EACN,QAAQ,EACR,qBAAqB,EACrB,aAAa,EACb,UAAU,EACV,SAAS,EACT,GAAG,SAAS,EACmF;IAC/F,OAAO,CACL,KAAC,wBAAwB,IAAC,mBAAmB,EAAE,4BAA4B,CAAC,EAAE,EAAE,aAAa,EAAE,aAAa,EAAE,qBAAqB,EAAE,qBAAqB,YACxJ,KAAC,cAAc,cACb,KAAC,mBAAmB,OAAK,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,GAAI,GAC3G,GACQ,CAC5B,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 { useActiveIModelConnection } from \"@itwin/appui-react\";\nimport { SchemaKey, SchemaMatchType } from \"@itwin/ecschema-metadata\";\nimport { TreeWidget } from \"../../../TreeWidget.js\";\nimport { SelectableTree } from \"../../tree-header/SelectableTree.js\";\nimport { SharedTreeContextProviderInternal } from \"../common/internal/SharedTreeContextProviderInternal.js\";\nimport { useActiveTreeWidgetViewport } from \"../common/internal/UseActiveTreeWidgetViewport.js\";\nimport { TelemetryContextProvider } from \"../common/UseTelemetryContext.js\";\nimport { ClassificationsTree } from \"./ClassificationsTree.js\";\n\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { TreeWidgetViewport } from \"../common/TreeWidgetViewport.js\";\nimport type { ClassificationsTreeProps } from \"./ClassificationsTree.js\";\n\n/** @alpha */\ninterface ClassificationsTreeComponentProps extends Pick<\n  ClassificationsTreeProps,\n  | \"selectionStorage\"\n  | \"hierarchyLevelConfig\"\n  | \"selectionMode\"\n  | \"searchText\"\n  | \"emptyTreeContent\"\n  | \"getInlineActions\"\n  | \"getMenuActions\"\n  | \"getContextMenuActions\"\n  | \"getTreeItemProps\"\n  | \"hierarchyConfig\"\n  | \"getEditingProps\"\n  | \"treeLabel\"\n> {\n  /**\n   * Viewport used for visibility controls.\n   *\n   * When viewport is not provided, `IModelApp.viewManager.selectedView` will be used.\n   */\n  viewport?: TreeWidgetViewport;\n  onPerformanceMeasured?: (featureId: string, duration: number) => void;\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * A component that renders `ClassificationsTree` with active iModel and viewport.\n *\n * **Note:** Wrap tree components with a single `SharedTreeContextProvider` to improve trees' performance.\n * @alpha\n */\nexport const ClassificationsTreeComponent = (props: ClassificationsTreeComponentProps) => {\n  const iModel = useActiveIModelConnection();\n  const viewport = useActiveTreeWidgetViewport({ treeWidgetViewport: props.viewport });\n\n  if (!iModel || !viewport) {\n    return null;\n  }\n\n  return (\n    <SharedTreeContextProviderInternal showWarning={true}>\n      <ClassificationsTreeComponentImpl {...props} iModel={iModel} viewport={viewport} />\n    </SharedTreeContextProviderInternal>\n  );\n};\n\n/**\n * Id of the component. May be used when a creating a `TreeDefinition`.\n * @alpha\n */\nClassificationsTreeComponent.id = \"classifications-tree\";\n\n/**\n * Label of the component. May be used when a creating a `TreeDefinition`.\n * @alpha\n */\nClassificationsTreeComponent.getLabel = () => TreeWidget.translate(\"classificationsTree.label\");\n\n/** @alpha */\nClassificationsTreeComponent.isSupportedByIModel = async (imodel: IModelConnection) => {\n  try {\n    return !!(await imodel.schemaContext.getSchemaInfo(new SchemaKey(\"ClassificationSystems\"), SchemaMatchType.Latest));\n  } catch {\n    return false;\n  }\n};\n\nfunction ClassificationsTreeComponentImpl({\n  iModel,\n  viewport,\n  onPerformanceMeasured,\n  onFeatureUsed,\n  searchText,\n  treeLabel,\n  ...treeProps\n}: ClassificationsTreeComponentProps & { iModel: IModelConnection; viewport: TreeWidgetViewport }) {\n  return (\n    <TelemetryContextProvider componentIdentifier={ClassificationsTreeComponent.id} onFeatureUsed={onFeatureUsed} onPerformanceMeasured={onPerformanceMeasured}>\n      <SelectableTree>\n        <ClassificationsTree {...treeProps} imodel={iModel} activeView={viewport} searchText={searchText} treeLabel={treeLabel} />\n      </SelectableTree>\n    </TelemetryContextProvider>\n  );\n}\n"]}