@itwin/tree-widget-react 4.0.0-alpha.6 → 4.0.0-alpha.8

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

@@ -0,0 +1,701 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { concat, concatAll, defaultIfEmpty, defer, EMPTY, filter, firstValueFrom, forkJoin, from, fromEventPattern, map, merge, mergeAll, mergeMap, of, reduce, shareReplay, startWith, Subject, take, takeUntil, tap, toArray, } from "rxjs";
+import { assert, Id64 } from "@itwin/core-bentley";
+import { PerModelCategoryVisibility } from "@itwin/core-frontend";
+import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { AlwaysAndNeverDrawnElementInfo } from "../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import { toVoidPromise } from "../../common/internal/Rxjs.js";
+import { createVisibilityStatus } from "../../common/internal/Tooltip.js";
+import { getClassesByView, releaseMainThreadOnItemsCount, setDifference, setIntersection } from "../../common/internal/Utils.js";
+import { createVisibilityChangeEventListener } from "../../common/internal/VisibilityChangeEventListener.js";
+import { changeElementStateNoChildrenOperator, enableCategoryDisplay, enableSubCategoryDisplay, filterSubModeledElementIds, getElementOverriddenVisibility, getElementVisibility, getSubModeledElementsVisibilityStatus, getVisibilityFromAlwaysAndNeverDrawnElementsImpl, mergeVisibilityStatuses, } from "../../common/internal/VisibilityUtils.js";
+import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility.js";
+import { CategoriesTreeNode } from "./CategoriesTreeNode.js";
+import { createFilteredTree, parseCategoryKey, parseSubCategoryKey } from "./FilteredTree.js";
+/**
+ * Creates an instance if `CategoriesTreeVisibilityHandler`.
+ * @internal
+ */
+export function createCategoriesTreeVisibilityHandler(props) {
+    return new CategoriesTreeVisibilityHandlerImpl(props);
+}
+class CategoriesTreeVisibilityHandlerImpl {
+    constructor(_props) {
+        this._props = _props;
+        this._elementChangeQueue = new Subject();
+        this._subscriptions = [];
+        this._changeRequest = new Subject();
+        this._eventListener = createVisibilityChangeEventListener({
+            viewport: _props.viewport,
+            listeners: {
+                models: true,
+                categories: true,
+                elements: _props.hierarchyConfig.showElements,
+                displayStyle: true,
+            },
+        });
+        this._alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo(_props.viewport);
+        this._idsCache = this._props.idsCache;
+        const { categoryClass, elementClass, modelClass } = getClassesByView(_props.viewport.view.is2d() ? "2d" : "3d");
+        if (_props.filteredPaths) {
+            this._filteredTree = createFilteredTree({
+                idsCache: this._idsCache,
+                filteringPaths: _props.filteredPaths,
+                categoryClassName: categoryClass,
+                categoryElementClassName: elementClass,
+                categoryModelClassName: modelClass,
+                imodelAccess: this._props.imodelAccess,
+            });
+        }
+        this._subscriptions.push(this._elementChangeQueue.pipe(concatAll()).subscribe());
+    }
+    get onVisibilityChange() {
+        return this._eventListener.onVisibilityChange;
+    }
+    async getVisibilityStatus(node) {
+        return firstValueFrom(this.getVisibilityStatusObs(node).pipe(
+        // unsubscribe from the observable if the change request for this node is received
+        takeUntil(this._changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))), 
+        // unsubscribe if visibility changes
+        takeUntil(fromEventPattern((handler) => {
+            this._eventListener.onVisibilityChange.addListener(handler);
+        }, (handler) => {
+            this._eventListener.onVisibilityChange.removeListener(handler);
+        })), defaultIfEmpty(createVisibilityStatus("hidden"))));
+    }
+    async changeVisibility(node, shouldDisplay) {
+        // notify about new change request
+        this._changeRequest.next({ key: node.key, depth: node.parentKeys.length });
+        const changeObservable = this.changeVisibilityObs(node, shouldDisplay).pipe(
+        // unsubscribe from the observable if the change request for this node is received
+        takeUntil(this._changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))), tap({
+            subscribe: () => {
+                this._eventListener.suppressChangeEvents();
+                this._alwaysAndNeverDrawnElements.suppressChangeEvents();
+            },
+            finalize: () => {
+                this._eventListener.resumeChangeEvents();
+                this._alwaysAndNeverDrawnElements.resumeChangeEvents();
+            },
+        }));
+        return toVoidPromise(changeObservable);
+    }
+    dispose() {
+        this[Symbol.dispose]();
+    }
+    [Symbol.dispose]() {
+        this._eventListener[Symbol.dispose]();
+        this._alwaysAndNeverDrawnElements[Symbol.dispose]();
+        this._subscriptions.forEach((x) => x.unsubscribe());
+    }
+    getVisibilityStatusObs(node) {
+        if (node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.getFilteredNodeVisibility({ node });
+        }
+        if (HierarchyNode.isClassGroupingNode(node)) {
+            return this.getClassGroupingNodeDisplayStatus(node);
+        }
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            return this.getDefinitionContainerDisplayStatus({
+                definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        if (CategoriesTreeNode.isModelNode(node)) {
+            return this.getModelVisibilityStatus({
+                modelId: node.key.instanceKeys[0].id,
+            });
+        }
+        if (CategoriesTreeNode.isCategoryNode(node)) {
+            return this.getCategoryDisplayStatus({
+                categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                modelId: CategoriesTreeNode.getModelId(node),
+                ignoreSubCategories: node.extendedData?.isCategoryOfSubModel,
+            });
+        }
+        const categoryId = CategoriesTreeNode.getCategoryId(node);
+        if (!categoryId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            return this.getSubCategoryDisplayStatus({
+                categoryId,
+                subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        const modelId = CategoriesTreeNode.getModelId(node);
+        if (!modelId) {
+            return of(createVisibilityStatus("disabled"));
+        }
+        return this.getElementDisplayStatus({
+            elementId: node.key.instanceKeys[0].id,
+            modelId,
+            categoryId,
+        });
+    }
+    getFilteredNodeVisibility(props) {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ definitionContainerIds: definitionContainers, subCategories, modelIds: models, categories, elements }) => {
+            const observables = new Array();
+            if (definitionContainers?.size) {
+                observables.push(from(definitionContainers).pipe(mergeMap((definitionContainerId) => this.getDefinitionContainerDisplayStatus({ definitionContainerIds: [definitionContainerId] }))));
+            }
+            if (models?.size) {
+                observables.push(from(models).pipe(mergeMap((modelId) => this.getModelVisibilityStatus({ modelId }))));
+            }
+            if (categories?.size) {
+                observables.push(from(categories).pipe(mergeMap((key) => {
+                    const { modelId, categoryId } = parseCategoryKey(key);
+                    return this.getCategoryDisplayStatus({ ignoreSubCategories: !!modelId, modelId, categoryIds: [categoryId] });
+                })));
+            }
+            if (subCategories?.size) {
+                observables.push(from(subCategories).pipe(mergeMap((key) => {
+                    const { subCategoryId, categoryId } = parseSubCategoryKey(key);
+                    return this.getSubCategoryDisplayStatus({ subCategoryIds: [subCategoryId], categoryId });
+                })));
+            }
+            if (elements?.size) {
+                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(([categoryKey, elementIds]) => {
+                    const { modelId, categoryId } = parseCategoryKey(categoryKey);
+                    assert(modelId !== undefined);
+                    return from(elementIds).pipe(releaseMainThreadOnItemsCount(1000), mergeMap((elementId) => this.getElementDisplayStatus({ modelId, categoryId, elementId })));
+                })));
+            }
+            return merge(...observables);
+        }), mergeVisibilityStatuses);
+    }
+    getModelVisibilityStatus({ modelId }) {
+        const result = defer(() => {
+            const viewport = this._props.viewport;
+            if (!viewport.view.viewsModel(modelId)) {
+                return from(this._idsCache.getModelCategoryIds(modelId)).pipe(mergeMap((categoryIds) => from(this._idsCache.getCategoriesModeledElements(modelId, categoryIds))), getSubModeledElementsVisibilityStatus({
+                    parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
+                }));
+            }
+            return from(this._idsCache.getModelCategoryIds(modelId)).pipe(mergeAll(), mergeMap((categoryId) => this.getCategoryDisplayStatus({ modelId, categoryIds: [categoryId], ignoreSubCategories: true })), mergeVisibilityStatuses);
+        });
+        return createVisibilityHandlerResult(this, { id: modelId }, result, undefined);
+    }
+    getDefinitionContainerDisplayStatus(props) {
+        const result = defer(() => {
+            return from(this._idsCache.getAllContainedCategories(props.definitionContainerIds)).pipe(mergeAll(), mergeMap((categoryId) => this.getCategoryDisplayStatus({ categoryIds: [categoryId] })), mergeVisibilityStatuses);
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    getSubCategoryDisplayStatus(props) {
+        const result = defer(async () => {
+            const { categoryId, subCategoryIds } = props;
+            let visibility = "unknown";
+            const categoryModels = [...(await this._idsCache.getCategoriesElementModels([categoryId], true)).values()].flat();
+            let nonDefaultModelDisplayStatesCount = 0;
+            for (const modelId of categoryModels) {
+                if (!this._props.viewport.view.viewsModel(modelId)) {
+                    if (visibility === "visible") {
+                        return createVisibilityStatus("partial");
+                    }
+                    visibility = "hidden";
+                    ++nonDefaultModelDisplayStatesCount;
+                    continue;
+                }
+                const override = this._props.viewport.perModelCategoryVisibility.getOverride(modelId, categoryId);
+                if (override === PerModelCategoryVisibility.Override.Show) {
+                    if (visibility === "hidden") {
+                        return createVisibilityStatus("partial");
+                    }
+                    visibility = "visible";
+                    ++nonDefaultModelDisplayStatesCount;
+                    continue;
+                }
+                if (override === PerModelCategoryVisibility.Override.Hide) {
+                    if (visibility === "visible") {
+                        return createVisibilityStatus("partial");
+                    }
+                    visibility = "hidden";
+                    ++nonDefaultModelDisplayStatesCount;
+                    continue;
+                }
+            }
+            if (categoryModels.length > 0 && nonDefaultModelDisplayStatesCount === categoryModels.length) {
+                assert(visibility === "visible" || visibility === "hidden");
+                return createVisibilityStatus(visibility);
+            }
+            if (!this._props.viewport.view.viewsCategory(categoryId)) {
+                return createVisibilityStatus(visibility === "visible" ? "partial" : "hidden");
+            }
+            if (subCategoryIds.length === 0) {
+                if (visibility === "hidden") {
+                    return createVisibilityStatus("partial");
+                }
+                return createVisibilityStatus("visible");
+            }
+            for (const subCategoryId of subCategoryIds) {
+                const isSubCategoryVisible = this._props.viewport.isSubCategoryVisible(subCategoryId);
+                if (isSubCategoryVisible && visibility === "hidden") {
+                    return createVisibilityStatus("partial");
+                }
+                if (!isSubCategoryVisible && visibility === "visible") {
+                    return createVisibilityStatus("partial");
+                }
+                visibility = isSubCategoryVisible ? "visible" : "hidden";
+            }
+            assert(visibility === "visible" || visibility === "hidden");
+            return createVisibilityStatus(visibility);
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    getDefaultModelsCategoryVisibilityStatus({ modelId, categoryIds }) {
+        const viewport = this._props.viewport;
+        if (!viewport.view.viewsModel(modelId)) {
+            return createVisibilityStatus("hidden");
+        }
+        let visibleCount = 0;
+        let hiddenCount = 0;
+        let visibleThroughCategorySelectorCount = 0;
+        for (const categoryId of categoryIds) {
+            if (viewport.view.viewsCategory(categoryId)) {
+                ++visibleThroughCategorySelectorCount;
+            }
+            const override = this._props.viewport.perModelCategoryVisibility.getOverride(modelId, categoryId);
+            if (override === PerModelCategoryVisibility.Override.Show) {
+                ++visibleCount;
+                continue;
+            }
+            if (override === PerModelCategoryVisibility.Override.Hide) {
+                ++hiddenCount;
+                continue;
+            }
+            if (visibleCount > 0 && hiddenCount > 0) {
+                return createVisibilityStatus("partial");
+            }
+        }
+        if (hiddenCount + visibleCount > 0) {
+            return createVisibilityStatus(hiddenCount > 0 ? "hidden" : "visible");
+        }
+        return createVisibilityStatus(visibleThroughCategorySelectorCount > 0 ? "visible" : "hidden");
+    }
+    getDefaultCategoryVisibilityStatus({ categoryIds }) {
+        return from(this._idsCache.getSubCategories(categoryIds)).pipe(mergeMap((categoriesSubCategoriesMap) => {
+            return from(categoryIds).pipe(mergeMap((categoryId) => {
+                const subCategoryIds = categoriesSubCategoriesMap.get(categoryId);
+                return this.getSubCategoryDisplayStatus({ subCategoryIds: subCategoryIds ?? [], categoryId });
+            }), mergeVisibilityStatuses);
+        }));
+    }
+    getCategoryDisplayStatus(props) {
+        const result = defer(() => {
+            if (!this._props.hierarchyConfig.showElements) {
+                return from(this.getDefaultCategoryVisibilityStatus({ categoryIds: props.categoryIds }));
+            }
+            const modelsObservable = props.modelId
+                ? of(new Map(props.categoryIds.map((id) => [id, [props.modelId]])))
+                : from(this._idsCache.getCategoriesElementModels(props.categoryIds));
+            return merge(
+            // get visibility status from always and never drawn elements
+            this._props.hierarchyConfig.showElements
+                ? modelsObservable.pipe(mergeMap((categoryModelsMap) => {
+                    if (categoryModelsMap.size === 0) {
+                        return props.modelId
+                            ? of(this.getDefaultModelsCategoryVisibilityStatus({ modelId: props.modelId, categoryIds: props.categoryIds }))
+                            : from(this.getDefaultCategoryVisibilityStatus({ categoryIds: props.categoryIds }));
+                    }
+                    return from(categoryModelsMap).pipe(mergeMap(([category, models]) => from(models).pipe(mergeMap((model) => {
+                        if (this._props.viewport.view.viewsModel(model)) {
+                            return this.getVisibilityFromAlwaysAndNeverDrawnElements({
+                                queryProps: props,
+                                defaultStatus: () => this.getDefaultModelsCategoryVisibilityStatus({ modelId: model, categoryIds: [category] }),
+                            }).pipe(mergeMap((visibilityStatusAlwaysAndNeverDraw) => {
+                                return from(this._idsCache.getCategoriesModeledElements(model, [category])).pipe(getSubModeledElementsVisibilityStatus({
+                                    parentNodeVisibilityStatus: visibilityStatusAlwaysAndNeverDraw,
+                                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
+                                }));
+                            }));
+                        }
+                        return from(this._idsCache.getCategoriesModeledElements(model, [category])).pipe(getSubModeledElementsVisibilityStatus({
+                            parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
+                            getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
+                        }));
+                    }), mergeVisibilityStatuses)), mergeVisibilityStatuses);
+                }), map((visibilityStatus) => {
+                    return { visibilityStatus, type: 0 };
+                }))
+                : EMPTY, 
+            // get category status
+            (props.modelId
+                ? of(this.getDefaultModelsCategoryVisibilityStatus({ modelId: props.modelId, categoryIds: props.categoryIds }))
+                : from(this.getDefaultCategoryVisibilityStatus({ categoryIds: props.categoryIds }))).pipe(map((visibilityStatus) => {
+                return { visibilityStatus, type: 1 };
+            }))).pipe(toArray(), mergeMap(async (visibilityStatusesInfo) => {
+                let defaultStatus;
+                let alwaysNeverDrawStatus;
+                visibilityStatusesInfo.forEach((visibilityStatusInfo) => {
+                    switch (visibilityStatusInfo.type) {
+                        case 0:
+                            alwaysNeverDrawStatus = visibilityStatusInfo.visibilityStatus;
+                            break;
+                        case 1:
+                            defaultStatus = visibilityStatusInfo.visibilityStatus;
+                            break;
+                    }
+                });
+                assert(defaultStatus !== undefined);
+                if (defaultStatus.state === "partial") {
+                    return defaultStatus;
+                }
+                // This can happen if:
+                // a) showElements is set to false
+                // b) root category does not have any elements (that dont have Parent)
+                // In both cases we don't need to look at modeled elements visibility
+                if (alwaysNeverDrawStatus === undefined) {
+                    return defaultStatus;
+                }
+                // In cases where
+                // a) SubCategories are hidden
+                // b) Category has model (it means that category is under hidden subModel)
+                // We dont need to look at default category status, it is already accounted for in always/never drawn visibility
+                if (this._props.hierarchyConfig.hideSubCategories || props.modelId) {
+                    return alwaysNeverDrawStatus;
+                }
+                if ((await this._idsCache.getSubCategories(props.categoryIds)).size === 0) {
+                    return alwaysNeverDrawStatus;
+                }
+                if (alwaysNeverDrawStatus.state === "partial" || alwaysNeverDrawStatus.state !== defaultStatus.state) {
+                    return createVisibilityStatus("partial");
+                }
+                return alwaysNeverDrawStatus;
+            }));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    getClassGroupingNodeDisplayStatus(node) {
+        const result = defer(() => {
+            const info = this.getGroupingNodeInfo(node);
+            const { modelElementsMap, categoryId } = info;
+            return from(modelElementsMap).pipe(mergeMap(([modelId, elementIds]) => {
+                if (!this._props.viewport.view.viewsModel(modelId)) {
+                    return of([...elementIds]).pipe(filterSubModeledElementIds({ doesSubModelExist: async (id) => this._idsCache.hasSubModel(id) }), getSubModeledElementsVisibilityStatus({
+                        parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
+                        getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
+                    }));
+                }
+                return this.getVisibilityFromAlwaysAndNeverDrawnElements({
+                    elements: elementIds,
+                    defaultStatus: () => this.getDefaultModelsCategoryVisibilityStatus({ categoryIds: [categoryId], modelId }),
+                }).pipe(mergeMap((visibilityStatusAlwaysAndNeverDraw) => {
+                    return of([...elementIds]).pipe(filterSubModeledElementIds({ doesSubModelExist: async (id) => this._idsCache.hasSubModel(id) }), getSubModeledElementsVisibilityStatus({
+                        parentNodeVisibilityStatus: visibilityStatusAlwaysAndNeverDraw,
+                        getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
+                    }));
+                }));
+            }), mergeVisibilityStatuses);
+        });
+        return createVisibilityHandlerResult(this, { node }, result, undefined);
+    }
+    getElementDisplayStatus(props) {
+        const result = defer(() => {
+            const viewport = this._props.viewport;
+            const { elementId, modelId, categoryId } = props;
+            const viewsModel = viewport.view.viewsModel(modelId);
+            const elementStatus = getElementOverriddenVisibility({
+                elementId,
+                viewport,
+            });
+            return from(this._idsCache.hasSubModel(elementId)).pipe(mergeMap((hasSubModel) => (hasSubModel ? this.getModelVisibilityStatus({ modelId: elementId }) : of(undefined))), map((subModelVisibilityStatus) => getElementVisibility(viewsModel, elementStatus, this.getDefaultModelsCategoryVisibilityStatus({ categoryIds: [categoryId], modelId }), subModelVisibilityStatus)));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    /** Changes visibility of the items represented by the tree node. */
+    changeVisibilityObs(node, on) {
+        if (node.filtering?.filteredChildrenIdentifierPaths?.length && !node.filtering.isFilterTarget) {
+            return this.changeFilteredNodeVisibility({ node, on });
+        }
+        if (HierarchyNode.isClassGroupingNode(node)) {
+            return this.changeElementGroupingNodeState(node, on);
+        }
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return EMPTY;
+        }
+        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            return this.changeDefinitionContainerState({
+                definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                on,
+            });
+        }
+        if (CategoriesTreeNode.isModelNode(node)) {
+            return this.changeModelState({
+                ids: node.key.instanceKeys.map(({ id }) => id),
+                on,
+            });
+        }
+        if (CategoriesTreeNode.isCategoryNode(node)) {
+            return this.changeCategoryState({
+                categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                modelId: CategoriesTreeNode.getModelId(node),
+                on,
+                ignoreSubCategories: node.extendedData?.isCategoryOfSubModel,
+            });
+        }
+        const categoryId = CategoriesTreeNode.getCategoryId(node);
+        if (!categoryId) {
+            return EMPTY;
+        }
+        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            if (this._props.hierarchyConfig.hideSubCategories) {
+                return EMPTY;
+            }
+            return this.changeSubCategoryState({
+                categoryId,
+                subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                on,
+            });
+        }
+        if (!this._props.hierarchyConfig.showElements) {
+            return EMPTY;
+        }
+        const modelId = CategoriesTreeNode.getModelId(node);
+        if (!modelId) {
+            return EMPTY;
+        }
+        return this.changeElementsState({
+            elementIds: new Set([...node.key.instanceKeys.map(({ id }) => id)]),
+            modelId,
+            categoryId,
+            on,
+        });
+    }
+    async getVisibilityChangeTargets({ node }) {
+        const filteredTree = await this._filteredTree;
+        return filteredTree ? filteredTree.getVisibilityChangeTargets(node) : {};
+    }
+    changeFilteredNodeVisibility({ on, ...props }) {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ definitionContainerIds: definitionContainers, subCategories, modelIds: models, categories, elements }) => {
+            const observables = new Array();
+            if (definitionContainers?.size) {
+                observables.push(this.changeDefinitionContainerState({ definitionContainerIds: [...definitionContainers], on }));
+            }
+            if (models?.size) {
+                observables.push(this.changeModelState({ ids: models, on }));
+            }
+            if (categories?.size) {
+                observables.push(from(categories).pipe(mergeMap((key) => {
+                    const { modelId, categoryId } = parseCategoryKey(key);
+                    return this.changeCategoryState({ modelId, categoryIds: [categoryId], ignoreSubCategories: false, on });
+                })));
+            }
+            if (subCategories?.size) {
+                observables.push(from(subCategories).pipe(mergeMap((key) => {
+                    const { subCategoryId, categoryId } = parseSubCategoryKey(key);
+                    return this.changeSubCategoryState({ categoryId, subCategoryIds: [subCategoryId], on });
+                })));
+            }
+            if (elements?.size) {
+                observables.push(from(elements).pipe(mergeMap(([categoryKey, elementIds]) => {
+                    const { modelId, categoryId } = parseCategoryKey(categoryKey);
+                    assert(modelId !== undefined);
+                    return this.changeElementsState({ modelId, categoryId, elementIds, on });
+                })));
+            }
+            return merge(...observables);
+        }));
+    }
+    changeModelState(props) {
+        const { ids, on } = props;
+        if (Id64.sizeOf(ids) === 0) {
+            return EMPTY;
+        }
+        const result = defer(() => {
+            const viewport = this._props.viewport;
+            viewport.perModelCategoryVisibility.clearOverrides(ids);
+            const idsObs = from(Id64.iterable(ids));
+            if (!on) {
+                viewport.changeModelDisplay(ids, false);
+                return idsObs.pipe(mergeMap(async (modelId) => ({ modelId, categoryIds: await this._idsCache.getModelCategoryIds(modelId) })), mergeMap(({ modelId, categoryIds }) => from(this._idsCache.getCategoriesModeledElements(modelId, categoryIds))), mergeMap((modeledElementIds) => this.changeModelState({ ids: modeledElementIds, on })));
+            }
+            return concat(from(viewport.addViewedModels(ids)), idsObs.pipe(mergeMap((modelId) => {
+                return from(this._idsCache.getModelCategoryIds(modelId)).pipe(mergeAll(), mergeMap((categoryId) => this.changeCategoryState({ categoryIds: [categoryId], modelId, on, ignoreSubCategories: true })));
+            })));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    showModelWithoutAnyCategoriesOrElements(modelId) {
+        const viewport = this._props.viewport;
+        return forkJoin({
+            categories: this._idsCache.getModelCategoryIds(modelId),
+            alwaysDrawnElements: this._alwaysAndNeverDrawnElements.getAlwaysDrawnElements({ modelId }),
+        }).pipe(mergeMap(async ({ categories, alwaysDrawnElements }) => {
+            const alwaysDrawn = this._props.viewport.alwaysDrawn;
+            if (alwaysDrawn && alwaysDrawnElements) {
+                viewport.setAlwaysDrawn(setDifference(alwaysDrawn, alwaysDrawnElements));
+            }
+            categories.forEach((categoryId) => {
+                this.changeCategoryStateInViewportAccordingToModelVisibility(modelId, categoryId, false, false);
+            });
+            await viewport.addViewedModels(modelId);
+        }));
+    }
+    changeCategoryStateInViewportAccordingToModelVisibility(modelId, categoryId, on, changeSubCategories) {
+        const viewport = this._props.viewport;
+        const isDisplayedInSelector = viewport.view.viewsCategory(categoryId);
+        const override = on === isDisplayedInSelector
+            ? PerModelCategoryVisibility.Override.None
+            : on
+                ? PerModelCategoryVisibility.Override.Show
+                : PerModelCategoryVisibility.Override.Hide;
+        viewport.perModelCategoryVisibility.setOverride(modelId, categoryId, override);
+        if (override === 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
+            viewport.changeCategoryDisplay(categoryId, true, changeSubCategories);
+        }
+    }
+    changeSubCategoryState(props) {
+        const result = defer(() => {
+            return concat(
+            // make sure parent category and models are enabled
+            props.on
+                ? concat(from(enableCategoryDisplay(this._props.viewport, [props.categoryId], props.on, false)), from(this.enableCategoriesElementModelsVisibility([props.categoryId])))
+                : EMPTY, from(props.subCategoryIds).pipe(map((subCategoryId) => enableSubCategoryDisplay(this._props.viewport, subCategoryId, props.on))));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    async enableCategoriesElementModelsVisibility(categoryIds) {
+        const categoriesModelsMap = await this._idsCache.getCategoriesElementModels(categoryIds, true);
+        const modelIds = [...categoriesModelsMap.values()].flat();
+        const hiddenModels = modelIds.filter((modelId) => !this._props.viewport.view.viewsModel(modelId));
+        if (hiddenModels.length > 0) {
+            this._props.viewport.changeModelDisplay(hiddenModels, true);
+        }
+    }
+    changeDefinitionContainerState(props) {
+        const result = defer(() => {
+            return from(this._idsCache.getAllContainedCategories(props.definitionContainerIds)).pipe(mergeMap((categoryIds) => {
+                return this.changeCategoryState({ categoryIds, on: props.on });
+            }));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    changeCategoryState(props) {
+        const result = defer(() => {
+            const { modelId, categoryIds, on, ignoreSubCategories } = props;
+            const viewport = this._props.viewport;
+            if (!this._props.hierarchyConfig.showElements) {
+                return concat(from(enableCategoryDisplay(viewport, categoryIds, on, on)), on ? from(this.enableCategoriesElementModelsVisibility(categoryIds)) : EMPTY);
+            }
+            const modelIdsObservable = modelId
+                ? of(new Map(categoryIds.map((id) => [id, [modelId]])))
+                : from(this._idsCache.getCategoriesElementModels(categoryIds, true));
+            return concat(modelId ? EMPTY : from(enableCategoryDisplay(viewport, categoryIds, on, ignoreSubCategories ? false : on)), on
+                ? modelIdsObservable.pipe(mergeMap((categoriesMap) => from(categoriesMap.values())), mergeAll(), filter((modelIdToCheck) => !viewport.view.viewsModel(modelIdToCheck)), mergeMap((modelIdToCheck) => this.showModelWithoutAnyCategoriesOrElements(modelIdToCheck)))
+                : EMPTY, modelIdsObservable.pipe(mergeMap((categoriesMap) => from(categoriesMap.entries())), mergeMap(([categoryId, modelIds]) => {
+                return from(modelIds).pipe(mergeMap((modelOfCategory) => {
+                    this.changeCategoryStateInViewportAccordingToModelVisibility(modelOfCategory, categoryId, on, !!ignoreSubCategories);
+                    return this._alwaysAndNeverDrawnElements.clearAlwaysAndNeverDrawnElements({ modelId: modelOfCategory, categoryIds: [categoryId] });
+                }));
+            })), modelIdsObservable.pipe(mergeMap((categoriesMap) => from(categoriesMap.entries())), mergeMap(([categoryId, modelIds]) => {
+                return from(modelIds).pipe(mergeMap((modelOfCategory) => from(this._idsCache.getCategoriesModeledElements(modelOfCategory, [categoryId])).pipe(mergeMap((modeledElementIds) => this.changeModelState({ ids: modeledElementIds, on })))));
+            })));
+        });
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    doChangeElementsState(props) {
+        return defer(() => {
+            const { modelId, categoryId, elementIds, on } = props;
+            const viewport = this._props.viewport;
+            return concat(on && !viewport.view.viewsModel(modelId) ? this.showModelWithoutAnyCategoriesOrElements(modelId) : EMPTY, defer(() => {
+                const categoryVisibility = this.getDefaultModelsCategoryVisibilityStatus({ categoryIds: [categoryId], modelId });
+                const isDisplayedByDefault = categoryVisibility.state === "visible";
+                return this.queueElementsVisibilityChange(elementIds, on, isDisplayedByDefault);
+            }), from(elementIds).pipe(mergeMap(async (elementId) => ({ elementId, isSubModel: await this._idsCache.hasSubModel(elementId) })), filter(({ isSubModel }) => isSubModel), map(({ elementId }) => elementId), toArray(), mergeMap((subModelIds) => this.changeModelState({ ids: subModelIds, on }))));
+        });
+    }
+    /**
+     * Updates visibility of all grouping node's elements.
+     * @see `changeElementState`
+     */
+    changeElementGroupingNodeState(node, on) {
+        const info = this.getGroupingNodeInfo(node);
+        const result = from(info.modelElementsMap).pipe(mergeMap(([modelId, elementIds]) => {
+            return this.doChangeElementsState({ modelId, elementIds, categoryId: info.categoryId, on });
+        }));
+        return createVisibilityHandlerResult(this, { node, on }, result, undefined);
+    }
+    /**
+     * Updates visibility of an element and all its child elements by adding them to the always/never drawn list.
+     * @note If element is to be enabled and model is hidden, it will be enabled.
+     */
+    changeElementsState(props) {
+        const result = this.doChangeElementsState(props);
+        return createVisibilityHandlerResult(this, props, result, undefined);
+    }
+    queueElementsVisibilityChange(elementIds, on, visibleByDefault) {
+        const finishedSubject = new Subject();
+        // observable to track if visibility change is finished/cancelled
+        const changeFinished = finishedSubject.pipe(startWith(false), shareReplay(1), filter((finished) => finished));
+        const changeObservable = from(elementIds).pipe(
+        // check if visibility change is not finished (cancelled) due to change overall change request being cancelled
+        takeUntil(changeFinished), changeElementStateNoChildrenOperator({ on, isDisplayedByDefault: visibleByDefault, viewport: this._props.viewport }), tap({
+            next: () => {
+                // notify that visibility change is finished
+                finishedSubject.next(true);
+            },
+        }));
+        // queue visibility change. `changeObservable` will be subscribed to when other queue changes are finished
+        this._elementChangeQueue.next(changeObservable);
+        // return observable that will emit when visibility change is finished
+        return changeFinished.pipe(take(1), tap({
+            unsubscribe: () => {
+                // if this observable is unsubscribed before visibility change is finished, we have to notify that it queued change request is cancelled
+                finishedSubject.next(true);
+            },
+        }), map(() => undefined));
+    }
+    getVisibilityFromAlwaysAndNeverDrawnElements(props) {
+        const viewport = this._props.viewport;
+        if (viewport.isAlwaysDrawnExclusive) {
+            if (!viewport?.alwaysDrawn?.size) {
+                return of(createVisibilityStatus("hidden"));
+            }
+        }
+        else if (!viewport?.neverDrawn?.size && !viewport?.alwaysDrawn?.size) {
+            return of(props.defaultStatus());
+        }
+        if ("elements" in props) {
+            return of(getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
+                ...props,
+                alwaysDrawn: viewport.alwaysDrawn?.size ? setIntersection(props.elements, viewport.alwaysDrawn) : undefined,
+                neverDrawn: viewport.neverDrawn?.size ? setIntersection(props.elements, viewport.neverDrawn) : undefined,
+                totalCount: props.elements.size,
+                viewport,
+            }));
+        }
+        const { modelId, categoryIds } = props.queryProps;
+        const totalCount = (modelId ? of(new Map(categoryIds.map((categoryId) => [categoryId, [modelId]]))) : from(this._idsCache.getCategoriesElementModels(categoryIds))).pipe(mergeMap((categoriesMap) => from(categoriesMap)), mergeMap(([categoryId, modelIds]) => {
+            return from(modelIds).pipe(mergeMap((modelOfCategory) => from(this._idsCache.getCategoryElementsCount(modelOfCategory, categoryId))), reduce((acc, specificModelCategoryCount) => {
+                return acc + specificModelCategoryCount;
+            }, 0));
+        }));
+        return forkJoin({
+            totalCount,
+            alwaysDrawn: this._alwaysAndNeverDrawnElements.getAlwaysDrawnElements(props.queryProps),
+            neverDrawn: this._alwaysAndNeverDrawnElements.getNeverDrawnElements(props.queryProps),
+        }).pipe(map((state) => {
+            return getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
+                ...props,
+                ...state,
+                viewport,
+            });
+        }));
+    }
+    getGroupingNodeInfo(node) {
+        const modelElementsMap = node.extendedData?.modelElementsMap;
+        const categoryId = node.extendedData?.categoryId;
+        assert(!!modelElementsMap && !!categoryId);
+        return { modelElementsMap, categoryId };
+    }
+}
+//# sourceMappingURL=CategoriesTreeVisibilityHandler.js.map