@itwin/tree-widget-react 3.0.0-dev.0 → 3.0.0-dev.1

package/lib/esm/components/trees/models-tree/ModelsTree.js

@@ -1,150 +1,214 @@
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
 /*---------------------------------------------------------------------------------------------
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import "../VisibilityTreeBase.scss";
-import classNames from "classnames";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
-import { SelectionMode } from "@itwin/components-react";
-import { isPresentationTreeNodeItem, PresentationTree } from "@itwin/presentation-components";
+import { IModelApp } from "@itwin/core-frontend";
+import { SvgFolder, SvgImodelHollow, SvgItem, SvgLayers, SvgModel } from "@itwin/itwinui-icons-react";
+import { Anchor, Icon, Text } from "@itwin/itwinui-react";
+import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
+import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
 import { TreeWidget } from "../../../TreeWidget";
-import { FilterableTreeRenderer } from "../common/TreeRenderer";
-import { ClassGroupingOption } from "../common/Types";
+import { VisibilityTree } from "../common/components/VisibilityTree";
+import { useFocusedInstancesContext } from "../common/FocusedInstancesContext";
 import { useFeatureReporting } from "../common/UseFeatureReporting";
-import { usePerformanceReporting } from "../common/UsePerformanceReporting";
-import { useVisibilityTreeState } from "../common/UseVisibilityTreeState";
-import { addCustomTreeNodeItemLabelRenderer, addTreeNodeItemCheckbox, combineTreeNodeItemCustomizations } from "../common/Utils";
-import { createVisibilityTreeRenderer, FilterableVisibilityTreeNodeRenderer, VisibilityTreeNoFilteredData } from "../VisibilityTreeRenderer";
+import { useIModelChangeListener } from "../common/UseIModelChangeListener";
+import { ModelsTreeIdsCache } from "./internal/ModelsTreeIdsCache";
+import { createModelsTreeVisibilityHandler } from "./internal/ModelsTreeVisibilityHandler";
 import { ModelsTreeComponent } from "./ModelsTreeComponent";
-import { ModelsTreeEventHandler } from "./ModelsTreeEventHandler";
-import { ModelsVisibilityHandler, SubjectModelIdsCache } from "./ModelsVisibilityHandler";
-import { addModelsTreeNodeItemIcons, createRuleset, createSearchRuleset } from "./Utils";
-const PAGING_SIZE = 20;
-/**
- * A tree component that shows a subject - model - category - element
- * hierarchy along with checkboxes that represent and allow changing
- * the display of those instances.
- * @public
- */
-export function ModelsTree(props) {
-    const { hierarchyLevelConfig, density, height, width, selectionMode, onFeatureUsed } = props;
-    const { reportUsage } = useFeatureReporting({ treeIdentifier: ModelsTreeComponent.id, onFeatureUsed });
-    const state = useModelsTreeState({ ...props, reportUsage });
-    const baseRendererProps = {
-        reportUsage,
-        contextMenuItems: props.contextMenuItems,
-        nodeLabelRenderer: props.nodeLabelRenderer,
-        density: props.density,
-        nodeRendererProps: {
-            iconsEnabled: true,
-            descriptionEnabled: false,
-            levelOffset: 10,
-            disableRootNodeCollapse: true,
-            onVisibilityToggled: () => reportUsage({ featureId: "visibility-change", reportInteraction: true }),
-        },
-    };
-    // istanbul ignore next
-    const noFilteredDataRenderer = useCallback(() => {
-        return (_jsx(VisibilityTreeNoFilteredData, { title: TreeWidget.translate("modelTree.noModelFound"), message: TreeWidget.translate("modelTree.noMatchingModelNames") }));
-    }, []);
-    if (!state) {
-        return null;
+import { defaultHierarchyConfiguration, ModelsTreeDefinition } from "./ModelsTreeDefinition";
+/** @internal */
+export function ModelsTree({ imodel, getSchemaContext, height, width, activeView, filter, density, hierarchyLevelConfig, hierarchyConfig, selectionMode, onPerformanceMeasured, onFeatureUsed, }) {
+    const [filteringError, setFilteringError] = useState(undefined);
+    const hierarchyConfiguration = useMemo(() => ({
+        ...defaultHierarchyConfiguration,
+        ...hierarchyConfig,
+    }), 
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    Object.values(hierarchyConfig ?? {}));
+    const { getModelsTreeIdsCache, visibilityHandlerFactory } = useCachedVisibility(activeView, hierarchyConfiguration);
+    const { loadInstanceKeys: loadFocusedInstancesKeys } = useFocusedInstancesContext();
+    const { reportUsage } = useFeatureReporting({ onFeatureUsed, treeIdentifier: ModelsTreeComponent.id });
+    const getHierarchyDefinition = useCallback(({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration }), [getModelsTreeIdsCache, hierarchyConfiguration]);
+    const onNodeDoubleClick = useCallback(async ({ nodeData, extendedData }) => {
+        if (!HierarchyNode.isInstancesNode(nodeData) || (extendedData && (extendedData.isSubject || extendedData.isModel || extendedData.isCategory))) {
+            return;
+        }
+        const instanceIds = nodeData.key.instanceKeys.map((instanceKey) => instanceKey.id);
+        await IModelApp.viewManager.selectedView?.zoomToElements(instanceIds);
+        reportUsage({ featureId: "zoom-to-node", reportInteraction: false });
+    }, [reportUsage]);
+    const getFocusedFilteredPaths = useMemo(() => {
+        setFilteringError(undefined);
+        if (!loadFocusedInstancesKeys) {
+            return undefined;
+        }
+        return async ({ imodelAccess }) => {
+            try {
+                const targetKeys = await collectTargetKeys(loadFocusedInstancesKeys);
+                return await ModelsTreeDefinition.createInstanceKeyPaths({
+                    imodelAccess,
+                    idsCache: getModelsTreeIdsCache(),
+                    keys: targetKeys,
+                    hierarchyConfig: hierarchyConfiguration,
+                });
+            }
+            catch (e) {
+                const newError = e instanceof Error && e.message.match(/Filter matches more than \d+ items/) ? "tooManyInstancesFocused" : "unknownInstanceFocusError";
+                setFilteringError(newError);
+                return [];
+            }
+        };
+    }, [loadFocusedInstancesKeys, getModelsTreeIdsCache, hierarchyConfiguration]);
+    const getSearchFilteredPaths = useMemo(() => {
+        setFilteringError(undefined);
+        if (!filter) {
+            return undefined;
+        }
+        return async ({ imodelAccess }) => {
+            reportUsage?.({ featureId: "filtering", reportInteraction: true });
+            try {
+                return await ModelsTreeDefinition.createInstanceKeyPaths({
+                    imodelAccess,
+                    label: filter,
+                    idsCache: getModelsTreeIdsCache(),
+                    hierarchyConfig: hierarchyConfiguration,
+                });
+            }
+            catch (e) {
+                const newError = e instanceof Error && e.message.match(/Filter matches more than \d+ items/) ? "tooManyFilterMatches" : "unknownFilterError";
+                setFilteringError(newError);
+                return [];
+            }
+        };
+    }, [filter, getModelsTreeIdsCache, reportUsage, hierarchyConfiguration]);
+    const getFilteredPaths = getFocusedFilteredPaths ?? getSearchFilteredPaths;
+    return (_jsx(VisibilityTree, { height: height, width: width, imodel: imodel, treeName: ModelsTreeComponent.id, getSchemaContext: getSchemaContext, visibilityHandlerFactory: visibilityHandlerFactory, getHierarchyDefinition: getHierarchyDefinition, getFilteredPaths: getFilteredPaths, hierarchyLevelSizeLimit: hierarchyLevelConfig?.sizeLimit, getIcon: getIcon, density: density, noDataMessage: getNoDataMessage(filter, filteringError), selectionMode: selectionMode, onPerformanceMeasured: (action, duration) => {
+            onPerformanceMeasured?.(`${ModelsTreeComponent.id}-${action}`, duration);
+        }, reportUsage: reportUsage, onNodeDoubleClick: onNodeDoubleClick, searchText: filter }));
+}
+function getNoDataMessage(filter, error) {
+    if (isInstanceFocusError(error)) {
+        return _jsx(InstanceFocusError, { error: error });
+    }
+    if (isFilterError(error)) {
+        return _jsx(Text, { children: TreeWidget.translate(`modelsTree.filtering.${error}`) });
     }
-    const isFilterApplied = state.filteringResult?.filteredProvider !== undefined;
-    const overlay = state.filteringResult?.isFiltering ? _jsx("div", { className: "filteredTreeOverlay" }) : undefined;
-    return (_jsxs("div", { className: classNames("tree-widget-visibility-tree-base", "tree-widget-tree-container"), children: [_jsx(PresentationTree, { state: state, selectionMode: selectionMode || SelectionMode.None, treeRenderer: hierarchyLevelConfig?.isFilteringEnabled
-                    ? (rendererProps) => (_jsx(FilterableTreeRenderer, { ...rendererProps, ...baseRendererProps, nodeLoader: state.nodeLoader, nodeRenderer: (nodeProps) => _jsx(ModelsTreeNodeRenderer, { ...nodeProps, density: density }) }))
-                    : createVisibilityTreeRenderer(baseRendererProps), noDataRenderer: isFilterApplied ? noFilteredDataRenderer : undefined, width: width, height: height }), overlay] }));
+    if (filter) {
+        return _jsx(Text, { children: TreeWidget.translate("modelsTree.filtering.noMatches", { filter }) });
+    }
+    return undefined;
 }
-function ModelsTreeNodeRenderer(props) {
-    return (_jsx(FilterableVisibilityTreeNodeRenderer, { ...props, iconsEnabled: true, descriptionEnabled: false, levelOffset: 10, disableRootNodeCollapse: true, isEnlarged: props.density === "enlarged", onVisibilityToggled: () => props.reportUsage?.({ featureId: "visibility-change", reportInteraction: true }) }));
+function isFilterError(error) {
+    return error === "tooManyFilterMatches" || error === "unknownFilterError";
 }
-function useModelsTreeState({ filterInfo, onFilterApplied, ...props }) {
-    const rulesets = {
-        general: useMemo(() => createRuleset({
-            enableElementsClassGrouping: !!props.hierarchyConfig?.enableElementsClassGrouping,
-            elementClassSpecification: props.hierarchyConfig?.elementClassSpecification,
-            showEmptyModels: props.hierarchyConfig?.showEmptyModels,
-        }), [props.hierarchyConfig?.enableElementsClassGrouping, props.hierarchyConfig?.elementClassSpecification, props.hierarchyConfig?.showEmptyModels]),
-        search: useMemo(() => createSearchRuleset({
-            elementClassSpecification: props.hierarchyConfig?.elementClassSpecification,
-            showEmptyModels: props.hierarchyConfig?.showEmptyModels,
-        }), [props.hierarchyConfig?.elementClassSpecification, props.hierarchyConfig?.showEmptyModels]),
-    };
-    const treeState = useTreeState({
-        ...props,
-        ruleset: rulesets.general,
-    });
-    const filteredTreeState = useTreeState({
-        ...props,
-        ruleset: rulesets.search,
-        filterInfo,
-        onFilterApplied,
-    });
-    return filterInfo?.filter ? filteredTreeState : treeState;
+function isInstanceFocusError(error) {
+    return error === "tooManyInstancesFocused" || error === "unknownInstanceFocusError";
 }
-function useTreeState({ modelsVisibilityHandler, activeView, selectionPredicate, hierarchyConfig, iModel, ruleset, filterInfo, onFilterApplied, hierarchyLevelConfig, onPerformanceMeasured, reportUsage, }) {
-    const visibilityHandler = useVisibilityHandler(ruleset.id, iModel, activeView, modelsVisibilityHandler);
-    const selectionPredicateRef = useRef(selectionPredicate);
-    useEffect(() => {
-        selectionPredicateRef.current = selectionPredicate;
-    }, [selectionPredicate]);
-    const onFilterChange = useCallback((dataProvider, matchesCount) => {
-        if (dataProvider && matchesCount !== undefined) {
-            reportUsage({ featureId: "filtering", reportInteraction: false });
-            onFilterApplied?.(dataProvider, matchesCount);
-        }
-        if (visibilityHandler) {
-            visibilityHandler.setFilteredDataProvider(dataProvider);
+function InstanceFocusError({ error }) {
+    const { toggle } = useFocusedInstancesContext();
+    const localizedMessage = createLocalizedMessage(TreeWidget.translate(`modelsTree.filtering.${error}`), () => toggle());
+    return _jsx(Text, { children: localizedMessage });
+}
+function getIcon(node) {
+    if (node.extendedData?.imageId === undefined) {
+        return undefined;
+    }
+    switch (node.extendedData.imageId) {
+        case "icon-layers":
+            return _jsx(SvgLayers, {});
+        case "icon-item":
+            return _jsx(SvgItem, {});
+        case "icon-ec-class":
+            return _jsx(SvgClassGrouping, {});
+        case "icon-imodel-hollow-2":
+            return _jsx(SvgImodelHollow, {});
+        case "icon-folder":
+            return _jsx(SvgFolder, {});
+        case "icon-model":
+            return _jsx(SvgModel, {});
+    }
+    return undefined;
+}
+function createVisibilityHandlerFactory(activeView, idsCacheGetter) {
+    return (imodelAccess) => createModelsTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess });
+}
+function useCachedVisibility(activeView, hierarchyConfig) {
+    const cacheRef = useRef();
+    const currentIModelRef = useRef(activeView.iModel);
+    const getModelsTreeIdsCache = useCallback(() => {
+        if (!cacheRef.current) {
+            cacheRef.current = new ModelsTreeIdsCache(createECSqlQueryExecutor(currentIModelRef.current), hierarchyConfig);
         }
-    }, [onFilterApplied, reportUsage, visibilityHandler]);
-    const { onNodeLoaded } = usePerformanceReporting({
-        treeIdentifier: ModelsTreeComponent.id,
-        onPerformanceMeasured,
+        return cacheRef.current;
+    }, [hierarchyConfig]);
+    const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));
+    useIModelChangeListener({
+        imodel: activeView.iModel,
+        action: useCallback(() => {
+            cacheRef.current = undefined;
+            setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));
+        }, [activeView, getModelsTreeIdsCache]),
     });
-    const eventHandlerFactory = useCallback((handlerProps) => {
-        return new ModelsTreeEventHandler({ ...handlerProps, reportUsage });
-    }, [reportUsage]);
-    return useVisibilityTreeState({
-        imodel: iModel,
-        ruleset,
-        pagingSize: PAGING_SIZE,
-        appendChildrenCountForGroupingNodes: hierarchyConfig?.enableElementsClassGrouping === ClassGroupingOption.YesWithCounts,
-        enableHierarchyAutoUpdate: true,
-        customizeTreeNodeItem,
-        visibilityHandler,
-        filterInfo,
-        onFilterChange,
-        selectionPredicate: useCallback((node) => !selectionPredicateRef.current || !isPresentationTreeNodeItem(node)
-            ? true
-            : selectionPredicateRef.current(node.key, ModelsVisibilityHandler.getNodeType(node)), []),
-        eventHandler: eventHandlerFactory,
-        hierarchyLevelSizeLimit: hierarchyLevelConfig?.sizeLimit,
-        onNodeLoaded: filterInfo ? undefined : onNodeLoaded,
-        reportUsage: filterInfo ? undefined : reportUsage,
-        onHierarchyLimitExceeded: () => reportUsage({ featureId: "hierarchy-level-size-limit-hit", reportInteraction: false }),
-    });
-}
-function useVisibilityHandler(rulesetId, iModel, activeView, visibilityHandler) {
-    const subjectModelIdsCache = useMemo(() => new SubjectModelIdsCache(iModel), [iModel]);
-    const [state, setState] = useState();
     useEffect(() => {
-        if (visibilityHandler && typeof visibilityHandler !== "function") {
-            return;
+        currentIModelRef.current = activeView.iModel;
+        cacheRef.current = undefined;
+        setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));
+    }, [activeView, getModelsTreeIdsCache]);
+    return {
+        getModelsTreeIdsCache,
+        visibilityHandlerFactory,
+    };
+}
+function SvgClassGrouping() {
+    return (_jsx(Icon, { children: _jsx("svg", { id: "Calque_1", "data-name": "Calque 1", xmlns: "http://www.w3.org/2000/svg", viewBox: "0 0 16 16", children: _jsx("path", { d: "M8.00933,0,0,3.97672V11.986L8.00933,16,16,11.93V3.97651ZM1.66173,11.27642c-.26155.03734-.59754-.26154-.76553-.69085-.168-.41066-.09334-.784.168-.82152.26154-.03734.59754.26154.76553.67219C1.99772,10.86577,1.92306,11.23909,1.66173,11.27642Zm0-3.32319c-.26155.03733-.59754-.28-.76553-.69086-.168-.42932-.09334-.80285.168-.84.26133-.03733.59754.28.76532.69086C1.99772,7.54236,1.92306,7.89723,1.66173,7.95323Zm4.31276,5.52621a.18186.18186,0,0,1-.16821-.01866L3.41657,12.15394a.94275.94275,0,0,1-.29887-.80285c.03754-.33621.22421-.52265.41108-.41066L5.9185,12.24727a.88656.88656,0,0,1,.28.80285A.5057.5057,0,0,1,5.97449,13.47944Zm0-3.37919a.18184.18184,0,0,1-.16821-.01867L3.41657,8.77475a.943.943,0,0,1-.29887-.80286c.03754-.3362.22421-.52286.41108-.42953L5.9185,8.86786a.83112.83112,0,0,1,.28.78419A.51684.51684,0,0,1,5.97449,10.10025Z" }) }) }));
+}
+async function collectTargetKeys(loadFocusedInstancesKeys) {
+    const targetKeys = [];
+    const groupingNodeInfos = [];
+    for await (const key of loadFocusedInstancesKeys()) {
+        if ("id" in key) {
+            targetKeys.push(key);
+            continue;
         }
-        const visibilityHandlerProps = {
-            rulesetId,
-            viewport: activeView,
-            subjectModelIdsCache,
-        };
-        const handler = visibilityHandler ? visibilityHandler(visibilityHandlerProps) : new ModelsVisibilityHandler(visibilityHandlerProps);
-        setState(handler);
-        return () => {
-            handler.dispose();
-        };
-    }, [rulesetId, activeView, subjectModelIdsCache, visibilityHandler]);
-    return visibilityHandler && typeof visibilityHandler !== "function" ? visibilityHandler : state;
+        if (!HierarchyNodeKey.isClassGrouping(key.key)) {
+            targetKeys.push(...key.groupedInstanceKeys);
+            continue;
+        }
+        if (!key.nonGroupingAncestor || !HierarchyNodeKey.isInstances(key.nonGroupingAncestor.key)) {
+            continue;
+        }
+        const parentKey = key.nonGroupingAncestor.key;
+        const type = key.nonGroupingAncestor.extendedData?.isCategory ? "category" : "element";
+        const modelIds = (key.nonGroupingAncestor.extendedData?.modelIds ?? []).flatMap((ids) => ids);
+        const groupInfo = groupingNodeInfos.find((group) => HierarchyNodeKey.equals(group.parentKey, parentKey));
+        if (groupInfo) {
+            groupInfo.classes.push(key.key.className);
+        }
+        else {
+            groupingNodeInfos.push({ classes: [key.key.className], parentType: type, parentKey, modelIds });
+        }
+    }
+    targetKeys.push(...groupingNodeInfos.map(({ parentKey, parentType, classes, modelIds }) => ({
+        parent: parentType === "element"
+            ? { type: "element", ids: parentKey.instanceKeys.map((key) => key.id) }
+            : { type: "category", ids: parentKey.instanceKeys.map((key) => key.id), modelIds },
+        classes,
+    })));
+    return targetKeys;
+}
+function createLocalizedMessage(message, onClick) {
+    const exp = new RegExp("<link>(.*)</link>");
+    const match = message.match(exp);
+    if (!match) {
+        return message;
+    }
+    const [fullText, innerText] = match;
+    const [textBefore, textAfter] = message.split(fullText);
+    return (_jsxs(_Fragment, { children: [textBefore ? textBefore : null, _jsx(Anchor, { underline: true, onClick: (e) => {
+                    e.stopPropagation();
+                    onClick?.();
+                }, children: innerText }), textAfter ? textAfter : null] }));
 }
