@itwin/appui-react 4.0.0-dev.3 → 4.0.0-dev.6

package/lib/cjs/appui-react/imodel-components/models-tree/ModelsVisibilityHandler.js

@@ -0,0 +1,482 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module IModelComponents
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SubjectModelIdsCache = exports.ModelsVisibilityHandler = exports.ModelsTreeNodeType = void 0;
+const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
+const core_frontend_1 = require("@itwin/core-frontend");
+const presentation_common_1 = require("@itwin/presentation-common");
+const presentation_frontend_1 = require("@itwin/presentation-frontend");
+const UiFramework_1 = require("../../UiFramework");
+const Utils_1 = require("./Utils");
+/**
+ * Visibility tree node types.
+ * @beta
+ * @deprecated in 3.6. Was moved to `@itwin/tree-widget-react` package.
+ */
+var ModelsTreeNodeType;
+(function (ModelsTreeNodeType) {
+    ModelsTreeNodeType[ModelsTreeNodeType["Unknown"] = 0] = "Unknown";
+    ModelsTreeNodeType[ModelsTreeNodeType["Subject"] = 1] = "Subject";
+    ModelsTreeNodeType[ModelsTreeNodeType["Model"] = 2] = "Model";
+    ModelsTreeNodeType[ModelsTreeNodeType["Category"] = 3] = "Category";
+    ModelsTreeNodeType[ModelsTreeNodeType["Element"] = 4] = "Element";
+    ModelsTreeNodeType[ModelsTreeNodeType["Grouping"] = 5] = "Grouping";
+})(ModelsTreeNodeType = exports.ModelsTreeNodeType || (exports.ModelsTreeNodeType = {}));
+/**
+ * Visibility handler used by [[ModelsTree]] to control visibility of the tree items.
+ * @alpha
+ * @deprecated in 3.6. Was moved to `@itwin/tree-widget-react` package.
+ */
+class ModelsVisibilityHandler {
+    constructor(props) {
+        var _a;
+        this._listeners = new Array();
+        this.onVisibilityChange = new core_bentley_1.BeEvent();
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onViewChanged = (_vp) => {
+            this.onVisibilityChangeInternal();
+        };
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onElementAlwaysDrawnChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onElementNeverDrawnChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        this._props = props;
+        this._subjectModelIdsCache = (_a = props.subjectModelIdsCache) !== null && _a !== void 0 ? _a : new SubjectModelIdsCache(this._props.viewport.iModel);
+        this._elementIdsCache = new ElementIdsCache(this._props.viewport.iModel, this._props.rulesetId);
+        this._listeners.push(this._props.viewport.onViewedCategoriesPerModelChanged.addListener(this.onViewChanged));
+        this._listeners.push(this._props.viewport.onViewedCategoriesChanged.addListener(this.onViewChanged));
+        this._listeners.push(this._props.viewport.onViewedModelsChanged.addListener(this.onViewChanged));
+        this._listeners.push(this._props.viewport.onAlwaysDrawnChanged.addListener(this.onElementAlwaysDrawnChanged));
+        this._listeners.push(this._props.viewport.onNeverDrawnChanged.addListener(this.onElementNeverDrawnChanged));
+        if (this._props.hierarchyAutoUpdateEnabled) {
+            this._listeners.push(presentation_frontend_1.Presentation.presentation.onIModelHierarchyChanged.addListener(/* istanbul ignore next */ () => this._elementIdsCache.clear()));
+        }
+    }
+    dispose() {
+        this._listeners.forEach((remove) => remove());
+        clearTimeout(this._pendingVisibilityChange);
+    }
+    /** Sets data provider that is used to get filtered tree hierarchy. */
+    setFilteredDataProvider(provider) { this._filteredDataProvider = provider; }
+    static getNodeType(item, dataProvider) {
+        if (presentation_common_1.NodeKey.isClassGroupingNodeKey(dataProvider.getNodeKey(item)))
+            return ModelsTreeNodeType.Grouping;
+        if (!item.extendedData)
+            return ModelsTreeNodeType.Unknown;
+        if (this.isSubjectNode(item))
+            return ModelsTreeNodeType.Subject;
+        if (this.isModelNode(item))
+            return ModelsTreeNodeType.Model;
+        if (this.isCategoryNode(item))
+            return ModelsTreeNodeType.Category;
+        return ModelsTreeNodeType.Element;
+    }
+    static isSubjectNode(node) {
+        return node.extendedData && node.extendedData.isSubject;
+    }
+    static isModelNode(node) {
+        return node.extendedData && node.extendedData.isModel;
+    }
+    static isCategoryNode(node) {
+        return node.extendedData && node.extendedData.isCategory;
+    }
+    /** Returns visibility status of the tree node. */
+    getVisibilityStatus(node, nodeKey) {
+        if (presentation_common_1.NodeKey.isClassGroupingNodeKey(nodeKey))
+            return this.getElementGroupingNodeDisplayStatus(node.id, nodeKey);
+        if (!presentation_common_1.NodeKey.isInstancesNodeKey(nodeKey))
+            return { state: "hidden", isDisabled: true };
+        if (ModelsVisibilityHandler.isSubjectNode(node)) {
+            // note: subject nodes may be merged to represent multiple subject instances
+            return this.getSubjectNodeVisibility(nodeKey.instanceKeys.map((key) => key.id), node);
+        }
+        if (ModelsVisibilityHandler.isModelNode(node))
+            return this.getModelDisplayStatus(nodeKey.instanceKeys[0].id);
+        if (ModelsVisibilityHandler.isCategoryNode(node))
+            return this.getCategoryDisplayStatus(nodeKey.instanceKeys[0].id, this.getCategoryParentModelId(node));
+        return this.getElementDisplayStatus(nodeKey.instanceKeys[0].id, this.getElementModelId(node), this.getElementCategoryId(node));
+    }
+    /** Changes visibility of the items represented by the tree node. */
+    async changeVisibility(node, nodeKey, on) {
+        if (presentation_common_1.NodeKey.isClassGroupingNodeKey(nodeKey)) {
+            await this.changeElementGroupingNodeState(nodeKey, on);
+            return;
+        }
+        if (!presentation_common_1.NodeKey.isInstancesNodeKey(nodeKey))
+            return;
+        if (ModelsVisibilityHandler.isSubjectNode(node)) {
+            await this.changeSubjectNodeState(nodeKey.instanceKeys.map((key) => key.id), node, on);
+        }
+        else if (ModelsVisibilityHandler.isModelNode(node)) {
+            await this.changeModelState(nodeKey.instanceKeys[0].id, on);
+        }
+        else if (ModelsVisibilityHandler.isCategoryNode(node)) {
+            this.changeCategoryState(nodeKey.instanceKeys[0].id, this.getCategoryParentModelId(node), on);
+        }
+        else {
+            await this.changeElementState(nodeKey.instanceKeys[0].id, this.getElementModelId(node), this.getElementCategoryId(node), on);
+        }
+    }
+    async getSubjectNodeVisibility(ids, node) {
+        if (!this._props.viewport.view.isSpatialView())
+            return { isDisabled: true, state: "hidden", tooltip: createTooltip("disabled", "subject.nonSpatialView") };
+        if (this._filteredDataProvider)
+            return this.getFilteredSubjectDisplayStatus(this._filteredDataProvider, ids, node);
+        return this.getSubjectDisplayStatus(ids);
+    }
+    async getSubjectDisplayStatus(ids) {
+        const modelIds = await this.getSubjectModelIds(ids);
+        const isDisplayed = modelIds.some((modelId) => this.getModelDisplayStatus(modelId).state === "visible");
+        if (isDisplayed)
+            return { state: "visible", tooltip: createTooltip("visible", "subject.atLeastOneModelVisible") };
+        return { state: "hidden", tooltip: createTooltip("hidden", "subject.allModelsHidden") };
+    }
+    async getFilteredSubjectDisplayStatus(provider, ids, node) {
+        if (provider.nodeMatchesFilter(node))
+            return this.getSubjectDisplayStatus(ids);
+        const children = await provider.getNodes(node);
+        const childrenDisplayStatuses = await Promise.all(children.map((childNode) => this.getVisibilityStatus(childNode, provider.getNodeKey(childNode))));
+        if (childrenDisplayStatuses.some((status) => status.state === "visible"))
+            return { state: "visible", tooltip: createTooltip("visible", "subject.atLeastOneModelVisible") };
+        return { state: "hidden", tooltip: createTooltip("hidden", "subject.allModelsHidden") };
+    }
+    getModelDisplayStatus(id) {
+        if (!this._props.viewport.view.isSpatialView())
+            return { isDisabled: true, state: "hidden", tooltip: createTooltip("disabled", "model.nonSpatialView") };
+        const isDisplayed = this._props.viewport.view.viewsModel(id);
+        return { state: isDisplayed ? "visible" : "hidden", tooltip: createTooltip(isDisplayed ? "visible" : "hidden", undefined) };
+    }
+    getCategoryDisplayStatus(id, parentModelId) {
+        if (parentModelId) {
+            if (this.getModelDisplayStatus(parentModelId).state === "hidden")
+                return { state: "hidden", isDisabled: true, tooltip: createTooltip("disabled", "category.modelNotDisplayed") };
+            const override = this._props.viewport.perModelCategoryVisibility.getOverride(parentModelId, id);
+            switch (override) {
+                case core_frontend_1.PerModelCategoryVisibility.Override.Show:
+                    return { state: "visible", tooltip: createTooltip("visible", "category.displayedThroughPerModelOverride") };
+                case core_frontend_1.PerModelCategoryVisibility.Override.Hide:
+                    return { state: "hidden", tooltip: createTooltip("hidden", "category.hiddenThroughPerModelOverride") };
+            }
+        }
+        const isDisplayed = this._props.viewport.view.viewsCategory(id);
+        return {
+            state: isDisplayed ? "visible" : "hidden",
+            tooltip: isDisplayed
+                ? createTooltip("visible", "category.displayedThroughCategorySelector")
+                : createTooltip("hidden", "category.hiddenThroughCategorySelector"),
+        };
+    }
+    async getElementGroupingNodeDisplayStatus(_id, key) {
+        const { modelId, categoryId, elementIds } = await this.getGroupedElementIds(key);
+        if (!modelId || !this._props.viewport.view.viewsModel(modelId))
+            return { isDisabled: true, state: "hidden", tooltip: createTooltip("disabled", "element.modelNotDisplayed") };
+        if (this._props.viewport.alwaysDrawn !== undefined && this._props.viewport.alwaysDrawn.size > 0) {
+            let atLeastOneElementForceDisplayed = false;
+            for await (const elementId of elementIds.getElementIds()) {
+                if (this._props.viewport.alwaysDrawn.has(elementId)) {
+                    atLeastOneElementForceDisplayed = true;
+                    break;
+                }
+            }
+            if (atLeastOneElementForceDisplayed)
+                return { state: "visible", tooltip: createTooltip("visible", "element.displayedThroughAlwaysDrawnList") };
+        }
+        if (this._props.viewport.alwaysDrawn !== undefined && this._props.viewport.alwaysDrawn.size !== 0 && this._props.viewport.isAlwaysDrawnExclusive)
+            return { state: "hidden", tooltip: createTooltip("hidden", "element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn") };
+        // istanbul ignore else
+        if (this._props.viewport.neverDrawn !== undefined && this._props.viewport.neverDrawn.size > 0) {
+            let allElementsForceHidden = true;
+            for await (const elementId of elementIds.getElementIds()) {
+                if (!this._props.viewport.neverDrawn.has(elementId)) {
+                    allElementsForceHidden = false;
+                    break;
+                }
+            }
+            if (allElementsForceHidden)
+                return { state: "hidden", tooltip: createTooltip("visible", "element.hiddenThroughNeverDrawnList") };
+        }
+        if (categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === "visible")
+            return { state: "visible", tooltip: createTooltip("visible", undefined) };
+        return { state: "hidden", tooltip: createTooltip("hidden", "element.hiddenThroughCategory") };
+    }
+    getElementDisplayStatus(elementId, modelId, categoryId) {
+        if (!modelId || !this._props.viewport.view.viewsModel(modelId))
+            return { isDisabled: true, state: "hidden", tooltip: createTooltip("disabled", "element.modelNotDisplayed") };
+        if (this._props.viewport.neverDrawn !== undefined && this._props.viewport.neverDrawn.has(elementId))
+            return { state: "hidden", tooltip: createTooltip("hidden", "element.hiddenThroughNeverDrawnList") };
+        if (this._props.viewport.alwaysDrawn !== undefined) {
+            if (this._props.viewport.alwaysDrawn.has(elementId))
+                return { state: "visible", tooltip: createTooltip("visible", "element.displayedThroughAlwaysDrawnList") };
+            if (this._props.viewport.alwaysDrawn.size !== 0 && this._props.viewport.isAlwaysDrawnExclusive)
+                return { state: "hidden", tooltip: createTooltip("hidden", "element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn") };
+        }
+        if (categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === "visible")
+            return { state: "visible", tooltip: createTooltip("visible", undefined) };
+        return { state: "hidden", tooltip: createTooltip("hidden", "element.hiddenThroughCategory") };
+    }
+    async changeSubjectNodeState(ids, node, on) {
+        if (!this._props.viewport.view.isSpatialView())
+            return;
+        if (this._filteredDataProvider)
+            return this.changeFilteredSubjectState(this._filteredDataProvider, ids, node, on);
+        return this.changeSubjectState(ids, on);
+    }
+    async changeFilteredSubjectState(provider, ids, node, on) {
+        if (provider.nodeMatchesFilter(node))
+            return this.changeSubjectState(ids, on);
+        const children = await provider.getNodes(node);
+        return Promise.all(children.map(async (childNode) => this.changeVisibility(childNode, provider.getNodeKey(childNode), on)));
+    }
+    async changeSubjectState(ids, on) {
+        const modelIds = await this.getSubjectModelIds(ids);
+        return this.changeModelsVisibility(modelIds, on);
+    }
+    async changeModelState(id, on) {
+        if (!this._props.viewport.view.isSpatialView())
+            return;
+        return this.changeModelsVisibility([id], on);
+    }
+    async changeModelsVisibility(ids, visible) {
+        if (visible)
+            return this._props.viewport.addViewedModels(ids);
+        else
+            this._props.viewport.changeModelDisplay(ids, false);
+    }
+    changeCategoryState(categoryId, parentModelId, on) {
+        if (parentModelId) {
+            const isDisplayedInSelector = this._props.viewport.view.viewsCategory(categoryId);
+            const ovr = (on === isDisplayedInSelector) ? core_frontend_1.PerModelCategoryVisibility.Override.None
+                : on ? core_frontend_1.PerModelCategoryVisibility.Override.Show : core_frontend_1.PerModelCategoryVisibility.Override.Hide;
+            this._props.viewport.perModelCategoryVisibility.setOverride(parentModelId, categoryId, ovr);
+            if (ovr === core_frontend_1.PerModelCategoryVisibility.Override.None && on) {
+                // we took off the override which means the category is displayed in selector, but
+                // doesn't mean all its subcategories are displayed - this call ensures that
+                this._props.viewport.changeCategoryDisplay([categoryId], true, true);
+            }
+            return;
+        }
+        this._props.viewport.changeCategoryDisplay([categoryId], on, on ? true : false);
+    }
+    async changeElementGroupingNodeState(key, on) {
+        const { modelId, categoryId, elementIds } = await this.getGroupedElementIds(key);
+        await this.changeElementsState(modelId, categoryId, elementIds.getElementIds(), on);
+    }
+    async changeElementState(id, modelId, categoryId, on) {
+        const childIdsContainer = this.getAssemblyElementIds(id);
+        async function* elementIds() {
+            yield id;
+            for await (const childId of childIdsContainer.getElementIds())
+                yield childId;
+        }
+        await this.changeElementsState(modelId, categoryId, elementIds(), on);
+    }
+    async changeElementsState(modelId, categoryId, elementIds, on) {
+        const isDisplayedByDefault = modelId && this.getModelDisplayStatus(modelId).state === "visible"
+            && categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === "visible";
+        const isHiddenDueToExclusiveAlwaysDrawnElements = this._props.viewport.isAlwaysDrawnExclusive && this._props.viewport.alwaysDrawn && 0 !== this._props.viewport.alwaysDrawn.size;
+        const currNeverDrawn = new Set(this._props.viewport.neverDrawn ? this._props.viewport.neverDrawn : []);
+        const currAlwaysDrawn = new Set(this._props.viewport.alwaysDrawn ?
+            this._props.viewport.alwaysDrawn : /* istanbul ignore next */ []);
+        for await (const elementId of elementIds) {
+            if (on) {
+                currNeverDrawn.delete(elementId);
+                if (!isDisplayedByDefault || isHiddenDueToExclusiveAlwaysDrawnElements)
+                    currAlwaysDrawn.add(elementId);
+            }
+            else {
+                currAlwaysDrawn.delete(elementId);
+                if (isDisplayedByDefault && !isHiddenDueToExclusiveAlwaysDrawnElements)
+                    currNeverDrawn.add(elementId);
+            }
+        }
+        this._props.viewport.setNeverDrawn(currNeverDrawn);
+        this._props.viewport.setAlwaysDrawn(currAlwaysDrawn, this._props.viewport.isAlwaysDrawnExclusive);
+    }
+    onVisibilityChangeInternal() {
+        if (this._pendingVisibilityChange)
+            return;
+        this._pendingVisibilityChange = setTimeout(() => {
+            this.onVisibilityChange.raiseEvent();
+            this._pendingVisibilityChange = undefined;
+        }, 0);
+    }
+    getCategoryParentModelId(categoryNode) {
+        return categoryNode.extendedData ? categoryNode.extendedData.modelId : /* istanbul ignore next */ undefined;
+    }
+    getElementModelId(elementNode) {
+        return elementNode.extendedData ? elementNode.extendedData.modelId : /* istanbul ignore next */ undefined;
+    }
+    getElementCategoryId(elementNode) {
+        return elementNode.extendedData ? elementNode.extendedData.categoryId : /* istanbul ignore next */ undefined;
+    }
+    async getSubjectModelIds(subjectIds) {
+        return (await Promise.all(subjectIds.map(async (id) => this._subjectModelIdsCache.getSubjectModelIds(id))))
+            .reduce((allModelIds, curr) => [...allModelIds, ...curr], []);
+    }
+    // istanbul ignore next
+    getAssemblyElementIds(assemblyId) {
+        return this._elementIdsCache.getAssemblyElementIds(assemblyId);
+    }
+    // istanbul ignore next
+    async getGroupedElementIds(groupingNodeKey) {
+        return this._elementIdsCache.getGroupedElementIds(groupingNodeKey);
+    }
+}
+exports.ModelsVisibilityHandler = ModelsVisibilityHandler;
+/** @internal */
+class SubjectModelIdsCache {
+    constructor(imodel) {
+        this._imodel = imodel;
+    }
+    async initSubjectModels() {
+        var _a;
+        const querySubjects = () => {
+            const subjectsQuery = `
+        SELECT ECInstanceId id, Parent.Id parentId, json_extract(JsonProperties, '$.Subject.Model.TargetPartition') targetPartitionId
+        FROM bis.Subject
+      `;
+            return this._imodel.query(subjectsQuery, undefined, { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames });
+        };
+        const queryModels = () => {
+            const modelsQuery = `
+        SELECT p.ECInstanceId id, p.Parent.Id parentId
+        FROM bis.InformationPartitionElement p
+        INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId
+        WHERE NOT m.IsPrivate
+      `;
+            return this._imodel.query(modelsQuery, undefined, { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames });
+        };
+        function pushToMap(map, key, value) {
+            let list = map.get(key);
+            if (!list) {
+                list = [];
+                map.set(key, list);
+            }
+            list.push(value);
+        }
+        this._subjectsHierarchy = new Map();
+        const targetPartitionSubjects = new Map();
+        for await (const subject of querySubjects()) {
+            // istanbul ignore else
+            if (subject.parentId)
+                pushToMap(this._subjectsHierarchy, subject.parentId, subject.id);
+            // istanbul ignore if
+            if (subject.targetPartitionId)
+                pushToMap(targetPartitionSubjects, subject.targetPartitionId, subject.id);
+        }
+        this._subjectModels = new Map();
+        for await (const model of queryModels()) {
+            // istanbul ignore next
+            const subjectIds = (_a = targetPartitionSubjects.get(model.id)) !== null && _a !== void 0 ? _a : [];
+            // istanbul ignore else
+            if (!subjectIds.includes(model.parentId))
+                subjectIds.push(model.parentId);
+            subjectIds.forEach((subjectId) => {
+                pushToMap(this._subjectModels, subjectId, model.id);
+            });
+        }
+    }
+    async initCache() {
+        if (!this._init) {
+            this._init = this.initSubjectModels().then(() => { });
+        }
+        return this._init;
+    }
+    appendSubjectModelsRecursively(modelIds, subjectId) {
+        const subjectModelIds = this._subjectModels.get(subjectId);
+        if (subjectModelIds)
+            modelIds.push(...subjectModelIds);
+        const childSubjectIds = this._subjectsHierarchy.get(subjectId);
+        if (childSubjectIds)
+            childSubjectIds.forEach((cs) => this.appendSubjectModelsRecursively(modelIds, cs));
+    }
+    async getSubjectModelIds(subjectId) {
+        await this.initCache();
+        const modelIds = new Array();
+        this.appendSubjectModelsRecursively(modelIds, subjectId);
+        return modelIds;
+    }
+}
+exports.SubjectModelIdsCache = SubjectModelIdsCache;
+// istanbul ignore next
+class ElementIdsCache {
+    constructor(_imodel, _rulesetId) {
+        this._imodel = _imodel;
+        this._rulesetId = _rulesetId;
+        this._assemblyElementIdsCache = new Map();
+        this._groupedElementIdsCache = new Map();
+    }
+    clear() {
+        this._assemblyElementIdsCache.clear();
+        this._groupedElementIdsCache.clear();
+    }
+    getAssemblyElementIds(assemblyId) {
+        const ids = this._assemblyElementIdsCache.get(assemblyId);
+        if (ids)
+            return ids;
+        const container = createAssemblyElementIdsContainer(this._imodel, this._rulesetId, assemblyId);
+        this._assemblyElementIdsCache.set(assemblyId, container);
+        return container;
+    }
+    async getGroupedElementIds(groupingNodeKey) {
+        const keyString = JSON.stringify(groupingNodeKey);
+        const ids = this._groupedElementIdsCache.get(keyString);
+        if (ids)
+            return ids;
+        const info = await createGroupedElementsInfo(this._imodel, this._rulesetId, groupingNodeKey);
+        this._groupedElementIdsCache.set(keyString, info);
+        return info;
+    }
+}
+// istanbul ignore next
+async function* createInstanceIdsGenerator(imodel, rulesetId, displayType, inputKeys) {
+    const res = await presentation_frontend_1.Presentation.presentation.getContentInstanceKeys({
+        imodel,
+        rulesetOrId: rulesetId,
+        displayType,
+        keys: new presentation_common_1.KeySet(inputKeys),
+    });
+    for await (const key of res.items()) {
+        yield key.id;
+    }
+}
+// istanbul ignore next
+function createAssemblyElementIdsContainer(imodel, rulesetId, assemblyId) {
+    return new Utils_1.CachingElementIdsContainer(createInstanceIdsGenerator(imodel, rulesetId, "AssemblyElementsRequest", [{ className: "BisCore:Element", id: assemblyId }]));
+}
+// istanbul ignore next
+async function createGroupedElementsInfo(imodel, rulesetId, groupingNodeKey) {
+    const groupedElementIdsContainer = new Utils_1.CachingElementIdsContainer(createInstanceIdsGenerator(imodel, rulesetId, "AssemblyElementsRequest", [groupingNodeKey]));
+    const elementId = await groupedElementIdsContainer.getElementIds().next();
+    if (elementId.done)
+        throw new Error("Invalid grouping node key");
+    let modelId, categoryId;
+    const query = `SELECT Model.Id AS modelId, Category.Id AS categoryId FROM bis.GeometricElement3d WHERE ECInstanceId = ? LIMIT 1`;
+    for await (const modelAndCategoryIds of imodel.query(query, core_common_1.QueryBinder.from([elementId.value]), { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames })) {
+        modelId = modelAndCategoryIds.modelId;
+        categoryId = modelAndCategoryIds.categoryId;
+        break;
+    }
+    return { modelId, categoryId, elementIds: groupedElementIdsContainer };
+}
+const createTooltip = (status, tooltipStringId) => {
+    const statusStringId = `UiFramework:modelTree.status.${status}`;
+    const statusString = UiFramework_1.UiFramework.localization.getLocalizedString(statusStringId);
+    if (!tooltipStringId)
+        return statusString;
+    tooltipStringId = `UiFramework:modelTree.tooltips.${tooltipStringId}`;
+    const tooltipString = UiFramework_1.UiFramework.localization.getLocalizedString(tooltipStringId);
+    return `${statusString}: ${tooltipString}`;
+};
+//# sourceMappingURL=ModelsVisibilityHandler.js.map

