@itwin/tree-widget-react 3.4.2 → 3.5.0

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

@@ -0,0 +1,23 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/**
+ * @internal
+ */
+export var CategoriesTreeNode;
+(function (CategoriesTreeNode) {
+    /**
+     * Determines if node represents a definition container.
+     */
+    CategoriesTreeNode.isDefinitionContainerNode = (node) => node.extendedData && "isDefinitionContainer" in node.extendedData && !!node.extendedData.isDefinitionContainer;
+    /**
+     * Determines if node represents a category.
+     */
+    CategoriesTreeNode.isCategoryNode = (node) => node.extendedData && "isCategory" in node.extendedData && !!node.extendedData.isCategory;
+    /**
+     * Determines if node represents a sub-category.
+     */
+    CategoriesTreeNode.isSubCategoryNode = (node) => node.extendedData && "isSubCategory" in node.extendedData && !!node.extendedData.isSubCategory;
+})(CategoriesTreeNode || (CategoriesTreeNode = {}));
+//# sourceMappingURL=CategoriesTreeNode.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"CategoriesTreeNode.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeNode.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAShG;;GAEG;AACH,MAAM,KAAW,kBAAkB,CAkBlC;AAlBD,WAAiB,kBAAkB;IACjC;;OAEG;IACU,4CAAyB,GAAG,CAAC,IAA8C,EAAE,EAAE,CAC1F,IAAI,CAAC,YAAY,IAAI,uBAAuB,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAAC;IAEjH;;OAEG;IACU,iCAAc,GAAG,CAAC,IAA8C,EAAE,EAAE,CAC/E,IAAI,CAAC,YAAY,IAAI,YAAY,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC;IAE3F;;OAEG;IACU,oCAAiB,GAAG,CAAC,IAA8C,EAAE,EAAE,CAClF,IAAI,CAAC,YAAY,IAAI,eAAe,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;AACnG,CAAC,EAlBgB,kBAAkB,KAAlB,kBAAkB,QAkBlC","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 type { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\n\ninterface CategoriesTreeNode {\n  key: HierarchyNodeKey;\n  extendedData?: { [id: string]: any };\n}\n\n/**\n * @internal\n */\nexport namespace CategoriesTreeNode {\n  /**\n   * Determines if node represents a definition container.\n   */\n  export const isDefinitionContainerNode = (node: Pick<CategoriesTreeNode, \"extendedData\">) =>\n    node.extendedData && \"isDefinitionContainer\" in node.extendedData && !!node.extendedData.isDefinitionContainer;\n\n  /**\n   * Determines if node represents a category.\n   */\n  export const isCategoryNode = (node: Pick<CategoriesTreeNode, \"extendedData\">) =>\n    node.extendedData && \"isCategory\" in node.extendedData && !!node.extendedData.isCategory;\n\n  /**\n   * Determines if node represents a sub-category.\n   */\n  export const isSubCategoryNode = (node: Pick<CategoriesTreeNode, \"extendedData\">) =>\n    node.extendedData && \"isSubCategory\" in node.extendedData && !!node.extendedData.isSubCategory;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/categories-tree/{CategoriesVisibilityHandler.d.ts → internal/CategoriesVisibilityHandler.d.ts}

@@ -1,28 +1,37 @@
 import { BeEvent } from "@itwin/core-bentley";
 import { HierarchyNode } from "@itwin/presentation-hierarchies";
 import type { Viewport } from "@itwin/core-frontend";
