@itwin/tree-widget-react 4.0.0-alpha.5 → 4.0.0-alpha.7

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

@@ -2,19 +2,20 @@
  * 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, defer, distinct, EMPTY, filter, firstValueFrom, forkJoin, from, fromEventPattern, map, merge, mergeMap, of, reduce, shareReplay, startWith, Subject, take, takeUntil, tap, toArray, } from "rxjs";
+import { concat, concatAll, defaultIfEmpty, defer, distinct, EMPTY, filter, firstValueFrom, forkJoin, from, fromEventPattern, map, merge, mergeMap, of, 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 { toggleAllCategories } from "../../common/CategoriesVisibilityUtils.js";
-import { reduceWhile, toVoidPromise } from "../../common/Rxjs.js";
-import { createVisibilityStatus } from "../../common/Tooltip.js";
+import { AlwaysAndNeverDrawnElementInfo } from "../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import { GEOMETRIC_ELEMENT_3D_CLASS_NAME } from "../../common/internal/ClassNameDefinitions.js";
+import { toVoidPromise } from "../../common/internal/Rxjs.js";
+import { createVisibilityStatus, getTooltipOptions } from "../../common/internal/Tooltip.js";
+import { releaseMainThreadOnItemsCount, setDifference, setIntersection } from "../../common/internal/Utils.js";
+import { createVisibilityChangeEventListener } from "../../common/internal/VisibilityChangeEventListener.js";
+import { changeElementStateNoChildrenOperator, filterSubModeledElementIds, getElementOverriddenVisibility, getElementVisibility, getSubModeledElementsVisibilityStatus, getVisibilityFromAlwaysAndNeverDrawnElementsImpl, mergeVisibilityStatuses, } from "../../common/internal/VisibilityUtils.js";
 import { createVisibilityHandlerResult } from "../../common/UseHierarchyVisibility.js";
-import { releaseMainThreadOnItemsCount } from "../Utils.js";
-import { AlwaysAndNeverDrawnElementInfo } from "./AlwaysAndNeverDrawnElementInfo.js";
 import { createFilteredTree, parseCategoryKey } from "./FilteredTree.js";
 import { ModelsTreeNode } from "./ModelsTreeNode.js";
-import { createVisibilityChangeEventListener } from "./VisibilityChangeEventListener.js";
 /**
  * Creates an instance if `ModelsTreeVisibilityHandler`.
  * @internal
@@ -28,8 +29,15 @@ class ModelsTreeVisibilityHandlerImpl {
         this._elementChangeQueue = new Subject();
         this._subscriptions = [];
         this._changeRequest = new Subject();
-        this._eventListener = createVisibilityChangeEventListener(_props.viewport);
-        this._alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo(_props.viewport);
+        this._eventListener = createVisibilityChangeEventListener({
+            viewport: _props.viewport,
+            listeners: {
+                models: true,
+                categories: true,
+                elements: true,
+            },
+        });
+        this._alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo(_props.viewport, GEOMETRIC_ELEMENT_3D_CLASS_NAME);
         this._idsCache = this._props.idsCache;
         this._filteredTree = _props.filteredPaths ? createFilteredTree(this._props.imodelAccess, _props.filteredPaths) : undefined;
         this._subscriptions.push(this._elementChangeQueue.pipe(concatAll()).subscribe());
@@ -46,7 +54,7 @@ class ModelsTreeVisibilityHandlerImpl {
             this._eventListener.onVisibilityChange.addListener(handler);
         }, (handler) => {
             this._eventListener.onVisibilityChange.removeListener(handler);
-        }))));
+        })), defaultIfEmpty(createVisibilityStatus("hidden"))));
     }
     async changeVisibility(node, shouldDisplay) {
         // notify about new change request
@@ -69,8 +77,8 @@ class ModelsTreeVisibilityHandlerImpl {
         this[Symbol.dispose]();
     }
     [Symbol.dispose]() {
-        this._eventListener.dispose();
-        this._alwaysAndNeverDrawnElements.dispose();
+        this._eventListener[Symbol.dispose]();
+        this._alwaysAndNeverDrawnElements[Symbol.dispose]();
         this._subscriptions.forEach((x) => x.unsubscribe());
     }
     getVisibilityStatusObs(node) {
@@ -111,7 +119,7 @@ class ModelsTreeVisibilityHandlerImpl {
         });
     }
     getFilteredNodeVisibility(props) {
-        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ subjects, models, categories, elements }) => {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ subjectIds: subjects, modelIds: models, categories, elements }) => {
             const observables = new Array();
             if (subjects?.size) {
                 observables.push(this.getSubjectNodeVisibilityStatus({ subjectIds: [...subjects], ignoreTooltip: true }));
@@ -132,17 +140,18 @@ class ModelsTreeVisibilityHandlerImpl {
                 })));
             }
             return merge(...observables);
-        }), map((x) => x.state), getVisibilityFromTreeNodeChildren, map((x) => {
-            assert(x !== "empty");
-            return createVisibilityStatus(x);
-        }));
+        }), mergeVisibilityStatuses({
+            visible: undefined,
+            hidden: undefined,
+            partial: undefined,
+        }, true));
     }
     getSubjectNodeVisibilityStatus({ subjectIds, ignoreTooltip }) {
         const result = defer(() => {
             if (!this._props.viewport.view.isSpatialView()) {
                 return of(createVisibilityStatus("disabled", getTooltipOptions("modelsTree.subject.nonSpatialView", ignoreTooltip)));
             }
-            return from(this._idsCache.getSubjectModelIds(subjectIds)).pipe(concatAll(), distinct(), mergeMap((modelId) => this.getModelVisibilityStatus({ modelId, ignoreTooltip: true })), map((x) => x.state), getVisibilityStatusFromTreeNodeChildren({
+            return from(this._idsCache.getSubjectModelIds(subjectIds)).pipe(concatAll(), distinct(), mergeMap((modelId) => this.getModelVisibilityStatus({ modelId, ignoreTooltip: true })), mergeVisibilityStatuses({
                 visible: "modelsTree.subject.allModelsVisible",
                 hidden: "modelsTree.subject.allModelsHidden",
                 partial: "modelsTree.subject.someModelsHidden",
@@ -157,16 +166,17 @@ class ModelsTreeVisibilityHandlerImpl {
                 return of(createVisibilityStatus("disabled", getTooltipOptions("modelsTree.model.nonSpatialView", ignoreTooltip)));
             }
             if (!viewport.view.viewsModel(modelId)) {
-                return from(this._idsCache.getModelCategories(modelId)).pipe(mergeMap((categoryIds) => from(this._idsCache.getCategoriesModeledElements(modelId, categoryIds))), this.getSubModeledElementsVisibilityStatus({
+                return from(this._idsCache.getModelCategoryIds(modelId)).pipe(mergeMap((categoryIds) => from(this._idsCache.getCategoriesModeledElements(modelId, categoryIds))), getSubModeledElementsVisibilityStatus({
                     ignoreTooltips: ignoreTooltip,
-                    haveSubModel: "yes",
                     tooltips: { visible: undefined, hidden: "modelsTree.model.hiddenThroughModelSelector", partial: "modelsTree.model.someSubModelsVisible" },
                     parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
                 }));
             }
-            return from(this._idsCache.getModelCategories(modelId)).pipe(concatAll(), mergeMap((categoryId) => this.getCategoryDisplayStatus({ modelId, categoryId, ignoreTooltip: true })), map((x) => x.state), getVisibilityFromTreeNodeChildren, map((visibilityByCategories) => {
-                const state = visibilityByCategories === "empty" ? "visible" : visibilityByCategories;
-                return createVisibilityStatus(state, getTooltipOptions(state === "partial" ? "modelsTree.model.someCategoriesHidden" : `modelsTree.model.allCategories${state ? "Visible" : "Hidden"}`, ignoreTooltip));
+            return from(this._idsCache.getModelCategoryIds(modelId)).pipe(concatAll(), mergeMap((categoryId) => this.getCategoryDisplayStatus({ modelId, categoryId, ignoreTooltip: true })), mergeVisibilityStatuses({
+                visible: "modelsTree.model.allCategoriesVisible",
+                hidden: "modelsTree.model.allCategoriesHidden",
+                partial: "modelsTree.model.someCategoriesHidden",
             }));
         });
         return createVisibilityHandlerResult(this, { id: modelId }, result, this._props.overrides?.getModelDisplayStatus);
@@ -190,7 +200,7 @@ class ModelsTreeVisibilityHandlerImpl {
     getCategoryDisplayStatus({ ignoreTooltip, ...props }) {
         const result = defer(() => {
             if (!this._props.viewport.view.viewsModel(props.modelId)) {
-                return from(this._idsCache.getCategoriesModeledElements(props.modelId, [props.categoryId])).pipe(this.getSubModeledElementsVisibilityStatus({
+                return from(this._idsCache.getCategoriesModeledElements(props.modelId, [props.categoryId])).pipe(getSubModeledElementsVisibilityStatus({
                     ignoreTooltips: ignoreTooltip,
                     parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
                     tooltips: {
@@ -198,11 +208,14 @@ class ModelsTreeVisibilityHandlerImpl {
                         hidden: "modelsTree.category.hiddenThroughModel",
                         partial: "modelsTree.category.someElementsOrSubModelsHidden",
                     },
-                    haveSubModel: "yes",
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
                 }));
             }
             return this.getVisibilityFromAlwaysAndNeverDrawnElements({
-                queryProps: props,
+                queryProps: {
+                    categoryIds: [props.categoryId],
+                    modelId: props.modelId,
+                },
                 tooltips: {
                     allElementsInAlwaysDrawnList: "modelsTree.category.allElementsVisible",
                     allElementsInNeverDrawnList: "modelsTree.category.allElementsHidden",
@@ -212,15 +225,15 @@ class ModelsTreeVisibilityHandlerImpl {
                 defaultStatus: () => this.getDefaultCategoryVisibilityStatus(props),
                 ignoreTooltip,
             }).pipe(mergeMap((visibilityStatusAlwaysAndNeverDraw) => {
-                return from(this._idsCache.getCategoriesModeledElements(props.modelId, [props.categoryId])).pipe(this.getSubModeledElementsVisibilityStatus({
+                return from(this._idsCache.getCategoriesModeledElements(props.modelId, [props.categoryId])).pipe(getSubModeledElementsVisibilityStatus({
                     tooltips: {
                         visible: undefined,
                         hidden: "modelsTree.category.allElementsAndSubModelsHidden",
                         partial: "modelsTree.category.someElementsOrSubModelsHidden",
                     },
-                    haveSubModel: "yes",
                     parentNodeVisibilityStatus: visibilityStatusAlwaysAndNeverDraw,
                     ignoreTooltips: ignoreTooltip,
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
                 }));
             }));
         });
@@ -231,14 +244,14 @@ class ModelsTreeVisibilityHandlerImpl {
             const info = this.getGroupingNodeInfo(node);
             const { modelId, categoryId, elementIds } = info;
             if (!this._props.viewport.view.viewsModel(modelId)) {
-                return of([...elementIds]).pipe(this.getSubModeledElementsVisibilityStatus({
+                return of([...elementIds]).pipe(filterSubModeledElementIds({ doesSubModelExist: async (id) => this._idsCache.hasSubModel(id) }), getSubModeledElementsVisibilityStatus({
                     tooltips: {
                         visible: undefined,
                         hidden: undefined,
                         partial: "modelsTree.groupingNode.someElementsOrSubModelsHidden",
                     },
                     parentNodeVisibilityStatus: createVisibilityStatus("hidden"),
-                    haveSubModel: "unknown",
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
                 }));
             }
             return this.getVisibilityFromAlwaysAndNeverDrawnElements({
@@ -254,74 +267,35 @@ class ModelsTreeVisibilityHandlerImpl {
                     noElementsInExclusiveAlwaysDrawnList: "modelsTree.groupingNode.allElementsHidden",
                 },
             }).pipe(mergeMap((visibilityStatusAlwaysAndNeverDraw) => {
-                return of([...elementIds]).pipe(this.getSubModeledElementsVisibilityStatus({
+                return of([...elementIds]).pipe(filterSubModeledElementIds({ doesSubModelExist: async (id) => this._idsCache.hasSubModel(id) }), getSubModeledElementsVisibilityStatus({
                     tooltips: {
                         visible: undefined,
                         hidden: "modelsTree.groupingNode.allElementsAndSubModelsHidden",
                         partial: "modelsTree.groupingNode.someElementsOrSubModelsHidden",
                     },
                     parentNodeVisibilityStatus: visibilityStatusAlwaysAndNeverDraw,
-                    haveSubModel: "unknown",
+                    getModelVisibilityStatus: (modelProps) => this.getModelVisibilityStatus(modelProps),
                 }));
             }));
         });
         return createVisibilityHandlerResult(this, { node }, result, this._props.overrides?.getElementGroupingNodeDisplayStatus);
     }
-    getElementOverriddenVisibility(elementId, ignoreTooltip) {
-        const viewport = this._props.viewport;
-        if (viewport.neverDrawn?.has(elementId)) {
-            return createVisibilityStatus("hidden", getTooltipOptions("modelsTree.element.hiddenThroughNeverDrawnList", ignoreTooltip));
-        }
-        if (viewport.alwaysDrawn?.size) {
-            if (viewport.alwaysDrawn.has(elementId)) {
-                return createVisibilityStatus("visible", getTooltipOptions("modelsTree.element.displayedThroughAlwaysDrawnList", ignoreTooltip));
-            }
-            if (viewport.isAlwaysDrawnExclusive) {
-                return createVisibilityStatus("hidden", getTooltipOptions("modelsTree.element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn", ignoreTooltip));
-            }
-        }
-        return undefined;
-    }
-    getElementVisibility(ignoreTooltip, viewsModel, overridenVisibility, categoryVisibility, subModelVisibilityStatus) {
-        if (subModelVisibilityStatus === undefined) {
-            if (!viewsModel) {
-                return createVisibilityStatus("hidden", getTooltipOptions("modelsTree.element.hiddenThroughModel", ignoreTooltip));
-            }
-            if (overridenVisibility) {
-                return overridenVisibility;
-            }
-            return createVisibilityStatus(categoryVisibility.state, getTooltipOptions(categoryVisibility.state === "visible" ? undefined : "modelsTree.element.hiddenThroughCategory", ignoreTooltip));
-        }
-        if (subModelVisibilityStatus.state === "partial") {
-            return createVisibilityStatus("partial", getTooltipOptions("modelsTree.element.someElementsAreHidden", ignoreTooltip));
-        }
-        if (subModelVisibilityStatus.state === "visible") {
-            if (!viewsModel || overridenVisibility?.state === "hidden" || (categoryVisibility.state === "hidden" && !overridenVisibility)) {
-                return createVisibilityStatus("partial", getTooltipOptions("modelsTree.element.partialThroughSubModel", ignoreTooltip));
-            }
-            return createVisibilityStatus("visible", getTooltipOptions(undefined, ignoreTooltip));
-        }
-        if (!viewsModel) {
-            return createVisibilityStatus("hidden", getTooltipOptions("modelsTree.element.hiddenThroughModel", ignoreTooltip));
-        }
-        if (overridenVisibility) {
-            if (overridenVisibility.state === "hidden") {
-                return overridenVisibility;
-            }
-            return createVisibilityStatus("partial", getTooltipOptions("modelsTree.element.partialThroughElement", ignoreTooltip));
-        }
-        if (categoryVisibility.state === "visible") {
-            return createVisibilityStatus("partial", getTooltipOptions("modelsTree.element.partialThroughCategory", ignoreTooltip));
-        }
-        return createVisibilityStatus("hidden", getTooltipOptions("modelsTree.element.hiddenThroughCategory", ignoreTooltip));
-    }
     getElementDisplayStatus({ ignoreTooltip, ...props }) {
         const result = defer(() => {
             const viewport = this._props.viewport;
             const { elementId, modelId, categoryId } = props;
             const viewsModel = viewport.view.viewsModel(modelId);
-            const elementStatus = this.getElementOverriddenVisibility(elementId, ignoreTooltip);
-            return from(this._idsCache.hasSubModel(elementId)).pipe(mergeMap((hasSubModel) => (hasSubModel ? this.getModelVisibilityStatus({ modelId: elementId }) : of(undefined))), map((subModelVisibilityStatus) => this.getElementVisibility(ignoreTooltip, viewsModel, elementStatus, this.getDefaultCategoryVisibilityStatus({ categoryId, modelId, ignoreTooltip: true }), subModelVisibilityStatus)));
+            const elementStatus = getElementOverriddenVisibility({
+                elementId,
+                ignoreTooltip,
+                viewport,
+                tooltips: {
+                    visibileThorughAlwaysDrawn: "modelsTree.element.displayedThroughAlwaysDrawnList",
+                    hiddenThroughAlwaysDrawnExclusive: "modelsTree.element.hiddenDueToOtherElementsExclusivelyAlwaysDrawn",
+                    hiddenThroughNeverDrawn: "modelsTree.element.hiddenThroughNeverDrawnList",
+                },
+            });
+            return from(this._idsCache.hasSubModel(elementId)).pipe(mergeMap((hasSubModel) => (hasSubModel ? this.getModelVisibilityStatus({ modelId: elementId }) : of(undefined))), map((subModelVisibilityStatus) => getElementVisibility(ignoreTooltip, viewsModel, elementStatus, this.getDefaultCategoryVisibilityStatus({ categoryId, modelId, ignoreTooltip: true }), "modelsTree", subModelVisibilityStatus)));
         });
         return createVisibilityHandlerResult(this, props, result, this._props.overrides?.getElementDisplayStatus);
     }
@@ -369,7 +343,7 @@ class ModelsTreeVisibilityHandlerImpl {
         return filteredTree ? filteredTree.getVisibilityChangeTargets(node) : {};
     }
     changeFilteredNodeVisibility({ on, ...props }) {
-        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ subjects, models, categories, elements }) => {
+        return from(this.getVisibilityChangeTargets(props)).pipe(mergeMap(({ subjectIds: subjects, modelIds: models, categories, elements }) => {
             const observables = new Array();
             if (subjects?.size) {
                 observables.push(this.changeSubjectNodeState([...subjects], on));
@@ -392,14 +366,14 @@ class ModelsTreeVisibilityHandlerImpl {
             return merge(...observables);
         }));
     }
-    changeSubjectNodeState(ids, on) {
+    changeSubjectNodeState(subjectIds, on) {
         const result = defer(() => {
             if (!this._props.viewport.view.isSpatialView()) {
                 return EMPTY;
             }
-            return from(this._idsCache.getSubjectModelIds(ids)).pipe(mergeMap((modelIds) => this.changeModelState({ ids: modelIds, on })));
+            return from(this._idsCache.getSubjectModelIds(subjectIds)).pipe(mergeMap((modelIds) => this.changeModelState({ ids: modelIds, on })));
         });
-        return createVisibilityHandlerResult(this, { ids, on }, result, this._props.overrides?.changeSubjectNodeState);
+        return createVisibilityHandlerResult(this, { ids: subjectIds, on }, result, this._props.overrides?.changeSubjectNodeState);
     }
     changeModelState(props) {
         const { ids, on } = props;
@@ -414,13 +388,13 @@ class ModelsTreeVisibilityHandlerImpl {
             const idsObs = from(Id64.iterable(ids));
             if (!on) {
                 viewport.changeModelDisplay(ids, false);
-                return idsObs.pipe(mergeMap(async (modelId) => ({ modelId, categoryIds: await this._idsCache.getModelCategories(modelId) })), mergeMap(({ modelId, categoryIds }) => from(this._idsCache.getCategoriesModeledElements(modelId, categoryIds))), mergeMap((modeledElementIds) => this.changeModelState({ ids: modeledElementIds, on })));
+                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(defer(() => {
                 viewport.perModelCategoryVisibility.clearOverrides(ids);
                 return from(viewport.addViewedModels(ids));
             }), idsObs.pipe(mergeMap((modelId) => {
-                return from(this._idsCache.getModelCategories(modelId)).pipe(concatAll(), mergeMap((categoryId) => this.changeCategoryState({ categoryId, modelId, on: true })));
+                return from(this._idsCache.getModelCategoryIds(modelId)).pipe(concatAll(), mergeMap((categoryId) => this.changeCategoryState({ categoryId, modelId, on: true })));
             })));
         });
         return createVisibilityHandlerResult(this, props, result, this._props.overrides?.changeModelState);
@@ -428,8 +402,8 @@ class ModelsTreeVisibilityHandlerImpl {
     showModelWithoutAnyCategoriesOrElements(modelId) {
         const viewport = this._props.viewport;
         return forkJoin({
-            categories: this._idsCache.getModelCategories(modelId),
-            alwaysDrawnElements: this.getAlwaysDrawnElements({ modelId }),
+            categories: this._idsCache.getModelCategoryIds(modelId),
+            alwaysDrawnElements: this._alwaysAndNeverDrawnElements.getAlwaysDrawnElements({ modelId }),
         }).pipe(mergeMap(async ({ categories, alwaysDrawnElements }) => {
             const alwaysDrawn = this._props.viewport.alwaysDrawn;
             if (alwaysDrawn && alwaysDrawnElements) {
@@ -462,7 +436,7 @@ class ModelsTreeVisibilityHandlerImpl {
             const { modelId, categoryId, on } = props;
             return concat(props.on && !viewport.view.viewsModel(modelId) ? this.showModelWithoutAnyCategoriesOrElements(modelId) : EMPTY, defer(() => {
                 this.changeCategoryStateInViewportAccordingToModelVisibility(modelId, categoryId, on);
-                return this.clearAlwaysAndNeverDrawnElements(props);
+                return this._alwaysAndNeverDrawnElements.clearAlwaysAndNeverDrawnElements({ categoryIds: [props.categoryId], modelId: props.modelId });
             }), from(this._idsCache.getCategoriesModeledElements(modelId, [categoryId])).pipe(mergeMap((modeledElementIds) => this.changeModelState({ ids: modeledElementIds, on }))));
         });
         return createVisibilityHandlerResult(this, props, result, this._props.overrides?.changeCategoryState);
@@ -500,7 +474,7 @@ class ModelsTreeVisibilityHandlerImpl {
         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), this.changeElementStateNoChildrenOperator({ on, isDisplayedByDefault: visibleByDefault }), tap({
+        takeUntil(changeFinished), changeElementStateNoChildrenOperator({ on, isDisplayedByDefault: visibleByDefault, viewport: this._props.viewport }), tap({
             next: () => {
                 // notify that visibility change is finished
                 finishedSubject.next(true);
@@ -516,65 +490,6 @@ class ModelsTreeVisibilityHandlerImpl {
             },
         }), map(() => undefined));
     }
-    changeElementStateNoChildrenOperator(props) {
-        return (elementIds) => {
-            const { on, isDisplayedByDefault } = props;
-            const isAlwaysDrawnExclusive = this._props.viewport.isAlwaysDrawnExclusive;
-            return elementIds.pipe(releaseMainThreadOnItemsCount(500), reduce((acc, elementId) => {
-                if (acc.alwaysDrawn === undefined || acc.neverDrawn === undefined) {
-                    acc.alwaysDrawn = new Set(this._props.viewport.alwaysDrawn || []);
-                    acc.neverDrawn = new Set(this._props.viewport.neverDrawn || []);
-                }
-                if (on) {
-                    const wasRemoved = acc.neverDrawn.delete(elementId);
-                    acc.changedNeverDrawn ||= wasRemoved;
-                    // If exclusive mode is enabled, we must add the element to the always drawn list.
-                    if ((!isDisplayedByDefault || isAlwaysDrawnExclusive) && !acc.alwaysDrawn.has(elementId)) {
-                        acc.alwaysDrawn.add(elementId);
-                        acc.changedAlwaysDrawn = true;
-                    }
-                }
-                else {
-                    const wasRemoved = acc.alwaysDrawn.delete(elementId);
-                    acc.changedAlwaysDrawn ||= wasRemoved;
-                    // If exclusive mode is not enabled, we have to add the element to the never drawn list.
-                    if (isDisplayedByDefault && !isAlwaysDrawnExclusive && !acc.neverDrawn.has(elementId)) {
-                        acc.neverDrawn.add(elementId);
-                        acc.changedNeverDrawn = true;
-                    }
-                }
-                return acc;
-            }, {
-                changedNeverDrawn: false,
-                changedAlwaysDrawn: false,
-                neverDrawn: undefined,
-                alwaysDrawn: undefined,
-            }), map((state) => {
-                state.changedNeverDrawn && state.neverDrawn && this._props.viewport.setNeverDrawn(state.neverDrawn);
-                state.changedAlwaysDrawn && state.alwaysDrawn && this._props.viewport.setAlwaysDrawn(state.alwaysDrawn, this._props.viewport.isAlwaysDrawnExclusive);
-            }));
-        };
-    }
-    getVisibilityFromAlwaysAndNeverDrawnElementsImpl(props) {
-        const { alwaysDrawn, neverDrawn, totalCount, ignoreTooltip } = props;
-        if (neverDrawn?.size === totalCount) {
-            return createVisibilityStatus("hidden", getTooltipOptions(props.tooltips.allElementsInNeverDrawnList, ignoreTooltip));
-        }
-        if (alwaysDrawn?.size === totalCount) {
-            return createVisibilityStatus("visible", getTooltipOptions(props.tooltips.allElementsInAlwaysDrawnList, ignoreTooltip));
-        }
-        const viewport = this._props.viewport;
-        if (viewport.isAlwaysDrawnExclusive && viewport.alwaysDrawn?.size) {
-            return alwaysDrawn?.size
-                ? createVisibilityStatus("partial", getTooltipOptions(props.tooltips.elementsInBothAlwaysAndNeverDrawn, ignoreTooltip))
-                : createVisibilityStatus("hidden", getTooltipOptions(props.tooltips.noElementsInExclusiveAlwaysDrawnList, ignoreTooltip));
-        }
-        const status = props.defaultStatus();
-        if ((status.state === "visible" && neverDrawn?.size) || (status.state === "hidden" && alwaysDrawn?.size)) {
-            return createVisibilityStatus("partial", getTooltipOptions(undefined, ignoreTooltip));
-        }
-        return status;
-    }
     getVisibilityFromAlwaysAndNeverDrawnElements({ ignoreTooltip, ...props }) {
         const viewport = this._props.viewport;
         if (viewport.isAlwaysDrawnExclusive) {
@@ -586,48 +501,31 @@ class ModelsTreeVisibilityHandlerImpl {
             return of(props.defaultStatus());
         }
         if ("elements" in props) {
-            return of(this.getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
+            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,
                 ignoreTooltip,
+                viewport,
             }));
         }
-        const { modelId, categoryId } = props.queryProps;
-        const totalCount = categoryId ? this._idsCache.getCategoryElementsCount(modelId, categoryId) : this._idsCache.getModelElementCount(modelId);
+        const { modelId, categoryIds } = props.queryProps;
+        assert(modelId !== undefined);
+        const totalCount = this._idsCache.getCategoryElementsCount(modelId, categoryIds[0]);
         return forkJoin({
             totalCount,
-            alwaysDrawn: this.getAlwaysDrawnElements(props.queryProps),
-            neverDrawn: this.getNeverDrawnElements(props.queryProps),
+            alwaysDrawn: this._alwaysAndNeverDrawnElements.getAlwaysDrawnElements(props.queryProps),
+            neverDrawn: this._alwaysAndNeverDrawnElements.getNeverDrawnElements(props.queryProps),
         }).pipe(map((state) => {
-            return this.getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
+            return getVisibilityFromAlwaysAndNeverDrawnElementsImpl({
                 ...props,
                 ...state,
+                viewport,
                 ignoreTooltip,
             });
         }));
     }
-    getAlwaysDrawnElements(props) {
-        return this._alwaysAndNeverDrawnElements.getElements({ ...props, setType: "always" });
-    }
-    getNeverDrawnElements(props) {
-        return this._alwaysAndNeverDrawnElements.getElements({ ...props, setType: "never" });
-    }
-    clearAlwaysAndNeverDrawnElements(props) {
-        return forkJoin({
-            alwaysDrawn: this.getAlwaysDrawnElements(props),
-            neverDrawn: this.getNeverDrawnElements(props),
-        }).pipe(map(({ alwaysDrawn, neverDrawn }) => {
-            const viewport = this._props.viewport;
-            if (viewport.alwaysDrawn?.size && alwaysDrawn.size) {
-                viewport.setAlwaysDrawn(setDifference(viewport.alwaysDrawn, alwaysDrawn));
-            }
-            if (viewport.neverDrawn?.size && neverDrawn.size) {
-                viewport.setNeverDrawn(setDifference(viewport.neverDrawn, neverDrawn));
-            }
-        }));
-    }
     getGroupingNodeInfo(node) {
         const modelId = ModelsTreeNode.getModelId(node);
         const categoryId = ModelsTreeNode.getCategoryId(node);
@@ -635,119 +533,5 @@ class ModelsTreeVisibilityHandlerImpl {
         const elementIds = new Set(node.groupedInstanceKeys.map((key) => key.id));
         return { modelId, categoryId, elementIds };
     }
-    getSubModeledElementsVisibilityStatus({ parentNodeVisibilityStatus, haveSubModel, tooltips, ignoreTooltips, }) {
-        return (obs) => {
-            return obs.pipe(
-            // ensure we're only looking at elements that have a sub-model
-            mergeMap((modeledElementIds) => {
-                if (haveSubModel === "yes") {
-                    return of(modeledElementIds);
-                }
-                return from(modeledElementIds).pipe(mergeMap(async (elementId) => ({ elementId, hasSubModel: await this._idsCache.hasSubModel(elementId) })), filter(({ hasSubModel }) => hasSubModel), map(({ elementId }) => elementId), toArray());
-            }), 
-            // combine visibility status of sub-models with visibility status of parent node
-            mergeMap((modeledElementIds) => {
-                if (modeledElementIds.length === 0) {
-                    return of(parentNodeVisibilityStatus);
-                }
-                return from(modeledElementIds).pipe(mergeMap((modeledElementId) => this.getModelVisibilityStatus({ modelId: modeledElementId })), startWith(parentNodeVisibilityStatus), map((visibilityStatus) => visibilityStatus.state), getVisibilityStatusFromTreeNodeChildren(tooltips, ignoreTooltips));
-            }));
-        };
-    }
-}
-function getVisibilityFromTreeNodeChildren(obs) {
-    return obs.pipe(reduceWhile((x) => x.allVisible || x.allHidden, (acc, val) => {
-        acc.allVisible &&= val === "visible";
-        acc.allHidden &&= val === "hidden";
-        return acc;
-    }, { allVisible: true, allHidden: true }), map((x) => {
-        if (!x) {
-            return "empty";
-        }
-        return x.allVisible ? "visible" : x.allHidden ? "hidden" : "partial";
-    }));
-}
-function getVisibilityStatusFromTreeNodeChildren(tooltipMap, ignoreTooltip) {
-    return (obs) => {
-        return getVisibilityFromTreeNodeChildren(obs).pipe(map((visibility) => {
-            if (visibility === "empty") {
-                visibility = "visible";
-            }
-            return createVisibilityStatus(visibility, getTooltipOptions(tooltipMap[visibility], ignoreTooltip));
-        }));
-    };
-}
-function setDifference(lhs, rhs) {
-    const result = new Set();
-    lhs.forEach((x) => !rhs.has(x) && result.add(x));
-    return result;
-}
-function setIntersection(lhs, rhs) {
-    const result = new Set();
-    lhs.forEach((x) => rhs.has(x) && result.add(x));
-    return result;
-}
-/**
- * Enables display of all given models. Also enables display of all categories and clears always and
- * never drawn lists in the viewport.
- * @public
- */
-export async function showAllModels(models, viewport) {
-    await viewport.addViewedModels(models);
-    viewport.clearNeverDrawn();
-    viewport.clearAlwaysDrawn();
-    await toggleAllCategories(viewport, true);
-}
-/**
- * Disables display of all given models.
- * @public
- */
-export async function hideAllModels(models, viewport) {
-    viewport.changeModelDisplay(models, false);
-}
-/**
- * Inverts display of all given models.
- * @public
- */
-export async function invertAllModels(models, viewport) {
-    const notViewedModels = [];
-    const viewedModels = [];
-    models.forEach((modelId) => {
-        if (viewport.viewsModel(modelId)) {
-            viewedModels.push(modelId);
-        }
-        else {
-            notViewedModels.push(modelId);
-        }
-    });
-    await viewport.addViewedModels(notViewedModels);
-    viewport.changeModelDisplay(viewedModels, false);
-}
-/**
- * Based on the value of `enable` argument, either enables or disables display of given models.
- * @public
- */
-export async function toggleModels(models, enable, viewport) {
-    if (!models) {
-        return;
-    }
-    if (enable) {
-        viewport.changeModelDisplay(models, false);
-    }
-    else {
-        await viewport.addViewedModels(models);
-    }
-}
-/**
- * Checks if all given models are displayed in given viewport.
- * @public
- */
-export function areAllModelsVisible(models, viewport) {
-    return models.length !== 0 ? models.every((id) => viewport.viewsModel(id)) : false;
-}
-function getTooltipOptions(key, ignoreTooltip) {
-    return {
-        useTooltip: ignoreTooltip ? false : key,
-    };
 }
 //# sourceMappingURL=ModelsTreeVisibilityHandler.js.map