@itwin/tree-widget-react 4.0.0-alpha.2 → 4.0.0-alpha.20

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

@@ -0,0 +1,60 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { BeEvent } from "@itwin/core-bentley";
+/** @internal */
+export function createVisibilityChangeEventListener(props) {
+    const onVisibilityChange = new BeEvent();
+    let pendingVisibilityChange;
+    let suppressChangeEvents = 0;
+    let hasFiredDuringSuppress = true;
+    const handleVisibilityChange = () => {
+        hasFiredDuringSuppress = true;
+        if (pendingVisibilityChange || suppressChangeEvents > 0) {
+            return;
+        }
+        pendingVisibilityChange = setTimeout(() => {
+            onVisibilityChange.raiseEvent();
+            pendingVisibilityChange = undefined;
+        });
+    };
+    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.onPerModelCategoriesOverridesChanged.addListener(handleVisibilityChange));
+        listeners.push(props.viewport.onDisplayedCategoriesChanged.addListener(handleVisibilityChange));
+    }
+    if (props.listeners.displayStyle) {
+        listeners.push(props.viewport.onDisplayStyleChanged.addListener(handleVisibilityChange));
+    }
+    if (props.listeners.models) {
+        listeners.push(props.viewport.onDisplayedModelsChanged.addListener(handleVisibilityChange));
+    }
+    return {
+        onVisibilityChange,
+        [Symbol.dispose]() {
+            if (pendingVisibilityChange) {
+                clearTimeout(pendingVisibilityChange);
+                pendingVisibilityChange = undefined;
+            }
+            listeners.forEach((x) => x());
+            listeners.length = 0;
+        },
+        suppressChangeEvents: () => {
+            hasFiredDuringSuppress = false;
+            ++suppressChangeEvents;
+        },
+        isVisibilityChangePending: () => pendingVisibilityChange !== undefined,
+        resumeChangeEvents: () => {
+            --suppressChangeEvents;
+            if (suppressChangeEvents === 0 && hasFiredDuringSuppress) {
+                handleVisibilityChange();
+            }
+        },
+    };
+}
+//# sourceMappingURL=VisibilityChangeEventListener.js.map

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;AAY9C,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,IAAI,sBAAsB,GAAG,IAAI,CAAC;IAClC,MAAM,sBAAsB,GAAG,GAAG,EAAE;QAClC,sBAAsB,GAAG,IAAI,CAAC;QAC9B,IAAI,uBAAuB,IAAI,oBAAoB,GAAG,CAAC,EAAE,CAAC;YACxD,OAAO;QACT,CAAC;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,CAAC;QAC7B,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;IACzF,CAAC;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QAC/B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,oCAAoC,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACxG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,4BAA4B,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;IAClG,CAAC;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC;QACjC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;IAC3F,CAAC;IACD,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QAC3B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;IAC9F,CAAC;IAED,OAAO;QACL,kBAAkB;QAClB,CAAC,MAAM,CAAC,OAAO,CAAC;YACd,IAAI,uBAAuB,EAAE,CAAC;gBAC5B,YAAY,CAAC,uBAAuB,CAAC,CAAC;gBACtC,uBAAuB,GAAG,SAAS,CAAC;YACtC,CAAC;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,sBAAsB,GAAG,KAAK,CAAC;YAC/B,EAAE,oBAAoB,CAAC;QACzB,CAAC;QACD,yBAAyB,EAAE,GAAG,EAAE,CAAC,uBAAuB,KAAK,SAAS;QACtE,kBAAkB,EAAE,GAAG,EAAE;YACvB,EAAE,oBAAoB,CAAC;YACvB,IAAI,oBAAoB,KAAK,CAAC,IAAI,sBAAsB,EAAE,CAAC;gBACzD,sBAAsB,EAAE,CAAC;YAC3B,CAAC;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 { TreeWidgetViewport } from \"../TreeWidgetViewport.js\";\n\n/** @internal */\nexport interface IVisibilityChangeEventListener extends Disposable {\n  onVisibilityChange: BeEvent<() => void>;\n  suppressChangeEvents(): void;\n  resumeChangeEvents(): void;\n  isVisibilityChangePending: () => boolean;\n}\n\n/** @internal */\nexport function createVisibilityChangeEventListener(props: {\n  viewport: TreeWidgetViewport;\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  let hasFiredDuringSuppress = true;\n  const handleVisibilityChange = () => {\n    hasFiredDuringSuppress = true;\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.onPerModelCategoriesOverridesChanged.addListener(handleVisibilityChange));\n    listeners.push(props.viewport.onDisplayedCategoriesChanged.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.onDisplayedModelsChanged.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      hasFiredDuringSuppress = false;\n      ++suppressChangeEvents;\n    },\n    isVisibilityChangePending: () => pendingVisibilityChange !== undefined,\n    resumeChangeEvents: () => {\n      --suppressChangeEvents;\n      if (suppressChangeEvents === 0 && hasFiredDuringSuppress) {\n        handleVisibilityChange();\n      }\n    },\n  };\n}\n"]}

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

