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

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

@@ -1,129 +1,62 @@
-import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
+import { jsx as _jsx } 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 { useCallback, useEffect, useMemo, useRef, useState } from "react";
-import { Anchor, Text } from "@itwin/itwinui-react/bricks";
+import { useCallback, useMemo } from "react";
 import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
-import { HierarchyNodeIdentifier, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
-import { TreeWidget } from "../../../TreeWidget.js";
-import { EmptyTreeContent } from "../common/components/EmptyTreeContent.js";
-import { useFocusedInstancesContext } from "../common/FocusedInstancesContext.js";
-import { FilterLimitExceededError } from "../common/TreeErrors.js";
-import { useIModelChangeListener } from "../common/UseIModelChangeListener.js";
-import { useTelemetryContext } from "../common/UseTelemetryContext.js";
+import { Icon } from "@stratakit/foundations";
+import categorySvg from "@stratakit/icons/bis-category-3d.svg";
+import classSvg from "@stratakit/icons/bis-class.svg";
+import elementSvg from "@stratakit/icons/bis-element.svg";
+import subjectSvg from "@stratakit/icons/bis-subject.svg";
+import modelSvg from "@stratakit/icons/model-cube.svg";
+import { EmptyTreeContent, FilterUnknownError, NoFilterMatches, SubTreeError, TooManyFilterMatches, TooManyInstancesFocused, UnknownInstanceFocusError, } from "../common/components/EmptyTree.js";
+import { useGuid } from "../common/internal/useGuid.js";
+import { useCachedVisibility } from "../common/internal/useTreeHooks/UseCachedVisibility.js";
+import { useIdsCache } from "../common/internal/useTreeHooks/UseIdsCache.js";
 import { ModelsTreeIdsCache } from "./internal/ModelsTreeIdsCache.js";
 import { ModelsTreeNode } from "./internal/ModelsTreeNode.js";
-import { createModelsTreeVisibilityHandler } from "./internal/ModelsTreeVisibilityHandler.js";
+import { useFilteredPaths } from "./internal/UseFilteredPaths.js";
+import { createFilteredModelsTree } from "./internal/visibility/FilteredTree.js";
+import { ModelsTreeVisibilityHandler } from "./internal/visibility/ModelsTreeVisibilityHandler.js";
 import { defaultHierarchyConfiguration, ModelsTreeDefinition } from "./ModelsTreeDefinition.js";
 /**
  * Custom hook to create and manage state for the models tree.
  * @beta
  */
-export function useModelsTree({ activeView, filter, hierarchyConfig, visibilityHandlerOverrides, getFilteredPaths, onModelsFiltered, selectionPredicate: nodeTypeSelectionPredicate, emptyTreeContent, }) {
-    const [filteringError, setFilteringError] = useState(undefined);
+export function useModelsTree({ activeView, filter, hierarchyConfig, visibilityHandlerOverrides, getFilteredPaths, onModelsFiltered, selectionPredicate: nodeTypeSelectionPredicate, emptyTreeContent, getSubTreePaths, }) {
     const hierarchyConfiguration = useMemo(() => ({
         ...defaultHierarchyConfiguration,
         ...hierarchyConfig,
     }), 
     // eslint-disable-next-line react-hooks/exhaustive-deps
     Object.values(hierarchyConfig ?? {}));
-    const { onFeatureUsed } = useTelemetryContext();
-    const { getModelsTreeIdsCache, visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility(activeView, hierarchyConfiguration, visibilityHandlerOverrides);
-    const { loadFocusedItems } = useFocusedInstancesContext();
-    const getHierarchyDefinition = useCallback(({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration }), [getModelsTreeIdsCache, hierarchyConfiguration]);
-    const getPaths = useMemo(() => {
-        setFilteringError(undefined);
-        onModelsFiltered?.(undefined);
-        // reset filtered paths if there is no filters applied. This allows to keep current filtered paths until new paths are loaded.
-        if (!loadFocusedItems && !getFilteredPaths && !filter) {
-            onFilteredPathsChanged(undefined);
-        }
-        const handlePaths = async (filteredPaths, classInspector) => {
-            onFilteredPathsChanged(filteredPaths);
-            if (!onModelsFiltered) {
-                return;
-            }
-            const modelIds = await getModels(filteredPaths, getModelsTreeIdsCache(), classInspector);
-            onModelsFiltered(modelIds);
-        };
-        if (loadFocusedItems) {
-            return async ({ imodelAccess }) => {
-                try {
-                    const focusedItems = await collectFocusedItems(loadFocusedItems);
-                    const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
-                        imodelAccess,
-                        idsCache: getModelsTreeIdsCache(),
-                        targetItems: focusedItems,
-                        hierarchyConfig: hierarchyConfiguration,
-                    });
-                    void handlePaths(paths, imodelAccess);
-                    return paths.map((path) => ("path" in path ? path : { path, options: { autoExpand: true } }));
-                }
-                catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyInstancesFocused" : "unknownInstanceFocusError";
-                    if (newError !== "tooManyInstancesFocused") {
-                        const feature = e instanceof Error && e.message.includes("query too long to execute or server is too busy") ? "error-timeout" : "error-unknown";
-                        onFeatureUsed({ featureId: feature, reportInteraction: false });
-                    }
-                    setFilteringError(newError);
-                    return [];
-                }
-            };
-        }
-        if (getFilteredPaths) {
-            return async ({ imodelAccess }) => {
-                try {
-                    const paths = await getFilteredPaths({
-                        createInstanceKeyPaths: async (props) => ModelsTreeDefinition.createInstanceKeyPaths({
-                            ...props,
-                            imodelAccess,
-                            idsCache: getModelsTreeIdsCache(),
-                            hierarchyConfig: hierarchyConfiguration,
-                            limit: "unbounded",
-                        }),
-                    });
-                    void handlePaths(paths, imodelAccess);
-                    return paths;
-                }
-                catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyFilterMatches" : "unknownFilterError";
-                    if (newError !== "tooManyFilterMatches") {
-                        const feature = e instanceof Error && e.message.includes("query too long to execute or server is too busy") ? "error-timeout" : "error-unknown";
-                        onFeatureUsed({ featureId: feature, reportInteraction: false });
-                    }
-                    setFilteringError(newError);
-                    return [];
-                }
-            };
-        }
-        if (filter) {
-            return async ({ imodelAccess }) => {
-                onFeatureUsed({ featureId: "filtering", reportInteraction: true });
-                try {
-                    const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
-                        imodelAccess,
-                        label: filter,
-                        idsCache: getModelsTreeIdsCache(),
-                        hierarchyConfig: hierarchyConfiguration,
-                    });
-                    void handlePaths(paths, imodelAccess);
-                    return paths.map((path) => ("path" in path ? path : { path, options: { autoExpand: true } }));
-                }
-                catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyFilterMatches" : "unknownFilterError";
-                    if (newError !== "tooManyFilterMatches") {
-                        const feature = e instanceof Error && e.message.includes("query too long to execute or server is too busy") ? "error-timeout" : "error-unknown";
-                        onFeatureUsed({ featureId: feature, reportInteraction: false });
-                    }
-                    setFilteringError(newError);
-                    return [];
-                }
-            };
-        }
-        return undefined;
-    }, [filter, loadFocusedItems, getModelsTreeIdsCache, onFeatureUsed, getFilteredPaths, hierarchyConfiguration, onModelsFiltered, onFilteredPathsChanged]);
+    const componentId = useGuid();
+    const { getCache: getModelsTreeIdsCache } = useIdsCache({
+        imodel: activeView.iModel,
+        createCache,
+        cacheSpecificProps: useMemo(() => ({ hierarchyConfig: hierarchyConfiguration }), [hierarchyConfiguration]),
+        componentId,
+    });
+    const { visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility({
+        activeView,
+        createFilteredTree,
+        createTreeSpecificVisibilityHandler: useCallback((treeProps) => createTreeSpecificVisibilityHandler({ ...treeProps, overrides: visibilityHandlerOverrides }), [visibilityHandlerOverrides]),
+        getCache: getModelsTreeIdsCache,
+        componentId,
+    });
+    const getHierarchyDefinition = useCallback(({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration, componentId }), [getModelsTreeIdsCache, hierarchyConfiguration, componentId]);
+    const { getPaths, filteringError, subTreeError } = useFilteredPaths({
+        hierarchyConfiguration,
+        filter,
+        getFilteredPaths,
+        getModelsTreeIdsCache,
+        onFilteredPathsChanged,
+        onModelsFiltered,
+        getSubTreePaths,
+        componentId,
+    });
     const nodeSelectionPredicate = useCallback((node) => {
         if (!nodeTypeSelectionPredicate) {
             return true;
@@ -137,57 +70,56 @@ export function useModelsTree({ activeView, filter, hierarchyConfig, visibilityH
             visibilityHandlerFactory,
             getHierarchyDefinition,
             getFilteredPaths: getPaths,
-            emptyTreeContent: getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent),
-            highlight: filter ? { text: filter } : undefined,
+            emptyTreeContent: useMemo(() => getEmptyTreeContentComponent(filter, subTreeError, filteringError, emptyTreeContent), [filter, subTreeError, filteringError, emptyTreeContent]),
+            highlightText: filter,
             selectionPredicate: nodeSelectionPredicate,
         },
         rendererProps: {
             // onDoubleClick,
-            getIcon,
+            getDecorations: useCallback((node) => _jsx(ModelsTreeIcon, { node: node }), []),
         },
     };
 }
-async function getModels(paths, idsCache, classInspector) {
-    if (!paths) {
-        return undefined;
-    }
-    const targetModels = new Set();
-    const targetSubjects = new Set();
-    for (const path of paths) {
-        const currPath = Array.isArray(path) ? path : path.path;
-        for (let i = 0; i < currPath.length; i++) {
-            const currStep = currPath[i];
-            if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currStep)) {
-                break;
-            }
-            // if paths end with subject need to get all models under that subject
-            if (i === currPath.length - 1 && currStep.className === "BisCore.Subject") {
-                targetSubjects.add(currStep.id);
-                break;
-            }
-            // collect all the models from the filtered path
-            if (await classInspector.classDerivesFrom(currStep.className, "BisCore.GeometricModel3d")) {
-                targetModels.add(currStep.id);
-            }
-        }
-    }
-    const matchingModels = await idsCache.getSubjectModelIds([...targetSubjects]);
-    return [...targetModels, ...matchingModels];
+async function createFilteredTree(props) {
+    const { filteringPaths, imodelAccess } = props;
+    return createFilteredModelsTree({
+        imodelAccess,
+        filteringPaths,
+    });
+}
+function createTreeSpecificVisibilityHandler(props) {
+    const { info, getCache, overrideHandler, overrides, viewport } = props;
+    return new ModelsTreeVisibilityHandler({
+        alwaysAndNeverDrawnElementInfo: info,
+        overrideHandler,
+        idsCache: getCache(),
+        viewport,
+        overrides,
+    });
 }
-function getEmptyTreeContentComponent(filter, error, emptyTreeContent) {
+function getEmptyTreeContentComponent(filter, subTreeError, error, emptyTreeContent) {
+    if (isSubTreeError(subTreeError)) {
+        return _jsx(SubTreeError, { base: "modelsTree", error: subTreeError });
+    }
     if (isInstanceFocusError(error)) {
         return _jsx(InstanceFocusError, { error: error });
     }
     if (isFilterError(error)) {
-        return _jsx(Text, { children: TreeWidget.translate(`modelsTree.filtering.${error}`) });
+        if (error === "tooManyFilterMatches") {
+            return _jsx(TooManyFilterMatches, { base: "modelsTree" });
+        }
+        return _jsx(FilterUnknownError, { base: "modelsTree" });
     }
     if (filter) {
-        return _jsx(Text, { children: TreeWidget.translate("modelsTree.filtering.noMatches", { filter }) });
+        return _jsx(NoFilterMatches, { base: "modelsTree" });
     }
     if (emptyTreeContent) {
         return emptyTreeContent;
     }
-    return _jsx(EmptyTreeContent, { icon: modelIcon });
+    return _jsx(EmptyTreeContent, { icon: modelSvg });
+}
+function isSubTreeError(error) {
+    return error === "unknownSubTreeError";
 }
 function isFilterError(error) {
     return error === "tooManyFilterMatches" || error === "unknownFilterError";
@@ -196,108 +128,35 @@ function isInstanceFocusError(error) {
     return error === "tooManyInstancesFocused" || error === "unknownInstanceFocusError";
 }
 function InstanceFocusError({ error }) {
-    const { toggle } = useFocusedInstancesContext();
-    const localizedMessage = createLocalizedMessage(TreeWidget.translate(`modelsTree.filtering.${error}`), () => toggle());
-    return _jsx(Text, { children: localizedMessage });
+    if (error === "tooManyInstancesFocused") {
+        return _jsx(TooManyInstancesFocused, { base: "modelsTree" });
+    }
+    return _jsx(UnknownInstanceFocusError, { base: "modelsTree" });
 }
-const subjectIcon = new URL("@itwin/itwinui-icons/tree-subject.svg", import.meta.url).href;
-const classIcon = new URL("@itwin/itwinui-icons/tree-class.svg", import.meta.url).href;
-const modelIcon = new URL("@itwin/itwinui-icons/model-cube.svg", import.meta.url).href;
-const categoryIcon = new URL("@itwin/itwinui-icons/tree-category.svg", import.meta.url).href;
-const elementIcon = new URL("@itwin/itwinui-icons/tree-element.svg", import.meta.url).href;
-function getIcon(node) {
-    if (node.extendedData?.imageId === undefined) {
+/** @beta */
+export function ModelsTreeIcon({ node }) {
+    if (node.nodeData.extendedData?.imageId === undefined) {
         return undefined;
     }
-    switch (node.extendedData.imageId) {
-        case "icon-layers":
-            return categoryIcon;
-        case "icon-item":
-            return elementIcon;
-        case "icon-ec-class":
-            return classIcon;
-        case "icon-folder":
-            return subjectIcon;
-        case "icon-model":
-            return modelIcon;
-    }
-    return undefined;
-}
-function createVisibilityHandlerFactory(activeView, idsCacheGetter, overrides, filteredPaths) {
-    return ({ imodelAccess }) => createModelsTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess, overrides, filteredPaths });
-}
-function useCachedVisibility(activeView, hierarchyConfig, overrides) {
-    const cacheRef = useRef();
-    const currentIModelRef = useRef(activeView.iModel);
-    const resetModelsTreeIdsCache = () => {
-        cacheRef.current?.[Symbol.dispose]();
-        cacheRef.current = undefined;
-    };
-    const getModelsTreeIdsCache = useCallback(() => {
-        if (!cacheRef.current) {
-            cacheRef.current = new ModelsTreeIdsCache(createECSqlQueryExecutor(currentIModelRef.current), hierarchyConfig);
+    const getIcon = () => {
+        switch (node.nodeData.extendedData.imageId) {
+            case "icon-layers":
+                return categorySvg;
+            case "icon-item":
+                return elementSvg;
+            case "icon-ec-class":
+                return classSvg;
+            case "icon-folder":
+                return subjectSvg;
+            case "icon-model":
+                return modelSvg;
+            default:
+                return undefined;
         }
-        return cacheRef.current;
-    }, [hierarchyConfig]);
-    const [filteredPaths, setFilteredPaths] = useState();
-    const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths));
-    useIModelChangeListener({
-        imodel: activeView.iModel,
-        action: useCallback(() => {
-            resetModelsTreeIdsCache();
-            setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths));
-        }, [activeView, getModelsTreeIdsCache, overrides, filteredPaths]),
-    });
-    useEffect(() => {
-        currentIModelRef.current = activeView.iModel;
-        setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths));
-        return () => resetModelsTreeIdsCache();
-    }, [activeView, getModelsTreeIdsCache, overrides, filteredPaths]);
-    return {
-        getModelsTreeIdsCache,
-        visibilityHandlerFactory,
-        onFilteredPathsChanged: useCallback((paths) => setFilteredPaths(paths), []),
     };