-import type { HierarchyVisibilityHandler, VisibilityStatus } from "../common/UseHierarchyVisibility.js";
-interface CategoriesVisibilityHandlerProps {
+import type { HierarchyVisibilityHandler, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
+import type { CategoriesTreeIdsCache } from "./CategoriesTreeIdsCache.js";
+/** @internal */
+export interface CategoriesVisibilityHandlerProps {
     viewport: Viewport;
+    idsCache: CategoriesTreeIdsCache;
 }
 /** @internal */
 export declare class CategoriesVisibilityHandler implements HierarchyVisibilityHandler {
     private _pendingVisibilityChange;
     private _viewport;
+    private _idsCache;
     constructor(props: CategoriesVisibilityHandlerProps);
     dispose(): void;
+    [Symbol.dispose](): void;
     onVisibilityChange: BeEvent<import("@itwin/core-bentley").Listener>;
     /** Returns visibility status of the tree node. */
-    getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;
+    getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus>;
     changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;
-    getSubCategoryVisibility(node: HierarchyNode): "visible" | "hidden";
-    getCategoryVisibility(node: HierarchyNode): "visible" | "hidden";
+    private getSubCategoryVisibility;
+    private getDefinitionContainerVisibility;
+    private getCategoriesVisibility;
+    private getCategoryVisibilityFromOverrides;
+    private changeSubCategoryVisibility;
+    private changeCategoryVisibility;
+    private changeDefinitionContainerVisibility;
     private onDisplayStyleChanged;
     private onViewedCategoriesChanged;
     private onVisibilityChangeInternal;
-    static getInstanceIdFromHierarchyNode(node: HierarchyNode): string;
-    enableCategory(ids: string[], enabled: boolean, enableAllSubCategories?: boolean): Promise<void>;
-    enableSubCategory(key: string, enabled: boolean): void;
+    private static getInstanceIdsFromHierarchyNode;
+    private changeCategoryState;
+    private changeSubCategoryState;
 }
-export {};
 //# sourceMappingURL=CategoriesVisibilityHandler.d.ts.map

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

@@ -0,0 +1,214 @@
+/*---------------------------------------------------------------------------------------------
+ * 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";
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { enableCategoryDisplay, enableSubCategoryDisplay } from "../../common/CategoriesVisibilityUtils.js";
+import { createVisibilityStatus } from "../../common/Tooltip.js";
+import { CategoriesTreeNode } from "./CategoriesTreeNode.js";
+/** @internal */
+export class CategoriesVisibilityHandler {
+    constructor(props) {
+        this.onVisibilityChange = new BeEvent();
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onDisplayStyleChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onViewedCategoriesChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        this._idsCache = props.idsCache;
+        this._viewport = props.viewport;
+        this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);
+        this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);
+    }
+    dispose() {
+        this[Symbol.dispose]();
+    }
+    [Symbol.dispose]() {
+        this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);
+        this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);
+        clearTimeout(this._pendingVisibilityChange);
+    }
+    /** Returns visibility status of the tree node. */
+    async getVisibilityStatus(node) {
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return { state: "hidden", isDisabled: true };
+        }
+        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            return createVisibilityStatus(this.getSubCategoryVisibility(node));
+        }
+        if (CategoriesTreeNode.isCategoryNode(node)) {
+            return createVisibilityStatus(await this.getCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node)));
+        }
+        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            return createVisibilityStatus(await this.getDefinitionContainerVisibility(node));
+        }
+        return { state: "hidden", isDisabled: true };
+    }
+    async changeVisibility(node, on) {
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return;
+        }
+        if (CategoriesTreeNode.isCategoryNode(node)) {
+            return this.changeCategoryVisibility(node, on);
+        }
+        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            return this.changeSubCategoryVisibility(node, on);
+        }
+        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            return this.changeDefinitionContainerVisibility(node, on);
+        }
+    }
+    getSubCategoryVisibility(node) {
+        const parentCategoryId = node.extendedData?.categoryId;
+        if (!parentCategoryId) {
+            return "hidden";
+        }
+        const categoryOverrideResult = this.getCategoryVisibilityFromOverrides(parentCategoryId);
+        if (categoryOverrideResult === "hidden" || categoryOverrideResult === "visible") {
+            return categoryOverrideResult;
+        }
+        if (!this._viewport.view.viewsCategory(parentCategoryId)) {
+            return "hidden";
+        }
+        const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
+        let visibleCount = 0;
+        let hiddenCount = 0;
+        for (const subCategoryId of subCategoryIds) {
+            const isVisible = this._viewport.isSubCategoryVisible(subCategoryId);
+            if (isVisible) {
+                ++visibleCount;
+            }
+            else {
+                ++hiddenCount;
+            }
+            if (visibleCount > 0 && hiddenCount > 0) {
+                return "partial";
+            }
+        }
+        return visibleCount > 0 ? "visible" : "hidden";
+    }
+    async getDefinitionContainerVisibility(node) {
+        const childrenResult = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));
+        let hiddenCount = 0;
+        let visibleCount = 0;
+        for (const categoryId of childrenResult) {
+            const categoryVisibility = await this.getCategoriesVisibility([categoryId]);
+            if (categoryVisibility === "partial") {
+                return "partial";
+            }
+            if (categoryVisibility === "hidden") {
+                ++hiddenCount;
+            }
+            else {
+                ++visibleCount;
+            }
+            if (hiddenCount > 0 && visibleCount > 0) {
+                return "partial";
+            }
+        }
+        return hiddenCount > 0 ? "hidden" : "visible";
+    }
+    async getCategoriesVisibility(categoryIds) {
+        const overrideResult = this.getCategoryVisibilityFromOverrides(categoryIds);
+        if (overrideResult !== "none") {
+            return overrideResult;
+        }
+        let visibleCount = 0;
+        let hiddenCount = 0;
+        for (const categoryId of categoryIds) {
+            const isVisible = this._viewport.view.viewsCategory(categoryId);
+            if (isVisible) {
+                ++visibleCount;
+            }
+            else {
+                ++hiddenCount;
+            }
+            if (visibleCount > 0 && hiddenCount > 0) {
+                return "partial";
+            }
+        }
+        if (hiddenCount > 0) {
+            return "hidden";
+        }
+        const subCategories = (await Promise.all(categoryIds.map(async (id) => this._idsCache.getSubCategories(id)))).reduce((acc, val) => acc.concat(val), []);
+        let visibleSubCategoryCount = 0;
+        let hiddenSubCategoryCount = 0;
+        for (const subCategory of subCategories) {
+            const isVisible = this._viewport.isSubCategoryVisible(subCategory);
+            if (isVisible) {
+                ++visibleSubCategoryCount;
+            }
+            else {
+                ++hiddenSubCategoryCount;
+            }
+            if (hiddenSubCategoryCount > 0 && visibleSubCategoryCount > 0) {
+                return "partial";
+            }
+        }
+        return hiddenSubCategoryCount > 0 ? "hidden" : "visible";
+    }
+    getCategoryVisibilityFromOverrides(categoryIds) {
+        let showOverrides = 0;
+        let hideOverrides = 0;
+        for (const currentOverride of this._viewport.perModelCategoryVisibility) {
+            if (categoryIds.includes(currentOverride.categoryId)) {
+                if (currentOverride.visible) {
+                    ++showOverrides;
+                }
+                else {
+                    ++hideOverrides;
+                }
+                if (showOverrides > 0 && hideOverrides > 0) {
+                    return "partial";
+                }
+            }
+        }
+        if (showOverrides === 0 && hideOverrides === 0) {
+            return "none";
+        }
+        return showOverrides > 0 ? "visible" : "hidden";
+    }
+    async changeSubCategoryVisibility(node, on) {
+        const parentCategoryId = node.extendedData?.categoryId;
+        // make sure parent category is enabled
+        if (on && parentCategoryId) {
+            await this.changeCategoryState([parentCategoryId], true, false);
+        }
+        const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
+        subCategoryIds.forEach((id) => {
+            this.changeSubCategoryState(id, on);
+        });
+    }
+    async changeCategoryVisibility(node, on) {
+        const categoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
+        return this.changeCategoryState(categoryIds, on, on);
+    }
+    async changeDefinitionContainerVisibility(node, on) {
+        const definitionContainerId = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);
+        const childCategories = await this._idsCache.getAllContainedCategories(definitionContainerId);
+        return this.changeCategoryState(childCategories, on, on);
+    }
+    onVisibilityChangeInternal() {
+        if (this._pendingVisibilityChange) {
+            return;
+        }
+        this._pendingVisibilityChange = setTimeout(() => {
+            this.onVisibilityChange.raiseEvent();
+            this._pendingVisibilityChange = undefined;
+        }, 0);
+    }
+    static getInstanceIdsFromHierarchyNode(node) {
+        return HierarchyNode.isInstancesNode(node) ? node.key.instanceKeys.map((instanceKey) => instanceKey.id) : /* istanbul ignore next */ [];
+    }
+    async changeCategoryState(ids, enabled, enableAllSubCategories) {
+        await enableCategoryDisplay(this._viewport, ids, enabled, enableAllSubCategories);
+    }
+    changeSubCategoryState(key, enabled) {
+        enableSubCategoryDisplay(this._viewport, key, enabled);
+    }
+}
+//# sourceMappingURL=CategoriesVisibilityHandler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"CategoriesVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAC5G,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAa7D,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IAKtC,YAAY,KAAuC;QAiB5C,uBAAkB,GAAG,IAAI,OAAO,EAAE,CAAC;QA4L1C,gEAAgE;QACxD,0BAAqB,GAAG,GAAG,EAAE;YACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,gEAAgE;QACxD,8BAAyB,GAAG,GAAG,EAAE;YACvC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QApNA,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7E,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,WAAW,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACvF,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACzB,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,cAAc,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxF,YAAY,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;IAID,kDAAkD;IAC3C,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC9C;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,OAAO,sBAAsB,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC;SACpE;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACtI;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,sBAAsB,CAAC,MAAM,IAAI,CAAC,gCAAgC,CAAC,IAAI,CAAC,CAAC,CAAC;SAClF;QAED,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAC/C,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,EAAW;QAC5D,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO;SACR;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YAC3C,OAAO,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SAChD;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YAC9C,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SACnD;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE;YACtD,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAEO,wBAAwB,CAAC,IAAmB;QAClD,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QACvD,IAAI,CAAC,gBAAgB,EAAE;YACrB,OAAO,QAAQ,CAAC;SACjB;QAED,MAAM,sBAAsB,GAAG,IAAI,CAAC,kCAAkC,CAAC,gBAAgB,CAAC,CAAC;QACzF,IAAI,sBAAsB,KAAK,QAAQ,IAAI,sBAAsB,KAAK,SAAS,EAAE;YAC/E,OAAO,sBAAsB,CAAC;SAC/B;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,EAAE;YACxD,OAAO,QAAQ,CAAC;SACjB;QACD,MAAM,cAAc,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACzF,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACrE,IAAI,SAAS,EAAE;gBACb,EAAE,YAAY,CAAC;aAChB;iBAAM;gBACL,EAAE,WAAW,CAAC;aACf;YACD,IAAI,YAAY,GAAG,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QACD,OAAO,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gCAAgC,CAAC,IAAmB;QAChE,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC,CAAC;QACzI,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,KAAK,MAAM,UAAU,IAAI,cAAc,EAAE;YACvC,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5E,IAAI,kBAAkB,KAAK,SAAS,EAAE;gBACpC,OAAO,SAAS,CAAC;aAClB;YAED,IAAI,kBAAkB,KAAK,QAAQ,EAAE;gBACnC,EAAE,WAAW,CAAC;aACf;iBAAM;gBACL,EAAE,YAAY,CAAC;aAChB;YAED,IAAI,WAAW,GAAG,CAAC,IAAI,YAAY,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QAED,OAAO,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,WAAsB;QAC1D,MAAM,cAAc,GAAG,IAAI,CAAC,kCAAkC,CAAC,WAAW,CAAC,CAAC;QAC5E,IAAI,cAAc,KAAK,MAAM,EAAE;YAC7B,OAAO,cAAc,CAAC;SACvB;QACD,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE;YACpC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAChE,IAAI,SAAS,EAAE;gBACb,EAAE,YAAY,CAAC;aAChB;iBAAM;gBACL,EAAE,WAAW,CAAC;aACf;YACD,IAAI,YAAY,GAAG,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE;gBACvC,OAAO,SAAS,CAAC;aAClB;SACF;QAED,IAAI,WAAW,GAAG,CAAC,EAAE;YACnB,OAAO,QAAQ,CAAC;SACjB;QAED,MAAM,aAAa,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QACxJ,IAAI,uBAAuB,GAAG,CAAC,CAAC;QAChC,IAAI,sBAAsB,GAAG,CAAC,CAAC;QAE/B,KAAK,MAAM,WAAW,IAAI,aAAa,EAAE;YACvC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACnE,IAAI,SAAS,EAAE;gBACb,EAAE,uBAAuB,CAAC;aAC3B;iBAAM;gBACL,EAAE,sBAAsB,CAAC;aAC1B;YACD,IAAI,sBAAsB,GAAG,CAAC,IAAI,uBAAuB,GAAG,CAAC,EAAE;gBAC7D,OAAO,SAAS,CAAC;aAClB;SACF;QAED,OAAO,sBAAsB,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3D,CAAC;IAEO,kCAAkC,CAAC,WAAsB;QAC/D,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,aAAa,GAAG,CAAC,CAAC;QAEtB,KAAK,MAAM,eAAe,IAAI,IAAI,CAAC,SAAS,CAAC,0BAA0B,EAAE;YACvE,IAAI,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE;gBACpD,IAAI,eAAe,CAAC,OAAO,EAAE;oBAC3B,EAAE,aAAa,CAAC;iBACjB;qBAAM;oBACL,EAAE,aAAa,CAAC;iBACjB;gBAED,IAAI,aAAa,GAAG,CAAC,IAAI,aAAa,GAAG,CAAC,EAAE;oBAC1C,OAAO,SAAS,CAAC;iBAClB;aACF;SACF;QAED,IAAI,aAAa,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC,EAAE;YAC9C,OAAO,MAAM,CAAC;SACf;QAED,OAAO,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,IAAmB,EAAE,EAAW;QACxE,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QAEvD,uCAAuC;QACvC,IAAI,EAAE,IAAI,gBAAgB,EAAE;YAC1B,MAAM,IAAI,CAAC,mBAAmB,CAAC,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;SACjE;QAED,MAAM,cAAc,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACzF,cAAc,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YAC5B,IAAI,CAAC,sBAAsB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAAC,IAAmB,EAAE,EAAW;QACrE,MAAM,WAAW,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QACtF,OAAO,IAAI,CAAC,mBAAmB,CAAC,WAAW,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,mCAAmC,CAAC,IAAmB,EAAE,EAAW;QAChF,MAAM,qBAAqB,GAAG,2BAA2B,CAAC,+BAA+B,CAAC,IAAI,CAAC,CAAC;QAChG,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,qBAAqB,CAAC,CAAC;QAC9F,OAAO,IAAI,CAAC,mBAAmB,CAAC,eAAe,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IAC3D,CAAC;IAYO,0BAA0B;QAChC,IAAI,IAAI,CAAC,wBAAwB,EAAE;YACjC,OAAO;SACR;QAED,IAAI,CAAC,wBAAwB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9C,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAC;QAC5C,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,MAAM,CAAC,+BAA+B,CAAC,IAAmB;QAChE,OAAO,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,EAAE,CAAC;IAC1I,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,GAAa,EAAE,OAAgB,EAAE,sBAA+B;QAChG,MAAM,qBAAqB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IACpF,CAAC;IAEO,sBAAsB,CAAC,GAAW,EAAE,OAAgB;QAC1D,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IACzD,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 { BeEvent } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { enableCategoryDisplay, enableSubCategoryDisplay } from \"../../common/CategoriesVisibilityUtils.js\";\nimport { createVisibilityStatus } from \"../../common/Tooltip.js\";\nimport { CategoriesTreeNode } from \"./CategoriesTreeNode.js\";\n\nimport type { Id64Array } from \"@itwin/core-bentley\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeIdsCache } from \"./CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport interface CategoriesVisibilityHandlerProps {\n  viewport: Viewport;\n  idsCache: CategoriesTreeIdsCache;\n}\n\n/** @internal */\nexport class CategoriesVisibilityHandler implements HierarchyVisibilityHandler {\n  private _pendingVisibilityChange: any;\n  private _viewport: Viewport;\n  private _idsCache: CategoriesTreeIdsCache;\n\n  constructor(props: CategoriesVisibilityHandlerProps) {\n    this._idsCache = props.idsCache;\n    this._viewport = props.viewport;\n    this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);\n  }\n\n  public dispose() {\n    this[Symbol.dispose]();\n  }\n\n  public [Symbol.dispose]() {\n    this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);\n    clearTimeout(this._pendingVisibilityChange);\n  }\n\n  public onVisibilityChange = new BeEvent();\n\n  /** Returns visibility status of the tree node. */\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return { state: \"hidden\", isDisabled: true };\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return createVisibilityStatus(this.getSubCategoryVisibility(node));\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return createVisibilityStatus(await this.getCategoriesVisibility(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node)));\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return createVisibilityStatus(await this.getDefinitionContainerVisibility(node));\n    }\n\n    return { state: \"hidden\", isDisabled: true };\n  }\n\n  public async changeVisibility(node: HierarchyNode, on: boolean) {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return;\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return this.changeCategoryVisibility(node, on);\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return this.changeSubCategoryVisibility(node, on);\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return this.changeDefinitionContainerVisibility(node, on);\n    }\n  }\n\n  private getSubCategoryVisibility(node: HierarchyNode): VisibilityStatus[\"state\"] {\n    const parentCategoryId = node.extendedData?.categoryId;\n    if (!parentCategoryId) {\n      return \"hidden\";\n    }\n\n    const categoryOverrideResult = this.getCategoryVisibilityFromOverrides(parentCategoryId);\n    if (categoryOverrideResult === \"hidden\" || categoryOverrideResult === \"visible\") {\n      return categoryOverrideResult;\n    }\n\n    if (!this._viewport.view.viewsCategory(parentCategoryId)) {\n      return \"hidden\";\n    }\n    const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    let visibleCount = 0;\n    let hiddenCount = 0;\n    for (const subCategoryId of subCategoryIds) {\n      const isVisible = this._viewport.isSubCategoryVisible(subCategoryId);\n      if (isVisible) {\n        ++visibleCount;\n      } else {\n        ++hiddenCount;\n      }\n      if (visibleCount > 0 && hiddenCount > 0) {\n        return \"partial\";\n      }\n    }\n    return visibleCount > 0 ? \"visible\" : \"hidden\";\n  }\n\n  private async getDefinitionContainerVisibility(node: HierarchyNode): Promise<VisibilityStatus[\"state\"]> {\n    const childrenResult = await this._idsCache.getAllContainedCategories(CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node));\n    let hiddenCount = 0;\n    let visibleCount = 0;\n    for (const categoryId of childrenResult) {\n      const categoryVisibility = await this.getCategoriesVisibility([categoryId]);\n      if (categoryVisibility === \"partial\") {\n        return \"partial\";\n      }\n\n      if (categoryVisibility === \"hidden\") {\n        ++hiddenCount;\n      } else {\n        ++visibleCount;\n      }\n\n      if (hiddenCount > 0 && visibleCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    return hiddenCount > 0 ? \"hidden\" : \"visible\";\n  }\n\n  private async getCategoriesVisibility(categoryIds: Id64Array): Promise<VisibilityStatus[\"state\"]> {\n    const overrideResult = this.getCategoryVisibilityFromOverrides(categoryIds);\n    if (overrideResult !== \"none\") {\n      return overrideResult;\n    }\n    let visibleCount = 0;\n    let hiddenCount = 0;\n    for (const categoryId of categoryIds) {\n      const isVisible = this._viewport.view.viewsCategory(categoryId);\n      if (isVisible) {\n        ++visibleCount;\n      } else {\n        ++hiddenCount;\n      }\n      if (visibleCount > 0 && hiddenCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    if (hiddenCount > 0) {\n      return \"hidden\";\n    }\n\n    const subCategories = (await Promise.all(categoryIds.map(async (id) => this._idsCache.getSubCategories(id)))).reduce((acc, val) => acc.concat(val), []);\n    let visibleSubCategoryCount = 0;\n    let hiddenSubCategoryCount = 0;\n\n    for (const subCategory of subCategories) {\n      const isVisible = this._viewport.isSubCategoryVisible(subCategory);\n      if (isVisible) {\n        ++visibleSubCategoryCount;\n      } else {\n        ++hiddenSubCategoryCount;\n      }\n      if (hiddenSubCategoryCount > 0 && visibleSubCategoryCount > 0) {\n        return \"partial\";\n      }\n    }\n\n    return hiddenSubCategoryCount > 0 ? \"hidden\" : \"visible\";\n  }\n\n  private getCategoryVisibilityFromOverrides(categoryIds: Id64Array): VisibilityStatus[\"state\"] | \"none\" {\n    let showOverrides = 0;\n    let hideOverrides = 0;\n\n    for (const currentOverride of this._viewport.perModelCategoryVisibility) {\n      if (categoryIds.includes(currentOverride.categoryId)) {\n        if (currentOverride.visible) {\n          ++showOverrides;\n        } else {\n          ++hideOverrides;\n        }\n\n        if (showOverrides > 0 && hideOverrides > 0) {\n          return \"partial\";\n        }\n      }\n    }\n\n    if (showOverrides === 0 && hideOverrides === 0) {\n      return \"none\";\n    }\n\n    return showOverrides > 0 ? \"visible\" : \"hidden\";\n  }\n\n  private async changeSubCategoryVisibility(node: HierarchyNode, on: boolean) {\n    const parentCategoryId = node.extendedData?.categoryId;\n\n    // make sure parent category is enabled\n    if (on && parentCategoryId) {\n      await this.changeCategoryState([parentCategoryId], true, false);\n    }\n\n    const subCategoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    subCategoryIds.forEach((id) => {\n      this.changeSubCategoryState(id, on);\n    });\n  }\n\n  private async changeCategoryVisibility(node: HierarchyNode, on: boolean) {\n    const categoryIds = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    return this.changeCategoryState(categoryIds, on, on);\n  }\n\n  private async changeDefinitionContainerVisibility(node: HierarchyNode, on: boolean) {\n    const definitionContainerId = CategoriesVisibilityHandler.getInstanceIdsFromHierarchyNode(node);\n    const childCategories = await this._idsCache.getAllContainedCategories(definitionContainerId);\n    return this.changeCategoryState(childCategories, on, on);\n  }\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onDisplayStyleChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onViewedCategoriesChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onVisibilityChangeInternal() {\n    if (this._pendingVisibilityChange) {\n      return;\n    }\n\n    this._pendingVisibilityChange = setTimeout(() => {\n      this.onVisibilityChange.raiseEvent();\n      this._pendingVisibilityChange = undefined;\n    }, 0);\n  }\n\n  private static getInstanceIdsFromHierarchyNode(node: HierarchyNode) {\n    return HierarchyNode.isInstancesNode(node) ? node.key.instanceKeys.map((instanceKey) => instanceKey.id) : /* istanbul ignore next */ [];\n  }\n\n  private async changeCategoryState(ids: string[], enabled: boolean, enableAllSubCategories: boolean) {\n    await enableCategoryDisplay(this._viewport, ids, enabled, enableAllSubCategories);\n  }\n\n  private changeSubCategoryState(key: string, enabled: boolean) {\n    enableSubCategoryDisplay(this._viewport, key, enabled);\n  }\n}\n"]}

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

