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

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

@@ -0,0 +1,148 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { assert } from "@itwin/core-bentley";
+import { CLASS_NAME_Category, CLASS_NAME_Model, CLASS_NAME_Subject } from "../../../common/internal/ClassNameDefinitions.js";
+import { createFilteredTree, FilteredNodesHandler } from "../../../common/internal/visibility/BaseFilteredTree.js";
+/** @internal */
+export async function createFilteredModelsTree(props) {
+    const { imodelAccess, filteringPaths } = props;
+    return createFilteredTree({
+        filteredNodesHandler: new ModelsTreeFilteredNodesHandler({ imodelAccess }),
+        filteringPaths,
+    });
+}
+class ModelsTreeFilteredNodesHandler extends FilteredNodesHandler {
+    #props;
+    constructor(props) {
+        super();
+        this.#props = props;
+    }
+    convertNodesToFilterTargets(filteredNodes) {
+        const filterTargets = {};
+        filteredNodes.forEach((filteredNode) => this.collectFilterTargets(filterTargets, filteredNode));
+        return this.convertInternalFilterTargets(filterTargets);
+    }
+    async getProcessedFilteredNodes() { }
+    convertInternalFilterTargets(filterTargets) {
+        if (!filterTargets.categories && !filterTargets.subjectIds && !filterTargets.elements && !filterTargets.modelIds) {
+            return undefined;
+        }
+        return {
+            categories: filterTargets.categories
+                ? [...filterTargets.categories.entries()].map(([modelId, categoryIds]) => {
+                    return { modelId, categoryIds };
+                })
+                : undefined,
+            elements: filterTargets.elements
+                ? [...filterTargets.elements.entries()].map(([modelCategoryKey, elements]) => {
+                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
+                    return { modelId, categoryId, elements };
+                })
+                : undefined,
+            modelIds: filterTargets.modelIds,
+            subjectIds: filterTargets.subjectIds,
+        };
+    }
+    collectFilterTargets(changeTargets, filteredNode) {
+        if (filteredNode.isFilterTarget) {
+            this.addTarget(changeTargets, filteredNode);
+            return;
+        }
+        if (filteredNode.type === "element") {
+            // need to add parent ids as filter target will be an element
+            this.addTarget(changeTargets, filteredNode);
+        }
+        if (!filteredNode.children) {
+            return;
+        }
+        for (const child of filteredNode.children.values()) {
+            this.collectFilterTargets(changeTargets, child);
+        }
+    }
+    addTarget(filterTargets, node) {
+        switch (node.type) {
+            case "subject":
+                (filterTargets.subjectIds ??= new Set()).add(node.id);
+                return;
+            case "model":
+                (filterTargets.modelIds ??= new Set()).add(node.id);
+                return;
+            case "category":
+                const categories = (filterTargets.categories ??= new Map()).get(node.modelId);
+                if (categories) {
+                    categories.add(node.id);
+                    return;
+                }
+                filterTargets.categories.set(node.modelId, new Set([node.id]));
+                return;
+            case "element":
+                const modelCategoryKey = this.createModelCategoryKey(node.modelId, node.categoryId);
+                const elements = (filterTargets.elements ??= new Map()).get(modelCategoryKey);
+                if (elements) {
+                    elements.set(node.id, { isFilterTarget: node.isFilterTarget });
+                }
+                else {
+                    filterTargets.elements.set(modelCategoryKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));
+                }
+        }
+    }
+    createModelCategoryKey(modelId, categoryId) {
+        return `${modelId}-${categoryId}`;
+    }
+    parseModelCategoryKey(key) {
+        const [modelId, categoryId] = key.split("-");
+        return { modelId, categoryId };
+    }
+    createFilteredTreeNode({ type, id, isFilterTarget, parent, }) {
+        if (type === "subject" || type === "model") {
+            return {
+                id,
+                isFilterTarget,
+                type,
+            };
+        }
+        if (type === "category") {
+            assert("type" in parent && parent.type === "model");
+            return {
+                id,
+                isFilterTarget,
+                type,
+                modelId: parent.id,
+            };
+        }
+        if ("type" in parent && parent.type === "category") {
+            return {
+                id,
+                isFilterTarget,
+                type,
+                modelId: parent.modelId,
+                categoryId: parent.id,
+            };
+        }
+        if ("type" in parent && parent.type === "element") {
+            return {
+                id,
+                isFilterTarget,
+                type,
+                modelId: parent.modelId,
+                categoryId: parent.categoryId,
+            };
+        }
+        throw new Error("Invalid parent node type");
+    }
+    async getType(className) {
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Subject)) {
+            return "subject";
+        }
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Model)) {
+            return "model";
+        }
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Category)) {
+            return "category";
+        }
+        return "element";
+    }
+}
+//# sourceMappingURL=FilteredTree.js.map

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/visibility/FilteredTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FilteredTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/visibility/FilteredTree.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,kDAAkD,CAAC;AAC7H,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,yDAAyD,CAAC;AAiCnH,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,KAG9C;IACC,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC;IAC/C,OAAO,kBAAkB,CAAC;QACxB,oBAAoB,EAAE,IAAI,8BAA8B,CAAC,EAAE,YAAY,EAAE,CAAC;QAC1E,cAAc;KACf,CAAC,CAAC;AACL,CAAC;AAeD,MAAM,8BAA+B,SAAQ,oBAAqE;IACvG,MAAM,CAAsC;IACrD,YAAY,KAA0C;QACpD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAEM,2BAA2B,CAAC,aAAiC;QAClE,MAAM,aAAa,GAA0B,EAAE,CAAC;QAEhD,aAAa,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC;QAEhG,OAAO,IAAI,CAAC,4BAA4B,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC;IAEM,KAAK,CAAC,yBAAyB,KAAmB,CAAC;IAElD,4BAA4B,CAAC,aAAoC;QACvE,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,CAAC,aAAa,CAAC,QAAQ,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;YACjH,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE;oBACrE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;gBAClC,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa,CAAC,QAAQ;gBAC9B,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBACzE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAC3C,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,aAAa,CAAC,QAAQ;YAChC,UAAU,EAAE,aAAa,CAAC,UAAU;SACrC,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,aAAoC,EAAE,YAA8B;QAC/F,IAAI,YAAY,CAAC,cAAc,EAAE,CAAC;YAChC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,YAAY,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACpC,6DAA6D;YAC7D,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YACnD,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,aAAoC,EAAE,IAAsB;QAC5E,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,SAAS;gBACZ,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACtD,OAAO;YACT,KAAK,OAAO;gBACV,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACpD,OAAO;YACT,KAAK,UAAU;gBACb,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9E,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACxB,OAAO;gBACT,CAAC;gBACD,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC/D,OAAO;YACT,KAAK,SAAS;gBACZ,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBACpF,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9E,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACjE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9G,CAAC;QACL,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,OAAmB,EAAE,UAAsB;QACxE,OAAO,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;IACpC,CAAC;IAEO,qBAAqB,CAAC,GAAqB;QACjD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IACjC,CAAC;IAEM,sBAAsB,CAAC,EAC5B,IAAI,EACJ,EAAE,EACF,cAAc,EACd,MAAM,GAMP;QACC,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3C,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;aACL,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;YACpD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,MAAM,CAAC,EAAE;aACnB,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACnD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,UAAU,EAAE,MAAM,CAAC,EAAE;aACtB,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAClD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC9C,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,kBAAkB,CAAC,EAAE,CAAC;YACnF,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,gBAAgB,CAAC,EAAE,CAAC;YACjF,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,mBAAmB,CAAC,EAAE,CAAC;YACpF,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,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 { assert } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_Category, CLASS_NAME_Model, CLASS_NAME_Subject } from \"../../../common/internal/ClassNameDefinitions.js\";\nimport { createFilteredTree, FilteredNodesHandler } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\n\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { BaseFilteredTreeNode, FilteredTree, FilteredTreeRootNode } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\n\ninterface GenericFilteredTreeNode extends BaseFilteredTreeNode<GenericFilteredTreeNode> {\n  type: \"subject\" | \"model\";\n}\n\ninterface CategoryFilteredTreeNode extends BaseFilteredTreeNode<CategoryFilteredTreeNode> {\n  type: \"category\";\n  modelId: Id64String;\n}\n\ninterface ElementFilteredTreeNode extends BaseFilteredTreeNode<ElementFilteredTreeNode> {\n  type: \"element\";\n  modelId: Id64String;\n  categoryId: Id64String;\n}\n\ntype FilteredTreeNode = GenericFilteredTreeNode | CategoryFilteredTreeNode | ElementFilteredTreeNode;\n\n/** @internal */\nexport interface ModelsTreeFilterTargets {\n  subjectIds?: Id64Set;\n  modelIds?: Id64Set;\n  categories?: Array<{ modelId: Id64String | undefined; categoryIds: Id64Set }>;\n  elements?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isFilterTarget: boolean }> }>;\n}\n\n/** @internal */\nexport async function createFilteredModelsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  filteringPaths: HierarchyFilteringPath[];\n}): Promise<FilteredTree<ModelsTreeFilterTargets>> {\n  const { imodelAccess, filteringPaths } = props;\n  return createFilteredTree({\n    filteredNodesHandler: new ModelsTreeFilteredNodesHandler({ imodelAccess }),\n    filteringPaths,\n  });\n}\n\ninterface FilterTargetsInternal {\n  subjectIds?: Id64Set;\n  modelIds?: Id64Set;\n  categories?: Map<ModelId, Set<CategoryId>>;\n  elements?: Map<ModelCategoryKey, Map<ElementId, { isFilterTarget: boolean }>>;\n}\n\ninterface ModelsTreeFilteredNodesHandlerProps {\n  imodelAccess: ECClassHierarchyInspector;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\nclass ModelsTreeFilteredNodesHandler extends FilteredNodesHandler<void, ModelsTreeFilterTargets, FilteredTreeNode> {\n  readonly #props: ModelsTreeFilteredNodesHandlerProps;\n  constructor(props: ModelsTreeFilteredNodesHandlerProps) {\n    super();\n    this.#props = props;\n  }\n\n  public convertNodesToFilterTargets(filteredNodes: FilteredTreeNode[]): ModelsTreeFilterTargets | undefined {\n    const filterTargets: FilterTargetsInternal = {};\n\n    filteredNodes.forEach((filteredNode) => this.collectFilterTargets(filterTargets, filteredNode));\n\n    return this.convertInternalFilterTargets(filterTargets);\n  }\n\n  public async getProcessedFilteredNodes(): Promise<void> {}\n\n  private convertInternalFilterTargets(filterTargets: FilterTargetsInternal): ModelsTreeFilterTargets | undefined {\n    if (!filterTargets.categories && !filterTargets.subjectIds && !filterTargets.elements && !filterTargets.modelIds) {\n      return undefined;\n    }\n\n    return {\n      categories: filterTargets.categories\n        ? [...filterTargets.categories.entries()].map(([modelId, categoryIds]) => {\n            return { modelId, categoryIds };\n          })\n        : undefined,\n      elements: filterTargets.elements\n        ? [...filterTargets.elements.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n      modelIds: filterTargets.modelIds,\n      subjectIds: filterTargets.subjectIds,\n    };\n  }\n\n  private collectFilterTargets(changeTargets: FilterTargetsInternal, filteredNode: FilteredTreeNode) {\n    if (filteredNode.isFilterTarget) {\n      this.addTarget(changeTargets, filteredNode);\n      return;\n    }\n\n    if (filteredNode.type === \"element\") {\n      // need to add parent ids as filter target will be an element\n      this.addTarget(changeTargets, filteredNode);\n    }\n\n    if (!filteredNode.children) {\n      return;\n    }\n\n    for (const child of filteredNode.children.values()) {\n      this.collectFilterTargets(changeTargets, child);\n    }\n  }\n\n  private addTarget(filterTargets: FilterTargetsInternal, node: FilteredTreeNode) {\n    switch (node.type) {\n      case \"subject\":\n        (filterTargets.subjectIds ??= new Set()).add(node.id);\n        return;\n      case \"model\":\n        (filterTargets.modelIds ??= new Set()).add(node.id);\n        return;\n      case \"category\":\n        const categories = (filterTargets.categories ??= new Map()).get(node.modelId);\n        if (categories) {\n          categories.add(node.id);\n          return;\n        }\n        filterTargets.categories.set(node.modelId, new Set([node.id]));\n        return;\n      case \"element\":\n        const modelCategoryKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements = (filterTargets.elements ??= new Map()).get(modelCategoryKey);\n        if (elements) {\n          elements.set(node.id, { isFilterTarget: node.isFilterTarget });\n        } else {\n          filterTargets.elements.set(modelCategoryKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));\n        }\n    }\n  }\n\n  private createModelCategoryKey(modelId: Id64String, categoryId: Id64String): ModelCategoryKey {\n    return `${modelId}-${categoryId}`;\n  }\n\n  private parseModelCategoryKey(key: ModelCategoryKey): { modelId: Id64String; categoryId: Id64String } {\n    const [modelId, categoryId] = key.split(\"-\");\n    return { modelId, categoryId };\n  }\n\n  public createFilteredTreeNode({\n    type,\n    id,\n    isFilterTarget,\n    parent,\n  }: {\n    type: FilteredTreeNode[\"type\"];\n    id: string;\n    isFilterTarget: boolean;\n    parent: FilteredTreeNode | FilteredTreeRootNode<FilteredTreeNode>;\n  }): FilteredTreeNode {\n    if (type === \"subject\" || type === \"model\") {\n      return {\n        id,\n        isFilterTarget,\n        type,\n      };\n    }\n\n    if (type === \"category\") {\n      assert(\"type\" in parent && parent.type === \"model\");\n      return {\n        id,\n        isFilterTarget,\n        type,\n        modelId: parent.id,\n      };\n    }\n\n    if (\"type\" in parent && parent.type === \"category\") {\n      return {\n        id,\n        isFilterTarget,\n        type,\n        modelId: parent.modelId,\n        categoryId: parent.id,\n      };\n    }\n\n    if (\"type\" in parent && parent.type === \"element\") {\n      return {\n        id,\n        isFilterTarget,\n        type,\n        modelId: parent.modelId,\n        categoryId: parent.categoryId,\n      };\n    }\n\n    throw new Error(\"Invalid parent node type\");\n  }\n\n  public async getType(className: string): Promise<FilteredTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Subject)) {\n      return \"subject\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Model)) {\n      return \"model\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Category)) {\n      return \"category\";\n    }\n    return \"element\";\n  }\n}\n"]}

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