+    return _jsx(Icon, { href: getIcon() });
 }
-async function collectFocusedItems(loadFocusedItems) {
-    const focusedItems = [];
-    const groupingNodeInfos = [];
-    for await (const key of loadFocusedItems()) {
-        if ("id" in key) {
-            focusedItems.push(key);
-            continue;
-        }
-        if (!HierarchyNodeKey.isClassGrouping(key.key)) {
-            continue;
-        }
-        const groupingNode = key;
-        if (!groupingNode.nonGroupingAncestor || !HierarchyNodeKey.isInstances(groupingNode.nonGroupingAncestor.key)) {
-            continue;
-        }
-        const parentKey = groupingNode.nonGroupingAncestor.key;
-        const type = groupingNode.nonGroupingAncestor.extendedData?.isCategory ? "category" : "element";
-        const modelIds = (groupingNode.nonGroupingAncestor.extendedData?.modelIds ?? []).flatMap((ids) => ids);
-        groupingNodeInfos.push({ groupingNode, parentType: type, parentKey, modelIds });
-    }
-    focusedItems.push(...groupingNodeInfos.map(({ parentKey, parentType, groupingNode, modelIds }) => ({
-        parent: parentType === "element"
-            ? { type: "element", ids: parentKey.instanceKeys.map((key) => key.id) }
-            : { type: "category", ids: parentKey.instanceKeys.map((key) => key.id), modelIds },
-        groupingNode,
-    })));
-    return focusedItems;
-}
-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, { onClick: (e) => {
-                    e.stopPropagation();
-                    onClick?.();
-                }, children: innerText }), textAfter ? textAfter : null] }));
+function createCache(props) {
+    return new ModelsTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.hierarchyConfig, props.componentId);
 }
 //# sourceMappingURL=UseModelsTree.js.map

