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

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHandler.js

@@ -0,0 +1,177 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { concat, defer, EMPTY, from, map, merge, mergeMap, of, toArray } from "rxjs";
+import { Id64 } from "@itwin/core-bentley";
+import { HierarchyNode } from "@itwin/presentation-hierarchies";
+import { releaseMainThreadOnItemsCount } from "../../../common/internal/Utils.js";
+import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
+import { ClassificationsTreeNode } from "../ClassificationsTreeNode.js";
+import { ClassificationsTreeVisibilityHelper } from "./ClassificationsTreeVisibilityHelper.js";
+/**
+ * Handles visibility status of classifications tree nodes.
+ *
+ * This handler knows how to get and change visibility status of nodes created by hierarchy definition.
+ * @internal
+ */
+export class ClassificationsTreeVisibilityHandler {
+    #props;
+    #visibilityHelper;
+    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 ClassificationsTreeVisibilityHelper({
+            viewport: this.#props.viewport,
+            idsCache: this.#props.idsCache,
+            alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
+            baseIdsCache,
+        });
+    }
+    [Symbol.dispose]() {
+        this.#visibilityHelper[Symbol.dispose]();
+    }
+    changeFilterTargetsVisibilityStatus(targets, on) {
+        return defer(() => {
+            const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;
+            const observables = new Array();
+            if (classificationTableIds?.size) {
+                observables.push(this.#visibilityHelper.changeClassificationTablesVisibilityStatus({ classificationTableIds, on }));
+            }
+            if (classificationIds?.size) {
+                observables.push(this.#visibilityHelper.changeClassificationsVisibilityStatus({ classificationIds, on }));
+            }
+            if (elements2d?.length) {
+                observables.push(from(elements2d).pipe(mergeMap(({ modelId, categoryId, elements }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }))));
+            }
+            if (elements3d?.length) {
+                observables.push(from(elements3d).pipe(mergeMap(({ modelId, categoryId, elements }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }))));
+            }
+            return merge(...observables);
+        });
+    }
+    getVisibilityStatus(node) {
+        if (!HierarchyNode.isInstancesNode(node)) {
+            return EMPTY;
+        }
+        if (ClassificationsTreeNode.isClassificationTableNode(node)) {
+            return this.#visibilityHelper.getClassificationTablesVisibilityStatus({
+                classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        if (ClassificationsTreeNode.isClassificationNode(node)) {
+            return this.#visibilityHelper.getClassificationsVisibilityStatus({
+                classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            });
+        }
+        return this.#visibilityHelper.getElementsVisibilityStatus({
+            elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+            modelId: ClassificationsTreeNode.getModelId(node),
+            categoryId: ClassificationsTreeNode.getCategoryId(node),
+            type: node.extendedData?.type,
+        });
+    }
+    /** Changes visibility of the items represented by the tree node. */
+    changeVisibilityStatus(node, on) {
+        const changeObs = defer(() => {
+            if (!HierarchyNode.isInstancesNode(node)) {
+                return EMPTY;
+            }
+            if (ClassificationsTreeNode.isClassificationTableNode(node)) {
+                return this.#visibilityHelper.changeClassificationTablesVisibilityStatus({
+                    classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                    on,
+                });
+            }
+            if (ClassificationsTreeNode.isClassificationNode(node)) {
+                return this.#visibilityHelper.changeClassificationsVisibilityStatus({
+                    classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),
+                    on,
+                });
+            }
+            return this.#visibilityHelper.changeElementsVisibilityStatus({
+                elementIds: node.key.instanceKeys.map(({ id }) => id),
+                modelId: ClassificationsTreeNode.getModelId(node),
+                categoryId: ClassificationsTreeNode.getCategoryId(node),
+                on,
+            });
+        });
+        if (this.#props.viewport.isAlwaysDrawnExclusive) {
+            return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);
+        }
+        return changeObs;
+    }
+    getFilterTargetsVisibilityStatus(targets) {
+        return defer(() => {
+            const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;
+            const observables = new Array();
+            if (classificationTableIds?.size) {
+                observables.push(this.#visibilityHelper.getClassificationTablesVisibilityStatus({ classificationTableIds }));
+            }
+            if (classificationIds?.size) {
+                observables.push(this.#visibilityHelper.getClassificationsVisibilityStatus({ classificationIds }));
+            }
+            if (elements2d?.length) {
+                observables.push(from(elements2d).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, categoryId, elements }) => {
+                    return from(elements.keys()).pipe(releaseMainThreadOnItemsCount(1000), mergeMap((elementId) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: "GeometricElement2d" })));
+                })));
+            }
+            if (elements3d?.length) {
+                observables.push(from(elements3d).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, categoryId, elements }) => {
+                    return from(elements.keys()).pipe(releaseMainThreadOnItemsCount(1000), mergeMap((elementId) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: "GeometricElement3d" })));
+                })));
+            }
+            return merge(...observables);
+        }).pipe(mergeVisibilityStatuses);
+    }
+    getCategories(props) {
+        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache
+            .getModelCategoryIds(modelId)
+            .pipe(map(({ spatial, drawing }) => ({ id: modelId, drawingCategories: drawing, spatialCategories: spatial })))));
+    }
+    getAllCategories() {
+        return this.#props.idsCache.getAllCategories().pipe(map(({ drawing, spatial }) => {
+            return { drawingCategories: drawing, spatialCategories: spatial };
+        }));
+    }
+    getElementsCount(props) {
+        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
+    }
+    getModels(props) {
+        return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesElementModels(categoryId, true).pipe(mergeMap((categoryModelsMap) => (categoryModelsMap.size > 0 ? categoryModelsMap.values() : of(new Array()))), map((categoryModels) => ({ id: categoryId, models: categoryModels })))));
+    }
+    getSubCategories(props) {
+        return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));
+    }
+    getSubModels(props) {
+        if ("modelIds" in props) {
+            return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => {
+                if (props.categoryId) {
+                    return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));
+                }
+                return from(this.#props.idsCache.getModelCategoryIds(modelId)).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)), map((subModels) => ({ id: modelId, subModels })));
+            }));
+        }
+        if (props.modelId) {
+            return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesModeledElements(props.modelId, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels })))));
+        }
+        return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesElementModels(categoryId).pipe(mergeMap((categoryModelsMap) => {
+            const models = categoryModelsMap.get(categoryId);
+            if (!models) {
+                return of({ id: categoryId, subModels: undefined });
+            }
+            return from(models).pipe(mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryId)), map((subModels) => ({ id: categoryId, subModels })));
+        }))));
+    }
+}
+//# sourceMappingURL=ClassificationsTreeVisibilityHandler.js.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClassificationsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrF,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AAClF,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AACxE,OAAO,EAAE,mCAAmC,EAAE,MAAM,0CAA0C,CAAC;AAkB/F;;;;;GAKG;AACH,MAAM,OAAO,oCAAoC;IACtC,MAAM,CAA4C;IAC3D,iBAAiB,CAAsC;IAEvD,YAAY,gBAA2D;QACrE,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,mCAAmC,CAAC;YAC/D,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;SACb,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,OAAyC,EAAE,EAAW;QAC/F,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;YACtF,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,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5G,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,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAC7C,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACrH,CACF,CACF,CAAC;YACJ,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,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAC7C,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACrH,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,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,uBAAuB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5D,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,uBAAuB,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBAC/D,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;aAC9E,CAAC,CAAC;QACL,CAAC;QAED,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,uBAAuB,CAAC,UAAU,CAAC,IAAI,CAAC;YACjD,UAAU,EAAE,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC;YACvD,IAAI,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI;SAC9B,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,uBAAuB,CAAC,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5D,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,uBAAuB,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAClE,iBAAiB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC7E,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO,EAAE,uBAAuB,CAAC,UAAU,CAAC,IAAI,CAAC;gBACjD,UAAU,EAAE,uBAAuB,CAAC,aAAa,CAAC,IAAI,CAAC;gBACvD,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,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,OAAyC;QAC/E,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;YACtF,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,iBAAiB,EAAE,IAAI,EAAE,CAAC;gBAC5B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC;YACrG,CAAC;YACD,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAC/B,6BAA6B,CAAC,IAAI,CAAC,EACnC,QAAQ,CAAC,CAAC,SAAS,EAAE,EAAE,CACrB,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC/H,CACF,CAAC;gBACJ,CAAC,CAAC,CACH,CACF,CAAC;YACJ,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAC/B,6BAA6B,CAAC,IAAI,CAAC,EACnC,QAAQ,CAAC,CAAC,SAAS,EAAE,EAAE,CACrB,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC/H,CACF,CAAC;gBACJ,CAAC,CAAC,CACH,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;aACjB,mBAAmB,CAAC,OAAO,CAAC;aAC5B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAClH,CACF,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CACjD,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE;YAC3B,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC;QACpE,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,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CACpE,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,KAAK,EAAW,CAAC,CAAC,CAAC,EACrH,GAAG,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC,CACtE,CACF,CACF,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACjE,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAAC,IAAI,CAC9D,QAAQ,CAAC,CAAC,iBAAiB,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,EAC7F,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CACpD,CAAC;QACJ,CAAC,CAAC,CACH,CACF,CACF,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, defer, EMPTY, from, map, merge, mergeMap, of, toArray } from \"rxjs\";\nimport { Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { releaseMainThreadOnItemsCount } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ClassificationsTreeNode } from \"../ClassificationsTreeNode.js\";\nimport { ClassificationsTreeVisibilityHelper } from \"./ClassificationsTreeVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { ModelId } from \"../../../common/internal/Types.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 { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\nimport type { ClassificationsTreeFilterTargets } from \"./FilteredTree.js\";\n\n/** @internal */\nexport interface ClassificationsTreeVisibilityHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n}\n\n/**\n * Handles visibility status of classifications 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 ClassificationsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ClassificationsTreeFilterTargets> {\n  readonly #props: ClassificationsTreeVisibilityHandlerProps;\n  #visibilityHelper: ClassificationsTreeVisibilityHelper;\n\n  constructor(constructorProps: ClassificationsTreeVisibilityHandlerProps) {\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 ClassificationsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      baseIdsCache,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeFilterTargetsVisibilityStatus(targets: ClassificationsTreeFilterTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;\n      const observables = new Array<Observable<void>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationTablesVisibilityStatus({ classificationTableIds, on }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.changeClassificationsVisibilityStatus({ classificationIds, on }));\n      }\n\n      if (elements2d?.length) {\n        observables.push(\n          from(elements2d).pipe(\n            mergeMap(({ modelId, categoryId, elements }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      if (elements3d?.length) {\n        observables.push(\n          from(elements3d).pipe(\n            mergeMap(({ modelId, categoryId, elements }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elements.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return EMPTY;\n    }\n\n    if (ClassificationsTreeNode.isClassificationTableNode(node)) {\n      return this.#visibilityHelper.getClassificationTablesVisibilityStatus({\n        classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    if (ClassificationsTreeNode.isClassificationNode(node)) {\n      return this.#visibilityHelper.getClassificationsVisibilityStatus({\n        classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      });\n    }\n\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n      modelId: ClassificationsTreeNode.getModelId(node),\n      categoryId: ClassificationsTreeNode.getCategoryId(node),\n      type: node.extendedData?.type,\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (!HierarchyNode.isInstancesNode(node)) {\n        return EMPTY;\n      }\n\n      if (ClassificationsTreeNode.isClassificationTableNode(node)) {\n        return this.#visibilityHelper.changeClassificationTablesVisibilityStatus({\n          classificationTableIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      if (ClassificationsTreeNode.isClassificationNode(node)) {\n        return this.#visibilityHelper.changeClassificationsVisibilityStatus({\n          classificationIds: node.key.instanceKeys.map((instanceKey) => instanceKey.id),\n          on,\n        });\n      }\n\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId: ClassificationsTreeNode.getModelId(node),\n        categoryId: ClassificationsTreeNode.getCategoryId(node),\n        on,\n      });\n    });\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getFilterTargetsVisibilityStatus(targets: ClassificationsTreeFilterTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { classificationIds, classificationTableIds, elements2d, elements3d } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (classificationTableIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationTablesVisibilityStatus({ classificationTableIds }));\n      }\n\n      if (classificationIds?.size) {\n        observables.push(this.#visibilityHelper.getClassificationsVisibilityStatus({ classificationIds }));\n      }\n      if (elements2d?.length) {\n        observables.push(\n          from(elements2d).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, categoryId, elements }) => {\n              return from(elements.keys()).pipe(\n                releaseMainThreadOnItemsCount(1000),\n                mergeMap((elementId) =>\n                  this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: \"GeometricElement2d\" }),\n                ),\n              );\n            }),\n          ),\n        );\n      }\n\n      if (elements3d?.length) {\n        observables.push(\n          from(elements3d).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, categoryId, elements }) => {\n              return from(elements.keys()).pipe(\n                releaseMainThreadOnItemsCount(1000),\n                mergeMap((elementId) =>\n                  this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: elementId, type: \"GeometricElement3d\" }),\n                ),\n              );\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\n          .getModelCategoryIds(modelId)\n          .pipe(map(({ spatial, drawing }) => ({ id: modelId, drawingCategories: drawing, spatialCategories: spatial }))),\n      ),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(\n      map(({ drawing, spatial }) => {\n        return { drawingCategories: drawing, spatialCategories: spatial };\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 from(Id64.iterable(props.categoryIds)).pipe(\n      mergeMap((categoryId) =>\n        this.#props.idsCache.getCategoriesElementModels(categoryId, true).pipe(\n          mergeMap((categoryModelsMap) => (categoryModelsMap.size > 0 ? categoryModelsMap.values() : of(new Array<ModelId>()))),\n          map((categoryModels) => ({ id: categoryId, models: categoryModels })),\n        ),\n      ),\n    );\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return from(this.#props.idsCache.getModelCategoryIds(modelId)).pipe(\n            mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n            toArray(),\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return from(Id64.iterable(props.categoryIds)).pipe(\n      mergeMap((categoryId) =>\n        this.#props.idsCache.getCategoriesElementModels(categoryId).pipe(\n          mergeMap((categoryModelsMap) => {\n            const models = categoryModelsMap.get(categoryId);\n            if (!models) {\n              return of({ id: categoryId, subModels: undefined });\n            }\n            return from(models).pipe(\n              mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryId)),\n              map((subModels) => ({ id: categoryId, subModels })),\n            );\n          }),\n        ),\n      ),\n    );\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.d.ts