-const customizeTreeNodeItem = combineTreeNodeItemCustomizations([addCustomTreeNodeItemLabelRenderer, addTreeNodeItemCheckbox, addModelsTreeNodeItemIcons]);
 //# sourceMappingURL=ModelsTree.js.map

package/lib/esm/components/trees/models-tree/ModelsTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"ModelsTree.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,4BAA4B,CAAC;AACpC,OAAO,UAAU,MAAM,YAAY,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAC9F,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,sBAAsB,EAAE,MAAM,kCAAkC,CAAC;AAC1E,OAAO,EAAE,kCAAkC,EAAE,uBAAuB,EAAE,iCAAiC,EAAE,MAAM,iBAAiB,CAAC;AACjI,OAAO,EAAE,4BAA4B,EAAE,oCAAoC,EAAE,4BAA4B,EAAE,MAAM,2BAA2B,CAAC;AAC7I,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAC1F,OAAO,EAAE,0BAA0B,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAWzF,MAAM,WAAW,GAAG,EAAE,CAAC;AAiEvB;;;;;GAKG;AACH,MAAM,UAAU,UAAU,CAAC,KAAsB;IAC/C,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;IAC7F,MAAM,EAAE,WAAW,EAAE,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,mBAAmB,CAAC,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;IACvG,MAAM,KAAK,GAAG,kBAAkB,CAAC,EAAE,GAAG,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IAE5D,MAAM,iBAAiB,GAAG;QACxB,WAAW;QACX,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;QACxC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;QAC1C,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,iBAAiB,EAAE;YACjB,YAAY,EAAE,IAAI;YAClB,kBAAkB,EAAE,KAAK;YACzB,WAAW,EAAE,EAAE;YACf,uBAAuB,EAAE,IAAI;YAC7B,mBAAmB,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;SACpG;KACF,CAAC;IAEF,uBAAuB;IACvB,MAAM,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC9C,OAAO,CACL,KAAC,4BAA4B,IAAC,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,wBAAwB,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC,gCAAgC,CAAC,GAAI,CACzJ,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,IAAI,CAAC;KACb;IAED,MAAM,eAAe,GAAG,KAAK,CAAC,eAAe,EAAE,gBAAgB,KAAK,SAAS,CAAC;IAC9E,MAAM,OAAO,GAAG,KAAK,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC,cAAK,SAAS,EAAC,qBAAqB,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;IACzG,OAAO,CACL,eAAK,SAAS,EAAE,UAAU,CAAC,kCAAkC,EAAE,4BAA4B,CAAC,aAC1F,KAAC,gBAAgB,IACf,KAAK,EAAE,KAAK,EACZ,aAAa,EAAE,aAAa,IAAI,aAAa,CAAC,IAAI,EAClD,YAAY,EACV,oBAAoB,EAAE,kBAAkB;oBACtC,CAAC,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,CACjB,KAAC,sBAAsB,OACjB,aAAa,KACb,iBAAiB,EACrB,UAAU,EAAE,KAAK,CAAC,UAAU,EAC5B,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,KAAC,sBAAsB,OAAK,SAAS,EAAE,OAAO,EAAE,OAAO,GAAI,GACxF,CACH;oBACH,CAAC,CAAC,4BAA4B,CAAC,iBAAiB,CAAC,EAErD,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,SAAS,EACpE,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,GACd,EACD,OAAO,IACJ,CACP,CAAC;AACJ,CAAC;AAMD,SAAS,sBAAsB,CAAC,KAAkC;IAChE,OAAO,CACL,KAAC,oCAAoC,OAC/B,KAAK,EACT,YAAY,EAAE,IAAI,EAClB,kBAAkB,EAAE,KAAK,EACzB,WAAW,EAAE,EAAE,EACf,uBAAuB,EAAE,IAAI,EAC7B,UAAU,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,EACxC,mBAAmB,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,GAC3G,CACH,CAAC;AACJ,CAAC;AAMD,SAAS,kBAAkB,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,KAAK,EAA2B;IAC5F,MAAM,QAAQ,GAAG;QACf,OAAO,EAAE,OAAO,CACd,GAAG,EAAE,CACH,aAAa,CAAC;YACZ,2BAA2B,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,EAAE,2BAA2B;YACjF,yBAAyB,EAAE,KAAK,CAAC,eAAe,EAAE,yBAAyB;YAC3E,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe;SACxD,CAAC,EACJ,CAAC,KAAK,CAAC,eAAe,EAAE,2BAA2B,EAAE,KAAK,CAAC,eAAe,EAAE,yBAAyB,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe,CAAC,CAC/I;QACD,MAAM,EAAE,OAAO,CACb,GAAG,EAAE,CACH,mBAAmB,CAAC;YAClB,yBAAyB,EAAE,KAAK,CAAC,eAAe,EAAE,yBAAyB;YAC3E,eAAe,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe;SACxD,CAAC,EACJ,CAAC,KAAK,CAAC,eAAe,EAAE,yBAAyB,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe,CAAC,CAC3F;KACF,CAAC;IAEF,MAAM,SAAS,GAAG,YAAY,CAAC;QAC7B,GAAG,KAAK;QACR,OAAO,EAAE,QAAQ,CAAC,OAAO;KAC1B,CAAC,CAAC;IAEH,MAAM,iBAAiB,GAAG,YAAY,CAAC;QACrC,GAAG,KAAK;QACR,OAAO,EAAE,QAAQ,CAAC,MAAM;QACxB,UAAU;QACV,eAAe;KAChB,CAAC,CAAC;IAEH,OAAO,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5D,CAAC;AAOD,SAAS,YAAY,CAAC,EACpB,uBAAuB,EACvB,UAAU,EACV,kBAAkB,EAClB,eAAe,EACf,MAAM,EACN,OAAO,EACP,UAAU,EACV,eAAe,EACf,oBAAoB,EACpB,qBAAqB,EACrB,WAAW,GACE;IACb,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,uBAAuB,CAAC,CAAC;IACxG,MAAM,qBAAqB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;IACzD,SAAS,CAAC,GAAG,EAAE;QACb,qBAAqB,CAAC,OAAO,GAAG,kBAAkB,CAAC;IACrD,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAEzB,MAAM,cAAc,GAAG,WAAW,CAChC,CAAC,YAAoD,EAAE,YAAqB,EAAE,EAAE;QAC9E,IAAI,YAAY,IAAI,YAAY,KAAK,SAAS,EAAE;YAC9C,WAAW,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;YAClE,eAAe,EAAE,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;SAC/C;QAED,IAAI,iBAAiB,EAAE;YACrB,iBAAiB,CAAC,uBAAuB,CAAC,YAAY,CAAC,CAAC;SACzD;IACH,CAAC,EACD,CAAC,eAAe,EAAE,WAAW,EAAE,iBAAiB,CAAC,CAClD,CAAC;IAEF,MAAM,EAAE,YAAY,EAAE,GAAG,uBAAuB,CAAC;QAC/C,cAAc,EAAE,mBAAmB,CAAC,EAAE;QACtC,qBAAqB;KACtB,CAAC,CAAC;IAEH,MAAM,mBAAmB,GAAG,WAAW,CACrC,CAAC,YAA8C,EAAE,EAAE;QACjD,OAAO,IAAI,sBAAsB,CAAC,EAAE,GAAG,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC;IACtE,CAAC,EACD,CAAC,WAAW,CAAC,CACd,CAAC;IAEF,OAAO,sBAAsB,CAAC;QAC5B,MAAM,EAAE,MAAM;QACd,OAAO;QACP,UAAU,EAAE,WAAW;QACvB,mCAAmC,EAAE,eAAe,EAAE,2BAA2B,KAAK,mBAAmB,CAAC,aAAa;QACvH,yBAAyB,EAAE,IAAI;QAC/B,qBAAqB;QACrB,iBAAiB;QACjB,UAAU;QACV,cAAc;QACd,kBAAkB,EAAE,WAAW,CAC7B,CAAC,IAAkB,EAAE,EAAE,CACrB,CAAC,qBAAqB,CAAC,OAAO,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC;YACjE,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,qBAAqB,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,uBAAuB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,EACxF,EAAE,CACH;QACD,YAAY,EAAE,mBAAmB;QACjC,uBAAuB,EAAE,oBAAoB,EAAE,SAAS;QACxD,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;QACnD,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW;QACjD,wBAAwB,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;KACvH,CAAC,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CAC3B,SAAiB,EACjB,MAAwB,EACxB,UAAoB,EACpB,iBAAgH;IAEhH,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,oBAAoB,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IACvF,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,EAA2B,CAAC;IAE9D,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,iBAAiB,IAAI,OAAO,iBAAiB,KAAK,UAAU,EAAE;YAChE,OAAO;SACR;QAED,MAAM,sBAAsB,GAAiC;YAC3D,SAAS;YACT,QAAQ,EAAE,UAAU;YACpB,oBAAoB;SACrB,CAAC;QAEF,MAAM,OAAO,GAAG,iBAAiB,CAAC,CAAC,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC,IAAI,uBAAuB,CAAC,sBAAsB,CAAC,CAAC;QACpI,QAAQ,CAAC,OAAO,CAAC,CAAC;QAClB,OAAO,GAAG,EAAE;YACV,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,SAAS,EAAE,UAAU,EAAE,oBAAoB,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAErE,OAAO,iBAAiB,IAAI,OAAO,iBAAiB,KAAK,UAAU,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,KAAK,CAAC;AAClG,CAAC;AAED,MAAM,qBAAqB,GAAG,iCAAiC,CAAC,CAAC,kCAAkC,EAAE,uBAAuB,EAAE,0BAA0B,CAAC,CAAC,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 \"../VisibilityTreeBase.scss\";\nimport classNames from \"classnames\";\nimport { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport { SelectionMode } from \"@itwin/components-react\";\nimport { isPresentationTreeNodeItem, PresentationTree } from \"@itwin/presentation-components\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { FilterableTreeRenderer } from \"../common/TreeRenderer\";\nimport { ClassGroupingOption } from \"../common/Types\";\nimport { useFeatureReporting } from \"../common/UseFeatureReporting\";\nimport { usePerformanceReporting } from \"../common/UsePerformanceReporting\";\nimport { useVisibilityTreeState } from \"../common/UseVisibilityTreeState\";\nimport { addCustomTreeNodeItemLabelRenderer, addTreeNodeItemCheckbox, combineTreeNodeItemCustomizations } from \"../common/Utils\";\nimport { createVisibilityTreeRenderer, FilterableVisibilityTreeNodeRenderer, VisibilityTreeNoFilteredData } from \"../VisibilityTreeRenderer\";\nimport { ModelsTreeComponent } from \"./ModelsTreeComponent\";\nimport { ModelsTreeEventHandler } from \"./ModelsTreeEventHandler\";\nimport { ModelsVisibilityHandler, SubjectModelIdsCache } from \"./ModelsVisibilityHandler\";\nimport { addModelsTreeNodeItemIcons, createRuleset, createSearchRuleset } from \"./Utils\";\n\nimport type { FilterableTreeNodeRendererProps } from \"../common/TreeRenderer\";\nimport type { UsageTrackedFeatures } from \"../common/UseFeatureReporting\";\nimport type { VisibilityTreeEventHandlerParams } from \"../VisibilityTreeEventHandler\";\nimport type { Ruleset, SingleSchemaClassSpecification } from \"@itwin/presentation-common\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\nimport type { TreeNodeItem } from \"@itwin/components-react\";\nimport type { IFilteredPresentationTreeDataProvider } from \"@itwin/presentation-components\";\nimport type { BaseFilterableTreeProps, HierarchyLevelConfig } from \"../common/Types\";\nimport type { ModelsTreeSelectionPredicate, ModelsVisibilityHandlerProps } from \"./ModelsVisibilityHandler\";\nconst PAGING_SIZE = 20;\n\n/**\n * Props for configuring the hierarchy in [[ModelsTree]].\n * @public\n */\nexport interface ModelsTreeHierarchyConfiguration {\n  /** Should the tree group displayed element nodes by class. Defaults to `ClassGroupingOption.No`. */\n  enableElementsClassGrouping?: ClassGroupingOption;\n  /**\n   * Defines the `bis.GeometricElement3d` sub-class that should be used to load element nodes.\n   * Defaults to `bis.GeometricElement3d`. It's expected for the given class to derive from it.\n   */\n  elementClassSpecification?: SingleSchemaClassSpecification;\n  /** Should the tree show models without elements. */\n  showEmptyModels?: boolean;\n}\n\n/**\n * Props for [[ModelsTree]] component.\n * @public\n */\nexport interface ModelsTreeProps extends BaseFilterableTreeProps {\n  /**\n   * Predicate which indicates whether node can be selected or no\n   */\n  selectionPredicate?: ModelsTreeSelectionPredicate;\n  /**\n   * Active view used to determine and control visibility\n   */\n  activeView: Viewport;\n  /**\n   * Configuration options for the hierarchy loaded in the component.\n   */\n  hierarchyConfig?: ModelsTreeHierarchyConfiguration;\n  /**\n   * Auto-update the hierarchy when data in the iModel changes.\n   * @alpha\n   * @deprecated in 2.0.1. It does not have any effect, auto update is always on.\n   */\n  enableHierarchyAutoUpdate?: boolean;\n  /**\n   * Custom visibility handler.\n   */\n  modelsVisibilityHandler?: ModelsVisibilityHandler | ((props: ModelsVisibilityHandlerProps) => ModelsVisibilityHandler);\n  /**\n   * Props for configuring hierarchy level.\n   * @beta\n   */\n  hierarchyLevelConfig?: HierarchyLevelConfig;\n  /**\n   * Reports performance of a feature.\n   * @param featureId ID of the feature.\n   * @param elapsedTime Elapsed time of the feature.\n   * @beta\n   */\n  onPerformanceMeasured?: (featureId: string, elapsedTime: number) => void;\n  /**\n   * Callback that is invoked when a tracked feature is used.\n   * @param featureId ID of the feature.\n   * @beta\n   */\n  onFeatureUsed?: (feature: string) => void;\n}\n\n/**\n * A tree component that shows a subject - model - category - element\n * hierarchy along with checkboxes that represent and allow changing\n * the display of those instances.\n * @public\n */\nexport function ModelsTree(props: ModelsTreeProps) {\n  const { hierarchyLevelConfig, density, height, width, selectionMode, onFeatureUsed } = props;\n  const { reportUsage } = useFeatureReporting({ treeIdentifier: ModelsTreeComponent.id, onFeatureUsed });\n  const state = useModelsTreeState({ ...props, reportUsage });\n\n  const baseRendererProps = {\n    reportUsage,\n    contextMenuItems: props.contextMenuItems,\n    nodeLabelRenderer: props.nodeLabelRenderer,\n    density: props.density,\n    nodeRendererProps: {\n      iconsEnabled: true,\n      descriptionEnabled: false,\n      levelOffset: 10,\n      disableRootNodeCollapse: true,\n      onVisibilityToggled: () => reportUsage({ featureId: \"visibility-change\", reportInteraction: true }),\n    },\n  };\n\n  // istanbul ignore next\n  const noFilteredDataRenderer = useCallback(() => {\n    return (\n      <VisibilityTreeNoFilteredData title={TreeWidget.translate(\"modelTree.noModelFound\")} message={TreeWidget.translate(\"modelTree.noMatchingModelNames\")} />\n    );\n  }, []);\n\n  if (!state) {\n    return null;\n  }\n\n  const isFilterApplied = state.filteringResult?.filteredProvider !== undefined;\n  const overlay = state.filteringResult?.isFiltering ? <div className=\"filteredTreeOverlay\" /> : undefined;\n  return (\n    <div className={classNames(\"tree-widget-visibility-tree-base\", \"tree-widget-tree-container\")}>\n      <PresentationTree\n        state={state}\n        selectionMode={selectionMode || SelectionMode.None}\n        treeRenderer={\n          hierarchyLevelConfig?.isFilteringEnabled\n            ? (rendererProps) => (\n                <FilterableTreeRenderer\n                  {...rendererProps}\n                  {...baseRendererProps}\n                  nodeLoader={state.nodeLoader}\n                  nodeRenderer={(nodeProps) => <ModelsTreeNodeRenderer {...nodeProps} density={density} />}\n                />\n              )\n            : createVisibilityTreeRenderer(baseRendererProps)\n        }\n        noDataRenderer={isFilterApplied ? noFilteredDataRenderer : undefined}\n        width={width}\n        height={height}\n      />\n      {overlay}\n    </div>\n  );\n}\n\ninterface ModelsTreeNodeRendererProps extends FilterableTreeNodeRendererProps {\n  density?: \"default\" | \"enlarged\";\n}\n\nfunction ModelsTreeNodeRenderer(props: ModelsTreeNodeRendererProps) {\n  return (\n    <FilterableVisibilityTreeNodeRenderer\n      {...props}\n      iconsEnabled={true}\n      descriptionEnabled={false}\n      levelOffset={10}\n      disableRootNodeCollapse={true}\n      isEnlarged={props.density === \"enlarged\"}\n      onVisibilityToggled={() => props.reportUsage?.({ featureId: \"visibility-change\", reportInteraction: true })}\n    />\n  );\n}\n\ninterface UseModelsTreeStateProps extends Omit<ModelsTreeProps, \"onFeatureUsed\"> {\n  reportUsage: (props: { featureId?: UsageTrackedFeatures; reportInteraction: boolean }) => void;\n}\n\nfunction useModelsTreeState({ filterInfo, onFilterApplied, ...props }: UseModelsTreeStateProps) {\n  const rulesets = {\n    general: useMemo(\n      () =>\n        createRuleset({\n          enableElementsClassGrouping: !!props.hierarchyConfig?.enableElementsClassGrouping,\n          elementClassSpecification: props.hierarchyConfig?.elementClassSpecification,\n          showEmptyModels: props.hierarchyConfig?.showEmptyModels,\n        }),\n      [props.hierarchyConfig?.enableElementsClassGrouping, props.hierarchyConfig?.elementClassSpecification, props.hierarchyConfig?.showEmptyModels],\n    ),\n    search: useMemo(\n      () =>\n        createSearchRuleset({\n          elementClassSpecification: props.hierarchyConfig?.elementClassSpecification,\n          showEmptyModels: props.hierarchyConfig?.showEmptyModels,\n        }),\n      [props.hierarchyConfig?.elementClassSpecification, props.hierarchyConfig?.showEmptyModels],\n    ),\n  };\n\n  const treeState = useTreeState({\n    ...props,\n    ruleset: rulesets.general,\n  });\n\n  const filteredTreeState = useTreeState({\n    ...props,\n    ruleset: rulesets.search,\n    filterInfo,\n    onFilterApplied,\n  });\n\n  return filterInfo?.filter ? filteredTreeState : treeState;\n}\n\ninterface UseTreeProps extends Omit<ModelsTreeProps, \"onFeatureUsed\"> {\n  ruleset: Ruleset;\n  reportUsage: (props: { featureId?: UsageTrackedFeatures; reportInteraction: boolean }) => void;\n}\n\nfunction useTreeState({\n  modelsVisibilityHandler,\n  activeView,\n  selectionPredicate,\n  hierarchyConfig,\n  iModel,\n  ruleset,\n  filterInfo,\n  onFilterApplied,\n  hierarchyLevelConfig,\n  onPerformanceMeasured,\n  reportUsage,\n}: UseTreeProps) {\n  const visibilityHandler = useVisibilityHandler(ruleset.id, iModel, activeView, modelsVisibilityHandler);\n  const selectionPredicateRef = useRef(selectionPredicate);\n  useEffect(() => {\n    selectionPredicateRef.current = selectionPredicate;\n  }, [selectionPredicate]);\n\n  const onFilterChange = useCallback(\n    (dataProvider?: IFilteredPresentationTreeDataProvider, matchesCount?: number) => {\n      if (dataProvider && matchesCount !== undefined) {\n        reportUsage({ featureId: \"filtering\", reportInteraction: false });\n        onFilterApplied?.(dataProvider, matchesCount);\n      }\n\n      if (visibilityHandler) {\n        visibilityHandler.setFilteredDataProvider(dataProvider);\n      }\n    },\n    [onFilterApplied, reportUsage, visibilityHandler],\n  );\n\n  const { onNodeLoaded } = usePerformanceReporting({\n    treeIdentifier: ModelsTreeComponent.id,\n    onPerformanceMeasured,\n  });\n\n  const eventHandlerFactory = useCallback(\n    (handlerProps: VisibilityTreeEventHandlerParams) => {\n      return new ModelsTreeEventHandler({ ...handlerProps, reportUsage });\n    },\n    [reportUsage],\n  );\n\n  return useVisibilityTreeState({\n    imodel: iModel,\n    ruleset,\n    pagingSize: PAGING_SIZE,\n    appendChildrenCountForGroupingNodes: hierarchyConfig?.enableElementsClassGrouping === ClassGroupingOption.YesWithCounts,\n    enableHierarchyAutoUpdate: true,\n    customizeTreeNodeItem,\n    visibilityHandler,\n    filterInfo,\n    onFilterChange,\n    selectionPredicate: useCallback(\n      (node: TreeNodeItem) =>\n        !selectionPredicateRef.current || !isPresentationTreeNodeItem(node)\n          ? true\n          : selectionPredicateRef.current(node.key, ModelsVisibilityHandler.getNodeType(node)),\n      [],\n    ),\n    eventHandler: eventHandlerFactory,\n    hierarchyLevelSizeLimit: hierarchyLevelConfig?.sizeLimit,\n    onNodeLoaded: filterInfo ? undefined : onNodeLoaded,\n    reportUsage: filterInfo ? undefined : reportUsage,\n    onHierarchyLimitExceeded: () => reportUsage({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n  });\n}\n\nfunction useVisibilityHandler(\n  rulesetId: string,\n  iModel: IModelConnection,\n  activeView: Viewport,\n  visibilityHandler?: ModelsVisibilityHandler | ((props: ModelsVisibilityHandlerProps) => ModelsVisibilityHandler),\n) {\n  const subjectModelIdsCache = useMemo(() => new SubjectModelIdsCache(iModel), [iModel]);\n  const [state, setState] = useState<ModelsVisibilityHandler>();\n\n  useEffect(() => {\n    if (visibilityHandler && typeof visibilityHandler !== \"function\") {\n      return;\n    }\n\n    const visibilityHandlerProps: ModelsVisibilityHandlerProps = {\n      rulesetId,\n      viewport: activeView,\n      subjectModelIdsCache,\n    };\n\n    const handler = visibilityHandler ? visibilityHandler(visibilityHandlerProps) : new ModelsVisibilityHandler(visibilityHandlerProps);\n    setState(handler);\n    return () => {\n      handler.dispose();\n    };\n  }, [rulesetId, activeView, subjectModelIdsCache, visibilityHandler]);\n\n  return visibilityHandler && typeof visibilityHandler !== \"function\" ? visibilityHandler : state;\n}\n\nconst customizeTreeNodeItem = combineTreeNodeItemCustomizations([addCustomTreeNodeItemLabelRenderer, addTreeNodeItemCheckbox, addModelsTreeNodeItemIcons]);\n"]}
+{"version":3,"file":"ModelsTree.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,mCAAmC,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,iCAAiC,EAAE,MAAM,wCAAwC,CAAC;AAC3F,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAgC7F,gBAAgB;AAChB,MAAM,UAAU,UAAU,CAAC,EACzB,MAAM,EACN,gBAAgB,EAChB,MAAM,EACN,KAAK,EACL,UAAU,EACV,MAAM,EACN,OAAO,EACP,oBAAoB,EACpB,eAAe,EACf,aAAa,EACb,qBAAqB,EACrB,aAAa,GACG;IAChB,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,CAAuC,SAAS,CAAC,CAAC;IACtG,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC;QACL,GAAG,6BAA6B;QAChC,GAAG,eAAe;KACnB,CAAC;IACF,uDAAuD;IACvD,MAAM,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CACrC,CAAC;IAEF,MAAM,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,GAAG,mBAAmB,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;IACpH,MAAM,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,GAAG,0BAA0B,EAAE,CAAC;IACpF,MAAM,EAAE,WAAW,EAAE,GAAG,mBAAmB,CAAC,EAAE,aAAa,EAAE,cAAc,EAAE,mBAAmB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEvG,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,IAAI,oBAAoB,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,qBAAqB,EAAE,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,EAC5I,CAAC,qBAAqB,EAAE,sBAAsB,CAAC,CAChD,CAAC;IAEF,MAAM,iBAAiB,GAAG,WAAW,CACnC,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAA6B,EAAE,EAAE;QAC9D,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC,EAAE;YAC7I,OAAO;SACR;QACD,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACnF,MAAM,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC;QACtE,WAAW,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC,EACD,CAAC,WAAW,CAAC,CACd,CAAC;IAEF,MAAM,uBAAuB,GAAG,OAAO,CAAuC,GAAG,EAAE;QACjF,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC,wBAAwB,EAAE;YAC7B,OAAO,SAAS,CAAC;SAClB;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;YAChC,IAAI;gBACF,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,wBAAwB,CAAC,CAAC;gBACrE,OAAO,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;oBACvD,YAAY;oBACZ,QAAQ,EAAE,qBAAqB,EAAE;oBACjC,IAAI,EAAE,UAAU;oBAChB,eAAe,EAAE,sBAAsB;iBACxC,CAAC,CAAC;aACJ;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,2BAA2B,CAAC;gBACvJ,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;aACX;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,EAAE,qBAAqB,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAE9E,MAAM,sBAAsB,GAAG,OAAO,CAAuC,GAAG,EAAE;QAChF,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,SAAS,CAAC;SAClB;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;YAChC,WAAW,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,IAAI;gBACF,OAAO,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;oBACvD,YAAY;oBACZ,KAAK,EAAE,MAAM;oBACb,QAAQ,EAAE,qBAAqB,EAAE;oBACjC,eAAe,EAAE,sBAAsB;iBACxC,CAAC,CAAC;aACJ;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;gBAC7I,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;aACX;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,EAAE,qBAAqB,EAAE,WAAW,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAEzE,MAAM,gBAAgB,GAAG,uBAAuB,IAAI,sBAAsB,CAAC;IAE3E,OAAO,CACL,KAAC,cAAc,IACb,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,mBAAmB,CAAC,EAAE,EAChC,gBAAgB,EAAE,gBAAgB,EAClC,wBAAwB,EAAE,wBAAwB,EAClD,sBAAsB,EAAE,sBAAsB,EAC9C,gBAAgB,EAAE,gBAAgB,EAClC,uBAAuB,EAAE,oBAAoB,EAAE,SAAS,EACxD,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,OAAO,EAChB,aAAa,EAAE,gBAAgB,CAAC,MAAM,EAAE,cAAc,CAAC,EACvD,aAAa,EAAE,aAAa,EAC5B,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,qBAAqB,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,IAAI,MAAM,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC3E,CAAC,EACD,WAAW,EAAE,WAAW,EACxB,iBAAiB,EAAE,iBAAiB,EACpC,UAAU,EAAE,MAAM,GAClB,CACH,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAe,EAAE,KAAgC;IACzE,IAAI,oBAAoB,CAAC,KAAK,CAAC,EAAE;QAC/B,OAAO,KAAC,kBAAkB,IAAC,KAAK,EAAE,KAAM,GAAI,CAAC;KAC9C;IACD,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE;QACxB,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,wBAAwB,KAAK,EAAE,CAAC,GAAQ,CAAC;KAC7E;IACD,IAAI,MAAM,EAAE;QACV,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,gCAAgC,EAAE,EAAE,MAAM,EAAE,CAAC,GAAQ,CAAC;KAC1F;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,aAAa,CAAC,KAA2C;IAChE,OAAO,KAAK,KAAK,sBAAsB,IAAI,KAAK,KAAK,oBAAoB,CAAC;AAC5E,CAAC;AAED,SAAS,oBAAoB,CAAC,KAA2C;IACvE,OAAO,KAAK,KAAK,yBAAyB,IAAI,KAAK,KAAK,2BAA2B,CAAC;AACtF,CAAC;AAED,SAAS,kBAAkB,CAAC,EAAE,KAAK,EAAuC;IACxE,MAAM,EAAE,MAAM,EAAE,GAAG,0BAA0B,EAAE,CAAC;IAChD,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,UAAU,CAAC,SAAS,CAAC,wBAAwB,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;IACvH,OAAO,KAAC,IAAI,cAAE,gBAAgB,GAAQ,CAAC;AACzC,CAAC;AAED,SAAS,OAAO,CAAC,IAA+B;IAC9C,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE;QAC5C,OAAO,SAAS,CAAC;KAClB;IAED,QAAQ,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE;QACjC,KAAK,aAAa;YAChB,OAAO,KAAC,SAAS,KAAG,CAAC;QACvB,KAAK,WAAW;YACd,OAAO,KAAC,OAAO,KAAG,CAAC;QACrB,KAAK,eAAe;YAClB,OAAO,KAAC,gBAAgB,KAAG,CAAC;QAC9B,KAAK,sBAAsB;YACzB,OAAO,KAAC,eAAe,KAAG,CAAC;QAC7B,KAAK,aAAa;YAChB,OAAO,KAAC,SAAS,KAAG,CAAC;QACvB,KAAK,YAAY;YACf,OAAO,KAAC,QAAQ,KAAG,CAAC;KACvB;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,8BAA8B,CAAC,UAAoB,EAAE,cAAwC;IACpG,OAAO,CAAC,YAAuC,EAAE,EAAE,CAAC,iCAAiC,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;AAC5J,CAAC;AAED,SAAS,mBAAmB,CAAC,UAAoB,EAAE,eAAiD;IAClG,MAAM,QAAQ,GAAG,MAAM,EAAsB,CAAC;IAC9C,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAEnD,MAAM,qBAAqB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE;YACrB,QAAQ,CAAC,OAAO,GAAG,IAAI,kBAAkB,CAAC,wBAAwB,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,eAAe,CAAC,CAAC;SAChH;QACD,OAAO,QAAQ,CAAC,OAAO,CAAC;IAC1B,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAElJ,uBAAuB,CAAC;QACtB,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE;YACvB,QAAQ,CAAC,OAAO,GAAG,SAAS,CAAC;YAC7B,2BAA2B,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,CAAC;QACvG,CAAC,EAAE,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC;KACxC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,gBAAgB,CAAC,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC;QAC7C,QAAQ,CAAC,OAAO,GAAG,SAAS,CAAC;QAC7B,2BAA2B,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,CAAC;IACvG,CAAC,EAAE,CAAC,UAAU,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAExC,OAAO;QACL,qBAAqB;QACrB,wBAAwB;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO,CACL,KAAC,IAAI,cACH,cAAK,EAAE,EAAC,UAAU,eAAW,UAAU,EAAC,KAAK,EAAC,4BAA4B,EAAC,OAAO,EAAC,WAAW,YAC5F,eAAM,CAAC,EAAC,k0BAAk0B,GAAG,GACz0B,GACD,CACR,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,wBAA0F;IACzH,MAAM,UAAU,GAA2C,EAAE,CAAC;IAC9D,MAAM,iBAAiB,GAA0H,EAAE,CAAC;IACpJ,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,wBAAwB,EAAE,EAAE;QAClD,IAAI,IAAI,IAAI,GAAG,EAAE;YACf,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,SAAS;SACV;QAED,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YAC9C,UAAU,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,mBAAmB,CAAC,CAAC;YAC5C,SAAS;SACV;QAED,IAAI,CAAC,GAAG,CAAC,mBAAmB,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE;YAC1F,SAAS;SACV;QAED,MAAM,SAAS,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC;QAC9C,MAAM,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QACvF,MAAM,QAAQ,GAAG,CAAE,GAAG,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAA2B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAClH,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QACzG,IAAI,SAAS,EAAE;YACb,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;SAC3C;aAAM;YACL,iBAAiB,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;SACjG;KACF;IACD,UAAU,CAAC,IAAI,CACb,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAC1E,MAAM,EACJ,UAAU,KAAK,SAAS;YACtB,CAAC,CAAC,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAChF,CAAC,CAAC,EAAE,IAAI,EAAE,UAAmB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE;QAC/F,OAAO;KACR,CAAC,CAAC,CACJ,CAAC;IACF,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAe,EAAE,OAAoB;IACnE,MAAM,GAAG,GAAG,IAAI,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,OAAO,CAAC;KAChB;IAED,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,GAAG,KAAK,CAAC;IACpC,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAExD,OAAO,CACL,8BACG,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EAC/B,KAAC,MAAM,IACL,SAAS,QACT,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;oBACb,CAAC,CAAC,eAAe,EAAE,CAAC;oBACpB,OAAO,EAAE,EAAE,CAAC;gBACd,CAAC,YAEA,SAAS,GACH,EACR,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAC5B,CACJ,CAAC;AACJ,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 { useCallback, useEffect, useMemo, useRef, useState } from \"react\";\nimport { IModelApp } from \"@itwin/core-frontend\";\nimport { SvgFolder, SvgImodelHollow, SvgItem, SvgLayers, SvgModel } from \"@itwin/itwinui-icons-react\";\nimport { Anchor, Icon, Text } from \"@itwin/itwinui-react\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { HierarchyNode, HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { VisibilityTree } from \"../common/components/VisibilityTree\";\nimport { useFocusedInstancesContext } from \"../common/FocusedInstancesContext\";\nimport { useFeatureReporting } from \"../common/UseFeatureReporting\";\nimport { useIModelChangeListener } from \"../common/UseIModelChangeListener\";\nimport { ModelsTreeIdsCache } from \"./internal/ModelsTreeIdsCache\";\nimport { createModelsTreeVisibilityHandler } from \"./internal/ModelsTreeVisibilityHandler\";\nimport { ModelsTreeComponent } from \"./ModelsTreeComponent\";\nimport { defaultHierarchyConfiguration, ModelsTreeDefinition } from \"./ModelsTreeDefinition\";\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ElementsGroupInfo } from \"./ModelsTreeDefinition\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ComponentPropsWithoutRef, ReactElement } from \"react\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\ntype ModelsTreeFilteringError = \"tooManyFilterMatches\" | \"tooManyInstancesFocused\" | \"unknownFilterError\" | \"unknownInstanceFocusError\";\n\ninterface ModelsTreeOwnProps {\n  activeView: Viewport;\n  hierarchyLevelConfig?: {\n    sizeLimit?: number;\n  };\n  filter?: string;\n  onPerformanceMeasured?: (featureId: string, duration: number) => void;\n  onFeatureUsed?: (feature: string) => void;\n}\n\ntype VisibilityTreeProps = ComponentPropsWithoutRef<typeof VisibilityTree>;\ntype GetFilteredPathsCallback = VisibilityTreeProps[\"getFilteredPaths\"];\ntype GetHierarchyDefinitionCallback = VisibilityTreeProps[\"getHierarchyDefinition\"];\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\ntype ModelsTreeProps = ModelsTreeOwnProps &\n  Pick<VisibilityTreeProps, \"imodel\" | \"getSchemaContext\" | \"height\" | \"width\" | \"density\" | \"selectionMode\"> & {\n    hierarchyConfig?: Partial<ModelsTreeHierarchyConfiguration>;\n  };\n\n/** @internal */\nexport function ModelsTree({\n  imodel,\n  getSchemaContext,\n  height,\n  width,\n  activeView,\n  filter,\n  density,\n  hierarchyLevelConfig,\n  hierarchyConfig,\n  selectionMode,\n  onPerformanceMeasured,\n  onFeatureUsed,\n}: ModelsTreeProps) {\n  const [filteringError, setFilteringError] = useState<ModelsTreeFilteringError | undefined>(undefined);\n  const hierarchyConfiguration = useMemo<ModelsTreeHierarchyConfiguration>(\n    () => ({\n      ...defaultHierarchyConfiguration,\n      ...hierarchyConfig,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    Object.values(hierarchyConfig ?? {}),\n  );\n\n  const { getModelsTreeIdsCache, visibilityHandlerFactory } = useCachedVisibility(activeView, hierarchyConfiguration);\n  const { loadInstanceKeys: loadFocusedInstancesKeys } = useFocusedInstancesContext();\n  const { reportUsage } = useFeatureReporting({ onFeatureUsed, treeIdentifier: ModelsTreeComponent.id });\n\n  const getHierarchyDefinition = useCallback<GetHierarchyDefinitionCallback>(\n    ({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration }),\n    [getModelsTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const onNodeDoubleClick = useCallback(\n    async ({ nodeData, extendedData }: PresentationHierarchyNode) => {\n      if (!HierarchyNode.isInstancesNode(nodeData) || (extendedData && (extendedData.isSubject || extendedData.isModel || extendedData.isCategory))) {\n        return;\n      }\n      const instanceIds = nodeData.key.instanceKeys.map((instanceKey) => instanceKey.id);\n      await IModelApp.viewManager.selectedView?.zoomToElements(instanceIds);\n      reportUsage({ featureId: \"zoom-to-node\", reportInteraction: false });\n    },\n    [reportUsage],\n  );\n\n  const getFocusedFilteredPaths = useMemo<GetFilteredPathsCallback | undefined>(() => {\n    setFilteringError(undefined);\n    if (!loadFocusedInstancesKeys) {\n      return undefined;\n    }\n    return async ({ imodelAccess }) => {\n      try {\n        const targetKeys = await collectTargetKeys(loadFocusedInstancesKeys);\n        return await ModelsTreeDefinition.createInstanceKeyPaths({\n          imodelAccess,\n          idsCache: getModelsTreeIdsCache(),\n          keys: targetKeys,\n          hierarchyConfig: hierarchyConfiguration,\n        });\n      } catch (e) {\n        const newError = e instanceof Error && e.message.match(/Filter matches more than \\d+ items/) ? \"tooManyInstancesFocused\" : \"unknownInstanceFocusError\";\n        setFilteringError(newError);\n        return [];\n      }\n    };\n  }, [loadFocusedInstancesKeys, getModelsTreeIdsCache, hierarchyConfiguration]);\n\n  const getSearchFilteredPaths = useMemo<GetFilteredPathsCallback | undefined>(() => {\n    setFilteringError(undefined);\n    if (!filter) {\n      return undefined;\n    }\n    return async ({ imodelAccess }) => {\n      reportUsage?.({ featureId: \"filtering\", reportInteraction: true });\n      try {\n        return await ModelsTreeDefinition.createInstanceKeyPaths({\n          imodelAccess,\n          label: filter,\n          idsCache: getModelsTreeIdsCache(),\n          hierarchyConfig: hierarchyConfiguration,\n        });\n      } catch (e) {\n        const newError = e instanceof Error && e.message.match(/Filter matches more than \\d+ items/) ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n        setFilteringError(newError);\n        return [];\n      }\n    };\n  }, [filter, getModelsTreeIdsCache, reportUsage, hierarchyConfiguration]);\n\n  const getFilteredPaths = getFocusedFilteredPaths ?? getSearchFilteredPaths;\n\n  return (\n    <VisibilityTree\n      height={height}\n      width={width}\n      imodel={imodel}\n      treeName={ModelsTreeComponent.id}\n      getSchemaContext={getSchemaContext}\n      visibilityHandlerFactory={visibilityHandlerFactory}\n      getHierarchyDefinition={getHierarchyDefinition}\n      getFilteredPaths={getFilteredPaths}\n      hierarchyLevelSizeLimit={hierarchyLevelConfig?.sizeLimit}\n      getIcon={getIcon}\n      density={density}\n      noDataMessage={getNoDataMessage(filter, filteringError)}\n      selectionMode={selectionMode}\n      onPerformanceMeasured={(action, duration) => {\n        onPerformanceMeasured?.(`${ModelsTreeComponent.id}-${action}`, duration);\n      }}\n      reportUsage={reportUsage}\n      onNodeDoubleClick={onNodeDoubleClick}\n      searchText={filter}\n    />\n  );\n}\n\nfunction getNoDataMessage(filter?: string, error?: ModelsTreeFilteringError) {\n  if (isInstanceFocusError(error)) {\n    return <InstanceFocusError error={error!} />;\n  }\n  if (isFilterError(error)) {\n    return <Text>{TreeWidget.translate(`modelsTree.filtering.${error}`)}</Text>;\n  }\n  if (filter) {\n    return <Text>{TreeWidget.translate(\"modelsTree.filtering.noMatches\", { filter })}</Text>;\n  }\n  return undefined;\n}\n\nfunction isFilterError(error: ModelsTreeFilteringError | undefined) {\n  return error === \"tooManyFilterMatches\" || error === \"unknownFilterError\";\n}\n\nfunction isInstanceFocusError(error: ModelsTreeFilteringError | undefined) {\n  return error === \"tooManyInstancesFocused\" || error === \"unknownInstanceFocusError\";\n}\n\nfunction InstanceFocusError({ error }: { error: ModelsTreeFilteringError }) {\n  const { toggle } = useFocusedInstancesContext();\n  const localizedMessage = createLocalizedMessage(TreeWidget.translate(`modelsTree.filtering.${error}`), () => toggle());\n  return <Text>{localizedMessage}</Text>;\n}\n\nfunction getIcon(node: PresentationHierarchyNode): ReactElement | undefined {\n  if (node.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  switch (node.extendedData.imageId) {\n    case \"icon-layers\":\n      return <SvgLayers />;\n    case \"icon-item\":\n      return <SvgItem />;\n    case \"icon-ec-class\":\n      return <SvgClassGrouping />;\n    case \"icon-imodel-hollow-2\":\n      return <SvgImodelHollow />;\n    case \"icon-folder\":\n      return <SvgFolder />;\n    case \"icon-model\":\n      return <SvgModel />;\n  }\n\n  return undefined;\n}\n\nfunction createVisibilityHandlerFactory(activeView: Viewport, idsCacheGetter: () => ModelsTreeIdsCache) {\n  return (imodelAccess: ECClassHierarchyInspector) => createModelsTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess });\n}\n\nfunction useCachedVisibility(activeView: Viewport, hierarchyConfig: ModelsTreeHierarchyConfiguration) {\n  const cacheRef = useRef<ModelsTreeIdsCache>();\n  const currentIModelRef = useRef(activeView.iModel);\n\n  const getModelsTreeIdsCache = useCallback(() => {\n    if (!cacheRef.current) {\n      cacheRef.current = new ModelsTreeIdsCache(createECSqlQueryExecutor(currentIModelRef.current), hierarchyConfig);\n    }\n    return cacheRef.current;\n  }, [hierarchyConfig]);\n\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));\n\n  useIModelChangeListener({\n    imodel: activeView.iModel,\n    action: useCallback(() => {\n      cacheRef.current = undefined;\n      setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));\n    }, [activeView, getModelsTreeIdsCache]),\n  });\n\n  useEffect(() => {\n    currentIModelRef.current = activeView.iModel;\n    cacheRef.current = undefined;\n    setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache));\n  }, [activeView, getModelsTreeIdsCache]);\n\n  return {\n    getModelsTreeIdsCache,\n    visibilityHandlerFactory,\n  };\n}\n\nfunction SvgClassGrouping() {\n  return (\n    <Icon>\n      <svg id=\"Calque_1\" data-name=\"Calque 1\" xmlns=\"http://www.w3.org/2000/svg\" viewBox=\"0 0 16 16\">\n        <path d=\"M8.00933,0,0,3.97672V11.986L8.00933,16,16,11.93V3.97651ZM1.66173,11.27642c-.26155.03734-.59754-.26154-.76553-.69085-.168-.41066-.09334-.784.168-.82152.26154-.03734.59754.26154.76553.67219C1.99772,10.86577,1.92306,11.23909,1.66173,11.27642Zm0-3.32319c-.26155.03733-.59754-.28-.76553-.69086-.168-.42932-.09334-.80285.168-.84.26133-.03733.59754.28.76532.69086C1.99772,7.54236,1.92306,7.89723,1.66173,7.95323Zm4.31276,5.52621a.18186.18186,0,0,1-.16821-.01866L3.41657,12.15394a.94275.94275,0,0,1-.29887-.80285c.03754-.33621.22421-.52265.41108-.41066L5.9185,12.24727a.88656.88656,0,0,1,.28.80285A.5057.5057,0,0,1,5.97449,13.47944Zm0-3.37919a.18184.18184,0,0,1-.16821-.01867L3.41657,8.77475a.943.943,0,0,1-.29887-.80286c.03754-.3362.22421-.52286.41108-.42953L5.9185,8.86786a.83112.83112,0,0,1,.28.78419A.51684.51684,0,0,1,5.97449,10.10025Z\" />\n      </svg>\n    </Icon>\n  );\n}\n\nasync function collectTargetKeys(loadFocusedInstancesKeys: () => AsyncIterableIterator<InstanceKey | GroupingHierarchyNode>) {\n  const targetKeys: Array<InstanceKey | ElementsGroupInfo> = [];\n  const groupingNodeInfos: Array<{ parentKey: InstancesNodeKey; parentType: \"element\" | \"category\"; classes: string[]; modelIds: Id64String[] }> = [];\n  for await (const key of loadFocusedInstancesKeys()) {\n    if (\"id\" in key) {\n      targetKeys.push(key);\n      continue;\n    }\n\n    if (!HierarchyNodeKey.isClassGrouping(key.key)) {\n      targetKeys.push(...key.groupedInstanceKeys);\n      continue;\n    }\n\n    if (!key.nonGroupingAncestor || !HierarchyNodeKey.isInstances(key.nonGroupingAncestor.key)) {\n      continue;\n    }\n\n    const parentKey = key.nonGroupingAncestor.key;\n    const type = key.nonGroupingAncestor.extendedData?.isCategory ? \"category\" : \"element\";\n    const modelIds = ((key.nonGroupingAncestor.extendedData?.modelIds as Id64String[][]) ?? []).flatMap((ids) => ids);\n    const groupInfo = groupingNodeInfos.find((group) => HierarchyNodeKey.equals(group.parentKey, parentKey));\n    if (groupInfo) {\n      groupInfo.classes.push(key.key.className);\n    } else {\n      groupingNodeInfos.push({ classes: [key.key.className], parentType: type, parentKey, modelIds });\n    }\n  }\n  targetKeys.push(\n    ...groupingNodeInfos.map(({ parentKey, parentType, classes, modelIds }) => ({\n      parent:\n        parentType === \"element\"\n          ? { type: \"element\" as const, ids: parentKey.instanceKeys.map((key) => key.id) }\n          : { type: \"category\" as const, ids: parentKey.instanceKeys.map((key) => key.id), modelIds },\n      classes,\n    })),\n  );\n  return targetKeys;\n}\n\nfunction createLocalizedMessage(message: string, onClick?: () => void) {\n  const exp = new RegExp(\"<link>(.*)</link>\");\n  const match = message.match(exp);\n  if (!match) {\n    return message;\n  }\n\n  const [fullText, innerText] = match;\n  const [textBefore, textAfter] = message.split(fullText);\n\n  return (\n    <>\n      {textBefore ? textBefore : null}\n      <Anchor\n        underline\n        onClick={(e) => {\n          e.stopPropagation();\n          onClick?.();\n        }}\n      >\n        {innerText}\n      </Anchor>\n      {textAfter ? textAfter : null}\n    </>\n  );\n}\n"]}

