@itwin/tree-widget-react 4.0.0-alpha.6 → 4.0.0-alpha.8

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

@@ -0,0 +1,20 @@
+/// <reference types="node" resolution-mode="require"/>
+import { BeEvent } from "@itwin/core-bentley";
+import type { Viewport } from "@itwin/core-frontend";
+/** @internal */
+export interface IVisibilityChangeEventListener extends Disposable {
+    onVisibilityChange: BeEvent<() => void>;
+    suppressChangeEvents(): void;
+    resumeChangeEvents(): void;
+}
+/** @internal */
+export declare function createVisibilityChangeEventListener(props: {
+    viewport: Viewport;
+    listeners: {
+        elements?: boolean;
+        categories?: boolean;
+        displayStyle?: boolean;
+        models?: boolean;
+    };
+}): IVisibilityChangeEventListener;
+//# sourceMappingURL=VisibilityChangeEventListener.d.ts.map

package/lib/esm/tree-widget-react/components/trees/{models-tree → common}/internal/VisibilityChangeEventListener.js

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { BeEvent } from "@itwin/core-bentley";
 /** @internal */
-export function createVisibilityChangeEventListener(viewport) {
+export function createVisibilityChangeEventListener(props) {
     const onVisibilityChange = new BeEvent();
     let pendingVisibilityChange;
     let suppressChangeEvents = 0;
@@ -17,16 +17,24 @@ export function createVisibilityChangeEventListener(viewport) {
             pendingVisibilityChange = undefined;
         });
     };