@@ -0,0 +1,55 @@
+import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { Observable } from "rxjs";
+import type { Id64Arg } from "@itwin/core-bentley";
+import type { BaseVisibilityHelperProps } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import type { VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
+import type { ClassificationsTreeIdsCache } from "../ClassificationsTreeIdsCache.js";
+/** @internal */
+export type ClassificationsTreeVisibilityHelperProps = BaseVisibilityHelperProps & {
+    idsCache: ClassificationsTreeIdsCache;
+};
+/**
+ * Visibility status helper for classifications tree.
+ *
+ * It extends base visibility status helper and provides methods to get and change visibility status of classification tables and classifications.
+ * @internal
+ */
+export declare class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
+    #private;
+    constructor(props: ClassificationsTreeVisibilityHelperProps);
+    /**
+     * Gets visibility status of classification tables.
+     *
+     * Determines visibility status by checking visibility status of related categories.
+     */
+    getClassificationTablesVisibilityStatus(props: {
+        classificationTableIds: Id64Arg;
+    }): Observable<VisibilityStatus>;
+    /**
+     * Gets visibility status of classifications.
+     *
+     * Determines visibility status by checking visibility status of related categories.
+     */
+    getClassificationsVisibilityStatus(props: {
+        classificationIds: Id64Arg;
+    }): Observable<VisibilityStatus>;
+    /**
+     * Changes visibility status of classification tables.
+     *
+     * Does this by changing visibility status of related categories.
+     */
+    changeClassificationTablesVisibilityStatus(props: {
+        classificationTableIds: Id64Arg;
+        on: boolean;
+    }): Observable<void>;
+    /**
+     * Changes visibility status of classifications.
+     *
+     * Does this by changing visibility status of related categories.
+     */
+    changeClassificationsVisibilityStatus(props: {
+        classificationIds: Id64Arg;
+        on: boolean;
+    }): Observable<void>;
+}
+//# sourceMappingURL=ClassificationsTreeVisibilityHelper.d.ts.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.js