package/lib/esm/components/trees/models-tree/ModelsTreeButtons.d.ts

@@ -10,7 +10,7 @@ export interface ModelInfo {
     isPlanProjection?: boolean;
 }
 /**
- * Props that get passed to [[ModelsTreeComponent]] header button renderer.
+ * Props that get passed to `ModelsTreeComponent` header button renderer.
  * @see ModelTreeComponentProps.headerButtons
  * @public
  */

package/lib/esm/components/trees/models-tree/ModelsTreeButtons.js

@@ -7,8 +7,7 @@ import { useEffect, useMemo, useState } from "react";
 import { SvgVisibilityHalf, SvgVisibilityHide, SvgVisibilityShow } from "@itwin/itwinui-icons-react";
 import { Button, IconButton } from "@itwin/itwinui-react";
 import { TreeWidget } from "../../../TreeWidget";
-import { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from "./ModelsVisibilityHandler";
-import { queryModelsForHeaderActions } from "./Utils";
+import { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from "./internal/ModelsTreeVisibilityHandler";
 /** @internal */
 export function useAvailableModels(imodel) {
     const [availableModels, setAvailableModels] = useState([]);
@@ -17,30 +16,46 @@ export function useAvailableModels(imodel) {
             .then((modelInfos) => {
             setAvailableModels(modelInfos);
         })
-            .catch((_e) => {
+            .catch(() => {
             setAvailableModels([]);
         });
     }, [imodel]);
     return availableModels;
 }
+async function queryModelsForHeaderActions(iModel) {
+    const queryParams = {
+        from: "BisCore.GeometricModel3d",
+        where: `
+        EXISTS (
+          SELECT 1
+          FROM BisCore.Element e
+          WHERE e.ECClassId IS (BisCore.GeometricElement3d, BisCore.InformationPartitionElement)
+            AND e.ECInstanceId = GeometricModel3d.ModeledElement.Id
+        )
+      `,
+        wantPrivate: false,
+    };
+    const modelProps = await iModel.models.queryProps(queryParams);
+    return modelProps.map(({ id, isPlanProjection }) => ({ id, isPlanProjection })).filter(({ id }) => id);
+}
 /** @internal */
 export function ShowAllButton(props) {
-    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("showAll"), onClick: () => {
-            props.onFeatureUsed?.("showall");
+    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("modelsTree.buttons.showAll.tooltip"), onClick: () => {
+            props.onFeatureUsed?.("models-tree-showall");
             void showAllModels(props.models.map((model) => model.id), props.viewport);
         }, children: _jsx(SvgVisibilityShow, {}) }));
 }
 /** @internal */
 export function HideAllButton(props) {
-    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("hideAll"), onClick: () => {
-            props.onFeatureUsed?.("hideall");
+    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("modelsTree.buttons.hideAll.tooltip"), onClick: () => {
+            props.onFeatureUsed?.("models-tree-hideall");
             void hideAllModels(props.models.map((model) => model.id), props.viewport);
         }, children: _jsx(SvgVisibilityHide, {}) }));
 }
 /** @internal */
 export function InvertButton(props) {
-    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("invert"), onClick: () => {
-            props.onFeatureUsed?.("invert");
+    return (_jsx(IconButton, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("modelsTree.buttons.invert.tooltip"), onClick: () => {
+            props.onFeatureUsed?.("models-tree-invert");
             void invertAllModels(props.models.map((model) => model.id), props.viewport);
         }, children: _jsx(SvgVisibilityHalf, {}) }));
 }