package/lib/cjs/appui-react/imodel-components/models-tree/ModelsVisibilityHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelsVisibilityHandler.js","sourceRoot":"","sources":["../../../../../src/appui-react/imodel-components/models-tree/ModelsVisibilityHandler.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;AAKH,sDAA0D;AAC1D,oDAAiE;AACjE,wDAA8F;AAC9F,oEAA4G;AAE5G,wEAA4D;AAC5D,mDAAgD;AAEhD,mCAAqD;AAErD;;;;GAIG;AACH,IAAY,kBAOX;AAPD,WAAY,kBAAkB;IAC5B,iEAAO,CAAA;IACP,iEAAO,CAAA;IACP,6DAAK,CAAA;IACL,mEAAQ,CAAA;IACR,iEAAO,CAAA;IACP,mEAAQ,CAAA;AACV,CAAC,EAPW,kBAAkB,GAAlB,0BAAkB,KAAlB,0BAAkB,QAO7B;AAqBD;;;;GAIG;AACH,MAAa,uBAAuB;IAQlC,YAAY,KAAmC;;QAFvC,eAAU,GAAG,IAAI,KAAK,EAAc,CAAC;QAqBtC,uBAAkB,GAAG,IAAI,sBAAO,EAA4B,CAAC;QA8RpE,gEAAgE;QACxD,kBAAa,GAAG,CAAC,GAAa,EAAE,EAAE;YACxC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,gEAAgE;QACxD,gCAA2B,GAAG,GAAG,EAAE;YACzC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,gEAAgE;QACxD,+BAA0B,GAAG,GAAG,EAAE;YACxC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QA7TA,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,qBAAqB,GAAG,MAAA,KAAK,CAAC,oBAAoB,mCAAI,IAAI,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACjH,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAChG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,iCAAiC,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7G,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACrG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACjG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC,CAAC;QAC9G,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC,CAAC;QAC5G,IAAI,IAAI,CAAC,MAAM,CAAC,0BAA0B,EAAE;YAC1C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,oCAAY,CAAC,YAAY,CAAC,wBAAwB,CAAC,WAAW,CAAC,0BAA0B,CAAA,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;SACrJ;IACH,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9C,YAAY,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;IAID,sEAAsE;IAC/D,uBAAuB,CAAC,QAA2D,IAAI,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC,CAAC,CAAC;IAE/H,MAAM,CAAC,WAAW,CAAC,IAAkB,EAAE,YAA2C;QACvF,IAAI,6BAAO,CAAC,sBAAsB,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC/D,OAAO,kBAAkB,CAAC,QAAQ,CAAC;QAErC,IAAI,CAAC,IAAI,CAAC,YAAY;YACpB,OAAO,kBAAkB,CAAC,OAAO,CAAC;QAEpC,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAC1B,OAAO,kBAAkB,CAAC,OAAO,CAAC;QACpC,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YACxB,OAAO,kBAAkB,CAAC,KAAK,CAAC;QAClC,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC;YAC3B,OAAO,kBAAkB,CAAC,QAAQ,CAAC;QACrC,OAAO,kBAAkB,CAAC,OAAO,CAAC;IACpC,CAAC;IAEM,MAAM,CAAC,aAAa,CAAC,IAAkB;QAC5C,OAAO,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC;IAC1D,CAAC;IACM,MAAM,CAAC,WAAW,CAAC,IAAkB;QAC1C,OAAO,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IACxD,CAAC;IACM,MAAM,CAAC,cAAc,CAAC,IAAkB;QAC7C,OAAO,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC;IAC3D,CAAC;IAED,kDAAkD;IAC3C,mBAAmB,CAAC,IAAkB,EAAE,OAAgB;QAC7D,IAAI,6BAAO,CAAC,sBAAsB,CAAC,OAAO,CAAC;YACzC,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAEpE,IAAI,CAAC,6BAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC;YACtC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAE/C,IAAI,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE;YAC/C,4EAA4E;YAC5E,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;SACvF;QACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,IAAI,CAAC;YAC3C,OAAO,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAChE,IAAI,uBAAuB,CAAC,cAAc,CAAC,IAAI,CAAC;YAC9C,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC;QACxG,OAAO,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC;IACjI,CAAC;IAED,oEAAoE;IAC7D,KAAK,CAAC,gBAAgB,CAAC,IAAkB,EAAE,OAAgB,EAAE,EAAW;QAC7E,IAAI,6BAAO,CAAC,sBAAsB,CAAC,OAAO,CAAC,EAAE;YAC3C,MAAM,IAAI,CAAC,8BAA8B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACvD,OAAO;SACR;QAED,IAAI,CAAC,6BAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC;YACtC,OAAO;QAET,IAAI,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE;YAC/C,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;SACxF;aAAM,IAAI,uBAAuB,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;YACpD,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;SAC7D;aAAM,IAAI,uBAAuB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE;YACvD,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;SAC/F;aAAM;YACL,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;SAC9H;IACH,CAAC;IAES,KAAK,CAAC,wBAAwB,CAAC,GAAiB,EAAE,IAAkB;QAC5E,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;YAC5C,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,UAAU,EAAE,wBAAwB,CAAC,EAAE,CAAC;QAE7G,IAAI,IAAI,CAAC,qBAAqB;YAC5B,OAAO,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QAErF,OAAO,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,GAAiB;QACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;QACpD,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;QACxG,IAAI,WAAW;YACb,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,gCAAgC,CAAC,EAAE,CAAC;QACnG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,yBAAyB,CAAC,EAAE,CAAC;IAC1F,CAAC;IAEO,KAAK,CAAC,+BAA+B,CAAC,QAA+C,EAAE,GAAiB,EAAE,IAAkB;QAClI,IAAI,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAClC,OAAO,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,uBAAuB,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACpJ,IAAI,uBAAuB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC;YACtE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,gCAAgC,CAAC,EAAE,CAAC;QACnG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,yBAAyB,CAAC,EAAE,CAAC;IAC1F,CAAC;IAES,qBAAqB,CAAC,EAAc;QAC5C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;YAC5C,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,UAAU,EAAE,sBAAsB,CAAC,EAAE,CAAC;QAC3G,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC7D,OAAO,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC;IAC9H,CAAC;IAES,wBAAwB,CAAC,EAAc,EAAE,aAAqC;QACtF,IAAI,aAAa,EAAE;YACjB,IAAI,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC,KAAK,KAAK,QAAQ;gBAC9D,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,CAAC,UAAU,EAAE,4BAA4B,CAAC,EAAE,CAAC;YAEjH,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YAChG,QAAQ,QAAQ,EAAE;gBAChB,KAAK,0CAA0B,CAAC,QAAQ,CAAC,IAAI;oBAC3C,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,2CAA2C,CAAC,EAAE,CAAC;gBAC9G,KAAK,0CAA0B,CAAC,QAAQ,CAAC,IAAI;oBAC3C,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,wCAAwC,CAAC,EAAE,CAAC;aAC1G;SACF;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAChE,OAAO;YACL,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;YACzC,OAAO,EAAE,WAAW;gBAClB,CAAC,CAAC,aAAa,CAAC,SAAS,EAAE,2CAA2C,CAAC;gBACvE,CAAC,CAAC,aAAa,CAAC,QAAQ,EAAE,wCAAwC,CAAC;SACtE,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,mCAAmC,CAAC,GAAW,EAAE,GAA2B;QAC1F,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAEjF,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC5D,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,UAAU,EAAE,2BAA2B,CAAC,EAAE,CAAC;QAEhH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;YAC/F,IAAI,+BAA+B,GAAG,KAAK,CAAC;YAC5C,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,UAAU,CAAC,aAAa,EAAE,EAAE;gBACxD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;oBACnD,+BAA+B,GAAG,IAAI,CAAC;oBACvC,MAAM;iBACP;aACF;YACD,IAAI,+BAA+B;gBACjC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,yCAAyC,CAAC,EAAE,CAAC;SAC7G;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB;YAC9I,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,wDAAwD,CAAC,EAAE,CAAC;QAEzH,uBAAuB;QACvB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,EAAE;YAC7F,IAAI,sBAAsB,GAAG,IAAI,CAAC;YAClC,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,UAAU,CAAC,aAAa,EAAE,EAAE;gBACxD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;oBACnD,sBAAsB,GAAG,KAAK,CAAC;oBAC/B,MAAM;iBACP;aACF;YACD,IAAI,sBAAsB;gBACxB,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,qCAAqC,CAAC,EAAE,CAAC;SACxG;QAED,IAAI,UAAU,IAAI,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS;YACtF,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC;QAE5E,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,+BAA+B,CAAC,EAAE,CAAC;IAChG,CAAC;IAES,uBAAuB,CAAC,SAAqB,EAAE,OAA+B,EAAE,UAAkC;QAC1H,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC5D,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,UAAU,EAAE,2BAA2B,CAAC,EAAE,CAAC;QAChH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC;YACjG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,qCAAqC,CAAC,EAAE,CAAC;QACtG,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS,EAAE;YAClD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC;gBACjD,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,yCAAyC,CAAC,EAAE,CAAC;YAC5G,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB;gBAC5F,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,wDAAwD,CAAC,EAAE,CAAC;SAC1H;QACD,IAAI,UAAU,IAAI,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS;YACtF,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC;QAC5E,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,QAAQ,EAAE,+BAA+B,CAAC,EAAE,CAAC;IAChG,CAAC;IAES,KAAK,CAAC,sBAAsB,CAAC,GAAiB,EAAE,IAAkB,EAAE,EAAW;QACvF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;YAC5C,OAAO;QAET,IAAI,IAAI,CAAC,qBAAqB;YAC5B,OAAO,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QAEpF,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,QAA+C,EAAE,GAAiB,EAAE,IAAkB,EAAE,EAAW;QAC1I,IAAI,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAClC,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9H,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,GAAiB,EAAE,EAAW;QAC7D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC;IAES,KAAK,CAAC,gBAAgB,CAAC,EAAc,EAAE,EAAW;QAC1D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,EAAE;YAC5C,OAAO;QAET,OAAO,IAAI,CAAC,sBAAsB,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;IAES,KAAK,CAAC,sBAAsB,CAAC,GAAiB,EAAE,OAAgB;QACxE,IAAI,OAAO;YACT,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;;YAEjD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC;IAES,mBAAmB,CAAC,UAAsB,EAAE,aAAqC,EAAE,EAAW;QACtG,IAAI,aAAa,EAAE;YACjB,MAAM,qBAAqB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAClF,MAAM,GAAG,GAAG,CAAC,EAAE,KAAK,qBAAqB,CAAC,CAAC,CAAC,CAAC,0CAA0B,CAAC,QAAQ,CAAC,IAAI;gBACnF,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,0CAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,0CAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC7F,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,WAAW,CAAC,aAAa,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;YAC5F,IAAI,GAAG,KAAK,0CAA0B,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,EAAE;gBAC1D,kFAAkF;gBAClF,4EAA4E;gBAC5E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;aACtE;YACD,OAAO;SACR;QACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAClF,CAAC;IAES,KAAK,CAAC,8BAA8B,CAAC,GAA2B,EAAE,EAAW;QACrF,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QACjF,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IACtF,CAAC;IAES,KAAK,CAAC,kBAAkB,CAAC,EAAc,EAAE,OAA+B,EAAE,UAAkC,EAAE,EAAW;QACjI,MAAM,iBAAiB,GAAG,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QACzD,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM,EAAE,CAAC;YACT,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,iBAAiB,CAAC,aAAa,EAAE;gBAC3D,MAAM,OAAO,CAAC;QAClB,CAAC;QACD,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;IACxE,CAAC;IAES,KAAK,CAAC,mBAAmB,CAAC,OAA+B,EAAE,UAAkC,EAAE,UAAsC,EAAE,EAAW;QAC1J,MAAM,oBAAoB,GAAG,OAAO,IAAI,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS;eAC1F,UAAU,IAAI,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC;QAC1F,MAAM,yCAAyC,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC;QACjL,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACvG,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAChE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,0BAA0B,CAAA,EAAE,CAChE,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,UAAU,EAAE;YACxC,IAAI,EAAE,EAAE;gBACN,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACjC,IAAI,CAAC,oBAAoB,IAAI,yCAAyC;oBACpE,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;aAClC;iBAAM;gBACL,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAClC,IAAI,oBAAoB,IAAI,CAAC,yCAAyC;oBACpE,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;aACjC;SACF;QACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IACpG,CAAC;IAEO,0BAA0B;QAChC,IAAI,IAAI,CAAC,wBAAwB;YAC/B,OAAO;QAET,IAAI,CAAC,wBAAwB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9C,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAC;QAC5C,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAiBO,wBAAwB,CAAC,YAA0B;QACzD,OAAO,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,SAAS,CAAC;IAC9G,CAAC;IAEO,iBAAiB,CAAC,WAAyB;QACjD,OAAO,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,SAAS,CAAC;IAC5G,CAAC;IAEO,oBAAoB,CAAC,WAAyB;QACpD,OAAO,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,0BAA0B,CAAC,SAAS,CAAC;IAC/G,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,UAAwB;QACvD,OAAO,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;aACxG,MAAM,CAAC,CAAC,WAAyB,EAAE,IAAkB,EAAE,EAAE,CAAC,CAAC,GAAG,WAAW,EAAE,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;IAC9F,CAAC;IAED,uBAAuB;IACf,qBAAqB,CAAC,UAAsB;QAClD,OAAO,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IAED,uBAAuB;IACf,KAAK,CAAC,oBAAoB,CAAC,eAAgC;QACjE,OAAO,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;IACrE,CAAC;CACF;AAlWD,0DAkWC;AAED,gBAAgB;AAChB,MAAa,oBAAoB;IAM/B,YAAY,MAAwB;QAClC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,iBAAiB;;QAC7B,MAAM,aAAa,GAAG,GAAqG,EAAE;YAC3H,MAAM,aAAa,GAAG;;;OAGrB,CAAC;YACF,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACxG,CAAC,CAAC;QACF,MAAM,WAAW,GAAG,GAAoE,EAAE;YACxF,MAAM,WAAW,GAAG;;;;;OAKnB,CAAC;YACF,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACtG,CAAC,CAAC;QAEF,SAAS,SAAS,CAAe,GAAwB,EAAE,GAAS,EAAE,KAAa;YACjF,IAAI,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,IAAI,EAAE;gBACT,IAAI,GAAG,EAAE,CAAC;gBACV,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;aACpB;YACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAC;QACpC,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAA4B,CAAC;QACpE,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,aAAa,EAAE,EAAE;YAC3C,uBAAuB;YACvB,IAAI,OAAO,CAAC,QAAQ;gBAClB,SAAS,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YACnE,qBAAqB;YACrB,IAAI,OAAO,CAAC,iBAAiB;gBAC3B,SAAS,CAAC,uBAAuB,EAAE,OAAO,CAAC,iBAAiB,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;SAC7E;QAED,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,EAAE;YACvC,uBAAuB;YACvB,MAAM,UAAU,GAAG,MAAA,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,mCAAI,EAAE,CAAC;YAC/D,uBAAuB;YACvB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACtC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAElC,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBAC/B,SAAS,CAAC,IAAI,CAAC,cAAe,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;SACvD;QACD,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEO,8BAA8B,CAAC,QAAsB,EAAE,SAAqB;QAClF,MAAM,eAAe,GAAG,IAAI,CAAC,cAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC5D,IAAI,eAAe;YACjB,QAAQ,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QAEpC,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAChE,IAAI,eAAe;YACjB,eAAe,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,8BAA8B,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;IACvF,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,SAAqB;QACnD,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,IAAI,CAAC,8BAA8B,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACzD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AArFD,oDAqFC;AAQD,uBAAuB;AACvB,MAAM,eAAe;IAInB,YAAoB,OAAyB,EAAU,UAAkB;QAArD,YAAO,GAAP,OAAO,CAAkB;QAAU,eAAU,GAAV,UAAU,CAAQ;QAHjE,6BAAwB,GAAG,IAAI,GAAG,EAAsC,CAAC;QACzE,4BAAuB,GAAG,IAAI,GAAG,EAA6B,CAAC;IAGvE,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,CAAC;QACtC,IAAI,CAAC,uBAAuB,CAAC,KAAK,EAAE,CAAC;IACvC,CAAC;IAEM,qBAAqB,CAAC,UAAsB;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1D,IAAI,GAAG;YACL,OAAO,GAAG,CAAC;QAEb,MAAM,SAAS,GAAG,iCAAiC,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAC/F,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QACzD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,eAAgC;QAChE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACxD,IAAI,GAAG;YACL,OAAO,GAAG,CAAC;QACb,MAAM,IAAI,GAAG,MAAM,yBAAyB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;QAC7F,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAClD,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,uBAAuB;AACvB,KAAK,SAAS,CAAC,CAAC,0BAA0B,CAAC,MAAwB,EAAE,SAAiB,EAAE,WAAmB,EAAE,SAAe;IAC1H,MAAM,GAAG,GAAG,MAAM,oCAAY,CAAC,YAAY,CAAC,sBAAsB,CAAC;QACjE,MAAM;QACN,WAAW,EAAE,SAAS;QACtB,WAAW;QACX,IAAI,EAAE,IAAI,4BAAM,CAAC,SAAS,CAAC;KAC5B,CAAC,CAAC;IACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,EAAE,EAAE;QACnC,MAAM,GAAG,CAAC,EAAE,CAAC;KACd;AACH,CAAC;AAED,uBAAuB;AACvB,SAAS,iCAAiC,CAAC,MAAwB,EAAE,SAAiB,EAAE,UAAsB;IAC5G,OAAO,IAAI,kCAA0B,CAAC,0BAA0B,CAAC,MAAM,EAAE,SAAS,EAAE,yBAAyB,EAAE,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;AACtK,CAAC;AAED,uBAAuB;AACvB,KAAK,UAAU,yBAAyB,CAAC,MAAwB,EAAE,SAAiB,EAAE,eAAgC;IACpH,MAAM,0BAA0B,GAAG,IAAI,kCAA0B,CAAC,0BAA0B,CAAC,MAAM,EAAE,SAAS,EAAE,yBAAyB,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC/J,MAAM,SAAS,GAAG,MAAM,0BAA0B,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,CAAC;IAC1E,IAAI,SAAS,CAAC,IAAI;QAChB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAE/C,IAAI,OAAO,EAAE,UAAU,CAAC;IACxB,MAAM,KAAK,GAAG,kHAAkH,CAAC;IACjI,IAAI,KAAK,EAAE,MAAM,mBAAmB,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,yBAAW,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,EAAE;QAClJ,OAAO,GAAG,mBAAmB,CAAC,OAAO,CAAC;QACtC,UAAU,GAAG,mBAAmB,CAAC,UAAU,CAAC;QAC5C,MAAM;KACP;IACD,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,0BAA0B,EAAE,CAAC;AACzE,CAAC;AAED,MAAM,aAAa,GAAG,CAAC,MAAyC,EAAE,eAAmC,EAAU,EAAE;IAC/G,MAAM,cAAc,GAAG,gCAAgC,MAAM,EAAE,CAAC;IAChE,MAAM,YAAY,GAAG,yBAAW,CAAC,YAAY,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;IACjF,IAAI,CAAC,eAAe;QAClB,OAAO,YAAY,CAAC;IAEtB,eAAe,GAAG,kCAAkC,eAAe,EAAE,CAAC;IACtE,MAAM,aAAa,GAAG,yBAAW,CAAC,YAAY,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC;IACnF,OAAO,GAAG,YAAY,KAAK,aAAa,EAAE,CAAC;AAC7C,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module IModelComponents\r\n */\r\n\r\n/* eslint-disable deprecation/deprecation */\r\n\r\nimport { TreeNodeItem } from \"@itwin/components-react\";\r\nimport { BeEvent, Id64String } from \"@itwin/core-bentley\";\r\nimport { QueryBinder, QueryRowFormat } from \"@itwin/core-common\";\r\nimport { IModelConnection, PerModelCategoryVisibility, Viewport } from \"@itwin/core-frontend\";\r\nimport { ECClassGroupingNodeKey, GroupingNodeKey, Keys, KeySet, NodeKey } from \"@itwin/presentation-common\";\r\nimport { IFilteredPresentationTreeDataProvider, IPresentationTreeDataProvider } from \"@itwin/presentation-components\";\r\nimport { Presentation } from \"@itwin/presentation-frontend\";\r\nimport { UiFramework } from \"../../UiFramework\";\r\nimport { IVisibilityHandler, VisibilityChangeListener, VisibilityStatus } from \"../VisibilityTreeEventHandler\";\r\nimport { CachingElementIdsContainer } from \"./Utils\";\r\n\r\n/**\r\n * Visibility tree node types.\r\n * @beta\r\n * @deprecated in 3.6. Was moved to `@itwin/tree-widget-react` package.\r\n */\r\nexport enum ModelsTreeNodeType {\r\n  Unknown,\r\n  Subject,\r\n  Model,\r\n  Category,\r\n  Element,\r\n  Grouping,\r\n}\r\n\r\n/**\r\n * Type definition of predicate used to decide if node can be selected\r\n * @beta\r\n * @deprecated in 3.6. Was moved to `@itwin/tree-widget-react` package.\r\n */\r\nexport type ModelsTreeSelectionPredicate = (key: NodeKey, type: ModelsTreeNodeType) => boolean;\r\n\r\n/**\r\n * Props for [[ModelsVisibilityHandler]]\r\n * @alpha\r\n */\r\nexport interface ModelsVisibilityHandlerProps {\r\n  rulesetId: string;\r\n  viewport: Viewport;\r\n  hierarchyAutoUpdateEnabled?: boolean;\r\n  /** @internal */\r\n  subjectModelIdsCache?: SubjectModelIdsCache;\r\n}\r\n\r\n/**\r\n * Visibility handler used by [[ModelsTree]] to control visibility of the tree items.\r\n * @alpha\r\n * @deprecated in 3.6. Was moved to `@itwin/tree-widget-react` package.\r\n */\r\nexport class ModelsVisibilityHandler implements IVisibilityHandler {\r\n  private _props: ModelsVisibilityHandlerProps;\r\n  private _pendingVisibilityChange: any | undefined;\r\n  private _subjectModelIdsCache: SubjectModelIdsCache;\r\n  private _filteredDataProvider?: IFilteredPresentationTreeDataProvider;\r\n  private _elementIdsCache: ElementIdsCache;\r\n  private _listeners = new Array<() => void>();\r\n\r\n  constructor(props: ModelsVisibilityHandlerProps) {\r\n    this._props = props;\r\n    this._subjectModelIdsCache = props.subjectModelIdsCache ?? new SubjectModelIdsCache(this._props.viewport.iModel);\r\n    this._elementIdsCache = new ElementIdsCache(this._props.viewport.iModel, this._props.rulesetId);\r\n    this._listeners.push(this._props.viewport.onViewedCategoriesPerModelChanged.addListener(this.onViewChanged));\r\n    this._listeners.push(this._props.viewport.onViewedCategoriesChanged.addListener(this.onViewChanged));\r\n    this._listeners.push(this._props.viewport.onViewedModelsChanged.addListener(this.onViewChanged));\r\n    this._listeners.push(this._props.viewport.onAlwaysDrawnChanged.addListener(this.onElementAlwaysDrawnChanged));\r\n    this._listeners.push(this._props.viewport.onNeverDrawnChanged.addListener(this.onElementNeverDrawnChanged));\r\n    if (this._props.hierarchyAutoUpdateEnabled) {\r\n      this._listeners.push(Presentation.presentation.onIModelHierarchyChanged.addListener(/* istanbul ignore next */() => this._elementIdsCache.clear()));\r\n    }\r\n  }\r\n\r\n  public dispose() {\r\n    this._listeners.forEach((remove) => remove());\r\n    clearTimeout(this._pendingVisibilityChange);\r\n  }\r\n\r\n  public onVisibilityChange = new BeEvent<VisibilityChangeListener>();\r\n\r\n  /** Sets data provider that is used to get filtered tree hierarchy. */\r\n  public setFilteredDataProvider(provider: IFilteredPresentationTreeDataProvider | undefined) { this._filteredDataProvider = provider; }\r\n\r\n  public static getNodeType(item: TreeNodeItem, dataProvider: IPresentationTreeDataProvider) {\r\n    if (NodeKey.isClassGroupingNodeKey(dataProvider.getNodeKey(item)))\r\n      return ModelsTreeNodeType.Grouping;\r\n\r\n    if (!item.extendedData)\r\n      return ModelsTreeNodeType.Unknown;\r\n\r\n    if (this.isSubjectNode(item))\r\n      return ModelsTreeNodeType.Subject;\r\n    if (this.isModelNode(item))\r\n      return ModelsTreeNodeType.Model;\r\n    if (this.isCategoryNode(item))\r\n      return ModelsTreeNodeType.Category;\r\n    return ModelsTreeNodeType.Element;\r\n  }\r\n\r\n  public static isSubjectNode(node: TreeNodeItem) {\r\n    return node.extendedData && node.extendedData.isSubject;\r\n  }\r\n  public static isModelNode(node: TreeNodeItem) {\r\n    return node.extendedData && node.extendedData.isModel;\r\n  }\r\n  public static isCategoryNode(node: TreeNodeItem) {\r\n    return node.extendedData && node.extendedData.isCategory;\r\n  }\r\n\r\n  /** Returns visibility status of the tree node. */\r\n  public getVisibilityStatus(node: TreeNodeItem, nodeKey: NodeKey): VisibilityStatus | Promise<VisibilityStatus> {\r\n    if (NodeKey.isClassGroupingNodeKey(nodeKey))\r\n      return this.getElementGroupingNodeDisplayStatus(node.id, nodeKey);\r\n\r\n    if (!NodeKey.isInstancesNodeKey(nodeKey))\r\n      return { state: \"hidden\", isDisabled: true };\r\n\r\n    if (ModelsVisibilityHandler.isSubjectNode(node)) {\r\n      // note: subject nodes may be merged to represent multiple subject instances\r\n      return this.getSubjectNodeVisibility(nodeKey.instanceKeys.map((key) => key.id), node);\r\n    }\r\n    if (ModelsVisibilityHandler.isModelNode(node))\r\n      return this.getModelDisplayStatus(nodeKey.instanceKeys[0].id);\r\n    if (ModelsVisibilityHandler.isCategoryNode(node))\r\n      return this.getCategoryDisplayStatus(nodeKey.instanceKeys[0].id, this.getCategoryParentModelId(node));\r\n    return this.getElementDisplayStatus(nodeKey.instanceKeys[0].id, this.getElementModelId(node), this.getElementCategoryId(node));\r\n  }\r\n\r\n  /** Changes visibility of the items represented by the tree node. */\r\n  public async changeVisibility(node: TreeNodeItem, nodeKey: NodeKey, on: boolean) {\r\n    if (NodeKey.isClassGroupingNodeKey(nodeKey)) {\r\n      await this.changeElementGroupingNodeState(nodeKey, on);\r\n      return;\r\n    }\r\n\r\n    if (!NodeKey.isInstancesNodeKey(nodeKey))\r\n      return;\r\n\r\n    if (ModelsVisibilityHandler.isSubjectNode(node)) {\r\n      await this.changeSubjectNodeState(nodeKey.instanceKeys.map((key) => key.id), node, on);\r\n    } else if (ModelsVisibilityHandler.isModelNode(node)) {\r\n      await this.changeModelState(nodeKey.instanceKeys[0].id, on);\r\n    } else if (ModelsVisibilityHandler.isCategoryNode(node)) {\r\n      this.changeCategoryState(nodeKey.instanceKeys[0].id, this.getCategoryParentModelId(node), on);\r\n    } else {\r\n      await this.changeElementState(nodeKey.instanceKeys[0].id, this.getElementModelId(node), this.getElementCategoryId(node), on);\r\n    }\r\n  }\r\n\r\n  protected async getSubjectNodeVisibility(ids: Id64String[], node: TreeNodeItem): Promise<VisibilityStatus> {\r\n    if (!this._props.viewport.view.isSpatialView())\r\n      return { isDisabled: true, state: \"hidden\", tooltip: createTooltip(\"disabled\", \"subject.nonSpatialView\") };\r\n\r\n    if (this._filteredDataProvider)\r\n      return this.getFilteredSubjectDisplayStatus(this._filteredDataProvider, ids, node);\r\n\r\n    return this.getSubjectDisplayStatus(ids);\r\n  }\r\n\r\n  private async getSubjectDisplayStatus(ids: Id64String[]): Promise<VisibilityStatus> {\r\n    const modelIds = await this.getSubjectModelIds(ids);\r\n    const isDisplayed = modelIds.some((modelId) => this.getModelDisplayStatus(modelId).state === \"visible\");\r\n    if (isDisplayed)\r\n      return { state: \"visible\", tooltip: createTooltip(\"visible\", \"subject.atLeastOneModelVisible\") };\r\n    return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"subject.allModelsHidden\") };\r\n  }\r\n\r\n  private async getFilteredSubjectDisplayStatus(provider: IFilteredPresentationTreeDataProvider, ids: Id64String[], node: TreeNodeItem): Promise<VisibilityStatus> {\r\n    if (provider.nodeMatchesFilter(node))\r\n      return this.getSubjectDisplayStatus(ids);\r\n\r\n    const children = await provider.getNodes(node);\r\n    const childrenDisplayStatuses = await Promise.all(children.map((childNode) => this.getVisibilityStatus(childNode, provider.getNodeKey(childNode))));\r\n    if (childrenDisplayStatuses.some((status) => status.state === \"visible\"))\r\n      return { state: \"visible\", tooltip: createTooltip(\"visible\", \"subject.atLeastOneModelVisible\") };\r\n    return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"subject.allModelsHidden\") };\r\n  }\r\n\r\n  protected getModelDisplayStatus(id: Id64String): VisibilityStatus {\r\n    if (!this._props.viewport.view.isSpatialView())\r\n      return { isDisabled: true, state: \"hidden\", tooltip: createTooltip(\"disabled\", \"model.nonSpatialView\") };\r\n    const isDisplayed = this._props.viewport.view.viewsModel(id);\r\n    return { state: isDisplayed ? \"visible\" : \"hidden\", tooltip: createTooltip(isDisplayed ? \"visible\" : \"hidden\", undefined) };\r\n  }\r\n\r\n  protected getCategoryDisplayStatus(id: Id64String, parentModelId: Id64String | undefined): VisibilityStatus {\r\n    if (parentModelId) {\r\n      if (this.getModelDisplayStatus(parentModelId).state === \"hidden\")\r\n        return { state: \"hidden\", isDisabled: true, tooltip: createTooltip(\"disabled\", \"category.modelNotDisplayed\") };\r\n\r\n      const override = this._props.viewport.perModelCategoryVisibility.getOverride(parentModelId, id);\r\n      switch (override) {\r\n        case PerModelCategoryVisibility.Override.Show:\r\n          return { state: \"visible\", tooltip: createTooltip(\"visible\", \"category.displayedThroughPerModelOverride\") };\r\n        case PerModelCategoryVisibility.Override.Hide:\r\n          return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"category.hiddenThroughPerModelOverride\") };\r\n      }\r\n    }\r\n    const isDisplayed = this._props.viewport.view.viewsCategory(id);\r\n    return {\r\n      state: isDisplayed ? \"visible\" : \"hidden\",\r\n      tooltip: isDisplayed\r\n        ? createTooltip(\"visible\", \"category.displayedThroughCategorySelector\")\r\n        : createTooltip(\"hidden\", \"category.hiddenThroughCategorySelector\"),\r\n    };\r\n  }\r\n\r\n  protected async getElementGroupingNodeDisplayStatus(_id: string, key: ECClassGroupingNodeKey): Promise<VisibilityStatus> {\r\n    const { modelId, categoryId, elementIds } = await this.getGroupedElementIds(key);\r\n\r\n    if (!modelId || !this._props.viewport.view.viewsModel(modelId))\r\n      return { isDisabled: true, state: \"hidden\", tooltip: createTooltip(\"disabled\", \"element.modelNotDisplayed\") };\r\n\r\n    if (this._props.viewport.alwaysDrawn !== undefined && this._props.viewport.alwaysDrawn.size > 0) {\r\n      let atLeastOneElementForceDisplayed = false;\r\n      for await (const elementId of elementIds.getElementIds()) {\r\n        if (this._props.viewport.alwaysDrawn.has(elementId)) {\r\n          atLeastOneElementForceDisplayed = true;\r\n          break;\r\n        }\r\n      }\r\n      if (atLeastOneElementForceDisplayed)\r\n        return { state: \"visible\", tooltip: createTooltip(\"visible\", \"element.displayedThroughAlwaysDrawnList\") };\r\n    }\r\n\r\n    if (this._props.viewport.alwaysDrawn !== undefined && this._props.viewport.alwaysDrawn.size !== 0 && this._props.viewport.isAlwaysDrawnExclusive)\r\n      return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn\") };\r\n\r\n    // istanbul ignore else\r\n    if (this._props.viewport.neverDrawn !== undefined && this._props.viewport.neverDrawn.size > 0) {\r\n      let allElementsForceHidden = true;\r\n      for await (const elementId of elementIds.getElementIds()) {\r\n        if (!this._props.viewport.neverDrawn.has(elementId)) {\r\n          allElementsForceHidden = false;\r\n          break;\r\n        }\r\n      }\r\n      if (allElementsForceHidden)\r\n        return { state: \"hidden\", tooltip: createTooltip(\"visible\", \"element.hiddenThroughNeverDrawnList\") };\r\n    }\r\n\r\n    if (categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === \"visible\")\r\n      return { state: \"visible\", tooltip: createTooltip(\"visible\", undefined) };\r\n\r\n    return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"element.hiddenThroughCategory\") };\r\n  }\r\n\r\n  protected getElementDisplayStatus(elementId: Id64String, modelId: Id64String | undefined, categoryId: Id64String | undefined): VisibilityStatus {\r\n    if (!modelId || !this._props.viewport.view.viewsModel(modelId))\r\n      return { isDisabled: true, state: \"hidden\", tooltip: createTooltip(\"disabled\", \"element.modelNotDisplayed\") };\r\n    if (this._props.viewport.neverDrawn !== undefined && this._props.viewport.neverDrawn.has(elementId))\r\n      return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"element.hiddenThroughNeverDrawnList\") };\r\n    if (this._props.viewport.alwaysDrawn !== undefined) {\r\n      if (this._props.viewport.alwaysDrawn.has(elementId))\r\n        return { state: \"visible\", tooltip: createTooltip(\"visible\", \"element.displayedThroughAlwaysDrawnList\") };\r\n      if (this._props.viewport.alwaysDrawn.size !== 0 && this._props.viewport.isAlwaysDrawnExclusive)\r\n        return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn\") };\r\n    }\r\n    if (categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === \"visible\")\r\n      return { state: \"visible\", tooltip: createTooltip(\"visible\", undefined) };\r\n    return { state: \"hidden\", tooltip: createTooltip(\"hidden\", \"element.hiddenThroughCategory\") };\r\n  }\r\n\r\n  protected async changeSubjectNodeState(ids: Id64String[], node: TreeNodeItem, on: boolean) {\r\n    if (!this._props.viewport.view.isSpatialView())\r\n      return;\r\n\r\n    if (this._filteredDataProvider)\r\n      return this.changeFilteredSubjectState(this._filteredDataProvider, ids, node, on);\r\n\r\n    return this.changeSubjectState(ids, on);\r\n  }\r\n\r\n  private async changeFilteredSubjectState(provider: IFilteredPresentationTreeDataProvider, ids: Id64String[], node: TreeNodeItem, on: boolean) {\r\n    if (provider.nodeMatchesFilter(node))\r\n      return this.changeSubjectState(ids, on);\r\n\r\n    const children = await provider.getNodes(node);\r\n    return Promise.all(children.map(async (childNode) => this.changeVisibility(childNode, provider.getNodeKey(childNode), on)));\r\n  }\r\n\r\n  private async changeSubjectState(ids: Id64String[], on: boolean) {\r\n    const modelIds = await this.getSubjectModelIds(ids);\r\n    return this.changeModelsVisibility(modelIds, on);\r\n  }\r\n\r\n  protected async changeModelState(id: Id64String, on: boolean) {\r\n    if (!this._props.viewport.view.isSpatialView())\r\n      return;\r\n\r\n    return this.changeModelsVisibility([id], on);\r\n  }\r\n\r\n  protected async changeModelsVisibility(ids: Id64String[], visible: boolean) {\r\n    if (visible)\r\n      return this._props.viewport.addViewedModels(ids);\r\n    else\r\n      this._props.viewport.changeModelDisplay(ids, false);\r\n  }\r\n\r\n  protected changeCategoryState(categoryId: Id64String, parentModelId: Id64String | undefined, on: boolean) {\r\n    if (parentModelId) {\r\n      const isDisplayedInSelector = this._props.viewport.view.viewsCategory(categoryId);\r\n      const ovr = (on === isDisplayedInSelector) ? PerModelCategoryVisibility.Override.None\r\n        : on ? PerModelCategoryVisibility.Override.Show : PerModelCategoryVisibility.Override.Hide;\r\n      this._props.viewport.perModelCategoryVisibility.setOverride(parentModelId, categoryId, ovr);\r\n      if (ovr === PerModelCategoryVisibility.Override.None && on) {\r\n        // we took off the override which means the category is displayed in selector, but\r\n        // doesn't mean all its subcategories are displayed - this call ensures that\r\n        this._props.viewport.changeCategoryDisplay([categoryId], true, true);\r\n      }\r\n      return;\r\n    }\r\n    this._props.viewport.changeCategoryDisplay([categoryId], on, on ? true : false);\r\n  }\r\n\r\n  protected async changeElementGroupingNodeState(key: ECClassGroupingNodeKey, on: boolean) {\r\n    const { modelId, categoryId, elementIds } = await this.getGroupedElementIds(key);\r\n    await this.changeElementsState(modelId, categoryId, elementIds.getElementIds(), on);\r\n  }\r\n\r\n  protected async changeElementState(id: Id64String, modelId: Id64String | undefined, categoryId: Id64String | undefined, on: boolean) {\r\n    const childIdsContainer = this.getAssemblyElementIds(id);\r\n    async function* elementIds() {\r\n      yield id;\r\n      for await (const childId of childIdsContainer.getElementIds())\r\n        yield childId;\r\n    }\r\n    await this.changeElementsState(modelId, categoryId, elementIds(), on);\r\n  }\r\n\r\n  protected async changeElementsState(modelId: Id64String | undefined, categoryId: Id64String | undefined, elementIds: AsyncGenerator<Id64String>, on: boolean) {\r\n    const isDisplayedByDefault = modelId && this.getModelDisplayStatus(modelId).state === \"visible\"\r\n      && categoryId && this.getCategoryDisplayStatus(categoryId, modelId).state === \"visible\";\r\n    const isHiddenDueToExclusiveAlwaysDrawnElements = this._props.viewport.isAlwaysDrawnExclusive && this._props.viewport.alwaysDrawn && 0 !== this._props.viewport.alwaysDrawn.size;\r\n    const currNeverDrawn = new Set(this._props.viewport.neverDrawn ? this._props.viewport.neverDrawn : []);\r\n    const currAlwaysDrawn = new Set(this._props.viewport.alwaysDrawn ?\r\n      this._props.viewport.alwaysDrawn : /* istanbul ignore next */[],\r\n    );\r\n    for await (const elementId of elementIds) {\r\n      if (on) {\r\n        currNeverDrawn.delete(elementId);\r\n        if (!isDisplayedByDefault || isHiddenDueToExclusiveAlwaysDrawnElements)\r\n          currAlwaysDrawn.add(elementId);\r\n      } else {\r\n        currAlwaysDrawn.delete(elementId);\r\n        if (isDisplayedByDefault && !isHiddenDueToExclusiveAlwaysDrawnElements)\r\n          currNeverDrawn.add(elementId);\r\n      }\r\n    }\r\n    this._props.viewport.setNeverDrawn(currNeverDrawn);\r\n    this._props.viewport.setAlwaysDrawn(currAlwaysDrawn, this._props.viewport.isAlwaysDrawnExclusive);\r\n  }\r\n\r\n  private onVisibilityChangeInternal() {\r\n    if (this._pendingVisibilityChange)\r\n      return;\r\n\r\n    this._pendingVisibilityChange = setTimeout(() => {\r\n      this.onVisibilityChange.raiseEvent();\r\n      this._pendingVisibilityChange = undefined;\r\n    }, 0);\r\n  }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  private onViewChanged = (_vp: Viewport) => {\r\n    this.onVisibilityChangeInternal();\r\n  };\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  private onElementAlwaysDrawnChanged = () => {\r\n    this.onVisibilityChangeInternal();\r\n  };\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  private onElementNeverDrawnChanged = () => {\r\n    this.onVisibilityChangeInternal();\r\n  };\r\n\r\n  private getCategoryParentModelId(categoryNode: TreeNodeItem): Id64String | undefined {\r\n    return categoryNode.extendedData ? categoryNode.extendedData.modelId : /* istanbul ignore next */ undefined;\r\n  }\r\n\r\n  private getElementModelId(elementNode: TreeNodeItem): Id64String | undefined {\r\n    return elementNode.extendedData ? elementNode.extendedData.modelId : /* istanbul ignore next */ undefined;\r\n  }\r\n\r\n  private getElementCategoryId(elementNode: TreeNodeItem): Id64String | undefined {\r\n    return elementNode.extendedData ? elementNode.extendedData.categoryId : /* istanbul ignore next */ undefined;\r\n  }\r\n\r\n  private async getSubjectModelIds(subjectIds: Id64String[]) {\r\n    return (await Promise.all(subjectIds.map(async (id) => this._subjectModelIdsCache.getSubjectModelIds(id))))\r\n      .reduce((allModelIds: Id64String[], curr: Id64String[]) => [...allModelIds, ...curr], []);\r\n  }\r\n\r\n  // istanbul ignore next\r\n  private getAssemblyElementIds(assemblyId: Id64String) {\r\n    return this._elementIdsCache.getAssemblyElementIds(assemblyId);\r\n  }\r\n\r\n  // istanbul ignore next\r\n  private async getGroupedElementIds(groupingNodeKey: GroupingNodeKey) {\r\n    return this._elementIdsCache.getGroupedElementIds(groupingNodeKey);\r\n  }\r\n}\r\n\r\n/** @internal */\r\nexport class SubjectModelIdsCache {\r\n  private _imodel: IModelConnection;\r\n  private _subjectsHierarchy: Map<Id64String, Id64String[]> | undefined;\r\n  private _subjectModels: Map<Id64String, Id64String[]> | undefined;\r\n  private _init: Promise<void> | undefined;\r\n\r\n  constructor(imodel: IModelConnection) {\r\n    this._imodel = imodel;\r\n  }\r\n\r\n  private async initSubjectModels() {\r\n    const querySubjects = (): AsyncIterableIterator<{ id: Id64String, parentId?: Id64String, targetPartitionId?: Id64String }> => {\r\n      const subjectsQuery = `\r\n        SELECT ECInstanceId id, Parent.Id parentId, json_extract(JsonProperties, '$.Subject.Model.TargetPartition') targetPartitionId\r\n        FROM bis.Subject\r\n      `;\r\n      return this._imodel.query(subjectsQuery, undefined, { rowFormat: QueryRowFormat.UseJsPropertyNames });\r\n    };\r\n    const queryModels = (): AsyncIterableIterator<{ id: Id64String, parentId: Id64String }> => {\r\n      const modelsQuery = `\r\n        SELECT p.ECInstanceId id, p.Parent.Id parentId\r\n        FROM bis.InformationPartitionElement p\r\n        INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\r\n        WHERE NOT m.IsPrivate\r\n      `;\r\n      return this._imodel.query(modelsQuery, undefined, { rowFormat: QueryRowFormat.UseJsPropertyNames });\r\n    };\r\n\r\n    function pushToMap<TKey, TValue>(map: Map<TKey, TValue[]>, key: TKey, value: TValue) {\r\n      let list = map.get(key);\r\n      if (!list) {\r\n        list = [];\r\n        map.set(key, list);\r\n      }\r\n      list.push(value);\r\n    }\r\n\r\n    this._subjectsHierarchy = new Map();\r\n    const targetPartitionSubjects = new Map<Id64String, Id64String[]>();\r\n    for await (const subject of querySubjects()) {\r\n      // istanbul ignore else\r\n      if (subject.parentId)\r\n        pushToMap(this._subjectsHierarchy, subject.parentId, subject.id);\r\n      // istanbul ignore if\r\n      if (subject.targetPartitionId)\r\n        pushToMap(targetPartitionSubjects, subject.targetPartitionId, subject.id);\r\n    }\r\n\r\n    this._subjectModels = new Map();\r\n    for await (const model of queryModels()) {\r\n      // istanbul ignore next\r\n      const subjectIds = targetPartitionSubjects.get(model.id) ?? [];\r\n      // istanbul ignore else\r\n      if (!subjectIds.includes(model.parentId))\r\n        subjectIds.push(model.parentId);\r\n\r\n      subjectIds.forEach((subjectId) => {\r\n        pushToMap(this._subjectModels!, subjectId, model.id);\r\n      });\r\n    }\r\n  }\r\n\r\n  private async initCache() {\r\n    if (!this._init) {\r\n      this._init = this.initSubjectModels().then(() => { });\r\n    }\r\n    return this._init;\r\n  }\r\n\r\n  private appendSubjectModelsRecursively(modelIds: Id64String[], subjectId: Id64String) {\r\n    const subjectModelIds = this._subjectModels!.get(subjectId);\r\n    if (subjectModelIds)\r\n      modelIds.push(...subjectModelIds);\r\n\r\n    const childSubjectIds = this._subjectsHierarchy!.get(subjectId);\r\n    if (childSubjectIds)\r\n      childSubjectIds.forEach((cs) => this.appendSubjectModelsRecursively(modelIds, cs));\r\n  }\r\n\r\n  public async getSubjectModelIds(subjectId: Id64String): Promise<Id64String[]> {\r\n    await this.initCache();\r\n    const modelIds = new Array<Id64String>();\r\n    this.appendSubjectModelsRecursively(modelIds, subjectId);\r\n    return modelIds;\r\n  }\r\n}\r\n\r\ninterface GroupedElementIds {\r\n  modelId?: string;\r\n  categoryId?: string;\r\n  elementIds: CachingElementIdsContainer;\r\n}\r\n\r\n// istanbul ignore next\r\nclass ElementIdsCache {\r\n  private _assemblyElementIdsCache = new Map<string, CachingElementIdsContainer>();\r\n  private _groupedElementIdsCache = new Map<string, GroupedElementIds>();\r\n\r\n  constructor(private _imodel: IModelConnection, private _rulesetId: string) {\r\n  }\r\n\r\n  public clear() {\r\n    this._assemblyElementIdsCache.clear();\r\n    this._groupedElementIdsCache.clear();\r\n  }\r\n\r\n  public getAssemblyElementIds(assemblyId: Id64String) {\r\n    const ids = this._assemblyElementIdsCache.get(assemblyId);\r\n    if (ids)\r\n      return ids;\r\n\r\n    const container = createAssemblyElementIdsContainer(this._imodel, this._rulesetId, assemblyId);\r\n    this._assemblyElementIdsCache.set(assemblyId, container);\r\n    return container;\r\n  }\r\n\r\n  public async getGroupedElementIds(groupingNodeKey: GroupingNodeKey): Promise<GroupedElementIds> {\r\n    const keyString = JSON.stringify(groupingNodeKey);\r\n    const ids = this._groupedElementIdsCache.get(keyString);\r\n    if (ids)\r\n      return ids;\r\n    const info = await createGroupedElementsInfo(this._imodel, this._rulesetId, groupingNodeKey);\r\n    this._groupedElementIdsCache.set(keyString, info);\r\n    return info;\r\n  }\r\n}\r\n\r\n// istanbul ignore next\r\nasync function* createInstanceIdsGenerator(imodel: IModelConnection, rulesetId: string, displayType: string, inputKeys: Keys) {\r\n  const res = await Presentation.presentation.getContentInstanceKeys({\r\n    imodel,\r\n    rulesetOrId: rulesetId,\r\n    displayType,\r\n    keys: new KeySet(inputKeys),\r\n  });\r\n  for await (const key of res.items()) {\r\n    yield key.id;\r\n  }\r\n}\r\n\r\n// istanbul ignore next\r\nfunction createAssemblyElementIdsContainer(imodel: IModelConnection, rulesetId: string, assemblyId: Id64String) {\r\n  return new CachingElementIdsContainer(createInstanceIdsGenerator(imodel, rulesetId, \"AssemblyElementsRequest\", [{ className: \"BisCore:Element\", id: assemblyId }]));\r\n}\r\n\r\n// istanbul ignore next\r\nasync function createGroupedElementsInfo(imodel: IModelConnection, rulesetId: string, groupingNodeKey: GroupingNodeKey) {\r\n  const groupedElementIdsContainer = new CachingElementIdsContainer(createInstanceIdsGenerator(imodel, rulesetId, \"AssemblyElementsRequest\", [groupingNodeKey]));\r\n  const elementId = await groupedElementIdsContainer.getElementIds().next();\r\n  if (elementId.done)\r\n    throw new Error(\"Invalid grouping node key\");\r\n\r\n  let modelId, categoryId;\r\n  const query = `SELECT Model.Id AS modelId, Category.Id AS categoryId FROM bis.GeometricElement3d WHERE ECInstanceId = ? LIMIT 1`;\r\n  for await (const modelAndCategoryIds of imodel.query(query, QueryBinder.from([elementId.value]), { rowFormat: QueryRowFormat.UseJsPropertyNames })) {\r\n    modelId = modelAndCategoryIds.modelId;\r\n    categoryId = modelAndCategoryIds.categoryId;\r\n    break;\r\n  }\r\n  return { modelId, categoryId, elementIds: groupedElementIdsContainer };\r\n}\r\n\r\nconst createTooltip = (status: \"visible\" | \"hidden\" | \"disabled\", tooltipStringId: string | undefined): string => {\r\n  const statusStringId = `UiFramework:modelTree.status.${status}`;\r\n  const statusString = UiFramework.localization.getLocalizedString(statusStringId);\r\n  if (!tooltipStringId)\r\n    return statusString;\r\n\r\n  tooltipStringId = `UiFramework:modelTree.tooltips.${tooltipStringId}`;\r\n  const tooltipString = UiFramework.localization.getLocalizedString(tooltipStringId);\r\n  return `${statusString}: ${tooltipString}`;\r\n};\r\n"]}

