@itwin/tree-widget-react 4.0.0-alpha.25 → 4.0.0-alpha.26

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

@@ -2,12 +2,14 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { concat, defer, from, map, merge, mergeMap, of } from "rxjs";
-import { assert, Guid, Id64 } from "@itwin/core-bentley";
+import { concat, defer, EMPTY, from, map, merge, mergeAll, mergeMap, of } from "rxjs";
+import { assert, Guid } from "@itwin/core-bentley";
+import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
 import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
-import { fromWithRelease, getClassesByView } from "../../../common/internal/Utils.js";
+import { fromWithRelease, getClassesByView, getIdsFromChildrenTree, getParentElementsIdsPath, setDifference, setIntersection, } from "../../../common/internal/Utils.js";
 import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
-import { CategoriesTreeNode } from "../../CategoriesTreeNode.js";
+import { CategoriesTreeNodeInternal } from "../../internal/CategoriesTreeNodeInternal.js";
 import { CategoriesTreeVisibilityHelper } from "./CategoriesTreeVisibilityHelper.js";
 import { createCategoriesSearchResultsTree } from "./SearchResultsTree.js";
 /**
@@ -19,38 +21,24 @@ import { createCategoriesSearchResultsTree } from "./SearchResultsTree.js";
 export class CategoriesTreeVisibilityHandler {
     #props;
     #visibilityHelper;
-    #elementType;
-    #categoryType;
-    #modelType;
     constructor(constructorProps) {
         this.#props = constructorProps;
-        // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.
-        // We won't need to create a custom base ids cache.
-        const baseIdsCache = {
-            getCategories: (props) => this.getCategories(props),
-            getAllCategories: () => this.getAllCategories(),
-            getElementsCount: (props) => this.getElementsCount(props),
-            getModels: (props) => this.getModels(props),
-            getSubCategories: (props) => this.getSubCategories(props),
-            getSubModels: (props) => this.getSubModels(props),
-            hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),
-        };
         this.#visibilityHelper = new CategoriesTreeVisibilityHelper({
             viewport: this.#props.viewport,
             idsCache: this.#props.idsCache,
             alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
-            baseIdsCache,
+            baseIdsCache: constructorProps.idsCache,
             hierarchyConfig: constructorProps.hierarchyConfig,
         });
-        this.#elementType = this.#props.viewport.viewType === "2d" ? "GeometricElement2d" : "GeometricElement3d";
-        this.#categoryType = this.#props.viewport.viewType === "2d" ? "DrawingCategory" : "SpatialCategory";
-        this.#modelType = this.#props.viewport.viewType === "2d" ? "GeometricModel2d" : "GeometricModel3d";
     }
     [Symbol.dispose]() {
         this.#visibilityHelper[Symbol.dispose]();
     }
     changeSearchTargetsVisibilityStatus(targets, on) {
         return defer(() => {
+            if (this.#props.viewport.viewType === "other") {
+                return EMPTY;
+            }
             const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;
             const observables = new Array();
             if (definitionContainerIds?.size) {
@@ -66,100 +54,174 @@ export class CategoriesTreeVisibilityHandler {
                 observables.push(from(subCategories).pipe(mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.changeSubCategoriesVisibilityStatus({ subCategoryIds, categoryId, on }))));
             }
             if (elements?.length) {
-                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }))));
+                const searchTargetElements = new Array();
+                const elementIdsSet = new Set();
+                const modelCategoryElementMap = new Map();
+                // elements is an array that stores elements grouped by:
+                // 1. Their path
+                // 2. Their modelId and categoryId
+                // When changing visibility of elements, visibility handler does not care about the path.
+                // So we can first get all elements and group them only by modelId and categoryId.
+                for (const { elements: elementsMap, categoryId, modelId } of elements) {
+                    const key = `${modelId}-${categoryId}`;
+                    let mapEntry = modelCategoryElementMap.get(key);
+                    if (!mapEntry) {
+                        mapEntry = [];
+                        modelCategoryElementMap.set(key, mapEntry);
+                    }
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        mapEntry.push(elementId);
+                        elementIdsSet.add(elementId);
+                        if (isSearchTarget) {
+                            searchTargetElements.push(elementId);
+                        }
+                    }
+                }
+                observables.push(
+                // Get children for search targets, since non search targets don't have all the children present in the hierarchy.
+                this.#props.idsCache.getChildElementsTree({ elementIds: searchTargetElements }).pipe(
+                // Need to filter out and keep only those children ids that are not part of elements that are present in search paths.
+                // Elements in search paths will have their visibility changed directly: they will be provided as elementIds to changeElementsVisibilityStatus.
+                map((childrenTree) => ({
+                    childrenNotInSearchPaths: setDifference(getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 }), elementIdsSet),
+                    childrenTree,
+                })), mergeMap(({ childrenNotInSearchPaths, childrenTree }) => fromWithRelease({ source: modelCategoryElementMap.entries(), size: modelCategoryElementMap.size, releaseOnCount: 50 }).pipe(mergeMap(([key, elementsInSearchPathsGroupedByModelAndCategory]) => {
+                    const [modelId, categoryId] = key.split("-");
+                    const childrenIds = new Set();
+                    // A shared children tree was created, need to get the children for each element in the group.
+                    for (const elementId of elementsInSearchPathsGroupedByModelAndCategory) {
+                        const elementChildrenTree = childrenTree.get(elementId)?.children;
+                        if (!elementChildrenTree) {
+                            continue;
+                        }
+                        for (const childId of getIdsFromChildrenTree({ tree: elementChildrenTree })) {
+                            childrenIds.add(childId);
+                        }
+                    }
+                    return this.#visibilityHelper.changeElementsVisibilityStatus({
+                        modelId,
+                        categoryId,
+                        elementIds: elementsInSearchPathsGroupedByModelAndCategory,
+                        // Pass only those children that are not part of search paths.
+                        children: setIntersection(childrenIds, childrenNotInSearchPaths),
+                        on,
+                    });
+                })))));
             }
             return merge(...observables);
         });
     }
     getVisibilityStatus(node) {
-        if (CategoriesTreeNode.isElementClassGroupingNode(node)) {
+        if (this.#props.viewport.viewType === "other") {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {
             return this.#visibilityHelper.getGroupedElementsVisibilityStatus({
                 categoryId: node.extendedData.categoryId,
                 modelElementsMap: node.extendedData.modelElementsMap,
+                parentKeys: node.parentKeys,
+                childrenCount: node.extendedData.childrenCount,
+                topMostParentElementId: node.extendedData.topMostParentElementId,
             });
         }
-        if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+        if (CategoriesTreeNodeInternal.isDefinitionContainerNode(node)) {
             return this.#visibilityHelper.getDefinitionContainersVisibilityStatus({
                 definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
             });
         }
-        if (CategoriesTreeNode.isModelNode(node)) {
+        if (CategoriesTreeNodeInternal.isModelNode(node)) {
             return this.#visibilityHelper.getModelsVisibilityStatus({
                 modelIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
-                type: this.#modelType,
             });
         }
-        if (CategoriesTreeNode.isCategoryNode(node)) {
+        if (CategoriesTreeNodeInternal.isCategoryNode(node)) {
             return this.#visibilityHelper.getCategoriesVisibilityStatus({
                 categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                 modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,
-                type: this.#categoryType,
             });
         }
-        if (CategoriesTreeNode.isSubCategoryNode(node)) {
+        if (CategoriesTreeNodeInternal.isSubCategoryNode(node)) {
             return this.#visibilityHelper.getSubCategoriesVisibilityStatus({
                 categoryId: node.extendedData.categoryId,
                 subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
             });
         }
-        assert(CategoriesTreeNode.isElementNode(node));
+        assert(CategoriesTreeNodeInternal.isElementNode(node));
+        const parentElementsIdsPath = getParentElementsIdsPath({
+            parentInstanceKeys: node.parentKeys.filter((parentKey) => HierarchyNodeKey.isInstances(parentKey)).map((key) => key.instanceKeys),
+            topMostParentElementId: node.extendedData.topMostParentElementId,
+        });
         return this.#visibilityHelper.getElementsVisibilityStatus({
             elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
             modelId: node.extendedData.modelId,
             categoryId: node.extendedData.categoryId,
-            type: this.#elementType,
+            parentElementsIdsPath,
+            childrenCount: node.extendedData.childrenCount,
+            categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,
         });
     }
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node, on) {
         const changeObs = defer(() => {
-            if (CategoriesTreeNode.isElementClassGroupingNode(node)) {
+            if (this.#props.viewport.viewType === "other") {
+                return EMPTY;
+            }
+            if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {
                 return this.#visibilityHelper.changeGroupedElementsVisibilityStatus({
                     categoryId: node.extendedData.categoryId,
                     modelElementsMap: node.extendedData.modelElementsMap,
                     on,
                 });
             }
-            if (CategoriesTreeNode.isDefinitionContainerNode(node)) {
+            if (CategoriesTreeNodeInternal.isDefinitionContainerNode(node)) {
                 return this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({
                     definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                     on,
                 });
             }
-            if (CategoriesTreeNode.isModelNode(node)) {
+            if (CategoriesTreeNodeInternal.isModelNode(node)) {
                 return this.#visibilityHelper.changeModelsVisibilityStatus({
                     modelIds: node.key.instanceKeys.map(({ id }) => id),
                     on,
                 });
             }
-            if (CategoriesTreeNode.isCategoryNode(node)) {
+            if (CategoriesTreeNodeInternal.isCategoryNode(node)) {
                 return this.#visibilityHelper.changeCategoriesVisibilityStatus({
                     categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                     modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,
                     on,
                 });
             }
-            if (CategoriesTreeNode.isSubCategoryNode(node)) {
+            if (CategoriesTreeNodeInternal.isSubCategoryNode(node)) {
                 return this.#visibilityHelper.changeSubCategoriesVisibilityStatus({
                     categoryId: node.extendedData.categoryId,
                     subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
                     on,
                 });
             }
-            assert(CategoriesTreeNode.isElementNode(node));
-            return this.#visibilityHelper.changeElementsVisibilityStatus({
-                elementIds: node.key.instanceKeys.map(({ id }) => id),
+            assert(CategoriesTreeNodeInternal.isElementNode(node));
+            const elementIds = node.key.instanceKeys.map(({ id }) => id);
+            return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(map((childrenTree) => {
+                // Children tree contains provided elementIds, they are at the root of this tree.
+                // We want to skip them and only get ids of children.
+                return getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 });
+            }), mergeMap((children) => this.#visibilityHelper.changeElementsVisibilityStatus({
+                elementIds,
                 modelId: node.extendedData.modelId,
+                children: children.size > 0 ? children : undefined,
                 categoryId: node.extendedData.categoryId,
                 on,
-            });
+            })));
         });
         if (this.#props.viewport.isAlwaysDrawnExclusive) {
             return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);
         }
         return changeObs;
     }
-    getSearchTargetsVisibilityStatus(targets) {
+    getSearchTargetsVisibilityStatus(targets, node) {
+        if (this.#props.viewport.viewType === "other") {
+            return of(createVisibilityStatus("disabled"));
+        }
         return defer(() => {
             const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;
             const observables = new Array();
@@ -167,65 +229,95 @@ export class CategoriesTreeVisibilityHandler {
                 observables.push(this.#visibilityHelper.getDefinitionContainersVisibilityStatus({ definitionContainerIds }));
             }
             if (modelIds?.size) {
-                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: this.#modelType }));
+                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds }));
             }
             if (categories?.length) {
                 observables.push(from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.getCategoriesVisibilityStatus({
                     categoryIds,
                     modelId,
-                    type: this.#categoryType,
                 }))));
             }
             if (subCategories?.length) {
                 observables.push(from(subCategories).pipe(mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.getSubCategoriesVisibilityStatus({ subCategoryIds, categoryId }))));
             }
             if (elements?.length) {
-                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: this.#elementType }))));
-            }
-            return merge(...observables);
-        }).pipe(mergeVisibilityStatuses);
-    }
-    getCategories(props) {
-        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categories) => ({
-            id: modelId,
-            ...(this.#props.viewport.viewType === "2d" ? { drawingCategories: categories } : { spatialCategories: categories }),
-        })))));
-    }
-    getAllCategories() {
-        return this.#props.idsCache.getAllCategories().pipe(map((categories) => {
-            if (this.#props.viewport.viewType === "2d") {
-                return { drawingCategories: categories };
-            }
-            return { spatialCategories: categories };
-        }));
-    }
-    getElementsCount(props) {
-        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
-    }
-    getModels(props) {
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds, true);
-    }
-    getSubCategories(props) {
-        return this.#props.idsCache.getSubCategories(props.categoryId);
-    }
-    getSubModels(props) {
-        if ("modelIds" in props) {
-            return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => {
-                if (props.categoryId) {
-                    return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));
+                const searchTargetElements = new Array();
+                for (const { elements: elementsMap } of elements) {
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        if (isSearchTarget) {
+                            searchTargetElements.push(elementId);
+                        }
+                    }
                 }
-                return this.#props.idsCache.getModelCategoryIds(modelId).pipe(mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)), map((subModels) => ({ id: modelId, subModels })));
-            }));
-        }
-        if (props.modelId) {
-            return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesModeledElements(props.modelId, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels })))));
-        }
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(mergeMap(({ id, models }) => {
-            if (!models) {
-                return of({ id, subModels: undefined });
+                let childrenCountMapObs;
+                if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {
+                    const groupingNodesSearchTargets = node.extendedData?.searchTargets;
+                    const nestedSearchTargetElements = searchTargetElements.filter((searchTarget) => !groupingNodesSearchTargets?.has(searchTarget));
+                    // Only need to request children count for indirect children search targets.
+                    // Direct children search targets already have children count stored in grouping nodes extended data.
+                    childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: nestedSearchTargetElements }).pipe(map((elementCountMap) => {
+                        // Direct children search targets already have children count stored in grouping nodes extended data.
+                        for (const [key, value] of node.extendedData.searchTargets ?? []) {
+                            elementCountMap.set(key, value.childrenCount);
+                        }
+                        return elementCountMap;
+                    }));
+                }
+                else {
+                    childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: searchTargetElements });
+                }
+                observables.push(childrenCountMapObs.pipe(mergeMap((elementsChildrenCountMap) => fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId, pathToElements, topMostParentElementId }) => {
+                    const parentElementsIdsPath = topMostParentElementId
+                        ? getParentElementsIdsPath({
+                            parentInstanceKeys: pathToElements.map((instanceKey) => [instanceKey]),
+                            topMostParentElementId,
+                        })
+                        : [];
+                    let totalSearchTargetsChildrenCount = 0;
+                    const nonSearchTargetIds = new Array();
+                    const searchTargetIds = new Array();
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        if (!isSearchTarget) {
+                            nonSearchTargetIds.push(elementId);
+                            continue;
+                        }
+                        searchTargetIds.push(elementId);
+                        const childCount = elementsChildrenCountMap.get(elementId);
+                        if (childCount) {
+                            totalSearchTargetsChildrenCount += childCount;
+                        }
+                    }
+                    return merge(searchTargetIds.length > 0
+                        ? this.#visibilityHelper.getElementsVisibilityStatus({
+                            modelId,
+                            categoryId,
+                            elementIds: searchTargetIds,
+                            parentElementsIdsPath,
+                            childrenCount: totalSearchTargetsChildrenCount,
+                            // Search results tree is created on search paths. Since search paths contain only categories that are directly under models
+                            // or at the root, categoryId can be used here here.
+                            categoryOfTopMostParentElement: categoryId,
+                        })
+                        : EMPTY, 
+                    // Set childrenCount to undefined for non search targets, as some of their child elements might be filtered out.
+                    // Since childrenCount is set to undefined, these elements won't check child always/never drawn child elements status.
+                    // Child always/never drawn elements will be in search paths, and their visibility status will be handled separately.
+                    nonSearchTargetIds.length > 0
+                        ? this.#visibilityHelper.getElementsVisibilityStatus({
+                            modelId,
+                            categoryId,
+                            elementIds: nonSearchTargetIds,
+                            parentElementsIdsPath,
+                            childrenCount: undefined,
+                            // Search results tree is created on search paths. Since search paths contain only categories that are directly under models
+                            // or at the root, categoryId can be used here here.
+                            categoryOfTopMostParentElement: categoryId,
+                        })
+                        : EMPTY).pipe(mergeVisibilityStatuses());
+                })))));
             }
-            return from(models).pipe(mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)), map((subModels) => ({ id, subModels })));
-        }));
+            return from(observables).pipe(mergeAll(), mergeVisibilityStatuses());
+        });
     }
 }
 /**

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,8BAA8B,EAAE,MAAM,8DAA8D,CAAC;AAC9G,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AACjE,OAAO,EAAE,8BAA8B,EAAE,MAAM,qCAAqC,CAAC;AACrF,OAAO,EAAE,iCAAiC,EAAE,MAAM,wBAAwB,CAAC;AAsB3E;;;;;GAKG;AACH,MAAM,OAAO,+BAA+B;IACjC,MAAM,CAAuC;IACtD,iBAAiB,CAAiC;IAClD,YAAY,CAA8C;IAC1D,aAAa,CAAwC;IACrD,UAAU,CAA0C;IACpD,YAAY,gBAAsD;QAChE,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,6EAA6E;QAC7E,mDAAmD;QACnD,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YACnD,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAC/C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAC3C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACjD,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;SAChE,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,8BAA8B,CAAC;YAC1D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,YAAY;YACZ,eAAe,EAAE,gBAAgB,CAAC,eAAe;SAClD,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,oBAAoB,CAAC;QACzG,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,iBAAiB,CAAC;QACpG,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC;IACrG,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAoC,EAAE,EAAW;QAC1F,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,sBAAsB,EAAE,aAAa,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC1F,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC,EAAE,sBAAsB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACtH,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CACrJ,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,mCAAmC,CAAC,EAAE,cAAc,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,CAC7I,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACxH,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,kBAAkB,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;YACxD,OAAO,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBAC/D,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,gBAAgB;aACrD,CAAC,CAAC;QACL,CAAC;QAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC;gBACpE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aACnF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,kBAAkB,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC;gBACtD,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpE,IAAI,EAAE,IAAI,CAAC,UAAU;aACtB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;gBAC1D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;gBACvE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC3F,IAAI,EAAE,IAAI,CAAC,aAAa;aACzB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;gBAC7D,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aAC3E,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;YACtE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;YACxC,IAAI,EAAE,IAAI,CAAC,YAAY;SACxB,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,kBAAkB,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAClE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;oBACxC,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,gBAAgB;oBACpD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,kBAAkB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC;oBACvE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAClF,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,kBAAkB,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC;oBACzD,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;oBACnD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBAC7D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBACvE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;oBAC3F,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,mCAAmC,CAAC;oBAChE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;oBACxC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC1E,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,MAAM,CAAC,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;gBAClC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAoC;QAC1E,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,sBAAsB,EAAE,aAAa,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC1F,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;YAC/G,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAC1G,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;oBACnD,WAAW;oBACX,OAAO;oBACP,IAAI,EAAE,IAAI,CAAC,aAAa;iBACzB,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC,CAAC,CACtI,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAC1I,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnC,CAAC;IAEO,aAAa,CAAC,KAAmD;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CACnB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CACpD,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACnB,EAAE,EAAE,OAAO;YACX,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;SACpH,CAAC,CAAC,CACJ,CACF,CACF,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CACjD,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACjB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAC3C,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;YAC3C,CAAC;YACD,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC;QAC3C,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAEO,SAAS,CAAC,KAA+C;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAClF,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAC5E,QAAQ,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EACrF,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CACxC,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,qCAAqC,CAAC,KAMrD;IACC,OAAO,IAAI,8BAA8B,CAA8B;QACrE,oBAAoB,EAAE,GAAwE,EAAE;YAC9F,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACvB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACrH,OAAO,iCAAiC,CAAC;gBACvC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,iBAAiB,EAAE,aAAa;gBAChC,wBAAwB,EAAE,YAAY;gBACtC,sBAAsB,EAAE,UAAU;aACnC,CAAC,CAAC;QACL,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,EAAE;YACzC,OAAO,IAAI,+BAA+B,CAAC;gBACzC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe,EAAE,KAAK,CAAC,eAAe;aACvC,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;KAChC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, defer, from, map, merge, mergeMap, of } from \"rxjs\";\nimport { assert, Guid, Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyVisibilityHandlerImpl } from \"../../../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { fromWithRelease, getClassesByView } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { CategoriesTreeNode } from \"../../CategoriesTreeNode.js\";\nimport { CategoriesTreeVisibilityHelper } from \"./CategoriesTreeVisibilityHelper.js\";\nimport { createCategoriesSearchResultsTree } from \"./SearchResultsTree.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { HierarchyNode, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { SearchResultsTree } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { BaseIdsCache, TreeSpecificVisibilityHandler } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"../../CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\nimport type { CategoriesTreeSearchTargets } from \"./SearchResultsTree.js\";\n\n/** @internal */\nexport interface CategoriesTreeVisibilityHandlerProps {\n  idsCache: CategoriesTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}\n\n/**\n * Handles visibility status of categories tree nodes.\n *\n * This handler knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class CategoriesTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<CategoriesTreeSearchTargets> {\n  readonly #props: CategoriesTreeVisibilityHandlerProps;\n  #visibilityHelper: CategoriesTreeVisibilityHelper;\n  #elementType: \"GeometricElement3d\" | \"GeometricElement2d\";\n  #categoryType: \"DrawingCategory\" | \"SpatialCategory\";\n  #modelType: \"GeometricModel3d\" | \"GeometricModel2d\";\n  constructor(constructorProps: CategoriesTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.\n    // We won't need to create a custom base ids cache.\n    const baseIdsCache: BaseIdsCache = {\n      getCategories: (props) => this.getCategories(props),\n      getAllCategories: () => this.getAllCategories(),\n      getElementsCount: (props) => this.getElementsCount(props),\n      getModels: (props) => this.getModels(props),\n      getSubCategories: (props) => this.getSubCategories(props),\n      getSubModels: (props) => this.getSubModels(props),\n      hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),\n    };\n    this.#visibilityHelper = new CategoriesTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      baseIdsCache,\n      hierarchyConfig: constructorProps.hierarchyConfig,\n    });\n\n    this.#elementType = this.#props.viewport.viewType === \"2d\" ? \"GeometricElement2d\" : \"GeometricElement3d\";\n    this.#categoryType = this.#props.viewport.viewType === \"2d\" ? \"DrawingCategory\" : \"SpatialCategory\";\n    this.#modelType = this.#props.viewport.viewType === \"2d\" ? \"GeometricModel2d\" : \"GeometricModel3d\";\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeSearchTargetsVisibilityStatus(targets: CategoriesTreeSearchTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (definitionContainerIds?.size) {\n        observables.push(this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({ definitionContainerIds, on }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.changeCategoriesVisibilityStatus({ categoryIds, modelId, on }))),\n        );\n      }\n\n      if (subCategories?.length) {\n        observables.push(\n          from(subCategories).pipe(\n            mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.changeSubCategoriesVisibilityStatus({ subCategoryIds, categoryId, on })),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (CategoriesTreeNode.isElementClassGroupingNode(node)) {\n      return this.#visibilityHelper.getGroupedElementsVisibilityStatus({\n        categoryId: node.extendedData.categoryId,\n        modelElementsMap: node.extendedData.modelElementsMap,\n      });\n    }\n\n    if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n      return this.#visibilityHelper.getDefinitionContainersVisibilityStatus({\n        definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (CategoriesTreeNode.isModelNode(node)) {\n      return this.#visibilityHelper.getModelsVisibilityStatus({\n        modelIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n        type: this.#modelType,\n      });\n    }\n\n    if (CategoriesTreeNode.isCategoryNode(node)) {\n      return this.#visibilityHelper.getCategoriesVisibilityStatus({\n        categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n        modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,\n        type: this.#categoryType,\n      });\n    }\n\n    if (CategoriesTreeNode.isSubCategoryNode(node)) {\n      return this.#visibilityHelper.getSubCategoriesVisibilityStatus({\n        categoryId: node.extendedData.categoryId,\n        subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    assert(CategoriesTreeNode.isElementNode(node));\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      modelId: node.extendedData.modelId,\n      categoryId: node.extendedData.categoryId,\n      type: this.#elementType,\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (CategoriesTreeNode.isElementClassGroupingNode(node)) {\n        return this.#visibilityHelper.changeGroupedElementsVisibilityStatus({\n          categoryId: node.extendedData.categoryId,\n          modelElementsMap: node.extendedData.modelElementsMap,\n          on,\n        });\n      }\n\n      if (CategoriesTreeNode.isDefinitionContainerNode(node)) {\n        return this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({\n          definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      if (CategoriesTreeNode.isModelNode(node)) {\n        return this.#visibilityHelper.changeModelsVisibilityStatus({\n          modelIds: node.key.instanceKeys.map(({ id }) => id),\n          on,\n        });\n      }\n\n      if (CategoriesTreeNode.isCategoryNode(node)) {\n        return this.#visibilityHelper.changeCategoriesVisibilityStatus({\n          categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,\n          on,\n        });\n      }\n\n      if (CategoriesTreeNode.isSubCategoryNode(node)) {\n        return this.#visibilityHelper.changeSubCategoriesVisibilityStatus({\n          categoryId: node.extendedData.categoryId,\n          subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      assert(CategoriesTreeNode.isElementNode(node));\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId: node.extendedData.modelId,\n        categoryId: node.extendedData.categoryId,\n        on,\n      });\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getSearchTargetsVisibilityStatus(targets: CategoriesTreeSearchTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (definitionContainerIds?.size) {\n        observables.push(this.#visibilityHelper.getDefinitionContainersVisibilityStatus({ definitionContainerIds }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: this.#modelType }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.getCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                type: this.#categoryType,\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (subCategories?.length) {\n        observables.push(\n          from(subCategories).pipe(\n            mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.getSubCategoriesVisibilityStatus({ subCategoryIds, categoryId })),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: this.#elementType }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    }).pipe(mergeVisibilityStatuses);\n  }\n\n  private getCategories(props: Parameters<BaseIdsCache[\"getCategories\"]>[0]): ReturnType<BaseIdsCache[\"getCategories\"]> {\n    return from(Id64.iterable(props.modelIds)).pipe(\n      mergeMap((modelId) =>\n        this.#props.idsCache.getModelCategoryIds(modelId).pipe(\n          map((categories) => ({\n            id: modelId,\n            ...(this.#props.viewport.viewType === \"2d\" ? { drawingCategories: categories } : { spatialCategories: categories }),\n          })),\n        ),\n      ),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(\n      map((categories) => {\n        if (this.#props.viewport.viewType === \"2d\") {\n          return { drawingCategories: categories };\n        }\n        return { spatialCategories: categories };\n      }),\n    );\n  }\n\n  private getElementsCount(props: Parameters<BaseIdsCache[\"getElementsCount\"]>[0]): ReturnType<BaseIdsCache[\"getElementsCount\"]> {\n    return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);\n  }\n\n  private getModels(props: Parameters<BaseIdsCache[\"getModels\"]>[0]): ReturnType<BaseIdsCache[\"getModels\"]> {\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds, true);\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return this.#props.idsCache.getSubCategories(props.categoryId);\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return this.#props.idsCache.getModelCategoryIds(modelId).pipe(\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(\n      mergeMap(({ id, models }) => {\n        if (!models) {\n          return of({ id, subModels: undefined });\n        }\n        return from(models).pipe(\n          mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)),\n          map((subModels) => ({ id, subModels })),\n        );\n      }),\n    );\n  }\n}\n\n/**\n * Creates categories tree visibility handler. Is used by integration and performance tests.\n * @internal\n */\nexport function createCategoriesTreeVisibilityHandler(props: {\n  viewport: TreeWidgetViewport;\n  idsCache: CategoriesTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths?: HierarchySearchPath[];\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}) {\n  return new HierarchyVisibilityHandlerImpl<CategoriesTreeSearchTargets>({\n    getSearchResultsTree: (): undefined | Promise<SearchResultsTree<CategoriesTreeSearchTargets>> => {\n      if (!props.searchPaths) {\n        return undefined;\n      }\n      const { categoryClass, elementClass, modelClass } = getClassesByView(props.viewport.viewType === \"2d\" ? \"2d\" : \"3d\");\n      return createCategoriesSearchResultsTree({\n        idsCache: props.idsCache,\n        searchPaths: props.searchPaths,\n        imodelAccess: props.imodelAccess,\n        categoryClassName: categoryClass,\n        categoryElementClassName: elementClass,\n        categoryModelClassName: modelClass,\n      });\n    },\n    getTreeSpecificVisibilityHandler: (info) => {\n      return new CategoriesTreeVisibilityHandler({\n        alwaysAndNeverDrawnElementInfo: info,\n        idsCache: props.idsCache,\n        viewport: props.viewport,\n        hierarchyConfig: props.hierarchyConfig,\n      });\n    },\n    viewport: props.viewport,\n    componentId: Guid.createValue(),\n  });\n}\n"]}
+{"version":3,"file":"CategoriesTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/visibility/CategoriesTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AACtF,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,8BAA8B,EAAE,MAAM,8DAA8D,CAAC;AAC9G,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,sBAAsB,EACtB,wBAAwB,EACxB,aAAa,EACb,eAAe,GAChB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,0BAA0B,EAAE,MAAM,8CAA8C,CAAC;AAC1F,OAAO,EAAE,8BAA8B,EAAE,MAAM,qCAAqC,CAAC;AACrF,OAAO,EAAE,iCAAiC,EAAE,MAAM,wBAAwB,CAAC;AAyB3E;;;;;GAKG;AACH,MAAM,OAAO,+BAA+B;IACjC,MAAM,CAAuC;IACtD,iBAAiB,CAAiC;IAClD,YAAY,gBAAsD;QAChE,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,IAAI,CAAC,iBAAiB,GAAG,IAAI,8BAA8B,CAAC;YAC1D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,YAAY,EAAE,gBAAgB,CAAC,QAAQ;YACvC,eAAe,EAAE,gBAAgB,CAAC,eAAe;SAClD,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAoC,EAAE,EAAW;QAC1F,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBAC9C,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,EAAE,sBAAsB,EAAE,aAAa,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC1F,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC,EAAE,sBAAsB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACtH,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CACrJ,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,mCAAmC,CAAC,EAAE,cAAc,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,CAC7I,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,MAAM,oBAAoB,GAAG,IAAI,KAAK,EAAc,CAAC;gBACrD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAc,CAAC;gBAC5C,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAgD,CAAC;gBACxF,wDAAwD;gBACxD,gBAAgB;gBAChB,kCAAkC;gBAClC,yFAAyF;gBACzF,kFAAkF;gBAClF,KAAK,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,QAAQ,EAAE,CAAC;oBACtE,MAAM,GAAG,GAA+B,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;oBACnE,IAAI,QAAQ,GAAG,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,QAAQ,GAAG,EAAE,CAAC;wBACd,uBAAuB,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBAC7C,CAAC;oBACD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC1D,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBACzB,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;wBAC7B,IAAI,cAAc,EAAE,CAAC;4BACnB,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBACvC,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,WAAW,CAAC,IAAI;gBACd,kHAAkH;gBAClH,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,oBAAoB,EAAE,CAAC,CAAC,IAAI;gBAClF,sHAAsH;gBACtH,+IAA+I;gBAC/I,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBACrB,wBAAwB,EAAE,aAAa,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,EAAE,aAAa,CAAC;oBAC3I,YAAY;iBACb,CAAC,CAAC,EACH,QAAQ,CAAC,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,EAAE,EAAE,CACtD,eAAe,CAAC,EAAE,MAAM,EAAE,uBAAuB,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,uBAAuB,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CACzH,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,8CAA8C,CAAC,EAAE,EAAE;oBACjE,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBAC7C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAc,CAAC;oBAC1C,8FAA8F;oBAC9F,KAAK,MAAM,SAAS,IAAI,8CAA8C,EAAE,CAAC;wBACvE,MAAM,mBAAmB,GAA6B,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC;wBAC5F,IAAI,CAAC,mBAAmB,EAAE,CAAC;4BACzB,SAAS;wBACX,CAAC;wBACD,KAAK,MAAM,OAAO,IAAI,sBAAsB,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE,CAAC;4BAC5E,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBAC3B,CAAC;oBACH,CAAC;oBACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;wBAC3D,OAAO;wBACP,UAAU;wBACV,UAAU,EAAE,8CAA8C;wBAC1D,8DAA8D;wBAC9D,QAAQ,EAAE,eAAe,CAAC,WAAW,EAAE,wBAAwB,CAAC;wBAChE,EAAE;qBACH,CAAC,CAAC;gBACL,CAAC,CAAC,CACH,CACF,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAC9C,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,0BAA0B,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;YAChE,OAAO,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBAC/D,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,gBAAgB;gBACpD,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,aAAa;gBAC9C,sBAAsB,EAAE,IAAI,CAAC,YAAY,CAAC,sBAAsB;aACjE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,0BAA0B,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/D,OAAO,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC;gBACpE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aACnF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,0BAA0B,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC;gBACtD,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aACrE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,0BAA0B,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YACpD,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;gBAC1D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;gBACvE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;aAC5F,CAAC,CAAC;QACL,CAAC;QAED,IAAI,0BAA0B,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;gBAC7D,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aAC3E,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,0BAA0B,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;QACvD,MAAM,qBAAqB,GAAG,wBAAwB,CAAC;YACrD,kBAAkB,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,gBAAgB,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC;YACjI,sBAAsB,EAAE,IAAI,CAAC,YAAY,CAAC,sBAAsB;SACjE,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;YACtE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;YAClC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;YACxC,qBAAqB;YACrB,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,aAAa;YAC9C,8BAA8B,EAAE,IAAI,CAAC,YAAY,CAAC,8BAA8B;SACjF,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBAC9C,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,0BAA0B,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChE,OAAO,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAClE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;oBACxC,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,gBAAgB;oBACpD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,0BAA0B,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,iBAAiB,CAAC,0CAA0C,CAAC;oBACvE,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAClF,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,0BAA0B,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjD,OAAO,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC;oBACzD,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;oBACnD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,0BAA0B,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBAC7D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBACvE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;oBAC3F,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,0BAA0B,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,mCAAmC,CAAC;oBAChE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;oBACxC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC1E,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YACD,MAAM,CAAC,0BAA0B,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAEvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CACnE,GAAG,CAAC,CAAC,YAAY,EAAW,EAAE;gBAC5B,iFAAiF;gBACjF,qDAAqD;gBACrD,OAAO,sBAAsB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7F,CAAC,CAAC,EACF,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CACpB,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBACpD,UAAU;gBACV,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO;gBAClC,QAAQ,EAAE,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;gBAClD,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU;gBACxC,EAAE;aACH,CAAC,CACH,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CACrC,OAAoC,EACpC,IAEC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAC9C,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,sBAAsB,EAAE,aAAa,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC1F,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,sBAAsB,EAAE,IAAI,EAAE,CAAC;gBACjC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,uCAAuC,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;YAC/G,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YACnF,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;oBACnD,WAAW;oBACX,OAAO;iBACR,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC,CAAC,CACtI,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,MAAM,oBAAoB,GAAG,IAAI,KAAK,EAAc,CAAC;gBACrD,KAAK,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,QAAQ,EAAE,CAAC;oBACjD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC1D,IAAI,cAAc,EAAE,CAAC;4BACnB,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBACvC,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,mBAAwD,CAAC;gBAC7D,IAAI,0BAA0B,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;oBAChE,MAAM,0BAA0B,GAA2D,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC;oBAC5H,MAAM,0BAA0B,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,0BAA0B,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;oBACjI,4EAA4E;oBAC5E,qGAAqG;oBACrG,mBAAmB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,0BAA0B,EAAE,CAAC,CAAC,IAAI,CAClH,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;wBACtB,qGAAqG;wBACrG,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,IAAI,EAAE,EAAE,CAAC;4BACjE,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;wBAChD,CAAC;wBACD,OAAO,eAAe,CAAC;oBACzB,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,mBAAmB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,oBAAoB,EAAE,CAAC,CAAC;gBAC5G,CAAC;gBACD,WAAW,CAAC,IAAI,CACd,mBAAmB,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,wBAAwB,EAAE,EAAE,CACpC,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAE,sBAAsB,EAAE,EAAE,EAAE;oBAClG,MAAM,qBAAqB,GAAG,sBAAsB;wBAClD,CAAC,CAAC,wBAAwB,CAAC;4BACvB,kBAAkB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC;4BACtE,sBAAsB;yBACvB,CAAC;wBACJ,CAAC,CAAC,EAAE,CAAC;oBACP,IAAI,+BAA+B,GAAG,CAAC,CAAC;oBACxC,MAAM,kBAAkB,GAAG,IAAI,KAAK,EAAc,CAAC;oBACnD,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;oBAChD,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC1D,IAAI,CAAC,cAAc,EAAE,CAAC;4BACpB,kBAAkB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;4BACnC,SAAS;wBACX,CAAC;wBACD,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBAChC,MAAM,UAAU,GAAG,wBAAwB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;wBAC3D,IAAI,UAAU,EAAE,CAAC;4BACf,+BAA+B,IAAI,UAAU,CAAC;wBAChD,CAAC;oBACH,CAAC;oBACD,OAAO,KAAK,CACV,eAAe,CAAC,MAAM,GAAG,CAAC;wBACxB,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;4BACjD,OAAO;4BACP,UAAU;4BACV,UAAU,EAAE,eAAe;4BAC3B,qBAAqB;4BACrB,aAAa,EAAE,+BAA+B;4BAC9C,4HAA4H;4BAC5H,oDAAoD;4BACpD,8BAA8B,EAAE,UAAU;yBAC3C,CAAC;wBACJ,CAAC,CAAC,KAAK;oBACT,gHAAgH;oBAChH,sHAAsH;oBACtH,qHAAqH;oBACrH,kBAAkB,CAAC,MAAM,GAAG,CAAC;wBAC3B,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;4BACjD,OAAO;4BACP,UAAU;4BACV,UAAU,EAAE,kBAAkB;4BAC9B,qBAAqB;4BACrB,aAAa,EAAE,SAAS;4BACxB,4HAA4H;4BAC5H,oDAAoD;4BACpD,8BAA8B,EAAE,UAAU;yBAC3C,CAAC;wBACJ,CAAC,CAAC,KAAK,CACV,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC;gBACpC,CAAC,CAAC,CACH,CACF,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,qCAAqC,CAAC,KAMrD;IACC,OAAO,IAAI,8BAA8B,CAA8B;QACrE,oBAAoB,EAAE,GAAwE,EAAE;YAC9F,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACvB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACrH,OAAO,iCAAiC,CAAC;gBACvC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,iBAAiB,EAAE,aAAa;gBAChC,wBAAwB,EAAE,YAAY;gBACtC,sBAAsB,EAAE,UAAU;aACnC,CAAC,CAAC;QACL,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,EAAE;YACzC,OAAO,IAAI,+BAA+B,CAAC;gBACzC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe,EAAE,KAAK,CAAC,eAAe;aACvC,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;KAChC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, defer, EMPTY, from, map, merge, mergeAll, mergeMap, of } from \"rxjs\";\nimport { assert, Guid } from \"@itwin/core-bentley\";\nimport { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { HierarchyVisibilityHandlerImpl } from \"../../../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport {\n  fromWithRelease,\n  getClassesByView,\n  getIdsFromChildrenTree,\n  getParentElementsIdsPath,\n  setDifference,\n  setIntersection,\n} from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { CategoriesTreeNodeInternal } from \"../../internal/CategoriesTreeNodeInternal.js\";\nimport { CategoriesTreeVisibilityHelper } from \"./CategoriesTreeVisibilityHelper.js\";\nimport { createCategoriesSearchResultsTree } from \"./SearchResultsTree.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { ClassGroupingNodeKey, HierarchyNode, HierarchySearchPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { AlwaysAndNeverDrawnElementInfoCache } from \"../../../common/internal/caches/AlwaysAndNeverDrawnElementInfoCache.js\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { ChildrenTree } from \"../../../common/internal/Utils.js\";\nimport type { SearchResultsTree } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type { TreeSpecificVisibilityHandler } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { CategoriesTreeHierarchyConfiguration } from \"../../CategoriesTreeDefinition.js\";\nimport type { CategoriesTreeIdsCache } from \"../CategoriesTreeIdsCache.js\";\nimport type { CategoriesTreeSearchTargets } from \"./SearchResultsTree.js\";\n\n/** @internal */\nexport interface CategoriesTreeVisibilityHandlerProps {\n  idsCache: CategoriesTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfoCache;\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}\n\n/**\n * Handles visibility status of categories tree nodes.\n *\n * This handler knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class CategoriesTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<CategoriesTreeSearchTargets> {\n  readonly #props: CategoriesTreeVisibilityHandlerProps;\n  #visibilityHelper: CategoriesTreeVisibilityHelper;\n  constructor(constructorProps: CategoriesTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    this.#visibilityHelper = new CategoriesTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      baseIdsCache: constructorProps.idsCache,\n      hierarchyConfig: constructorProps.hierarchyConfig,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeSearchTargetsVisibilityStatus(targets: CategoriesTreeSearchTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      if (this.#props.viewport.viewType === \"other\") {\n        return EMPTY;\n      }\n      const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (definitionContainerIds?.size) {\n        observables.push(this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({ definitionContainerIds, on }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.changeCategoriesVisibilityStatus({ categoryIds, modelId, on }))),\n        );\n      }\n\n      if (subCategories?.length) {\n        observables.push(\n          from(subCategories).pipe(\n            mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.changeSubCategoriesVisibilityStatus({ subCategoryIds, categoryId, on })),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        const searchTargetElements = new Array<Id64String>();\n        const elementIdsSet = new Set<Id64String>();\n        const modelCategoryElementMap = new Map<`${ModelId}-${CategoryId}`, Array<ElementId>>();\n        // elements is an array that stores elements grouped by:\n        // 1. Their path\n        // 2. Their modelId and categoryId\n        // When changing visibility of elements, visibility handler does not care about the path.\n        // So we can first get all elements and group them only by modelId and categoryId.\n        for (const { elements: elementsMap, categoryId, modelId } of elements) {\n          const key: `${ModelId}-${CategoryId}` = `${modelId}-${categoryId}`;\n          let mapEntry = modelCategoryElementMap.get(key);\n          if (!mapEntry) {\n            mapEntry = [];\n            modelCategoryElementMap.set(key, mapEntry);\n          }\n          for (const [elementId, { isSearchTarget }] of elementsMap) {\n            mapEntry.push(elementId);\n            elementIdsSet.add(elementId);\n            if (isSearchTarget) {\n              searchTargetElements.push(elementId);\n            }\n          }\n        }\n        observables.push(\n          // Get children for search targets, since non search targets don't have all the children present in the hierarchy.\n          this.#props.idsCache.getChildElementsTree({ elementIds: searchTargetElements }).pipe(\n            // Need to filter out and keep only those children ids that are not part of elements that are present in search paths.\n            // Elements in search paths will have their visibility changed directly: they will be provided as elementIds to changeElementsVisibilityStatus.\n            map((childrenTree) => ({\n              childrenNotInSearchPaths: setDifference(getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 }), elementIdsSet),\n              childrenTree,\n            })),\n            mergeMap(({ childrenNotInSearchPaths, childrenTree }) =>\n              fromWithRelease({ source: modelCategoryElementMap.entries(), size: modelCategoryElementMap.size, releaseOnCount: 50 }).pipe(\n                mergeMap(([key, elementsInSearchPathsGroupedByModelAndCategory]) => {\n                  const [modelId, categoryId] = key.split(\"-\");\n                  const childrenIds = new Set<Id64String>();\n                  // A shared children tree was created, need to get the children for each element in the group.\n                  for (const elementId of elementsInSearchPathsGroupedByModelAndCategory) {\n                    const elementChildrenTree: ChildrenTree | undefined = childrenTree.get(elementId)?.children;\n                    if (!elementChildrenTree) {\n                      continue;\n                    }\n                    for (const childId of getIdsFromChildrenTree({ tree: elementChildrenTree })) {\n                      childrenIds.add(childId);\n                    }\n                  }\n                  return this.#visibilityHelper.changeElementsVisibilityStatus({\n                    modelId,\n                    categoryId,\n                    elementIds: elementsInSearchPathsGroupedByModelAndCategory,\n                    // Pass only those children that are not part of search paths.\n                    children: setIntersection(childrenIds, childrenNotInSearchPaths),\n                    on,\n                  });\n                }),\n              ),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (this.#props.viewport.viewType === \"other\") {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {\n      return this.#visibilityHelper.getGroupedElementsVisibilityStatus({\n        categoryId: node.extendedData.categoryId,\n        modelElementsMap: node.extendedData.modelElementsMap,\n        parentKeys: node.parentKeys,\n        childrenCount: node.extendedData.childrenCount,\n        topMostParentElementId: node.extendedData.topMostParentElementId,\n      });\n    }\n\n    if (CategoriesTreeNodeInternal.isDefinitionContainerNode(node)) {\n      return this.#visibilityHelper.getDefinitionContainersVisibilityStatus({\n        definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (CategoriesTreeNodeInternal.isModelNode(node)) {\n      return this.#visibilityHelper.getModelsVisibilityStatus({\n        modelIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (CategoriesTreeNodeInternal.isCategoryNode(node)) {\n      return this.#visibilityHelper.getCategoriesVisibilityStatus({\n        categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n        modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,\n      });\n    }\n\n    if (CategoriesTreeNodeInternal.isSubCategoryNode(node)) {\n      return this.#visibilityHelper.getSubCategoriesVisibilityStatus({\n        categoryId: node.extendedData.categoryId,\n        subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    assert(CategoriesTreeNodeInternal.isElementNode(node));\n    const parentElementsIdsPath = getParentElementsIdsPath({\n      parentInstanceKeys: node.parentKeys.filter((parentKey) => HierarchyNodeKey.isInstances(parentKey)).map((key) => key.instanceKeys),\n      topMostParentElementId: node.extendedData.topMostParentElementId,\n    });\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      modelId: node.extendedData.modelId,\n      categoryId: node.extendedData.categoryId,\n      parentElementsIdsPath,\n      childrenCount: node.extendedData.childrenCount,\n      categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (this.#props.viewport.viewType === \"other\") {\n        return EMPTY;\n      }\n      if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {\n        return this.#visibilityHelper.changeGroupedElementsVisibilityStatus({\n          categoryId: node.extendedData.categoryId,\n          modelElementsMap: node.extendedData.modelElementsMap,\n          on,\n        });\n      }\n\n      if (CategoriesTreeNodeInternal.isDefinitionContainerNode(node)) {\n        return this.#visibilityHelper.changeDefinitionContainersVisibilityStatus({\n          definitionContainerIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      if (CategoriesTreeNodeInternal.isModelNode(node)) {\n        return this.#visibilityHelper.changeModelsVisibilityStatus({\n          modelIds: node.key.instanceKeys.map(({ id }) => id),\n          on,\n        });\n      }\n\n      if (CategoriesTreeNodeInternal.isCategoryNode(node)) {\n        return this.#visibilityHelper.changeCategoriesVisibilityStatus({\n          categoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          modelId: node.extendedData.isCategoryOfSubModel ? node.extendedData.modelIds[0] : undefined,\n          on,\n        });\n      }\n\n      if (CategoriesTreeNodeInternal.isSubCategoryNode(node)) {\n        return this.#visibilityHelper.changeSubCategoriesVisibilityStatus({\n          categoryId: node.extendedData.categoryId,\n          subCategoryIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n      assert(CategoriesTreeNodeInternal.isElementNode(node));\n\n      const elementIds = node.key.instanceKeys.map(({ id }) => id);\n      return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(\n        map((childrenTree): Id64Set => {\n          // Children tree contains provided elementIds, they are at the root of this tree.\n          // We want to skip them and only get ids of children.\n          return getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 });\n        }),\n        mergeMap((children) =>\n          this.#visibilityHelper.changeElementsVisibilityStatus({\n            elementIds,\n            modelId: node.extendedData.modelId,\n            children: children.size > 0 ? children : undefined,\n            categoryId: node.extendedData.categoryId,\n            on,\n          }),\n        ),\n      );\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getSearchTargetsVisibilityStatus(\n    targets: CategoriesTreeSearchTargets,\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    },\n  ): Observable<VisibilityStatus> {\n    if (this.#props.viewport.viewType === \"other\") {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n    return defer(() => {\n      const { definitionContainerIds, subCategories, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (definitionContainerIds?.size) {\n        observables.push(this.#visibilityHelper.getDefinitionContainersVisibilityStatus({ definitionContainerIds }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.getCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (subCategories?.length) {\n        observables.push(\n          from(subCategories).pipe(\n            mergeMap(({ categoryId, subCategoryIds }) => this.#visibilityHelper.getSubCategoriesVisibilityStatus({ subCategoryIds, categoryId })),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        const searchTargetElements = new Array<Id64String>();\n        for (const { elements: elementsMap } of elements) {\n          for (const [elementId, { isSearchTarget }] of elementsMap) {\n            if (isSearchTarget) {\n              searchTargetElements.push(elementId);\n            }\n          }\n        }\n        let childrenCountMapObs: Observable<Map<Id64String, number>>;\n        if (CategoriesTreeNodeInternal.isElementClassGroupingNode(node)) {\n          const groupingNodesSearchTargets: Map<Id64String, { childrenCount: number }> | undefined = node.extendedData?.searchTargets;\n          const nestedSearchTargetElements = searchTargetElements.filter((searchTarget) => !groupingNodesSearchTargets?.has(searchTarget));\n          // Only need to request children count for indirect children search targets.\n          // Direct children search targets already have children count stored in grouping nodes extended data.\n          childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: nestedSearchTargetElements }).pipe(\n            map((elementCountMap) => {\n              // Direct children search targets already have children count stored in grouping nodes extended data.\n              for (const [key, value] of node.extendedData.searchTargets ?? []) {\n                elementCountMap.set(key, value.childrenCount);\n              }\n              return elementCountMap;\n            }),\n          );\n        } else {\n          childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: searchTargetElements });\n        }\n        observables.push(\n          childrenCountMapObs.pipe(\n            mergeMap((elementsChildrenCountMap) =>\n              fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n                mergeMap(({ modelId, elements: elementsMap, categoryId, pathToElements, topMostParentElementId }) => {\n                  const parentElementsIdsPath = topMostParentElementId\n                    ? getParentElementsIdsPath({\n                        parentInstanceKeys: pathToElements.map((instanceKey) => [instanceKey]),\n                        topMostParentElementId,\n                      })\n                    : [];\n                  let totalSearchTargetsChildrenCount = 0;\n                  const nonSearchTargetIds = new Array<Id64String>();\n                  const searchTargetIds = new Array<Id64String>();\n                  for (const [elementId, { isSearchTarget }] of elementsMap) {\n                    if (!isSearchTarget) {\n                      nonSearchTargetIds.push(elementId);\n                      continue;\n                    }\n                    searchTargetIds.push(elementId);\n                    const childCount = elementsChildrenCountMap.get(elementId);\n                    if (childCount) {\n                      totalSearchTargetsChildrenCount += childCount;\n                    }\n                  }\n                  return merge(\n                    searchTargetIds.length > 0\n                      ? this.#visibilityHelper.getElementsVisibilityStatus({\n                          modelId,\n                          categoryId,\n                          elementIds: searchTargetIds,\n                          parentElementsIdsPath,\n                          childrenCount: totalSearchTargetsChildrenCount,\n                          // Search results tree is created on search paths. Since search paths contain only categories that are directly under models\n                          // or at the root, categoryId can be used here here.\n                          categoryOfTopMostParentElement: categoryId,\n                        })\n                      : EMPTY,\n                    // Set childrenCount to undefined for non search targets, as some of their child elements might be filtered out.\n                    // Since childrenCount is set to undefined, these elements won't check child always/never drawn child elements status.\n                    // Child always/never drawn elements will be in search paths, and their visibility status will be handled separately.\n                    nonSearchTargetIds.length > 0\n                      ? this.#visibilityHelper.getElementsVisibilityStatus({\n                          modelId,\n                          categoryId,\n                          elementIds: nonSearchTargetIds,\n                          parentElementsIdsPath,\n                          childrenCount: undefined,\n                          // Search results tree is created on search paths. Since search paths contain only categories that are directly under models\n                          // or at the root, categoryId can be used here here.\n                          categoryOfTopMostParentElement: categoryId,\n                        })\n                      : EMPTY,\n                  ).pipe(mergeVisibilityStatuses());\n                }),\n              ),\n            ),\n          ),\n        );\n      }\n\n      return from(observables).pipe(mergeAll(), mergeVisibilityStatuses());\n    });\n  }\n}\n\n/**\n * Creates categories tree visibility handler. Is used by integration and performance tests.\n * @internal\n */\nexport function createCategoriesTreeVisibilityHandler(props: {\n  viewport: TreeWidgetViewport;\n  idsCache: CategoriesTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths?: HierarchySearchPath[];\n  hierarchyConfig: CategoriesTreeHierarchyConfiguration;\n}) {\n  return new HierarchyVisibilityHandlerImpl<CategoriesTreeSearchTargets>({\n    getSearchResultsTree: (): undefined | Promise<SearchResultsTree<CategoriesTreeSearchTargets>> => {\n      if (!props.searchPaths) {\n        return undefined;\n      }\n      const { categoryClass, elementClass, modelClass } = getClassesByView(props.viewport.viewType === \"2d\" ? \"2d\" : \"3d\");\n      return createCategoriesSearchResultsTree({\n        idsCache: props.idsCache,\n        searchPaths: props.searchPaths,\n        imodelAccess: props.imodelAccess,\n        categoryClassName: categoryClass,\n        categoryElementClassName: elementClass,\n        categoryModelClassName: modelClass,\n      });\n    },\n    getTreeSpecificVisibilityHandler: (info) => {\n      return new CategoriesTreeVisibilityHandler({\n        alwaysAndNeverDrawnElementInfo: info,\n        idsCache: props.idsCache,\n        viewport: props.viewport,\n        hierarchyConfig: props.hierarchyConfig,\n      });\n    },\n    viewport: props.viewport,\n    componentId: Guid.createValue(),\n  });\n}\n"]}

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

@@ -1,7 +1,8 @@
+import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
 import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
 import type { Observable } from "rxjs";
 import type { Id64Arg, Id64String } from "@itwin/core-bentley";
-import type { ElementId, ModelId } from "../../../common/internal/Types.js";
+import type { CategoryId, ElementId, ModelId } from "../../../common/internal/Types.js";
 import type { BaseVisibilityHelperProps } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
 import type { VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
 import type { CategoriesTreeHierarchyConfiguration } from "../../CategoriesTreeDefinition.js";
@@ -30,8 +31,14 @@ export declare class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper
     }): Observable<VisibilityStatus>;
     /** Gets grouped elements visibility status. */
     getGroupedElementsVisibilityStatus(props: {
-        modelElementsMap: Map<ModelId, Set<ElementId>>;
+        modelElementsMap: Map<ModelId, {
+            elementIds: Set<ElementId>;
+            categoryOfTopMostParentElement: CategoryId;
+        }>;
         categoryId: Id64String;
+        parentKeys: HierarchyNodeKey[];
+        childrenCount: number;
+        topMostParentElementId?: ElementId;
     }): Observable<VisibilityStatus>;
     /**
      * Changes visibility status of definition containers.
@@ -54,11 +61,13 @@ export declare class CategoriesTreeVisibilityHelper extends BaseVisibilityHelper
     }): Observable<void>;
     /** Changes grouped elements visibility status. */
     changeGroupedElementsVisibilityStatus(props: {
-        modelElementsMap: Map<ModelId, Set<ElementId>>;
+        modelElementsMap: Map<ModelId, {
+            elementIds: Set<ElementId>;
+        }>;
         categoryId: Id64String;
         on: boolean;
     }): Observable<void>;
-    /** Turns on visibility status of models (that are not yet turned on) that are related to categories. */
-    private enableCategoriesElementModelsVisibilityStatus;
+    /** Turns on category and its' related models. Does not turn on other categories contained in those models.*/
+    private enableCategoryWithoutEnablingOtherCategories;
 }
 //# sourceMappingURL=CategoriesTreeVisibilityHelper.d.ts.map