@@ -54,10 +69,10 @@ export function View2DButton(props) {
         setIs2dToggleActive(areAllModelsVisible(models2d, props.viewport));
         return props.viewport.onViewedModelsChanged.addListener((vp) => setIs2dToggleActive(areAllModelsVisible(models2d, vp)));
     }, [models2d, props.viewport]);
-    return (_jsx(Button, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("toggle2DViews"), onClick: () => {
-            props.onFeatureUsed?.("view2d");
+    return (_jsx(Button, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("modelsTree.buttons.toggle2d.tooltip"), onClick: () => {
+            props.onFeatureUsed?.("models-tree-view2d");
             void toggleModels(models2d, is2dToggleActive, props.viewport);
-        }, disabled: models2d.length === 0, endIcon: is2dToggleActive ? _jsx(SvgVisibilityShow, {}) : _jsx(SvgVisibilityHide, {}), children: TreeWidget.translate("label2D") }));
+        }, disabled: models2d.length === 0, endIcon: is2dToggleActive ? _jsx(SvgVisibilityShow, {}) : _jsx(SvgVisibilityHide, {}), children: TreeWidget.translate("modelsTree.buttons.toggle2d.label") }));
 }
 /** @internal */
 export function View3DButton(props) {
@@ -69,9 +84,9 @@ export function View3DButton(props) {
         setIs3dToggleActive(areAllModelsVisible(models3d, props.viewport));
         return props.viewport.onViewedModelsChanged.addListener((vp) => setIs3dToggleActive(areAllModelsVisible(models3d, vp)));
     }, [models3d, props.viewport]);
