@itwin/tree-widget-react 4.0.0-alpha.22 → 4.0.0-alpha.24

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/{UseFilteredPaths.js → UseSearchPaths.js}

@@ -4,28 +4,28 @@
  *--------------------------------------------------------------------------------------------*/
 import { useEffect, useMemo, useState } from "react";
 import { firstValueFrom } from "rxjs";
-import { HierarchyFilteringPath, HierarchyNodeIdentifier, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { HierarchyNodeIdentifier, HierarchyNodeKey, HierarchySearchPath } from "@itwin/presentation-hierarchies";
 import { useFocusedInstancesContext } from "../../common/FocusedInstancesContext.js";
 import { CLASS_NAME_GeometricModel3d, CLASS_NAME_Subject } from "../../common/internal/ClassNameDefinitions.js";
-import { FilterLimitExceededError } from "../../common/TreeErrors.js";
+import { SearchLimitExceededError } from "../../common/TreeErrors.js";
 import { useTelemetryContext } from "../../common/UseTelemetryContext.js";
-import { joinHierarchyFilteringPaths } from "../../common/Utils.js";
+import { joinHierarchySearchPaths } from "../../common/Utils.js";
 import { ModelsTreeDefinition } from "../ModelsTreeDefinition.js";
 /** @internal */
-export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPaths, getSubTreePaths, getModelsTreeIdsCache, onModelsFiltered, onFilteredPathsChanged, componentId, }) {
-    const [filteringError, setFilteringError] = useState(undefined);
+export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPaths, getSubTreePaths, getModelsTreeIdsCache, onModelsFiltered, onSearchPathsChanged, componentId, }) {
+    const [searchError, setSearchError] = useState(undefined);
     const [subTreeError, setSubTreeError] = useState(undefined);
     const { onFeatureUsed } = useTelemetryContext();
     const { loadFocusedItems } = useFocusedInstancesContext();
     useEffect(() => {
-        setFilteringError(undefined);
+        setSearchError(undefined);
         setSubTreeError(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 && !getSubTreePaths) {
-            onFilteredPathsChanged(undefined);
+        // reset search paths if there is no search applied. This allows to keep current search paths until new paths are loaded.
+        if (!loadFocusedItems && !getSearchPaths && !searchText && !getSubTreePaths) {
+            onSearchPathsChanged(undefined);
         }
-    }, [loadFocusedItems, getFilteredPaths, getSubTreePaths, filter, onModelsFiltered, onFilteredPathsChanged]);
+    }, [loadFocusedItems, getSearchPaths, getSubTreePaths, searchText, onModelsFiltered, onSearchPathsChanged]);
     const getSubTreePathsInternal = useMemo(() => {
         if (!getSubTreePaths) {
             return undefined;
@@ -43,7 +43,7 @@ export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPa
                         componentId: `${componentId}/subTree`,
                     }),
                 });