package/lib/cjs/appui-react/imodel-components/models-tree/Utils.d.ts

@@ -0,0 +1,10 @@
+import { Id64String } from "@itwin/core-bentley";
+/** @internal */
+export declare class CachingElementIdsContainer {
+    private _generator;
+    private _ids;
+    constructor(_generator: AsyncGenerator<Id64String>);
+    private next;
+    getElementIds(): AsyncGenerator<any, void, unknown>;
+}
+//# sourceMappingURL=Utils.d.ts.map

package/lib/cjs/appui-react/imodel-components/models-tree/Utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Utils.d.ts","sourceRoot":"","sources":["../../../../../src/appui-react/imodel-components/models-tree/Utils.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,gBAAgB;AAChB,qBAAa,0BAA0B;IAEzB,OAAO,CAAC,UAAU;IAD9B,OAAO,CAAC,IAAI,CAA2B;gBACnB,UAAU,EAAE,cAAc,CAAC,UAAU,CAAC;YAG5C,IAAI;IAEJ,aAAa;CAW5B"}

package/lib/cjs/appui-react/imodel-components/models-tree/Utils.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CachingElementIdsContainer = void 0;
+/** @internal */
+class CachingElementIdsContainer {
+    constructor(_generator) {
+        this._generator = _generator;
+        this._ids = new Array();
+    }
+    async next() { return (await this._generator.next()).value; }
+    async *getElementIds() {
+        for (const id of this._ids) {
+            yield id;
+        }
+        let nextId;
+        while (nextId = await this.next()) {
+            this._ids.push(nextId);
+            yield nextId;
+        }
+    }
+}
+exports.CachingElementIdsContainer = CachingElementIdsContainer;
+//# sourceMappingURL=Utils.js.map