-    return (_jsx(Button, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("toggle3DViews"), onClick: () => {
-            props.onFeatureUsed?.("view3d");
+    return (_jsx(Button, { size: props.density === "enlarged" ? "large" : "small", styleType: "borderless", title: TreeWidget.translate("modelsTree.buttons.toggle3d.tooltip"), onClick: () => {
+            props.onFeatureUsed?.("models-tree-view3d");
             void toggleModels(models3d, is3dToggleActive, props.viewport);
-        }, disabled: models3d.length === 0, endIcon: is3dToggleActive ? _jsx(SvgVisibilityShow, {}) : _jsx(SvgVisibilityHide, {}), children: TreeWidget.translate("label3D") }));
+        }, disabled: models3d.length === 0, endIcon: is3dToggleActive ? _jsx(SvgVisibilityShow, {}) : _jsx(SvgVisibilityHide, {}), children: TreeWidget.translate("modelsTree.buttons.toggle3d.label") }));
 }
 //# sourceMappingURL=ModelsTreeButtons.js.map

package/lib/esm/components/trees/models-tree/ModelsTreeButtons.js.map

@@ -1 +1 @@
-{"version":3,"file":"ModelsTreeButtons.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTreeButtons.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,eAAe,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC7H,OAAO,EAAE,2BAA2B,EAAE,MAAM,SAAS,CAAC;AAwBtD,gBAAgB;AAChB,MAAM,UAAU,kBAAkB,CAAC,MAAwB;IACzD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAc,EAAE,CAAC,CAAC;IAExE,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,MAAM,CAAC;aAChC,IAAI,CAAC,CAAC,UAAuB,EAAE,EAAE;YAChC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE;YACZ,kBAAkB,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAAC,KAAkC;IAC9D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;YACjC,KAAK,aAAa,CAChB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAAC,KAAkC;IAC9D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,EACtC,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,CAAC;YACjC,KAAK,aAAa,CAChB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,EACrC,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,eAAe,CAClB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,KAAC,MAAM,IACL,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,YAAY,CAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,EACD,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,YAExE,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1F,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,KAAC,MAAM,IACL,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,eAAe,CAAC,EAC5C,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,CAAC;YAChC,KAAK,YAAY,CAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,EACD,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,YAExE,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACzB,CACV,CAAC;AACJ,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 { useEffect, useMemo, useState } from \"react\";\nimport { SvgVisibilityHalf, SvgVisibilityHide, SvgVisibilityShow } from \"@itwin/itwinui-icons-react\";\nimport { Button, IconButton } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from \"./ModelsVisibilityHandler\";\nimport { queryModelsForHeaderActions } from \"./Utils\";\n\nimport type { TreeHeaderButtonProps } from \"../../tree-header/TreeHeader\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\n\n/**\n * Information about a single Model.\n * @public\n */\nexport interface ModelInfo {\n  id: string;\n  isPlanProjection?: boolean;\n}\n\n/**\n * Props that get passed to [[ModelsTreeComponent]] header button renderer.\n * @see ModelTreeComponentProps.headerButtons\n * @public\n */\nexport interface ModelsTreeHeaderButtonProps extends TreeHeaderButtonProps {\n  /** A list of models available in the iModel. */\n  models: ModelInfo[];\n}\n\n/** @internal */\nexport function useAvailableModels(imodel: IModelConnection) {\n  const [availableModels, setAvailableModels] = useState<ModelInfo[]>([]);\n\n  useEffect(() => {\n    queryModelsForHeaderActions(imodel)\n      .then((modelInfos: ModelInfo[]) => {\n        setAvailableModels(modelInfos);\n      })\n      .catch((_e) => {\n        setAvailableModels([]);\n      });\n  }, [imodel]);\n\n  return availableModels;\n}\n\n/** @internal */\nexport function ShowAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"showAll\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"showall\");\n        void showAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityShow />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function HideAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"hideAll\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"hideall\");\n        void hideAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityHide />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function InvertButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"invert\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"invert\");\n        void invertAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityHalf />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function View2DButton(props: ModelsTreeHeaderButtonProps) {\n  const models2d = useMemo(() => {\n    return props.models.filter((model) => model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is2dToggleActive, setIs2dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs2dToggleActive(areAllModelsVisible(models2d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs2dToggleActive(areAllModelsVisible(models2d, vp)));\n  }, [models2d, props.viewport]);\n\n  return (\n    <Button\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle2DViews\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"view2d\");\n        void toggleModels(models2d, is2dToggleActive, props.viewport);\n      }}\n      disabled={models2d.length === 0}\n      endIcon={is2dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label2D\")}\n    </Button>\n  );\n}\n\n/** @internal */\nexport function View3DButton(props: ModelsTreeHeaderButtonProps) {\n  const models3d = useMemo(() => {\n    return props.models.filter((model) => !model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is3dToggleActive, setIs3dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs3dToggleActive(areAllModelsVisible(models3d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs3dToggleActive(areAllModelsVisible(models3d, vp)));\n  }, [models3d, props.viewport]);\n\n  return (\n    <Button\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"toggle3DViews\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"view3d\");\n        void toggleModels(models3d, is3dToggleActive, props.viewport);\n      }}\n      disabled={models3d.length === 0}\n      endIcon={is3dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"label3D\")}\n    </Button>\n  );\n}\n"]}
+{"version":3,"file":"ModelsTreeButtons.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTreeButtons.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,eAAe,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AAyB1I,gBAAgB;AAChB,MAAM,UAAU,kBAAkB,CAAC,MAAwB;IACzD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAc,EAAE,CAAC,CAAC;IAExE,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,MAAM,CAAC;aAChC,IAAI,CAAC,CAAC,UAAuB,EAAE,EAAE;YAChC,kBAAkB,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE;YACV,kBAAkB,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACP,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,KAAK,UAAU,2BAA2B,CAAC,MAAwB;IACjE,MAAM,WAAW,GAAqB;QACpC,IAAI,EAAE,0BAA0B;QAChC,KAAK,EAAE;;;;;;;OAOJ;QACH,WAAW,EAAE,KAAK;KACnB,CAAC;IAEF,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC/D,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAyB,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAgB,CAAC;AAC/I,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAAC,KAAkC;IAC9D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,oCAAoC,CAAC,EACjE,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,qBAAqB,CAAC,CAAC;YAC7C,KAAK,aAAa,CAChB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,aAAa,CAAC,KAAkC;IAC9D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,oCAAoC,CAAC,EACjE,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,qBAAqB,CAAC,CAAC;YAC7C,KAAK,aAAa,CAChB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,OAAO,CACL,KAAC,UAAU,IACT,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,mCAAmC,CAAC,EAChE,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,oBAAoB,CAAC,CAAC;YAC5C,KAAK,eAAe,CAClB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,EACrC,KAAK,CAAC,QAAQ,CACf,CAAC;QACJ,CAAC,YAED,KAAC,iBAAiB,KAAG,GACV,CACd,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzF,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,KAAC,MAAM,IACL,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,qCAAqC,CAAC,EAClE,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,oBAAoB,CAAC,CAAC;YAC5C,KAAK,YAAY,CAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,EACD,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,YAExE,UAAU,CAAC,SAAS,CAAC,mCAAmC,CAAC,GACnD,CACV,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,YAAY,CAAC,KAAkC;IAC7D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,OAAO,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1F,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEnB,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,SAAS,CAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC,QAAQ,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC,EAAY,EAAE,EAAE,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IACpI,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAE/B,OAAO,CACL,KAAC,MAAM,IACL,IAAI,EAAE,KAAK,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EACtD,SAAS,EAAC,YAAY,EACtB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,qCAAqC,CAAC,EAClE,OAAO,EAAE,GAAG,EAAE;YACZ,KAAK,CAAC,aAAa,EAAE,CAAC,oBAAoB,CAAC,CAAC;YAC5C,KAAK,YAAY,CAAC,QAAQ,EAAE,gBAAgB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChE,CAAC,EACD,QAAQ,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC,EAC/B,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,CAAC,CAAC,CAAC,KAAC,iBAAiB,KAAG,YAExE,UAAU,CAAC,SAAS,CAAC,mCAAmC,CAAC,GACnD,CACV,CAAC;AACJ,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 { useEffect, useMemo, useState } from \"react\";\nimport { SvgVisibilityHalf, SvgVisibilityHide, SvgVisibilityShow } from \"@itwin/itwinui-icons-react\";\nimport { Button, IconButton } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { areAllModelsVisible, hideAllModels, invertAllModels, showAllModels, toggleModels } from \"./internal/ModelsTreeVisibilityHandler\";\n\nimport type { GeometricModel3dProps, ModelQueryParams } from \"@itwin/core-common\";\nimport type { TreeHeaderButtonProps } from \"../../tree-header/TreeHeader\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\n\n/**\n * Information about a single Model.\n * @public\n */\nexport interface ModelInfo {\n  id: string;\n  isPlanProjection?: boolean;\n}\n\n/**\n * Props that get passed to `ModelsTreeComponent` header button renderer.\n * @see ModelTreeComponentProps.headerButtons\n * @public\n */\nexport interface ModelsTreeHeaderButtonProps extends TreeHeaderButtonProps {\n  /** A list of models available in the iModel. */\n  models: ModelInfo[];\n}\n\n/** @internal */\nexport function useAvailableModels(imodel: IModelConnection) {\n  const [availableModels, setAvailableModels] = useState<ModelInfo[]>([]);\n\n  useEffect(() => {\n    queryModelsForHeaderActions(imodel)\n      .then((modelInfos: ModelInfo[]) => {\n        setAvailableModels(modelInfos);\n      })\n      .catch(() => {\n        setAvailableModels([]);\n      });\n  }, [imodel]);\n\n  return availableModels;\n}\n\nasync function queryModelsForHeaderActions(iModel: IModelConnection) {\n  const queryParams: ModelQueryParams = {\n    from: \"BisCore.GeometricModel3d\",\n    where: `\n        EXISTS (\n          SELECT 1\n          FROM BisCore.Element e\n          WHERE e.ECClassId IS (BisCore.GeometricElement3d, BisCore.InformationPartitionElement)\n            AND e.ECInstanceId = GeometricModel3d.ModeledElement.Id\n        )\n      `,\n    wantPrivate: false,\n  };\n\n  const modelProps = await iModel.models.queryProps(queryParams);\n  return modelProps.map(({ id, isPlanProjection }: GeometricModel3dProps) => ({ id, isPlanProjection })).filter(({ id }) => id) as ModelInfo[];\n}\n\n/** @internal */\nexport function ShowAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"modelsTree.buttons.showAll.tooltip\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"models-tree-showall\");\n        void showAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityShow />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function HideAllButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"modelsTree.buttons.hideAll.tooltip\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"models-tree-hideall\");\n        void hideAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityHide />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function InvertButton(props: ModelsTreeHeaderButtonProps) {\n  return (\n    <IconButton\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"modelsTree.buttons.invert.tooltip\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"models-tree-invert\");\n        void invertAllModels(\n          props.models.map((model) => model.id),\n          props.viewport,\n        );\n      }}\n    >\n      <SvgVisibilityHalf />\n    </IconButton>\n  );\n}\n\n/** @internal */\nexport function View2DButton(props: ModelsTreeHeaderButtonProps) {\n  const models2d = useMemo(() => {\n    return props.models.filter((model) => model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is2dToggleActive, setIs2dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs2dToggleActive(areAllModelsVisible(models2d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs2dToggleActive(areAllModelsVisible(models2d, vp)));\n  }, [models2d, props.viewport]);\n\n  return (\n    <Button\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"modelsTree.buttons.toggle2d.tooltip\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"models-tree-view2d\");\n        void toggleModels(models2d, is2dToggleActive, props.viewport);\n      }}\n      disabled={models2d.length === 0}\n      endIcon={is2dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"modelsTree.buttons.toggle2d.label\")}\n    </Button>\n  );\n}\n\n/** @internal */\nexport function View3DButton(props: ModelsTreeHeaderButtonProps) {\n  const models3d = useMemo(() => {\n    return props.models.filter((model) => !model.isPlanProjection).map((model) => model.id);\n  }, [props.models]);\n\n  const [is3dToggleActive, setIs3dToggleActive] = useState(false);\n\n  useEffect(() => {\n    setIs3dToggleActive(areAllModelsVisible(models3d, props.viewport));\n    return props.viewport.onViewedModelsChanged.addListener((vp: Viewport) => setIs3dToggleActive(areAllModelsVisible(models3d, vp)));\n  }, [models3d, props.viewport]);\n\n  return (\n    <Button\n      size={props.density === \"enlarged\" ? \"large\" : \"small\"}\n      styleType=\"borderless\"\n      title={TreeWidget.translate(\"modelsTree.buttons.toggle3d.tooltip\")}\n      onClick={() => {\n        props.onFeatureUsed?.(\"models-tree-view3d\");\n        void toggleModels(models3d, is3dToggleActive, props.viewport);\n      }}\n      disabled={models3d.length === 0}\n      endIcon={is3dToggleActive ? <SvgVisibilityShow /> : <SvgVisibilityHide />}\n    >\n      {TreeWidget.translate(\"modelsTree.buttons.toggle3d.label\")}\n    </Button>\n  );\n}\n"]}