@@ -0,0 +1,53 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { merge, mergeMap, of, toArray } from "rxjs";
+import { BaseVisibilityHelper } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
+import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
+/**
+ * Visibility status helper for classifications tree.
+ *
+ * It extends base visibility status helper and provides methods to get and change visibility status of classification tables and classifications.
+ * @internal
+ */
+export class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {
+    #props;
+    constructor(props) {
+        super(props);
+        this.#props = props;
+    }
+    /**
+     * Gets visibility status of classification tables.
+     *
+     * Determines visibility status by checking visibility status of related categories.
+     */
+    getClassificationTablesVisibilityStatus(props) {
+        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap(({ drawing, spatial }) => merge(of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "DrawingCategory" }))), of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "SpatialCategory" }))))), mergeVisibilityStatuses);
+    }
+    /**
+     * Gets visibility status of classifications.
+     *
+     * Determines visibility status by checking visibility status of related categories.
+     */
+    getClassificationsVisibilityStatus(props) {
+        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap(({ drawing, spatial }) => merge(of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "DrawingCategory" }))), of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: "SpatialCategory" }))))), mergeVisibilityStatuses);
+    }
+    /**
+     * Changes visibility status of classification tables.
+     *
+     * Does this by changing visibility status of related categories.
+     */
+    changeClassificationTablesVisibilityStatus(props) {
+        return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })));
+    }
+    /**
+     * Changes visibility status of classifications.
+     *
+     * Does this by changing visibility status of related categories.
+     */
+    changeClassificationsVisibilityStatus(props) {
+        return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(mergeMap(({ drawing, spatial }) => merge(drawing, spatial)), toArray(), mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })));
+    }
+}
+//# sourceMappingURL=ClassificationsTreeVisibilityHelper.js.map

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClassificationsTreeVisibilityHelper.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/ClassificationsTreeVisibilityHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,6DAA6D,CAAC;AACnG,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAatF;;;;;GAKG;AACH,MAAM,OAAO,mCAAoC,SAAQ,oBAAoB;IAC3E,MAAM,CAA2C;IACjD,YAAY,KAA+C;QACzD,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,uCAAuC,CAAC,KAA0C;QACvF,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAChC,KAAK,CACH,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAC7I,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAC9I,CACF,EACD,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,kCAAkC,CAAC,KAAqC;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAChC,KAAK,CACH,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAC7I,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAC9I,CACF,EACD,uBAAuB,CACxB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,0CAA0C,CAAC,KAAuD;QACvG,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,IAAI,CACtF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CACpH,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACI,qCAAqC,CAAC,KAAkD;QAC7F,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CACjF,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,EAC3D,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CACpH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { merge, mergeMap, of, toArray } from \"rxjs\";\nimport { BaseVisibilityHelper } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { BaseVisibilityHelperProps } from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { VisibilityStatus } from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\n/** @internal */\nexport type ClassificationsTreeVisibilityHelperProps = BaseVisibilityHelperProps & {\n  idsCache: ClassificationsTreeIdsCache;\n};\n\n/**\n * Visibility status helper for classifications tree.\n *\n * It extends base visibility status helper and provides methods to get and change visibility status of classification tables and classifications.\n * @internal\n */\nexport class ClassificationsTreeVisibilityHelper extends BaseVisibilityHelper {\n  #props: ClassificationsTreeVisibilityHelperProps;\n  constructor(props: ClassificationsTreeVisibilityHelperProps) {\n    super(props);\n    this.#props = props;\n  }\n\n  /**\n   * Gets visibility status of classification tables.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap(({ drawing, spatial }) =>\n        merge(\n          of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"DrawingCategory\" }))),\n          of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"SpatialCategory\" }))),\n        ),\n      ),\n      mergeVisibilityStatuses,\n    );\n  }\n\n  /**\n   * Gets visibility status of classifications.\n   *\n   * Determines visibility status by checking visibility status of related categories.\n   */\n  public getClassificationsVisibilityStatus(props: { classificationIds: Id64Arg }): Observable<VisibilityStatus> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap(({ drawing, spatial }) =>\n        merge(\n          of(drawing).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"DrawingCategory\" }))),\n          of(spatial).pipe(mergeMap((categoryIds) => this.getCategoriesVisibilityStatus({ modelId: undefined, categoryIds, type: \"SpatialCategory\" }))),\n        ),\n      ),\n      mergeVisibilityStatuses,\n    );\n  }\n\n  /**\n   * Changes visibility status of classification tables.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationTablesVisibilityStatus(props: { classificationTableIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationTableIds).pipe(\n      mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n      toArray(),\n      mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })),\n    );\n  }\n\n  /**\n   * Changes visibility status of classifications.\n   *\n   * Does this by changing visibility status of related categories.\n   */\n  public changeClassificationsVisibilityStatus(props: { classificationIds: Id64Arg; on: boolean }): Observable<void> {\n    return this.#props.idsCache.getAllContainedCategories(props.classificationIds).pipe(\n      mergeMap(({ drawing, spatial }) => merge(drawing, spatial)),\n      toArray(),\n      mergeMap((categoryIds) => this.changeCategoriesVisibilityStatus({ modelId: undefined, categoryIds, on: props.on })),\n    );\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/classifications-tree/internal/visibility/FilteredTree.d.ts

