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

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

@@ -0,0 +1,41 @@
+import type { Id64String } from "@itwin/core-bentley";
+import type { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+interface CategoriesTreeNode {
+    key: HierarchyNodeKey;
+    extendedData?: {
+        [id: string]: any;
+    };
+}
+/** @internal */
+export declare namespace CategoriesTreeNode {
+    /**
+     * Determines if node represents a definition container.
+     */
+    const isDefinitionContainerNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a category.
+     */
+    const isCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a model.
+     */
+    const isModelNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents an element.
+     */
+    const isElementNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a sub-category.
+     */
+    const isSubCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    const getModelId: (node: Pick<CategoriesTreeNode, "extendedData">) => Id64String | undefined;
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    const getCategoryId: (node: Pick<CategoriesTreeNode, "extendedData">) => Id64String | undefined;
+}
+export {};
+//# sourceMappingURL=CategoriesTreeNode.d.ts.map

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

@@ -0,0 +1,50 @@
+/*---------------------------------------------------------------------------------------------
+ * 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 model.
+     */
+    CategoriesTreeNode.isModelNode = (node) => node.extendedData && "isModel" in node.extendedData && !!node.extendedData.isModel;
+    /**
+     * Determines if node represents an element.
+     */
+    CategoriesTreeNode.isElementNode = (node) => node.extendedData && "isElement" in node.extendedData && !!node.extendedData.isElement;
+    /**
+     * Determines if node represents a sub-category.
+     */
+    CategoriesTreeNode.isSubCategoryNode = (node) => node.extendedData && "isSubCategory" in node.extendedData && !!node.extendedData.isSubCategory;
+    /**
+     * Retrieves model ID from node's extended data.
+     */
+    CategoriesTreeNode.getModelId = (node) => {
+        // Elements have modelId property set
+        if (node.extendedData?.modelId) {
+            return node.extendedData?.modelId;
+        }
+        // This is for categories:
+        // a) If category is at root, then it won't have 'modelId' or 'modelIds' set - it will return undefined;
+        // b) If category is under subModel then it will have 'modelIds' set just like in models tree.
+        const modelIds = node.extendedData?.modelIds?.[0];
+        return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);
+    };
+    /**
+     * Retrieves category ID from node's extended data.
+     */
+    CategoriesTreeNode.getCategoryId = (node) => {
+        // Elements have categoryId property set
+        return node.extendedData?.categoryId;
+    };
+})(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;AAUhG,gBAAgB;AAChB,MAAM,KAAW,kBAAkB,CAsDlC;AAtDD,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,8BAAW,GAAG,CAAC,IAA8C,EAAE,EAAE,CAC5E,IAAI,CAAC,YAAY,IAAI,SAAS,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IAErF;;OAEG;IACU,gCAAa,GAAG,CAAC,IAA8C,EAAE,EAAE,CAC9E,IAAI,CAAC,YAAY,IAAI,WAAW,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC;IAEzF;;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;IAEjG;;OAEG;IACU,6BAAU,GAAG,CAAC,IAA8C,EAA0B,EAAE;QACnG,qCAAqC;QACrC,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;QACpC,CAAC;QAED,0BAA0B;QAC1B,wGAAwG;QACxG,8FAA8F;QAC9F,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QAClD,OAAO,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC,CAAC;IAEF;;OAEG;IACU,gCAAa,GAAG,CAAC,IAA8C,EAA0B,EAAE;QACtG,wCAAwC;QACxC,OAAO,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;IACvC,CAAC,CAAC;AACJ,CAAC,EAtDgB,kBAAkB,KAAlB,kBAAkB,QAsDlC","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 { Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\n\ninterface CategoriesTreeNode {\n  key: HierarchyNodeKey;\n  extendedData?: { [id: string]: any };\n}\n\n/** @internal */\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 model.\n   */\n  export const isModelNode = (node: Pick<CategoriesTreeNode, \"extendedData\">) =>\n    node.extendedData && \"isModel\" in node.extendedData && !!node.extendedData.isModel;\n\n  /**\n   * Determines if node represents an element.\n   */\n  export const isElementNode = (node: Pick<CategoriesTreeNode, \"extendedData\">) =>\n    node.extendedData && \"isElement\" in node.extendedData && !!node.extendedData.isElement;\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  /**\n   * Retrieves model ID from node's extended data.\n   */\n  export const getModelId = (node: Pick<CategoriesTreeNode, \"extendedData\">): Id64String | undefined => {\n    // Elements have modelId property set\n    if (node.extendedData?.modelId) {\n      return node.extendedData?.modelId;\n    }\n\n    // This is for categories:\n    // a) If category is at root, then it won't have 'modelId' or 'modelIds' set - it will return undefined;\n    // b) If category is under subModel then it will have 'modelIds' set just like in models tree.\n    const modelIds = node.extendedData?.modelIds?.[0];\n    return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);\n  };\n\n  /**\n   * Retrieves category ID from node's extended data.\n   */\n  export const getCategoryId = (node: Pick<CategoriesTreeNode, \"extendedData\">): Id64String | undefined => {\n    // Elements have categoryId property set\n    return node.extendedData?.categoryId;\n  };\n}\n"]}

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

@@ -0,0 +1,27 @@
+import type { GuidString } from "@itwin/core-bentley";
+import type { CategoryInfo } from "../../common/CategoriesVisibilityUtils.js";
+import type { VisibilityTreeProps } from "../../common/components/VisibilityTree.js";
+import type { ModelId } from "../../common/internal/Types.js";
+import type { CategoriesTreeHierarchyConfiguration } from "../CategoriesTreeDefinition.js";
+import type { CategoriesTreeIdsCache } from "./CategoriesTreeIdsCache.js";
+/** @internal */
+export type CategoriesTreeFilteringError = "tooManyFilterMatches" | "unknownFilterError";
+type HierarchyFilteringPaths = Awaited<ReturnType<Required<VisibilityTreeProps>["getFilteredPaths"]>>;
+/** @internal */
+export declare function useFilteredPaths({ filter, viewType, hierarchyConfiguration, getCategoriesTreeIdsCache, onCategoriesFiltered, onFilteredPathsChanged, componentId, }: {
+    viewType: "2d" | "3d";
+    filter?: string;
+    hierarchyConfiguration: CategoriesTreeHierarchyConfiguration;
+    getCategoriesTreeIdsCache: () => CategoriesTreeIdsCache;
+    onCategoriesFiltered?: (categories: {
+        categories: CategoryInfo[] | undefined;
+        models?: Array<ModelId>;
+    }) => void;
+    onFilteredPathsChanged: (paths: HierarchyFilteringPaths | undefined) => void;
+    componentId: GuidString;
+}): {
+    getPaths: VisibilityTreeProps["getFilteredPaths"] | undefined;
+    filteringError: CategoriesTreeFilteringError | undefined;
+};
+export {};
+//# sourceMappingURL=UseFilteredPaths.d.ts.map

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

@@ -0,0 +1,136 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { useEffect, useMemo, useState } from "react";
+import { firstValueFrom } from "rxjs";
+import { assert } from "@itwin/core-bentley";
+import { HierarchyFilteringPath, HierarchyNodeIdentifier } from "@itwin/presentation-hierarchies";
+import { CLASS_NAME_DefinitionContainer, CLASS_NAME_SubCategory } from "../../common/internal/ClassNameDefinitions.js";
+import { getClassesByView } from "../../common/internal/Utils.js";
+import { FilterLimitExceededError } from "../../common/TreeErrors.js";
+import { useTelemetryContext } from "../../common/UseTelemetryContext.js";
+import { CategoriesTreeDefinition } from "../CategoriesTreeDefinition.js";
+/** @internal */
+export function useFilteredPaths({ filter, viewType, hierarchyConfiguration, getCategoriesTreeIdsCache, onCategoriesFiltered, onFilteredPathsChanged, componentId, }) {
+    const [filteringError, setFilteringError] = useState();
+    const { onFeatureUsed } = useTelemetryContext();
+    useEffect(() => {
+        setFilteringError(undefined);
+        onCategoriesFiltered?.({ categories: undefined, models: undefined });
+        if (!filter) {
+            onFilteredPathsChanged(undefined);
+        }
+    }, [filter, onCategoriesFiltered, onFilteredPathsChanged]);
+    const getFilteredPaths = useMemo(() => {
+        if (!filter) {
+            return undefined;
+        }
+        return async ({ imodelAccess, abortSignal }) => {
+            onFeatureUsed({ featureId: "filtering", reportInteraction: true });
+            try {
+                const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({
+                    imodelAccess,
+                    abortSignal,
+                    label: filter,
+                    viewType,
+                    idsCache: getCategoriesTreeIdsCache(),
+                    hierarchyConfig: hierarchyConfiguration,
+                    componentId,
+                });
+                onFilteredPathsChanged(paths);
+                const { elementClass, modelClass } = getClassesByView(viewType);
+                onCategoriesFiltered?.(await getCategoriesFromPaths(paths, getCategoriesTreeIdsCache(), elementClass, modelClass, hierarchyConfiguration));
+                return paths;
+            }
+            catch (e) {
+                const newError = e instanceof FilterLimitExceededError ? "tooManyFilterMatches" : "unknownFilterError";
+                if (newError !== "tooManyFilterMatches") {
+                    const feature = e instanceof Error && e.message.includes("query too long to execute or server is too busy") ? "error-timeout" : "error-unknown";
+                    onFeatureUsed({ featureId: feature, reportInteraction: false });
+                }
+                setFilteringError(newError);
+                return [];
+            }
+        };
+    }, [onCategoriesFiltered, filter, onFilteredPathsChanged, onFeatureUsed, viewType, getCategoriesTreeIdsCache, hierarchyConfiguration, componentId]);
+    return {
+        getPaths: getFilteredPaths,
+        filteringError,
+    };
+}
+async function getCategoriesFromPaths(paths, idsCache, elementClassName, modelsClassName, hierarchyConfig) {
+    if (!paths) {
+        return { categories: undefined };
+    }
+    const rootFilteredElementIds = new Set();
+    const subModelIds = new Set();
+    const categories = new Map();
+    for (const path of paths) {
+        const currPath = HierarchyFilteringPath.normalize(path).path;
+        if (currPath.length === 0) {
+            continue;
+        }
+        let category;
+        let subCategory;
+        let lastNodeInfo;
+        for (let i = 0; i < currPath.length; ++i) {
+            const currentNode = currPath[i];
+            if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currentNode)) {
+                continue;
+            }
+            if (currentNode.className === elementClassName) {
+                rootFilteredElementIds.add(currentNode.id);
+                for (let j = i + 1; j < currPath.length; ++j) {
+                    const childNode = currPath[j];
+                    if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(childNode)) {
+                        continue;
+                    }
+                    if (childNode.className === modelsClassName) {
+                        subModelIds.add(childNode.id);
+                    }
+                }
+                break;
+            }
+            lastNodeInfo = { lastNode: currentNode, nodeIndex: i };
+        }
+        assert(lastNodeInfo !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(lastNodeInfo.lastNode));
+        if (lastNodeInfo.lastNode.className === CLASS_NAME_DefinitionContainer) {
+            const definitionContainerCategories = await firstValueFrom(idsCache.getAllContainedCategories({ definitionContainerIds: lastNodeInfo.lastNode.id, includeEmptyCategories: hierarchyConfig.showEmptyCategories }));
+            for (const categoryId of definitionContainerCategories) {
+                const value = categories.get(categoryId);
+                if (value === undefined) {
+                    categories.set(categoryId, []);
+                }
+            }
+            continue;
+        }
+        if (lastNodeInfo.lastNode.className === CLASS_NAME_SubCategory) {
+            const secondToLastNode = lastNodeInfo.nodeIndex > 0 ? currPath[lastNodeInfo.nodeIndex - 1] : undefined;
+            assert(secondToLastNode !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(secondToLastNode));
+            subCategory = lastNodeInfo.lastNode;
+            category = secondToLastNode;
+        }
+        else {
+            category = lastNodeInfo.lastNode;
+        }
+        let entry = categories.get(category.id);
+        if (entry === undefined) {
+            entry = [];
+            categories.set(category.id, entry);
+        }
+        if (subCategory) {
+            entry.push(subCategory.id);
+        }
+    }
+    const rootElementModelMap = await firstValueFrom(idsCache.getFilteredElementsModels(rootFilteredElementIds));
+    const models = [...subModelIds, ...new Set(rootElementModelMap.values())];
+    return {
+        categories: [...categories.entries()].map(([categoryId, subCategoryIds]) => ({
+            categoryId,
+            subCategoryIds: subCategoryIds.length === 0 ? undefined : subCategoryIds,
+        })),
+        models,
+    };
+}
+//# sourceMappingURL=UseFilteredPaths.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"UseFilteredPaths.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/UseFilteredPaths.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAClG,OAAO,EAAE,8BAA8B,EAAE,sBAAsB,EAAE,MAAM,+CAA+C,CAAC;AACvH,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAc1E,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,EAC/B,MAAM,EACN,QAAQ,EACR,sBAAsB,EACtB,yBAAyB,EACzB,oBAAoB,EACpB,sBAAsB,EACtB,WAAW,GASZ;IAIC,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,EAA4C,CAAC;IACjG,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,SAAS,CAAC,GAAG,EAAE;QACb,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,oBAAoB,EAAE,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,sBAAsB,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,EAAE,CAAC,MAAM,EAAE,oBAAoB,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAE3D,MAAM,gBAAgB,GAAG,OAAO,CAAsD,GAAG,EAAE;QACzF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;YAC7C,aAAa,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC;oBAClE,YAAY;oBACZ,WAAW;oBACX,KAAK,EAAE,MAAM;oBACb,QAAQ;oBACR,QAAQ,EAAE,yBAAyB,EAAE;oBACrC,eAAe,EAAE,sBAAsB;oBACvC,WAAW;iBACZ,CAAC,CAAC;gBACH,sBAAsB,CAAC,KAAK,CAAC,CAAC;gBAC9B,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAChE,oBAAoB,EAAE,CAAC,MAAM,sBAAsB,CAAC,KAAK,EAAE,yBAAyB,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,sBAAsB,CAAC,CAAC,CAAC;gBAC3I,OAAO,KAAK,CAAC;YACf,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;gBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;oBACxC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;oBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;gBAClE,CAAC;gBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,oBAAoB,EAAE,MAAM,EAAE,sBAAsB,EAAE,aAAa,EAAE,QAAQ,EAAE,yBAAyB,EAAE,sBAAsB,EAAE,WAAW,CAAC,CAAC,CAAC;IAEpJ,OAAO;QACL,QAAQ,EAAE,gBAAgB;QAC1B,cAAc;KACf,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,KAA8B,EAC9B,QAAgC,EAChC,gBAAwB,EACxB,eAAuB,EACvB,eAAqD;IAErD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;IACnC,CAAC;IAED,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAa,CAAC;IACpD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAW,CAAC;IAEvC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAoC,CAAC;IAC/D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QAC7D,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,SAAS;QACX,CAAC;QAED,IAAI,QAAiC,CAAC;QACtC,IAAI,WAAgD,CAAC;QAErD,IAAI,YAAkF,CAAC;QAEvF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YACzC,MAAM,WAAW,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACnE,SAAS;YACX,CAAC;YACD,IAAI,WAAW,CAAC,SAAS,KAAK,gBAAgB,EAAE,CAAC;gBAC/C,sBAAsB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;gBAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;oBAC7C,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;oBAC9B,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,SAAS,CAAC,EAAE,CAAC;wBACjE,SAAS;oBACX,CAAC;oBACD,IAAI,SAAS,CAAC,SAAS,KAAK,eAAe,EAAE,CAAC;wBAC5C,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;YACD,YAAY,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;QACzD,CAAC;QAED,MAAM,CAAC,YAAY,KAAK,SAAS,IAAI,uBAAuB,CAAC,wBAAwB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE9G,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,KAAK,8BAA8B,EAAE,CAAC;YACvE,MAAM,6BAA6B,GAAG,MAAM,cAAc,CACxD,QAAQ,CAAC,yBAAyB,CAAC,EAAE,sBAAsB,EAAE,YAAY,CAAC,QAAQ,CAAC,EAAE,EAAE,sBAAsB,EAAE,eAAe,CAAC,mBAAmB,EAAE,CAAC,CACtJ,CAAC;YACF,KAAK,MAAM,UAAU,IAAI,6BAA6B,EAAE,CAAC;gBACvD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACzC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,KAAK,sBAAsB,EAAE,CAAC;YAC/D,MAAM,gBAAgB,GAAG,YAAY,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACvG,MAAM,CAAC,gBAAgB,KAAK,SAAS,IAAI,uBAAuB,CAAC,wBAAwB,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAE7G,WAAW,GAAG,YAAY,CAAC,QAAQ,CAAC;YACpC,QAAQ,GAAG,gBAAgB,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;QACnC,CAAC;QAED,IAAI,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,KAAK,GAAG,EAAE,CAAC;YACX,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,WAAW,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,MAAM,mBAAmB,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,yBAAyB,CAAC,sBAAsB,CAAC,CAAC,CAAC;IAC7G,MAAM,MAAM,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,IAAI,GAAG,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC1E,OAAO;QACL,UAAU,EAAE,CAAC,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3E,UAAU;YACV,cAAc,EAAE,cAAc,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;SACzE,CAAC,CAAC;QACH,MAAM;KACP,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 { useEffect, useMemo, useState } from \"react\";\nimport { firstValueFrom } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { HierarchyFilteringPath, HierarchyNodeIdentifier } from \"@itwin/presentation-hierarchies\";\nimport { CLASS_NAME_DefinitionContainer, CLASS_NAME_SubCategory } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { getClassesByView } from \"../../common/internal/Utils.js\";\nimport { FilterLimitExceededError } from \"../../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../../common/UseTelemetryContext.js\";\nimport { CategoriesTreeDefinition } from \"../CategoriesTreeDefinition.js\";\n\nimport type { GuidString } from \"@itwin/core-bentley\";\nimport type { CategoryInfo } from \"../../common/CategoriesVisibilityUtils.js\";\nimport type { VisibilityTreeProps } from \"../../common/components/VisibilityTree.js\";\nimport type { CategoryId, ElementId, ModelId, SubCategoryId } from \"../../common/internal/Types.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"../CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeIdsCache } from \"./CategoriesTreeIdsCache.js\";\n\n/** @internal */\nexport type CategoriesTreeFilteringError = \"tooManyFilterMatches\" | \"unknownFilterError\";\n\ntype HierarchyFilteringPaths = Awaited<ReturnType<Required<VisibilityTreeProps>[\"getFilteredPaths\"]>>;\n\n/** @internal */\nexport function useFilteredPaths({\n  filter,\n  viewType,\n  hierarchyConfiguration,\n  getCategoriesTreeIdsCache,\n  onCategoriesFiltered,\n  onFilteredPathsChanged,\n  componentId,\n}: {\n  viewType: \"2d\" | \"3d\";\n  filter?: string;\n  hierarchyConfiguration: CategoriesTreeHierarchyConfiguration;\n  getCategoriesTreeIdsCache: () => CategoriesTreeIdsCache;\n  onCategoriesFiltered?: (categories: { categories: CategoryInfo[] | undefined; models?: Array<ModelId> }) => void;\n  onFilteredPathsChanged: (paths: HierarchyFilteringPaths | undefined) => void;\n  componentId: GuidString;\n}): {\n  getPaths: VisibilityTreeProps[\"getFilteredPaths\"] | undefined;\n  filteringError: CategoriesTreeFilteringError | undefined;\n} {\n  const [filteringError, setFilteringError] = useState<CategoriesTreeFilteringError | undefined>();\n  const { onFeatureUsed } = useTelemetryContext();\n\n  useEffect(() => {\n    setFilteringError(undefined);\n    onCategoriesFiltered?.({ categories: undefined, models: undefined });\n    if (!filter) {\n      onFilteredPathsChanged(undefined);\n    }\n  }, [filter, onCategoriesFiltered, onFilteredPathsChanged]);\n\n  const getFilteredPaths = useMemo<VisibilityTreeProps[\"getFilteredPaths\"] | undefined>(() => {\n    if (!filter) {\n      return undefined;\n    }\n\n    return async ({ imodelAccess, abortSignal }) => {\n      onFeatureUsed({ featureId: \"filtering\", reportInteraction: true });\n      try {\n        const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({\n          imodelAccess,\n          abortSignal,\n          label: filter,\n          viewType,\n          idsCache: getCategoriesTreeIdsCache(),\n          hierarchyConfig: hierarchyConfiguration,\n          componentId,\n        });\n        onFilteredPathsChanged(paths);\n        const { elementClass, modelClass } = getClassesByView(viewType);\n        onCategoriesFiltered?.(await getCategoriesFromPaths(paths, getCategoriesTreeIdsCache(), elementClass, modelClass, hierarchyConfiguration));\n        return paths;\n      } catch (e) {\n        const newError = e instanceof FilterLimitExceededError ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n        if (newError !== \"tooManyFilterMatches\") {\n          const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n          onFeatureUsed({ featureId: feature, reportInteraction: false });\n        }\n        setFilteringError(newError);\n        return [];\n      }\n    };\n  }, [onCategoriesFiltered, filter, onFilteredPathsChanged, onFeatureUsed, viewType, getCategoriesTreeIdsCache, hierarchyConfiguration, componentId]);\n\n  return {\n    getPaths: getFilteredPaths,\n    filteringError,\n  };\n}\n\nasync function getCategoriesFromPaths(\n  paths: HierarchyFilteringPaths,\n  idsCache: CategoriesTreeIdsCache,\n  elementClassName: string,\n  modelsClassName: string,\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration,\n): Promise<{ categories: CategoryInfo[] | undefined; models?: Array<ModelId> }> {\n  if (!paths) {\n    return { categories: undefined };\n  }\n\n  const rootFilteredElementIds = new Set<ElementId>();\n  const subModelIds = new Set<ModelId>();\n\n  const categories = new Map<CategoryId, Array<SubCategoryId>>();\n  for (const path of paths) {\n    const currPath = HierarchyFilteringPath.normalize(path).path;\n    if (currPath.length === 0) {\n      continue;\n    }\n\n    let category: HierarchyNodeIdentifier;\n    let subCategory: HierarchyNodeIdentifier | undefined;\n\n    let lastNodeInfo: { lastNode: HierarchyNodeIdentifier; nodeIndex: number } | undefined;\n\n    for (let i = 0; i < currPath.length; ++i) {\n      const currentNode = currPath[i];\n      if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currentNode)) {\n        continue;\n      }\n      if (currentNode.className === elementClassName) {\n        rootFilteredElementIds.add(currentNode.id);\n        for (let j = i + 1; j < currPath.length; ++j) {\n          const childNode = currPath[j];\n          if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(childNode)) {\n            continue;\n          }\n          if (childNode.className === modelsClassName) {\n            subModelIds.add(childNode.id);\n          }\n        }\n        break;\n      }\n      lastNodeInfo = { lastNode: currentNode, nodeIndex: i };\n    }\n\n    assert(lastNodeInfo !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(lastNodeInfo.lastNode));\n\n    if (lastNodeInfo.lastNode.className === CLASS_NAME_DefinitionContainer) {\n      const definitionContainerCategories = await firstValueFrom(\n        idsCache.getAllContainedCategories({ definitionContainerIds: lastNodeInfo.lastNode.id, includeEmptyCategories: hierarchyConfig.showEmptyCategories }),\n      );\n      for (const categoryId of definitionContainerCategories) {\n        const value = categories.get(categoryId);\n        if (value === undefined) {\n          categories.set(categoryId, []);\n        }\n      }\n      continue;\n    }\n\n    if (lastNodeInfo.lastNode.className === CLASS_NAME_SubCategory) {\n      const secondToLastNode = lastNodeInfo.nodeIndex > 0 ? currPath[lastNodeInfo.nodeIndex - 1] : undefined;\n      assert(secondToLastNode !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(secondToLastNode));\n\n      subCategory = lastNodeInfo.lastNode;\n      category = secondToLastNode;\n    } else {\n      category = lastNodeInfo.lastNode;\n    }\n\n    let entry = categories.get(category.id);\n    if (entry === undefined) {\n      entry = [];\n      categories.set(category.id, entry);\n    }\n\n    if (subCategory) {\n      entry.push(subCategory.id);\n    }\n  }\n  const rootElementModelMap = await firstValueFrom(idsCache.getFilteredElementsModels(rootFilteredElementIds));\n  const models = [...subModelIds, ...new Set(rootElementModelMap.values())];\n  return {\n    categories: [...categories.entries()].map(([categoryId, subCategoryIds]) => ({\n      categoryId,\n      subCategoryIds: subCategoryIds.length === 0 ? undefined : subCategoryIds,\n    })),\n    models,\n  };\n}\n"]}

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