package/lib/cjs/appui-react/imodel-components/models-tree/Utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Utils.js","sourceRoot":"","sources":["../../../../../src/appui-react/imodel-components/models-tree/Utils.ts"],"names":[],"mappings":";;;AAMA,gBAAgB;AAChB,MAAa,0BAA0B;IAErC,YAAoB,UAAsC;QAAtC,eAAU,GAAV,UAAU,CAA4B;QADlD,SAAI,GAAG,IAAI,KAAK,EAAc,CAAC;IAEvC,CAAC;IAEO,KAAK,CAAC,IAAI,KAAI,OAAO,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAA,CAAC;IAE5D,KAAK,CAAA,CAAE,aAAa;QACzB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE;YAC1B,MAAM,EAAE,CAAC;SACV;QAED,IAAI,MAAM,CAAC;QACX,OAAO,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,EAAE;YACjC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,MAAM,MAAM,CAAC;SACd;IACH,CAAC;CACF;AAlBD,gEAkBC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\nimport { Id64String } from \"@itwin/core-bentley\";\r\n\r\n/** @internal */\r\nexport class CachingElementIdsContainer {\r\n  private _ids = new Array<Id64String>();\r\n  constructor(private _generator: AsyncGenerator<Id64String>) {\r\n  }\r\n\r\n  private async next() {return (await this._generator.next()).value;}\r\n\r\n  public async* getElementIds() {\r\n    for (const id of this._ids) {\r\n      yield id;\r\n    }\r\n\r\n    let nextId;\r\n    while (nextId = await this.next()) {\r\n      this._ids.push(nextId);\r\n      yield nextId;\r\n    }\r\n  }\r\n}\r\n"]}