@@ -0,0 +1,32 @@
+import type { Id64Set, Id64String } from "@itwin/core-bentley";
+import type { HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
+import type { ElementId } from "../../../common/internal/Types.js";
+import type { FilteredTree } from "../../../common/internal/visibility/BaseFilteredTree.js";
+import type { ClassificationsTreeIdsCache } from "../ClassificationsTreeIdsCache.js";
+/** @internal */
+export interface ClassificationsTreeFilterTargets {
+    elements2d?: Array<{
+        modelId: Id64String;
+        categoryId: Id64String;
+        elements: Map<ElementId, {
+            isFilterTarget: boolean;
+        }>;
+    }>;
+    elements3d?: Array<{
+        modelId: Id64String;
+        categoryId: Id64String;
+        elements: Map<ElementId, {
+            isFilterTarget: boolean;
+        }>;
+    }>;
+    classificationTableIds?: Id64Set;
+    classificationIds?: Id64Set;
+}
+/** @internal */
+export declare function createFilteredClassificationsTree(props: {
+    imodelAccess: ECClassHierarchyInspector;
+    filteringPaths: HierarchyFilteringPath[];
+    idsCache: ClassificationsTreeIdsCache;
+}): Promise<FilteredTree<ClassificationsTreeFilterTargets>>;
+//# sourceMappingURL=FilteredTree.d.ts.map

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

@@ -0,0 +1,168 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { firstValueFrom } from "rxjs";
+import { assert } from "@itwin/core-bentley";
+import { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement2d } from "../../../common/internal/ClassNameDefinitions.js";
+import { createFilteredTree, FilteredNodesHandler } from "../../../common/internal/visibility/BaseFilteredTree.js";
+/** @internal */
+export async function createFilteredClassificationsTree(props) {
+    const { imodelAccess, filteringPaths, idsCache } = props;
+    return createFilteredTree({
+        filteredNodesHandler: new ClassificationsTreeFilteredNodesHandler({ idsCache, imodelAccess }),
+        filteringPaths,
+    });
+}
+class ClassificationsTreeFilteredNodesHandler extends FilteredNodesHandler {
+    #props;
+    constructor(props) {
+        super();
+        this.#props = props;
+    }
+    async getProcessedFilteredNodes() {
+        const filteredTemporary2dElements = new Map();
+        const filteredTemporary3dElements = new Map();
+        const result = {
+            filtered2dElements: new Map(),
+            filtered3dElements: new Map(),
+        };
+        for (const node of this.filteredNodesArr) {
+            if (node.type === "element2d") {
+                filteredTemporary2dElements.set(node.id, node);
+            }
+            else if (node.type === "element3d") {
+                filteredTemporary3dElements.set(node.id, node);
+            }
+        }
+        const filteredElementsModels = await firstValueFrom(this.#props.idsCache.getFilteredElementsData({
+            element2dIds: [...filteredTemporary2dElements.keys()],
+            element3dIds: [...filteredTemporary3dElements.keys()],
+        }));
+        filteredTemporary2dElements.forEach((element, id) => {
+            const entry = filteredElementsModels.get(element.id);
+            assert(entry !== undefined);
+            result.filtered2dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });
+        });
+        filteredTemporary3dElements.forEach((element, id) => {
+            const entry = filteredElementsModels.get(element.id);
+            assert(entry !== undefined);
+            result.filtered3dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });
+        });
+        return result;
+    }
+    convertNodesToFilterTargets(filteredNodes, processedFilteredNodes) {
+        const filterTargets = {};
+        filteredNodes.forEach((filteredNode) => this.collectFilterTargets(filterTargets, filteredNode, processedFilteredNodes));
+        return this.convertInternalFilterTargets(filterTargets);
+    }
+    convertInternalFilterTargets(filterTargets) {
+        if (!filterTargets.classificationIds && !filterTargets.classificationIds && !filterTargets.elements2d && !filterTargets.elements3d) {
+            return undefined;
+        }
+        return {
+            classificationIds: filterTargets.classificationIds,
+            classificationTableIds: filterTargets.classificationIds,
+            elements2d: filterTargets.elements2d
+                ? [...filterTargets.elements2d?.entries()].map(([modelCategoryKey, elements]) => {
+                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
+                    return { modelId, categoryId, elements };
+                })
+                : undefined,
+            elements3d: filterTargets.elements3d
+                ? [...filterTargets.elements3d?.entries()].map(([modelCategoryKey, elements]) => {
+                    const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);
+                    return { modelId, categoryId, elements };
+                })
+                : undefined,
+        };
+    }
+    collectFilterTargets(changeTargets, node, processedFilteredNodes) {
+        const filteredNode = node.type === "element2d"
+            ? processedFilteredNodes.filtered2dElements.get(node.id)
+            : node.type === "element3d"
+                ? processedFilteredNodes.filtered3dElements.get(node.id)
+                : node;
+        assert(filteredNode !== undefined);
+        if (filteredNode.isFilterTarget) {
+            this.addTarget(changeTargets, filteredNode);
+            return;
+        }
+        if (filteredNode.type === "element2d" || filteredNode.type === "element3d") {
+            // need to add parent ids as filter target will be an element
+            this.addTarget(changeTargets, filteredNode);
+        }
+        if (!node.children) {
+            return;
+        }
+        for (const child of node.children.values()) {
+            this.collectFilterTargets(changeTargets, child, processedFilteredNodes);
+        }
+    }
+    addTarget(filterTargets, node) {
+        switch (node.type) {
+            case "classificationTable":
+                (filterTargets.classificationTableIds ??= new Set()).add(node.id);
+                return;
+            case "classification":
+                (filterTargets.classificationIds ??= new Set()).add(node.id);
+                return;
+            case "element2d":
+                const element2dKey = this.createModelCategoryKey(node.modelId, node.categoryId);
+                const elements2d = (filterTargets.elements2d ??= new Map()).get(element2dKey);
+                if (elements2d) {
+                    elements2d.set(node.id, { isFilterTarget: node.isFilterTarget });
+                }
+                else {
+                    filterTargets.elements2d.set(element2dKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));
+                }
+                return;
+            case "element3d":
+                const element3dKey = this.createModelCategoryKey(node.modelId, node.categoryId);
+                const elements3d = (filterTargets.elements3d ??= new Map()).get(element3dKey);
+                if (elements3d) {
+                    elements3d.set(node.id, { isFilterTarget: node.isFilterTarget });
+                }
+                else {
+                    filterTargets.elements3d.set(element3dKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));
+                }
+                return;
+        }
+    }
+    createModelCategoryKey(modelId, categoryId) {
+        return `${modelId}-${categoryId}`;
+    }
+    parseModelCategoryKey(key) {
+        const [modelId, categoryId] = key.split("-");
+        return { modelId, categoryId };
+    }
+    createFilteredTreeNode({ type, id, isFilterTarget, }) {
+        if (type === "element2d" || type === "element3d") {
+            return {
+                id,
+                isFilterTarget,
+                type,
+                modelId: undefined,
+                categoryId: undefined,
+            };
+        }
+        return {
+            id,
+            isFilterTarget,
+            type,
+        };
+    }
+    async getType(className) {
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_ClassificationTable)) {
+            return "classificationTable";
+        }
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Classification)) {
+            return "classification";
+        }
+        if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_GeometricElement2d)) {
+            return "element2d";
+        }
+        return "element3d";
+    }
+}
+//# sourceMappingURL=FilteredTree.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"FilteredTree.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/classifications-tree/internal/visibility/FilteredTree.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,8BAA8B,EAAE,6BAA6B,EAAE,MAAM,kDAAkD,CAAC;AAC5J,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,yDAAyD,CAAC;AAyDnH,gBAAgB;AAChB,MAAM,CAAC,KAAK,UAAU,iCAAiC,CAAC,KAIvD;IACC,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACzD,OAAO,kBAAkB,CAAC;QACxB,oBAAoB,EAAE,IAAI,uCAAuC,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QAC7F,cAAc;KACf,CAAC,CAAC;AACL,CAAC;AAqBD,MAAM,uCAAwC,SAAQ,oBAIrD;IACU,MAAM,CAA+C;IAC9D,YAAY,KAAmD;QAC7D,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,yBAAyB;QACpC,MAAM,2BAA2B,GAAG,IAAI,GAAG,EAAgE,CAAC;QAC5G,MAAM,2BAA2B,GAAG,IAAI,GAAG,EAAgE,CAAC;QAC5G,MAAM,MAAM,GAA2B;YACrC,kBAAkB,EAAE,IAAI,GAAG,EAAE;YAC7B,kBAAkB,EAAE,IAAI,GAAG,EAAE;SAC9B,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACzC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC9B,2BAA2B,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACjD,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrC,2BAA2B,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QAED,MAAM,sBAAsB,GAAG,MAAM,cAAc,CACjD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YAC3C,YAAY,EAAE,CAAC,GAAG,2BAA2B,CAAC,IAAI,EAAE,CAAC;YACrD,YAAY,EAAE,CAAC,GAAG,2BAA2B,CAAC,IAAI,EAAE,CAAC;SACtD,CAAC,CACH,CAAC;QACF,2BAA2B,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE;YAClD,MAAM,KAAK,GAAG,sBAAsB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC5B,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;QACH,2BAA2B,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,EAAE,EAAE;YAClD,MAAM,KAAK,GAAG,sBAAsB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC5B,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,2BAA2B,CAChC,aAA0C,EAC1C,sBAA8C;QAE9C,MAAM,aAAa,GAA0B,EAAE,CAAC;QAEhD,aAAa,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,YAAY,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAExH,OAAO,IAAI,CAAC,4BAA4B,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC;IAEO,4BAA4B,CAAC,aAAoC;QACvE,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,iBAAiB,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YACnI,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,iBAAiB,EAAE,aAAa,CAAC,iBAAiB;YAClD,sBAAsB,EAAE,aAAa,CAAC,iBAAiB;YACvD,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBAC5E,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAC3C,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;YACb,UAAU,EAAE,aAAa,CAAC,UAAU;gBAClC,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB,EAAE,QAAQ,CAAC,EAAE,EAAE;oBAC5E,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC7E,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;gBAC3C,CAAC,CAAC;gBACJ,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,aAAoC,EAAE,IAA+B,EAAE,sBAA8C;QAChJ,MAAM,YAAY,GAChB,IAAI,CAAC,IAAI,KAAK,WAAW;YACvB,CAAC,CAAC,sBAAsB,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACxD,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW;gBACzB,CAAC,CAAC,sBAAsB,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxD,CAAC,CAAC,IAAI,CAAC;QACb,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC;QACnC,IAAI,YAAY,CAAC,cAAc,EAAE,CAAC;YAChC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,YAAY,CAAC,IAAI,KAAK,WAAW,IAAI,YAAY,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3E,6DAA6D;YAC7D,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,oBAAoB,CAAC,aAAa,EAAE,KAAK,EAAE,sBAAsB,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,aAAoC,EAAE,IAAsB;QAC5E,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,qBAAqB;gBACxB,CAAC,aAAa,CAAC,sBAAsB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClE,OAAO;YACT,KAAK,gBAAgB;gBACnB,CAAC,aAAa,CAAC,iBAAiB,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7D,OAAO;YACT,KAAK,WAAW;gBACd,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChF,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9E,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5G,CAAC;gBACD,OAAO;YACT,KAAK,WAAW;gBACd,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;gBAChF,MAAM,UAAU,GAAG,CAAC,aAAa,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC9E,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5G,CAAC;gBACD,OAAO;QACX,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,OAAmB,EAAE,UAAsB;QACxE,OAAO,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;IACpC,CAAC;IAEO,qBAAqB,CAAC,GAAqB;QACjD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;IACjC,CAAC;IAEM,sBAAsB,CAAC,EAC5B,IAAI,EACJ,EAAE,EACF,cAAc,GAKf;QACC,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACjD,OAAO;gBACL,EAAE;gBACF,cAAc;gBACd,IAAI;gBACJ,OAAO,EAAE,SAAS;gBAClB,UAAU,EAAE,SAAS;aACtB,CAAC;QACJ,CAAC;QACD,OAAO;YACL,EAAE;YACF,cAAc;YACd,IAAI;SACL,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,SAAiB;QACpC,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,8BAA8B,CAAC,EAAE,CAAC;YAC/F,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,yBAAyB,CAAC,EAAE,CAAC;YAC1F,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,6BAA6B,CAAC,EAAE,CAAC;YAC9F,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { firstValueFrom } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_Classification, CLASS_NAME_ClassificationTable, CLASS_NAME_GeometricElement2d } from \"../../../common/internal/ClassNameDefinitions.js\";\nimport { createFilteredTree, FilteredNodesHandler } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\n\nimport type { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { CategoryId, ElementId, ModelId } from \"../../../common/internal/Types.js\";\nimport type { BaseFilteredTreeNode, FilteredTree, FilteredTreeNodeChildren } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\nimport type { ClassificationsTreeIdsCache } from \"../ClassificationsTreeIdsCache.js\";\n\ninterface ClassificationTableFilteredTreeNode extends BaseFilteredTreeNode<ClassificationTableFilteredTreeNode> {\n  type: \"classificationTable\";\n}\n\ninterface ClassificationFilteredTreeNode extends BaseFilteredTreeNode<ClassificationFilteredTreeNode> {\n  type: \"classification\";\n}\n\ninterface Element2dFilteredTreeNode extends BaseFilteredTreeNode<Element2dFilteredTreeNode> {\n  type: \"element2d\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ninterface Element3dFilteredTreeNode extends BaseFilteredTreeNode<Element3dFilteredTreeNode> {\n  type: \"element3d\";\n  categoryId: Id64String;\n  modelId: Id64String;\n}\n\ntype FilteredTreeNode = ClassificationTableFilteredTreeNode | ClassificationFilteredTreeNode | Element2dFilteredTreeNode | Element3dFilteredTreeNode;\n\ntype TemporaryElement2dFilteredNode = Omit<Element2dFilteredTreeNode, \"modelId\" | \"categoryId\" | \"children\"> & {\n  modelId: string | undefined;\n  categoryId: string | undefined;\n  children?: FilteredTreeNodeChildren<TemporaryElement2dFilteredNode>;\n};\n\ntype TemporaryElement3dFilteredNode = Omit<Element3dFilteredTreeNode, \"modelId\" | \"categoryId\" | \"children\"> & {\n  modelId: string | undefined;\n  categoryId: string | undefined;\n  children?: FilteredTreeNodeChildren<TemporaryElement3dFilteredNode>;\n};\n\ntype TemporaryFilteredTreeNode =\n  | ClassificationTableFilteredTreeNode\n  | ClassificationFilteredTreeNode\n  | TemporaryElement2dFilteredNode\n  | TemporaryElement3dFilteredNode;\n\n/** @internal */\nexport interface ClassificationsTreeFilterTargets {\n  elements2d?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isFilterTarget: boolean }> }>;\n  elements3d?: Array<{ modelId: Id64String; categoryId: Id64String; elements: Map<ElementId, { isFilterTarget: boolean }> }>;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\n/** @internal */\nexport async function createFilteredClassificationsTree(props: {\n  imodelAccess: ECClassHierarchyInspector;\n  filteringPaths: HierarchyFilteringPath[];\n  idsCache: ClassificationsTreeIdsCache;\n}): Promise<FilteredTree<ClassificationsTreeFilterTargets>> {\n  const { imodelAccess, filteringPaths, idsCache } = props;\n  return createFilteredTree({\n    filteredNodesHandler: new ClassificationsTreeFilteredNodesHandler({ idsCache, imodelAccess }),\n    filteringPaths,\n  });\n}\n\ninterface FilterTargetsInternal {\n  elements2d?: Map<ModelCategoryKey, Map<ElementId, { isFilterTarget: boolean }>>;\n  elements3d?: Map<ModelCategoryKey, Map<ElementId, { isFilterTarget: boolean }>>;\n  classificationTableIds?: Id64Set;\n  classificationIds?: Id64Set;\n}\n\ninterface ClassificationsTreeFilteredNodesHandlerProps {\n  idsCache: ClassificationsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n}\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\ninterface ProcessedFilteredNodes {\n  filtered2dElements: Map<Id64String, Omit<Element2dFilteredTreeNode, \"children\">>;\n  filtered3dElements: Map<Id64String, Omit<Element3dFilteredTreeNode, \"children\">>;\n}\n\nclass ClassificationsTreeFilteredNodesHandler extends FilteredNodesHandler<\n  ProcessedFilteredNodes,\n  ClassificationsTreeFilterTargets,\n  TemporaryFilteredTreeNode\n> {\n  readonly #props: ClassificationsTreeFilteredNodesHandlerProps;\n  constructor(props: ClassificationsTreeFilteredNodesHandlerProps) {\n    super();\n    this.#props = props;\n  }\n\n  public async getProcessedFilteredNodes(): Promise<ProcessedFilteredNodes> {\n    const filteredTemporary2dElements = new Map<Id64String, Omit<TemporaryElement2dFilteredNode, \"children\">>();\n    const filteredTemporary3dElements = new Map<Id64String, Omit<TemporaryElement3dFilteredNode, \"children\">>();\n    const result: ProcessedFilteredNodes = {\n      filtered2dElements: new Map(),\n      filtered3dElements: new Map(),\n    };\n    for (const node of this.filteredNodesArr) {\n      if (node.type === \"element2d\") {\n        filteredTemporary2dElements.set(node.id, node);\n      } else if (node.type === \"element3d\") {\n        filteredTemporary3dElements.set(node.id, node);\n      }\n    }\n\n    const filteredElementsModels = await firstValueFrom(\n      this.#props.idsCache.getFilteredElementsData({\n        element2dIds: [...filteredTemporary2dElements.keys()],\n        element3dIds: [...filteredTemporary3dElements.keys()],\n      }),\n    );\n    filteredTemporary2dElements.forEach((element, id) => {\n      const entry = filteredElementsModels.get(element.id);\n      assert(entry !== undefined);\n      result.filtered2dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });\n    });\n    filteredTemporary3dElements.forEach((element, id) => {\n      const entry = filteredElementsModels.get(element.id);\n      assert(entry !== undefined);\n      result.filtered3dElements.set(id, { ...element, modelId: entry.modelId, categoryId: entry.categoryId });\n    });\n    return result;\n  }\n\n  public convertNodesToFilterTargets(\n    filteredNodes: TemporaryFilteredTreeNode[],\n    processedFilteredNodes: ProcessedFilteredNodes,\n  ): ClassificationsTreeFilterTargets | undefined {\n    const filterTargets: FilterTargetsInternal = {};\n\n    filteredNodes.forEach((filteredNode) => this.collectFilterTargets(filterTargets, filteredNode, processedFilteredNodes));\n\n    return this.convertInternalFilterTargets(filterTargets);\n  }\n\n  private convertInternalFilterTargets(filterTargets: FilterTargetsInternal): ClassificationsTreeFilterTargets | undefined {\n    if (!filterTargets.classificationIds && !filterTargets.classificationIds && !filterTargets.elements2d && !filterTargets.elements3d) {\n      return undefined;\n    }\n\n    return {\n      classificationIds: filterTargets.classificationIds,\n      classificationTableIds: filterTargets.classificationIds,\n      elements2d: filterTargets.elements2d\n        ? [...filterTargets.elements2d?.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n      elements3d: filterTargets.elements3d\n        ? [...filterTargets.elements3d?.entries()].map(([modelCategoryKey, elements]) => {\n            const { modelId, categoryId } = this.parseModelCategoryKey(modelCategoryKey);\n            return { modelId, categoryId, elements };\n          })\n        : undefined,\n    };\n  }\n\n  private collectFilterTargets(changeTargets: FilterTargetsInternal, node: TemporaryFilteredTreeNode, processedFilteredNodes: ProcessedFilteredNodes) {\n    const filteredNode =\n      node.type === \"element2d\"\n        ? processedFilteredNodes.filtered2dElements.get(node.id)\n        : node.type === \"element3d\"\n          ? processedFilteredNodes.filtered3dElements.get(node.id)\n          : node;\n    assert(filteredNode !== undefined);\n    if (filteredNode.isFilterTarget) {\n      this.addTarget(changeTargets, filteredNode);\n      return;\n    }\n\n    if (filteredNode.type === \"element2d\" || filteredNode.type === \"element3d\") {\n      // need to add parent ids as filter target will be an element\n      this.addTarget(changeTargets, filteredNode);\n    }\n\n    if (!node.children) {\n      return;\n    }\n\n    for (const child of node.children.values()) {\n      this.collectFilterTargets(changeTargets, child, processedFilteredNodes);\n    }\n  }\n\n  private addTarget(filterTargets: FilterTargetsInternal, node: FilteredTreeNode) {\n    switch (node.type) {\n      case \"classificationTable\":\n        (filterTargets.classificationTableIds ??= new Set()).add(node.id);\n        return;\n      case \"classification\":\n        (filterTargets.classificationIds ??= new Set()).add(node.id);\n        return;\n      case \"element2d\":\n        const element2dKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements2d = (filterTargets.elements2d ??= new Map()).get(element2dKey);\n        if (elements2d) {\n          elements2d.set(node.id, { isFilterTarget: node.isFilterTarget });\n        } else {\n          filterTargets.elements2d.set(element2dKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));\n        }\n        return;\n      case \"element3d\":\n        const element3dKey = this.createModelCategoryKey(node.modelId, node.categoryId);\n        const elements3d = (filterTargets.elements3d ??= new Map()).get(element3dKey);\n        if (elements3d) {\n          elements3d.set(node.id, { isFilterTarget: node.isFilterTarget });\n        } else {\n          filterTargets.elements3d.set(element3dKey, new Map([[node.id, { isFilterTarget: node.isFilterTarget }]]));\n        }\n        return;\n    }\n  }\n\n  private createModelCategoryKey(modelId: Id64String, categoryId: Id64String): ModelCategoryKey {\n    return `${modelId}-${categoryId}`;\n  }\n\n  private parseModelCategoryKey(key: ModelCategoryKey): { modelId: Id64String; categoryId: Id64String } {\n    const [modelId, categoryId] = key.split(\"-\");\n    return { modelId, categoryId };\n  }\n\n  public createFilteredTreeNode({\n    type,\n    id,\n    isFilterTarget,\n  }: {\n    type: FilteredTreeNode[\"type\"];\n    id: Id64String;\n    isFilterTarget: boolean;\n  }): TemporaryFilteredTreeNode {\n    if (type === \"element2d\" || type === \"element3d\") {\n      return {\n        id,\n        isFilterTarget,\n        type,\n        modelId: undefined,\n        categoryId: undefined,\n      };\n    }\n    return {\n      id,\n      isFilterTarget,\n      type,\n    };\n  }\n\n  public async getType(className: string): Promise<TemporaryFilteredTreeNode[\"type\"]> {\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_ClassificationTable)) {\n      return \"classificationTable\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_Classification)) {\n      return \"classification\";\n    }\n    if (await this.#props.imodelAccess.classDerivesFrom(className, CLASS_NAME_GeometricElement2d)) {\n      return \"element2d\";\n    }\n    return \"element3d\";\n  }\n}\n"]}