@@ -0,0 +1,39 @@
+import type { Observable, OperatorFunction } from "rxjs";
+import type { GuidString, Id64Arg, Id64String } from "@itwin/core-bentley";
+import type { CategoryInfo } from "../CategoriesVisibilityUtils.js";
+import type { TreeWidgetViewport } from "../TreeWidgetViewport.js";
+import type { VisibilityStatus } from "../UseHierarchyVisibility.js";
+import type { ElementId } from "./Types.js";
+/** @internal */
+export declare function mergeVisibilityStatuses(obs: Observable<VisibilityStatus>): Observable<VisibilityStatus>;
+/** @internal */
+export declare function changeElementStateNoChildrenOperator(props: {
+    on: boolean;
+    isDisplayedByDefault: boolean;
+    viewport: TreeWidgetViewport;
+}): OperatorFunction<string, void>;
+/** @internal */
+export declare function getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props: {
+    alwaysDrawn: Set<ElementId> | undefined;
+    neverDrawn: Set<ElementId> | undefined;
+    totalCount: number;
+    viewport: TreeWidgetViewport;
+} & GetVisibilityFromAlwaysAndNeverDrawnElementsProps): VisibilityStatus;
+/** @internal */
+export interface GetVisibilityFromAlwaysAndNeverDrawnElementsProps {
+    /** Status when always/never lists are empty and exclusive mode is off */
+    defaultStatus: (categoryId?: string) => VisibilityStatus;
+}
+/**
+ * Changes category display in the viewport.
+ * @internal
+ */
+export declare function enableCategoryDisplay(viewport: TreeWidgetViewport, categoryIds: Id64Arg, enabled: boolean, enableAllSubCategories?: boolean): Promise<void>;
+/**
+ * Changes subcategory display in the viewport
+ * @internal
+ */
+export declare function enableSubCategoryDisplay(viewport: TreeWidgetViewport, subCategoryId: Id64String, enabled: boolean): void;
+/** @internal */
+export declare function loadCategoriesFromViewport(vp: TreeWidgetViewport, componentId?: GuidString): Promise<CategoryInfo[]>;
+//# sourceMappingURL=VisibilityUtils.d.ts.map

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