@@ -0,0 +1,76 @@
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
+import type { Observable } from "rxjs";
+import type { Id64Arg } from "@itwin/core-bentley";
+import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { AlwaysAndNeverDrawnElementInfo } from "../../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import type { BaseTreeVisibilityHandlerOverrides, TreeSpecificVisibilityHandler } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { TreeWidgetViewport } from "../../../common/TreeWidgetViewport.js";
+import type { HierarchyVisibilityHandlerOverridableMethod, HierarchyVisibilityOverrideHandler, VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
+import type { ModelsTreeIdsCache } from "../ModelsTreeIdsCache.js";
+import type { ModelsTreeFilterTargets } from "./FilteredTree.js";
+/**
+ * Functionality of Models tree visibility handler that can be overridden.
+ * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.
+ * @beta
+ */
+export interface ModelsTreeVisibilityHandlerOverrides extends BaseTreeVisibilityHandlerOverrides {
+    getSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
+        subjectIds: Id64Arg;
+    }) => Promise<VisibilityStatus>>;
+    getElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
+        node: GroupingHierarchyNode;
+    }) => Promise<VisibilityStatus>>;
+    changeSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
+        subjectIds: Id64Arg;
+        on: boolean;
+    }) => Promise<void>>;
+    changeElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
+        node: GroupingHierarchyNode;
+        on: boolean;
+    }) => Promise<void>>;
+}
+/** @internal */
+export interface ModelsTreeVisibilityHandlerProps {
+    idsCache: ModelsTreeIdsCache;
+    viewport: TreeWidgetViewport;
+    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;
+    overrideHandler: HierarchyVisibilityOverrideHandler;
+    overrides?: ModelsTreeVisibilityHandlerOverrides;
+}
+/**
+ * Handles visibility status of models tree nodes.
+ *
+ * It knows how to get and change visibility status of nodes created by hierarchy definition.
+ * @internal
+ */
+export declare class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeFilterTargets> {
+    #private;
+    constructor(constructorProps: ModelsTreeVisibilityHandlerProps);
+    [Symbol.dispose](): void;
+    changeFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets, 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: ModelsTreeFilterTargets): Observable<VisibilityStatus>;
+    private getCategories;
+    private getAllCategories;
+    private getElementsCount;
+    private getModels;
+    private getSubCategories;
+    private getSubModels;
+    private getGroupingNodeInfo;
+}
+/**
+ * Creates models tree visibility handler. Is used by integration and performance tests.
+ * @internal
+ */
+export declare function createModelsTreeVisibilityHandler(props: {
+    viewport: TreeWidgetViewport;
+    idsCache: ModelsTreeIdsCache;
+    imodelAccess: ECClassHierarchyInspector;
+    overrides?: ModelsTreeVisibilityHandlerOverrides;
+    filteredPaths?: HierarchyFilteringPath[];
+}): HierarchyVisibilityHandlerImpl<ModelsTreeFilterTargets>;
+//# sourceMappingURL=ModelsTreeVisibilityHandler.d.ts.map

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