package/lib/esm/tree-widget-react/components/trees/models-tree/UseModelsTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseModelsTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/models-tree/UseModelsTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC1E,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAC5F,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAC5E,OAAO,EAAE,0BAA0B,EAAE,MAAM,sCAAsC,CAAC;AAClF,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AA0ChG;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,EAC5B,UAAU,EACV,MAAM,EACN,eAAe,EACf,0BAA0B,EAC1B,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAAE,0BAA0B,EAC9C,gBAAgB,GACG;IACnB,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;IACF,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,MAAM,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,GAAG,mBAAmB,CACrG,UAAU,EACV,sBAAsB,EACtB,0BAA0B,CAC3B,CAAC;IACF,MAAM,EAAE,gBAAgB,EAAE,GAAG,0BAA0B,EAAE,CAAC;IAE1D,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,QAAQ,GAAG,OAAO,CAAsD,GAAG,EAAE;QACjF,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,gBAAgB,EAAE,CAAC,SAAS,CAAC,CAAC;QAE9B,8HAA8H;QAC9H,IAAI,CAAC,gBAAgB,IAAI,CAAC,gBAAgB,IAAI,CAAC,MAAM,EAAE;YACrD,sBAAsB,CAAC,SAAS,CAAC,CAAC;SACnC;QAED,MAAM,WAAW,GAAG,KAAK,EAAE,aAAuC,EAAE,cAAyC,EAAE,EAAE;YAC/G,sBAAsB,CAAC,aAAa,CAAC,CAAC;YACtC,IAAI,CAAC,gBAAgB,EAAE;gBACrB,OAAO;aACR;YAED,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,aAAa,EAAE,qBAAqB,EAAE,EAAE,cAAc,CAAC,CAAC;YACzF,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,gBAAgB,EAAE;YACpB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBAChC,IAAI;oBACF,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;oBACjE,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;wBAC9D,YAAY;wBACZ,QAAQ,EAAE,qBAAqB,EAAE;wBACjC,WAAW,EAAE,YAAY;wBACzB,eAAe,EAAE,sBAAsB;qBACxC,CAAC,CAAC;oBACH,KAAK,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;oBACtC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;iBAC/F;gBAAC,OAAO,CAAC,EAAE;oBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,2BAA2B,CAAC;oBACjH,IAAI,QAAQ,KAAK,yBAAyB,EAAE;wBAC1C,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;qBACjE;oBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;oBAC5B,OAAO,EAAE,CAAC;iBACX;YACH,CAAC,CAAC;SACH;QAED,IAAI,gBAAgB,EAAE;YACpB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBAChC,IAAI;oBACF,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAAC;wBACnC,sBAAsB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CACtC,oBAAoB,CAAC,sBAAsB,CAAC;4BAC1C,GAAG,KAAK;4BACR,YAAY;4BACZ,QAAQ,EAAE,qBAAqB,EAAE;4BACjC,eAAe,EAAE,sBAAsB;4BACvC,KAAK,EAAE,WAAW;yBACnB,CAAC;qBACL,CAAC,CAAC;oBACH,KAAK,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;oBACtC,OAAO,KAAK,CAAC;iBACd;gBAAC,OAAO,CAAC,EAAE;oBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE;wBACvC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;qBACjE;oBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;oBAC5B,OAAO,EAAE,CAAC;iBACX;YACH,CAAC,CAAC;SACH;QAED,IAAI,MAAM,EAAE;YACV,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBAChC,aAAa,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnE,IAAI;oBACF,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;wBAC9D,YAAY;wBACZ,KAAK,EAAE,MAAM;wBACb,QAAQ,EAAE,qBAAqB,EAAE;wBACjC,eAAe,EAAE,sBAAsB;qBACxC,CAAC,CAAC;oBACH,KAAK,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;oBACtC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;iBAC/F;gBAAC,OAAO,CAAC,EAAE;oBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE;wBACvC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;qBACjE;oBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;oBAC5B,OAAO,EAAE,CAAC;iBACX;YACH,CAAC,CAAC;SACH;QACD,OAAO,SAAS,CAAC;IACnB,CAAC,EAAE,CAAC,MAAM,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,aAAa,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,sBAAsB,CAAC,CAAC,CAAC;IAEzJ,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,IAAI,EAAE,EAAE;QACP,IAAI,CAAC,0BAA0B,EAAE;YAC/B,OAAO,IAAI,CAAC;SACb;QACD,OAAO,0BAA0B,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC,EACD,CAAC,0BAA0B,CAAC,CAC7B,CAAC;IAEF,+BAA+B;IAC/B,OAAO;QACL,eAAe,EAAE;YACf,QAAQ,EAAE,gBAAgB;YAC1B,wBAAwB;YACxB,sBAAsB;YACtB,gBAAgB,EAAE,QAAQ;YAC1B,gBAAgB,EAAE,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC;YACxF,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;YAChD,kBAAkB,EAAE,sBAAsB;SAC3C;QACD,aAAa,EAAE;YACb,iBAAiB;YACjB,OAAO;SACR;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAA+B,EAAE,QAA4B,EAAE,cAAyC;IAC/H,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,YAAY,GAAG,IAAI,GAAG,EAAc,CAAC;IAC3C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAc,CAAC;IAC7C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE;gBAC/D,MAAM;aACP;YAED,sEAAsE;YACtE,IAAI,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,SAAS,KAAK,iBAAiB,EAAE;gBACzE,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAChC,MAAM;aACP;YAED,gDAAgD;YAChD,IAAI,MAAM,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,0BAA0B,CAAC,EAAE;gBACzF,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;aAC/B;SACF;KACF;IAED,MAAM,cAAc,GAAG,MAAM,QAAQ,CAAC,kBAAkB,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;IAC9E,OAAO,CAAC,GAAG,YAAY,EAAE,GAAG,cAAc,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAgC,EAAE,gBAAkC;IACzH,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,IAAI,gBAAgB,EAAE;QACpB,OAAO,gBAAgB,CAAC;KACzB;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,SAAS,GAAI,CAAC;AAC/C,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,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,uCAAuC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAC3F,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,qCAAqC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACvF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,qCAAqC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACvF,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,wCAAwC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAC7F,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,uCAAuC,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAE3F,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,YAAY,CAAC;QACtB,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,eAAe;YAClB,OAAO,SAAS,CAAC;QACnB,KAAK,aAAa;YAChB,OAAO,WAAW,CAAC;QACrB,KAAK,YAAY;YACf,OAAO,SAAS,CAAC;KACpB;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,8BAA8B,CACrC,UAAoB,EACpB,cAAwC,EACxC,SAAgD,EAChD,aAAwC;IAExC,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,iCAAiC,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;AAC/J,CAAC;AAED,SAAS,mBAAmB,CAAC,UAAoB,EAAE,eAAiD,EAAE,SAAgD;IACpJ,MAAM,QAAQ,GAAG,MAAM,EAAsB,CAAC;IAC9C,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAEnD,MAAM,uBAAuB,GAAG,GAAG,EAAE;QACnC,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACrC,QAAQ,CAAC,OAAO,GAAG,SAAS,CAAC;IAC/B,CAAC,CAAC;IACF,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,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,EAA4B,CAAC;IAC/E,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAkD,GAAG,EAAE,CAC7H,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,EAAE,aAAa,CAAC,CAC5F,CAAC;IAEF,uBAAuB,CAAC;QACtB,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE;YACvB,uBAAuB,EAAE,CAAC;YAC1B,2BAA2B,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC;QACjI,CAAC,EAAE,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;KAClE,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,gBAAgB,CAAC,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC;QAC7C,2BAA2B,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC;QAC/H,OAAO,GAAG,EAAE,CAAC,uBAAuB,EAAE,CAAC;IACzC,CAAC,EAAE,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC;IAElE,OAAO;QACL,qBAAqB;QACrB,wBAAwB;QACxB,sBAAsB,EAAE,WAAW,CAAC,CAAC,KAA2C,EAAE,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;KAClH,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,gBAAkF;IACnH,MAAM,YAAY,GAA2C,EAAE,CAAC;IAChE,MAAM,iBAAiB,GAKlB,EAAE,CAAC;IACR,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,gBAAgB,EAAE,EAAE;QAC1C,IAAI,IAAI,IAAI,GAAG,EAAE;YACf,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,SAAS;SACV;QAED,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YAC9C,SAAS;SACV;QAED,MAAM,YAAY,GAAG,GAAiC,CAAC;QACvD,IAAI,CAAC,YAAY,CAAC,mBAAmB,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE;YAC5G,SAAS;SACV;QAED,MAAM,SAAS,GAAG,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC;QACvD,MAAM,IAAI,GAAG,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAChG,MAAM,QAAQ,GAAG,CAAE,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAA2B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAC3H,iBAAiB,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;KACjF;IACD,YAAY,CAAC,IAAI,CACf,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/E,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,YAAY;KACb,CAAC,CAAC,CACJ,CAAC;IACF,OAAO,YAAY,CAAC;AACtB,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,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 { Anchor, Text } from \"@itwin/itwinui-react/bricks\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { HierarchyNodeIdentifier, HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { TreeWidget } from \"../../../TreeWidget.js\";\nimport { EmptyTreeContent } from \"../common/components/EmptyTreeContent.js\";\nimport { useFocusedInstancesContext } from \"../common/FocusedInstancesContext.js\";\nimport { FilterLimitExceededError } from \"../common/TreeErrors.js\";\nimport { useIModelChangeListener } from \"../common/UseIModelChangeListener.js\";\nimport { useTelemetryContext } from \"../common/UseTelemetryContext.js\";\nimport { ModelsTreeIdsCache } from \"./internal/ModelsTreeIdsCache.js\";\nimport { ModelsTreeNode } from \"./internal/ModelsTreeNode.js\";\nimport { createModelsTreeVisibilityHandler } from \"./internal/ModelsTreeVisibilityHandler.js\";\nimport { defaultHierarchyConfiguration, ModelsTreeDefinition } from \"./ModelsTreeDefinition.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { GroupingHierarchyNode, HierarchyFilteringPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { ClassGroupingHierarchyNode, ElementsGroupInfo, ModelsTreeHierarchyConfiguration } from \"./ModelsTreeDefinition.js\";\nimport type { ModelsTreeVisibilityHandlerOverrides } from \"./internal/ModelsTreeVisibilityHandler.js\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\n\ntype ModelsTreeFilteringError = \"tooManyFilterMatches\" | \"tooManyInstancesFocused\" | \"unknownFilterError\" | \"unknownInstanceFocusError\";\n\n/** @beta */\nexport interface UseModelsTreeProps {\n  filter?: string;\n  activeView: Viewport;\n  hierarchyConfig?: Partial<ModelsTreeHierarchyConfiguration>;\n  visibilityHandlerOverrides?: ModelsTreeVisibilityHandlerOverrides;\n  getFilteredPaths?: (props: {\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> } | { label: string }) => Promise<HierarchyFilteringPath[]>;\n  }) => Promise<HierarchyFilteringPath[]>;\n  onModelsFiltered?: (modelIds: Id64String[] | undefined) => void;\n  /**\n   * An optional predicate to allow or prohibit selection of a node.\n   * When not supplied, all nodes are selectable.\n   */\n  selectionPredicate?: (props: { node: PresentationHierarchyNode; type: \"subject\" | \"model\" | \"category\" | \"element\" | \"elements-class-group\" }) => boolean;\n  emptyTreeContent?: ReactNode;\n}\n\n/** @beta */\ninterface UseModelsTreeResult {\n  modelsTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlight\" | \"emptyTreeContent\" | \"selectionPredicate\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getIcon\">>;\n}\n\n/**\n * Custom hook to create and manage state for the models tree.\n * @beta\n */\nexport function useModelsTree({\n  activeView,\n  filter,\n  hierarchyConfig,\n  visibilityHandlerOverrides,\n  getFilteredPaths,\n  onModelsFiltered,\n  selectionPredicate: nodeTypeSelectionPredicate,\n  emptyTreeContent,\n}: UseModelsTreeProps): UseModelsTreeResult {\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  const { onFeatureUsed } = useTelemetryContext();\n\n  const { getModelsTreeIdsCache, visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility(\n    activeView,\n    hierarchyConfiguration,\n    visibilityHandlerOverrides,\n  );\n  const { loadFocusedItems } = useFocusedInstancesContext();\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    ({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration }),\n    [getModelsTreeIdsCache, hierarchyConfiguration],\n  );\n\n  const getPaths = useMemo<VisibilityTreeProps[\"getFilteredPaths\"] | undefined>(() => {\n    setFilteringError(undefined);\n    onModelsFiltered?.(undefined);\n\n    // reset filtered paths if there is no filters applied. This allows to keep current filtered paths until new paths are loaded.\n    if (!loadFocusedItems && !getFilteredPaths && !filter) {\n      onFilteredPathsChanged(undefined);\n    }\n\n    const handlePaths = async (filteredPaths: HierarchyFilteringPath[], classInspector: ECClassHierarchyInspector) => {\n      onFilteredPathsChanged(filteredPaths);\n      if (!onModelsFiltered) {\n        return;\n      }\n\n      const modelIds = await getModels(filteredPaths, getModelsTreeIdsCache(), classInspector);\n      onModelsFiltered(modelIds);\n    };\n\n    if (loadFocusedItems) {\n      return async ({ imodelAccess }) => {\n        try {\n          const focusedItems = await collectFocusedItems(loadFocusedItems);\n          const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n            imodelAccess,\n            idsCache: getModelsTreeIdsCache(),\n            targetItems: focusedItems,\n            hierarchyConfig: hierarchyConfiguration,\n          });\n          void handlePaths(paths, imodelAccess);\n          return paths.map((path) => (\"path\" in path ? path : { path, options: { autoExpand: true } }));\n        } catch (e) {\n          const newError = e instanceof FilterLimitExceededError ? \"tooManyInstancesFocused\" : \"unknownInstanceFocusError\";\n          if (newError !== \"tooManyInstancesFocused\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setFilteringError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (getFilteredPaths) {\n      return async ({ imodelAccess }) => {\n        try {\n          const paths = await getFilteredPaths({\n            createInstanceKeyPaths: async (props) =>\n              ModelsTreeDefinition.createInstanceKeyPaths({\n                ...props,\n                imodelAccess,\n                idsCache: getModelsTreeIdsCache(),\n                hierarchyConfig: hierarchyConfiguration,\n                limit: \"unbounded\",\n              }),\n          });\n          void handlePaths(paths, imodelAccess);\n          return paths;\n        } catch (e) {\n          const newError = e instanceof FilterLimitExceededError ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n          if (newError !== \"tooManyFilterMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setFilteringError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (filter) {\n      return async ({ imodelAccess }) => {\n        onFeatureUsed({ featureId: \"filtering\", reportInteraction: true });\n        try {\n          const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n            imodelAccess,\n            label: filter,\n            idsCache: getModelsTreeIdsCache(),\n            hierarchyConfig: hierarchyConfiguration,\n          });\n          void handlePaths(paths, imodelAccess);\n          return paths.map((path) => (\"path\" in path ? path : { path, options: { autoExpand: true } }));\n        } catch (e) {\n          const newError = e instanceof FilterLimitExceededError ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n          if (newError !== \"tooManyFilterMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setFilteringError(newError);\n          return [];\n        }\n      };\n    }\n    return undefined;\n  }, [filter, loadFocusedItems, getModelsTreeIdsCache, onFeatureUsed, getFilteredPaths, hierarchyConfiguration, onModelsFiltered, onFilteredPathsChanged]);\n\n  const nodeSelectionPredicate = useCallback<NonNullable<VisibilityTreeProps[\"selectionPredicate\"]>>(\n    (node) => {\n      if (!nodeTypeSelectionPredicate) {\n        return true;\n      }\n      return nodeTypeSelectionPredicate({ node, type: ModelsTreeNode.getType(node.nodeData) });\n    },\n    [nodeTypeSelectionPredicate],\n  );\n\n  // TODO: add double click logic\n  return {\n    modelsTreeProps: {\n      treeName: \"models-tree-v2\",\n      visibilityHandlerFactory,\n      getHierarchyDefinition,\n      getFilteredPaths: getPaths,\n      emptyTreeContent: getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent),\n      highlight: filter ? { text: filter } : undefined,\n      selectionPredicate: nodeSelectionPredicate,\n    },\n    rendererProps: {\n      // onDoubleClick,\n      getIcon,\n    },\n  };\n}\n\nasync function getModels(paths: HierarchyFilteringPath[], idsCache: ModelsTreeIdsCache, classInspector: ECClassHierarchyInspector) {\n  if (!paths) {\n    return undefined;\n  }\n\n  const targetModels = new Set<Id64String>();\n  const targetSubjects = new Set<Id64String>();\n  for (const path of paths) {\n    const currPath = Array.isArray(path) ? path : path.path;\n    for (let i = 0; i < currPath.length; i++) {\n      const currStep = currPath[i];\n      if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currStep)) {\n        break;\n      }\n\n      // if paths end with subject need to get all models under that subject\n      if (i === currPath.length - 1 && currStep.className === \"BisCore.Subject\") {\n        targetSubjects.add(currStep.id);\n        break;\n      }\n\n      // collect all the models from the filtered path\n      if (await classInspector.classDerivesFrom(currStep.className, \"BisCore.GeometricModel3d\")) {\n        targetModels.add(currStep.id);\n      }\n    }\n  }\n\n  const matchingModels = await idsCache.getSubjectModelIds([...targetSubjects]);\n  return [...targetModels, ...matchingModels];\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: ModelsTreeFilteringError, emptyTreeContent?: React.ReactNode) {\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  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={modelIcon} />;\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\nconst subjectIcon = new URL(\"@itwin/itwinui-icons/tree-subject.svg\", import.meta.url).href;\nconst classIcon = new URL(\"@itwin/itwinui-icons/tree-class.svg\", import.meta.url).href;\nconst modelIcon = new URL(\"@itwin/itwinui-icons/model-cube.svg\", import.meta.url).href;\nconst categoryIcon = new URL(\"@itwin/itwinui-icons/tree-category.svg\", import.meta.url).href;\nconst elementIcon = new URL(\"@itwin/itwinui-icons/tree-element.svg\", import.meta.url).href;\n\nfunction getIcon(node: PresentationHierarchyNode): string | undefined {\n  if (node.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  switch (node.extendedData.imageId) {\n    case \"icon-layers\":\n      return categoryIcon;\n    case \"icon-item\":\n      return elementIcon;\n    case \"icon-ec-class\":\n      return classIcon;\n    case \"icon-folder\":\n      return subjectIcon;\n    case \"icon-model\":\n      return modelIcon;\n  }\n\n  return undefined;\n}\n\nfunction createVisibilityHandlerFactory(\n  activeView: Viewport,\n  idsCacheGetter: () => ModelsTreeIdsCache,\n  overrides?: ModelsTreeVisibilityHandlerOverrides,\n  filteredPaths?: HierarchyFilteringPath[],\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  return ({ imodelAccess }) => createModelsTreeVisibilityHandler({ viewport: activeView, idsCache: idsCacheGetter(), imodelAccess, overrides, filteredPaths });\n}\n\nfunction useCachedVisibility(activeView: Viewport, hierarchyConfig: ModelsTreeHierarchyConfiguration, overrides?: ModelsTreeVisibilityHandlerOverrides) {\n  const cacheRef = useRef<ModelsTreeIdsCache>();\n  const currentIModelRef = useRef(activeView.iModel);\n\n  const resetModelsTreeIdsCache = () => {\n    cacheRef.current?.[Symbol.dispose]();\n    cacheRef.current = undefined;\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 [filteredPaths, setFilteredPaths] = useState<HierarchyFilteringPath[]>();\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState<VisibilityTreeProps[\"visibilityHandlerFactory\"]>(() =>\n    createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths),\n  );\n\n  useIModelChangeListener({\n    imodel: activeView.iModel,\n    action: useCallback(() => {\n      resetModelsTreeIdsCache();\n      setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths));\n    }, [activeView, getModelsTreeIdsCache, overrides, filteredPaths]),\n  });\n\n  useEffect(() => {\n    currentIModelRef.current = activeView.iModel;\n    setVisibilityHandlerFactory(() => createVisibilityHandlerFactory(activeView, getModelsTreeIdsCache, overrides, filteredPaths));\n    return () => resetModelsTreeIdsCache();\n  }, [activeView, getModelsTreeIdsCache, overrides, filteredPaths]);\n\n  return {\n    getModelsTreeIdsCache,\n    visibilityHandlerFactory,\n    onFilteredPathsChanged: useCallback((paths: HierarchyFilteringPath[] | undefined) => setFilteredPaths(paths), []),\n  };\n}\n\nasync function collectFocusedItems(loadFocusedItems: () => AsyncIterableIterator<InstanceKey | GroupingHierarchyNode>) {\n  const focusedItems: Array<InstanceKey | ElementsGroupInfo> = [];\n  const groupingNodeInfos: Array<{\n    parentKey: InstancesNodeKey;\n    parentType: \"element\" | \"category\";\n    groupingNode: ClassGroupingHierarchyNode;\n    modelIds: Id64String[];\n  }> = [];\n  for await (const key of loadFocusedItems()) {\n    if (\"id\" in key) {\n      focusedItems.push(key);\n      continue;\n    }\n\n    if (!HierarchyNodeKey.isClassGrouping(key.key)) {\n      continue;\n    }\n\n    const groupingNode = key as ClassGroupingHierarchyNode;\n    if (!groupingNode.nonGroupingAncestor || !HierarchyNodeKey.isInstances(groupingNode.nonGroupingAncestor.key)) {\n      continue;\n    }\n\n    const parentKey = groupingNode.nonGroupingAncestor.key;\n    const type = groupingNode.nonGroupingAncestor.extendedData?.isCategory ? \"category\" : \"element\";\n    const modelIds = ((groupingNode.nonGroupingAncestor.extendedData?.modelIds as Id64String[][]) ?? []).flatMap((ids) => ids);\n    groupingNodeInfos.push({ groupingNode, parentType: type, parentKey, modelIds });\n  }\n  focusedItems.push(\n    ...groupingNodeInfos.map(({ parentKey, parentType, groupingNode, 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      groupingNode,\n    })),\n  );\n  return focusedItems;\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        onClick={(e) => {\n          e.stopPropagation();\n          onClick?.();\n        }}\n      >\n        {innerText}\n      </Anchor>\n      {textAfter ? textAfter : null}\n    </>\n  );\n}\n"]}
+{"version":3,"file":"UseModelsTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/models-tree/UseModelsTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAC9C,OAAO,WAAW,MAAM,sCAAsC,CAAC;AAC/D,OAAO,QAAQ,MAAM,gCAAgC,CAAC;AACtD,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,UAAU,MAAM,kCAAkC,CAAC;AAC1D,OAAO,QAAQ,MAAM,iCAAiC,CAAC;AACvD,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,eAAe,EACf,YAAY,EACZ,oBAAoB,EACpB,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wDAAwD,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,MAAM,gDAAgD,CAAC;AAC7E,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AACjF,OAAO,EAAE,2BAA2B,EAAE,MAAM,sDAAsD,CAAC;AACnG,OAAO,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAyFhG;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,EAC5B,UAAU,EACV,MAAM,EACN,eAAe,EACf,0BAA0B,EAC1B,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAAE,0BAA0B,EAC9C,gBAAgB,EAChB,eAAe,GACI;IACnB,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;IACF,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC;IAC9B,MAAM,EAAE,QAAQ,EAAE,qBAAqB,EAAE,GAAG,WAAW,CAA4E;QACjI,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,WAAW;QACX,kBAAkB,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,sBAAsB,EAAE,CAAC,EAAE,CAAC,sBAAsB,CAAC,CAAC;QAC1G,WAAW;KACZ,CAAC,CAAC;IAEH,MAAM,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,GAAG,mBAAmB,CAA8C;QAC5H,UAAU;QACV,kBAAkB;QAClB,mCAAmC,EAAE,WAAW,CAC9C,CAAC,SAAS,EAAE,EAAE,CAAC,mCAAmC,CAAC,EAAE,GAAG,SAAS,EAAE,SAAS,EAAE,0BAA0B,EAAE,CAAC,EAC3G,CAAC,0BAA0B,CAAC,CAC7B;QACD,QAAQ,EAAE,qBAAqB;QAC/B,WAAW;KACZ,CAAC,CAAC;IAEH,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,WAAW,EAAE,CAAC,EACzJ,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,WAAW,CAAC,CAC7D,CAAC;IAEF,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,gBAAgB,CAAC;QAClE,sBAAsB;QACtB,MAAM;QACN,gBAAgB;QAChB,qBAAqB;QACrB,sBAAsB;QACtB,gBAAgB;QAChB,eAAe;QACf,WAAW;KACZ,CAAC,CAAC;IAEH,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,IAAI,EAAE,EAAE;QACP,IAAI,CAAC,0BAA0B,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,0BAA0B,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC,EACD,CAAC,0BAA0B,CAAC,CAC7B,CAAC;IAEF,+BAA+B;IAC/B,OAAO;QACL,eAAe,EAAE;YACf,QAAQ,EAAE,gBAAgB;YAC1B,wBAAwB;YACxB,sBAAsB;YACtB,gBAAgB,EAAE,QAAQ;YAC1B,gBAAgB,EAAE,OAAO,CACvB,GAAG,EAAE,CAAC,4BAA4B,CAAC,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,gBAAgB,CAAC,EAC1F,CAAC,MAAM,EAAE,YAAY,EAAE,cAAc,EAAE,gBAAgB,CAAC,CACzD;YACD,aAAa,EAAE,MAAM;YACrB,kBAAkB,EAAE,sBAAsB;SAC3C;QACD,aAAa,EAAE;YACb,iBAAiB;YACjB,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAC,cAAc,IAAC,IAAI,EAAE,IAAI,GAAI,EAAE,EAAE,CAAC;SAC1E;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,KAAkD;IAClF,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC;IAC/C,OAAO,wBAAwB,CAAC;QAC9B,YAAY;QACZ,cAAc;KACf,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mCAAmC,CAC1C,KAA0H;IAE1H,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,eAAe,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACvE,OAAO,IAAI,2BAA2B,CAAC;QACrC,8BAA8B,EAAE,IAAI;QACpC,eAAe;QACf,QAAQ,EAAE,QAAQ,EAAE;QACpB,QAAQ;QACR,SAAS;KACV,CAAC,CAAC;AACL,CAAC;AAED,SAAS,4BAA4B,CACnC,MAAe,EACf,YAAqC,EACrC,KAAgC,EAChC,gBAAkC;IAElC,IAAI,cAAc,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,OAAO,KAAC,YAAY,IAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,GAAI,CAAC;IACnE,CAAC;IACD,IAAI,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;QAChC,OAAO,KAAC,kBAAkB,IAAC,KAAK,EAAE,KAAK,GAAI,CAAC;IAC9C,CAAC;IACD,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,OAAO,KAAC,oBAAoB,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;QACtD,CAAC;QACD,OAAO,KAAC,kBAAkB,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;IACpD,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,KAAC,eAAe,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;IACjD,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,QAAQ,GAAI,CAAC;AAC9C,CAAC;AAED,SAAS,cAAc,CAAC,KAAyC;IAC/D,OAAO,KAAK,KAAK,qBAAqB,CAAC;AACzC,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,IAAI,KAAK,KAAK,yBAAyB,EAAE,CAAC;QACxC,OAAO,KAAC,uBAAuB,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;IACzD,CAAC;IACD,OAAO,KAAC,yBAAyB,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;AAC3D,CAAC;AAED,YAAY;AACZ,MAAM,UAAU,cAAc,CAAC,EAAE,IAAI,EAAuC;IAC1E,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,IAAI,CAAC,QAAQ,CAAC,YAAa,CAAC,OAAO,EAAE,CAAC;YAC5C,KAAK,aAAa;gBAChB,OAAO,WAAW,CAAC;YACrB,KAAK,WAAW;gBACd,OAAO,UAAU,CAAC;YACpB,KAAK,eAAe;gBAClB,OAAO,QAAQ,CAAC;YAClB,KAAK,aAAa;gBAChB,OAAO,UAAU,CAAC;YACpB,KAAK,YAAY;gBACf,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,KAAC,IAAI,IAAC,IAAI,EAAE,OAAO,EAAE,GAAI,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,KAA8E;IACjG,OAAO,IAAI,kBAAkB,CAAC,wBAAwB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,eAAe,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;AAChI,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, useMemo } from \"react\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { Icon } from \"@stratakit/foundations\";\nimport categorySvg from \"@stratakit/icons/bis-category-3d.svg\";\nimport classSvg from \"@stratakit/icons/bis-class.svg\";\nimport elementSvg from \"@stratakit/icons/bis-element.svg\";\nimport subjectSvg from \"@stratakit/icons/bis-subject.svg\";\nimport modelSvg from \"@stratakit/icons/model-cube.svg\";\nimport {\n  EmptyTreeContent,\n  FilterUnknownError,\n  NoFilterMatches,\n  SubTreeError,\n  TooManyFilterMatches,\n  TooManyInstancesFocused,\n  UnknownInstanceFocusError,\n} from \"../common/components/EmptyTree.js\";\nimport { useGuid } from \"../common/internal/useGuid.js\";\nimport { useCachedVisibility } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { useIdsCache } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport { ModelsTreeIdsCache } from \"./internal/ModelsTreeIdsCache.js\";\nimport { ModelsTreeNode } from \"./internal/ModelsTreeNode.js\";\nimport { useFilteredPaths } from \"./internal/UseFilteredPaths.js\";\nimport { createFilteredModelsTree } from \"./internal/visibility/FilteredTree.js\";\nimport { ModelsTreeVisibilityHandler } from \"./internal/visibility/ModelsTreeVisibilityHandler.js\";\nimport { defaultHierarchyConfiguration, ModelsTreeDefinition } from \"./ModelsTreeDefinition.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CreateFilteredTreeProps, CreateTreeSpecificVisibilityHandlerProps } from \"../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport type { CreateCacheProps } from \"../common/internal/useTreeHooks/UseIdsCache.js\";\nimport type { FilteredTree } from \"../common/internal/visibility/BaseFilteredTree.js\";\nimport type { TreeWidgetViewport } from \"../common/TreeWidgetViewport.js\";\nimport type { NormalizedHierarchyFilteringPath } from \"../common/Utils.js\";\nimport type { ModelsTreeFilteringError, ModelsTreeSubTreeError } from \"./internal/UseFilteredPaths.js\";\nimport type { ModelsTreeFilterTargets } from \"./internal/visibility/FilteredTree.js\";\nimport type { ModelsTreeVisibilityHandlerOverrides } from \"./internal/visibility/ModelsTreeVisibilityHandler.js\";\nimport type { ElementsGroupInfo, ModelsTreeHierarchyConfiguration } from \"./ModelsTreeDefinition.js\";\n\n/** @beta */\nexport interface UseModelsTreeProps {\n  /**\n   * Optional search string used to filter tree nodes by label, as well as highlight matching substrings in the tree.\n   * Nodes that do not contain this string in their label will be filtered out.\n   *\n   * If `getFilteredPaths` function is provided, it will take precedence and automatic filtering by this string will not be applied.\n   * Instead, the string will be supplied to the given `getFilteredPaths` function for consumers to apply the filtering.\n   */\n  filter?: string;\n  activeView: TreeWidgetViewport;\n  hierarchyConfig?: Partial<ModelsTreeHierarchyConfiguration>;\n  visibilityHandlerOverrides?: ModelsTreeVisibilityHandlerOverrides;\n  /**\n   * Optional function for applying custom filtering on the hierarchy. Use it when you want full control over which nodes should be displayed, based on more complex logic or known instance keys.\n   *\n   * When defined, this function takes precedence over filtering by `filter` string. If both are supplied, the `filter` is provided as an argument to `getFilteredPaths`.\n   *\n   * @param props Parameters provided when `getFilteredPaths` is called:\n   * - `createInstanceKeyPaths`: Helper function to create filter paths.\n   * - `filter`: The filter string which would otherwise be used for default filtering.\n   *\n   * **Example use cases:**\n   * - You have a list of `InstanceKey` items, which you want to use for filtering the hierarchy.\n   * - You want to create filter paths based on node label, but also apply some extra conditions (for example exclude paths with sub-models).\n   * - You want to construct custom filtered paths. For example: create a filter path for each geometric element which has a parent element.\n   *\n   * @note Paths returned  by `createInstanceKeyPaths` will not have `reveal` flag set. If you want nodes to be expanded, iterate over the paths and\n   * set `reveal: true` manually.\n   */\n  getFilteredPaths?: (props: {\n    /** A function that creates filtering paths based on provided target instance keys or node label. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> } | { label: string }) => Promise<NormalizedHierarchyFilteringPath[]>;\n    /** Filter which would be used to create filter paths if `getFilteredPaths` wouldn't be provided. */\n    filter?: string;\n  }) => Promise<HierarchyFilteringPath[] | undefined>;\n  /**\n   * Optional function for restricting the visible hierarchy to a specific sub-tree of nodes, without changing how filtering works.\n   *\n   * Use when you want to display only part of the hierarchy, but still allow normal filtering within that sub-tree.\n   *\n   * When defined, only nodes that are in the provided paths or children of target nodes will be part of the hierarchy.\n   * Filtering (by label or custom logic) will still apply within this sub-tree.\n   *\n   * Key difference:\n   * - `getFilteredPaths` determines which nodes should be shown, giving you full control over filtering logic.\n   * - `getSubTreePaths` restricts the hierarchy to a sub-tree, but does not override the filtering logic — filtering is still applied within the restricted sub-tree.\n   */\n  getSubTreePaths?: (props: {\n    /** A function that creates filtering paths based on provided target instance keys. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> }) => Promise<NormalizedHierarchyFilteringPath[]>;\n  }) => Promise<HierarchyFilteringPath[]>;\n  onModelsFiltered?: (modelIds: Id64String[] | undefined) => void;\n  /**\n   * An optional predicate to allow or prohibit selection of a node.\n   * When not supplied, all nodes are selectable.\n   */\n  selectionPredicate?: (props: { node: PresentationHierarchyNode; type: \"subject\" | \"model\" | \"category\" | \"element\" | \"elements-class-group\" }) => boolean;\n  emptyTreeContent?: ReactNode;\n}\n\n/** @beta */\ninterface UseModelsTreeResult {\n  modelsTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlightText\" | \"emptyTreeContent\" | \"selectionPredicate\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getDecorations\">>;\n}\n\n/**\n * Custom hook to create and manage state for the models tree.\n * @beta\n */\nexport function useModelsTree({\n  activeView,\n  filter,\n  hierarchyConfig,\n  visibilityHandlerOverrides,\n  getFilteredPaths,\n  onModelsFiltered,\n  selectionPredicate: nodeTypeSelectionPredicate,\n  emptyTreeContent,\n  getSubTreePaths,\n}: UseModelsTreeProps): UseModelsTreeResult {\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  const componentId = useGuid();\n  const { getCache: getModelsTreeIdsCache } = useIdsCache<ModelsTreeIdsCache, { hierarchyConfig: ModelsTreeHierarchyConfiguration }>({\n    imodel: activeView.iModel,\n    createCache,\n    cacheSpecificProps: useMemo(() => ({ hierarchyConfig: hierarchyConfiguration }), [hierarchyConfiguration]),\n    componentId,\n  });\n\n  const { visibilityHandlerFactory, onFilteredPathsChanged } = useCachedVisibility<ModelsTreeIdsCache, ModelsTreeFilterTargets>({\n    activeView,\n    createFilteredTree,\n    createTreeSpecificVisibilityHandler: useCallback(\n      (treeProps) => createTreeSpecificVisibilityHandler({ ...treeProps, overrides: visibilityHandlerOverrides }),\n      [visibilityHandlerOverrides],\n    ),\n    getCache: getModelsTreeIdsCache,\n    componentId,\n  });\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    ({ imodelAccess }) => new ModelsTreeDefinition({ imodelAccess, idsCache: getModelsTreeIdsCache(), hierarchyConfig: hierarchyConfiguration, componentId }),\n    [getModelsTreeIdsCache, hierarchyConfiguration, componentId],\n  );\n\n  const { getPaths, filteringError, subTreeError } = useFilteredPaths({\n    hierarchyConfiguration,\n    filter,\n    getFilteredPaths,\n    getModelsTreeIdsCache,\n    onFilteredPathsChanged,\n    onModelsFiltered,\n    getSubTreePaths,\n    componentId,\n  });\n\n  const nodeSelectionPredicate = useCallback<NonNullable<VisibilityTreeProps[\"selectionPredicate\"]>>(\n    (node) => {\n      if (!nodeTypeSelectionPredicate) {\n        return true;\n      }\n      return nodeTypeSelectionPredicate({ node, type: ModelsTreeNode.getType(node.nodeData) });\n    },\n    [nodeTypeSelectionPredicate],\n  );\n\n  // TODO: add double click logic\n  return {\n    modelsTreeProps: {\n      treeName: \"models-tree-v2\",\n      visibilityHandlerFactory,\n      getHierarchyDefinition,\n      getFilteredPaths: getPaths,\n      emptyTreeContent: useMemo(\n        () => getEmptyTreeContentComponent(filter, subTreeError, filteringError, emptyTreeContent),\n        [filter, subTreeError, filteringError, emptyTreeContent],\n      ),\n      highlightText: filter,\n      selectionPredicate: nodeSelectionPredicate,\n    },\n    rendererProps: {\n      // onDoubleClick,\n      getDecorations: useCallback((node) => <ModelsTreeIcon node={node} />, []),\n    },\n  };\n}\n\nasync function createFilteredTree(props: CreateFilteredTreeProps<ModelsTreeIdsCache>): Promise<FilteredTree<ModelsTreeFilterTargets>> {\n  const { filteringPaths, imodelAccess } = props;\n  return createFilteredModelsTree({\n    imodelAccess,\n    filteringPaths,\n  });\n}\n\nfunction createTreeSpecificVisibilityHandler(\n  props: CreateTreeSpecificVisibilityHandlerProps<ModelsTreeIdsCache> & { overrides?: ModelsTreeVisibilityHandlerOverrides },\n) {\n  const { info, getCache, overrideHandler, overrides, viewport } = props;\n  return new ModelsTreeVisibilityHandler({\n    alwaysAndNeverDrawnElementInfo: info,\n    overrideHandler,\n    idsCache: getCache(),\n    viewport,\n    overrides,\n  });\n}\n\nfunction getEmptyTreeContentComponent(\n  filter?: string,\n  subTreeError?: ModelsTreeSubTreeError,\n  error?: ModelsTreeFilteringError,\n  emptyTreeContent?: React.ReactNode,\n) {\n  if (isSubTreeError(subTreeError)) {\n    return <SubTreeError base={\"modelsTree\"} error={subTreeError} />;\n  }\n  if (isInstanceFocusError(error)) {\n    return <InstanceFocusError error={error} />;\n  }\n  if (isFilterError(error)) {\n    if (error === \"tooManyFilterMatches\") {\n      return <TooManyFilterMatches base={\"modelsTree\"} />;\n    }\n    return <FilterUnknownError base={\"modelsTree\"} />;\n  }\n  if (filter) {\n    return <NoFilterMatches base={\"modelsTree\"} />;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={modelSvg} />;\n}\n\nfunction isSubTreeError(error: ModelsTreeSubTreeError | undefined) {\n  return error === \"unknownSubTreeError\";\n}\n\nfunction isFilterError(error: ModelsTreeFilteringError | undefined) {\n  return error === \"tooManyFilterMatches\" || error === \"unknownFilterError\";\n}\n\nfunction isInstanceFocusError(error: ModelsTreeFilteringError | undefined): error is \"tooManyInstancesFocused\" | \"unknownInstanceFocusError\" {\n  return error === \"tooManyInstancesFocused\" || error === \"unknownInstanceFocusError\";\n}\n\nfunction InstanceFocusError({ error }: { error: ModelsTreeFilteringError }) {\n  if (error === \"tooManyInstancesFocused\") {\n    return <TooManyInstancesFocused base={\"modelsTree\"} />;\n  }\n  return <UnknownInstanceFocusError base={\"modelsTree\"} />;\n}\n\n/** @beta */\nexport function ModelsTreeIcon({ node }: { node: PresentationHierarchyNode }) {\n  if (node.nodeData.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  const getIcon = () => {\n    switch (node.nodeData.extendedData!.imageId) {\n      case \"icon-layers\":\n        return categorySvg;\n      case \"icon-item\":\n        return elementSvg;\n      case \"icon-ec-class\":\n        return classSvg;\n      case \"icon-folder\":\n        return subjectSvg;\n      case \"icon-model\":\n        return modelSvg;\n      default:\n        return undefined;\n    }\n  };\n\n  return <Icon href={getIcon()} />;\n}\n\nfunction createCache(props: CreateCacheProps<{ hierarchyConfig: ModelsTreeHierarchyConfiguration }>) {\n  return new ModelsTreeIdsCache(createECSqlQueryExecutor(props.imodel), props.specificProps.hierarchyConfig, props.componentId);\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.d.ts

@@ -1,44 +1,41 @@
-import type { ModelsTreeDefinition } from "../ModelsTreeDefinition.js";
-import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { Observable } from "rxjs";
+import type { GuidString, Id64Arg, Id64Array, Id64Set, Id64String } from "@itwin/core-bentley";
 import type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { CategoryId, ModelId } from "../../common/internal/Types.js";
+import type { ModelsTreeDefinition } from "../ModelsTreeDefinition.js";
 type ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0]["hierarchyConfig"];
 /** @internal */