@@ -0,0 +1,7 @@
+/** @internal */
+export declare const SUB_CATEGORY_CLASS = "BisCore.SubCategory";
+/** @internal */
+export declare const DEFINITION_CONTAINER_CLASS = "BisCore.DefinitionContainer";
+/** @internal */
+export declare const DEFINITION_ELEMENT_CLASS = "BisCore.DefinitionElement";
+//# sourceMappingURL=ClassNameDefinitions.d.ts.map

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

@@ -0,0 +1,11 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/** @internal */
+export const SUB_CATEGORY_CLASS = "BisCore.SubCategory";
+/** @internal */
+export const DEFINITION_CONTAINER_CLASS = "BisCore.DefinitionContainer";
+/** @internal */
+export const DEFINITION_ELEMENT_CLASS = "BisCore.DefinitionElement";
+//# sourceMappingURL=ClassNameDefinitions.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ClassNameDefinitions.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/ClassNameDefinitions.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,gBAAgB;AAChB,MAAM,CAAC,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;AAExD,gBAAgB;AAChB,MAAM,CAAC,MAAM,0BAA0B,GAAG,6BAA6B,CAAC;AAExE,gBAAgB;AAChB,MAAM,CAAC,MAAM,wBAAwB,GAAG,2BAA2B,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\n/** @internal */\nexport const SUB_CATEGORY_CLASS = \"BisCore.SubCategory\";\n\n/** @internal */\nexport const DEFINITION_CONTAINER_CLASS = \"BisCore.DefinitionContainer\";\n\n/** @internal */\nexport const DEFINITION_ELEMENT_CLASS = \"BisCore.DefinitionElement\";\n"]}

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