@@ -0,0 +1,269 @@
+/*---------------------------------------------------------------------------------------------
+ * 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 { releaseMainThreadOnItemsCount } from "../../../common/internal/Utils.js";
+import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
+import { ModelsTreeNode } from "../ModelsTreeNode.js";
+import { createFilteredModelsTree } from "./FilteredTree.js";
+import { ModelsTreeVisibilityHelper } from "./ModelsTreeVisibilityHelper.js";
+/**
+ * Handles visibility status of models tree nodes.
+ *
+ * It knows how to get and change visibility status of nodes created by hierarchy definition.
+ * @internal
+ */
+export class ModelsTreeVisibilityHandler {
+    #visibilityHelper;
+    #props;
+    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),
+            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),
+            getAllCategories: () => this.getAllCategories(),
+        };
+        this.#visibilityHelper = new ModelsTreeVisibilityHelper({
+            viewport: this.#props.viewport,
+            idsCache: this.#props.idsCache,
+            alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
+            overrideHandler: this.#props.overrideHandler,
+            baseIdsCache,
+            overrides: this.#props.overrides,
+        });
+    }
+    [Symbol.dispose]() {
+        this.#visibilityHelper[Symbol.dispose]();
+    }
+    changeFilterTargetsVisibilityStatus(targets, on) {
+        return defer(() => {
+            const { subjectIds, modelIds, categories, elements } = targets;
+            const observables = new Array();
+            if (subjectIds?.size) {
+                observables.push(this.#visibilityHelper.changeSubjectsVisibilityStatus({ subjectIds, 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 (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);
+            const result = this.#visibilityHelper.getGroupedElementsVisibilityStatus({
+                categoryId: nodeInfo.categoryId,
+                modelId: nodeInfo.modelId,
+                elementIds: nodeInfo.elementIds,
+            });
+            return this.#props.overrideHandler.createVisibilityHandlerResult({
+                overrideProps: { node },
+                nonOverriddenResult: result,
+                override: this.#props.overrides?.getElementGroupingNodeVisibilityStatus,
+            });
+        }
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (ModelsTreeNode.isSubjectNode(node)) {
+            // note: subject nodes may be merged to represent multiple subject instances
+            return this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds: node.key.instanceKeys.map((key) => key.id) });
+        }
+        if (ModelsTreeNode.isModelNode(node)) {
+            return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), type: "GeometricModel3d" });
+        }
+        const modelId = ModelsTreeNode.getModelId(node);
+        if (!modelId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (ModelsTreeNode.isCategoryNode(node)) {
+            return this.#visibilityHelper.getCategoriesVisibilityStatus({
+                categoryIds: node.key.instanceKeys.map(({ id }) => id),
+                modelId,
+                type: "SpatialCategory",
+            });
+        }
+        const categoryId = ModelsTreeNode.getCategoryId(node);
+        if (!categoryId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        return this.#visibilityHelper.getElementsVisibilityStatus({
+            elementIds: node.key.instanceKeys.map(({ id }) => id),
+            modelId,
+            categoryId,
+            type: "GeometricElement3d",
+        });
+    }
+    /** 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);
+                const result = this.#visibilityHelper.changeGroupedElementsVisibilityStatus({
+                    categoryId: nodeInfo.categoryId,
+                    modelId: nodeInfo.modelId,
+                    elementIds: nodeInfo.elementIds,
+                    on,
+                });
+                return this.#props.overrideHandler.createVisibilityHandlerResult({
+                    overrideProps: { node, on },
+                    nonOverriddenResult: result,
+                    override: this.#props.overrides?.changeElementGroupingNodeVisibilityStatus,
+                });
+            }
+            if (!HierarchyNode.isInstancesNode(node)) {
+                return EMPTY;
+            }
+            if (ModelsTreeNode.isSubjectNode(node)) {
+                return this.#visibilityHelper.changeSubjectsVisibilityStatus({
+                    subjectIds: node.key.instanceKeys.map((key) => key.id),
+                    on,
+                });
+            }
+            if (ModelsTreeNode.isModelNode(node)) {
+                return this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), on });
+            }
+            const modelId = ModelsTreeNode.getModelId(node);
+            if (!modelId) {
+                return EMPTY;
+            }
+            if (ModelsTreeNode.isCategoryNode(node)) {
+                return this.#visibilityHelper.changeCategoriesVisibilityStatus({
+                    categoryIds: node.key.instanceKeys.map(({ id }) => id),
+                    modelId,
+                    on,
+                });
+            }
+            const categoryId = ModelsTreeNode.getCategoryId(node);
+            if (!categoryId) {
+                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 { subjectIds, modelIds, categories, elements } = targets;
+            const observables = new Array();
+            if (subjectIds?.size) {
+                observables.push(this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds }));
+            }
+            if (modelIds?.size) {
+                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: "GeometricModel3d" }));
+            }
+            if (categories?.length) {
+                observables.push(from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.getCategoriesVisibilityStatus({
+                    categoryIds,
+                    modelId,
+                    type: "SpatialCategory",
+                }))));
+            }
+            if (elements?.length) {
+                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: "GeometricElement3d" }))));
+            }
+            return merge(...observables);
+        }).pipe(mergeVisibilityStatuses);
+    }
+    getCategories(props) {
+        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categoryIds) => ({ id: modelId, spatialCategories: categoryIds })))));
+    }
+    getAllCategories() {
+        return this.#props.idsCache.getAllCategories().pipe(map((categories) => ({ spatialCategories: categories })));
+    }
+    getElementsCount(props) {
+        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
+    }
+    getModels(props) {
+        // Models cache for categories that don't have models still adds them to the final map
+        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(mergeMap((categoryModelsMap) => categoryModelsMap.entries()), map(([categoryId, categoryModels]) => ({ id: categoryId, models: categoryModels })));
+    }
+    getSubCategories(props) {
+        return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));
+    }
+    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 modelId = ModelsTreeNode.getModelId(node);
+        const categoryId = ModelsTreeNode.getCategoryId(node);
+        assert(!!modelId && !!categoryId);
+        const elementIds = node.groupedInstanceKeys.map((key) => key.id);
+        return { modelId, categoryId, elementIds };
+    }
+}
+/**
+ * Creates models tree visibility handler. Is used by integration and performance tests.
+ * @internal
+ */
+export function createModelsTreeVisibilityHandler(props) {
+    return new HierarchyVisibilityHandlerImpl({
+        getFilteredTree: () => {
+            if (!props.filteredPaths) {
+                return undefined;
+            }
+            return createFilteredModelsTree({
+                filteringPaths: props.filteredPaths,
+                imodelAccess: props.imodelAccess,
+            });
+        },
+        getTreeSpecificVisibilityHandler: (info, overrideHandler) => {
+            return new ModelsTreeVisibilityHandler({
+                alwaysAndNeverDrawnElementInfo: info,
+                idsCache: props.idsCache,
+                viewport: props.viewport,
+                overrideHandler,
+                overrides: props.overrides,
+            });
+        },
+        viewport: props.viewport,
+        componentId: "Test",
+    });
+}
+//# sourceMappingURL=ModelsTreeVisibilityHandler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ModelsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/visibility/ModelsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,8BAA8B,EAAE,MAAM,8DAA8D,CAAC;AAC9G,OAAO,EAAE,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AAClF,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AA8C7E;;;;;GAKG;AACH,MAAM,OAAO,2BAA2B;IACtC,iBAAiB,CAA6B;IACrC,MAAM,CAAmC;IAElD,YAAY,gBAAkD;QAC5D,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,6EAA6E;QAC7E,mDAAmD;QACnD,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YACnD,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAC3C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACjD,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;YAC/D,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE;SAChD,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,0BAA0B,CAAC;YACtD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YAC5C,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;SACjC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAgC,EAAE,EAAW;QACtF,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC9F,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBACtD,WAAW;oBACX,OAAO;oBACP,EAAE;iBACH,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACxH,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBACvE,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;gBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE;gBACvB,mBAAmB,EAAE,MAAM;gBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,sCAAsC;aACxE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,4EAA4E;YAC5E,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACxH,CAAC;QAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC7I,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;gBAC1D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACtD,OAAO;gBACP,IAAI,EAAE,iBAAiB;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACrD,OAAO;YACP,UAAU;YACV,IAAI,EAAE,oBAAoB;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAC1E,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,EAAE;iBACH,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;oBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;oBAC3B,mBAAmB,EAAE,MAAM;oBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,yCAAyC;iBAC3E,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;oBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1H,CAAC;YAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBAC7D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;oBACtD,OAAO;oBACP,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,KAAK,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO;gBACP,UAAU;gBACV,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAgC;QACtE,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YACvF,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC7G,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;oBACnD,WAAW;oBACX,OAAO;oBACP,IAAI,EAAE,iBAAiB;iBACxB,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC7I,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnC,CAAC;IAEO,aAAa,CAAC,KAAmD;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CACvJ,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAChH,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAEO,SAAS,CAAC,KAA+C;QAC/D,sFAAsF;QACtF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAC5E,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,EAC5D,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC,CACpF,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAAC,IAAI,CAC9D,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,EAC7F,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,IAA2B;QACrD,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC;QAElC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;IAC7C,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,iCAAiC,CAAC,KAMjD;IACC,OAAO,IAAI,8BAA8B,CAA0B;QACjE,eAAe,EAAE,GAA+D,EAAE;YAChF,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;gBACzB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,wBAAwB,CAAC;gBAC9B,cAAc,EAAE,KAAK,CAAC,aAAa;gBACnC,YAAY,EAAE,KAAK,CAAC,YAAY;aACjC,CAAC,CAAC;QACL,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE;YAC1D,OAAO,IAAI,2BAA2B,CAAC;gBACrC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe;gBACf,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,MAAM;KACpB,CAAC,CAAC;AACL,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 { concat, defer, EMPTY, from, map, merge, mergeMap, of } from \"rxjs\";\nimport { assert, Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { HierarchyVisibilityHandlerImpl } from \"../../../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { releaseMainThreadOnItemsCount } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ModelsTreeNode } from \"../ModelsTreeNode.js\";\nimport { createFilteredModelsTree } from \"./FilteredTree.js\";\nimport { ModelsTreeVisibilityHelper } from \"./ModelsTreeVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { FilteredTree } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\nimport type {\n  BaseIdsCache,\n  BaseTreeVisibilityHandlerOverrides,\n  TreeSpecificVisibilityHandler,\n} from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type {\n  HierarchyVisibilityHandlerOverridableMethod,\n  HierarchyVisibilityOverrideHandler,\n  VisibilityStatus,\n} from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ModelsTreeIdsCache } from \"../ModelsTreeIdsCache.js\";\nimport type { ModelsTreeFilterTargets } from \"./FilteredTree.js\";\n\n/**\n * Functionality of Models tree visibility handler that can be overridden.\n * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.\n * @beta\n */\nexport interface ModelsTreeVisibilityHandlerOverrides extends BaseTreeVisibilityHandlerOverrides {\n  getSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg }) => Promise<VisibilityStatus>>;\n  getElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { node: GroupingHierarchyNode }) => Promise<VisibilityStatus>>;\n\n  changeSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg; on: boolean }) => Promise<void>>;\n  changeElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<\n    (props: { node: GroupingHierarchyNode; on: boolean }) => Promise<void>\n  >;\n}\n\n/** @internal */\nexport interface ModelsTreeVisibilityHandlerProps {\n  idsCache: ModelsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n}\n\n/**\n * Handles visibility status of models tree nodes.\n *\n * It knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeFilterTargets> {\n  #visibilityHelper: ModelsTreeVisibilityHelper;\n  readonly #props: ModelsTreeVisibilityHandlerProps;\n\n  constructor(constructorProps: ModelsTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.\n    // We won't need to create a custom base ids cache.\n    const baseIdsCache: BaseIdsCache = {\n      getCategories: (props) => this.getCategories(props),\n      getElementsCount: (props) => this.getElementsCount(props),\n      getModels: (props) => this.getModels(props),\n      getSubCategories: (props) => this.getSubCategories(props),\n      getSubModels: (props) => this.getSubModels(props),\n      hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),\n      getAllCategories: () => this.getAllCategories(),\n    };\n    this.#visibilityHelper = new ModelsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      overrideHandler: this.#props.overrideHandler,\n      baseIdsCache,\n      overrides: this.#props.overrides,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.changeSubjectsVisibilityStatus({ subjectIds, on }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.changeCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                on,\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          from(elements).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      const nodeInfo = this.getGroupingNodeInfo(node);\n      const result = this.#visibilityHelper.getGroupedElementsVisibilityStatus({\n        categoryId: nodeInfo.categoryId,\n        modelId: nodeInfo.modelId,\n        elementIds: nodeInfo.elementIds,\n      });\n      return this.#props.overrideHandler.createVisibilityHandlerResult({\n        overrideProps: { node },\n        nonOverriddenResult: result,\n        override: this.#props.overrides?.getElementGroupingNodeVisibilityStatus,\n      });\n    }\n\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isSubjectNode(node)) {\n      // note: subject nodes may be merged to represent multiple subject instances\n      return this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds: node.key.instanceKeys.map((key) => key.id) });\n    }\n\n    if (ModelsTreeNode.isModelNode(node)) {\n      return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), type: \"GeometricModel3d\" });\n    }\n\n    const modelId = ModelsTreeNode.getModelId(node);\n    if (!modelId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isCategoryNode(node)) {\n      return this.#visibilityHelper.getCategoriesVisibilityStatus({\n        categoryIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        type: \"SpatialCategory\",\n      });\n    }\n\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    if (!categoryId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map(({ id }) => id),\n      modelId,\n      categoryId,\n      type: \"GeometricElement3d\",\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (HierarchyNode.isClassGroupingNode(node)) {\n        const nodeInfo = this.getGroupingNodeInfo(node);\n        const result = this.#visibilityHelper.changeGroupedElementsVisibilityStatus({\n          categoryId: nodeInfo.categoryId,\n          modelId: nodeInfo.modelId,\n          elementIds: nodeInfo.elementIds,\n          on,\n        });\n        return this.#props.overrideHandler.createVisibilityHandlerResult({\n          overrideProps: { node, on },\n          nonOverriddenResult: result,\n          override: this.#props.overrides?.changeElementGroupingNodeVisibilityStatus,\n        });\n      }\n\n      if (!HierarchyNode.isInstancesNode(node)) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isSubjectNode(node)) {\n        return this.#visibilityHelper.changeSubjectsVisibilityStatus({\n          subjectIds: node.key.instanceKeys.map((key) => key.id),\n          on,\n        });\n      }\n\n      if (ModelsTreeNode.isModelNode(node)) {\n        return this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), on });\n      }\n\n      const modelId = ModelsTreeNode.getModelId(node);\n      if (!modelId) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isCategoryNode(node)) {\n        return this.#visibilityHelper.changeCategoriesVisibilityStatus({\n          categoryIds: node.key.instanceKeys.map(({ id }) => id),\n          modelId,\n          on,\n        });\n      }\n\n      const categoryId = ModelsTreeNode.getCategoryId(node);\n      if (!categoryId) {\n        return EMPTY;\n      }\n\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        categoryId,\n        on,\n      });\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: \"GeometricModel3d\" }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.getCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                type: \"SpatialCategory\",\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          from(elements).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: \"GeometricElement3d\" }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    }).pipe(mergeVisibilityStatuses);\n  }\n\n  private getCategories(props: Parameters<BaseIdsCache[\"getCategories\"]>[0]): ReturnType<BaseIdsCache[\"getCategories\"]> {\n    return from(Id64.iterable(props.modelIds)).pipe(\n      mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categoryIds) => ({ id: modelId, spatialCategories: categoryIds })))),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(map((categories) => ({ spatialCategories: categories })));\n  }\n\n  private getElementsCount(props: Parameters<BaseIdsCache[\"getElementsCount\"]>[0]): ReturnType<BaseIdsCache[\"getElementsCount\"]> {\n    return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);\n  }\n\n  private getModels(props: Parameters<BaseIdsCache[\"getModels\"]>[0]): ReturnType<BaseIdsCache[\"getModels\"]> {\n    // Models cache for categories that don't have models still adds them to the final map\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(\n      mergeMap((categoryModelsMap) => categoryModelsMap.entries()),\n      map(([categoryId, categoryModels]) => ({ id: categoryId, models: categoryModels })),\n    );\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return this.#props.idsCache.getModelCategoryIds(modelId).pipe(\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return from(Id64.iterable(props.categoryIds)).pipe(\n      mergeMap((categoryId) =>\n        this.#props.idsCache.getCategoriesElementModels(categoryId).pipe(\n          mergeMap((categoryModelsMap) => {\n            const models = categoryModelsMap.get(categoryId);\n            if (!models) {\n              return of({ id: categoryId, subModels: undefined });\n            }\n            return from(models).pipe(\n              mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryId)),\n              map((subModels) => ({ id: categoryId, subModels })),\n            );\n          }),\n        ),\n      ),\n    );\n  }\n\n  private getGroupingNodeInfo(node: GroupingHierarchyNode) {\n    const modelId = ModelsTreeNode.getModelId(node);\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    assert(!!modelId && !!categoryId);\n\n    const elementIds = node.groupedInstanceKeys.map((key) => key.id);\n    return { modelId, categoryId, elementIds };\n  }\n}\n\n/**\n * Creates models tree visibility handler. Is used by integration and performance tests.\n * @internal\n */\nexport function createModelsTreeVisibilityHandler(props: {\n  viewport: TreeWidgetViewport;\n  idsCache: ModelsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n  filteredPaths?: HierarchyFilteringPath[];\n}) {\n  return new HierarchyVisibilityHandlerImpl<ModelsTreeFilterTargets>({\n    getFilteredTree: (): undefined | Promise<FilteredTree<ModelsTreeFilterTargets>> => {\n      if (!props.filteredPaths) {\n        return undefined;\n      }\n      return createFilteredModelsTree({\n        filteringPaths: props.filteredPaths,\n        imodelAccess: props.imodelAccess,\n      });\n    },\n    getTreeSpecificVisibilityHandler: (info, overrideHandler) => {\n      return new ModelsTreeVisibilityHandler({\n        alwaysAndNeverDrawnElementInfo: info,\n        idsCache: props.idsCache,\n        viewport: props.viewport,\n        overrideHandler,\n        overrides: props.overrides,\n      });\n    },\n    viewport: props.viewport,\n    componentId: \"Test\",\n  });\n}\n"]}

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