@@ -0,0 +1,160 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { bufferCount, from, map, mergeMap, reduce } from "rxjs";
+import { Guid, Id64 } from "@itwin/core-bentley";
+import { QueryRowFormat } from "@itwin/core-common";
+import { reduceWhile, toVoidPromise } from "./Rxjs.js";
+import { createVisibilityStatus } from "./Tooltip.js";
+import { getClassesByView, getOptimalBatchSize, 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) => createVisibilityStatus(visibility === "empty" ? "disabled" : visibility)));
+}
+/** @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({ elementIds: state.neverDrawn });
+            state.changedAlwaysDrawn &&
+                state.alwaysDrawn &&
+                props.viewport.setAlwaysDrawn({ elementIds: state.alwaysDrawn, exclusive: props.viewport.isAlwaysDrawnExclusive });
+        }));
+    };
+}
+/** @internal */
+export function getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props) {
+    const { alwaysDrawn, neverDrawn, totalCount, viewport } = props;
+    if (totalCount === 0) {
+        return props.defaultStatus();
+    }
+    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;
+}
+/**
+ * Changes category display in the viewport.
+ * @internal
+ */
+export async function enableCategoryDisplay(viewport, categoryIds, enabled, enableAllSubCategories = true) {
+    const removeOverrides = (bufferedCategories) => {
+        const modelsContainingOverrides = [];
+        for (const ovr of viewport.perModelCategoryOverrides) {
+            if (Id64.has(bufferedCategories, ovr.categoryId)) {
+                modelsContainingOverrides.push(ovr.modelId);
+            }
+        }
+        viewport.setPerModelCategoryOverride({ modelIds: modelsContainingOverrides, categoryIds: bufferedCategories, override: "none" });
+    };
+    const disableSubCategories = async (bufferedCategories) => {
+        // changeCategoryDisplay only enables subcategories, it does not disabled them. So we must do that ourselves.
+        (await viewport.iModel.categories.getCategoryInfo(bufferedCategories)).forEach((categoryInfo) => {
+            categoryInfo.subCategories.forEach((value) => enableSubCategoryDisplay(viewport, value.id, false));
+        });
+    };
+    return toVoidPromise(from(Id64.iterable(categoryIds)).pipe(releaseMainThreadOnItemsCount(500), bufferCount(getOptimalBatchSize({ totalSize: Id64.sizeOf(categoryIds), maximumBatchSize: 500 })), mergeMap(async (bufferedCategories) => {
+        viewport.changeCategoryDisplay({ categoryIds: bufferedCategories, display: enabled, enableAllSubCategories });
+        removeOverrides(bufferedCategories);
+        if (!enabled) {
+            await disableSubCategories(bufferedCategories);
+        }
+    })));
+}
+/**
+ * Changes subcategory display in the viewport
+ * @internal
+ */
+export function enableSubCategoryDisplay(viewport, subCategoryId, enabled) {
+    viewport.changeSubCategoryDisplay({ subCategoryId, display: enabled });
+}
+/** @internal */
+export async function loadCategoriesFromViewport(vp, componentId) {
+    // Query categories and add them to state
+    if (vp.viewType === "other") {
+        return [];
+    }
+    const { categoryClass, elementClass } = getClassesByView(vp.viewType);
+    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.viewType !== "2d" ? "" : "AND Model.Id=?"}
+      )
+  `;
+    const categories = [];
+    const rows = await (async () => {
+        const result = new Array();
+        for await (const row of vp.iModel.createQueryReader(ecsql, undefined, {
+            rowFormat: QueryRowFormat.UseJsPropertyNames,
+            restartToken: `CategoriesVisibilityUtils/${componentId ?? Guid.createValue()}/categories`,
+        })) {
+            result.push(row.id);
+        }
+        return result;
+    })();
+    (await vp.iModel.categories.getCategoryInfo(rows)).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,WAAW,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,MAAM,YAAY,CAAC;AAUlG,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,CAAC;YACP,OAAO,OAAO,CAAC;QACjB,CAAC;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,CAAC,sBAAsB,CAAC,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAC9F,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,CAAC;gBAClE,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;YAC5D,CAAC;YACD,IAAI,EAAE,EAAE,CAAC;gBACP,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,CAAC;oBACzF,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC/B,GAAG,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAChC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,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,CAAC;oBACtF,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC9B,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC;gBAC/B,CAAC;YACH,CAAC;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,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;YAC9G,KAAK,CAAC,kBAAkB;gBACtB,KAAK,CAAC,WAAW;gBACjB,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,WAAW,EAAE,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC,CAAC;QACvH,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;IAChE,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC,aAAa,EAAE,CAAC;IAC/B,CAAC;IACD,IAAI,UAAU,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;QACpC,OAAO,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;QACrC,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,QAAQ,CAAC,sBAAsB,EAAE,CAAC;QACpC,OAAO,sBAAsB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC1E,CAAC;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,CAAC;QACzG,OAAO,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAQD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAA4B,EAAE,WAAoB,EAAE,OAAgB,EAAE,sBAAsB,GAAG,IAAI;IAC7I,MAAM,eAAe,GAAG,CAAC,kBAA6B,EAAE,EAAE;QACxD,MAAM,yBAAyB,GAAa,EAAE,CAAC;QAC/C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,yBAAyB,EAAE,CAAC;YACrD,IAAI,IAAI,CAAC,GAAG,CAAC,kBAAkB,EAAE,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBACjD,yBAAyB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QACD,QAAQ,CAAC,2BAA2B,CAAC,EAAE,QAAQ,EAAE,yBAAyB,EAAE,WAAW,EAAE,kBAAkB,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IACnI,CAAC,CAAC;IACF,MAAM,oBAAoB,GAAG,KAAK,EAAE,kBAA6B,EAAE,EAAE;QACnE,6GAA6G;QAC7G,CAAC,MAAM,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;YAC9F,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;IACL,CAAC,CAAC;IACF,OAAO,aAAa,CAClB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACnC,6BAA6B,CAAC,GAAG,CAAC,EAClC,WAAW,CAAC,mBAAmB,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC,EAChG,QAAQ,CAAC,KAAK,EAAE,kBAAkB,EAAE,EAAE;QACpC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAC9G,eAAe,CAAC,kBAAkB,CAAC,CAAC;QACpC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC,CACH,CACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAA4B,EAAE,aAAyB,EAAE,OAAgB;IAChH,QAAQ,CAAC,wBAAwB,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAsB,EAAE,WAAwB;IAC/F,yCAAyC;IACzC,IAAI,EAAE,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,GAAG,gBAAgB,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;IACtE,MAAM,KAAK,GAAG;;WAEL,aAAa;;;;eAIT,YAAY;;qDAE0B,aAAa;YACtD,EAAE,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,gBAAgB;;GAErD,CAAC;IAEF,MAAM,UAAU,GAAmB,EAAE,CAAC;IACtC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE;QAC7B,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE;YACpE,SAAS,EAAE,cAAc,CAAC,kBAAkB;YAC5C,YAAY,EAAE,6BAA6B,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,aAAa;SAC1F,CAAC,EAAE,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,EAAE,CAAC;IACL,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;QACjE,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 { bufferCount, from, map, mergeMap, reduce } from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { QueryRowFormat } from \"@itwin/core-common\";\nimport { reduceWhile, toVoidPromise } from \"./Rxjs.js\";\nimport { createVisibilityStatus } from \"./Tooltip.js\";\nimport { getClassesByView, getOptimalBatchSize, releaseMainThreadOnItemsCount } from \"./Utils.js\";\n\nimport type { Observable, OperatorFunction } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { CategoryInfo } from \"../CategoriesVisibilityUtils.js\";\nimport type { TreeWidgetViewport } from \"../TreeWidgetViewport.js\";\nimport type { VisibilityStatus } from \"../UseHierarchyVisibility.js\";\nimport type { Visibility } from \"./Tooltip.js\";\nimport type { ElementId } from \"./Types.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) => createVisibilityStatus(visibility === \"empty\" ? \"disabled\" : visibility)),\n  );\n}\n\n/** @internal */\nexport function changeElementStateNoChildrenOperator(props: {\n  on: boolean;\n  isDisplayedByDefault: boolean;\n  viewport: TreeWidgetViewport;\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({ elementIds: state.neverDrawn });\n        state.changedAlwaysDrawn &&\n          state.alwaysDrawn &&\n          props.viewport.setAlwaysDrawn({ elementIds: state.alwaysDrawn, exclusive: 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: TreeWidgetViewport;\n  } & GetVisibilityFromAlwaysAndNeverDrawnElementsProps,\n): VisibilityStatus {\n  const { alwaysDrawn, neverDrawn, totalCount, viewport } = props;\n  if (totalCount === 0) {\n    return props.defaultStatus();\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 interface GetVisibilityFromAlwaysAndNeverDrawnElementsProps {\n  /** Status when always/never lists are empty and exclusive mode is off */\n  defaultStatus: (categoryId?: string) => VisibilityStatus;\n}\n\n/**\n * Changes category display in the viewport.\n * @internal\n */\nexport async function enableCategoryDisplay(viewport: TreeWidgetViewport, categoryIds: Id64Arg, enabled: boolean, enableAllSubCategories = true) {\n  const removeOverrides = (bufferedCategories: Id64Array) => {\n    const modelsContainingOverrides: string[] = [];\n    for (const ovr of viewport.perModelCategoryOverrides) {\n      if (Id64.has(bufferedCategories, ovr.categoryId)) {\n        modelsContainingOverrides.push(ovr.modelId);\n      }\n    }\n    viewport.setPerModelCategoryOverride({ modelIds: modelsContainingOverrides, categoryIds: bufferedCategories, override: \"none\" });\n  };\n  const disableSubCategories = async (bufferedCategories: Id64Array) => {\n    // changeCategoryDisplay only enables subcategories, it does not disabled them. So we must do that ourselves.\n    (await viewport.iModel.categories.getCategoryInfo(bufferedCategories)).forEach((categoryInfo) => {\n      categoryInfo.subCategories.forEach((value) => enableSubCategoryDisplay(viewport, value.id, false));\n    });\n  };\n  return toVoidPromise(\n    from(Id64.iterable(categoryIds)).pipe(\n      releaseMainThreadOnItemsCount(500),\n      bufferCount(getOptimalBatchSize({ totalSize: Id64.sizeOf(categoryIds), maximumBatchSize: 500 })),\n      mergeMap(async (bufferedCategories) => {\n        viewport.changeCategoryDisplay({ categoryIds: bufferedCategories, display: enabled, enableAllSubCategories });\n        removeOverrides(bufferedCategories);\n        if (!enabled) {\n          await disableSubCategories(bufferedCategories);\n        }\n      }),\n    ),\n  );\n}\n\n/**\n * Changes subcategory display in the viewport\n * @internal\n */\nexport function enableSubCategoryDisplay(viewport: TreeWidgetViewport, subCategoryId: Id64String, enabled: boolean) {\n  viewport.changeSubCategoryDisplay({ subCategoryId, display: enabled });\n}\n\n/** @internal */\nexport async function loadCategoriesFromViewport(vp: TreeWidgetViewport, componentId?: GuidString) {\n  // Query categories and add them to state\n  if (vp.viewType === \"other\") {\n    return [];\n  }\n  const { categoryClass, elementClass } = getClassesByView(vp.viewType);\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.viewType !== \"2d\" ? \"\" : \"AND Model.Id=?\"}\n      )\n  `;\n\n  const categories: CategoryInfo[] = [];\n  const rows = await (async () => {\n    const result = new Array<Id64String>();\n    for await (const row of vp.iModel.createQueryReader(ecsql, undefined, {\n      rowFormat: QueryRowFormat.UseJsPropertyNames,\n      restartToken: `CategoriesVisibilityUtils/${componentId ?? Guid.createValue()}/categories`,\n    })) {\n      result.push(row.id);\n    }\n    return result;\n  })();\n  (await vp.iModel.categories.getCategoryInfo(rows)).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/common/internal/useGuid.d.ts