-export declare class ModelsTreeIdsCache {
-    private _queryExecutor;
-    private _hierarchyConfig;
-    private readonly _categoryElementCounts;
-    private _subjectInfos;
-    private _parentSubjectIds;
-    private _modelInfos;
-    private _modelKeyPaths;
-    private _subjectKeyPaths;
-    private _categoryKeyPaths;
-    constructor(_queryExecutor: LimitingECSqlQueryExecutor, _hierarchyConfig: ModelsTreeHierarchyConfiguration);
+export declare class ModelsTreeIdsCache implements Disposable {
+    #private;
+    constructor(queryExecutor: LimitingECSqlQueryExecutor, hierarchyConfig: ModelsTreeHierarchyConfiguration, componentId?: GuidString);
     [Symbol.dispose](): void;
     private querySubjects;
     private queryModels;
     private getSubjectInfos;
     /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */
-    getParentSubjectIds(): Promise<Id64String[]>;
+    getParentSubjectIds(): Observable<Id64Array>;
     /**
      * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into
      * account `hideInHierarchy` flag.
      */
-    getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]>;
+    getChildSubjectIds(parentSubjectIds: Id64Arg): Observable<Id64Array>;
     /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */
-    getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array>;
+    getSubjectModelIds(subjectIds: Id64Arg): Observable<Id64Array>;
     /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */
-    getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]>;
-    createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath>;
-    private queryModelElementCounts;
+    getChildSubjectModelIds(parentSubjectIds: Id64Arg): Observable<Id64Array>;
+    createSubjectInstanceKeysPath(targetSubjectId: Id64String): Observable<HierarchyNodeIdentifiersPath>;
     private queryModelCategories;
+    private queryModeledElements;
+    private getModelWithCategoryModeledElements;
     private getModelInfos;
-    getModelCategories(modelId: Id64String): Promise<Id64Array>;
-    getModelElementCount(modelId: Id64String): Promise<number>;
-    createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]>;
-    private queryCategoryElementCounts;
-    getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number>;
-    createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]>;
+    getAllCategories(): Observable<Id64Set>;
+    getModelCategoryIds(modelId: Id64String): Observable<Id64Array>;
+    hasSubModel(elementId: Id64String): Observable<boolean>;
+    getCategoriesModeledElements(modelId: Id64String, categoryIds: Id64Arg): Observable<Id64Array>;
+    getCategoriesElementModels(categoryIds: Id64Arg): Observable<Map<CategoryId, Array<ModelId>>>;
+    createModelInstanceKeyPaths(modelId: Id64String): Observable<HierarchyNodeIdentifiersPath[]>;
+    getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number>;
+    createCategoryInstanceKeyPaths(categoryId: Id64String): Observable<HierarchyNodeIdentifiersPath[]>;
 }
 export {};
 //# sourceMappingURL=ModelsTreeIdsCache.d.ts.map