@@ -0,0 +1,53 @@
+import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { Observable } from "rxjs";
+import type { Id64Arg, Id64String } from "@itwin/core-bentley";
+import type { BaseVisibilityHelperProps } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
+import type { ModelsTreeIdsCache } from "../ModelsTreeIdsCache.js";
+import type { ModelsTreeVisibilityHandlerOverrides } from "./ModelsTreeVisibilityHandler.js";
+/** @internal */
+export type ModelsTreeVisibilityHelperProps = BaseVisibilityHelperProps & {
+    idsCache: ModelsTreeIdsCache;
+    overrides?: ModelsTreeVisibilityHandlerOverrides;
+};
+/**
+ * Visibility status helper for models tree.
+ *
+ * It extends base visibility status helper and provides methods to get and change visibility status of subjects and grouped elements.
+ * @internal
+ */
+export declare class ModelsTreeVisibilityHelper extends BaseVisibilityHelper {
+    #private;
+    constructor(props: ModelsTreeVisibilityHelperProps);
+    /**
+     * Gets visibility status of subjects.
+     *
+     * Determines visibility status by checking visibility status of related models.
+     */
+    getSubjectsVisibilityStatus(props: {
+        subjectIds: Id64Arg;
+    }): Observable<VisibilityStatus>;
+    /** Gets visibility status of grouped elements */
+    getGroupedElementsVisibilityStatus(props: {
+        modelId: Id64String;
+        categoryId: Id64String;
+        elementIds: Id64Arg;
+    }): Observable<VisibilityStatus>;
+    /**
+     * Changes visibility status of subjects.
+     *
+     * Does this by changing visibility status of related models.
+     */
+    changeSubjectsVisibilityStatus(props: {
+        subjectIds: Id64Arg;
+        on: boolean;
+    }): Observable<void>;
+    /** Changes visibility of grouped elements. */
+    changeGroupedElementsVisibilityStatus(props: {
+        modelId: Id64String;
+        categoryId: Id64String;
+        elementIds: Id64Arg;
+        on: boolean;
+    }): Observable<void>;
+}
+//# sourceMappingURL=ModelsTreeVisibilityHelper.d.ts.map