@@ -0,0 +1,3 @@
+/** @internal */
+export declare function useGuid(): string;
+//# sourceMappingURL=useGuid.d.ts.map

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

@@ -0,0 +1,12 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { useState } from "react";
+import { Guid } from "@itwin/core-bentley";
+/** @internal */
+export function useGuid() {
+    const [componentId] = useState(() => Guid.createValue());
+    return componentId;
+}
+//# sourceMappingURL=useGuid.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useGuid.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/internal/useGuid.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAE3C,gBAAgB;AAChB,MAAM,UAAU,OAAO;IACrB,MAAM,CAAC,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IACzD,OAAO,WAAW,CAAC;AACrB,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 { useState } from \"react\";\nimport { Guid } from \"@itwin/core-bentley\";\n\n/** @internal */\nexport function useGuid() {\n  const [componentId] = useState(() => Guid.createValue());\n  return componentId;\n}\n"]}

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

@@ -0,0 +1,68 @@
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { HierarchyVisibilityOverrideHandler } from "../../UseHierarchyVisibility.js";
+import { AlwaysAndNeverDrawnElementInfo } from "../AlwaysAndNeverDrawnElementInfo.js";
+import type { GuidString } from "@itwin/core-bentley";
+import type { HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { TreeWidgetViewport } from "../../TreeWidgetViewport.js";
+import type { HierarchyVisibilityHandler, VisibilityStatus } from "../../UseHierarchyVisibility.js";
+import type { FilteredTree } from "../visibility/BaseFilteredTree.js";
+import type { TreeSpecificVisibilityHandler } from "../visibility/BaseVisibilityHelper.js";
+/** @internal */
+export interface CreateFilteredTreeProps<TCache> {
+    getCache: () => TCache;
+    imodelAccess: ECClassHierarchyInspector;
+    filteringPaths: HierarchyFilteringPath[];
+}
+/** @internal */
+export interface CreateTreeSpecificVisibilityHandlerProps<TCache> {
+    info: AlwaysAndNeverDrawnElementInfo;
+    getCache: () => TCache;
+    viewport: TreeWidgetViewport;
+    overrideHandler: HierarchyVisibilityOverrideHandler;
+}
+/** @internal */
+export interface UseCachedVisibilityProps<TCache, TFilterTargets> {
+    componentId: GuidString;
+    activeView: TreeWidgetViewport;
+    getCache: () => TCache;
+    createFilteredTree: (props: CreateFilteredTreeProps<TCache>) => Promise<FilteredTree<TFilterTargets>>;
+    createTreeSpecificVisibilityHandler: (props: CreateTreeSpecificVisibilityHandlerProps<TCache>) => TreeSpecificVisibilityHandler<TFilterTargets> & Disposable;
+}
+/** @internal */
+export declare function useCachedVisibility<TCache, TFilterTargets>(props: UseCachedVisibilityProps<TCache, TFilterTargets>): {
+    visibilityHandlerFactory: (props: {
+        imodelAccess: ECClassHierarchyInspector;
+    }) => HierarchyVisibilityHandler;
+    onFilteredPathsChanged: (paths: HierarchyFilteringPath[] | undefined) => void;
+    filteredPaths: HierarchyFilteringPath[] | undefined;
+};
+/** @internal */
+export interface HierarchyVisibilityHandlerImplProps<TFilterTargets> {
+    viewport: TreeWidgetViewport;
+    getTreeSpecificVisibilityHandler: (info: AlwaysAndNeverDrawnElementInfo, overrideHandler: HierarchyVisibilityOverrideHandler) => TreeSpecificVisibilityHandler<TFilterTargets> & Disposable;
+    getFilteredTree: () => Promise<FilteredTree<TFilterTargets>> | undefined;
+    componentId?: GuidString;
+}
+/**
+ * Class that handles visibility of hierarchy nodes in a tree.
+ *
+ * - It provides methods to get and change visibility status of nodes.
+ * - Also handles filtered tree nodes visibility.
+ * - Listens to visibility change events and updates visibility status accordingly.
+ * @internal
+ */
+export declare class HierarchyVisibilityHandlerImpl<TFilterTargets> implements HierarchyVisibilityHandler, Disposable {
+    #private;
+    constructor(props: HierarchyVisibilityHandlerImplProps<TFilterTargets>);
+    get onVisibilityChange(): import("@itwin/core-bentley").BeEvent<() => void>;
+    getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus>;
+    changeVisibility(node: HierarchyNode, shouldDisplay: boolean): Promise<void>;
+    [Symbol.dispose](): void;
+    private getVisibilityStatusInternal;
+    private changeVisibilityStatusInternal;
+    private getFilteredNodeVisibility;
+    private getFilteredTreeTargets;
+    private changeFilteredNodeVisibility;
+}
+//# sourceMappingURL=UseCachedVisibility.d.ts.map

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

@@ -0,0 +1,170 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { useCallback, useEffect, useState } from "react";
+import { defaultIfEmpty, EMPTY, filter, firstValueFrom, from, fromEventPattern, map, mergeMap, Subject, takeUntil, tap } from "rxjs";
+import { assert } from "@itwin/core-bentley";
+import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { HierarchyVisibilityOverrideHandler } from "../../UseHierarchyVisibility.js";
+import { AlwaysAndNeverDrawnElementInfo } from "../AlwaysAndNeverDrawnElementInfo.js";
+import { toVoidPromise } from "../Rxjs.js";
+import { createVisibilityStatus } from "../Tooltip.js";
+import { createVisibilityChangeEventListener } from "../VisibilityChangeEventListener.js";
+/** @internal */
+export function useCachedVisibility(props) {
+    const [filteredPaths, setFilteredPaths] = useState(undefined);
+    const { activeView, getCache, createFilteredTree, createTreeSpecificVisibilityHandler, componentId } = props;
+    const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory({
+        activeView,
+        getCache,
+        createFilteredTree,
+        createTreeSpecificVisibilityHandler,
+        filteringPaths: filteredPaths,
+        componentId,
+    }));
+    useEffect(() => {
+        setVisibilityHandlerFactory(() => createVisibilityHandlerFactory({
+            activeView,
+            getCache,
+            createFilteredTree,
+            createTreeSpecificVisibilityHandler,
+            filteringPaths: filteredPaths,
+            componentId,
+        }));
+    }, [activeView, getCache, filteredPaths, createFilteredTree, createTreeSpecificVisibilityHandler, componentId]);
+    return {
+        visibilityHandlerFactory,
+        onFilteredPathsChanged: useCallback((paths) => setFilteredPaths(paths), []),
+        filteredPaths,
+    };
+}
+function createVisibilityHandlerFactory(props) {
+    const { activeView, createFilteredTree, createTreeSpecificVisibilityHandler, getCache, filteringPaths, componentId } = props;
+    return ({ imodelAccess }) => new HierarchyVisibilityHandlerImpl({
+        componentId,
+        viewport: activeView,
+        getFilteredTree: () => {
+            if (filteringPaths) {
+                return createFilteredTree({ imodelAccess, filteringPaths, getCache });
+            }
+            return undefined;
+        },
+        getTreeSpecificVisibilityHandler: (info, overrideHandler) => createTreeSpecificVisibilityHandler({
+            info,
+            getCache,
+            viewport: activeView,
+            overrideHandler,
+        }),
+    });
+}
+/**
+ * Class that handles visibility of hierarchy nodes in a tree.
+ *
+ * - It provides methods to get and change visibility status of nodes.
+ * - Also handles filtered tree nodes visibility.
+ * - Listens to visibility change events and updates visibility status accordingly.
+ * @internal
+ */
+export class HierarchyVisibilityHandlerImpl {
+    #props;
+    #eventListener;
+    #alwaysAndNeverDrawnElements;
+    #treeSpecificVisibilityHandler;
+    #changeRequest = new Subject();
+    #filteredTree;
+    constructor(props) {
+        this.#props = props;
+        this.#eventListener = createVisibilityChangeEventListener({
+            viewport: this.#props.viewport,
+            listeners: {
+                models: true,
+                categories: true,
+                elements: true,
+                displayStyle: true,
+            },
+        });
+        this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo({ viewport: this.#props.viewport, componentId: props.componentId });
+        this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(this.#alwaysAndNeverDrawnElements, new HierarchyVisibilityOverrideHandler(this));
+        this.#filteredTree = this.#props.getFilteredTree();
+    }
+    get onVisibilityChange() {
+        return this.#eventListener.onVisibilityChange;
+    }
+    async getVisibilityStatus(node) {
+        return firstValueFrom(this.getVisibilityStatusInternal(node).pipe(
+        // unsubscribe from the observable if the change request for this node is received
+        takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))), 
+        // unsubscribe if visibility changes
+        takeUntil(fromEventPattern((handler) => {
+            this.#eventListener.onVisibilityChange.addListener(handler);
+        }, (handler) => {
+            this.#eventListener.onVisibilityChange.removeListener(handler);
+        })), defaultIfEmpty(createVisibilityStatus("hidden"))));
+    }
+    async changeVisibility(node, shouldDisplay) {
+        // notify about new change request
+        this.#changeRequest.next({ key: node.key, depth: node.parentKeys.length });
+        const changeObservable = this.changeVisibilityStatusInternal(node, shouldDisplay).pipe(
+        // unsubscribe from the observable if the change request for this node is received
+        takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))), tap({
+            subscribe: () => {
+                this.#eventListener.suppressChangeEvents();
+                this.#alwaysAndNeverDrawnElements.suppressChangeEvents();
+            },
+            finalize: () => {
+                this.#eventListener.resumeChangeEvents();
+                this.#alwaysAndNeverDrawnElements.resumeChangeEvents();
+            },
+        }));
+        return toVoidPromise(changeObservable);
+    }
+    [Symbol.dispose]() {
+        this.#eventListener[Symbol.dispose]();
+        this.#alwaysAndNeverDrawnElements[Symbol.dispose]();
+        this.#treeSpecificVisibilityHandler[Symbol.dispose]();
+    }
+    getVisibilityStatusInternal(node) {
+        if (HierarchyNode.isClassGroupingNode(node)) {
+            if (node.extendedData?.hasDirectNonFilteredTargets && !node.extendedData?.hasFilterTargetAncestor) {
+                return this.getFilteredNodeVisibility({ node });
+            }
+        }
+        if (HierarchyNode.isInstancesNode(node) && node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.getFilteredNodeVisibility({ node });
+        }
+        return this.#treeSpecificVisibilityHandler.getVisibilityStatus(node);
+    }
+    changeVisibilityStatusInternal(node, on) {
+        if (HierarchyNode.isClassGroupingNode(node)) {
+            if (node.extendedData?.hasDirectNonFilteredTargets && !node.extendedData?.hasFilterTargetAncestor) {
+                return this.changeFilteredNodeVisibility({ node, on });
+            }
+        }
+        if (HierarchyNode.isInstancesNode(node) && node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.changeFilteredNodeVisibility({ node, on });
+        }
+        return this.#treeSpecificVisibilityHandler.changeVisibilityStatus(node, on);
+    }
+    getFilteredNodeVisibility(props) {
+        return this.getFilteredTreeTargets(props).pipe(mergeMap((targets) => {
+            if (!targets) {
+                return EMPTY;
+            }
+            return this.#treeSpecificVisibilityHandler.getFilterTargetsVisibilityStatus(targets);
+        }));
+    }
+    getFilteredTreeTargets({ node, }) {
+        assert(this.#filteredTree !== undefined);
+        return from(this.#filteredTree).pipe(map((filteredTree) => filteredTree.getFilterTargets(node)));
+    }
+    changeFilteredNodeVisibility({ on, node, }) {
+        return this.getFilteredTreeTargets({ node }).pipe(mergeMap((targets) => {
+            if (!targets) {
+                return EMPTY;
+            }
+            return this.#treeSpecificVisibilityHandler.changeFilterTargetsVisibilityStatus(targets, on);
+        }));
+    }
+}
+//# sourceMappingURL=UseCachedVisibility.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"UseCachedVisibility.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/useTreeHooks/UseCachedVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACrI,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,kCAAkC,EAAE,MAAM,iCAAiC,CAAC;AACrF,OAAO,EAAE,8BAA8B,EAAE,MAAM,sCAAsC,CAAC;AACtF,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AACvD,OAAO,EAAE,mCAAmC,EAAE,MAAM,qCAAqC,CAAC;AAqC1F,gBAAgB;AAChB,MAAM,UAAU,mBAAmB,CAAyB,KAAuD;IACjH,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAuC,SAAS,CAAC,CAAC;IACpG,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,kBAAkB,EAAE,mCAAmC,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAE7G,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAkD,GAAG,EAAE,CAC7H,8BAA8B,CAAC;QAC7B,UAAU;QACV,QAAQ;QACR,kBAAkB;QAClB,mCAAmC;QACnC,cAAc,EAAE,aAAa;QAC7B,WAAW;KACZ,CAAC,CACH,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,GAAG,EAAE,CAC/B,8BAA8B,CAAC;YAC7B,UAAU;YACV,QAAQ;YACR,kBAAkB;YAClB,mCAAmC;YACnC,cAAc,EAAE,aAAa;YAC7B,WAAW;SACZ,CAAC,CACH,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,kBAAkB,EAAE,mCAAmC,EAAE,WAAW,CAAC,CAAC,CAAC;IAEhH,OAAO;QACL,wBAAwB;QACxB,sBAAsB,EAAE,WAAW,CAAC,CAAC,KAA2C,EAAE,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QACjH,aAAa;KACd,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,KAEC;IAED,MAAM,EAAE,UAAU,EAAE,kBAAkB,EAAE,mCAAmC,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC7H,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAC1B,IAAI,8BAA8B,CAAiB;QACjD,WAAW;QACX,QAAQ,EAAE,UAAU;QACpB,eAAe,EAAE,GAAsD,EAAE;YACvE,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,kBAAkB,CAAC,EAAE,YAAY,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE,CAC1D,mCAAmC,CAAC;YAClC,IAAI;YACJ,QAAQ;YACR,QAAQ,EAAE,UAAU;YACpB,eAAe;SAChB,CAAC;KACL,CAAC,CAAC;AACP,CAAC;AAaD;;;;;;;GAOG;AACH,MAAM,OAAO,8BAA8B;IAChC,MAAM,CAAsD;IAC5D,cAAc,CAAiC;IAC/C,4BAA4B,CAAiC;IACtE,8BAA8B,CAA6D;IAC3F,cAAc,GAAG,IAAI,OAAO,EAA4C,CAAC;IACzE,aAAa,CAAoD;IAEjE,YAAY,KAA0D;QACpE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,mCAAmC,CAAC;YACxD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,SAAS,EAAE;gBACT,MAAM,EAAE,IAAI;gBACZ,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,4BAA4B,GAAG,IAAI,8BAA8B,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;QAC3I,IAAI,CAAC,8BAA8B,GAAG,IAAI,CAAC,MAAM,CAAC,gCAAgC,CAChF,IAAI,CAAC,4BAA4B,EACjC,IAAI,kCAAkC,CAAC,IAAI,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;IACrD,CAAC;IAED,IAAW,kBAAkB;QAC3B,OAAO,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,OAAO,cAAc,CACnB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,IAAI;QACzC,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3I,oCAAoC;QACpC,SAAS,CACP,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC,EACD,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CACF,CACF,EACD,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CACjD,CACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,aAAsB;QACvE,kCAAkC;QAClC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3E,MAAM,gBAAgB,GAAG,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC,IAAI;QACpF,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAC3I,GAAG,CAAC;YACF,SAAS,EAAE,GAAG,EAAE;gBACd,IAAI,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC;gBAC3C,IAAI,CAAC,4BAA4B,CAAC,oBAAoB,EAAE,CAAC;YAC3D,CAAC;YACD,QAAQ,EAAE,GAAG,EAAE;gBACb,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;gBACzC,IAAI,CAAC,4BAA4B,CAAC,kBAAkB,EAAE,CAAC;YACzD,CAAC;SACF,CAAC,CACH,CAAC;QAEF,OAAO,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACzC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,IAAI,CAAC,4BAA4B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,8BAA8B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACxD,CAAC;IAEO,2BAA2B,CAAC,IAAmB;QACrD,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,2BAA2B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAClG,OAAO,IAAI,CAAC,yBAAyB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAED,IAAI,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,+BAA+B,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC;YACrI,OAAO,IAAI,CAAC,yBAAyB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC;IAEO,8BAA8B,CAAC,IAAmB,EAAE,EAAW;QACrE,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,2BAA2B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAClG,OAAO,IAAI,CAAC,4BAA4B,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QACD,IAAI,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,+BAA+B,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC;YACrI,OAAO,IAAI,CAAC,4BAA4B,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,sBAAsB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAEO,yBAAyB,CAAC,KAIjC;QACC,OAAO,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,IAAI,CAC5C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,gCAAgC,CAAC,OAAO,CAAC,CAAC;QACvF,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAAC,EAC7B,IAAI,GAKL;QACC,MAAM,CAAC,IAAI,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACnG,CAAC;IAEO,4BAA4B,CAAC,EACnC,EAAE,EACF,IAAI,GAML;QACC,OAAO,IAAI,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAC/C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mCAAmC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9F,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 { useCallback, useEffect, useState } from \"react\";\nimport { defaultIfEmpty, EMPTY, filter, firstValueFrom, from, fromEventPattern, map, mergeMap, Subject, takeUntil, tap } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { HierarchyNode, HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { HierarchyVisibilityOverrideHandler } from \"../../UseHierarchyVisibility.js\";\nimport { AlwaysAndNeverDrawnElementInfo } from \"../AlwaysAndNeverDrawnElementInfo.js\";\nimport { toVoidPromise } from \"../Rxjs.js\";\nimport { createVisibilityStatus } from \"../Tooltip.js\";\nimport { createVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString } from \"@itwin/core-bentley\";\nimport type { ClassGroupingNodeKey, HierarchyFilteringPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../components/VisibilityTree.js\";\nimport type { TreeWidgetViewport } from \"../../TreeWidgetViewport.js\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../UseHierarchyVisibility.js\";\nimport type { FilteredTree } from \"../visibility/BaseFilteredTree.js\";\nimport type { TreeSpecificVisibilityHandler } from \"../visibility/BaseVisibilityHelper.js\";\nimport type { IVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\n/** @internal */\nexport interface CreateFilteredTreeProps<TCache> {\n  getCache: () => TCache;\n  imodelAccess: ECClassHierarchyInspector;\n  filteringPaths: HierarchyFilteringPath[];\n}\n\n/** @internal */\nexport interface CreateTreeSpecificVisibilityHandlerProps<TCache> {\n  info: AlwaysAndNeverDrawnElementInfo;\n  getCache: () => TCache;\n  viewport: TreeWidgetViewport;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n}\n\n/** @internal */\nexport interface UseCachedVisibilityProps<TCache, TFilterTargets> {\n  componentId: GuidString;\n  activeView: TreeWidgetViewport;\n  getCache: () => TCache;\n  createFilteredTree: (props: CreateFilteredTreeProps<TCache>) => Promise<FilteredTree<TFilterTargets>>;\n  createTreeSpecificVisibilityHandler: (props: CreateTreeSpecificVisibilityHandlerProps<TCache>) => TreeSpecificVisibilityHandler<TFilterTargets> & Disposable;\n}\n\n/** @internal */\nexport function useCachedVisibility<TCache, TFilterTargets>(props: UseCachedVisibilityProps<TCache, TFilterTargets>) {\n  const [filteredPaths, setFilteredPaths] = useState<HierarchyFilteringPath[] | undefined>(undefined);\n  const { activeView, getCache, createFilteredTree, createTreeSpecificVisibilityHandler, componentId } = props;\n\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState<VisibilityTreeProps[\"visibilityHandlerFactory\"]>(() =>\n    createVisibilityHandlerFactory({\n      activeView,\n      getCache,\n      createFilteredTree,\n      createTreeSpecificVisibilityHandler,\n      filteringPaths: filteredPaths,\n      componentId,\n    }),\n  );\n\n  useEffect(() => {\n    setVisibilityHandlerFactory(() =>\n      createVisibilityHandlerFactory({\n        activeView,\n        getCache,\n        createFilteredTree,\n        createTreeSpecificVisibilityHandler,\n        filteringPaths: filteredPaths,\n        componentId,\n      }),\n    );\n  }, [activeView, getCache, filteredPaths, createFilteredTree, createTreeSpecificVisibilityHandler, componentId]);\n\n  return {\n    visibilityHandlerFactory,\n    onFilteredPathsChanged: useCallback((paths: HierarchyFilteringPath[] | undefined) => setFilteredPaths(paths), []),\n    filteredPaths,\n  };\n}\n\nfunction createVisibilityHandlerFactory<TCache, TFilterTargets>(\n  props: UseCachedVisibilityProps<TCache, TFilterTargets> & {\n    filteringPaths: HierarchyFilteringPath[] | undefined;\n  },\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  const { activeView, createFilteredTree, createTreeSpecificVisibilityHandler, getCache, filteringPaths, componentId } = props;\n  return ({ imodelAccess }) =>\n    new HierarchyVisibilityHandlerImpl<TFilterTargets>({\n      componentId,\n      viewport: activeView,\n      getFilteredTree: (): Promise<FilteredTree<TFilterTargets>> | undefined => {\n        if (filteringPaths) {\n          return createFilteredTree({ imodelAccess, filteringPaths, getCache });\n        }\n        return undefined;\n      },\n      getTreeSpecificVisibilityHandler: (info, overrideHandler) =>\n        createTreeSpecificVisibilityHandler({\n          info,\n          getCache,\n          viewport: activeView,\n          overrideHandler,\n        }),\n    });\n}\n\n/** @internal */\nexport interface HierarchyVisibilityHandlerImplProps<TFilterTargets> {\n  viewport: TreeWidgetViewport;\n  getTreeSpecificVisibilityHandler: (\n    info: AlwaysAndNeverDrawnElementInfo,\n    overrideHandler: HierarchyVisibilityOverrideHandler,\n  ) => TreeSpecificVisibilityHandler<TFilterTargets> & Disposable;\n  getFilteredTree: () => Promise<FilteredTree<TFilterTargets>> | undefined;\n  componentId?: GuidString;\n}\n\n/**\n * Class that handles visibility of hierarchy nodes in a tree.\n *\n * - It provides methods to get and change visibility status of nodes.\n * - Also handles filtered tree nodes visibility.\n * - Listens to visibility change events and updates visibility status accordingly.\n * @internal\n */\nexport class HierarchyVisibilityHandlerImpl<TFilterTargets> implements HierarchyVisibilityHandler, Disposable {\n  readonly #props: HierarchyVisibilityHandlerImplProps<TFilterTargets>;\n  readonly #eventListener: IVisibilityChangeEventListener;\n  readonly #alwaysAndNeverDrawnElements: AlwaysAndNeverDrawnElementInfo;\n  #treeSpecificVisibilityHandler: TreeSpecificVisibilityHandler<TFilterTargets> & Disposable;\n  #changeRequest = new Subject<{ key: HierarchyNodeKey; depth: number }>();\n  #filteredTree: Promise<FilteredTree<TFilterTargets>> | undefined;\n\n  constructor(props: HierarchyVisibilityHandlerImplProps<TFilterTargets>) {\n    this.#props = props;\n    this.#eventListener = createVisibilityChangeEventListener({\n      viewport: this.#props.viewport,\n      listeners: {\n        models: true,\n        categories: true,\n        elements: true,\n        displayStyle: true,\n      },\n    });\n    this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo({ viewport: this.#props.viewport, componentId: props.componentId });\n    this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(\n      this.#alwaysAndNeverDrawnElements,\n      new HierarchyVisibilityOverrideHandler(this),\n    );\n    this.#filteredTree = this.#props.getFilteredTree();\n  }\n\n  public get onVisibilityChange() {\n    return this.#eventListener.onVisibilityChange;\n  }\n\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    return firstValueFrom(\n      this.getVisibilityStatusInternal(node).pipe(\n        // unsubscribe from the observable if the change request for this node is received\n        takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n        // unsubscribe if visibility changes\n        takeUntil(\n          fromEventPattern(\n            (handler) => {\n              this.#eventListener.onVisibilityChange.addListener(handler);\n            },\n            (handler) => {\n              this.#eventListener.onVisibilityChange.removeListener(handler);\n            },\n          ),\n        ),\n        defaultIfEmpty(createVisibilityStatus(\"hidden\")),\n      ),\n    );\n  }\n\n  public async changeVisibility(node: HierarchyNode, shouldDisplay: boolean): Promise<void> {\n    // notify about new change request\n    this.#changeRequest.next({ key: node.key, depth: node.parentKeys.length });\n\n    const changeObservable = this.changeVisibilityStatusInternal(node, shouldDisplay).pipe(\n      // unsubscribe from the observable if the change request for this node is received\n      takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n      tap({\n        subscribe: () => {\n          this.#eventListener.suppressChangeEvents();\n          this.#alwaysAndNeverDrawnElements.suppressChangeEvents();\n        },\n        finalize: () => {\n          this.#eventListener.resumeChangeEvents();\n          this.#alwaysAndNeverDrawnElements.resumeChangeEvents();\n        },\n      }),\n    );\n\n    return toVoidPromise(changeObservable);\n  }\n\n  public [Symbol.dispose]() {\n    this.#eventListener[Symbol.dispose]();\n    this.#alwaysAndNeverDrawnElements[Symbol.dispose]();\n    this.#treeSpecificVisibilityHandler[Symbol.dispose]();\n  }\n\n  private getVisibilityStatusInternal(node: HierarchyNode) {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonFilteredTargets && !node.extendedData?.hasFilterTargetAncestor) {\n        return this.getFilteredNodeVisibility({ node });\n      }\n    }\n\n    if (HierarchyNode.isInstancesNode(node) && node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {\n      return this.getFilteredNodeVisibility({ node });\n    }\n    return this.#treeSpecificVisibilityHandler.getVisibilityStatus(node);\n  }\n\n  private changeVisibilityStatusInternal(node: HierarchyNode, on: boolean): Observable<void> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonFilteredTargets && !node.extendedData?.hasFilterTargetAncestor) {\n        return this.changeFilteredNodeVisibility({ node, on });\n      }\n    }\n    if (HierarchyNode.isInstancesNode(node) && node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {\n      return this.changeFilteredNodeVisibility({ node, on });\n    }\n    return this.#treeSpecificVisibilityHandler.changeVisibilityStatus(node, on);\n  }\n\n  private getFilteredNodeVisibility(props: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getFilteredTreeTargets(props).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.getFilterTargetsVisibilityStatus(targets);\n      }),\n    );\n  }\n\n  private getFilteredTreeTargets({\n    node,\n  }: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }): Observable<TFilterTargets | undefined> {\n    assert(this.#filteredTree !== undefined);\n    return from(this.#filteredTree).pipe(map((filteredTree) => filteredTree.getFilterTargets(node)));\n  }\n\n  private changeFilteredNodeVisibility({\n    on,\n    node,\n  }: {\n    on: boolean;\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getFilteredTreeTargets({ node }).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.changeFilterTargetsVisibilityStatus(targets, on);\n      }),\n    );\n  }\n}\n"]}

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

@@ -0,0 +1,20 @@
+import type { GuidString } from "@itwin/core-bentley";
+import type { IModelConnection } from "@itwin/core-frontend";
+/** @internal */
+export interface CreateCacheProps<TCacheSpecificProps> {
+    imodel: IModelConnection;
+    specificProps: TCacheSpecificProps;
+    componentId: GuidString;
+}
+/** @internal */
+export interface UseIdsCacheProps<TCache, TCacheSpecificProps> {
+    imodel: IModelConnection;
+    createCache: (props: CreateCacheProps<TCacheSpecificProps>) => TCache;
+    cacheSpecificProps: TCacheSpecificProps;
+    componentId: GuidString;
+}
+/** @internal */
+export declare function useIdsCache<TCache extends Disposable, TCacheSpecificProps extends object>(props: UseIdsCacheProps<TCache, TCacheSpecificProps>): {
+    getCache: () => TCache;
+};
+//# sourceMappingURL=UseIdsCache.d.ts.map