@@ -11,10 +11,6 @@ export interface CategoryInfo {
  * Toggles visibility of categories to show or hide.
  */
 export declare function toggleAllCategories(viewport: Viewport, display: boolean): Promise<void>;
-/**
- * Gets ids of all categories from specified imodel and viewport.
- */
-export declare function getCategories(viewport: Viewport): Promise<string[]>;
 /**
  * Changes category display in the viewport.
  */

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

@@ -17,7 +17,7 @@ export async function toggleAllCategories(viewport, display) {
 /**
  * Gets ids of all categories from specified imodel and viewport.
  */
-export async function getCategories(viewport) {
+async function getCategories(viewport) {
     const categories = await loadCategoriesFromViewport(viewport);
     return categories.map((category) => category.categoryId);
 }

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesVisibilityUtils.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/CategoriesVisibilityUtils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAalE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,QAAkB,EAAE,OAAgB;IAC5E,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;QACpB,OAAO;KACR;IAED,MAAM,qBAAqB,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,QAAkB;IACpD,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;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAAkB,EAAE,GAAa,EAAE,OAAgB,EAAE,sBAAsB,GAAG,IAAI;IAC5H,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IAErE,sCAAsC;IACtC,MAAM,yBAAyB,GAAa,EAAE,CAAC;IAC/C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,0BAA0B,EAAE;QACrD,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE;YACvD,yBAAyB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC7C;KACF;IACD,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,yBAAyB,EAAE,GAAG,EAAE,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAE1H,6GAA6G;IAC7G,IAAI,KAAK,KAAK,OAAO,EAAE;QACrB,CAAC,MAAM,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;YAC/E,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;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAkB,EAAE,GAAW,EAAE,OAAgB;IACxF,QAAQ,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAY;IAC3D,yCAAyC;IACzC,MAAM,4BAA4B,GAChC,2IAA2I,CAAC;IAC9I,MAAM,4BAA4B,GAChC,0JAA0J,CAAC;IAC7J,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,4BAA4B,CAAC;IAC3F,MAAM,MAAM,GAAG,kCAAkC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,yBAAyB,2BAA2B,KAAK,GAAG,CAAC;IAE3J,MAAM,UAAU,GAAmB,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9H,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;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,UAAoB,EAAE,QAAkB;IAC9E,MAAM,qBAAqB,CAAC,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAChE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,UAAoB,EAAE,QAAkB;IAC9E,MAAM,qBAAqB,CAAC,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;AACjE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,UAA0B,EAAE,QAAkB;IACtF,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,oBAAoB,GAAa,EAAE,CAAC;IAC1C,MAAM,qBAAqB,GAAa,EAAE,CAAC;IAE3C,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE;QACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnC,SAAS;SACV;QACD,iJAAiJ;QACjJ,IAAI,QAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,EAAE;YAC/F,KAAK,MAAM,WAAW,IAAI,QAAQ,CAAC,cAAc,EAAE;gBACjD,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;aAC/H;SACF;aAAM;YACL,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;SACnC;KACF;IAED,kBAAkB;IAClB,oBAAoB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;IAEtG,MAAM,qBAAqB,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IAE5D,kBAAkB;IAClB,qBAAqB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC;IAEtG,MAAM,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9D,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 { QueryRowFormat } from \"@itwin/core-common\";\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\n\n/**\n * Data structure that describes category.\n * @beta\n */\nexport interface CategoryInfo {\n  categoryId: string;\n  subCategoryIds?: string[];\n}\n\n/**\n * Toggles visibility of categories to show or hide.\n */\nexport async function toggleAllCategories(viewport: Viewport, display: boolean) {\n  const ids = await getCategories(viewport);\n  if (ids.length === 0) {\n    return;\n  }\n\n  await enableCategoryDisplay(viewport, ids, display);\n}\n\n/**\n * Gets ids of all categories from specified imodel and viewport.\n */\nexport async function getCategories(viewport: Viewport) {\n  const categories = await loadCategoriesFromViewport(viewport);\n  return categories.map((category) => category.categoryId);\n}\n\n/**\n * Changes category display in the viewport.\n */\nexport async function enableCategoryDisplay(viewport: Viewport, ids: string[], enabled: boolean, enableAllSubCategories = true) {\n  viewport.changeCategoryDisplay(ids, enabled, enableAllSubCategories);\n\n  // remove category overrides per model\n  const modelsContainingOverrides: string[] = [];\n  for (const ovr of viewport.perModelCategoryVisibility) {\n    if (ids.findIndex((id) => id === ovr.categoryId) !== -1) {\n      modelsContainingOverrides.push(ovr.modelId);\n    }\n  }\n  viewport.perModelCategoryVisibility.setOverride(modelsContainingOverrides, ids, 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(ids)).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 */\nexport function enableSubCategoryDisplay(viewport: Viewport, key: string, enabled: boolean) {\n  viewport.changeSubCategoryDisplay(key, enabled);\n}\n\nexport async function loadCategoriesFromViewport(vp: Viewport) {\n  // Query categories and add them to state\n  const selectUsedSpatialCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement3d WHERE Category.Id IN (SELECT ECInstanceId from BisCore.SpatialCategory)\";\n  const selectUsedDrawingCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement2d WHERE Model.Id=? AND Category.Id IN (SELECT ECInstanceId from BisCore.DrawingCategory)\";\n  const ecsql = vp.view.is3d() ? selectUsedSpatialCategoryIds : selectUsedDrawingCategoryIds;\n  const ecsql2 = `SELECT ECInstanceId as id FROM ${vp.view.is3d() ? \"BisCore.SpatialCategory\" : \"BisCore.DrawingCategory\"} WHERE ECInstanceId IN (${ecsql})`;\n\n  const categories: CategoryInfo[] = [];\n\n  const rows = await vp.iModel.createQueryReader(ecsql2, 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\n/**\n * Enable display of all given categories.\n * @public\n */\nexport async function showAllCategories(categories: string[], viewport: Viewport) {\n  await enableCategoryDisplay(viewport, categories, true, true);\n}\n\n/**\n * Disable display of all given categories.\n * @public\n */\nexport async function hideAllCategories(categories: string[], viewport: Viewport) {\n  await enableCategoryDisplay(viewport, categories, false, true);\n}\n\n/**\n * Invert display of all given categories.\n * @public\n */\nexport async function invertAllCategories(categories: CategoryInfo[], viewport: Viewport) {\n  const enabled: string[] = [];\n  const disabled: string[] = [];\n  const enabledSubCategories: string[] = [];\n  const disabledSubCategories: string[] = [];\n\n  for (const category of categories) {\n    if (!viewport.view.viewsCategory(category.categoryId)) {\n      disabled.push(category.categoryId);\n      continue;\n    }\n    // First, we need to check if at least one subcategory is disabled. If it is true, then only subcategories should change display, not categories.\n    if (category.subCategoryIds?.some((subCategory) => !viewport.isSubCategoryVisible(subCategory))) {\n      for (const subCategory of category.subCategoryIds) {\n        viewport.isSubCategoryVisible(subCategory) ? enabledSubCategories.push(subCategory) : disabledSubCategories.push(subCategory);\n      }\n    } else {\n      enabled.push(category.categoryId);\n    }\n  }\n\n  // Disable enabled\n  enabledSubCategories.forEach((subCategory) => enableSubCategoryDisplay(viewport, subCategory, false));\n\n  await enableCategoryDisplay(viewport, enabled, false, true);\n\n  // Enable disabled\n  disabledSubCategories.forEach((subCategory) => enableSubCategoryDisplay(viewport, subCategory, true));\n\n  await enableCategoryDisplay(viewport, disabled, true, true);\n}\n"]}
+{"version":3,"file":"CategoriesVisibilityUtils.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/CategoriesVisibilityUtils.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAalE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,QAAkB,EAAE,OAAgB;IAC5E,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;QACpB,OAAO;KACR;IAED,MAAM,qBAAqB,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,aAAa,CAAC,QAAkB;IAC7C,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;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAAkB,EAAE,GAAa,EAAE,OAAgB,EAAE,sBAAsB,GAAG,IAAI;IAC5H,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;IAErE,sCAAsC;IACtC,MAAM,yBAAyB,GAAa,EAAE,CAAC;IAC/C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,0BAA0B,EAAE;QACrD,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE;YACvD,yBAAyB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC7C;KACF;IACD,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,yBAAyB,EAAE,GAAG,EAAE,0BAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAE1H,6GAA6G;IAC7G,IAAI,KAAK,KAAK,OAAO,EAAE;QACrB,CAAC,MAAM,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;YAC/E,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;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAkB,EAAE,GAAW,EAAE,OAAgB;IACxF,QAAQ,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAAC,EAAY;IAC3D,yCAAyC;IACzC,MAAM,4BAA4B,GAChC,2IAA2I,CAAC;IAC9I,MAAM,4BAA4B,GAChC,0JAA0J,CAAC;IAC7J,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,4BAA4B,CAAC;IAC3F,MAAM,MAAM,GAAG,kCAAkC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,yBAAyB,2BAA2B,KAAK,GAAG,CAAC;IAE3J,MAAM,UAAU,GAAmB,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9H,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;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,UAAoB,EAAE,QAAkB;IAC9E,MAAM,qBAAqB,CAAC,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAChE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,UAAoB,EAAE,QAAkB;IAC9E,MAAM,qBAAqB,CAAC,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;AACjE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,UAA0B,EAAE,QAAkB;IACtF,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,oBAAoB,GAAa,EAAE,CAAC;IAC1C,MAAM,qBAAqB,GAAa,EAAE,CAAC;IAE3C,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE;QACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnC,SAAS;SACV;QACD,iJAAiJ;QACjJ,IAAI,QAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,EAAE;YAC/F,KAAK,MAAM,WAAW,IAAI,QAAQ,CAAC,cAAc,EAAE;gBACjD,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;aAC/H;SACF;aAAM;YACL,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;SACnC;KACF;IAED,kBAAkB;IAClB,oBAAoB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;IAEtG,MAAM,qBAAqB,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IAE5D,kBAAkB;IAClB,qBAAqB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,wBAAwB,CAAC,QAAQ,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC;IAEtG,MAAM,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9D,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 { QueryRowFormat } from \"@itwin/core-common\";\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\n\n/**\n * Data structure that describes category.\n * @beta\n */\nexport interface CategoryInfo {\n  categoryId: string;\n  subCategoryIds?: string[];\n}\n\n/**\n * Toggles visibility of categories to show or hide.\n */\nexport async function toggleAllCategories(viewport: Viewport, display: boolean) {\n  const ids = await getCategories(viewport);\n  if (ids.length === 0) {\n    return;\n  }\n\n  await enableCategoryDisplay(viewport, ids, display);\n}\n\n/**\n * Gets ids of all categories from specified imodel and viewport.\n */\nasync function getCategories(viewport: Viewport) {\n  const categories = await loadCategoriesFromViewport(viewport);\n  return categories.map((category) => category.categoryId);\n}\n\n/**\n * Changes category display in the viewport.\n */\nexport async function enableCategoryDisplay(viewport: Viewport, ids: string[], enabled: boolean, enableAllSubCategories = true) {\n  viewport.changeCategoryDisplay(ids, enabled, enableAllSubCategories);\n\n  // remove category overrides per model\n  const modelsContainingOverrides: string[] = [];\n  for (const ovr of viewport.perModelCategoryVisibility) {\n    if (ids.findIndex((id) => id === ovr.categoryId) !== -1) {\n      modelsContainingOverrides.push(ovr.modelId);\n    }\n  }\n  viewport.perModelCategoryVisibility.setOverride(modelsContainingOverrides, ids, 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(ids)).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 */\nexport function enableSubCategoryDisplay(viewport: Viewport, key: string, enabled: boolean) {\n  viewport.changeSubCategoryDisplay(key, enabled);\n}\n\nexport async function loadCategoriesFromViewport(vp: Viewport) {\n  // Query categories and add them to state\n  const selectUsedSpatialCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement3d WHERE Category.Id IN (SELECT ECInstanceId from BisCore.SpatialCategory)\";\n  const selectUsedDrawingCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement2d WHERE Model.Id=? AND Category.Id IN (SELECT ECInstanceId from BisCore.DrawingCategory)\";\n  const ecsql = vp.view.is3d() ? selectUsedSpatialCategoryIds : selectUsedDrawingCategoryIds;\n  const ecsql2 = `SELECT ECInstanceId as id FROM ${vp.view.is3d() ? \"BisCore.SpatialCategory\" : \"BisCore.DrawingCategory\"} WHERE ECInstanceId IN (${ecsql})`;\n\n  const categories: CategoryInfo[] = [];\n\n  const rows = await vp.iModel.createQueryReader(ecsql2, 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\n/**\n * Enable display of all given categories.\n * @public\n */\nexport async function showAllCategories(categories: string[], viewport: Viewport) {\n  await enableCategoryDisplay(viewport, categories, true, true);\n}\n\n/**\n * Disable display of all given categories.\n * @public\n */\nexport async function hideAllCategories(categories: string[], viewport: Viewport) {\n  await enableCategoryDisplay(viewport, categories, false, true);\n}\n\n/**\n * Invert display of all given categories.\n * @public\n */\nexport async function invertAllCategories(categories: CategoryInfo[], viewport: Viewport) {\n  const enabled: string[] = [];\n  const disabled: string[] = [];\n  const enabledSubCategories: string[] = [];\n  const disabledSubCategories: string[] = [];\n\n  for (const category of categories) {\n    if (!viewport.view.viewsCategory(category.categoryId)) {\n      disabled.push(category.categoryId);\n      continue;\n    }\n    // First, we need to check if at least one subcategory is disabled. If it is true, then only subcategories should change display, not categories.\n    if (category.subCategoryIds?.some((subCategory) => !viewport.isSubCategoryVisible(subCategory))) {\n      for (const subCategory of category.subCategoryIds) {\n        viewport.isSubCategoryVisible(subCategory) ? enabledSubCategories.push(subCategory) : disabledSubCategories.push(subCategory);\n      }\n    } else {\n      enabled.push(category.categoryId);\n    }\n  }\n\n  // Disable enabled\n  enabledSubCategories.forEach((subCategory) => enableSubCategoryDisplay(viewport, subCategory, false));\n\n  await enableCategoryDisplay(viewport, enabled, false, true);\n\n  // Enable disabled\n  disabledSubCategories.forEach((subCategory) => enableSubCategoryDisplay(viewport, subCategory, true));\n\n  await enableCategoryDisplay(viewport, disabled, true, true);\n}\n"]}

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