-                return paths.map(HierarchyFilteringPath.normalize).map(({ path }) => path);
+                return paths.map(HierarchySearchPath.normalize).map(({ path }) => path);
             }
             catch {
                 const newError = "unknownSubTreeError";
@@ -53,20 +53,20 @@ export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPa
         };
     }, [getModelsTreeIdsCache, hierarchyConfiguration, getSubTreePaths, componentId]);
     const getPaths = useMemo(() => {
-        const handlePaths = async (filteredPaths, classInspector) => {
-            onFilteredPathsChanged(filteredPaths);
+        const handlePaths = async (searchPaths, classInspector) => {
+            onSearchPathsChanged(searchPaths);
             if (!onModelsFiltered) {
                 return;
             }
-            const modelIds = filteredPaths ? await getModels(filteredPaths, getModelsTreeIdsCache(), classInspector) : undefined;
+            const modelIds = searchPaths ? await getModels(searchPaths, getModelsTreeIdsCache(), classInspector) : undefined;
             onModelsFiltered(modelIds);
         };
         if (loadFocusedItems) {
             return async ({ imodelAccess, abortSignal }) => {
                 try {
                     const focusedItems = await collectFocusedItems(loadFocusedItems);
-                    return await createFilteringPathsResult({
-                        getFilteringPaths: async () => {
+                    return await createSearchPathsResult({
+                        getSearchPaths: async () => {
                             const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
                                 imodelAccess,
                                 idsCache: getModelsTreeIdsCache(),
@@ -82,22 +82,22 @@ export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPa
                     });
                 }
                 catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyInstancesFocused" : "unknownInstanceFocusError";
+                    const newError = e instanceof SearchLimitExceededError ? "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);
+                    setSearchError(newError);
                     return [];
                 }
             };
         }
-        if (getFilteredPaths) {
+        if (getSearchPaths) {
             return async ({ imodelAccess, abortSignal }) => {
                 try {
-                    return await createFilteringPathsResult({
-                        getFilteringPaths: async () => {
-                            const paths = await getFilteredPaths({
+                    return await createSearchPathsResult({
+                        getSearchPaths: async () => {
+                            const paths = await getSearchPaths({
                                 createInstanceKeyPaths: async (props) => ModelsTreeDefinition.createInstanceKeyPaths({
                                     ...props,
                                     imodelAccess,
@@ -107,34 +107,34 @@ export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPa
                                     abortSignal,
                                     componentId,
                                 }),
-                                filter,
+                                searchText,
                             });
-                            return paths?.map(HierarchyFilteringPath.normalize);
+                            return paths?.map(HierarchySearchPath.normalize);
                         },
                         getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),
                         handlePaths: async (paths) => handlePaths(paths, imodelAccess),
                     });
                 }
                 catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyFilterMatches" : "unknownFilterError";
-                    if (newError !== "tooManyFilterMatches") {
+                    const newError = e instanceof SearchLimitExceededError ? "tooManySearchMatches" : "unknownSearchError";
+                    if (newError !== "tooManySearchMatches") {
                         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);
+                    setSearchError(newError);
                     return [];
                 }
             };
         }
-        if (filter) {
+        if (searchText) {
             return async ({ imodelAccess, abortSignal }) => {
-                onFeatureUsed({ featureId: "filtering", reportInteraction: true });
+                onFeatureUsed({ featureId: "search", reportInteraction: true });
                 try {
-                    return await createFilteringPathsResult({
-                        getFilteringPaths: async () => {
+                    return await createSearchPathsResult({
+                        getSearchPaths: async () => {
                             const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
                                 imodelAccess,
-                                label: filter,
+                                label: searchText,
                                 idsCache: getModelsTreeIdsCache(),
                                 hierarchyConfig: hierarchyConfiguration,
                                 abortSignal,
@@ -147,32 +147,32 @@ export function useFilteredPaths({ hierarchyConfiguration, filter, getFilteredPa
                     });
                 }
                 catch (e) {
-                    const newError = e instanceof FilterLimitExceededError ? "tooManyFilterMatches" : "unknownFilterError";
-                    if (newError !== "tooManyFilterMatches") {
+                    const newError = e instanceof SearchLimitExceededError ? "tooManySearchMatches" : "unknownSearchError";
+                    if (newError !== "tooManySearchMatches") {
                         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);
+                    setSearchError(newError);
                     return [];
                 }
             };
         }
         return getSubTreePathsInternal;
     }, [
-        filter,
+        searchText,
         loadFocusedItems,
         getModelsTreeIdsCache,
         onFeatureUsed,
-        getFilteredPaths,
+        getSearchPaths,
         hierarchyConfiguration,
         onModelsFiltered,
-        onFilteredPathsChanged,
+        onSearchPathsChanged,
         getSubTreePathsInternal,
         componentId,
     ]);
     return {
         getPaths,
-        filteringError,
+        searchError,
         subTreeError,
     };
 }
@@ -194,7 +194,7 @@ async function getModels(paths, idsCache, classInspector) {
                 targetSubjectIds.add(currStep.id);
                 break;
             }
-            // collect all the models from the filtered path
+            // collect all the models from the search path
             if (await classInspector.classDerivesFrom(currStep.className, CLASS_NAME_GeometricModel3d)) {
                 targetModelIds.add(currStep.id);
             }
@@ -231,18 +231,18 @@ async function collectFocusedItems(loadFocusedItems) {
     })));
     return focusedItems;
 }
-async function createFilteringPathsResult({ getSubTreePaths, getFilteringPaths, handlePaths, }) {
-    const [subTreePaths, filterPaths] = await Promise.all([getSubTreePaths(), getFilteringPaths()]);
+async function createSearchPathsResult({ getSubTreePaths, getSearchPaths, handlePaths, }) {
+    const [subTreePaths, searchPaths] = await Promise.all([getSubTreePaths(), getSearchPaths()]);
     let joinedPaths;
     try {
-        if (subTreePaths && filterPaths) {
-            return (joinedPaths = joinHierarchyFilteringPaths(subTreePaths, filterPaths));
+        if (subTreePaths && searchPaths) {
+            return (joinedPaths = joinHierarchySearchPaths(subTreePaths, searchPaths));
         }
         if (subTreePaths) {
             return (joinedPaths = subTreePaths);
         }
-        if (filterPaths) {
-            return (joinedPaths = filterPaths);
+        if (searchPaths) {
+            return (joinedPaths = searchPaths);
         }
     }
     finally {
@@ -250,4 +250,4 @@ async function createFilteringPathsResult({ getSubTreePaths, getFilteringPaths,
     }
     return joinedPaths;
 }
-//# sourceMappingURL=UseFilteredPaths.js.map
+//# sourceMappingURL=UseSearchPaths.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"UseSearchPaths.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/UseSearchPaths.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AACjH,OAAO,EAAE,0BAA0B,EAAE,MAAM,yCAAyC,CAAC;AACrF,OAAO,EAAE,2BAA2B,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AAChH,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAelE,gBAAgB;AAChB,MAAM,UAAU,cAAc,CAAC,EAC7B,sBAAsB,EACtB,UAAU,EACV,cAAc,EACd,eAAe,EACf,qBAAqB,EACrB,gBAAgB,EAChB,oBAAoB,EACpB,WAAW,GAkBZ;IAKC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAoC,SAAS,CAAC,CAAC;IAC7F,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAqC,SAAS,CAAC,CAAC;IAEhG,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAChD,MAAM,EAAE,gBAAgB,EAAE,GAAG,0BAA0B,EAAE,CAAC;IAE1D,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,CAAC,SAAS,CAAC,CAAC;QAC1B,eAAe,CAAC,SAAS,CAAC,CAAC;QAC3B,gBAAgB,EAAE,CAAC,SAAS,CAAC,CAAC;QAE9B,yHAAyH;QACzH,IAAI,CAAC,gBAAgB,IAAI,CAAC,cAAc,IAAI,CAAC,UAAU,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5E,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,gBAAgB,EAAE,cAAc,EAAE,eAAe,EAAE,UAAU,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAE5G,MAAM,uBAAuB,GAAG,OAAO,CAErC,GAAG,EAAE;QACL,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;YAC7C,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;oBAClC,sBAAsB,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAChD,oBAAoB,CAAC,sBAAsB,CAAC;wBAC1C,YAAY;wBACZ,WAAW;wBACX,QAAQ,EAAE,qBAAqB,EAAE;wBACjC,eAAe,EAAE,sBAAsB;wBACvC,KAAK,EAAE,WAAW;wBAClB,WAAW;wBACX,WAAW,EAAE,GAAG,WAAW,UAAU;qBACtC,CAAC;iBACL,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,qBAAqB,CAAC;gBACvC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAC1B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC;IAElF,MAAM,QAAQ,GAAG,OAAO,CAAoD,GAAG,EAAE;QAC/E,MAAM,WAAW,GAAG,KAAK,EAAE,WAA8C,EAAE,cAAyC,EAAE,EAAE;YACtH,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,SAAS,CAAC,WAAW,EAAE,qBAAqB,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACjH,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;oBACjE,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,QAAQ,EAAE,qBAAqB,EAAE;gCACjC,WAAW,EAAE,YAAY;gCACzB,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,2BAA2B,CAAC;oBACjH,IAAI,QAAQ,KAAK,yBAAyB,EAAE,CAAC;wBAC3C,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;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC;gCACjC,sBAAsB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CACtC,oBAAoB,CAAC,sBAAsB,CAAC;oCAC1C,GAAG,KAAK;oCACR,YAAY;oCACZ,QAAQ,EAAE,qBAAqB,EAAE;oCACjC,eAAe,EAAE,sBAAsB;oCACvC,KAAK,EAAE,WAAW;oCAClB,WAAW;oCACX,WAAW;iCACZ,CAAC;gCACJ,UAAU;6BACX,CAAC,CAAC;4BACH,OAAO,KAAK,EAAE,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;wBACnD,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,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;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,aAAa,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;gBAChE,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,KAAK,EAAE,UAAU;gCACjB,QAAQ,EAAE,qBAAqB,EAAE;gCACjC,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,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;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,uBAAuB,CAAC;IACjC,CAAC,EAAE;QACD,UAAU;QACV,gBAAgB;QAChB,qBAAqB;QACrB,aAAa;QACb,cAAc;QACd,sBAAsB;QACtB,gBAAgB;QAChB,oBAAoB;QACpB,uBAAuB;QACvB,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;QACR,WAAW;QACX,YAAY;KACb,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAA4B,EAAE,QAA4B,EAAE,cAAyC;IAC5H,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,IAAI,GAAG,EAAc,CAAC;IAC7C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,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,CAAC;YACzC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChE,MAAM;YACR,CAAC;YAED,sEAAsE;YACtE,IAAI,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,SAAS,KAAK,kBAAkB,EAAE,CAAC;gBAC3E,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,8CAA8C;YAC9C,IAAI,MAAM,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,2BAA2B,CAAC,EAAE,CAAC;gBAC3F,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC3F,OAAO,CAAC,GAAG,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;AAChD,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,CAAC;QAC3C,IAAI,IAAI,IAAI,GAAG,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,SAAS;QACX,CAAC;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,CAAC;YAC7G,SAAS;QACX,CAAC;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;IAClF,CAAC;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,KAAK,UAAU,uBAAuB,CAAC,EACrC,eAAe,EACf,cAAc,EACd,WAAW,GAKZ;IACC,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IAC7F,IAAI,WAA8C,CAAC;IACnD,IAAI,CAAC;QACH,IAAI,YAAY,IAAI,WAAW,EAAE,CAAC;YAChC,OAAO,CAAC,WAAW,GAAG,wBAAwB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,WAAW,GAAG,YAAY,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;YAAS,CAAC;QACT,KAAK,WAAW,CAAC,WAAW,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,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 { firstValueFrom } from \"rxjs\";\nimport { HierarchyNodeIdentifier, HierarchyNodeKey, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport { useFocusedInstancesContext } from \"../../common/FocusedInstancesContext.js\";\nimport { CLASS_NAME_GeometricModel3d, CLASS_NAME_Subject } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { SearchLimitExceededError } from \"../../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../../common/UseTelemetryContext.js\";\nimport { joinHierarchySearchPaths } from \"../../common/Utils.js\";\nimport { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\n\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchyNodeIdentifiersPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../common/components/VisibilityTree.js\";\nimport type { NormalizedHierarchySearchPath } from \"../../common/Utils.js\";\nimport type { ClassGroupingHierarchyNode, ElementsGroupInfo, ModelsTreeHierarchyConfiguration } from \"../ModelsTreeDefinition.js\";\nimport type { ModelsTreeIdsCache } from \"./ModelsTreeIdsCache.js\";\n\n/** @internal */\nexport type ModelsTreeSearchError = \"tooManySearchMatches\" | \"tooManyInstancesFocused\" | \"unknownSearchError\" | \"unknownInstanceFocusError\";\n/** @internal */\nexport type ModelsTreeSubTreeError = \"unknownSubTreeError\";\n\n/** @internal */\nexport function useSearchPaths({\n  hierarchyConfiguration,\n  searchText,\n  getSearchPaths,\n  getSubTreePaths,\n  getModelsTreeIdsCache,\n  onModelsFiltered,\n  onSearchPathsChanged,\n  componentId,\n}: {\n  hierarchyConfiguration: ModelsTreeHierarchyConfiguration;\n  searchText?: string;\n  getSearchPaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys or node label. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> } | { label: string }) => Promise<NormalizedHierarchySearchPath[]>;\n    /** Search text which would be used to create search paths if `getSearchPaths` wouldn't be provided. */\n    searchText?: string;\n  }) => Promise<HierarchySearchPath[] | undefined>;\n  getSubTreePaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> }) => Promise<NormalizedHierarchySearchPath[]>;\n  }) => Promise<HierarchySearchPath[]>;\n  getModelsTreeIdsCache: () => ModelsTreeIdsCache;\n  onModelsFiltered?: (modelIds: Id64String[] | undefined) => void;\n  onSearchPathsChanged: (paths: HierarchySearchPath[] | undefined) => void;\n  componentId: GuidString;\n}): {\n  getPaths: VisibilityTreeProps[\"getSearchPaths\"] | undefined;\n  searchError: ModelsTreeSearchError | undefined;\n  subTreeError: ModelsTreeSubTreeError | undefined;\n} {\n  const [searchError, setSearchError] = useState<ModelsTreeSearchError | undefined>(undefined);\n  const [subTreeError, setSubTreeError] = useState<ModelsTreeSubTreeError | undefined>(undefined);\n\n  const { onFeatureUsed } = useTelemetryContext();\n  const { loadFocusedItems } = useFocusedInstancesContext();\n\n  useEffect(() => {\n    setSearchError(undefined);\n    setSubTreeError(undefined);\n    onModelsFiltered?.(undefined);\n\n    // reset search paths if there is no search applied. This allows to keep current search paths until new paths are loaded.\n    if (!loadFocusedItems && !getSearchPaths && !searchText && !getSubTreePaths) {\n      onSearchPathsChanged(undefined);\n    }\n  }, [loadFocusedItems, getSearchPaths, getSubTreePaths, searchText, onModelsFiltered, onSearchPathsChanged]);\n\n  const getSubTreePathsInternal = useMemo<\n    ((...props: Parameters<Required<VisibilityTreeProps>[\"getSearchPaths\"]>) => Promise<HierarchyNodeIdentifiersPath[]>) | undefined\n  >(() => {\n    if (!getSubTreePaths) {\n      return undefined;\n    }\n    return async ({ imodelAccess, abortSignal }) => {\n      try {\n        const paths = await getSubTreePaths({\n          createInstanceKeyPaths: async ({ targetItems }) =>\n            ModelsTreeDefinition.createInstanceKeyPaths({\n              imodelAccess,\n              targetItems,\n              idsCache: getModelsTreeIdsCache(),\n              hierarchyConfig: hierarchyConfiguration,\n              limit: \"unbounded\",\n              abortSignal,\n              componentId: `${componentId}/subTree`,\n            }),\n        });\n        return paths.map(HierarchySearchPath.normalize).map(({ path }) => path);\n      } catch {\n        const newError = \"unknownSubTreeError\";\n        setSubTreeError(newError);\n        return [];\n      }\n    };\n  }, [getModelsTreeIdsCache, hierarchyConfiguration, getSubTreePaths, componentId]);\n\n  const getPaths = useMemo<VisibilityTreeProps[\"getSearchPaths\"] | undefined>(() => {\n    const handlePaths = async (searchPaths: HierarchySearchPath[] | undefined, classInspector: ECClassHierarchyInspector) => {\n      onSearchPathsChanged(searchPaths);\n      if (!onModelsFiltered) {\n        return;\n      }\n\n      const modelIds = searchPaths ? await getModels(searchPaths, getModelsTreeIdsCache(), classInspector) : undefined;\n      onModelsFiltered(modelIds);\n    };\n\n    if (loadFocusedItems) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          const focusedItems = await collectFocusedItems(loadFocusedItems);\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                idsCache: getModelsTreeIdsCache(),\n                targetItems: focusedItems,\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"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          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (getSearchPaths) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await getSearchPaths({\n                createInstanceKeyPaths: async (props) =>\n                  ModelsTreeDefinition.createInstanceKeyPaths({\n                    ...props,\n                    imodelAccess,\n                    idsCache: getModelsTreeIdsCache(),\n                    hierarchyConfig: hierarchyConfiguration,\n                    limit: \"unbounded\",\n                    abortSignal,\n                    componentId,\n                  }),\n                searchText,\n              });\n              return paths?.map(HierarchySearchPath.normalize);\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\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          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (searchText) {\n      return async ({ imodelAccess, abortSignal }) => {\n        onFeatureUsed({ featureId: \"search\", reportInteraction: true });\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                label: searchText,\n                idsCache: getModelsTreeIdsCache(),\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\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          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n    return getSubTreePathsInternal;\n  }, [\n    searchText,\n    loadFocusedItems,\n    getModelsTreeIdsCache,\n    onFeatureUsed,\n    getSearchPaths,\n    hierarchyConfiguration,\n    onModelsFiltered,\n    onSearchPathsChanged,\n    getSubTreePathsInternal,\n    componentId,\n  ]);\n\n  return {\n    getPaths,\n    searchError,\n    subTreeError,\n  };\n}\n\nasync function getModels(paths: HierarchySearchPath[], idsCache: ModelsTreeIdsCache, classInspector: ECClassHierarchyInspector) {\n  if (!paths) {\n    return undefined;\n  }\n\n  const targetModelIds = new Set<Id64String>();\n  const targetSubjectIds = 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 === CLASS_NAME_Subject) {\n        targetSubjectIds.add(currStep.id);\n        break;\n      }\n\n      // collect all the models from the search path\n      if (await classInspector.classDerivesFrom(currStep.className, CLASS_NAME_GeometricModel3d)) {\n        targetModelIds.add(currStep.id);\n      }\n    }\n  }\n\n  const matchingModels = await firstValueFrom(idsCache.getSubjectModelIds(targetSubjectIds));\n  return [...targetModelIds, ...matchingModels];\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: Id64Array;\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\nasync function createSearchPathsResult({\n  getSubTreePaths,\n  getSearchPaths,\n  handlePaths,\n}: {\n  getSubTreePaths: () => Promise<HierarchyNodeIdentifiersPath[] | undefined>;\n  getSearchPaths: () => Promise<NormalizedHierarchySearchPath[] | undefined>;\n  handlePaths: (searchPaths: HierarchySearchPath[] | undefined) => Promise<void>;\n}): Promise<HierarchySearchPath[] | undefined> {\n  const [subTreePaths, searchPaths] = await Promise.all([getSubTreePaths(), getSearchPaths()]);\n  let joinedPaths: HierarchySearchPath[] | undefined;\n  try {\n    if (subTreePaths && searchPaths) {\n      return (joinedPaths = joinHierarchySearchPaths(subTreePaths, searchPaths));\n    }\n    if (subTreePaths) {\n      return (joinedPaths = subTreePaths);\n    }\n    if (searchPaths) {\n      return (joinedPaths = searchPaths);\n    }\n  } finally {\n    void handlePaths(joinedPaths);\n  }\n  return joinedPaths;\n}\n"]}

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

@@ -2,14 +2,14 @@ import { HierarchyNode } from "@itwin/presentation-hierarchies";
 import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
 import type { Observable } from "rxjs";
 import type { Id64Arg } from "@itwin/core-bentley";
-import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { GroupingHierarchyNode, HierarchySearchPath } from "@itwin/presentation-hierarchies";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
 import type { AlwaysAndNeverDrawnElementInfo } from "../../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
 import type { BaseTreeVisibilityHandlerOverrides, TreeSpecificVisibilityHandler } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
 import type { TreeWidgetViewport } from "../../../common/TreeWidgetViewport.js";
 import type { HierarchyVisibilityHandlerOverridableMethod, HierarchyVisibilityOverrideHandler, VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
 import type { ModelsTreeIdsCache } from "../ModelsTreeIdsCache.js";
-import type { ModelsTreeFilterTargets } from "./FilteredTree.js";
+import type { ModelsTreeSearchTargets } from "./SearchResultsTree.js";
 /**
  * Functionality of Models tree visibility handler that can be overridden.
  * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.
@@ -45,15 +45,15 @@ export interface ModelsTreeVisibilityHandlerProps {
  * It knows how to get and change visibility status of nodes created by hierarchy definition.
  * @internal
  */
-export declare class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeFilterTargets> {
+export declare class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeSearchTargets> {
     #private;
     constructor(constructorProps: ModelsTreeVisibilityHandlerProps);
     [Symbol.dispose](): void;
-    changeFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets, on: boolean): Observable<void>;
+    changeSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets, on: boolean): Observable<void>;
     getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus>;
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void>;
-    getFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets): Observable<VisibilityStatus>;
+    getSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets): Observable<VisibilityStatus>;
     private getCategories;
     private getAllCategories;
     private getElementsCount;
@@ -71,6 +71,6 @@ export declare function createModelsTreeVisibilityHandler(props: {
     idsCache: ModelsTreeIdsCache;
     imodelAccess: ECClassHierarchyInspector;
     overrides?: ModelsTreeVisibilityHandlerOverrides;
-    filteredPaths?: HierarchyFilteringPath[];
-}): HierarchyVisibilityHandlerImpl<ModelsTreeFilterTargets>;
+    searchPaths?: HierarchySearchPath[];
+}): HierarchyVisibilityHandlerImpl<ModelsTreeSearchTargets>;
 //# sourceMappingURL=ModelsTreeVisibilityHandler.d.ts.map

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

@@ -3,15 +3,15 @@
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import { concat, defer, EMPTY, from, map, merge, mergeMap, of } from "rxjs";
-import { assert, Id64 } from "@itwin/core-bentley";
+import { assert, Guid, Id64 } from "@itwin/core-bentley";
 import { HierarchyNode } from "@itwin/presentation-hierarchies";
 import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
 import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
-import { releaseMainThreadOnItemsCount } from "../../../common/internal/Utils.js";
+import { fromWithRelease } from "../../../common/internal/Utils.js";
 import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
 import { ModelsTreeNode } from "../ModelsTreeNode.js";
-import { createFilteredModelsTree } from "./FilteredTree.js";
 import { ModelsTreeVisibilityHelper } from "./ModelsTreeVisibilityHelper.js";
+import { createModelsSearchResultsTree } from "./SearchResultsTree.js";
 /**
  * Handles visibility status of models tree nodes.
  *
@@ -46,7 +46,7 @@ export class ModelsTreeVisibilityHandler {
     [Symbol.dispose]() {
         this.#visibilityHelper[Symbol.dispose]();
     }
-    changeFilterTargetsVisibilityStatus(targets, on) {
+    changeSearchTargetsVisibilityStatus(targets, on) {
         return defer(() => {
             const { subjectIds, modelIds, categories, elements } = targets;
             const observables = new Array();
@@ -64,7 +64,7 @@ export class ModelsTreeVisibilityHandler {
                 }))));
             }
             if (elements?.length) {
-                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }))));
+                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }))));
             }
             return merge(...observables);
         });
@@ -171,7 +171,7 @@ export class ModelsTreeVisibilityHandler {
         }
         return changeObs;
     }
-    getFilterTargetsVisibilityStatus(targets) {
+    getSearchTargetsVisibilityStatus(targets) {
         return defer(() => {
             const { subjectIds, modelIds, categories, elements } = targets;
             const observables = new Array();
@@ -189,7 +189,7 @@ export class ModelsTreeVisibilityHandler {
                 }))));
             }
             if (elements?.length) {
-                observables.push(from(elements).pipe(releaseMainThreadOnItemsCount(50), mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: "GeometricElement3d" }))));
+                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: "GeometricElement3d" }))));
             }
             return merge(...observables);
         }).pipe(mergeVisibilityStatuses);
@@ -207,7 +207,7 @@ export class ModelsTreeVisibilityHandler {
         return this.#props.idsCache.getCategoriesElementModels(props.categoryIds);
     }
     getSubCategories(props) {
-        return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));
+        return this.#props.idsCache.getSubCategories(props.categoryId);
     }
     getSubModels(props) {
         if ("modelIds" in props) {
@@ -242,12 +242,12 @@ export class ModelsTreeVisibilityHandler {
  */
 export function createModelsTreeVisibilityHandler(props) {
     return new HierarchyVisibilityHandlerImpl({
-        getFilteredTree: () => {
-            if (!props.filteredPaths) {
+        getSearchResultsTree: () => {
+            if (!props.searchPaths) {
                 return undefined;
             }
-            return createFilteredModelsTree({
-                filteringPaths: props.filteredPaths,
+            return createModelsSearchResultsTree({
+                searchPaths: props.searchPaths,
                 imodelAccess: props.imodelAccess,
             });
         },
@@ -261,7 +261,7 @@ export function createModelsTreeVisibilityHandler(props) {
             });
         },
         viewport: props.viewport,
-        componentId: "Test",
+        componentId: Guid.createValue(),
     });
 }
 //# sourceMappingURL=ModelsTreeVisibilityHandler.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"ModelsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/visibility/ModelsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,8BAA8B,EAAE,MAAM,8DAA8D,CAAC;AAC9G,OAAO,EAAE,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AAClF,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AA8C7E;;;;;GAKG;AACH,MAAM,OAAO,2BAA2B;IACtC,iBAAiB,CAA6B;IACrC,MAAM,CAAmC;IAElD,YAAY,gBAAkD;QAC5D,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,6EAA6E;QAC7E,mDAAmD;QACnD,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YACnD,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAC3C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACjD,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;YAC/D,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE;SAChD,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,0BAA0B,CAAC;YACtD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YAC5C,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;SACjC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAgC,EAAE,EAAW;QACtF,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC9F,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBACtD,WAAW;oBACX,OAAO;oBACP,EAAE;iBACH,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACxH,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBACvE,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;gBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE;gBACvB,mBAAmB,EAAE,MAAM;gBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,sCAAsC;aACxE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,4EAA4E;YAC5E,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACxH,CAAC;QAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC7I,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;gBAC1D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACtD,OAAO;gBACP,IAAI,EAAE,iBAAiB;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACrD,OAAO;YACP,UAAU;YACV,IAAI,EAAE,oBAAoB;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAC1E,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,EAAE;iBACH,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;oBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;oBAC3B,mBAAmB,EAAE,MAAM;oBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,yCAAyC;iBAC3E,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;oBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1H,CAAC;YAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBAC7D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;oBACtD,OAAO;oBACP,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,KAAK,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO;gBACP,UAAU;gBACV,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAgC;QACtE,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YACvF,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC7G,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;oBACnD,WAAW;oBACX,OAAO;oBACP,IAAI,EAAE,iBAAiB;iBACxB,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjB,6BAA6B,CAAC,EAAE,CAAC,EACjC,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC7I,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnC,CAAC;IAEO,aAAa,CAAC,KAAmD;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CACvJ,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAChH,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAEO,SAAS,CAAC,KAA+C;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC5E,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1H,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAC5E,QAAQ,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EACrF,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CACxC,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,IAA2B;QACrD,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC;QAElC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;IAC7C,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,iCAAiC,CAAC,KAMjD;IACC,OAAO,IAAI,8BAA8B,CAA0B;QACjE,eAAe,EAAE,GAA+D,EAAE;YAChF,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;gBACzB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,wBAAwB,CAAC;gBAC9B,cAAc,EAAE,KAAK,CAAC,aAAa;gBACnC,YAAY,EAAE,KAAK,CAAC,YAAY;aACjC,CAAC,CAAC;QACL,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE;YAC1D,OAAO,IAAI,2BAA2B,CAAC;gBACrC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe;gBACf,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,MAAM;KACpB,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, defer, EMPTY, from, map, merge, mergeMap, of } from \"rxjs\";\nimport { assert, Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { HierarchyVisibilityHandlerImpl } from \"../../../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { releaseMainThreadOnItemsCount } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ModelsTreeNode } from \"../ModelsTreeNode.js\";\nimport { createFilteredModelsTree } from \"./FilteredTree.js\";\nimport { ModelsTreeVisibilityHelper } from \"./ModelsTreeVisibilityHelper.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchyFilteringPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { FilteredTree } from \"../../../common/internal/visibility/BaseFilteredTree.js\";\nimport type {\n  BaseIdsCache,\n  BaseTreeVisibilityHandlerOverrides,\n  TreeSpecificVisibilityHandler,\n} from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type {\n  HierarchyVisibilityHandlerOverridableMethod,\n  HierarchyVisibilityOverrideHandler,\n  VisibilityStatus,\n} from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ModelsTreeIdsCache } from \"../ModelsTreeIdsCache.js\";\nimport type { ModelsTreeFilterTargets } from \"./FilteredTree.js\";\n\n/**\n * Functionality of Models tree visibility handler that can be overridden.\n * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.\n * @beta\n */\nexport interface ModelsTreeVisibilityHandlerOverrides extends BaseTreeVisibilityHandlerOverrides {\n  getSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg }) => Promise<VisibilityStatus>>;\n  getElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { node: GroupingHierarchyNode }) => Promise<VisibilityStatus>>;\n\n  changeSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg; on: boolean }) => Promise<void>>;\n  changeElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<\n    (props: { node: GroupingHierarchyNode; on: boolean }) => Promise<void>\n  >;\n}\n\n/** @internal */\nexport interface ModelsTreeVisibilityHandlerProps {\n  idsCache: ModelsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n}\n\n/**\n * Handles visibility status of models tree nodes.\n *\n * It knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeFilterTargets> {\n  #visibilityHelper: ModelsTreeVisibilityHelper;\n  readonly #props: ModelsTreeVisibilityHandlerProps;\n\n  constructor(constructorProps: ModelsTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.\n    // We won't need to create a custom base ids cache.\n    const baseIdsCache: BaseIdsCache = {\n      getCategories: (props) => this.getCategories(props),\n      getElementsCount: (props) => this.getElementsCount(props),\n      getModels: (props) => this.getModels(props),\n      getSubCategories: (props) => this.getSubCategories(props),\n      getSubModels: (props) => this.getSubModels(props),\n      hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),\n      getAllCategories: () => this.getAllCategories(),\n    };\n    this.#visibilityHelper = new ModelsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      overrideHandler: this.#props.overrideHandler,\n      baseIdsCache,\n      overrides: this.#props.overrides,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.changeSubjectsVisibilityStatus({ subjectIds, on }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.changeCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                on,\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          from(elements).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      const nodeInfo = this.getGroupingNodeInfo(node);\n      const result = this.#visibilityHelper.getGroupedElementsVisibilityStatus({\n        categoryId: nodeInfo.categoryId,\n        modelId: nodeInfo.modelId,\n        elementIds: nodeInfo.elementIds,\n      });\n      return this.#props.overrideHandler.createVisibilityHandlerResult({\n        overrideProps: { node },\n        nonOverriddenResult: result,\n        override: this.#props.overrides?.getElementGroupingNodeVisibilityStatus,\n      });\n    }\n\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isSubjectNode(node)) {\n      // note: subject nodes may be merged to represent multiple subject instances\n      return this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds: node.key.instanceKeys.map((key) => key.id) });\n    }\n\n    if (ModelsTreeNode.isModelNode(node)) {\n      return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), type: \"GeometricModel3d\" });\n    }\n\n    const modelId = ModelsTreeNode.getModelId(node);\n    if (!modelId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isCategoryNode(node)) {\n      return this.#visibilityHelper.getCategoriesVisibilityStatus({\n        categoryIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        type: \"SpatialCategory\",\n      });\n    }\n\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    if (!categoryId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map(({ id }) => id),\n      modelId,\n      categoryId,\n      type: \"GeometricElement3d\",\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (HierarchyNode.isClassGroupingNode(node)) {\n        const nodeInfo = this.getGroupingNodeInfo(node);\n        const result = this.#visibilityHelper.changeGroupedElementsVisibilityStatus({\n          categoryId: nodeInfo.categoryId,\n          modelId: nodeInfo.modelId,\n          elementIds: nodeInfo.elementIds,\n          on,\n        });\n        return this.#props.overrideHandler.createVisibilityHandlerResult({\n          overrideProps: { node, on },\n          nonOverriddenResult: result,\n          override: this.#props.overrides?.changeElementGroupingNodeVisibilityStatus,\n        });\n      }\n\n      if (!HierarchyNode.isInstancesNode(node)) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isSubjectNode(node)) {\n        return this.#visibilityHelper.changeSubjectsVisibilityStatus({\n          subjectIds: node.key.instanceKeys.map((key) => key.id),\n          on,\n        });\n      }\n\n      if (ModelsTreeNode.isModelNode(node)) {\n        return this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), on });\n      }\n\n      const modelId = ModelsTreeNode.getModelId(node);\n      if (!modelId) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isCategoryNode(node)) {\n        return this.#visibilityHelper.changeCategoriesVisibilityStatus({\n          categoryIds: node.key.instanceKeys.map(({ id }) => id),\n          modelId,\n          on,\n        });\n      }\n\n      const categoryId = ModelsTreeNode.getCategoryId(node);\n      if (!categoryId) {\n        return EMPTY;\n      }\n\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        categoryId,\n        on,\n      });\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getFilterTargetsVisibilityStatus(targets: ModelsTreeFilterTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: \"GeometricModel3d\" }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.getCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                type: \"SpatialCategory\",\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          from(elements).pipe(\n            releaseMainThreadOnItemsCount(50),\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: \"GeometricElement3d\" }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    }).pipe(mergeVisibilityStatuses);\n  }\n\n  private getCategories(props: Parameters<BaseIdsCache[\"getCategories\"]>[0]): ReturnType<BaseIdsCache[\"getCategories\"]> {\n    return from(Id64.iterable(props.modelIds)).pipe(\n      mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categoryIds) => ({ id: modelId, spatialCategories: categoryIds })))),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(map((categories) => ({ spatialCategories: categories })));\n  }\n\n  private getElementsCount(props: Parameters<BaseIdsCache[\"getElementsCount\"]>[0]): ReturnType<BaseIdsCache[\"getElementsCount\"]> {\n    return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);\n  }\n\n  private getModels(props: Parameters<BaseIdsCache[\"getModels\"]>[0]): ReturnType<BaseIdsCache[\"getModels\"]> {\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds);\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return from(Id64.iterable(props.categoryIds)).pipe(map((categoryId) => ({ id: categoryId, subCategories: undefined })));\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return this.#props.idsCache.getModelCategoryIds(modelId).pipe(\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(\n      mergeMap(({ id, models }) => {\n        if (!models) {\n          return of({ id, subModels: undefined });\n        }\n        return from(models).pipe(\n          mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)),\n          map((subModels) => ({ id, subModels })),\n        );\n      }),\n    );\n  }\n\n  private getGroupingNodeInfo(node: GroupingHierarchyNode) {\n    const modelId = ModelsTreeNode.getModelId(node);\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    assert(!!modelId && !!categoryId);\n\n    const elementIds = node.groupedInstanceKeys.map((key) => key.id);\n    return { modelId, categoryId, elementIds };\n  }\n}\n\n/**\n * Creates models tree visibility handler. Is used by integration and performance tests.\n * @internal\n */\nexport function createModelsTreeVisibilityHandler(props: {\n  viewport: TreeWidgetViewport;\n  idsCache: ModelsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n  filteredPaths?: HierarchyFilteringPath[];\n}) {\n  return new HierarchyVisibilityHandlerImpl<ModelsTreeFilterTargets>({\n    getFilteredTree: (): undefined | Promise<FilteredTree<ModelsTreeFilterTargets>> => {\n      if (!props.filteredPaths) {\n        return undefined;\n      }\n      return createFilteredModelsTree({\n        filteringPaths: props.filteredPaths,\n        imodelAccess: props.imodelAccess,\n      });\n    },\n    getTreeSpecificVisibilityHandler: (info, overrideHandler) => {\n      return new ModelsTreeVisibilityHandler({\n        alwaysAndNeverDrawnElementInfo: info,\n        idsCache: props.idsCache,\n        viewport: props.viewport,\n        overrideHandler,\n        overrides: props.overrides,\n      });\n    },\n    viewport: props.viewport,\n    componentId: \"Test\",\n  });\n}\n"]}
+{"version":3,"file":"ModelsTreeVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/visibility/ModelsTreeVisibilityHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,EAAE,8BAA8B,EAAE,MAAM,8DAA8D,CAAC;AAC9G,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,EAAE,6BAA6B,EAAE,MAAM,wBAAwB,CAAC;AA8CvE;;;;;GAKG;AACH,MAAM,OAAO,2BAA2B;IACtC,iBAAiB,CAA6B;IACrC,MAAM,CAAmC;IAElD,YAAY,gBAAkD;QAC5D,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC;QAC/B,6EAA6E;QAC7E,mDAAmD;QACnD,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YACnD,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YAC3C,gBAAgB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACzD,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACjD,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC;YAC/D,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE;SAChD,CAAC;QACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,0BAA0B,CAAC;YACtD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,8BAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B;YAC1E,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YAC5C,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;SACjC,CAAC,CAAC;IACL,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,mCAAmC,CAAC,OAAgC,EAAE,EAAW;QACtF,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAoB,CAAC;YAClD,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC9F,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBACtD,WAAW;oBACX,OAAO;oBACP,EAAE;iBACH,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CACxH,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,kCAAkC,CAAC;gBACvE,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;gBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE;gBACvB,mBAAmB,EAAE,MAAM;gBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,sCAAsC;aACxE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,4EAA4E;YAC5E,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACxH,CAAC;QAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC7I,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;gBAC1D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACtD,OAAO;gBACP,IAAI,EAAE,iBAAiB;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC;YACxD,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACrD,OAAO;YACP,UAAU;YACV,IAAI,EAAE,oBAAoB;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,oEAAoE;IAC7D,sBAAsB,CAAC,IAAmB,EAAE,EAAW;QAC5D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE;YAC3B,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,qCAAqC,CAAC;oBAC1E,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,EAAE;iBACH,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,6BAA6B,CAAC;oBAC/D,aAAa,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;oBAC3B,mBAAmB,EAAE,MAAM;oBAC3B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,yCAAyC;iBAC3E,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;oBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtD,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,IAAI,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,OAAO,IAAI,CAAC,iBAAiB,CAAC,4BAA4B,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1H,CAAC;YAED,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YAED,IAAI,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,gCAAgC,CAAC;oBAC7D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;oBACtD,OAAO;oBACP,EAAE;iBACH,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACtD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,KAAK,CAAC;YACf,CAAC;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,8BAA8B,CAAC;gBAC3D,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACrD,OAAO;gBACP,UAAU;gBACV,EAAE;aACH,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,EAAE,SAAS,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,gCAAgC,CAAC,OAAgC;QACtE,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;YAC/D,MAAM,WAAW,GAAG,IAAI,KAAK,EAAgC,CAAC;YAC9D,IAAI,UAAU,EAAE,IAAI,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YACvF,CAAC;YAED,IAAI,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YAC7G,CAAC;YAED,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CACd,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CACnB,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,IAAI,CAAC,iBAAiB,CAAC,6BAA6B,CAAC;oBACnD,WAAW;oBACX,OAAO;oBACP,IAAI,EAAE,iBAAiB;iBACxB,CAAC,CACH,CACF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,QAAQ,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,CAC1D,IAAI,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAC7I,CACF,CACF,CAAC;YACJ,CAAC;YAED,OAAO,KAAK,CAAC,GAAG,WAAW,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnC,CAAC;IAEO,aAAa,CAAC,KAAmD;QACvE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CACvJ,CAAC;IACJ,CAAC;IAEO,gBAAgB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAChH,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IAEO,SAAS,CAAC,KAA+C;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC5E,CAAC;IAEO,gBAAgB,CAAC,KAAsD;QAC7E,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IAEO,YAAY,CAAC,KAAkD;QACrE,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAC7C,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;gBACnB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC7I,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,CAC3D,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,EAClG,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CACjD,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAChD,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,KAAK,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CACxI,CACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAC5E,QAAQ,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;YAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1C,CAAC;YACD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EACrF,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,CACxC,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,IAA2B;QACrD,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACtD,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC;QAElC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;IAC7C,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,iCAAiC,CAAC,KAMjD;IACC,OAAO,IAAI,8BAA8B,CAA0B;QACjE,oBAAoB,EAAE,GAAoE,EAAE;YAC1F,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACvB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,6BAA6B,CAAC;gBACnC,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;aACjC,CAAC,CAAC;QACL,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE;YAC1D,OAAO,IAAI,2BAA2B,CAAC;gBACrC,8BAA8B,EAAE,IAAI;gBACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe;gBACf,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;KAChC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { concat, defer, EMPTY, from, map, merge, mergeMap, of } from \"rxjs\";\nimport { assert, Guid, Id64 } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { createVisibilityStatus } from \"../../../common/internal/Tooltip.js\";\nimport { HierarchyVisibilityHandlerImpl } from \"../../../common/internal/useTreeHooks/UseCachedVisibility.js\";\nimport { fromWithRelease } from \"../../../common/internal/Utils.js\";\nimport { mergeVisibilityStatuses } from \"../../../common/internal/VisibilityUtils.js\";\nimport { ModelsTreeNode } from \"../ModelsTreeNode.js\";\nimport { ModelsTreeVisibilityHelper } from \"./ModelsTreeVisibilityHelper.js\";\nimport { createModelsSearchResultsTree } from \"./SearchResultsTree.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Arg } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { AlwaysAndNeverDrawnElementInfo } from \"../../../common/internal/AlwaysAndNeverDrawnElementInfo.js\";\nimport type { SearchResultsTree } from \"../../../common/internal/visibility/BaseSearchResultsTree.js\";\nimport type {\n  BaseIdsCache,\n  BaseTreeVisibilityHandlerOverrides,\n  TreeSpecificVisibilityHandler,\n} from \"../../../common/internal/visibility/BaseVisibilityHelper.js\";\nimport type { TreeWidgetViewport } from \"../../../common/TreeWidgetViewport.js\";\nimport type {\n  HierarchyVisibilityHandlerOverridableMethod,\n  HierarchyVisibilityOverrideHandler,\n  VisibilityStatus,\n} from \"../../../common/UseHierarchyVisibility.js\";\nimport type { ModelsTreeIdsCache } from \"../ModelsTreeIdsCache.js\";\nimport type { ModelsTreeSearchTargets } from \"./SearchResultsTree.js\";\n\n/**\n * Functionality of Models tree visibility handler that can be overridden.\n * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.\n * @beta\n */\nexport interface ModelsTreeVisibilityHandlerOverrides extends BaseTreeVisibilityHandlerOverrides {\n  getSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg }) => Promise<VisibilityStatus>>;\n  getElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { node: GroupingHierarchyNode }) => Promise<VisibilityStatus>>;\n\n  changeSubjectsVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: { subjectIds: Id64Arg; on: boolean }) => Promise<void>>;\n  changeElementGroupingNodeVisibilityStatus?: HierarchyVisibilityHandlerOverridableMethod<\n    (props: { node: GroupingHierarchyNode; on: boolean }) => Promise<void>\n  >;\n}\n\n/** @internal */\nexport interface ModelsTreeVisibilityHandlerProps {\n  idsCache: ModelsTreeIdsCache;\n  viewport: TreeWidgetViewport;\n  alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n}\n\n/**\n * Handles visibility status of models tree nodes.\n *\n * It knows how to get and change visibility status of nodes created by hierarchy definition.\n * @internal\n */\nexport class ModelsTreeVisibilityHandler implements Disposable, TreeSpecificVisibilityHandler<ModelsTreeSearchTargets> {\n  #visibilityHelper: ModelsTreeVisibilityHelper;\n  readonly #props: ModelsTreeVisibilityHandlerProps;\n\n  constructor(constructorProps: ModelsTreeVisibilityHandlerProps) {\n    this.#props = constructorProps;\n    // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.\n    // We won't need to create a custom base ids cache.\n    const baseIdsCache: BaseIdsCache = {\n      getCategories: (props) => this.getCategories(props),\n      getElementsCount: (props) => this.getElementsCount(props),\n      getModels: (props) => this.getModels(props),\n      getSubCategories: (props) => this.getSubCategories(props),\n      getSubModels: (props) => this.getSubModels(props),\n      hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),\n      getAllCategories: () => this.getAllCategories(),\n    };\n    this.#visibilityHelper = new ModelsTreeVisibilityHelper({\n      viewport: this.#props.viewport,\n      idsCache: this.#props.idsCache,\n      alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,\n      overrideHandler: this.#props.overrideHandler,\n      baseIdsCache,\n      overrides: this.#props.overrides,\n    });\n  }\n\n  public [Symbol.dispose]() {\n    this.#visibilityHelper[Symbol.dispose]();\n  }\n\n  public changeSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets, on: boolean): Observable<void> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<void>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.changeSubjectsVisibilityStatus({ subjectIds, on }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds, on }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.changeCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                on,\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    });\n  }\n\n  public getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      const nodeInfo = this.getGroupingNodeInfo(node);\n      const result = this.#visibilityHelper.getGroupedElementsVisibilityStatus({\n        categoryId: nodeInfo.categoryId,\n        modelId: nodeInfo.modelId,\n        elementIds: nodeInfo.elementIds,\n      });\n      return this.#props.overrideHandler.createVisibilityHandlerResult({\n        overrideProps: { node },\n        nonOverriddenResult: result,\n        override: this.#props.overrides?.getElementGroupingNodeVisibilityStatus,\n      });\n    }\n\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isSubjectNode(node)) {\n      // note: subject nodes may be merged to represent multiple subject instances\n      return this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds: node.key.instanceKeys.map((key) => key.id) });\n    }\n\n    if (ModelsTreeNode.isModelNode(node)) {\n      return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), type: \"GeometricModel3d\" });\n    }\n\n    const modelId = ModelsTreeNode.getModelId(node);\n    if (!modelId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    if (ModelsTreeNode.isCategoryNode(node)) {\n      return this.#visibilityHelper.getCategoriesVisibilityStatus({\n        categoryIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        type: \"SpatialCategory\",\n      });\n    }\n\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    if (!categoryId) {\n      return of(createVisibilityStatus(\"disabled\"));\n    }\n\n    return this.#visibilityHelper.getElementsVisibilityStatus({\n      elementIds: node.key.instanceKeys.map(({ id }) => id),\n      modelId,\n      categoryId,\n      type: \"GeometricElement3d\",\n    });\n  }\n\n  /** Changes visibility of the items represented by the tree node. */\n  public changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void> {\n    const changeObs = defer(() => {\n      if (HierarchyNode.isClassGroupingNode(node)) {\n        const nodeInfo = this.getGroupingNodeInfo(node);\n        const result = this.#visibilityHelper.changeGroupedElementsVisibilityStatus({\n          categoryId: nodeInfo.categoryId,\n          modelId: nodeInfo.modelId,\n          elementIds: nodeInfo.elementIds,\n          on,\n        });\n        return this.#props.overrideHandler.createVisibilityHandlerResult({\n          overrideProps: { node, on },\n          nonOverriddenResult: result,\n          override: this.#props.overrides?.changeElementGroupingNodeVisibilityStatus,\n        });\n      }\n\n      if (!HierarchyNode.isInstancesNode(node)) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isSubjectNode(node)) {\n        return this.#visibilityHelper.changeSubjectsVisibilityStatus({\n          subjectIds: node.key.instanceKeys.map((key) => key.id),\n          on,\n        });\n      }\n\n      if (ModelsTreeNode.isModelNode(node)) {\n        return this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), on });\n      }\n\n      const modelId = ModelsTreeNode.getModelId(node);\n      if (!modelId) {\n        return EMPTY;\n      }\n\n      if (ModelsTreeNode.isCategoryNode(node)) {\n        return this.#visibilityHelper.changeCategoriesVisibilityStatus({\n          categoryIds: node.key.instanceKeys.map(({ id }) => id),\n          modelId,\n          on,\n        });\n      }\n\n      const categoryId = ModelsTreeNode.getCategoryId(node);\n      if (!categoryId) {\n        return EMPTY;\n      }\n\n      return this.#visibilityHelper.changeElementsVisibilityStatus({\n        elementIds: node.key.instanceKeys.map(({ id }) => id),\n        modelId,\n        categoryId,\n        on,\n      });\n    });\n\n    if (this.#props.viewport.isAlwaysDrawnExclusive) {\n      return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);\n    }\n    return changeObs;\n  }\n\n  public getSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets): Observable<VisibilityStatus> {\n    return defer(() => {\n      const { subjectIds, modelIds, categories, elements } = targets;\n      const observables = new Array<Observable<VisibilityStatus>>();\n      if (subjectIds?.size) {\n        observables.push(this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds }));\n      }\n\n      if (modelIds?.size) {\n        observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: \"GeometricModel3d\" }));\n      }\n\n      if (categories?.length) {\n        observables.push(\n          from(categories).pipe(\n            mergeMap(({ modelId, categoryIds }) =>\n              this.#visibilityHelper.getCategoriesVisibilityStatus({\n                categoryIds,\n                modelId,\n                type: \"SpatialCategory\",\n              }),\n            ),\n          ),\n        );\n      }\n\n      if (elements?.length) {\n        observables.push(\n          fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(\n            mergeMap(({ modelId, elements: elementsMap, categoryId }) =>\n              this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: \"GeometricElement3d\" }),\n            ),\n          ),\n        );\n      }\n\n      return merge(...observables);\n    }).pipe(mergeVisibilityStatuses);\n  }\n\n  private getCategories(props: Parameters<BaseIdsCache[\"getCategories\"]>[0]): ReturnType<BaseIdsCache[\"getCategories\"]> {\n    return from(Id64.iterable(props.modelIds)).pipe(\n      mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categoryIds) => ({ id: modelId, spatialCategories: categoryIds })))),\n    );\n  }\n\n  private getAllCategories(): ReturnType<BaseIdsCache[\"getAllCategories\"]> {\n    return this.#props.idsCache.getAllCategories().pipe(map((categories) => ({ spatialCategories: categories })));\n  }\n\n  private getElementsCount(props: Parameters<BaseIdsCache[\"getElementsCount\"]>[0]): ReturnType<BaseIdsCache[\"getElementsCount\"]> {\n    return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);\n  }\n\n  private getModels(props: Parameters<BaseIdsCache[\"getModels\"]>[0]): ReturnType<BaseIdsCache[\"getModels\"]> {\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds);\n  }\n\n  private getSubCategories(props: Parameters<BaseIdsCache[\"getSubCategories\"]>[0]): ReturnType<BaseIdsCache[\"getSubCategories\"]> {\n    return this.#props.idsCache.getSubCategories(props.categoryId);\n  }\n\n  private getSubModels(props: Parameters<BaseIdsCache[\"getSubModels\"]>[0]): ReturnType<BaseIdsCache[\"getSubModels\"]> {\n    if (\"modelIds\" in props) {\n      return from(Id64.iterable(props.modelIds)).pipe(\n        mergeMap((modelId) => {\n          if (props.categoryId) {\n            return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));\n          }\n          return this.#props.idsCache.getModelCategoryIds(modelId).pipe(\n            mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)),\n            map((subModels) => ({ id: modelId, subModels })),\n          );\n        }),\n      );\n    }\n\n    if (props.modelId) {\n      return from(Id64.iterable(props.categoryIds)).pipe(\n        mergeMap((categoryId) =>\n          this.#props.idsCache.getCategoriesModeledElements(props.modelId!, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels }))),\n        ),\n      );\n    }\n\n    return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(\n      mergeMap(({ id, models }) => {\n        if (!models) {\n          return of({ id, subModels: undefined });\n        }\n        return from(models).pipe(\n          mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)),\n          map((subModels) => ({ id, subModels })),\n        );\n      }),\n    );\n  }\n\n  private getGroupingNodeInfo(node: GroupingHierarchyNode) {\n    const modelId = ModelsTreeNode.getModelId(node);\n    const categoryId = ModelsTreeNode.getCategoryId(node);\n    assert(!!modelId && !!categoryId);\n\n    const elementIds = node.groupedInstanceKeys.map((key) => key.id);\n    return { modelId, categoryId, elementIds };\n  }\n}\n\n/**\n * Creates models tree visibility handler. Is used by integration and performance tests.\n * @internal\n */\nexport function createModelsTreeVisibilityHandler(props: {\n  viewport: TreeWidgetViewport;\n  idsCache: ModelsTreeIdsCache;\n  imodelAccess: ECClassHierarchyInspector;\n  overrides?: ModelsTreeVisibilityHandlerOverrides;\n  searchPaths?: HierarchySearchPath[];\n}) {\n  return new HierarchyVisibilityHandlerImpl<ModelsTreeSearchTargets>({\n    getSearchResultsTree: (): undefined | Promise<SearchResultsTree<ModelsTreeSearchTargets>> => {\n      if (!props.searchPaths) {\n        return undefined;\n      }\n      return createModelsSearchResultsTree({\n        searchPaths: props.searchPaths,\n        imodelAccess: props.imodelAccess,\n      });\n    },\n    getTreeSpecificVisibilityHandler: (info, overrideHandler) => {\n      return new ModelsTreeVisibilityHandler({\n        alwaysAndNeverDrawnElementInfo: info,\n        idsCache: props.idsCache,\n        viewport: props.viewport,\n        overrideHandler,\n        overrides: props.overrides,\n      });\n    },\n    viewport: props.viewport,\n    componentId: Guid.createValue(),\n  });\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/visibility/{FilteredTree.d.ts → SearchResultsTree.d.ts}

@@ -1,10 +1,10 @@
 import type { Id64Set, Id64String } from "@itwin/core-bentley";
-import type { HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
+import type { HierarchySearchPath } from "@itwin/presentation-hierarchies";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
 import type { ElementId } from "../../../common/internal/Types.js";
-import type { FilteredTree } from "../../../common/internal/visibility/BaseFilteredTree.js";
+import type { SearchResultsTree } from "../../../common/internal/visibility/BaseSearchResultsTree.js";
 /** @internal */
-export interface ModelsTreeFilterTargets {
+export interface ModelsTreeSearchTargets {
     subjectIds?: Id64Set;
     modelIds?: Id64Set;
     categories?: Array<{
@@ -15,13 +15,13 @@ export interface ModelsTreeFilterTargets {
         modelId: Id64String;
         categoryId: Id64String;
         elements: Map<ElementId, {
-            isFilterTarget: boolean;
+            isSearchTarget: boolean;
         }>;
     }>;
 }
 /** @internal */
-export declare function createFilteredModelsTree(props: {
+export declare function createModelsSearchResultsTree(props: {
     imodelAccess: ECClassHierarchyInspector;
-    filteringPaths: HierarchyFilteringPath[];
-}): Promise<FilteredTree<ModelsTreeFilterTargets>>;
-//# sourceMappingURL=FilteredTree.d.ts.map
+    searchPaths: HierarchySearchPath[];
+}): Promise<SearchResultsTree<ModelsTreeSearchTargets>>;
+//# sourceMappingURL=SearchResultsTree.d.ts.map