-    const listeners = [
-        viewport.onViewedCategoriesPerModelChanged.addListener(handleVisibilityChange),
-        viewport.onViewedCategoriesChanged.addListener(handleVisibilityChange),
-        viewport.onViewedModelsChanged.addListener(handleVisibilityChange),
-        viewport.onAlwaysDrawnChanged.addListener(handleVisibilityChange),
-        viewport.onNeverDrawnChanged.addListener(handleVisibilityChange),
-    ];
+    const listeners = new Array();
+    if (props.listeners.elements) {
+        listeners.push(props.viewport.onAlwaysDrawnChanged.addListener(handleVisibilityChange));
+        listeners.push(props.viewport.onNeverDrawnChanged.addListener(handleVisibilityChange));
+    }
+    if (props.listeners.categories) {
+        listeners.push(props.viewport.onViewedCategoriesPerModelChanged.addListener(handleVisibilityChange));
+        listeners.push(props.viewport.onViewedCategoriesChanged.addListener(handleVisibilityChange));
+    }
+    if (props.listeners.displayStyle) {
+        listeners.push(props.viewport.onDisplayStyleChanged.addListener(handleVisibilityChange));
+    }
+    if (props.listeners.models) {
+        listeners.push(props.viewport.onViewedModelsChanged.addListener(handleVisibilityChange));
+    }
     return {
         onVisibilityChange,
-        dispose: () => {
+        [Symbol.dispose]() {
             if (pendingVisibilityChange) {
                 clearTimeout(pendingVisibilityChange);
                 pendingVisibilityChange = undefined;

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

@@ -0,0 +1 @@
+{"version":3,"file":"VisibilityChangeEventListener.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/VisibilityChangeEventListener.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAW9C,gBAAgB;AAChB,MAAM,UAAU,mCAAmC,CAAC,KAQnD;IACC,MAAM,kBAAkB,GAAG,IAAI,OAAO,EAAc,CAAC;IACrD,IAAI,uBAAkE,CAAC;IACvE,IAAI,oBAAoB,GAAW,CAAC,CAAC;IACrC,MAAM,sBAAsB,GAAG,GAAG,EAAE;QAClC,IAAI,uBAAuB,IAAI,oBAAoB,GAAG,CAAC,EAAE;YACvD,OAAO;SACR;QACD,uBAAuB,GAAG,UAAU,CAAC,GAAG,EAAE;YACxC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YAChC,uBAAuB,GAAG,SAAS,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,KAAK,EAAc,CAAC;IAE1C,IAAI,KAAK,CAAC,SAAS,CAAC,QAAQ,EAAE;QAC5B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACxF,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;KACxF;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE;QAC9B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,iCAAiC,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACrG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,yBAAyB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;KAC9F;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,YAAY,EAAE;QAChC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;KAC1F;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE;QAC1B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;KAC1F;IAED,OAAO;QACL,kBAAkB;QAClB,CAAC,MAAM,CAAC,OAAO,CAAC;YACd,IAAI,uBAAuB,EAAE;gBAC3B,YAAY,CAAC,uBAAuB,CAAC,CAAC;gBACtC,uBAAuB,GAAG,SAAS,CAAC;aACrC;YACD,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAC9B,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;QACvB,CAAC;QACD,oBAAoB,EAAE,GAAG,EAAE;YACzB,oBAAoB,EAAE,CAAC;QACzB,CAAC;QACD,kBAAkB,EAAE,GAAG,EAAE;YACvB,oBAAoB,EAAE,CAAC;YACvB,IAAI,oBAAoB,KAAK,CAAC,EAAE;gBAC9B,sBAAsB,EAAE,CAAC;aAC1B;QACH,CAAC;KACF,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { BeEvent } from \"@itwin/core-bentley\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\n\n/** @internal */\nexport interface IVisibilityChangeEventListener extends Disposable {\n  onVisibilityChange: BeEvent<() => void>;\n  suppressChangeEvents(): void;\n  resumeChangeEvents(): void;\n}\n\n/** @internal */\nexport function createVisibilityChangeEventListener(props: {\n  viewport: Viewport;\n  listeners: {\n    elements?: boolean;\n    categories?: boolean;\n    displayStyle?: boolean;\n    models?: boolean;\n  };\n}): IVisibilityChangeEventListener {\n  const onVisibilityChange = new BeEvent<() => void>();\n  let pendingVisibilityChange: undefined | ReturnType<typeof setTimeout>;\n  let suppressChangeEvents: number = 0;\n  const handleVisibilityChange = () => {\n    if (pendingVisibilityChange || suppressChangeEvents > 0) {\n      return;\n    }\n    pendingVisibilityChange = setTimeout(() => {\n      onVisibilityChange.raiseEvent();\n      pendingVisibilityChange = undefined;\n    });\n  };\n\n  const listeners = new Array<() => void>();\n\n  if (props.listeners.elements) {\n    listeners.push(props.viewport.onAlwaysDrawnChanged.addListener(handleVisibilityChange));\n    listeners.push(props.viewport.onNeverDrawnChanged.addListener(handleVisibilityChange));\n  }\n  if (props.listeners.categories) {\n    listeners.push(props.viewport.onViewedCategoriesPerModelChanged.addListener(handleVisibilityChange));\n    listeners.push(props.viewport.onViewedCategoriesChanged.addListener(handleVisibilityChange));\n  }\n  if (props.listeners.displayStyle) {\n    listeners.push(props.viewport.onDisplayStyleChanged.addListener(handleVisibilityChange));\n  }\n  if (props.listeners.models) {\n    listeners.push(props.viewport.onViewedModelsChanged.addListener(handleVisibilityChange));\n  }\n\n  return {\n    onVisibilityChange,\n    [Symbol.dispose]() {\n      if (pendingVisibilityChange) {\n        clearTimeout(pendingVisibilityChange);\n        pendingVisibilityChange = undefined;\n      }\n      listeners.forEach((x) => x());\n      listeners.length = 0;\n    },\n    suppressChangeEvents: () => {\n      suppressChangeEvents++;\n    },\n    resumeChangeEvents: () => {\n      suppressChangeEvents--;\n      if (suppressChangeEvents === 0) {\n        handleVisibilityChange();\n      }\n    },\n  };\n}\n"]}

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

@@ -0,0 +1,63 @@
+import type { Viewport } from "@itwin/core-frontend";
+import type { Observable, OperatorFunction } from "rxjs";
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { NonPartialVisibilityStatus } from "./Tooltip.js";
+import type { VisibilityStatus } from "../UseHierarchyVisibility.js";
+import type { ElementId, ModelId } from "./Types.js";
+import type { CategoryInfo } from "../CategoriesVisibilityUtils.js";
+/** @internal */
+export declare function mergeVisibilityStatuses(obs: Observable<VisibilityStatus>): Observable<VisibilityStatus>;
+/** @internal */
+export declare function getSubModeledElementsVisibilityStatus({ parentNodeVisibilityStatus, getModelVisibilityStatus, }: {
+    parentNodeVisibilityStatus: VisibilityStatus;
+    getModelVisibilityStatus: ({ modelId }: {
+        modelId: Id64String;
+    }) => Observable<VisibilityStatus>;
+}): OperatorFunction<Id64Array, VisibilityStatus>;
+/** @internal */
+export declare function filterSubModeledElementIds({ doesSubModelExist, }: {
+    doesSubModelExist: (elementId: Id64String) => Promise<boolean>;
+}): OperatorFunction<Array<ElementId>, Array<ModelId>>;
+/** @internal */
+export declare function changeElementStateNoChildrenOperator(props: {
+    on: boolean;
+    isDisplayedByDefault: boolean;
+    viewport: Viewport;
+}): OperatorFunction<string, void>;
+/** @internal */
+export declare function getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props: {
+    alwaysDrawn: Set<ElementId> | undefined;
+    neverDrawn: Set<ElementId> | undefined;
+    totalCount: number;
+    viewport: Viewport;
+} & GetVisibilityFromAlwaysAndNeverDrawnElementsProps): VisibilityStatus;
+/** @internal */
+export declare function getElementOverriddenVisibility(props: {
+    elementId: Id64String;
+    viewport: Viewport;
+}): NonPartialVisibilityStatus | undefined;
+/** @internal */
+export interface GetVisibilityFromAlwaysAndNeverDrawnElementsProps {
+    /** Status when always/never lists are empty and exclusive mode is off */
+    defaultStatus: () => VisibilityStatus;
+}
+/** @internal */
+export declare function getElementVisibility(viewsModel: boolean, overridenVisibility: NonPartialVisibilityStatus | undefined, categoryVisibility: NonPartialVisibilityStatus, subModelVisibilityStatus?: VisibilityStatus): VisibilityStatus;
+/**
+ * Toggles visibility of categories to show or hide.
+ * @internal
+ */
+export declare function toggleAllCategories(viewport: Viewport, display: boolean): Promise<void>;
+/**
+ * Changes category display in the viewport.
+ * @internal
+ */
+export declare function enableCategoryDisplay(viewport: Viewport, categoryIds: Id64Array, enabled: boolean, enableAllSubCategories?: boolean): Promise<void>;
+/**
+ * Changes subcategory display in the viewport
+ * @internal
+ */
+export declare function enableSubCategoryDisplay(viewport: Viewport, subCategoryId: Id64String, enabled: boolean): void;
+/** @internal */
+export declare function loadCategoriesFromViewport(vp: Viewport): Promise<CategoryInfo[]>;
+//# sourceMappingURL=VisibilityUtils.d.ts.map

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

@@ -0,0 +1,210 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { filter, from, map, mergeAll, mergeMap, of, reduce, startWith, toArray } from "rxjs";
+import { QueryRowFormat } from "@itwin/core-common";
+import { PerModelCategoryVisibility } from "@itwin/core-frontend";
+import { reduceWhile } from "./Rxjs.js";
+import { createVisibilityStatus } from "./Tooltip.js";
+import { getClassesByView, releaseMainThreadOnItemsCount } from "./Utils.js";
+function mergeVisibilities(obs) {
+    return obs.pipe(reduceWhile((x) => x.allVisible || x.allHidden, (acc, val) => {
+        acc.allVisible &&= val === "visible";
+        acc.allHidden &&= val === "hidden";
+        return acc;
+    }, { allVisible: true, allHidden: true }), map((x) => {
+        if (!x) {
+            return "empty";
+        }
+        return x.allVisible ? "visible" : x.allHidden ? "hidden" : "partial";
+    }));
+}
+/** @internal */
+export function mergeVisibilityStatuses(obs) {
+    return obs.pipe(map((visibilityStatus) => visibilityStatus.state), mergeVisibilities, map((visibility) => {
+        if (visibility === "empty") {
+            visibility = "visible";
+        }
+        return createVisibilityStatus(visibility);
+    }));
+}
+/** @internal */
+export function getSubModeledElementsVisibilityStatus({ parentNodeVisibilityStatus, getModelVisibilityStatus, }) {
+    return (obs) => {
+        return obs.pipe(
+        // combine visibility status of sub-models with visibility status of parent node
+        mergeMap((modeledElementIds) => {
+            if (modeledElementIds.length === 0) {
+                return of(parentNodeVisibilityStatus);
+            }
+            return from(modeledElementIds).pipe(mergeMap((modeledElementId) => getModelVisibilityStatus({ modelId: modeledElementId })), startWith(parentNodeVisibilityStatus), mergeVisibilityStatuses);
+        }));
+    };
+}
+/** @internal */
+export function filterSubModeledElementIds({ doesSubModelExist, }) {
+    return (obs) => {
+        return obs.pipe(mergeAll(), mergeMap(async (elementId) => ({ elementId, hasSubModel: await doesSubModelExist(elementId) })), filter(({ hasSubModel }) => hasSubModel), map(({ elementId }) => elementId), toArray());
+    };
+}
+/** @internal */
+export function changeElementStateNoChildrenOperator(props) {
+    return (elementIds) => {
+        const { on, isDisplayedByDefault } = props;
+        const isAlwaysDrawnExclusive = props.viewport.isAlwaysDrawnExclusive;
+        return elementIds.pipe(releaseMainThreadOnItemsCount(500), reduce((acc, elementId) => {
+            if (acc.alwaysDrawn === undefined || acc.neverDrawn === undefined) {
+                acc.alwaysDrawn = new Set(props.viewport.alwaysDrawn || []);
+                acc.neverDrawn = new Set(props.viewport.neverDrawn || []);
+            }
+            if (on) {
+                const wasRemoved = acc.neverDrawn.delete(elementId);
+                acc.changedNeverDrawn ||= wasRemoved;
+                // If exclusive mode is enabled, we must add the element to the always drawn list.
+                if ((!isDisplayedByDefault || isAlwaysDrawnExclusive) && !acc.alwaysDrawn.has(elementId)) {
+                    acc.alwaysDrawn.add(elementId);
+                    acc.changedAlwaysDrawn = true;
+                }
+            }
+            else {
+                const wasRemoved = acc.alwaysDrawn.delete(elementId);
+                acc.changedAlwaysDrawn ||= wasRemoved;
+                // If exclusive mode is not enabled, we have to add the element to the never drawn list.
+                if (isDisplayedByDefault && !isAlwaysDrawnExclusive && !acc.neverDrawn.has(elementId)) {
+                    acc.neverDrawn.add(elementId);
+                    acc.changedNeverDrawn = true;
+                }
+            }
+            return acc;
+        }, {
+            changedNeverDrawn: false,
+            changedAlwaysDrawn: false,
+            neverDrawn: undefined,
+            alwaysDrawn: undefined,
+        }), map((state) => {
+            state.changedNeverDrawn && state.neverDrawn && props.viewport.setNeverDrawn(state.neverDrawn);
+            state.changedAlwaysDrawn && state.alwaysDrawn && props.viewport.setAlwaysDrawn(state.alwaysDrawn, props.viewport.isAlwaysDrawnExclusive);
+        }));
+    };
+}
+/** @internal */
+export function getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props) {
+    const { alwaysDrawn, neverDrawn, totalCount, viewport } = props;
+    if (neverDrawn?.size === totalCount) {
+        return createVisibilityStatus("hidden");
+    }
+    if (alwaysDrawn?.size === totalCount) {
+        return createVisibilityStatus("visible");
+    }
+    if (viewport.isAlwaysDrawnExclusive) {
+        return createVisibilityStatus(alwaysDrawn?.size ? "partial" : "hidden");
+    }
+    const status = props.defaultStatus();
+    if ((status.state === "visible" && neverDrawn?.size) || (status.state === "hidden" && alwaysDrawn?.size)) {
+        return createVisibilityStatus("partial");
+    }
+    return status;
+}
+/** @internal */
+export function getElementOverriddenVisibility(props) {
+    const { viewport, elementId } = props;
+    if (viewport.neverDrawn?.has(elementId)) {
+        return createVisibilityStatus("hidden");
+    }
+    if (viewport.alwaysDrawn?.has(elementId)) {
+        return createVisibilityStatus("visible");
+    }
+    if (viewport.isAlwaysDrawnExclusive) {
+        return createVisibilityStatus("hidden");
+    }
+    return undefined;
+}
+/** @internal */
+export function getElementVisibility(viewsModel, overridenVisibility, categoryVisibility, subModelVisibilityStatus) {
+    if (subModelVisibilityStatus?.state === "partial") {
+        return createVisibilityStatus("partial");
+    }
+    if (!viewsModel) {
+        return createVisibilityStatus(subModelVisibilityStatus?.state === "visible" ? "partial" : "hidden");
+    }
+    const elementVisibilityWithoutSubModels = overridenVisibility ?? categoryVisibility;
+    if (!subModelVisibilityStatus) {
+        return elementVisibilityWithoutSubModels;
+    }
+    if (elementVisibilityWithoutSubModels.state === subModelVisibilityStatus.state) {
+        return elementVisibilityWithoutSubModels;
+    }
+    return createVisibilityStatus("partial");
+}
+/**
+ * Toggles visibility of categories to show or hide.
+ * @internal
+ */
+export async function toggleAllCategories(viewport, display) {
+    const categoryIds = await getCategoryIds(viewport);
+    if (categoryIds.length === 0) {
+        return;
+    }
+    await enableCategoryDisplay(viewport, categoryIds, display);
+}
+/**
+ * Gets ids of all categories from specified imodel and viewport.
+ * @internal
+ */
+async function getCategoryIds(viewport) {
+    const categories = await loadCategoriesFromViewport(viewport);
+    return categories.map((category) => category.categoryId);
+}
+/**
+ * Changes category display in the viewport.
+ * @internal
+ */
+export async function enableCategoryDisplay(viewport, categoryIds, enabled, enableAllSubCategories = true) {
+    viewport.changeCategoryDisplay(categoryIds, enabled, enableAllSubCategories);
+    // remove category overrides per model
+    const modelsContainingOverrides = new Array();
+    for (const ovr of viewport.perModelCategoryVisibility) {
+        if (categoryIds.findIndex((id) => id === ovr.categoryId) !== -1) {
+            modelsContainingOverrides.push(ovr.modelId);
+        }
+    }
+    viewport.perModelCategoryVisibility.setOverride(modelsContainingOverrides, categoryIds, PerModelCategoryVisibility.Override.None);
+    // changeCategoryDisplay only enables subcategories, it does not disabled them. So we must do that ourselves.
+    if (false === enabled) {
+        (await viewport.iModel.categories.getCategoryInfo(categoryIds)).forEach((categoryInfo) => {
+            categoryInfo.subCategories.forEach((value) => enableSubCategoryDisplay(viewport, value.id, false));
+        });
+    }
+}
+/**
+ * Changes subcategory display in the viewport
+ * @internal
+ */
+export function enableSubCategoryDisplay(viewport, subCategoryId, enabled) {
+    viewport.changeSubCategoryDisplay(subCategoryId, enabled);
+}
+/** @internal */
+export async function loadCategoriesFromViewport(vp) {
+    // Query categories and add them to state
+    const { categoryClass, elementClass } = getClassesByView(vp.view.is3d() ? "3d" : "2d");
+    const ecsql = `
+    SELECT ECInstanceId as id
+    FROM ${categoryClass}
+    WHERE
+      ECInstanceId IN (
+        SELECT DISTINCT Category.Id
+        FROM ${elementClass}
+        WHERE
+          Category.Id IN (SELECT ECInstanceId FROM ${categoryClass})
+          ${vp.view.is3d() ? "" : "AND Model.Id=?"}
+      )
+  `;
+    const categories = [];
+    const rows = await vp.iModel.createQueryReader(ecsql, undefined, { rowFormat: QueryRowFormat.UseJsPropertyNames }).toArray();
+    (await vp.iModel.categories.getCategoryInfo(rows.map((row) => row.id))).forEach((val) => {
+        categories.push({ categoryId: val.id, subCategoryIds: val.subCategories.size ? [...val.subCategories.keys()] : undefined });
+    });
+    return categories;
+}
+//# sourceMappingURL=VisibilityUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"VisibilityUtils.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/VisibilityUtils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC7F,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,6BAA6B,EAAE,MAAM,YAAY,CAAC;AAU7E,SAAS,iBAAiB,CAAC,GAA2B;IACpD,OAAO,GAAG,CAAC,IAAI,CACb,WAAW,CACT,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,SAAS,EAClC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACX,GAAG,CAAC,UAAU,KAAK,GAAG,KAAK,SAAS,CAAC;QACrC,GAAG,CAAC,SAAS,KAAK,GAAG,KAAK,QAAQ,CAAC;QACnC,OAAO,GAAG,CAAC;IACb,CAAC,EACD,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CACtC,EACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACR,IAAI,CAAC,CAAC,EAAE;YACN,OAAO,OAAO,CAAC;SAChB;QACD,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IACvE,CAAC,CAAC,CACH,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,uBAAuB,CAAC,GAAiC;IACvE,OAAO,GAAG,CAAC,IAAI,CACb,GAAG,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EACjD,iBAAiB,EACjB,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;QACjB,IAAI,UAAU,KAAK,OAAO,EAAE;YAC1B,UAAU,GAAG,SAAS,CAAC;SACxB;QACD,OAAO,sBAAsB,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC,CAAC,CACH,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,qCAAqC,CAAC,EACpD,0BAA0B,EAC1B,wBAAwB,GAIzB;IACC,OAAO,CAAC,GAAG,EAAE,EAAE;QACb,OAAO,GAAG,CAAC,IAAI;QACb,gFAAgF;QAChF,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE;YAC7B,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE;gBAClC,OAAO,EAAE,CAAC,0BAA0B,CAAC,CAAC;aACvC;YACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,wBAAwB,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,EACvF,SAAS,CAAmB,0BAA0B,CAAC,EACvD,uBAAuB,CACxB,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,0BAA0B,CAAC,EACzC,iBAAiB,GAGlB;IACC,OAAO,CAAC,GAAG,EAAE,EAAE;QACb,OAAO,GAAG,CAAC,IAAI,CACb,QAAQ,EAAE,EACV,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,EAC/F,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,EACxC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,EACjC,OAAO,EAAE,CACV,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,oCAAoC,CAAC,KAIpD;IACC,OAAO,CAAC,UAAkC,EAAE,EAAE;QAC5C,MAAM,EAAE,EAAE,EAAE,oBAAoB,EAAE,GAAG,KAAK,CAAC;QAC3C,MAAM,sBAAsB,GAAG,KAAK,CAAC,QAAQ,CAAC,sBAAsB,CAAC;QACrE,OAAO,UAAU,CAAC,IAAI,CACpB,6BAA6B,CAAC,GAAG,CAAC,EAClC,MAAM,CAIJ,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACjB,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,GAAG,CAAC,UAAU,KAAK,SAAS,EAAE;gBACjE,GAAG,CAAC,WAAW,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBAC5D,GAAG,CAAC,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;aAC3D;YACD,IAAI,EAAE,EAAE;gBACN,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACpD,GAAG,CAAC,iBAAiB,KAAK,UAAU,CAAC;gBACrC,kFAAkF;gBAClF,IAAI,CAAC,CAAC,oBAAoB,IAAI,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;oBACxF,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC/B,GAAG,CAAC,kBAAkB,GAAG,IAAI,CAAC;iBAC/B;aACF;iBAAM;gBACL,MAAM,UAAU,GAAG,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACrD,GAAG,CAAC,kBAAkB,KAAK,UAAU,CAAC;gBACtC,wFAAwF;gBACxF,IAAI,oBAAoB,IAAI,CAAC,sBAAsB,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;oBACrF,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC9B,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC;iBAC9B;aACF;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD;YACE,iBAAiB,EAAE,KAAK;YACxB,kBAAkB,EAAE,KAAK;YACzB,UAAU,EAAE,SAAS;YACrB,WAAW,EAAE,SAAS;SACvB,CACF,EACD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACZ,KAAK,CAAC,iBAAiB,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC9F,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QAC3I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,gDAAgD,CAC9D,KAKqD;IAErD,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAEhE,IAAI,UAAU,EAAE,IAAI,KAAK,UAAU,EAAE;QACnC,OAAO,sBAAsB,CAAC,QAAQ,CAAC,CAAC;KACzC;IAED,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU,EAAE;QACpC,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;KAC1C;IAED,IAAI,QAAQ,CAAC,sBAAsB,EAAE;QACnC,OAAQ,sBAAsB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAA;KACzE;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ,IAAI,WAAW,EAAE,IAAI,CAAC,EAAE;QACxG,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;KAC1C;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,8BAA8B,CAAC,KAAoD;IACjG,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IACtC,IAAI,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,EAAE;QACvC,OAAO,sBAAsB,CAAC,QAAQ,CAAC,CAAC;KACzC;IACD,IAAI,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,EAAE;QACxC,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;KAC1C;IAED,IAAI,QAAQ,CAAC,sBAAsB,EAAE;QACnC,OAAO,sBAAsB,CAAC,QAAQ,CAAC,CAAC;KACzC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAQD,gBAAgB;AAChB,MAAM,UAAU,oBAAoB,CAClC,UAAmB,EACnB,mBAA2D,EAC3D,kBAA8C,EAC9C,wBAA2C;IAE3C,IAAI,wBAAwB,EAAE,KAAK,KAAK,SAAS,EAAE;QACjD,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;KAC1C;IAED,IAAI,CAAC,UAAU,EAAE;QACf,OAAO,sBAAsB,CAAC,wBAAwB,EAAE,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;KACrG;IAED,MAAM,iCAAiC,GAAG,mBAAmB,IAAI,kBAAkB,CAAC;IACpF,IAAI,CAAC,wBAAwB,EAAE;QAC7B,OAAO,iCAAiC,CAAC;KAC1C;IAED,IAAI,iCAAiC,CAAC,KAAK,KAAK,wBAAwB,CAAC,KAAK,EAAE;QAC9E,OAAO,iCAAiC,CAAC;KAC1C;IAED,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;AAC3C,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,QAAkB,EAAE,OAAgB;IAC5E,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;IACnD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE;QAC5B,OAAO;KACR;IAED,MAAM,qBAAqB,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;AAC9D,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,cAAc,CAAC,QAAkB;IAC9C,MAAM,UAAU,GAAG,MAAM,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IAC9D,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;AAC3D,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAAkB,EAAE,WAAsB,EAAE,OAAgB,EAAE,sBAAsB,GAAG,IAAI;IACrI,QAAQ,CAAC,qBAAqB,CAAC,WAAW,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IAE7E,sCAAsC;IACtC,MAAM,yBAAyB,GAAG,IAAI,KAAK,EAAW,CAAC;IACvD,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,0BAA0B,EAAE;QACrD,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE;YAC/D,yBAAyB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC7C;KACF;IACD,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,yBAAyB,EAAE,WAAW,EAAE,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAElI,6GAA6G;IAC7G,IAAI,KAAK,KAAK,OAAO,EAAE;QACrB,CAAC,MAAM,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;YACvF,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QACrG,CAAC,CAAC,CAAC;KACJ;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAkB,EAAE,aAAyB,EAAE,OAAgB;IACtG,QAAQ,CAAC,wBAAwB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;AAC5D,CAAC;AAED,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAY;IAC3D,yCAAyC;IACzC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,GAAG,gBAAgB,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACvF,MAAM,KAAK,GAAG;;WAEL,aAAa;;;;eAIT,YAAY;;qDAE0B,aAAa;YACtD,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,gBAAgB;;GAE/C,CAAC;IAEF,MAAM,UAAU,GAAmB,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC7H,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;QACtF,UAAU,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;IAC9H,CAAC,CAAC,CAAC;IACH,OAAO,UAAU,CAAC;AACpB,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 { filter, from, map, mergeAll, mergeMap, of, reduce, startWith, toArray } from \"rxjs\";\nimport { QueryRowFormat } from \"@itwin/core-common\";\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\nimport { reduceWhile } from \"./Rxjs.js\";\nimport { createVisibilityStatus } from \"./Tooltip.js\";\nimport { getClassesByView, releaseMainThreadOnItemsCount } from \"./Utils.js\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { Observable, OperatorFunction } from \"rxjs\";\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { NonPartialVisibilityStatus, Visibility } from \"./Tooltip.js\";\nimport type { VisibilityStatus } from \"../UseHierarchyVisibility.js\";\nimport type { ElementId, ModelId } from \"./Types.js\";\nimport type { CategoryInfo } from \"../CategoriesVisibilityUtils.js\";\n\nfunction mergeVisibilities(obs: Observable<Visibility>): Observable<Visibility | \"empty\"> {\n  return obs.pipe(\n    reduceWhile(\n      (x) => x.allVisible || x.allHidden,\n      (acc, val) => {\n        acc.allVisible &&= val === \"visible\";\n        acc.allHidden &&= val === \"hidden\";\n        return acc;\n      },\n      { allVisible: true, allHidden: true },\n    ),\n    map((x) => {\n      if (!x) {\n        return \"empty\";\n      }\n      return x.allVisible ? \"visible\" : x.allHidden ? \"hidden\" : \"partial\";\n    }),\n  );\n}\n\n/** @internal */\nexport function mergeVisibilityStatuses(obs: Observable<VisibilityStatus>): Observable<VisibilityStatus> {\n  return obs.pipe(\n    map((visibilityStatus) => visibilityStatus.state),\n    mergeVisibilities,\n    map((visibility) => {\n      if (visibility === \"empty\") {\n        visibility = \"visible\";\n      }\n      return createVisibilityStatus(visibility);\n    }),\n  );\n}\n\n/** @internal */\nexport function getSubModeledElementsVisibilityStatus({\n  parentNodeVisibilityStatus,\n  getModelVisibilityStatus,\n}: {\n  parentNodeVisibilityStatus: VisibilityStatus;\n  getModelVisibilityStatus: ({ modelId }: { modelId: Id64String }) => Observable<VisibilityStatus>;\n}): OperatorFunction<Id64Array, VisibilityStatus> {\n  return (obs) => {\n    return obs.pipe(\n      // combine visibility status of sub-models with visibility status of parent node\n      mergeMap((modeledElementIds) => {\n        if (modeledElementIds.length === 0) {\n          return of(parentNodeVisibilityStatus);\n        }\n        return from(modeledElementIds).pipe(\n          mergeMap((modeledElementId) => getModelVisibilityStatus({ modelId: modeledElementId })),\n          startWith<VisibilityStatus>(parentNodeVisibilityStatus),\n          mergeVisibilityStatuses,\n        );\n      }),\n    );\n  };\n}\n\n/** @internal */\nexport function filterSubModeledElementIds({\n  doesSubModelExist,\n}: {\n  doesSubModelExist: (elementId: Id64String) => Promise<boolean>;\n}): OperatorFunction<Array<ElementId>, Array<ModelId>> {\n  return (obs) => {\n    return obs.pipe(\n      mergeAll(),\n      mergeMap(async (elementId) => ({ elementId, hasSubModel: await doesSubModelExist(elementId) })),\n      filter(({ hasSubModel }) => hasSubModel),\n      map(({ elementId }) => elementId),\n      toArray(),\n    );\n  };\n}\n\n/** @internal */\nexport function changeElementStateNoChildrenOperator(props: {\n  on: boolean;\n  isDisplayedByDefault: boolean;\n  viewport: Viewport;\n}): OperatorFunction<string, void> {\n  return (elementIds: Observable<Id64String>) => {\n    const { on, isDisplayedByDefault } = props;\n    const isAlwaysDrawnExclusive = props.viewport.isAlwaysDrawnExclusive;\n    return elementIds.pipe(\n      releaseMainThreadOnItemsCount(500),\n      reduce<\n        string,\n        { changedNeverDrawn: boolean; changedAlwaysDrawn: boolean; neverDrawn: Set<ElementId> | undefined; alwaysDrawn: Set<ElementId> | undefined }\n      >(\n        (acc, elementId) => {\n          if (acc.alwaysDrawn === undefined || acc.neverDrawn === undefined) {\n            acc.alwaysDrawn = new Set(props.viewport.alwaysDrawn || []);\n            acc.neverDrawn = new Set(props.viewport.neverDrawn || []);\n          }\n          if (on) {\n            const wasRemoved = acc.neverDrawn.delete(elementId);\n            acc.changedNeverDrawn ||= wasRemoved;\n            // If exclusive mode is enabled, we must add the element to the always drawn list.\n            if ((!isDisplayedByDefault || isAlwaysDrawnExclusive) && !acc.alwaysDrawn.has(elementId)) {\n              acc.alwaysDrawn.add(elementId);\n              acc.changedAlwaysDrawn = true;\n            }\n          } else {\n            const wasRemoved = acc.alwaysDrawn.delete(elementId);\n            acc.changedAlwaysDrawn ||= wasRemoved;\n            // If exclusive mode is not enabled, we have to add the element to the never drawn list.\n            if (isDisplayedByDefault && !isAlwaysDrawnExclusive && !acc.neverDrawn.has(elementId)) {\n              acc.neverDrawn.add(elementId);\n              acc.changedNeverDrawn = true;\n            }\n          }\n          return acc;\n        },\n        {\n          changedNeverDrawn: false,\n          changedAlwaysDrawn: false,\n          neverDrawn: undefined,\n          alwaysDrawn: undefined,\n        },\n      ),\n      map((state) => {\n        state.changedNeverDrawn && state.neverDrawn && props.viewport.setNeverDrawn(state.neverDrawn);\n        state.changedAlwaysDrawn && state.alwaysDrawn && props.viewport.setAlwaysDrawn(state.alwaysDrawn, props.viewport.isAlwaysDrawnExclusive);\n      }),\n    );\n  };\n}\n\n/** @internal */\nexport function getVisibilityFromAlwaysAndNeverDrawnElementsImpl(\n  props: {\n    alwaysDrawn: Set<ElementId> | undefined;\n    neverDrawn: Set<ElementId> | undefined;\n    totalCount: number;\n    viewport: Viewport;\n  } & GetVisibilityFromAlwaysAndNeverDrawnElementsProps,\n): VisibilityStatus {\n  const { alwaysDrawn, neverDrawn, totalCount, viewport } = props;\n\n  if (neverDrawn?.size === totalCount) {\n    return createVisibilityStatus(\"hidden\");\n  }\n\n  if (alwaysDrawn?.size === totalCount) {\n    return createVisibilityStatus(\"visible\");\n  }\n\n  if (viewport.isAlwaysDrawnExclusive) {\n    return  createVisibilityStatus(alwaysDrawn?.size ? \"partial\" : \"hidden\")\n  }\n\n  const status = props.defaultStatus();\n  if ((status.state === \"visible\" && neverDrawn?.size) || (status.state === \"hidden\" && alwaysDrawn?.size)) {\n    return createVisibilityStatus(\"partial\");\n  }\n  return status;\n}\n\n/** @internal */\nexport function getElementOverriddenVisibility(props: { elementId: Id64String; viewport: Viewport }): NonPartialVisibilityStatus | undefined {\n  const { viewport, elementId } = props;\n  if (viewport.neverDrawn?.has(elementId)) {\n    return createVisibilityStatus(\"hidden\");\n  }\n  if (viewport.alwaysDrawn?.has(elementId)) {\n    return createVisibilityStatus(\"visible\");\n  }\n\n  if (viewport.isAlwaysDrawnExclusive) {\n    return createVisibilityStatus(\"hidden\");\n  }\n\n  return undefined;\n}\n\n/** @internal */\nexport interface GetVisibilityFromAlwaysAndNeverDrawnElementsProps {\n  /** Status when always/never lists are empty and exclusive mode is off */\n  defaultStatus: () => VisibilityStatus;\n}\n\n/** @internal */\nexport function getElementVisibility(\n  viewsModel: boolean,\n  overridenVisibility: NonPartialVisibilityStatus | undefined,\n  categoryVisibility: NonPartialVisibilityStatus,\n  subModelVisibilityStatus?: VisibilityStatus,\n): VisibilityStatus {\n  if (subModelVisibilityStatus?.state === \"partial\") {\n    return createVisibilityStatus(\"partial\");\n  }\n\n  if (!viewsModel) {\n    return createVisibilityStatus(subModelVisibilityStatus?.state === \"visible\" ? \"partial\" : \"hidden\");\n  }\n\n  const elementVisibilityWithoutSubModels = overridenVisibility ?? categoryVisibility;\n  if (!subModelVisibilityStatus) {\n    return elementVisibilityWithoutSubModels;\n  }\n\n  if (elementVisibilityWithoutSubModels.state === subModelVisibilityStatus.state) {\n    return elementVisibilityWithoutSubModels;\n  }\n\n  return createVisibilityStatus(\"partial\");\n}\n\n/**\n * Toggles visibility of categories to show or hide.\n * @internal\n */\nexport async function toggleAllCategories(viewport: Viewport, display: boolean) {\n  const categoryIds = await getCategoryIds(viewport);\n  if (categoryIds.length === 0) {\n    return;\n  }\n\n  await enableCategoryDisplay(viewport, categoryIds, display);\n}\n\n/**\n * Gets ids of all categories from specified imodel and viewport.\n * @internal\n */\nasync function getCategoryIds(viewport: Viewport): Promise<Id64Array> {\n  const categories = await loadCategoriesFromViewport(viewport);\n  return categories.map((category) => category.categoryId);\n}\n\n/**\n * Changes category display in the viewport.\n * @internal\n */\nexport async function enableCategoryDisplay(viewport: Viewport, categoryIds: Id64Array, enabled: boolean, enableAllSubCategories = true) {\n  viewport.changeCategoryDisplay(categoryIds, enabled, enableAllSubCategories);\n\n  // remove category overrides per model\n  const modelsContainingOverrides = new Array<ModelId>();\n  for (const ovr of viewport.perModelCategoryVisibility) {\n    if (categoryIds.findIndex((id) => id === ovr.categoryId) !== -1) {\n      modelsContainingOverrides.push(ovr.modelId);\n    }\n  }\n  viewport.perModelCategoryVisibility.setOverride(modelsContainingOverrides, categoryIds, PerModelCategoryVisibility.Override.None);\n\n  // changeCategoryDisplay only enables subcategories, it does not disabled them. So we must do that ourselves.\n  if (false === enabled) {\n    (await viewport.iModel.categories.getCategoryInfo(categoryIds)).forEach((categoryInfo) => {\n      categoryInfo.subCategories.forEach((value) => enableSubCategoryDisplay(viewport, value.id, false));\n    });\n  }\n}\n\n/**\n * Changes subcategory display in the viewport\n * @internal\n */\nexport function enableSubCategoryDisplay(viewport: Viewport, subCategoryId: Id64String, enabled: boolean) {\n  viewport.changeSubCategoryDisplay(subCategoryId, enabled);\n}\n\n/** @internal */\nexport async function loadCategoriesFromViewport(vp: Viewport) {\n  // Query categories and add them to state\n  const { categoryClass, elementClass } = getClassesByView(vp.view.is3d() ? \"3d\" : \"2d\");\n  const ecsql = `\n    SELECT ECInstanceId as id\n    FROM ${categoryClass}\n    WHERE\n      ECInstanceId IN (\n        SELECT DISTINCT Category.Id\n        FROM ${elementClass}\n        WHERE\n          Category.Id IN (SELECT ECInstanceId FROM ${categoryClass})\n          ${vp.view.is3d() ? \"\" : \"AND Model.Id=?\"}\n      )\n  `;\n\n  const categories: CategoryInfo[] = [];\n\n  const rows = await vp.iModel.createQueryReader(ecsql, undefined, { rowFormat: QueryRowFormat.UseJsPropertyNames }).toArray();\n  (await vp.iModel.categories.getCategoryInfo(rows.map((row) => row.id))).forEach((val) => {\n    categories.push({ categoryId: val.id, subCategoryIds: val.subCategories.size ? [...val.subCategories.keys()] : undefined });\n  });\n  return categories;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTree.d.ts

@@ -2,13 +2,13 @@ import type { PresentationHierarchyNode } from "@itwin/presentation-hierarchies-
 import type { BaseTreeRendererProps } from "../common/components/BaseTreeRenderer.js";
 import type { TreeProps } from "../common/components/Tree.js";
 /** @beta */
-export type ExternalSourcesTreeProps = Pick<TreeProps, "imodel" | "getSchemaContext" | "selectionStorage" | "selectionMode" | "emptyTreeContent"> & Pick<BaseTreeRendererProps, "actions" | "getDecorations"> & {
+export type ExternalSourcesTreeProps = Pick<TreeProps, "imodel" | "getSchemaContext" | "selectionStorage" | "selectionMode" | "emptyTreeContent"> & Pick<BaseTreeRendererProps, "getActions" | "getDecorations"> & {
     hierarchyLevelConfig?: {
         sizeLimit?: number;
     };
 };
 /** @beta */
-export declare function ExternalSourcesTree({ actions, getDecorations, selectionMode, ...rest }: ExternalSourcesTreeProps): import("react/jsx-runtime").JSX.Element;
+export declare function ExternalSourcesTree({ getActions, getDecorations, selectionMode, ...rest }: ExternalSourcesTreeProps): import("react/jsx-runtime").JSX.Element;
 /** @beta */
 export declare function ExternalSourcesTreeIcon({ node }: {
     node: PresentationHierarchyNode;

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTree.js

@@ -3,6 +3,10 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
+import classSvg from "@itwin/itwinui-icons/bis-class.svg";
+import elementSvg from "@itwin/itwinui-icons/bis-element.svg";
+import documentSvg from "@itwin/itwinui-icons/document.svg";
+import ecSchemaSvg from "@itwin/itwinui-icons/selection-children.svg";
 import { Icon } from "@itwin/itwinui-react/bricks";
 import { EmptyTreeContent } from "../common/components/EmptyTree.js";
 import { Tree } from "../common/components/Tree.js";
@@ -10,23 +14,19 @@ import { TreeRenderer } from "../common/components/TreeRenderer.js";
 import { ExternalSourcesTreeComponent } from "./ExternalSourcesTreeComponent.js";
 import { ExternalSourcesTreeDefinition } from "./ExternalSourcesTreeDefinition.js";
 /** @beta */
-export function ExternalSourcesTree({ actions, getDecorations, selectionMode, ...rest }) {
-    return (_jsx(Tree, { emptyTreeContent: _jsx(EmptyTreeContent, { icon: documentSvg }), ...rest, treeName: ExternalSourcesTreeComponent.id, getHierarchyDefinition: getDefinitionsProvider, selectionMode: selectionMode ?? "none", treeRenderer: (treeProps) => (_jsx(TreeRenderer, { ...treeProps, actions: actions, getDecorations: getDecorations ?? ((node) => _jsx(ExternalSourcesTreeIcon, { node: node })) })) }));
+export function ExternalSourcesTree({ getActions, getDecorations, selectionMode, ...rest }) {
+    return (_jsx(Tree, { emptyTreeContent: _jsx(EmptyTreeContent, { icon: documentSvg }), ...rest, treeName: ExternalSourcesTreeComponent.id, getHierarchyDefinition: getDefinitionsProvider, selectionMode: selectionMode ?? "none", treeRenderer: (treeProps) => (_jsx(TreeRenderer, { ...treeProps, getActions: getActions, getDecorations: getDecorations ?? ((node) => _jsx(ExternalSourcesTreeIcon, { node: node })) })) }));
 }
 const getDefinitionsProvider = (props) => {
     return new ExternalSourcesTreeDefinition(props);
 };
-const classSvg = new URL("@itwin/itwinui-icons/bis-class.svg", import.meta.url).href;
-const elementSvg = new URL("@itwin/itwinui-icons/bis-element.svg", import.meta.url).href;
-const documentSvg = new URL("@itwin/itwinui-icons/document.svg", import.meta.url).href;
-const ecSchemaSvg = new URL("@itwin/itwinui-icons/selection-children.svg", import.meta.url).href;
 /** @beta */
 export function ExternalSourcesTreeIcon({ node }) {
-    if (node.extendedData?.imageId === undefined) {
+    if (node.nodeData.extendedData?.imageId === undefined) {
         return undefined;
     }
     const getIcon = () => {
-        switch (node.extendedData.imageId) {
+        switch (node.nodeData.extendedData.imageId) {
             case "icon-item":
                 return elementSvg;
             case "icon-ec-class":

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"ExternalSourcesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,IAAI,EAAE,MAAM,8BAA8B,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,sCAAsC,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAanF,YAAY;AACZ,MAAM,UAAU,mBAAmB,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,GAAG,IAAI,EAA4B;IAC/G,OAAO,CACL,KAAC,IAAI,IACH,gBAAgB,EAAE,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,KACrD,IAAI,EACR,QAAQ,EAAE,4BAA4B,CAAC,EAAE,EACzC,sBAAsB,EAAE,sBAAsB,EAC9C,aAAa,EAAE,aAAa,IAAI,MAAM,EACtC,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,CAC3B,KAAC,YAAY,OAAK,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,uBAAuB,IAAC,IAAI,EAAE,IAAI,GAAI,CAAC,GAAI,CACzI,GACD,CACH,CAAC;AACJ,CAAC;AAED,MAAM,sBAAsB,GAAwC,CAAC,KAAK,EAAE,EAAE;IAC5E,OAAO,IAAI,6BAA6B,CAAC,KAAK,CAAC,CAAC;AAClD,CAAC,CAAC;AAEF,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,oCAAoC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACrF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,sCAAsC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACzF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,mCAAmC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACvF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,6CAA6C,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAEjG,YAAY;AACZ,MAAM,UAAU,uBAAuB,CAAC,EAAE,IAAI,EAAuC;IACnF,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE;QAC5C,OAAO,SAAS,CAAC;KAClB;IACD,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,YAAa,CAAC,OAAO,EAAE;YAClC,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB,KAAK,eAAe;gBAClB,OAAO,WAAW,CAAC;YACrB,KAAK,gBAAgB;gBACnB,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,SAAS,CAAC;SACpB;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,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 { Icon } from \"@itwin/itwinui-react/bricks\";\nimport { EmptyTreeContent } from \"../common/components/EmptyTree.js\";\nimport { Tree } from \"../common/components/Tree.js\";\nimport { TreeRenderer } from \"../common/components/TreeRenderer.js\";\nimport { ExternalSourcesTreeComponent } from \"./ExternalSourcesTreeComponent.js\";\nimport { ExternalSourcesTreeDefinition } from \"./ExternalSourcesTreeDefinition.js\";\n\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { BaseTreeRendererProps } from \"../common/components/BaseTreeRenderer.js\";\nimport type { TreeProps } from \"../common/components/Tree.js\";\n/** @beta */\nexport type ExternalSourcesTreeProps = Pick<TreeProps, \"imodel\" | \"getSchemaContext\" | \"selectionStorage\" | \"selectionMode\" | \"emptyTreeContent\"> &\n  Pick<BaseTreeRendererProps, \"actions\" | \"getDecorations\"> & {\n    hierarchyLevelConfig?: {\n      sizeLimit?: number;\n    };\n  };\n\n/** @beta */\nexport function ExternalSourcesTree({ actions, getDecorations, selectionMode, ...rest }: ExternalSourcesTreeProps) {\n  return (\n    <Tree\n      emptyTreeContent={<EmptyTreeContent icon={documentSvg} />}\n      {...rest}\n      treeName={ExternalSourcesTreeComponent.id}\n      getHierarchyDefinition={getDefinitionsProvider}\n      selectionMode={selectionMode ?? \"none\"}\n      treeRenderer={(treeProps) => (\n        <TreeRenderer {...treeProps} actions={actions} getDecorations={getDecorations ?? ((node) => <ExternalSourcesTreeIcon node={node} />)} />\n      )}\n    />\n  );\n}\n\nconst getDefinitionsProvider: TreeProps[\"getHierarchyDefinition\"] = (props) => {\n  return new ExternalSourcesTreeDefinition(props);\n};\n\nconst classSvg = new URL(\"@itwin/itwinui-icons/bis-class.svg\", import.meta.url).href;\nconst elementSvg = new URL(\"@itwin/itwinui-icons/bis-element.svg\", import.meta.url).href;\nconst documentSvg = new URL(\"@itwin/itwinui-icons/document.svg\", import.meta.url).href;\nconst ecSchemaSvg = new URL(\"@itwin/itwinui-icons/selection-children.svg\", import.meta.url).href;\n\n/** @beta */\nexport function ExternalSourcesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n  const getIcon = () => {\n    switch (node.extendedData!.imageId) {\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      case \"icon-document\":\n        return documentSvg;\n      case \"icon-ec-schema\":\n        return ecSchemaSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n"]}
+{"version":3,"file":"ExternalSourcesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,QAAQ,MAAM,oCAAoC,CAAC;AAC1D,OAAO,UAAU,MAAM,sCAAsC,CAAC;AAC9D,OAAO,WAAW,MAAM,mCAAmC,CAAC;AAC5D,OAAO,WAAW,MAAM,6CAA6C,CAAC;AACtE,OAAO,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,IAAI,EAAE,MAAM,8BAA8B,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,sCAAsC,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAcnF,YAAY;AACZ,MAAM,UAAU,mBAAmB,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,GAAG,IAAI,EAA4B;IAClH,OAAO,CACL,KAAC,IAAI,IACH,gBAAgB,EAAE,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,KACrD,IAAI,EACR,QAAQ,EAAE,4BAA4B,CAAC,EAAE,EACzC,sBAAsB,EAAE,sBAAsB,EAC9C,aAAa,EAAE,aAAa,IAAI,MAAM,EACtC,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,CAC3B,KAAC,YAAY,OAAK,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,cAAc,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,uBAAuB,IAAC,IAAI,EAAE,IAAI,GAAI,CAAC,GAAI,CAC/I,GACD,CACH,CAAC;AACJ,CAAC;AAED,MAAM,sBAAsB,GAAwC,CAAC,KAAK,EAAE,EAAE;IAC5E,OAAO,IAAI,6BAA6B,CAAC,KAAK,CAAC,CAAC;AAClD,CAAC,CAAC;AAEF,YAAY;AACZ,MAAM,UAAU,uBAAuB,CAAC,EAAE,IAAI,EAAuC;IACnF,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE;QACrD,OAAO,SAAS,CAAC;KAClB;IACD,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE;YAC3C,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB,KAAK,eAAe;gBAClB,OAAO,WAAW,CAAC;YACrB,KAAK,gBAAgB;gBACnB,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,SAAS,CAAC;SACpB;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,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 classSvg from \"@itwin/itwinui-icons/bis-class.svg\";\nimport elementSvg from \"@itwin/itwinui-icons/bis-element.svg\";\nimport documentSvg from \"@itwin/itwinui-icons/document.svg\";\nimport ecSchemaSvg from \"@itwin/itwinui-icons/selection-children.svg\";\nimport { Icon } from \"@itwin/itwinui-react/bricks\";\nimport { EmptyTreeContent } from \"../common/components/EmptyTree.js\";\nimport { Tree } from \"../common/components/Tree.js\";\nimport { TreeRenderer } from \"../common/components/TreeRenderer.js\";\nimport { ExternalSourcesTreeComponent } from \"./ExternalSourcesTreeComponent.js\";\nimport { ExternalSourcesTreeDefinition } from \"./ExternalSourcesTreeDefinition.js\";\n\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { BaseTreeRendererProps } from \"../common/components/BaseTreeRenderer.js\";\nimport type { TreeProps } from \"../common/components/Tree.js\";\n\n/** @beta */\nexport type ExternalSourcesTreeProps = Pick<TreeProps, \"imodel\" | \"getSchemaContext\" | \"selectionStorage\" | \"selectionMode\" | \"emptyTreeContent\"> &\n  Pick<BaseTreeRendererProps, \"getActions\" | \"getDecorations\"> & {\n    hierarchyLevelConfig?: {\n      sizeLimit?: number;\n    };\n  };\n\n/** @beta */\nexport function ExternalSourcesTree({ getActions, getDecorations, selectionMode, ...rest }: ExternalSourcesTreeProps) {\n  return (\n    <Tree\n      emptyTreeContent={<EmptyTreeContent icon={documentSvg} />}\n      {...rest}\n      treeName={ExternalSourcesTreeComponent.id}\n      getHierarchyDefinition={getDefinitionsProvider}\n      selectionMode={selectionMode ?? \"none\"}\n      treeRenderer={(treeProps) => (\n        <TreeRenderer {...treeProps} getActions={getActions} getDecorations={getDecorations ?? ((node) => <ExternalSourcesTreeIcon node={node} />)} />\n      )}\n    />\n  );\n}\n\nconst getDefinitionsProvider: TreeProps[\"getHierarchyDefinition\"] = (props) => {\n  return new ExternalSourcesTreeDefinition(props);\n};\n\n/** @beta */\nexport function ExternalSourcesTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      case \"icon-document\":\n        return documentSvg;\n      case \"icon-ec-schema\":\n        return ecSchemaSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeComponent.d.ts

@@ -1,6 +1,6 @@
 import type { ExternalSourcesTreeProps } from "./ExternalSourcesTree.js";
 /** @beta */
-interface ExternalSourcesTreeComponentProps extends Pick<ExternalSourcesTreeProps, "getSchemaContext" | "selectionStorage" | "selectionMode" | "hierarchyLevelConfig" | "selectionMode" | "emptyTreeContent" | "actions" | "getDecorations"> {
+interface ExternalSourcesTreeComponentProps extends Pick<ExternalSourcesTreeProps, "getSchemaContext" | "selectionStorage" | "selectionMode" | "hierarchyLevelConfig" | "selectionMode" | "emptyTreeContent" | "getActions" | "getDecorations"> {
     onPerformanceMeasured?: (featureId: string, duration: number) => void;
     onFeatureUsed?: (feature: string) => void;
 }

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeComponent.js.map

@@ -1 +1 @@
-{"version":3,"file":"ExternalSourcesTreeComponent.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeComponent.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAa/D;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,KAAK,EAAqC,EAAE,EAAE;IACpI,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;IAE3C,IAAI,CAAC,MAAM,EAAE;QACX,OAAO,IAAI,CAAC;KACb;IAED,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,KAAK,EAAE,MAAM,EAAE,MAAM,GAAI,GACnC,GACQ,CAC5B,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,4BAA4B,CAAC,EAAE,GAAG,0BAA0B,CAAC;AAE7D;;;GAGG;AACH,4BAA4B,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,2BAA2B,CAAC,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 { TreeWidget } from \"../../../TreeWidget.js\";\nimport { SelectableTree } from \"../../tree-header/SelectableTree.js\";\nimport { TelemetryContextProvider } from \"../common/UseTelemetryContext.js\";\nimport { ExternalSourcesTree } from \"./ExternalSourcesTree.js\";\n\nimport type { ExternalSourcesTreeProps } from \"./ExternalSourcesTree.js\";\n/** @beta */\ninterface ExternalSourcesTreeComponentProps\n  extends Pick<\n    ExternalSourcesTreeProps,\n    \"getSchemaContext\" | \"selectionStorage\" | \"selectionMode\" | \"hierarchyLevelConfig\" | \"selectionMode\" | \"emptyTreeContent\" | \"actions\" | \"getDecorations\"\n  > {\n  onPerformanceMeasured?: (featureId: string, duration: number) => void;\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * A component that renders `ExternalSourcesTree`.\n * @beta\n */\nexport const ExternalSourcesTreeComponent = ({ onFeatureUsed, onPerformanceMeasured, ...props }: ExternalSourcesTreeComponentProps) => {\n  const imodel = useActiveIModelConnection();\n\n  if (!imodel) {\n    return null;\n  }\n\n  return (\n    <TelemetryContextProvider componentIdentifier={ExternalSourcesTreeComponent.id} onFeatureUsed={onFeatureUsed} onPerformanceMeasured={onPerformanceMeasured}>\n      <SelectableTree>\n        <ExternalSourcesTree {...props} imodel={imodel} />\n      </SelectableTree>\n    </TelemetryContextProvider>\n  );\n};\n\n/**\n * Id of the component. May be used when a creating a `TreeDefinition` for `SelectableTree`.\n * @beta\n */\nExternalSourcesTreeComponent.id = \"external-sources-tree-v2\";\n\n/**\n * Label of the component. May be used when a creating a `TreeDefinition` for `SelectableTree`.\n * @beta\n */\nExternalSourcesTreeComponent.getLabel = () => TreeWidget.translate(\"externalSourcesTree.label\");\n"]}
+{"version":3,"file":"ExternalSourcesTreeComponent.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeComponent.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAc/D;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,KAAK,EAAqC,EAAE,EAAE;IACpI,MAAM,MAAM,GAAG,yBAAyB,EAAE,CAAC;IAE3C,IAAI,CAAC,MAAM,EAAE;QACX,OAAO,IAAI,CAAC;KACb;IAED,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,KAAK,EAAE,MAAM,EAAE,MAAM,GAAI,GACnC,GACQ,CAC5B,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,4BAA4B,CAAC,EAAE,GAAG,0BAA0B,CAAC;AAE7D;;;GAGG;AACH,4BAA4B,CAAC,QAAQ,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,2BAA2B,CAAC,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 { TreeWidget } from \"../../../TreeWidget.js\";\nimport { SelectableTree } from \"../../tree-header/SelectableTree.js\";\nimport { TelemetryContextProvider } from \"../common/UseTelemetryContext.js\";\nimport { ExternalSourcesTree } from \"./ExternalSourcesTree.js\";\n\nimport type { ExternalSourcesTreeProps } from \"./ExternalSourcesTree.js\";\n\n/** @beta */\ninterface ExternalSourcesTreeComponentProps\n  extends Pick<\n    ExternalSourcesTreeProps,\n    \"getSchemaContext\" | \"selectionStorage\" | \"selectionMode\" | \"hierarchyLevelConfig\" | \"selectionMode\" | \"emptyTreeContent\" | \"getActions\" | \"getDecorations\"\n  > {\n  onPerformanceMeasured?: (featureId: string, duration: number) => void;\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * A component that renders `ExternalSourcesTree`.\n * @beta\n */\nexport const ExternalSourcesTreeComponent = ({ onFeatureUsed, onPerformanceMeasured, ...props }: ExternalSourcesTreeComponentProps) => {\n  const imodel = useActiveIModelConnection();\n\n  if (!imodel) {\n    return null;\n  }\n\n  return (\n    <TelemetryContextProvider componentIdentifier={ExternalSourcesTreeComponent.id} onFeatureUsed={onFeatureUsed} onPerformanceMeasured={onPerformanceMeasured}>\n      <SelectableTree>\n        <ExternalSourcesTree {...props} imodel={imodel} />\n      </SelectableTree>\n    </TelemetryContextProvider>\n  );\n};\n\n/**\n * Id of the component. May be used when a creating a `TreeDefinition` for `SelectableTree`.\n * @beta\n */\nExternalSourcesTreeComponent.id = \"external-sources-tree-v2\";\n\n/**\n * Label of the component. May be used when a creating a `TreeDefinition` for `SelectableTree`.\n * @beta\n */\nExternalSourcesTreeComponent.getLabel = () => TreeWidget.translate(\"externalSourcesTree.label\");\n"]}

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeDefinition.d.ts

@@ -3,6 +3,7 @@ import type { DefineHierarchyLevelProps, HierarchyDefinition, HierarchyLevelDefi
 interface ExternalSourcesTreeDefinitionProps {
     imodelAccess: ECSchemaProvider & ECClassHierarchyInspector & LimitingECSqlQueryExecutor;
 }
+/** @internal */
 export declare class ExternalSourcesTreeDefinition implements HierarchyDefinition {
     private _impl;
     private _selectQueryFactory;

package/lib/esm/tree-widget-react/components/trees/external-sources-tree/ExternalSourcesTreeDefinition.js

@@ -4,6 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { createNodesQueryClauseFactory, createPredicateBasedHierarchyDefinition, HierarchyNode } from "@itwin/presentation-hierarchies";
 import { createBisInstanceLabelSelectClauseFactory, ECSql } from "@itwin/presentation-shared";
+/** @internal */
 export class ExternalSourcesTreeDefinition {
     constructor(props) {
         this._impl = createPredicateBasedHierarchyDefinition({