@@ -1 +1 @@
-{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EACL,cAAc,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,qBAAqB,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,YAAY,GAC/I,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAsC/D,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,mBAAmB,GAAG,MAAM,CAAC,IAAI,GAAG,EAAgG,CAAC,CAAC;IAC5I,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAqD;QACrF,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC5D,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;QAC3B,cAAc,EAAE,GAAG,EAAE,GAAE,CAAC;KACzB,CAAC,CAAC;IACH,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,wBAAwB,EAAE,CAAC;QAE3C,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAQ,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,OAAO,EAA6B,CAAC;QAC3D,MAAM,mBAAmB,GAAG,CAAC,IAA+B,EAAE,EAAE;YAC9D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,GAAG,EAAE;YACtB,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC5C,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YACH,iBAAiB,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;aAC9E,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,SAAS,CAAC,cAAc,CAAC,EACzB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAChB,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChE,GAAG,CAAC;YACF,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvC,IAAI;oBACJ,MAAM;oBACN,YAAY,EAAE,KAAK;iBACpB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,EACF,SAAS,CAAC,iBAAiB,CAAC,EAC5B,qBAAqB,CAAC,KAAK,CAAC,CAC7B,CACF,EACD,YAAY,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CACjE;aACA,SAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE;gBACT,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAG,CAAC,IAA+B,EAAE,OAAgB,EAAE,EAAE;YAC7E,aAAa,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,KAAK,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,KAAK,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK,EAAE;gBACV,OAAO;aACR;YACD,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;YACpD,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;YACjC,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,iBAAiB,EAAE,gBAAgB;YACnC,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YAC7E,cAAc,EAAE,GAAG,EAAE;gBACnB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,UAAU,EAAE,CAAC;YACb,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,YAAY,CAAC,WAAW,EAAE,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,EAAE,aAAa,CAAC,CAAC,CAAC;IAE9C,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CACxB,GAAwH,EACxH,mBAA8D;IAE9D,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC3C;QACD,IAAI,KAAK,CAAC,YAAY,EAAE;YACtB,mBAAmB,CAAC,IAAI,CAAC,CAAC;SAC3B;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACxF,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AA0BD,MAAM,UAAU,6BAA6B,CAC3C,OAAmC,EACnC,KAAqB,EACrB,GAAwB,EACxB,QAA8G;IAE9G,OAAO,QAAQ;QACb,CAAC,CAAC,IAAI,CACF,QAAQ,CAAC;YACP,GAAG,KAAK;YACR,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,EAAE,YAAY,EAAE,SAAoB,EAAE,CAAC;YAC9F,OAAO;SACR,CAAC,CACH;QACH,CAAC,CAAC,GAAG,CAAC;AACV,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 { useEffect, useRef, useState } from \"react\";\nimport {\n  asyncScheduler, defer, distinct, EMPTY, from, lastValueFrom, mergeMap, observeOn, onErrorResumeNextWith, Subject, takeUntil, tap, throttleTime,\n} from \"rxjs\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { MutableRefObject } from \"react\";\nimport type { BeEvent, IDisposable } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeCheckboxProps } from \"./components/TreeNodeCheckbox.js\";\n\n/**\n * Data structure that describes instance visibility status.\n * @beta\n */\nexport interface VisibilityStatus {\n  /** Instance visibility state. */\n  state: \"visible\" | \"partial\" | \"hidden\";\n  /** Specifies whether visibility changing is disabled or not. */\n  isDisabled?: boolean;\n  /** Tooltip that should be displayed when hovering over the visibility checkbox. */\n  tooltip?: string;\n}\n\n/**\n * Handler that can be used to determine and change visibility of instances represented by tree nodes.\n * @beta\n */\nexport interface HierarchyVisibilityHandler extends IDisposable {\n  /** Event used to notify tree about visibility changes from outside. */\n  readonly onVisibilityChange: BeEvent<() => void>;\n  /** Returns current visibility status for tree node. */\n  getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;\n  /** Changes visibility of the instance represented by tree node. */\n  changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): TreeCheckboxProps & { triggerRefresh: () => void } {\n  const visibilityStatusMap = useRef(new Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>());\n  const [state, setState] = useState<TreeCheckboxProps & { triggerRefresh: () => void }>({\n    getCheckboxState: () => ({ state: \"off\", isDisabled: true }),\n    onCheckboxClicked: () => {},\n    triggerRefresh: () => {},\n  });\n  const { onFeatureUsed } = useTelemetryContext();\n\n  useEffect(() => {\n    visibilityStatusMap.current.clear();\n    const handler = visibilityHandlerFactory();\n\n    const visibilityChanged = new Subject<void>();\n    const calculate = new Subject<PresentationHierarchyNode>();\n    const calculateNodeStatus = (node: PresentationHierarchyNode) => {\n      calculate.next(node);\n    };\n\n    const resetCache = () => {\n      visibilityStatusMap.current.forEach((value) => {\n        value.needsRefresh = true;\n      });\n      visibilityChanged.next();\n    };\n\n    const triggerCheckboxUpdate = () => {\n      setState((prev) => ({\n        ...prev,\n        getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      }));\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        observeOn(asyncScheduler),\n        mergeMap((node) =>\n          defer(async () => handler.getVisibilityStatus(node.nodeData)).pipe(\n            tap({\n              next: (status) => {\n                visibilityStatusMap.current.set(node.id, {\n                  node,\n                  status,\n                  needsRefresh: false,\n                });\n              },\n            }),\n            takeUntil(visibilityChanged),\n            onErrorResumeNextWith(EMPTY),\n          ),\n        ),\n        throttleTime(100, undefined, { leading: false, trailing: true }),\n      )\n      .subscribe({\n        next: () => {\n          triggerCheckboxUpdate();\n        },\n      });\n\n    const changeVisibility = (node: PresentationHierarchyNode, checked: boolean) => {\n      onFeatureUsed({ featureId: \"visibility-change\", reportInteraction: true });\n      void handler.changeVisibility(node.nodeData, checked);\n      const entry = visibilityStatusMap.current.get(node.id);\n      if (!entry) {\n        return;\n      }\n      entry.status.state = checked ? \"visible\" : \"hidden\";\n      entry.status.tooltip = undefined;\n      triggerCheckboxUpdate();\n    };\n\n    setState({\n      onCheckboxClicked: changeVisibility,\n      getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      triggerRefresh: () => {\n        resetCache();\n        triggerCheckboxUpdate();\n      },\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      resetCache();\n      triggerCheckboxUpdate();\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler.dispose();\n    };\n  }, [visibilityHandlerFactory, onFeatureUsed]);\n\n  return state;\n}\n\nfunction createStateGetter(\n  map: MutableRefObject<Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>>,\n  calculateVisibility: (node: PresentationHierarchyNode) => void,\n): TreeCheckboxProps[\"getCheckboxState\"] {\n  return (node) => {\n    const entry = map.current.get(node.id);\n    if (entry === undefined) {\n      calculateVisibility(node);\n      return { state: \"off\", isDisabled: true };\n    }\n    if (entry.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    const status = entry.status;\n    return {\n      state: status.state === \"visible\" ? \"on\" : status.state === \"hidden\" ? \"off\" : \"partial\",\n      tooltip: status.tooltip,\n      isDisabled: status.isDisabled,\n    };\n  };\n}\n\n/**\n * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult\n  ? TProps & {\n      /** A callback that produces the value from the original implementation. */\n      readonly originalImplementation: () => TResult;\n      /**\n       * Reference to the hierarchy based handler.\n       * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.\n       */\n      readonly handler: HierarchyVisibilityHandler;\n    }\n  : never;\n\n/**\n * Function type for an overridden method of `HierarchyVisibilityHandler`.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult\n  ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult\n  : never;\n\nexport function createVisibilityHandlerResult<TResult, TOverrideProps>(\n  handler: HierarchyVisibilityHandler,\n  props: TOverrideProps,\n  obs: Observable<TResult>,\n  override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined,\n): Observable<TResult> {\n  return override\n    ? from(\n        override({\n          ...props,\n          originalImplementation: async () => lastValueFrom(obs, { defaultValue: undefined as TResult }),\n          handler,\n        }),\n      )\n    : obs;\n}\n"]}
+{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EACL,cAAc,EACd,KAAK,EACL,QAAQ,EACR,KAAK,EACL,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,SAAS,EACT,qBAAqB,EACrB,OAAO,EACP,SAAS,EACT,GAAG,EACH,YAAY,GACb,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAsC/D,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,mBAAmB,GAAG,MAAM,CAAC,IAAI,GAAG,EAAgG,CAAC,CAAC;IAC5I,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAqD;QACrF,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC5D,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;QAC3B,cAAc,EAAE,GAAG,EAAE,GAAE,CAAC;KACzB,CAAC,CAAC;IACH,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,wBAAwB,EAAE,CAAC;QAE3C,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAAQ,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,OAAO,EAA6B,CAAC;QAC3D,MAAM,mBAAmB,GAAG,CAAC,IAA+B,EAAE,EAAE;YAC9D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,GAAG,EAAE;YACtB,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC5C,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YACH,iBAAiB,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;aAC9E,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,QAAQ,CAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,SAAS,CAAC,cAAc,CAAC,EACzB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAChB,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChE,GAAG,CAAC;YACF,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvC,IAAI;oBACJ,MAAM;oBACN,YAAY,EAAE,KAAK;iBACpB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,EACF,SAAS,CAAC,iBAAiB,CAAC,EAC5B,qBAAqB,CAAC,KAAK,CAAC,CAC7B,CACF,EACD,YAAY,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CACjE;aACA,SAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE;gBACT,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAG,CAAC,IAA+B,EAAE,OAAgB,EAAE,EAAE;YAC7E,aAAa,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,KAAK,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,KAAK,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK,EAAE;gBACV,OAAO;aACR;YACD,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;YACpD,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;YACjC,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,iBAAiB,EAAE,gBAAgB;YACnC,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YAC7E,cAAc,EAAE,GAAG,EAAE;gBACnB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,UAAU,EAAE,CAAC;YACb,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,YAAY,CAAC,WAAW,EAAE,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,EAAE,aAAa,CAAC,CAAC,CAAC;IAE9C,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CACxB,GAAwH,EACxH,mBAA8D;IAE9D,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC3C;QACD,IAAI,KAAK,CAAC,YAAY,EAAE;YACtB,mBAAmB,CAAC,IAAI,CAAC,CAAC;SAC3B;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACxF,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AA0BD,MAAM,UAAU,6BAA6B,CAC3C,OAAmC,EACnC,KAAqB,EACrB,GAAwB,EACxB,QAA8G;IAE9G,OAAO,QAAQ;QACb,CAAC,CAAC,IAAI,CACF,QAAQ,CAAC;YACP,GAAG,KAAK;YACR,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,EAAE,YAAY,EAAE,SAAoB,EAAE,CAAC;YAC9F,OAAO;SACR,CAAC,CACH;QACH,CAAC,CAAC,GAAG,CAAC;AACV,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 { useEffect, useRef, useState } from \"react\";\nimport {\n  asyncScheduler,\n  defer,\n  distinct,\n  EMPTY,\n  from,\n  lastValueFrom,\n  mergeMap,\n  observeOn,\n  onErrorResumeNextWith,\n  Subject,\n  takeUntil,\n  tap,\n  throttleTime,\n} from \"rxjs\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { MutableRefObject } from \"react\";\nimport type { BeEvent, IDisposable } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeCheckboxProps } from \"./components/TreeNodeCheckbox.js\";\n\n/**\n * Data structure that describes instance visibility status.\n * @beta\n */\nexport interface VisibilityStatus {\n  /** Instance visibility state. */\n  state: \"visible\" | \"partial\" | \"hidden\";\n  /** Specifies whether visibility changing is disabled or not. */\n  isDisabled?: boolean;\n  /** Tooltip that should be displayed when hovering over the visibility checkbox. */\n  tooltip?: string;\n}\n\n/**\n * Handler that can be used to determine and change visibility of instances represented by tree nodes.\n * @beta\n */\nexport interface HierarchyVisibilityHandler extends IDisposable {\n  /** Event used to notify tree about visibility changes from outside. */\n  readonly onVisibilityChange: BeEvent<() => void>;\n  /** Returns current visibility status for tree node. */\n  getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;\n  /** Changes visibility of the instance represented by tree node. */\n  changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): TreeCheckboxProps & { triggerRefresh: () => void } {\n  const visibilityStatusMap = useRef(new Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>());\n  const [state, setState] = useState<TreeCheckboxProps & { triggerRefresh: () => void }>({\n    getCheckboxState: () => ({ state: \"off\", isDisabled: true }),\n    onCheckboxClicked: () => {},\n    triggerRefresh: () => {},\n  });\n  const { onFeatureUsed } = useTelemetryContext();\n\n  useEffect(() => {\n    visibilityStatusMap.current.clear();\n    const handler = visibilityHandlerFactory();\n\n    const visibilityChanged = new Subject<void>();\n    const calculate = new Subject<PresentationHierarchyNode>();\n    const calculateNodeStatus = (node: PresentationHierarchyNode) => {\n      calculate.next(node);\n    };\n\n    const resetCache = () => {\n      visibilityStatusMap.current.forEach((value) => {\n        value.needsRefresh = true;\n      });\n      visibilityChanged.next();\n    };\n\n    const triggerCheckboxUpdate = () => {\n      setState((prev) => ({\n        ...prev,\n        getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      }));\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        observeOn(asyncScheduler),\n        mergeMap((node) =>\n          defer(async () => handler.getVisibilityStatus(node.nodeData)).pipe(\n            tap({\n              next: (status) => {\n                visibilityStatusMap.current.set(node.id, {\n                  node,\n                  status,\n                  needsRefresh: false,\n                });\n              },\n            }),\n            takeUntil(visibilityChanged),\n            onErrorResumeNextWith(EMPTY),\n          ),\n        ),\n        throttleTime(100, undefined, { leading: false, trailing: true }),\n      )\n      .subscribe({\n        next: () => {\n          triggerCheckboxUpdate();\n        },\n      });\n\n    const changeVisibility = (node: PresentationHierarchyNode, checked: boolean) => {\n      onFeatureUsed({ featureId: \"visibility-change\", reportInteraction: true });\n      void handler.changeVisibility(node.nodeData, checked);\n      const entry = visibilityStatusMap.current.get(node.id);\n      if (!entry) {\n        return;\n      }\n      entry.status.state = checked ? \"visible\" : \"hidden\";\n      entry.status.tooltip = undefined;\n      triggerCheckboxUpdate();\n    };\n\n    setState({\n      onCheckboxClicked: changeVisibility,\n      getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      triggerRefresh: () => {\n        resetCache();\n        triggerCheckboxUpdate();\n      },\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      resetCache();\n      triggerCheckboxUpdate();\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler.dispose();\n    };\n  }, [visibilityHandlerFactory, onFeatureUsed]);\n\n  return state;\n}\n\nfunction createStateGetter(\n  map: MutableRefObject<Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>>,\n  calculateVisibility: (node: PresentationHierarchyNode) => void,\n): TreeCheckboxProps[\"getCheckboxState\"] {\n  return (node) => {\n    const entry = map.current.get(node.id);\n    if (entry === undefined) {\n      calculateVisibility(node);\n      return { state: \"off\", isDisabled: true };\n    }\n    if (entry.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    const status = entry.status;\n    return {\n      state: status.state === \"visible\" ? \"on\" : status.state === \"hidden\" ? \"off\" : \"partial\",\n      tooltip: status.tooltip,\n      isDisabled: status.isDisabled,\n    };\n  };\n}\n\n/**\n * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult\n  ? TProps & {\n      /** A callback that produces the value from the original implementation. */\n      readonly originalImplementation: () => TResult;\n      /**\n       * Reference to the hierarchy based handler.\n       * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.\n       */\n      readonly handler: HierarchyVisibilityHandler;\n    }\n  : never;\n\n/**\n * Function type for an overridden method of `HierarchyVisibilityHandler`.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult\n  ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult\n  : never;\n\nexport function createVisibilityHandlerResult<TResult, TOverrideProps>(\n  handler: HierarchyVisibilityHandler,\n  props: TOverrideProps,\n  obs: Observable<TResult>,\n  override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined,\n): Observable<TResult> {\n  return override\n    ? from(\n        override({\n          ...props,\n          originalImplementation: async () => lastValueFrom(obs, { defaultValue: undefined as TResult }),\n          handler,\n        }),\n      )\n    : obs;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/ModelsTreeVisibilityHandler.d.ts

@@ -1,3 +1,4 @@
+/// <reference types="node" resolution-mode="require"/>
 import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
 import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
 import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache.js";
@@ -75,7 +76,7 @@ export interface ModelsTreeVisibilityHandlerProps {
  * Creates an instance if `ModelsTreeVisibilityHandler`.
  * @internal
  */
-export declare function createModelsTreeVisibilityHandler(props: ModelsTreeVisibilityHandlerProps): HierarchyVisibilityHandler;
+export declare function createModelsTreeVisibilityHandler(props: ModelsTreeVisibilityHandlerProps): HierarchyVisibilityHandler & Disposable;
 /**
  * Enables display of all given models. Also enables display of all categories and clears always and
  * never drawn lists in the viewport.

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/ModelsTreeVisibilityHandler.js

@@ -66,6 +66,9 @@ class ModelsTreeVisibilityHandlerImpl {
         return toVoidPromise(changeObservable);
     }
     dispose() {
+        this[Symbol.dispose]();
+    }
+    [Symbol.dispose]() {
         this._eventListener.dispose();
         this._alwaysAndNeverDrawnElements.dispose();
         this._subscriptions.forEach((x) => x.unsubscribe());