@@ -0,0 +1,54 @@
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
+import type { Observable } from "rxjs";
+import type { HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { AlwaysAndNeverDrawnElementInfo } from "../../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import type { TreeSpecificVisibilityHandler } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { TreeWidgetViewport } from "../../../common/TreeWidgetViewport.js";
+import type { VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
+import type { CategoriesTreeHierarchyConfiguration } from "../../CategoriesTreeDefinition.js";
+import type { CategoriesTreeIdsCache } from "../CategoriesTreeIdsCache.js";
+import type { CategoriesTreeFilterTargets } from "./FilteredTree.js";
+/** @internal */
+export interface CategoriesTreeVisibilityHandlerProps {
+    idsCache: CategoriesTreeIdsCache;
+    viewport: TreeWidgetViewport;
+    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;
+    hierarchyConfig: CategoriesTreeHierarchyConfiguration;
+}
+/**
+ * Handles visibility status of categories tree nodes.
+ *
+ * This handler knows how to get and change visibility status of nodes created by hierarchy definition.
+ * @internal
+ */
+export declare class CategoriesTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<CategoriesTreeFilterTargets> {
+    #private;
+    constructor(constructorProps: CategoriesTreeVisibilityHandlerProps);
+    [Symbol.dispose](): void;
+    changeFilterTargetsVisibilityStatus(targets: CategoriesTreeFilterTargets, on: boolean): Observable<void>;
+    getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus>;
+    /** Changes visibility of the items represented by the tree node. */
+    changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void>;
+    getFilterTargetsVisibilityStatus(targets: CategoriesTreeFilterTargets): Observable<VisibilityStatus>;
+    private getCategories;
+    private getAllCategories;
+    private getElementsCount;
+    private getModels;
+    private getSubCategories;
+    private getSubModels;
+    private getGroupingNodeInfo;
+}
+/**
+ * Creates categories tree visibility handler. Is used by integration and performance tests.
+ * @internal
+ */
+export declare function createCategoriesTreeVisibilityHandler(props: {
+    viewport: TreeWidgetViewport;
+    idsCache: CategoriesTreeIdsCache;
+    imodelAccess: ECClassHierarchyInspector;
+    filteredPaths?: HierarchyFilteringPath[];
+    hierarchyConfig: CategoriesTreeHierarchyConfiguration;
+}): HierarchyVisibilityHandlerImpl<CategoriesTreeFilterTargets>;
+//# sourceMappingURL=CategoriesTreeVisibilityHandler.d.ts.map

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

@@ -0,0 +1,291 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { concat, defer, EMPTY, from, map, merge, mergeMap, of } from "rxjs";
+import { assert, Id64 } from "@itwin/core-bentley";
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
+import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
+import { getClassesByView, releaseMainThreadOnItemsCount } from "../../../common/internal/Utils.js";
+import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
+import { CategoriesTreeNode } from "../CategoriesTreeNode.js";
+import { CategoriesTreeVisibilityHelper } from "./CategoriesTreeVisibilityHelper.js";
+import { createFilteredCategoriesTree } from "./FilteredTree.js";
+/**
+ * Handles visibility status of categories tree nodes.
+ *
+ * This handler knows how to get and change visibility status of nodes created by hierarchy definition.
+ * @internal
+ */
+export class CategoriesTreeVisibilityHandler {
+    #props;
+    #visibilityHelper;
+    #elementType;
+    #categoryType;
+    #modelType;
+    constructor(constructorProps) {
+        this.#props = constructorProps;
+        // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.
+        // We won't need to create a custom base ids cache.
+        const baseIdsCache = {
+            getCategories: (props) => this.getCategories(props),
+            getAllCategories: () => this.getAllCategories(),
+            getElementsCount: (props) => this.getElementsCount(props),
+            getModels: (props) => this.getModels(props),
+            getSubCategories: (props) => this.getSubCategories(props),
+            getSubModels: (props) => this.getSubModels(props),
+            hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),
+        };
+        this.#visibilityHelper = new CategoriesTreeVisibilityHelper({
+            viewport: this.#props.viewport,
+            idsCache: this.#props.idsCache,
+            alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
+            baseIdsCache,
+            hierarchyConfig: constructorProps.hierarchyConfig,
+        });
+        this.#elementType = this.#props.viewport.viewType === "2d" ? "GeometricElement2d" : "GeometricElement3d";
+        this.#categoryType = this.#props.viewport.viewType === "2d" ? "DrawingCategory" : "SpatialCategory";
+        this.#modelType = this.#props.viewport.viewType === "2d" ? "GeometricModel2d" : "GeometricModel3d";
+    }
+    [Symbol.dispose]() {
+        this.#visibilityHelper[Symbol.dispose]();
+    }
+    changeFilterTargetsVisibilityStatus(targets, on) {
+        return defer(() => {
+            const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;
+            const observables = new Array();
+            if (definitionContainerIds?.size) {
+                observables.push(this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({ definitionContainerIds, on }));
+            }
+            if (modelIds?.size) {
+                observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));
+            }
+            if (categories?.length) {
+                observables.push(from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.changeCategoriesVisibilityStatus({ categoryIds, modelId, on }))));
+            }
+            if (subCategories?.length) {
+                observables.push(from(subCategories).pipe(mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.changeSubCategoriesVisibilityStatus({ subCategoryIds, categoryId, on }))));
+            }
+            if (elements?.length) {
+                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }))));
+            }
+            return merge(...observables);
+        });
+    }
+    getVisibilityStatus(node) {
+        if (HierarchyNode.isClassGroupingNode(node)) {
+            const nodeInfo = this.getGroupingNodeInfo(node);
+            return this.#visibilityHelper.getGroupedElementsVisibilityStatus({ categoryId: nodeInfo.categoryId, modelElementsMap: nodeInfo.modelElementsMap });
+        }
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            return this.#visibilityHelper.getDefinitionContainersVisibilityStatus({
+                definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        if (CategoriesTreeNode.isModelNode(node)) {
+            return this.#visibilityHelper.getModelsVisibilityStatus({
+                modelIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                type: this.#modelType,
+            });
+        }
+        if (CategoriesTreeNode.isCategoryNode(node)) {
+            return this.#visibilityHelper.getCategoriesVisibilityStatus({
+                categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                modelId: CategoriesTreeNode.getModelId(node),
+                type: this.#categoryType,
+            });
+        }
+        const categoryId = CategoriesTreeNode.getCategoryId(node);
+        if (!categoryId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            return this.#visibilityHelper.getSubCategoriesVisibilityStatus({
+                categoryId,
+                subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        const modelId = CategoriesTreeNode.getModelId(node);
+        if (!modelId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        return this.#visibilityHelper.getElementsVisibilityStatus({
+            elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            modelId,
+            categoryId,
+            type: this.#elementType,
+        });
+    }
+    /** Changes visibility of the items represented by the tree node. */
+    changeVisibilityStatus(node, on) {
+        const changeObs = defer(() => {
+            if (HierarchyNode.isClassGroupingNode(node)) {
+                const nodeInfo = this.getGroupingNodeInfo(node);
+                return this.#visibilityHelper.changeGroupedElementsVisibilityStatus({
+                    categoryId: nodeInfo.categoryId,
+                    modelElementsMap: nodeInfo.modelElementsMap,
+                    on,
+                });
+            }
+            if (!HierarchyNode.isInstancesNode(node)) {
+                return EMPTY;
+            }
+            if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+                return this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({
+                    definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                    on,
+                });
+            }
+            if (CategoriesTreeNode.isModelNode(node)) {
+                return this.#visibilityHelper.changeModelsVisibilityStatus({
+                    modelIds: node.key.instanceKeys.map(({ id }) => id),
+                    on,
+                });
+            }
+            if (CategoriesTreeNode.isCategoryNode(node)) {
+                return this.#visibilityHelper.changeCategoriesVisibilityStatus({
+                    categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                    modelId: CategoriesTreeNode.getModelId(node),
+                    on,
+                });
+            }
+            const categoryId = CategoriesTreeNode.getCategoryId(node);
+            if (!categoryId) {
+                return EMPTY;
+            }
+            if (CategoriesTreeNode.isSubCategoryNode(node)) {
+                return this.#visibilityHelper.changeSubCategoriesVisibilityStatus({
+                    categoryId,
+                    subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                    on,
+                });
+            }
+            const modelId = CategoriesTreeNode.getModelId(node);
+            if (!modelId) {
+                return EMPTY;
+            }
+            return this.#visibilityHelper.changeElementsVisibilityStatus({
+                elementIds: node.key.instanceKeys.map(({ id }) => id),
+                modelId,
+                categoryId,
+                on,
+            });
+        });
+        if (this.#props.viewport.isAlwaysDrawnExclusive) {
+            return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);
+        }
+        return changeObs;
+    }
+    getFilterTargetsVisibilityStatus(targets) {
+        return defer(() => {
+            const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;
+            const observables = new Array();
+            if (definitionContainerIds?.size) {
+                observables.push(this.#visibilityHelper.getDefinitionContainersVisibilityStatus({ definitionContainerIds }));
+            }
+            if (modelIds?.size) {
+                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: this.#modelType }));
+            }
+            if (categories?.length) {
+                observables.push(from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.getCategoriesVisibilityStatus({
+                    categoryIds,
+                    modelId,
+                    type: this.#categoryType,
+                }))));
+            }
+            if (subCategories?.length) {
+                observables.push(from(subCategories).pipe(mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.getSubCategoriesVisibilityStatus({ subCategoryIds, categoryId }))));
+            }
+            if (elements?.length) {
+                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: this.#elementType }))));
+            }
+            return merge(...observables);
+        }).pipe(mergeVisibilityStatuses);
+    }
+    getCategories(props) {
+        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categories) => ({
+            id: modelId,
+            ...(this.#props.viewport.viewType === "2d" ? { drawingCategories: categories } : { spatialCategories: categories }),
+        })))));
+    }
+    getAllCategories() {
+        return this.#props.idsCache.getAllCategories().pipe(map((categories) => {
+            if (this.#props.viewport.viewType === "2d") {
+                return { drawingCategories: categories };
+            }
+            return { spatialCategories: categories };
+        }));
+    }
+    getElementsCount(props) {
+        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
+    }
+    getModels(props) {
+        return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesElementModels(categoryId, true).pipe(mergeMap((categoryModelsMap) => (categoryModelsMap.size > 0 ? categoryModelsMap.values() : of(undefined))), map((categoryModels) => ({ id: categoryId, models: categoryModels })))));
+    }
+    getSubCategories(props) {
+        return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getSubCategories(categoryId).pipe(mergeMap((categorySubCategoriesMap) => (categorySubCategoriesMap.size > 0 ? categorySubCategoriesMap.values() : of(undefined))), map((subCategories) => ({ id: categoryId, subCategories })))));
+    }
+    getSubModels(props) {
+        if ("modelIds" in props) {
+            return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => {
+                if (props.categoryId) {
+                    return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));
+                }
+                return this.#props.idsCache.getModelCategoryIds(modelId).pipe(mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)), map((subModels) => ({ id: modelId, subModels })));
+            }));
+        }
+        if (props.modelId) {
+            return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesModeledElements(props.modelId, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels })))));
+        }
+        return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesElementModels(categoryId).pipe(mergeMap((categoryModelsMap) => {
+            const models = categoryModelsMap.get(categoryId);
+            if (!models) {
+                return of({ id: categoryId, subModels: undefined });
+            }
+            return from(models).pipe(mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryId)), map((subModels) => ({ id: categoryId, subModels })));
+        }))));
+    }
+    getGroupingNodeInfo(node) {
+        const modelElementsMap = node.extendedData?.modelElementsMap;
+        const categoryId = node.extendedData?.categoryId;
+        assert(!!modelElementsMap && !!categoryId);
+        return { modelElementsMap, categoryId };
+    }
+}
+/**
+ * Creates categories tree visibility handler. Is used by integration and performance tests.
+ * @internal
+ */
+export function createCategoriesTreeVisibilityHandler(props) {
+    return new HierarchyVisibilityHandlerImpl({
+        getFilteredTree: () => {
+            if (!props.filteredPaths) {
+                return undefined;
+            }
+            const { categoryClass, elementClass, modelClass } = getClassesByView(props.viewport.viewType === "2d" ? "2d" : "3d");
+            return createFilteredCategoriesTree({
+                idsCache: props.idsCache,
+                filteringPaths: props.filteredPaths,
+                imodelAccess: props.imodelAccess,
+                categoryClassName: categoryClass,
+                categoryElementClassName: elementClass,
+                categoryModelClassName: modelClass,
+            });
+        },
+        getTreeSpecificVisibilityHandler: (info) => {
+            return new CategoriesTreeVisibilityHandler({
+                alwaysAndNeverDrawnElementInfo: info,
+                idsCache: props.idsCache,
+                viewport: props.viewport,
+                hierarchyConfig: props.hierarchyConfig,
+            });
+        },
+        viewport: props.viewport,
+        componentId: "Test",
+    });
+}
+//# sourceMappingURL=CategoriesTreeVisibilityHandler.js.map